Home | History | Annotate | Line # | Download | only in cgdconfig
cgdconfig.c revision 1.56
      1  1.56  riastrad /* $NetBSD: cgdconfig.c,v 1.56 2022/08/12 10:49:17 riastradh 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.56  riastrad __RCSID("$NetBSD: cgdconfig.c,v 1.56 2022/08/12 10:49:17 riastradh Exp $");
     37   1.1     elric #endif
     38   1.1     elric 
     39  1.53       nia #ifdef HAVE_ARGON2
     40  1.53       nia #include <argon2.h>
     41  1.53       nia #endif
     42  1.56  riastrad #include <assert.h>
     43   1.5     elric #include <err.h>
     44   1.1     elric #include <errno.h>
     45   1.1     elric #include <fcntl.h>
     46   1.1     elric #include <libgen.h>
     47   1.1     elric #include <stdio.h>
     48   1.1     elric #include <stdlib.h>
     49   1.1     elric #include <string.h>
     50   1.1     elric #include <unistd.h>
     51   1.1     elric #include <util.h>
     52  1.35  christos #include <paths.h>
     53  1.35  christos #include <dirent.h>
     54   1.1     elric 
     55  1.54  riastrad /* base64 gunk */
     56  1.54  riastrad #include <netinet/in.h>
     57  1.54  riastrad #include <arpa/nameser.h>
     58  1.54  riastrad #include <resolv.h>
     59  1.54  riastrad 
     60   1.1     elric #include <sys/ioctl.h>
     61  1.41  christos #include <sys/stat.h>
     62  1.37   mlelstv #include <sys/bootblock.h>
     63   1.3     elric #include <sys/disklabel.h>
     64  1.37   mlelstv #include <sys/disklabel_gpt.h>
     65  1.14     elric #include <sys/mman.h>
     66   1.1     elric #include <sys/param.h>
     67  1.13     elric #include <sys/resource.h>
     68  1.34  christos #include <sys/statvfs.h>
     69  1.35  christos #include <sys/bitops.h>
     70  1.56  riastrad #include <sys/queue.h>
     71   1.1     elric 
     72   1.1     elric #include <dev/cgdvar.h>
     73   1.1     elric 
     74   1.5     elric #include <ufs/ffs/fs.h>
     75   1.5     elric 
     76   1.1     elric #include "params.h"
     77   1.1     elric #include "pkcs5_pbkdf2.h"
     78   1.1     elric #include "utils.h"
     79  1.28     pooka #include "cgdconfig.h"
     80  1.32     pooka #include "prog_ops.h"
     81  1.56  riastrad #include "hkdf_hmac_sha256.h"
     82   1.1     elric 
     83   1.1     elric #define CGDCONFIG_CFILE		CGDCONFIG_DIR "/cgd.conf"
     84   1.1     elric 
     85  1.17    cbiere enum action {
     86  1.17    cbiere 	 ACTION_DEFAULT,		/* default -> configure */
     87  1.17    cbiere 	 ACTION_CONFIGURE,		/* configure, with paramsfile */
     88  1.17    cbiere 	 ACTION_UNCONFIGURE,		/* unconfigure */
     89  1.17    cbiere 	 ACTION_GENERATE,		/* generate a paramsfile */
     90  1.17    cbiere 	 ACTION_GENERATE_CONVERT,	/* generate a ``dup'' paramsfile */
     91  1.17    cbiere 	 ACTION_CONFIGALL,		/* configure all from config file */
     92  1.17    cbiere 	 ACTION_UNCONFIGALL,		/* unconfigure all from config file */
     93  1.34  christos 	 ACTION_CONFIGSTDIN,		/* configure, key from stdin */
     94  1.54  riastrad 	 ACTION_LIST,			/* list configured devices */
     95  1.54  riastrad 	 ACTION_PRINTKEY,		/* print key to stdout */
     96  1.55  riastrad 	 ACTION_PRINTALLKEYS,		/* print all keys to stdout */
     97  1.17    cbiere };
     98   1.1     elric 
     99   1.1     elric /* if nflag is set, do not configure/unconfigure the cgd's */
    100   1.1     elric 
    101   1.1     elric int	nflag = 0;
    102   1.1     elric 
    103  1.22     elric /* if pflag is set to PFLAG_STDIN read from stdin rather than getpass(3) */
    104  1.22     elric 
    105  1.46     alnsn #define	PFLAG_GETPASS		0x01
    106  1.46     alnsn #define	PFLAG_GETPASS_ECHO	0x02
    107  1.46     alnsn #define	PFLAG_GETPASS_MASK	0x03
    108  1.46     alnsn #define	PFLAG_STDIN		0x04
    109  1.22     elric int	pflag = PFLAG_GETPASS;
    110  1.22     elric 
    111  1.56  riastrad /*
    112  1.56  riastrad  * When configuring all cgds, save a cache of shared keys for key
    113  1.56  riastrad  * derivation.
    114  1.56  riastrad  */
    115  1.56  riastrad 
    116  1.56  riastrad struct sharedkey {
    117  1.56  riastrad 	int			 alg;
    118  1.56  riastrad 	string_t		*id;
    119  1.56  riastrad 	bits_t			*key;
    120  1.56  riastrad 	LIST_ENTRY(sharedkey)	 list;
    121  1.56  riastrad };
    122  1.56  riastrad LIST_HEAD(, sharedkey) sharedkeys;
    123  1.56  riastrad 
    124   1.3     elric static int	configure(int, char **, struct params *, int);
    125   1.1     elric static int	configure_stdin(struct params *, int argc, char **);
    126   1.1     elric static int	generate(struct params *, int, char **, const char *);
    127   1.5     elric static int	generate_convert(struct params *, int, char **, const char *);
    128   1.3     elric static int	unconfigure(int, char **, struct params *, int);
    129   1.3     elric static int	do_all(const char *, int, char **,
    130   1.3     elric 		       int (*)(int, char **, struct params *, int));
    131  1.34  christos static int	do_list(int, char **);
    132  1.55  riastrad static int	printkey(const char *, const char *, const char *, ...)
    133  1.55  riastrad 		    __printflike(3,4);
    134  1.55  riastrad static int	printkey1(int, char **, struct params *, int);
    135  1.54  riastrad static int	do_printkey(int, char **);
    136   1.1     elric 
    137   1.1     elric #define CONFIG_FLAGS_FROMALL	1	/* called from configure_all() */
    138   1.1     elric #define CONFIG_FLAGS_FROMMAIN	2	/* called from main() */
    139   1.1     elric 
    140   1.2     elric static int	 configure_params(int, const char *, const char *,
    141   1.2     elric 				  struct params *);
    142  1.13     elric static void	 eliminate_cores(void);
    143  1.19  christos static bits_t	*getkey(const char *, struct keygen *, size_t);
    144  1.19  christos static bits_t	*getkey_storedkey(const char *, struct keygen *, size_t);
    145  1.19  christos static bits_t	*getkey_randomkey(const char *, struct keygen *, size_t, int);
    146  1.53       nia #ifdef HAVE_ARGON2
    147  1.53       nia static bits_t	*getkey_argon2id(const char *, struct keygen *, size_t);
    148  1.53       nia #endif
    149  1.23     elric static bits_t	*getkey_pkcs5_pbkdf2(const char *, struct keygen *, size_t,
    150  1.23     elric 				     int);
    151  1.23     elric static bits_t	*getkey_shell_cmd(const char *, struct keygen *, size_t);
    152  1.22     elric static char	*maybe_getpass(char *);
    153  1.17    cbiere static int	 opendisk_werror(const char *, char *, size_t);
    154   1.3     elric static int	 unconfigure_fd(int);
    155   1.3     elric static int	 verify(struct params *, int);
    156   1.5     elric static int	 verify_disklabel(int);
    157   1.5     elric static int	 verify_ffs(int);
    158   1.9        cb static int	 verify_reenter(struct params *);
    159  1.37   mlelstv static int	 verify_mbr(int);
    160  1.37   mlelstv static int	 verify_gpt(int);
    161   1.1     elric 
    162  1.33     joerg __dead static void	 usage(void);
    163   1.1     elric 
    164   1.1     elric /* Verbose Framework */
    165  1.17    cbiere unsigned	verbose = 0;
    166   1.1     elric 
    167   1.1     elric #define VERBOSE(x,y)	if (verbose >= x) y
    168  1.19  christos #define VPRINTF(x,y)	if (verbose >= x) (void)printf y
    169   1.1     elric 
    170   1.1     elric static void
    171   1.1     elric usage(void)
    172   1.1     elric {
    173   1.1     elric 
    174  1.46     alnsn 	(void)fprintf(stderr, "usage: %s [-enpv] [-V vmeth] cgd dev "
    175  1.45       kre 	    "[paramsfile]\n", getprogname());
    176  1.46     alnsn 	(void)fprintf(stderr, "       %s -C [-enpv] [-f configfile]\n",
    177   1.1     elric 	    getprogname());
    178  1.46     alnsn 	(void)fprintf(stderr, "       %s -G [-enpv] [-i ivmeth] [-k kgmeth] "
    179   1.5     elric 	    "[-o outfile] paramsfile\n", getprogname());
    180  1.51      maya 	(void)fprintf(stderr, "       %s -g [-v] [-i ivmeth] [-k kgmeth] "
    181   1.5     elric 	    "[-o outfile] alg [keylen]\n", getprogname());
    182  1.45       kre 	(void)fprintf(stderr, "       %s -l [-v[v]] [cgd]\n", getprogname());
    183  1.19  christos 	(void)fprintf(stderr, "       %s -s [-nv] [-i ivmeth] cgd dev alg "
    184   1.1     elric 	    "[keylen]\n", getprogname());
    185  1.54  riastrad 	(void)fprintf(stderr, "       %s -t paramsfile\n", getprogname());
    186  1.55  riastrad 	(void)fprintf(stderr, "       %s -T [-f configfile]\n", getprogname());
    187  1.45       kre 	(void)fprintf(stderr, "       %s -U [-nv] [-f configfile]\n",
    188  1.45       kre 	    getprogname());
    189  1.19  christos 	(void)fprintf(stderr, "       %s -u [-nv] cgd\n", getprogname());
    190  1.17    cbiere 	exit(EXIT_FAILURE);
    191  1.17    cbiere }
    192  1.17    cbiere 
    193  1.17    cbiere static int
    194  1.19  christos parse_size_t(const char *s, size_t *l)
    195  1.17    cbiere {
    196  1.17    cbiere 	char *endptr;
    197  1.17    cbiere 	long v;
    198  1.17    cbiere 
    199  1.17    cbiere 	errno = 0;
    200  1.17    cbiere 	v = strtol(s, &endptr, 10);
    201  1.17    cbiere 	if ((v == LONG_MIN || v == LONG_MAX) && errno)
    202  1.17    cbiere 		return -1;
    203  1.17    cbiere 	if (v < INT_MIN || v > INT_MAX) {
    204  1.17    cbiere 		errno = ERANGE;
    205  1.17    cbiere 		return -1;
    206  1.17    cbiere 	}
    207  1.17    cbiere 	if (endptr == s) {
    208  1.17    cbiere 		errno = EINVAL;
    209  1.17    cbiere 		return -1;
    210  1.17    cbiere 	}
    211  1.19  christos 	*l = (size_t)v;
    212  1.19  christos 	return 0;
    213  1.17    cbiere }
    214  1.17    cbiere 
    215  1.17    cbiere static void
    216  1.17    cbiere set_action(enum action *action, enum action value)
    217  1.17    cbiere {
    218  1.17    cbiere 	if (*action != ACTION_DEFAULT)
    219  1.17    cbiere 		usage();
    220  1.17    cbiere 	*action = value;
    221   1.1     elric }
    222   1.1     elric 
    223   1.1     elric int
    224   1.1     elric main(int argc, char **argv)
    225   1.1     elric {
    226   1.5     elric 	struct params *p;
    227   1.5     elric 	struct params *tp;
    228   1.5     elric 	struct keygen *kg;
    229  1.17    cbiere 	enum action action = ACTION_DEFAULT;
    230   1.1     elric 	int	ch;
    231  1.17    cbiere 	const char	*cfile = NULL;
    232  1.17    cbiere 	const char	*outfile = NULL;
    233   1.1     elric 
    234  1.15     elric 	setprogname(*argv);
    235  1.56  riastrad 	if (hkdf_hmac_sha256_selftest())
    236  1.56  riastrad 		err(EXIT_FAILURE, "Crypto self-test failed");
    237  1.13     elric 	eliminate_cores();
    238  1.14     elric 	if (mlockall(MCL_FUTURE))
    239  1.14     elric 		err(EXIT_FAILURE, "Can't lock memory");
    240   1.5     elric 	p = params_new();
    241   1.5     elric 	kg = NULL;
    242   1.1     elric 
    243  1.55  riastrad 	while ((ch = getopt(argc, argv, "CGTUV:b:ef:gi:k:lno:sptuv")) != -1)
    244   1.1     elric 		switch (ch) {
    245   1.1     elric 		case 'C':
    246  1.17    cbiere 			set_action(&action, ACTION_CONFIGALL);
    247   1.1     elric 			break;
    248   1.5     elric 		case 'G':
    249  1.17    cbiere 			set_action(&action, ACTION_GENERATE_CONVERT);
    250   1.5     elric 			break;
    251  1.55  riastrad 		case 'T':
    252  1.55  riastrad 			set_action(&action, ACTION_PRINTALLKEYS);
    253  1.55  riastrad 			break;
    254   1.1     elric 		case 'U':
    255  1.17    cbiere 			set_action(&action, ACTION_UNCONFIGALL);
    256   1.1     elric 			break;
    257   1.3     elric 		case 'V':
    258   1.5     elric 			tp = params_verify_method(string_fromcharstar(optarg));
    259   1.5     elric 			if (!tp)
    260   1.3     elric 				usage();
    261   1.5     elric 			p = params_combine(p, tp);
    262   1.3     elric 			break;
    263   1.1     elric 		case 'b':
    264  1.17    cbiere 			{
    265  1.19  christos 				size_t size;
    266  1.17    cbiere 
    267  1.19  christos 				if (parse_size_t(optarg, &size) == -1)
    268  1.17    cbiere 					usage();
    269  1.17    cbiere 				tp = params_bsize(size);
    270  1.17    cbiere 				if (!tp)
    271  1.17    cbiere 					usage();
    272  1.17    cbiere 				p = params_combine(p, tp);
    273  1.17    cbiere 			}
    274   1.1     elric 			break;
    275  1.46     alnsn 		case 'e':
    276  1.46     alnsn 			pflag = PFLAG_GETPASS_ECHO;
    277  1.46     alnsn 			break;
    278   1.1     elric 		case 'f':
    279  1.17    cbiere 			if (cfile)
    280  1.17    cbiere 				usage();
    281  1.17    cbiere 			cfile = estrdup(optarg);
    282   1.1     elric 			break;
    283   1.1     elric 		case 'g':
    284  1.17    cbiere 			set_action(&action, ACTION_GENERATE);
    285   1.1     elric 			break;
    286   1.1     elric 		case 'i':
    287   1.5     elric 			tp = params_ivmeth(string_fromcharstar(optarg));
    288   1.5     elric 			p = params_combine(p, tp);
    289   1.1     elric 			break;
    290   1.1     elric 		case 'k':
    291   1.5     elric 			kg = keygen_method(string_fromcharstar(optarg));
    292   1.5     elric 			if (!kg)
    293   1.1     elric 				usage();
    294   1.5     elric 			keygen_addlist(&p->keygen, kg);
    295   1.1     elric 			break;
    296  1.34  christos 		case 'l':
    297  1.34  christos 			set_action(&action, ACTION_LIST);
    298  1.34  christos 			break;
    299   1.1     elric 		case 'n':
    300   1.1     elric 			nflag = 1;
    301   1.1     elric 			break;
    302   1.1     elric 		case 'o':
    303  1.17    cbiere 			if (outfile)
    304  1.17    cbiere 				usage();
    305  1.17    cbiere 			outfile = estrdup(optarg);
    306   1.1     elric 			break;
    307  1.22     elric 		case 'p':
    308  1.22     elric 			pflag = PFLAG_STDIN;
    309  1.22     elric 			break;
    310   1.1     elric 		case 's':
    311  1.17    cbiere 			set_action(&action, ACTION_CONFIGSTDIN);
    312   1.1     elric 			break;
    313  1.54  riastrad 		case 't':
    314  1.54  riastrad 			set_action(&action, ACTION_PRINTKEY);
    315  1.54  riastrad 			break;
    316   1.1     elric 		case 'u':
    317  1.17    cbiere 			set_action(&action, ACTION_UNCONFIGURE);
    318   1.1     elric 			break;
    319   1.1     elric 		case 'v':
    320   1.1     elric 			verbose++;
    321   1.1     elric 			break;
    322   1.1     elric 		default:
    323   1.1     elric 			usage();
    324   1.1     elric 			/* NOTREACHED */
    325   1.1     elric 		}
    326   1.1     elric 
    327   1.1     elric 	argc -= optind;
    328   1.1     elric 	argv += optind;
    329   1.1     elric 
    330  1.17    cbiere 	if (!outfile)
    331  1.17    cbiere 		outfile = "";
    332  1.17    cbiere 	if (!cfile)
    333  1.17    cbiere 		cfile = "";
    334  1.17    cbiere 
    335  1.32     pooka 	if (prog_init && prog_init() == -1)
    336  1.32     pooka 		err(1, "init failed");
    337  1.32     pooka 
    338   1.1     elric 	/* validate the consistency of the arguments */
    339   1.1     elric 
    340   1.1     elric 	switch (action) {
    341  1.17    cbiere 	case ACTION_DEFAULT:	/* ACTION_CONFIGURE is the default */
    342   1.1     elric 	case ACTION_CONFIGURE:
    343   1.5     elric 		return configure(argc, argv, p, CONFIG_FLAGS_FROMMAIN);
    344   1.1     elric 	case ACTION_UNCONFIGURE:
    345   1.3     elric 		return unconfigure(argc, argv, NULL, CONFIG_FLAGS_FROMMAIN);
    346   1.1     elric 	case ACTION_GENERATE:
    347   1.5     elric 		return generate(p, argc, argv, outfile);
    348   1.5     elric 	case ACTION_GENERATE_CONVERT:
    349   1.5     elric 		return generate_convert(p, argc, argv, outfile);
    350   1.1     elric 	case ACTION_CONFIGALL:
    351   1.1     elric 		return do_all(cfile, argc, argv, configure);
    352   1.1     elric 	case ACTION_UNCONFIGALL:
    353   1.1     elric 		return do_all(cfile, argc, argv, unconfigure);
    354   1.1     elric 	case ACTION_CONFIGSTDIN:
    355   1.5     elric 		return configure_stdin(p, argc, argv);
    356  1.34  christos 	case ACTION_LIST:
    357  1.34  christos 		return do_list(argc, argv);
    358  1.54  riastrad 	case ACTION_PRINTKEY:
    359  1.54  riastrad 		return do_printkey(argc, argv);
    360  1.55  riastrad 	case ACTION_PRINTALLKEYS:
    361  1.55  riastrad 		return do_all(cfile, argc, argv, printkey1);
    362  1.19  christos 	default:
    363  1.19  christos 		errx(EXIT_FAILURE, "undefined action");
    364  1.19  christos 		/* NOTREACHED */
    365   1.1     elric 	}
    366   1.1     elric }
    367   1.1     elric 
    368   1.5     elric static bits_t *
    369  1.56  riastrad getsubkey_hkdf_hmac_sha256(bits_t *key, bits_t *info, size_t subkeylen)
    370  1.56  riastrad {
    371  1.56  riastrad 	bits_t		*ret = NULL;
    372  1.56  riastrad 	uint8_t		*tmp;
    373  1.56  riastrad 
    374  1.56  riastrad 	tmp = emalloc(BITS2BYTES(subkeylen));
    375  1.56  riastrad 	if (hkdf_hmac_sha256(tmp, BITS2BYTES(subkeylen),
    376  1.56  riastrad 		bits_getbuf(key), BITS2BYTES(bits_len(key)),
    377  1.56  riastrad 		bits_getbuf(info), BITS2BYTES(bits_len(info)))) {
    378  1.56  riastrad 		warnx("failed to derive HKDF-HMAC-SHA256 subkey");
    379  1.56  riastrad 		goto out;
    380  1.56  riastrad 	}
    381  1.56  riastrad 
    382  1.56  riastrad 	ret = bits_new(tmp, subkeylen);
    383  1.56  riastrad 
    384  1.56  riastrad out:	free(tmp);
    385  1.56  riastrad 	return ret;
    386  1.56  riastrad }
    387  1.56  riastrad 
    388  1.56  riastrad static bits_t *
    389  1.56  riastrad getsubkey(int alg, bits_t *key, bits_t *info, size_t subkeylen)
    390  1.56  riastrad {
    391  1.56  riastrad 
    392  1.56  riastrad 	switch (alg) {
    393  1.56  riastrad 	case SHARED_ALG_HKDF_HMAC_SHA256:
    394  1.56  riastrad 		return getsubkey_hkdf_hmac_sha256(key, info, subkeylen);
    395  1.56  riastrad 	default:
    396  1.56  riastrad 		warnx("unrecognised shared key derivation method %d", alg);
    397  1.56  riastrad 		return NULL;
    398  1.56  riastrad 	}
    399  1.56  riastrad }
    400  1.56  riastrad 
    401  1.56  riastrad static bits_t *
    402  1.56  riastrad getkey(const char *dev, struct keygen *kg, size_t len0)
    403   1.1     elric {
    404   1.5     elric 	bits_t	*ret = NULL;
    405   1.5     elric 	bits_t	*tmp;
    406   1.1     elric 
    407  1.56  riastrad 	VPRINTF(3, ("getkey(\"%s\", %p, %zu) called\n", dev, kg, len0));
    408   1.5     elric 	for (; kg; kg=kg->next) {
    409  1.56  riastrad 		struct sharedkey *sk = NULL;
    410  1.56  riastrad 		size_t len = len0;
    411  1.56  riastrad 
    412  1.56  riastrad 		/*
    413  1.56  riastrad 		 * If shared, determine the shared key's length and
    414  1.56  riastrad 		 * probe the cache of shared keys.
    415  1.56  riastrad 		 */
    416  1.56  riastrad 		if (kg->kg_sharedid) {
    417  1.56  riastrad 			const char *id = string_tocharstar(kg->kg_sharedid);
    418  1.56  riastrad 
    419  1.56  riastrad 			len = kg->kg_sharedlen;
    420  1.56  riastrad 			LIST_FOREACH(sk, &sharedkeys, list) {
    421  1.56  riastrad 				if (kg->kg_sharedalg == sk->alg &&
    422  1.56  riastrad 				    kg->kg_sharedlen == bits_len(sk->key) &&
    423  1.56  riastrad 				    strcmp(id, string_tocharstar(sk->id)) == 0)
    424  1.56  riastrad 					break;
    425  1.56  riastrad 			}
    426  1.56  riastrad 			if (sk) {
    427  1.56  riastrad 				tmp = sk->key;
    428  1.56  riastrad 				goto derive;
    429  1.56  riastrad 			}
    430  1.56  riastrad 		}
    431  1.56  riastrad 
    432   1.5     elric 		switch (kg->kg_method) {
    433   1.5     elric 		case KEYGEN_STOREDKEY:
    434   1.5     elric 			tmp = getkey_storedkey(dev, kg, len);
    435   1.5     elric 			break;
    436   1.5     elric 		case KEYGEN_RANDOMKEY:
    437  1.12        tv 			tmp = getkey_randomkey(dev, kg, len, 1);
    438  1.12        tv 			break;
    439  1.12        tv 		case KEYGEN_URANDOMKEY:
    440  1.12        tv 			tmp = getkey_randomkey(dev, kg, len, 0);
    441   1.5     elric 			break;
    442  1.53       nia #ifdef HAVE_ARGON2
    443  1.53       nia 		case KEYGEN_ARGON2ID:
    444  1.53       nia 			tmp = getkey_argon2id(dev, kg, len);
    445  1.53       nia 			break;
    446  1.53       nia #endif
    447  1.10       dan 		case KEYGEN_PKCS5_PBKDF2_SHA1:
    448  1.10       dan 			tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 0);
    449  1.10       dan 			break;
    450  1.10       dan 		/* provide backwards compatibility for old config files */
    451  1.10       dan 		case KEYGEN_PKCS5_PBKDF2_OLD:
    452  1.10       dan 			tmp = getkey_pkcs5_pbkdf2(dev, kg, len, 1);
    453   1.5     elric 			break;
    454  1.23     elric 		case KEYGEN_SHELL_CMD:
    455  1.23     elric 			tmp = getkey_shell_cmd(dev, kg, len);
    456  1.23     elric 			break;
    457   1.5     elric 		default:
    458   1.5     elric 			warnx("unrecognised keygen method %d in getkey()",
    459   1.5     elric 			    kg->kg_method);
    460   1.5     elric 			if (ret)
    461   1.5     elric 				bits_free(ret);
    462   1.5     elric 			return NULL;
    463   1.5     elric 		}
    464   1.5     elric 
    465  1.56  riastrad 		/*
    466  1.56  riastrad 		 * If shared, cache the key.
    467  1.56  riastrad 		 */
    468  1.56  riastrad 		if (kg->kg_sharedid) {
    469  1.56  riastrad 			assert(sk == NULL);
    470  1.56  riastrad 			sk = ecalloc(1, sizeof(*sk));
    471  1.56  riastrad 			sk->alg = kg->kg_sharedalg;
    472  1.56  riastrad 			sk->id = string_dup(kg->kg_sharedid);
    473  1.56  riastrad 			sk->key = tmp;
    474  1.56  riastrad 			LIST_INSERT_HEAD(&sharedkeys, sk, list);
    475  1.56  riastrad 		}
    476  1.56  riastrad 
    477  1.56  riastrad derive:		if (kg->kg_sharedid) {
    478  1.56  riastrad 			/*
    479  1.56  riastrad 			 * tmp holds the master key, owned by the
    480  1.56  riastrad 			 * struct sharedkey record; replace it by the
    481  1.56  riastrad 			 * derived subkey.
    482  1.56  riastrad 			 */
    483  1.56  riastrad 			tmp = getsubkey(kg->kg_sharedalg, tmp,
    484  1.56  riastrad 			    kg->kg_sharedinfo, len0);
    485  1.56  riastrad 			if (tmp == NULL) {
    486  1.56  riastrad 				if (ret)
    487  1.56  riastrad 					bits_free(ret);
    488  1.56  riastrad 				return NULL;
    489  1.56  riastrad 			}
    490  1.56  riastrad 		}
    491   1.5     elric 		if (ret)
    492   1.5     elric 			ret = bits_xor_d(tmp, ret);
    493   1.5     elric 		else
    494   1.5     elric 			ret = tmp;
    495   1.1     elric 	}
    496   1.5     elric 
    497   1.5     elric 	return ret;
    498   1.5     elric }
    499   1.5     elric 
    500   1.5     elric /*ARGSUSED*/
    501   1.5     elric static bits_t *
    502  1.19  christos getkey_storedkey(const char *target, struct keygen *kg, size_t keylen)
    503   1.5     elric {
    504   1.5     elric 	return bits_dup(kg->kg_key);
    505   1.1     elric }
    506   1.1     elric 
    507   1.5     elric /*ARGSUSED*/
    508   1.5     elric static bits_t *
    509  1.19  christos getkey_randomkey(const char *target, struct keygen *kg, size_t keylen, int hard)
    510   1.1     elric {
    511  1.12        tv 	return bits_getrandombits(keylen, hard);
    512   1.1     elric }
    513   1.1     elric 
    514  1.22     elric static char *
    515  1.22     elric maybe_getpass(char *prompt)
    516  1.22     elric {
    517  1.22     elric 	char	 buf[1024];
    518  1.46     alnsn 	char	*p = NULL;
    519  1.46     alnsn 	char	*tmp, *pass;
    520  1.22     elric 
    521  1.22     elric 	switch (pflag) {
    522  1.22     elric 	case PFLAG_GETPASS:
    523  1.46     alnsn 		p = getpass_r(prompt, buf, sizeof(buf));
    524  1.46     alnsn 		break;
    525  1.46     alnsn 
    526  1.46     alnsn 	case PFLAG_GETPASS_ECHO:
    527  1.46     alnsn 		p = getpassfd(prompt, buf, sizeof(buf), NULL,
    528  1.46     alnsn 		    GETPASS_ECHO|GETPASS_ECHO_NL|GETPASS_NEED_TTY, 0);
    529  1.22     elric 		break;
    530  1.22     elric 
    531  1.22     elric 	case PFLAG_STDIN:
    532  1.22     elric 		p = fgets(buf, sizeof(buf), stdin);
    533  1.22     elric 		if (p) {
    534  1.22     elric 			tmp = strchr(p, '\n');
    535  1.22     elric 			if (tmp)
    536  1.22     elric 				*tmp = '\0';
    537  1.22     elric 		}
    538  1.22     elric 		break;
    539  1.22     elric 
    540  1.22     elric 	default:
    541  1.22     elric 		errx(EXIT_FAILURE, "pflag set inappropriately?");
    542  1.22     elric 	}
    543  1.22     elric 
    544  1.22     elric 	if (!p)
    545  1.22     elric 		err(EXIT_FAILURE, "failed to read passphrase");
    546  1.22     elric 
    547  1.46     alnsn 	pass = estrdup(p);
    548  1.48     alnsn 	explicit_memset(buf, 0, sizeof(buf));
    549  1.46     alnsn 
    550  1.46     alnsn 	return pass;
    551  1.22     elric }
    552  1.22     elric 
    553   1.5     elric /*ARGSUSED*/
    554  1.29     elric /*
    555  1.10       dan  * XXX take, and pass through, a compat flag that indicates whether we
    556  1.10       dan  * provide backwards compatibility with a previous bug.  The previous
    557  1.10       dan  * behaviour is indicated by the keygen method pkcs5_pbkdf2, and a
    558  1.10       dan  * non-zero compat flag. The new default, and correct keygen method is
    559  1.10       dan  * called pcks5_pbkdf2/sha1.  When the old method is removed, so will
    560  1.10       dan  * be the compat argument.
    561  1.10       dan  */
    562   1.5     elric static bits_t *
    563  1.19  christos getkey_pkcs5_pbkdf2(const char *target, struct keygen *kg, size_t keylen,
    564  1.19  christos     int compat)
    565   1.1     elric {
    566   1.5     elric 	bits_t		*ret;
    567  1.25  dholland 	char		*passp;
    568   1.5     elric 	char		 buf[1024];
    569   1.5     elric 	u_int8_t	*tmp;
    570   1.1     elric 
    571  1.46     alnsn 	snprintf(buf, sizeof(buf), "%s's passphrase%s:", target,
    572  1.46     alnsn 	    pflag & PFLAG_GETPASS_ECHO ? " (echo)" : "");
    573  1.25  dholland 	passp = maybe_getpass(buf);
    574  1.25  dholland 	if (pkcs5_pbkdf2(&tmp, BITS2BYTES(keylen), (uint8_t *)passp,
    575  1.25  dholland 	    strlen(passp),
    576   1.5     elric 	    bits_getbuf(kg->kg_salt), BITS2BYTES(bits_len(kg->kg_salt)),
    577  1.10       dan 	    kg->kg_iterations, compat)) {
    578   1.5     elric 		warnx("failed to generate PKCS#5 PBKDF2 key");
    579   1.5     elric 		return NULL;
    580   1.5     elric 	}
    581   1.5     elric 
    582   1.5     elric 	ret = bits_new(tmp, keylen);
    583   1.9        cb 	kg->kg_key = bits_dup(ret);
    584  1.48     alnsn 	explicit_memset(passp, 0, strlen(passp));
    585  1.22     elric 	free(passp);
    586   1.5     elric 	free(tmp);
    587   1.1     elric 	return ret;
    588   1.1     elric }
    589   1.1     elric 
    590  1.53       nia #ifdef HAVE_ARGON2
    591  1.53       nia static bits_t *
    592  1.53       nia getkey_argon2id(const char *target, struct keygen *kg, size_t keylen)
    593  1.53       nia {
    594  1.53       nia 	bits_t *ret;
    595  1.53       nia 	char *passp;
    596  1.53       nia 	char buf[1024];
    597  1.53       nia 	uint8_t	raw[256];
    598  1.53       nia 	int err;
    599  1.53       nia 
    600  1.53       nia 	snprintf(buf, sizeof(buf), "%s's passphrase%s:", target,
    601  1.53       nia 	    pflag & PFLAG_GETPASS_ECHO ? " (echo)" : "");
    602  1.53       nia 	passp = maybe_getpass(buf);
    603  1.53       nia 	if ((err = argon2_hash(kg->kg_iterations, kg->kg_memory,
    604  1.53       nia 	    kg->kg_parallelism,
    605  1.53       nia 	    passp, strlen(passp),
    606  1.53       nia 	    bits_getbuf(kg->kg_salt),
    607  1.53       nia 	    BITS2BYTES(bits_len(kg->kg_salt)),
    608  1.53       nia 	    raw, sizeof(raw),
    609  1.53       nia 	    NULL, 0,
    610  1.53       nia 	    Argon2_id, kg->kg_version)) != ARGON2_OK) {
    611  1.53       nia 		warnx("failed to generate Argon2id key, error code %d", err);
    612  1.53       nia 		return NULL;
    613  1.53       nia 	}
    614  1.53       nia 
    615  1.53       nia 	ret = bits_new(raw, keylen);
    616  1.53       nia 	kg->kg_key = bits_dup(ret);
    617  1.53       nia 	explicit_memset(passp, 0, strlen(passp));
    618  1.53       nia 	explicit_memset(raw, 0, sizeof(raw));
    619  1.53       nia 	free(passp);
    620  1.53       nia 	return ret;
    621  1.53       nia }
    622  1.53       nia #endif
    623  1.53       nia 
    624   1.5     elric /*ARGSUSED*/
    625  1.23     elric static bits_t *
    626  1.23     elric getkey_shell_cmd(const char *target, struct keygen *kg, size_t keylen)
    627  1.23     elric {
    628  1.23     elric 	FILE	*f;
    629  1.23     elric 	bits_t	*ret;
    630  1.52  riastrad 	int	status;
    631  1.23     elric 
    632  1.52  riastrad 	if ((f = popen(string_tocharstar(kg->kg_cmd), "r")) == NULL)
    633  1.52  riastrad 		errx(1, "command failed");
    634  1.52  riastrad 	if ((ret = bits_fget(f, keylen)) == NULL)
    635  1.52  riastrad 		errx(1, "command output too short");
    636  1.52  riastrad 	if ((status = pclose(f)) != 0)
    637  1.52  riastrad 		err(1, "command failed with status %d", status);
    638  1.23     elric 
    639  1.23     elric 	return ret;
    640  1.23     elric }
    641  1.23     elric 
    642  1.23     elric /*ARGSUSED*/
    643   1.1     elric static int
    644   1.3     elric unconfigure(int argc, char **argv, struct params *inparams, int flags)
    645   1.1     elric {
    646   1.1     elric 	int	fd;
    647   1.1     elric 	int	ret;
    648   1.1     elric 	char	buf[MAXPATHLEN] = "";
    649   1.1     elric 
    650   1.1     elric 	/* only complain about additional arguments, if called from main() */
    651   1.1     elric 	if (flags == CONFIG_FLAGS_FROMMAIN && argc != 1)
    652   1.1     elric 		usage();
    653   1.1     elric 
    654   1.1     elric 	/* if called from do_all(), then ensure that 2 or 3 args exist */
    655   1.1     elric 	if (flags == CONFIG_FLAGS_FROMALL && (argc < 2 || argc > 3))
    656   1.1     elric 		return -1;
    657   1.1     elric 
    658  1.32     pooka 	fd = opendisk1(*argv, O_RDWR, buf, sizeof(buf), 1, prog_open);
    659   1.1     elric 	if (fd == -1) {
    660  1.17    cbiere 		int saved_errno = errno;
    661  1.17    cbiere 
    662   1.5     elric 		warn("can't open cgd \"%s\", \"%s\"", *argv, buf);
    663   1.1     elric 
    664   1.1     elric 		/* this isn't fatal with nflag != 0 */
    665   1.1     elric 		if (!nflag)
    666  1.17    cbiere 			return saved_errno;
    667   1.1     elric 	}
    668   1.1     elric 
    669   1.1     elric 	VPRINTF(1, ("%s (%s): clearing\n", *argv, buf));
    670   1.1     elric 
    671   1.1     elric 	if (nflag)
    672   1.1     elric 		return 0;
    673   1.1     elric 
    674   1.3     elric 	ret = unconfigure_fd(fd);
    675  1.32     pooka 	(void)prog_close(fd);
    676   1.3     elric 	return ret;
    677   1.3     elric }
    678   1.3     elric 
    679   1.3     elric static int
    680   1.3     elric unconfigure_fd(int fd)
    681   1.3     elric {
    682   1.3     elric 	struct	cgd_ioctl ci;
    683   1.3     elric 
    684  1.32     pooka 	if (prog_ioctl(fd, CGDIOCCLR, &ci) == -1) {
    685  1.17    cbiere 		warn("ioctl");
    686   1.3     elric 		return -1;
    687   1.1     elric 	}
    688   1.1     elric 
    689   1.1     elric 	return 0;
    690   1.1     elric }
    691   1.1     elric 
    692   1.5     elric /*ARGSUSED*/
    693   1.1     elric static int
    694   1.3     elric configure(int argc, char **argv, struct params *inparams, int flags)
    695   1.1     elric {
    696   1.5     elric 	struct params	*p;
    697  1.22     elric 	struct keygen	*kg;
    698   1.5     elric 	int		 fd;
    699  1.22     elric 	int		 loop = 0;
    700   1.5     elric 	int		 ret;
    701   1.5     elric 	char		 cgdname[PATH_MAX];
    702  1.38  christos 	char		 devicename[PATH_MAX];
    703  1.39  christos 	const char	*dev = NULL;	/* XXX: gcc */
    704  1.36   mlelstv 
    705  1.44       kre 	if (argc < 2 || argc > 3) {
    706  1.44       kre 		/* print usage and exit, only if called from main() */
    707  1.44       kre 		if (flags == CONFIG_FLAGS_FROMMAIN) {
    708  1.44       kre 			warnx("wrong number of args");
    709  1.44       kre 			usage();
    710  1.44       kre 		}
    711  1.44       kre 		return -1;
    712  1.44       kre 	}
    713  1.44       kre 
    714  1.42       kre 	if ((
    715  1.42       kre 	  fd = opendisk1(*argv, O_RDWR, cgdname, sizeof(cgdname), 1, prog_open)
    716  1.42       kre 	    ) != -1) {
    717  1.42       kre 		struct cgd_user cgu;
    718  1.42       kre 
    719  1.42       kre 		cgu.cgu_unit = -1;
    720  1.42       kre 		if (prog_ioctl(fd, CGDIOCGET, &cgu) != -1 && cgu.cgu_dev != 0) {
    721  1.42       kre 			warnx("device %s already in use", *argv);
    722  1.43       kre 			prog_close(fd);
    723  1.42       kre 			return -1;
    724  1.42       kre 		}
    725  1.43       kre 		prog_close(fd);
    726  1.42       kre 	}
    727  1.42       kre 
    728  1.44       kre 	dev = getfsspecname(devicename, sizeof(devicename), argv[1]);
    729  1.44       kre 	if (dev == NULL) {
    730  1.44       kre 		warnx("getfsspecname failed: %s", devicename);
    731  1.44       kre 		return -1;
    732  1.36   mlelstv 	}
    733   1.1     elric 
    734  1.34  christos 	if (argc == 2) {
    735  1.40  christos 		char pfile[MAXPATHLEN];
    736  1.36   mlelstv 
    737  1.36   mlelstv 		/* make string writable for basename */
    738  1.40  christos 		strlcpy(pfile, dev, sizeof(pfile));
    739  1.40  christos 		p = params_cget(basename(pfile));
    740  1.44       kre 	} else
    741  1.17    cbiere 		p = params_cget(argv[2]);
    742   1.1     elric 
    743   1.5     elric 	if (!p)
    744   1.5     elric 		return -1;
    745   1.2     elric 
    746   1.3     elric 	/*
    747   1.5     elric 	 * over-ride with command line specifications and fill in default
    748   1.5     elric 	 * values.
    749   1.3     elric 	 */
    750   1.3     elric 
    751   1.5     elric 	p = params_combine(p, inparams);
    752   1.5     elric 	ret = params_filldefaults(p);
    753   1.5     elric 	if (ret) {
    754   1.5     elric 		params_free(p);
    755   1.5     elric 		return ret;
    756   1.5     elric 	}
    757   1.5     elric 
    758   1.5     elric 	if (!params_verify(p)) {
    759   1.5     elric 		warnx("params invalid");
    760   1.5     elric 		return -1;
    761   1.5     elric 	}
    762   1.3     elric 
    763   1.3     elric 	/*
    764   1.3     elric 	 * loop over configuring the disk and checking to see if it
    765   1.3     elric 	 * verifies properly.  We open and close the disk device each
    766   1.3     elric 	 * time, because if the user passes us the block device we
    767   1.3     elric 	 * need to flush the buffer cache.
    768  1.22     elric 	 *
    769  1.22     elric 	 * We only loop if one of the verification methods prompts for
    770  1.22     elric 	 * a password.
    771   1.3     elric 	 */
    772   1.3     elric 
    773  1.47     alnsn 	for (kg = p->keygen;
    774  1.47     alnsn 	    (pflag & PFLAG_GETPASS_MASK) && kg;
    775  1.47     alnsn 	    kg = kg->next)
    776  1.53       nia 		if (kg->kg_method == KEYGEN_ARGON2ID ||
    777  1.53       nia 		    kg->kg_method == KEYGEN_PKCS5_PBKDF2_SHA1 ||
    778  1.53       nia 		    kg->kg_method == KEYGEN_PKCS5_PBKDF2_OLD) {
    779  1.22     elric 			loop = 1;
    780  1.22     elric 			break;
    781  1.22     elric 		}
    782  1.22     elric 
    783   1.3     elric 	for (;;) {
    784   1.3     elric 		fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
    785   1.3     elric 		if (fd == -1)
    786   1.3     elric 			return -1;
    787   1.3     elric 
    788   1.5     elric 		if (p->key)
    789   1.5     elric 			bits_free(p->key);
    790   1.5     elric 
    791   1.5     elric 		p->key = getkey(argv[1], p->keygen, p->keylen);
    792   1.5     elric 		if (!p->key)
    793   1.3     elric 			goto bail_err;
    794   1.3     elric 
    795  1.36   mlelstv 		ret = configure_params(fd, cgdname, dev, p);
    796   1.3     elric 		if (ret)
    797   1.3     elric 			goto bail_err;
    798   1.3     elric 
    799   1.5     elric 		ret = verify(p, fd);
    800  1.49       kre 		if (ret == -1) {
    801  1.49       kre 			(void)unconfigure_fd(fd);
    802   1.3     elric 			goto bail_err;
    803  1.49       kre 		}
    804  1.50       kre 		if (ret == 0)		/* success */
    805   1.3     elric 			break;
    806   1.2     elric 
    807  1.19  christos 		(void)unconfigure_fd(fd);
    808  1.32     pooka 		(void)prog_close(fd);
    809  1.22     elric 
    810  1.22     elric 		if (!loop) {
    811  1.22     elric 			warnx("verification failed permanently");
    812  1.22     elric 			goto bail_err;
    813  1.22     elric 		}
    814  1.22     elric 
    815  1.22     elric 		warnx("verification failed, please reenter passphrase");
    816   1.3     elric 	}
    817   1.2     elric 
    818   1.5     elric 	params_free(p);
    819  1.32     pooka 	(void)prog_close(fd);
    820   1.3     elric 	return 0;
    821  1.50       kre 
    822  1.50       kre  bail_err:;
    823   1.5     elric 	params_free(p);
    824  1.32     pooka 	(void)prog_close(fd);
    825   1.3     elric 	return -1;
    826   1.1     elric }
    827   1.1     elric 
    828   1.1     elric static int
    829   1.1     elric configure_stdin(struct params *p, int argc, char **argv)
    830   1.1     elric {
    831  1.36   mlelstv 	int		 fd;
    832  1.36   mlelstv 	int		 ret;
    833  1.36   mlelstv 	char		 cgdname[PATH_MAX];
    834  1.38  christos 	char		 devicename[PATH_MAX];
    835  1.36   mlelstv 	const char	*dev;
    836   1.1     elric 
    837   1.1     elric 	if (argc < 3 || argc > 4)
    838   1.1     elric 		usage();
    839   1.1     elric 
    840  1.38  christos 	dev = getfsspecname(devicename, sizeof(devicename), argv[1]);
    841  1.36   mlelstv 	if (dev == NULL) {
    842  1.38  christos 		warnx("getfsspecname failed: %s", devicename);
    843  1.36   mlelstv 		return -1;
    844  1.36   mlelstv 	}
    845  1.36   mlelstv 
    846   1.5     elric 	p->algorithm = string_fromcharstar(argv[2]);
    847  1.17    cbiere 	if (argc > 3) {
    848  1.19  christos 		size_t keylen;
    849  1.17    cbiere 
    850  1.19  christos 		if (parse_size_t(argv[3], &keylen) == -1) {
    851  1.17    cbiere 			warn("failed to parse key length");
    852  1.17    cbiere 			return -1;
    853  1.17    cbiere 		}
    854  1.17    cbiere 		p->keylen = keylen;
    855  1.17    cbiere 	}
    856   1.1     elric 
    857   1.1     elric 	ret = params_filldefaults(p);
    858   1.1     elric 	if (ret)
    859   1.1     elric 		return ret;
    860   1.1     elric 
    861   1.2     elric 	fd = opendisk_werror(argv[0], cgdname, sizeof(cgdname));
    862   1.2     elric 	if (fd == -1)
    863   1.2     elric 		return -1;
    864   1.2     elric 
    865   1.5     elric 	p->key = bits_fget(stdin, p->keylen);
    866   1.5     elric 	if (!p->key) {
    867   1.5     elric 		warnx("failed to read key from stdin");
    868   1.1     elric 		return -1;
    869   1.5     elric 	}
    870   1.1     elric 
    871  1.36   mlelstv 	return configure_params(fd, cgdname, dev, p);
    872   1.1     elric }
    873   1.1     elric 
    874   1.1     elric static int
    875  1.17    cbiere opendisk_werror(const char *cgd, char *buf, size_t buflen)
    876   1.2     elric {
    877   1.2     elric 	int	fd;
    878   1.2     elric 
    879  1.50       kre 	VPRINTF(3, ("opendisk_werror(%s, %s, %zu) called.\n", cgd,buf,buflen));
    880   1.5     elric 
    881   1.2     elric 	/* sanity */
    882   1.2     elric 	if (!cgd || !buf)
    883   1.2     elric 		return -1;
    884   1.2     elric 
    885   1.2     elric 	if (nflag) {
    886  1.17    cbiere 		if (strlcpy(buf, cgd, buflen) >= buflen)
    887  1.17    cbiere 			return -1;
    888   1.2     elric 		return 0;
    889   1.2     elric 	}
    890   1.2     elric 
    891  1.32     pooka 	fd = opendisk1(cgd, O_RDWR, buf, buflen, 0, prog_open);
    892   1.2     elric 	if (fd == -1)
    893   1.5     elric 		warnx("can't open cgd \"%s\", \"%s\"", cgd, buf);
    894   1.5     elric 
    895   1.2     elric 	return fd;
    896   1.2     elric }
    897   1.2     elric 
    898   1.2     elric static int
    899   1.2     elric configure_params(int fd, const char *cgd, const char *dev, struct params *p)
    900   1.1     elric {
    901   1.1     elric 	struct cgd_ioctl ci;
    902   1.1     elric 
    903   1.1     elric 	/* sanity */
    904   1.1     elric 	if (!cgd || !dev)
    905   1.1     elric 		return -1;
    906   1.1     elric 
    907  1.19  christos 	(void)memset(&ci, 0x0, sizeof(ci));
    908  1.16  christos 	ci.ci_disk = dev;
    909  1.16  christos 	ci.ci_alg = string_tocharstar(p->algorithm);
    910  1.16  christos 	ci.ci_ivmethod = string_tocharstar(p->ivmeth);
    911  1.16  christos 	ci.ci_key = bits_getbuf(p->key);
    912   1.1     elric 	ci.ci_keylen = p->keylen;
    913   1.1     elric 	ci.ci_blocksize = p->bsize;
    914   1.1     elric 
    915  1.20    martin 	VPRINTF(1, ("    with alg %s keylen %zu blocksize %zu ivmethod %s\n",
    916   1.5     elric 	    string_tocharstar(p->algorithm), p->keylen, p->bsize,
    917   1.5     elric 	    string_tocharstar(p->ivmeth)));
    918   1.5     elric 	VPRINTF(2, ("key: "));
    919   1.5     elric 	VERBOSE(2, bits_fprint(stdout, p->key));
    920   1.5     elric 	VPRINTF(2, ("\n"));
    921   1.1     elric 
    922   1.1     elric 	if (nflag)
    923   1.1     elric 		return 0;
    924   1.1     elric 
    925  1.32     pooka 	if (prog_ioctl(fd, CGDIOCSET, &ci) == -1) {
    926  1.17    cbiere 		int saved_errno = errno;
    927  1.17    cbiere 		warn("ioctl");
    928  1.17    cbiere 		return saved_errno;
    929   1.1     elric 	}
    930   1.1     elric 
    931   1.1     elric 	return 0;
    932   1.1     elric }
    933   1.1     elric 
    934   1.3     elric /*
    935   1.3     elric  * verify returns 0 for success, -1 for unrecoverable error, or 1 for retry.
    936   1.3     elric  */
    937   1.3     elric 
    938   1.3     elric #define SCANSIZE	8192
    939   1.3     elric 
    940   1.3     elric static int
    941   1.3     elric verify(struct params *p, int fd)
    942   1.3     elric {
    943   1.3     elric 
    944   1.3     elric 	switch (p->verify_method) {
    945   1.3     elric 	case VERIFY_NONE:
    946   1.3     elric 		return 0;
    947   1.3     elric 	case VERIFY_DISKLABEL:
    948   1.5     elric 		return verify_disklabel(fd);
    949   1.5     elric 	case VERIFY_FFS:
    950   1.5     elric 		return verify_ffs(fd);
    951   1.9        cb 	case VERIFY_REENTER:
    952   1.9        cb 		return verify_reenter(p);
    953  1.37   mlelstv 	case VERIFY_MBR:
    954  1.37   mlelstv 		return verify_mbr(fd);
    955  1.37   mlelstv 	case VERIFY_GPT:
    956  1.37   mlelstv 		return verify_gpt(fd);
    957   1.3     elric 	default:
    958   1.5     elric 		warnx("unimplemented verification method");
    959   1.3     elric 		return -1;
    960   1.3     elric 	}
    961   1.5     elric }
    962   1.5     elric 
    963   1.5     elric static int
    964   1.5     elric verify_disklabel(int fd)
    965   1.5     elric {
    966   1.5     elric 	struct	disklabel l;
    967  1.17    cbiere 	ssize_t	ret;
    968   1.5     elric 	char	buf[SCANSIZE];
    969   1.3     elric 
    970   1.3     elric 	/*
    971   1.3     elric 	 * we simply scan the first few blocks for a disklabel, ignoring
    972   1.3     elric 	 * any MBR/filecore sorts of logic.  MSDOS and RiscOS can't read
    973   1.3     elric 	 * a cgd, anyway, so it is unlikely that there will be non-native
    974   1.3     elric 	 * partition information.
    975   1.3     elric 	 */
    976   1.3     elric 
    977  1.37   mlelstv 	ret = prog_pread(fd, buf, SCANSIZE, 0);
    978  1.17    cbiere 	if (ret < 0) {
    979   1.5     elric 		warn("can't read disklabel area");
    980   1.3     elric 		return -1;
    981   1.3     elric 	}
    982   1.3     elric 
    983   1.3     elric 	/* now scan for the disklabel */
    984   1.3     elric 
    985  1.17    cbiere 	return disklabel_scan(&l, buf, (size_t)ret);
    986   1.3     elric }
    987   1.3     elric 
    988  1.37   mlelstv static int
    989  1.37   mlelstv verify_mbr(int fd)
    990  1.37   mlelstv {
    991  1.37   mlelstv 	struct mbr_sector mbr;
    992  1.37   mlelstv 	ssize_t	ret;
    993  1.37   mlelstv 	char	buf[SCANSIZE];
    994  1.37   mlelstv 
    995  1.37   mlelstv 	/*
    996  1.37   mlelstv 	 * we read the first blocks to avoid sector size issues and
    997  1.37   mlelstv 	 * verify the MBR in the beginning
    998  1.37   mlelstv 	 */
    999  1.37   mlelstv 
   1000  1.37   mlelstv 	ret = prog_pread(fd, buf, SCANSIZE, 0);
   1001  1.37   mlelstv 	if (ret < 0) {
   1002  1.37   mlelstv 		warn("can't read mbr area");
   1003  1.37   mlelstv 		return -1;
   1004  1.37   mlelstv 	}
   1005  1.37   mlelstv 
   1006  1.37   mlelstv 	memcpy(&mbr, buf, sizeof(mbr));
   1007  1.37   mlelstv 	if (le16toh(mbr.mbr_magic) != MBR_MAGIC)
   1008  1.49       kre 		return 1;
   1009  1.37   mlelstv 
   1010  1.37   mlelstv 	return 0;
   1011  1.37   mlelstv }
   1012  1.37   mlelstv 
   1013  1.37   mlelstv static uint32_t crc32_tab[] = {
   1014  1.37   mlelstv 	0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,
   1015  1.37   mlelstv 	0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
   1016  1.37   mlelstv 	0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
   1017  1.37   mlelstv 	0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
   1018  1.37   mlelstv 	0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
   1019  1.37   mlelstv 	0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
   1020  1.37   mlelstv 	0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,
   1021  1.37   mlelstv 	0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
   1022  1.37   mlelstv 	0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
   1023  1.37   mlelstv 	0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
   1024  1.37   mlelstv 	0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,
   1025  1.37   mlelstv 	0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
   1026  1.37   mlelstv 	0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,
   1027  1.37   mlelstv 	0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
   1028  1.37   mlelstv 	0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
   1029  1.37   mlelstv 	0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
   1030  1.37   mlelstv 	0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,
   1031  1.37   mlelstv 	0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
   1032  1.37   mlelstv 	0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,
   1033  1.37   mlelstv 	0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
   1034  1.37   mlelstv 	0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
   1035  1.37   mlelstv 	0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
   1036  1.37   mlelstv 	0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,
   1037  1.37   mlelstv 	0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
   1038  1.37   mlelstv 	0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
   1039  1.37   mlelstv 	0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
   1040  1.37   mlelstv 	0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
   1041  1.37   mlelstv 	0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
   1042  1.37   mlelstv 	0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,
   1043  1.37   mlelstv 	0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
   1044  1.37   mlelstv 	0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,
   1045  1.37   mlelstv 	0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
   1046  1.37   mlelstv 	0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
   1047  1.37   mlelstv 	0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
   1048  1.37   mlelstv 	0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
   1049  1.37   mlelstv 	0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
   1050  1.37   mlelstv 	0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,
   1051  1.37   mlelstv 	0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
   1052  1.37   mlelstv 	0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
   1053  1.37   mlelstv 	0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
   1054  1.37   mlelstv 	0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,
   1055  1.37   mlelstv 	0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
   1056  1.37   mlelstv 	0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d
   1057  1.37   mlelstv };
   1058  1.37   mlelstv 
   1059  1.37   mlelstv static uint32_t
   1060  1.37   mlelstv crc32(const void *buf, size_t size)
   1061  1.37   mlelstv {
   1062  1.37   mlelstv 	const uint8_t *p;
   1063  1.37   mlelstv 	uint32_t crc;
   1064  1.37   mlelstv 
   1065  1.37   mlelstv 	p = buf;
   1066  1.37   mlelstv 	crc = ~0U;
   1067  1.37   mlelstv 
   1068  1.37   mlelstv 	while (size--)
   1069  1.37   mlelstv 		crc = crc32_tab[(crc ^ *p++) & 0xFF] ^ (crc >> 8);
   1070  1.37   mlelstv 
   1071  1.37   mlelstv 	return crc ^ ~0U;
   1072  1.37   mlelstv }
   1073  1.37   mlelstv 
   1074  1.37   mlelstv static int
   1075  1.37   mlelstv verify_gpt(int fd)
   1076  1.37   mlelstv {
   1077  1.37   mlelstv 	struct	 gpt_hdr hdr;
   1078  1.37   mlelstv 	ssize_t	 ret;
   1079  1.37   mlelstv 	char	 buf[SCANSIZE];
   1080  1.37   mlelstv 	unsigned blksize;
   1081  1.37   mlelstv 	size_t	 off;
   1082  1.37   mlelstv 
   1083  1.37   mlelstv 	/*
   1084  1.37   mlelstv 	 * we read the first blocks to avoid sector size issues and
   1085  1.37   mlelstv 	 * verify the GPT header.
   1086  1.37   mlelstv 	 */
   1087  1.37   mlelstv 
   1088  1.37   mlelstv 	ret = prog_pread(fd, buf, SCANSIZE, 0);
   1089  1.37   mlelstv 	if (ret < 0) {
   1090  1.37   mlelstv 		warn("can't read gpt area");
   1091  1.37   mlelstv 		return -1;
   1092  1.37   mlelstv 	}
   1093  1.37   mlelstv 
   1094  1.49       kre 	ret = 1;
   1095  1.50       kre 	for (blksize = DEV_BSIZE;
   1096  1.50       kre              (off = (blksize * GPT_HDR_BLKNO)) <= SCANSIZE - sizeof(hdr);
   1097  1.37   mlelstv              blksize <<= 1) {
   1098  1.37   mlelstv 
   1099  1.37   mlelstv 		memcpy(&hdr, &buf[off], sizeof(hdr));
   1100  1.50       kre 		if (memcmp(hdr.hdr_sig, GPT_HDR_SIG, sizeof(hdr.hdr_sig)) == 0
   1101  1.50       kre 		    && le32toh(hdr.hdr_revision) == GPT_HDR_REVISION
   1102  1.50       kre 		    && le32toh(hdr.hdr_size) == GPT_HDR_SIZE) {
   1103  1.37   mlelstv 
   1104  1.37   mlelstv 			hdr.hdr_crc_self = 0;
   1105  1.37   mlelstv 			if (crc32(&hdr, sizeof(hdr))) {
   1106  1.37   mlelstv 				ret = 0;
   1107  1.37   mlelstv 				break;
   1108  1.37   mlelstv 			}
   1109  1.37   mlelstv 		}
   1110  1.37   mlelstv 	}
   1111  1.37   mlelstv 
   1112  1.37   mlelstv 	return ret;
   1113  1.37   mlelstv }
   1114  1.37   mlelstv 
   1115   1.7      fvdl static off_t sblock_try[] = SBLOCKSEARCH;
   1116   1.7      fvdl 
   1117   1.1     elric static int
   1118   1.5     elric verify_ffs(int fd)
   1119   1.5     elric {
   1120  1.19  christos 	size_t	i;
   1121   1.5     elric 
   1122   1.7      fvdl 	for (i = 0; sblock_try[i] != -1; i++) {
   1123  1.19  christos 		union {
   1124  1.19  christos 		    char	buf[SBLOCKSIZE];
   1125  1.19  christos 		    struct	fs fs;
   1126  1.19  christos 		} u;
   1127  1.17    cbiere 		ssize_t ret;
   1128  1.17    cbiere 
   1129  1.32     pooka 		ret = prog_pread(fd, &u, sizeof(u), sblock_try[i]);
   1130  1.17    cbiere 		if (ret < 0) {
   1131   1.7      fvdl 			warn("pread");
   1132  1.17    cbiere 			break;
   1133  1.19  christos 		} else if ((size_t)ret < sizeof(u)) {
   1134  1.17    cbiere 			warnx("pread: incomplete block");
   1135  1.17    cbiere 			break;
   1136   1.7      fvdl 		}
   1137  1.19  christos 		switch (u.fs.fs_magic) {
   1138   1.7      fvdl 		case FS_UFS1_MAGIC:
   1139   1.7      fvdl 		case FS_UFS2_MAGIC:
   1140   1.7      fvdl 		case FS_UFS1_MAGIC_SWAPPED:
   1141   1.7      fvdl 		case FS_UFS2_MAGIC_SWAPPED:
   1142   1.7      fvdl 			return 0;
   1143   1.7      fvdl 		default:
   1144   1.7      fvdl 			continue;
   1145   1.7      fvdl 		}
   1146   1.5     elric 	}
   1147  1.17    cbiere 
   1148  1.17    cbiere 	return 1;	/* failure */
   1149   1.9        cb }
   1150   1.9        cb 
   1151   1.9        cb static int
   1152   1.9        cb verify_reenter(struct params *p)
   1153   1.9        cb {
   1154   1.9        cb 	struct keygen *kg;
   1155  1.53       nia 	bits_t *orig_key, *key = NULL;
   1156   1.9        cb 	int ret;
   1157   1.9        cb 
   1158   1.9        cb 	ret = 0;
   1159   1.9        cb 	for (kg = p->keygen; kg && !ret; kg = kg->next) {
   1160  1.53       nia 		if (kg->kg_method != KEYGEN_ARGON2ID &&
   1161  1.53       nia 		    kg->kg_method != KEYGEN_PKCS5_PBKDF2_SHA1 &&
   1162  1.53       nia 		    kg->kg_method != KEYGEN_PKCS5_PBKDF2_OLD)
   1163   1.9        cb 			continue;
   1164   1.9        cb 
   1165   1.9        cb 		orig_key = kg->kg_key;
   1166   1.9        cb 		kg->kg_key = NULL;
   1167   1.9        cb 
   1168  1.53       nia 		switch (kg->kg_method) {
   1169  1.53       nia #ifdef HAVE_ARGON2
   1170  1.53       nia 		case KEYGEN_ARGON2ID:
   1171  1.53       nia 			key = getkey_argon2id("re-enter device", kg,
   1172  1.53       nia 				bits_len(orig_key));
   1173  1.53       nia 			break;
   1174  1.53       nia #endif
   1175  1.53       nia 		case KEYGEN_PKCS5_PBKDF2_SHA1:
   1176  1.53       nia 			key = getkey_pkcs5_pbkdf2("re-enter device", kg,
   1177  1.53       nia 				bits_len(orig_key), 0);
   1178  1.53       nia 			break;
   1179  1.53       nia 		case KEYGEN_PKCS5_PBKDF2_OLD:
   1180  1.53       nia 			key = getkey_pkcs5_pbkdf2("re-enter device", kg,
   1181  1.53       nia 				bits_len(orig_key), 1);
   1182  1.53       nia 			break;
   1183  1.53       nia 		default:
   1184  1.53       nia 			warnx("unsupported keygen method");
   1185  1.53       nia 			kg->kg_key = orig_key;
   1186  1.53       nia 			return -1;
   1187  1.53       nia 		}
   1188  1.50       kre 
   1189   1.9        cb 		ret = !bits_match(key, orig_key);
   1190   1.9        cb 
   1191   1.9        cb 		bits_free(key);
   1192   1.9        cb 		bits_free(kg->kg_key);
   1193   1.9        cb 		kg->kg_key = orig_key;
   1194   1.9        cb 	}
   1195   1.9        cb 
   1196   1.9        cb 	return ret;
   1197   1.5     elric }
   1198   1.5     elric 
   1199   1.5     elric static int
   1200   1.1     elric generate(struct params *p, int argc, char **argv, const char *outfile)
   1201   1.1     elric {
   1202   1.1     elric 	int	 ret;
   1203   1.1     elric 
   1204   1.1     elric 	if (argc < 1 || argc > 2)
   1205   1.1     elric 		usage();
   1206   1.1     elric 
   1207   1.5     elric 	p->algorithm = string_fromcharstar(argv[0]);
   1208  1.17    cbiere 	if (argc > 1) {
   1209  1.19  christos 		size_t keylen;
   1210  1.17    cbiere 
   1211  1.19  christos 		if (parse_size_t(argv[1], &keylen) == -1) {
   1212  1.17    cbiere 			warn("Failed to parse key length");
   1213  1.17    cbiere 			return -1;
   1214  1.17    cbiere 		}
   1215  1.17    cbiere 		p->keylen = keylen;
   1216  1.17    cbiere 	}
   1217   1.5     elric 
   1218   1.5     elric 	ret = params_filldefaults(p);
   1219   1.1     elric 	if (ret)
   1220   1.1     elric 		return ret;
   1221   1.5     elric 
   1222   1.5     elric 	if (!p->keygen) {
   1223  1.10       dan 		p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
   1224   1.5     elric 		if (!p->keygen)
   1225   1.5     elric 			return -1;
   1226   1.5     elric 	}
   1227   1.5     elric 
   1228   1.5     elric 	if (keygen_filldefaults(p->keygen, p->keylen)) {
   1229   1.5     elric 		warnx("Failed to generate defaults for keygen");
   1230   1.5     elric 		return -1;
   1231   1.5     elric 	}
   1232   1.5     elric 
   1233   1.5     elric 	if (!params_verify(p)) {
   1234   1.5     elric 		warnx("invalid parameters generated");
   1235   1.5     elric 		return -1;
   1236   1.5     elric 	}
   1237   1.5     elric 
   1238   1.5     elric 	return params_cput(p, outfile);
   1239   1.5     elric }
   1240   1.5     elric 
   1241   1.5     elric static int
   1242   1.5     elric generate_convert(struct params *p, int argc, char **argv, const char *outfile)
   1243   1.5     elric {
   1244   1.5     elric 	struct params	*oldp;
   1245   1.5     elric 	struct keygen	*kg;
   1246   1.5     elric 
   1247   1.5     elric 	if (argc != 1)
   1248   1.5     elric 		usage();
   1249   1.5     elric 
   1250   1.5     elric 	oldp = params_cget(*argv);
   1251   1.5     elric 	if (!oldp)
   1252   1.5     elric 		return -1;
   1253   1.5     elric 
   1254   1.5     elric 	/* for sanity, we ensure that none of the keygens are randomkey */
   1255   1.5     elric 	for (kg=p->keygen; kg; kg=kg->next)
   1256  1.30     elric 		if ((kg->kg_method == KEYGEN_RANDOMKEY) ||
   1257  1.30     elric 		    (kg->kg_method == KEYGEN_URANDOMKEY)) {
   1258  1.30     elric 			warnx("can't preserve randomly generated key");
   1259   1.5     elric 			goto bail;
   1260  1.30     elric 		}
   1261   1.5     elric 	for (kg=oldp->keygen; kg; kg=kg->next)
   1262  1.30     elric 		if ((kg->kg_method == KEYGEN_RANDOMKEY) ||
   1263  1.30     elric 		    (kg->kg_method == KEYGEN_URANDOMKEY)) {
   1264  1.30     elric 			warnx("can't preserve randomly generated key");
   1265   1.5     elric 			goto bail;
   1266  1.30     elric 		}
   1267   1.5     elric 
   1268   1.5     elric 	if (!params_verify(oldp)) {
   1269   1.5     elric 		warnx("invalid old parameters file \"%s\"", *argv);
   1270   1.5     elric 		return -1;
   1271   1.1     elric 	}
   1272   1.1     elric 
   1273   1.5     elric 	oldp->key = getkey("old file", oldp->keygen, oldp->keylen);
   1274   1.5     elric 
   1275   1.5     elric 	/* we copy across the non-keygen info, here. */
   1276   1.5     elric 
   1277   1.5     elric 	string_free(p->algorithm);
   1278   1.5     elric 	string_free(p->ivmeth);
   1279   1.5     elric 
   1280   1.5     elric 	p->algorithm = string_dup(oldp->algorithm);
   1281   1.5     elric 	p->ivmeth = string_dup(oldp->ivmeth);
   1282   1.5     elric 	p->keylen = oldp->keylen;
   1283   1.5     elric 	p->bsize = oldp->bsize;
   1284   1.5     elric 	if (p->verify_method == VERIFY_UNKNOWN)
   1285   1.5     elric 		p->verify_method = oldp->verify_method;
   1286   1.5     elric 
   1287   1.5     elric 	params_free(oldp);
   1288   1.1     elric 
   1289   1.5     elric 	if (!p->keygen) {
   1290  1.10       dan 		p->keygen = keygen_generate(KEYGEN_PKCS5_PBKDF2_SHA1);
   1291   1.5     elric 		if (!p->keygen)
   1292   1.1     elric 			return -1;
   1293   1.5     elric 	}
   1294  1.19  christos 	(void)params_filldefaults(p);
   1295  1.19  christos 	(void)keygen_filldefaults(p->keygen, p->keylen);
   1296   1.5     elric 	p->key = getkey("new file", p->keygen, p->keylen);
   1297   1.5     elric 
   1298   1.5     elric 	kg = keygen_generate(KEYGEN_STOREDKEY);
   1299   1.5     elric 	kg->kg_key = bits_xor(p->key, oldp->key);
   1300   1.5     elric 	keygen_addlist(&p->keygen, kg);
   1301   1.5     elric 
   1302   1.5     elric 	if (!params_verify(p)) {
   1303   1.5     elric 		warnx("can't generate new parameters file");
   1304   1.5     elric 		return -1;
   1305   1.1     elric 	}
   1306   1.1     elric 
   1307   1.5     elric 	return params_cput(p, outfile);
   1308  1.50       kre  bail:;
   1309   1.5     elric 	params_free(oldp);
   1310   1.5     elric 	return -1;
   1311   1.1     elric }
   1312   1.1     elric 
   1313   1.1     elric static int
   1314  1.19  christos /*ARGSUSED*/
   1315   1.1     elric do_all(const char *cfile, int argc, char **argv,
   1316   1.3     elric        int (*conf)(int, char **, struct params *, int))
   1317   1.1     elric {
   1318   1.1     elric 	FILE		 *f;
   1319   1.1     elric 	size_t		  len;
   1320   1.1     elric 	size_t		  lineno;
   1321   1.1     elric 	int		  my_argc;
   1322   1.1     elric 	int		  ret;
   1323   1.1     elric 	const char	 *fn;
   1324   1.1     elric 	char		 *line;
   1325   1.1     elric 	char		**my_argv;
   1326   1.1     elric 
   1327   1.1     elric 	if (argc > 0)
   1328   1.1     elric 		usage();
   1329   1.1     elric 
   1330   1.1     elric 	if (!cfile[0])
   1331   1.1     elric 		fn = CGDCONFIG_CFILE;
   1332   1.1     elric 	else
   1333   1.1     elric 		fn = cfile;
   1334   1.1     elric 
   1335   1.1     elric 	f = fopen(fn, "r");
   1336  1.50       kre 	if (f == NULL) {
   1337   1.5     elric 		warn("could not open config file \"%s\"", fn);
   1338   1.1     elric 		return -1;
   1339   1.1     elric 	}
   1340   1.1     elric 
   1341   1.1     elric 	ret = 0;
   1342   1.1     elric 	lineno = 0;
   1343   1.1     elric 	for (;;) {
   1344   1.1     elric 		line = fparseln(f, &len, &lineno, "\\\\#", FPARSELN_UNESCALL);
   1345   1.1     elric 		if (!line)
   1346   1.1     elric 			break;
   1347   1.1     elric 		if (!*line)
   1348   1.1     elric 			continue;
   1349   1.1     elric 
   1350   1.1     elric 		my_argv = words(line, &my_argc);
   1351   1.3     elric 		ret = conf(my_argc, my_argv, NULL, CONFIG_FLAGS_FROMALL);
   1352   1.1     elric 		if (ret) {
   1353   1.5     elric 			warnx("action failed on \"%s\" line %lu", fn,
   1354   1.1     elric 			    (u_long)lineno);
   1355   1.1     elric 			break;
   1356   1.1     elric 		}
   1357   1.1     elric 		words_free(my_argv, my_argc);
   1358   1.1     elric 	}
   1359   1.1     elric 	return ret;
   1360   1.1     elric }
   1361  1.13     elric 
   1362  1.34  christos static const char *
   1363  1.34  christos iv_method(int mode)
   1364  1.34  christos {
   1365  1.34  christos 
   1366  1.34  christos 	switch (mode) {
   1367  1.34  christos 	case CGD_CIPHER_CBC_ENCBLKNO8:
   1368  1.34  christos 		return "encblkno8";
   1369  1.34  christos 	case CGD_CIPHER_CBC_ENCBLKNO1:
   1370  1.34  christos 		return "encblkno1";
   1371  1.34  christos 	default:
   1372  1.34  christos 		return "unknown";
   1373  1.34  christos 	}
   1374  1.34  christos }
   1375  1.34  christos 
   1376  1.35  christos 
   1377  1.35  christos static void
   1378  1.35  christos show(const char *dev) {
   1379  1.35  christos 	char path[64];
   1380  1.35  christos 	struct cgd_user cgu;
   1381  1.35  christos 	int fd;
   1382  1.35  christos 
   1383  1.35  christos 	fd = opendisk(dev, O_RDONLY, path, sizeof(path), 0);
   1384  1.35  christos 	if (fd == -1) {
   1385  1.35  christos 		warn("open: %s", dev);
   1386  1.35  christos 		return;
   1387  1.35  christos 	}
   1388  1.35  christos 
   1389  1.35  christos 	cgu.cgu_unit = -1;
   1390  1.35  christos 	if (prog_ioctl(fd, CGDIOCGET, &cgu) == -1) {
   1391  1.35  christos 		close(fd);
   1392  1.35  christos 		err(1, "CGDIOCGET");
   1393  1.35  christos 	}
   1394  1.35  christos 
   1395  1.35  christos 	printf("%s: ", dev);
   1396  1.35  christos 
   1397  1.35  christos 	if (cgu.cgu_dev == 0) {
   1398  1.35  christos 		printf("not in use");
   1399  1.35  christos 		goto out;
   1400  1.35  christos 	}
   1401  1.35  christos 
   1402  1.35  christos 	dev = devname(cgu.cgu_dev, S_IFBLK);
   1403  1.35  christos 	if (dev != NULL)
   1404  1.35  christos 		printf("%s ", dev);
   1405  1.35  christos 	else
   1406  1.35  christos 		printf("dev %llu,%llu ", (unsigned long long)major(cgu.cgu_dev),
   1407  1.35  christos 		    (unsigned long long)minor(cgu.cgu_dev));
   1408  1.35  christos 
   1409  1.35  christos 	if (verbose)
   1410  1.35  christos 		printf("%s ", cgu.cgu_alg);
   1411  1.35  christos 	if (verbose > 1) {
   1412  1.35  christos 		printf("keylen %d ", cgu.cgu_keylen);
   1413  1.35  christos 		printf("blksize %zd ", cgu.cgu_blocksize);
   1414  1.35  christos 		printf("%s ", iv_method(cgu.cgu_mode));
   1415  1.35  christos 	}
   1416  1.35  christos 
   1417  1.50       kre  out:;
   1418  1.35  christos 	putchar('\n');
   1419  1.35  christos 	close(fd);
   1420  1.35  christos }
   1421  1.35  christos 
   1422  1.34  christos static int
   1423  1.34  christos do_list(int argc, char **argv)
   1424  1.34  christos {
   1425  1.34  christos 
   1426  1.34  christos 	if (argc != 0 && argc != 1)
   1427  1.34  christos 		usage();
   1428  1.34  christos 
   1429  1.35  christos 	if (argc) {
   1430  1.35  christos 		show(argv[0]);
   1431  1.35  christos 		return 0;
   1432  1.35  christos 	}
   1433  1.34  christos 
   1434  1.35  christos 	DIR *dirp;
   1435  1.35  christos 	struct dirent *dp;
   1436  1.35  christos 	__BITMAP_TYPE(, uint32_t, 65536) bm;
   1437  1.35  christos 
   1438  1.35  christos 	__BITMAP_ZERO(&bm);
   1439  1.35  christos 
   1440  1.35  christos 	if ((dirp = opendir(_PATH_DEV)) == NULL)
   1441  1.35  christos 		err(1, "opendir: %s", _PATH_DEV);
   1442  1.35  christos 
   1443  1.35  christos 	while ((dp = readdir(dirp)) != NULL) {
   1444  1.35  christos 		char *ep;
   1445  1.35  christos 		if (strncmp(dp->d_name, "rcgd", 4) != 0)
   1446  1.35  christos 			continue;
   1447  1.35  christos 		errno = 0;
   1448  1.35  christos 		int n = (int)strtol(dp->d_name + 4, &ep, 0);
   1449  1.35  christos 		if (ep == dp->d_name + 4 || errno != 0) {
   1450  1.35  christos 			warnx("bad name %s", dp->d_name);
   1451  1.35  christos 			continue;
   1452  1.34  christos 		}
   1453  1.35  christos 		*ep = '\0';
   1454  1.35  christos 		if (__BITMAP_ISSET(n, &bm))
   1455  1.35  christos 			continue;
   1456  1.35  christos 		__BITMAP_SET(n, &bm);
   1457  1.35  christos 		show(dp->d_name + 1);
   1458  1.34  christos 	}
   1459  1.34  christos 
   1460  1.35  christos 	closedir(dirp);
   1461  1.34  christos 	return 0;
   1462  1.34  christos }
   1463  1.34  christos 
   1464  1.54  riastrad static int
   1465  1.55  riastrad printkey(const char *dev, const char *paramsfile, const char *fmt, ...)
   1466  1.54  riastrad {
   1467  1.55  riastrad 	va_list va;
   1468  1.54  riastrad 	struct params *p;
   1469  1.54  riastrad 	const uint8_t *raw;
   1470  1.54  riastrad 	size_t nbits, nbytes;
   1471  1.54  riastrad 	size_t nb64;
   1472  1.54  riastrad 	char *b64;
   1473  1.54  riastrad 	int ret;
   1474  1.54  riastrad 
   1475  1.55  riastrad 	p = params_cget(paramsfile);
   1476  1.54  riastrad 	if (p == NULL)
   1477  1.54  riastrad 		return -1;
   1478  1.54  riastrad 	if (!params_verify(p)) {
   1479  1.55  riastrad 		warnx("invalid parameters file \"%s\"", paramsfile);
   1480  1.54  riastrad 		return -1;
   1481  1.54  riastrad 	}
   1482  1.55  riastrad 	p->key = getkey(dev, p->keygen, p->keylen);
   1483  1.54  riastrad 	raw = bits_getbuf(p->key);
   1484  1.54  riastrad 	nbits = bits_len(p->key);
   1485  1.54  riastrad 	assert(nbits <= INT_MAX - 7);
   1486  1.54  riastrad 	nbytes = BITS2BYTES(nbits);
   1487  1.54  riastrad 	assert(nbytes <= 3*(INT_MAX/4) - 2);
   1488  1.54  riastrad 
   1489  1.54  riastrad 	nb64 = 4*((nbytes + 2)/3);
   1490  1.54  riastrad 	b64 = emalloc(nb64 + 2);
   1491  1.54  riastrad 	ret = __b64_ntop(raw, nbytes, b64, nb64 + 1);
   1492  1.54  riastrad 	assert(ret == (int)nb64);
   1493  1.54  riastrad 	b64[nb64] = '\n';
   1494  1.54  riastrad 	b64[nb64 + 1] = '\0';
   1495  1.54  riastrad 
   1496  1.55  riastrad 	va_start(va, fmt);
   1497  1.55  riastrad 	vprintf(fmt, va);
   1498  1.55  riastrad 	va_end(va);
   1499  1.54  riastrad 	if (fwrite(b64, nb64 + 1, 1, stdout) != 1)
   1500  1.54  riastrad 		err(1, "fwrite");
   1501  1.54  riastrad 	fflush(stdout);
   1502  1.54  riastrad 	return ferror(stdout);
   1503  1.54  riastrad }
   1504  1.54  riastrad 
   1505  1.55  riastrad static int
   1506  1.55  riastrad printkey1(int argc, char **argv, struct params *inparams, int flags)
   1507  1.55  riastrad {
   1508  1.55  riastrad 	char devicename[PATH_MAX], paramsfilebuf[PATH_MAX];
   1509  1.55  riastrad 	const char *dev, *paramsfile;
   1510  1.55  riastrad 
   1511  1.55  riastrad 	assert(flags == CONFIG_FLAGS_FROMALL);
   1512  1.55  riastrad 
   1513  1.55  riastrad 	if (argc < 2 || argc > 3)
   1514  1.55  riastrad 		return -1;
   1515  1.55  riastrad 
   1516  1.55  riastrad 	dev = getfsspecname(devicename, sizeof(devicename), argv[1]);
   1517  1.55  riastrad 	if (dev == NULL) {
   1518  1.55  riastrad 		warnx("getfsspecname failed: %s", devicename);
   1519  1.55  riastrad 		return -1;
   1520  1.55  riastrad 	}
   1521  1.55  riastrad 
   1522  1.55  riastrad 	if (argc == 2) {
   1523  1.55  riastrad 		strlcpy(paramsfilebuf, dev, sizeof(paramsfilebuf));
   1524  1.55  riastrad 		paramsfile = basename(paramsfilebuf);
   1525  1.55  riastrad 	} else {
   1526  1.55  riastrad 		paramsfile = argv[2];
   1527  1.55  riastrad 	}
   1528  1.55  riastrad 
   1529  1.55  riastrad 	return printkey(dev, paramsfile, "%s: ", dev);
   1530  1.55  riastrad }
   1531  1.55  riastrad 
   1532  1.55  riastrad static int
   1533  1.55  riastrad do_printkey(int argc, char **argv)
   1534  1.55  riastrad {
   1535  1.55  riastrad 
   1536  1.55  riastrad 	if (argc != 1)
   1537  1.55  riastrad 		usage();
   1538  1.55  riastrad 	return printkey("key", argv[0], "");
   1539  1.55  riastrad }
   1540  1.55  riastrad 
   1541  1.13     elric static void
   1542  1.13     elric eliminate_cores(void)
   1543  1.13     elric {
   1544  1.13     elric 	struct rlimit	rlp;
   1545  1.13     elric 
   1546  1.13     elric 	rlp.rlim_cur = 0;
   1547  1.13     elric 	rlp.rlim_max = 0;
   1548  1.17    cbiere 	if (setrlimit(RLIMIT_CORE, &rlp) == -1)
   1549  1.13     elric 		err(EXIT_FAILURE, "Can't disable cores");
   1550  1.13     elric }
   1551