python socket 实现 TCP客户端, 以及TCP服务端

开发人员可以使用大量的第三方 python 工具创建网络客户端和服务器, 这些第三方工具的核心模块是 socket 模块

这个模块展示了快速创建 TCP 服务端及客户端, 使用原始套接字等所必须的代码.

import socket

import threading

def tcp_client(target_host=None,target_port=None,send_data=None):

""" TCP 客户端 """

target_host = target_host if target_host else "www.baidu.com"

target_port = target_port if target_port else 80

send_data = send_data if send_data else b"GET / HTTP/1.1\r\nHost: baidu.com\r\n\r\n"

# 建立一个 socket 对象

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接客户端

client.connect((target_host, target_port))

# 发送数据

client.send(send_data)

# 接收数据

response = b''

while 1:

res = client.recv(1024)

response += res

if len(res)<1024:

break

print(response)

def tcp_server(bind_ip=None,bind_port=None,response_data=None):

""" TCP 服务器 """

bind_ip = bind_ip if bind_ip else "192.168.1.80"

bind_port = bind_port if bind_port else 8000

response_data = response_data if response_data else b"Yes!"

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

server.bind((bind_ip, bind_port))

server.listen(5)

print("[*] Listening on %s:%d"%(bind_ip, bind_port))

# 客户端处理线程

def handle_client(client_socket):

# 打印客户端发送的内容

request = client_socket.recv(1024)

print("[*] Received: %s"%request)

# 返还一个数据包

client_socket.send(response_data)

client_socket.close()

while 1:

client, addr = server.accept()

print("[*] Accepted connection from: %s:%d"%(addr[0],addr[1]))

# 挂起客户端线程, 处理传入的数据

client_handler = threading.Thread(target=handle_client,args=(client,))

client_handler.start()

if __name__ == '__main__':

host = "192.168.1.80"

port = 8000

# 开启服务端

server = threading.Thread(target=tcp_server,args=(host,port,b"Server: Hello client"))

server.start()

# 开启客户端

client = threading.Thread(target=tcp_client,args=(host,port,b'Client: My name is client'))

client.start()