Python_TCP编程

1.客户端

# 导入socket库:import socket# 创建一个socket:s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 建立连接:s.connect(('www.sina.com.cn', 80))# 发送数据:s.send('GET / HTTP/1.1\r\nHost: www.sina.com.cn\r\nConnection: close\r\n\r\n')# 接收数据:buffer = []while True:    # 每次最多接收1k字节:    d = s.recv(1024)    if d:        buffer.append(d)    else:        breakdata = ''.join(buffer)# 关闭连接:s.close()header, html = data.split('\r\n\r\n', 1)print header# 把接收的数据写入文件:with open('sina.html', 'wb') as f:    f.write(html)

现在，只需要在浏览器中打开这个sina.html文件，就可以看到新浪的首页了。

---

服务器

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 监听端口:s.bind(('127.0.0.1', 9999))s.listen(5)print 'Waiting for connection...'def tcplink(sock, addr):    print 'Accept new connection from %s:%s...' % addr    sock.send('Welcome!')    while True:        data = sock.recv(1024)        time.sleep(1)        if data == 'exit' or not data:            break        sock.send('Hello, %s!' % data)    sock.close()    print 'Connection from %s:%s closed.' % addrwhile True:    # 接受一个新连接:    sock, addr = s.accept()    # 创建新线程来处理TCP连接:    t = threading.Thread(target=tcplink, args=(sock, addr))    t.start()

连接建立后，服务器首先发一条欢迎消息，然后等待客户端数据，并加上Hello再发送给客户端。如果客户端发送了exit字符串，就直接关闭连接。

要测试这个服务器程序，我们还需要编写一个客户端程序：

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 建立连接:s.connect(('127.0.0.1', 9999))# 接收欢迎消息:print s.recv(1024)for data in ['Michael', 'Tracy', 'Sarah']:    # 发送数据:    s.send(data)    print s.recv(1024)s.send('exit')s.close()