1132904Spjd/*- 2196879Spjd * Copyright (c) 2004-2009 Pawel Jakub Dawidek <pjd@FreeBSD.org> 3132904Spjd * All rights reserved. 4132904Spjd * 5132904Spjd * Redistribution and use in source and binary forms, with or without 6132904Spjd * modification, are permitted provided that the following conditions 7132904Spjd * are met: 8132904Spjd * 1. Redistributions of source code must retain the above copyright 9132904Spjd * notice, this list of conditions and the following disclaimer. 10132904Spjd * 2. Redistributions in binary form must reproduce the above copyright 11132904Spjd * notice, this list of conditions and the following disclaimer in the 12132904Spjd * documentation and/or other materials provided with the distribution. 13155175Spjd * 14132904Spjd * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND 15132904Spjd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16132904Spjd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17132904Spjd * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE 18132904Spjd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19132904Spjd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20132904Spjd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21132904Spjd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22132904Spjd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23132904Spjd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24132904Spjd * SUCH DAMAGE. 25132904Spjd */ 26132904Spjd 27132904Spjd#include <sys/cdefs.h> 28132904Spjd__FBSDID("$FreeBSD$"); 29132904Spjd 30132904Spjd#include <sys/param.h> 31132904Spjd#include <errno.h> 32133447Spjd#include <paths.h> 33132904Spjd#include <stdio.h> 34132904Spjd#include <stdlib.h> 35132904Spjd#include <stdint.h> 36132904Spjd#include <string.h> 37132904Spjd#include <strings.h> 38132904Spjd#include <assert.h> 39132904Spjd#include <libgeom.h> 40132904Spjd#include <geom/mirror/g_mirror.h> 41132904Spjd#include <core/geom.h> 42132904Spjd#include <misc/subr.h> 43132904Spjd 44132904Spjduint32_t lib_version = G_LIB_VERSION; 45132904Spjduint32_t version = G_MIRROR_VERSION; 46132904Spjd 47212554Spjd#define GMIRROR_BALANCE "load" 48212554Spjd#define GMIRROR_SLICE "4096" 49212554Spjd#define GMIRROR_PRIORITY "0" 50212547Spjd 51133447Spjdstatic void mirror_main(struct gctl_req *req, unsigned flags); 52132904Spjdstatic void mirror_activate(struct gctl_req *req); 53132904Spjdstatic void mirror_clear(struct gctl_req *req); 54132904Spjdstatic void mirror_dump(struct gctl_req *req); 55132904Spjdstatic void mirror_label(struct gctl_req *req); 56132904Spjd 57132904Spjdstruct g_command class_commands[] = { 58212554Spjd { "activate", G_FLAG_VERBOSE, mirror_main, G_NULL_OPTS, 59143586Spjd "[-v] name prov ..." 60143586Spjd }, 61212554Spjd { "clear", G_FLAG_VERBOSE, mirror_main, G_NULL_OPTS, 62143586Spjd "[-v] prov ..." 63143586Spjd }, 64132904Spjd { "configure", G_FLAG_VERBOSE, NULL, 65132904Spjd { 66162868Spjd { 'a', "autosync", NULL, G_TYPE_BOOL }, 67212547Spjd { 'b', "balance", "", G_TYPE_STRING }, 68162868Spjd { 'd', "dynamic", NULL, G_TYPE_BOOL }, 69163888Spjd { 'f', "failsync", NULL, G_TYPE_BOOL }, 70163888Spjd { 'F', "nofailsync", NULL, G_TYPE_BOOL }, 71162868Spjd { 'h', "hardcode", NULL, G_TYPE_BOOL }, 72162868Spjd { 'n', "noautosync", NULL, G_TYPE_BOOL }, 73212554Spjd { 'p', "priority", "-1", G_TYPE_NUMBER }, 74212554Spjd { 's', "slice", "-1", G_TYPE_NUMBER }, 75132904Spjd G_OPT_SENTINEL 76143586Spjd }, 77212554Spjd "[-adfFhnv] [-b balance] [-s slice] name\n" 78212554Spjd "[-v] -p priority name prov" 79132904Spjd }, 80212554Spjd { "deactivate", G_FLAG_VERBOSE, NULL, G_NULL_OPTS, 81143586Spjd "[-v] name prov ..." 82143586Spjd }, 83212554Spjd { "dump", 0, mirror_main, G_NULL_OPTS, 84143586Spjd "prov ..." 85143586Spjd }, 86212554Spjd { "forget", G_FLAG_VERBOSE, NULL, G_NULL_OPTS, 87143586Spjd "name ..." 88143586Spjd }, 89132904Spjd { "label", G_FLAG_VERBOSE, mirror_main, 90132904Spjd { 91212547Spjd { 'b', "balance", GMIRROR_BALANCE, G_TYPE_STRING }, 92163888Spjd { 'F', "nofailsync", NULL, G_TYPE_BOOL }, 93162868Spjd { 'h', "hardcode", NULL, G_TYPE_BOOL }, 94162868Spjd { 'n', "noautosync", NULL, G_TYPE_BOOL }, 95212554Spjd { 's', "slice", GMIRROR_SLICE, G_TYPE_NUMBER }, 96132904Spjd G_OPT_SENTINEL 97143586Spjd }, 98212554Spjd "[-Fhnv] [-b balance] [-s slice] name prov ..." 99132904Spjd }, 100132904Spjd { "insert", G_FLAG_VERBOSE, NULL, 101132904Spjd { 102162868Spjd { 'h', "hardcode", NULL, G_TYPE_BOOL }, 103162868Spjd { 'i', "inactive", NULL, G_TYPE_BOOL }, 104212554Spjd { 'p', "priority", GMIRROR_PRIORITY, G_TYPE_NUMBER }, 105132904Spjd G_OPT_SENTINEL 106143586Spjd }, 107212554Spjd "[-hiv] [-p priority] name prov ..." 108132904Spjd }, 109212554Spjd { "rebuild", G_FLAG_VERBOSE, NULL, G_NULL_OPTS, 110143586Spjd "[-v] name prov ..." 111143586Spjd }, 112212554Spjd { "remove", G_FLAG_VERBOSE, NULL, G_NULL_OPTS, 113143586Spjd "[-v] name prov ..." 114143586Spjd }, 115132904Spjd { "stop", G_FLAG_VERBOSE, NULL, 116132904Spjd { 117162868Spjd { 'f', "force", NULL, G_TYPE_BOOL }, 118132904Spjd G_OPT_SENTINEL 119143586Spjd }, 120212554Spjd "[-fv] name ..." 121132904Spjd }, 122132904Spjd G_CMD_SENTINEL 123132904Spjd}; 124132904Spjd 125132904Spjdstatic int verbose = 0; 126132904Spjd 127132904Spjdstatic void 128133447Spjdmirror_main(struct gctl_req *req, unsigned flags) 129132904Spjd{ 130132904Spjd const char *name; 131132904Spjd 132133447Spjd if ((flags & G_FLAG_VERBOSE) != 0) 133132904Spjd verbose = 1; 134132904Spjd 135153190Spjd name = gctl_get_ascii(req, "verb"); 136132904Spjd if (name == NULL) { 137132904Spjd gctl_error(req, "No '%s' argument.", "verb"); 138132904Spjd return; 139132904Spjd } 140132904Spjd if (strcmp(name, "label") == 0) 141132904Spjd mirror_label(req); 142132904Spjd else if (strcmp(name, "clear") == 0) 143132904Spjd mirror_clear(req); 144132904Spjd else if (strcmp(name, "dump") == 0) 145132904Spjd mirror_dump(req); 146132904Spjd else if (strcmp(name, "activate") == 0) 147132904Spjd mirror_activate(req); 148132904Spjd else 149132904Spjd gctl_error(req, "Unknown command: %s.", name); 150132904Spjd} 151132904Spjd 152132904Spjdstatic void 153132904Spjdmirror_label(struct gctl_req *req) 154132904Spjd{ 155132904Spjd struct g_mirror_metadata md; 156132904Spjd u_char sector[512]; 157132904Spjd const char *str; 158132904Spjd unsigned sectorsize; 159132904Spjd off_t mediasize; 160153190Spjd intmax_t val; 161163888Spjd int error, i, nargs, bal, hardcode; 162132904Spjd 163153190Spjd nargs = gctl_get_int(req, "nargs"); 164153190Spjd if (nargs < 2) { 165132904Spjd gctl_error(req, "Too few arguments."); 166132904Spjd return; 167132904Spjd } 168132904Spjd 169132904Spjd strlcpy(md.md_magic, G_MIRROR_MAGIC, sizeof(md.md_magic)); 170132904Spjd md.md_version = G_MIRROR_VERSION; 171153190Spjd str = gctl_get_ascii(req, "arg0"); 172132904Spjd strlcpy(md.md_name, str, sizeof(md.md_name)); 173132904Spjd md.md_mid = arc4random(); 174153190Spjd md.md_all = nargs - 1; 175132904Spjd md.md_mflags = 0; 176132904Spjd md.md_dflags = 0; 177139213Spjd md.md_genid = 0; 178132904Spjd md.md_syncid = 1; 179132904Spjd md.md_sync_offset = 0; 180153190Spjd val = gctl_get_intmax(req, "slice"); 181153190Spjd md.md_slice = val; 182153190Spjd str = gctl_get_ascii(req, "balance"); 183132904Spjd bal = balance_id(str); 184132904Spjd if (bal == -1) { 185151651Sceri gctl_error(req, "Invalid balance algorithm."); 186132904Spjd return; 187132904Spjd } 188132904Spjd md.md_balance = bal; 189163888Spjd if (gctl_get_int(req, "noautosync")) 190132904Spjd md.md_mflags |= G_MIRROR_DEVICE_FLAG_NOAUTOSYNC; 191163888Spjd if (gctl_get_int(req, "nofailsync")) 192163888Spjd md.md_mflags |= G_MIRROR_DEVICE_FLAG_NOFAILSYNC; 193153190Spjd hardcode = gctl_get_int(req, "hardcode"); 194132904Spjd 195132904Spjd /* 196132904Spjd * Calculate sectorsize by finding least common multiple from 197132904Spjd * sectorsizes of every disk and find the smallest mediasize. 198132904Spjd */ 199132904Spjd mediasize = 0; 200132904Spjd sectorsize = 0; 201153190Spjd for (i = 1; i < nargs; i++) { 202132904Spjd unsigned ssize; 203132904Spjd off_t msize; 204132904Spjd 205153190Spjd str = gctl_get_ascii(req, "arg%d", i); 206132904Spjd msize = g_get_mediasize(str); 207132904Spjd ssize = g_get_sectorsize(str); 208132904Spjd if (msize == 0 || ssize == 0) { 209132904Spjd gctl_error(req, "Can't get informations about %s: %s.", 210132904Spjd str, strerror(errno)); 211132904Spjd return; 212132904Spjd } 213132904Spjd msize -= ssize; 214132904Spjd if (mediasize == 0 || (mediasize > 0 && msize < mediasize)) 215132904Spjd mediasize = msize; 216132904Spjd if (sectorsize == 0) 217132904Spjd sectorsize = ssize; 218132904Spjd else 219132904Spjd sectorsize = g_lcm(sectorsize, ssize); 220132904Spjd } 221132904Spjd md.md_mediasize = mediasize; 222132904Spjd md.md_sectorsize = sectorsize; 223163205Spjd md.md_mediasize -= (md.md_mediasize % md.md_sectorsize); 224132904Spjd 225132904Spjd /* 226132904Spjd * Clear last sector first, to spoil all components if device exists. 227132904Spjd */ 228153190Spjd for (i = 1; i < nargs; i++) { 229153190Spjd str = gctl_get_ascii(req, "arg%d", i); 230132904Spjd error = g_metadata_clear(str, NULL); 231132904Spjd if (error != 0) { 232132904Spjd gctl_error(req, "Can't store metadata on %s: %s.", str, 233132904Spjd strerror(error)); 234132904Spjd return; 235132904Spjd } 236132904Spjd } 237132904Spjd 238132904Spjd /* 239132904Spjd * Ok, store metadata (use disk number as priority). 240132904Spjd */ 241153190Spjd for (i = 1; i < nargs; i++) { 242153190Spjd str = gctl_get_ascii(req, "arg%d", i); 243132904Spjd md.md_did = arc4random(); 244132904Spjd md.md_priority = i - 1; 245142727Spjd md.md_provsize = g_get_mediasize(str); 246142727Spjd assert(md.md_provsize != 0); 247153190Spjd if (!hardcode) 248133447Spjd bzero(md.md_provider, sizeof(md.md_provider)); 249133447Spjd else { 250213662Sae if (strncmp(str, _PATH_DEV, sizeof(_PATH_DEV) - 1) == 0) 251213662Sae str += sizeof(_PATH_DEV) - 1; 252133447Spjd strlcpy(md.md_provider, str, sizeof(md.md_provider)); 253133447Spjd } 254132904Spjd mirror_metadata_encode(&md, sector); 255132904Spjd error = g_metadata_store(str, sector, sizeof(sector)); 256132904Spjd if (error != 0) { 257132904Spjd fprintf(stderr, "Can't store metadata on %s: %s.\n", 258132904Spjd str, strerror(error)); 259132904Spjd gctl_error(req, "Not fully done."); 260132904Spjd continue; 261132904Spjd } 262132904Spjd if (verbose) 263132904Spjd printf("Metadata value stored on %s.\n", str); 264132904Spjd } 265132904Spjd} 266132904Spjd 267132904Spjdstatic void 268132904Spjdmirror_clear(struct gctl_req *req) 269132904Spjd{ 270132904Spjd const char *name; 271153190Spjd int error, i, nargs; 272132904Spjd 273153190Spjd nargs = gctl_get_int(req, "nargs"); 274153190Spjd if (nargs < 1) { 275132904Spjd gctl_error(req, "Too few arguments."); 276132904Spjd return; 277132904Spjd } 278132904Spjd 279153190Spjd for (i = 0; i < nargs; i++) { 280153190Spjd name = gctl_get_ascii(req, "arg%d", i); 281132904Spjd error = g_metadata_clear(name, G_MIRROR_MAGIC); 282132904Spjd if (error != 0) { 283132904Spjd fprintf(stderr, "Can't clear metadata on %s: %s.\n", 284132904Spjd name, strerror(error)); 285132904Spjd gctl_error(req, "Not fully done."); 286132904Spjd continue; 287132904Spjd } 288132904Spjd if (verbose) 289155175Spjd printf("Metadata cleared on %s.\n", name); 290132904Spjd } 291132904Spjd} 292132904Spjd 293132904Spjdstatic void 294132904Spjdmirror_dump(struct gctl_req *req) 295132904Spjd{ 296132904Spjd struct g_mirror_metadata md, tmpmd; 297132904Spjd const char *name; 298153190Spjd int error, i, nargs; 299132904Spjd 300153190Spjd nargs = gctl_get_int(req, "nargs"); 301153190Spjd if (nargs < 1) { 302132904Spjd gctl_error(req, "Too few arguments."); 303132904Spjd return; 304132904Spjd } 305132904Spjd 306153190Spjd for (i = 0; i < nargs; i++) { 307153190Spjd name = gctl_get_ascii(req, "arg%d", i); 308132904Spjd error = g_metadata_read(name, (u_char *)&tmpmd, sizeof(tmpmd), 309132904Spjd G_MIRROR_MAGIC); 310132904Spjd if (error != 0) { 311132904Spjd fprintf(stderr, "Can't read metadata from %s: %s.\n", 312132904Spjd name, strerror(error)); 313132904Spjd gctl_error(req, "Not fully done."); 314132904Spjd continue; 315132904Spjd } 316132904Spjd if (mirror_metadata_decode((u_char *)&tmpmd, &md) != 0) { 317132904Spjd fprintf(stderr, "MD5 hash mismatch for %s, skipping.\n", 318132904Spjd name); 319132904Spjd gctl_error(req, "Not fully done."); 320132904Spjd continue; 321132904Spjd } 322132904Spjd printf("Metadata on %s:\n", name); 323132904Spjd mirror_metadata_dump(&md); 324132904Spjd printf("\n"); 325132904Spjd } 326132904Spjd} 327132904Spjd 328132904Spjdstatic void 329132904Spjdmirror_activate(struct gctl_req *req) 330132904Spjd{ 331132904Spjd struct g_mirror_metadata md, tmpmd; 332132904Spjd const char *name, *path; 333153190Spjd int error, i, nargs; 334132904Spjd 335153190Spjd nargs = gctl_get_int(req, "nargs"); 336153190Spjd if (nargs < 2) { 337132904Spjd gctl_error(req, "Too few arguments."); 338132904Spjd return; 339132904Spjd } 340153190Spjd name = gctl_get_ascii(req, "arg0"); 341132904Spjd 342153190Spjd for (i = 1; i < nargs; i++) { 343153190Spjd path = gctl_get_ascii(req, "arg%d", i); 344132904Spjd error = g_metadata_read(path, (u_char *)&tmpmd, sizeof(tmpmd), 345132904Spjd G_MIRROR_MAGIC); 346132904Spjd if (error != 0) { 347132904Spjd fprintf(stderr, "Cannot read metadata from %s: %s.\n", 348132904Spjd path, strerror(error)); 349132904Spjd gctl_error(req, "Not fully done."); 350132904Spjd continue; 351132904Spjd } 352132904Spjd if (mirror_metadata_decode((u_char *)&tmpmd, &md) != 0) { 353132904Spjd fprintf(stderr, 354132904Spjd "MD5 hash mismatch for provider %s, skipping.\n", 355132904Spjd path); 356132904Spjd gctl_error(req, "Not fully done."); 357132904Spjd continue; 358132904Spjd } 359132904Spjd if (strcmp(md.md_name, name) != 0) { 360132904Spjd fprintf(stderr, 361132904Spjd "Provider %s is not the mirror %s component.\n", 362132904Spjd path, name); 363132904Spjd gctl_error(req, "Not fully done."); 364132904Spjd continue; 365132904Spjd } 366132904Spjd md.md_dflags &= ~G_MIRROR_DISK_FLAG_INACTIVE; 367132904Spjd mirror_metadata_encode(&md, (u_char *)&tmpmd); 368132904Spjd error = g_metadata_store(path, (u_char *)&tmpmd, sizeof(tmpmd)); 369132904Spjd if (error != 0) { 370132904Spjd fprintf(stderr, "Cannot write metadata from %s: %s.\n", 371132904Spjd path, strerror(error)); 372132904Spjd gctl_error(req, "Not fully done."); 373132904Spjd continue; 374132904Spjd } 375132904Spjd if (verbose) 376155175Spjd printf("Provider %s activated.\n", path); 377132904Spjd } 378132904Spjd} 379