113 lines
3.2 KiB
Diff
113 lines
3.2 KiB
Diff
|
From de9a74e1c60210bee229fcf55b1678a99d1b44dd Mon Sep 17 00:00:00 2001
|
||
|
From: Jean Delvare <jdelvare@suse.de>
|
||
|
Date: Mon, 2 Nov 2015 09:45:26 +0100
|
||
|
Subject: [PATCH 7/9] Let read_file return the actual data size
|
||
|
|
||
|
Let read_file return the actual data size to the caller. This gives
|
||
|
the caller the possibility to check that the data size is as expected
|
||
|
and large enough for the purpose, and report to the user if not.
|
||
|
---
|
||
|
CHANGELOG | 5 +++++
|
||
|
dmidecode.c | 4 +++-
|
||
|
util.c | 11 +++++++----
|
||
|
util.h | 2 +-
|
||
|
4 files changed, 16 insertions(+), 6 deletions(-)
|
||
|
|
||
|
diff --git a/CHANGELOG b/CHANGELOG
|
||
|
index be2092a..1e5437a 100644
|
||
|
--- a/CHANGELOG
|
||
|
+++ b/CHANGELOG
|
||
|
@@ -1,3 +1,8 @@
|
||
|
+2015-11-02 Jean Delvare <jdelvare@suse.de>
|
||
|
+
|
||
|
+ * dmidecode.c, util.c, util.h: Let read_file return the actual data
|
||
|
+ size.
|
||
|
+
|
||
|
2015-10-21 Xie XiuQi <xiexiuqi@huawei.com>
|
||
|
|
||
|
* dmidecode.c: Handle SMBIOS 3.0 entry points on EFI systems.
|
||
|
diff --git a/dmidecode.c b/dmidecode.c
|
||
|
index 183ced4..a43cfd1 100644
|
||
|
--- a/dmidecode.c
|
||
|
+++ b/dmidecode.c
|
||
|
@@ -4751,6 +4751,7 @@ int main(int argc, char * const argv[])
|
||
|
int ret = 0; /* Returned value */
|
||
|
int found = 0;
|
||
|
off_t fp;
|
||
|
+ size_t size;
|
||
|
int efi;
|
||
|
u8 *buf;
|
||
|
|
||
|
@@ -4820,8 +4821,9 @@ int main(int argc, char * const argv[])
|
||
|
* contain one of several types of entry points, so read enough for
|
||
|
* the largest one, then determine what type it contains.
|
||
|
*/
|
||
|
+ size = 0x20;
|
||
|
if (!(opt.flags & FLAG_NO_SYSFS)
|
||
|
- && (buf = read_file(0x20, SYS_ENTRY_FILE)) != NULL)
|
||
|
+ && (buf = read_file(&size, SYS_ENTRY_FILE)) != NULL)
|
||
|
{
|
||
|
if (!(opt.flags & FLAG_QUIET))
|
||
|
printf("Getting SMBIOS data from sysfs.\n");
|
||
|
diff --git a/util.c b/util.c
|
||
|
index f97ac0d..52ed413 100644
|
||
|
--- a/util.c
|
||
|
+++ b/util.c
|
||
|
@@ -94,10 +94,11 @@ int checksum(const u8 *buf, size_t len)
|
||
|
* needs to be freed by the caller.
|
||
|
* This provides a similar usage model to mem_chunk()
|
||
|
*
|
||
|
- * Returns pointer to buffer of max_len bytes, or NULL on error
|
||
|
+ * Returns pointer to buffer of max_len bytes, or NULL on error, and
|
||
|
+ * sets max_len to the length actually read.
|
||
|
*
|
||
|
*/
|
||
|
-void *read_file(size_t max_len, const char *filename)
|
||
|
+void *read_file(size_t *max_len, const char *filename)
|
||
|
{
|
||
|
int fd;
|
||
|
size_t r2 = 0;
|
||
|
@@ -115,7 +116,7 @@ void *read_file(size_t max_len, const char *filename)
|
||
|
return(NULL);
|
||
|
}
|
||
|
|
||
|
- if ((p = malloc(max_len)) == NULL)
|
||
|
+ if ((p = malloc(*max_len)) == NULL)
|
||
|
{
|
||
|
perror("malloc");
|
||
|
return NULL;
|
||
|
@@ -123,7 +124,7 @@ void *read_file(size_t max_len, const char *filename)
|
||
|
|
||
|
do
|
||
|
{
|
||
|
- r = read(fd, p + r2, max_len - r2);
|
||
|
+ r = read(fd, p + r2, *max_len - r2);
|
||
|
if (r == -1)
|
||
|
{
|
||
|
if (errno != EINTR)
|
||
|
@@ -140,6 +141,8 @@ void *read_file(size_t max_len, const char *filename)
|
||
|
while (r != 0);
|
||
|
|
||
|
close(fd);
|
||
|
+ *max_len = r2;
|
||
|
+
|
||
|
return p;
|
||
|
}
|
||
|
|
||
|
diff --git a/util.h b/util.h
|
||
|
index 9d409cd..b8748f1 100644
|
||
|
--- a/util.h
|
||
|
+++ b/util.h
|
||
|
@@ -25,7 +25,7 @@
|
||
|
#define ARRAY_SIZE(x) (sizeof(x)/sizeof((x)[0]))
|
||
|
|
||
|
int checksum(const u8 *buf, size_t len);
|
||
|
-void *read_file(size_t len, const char *filename);
|
||
|
+void *read_file(size_t *len, const char *filename);
|
||
|
void *mem_chunk(off_t base, size_t len, const char *devmem);
|
||
|
int write_dump(size_t base, size_t len, const void *data, const char *dumpfile, int add);
|
||
|
u64 u64_range(u64 start, u64 end);
|
||
|
--
|
||
|
2.5.0
|
||
|
|