git.oblomov.eu Git - wine/blob - dlls/advapi32/tests/crypt_sha.c

   1 /*
   2  * Unit tests for SHA functions
   3  *
   4  * Copyright (c) 2004 Filip Navara
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library 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 GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdio.h>
  22
  23 #include "wine/test.h"
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winerror.h"
  27
  28 typedef struct {
  29    ULONG Unknown[6];
  30    ULONG State[5];
  31    ULONG Count[2];
  32    UCHAR Buffer[64];
  33 } SHA_CTX, *PSHA_CTX;
  34
  35 typedef VOID (WINAPI *fnA_SHAInit)(PSHA_CTX Context);
  36 typedef VOID (WINAPI *fnA_SHAUpdate)(PSHA_CTX Context, PCHAR Buffer, UINT BufferSize);
  37 typedef VOID (WINAPI *fnA_SHAFinal)(PSHA_CTX Context, PULONG Result);
  38
  39 fnA_SHAInit pA_SHAInit;
  40 fnA_SHAUpdate pA_SHAUpdate;
  41 fnA_SHAFinal pA_SHAFinal;
  42
  43 #define ctxcmp(a,b) memcmp((char*)a, (char*)b, FIELD_OFFSET(SHA_CTX, Buffer))
  44
  45 void test_sha_ctx()
  46 {
  47    PCHAR test_buffer = "In our Life there's If"
  48                        "In our beliefs there's Lie"
  49                        "In our business there is Sin"
  50                        "In our bodies, there is Die";
  51    ULONG test_buffer_size = strlen(test_buffer);
  52    HMODULE hmod;
  53    SHA_CTX ctx;
  54    SHA_CTX ctx_initialized = {{0, 0, 0, 0, 0}, {0x67452301, 0xefcdab89, 0x98badcfe, 0x10325476, 0xc3d2e1f0}, {0, 0}};
  55    SHA_CTX ctx_update1 = {{0, 0, 0, 0, 0}, {0xdbe5eba8, 0x6b4335ca, 0xf7c94abe, 0xc9f34e31, 0x311023f0}, {0, 0x67}};
  56    SHA_CTX ctx_update2 = {{0, 0, 0, 0, 0}, {0x5ecc818d, 0x52498169, 0xf6758559, 0xd035a164, 0x871dd125}, {0, 0xce}};
  57    ULONG result[5];
  58    ULONG result_correct[5] = {0xe014f93, 0xe09791ec, 0x6dcf96c8, 0x8e9385fc, 0x1611c1bb};
  59
  60    hmod = LoadLibrary("advapi32.dll");
  61    pA_SHAInit = (fnA_SHAInit)GetProcAddress(hmod, "A_SHAInit");
  62    pA_SHAUpdate = (fnA_SHAUpdate)GetProcAddress(hmod, "A_SHAUpdate");
  63    pA_SHAFinal = (fnA_SHAFinal)GetProcAddress(hmod, "A_SHAFinal");
  64
  65    RtlZeroMemory(&ctx, sizeof(ctx));
  66    pA_SHAInit(&ctx);
  67    ok(!ctxcmp(&ctx, &ctx_initialized), "invalid initialization\n");
  68
  69    pA_SHAUpdate(&ctx, test_buffer, test_buffer_size);
  70    ok(!ctxcmp(&ctx, &ctx_update1), "update doesn't work correctly\n");
  71
  72    pA_SHAUpdate(&ctx, test_buffer, test_buffer_size);
  73    ok(!ctxcmp(&ctx, &ctx_update2), "update doesn't work correctly\n");
  74
  75    pA_SHAFinal(&ctx, result);
  76    ok(!ctxcmp(&ctx, &ctx_initialized), "context hasn't been reinitialized\n");
  77    ok(!memcmp(result, result_correct, sizeof(result)), "incorrect result\n");
  78
  79    FreeLibrary(hmod);
  80 }
  81
  82 START_TEST(crypt_sha)
  83 {
  84     test_sha_ctx();
  85 }