/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2005 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
#include <sys/types.h>
#include <sys/wanboot_impl.h>
#include <libinetutil.h>
#include <wanbootutil.h>
#include <libintl.h>
#include <locale.h>
#include <unistd.h>
#include <stdlib.h>
#include <strings.h>
#include <stdio.h>
#include <fcntl.h>
#include <ctype.h>
#include <assert.h>
#include <sys/openpromio.h>
#define TYPE 0
static char *progopts[] = {
"type",
NULL
};
/*
* The key's handle is the name by which a user knows the key (i.e. the
* name specified on the command line. The keyname is the name this
* utility uses to store the keys and the name OBP and wanboot use to
* retrieve them.
*/
static struct keylist {
const char *handle;
const char *keyname;
const int keysize; /* size of hex string representation */
} keylist[] = {
WBKU_KW_3DES, WANBOOT_DES3_KEY_NAME,
(DES3_KEY_SIZE * 2),
WBKU_KW_AES_128, WANBOOT_AES_128_KEY_NAME,
(AES_128_KEY_SIZE * 2),
WBKU_KW_HMAC_SHA1, WANBOOT_HMAC_SHA1_KEY_NAME,
(WANBOOT_HMAC_KEY_SIZE * 2)
};
static const struct keylist *knownkeytype(char *);
static char *getkey(const struct keylist *);
static void deletekey(const struct keylist *);
static void installkey(const struct keylist *);
static void usage(const char *) __NORETURN;
static boolean_t delete = B_FALSE;
int
main(int ac, char **av)
{
int i;
const struct keylist *k;
char *typestring = NULL;
char *options;
char *value;
/*
* Do the necessary magic for localization support.
*/
(void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN)
#define TEXT_DOMAIN "SYS_TEST"
#endif
(void) textdomain(TEXT_DOMAIN);
/*
* Initialize program name for use by wbku_printerr().
*/
wbku_errinit(av[0]);
while ((i = getopt(ac, av, "do:")) != -1)
switch (i) {
case 'd':
delete = B_TRUE;
break;
case 'o':
options = optarg;
while (*options != '\0') {
switch (getsubopt(&options, progopts,
&value)) {
case TYPE:
typestring = value;
break;
default:
/* unknown token */
usage(*av);
/* NOTREACHED */
}
}
break;
case '?':
usage(*av);
/* NOTREACHED */
}
if ((optind >= ac) && (typestring != NULL) &&
((k = knownkeytype(typestring)) != NULL)) {
if (delete == B_TRUE)
deletekey(k);
else
installkey(k);
return (0);
} else {
usage(*av);
/* NOTREACHED */
}
}
static const struct keylist *
knownkeytype(char *type)
{
int i;
for (i = 0; i < sizeof (keylist)/sizeof (keylist[0]); i++) {
if (strcmp(keylist[i].handle, type) == 0)
return (&keylist[i]);
}
return (NULL);
}
static void
deletekey(const struct keylist *k)
{
int fd;
struct wankeyio wkio;
struct openpromio *oio;
(void) strlcpy(wkio.wk_keyname, k->keyname, WANBOOT_MAXKEYNAMELEN);
wkio.wk_keysize = 0; /* zero key size indicates a deletion */
oio = malloc(sizeof (struct openpromio) + sizeof (struct wankeyio));
if (oio == NULL) {
wbku_printerr("openpromio malloc (%d) failed\n",
sizeof (struct openpromio) +
sizeof (struct wankeyio));
exit(1);
}
oio->oprom_size = sizeof (struct wankeyio);
bcopy(&wkio, oio->oprom_array, sizeof (struct wankeyio));
fd = open("/dev/openprom", O_RDWR);
if (fd == -1) {
wbku_printerr("open: /dev/openprom");
exit(1);
}
if (ioctl(fd, WANBOOT_SETKEY, oio) == -1) {
wbku_printerr("setkey: ioctl");
exit(1);
}
(void) close(fd);
}
static void
installkey(const struct keylist *k)
{
char *keyptr;
int fd;
struct wankeyio wkio;
struct openpromio *oio;
uint_t rawkeysize;
int err;
(void) strlcpy(wkio.wk_keyname, k->keyname, WANBOOT_MAXKEYNAMELEN);
assert((k->keysize % 2) == 0);
wkio.wk_keysize = k->keysize / 2;
if ((keyptr = getkey(k)) != NULL) {
rawkeysize = sizeof (wkio.wk_u);
if ((err = hexascii_to_octet(keyptr, strlen(keyptr),
wkio.wk_u.key, &rawkeysize)) != 0) {
wbku_printerr(
"internal error: hexascii_to_octet returned %d\n",
err);
exit(1);
} else if (rawkeysize != wkio.wk_keysize) {
wbku_printerr("internal error: key size mismatch\n");
exit(1);
}
oio = malloc(sizeof (struct openpromio) +
sizeof (struct wankeyio));
if (oio == NULL) {
wbku_printerr("openpromio malloc (%d) failed\n",
sizeof (struct openpromio) +
sizeof (struct wankeyio));
exit(1);
}
oio->oprom_size = sizeof (struct wankeyio);
bcopy(&wkio, oio->oprom_array, sizeof (struct wankeyio));
fd = open("/dev/openprom", O_RDWR);
if (fd == -1) {
wbku_printerr("open: /dev/openprom");
exit(1);
}
if (ioctl(fd, WANBOOT_SETKEY, oio) == -1) {
wbku_printerr("setkey: ioctl");
exit(1);
}
(void) close(fd);
} else {
wbku_printerr("getpassphrase"); /* getpassphrase() failed */
exit(1);
}
}
static char *
getkey(const struct keylist *k)
{
char prompt[BUFSIZ];
char *p;
char *q;
int len;
(void) snprintf(prompt, sizeof (prompt),
gettext("Enter %s key: "), k->handle);
p = getpassphrase(prompt);
if (p) {
/* skip over initial "0[xX]" */
if ((p[0] == '0') && (p[1] == 'x' || p[1] == 'X'))
p += 2;
len = strlen(p);
if (len != k->keysize) {
wbku_printerr(
"key length mismatch (expected %d, got %d)\n",
k->keysize, len);
exit(1);
}
for (q = p; q < p + len; q++)
if (!isxdigit(*q)) {
wbku_printerr(
"non-hexadecimal characters in key\n");
exit(1);
}
}
return (p);
}
static void
usage(const char *progname)
{
int i;
(void) fprintf(stderr, gettext(
"usage: %s [ -d ] -o type=keytype\nwhere keytype is one of "),
progname);
for (i = 0; i < sizeof (keylist)/sizeof (keylist[0]); i++)
(void) fprintf(stderr, "%s ", keylist[i].handle);
(void) fputc('\n', stderr);
exit(1);
}