[PATCH v2] xfsdump: Judge the return value of malloc function

[zhanchengbin <zhanchengbin1@huawei.com>]

Add judgment on the return value of malloc function.

Signed-off-by: zhanchengbin <zhanchengbin1@huawei.com>
---
  common/inventory.c    |  4 ++++
  inventory/inv_stobj.c |  4 ++++
  invutil/invidx.c      | 24 ++++++++++++++++++++++++
  invutil/invutil.c     | 12 ++++++++++++
  4 files changed, 44 insertions(+)

diff --git a/common/inventory.c b/common/inventory.c
index 6ffe9fe..0de7f6f 100644
--- a/common/inventory.c
+++ b/common/inventory.c
@@ -364,6 +364,10 @@ inv_stream_open(

  	/* XXX yukk... make the token descriptors not pointers */
  	stok = (inv_stmtoken_t) malloc(sizeof(invt_strdesc_entry_t));
+	if (stok == NULL) {
+		fprintf(stderr, "%s: internal memory error: stok\n", g_programName);
+		exit(1);
+	}

  	stok->md_sesstok = tok;
  	stok->md_lastmfile = 0;
diff --git a/inventory/inv_stobj.c b/inventory/inv_stobj.c
index e2e8767..5a6786a 100644
--- a/inventory/inv_stobj.c
+++ b/inventory/inv_stobj.c
@@ -1015,6 +1015,10 @@ stobj_unpack_sessinfo(
  	assert (bufp);

  	tmpbuf = (char *)malloc(bufsz);
+	if (tmpbuf == NULL) {
+		fprintf(stderr, "internal memory error: tmpbuf\n");
+		exit(1);
+	}

  	/* first make sure that the magic cookie at the beginning is right.
  	   this isn't null-terminated */
diff --git a/invutil/invidx.c b/invutil/invidx.c
index 5874e8d..5e4c79c 100644
--- a/invutil/invidx.c
+++ b/invutil/invidx.c
@@ -363,15 +363,27 @@ read_stobj_info(int fd, int idx, invt_seshdr_t 
**out_hdr,

      lseek(fd, STOBJ_OFFSET(idx, 0), SEEK_SET);
      hdr = malloc(sizeof(*hdr));
+    if (hdr == NULL) {
+        fprintf(stderr, "%s: internal memory error: hdr\n", g_programName);
+        exit(1);
+    }
      read_n_bytes(fd, (char *)hdr, sizeof(*hdr), "stobj file");

      lseek(fd, hdr->sh_sess_off, SEEK_SET);
      ses = malloc(sizeof(*ses));
+    if (ses == NULL) {
+        fprintf(stderr, "%s: internal memory error: ses\n", g_programName);
+        exit(1);
+    }
      read_n_bytes(fd, (char *)ses, sizeof(*ses), "stobj file");

      if(ses->s_cur_nstreams > 0) {
  	lseek(fd, hdr->sh_streams_off, SEEK_SET);
  	strms = malloc(sizeof(*strms) * ses->s_cur_nstreams);
+	if (strms == NULL) {
+	    fprintf(stderr, "%s: internal memory error: strms\n", g_programName);
+    	    exit(1);
+	}
  	read_n_bytes(fd, (char *)strms, sizeof(*strms) * ses->s_cur_nstreams, 
"stobj file");

  	nmfiles = 0;
@@ -381,6 +393,10 @@ read_stobj_info(int fd, int idx, invt_seshdr_t 
**out_hdr,

  	if(nmfiles > 0) {
  	    mfiles = malloc(sizeof(*mfiles) * nmfiles);
+	    if (mfiles == NULL) {
+		fprintf(stderr, "%s: internal memory error: mfiles\n", g_programName);
+	        exit(1);
+            }
  	    read_n_bytes(fd, (char *)mfiles, sizeof(*mfiles) * nmfiles, 
"stobj file");
  	}
  	else {
@@ -552,6 +568,10 @@ insert_stobj_into_stobjfile(int invidx_fileidx, 
char *filename, int fd,

      /* for seshdr: malloc buffer, copy new entry into buffer, read 
file entries into buffer, write the lot out */
      buf = malloc(((sescnt.ic_curnum - pos) + 1) * sizeof(invt_seshdr_t));
+    if (buf == NULL) {
+	fprintf(stderr, "%s: internal memory error: buf\n", g_programName);
+	exit(1);
+    }
      memmove(buf, hdr, sizeof(invt_seshdr_t));
      lseek(fd, STOBJ_OFFSET(pos, 0), SEEK_SET);
      read_n_bytes(fd, buf + sizeof(invt_seshdr_t), (sescnt.ic_curnum - 
pos) * sizeof(invt_seshdr_t), "stobj file");
@@ -561,6 +581,10 @@ insert_stobj_into_stobjfile(int invidx_fileidx, 
char *filename, int fd,

      /* for session: malloc buffer, copy new entry into buffer, read 
file entries into buffer, write the lot out */
      buf = malloc(((sescnt.ic_curnum - pos) + 1) * sizeof(invt_session_t));
+    if (buf == NULL) {
+	fprintf(stderr, "%s: internal memory error: buf\n", g_programName);
+	exit(1);
+    }
      memmove(buf, ses, sizeof(invt_session_t));
      lseek(fd, STOBJ_OFFSET(sescnt.ic_maxnum, pos), SEEK_SET);
      read_n_bytes(fd, buf + sizeof(invt_session_t), (sescnt.ic_curnum - 
pos) * sizeof(invt_session_t), "stobj file");
diff --git a/invutil/invutil.c b/invutil/invutil.c
index 0d27a0b..304edc0 100644
--- a/invutil/invutil.c
+++ b/invutil/invutil.c
@@ -409,6 +409,10 @@ GetNameOfStobj (char *inv_path, char *filename)

      str = basename(filename);
      name = (char *) malloc(strlen(inv_path) + 1  + strlen(str) + 1);
+    if (name == NULL) {
+	fprintf(stderr, "%s: internal memory error: name\n", g_programName);
+	exit(1);
+    }
      strcpy(name, inv_path);
      strcat(name, "/");
      strcat(name, str);
@@ -425,6 +429,10 @@ GetNameOfInvIndex (char *inv_path, uuid_t uuid)
      uuid_unparse(uuid, str);
      name = (char *) malloc(strlen(inv_path) + 1  + strlen(str)
  			     + strlen(INV_INVINDEX_PREFIX) + 1);
+    if (name == NULL) {
+	fprintf(stderr, "%s: internal memory error: name\n", g_programName);
+	exit(1);
+    }
      strcpy(name, inv_path);
      strcat(name, "/");
      strcat(name, str);
@@ -440,6 +448,10 @@ GetFstabFullPath(char *inv_path)

      fstabname = (char *) malloc(strlen(inv_path) + 1 /* one for the "/" */
  				   + strlen("fstab") + 1);
+    if (fstabname == NULL) {
+	fprintf(stderr, "%s: internal memory error: fstabname\n", g_programName);
+	exit(1);
+    }
      strcpy(fstabname, inv_path);
      strcat(fstabname, "/");
      strcat(fstabname, "fstab");
-- 
2.33.0