ceph/src/crypto/isa-l/isa-l_crypto/sha256_mb/sha256_mb_rand_test.c

   1 /**********************************************************************
   2   Copyright(c) 2011-2016 Intel Corporation All rights reserved.
   3
   4   Redistribution and use in source and binary forms, with or without
   5   modification, are permitted provided that the following conditions
   6   are met:
   7     * Redistributions of source code must retain the above copyright
   8       notice, this list of conditions and the following disclaimer.
   9     * Redistributions in binary form must reproduce the above copyright
  10       notice, this list of conditions and the following disclaimer in
  11       the documentation and/or other materials provided with the
  12       distribution.
  13     * Neither the name of Intel Corporation nor the names of its
  14       contributors may be used to endorse or promote products derived
  15       from this software without specific prior written permission.
  16
  17   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  20   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  21   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28 **********************************************************************/
  29
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include "sha256_mb.h"
  33
  34 #define TEST_LEN  (1024*1024)
  35 #define TEST_BUFS 100
  36 #ifndef RANDOMS
  37 # define RANDOMS  10
  38 #endif
  39 #ifndef TEST_SEED
  40 # define TEST_SEED 0x1234
  41 #endif
  42
  43 static uint32_t digest_ref[TEST_BUFS][SHA256_DIGEST_NWORDS];
  44
  45 // Compare against reference function
  46 extern void sha256_ref(uint8_t * input_data, uint32_t * digest, uint32_t len);
  47
  48 // Generates pseudo-random data
  49 void rand_buffer(unsigned char *buf, const long buffer_size)
  50 {
  51         long i;
  52         for (i = 0; i < buffer_size; i++)
  53                 buf[i] = rand();
  54 }
  55
  56 int main(void)
  57 {
  58         SHA256_HASH_CTX_MGR *mgr = NULL;
  59         SHA256_HASH_CTX ctxpool[TEST_BUFS];
  60         uint32_t i, j, fail = 0;
  61         unsigned char *bufs[TEST_BUFS];
  62         uint32_t lens[TEST_BUFS];
  63         unsigned int jobs, t;
  64         uint8_t *tmp_buf;
  65
  66         printf("multibinary_sha256 test, %d sets of %dx%d max: ", RANDOMS, TEST_BUFS,
  67                TEST_LEN);
  68
  69         posix_memalign((void *)&mgr, 16, sizeof(SHA256_HASH_CTX_MGR));
  70         sha256_ctx_mgr_init(mgr);
  71
  72         srand(TEST_SEED);
  73
  74         for (i = 0; i < TEST_BUFS; i++) {
  75                 // Allocate  and fill buffer
  76                 bufs[i] = (unsigned char *)malloc(TEST_LEN);
  77                 if (bufs[i] == NULL) {
  78                         printf("malloc failed test aborted\n");
  79                         return 1;
  80                 }
  81                 rand_buffer(bufs[i], TEST_LEN);
  82
  83                 // Init ctx contexts
  84                 hash_ctx_init(&ctxpool[i]);
  85                 ctxpool[i].user_data = (void *)((uint64_t) i);
  86
  87                 // Run reference test
  88                 sha256_ref(bufs[i], digest_ref[i], TEST_LEN);
  89
  90                 // Run sb_sha256 test
  91                 sha256_ctx_mgr_submit(mgr, &ctxpool[i], bufs[i], TEST_LEN, HASH_ENTIRE);
  92         }
  93
  94         while (sha256_ctx_mgr_flush(mgr)) ;
  95
  96         for (i = 0; i < TEST_BUFS; i++) {
  97                 for (j = 0; j < SHA256_DIGEST_NWORDS; j++) {
  98                         if (ctxpool[i].job.result_digest[j] != digest_ref[i][j]) {
  99                                 fail++;
 100                                 printf("Test%d fixed size, digest%d "
 101                                        "fail 0x%08X <=> 0x%08X \n",
 102                                        i, j, ctxpool[i].job.result_digest[j],
 103                                        digest_ref[i][j]);
 104                         }
 105                 }
 106         }
 107
 108         if (fail) {
 109                 printf("Test failed function check %d\n", fail);
 110                 return fail;
 111         }
 112         // Run tests with random size and number of jobs
 113         for (t = 0; t < RANDOMS; t++) {
 114                 jobs = rand() % (TEST_BUFS);
 115
 116                 sha256_ctx_mgr_init(mgr);
 117
 118                 for (i = 0; i < jobs; i++) {
 119                         // Use buffer with random len and contents
 120                         lens[i] = rand() % (TEST_LEN);
 121                         rand_buffer(bufs[i], lens[i]);
 122
 123                         // Run reference test
 124                         sha256_ref(bufs[i], digest_ref[i], lens[i]);
 125
 126                         // Run sha256_mb test
 127                         sha256_ctx_mgr_submit(mgr, &ctxpool[i], bufs[i], lens[i], HASH_ENTIRE);
 128                 }
 129
 130                 while (sha256_ctx_mgr_flush(mgr)) ;
 131
 132                 for (i = 0; i < jobs; i++) {
 133                         for (j = 0; j < SHA256_DIGEST_NWORDS; j++) {
 134                                 if (ctxpool[i].job.result_digest[j] != digest_ref[i][j]) {
 135                                         fail++;
 136                                         printf("Test%d, digest%d fail "
 137                                                "0x%08X <=> 0x%08X\n",
 138                                                i, j, ctxpool[i].job.result_digest[j],
 139                                                digest_ref[i][j]);
 140                                 }
 141                         }
 142                 }
 143                 if (fail) {
 144                         printf("Test failed function check %d\n", fail);
 145                         return fail;
 146                 }
 147
 148                 putchar('.');
 149                 fflush(0);
 150         }                       // random test t
 151
 152         // Test at the end of buffer
 153         jobs = rand() % TEST_BUFS;
 154         tmp_buf = (uint8_t *) malloc(sizeof(uint8_t) * jobs);
 155         if (!tmp_buf) {
 156                 printf("malloc failed, end test aborted.\n");
 157                 return 1;
 158         }
 159
 160         rand_buffer(tmp_buf, jobs);
 161
 162         sha256_ctx_mgr_init(mgr);
 163
 164         // Extend to the end of allocated buffer to construct jobs
 165         for (i = 0; i < jobs; i++) {
 166                 bufs[i] = (uint8_t *) & tmp_buf[i];
 167                 lens[i] = jobs - i;
 168
 169                 // Reference test
 170                 sha256_ref(bufs[i], digest_ref[i], lens[i]);
 171
 172                 // sb_sha256 test
 173                 sha256_ctx_mgr_submit(mgr, &ctxpool[i], bufs[i], lens[i], HASH_ENTIRE);
 174         }
 175
 176         while (sha256_ctx_mgr_flush(mgr)) ;
 177
 178         for (i = 0; i < jobs; i++) {
 179                 for (j = 0; j < SHA256_DIGEST_NWORDS; j++) {
 180                         if (ctxpool[i].job.result_digest[j] != digest_ref[i][j]) {
 181                                 fail++;
 182                                 printf("End test failed at offset %d - result: 0x%08X"
 183                                        ", ref: 0x%08X\n", i, ctxpool[i].job.result_digest[j],
 184                                        digest_ref[i][j]);
 185                         }
 186                 }
 187         }
 188
 189         putchar('.');
 190
 191         if (fail)
 192                 printf("Test failed function check %d\n", fail);
 193         else
 194                 printf(" multibinary_sha256 rand: Pass\n");
 195
 196         return fail;
 197 }