src/threadengines/threadengine_win32.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *          the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "threadengines/threadengine_win32.h"
  16
  17 ThreadEngine::ThreadEngine(InspIRCd* Instance)
  18 {
  19 }
  20
  21 void ThreadEngine::Start(Thread* thread)
  22 {
  23         ThreadData* data = new ThreadData;
  24         thread->state = data;
  25
  26         DWORD ThreadId = 0;
  27         data->handle = CreateThread(NULL,0,ThreadEngine::Entry,thread,0,&ThreadId);
  28
  29         if (data->handle == NULL)
  30         {
  31                 thread->state = NULL;
  32                 delete data;
  33                 throw CoreException(std::string("Unable to create new thread: ") + dlerror());
  34         }
  35 }
  36
  37 ThreadEngine::~ThreadEngine()
  38 {
  39 }
  40
  41 DWORD WINAPI ThreadEngine::Entry(void* parameter)
  42 {
  43         Thread* pt = reinterpret_cast<Thread*>(parameter);
  44         pt->Run();
  45         return 0;
  46 }
  47
  48 void ThreadData::FreeThread(Thread* thread)
  49 {
  50         thread->SetExitFlag();
  51         WaitForSingleObject(handle,INFINITE);
  52 }
  53
  54 class ThreadSignalSocket : public BufferedSocket
  55 {
  56         SocketThread* parent;
  57  public:
  58         ThreadSignalSocket(SocketThread* t, InspIRCd* SI, int newfd, const char* ip)
  59                 : BufferedSocket(SI, newfd, ip), parent(t)
  60         {
  61         }
  62
  63         virtual bool OnDataReady()
  64         {
  65                 char data = 0;
  66                 if (ServerInstance->SE->Recv(this, &data, 1, 0) > 0)
  67                 {
  68                         parent->OnNotify();
  69                         return true;
  70                 }
  71                 return false;
  72         }
  73 };
  74
  75 class ThreadSignalListener : public ListenSocketBase
  76 {
  77         SocketThread* parent;
  78         irc::sockets::insp_sockaddr sock_us;
  79  public:
  80         ThreadSignalListener(SocketThread* t, InspIRCd* Instance, int port, const std::string &addr) : ListenSocketBase(Instance, port, addr), parent(t)
  81         {
  82                 socklen_t uslen = sizeof(sock_us);
  83                 if (getsockname(this->fd,(sockaddr*)&sock_us,&uslen))
  84                 {
  85                         throw ModuleException("Could not getsockname() to find out port number for ITC port");
  86                 }
  87         }
  88
  89         virtual void OnAcceptReady(int nfd)
  90         {
  91                 new ThreadSignalSocket(parent, ServerInstance, nfd, "");
  92                 ServerInstance->SE->DelFd(this);
  93         }
  94 /* Using getsockname and ntohs, we can determine which port number we were allocated */
  95         int GetPort()
  96         {
  97 #ifdef IPV6
  98                 return ntohs(sock_us.sin6_port);
  99 #else
 100                 return ntohs(sock_us.sin_port);
 101 #endif
 102         }
 103 };
 104
 105 SocketThread::SocketThread(InspIRCd* SI)
 106 {
 107         ThreadSignalListener* listener = new ThreadSignalListener(this, SI, 0, "127.0.0.1");
 108         if (listener->GetFd() == -1)
 109                 throw CoreException("Could not create ITC pipe");
 110         int connFD = socket(AF_INET, SOCK_STREAM, 0);
 111         if (connFD == -1)
 112                 throw CoreException("Could not create ITC pipe");
 113
 114         irc::sockets::insp_sockaddr addr;
 115
 116 #ifdef IPV6
 117         irc::sockets::insp_aton("::1", &addr.sin6_addr);
 118         addr.sin6_family = AF_INET6;
 119         addr.sin6_port = htons(listener->GetPort());
 120 #else
 121         irc::sockets::insp_aton("127.0.0.1", &addr.sin_addr);
 122         addr.sin_family = AF_INET;
 123         addr.sin_port = htons(listener->GetPort());
 124 #endif
 125
 126         if (connect(connFD, reinterpret_cast<struct sockaddr*>(&addr), sizeof(addr)) == -1)
 127         {
 128                 SI->SE->DelFd(listener);
 129                 closesocket(connFD);
 130                 throw CoreException("Could not connet to ITC pipe");
 131         }
 132         this->signal.connFD = connFD;
 133 }
 134
 135 void SocketThread::NotifyParent()
 136 {
 137         char dummy = '*';
 138         send(signal.connFD, &dummy, 1, 0);
 139 }
 140
 141 SocketThread::~SocketThread()
 142 {
 143         if (signal.connFD >= 0)
 144         {
 145                 shutdown(signal.connFD, 2);
 146                 closesocket(signal.connFD);
 147         }
 148 }