lib/krb5/test_addr.c

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