Merge branch 'master' of ssh://crater.dragonflybsd.org/repository/git/dragonfly
[dragonfly.git] / sbin / hammer / cmd_config.c
blob9a60146616b4225899ed1c84ccf13ca28b18d14f
1 /*
2 * Copyright (c) 2009 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.
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>
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);
48 char *ConfigPath;
51 * hammer config [<fs> [configfile]]
53 * Prints out the hammer cleanup configuration for the specified HAMMER
54 * filesystem(s) or the current filesystem.
56 void
57 hammer_cmd_config(char **av, int ac)
59 struct hammer_ioc_config config;
60 char *dirpath;
61 ssize_t n;
62 int fd;
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 */
73 return;
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 */
84 return;
86 config_get(dirpath, &config); /* ignore errors */
87 config.head.error = 0;
89 fd = open(av[1], O_RDONLY);
90 n = read(fd, config.config.text, sizeof(config.config.text) - 1);
91 if (n == sizeof(config.config.text) - 1) {
92 err(2, "hammer config: config file too big, limit %zu bytes",
93 sizeof(config.config.text) - 1);
94 /* not reached */
96 bzero(config.config.text + n, sizeof(config.config.text) - n);
97 config_set(dirpath, &config);
98 close(fd);
102 * hammer viconfig [<fs>]
104 void
105 hammer_cmd_viconfig(char **av, int ac)
107 struct hammer_ioc_config config;
108 struct timeval times[2];
109 const char *dirpath;
110 struct stat st;
111 char *runcmd, *editor, *tmp;
112 char path[32];
113 ssize_t n;
114 int fd;
116 if (ac > 1) {
117 errx(1, "hammer viconfig: 0 or 1 argument (<fs>) only");
118 /* not reached */
120 if (ac == 0)
121 dirpath = ".";
122 else
123 dirpath = av[0];
124 config_get(dirpath, &config);
125 if (config.head.error == ENOENT) {
126 snprintf(config.config.text, sizeof(config.config.text),
127 "%s",
128 "# No configuration present, here are some defaults\n"
129 "# you can uncomment. Also remove these instructions\n"
130 "#\n"
131 "#snapshots 1d 60d\n"
132 "#prune 1d 5m\n"
133 "#rebalance 1d 5m\n"
134 "#reblock 1d 5m\n"
135 "#recopy 30d 10m\n");
136 config.head.error = 0;
138 if (config.head.error) {
139 errx(2, "hammer viconfig: read config failed error: %s",
140 strerror(config.head.error));
141 /* not reached */
145 * Edit a temporary file and write back if it was modified.
146 * Adjust the mtime back one second so a quick edit is not
147 * improperly detected as not having been modified.
149 snprintf(path, sizeof(path), "/tmp/configXXXXXXXXXX");
150 mkstemp(path);
151 ConfigPath = path;
152 atexit(config_remove_path);
154 fd = open(path, O_RDWR|O_CREAT|O_TRUNC, 0600);
155 if (fd < 0)
156 err(2, "hammer viconfig: creating temporary file %s", path);
157 write(fd, config.config.text, strlen(config.config.text));
158 if (fstat(fd, &st) < 0)
159 err(2, "hammer viconfig");
160 times[0].tv_sec = st.st_mtime - 1;
161 times[0].tv_usec = 0;
162 times[1] = times[0];
163 close(fd);
164 utimes(path, times);
166 if ((tmp = getenv("EDITOR")) != NULL ||
167 (tmp = getenv("VISUAL")) != NULL)
168 editor = strdup(tmp);
169 else
170 editor = strdup("vi");
172 asprintf(&runcmd, "%s %s", editor, path);
173 system(runcmd);
175 if (stat(path, &st) < 0)
176 err(2, "hammer viconfig: unable to stat file after vi");
177 if (times[0].tv_sec == st.st_mtime) {
178 printf("hammer viconfig: no changes were made\n");
179 remove(path);
180 return;
182 fd = open(path, O_RDONLY);
183 if (fd < 0)
184 err(2, "hammer viconfig: unable to read %s", path);
185 remove(path);
186 n = read(fd, config.config.text, sizeof(config.config.text) - 1);
187 if (n < 0)
188 err(2, "hammer viconfig: unable to read %s", path);
189 if (n == sizeof(config.config.text) - 1) {
190 err(2, "hammer config: config file too big, limit %zu bytes",
191 sizeof(config.config.text) - 1);
192 /* not reached */
194 bzero(config.config.text + n, sizeof(config.config.text) - n);
195 config_set(dirpath, &config);
196 free(editor);
197 free(runcmd);
200 static void
201 config_get(const char *dirpath, struct hammer_ioc_config *config)
203 struct hammer_ioc_version version;
204 int fd;
206 bzero(&version, sizeof(version));
207 if ((fd = open(dirpath, O_RDONLY)) < 0)
208 err(2, "hammer config: unable to open directory %s", dirpath);
209 if (ioctl(fd, HAMMERIOC_GET_VERSION, &version) < 0)
210 errx(2, "hammer config: not a HAMMER filesystem!");
212 if (ioctl(fd, HAMMERIOC_GET_CONFIG, config) < 0)
213 errx(2, "hammer config: config_get");
214 close(fd);
217 static void
218 config_set(const char *dirpath, struct hammer_ioc_config *config)
220 struct hammer_ioc_version version;
221 int fd;
223 bzero(&version, sizeof(version));
224 if ((fd = open(dirpath, O_RDONLY)) < 0)
225 errx(2, "hammer config: unable to open directory %s", dirpath);
226 if (ioctl(fd, HAMMERIOC_GET_VERSION, &version) < 0)
227 errx(2, "hammer config: not a HAMMER filesystem!");
228 if (ioctl(fd, HAMMERIOC_SET_CONFIG, config) < 0)
229 err(2, "hammer config");
230 close(fd);
233 static void
234 config_remove_path(void)
236 remove(ConfigPath);