Release 20000801.
[wine] / scheduler / sysdeps.c
1 /*
2  * System-dependent scheduler support
3  *
4  * Copyright 1998 Alexandre Julliard
5  */
6
7 #include "config.h"
8
9 /* Get pointers to the static errno and h_errno variables used by Xlib. This
10    must be done before including <errno.h> makes the variables invisible.  */
11 extern int errno;
12 static int *perrno = &errno;
13 extern int h_errno;
14 static int *ph_errno = &h_errno;
15
16 #include <signal.h>
17 #include <stdio.h>
18 #include <unistd.h>
19 #include <sys/time.h>
20 #include <sys/resource.h>
21 #ifdef HAVE_SYS_SYSCALL_H
22 # include <sys/syscall.h>
23 #endif
24 #ifdef HAVE_SYS_LWP_H
25 # include <sys/lwp.h>
26 #endif
27 #ifdef HAVE_UCONTEXT_H
28 # include <ucontext.h>
29 #endif
30 #include "wine/port.h"
31 #include "thread.h"
32 #include "selectors.h"
33 #include "server.h"
34 #include "winbase.h"
35 #include "global.h"
36 #include "wine/exception.h"
37 #include "debugtools.h"
38
39 DEFAULT_DEBUG_CHANNEL(thread)
40
41 /* Xlib critical section (FIXME: does not belong here) */
42 CRITICAL_SECTION X11DRV_CritSection = CRITICAL_SECTION_INIT;
43
44 #ifdef linux
45 # ifdef HAVE_SCHED_H
46 #  include <sched.h>
47 # endif
48 # ifndef CLONE_VM
49 #  define CLONE_VM      0x00000100
50 #  define CLONE_FS      0x00000200
51 #  define CLONE_FILES   0x00000400
52 #  define CLONE_SIGHAND 0x00000800
53 #  define CLONE_PID     0x00001000
54 # endif  /* CLONE_VM */
55 #endif  /* linux */
56
57 static int init_done;
58
59 #ifndef NO_REENTRANT_LIBC
60
61 /***********************************************************************
62  *           __errno_location/__error/___errno
63  *
64  * Get the per-thread errno location.
65  */
66 #ifdef HAVE__ERRNO_LOCATION
67 int *__errno_location()
68 #endif
69 #ifdef HAVE__ERROR
70 int *__error()
71 #endif
72 #ifdef HAVE___ERRNO
73 int *___errno()
74 #endif
75 #ifdef HAVE__THR_ERRNO
76 int *__thr_errno()
77 #endif
78 {
79     if (!init_done) return perrno;
80 #ifdef NO_REENTRANT_X11
81     /* Use static libc errno while running in Xlib. */
82     if (X11DRV_CritSection.OwningThread == GetCurrentThreadId())
83         return perrno;
84 #endif
85     return &NtCurrentTeb()->thread_errno;
86 }
87
88 /***********************************************************************
89  *           __h_errno_location
90  *
91  * Get the per-thread h_errno location.
92  */
93 int *__h_errno_location()
94 {
95     if (!init_done) return ph_errno;
96 #ifdef NO_REENTRANT_X11
97     /* Use static libc h_errno while running in Xlib. */
98     if (X11DRV_CritSection.OwningThread == GetCurrentThreadId())
99         return ph_errno;
100 #endif
101     return &NtCurrentTeb()->thread_h_errno;
102 }
103
104 #endif /* NO_REENTRANT_LIBC */
105
106 /***********************************************************************
107  *           SYSDEPS_SetCurThread
108  *
109  * Make 'thread' the current thread.
110  */
111 void SYSDEPS_SetCurThread( TEB *teb )
112 {
113 #if defined(__i386__)
114     /* On the i386, the current thread is in the %fs register */
115     __set_fs( teb->teb_sel );
116 #elif defined(HAVE__LWP_CREATE)
117     /* On non-i386 Solaris, we use the LWP private pointer */
118     _lwp_setprivate( teb );
119 #endif
120
121     init_done = 1;  /* now we can use threading routines */
122 }
123
124 /***********************************************************************
125  *           SYSDEPS_StartThread
126  *
127  * Startup routine for a new thread.
128  */
129 static void SYSDEPS_StartThread( TEB *teb )
130 {
131     SYSDEPS_SetCurThread( teb );
132     CLIENT_InitThread();
133     SIGNAL_Init();
134     __TRY
135     {
136         teb->startup();
137     }
138     __EXCEPT(UnhandledExceptionFilter)
139     {
140         TerminateThread( GetCurrentThread(), GetExceptionCode() );
141     }
142     __ENDTRY
143     SYSDEPS_ExitThread(0);  /* should never get here */
144 }
145
146
147 /***********************************************************************
148  *           SYSDEPS_SpawnThread
149  *
150  * Start running a new thread.
151  * Return -1 on error, 0 if OK.
152  */
153 int SYSDEPS_SpawnThread( TEB *teb )
154 {
155 #ifndef NO_REENTRANT_LIBC
156
157 #ifdef linux
158     const int flags = CLONE_VM | CLONE_FS | CLONE_FILES | SIGCHLD;
159     if (clone( (int (*)(void *))SYSDEPS_StartThread, teb->stack_top, flags, teb ) < 0)
160         return -1;
161     if (!(flags & CLONE_FILES)) close( teb->socket );  /* close the child socket in the parent */
162     return 0;
163 #endif
164
165 #ifdef HAVE_RFORK
166     const int flags = RFPROC | RFMEM; /*|RFFDG*/
167     void **sp = (void **)teb->stack_top;
168     *--sp = teb;
169     *--sp = 0;
170     *--sp = SYSDEPS_StartThread;
171     __asm__ __volatile__(
172     "pushl %2;\n\t"             /* flags */
173     "pushl $0;\n\t"             /* 0 ? */
174     "movl %1,%%eax;\n\t"        /* SYS_rfork */
175     ".byte 0x9a; .long 0; .word 7;\n\t" /* lcall 7:0... FreeBSD syscall */
176     "cmpl $0, %%edx;\n\t"
177     "je 1f;\n\t"
178     "movl %0,%%esp;\n\t"        /* child -> new thread */
179     "ret;\n"
180     "1:\n\t"            /* parent -> caller thread */
181     "addl $8,%%esp" :
182     : "r" (sp), "g" (SYS_rfork), "g" (flags)
183     : "eax", "edx");
184     if (flags & RFFDG) close( teb->socket );  /* close the child socket in the parent */
185     return 0;
186 #endif
187
188 #ifdef HAVE__LWP_CREATE
189     ucontext_t context;
190     _lwp_makecontext( &context, (void(*)(void *))SYSDEPS_StartThread, teb,
191                       NULL, teb->stack_base, (char *)teb->stack_top - (char *)teb->stack_base );
192     if ( _lwp_create( &context, 0, NULL ) )
193         return -1;
194     return 0;
195 #endif
196
197 #endif /* NO_REENTRANT_LIBC */
198
199     FIXME("CreateThread: stub\n" );
200     return 0;
201 }
202
203
204
205 /***********************************************************************
206  *           SYSDEPS_ExitThread
207  *
208  * Exit a running thread; must not return.
209  */
210 void SYSDEPS_ExitThread( int status )
211 {
212     int socket = NtCurrentTeb()->socket;
213     NtCurrentTeb()->socket = -1;
214     close( socket );
215 #ifdef HAVE__LWP_CREATE
216     _lwp_exit();
217 #endif
218     _exit( status );
219     /*
220      * It is of course impossible to come here,
221      * but it eliminates a compiler warning.
222      */
223     exit( status );
224 }
225
226
227 /***********************************************************************
228  *           SYSDEPS_CallOnStack
229  */
230 int SYSDEPS_DoCallOnStack( int (*func)(LPVOID), LPVOID arg )
231 {
232     int retv = 0;
233
234     __TRY
235     {
236         retv = func( arg );
237     }
238     __EXCEPT(UnhandledExceptionFilter)
239     {
240         TerminateThread( GetCurrentThread(), GetExceptionCode() );
241         return 0;
242     }
243     __ENDTRY
244
245     return retv;
246 }
247
248 #ifdef __i386__
249 int SYSDEPS_CallOnStack( LPVOID stackTop, LPVOID stackLow,
250                          int (*func)(LPVOID), LPVOID arg );
251 __ASM_GLOBAL_FUNC( SYSDEPS_CallOnStack,
252                    "pushl %ebp\n\t"
253                    "movl %esp, %ebp\n\t"
254                    ".byte 0x64; pushl 0x04\n\t"
255                    ".byte 0x64; pushl 0x08\n\t"
256                    "movl 8(%ebp), %esp\n\t"
257                    "movl 12(%ebp), %eax\n\t"
258                    ".byte 0x64; movl %esp, 0x04\n\t"
259                    ".byte 0x64; movl %eax, 0x08\n\t"
260                    "pushl 20(%ebp)\n\t"
261                    "pushl 16(%ebp)\n\t"
262                    "call " __ASM_NAME("SYSDEPS_DoCallOnStack") "\n\t"
263                    "leal -8(%ebp), %esp\n\t"
264                    ".byte 0x64; popl 0x08\n\t"
265                    ".byte 0x64; popl 0x04\n\t"
266                    "popl %ebp\n\t"
267                    "ret" );
268 #else
269 int SYSDEPS_CallOnStack( LPVOID stackTop, LPVOID stackLow,
270                          int (*func)(LPVOID), LPVOID arg )
271 {
272     return SYSDEPS_DoCallOnStack( func, arg );
273 }
274 #endif
275
276 /***********************************************************************
277  *           SYSDEPS_SwitchToThreadStack
278  */
279
280 static LPVOID SYSDEPS_LargeStackTop = NULL;
281 static LPVOID SYSDEPS_LargeStackLow = NULL;
282
283 void SYSDEPS_SwitchToThreadStack( void (*func)(void) )
284 {
285     DWORD page_size = VIRTUAL_GetPageSize();
286     DWORD cur_stack = (((DWORD)&func) + (page_size-1)) & ~(page_size-1);
287
288     TEB *teb = NtCurrentTeb();
289     LPVOID stackTop = teb->stack_top;
290     LPVOID stackLow = teb->stack_low;
291
292     struct rlimit rl;
293
294     if ( getrlimit(RLIMIT_STACK, &rl) < 0 ) 
295     {
296         WARN("Can't get rlimit\n");
297         rl.rlim_cur = 8*1024*1024;
298     }
299
300     teb->stack_top = (LPVOID) cur_stack;
301     teb->stack_low = (LPVOID)(cur_stack - rl.rlim_cur);
302
303     SYSDEPS_LargeStackTop = (LPVOID)(cur_stack - 2*page_size);
304     SYSDEPS_LargeStackLow = (LPVOID)(cur_stack - rl.rlim_cur);
305
306     SYSDEPS_CallOnStack( stackTop, stackLow, 
307                          (int (*)(void *))func, NULL );
308 }
309
310 /***********************************************************************
311  *           SYSDEPS_CallOnLargeStack
312  */
313 int SYSDEPS_CallOnLargeStack( int (*func)(LPVOID), LPVOID arg )
314 {
315     static int recurse = 0;
316     int retv;
317
318     if ( recurse++ == 0 && SYSDEPS_LargeStackTop )
319         retv = SYSDEPS_CallOnStack( SYSDEPS_LargeStackTop,
320                                     SYSDEPS_LargeStackLow, func, arg );
321     else
322         retv = func( arg );
323
324     recurse--;
325     return retv;
326 }
327
328
329 /**********************************************************************
330  *           NtCurrentTeb   (NTDLL.89)
331  *
332  * This will crash and burn if called before threading is initialized
333  */
334 #ifdef __i386__
335 __ASM_GLOBAL_FUNC( NtCurrentTeb, ".byte 0x64\n\tmovl 0x18,%eax\n\tret" );
336 #elif defined(HAVE__LWP_CREATE)
337 struct _TEB * WINAPI NtCurrentTeb(void)
338 {
339     extern void *_lwp_getprivate(void);
340     return (struct _TEB *)_lwp_getprivate();
341 }
342 #else
343 # error NtCurrentTeb not defined for this architecture
344 #endif  /* __i386__ */