Python 的 Socket 編程教程

來源:互聯網
上載者:User

  Python 的 Socket 編程教程

[複製連結]

 
  電梯直達

樓主

發表於 2012-11-10 14:56:10|只看該作者|倒序瀏覽

分享到:
這是用來快速學習 Python Socket 通訊端編程的指南和教程。Python 的 Socket 編程跟 C 語言很像。  Python 官方關於 Socket 的函數請看
http://docs.python.org/library/socket.html

基本上,Socket 是任何一種電腦網路通訊中最基礎的內容。例如當你在瀏覽器地址欄中輸入
www.oschina.net 時,你會開啟一個通訊端,然後串連到
www.oschina.net 並讀取響應的頁面然後然後顯示出來。而其他一些聊天用戶端如 gtalk 和 skype 也是類似。任何網路通訊都是通過 Socket 來完成的。

寫在開頭 本教程假設你已經有一些基本的 Python 編程的知識。
讓我們開始 Socket 編程吧。
建立 Socket 首先要做的就是建立一個 Socket,socket 的 socket 函數可以實現,代碼如下:

  1. #Socket client example in python
  2. import socket        #for sockets
  3. #create an AF_INET, STREAM socket (TCP)
  4. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  5. print 'Socket Created'

複製代碼

函數 socket.socket 建立了一個 Socket,並返回 Socket 的描述符可用於其他 Socket 相關的函數。
上述代碼使用了下面兩個屬性來建立 Socket:
地址簇 : AF_INET (IPv4)
類型: SOCK_STREAM (使用 TCP 傳輸控制通訊協定)
錯誤處理
如果 socket 函數失敗了,python 將拋出一個名為 socket.error 的異常,這個異常必須予以處理:

  1. #handling errors in python socket programs
  2. import socket        #for sockets
  3. import sys        #for exit
  4. try:
  5.         #create an AF_INET, STREAM socket (TCP)
  6.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  7. except socket.error, msg:
  8.         print 'Failed to create socket. Error code: ' + str(msg[0]) + ' , Error message : ' + msg[1]
  9.         sys.exit();
  10. print 'Socket Created'

複製代碼

好了,假設你已經成功建立了 Socket,下一步該做什麼呢?接下來我們將使用這個 Socket 來串連到伺服器。
注意
與 SOCK_STREAM 相對應的其他類型是 SOCK_DGRAM 用於 UDP 通訊協議,UDP 通訊是非串連 Socket,在這篇文章中我們只討論 SOCK_STREAM ,或者叫 TCP 。

串連到伺服器 串連到伺服器需要伺服器位址和連接埠號碼,這裡使用的是
www.oschina.net 和 80 連接埠。
首先擷取遠程主機的 IP 位址
串連到遠程主機之前,我們需要知道它的 IP 位址,在 Python 中,擷取 IP 位址是很簡單的:

  1. import socket        #for sockets
  2. import sys        #for exit
  3. try:
  4.         #create an AF_INET, STREAM socket (TCP)
  5.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. except socket.error, msg:
  7.         print 'Failed to create socket. Error code: ' + str(msg[0]) + ' , Error message : ' + msg[1]
  8.         sys.exit();
  9. print 'Socket Created'
  10. host = 'www.oschina.net'
  11. try:
  12.         remote_ip = socket.gethostbyname( host )
  13. except socket.gaierror:
  14.         #could not resolve
  15.         print 'Hostname could not be resolved. Exiting'
  16.         sys.exit()
  17.        
  18. print 'Ip address of ' + host + ' is ' + remote_ip

複製代碼

