[Python]TCP编程的简单示例

套接字模块是一个非常简单的基于对象的接口，它提供对低层BSD套接字样式网络的访问。使用该模块可以实现客户机和服务器套接字。要在Python 中建立具有TCP和流套接字的简单服务器，需要使用Socket模块。利用该模块包含的函数和类定义，可生成通过网络通信的程序。

建立一个TCP服务器连接需要6个步骤：

1. 创建socket对象。调用socket构造函数：

socket=socket.socket(familly,type)

family的值可以是AF_UNIX(Unix域，用于同一台机器上的进程间通讯)，也可以是AF_INET（对于IPV4协议的TCP和 UDP），至于type参数，SOCK_STREAM（流套接字）或者 SOCK_DGRAM（数据报文套接字）,SOCK_RAW（raw套接字）。

2. 将socket绑定（指派）到指定地址上，socket.bind(address)

address必须是一个双元素元组,((host,port)),主机名或者ip地址+端口号。如果端口号正在被使用或者保留，或者主机名或ip地址错误，则引发socke.error异常。

3. 绑定后，必须准备好套接字，以便接受连接请求。

socket.listen(backlog)

backlog指定了最多连接数，至少为1，接到连接请求后，这些请求必须排队，如果队列已满，则拒绝请求。

4. 服务器套接字通过socket的accept方法等待客户请求一个连接：

connection,address=socket.accept()

调用accept方法时，socket会进入'waiting'（或阻塞）状态。客户请求连接时，方法建立连接并返回服务器。accept方法返回一个含有俩个元素的元组，形如(connection,address)。第一个元素（connection）是新的socket对象，服务器通过它与客户通信；第二个元素（address）是客户的internet地址。

5. 处理阶段，服务器和客户通过send和recv方法通信（传输数据）。

服务器调用send，并采用字符串形式向客户发送信息。send方法返回已发送的字符个数。服务器使用recv方法从客户接受信息。调用recv时，必须指定一个整数来控制本次调用所接受的最大数据量。recv方法在接受数据时会进入'blocked'状态，最后返回一个字符串，用它来表示收到的数据。如果发送的量超过recv所允许，数据会被截断。多余的数据将缓冲于接受端。以后调用recv时，多余的数据会从缓冲区删除。

6. 传输结束，服务器调用socket的close方法以关闭连接。

TCP服务器端设计（伪码流程）:

ss = socket()

ss.bind()

ss.listen()

inf_loop:

cs = ss.accept()

comm_loop:

cs.recv()/cs.send()

cs.close()

ss.close()

示例代码一(时间戳服务器tsTserv.py):

#! /usr/bin/env python
#coding=utf-8
from socket import *
from time import ctime

HOST = ''
PORT = 21567
BUFSIZE = 1024
ADDR = (HOST,PORT)

tcpServSock = socket(AF_INET,SOCK_STREAM)
tcpServSock.bind(ADDR)
tcpServSock.listen(5)

while 1:
print('waiting for connection...')
tcpCliSock,addr = tcpServSock.accept()
print('...connected from :',addr)

while 1:
data = tcpCliSock.recv(BUFSIZE)
if not data:
break
tcpCliSock.send('[%s] %s' %(ctime(),data))
tcpCliSock.close()

tcpServSock.close()

建立一个TCP客户端连接则需要4个步骤：

1. 创建一个socket以连接服务器 socket=socket.socket(family,type)

2. 使用socket的connect方法连接服务器 socket.connect((host,port))

3. 客户和服务器通过send和recv方法通信。

4. 结束后，客户通过调用socket的close方法来关闭连接。

TCP客户端设计（伪码流程）：

cs = socket()

cs.connect()

comm_loop:

cs.send()/cs.recv()

cs.close()

示例代码二(时间戳客户端tsTclnt.py):

#! /usr/bin/env python
#coding=utf-8
from socket import *

HOST = 'localhost'
PORT = 21567
BUFSIZE = 1024
ADDR = (HOST,PORT)

tcpCliSock = socket(AF_INET,SOCK_STREAM)
tcpCliSock.connect(ADDR)

while 1:
data = raw_input('> ')
if not data:
break
tcpCliSock.send(data)
data = tcpCliSock.recv(BUFSIZE)
if not data:
break
print(data)

tcpCliSock.close()

运行结果：






REF:

1，Core Python Programming

2，http://blog.sina.com.cn/s/blog_6826662b0100yf3g.html