copy_cred_cache.c revision 178825
1/*
2 * Copyright (c) 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#ifdef HAVE_CONFIG_H
35#include <config.h>
36RCSID("$Id: copy_cred_cache.c 15542 2005-07-01 07:20:54Z lha $");
37#endif
38
39#include <stdlib.h>
40#include <krb5.h>
41#include <roken.h>
42#include <getarg.h>
43#include <parse_units.h>
44#include <parse_time.h>
45
46static int krbtgt_only_flag;
47static char *service_string;
48static char *enctype_string;
49static char *flags_string;
50static char *valid_string;
51static int fcache_version;
52static int help_flag;
53static int version_flag;
54
55static struct getargs args[] = {
56    { "krbtgt-only", 0, arg_flag, &krbtgt_only_flag,
57      "only copy local krbtgt" },
58    { "service", 0, arg_string, &service_string,
59      "limit to this service", "principal" },
60    { "enctype", 0, arg_string, &enctype_string,
61      "limit to this enctype", "enctype" },
62    { "flags", 0, arg_string, &flags_string,
63      "limit to these flags", "ticketflags" },
64    { "valid-for", 0, arg_string, &valid_string,
65      "limit to creds valid for at least this long", "time" },
66    { "fcache-version", 0, arg_integer, &fcache_version,
67      "file cache version to create" },
68    { "version", 0, arg_flag, &version_flag },
69    { "help", 'h', arg_flag, &help_flag }
70};
71
72static void
73usage(int ret)
74{
75    arg_printusage(args,
76		   sizeof(args) / sizeof(*args),
77		   NULL,
78		   "[from-cache] to-cache");
79    exit(ret);
80}
81
82static int32_t
83bitswap32(int32_t b)
84{
85    int32_t r = 0;
86    int i;
87    for (i = 0; i < 32; i++) {
88	r = r << 1 | (b & 1);
89	b = b >> 1;
90    }
91    return r;
92}
93
94static void
95parse_ticket_flags(krb5_context context,
96		   const char *string, krb5_ticket_flags *ret_flags)
97{
98    TicketFlags ff;
99    int flags = parse_flags(string, asn1_TicketFlags_units(), 0);
100    if (flags == -1)	/* XXX */
101	krb5_errx(context, 1, "bad flags specified: \"%s\"", string);
102
103    memset(&ff, 0, sizeof(ff));
104    ff.proxy = 1;
105    if (parse_flags("proxy", asn1_TicketFlags_units(), 0) == TicketFlags2int(ff))
106	ret_flags->i = flags;
107    else
108	ret_flags->i = bitswap32(flags);
109}
110
111int
112main(int argc, char **argv)
113{
114    krb5_error_code ret;
115    krb5_context context;
116    int optidx = 0;
117    const char *from_name, *to_name;
118    krb5_ccache from_ccache, to_ccache;
119    krb5_flags whichfields = 0;
120    krb5_creds mcreds;
121    unsigned int matched;
122
123    setprogname(argv[0]);
124
125    memset(&mcreds, 0, sizeof(mcreds));
126
127    if (getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
128	usage(1);
129
130    if (help_flag)
131	usage(0);
132
133    if (version_flag) {
134	print_version(NULL);
135	exit(0);
136    }
137    argc -= optidx;
138    argv += optidx;
139
140    if (argc < 1 || argc > 2)
141	usage(1);
142
143    if (krb5_init_context(&context))
144	errx(1, "krb5_init_context failed");
145
146    if (service_string) {
147	ret = krb5_parse_name(context, service_string, &mcreds.server);
148	if (ret)
149	    krb5_err(context, 1, ret, "%s", service_string);
150    }
151    if (enctype_string) {
152	krb5_enctype enctype;
153	ret = krb5_string_to_enctype(context, enctype_string, &enctype);
154	if (ret)
155	    krb5_err(context, 1, ret, "%s", enctype_string);
156	whichfields |= KRB5_TC_MATCH_KEYTYPE;
157	mcreds.session.keytype = enctype;
158    }
159    if (flags_string) {
160	parse_ticket_flags(context, flags_string, &mcreds.flags);
161	whichfields |= KRB5_TC_MATCH_FLAGS;
162    }
163    if (valid_string) {
164	time_t t = parse_time(valid_string, "s");
165	if(t < 0)
166	    errx(1, "unknown time \"%s\"", valid_string);
167	mcreds.times.endtime = time(NULL) + t;
168	whichfields |= KRB5_TC_MATCH_TIMES;
169    }
170    if (fcache_version)
171	krb5_set_fcache_version(context, fcache_version);
172
173    if (argc == 1) {
174	from_name = krb5_cc_default_name(context);
175	to_name = argv[0];
176    } else {
177	from_name = argv[0];
178	to_name = argv[1];
179    }
180
181    ret = krb5_cc_resolve(context, from_name, &from_ccache);
182    if (ret)
183	krb5_err(context, 1, ret, "%s", from_name);
184
185    if (krbtgt_only_flag) {
186	krb5_principal client;
187	ret = krb5_cc_get_principal(context, from_ccache, &client);
188	if (ret)
189	    krb5_err(context, 1, ret, "getting default principal");
190	ret = krb5_make_principal(context, &mcreds.server,
191				  krb5_principal_get_realm(context, client),
192				  KRB5_TGS_NAME,
193				  krb5_principal_get_realm(context, client),
194				  NULL);
195	if (ret)
196	    krb5_err(context, 1, ret, "constructing krbtgt principal");
197	krb5_free_principal(context, client);
198    }
199    ret = krb5_cc_resolve(context, to_name, &to_ccache);
200    if (ret)
201	krb5_err(context, 1, ret, "%s", to_name);
202
203    ret = krb5_cc_copy_cache_match(context, from_ccache, to_ccache,
204				   whichfields, &mcreds, &matched);
205    if (ret)
206	krb5_err(context, 1, ret, "copying cred cache");
207
208    krb5_cc_close(context, from_ccache);
209    if(matched == 0)
210	krb5_cc_destroy(context, to_ccache);
211    else
212	krb5_cc_close(context, to_ccache);
213    krb5_free_context(context);
214    return matched == 0;
215}
216