Python 之 socket 入门

**************************创建 TCP socket***************************

python 编写server的步骤：

   1.
      第一步是创建socket对象。调用socket构造函数。如：
      socket = socket.socket( family, type )
      family参数代表地址家族，可为AF_INET或AF_UNIX。AF_INET家族包括Internet地址，AF_UNIX家族用于同一台机器上的进程间通信。

      type参数代表套接字类型，可为SOCK_STREAM(流套接字)和SOCK_DGRAM(数据报套接字)。

   2.
      第二步是将socket绑定到指定地址。这是通过socket对象的bind方法来实现的：
      socket.bind( address )

      由AF_INET所创建的套接字，address地址必须是一个双元素元组，格式是(host,port)。host代表主机，port代表端口号。如果端口号正在使用、主机名不正确或端口已被保留，bind方法将引发socket.error异常。

   3.
      第三步是使用socket套接字的listen方法接收连接请求。
      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时，多余的数据会从缓冲区 删除(以及自上次调用recv以来，客户可能发送的其它任何数据)。   

   6. 

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

python编写client的步骤：
   1. 创建一个socket以连接服务器：socket = socket.socket( family, type )
   2.  使用socket的connect方法连接服务器。对于AF_INET家族,连接格式如下：
      socket.connect( (host,port) )
      host代表服务器主机名或IP，port代表服务器进程所绑定的端口号。如连接成功，客户就可通过套接字与服务器通信，如果连接失败，会引发socket.error异常。
   3. 处理阶段，客户和服务器将通过send方法和recv方法通信。
   4. 传输结束，客户通过调用socket的close方法关闭连接。
下面给个简单的例子：
server.py
python 代码

1. if __name__ == '__main__':  
   
2.     import socket  
   
3.     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  
   
4.     sock.bind(('localhost', 8001))  
   
5.     sock.listen(5)  
   
6.     while True:  
   
7.         connection,address = sock.accept()  
   
8.         try:  
   
9.             connection.settimeout(5)  
   
10.             buf = connection.recv(1024)  
    
11.             if buf == '1':  
    
12.                 connection.send('welcome to server!')  
    
13.             else:  
    
14.                 connection.send('please go out!')  
    
15.         except socket.timeout:  
    
16.             print 'time out'  
    
17.         connection.close()  

复制代码

client.py
python 代码

1. if __name__ == '__main__':  
   
2.     import socket  
   
3.     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  
   
4.     sock.connect(('localhost', 8001))  
   
5.     import time  
   
6.     time.sleep(2)  
   
7.     sock.send('1')  
   
8.     print sock.recv(1024)  
   
9.     sock.close()  

复制代码

在终端运行server.py，然后运行clien.py，会在终端打印“welcome to server!"。如果更改client.py的sock.send('1')为其它值在终端会打印”please go out!“，更改time.sleep(2)为大于5的数值， 服务器将会超时。

第二个例子：

tcp sever.py

tcp client.py:

**************************创建 UDP socket***************************

UDP服务器代码：

[python]  view plain  copy

1. #!/usr/bin/env python  
2. # -*- coding:UTF-8 -*-  
3.   
4. from socket import *  
5. from time import ctime  
6.   
7. HOST = ''  
8. PORT = 21567  
9. BUFSIZE = 1024  
10.   
11. ADDR = (HOST,PORT)  
12.   
13. udpSerSock = socket(AF_INET, SOCK_DGRAM)  
14. udpSerSock.bind(ADDR)  
15.   
16. while True:  
17.     print 'wating for message...'  
18.     data, addr = udpSerSock.recvfrom(BUFSIZE)  
19.     udpSerSock.sendto('[%s] %s'%(ctime(),data),addr)  
20.     print '...received from and retuned to:',addr  
21.   
22. udpSerSock.close()  

需要注意的是，server中address元组中的引号表示可以接受任何地址来的数据报，TCP例子中的则表示可以接受任意地址发起的连接。

UDP客服端代码：

[python]  view plain  copy

1. #!/usr/bin/env python  
2.   
3. from socket import *  
4.   
5. HOST = 'localhost'  
6. PORT = 21567  
7. BUFSIZE = 1024  
8.   
9. ADDR = (HOST, PORT)  
10.   
11. udpCliSock = socket(AF_INET, SOCK_DGRAM)  
12.   
13. while True:  
14.     data = raw_input('>')  
15.     if not data:  
16.         break  
17.     udpCliSock.sendto(data,ADDR)  
18.     data,ADDR = udpCliSock.recvfrom(BUFSIZE)  
19.     if not data:  
20.         break  
21.     print data  
22.   
23. udpCliSock.close()  

参考：

http://blog.youkuaiyun.com/hu330459076/article/details/7868028

http://blog.163.com/yi_yixinyiyi/blog/static/136286889201152814341144/