summaryrefslogtreecommitdiff
path: root/util/cbfstool
diff options
context:
space:
mode:
Diffstat (limited to 'util/cbfstool')
-rw-r--r--util/cbfstool/common.c16
-rw-r--r--util/cbfstool/common.h5
2 files changed, 18 insertions, 3 deletions
diff --git a/util/cbfstool/common.c b/util/cbfstool/common.c
index 539d0baccf..5889e2a937 100644
--- a/util/cbfstool/common.c
+++ b/util/cbfstool/common.c
@@ -49,7 +49,8 @@ int buffer_create(struct buffer *buffer, size_t size, const char *name)
return (buffer->data == NULL);
}
-int buffer_from_file(struct buffer *buffer, const char *filename)
+int buffer_from_file_aligned_size(struct buffer *buffer, const char *filename,
+ size_t size_granularity)
{
FILE *fp = fopen(filename, "rb");
if (!fp) {
@@ -63,20 +64,29 @@ int buffer_from_file(struct buffer *buffer, const char *filename)
fclose(fp);
return -1;
}
- buffer->size = file_size;
+ buffer->size = ALIGN_UP(file_size, size_granularity);
buffer->name = strdup(filename);
buffer->data = (char *)malloc(buffer->size);
assert(buffer->data);
- if (fread(buffer->data, 1, buffer->size, fp) != buffer->size) {
+ if (fread(buffer->data, 1, file_size, fp) != (size_t)file_size) {
fprintf(stderr, "incomplete read: %s\n", filename);
fclose(fp);
buffer_delete(buffer);
return -1;
}
fclose(fp);
+
+ if (buffer->size > (size_t)file_size)
+ memset(buffer->data + file_size, 0xff, buffer->size - file_size);
+
return 0;
}
+int buffer_from_file(struct buffer *buffer, const char *filename)
+{
+ return buffer_from_file_aligned_size(buffer, filename, 1);
+}
+
int buffer_write_file(struct buffer *buffer, const char *filename)
{
FILE *fp = fopen(filename, "wb");
diff --git a/util/cbfstool/common.h b/util/cbfstool/common.h
index 07ffdf8231..b23c8d21b0 100644
--- a/util/cbfstool/common.h
+++ b/util/cbfstool/common.h
@@ -132,6 +132,11 @@ int buffer_create(struct buffer *buffer, size_t size, const char *name);
/* Loads a file into memory buffer. Returns 0 on success, otherwise non-zero. */
int buffer_from_file(struct buffer *buffer, const char *filename);
+/* Loads a file into memory buffer (with buffer size rounded up to a multiple of
+ size_granularity). Returns 0 on success, otherwise non-zero. */
+int buffer_from_file_aligned_size(struct buffer *buffer, const char *filename,
+ size_t size_granularity);
+
/* Writes memory buffer content into file.
* Returns 0 on success, otherwise non-zero. */
int buffer_write_file(struct buffer *buffer, const char *filename);