我們已經有 IP 位址了,接下來需要指定要串連的連接埠。
代碼:

  1. import socket        #for sockets
  2. import sys        #for exit
  3. try:
  4.         #create an AF_INET, STREAM socket (TCP)
  5.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. except socket.error, msg:
  7.         print 'Failed to create socket. Error code: ' + str(msg[0]) + ' , Error message : ' + msg[1]
  8.         sys.exit();
  9. print 'Socket Created'
  10. host = 'www.oschina.net'
  11. port = 80
  12. try:
  13.         remote_ip = socket.gethostbyname( host )
  14. except socket.gaierror:
  15.         #could not resolve
  16.         print 'Hostname could not be resolved. Exiting'
  17.         sys.exit()
  18.        
  19. print 'Ip address of ' + host + ' is ' + remote_ip
  20. #Connect to remote server
  21. s.connect((remote_ip , port))
  22. print 'Socket Connected to ' + host + ' on ip ' + remote_ip

複製代碼

現在運行程式

  1. $ python client.py
  2. Socket Created
  3. Ip address of www.oschina.net is 61.145.122.155
  4. Socket Connected to www.oschina.net on ip 61.145.122.155

複製代碼

這段程式建立了一個 Socket 並進行串連,試試使用其他一些不存在的連接埠(如81)會是怎樣?這個邏輯相當於構建了一個連接埠掃描器。
已經串連上了,接下來就是往伺服器上發送資料。
免費提示
使用 SOCK_STREAM/TCP 通訊端才有“串連”的概念。串連意味著可靠的資料流通訊機制,可以同時有多個資料流。可以想象成一個資料互不干擾的管道。另外一個重要的提示是:資料包的發送和接收是有順序的。

其他一些 Socket 如 UDP、ICMP 和 ARP 沒有“串連”的概念,它們是無串連通訊,意味著你可從任何人或者給任何人發送和接收資料包。
發送資料
sendall 函數用於簡單的發送資料,我們來向 oschina 發送一些資料:

  1. import socket        #for sockets
  2. import sys        #for exit
  3. try:
  4.         #create an AF_INET, STREAM socket (TCP)
  5.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. except socket.error, msg:
  7.         print 'Failed to create socket. Error code: ' + str(msg[0]) + ' , Error message : ' + msg[1]
  8.         sys.exit();
  9. print 'Socket Created'
  10. host = 'www.oschina.net'
  11. port = 80
  12. try:
  13.         remote_ip = socket.gethostbyname( host )
  14. except socket.gaierror:
  15.         #could not resolve
  16.         print 'Hostname could not be resolved. Exiting'
  17.         sys.exit()
  18.        
  19. print 'Ip address of ' + host + ' is ' + remote_ip
  20. #Connect to remote server
  21. s.connect((remote_ip , port))
  22. print 'Socket Connected to ' + host + ' on ip ' + remote_ip
  23. #Send some data to remote server
  24. message = "GET / HTTP/1.1\r\n\r\n"
  25. try :
  26.         #Set the whole string
  27.         s.sendall(message)
  28. except socket.error:
  29.         #Send failed
  30.         print 'Send failed'
  31.         sys.exit()
  32. print 'Message send successfully'

複製代碼

上述例子中,首先串連到目標伺服器,然後發送字串資料 "GET / HTTP/1.1\r\n\r\n" ,這是一個 HTTP 協議的命令,用來擷取網站首頁的內容。

接下來需要讀取伺服器返回的資料。
接收資料 recv 函數用於從 socket 接收資料:

  1. #Socket client example in python
  2. import socket        #for sockets
  3. import sys        #for exit
  4. #create an INET, STREAMing socket
  5. try:
  6.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  7. except socket.error:
  8.         print 'Failed to create socket'
  9.         sys.exit()
  10.        
  11. print 'Socket Created'
  12. host = 'oschina.net';
  13. port = 80;
  14. try:
  15.         remote_ip = socket.gethostbyname( host )
  16. except socket.gaierror:
  17.         #could not resolve
  18.         print 'Hostname could not be resolved. Exiting'
  19.         sys.exit()
  20. #Connect to remote server
  21. s.connect((remote_ip , port))
  22. print 'Socket Connected to ' + host + ' on ip ' + remote_ip
  23. #Send some data to remote server
  24. message = "GET / HTTP/1.1\r\nHost: oschina.net\r\n\r\n"
  25. try :
  26.         #Set the whole string
  27.         s.sendall(message)
  28. except socket.error:
  29.         #Send failed
  30.         print 'Send failed'
  31.         sys.exit()
  32. print 'Message send successfully'
  33. #Now receive data
  34. reply = s.recv(4096)
  35. print reply

複製代碼

下面是上述程式執行的結果:

  1. $ python client.py
  2. Socket Created
  3. Ip address of oschina.net is 61.145.122.
  4. Socket Connected to oschina.net on ip 61.145.122.155
  5. Message send successfully
  6. HTTP/1.1 301 Moved Permanently
  7. Server: nginx
  8. Date: Wed, 24 Oct 2012 13:26:46 GMT
  9. Content-Type: text/html
  10. Content-Length: 178
  11. Connection: keep-alive
  12. Keep-Alive: timeout=20
  13. Location: http://www.oschina.net/

複製代碼

oschina.net 回應了我們所請求的 URL 的內容,很簡單。資料接收完了,可以關閉 Socket 了。
關閉 socket close 函數用於關閉 Socket:

  1. s.close()

複製代碼

這就是了。
讓我們回顧一下 上述的樣本中我們學到了如何:
1. 建立 Socket
2. 串連到遠程伺服器
3. 發送資料
4. 接收回應
當你用瀏覽器開啟 www.oschina.net 時,其過程也是一樣。包含兩種類型,分別是用戶端和伺服器,用戶端串連到伺服器並讀取資料,伺服器使用 Socket 接收進入的串連並提供資料。因此在這裡
www.oschina.net 是伺服器端,而你的瀏覽器是用戶端。

接下來我們開始在伺服器端做點編碼。
伺服器端編程 伺服器端編程主要包括下面幾步:
1. 開啟 socket
2. 綁定到一個地址和連接埠
3. 偵聽進來的串連
4. 接受串連
5. 讀寫資料
我們已經學習過如何開啟 Socket 了,下面是綁定到指定的地址和連接埠上。
綁定 Socket bind 函數用於將 Socket 綁定到一個特定的地址和連接埠,它需要一個類似 connect 函數所需的 sockaddr_in 結構體。

範例程式碼:

  1. import socket
  2. import sys
  3. HOST = ''        # Symbolic name meaning all available interfaces
  4. PORT = 8888        # Arbitrary non-privileged port
  5. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. print 'Socket created'
  7. try:
  8.         s.bind((HOST, PORT))
  9. except socket.error , msg:
  10.         print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
  11.         sys.exit()
  12.        
  13. print 'Socket bind complete'

複製代碼

綁定完成後,就需要讓 Socket 開始偵聽串連。很顯然,你不能將兩個不同的 Socket 綁定到同一個連接埠之上。
串連偵聽 綁定 Socket 之後就可以開始偵聽串連,我們需要將 Socket 變成偵聽模式。socket 的 listen 函數用於實現偵聽模式:

  1. s.listen(10)
  2. print 'Socket now listening'

複製代碼

listen 函數所需的參數成為 backlog,用來控製程序忙時可保持等待狀態的串連數。這裡我們傳遞的是 10,意味著如果已經有 10 個串連在等待處理,那麼第 11 個串連將會被拒絕。當檢查了 socket_accept 後這個會更加清晰。

接受串連 範例程式碼:

  1. import socket
  2. import sys
  3. HOST = ''        # Symbolic name meaning all available interfaces
  4. PORT = 8888        # Arbitrary non-privileged port
  5. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. print 'Socket created'
  7. try:
  8.         s.bind((HOST, PORT))
  9. except socket.error , msg:
  10.         print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
  11.         sys.exit()
  12.        
  13. print 'Socket bind complete'
  14. s.listen(10)
  15. print 'Socket now listening'
  16. #wait to accept a connection - blocking call
  17. conn, addr = s.accept()
  18. #display client information
  19. print 'Connected with ' + addr[0] + ':' + str(addr[1])

