test_crypto_wrapping.c revision 178825
1178825Sdfr/* 2178825Sdfr * Copyright (c) 2005 Kungliga Tekniska H�gskolan 3178825Sdfr * (Royal Institute of Technology, Stockholm, Sweden). 4178825Sdfr * All rights reserved. 5178825Sdfr * 6178825Sdfr * Redistribution and use in source and binary forms, with or without 7178825Sdfr * modification, are permitted provided that the following conditions 8178825Sdfr * are met: 9178825Sdfr * 10178825Sdfr * 1. Redistributions of source code must retain the above copyright 11178825Sdfr * notice, this list of conditions and the following disclaimer. 12178825Sdfr * 13178825Sdfr * 2. Redistributions in binary form must reproduce the above copyright 14178825Sdfr * notice, this list of conditions and the following disclaimer in the 15178825Sdfr * documentation and/or other materials provided with the distribution. 16178825Sdfr * 17178825Sdfr * 3. Neither the name of KTH nor the names of its contributors may be 18178825Sdfr * used to endorse or promote products derived from this software without 19178825Sdfr * specific prior written permission. 20178825Sdfr * 21178825Sdfr * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY 22178825Sdfr * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23178825Sdfr * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 24178825Sdfr * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE 25178825Sdfr * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 26178825Sdfr * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 27178825Sdfr * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 28178825Sdfr * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 29178825Sdfr * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 30178825Sdfr * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 31178825Sdfr * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ 32178825Sdfr 33178825Sdfr#include "krb5_locl.h" 34178825Sdfr#include <err.h> 35178825Sdfr#include <getarg.h> 36178825Sdfr 37178825SdfrRCSID("$Id: test_crypto_wrapping.c 18809 2006-10-22 07:11:43Z lha $"); 38178825Sdfr 39178825Sdfrstatic void 40178825Sdfrtest_wrapping(krb5_context context, 41178825Sdfr size_t min_size, 42178825Sdfr size_t max_size, 43178825Sdfr size_t step, 44178825Sdfr krb5_enctype etype) 45178825Sdfr{ 46178825Sdfr krb5_error_code ret; 47178825Sdfr krb5_keyblock key; 48178825Sdfr krb5_crypto crypto; 49178825Sdfr krb5_data data; 50178825Sdfr char *etype_name; 51178825Sdfr void *buf; 52178825Sdfr size_t size; 53178825Sdfr 54178825Sdfr ret = krb5_generate_random_keyblock(context, etype, &key); 55178825Sdfr if (ret) 56178825Sdfr krb5_err(context, 1, ret, "krb5_generate_random_keyblock"); 57178825Sdfr 58178825Sdfr ret = krb5_enctype_to_string(context, etype, &etype_name); 59178825Sdfr if (ret) 60178825Sdfr krb5_err(context, 1, ret, "krb5_enctype_to_string"); 61178825Sdfr 62178825Sdfr buf = malloc(max_size); 63178825Sdfr if (buf == NULL) 64178825Sdfr krb5_errx(context, 1, "out of memory"); 65178825Sdfr memset(buf, 0, max_size); 66178825Sdfr 67178825Sdfr ret = krb5_crypto_init(context, &key, 0, &crypto); 68178825Sdfr if (ret) 69178825Sdfr krb5_err(context, 1, ret, "krb5_crypto_init"); 70178825Sdfr 71178825Sdfr for (size = min_size; size < max_size; size += step) { 72178825Sdfr size_t wrapped_size; 73178825Sdfr 74178825Sdfr ret = krb5_encrypt(context, crypto, 0, buf, size, &data); 75178825Sdfr if (ret) 76178825Sdfr krb5_err(context, 1, ret, "encrypt size %lu using %s", 77178825Sdfr (unsigned long)size, etype_name); 78178825Sdfr 79178825Sdfr wrapped_size = krb5_get_wrapped_length(context, crypto, size); 80178825Sdfr 81178825Sdfr if (wrapped_size != data.length) 82178825Sdfr krb5_errx(context, 1, "calculated wrapped length %lu != " 83178825Sdfr "real wrapped length %lu for data length %lu using " 84178825Sdfr "enctype %s", 85178825Sdfr (unsigned long)wrapped_size, 86178825Sdfr (unsigned long)data.length, 87178825Sdfr (unsigned long)size, 88178825Sdfr etype_name); 89178825Sdfr krb5_data_free(&data); 90178825Sdfr } 91178825Sdfr 92178825Sdfr free(etype_name); 93178825Sdfr free(buf); 94178825Sdfr krb5_crypto_destroy(context, crypto); 95178825Sdfr krb5_free_keyblock_contents(context, &key); 96178825Sdfr} 97178825Sdfr 98178825Sdfr 99178825Sdfr 100178825Sdfrstatic int version_flag = 0; 101178825Sdfrstatic int help_flag = 0; 102178825Sdfr 103178825Sdfrstatic struct getargs args[] = { 104178825Sdfr {"version", 0, arg_flag, &version_flag, 105178825Sdfr "print version", NULL }, 106178825Sdfr {"help", 0, arg_flag, &help_flag, 107178825Sdfr NULL, NULL } 108178825Sdfr}; 109178825Sdfr 110178825Sdfrstatic void 111178825Sdfrusage (int ret) 112178825Sdfr{ 113178825Sdfr arg_printusage (args, 114178825Sdfr sizeof(args)/sizeof(*args), 115178825Sdfr NULL, 116178825Sdfr ""); 117178825Sdfr exit (ret); 118178825Sdfr} 119178825Sdfr 120178825Sdfrint 121178825Sdfrmain(int argc, char **argv) 122178825Sdfr{ 123178825Sdfr krb5_context context; 124178825Sdfr krb5_error_code ret; 125178825Sdfr int i, optidx = 0; 126178825Sdfr 127178825Sdfr krb5_enctype enctypes[] = { 128178825Sdfr ETYPE_DES_CBC_CRC, 129178825Sdfr ETYPE_DES_CBC_MD4, 130178825Sdfr ETYPE_DES_CBC_MD5, 131178825Sdfr ETYPE_DES3_CBC_SHA1, 132178825Sdfr ETYPE_ARCFOUR_HMAC_MD5, 133178825Sdfr ETYPE_AES128_CTS_HMAC_SHA1_96, 134178825Sdfr ETYPE_AES256_CTS_HMAC_SHA1_96 135178825Sdfr }; 136178825Sdfr 137178825Sdfr setprogname(argv[0]); 138178825Sdfr 139178825Sdfr if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx)) 140178825Sdfr usage(1); 141178825Sdfr 142178825Sdfr if (help_flag) 143178825Sdfr usage (0); 144178825Sdfr 145178825Sdfr if(version_flag){ 146178825Sdfr print_version(NULL); 147178825Sdfr exit(0); 148178825Sdfr } 149178825Sdfr 150178825Sdfr argc -= optidx; 151178825Sdfr argv += optidx; 152178825Sdfr 153178825Sdfr ret = krb5_init_context(&context); 154178825Sdfr if (ret) 155178825Sdfr errx (1, "krb5_init_context failed: %d", ret); 156178825Sdfr 157178825Sdfr for (i = 0; i < sizeof(enctypes)/sizeof(enctypes[0]); i++) { 158178825Sdfr test_wrapping(context, 0, 1024, 1, enctypes[i]); 159178825Sdfr test_wrapping(context, 1024, 1024 * 100, 1024, enctypes[i]); 160178825Sdfr } 161178825Sdfr krb5_free_context(context); 162178825Sdfr 163178825Sdfr return 0; 164178825Sdfr} 165