mirror of
https://github.com/xcat2/xNBA.git
synced 2024-11-22 09:31:51 +00:00
8406115834
Access to the gpxe.org and etherboot.org domains and associated resources has been revoked by the registrant of the domain. Work around this problem by renaming project from gPXE to iPXE, and updating URLs to match. Also update README, LOG and COPYRIGHTS to remove obsolete information. Signed-off-by: Michael Brown <mcb30@ipxe.org>
125 lines
3.2 KiB
Python
Executable File
125 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright (C) 2008 Stefan Hajnoczi <stefanha@gmail.com>.
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as
|
|
# published by the Free Software Foundation; either version 2 of the
|
|
# License, or any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
import re
|
|
import socket
|
|
import errcode
|
|
|
|
HOST = 'irc.freenode.net'
|
|
PORT = 6667
|
|
NICK = 'ipxebot'
|
|
CHAN = '#etherboot'
|
|
NICKSERV_PASSWORD = None
|
|
IDENT = 'ipxebot'
|
|
REALNAME = 'iPXE bot'
|
|
|
|
ERRCODE_RE = re.compile(r'(errcode|Error)\s+((0x)?[0-9a-fA-F]{8})')
|
|
|
|
NO_ARGS = -1
|
|
|
|
handlers = {}
|
|
|
|
def nick_from_mask(mask):
|
|
return (mask.find('!') > -1 and mask.split('!', 1)[0]) or mask
|
|
|
|
def autojoin():
|
|
del handlers['376']
|
|
if NICKSERV_PASSWORD:
|
|
pmsg('nickserv', 'identify %s' % NICKSERV_PASSWORD)
|
|
if CHAN:
|
|
cmd('JOIN %s' % CHAN)
|
|
|
|
def ping(_, arg):
|
|
cmd('PONG %s' % arg)
|
|
|
|
def privmsg(_, target, msg):
|
|
if target == CHAN:
|
|
replyto = target
|
|
if msg.find(NICK) == -1:
|
|
return
|
|
elif target == NICK:
|
|
replyto = nick_from_mask(who)
|
|
m = ERRCODE_RE.search(msg)
|
|
if m:
|
|
try:
|
|
pmsg(replyto, str(errcode.Errcode(int(m.groups()[1], 16))))
|
|
except ValueError:
|
|
pass
|
|
if msg.find('help') > -1:
|
|
pmsg(replyto, 'I look up iPXE error codes. Message me like this:')
|
|
pmsg(replyto, 'errcode 0x12345678 OR Error 0x12345678')
|
|
|
|
def add_handler(command, handler, nargs):
|
|
handlers[command] = (handler, nargs)
|
|
|
|
def cmd(msg):
|
|
sock.sendall('%s\r\n' % msg)
|
|
|
|
def pmsg(target, msg):
|
|
cmd('PRIVMSG %s :%s' % (target, msg))
|
|
|
|
def dispatch(args):
|
|
command = args[0]
|
|
if command in handlers:
|
|
h = handlers[command]
|
|
if h[1] == NO_ARGS:
|
|
h[0]()
|
|
elif len(args) == h[1]:
|
|
h[0](*args)
|
|
|
|
def parse(line):
|
|
if line[0] == ':':
|
|
who, line = line.split(None, 1)
|
|
who = who[1:]
|
|
else:
|
|
who = None
|
|
args = []
|
|
while line and line[0] != ':' and line.find(' ') != -1:
|
|
fields = line.split(None, 1)
|
|
if len(fields) == 1:
|
|
fields.append(None)
|
|
arg, line = fields
|
|
args.append(arg)
|
|
if line:
|
|
if line[0] == ':':
|
|
args.append(line[1:])
|
|
else:
|
|
args.append(line)
|
|
return who, args
|
|
|
|
add_handler('376', autojoin, NO_ARGS)
|
|
add_handler('PING', ping, 2)
|
|
add_handler('PRIVMSG', privmsg, 3)
|
|
|
|
sock = socket.socket()
|
|
sock.connect((HOST, PORT))
|
|
cmd('NICK %s' % NICK)
|
|
cmd('USER %s none none :%s' % (IDENT, REALNAME))
|
|
|
|
rbuf = ''
|
|
while True:
|
|
r = sock.recv(4096)
|
|
if not r:
|
|
break
|
|
rbuf += r
|
|
|
|
while rbuf.find('\r\n') != -1:
|
|
line, rbuf = rbuf.split('\r\n', 1)
|
|
if not line:
|
|
continue
|
|
who, args = parse(line)
|
|
dispatch(args)
|