pythonsocket自动化教程_Python 的 Socket 编程教程
這是用來快速學(xué)習(xí) Python Socket 套接字編程的指南和教程。Python 的 Socket 編程跟 C 語言很像。
基本上，Socket 是任何一種計算機(jī)網(wǎng)絡(luò)通訊中最基礎(chǔ)的內(nèi)容。例如當(dāng)你在瀏覽器地址欄中輸入 www.oschina.net 時，你會打開一個套接字，然后連接到 www.oschina.net 并讀取響應(yīng)的頁面然后然后顯示出來。而其他一些聊天客戶端如 gtalk 和 skype 也是類似。任何網(wǎng)絡(luò)通訊都是通過 Socket 來完成的。
寫在開頭
本教程假設(shè)你已經(jīng)有一些基本的 Python 編程的知識。
讓我們開始 Socket 編程吧。
創(chuàng)建 Socket
首先要做的就是創(chuàng)建一個 Socket，socket 的 socket 函數(shù)可以實(shí)現(xiàn)，代碼如下：
1
#Socket client example in python
2
3
import socket#for sockets
4
5
#create an AF_INET, STREAM socket (TCP)
6
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
7
8
print 'Socket Created'
函數(shù) socket.socket 創(chuàng)建了一個 Socket，并返回 Socket 的描述符可用于其他 Socket 相關(guān)的函數(shù)。
上述代碼使用了下面兩個屬性來創(chuàng)建 Socket：
地址簇 : AF_INET (IPv4)
類型: SOCK_STREAM (使用 TCP 傳輸控制協(xié)議)
錯誤處理
如果 socket 函數(shù)失敗了，python 將拋出一個名為 socket.error 的異常，這個異常必須予以處理：
01
#handling errors in python socket programs
02
03
import socket#for sockets
04
import sys#for exit
05
06
try:
07
#create an AF_INET, STREAM socket (TCP)
08
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
09
except socket.error, msg:
10
print 'Failed to create socket. Error code: ' + str(msg[0])+ ' , Error message : ' + msg[1]
11
sys.exit();
12
13
print 'Socket Created'
好了，假設(shè)你已經(jīng)成功創(chuàng)建了 Socket，下一步該做什么呢？接下來我們將使用這個 Socket 來連接到服務(wù)器。
注意
與 SOCK_STREAM 相對應(yīng)的其他類型是 SOCK_DGRAM 用于 UDP 通訊協(xié)議，UDP 通訊是非連接 Socket，在這篇文章中我們只討論 SOCK_STREAM ，或者叫 TCP 。
連接到服務(wù)器
連接到服務(wù)器需要服務(wù)器地址和端口號，這里使用的是 www.oschina.net 和 80 端口。
首先獲取遠(yuǎn)程主機(jī)的 IP 地址
連接到遠(yuǎn)程主機(jī)之前，我們需要知道它的 IP 地址，在 Python 中，獲取 IP 地址是很簡單的：
01
import socket#for sockets
02
import sys#for exit
03
04
try:
05
#create an AF_INET, STREAM socket (TCP)
06
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
07
except socket.error, msg:
08
print 'Failed to create socket. Error code: ' + str(msg[0])+ ' , Error message : ' + msg[1]
09
sys.exit();
10
11
print 'Socket Created'
12
13
host= 'www.oschina.net'
14
15
try:
16
remote_ip= socket.gethostbyname( host )
17
18
except socket.gaierror:
19
#could not resolve
20
print 'Hostname could not be resolved. Exiting'
21
sys.exit()
22
23
print 'Ip address of ' + host+ ' is ' + remote_ip
我們已經(jīng)有 IP 地址了，接下來需要指定要連接的端口。
代碼：
01
import socket#for sockets
02
import sys#for exit
03
04
try:
05
#create an AF_INET, STREAM socket (TCP)
06
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
07
except socket.error, msg:
08
print 'Failed to create socket. Error code: ' + str(msg[0])+ ' , Error message : ' + msg[1]
09
sys.exit();
10
11
print 'Socket Created'
12
13
host= 'www.oschina.net'
14
port= 80
15
16
try:
17
remote_ip= socket.gethostbyname( host )
18
19
except socket.gaierror:
20
#could not resolve
21
print 'Hostname could not be resolved. Exiting'
22
sys.exit()
23
24
print 'Ip address of ' + host+ ' is ' + remote_ip
25
26
#Connect to remote server
27
s.connect((remote_ip , port))
28
29
print 'Socket Connected to ' + host+ ' on ip ' + remote_ip
現(xiàn)在運(yùn)行程序
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
這段程序創(chuàng)建了一個 Socket 并進(jìn)行連接，試試使用其他一些不存在的端口(如81)會是怎樣？這個邏輯相當(dāng)于構(gòu)建了一個端口掃描器。
已經(jīng)連接上了，接下來就是往服務(wù)器上發(fā)送數(shù)據(jù)。
免費(fèi)提示
使用 SOCK_STREAM/TCP 套接字才有“連接”的概念。連接意味著可靠的數(shù)據(jù)流通訊機(jī)制，可以同時有多個數(shù)據(jù)流。可以想象成一個數(shù)據(jù)互不干擾的管道。另外一個重要的提示是：數(shù)據(jù)包的發(fā)送和接收是有順序的。
其他一些 Socket 如 UDP、ICMP 和 ARP 沒有“連接”的概念，它們是無連接通訊，意味著你可從任何人或者給任何人發(fā)送和接收數(shù)據(jù)包。
發(fā)送數(shù)據(jù)
sendall 函數(shù)用于簡單的發(fā)送數(shù)據(jù)，我們來向 oschina 發(fā)送一些數(shù)據(jù)：
01
import socket#for sockets
02
import sys#for exit
03
04
try:
05
#create an AF_INET, STREAM socket (TCP)
06
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
07
except socket.error, msg:
08
print 'Failed to create socket. Error code: ' + str(msg[0])+ ' , Error message : ' + msg[1]
09
sys.exit();
10
11
print 'Socket Created'
12
13
host= 'www.oschina.net'
14
port= 80
15
16
try:
17
remote_ip= socket.gethostbyname( host )
18
19
except socket.gaierror:
20
#could not resolve
21
print 'Hostname could not be resolved. Exiting'
22
sys.exit()
23
24
print 'Ip address of ' + host+ ' is ' + remote_ip
25
26
#Connect to remote server
27
s.connect((remote_ip , port))
28
29
print 'Socket Connected to ' + host+ ' on ip ' + remote_ip
30
31
#Send some data to remote server
32
message= "GET / HTTP/1.1\r\n\r\n"
33
34
try :
35
#Set the whole string
36
s.sendall(message)
37
except socket.error:
38
#Send failed
39
print 'Send failed'
40
sys.exit()
41
42
print 'Message send successfully'
上述例子中，首先連接到目標(biāo)服務(wù)器，然后發(fā)送字符串?dāng)?shù)據(jù) "GET / HTTP/1.1\r\n\r\n" ，這是一個 HTTP 協(xié)議的命令，用來獲取網(wǎng)站首頁的內(nèi)容。
接下來需要讀取服務(wù)器返回的數(shù)據(jù)。
接收數(shù)據(jù)
recv 函數(shù)用于從 socket 接收數(shù)據(jù)：
01
#Socket client example in python
02
03
import socket#for sockets
04
import sys#for exit
05
06
#create an INET, STREAMing socket
07
try:
08
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
09
except socket.error:
10
print 'Failed to create socket'
11
sys.exit()
12
13
print 'Socket Created'
14
15
host= 'oschina.net';
16
port= 80;
17
18
try:
19
remote_ip= socket.gethostbyname( host )
20
21
except socket.gaierror:
22
#could not resolve
23
print 'Hostname could not be resolved. Exiting'
24
sys.exit()
25
26
#Connect to remote server
27
s.connect((remote_ip , port))
28
29
print 'Socket Connected to ' + host+ ' on ip ' + remote_ip
30
31
#Send some data to remote server
32
message= "GET / HTTP/1.1\r\nHost: oschina.net\r\n\r\n"
33
34
try :
35
#Set the whole string
36
s.sendall(message)
37
except socket.error:
38
#Send failed
39
print 'Send failed'
40
sys.exit()
41
42
print 'Message send successfully'
43
44
#Now receive data
45
reply= s.recv(4096)
46
47
print reply
下面是上述程序執(zhí)行的結(jié)果：
01
$ python client.py
02
Socket Created
03
Ip address of oschina.net is 61.145.122.
04
Socket Connected to oschina.net on ip 61.145.122.155
05
Message send successfully
06
HTTP/1.1 301 Moved Permanently
07
Server: nginx
08
Date: Wed, 24 Oct 2012 13:26:46 GMT
09
Content-Type: text/html
10
Content-Length: 178
11
Connection: keep-alive
12
Keep-Alive: timeout=20
oschina.net 回應(yīng)了我們所請求的 URL 的內(nèi)容，很簡單。數(shù)據(jù)接收完了，可以關(guān)閉 Socket 了。
關(guān)閉 socket
close 函數(shù)用于關(guān)閉 Socket：
1
s.close()
這就是了。
讓我們回顧一下
上述的示例中我們學(xué)到了如何：
1. 創(chuàng)建 Socket
2. 連接到遠(yuǎn)程服務(wù)器
3. 發(fā)送數(shù)據(jù)
4. 接收回應(yīng)
當(dāng)你用瀏覽器打開 www.oschina.net
時，其過程也是一樣。包含兩種類型，分別是客戶端和服務(wù)器，客戶端連接到服務(wù)器并讀取數(shù)據(jù)，服務(wù)器使用 Socket
接收進(jìn)入的連接并提供數(shù)據(jù)。因此在這里 www.oschina.net 是服務(wù)器端，而你的瀏覽器是客戶端。
接下來我們開始在服務(wù)器端做點(diǎn)編碼。
服務(wù)器端編程
服務(wù)器端編程主要包括下面幾步：
1. 打開 socket
2. 綁定到一個地址和端口
3. 偵聽進(jìn)來的連接
4. 接受連接
5. 讀寫數(shù)據(jù)
我們已經(jīng)學(xué)習(xí)過如何打開 Socket 了，下面是綁定到指定的地址和端口上。
綁定 Socket
bind 函數(shù)用于將 Socket 綁定到一個特定的地址和端口，它需要一個類似 connect 函數(shù)所需的 sockaddr_in 結(jié)構(gòu)體。
示例代碼：
01
import socket
02
import sys
03
04
HOST= ''# Symbolic name meaning all available interfaces
05
PORT= 8888 # Arbitrary non-privileged port
06
07
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
08
print 'Socket created'
09
10
try:
11
s.bind((HOST, PORT))
12
except socket.error , msg:
13
print 'Bind failed. Error Code : ' + str(msg[0])+ ' Message ' + msg[1]
14
sys.exit()
15
16
print 'Socket bind complete'
綁定完成后，就需要讓 Socket 開始偵聽連接。很顯然，你不能將兩個不同的 Socket 綁定到同一個端口之上。
連接偵聽
綁定 Socket 之后就可以開始偵聽連接，我們需要將 Socket 變成偵聽模式。socket 的 listen 函數(shù)用于實(shí)現(xiàn)偵聽模式：
1
s.listen(10)
2
print'Socket now listening'
listen 函數(shù)所需的參數(shù)成為 backlog，用來控制程序忙時可保持等待狀態(tài)的連接數(shù)。這里我們傳遞的是 10，意味著如果已經(jīng)有 10 個連接在等待處理，那么第 11 個連接將會被拒絕。當(dāng)檢查了 socket_accept 后這個會更加清晰。
接受連接
示例代碼：
01
import socket
02
import sys
03
04
HOST= ''# Symbolic name meaning all available interfaces
05
PORT= 8888 # Arbitrary non-privileged port
06
07
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
08
print 'Socket created'
09
10
try:
11
s.bind((HOST, PORT))
12
except socket.error , msg:
13
print 'Bind failed. Error Code : ' + str(msg[0])+ ' Message ' + msg[1]
14
sys.exit()
15
16
print 'Socket bind complete'
17
18
s.listen(10)
19
print 'Socket now listening'
20
21
#wait to accept a connection - blocking call
22
conn, addr= s.accept()
23
24
#display client information
25
print 'Connected with ' + addr[0]+ ':' + str(addr[1])
輸出
運(yùn)行該程序?qū)@示：
1
$ python server.py
2
Socket created
3
Socket bind complete
4
Socket now listening
現(xiàn)在這個程序開始等待連接進(jìn)入，端口是 8888，請不要關(guān)閉這個程序，我們來通過 telnet 程序來進(jìn)行測試。
打開命令行窗口并輸入：
1
$ telnet localhost 8888
2
3
It will immediately show
4
$ telnet localhost 8888
5
Trying 127.0.0.1...
6
Connected to localhost.
7
Escape character is'^]'.
8
Connection closed by foreign host.
而服務(wù)器端窗口顯示的是：
1
$ python server.py
2
Socket created
3
Socket bind complete
4
Socket now listening
5
Connected with 127.0.0.1:59954
我們可看到客戶端已經(jīng)成功連接到服務(wù)器。
上面例子我們接收到連接并立即關(guān)閉，這樣的程序沒什么實(shí)際的價值，連接建立后一般會有大量的事情需要處理，因此讓我們來給客戶端做出點(diǎn)回應(yīng)吧。
sendall 函數(shù)可通過 Socket 給客戶端發(fā)送數(shù)據(jù)：
01
import socket
02
import sys
03
04
HOST= ''# Symbolic name meaning all available interfaces
05
PORT= 8888 # Arbitrary non-privileged port
06
07
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
08
print 'Socket created'
09
10
try:
11
s.bind((HOST, PORT))
12
except socket.error , msg:
13
print 'Bind failed. Error Code : ' + str(msg[0])+ ' Message ' + msg[1]
14
sys.exit()
15
16
print 'Socket bind complete'
17
18
s.listen(10)
19
print 'Socket now listening'
20
21
#wait to accept a connection - blocking call
22
conn, addr= s.accept()
23
24
print 'Connected with ' + addr[0]+ ':' + str(addr[1])
25
26
#now keep talking with the client
27
data= conn.recv(1024)
28
conn.sendall(data)
29
30
conn.close()
31
s.close()
繼續(xù)運(yùn)行上述代碼，然后打開另外一個命令行窗口輸入下面命令：
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.
可看到客戶端接收到來自服務(wù)器端的回應(yīng)內(nèi)容。
上面的例子還是一樣，服務(wù)器端回應(yīng)后就立即退出了。而一些真正的服務(wù)器像 www.oschina.net 是一直在運(yùn)行的，時刻接受連接請求。
也就是說服務(wù)器端應(yīng)該一直處于運(yùn)行狀態(tài)，否則就不能成為“服務(wù)”，因此我們要讓服務(wù)器端一直運(yùn)行，最簡單的方法就是把 accept 方法放在一個循環(huán)內(nèi)。
一直在運(yùn)行的服務(wù)器
對上述代碼稍作改動：
01
import socket
02
import sys
03
04
HOST= ''# Symbolic name meaning all available interfaces
05
PORT= 8888 # Arbitrary non-privileged port
06
07
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
08
print 'Socket created'
09
10
try:
11
s.bind((HOST, PORT))
12
except socket.error , msg:
13
print 'Bind failed. Error Code : ' + str(msg[0])+ ' Message ' + msg[1]
14
sys.exit()
15
16
print 'Socket bind complete'
17
18
s.listen(10)
19
print 'Socket now listening'
20
21
#now keep talking with the client
22
while 1:
23
#wait to accept a connection - blocking call
24
conn, addr= s.accept()
25
print 'Connected with ' + addr[0]+ ':' + str(addr[1])
26
27
data= conn.recv(1024)
28
reply= 'OK...' + data
29
if not data:
30
break
31
32
conn.sendall(reply)
33
34
conn.close()
35
s.close()
很簡單只是加多一個 while 1 語句而已。
繼續(xù)運(yùn)行服務(wù)器，然后打開另外三個命令行窗口。每個窗口都使用 telnet 命令連接到服務(wù)器：
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.
服務(wù)器所在的終端窗口顯示的是：
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
你看服務(wù)器再也不退出了，好吧，用 Ctrl+C 關(guān)閉服務(wù)器，所有的 telnet 終端將會顯示 "Connection closed by foreign host."
已經(jīng)很不錯了，但是這樣的通訊效率太低了，服務(wù)器程序使用循環(huán)來接受連接并發(fā)送回應(yīng)，這相當(dāng)于是一次最多處理一個客戶端的請求，而我們要求服務(wù)器可同時處理多個請求。
處理多個連接
為了處理多個連接，我們需要一個獨(dú)立的處理代碼在主服務(wù)器接收到連接時運(yùn)行。一種方法是使用線程，服務(wù)器接收到連接然后創(chuàng)建一個線程來處理連接收發(fā)數(shù)據(jù)，然后主服務(wù)器程序返回去接收新的連接。
下面是我們使用線程來處理連接請求：
01
import socket
02
import sys
03
from threadimport *
04
05
HOST= ''# Symbolic name meaning all available interfaces
06
PORT= 8888 # Arbitrary non-privileged port
07
08
s= socket.socket(socket.AF_INET, socket.SOCK_STREAM)
09
print 'Socket created'
10
11
#Bind socket to local host and port
12
try:
13
s.bind((HOST, PORT))
14
except socket.error , msg:
15
print 'Bind failed. Error Code : ' + str(msg[0])+ ' Message ' + msg[1]
16
sys.exit()
17
18
print 'Socket bind complete'
19
20
#Start listening on socket
21
s.listen(10)
22
print 'Socket now listening'
23
24
#Function for handling connections. This will be used to create threads
25
def clientthread(conn):
26
#Sending message to connected client
27
conn.send('Welcome to the server. Type something and hit enter\n')#send only takes string
28
29
#infinite loop so that function do not terminate and thread do not end.
30
while True:
31
32
#Receiving from client
33
data= conn.recv(1024)
34
reply= 'OK...' + data
35
if not data:
36
break
37
38
conn.sendall(reply)
39
40
#came out of loop
41
conn.close()
42
43
#now keep talking with the client
44
while 1:
45
#wait to accept a connection - blocking call
46
conn, addr= s.accept()
47
print 'Connected with ' + addr[0]+ ':' + str(addr[1])
48
49
#start new thread takes 1st argument as a function name to be run, second is the tuple of arguments to the function.
50
start_new_thread(clientthread ,(conn,))
51
52
s.close()
運(yùn)行上述服務(wù)端程序，然后像之前一樣打開三個終端窗口并執(zhí)行 telent 命令：
01
$ telnet localhost 8888
02
Trying 127.0.0.1...
03
Connected to localhost.
04
Escape character is'^]'.
05
Welcome to the server. Type something and hit enter
06
hi
07
OK...hi
08
asd
09
OK...asd
10
cv
11
OK...cv
服務(wù)器端所在終端窗口輸出信息如下：
1
$ python server.py
2
Socket created
3
Socket bind complete
4
Socket now listening
5
Connected with127.0.0.1:60730
6
Connected with127.0.0.1:60731
線程接管了連接并返回相應(yīng)數(shù)據(jù)給客戶端。
這便是我們所要介紹的服務(wù)器端編程。
結(jié)論
到這里為止，你已經(jīng)學(xué)習(xí)了 Python 的 Socket 基本編程，你可自己動手編寫一些例子來強(qiáng)化這些知識。
你可能會遇見一些問題：Bind failed. Error Code : 98 Message Address already in use，碰見這種問題只需要簡單更改服務(wù)器端口即可。
總結(jié)
以上是生活随笔為你收集整理的pythonsocket自动化教程_Python 的 Socket 编程教程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： php拿站,从建站到拿站
- 下一篇： ThinkPHP框架在linux服务器n