]> git.ipfire.org Git - thirdparty/mdadm.git/blobdiff - mapfile.c
Remove scattered checks for malloc success.
[thirdparty/mdadm.git] / mapfile.c
index 112d3854c225181bafec17a619161d00b6f7ba54..fa238832ab6ba633ef08960125f37242945cfd09 100644 (file)
--- a/mapfile.c
+++ b/mapfile.c
@@ -1,8 +1,8 @@
 /*
- * mapfile - manage /var/run/mdadm.map. Part of:
+ * mapfile - keep track of uuid <-> array mapping. Part of:
  * mdadm - manage Linux "md" devices aka RAID arrays.
  *
- * Copyright (C) 2006-2009 Neil Brown <neilb@suse.de>
+ * Copyright (C) 2006-2010 Neil Brown <neilb@suse.de>
  *
  *
  *    This program is free software; you can redistribute it and/or modify
@@ -28,8 +28,8 @@
  *           Australia
  */
 
-/* /var/run/mdadm.map is used to track arrays being created in --incremental
- * more.  It particularly allows lookup from UUID to array device, but
+/* The mapfile is used to track arrays being created in --incremental
+ * mode.  It particularly allows lookup from UUID to array device, but
  * also allows the array device name to be easily found.
  *
  * The map file is line based with space separated fields.  The fields are:
  *  UUID       -  uuid of the array
  *  path       -  path where device created: /dev/md/home
  *
- * The preferred location for the map file is /var/run/mdadm.map.
- * However /var/run may not exist or be writable in early boot.  And if
- * no-one has created /var/run/mdadm, we still want to survive.
- * So possible locations are:
- *   /var/run/mdadm/map  /var/run/mdadm.map  /dev/.mdadm.map
- * the last, because udev requires a writable /dev very early.
- * We read from the first one that exists and write to the first
- * one that we can.
+ * The best place for the mapfile is /run/mdadm/map.  Distros and users
+ * which have not switched to /run yet can choose a different location
+ * at compile time via MAP_DIR and MAP_FILE.
  */
-#include "mdadm.h"
+#include       "mdadm.h"
+#include       <sys/file.h>
+#include       <ctype.h>
 
