Python socket服務常用操作程式碼例項
阿新 • • 發佈:2020-06-22
套接字(socket)是一個抽象層,應用程式可以通過它傳送或接收資料,可對其進行像對檔案一樣的開啟、讀寫和關閉等操作。
1. 實現客戶端傳送字元,伺服器返回大寫的字元:
伺服器:
import socketserver class MyTCPHandler(socketserver.BaseRequestHandler): # 通過類的繼承,實現 def handle(self): # 重寫父類的handle方法,所有的操作都在此方法中 while True: # 迴圈,不停的接收從客戶端來的資料 try: self.data = self.request.recv(1024).strip() # 從客戶端接收資料,每次收1024位元組 print("{} send:".format(self.client_address),self.data) self.request.send(self.data.upper()) # 從伺服器傳送給客戶端資料 except ConnectionResetError as e: print('Error: ',e) break if __name__ == '__main__': host,port = 'localhost',9999 server = socketserver.ThreadingTCPServer((host,port),MyTCPHandler) # 通過多執行緒實現多個客戶端連線,每個客戶端連線都是一個執行緒 server.serve_forever() # 一直執行服務
客戶端:
import socket client = socket.socket() # socket物件 client.connect(('localhost',9999)) # 連線伺服器地址和埠 while True: # 迴圈,不停的輸入傳送資料 con = input('>>>:').strip() if len(con) ==0: continue # 不能傳送空資料,否則會阻塞 client.send(con.encode('utf-8')) # 傳送資料,必須是二進位制的 data = client.recv(1024) # 接收伺服器返回的資料 print(data.decode()) # 列印 解碼後的資料 client.close() # 關閉
2. 通過socket執行伺服器命令:
用法:直接在客戶端輸入處輸入命令如:ipconfig
伺服器:
import socket import os import threading def tcplink(sock,addr): print('Accept new connection from %s:%s...' % addr) while True: # 和每個接入的客戶端,進行多次資料通訊 data = sock.recv(1024) # 接收客戶端資料 if not data or data.decode('utf-8') == 'exit': # 如果客戶端不傳送資料或者傳送了exit print('client disconnected.') break content = os.popen(data.decode('utf-8')).read() # 對傳送來的資料執行cmd命令,獲取結果 if len(content) == 0: #如果執行的命令結果為空的,就手動造一個結果。因為如果為空資料,會掛起,無法正常傳送。 content = 'cmd not exists.' sock.send(str(len(content.encode('utf-8'))).encode('utf-8')) # 傳送資料的長度 print('send length:',(len(content.encode('utf-8')))) # print('content,',content.encode('utf-8')) recv = sock.recv(1024) # 因為上下都有一個send連在一起,可能發生粘包現象,為了防止這種情況,可以讓客戶端重新應答一下 print('Answer:',recv.decode('utf-8')) sock.send(content.encode('utf-8')) # 傳送資料 print('send finished.') sock.close() print('Connection from %s:%s closed.' % addr) s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 監聽埠: s.bind(('127.0.0.1',9999)) s.listen(3) print('Waiting for connection...') while True: # 接受一個新連線: sock,addr = s.accept() # 建立新執行緒來處理TCP連線: t = threading.Thread(target=tcplink,args=(sock,addr)) t.start()
客戶端:
import socket # AF_INET 代表ipv4,SOCK_STREAM 代表TCP s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 確定網路協議,生成物件 s.connect(('127.0.0.1',9999)) # 連線伺服器的地址和埠,元組的形式。 while True: msg = input('>>:').strip() if len(msg) != 0: # 如果訊息為空,會一直掛起,所以不能為空 if msg =='exit': s.close() # 關閉連線 print('Connection closed.') break s.send(msg.encode('utf-8')) # 給伺服器傳送資料,必須是二進位制的 length = s.recv(1024) # 首先接收伺服器返回的將要接收的資料的長度資訊。 s.send(b'Ready to receive...') # 傳送接收命令 length = int(length.decode('utf-8')) print('receive len:',length) data_len = 0 data_recv = b'' while data_len < length: # 已經接收的資訊的長度,如果小於總長度 data = s.recv(1024) # 從伺服器接收資料 data_recv += data data_len += len(data) print(data_recv.decode('utf-8')) # 列印返回的資料。
3. 通過socket傳輸檔案:
用法:get 檔名
伺服器:
import socket import os import hashlib import threading def tcplink(sock,addr): print('Accept new connection from %s:%s...' % addr) while True: # 和每個接入的客戶端,進行多次資料通訊 data = sock.recv(1024) # 接收客戶端資料 if not data or data.decode('utf-8') == 'exit': # 如果客戶端不傳送資料或者傳送了exit print('client disconnected.') break oper,filename = data.decode('utf-8').split() # 對接收的資料按照空格分割 if oper == 'get': m = hashlib.md5() if os.path.isfile(filename): size = os.stat(filename).st_size # 獲取檔案大小 print('Send size:',size) sock.send(str(size).encode('utf-8')) # 傳送檔案大小 recv = sock.recv(1024) # 接收客戶端確認資訊(因為上下文兩個send是連著的,所以為了防止粘包,接收一次資訊) f = open(filename,'rb') for line in f: sock.send(line) #讀取檔案,傳送給客戶端 m.update(line) # print('Send finished.',m.hexdigest()) # 列印md5的值 sock.send(m.hexdigest().encode('utf-8')) # 把md5的值傳送給客戶端 sock.close() print('Connection from %s:%s closed.' % addr) s = socket.socket(socket.AF_INET,addr)) t.start()
客戶端:
import socket import hashlib # AF_INET 代表ipv4,SOCK_STREAM 代表TCP s = socket.socket(socket.AF_INET,9999)) # 連線伺服器的地址和埠,元組的形式。 while True: msg = input('>>:').strip() if len(msg) != 0: # 如果訊息為空,會一直掛起,所以不能為空 if msg =='exit': s.close() # 關閉連線 print('Connection closed.') break s.send(msg.encode('utf-8')) # 給伺服器傳送資料,必須是二進位制的 length = s.recv(1024) # 首先接收伺服器返回的將要接收的資料的長度資訊。 s.send(b'Ready to receive...') # 傳送接收確認命令 length = int(length.decode('utf-8')) print('Recv size:',length) data_len = 0 data_recv = b'' # 新檔名 fileName = msg.split()[-1].split('.')[0] fileExt = msg.split()[-1].split('.')[-1] newFile = fileName+'-1.'+fileExt f = open(newFile,'wb') # 開啟檔案,準備寫入伺服器發過來的檔案 m = hashlib.md5() while data_len < length: # 已經接收的資訊的長度,如果小於總長度 size = length - data_len if size > 1024: # 如果剩下的資訊長度大於1024,即不能一次性發完。 size = 1024 else: # 如果能一次性發完,就只收剩下的資訊。目的是準確的接收檔案的大小,把可能粘連的send的資料留給下一次recv size = length-data_len data = s.recv(size) # 從伺服器接收資料 f.write(data) m.update(data) data_len += len(data) f.close() print('recv_md5:',m.hexdigest()) # 列印返回的資料。 recv = s.recv(1024) # 接收下一次send的資料,即md5的值。 print('orig_md5:',recv.decode())
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。