python初步实现socket服务端数据流转（1对1网络聊天室）

关键词：管道、多进程、socket、内网穿透

程序运行流程示意图：

 

 附code如下；

import socket as sk
import multiprocessing as mp
import time as t


def c_s_s(sock, addr,conn1,conn2): #client to sever_sockfile1 to sever_sockfile2 
    while True:
        conn2.close()
        readdata = sock.makefile().readline() #进程1读取客户端1发来的信息
        if readdata != "":
            print(readdata)
            conn1.send(readdata) #发送至管道内
        t.sleep(0.2) 
            
def s_c(sock, addr,conn1,conn2): #sever_file2 to client
    while True:
        conn1.close()
        recvdata = conn2.recv() #进程2读取管道信息
        if recvdata != "":
            print(recvdata)
            sock.send(recvdata.encode()) #进程2发送信息至客户端2
            #sock.send(b'1212\x0d')
        t.sleep(0.2) 
def main():
    while True:
        conn1,conn2 = mp.Pipe() #创建管道
        s = sk.socket(sk.AF_INET ,sk.SOCK_STREAM)
        s.bind(('10.238.2.208',2333))
        s.listen(5)
        sock,addr = s.accept()
        sock.send(b"hello-usr1")
        t1 = mp.Process(target=c_s_s, args=(sock, addr,conn1,conn2))
        t1.start()
        t11 = mp.Process(target=s_c, args=(sock, addr,conn2,conn1))
        t11.start()
        #t1.join()
        sock,addr = s.accept()
        sock.send(b"hello-usr2")
        t22 = mp.Process(target=c_s_s, args=(sock, addr,conn2,conn1))
        t22.start()
        t2 = mp.Process(target=s_c, args=(sock, addr,conn1,conn2))
        t2.start()
        #t2.join()

if __name__ == '__main__':
    main()
        

函数C_S_S()功能示意图如下：

 

 函数S_C()功能示意图如下；

 

准备两个客户端

 

 

 运行服务程序后连接

客户端1：

 

 客户端2：

 

 互相通信：

 

 将服务端放入公网服务器，可实现类似内网穿透的功能~

 -----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

如下为优化版本1.1

import socket as sk
import multiprocessing as mp
import time as t


def c_s_s(sock, addr,conn1,conn2): #client to sever_sockfile1 to sever_sockfile2 
    while True:
        conn2.close()
        #readdata = sock.makefile().readline() #进程1读取客户端1发来的信息
        readdata = sock.recv(1024)
        print(readdata)
        if readdata != "":
            print(readdata)
            conn1.send(readdata) #发送至管道内
        #t.sleep(0.2) 
            
def s_c(sock, addr,conn1,conn2): #sever_file2 to client
    while True:
        conn1.close()
        recvdata = conn2.recv() #进程2读取管道信息
        if recvdata != "":
            print(recvdata)
            sock.send(recvdata) #进程2发送信息至客户端2
            #sock.send(b'1212\x0d')
        #t.sleep(0.2) 
def main():
    while True:
        conn1,conn2 = mp.Pipe() #创建管道
        s = sk.socket(sk.AF_INET ,sk.SOCK_STREAM)
        s.bind(('10.238.2.208',2333))
        s.listen(5)
        sock,addr = s.accept()
        sock.send(b"hello-usr1")
        t1 = mp.Process(target=c_s_s, args=(sock, addr,conn1,conn2))
        t1.start()
        t11 = mp.Process(target=s_c, args=(sock, addr,conn2,conn1))
        t11.start()
        #t1.join()
        sock,addr = s.accept()
        sock.send(b"hello-usr2")
        t22 = mp.Process(target=c_s_s, args=(sock, addr,conn2,conn1))
        t22.start()
        t2 = mp.Process(target=s_c, args=(sock, addr,conn1,conn2))
        t2.start()
        #t2.join()

if __name__ == '__main__':
    main()
        

 优化版本1.2

进程断开自动杀进程；

import socket as sk
import multiprocessing as mp
import time as t
import os

def c_s_s(sock, addr,conn1,conn2): #client to sever_sockfile1 to sever_sockfile2 
    while True:
        conn2.close()
        readdata = sock.recv(1024) #进程1读取客户端1发来的信息
        if readdata:            #判断TCP是否断开和是否接收信息-----------
            print(readdata)
            conn1.send(readdata) #发送至管道内
        else:                   #断开的处理部分-----------
            conn1.send(b"212") #告知另一端已断开
            sock.close()
            os._exit(0)         #强制退出进程
            break
        t.sleep(0.2) 
            
def s_c(sock, addr,conn1,conn2): #sever_file2 to client
    while True:
        conn1.close()
        recvdata = conn2.recv() #进程2读取管道信息
        if recvdata == b"212":  #另一端断开的处理部分--------------
            print(recvdata)
            sock.send(b"break") #告知另一端已断开
            sock.close()
            os._exit(0)         #强制退出进程
            break
        else:
            sock.send(recvdata) #另一端未断开 进行信息转发
            #sock.send(b'1212\x0d')
        t.sleep(0.2) 
def main():
    s = sk.socket(sk.AF_INET ,sk.SOCK_STREAM)
    s.bind(('10.238.2.208',2333))
    s.listen()
    while True:
        conn1,conn2 = mp.Pipe() #创建管道
        sock,addr = s.accept()
        sock.send(b"hello-usr1")
        t1 = mp.Process(target=c_s_s, args=(sock, addr,conn1,conn2))
        t1.start()
        t11 = mp.Process(target=s_c, args=(sock, addr,conn2,conn1))
        t11.start()
        #t1.join()
        sock,addr = s.accept()
        sock.send(b"hello-usr2")
        t22 = mp.Process(target=c_s_s, args=(sock, addr,conn2,conn1))
        t22.start()
        t2 = mp.Process(target=s_c, args=(sock, addr,conn1,conn2))
        t2.start()
        #t2.join()

if __name__ == '__main__':
    main()