实现端口转发功能，经测试，ssh、http可以正常使用

.gitignore

@@ -0,0 +1 @@
+__pycache__/

ReadMe.txt

@@ -0,0 +1,7 @@
+
+
+
+
+2024.5.22
+  实现端口转发功能，经测试，ssh、http可以正常使用
+

mytcp.service

@@ -0,0 +1,14 @@
+[Unit]
+Description=mytcp service
+After=network.target
+
+[Service]
+Type=simple
+User=root
+Group=root
+WorkingDirectory=/root/work/mytcp
+ExecStart=/usr/bin/python3 server.py
+Restart=always
+
+[Install]
+WantedBy=multi-user.target

prot_codec.py

@@ -0,0 +1,114 @@
+
+import os
+import sys
+import json
+
+
+
+
+# 数据包编解码
+
+
+
+
+
+
+
+# 计算整个数据流的CRC
+def _crc16(data:bytearray):
+  lenth=len(data)
+  if(lenth>0):
+    crc=0xffff
+    for i in range(lenth):
+      crc=(crc^data[i])&0xffff
+      for j in range(8):
+        if(crc&1)!=0:
+          crc=((crc>>1)^0xa001)&0xffff
+        else:
+          crc=(crc>>1)&0xffff
+    return bytearray([crc&0xff,(crc>>8)&0xff])
+  return bytearray([0,0])
+
+
+# 0xff帧起始，0xfe帧结尾
+# 0xf0转义标识
+# 0xf1命令数据分割
+# 0xf2数据校验分割
+
+# 字节数据大于等于0xf0 之后要转义
+def _tran(data:bytearray):
+  ret=bytearray()
+  for item in data:
+    if(item>=0xf0):
+      ret.append(0xf0)
+      ret.append(item-0xf0)
+    else:
+      ret.append(item)
+  return ret
+
+# 去转义
+def _untran(data:bytearray):
+  ret=bytearray()
+  i=0
+  length=len(data)
+  while i<length:
+    if(data[i]>=0xf0):
+      ret.append(data[i]+data[i+1])
+      i+=1
+    else:
+      ret.append(data[i])
+    i+=1
+  return ret
+
+
+# 以命令和数据编码
+def encode(cmd:bytearray,data:bytearray):
+  crc=_crc16(cmd+data)
+  ret=bytearray()
+  ret.append(0xff)
+  ret+=_tran(cmd)
+  ret.append(0xf1)
+  ret+=_tran(data)
+  ret.append(0xf2)
+  ret+=_tran(crc)
+  ret.append(0xfe)
+  return ret
+
+
+# 解码为命令和数据
+def decode(data:bytearray):
+  # print('prot_decode:',data.hex(' '))
+  while len(data)>0:
+    if(data[0]!=0xff):
+      data=data[1:]
+    else:
+      break
+  if(len(data)==0):
+    return (None,None)
+  if(data[0]==0xff and data[-1]==0xfe):
+    data=data[1:-1]
+    cmd_index=data.find(0xf1)
+    data_index=data.find(0xf2)
+    cmd=_untran(data[:cmd_index])
+    da=_untran(data[cmd_index+1:data_index])
+    crc=_untran(data[data_index+1:])
+    if(crc==_crc16(cmd+da)):
+      return (cmd,da)
+  return (None,None)
+
+
+
+
+
+
+
+
+if __name__ == "__main__":
+  data=encode(json.dumps({'device':'client','option':'data'}).encode('utf-8'),bytearray([0xff,0xfd,0xff,0xde,0xde]))
+  print(data.hex(' '))
+  print(decode(data))
+  print(b'\xff'.hex(' '))
+
+
+
+

server.py

