bancache.h

/*
 * InspIRCd -- Internet Relay Chat Daemon
 *
 *   Copyright (C) 2008 Craig Edwards <[email protected]>
 *   Copyright (C) 2007 Robin Burchell <[email protected]>
 *
 * 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/>.
 */
 
 
#ifndef BANCACHE_H
#define BANCACHE_H
 
class CoreExport BanCacheHit
{
 public:
        std::string Type;
        std::string Reason;
        std::string IP;
        time_t Expiry;
 
        BanCacheHit(const std::string &ip, const std::string &type, const std::string &reason)
        {
                this->Type = type;
                this->Reason = reason;
                this->IP = ip;
                this->Expiry = ServerInstance->Time() + 86400; // a day. this might seem long, but entries will be removed as glines/etc expire.
        }
 
        // overridden to allow custom time
        BanCacheHit(const std::string &ip, const std::string &type, const std::string &reason, time_t seconds)
        {
                this->Type = type;
                this->Reason = reason;
                this->IP = ip;
                this->Expiry = ServerInstance->Time() + seconds;
        }
};
 
/* A container of ban cache items.
 * must be defined after class BanCacheHit.
 */
typedef nspace::hash_map<std::string, BanCacheHit*, nspace::hash<std::string> > BanCacheHash;
 
class CoreExport BanCacheManager
{
 private:
        BanCacheHash* BanHash;
 public:
 
        BanCacheHit *AddHit(const std::string &ip, const std::string &type, const std::string &reason);
 
        // Overridden to allow an optional number of seconds before expiry
        BanCacheHit *AddHit(const std::string &ip, const std::string &type, const std::string &reason, time_t seconds);
        BanCacheHit *GetHit(const std::string &ip);
        bool RemoveHit(BanCacheHit *b);
 
        unsigned int RemoveEntries(const std::string &type, bool positive);
 
        BanCacheManager()
        {
                this->BanHash = new BanCacheHash();
        }
        ~BanCacheManager();
        void RehashCache();
};
 
#endif