Home | History | Annotate | Line # | Download | only in krb5
test_addr.c revision 1.1.1.1.28.1
      1 /*	$NetBSD: test_addr.c,v 1.1.1.1.28.1 2017/03/20 06:51:47 pgoyette Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2005 Kungliga Tekniska Hgskolan
      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 }
    242