@@ -0,0 +1,63 @@
+import socket
+import threading
+import os
+import sys
+import json
+
+
+import prot_codec as pc
+import target as tg
+
+
+
+# 服务器端脚本，用于转发端口数据
+
+SERVER_PORT = 5345
+
+
+
+# 定义个函数,使其专门重复处理客户的请求数据（也就是重复接受一个用户的消息并且重复回答，直到用户选择下线）
+def dispose_client_request(tcp_client_1,tcp_addr):
+    print(f"客户端：{tcp_addr} 已连接")
+    recv_data=bytearray()
+    target =tg.tcp_target(tcp_client_1)
+    while True:
+        try:
+            recv = tcp_client_1.recv(4096)
+        except Exception as e:
+            print(str(e))
+            break
+        if recv:
+            recv_data+=recv
+            while True:
+                start=recv_data.find(b'\xff')
+                end=recv_data.find(b'\xfe')
+                if(start==-1 or end ==-1):
+                    break
+                target.recv_handler(recv_data[start:end+1])
+                recv_data=recv_data[end+1:]
+        else:
+            break
+    print(f"客户端：{tcp_addr} 已下线")
+    tcp_client_1.close()
+
+
+
+def start_service():
+    tcp_server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+    tcp_server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)
+    tcp_server.bind(("",SERVER_PORT))
+
+    print(f"开始监听 ({SERVER_PORT})")
+    tcp_server.listen(128)
+
+    while True:
+        tcp_client_1 , tcp_client_address = tcp_server.accept()
+        thd = threading.Thread(target = dispose_client_request, args = (tcp_client_1,tcp_client_address))
+        # 设置守护主线程  即如果主线程结束了 那子线程中也都销毁了  防止主线程无法退出
+        # thd.setDaemon(True)
+        thd.start()
+
+
+if __name__ == "__main__":
+    start_service()

target.py

@@ -0,0 +1,57 @@
+
+
+import os
+import sys
+import json
+import socket
+
+
+import prot_codec as pc
+
+
+
+# 定义服务器和客户端
+# device:
+# server
+# client
+# 定义操作类型
+# option:
+# connect
+# disconnect
+# data
+# login
+
+
+_tcp_server=None
+_tcp_client=None
+
+
+
+class tcp_target(object):
+  def __init__(self,tcp_handler:socket):
+    self.handler=tcp_handler
+    self.state="idle"
+  def recv_handler(self,recv_data:bytearray):
+    global _tcp_client
+    global _tcp_server
+    cmd,data=pc.decode(recv_data)
+    print(cmd.decode('utf-8'))
+    j=json.loads(cmd)
+    if(j["device"]=="client"):
+      if(j["option"]=="login"):
+        if(_tcp_client is not None):
+          _tcp_client.close()
+        _tcp_client=self.handler
+      else:
+        if(_tcp_server is not None):
+          _tcp_server.send(recv_data)
+    if(j["device"]=="server"):
+      if(j["option"]=="login"):
+        if(_tcp_server is not None):
+          _tcp_server.close()
+        _tcp_server=self.handler
+      else:
+        if(_tcp_client is not None):
+          _tcp_client.send(recv_data)
+
+

target_client.py

