dlls/advapi32/tests/crypt_md4.c

   1 /*
   2  * Unit tests for MD4 functions
   3  *
   4  * Copyright 2004 Hans Leidekker
   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdio.h>
  22
  23 #include "ntstatus.h"
  24 #define WIN32_NO_STATUS
  25 #include "wine/test.h"
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winerror.h"
  29 #include "winternl.h"
  30
  31 typedef struct
  32 {
  33     unsigned int buf[4];
  34     unsigned int i[2];
  35     unsigned char in[64];
  36     unsigned char digest[16];
  37 } MD4_CTX;
  38
  39 typedef VOID (WINAPI *fnMD4Init)( MD4_CTX *ctx );
  40 typedef VOID (WINAPI *fnMD4Update)( MD4_CTX *ctx, const unsigned char *src, const int len );
  41 typedef VOID (WINAPI *fnMD4Final)( MD4_CTX *ctx );
  42 typedef int (WINAPI *fnSystemFunction007)(const UNICODE_STRING *, LPBYTE);
  43 typedef int (WINAPI *md4hashfunc)(LPVOID, const LPBYTE, LPBYTE);
  44
  45 fnMD4Init pMD4Init;
  46 fnMD4Update pMD4Update;
  47 fnMD4Final pMD4Final;
  48 fnSystemFunction007 pSystemFunction007;
  49 md4hashfunc pSystemFunction010;
  50 md4hashfunc pSystemFunction011;
  51
  52 #define ctxcmp( a, b ) memcmp( a, b, FIELD_OFFSET( MD4_CTX, in ) )
  53
  54 static void test_md4_ctx(void)
  55 {
  56     static unsigned char message[] =
  57         "In our Life there's If"
  58         "In our beliefs there's Lie"
  59         "In our business there is Sin"
  60         "In our bodies, there is Die";
  61
  62     int size = sizeof(message) - 1;
  63
  64     MD4_CTX ctx;
  65     MD4_CTX ctx_initialized =
  66     {
  67         { 0x67452301, 0xefcdab89, 0x98badcfe, 0x10325476 },
  68         { 0, 0 }
  69     };
  70
  71     MD4_CTX ctx_update1 =
  72     {
  73         { 0x5e592ef7, 0xbdcb1567, 0x2b626d17, 0x7d1198bd },
  74         { 0x00000338, 0 }
  75     };
  76
  77     MD4_CTX ctx_update2 =
  78     {
  79         { 0x05dcfd65, 0xb3711c0d, 0x9e3369c2, 0x903ead11 },
  80         { 0x00000670, 0 }
  81     };
  82
  83     unsigned char expect[16] =
  84         { 0x5f, 0xd3, 0x9b, 0x29, 0x47, 0x53, 0x47, 0xaf,
  85           0xa5, 0xba, 0x0c, 0x05, 0xff, 0xc0, 0xc7, 0xda };
  86
  87
  88     memset( &ctx, 0, sizeof(ctx) );
  89     pMD4Init( &ctx );
  90     ok( !ctxcmp( &ctx, &ctx_initialized ), "invalid initialization\n" );
  91
  92     pMD4Update( &ctx, message, size );
  93     ok( !ctxcmp( &ctx, &ctx_update1 ), "update doesn't work correctly\n" );
  94
  95     pMD4Update( &ctx, message, size );
  96     ok( !ctxcmp( &ctx, &ctx_update2 ), "update doesn't work correctly\n" );
  97
  98     pMD4Final( &ctx );
  99     ok( ctxcmp( &ctx, &ctx_initialized ), "context has changed\n" );
 100     ok( !memcmp( ctx.digest, expect, sizeof(expect) ), "incorrect result\n" );
 101
 102 }
 103
 104 static void test_SystemFunction007(void)
 105 {
 106     int r;
 107     UNICODE_STRING str;
 108     BYTE output[0x10];
 109     BYTE expected[0x10] = { 0x24, 0x0a, 0xf0, 0x9d, 0x84, 0x1c, 0xda, 0xcf,
 110                             0x56, 0xeb, 0x6b, 0x96, 0x55, 0xec, 0xcf, 0x0a };
 111     WCHAR szFoo[] = {'f','o','o',0 };
 112
 113     if (0)
 114     {
 115     /* crashes on Windows */
 116     r = pSystemFunction007(NULL, NULL);
 117     ok( r == STATUS_UNSUCCESSFUL, "wrong error code\n");
 118     }
 119
 120     str.Buffer = szFoo;
 121     str.Length = 4*sizeof(WCHAR);
 122     str.MaximumLength = str.Length;
 123
 124     memset(output, 0, sizeof output);
 125     r = pSystemFunction007(&str, output);
 126     ok( r == STATUS_SUCCESS, "wrong error code\n");
 127
 128     ok(!memcmp(output, expected, sizeof expected), "response wrong\n");
 129 }
 130
 131 static void test_md4hashfunc(md4hashfunc func)
 132 {
 133     unsigned char expected[0x10] = {
 134         0x48, 0x7c, 0x3f, 0x5e, 0x2b, 0x0d, 0x6a, 0x79,
 135         0x32, 0x4e, 0xcd, 0xbe, 0x9c, 0x15, 0x16, 0x6f };
 136     unsigned char in[0x10], output[0x10];
 137     int r;
 138
 139     memset(in, 0, sizeof in);
 140     memset(output, 0, sizeof output);
 141     r = func(0, in, output);
 142     ok( r == STATUS_SUCCESS, "wrong error code\n");
 143     ok( !memcmp(expected, output, sizeof output), "output wrong\n");
 144 }
 145
 146 START_TEST(crypt_md4)
 147 {
 148     HMODULE module;
 149
 150     module = GetModuleHandleA( "advapi32.dll" );
 151
 152     pMD4Init = (fnMD4Init)GetProcAddress( module, "MD4Init" );
 153     pMD4Update = (fnMD4Update)GetProcAddress( module, "MD4Update" );
 154     pMD4Final = (fnMD4Final)GetProcAddress( module, "MD4Final" );
 155
 156     if (pMD4Init && pMD4Update && pMD4Final)
 157         test_md4_ctx();
 158     else
 159         win_skip("MD4Init and/or MD4Update and/or MD4Final are not available\n");
 160
 161     pSystemFunction007 = (fnSystemFunction007)GetProcAddress( module, "SystemFunction007" );
 162     if (pSystemFunction007)
 163         test_SystemFunction007();
 164     else
 165         win_skip("SystemFunction007 is not available\n");
 166
 167     pSystemFunction010 = (md4hashfunc)GetProcAddress( module, "SystemFunction010" );
 168     if (pSystemFunction010)
 169         test_md4hashfunc(pSystemFunction010);
 170     else
 171         win_skip("SystemFunction010 is not available\n");
 172
 173     pSystemFunction011 = (md4hashfunc)GetProcAddress( module, "SystemFunction011" );
 174     if (pSystemFunction011)
 175         test_md4hashfunc(pSystemFunction011);
 176     else
 177         win_skip("SystemFunction011 is not available\n");
 178 }