xref: /freebsd/crypto/heimdal/lib/krb5/test_alname.c (revision 6a068746777241722b2b32c5d0bc443a2a64d80b)
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 
37 static void
test_alname(krb5_context context,krb5_const_realm realm,const char * user,const char * inst,const char * localuser,int ok)38 test_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 
77 static int version_flag = 0;
78 static int help_flag	= 0;
79 
80 static 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 
87 static void
usage(int ret)88 usage (int ret)
89 {
90     arg_printusage (args,
91 		    sizeof(args)/sizeof(*args),
92 		    NULL,
93 		    "");
94     exit (ret);
95 }
96 
97 int
main(int argc,char ** argv)98 main(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