Add and use xmmap.

Everyone forgets that mmap returns MAP_FAILED rather than NULL on failure.
Every use of mmap in toybox was either doing the wrong check, or no check
at all (including the two I personally added).

1 files changed, 87 insertions, 88 deletions

diff --git a/toys/pending/mdev.c b/toys/pending/mdev.c
index 2688cf35..1cf4e7cd 100644
--- a/toys/pending/mdev.c
+++ b/toys/pending/mdev.c
@@ -82,103 +82,102 @@ static void make_device(char *path)
 
     // mmap the config file
     if (-1!=(fd = open("/etc/mdev.conf", O_RDONLY))) {
+      int line = 0;
+
       len = fdlength(fd);
-      conf = mmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);
-      if (conf) {
-        int line = 0;
-
-        // Loop through lines in mmaped file
-        for (pos = conf; pos-conf<len;) {
-          int field;
-          char *end2;
-
-          line++;
-          // find end of this line
-          for(end = pos; end-conf<len && *end!='\n'; end++);
-
-          // Three fields: regex, uid:gid, mode
-          for (field = 3; field; field--) {
-            // Skip whitespace
-            while (pos<end && isspace(*pos)) pos++;
-            if (pos==end || *pos=='#') break;
-            for (end2 = pos;
-              end2<end && !isspace(*end2) && *end2!='#'; end2++);
-            switch(field) {
-              // Regex to match this device
-              case 3:
-              {
-                char *regex = strndup(pos, end2-pos);
-                regex_t match;
-                regmatch_t off;
-                int result;
-
-                // Is this it?
-                xregcomp(&match, regex, REG_EXTENDED);
-                result=regexec(&match, device_name, 1, &off, 0);
-                regfree(&match);
-                free(regex);
-
-                // If not this device, skip rest of line
-                if (result || off.rm_so
-                  || off.rm_eo!=strlen(device_name))
-                    goto end_line;
-
-                break;
-              }
-              // uid:gid
-              case 2:
-              {
-                char *s2;
-
-                // Find :
-                for(s = pos; s<end2 && *s!=':'; s++);
-                if (s==end2) goto end_line;
-
-                // Parse UID
-                uid = strtoul(pos,&s2,10);
-                if (s!=s2) {
-                  struct passwd *pass;
-                  char *str = strndup(pos, s-pos);
-                  pass = getpwnam(str);
-                  free(str);
-                  if (!pass) goto end_line;
-                  uid = pass->pw_uid;
-                }
-                s++;
-                // parse GID
-                gid = strtoul(s,&s2,10);
-                if (end2!=s2) {
-                  struct group *grp;
-                  char *str = strndup(s, end2-s);
-                  grp = getgrnam(str);
-                  free(str);
-                  if (!grp) goto end_line;
-                  gid = grp->gr_gid;
-                }
-                break;
+      conf = xmmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);
+
+      // Loop through lines in mmaped file
+      for (pos = conf; pos-conf<len;) {
+        int field;
+        char *end2;
+
+        line++;
+        // find end of this line
+        for(end = pos; end-conf<len && *end!='\n'; end++);
+
+        // Three fields: regex, uid:gid, mode
+        for (field = 3; field; field--) {
+          // Skip whitespace
+          while (pos<end && isspace(*pos)) pos++;
+          if (pos==end || *pos=='#') break;
+          for (end2 = pos;
+            end2<end && !isspace(*end2) && *end2!='#'; end2++);
+          switch(field) {
+            // Regex to match this device
+            case 3:
+            {
+              char *regex = strndup(pos, end2-pos);
+              regex_t match;
+              regmatch_t off;
+              int result;
+
+              // Is this it?
+              xregcomp(&match, regex, REG_EXTENDED);
+              result=regexec(&match, device_name, 1, &off, 0);
+              regfree(&match);
+              free(regex);
+
+              // If not this device, skip rest of line
+              if (result || off.rm_so
+                || off.rm_eo!=strlen(device_name))
+                  goto end_line;
+
+              break;
+            }
+            // uid:gid
+            case 2:
+            {
+              char *s2;
+
+              // Find :
+              for(s = pos; s<end2 && *s!=':'; s++);
+              if (s==end2) goto end_line;
+
+              // Parse UID
+              uid = strtoul(pos,&s2,10);
+              if (s!=s2) {
+                struct passwd *pass;
+                char *str = strndup(pos, s-pos);
+                pass = getpwnam(str);
+                free(str);
+                if (!pass) goto end_line;
+                uid = pass->pw_uid;
               }
-              // mode
-              case 1:
-              {
-                mode = strtoul(pos, &pos, 8);
-                if (pos!=end2) goto end_line;
-                goto found_device;
+              s++;
+              // parse GID
+              gid = strtoul(s,&s2,10);
+              if (end2!=s2) {
+                struct group *grp;
+                char *str = strndup(s, end2-s);
+                grp = getgrnam(str);
+                free(str);
+                if (!grp) goto end_line;
+                gid = grp->gr_gid;
               }
+              break;
+            }
+            // mode
+            case 1:
+            {
+              mode = strtoul(pos, &pos, 8);
+              if (pos!=end2) goto end_line;
+              goto found_device;
             }
-            pos=end2;
           }
+          pos=end2;
+        }
 end_line:
-          // Did everything parse happily?
-          if (field && field!=3) error_exit("Bad line %d", line);
+        // Did everything parse happily?
+        if (field && field!=3) error_exit("Bad line %d", line);
 
-          // Next line
-          pos = ++end;
-        }
-found_device:
-        munmap(conf, len);
+        // Next line
+        pos = ++end;
       }
-      close(fd);
+found_device:
+      munmap(conf, len);
     }
+    close(fd);
   }
 
   sprintf(toybuf, "/dev/%s", device_name);