Python socket模拟TCP通讯

本文将介绍使用Python的socket模块创建TCP客户端和服务端的方法。详细的模块介绍参见Python官方文档。

对于TCP server端的创建而言，分为如下几个步骤：

• 创建socket对象（socket）：其中两个参数分别为Address Family（如AF_INET为IPV4，AF_INET6为IPV6，AF_UNIX为UNIX域协议族）、socket类型（如SOCK_STREAM为TCP，SOCK_DGRAM为UDP）。
• 绑定服务器地址（bind）：参数为服务器地址二元组。
• 监听（listen）：参数为允许的连接数。
• 等待请求（accept）。
• 接收数据（recv、recvfrom、recvfrom_into、recv_into）、发送数据（send、sendall、sendto）。
• 关闭连接（close）。

示例代码如下：

Python socket: TCP server

! /usr/bin/python

-- coding: utf-8 --

import socket

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

server_address = ('127.0.0.1', 12345)

print "Starting up on %s:%s" % server_address

sock.bind(server_address)

sock.listen(1)

while True:

print  "Waiting for a connection"

connection,  client_address  =  sock.accept()

try:

    print  "Connection from",  client_address

    data  =  connection.recv(1024)

    print  "Receive '%s'"  %  data

finally:

    connection.close()

|

其中，服务器地址二元组中，第一个元素为服务器IP（留空为在任意IP监听），第二个元素为服务器端口号。

而对于TCP client而言，通常包括如下几个步骤：

• 创建socket对象（socket）：同server端。
• 连接服务器（connect）：参数为服务器地址二元组。
• 发送和接收数据：同server端。
• 关闭连接：同server端。

示例代码如下：

Python socket: TCP client

# /usr/bin/python

# -*- coding: utf-8 -*-

import  socket

def  check_tcp_status(ip,  port):

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

    server_address  =  (ip,  port)

    print  'Connecting to %s:%s.'  %  server_address

    sock.connect(server_address)

    message  =  "I'm TCP client"

    print  'Sending "%s".'  %  message

    sock.sendall(message)

    print  'Closing socket.'

    sock.close()

if  __name__  ==  "__main__":

    print  check_tcp_status("127.0.0.1",  12345)