1 /*
2  * Copyright (c) 2002 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 <err.h>
35 
36 enum { MAX_COMPONENTS = 3 };
37 
38 static struct testcase {
39     const char *input_string;
40     const char *output_string;
41     krb5_realm realm;
42     unsigned ncomponents;
43     char *comp_val[MAX_COMPONENTS];
44     int realmp;
45 } tests[] = {
46     {"", "@", "", 1, {""}, FALSE},
47     {"a", "a@", "", 1, {"a"}, FALSE},
48     {"\\n", "\\n@", "", 1, {"\n"}, FALSE},
49     {"\\ ", "\\ @", "", 1, {" "}, FALSE},
50     {"\\t", "\\t@", "", 1, {"\t"}, FALSE},
51     {"\\b", "\\b@", "", 1, {"\b"}, FALSE},
52     {"\\\\", "\\\\@", "", 1, {"\\"}, FALSE},
53     {"\\/", "\\/@", "", 1, {"/"}, FALSE},
54     {"\\@", "\\@@", "", 1, {"@"}, FALSE},
55     {"@", "@", "", 1, {""}, TRUE},
56     {"a/b", "a/b@", "", 2, {"a", "b"}, FALSE},
57     {"a/", "a/@", "", 2, {"a", ""}, FALSE},
58     {"a\\//\\/", "a\\//\\/@", "", 2, {"a/", "/"}, FALSE},
59     {"/a", "/a@", "", 2, {"", "a"}, FALSE},
60     {"\\@@\\@", "\\@@\\@", "@", 1, {"@"}, TRUE},
61     {"a/b/c", "a/b/c@", "", 3, {"a", "b", "c"}, FALSE},
62     {NULL, NULL, "", 0, { NULL }, FALSE}};
63 
64 int
65 main(int argc, char **argv)
66 {
67     struct testcase *t;
68     krb5_context context;
69     krb5_error_code ret;
70     int val = 0;
71 
72     ret = krb5_init_context (&context);
73     if (ret)
74 	errx (1, "krb5_init_context failed: %d", ret);
75 
76     /* to enable realm-less principal name above */
77 
78     krb5_set_default_realm(context, "");
79 
80     for (t = tests; t->input_string; ++t) {
81 	krb5_principal princ;
82 	int i, j;
83 	char name_buf[1024];
84 	char *s;
85 
86 	ret = krb5_parse_name(context, t->input_string, &princ);
87 	if (ret)
88 	    krb5_err (context, 1, ret, "krb5_parse_name %s",
89 		      t->input_string);
90 	if (strcmp (t->realm, princ->realm) != 0) {
91 	    printf ("wrong realm (\"%s\" should be \"%s\")"
92 		    " for \"%s\"\n",
93 		    princ->realm, t->realm,
94 		    t->input_string);
95 	    val = 1;
96 	}
97 
98 	if (t->ncomponents != princ->name.name_string.len) {
99 	    printf ("wrong number of components (%u should be %u)"
100 		    " for \"%s\"\n",
101 		    princ->name.name_string.len, t->ncomponents,
102 		    t->input_string);
103 	    val = 1;
104 	} else {
105 	    for (i = 0; i < t->ncomponents; ++i) {
106 		if (strcmp(t->comp_val[i],
107 			   princ->name.name_string.val[i]) != 0) {
108 		    printf ("bad component %d (\"%s\" should be \"%s\")"
109 			    " for \"%s\"\n",
110 			    i,
111 			    princ->name.name_string.val[i],
112 			    t->comp_val[i],
113 			    t->input_string);
114 		    val = 1;
115 		}
116 	    }
117 	}
118 	for (j = 0; j < strlen(t->output_string); ++j) {
119 	    ret = krb5_unparse_name_fixed(context, princ,
120 					  name_buf, j);
121 	    if (ret != ERANGE) {
122 		printf ("unparse_name %s with length %d should have failed\n",
123 			t->input_string, j);
124 		val = 1;
125 		break;
126 	    }
127 	}
128 	ret = krb5_unparse_name_fixed(context, princ,
129 				      name_buf, sizeof(name_buf));
130 	if (ret)
131 	    krb5_err (context, 1, ret, "krb5_unparse_name_fixed");
132 
133 	if (strcmp (t->output_string, name_buf) != 0) {
134 	    printf ("failed comparing the re-parsed"
135 		    " (\"%s\" should be \"%s\")\n",
136 		    name_buf, t->output_string);
137 	    val = 1;
138 	}
139 
140 	ret = krb5_unparse_name(context, princ, &s);
141 	if (ret)
142 	    krb5_err (context, 1, ret, "krb5_unparse_name");
143 
144 	if (strcmp (t->output_string, s) != 0) {
145 	    printf ("failed comparing the re-parsed"
146 		    " (\"%s\" should be \"%s\"\n",
147 		    s, t->output_string);
148 	    val = 1;
149 	}
150 	free(s);
151 
152 	if (!t->realmp) {
153 	    for (j = 0; j < strlen(t->input_string); ++j) {
154 		ret = krb5_unparse_name_fixed_short(context, princ,
155 						    name_buf, j);
156 		if (ret != ERANGE) {
157 		    printf ("unparse_name_short %s with length %d"
158 			    " should have failed\n",
159 			    t->input_string, j);
160 		    val = 1;
161 		    break;
162 		}
163 	    }
164 	    ret = krb5_unparse_name_fixed_short(context, princ,
165 						name_buf, sizeof(name_buf));
166 	    if (ret)
167 		krb5_err (context, 1, ret, "krb5_unparse_name_fixed");
168 
169 	    if (strcmp (t->input_string, name_buf) != 0) {
170 		printf ("failed comparing the re-parsed"
171 			" (\"%s\" should be \"%s\")\n",
172 			name_buf, t->input_string);
173 		val = 1;
174 	    }
175 
176 	    ret = krb5_unparse_name_short(context, princ, &s);
177 	    if (ret)
178 		krb5_err (context, 1, ret, "krb5_unparse_name_short");
179 
180 	    if (strcmp (t->input_string, s) != 0) {
181 		printf ("failed comparing the re-parsed"
182 			" (\"%s\" should be \"%s\"\n",
183 			s, t->input_string);
184 		val = 1;
185 	    }
186 	    free(s);
187 	}
188 	krb5_free_principal (context, princ);
189     }
190     krb5_free_context(context);
191     return val;
192 }
193