blob: f1174c4fe41c0979c923ba908557c4cea4b85fbe (
plain)
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
|
#pragma once
#include <Main.h>
namespace Balau {
class TLSManager {
public:
TLSManager() { }
virtual void * getTLS();
virtual void * setTLS(void * val);
private:
TLSManager(const TLSManager &) = delete;
TLSManager & operator=(const TLSManager &) = delete;
};
class PThreadsTLSManager : public TLSManager {
public:
virtual void * getTLS();
virtual void * setTLS(void * val);
void init();
private:
pthread_key_t m_key;
};
extern TLSManager * g_tlsManager;
class Local : public AtStart {
public:
static int getSize() { return s_size; }
static void * createTLS(void * c = NULL) {
void * r = calloc(s_size * sizeof(void *), 1);
if (c)
memcpy(r, c, s_size * sizeof(void *));
return r;
}
protected:
Local() : AtStart(0) { }
void * getGlobal() { return m_globals[m_idx]; }
void * getLocal() { return reinterpret_cast<void **>(getTLS())[m_idx]; }
void * get() { if (getTLS()) { void * l = getLocal(); return l ? l : getGlobal(); } else return getGlobal(); }
void setGlobal(void * obj) { m_globals[m_idx] = obj; }
void setLocal(void * obj) { void * r = getTLS(); reinterpret_cast<void **>(r)[m_idx] = obj; }
void set(void * obj) { void * r = getTLS(); if (r) setLocal(obj); else setGlobal(obj); }
int getIndex() { return m_idx; }
private:
static void * getTLS() { return g_tlsManager->getTLS(); }
static void * setTLS(void * val) { return g_tlsManager->setTLS(val); }
virtual void doStart();
int m_idx;
static int s_size;
static void ** m_globals;
friend class TLSManager;
};
template<class T>
class DefaultTmpl : public AtStart {
public:
DefaultTmpl(int pri) : AtStart(pri) { }
protected:
virtual void doStart() { new T; }
};
template<class T>
class LocalTmpl : public Local {
public:
LocalTmpl() { }
T * getGlobal() { return reinterpret_cast<T *>(Local::getGlobal()); }
T * get() { return reinterpret_cast<T *>(Local::get()); }
void setGlobal(T * obj) { Local::setGlobal(obj); }
void set(T * obj) { Local::set(obj); }
};
};
|
