src/cidr.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
   6  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   7  *
   8  * This file is part of InspIRCd.  InspIRCd is free software: you can
   9  * redistribute it and/or modify it under the terms of the GNU General Public
  10  * License as published by the Free Software Foundation, version 2.
  11  *
  12  * This program is distributed in the hope that it will be useful, but WITHOUT
  13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 /* $Core */
  23
  24 #include "inspircd.h"
  25
  26 /* Match CIDR strings, e.g. 127.0.0.1 to 127.0.0.0/8 or 3ffe:1:5:6::8 to 3ffe:1::0/32
  27  *
  28  * This will also attempt to match any leading usernames or nicknames on the mask, using
  29  * match(), when match_with_username is true.
  30  */
  31 bool irc::sockets::MatchCIDR(const std::string &address, const std::string &cidr_mask, bool match_with_username)
  32 {
  33         std::string address_copy;
  34         std::string cidr_copy;
  35
  36         /* The caller is trying to match ident@<mask>/bits.
  37          * Chop off the ident@ portion, use match() on it
  38          * seperately.
  39          */
  40         if (match_with_username)
  41         {
  42                 /* Use strchr not strrchr, because its going to be nearer to the left */
  43                 std::string::size_type username_mask_pos = cidr_mask.rfind('@');
  44                 std::string::size_type username_addr_pos = address.rfind('@');
  45
  46                 /* Both strings have an @ symbol in them */
  47                 if (username_mask_pos != std::string::npos && username_addr_pos != std::string::npos)
  48                 {
  49                         /* Try and match() the strings before the @
  50                          * symbols, and recursively call MatchCIDR without
  51                          * username matching enabled to match the host part.
  52                          */
  53                         return (InspIRCd::Match(address.substr(0, username_addr_pos), cidr_mask.substr(0, username_mask_pos), ascii_case_insensitive_map) &&
  54                                         MatchCIDR(address.substr(username_addr_pos + 1), cidr_mask.substr(username_mask_pos + 1), false));
  55                 }
  56                 else
  57                 {
  58                         address_copy = address.substr(username_addr_pos + 1);
  59                         cidr_copy = cidr_mask.substr(username_mask_pos + 1);
  60                 }
  61         }
  62         else
  63         {
  64                 address_copy.assign(address);
  65                 cidr_copy.assign(cidr_mask);
  66         }
  67
  68         const std::string::size_type per_pos = cidr_copy.rfind('/');
  69         if ((per_pos == std::string::npos) || (per_pos == cidr_copy.length()-1)
  70                 || (cidr_copy.find_first_not_of("0123456789", per_pos+1) != std::string::npos)
  71                 || (cidr_copy.find_first_not_of("0123456789abcdefABCDEF.:") < per_pos))
  72         {
  73                 // The CIDR mask is invalid
  74                 return false;
  75         }
  76
  77         irc::sockets::sockaddrs addr;
  78         irc::sockets::aptosa(address_copy, 0, addr);
  79
  80         irc::sockets::cidr_mask mask(cidr_copy);
  81         irc::sockets::cidr_mask mask2(addr, mask.length);
  82
  83         return mask == mask2;
  84 }
  85
  86