@@ -0,0 +1,144 @@
+
+
+
+import os
+import sys
+import json
+import socket
+import threading
+import time
+
+
+import prot_codec as pc
+
+
+
+# tcp 客户端脚本
+# 一个本地服务端
+# 一个远程客户端
+
+
+
+_LOCAL_PORT = 31234
+
+_remote_client=None
+_local_client=[]
+
+
+
+# 发送数据到指定ip地址和端口
+def send_to(ip,port,data:bytearray):
+  for item in _local_client:
+    if(item[1]==ip and item[2]==port):
+      print(f"recv from remote {ip},{port}")
+      item[0].send(data)
+      return
+  print(f"can not fond {ip},{port}")
+
+
+# 关闭指定地址的端口
+def close(ip,port):
+  for item in _local_client:
+    if(item[1]==ip and item[2]==port):
+      item[0].close()
+      break
+
+
+
+
+
+# 远端数据处理，解包，把负载数据发送到本地服务器
+def remote_client_handler(tcp_remote:socket):
+  global _remote_client
+  global _local_client
+  print("已连接代理服务器")
+  recv_data=bytearray()
+  cmd={'device':'client','option':'login'}
+  data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
+  _remote_client.send(data)
+  while True:
+    try:
+      recv = tcp_remote.recv(4096)
+    except Exception as e:
+      print(str(e))
+      break
+    if recv:
+      recv_data+=recv
+      while True:
+        start=recv_data.find(b'\xff')
+        end=recv_data.find(b'\xfe')
+        if(start==-1 or end==-1):
+          break
+        cmd,data=pc.decode(recv_data[start:end+1])
+        print(cmd.decode('utf-8'))
+        try:
+          j=json.loads(cmd)
+          if(j['device']=='server'):
+            if(j['option']=='data'):
+              send_to(j['ip'],j['port'],data)
+            elif(j['option']=='disconnect'):
+              close(j['ip'],j['port'])
+        except Exception as e:
+          print(str(e))
+        recv_data=recv_data[end+1:]
+    else:
+      break
+  tcp_remote.close()
+
+
+# 本地数据处理，解包，把负载数据发送到本地服务器
+def local_client_handler(tcp_server:socket,addr):
+  global _remote_client
+  global _local_client
+  print("addr:",addr)
+  addr_info=(tcp_server,addr[0],addr[1])
+  _local_client.append(addr_info)
+  if(_remote_client is not None):
+    cmd={'device':'client','option':'connect','ip':addr[0],'port':addr[1]}
+    data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
+    _remote_client.send(data)
+  while True:
+    try:
+      recv = tcp_server.recv(4096)
+    except Exception as e:
+      print(str(e))
+      break
+    if recv:
+      cmd={'device':'client','option':'data','ip':addr[0],'port':addr[1]}
+      data=pc.encode(json.dumps(cmd).encode('utf-8'),recv)
+      if _remote_client is not None:
+        _remote_client.send(data)
+        print(f"send to remote {addr[0]},{addr[1]}")
+    else:
+      break
+  tcp_server.close()
+  if(_remote_client is not None):
+    cmd={'device':'client','option':'disconnect','ip':addr[0],'port':addr[1]}
+    data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
+    _remote_client.send(data)
+  _local_client.remove(addr_info)
+
+
+
+def main():
+  global _remote_client
+  global _local_client
+
+  _remote_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+  _remote_client.connect(("1.92.113.30",5345))
+  thd = threading.Thread(target = remote_client_handler, args = (_remote_client,))
+  thd.start()
+  tcp_server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+  tcp_server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)
+  tcp_server.bind(("",_LOCAL_PORT))
+  print(f"开始监听({_LOCAL_PORT})")
+  tcp_server.listen(128)
+  while True:
+      temp , temp_address = tcp_server.accept()
+      thd = threading.Thread(target = local_client_handler, args = (temp,temp_address))
+      thd.start()
+
+
+if __name__ == "__main__":
+  main()
+

target_server.py

