xref: /freebsd/crypto/heimdal/lib/krb5/test_store.c (revision aa0a1e58)
1 /*
2  * Copyright (c) 2006 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 
36 RCSID("$Id: test_store.c 20192 2007-02-05 23:21:03Z lha $");
37 
38 static void
39 test_int8(krb5_context context, krb5_storage *sp)
40 {
41     krb5_error_code ret;
42     int i;
43     int8_t val[] = {
44 	0, 1, -1, 128, -127
45     }, v;
46 
47     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
48 
49 	ret = krb5_store_int8(sp, val[i]);
50 	if (ret)
51 	    krb5_err(context, 1, ret, "krb5_store_int8");
52 	krb5_storage_seek(sp, 0, SEEK_SET);
53 	ret = krb5_ret_int8(sp, &v);
54 	if (ret)
55 	    krb5_err(context, 1, ret, "krb5_ret_int8");
56 	if (v != val[i])
57 	    krb5_errx(context, 1, "store and ret mismatch");
58     }
59 }
60 
61 static void
62 test_int16(krb5_context context, krb5_storage *sp)
63 {
64     krb5_error_code ret;
65     int i;
66     int16_t val[] = {
67 	0, 1, -1, 32768, -32767
68     }, v;
69 
70     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
71 
72 	ret = krb5_store_int16(sp, val[i]);
73 	if (ret)
74 	    krb5_err(context, 1, ret, "krb5_store_int16");
75 	krb5_storage_seek(sp, 0, SEEK_SET);
76 	ret = krb5_ret_int16(sp, &v);
77 	if (ret)
78 	    krb5_err(context, 1, ret, "krb5_ret_int16");
79 	if (v != val[i])
80 	    krb5_errx(context, 1, "store and ret mismatch");
81     }
82 }
83 
84 static void
85 test_int32(krb5_context context, krb5_storage *sp)
86 {
87     krb5_error_code ret;
88     int i;
89     int32_t val[] = {
90 	0, 1, -1, 2147483647, -2147483646
91     }, v;
92 
93     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
94 
95 	ret = krb5_store_int32(sp, val[i]);
96 	if (ret)
97 	    krb5_err(context, 1, ret, "krb5_store_int32");
98 	krb5_storage_seek(sp, 0, SEEK_SET);
99 	ret = krb5_ret_int32(sp, &v);
100 	if (ret)
101 	    krb5_err(context, 1, ret, "krb5_ret_int32");
102 	if (v != val[i])
103 	    krb5_errx(context, 1, "store and ret mismatch");
104     }
105 }
106 
107 static void
108 test_uint8(krb5_context context, krb5_storage *sp)
109 {
110     krb5_error_code ret;
111     int i;
112     uint8_t val[] = {
113 	0, 1, 255
114     }, v;
115 
116     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
117 
118 	ret = krb5_store_uint8(sp, val[i]);
119 	if (ret)
120 	    krb5_err(context, 1, ret, "krb5_store_uint8");
121 	krb5_storage_seek(sp, 0, SEEK_SET);
122 	ret = krb5_ret_uint8(sp, &v);
123 	if (ret)
124 	    krb5_err(context, 1, ret, "krb5_ret_uint8");
125 	if (v != val[i])
126 	    krb5_errx(context, 1, "store and ret mismatch");
127     }
128 }
129 
130 static void
131 test_uint16(krb5_context context, krb5_storage *sp)
132 {
133     krb5_error_code ret;
134     int i;
135     uint16_t val[] = {
136 	0, 1, 65535
137     }, v;
138 
139     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
140 
141 	ret = krb5_store_uint16(sp, val[i]);
142 	if (ret)
143 	    krb5_err(context, 1, ret, "krb5_store_uint16");
144 	krb5_storage_seek(sp, 0, SEEK_SET);
145 	ret = krb5_ret_uint16(sp, &v);
146 	if (ret)
147 	    krb5_err(context, 1, ret, "krb5_ret_uint16");
148 	if (v != val[i])
149 	    krb5_errx(context, 1, "store and ret mismatch");
150     }
151 }
152 
153 static void
154 test_uint32(krb5_context context, krb5_storage *sp)
155 {
156     krb5_error_code ret;
157     int i;
158     uint32_t val[] = {
159 	0, 1, 4294967295UL
160     }, v;
161 
162     for (i = 0; i < sizeof(val[0])/sizeof(val); i++) {
163 
164 	ret = krb5_store_uint32(sp, val[i]);
165 	if (ret)
166 	    krb5_err(context, 1, ret, "krb5_store_uint32");
167 	krb5_storage_seek(sp, 0, SEEK_SET);
168 	ret = krb5_ret_uint32(sp, &v);
169 	if (ret)
170 	    krb5_err(context, 1, ret, "krb5_ret_uint32");
171 	if (v != val[i])
172 	    krb5_errx(context, 1, "store and ret mismatch");
173     }
174 }
175 
176 
177 static void
178 test_storage(krb5_context context)
179 {
180     krb5_storage *sp;
181 
182     sp = krb5_storage_emem();
183     if (sp == NULL)
184 	krb5_errx(context, 1, "krb5_storage_emem: no mem");
185 
186     test_int8(context, sp);
187     test_int16(context, sp);
188     test_int32(context, sp);
189     test_uint8(context, sp);
190     test_uint16(context, sp);
191     test_uint32(context, sp);
192 
193     krb5_storage_free(sp);
194 }
195 
196 /*
197  *
198  */
199 
200 static int version_flag = 0;
201 static int help_flag	= 0;
202 
203 static struct getargs args[] = {
204     {"version",	0,	arg_flag,	&version_flag,
205      "print version", NULL },
206     {"help",	0,	arg_flag,	&help_flag,
207      NULL, NULL }
208 };
209 
210 static void
211 usage (int ret)
212 {
213     arg_printusage (args,
214 		    sizeof(args)/sizeof(*args),
215 		    NULL,
216 		    "");
217     exit (ret);
218 }
219 
220 int
221 main(int argc, char **argv)
222 {
223     krb5_context context;
224     krb5_error_code ret;
225     int optidx = 0;
226 
227     setprogname(argv[0]);
228 
229     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
230 	usage(1);
231 
232     if (help_flag)
233 	usage (0);
234 
235     if(version_flag){
236 	print_version(NULL);
237 	exit(0);
238     }
239 
240     argc -= optidx;
241     argv += optidx;
242 
243     ret = krb5_init_context (&context);
244     if (ret)
245 	errx (1, "krb5_init_context failed: %d", ret);
246 
247     test_storage(context);
248 
249     krb5_free_context(context);
250 
251     return 0;
252 }
253