複製代碼

輸出
運行該程式將會顯示:

  1. $ python server.py
  2. Socket created
  3. Socket bind complete
  4. Socket now listening

複製代碼

現在這個程式開始等待串連進入,連接埠是 8888,請不要關閉這個程式,我們來通過 telnet 程式來進行測試。
開啟命令列視窗並輸入:

  1. $ telnet localhost 8888
  2. It will immediately show
  3. $ telnet localhost 8888
  4. Trying 127.0.0.1...
  5. Connected to localhost.
  6. Escape character is '^]'.
  7. Connection closed by foreign host.

複製代碼

而伺服器端視窗顯示的是:

  1. $ python server.py
  2. Socket created
  3. Socket bind complete
  4. Socket now listening
  5. Connected with 127.0.0.1:59954

複製代碼

我們可看到用戶端已經成功串連到伺服器。
上面例子我們接收到串連並立即關閉,這樣的程式沒什麼實際的價值,串連建立後一般會有大量的事情需要處理,因此讓我們來給用戶端做出點回應吧。
sendall 函數可通過 Socket 給用戶端發送資料:

  1. import socket
  2. import sys
  3. HOST = ''        # Symbolic name meaning all available interfaces
  4. PORT = 8888        # Arbitrary non-privileged port
  5. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. print 'Socket created'
  7. try:
  8.         s.bind((HOST, PORT))
  9. except socket.error , msg:
  10.         print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
  11.         sys.exit()
  12.        
  13. print 'Socket bind complete'
  14. s.listen(10)
  15. print 'Socket now listening'
  16. #wait to accept a connection - blocking call
  17. conn, addr = s.accept()
  18. print 'Connected with ' + addr[0] + ':' + str(addr[1])
  19. #now keep talking with the client
  20. data = conn.recv(1024)
  21. conn.sendall(data)
  22. conn.close()
  23. s.close()

複製代碼

繼續運行上述代碼,然後開啟另外一個命令列視窗輸入下面命令:

  1. $ telnet localhost 8888
  2. Trying 127.0.0.1...
  3. Connected to localhost.
  4. Escape character is '^]'.
  5. happy
  6. happy
  7. Connection closed by foreign host.

複製代碼

可看到用戶端接收到來自伺服器端的回應內容。
上面的例子還是一樣,伺服器端回應後就立即退出了。而一些真正的伺服器像
www.oschina.net 是一直在啟動並執行,時刻接受串連請求。
也就是說伺服器端應該一直處於運行狀態,否則就不能成為“服務”,因此我們要讓伺服器端一直運行,最簡單的方法就是把 accept 方法放在一個迴圈內。
一直在啟動並執行伺服器 對上述代碼稍作改動:

  1. import socket
  2. import sys
  3. HOST = ''        # Symbolic name meaning all available interfaces
  4. PORT = 8888        # Arbitrary non-privileged port
  5. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. print 'Socket created'
  7. try:
  8.         s.bind((HOST, PORT))
  9. except socket.error , msg:
  10.         print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
  11.         sys.exit()
  12.        
  13. print 'Socket bind complete'
  14. s.listen(10)
  15. print 'Socket now listening'
  16. #now keep talking with the client
  17. while 1:
  18.     #wait to accept a connection - blocking call
  19.         conn, addr = s.accept()
  20.         print 'Connected with ' + addr[0] + ':' + str(addr[1])
  21.        
  22.         data = conn.recv(1024)
  23.         reply = 'OK...' + data
  24.         if not data:
  25.                 break
  26.        
  27.         conn.sendall(reply)
  28. conn.close()
  29. s.close()

複製代碼

