]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/inspircd.cpp
Add duplicate UID error in ParseUID(), seems pippijn was triggering it and it was...
[user/henk/code/inspircd.git] / src / inspircd.cpp
index 449bb005b2a2a68659f6c46f28c750af408af278..fe36208d13670e58f43584aefe350ca8ac1c5b60 100644 (file)
@@ -6,13 +6,12 @@
  * See: http://www.inspircd.org/wiki/index.php/Credits
  *
  * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ *         the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
 #include "inspircd.h"
-#include "configreader.h"
 #include <signal.h>
 
 #ifndef WIN32
        #endif
 #endif
 
-#include <exception>
 #include <fstream>
-#include "modules.h"
-#include "mode.h"
 #include "xline.h"
 #include "socketengine.h"
 #include "inspircd_se_config.h"
 #include "socket.h"
-#include "typedefs.h"
 #include "command_parse.h"
 #include "exitcodes.h"
 #include "caller.h"
 
-#ifdef WIN32
-
-/* This MUST remain static and delcared outside the class, so that WriteProcessMemory can reference it properly */
-static DWORD owner_processid = 0;
-
-DWORD WindowsForkStart(InspIRCd * Instance)
-{
-       /* Windows implementation of fork() :P */
-
-       char module[MAX_PATH];
-       if(!GetModuleFileName(NULL, module, MAX_PATH))
-       {
-               printf("GetModuleFileName() failed.\n");
-               return false;
-       }
-
-       STARTUPINFO startupinfo;
-       PROCESS_INFORMATION procinfo;
-       ZeroMemory(&startupinfo, sizeof(STARTUPINFO));
-       ZeroMemory(&procinfo, sizeof(PROCESS_INFORMATION));
-
-       // Fill in the startup info struct
-       GetStartupInfo(&startupinfo);
-
-       /* Default creation flags create the processes suspended */
-       DWORD startupflags = CREATE_SUSPENDED;
-
-       /* On windows 2003/XP and above, we can use the value
-        * CREATE_PRESERVE_CODE_AUTHZ_LEVEL which gives more access
-        * to the process which we may require on these operating systems.
-        */
-       OSVERSIONINFO vi;
-       vi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
-       GetVersionEx(&vi);
-       if ((vi.dwMajorVersion >= 5) && (vi.dwMinorVersion > 0))
-               startupflags |= CREATE_PRESERVE_CODE_AUTHZ_LEVEL;
-
-       // Launch our "forked" process.
-       BOOL bSuccess = CreateProcess ( module, // Module (exe) filename
-               strdup(GetCommandLine()),       // Command line (exe plus parameters from the OS)
-                                               // NOTE: We cannot return the direct value of the
-                                               // GetCommandLine function here, as the pointer is
-                                               // passed straight to the child process, and will be
-                                               // invalid once we exit as it goes out of context.
-                                               // strdup() seems ok, though.
-               0,                              // PROCESS_SECURITY_ATTRIBUTES
-               0,                              // THREAD_SECURITY_ATTRIBUTES
-               TRUE,                           // We went to inherit handles.
-               startupflags,                   // Allow us full access to the process and suspend it.
-               0,                              // ENVIRONMENT
-               0,                              // CURRENT_DIRECTORY
-               &startupinfo,                   // startup info
-               &procinfo);                     // process info
-
-       if(!bSuccess)
-       {
-               printf("CreateProcess() error: %s\n", dlerror());
-               return false;
-       }
-
-       // Set the owner process id in the target process.
-       SIZE_T written = 0;
-       DWORD pid = GetCurrentProcessId();
-       if(!WriteProcessMemory(procinfo.hProcess, &owner_processid, &pid, sizeof(DWORD), &written) || written != sizeof(DWORD))
-       {
-               printf("WriteProcessMemory() failed: %s\n", dlerror());
-               return false;
-       }
-
-       // Resume the other thread (let it start)
-       ResumeThread(procinfo.hThread);
-
-       // Wait for the new process to kill us. If there is some error, the new process will end and we will end up at the next line.
-       WaitForSingleObject(procinfo.hProcess, INFINITE);
-
-       // If we hit this it means startup failed, default to 14 if this fails.
-       DWORD ExitCode = 14;
-       GetExitCodeProcess(procinfo.hProcess, &ExitCode);
-       CloseHandle(procinfo.hThread);
-       CloseHandle(procinfo.hProcess);
-       return ExitCode;
-}
-
-void WindowsForkKillOwner(InspIRCd * Instance)
-{
-       HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, owner_processid);
-       if(!hProcess || !owner_processid)
-       {
-               printf("Could not open process id %u: %s.\n", owner_processid, dlerror());
-               Instance->Exit(14);
-       }
-
-       // die die die
-       if(!TerminateProcess(hProcess, 0))
-       {
-               printf("Could not TerminateProcess(): %s\n", dlerror());
-               Instance->Exit(14);
-       }
-
-       CloseHandle(hProcess);
-}
-
-#endif
-
-using irc::sockets::NonBlocking;
-using irc::sockets::Blocking;
 using irc::sockets::insp_ntoa;
 using irc::sockets::insp_inaddr;
 using irc::sockets::insp_sockaddr;
 
 InspIRCd* SI = NULL;
