[CONSOLE.CPL-KERNEL32-USER32-NTDLL-CSRSRV-CONSRV-BASESRV-WINSRV]
[reactos.git] / dll / win32 / kernel32 / client / dllmain.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS system libraries
4 * FILE: lib/kernel32/misc/dllmain.c
5 * PURPOSE: Initialization
6 * PROGRAMMERS: Ariadne (ariadne@xs4all.nl)
7 * Aleksey Bragin (aleksey@reactos.org)
8 */
9
10 /* INCLUDES ******************************************************************/
11
12 #include <k32.h>
13
14 #define NDEBUG
15 #include <debug.h>
16
17 /* GLOBALS *******************************************************************/
18
19 PBASE_STATIC_SERVER_DATA BaseStaticServerData;
20
21 BOOLEAN BaseRunningInServerProcess;
22
23 WCHAR BaseDefaultPathBuffer[6140];
24
25 HANDLE BaseNamedObjectDirectory;
26 HMODULE hCurrentModule = NULL;
27 HMODULE kernel32_handle = NULL;
28 PPEB Peb;
29 ULONG SessionId;
30 static BOOL DllInitialized = FALSE;
31
32 /* Critical section for various kernel32 data structures */
33 RTL_CRITICAL_SECTION BaseDllDirectoryLock;
34
35 extern BOOL FASTCALL NlsInit(VOID);
36 extern VOID FASTCALL NlsUninit(VOID);
37
38 #define WIN_OBJ_DIR L"\\Windows"
39 #define SESSION_DIR L"\\Sessions"
40
41 /* FUNCTIONS *****************************************************************/
42
43 NTSTATUS
44 NTAPI
45 BaseCreateThreadPoolThread(IN PTHREAD_START_ROUTINE Function,
46 IN PVOID Parameter,
47 OUT PHANDLE ThreadHandle)
48 {
49 NTSTATUS Status;
50
51 /* Create a Win32 thread */
52 *ThreadHandle = CreateRemoteThread(NtCurrentProcess(),
53 NULL,
54 0,
55 Function,
56 Parameter,
57 CREATE_SUSPENDED,
58 NULL);
59 if (!(*ThreadHandle))
60 {
61 /* Get the status value if we couldn't get a handle */
62 Status = NtCurrentTeb()->LastStatusValue;
63 if (NT_SUCCESS(Status)) Status = STATUS_UNSUCCESSFUL;
64 }
65 else
66 {
67 /* Set success code */
68 Status = STATUS_SUCCESS;
69 }
70
71 /* All done */
72 return Status;
73 }
74
75 NTSTATUS
76 NTAPI
77 BaseExitThreadPoolThread(IN NTSTATUS ExitStatus)
78 {
79 /* Exit the thread */
80 ExitThread(ExitStatus);
81 return STATUS_SUCCESS;
82 }
83
84 BOOL
85 WINAPI
86 DllMain(HANDLE hDll,
87 DWORD dwReason,
88 LPVOID lpReserved)
89 {
90 NTSTATUS Status;
91 ULONG Dummy;
92 ULONG DummySize = sizeof(Dummy);
93 WCHAR SessionDir[256];
94
95 DPRINT("DllMain(hInst %lx, dwReason %lu)\n",
96 hDll, dwReason);
97
98 Basep8BitStringToUnicodeString = RtlAnsiStringToUnicodeString;
99
100 /* Cache the PEB and Session ID */
101 Peb = NtCurrentPeb();
102 SessionId = Peb->SessionId;
103
104 switch (dwReason)
105 {
106 case DLL_PROCESS_ATTACH:
107 {
108 /* Set no filter intially */
109 GlobalTopLevelExceptionFilter = RtlEncodePointer(NULL);
110
111 /* Enable the Rtl thread pool and timer queue to use proper Win32 thread */
112 RtlSetThreadPoolStartFunc(BaseCreateThreadPoolThread, BaseExitThreadPoolThread);
113
114 /* Don't bother us for each thread */
115 LdrDisableThreadCalloutsForDll((PVOID)hDll);
116
117 /* Initialize default path to NULL */
118 RtlInitUnicodeString(&BaseDefaultPath, NULL);
119
120 /* Setup the right Object Directory path */
121 if (!SessionId)
122 {
123 /* Use the raw path */
124 wcscpy(SessionDir, WIN_OBJ_DIR);
125 }
126 else
127 {
128 /* Use the session path */
129 swprintf(SessionDir,
130 L"%ws\\%ld%ws",
131 SESSION_DIR,
132 SessionId,
133 WIN_OBJ_DIR);
134 }
135
136 /* Connect to the base server */
137 Status = CsrClientConnectToServer(SessionDir,
138 BASESRV_SERVERDLL_INDEX,
139 &Dummy,
140 &DummySize,
141 &BaseRunningInServerProcess);
142 if (!NT_SUCCESS(Status))
143 {
144 DPRINT1("Failed to connect to CSR (Status %lx)\n", Status);
145 NtTerminateProcess(NtCurrentProcess(), Status);
146 return FALSE;
147 }
148
149 /* Get the server data */
150 ASSERT(Peb->ReadOnlyStaticServerData);
151 BaseStaticServerData = Peb->ReadOnlyStaticServerData[BASESRV_SERVERDLL_INDEX];
152 ASSERT(BaseStaticServerData);
153
154 /* Check if we are running a CSR Server */
155 if (!BaseRunningInServerProcess)
156 {
157 /* Set the termination port for the thread */
158 DPRINT("Creating new thread for CSR\n");
159 CsrNewThread();
160 }
161
162 /* Initialize heap handle table */
163 BaseDllInitializeMemoryManager();
164
165 /* Set HMODULE for our DLL */
166 kernel32_handle = hCurrentModule = hDll;
167
168 /* Set the directories */
169 BaseWindowsDirectory = BaseStaticServerData->WindowsDirectory;
170 BaseWindowsSystemDirectory = BaseStaticServerData->WindowsSystemDirectory;
171
172 /* Construct the default path (using the static buffer) */
173 _snwprintf(BaseDefaultPathBuffer,
174 sizeof(BaseDefaultPathBuffer) / sizeof(WCHAR),
175 L".;%wZ;%wZ\\system;%wZ;",
176 &BaseWindowsSystemDirectory,
177 &BaseWindowsDirectory,
178 &BaseWindowsDirectory);
179
180 BaseDefaultPath.Buffer = BaseDefaultPathBuffer;
181 BaseDefaultPath.Length = wcslen(BaseDefaultPathBuffer) * sizeof(WCHAR);
182 BaseDefaultPath.MaximumLength = sizeof(BaseDefaultPathBuffer);
183
184 /* Use remaining part of the default path buffer for the append path */
185 BaseDefaultPathAppend.Buffer = (PWSTR)((ULONG_PTR)BaseDefaultPathBuffer + BaseDefaultPath.Length);
186 BaseDefaultPathAppend.Length = 0;
187 BaseDefaultPathAppend.MaximumLength = BaseDefaultPath.MaximumLength - BaseDefaultPath.Length;
188
189 /* Initialize command line */
190 InitCommandLines();
191
192 /* Initialize the DLL critical section */
193 RtlInitializeCriticalSection(&BaseDllDirectoryLock);
194
195 /* Initialize the National Language Support routines */
196 if (!NlsInit())
197 {
198 DPRINT1("NLS Init failed\n");
199 return FALSE;
200 }
201
202 /* Initialize Console Support */
203 if (!BasepInitConsole())
204 {
205 DPRINT1("Failed to set up console\n");
206 return FALSE;
207 }
208
209 /* Initialize application certification globals */
210 InitializeListHead(&BasepAppCertDllsList);
211 RtlInitializeCriticalSection(&gcsAppCert);
212
213 /* Insert more dll attach stuff here! */
214 DllInitialized = TRUE;
215 break;
216 }
217
218 case DLL_PROCESS_DETACH:
219 {
220 if (DllInitialized == TRUE)
221 {
222 /* Insert more dll detach stuff here! */
223 NlsUninit();
224
225 /* Uninitialize console support */
226 BasepUninitConsole();
227
228 /* Delete DLL critical section */
229 RtlDeleteCriticalSection(&BaseDllDirectoryLock);
230 }
231 break;
232 }
233
234 default:
235 break;
236 }
237
238 return TRUE;
239 }
240
241 /* EOF */