Initial import from FreeBSD RELENG_4:
[dragonfly.git] / crypto / heimdal / kuser / kinit.c
1 /*
2  * Copyright (c) 1997-2002 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
34 #include "kuser_locl.h"
35 RCSID("$Id: kinit.c,v 1.90 2002/09/09 22:17:53 joda Exp $");
36
37 int forwardable_flag    = -1;
38 int proxiable_flag      = -1;
39 int renewable_flag      = -1;
40 int renew_flag          = 0;
41 int validate_flag       = 0;
42 int version_flag        = 0;
43 int help_flag           = 0;
44 int addrs_flag          = 1;
45 struct getarg_strings extra_addresses;
46 int anonymous_flag      = 0;
47 char *lifetime          = NULL;
48 char *renew_life        = NULL;
49 char *server            = NULL;
50 char *cred_cache        = NULL;
51 char *start_str         = NULL;
52 struct getarg_strings etype_str;
53 int use_keytab          = 0;
54 char *keytab_str        = NULL;
55 #ifdef KRB4
56 int get_v4_tgt          = -1;
57 int do_afslog           = -1;
58 int convert_524;
59 #endif
60 int fcache_version;
61
62 static struct getargs args[] = {
63 #ifdef KRB4
64     { "524init",        '4', arg_flag, &get_v4_tgt,
65       "obtain version 4 TGT" },
66     
67     { "524convert",     '9', arg_flag, &convert_524,
68       "only convert ticket to version 4" },
69     
70     { "afslog",         0  , arg_flag, &do_afslog,
71       "obtain afs tokens"  },
72 #endif
73     { "cache",          'c', arg_string, &cred_cache,
74       "credentials cache", "cachename" },
75
76     { "forwardable",    'f', arg_flag, &forwardable_flag,
77       "get forwardable tickets"},
78
79     { "keytab",         't', arg_string, &keytab_str,
80       "keytab to use", "keytabname" },
81
82     { "lifetime",       'l', arg_string, &lifetime,
83       "lifetime of tickets", "time"},
84
85     { "proxiable",      'p', arg_flag, &proxiable_flag,
86       "get proxiable tickets" },
87
88     { "renew",          'R', arg_flag, &renew_flag,
89       "renew TGT" },
90
91     { "renewable",      0,   arg_flag, &renewable_flag,
92       "get renewable tickets" },
93
94     { "renewable-life", 'r', arg_string, &renew_life,
95       "renewable lifetime of tickets", "time" },
96
97     { "server",         'S', arg_string, &server,
98       "server to get ticket for", "principal" },
99
100     { "start-time",     's', arg_string, &start_str,
101       "when ticket gets valid", "time" },
102
103     { "use-keytab",     'k', arg_flag, &use_keytab,
104       "get key from keytab" },
105
106     { "validate",       'v', arg_flag, &validate_flag,
107       "validate TGT" },
108
109     { "enctypes",       'e', arg_strings, &etype_str,
110       "encryption types to use", "enctypes" },
111
112     { "fcache-version", 0,   arg_integer, &fcache_version,
113       "file cache version to create" },
114
115     { "addresses",      0,   arg_negative_flag, &addrs_flag,
116       "request a ticket with no addresses" },
117
118     { "extra-addresses",'a', arg_strings,       &extra_addresses,
119       "include these extra addresses", "addresses" },
120
121     { "anonymous",      0,   arg_flag,  &anonymous_flag,
122       "request an anonymous ticket" },
123
124     { "version",        0,   arg_flag, &version_flag },
125     { "help",           0,   arg_flag, &help_flag }
126 };
127
128 static void
129 usage (int ret)
130 {
131     arg_printusage (args,
132                     sizeof(args)/sizeof(*args),
133                     NULL,
134                     "[principal [command]]");
135     exit (ret);
136 }
137
138 #ifdef KRB4
139 /* for when the KDC tells us it's a v4 one, we try to talk that */
140
141 static int
142 key_to_key(const char *user,
143            char *instance,
144            const char *realm,
145            const void *arg,
146            des_cblock *key)
147 {
148     memcpy(key, arg, sizeof(des_cblock));
149     return 0;
150 }
151
152 static int
153 do_v4_fallback (krb5_context context,
154                 const krb5_principal principal,
155                 int lifetime,
156                 int use_srvtab, const char *srvtab_str,
157                 const char *passwd)
158 {
159     int ret;
160     krb_principal princ;
161     des_cblock key;
162     krb5_error_code kret;
163
164     if (lifetime == 0)
165         lifetime = DEFAULT_TKT_LIFE;
166     else
167         lifetime = krb_time_to_life (0, lifetime);
168
169     kret = krb5_524_conv_principal (context, principal,
170                                     princ.name,
171                                     princ.instance,
172                                     princ.realm);
173     if (kret) {
174         krb5_warn (context, kret, "krb5_524_conv_principal");
175         return 1;
176     }
177
178     if (use_srvtab || srvtab_str) {
179         if (srvtab_str == NULL)
180             srvtab_str = KEYFILE;
181
182         ret = read_service_key (princ.name, princ.instance, princ.realm,
183                                 0, srvtab_str, (char *)&key);
184         if (ret) {
185             warnx ("read_service_key %s: %s", srvtab_str,
186                    krb_get_err_text (ret));
187             return 1;
188         }
189         ret = krb_get_in_tkt (princ.name, princ.instance, princ.realm,
190                               KRB_TICKET_GRANTING_TICKET, princ.realm,
191                               lifetime, key_to_key, NULL, key);
192     } else {
193         ret = krb_get_pw_in_tkt(princ.name, princ.instance, princ.realm, 
194                                 KRB_TICKET_GRANTING_TICKET, princ.realm, 
195                                 lifetime, passwd);
196     }
197     memset (key, 0, sizeof(key));
198     if (ret) {
199         warnx ("%s", krb_get_err_text(ret));
200         return 1;
201     }
202     if (do_afslog && k_hasafs()) {
203         if ((ret = krb_afslog(NULL, NULL)) != 0 && ret != KDC_PR_UNKNOWN) {
204             if(ret > 0)
205                 warnx ("%s", krb_get_err_text(ret));
206             else
207                 warnx ("failed to store AFS token");
208         }
209     }
210     return 0;
211 }
212
213
214 /*
215  * the special version of get_default_principal that takes v4 into account
216  */
217
218 static krb5_error_code
219 kinit_get_default_principal (krb5_context context,
220                              krb5_principal *princ)
221 {
222     krb5_error_code ret;
223     krb5_ccache id;
224     krb_principal v4_princ;
225     int kret;
226
227     ret = krb5_cc_default (context, &id);
228     if (ret == 0) {
229         ret = krb5_cc_get_principal (context, id, princ);
230         krb5_cc_close (context, id);
231         if (ret == 0)
232             return 0;
233     }
234
235     kret = krb_get_tf_fullname (tkt_string(),
236                                 v4_princ.name,
237                                 v4_princ.instance,
238                                 v4_princ.realm);
239     if (kret == KSUCCESS) {
240         ret = krb5_425_conv_principal (context,
241                                        v4_princ.name,
242                                        v4_princ.instance,
243                                        v4_princ.realm,
244                                        princ);
245         if (ret == 0)
246             return 0;
247     }
248     return krb5_get_default_principal (context, princ);
249 }
250
251 #else /* !KRB4 */
252
253 static krb5_error_code
254 kinit_get_default_principal (krb5_context context,
255                              krb5_principal *princ)
256 {
257     return krb5_get_default_principal (context, princ);
258 }
259
260 #endif /* !KRB4 */
261
262 static krb5_error_code
263 get_server(krb5_context context,
264            krb5_principal client,
265            const char *server,
266            krb5_principal *princ)
267 {
268     krb5_realm *client_realm;
269     if(server)
270         return krb5_parse_name(context, server, princ);
271
272     client_realm = krb5_princ_realm (context, client);
273     return krb5_make_principal(context, princ, *client_realm,
274                                KRB5_TGS_NAME, *client_realm, NULL);
275 }
276
277 #ifdef KRB4
278 static krb5_error_code
279 do_524init(krb5_context context, krb5_ccache ccache, 
280            krb5_creds *creds, const char *server)
281 {
282     krb5_error_code ret;
283     CREDENTIALS c;
284     krb5_creds in_creds, *real_creds;
285
286     if(creds != NULL)
287         real_creds = creds;
288     else {
289         krb5_principal client;
290         krb5_cc_get_principal(context, ccache, &client);
291         memset(&in_creds, 0, sizeof(in_creds));
292         ret = get_server(context, client, server, &in_creds.server);
293         krb5_free_principal(context, client);
294         if(ret)
295             return ret;
296         ret = krb5_get_credentials(context, 0, ccache, &in_creds, &real_creds);
297         krb5_free_principal(context, in_creds.server);
298         if(ret)
299             return ret;
300     }
301     ret = krb524_convert_creds_kdc_ccache(context, ccache, real_creds, &c);
302     if(ret)
303         krb5_warn(context, ret, "converting creds");
304     else {
305         int tret = tf_setup(&c, c.pname, c.pinst);
306         if(tret)
307             krb5_warnx(context, "saving v4 creds: %s", krb_get_err_text(tret));
308     }
309
310     if(creds == NULL)
311         krb5_free_creds(context, real_creds);
312     memset(&c, 0, sizeof(c));
313
314     return ret;
315 }
316 #endif
317
318 static int
319 renew_validate(krb5_context context, 
320                int renew,
321                int validate,
322                krb5_ccache cache, 
323                const char *server,
324                krb5_deltat life)
325 {
326     krb5_error_code ret;
327     krb5_creds in, *out;
328     krb5_kdc_flags flags;
329
330     memset(&in, 0, sizeof(in));
331
332     ret = krb5_cc_get_principal(context, cache, &in.client);
333     if(ret) {
334         krb5_warn(context, ret, "krb5_cc_get_principal");
335         return ret;
336     }
337     ret = get_server(context, in.client, server, &in.server);
338     if(ret) {
339         krb5_warn(context, ret, "get_server");
340         goto out;
341     }
342     flags.i = 0;
343     flags.b.renewable         = flags.b.renew = renew;
344     flags.b.validate          = validate;
345     if (forwardable_flag != -1)
346         flags.b.forwardable       = forwardable_flag;
347     if (proxiable_flag != -1)
348         flags.b.proxiable         = proxiable_flag;
349     if (anonymous_flag != -1)
350         flags.b.request_anonymous = anonymous_flag;
351     if(life)
352         in.times.endtime = time(NULL) + life;
353
354     ret = krb5_get_kdc_cred(context,
355                             cache,
356                             flags,
357                             NULL,
358                             NULL,
359                             &in,
360                             &out);
361     if(ret) {
362         krb5_warn(context, ret, "krb5_get_kdc_cred");
363         goto out;
364     }
365     ret = krb5_cc_initialize(context, cache, in.client);
366     if(ret) {
367         krb5_free_creds (context, out);
368         krb5_warn(context, ret, "krb5_cc_initialize");
369         goto out;
370     }
371     ret = krb5_cc_store_cred(context, cache, out);
372
373 #ifdef KRB4
374     if(ret == 0 && server == NULL) {
375         /* only do this if it's a general renew-my-tgt request */
376         if(get_v4_tgt)
377             do_524init(context, cache, out, NULL);
378
379         if(do_afslog && k_hasafs())
380             krb5_afslog(context, cache, NULL, NULL);
381     }
382 #endif
383
384     krb5_free_creds (context, out);
385     if(ret) {
386         krb5_warn(context, ret, "krb5_cc_store_cred");
387         goto out;
388     }
389 out:
390     krb5_free_creds_contents(context, &in);
391     return ret;
392 }
393
394 static krb5_error_code
395 get_new_tickets(krb5_context context, 
396                 krb5_principal principal,
397                 krb5_ccache ccache,
398                 krb5_deltat ticket_life)
399 {
400     krb5_error_code ret;
401     krb5_get_init_creds_opt opt;
402     krb5_addresses no_addrs;
403     krb5_creds cred;
404     char passwd[256];
405     krb5_deltat start_time = 0;
406     krb5_deltat renew = 0;
407
408     memset(&cred, 0, sizeof(cred));
409
410     krb5_get_init_creds_opt_init (&opt);
411     
412     krb5_get_init_creds_opt_set_default_flags(context, "kinit", 
413                                               /* XXX */principal->realm, &opt);
414
415     if(forwardable_flag != -1)
416         krb5_get_init_creds_opt_set_forwardable (&opt, forwardable_flag);
417     if(proxiable_flag != -1)
418         krb5_get_init_creds_opt_set_proxiable (&opt, proxiable_flag);
419     if(anonymous_flag != -1)
420         krb5_get_init_creds_opt_set_anonymous (&opt, anonymous_flag);
421
422     if (!addrs_flag) {
423         no_addrs.len = 0;
424         no_addrs.val = NULL;
425
426         krb5_get_init_creds_opt_set_address_list (&opt, &no_addrs);
427     }
428
429     if(renew_life) {
430         renew = parse_time (renew_life, "s");
431         if (renew < 0)
432             errx (1, "unparsable time: %s", renew_life);
433
434         krb5_get_init_creds_opt_set_renew_life (&opt, renew);
435     } else if (renewable_flag == 1)
436         krb5_get_init_creds_opt_set_renew_life (&opt, 1 << 30);
437
438
439     if(ticket_life != 0)
440         krb5_get_init_creds_opt_set_tkt_life (&opt, ticket_life);
441
442     if(start_str) {
443         int tmp = parse_time (start_str, "s");
444         if (tmp < 0)
445             errx (1, "unparsable time: %s", start_str);
446
447         start_time = tmp;
448     }
449
450     if(etype_str.num_strings) {
451         krb5_enctype *enctype = NULL;
452         int i;
453         enctype = malloc(etype_str.num_strings * sizeof(*enctype));
454         if(enctype == NULL)
455             errx(1, "out of memory");
456         for(i = 0; i < etype_str.num_strings; i++) {
457             ret = krb5_string_to_enctype(context, 
458                                          etype_str.strings[i], 
459                                          &enctype[i]);
460             if(ret)
461                 errx(1, "unrecognized enctype: %s", etype_str.strings[i]);
462         }
463         krb5_get_init_creds_opt_set_etype_list(&opt, enctype, 
464                                                etype_str.num_strings);
465     }
466
467     if(use_keytab || keytab_str) {
468         krb5_keytab kt;
469         if(keytab_str)
470             ret = krb5_kt_resolve(context, keytab_str, &kt);
471         else
472             ret = krb5_kt_default(context, &kt);
473         if (ret)
474             krb5_err (context, 1, ret, "resolving keytab");
475         ret = krb5_get_init_creds_keytab (context,
476                                           &cred,
477                                           principal,
478                                           kt,
479                                           start_time,
480                                           server,
481                                           &opt);
482         krb5_kt_close(context, kt);
483     } else {
484         char *p, *prompt;
485
486         krb5_unparse_name (context, principal, &p);
487         asprintf (&prompt, "%s's Password: ", p);
488         free (p);
489
490         if (des_read_pw_string(passwd, sizeof(passwd)-1, prompt, 0)){
491             memset(passwd, 0, sizeof(passwd));
492             exit(1);
493         }
494
495         free (prompt);
496
497         ret = krb5_get_init_creds_password (context,
498                                             &cred,
499                                             principal,
500                                             passwd,
501                                             krb5_prompter_posix,
502                                             NULL,
503                                             start_time,
504                                             server,
505                                             &opt);
506     }
507 #ifdef KRB4
508     if (ret == KRB5KRB_AP_ERR_V4_REPLY || ret == KRB5_KDC_UNREACH) {
509         int exit_val;
510
511         exit_val = do_v4_fallback (context, principal, ticket_life,
512                                    use_keytab, keytab_str, passwd);
513         get_v4_tgt = 0;
514         do_afslog  = 0;
515         memset(passwd, 0, sizeof(passwd));
516         if (exit_val == 0 || ret == KRB5KRB_AP_ERR_V4_REPLY)
517             return exit_val;
518     }
519 #endif
520     memset(passwd, 0, sizeof(passwd));
521
522     switch(ret){
523     case 0:
524         break;
525     case KRB5_LIBOS_PWDINTR: /* don't print anything if it was just C-c:ed */
526         exit(1);
527     case KRB5KRB_AP_ERR_BAD_INTEGRITY:
528     case KRB5KRB_AP_ERR_MODIFIED:
529         krb5_errx(context, 1, "Password incorrect");
530         break;
531     default:
532         krb5_err(context, 1, ret, "krb5_get_init_creds");
533     }
534
535     if(ticket_life != 0) {
536         if(abs(cred.times.endtime - cred.times.starttime - ticket_life) > 30) {
537             char life[32];
538             unparse_time(cred.times.endtime - cred.times.starttime, 
539                          life, sizeof(life));
540             krb5_warnx(context, "NOTICE: ticket lifetime is %s", life);
541         }
542     }
543     if(renew != 0) {
544         if(abs(cred.times.renew_till - cred.times.starttime - renew) > 30) {
545             char life[32];
546             unparse_time(cred.times.renew_till - cred.times.starttime, 
547                          life, sizeof(life));
548             krb5_warnx(context, "NOTICE: ticket renewable lifetime is %s", 
549                        life);
550         }
551     }
552
553     ret = krb5_cc_initialize (context, ccache, cred.client);
554     if (ret)
555         krb5_err (context, 1, ret, "krb5_cc_initialize");
556     
557     ret = krb5_cc_store_cred (context, ccache, &cred);
558     if (ret)
559         krb5_err (context, 1, ret, "krb5_cc_store_cred");
560
561     krb5_free_creds_contents (context, &cred);
562
563     return 0;
564 }
565
566 int
567 main (int argc, char **argv)
568 {
569     krb5_error_code ret;
570     krb5_context context;
571     krb5_ccache  ccache;
572     krb5_principal principal;
573     int optind = 0;
574     krb5_deltat ticket_life = 0;
575
576     setprogname (argv[0]);
577     
578     ret = krb5_init_context (&context);
579     if (ret)
580         errx(1, "krb5_init_context failed: %d", ret);
581   
582     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optind))
583         usage(1);
584     
585     if (help_flag)
586         usage (0);
587
588     if(version_flag) {
589         print_version(NULL);
590         exit(0);
591     }
592
593     argc -= optind;
594     argv += optind;
595
596     if (argv[0]) {
597         ret = krb5_parse_name (context, argv[0], &principal);
598         if (ret)
599             krb5_err (context, 1, ret, "krb5_parse_name");
600     } else {
601         ret = kinit_get_default_principal (context, &principal);
602         if (ret)
603             krb5_err (context, 1, ret, "krb5_get_default_principal");
604     }
605
606     if(fcache_version)
607         krb5_set_fcache_version(context, fcache_version);
608
609     if(cred_cache) 
610         ret = krb5_cc_resolve(context, cred_cache, &ccache);
611     else {
612         if(argc > 1) {
613             char s[1024];
614             ret = krb5_cc_gen_new(context, &krb5_fcc_ops, &ccache);
615             if(ret)
616                 krb5_err(context, 1, ret, "creating cred cache");
617             snprintf(s, sizeof(s), "%s:%s",
618                      krb5_cc_get_type(context, ccache),
619                      krb5_cc_get_name(context, ccache));
620             setenv("KRB5CCNAME", s, 1);
621 #ifdef KRB4
622             {
623                 int fd;
624                 snprintf(s, sizeof(s), "%s_XXXXXX", TKT_ROOT);
625                 if((fd = mkstemp(s)) >= 0) {
626                     close(fd);
627                     setenv("KRBTKFILE", s, 1);
628                     if (k_hasafs ())
629                         k_setpag();
630                 }
631             }
632 #endif
633         } else
634             ret = krb5_cc_default (context, &ccache);
635     }
636     if (ret)
637         krb5_err (context, 1, ret, "resolving credentials cache");
638
639     if (lifetime) {
640         int tmp = parse_time (lifetime, "s");
641         if (tmp < 0)
642             errx (1, "unparsable time: %s", lifetime);
643
644         ticket_life = tmp;
645     }
646 #ifdef KRB4
647     if(get_v4_tgt == -1)
648         krb5_appdefault_boolean(context, "kinit", 
649                                 krb5_principal_get_realm(context, principal), 
650                                 "krb4_get_tickets", TRUE, &get_v4_tgt);
651     if(do_afslog == -1)
652         krb5_appdefault_boolean(context, "kinit", 
653                                 krb5_principal_get_realm(context, principal), 
654                                 "afslog", TRUE, &do_afslog);
655 #endif
656
657     if(!addrs_flag && extra_addresses.num_strings > 0)
658         krb5_errx(context, 1, "specifying both extra addresses and "
659                   "no addresses makes no sense");
660     {
661         int i;
662         krb5_addresses addresses;
663         memset(&addresses, 0, sizeof(addresses));
664         for(i = 0; i < extra_addresses.num_strings; i++) {
665             ret = krb5_parse_address(context, extra_addresses.strings[i], 
666                                      &addresses);
667             if (ret == 0) {
668                 krb5_add_extra_addresses(context, &addresses);
669                 krb5_free_addresses(context, &addresses);
670             }
671         }
672         free_getarg_strings(&extra_addresses);
673     }
674
675     
676     if(renew_flag || validate_flag) {
677         ret = renew_validate(context, renew_flag, validate_flag, 
678                              ccache, server, ticket_life);
679         exit(ret != 0);
680     }
681
682 #ifdef KRB4
683     if(!convert_524)
684 #endif
685         get_new_tickets(context, principal, ccache, ticket_life);
686
687 #ifdef KRB4
688     if(get_v4_tgt)
689         do_524init(context, ccache, NULL, server);
690     if(do_afslog && k_hasafs())
691         krb5_afslog(context, ccache, NULL, NULL);
692 #endif
693     if(argc > 1) {
694         simple_execvp(argv[1], argv+1);
695         krb5_cc_destroy(context, ccache);
696 #ifdef KRB4
697         dest_tkt();
698         if(k_hasafs())
699             k_unlog();
700 #endif
701     } else 
702         krb5_cc_close (context, ccache);
703     krb5_free_principal(context, principal);
704     krb5_free_context (context);
705     return 0;
706 }