很簡單只是加多一個 while 1 語句而已。
繼續運行伺服器,然後開啟另外三個命令列視窗。每個視窗都使用 telnet 命令串連到伺服器:

  1. $ telnet localhost 5000
  2. Trying 127.0.0.1...
  3. Connected to localhost.
  4. Escape character is '^]'.
  5. happy
  6. OK .. happy
  7. Connection closed by foreign host.

複製代碼

伺服器所在的終端視窗顯示的是:

  1. $ python server.py
  2. Socket created
  3. Socket bind complete
  4. Socket now listening
  5. Connected with 127.0.0.1:60225
  6. Connected with 127.0.0.1:60237
  7. Connected with 127.0.0.1:60239

複製代碼

你看伺服器再也不退出了,好吧,用 Ctrl+C 關閉伺服器,所有的 telnet 終端將會顯示 "Connection closed by foreign host."

已經很不錯了,但是這樣的通訊效率太低了,伺服器程式使用迴圈來接受串連並發送回應,這相當於是一次最多處理一個用戶端的請求,而我們要求伺服器可同時處理多個請求。
處理多個串連 為了處理多個串連,我們需要一個獨立的處理代碼在主伺服器接收到串連時運行。一種方法是使用線程,伺服器接收到串連然後建立一個線程來處理串連收發資料,然後主伺服器程式返回去接收新的串連。

下面是我們使用線程來處理串連請求:

  1. import socket
  2. import sys
  3. from thread import *
  4. HOST = ''        # Symbolic name meaning all available interfaces
  5. PORT = 8888        # Arbitrary non-privileged port
  6. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  7. print 'Socket created'
  8. #Bind socket to local host and port
  9. try:
  10.         s.bind((HOST, PORT))
  11. except socket.error , msg:
  12.         print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
  13.         sys.exit()
  14.        
  15. print 'Socket bind complete'
  16. #Start listening on socket
  17. s.listen(10)
  18. print 'Socket now listening'
  19. #Function for handling connections. This will be used to create threads
  20. def clientthread(conn):
  21.         #Sending message to connected client
  22.         conn.send('Welcome to the server. Type something and hit enter\n') #send only takes string
  23.        
  24.         #infinite loop so that function do not terminate and thread do not end.
  25.         while True:
  26.                
  27.                 #Receiving from client
  28.                 data = conn.recv(1024)
  29.                 reply = 'OK...' + data
  30.                 if not data:
  31.                         break
  32.        
  33.                 conn.sendall(reply)
  34.        
  35.         #came out of loop
  36.         conn.close()
  37. #now keep talking with the client
  38. while 1:
  39.     #wait to accept a connection - blocking call
  40.         conn, addr = s.accept()
  41.         print 'Connected with ' + addr[0] + ':' + str(addr[1])
  42.        
  43.         #start new thread takes 1st argument as a function name to be run, second is the tuple of arguments to the function.
  44.         start_new_thread(clientthread ,(conn,))
  45. s.close()

複製代碼

運行上述服務端程式,然後像之前一樣開啟三個終端視窗並執行 telent 命令:

  1. $ telnet localhost 8888
  2. Trying 127.0.0.1...
  3. Connected to localhost.
  4. Escape character is '^]'.
  5. Welcome to the server. Type something and hit enter
  6. hi
  7. OK...hi
  8. asd
  9. OK...asd
  10. cv
  11. OK...cv

複製代碼

伺服器端所在終端視窗輸出資訊如下:

  1. $ python server.py
  2. Socket created
  3. Socket bind complete
  4. Socket now listening
  5. Connected with 127.0.0.1:60730
  6. Connected with 127.0.0.1:60731

複製代碼

線程接管了串連並返回相應資料給用戶端。
這便是我們所要介紹的伺服器端編程。
結論
到這裡為止,你已經學習了 Python 的 Socket 基本編程,你可自己動手編寫一些例子來強化這些知識。
你可能會遇見一些問題:Bind failed. Error Code : 98 Message Address already in use,碰見這種問題只需要簡單更改伺服器連接埠即可。

 
   
   

相關文章

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.