sbin/hammer/cmd_config.c

   1 /*
   2  * Copyright (c) 2009 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
  35 #include "hammer.h"
  36 #include <sys/param.h>
  37 #include <sys/mount.h>
  38 #include <sys/types.h>
  39 #include <sys/stat.h>
  40 #include <unistd.h>
  41 #include <string.h>
  42 #include <time.h>
  43
  44 static void config_get(const char *dirpath, struct hammer_ioc_config *config);
  45 static void config_set(const char *dirpath, struct hammer_ioc_config *config);
  46 static void config_remove_path(void);
  47
  48 char *ConfigPath;
  49
  50 /*
  51  * hammer config [<fs> [configfile]]
  52  *
  53  * Prints out the hammer cleanup configuration for the specified HAMMER
  54  * filesystem(s) or the current filesystem.
  55  */
  56 void
  57 hammer_cmd_config(char **av, int ac)
  58 {
  59         struct hammer_ioc_config config;
  60         char *dirpath;
  61         ssize_t n;
  62         int fd;
  63
  64         bzero(&config, sizeof(config));
  65         if (ac == 0) {
  66                 config_get(".", &config);
  67                 if (config.head.error == 0) {
  68                         printf("%s", config.config.text);
  69                 } else {
  70                         errx(2, "hammer config: no configuration found");
  71                         /* not reached */
  72                 }
  73                 return;
  74         }
  75         dirpath = av[0];
  76         if (ac == 1) {
  77                 config_get(dirpath, &config);
  78                 if (config.head.error == 0) {
  79                         printf("%s", config.config.text);
  80                 } else {
  81                         errx(2, "hammer config: no configuration found");
  82                         /* not reached */
  83                 }
  84                 return;
  85         }
  86         config_get(dirpath, &config);   /* ignore errors */
  87         config.head.error = 0;
  88
  89         fd = open(av[1], O_RDONLY);
  90         if (fd < 0) {
  91                 err(2, "hammer config: %s", av[1]);
  92                 /* not reached */
  93         }
  94         n = read(fd, config.config.text, sizeof(config.config.text) - 1);
  95         if (n == sizeof(config.config.text) - 1) {
  96                 err(2, "hammer config: config file too big, limit %zu bytes",
  97                     sizeof(config.config.text) - 1);
  98                 /* not reached */
  99         }
 100         bzero(config.config.text + n, sizeof(config.config.text) - n);
 101         config_set(dirpath, &config);
 102         close(fd);
 103 }
 104
 105 /*
 106  * hammer viconfig [<fs>]
 107  */
 108 void
 109 hammer_cmd_viconfig(char **av, int ac)
 110 {
 111         struct hammer_ioc_config config;
 112         struct timeval times[2];
 113         const char *dirpath;
 114         struct stat st;
 115         char *runcmd, *editor, *tmp;
 116         char path[32];
 117         ssize_t n;
 118         int fd;
 119
 120         if (ac > 1) {
 121                 errx(1, "hammer viconfig: 0 or 1 argument (<fs>) only");
 122                 /* not reached */
 123         }
 124         if (ac == 0)
 125                 dirpath = ".";
 126         else
 127                 dirpath = av[0];
 128         config_get(dirpath, &config);
 129         if (config.head.error == ENOENT) {
 130                 snprintf(config.config.text, sizeof(config.config.text),
 131                         "%s",
 132                         "# No configuration present, here are some defaults\n"
 133                         "# you can uncomment.  Also remove these instructions\n"
 134                         "#\n"
 135                         "#snapshots 1d 60d\n"
 136                         "#prune     1d 5m\n"
 137                         "#rebalance 1d 5m\n"
 138                         "#dedup     1d 5m\n"
 139                         "#reblock   1d 5m\n"
 140                         "#recopy    30d 10m\n");
 141                 config.head.error = 0;
 142         }
 143         if (config.head.error) {
 144                 errx(2, "hammer viconfig: read config failed error: %s",
 145                         strerror(config.head.error));
 146                 /* not reached */
 147         }
 148
 149         /*
 150          * Edit a temporary file and write back if it was modified.
 151          * Adjust the mtime back one second so a quick edit is not
 152          * improperly detected as not having been modified.
 153          */
 154         snprintf(path, sizeof(path), "/tmp/configXXXXXXXXXX");
 155         mkstemp(path);
 156         ConfigPath = path;
 157         atexit(config_remove_path);
 158
 159         fd = open(path, O_RDWR|O_CREAT|O_TRUNC, 0600);
 160         if (fd < 0)
 161                 err(2, "hammer viconfig: creating temporary file %s", path);
 162         write(fd, config.config.text, strlen(config.config.text));
 163         if (fstat(fd, &st) < 0)
 164                 err(2, "hammer viconfig");
 165         times[0].tv_sec = st.st_mtime - 1;
 166         times[0].tv_usec = 0;
 167         times[1] = times[0];
 168         close(fd);
 169         utimes(path, times);
 170
 171         if ((tmp = getenv("EDITOR")) != NULL ||
 172             (tmp = getenv("VISUAL")) != NULL)
 173                 editor = strdup(tmp);
 174         else
 175                 editor = strdup("vi");
 176
 177         asprintf(&runcmd, "%s %s", editor, path);
 178         system(runcmd);
 179
 180         if (stat(path, &st) < 0)
 181                 err(2, "hammer viconfig: unable to stat file after vi");
 182         if (times[0].tv_sec == st.st_mtime) {
 183                 printf("hammer viconfig: no changes were made\n");
 184                 remove(path);
 185                 return;
 186         }
 187         fd = open(path, O_RDONLY);
 188         if (fd < 0)
 189                 err(2, "hammer viconfig: unable to read %s", path);
 190         remove(path);
 191         n = read(fd, config.config.text, sizeof(config.config.text) - 1);
 192         if (n < 0)
 193                 err(2, "hammer viconfig: unable to read %s", path);
 194         if (n == sizeof(config.config.text) - 1) {
 195                 err(2, "hammer config: config file too big, limit %zu bytes",
 196                     sizeof(config.config.text) - 1);
 197                 /* not reached */
 198         }
 199         bzero(config.config.text + n, sizeof(config.config.text) - n);
 200         config_set(dirpath, &config);
 201         free(editor);
 202         free(runcmd);
 203 }
 204
 205 static void
 206 config_get(const char *dirpath, struct hammer_ioc_config *config)
 207 {
 208         struct hammer_ioc_version version;
 209         int fd;
 210
 211         bzero(&version, sizeof(version));
 212         if ((fd = open(dirpath, O_RDONLY)) < 0)
 213                 err(2, "hammer config: unable to open directory %s", dirpath);
 214         if (ioctl(fd, HAMMERIOC_GET_VERSION, &version) < 0)
 215                 errx(2, "hammer config: not a HAMMER filesystem!");
 216
 217         if (ioctl(fd, HAMMERIOC_GET_CONFIG, config) < 0)
 218                 errx(2, "hammer config: config_get");
 219         close(fd);
 220 }
 221
 222 static void
 223 config_set(const char *dirpath, struct hammer_ioc_config *config)
 224 {
 225         struct hammer_ioc_version version;
 226         int fd;
 227
 228         bzero(&version, sizeof(version));
 229         if ((fd = open(dirpath, O_RDONLY)) < 0)
 230                 errx(2, "hammer config: unable to open directory %s", dirpath);
 231         if (ioctl(fd, HAMMERIOC_GET_VERSION, &version) < 0)
 232                 errx(2, "hammer config: not a HAMMER filesystem!");
 233         if (ioctl(fd, HAMMERIOC_SET_CONFIG, config) < 0)
 234                 err(2, "hammer config");
 235         close(fd);
 236 }
 237
 238 static void
 239 config_remove_path(void)
 240 {
 241         remove(ConfigPath);
 242 }