src/threadengines/threadengine_win32.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22 #include "threadengines/threadengine_win32.h"
  23
  24 void ThreadEngine::Start(Thread* thread)
  25 {
  26         thread->state.handle = CreateThread(NULL, 0, ThreadEngine::Entry, thread, 0, NULL);
  27
  28         if (thread->state.handle == NULL)
  29         {
  30                 DWORD lasterr = GetLastError();
  31                 std::string err = "Unable to create new thread: " + ConvToStr(lasterr);
  32                 SetLastError(ERROR_SUCCESS);
  33                 throw CoreException(err);
  34         }
  35 }
  36
  37 DWORD WINAPI ThreadEngine::Entry(void* parameter)
  38 {
  39         Thread* pt = static_cast<Thread*>(parameter);
  40         pt->Run();
  41         return 0;
  42 }
  43
  44 void ThreadEngine::Stop(Thread* thread)
  45 {
  46         thread->SetExitFlag();
  47         HANDLE handle = thread->state.handle;
  48         WaitForSingleObject(handle,INFINITE);
  49         CloseHandle(handle);
  50 }
  51
  52 class ThreadSignalSocket : public BufferedSocket
  53 {
  54         SocketThread* parent;
  55  public:
  56         ThreadSignalSocket(SocketThread* t, int newfd)
  57                 : BufferedSocket(newfd), parent(t)
  58         {
  59         }
  60
  61         void OnDataReady()
  62         {
  63                 recvq.clear();
  64                 parent->OnNotify();
  65         }
  66
  67         void OnError(BufferedSocketError)
  68         {
  69                 ServerInstance->GlobalCulls.AddItem(this);
  70         }
  71 };
  72
  73 static bool BindAndListen(int sockfd, int port, const char* addr)
  74 {
  75         irc::sockets::sockaddrs servaddr;
  76         if (!irc::sockets::aptosa(addr, port, servaddr))
  77                 return false;
  78
  79         if (SocketEngine::Bind(sockfd, servaddr) != 0)
  80                 return false;
  81
  82         if (SocketEngine::Listen(sockfd, ServerInstance->Config->MaxConn) != 0)
  83         {
  84                 ServerInstance->Logs->Log("SOCKET", LOG_DEFAULT, "ERROR in listen(): %s", strerror(errno));
  85                 return false;
  86         }
  87
  88         return true;
  89 }
  90
  91 SocketThread::SocketThread()
  92 {
  93         int listenFD = socket(AF_INET, SOCK_STREAM, 0);
  94         if (listenFD == -1)
  95                 throw CoreException("Could not create ITC pipe");
  96         int connFD = socket(AF_INET, SOCK_STREAM, 0);
  97         if (connFD == -1)
  98                 throw CoreException("Could not create ITC pipe");
  99
 100         if (!BindAndListen(listenFD, 0, "127.0.0.1"))
 101                 throw CoreException("Could not create ITC pipe");
 102         SocketEngine::NonBlocking(connFD);
 103
 104         struct sockaddr_in addr;
 105         socklen_t sz = sizeof(addr);
 106         getsockname(listenFD, reinterpret_cast<struct sockaddr*>(&addr), &sz);
 107         connect(connFD, reinterpret_cast<struct sockaddr*>(&addr), sz);
 108         SocketEngine::Blocking(listenFD);
 109         int nfd = accept(listenFD, reinterpret_cast<struct sockaddr*>(&addr), &sz);
 110         if (nfd < 0)
 111                 throw CoreException("Could not create ITC pipe");
 112         new ThreadSignalSocket(this, nfd);
 113         closesocket(listenFD);
 114
 115         SocketEngine::Blocking(connFD);
 116         this->signal.connFD = connFD;
 117 }
 118
 119 void SocketThread::NotifyParent()
 120 {
 121         char dummy = '*';
 122         send(signal.connFD, &dummy, 1, 0);
 123 }
 124
 125 SocketThread::~SocketThread()
 126 {
 127         if (signal.connFD >= 0)
 128         {
 129                 shutdown(signal.connFD, 2);
 130                 closesocket(signal.connFD);
 131         }
 132 }