@@ -0,0 +1,145 @@
+
+
+import os
+import sys
+import json
+import socket
+import threading
+
+
+import prot_codec as pc
+
+
+
+
+
+
+
+# tcp服务端脚本
+# 一个连接远端转发服务器
+# 另一个连接本地tcp服务器
+
+
+_remote_client=None
+_local_client=[]
+
+
+LOCAL_SERVER_IP = ("192.168.3.166",80)
+# LOCAL_SERVER_IP = ("192.168.3.167",22)
+
+
+
+# 发送数据到指定ip地址和端口
+def send_to(ip,port,data:bytearray):
+  for item in _local_client:
+    if(item[1]==ip and item[2]==port):
+      print(f"recv from remote {ip},{port}")
+      item[0].send(data)
+      break
+
+# 关闭指定地址的端口
+def close(ip,port):
+  for item in _local_client:
+    if(item[1]==ip and item[2]==port):
+      item[0].close()
+      break
+
+
+
+
+# 本地数据处理，解包，把负载数据发送到本地服务器
+def local_client_handler(tcp_server:socket,ip,port):
+  global _remote_client
+  global _local_client
+  self_info=(tcp_server,ip,port)
+  _local_client.append(self_info)
+  while True:
+    try:
+      recv = tcp_server.recv(4096)
+    except Exception as e:
+      print(str(e))
+      break
+    if recv:
+      cmd={'device':'server','option':'data','ip':ip,'port':port}
+      data=pc.encode(json.dumps(cmd).encode('utf-8'),recv)
+      if _remote_client is not None:
+        _remote_client.send(data)
+        print(f"send to remote {ip},{port}")
+
+    else:
+      break
+  tcp_server.close()
+  cmd={'device':'server','option':'disconnect','ip':ip,'port':port}
+  data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
+  if _remote_client is not None:
+    _remote_client.send(data)
+  _local_client.remove(self_info)
+
+
+
+
+# 远端数据处理，解包，把负载数据发送到本地服务器
+def remote_client_handler(tcp_client_1:socket):
+  global _remote_client
+  global _local_client
+  print("已连接代理服务器")
+  cmd={'device':'server','option':'login'}
+  data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
+  _remote_client.send(data)
+  recv_data=bytearray()
+  while True:
+    try:
+      recv = tcp_client_1.recv(4096)
+    except Exception as e:
+      print(str(e))
+      break
+    if recv:
+      recv_data+=recv
+      while True:
+        start=recv_data.find(b'\xff')
+        end=recv_data.find(b'\xfe')
+        if(start == -1 or end == -1):
+          break
+        cmd,data=pc.decode(recv_data[start:end+1])
+        print(cmd.decode('utf-8'))
+        try:
+          j=json.loads(cmd)
+          if(j['device']=='client'):
+            if(j['option']=='connect'):
+              temp = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+              temp.connect(LOCAL_SERVER_IP)
+              thd = threading.Thread(target = local_client_handler, args = (temp,j['ip'],j['port']))
+              thd.start()
+            elif(j['option']=='disconnect'):
+              close(j['ip'],j['port'])
+            elif(j['option']=='data'):
+              send_to(j['ip'],j['port'],data)
+        except Exception as e:
+          print(str(e))
+        recv_data=recv_data[end+1:]
+    else:
+      break
+  tcp_client_1.close()
+  print("socket close.")
+
+
+
+def main():
+  global _remote_client
+  global _local_client
+
+  _remote_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+  _remote_client.connect(("1.92.113.30",5345))
+  # thd = threading.Thread(target = remote_client_handler, args = (_remote_client,))
+  # thd.start()
+  # thd.join()
+
+  remote_client_handler(_remote_client)
+
+
+
+
+
+if __name__ == "__main__":
+  main()
+

test.py

@@ -0,0 +1,49 @@
+import os
+import sys
+
+# import PyQt5.QtCore as core
+# import PyQt5.QtGui as gui
+# import PyQt5.QtWidgets as wid
+
+
+
+
+# app=wid.QApplication(sys.argv)
+
+# qd=wid.QDialog()
+
+# qd.show()
+
+# app.exec()
+
+import socket
+
+if __name__ == '__main__':
+
+    # 1 创建客户端套接字对象tcp_client_1
+    # 参数介绍：AF_INET 代表IPV4类型, SOCK_STREAM代表tcp传输协议类型 ,注：AF_INET6代表IPV6
+
+    tcp_client_1 = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+
+    # 2 通过客户端套接字的connect方法与服务器套接字建立连接
+    # 参数介绍：前面的ip地址代表服务器的ip地址，后面的61234代表服务端的端口号 。
+
+    # tcp_client_1.connect(("10.0.15.79",61234))
+    tcp_client_1.connect(("1.92.113.30",5345))
+
+    # 将编号好的数据存到变量send_data中，注：encode(encoding='utf-8)是将数据转换成utf-8的格式发送给服务器
+    send_data = "你好，服务器，我是客户端1号".encode(encoding='utf-8')
+
+    # 3 通过客户端套接字的send方法将数据发送给服务器
+    tcp_client_1.send(send_data)
+
+    # 4 通过客户端套接字的recv方法来接受服务器返回的数据存到变量recv_data中，1024是可接收的最大字节数。
+    recv_data = tcp_client_1.recv(1024)
+
+    # 将接收到的服务器数据recv_data通过decode方法解码为utf-8
+    print(recv_data.decode(encoding = 'utf-8'))
+
+    # 5 最后关闭客户端套接字连接
+    tcp_client_1.close()
+
+