[user/henk/code/inspircd.git] / include / inspircd.h

/*
 * InspIRCd -- Internet Relay Chat Daemon
 *
 *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
 *   Copyright (C) 2007-2008 Robin Burchell <robin+git@viroteck.net>
 *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
 *   Copyright (C) 2003-2008 Craig Edwards <craigedwards@brainbox.cc>
 *   Copyright (C) 2006-2007 Oliver Lupton <oliverlupton@gmail.com>
 *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
 *   Copyright (C) 2003 randomdan <???@???>
 *
 * 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 <http://www.gnu.org/licenses/>.
 */


#pragma once

#include <climits>
#include <cmath>
#include <csignal>
#include <cstdarg>
#include <cstdio>
#include <cstring>
#include <ctime>

#include <algorithm>
#include <bitset>
#include <deque>
#include <list>
#include <map>
#include <set>
#include <sstream>
#include <string>
#include <vector>

#include "compat.h"
#include "typedefs.h"

CoreExport extern InspIRCd* ServerInstance;

#include "config.h"
#include "consolecolors.h"
#include "caller.h"
#include "cull_list.h"
#include "extensible.h"
#include "numerics.h"
#include "uid.h"
#include "users.h"
#include "channels.h"
#include "timer.h"
#include "hashcomp.h"
#include "logger.h"
#include "usermanager.h"
#include "socket.h"
#include "ctables.h"
#include "command_parse.h"
#include "mode.h"
#include "socketengine.h"
#include "snomasks.h"
#include "filelogger.h"
#include "modules.h"
#include "threadengine.h"
#include "configreader.h"
#include "inspstring.h"
#include "protocol.h"

/** Returned by some functions to indicate failure.
 */
#define ERROR -1

/** Template function to convert any input type to std::string
 */
template<typename T> inline std::string ConvNumeric(const T &in)
{
        if (in == 0)
                return "0";
        T quotient = in;
        std::string out;
        while (quotient)
        {
                out += "0123456789"[ std::abs( (long)quotient % 10 ) ];
                quotient /= 10;
        }
        if (in < 0)
                out += '-';
        std::reverse(out.begin(), out.end());
        return out;
}

/** Template function to convert any input type to std::string
 */
inline std::string ConvToStr(const int in)
{
        return ConvNumeric(in);
}

/** Template function to convert any input type to std::string
 */
inline std::string ConvToStr(const long in)
{
        return ConvNumeric(in);
}

/** Template function to convert any input type to std::string
 */
inline std::string ConvToStr(const char* in)
{
        return in;
}

/** Template function to convert any input type to std::string
 */
inline std::string ConvToStr(const bool in)
{
        return (in ? "1" : "0");
}

/** Template function to convert any input type to std::string
 */
inline std::string ConvToStr(char in)
{
        return std::string(in,1);
}

/** Template function to convert any input type to std::string
 */
template <class T> inline std::string ConvToStr(const T &in)
{
        std::stringstream tmp;
        if (!(tmp << in)) return std::string();
        return tmp.str();
}

/** Template function to convert any input type to any other type
 * (usually an integer or numeric type)
 */
template<typename T> inline long ConvToInt(const T &in)
{
        std::stringstream tmp;
        if (!(tmp << in)) return 0;
        return atol(tmp.str().c_str());
}

/** This class contains various STATS counters
 * It is used by the InspIRCd class, which internally
 * has an instance of it.
 */
class serverstats
{
  public:
        /** Number of accepted connections
         */
        unsigned long statsAccept;
        /** Number of failed accepts
         */
        unsigned long statsRefused;
        /** Number of unknown commands seen
         */
        unsigned long statsUnknown;
        /** Number of nickname collisions handled
         */
        unsigned long statsCollisions;
        /** Number of DNS queries sent out
         */
        unsigned long statsDns;
        /** Number of good DNS replies received
         * NOTE: This may not tally to the number sent out,
         * due to timeouts and other latency issues.
         */
        unsigned long statsDnsGood;
        /** Number of bad (negative) DNS replies received
         * NOTE: This may not tally to the number sent out,
         * due to timeouts and other latency issues.
         */
        unsigned long statsDnsBad;
        /** Number of inbound connections seen
         */
        unsigned long statsConnects;
        /** Total bytes of data transmitted
         */
        unsigned long statsSent;
        /** Total bytes of data received
         */
        unsigned long statsRecv;
#ifdef _WIN32
        /** Cpu usage at last sample
        */
        FILETIME LastCPU;
        /** Time QP sample was read
        */
        LARGE_INTEGER LastSampled;
        /** QP frequency
        */
        LARGE_INTEGER QPFrequency;
#else
        /** Cpu usage at last sample
         */
        timeval LastCPU;
        /** Time last sample was read
         */
        timespec LastSampled;
#endif
        /** The constructor initializes all the counts to zero
         */
        serverstats()
                : statsAccept(0), statsRefused(0), statsUnknown(0), statsCollisions(0), statsDns(0),
                statsDnsGood(0), statsDnsBad(0), statsConnects(0), statsSent(0), statsRecv(0)
        {
        }
};

