]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/inspircd.cpp
Add a progress meter for loading of core commands, fix broken implementation for...
[user/henk/code/inspircd.git] / src / inspircd.cpp
index 47b6d278b44d8b29f66a74e1df1524bea33382e4..069c8634d0c86ee8de854e916aac5326ccedac91 100644 (file)
@@ -41,8 +41,6 @@ InspIRCd* SI = NULL;
 
 void InspIRCd::AddServerName(const std::string &servername)
 {
-       this->Log(DEBUG,"Adding server name: %s",servername.c_str());
-       
        if(find(servernames.begin(), servernames.end(), servername) == servernames.end())
                servernames.push_back(servername); /* Wasn't already there. */
 }
@@ -50,9 +48,9 @@ void InspIRCd::AddServerName(const std::string &servername)
 const char* InspIRCd::FindServerNamePtr(const std::string &servername)
 {
        servernamelist::iterator iter = find(servernames.begin(), servernames.end(), servername);
-       
+
        if(iter == servernames.end())
-       {               
+       {
                AddServerName(servername);
                iter = --servernames.end();
        }
@@ -80,13 +78,20 @@ void InspIRCd::Cleanup()
        std::vector<std::string> mymodnames;
        int MyModCount = this->GetModuleCount();
 
-       for (unsigned int i = 0; i < stats->BoundPortCount; i++)
+       for (unsigned int i = 0; i < Config->ports.size(); i++)
        {
                /* This calls the constructor and closes the listening socket */
-               delete Config->openSockfd[i];
-               Config->openSockfd[i] = NULL;
+               delete Config->ports[i];
+       }
+
+       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++)
+       {
+               (*i)->SetWriteError("Server shutdown");
+               (*i)->CloseSocket();
        }
-       stats->BoundPortCount = 0;
 
        /* We do this more than once, so that any service providers get a
         * chance to be unhooked by the modules using them, but then get
@@ -105,10 +110,6 @@ void InspIRCd::Cleanup()
                        this->UnloadModule(mymodnames[k].c_str());
        }
 
-       /* 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++)
-               (*i)->CloseSocket();
-
        /* Close logging */
        this->Logger->Close();
 }
@@ -142,16 +143,25 @@ void InspIRCd::Start()
 
 void InspIRCd::Rehash(int status)
 {
-       SI->WriteOpers("Rehashing config file %s due to SIGHUP",ServerConfig::CleanFilename(CONFIG_FILE));
+       SI->WriteOpers("*** Rehashing config file %s due to SIGHUP",ServerConfig::CleanFilename(SI->ConfigFileName));
        SI->CloseLog();
        SI->OpenLog(SI->Config->argv, SI->Config->argc);
        SI->RehashUsersAndChans();
        FOREACH_MOD_I(SI, I_OnGarbageCollect, OnGarbageCollect());
        SI->Config->Read(false,NULL);
+       SI->ResetMaxBans();
        SI->Res->Rehash();
+       SI->BuildISupport();
        FOREACH_MOD_I(SI,I_OnRehash,OnRehash(NULL,""));
 }
 
+void InspIRCd::ResetMaxBans()
+{
+       for (chan_hash::const_iterator i = chanlist->begin(); i != chanlist->end(); i++)
+               i->second->ResetMaxBans();
+}
+
+
 /** Because hash_map doesnt free its buckets when we delete items (this is a 'feature')
  * we must occasionally rehash the hash (yes really).
  * We do this by copying the entries from the old hash to a new hash, causing all
@@ -192,8 +202,15 @@ void InspIRCd::SetSignals()
        signal(SIGCHLD, SIG_IGN);
 }
 
+void InspIRCd::QuickExit(int status)
+{
+       exit(0);
+}
+
 bool InspIRCd::DaemonSeed()
 {
+       signal(SIGTERM, InspIRCd::QuickExit);
+
        int childpid;
        if ((childpid = fork ()) < 0)
                return false;
@@ -214,6 +231,8 @@ bool InspIRCd::DaemonSeed()
        umask (007);
        printf("InspIRCd Process ID: \033[1;32m%lu\033[0m\n",(unsigned long)getpid());
 
+       signal(SIGTERM, InspIRCd::Exit);
+
        rlimit rl;
        if (getrlimit(RLIMIT_CORE, &rl) == -1)
        {
@@ -236,13 +255,13 @@ void InspIRCd::WritePID(const std::string &filename)
        if (*(fname.begin()) != '/')
        {
                std::string::size_type pos;
-               std::string confpath = CONFIG_FILE;
-               if ((pos = confpath.find("/inspircd.conf")) != std::string::npos)
+               std::string confpath = this->ConfigFileName;
+               if ((pos = confpath.rfind("/")) != std::string::npos)
                {
                        /* Leaves us with just the path */
                        fname = confpath.substr(0, pos) + std::string("/") + fname;
                }
