unit/test-prf-sha1.c

   1 /*
   2  *
   3  *  Connection Manager
   4  *
   5  *  Copyright (C) 2007-2012  Intel Corporation. All rights reserved.
   6  *
   7  *  This program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License version 2 as
   9  *  published by the Free Software Foundation.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <stdio.h>
  27 #include <string.h>
  28
  29 #include <glib.h>
  30
  31 #include "src/shared/sha1.h"
  32
  33 struct prf_data {
  34         const char *key;
  35         unsigned int key_len;
  36         const char *prefix;
  37         unsigned int prefix_len;
  38         const char *data;
  39         unsigned int data_len;
  40         const char *prf;
  41 };
  42
  43 static void prf_test(gconstpointer data)
  44 {
  45         const struct prf_data *test = data;
  46         unsigned int prf_len;
  47         unsigned char output[512];
  48         char prf[128];
  49         unsigned int i;
  50         int result;
  51
  52         prf_len = strlen(test->prf) / 2;
  53
  54         if (g_test_verbose())
  55                 g_print("PRF    = %s (%d octects)\n", test->prf, prf_len);
  56
  57         result = prf_sha1(test->key, test->key_len, test->prefix,
  58                                 test->prefix_len, test->data, test->data_len,
  59                                                 output, prf_len);
  60
  61         g_assert(result == 0);
  62
  63         for (i = 0; i < prf_len; i++)
  64                 sprintf(prf + (i * 2), "%02x", output[i]);
  65
  66         if (g_test_verbose())
  67                 g_print("Result = %s\n", prf);
  68
  69         g_assert(strcmp(test->prf, prf) == 0);
  70 }
  71
  72 static const struct prf_data test_case_1 = {
  73         .key            = "\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
  74                           "\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b",
  75         .key_len        = 20,
  76         .prefix         = "prefix",
  77         .prefix_len     = 6,
  78         .data           = "Hi There",
  79         .data_len       = 8,
  80         .prf            = "bcd4c650b30b9684951829e0d75f9d54"
  81                           "b862175ed9f00606e17d8da35402ffee"
  82                           "75df78c3d31e0f889f012120c0862beb"
  83                           "67753e7439ae242edb8373698356cf5a",
  84 };
  85
  86 static const struct prf_data test_case_2 = {
  87         .key            = "Jefe",
  88         .key_len        = 4,
  89         .prefix         = "prefix",
  90         .prefix_len     = 6,
  91         .data           = "what do ya want for nothing?",
  92         .data_len       = 28,
  93         .prf            = "51f4de5b33f249adf81aeb713a3c20f4"
  94                           "fe631446fabdfa58244759ae58ef9009"
  95                           "a99abf4eac2ca5fa87e692c440eb4002"
  96                           "3e7babb206d61de7b92f41529092b8fc",
  97 };
  98
  99 static const struct prf_data test_case_3 = {
 100         .key            = "\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa"
 101                           "\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa",
 102         .key_len        = 20,
 103         .prefix         = "prefix",
 104         .prefix_len     = 6,
 105         .data           = "\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd"
 106                           "\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd"
 107                           "\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd"
 108                           "\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd"
 109                           "\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd\xdd",
 110         .data_len       = 50,
 111         .prf            = "e1ac546ec4cb636f9976487be5c86be1"
 112                           "7a0252ca5d8d8df12cfb0473525249ce"
 113                           "9dd8d177ead710bc9b590547239107ae"
 114                           "f7b4abd43d87f0a68f1cbd9e2b6f7607",
 115 };
 116
 117 int main(int argc, char *argv[])
 118 {
 119         g_test_init(&argc, &argv, NULL);
 120
 121         g_test_add_data_func("/prf-sha1/Test case 1",
 122                                         &test_case_1, prf_test);
 123         g_test_add_data_func("/prf-sha1/Test case 2",
 124                                         &test_case_2, prf_test);
 125         g_test_add_data_func("/prf-sha1/Test case 3",
 126                                         &test_case_3, prf_test);
 127
 128         return g_test_run();
 129 }