sbin/hammer/hammer.c

   1 /*
   2  * Copyright (c) 2007 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Matthew Dillon <dillon@backplane.com>
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $DragonFly: src/sbin/hammer/hammer.c,v 1.30 2008/07/07 00:27:22 dillon Exp $
  35  */
  36
  37 #include "hammer.h"
  38 #include <signal.h>
  39 #include <math.h>
  40
  41 static void hammer_parsedevs(const char *blkdevs);
  42 static void sigalrm(int signo);
  43 static void usage(int exit_code);
  44
  45 int RecurseOpt;
  46 int VerboseOpt;
  47 int NoSyncOpt;
  48 int TwoWayPipeOpt;
  49 int TimeoutOpt;
  50 const char *CyclePath;
  51 const char *LinkPath;
  52
  53 int
  54 main(int ac, char **av)
  55 {
  56         int ch;
  57         u_int32_t status;
  58         char *blkdevs = NULL;
  59
  60         while ((ch = getopt(ac, av, "c:dhf:rs:t:v2")) != -1) {
  61                 switch(ch) {
  62                 case '2':
  63                         TwoWayPipeOpt = 1;
  64                         break;
  65                 case 'c':
  66                         CyclePath = optarg;
  67                         break;
  68                 case 'd':
  69                         ++DebugOpt;
  70                         break;
  71                 case 'h':
  72                         usage(0);
  73                         /* not reached */
  74                 case 'r':
  75                         RecurseOpt = 1;
  76                         break;
  77                 case 'f':
  78                         blkdevs = optarg;
  79                         break;
  80                 case 's':
  81                         LinkPath = optarg;
  82                         break;
  83                 case 't':
  84                         TimeoutOpt = strtol(optarg, NULL, 0);
  85                         break;
  86                 case 'v':
  87                         ++VerboseOpt;
  88                         break;
  89                 default:
  90                         usage(1);
  91                         /* not reached */
  92                 }
  93         }
  94         ac -= optind;
  95         av += optind;
  96         if (ac < 1) {
  97                 usage(1);
  98                 /* not reached */
  99         }
 100
 101         signal(SIGALRM, sigalrm);
 102
 103         if (strcmp(av[0], "synctid") == 0) {
 104                 hammer_cmd_synctid(av + 1, ac - 1);
 105                 exit(0);
 106         }
 107         if (strcmp(av[0], "namekey") == 0) {
 108                 int64_t key;
 109
 110                 if (av[1] == NULL)
 111                         usage(1);
 112                 key = (int64_t)(crc32(av[1], strlen(av[1])) & 0x7FFFFFFF) << 32;
 113                 if (key == 0)
 114                         key |= 0x100000000LL;
 115                 printf("0x%016llx\n", key);
 116                 exit(0);
 117         }
 118         if (strcmp(av[0], "namekey32") == 0) {
 119                 int32_t key;
 120
 121                 if (av[1] == NULL)
 122                         usage(1);
 123                 key = crc32(av[1], strlen(av[1])) & 0x7FFFFFFF;
 124                 if (key == 0)
 125                         ++key;
 126                 printf("0x%08x\n", key);
 127                 exit(0);
 128         }
 129         if (strcmp(av[0], "pfs-status") == 0) {
 130                 hammer_cmd_pseudofs_status(av + 1, ac - 1);
 131                 exit(0);
 132         }
 133         if (strcmp(av[0], "pfs-create") == 0) {
 134                 hammer_cmd_pseudofs_create(av + 1, ac - 1);
 135                 exit(0);
 136         }
 137         if (strcmp(av[0], "pfs-update") == 0) {
 138                 hammer_cmd_pseudofs_update(av + 1, ac - 1, 0);
 139                 exit(0);
 140         }
 141         if (strcmp(av[0], "pfs-destroy") == 0) {
 142                 hammer_cmd_pseudofs_destroy(av + 1, ac - 1);
 143                 exit(0);
 144         }
 145         if (strcmp(av[0], "status") == 0) {
 146                 hammer_cmd_status(av + 1, ac - 1);
 147                 exit(0);
 148         }
 149         if (strcmp(av[0], "prune") == 0) {
 150                 hammer_cmd_softprune(av + 1, ac - 1, 0);
 151                 exit(0);
 152         }
 153         if (strcmp(av[0], "prune-everything") == 0) {
 154                 hammer_cmd_softprune(av + 1, ac - 1, 1);
 155                 exit(0);
 156         }
 157         if (strcmp(av[0], "snapshot") == 0) {
 158                 hammer_cmd_snapshot(av + 1, ac - 1);
 159                 exit(0);
 160         }
 161         if (strcmp(av[0], "bstats") == 0) {
 162                 hammer_cmd_bstats(av + 1, ac - 1);
 163                 exit(0);
 164         }
 165         if (strcmp(av[0], "iostats") == 0) {
 166                 hammer_cmd_iostats(av + 1, ac - 1);
 167                 exit(0);
 168         }
 169
 170         if (strncmp(av[0], "history", 7) == 0) {
 171                 hammer_cmd_history(av[0] + 7, av + 1, ac - 1);
 172                 exit(0);
 173         }
 174         if (strncmp(av[0], "reblock", 7) == 0) {
 175                 if (strcmp(av[0], "reblock") == 0)
 176                         hammer_cmd_reblock(av + 1, ac - 1, -1);
 177                 else if (strcmp(av[0], "reblock-btree") == 0)
 178                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_BTREE);
 179                 else if (strcmp(av[0], "reblock-inodes") == 0)
 180                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_INODES);
 181                 else if (strcmp(av[0], "reblock-dirs") == 0)
 182                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DIRS);
 183                 else if (strcmp(av[0], "reblock-data") == 0)
 184                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DATA);
 185                 else
 186                         usage(1);
 187                 exit(0);
 188         }
 189         if (strncmp(av[0], "mirror", 6) == 0) {
 190                 if (strcmp(av[0], "mirror-read") == 0)
 191                         hammer_cmd_mirror_read(av + 1, ac - 1);
 192                 else if (strcmp(av[0], "mirror-write") == 0)
 193                         hammer_cmd_mirror_write(av + 1, ac - 1);
 194                 else if (strcmp(av[0], "mirror-copy") == 0)
 195                         hammer_cmd_mirror_copy(av + 1, ac - 1);
 196                 else if (strcmp(av[0], "mirror-dump") == 0)
 197                         hammer_cmd_mirror_dump();
 198                 else
 199                         usage(1);
 200                 exit(0);
 201         }
 202
 203         uuid_name_lookup(&Hammer_FSType, "DragonFly HAMMER", &status);
 204         if (status != uuid_s_ok) {
 205                 errx(1, "uuids file does not have the DragonFly "
 206                         "HAMMER filesystem type");
 207         }
 208
 209         if (strcmp(av[0], "show") == 0) {
 210                 hammer_off_t node_offset = (hammer_off_t)-1;
 211
 212                 hammer_parsedevs(blkdevs);
 213                 if (ac > 1)
 214                         sscanf(av[1], "%llx", &node_offset);
 215                 hammer_cmd_show(node_offset, 0, NULL, NULL);
 216                 exit(0);
 217         }
 218         if (strcmp(av[0], "blockmap") == 0) {
 219                 hammer_parsedevs(blkdevs);
 220                 hammer_cmd_blockmap();
 221                 exit(0);
 222         }
 223         usage(1);
 224         /* not reached */
 225         return(0);
 226 }
 227
 228 static
 229 void
 230 hammer_parsedevs(const char *blkdevs)
 231 {
 232         char *copy;
 233         char *volname;
 234
 235         if (blkdevs == NULL) {
 236                 errx(1, "A -f blkdevs specification is required "
 237                         "for this command");
 238         }
 239
 240         copy = strdup(blkdevs);
 241         while ((volname = copy) != NULL) {
 242                 if ((copy = strchr(copy, ':')) != NULL)
 243                         *copy++ = 0;
 244                 setup_volume(-1, volname, 0, O_RDONLY);
 245         }
 246 }
 247
 248 static
 249 void
 250 sigalrm(int signo __unused)
 251 {
 252         /* do nothing (interrupts HAMMER ioctl) */
 253 }
 254
 255 static
 256 void
 257 usage(int exit_code)
 258 {
 259         fprintf(stderr,
 260                 "hammer -h\n"
 261                 "hammer [-t timeout] [-c cyclefile] ....\n"
 262                 "hammer prune <dir-holding-softlinks>\n"
 263                 "hammer prune-everything <filesystem>\n"
 264                 "hammer snapshot <softlink-dir> [<filesystem>]\n"
 265                 "hammer bstats <interval>\n"
 266                 "hammer iostats <interval>\n"
 267                 "hammer mirror-read <filesystem>\n"
 268                 "hammer mirror-write <filesystem>\n"
 269                 "hammer mirror-dump\n"
 270                 "hammer mirror-copy [[user@]host:]<filesystem>"
 271                                   " [[user@]host:]<filesystem>\n"
 272                 "hammer reblock[-btree/inodes/dirs/data] "
 273                         "<filesystem> [pack%%]\n"
 274                 "hammer pfs-status <dirpath>\n"
 275                 "hammer pfs-create <dirpath> [options]\n"
 276                 "hammer pfs-update <dirpath> [options]\n"
 277                 "hammer history[@offset[,len]] <file-1>...<file-N>\n"
 278                 "hammer -f blkdevs [-r] show\n"
 279                 "hammer -f blkdevs blockmap\n"
 280         );
 281         exit(exit_code);
 282 }
 283