Limnoria/plugins/Internet/plugin.py

218 lines
8.6 KiB
Python
Raw Normal View History

###
# 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':
2013-01-06 16:58:02 +01:00
sock.send(b'=')
elif usertld == 'ipaddress':
sock.send(b'n + ')
sock.send(domain.encode('ascii'))
sock.send(b'\r\n')
2013-01-06 16:58:02 +01:00
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():
2013-08-06 12:16:52 +02:00
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
2012-08-04 20:07:24 +02:00
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():
2012-08-04 20:07:24 +02:00
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: