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）。

示例代码如下：

#! /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()

#! /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端。

示例代码如下：

# /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)