1/*
2 * Copyright (c) 2003 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 KTH nor the names of its contributors may be
18 *    used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
22 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
25 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
28 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
30 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
31 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
32
33#include "krb5_locl.h"
34#include <getarg.h>
35#include <err.h>
36
37static void
38test_alname(krb5_context context, krb5_const_realm realm,
39	    const char *user, const char *inst,
40	    const char *localuser, int ok)
41{
42    krb5_principal p;
43    char localname[1024];
44    krb5_error_code ret;
45    char *princ;
46
47    ret = krb5_make_principal(context, &p, realm, user, inst, NULL);
48    if (ret)
49	krb5_err(context, 1, ret, "krb5_build_principal");
50
51    ret = krb5_unparse_name(context, p, &princ);
52    if (ret)
53	krb5_err(context, 1, ret, "krb5_unparse_name");
54
55    ret = krb5_aname_to_localname(context, p, sizeof(localname), localname);
56    krb5_free_principal(context, p);
57    free(princ);
58    if (ret) {
59	if (!ok)
60	    return;
61	krb5_err(context, 1, ret, "krb5_aname_to_localname: %s -> %s",
62		 princ, localuser);
63    }
64
65    if (strcmp(localname, localuser) != 0) {
66	if (ok)
67	    errx(1, "compared failed %s != %s (should have succeded)",
68		 localname, localuser);
69    } else {
70	if (!ok)
71	    errx(1, "compared failed %s == %s (should have failed)",
72		 localname, localuser);
73    }
74
75}
76
77static int version_flag = 0;
78static int help_flag	= 0;
79
80static struct getargs args[] = {
81    {"version",	0,	arg_flag,	&version_flag,
82     "print version", NULL },
83    {"help",	0,	arg_flag,	&help_flag,
84     NULL, NULL }
85};
86
87static void
88usage (int ret)
89{
90    arg_printusage (args,
91		    sizeof(args)/sizeof(*args),
92		    NULL,
93		    "");
94    exit (ret);
95}
96
97int
98main(int argc, char **argv)
99{
100    krb5_context context;
101    krb5_error_code ret;
102    krb5_realm realm;
103    int optidx = 0;
104    char *user;
105
106    setprogname(argv[0]);
107
108    if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
109	usage(1);
110
111    if (help_flag)
112	usage (0);
113
114    if(version_flag){
115	print_version(NULL);
116	exit(0);
117    }
118
119    argc -= optidx;
120    argv += optidx;
121
122    if (argc != 1)
123	errx(1, "first argument should be a local user that in root .k5login");
124
125    user = argv[0];
126
127    ret = krb5_init_context(&context);
128    if (ret)
129	errx (1, "krb5_init_context failed: %d", ret);
130
131    ret = krb5_get_default_realm(context, &realm);
132    if (ret)
133	krb5_err(context, 1, ret, "krb5_get_default_realm");
134
135    test_alname(context, realm, user, NULL, user, 1);
136    test_alname(context, realm, user, "root", "root", 1);
137
138    test_alname(context, "FOO.BAR.BAZ.KAKA", user, NULL, user, 0);
139    test_alname(context, "FOO.BAR.BAZ.KAKA", user, "root", "root", 0);
140
141    test_alname(context, realm, user, NULL,
142		"not-same-as-user", 0);
143    test_alname(context, realm, user, "root",
144		"not-same-as-user", 0);
145
146    test_alname(context, "FOO.BAR.BAZ.KAKA", user, NULL,
147		"not-same-as-user", 0);
148    test_alname(context, "FOO.BAR.BAZ.KAKA", user, "root",
149		"not-same-as-user", 0);
150
151    krb5_free_context(context);
152
153    return 0;
154}
155