python通過socket實現多個連接并實現ssh功能詳解

一、前言
　　上一篇中我們已經知道了客戶端通過socket來連接服務端，進行了一次數據傳輸，那如何實現客戶端多次發生數據？而服務端接受多個客戶端呢？
二、發送中文信息
　　在python3中，socket只能發送bytes類型的數據，bytes類型只能表示0-225的ASCII碼的值，并不能表示中文，所以當我們需要發送中文時，需要使用到編碼和解碼。

客戶端：    
import socket
# 客戶端
# 聲明協議類型,同時生成socket對象
client = socket.socket()
#
client.connect(('localhost', 8888))
# python3 接收字節流數據
msg = input('>>:').strip()
client.send(msg.encode('utf-8')) # 先編碼成utf-8格式
data = client.recv(1024) # 1024字節的數據
print(data)
print(data.decode())

服務端：    
import socket
# 聲明協議類型
server = socket.socket()
# 綁定本地網卡（多網卡選擇），端口
server.bind(('localhost', 8888))
# 監聽端口
server.listen() # 監聽
# conn 就是客戶端連接后，在服務器端為其生成的一個連接實例
# address 是客戶端的 hostaddr,port
conn, address = server.accept()
print("進入等待時間....")
# 等待
# print(conn, address)
print("收到連接....")
# 接收數據
data_server = conn.recv(1024)
print('receive:', data_server.decode()) # 解碼
# 返回一個值
conn.send(data_server)

三、Socket實現多個連接

　　這個事例需要在Linux環境下測試，在windows中測試時，多個客戶端同時連接(也就是同時運行多個socket_client.py程序)，其中一個斷開，服務端會報錯。Linux環境python版本為3.5。

　　事例代碼：

客戶端：    
import socket
# 客戶端
# 聲明協議類型,同時生成socket對象
client = socket.socket()
#
client.connect(('localhost', 8888))
# python3 接收字節流數據
while True:
  msg = input('>>:').strip()
  if len(msg) == 0:   # 輸入不能為空
    continue
  client.send(msg.encode('utf-8'))
  data = client.recv(1024) # 1024字節的數據
  print(data)
  print(data.decode())

　　事例中輸入為空（即len(msg==0)）是不可以的，如果不輸入任何東西，socket程序默認等待你的輸入，所以程序會卡掉。

服務端：    
# -*- coding: UTF-8 -*-
import socket
# 聲明協議類型
server = socket.socket()
# 綁定本地網卡（多網卡選擇），端口
server.bind(('localhost', 8888))
# 監聽端口
server.listen(5) # 監聽
while True:
  conn, address = server.accept()
  print("進入等待時間....")
  while True:
    print("收到連接....")
    # 接收數據
    data_server = conn.recv(1024)
    if not data_server:  # 這里判斷客戶端斷開的情況，不控制會無限循環
      print('client is lost...')
      break
    print('receive:', data_server.decode())
    # 返回一個值

conn.send(data_server)

在Linux服務器上，我開啟了6個客戶端，每個客戶端輸入一次就斷開（即顯示了client has lost...），服務端分別和六個客戶端連接并接收數據。
四、實現ssh功能
4.1 測試環境
　　服務端： 172.16.200.49，監聽端口（'0.0.0.0', 8888），Linux系統
　　客戶端：本機win10
4.2 測試代碼
　　服務端：    
# -*- coding: UTF-8 -*-
import socket
import os
# 聲明協議類型
server = socket.socket()
# 綁定本地網卡（多網卡選擇），端口
server.bind(('0.0.0.0', 8888))
# 監聽端口
server.listen() # 監聽
while True:
  conn, address = server.accept()
  print("進入等待時間....")
  while True:
    print("收到連接....")
    # 接收數據
    data_server = conn.recv(1024)
    if not data_server:
      print('client is lost...')
      break
    res = os.popen("{}".format(data_server.decode())).read() # 將執行命令的結果存儲返回
    # 返回結果
    conn.send(res.encode('utf-8'))

　　客戶端：    
# -*- coding: UTF-8 -*-
import socket
# 客戶端
# 聲明協議類型,同時生成socket對象
client = socket.socket()
#
client.connect(('172.16.200.49', 8888)) # 服務端ip和端口
# python3 接收字節流數據
while True:
  msg = input('>>:').strip()
  if len(msg) == 0:
    continue
  client.send(msg.encode('utf-8'))
  data = client.recv(1024) # 1024字節的數據
  print(data.decode())

　　結果如下：

注：socket中recv()和send()函數接收和發送數據大小都是有限制的。如果一次發送太大，客戶端接收不完，就會先存儲在緩存當中。但是下一次命令，客戶端接收的還是上次命令沒有發完的數據。
四、模擬FTP上傳文件
　　本例當服務端在Linux環境中，struck模塊有些問題....，目前在win10中可以正常運行
　　服務端：　    
# -*- coding: UTF-8 -*-
import struct
import socket
class FtpServer(object):
  def __init__(self, host, port):
    self.host = host
    self.port = port
  def ftp_server(self):
    # 聲明協議類型
    ftp_server = socket.socket()
    # 綁定本地網卡（多網卡選擇），端口
    ftp_server.bind((self.host, self.port))
    # 監聽端口
    ftp_server.listen() # 監聽
    while True:
      print('等待...')
      conn, address = ftp_server.accept()
      while True:
        file_info = struct.calcsize('128sl')
        buf = conn.recv(file_info)
        if buf:
          file_name, file_size = struct.unpack('128sl', buf)
          # 使用strip()刪除打包時附加的多余空字符
          file_new_name = file_name.decode().strip('\00')
          print('start receiving...')
          fw = open(file_new_name, 'wb')
          received_size = 0 # 接收文件的大小
          while not received_size == file_size:
            if file_size - received_size > 1024:
              r_data = conn.recv(1024)
              received_size += len(r_data)
            else:
              r_data = conn.recv(file_size - received_size)
              received_size = file_size
            fw.write(r_data)
          fw.close()
if __name__ == '__main__':
  server = FtpServer('localhost', 8888)
  server.ftp_server()

　　客戶端：    
# -*- coding: UTF-8 -*-
import socket
import os
import struct
class FtpClient(object):
  # 定義一個FtpClien類
  def __init__(self, host, port):
    self.host = host
    self.port = port
  def client_push(self):
    # 聲明協議類型,同時生成socket對象
    ftp_client = socket.socket()
    # 連接服務端
    ftp_client.connect((self.host, self.port))
    while True:
      # 切換文件目錄路徑
      print("輸入文件目錄路徑")
      pwd = input(">>:").strip()
      # 列出文件名稱
      files_list = os.listdir('{}'.format(pwd))
      for i in files_list:
        print(i)
      file_name = input('輸入上傳的文件名:').strip()
      file_path = os.path.join(pwd, file_name)
      if os.path.isfile(file_path):
        file_info = struct.calcsize('128sl') # 定義打包規則
        f_head = struct.pack('128sl', file_name.encode('utf-8'), os.stat(file_path).st_size)
        ftp_client.send(f_head)
        fo = open(file_path, 'rb')
        while True:
          file_data = fo.read(1024)
          if not file_data:
            break
          ftp_client.send(file_data)
        fo.close()
        # 上傳文件
        ftp_client.send(file_data)
# client.close()
if __name__ == '__main__':
  client = FtpClient('localhost', 8888)
  client.client_push()

　　結果：

在socket_server.py文件位置處能看到上傳的文件

總結

以上就是本文關于python通過socket實現多個連接并實現ssh功能詳解的全部內容，希望對大家有所幫助。