crypto/external/bsd/heimdal/dist/lib/krb5/test_addr.c

   1 /*      $NetBSD: test_addr.c,v 1.1.1.1 2011/04/13 18:15:38 elric Exp $  */
   2
   3 /*
   4  * Copyright (c) 2005 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 <err.h>
  37
  38 static void
  39 print_addr(krb5_context context, const char *addr)
  40 {
  41     krb5_addresses addresses;
  42     krb5_error_code ret;
  43     char buf[38];
  44     char buf2[1000];
  45     size_t len;
  46     int i;
  47
  48     ret = krb5_parse_address(context, addr, &addresses);
  49     if (ret)
  50         krb5_err(context, 1, ret, "krb5_parse_address");
  51
  52     if (addresses.len < 1)
  53         krb5_err(context, 1, ret, "too few addresses");
  54
  55     for (i = 0; i < addresses.len; i++) {
  56         krb5_print_address(&addresses.val[i], buf, sizeof(buf), &len);
  57 #if 0
  58         printf("addr %d: %s (%d/%d)\n", i, buf, (int)len, (int)strlen(buf));
  59 #endif
  60         if (strlen(buf) > sizeof(buf))
  61             krb5_err(context, 1, ret, "len %d larger then buf %d",
  62                      (int)strlen(buf), (int)sizeof(buf));
  63         krb5_print_address(&addresses.val[i], buf2, sizeof(buf2), &len);
  64 #if 0
  65         printf("addr %d: %s (%d/%d)\n", i, buf2, (int)len, (int)strlen(buf2));
  66 #endif
  67         if (strlen(buf2) > sizeof(buf2))
  68             krb5_err(context, 1, ret, "len %d larger then buf %d",
  69                      (int)strlen(buf2), (int)sizeof(buf2));
  70
  71     }
  72     krb5_free_addresses(context, &addresses);
  73
  74 }
  75
  76 static void
  77 truncated_addr(krb5_context context, const char *addr,
  78                size_t truncate_len, size_t outlen)
  79 {
  80     krb5_addresses addresses;
  81     krb5_error_code ret;
  82     char *buf;
  83     size_t len;
  84
  85     buf = ecalloc(1, outlen + 1);
  86
  87     ret = krb5_parse_address(context, addr, &addresses);
  88     if (ret)
  89         krb5_err(context, 1, ret, "krb5_parse_address");
  90
  91     if (addresses.len != 1)
  92         krb5_err(context, 1, ret, "addresses should be one");
  93
  94     krb5_print_address(&addresses.val[0], buf, truncate_len, &len);
  95
  96 #if 0
  97     printf("addr %s (%d/%d) should be %d\n", buf, (int)len, (int)strlen(buf), (int)outlen);
  98 #endif
  99
 100     if (truncate_len > strlen(buf) + 1)
 101         krb5_err(context, 1, ret, "%s truncate_len %d larger then strlen %d source %s",
 102                  buf, (int)truncate_len, (int)strlen(buf), addr);
 103
 104     if (outlen != len)
 105         krb5_err(context, 1, ret, "%s: outlen %d != len %d",
 106                  buf, (int)outlen, (int)strlen(buf));
 107
 108     krb5_print_address(&addresses.val[0], buf, outlen + 1, &len);
 109
 110 #if 0
 111     printf("addr %s (%d/%d)\n", buf, (int)len, (int)strlen(buf));
 112 #endif
 113
 114     if (len != outlen)
 115         abort();
 116     if (strlen(buf) != len)
 117         abort();
 118
 119     krb5_free_addresses(context, &addresses);
 120     free(buf);
 121 }
 122
 123 static void
 124 check_truncation(krb5_context context, const char *addr)
 125 {
 126     int i, len = strlen(addr);
 127
 128     truncated_addr(context, addr, len, len);
 129
 130     for (i = 0; i < len; i++)
 131         truncated_addr(context, addr, i, len);
 132 }
 133
 134 static void
 135 match_addr(krb5_context context, const char *range_addr,
 136            const char *one_addr, int match)
 137 {
 138     krb5_addresses range, one;
 139     krb5_error_code ret;
 140
 141     ret = krb5_parse_address(context, range_addr, &range);
 142     if (ret)
 143         krb5_err(context, 1, ret, "krb5_parse_address");
 144
 145     if (range.len != 1)
 146         krb5_err(context, 1, ret, "wrong num of addresses");
 147
 148     ret = krb5_parse_address(context, one_addr, &one);
 149     if (ret)
 150         krb5_err(context, 1, ret, "krb5_parse_address");
 151
 152     if (one.len != 1)
 153         krb5_err(context, 1, ret, "wrong num of addresses");
 154
 155     if (krb5_address_order(context, &range.val[0], &one.val[0]) == 0) {
 156         if (!match)
 157             krb5_errx(context, 1, "match when one shouldn't be");
 158     } else {
 159         if (match)
 160             krb5_errx(context, 1, "no match when one should be");
 161     }
 162
 163     krb5_free_addresses(context, &range);
 164     krb5_free_addresses(context, &one);
 165 }
 166
 167 #ifdef _MSC_VER
 168
 169 /* For the truncation tests, calling strcpy_s() or strcat_s() with a
 170    size of 0 results in the invalid parameter handler being invoked.
 171    For the debug version, the runtime also throws an assert. */
 172
 173 static void
 174 inv_param_handler(const wchar_t* expression,
 175                   const wchar_t* function,
 176                   const wchar_t* file,
 177                   unsigned int line,
 178                   uintptr_t pReserved)
 179 {
 180     printf("Invalid parameter handler invoked for: %S in %S(%d) [%S]\n",
 181            function, file, line, expression);
 182 }
 183
 184 static _invalid_parameter_handler _inv_old = NULL;
 185
 186 #define SET_INVALID_PARAM_HANDLER _inv_old = _set_invalid_parameter_handler(inv_param_handler)
 187
 188 #else
 189
 190 #define SET_INVALID_PARAM_HANDLER ((void) 0)
 191
 192 #endif
 193
 194 int
 195 main(int argc, char **argv)
 196 {
 197     krb5_context context;
 198     krb5_error_code ret;
 199
 200     SET_INVALID_PARAM_HANDLER;
 201
 202     setprogname(argv[0]);
 203
 204     ret = krb5_init_context(&context);
 205     if (ret)
 206         errx (1, "krb5_init_context failed: %d", ret);
 207
 208     print_addr(context, "RANGE:127.0.0.0/8");
 209     print_addr(context, "RANGE:127.0.0.0/24");
 210     print_addr(context, "RANGE:IPv4:127.0.0.0-IPv4:127.0.0.255");
 211     print_addr(context, "RANGE:130.237.237.4/29");
 212 #ifdef HAVE_IPV6
 213     print_addr(context, "RANGE:2001:db8:1:2:3:4:1428:7ab/64");
 214     print_addr(context, "RANGE:IPv6:fe80::209:6bff:fea0:e522/64");
 215     print_addr(context, "RANGE:IPv6:fe80::-IPv6:fe80::ffff:ffff:ffff:ffff");
 216     print_addr(context, "RANGE:fe80::-fe80::ffff:ffff:ffff:ffff");
 217 #endif
 218
 219     check_truncation(context, "IPv4:127.0.0.0");
 220     check_truncation(context, "RANGE:IPv4:127.0.0.0-IPv4:127.0.0.255");
 221 #ifdef HAVE_IPV6
 222     check_truncation(context, "IPv6:::");
 223     check_truncation(context, "IPv6:::1");
 224     check_truncation(context, "IPv6:2001:db8:1:2:3:4:1428:7ab");
 225     check_truncation(context, "IPv6:fe80::209:0:0:0");
 226     check_truncation(context, "IPv6:fe80::ffff:ffff:ffff:ffff");
 227 #endif
 228
 229     match_addr(context, "RANGE:127.0.0.0/8", "inet:127.0.0.0", 1);
 230     match_addr(context, "RANGE:127.0.0.0/8", "inet:127.255.255.255", 1);
 231     match_addr(context, "RANGE:127.0.0.0/8", "inet:128.0.0.0", 0);
 232
 233     match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.7", 0);
 234     match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.8", 1);
 235     match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.15", 1);
 236     match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.16", 0);
 237
 238     krb5_free_context(context);
 239
 240     return 0;
 241 }