+int* mysig = NULL;
+
 
 /* Burlex: Moved from exitcodes.h -- due to duplicate symbols */
 const char* ExitCodes[] =
@@ -174,31 +65,21 @@ const char* ExitCodes[] =
                "Received SIGTERM", /* 15 */
 };
 
-void InspIRCd::Exit(int status)
-{
-#ifdef WINDOWS
-       CloseIPC();
-#endif
-       if (SI)
-       {
-               SI->SendError("Exiting with status " + ConvToStr(status) + " (" + std::string(ExitCodes[status]) + ")");
-               SI->Cleanup();
-       }
-       exit (status);
-}
-
 void InspIRCd::Cleanup()
 {
        std::vector<std::string> mymodnames;
-       int MyModCount = this->GetModuleCount();
+       int MyModCount = this->Modules->GetCount();
 
-       for (unsigned int i = 0; i < Config->ports.size(); i++)
+       if (Config)
        {
-               /* This calls the constructor and closes the listening socket */
-               delete Config->ports[i];
-       }
+               for (unsigned int i = 0; i < Config->ports.size(); i++)
+               {
+                       /* This calls the constructor and closes the listening socket */
+                       delete Config->ports[i];
+               }
 
-       Config->ports.clear();
+               Config->ports.clear();
+       }
 
        /* Close all client sockets, or the new process inherits them */
        for (std::vector<userrec*>::const_iterator i = this->local_users.begin(); i != this->local_users.end(); i++)
@@ -213,28 +94,27 @@ void InspIRCd::Cleanup()
         */
        for (int tries = 0; tries < 3; tries++)
        {
-               MyModCount = this->GetModuleCount();
+               MyModCount = this->Modules->GetCount();
                mymodnames.clear();
 
-               /* Unload all modules, so they get a chance to clean up their listeners */
-               for (int j = 0; j <= MyModCount; j++)
-                       mymodnames.push_back(Config->module_names[j]);
+               if (MyModCount)
+               {
+                       /* Unload all modules, so they get a chance to clean up their listeners */
+                       for (int j = 0; j <= MyModCount; j++)
+                               mymodnames.push_back(Config->module_names[j]);
 
-               for (int k = 0; k <= MyModCount; k++)
-                       this->UnloadModule(mymodnames[k].c_str());
+                       for (int k = 0; k <= MyModCount; k++)
+                               this->Modules->Unload(mymodnames[k].c_str());
+               }
        }
 
        /* Close logging */
-       this->Logger->Close();
+       if (this->Logger)
+               this->Logger->Close();
 
        /* Cleanup Server Names */
        for(servernamelist::iterator itr = servernames.begin(); itr != servernames.end(); ++itr)
                delete (*itr);
-
-#ifdef WINDOWS
-       /* WSACleanup */
-       WSACleanup();
-#endif
 }
 
 void InspIRCd::Restart(const std::string &reason)
