JLU5-SupyPlugins/Hostmasks/plugin.py

###
# Copyright (c) 2013, GLolol (GLolol1@hotmail.com)
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice, this
# list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
#
# * Neither the name of the software nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
###

import supybot.utils as utils
from supybot.commands import *
import supybot.plugins as plugins
import supybot.ircutils as ircutils
import supybot.callbacks as callbacks
import re
import socket
try:
    from supybot.i18n import PluginInternationalization
    _ = PluginInternationalization('Hostmasks')
except:
    # Placeholder that allows to run the plugin on a bot
    # without the i18n module
    _ = lambda x:x

class Hostmasks(callbacks.Plugin):
    """Add the help for "@plugin help Hostmasks" here
    This should describe *how* to use this plugin."""
    pass
    
    def _SplitHostmask(self, irc, nick):
        # Split the hostmask into 3 sections: nick, ident, and host.
        try: 
            splithostmask = re.split('[!@]', irc.state.nickToHostmask(nick))
        except KeyError:
            irc.error('There is no such nick \'%s\'%s.' % nick, Raise=True)
        if len(splithostmask) != 3:
            # This is just here so in case this does happen, the bot doesn't
            # error out when retrieving parts of the hostmask.
            self.log.warning('Hostmasks: Invalid hostmask length received' 
                ' for %s on %s. This should not be happening!'
                % (nick, irc.network))
            irc.error('Invalid hostmask length received! (this should not '
                'be happening!)', Raise=True)
        return splithostmask
    
    def gethost(self, irc, msg, args, nick):
        """[<nick>]
        Returns the host of <nick>. If <nick> is not given, returns the host
        of the person who called the command.
        """
        if not nick:
            nick = msg.nick
        irc.reply(self._SplitHostmask(irc, nick)[2])
    gethost = wrap(gethost, [(additional('nick'))])
    
    def getident(self, irc, msg, args, nick):
        """[<nick>]
        Returns the ident of <nick>. If <nick> is not given, returns the host
        of the person who called the command.
        """
        if not nick:
            nick = msg.nick
        irc.reply(self._SplitHostmask(irc, nick)[1])
    getident = wrap(getident, [(additional('nick'))])
    
    def me(self, irc, msg, args):
        """takes no arguments.
        Returns the nick of the person who called the command.
        """
        irc.reply(msg.nick)
    me = wrap(me)
    
    def _isv4IP(self, ipstr):
        try:
            socket.inet_aton(ipstr)
            return True
        except socket.error:
            return False
            
    def _isv4cloak(self, hostname):
        # Smart bans: look for charybdis-style IP cloaks, Unreal/InspIRCd
        # styles will work (hopefully) using the regular wildhost parsing.
        v4cloak = re.match("^(?:[0-9]{1,3}\.){2}[a-z]{1,3}\.[a-z]{1,3}", hostname)
        if v4cloak:
            return True
        else:
            return False

    def _isv6IP(self, ipstr):    
        try:
            socket.inet_pton(socket.AF_INET6, ipstr)
            return True
        except socket.error:
            return False
        except AttributeError:
            # if inet_pton not implemented in the OS used (currently only
            # works on Unix), use our super-duper lazy regexp instead!
            v6ip = re.match("([0-9]{1,4}:{1,2}){2,8}", ipstr)
            if v6ip:
                return True
            else:
                return False         
            
    def _isvHost(self, hostname):
        isvHost = re.search("/", hostname)
        if isvHost:
            return True
        else:
            return False
    
    def _isv6cloak(self, hostname):
        if re.search(":", hostname):
            # Look for unreal-style cloaks (1234abcd:2345bcde:3456cdef:IP)
            v6cloaku = re.match("([0-9A-F]{8}:){3}IP", hostname)
            # Use our super lazy regexp for charybdis-style v6 cloaks
            v6cloakc = re.match("([0-9a-z]{1,4}:{1,2}){2,8}", hostname)       
            if v6cloaku:
                return 'u'
            elif v6cloakc:
                return 'c'
        else: # doesn't even include a : , why bother checking?
            return False

    def banmask(self, irc, msg, args, nick):
        """[<nick>]
        Returns a nice banmask for <nick>. If <nick> is not given, returns a
         banmask for the person who called the command.
        """
        if not nick:
            nick = msg.nick
        splithostmask = self._SplitHostmask(irc, nick)
        bantype = self.registryValue('banType')
        if bantype == '1':
            banmask = '%s%s' % ('*!*@', splithostmask[2])
        elif bantype == '2':
            banmask = '%s%s%s%s' % ('*!', splithostmask[1], '@', 
                splithostmask[2])
        else:
            splithost = re.split(r"[.]", splithostmask[2], 2)
            wildhost = ''
            if self.registryValue('smartBans'):
                if self._isv4IP(splithostmask[2]) or self._isv4cloak(splithostmask[2]):
                    v4cloak = re.split(r"\.", splithostmask[2], 2)
                    wildhost = v4cloak[0] + '.' + v4cloak[1] + '.*'
                elif self._isvHost(splithostmask[2]):
                    wildhost = splithostmask[2]
                elif self._isv6IP(splithostmask[2]) or self._isv6cloak(splithostmask[2]):
                    wildhost = splithostmask[2] # TODO: support ipv6 ranges
            if not wildhost:
                if len(splithost) <= 2:
                    wildhost = splithostmask[2] # Hostmask is too short to split
                else:
                    wildhost = '%s%s%s%s' % ('*.', splithost[1], '.', 
                        splithost[2])
            if bantype == '3':
                banmask = '%s%s' % ('*!*@', wildhost) 
            if bantype == '4':
                banmask = '%s%s%s%s' % ('*!', splithostmask[1], '@', wildhost)
        irc.reply(banmask)
    banmask = wrap(banmask, [(additional('nick'))])
    
Class = Hostmasks


# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: