src/microhttpd/test_sha1.c - manifest_repos/libmicrohttpd - Git at Google

 /*
   This file is part of libmicrohttpd
   Copyright (C) 2019-2021 Karlson2k (Evgeny Grin)

   This test tool is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License as
   published by the Free Software Foundation; either version 2, or
   (at your option) any later version.

   This test tool is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

 /**
  * @file microhttpd/test_sha1.h
  * @brief  Unit tests for SHA-1 functions
  * @author Karlson2k (Evgeny Grin)
  */

 #include "mhd_options.h"
 #include "sha1.h"
 #include "test_helpers.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>

 static int verbose = 0; /* verbose level (0-1)*/


 struct str_with_len
 {
   const char *const str;
   const size_t len;
 };

 #define D_STR_W_LEN(s) {(s), (sizeof((s)) / sizeof(char)) - 1}

 struct data_unit1
 {
   const struct str_with_len str_l;
   const uint8_t digest[SHA1_DIGEST_SIZE];
 };

 static const struct data_unit1 data_units1[] = {
   {D_STR_W_LEN ("abc"),
    {0xA9, 0x99, 0x3E, 0x36, 0x47, 0x06, 0x81, 0x6A, 0xBA, 0x3E, 0x25, 0x71,
     0x78, 0x50, 0xC2, 0x6C, 0x9C, 0xD0, 0xD8, 0x9D}},
   {D_STR_W_LEN ("abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"),
    {0x84, 0x98, 0x3E, 0x44, 0x1C, 0x3B, 0xD2, 0x6E, 0xBA, 0xAE, 0x4A, 0xA1,
     0xF9, 0x51, 0x29, 0xE5, 0xE5, 0x46, 0x70, 0xF1}},
   {D_STR_W_LEN (""),
    {0xda, 0x39, 0xa3, 0xee, 0x5e, 0x6b, 0x4b, 0x0d, 0x32, 0x55, 0xbf, 0xef,
     0x95, 0x60, 0x18, 0x90, 0xaf, 0xd8, 0x07, 0x09}},
   {D_STR_W_LEN ("The quick brown fox jumps over the lazy dog"),
    {0x2f, 0xd4, 0xe1, 0xc6, 0x7a, 0x2d, 0x28, 0xfc, 0xed, 0x84, 0x9e, 0xe1,
     0xbb, 0x76, 0xe7, 0x39, 0x1b, 0x93, 0xeb, 0x12}},
   {D_STR_W_LEN ("1234567890!@~%&$@#{}[]\\/!?`."),
    {0xa7, 0x08, 0x9e, 0x3d, 0xe1, 0x6b, 0x63, 0xa3, 0x2a, 0x43, 0xa5, 0xe7,
     0xf3, 0xb5, 0x4f, 0x80, 0x6a, 0xc8, 0x4f, 0x53}},
   {D_STR_W_LEN ("Simple string."),
    {0x6c, 0x6c, 0x9c, 0xef, 0x53, 0xff, 0x22, 0x39, 0x0c, 0x54, 0xc7, 0xba,
     0x6d, 0x98, 0xe0, 0xd5, 0x6c, 0x24, 0x0d, 0x55}},
   {D_STR_W_LEN ("abcdefghijklmnopqrstuvwxyz"),
    {0x32, 0xd1, 0x0c, 0x7b, 0x8c, 0xf9, 0x65, 0x70, 0xca, 0x04, 0xce, 0x37,
     0xf2, 0xa1, 0x9d, 0x84, 0x24, 0x0d, 0x3a, 0x89}},
   {D_STR_W_LEN ("zyxwvutsrqponMLKJIHGFEDCBA"),
    {0x59, 0x0f, 0xc8, 0xea, 0xde, 0xa2, 0x78, 0x65, 0x5a, 0xf2, 0xa1, 0xe5,
     0xb4, 0xc9, 0x61, 0xa4, 0x3a, 0xc3, 0x6a, 0x83}},
   {D_STR_W_LEN ("abcdefghijklmnopqrstuvwxyzzyxwvutsrqponMLKJIHGFEDCBA"
                 "abcdefghijklmnopqrstuvwxyzzyxwvutsrqponMLKJIHGFEDCBA"),
    {0xa7, 0x87, 0x4c, 0x16, 0xc0, 0x4e, 0xd6, 0x24, 0x5f, 0x25, 0xbe, 0x06,
     0x7b, 0x1b, 0x6b, 0xaf, 0xf4, 0x0e, 0x74, 0x0b}},
 };

 static const size_t units1_num = sizeof(data_units1) / sizeof(data_units1[0]);

 struct bin_with_len
 {
   const uint8_t bin[512];
   const size_t len;
 };

 struct data_unit2
 {
   const struct bin_with_len bin_l;
   const uint8_t digest[SHA1_DIGEST_SIZE];
 };

 /* Size must be less than 512 bytes! */
 static const struct data_unit2 data_units2[] = {
   { { {97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111,
        112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122}, 26}, /* a..z ASCII sequence */
     {0x32, 0xd1, 0x0c, 0x7b, 0x8c, 0xf9, 0x65, 0x70, 0xca, 0x04, 0xce, 0x37,
      0xf2, 0xa1, 0x9d, 0x84, 0x24, 0x0d, 0x3a, 0x89}},
   { { {65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
        65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
        65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
        65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65},
       72 }, /* 'A' x 72 times */
     {0x41, 0xf9, 0x07, 0x05, 0x04, 0xf9, 0xc8, 0x1a, 0xbf, 0xbb, 0x61, 0x4d,
      0xaa, 0xec, 0x3b, 0x26, 0xa2, 0xf9, 0x23, 0x7e}},
   { { {19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,
        37, 38, 39, 40, 41, 42,43, 44, 45, 46, 47, 48, 49, 50, 51, 52,53, 54,
        55, 56, 57, 58, 59, 60,61, 62, 63, 64, 65, 66, 67,68, 69, 70, 71, 72,
        73}, 55}, /* 19..73 sequence */
     {0xf2, 0x50, 0xb2, 0x79, 0x62, 0xcc, 0xff, 0x4b, 0x1b, 0x61, 0x4a, 0x6f,
      0x80, 0xec, 0x9b, 0x4d, 0xd0, 0xc0, 0xfb, 0x7b}},
   { { {7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,
        26, 27, 28, 29, 30, 31,32, 33, 34, 35, 36, 37, 38, 39, 40, 41,42, 43,
        44, 45, 46, 47, 48, 49,50, 51, 52, 53, 54, 55, 56,57, 58, 59, 60, 61,
        62, 63, 64, 65, 66, 67, 68, 69}, 63}, /* 7..69 sequence */
     {0xf0, 0xa3, 0xfc, 0x4b, 0x90, 0x6f, 0x1b, 0x41, 0x68, 0xc8, 0x3e, 0x05,
      0xb0, 0xc5, 0xac, 0xb7, 0x3d, 0xcd, 0x6b, 0x0f}},
   { { {38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55,
        56, 57, 58, 59, 60, 61,62, 63, 64, 65, 66, 67, 68, 69, 70, 71,72, 73,
        74, 75, 76, 77, 78, 79,80, 81, 82, 83, 84, 85, 86,87, 88, 89, 90, 91,
        92}, 55},  /* 38..92 sequence */
     {0x93, 0xa4, 0x9c, 0x5f, 0xda, 0x22, 0x63, 0x73, 0x85, 0xeb, 0x70, 0xd2,
      0x00, 0x52, 0x0c, 0xb0, 0x2e, 0x86, 0x9b, 0xa0 }},
   { { {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,21,
        22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,37, 38, 39,
        40, 41, 42, 43, 44, 45,46, 47, 48, 49, 50, 51, 52,53, 54, 55, 56, 57,
        58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70,71, 72},72}, /* 1..72 sequence */
     {0x8b, 0x30, 0xd3, 0x41, 0x89, 0xb6, 0x1b, 0x66, 0x5a, 0x1a, 0x9a, 0x51,
      0x64, 0x93, 0xab, 0x5e, 0x78, 0x81, 0x52, 0xb5}},
   { { {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
        21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,37, 38,
        39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,52, 53, 54, 55, 56,
        57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69,70, 71, 72, 73, 74,
        75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85,86, 87, 88, 89, 90, 91, 92,
        93, 94, 95, 96, 97, 98, 99, 100,101, 102, 103, 104, 105, 106, 107, 108,
        109, 110, 111, 112, 113, 114,115, 116, 117, 118, 119, 120, 121, 122,
        123, 124, 125, 126, 127,128, 129, 130, 131, 132, 133, 134, 135, 136,
        137, 138, 139, 140,141, 142, 143, 144, 145, 146, 147, 148, 149, 150,
        151, 152, 153, 154,155, 156, 157, 158, 159, 160, 161, 162, 163, 164,
        165, 166, 167,168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178,
        179, 180,181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192,
        193, 194,195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206,
        207,208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220,221,
        222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234,235,
        236, 237, 238, 239, 240,241, 242, 243, 244, 245, 246, 247,248, 249, 250,
        251, 252, 253, 254, 255}, 256}, /* 0..255 sequence */
     {0x49, 0x16, 0xd6, 0xbd, 0xb7, 0xf7, 0x8e, 0x68, 0x03, 0x69, 0x8c, 0xab,
      0x32, 0xd1, 0x58, 0x6e, 0xa4, 0x57, 0xdf, 0xc8}},
   { { {199, 198, 197, 196, 195, 194, 193, 192, 191, 190, 189, 188, 187, 186,185,
        184, 183, 182, 181, 180,179, 178, 177, 176, 175, 174, 173,172, 171, 170,
        169, 168, 167, 166,165, 164, 163, 162, 161, 160,159, 158, 157, 156, 155,
        154, 153, 152, 151, 150, 149, 148, 147, 146,145, 144, 143, 142, 141,
        140,139}, 61}, /* 199..139 sequence */
     {0xb3, 0xec, 0x61, 0x3c, 0xf7, 0x36, 0x57, 0x94, 0x61, 0xdb, 0xb2, 0x16,
      0x75, 0xfe, 0x34, 0x60, 0x99, 0x94, 0xb5, 0xfc}},
   { { {255, 254, 253, 252, 251, 250, 249, 248, 247, 246, 245, 244, 243, 242,
        241, 240, 239, 238, 237, 236, 235, 234, 233, 232, 231, 230, 229, 228,
        227, 226, 225, 224, 223, 222, 221, 220, 219, 218, 217, 216, 215, 214,
        213, 212, 211, 210, 209, 208, 207, 206, 205, 204, 203, 202, 201, 200,
        199, 198, 197, 196, 195, 194, 193, 192, 191, 190, 189, 188, 187, 186,
        185, 184, 183, 182, 181, 180, 179, 178, 177, 176, 175, 174, 173, 172,
        171, 170, 169, 168, 167, 166, 165, 164, 163, 162, 161, 160, 159, 158,
        157, 156, 155, 154, 153, 152, 151, 150, 149, 148, 147, 146, 145, 144,
        143, 142, 141, 140, 139, 138, 137, 136, 135, 134, 133, 132, 131, 130,
        129, 128, 127, 126, 125, 124, 123, 122, 121, 120, 119, 118, 117, 116,
        115, 114, 113, 112, 111, 110, 109, 108, 107, 106, 105, 104, 103, 102,
        101, 100, 99, 98, 97, 96, 95, 94, 93, 92, 91, 90, 89, 88, 87, 86, 85,
        84, 83, 82, 81, 80, 79, 78, 77, 76, 75, 74, 73, 72, 71, 70, 69, 68, 67,
        66, 65, 64, 63, 62, 61, 60, 59, 58, 57, 56, 55, 54, 53, 52, 51, 50, 49,
        48, 47, 46, 45, 44, 43, 42, 41, 40, 39, 38, 37, 36, 35, 34, 33, 32, 31,
        30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14, 13,
        12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1}, 255}, /* 255..1 sequence */
     {0x99, 0x1d, 0xc6, 0x95, 0xe1, 0xaa, 0xcc, 0xc9, 0x35, 0x60, 0xbe, 0xe3,
      0xe2, 0x41, 0x2b, 0xa7, 0xab, 0x49, 0x32, 0xf7}},
   { { {41, 35, 190, 132, 225, 108, 214, 174, 82, 144, 73, 241, 241, 187, 233,
        235, 179, 166, 219, 60, 135, 12, 62, 153, 36, 94, 13, 28, 6, 183, 71,
        222, 179, 18, 77, 200, 67, 187, 139, 166, 31, 3, 90, 125, 9, 56, 37, 31,
        93, 212, 203, 252, 150, 245, 69, 59, 19, 13, 137, 10, 28, 219, 174, 50,
        32, 154, 80, 238, 64, 120, 54, 253, 18, 73, 50, 246, 158, 125, 73, 220,
        173, 79, 20, 242, 68, 64, 102, 208, 107, 196, 48, 183, 50, 59, 161, 34,
        246, 34, 145, 157, 225, 139, 31, 218, 176, 202, 153, 2, 185, 114, 157,
        73, 44, 128, 126, 197, 153, 213, 233, 128, 178, 234, 201, 204, 83, 191,
        103, 214, 191, 20, 214, 126, 45, 220, 142, 102, 131, 239, 87, 73, 97,
        255, 105, 143, 97, 205, 209, 30, 157, 156, 22, 114, 114, 230, 29, 240,
        132, 79, 74, 119, 2, 215, 232, 57, 44, 83, 203, 201, 18, 30, 51, 116,
        158, 12, 244, 213, 212, 159, 212, 164, 89, 126, 53, 207, 50, 34, 244,
        204, 207, 211, 144, 45, 72, 211, 143, 117, 230, 217, 29, 42, 229, 192,
        247, 43, 120, 129, 135, 68, 14, 95, 80, 0, 212, 97, 141, 190, 123, 5,
        21, 7, 59, 51, 130, 31, 24, 112, 146, 218, 100, 84, 206, 177, 133, 62,
        105, 21, 248, 70, 106, 4, 150, 115, 14, 217, 22, 47, 103, 104, 212, 247,
        74, 74, 208, 87, 104}, 255}, /* pseudo-random data */
     {0x9e, 0xb6, 0xce, 0x48, 0xf4, 0x6e, 0x9c, 0xf4, 0x1b, 0x4f, 0x9f, 0x66,
      0xdd, 0xe8, 0x41, 0x01, 0x71, 0xf6, 0xf5, 0xd6}}
 };

 static const size_t units2_num = sizeof(data_units2) / sizeof(data_units2[0]);


 /*
  *  Helper functions
  */

 /**
  * Print bin as hex
  *
  * @param bin binary data
  * @param len number of bytes in bin
  * @param hex pointer to len*2+1 bytes buffer
  */
 static void
 bin2hex (const uint8_t *bin,
          size_t len,
          char *hex)
 {
   while (len-- > 0)
   {
     unsigned int b1, b2;
     b1 = (*bin >> 4) & 0xf;
     *hex++ = (char) ((b1 > 9) ? (b1 + 'A' - 10) : (b1 + '0'));
     b2 = *bin++ & 0xf;
     *hex++ = (char) ((b2 > 9) ? (b2 + 'A' - 10) : (b2 + '0'));
   }
   *hex = 0;
 }


 static int
 check_result (const char *test_name,
               unsigned int check_num,
               const uint8_t calculated[SHA1_DIGEST_SIZE],
               const uint8_t expected[SHA1_DIGEST_SIZE])
 {
   int failed = memcmp (calculated, expected, SHA1_DIGEST_SIZE);
   check_num++; /* Print 1-based numbers */
   if (failed)
   {
     char calc_str[SHA1_DIGEST_STRING_SIZE];
     char expc_str[SHA1_DIGEST_STRING_SIZE];
     bin2hex (calculated, SHA1_DIGEST_SIZE, calc_str);
     bin2hex (expected, SHA1_DIGEST_SIZE, expc_str);
     fprintf (stderr,
              "FAILED: %s check %u: calculated digest %s, expected digest %s.\n",
              test_name, check_num, calc_str, expc_str);
     fflush (stderr);
   }
   else if (verbose)
   {
     char calc_str[SHA1_DIGEST_STRING_SIZE];
     bin2hex (calculated, SHA1_DIGEST_SIZE, calc_str);
     printf ("PASSED: %s check %u: calculated digest %s matches " \
             "expected digest.\n",
             test_name, check_num, calc_str);
     fflush (stdout);
   }
   return failed ? 1 : 0;
 }


 /*
  *  Tests
  */

 /* Calculated SHA-256 as one pass for whole data */
 static int
 test1_str (void)
 {
   int num_failed = 0;
   unsigned int i;

   for (i = 0; i < units1_num; i++)
   {
     struct sha1_ctx ctx;
     uint8_t digest[SHA1_DIGEST_SIZE];

     MHD_SHA1_init (&ctx);
     MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str,
                      data_units1[i].str_l.len);
     MHD_SHA1_finish (&ctx, digest);
     num_failed += check_result (MHD_FUNC_, i, digest,
                                 data_units1[i].digest);
   }
   return num_failed;
 }


 static int
 test1_bin (void)
 {
   int num_failed = 0;
   unsigned int i;

   for (i = 0; i < units2_num; i++)
   {
     struct sha1_ctx ctx;
     uint8_t digest[SHA1_DIGEST_SIZE];

     MHD_SHA1_init (&ctx);
     MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin,
                      data_units2[i].bin_l.len);
     MHD_SHA1_finish (&ctx, digest);
     num_failed += check_result (MHD_FUNC_, i, digest,
                                 data_units2[i].digest);
   }
   return num_failed;
 }


 /* Calculated SHA-256 as two iterations for whole data */
 static int
 test2_str (void)
 {
   int num_failed = 0;
   unsigned int i;

   for (i = 0; i < units1_num; i++)
   {
     struct sha1_ctx ctx;
     uint8_t digest[SHA1_DIGEST_SIZE];
     size_t part_s = data_units1[i].str_l.len / 4;

     MHD_SHA1_init (&ctx);
     MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str, part_s);
     MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str + part_s,
                      data_units1[i].str_l.len - part_s);
     MHD_SHA1_finish (&ctx, digest);
     num_failed += check_result (MHD_FUNC_, i, digest,
                                 data_units1[i].digest);
   }
   return num_failed;
 }


 static int
 test2_bin (void)
 {
   int num_failed = 0;
   unsigned int i;

   for (i = 0; i < units2_num; i++)
   {
     struct sha1_ctx ctx;
     uint8_t digest[SHA1_DIGEST_SIZE];
     size_t part_s = data_units2[i].bin_l.len * 2 / 3;

     MHD_SHA1_init (&ctx);
     MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin, part_s);
     MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin + part_s,
                      data_units2[i].bin_l.len - part_s);
     MHD_SHA1_finish (&ctx, digest);
     num_failed += check_result (MHD_FUNC_, i, digest,
                                 data_units2[i].digest);
   }
   return num_failed;
 }


 /* Use data set number 7 as it has the longest sequence */
 #define DATA_POS 6
 #define MAX_OFFSET 31

 static int
 test_unaligned (void)
 {
   int num_failed = 0;
   unsigned int offset;
   uint8_t *buf;
   uint8_t *digest_buf;

   const struct data_unit2 *const tdata = data_units2 + DATA_POS;

   buf = malloc (tdata->bin_l.len + MAX_OFFSET);
   digest_buf = malloc (SHA1_DIGEST_SIZE + MAX_OFFSET);
   if ((NULL == buf) || (NULL == digest_buf))
     exit (99);

   for (offset = MAX_OFFSET; offset >= 1; --offset)
   {
     struct sha1_ctx ctx;
     uint8_t *unaligned_digest;
     uint8_t *unaligned_buf;

     unaligned_buf = buf + offset;
     memcpy (unaligned_buf, tdata->bin_l.bin, tdata->bin_l.len);
     unaligned_digest = digest_buf + MAX_OFFSET - offset;
     memset (unaligned_digest, 0, SHA1_DIGEST_SIZE);

     MHD_SHA1_init (&ctx);
     MHD_SHA1_update (&ctx, unaligned_buf, tdata->bin_l.len);
     MHD_SHA1_finish (&ctx, unaligned_digest);
     num_failed += check_result (MHD_FUNC_, MAX_OFFSET - offset,
                                 unaligned_digest, tdata->digest);
   }
   free (digest_buf);
   free (buf);
   return num_failed;
 }


 int
 main (int argc, char *argv[])
 {
   int num_failed = 0;
   (void) has_in_name; /* Mute compiler warning. */
   if (has_param (argc, argv, "-v") || has_param (argc, argv, "--verbose"))
     verbose = 1;

   num_failed += test1_str ();
   num_failed += test1_bin ();

   num_failed += test2_str ();
   num_failed += test2_bin ();

   num_failed += test_unaligned ();

   return num_failed ? 1 : 0;
 }
	/*
	This file is part of libmicrohttpd
	Copyright (C) 2019-2021 Karlson2k (Evgeny Grin)

	This test tool is free software; you can redistribute it and/or
	modify it under the terms of the GNU General Public License as
	published by the Free Software Foundation; either version 2, or
	(at your option) any later version.

	This test tool is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public
	License along with this library; if not, write to the Free Software
	Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	*/

	/**
	* @file microhttpd/test_sha1.h
	* @brief Unit tests for SHA-1 functions
	* @author Karlson2k (Evgeny Grin)
	*/

	#include "mhd_options.h"
	#include "sha1.h"
	#include "test_helpers.h"
	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>

	static int verbose = 0; /* verbose level (0-1)*/


	struct str_with_len
	{
	const char *const str;
	const size_t len;
	};

	#define D_STR_W_LEN(s) {(s), (sizeof((s)) / sizeof(char)) - 1}

	struct data_unit1
	{
	const struct str_with_len str_l;
	const uint8_t digest[SHA1_DIGEST_SIZE];
	};

	static const struct data_unit1 data_units1[] = {
	{D_STR_W_LEN ("abc"),
	{0xA9, 0x99, 0x3E, 0x36, 0x47, 0x06, 0x81, 0x6A, 0xBA, 0x3E, 0x25, 0x71,
	0x78, 0x50, 0xC2, 0x6C, 0x9C, 0xD0, 0xD8, 0x9D}},
	{D_STR_W_LEN ("abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"),
	{0x84, 0x98, 0x3E, 0x44, 0x1C, 0x3B, 0xD2, 0x6E, 0xBA, 0xAE, 0x4A, 0xA1,
	0xF9, 0x51, 0x29, 0xE5, 0xE5, 0x46, 0x70, 0xF1}},
	{D_STR_W_LEN (""),
	{0xda, 0x39, 0xa3, 0xee, 0x5e, 0x6b, 0x4b, 0x0d, 0x32, 0x55, 0xbf, 0xef,
	0x95, 0x60, 0x18, 0x90, 0xaf, 0xd8, 0x07, 0x09}},
	{D_STR_W_LEN ("The quick brown fox jumps over the lazy dog"),
	{0x2f, 0xd4, 0xe1, 0xc6, 0x7a, 0x2d, 0x28, 0xfc, 0xed, 0x84, 0x9e, 0xe1,
	0xbb, 0x76, 0xe7, 0x39, 0x1b, 0x93, 0xeb, 0x12}},
	{D_STR_W_LEN ("1234567890!@~%&$@#{}[]\\/!?`."),
	{0xa7, 0x08, 0x9e, 0x3d, 0xe1, 0x6b, 0x63, 0xa3, 0x2a, 0x43, 0xa5, 0xe7,
	0xf3, 0xb5, 0x4f, 0x80, 0x6a, 0xc8, 0x4f, 0x53}},
	{D_STR_W_LEN ("Simple string."),
	{0x6c, 0x6c, 0x9c, 0xef, 0x53, 0xff, 0x22, 0x39, 0x0c, 0x54, 0xc7, 0xba,
	0x6d, 0x98, 0xe0, 0xd5, 0x6c, 0x24, 0x0d, 0x55}},
	{D_STR_W_LEN ("abcdefghijklmnopqrstuvwxyz"),
	{0x32, 0xd1, 0x0c, 0x7b, 0x8c, 0xf9, 0x65, 0x70, 0xca, 0x04, 0xce, 0x37,
	0xf2, 0xa1, 0x9d, 0x84, 0x24, 0x0d, 0x3a, 0x89}},
	{D_STR_W_LEN ("zyxwvutsrqponMLKJIHGFEDCBA"),
	{0x59, 0x0f, 0xc8, 0xea, 0xde, 0xa2, 0x78, 0x65, 0x5a, 0xf2, 0xa1, 0xe5,
	0xb4, 0xc9, 0x61, 0xa4, 0x3a, 0xc3, 0x6a, 0x83}},
	{D_STR_W_LEN ("abcdefghijklmnopqrstuvwxyzzyxwvutsrqponMLKJIHGFEDCBA"
	"abcdefghijklmnopqrstuvwxyzzyxwvutsrqponMLKJIHGFEDCBA"),
	{0xa7, 0x87, 0x4c, 0x16, 0xc0, 0x4e, 0xd6, 0x24, 0x5f, 0x25, 0xbe, 0x06,
	0x7b, 0x1b, 0x6b, 0xaf, 0xf4, 0x0e, 0x74, 0x0b}},
	};

	static const size_t units1_num = sizeof(data_units1) / sizeof(data_units1[0]);

	struct bin_with_len
	{
	const uint8_t bin[512];
	const size_t len;
	};

	struct data_unit2
	{
	const struct bin_with_len bin_l;
	const uint8_t digest[SHA1_DIGEST_SIZE];
	};

	/* Size must be less than 512 bytes! */
	static const struct data_unit2 data_units2[] = {
	{ { {97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111,
	112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122}, 26}, /* a..z ASCII sequence */
	{0x32, 0xd1, 0x0c, 0x7b, 0x8c, 0xf9, 0x65, 0x70, 0xca, 0x04, 0xce, 0x37,
	0xf2, 0xa1, 0x9d, 0x84, 0x24, 0x0d, 0x3a, 0x89}},
	{ { {65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
	65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
	65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65,
	65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65, 65},
	72 }, /* 'A' x 72 times */
	{0x41, 0xf9, 0x07, 0x05, 0x04, 0xf9, 0xc8, 0x1a, 0xbf, 0xbb, 0x61, 0x4d,
	0xaa, 0xec, 0x3b, 0x26, 0xa2, 0xf9, 0x23, 0x7e}},
	{ { {19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,
	37, 38, 39, 40, 41, 42,43, 44, 45, 46, 47, 48, 49, 50, 51, 52,53, 54,
	55, 56, 57, 58, 59, 60,61, 62, 63, 64, 65, 66, 67,68, 69, 70, 71, 72,
	73}, 55}, /* 19..73 sequence */
	{0xf2, 0x50, 0xb2, 0x79, 0x62, 0xcc, 0xff, 0x4b, 0x1b, 0x61, 0x4a, 0x6f,
	0x80, 0xec, 0x9b, 0x4d, 0xd0, 0xc0, 0xfb, 0x7b}},
	{ { {7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,
	26, 27, 28, 29, 30, 31,32, 33, 34, 35, 36, 37, 38, 39, 40, 41,42, 43,
	44, 45, 46, 47, 48, 49,50, 51, 52, 53, 54, 55, 56,57, 58, 59, 60, 61,
	62, 63, 64, 65, 66, 67, 68, 69}, 63}, /* 7..69 sequence */
	{0xf0, 0xa3, 0xfc, 0x4b, 0x90, 0x6f, 0x1b, 0x41, 0x68, 0xc8, 0x3e, 0x05,
	0xb0, 0xc5, 0xac, 0xb7, 0x3d, 0xcd, 0x6b, 0x0f}},
	{ { {38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55,
	56, 57, 58, 59, 60, 61,62, 63, 64, 65, 66, 67, 68, 69, 70, 71,72, 73,
	74, 75, 76, 77, 78, 79,80, 81, 82, 83, 84, 85, 86,87, 88, 89, 90, 91,
	92}, 55}, /* 38..92 sequence */
	{0x93, 0xa4, 0x9c, 0x5f, 0xda, 0x22, 0x63, 0x73, 0x85, 0xeb, 0x70, 0xd2,
	0x00, 0x52, 0x0c, 0xb0, 0x2e, 0x86, 0x9b, 0xa0 }},
	{ { {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,21,
	22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,37, 38, 39,
	40, 41, 42, 43, 44, 45,46, 47, 48, 49, 50, 51, 52,53, 54, 55, 56, 57,
	58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70,71, 72},72}, /* 1..72 sequence */
	{0x8b, 0x30, 0xd3, 0x41, 0x89, 0xb6, 0x1b, 0x66, 0x5a, 0x1a, 0x9a, 0x51,
	0x64, 0x93, 0xab, 0x5e, 0x78, 0x81, 0x52, 0xb5}},
	{ { {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
	21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,37, 38,
	39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,52, 53, 54, 55, 56,
	57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69,70, 71, 72, 73, 74,
	75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85,86, 87, 88, 89, 90, 91, 92,
	93, 94, 95, 96, 97, 98, 99, 100,101, 102, 103, 104, 105, 106, 107, 108,
	109, 110, 111, 112, 113, 114,115, 116, 117, 118, 119, 120, 121, 122,
	123, 124, 125, 126, 127,128, 129, 130, 131, 132, 133, 134, 135, 136,
	137, 138, 139, 140,141, 142, 143, 144, 145, 146, 147, 148, 149, 150,
	151, 152, 153, 154,155, 156, 157, 158, 159, 160, 161, 162, 163, 164,
	165, 166, 167,168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178,
	179, 180,181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192,
	193, 194,195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206,
	207,208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220,221,
	222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234,235,
	236, 237, 238, 239, 240,241, 242, 243, 244, 245, 246, 247,248, 249, 250,
	251, 252, 253, 254, 255}, 256}, /* 0..255 sequence */
	{0x49, 0x16, 0xd6, 0xbd, 0xb7, 0xf7, 0x8e, 0x68, 0x03, 0x69, 0x8c, 0xab,
	0x32, 0xd1, 0x58, 0x6e, 0xa4, 0x57, 0xdf, 0xc8}},
	{ { {199, 198, 197, 196, 195, 194, 193, 192, 191, 190, 189, 188, 187, 186,185,
	184, 183, 182, 181, 180,179, 178, 177, 176, 175, 174, 173,172, 171, 170,
	169, 168, 167, 166,165, 164, 163, 162, 161, 160,159, 158, 157, 156, 155,
	154, 153, 152, 151, 150, 149, 148, 147, 146,145, 144, 143, 142, 141,
	140,139}, 61}, /* 199..139 sequence */
	{0xb3, 0xec, 0x61, 0x3c, 0xf7, 0x36, 0x57, 0x94, 0x61, 0xdb, 0xb2, 0x16,
	0x75, 0xfe, 0x34, 0x60, 0x99, 0x94, 0xb5, 0xfc}},
	{ { {255, 254, 253, 252, 251, 250, 249, 248, 247, 246, 245, 244, 243, 242,
	241, 240, 239, 238, 237, 236, 235, 234, 233, 232, 231, 230, 229, 228,
	227, 226, 225, 224, 223, 222, 221, 220, 219, 218, 217, 216, 215, 214,
	213, 212, 211, 210, 209, 208, 207, 206, 205, 204, 203, 202, 201, 200,
	199, 198, 197, 196, 195, 194, 193, 192, 191, 190, 189, 188, 187, 186,
	185, 184, 183, 182, 181, 180, 179, 178, 177, 176, 175, 174, 173, 172,
	171, 170, 169, 168, 167, 166, 165, 164, 163, 162, 161, 160, 159, 158,
	157, 156, 155, 154, 153, 152, 151, 150, 149, 148, 147, 146, 145, 144,
	143, 142, 141, 140, 139, 138, 137, 136, 135, 134, 133, 132, 131, 130,
	129, 128, 127, 126, 125, 124, 123, 122, 121, 120, 119, 118, 117, 116,
	115, 114, 113, 112, 111, 110, 109, 108, 107, 106, 105, 104, 103, 102,
	101, 100, 99, 98, 97, 96, 95, 94, 93, 92, 91, 90, 89, 88, 87, 86, 85,
	84, 83, 82, 81, 80, 79, 78, 77, 76, 75, 74, 73, 72, 71, 70, 69, 68, 67,
	66, 65, 64, 63, 62, 61, 60, 59, 58, 57, 56, 55, 54, 53, 52, 51, 50, 49,
	48, 47, 46, 45, 44, 43, 42, 41, 40, 39, 38, 37, 36, 35, 34, 33, 32, 31,
	30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14, 13,
	12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1}, 255}, /* 255..1 sequence */
	{0x99, 0x1d, 0xc6, 0x95, 0xe1, 0xaa, 0xcc, 0xc9, 0x35, 0x60, 0xbe, 0xe3,
	0xe2, 0x41, 0x2b, 0xa7, 0xab, 0x49, 0x32, 0xf7}},
	{ { {41, 35, 190, 132, 225, 108, 214, 174, 82, 144, 73, 241, 241, 187, 233,
	235, 179, 166, 219, 60, 135, 12, 62, 153, 36, 94, 13, 28, 6, 183, 71,
	222, 179, 18, 77, 200, 67, 187, 139, 166, 31, 3, 90, 125, 9, 56, 37, 31,
	93, 212, 203, 252, 150, 245, 69, 59, 19, 13, 137, 10, 28, 219, 174, 50,
	32, 154, 80, 238, 64, 120, 54, 253, 18, 73, 50, 246, 158, 125, 73, 220,
	173, 79, 20, 242, 68, 64, 102, 208, 107, 196, 48, 183, 50, 59, 161, 34,
	246, 34, 145, 157, 225, 139, 31, 218, 176, 202, 153, 2, 185, 114, 157,
	73, 44, 128, 126, 197, 153, 213, 233, 128, 178, 234, 201, 204, 83, 191,
	103, 214, 191, 20, 214, 126, 45, 220, 142, 102, 131, 239, 87, 73, 97,
	255, 105, 143, 97, 205, 209, 30, 157, 156, 22, 114, 114, 230, 29, 240,
	132, 79, 74, 119, 2, 215, 232, 57, 44, 83, 203, 201, 18, 30, 51, 116,
	158, 12, 244, 213, 212, 159, 212, 164, 89, 126, 53, 207, 50, 34, 244,
	204, 207, 211, 144, 45, 72, 211, 143, 117, 230, 217, 29, 42, 229, 192,
	247, 43, 120, 129, 135, 68, 14, 95, 80, 0, 212, 97, 141, 190, 123, 5,
	21, 7, 59, 51, 130, 31, 24, 112, 146, 218, 100, 84, 206, 177, 133, 62,
	105, 21, 248, 70, 106, 4, 150, 115, 14, 217, 22, 47, 103, 104, 212, 247,
	74, 74, 208, 87, 104}, 255}, /* pseudo-random data */
	{0x9e, 0xb6, 0xce, 0x48, 0xf4, 0x6e, 0x9c, 0xf4, 0x1b, 0x4f, 0x9f, 0x66,
	0xdd, 0xe8, 0x41, 0x01, 0x71, 0xf6, 0xf5, 0xd6}}
	};

	static const size_t units2_num = sizeof(data_units2) / sizeof(data_units2[0]);


	/*
	* Helper functions
	*/

	/**
	* Print bin as hex
	*
	* @param bin binary data
	* @param len number of bytes in bin
	* @param hex pointer to len*2+1 bytes buffer
	*/
	static void
	bin2hex (const uint8_t *bin,
	size_t len,
	char *hex)
	{
	while (len-- > 0)
	{
	unsigned int b1, b2;
	b1 = (*bin >> 4) & 0xf;
	*hex++ = (char) ((b1 > 9) ? (b1 + 'A' - 10) : (b1 + '0'));
	b2 = *bin++ & 0xf;
	*hex++ = (char) ((b2 > 9) ? (b2 + 'A' - 10) : (b2 + '0'));
	}
	*hex = 0;
	}


	static int
	check_result (const char *test_name,
	unsigned int check_num,
	const uint8_t calculated[SHA1_DIGEST_SIZE],
	const uint8_t expected[SHA1_DIGEST_SIZE])
	{
	int failed = memcmp (calculated, expected, SHA1_DIGEST_SIZE);
	check_num++; /* Print 1-based numbers */
	if (failed)
	{
	char calc_str[SHA1_DIGEST_STRING_SIZE];
	char expc_str[SHA1_DIGEST_STRING_SIZE];
	bin2hex (calculated, SHA1_DIGEST_SIZE, calc_str);
	bin2hex (expected, SHA1_DIGEST_SIZE, expc_str);
	fprintf (stderr,
	"FAILED: %s check %u: calculated digest %s, expected digest %s.\n",
	test_name, check_num, calc_str, expc_str);
	fflush (stderr);
	}
	else if (verbose)
	{
	char calc_str[SHA1_DIGEST_STRING_SIZE];
	bin2hex (calculated, SHA1_DIGEST_SIZE, calc_str);
	printf ("PASSED: %s check %u: calculated digest %s matches " \
	"expected digest.\n",
	test_name, check_num, calc_str);
	fflush (stdout);
	}
	return failed ? 1 : 0;
	}


	/*
	* Tests
	*/

	/* Calculated SHA-256 as one pass for whole data */
	static int
	test1_str (void)
	{
	int num_failed = 0;
	unsigned int i;

	for (i = 0; i < units1_num; i++)
	{
	struct sha1_ctx ctx;
	uint8_t digest[SHA1_DIGEST_SIZE];

	MHD_SHA1_init (&ctx);
	MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str,
	data_units1[i].str_l.len);
	MHD_SHA1_finish (&ctx, digest);
	num_failed += check_result (MHD_FUNC_, i, digest,
	data_units1[i].digest);
	}
	return num_failed;
	}


	static int
	test1_bin (void)
	{
	int num_failed = 0;
	unsigned int i;

	for (i = 0; i < units2_num; i++)
	{
	struct sha1_ctx ctx;
	uint8_t digest[SHA1_DIGEST_SIZE];

	MHD_SHA1_init (&ctx);
	MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin,
	data_units2[i].bin_l.len);
	MHD_SHA1_finish (&ctx, digest);
	num_failed += check_result (MHD_FUNC_, i, digest,
	data_units2[i].digest);
	}
	return num_failed;
	}


	/* Calculated SHA-256 as two iterations for whole data */
	static int
	test2_str (void)
	{
	int num_failed = 0;
	unsigned int i;

	for (i = 0; i < units1_num; i++)
	{
	struct sha1_ctx ctx;
	uint8_t digest[SHA1_DIGEST_SIZE];
	size_t part_s = data_units1[i].str_l.len / 4;

	MHD_SHA1_init (&ctx);
	MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str, part_s);
	MHD_SHA1_update (&ctx, (const uint8_t *) data_units1[i].str_l.str + part_s,
	data_units1[i].str_l.len - part_s);
	MHD_SHA1_finish (&ctx, digest);
	num_failed += check_result (MHD_FUNC_, i, digest,
	data_units1[i].digest);
	}
	return num_failed;
	}


	static int
	test2_bin (void)
	{
	int num_failed = 0;
	unsigned int i;

	for (i = 0; i < units2_num; i++)
	{
	struct sha1_ctx ctx;
	uint8_t digest[SHA1_DIGEST_SIZE];
	size_t part_s = data_units2[i].bin_l.len * 2 / 3;

	MHD_SHA1_init (&ctx);
	MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin, part_s);
	MHD_SHA1_update (&ctx, data_units2[i].bin_l.bin + part_s,
	data_units2[i].bin_l.len - part_s);
	MHD_SHA1_finish (&ctx, digest);
	num_failed += check_result (MHD_FUNC_, i, digest,
	data_units2[i].digest);
	}
	return num_failed;
	}


	/* Use data set number 7 as it has the longest sequence */
	#define DATA_POS 6
	#define MAX_OFFSET 31

	static int
	test_unaligned (void)
	{
	int num_failed = 0;
	unsigned int offset;
	uint8_t *buf;
	uint8_t *digest_buf;

	const struct data_unit2 *const tdata = data_units2 + DATA_POS;

	buf = malloc (tdata->bin_l.len + MAX_OFFSET);
	digest_buf = malloc (SHA1_DIGEST_SIZE + MAX_OFFSET);
	if ((NULL == buf) \|\| (NULL == digest_buf))
	exit (99);

	for (offset = MAX_OFFSET; offset >= 1; --offset)
	{
	struct sha1_ctx ctx;
	uint8_t *unaligned_digest;
	uint8_t *unaligned_buf;

	unaligned_buf = buf + offset;
	memcpy (unaligned_buf, tdata->bin_l.bin, tdata->bin_l.len);
	unaligned_digest = digest_buf + MAX_OFFSET - offset;
	memset (unaligned_digest, 0, SHA1_DIGEST_SIZE);

	MHD_SHA1_init (&ctx);
	MHD_SHA1_update (&ctx, unaligned_buf, tdata->bin_l.len);
	MHD_SHA1_finish (&ctx, unaligned_digest);
	num_failed += check_result (MHD_FUNC_, MAX_OFFSET - offset,
	unaligned_digest, tdata->digest);
	}
	free (digest_buf);
	free (buf);
	return num_failed;
	}


	int
	main (int argc, char *argv[])
	{
	int num_failed = 0;
	(void) has_in_name; /* Mute compiler warning. */
	if (has_param (argc, argv, "-v") \|\| has_param (argc, argv, "--verbose"))
	verbose = 1;

	num_failed += test1_str ();
	num_failed += test1_bin ();

	num_failed += test2_str ();
	num_failed += test2_bin ();

	num_failed += test_unaligned ();

	return num_failed ? 1 : 0;
	}