/** This class manages the generation and transmission of ISUPPORT. */
class CoreExport ISupportManager
{
private:
        /** The generated lines which are sent to clients. */
        std::vector<std::string> Lines;

public:
        /** (Re)build the ISUPPORT vector. */
        void Build();

        /** Returns the std::vector of ISUPPORT lines. */
        const std::vector<std::string>& GetLines()
        {
                return this->Lines;
        }

        /** Send the 005 numerics (ISUPPORT) to a user. */
        void SendTo(LocalUser* user);
};

DEFINE_HANDLER1(IsNickHandler, bool, const std::string&);
DEFINE_HANDLER2(GenRandomHandler, void, char*, size_t);
DEFINE_HANDLER1(IsIdentHandler, bool, const std::string&);
DEFINE_HANDLER1(IsChannelHandler, bool, const std::string&);
DEFINE_HANDLER1(RehashHandler, void, const std::string&);
DEFINE_HANDLER3(OnCheckExemptionHandler, ModResult, User*, Channel*, const std::string&);

/** The main class of the irc server.
 * This class contains instances of all the other classes in this software.
 * Amongst other things, it contains a ModeParser, a DNS object, a CommandParser
 * object, and a list of active Module objects, and facilities for Module
 * objects to interact with the core system it implements.
 */
class CoreExport InspIRCd
{
 private:
        /** Set up the signal handlers
         */
        void SetSignals();

        /** Daemonize the ircd and close standard input/output streams
         * @return True if the program daemonized succesfully
         */
        bool DaemonSeed();

        /** Iterate the list of BufferedSocket objects, removing ones which have timed out
         * @param TIME the current time
         */
        void DoSocketTimeouts(time_t TIME);

        /** The current time, updated in the mainloop
         */
        struct timespec TIME;

        /** A 64k buffer used to read socket data into
         * NOTE: update ValidateNetBufferSize if you change this
         */
        char ReadBuffer[65535];

 public:

        UIDGenerator UIDGen;

        /** Global cull list, will be processed on next iteration
         */
        CullList GlobalCulls;
        /** Actions that must happen outside of the current call stack */
        ActionList AtomicActions;

        /**** Functors ****/

        IsNickHandler HandleIsNick;
        IsIdentHandler HandleIsIdent;
        OnCheckExemptionHandler HandleOnCheckExemption;
        IsChannelHandler HandleIsChannel;
        RehashHandler HandleRehash;
        GenRandomHandler HandleGenRandom;

        /** Globally accessible fake user record. This is used to force mode changes etc across s2s, etc.. bit ugly, but.. better than how this was done in 1.1
         * Reason for it:
         * kludge alert!
         * SendMode expects a User* to send the numeric replies
         * back to, so we create it a fake user that isnt in the user
         * hash and set its descriptor to FD_MAGIC_NUMBER so the data
         * falls into the abyss :p
         */
        FakeUser* FakeClient;

        /** Find a user in the UUID hash
         * @param uid The UUID to find
         * @return A pointer to the user, or NULL if the user does not exist
         */
        User* FindUUID(const std::string &uid);

        /** Time this ircd was booted
         */
        time_t startup_time;

        /** Config file pathname specified on the commandline or via ./configure
         */
        std::string ConfigFileName;

        ExtensionManager Extensions;

        /** Mode handler, handles mode setting and removal
         */
        ModeParser* Modes;

        /** Command parser, handles client to server commands
         */
        CommandParser* Parser;

        /** Socket engine, handles socket activity events
         */
        SocketEngine* SE;

        /** Thread engine, Handles threading where required
         */
        ThreadEngine* Threads;

