sbin/hammer: Cleanup blocks with a single statement
[dragonfly.git] / sbin / hammer / cmd_pfs.c
blobe4f623d8af70ff81102916d929c1bb62447bf6e5
1 /*
2 * Copyright (c) 2008 The DragonFly Project. All rights reserved.
4 * This code is derived from software contributed to The DragonFly Project
5 * by Matthew Dillon <dillon@backplane.com>
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
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.
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.
34 * $DragonFly: src/sbin/hammer/cmd_pseudofs.c,v 1.12 2008/10/08 21:01:54 thomas Exp $
37 #include <libgen.h>
39 #include "hammer.h"
41 static int scanpfsid(struct hammer_ioc_pseudofs_rw *pfs, const char *path);
42 static void parse_pfsd_options(char **av, int ac, hammer_pseudofs_data_t pfsd);
43 static void init_pfsd(hammer_pseudofs_data_t pfsd, int is_slave);
44 static void pseudofs_usage(int code);
45 static int timetosecs(char *str);
47 void
48 clrpfs(struct hammer_ioc_pseudofs_rw *pfs, hammer_pseudofs_data_t pfsd,
49 int pfs_id)
51 bzero(pfs, sizeof(*pfs));
53 if (pfsd)
54 pfs->ondisk = pfsd;
55 else
56 pfs->ondisk = malloc(sizeof(*pfs->ondisk));
57 bzero(pfs->ondisk, sizeof(*pfs->ondisk));
59 pfs->pfs_id = pfs_id;
60 pfs->bytes = sizeof(*pfs->ondisk);
61 pfs->version = HAMMER_IOC_PSEUDOFS_VERSION;
65 * If path is a symlink, return strdup'd path.
66 * If it's a directory via symlink, strip trailing /
67 * from strdup'd path and return the symlink.
69 static char*
70 getlink(const char *path)
72 int i;
73 char *linkpath;
74 struct stat st;
76 if (lstat(path, &st))
77 return(NULL);
78 linkpath = strdup(path);
80 if (S_ISDIR(st.st_mode)) {
81 i = strlen(linkpath) - 1;
82 while (i > 0 && linkpath[i] == '/')
83 linkpath[i--] = 0;
84 lstat(linkpath, &st);
86 if (S_ISLNK(st.st_mode))
87 return(linkpath);
89 free(linkpath);
90 return(NULL);
94 * Calculate the PFS id given a path to a file/directory or
95 * a @@%llx:%d softlink.
97 int
98 getpfs(struct hammer_ioc_pseudofs_rw *pfs, const char *path)
100 int fd;
102 clrpfs(pfs, NULL, -1);
105 * Extract the PFS id.
106 * dirname(path) is supposed to be a directory in root PFS.
108 if (scanpfsid(pfs, path) == 0)
109 path = dirname(path); /* strips trailing / first if any */
112 * Open the path regardless of scanpfsid() result, since some
113 * commands can take a regular file/directory (e.g. pfs-status).
115 fd = open(path, O_RDONLY);
116 if (fd < 0)
117 err(1, "Failed to open %s", path);
120 * If pfs.pfs_id has been set to non -1, the file descriptor fd
121 * could be any fd of HAMMER inodes since HAMMERIOC_GET_PSEUDOFS
122 * doesn't depend on inode attributes if it's set to a valid id.
124 if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, pfs) < 0)
125 err(1, "Cannot access %s", path);
127 return(fd);
131 * Extract the PFS id from path.
133 static int
134 scanpfsid(struct hammer_ioc_pseudofs_rw *pfs, const char *path)
136 char *linkpath;
137 char buf[64];
138 uintmax_t dummy_tid;
139 struct stat st;
141 if (stat(path, &st))
142 ; /* possibly slave PFS */
143 else if (S_ISDIR(st.st_mode))
144 ; /* possibly master or slave PFS */
145 else
146 return(-1); /* neither */
148 linkpath = getlink(path);
149 if (linkpath) {
151 * Read the symlink assuming it's a link to PFS.
153 bzero(buf, sizeof(buf));
154 if (readlink(linkpath, buf, sizeof(buf) - 1) < 0) {
155 free(linkpath);
156 return(-1);
158 free(linkpath);
159 path = buf;
163 * The symlink created by pfs-master|slave is just a symlink.
164 * One could happen to remove a symlink and relink PFS as
165 * # ln -s ./@@-1:00001 ./link
166 * which results PFS having something extra before @@.
167 * One could also directly use the PFS and results the same.
168 * Get rid of it before we extract the PFS id.
170 if (strchr(path, '/')) {
171 path = basename(path); /* strips trailing / first if any */
172 if (path == NULL)
173 err(1, "basename");
177 * Test and extract the PFS id from the link.
178 * "@@%jx:%d" covers both "@@-1:%05d" format for master PFS
179 * and "@@0x%016jx:%05d" format for slave PFS.
181 if (sscanf(path, "@@%jx:%d", &dummy_tid, &pfs->pfs_id) == 2) {
182 assert(pfs->pfs_id > 0);
183 return(0);
186 return(-1);
189 void
190 relpfs(int fd, struct hammer_ioc_pseudofs_rw *pfs)
192 if (fd >= 0)
193 close(fd);
194 if (pfs->ondisk) {
195 free(pfs->ondisk);
196 pfs->ondisk = NULL;
200 static void
201 print_pfs_status(char *path)
203 struct hammer_ioc_pseudofs_rw pfs;
204 int fd;
206 fd = getpfs(&pfs, path);
207 printf("%s\t", path);
208 if (fd < 0 || ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) < 0) {
209 printf("Invalid PFS path %s\n", path);
210 } else {
211 printf("PFS#%d {\n", pfs.pfs_id);
212 dump_pfsd(pfs.ondisk, fd);
213 printf("}\n");
215 if (fd >= 0)
216 close(fd);
217 if (pfs.ondisk)
218 free(pfs.ondisk);
219 relpfs(fd, &pfs);
222 void
223 hammer_cmd_pseudofs_status(char **av, int ac)
225 int i;
227 if (ac == 0) {
228 char buf[2] = "."; /* can't be readonly string */
229 print_pfs_status(buf);
230 return;
233 for (i = 0; i < ac; ++i)
234 print_pfs_status(av[i]);
237 void
238 hammer_cmd_pseudofs_create(char **av, int ac, int is_slave)
240 struct hammer_ioc_pseudofs_rw pfs;
241 struct hammer_pseudofs_data pfsd;
242 struct stat st;
243 const char *path;
244 char *dirpath;
245 char *linkpath;
246 int pfs_id;
247 int fd;
249 if (ac == 0)
250 pseudofs_usage(1);
251 path = av[0];
252 if (lstat(path, &st) == 0)
253 errx(1, "Cannot create %s, file exists!", path);
254 else if (path[strlen(path) - 1] == '/')
255 errx(1, "Invalid PFS path %s with trailing /", path);
258 * Figure out the directory prefix, taking care of degenerate
259 * cases.
261 dirpath = dirname(path);
262 fd = open(dirpath, O_RDONLY);
263 if (fd < 0)
264 err(1, "Cannot open directory %s", dirpath);
267 * Avoid foot-shooting. Don't let the user create a PFS
268 * softlink via a PFS. PFS softlinks may only be accessed
269 * via the master filesystem. Checking it here ensures
270 * other PFS commands access PFS under the master filesystem.
272 clrpfs(&pfs, &pfsd, -1);
274 ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs);
275 if (pfs.pfs_id != HAMMER_ROOT_PFSID) {
276 fprintf(stderr,
277 "You are attempting to access a PFS softlink "
278 "from a PFS. It may not represent the PFS\n"
279 "on the main filesystem mount that you "
280 "expect! You may only access PFS softlinks\n"
281 "via the main filesystem mount!\n");
282 exit(1);
285 for (pfs_id = 0; pfs_id < HAMMER_MAX_PFS; ++pfs_id) {
286 clrpfs(&pfs, &pfsd, pfs_id);
287 if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) < 0) {
288 if (errno != ENOENT)
289 err(1, "Cannot create %s", path);
290 break;
293 if (pfs_id == HAMMER_MAX_PFS)
294 errx(1, "Cannot create %s, all PFSs in use", path);
295 else if (pfs_id == HAMMER_ROOT_PFSID)
296 errx(1, "Fatal error: PFS#%d must exist", HAMMER_ROOT_PFSID);
299 * Create the new PFS
301 printf("Creating PFS#%d\t", pfs_id);
302 clrpfs(&pfs, &pfsd, pfs_id);
303 init_pfsd(&pfsd, is_slave);
305 if (ioctl(fd, HAMMERIOC_SET_PSEUDOFS, &pfs) < 0) {
306 printf("failed: %s\n", strerror(errno));
307 } else {
308 /* special symlink, must be exactly 10 characters */
309 asprintf(&linkpath, "@@PFS%05d", pfs_id);
310 if (symlink(linkpath, path) < 0) {
311 printf("failed: cannot create symlink: %s\n",
312 strerror(errno));
313 } else {
314 printf("succeeded!\n");
315 hammer_cmd_pseudofs_update(av, ac);
318 free(dirpath);
319 close(fd);
322 void
323 hammer_cmd_pseudofs_destroy(char **av, int ac)
325 struct hammer_ioc_pseudofs_rw pfs;
326 char *linkpath;
327 int fd;
328 int i;
330 if (ac == 0)
331 pseudofs_usage(1);
332 fd = getpfs(&pfs, av[0]);
334 if (pfs.pfs_id == HAMMER_ROOT_PFSID)
335 errx(1, "You cannot destroy PFS#%d", HAMMER_ROOT_PFSID);
337 printf("You have requested that PFS#%d (%s) be destroyed\n",
338 pfs.pfs_id, pfs.ondisk->label);
339 printf("This will irrevocably destroy all data on this PFS!!!!!\n");
340 printf("Do you really want to do this? [y/n] ");
341 fflush(stdout);
342 if (getyn() == 0)
343 errx(1, "No action taken on PFS#%d", pfs.pfs_id);
345 if (hammer_is_pfs_master(pfs.ondisk)) {
346 printf("This PFS is currently setup as a MASTER!\n");
347 printf("Are you absolutely sure you want to destroy it? [y/n] ");
348 fflush(stdout);
349 if (getyn() == 0)
350 errx(1, "No action taken on PFS#%d", pfs.pfs_id);
353 printf("Destroying PFS#%d (%s)", pfs.pfs_id, pfs.ondisk->label);
354 if (DebugOpt) {
355 printf("\n");
356 } else {
357 printf(" in");
358 for (i = 5; i; --i) {
359 printf(" %d", i);
360 fflush(stdout);
361 sleep(1);
363 printf(".. starting destruction pass\n");
367 * Remove the softlink on success.
369 if (ioctl(fd, HAMMERIOC_RMR_PSEUDOFS, &pfs) == 0) {
370 printf("pfs-destroy of PFS#%d succeeded!\n", pfs.pfs_id);
371 linkpath = getlink(av[0]);
372 if (linkpath) {
373 if (remove(linkpath) < 0)
374 err(1, "Unable to remove softlink %s", linkpath);
375 free(linkpath);
377 } else {
378 printf("pfs-destroy of PFS#%d failed: %s\n",
379 pfs.pfs_id, strerror(errno));
381 relpfs(fd, &pfs);
384 void
385 hammer_cmd_pseudofs_upgrade(char **av, int ac)
387 struct hammer_ioc_pseudofs_rw pfs;
388 int fd;
390 if (ac == 0)
391 pseudofs_usage(1);
392 fd = getpfs(&pfs, av[0]);
394 if (pfs.pfs_id == HAMMER_ROOT_PFSID)
395 errx(1, "You cannot upgrade PFS#%d"
396 " (It should already be a master)",
397 HAMMER_ROOT_PFSID);
398 else if (hammer_is_pfs_master(pfs.ondisk))
399 errx(1, "It is already a master");
401 if (ioctl(fd, HAMMERIOC_UPG_PSEUDOFS, &pfs) == 0)
402 printf("pfs-upgrade of PFS#%d (%s) succeeded\n",
403 pfs.pfs_id, pfs.ondisk->label);
404 else
405 err(1, "pfs-upgrade of PFS#%d (%s) failed",
406 pfs.pfs_id, pfs.ondisk->label);
407 relpfs(fd, &pfs);
410 void
411 hammer_cmd_pseudofs_downgrade(char **av, int ac)
413 struct hammer_ioc_pseudofs_rw pfs;
414 int fd;
416 if (ac == 0)
417 pseudofs_usage(1);
418 fd = getpfs(&pfs, av[0]);
420 if (pfs.pfs_id == HAMMER_ROOT_PFSID)
421 errx(1, "You cannot downgrade PFS#%d", HAMMER_ROOT_PFSID);
422 else if (hammer_is_pfs_slave(pfs.ondisk))
423 errx(1, "It is already a slave");
425 if (ioctl(fd, HAMMERIOC_DGD_PSEUDOFS, &pfs) == 0)
426 printf("pfs-downgrade of PFS#%d (%s) succeeded\n",
427 pfs.pfs_id, pfs.ondisk->label);
428 else
429 err(1, "pfs-downgrade of PFS#%d (%s) failed",
430 pfs.pfs_id, pfs.ondisk->label);
431 relpfs(fd, &pfs);
434 void
435 hammer_cmd_pseudofs_update(char **av, int ac)
437 struct hammer_ioc_pseudofs_rw pfs;
438 int fd;
440 if (ac == 0)
441 pseudofs_usage(1);
442 fd = getpfs(&pfs, av[0]);
444 printf("%s\n", av[0]);
445 fflush(stdout);
447 if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) == 0) {
448 parse_pfsd_options(av + 1, ac - 1, pfs.ondisk);
449 if (hammer_is_pfs_slave(pfs.ondisk) &&
450 pfs.pfs_id == HAMMER_ROOT_PFSID)
451 errx(1, "The real mount point cannot be made a PFS "
452 "slave, only PFS sub-directories can be made "
453 "slaves");
454 pfs.bytes = sizeof(*pfs.ondisk);
455 if (ioctl(fd, HAMMERIOC_SET_PSEUDOFS, &pfs) == 0) {
456 if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) == 0)
457 dump_pfsd(pfs.ondisk, fd);
458 else
459 err(1, "Unable to retrieve PFS configuration "
460 "after successful update");
461 } else {
462 err(1, "Unable to adjust PFS configuration");
465 relpfs(fd, &pfs);
468 static void
469 init_pfsd(hammer_pseudofs_data_t pfsd, int is_slave)
471 uint32_t status;
473 bzero(pfsd, sizeof(*pfsd));
474 pfsd->sync_beg_tid = 1;
475 pfsd->sync_end_tid = 1;
476 pfsd->sync_beg_ts = 0;
477 pfsd->sync_end_ts = 0;
478 uuid_create(&pfsd->shared_uuid, &status);
479 uuid_create(&pfsd->unique_uuid, &status);
480 if (is_slave)
481 pfsd->mirror_flags |= HAMMER_PFSD_SLAVE;
484 void
485 dump_pfsd(hammer_pseudofs_data_t pfsd, int fd)
487 struct hammer_ioc_version version;
488 uint32_t status;
489 char *str = NULL;
491 printf(" sync-beg-tid=0x%016jx\n", (uintmax_t)pfsd->sync_beg_tid);
492 printf(" sync-end-tid=0x%016jx\n", (uintmax_t)pfsd->sync_end_tid);
493 uuid_to_string(&pfsd->shared_uuid, &str, &status);
494 printf(" shared-uuid=%s\n", str);
495 free(str);
496 uuid_to_string(&pfsd->unique_uuid, &str, &status);
497 printf(" unique-uuid=%s\n", str);
498 free(str);
499 printf(" label=\"%s\"\n", pfsd->label);
500 if (pfsd->snapshots[0])
501 printf(" snapshots=\"%s\"\n", pfsd->snapshots);
502 if (pfsd->prune_min < (60 * 60 * 24))
503 printf(" prune-min=%02d:%02d:%02d\n",
504 pfsd->prune_min / 60 / 60 % 24,
505 pfsd->prune_min / 60 % 60,
506 pfsd->prune_min % 60);
507 else if (pfsd->prune_min % (60 * 60 * 24))
508 printf(" prune-min=%dd/%02d:%02d:%02d\n",
509 pfsd->prune_min / 60 / 60 / 24,
510 pfsd->prune_min / 60 / 60 % 24,
511 pfsd->prune_min / 60 % 60,
512 pfsd->prune_min % 60);
513 else
514 printf(" prune-min=%dd\n", pfsd->prune_min / 60 / 60 / 24);
516 if (hammer_is_pfs_slave(pfsd))
517 printf(" operating as a SLAVE\n");
518 else
519 printf(" operating as a MASTER\n");
522 * Snapshots directory cannot be shown when there is no fd since
523 * hammer version can't be retrieved. mirror-dump passes -1 because
524 * its input came from mirror-read output thus no path is available
525 * to open(2).
527 if (fd >= 0 && pfsd->snapshots[0] == 0) {
528 bzero(&version, sizeof(version));
529 if (ioctl(fd, HAMMERIOC_GET_VERSION, &version) < 0)
530 return;
531 HammerVersion = version.cur_version;
532 if (version.cur_version < 3) {
533 if (hammer_is_pfs_slave(pfsd))
534 printf(" snapshots directory not set for "
535 "slave\n");
536 else
537 printf(" snapshots directory for master "
538 "defaults to <pfs>/snapshots\n");
539 } else {
540 printf(" snapshots directory defaults to "
541 "/var/hammer/<pfs>\n");
546 static void
547 parse_pfsd_options(char **av, int ac, hammer_pseudofs_data_t pfsd)
549 char *cmd;
550 char *ptr;
551 int len;
552 uint32_t status;
554 while (ac) {
555 cmd = *av;
556 if ((ptr = strchr(cmd, '=')) != NULL)
557 *ptr++ = 0;
560 * Basic assignment value test
562 if (ptr == NULL)
563 errx(1, "option %s requires an assignment", cmd);
565 status = uuid_s_ok;
566 if (strcmp(cmd, "sync-beg-tid") == 0) {
567 pfsd->sync_beg_tid = strtoull(ptr, NULL, 16);
568 } else if (strcmp(cmd, "sync-end-tid") == 0) {
569 pfsd->sync_end_tid = strtoull(ptr, NULL, 16);
570 } else if (strcmp(cmd, "shared-uuid") == 0) {
571 uuid_from_string(ptr, &pfsd->shared_uuid, &status);
572 } else if (strcmp(cmd, "unique-uuid") == 0) {
573 uuid_from_string(ptr, &pfsd->unique_uuid, &status);
574 } else if (strcmp(cmd, "label") == 0) {
575 len = strlen(ptr);
576 if (ptr[0] == '"' && ptr[len-1] == '"') {
577 ptr[len-1] = 0;
578 ++ptr;
579 } else if (ptr[0] == '"') {
580 errx(1, "option %s: malformed string", cmd);
582 snprintf(pfsd->label, sizeof(pfsd->label), "%s", ptr);
583 } else if (strcmp(cmd, "snapshots") == 0) {
584 len = strlen(ptr);
585 if (ptr[0] != '/') {
586 fprintf(stderr,
587 "option %s: '%s' must be an "
588 "absolute path\n", cmd, ptr);
589 if (ptr[0] == 0)
590 fprintf(stderr,
591 "use 'snapshots-clear' "
592 "to unset snapshots dir\n");
593 exit(1);
595 if (len >= (int)sizeof(pfsd->snapshots))
596 errx(1, "option %s: path too long, %d "
597 "character limit", cmd, len);
598 snprintf(pfsd->snapshots, sizeof(pfsd->snapshots),
599 "%s", ptr);
600 } else if (strcmp(cmd, "snapshots-clear") == 0) {
601 pfsd->snapshots[0] = 0;
602 } else if (strcmp(cmd, "prune-min") == 0) {
603 pfsd->prune_min = timetosecs(ptr);
604 if (pfsd->prune_min < 0)
605 errx(1, "option %s: illegal time spec, "
606 "use Nd or [Nd/]hh[:mm[:ss]]", ptr);
607 } else {
608 errx(1, "invalid option: %s", cmd);
610 if (status != uuid_s_ok)
611 errx(1, "option %s: error parsing uuid %s", cmd, ptr);
612 --ac;
613 ++av;
617 static
618 void
619 pseudofs_usage(int code)
621 fprintf(stderr,
622 "hammer pfs-status <dirpath> ...\n"
623 "hammer pfs-master <dirpath> [options]\n"
624 "hammer pfs-slave <dirpath> [options]\n"
625 "hammer pfs-update <dirpath> [options]\n"
626 "hammer pfs-upgrade <dirpath>\n"
627 "hammer pfs-downgrade <dirpath>\n"
628 "hammer pfs-destroy <dirpath>\n"
629 "\n"
630 "options:\n"
631 " sync-beg-tid=0x16llx\n"
632 " sync-end-tid=0x16llx\n"
633 " shared-uuid=0x16llx\n"
634 " unique-uuid=0x16llx\n"
635 " label=\"string\"\n"
636 " snapshots=\"/path\"\n"
637 " snapshots-clear\n"
638 " prune-min=Nd\n"
639 " prune-min=[Nd/]hh[:mm[:ss]]\n"
641 exit(code);
645 * Convert time in the form [Nd/]hh[:mm[:ss]] to seconds.
647 * Return -1 if a parse error occurs.
648 * Return 0x7FFFFFFF if the time exceeds the maximum allowed.
650 static
652 timetosecs(char *str)
654 int days = 0;
655 int hrs = 0;
656 int mins = 0;
657 int secs = 0;
658 int n;
659 long long v;
660 char *ptr;
662 n = strtol(str, &ptr, 10);
663 if (n < 0)
664 return(-1);
665 if (*ptr == 'd') {
666 days = n;
667 ++ptr;
668 if (*ptr == '/')
669 n = strtol(ptr + 1, &ptr, 10);
670 else
671 n = 0;
673 if (n < 0)
674 return(-1);
675 hrs = n;
676 if (*ptr == ':') {
677 n = strtol(ptr + 1, &ptr, 10);
678 if (n < 0)
679 return(-1);
680 mins = n;
681 if (*ptr == ':') {
682 n = strtol(ptr + 1, &ptr, 10);
683 if (n < 0)
684 return(-1);
685 secs = n;
688 if (*ptr)
689 return(-1);
690 v = days * 24 * 60 * 60 + hrs * 60 * 60 + mins * 60 + secs;
691 if (v > 0x7FFFFFFF)
692 v = 0x7FFFFFFF;
693 return((int)v);