cgdconfig.c revision 1.34 1 1.34 christos /* $NetBSD: cgdconfig.c,v 1.34 2012/12/05 02:23:20 christos 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 *
19 1.1 elric * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.1 elric * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.1 elric * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.1 elric * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.1 elric * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.1 elric * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.1 elric * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.1 elric * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.1 elric * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.1 elric * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.1 elric * POSSIBILITY OF SUCH DAMAGE.
30 1.1 elric */
31 1.1 elric
32 1.1 elric #include <sys/cdefs.h>
33 1.1 elric #ifndef lint
34 1.24 lukem __COPYRIGHT("@(#) Copyright (c) 2002, 2003\
35 1.24 lukem The NetBSD Foundation, Inc. All rights reserved.");
36 1.34 christos __RCSID("$NetBSD: cgdconfig.c,v 1.34 2012/12/05 02:23:20 christos Exp $");
37 1.1 elric #endif
38 1.1 elric
39 1.5 elric #include <err.h>
40 1.1 elric #include <errno.h>
41 1.1 elric #include <fcntl.h>
42 1.1 elric #include <libgen.h>
43 1.1 elric #include <stdio.h>
44 1.1 elric #include <stdlib.h>
45 1.1 elric #include <string.h>
46 1.1 elric #include <unistd.h>
47 1.1 elric #include <util.h>
48 1.1 elric
49 1.1 elric #include <sys/ioctl.h>
50 1.3 elric #include <sys/disklabel.h>
51 1.14 elric #include <sys/mman.h>
52 1.1 elric #include <sys/param.h>
53 1.13 elric #include <sys/resource.h>
54 1.34 christos #include <sys/statvfs.h>
55 1.1 elric
56 1.1 elric #include <dev/cgdvar.h>
57 1.1 elric
58 1.5 elric #include <ufs/ffs/fs.h>
59 1.5 elric
60 1.1 elric #include "params.h"
61 1.1 elric #include "pkcs5_pbkdf2.h"
62 1.1 elric #include "utils.h"
63 1.28 pooka #include "cgdconfig.h"
64 1.32 pooka #include "prog_ops.h"
65 1.1 elric
66 1.1 elric #define CGDCONFIG_DIR "/etc/cgd"
67 1.1 elric #define CGDCONFIG_CFILE CGDCONFIG_DIR "/cgd.conf"
68 1.1 elric
69 1.17 cbiere enum action {
70 1.17 cbiere ACTION_DEFAULT, /* default -> configure */
71 1.17 cbiere ACTION_CONFIGURE, /* configure, with paramsfile */
72 1.17 cbiere ACTION_UNCONFIGURE, /* unconfigure */
73 1.17 cbiere ACTION_GENERATE, /* generate a paramsfile */
74 1.17 cbiere ACTION_GENERATE_CONVERT, /* generate a ``dup'' paramsfile */
75 1.17 cbiere ACTION_CONFIGALL, /* configure all from config file */
76 1.17 cbiere ACTION_UNCONFIGALL, /* unconfigure all from config file */
77 1.34 christos ACTION_CONFIGSTDIN, /* configure, key from stdin */
78 1.34 christos ACTION_LIST /* list configured devices */
79 1.17 cbiere };
80 1.1 elric
81 1.1 elric /* if nflag is set, do not configure/unconfigure the cgd's */
82 1.1 elric
83 1.1 elric int nflag = 0;
84 1.1 elric
85 1.22 elric /* if pflag is set to PFLAG_STDIN read from stdin rather than getpass(3) */
86 1.22 elric
87 1.22 elric #define PFLAG_GETPASS 0x01
88 1.22 elric #define PFLAG_STDIN 0x02
89 1.22 elric int pflag = PFLAG_GETPASS;
90 1.22 elric
91 1.3 elric static int configure(int, char **, struct params *, int);
92 1.1 elric static int configure_stdin(struct params *, int argc, char **);
93 1.1 elric static int generate(struct params *, int, char **, const char *);
94 1.5 elric static int generate_convert(struct params *, int, char **, const char *);
95 1.3 elric static int unconfigure(int, char **, struct params *, int);
96 1.3 elric static int do_all(const char *, int, char **,
97 1.3 elric int (*)(int, char **, struct params *, int));
98 1.34 christos static int do_list(int, char **);
99 1.1 elric
100 1.1 elric #define CONFIG_FLAGS_FROMALL 1 /* called from configure_all() */
101 1.1 elric #define CONFIG_FLAGS_FROMMAIN 2 /* called from main() */
102 1.1 elric
103 1.2 elric static int configure_params(int, const char *, const char *,
104 1.2 elric struct params *);
105 1.13 elric static void eliminate_cores(void);
106 1.19 christos static bits_t *getkey(const char *, struct keygen *, size_t);
107 1.19 christos static bits_t *getkey_storedkey(const char *, struct keygen *, size_t);
108 1.19 christos static bits_t *getkey_randomkey(const char *, struct keygen *, size_t, int);
109 1.23 elric static bits_t *getkey_pkcs5_pbkdf2(const char *, struct keygen *, size_t,
110 1.23 elric int);
111 1.23 elric static bits_t *getkey_shell_cmd(const char *, struct keygen *, size_t);
112 1.22 elric static char *maybe_getpass(char *);
113 1.17 cbiere static int opendisk_werror(const char *, char *, size_t);
114 1.3 elric static int unconfigure_fd(int);
115 1.3 elric static int verify(struct params *, int);
116 1.5 elric static int verify_disklabel(int);
117 1.5 elric static int verify_ffs(int);
118 1.9 cb static int verify_reenter(struct params *);
119 1.1 elric
120 1.33 joerg __dead static void usage(void);
121 1.1 elric
122 1.1 elric /* Verbose Framework */
123 1.17 cbiere unsigned verbose = 0;
124 1.1 elric
125 1.1 elric #define VERBOSE(x,y) if (verbose >= x) y
126 1.19 christos #define VPRINTF(x,y) if (verbose >= x) (void)printf y
127 1.1 elric
128 1.1 elric static void
129 1.1 elric usage(void)
130 1.1 elric {
131 1.1 elric
132 1.19 christos (void)fprintf(stderr, "usage: %s [-nv] [-V vmeth] cgd dev [paramsfile]\n",
133 1.1 elric getprogname());
134 1.19 christos (void)fprintf(stderr, " %s -C [-nv] [-f configfile]\n", getprogname());
135 1.19 christos (void)fprintf(stderr, " %s -G [-nv] [-i ivmeth] [-k kgmeth] "
136 1.5 elric "[-o outfile] paramsfile\n", getprogname());
137 1.19 christos (void)fprintf(stderr, " %s -g [-nv] [-i ivmeth] [-k kgmeth] "
138 1.5 elric "[-o outfile] alg [keylen]\n", getprogname());
139 1.34 christos (void)fprintf(stderr, " %s -l\n", getprogname());
140 1.19 christos (void)fprintf(stderr, " %s -s [-nv] [-i ivmeth] cgd dev alg "
141 1.1 elric "[keylen]\n", getprogname());
142 1.34 christos (void)fprintf(stderr, " %s -U [-nv] [-f configfile]\n", getprogname());
143 1.19 christos (void)fprintf(stderr, " %s -u [-nv] cgd\n", getprogname());
144 1.17 cbiere exit(EXIT_FAILURE);
145 1.17 cbiere }
146 1.17 cbiere
147 1.17 cbiere static int
148 1.19 christos parse_size_t(const char *s, size_t *l)
149 1.17 cbiere {
150 1.17 cbiere char *endptr;
151 1.17 cbiere long v;
152 1.17 cbiere
153 1.17 cbiere errno = 0;
154 1.17 cbiere v = strtol(s, &endptr, 10);
155 1.17 cbiere if ((v == LONG_MIN || v == LONG_MAX) && errno)
156 1.17 cbiere return -1;
157 1.17 cbiere if (v < INT_MIN || v > INT_MAX) {
158 1.17 cbiere errno = ERANGE;
159 1.17 cbiere return -1;
160 1.17 cbiere }
161 1.17 cbiere if (endptr == s) {
162 1.17 cbiere errno = EINVAL;
163 1.17 cbiere return -1;
164 1.17 cbiere }
165 1.19 christos *l = (size_t)v;
166 1.19 christos return 0;
167 1.17 cbiere }
168 1.17 cbiere
169 1.17 cbiere static void
170 1.17 cbiere set_action(enum action *action, enum action value)
171 1.17 cbiere {
172 1.17 cbiere if (*action != ACTION_DEFAULT)
173 1.17 cbiere usage();
174 1.17 cbiere *action = value;
175 1.1 elric }
176 1.1 elric
177 1.1 elric int
178 1.1 elric main(int argc, char **argv)
179 1.1 elric {
180 1.5 elric struct params *p;
181 1.5 elric struct params *tp;
182 1.5 elric struct keygen *kg;
183 1.17 cbiere enum action action = ACTION_DEFAULT;
184 1.1 elric int ch;
185 1.17 cbiere const char *cfile = NULL;
186 1.17 cbiere const char *outfile = NULL;
187 1.1 elric
188 1.15 elric setprogname(*argv);
189 1.13 elric eliminate_cores();
190 1.14 elric if (mlockall(MCL_FUTURE))
191 1.14 elric err(EXIT_FAILURE, "Can't lock memory");
192 1.5 elric p = params_new();
193 1.5 elric kg = NULL;
194 1.1 elric
195 1.34 christos while ((ch = getopt(argc, argv, "CGUV:b:f:gi:k:lno:spuv")) != -1)
196 1.1 elric switch (ch) {
197 1.1 elric case 'C':
198 1.17 cbiere set_action(&action, ACTION_CONFIGALL);
199 1.1 elric break;
200 1.5 elric case 'G':
201 1.17 cbiere set_action(&action, ACTION_GENERATE_CONVERT);
202 1.5 elric break;
203 1.1 elric case 'U':
204 1.17 cbiere set_action(&action, ACTION_UNCONFIGALL);
205 1.1 elric break;
206 1.3 elric case 'V':
207 1.5 elric tp = params_verify_method(string_fromcharstar(optarg));
208 1.5 elric if (!tp)
209 1.3 elric usage();
210 1.5 elric p = params_combine(p, tp);
211 1.3 elric break;
212 1.1 elric case 'b':
213 1.17 cbiere {
214 1.19 christos size_t size;
215 1.17 cbiere
216 1.19 christos if (parse_size_t(optarg, &size) == -1)
217 1.17 cbiere usage();
218 1.17 cbiere tp = params_bsize(size);
219 1.17 cbiere if (!tp)
220 1.17 cbiere usage();
221 1.17 cbiere p = params_combine(p, tp);
222 1.17 cbiere }
223 1.1 elric break;
224 1.1 elric case 'f':
225 1.17 cbiere if (cfile)
226 1.17 cbiere usage();
227 1.17 cbiere cfile = estrdup(optarg);
228 1.1 elric break;
229 1.1 elric case 'g':
230 1.17 cbiere set_action(&action, ACTION_GENERATE);
231 1.1 elric break;
232 1.1 elric case 'i':
233 1.5 elric tp = params_ivmeth(string_fromcharstar(optarg));
234 1.5 elric p = params_combine(p, tp);
235 1.1 elric break;
236 1.1 elric case 'k':
237 1.5 elric kg = keygen_method(string_fromcharstar(optarg));
238 1.5 elric if (!kg)
239 1.1 elric usage();
240 1.5 elric keygen_addlist(&p->keygen, kg);
241 1.1 elric break;
242 1.34 christos case 'l':
243 1.34 christos set_action(&action, ACTION_LIST);
244 1.34 christos break;
245 1.1 elric case 'n':
246 1.1 elric nflag = 1;
247 1.1 elric break;
248 1.1 elric case 'o':
249 1.17 cbiere if (outfile)
250 1.17 cbiere usage();
251 1.17 cbiere outfile = estrdup(optarg);
252 1.1 elric break;
253 1.22 elric case 'p':
254 1.22 elric pflag = PFLAG_STDIN;
255 1.22 elric break;
256 1.1 elric case 's':
257 1.17 cbiere set_action(&action, ACTION_CONFIGSTDIN);
258 1.1 elric break;
259 1.1 elric
260 1.1 elric case 'u':
261 1.17 cbiere set_action(&action, ACTION_UNCONFIGURE);
262 1.1 elric break;
263 1.1 elric case 'v':
264 1.1 elric verbose++;
265 1.1 elric break;
266 1.1 elric default:
267 1.1 elric usage();
268 1.1 elric /* NOTREACHED */
269 1.1 elric }
270 1.1 elric
271 1.1 elric argc -= optind;
272 1.1 elric argv += optind;
273 1.1 elric
274 1.17 cbiere if (!outfile)
275 1.17 cbiere outfile = "";
276 1.17 cbiere if (!cfile)
277 1.17 cbiere cfile = "";
278 1.17 cbiere
279 1.32 pooka if (prog_init && prog_init() == -1)
280 1.32 pooka err(1, "init failed");
281 1.32 pooka
282 1.1 elric /* validate the consistency of the arguments */
283 1.1 elric
284 1.1 elric switch (action) {
285 1.17 cbiere case ACTION_DEFAULT: /* ACTION_CONFIGURE is the default */
286 1.1 elric case ACTION_CONFIGURE:
287 1.5 elric return configure(argc, argv, p, CONFIG_FLAGS_FROMMAIN);
288 1.1 elric case ACTION_UNCONFIGURE:
289 1.3 elric return unconfigure(argc, argv, NULL, CONFIG_FLAGS_FROMMAIN);
290 1.1 elric case ACTION_GENERATE:
291 1.5 elric return generate(p, argc, argv, outfile);
292 1.5 elric case ACTION_GENERATE_CONVERT:
293 1.5 elric return generate_convert(p, argc, argv, outfile);
294 1.1 elric case ACTION_CONFIGALL:
295 1.1 elric return do_all(cfile, argc, argv, configure);
296 1.1 elric case ACTION_UNCONFIGALL:
297 1.1 elric return do_all(cfile, argc, argv, unconfigure);
298 1.1 elric case ACTION_CONFIGSTDIN:
299 1.5 elric return configure_stdin(p, argc, argv);
300 1.34 christos case ACTION_LIST:
301 1.34 christos return do_list(argc, argv);
302 1.19 christos default:
303 1.19 christos errx(EXIT_FAILURE, "undefined action");
304 1.19 christos /* NOTREACHED */
305 1.1 elric }
306 1.1 elric }
307 1.1 elric
308 1.5 elric static bits_t *
309 1.19 christos getkey(const char *dev, struct keygen *kg, size_t len)
310 1.1 elric {
311 1.5 elric bits_t *ret = NULL;
312 1.5 elric bits_t *tmp;
313 1.1 elric
314 1.19 christos VPRINTF(3, ("getkey(\"%s\", %p, %zu) called\n", dev, kg, len));
315 1.5 elric for (; kg; kg=kg->next) {
316 1.5 elric switch (kg->kg_method) {
317 1.5 elric case KEYGEN_STOREDKEY:
318 1.5 elric tmp = getkey_storedkey(dev, kg, len);
319 1.5 elric break;
320 1.5 elric case KEYGEN_RANDOMKEY:
321 1.12 tv tmp = getkey_randomkey(dev, kg, len, 1);
322 1.12 tv break;
323 1.12 tv case KEYGEN_URANDOMKEY:
324 1.12 tv tmp = getkey_randomkey(dev, kg, len, 0);
325 1.5 elric break;
326 1.10 dan case KEYGEN_PKCS5_PBKDF2_SHA1:
327 1.10 dan tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 0);
328 1.10 dan break;
329 1.10 dan /* provide backwards compatibility for old config files */
330 1.10 dan case KEYGEN_PKCS5_PBKDF2_OLD:
331 1.10 dan tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 1);
332 1.5 elric break;
333 1.23 elric case KEYGEN_SHELL_CMD:
334 1.23 elric tmp = getkey_shell_cmd(dev, kg, len);
335 1.23 elric break;
336 1.5 elric default:
337 1.5 elric warnx("unrecognised keygen method %d in getkey()",
338 1.5 elric kg->kg_method);
339 1.5 elric if (ret)
340 1.5 elric bits_free(ret);
341 1.5 elric return NULL;
342 1.5 elric }
343 1.5 elric
344 1.5 elric if (ret)
345 1.5 elric ret = bits_xor_d(tmp, ret);
346 1.5 elric else
347 1.5 elric ret = tmp;
348 1.1 elric }
349 1.5 elric
350 1.5 elric return ret;
351 1.5 elric }
352 1.5 elric
353 1.5 elric /*ARGSUSED*/
354 1.5 elric static bits_t *
355 1.19 christos getkey_storedkey(const char *target, struct keygen *kg, size_t keylen)
356 1.5 elric {
357 1.5 elric return bits_dup(kg->kg_key);
358 1.1 elric }
359 1.1 elric
360 1.5 elric /*ARGSUSED*/
361 1.5 elric static bits_t *
362 1.19 christos getkey_randomkey(const char *target, struct keygen *kg, size_t keylen, int hard)
363 1.1 elric {
364 1.12 tv return bits_getrandombits(keylen, hard);
365 1.1 elric }
366 1.1 elric
367 1.22 elric static char *
368 1.22 elric maybe_getpass(char *prompt)
369 1.22 elric {
370 1.22 elric char buf[1024];
371 1.22 elric char *p = buf;
372 1.22 elric char *tmp;
373 1.22 elric
374 1.22 elric switch (pflag) {
375 1.22 elric case PFLAG_GETPASS:
376 1.22 elric p = getpass(prompt);
377 1.22 elric break;
378 1.22 elric
379 1.22 elric case PFLAG_STDIN:
380 1.22 elric p = fgets(buf, sizeof(buf), stdin);
381 1.22 elric if (p) {
382 1.22 elric tmp = strchr(p, '\n');
383 1.22 elric if (tmp)
384 1.22 elric *tmp = '\0';
385 1.22 elric }
386 1.22 elric break;
387 1.22 elric
388 1.22 elric default:
389 1.22 elric errx(EXIT_FAILURE, "pflag set inappropriately?");
390 1.22 elric }
391 1.22 elric
392 1.22 elric if (!p)
393 1.22 elric err(EXIT_FAILURE, "failed to read passphrase");
394 1.22 elric
395 1.22 elric return estrdup(p);
396 1.22 elric }
397 1.22 elric
398 1.5 elric /*ARGSUSED*/
399 1.29 elric /*
400 1.10 dan * XXX take, and pass through, a compat flag that indicates whether we
401 1.10 dan * provide backwards compatibility with a previous bug. The previous
402 1.10 dan * behaviour is indicated by the keygen method pkcs5_pbkdf2, and a
403 1.10 dan * non-zero compat flag. The new default, and correct keygen method is
404 1.10 dan * called pcks5_pbkdf2/sha1. When the old method is removed, so will
405 1.10 dan * be the compat argument.
406 1.10 dan */
407 1.5 elric static bits_t *
408 1.19 christos getkey_pkcs5_pbkdf2(const char *target, struct keygen *kg, size_t keylen,
409 1.19 christos int compat)
410 1.1 elric {
411 1.5 elric bits_t *ret;
412 1.25 dholland char *passp;
413 1.5 elric char buf[1024];
414 1.5 elric u_int8_t *tmp;
415 1.1 elric
416 1.5 elric snprintf(buf, sizeof(buf), "%s's passphrase:", target);
417 1.25 dholland passp = maybe_getpass(buf);
418 1.25 dholland if (pkcs5_pbkdf2(&tmp, BITS2BYTES(keylen), (uint8_t *)passp,
419 1.25 dholland strlen(passp),
420 1.5 elric bits_getbuf(kg->kg_salt), BITS2BYTES(bits_len(kg->kg_salt)),
421 1.10 dan kg->kg_iterations, compat)) {
422 1.5 elric warnx("failed to generate PKCS#5 PBKDF2 key");
423 1.5 elric return NULL;
424 1.5 elric }
425 1.5 elric
426 1.5 elric ret = bits_new(tmp, keylen);
427 1.9 cb kg->kg_key = bits_dup(ret);
428 1.27 christos memset(passp, 0, strlen(passp));
429 1.22 elric free(passp);
430 1.5 elric free(tmp);
431 1.1 elric return ret;
432 1.1 elric }
433 1.1 elric
434 1.5 elric /*ARGSUSED*/
435 1.23 elric static bits_t *
436 1.23 elric getkey_shell_cmd(const char *target, struct keygen *kg, size_t keylen)
437 1.23 elric {
438 1.23 elric FILE *f;
439 1.23 elric bits_t *ret;
440 1.23 elric
441 1.23 elric f = popen(string_tocharstar(kg->kg_cmd), "r");
442 1.23 elric ret = bits_fget(f, keylen);
443 1.23 elric pclose(f);
444 1.23 elric
445 1.23 elric return ret;
446 1.23 elric }
447 1.23 elric
448 1.23 elric /*ARGSUSED*/
449 1.1 elric static int
450 1.3 elric unconfigure(int argc, char **argv, struct params *inparams, int flags)
451 1.1 elric {
452 1.1 elric int fd;
453 1.1 elric int ret;
454 1.1 elric char buf[MAXPATHLEN] = "";
455 1.1 elric
456 1.1 elric /* only complain about additional arguments, if called from main() */
457 1.1 elric if (flags == CONFIG_FLAGS_FROMMAIN && argc != 1)
458 1.1 elric usage();
459 1.1 elric
460 1.1 elric /* if called from do_all(), then ensure that 2 or 3 args exist */
461 1.1 elric if (flags == CONFIG_FLAGS_FROMALL && (argc < 2 || argc > 3))
462 1.1 elric return -1;
463 1.1 elric
464 1.32 pooka fd = opendisk1(*argv, O_RDWR, buf, sizeof(buf), 1, prog_open);
465 1.1 elric if (fd == -1) {
466 1.17 cbiere int saved_errno = errno;
467 1.17 cbiere
468 1.5 elric warn("can't open cgd \"%s\", \"%s\"", *argv, buf);
469 1.1 elric
470 1.1 elric /* this isn't fatal with nflag != 0 */
471 1.1 elric if (!nflag)
472 1.17 cbiere return saved_errno;
473 1.1 elric }
474 1.1 elric
475 1.1 elric VPRINTF(1, ("%s (%s): clearing\n", *argv, buf));
476 1.1 elric
477 1.1 elric if (nflag)
478 1.1 elric return 0;
479 1.1 elric
480 1.3 elric ret = unconfigure_fd(fd);
481 1.32 pooka (void)prog_close(fd);
482 1.3 elric return ret;
483 1.3 elric }
484 1.3 elric
485 1.3 elric static int
486 1.3 elric unconfigure_fd(int fd)
487 1.3 elric {
488 1.3 elric struct cgd_ioctl ci;
489 1.3 elric
490 1.32 pooka if (prog_ioctl(fd, CGDIOCCLR, &ci) == -1) {
491 1.17 cbiere warn("ioctl");
492 1.3 elric return -1;
493 1.1 elric }
494 1.1 elric
495 1.1 elric return 0;
496 1.1 elric }
497 1.1 elric
498 1.5 elric /*ARGSUSED*/
499 1.1 elric static int
500 1.3 elric configure(int argc, char **argv, struct params *inparams, int flags)
501 1.1 elric {
502 1.5 elric struct params *p;
503 1.22 elric struct keygen *kg;
504 1.5 elric int fd;
505 1.22 elric int loop = 0;
506 1.5 elric int ret;
507 1.5 elric char cgdname[PATH_MAX];
508 1.1 elric
509 1.34 christos if (argc == 2) {
510 1.17 cbiere char *pfile;
511 1.17 cbiere
512 1.17 cbiere if (asprintf(&pfile, "%s/%s",
513 1.17 cbiere CGDCONFIG_DIR, basename(argv[1])) == -1)
514 1.17 cbiere return -1;
515 1.17 cbiere
516 1.17 cbiere p = params_cget(pfile);
517 1.17 cbiere free(pfile);
518 1.17 cbiere } else if (argc == 3) {
519 1.17 cbiere p = params_cget(argv[2]);
520 1.17 cbiere } else {
521 1.1 elric /* print usage and exit, only if called from main() */
522 1.5 elric if (flags == CONFIG_FLAGS_FROMMAIN) {
523 1.5 elric warnx("wrong number of args");
524 1.1 elric usage();
525 1.5 elric }
526 1.1 elric return -1;
527 1.1 elric }
528 1.1 elric
529 1.5 elric if (!p)
530 1.5 elric return -1;
531 1.2 elric
532 1.3 elric /*
533 1.5 elric * over-ride with command line specifications and fill in default
534 1.5 elric * values.
535 1.3 elric */
536 1.3 elric
537 1.5 elric p = params_combine(p, inparams);
538 1.5 elric ret = params_filldefaults(p);
539 1.5 elric if (ret) {
540 1.5 elric params_free(p);
541 1.5 elric return ret;
542 1.5 elric }
543 1.5 elric
544 1.5 elric if (!params_verify(p)) {
545 1.5 elric warnx("params invalid");
546 1.5 elric return -1;
547 1.5 elric }
548 1.3 elric
549 1.3 elric /*
550 1.3 elric * loop over configuring the disk and checking to see if it
551 1.3 elric * verifies properly. We open and close the disk device each
552 1.3 elric * time, because if the user passes us the block device we
553 1.3 elric * need to flush the buffer cache.
554 1.22 elric *
555 1.22 elric * We only loop if one of the verification methods prompts for
556 1.22 elric * a password.
557 1.3 elric */
558 1.3 elric
559 1.22 elric for (kg = p->keygen; pflag == PFLAG_GETPASS && kg; kg = kg->next)
560 1.22 elric if ((kg->kg_method == KEYGEN_PKCS5_PBKDF2_SHA1) ||
561 1.22 elric (kg->kg_method == KEYGEN_PKCS5_PBKDF2_OLD )) {
562 1.22 elric loop = 1;
563 1.22 elric break;
564 1.22 elric }
565 1.22 elric
566 1.3 elric for (;;) {
567 1.3 elric fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
568 1.3 elric if (fd == -1)
569 1.3 elric return -1;
570 1.3 elric
571 1.5 elric if (p->key)
572 1.5 elric bits_free(p->key);
573 1.5 elric
574 1.5 elric p->key = getkey(argv[1], p->keygen, p->keylen);
575 1.5 elric if (!p->key)
576 1.3 elric goto bail_err;
577 1.3 elric
578 1.5 elric ret = configure_params(fd, cgdname, argv[1], p);
579 1.3 elric if (ret)
580 1.3 elric goto bail_err;
581 1.3 elric
582 1.5 elric ret = verify(p, fd);
583 1.3 elric if (ret == -1)
584 1.3 elric goto bail_err;
585 1.3 elric if (!ret)
586 1.3 elric break;
587 1.2 elric
588 1.19 christos (void)unconfigure_fd(fd);
589 1.32 pooka (void)prog_close(fd);
590 1.22 elric
591 1.22 elric if (!loop) {
592 1.22 elric warnx("verification failed permanently");
593 1.22 elric goto bail_err;
594 1.22 elric }
595 1.22 elric
596 1.22 elric warnx("verification failed, please reenter passphrase");
597 1.3 elric }
598 1.2 elric
599 1.5 elric params_free(p);
600 1.32 pooka (void)prog_close(fd);
601 1.3 elric return 0;
602 1.3 elric bail_err:
603 1.5 elric params_free(p);
604 1.32 pooka (void)prog_close(fd);
605 1.3 elric return -1;
606 1.1 elric }
607 1.1 elric
608 1.1 elric static int
609 1.1 elric configure_stdin(struct params *p, int argc, char **argv)
610 1.1 elric {
611 1.2 elric int fd;
612 1.1 elric int ret;
613 1.2 elric char cgdname[PATH_MAX];
614 1.1 elric
615 1.1 elric if (argc < 3 || argc > 4)
616 1.1 elric usage();
617 1.1 elric
618 1.5 elric p->algorithm = string_fromcharstar(argv[2]);
619 1.17 cbiere if (argc > 3) {
620 1.19 christos size_t keylen;
621 1.17 cbiere
622 1.19 christos if (parse_size_t(argv[3], &keylen) == -1) {
623 1.17 cbiere warn("failed to parse key length");
624 1.17 cbiere return -1;
625 1.17 cbiere }
626 1.17 cbiere p->keylen = keylen;
627 1.17 cbiere }
628 1.1 elric
629 1.1 elric ret = params_filldefaults(p);
630 1.1 elric if (ret)
631 1.1 elric return ret;
632 1.1 elric
633 1.2 elric fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
634 1.2 elric if (fd == -1)
635 1.2 elric return -1;
636 1.2 elric
637 1.5 elric p->key = bits_fget(stdin, p->keylen);
638 1.5 elric if (!p->key) {
639 1.5 elric warnx("failed to read key from stdin");
640 1.1 elric return -1;
641 1.5 elric }
642 1.1 elric
643 1.2 elric return configure_params(fd, cgdname, argv[1], p);
644 1.1 elric }
645 1.1 elric
646 1.1 elric static int
647 1.17 cbiere opendisk_werror(const char *cgd, char *buf, size_t buflen)
648 1.2 elric {
649 1.2 elric int fd;
650 1.2 elric
651 1.18 cbiere VPRINTF(3, ("opendisk_werror(%s, %s, %zu) called.\n", cgd, buf, buflen));
652 1.5 elric
653 1.2 elric /* sanity */
654 1.2 elric if (!cgd || !buf)
655 1.2 elric return -1;
656 1.2 elric
657 1.2 elric if (nflag) {
658 1.17 cbiere if (strlcpy(buf, cgd, buflen) >= buflen)
659 1.17 cbiere return -1;
660 1.2 elric return 0;
661 1.2 elric }
662 1.2 elric
663 1.32 pooka fd = opendisk1(cgd, O_RDWR, buf, buflen, 0, prog_open);
664 1.2 elric if (fd == -1)
665 1.5 elric warnx("can't open cgd \"%s\", \"%s\"", cgd, buf);
666 1.5 elric
667 1.2 elric return fd;
668 1.2 elric }
669 1.2 elric
670 1.2 elric static int
671 1.2 elric configure_params(int fd, const char *cgd, const char *dev, struct params *p)
672 1.1 elric {
673 1.1 elric struct cgd_ioctl ci;
674 1.1 elric
675 1.1 elric /* sanity */
676 1.1 elric if (!cgd || !dev)
677 1.1 elric return -1;
678 1.1 elric
679 1.19 christos (void)memset(&ci, 0x0, sizeof(ci));
680 1.16 christos ci.ci_disk = dev;
681 1.16 christos ci.ci_alg = string_tocharstar(p->algorithm);
682 1.16 christos ci.ci_ivmethod = string_tocharstar(p->ivmeth);
683 1.16 christos ci.ci_key = bits_getbuf(p->key);
684 1.1 elric ci.ci_keylen = p->keylen;
685 1.1 elric ci.ci_blocksize = p->bsize;
686 1.1 elric
687 1.20 martin VPRINTF(1, (" with alg %s keylen %zu blocksize %zu ivmethod %s\n",
688 1.5 elric string_tocharstar(p->algorithm), p->keylen, p->bsize,
689 1.5 elric string_tocharstar(p->ivmeth)));
690 1.5 elric VPRINTF(2, ("key: "));
691 1.5 elric VERBOSE(2, bits_fprint(stdout, p->key));
692 1.5 elric VPRINTF(2, ("\n"));
693 1.1 elric
694 1.1 elric if (nflag)
695 1.1 elric return 0;
696 1.1 elric
697 1.32 pooka if (prog_ioctl(fd, CGDIOCSET, &ci) == -1) {
698 1.17 cbiere int saved_errno = errno;
699 1.17 cbiere warn("ioctl");
700 1.17 cbiere return saved_errno;
701 1.1 elric }
702 1.1 elric
703 1.1 elric return 0;
704 1.1 elric }
705 1.1 elric
706 1.3 elric /*
707 1.3 elric * verify returns 0 for success, -1 for unrecoverable error, or 1 for retry.
708 1.3 elric */
709 1.3 elric
710 1.3 elric #define SCANSIZE 8192
711 1.3 elric
712 1.3 elric static int
713 1.3 elric verify(struct params *p, int fd)
714 1.3 elric {
715 1.3 elric
716 1.3 elric switch (p->verify_method) {
717 1.3 elric case VERIFY_NONE:
718 1.3 elric return 0;
719 1.3 elric case VERIFY_DISKLABEL:
720 1.5 elric return verify_disklabel(fd);
721 1.5 elric case VERIFY_FFS:
722 1.5 elric return verify_ffs(fd);
723 1.9 cb case VERIFY_REENTER:
724 1.9 cb return verify_reenter(p);
725 1.3 elric default:
726 1.5 elric warnx("unimplemented verification method");
727 1.3 elric return -1;
728 1.3 elric }
729 1.5 elric }
730 1.5 elric
731 1.5 elric static int
732 1.5 elric verify_disklabel(int fd)
733 1.5 elric {
734 1.5 elric struct disklabel l;
735 1.17 cbiere ssize_t ret;
736 1.5 elric char buf[SCANSIZE];
737 1.3 elric
738 1.3 elric /*
739 1.3 elric * we simply scan the first few blocks for a disklabel, ignoring
740 1.3 elric * any MBR/filecore sorts of logic. MSDOS and RiscOS can't read
741 1.3 elric * a cgd, anyway, so it is unlikely that there will be non-native
742 1.3 elric * partition information.
743 1.3 elric */
744 1.3 elric
745 1.32 pooka ret = prog_pread(fd, buf, 8192, 0);
746 1.17 cbiere if (ret < 0) {
747 1.5 elric warn("can't read disklabel area");
748 1.3 elric return -1;
749 1.3 elric }
750 1.3 elric
751 1.3 elric /* now scan for the disklabel */
752 1.3 elric
753 1.17 cbiere return disklabel_scan(&l, buf, (size_t)ret);
754 1.3 elric }
755 1.3 elric
756 1.7 fvdl static off_t sblock_try[] = SBLOCKSEARCH;
757 1.7 fvdl
758 1.1 elric static int
759 1.5 elric verify_ffs(int fd)
760 1.5 elric {
761 1.19 christos size_t i;
762 1.5 elric
763 1.7 fvdl for (i = 0; sblock_try[i] != -1; i++) {
764 1.19 christos union {
765 1.19 christos char buf[SBLOCKSIZE];
766 1.19 christos struct fs fs;
767 1.19 christos } u;
768 1.17 cbiere ssize_t ret;
769 1.17 cbiere
770 1.32 pooka ret = prog_pread(fd, &u, sizeof(u), sblock_try[i]);
771 1.17 cbiere if (ret < 0) {
772 1.7 fvdl warn("pread");
773 1.17 cbiere break;
774 1.19 christos } else if ((size_t)ret < sizeof(u)) {
775 1.17 cbiere warnx("pread: incomplete block");
776 1.17 cbiere break;
777 1.7 fvdl }
778 1.19 christos switch (u.fs.fs_magic) {
779 1.7 fvdl case FS_UFS1_MAGIC:
780 1.7 fvdl case FS_UFS2_MAGIC:
781 1.7 fvdl case FS_UFS1_MAGIC_SWAPPED:
782 1.7 fvdl case FS_UFS2_MAGIC_SWAPPED:
783 1.7 fvdl return 0;
784 1.7 fvdl default:
785 1.7 fvdl continue;
786 1.7 fvdl }
787 1.5 elric }
788 1.17 cbiere
789 1.17 cbiere return 1; /* failure */
790 1.9 cb }
791 1.9 cb
792 1.9 cb static int
793 1.9 cb verify_reenter(struct params *p)
794 1.9 cb {
795 1.9 cb struct keygen *kg;
796 1.9 cb bits_t *orig_key, *key;
797 1.9 cb int ret;
798 1.9 cb
799 1.9 cb ret = 0;
800 1.9 cb for (kg = p->keygen; kg && !ret; kg = kg->next) {
801 1.29 elric if ((kg->kg_method != KEYGEN_PKCS5_PBKDF2_SHA1) &&
802 1.10 dan (kg->kg_method != KEYGEN_PKCS5_PBKDF2_OLD ))
803 1.9 cb continue;
804 1.9 cb
805 1.9 cb orig_key = kg->kg_key;
806 1.9 cb kg->kg_key = NULL;
807 1.9 cb
808 1.10 dan /* add a compat flag till the _OLD method goes away */
809 1.9 cb key = getkey_pkcs5_pbkdf2("re-enter device", kg,
810 1.10 dan bits_len(orig_key), kg->kg_method == KEYGEN_PKCS5_PBKDF2_OLD);
811 1.9 cb ret = !bits_match(key, orig_key);
812 1.9 cb
813 1.9 cb bits_free(key);
814 1.9 cb bits_free(kg->kg_key);
815 1.9 cb kg->kg_key = orig_key;
816 1.9 cb }
817 1.9 cb
818 1.9 cb return ret;
819 1.5 elric }
820 1.5 elric
821 1.5 elric static int
822 1.1 elric generate(struct params *p, int argc, char **argv, const char *outfile)
823 1.1 elric {
824 1.1 elric int ret;
825 1.1 elric
826 1.1 elric if (argc < 1 || argc > 2)
827 1.1 elric usage();
828 1.1 elric
829 1.5 elric p->algorithm = string_fromcharstar(argv[0]);
830 1.17 cbiere if (argc > 1) {
831 1.19 christos size_t keylen;
832 1.17 cbiere
833 1.19 christos if (parse_size_t(argv[1], &keylen) == -1) {
834 1.17 cbiere warn("Failed to parse key length");
835 1.17 cbiere return -1;
836 1.17 cbiere }
837 1.17 cbiere p->keylen = keylen;
838 1.17 cbiere }
839 1.5 elric
840 1.5 elric ret = params_filldefaults(p);
841 1.1 elric if (ret)
842 1.1 elric return ret;
843 1.5 elric
844 1.5 elric if (!p->keygen) {
845 1.10 dan p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
846 1.5 elric if (!p->keygen)
847 1.5 elric return -1;
848 1.5 elric }
849 1.5 elric
850 1.5 elric if (keygen_filldefaults(p->keygen, p->keylen)) {
851 1.5 elric warnx("Failed to generate defaults for keygen");
852 1.5 elric return -1;
853 1.5 elric }
854 1.5 elric
855 1.5 elric if (!params_verify(p)) {
856 1.5 elric warnx("invalid parameters generated");
857 1.5 elric return -1;
858 1.5 elric }
859 1.5 elric
860 1.5 elric return params_cput(p, outfile);
861 1.5 elric }
862 1.5 elric
863 1.5 elric static int
864 1.5 elric generate_convert(struct params *p, int argc, char **argv, const char *outfile)
865 1.5 elric {
866 1.5 elric struct params *oldp;
867 1.5 elric struct keygen *kg;
868 1.5 elric
869 1.5 elric if (argc != 1)
870 1.5 elric usage();
871 1.5 elric
872 1.5 elric oldp = params_cget(*argv);
873 1.5 elric if (!oldp)
874 1.5 elric return -1;
875 1.5 elric
876 1.5 elric /* for sanity, we ensure that none of the keygens are randomkey */
877 1.5 elric for (kg=p->keygen; kg; kg=kg->next)
878 1.30 elric if ((kg->kg_method == KEYGEN_RANDOMKEY) ||
879 1.30 elric (kg->kg_method == KEYGEN_URANDOMKEY)) {
880 1.30 elric warnx("can't preserve randomly generated key");
881 1.5 elric goto bail;
882 1.30 elric }
883 1.5 elric for (kg=oldp->keygen; kg; kg=kg->next)
884 1.30 elric if ((kg->kg_method == KEYGEN_RANDOMKEY) ||
885 1.30 elric (kg->kg_method == KEYGEN_URANDOMKEY)) {
886 1.30 elric warnx("can't preserve randomly generated key");
887 1.5 elric goto bail;
888 1.30 elric }
889 1.5 elric
890 1.5 elric if (!params_verify(oldp)) {
891 1.5 elric warnx("invalid old parameters file \"%s\"", *argv);
892 1.5 elric return -1;
893 1.1 elric }
894 1.1 elric
895 1.5 elric oldp->key = getkey("old file", oldp->keygen, oldp->keylen);
896 1.5 elric
897 1.5 elric /* we copy across the non-keygen info, here. */
898 1.5 elric
899 1.5 elric string_free(p->algorithm);
900 1.5 elric string_free(p->ivmeth);
901 1.5 elric
902 1.5 elric p->algorithm = string_dup(oldp->algorithm);
903 1.5 elric p->ivmeth = string_dup(oldp->ivmeth);
904 1.5 elric p->keylen = oldp->keylen;
905 1.5 elric p->bsize = oldp->bsize;
906 1.5 elric if (p->verify_method == VERIFY_UNKNOWN)
907 1.5 elric p->verify_method = oldp->verify_method;
908 1.5 elric
909 1.5 elric params_free(oldp);
910 1.1 elric
911 1.5 elric if (!p->keygen) {
912 1.10 dan p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
913 1.5 elric if (!p->keygen)
914 1.1 elric return -1;
915 1.5 elric }
916 1.19 christos (void)params_filldefaults(p);
917 1.19 christos (void)keygen_filldefaults(p->keygen, p->keylen);
918 1.5 elric p->key = getkey("new file", p->keygen, p->keylen);
919 1.5 elric
920 1.5 elric kg = keygen_generate(KEYGEN_STOREDKEY);
921 1.5 elric kg->kg_key = bits_xor(p->key, oldp->key);
922 1.5 elric keygen_addlist(&p->keygen, kg);
923 1.5 elric
924 1.5 elric if (!params_verify(p)) {
925 1.5 elric warnx("can't generate new parameters file");
926 1.5 elric return -1;
927 1.1 elric }
928 1.1 elric
929 1.5 elric return params_cput(p, outfile);
930 1.5 elric bail:
931 1.5 elric params_free(oldp);
932 1.5 elric return -1;
933 1.1 elric }
934 1.1 elric
935 1.1 elric static int
936 1.19 christos /*ARGSUSED*/
937 1.1 elric do_all(const char *cfile, int argc, char **argv,
938 1.3 elric int (*conf)(int, char **, struct params *, int))
939 1.1 elric {
940 1.1 elric FILE *f;
941 1.1 elric size_t len;
942 1.1 elric size_t lineno;
943 1.1 elric int my_argc;
944 1.1 elric int ret;
945 1.1 elric const char *fn;
946 1.1 elric char *line;
947 1.1 elric char **my_argv;
948 1.1 elric
949 1.1 elric if (argc > 0)
950 1.1 elric usage();
951 1.1 elric
952 1.1 elric if (!cfile[0])
953 1.1 elric fn = CGDCONFIG_CFILE;
954 1.1 elric else
955 1.1 elric fn = cfile;
956 1.1 elric
957 1.1 elric f = fopen(fn, "r");
958 1.1 elric if (!f) {
959 1.5 elric warn("could not open config file \"%s\"", fn);
960 1.1 elric return -1;
961 1.1 elric }
962 1.1 elric
963 1.5 elric ret = chdir(CGDCONFIG_DIR);
964 1.5 elric if (ret == -1)
965 1.5 elric warn("could not chdir to %s", CGDCONFIG_DIR);
966 1.5 elric
967 1.1 elric ret = 0;
968 1.1 elric lineno = 0;
969 1.1 elric for (;;) {
970 1.1 elric line = fparseln(f, &len, &lineno, "\\\\#", FPARSELN_UNESCALL);
971 1.1 elric if (!line)
972 1.1 elric break;
973 1.1 elric if (!*line)
974 1.1 elric continue;
975 1.1 elric
976 1.1 elric my_argv = words(line, &my_argc);
977 1.3 elric ret = conf(my_argc, my_argv, NULL, CONFIG_FLAGS_FROMALL);
978 1.1 elric if (ret) {
979 1.5 elric warnx("action failed on \"%s\" line %lu", fn,
980 1.1 elric (u_long)lineno);
981 1.1 elric break;
982 1.1 elric }
983 1.1 elric words_free(my_argv, my_argc);
984 1.1 elric }
985 1.1 elric return ret;
986 1.1 elric }
987 1.13 elric
988 1.34 christos static const char *
989 1.34 christos iv_method(int mode)
990 1.34 christos {
991 1.34 christos
992 1.34 christos switch (mode) {
993 1.34 christos case CGD_CIPHER_CBC_ENCBLKNO8:
994 1.34 christos return "encblkno8";
995 1.34 christos case CGD_CIPHER_CBC_ENCBLKNO1:
996 1.34 christos return "encblkno1";
997 1.34 christos default:
998 1.34 christos return "unknown";
999 1.34 christos }
1000 1.34 christos }
1001 1.34 christos
1002 1.34 christos static int
1003 1.34 christos do_list(int argc, char **argv)
1004 1.34 christos {
1005 1.34 christos char path[64], buf[16];
1006 1.34 christos struct cgd_user cgu;
1007 1.34 christos const char *fn;
1008 1.34 christos int fd, n, rv;
1009 1.34 christos
1010 1.34 christos if (argc != 0 && argc != 1)
1011 1.34 christos usage();
1012 1.34 christos
1013 1.34 christos fn = argc ? argv[0] : "cgd0";
1014 1.34 christos n = 0;
1015 1.34 christos for (;;) {
1016 1.34 christos fd = opendisk(fn, O_RDONLY, path, sizeof(path), 0);
1017 1.34 christos if (fd == -1) {
1018 1.34 christos if (argc)
1019 1.34 christos err(1, "open: %s", fn);
1020 1.34 christos break;
1021 1.34 christos }
1022 1.34 christos
1023 1.34 christos cgu.cgu_unit = argc ? -1 : n;
1024 1.34 christos rv = prog_ioctl(fd, CGDIOCGET, &cgu);
1025 1.34 christos if (rv == -1) {
1026 1.34 christos close(fd);
1027 1.34 christos err(1, "CGDIOCGET");
1028 1.34 christos }
1029 1.34 christos
1030 1.34 christos printf("%s: ", fn);
1031 1.34 christos
1032 1.34 christos if (cgu.cgu_dev == 0)
1033 1.34 christos printf("not in use");
1034 1.34 christos else {
1035 1.34 christos char *dev;
1036 1.34 christos
1037 1.34 christos dev = devname(cgu.cgu_dev, S_IFBLK);
1038 1.34 christos if (dev != NULL)
1039 1.34 christos printf("%s ", dev);
1040 1.34 christos else
1041 1.34 christos printf("dev %llu,%llu ",
1042 1.34 christos (unsigned long long)major(cgu.cgu_dev),
1043 1.34 christos (unsigned long long)minor(cgu.cgu_dev));
1044 1.34 christos
1045 1.34 christos if (verbose)
1046 1.34 christos printf("%s ", cgu.cgu_alg);
1047 1.34 christos if (verbose > 1) {
1048 1.34 christos printf("keylen %d ", cgu.cgu_keylen);
1049 1.34 christos printf("blksize %zd ", cgu.cgu_blocksize);
1050 1.34 christos printf("%s ", iv_method(cgu.cgu_mode));
1051 1.34 christos }
1052 1.34 christos }
1053 1.34 christos putchar('\n');
1054 1.34 christos close(fd);
1055 1.34 christos
1056 1.34 christos if (argc)
1057 1.34 christos break;
1058 1.34 christos
1059 1.34 christos n++;
1060 1.34 christos snprintf(buf, sizeof(buf), "cgd%d", n);
1061 1.34 christos fn = buf;
1062 1.34 christos }
1063 1.34 christos
1064 1.34 christos return 0;
1065 1.34 christos }
1066 1.34 christos
1067 1.13 elric static void
1068 1.13 elric eliminate_cores(void)
1069 1.13 elric {
1070 1.13 elric struct rlimit rlp;
1071 1.13 elric
1072 1.13 elric rlp.rlim_cur = 0;
1073 1.13 elric rlp.rlim_max = 0;
1074 1.17 cbiere if (setrlimit(RLIMIT_CORE, &rlp) == -1)
1075 1.13 elric err(EXIT_FAILURE, "Can't disable cores");
1076 1.13 elric }
1077