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