-       }                                                                       
+       }
        std::ofstream outfile(fname.c_str());
        if (outfile.is_open())
        {
@@ -263,11 +282,11 @@ std::string InspIRCd::GetRevision()
 }
 
 InspIRCd::InspIRCd(int argc, char** argv)
-       : ModCount(-1), duration_m(60), duration_h(60*60), duration_d(60*60*24), duration_w(60*60*24*7), duration_y(60*60*24*365)
+       : ModCount(-1), duration_m(60), duration_h(60*60), duration_d(60*60*24), duration_w(60*60*24*7), duration_y(60*60*24*365), GlobalCulls(this)
 {
        int found_ports = 0;
        FailedPortList pl;
-       int do_nofork = 0, do_debug = 0, do_nolog = 0;    /* flag variables */
+       int do_version = 0, do_nofork = 0, do_debug = 0, do_nolog = 0, do_root = 0;    /* flag variables */
        char c = 0;
 
        modules.resize(255);
@@ -286,28 +305,23 @@ InspIRCd::InspIRCd(int argc, char** argv)
        this->Config->opertypes.clear();
        this->Config->operclass.clear();
        this->SNO = new SnomaskManager(this);
-       this->Start();
        this->TIME = this->OLDTIME = this->startup_time = time(NULL);
        this->time_delta = 0;
        this->next_call = this->TIME + 3;
        srand(this->TIME);
 
-       if (!ServerConfig::FileExists(CONFIG_FILE))
-       {
-               printf("ERROR: Cannot open config file: %s\nExiting...\n",CONFIG_FILE);
-               this->Log(DEFAULT,"main: no config");
-               printf("ERROR: Your config file is missing, this IRCd will self destruct in 10 seconds!\n");
-               Exit(EXIT_STATUS_CONFIG);
-       }
-
        *this->LogFileName = 0;
+       strlcpy(this->ConfigFileName, CONFIG_FILE, MAXBUF);
 
        struct option longopts[] =
        {
                { "nofork",     no_argument,            &do_nofork,     1       },
                { "logfile",    required_argument,      NULL,           'f'     },
+               { "config",     required_argument,      NULL,           'c'     },
                { "debug",      no_argument,            &do_debug,      1       },
                { "nolog",      no_argument,            &do_nolog,      1       },
+               { "runasroot",  no_argument,            &do_root,       1       },
+               { "version",    no_argument,            &do_version,    1       },
                { 0, 0, 0, 0 }
        };
 
@@ -320,17 +334,37 @@ InspIRCd::InspIRCd(int argc, char** argv)
                                strlcpy(LogFileName, optarg, MAXBUF);
                                printf("LOG: Setting logfile to %s\n", LogFileName);
                        break;
+                       case 'c':
+                               /* Config filename was set */
+                               strlcpy(ConfigFileName, optarg, MAXBUF);
+                               printf("CONFIG: Setting config file to %s\n", ConfigFileName);
+                       break;
                        case 0:
                                /* getopt_long_only() set an int variable, just keep going */
                        break;
                        default:
                                /* Unknown parameter! DANGER, INTRUDER.... err.... yeah. */
-                               printf("Usage: %s [--nofork] [--nolog] [--debug] [--logfile <filename>]\n", argv[0]);
+                               printf("Usage: %s [--nofork] [--nolog] [--debug] [--logfile <filename>] [--runasroot] [--version] [--config <config>]\n", argv[0]);
                                Exit(EXIT_STATUS_ARGV);
                        break;
                }
        }
 
