36642229be7271e4acac07867848b802ea73f639
[openafs.git] / src / aklog / asetkey.c
1 /*
2  * $Id$
3  *
4  * asetkey - Manipulates an AFS KeyFile
5  *
6  * Updated for Kerberos 5
7  */
8
9 #include <stdio.h>
10 #include <sys/types.h>
11 #include <netinet/in.h>
12 #include <netdb.h>
13 #ifdef HAVE_MEMORY_H
14 #include <memory.h>
15 #endif /* HAVE_MEMORY_H */
16 #ifdef HAVE_STRING_H
17 #include <string.h>
18 #else /* HAVE_STRING_H */
19 #ifdef HAVE_STRINGS_H
20 #include <strings.h>
21 #endif /* HAVE_STRINGS_H */
22 #endif /* HAVE_STRING_H */
23
24 #include <krb5.h>
25 #include <com_err.h>
26
27 #include <afs/stds.h>
28 #include <afs/cellconfig.h>
29 #include <afs/keys.h>
30 #include <afs/dirpath.h>
31
32 int
33 main(int argc, char *argv[])
34 {
35     struct afsconf_dir *tdir;
36     register long code;
37     const char *confdir;
38
39     if (argc == 1) {
40         fprintf(stderr, "%s: usage is '%s <opcode> options, e.g.\n",
41                 argv[0], argv[0]);
42         fprintf(stderr, "\t%s add <kvno> <keyfile> <princ>\n", argv[0]);
43         fprintf(stderr, "\t%s delete <kvno>\n", argv[0]);
44         fprintf(stderr, "\t%s list\n", argv[0]);
45         exit(1);
46     }
47
48     confdir = AFSDIR_SERVER_ETC_DIRPATH;
49
50     tdir = afsconf_Open(confdir);
51     if (!tdir) {
52         fprintf(stderr, "%s: can't initialize conf dir '%s'\n", argv[0],
53                 confdir);
54         exit(1);
55     }
56     if (strcmp(argv[1], "add")==0) {
57         krb5_context context;
58         krb5_principal principal;
59         krb5_keyblock *key;
60         krb5_error_code retval;
61         int kvno;
62
63         if (argc != 5) {
64             fprintf(stderr, "%s add: usage is '%s add <kvno> <keyfile> "
65                     "<princ>\n", argv[0], argv[0]);
66             exit(1);
67         }
68
69         krb5_init_context(&context);
70
71         kvno = atoi(argv[2]);
72         retval = krb5_parse_name(context, argv[4], &principal);
73         if (retval != 0) {
74                 com_err(argv[0], retval, "while parsing AFS principal");
75                 exit(1);
76         }
77         retval = krb5_kt_read_service_key(context, argv[3], principal, kvno,
78                                           ENCTYPE_DES_CBC_CRC, &key);
79         if (retval != 0) {
80                 com_err(argv[0], retval, "while extracting AFS service key");
81                 exit(1);
82         }
83
84         if (key->length != 8) {
85                 fprintf(stderr, "Key length should be 8, but is really %d!\n",
86                        key->length);
87                 exit(1);
88         }
89
90         code = afsconf_AddKey(tdir, kvno, (char *) key->contents, 1);
91         if (code) {
92             fprintf(stderr, "%s: failed to set key, code %d.\n", argv[0], code);
93             exit(1);
94         }
95         krb5_free_principal(context, principal);
96         krb5_free_keyblock(context, key);
97     }
98     else if (strcmp(argv[1], "delete")==0) {
99         long kvno;
100         if (argc != 3) {
101             fprintf(stderr, "%s delete: usage is '%s delete <kvno>\n",
102                     argv[0], argv[0]);
103             exit(1);
104         }
105         kvno = atoi(argv[2]);
106         code = afsconf_DeleteKey(tdir, kvno);
107         if (code) {
108             fprintf(stderr, "%s: failed to delete key %d, (code %d)\n",
109                     argv[0], kvno, code);
110             exit(1);
111         }
112     }
113     else if (strcmp(argv[1], "list") == 0) {
114         struct afsconf_keys tkeys;
115         register int i, j;
116         
117         code = afsconf_GetKeys(tdir, &tkeys);
118         if (code) {
119             fprintf(stderr, "%s: failed to get keys, code %d\n", argv[0], code);
120             exit(1);
121         }
122         for(i=0;i<tkeys.nkeys;i++) {
123             if (tkeys.key[i].kvno != -1) {
124                 printf("kvno %4d: key is: ", tkeys.key[i].kvno);
125                 for (j = 0; j < 8; j++)
126                         printf("%02x", (unsigned char) tkeys.key[i].key[j]);
127                 printf("\n");
128             }
129         }
130         printf("All done.\n");
131     }
132     else {
133         fprintf(stderr, "%s: unknown operation '%s', type '%s' for "
134                 "assistance\n", argv[0], argv[1], argv[0]);
135         exit(1);
136     }
137     exit(0);
138 }