git.oblomov.eu Git - wine/blob - dlls/msvcrt/tests/data.c

   1 /*
   2  * Tests msvcrt/data.c
   3  *
   4  * Copyright 2006 Andrew Ziem
   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 "wine/test.h"
  22 #include <stdarg.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <fcntl.h>
  26 #include <sys/stat.h>
  27 #include <io.h>
  28 #include <windef.h>
  29 #include <winbase.h>
  30 #include <winnls.h>
  31 #include <process.h>
  32 #include <errno.h>
  33 #include <direct.h>
  34
  35 void __cdecl __getmainargs(int *, char ***, char ***, int, int *);
  36 static int* (__cdecl *p___p___argc)(void);
  37 static char*** (__cdecl *p___p___argv)(void);
  38
  39 typedef void (__cdecl *_INITTERMFUN)(void);
  40 static void (__cdecl *p_initterm)(_INITTERMFUN *start, _INITTERMFUN *end);
  41
  42 static int (__cdecl *p_get_pgmptr)(char **p);
  43
  44 static int callbacked;
  45
  46 static void __cdecl initcallback(void)
  47 {
  48    callbacked++;
  49 }
  50
  51 #define initterm_test(start, end, expected) \
  52     callbacked = 0; \
  53     p_initterm(start, end); \
  54     ok(expected == callbacked,"_initterm: callbacks count mismatch: got %i, expected %i\n", callbacked, expected);
  55
  56 static void test_initterm(void)
  57 {
  58     int i;
  59     static _INITTERMFUN callbacks[4];
  60
  61     if (!p_initterm)
  62         return;
  63
  64     for (i = 0; i < 4; i++)
  65     {
  66         callbacks[i] = initcallback;
  67     }
  68
  69     initterm_test(&callbacks[0], &callbacks[1], 1);
  70     initterm_test(&callbacks[0], &callbacks[2], 2);
  71     initterm_test(&callbacks[0], &callbacks[3], 3);
  72
  73     callbacks[1] = NULL;
  74     initterm_test(&callbacks[0], &callbacks[3], 2);
  75 }
  76
  77 static void test_initvar( HMODULE hmsvcrt )
  78 {
  79     OSVERSIONINFO osvi = { sizeof(OSVERSIONINFO) };
  80     int *pp_winver   = (int*)GetProcAddress(hmsvcrt, "_winver");
  81     int *pp_winmajor = (int*)GetProcAddress(hmsvcrt, "_winmajor");
  82     int *pp_winminor = (int*)GetProcAddress(hmsvcrt, "_winminor");
  83     int *pp_osver    = (int*)GetProcAddress(hmsvcrt, "_osver");
  84     int *pp_osplatform = (int*)GetProcAddress(hmsvcrt, "_osplatform");
  85     unsigned int winver, winmajor, winminor, osver, osplatform;
  86
  87     if( !( pp_winmajor && pp_winminor && pp_winver)) {
  88         win_skip("_winver variables are not available\n");
  89         return;
  90     }
  91     winver = *pp_winver;
  92     winminor = *pp_winminor;
  93     winmajor = *pp_winmajor;
  94     GetVersionEx( &osvi);
  95     ok( winminor == osvi.dwMinorVersion, "Wrong value for _winminor %02x expected %02x\n",
  96             winminor, osvi.dwMinorVersion);
  97     ok( winmajor == osvi.dwMajorVersion, "Wrong value for _winmajor %02x expected %02x\n",
  98             winmajor, osvi.dwMajorVersion);
  99     ok( winver == ((osvi.dwMajorVersion << 8) | osvi.dwMinorVersion),
 100             "Wrong value for _winver %02x expected %02x\n",
 101             winver, ((osvi.dwMajorVersion << 8) | osvi.dwMinorVersion));
 102     if( !pp_osver || !pp_osplatform ) {
 103         win_skip("_osver variables are not available\n");
 104         return;
 105     }
 106     osver = *pp_osver;
 107     osplatform = *pp_osplatform;
 108     ok( osver == (osvi.dwBuildNumber & 0xffff) ||
 109             ((osvi.dwBuildNumber >> 24) == osvi.dwMajorVersion &&
 110                  ((osvi.dwBuildNumber >> 16) & 0xff) == osvi.dwMinorVersion), /* 95/98/ME */
 111             "Wrong value for _osver %04x expected %04x\n",
 112             osver, osvi.dwBuildNumber);
 113     ok(osplatform == osvi.dwPlatformId,
 114             "Wrong value for _osplatform %x exprected %x\n",
 115             osplatform, osvi.dwPlatformId);
 116 }
 117
 118 static void test_get_pgmptr(void)
 119 {
 120     char *pgm = NULL;
 121     int res;
 122
 123     if (!p_get_pgmptr)
 124         return;
 125
 126     res = p_get_pgmptr(&pgm);
 127
 128     ok( res == 0, "Wrong _get_pgmptr return value %d expected 0\n", res);
 129     ok( pgm != NULL, "_get_pgmptr returned a NULL pointer\n" );
 130 }
 131
 132 static void test___getmainargs(void)
 133 {
 134     int argc, new_argc, mode;
 135     char **argv, **new_argv, **envp;
 136     char tmppath[MAX_PATH], filepath[MAX_PATH];
 137     FILE *f;
 138
 139     ok(GetTempPathA(MAX_PATH, tmppath) != 0, "GetTempPath failed\n");
 140
 141     mode = 0;
 142     __getmainargs(&argc, &argv, &envp, 0, &mode);
 143     ok(argc == 4, "argc = %d\n", argc);
 144     ok(!strcmp(argv[1], "data"), "argv[1] = %s\n", argv[1]);
 145     sprintf(filepath, "%s*\\*", tmppath);
 146     ok(!strcmp(argv[2], filepath), "argv[2] = %s\n", argv[2]);
 147     sprintf(filepath, "%swine_test/*", tmppath);
 148     ok(!strcmp(argv[3], filepath), "argv[3] = %s\n", argv[3]);
 149
 150     if(p___p___argc && p___p___argv) {
 151         new_argc = *p___p___argc();
 152         new_argv = *p___p___argv();
 153         ok(new_argc == 4, "*__p___argc() = %d\n", new_argc);
 154         ok(new_argv == argv, "*__p___argv() = %p, epxected %p\n", new_argv, argv);
 155     }else {
 156         win_skip("__p___argc or __p___argv is not available\n");
 157     }
 158
 159     mode = 0;
 160     __getmainargs(&argc, &argv, &envp, 1, &mode);
 161     ok(argc == 5, "argc = %d\n", argc);
 162     ok(!strcmp(argv[1], "data"), "argv[1] = %s\n", argv[1]);
 163     sprintf(filepath, "%s*\\*", tmppath);
 164     ok(!strcmp(argv[2], filepath), "argv[2] = %s\n", argv[2]);
 165     sprintf(filepath, "%swine_test/a", tmppath);
 166     if(argv[3][strlen(argv[3])-1] == 'a') {
 167         ok(!strcmp(argv[3], filepath), "argv[3] = %s\n", argv[3]);
 168         sprintf(filepath, "%swine_test/test", tmppath);
 169         ok(!strcmp(argv[4], filepath), "argv[4] = %s\n", argv[4]);
 170     }else {
 171         ok(!strcmp(argv[4], filepath), "argv[4] = %s\n", argv[4]);
 172         sprintf(filepath, "%swine_test/test", tmppath);
 173         ok(!strcmp(argv[3], filepath), "argv[3] = %s\n", argv[3]);
 174     }
 175
 176     if(p___p___argc && p___p___argv) {
 177         new_argc = *p___p___argc();
 178         new_argv = *p___p___argv();
 179         ok(new_argc == argc, "*__p___argc() = %d, expected %d\n", new_argc, argc);
 180         ok(new_argv == argv, "*__p___argv() = %p, epxected %p\n", new_argv, argv);
 181     }
 182
 183     sprintf(filepath, "%swine_test/b", tmppath);
 184     f = fopen(filepath, "w");
 185     ok(f != NULL, "fopen(%s) failed: %d\n", filepath, errno);
 186     fclose(f);
 187     mode = 0;
 188     __getmainargs(&new_argc, &new_argv, &envp, 1, &mode);
 189     ok(new_argc == argc+1, "new_argc = %d, expected %d\n", new_argc, argc+1);
 190     _unlink(filepath);
 191 }
 192
 193 static void test___getmainargs_parent(char *name)
 194 {
 195     char cmdline[3*MAX_PATH];
 196     char tmppath[MAX_PATH], filepath[MAX_PATH];
 197     STARTUPINFO startup;
 198     PROCESS_INFORMATION proc;
 199     FILE *f;
 200
 201     ok(GetTempPathA(MAX_PATH, tmppath) != 0, "GetTempPath failed\n");
 202     sprintf(cmdline, "%s data %s*\\* %swine_test/*", name, tmppath, tmppath);
 203
 204     sprintf(filepath, "%swine_test", tmppath);
 205     ok(!_mkdir(filepath), "_mkdir failed: %d\n", errno);
 206     sprintf(filepath, "%swine_test\\a", tmppath);
 207     f = fopen(filepath, "w");
 208     ok(f != NULL, "fopen(%s) failed: %d\n", filepath, errno);
 209     fclose(f);
 210     sprintf(filepath, "%swine_test\\test", tmppath);
 211     f = fopen(filepath, "w");
 212     ok(f != NULL, "fopen(%s) failed: %d\n", filepath, errno);
 213     fclose(f);
 214
 215     memset(&startup, 0, sizeof(startup));
 216     startup.cb = sizeof(startup);
 217     CreateProcessA(NULL, cmdline, NULL, NULL, TRUE, CREATE_DEFAULT_ERROR_MODE|NORMAL_PRIORITY_CLASS, NULL, NULL, &startup, &proc);
 218     winetest_wait_child_process(proc.hProcess);
 219
 220     _unlink(filepath);
 221     sprintf(filepath, "%swine_test\\a", tmppath);
 222     _unlink(filepath);
 223     sprintf(filepath, "%swine_test", tmppath);
 224     _rmdir(filepath);
 225 }
 226
 227 START_TEST(data)
 228 {
 229     HMODULE hmsvcrt;
 230     int arg_c;
 231     char** arg_v;
 232
 233     hmsvcrt = GetModuleHandleA("msvcrt.dll");
 234     if (!hmsvcrt)
 235         hmsvcrt = GetModuleHandleA("msvcrtd.dll");
 236     if (hmsvcrt)
 237     {
 238         p_initterm=(void*)GetProcAddress(hmsvcrt, "_initterm");
 239         p_get_pgmptr=(void*)GetProcAddress(hmsvcrt, "_get_pgmptr");
 240         p___p___argc=(void*)GetProcAddress(hmsvcrt, "__p___argc");
 241         p___p___argv=(void*)GetProcAddress(hmsvcrt, "__p___argv");
 242     }
 243
 244     arg_c = winetest_get_mainargs(&arg_v);
 245     if(arg_c >= 3) {
 246         test___getmainargs();
 247         return;
 248     }
 249
 250     test_initterm();
 251     test_initvar(hmsvcrt);
 252     test_get_pgmptr();
 253     test___getmainargs_parent(arg_v[0]);
 254 }