1 /* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */ 2 /* lib/krad/t_attrset.c - Attribute set test program */ 3 /* 4 * Copyright 2013 Red Hat, Inc. 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 are met: 8 * 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in 14 * the documentation and/or other materials provided with the 15 * distribution. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 18 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 19 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 20 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 21 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 22 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 23 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 24 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 25 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 26 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 27 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30 #include "t_test.h" 31 32 const static unsigned char auth[] = { 33 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 34 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 35 }; 36 37 const static unsigned char encpass[] = { 38 0x58, 0x8d, 0xff, 0xda, 0x37, 0xf9, 0xe4, 0xca, 39 0x19, 0xae, 0x49, 0xb7, 0x16, 0x6d, 0x58, 0x27 40 }; 41 42 int 43 main() 44 { 45 unsigned char buffer[KRAD_PACKET_SIZE_MAX], encoded[MAX_ATTRSETSIZE]; 46 const char *username = "testUser", *password = "accept"; 47 const krb5_data *tmpp; 48 krad_attrset *set; 49 krb5_context ctx; 50 size_t len = 0, encode_len; 51 krb5_data tmp; 52 53 noerror(krb5_init_context(&ctx)); 54 noerror(krad_attrset_new(ctx, &set)); 55 56 /* Add username. */ 57 tmp = string2data((char *)username); 58 noerror(krad_attrset_add(set, krad_attr_name2num("User-Name"), &tmp)); 59 60 /* Add password. */ 61 tmp = string2data((char *)password); 62 noerror(krad_attrset_add(set, krad_attr_name2num("User-Password"), &tmp)); 63 64 /* Encode attrset. */ 65 noerror(kr_attrset_encode(set, "foo", auth, buffer, &encode_len)); 66 krad_attrset_free(set); 67 68 /* Manually encode User-Name. */ 69 encoded[len + 0] = krad_attr_name2num("User-Name"); 70 encoded[len + 1] = strlen(username) + 2; 71 memcpy(encoded + len + 2, username, strlen(username)); 72 len += encoded[len + 1]; 73 74 /* Manually encode User-Password. */ 75 encoded[len + 0] = krad_attr_name2num("User-Password"); 76 encoded[len + 1] = sizeof(encpass) + 2; 77 memcpy(encoded + len + 2, encpass, sizeof(encpass)); 78 len += encoded[len + 1]; 79 80 /* Compare output. */ 81 insist(len == encode_len); 82 insist(memcmp(encoded, buffer, len) == 0); 83 84 /* Decode output. */ 85 tmp = make_data(buffer, len); 86 noerror(kr_attrset_decode(ctx, &tmp, "foo", auth, &set)); 87 88 /* Test getting an attribute. */ 89 tmp = string2data((char *)username); 90 tmpp = krad_attrset_get(set, krad_attr_name2num("User-Name"), 0); 91 insist(tmpp != NULL); 92 insist(tmpp->length == tmp.length); 93 insist(strncmp(tmpp->data, tmp.data, tmp.length) == 0); 94 95 krad_attrset_free(set); 96 krb5_free_context(ctx); 97 return 0; 98 } 99