Initial import from FreeBSD RELENG_4:
[dragonfly.git] / crypto / kerberosIV / kadmin / ksrvutil_get.c
1 /*
2  * Copyright (c) 1995, 1996, 1997, 1998, 1999 Kungliga Tekniska Högskolan
3  * (Royal Institute of Technology, Stockholm, Sweden).
4  * All rights reserved.
5  * 
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 
17  * 3. Neither the name of the Institute nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  * 
21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 /* $FreeBSD: src/crypto/kerberosIV/kadmin/ksrvutil_get.c,v 1.1.1.3.2.1 2003/02/14 22:37:37 nectar Exp $ */
34
35 #include "kadm_locl.h"
36 #include "ksrvutil.h"
37
38 RCSID("$Id: ksrvutil_get.c,v 1.43 1999/12/02 16:58:36 joda Exp $");
39
40 #define BAD_PW 1
41 #define GOOD_PW 0
42 #define FUDGE_VALUE 15          /* for ticket expiration time */
43 #define PE_NO 0
44 #define PE_YES 1
45 #define PE_UNSURE 2
46
47 static char tktstring[MaxPathLen];
48
49 static int
50 princ_exists(char *name, char *instance, char *realm)
51 {
52     int status;
53
54     status = krb_get_pw_in_tkt(name, instance, realm,
55                                KRB_TICKET_GRANTING_TICKET,
56                                realm, 1, "");
57
58     if ((status == KSUCCESS) || (status == INTK_BADPW))
59         return(PE_YES);
60     else if (status == KDC_PR_UNKNOWN)
61         return(PE_NO);
62     else
63         return(PE_UNSURE);
64 }
65
66 static int
67 get_admin_password(char *myname, char *myinst, char *myrealm)
68 {
69   int status;
70   char admin_passwd[MAX_KPW_LEN];       /* Admin's password */
71   int ticket_life = 1;  /* minimum ticket lifetime */
72   char buf[1024];
73   CREDENTIALS c;
74
75   if (princ_exists(myname, myinst, myrealm) != PE_NO) {
76     snprintf(buf, sizeof(buf), "Password for %s: ",
77             krb_unparse_name_long (myname, myinst, myrealm));
78     if (des_read_pw_string(admin_passwd, sizeof(admin_passwd)-1,
79                             buf, 0)) {
80       fprintf(stderr, "Error reading admin password.\n");
81       goto bad;
82     }
83     status = krb_get_pw_in_tkt(myname, myinst, myrealm, PWSERV_NAME, 
84                                KADM_SINST, ticket_life, admin_passwd);
85     memset(admin_passwd, 0, sizeof(admin_passwd));
86   } else
87     status = KDC_PR_UNKNOWN;
88   
89   switch(status) {
90   case GT_PW_OK:
91     return(GOOD_PW);
92   case KDC_PR_UNKNOWN:
93     printf("Principal %s does not exist.\n",
94            krb_unparse_name_long(myname, myinst, myrealm));
95     goto bad;
96   case GT_PW_BADPW:
97     printf("Incorrect admin password.\n");
98     goto bad;
99   default:
100     com_err("kadmin", status+krb_err_base,
101             "while getting password tickets");
102     goto bad;
103   }
104   
105 bad:
106   memset(admin_passwd, 0, sizeof(admin_passwd));
107   dest_tkt();
108   return(BAD_PW);
109 }
110
111 static void
112 srvtab_put_key (int fd, char *filename, char *name, char *inst, char *realm,
113                 int8_t kvno, des_cblock key)
114 {
115   char sname[ANAME_SZ];         /* name of service */
116   char sinst[INST_SZ];          /* instance of service */
117   char srealm[REALM_SZ];        /* realm of service */
118   int8_t skvno;
119   des_cblock skey;
120
121   lseek(fd, 0, SEEK_SET);
122
123   while(getst(fd, sname,  SNAME_SZ) > 0 &&
124         getst(fd, sinst,  INST_SZ) > 0  &&
125         getst(fd, srealm, REALM_SZ) > 0 &&
126         read(fd, &skvno,  sizeof(skvno)) > 0 &&
127         read(fd, skey,    sizeof(skey)) > 0) {
128     if(strcmp(name,  sname)  == 0 &&
129        strcmp(inst,  sinst)  == 0 &&
130        strcmp(realm, srealm) == 0) {
131       lseek(fd, lseek(fd,0,SEEK_CUR)-(sizeof(skvno) + sizeof(skey)), SEEK_SET);
132       safe_write(filename, fd, &kvno, sizeof(kvno));
133       safe_write(filename, fd, key,   sizeof(des_cblock));
134       return;
135     }
136   }
137   safe_write(filename, fd, name,  strlen(name) + 1);
138   safe_write(filename, fd, inst,  strlen(inst) + 1);
139   safe_write(filename, fd, realm, strlen(realm) + 1);
140   safe_write(filename, fd, &kvno, sizeof(kvno));
141   safe_write(filename, fd, key,   sizeof(des_cblock));
142 }
143
144 /* 
145  * node list of services 
146  */
147
148 struct srv_ent{
149   char name[SNAME_SZ];
150   char inst[INST_SZ];
151   char realm[REALM_SZ];
152   struct srv_ent *next;
153 };
154
155 static int
156 key_to_key(const char *user,
157            char *instance,
158            const char *realm,
159            const void *arg,
160            des_cblock *key)
161 {
162   memcpy(key, arg, sizeof(des_cblock));
163   return 0;
164 }
165
166 static void
167 get_srvtab_ent(int unique_filename, int fd, char *filename, 
168                char *name, char *inst, char *realm)
169 {
170     char chname[128];
171     des_cblock newkey;
172     char old_tktfile[MaxPathLen], new_tktfile[MaxPathLen];
173     char garbage_name[ANAME_SZ];
174     char garbage_inst[ANAME_SZ];
175     CREDENTIALS c;
176     u_int8_t kvno;
177     Kadm_vals values;
178     int ret;
179
180     strlcpy(chname, krb_get_phost(inst), sizeof(chname));
181     if(strcmp(inst, chname))
182         fprintf(stderr, 
183                 "Warning: Are you sure `%s' should not be `%s'?\n",
184                 inst, chname);
185     
186     memset(&values, 0, sizeof(values));
187     strlcpy(values.name, name, sizeof(values.name));
188     strlcpy(values.instance, inst, sizeof(values.instance));
189     des_random_key(newkey);
190     values.key_low = (newkey[0] << 24) | (newkey[1] << 16)
191         | (newkey[2] << 8) | (newkey[3] << 0);
192     values.key_high = (newkey[4] << 24) | (newkey[5] << 16)
193         | (newkey[6] << 8) | (newkey[7] << 0);
194
195     SET_FIELD(KADM_NAME,values.fields);
196     SET_FIELD(KADM_INST,values.fields);
197     SET_FIELD(KADM_DESKEY,values.fields);
198
199     ret = kadm_mod(&values, &values);
200     if(ret == KADM_NOENTRY)
201         ret = kadm_add(&values);
202     if (ret != KSUCCESS) {
203         warnx ("Couldn't get srvtab entry for %s.%s: %s",
204                name, inst, error_message(ret));
205         return;
206     }
207   
208     values.key_low = values.key_high = 0;
209
210     /* get the key version number */
211     { 
212         int old = krb_use_admin_server(1);
213
214         strlcpy(old_tktfile, tkt_string(), sizeof(old_tktfile));
215         snprintf(new_tktfile, sizeof(new_tktfile), "%s_ksrvutil-get.%u",
216                  TKT_ROOT, (unsigned)getpid());
217         krb_set_tkt_string(new_tktfile);
218       
219         ret = krb_get_in_tkt(name, inst, realm, name, inst,
220                              1, key_to_key, NULL, &newkey);
221         krb_use_admin_server(old);
222         if (ret) {
223             warnx ("getting tickets for %s: %s", 
224                    krb_unparse_name_long(name, inst, realm),
225                    krb_get_err_text(ret));
226             return;
227         }
228     }
229       
230     if (ret == KSUCCESS &&
231         (ret = tf_init(tkt_string(), R_TKT_FIL)) == KSUCCESS &&
232         (ret = tf_get_pname(garbage_name)) == KSUCCESS &&
233         (ret = tf_get_pinst(garbage_inst)) == KSUCCESS &&
234         (ret = tf_get_cred(&c)) == KSUCCESS)
235         kvno = c.kvno;
236     else {
237         warnx ("Could not find the cred in the ticket file: %s",
238                krb_get_err_text(ret));
239         return;
240     }
241
242     tf_close();
243     krb_set_tkt_string(old_tktfile);
244     unlink(new_tktfile);
245     
246     if(ret != KSUCCESS) {
247         memset(&newkey, 0, sizeof(newkey));
248         warnx ("Could not get a ticket for %s: %s\n",
249                krb_unparse_name_long(name, inst, realm),
250                krb_get_err_text(ret));
251         return;
252     }
253
254     /* Write the new key & c:o to the srvtab file */
255
256     if(unique_filename){
257         char *fn;
258         asprintf(&fn, "%s-%s", filename, 
259                  krb_unparse_name_long(name, inst, realm));
260         if(fn == NULL){
261             warnx("Out of memory");
262             leave(NULL, 1);
263         }
264         fd = open(fn, O_RDWR | O_CREAT | O_TRUNC, 0600); /* XXX flags, mode? */
265         if(fd < 0){
266             warn("%s", fn);
267             leave(NULL, 1);
268         }
269         srvtab_put_key (fd, fn, name, inst, realm, kvno, newkey);
270         close(fd);
271         fprintf (stderr, "Created %s\n", fn);
272         free(fn);
273     }else{
274         srvtab_put_key (fd, filename, name, inst, realm, kvno, newkey);
275         fprintf (stderr, "Added %s\n", 
276                  krb_unparse_name_long (name, inst, realm));
277     }
278     memset(&newkey, 0, sizeof(newkey));
279 }
280
281 static void
282 ksrvutil_kadm(int unique_filename, int fd, char *filename, struct srv_ent *p)
283 {
284   int ret;
285   CREDENTIALS c;
286   
287   ret = kadm_init_link(PWSERV_NAME, KADM_SINST, u_realm);
288   if (ret != KADM_SUCCESS) {
289     warnx("Couldn't initialize kadmin link: %s", error_message(ret));
290     leave(NULL, 1);
291   }
292   
293   ret = krb_get_cred (PWSERV_NAME, KADM_SINST, u_realm, &c);
294   if (ret != KSUCCESS) {
295     umask(077);
296        
297     /*
298      *  create ticket file and get admin tickets
299      */
300     snprintf(tktstring, sizeof(tktstring), "%s_ksrvutil_%d",
301              TKT_ROOT, (int)getpid());
302     krb_set_tkt_string(tktstring);
303     destroyp = TRUE;
304        
305     ret = get_admin_password(u_name, u_inst, u_realm);
306     if (ret) {
307       warnx("Couldn't get admin password.");
308       leave(NULL, 1);
309     }
310   }  
311   for(;p;){
312     get_srvtab_ent(unique_filename, fd, filename, p->name, p->inst, p->realm);
313     p=p->next;
314   }
315   unlink(tktstring);
316 }
317
318 static void
319 parseinput (char *result, size_t sz, char *val, char *def)
320 {
321   char *lim;
322   int inq;
323
324   if (val[0] == '\0') {
325     strlcpy (result, def, sz);
326     return;
327   }
328   lim = result + sz - 1;
329   inq = 0;
330   while(*val && result < lim) {
331     switch(*val) {
332     case '\'' :
333       inq = !inq;
334       ++val;
335       break;
336     case '\\' :
337       if(!inq)
338         val++;
339     default:
340       *result++ = *val++;
341       break;
342     }
343   }
344   *result = '\0';
345 }
346
347 void
348 ksrvutil_get(int unique_filename, int fd, char *filename, int argc, char **argv)
349 {
350   char sname[ANAME_SZ];         /* name of service */
351   char sinst[INST_SZ];          /* instance of service */
352   char srealm[REALM_SZ];        /* realm of service */
353   char databuf[BUFSIZ];
354   char local_hostname[100];
355   char prompt[100];
356   struct srv_ent *head=NULL;
357   int i;
358
359   gethostname(local_hostname, sizeof(local_hostname));
360   strlcpy(local_hostname,
361                   krb_get_phost(local_hostname),
362                   sizeof(local_hostname));
363
364   if (argc)
365     for(i=0; i < argc; ++i) {
366       struct srv_ent *p=malloc(sizeof(*p));
367
368       if(p == NULL) {
369         warnx ("out of memory in malloc");
370         leave(NULL,1);
371       }
372       p->next = head;
373       strlcpy (p->realm, u_realm, sizeof(p->realm));
374       if (kname_parse (p->name, p->inst, p->realm, argv[i]) !=
375           KSUCCESS) {
376         warnx ("parse error on '%s'\n", argv[i]);
377         free(p);
378         continue;
379       }
380       if (p->name[0] == '\0')
381         strlcpy(p->name, "rcmd", sizeof(p->name));
382       if (p->inst[0] == '\0')
383         strlcpy(p->inst, local_hostname, sizeof(p->inst));
384       if (p->realm[0] == '\0')
385         strlcpy(p->realm, u_realm, sizeof(p->realm));
386       head = p;
387     }
388
389   else
390     do{
391       safe_read_stdin("Name [rcmd]: ", databuf, sizeof(databuf));
392       parseinput (sname, sizeof(sname), databuf, "rcmd");
393     
394       snprintf(prompt, sizeof(prompt), "Instance [%s]: ", local_hostname);
395       safe_read_stdin(prompt, databuf, sizeof(databuf));
396       parseinput (sinst, sizeof(sinst), databuf, local_hostname);
397     
398       snprintf(prompt, sizeof(prompt), "Realm [%s]: ", u_realm);
399       safe_read_stdin(prompt, databuf, sizeof(databuf));
400       parseinput (srealm, sizeof(srealm), databuf, u_realm);
401
402       if(yn("Is this correct?")){
403         struct srv_ent *p=(struct srv_ent*)malloc(sizeof(struct srv_ent));
404         if (p == NULL) {
405             warnx ("out of memory in malloc");
406             leave(NULL,1);
407         }
408         p->next=head;
409         head=p;
410         strlcpy(p->name, sname, sizeof(p->name));
411         strlcpy(p->inst, sinst, sizeof(p->inst));
412         strlcpy(p->realm, srealm, sizeof(p->realm));
413       }
414     }while(ny("Add more keys?"));
415   
416   
417   ksrvutil_kadm(unique_filename, fd, filename, head);
418
419   {
420     struct srv_ent *p=head, *q;
421     while(p){
422       q=p;
423       p=p->next;
424       free(q);
425     }
426   }
427
428 }