thread_nt.h 7.36 KB
Newer Older
Guido van Rossum's avatar
Guido van Rossum committed
1 2 3 4 5 6
/***********************************************************
Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
The Netherlands.

                        All Rights Reserved

7 8
Permission to use, copy, modify, and distribute this software and its
documentation for any purpose and without fee is hereby granted,
Guido van Rossum's avatar
Guido van Rossum committed
9
provided that the above copyright notice appear in all copies and that
10
both that copyright notice and this permission notice appear in
Guido van Rossum's avatar
Guido van Rossum committed
11
supporting documentation, and that the names of Stichting Mathematisch
12 13 14 15
Centrum or CWI or Corporation for National Research Initiatives or
CNRI not be used in advertising or publicity pertaining to
distribution of the software without specific, written prior
permission.
Guido van Rossum's avatar
Guido van Rossum committed
16

17 18 19 20 21 22 23 24 25 26 27 28
While CWI is the initial source for this software, a modified version
is made available by the Corporation for National Research Initiatives
(CNRI) at the Internet address ftp://ftp.python.org.

STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
Guido van Rossum's avatar
Guido van Rossum committed
29 30 31 32 33

******************************************************************/

/* This code implemented by Dag.Gruneau@elsa.preseco.comm.se */

34 35 36
#include <windows.h>
#include <limits.h>
#include <process.h>
Guido van Rossum's avatar
Guido van Rossum committed
37

38
long PyThread_get_thread_ident(void);
Guido van Rossum's avatar
Guido van Rossum committed
39 40 41 42 43 44 45 46 47

/*
 * Change all headers to pure ANSI as no one will use K&R style on an
 * NT
 */

/*
 * Initialization of the C package, should not be needed.
 */
48
static void PyThread__init_thread(void)
Guido van Rossum's avatar
Guido van Rossum committed
49 50 51 52 53 54
{
}

/*
 * Thread support.
 */
55
int PyThread_start_new_thread(void (*func)(void *), void *arg)
Guido van Rossum's avatar
Guido van Rossum committed
56
{
57 58
	long rv;
	int success = 0;
Guido van Rossum's avatar
Guido van Rossum committed
59

60
	dprintf(("%ld: PyThread_start_new_thread called\n", PyThread_get_thread_ident()));
Guido van Rossum's avatar
Guido van Rossum committed
61
	if (!initialized)
62
		PyThread_init_thread();
Guido van Rossum's avatar
Guido van Rossum committed
63

64
	rv = _beginthread(func, 0, arg); /* use default stack size */
Guido van Rossum's avatar
Guido van Rossum committed
65
 
66
	if (rv != -1) {
Guido van Rossum's avatar
Guido van Rossum committed
67
		success = 1;
68
		dprintf(("%ld: PyThread_start_new_thread succeeded: %ld\n", PyThread_get_thread_ident(), aThreadId));
Guido van Rossum's avatar
Guido van Rossum committed
69 70 71 72 73 74 75 76 77
	}

	return success;
}

/*
 * Return the thread Id instead of an handle. The Id is said to uniquely identify the
 * thread in the system
 */
78
long PyThread_get_thread_ident(void)
Guido van Rossum's avatar
Guido van Rossum committed
79 80
{
	if (!initialized)
81
		PyThread_init_thread();
Guido van Rossum's avatar
Guido van Rossum committed
82 83 84 85
        
	return GetCurrentThreadId();
}

86
static void do_PyThread_exit_thread(int no_cleanup)
Guido van Rossum's avatar
Guido van Rossum committed
87
{
88
	dprintf(("%ld: PyThread_exit_thread called\n", PyThread_get_thread_ident()));
Guido van Rossum's avatar
Guido van Rossum committed
89 90 91 92 93
	if (!initialized)
		if (no_cleanup)
			_exit(0);
		else
			exit(0);
94
	_endthread();
Guido van Rossum's avatar
Guido van Rossum committed
95 96
}