+       if (do_version)
+       {
+               printf("\n%s r%s\n", VERSION, REVISION);
+               Exit(EXIT_STATUS_NOERROR);
+       }
+
+       if (!ServerConfig::FileExists(this->ConfigFileName))
+       {
+               printf("ERROR: Cannot open config file: %s\nExiting...\n", this->ConfigFileName);
+               this->Log(DEFAULT,"Unable to open config file %s", this->ConfigFileName);
+               Exit(EXIT_STATUS_CONFIG);
+       }
+
+       this->Start();
+
        /* Set the finished argument values */
        Config->nofork = do_nofork;
        Config->forcedebug = do_debug;
@@ -339,58 +373,79 @@ InspIRCd::InspIRCd(int argc, char** argv)
        strlcpy(Config->MyExecutable,argv[0],MAXBUF);
 
        this->OpenLog(argv, argc);
+
        this->stats = new serverstats();
-       this->Timers = new TimerManager();
+       this->Timers = new TimerManager(this);
        this->Parser = new CommandParser(this);
        this->XLines = new XLineManager(this);
        Config->ClearStack();
        Config->Read(true, NULL);
-       this->CheckRoot();
-       this->Modes = new ModeParser(this);
-       this->AddServerName(Config->ServerName);        
-       CheckDie();
-       InitializeDisabledCommands(Config->DisabledCommands, this);
-       stats->BoundPortCount = BindPorts(true, found_ports, pl);
-
-       for(int t = 0; t < 255; t++)
-               Config->global_implementation[t] = 0;
 
-       memset(&Config->implement_lists,0,sizeof(Config->implement_lists));
+       if (!do_root)
+               this->CheckRoot();
+       else
+       {
+               printf("* WARNING * WARNING * WARNING * WARNING * WARNING * \n\n");
+               printf("YOU ARE RUNNING INSPIRCD AS ROOT. THIS IS UNSUPPORTED\n");
+               printf("AND IF YOU ARE HACKED, CRACKED, SPINDLED OR MUTILATED\n");
+               printf("OR ANYTHING ELSE UNEXPECTED HAPPENS TO YOU OR YOUR\n");
+               printf("SERVER, THEN IT IS YOUR OWN FAULT. IF YOU DID NOT MEAN\n");
+               printf("TO START INSPIRCD AS ROOT, HIT CTRL+C NOW AND RESTART\n");
+               printf("THE PROGRAM AS A NORMAL USER. YOU HAVE BEEN WARNED!\n");
+               printf("\nInspIRCd starting in 20 seconds, ctrl+c to abort...\n");
+               sleep(20);
+       }
 
-       printf("\n");
        this->SetSignals();
+
        if (!Config->nofork)
        {
                if (!this->DaemonSeed())
                {
                        printf("ERROR: could not go into daemon mode. Shutting down.\n");
+                       Log(DEFAULT,"ERROR: could not go into daemon mode. Shutting down.");
                        Exit(EXIT_STATUS_FORK);
                }
        }
 
-       /* Because of limitations in kqueue on freebsd, we must fork BEFORE we
+
+        /* 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;
 
+       this->Modes = new ModeParser(this);
+       this->AddServerName(Config->ServerName);
+       CheckDie();
+       int bounditems = BindPorts(true, found_ports, pl);
+
+       for(int t = 0; t < 255; t++)
+               Config->global_implementation[t] = 0;
+
+       memset(&Config->implement_lists,0,sizeof(Config->implement_lists));
+
+       printf("\n");
+
        this->Res = new DNS(this);
 
        this->LoadAllModules();
        /* Just in case no modules were loaded - fix for bug #101 */
        this->BuildISupport();
+       InitializeDisabledCommands(Config->DisabledCommands, this);
 
-       if ((stats->BoundPortCount == 0) && (found_ports > 0))
+       if ((Config->ports.size() == 0) && (found_ports > 0))
        {
                printf("\nERROR: I couldn't bind any ports! Are you sure you didn't start InspIRCd twice?\n");
+               Log(DEFAULT,"ERROR: I couldn't bind any ports! Are you sure you didn't start InspIRCd twice?");
                Exit(EXIT_STATUS_BIND);
        }
