crypto/dist/heimdal/lib/hcrypto/test_bn.c

   1 /*
   2  * Copyright (c) 2006 - 2007 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 the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #ifdef HAVE_CONFIG_H
  35 #include <config.h>
  36 #endif
  37
  38 #ifdef RCSID
  39 __RCSID("$Heimdal: test_bn.c 21047 2007-06-10 19:11:54Z lha $"
  40         "$NetBSD$");
  41 #endif
  42
  43 #include <sys/types.h>
  44 #include <limits.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <string.h>
  48
  49 #include <bn.h>
  50 #include <rand.h>
  51
  52 static int
  53 set_get(unsigned long num)
  54 {
  55     BIGNUM *bn;
  56
  57     bn = BN_new();
  58     if (!BN_set_word(bn, num))
  59         return 1;
  60
  61     if (BN_get_word(bn) != num)
  62         return 1;
  63
  64     BN_free(bn);
  65     return 0;
  66 }
  67
  68 #define CHECK(x) do { ret += x; } while(0)
  69
  70 static int
  71 test_BN_set_get(void)
  72 {
  73     int ret = 0;
  74     CHECK(set_get(0));
  75     CHECK(set_get(1));
  76     CHECK(set_get(0xff));
  77     CHECK(set_get(0x1ff));
  78     CHECK(set_get(0xffff));
  79     CHECK(set_get(0xf000));
  80     CHECK(set_get(ULONG_MAX / 2));
  81     CHECK(set_get(ULONG_MAX - 1));
  82
  83     return ret;
  84 }
  85
  86 static int
  87 test_BN_bit(void)
  88 {
  89     BIGNUM *bn;
  90     int ret = 0;
  91
  92     bn = BN_new();
  93
  94     /* test setting and getting of "word" */
  95     if (!BN_set_word(bn, 1))
  96         return 1;
  97     if (!BN_is_bit_set(bn, 0))
  98         ret += 1;
  99     if (!BN_is_bit_set(bn, 0))
 100         ret += 1;
 101
 102     if (!BN_set_word(bn, 2))
 103         return 1;
 104     if (!BN_is_bit_set(bn, 1))
 105         ret += 1;
 106
 107     if (!BN_set_word(bn, 3))
 108         return 1;
 109     if (!BN_is_bit_set(bn, 0))
 110         ret += 1;
 111     if (!BN_is_bit_set(bn, 1))
 112         ret += 1;
 113
 114     if (!BN_set_word(bn, 0x100))
 115         return 1;
 116     if (!BN_is_bit_set(bn, 8))
 117         ret += 1;
 118
 119     if (!BN_set_word(bn, 0x1000))
 120         return 1;
 121     if (!BN_is_bit_set(bn, 12))
 122         ret += 1;
 123
 124     /* test bitsetting */
 125     if (!BN_set_word(bn, 1))
 126         return 1;
 127     if (!BN_set_bit(bn, 1))
 128         return 1;
 129     if (BN_get_word(bn) != 3)
 130         return 1;
 131     if (!BN_clear_bit(bn, 0))
 132         return 1;
 133     if (BN_get_word(bn) != 2)
 134         return 1;
 135
 136     /* test bitsetting past end of current end */
 137     BN_clear(bn);
 138     if (!BN_set_bit(bn, 12))
 139         return 1;
 140     if (BN_get_word(bn) != 0x1000)
 141         return 1;
 142
 143     /* test bit and byte counting functions */
 144     if (BN_num_bits(bn) != 13)
 145         return 1;
 146     if (BN_num_bytes(bn) != 2)
 147         return 1;
 148
 149     BN_free(bn);
 150     return ret;
 151 }
 152
 153 struct ietest {
 154     char *data;
 155     size_t len;
 156     unsigned long num;
 157 } ietests[] = {
 158     { "", 0, 0 },
 159     { "\x01", 1, 1 },
 160     { "\x02", 1, 2 },
 161     { "\xf2", 1, 0xf2 },
 162     { "\x01\x00", 2, 256 }
 163 };
 164
 165 static int
 166 test_BN_import_export(void)
 167 {
 168     BIGNUM *bn;
 169     int ret = 0;
 170     int i;
 171
 172     bn = BN_new();
 173
 174     for (i = 0; i < sizeof(ietests)/sizeof(ietests[0]); i++) {
 175         size_t len;
 176         unsigned char *p;
 177         if (!BN_bin2bn((unsigned char*)ietests[i].data, ietests[i].len, bn))
 178             return 1;
 179         if (BN_get_word(bn) != ietests[i].num)
 180             return 1;
 181         len = BN_num_bytes(bn);
 182         if (len != ietests[i].len)
 183             return 1;
 184         p = malloc(len + 1);
 185         p[len] = 0xf4;
 186         BN_bn2bin(bn, p);
 187         if (p[len] != 0xf4)
 188             return 1;
 189         if (memcmp(p, ietests[i].data, ietests[i].len) != 0)
 190             return 1;
 191         free(p);
 192     }
 193
 194     BN_free(bn);
 195     return ret;
 196 }
 197
 198 static int
 199 test_BN_uadd(void)
 200 {
 201     BIGNUM *a, *b, *c;
 202     char *p;
 203
 204     a = BN_new();
 205     b = BN_new();
 206     c = BN_new();
 207
 208     BN_set_word(a, 1);
 209     BN_set_word(b, 2);
 210
 211     BN_uadd(c, a, b);
 212
 213     if (BN_get_word(c) != 3)
 214         return 1;
 215
 216     BN_uadd(c, b, a);
 217
 218     if (BN_get_word(c) != 3)
 219         return 1;
 220
 221     BN_set_word(b, 0xff);
 222
 223     BN_uadd(c, a, b);
 224     if (BN_get_word(c) != 0x100)
 225         return 1;
 226
 227     BN_uadd(c, b, a);
 228     if (BN_get_word(c) != 0x100)
 229         return 1;
 230
 231     BN_set_word(a, 0xff);
 232
 233     BN_uadd(c, a, b);
 234     if (BN_get_word(c) != 0x1fe)
 235         return 1;
 236
 237     BN_uadd(c, b, a);
 238     if (BN_get_word(c) != 0x1fe)
 239         return 1;
 240
 241
 242     BN_free(a);
 243     BN_free(b);
 244
 245     BN_hex2bn(&a, "50212A3B611D46642C825A16A354CE0FD4D85DD2");
 246     BN_hex2bn(&b, "84B6C7E8D28ACA1614954DA");
 247
 248     BN_uadd(c, b, a);
 249     p = BN_bn2hex(c);
 250     if (strcmp(p, "50212A3B611D466434CDC695307D7AB13621B2AC") != 0) {
 251         free(p);
 252         return 1;
 253     }
 254     free(p);
 255
 256     BN_uadd(c, a, b);
 257     p = BN_bn2hex(c);
 258     if (strcmp(p, "50212A3B611D466434CDC695307D7AB13621B2AC") != 0) {
 259         free(p);
 260         return 1;
 261     }
 262     free(p);
 263
 264     BN_free(a);
 265     BN_free(b);
 266     BN_free(c);
 267
 268     return 0;
 269 }
 270
 271 static int
 272 test_BN_cmp(void)
 273 {
 274     BIGNUM *a, *b;
 275
 276     a = BN_new();
 277     b = BN_new();
 278
 279     if (!BN_set_word(a, 1))
 280         return 1;
 281     if (!BN_set_word(b, 1))
 282         return 1;
 283
 284     if (BN_cmp(a, b) != 0)
 285         return 1;
 286     if (BN_cmp(b, a) != 0)
 287         return 1;
 288
 289     if (!BN_set_word(b, 2))
 290         return 1;
 291
 292     if (BN_cmp(a, b) >= 0)
 293         return 1;
 294     if (BN_cmp(b, a) <= 0)
 295         return 1;
 296
 297     BN_set_negative(b, 1);
 298
 299     if (BN_cmp(a, b) <= 0)
 300         return 1;
 301     if (BN_cmp(b, a) >= 0)
 302         return 1;
 303
 304     BN_free(a);
 305     BN_free(b);
 306
 307     BN_hex2bn(&a, "50212A3B611D46642C825A16A354CE0FD4D85DD1");
 308     BN_hex2bn(&b, "50212A3B611D46642C825A16A354CE0FD4D85DD2");
 309
 310     if (BN_cmp(a, b) >= 0)
 311         return 1;
 312     if (BN_cmp(b, a) <= 0)
 313         return 1;
 314
 315     BN_set_negative(b, 1);
 316
 317     if (BN_cmp(a, b) <= 0)
 318         return 1;
 319     if (BN_cmp(b, a) >= 0)
 320         return 1;
 321
 322     BN_free(a);
 323     BN_free(b);
 324     return 0;
 325 }
 326
 327 static int
 328 test_BN_rand(void)
 329 {
 330     BIGNUM *bn;
 331
 332     if (RAND_status() != 1)
 333         return 0;
 334
 335     bn = BN_new();
 336     if (bn == NULL)
 337         return 1;
 338
 339     if (!BN_rand(bn, 1024, 0, 0))
 340         return 1;
 341
 342     BN_free(bn);
 343     return 0;
 344 }
 345
 346 int
 347 main(int argc, char **argv)
 348 {
 349     int ret = 0;
 350
 351     ret += test_BN_set_get();
 352     ret += test_BN_bit();
 353     ret += test_BN_import_export();
 354     ret += test_BN_uadd();
 355     ret += test_BN_cmp();
 356     ret += test_BN_rand();
 357
 358     return ret;
 359 }