mdmon.c

   1 /*
   2  * mdmon - monitor external metadata arrays
   3  *
   4  * Copyright (C) 2007-2009 Neil Brown <neilb@suse.de>
   5  * Copyright (C) 2007-2009 Intel Corporation
   6  *
   7  * This program is free software; you can redistribute it and/or modify it
   8  * under the terms and conditions of the GNU General Public License,
   9  * version 2, as published by the Free Software Foundation.
  10  *
  11  * This program is distributed in the hope it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  14  * more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along with
  17  * this program; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 /*
  22  * md array manager.
  23  * When md arrays have user-space managed metadata, this is the program
  24  * that does the managing.
  25  *
  26  * Given one argument: the name of the array (e.g. /dev/md0) that is
  27  * the container.
  28  * We fork off a helper that runs high priority and mlocked.  It responds to
  29  * device failures and other events that might stop writeout, or that are
  30  * trivial to deal with.
  31  * The main thread then watches for new arrays being created in the container
  32  * and starts monitoring them too ... along with a few other tasks.
  33  *
  34  * The main thread communicates with the priority thread by writing over
  35  * a pipe.
  36  * Separate programs can communicate with the main thread via Unix-domain
  37  * socket.
  38  * The two threads share address space and open file table.
  39  *
  40  */
  41
  42 #ifndef _GNU_SOURCE
  43 #define _GNU_SOURCE
  44 #endif
  45
  46 #include        <unistd.h>
  47 #include        <stdlib.h>
  48 #include        <sys/types.h>
  49 #include        <sys/stat.h>
  50 #include        <sys/socket.h>
  51 #include        <sys/un.h>
  52 #include        <sys/mman.h>
  53 #include        <sys/syscall.h>
  54 #include        <sys/wait.h>
  55 #include        <stdio.h>
  56 #include        <errno.h>
  57 #include        <string.h>
  58 #include        <fcntl.h>
  59 #include        <dirent.h>
  60 #ifdef USE_PTHREADS
  61 #include        <pthread.h>
  62 #else
  63 #include        <sched.h>
  64 #endif
  65
  66 #include        "mdadm.h"
  67 #include        "mdmon.h"
  68
  69 char const Name[] = "mdmon";
  70
  71 struct active_array *discard_this;
  72 struct active_array *pending_discard;
  73
  74 int mon_tid, mgr_tid;
  75
  76 int sigterm;
  77
  78 #ifdef USE_PTHREADS
  79 static void *run_child(void *v)
  80 {
  81         struct supertype *c = v;
  82
  83         mon_tid = syscall(SYS_gettid);
  84         do_monitor(c);
  85         return 0;
  86 }
  87
  88 static int clone_monitor(struct supertype *container)
  89 {
  90         pthread_attr_t attr;
  91         pthread_t thread;
  92         int rc;
  93
  94         mon_tid = -1;
  95         pthread_attr_init(&attr);
  96         pthread_attr_setstacksize(&attr, 4096);
  97         pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
  98         rc = pthread_create(&thread, &attr, run_child, container);
  99         if (rc)
 100                 return rc;
 101         while (mon_tid == -1)
 102                 usleep(10);
 103         pthread_attr_destroy(&attr);
 104
 105         mgr_tid = syscall(SYS_gettid);
 106
 107         return mon_tid;
 108 }
 109 #else /* USE_PTHREADS */
 110 static int run_child(void *v)
 111 {
 112         struct supertype *c = v;
 113
 114         do_monitor(c);
 115         return 0;
 116 }
 117
 118 #ifdef __ia64__
 119 int __clone2(int (*fn)(void *),
 120             void *child_stack_base, size_t stack_size,
 121             int flags, void *arg, ...
 122          /* pid_t *pid, struct user_desc *tls, pid_t *ctid */ );
 123 #endif
 124 static int clone_monitor(struct supertype *container)
 125 {
 126         static char stack[4096];
 127
 128 #ifdef __ia64__
 129         mon_tid = __clone2(run_child, stack, sizeof(stack),
 130                    CLONE_FS|CLONE_FILES|CLONE_VM|CLONE_SIGHAND|CLONE_THREAD,
 131                    container);
 132 #else
 133         mon_tid = clone(run_child, stack+4096-64,
 134                    CLONE_FS|CLONE_FILES|CLONE_VM|CLONE_SIGHAND|CLONE_THREAD,
 135                    container);
 136 #endif
 137
 138         mgr_tid = syscall(SYS_gettid);
 139
 140         return mon_tid;
 141 }
 142 #endif /* USE_PTHREADS */
 143
 144 static int make_pidfile(char *devname)
 145 {
 146         char path[100];
 147         char pid[10];
 148         int fd;
 149         int n;
 150
 151         if (mkdir(MDMON_DIR, 0755) < 0 &&
 152             errno != EEXIST)
 153                 return -errno;
 154         sprintf(path, "%s/%s.pid", MDMON_DIR, devname);
 155
 156         fd = open(path, O_RDWR|O_CREAT|O_EXCL, 0600);
 157         if (fd < 0)
 158                 return -errno;
 159         sprintf(pid, "%d\n", getpid());
 160         n = write(fd, pid, strlen(pid));
 161         close(fd);
 162         if (n < 0)
 163                 return -errno;
 164         return 0;
 165 }
 166
 167 static void try_kill_monitor(pid_t pid, char *devname, int sock)
 168 {
 169         char buf[100];
 170         int fd;
 171         int n;
 172         long fl;
 173         int rv;
 174
 175         /* first rule of survival... don't off yourself */
 176         if (pid == getpid())
 177                 return;
 178
 179         /* kill this process if it is mdmon */
 180         sprintf(buf, "/proc/%lu/cmdline", (unsigned long) pid);
 181         fd = open(buf, O_RDONLY);
 182         if (fd < 0)
 183                 return;
 184
 185         n = read(fd, buf, sizeof(buf)-1);
 186         buf[sizeof(buf)-1] = 0;
 187         close(fd);
 188
 189         if (n < 0 || !(strstr(buf, "mdmon") ||
 190                        strstr(buf, "@dmon")))
 191                 return;
 192
 193         kill(pid, SIGTERM);
 194
 195         if (sock < 0)
 196                 return;
 197
 198         /* Wait for monitor to exit by reading from the socket, after
 199          * clearing the non-blocking flag */
 200         fl = fcntl(sock, F_GETFL, 0);
 201         fl &= ~O_NONBLOCK;
 202         fcntl(sock, F_SETFL, fl);
 203         n = read(sock, buf, 100);
 204
 205         /* If there is I/O going on it might took some time to get to
 206          * clean state. Wait for monitor to exit fully to avoid races.
 207          * Ping it with SIGUSR1 in case that it is sleeping  */
 208         for (n = 0; n < 25; n++) {
 209                 rv = kill(pid, SIGUSR1);
 210                 if (rv < 0)
 211                         break;
 212                 usleep(200000);
 213         }
 214 }
 215
 216 void remove_pidfile(char *devname)
 217 {
 218         char buf[100];
 219
 220         sprintf(buf, "%s/%s.pid", MDMON_DIR, devname);
 221         unlink(buf);
 222         sprintf(buf, "%s/%s.sock", MDMON_DIR, devname);
 223         unlink(buf);
 224 }
 225
 226 static int make_control_sock(char *devname)
 227 {
 228         char path[100];
 229         int sfd;
 230         long fl;
 231         struct sockaddr_un addr;
 232
 233         if (sigterm)
 234                 return -1;
 235
 236         sprintf(path, "%s/%s.sock", MDMON_DIR, devname);
 237         unlink(path);
 238         sfd = socket(PF_LOCAL, SOCK_STREAM, 0);
 239         if (sfd < 0)
 240                 return -1;
 241
 242         addr.sun_family = PF_LOCAL;
 243         strcpy(addr.sun_path, path);
 244         umask(077); /* ensure no world write access */
 245         if (bind(sfd, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
 246                 close(sfd);
 247                 return -1;
 248         }
 249         listen(sfd, 10);
 250         fl = fcntl(sfd, F_GETFL, 0);
 251         fl |= O_NONBLOCK;
 252         fcntl(sfd, F_SETFL, fl);
 253         return sfd;
 254 }
 255
 256 static void term(int sig)
 257 {
 258         sigterm = 1;
 259 }
 260
 261 static void wake_me(int sig)
 262 {
 263
 264 }
 265
 266 /* if we are debugging and starting mdmon by hand then don't fork */
 267 static int do_fork(void)
 268 {
 269         #ifdef DEBUG
 270         if (check_env("MDADM_NO_MDMON"))
 271                 return 0;
 272         #endif
 273
 274         return 1;
 275 }
 276
 277 void usage(void)
 278 {
 279         fprintf(stderr,
 280 "Usage: mdmon [options] CONTAINER\n"
 281 "\n"
 282 "Options are:\n"
 283 "  --help        -h   : This message\n"
 284 "  --all         -a   : All devices\n"
 285 "  --foreground  -F   : Run in foreground (do not fork)\n"
 286 "  --takeover    -t   : Takeover container\n"
 287 );
 288         exit(2);
 289 }
 290
 291 static bool is_duplicate_opt(const int opt, const int set_val, const char *long_name)
 292 {
 293         if (opt == set_val) {
 294                 pr_err("--%s option duplicated!\n", long_name);
 295                 return true;
 296         }
 297         return false;
 298 }
 299
 300 static int mdmon(char *devnm, int must_fork, int takeover);
 301
 302 int main(int argc, char *argv[])
 303 {
 304         char *container_name = NULL;
 305         char *devnm = NULL;
 306         int status = 0;
 307         int opt;
 308         int all = 0;
 309         int takeover = 0;
 310         int dofork = 1;
 311         bool help = false;
 312         static struct option options[] = {
 313                 {"all", 0, NULL, 'a'},
 314                 {"takeover", 0, NULL, 't'},
 315                 {"help", 0, NULL, 'h'},
 316                 {"offroot", 0, NULL, OffRootOpt},
 317                 {"foreground", 0, NULL, 'F'},
 318                 {NULL, 0, NULL, 0}
 319         };
 320
 321         while ((opt = getopt_long(argc, argv, "thaF", options, NULL)) != -1) {
 322                 switch (opt) {
 323                 case 'a':
 324                         if (is_duplicate_opt(all, 1, "all"))
 325                                 exit(1);
 326                         container_name = argv[optind-1];
 327                         all = 1;
 328                         break;
 329                 case 't':
 330                         if (is_duplicate_opt(takeover, 1, "takeover"))
 331                                 exit(1);
 332                         takeover = 1;
 333                         break;
 334                 case 'F':
 335                         if (is_duplicate_opt(dofork, 0, "foreground"))
 336                                 exit(1);
 337                         dofork = 0;
 338                         break;
 339                 case OffRootOpt:
 340                         if (is_duplicate_opt(argv[0][0], '@', "offroot"))
 341                                 exit(1);
 342                         argv[0][0] = '@';
 343                         break;
 344                 case 'h':
 345                         if (is_duplicate_opt(help, true, "help"))
 346                                 exit(1);
 347                         help = true;
 348                         break;
 349                 default:
 350                         usage();
 351                         break;
 352                 }
 353         }
 354
 355
 356         if (in_initrd()) {
 357                 /*
 358                  * set first char of argv[0] to @. This is used by
 359                  * systemd to signal that the task was launched from
 360                  * initrd/initramfs and should be preserved during shutdown
 361                  */
 362                 argv[0][0] = '@';
 363         }
 364
 365         if (all == 0 && container_name == NULL) {
 366                 if (argv[optind])
 367                         container_name = argv[optind];
 368         }
 369
 370         if (container_name == NULL)
 371                 usage();
 372
 373         if (argc - optind > 1)
 374                 usage();
 375
 376         if (strcmp(container_name, "/proc/mdstat") == 0)
 377                 all = 1;
 378
 379         if (help)
 380                 usage();
 381
 382         if (all) {
 383                 struct mdstat_ent *mdstat, *e;
 384                 int container_len = strlen(container_name);
 385
 386                 /* launch an mdmon instance for each container found */
 387                 mdstat = mdstat_read(0, 0);
 388                 for (e = mdstat; e; e = e->next) {
 389                         if (e->metadata_version &&
 390                             strncmp(e->metadata_version, "external:", 9) == 0 &&
 391                             !is_subarray(&e->metadata_version[9])) {
 392                                 /* update cmdline so this mdmon instance can be
 393                                  * distinguished from others in a call to ps(1)
 394                                  */
 395                                 if (strlen(e->devnm) <= (unsigned)container_len) {
 396                                         memset(container_name, 0, container_len);
 397                                         sprintf(container_name, "%s", e->devnm);
 398                                 }
 399                                 status |= mdmon(e->devnm, 1, takeover);
 400                         }
 401                 }
 402                 free_mdstat(mdstat);
 403
 404                 return status;
 405         } else if (strncmp(container_name, "md", 2) == 0) {
 406                 int id = devnm2devid(container_name);
 407                 if (id)
 408                         devnm = container_name;
 409         } else {
 410                 struct stat st;
 411
 412                 if (stat(container_name, &st) == 0)
 413                         devnm = xstrdup(stat2devnm(&st));
 414         }
 415
 416         if (!devnm) {
 417                 pr_err("%s is not a valid md device name\n",
 418                         container_name);
 419                 exit(1);
 420         }
 421         return mdmon(devnm, dofork && do_fork(), takeover);
 422 }
 423
 424 static int mdmon(char *devnm, int must_fork, int takeover)
 425 {
 426         int mdfd;
 427         struct mdinfo *mdi, *di;
 428         struct supertype *container;
 429         sigset_t set;
 430         struct sigaction act;
 431         int pfd[2];
 432         int status;
 433         int ignore;
 434         pid_t victim = -1;
 435         int victim_sock = -1;
 436
 437         dprintf("starting mdmon for %s\n", devnm);
 438
 439         mdfd = open_dev(devnm);
 440         if (mdfd < 0) {
 441                 pr_err("%s: %s\n", devnm, strerror(errno));
 442                 return 1;
 443         }
 444
 445         /* Fork, and have the child tell us when they are ready */
 446         if (must_fork) {
 447                 if (pipe(pfd) != 0) {
 448                         pr_err("failed to create pipe\n");
 449                         return 1;
 450                 }
 451                 switch(fork()) {
 452                 case -1:
 453                         pr_err("failed to fork: %s\n", strerror(errno));
 454                         return 1;
 455                 case 0: /* child */
 456                         close(pfd[0]);
 457                         break;
 458                 default: /* parent */
 459                         close(pfd[1]);
 460                         if (read(pfd[0], &status, sizeof(status)) != sizeof(status)) {
 461                                 wait(&status);
 462                                 status = WEXITSTATUS(status);
 463                         }
 464                         close(pfd[0]);
 465                         return status;
 466                 }
 467         } else
 468                 pfd[0] = pfd[1] = -1;
 469
 470         container = xcalloc(1, sizeof(*container));
 471         strcpy(container->devnm, devnm);
 472         container->arrays = NULL;
 473         container->sock = -1;
 474
 475         mdi = sysfs_read(mdfd, container->devnm, GET_VERSION|GET_LEVEL|GET_DEVS);
 476
 477         if (!mdi) {
 478                 pr_err("failed to load sysfs info for %s\n", container->devnm);
 479                 exit(3);
 480         }
 481         if (mdi->array.level != UnSet) {
 482                 pr_err("%s is not a container - cannot monitor\n", devnm);
 483                 exit(3);
 484         }
 485         if (mdi->array.major_version != -1 ||
 486             mdi->array.minor_version != -2) {
 487                 pr_err("%s does not use external metadata - cannot monitor\n",
 488                         devnm);
 489                 exit(3);
 490         }
 491
 492         container->ss = version_to_superswitch(mdi->text_version);
 493         if (container->ss == NULL) {
 494                 pr_err("%s uses unsupported metadata: %s\n",
 495                         devnm, mdi->text_version);
 496                 exit(3);
 497         }
 498
 499         container->devs = NULL;
 500         for (di = mdi->devs; di; di = di->next) {
 501                 struct mdinfo *cd = xmalloc(sizeof(*cd));
 502                 *cd = *di;
 503                 cd->next = container->devs;
 504                 container->devs = cd;
 505         }
 506         sysfs_free(mdi);
 507
 508         /* SIGUSR is sent between parent and child.  So both block it
 509          * and enable it only with pselect.
 510          */
 511         sigemptyset(&set);
 512         sigaddset(&set, SIGUSR1);
 513         sigaddset(&set, SIGTERM);
 514         sigprocmask(SIG_BLOCK, &set, NULL);
 515         act.sa_handler = wake_me;
 516         act.sa_flags = 0;
 517         sigaction(SIGUSR1, &act, NULL);
 518         act.sa_handler = term;
 519         sigaction(SIGTERM, &act, NULL);
 520         act.sa_handler = SIG_IGN;
 521         sigaction(SIGPIPE, &act, NULL);
 522
 523         victim = mdmon_pid(container->devnm);
 524         if (victim >= 0)
 525                 victim_sock = connect_monitor(container->devnm);
 526
 527         ignore = chdir("/");
 528         if (!takeover && victim > 0 && victim_sock >= 0) {
 529                 if (fping_monitor(victim_sock) == 0) {
 530                         pr_err("%s already managed\n", container->devnm);
 531                         exit(3);
 532                 }
 533                 close(victim_sock);
 534                 victim_sock = -1;
 535         }
 536         if (container->ss->load_container(container, mdfd, devnm)) {
 537                 pr_err("Cannot load metadata for %s\n", devnm);
 538                 exit(3);
 539         }
 540         close(mdfd);
 541
 542         /* Ok, this is close enough.  We can say goodbye to our parent now.
 543          */
 544         if (victim > 0)
 545                 remove_pidfile(devnm);
 546         if (make_pidfile(devnm) < 0) {
 547                 exit(3);
 548         }
 549         container->sock = make_control_sock(devnm);
 550
 551         status = 0;
 552         if (pfd[1] >= 0) {
 553                 if (write(pfd[1], &status, sizeof(status)) < 0)
 554                         pr_err("failed to notify our parent: %d\n",
 555                                getppid());
 556                 close(pfd[1]);
 557         }
 558
 559         mlockall(MCL_CURRENT | MCL_FUTURE);
 560
 561         if (clone_monitor(container) < 0) {
 562                 pr_err("failed to start monitor process: %s\n",
 563                         strerror(errno));
 564                 exit(2);
 565         }
 566
 567         if (victim > 0) {
 568                 try_kill_monitor(victim, container->devnm, victim_sock);
 569                 if (victim_sock >= 0)
 570                         close(victim_sock);
 571         }
 572
 573         setsid();
 574         manage_fork_fds(0);
 575
 576         /* This silliness is to stop the compiler complaining
 577          * that we ignore 'ignore'
 578          */
 579         if (ignore)
 580                 ignore++;
 581
 582         do_manager(container);
 583
 584         exit(0);
 585 }
 586
 587 /* Some stub functions so super-* can link with us */
 588 int child_monitor(int afd, struct mdinfo *sra, struct reshape *reshape,
 589                   struct supertype *st, unsigned long blocks,
 590                   int *fds, unsigned long long *offsets,
 591                   int dests, int *destfd, unsigned long long *destoffsets)
 592 {
 593         return 0;
 594 }
 595
 596 int restore_stripes(int *dest, unsigned long long *offsets,
 597                     int raid_disks, int chunk_size, int level, int layout,
 598                     int source, unsigned long long read_offset,
 599                     unsigned long long start, unsigned long long length,
 600                     char *src_buf)
 601 {
 602         return 1;
 603 }
 604
 605 int save_stripes(int *source, unsigned long long *offsets,
 606                  int raid_disks, int chunk_size, int level, int layout,
 607                  int nwrites, int *dest,
 608                  unsigned long long start, unsigned long long length,
 609                  char *buf)
 610 {
 611         return 0;
 612 }
 613
 614 struct superswitch super0 = {
 615         .name = "0.90",
 616 };
 617 struct superswitch super1 = {
 618         .name = "1.x",
 619 };