@@ -280,9 +160,11 @@ void InspIRCd::ResetMaxBans()
 void InspIRCd::RehashUsersAndChans()
 {
        user_hash* old_users = this->clientlist;
+       user_hash* old_uuid  = this->uuidlist;
        chan_hash* old_chans = this->chanlist;
 
        this->clientlist = new user_hash();
+       this->uuidlist = new user_hash();
        this->chanlist = new chan_hash();
 
        for (user_hash::const_iterator n = old_users->begin(); n != old_users->end(); n++)
@@ -290,6 +172,11 @@ void InspIRCd::RehashUsersAndChans()
 
        delete old_users;
 
+       for (user_hash::const_iterator n = old_uuid->begin(); n != old_uuid->end(); n++)
+               this->uuidlist->insert(*n);
+
+       delete old_uuid;
+
        for (chan_hash::const_iterator n = old_chans->begin(); n != old_chans->end(); n++)
                this->chanlist->insert(*n);
 
@@ -298,18 +185,19 @@ void InspIRCd::RehashUsersAndChans()
 
 void InspIRCd::CloseLog()
 {
-       this->Logger->Close();
+       if (this->Logger)
+               this->Logger->Close();
 }
 
 void InspIRCd::SetSignals()
 {
 #ifndef WIN32
        signal(SIGALRM, SIG_IGN);
-       signal(SIGHUP, InspIRCd::Rehash);
+       signal(SIGHUP, InspIRCd::SetSignal);
        signal(SIGPIPE, SIG_IGN);
        signal(SIGCHLD, SIG_IGN);
 #endif
-       signal(SIGTERM, InspIRCd::Exit);
+       signal(SIGTERM, InspIRCd::SetSignal);
 }
 
 void InspIRCd::QuickExit(int status)
@@ -345,7 +233,7 @@ bool InspIRCd::DaemonSeed()
        umask (007);
        printf("InspIRCd Process ID: \033[1;32m%lu\033[0m\n",(unsigned long)getpid());
 
-       signal(SIGTERM, InspIRCd::Exit);
+       signal(SIGTERM, InspIRCd::SetSignal);
 
        rlimit rl;
        if (getrlimit(RLIMIT_CORE, &rl) == -1)
@@ -392,28 +280,42 @@ void InspIRCd::WritePID(const std::string &filename)
 }
 
 InspIRCd::InspIRCd(int argc, char** argv)
-       : ModCount(0),
-         GlobalCulls(this),
+       : GlobalCulls(this),
+
+        /* Functor initialisation. Note that the ordering here is very important. */
+        HandleProcessUser(this),
         HandleIsNick(this),
         HandleIsIdent(this),
         HandleFindDescriptor(this),
+        HandleFloodQuitUser(this),
+
+        /* Functor pointer initialisation. Must match the order of the list above */
+        ProcessUser(&HandleProcessUser),
         IsNick(&HandleIsNick),
         IsIdent(&HandleIsIdent),
-        FindDescriptor(&HandleFindDescriptor)
+        FindDescriptor(&HandleFindDescriptor),
+        FloodQuitUser(&HandleFloodQuitUser)
+
 {
+
        int found_ports = 0;
        FailedPortList pl;
        int do_version = 0, do_nofork = 0, do_debug = 0, do_nolog = 0, do_root = 0;    /* flag variables */
        char c = 0;
 
-       modules.resize(255);
-       factory.resize(255);
        memset(&server, 0, sizeof(server));
        memset(&client, 0, sizeof(client));
 
+       SocketEngineFactory* SEF = new SocketEngineFactory();
+       SE = SEF->Create(this);
+       delete SEF;
+
+       this->s_signal = 0;
+
        this->unregistered_count = 0;
 
        this->clientlist = new user_hash();
+       this->uuidlist = new user_hash();
        this->chanlist = new chan_hash();
 
        this->Config = new ServerConfig(this);
@@ -421,7 +323,11 @@ InspIRCd::InspIRCd(int argc, char** argv)
        this->Config->argv = argv;
        this->Config->argc = argc;
 
-       chdir(Config->GetFullProgDir().c_str());
+       if (chdir(Config->GetFullProgDir().c_str()))
+       {
+               printf("Unable to change to my directory: %s\nAborted.", strerror(errno));
+               exit(0);
+       }
 
        this->Config->opertypes.clear();
        this->Config->operclass.clear();
@@ -478,7 +384,7 @@ InspIRCd::InspIRCd(int argc, char** argv)
 #ifdef WIN32
 
        // Handle forking
-       if(!do_nofork && !owner_processid)
+       if(!do_nofork)
        {
                DWORD ExitCode = WindowsForkStart(this);
                if(ExitCode)
@@ -489,6 +395,7 @@ InspIRCd::InspIRCd(int argc, char** argv)
        WSADATA wsadata;
        WSAStartup(MAKEWORD(2,0), &wsadata);
 
+       ChangeWindowsSpecificPointers(this);
 #endif
        if (!ServerConfig::FileExists(this->ConfigFileName))
        {
@@ -509,14 +416,58 @@ InspIRCd::InspIRCd(int argc, char** argv)
 
        strlcpy(Config->MyExecutable,argv[0],MAXBUF);
 
-       this->OpenLog(argv, argc);
+       if (!this->OpenLog(argv, argc))
+       {
+               printf("ERROR: Could not open logfile %s: %s\n\n", Config->logpath.c_str(), strerror(errno));
+               Exit(EXIT_STATUS_LOG);
+       }
 
+       this->Modules = new ModuleManager(this);
        this->stats = new serverstats();
        this->Timers = new TimerManager(this);
        this->Parser = new CommandParser(this);
        this->XLines = new XLineManager(this);
+       
        Config->ClearStack();
        Config->Read(true, NULL);
+       
+       this->Modules->modules.resize(255);
+       this->Modules->handles.resize(255);
+
+       /*
+        * Initialise UID. XXX, we need to read SID from config, and use it instead of 000.
+        * For an explanation as to exactly how this works, and why it works this way, see GetUID().
+        *   -- w00t
+        */
+       int i;
+
+       /* Generate SID */
+       size_t sid = 0;
+       if (Config->sid)
+       {
+               sid = Config->sid;
+       }
+       else
+       {
+               for (const char* x = Config->ServerName; *x; ++x)
+                       sid = 5 * sid + *x;
+               for (const char* y = Config->ServerDesc; *y; ++y)
+                       sid = 5 * sid + *y;
+               sid = sid % 999;
+
+               Config->sid = sid;
+       }
+       current_uid[0] = sid / 100 + 48;
+       current_uid[1] = ((sid / 10) % 10) + 48;
+       current_uid[2] = sid % 10 + 48;
+
+       /* Initialise UID */
+       for(i = 3; i < UUID_LENGTH - 1; i++)
+               current_uid[i] = 'A';
+
+       /* set up fake client */
+       this->FakeClient = new userrec(this);
+       this->FakeClient->SetFd(FD_MAGIC_NUMBER);
 
        if (!do_root)
                this->CheckRoot();
@@ -545,13 +496,7 @@ InspIRCd::InspIRCd(int argc, char** argv)
                }
        }
 
-
-       /* Because of limitations in kqueue on freebsd, we must fork BEFORE we
-        * initialize the socket engine.
-        */
-       SocketEngineFactory* SEF = new SocketEngineFactory();
-       SE = SEF->Create(this);
-       delete SEF;
+       SE->RecoverFromFork();
 
        this->Modes = new ModeParser(this);
        this->AddServerName(Config->ServerName);
@@ -567,7 +512,8 @@ InspIRCd::InspIRCd(int argc, char** argv)
 
        this->Res = new DNS(this);
 
-       this->LoadAllModules();
+       this->Modules->LoadAll();
+       
        /* Just in case no modules were loaded - fix for bug #101 */
        this->BuildISupport();
        InitializeDisabledCommands(Config->DisabledCommands, this);
@@ -617,13 +563,14 @@ InspIRCd::InspIRCd(int argc, char** argv)
                }
        }
 #else
