X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=libbb%2Fmtab.c;h=18386efb5acb578ba2f6d734b0e52ee116c4d152;hb=6f0540e7eca61fa10344699d023f4a90f705601d;hp=cce1aac5c45396212bd60c678de523017cd8357e;hpb=5d8843e451c01d8abfe6b5be772637310e9e581e;p=oweals%2Fbusybox.git

diff --git a/libbb/mtab.c b/libbb/mtab.c
index cce1aac5c..18386efb5 100644
--- a/libbb/mtab.c
+++ b/libbb/mtab.c
@@ -7,53 +7,42 @@
  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
-#include <stdlib.h>
-#include <unistd.h>
-#include <errno.h>
-#include <string.h>
-#include <stdio.h>
 #include <mntent.h>
 #include "libbb.h"
 
-#define MTAB_MAX_ENTRIES 40
-
-#ifdef CONFIG_FEATURE_MTAB_SUPPORT
+#if ENABLE_FEATURE_MTAB_SUPPORT
 void erase_mtab(const char *name)
 {
-	struct mntent entries[MTAB_MAX_ENTRIES];
-	int count = 0;
-	FILE *mountTable = setmntent(bb_path_mtab_file, "r");
+	struct mntent *entries = NULL;
+	int i, count = 0;
+	FILE *mountTable;
 	struct mntent *m;
 
-	/* Check if reading the mtab file failed */
-	if (mountTable == 0
-			/* Bummer.  fall back on trying the /proc filesystem */
-			&& (mountTable = setmntent("/proc/mounts", "r")) == 0) {
+	mountTable = setmntent(bb_path_mtab_file, "r");
+	/* Bummer. Fall back on trying the /proc filesystem */
+	if (!mountTable) mountTable = setmntent("/proc/mounts", "r");
+	if (!mountTable) {
 		bb_perror_msg(bb_path_mtab_file);
 		return;
 	}
 
-	while (((m = getmntent(mountTable)) != 0) && (count < MTAB_MAX_ENTRIES))
-	{
-		entries[count].mnt_fsname = strdup(m->mnt_fsname);
-		entries[count].mnt_dir = strdup(m->mnt_dir);
-		entries[count].mnt_type = strdup(m->mnt_type);
-		entries[count].mnt_opts = strdup(m->mnt_opts);
-		entries[count].mnt_freq = m->mnt_freq;
-		entries[count].mnt_passno = m->mnt_passno;
-		count++;
+	while ((m = getmntent(mountTable)) != 0) {
+		i = count++;
+		entries = xrealloc(entries, count * sizeof(entries[0]));
+		entries[i].mnt_fsname = xstrdup(m->mnt_fsname);
+		entries[i].mnt_dir = xstrdup(m->mnt_dir);
+		entries[i].mnt_type = xstrdup(m->mnt_type);
+		entries[i].mnt_opts = xstrdup(m->mnt_opts);
+		entries[i].mnt_freq = m->mnt_freq;
+		entries[i].mnt_passno = m->mnt_passno;
 	}
 	endmntent(mountTable);
-	if ((mountTable = setmntent(bb_path_mtab_file, "w"))) {
-		int i;
 
+	mountTable = setmntent(bb_path_mtab_file, "w");
+	if (mountTable) {
 		for (i = 0; i < count; i++) {
-			int result = (strcmp(entries[i].mnt_fsname, name) == 0
-						  || strcmp(entries[i].mnt_dir, name) == 0);
-
-			if (result)
-				continue;
-			else
+			if (strcmp(entries[i].mnt_fsname, name) != 0
+			 && strcmp(entries[i].mnt_dir, name) != 0)
 				addmntent(mountTable, &entries[i]);
 		}
 		endmntent(mountTable);