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