-       
-       if (stats->BoundPortCount != (unsigned int)found_ports)
+
+       if (Config->ports.size() != (unsigned int)found_ports)
        {
-               printf("\nWARNING: Not all your client ports could be bound --\nstarting anyway with %ld of %d client ports bound.\n\n", stats->BoundPortCount, found_ports);
-               printf("The following port%s failed to bind:\n", found_ports - stats->BoundPortCount != 1 ? "s" : "");
+               printf("\nWARNING: Not all your client ports could be bound --\nstarting anyway with %d of %d client ports bound.\n\n", bounditems, found_ports);
+               printf("The following port(s) failed to bind:\n");
                int j = 1;
                for (FailedPortList::iterator i = pl.begin(); i != pl.end(); i++, j++)
                {
@@ -398,30 +453,35 @@ InspIRCd::InspIRCd(int argc, char** argv)
                }
        }
 
-       /* Add the listening sockets used for client inbound connections
-        * to the socket engine
-        */
-       this->Log(DEBUG,"%d listeners",stats->BoundPortCount);
-       for (unsigned long count = 0; count < stats->BoundPortCount; count++)
+       if (!Config->nofork)
        {
-               this->Log(DEBUG,"Add listener: %d",Config->openSockfd[count]->GetFd());
-               if (!SE->AddFd(Config->openSockfd[count]))
+               if (kill(getppid(), SIGTERM) == -1)
                {
-                       printf("\nEH? Could not add listener to socketengine. You screwed up, aborting.\n");
-                       Exit(EXIT_STATUS_INTERNAL);
+                       printf("Error killing parent process: %s\n",strerror(errno));
+                       Log(DEFAULT,"Error killing parent process: %s",strerror(errno));
                }
        }
 
-       if (!Config->nofork)
+       if (isatty(0) && isatty(1) && isatty(2))
        {
-               if (kill(getppid(), SIGTERM) == -1)
-                       printf("Error killing parent process: %s\n",strerror(errno));
-               fclose(stdin);
-               fclose(stderr);
-               fclose(stdout);
+               /* We didn't start from a TTY, we must have started from a background process -
+                * e.g. we are restarting, or being launched by cron. Dont kill parent, and dont
+                * close stdin/stdout
+                */
+               if (!do_nofork)
+               {
+                       fclose(stdin);
+                       fclose(stderr);
+                       fclose(stdout);
+               }
+               else
+               {
+                       Log(DEFAULT,"Keeping pseudo-tty open as we are running in the foreground.");
+               }
        }
 
        printf("\nInspIRCd is now running!\n");
+       Log(DEFAULT,"Startup complete.");
 
        this->WritePID(Config->PID);
 }
@@ -430,13 +490,14 @@ std::string InspIRCd::GetVersionString()
 {
        char versiondata[MAXBUF];
        char dnsengine[] = "singlethread-object";
+
        if (*Config->CustomVersion)
        {
                snprintf(versiondata,MAXBUF,"%s %s :%s",VERSION,Config->ServerName,Config->CustomVersion);
        }
        else
        {
-               snprintf(versiondata,MAXBUF,"%s %s :%s [FLAGS=%lu,%s,%s]",VERSION,Config->ServerName,SYSTEM,(unsigned long)OPTIMISATION,SE->GetName().c_str(),dnsengine);
+               snprintf(versiondata,MAXBUF,"%s %s :%s [FLAGS=%s,%s,%s]",VERSION,Config->ServerName,SYSTEM,REVISION,SE->GetName().c_str(),dnsengine);
        }
        return versiondata;
 }
@@ -523,10 +584,6 @@ void InspIRCd::MoveTo(std::string modulename,int slot)
                        Config->implement_lists[slot][n] = x;
                }
        }
-       else
-       {
-               this->Log(DEBUG,"Move of %s to slot failed!",modulename.c_str());
-       }
 }
 
 void InspIRCd::MoveAfter(std::string modulename, std::string after)
