X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fthreadengines%2Fthreadengine_pthread.cpp;h=8527907c491ca249dc40693e7e371ee5ac912e10;hb=4ec65c6231df9fcb38210f9b885cdf73b72cc176;hp=2f09cc30556e7b4cfe189bf4ed675ecc6954963c;hpb=e71625d79ce4bf025125a03b1dfaf5cf9086c531;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/threadengines/threadengine_pthread.cpp b/src/threadengines/threadengine_pthread.cpp index 2f09cc305..8527907c4 100644 --- a/src/threadengines/threadengine_pthread.cpp +++ b/src/threadengines/threadengine_pthread.cpp @@ -1,107 +1,178 @@ -/* +------------------------------------+ - * | Inspire Internet Relay Chat Daemon | - * +------------------------------------+ +/* + * InspIRCd -- Internet Relay Chat Daemon * - * InspIRCd: (C) 2002-2008 InspIRCd Development Team - * See: http://www.inspircd.org/wiki/index.php/Credits + * Copyright (C) 2009-2010 Daniel De Graaf + * Copyright (C) 2008 Craig Edwards * - * This program is free but copyrighted software; see - * the file COPYING for details. + * This file is part of InspIRCd. InspIRCd 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, version 2. * - * --------------------------------------------------- + * This program 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 this program. If not, see . */ + #include "inspircd.h" #include "threadengines/threadengine_pthread.h" #include +#include +#include -pthread_mutex_t MyMutex = PTHREAD_MUTEX_INITIALIZER; - -PThreadEngine::PThreadEngine(InspIRCd* Instance) : ThreadEngine(Instance) +ThreadEngine::ThreadEngine() { } -void PThreadEngine::Create(Thread* thread_to_init) +static void* entry_point(void* parameter) { - pthread_attr_t attribs; - pthread_attr_init(&attribs); - pthread_attr_setdetachstate(&attribs, PTHREAD_CREATE_JOINABLE); - pthread_t* MyPThread = new pthread_t; - - /* Create a thread in a mutex. This prevents whacking the member value NewThread, - * and also prevents recursive creation of threads by mistake (instead, the thread - * will just deadlock itself) - */ - Mutex(true); - - if (pthread_create(MyPThread, &attribs, PThreadEngine::Entry, (void*)this) != 0) - { - delete MyPThread; - Mutex(false); - throw CoreException("Unable to create new PThreadEngine: " + std::string(strerror(errno))); - } + /* Recommended by nenolod, signal safety on a per-thread basis */ + sigset_t set; + sigemptyset(&set); + sigaddset(&set, SIGPIPE); + pthread_sigmask(SIG_BLOCK, &set, NULL); - pthread_attr_destroy(&attribs); + Thread* pt = static_cast(parameter); + pt->Run(); + return parameter; +} - NewThread = thread_to_init; - NewThread->Creator = this; - NewThread->Extend("pthread", MyPThread); - /* Always unset a mutex if you set it */ - Mutex(false); +void ThreadEngine::Start(Thread* thread) +{ + ThreadData* data = new ThreadData; + thread->state = data; - /* Wait for the PThreadEngine::Run method to take a copy of the - * pointer and clear this member value - */ - while (NewThread) - usleep(1000); + if (pthread_create(&data->pthread_id, NULL, entry_point, thread) != 0) + { + thread->state = NULL; + delete data; + throw CoreException("Unable to create new thread: " + std::string(strerror(errno))); + } } -PThreadEngine::~PThreadEngine() +ThreadEngine::~ThreadEngine() { } -void PThreadEngine::Run() +void ThreadData::FreeThread(Thread* thread) { - /* Take a copy of the member value, then clear it. Do this - * in a mutex so that we can be sure nothing else is looking - * at it. - */ - Mutex(true); - Thread* nt = NewThread; - NewThread = NULL; - Mutex(false); - /* Now we have our own safe copy, call the object on it */ - nt->Run(); + thread->SetExitFlag(); + pthread_join(pthread_id, NULL); } -bool PThreadEngine::Mutex(bool enable) +#ifdef HAS_EVENTFD +#include + +class ThreadSignalSocket : public EventHandler { - if (enable) - pthread_mutex_lock(&MyMutex); - else - pthread_mutex_unlock(&MyMutex); + SocketThread* parent; + public: + ThreadSignalSocket(SocketThread* p, int newfd) : parent(p) + { + SetFd(newfd); + SocketEngine::AddFd(this, FD_WANT_FAST_READ | FD_WANT_NO_WRITE); + } - return false; -} + ~ThreadSignalSocket() + { + SocketEngine::Close(this); + } -void* PThreadEngine::Entry(void* parameter) + void Notify() + { + eventfd_write(fd, 1); + } + + void HandleEvent(EventType et, int errornum) + { + if (et == EVENT_READ) + { + eventfd_t dummy; + eventfd_read(fd, &dummy); + parent->OnNotify(); + } + else + { + ServerInstance->GlobalCulls.AddItem(this); + } + } +}; + +SocketThread::SocketThread() { - ThreadEngine * pt = (ThreadEngine*)parameter; - pt->Run(); - return NULL; + signal.sock = NULL; + int fd = eventfd(0, EFD_NONBLOCK); + if (fd < 0) + throw new CoreException("Could not create pipe " + std::string(strerror(errno))); + signal.sock = new ThreadSignalSocket(this, fd); } +#else -void PThreadEngine::FreeThread(Thread* thread) +class ThreadSignalSocket : public EventHandler { - pthread_t* pthread = NULL; - if (thread->GetExt("pthread", pthread)) + SocketThread* parent; + int send_fd; + public: + ThreadSignalSocket(SocketThread* p, int recvfd, int sendfd) : + parent(p), send_fd(sendfd) + { + SetFd(recvfd); + SocketEngine::NonBlocking(fd); + SocketEngine::AddFd(this, FD_WANT_FAST_READ | FD_WANT_NO_WRITE); + } + + ~ThreadSignalSocket() + { + close(send_fd); + SocketEngine::Close(this); + } + + void Notify() + { + static const char dummy = '*'; + write(send_fd, &dummy, 1); + } + + void HandleEvent(EventType et, int errornum) { - thread->SetExitFlag(); - int rc; - void* status; - rc = pthread_join(*pthread, &status); - delete pthread; + if (et == EVENT_READ) + { + char dummy[128]; + read(fd, dummy, 128); + parent->OnNotify(); + } + else + { + ServerInstance->GlobalCulls.AddItem(this); + } } +}; + +SocketThread::SocketThread() +{ + signal.sock = NULL; + int fds[2]; + if (pipe(fds)) + throw new CoreException("Could not create pipe " + std::string(strerror(errno))); + signal.sock = new ThreadSignalSocket(this, fds[0], fds[1]); +} +#endif + +void SocketThread::NotifyParent() +{ + signal.sock->Notify(); } +SocketThread::~SocketThread() +{ + if (signal.sock) + { + signal.sock->cull(); + delete signal.sock; + } +}