xref: /netbsd-src/crypto/external/bsd/heimdal/dist/kuser/kswitch.c (revision aceb213538ec08a74028e213127af18aa17bf1cf)
1 /*	$NetBSD: kswitch.c,v 1.1.1.1 2011/04/13 18:14:39 elric Exp $	*/
2 
3 /*
4  * Copyright (c) 2008 - 2010 Kungliga Tekniska Högskolan
5  * (Royal Institute of Technology, Stockholm, Sweden).
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * 3. Neither the name of the Institute nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include "kuser_locl.h"
37 #include "kcc-commands.h"
38 
39 #ifdef HAVE_READLINE
40 char *readline(char *prompt);
41 #else
42 
43 static char *
44 readline(char *prompt)
45 {
46     char buf[BUFSIZ];
47     printf ("%s", prompt);
48     fflush (stdout);
49     if(fgets(buf, sizeof(buf), stdin) == NULL)
50 	return NULL;
51     buf[strcspn(buf, "\r\n")] = '\0';
52     return strdup(buf);
53 }
54 
55 #endif
56 
57 /*
58  *
59  */
60 
61 int
62 kswitch(struct kswitch_options *opt, int argc, char **argv)
63 {
64     krb5_error_code ret;
65     krb5_ccache id = NULL;
66 
67     if (opt->cache_string && opt->principal_string)
68 	krb5_errx(kcc_context, 1,
69 		  N_("Both --cache and --principal given, choose one", ""));
70 
71     if (opt->interactive_flag) {
72 	krb5_cc_cache_cursor cursor;
73 	krb5_ccache *ids = NULL;
74 	size_t i, len = 0;
75 	char *name;
76 	rtbl_t ct;
77 
78 	ct = rtbl_create();
79 
80 	rtbl_add_column(ct, "", 0);
81 	rtbl_add_column(ct, "Principal", 0);
82 	rtbl_set_column_prefix(ct, "Principal", "    ");
83 
84 	ret = krb5_cc_cache_get_first(kcc_context, NULL, &cursor);
85 	if (ret)
86 	    krb5_err(kcc_context, 1, ret, "krb5_cc_cache_get_first");
87 
88 	while (krb5_cc_cache_next(kcc_context, cursor, &id) == 0) {
89 	    krb5_principal p;
90 	    char num[10];
91 
92 	    ret = krb5_cc_get_principal(kcc_context, id, &p);
93 	    if (ret)
94 		continue;
95 
96 	    ret = krb5_unparse_name(kcc_context, p, &name);
97 	    krb5_free_principal(kcc_context, p);
98 
99 	    snprintf(num, sizeof(num), "%d", (int)(len + 1));
100 	    rtbl_add_column_entry(ct, "", num);
101 	    rtbl_add_column_entry(ct, "Principal", name);
102 	    free(name);
103 
104 	    ids = erealloc(ids, (len + 1) * sizeof(ids[0]));
105 	    ids[len] = id;
106 	    len++;
107 	}
108 	krb5_cc_cache_end_seq_get(kcc_context, cursor);
109 
110 	rtbl_format(ct, stdout);
111 	rtbl_destroy(ct);
112 
113 	name = readline("Select number: ");
114 	if (name) {
115 	    i = atoi(name);
116 	    if (i == 0)
117 		krb5_errx(kcc_context, 1, "Cache number '%s' is invalid", name);
118 	    if (i > len)
119 		krb5_errx(kcc_context, 1, "Cache number '%s' is too large", name);
120 
121 	    id = ids[i - 1];
122 	    ids[i - 1] = NULL;
123 	} else
124 	    krb5_errx(kcc_context, 1, "No cache selected");
125 	for (i = 0; i < len; i++)
126 	    if (ids[i])
127 		krb5_cc_close(kcc_context, ids[i]);
128 
129     } else if (opt->principal_string) {
130 	krb5_principal p;
131 
132 	ret = krb5_parse_name(kcc_context, opt->principal_string, &p);
133 	if (ret)
134 	    krb5_err(kcc_context, 1, ret, "krb5_parse_name: %s",
135 		     opt->principal_string);
136 
137 	ret = krb5_cc_cache_match(kcc_context, p, &id);
138 	if (ret)
139 	    krb5_err(kcc_context, 1, ret,
140 		     N_("Did not find principal: %s", ""),
141 		     opt->principal_string);
142 
143 	krb5_free_principal(kcc_context, p);
144 
145     } else if (opt->cache_string) {
146 	const krb5_cc_ops *ops;
147 	char *str;
148 
149 	ops = krb5_cc_get_prefix_ops(kcc_context, opt->type_string);
150 	if (ops == NULL)
151 	    krb5_err(kcc_context, 1, 0, "krb5_cc_get_prefix_ops");
152 
153 	asprintf(&str, "%s:%s", ops->prefix, opt->cache_string);
154 	if (str == NULL)
155 	    krb5_errx(kcc_context, 1, N_("out of memory", ""));
156 
157 	ret = krb5_cc_resolve(kcc_context, str, &id);
158 	if (ret)
159 	    krb5_err(kcc_context, 1, ret, "krb5_cc_resolve: %s", str);
160 
161 	free(str);
162     } else {
163 	krb5_errx(kcc_context, 1, "missing option for kswitch");
164     }
165 
166     ret = krb5_cc_switch(kcc_context, id);
167     if (ret)
168 	krb5_err(kcc_context, 1, ret, "krb5_cc_switch");
169 
170     return 0;
171 }
172