cgdconfig.c revision 1.14 1 1.14 elric /* $NetBSD: cgdconfig.c,v 1.14 2005/03/30 19:56:05 elric Exp $ */
2 1.1 elric
3 1.1 elric /*-
4 1.5 elric * Copyright (c) 2002, 2003 The NetBSD Foundation, Inc.
5 1.1 elric * All rights reserved.
6 1.1 elric *
7 1.1 elric * This code is derived from software contributed to The NetBSD Foundation
8 1.1 elric * by Roland C. Dowdeswell.
9 1.1 elric *
10 1.1 elric * Redistribution and use in source and binary forms, with or without
11 1.1 elric * modification, are permitted provided that the following conditions
12 1.1 elric * are met:
13 1.1 elric * 1. Redistributions of source code must retain the above copyright
14 1.1 elric * notice, this list of conditions and the following disclaimer.
15 1.1 elric * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 elric * notice, this list of conditions and the following disclaimer in the
17 1.1 elric * documentation and/or other materials provided with the distribution.
18 1.1 elric * 3. All advertising materials mentioning features or use of this software
19 1.1 elric * must display the following acknowledgement:
20 1.1 elric * This product includes software developed by the NetBSD
21 1.1 elric * Foundation, Inc. and its contributors.
22 1.1 elric * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.1 elric * contributors may be used to endorse or promote products derived
24 1.1 elric * from this software without specific prior written permission.
25 1.1 elric *
26 1.1 elric * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.1 elric * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.1 elric * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.1 elric * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.1 elric * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.1 elric * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.1 elric * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.1 elric * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.1 elric * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.1 elric * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.1 elric * POSSIBILITY OF SUCH DAMAGE.
37 1.1 elric */
38 1.1 elric
39 1.1 elric #include <sys/cdefs.h>
40 1.1 elric #ifndef lint
41 1.1 elric __COPYRIGHT(
42 1.5 elric "@(#) Copyright (c) 2002, 2003\
43 1.1 elric The NetBSD Foundation, Inc. All rights reserved.");
44 1.14 elric __RCSID("$NetBSD: cgdconfig.c,v 1.14 2005/03/30 19:56:05 elric Exp $");
45 1.1 elric #endif
46 1.1 elric
47 1.5 elric #include <err.h>
48 1.1 elric #include <errno.h>
49 1.1 elric #include <fcntl.h>
50 1.1 elric #include <libgen.h>
51 1.1 elric #include <stdio.h>
52 1.1 elric #include <stdlib.h>
53 1.1 elric #include <string.h>
54 1.1 elric #include <unistd.h>
55 1.1 elric #include <util.h>
56 1.1 elric
57 1.1 elric #include <sys/ioctl.h>
58 1.3 elric #include <sys/disklabel.h>
59 1.14 elric #include <sys/mman.h>
60 1.1 elric #include <sys/param.h>
61 1.13 elric #include <sys/resource.h>
62 1.1 elric
63 1.1 elric #include <dev/cgdvar.h>
64 1.1 elric
65 1.5 elric #include <ufs/ffs/fs.h>
66 1.5 elric
67 1.1 elric #include "params.h"
68 1.1 elric #include "pkcs5_pbkdf2.h"
69 1.1 elric #include "utils.h"
70 1.1 elric
71 1.1 elric #define CGDCONFIG_DIR "/etc/cgd"
72 1.1 elric #define CGDCONFIG_CFILE CGDCONFIG_DIR "/cgd.conf"
73 1.1 elric
74 1.1 elric #define ACTION_CONFIGURE 0x1 /* configure, with paramsfile */
75 1.1 elric #define ACTION_UNCONFIGURE 0x2 /* unconfigure */
76 1.1 elric #define ACTION_GENERATE 0x3 /* generate a paramsfile */
77 1.5 elric #define ACTION_GENERATE_CONVERT 0x4 /* generate a ``dup'' paramsfile */
78 1.5 elric #define ACTION_CONFIGALL 0x5 /* configure all from config file */
79 1.5 elric #define ACTION_UNCONFIGALL 0x6 /* unconfigure all from config file */
80 1.5 elric #define ACTION_CONFIGSTDIN 0x7 /* configure, key from stdin */
81 1.1 elric
82 1.1 elric /* if nflag is set, do not configure/unconfigure the cgd's */
83 1.1 elric
84 1.1 elric int nflag = 0;
85 1.1 elric
86 1.3 elric static int configure(int, char **, struct params *, int);
87 1.1 elric static int configure_stdin(struct params *, int argc, char **);
88 1.1 elric static int generate(struct params *, int, char **, const char *);
89 1.5 elric static int generate_convert(struct params *, int, char **, const char *);
90 1.3 elric static int unconfigure(int, char **, struct params *, int);
91 1.3 elric static int do_all(const char *, int, char **,
92 1.3 elric int (*)(int, char **, struct params *, int));
93 1.1 elric
94 1.1 elric #define CONFIG_FLAGS_FROMALL 1 /* called from configure_all() */
95 1.1 elric #define CONFIG_FLAGS_FROMMAIN 2 /* called from main() */
96 1.1 elric
97 1.2 elric static int configure_params(int, const char *, const char *,
98 1.2 elric struct params *);
99 1.13 elric static void eliminate_cores(void);
100 1.5 elric static bits_t *getkey(const char *, struct keygen *, int);
101 1.5 elric static bits_t *getkey_storedkey(const char *, struct keygen *, int);
102 1.12 tv static bits_t *getkey_randomkey(const char *, struct keygen *, int, int);
103 1.10 dan static bits_t *getkey_pkcs5_pbkdf2(const char *, struct keygen *, int, int);
104 1.2 elric static int opendisk_werror(const char *, char *, int);
105 1.3 elric static int unconfigure_fd(int);
106 1.3 elric static int verify(struct params *, int);
107 1.5 elric static int verify_disklabel(int);
108 1.5 elric static int verify_ffs(int);
109 1.9 cb static int verify_reenter(struct params *);
110 1.1 elric
111 1.3 elric static void usage(void);
112 1.1 elric
113 1.1 elric /* Verbose Framework */
114 1.1 elric int verbose = 0;
115 1.1 elric
116 1.1 elric #define VERBOSE(x,y) if (verbose >= x) y
117 1.1 elric #define VPRINTF(x,y) if (verbose >= x) printf y
118 1.1 elric
119 1.1 elric static void
120 1.1 elric usage(void)
121 1.1 elric {
122 1.1 elric
123 1.3 elric fprintf(stderr, "usage: %s [-nv] [-V vmeth] cgd dev [paramsfile]\n",
124 1.1 elric getprogname());
125 1.1 elric fprintf(stderr, " %s -C [-nv] [-f configfile]\n", getprogname());
126 1.1 elric fprintf(stderr, " %s -U [-nv] [-f configfile]\n", getprogname());
127 1.5 elric fprintf(stderr, " %s -G [-nv] [-i ivmeth] [-k kgmeth] "
128 1.5 elric "[-o outfile] paramsfile\n", getprogname());
129 1.1 elric fprintf(stderr, " %s -g [-nv] [-i ivmeth] [-k kgmeth] "
130 1.5 elric "[-o outfile] alg [keylen]\n", getprogname());
131 1.1 elric fprintf(stderr, " %s -s [-nv] [-i ivmeth] cgd dev alg "
132 1.1 elric "[keylen]\n", getprogname());
133 1.1 elric fprintf(stderr, " %s -u [-nv] cgd\n", getprogname());
134 1.1 elric exit(1);
135 1.1 elric }
136 1.1 elric
137 1.1 elric int
138 1.1 elric main(int argc, char **argv)
139 1.1 elric {
140 1.5 elric struct params *p;
141 1.5 elric struct params *tp;
142 1.5 elric struct keygen *kg;
143 1.1 elric int action = ACTION_CONFIGURE;
144 1.1 elric int actions = 0;
145 1.1 elric int ch;
146 1.1 elric char cfile[FILENAME_MAX] = "";
147 1.1 elric char outfile[FILENAME_MAX] = "";
148 1.1 elric
149 1.13 elric eliminate_cores();
150 1.14 elric if (mlockall(MCL_FUTURE))
151 1.14 elric err(EXIT_FAILURE, "Can't lock memory");
152 1.1 elric setprogname(*argv);
153 1.5 elric p = params_new();
154 1.5 elric kg = NULL;
155 1.1 elric
156 1.5 elric while ((ch = getopt(argc, argv, "CGUV:b:f:gi:k:no:usv")) != -1)
157 1.1 elric switch (ch) {
158 1.1 elric case 'C':
159 1.1 elric action = ACTION_CONFIGALL;
160 1.1 elric actions++;
161 1.1 elric break;
162 1.5 elric case 'G':
163 1.5 elric action = ACTION_GENERATE_CONVERT;
164 1.5 elric actions++;
165 1.5 elric break;
166 1.1 elric case 'U':
167 1.1 elric action = ACTION_UNCONFIGALL;
168 1.1 elric actions++;
169 1.1 elric break;
170 1.3 elric case 'V':
171 1.5 elric tp = params_verify_method(string_fromcharstar(optarg));
172 1.5 elric if (!tp)
173 1.3 elric usage();
174 1.5 elric p = params_combine(p, tp);
175 1.3 elric break;
176 1.1 elric case 'b':
177 1.5 elric tp = params_bsize(atoi(optarg));
178 1.5 elric if (!tp)
179 1.1 elric usage();
180 1.5 elric p = params_combine(p, tp);
181 1.1 elric break;
182 1.1 elric case 'f':
183 1.8 itojun strlcpy(cfile, optarg, sizeof(cfile));
184 1.1 elric break;
185 1.1 elric case 'g':
186 1.1 elric action = ACTION_GENERATE;
187 1.1 elric actions++;
188 1.1 elric break;
189 1.1 elric case 'i':
190 1.5 elric tp = params_ivmeth(string_fromcharstar(optarg));
191 1.5 elric p = params_combine(p, tp);
192 1.1 elric break;
193 1.1 elric case 'k':
194 1.5 elric kg = keygen_method(string_fromcharstar(optarg));
195 1.5 elric if (!kg)
196 1.1 elric usage();
197 1.5 elric keygen_addlist(&p->keygen, kg);
198 1.1 elric break;
199 1.1 elric case 'n':
200 1.1 elric nflag = 1;
201 1.1 elric break;
202 1.1 elric case 'o':
203 1.8 itojun strlcpy(outfile, optarg, sizeof(outfile));
204 1.1 elric break;
205 1.1 elric case 's':
206 1.1 elric action = ACTION_CONFIGSTDIN;
207 1.1 elric actions++;
208 1.1 elric break;
209 1.1 elric
210 1.1 elric case 'u':
211 1.1 elric action = ACTION_UNCONFIGURE;
212 1.1 elric actions++;
213 1.1 elric break;
214 1.1 elric case 'v':
215 1.1 elric verbose++;
216 1.1 elric break;
217 1.1 elric default:
218 1.1 elric usage();
219 1.1 elric /* NOTREACHED */
220 1.1 elric }
221 1.1 elric
222 1.1 elric argc -= optind;
223 1.1 elric argv += optind;
224 1.1 elric
225 1.1 elric /* validate the consistency of the arguments */
226 1.1 elric
227 1.1 elric if (actions > 1)
228 1.1 elric usage();
229 1.1 elric
230 1.1 elric switch (action) {
231 1.1 elric case ACTION_CONFIGURE:
232 1.5 elric return configure(argc, argv, p, CONFIG_FLAGS_FROMMAIN);
233 1.1 elric case ACTION_UNCONFIGURE:
234 1.3 elric return unconfigure(argc, argv, NULL, CONFIG_FLAGS_FROMMAIN);
235 1.1 elric case ACTION_GENERATE:
236 1.5 elric return generate(p, argc, argv, outfile);
237 1.5 elric case ACTION_GENERATE_CONVERT:
238 1.5 elric return generate_convert(p, argc, argv, outfile);
239 1.1 elric case ACTION_CONFIGALL:
240 1.1 elric return do_all(cfile, argc, argv, configure);
241 1.1 elric case ACTION_UNCONFIGALL:
242 1.1 elric return do_all(cfile, argc, argv, unconfigure);
243 1.1 elric case ACTION_CONFIGSTDIN:
244 1.5 elric return configure_stdin(p, argc, argv);
245 1.1 elric default:
246 1.5 elric errx(EXIT_FAILURE, "undefined action");
247 1.1 elric }
248 1.1 elric /* NOTREACHED */
249 1.1 elric }
250 1.1 elric
251 1.5 elric static bits_t *
252 1.5 elric getkey(const char *dev, struct keygen *kg, int len)
253 1.1 elric {
254 1.5 elric bits_t *ret = NULL;
255 1.5 elric bits_t *tmp;
256 1.1 elric
257 1.5 elric VPRINTF(3, ("getkey(\"%s\", %p, %d) called\n", dev, kg, len));
258 1.5 elric for (; kg; kg=kg->next) {
259 1.5 elric switch (kg->kg_method) {
260 1.5 elric case KEYGEN_STOREDKEY:
261 1.5 elric tmp = getkey_storedkey(dev, kg, len);
262 1.5 elric break;
263 1.5 elric case KEYGEN_RANDOMKEY:
264 1.12 tv tmp = getkey_randomkey(dev, kg, len, 1);
265 1.12 tv break;
266 1.12 tv case KEYGEN_URANDOMKEY:
267 1.12 tv tmp = getkey_randomkey(dev, kg, len, 0);
268 1.5 elric break;
269 1.10 dan case KEYGEN_PKCS5_PBKDF2_SHA1:
270 1.10 dan tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 0);
271 1.10 dan break;
272 1.10 dan /* provide backwards compatibility for old config files */
273 1.10 dan case KEYGEN_PKCS5_PBKDF2_OLD:
274 1.10 dan tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 1);
275 1.5 elric break;
276 1.5 elric default:
277 1.5 elric warnx("unrecognised keygen method %d in getkey()",
278 1.5 elric kg->kg_method);
279 1.5 elric if (ret)
280 1.5 elric bits_free(ret);
281 1.5 elric return NULL;
282 1.5 elric }
283 1.5 elric
284 1.5 elric if (ret)
285 1.5 elric ret = bits_xor_d(tmp, ret);
286 1.5 elric else
287 1.5 elric ret = tmp;
288 1.1 elric }
289 1.5 elric
290 1.5 elric return ret;
291 1.5 elric }
292 1.5 elric
293 1.5 elric /*ARGSUSED*/
294 1.5 elric static bits_t *
295 1.5 elric getkey_storedkey(const char *target, struct keygen *kg, int keylen)
296 1.5 elric {
297 1.5 elric
298 1.5 elric return bits_dup(kg->kg_key);
299 1.1 elric }
300 1.1 elric
301 1.5 elric /*ARGSUSED*/
302 1.5 elric static bits_t *
303 1.12 tv getkey_randomkey(const char *target, struct keygen *kg, int keylen, int hard)
304 1.1 elric {
305 1.1 elric
306 1.12 tv return bits_getrandombits(keylen, hard);
307 1.1 elric }
308 1.1 elric
309 1.5 elric /*ARGSUSED*/
310 1.10 dan /*
311 1.10 dan * XXX take, and pass through, a compat flag that indicates whether we
312 1.10 dan * provide backwards compatibility with a previous bug. The previous
313 1.10 dan * behaviour is indicated by the keygen method pkcs5_pbkdf2, and a
314 1.10 dan * non-zero compat flag. The new default, and correct keygen method is
315 1.10 dan * called pcks5_pbkdf2/sha1. When the old method is removed, so will
316 1.10 dan * be the compat argument.
317 1.10 dan */
318 1.5 elric static bits_t *
319 1.10 dan getkey_pkcs5_pbkdf2(const char *target, struct keygen *kg, int keylen, int compat)
320 1.1 elric {
321 1.5 elric bits_t *ret;
322 1.5 elric char *passp;
323 1.5 elric char buf[1024];
324 1.5 elric u_int8_t *tmp;
325 1.1 elric
326 1.5 elric snprintf(buf, sizeof(buf), "%s's passphrase:", target);
327 1.1 elric passp = getpass(buf);
328 1.5 elric if (pkcs5_pbkdf2(&tmp, BITS2BYTES(keylen), passp, strlen(passp),
329 1.5 elric bits_getbuf(kg->kg_salt), BITS2BYTES(bits_len(kg->kg_salt)),
330 1.10 dan kg->kg_iterations, compat)) {
331 1.5 elric warnx("failed to generate PKCS#5 PBKDF2 key");
332 1.5 elric return NULL;
333 1.5 elric }
334 1.5 elric
335 1.5 elric ret = bits_new(tmp, keylen);
336 1.9 cb kg->kg_key = bits_dup(ret);
337 1.5 elric free(tmp);
338 1.1 elric return ret;
339 1.1 elric }
340 1.1 elric
341 1.5 elric /*ARGSUSED*/
342 1.1 elric static int
343 1.3 elric unconfigure(int argc, char **argv, struct params *inparams, int flags)
344 1.1 elric {
345 1.1 elric int fd;
346 1.1 elric int ret;
347 1.1 elric char buf[MAXPATHLEN] = "";
348 1.1 elric
349 1.1 elric /* only complain about additional arguments, if called from main() */
350 1.1 elric if (flags == CONFIG_FLAGS_FROMMAIN && argc != 1)
351 1.1 elric usage();
352 1.1 elric
353 1.1 elric /* if called from do_all(), then ensure that 2 or 3 args exist */
354 1.1 elric if (flags == CONFIG_FLAGS_FROMALL && (argc < 2 || argc > 3))
355 1.1 elric return -1;
356 1.1 elric
357 1.1 elric fd = opendisk(*argv, O_RDWR, buf, sizeof(buf), 1);
358 1.1 elric if (fd == -1) {
359 1.5 elric warn("can't open cgd \"%s\", \"%s\"", *argv, buf);
360 1.1 elric
361 1.1 elric /* this isn't fatal with nflag != 0 */
362 1.1 elric if (!nflag)
363 1.1 elric return errno;
364 1.1 elric }
365 1.1 elric
366 1.1 elric VPRINTF(1, ("%s (%s): clearing\n", *argv, buf));
367 1.1 elric
368 1.1 elric if (nflag)
369 1.1 elric return 0;
370 1.1 elric
371 1.3 elric ret = unconfigure_fd(fd);
372 1.3 elric close(fd);
373 1.3 elric return ret;
374 1.3 elric }
375 1.3 elric
376 1.3 elric static int
377 1.3 elric unconfigure_fd(int fd)
378 1.3 elric {
379 1.3 elric struct cgd_ioctl ci;
380 1.3 elric int ret;
381 1.3 elric
382 1.1 elric ret = ioctl(fd, CGDIOCCLR, &ci);
383 1.1 elric if (ret == -1) {
384 1.1 elric perror("ioctl");
385 1.3 elric return -1;
386 1.1 elric }
387 1.1 elric
388 1.1 elric return 0;
389 1.1 elric }
390 1.1 elric
391 1.5 elric /*ARGSUSED*/
392 1.1 elric static int
393 1.3 elric configure(int argc, char **argv, struct params *inparams, int flags)
394 1.1 elric {
395 1.5 elric struct params *p;
396 1.5 elric int fd;
397 1.5 elric int ret;
398 1.5 elric char pfile[FILENAME_MAX];
399 1.5 elric char cgdname[PATH_MAX];
400 1.1 elric
401 1.1 elric switch (argc) {
402 1.1 elric case 2:
403 1.1 elric strlcpy(pfile, CGDCONFIG_DIR, FILENAME_MAX);
404 1.1 elric strlcat(pfile, "/", FILENAME_MAX);
405 1.1 elric strlcat(pfile, basename(argv[1]), FILENAME_MAX);
406 1.1 elric break;
407 1.1 elric case 3:
408 1.1 elric strlcpy(pfile, argv[2], FILENAME_MAX);
409 1.1 elric break;
410 1.1 elric default:
411 1.1 elric /* print usage and exit, only if called from main() */
412 1.5 elric if (flags == CONFIG_FLAGS_FROMMAIN) {
413 1.5 elric warnx("wrong number of args");
414 1.1 elric usage();
415 1.5 elric }
416 1.1 elric return -1;
417 1.1 elric /* NOTREACHED */
418 1.1 elric }
419 1.1 elric
420 1.5 elric p = params_cget(pfile);
421 1.5 elric if (!p)
422 1.5 elric return -1;
423 1.2 elric
424 1.3 elric /*
425 1.5 elric * over-ride with command line specifications and fill in default
426 1.5 elric * values.
427 1.3 elric */
428 1.3 elric
429 1.5 elric p = params_combine(p, inparams);
430 1.5 elric ret = params_filldefaults(p);
431 1.5 elric if (ret) {
432 1.5 elric params_free(p);
433 1.5 elric return ret;
434 1.5 elric }
435 1.5 elric
436 1.5 elric if (!params_verify(p)) {
437 1.5 elric warnx("params invalid");
438 1.5 elric return -1;
439 1.5 elric }
440 1.3 elric
441 1.3 elric /*
442 1.3 elric * loop over configuring the disk and checking to see if it
443 1.3 elric * verifies properly. We open and close the disk device each
444 1.3 elric * time, because if the user passes us the block device we
445 1.3 elric * need to flush the buffer cache.
446 1.3 elric */
447 1.3 elric
448 1.3 elric for (;;) {
449 1.3 elric fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
450 1.3 elric if (fd == -1)
451 1.3 elric return -1;
452 1.3 elric
453 1.5 elric if (p->key)
454 1.5 elric bits_free(p->key);
455 1.5 elric
456 1.5 elric p->key = getkey(argv[1], p->keygen, p->keylen);
457 1.5 elric if (!p->key)
458 1.3 elric goto bail_err;
459 1.3 elric
460 1.5 elric ret = configure_params(fd, cgdname, argv[1], p);
461 1.3 elric if (ret)
462 1.3 elric goto bail_err;
463 1.3 elric
464 1.5 elric ret = verify(p, fd);
465 1.3 elric if (ret == -1)
466 1.3 elric goto bail_err;
467 1.3 elric if (!ret)
468 1.3 elric break;
469 1.2 elric
470 1.3 elric fprintf(stderr, "verification failed, please reenter "
471 1.3 elric "passphrase\n");
472 1.1 elric
473 1.3 elric unconfigure_fd(fd);
474 1.3 elric close(fd);
475 1.3 elric }
476 1.2 elric
477 1.5 elric params_free(p);
478 1.3 elric close(fd);
479 1.3 elric return 0;
480 1.3 elric bail_err:
481 1.5 elric params_free(p);
482 1.3 elric close(fd);
483 1.3 elric return -1;
484 1.1 elric }
485 1.1 elric
486 1.1 elric static int
487 1.1 elric configure_stdin(struct params *p, int argc, char **argv)
488 1.1 elric {
489 1.2 elric int fd;
490 1.1 elric int ret;
491 1.2 elric char cgdname[PATH_MAX];
492 1.1 elric
493 1.1 elric if (argc < 3 || argc > 4)
494 1.1 elric usage();
495 1.1 elric
496 1.5 elric p->algorithm = string_fromcharstar(argv[2]);
497 1.5 elric if (argc > 3)
498 1.5 elric p->keylen = atoi(argv[3]);
499 1.1 elric
500 1.1 elric ret = params_filldefaults(p);
501 1.1 elric if (ret)
502 1.1 elric return ret;
503 1.1 elric
504 1.2 elric fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
505 1.2 elric if (fd == -1)
506 1.2 elric return -1;
507 1.2 elric
508 1.5 elric p->key = bits_fget(stdin, p->keylen);
509 1.5 elric if (!p->key) {
510 1.5 elric warnx("failed to read key from stdin");
511 1.1 elric return -1;
512 1.5 elric }
513 1.1 elric
514 1.2 elric return configure_params(fd, cgdname, argv[1], p);
515 1.1 elric }
516 1.1 elric
517 1.1 elric static int
518 1.2 elric opendisk_werror(const char *cgd, char *buf, int buflen)
519 1.2 elric {
520 1.2 elric int fd;
521 1.2 elric
522 1.5 elric VPRINTF(3, ("opendisk_werror(%s, %s, %d) called.\n", cgd, buf, buflen));
523 1.5 elric
524 1.2 elric /* sanity */
525 1.2 elric if (!cgd || !buf)
526 1.2 elric return -1;
527 1.2 elric
528 1.2 elric if (nflag) {
529 1.8 itojun strlcpy(buf, cgd, buflen);
530 1.2 elric return 0;
531 1.2 elric }
532 1.2 elric
533 1.3 elric fd = opendisk(cgd, O_RDWR, buf, buflen, 0);
534 1.2 elric if (fd == -1)
535 1.5 elric warnx("can't open cgd \"%s\", \"%s\"", cgd, buf);
536 1.5 elric
537 1.2 elric return fd;
538 1.2 elric }
539 1.2 elric
540 1.2 elric static int
541 1.2 elric configure_params(int fd, const char *cgd, const char *dev, struct params *p)
542 1.1 elric {
543 1.1 elric struct cgd_ioctl ci;
544 1.1 elric int ret;
545 1.1 elric
546 1.1 elric /* sanity */
547 1.1 elric if (!cgd || !dev)
548 1.1 elric return -1;
549 1.1 elric
550 1.1 elric memset(&ci, 0x0, sizeof(ci));
551 1.1 elric ci.ci_disk = (char *)dev;
552 1.5 elric ci.ci_alg = (char *)string_tocharstar(p->algorithm);
553 1.5 elric ci.ci_ivmethod = (char *)string_tocharstar(p->ivmeth);
554 1.5 elric ci.ci_key = (char *)bits_getbuf(p->key);
555 1.1 elric ci.ci_keylen = p->keylen;
556 1.1 elric ci.ci_blocksize = p->bsize;
557 1.1 elric
558 1.1 elric VPRINTF(1, (" with alg %s keylen %d blocksize %d ivmethod %s\n",
559 1.5 elric string_tocharstar(p->algorithm), p->keylen, p->bsize,
560 1.5 elric string_tocharstar(p->ivmeth)));
561 1.5 elric VPRINTF(2, ("key: "));
562 1.5 elric VERBOSE(2, bits_fprint(stdout, p->key));
563 1.5 elric VPRINTF(2, ("\n"));
564 1.1 elric
565 1.1 elric if (nflag)
566 1.1 elric return 0;
567 1.1 elric
568 1.1 elric ret = ioctl(fd, CGDIOCSET, &ci);
569 1.1 elric if (ret == -1) {
570 1.1 elric perror("ioctl");
571 1.1 elric return errno;
572 1.1 elric }
573 1.1 elric
574 1.1 elric return 0;
575 1.1 elric }
576 1.1 elric
577 1.3 elric /*
578 1.3 elric * verify returns 0 for success, -1 for unrecoverable error, or 1 for retry.
579 1.3 elric */
580 1.3 elric
581 1.3 elric #define SCANSIZE 8192
582 1.3 elric
583 1.3 elric static int
584 1.3 elric verify(struct params *p, int fd)
585 1.3 elric {
586 1.3 elric
587 1.3 elric switch (p->verify_method) {
588 1.3 elric case VERIFY_NONE:
589 1.3 elric return 0;
590 1.3 elric case VERIFY_DISKLABEL:
591 1.5 elric return verify_disklabel(fd);
592 1.5 elric case VERIFY_FFS:
593 1.5 elric return verify_ffs(fd);
594 1.9 cb case VERIFY_REENTER:
595 1.9 cb return verify_reenter(p);
596 1.3 elric default:
597 1.5 elric warnx("unimplemented verification method");
598 1.3 elric return -1;
599 1.3 elric }
600 1.5 elric }
601 1.5 elric
602 1.5 elric static int
603 1.5 elric verify_disklabel(int fd)
604 1.5 elric {
605 1.5 elric struct disklabel l;
606 1.5 elric int ret;
607 1.5 elric char buf[SCANSIZE];
608 1.3 elric
609 1.3 elric /*
610 1.3 elric * we simply scan the first few blocks for a disklabel, ignoring
611 1.3 elric * any MBR/filecore sorts of logic. MSDOS and RiscOS can't read
612 1.3 elric * a cgd, anyway, so it is unlikely that there will be non-native
613 1.3 elric * partition information.
614 1.3 elric */
615 1.3 elric
616 1.3 elric ret = pread(fd, buf, 8192, 0);
617 1.3 elric if (ret == -1) {
618 1.5 elric warn("can't read disklabel area");
619 1.3 elric return -1;
620 1.3 elric }
621 1.3 elric
622 1.3 elric /* now scan for the disklabel */
623 1.3 elric
624 1.3 elric return disklabel_scan(&l, buf, sizeof(buf));
625 1.3 elric }
626 1.3 elric
627 1.7 fvdl static off_t sblock_try[] = SBLOCKSEARCH;
628 1.7 fvdl
629 1.1 elric static int
630 1.5 elric verify_ffs(int fd)
631 1.5 elric {
632 1.5 elric struct fs *fs;
633 1.7 fvdl int ret, i;
634 1.7 fvdl char buf[SBLOCKSIZE];
635 1.5 elric
636 1.7 fvdl for (i = 0; sblock_try[i] != -1; i++) {
637 1.7 fvdl ret = pread(fd, buf, sizeof(buf), sblock_try[i]);
638 1.7 fvdl if (ret == -1) {
639 1.7 fvdl warn("pread");
640 1.7 fvdl return 0;
641 1.7 fvdl }
642 1.7 fvdl fs = (struct fs *)buf;
643 1.7 fvdl switch (fs->fs_magic) {
644 1.7 fvdl case FS_UFS1_MAGIC:
645 1.7 fvdl case FS_UFS2_MAGIC:
646 1.7 fvdl case FS_UFS1_MAGIC_SWAPPED:
647 1.7 fvdl case FS_UFS2_MAGIC_SWAPPED:
648 1.7 fvdl return 0;
649 1.7 fvdl default:
650 1.7 fvdl continue;
651 1.7 fvdl }
652 1.5 elric }
653 1.5 elric return 1;
654 1.9 cb }
655 1.9 cb
656 1.9 cb static int
657 1.9 cb verify_reenter(struct params *p)
658 1.9 cb {
659 1.9 cb struct keygen *kg;
660 1.9 cb bits_t *orig_key, *key;
661 1.9 cb int ret;
662 1.9 cb
663 1.9 cb ret = 0;
664 1.9 cb for (kg = p->keygen; kg && !ret; kg = kg->next) {
665 1.10 dan if ((kg->kg_method != KEYGEN_PKCS5_PBKDF2_SHA1) &&
666 1.10 dan (kg->kg_method != KEYGEN_PKCS5_PBKDF2_OLD ))
667 1.9 cb continue;
668 1.9 cb
669 1.9 cb orig_key = kg->kg_key;
670 1.9 cb kg->kg_key = NULL;
671 1.9 cb
672 1.10 dan /* add a compat flag till the _OLD method goes away */
673 1.9 cb key = getkey_pkcs5_pbkdf2("re-enter device", kg,
674 1.10 dan bits_len(orig_key), kg->kg_method == KEYGEN_PKCS5_PBKDF2_OLD);
675 1.9 cb ret = !bits_match(key, orig_key);
676 1.9 cb
677 1.9 cb bits_free(key);
678 1.9 cb bits_free(kg->kg_key);
679 1.9 cb kg->kg_key = orig_key;
680 1.9 cb }
681 1.9 cb
682 1.9 cb return ret;
683 1.5 elric }
684 1.5 elric
685 1.5 elric static int
686 1.1 elric generate(struct params *p, int argc, char **argv, const char *outfile)
687 1.1 elric {
688 1.1 elric int ret;
689 1.1 elric
690 1.1 elric if (argc < 1 || argc > 2)
691 1.1 elric usage();
692 1.1 elric
693 1.5 elric p->algorithm = string_fromcharstar(argv[0]);
694 1.5 elric if (argc > 1)
695 1.5 elric p->keylen = atoi(argv[1]);
696 1.5 elric
697 1.5 elric ret = params_filldefaults(p);
698 1.1 elric if (ret)
699 1.1 elric return ret;
700 1.5 elric
701 1.5 elric if (!p->keygen) {
702 1.10 dan p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
703 1.5 elric if (!p->keygen)
704 1.5 elric return -1;
705 1.5 elric }
706 1.5 elric
707 1.5 elric if (keygen_filldefaults(p->keygen, p->keylen)) {
708 1.5 elric warnx("Failed to generate defaults for keygen");
709 1.5 elric return -1;
710 1.5 elric }
711 1.5 elric
712 1.5 elric if (!params_verify(p)) {
713 1.5 elric warnx("invalid parameters generated");
714 1.5 elric return -1;
715 1.5 elric }
716 1.5 elric
717 1.5 elric return params_cput(p, outfile);
718 1.5 elric }
719 1.5 elric
720 1.5 elric static int
721 1.5 elric generate_convert(struct params *p, int argc, char **argv, const char *outfile)
722 1.5 elric {
723 1.5 elric struct params *oldp;
724 1.5 elric struct keygen *kg;
725 1.5 elric
726 1.5 elric if (argc != 1)
727 1.5 elric usage();
728 1.5 elric
729 1.5 elric oldp = params_cget(*argv);
730 1.5 elric if (!oldp)
731 1.5 elric return -1;
732 1.5 elric
733 1.5 elric /* for sanity, we ensure that none of the keygens are randomkey */
734 1.5 elric for (kg=p->keygen; kg; kg=kg->next)
735 1.5 elric if (kg->kg_method == KEYGEN_RANDOMKEY)
736 1.5 elric goto bail;
737 1.5 elric for (kg=oldp->keygen; kg; kg=kg->next)
738 1.5 elric if (kg->kg_method == KEYGEN_RANDOMKEY)
739 1.5 elric goto bail;
740 1.5 elric
741 1.5 elric if (!params_verify(oldp)) {
742 1.5 elric warnx("invalid old parameters file \"%s\"", *argv);
743 1.5 elric return -1;
744 1.1 elric }
745 1.1 elric
746 1.5 elric oldp->key = getkey("old file", oldp->keygen, oldp->keylen);
747 1.5 elric
748 1.5 elric /* we copy across the non-keygen info, here. */
749 1.5 elric
750 1.5 elric string_free(p->algorithm);
751 1.5 elric string_free(p->ivmeth);
752 1.5 elric
753 1.5 elric p->algorithm = string_dup(oldp->algorithm);
754 1.5 elric p->ivmeth = string_dup(oldp->ivmeth);
755 1.5 elric p->keylen = oldp->keylen;
756 1.5 elric p->bsize = oldp->bsize;
757 1.5 elric if (p->verify_method == VERIFY_UNKNOWN)
758 1.5 elric p->verify_method = oldp->verify_method;
759 1.5 elric
760 1.5 elric params_free(oldp);
761 1.1 elric
762 1.5 elric if (!p->keygen) {
763 1.10 dan p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
764 1.5 elric if (!p->keygen)
765 1.1 elric return -1;
766 1.5 elric }
767 1.5 elric params_filldefaults(p);
768 1.6 elric keygen_filldefaults(p->keygen, p->keylen);
769 1.5 elric p->key = getkey("new file", p->keygen, p->keylen);
770 1.5 elric
771 1.5 elric kg = keygen_generate(KEYGEN_STOREDKEY);
772 1.5 elric kg->kg_key = bits_xor(p->key, oldp->key);
773 1.5 elric keygen_addlist(&p->keygen, kg);
774 1.5 elric
775 1.5 elric if (!params_verify(p)) {
776 1.5 elric warnx("can't generate new parameters file");
777 1.5 elric return -1;
778 1.1 elric }
779 1.1 elric
780 1.5 elric return params_cput(p, outfile);
781 1.5 elric bail:
782 1.5 elric params_free(oldp);
783 1.5 elric return -1;
784 1.1 elric }
785 1.1 elric
786 1.1 elric static int
787 1.1 elric do_all(const char *cfile, int argc, char **argv,
788 1.3 elric int (*conf)(int, char **, struct params *, int))
789 1.1 elric {
790 1.1 elric FILE *f;
791 1.1 elric size_t len;
792 1.1 elric size_t lineno;
793 1.1 elric int my_argc;
794 1.1 elric int ret;
795 1.1 elric const char *fn;
796 1.1 elric char *line;
797 1.1 elric char **my_argv;
798 1.1 elric
799 1.1 elric if (argc > 0)
800 1.1 elric usage();
801 1.1 elric
802 1.1 elric if (!cfile[0])
803 1.1 elric fn = CGDCONFIG_CFILE;
804 1.1 elric else
805 1.1 elric fn = cfile;
806 1.1 elric
807 1.1 elric f = fopen(fn, "r");
808 1.1 elric if (!f) {
809 1.5 elric warn("could not open config file \"%s\"", fn);
810 1.1 elric return -1;
811 1.1 elric }
812 1.1 elric
813 1.5 elric ret = chdir(CGDCONFIG_DIR);
814 1.5 elric if (ret == -1)
815 1.5 elric warn("could not chdir to %s", CGDCONFIG_DIR);
816 1.5 elric
817 1.1 elric ret = 0;
818 1.1 elric lineno = 0;
819 1.1 elric for (;;) {
820 1.1 elric line = fparseln(f, &len, &lineno, "\\\\#", FPARSELN_UNESCALL);
821 1.1 elric if (!line)
822 1.1 elric break;
823 1.1 elric if (!*line)
824 1.1 elric continue;
825 1.1 elric
826 1.1 elric my_argv = words(line, &my_argc);
827 1.3 elric ret = conf(my_argc, my_argv, NULL, CONFIG_FLAGS_FROMALL);
828 1.1 elric if (ret) {
829 1.5 elric warnx("action failed on \"%s\" line %lu", fn,
830 1.1 elric (u_long)lineno);
831 1.1 elric break;
832 1.1 elric }
833 1.1 elric words_free(my_argv, my_argc);
834 1.1 elric }
835 1.1 elric return ret;
836 1.1 elric }
837 1.13 elric
838 1.13 elric static void
839 1.13 elric eliminate_cores(void)
840 1.13 elric {
841 1.13 elric struct rlimit rlp;
842 1.13 elric int ret;
843 1.13 elric
844 1.13 elric rlp.rlim_cur = 0;
845 1.13 elric rlp.rlim_max = 0;
846 1.13 elric ret = setrlimit(RLIMIT_CORE, &rlp);
847 1.13 elric if (ret)
848 1.13 elric err(EXIT_FAILURE, "Can't disable cores");
849 1.13 elric }
850