Assorted fixes

[thirdparty/mdadm.git] / Grow.c

diff --git a/Grow.c b/Grow.c
index 41c7dc0317aa539884c94a2586ac3c70d69449cb..615726133299053f97e47132d65bec2e528bfe9a 100644 (file)
--- a/Grow.c
+++ b/Grow.c
@@ -123,7 +123,7 @@ int Grow_Add_device(char *devname, int fd, char *newdev)
        info.disk.state = (1 << MD_DISK_SYNC) | (1 << MD_DISK_ACTIVE);
        st->ss->update_super(&info, super, "grow", newdev, 0);
 
-       if (st->ss->store_super(nfd, super)) {
+       if (st->ss->store_super(st, nfd, super)) {
                fprintf(stderr, Name ": Cannot store new superblock on %s\n", newdev);
                close(nfd);
                return 1;
@@ -181,7 +181,7 @@ int Grow_Add_device(char *devname, int fd, char *newdev)
                info.disk.state = (1 << MD_DISK_SYNC) | (1 << MD_DISK_ACTIVE);
                st->ss->update_super(&info, super, "grow", dv, 0);
                
-               if (st->ss->store_super(fd2, super)) {
+               if (st->ss->store_super(st, fd2, super)) {
                        fprintf(stderr, Name ": Cannot store new superblock on %s\n", dv);
                        close(fd2);
                        return 1;
@@ -191,3 +191,154 @@ int Grow_Add_device(char *devname, int fd, char *newdev)
 
        return 0;
 }
+
+int Grow_addbitmap(char *devname, int fd, char *file, int chunk, int delay, int write_behind)
+{
+       /*
+        * First check that array doesn't have a bitmap
+        * Then create the bitmap
+        * Then add it
+        *
+        * For internal bitmaps, we need to check the version,
+        * find all the active devices, and write the bitmap block
+        * to all devices
+        */
+       mdu_bitmap_file_t bmf;
+       mdu_array_info_t array;
+       struct supertype *st;
+
+       if (ioctl(fd, GET_BITMAP_FILE, &bmf) != 0) {
+               if (errno == ENOMEM) 
+                       fprintf(stderr, Name ": Memory allocation failure.\n");
+               else
+                       fprintf(stderr, Name ": bitmaps not supported by this kernel.\n");
+               return 1;
+       }
+       if (bmf.pathname[0]) {
+               if (strcmp(file,"none")==0) {
+                       if (ioctl(fd, SET_BITMAP_FILE, -1)!= 0) {
+                               fprintf(stderr, Name ": failed to remove bitmap %s\n",
+                                       bmf.pathname);
+                               return 1;
+                       }
+                       return 0;
+               }
+               fprintf(stderr, Name ": %s already has a bitmap (%s)\n",
+                       devname, bmf.pathname);
+               return 1;
+       }
+       if (ioctl(fd, GET_ARRAY_INFO, &array) != 0) {
+               fprintf(stderr, Name ": cannot get array status for %s\n", devname);
+               return 1;
+       }
+       if (array.state & (1<<MD_SB_BITMAP_PRESENT)) {
+               if (strcmp(file, "none")==0) {
+                       array.state &= ~(1<<MD_SB_BITMAP_PRESENT);
+                       if (ioctl(fd, SET_ARRAY_INFO, &array)!= 0) {
+                               fprintf(stderr, Name ": failed to remove internal bitmap.\n");
+                               return 1;
+                       }
+                       return 0;
+               }
+               fprintf(stderr, Name ": Internal bitmap already present on %s\n",
+                       devname);
+               return 1;
+       }
+       st = super_by_version(array.major_version, array.minor_version);
+       if (!st) {
+               fprintf(stderr, Name ": Cannot understand version %d.%d\n",
+                       array.major_version, array.minor_version);
+               return 1;
+       }
+       if (strcmp(file, "none") == 0) {
+               fprintf(stderr, Name ": no bitmap found on %s\n", devname);
+               return 1;
+       } else if (strcmp(file, "internal") == 0) {
+               int d;
+               for (d=0; d< st->max_devs; d++) {
+                       mdu_disk_info_t disk;
+                       char *dv;
+                       disk.number = d;
+                       if (ioctl(fd, GET_DISK_INFO, &disk) < 0)
+                               continue;
+                       if (disk.major == 0 &&
+                           disk.minor == 0)
+                               continue;
+                       if ((disk.state & (1<<MD_DISK_SYNC))==0)
+                               continue;
+                       dv = map_dev(disk.major, disk.minor);
+                       if (dv) {
+                               void *super;
+                               int fd2 = open(dv, O_RDWR);
+                               if (fd2 < 0)
+                                       continue;
+                               if (st->ss->load_super(st, fd2, &super, NULL)==0) {
+                                       st->ss->add_internal_bitmap(super, 
+                                                                   chunk, delay, write_behind,
+                                                                   array.size);
+                                       st->ss->write_bitmap(st, fd2, super);
+                               }
+                               close(fd2);
+                       }
+               }
+               array.state |= (1<<MD_SB_BITMAP_PRESENT);
+               if (ioctl(fd, SET_ARRAY_INFO, &array)!= 0) {
+                       fprintf(stderr, Name ": failed to set internal bitmap.\n");
+                       return 1;
+               }
+       } else {
+               int uuid[4];
+               int bitmap_fd;
+               int d;
+               int max_devs = st->max_devs;
+               void *super = NULL;
+               if (chunk == UnSet)
+                       chunk = DEFAULT_BITMAP_CHUNK;
+
+               /* try to load a superblock */
+               for (d=0; d<max_devs; d++) {
+                       mdu_disk_info_t disk;
+                       char *dv;
+                       int fd2;
+                       disk.number = d;
+                       if (ioctl(fd, GET_DISK_INFO, &disk) < 0)
+                               continue;
+                       if ((disk.major==0 && disk.minor==0) ||
+                           (disk.state & (1<<MD_DISK_REMOVED)))
+                               continue;
+                       dv = map_dev(disk.major, disk.minor);
+                       if (!dv) continue;
+                       fd2 = open(dv, O_RDONLY);
+                       if (fd2 >= 0 &&
+                           st->ss->load_super(st, fd2, &super, NULL) == 0) {
+                               close(fd2);
+                               st->ss->uuid_from_super(uuid, super);
+                               break;
+                       }
+                       close(fd2);
+               }
+               if (d == max_devs) {
+                       fprintf(stderr, Name ": cannot find UUID for array!\n");
+                       return 1;
+               }
+               if (CreateBitmap(file, 0, (char*)uuid, chunk,
+                                delay, write_behind, array.size*2ULL)) {
+                       return 1;
+               }
+               bitmap_fd = open(file, O_RDWR);
+               if (bitmap_fd < 0) {
+                       fprintf(stderr, Name ": weird: %s cannot be openned\n",
+                               file);
+                       return 1;
+               }
+               if (ioctl(fd, SET_BITMAP_FILE, bitmap_fd) < 0) {
+                       fprintf(stderr, Name ": Cannot set bitmap file for %s: %s\n",
+                               devname, strerror(errno));
+                       return 1;
+               }
+       }
+
+       return 0;
+}
+
+