msg.c

   1 /*
   2  * Copyright (C) 2008 Intel Corporation
   3  *
   4  *      mdmon socket / message handling
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms and conditions of the GNU General Public License,
   8  * version 2, as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along with
  16  * this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19 #ifndef _GNU_SOURCE
  20 #define _GNU_SOURCE
  21 #endif
  22 #include <unistd.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <stdlib.h>
  26 #include <errno.h>
  27 #include <fcntl.h>
  28 #include <sys/types.h>
  29 #include <sys/socket.h>
  30 #include <sys/un.h>
  31 #include "mdadm.h"
  32 #include "mdmon.h"
  33
  34 static const __u32 start_magic = 0x5a5aa5a5;
  35 static const __u32 end_magic = 0xa5a55a5a;
  36
  37 static int send_buf(int fd, const void* buf, int len, int tmo)
  38 {
  39         fd_set set;
  40         int rv;
  41         struct timeval timeout = {tmo, 0};
  42         struct timeval *ptmo = tmo ? &timeout : NULL;
  43
  44         while (len) {
  45                 FD_ZERO(&set);
  46                 FD_SET(fd, &set);
  47                 rv = select(fd+1, NULL, &set, NULL, ptmo);
  48                 if (rv <= 0)
  49                         return -1;
  50                 rv = write(fd, buf, len);
  51                 if (rv <= 0)
  52                         return -1;
  53                 len -= rv;
  54                 buf += rv;
  55         }
  56         return 0;
  57 }
  58
  59 static int recv_buf(int fd, void* buf, int len, int tmo)
  60 {
  61         fd_set set;
  62         int rv;
  63         struct timeval timeout = {tmo, 0};
  64         struct timeval *ptmo = tmo ? &timeout : NULL;
  65
  66         while (len) {
  67                 FD_ZERO(&set);
  68                 FD_SET(fd, &set);
  69                 rv = select(fd+1, &set, NULL, NULL, ptmo);
  70                 if (rv <= 0)
  71                         return -1;
  72                 rv = read(fd, buf, len);
  73                 if (rv <= 0)
  74                         return -1;
  75                 len -= rv;
  76                 buf += rv;
  77         }
  78         return 0;
  79 }
  80
  81
  82 int send_message(int fd, struct metadata_update *msg, int tmo)
  83 {
  84         __s32 len = msg->len;
  85         int rv;
  86
  87         rv = send_buf(fd, &start_magic, 4, tmo);
  88         rv = rv ?: send_buf(fd, &len, 4, tmo);
  89         if (len > 0)
  90                 rv = rv ?: send_buf(fd, msg->buf, msg->len, tmo);
  91         rv = send_buf(fd, &end_magic, 4, tmo);
  92
  93         return rv;
  94 }
  95
  96 int receive_message(int fd, struct metadata_update *msg, int tmo)
  97 {
  98         __u32 magic;
  99         __s32 len;
 100         int rv;
 101
 102         rv = recv_buf(fd, &magic, 4, tmo);
 103         if (rv < 0 || magic != start_magic)
 104                 return -1;
 105         rv = recv_buf(fd, &len, 4, tmo);
 106         if (rv < 0 || len > MSG_MAX_LEN)
 107                 return -1;
 108         if (len > 0) {
 109                 msg->buf = malloc(len);
 110                 if (msg->buf == NULL)
 111                         return -1;
 112                 rv = recv_buf(fd, msg->buf, len, tmo);
 113                 if (rv < 0) {
 114                         free(msg->buf);
 115                         return -1;
 116                 }
 117         } else
 118                 msg->buf = NULL;
 119         rv = recv_buf(fd, &magic, 4, tmo);
 120         if (rv < 0 || magic != end_magic) {
 121                 free(msg->buf);
 122                 return -1;
 123         }
 124         msg->len = len;
 125         return 0;
 126 }
 127
 128 int ack(int fd, int tmo)
 129 {
 130         struct metadata_update msg = { .len = 0 };
 131
 132         return send_message(fd, &msg, tmo);
 133 }
 134
 135 int wait_reply(int fd, int tmo)
 136 {
 137         struct metadata_update msg;
 138         int err = receive_message(fd, &msg, tmo);
 139
 140         /* mdmon sent extra data, but caller only cares that we got a
 141          * successful reply
 142          */
 143         if (err == 0 && msg.len > 0)
 144                 free(msg.buf);
 145
 146         return err;
 147 }
 148
 149 int connect_monitor(char *devname)
 150 {
 151         char path[100];
 152         int sfd;
 153         long fl;
 154         struct sockaddr_un addr;
 155         int pos;
 156         char *c;
 157
 158         pos = sprintf(path, "%s/", MDMON_DIR);
 159         if (is_subarray(devname)) {
 160                 devname++;
 161                 c = strchr(devname, '/');
 162                 if (!c)
 163                         return -1;
 164                 snprintf(&path[pos], c - devname + 1, "%s", devname);
 165                 pos += c - devname;
 166         } else
 167                 pos += sprintf(&path[pos], "%s", devname);
 168         sprintf(&path[pos], ".sock");
 169
 170         sfd = socket(PF_LOCAL, SOCK_STREAM, 0);
 171         if (sfd < 0)
 172                 return -1;
 173
 174         addr.sun_family = PF_LOCAL;
 175         strcpy(addr.sun_path, path);
 176         if (connect(sfd, &addr, sizeof(addr)) < 0) {
 177                 close(sfd);
 178                 return -1;
 179         }
 180
 181         fl = fcntl(sfd, F_GETFL, 0);
 182         fl |= O_NONBLOCK;
 183         fcntl(sfd, F_SETFL, fl);
 184
 185         return sfd;
 186 }
 187
 188 int fping_monitor(int sfd)
 189 {
 190         int err = 0;
 191
 192         if (sfd < 0)
 193                 return sfd;
 194
 195         /* try to ping existing socket */
 196         if (ack(sfd, 20) != 0)
 197                 err = -1;
 198
 199         /* check the reply */
 200         if (!err && wait_reply(sfd, 20) != 0)
 201                 err = -1;
 202
 203         return err;
 204 }
 205
 206 /* give the monitor a chance to update the metadata */
 207 int ping_monitor(char *devname)
 208 {
 209         int sfd = connect_monitor(devname);
 210         int err;
 211
 212         if (sfd >= 0) {
 213                 err = fping_monitor(sfd);
 214                 close(sfd);
 215         } else
 216                 err = -1;
 217
 218         return err;
 219 }
 220
 221 /* ping monitor using device number */
 222 int ping_monitor_by_id(int devnum)
 223 {
 224         int err = -1;
 225         char *container = devnum2devname(devnum);
 226
 227         if (container) {
 228                 err = ping_monitor(container);
 229                 free(container);
 230         }
 231
 232         return err;
 233 }
 234
 235 static char *ping_monitor_version(char *devname)
 236 {
 237         int sfd = connect_monitor(devname);
 238         struct metadata_update msg;
 239         int err = 0;
 240
 241         if (sfd < 0)
 242                 return NULL;
 243
 244         if (ack(sfd, 20) != 0)
 245                 err = -1;
 246
 247         if (!err && receive_message(sfd, &msg, 20) != 0)
 248                 err = -1;
 249
 250         close(sfd);
 251
 252         if (err || !msg.len || !msg.buf)
 253                 return NULL;
 254         return msg.buf;
 255 }
 256
 257 int unblock_subarray(struct mdinfo *sra, const int unfreeze)
 258 {
 259         char buf[64];
 260         int rc = 0;
 261
 262         if (sra) {
 263                 sprintf(buf, "external:%s\n", sra->text_version);
 264                 buf[9] = '/';
 265         } else
 266                 buf[9] = '-';
 267
 268         if (buf[9] == '-' ||
 269             sysfs_set_str(sra, NULL, "metadata_version", buf) ||
 270             (unfreeze &&
 271              sysfs_attribute_available(sra, NULL, "sync_action") &&
 272              sysfs_set_str(sra, NULL, "sync_action", "idle")))
 273                 rc = -1;
 274         return rc;
 275 }
 276
 277 int block_subarray(struct mdinfo *sra)
 278 {
 279         char buf[64];
 280         int rc = 0;
 281
 282         sprintf(buf, "external:%s\n", sra->text_version);
 283         buf[9] = '-';
 284         if (sysfs_set_str(sra, NULL, "metadata_version", buf))
 285                 rc = -1;
 286
 287         return rc;
 288 }
 289
 290 /* check mdmon version if it supports
 291  * array blocking mechanism
 292  */
 293 int check_mdmon_version(char *container)
 294 {
 295         char *version = NULL;
 296         int devnum = devname2devnum(container);
 297
 298         if (!mdmon_running(devnum)) {
 299                 /* if mdmon is not active we assume that any instance that is
 300                  * later started will match the current mdadm version, if this
 301                  * assumption is violated we may inadvertantly rebuild an array
 302                  * that was meant for reshape, or start rebuild on a spare that
 303                  * was to be moved to another container
 304                  */
 305                 /* pass */;
 306         } else {
 307                 int ver;
 308
 309                 version = ping_monitor_version(container);
 310                 ver = version ? mdadm_version(version) : -1;
 311                 free(version);
 312                 if (ver < 3002000) {
 313                         fprintf(stderr, Name
 314                                 ": mdmon instance for %s cannot be disabled\n",
 315                                 container);
 316                         return -1;
 317                 }
 318         }
 319
 320         return 0;
 321 }
 322
 323 /**
 324  * block_monitor - prevent mdmon spare assignment
 325  * @container - container to block
 326  * @freeze - flag to additionally freeze sync_action
 327  *
 328  * This is used by the reshape code to freeze the container, and the
 329  * auto-rebuild implementation to atomically move spares.
 330  * In both cases we need to stop mdmon from assigning spares to replace
 331  * failed devices as we might have other plans for the spare.
 332  * For the reshape case we also need to 'freeze' sync_action so that
 333  * no recovery happens until we have fully prepared for the reshape.
 334  *
 335  * We tell mdmon that the array is frozen by marking the 'metadata' name
 336  * with a leading '-'.  The previously told mdmon "Don't make this array
 337  * read/write, leave it readonly".  Now it means a more general "Don't
 338  * reconfigure this array at all".
 339  * As older versions of mdmon (which might run from initrd) don't understand
 340  * this, we first check that the running mdmon is new enough.
 341  */
 342 int block_monitor(char *container, const int freeze)
 343 {
 344         struct mdstat_ent *ent, *e, *e2;
 345         struct mdinfo *sra = NULL;
 346         char buf[64];
 347         int rv = 0;
 348
 349         if (check_mdmon_version(container))
 350                 return -1;
 351
 352         ent = mdstat_read(0, 0);
 353         if (!ent) {
 354                 fprintf(stderr, Name
 355                         ": failed to read /proc/mdstat while disabling mdmon\n");
 356                 return -1;
 357         }
 358
 359         /* freeze container contents */
 360         for (e = ent; e; e = e->next) {
 361                 if (!is_container_member(e, container))
 362                         continue;
 363                 sysfs_free(sra);
 364                 sra = sysfs_read(-1, e->devnum, GET_VERSION);
 365                 if (!sra) {
 366                         fprintf(stderr, Name
 367                                 ": failed to read sysfs for subarray%s\n",
 368                                 to_subarray(e, container));
 369                         break;
 370                 }
 371                 /* can't reshape an array that we can't monitor */
 372                 if (sra->text_version[0] == '-')
 373                         break;
 374
 375                 if (freeze && sysfs_freeze_array(sra) < 1)
 376                         break;
 377                 /* flag this array to not be modified by mdmon (close race with
 378                  * takeover in reshape case and spare reassignment in the
 379                  * auto-rebuild case)
 380                  */
 381                 if (block_subarray(sra))
 382                         break;
 383                 ping_monitor(container);
 384
 385                 /* check that we did not race with recovery */
 386                 if ((freeze &&
 387                      !sysfs_attribute_available(sra, NULL, "sync_action")) ||
 388                     (freeze &&
 389                      sysfs_attribute_available(sra, NULL, "sync_action") &&
 390                      sysfs_get_str(sra, NULL, "sync_action", buf, 20) > 0 &&
 391                      strcmp(buf, "frozen\n") == 0))
 392                         /* pass */;
 393                 else {
 394                         unblock_subarray(sra, 0);
 395                         break;
 396                 }
 397                 /* Double check against races - there should be no spares
 398                  * or part-spares
 399                  */
 400                 sysfs_free(sra);
 401                 sra = sysfs_read(-1, e->devnum, GET_DEVS | GET_STATE);
 402                 if (sra && sra->array.spare_disks > 0) {
 403                         unblock_subarray(sra, freeze);
 404                         break;
 405                 }
 406         }
 407
 408         if (e) {
 409                 fprintf(stderr, Name ": failed to freeze subarray%s\n",
 410                         to_subarray(e, container));
 411
 412                 /* thaw the partially frozen container */
 413                 for (e2 = ent; e2 && e2 != e; e2 = e2->next) {
 414                         if (!is_container_member(e2, container))
 415                                 continue;
 416                         sysfs_free(sra);
 417                         sra = sysfs_read(-1, e2->devnum, GET_VERSION);
 418                         if (unblock_subarray(sra, freeze))
 419                                 fprintf(stderr, Name ": Failed to unfreeze %s\n", e2->dev);
 420                 }
 421
 422                 ping_monitor(container); /* cleared frozen */
 423                 rv = -1;
 424         }
 425
 426         sysfs_free(sra);
 427         free_mdstat(ent);
 428
 429         return rv;
 430 }
 431
 432 void unblock_monitor(char *container, const int unfreeze)
 433 {
 434         struct mdstat_ent *ent, *e;
 435         struct mdinfo *sra = NULL;
 436         int to_ping = 0;
 437
 438         ent = mdstat_read(0, 0);
 439         if (!ent) {
 440                 fprintf(stderr, Name
 441                         ": failed to read /proc/mdstat while unblocking container\n");
 442                 return;
 443         }
 444
 445         /* unfreeze container contents */
 446         for (e = ent; e; e = e->next) {
 447                 if (!is_container_member(e, container))
 448                         continue;
 449                 sysfs_free(sra);
 450                 sra = sysfs_read(-1, e->devnum, GET_VERSION|GET_LEVEL);
 451                 if (!sra)
 452                         continue;
 453                 if (sra->array.level > 0)
 454                         to_ping++;
 455                 if (unblock_subarray(sra, unfreeze))
 456                         fprintf(stderr, Name ": Failed to unfreeze %s\n", e->dev);
 457         }
 458         if (to_ping)
 459                 ping_monitor(container);
 460
 461         sysfs_free(sra);
 462         free_mdstat(ent);
 463 }
 464
 465
 466
 467 /* give the manager a chance to view the updated container state.  This
 468  * would naturally happen due to the manager noticing a change in
 469  * /proc/mdstat; however, pinging encourages this detection to happen
 470  * while an exclusive open() on the container is active
 471  */
 472 int ping_manager(char *devname)
 473 {
 474         int sfd = connect_monitor(devname);
 475         struct metadata_update msg = { .len = -1 };
 476         int err = 0;
 477
 478         if (sfd < 0)
 479                 return sfd;
 480
 481         err = send_message(sfd, &msg, 20);
 482
 483         /* check the reply */
 484         if (!err && wait_reply(sfd, 20) != 0)
 485                 err = -1;
 486
 487         close(sfd);
 488         return err;
 489 }
 490
 491 /* using takeover operation for grow purposes, mdadm has to be sure
 492  * that mdmon processes all updates, and if necessary it will be closed
 493  * at takeover to raid0 operation
 494   */
 495 void flush_mdmon(char *container)
 496 {
 497         ping_manager(container);
 498         ping_monitor(container);
 499 }