sbin/hammer/cmd_snapshot.c

   1 /*
   2  * Copyright (c) 2008 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/cmd_snapshot.c,v 1.5 2008/06/26 20:43:45 mneumann Exp $
  35  */
  36
  37 #include "hammer.h"
  38 #include <sys/param.h>
  39 #include <sys/mount.h>
  40 #include <sys/types.h>
  41 #include <sys/stat.h>
  42 #include <unistd.h>
  43 #include <string.h>
  44 #include <time.h>
  45
  46 #define DEFAULT_SNAPSHOT_NAME "snap-%Y%m%d-%H%M"
  47
  48 static void snapshot_usage(int exit_code);
  49
  50 /*
  51  * snapshot <softlink-dir-in-filesystem>
  52  * snapshot <filesystem> <softlink-dir>
  53  */
  54 void
  55 hammer_cmd_snapshot(char **av, int ac)
  56 {
  57         const char *filesystem;
  58         const char *softlink_dir;
  59         char *softlink_fmt;
  60         struct statfs buf;
  61         struct stat st;
  62         struct hammer_ioc_synctid synctid;
  63         char *from;
  64         char *to;
  65
  66         if (ac == 1) {
  67                 filesystem = NULL;
  68                 softlink_dir = av[0];
  69         }
  70         else if (ac == 2) {
  71                 filesystem = av[0];
  72                 softlink_dir = av[1];
  73         }
  74         else {
  75                 snapshot_usage(1);
  76         }
  77
  78         if (stat(softlink_dir, &st) == 0) {
  79                 if (!S_ISDIR(st.st_mode))
  80                         err(2, "File %s already exists", softlink_dir);
  81
  82                 if (filesystem == NULL) {
  83                         if (statfs(softlink_dir, &buf) != 0) {
  84                                 err(2, "Unable to determine filesystem of %s",
  85                                     softlink_dir);
  86                         }
  87                         filesystem = buf.f_mntonname;
  88                 }
  89
  90                 softlink_fmt = malloc(strlen(softlink_dir) + 1 + 1 +
  91                                       sizeof(DEFAULT_SNAPSHOT_NAME));
  92                 if (softlink_fmt == NULL)
  93                         err(2, "Failed to allocate string");
  94
  95                 strcpy(softlink_fmt, softlink_dir);
  96                 if (softlink_fmt[strlen(softlink_fmt)-1] != '/')
  97                         strcat(softlink_fmt, "/");
  98                 strcat(softlink_fmt, DEFAULT_SNAPSHOT_NAME);
  99         }
 100         else {
 101                 softlink_fmt = strdup(softlink_dir);
 102
 103                 if (filesystem == NULL) {
 104                         /*
 105                          * strip-off last '/path' segment to get the softlink
 106                          * directory, which we need to determine the filesystem
 107                          * we are on.
 108                          */
 109                         char *pos = strrchr(softlink_fmt, '/');
 110                         if (pos != NULL)
 111                                 *pos = '\0';
 112
 113                         if (stat(softlink_fmt, &st) != 0 ||
 114                             !S_ISDIR(st.st_mode)) {
 115                                 err(2, "Unable to determine softlink dir %s",
 116                                     softlink_fmt);
 117                         }
 118                         if (statfs(softlink_fmt, &buf) != 0) {
 119                                 err(2, "Unable to determine filesystem of %s",
 120                                     softlink_fmt);
 121                         }
 122                         filesystem = buf.f_mntonname;
 123
 124                         if (pos != NULL)
 125                                 *pos = '/';
 126                 }
 127         }
 128
 129         /*
 130          * Synctid
 131          */
 132         bzero(&synctid, sizeof(synctid));
 133         synctid.op = HAMMER_SYNCTID_SYNC2;
 134         int fd = open(filesystem, O_RDONLY);
 135         if (fd < 0)
 136                 err(2, "Unable to open %s", filesystem);
 137         if (ioctl(fd, HAMMERIOC_SYNCTID, &synctid) < 0)
 138                 err(2, "Synctid %s failed", filesystem);
 139         close(fd);
 140
 141         asprintf(&from, "%s@@0x%016llx", filesystem, synctid.tid);
 142         if (from == NULL)
 143                 err(2, "Couldn't generate string");
 144
 145         int sz = strlen(softlink_fmt) + 50;
 146         to = malloc(sz);
 147         if (to == NULL)
 148                 err(2, "Failed to allocate string");
 149
 150         time_t t = time(NULL);
 151         if (strftime(to, sz, softlink_fmt, localtime(&t)) == 0)
 152                 err(2, "String buffer too small");
 153
 154         if (symlink(from, to) != 0)
 155                 err(2, "Unable to symlink %s to %s", from, to);
 156
 157         printf("%s\n", to);
 158
 159         free(softlink_fmt);
 160         free(from);
 161         free(to);
 162 }
 163
 164 static
 165 void
 166 snapshot_usage(int exit_code)
 167 {
 168         fprintf(stderr, "hammer snapshot <snapshot-dir-in-filesystem>\n");
 169         fprintf(stderr, "hammer snapshot <filesystem> <snapshot-dir>\n");
 170         exit(exit_code);
 171 }
 172