97
void PyThread_exit_thread(void)
Guido van Rossum's avatar
Guido van Rossum committed
98
{
99
	do_PyThread_exit_thread(0);
Guido van Rossum's avatar
Guido van Rossum committed
100 101
}

102
void PyThread__exit_thread(void)
Guido van Rossum's avatar
Guido van Rossum committed
103
{
104
	do_PyThread_exit_thread(1);
Guido van Rossum's avatar
Guido van Rossum committed
105 106 107
}

#ifndef NO_EXIT_PROG
108
static void do_PyThread_exit_prog(int status, int no_cleanup)
Guido van Rossum's avatar
Guido van Rossum committed
109
{
110
	dprintf(("PyThread_exit_prog(%d) called\n", status));
Guido van Rossum's avatar
Guido van Rossum committed
111 112 113 114 115 116 117
	if (!initialized)
		if (no_cleanup)
			_exit(status);
		else
			exit(status);
}

118
void PyThread_exit_prog(int status)
Guido van Rossum's avatar
Guido van Rossum committed
119
{
120
	do_PyThread_exit_prog(status, 0);
Guido van Rossum's avatar
Guido van Rossum committed
121 122
}

123
void PyThread__exit_prog _P1(int status)
Guido van Rossum's avatar
Guido van Rossum committed
124
{
125
	do_PyThread_exit_prog(status, 1);
Guido van Rossum's avatar
Guido van Rossum committed
126 127 128 129 130 131 132 133
}
#endif /* NO_EXIT_PROG */

/*
 * Lock support. It has too be implemented as semaphores.
 * I [Dag] tried to implement it with mutex but I could find a way to
 * tell whether a thread already own the lock or not.
 */
134
PyThread_type_lock PyThread_allocate_lock(void)
Guido van Rossum's avatar
Guido van Rossum committed
135 136 137
{
	HANDLE aLock;

138
	dprintf(("PyThread_allocate_lock called\n"));
Guido van Rossum's avatar
Guido van Rossum committed
139
	if (!initialized)
140
		PyThread_init_thread();
Guido van Rossum's avatar
Guido van Rossum committed
141 142 143 144 145 146 147

		aLock = CreateSemaphore(NULL,           /* Security attributes          */
                         1,                     /* Initial value                */
                         1,                     /* Maximum value                */
                         NULL);       
  /* Name of semaphore            */

148
	dprintf(("%ld: PyThread_allocate_lock() -> %lx\n", PyThread_get_thread_ident(), (long)aLock));
Guido van Rossum's avatar
Guido van Rossum committed
149

150
	return (PyThread_type_lock) aLock;
Guido van Rossum's avatar
Guido van Rossum committed
151 152
}

153
void PyThread_free_lock(PyThread_type_lock aLock)
Guido van Rossum's avatar
Guido van Rossum committed
154
{
155
	dprintf(("%ld: PyThread_free_lock(%lx) called\n", PyThread_get_thread_ident(),(long)aLock));
Guido van Rossum's avatar
Guido van Rossum committed
156 157 158 159 160 161 162 163 164 165

	CloseHandle((HANDLE) aLock);
}

/*
 * Return 1 on success if the lock was acquired
 *
 * and 0 if the lock was not acquired. This means a 0 is returned
 * if the lock has already been acquired by this thread!
 */
166
int PyThread_acquire_lock(PyThread_type_lock aLock, int waitflag)
Guido van Rossum's avatar
Guido van Rossum committed
167 168 169 170
{
	int success = 1;
	DWORD waitResult;

171
	dprintf(("%ld: PyThread_acquire_lock(%lx, %d) called\n", PyThread_get_thread_ident(),(long)aLock, waitflag));
Guido van Rossum's avatar
Guido van Rossum committed
172 173 174 175 176 177 178

	waitResult = WaitForSingleObject((HANDLE) aLock, (waitflag == 1 ? INFINITE : 0));

	if (waitResult != WAIT_OBJECT_0) {
		success = 0;    /* We failed */
	}

179
	dprintf(("%ld: PyThread_acquire_lock(%lx, %d) -> %d\n", PyThread_get_thread_ident(),(long)aLock, waitflag, success));
Guido van Rossum's avatar
Guido van Rossum committed
180 181 182 183

	return success;
}

