thread_win32.cpp

来自「This software aims to create an applet a」· C++ 代码 · 共 164 行

//
// Thread_WIN32.h
//
// $Id: //poco/Main/Foundation/src/Thread_WIN32.cpp#5 $
//
// Copyright (c) 2004, Guenter Obiltschnig/Applied Informatics.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//
// 1. Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//
// 3. Redistributions in any form must be accompanied by information on
//    how to obtain complete source code for this software and any
//    accompanying software that uses this software.  The source code
//    must either be included in the distribution or be available for no
//    more than the cost of distribution plus a nominal fee, and must be
//    freely redistributable under reasonable conditions.  For an
//    executable file, complete source code means the source code for all
//    modules it contains.  It does not include source code for modules or
//    files that typically accompany the major components of the operating
//    system on which the executable file runs.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
// ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//


#include "Foundation/Thread_WIN32.h"
#include "Foundation/Exception.h"
#include <process.h>


Foundation_BEGIN


DWORD ThreadImp::_currentKey = TLS_OUT_OF_INDEXES;


ThreadImp::ThreadImp(): _pTarget(0), _thread(0), _prio(PRIO_NORMAL_IMP)
{
	if (_currentKey == TLS_OUT_OF_INDEXES)
	{
		_currentKey = TlsAlloc();
		if (_currentKey == TLS_OUT_OF_INDEXES)
			throw SystemException("cannot allocate thread context key");
	}
}

			
ThreadImp::~ThreadImp()
{
	if (_thread) CloseHandle(_thread);
}


void ThreadImp::setPriorityImp(int prio)
{
	if (prio != _prio)
	{
		_prio = prio;
		if (_thread)
		{
			if (SetThreadPriority(_thread, _prio) == 0)
				throw SystemException("cannot set thread priority");
		}
	}
}


void ThreadImp::startImp(Runnable& target)
{
	if (_thread) throw SystemException("thread already running");

	_pTarget = &target;
#if defined(_DLL)
	DWORD threadId;
	_thread = CreateThread(NULL, 0, entry, this, 0, &threadId);
#else
	unsigned threadId;
	_thread = (HANDLE) _beginthreadex(NULL, 0, entry, this, 0, &threadId);
#endif
	if (!_thread)
		throw SystemException("cannot create thread");
	if (_prio != PRIO_NORMAL_IMP && !SetThreadPriority(_thread, _prio))
		throw SystemException("cannot set thread priority");
}


void ThreadImp::joinImp()
{
	if (!_thread) return;

	switch (WaitForSingleObject(_thread, INFINITE))
	{
	case WAIT_OBJECT_0:
		return;
	default:
		throw SystemException("cannot join thread");
	}
}


bool ThreadImp::isRunningImp() const
{
	if (_thread)
	{
		DWORD ec = 0;
		return GetExitCodeThread(_thread, &ec) && ec == STILL_ACTIVE;
	}
	return false;
}


ThreadImp* ThreadImp::currentImp()
{
	if (_currentKey == TLS_OUT_OF_INDEXES)
		return 0;
	else
		return (ThreadImp*) TlsGetValue(_currentKey);
}


#if defined(_DLL)
DWORD WINAPI ThreadImp::entry(LPVOID pThread)
#else
unsigned __stdcall ThreadImp::entry(void* pThread)
#endif
{
	TlsSetValue(_currentKey, pThread);
	try
	{
		reinterpret_cast<ThreadImp*>(pThread)->_pTarget->run();
	}
	catch (std::exception&)
	{
	}
	catch (...)
	{
		poco_bugcheck();
	}
	return 0;
}


Foundation_END