-       InitIPC();
+       WindowsIPC = new IPC(this);
        if(!Config->nofork)
        {
                WindowsForkKillOwner(this);
                FreeConsole();
        }
 #endif
+
        printf("\nInspIRCd is now running!\n");
        Log(DEFAULT,"Startup complete.");
 
@@ -677,7 +624,7 @@ void InspIRCd::DoOneIteration(bool process_module_sockets)
                        this->stats->LastCPU = ru.ru_utime;
                }
 #else
-               CheckIPC(this);
+               WindowsIPC->Check();
 
                if(Config->nofork)
                {
@@ -704,6 +651,13 @@ void InspIRCd::DoOneIteration(bool process_module_sockets)
 
        /* If any inspsockets closed, remove them */
        this->InspSocketCull();
+
+       if (this->s_signal)
+       {
+               this->SignalHandler(s_signal);
+               this->s_signal = 0;
+       }
+
 }
 
 void InspIRCd::InspSocketCull()
@@ -736,6 +690,7 @@ int InspIRCd::Run()
 int main(int argc, char** argv)
 {
        SI = new InspIRCd(argc, argv);
+       mysig = &SI->s_signal;
        SI->Run();
        delete SI;
        return 0;
@@ -750,11 +705,11 @@ bool InspIRCd::AllModulesReportReady(userrec* user)
        if (!Config->global_implementation[I_OnCheckReady])
                return true;
 
-       for (int i = 0; i <= this->GetModuleCount(); i++)
+       for (int i = 0; i <= this->Modules->GetCount(); i++)
        {
                if (Config->implement_lists[i][I_OnCheckReady])
                {
-                       int res = modules[i]->OnCheckReady(user);
+                       int res = this->Modules->modules[i]->OnCheckReady(user);
                        if (!res)
                                return false;
                }
@@ -762,11 +717,6 @@ bool InspIRCd::AllModulesReportReady(userrec* user)
        return true;
 }
 
-int InspIRCd::GetModuleCount()
-{
-       return this->ModCount;
-}
-
 time_t InspIRCd::Time(bool delta)
 {
        if (delta)
@@ -804,3 +754,8 @@ int InspIRCd::GetTimeDelta()
 {
        return time_delta;
 }
+
+void InspIRCd::SetSignal(int signal)
+{
+       *mysig = signal;
+}