-#define mapnames(base) { #base, #base ".new", #base ".lock"}
-char *mapname[3][3] = {
-       mapnames(/var/run/mdadm/map),
-       mapnames(/var/run/mdadm.map),
-       mapnames(/dev/.mdadm.map)
+
+#define MAP_READ 0
+#define MAP_NEW 1
+#define MAP_LOCK 2
+#define MAP_DIRNAME 3
+
+char *mapname[4] = {
+       MAP_DIR "/" MAP_FILE,
+       MAP_DIR "/" MAP_FILE ".new",
+       MAP_DIR "/" MAP_FILE ".lock",
+       MAP_DIR
 };
 
-int mapmode[3] = { O_RDONLY, O_RDWR|O_CREAT, O_RDWR|O_CREAT | O_TRUNC };
+int mapmode[3] = { O_RDONLY, O_RDWR|O_CREAT, O_RDWR|O_CREAT|O_TRUNC };
 char *mapsmode[3] = { "r", "w", "w"};
 
-FILE *open_map(int modenum, int *choice)
+FILE *open_map(int modenum)
 {
-       int i;
-       for (i = 0 ; i < 3 ; i++) {
-               int fd = open(mapname[i][modenum], mapmode[modenum], 0600);
-               if (fd >= 0) {
-                       *choice = i;
-                       return fdopen(fd, mapsmode[modenum]);
-               }
-       }
+       int fd;
+       if ((mapmode[modenum] & O_CREAT))
+               /* Attempt to create directory, don't worry about
+                * failure.
+                */
+               (void)mkdir(mapname[MAP_DIRNAME], 0755);
+       fd = open(mapname[modenum], mapmode[modenum], 0600);
+       if (fd >= 0)
+               return fdopen(fd, mapsmode[modenum]);
        return NULL;
 }
 
@@ -76,9 +80,8 @@ int map_write(struct map_ent *mel)
 {
        FILE *f;
        int err;
-       int which;
 
-       f = open_map(1, &which);
+       f = open_map(MAP_NEW);
 
        if (!f)
                return 0;
@@ -98,27 +101,36 @@ int map_write(struct map_ent *mel)
        err = ferror(f);
        fclose(f);
        if (err) {
-               unlink(mapname[which][1]);
+               unlink(mapname[1]);
                return 0;
        }
-       return rename(mapname[which][1],
-                     mapname[which][0]) == 0;
+       return rename(mapname[1],
+                     mapname[0]) == 0;
 }
 
 
 static FILE *lf = NULL;
-static int lwhich = 0;
 int map_lock(struct map_ent **melp)
 {
-       if (lf == NULL) {
-               lf = open_map(2, &lwhich);
+       while (lf == NULL) {
+               struct stat buf;
+               lf = open_map(MAP_LOCK);
                if (lf == NULL)
                        return -1;
-               if (lockf(fileno(lf), F_LOCK, 0) != 0) {
+               if (flock(fileno(lf), LOCK_EX) != 0) {
                        fclose(lf);
                        lf = NULL;
                        return -1;
                }
+               if (fstat(fileno(lf), &buf) != 0 ||
+                   buf.st_nlink == 0) {
+                       /* The owner of the lock unlinked it,
+                        * so we have a lock on a stale file,
+                        * try again
+                        */
+                       fclose(lf);
+                       lf = NULL;
+               }
        }
        if (*melp)
                map_free(*melp);
@@ -128,21 +140,38 @@ int map_lock(struct map_ent **melp)
 
 void map_unlock(struct map_ent **melp)
 {
-       if (lf)
+       if (lf) {
+               /* must unlink before closing the file,
+                * as only the owner of the lock may
+                * unlink the file
+                */
+               unlink(mapname[2]);
                fclose(lf);
-       unlink(mapname[lwhich][2]);
+       }
        lf = NULL;
 }
 
+void map_fork(void)
+{
+       /* We are forking, so must close the lock file.
+        * Don't risk flushing anything though.
+        */
+       if (lf) {
+               close(fileno(lf));
+               fclose(lf);
+               lf = NULL;
+       }
+}
+
 void map_add(struct map_ent **melp,
            int devnum, char *metadata, int uuid[4], char *path)
 {
-       struct map_ent *me = malloc(sizeof(*me));
+       struct map_ent *me = xmalloc(sizeof(*me));
 
        me->devnum = devnum;
        strcpy(me->metadata, metadata);
        memcpy(me->uuid, uuid, 16);
-       me->path = path ? strdup(path) : NULL;
+       me->path = path ? xstrdup(path) : NULL;
        me->next = *melp;
        me->bad = 0;
        *melp = me;
@@ -156,14 +185,13 @@ void map_read(struct map_ent **melp)
        int devnum, uuid[4];
        char metadata[30];
        char nam[4];
-       int which;
 
        *melp = NULL;
 
-       f = open_map(0, &which);
+       f = open_map(MAP_READ);
        if (!f) {
                RebuildMap();
-               f = open_map(0, &which);
+               f = open_map(MAP_READ);
        }
        if (!f)
                return;
@@ -209,7 +237,8 @@ int map_update(struct map_ent **mpp, int devnum, char *metadata,
                        strcpy(mp->metadata, metadata);
                        memcpy(mp->uuid, uuid, 16);
                        free(mp->path);
-                       mp->path = path ? strdup(path) : NULL;
+                       mp->path = path ? xstrdup(path) : NULL;
+                       mp->bad = 0;
                        break;
                }
        if (!mp)
@@ -238,6 +267,16 @@ void map_delete(struct map_ent **mapp, int devnum)
        }
 }
 
+void map_remove(struct map_ent **mapp, int devnum)
+{
+       if (devnum == NoMdDev)
+               return;
+
+       map_delete(mapp, devnum);
+       map_write(*mapp);
+       map_free(*mapp);
+}
+
 struct map_ent *map_by_uuid(struct map_ent **map, int uuid[4])
 {
        struct map_ent *mp;
@@ -296,27 +335,59 @@ struct map_ent *map_by_name(struct map_ent **map, char *name)
        return NULL;
 }
 
+/* sets the proper subarray and container_dev according to the metadata
+ * version super_by_fd does this automatically, this routine is meant as
+ * a supplement for guess_super()
+ */
+static char *get_member_info(struct mdstat_ent *ent)
+{
+
+       if (ent->metadata_version == NULL ||
+           strncmp(ent->metadata_version, "external:", 9) != 0)
+               return NULL;
+
+       if (is_subarray(&ent->metadata_version[9])) {
+               char *subarray;
+
+               subarray = strrchr(ent->metadata_version, '/');
+               return subarray + 1;
+       }
+       return NULL;
+}
+
 void RebuildMap(void)
 {
        struct mdstat_ent *mdstat = mdstat_read(0, 0);
        struct mdstat_ent *md;
        struct map_ent *map = NULL;
        int mdp = get_mdp_major();
+       int require_homehost;
+       char sys_hostname[256];
+       char *homehost = conf_get_homehost(&require_homehost);
+
+       if (homehost == NULL || strcmp(homehost, "<system>")==0) {
+               if (gethostname(sys_hostname, sizeof(sys_hostname)) == 0) {
+                       sys_hostname[sizeof(sys_hostname)-1] = 0;
+                       homehost = sys_hostname;
+               }
+       }
 
        for (md = mdstat ; md ; md = md->next) {
-               struct mdinfo *sra = sysfs_read(-1, md->devnum, GET_DEVS|SKIP_GONE_DEVS);
+               struct mdinfo *sra = sysfs_read(-1, md->devnum, GET_DEVS);
                struct mdinfo *sd;
 
                if (!sra)
                        continue;
 
                for (sd = sra->devs ; sd ; sd = sd->next) {
+                       char namebuf[100];
                        char dn[30];
                        int dfd;
                        int ok;
                        struct supertype *st;
+                       char *subarray = NULL;
                        char *path;
-                       struct mdinfo info;
+                       struct mdinfo *info;
 
                        sprintf(dn, "%d:%d", sd->disk.major, sd->disk.minor);
                        dfd = dev_open(dn, O_RDONLY);
@@ -325,30 +396,112 @@ void RebuildMap(void)
                        st = guess_super(dfd);
                        if ( st == NULL)
                                ok = -1;
-                       else
+                       else {
+                               subarray = get_member_info(md);
                                ok = st->ss->load_super(st, dfd, NULL);
+                       }
                        close(dfd);
                        if (ok != 0)
                                continue;
-                       st->ss->getinfo_super(st, &info);
+                       info = st->ss->container_content(st, subarray);
+                       if (!info)
+                               continue;
+
                        if (md->devnum >= 0)
                                path = map_dev(MD_MAJOR, md->devnum, 0);
                        else
                                path = map_dev(mdp, (-1-md->devnum)<< 6, 0);
+                       if (path == NULL ||
+                           strncmp(path, "/dev/md/", 8) != 0) {
+                               /* We would really like a name that provides
+                                * an MD_DEVNAME for udev.
+                                * The name needs to be unique both in /dev/md/
+                                * and in this mapfile.
+                                * It needs to match watch -I or -As would come
+                                * up with.
+                                * That means:
+                                *   Check if array is in mdadm.conf 
+                                *        - if so use that.
+                                *   determine trustworthy from homehost etc
+                                *   find a unique name based on metadata name.
+                                *   
+                                */
+                               struct mddev_ident *match = conf_match(st, info,
+                                                                      NULL, 0,
+                                                                      NULL);
+                               struct stat stb;
+                               if (match && match->devname && match->devname[0] == '/') {
+                                       path = match->devname;
+                                       if (path[0] != '/') {
+                                               strcpy(namebuf, "/dev/md/");
+                                               strcat(namebuf, path);
+                                               path = namebuf;
+                                       }
+                               } else {
+                                       int unum = 0;
+                                       char *sep = "_";
+                                       const char *name;
+                                       int conflict = 1;
+                                       if ((homehost == NULL ||
+                                            st->ss->match_home(st, homehost) != 1) &&
+                                           st->ss->match_home(st, "any") != 1 &&
+                                           (require_homehost
+                                            || ! conf_name_is_free(info->name)))
+                                               /* require a numeric suffix */
+                                               unum = 0;
+                                       else
+                                               /* allow name to be used as-is if no conflict */
+                                               unum = -1;
+                                       name = info->name;
+                                       if (!*name) {
+                                               name = st->ss->name;
+                                               if (!isdigit(name[strlen(name)-1]) &&
+                                                   unum == -1) {
+                                                       unum = 0;
+                                                       sep = "";
+                                               }
+                                       }
+                                       if (strchr(name, ':'))
+                                               /* probably a uniquifying
+                                                * hostname prefix.  Allow
+                                                * without a suffix
+                                                */
+                                               unum = -1;
+
+                                       while (conflict) {
+                                               if (unum >= 0)
+                                                       sprintf(namebuf, "/dev/md/%s%s%d",
+                                                               name, sep, unum);
+                                               else
+                                                       sprintf(namebuf, "/dev/md/%s",
+                                                               name);
+                                               unum++;
+                                               if (lstat(namebuf, &stb) != 0 &&
+                                                   (map == NULL ||
+                                                    !map_by_name(&map, namebuf+8)))
+                                                       conflict = 0;
+                                       }
+                                       path = namebuf;
+                               }
+                       }
                        map_add(&map, md->devnum,
-                               info.text_version,
-                               info.uuid, path);
+                               info->text_version,
+                               info->uuid, path);
                        st->ss->free_super(st);
+                       free(info);
                        break;
                }
                sysfs_free(sra);
        }
-       map_write(map);
+       /* Only trigger a change if we wrote a new map file */
+       if (map_write(map))
+               for (md = mdstat ; md ; md = md->next) {
+                       struct mdinfo *sra = sysfs_read(-1, md->devnum,
+                                                       GET_VERSION);
+                       if (sra)
+                               sysfs_uevent(sra, "change");
+                       sysfs_free(sra);
+               }
        map_free(map);
-       for (md = mdstat ; md ; md = md->next) {
-               struct mdinfo *sra = sysfs_read(-1, md->devnum, GET_VERSION);
-               sysfs_uevent(sra, "change");
-               sysfs_free(sra);
-       }
        free_mdstat(mdstat);
 }