src/modules/extra/m_regex_posix.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22 #include "m_regex.h"
  23 #include <sys/types.h>
  24 #include <regex.h>
  25
  26 /* $ModDesc: Regex Provider Module for POSIX Regular Expressions */
  27 /* $ModDep: m_regex.h */
  28
  29 class POSIXRegexException : public ModuleException
  30 {
  31 public:
  32         POSIXRegexException(const std::string& rx, const std::string& error)
  33                 : ModuleException(std::string("Error in regex ") + rx + ": " + error)
  34         {
  35         }
  36 };
  37
  38 class POSIXRegex : public Regex
  39 {
  40 private:
  41         regex_t regbuf;
  42
  43 public:
  44         POSIXRegex(const std::string& rx, bool extended) : Regex(rx)
  45         {
  46                 int flags = (extended ? REG_EXTENDED : 0) | REG_NOSUB;
  47                 int errcode;
  48                 errcode = regcomp(&regbuf, rx.c_str(), flags);
  49                 if (errcode)
  50                 {
  51                         // Get the error string into a std::string. YUCK this involves at least 2 string copies.
  52                         std::string error;
  53                         char* errbuf;
  54                         size_t sz = regerror(errcode, &regbuf, NULL, 0);
  55                         errbuf = new char[sz + 1];
  56                         memset(errbuf, 0, sz + 1);
  57                         regerror(errcode, &regbuf, errbuf, sz + 1);
  58                         error = errbuf;
  59                         delete[] errbuf;
  60                         regfree(&regbuf);
  61                         throw POSIXRegexException(rx, error);
  62                 }
  63         }
  64
  65         virtual ~POSIXRegex()
  66         {
  67                 regfree(&regbuf);
  68         }
  69
  70         virtual bool Matches(const std::string& text)
  71         {
  72                 if (regexec(&regbuf, text.c_str(), 0, NULL, 0) == 0)
  73                 {
  74                         // Bang. :D
  75                         return true;
  76                 }
  77                 return false;
  78         }
  79 };
  80
  81 class PosixFactory : public RegexFactory
  82 {
  83  public:
  84         bool extended;
  85         PosixFactory(Module* m) : RegexFactory(m, "regex/posix") {}
  86         Regex* Create(const std::string& expr)
  87         {
  88                 return new POSIXRegex(expr, extended);
  89         }
  90 };
  91
  92 class ModuleRegexPOSIX : public Module
  93 {
  94         PosixFactory ref;
  95 public:
  96         ModuleRegexPOSIX() : ref(this) {
  97                 ServerInstance->Modules->AddService(ref);
  98                 Implementation eventlist[] = { I_OnRehash };
  99                 ServerInstance->Modules->Attach(eventlist, this, 1);
 100                 OnRehash(NULL);
 101         }
 102
 103         Version GetVersion()
 104         {
 105                 return Version("Regex Provider Module for POSIX Regular Expressions", VF_VENDOR);
 106         }
 107
 108         void OnRehash(User* u)
 109         {
 110                 ConfigReader Conf;
 111                 ref.extended = Conf.ReadFlag("posix", "extended", 0);
 112         }
 113 };
 114
 115 MODULE_INIT(ModuleRegexPOSIX)