@@ -574,9 +631,9 @@ void InspIRCd::BuildISupport()
 {
        // the neatest way to construct the initial 005 numeric, considering the number of configure constants to go in it...
        std::stringstream v;
-       v << "WALLCHOPS WALLVOICES MODES=" << MAXMODES << " CHANTYPES=# PREFIX=" << this->Modes->BuildPrefixes() << " MAP MAXCHANNELS=" << MAXCHANS << " MAXBANS=60 VBANLIST NICKLEN=" << NICKMAX-1;
+       v << "WALLCHOPS WALLVOICES MODES=" << MAXMODES-1 << " CHANTYPES=# PREFIX=" << this->Modes->BuildPrefixes() << " MAP MAXCHANNELS=" << Config->MaxChans << " MAXBANS=60 VBANLIST NICKLEN=" << NICKMAX-1;
        v << " CASEMAPPING=rfc1459 STATUSMSG=@%+ CHARSET=ascii TOPICLEN=" << MAXTOPIC << " KICKLEN=" << MAXKICK << " MAXTARGETS=" << Config->MaxTargets << " AWAYLEN=";
-       v << MAXAWAY << " CHANMODES=" << this->Modes->ChanModes() << " FNC NETWORK=" << Config->Network << " MAXPARA=32";
+       v << MAXAWAY << " CHANMODES=" << this->Modes->ChanModes() << " FNC NETWORK=" << Config->Network << " MAXPARA=32 ELIST=MU";
        Config->data005 = v.str();
        FOREACH_MOD_I(this,I_On005Numeric,On005Numeric(Config->data005));
        Config->Update005();
@@ -692,8 +749,6 @@ bool InspIRCd::LoadModule(const char* filename)
                snprintf(MODERR,MAXBUF,"Module %s is not within the modules directory.",modfile);
                return false;
        }
-       this->Log(DEBUG,"Loading module: %s",modfile);
-
        if (ServerConfig::FileExists(modfile))
        {
 
@@ -758,7 +813,7 @@ bool InspIRCd::LoadModule(const char* filename)
                }
                catch (CoreException& modexcept)
                {
-                       this->Log(DEFAULT,"Unable to load %s: ",modfile,modexcept.GetReason());
+                       this->Log(DEFAULT,"Unable to load %s: %s",modfile,modexcept.GetReason());
                        snprintf(MODERR,MAXBUF,"Factory function of %s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
                        return false;
                }
@@ -821,7 +876,6 @@ void InspIRCd::DoOneIteration(bool process_module_sockets)
                        WriteOpers("*** \002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %d secs.",abs(OLDTIME-TIME));
                if ((TIME % 3600) == 0)
                {
-                       //MaintainWhoWas(this, TIME);
                        this->RehashUsersAndChans();
                        FOREACH_MOD_I(this, I_OnGarbageCollect, OnGarbageCollect());
                }
@@ -846,33 +900,24 @@ void InspIRCd::DoOneIteration(bool process_module_sockets)
         * file descriptors. The socket engine has everything's
         * descriptors in its list... dns, modules, users,
         * servers... so its nice and easy, just one call.
-        * This will cause any read or write events to be 
+        * This will cause any read or write events to be
         * dispatched to their handlers.
         */
        SE->DispatchEvents();
-}
 
-bool InspIRCd::IsIdent(const char* n)
-{
-       if (!n || !*n)
-               return false;
+       /* if any users was quit, take them out */
+       GlobalCulls.Apply();
 
-       for (char* i = (char*)n; *i; i++)
+       /* If any inspsockets closed, remove them */
+       for (std::map<InspSocket*,InspSocket*>::iterator x = SocketCull.begin(); x != SocketCull.end(); ++x)
        {
-               if ((*i >= 'A') && (*i <= '}'))
-               {
-                       continue;
-               }
-               if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.'))
-               {
-                       continue;
-               }
-               return false;
+               SE->DelFd(x->second);
+               x->second->Close();
+               delete x->second;
        }
-       return true;
+       SocketCull.clear();
 }
 
-
 int InspIRCd::Run()
 {
        while (true)
@@ -933,7 +978,7 @@ time_t InspIRCd::Time(bool delta)
 int InspIRCd::SetTimeDelta(int delta)
 {
        int old = time_delta;
-       time_delta += delta;
+       time_delta = delta;
        this->Log(DEBUG, "Time delta set to %d (was %d)", time_delta, old);
        return old;
 }
@@ -996,10 +1041,11 @@ void FileLogger::WriteLogLine(const std::string &line)
                        /* Wrote the whole buffer, and no need for write callback */
                        buffer = "";
                }
-       }
-       if (writeops++ % 20)
-       {
-               fflush(log);
+
+               if (writeops++ % 20)
+               {
+                       fflush(log);
+               }
        }
 }