kernel32/tests: Fix some profile test failures on NT4.
[wine] / dlls / kernel32 / kernel_main.c
1 /*
2  * Kernel initialization code
3  *
4  * Copyright 2000 Alexandre Julliard
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 <assert.h>
25 #include <ctype.h>
26 #include <stdarg.h>
27 #include <string.h>
28 #include <signal.h>
29
30 #include "windef.h"
31 #include "winbase.h"
32 #include "wincon.h"
33 #include "winternl.h"
34 #include "wownt32.h"
35
36 #include "wine/winbase16.h"
37 #include "wine/library.h"
38 #include "toolhelp.h"
39 #include "kernel_private.h"
40 #include "kernel16_private.h"
41 #include "console_private.h"
42 #include "wine/debug.h"
43
44 WINE_DEFAULT_DEBUG_CHANNEL(process);
45
46 extern  int __wine_set_signal_handler(unsigned, int (*)(unsigned));
47
48 static ULONGLONG server_start_time;
49
50 /***********************************************************************
51  *           KERNEL thread initialisation routine
52  */
53 static void thread_attach(void)
54 {
55     /* allocate the 16-bit stack (FIXME: should be done lazily) */
56     HGLOBAL16 hstack = WOWGlobalAlloc16( GMEM_FIXED, 0x10000 );
57     kernel_get_thread_data()->stack_sel = GlobalHandleToSel16( hstack );
58     NtCurrentTeb()->WOW32Reserved = (void *)MAKESEGPTR( kernel_get_thread_data()->stack_sel,
59                                                         0x10000 - sizeof(STACK16FRAME) );
60     memset( (char *)GlobalLock16(hstack) + 0x10000 - sizeof(STACK16FRAME), 0, sizeof(STACK16FRAME) );
61 }
62
63
64 /***********************************************************************
65  *           KERNEL thread finalisation routine
66  */
67 static void thread_detach(void)
68 {
69     /* free the 16-bit stack */
70     WOWGlobalFree16( kernel_get_thread_data()->stack_sel );
71     NtCurrentTeb()->WOW32Reserved = 0;
72     if (NtCurrentTeb()->Tib.SubSystemTib) TASK_ExitTask();
73 }
74
75
76 /***********************************************************************
77  *           set_entry_point
78  */
79 static void set_entry_point( HMODULE module, const char *name, DWORD rva )
80 {
81     IMAGE_EXPORT_DIRECTORY *exports;
82     DWORD exp_size;
83
84     if ((exports = RtlImageDirectoryEntryToData( module, TRUE,
85                                                   IMAGE_DIRECTORY_ENTRY_EXPORT, &exp_size )))
86     {
87         DWORD *functions = (DWORD *)((char *)module + exports->AddressOfFunctions);
88         const WORD *ordinals = (const WORD *)((const char *)module + exports->AddressOfNameOrdinals);
89         const DWORD *names = (const DWORD *)((const char *)module +  exports->AddressOfNames);
90         int min = 0, max = exports->NumberOfNames - 1;
91
92         while (min <= max)
93         {
94             int res, pos = (min + max) / 2;
95             const char *ename = (const char *)module + names[pos];
96             if (!(res = strcmp( ename, name )))
97             {
98                 WORD ordinal = ordinals[pos];
99                 assert( ordinal < exports->NumberOfFunctions );
100                 TRACE( "setting %s at %p to %08x\n", name, &functions[ordinal], rva );
101                 functions[ordinal] = rva;
102                 return;
103             }
104             if (res > 0) max = pos - 1;
105             else min = pos + 1;
106         }
107     }
108 }
109
110
111 /***********************************************************************
112  *           KERNEL process initialisation routine
113  */
114 static BOOL process_attach( HMODULE module )
115 {
116     SYSTEM_INFO si;
117     SYSTEM_TIMEOFDAY_INFORMATION ti;
118     RTL_USER_PROCESS_PARAMETERS *params = NtCurrentTeb()->Peb->ProcessParameters;
119
120     /* FIXME: should probably be done in ntdll */
121     GetSystemInfo( &si );
122     NtCurrentTeb()->Peb->NumberOfProcessors = si.dwNumberOfProcessors;
123
124     NtQuerySystemInformation( SystemTimeOfDayInformation, &ti, sizeof(ti), NULL );
125     server_start_time = ti.liKeBootTime.QuadPart;
126
127     /* Setup registry locale information */
128     LOCALE_InitRegistry();
129
130     /* Setup computer name */
131     COMPUTERNAME_Init();
132
133     /* convert value from server:
134      * + 0 => INVALID_HANDLE_VALUE
135      * + console handle needs to be mapped
136      */
137     if (!params->hStdInput)
138         params->hStdInput = INVALID_HANDLE_VALUE;
139     else if (VerifyConsoleIoHandle(console_handle_map(params->hStdInput)))
140         params->hStdInput = console_handle_map(params->hStdInput);
141
142     if (!params->hStdOutput)
143         params->hStdOutput = INVALID_HANDLE_VALUE;
144     else if (VerifyConsoleIoHandle(console_handle_map(params->hStdOutput)))
145         params->hStdOutput = console_handle_map(params->hStdOutput);
146
147     if (!params->hStdError)
148         params->hStdError = INVALID_HANDLE_VALUE;
149     else if (VerifyConsoleIoHandle(console_handle_map(params->hStdError)))
150         params->hStdError = console_handle_map(params->hStdError);
151
152     /* copy process information from ntdll */
153     ENV_CopyStartupInformation();
154
155     if (!(GetVersion() & 0x80000000))
156     {
157         /* Securom checks for this one when version is NT */
158         set_entry_point( module, "FT_Thunk", 0 );
159     }
160 #ifdef __i386__
161     else
162     {
163         /* create the shared heap for broken win95 native dlls */
164         HeapCreate( HEAP_SHARED, 0, 0 );
165         /* setup emulation of protected instructions from 32-bit code */
166         RtlAddVectoredExceptionHandler( TRUE, INSTR_vectored_handler );
167     }
168 #endif
169
170     /* finish the process initialisation for console bits, if needed */
171     __wine_set_signal_handler(SIGINT, CONSOLE_HandleCtrlC);
172
173     if (params->ConsoleHandle == (HANDLE)1)  /* FIXME */
174     {
175         HMODULE mod = GetModuleHandleA(0);
176         if (RtlImageNtHeader(mod)->OptionalHeader.Subsystem == IMAGE_SUBSYSTEM_WINDOWS_CUI)
177             AllocConsole();
178     }
179     /* else TODO for DETACHED_PROCESS:
180      * 1/ inherit console + handles
181      * 2/ create std handles, if handles are not inherited
182      * TBD when not using wineserver handles for console handles
183      */
184
185     /* Create 16-bit task */
186     LoadLibrary16( "krnl386.exe" );
187     thread_attach();
188     TASK_CreateMainTask();
189     return TRUE;
190 }
191
192 /***********************************************************************
193  *           KERNEL initialisation routine
194  */
195 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
196 {
197     switch(reason)
198     {
199     case DLL_PROCESS_ATTACH:
200         return process_attach( hinst );
201     case DLL_THREAD_ATTACH:
202         thread_attach();
203         break;
204     case DLL_THREAD_DETACH:
205         thread_detach();
206         break;
207     case DLL_PROCESS_DETACH:
208         WriteOutProfiles16();
209         break;
210     }
211     return TRUE;
212 }
213
214 /***********************************************************************
215  *           MulDiv   (KERNEL32.@)
216  * RETURNS
217  *      Result of multiplication and division
218  *      -1: Overflow occurred or Divisor was 0
219  */
220 INT WINAPI MulDiv( INT nMultiplicand, INT nMultiplier, INT nDivisor)
221 {
222     LONGLONG ret;
223
224     if (!nDivisor) return -1;
225
226     /* We want to deal with a positive divisor to simplify the logic. */
227     if (nDivisor < 0)
228     {
229       nMultiplicand = - nMultiplicand;
230       nDivisor = -nDivisor;
231     }
232
233     /* If the result is positive, we "add" to round. else, we subtract to round. */
234     if ( ( (nMultiplicand <  0) && (nMultiplier <  0) ) ||
235          ( (nMultiplicand >= 0) && (nMultiplier >= 0) ) )
236       ret = (((LONGLONG)nMultiplicand * nMultiplier) + (nDivisor/2)) / nDivisor;
237     else
238       ret = (((LONGLONG)nMultiplicand * nMultiplier) - (nDivisor/2)) / nDivisor;
239
240     if ((ret > 2147483647) || (ret < -2147483647)) return -1;
241     return ret;
242 }
243
244
245 /******************************************************************************
246  *           GetTickCount64       (KERNEL32.@)
247  */
248 ULONGLONG WINAPI GetTickCount64(void)
249 {
250     LARGE_INTEGER now;
251
252     NtQuerySystemTime( &now );
253     return (now.QuadPart - server_start_time) / 10000;
254 }
255
256
257 /***********************************************************************
258  *           GetTickCount       (KERNEL32.@)
259  *
260  * Get the number of milliseconds the system has been running.
261  *
262  * PARAMS
263  *  None.
264  *
265  * RETURNS
266  *  The current tick count.
267  *
268  * NOTES
269  *  The value returned will wrap around every 2^32 milliseconds.
270  *  Under Windows, tick 0 is the moment at which the system is rebooted.
271  *  Under Wine, tick 0 begins at the moment the wineserver process is started.
272  */
273 DWORD WINAPI GetTickCount(void)
274 {
275     return GetTickCount64();
276 }