### # Copyright (c) 2003-2005, Jeremiah Fincher # Copyright (c) 2010-2011, James McCoy # Copyright (c) 2010-2021, Valentin Lorentz # 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): """ Returns the ip of or the reverse DNS hostname of . """ 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): """ Returns WHOIS information on the registration of . """ 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.iana.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 = _(' ') % line.split('@')[-1] elif line.startswith('Registrar Organization:'): url = _(' ') % 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): """ 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: