python通過socket實現(xiàn)多個連接并實現(xiàn)ssh功能詳解

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

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

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

三、Socket實現(xiàn)多個連接

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

　　事例代碼：

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

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

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

conn.send(data_server)

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

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

　　結(jié)果如下：

注：socket中recv()和send()函數(shù)接收和發(fā)送數(shù)據(jù)大小都是有限制的。如果一次發(fā)送太大，客戶端接收不完，就會先存儲在緩存當(dāng)中。但是下一次命令，客戶端接收的還是上次命令沒有發(fā)完的數(shù)據(jù)。
四、模擬FTP上傳文件
　　本例當(dāng)服務(wù)端在Linux環(huán)境中，struck模塊有些問題....，目前在win10中可以正常運行
　　服務(wù)端：　    
# -*- 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):
    # 聲明協(xié)議類型
    ftp_server = socket.socket()
    # 綁定本地網(wǎng)卡（多網(wǎng)卡選擇），端口
    ftp_server.bind((self.host, self.port))
    # 監(jiān)聽端口
    ftp_server.listen() # 監(jiān)聽
    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):
    # 聲明協(xié)議類型,同時生成socket對象
    ftp_client = socket.socket()
    # 連接服務(wù)端
    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') # 定義打包規(guī)則
        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()

　　結(jié)果：

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

總結(jié)

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