Created
April 5, 2020 05:42
-
-
Save compwron/bd963a1632ae3a67f9f568b7748d4a58 to your computer and use it in GitHub Desktop.
python2 port forward proxy
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/python | |
# Thanks https://gist.github.com/gschizas/3731989 | |
# This is a simple port-forward / proxy, written using only the default python | |
# library. If you want to make a suggestion or fix something you can contact-me | |
# at voorloop_at_gmail.com | |
# Distributed over IDC(I Don't Care) license | |
import socket | |
import select | |
import time | |
import sys | |
# Changing the buffer_size and delay, you can improve the speed and bandwidth. | |
# But when buffer get to high or delay go too down, you can broke things | |
buffer_size = 4096 | |
delay = 0 | |
forward_to = ('localhost', 64960) | |
class Forward: | |
def __init__(self): | |
self.forward = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
def start(self, host, port): | |
try: | |
self.forward.connect((host, port)) | |
return self.forward | |
except Exception, e: | |
print e | |
return False | |
class TheServer: | |
input_list = [] | |
channel = {} | |
def __init__(self, host, port): | |
self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) | |
self.server.bind((host, port)) | |
self.server.listen(200) | |
def main_loop(self): | |
self.input_list.append(self.server) | |
while 1: | |
ss = select.select | |
inputready, outputready, exceptready = ss(self.input_list, [], []) | |
for self.s in inputready: | |
if self.s == self.server: | |
self.on_accept() | |
break | |
self.data = self.s.recv(buffer_size) | |
if len(self.data) == 0: | |
self.on_close() | |
break | |
else: | |
self.on_recv() | |
def on_accept(self): | |
forward = Forward().start(forward_to[0], forward_to[1]) | |
clientsock, clientaddr = self.server.accept() | |
if forward: | |
print clientaddr, "has connected" | |
self.input_list.append(clientsock) | |
self.input_list.append(forward) | |
self.channel[clientsock] = forward | |
self.channel[forward] = clientsock | |
else: | |
print "Can't establish connection with remote server.", | |
print "Closing connection with client side", clientaddr | |
clientsock.close() | |
def on_close(self): | |
print self.s.getpeername(), "has disconnected" | |
#remove objects from input_list | |
self.input_list.remove(self.s) | |
self.input_list.remove(self.channel[self.s]) | |
out = self.channel[self.s] | |
# close the connection with client | |
self.channel[out].close() # equivalent to do self.s.close() | |
# close the connection with remote server | |
self.channel[self.s].close() | |
# delete both objects from channel dict | |
del self.channel[out] | |
del self.channel[self.s] | |
def on_recv(self): | |
data = self.data | |
# here we can parse and/or modify the data before send forward | |
self.channel[self.s].send(data) | |
if __name__ == '__main__': | |
server = TheServer('', 12345) | |
try: | |
server.main_loop() | |
except KeyboardInterrupt: | |
print "Ctrl C - Stopping server" | |
sys.exit(1) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment