1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
/* $Id: shthread.c 3505 2021-12-15 22:53:57Z bird $ */
/** @file
*
* Shell Thread Management.
*
* Copyright (c) 2007-2010 knut st. osmundsen <bird-kBuild-spamx@anduin.net>
*
*
* This file is part of kBuild.
*
* kBuild is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* kBuild is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with kBuild; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
#include "shthread.h"
#include "shinstance.h"
#if K_OS == K_OS_WINDOWS
# include <Windows.h>
#elif K_OS == K_OS_OS2
# include <InnoTekLIBC/FastInfoBlocks.h>
# include <InnoTekLIBC/thread.h>
#else
# include <pthread.h>
#endif
/*******************************************************************************
* Global Variables *
*******************************************************************************/
#if K_OS == K_OS_WINDOWS
static DWORD sh_tls = TLS_OUT_OF_INDEXES;
#elif K_OS == K_OS_OS2
static int sh_tls = -1;
#else
static int sh_tls_inited = 0;
static pthread_key_t sh_tls;
#endif
/**
* Stores the shell instance pointer in a TLS entry.
*
* This will allocate the TLS entry on the first call. We assume
* there will no be races at that time.
*
* @param psh The shell instance.
*/
void shthread_set_shell(struct shinstance *psh)
{
#if K_OS == K_OS_WINDOWS
if (sh_tls == TLS_OUT_OF_INDEXES)
{
sh_tls = TlsAlloc();
kHlpAssert(sh_tls != TLS_OUT_OF_INDEXES);
}
if (!TlsSetValue(sh_tls, psh))
kHlpAssert(0);
#elif K_OS == K_OS_OS2
if (sh_tls == -1)
{
sh_tls = __libc_TLSAlloc();
kHlpAssert(sh_tls != -1);
}
if (__libc_TLSSet(sh_tls, psh) == -1)
kHlpAssert(0);
#else
if (!sh_tls_inited)
{
if (pthread_key_create(&sh_tls, NULL) != 0)
kHlpAssert(0);
sh_tls_inited = 1;
}
if (pthread_setspecific(sh_tls, psh) != 0)
kHlpAssert(0);
#endif
}
/**
* Get the shell instance pointer from TLS.
*
* @returns The shell instance.
*/
struct shinstance *shthread_get_shell(void)
{
shinstance *psh;
#if K_OS == K_OS_WINDOWS
psh = (shinstance *)TlsGetValue(sh_tls);
#elif K_OS == K_OS_OS2
psh = (shinstance *)__libc_TLSGet(sh_tls);
#else
psh = (shinstance *)pthread_getspecific(sh_tls);
#endif
return psh;
}
/**
* Sets the name of the current thread if supported by the OS.
*/
void shthread_set_name(const char *name)
{
#if K_OS == K_OS_WINDOWS
typedef BOOL (WINAPI * PFNSETTHREADDESCRIPTION)(HANDLE, WCHAR *);
static KBOOL volatile s_initialized = K_FALSE;
static PFNSETTHREADDESCRIPTION volatile s_pfnSetThreadDescription = NULL;
PFNSETTHREADDESCRIPTION pfnSetThreadDescription = s_pfnSetThreadDescription;
WCHAR wszName[32];
size_t i;
/* Get the function pointer, return if not available. */
if (pfnSetThreadDescription)
{ }
else if (s_initialized)
return;
else
{
pfnSetThreadDescription = (PFNSETTHREADDESCRIPTION)GetProcAddress(GetModuleHandleW(L"KERNEL32.DLL"),
"SetThreadDescription");
s_pfnSetThreadDescription = pfnSetThreadDescription;
s_initialized = K_TRUE;
if (!pfnSetThreadDescription)
return;
}
/* Convert the name to UTF-16 and call the API. */
i = strlen(name);
kHlpAssertStmt(i < K_ELEMENTS(wszName), i = K_ELEMENTS(wszName));
wszName[i] = '\0';
while (i-- > 0)
wszName[i] = name[i];
pfnSetThreadDescription(GetCurrentThread(), wszName);
#else
K_NOREF(name);
#endif
}
|
