mirror of https://github.com/Mikaela/Limnoria.git
218 lines
8.6 KiB
Python
218 lines
8.6 KiB
Python
###
|
|
# Copyright (c) 2003-2005, Jeremiah Fincher
|
|
# Copyright (c) 2010-2011, James McCoy
|
|
# 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 author of this software nor the name of
|
|
# contributors to this software may be used to endorse or promote products
|
|
# derived from this software without specific prior written consent.
|
|
#
|
|
# 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 OWNER 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 time
|
|
import socket
|
|
import telnetlib
|
|
|
|
import supybot.conf as conf
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
from supybot.utils.iter import any
|
|
import supybot.callbacks as callbacks
|
|
from supybot.i18n import PluginInternationalization, internationalizeDocstring
|
|
_ = PluginInternationalization('Internet')
|
|
|
|
class Internet(callbacks.Plugin):
|
|
"""Provides commands to query DNS, search WHOIS databases,
|
|
and convert IPs to hex."""
|
|
threaded = True
|
|
@internationalizeDocstring
|
|
def dns(self, irc, msg, args, host):
|
|
"""<host|ip>
|
|
|
|
Returns the ip of <host> or the reverse DNS hostname of <ip>.
|
|
"""
|
|
if utils.net.isIP(host):
|
|
hostname = socket.getfqdn(host)
|
|
if hostname == host:
|
|
irc.reply(_('Host not found.'))
|
|
else:
|
|
irc.reply(hostname)
|
|
else:
|
|
try:
|
|
ips = socket.getaddrinfo(host, None)
|
|
ips = map(lambda x:x[4][0], ips)
|
|
ordered_unique_ips = []
|
|
unique_ips = set()
|
|
for ip in ips:
|
|
if ip not in unique_ips:
|
|
ordered_unique_ips.append(ip)
|
|
unique_ips.add(ip)
|
|
irc.reply(format('%L', ordered_unique_ips))
|
|
except socket.error:
|
|
irc.reply(_('Host not found.'))
|
|
dns = wrap(dns, ['something'])
|
|
|
|
_domain = ['Domain Name', 'Server Name', 'domain']
|
|
_netrange = ['NetRange', 'inetnum']
|
|
_registrar = ['Sponsoring Registrar', 'Registrar', 'source']
|
|
_netname = ['NetName', 'inetname']
|
|
_updated = ['Last Updated On', 'Domain Last Updated Date', 'Updated Date',
|
|
'Last Modified', 'changed', 'last-modified']
|
|
_created = ['Created On', 'Domain Registration Date', 'Creation Date',
|
|
'created', 'RegDate']
|
|
_expires = ['Expiration Date', 'Domain Expiration Date']
|
|
_status = ['Status', 'Domain Status', 'status']
|
|
@internationalizeDocstring
|
|
def whois(self, irc, msg, args, domain):
|
|
"""<domain>
|
|
|
|
Returns WHOIS information on the registration of <domain>.
|
|
"""
|
|
if utils.net.isIP(domain):
|
|
whois_server = 'whois.arin.net'
|
|
usertld = 'ipaddress'
|
|
elif '.' in domain:
|
|
usertld = domain.split('.')[-1]
|
|
whois_server = '%s.whois-servers.net' % usertld
|
|
else:
|
|
usertld = None
|
|
whois_server = 'whois.iana.org'
|
|
try:
|
|
sock = utils.net.getSocket(whois_server,
|
|
vhost=conf.supybot.protocols.irc.vhost(),
|
|
vhostv6=conf.supybot.protocols.irc.vhostv6(),
|
|
)
|
|
sock.connect((whois_server, 43))
|
|
except socket.error as e:
|
|
irc.error(str(e))
|
|
return
|
|
sock.settimeout(5)
|
|
if usertld == 'com':
|
|
sock.send(b'=')
|
|
elif usertld == 'ipaddress':
|
|
sock.send(b'n + ')
|
|
sock.send(domain.encode('ascii'))
|
|
sock.send(b'\r\n')
|
|
|
|
s = b''
|
|
end_time = time.time() + 5
|
|
try:
|
|
while end_time>time.time():
|
|
time.sleep(0.1)
|
|
s += sock.recv(4096)
|
|
except socket.error:
|
|
pass
|
|
sock.close()
|
|
server = netrange = netname = registrar = updated = created = expires = status = ''
|
|
for line in s.splitlines():
|
|
line = line.decode('utf8').strip()
|
|
if not line or ':' not in line:
|
|
continue
|
|
if not server and any(line.startswith, self._domain):
|
|
server = ':'.join(line.split(':')[1:]).strip().lower()
|
|
# Let's add this check so that we don't respond with info for
|
|
# a different domain. E.g., doing a whois for microsoft.com
|
|
# and replying with the info for microsoft.com.wanadoodoo.com
|
|
if server != domain:
|
|
server = ''
|
|
continue
|
|
if not netrange and any(line.startswith, self._netrange):
|
|
netrange = ':'.join(line.split(':')[1:]).strip()
|
|
if not server and not netrange:
|
|
continue
|
|
if not registrar and any(line.startswith, self._registrar):
|
|
registrar = ':'.join(line.split(':')[1:]).strip()
|
|
elif not netname and any(line.startswith, self._netname):
|
|
netname = ':'.join(line.split(':')[1:]).strip()
|
|
elif not updated and any(line.startswith, self._updated):
|
|
s = ':'.join(line.split(':')[1:]).strip()
|
|
updated = _('updated %s') % s
|
|
elif not created and any(line.startswith, self._created):
|
|
s = ':'.join(line.split(':')[1:]).strip()
|
|
created = _('registered %s') % s
|
|
elif not expires and any(line.startswith, self._expires):
|
|
s = ':'.join(line.split(':')[1:]).strip()
|
|
expires = _('expires %s') % s
|
|
elif not status and any(line.startswith, self._status):
|
|
status = ':'.join(line.split(':')[1:]).strip().lower()
|
|
if not status:
|
|
status = 'unknown'
|
|
try:
|
|
t = telnetlib.Telnet('whois.pir.org', 43)
|
|
except socket.error as e:
|
|
irc.error(str(e))
|
|
return
|
|
t.write(b'registrar ')
|
|
t.write(registrar.split('(')[0].strip().encode('ascii'))
|
|
t.write(b'\n')
|
|
s = t.read_all()
|
|
url = ''
|
|
for line in s.splitlines():
|
|
line = line.decode('ascii').strip()
|
|
if not line:
|
|
continue
|
|
if line.startswith('Email'):
|
|
url = _(' <registered at %s>') % line.split('@')[-1]
|
|
elif line.startswith('Registrar Organization:'):
|
|
url = _(' <registered by %s>') % line.split(':')[1].strip()
|
|
elif line == 'Not a valid ID pattern':
|
|
url = ''
|
|
if (server or netrange) and status:
|
|
entity = server or 'Net range %s%s' % \
|
|
(netrange, ' (%s)' % netname if netname else '')
|
|
info = filter(None, [status, created, updated, expires])
|
|
s = format(_('%s%s is %L.'), entity, url, info)
|
|
irc.reply(s)
|
|
else:
|
|
irc.error(_('I couldn\'t find such a domain.'))
|
|
whois = wrap(whois, ['lowered'])
|
|
|
|
@internationalizeDocstring
|
|
def hexip(self, irc, msg, args, ip):
|
|
"""<ip>
|
|
|
|
Returns the hexadecimal IP for that IP.
|
|
"""
|
|
ret = ""
|
|
if utils.net.isIPV4(ip):
|
|
quads = ip.split('.')
|
|
for quad in quads:
|
|
i = int(quad)
|
|
ret += '%02X' % i
|
|
else:
|
|
octets = ip.split(':')
|
|
for octet in octets:
|
|
if octet:
|
|
i = int(octet, 16)
|
|
ret += '%04X' % i
|
|
else:
|
|
missing = (8 - len(octets)) * 4
|
|
ret += '0' * missing
|
|
irc.reply(ret)
|
|
hexip = wrap(hexip, ['ip'])
|
|
Internet = internationalizeDocstring(Internet)
|
|
|
|
Class = Internet
|
|
|
|
|
|
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79:
|