riched20: Fix text replacement with text limit.
[wine] / loader / pthread.c
1 /*
2  * Wine threading routines using the pthread library
3  *
4  * Copyright 2003 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 #ifdef HAVE_PTHREAD_H
25
26 #include <assert.h>
27 #include <stdlib.h>
28 #include <signal.h>
29 #ifdef HAVE_UNISTD_H
30 # include <unistd.h>
31 #endif
32 #include <string.h>
33 #include <sys/types.h>
34 #ifdef HAVE_SYS_MMAN_H
35 #include <sys/mman.h>
36 #endif
37 #ifdef HAVE_MACH_MACH_H
38 #include <mach/mach.h>
39 #endif
40
41 #include "wine/library.h"
42 #include "wine/pthread.h"
43
44 static int init_done;
45 static int nb_threads = 1;
46
47 #ifndef __i386__
48 static pthread_key_t teb_key;
49 #endif
50
51 /***********************************************************************
52  *           init_process
53  *
54  * Initialization for a newly created process.
55  */
56 static void init_process( const struct wine_pthread_callbacks *callbacks, size_t size )
57 {
58     init_done = 1;
59 }
60
61
62 /***********************************************************************
63  *           init_thread
64  *
65  * Initialization for a newly created thread.
66  */
67 static void init_thread( struct wine_pthread_thread_info *info )
68 {
69     /* retrieve the stack info (except for main thread) */
70     if (init_done)
71     {
72 #ifdef HAVE_PTHREAD_GETATTR_NP
73         pthread_attr_t attr;
74         pthread_getattr_np( pthread_self(), &attr );
75         pthread_attr_getstack( &attr, &info->stack_base, &info->stack_size );
76         pthread_attr_destroy( &attr );
77 #elif defined(HAVE_PTHREAD_ATTR_GET_NP)
78         pthread_attr_t attr;
79         pthread_attr_init( &attr );
80         pthread_attr_get_np( pthread_self(), &attr );
81         pthread_attr_getstack( &attr, &info->stack_base, &info->stack_size );
82         pthread_attr_destroy( &attr );
83 #elif defined(HAVE_PTHREAD_GET_STACKSIZE_NP) && defined(HAVE_PTHREAD_GET_STACKADDR_NP)
84         char dummy;
85         info->stack_size = pthread_get_stacksize_np(pthread_self());
86         info->stack_base = pthread_get_stackaddr_np(pthread_self());
87         /* if base is too large assume it's the top of the stack instead */
88         if ((char *)info->stack_base > &dummy)
89             info->stack_base = (char *)info->stack_base - info->stack_size;
90 #else
91         /* assume that the stack allocation is page aligned */
92         char dummy;
93         size_t page_size = getpagesize();
94         char *stack_top = (char *)((unsigned long)&dummy & ~(page_size - 1)) + page_size;
95         info->stack_base = stack_top - info->stack_size;
96 #endif
97     }
98 }
99
100
101 /***********************************************************************
102  *           create_thread
103  */
104 static int create_thread( struct wine_pthread_thread_info *info )
105 {
106     pthread_t id;
107     pthread_attr_t attr;
108     int ret = 0;
109
110     pthread_attr_init( &attr );
111     pthread_attr_setstacksize( &attr, info->stack_size );
112     pthread_attr_setdetachstate( &attr, PTHREAD_CREATE_DETACHED );
113     pthread_attr_setscope( &attr, PTHREAD_SCOPE_SYSTEM ); /* force creating a kernel thread on Solaris */
114     interlocked_xchg_add( &nb_threads, 1 );
115     if (pthread_create( &id, &attr, (void * (*)(void *))info->entry, info ))
116     {
117         interlocked_xchg_add( &nb_threads, -1 );
118         ret = -1;
119     }
120     pthread_attr_destroy( &attr );
121     return ret;
122 }
123
124
125 /***********************************************************************
126  *           init_current_teb
127  *
128  * Set the current TEB for a new thread.
129  */
130 static void init_current_teb( struct wine_pthread_thread_info *info )
131 {
132 #ifdef __i386__
133     /* On the i386, the current thread is in the %fs register */
134     LDT_ENTRY fs_entry;
135
136     wine_ldt_set_base( &fs_entry, info->teb_base );
137     wine_ldt_set_limit( &fs_entry, info->teb_size - 1 );
138     wine_ldt_set_flags( &fs_entry, WINE_LDT_FLAGS_DATA|WINE_LDT_FLAGS_32BIT );
139     wine_ldt_init_fs( info->teb_sel, &fs_entry );
140 #else
141     if (!init_done)  /* first thread */
142         pthread_key_create( &teb_key, NULL );
143     pthread_setspecific( teb_key, info->teb_base );
144 #endif
145
146     /* set pid and tid */
147     info->pid = getpid();
148 #ifdef __sun
149     info->tid = pthread_self();  /* this should return the lwp id on solaris */
150 #elif defined(__APPLE__)
151     info->tid = mach_thread_self();
152 #else
153     info->tid = gettid();
154 #endif
155 }
156
157
158 /***********************************************************************
159  *           get_current_teb
160  */
161 static void *get_current_teb(void)
162 {
163 #ifdef __i386__
164     void *ret;
165     __asm__( ".byte 0x64\n\tmovl 0x18,%0" : "=r" (ret) );
166     return ret;
167 #else
168     return pthread_getspecific( teb_key );
169 #endif
170 }
171
172
173 /***********************************************************************
174  *           exit_thread
175  */
176 static void DECLSPEC_NORETURN exit_thread( struct wine_pthread_thread_info *info )
177 {
178     if (interlocked_xchg_add( &nb_threads, -1 ) <= 1) exit( info->exit_status );
179     wine_ldt_free_fs( info->teb_sel );
180     if (info->teb_size) munmap( info->teb_base, info->teb_size );
181     pthread_exit( (void *)info->exit_status );
182 }
183
184
185 /***********************************************************************
186  *           abort_thread
187  */
188 static void DECLSPEC_NORETURN abort_thread( long status )
189 {
190     if (interlocked_xchg_add( &nb_threads, -1 ) <= 1) _exit( status );
191     pthread_exit( (void *)status );
192 }
193
194
195 /***********************************************************************
196  *           pthread_functions
197  */
198 const struct wine_pthread_functions pthread_functions =
199 {
200     init_process,
201     init_thread,
202     create_thread,
203     init_current_teb,
204     get_current_teb,
205     exit_thread,
206     abort_thread,
207     pthread_sigmask
208 };
209
210 #endif  /* HAVE_PTHREAD_H */