diff options
Diffstat (limited to 'proxy.py')
| -rw-r--r-- | proxy.py | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/proxy.py b/proxy.py new file mode 100644 index 0000000..b583167 --- /dev/null +++ b/proxy.py @@ -0,0 +1,192 @@ +#!/usr/bin/python -t +import asynchat, asyncore, socket +import pwd, os, sys, logging, logging.handlers + + + +class HTTPChannel(asynchat.async_chat): + def __init__(self, server, sock, addr): + asynchat.async_chat.__init__(self, sock) + self.server = server + self.set_terminator("\r\n\r\n") + self.request = None + self.data = StringIO.StringIO() + self.shutdown = 0 + + def collect_incoming_data(self, data): + self.data.write(data) + if self.data.tell() > 16384: + self.shutdown = 1 + + def found_terminator(self): + if not self.request: + # parse http header + self.data.seek(0) + self.request = string.split(self.data.readline(), None, 2) + if len(self.request) != 3: + # badly formed request; just shut down + self.shutdown = 1 + else: + self.server.handle_request(self, self.request[0], self.request[1]) + self.close_when_done() + else: + pass # ignore body data, for now + +class HTTPProxyServer(asyncore.dispatcher): + def __init__(self): + asyncore.dispatcher.__init__(self) + self.port = 8080 + self.create_socket(socket.AF_INET, socket.SOCK_STREAM) + self.set_reuse_addr() + self.bind(("", 8080)) + self.listen(5) + + def handle_accept(self): + conn, addr = self.accept() + HTTPChannel(self, conn, addr) + + def handle_request(self, channel, method, path): + rss = self.es.to_rss() + channel.push("HTTP/1.0 200 OK\r\nContent-type: text/xml\r\n\r\n") + channel.push(rss) + + +if __name__ == "__main__": + proxy = HTTPProxyServer() + asyncore.loop() + + + + +sys.exit(0) + +class SMTPChannel(asynchat.async_chat): + def __init__(self, server, sock, addr): + asynchat.async_chat.__init__(self, sock) + self.server = server + self.set_terminator("\n") + self.data = "" + self.read_data = False + self.push("220 Mailserver terrorist.at. Nutzung ohne vorige Erlaubnis verboten!\n") + + def collect_incoming_data(self, data): + if len(self.data) > 0: + self.data += "\n" + self.data += data + self.server.logger.debug("Read: " + data.strip()) + if self.data.__len__() > 16384: + self.server.logger.error("too much data, shutdown") + self.close_when_done() + + def push(self,data): + self.server.logger.debug("Write: " + data) + asynchat.async_chat.push(self,data) + + def handle_close(self): + asynchat.async_chat.handle_close(self) + if self.read_data and len(self.data) > 10: + #QUIT oder "." wurde nicht erkannt + mail = email.message_from_string(self.data) + key=self.server.maildir.add(mail) + self.server.logger.info("New Mail: %s" % key) + self.server.maildir.flush() + + + def found_terminator(self): + try: + if self.read_data: + if self.data.endswith("\n.") or self.data.endswith("\n.\r"): + self.read_data = False + self.push("250 Ok: queued as 12345\n") + mail = email.message_from_string(self.data) + key=self.server.maildir.add(mail) + self.server.logger.info("New Mail: %s" % key) + self.server.maildir.flush() + self.data = "" + else: + pass + + elif self.data.startswith("EHLO") \ + or self.data.startswith("HELO"): + self.push("250 Mailserver terrorist.at. Nutzung ohne vorige Erlaubnis verboten!\n") + self.data = "" + elif self.data.startswith("DATA"): + self.push("354 End data with <CR><LF>.<CR><LF>\n") + self.read_data = True + self.data="" + elif self.data.startswith("QUIT"): + self.data = "" + self.push("221 Bye\n") + self.close_when_done() + else: + self.push("250 OK\n") + self.data = "" + except Exception,e: + self.server.logger.error(str(e)) + + +class SMTPServer(asyncore.dispatcher): + def __init__(self): + asyncore.dispatcher.__init__(self) + self.port = 25 + self.create_socket(socket.AF_INET, socket.SOCK_STREAM) + self.set_reuse_addr() + self.bind(("", self.port)) + self.listen(5) + self.maildir = mailbox.Maildir("Maildir", create=True) + self.logger = logging.getLogger('MyLogger') + self.logger.setLevel(logging.DEBUG) + handler = logging.handlers.TimedRotatingFileHandler("log/smtp.log", "D", 1) + self.logger.addHandler(handler) + + def handle_accept(self): + conn, addr = self.accept() + self.logger.info("New Client %s:%s" % conn.getpeername()) + try: + SMTPChannel(self, conn, addr) + except Exception,e: + self.logger.error(str(e)) + + + +if __name__ == '__main__': + s = SMTPServer() + + #user change + pwinfo = pwd.getpwnam('nobody') + os.setregid(pwinfo[3],pwinfo[3]) + os.setreuid(pwinfo[2],pwinfo[2]) + + #"Robustly turn into a UNIX daemon, running in our_home_dir." + # First fork + try: + if os.fork() > 0: + sys.exit(0) # kill off parent + except OSError, e: + sys.stderr.write("fork #1 failed: (%d) %s\n" % (e.errno, e.strerror)) + sys.exit(1) + os.setsid() + os.chdir('/') + os.umask(0) + + # Second fork + try: + pid = os.fork() + if pid > 0: + os._exit(0) + except OSError, e: + sys.stderr.write("fork #2 failed: (%d) %s\n" % (e.errno, e.strerror)) + os._exit(1) + + si = open('/dev/null', 'r') + so = open('/dev/null', 'a+', 0) + se = open('/dev/null', 'a+', 0) + os.dup2(si.fileno(), sys.stdin.fileno()) + os.dup2(so.fileno(), sys.stdout.fileno()) + os.dup2(se.fileno(), sys.stderr.fileno()) + # Set custom file descriptors so that they get proper buffering. + sys.stdout, sys.stderr = so, se + #""" + + s.logger.info("Startup; pid=%s" % os.getpid()) + asyncore.loop() |