1/*
2 * Copyright (c) 1997-2004 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 "kpasswd_locl.h"
35RCSID("$Id$");
36
37static int version_flag;
38static int help_flag;
39static char *admin_principal_str;
40static char *cred_cache_str;
41
42static struct getargs args[] = {
43    { "admin-principal",	0,   arg_string, &admin_principal_str, NULL,
44   	 NULL },
45    { "cache",			'c', arg_string, &cred_cache_str, NULL, NULL },
46    { "version", 		0,   arg_flag, &version_flag, NULL, NULL },
47    { "help",			0,   arg_flag, &help_flag, NULL, NULL }
48};
49
50static void
51usage (int ret, struct getargs *a, int num_args)
52{
53    arg_printusage (a, num_args, NULL, "[principal ...]");
54    exit (ret);
55}
56
57static int
58change_password(krb5_context context,
59		krb5_principal principal,
60		krb5_ccache id)
61{
62    krb5_data result_code_string, result_string;
63    int result_code;
64    krb5_error_code ret;
65    char pwbuf[BUFSIZ];
66    char *msg, *name;
67
68    krb5_data_zero (&result_code_string);
69    krb5_data_zero (&result_string);
70
71    name = msg = NULL;
72    if (principal == NULL)
73	asprintf(&msg, "New password: ");
74    else {
75	ret = krb5_unparse_name(context, principal, &name);
76	if (ret)
77	    krb5_err(context, 1, ret, "krb5_unparse_name");
78
79	asprintf(&msg, "New password for %s: ", name);
80    }
81
82    if (msg == NULL)
83	krb5_errx (context, 1, "out of memory");
84
85    ret = UI_UTIL_read_pw_string (pwbuf, sizeof(pwbuf), msg, 1);
86    free(msg);
87    if (name)
88	free(name);
89    if (ret != 0) {
90	return 1;
91    }
92
93    ret = krb5_set_password_using_ccache (context, id, pwbuf,
94					  principal,
95					  &result_code,
96					  &result_code_string,
97					  &result_string);
98    if (ret) {
99	krb5_warn (context, ret, "krb5_set_password_using_ccache");
100	return 1;
101    }
102
103    printf ("%s%s%.*s\n", krb5_passwd_result_to_string(context, result_code),
104	    result_string.length > 0 ? " : " : "",
105	    (int)result_string.length,
106	    result_string.length > 0 ? (char *)result_string.data : "");
107
108    krb5_data_free (&result_code_string);
109    krb5_data_free (&result_string);
110
111    return ret != 0;
112}
113
114
115int
116main (int argc, char **argv)
117{
118    krb5_error_code ret;
119    krb5_context context;
120    krb5_principal principal;
121    krb5_get_init_creds_opt *opt;
122    krb5_ccache id = NULL;
123    int exit_value;
124    int optidx = 0;
125
126    setprogname(argv[0]);
127
128    if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
129	usage(1, args, sizeof(args) / sizeof(args[0]));
130    if (help_flag)
131	usage(0, args, sizeof(args) / sizeof(args[0]));
132    if (version_flag) {
133	print_version(NULL);
134	return 0;
135    }
136    argc -= optidx;
137    argv += optidx;
138
139    ret = krb5_init_context (&context);
140    if (ret)
141	errx (1, "krb5_init_context failed: %d", ret);
142
143    ret = krb5_get_init_creds_opt_alloc (context, &opt);
144    if (ret)
145	krb5_err(context, 1, ret, "krb5_get_init_creds_opt_alloc");
146
147    krb5_get_init_creds_opt_set_tkt_life (opt, 300);
148    krb5_get_init_creds_opt_set_forwardable (opt, FALSE);
149    krb5_get_init_creds_opt_set_proxiable (opt, FALSE);
150
151    if (cred_cache_str) {
152	ret = krb5_cc_resolve(context, cred_cache_str, &id);
153	if (ret)
154	    krb5_err (context, 1, ret, "krb5_cc_resolve");
155    } else {
156	ret = krb5_cc_new_unique(context, krb5_cc_type_memory, NULL, &id);
157	if (ret)
158	    krb5_err (context, 1, ret, "krb5_cc_new_unique");
159    }
160
161    if (cred_cache_str == NULL) {
162	krb5_principal admin_principal = NULL;
163	krb5_creds cred;
164
165	if (admin_principal_str) {
166	    ret = krb5_parse_name (context, admin_principal_str,
167				   &admin_principal);
168	    if (ret)
169		krb5_err (context, 1, ret, "krb5_parse_name");
170	} else if (argc == 1) {
171	    ret = krb5_parse_name (context, argv[0], &admin_principal);
172	    if (ret)
173		krb5_err (context, 1, ret, "krb5_parse_name");
174	} else {
175	    ret = krb5_get_default_principal (context, &admin_principal);
176	    if (ret)
177		krb5_err (context, 1, ret, "krb5_get_default_principal");
178	}
179
180	ret = krb5_get_init_creds_password (context,
181					    &cred,
182					    admin_principal,
183					    NULL,
184					    krb5_prompter_posix,
185					    NULL,
186					    0,
187					    "kadmin/changepw",
188					    opt);
189	switch (ret) {
190	case 0:
191	    break;
192	case KRB5_LIBOS_PWDINTR :
193	    return 1;
194	case KRB5KRB_AP_ERR_BAD_INTEGRITY :
195	case KRB5KRB_AP_ERR_MODIFIED :
196	    krb5_errx(context, 1, "Password incorrect");
197	    break;
198	default:
199	    krb5_err(context, 1, ret, "krb5_get_init_creds");
200	}
201
202	krb5_get_init_creds_opt_free(context, opt);
203
204	ret = krb5_cc_initialize(context, id, admin_principal);
205	krb5_free_principal(context, admin_principal);
206	if (ret)
207	    krb5_err(context, 1, ret, "krb5_cc_initialize");
208
209	ret = krb5_cc_store_cred(context, id, &cred);
210	if (ret)
211	    krb5_err(context, 1, ret, "krb5_cc_store_cred");
212
213	krb5_free_cred_contents (context, &cred);
214    }
215
216    if (argc == 0) {
217	exit_value = change_password(context, NULL, id);
218    } else {
219	exit_value = 0;
220
221	while (argc-- > 0) {
222
223	    ret = krb5_parse_name (context, argv[0], &principal);
224	    if (ret)
225		krb5_err (context, 1, ret, "krb5_parse_name");
226
227	    ret = change_password(context, principal, id);
228	    if (ret)
229		exit_value = 1;
230	    krb5_free_principal(context, principal);
231	    argv++;
232	}
233    }
234
235    if (cred_cache_str == NULL) {
236	ret = krb5_cc_destroy(context, id);
237	if (ret)
238	    krb5_err (context, 1, ret, "krb5_cc_destroy");
239    } else {
240	ret = krb5_cc_close(context, id);
241	if (ret)
242	    krb5_err (context, 1, ret, "krb5_cc_close");
243    }
244
245    krb5_free_context (context);
246    return exit_value;
247}
248