184
void PyThread_release_lock(PyThread_type_lock aLock)
Guido van Rossum's avatar
Guido van Rossum committed
185
{
186
	dprintf(("%ld: PyThread_release_lock(%lx) called\n", PyThread_get_thread_ident(),(long)aLock));
Guido van Rossum's avatar
Guido van Rossum committed
187 188 189 190 191 192

	if (!ReleaseSemaphore(
                        (HANDLE) aLock,                         /* Handle of semaphore                          */
                        1,                                      /* increment count by one                       */
                        NULL))                                  /* not interested in previous count             */
		{
193
		dprintf(("%ld: Could not PyThread_release_lock(%lx) error: %l\n", PyThread_get_thread_ident(), (long)aLock, GetLastError()));
Guido van Rossum's avatar
Guido van Rossum committed
194 195 196 197 198 199
		}
}

/*
 * Semaphore support.
 */
200
PyThread_type_sema PyThread_allocate_sema(int value)
Guido van Rossum's avatar
Guido van Rossum committed
201 202 203
{
	HANDLE aSemaphore;

204
	dprintf(("%ld: PyThread_allocate_sema called\n", PyThread_get_thread_ident()));
Guido van Rossum's avatar
Guido van Rossum committed
205
	if (!initialized)
206
		PyThread_init_thread();
Guido van Rossum's avatar
Guido van Rossum committed
207 208 209 210 211 212

	aSemaphore = CreateSemaphore( NULL,           /* Security attributes          */
                                  value,          /* Initial value                */
                                  INT_MAX,        /* Maximum value                */
                                  NULL);          /* Name of semaphore            */

213
	dprintf(("%ld: PyThread_allocate_sema() -> %lx\n", PyThread_get_thread_ident(), (long)aSemaphore));
Guido van Rossum's avatar
Guido van Rossum committed
214

215
	return (PyThread_type_sema) aSemaphore;
Guido van Rossum's avatar
Guido van Rossum committed
216 217
}

218
void PyThread_free_sema(PyThread_type_sema aSemaphore)
Guido van Rossum's avatar
Guido van Rossum committed
219
{
220
	dprintf(("%ld: PyThread_free_sema(%lx) called\n", PyThread_get_thread_ident(), (long)aSemaphore));
Guido van Rossum's avatar
Guido van Rossum committed
221 222 223 224

	CloseHandle((HANDLE) aSemaphore);
}

225 226 227
/*
  XXX must do something about waitflag
 */
228
int PyThread_down_sema(PyThread_type_sema aSemaphore, int waitflag)
Guido van Rossum's avatar
Guido van Rossum committed
229 230 231
{
	DWORD waitResult;

232
	dprintf(("%ld: PyThread_down_sema(%lx) called\n", PyThread_get_thread_ident(), (long)aSemaphore));
Guido van Rossum's avatar
Guido van Rossum committed
233 234 235

	waitResult = WaitForSingleObject( (HANDLE) aSemaphore, INFINITE);

236
	dprintf(("%ld: PyThread_down_sema(%lx) return: %l\n", PyThread_get_thread_ident(),(long) aSemaphore, waitResult));
237
	return 0;
Guido van Rossum's avatar
Guido van Rossum committed
238 239
}

240
void PyThread_up_sema(PyThread_type_sema aSemaphore)
Guido van Rossum's avatar
Guido van Rossum committed
241 242 243 244 245 246
{
	ReleaseSemaphore(
                (HANDLE) aSemaphore,            /* Handle of semaphore                          */
                1,                              /* increment count by one                       */
                NULL);                          /* not interested in previous count             */
                                                
247
	dprintf(("%ld: PyThread_up_sema(%lx)\n", PyThread_get_thread_ident(), (long)aSemaphore));
Guido van Rossum's avatar
Guido van Rossum committed
248
}