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 34#include "gsskrb5_locl.h" 35 36struct range { 37 size_t lower; 38 size_t upper; 39}; 40 41struct range tests[] = { 42 { 0, 1040 }, 43 { 2040, 2080 }, 44 { 4080, 5000 }, 45 { 8180, 8292 }, 46 { 9980, 10010 } 47}; 48 49static void 50test_range(const struct range *r, int integ, 51 krb5_context context, krb5_crypto crypto) 52{ 53 krb5_error_code ret; 54 size_t size, rsize; 55 struct gsskrb5_crypto ctx; 56 57 for (size = r->lower; size < r->upper; size++) { 58 size_t cksumsize; 59 uint16_t padsize; 60 OM_uint32 minor; 61 OM_uint32 max_wrap_size; 62 63 ctx.crypto = crypto; 64 65 ret = _gssapi_wrap_size_cfx(&minor, 66 &ctx, 67 context, 68 integ, 69 0, 70 size, 71 &max_wrap_size); 72 if (ret) 73 krb5_errx(context, 1, "_gsskrb5cfx_max_wrap_length_cfx: %d", ret); 74 if (max_wrap_size == 0) 75 continue; 76 77 ret = _gsskrb5cfx_wrap_length_cfx(context, 78 crypto, 79 integ, 80 max_wrap_size, 81 &rsize, &cksumsize, &padsize); 82 if (ret) 83 krb5_errx(context, 1, "_gsskrb5cfx_wrap_length_cfx: %d", ret); 84 85 if (size < rsize) 86 krb5_errx(context, 1, 87 "size (%d) < rsize (%d) for max_wrap_size %d", 88 (int)size, (int)rsize, (int)max_wrap_size); 89 } 90} 91 92static void 93test_special(krb5_context context, krb5_crypto crypto, 94 int integ, size_t testsize) 95{ 96 krb5_error_code ret; 97 size_t rsize; 98 OM_uint32 max_wrap_size; 99 size_t cksumsize; 100 uint16_t padsize; 101 struct gsskrb5_crypto ctx; 102 OM_uint32 minor; 103 104 ctx.crypto = crypto; 105 106 ret = _gssapi_wrap_size_cfx(&minor, 107 &ctx, 108 context, 109 integ, 110 0, 111 testsize, 112 &max_wrap_size); 113 if (ret) 114 krb5_errx(context, 1, "_gsskrb5cfx_max_wrap_length_cfx: %d", ret); 115 if (ret) 116 krb5_errx(context, 1, "_gsskrb5cfx_max_wrap_length_cfx: %d", ret); 117 118 ret = _gsskrb5cfx_wrap_length_cfx(context, 119 crypto, 120 integ, 121 max_wrap_size, 122 &rsize, &cksumsize, &padsize); 123 if (ret) 124 krb5_errx(context, 1, "_gsskrb5cfx_wrap_length_cfx: %d", ret); 125 126 if (testsize < rsize) 127 krb5_errx(context, 1, 128 "testsize (%d) < rsize (%d) for max_wrap_size %d", 129 (int)testsize, (int)rsize, (int)max_wrap_size); 130} 131 132 133 134 135int 136main(int argc, char **argv) 137{ 138 krb5_keyblock keyblock; 139 krb5_error_code ret; 140 krb5_context context; 141 krb5_crypto crypto; 142 int i; 143 144 ret = krb5_init_context(&context); 145 if (ret) 146 errx(1, "krb5_context_init: %d", ret); 147 148 ret = krb5_generate_random_keyblock(context, 149 KRB5_ENCTYPE_AES256_CTS_HMAC_SHA1_96, 150 &keyblock); 151 if (ret) 152 krb5_err(context, 1, ret, "krb5_generate_random_keyblock"); 153 154 ret = krb5_crypto_init(context, &keyblock, 0, &crypto); 155 if (ret) 156 krb5_err(context, 1, ret, "krb5_crypto_init"); 157 158 test_special(context, crypto, 1, 60); 159 test_special(context, crypto, 0, 60); 160 161 for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { 162 test_range(&tests[i], 1, context, crypto); 163 test_range(&tests[i], 0, context, crypto); 164 } 165 166 krb5_free_keyblock_contents(context, &keyblock); 167 krb5_crypto_destroy(context, crypto); 168 krb5_free_context(context); 169 170 return 0; 171} 172