msvcrt: Fixed (_w)fopen_s with test.
[wine] / dlls / msvcrt / data.c
1 /*
2  * msvcrt.dll dll data items
3  *
4  * Copyright 2000 Jon Griffiths
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 "config.h"
22 #include "wine/port.h"
23
24 #include <math.h>
25 #include "msvcrt.h"
26 #include "wine/library.h"
27 #include "wine/unicode.h"
28 #include "wine/debug.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
31
32 int MSVCRT___argc = 0;
33 unsigned int MSVCRT_basemajor = 0;/* FIXME: */
34 unsigned int MSVCRT_baseminor = 0;/* FIXME: */
35 unsigned int MSVCRT_baseversion = 0; /* FIXME: */
36 unsigned int MSVCRT__commode = 0;
37 unsigned int MSVCRT__fmode = 0;
38 unsigned int MSVCRT_osmajor = 0;/* FIXME: */
39 unsigned int MSVCRT_osminor = 0;/* FIXME: */
40 unsigned int MSVCRT_osmode = 0;/* FIXME: */
41 unsigned int MSVCRT__osver = 0;
42 unsigned int MSVCRT__osplatform = 0;
43 unsigned int MSVCRT_osversion = 0; /* FIXME: */
44 unsigned int MSVCRT__winmajor = 0;
45 unsigned int MSVCRT__winminor = 0;
46 unsigned int MSVCRT__winver = 0;
47 unsigned int MSVCRT___setlc_active = 0;
48 unsigned int MSVCRT___unguarded_readlc_active = 0;
49 double MSVCRT__HUGE = 0;
50 char **MSVCRT___argv = NULL;
51 MSVCRT_wchar_t **MSVCRT___wargv = NULL;
52 char *MSVCRT__acmdln = NULL;
53 MSVCRT_wchar_t *MSVCRT__wcmdln = NULL;
54 char **MSVCRT__environ = NULL;
55 MSVCRT_wchar_t **MSVCRT__wenviron = NULL;
56 char **MSVCRT___initenv = NULL;
57 MSVCRT_wchar_t **MSVCRT___winitenv = NULL;
58 int MSVCRT_app_type = 0;
59 char* MSVCRT__pgmptr = NULL;
60 WCHAR* MSVCRT__wpgmptr = NULL;
61
62 /* Get a snapshot of the current environment
63  * and construct the __p__environ array
64  *
65  * The pointer returned from GetEnvironmentStrings may get invalid when
66  * some other module cause a reallocation of the env-variable block
67  *
68  * blk is an array of pointers to environment strings, ending with a NULL
69  * and after that the actual copy of the environment strings, ending in a \0
70  */
71 char ** msvcrt_SnapshotOfEnvironmentA(char **blk)
72 {
73   char* environ_strings = GetEnvironmentStringsA();
74   int count = 1, len = 1, i = 0; /* keep space for the trailing NULLS */
75   char *ptr;
76
77   for (ptr = environ_strings; *ptr; ptr += strlen(ptr) + 1)
78   {
79     count++;
80     len += strlen(ptr) + 1;
81   }
82   if (blk)
83       blk = HeapReAlloc( GetProcessHeap(), 0, blk, count* sizeof(char*) + len );
84   else
85     blk = HeapAlloc(GetProcessHeap(), 0, count* sizeof(char*) + len );
86
87   if (blk)
88     {
89       if (count)
90         {
91           memcpy(&blk[count],environ_strings,len);
92           for (ptr = (char*) &blk[count]; *ptr; ptr += strlen(ptr) + 1)
93             {
94               blk[i++] = ptr;
95             }
96         }
97       blk[i] = NULL;
98     }
99   FreeEnvironmentStringsA(environ_strings);
100   return blk;
101 }
102
103 MSVCRT_wchar_t ** msvcrt_SnapshotOfEnvironmentW(MSVCRT_wchar_t **wblk)
104 {
105   MSVCRT_wchar_t* wenviron_strings = GetEnvironmentStringsW();
106   int count = 1, len = 1, i = 0; /* keep space for the trailing NULLS */
107   MSVCRT_wchar_t *wptr;
108
109   for (wptr = wenviron_strings; *wptr; wptr += strlenW(wptr) + 1)
110   {
111     count++;
112     len += strlenW(wptr) + 1;
113   }
114   if (wblk)
115       wblk = HeapReAlloc( GetProcessHeap(), 0, wblk, count* sizeof(MSVCRT_wchar_t*) + len * sizeof(MSVCRT_wchar_t));
116   else
117     wblk = HeapAlloc(GetProcessHeap(), 0, count* sizeof(MSVCRT_wchar_t*) + len * sizeof(MSVCRT_wchar_t));
118   if (wblk)
119     {
120       if (count)
121         {
122           memcpy(&wblk[count],wenviron_strings,len * sizeof(MSVCRT_wchar_t));
123           for (wptr = (MSVCRT_wchar_t*)&wblk[count]; *wptr; wptr += strlenW(wptr) + 1)
124             {
125               wblk[i++] = wptr;
126             }
127         }
128       wblk[i] = NULL;
129     }
130   FreeEnvironmentStringsW(wenviron_strings);
131   return wblk;
132 }
133
134 typedef void (CDECL *_INITTERMFUN)(void);
135
136 /***********************************************************************
137  *              __p___argc (MSVCRT.@)
138  */
139 int* CDECL __p___argc(void) { return &MSVCRT___argc; }
140
141 /***********************************************************************
142  *              __p__commode (MSVCRT.@)
143  */
144 unsigned int* CDECL __p__commode(void) { return &MSVCRT__commode; }
145
146
147 /***********************************************************************
148  *              __p__pgmptr (MSVCRT.@)
149  */
150 char** CDECL __p__pgmptr(void) { return &MSVCRT__pgmptr; }
151
152 /***********************************************************************
153  *              __p__wpgmptr (MSVCRT.@)
154  */
155 WCHAR** CDECL __p__wpgmptr(void) { return &MSVCRT__wpgmptr; }
156
157 /***********************************************************************
158  *              __p__fmode (MSVCRT.@)
159  */
160 unsigned int* CDECL __p__fmode(void) { return &MSVCRT__fmode; }
161
162 /***********************************************************************
163  *              __p__osver (MSVCRT.@)
164  */
165 unsigned int* CDECL __p__osver(void) { return &MSVCRT__osver; }
166
167 /***********************************************************************
168  *              __p__winmajor (MSVCRT.@)
169  */
170 unsigned int* CDECL __p__winmajor(void) { return &MSVCRT__winmajor; }
171
172 /***********************************************************************
173  *              __p__winminor (MSVCRT.@)
174  */
175 unsigned int* CDECL __p__winminor(void) { return &MSVCRT__winminor; }
176
177 /***********************************************************************
178  *              __p__winver (MSVCRT.@)
179  */
180 unsigned int* CDECL __p__winver(void) { return &MSVCRT__winver; }
181
182 /*********************************************************************
183  *              __p__acmdln (MSVCRT.@)
184  */
185 char** CDECL __p__acmdln(void) { return &MSVCRT__acmdln; }
186
187 /*********************************************************************
188  *              __p__wcmdln (MSVCRT.@)
189  */
190 MSVCRT_wchar_t** CDECL __p__wcmdln(void) { return &MSVCRT__wcmdln; }
191
192 /*********************************************************************
193  *              __p___argv (MSVCRT.@)
194  */
195 char*** CDECL __p___argv(void) { return &MSVCRT___argv; }
196
197 /*********************************************************************
198  *              __p___wargv (MSVCRT.@)
199  */
200 MSVCRT_wchar_t*** CDECL __p___wargv(void) { return &MSVCRT___wargv; }
201
202 /*********************************************************************
203  *              __p__environ (MSVCRT.@)
204  */
205 char*** CDECL __p__environ(void)
206 {
207   return &MSVCRT__environ;
208 }
209
210 /*********************************************************************
211  *              __p__wenviron (MSVCRT.@)
212  */
213 MSVCRT_wchar_t*** CDECL __p__wenviron(void)
214 {
215   return &MSVCRT__wenviron;
216 }
217
218 /*********************************************************************
219  *              __p___initenv (MSVCRT.@)
220  */
221 char*** CDECL __p___initenv(void) { return &MSVCRT___initenv; }
222
223 /*********************************************************************
224  *              __p___winitenv (MSVCRT.@)
225  */
226 MSVCRT_wchar_t*** CDECL __p___winitenv(void) { return &MSVCRT___winitenv; }
227
228 /*********************************************************************
229  *              _get_osplatform (MSVCRT.@)
230  */
231 int CDECL MSVCRT__get_osplatform(int *pValue)
232 {
233     if (!MSVCRT_CHECK_PMT(pValue != NULL)) {
234         *MSVCRT__errno() = MSVCRT_EINVAL;
235         return MSVCRT_EINVAL;
236     }
237
238     *pValue = MSVCRT__osplatform;
239     return 0;
240 }
241
242 /* INTERNAL: Create a wide string from an ascii string */
243 MSVCRT_wchar_t *msvcrt_wstrdupa(const char *str)
244 {
245   const unsigned int len = strlen(str) + 1 ;
246   MSVCRT_wchar_t *wstr = MSVCRT_malloc(len* sizeof (MSVCRT_wchar_t));
247   if (!wstr)
248     return NULL;
249    MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED,str,len,wstr,len);
250   return wstr;
251 }
252
253 /*********************************************************************
254  *              ___unguarded_readlc_active_add_func (MSVCRT.@)
255  */
256 unsigned int * CDECL MSVCRT____unguarded_readlc_active_add_func(void)
257 {
258   return &MSVCRT___unguarded_readlc_active;
259 }
260
261 /*********************************************************************
262  *              ___setlc_active_func (MSVCRT.@)
263  */
264 unsigned int CDECL MSVCRT____setlc_active_func(void)
265 {
266   return MSVCRT___setlc_active;
267 }
268
269 /* INTERNAL: Since we can't rely on Winelib startup code calling w/getmainargs,
270  * we initialise data values during DLL loading. When called by a native
271  * program we simply return the data we've already initialised. This also means
272  * you can call multiple times without leaking
273  */
274 void msvcrt_init_args(void)
275 {
276   OSVERSIONINFOW osvi;
277
278   MSVCRT__acmdln = _strdup( GetCommandLineA() );
279   MSVCRT__wcmdln = msvcrt_wstrdupa(MSVCRT__acmdln);
280   MSVCRT___argc = __wine_main_argc;
281   MSVCRT___argv = __wine_main_argv;
282   MSVCRT___wargv = __wine_main_wargv;
283
284   TRACE("got %s, wide = %s argc=%d\n", debugstr_a(MSVCRT__acmdln),
285         debugstr_w(MSVCRT__wcmdln),MSVCRT___argc);
286
287   osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOW);
288   GetVersionExW( &osvi );
289   MSVCRT__winver     = (osvi.dwMajorVersion << 8) | osvi.dwMinorVersion;
290   MSVCRT__winmajor   = osvi.dwMajorVersion;
291   MSVCRT__winminor   = osvi.dwMinorVersion;
292   MSVCRT__osver      = osvi.dwBuildNumber;
293   MSVCRT__osplatform = osvi.dwPlatformId;
294   MSVCRT_osversion   = MSVCRT__winver;
295   MSVCRT_osmajor     = MSVCRT__winmajor;
296   MSVCRT_osminor     = MSVCRT__winminor;
297   MSVCRT_baseversion = MSVCRT__osver;
298   MSVCRT_baseminor   = MSVCRT_baseversion & 0xFF;
299   MSVCRT_basemajor   = (MSVCRT_baseversion >> 8) & 0xFF;
300   TRACE( "winver %08x winmajor %08x winminor %08x osver%08x baseversion %08x basemajor %08x baseminor %08x\n",
301           MSVCRT__winver, MSVCRT__winmajor, MSVCRT__winminor, MSVCRT__osver, MSVCRT_baseversion,
302           MSVCRT_basemajor, MSVCRT_baseminor);
303   TRACE( "osversion %08x osmajor %08x osminor %08x\n", MSVCRT_osversion, MSVCRT_osmajor, MSVCRT_osminor);
304
305   MSVCRT__HUGE = HUGE_VAL;
306   MSVCRT___setlc_active = 0;
307   MSVCRT___unguarded_readlc_active = 0;
308   MSVCRT__fmode = MSVCRT__O_TEXT;
309
310   MSVCRT__environ = msvcrt_SnapshotOfEnvironmentA(NULL);
311   MSVCRT___initenv = msvcrt_SnapshotOfEnvironmentA(NULL);
312   MSVCRT___winitenv = msvcrt_SnapshotOfEnvironmentW(NULL);
313
314   MSVCRT__pgmptr = HeapAlloc(GetProcessHeap(), 0, MAX_PATH);
315   if (MSVCRT__pgmptr)
316   {
317     if (!GetModuleFileNameA(0, MSVCRT__pgmptr, MAX_PATH))
318       MSVCRT__pgmptr[0] = '\0';
319     else
320       MSVCRT__pgmptr[MAX_PATH - 1] = '\0';
321   }
322
323   MSVCRT__wpgmptr = HeapAlloc(GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR));
324   if (MSVCRT__wpgmptr)
325   {
326     if (!GetModuleFileNameW(0, MSVCRT__wpgmptr, MAX_PATH))
327       MSVCRT__wpgmptr[0] = '\0';
328     else
329       MSVCRT__wpgmptr[MAX_PATH - 1] = '\0';
330   }
331 }
332
333
334 /* INTERNAL: free memory used by args */
335 void msvcrt_free_args(void)
336 {
337   /* FIXME: more things to free */
338   HeapFree(GetProcessHeap(), 0, MSVCRT___initenv);
339   HeapFree(GetProcessHeap(), 0, MSVCRT___winitenv);
340   HeapFree(GetProcessHeap(), 0, MSVCRT__environ);
341   HeapFree(GetProcessHeap(), 0, MSVCRT__wenviron);
342   HeapFree(GetProcessHeap(), 0, MSVCRT__pgmptr);
343   HeapFree(GetProcessHeap(), 0, MSVCRT__wpgmptr);
344 }
345
346 /*********************************************************************
347  *              __getmainargs (MSVCRT.@)
348  */
349 void CDECL __getmainargs(int *argc, char** *argv, char** *envp,
350                          int expand_wildcards, int *new_mode)
351 {
352   TRACE("(%p,%p,%p,%d,%p).\n", argc, argv, envp, expand_wildcards, new_mode);
353   *argc = MSVCRT___argc;
354   *argv = MSVCRT___argv;
355   *envp = MSVCRT___initenv;
356   if (new_mode)
357     MSVCRT__set_new_mode( *new_mode );
358 }
359
360 /*********************************************************************
361  *              __wgetmainargs (MSVCRT.@)
362  */
363 void CDECL __wgetmainargs(int *argc, MSVCRT_wchar_t** *wargv, MSVCRT_wchar_t** *wenvp,
364                           int expand_wildcards, int *new_mode)
365 {
366   TRACE("(%p,%p,%p,%d,%p).\n", argc, wargv, wenvp, expand_wildcards, new_mode);
367
368   /* Initialize the _wenviron array if it's not already created. */
369   if (!MSVCRT__wenviron)
370     MSVCRT__wenviron = msvcrt_SnapshotOfEnvironmentW(NULL);
371   *argc = MSVCRT___argc;
372   *wargv = MSVCRT___wargv;
373   *wenvp = MSVCRT___winitenv;
374   if (new_mode)
375     MSVCRT__set_new_mode( *new_mode );
376 }
377
378 /*********************************************************************
379  *              _initterm (MSVCRT.@)
380  */
381 void CDECL _initterm(_INITTERMFUN *start,_INITTERMFUN *end)
382 {
383   _INITTERMFUN* current = start;
384
385   TRACE("(%p,%p)\n",start,end);
386   while (current<end)
387   {
388     if (*current)
389     {
390       TRACE("Call init function %p\n",*current);
391       (**current)();
392       TRACE("returned\n");
393     }
394     current++;
395   }
396 }
397
398 /*********************************************************************
399  *              __set_app_type (MSVCRT.@)
400  */
401 void CDECL MSVCRT___set_app_type(int app_type)
402 {
403   TRACE("(%d) %s application\n", app_type, app_type == 2 ? "Gui" : "Console");
404   MSVCRT_app_type = app_type;
405 }