37,721
社区成员
发帖
与我相关
我的任务
分享
#!/usr/bin/env python
import sys, socket, time, threading
loglock = threading.Lock()
def log(msg):
loglock.acquire()
try:
print '[%s]: \n%s\n' % (time.ctime(), msg.strip())
sys.stdout.flush()
finally:
loglock.release()
class PipeThread(threading.Thread):
def __init__(self, source, target):
threading.Thread.__init__(self)
self.source = source
self.target = target
def run(self):
while True:
try:
data = self.source.recv(1024)
log(data)
if not data: break
self.target.send(data)
except:
break
log('PipeThread done')
class Forwarding(threading.Thread):
def __init__(self, port, targethost, targetport,targethost1, targetport1):
threading.Thread.__init__(self)
self.targethost = targethost
self.targetport = targetport
self.targethost1 = targethost1
self.targetport1= targetport1
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.bind(('0.0.0.0', port))
self.sock.listen(1000)
def run(self):
while True:
client_fd, client_addr = self.sock.accept()
target_fd = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
target_fd.connect((self.targethost, self.targetport))
target_fd1 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
target_fd1.connect((self.targethost1, self.targetport1))
log('new connect')
# two direct pipe
PipeThread(target_fd, client_fd).start()
PipeThread(target_fd1, client_fd).start()
PipeThread(client_fd, target_fd).start()
if __name__ == '__main__':
print 'Starting'
import sys
try:
port = 90
targethost1 = '10.167.0.3'
targetport1 = 12000
targethost = '119.185.1.182'
targetport = 6977
print port, targethost, targetport,targethost1,targetport1
except (ValueError, IndexError):
print 'Usage: %s port targethost [targetport]' % sys.argv[0]
sys.exit(1)
#sys.stdout = open('forwaring.log', 'w')
Forwarding(port, targethost, targetport,targethost1,targetport1).start()
log('new connect')
# two direct pipe
PipeThread(target_fd, client_fd).start()
PipeThread(target_fd1, client_fd).start()
PipeThread(client_fd, target_fd).start()
改成
log('new connect')
# two direct pipe
PipeThread(target_fd, client_fd).start()
PipeThread(target_fd1, client_fd).start()
PipeThread(client_fd, target_fd).start()
PipeThread(client_fd, target_fd1).start()
这样才对称吧?