        /** The thread/class used to read config files in REHASH and on startup
         */
        ConfigReaderThread* ConfigThread;

        /** LogManager handles logging.
         */
        LogManager *Logs;

        /** ModuleManager contains everything related to loading/unloading
         * modules.
         */
        ModuleManager* Modules;

        /** BanCacheManager is used to speed up checking of restrictions on connection
         * to the IRCd.
         */
        BanCacheManager *BanCache;

        /** Stats class, holds miscellaneous stats counters
         */
        serverstats* stats;

        /**  Server Config class, holds configuration file data
         */
        ServerConfig* Config;

        /** Snomask manager - handles routing of snomask messages
         * to opers.
         */
        SnomaskManager* SNO;

        /** Timer manager class, triggers Timer timer events
         */
        TimerManager* Timers;

        /** X-Line manager. Handles G/K/Q/E line setting, removal and matching
         */
        XLineManager* XLines;

        /** User manager. Various methods and data associated with users.
         */
        UserManager *Users;

        /** Channel list, a hash_map containing all channels XXX move to channel manager class
         */
        chan_hash* chanlist;

        /** List of the open ports
         */
        std::vector<ListenSocket*> ports;

        /** Set to the current signal recieved
         */
        static sig_atomic_t s_signal;

        /** Protocol interface, overridden by server protocol modules
         */
        ProtocolInterface* PI;

        /** Holds extensible for user operquit
         */
        LocalStringExt OperQuit;

        /** Manages the generation and transmission of ISUPPORT. */
        ISupportManager ISupport;

        /** Get the current time
         * Because this only calls time() once every time around the mainloop,
         * it is much faster than calling time() directly.
         * @return The current time as an epoch value (time_t)
         */
        inline time_t Time() { return TIME.tv_sec; }
        /** The fractional time at the start of this mainloop iteration (nanoseconds) */
        inline long Time_ns() { return TIME.tv_nsec; }
        /** Update the current time. Don't call this unless you have reason to do so. */
        void UpdateTime();

        /** Generate a random string with the given length
         * @param length The length in bytes
         * @param printable if false, the string will use characters 0-255; otherwise,
         * it will be limited to 0x30-0x7E ('0'-'~', nonspace printable characters)
         */
        std::string GenRandomStr(int length, bool printable = true);
        /** Generate a random integer.
         * This is generally more secure than rand()
         */
        unsigned long GenRandomInt(unsigned long max);

        /** Fill a buffer with random bits */
        caller2<void, char*, size_t> GenRandom;

        /** Bind all ports specified in the configuration file.
         * @return The number of ports bound without error
         */
        int BindPorts(FailedPortList &failed_ports);

        /** Binds a socket on an already open file descriptor
         * @param sockfd A valid file descriptor of an open socket
         * @param port The port number to bind to
         * @param addr The address to bind to (IP only)
         * @param dolisten Should this port be listened on?
         * @return True if the port was bound successfully
         */
        bool BindSocket(int sockfd, int port, const char* addr, bool dolisten = true);

        /** Gets the GECOS (description) field of the given server.
         * If the servername is not that of the local server, the name
         * is passed to handling modules which will attempt to determine
         * the GECOS that bleongs to the given servername.
         * @param servername The servername to find the description of
         * @return The description of this server, or of the local server
         */
        std::string GetServerDescription(const std::string& servername);

        /** Find a user in the nick hash.
         * If the user cant be found in the nick hash check the uuid hash
         * @param nick The nickname to find
         * @return A pointer to the user, or NULL if the user does not exist
         */
        User* FindNick(const std::string &nick);

        /** Find a user in the nick hash ONLY
         */
        User* FindNickOnly(const std::string &nick);

        /** Find a channel in the channels hash
         * @param chan The channel to find
         * @return A pointer to the channel, or NULL if the channel does not exist
         */
        Channel* FindChan(const std::string &chan);

        /** Check we aren't running as root, and exit if we are
         * @return Depending on the configuration, this function may never return
         */
        void CheckRoot();

        /** Return true if a channel name is valid
         * @param chname A channel name to verify
         * @return True if the name is valid
         */
        caller1<bool, const std::string&> IsChannel;

        /** Return true if str looks like a server ID
         * @param string to check against
         */
        static bool IsSID(const std::string& sid);

        /** Rehash the local server
         */
        caller1<void, const std::string&> Rehash;

        /** Handles incoming signals after being set
         * @param signal the signal recieved
         */
        void SignalHandler(int signal);

        /** Sets the signal recieved
         * @param signal the signal recieved
         */
        static void SetSignal(int signal);

        /** Causes the server to exit after unloading modules and
         * closing all open file descriptors.
         *
         * @param status The exit code to give to the operating system
         * (See the ExitStatus enum for valid values)
         */
        void Exit(int status);

        /** Causes the server to exit immediately with exit code 0.
         * The status code is required for signal handlers, and ignored.
         */

        /** Printf-wrapper.
        * @param How you want it formatted
        * @param ...
        * @return The formatted string
        */
        static const char* Format(const char* formatString, ...) CUSTOM_PRINTF(1, 2);
        static const char* Format(va_list &vaList, const char* formatString) CUSTOM_PRINTF(2, 0);

        static void QuickExit(int status);

        /** Return a count of channels on the network
         * @return The number of channels
         */
        long ChannelCount() const { return chanlist->size(); }

        /** Send an error notice to all local users, opered and unopered
         * @param s The error string to send
         */
        void SendError(const std::string &s);

        /** Return true if a nickname is valid
         * @param n A nickname to verify
         * @return True if the nick is valid
         */
        caller1<bool, const std::string&> IsNick;

        /** Return true if an ident is valid
         * @param An ident to verify
         * @return True if the ident is valid
         */
        caller1<bool, const std::string&> IsIdent;

        /** Add a command to this server's command parser
         * @param f A Command command handler object to add
         * @throw ModuleException Will throw ModuleExcption if the command already exists
         */
        inline void AddCommand(Command *f)
        {
                Modules->AddService(*f);
        }

        /** Send a modechange.
         * The parameters provided are identical to that sent to the
         * handler for class cmd_mode.
         * @param parameters The mode parameters
         * @param user The user to send error messages to
         */
        void SendMode(const std::vector<std::string>& parameters, User *user);

        /** Send a modechange and route it to the network.
         * The parameters provided are identical to that sent to the
         * handler for class cmd_mode.
         * @param parameters The mode parameters
         * @param user The user to send error messages to
         */
        void SendGlobalMode(const std::vector<std::string>& parameters, User *user);

        /** Match two strings using pattern matching, optionally, with a map
         * to check case against (may be NULL). If map is null, match will be case insensitive.
         * @param str The literal string to match against
         * @param mask The glob pattern to match against.
         * @param map The character map to use when matching.
         */
        static bool Match(const std::string &str, const std::string &mask, unsigned const char *map = NULL);
        static bool Match(const char *str, const char *mask, unsigned const char *map = NULL);

        /** Match two strings using pattern matching, optionally, with a map
         * to check case against (may be NULL). If map is null, match will be case insensitive.
         * Supports CIDR patterns as well as globs.
         * @param str The literal string to match against
         * @param mask The glob or CIDR pattern to match against.
         * @param map The character map to use when matching.
         */
        static bool MatchCIDR(const std::string &str, const std::string &mask, unsigned const char *map = NULL);
        static bool MatchCIDR(const char *str, const char *mask, unsigned const char *map = NULL);

        /** Matches a hostname and IP against a space delimited list of hostmasks.
         * @param masks The space delimited masks to match against.
         * @param hostname The hostname to try and match.
         * @param ipaddr The IP address to try and match.
         */
        static bool MatchMask(const std::string& masks, const std::string& hostname, const std::string& ipaddr);

        /** Return true if the given parameter is a valid nick!user\@host mask
         * @param mask A nick!user\@host masak to match against
         * @return True i the mask is valid
         */
        bool IsValidMask(const std::string &mask);

        /** Strips all color codes from the given string
         * @param sentence The string to strip from
         */
        static void StripColor(std::string &sentence);

        /** Parses color codes from string values to actual color codes
         * @param input The data to process
         */
        static void ProcessColors(file_cache& input);

        /** Rehash the local server
         */
        void RehashServer();

        /** Check if the given nickmask matches too many users, send errors to the given user
         * @param nick A nickmask to match against
         * @param user A user to send error text to
         * @return True if the nick matches too many users
         */
        bool NickMatchesEveryone(const std::string &nick, User* user);

        /** Check if the given IP mask matches too many users, send errors to the given user
         * @param ip An ipmask to match against
         * @param user A user to send error text to
         * @return True if the ip matches too many users
         */
        bool IPMatchesEveryone(const std::string &ip, User* user);

