Charles Davis | 53ca1f3 | 2010-11-29 19:44:50 +0000 | [diff] [blame] | 1 | //= llvm/Support/Win32/ThreadLocal.inc - Win32 Thread Local Data -*- C++ -*-===// |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 2 | // |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 7 | // |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file implements the Win32 specific (non-pthread) ThreadLocal class. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | //=== WARNING: Implementation here must contain only generic Win32 code that |
| 16 | //=== is guaranteed to work on *all* Win32 variants. |
| 17 | //===----------------------------------------------------------------------===// |
| 18 | |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 19 | #include "Windows.h" |
| 20 | #include "llvm/Support/ThreadLocal.h" |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 21 | |
| 22 | namespace llvm { |
| 23 | using namespace sys; |
| 24 | |
| 25 | ThreadLocalImpl::ThreadLocalImpl() { |
| 26 | DWORD* tls = new DWORD; |
| 27 | *tls = TlsAlloc(); |
| 28 | assert(*tls != TLS_OUT_OF_INDEXES); |
| 29 | data = tls; |
| 30 | } |
| 31 | |
| 32 | ThreadLocalImpl::~ThreadLocalImpl() { |
| 33 | DWORD* tls = static_cast<DWORD*>(data); |
| 34 | TlsFree(*tls); |
| 35 | delete tls; |
| 36 | } |
| 37 | |
Owen Anderson | 438d394 | 2009-06-25 23:31:18 +0000 | [diff] [blame] | 38 | const void* ThreadLocalImpl::getInstance() { |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 39 | DWORD* tls = static_cast<DWORD*>(data); |
| 40 | return TlsGetValue(*tls); |
| 41 | } |
| 42 | |
Owen Anderson | 438d394 | 2009-06-25 23:31:18 +0000 | [diff] [blame] | 43 | void ThreadLocalImpl::setInstance(const void* d){ |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 44 | DWORD* tls = static_cast<DWORD*>(data); |
Duncan Sands | 05897c8 | 2010-10-25 13:10:03 +0000 | [diff] [blame] | 45 | int errorcode = TlsSetValue(*tls, const_cast<void*>(d)); |
Oscar Fuentes | 2b1a6e7 | 2009-06-30 14:12:28 +0000 | [diff] [blame] | 46 | assert(errorcode != 0); |
Duncan Sands | 05897c8 | 2010-10-25 13:10:03 +0000 | [diff] [blame] | 47 | (void)errorcode; |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 48 | } |
| 49 | |
Owen Anderson | 826c148 | 2010-07-28 22:49:43 +0000 | [diff] [blame] | 50 | void ThreadLocalImpl::removeInstance() { |
| 51 | setInstance(0); |
| 52 | } |
| 53 | |
Owen Anderson | 4a28522 | 2009-06-25 21:58:01 +0000 | [diff] [blame] | 54 | } |