Import initial version of 1:1 pthread library.
[dragonfly.git] / lib / libthread_xu / arch / i386 / i386 / pthread_md.c
1 /*-
2  * Copyright (C) 2003 David Xu <davidxu@freebsd.org>
3  * Copyright (c) 2001,2003 Daniel Eischen <deischen@freebsd.org>
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Neither the name of the author nor the names of its contributors
12  *    may be used to endorse or promote products derived from this software
13  *    without specific prior written permission.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25  * SUCH DAMAGE.
26  *
27  * $DragonFly: src/lib/libthread_xu/arch/i386/i386/pthread_md.c,v 1.1 2005/02/01 12:38:27 davidxu Exp $
28  */
29
30 #include <sys/cdefs.h>
31 #include <sys/types.h>
32 #include <machine/cpufunc.h>
33 #include <machine/segments.h>
34 #include <machine/sysarch.h>
35
36 #include <unistd.h>
37 #include <signal.h>
38 #include <stdlib.h>
39 #include <string.h>
40
41 #ifndef __DragonFly__
42 #include "rtld_tls.h"
43 #endif
44
45 #include "pthread_md.h"
46
47 #ifdef __DragonFly__
48
49 #define LDT_ENTRIES 8192
50 #define LDT_WORDS   (8192/sizeof(unsigned int))
51 #define LDT_RESERVED NLDT
52
53 static unsigned int ldt_mask[LDT_WORDS];
54 static int initialized = 0;
55
56 static void     initialize(void);
57
58 static void
59 initialize(void)
60 {
61         int i, j;
62
63         memset(ldt_mask, 0xFF, sizeof(ldt_mask));
64         /* Reserve system predefined LDT entries */
65         for (i = 0; i < LDT_RESERVED; ++i) {
66                 j = i / 32;
67                 ldt_mask[j] &= ~(1 << (i % 32));
68         }
69         initialized = 1;
70 }
71
72 static u_int
73 alloc_ldt_entry(void)
74 {
75         u_int i, j, index;
76         
77         index = 0;
78         for (i = 0; i < LDT_WORDS; ++i) {
79                 if (ldt_mask[i] != 0) {
80                         j = bsfl(ldt_mask[i]);
81                         ldt_mask[i] &= ~(1 << j);
82                         index = i * 32 + j;
83                         break;
84                 }
85         }
86         return (index);
87 }
88
89 static void
90 free_ldt_entry(u_int index)
91 {
92         u_int i, j;
93
94         if (index < LDT_RESERVED || index >= LDT_ENTRIES)
95                 return;
96         i = index / 32;
97         j = index % 32;
98         ldt_mask[i] |= (1 << j);
99 }
100
101 #endif
102
103 struct tcb *
104 _tcb_ctor(struct pthread *thread, int initial)
105 {
106 #ifndef COMPAT_32BIT
107         union descriptor ldt;
108 #endif
109         struct tcb *tcb;
110
111 #ifndef __DragonFly__
112         void *oldtls;
113
114         if (initial)
115                 __asm __volatile("movl %%gs:0, %0" : "=r" (oldtls));
116         else
117                 oldtls = NULL;
118
119         tcb = _rtld_allocate_tls(oldtls, sizeof(struct tcb), 16);
120 #else
121         if (!initialized)
122                 initialize();
123         if ((tcb = malloc(sizeof(struct tcb))) == NULL)
124                 return (NULL);
125         /* tcb_self and tcb_dtv should assigned by rtld tls code. */
126         tcb->tcb_self = tcb;
127         tcb->tcb_dtv = NULL;
128         if ((tcb->tcb_ldt = alloc_ldt_entry()) == 0) {
129                 free(tcb);
130                 return (NULL);
131         }
132 #endif
133
134         if (tcb) {
135                 tcb->tcb_thread = thread;
136 #ifndef COMPAT_32BIT
137                 ldt.sd.sd_hibase = (unsigned int)tcb >> 24;
138                 ldt.sd.sd_lobase = (unsigned int)tcb & 0xFFFFFF;
139                 ldt.sd.sd_hilimit = (sizeof(struct tcb) >> 16) & 0xF;
140                 ldt.sd.sd_lolimit = sizeof(struct tcb) & 0xFFFF;
141                 ldt.sd.sd_type = SDT_MEMRWA;
142                 ldt.sd.sd_dpl = SEL_UPL;
143                 ldt.sd.sd_p = 1;
144                 ldt.sd.sd_xx = 0;
145                 ldt.sd.sd_def32 = 1;
146                 ldt.sd.sd_gran = 0;     /* no more than 1M */
147
148 #ifdef __DragonFly__
149                 if (i386_set_ldt(tcb->tcb_ldt, &ldt, 1) == -1) { 
150                         free_ldt_entry(tcb->tcb_ldt);
151                         free(tcb);
152                         tcb = NULL;
153                 }
154 #else
155                 tcb->tcb_ldt = i386_set_ldt(LDT_AUTO_ALLOC, &ldt, 1);
156                 if (tcb->tcb_ldt < 0) {
157                         free(tcb);
158                         tcb = NULL;
159                 }
160 #endif
161 #endif
162         }
163         return (tcb);
164 }
165
166 void
167 _tcb_dtor(struct tcb *tcb)
168 {
169 #ifdef __DragonFly__
170         if (tcb->tcb_ldt > 0) {
171                 free_ldt_entry(tcb->tcb_ldt);
172                 free(tcb);      
173         }
174 #else
175 #ifndef COMPAT_32BIT
176         if (tcb->tcb_ldt > 0)
177                 i386_set_ldt(tcb->tcb_ldt, NULL, 1);
178 #endif
179         _rtld_free_tls(tcb, sizeof(struct tcb), 16);
180 #endif
181 }