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.39 2008/09/20 04:23:21 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 int DelayOpt = 5;
  51 u_int64_t BandwidthOpt;
  52 const char *CyclePath;
  53 const char *LinkPath;
  54
  55 int
  56 main(int ac, char **av)
  57 {
  58         char *blkdevs = NULL;
  59         char *ptr;
  60         u_int32_t status;
  61         int ch;
  62
  63         while ((ch = getopt(ac, av, "b:c:dhf:i:rs:t:v2")) != -1) {
  64                 switch(ch) {
  65                 case '2':
  66                         TwoWayPipeOpt = 1;
  67                         break;
  68                 case 'b':
  69                         BandwidthOpt = strtoull(optarg, &ptr, 0);
  70                         switch(*ptr) {
  71                         case 'g':
  72                         case 'G':
  73                                 BandwidthOpt *= 1024;
  74                                 /* fall through */
  75                         case 'm':
  76                         case 'M':
  77                                 BandwidthOpt *= 1024;
  78                                 /* fall through */
  79                         case 'k':
  80                         case 'K':
  81                                 BandwidthOpt *= 1024;
  82                                 break;
  83                         default:
  84                                 usage(1);
  85                         }
  86                         break;
  87                 case 'c':
  88                         CyclePath = optarg;
  89                         break;
  90                 case 'd':
  91                         ++DebugOpt;
  92                         break;
  93                 case 'h':
  94                         usage(0);
  95                         /* not reached */
  96                 case 'i':
  97                         DelayOpt = strtol(optarg, NULL, 0);
  98                         break;
  99                 case 'r':
 100                         RecurseOpt = 1;
 101                         break;
 102                 case 'f':
 103                         blkdevs = optarg;
 104                         break;
 105                 case 's':
 106                         LinkPath = optarg;
 107                         break;
 108                 case 't':
 109                         TimeoutOpt = strtol(optarg, NULL, 0);
 110                         break;
 111                 case 'v':
 112                         ++VerboseOpt;
 113                         break;
 114                 default:
 115                         usage(1);
 116                         /* not reached */
 117                 }
 118         }
 119         ac -= optind;
 120         av += optind;
 121         if (ac < 1) {
 122                 usage(1);
 123                 /* not reached */
 124         }
 125
 126         signal(SIGALRM, sigalrm);
 127
 128         if (strcmp(av[0], "synctid") == 0) {
 129                 hammer_cmd_synctid(av + 1, ac - 1);
 130                 exit(0);
 131         }
 132         if (strcmp(av[0], "namekey") == 0) {
 133                 int64_t key;
 134
 135                 if (av[1] == NULL)
 136                         usage(1);
 137                 key = (int64_t)(crc32(av[1], strlen(av[1])) & 0x7FFFFFFF) << 32;
 138                 if (key == 0)
 139                         key |= 0x100000000LL;
 140                 printf("0x%016llx\n", key);
 141                 exit(0);
 142         }
 143         if (strcmp(av[0], "namekey32") == 0) {
 144                 int32_t key;
 145
 146                 if (av[1] == NULL)
 147                         usage(1);
 148                 key = crc32(av[1], strlen(av[1])) & 0x7FFFFFFF;
 149                 if (key == 0)
 150                         ++key;
 151                 printf("0x%08x\n", key);
 152                 exit(0);
 153         }
 154         if (strcmp(av[0], "pfs-status") == 0) {
 155                 hammer_cmd_pseudofs_status(av + 1, ac - 1);
 156                 exit(0);
 157         }
 158         if (strcmp(av[0], "pfs-master") == 0) {
 159                 hammer_cmd_pseudofs_create(av + 1, ac - 1, 0);
 160                 exit(0);
 161         }
 162         if (strcmp(av[0], "pfs-slave") == 0) {
 163                 hammer_cmd_pseudofs_create(av + 1, ac - 1, 1);
 164                 exit(0);
 165         }
 166         if (strcmp(av[0], "pfs-update") == 0) {
 167                 hammer_cmd_pseudofs_update(av + 1, ac - 1);
 168                 exit(0);
 169         }
 170         if (strcmp(av[0], "pfs-upgrade") == 0) {
 171                 hammer_cmd_pseudofs_upgrade(av + 1, ac - 1);
 172                 exit(0);
 173         }
 174         if (strcmp(av[0], "pfs-downgrade") == 0) {
 175                 hammer_cmd_pseudofs_downgrade(av + 1, ac - 1);
 176                 exit(0);
 177         }
 178         if (strcmp(av[0], "pfs-destroy") == 0) {
 179                 hammer_cmd_pseudofs_destroy(av + 1, ac - 1);
 180                 exit(0);
 181         }
 182         if (strcmp(av[0], "status") == 0) {
 183                 hammer_cmd_status(av + 1, ac - 1);
 184                 exit(0);
 185         }
 186         if (strcmp(av[0], "prune") == 0) {
 187                 hammer_cmd_softprune(av + 1, ac - 1, 0);
 188                 exit(0);
 189         }
 190         if (strcmp(av[0], "cleanup") == 0) {
 191                 hammer_cmd_cleanup(av + 1, ac - 1);
 192                 exit(0);
 193         }
 194         if (strcmp(av[0], "prune-everything") == 0) {
 195                 hammer_cmd_softprune(av + 1, ac - 1, 1);
 196                 exit(0);
 197         }
 198         if (strcmp(av[0], "snapshot") == 0) {
 199                 hammer_cmd_snapshot(av + 1, ac - 1);
 200                 exit(0);
 201         }
 202         if (strcmp(av[0], "bstats") == 0) {
 203                 hammer_cmd_bstats(av + 1, ac - 1);
 204                 exit(0);
 205         }
 206         if (strcmp(av[0], "iostats") == 0) {
 207                 hammer_cmd_iostats(av + 1, ac - 1);
 208                 exit(0);
 209         }
 210
 211         if (strncmp(av[0], "history", 7) == 0) {
 212                 hammer_cmd_history(av[0] + 7, av + 1, ac - 1);
 213                 exit(0);
 214         }
 215         if (strncmp(av[0], "reblock", 7) == 0) {
 216                 signal(SIGINT, sigalrm);
 217                 if (strcmp(av[0], "reblock") == 0)
 218                         hammer_cmd_reblock(av + 1, ac - 1, -1);
 219                 else if (strcmp(av[0], "reblock-btree") == 0)
 220                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_BTREE);
 221                 else if (strcmp(av[0], "reblock-inodes") == 0)
 222                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_INODES);
 223                 else if (strcmp(av[0], "reblock-dirs") == 0)
 224                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DIRS);
 225                 else if (strcmp(av[0], "reblock-data") == 0)
 226                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DATA);
 227                 else
 228                         usage(1);
 229                 exit(0);
 230         }
 231         if (strncmp(av[0], "mirror", 6) == 0) {
 232                 if (strcmp(av[0], "mirror-read") == 0)
 233                         hammer_cmd_mirror_read(av + 1, ac - 1, 0);
 234                 else if (strcmp(av[0], "mirror-read-stream") == 0)
 235                         hammer_cmd_mirror_read(av + 1, ac - 1, 1);
 236                 else if (strcmp(av[0], "mirror-write") == 0)
 237                         hammer_cmd_mirror_write(av + 1, ac - 1);
 238                 else if (strcmp(av[0], "mirror-copy") == 0)
 239                         hammer_cmd_mirror_copy(av + 1, ac - 1, 0);
 240                 else if (strcmp(av[0], "mirror-stream") == 0)
 241                         hammer_cmd_mirror_copy(av + 1, ac - 1, 1);
 242                 else if (strcmp(av[0], "mirror-dump") == 0)
 243                         hammer_cmd_mirror_dump();
 244                 else
 245                         usage(1);
 246                 exit(0);
 247         }
 248
 249         uuid_name_lookup(&Hammer_FSType, "DragonFly HAMMER", &status);
 250         if (status != uuid_s_ok) {
 251                 errx(1, "uuids file does not have the DragonFly "
 252                         "HAMMER filesystem type");
 253         }
 254
 255         if (strcmp(av[0], "show") == 0) {
 256                 hammer_off_t node_offset = (hammer_off_t)-1;
 257
 258                 hammer_parsedevs(blkdevs);
 259                 if (ac > 1)
 260                         sscanf(av[1], "%llx", &node_offset);
 261                 hammer_cmd_show(node_offset, 0, NULL, NULL);
 262                 exit(0);
 263         }
 264         if (strcmp(av[0], "blockmap") == 0) {
 265                 hammer_parsedevs(blkdevs);
 266                 hammer_cmd_blockmap();
 267                 exit(0);
 268         }
 269         usage(1);
 270         /* not reached */
 271         return(0);
 272 }
 273
 274 static
 275 void
 276 hammer_parsedevs(const char *blkdevs)
 277 {
 278         char *copy;
 279         char *volname;
 280
 281         if (blkdevs == NULL) {
 282                 errx(1, "A -f blkdev[:blkdev]* specification is required "
 283                         "for this command");
 284         }
 285
 286         copy = strdup(blkdevs);
 287         while ((volname = copy) != NULL) {
 288                 if ((copy = strchr(copy, ':')) != NULL)
 289                         *copy++ = 0;
 290                 setup_volume(-1, volname, 0, O_RDONLY);
 291         }
 292 }
 293
 294 static
 295 void
 296 sigalrm(int signo __unused)
 297 {
 298         /* do nothing (interrupts HAMMER ioctl) */
 299 }
 300
 301 static
 302 void
 303 usage(int exit_code)
 304 {
 305         fprintf(stderr,
 306                 "hammer -h\n"
 307                 "hammer [-v] [-t seconds] [-c cyclefile] command [argument ...]\n"
 308                 "hammer cleanup [<filesystem>]\n"
 309                 "hammer synctid <filesystem> [quick]\n"
 310                 "hammer namekey[32] <path>\n"
 311                 "hammer prune <softlink-dir>\n"
 312                 "hammer prune-everything <filesystem>\n"
 313                 "hammer snapshot [filesystem] <snapshot-dir>\n"
 314                 "hammer bstats [interval]\n"
 315                 "hammer iostats [interval]\n"
 316                 "hammer mirror-read <filesystem> [begin-tid]\n"
 317                 "hammer mirror-read-stream <filesystem> [begin-tid]\n"
 318                 "hammer mirror-write <filesystem>\n"
 319                 "hammer mirror-dump\n"
 320                 "hammer mirror-copy [[user@]host:]<filesystem>"
 321                                   " [[user@]host:]<filesystem>\n"
 322                 "hammer mirror-stream [[user@]host:]<filesystem>"
 323                                     " [[user@]host:]<filesystem>\n"
 324                 "hammer reblock[-btree/inodes/dirs/data] "
 325                         "<filesystem> [fill_percentage]\n"
 326                 "hammer pfs-status <dirpath> ...\n"
 327                 "hammer pfs-master <dirpath> [options]\n"
 328                 "hammer pfs-slave <dirpath> [options]\n"
 329                 "hammer pfs-update <dirpath> [options]\n"
 330                 "hammer pfs-upgrade <dirpath>\n"
 331                 "hammer pfs-downgrade <dirpath>\n"
 332                 "hammer pfs-destroy <dirpath>\n"
 333                 "hammer history[@offset[,len]] <file> ...\n"
 334                 "hammer -f blkdev[:blkdev]* [-r] show [offset]\n"
 335 #if 0
 336                 "hammer -f blkdev[:blkdev]* blockmap\n"
 337 #endif
 338         );
 339         exit(exit_code);
 340 }
 341