        /** Check if the given hostmask matches too many users, send errors to the given user
         * @param mask A hostmask to match against
         * @param user A user to send error text to
         * @return True if the host matches too many users
         */
        bool HostMatchesEveryone(const std::string &mask, User* user);

        /** Calculate a duration in seconds from a string in the form 1y2w3d4h6m5s
         * @param str A string containing a time in the form 1y2w3d4h6m5s
         * (one year, two weeks, three days, four hours, six minutes and five seconds)
         * @return The total number of seconds
         */
        static unsigned long Duration(const std::string& str);

        /** Attempt to compare a password to a string from the config file.
         * This will be passed to handling modules which will compare the data
         * against possible hashed equivalents in the input string.
         * @param ex The object (user, server, whatever) causing the comparison.
         * @param data The data from the config file
         * @param input The data input by the oper
         * @param hashtype The hash from the config file
         * @return 0 if the strings match, 1 or -1 if they do not
         */
        int PassCompare(Extensible* ex, const std::string &data, const std::string &input, const std::string &hashtype);

        /** Check if a given server is a uline.
         * An empty string returns true, this is by design.
         * @param server The server to check for uline status
         * @return True if the server is a uline OR the string is empty
         */
        bool ULine(const std::string& server);

        /** Returns true if the uline is 'silent' (doesnt generate
         * remote connect notices etc).
         */
        bool SilentULine(const std::string& server);

        /** Returns the full version string of this ircd
         * @return The version string
         */
        std::string GetVersionString(bool getFullVersion = false);

        /** Attempt to write the process id to a given file
         * @param filename The PID file to attempt to write to
         * @return This function may bail if the file cannot be written
         */
        void WritePID(const std::string &filename);

        /** This constructor initialises all the subsystems and reads the config file.
         * @param argc The argument count passed to main()
         * @param argv The argument list passed to main()
         * @throw <anything> If anything is thrown from here and makes it to
         * you, you should probably just give up and go home. Yes, really.
         * It's that bad. Higher level classes should catch any non-fatal exceptions.
         */
        InspIRCd(int argc, char** argv);

        /** Send a line of WHOIS data to a user.
         * @param user user to send the line to
         * @param dest user being WHOISed
         * @param numeric Numeric to send
         * @param text Text of the numeric
         */
        void SendWhoisLine(User* user, User* dest, int numeric, const std::string &text);

        /** Send a line of WHOIS data to a user.
         * @param user user to send the line to
         * @param dest user being WHOISed
         * @param numeric Numeric to send
         * @param format Format string for the numeric
         * @param ... Parameters for the format string
         */
        void SendWhoisLine(User* user, User* dest, int numeric, const char* format, ...) CUSTOM_PRINTF(5, 6);

        /** Called to check whether a channel restriction mode applies to a user
         * @param User that is attempting some action
         * @param Channel that the action is being performed on
         * @param Action name
         */
        caller3<ModResult, User*, Channel*, const std::string&> OnCheckExemption;

        /** Restart the server.
         * This function will not return. If an error occurs,
         * it will throw an instance of CoreException.
         * @param reason The restart reason to show to all clients
         * @throw CoreException An instance of CoreException indicating the error from execv().
         */
        void Restart(const std::string &reason);

        /** Prepare the ircd for restart or shutdown.
         * This function unloads all modules which can be unloaded,
         * closes all open sockets, and closes the logfile.
         */
        void Cleanup();

        /** Return a time_t as a human-readable string.
         */
        std::string TimeString(time_t curtime);

        /** Begin execution of the server.
         * NOTE: this function NEVER returns. Internally,
         * it will repeatedly loop.
         * @return The return value for this function is undefined.
         */
        int Run();

        char* GetReadBuffer()
        {
                return this->ReadBuffer;
        }
};

ENTRYPOINT;

template<class Cmd>
class CommandModule : public Module
{
        Cmd cmd;
 public:
        CommandModule() : cmd(this)
        {
        }

        void init()
        {
                ServerInstance->Modules->AddService(cmd);
        }

        Version GetVersion()
        {
                return Version(cmd.name, VF_VENDOR|VF_CORE);
        }
};

inline void InspIRCd::SendMode(const std::vector<std::string>& parameters, User* user)
{
        this->Modes->Process(parameters, user);
}