diff options
Diffstat (limited to 'module/plugins/hoster/XDCC.py')
| -rw-r--r-- | module/plugins/hoster/XDCC.py | 206 |
1 files changed, 206 insertions, 0 deletions
diff --git a/module/plugins/hoster/XDCC.py b/module/plugins/hoster/XDCC.py new file mode 100644 index 000000000..3985a050c --- /dev/null +++ b/module/plugins/hoster/XDCC.py @@ -0,0 +1,206 @@ +# -*- coding: utf-8 -*- + +import re +import select +import socket +import struct +import sys +import time + +from module.plugins.internal.Hoster import Hoster +from module.plugins.internal.misc import fsjoin + + +class XDCC(Hoster): + __name__ = "XDCC" + __type__ = "hoster" + __version__ = "0.38" + __status__ = "testing" + + __config__ = [("nick", "str", "Nickname", "pyload"), + ("ident", "str", "Ident", "pyloadident"), + ("realname", "str", "Realname", "pyloadreal")] + + __description__ = """Download from IRC XDCC bot""" + __license__ = "GPLv3" + __authors__ = [("jeix", "jeix@hasnomail.com")] + + + def setup(self): + self.timeout = 30 + self.multiDL = False + + + def process(self, pyfile): + #: Change request type + self.req = self.pyload.requestFactory.getRequest(self.classname, type="XDCC") + + self.pyfile = pyfile + for _i in xrange(0, 3): + try: + nmn = self.do_download(pyfile.url) + self.log_debug("Download of %s finished." % nmn) + return + + except socket.error, e: + if hasattr(e, "errno"): + errno = e.errno + else: + errno = e.args[0] + + if errno is 10054: + self.log_debug("Server blocked our ip, retry in 5 min") + self.wait(300) + continue + + self.fail(_("Failed due to socket errors. Code: %d") % errno) + + self.fail(_("Server blocked our ip, retry again later manually")) + + + def do_download(self, url): + self.pyfile.setStatus("waiting") #: Real link + + m = re.match(r'xdcc://(.*?)/#?(.*?)/(.*?)/#?(\d+)/?', url) + server = m.group(1) + chan = m.group(2) + bot = m.group(3) + pack = m.group(4) + nick = self.config.get('nick') + ident = self.config.get('ident') + real = self.config.get('realname') + + temp = server.split(':') + ln = len(temp) + if ln is 2: + host, port = temp + elif ln is 1: + host, port = temp[0], 6667 + else: + self.fail(_("Invalid hostname for IRC Server: %s") % server) + + ####################### + #: CONNECT TO IRC AND IDLE FOR REAL LINK + dl_time = time.time() + + sock = socket.socket() + sock.connect((host, int(port))) + if nick == "pyload": + nick = "pyload-%d" % (time.time() % 1000) #: last 3 digits + sock.send("NICK %s\r\n" % nick) + sock.send("USER %s %s bla :%s\r\n" % (ident, host, real)) + + self.wait(3) + + sock.send("JOIN #%s\r\n" % chan) + sock.send("PRIVMSG %s :xdcc send #%s\r\n" % (bot, pack)) + + #: IRC recv loop + readbuffer = "" + done = False + retry = None + m = None + while True: + + #: Done is set if we got our real link + if done: + break + + if retry: + if time.time() > retry: + retry = None + dl_time = time.time() + sock.send("PRIVMSG %s :xdcc send #%s\r\n" % (bot, pack)) + + else: + if (dl_time + self.timeout) < time.time(): #@TODO: add in config + sock.send("QUIT :byebye\r\n") + sock.close() + self.fail(_("XDCC Bot did not answer")) + + fdset = select.select([sock], [], [], 0) + if sock not in fdset[0]: + continue + + readbuffer += sock.recv(1024) + temp = readbuffer.split("\n") + readbuffer = temp.pop() + + for line in temp: + # if self.pyload.debug: + # self.log_debug("*> " + decode(line)) + line = line.rstrip() + first = line.split() + + if first[0] == "PING": + sock.send("PONG %s\r\n" % first[1]) + + if first[0] == "ERROR": + self.fail(_("IRC-Error: %s") % line) + + msg = line.split(None, 3) + if len(msg) is not 4: + continue + + msg = { + 'origin': msg[0][1:], + 'action': msg[1], + 'target': msg[2], + 'text': msg[3][1:] + } + + if nick is msg['target'][0:len(nick)] and "PRIVMSG" is msg['action']: + if msg['text'] == "\x01VERSION\x01": + self.log_debug("Sending CTCP VERSION") + sock.send("NOTICE %s :%s\r\n" % (msg['origin'], "pyLoad! IRC Interface")) + elif msg['text'] == "\x01TIME\x01": + self.log_debug("Sending CTCP TIME") + sock.send("NOTICE %s :%d\r\n" % (msg['origin'], time.time())) + elif msg['text'] == "\x01LAG\x01": + pass #: don't know how to answer + + if not (bot is msg['origin'][0:len(bot)] + and nick is msg['target'][0:len(nick)] + and msg['action'] in ("PRIVMSG", "NOTICE")): + continue + + if self.pyload.debug: + self.log_debug(msg['origin'], msg['text']) + + if "You already requested that pack" in msg['text']: + retry = time.time() + 300 + + if "you must be on a known channel to request a pack" in msg['text']: + self.fail(_("Invalid channel")) + + m = re.match('\x01DCC SEND (.*?) (\d+) (\d+)(?: (\d+))?\x01', msg['text']) + if m is not None: + done = True + + #: Get connection data + ip = socket.inet_ntoa(struct.pack('L', socket.ntohl(int(m.group(2))))) + port = int(m.group(3)) + packname = m.group(1) + + if len(m.groups()) > 3: + self.req.filesize = int(m.group(4)) + + self.pyfile.name = packname + + dl_folder = self.pyload.config.get("general", "download_folder") + filename = fsjoin(dl_folder, packname) + + self.log_info(_("Downloading %s from %s:%d") % (packname, ip, port)) + + self.pyfile.setStatus("downloading") + newname = self.req.download(ip, port, filename, sock, self.pyfile.setProgress) + if newname and newname is not filename: + self.log_info(_("%(name)s saved as %(newname)s") % {'name': self.pyfile.name, 'newname': newname}) + filename = newname + + #: kill IRC socket + #: sock.send("QUIT :byebye\r\n") + sock.close() + + self.last_download = filename + return self.last_download |