1. 程式人生 > >python實現TCP通訊

python實現TCP通訊

1.    TCP是一種面向連線的可靠地協議,在一方傳送資料之前,必須在雙方之間建立一個連線,建立的過程需要經過三次握手,通訊完成後要拆除連線,需要經過四次握手,這是由TCP的半關閉造成的,一方在完成資料傳送後要傳送一個FIN來終止這個方向的連線,一個TCP連線在收到一個FIN後仍能傳送資料,但應用程式很少這麼做,下面是TCP連線建立和拆除的過程:

                                             

2.python可以實現TCP伺服器和客戶端的程式設計,下面是程式碼:

伺服器端:

#!/usr/bin/env python
import socket
host="localhost"
port=10000
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind((host,port))
s.listen(5)
while 1:
	sock,addr=s.accept()
	print "got connection form ",sock.getpeername()
	data=sock.recv(1024)
	if not data:
		break
	else:
		print data

客戶端:

#!/usr/bin/env python
import socket
host="localhost"
port=10000
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((host,port))
s.send("hello from client")
s.close()

3.使用socketserver框架編寫TCP伺服器

    Socketserver模組可以簡化網路伺服器的編寫,它包含了四種伺服器類,TCPServer使用TCP協議,UDPServer使用UDP協議,還有兩個不常使用的,即UnixStreamServer

UnixDatagramServer,這兩個類僅僅在unix環境下有用。

    使用伺服器程式設計,需要進行一下步驟,先建立一個請求控制代碼類,這個類繼承自BaseRequestHandler類,建立這個類後重寫它的handle方法,然後例項化伺服器類,把主機名,埠號和控制代碼類傳給它,然後呼叫server_forever()方法來處理請求。

使用socketserver框架的伺服器:

import SocketServer
host=''
port=10000
class Handler(SocketServer.StreamRequestHandler):
	
	def handler(self):
		addr=self.request.getpeername()
		print "got connection from",addr
		self.wfile.write("connected")
		
server=SocketServer.TCPServer((host,port),Handler)
server.serve_forever()

    上面的socketserver伺服器只能處理一個請求,如果要處理多個請求,可以用forking或者threading實現,實現多程序或者多執行緒伺服器。下面是使用forkingthreading的伺服器程式碼:

使用forking的伺服器

from SocketServer import TCPServer,ForkingMixIn,StreamRequestHandler
class Server(ForkingMixIn,TCPServer):pass
class Handler(StreamRequestHandler):
	
	def handle(self):
		addr=self.request.getpeername()
		print "got connection from",addr
		self.wfile.write('connected')
		
server=Server((''.10000),Handler)
server.serve_forever()

使用多執行緒的伺服器:

from SocketServer import TCPServer,ThreadingMixIn,StreamRequestHandler
class Server(ThreadingMixIn,TCPServer):pass

class Handler(StreamRequestHandler):
	def handle(self):
		addr=self.request.getpeername()
		print "got connection from",addr
		self.wfile.write("connected")
		
server=Server(('',10000),Handler)
server.serve_forever()