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.22 2008/06/17 04:03:38 dillon Exp $
  35  */
  36
  37 #include "hammer.h"
  38 #include <signal.h>
  39 #include <math.h>
  40
  41 static void hammer_parsetime(u_int64_t *tidp, const char *timestr);
  42 static void hammer_parsedevs(const char *blkdevs);
  43 static void sigalrm(int signo);
  44 static void usage(int exit_code);
  45
  46 int RecurseOpt;
  47 int VerboseOpt;
  48 int NoSyncOpt;
  49 const char *CyclePath;
  50 const char *LinkPath;
  51
  52 int
  53 main(int ac, char **av)
  54 {
  55         struct timeval tv;
  56         u_int64_t tid;
  57         int ch;
  58         int timeout = 0;
  59         u_int32_t status;
  60         char *blkdevs = NULL;
  61
  62         while ((ch = getopt(ac, av, "c:dhf:rs:t:v")) != -1) {
  63                 switch(ch) {
  64                 case 'c':
  65                         CyclePath = optarg;
  66                         break;
  67                 case 'd':
  68                         ++DebugOpt;
  69                         break;
  70                 case 'h':
  71                         usage(0);
  72                         /* not reached */
  73                 case 'r':
  74                         RecurseOpt = 1;
  75                         break;
  76                 case 'f':
  77                         blkdevs = optarg;
  78                         break;
  79                 case 's':
  80                         LinkPath = optarg;
  81                         break;
  82                 case 't':
  83                         timeout = strtol(optarg, NULL, 0);
  84                         break;
  85                 case 'v':
  86                         ++VerboseOpt;
  87                         break;
  88                 default:
  89                         usage(1);
  90                         /* not reached */
  91                 }
  92         }
  93         ac -= optind;
  94         av += optind;
  95         if (ac < 1) {
  96                 usage(1);
  97                 /* not reached */
  98         }
  99
 100         if (timeout > 0) {
 101                 signal(SIGALRM, sigalrm);
 102                 alarm(timeout);
 103         }
 104
 105         if (strcmp(av[0], "now") == 0) {
 106                 tid = (hammer_tid_t)time(NULL) * 1000000000LLU;
 107                 printf("0x%08x\n", (int)(tid / 1000000000LL));
 108                 exit(0);
 109         }
 110         if (strcmp(av[0], "now64") == 0) {
 111                 gettimeofday(&tv, NULL);
 112                 tid = (hammer_tid_t)tv.tv_sec * 1000000000LLU +
 113                         tv.tv_usec * 1000LLU;
 114                 printf("0x%016llx\n", tid);
 115                 exit(0);
 116         }
 117         if (strcmp(av[0], "synctid") == 0) {
 118                 hammer_cmd_synctid(av + 1, ac - 1);
 119                 exit(0);
 120         }
 121         if (strcmp(av[0], "stamp") == 0) {
 122                 if (av[1] == NULL)
 123                         usage(1);
 124                 hammer_parsetime(&tid, av[1]);
 125                 printf("0x%08x\n", (int)(tid / 1000000000LL));
 126                 exit(0);
 127         }
 128         if (strcmp(av[0], "stamp64") == 0) {
 129                 if (av[1] == NULL)
 130                         usage(1);
 131                 hammer_parsetime(&tid, av[1]);
 132                 printf("0x%016llx\n", tid);
 133                 exit(0);
 134         }
 135         if (strcmp(av[0], "date") == 0) {
 136                 time_t t;
 137
 138                 if (av[1] == NULL)
 139                         usage(1);
 140                 tid = strtoull(av[1], NULL, 16);
 141                 if (tid >= 0x100000000LLU)
 142                         tid /= 1000000000LLU;
 143                 t = (time_t)tid;
 144                 printf("%s", ctime(&t));
 145                 exit(0);
 146         }
 147         if (strcmp(av[0], "namekey") == 0) {
 148                 int64_t key;
 149
 150                 if (av[1] == NULL)
 151                         usage(1);
 152                 key = (int64_t)(crc32(av[1], strlen(av[1])) & 0x7FFFFFFF) << 32;
 153                 if (key == 0)
 154                         key |= 0x100000000LL;
 155                 printf("0x%016llx\n", key);
 156                 exit(0);
 157         }
 158         if (strcmp(av[0], "namekey32") == 0) {
 159                 int32_t key;
 160
 161                 if (av[1] == NULL)
 162                         usage(1);
 163                 key = crc32(av[1], strlen(av[1])) & 0x7FFFFFFF;
 164                 if (key == 0)
 165                         ++key;
 166                 printf("0x%08x\n", key);
 167                 exit(0);
 168         }
 169         if (strcmp(av[0], "prune") == 0) {
 170                 hammer_cmd_prune(av + 1, ac - 1);
 171                 exit(0);
 172         }
 173         if (strcmp(av[0], "softprune") == 0) {
 174                 hammer_cmd_softprune(av + 1, ac - 1);
 175                 exit(0);
 176         }
 177         if (strcmp(av[0], "bstats") == 0) {
 178                 hammer_cmd_bstats(av + 1, ac - 1);
 179                 exit(0);
 180         }
 181         if (strcmp(av[0], "iostats") == 0) {
 182                 hammer_cmd_iostats(av + 1, ac - 1);
 183                 exit(0);
 184         }
 185
 186         if (strncmp(av[0], "history", 7) == 0) {
 187                 hammer_cmd_history(av[0] + 7, av + 1, ac - 1);
 188                 exit(0);
 189         }
 190         if (strncmp(av[0], "reblock", 7) == 0) {
 191                 if (strcmp(av[0], "reblock") == 0)
 192                         hammer_cmd_reblock(av + 1, ac - 1, -1);
 193                 else if (strcmp(av[0], "reblock-btree") == 0)
 194                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_BTREE);
 195                 else if (strcmp(av[0], "reblock-inodes") == 0)
 196                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_INODES);
 197                 else if (strcmp(av[0], "reblock-dirs") == 0)
 198                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DIRS);
 199                 else if (strcmp(av[0], "reblock-data") == 0)
 200                         hammer_cmd_reblock(av + 1, ac - 1, HAMMER_IOC_DO_DATA);
 201                 else
 202                         usage(1);
 203                 exit(0);
 204         }
 205
 206         uuid_name_lookup(&Hammer_FSType, "DragonFly HAMMER", &status);
 207         if (status != uuid_s_ok) {
 208                 errx(1, "uuids file does not have the DragonFly "
 209                         "HAMMER filesystem type");
 210         }
 211
 212         if (strcmp(av[0], "show") == 0) {
 213                 hammer_off_t node_offset = (hammer_off_t)-1;
 214
 215                 hammer_parsedevs(blkdevs);
 216                 if (ac > 1)
 217                         sscanf(av[1], "%llx", &node_offset);
 218                 hammer_cmd_show(node_offset, 0, NULL, NULL);
 219                 exit(0);
 220         }
 221         if (strcmp(av[0], "blockmap") == 0) {
 222                 hammer_parsedevs(blkdevs);
 223                 hammer_cmd_blockmap();
 224                 exit(0);
 225         }
 226         usage(1);
 227         /* not reached */
 228         return(0);
 229 }
 230
 231 /*
 232  * Parse a timestamp for the mount point
 233  *
 234  * yyyymmddhhmmss
 235  * -N[s/h/d/m/y]
 236  */
 237 static
 238 void
 239 hammer_parsetime(u_int64_t *tidp, const char *timestr)
 240 {
 241         struct timeval tv;
 242         struct tm tm;
 243         int32_t n;
 244         char c;
 245
 246         gettimeofday(&tv, NULL);
 247
 248         if (*timestr == 0)
 249                 usage(1);
 250
 251         if (isalpha(timestr[strlen(timestr)-1])) {
 252                 if (sscanf(timestr, "%d%c", &n, &c) != 2)
 253                         usage(1);
 254                 switch(c) {
 255                 case 'Y':
 256                         n *= 365;
 257                         goto days;
 258                 case 'M':
 259                         n *= 30;
 260                         /* fall through */
 261                 case 'D':
 262                 days:
 263                         n *= 24;
 264                         /* fall through */
 265                 case 'h':
 266                         n *= 60;
 267                         /* fall through */
 268                 case 'm':
 269                         n *= 60;
 270                         /* fall through */
 271                 case 's':
 272                         tv.tv_sec -= n;
 273                         break;
 274                 default:
 275                         usage(1);
 276                 }
 277         } else {
 278                 double seconds = 0;
 279
 280                 localtime_r(&tv.tv_sec, &tm);
 281                 seconds = (double)tm.tm_sec;
 282                 tm.tm_year += 1900;
 283                 tm.tm_mon += 1;
 284                 n = sscanf(timestr, "%4d%2d%2d:%2d%2d%lf",
 285                            &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
 286                            &tm.tm_hour, &tm.tm_min, &seconds);
 287                 tm.tm_mon -= 1;
 288                 tm.tm_year -= 1900;
 289                 /* if [:hhmmss] is omitted, assume :000000.0 */
 290                 if (n < 4)
 291                         tm.tm_hour = tm.tm_min = tm.tm_sec = 0;
 292                 else
 293                         tm.tm_sec = (int)seconds;
 294                 tv.tv_sec = mktime(&tm);
 295                 tv.tv_usec = (int)((seconds - floor(seconds)) * 1000000.0);
 296         }
 297         *tidp = (u_int64_t)tv.tv_sec * 1000000000LLU +
 298                 tv.tv_usec * 1000LLU;
 299 }
 300
 301 static
 302 void
 303 hammer_parsedevs(const char *blkdevs)
 304 {
 305         char *copy;
 306         char *volname;
 307
 308         if (blkdevs == NULL) {
 309                 errx(1, "A -f blkdevs specification is required "
 310                         "for this command");
 311         }
 312
 313         copy = strdup(blkdevs);
 314         while ((volname = copy) != NULL) {
 315                 if ((copy = strchr(copy, ':')) != NULL)
 316                         *copy++ = 0;
 317                 setup_volume(-1, volname, 0, O_RDONLY);
 318         }
 319 }
 320
 321 static
 322 void
 323 sigalrm(int signo __unused)
 324 {
 325         /* do nothing (interrupts HAMMER ioctl) */
 326 }
 327
 328 static
 329 void
 330 usage(int exit_code)
 331 {
 332         fprintf(stderr,
 333                 "hammer -h\n"
 334                 "hammer [-x] now[64]\n"
 335                 "hammer [-t timeout] [-c cyclefile] ....\n"
 336                 "hammer stamp[64] <time>\n"
 337                 "hammer softprune <softlink-dir>\n"
 338                 "hammer bstats <interval>\n"
 339                 "hammer iostats <interval>\n"
 340                 "hammer prune <filesystem> [using <configfile>]\n"
 341                 "hammer prune <filesystem> from <modulo_time> to "
 342                                 "<modulo_time> every <modulo_time>\n"
 343                 "hammer prune <filesystem> from <modulo_time> everything\n"
 344                 "hammer prune <filesystem> everything\n"
 345                 "hammer reblock <filesystem> [compact%%] (default 90%%)\n"
 346                 "hammer history[@offset[,len]] <file-1>...<file-N>\n"
 347                 "hammer -f blkdevs [-r] show\n"
 348                 "hammer -f blkdevs blockmap\n"
 349         );
 350         fprintf(stderr, "time: +n[s/m/h/D/M/Y]\n"
 351                         "time: yyyymmdd[:hhmmss]\n"
 352                         "modulo_time: n{s,m,h,d,M,y}\n");
 353         exit(exit_code);
 354 }
 355