python通過(guò)socket實(shí)現(xiàn)多個(gè)連接并實(shí)現(xiàn)ssh功能詳解

更新時(shí)間：2017年11月08日 14:20:02 作者：Bigberg

這篇文章主要介紹了python通過(guò)socket實(shí)現(xiàn)多個(gè)連接并實(shí)現(xiàn)ssh功能詳解，具有一定參考價(jià)值，需要的朋友可以了解下。

一、前言

　　上一篇中我們已經(jīng)知道了客戶端通過(guò)socket來(lái)連接服務(wù)端，進(jìn)行了一次數(shù)據(jù)傳輸，那如何實(shí)現(xiàn)客戶端多次發(fā)生數(shù)據(jù)？而服務(wù)端接受多個(gè)客戶端呢？

二、發(fā)送中文信息

　　在python3中，socket只能發(fā)送bytes類型的數(shù)據(jù)，bytes類型只能表示0-225的ASCII碼的值，并不能表示中文，所以當(dāng)我們需要發(fā)送中文時(shí)，需要使用到編碼和解碼。

客戶端：

import socket
# 客戶端
# 聲明協(xié)議類型,同時(shí)生成socket對(duì)象
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ù)器端為其生成的一個(gè)連接實(shí)例
# address 是客戶端的 hostaddr,port
conn, address = server.accept()
print("進(jìn)入等待時(shí)間....")
# 等待
# print(conn, address)
print("收到連接....")
# 接收數(shù)據(jù)
data_server = conn.recv(1024)
print('receive:', data_server.decode()) # 解碼
# 返回一個(gè)值
conn.send(data_server)

三、Socket實(shí)現(xiàn)多個(gè)連接

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

　　事例代碼：

客戶端：

import socket
# 客戶端
# 聲明協(xié)議類型,同時(shí)生成socket對(duì)象
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程序默認(rèn)等待你的輸入，所以程序會(huì)卡掉。

服務(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("進(jìn)入等待時(shí)間....")
  while True:
    print("收到連接....")
    # 接收數(shù)據(jù)
    data_server = conn.recv(1024)
    if not data_server:  # 這里判斷客戶端斷開的情況，不控制會(huì)無(wú)限循環(huán)
      print('client is lost...')
      break
    print('receive:', data_server.decode())
    # 返回一個(gè)值
    conn.send(data_server)

　　在Linux服務(wù)器上，我開啟了6個(gè)客戶端，每個(gè)客戶端輸入一次就斷開（即顯示了client has lost...），服務(wù)端分別和六個(gè)客戶端連接并接收數(shù)據(jù)。

四、實(shí)現(xiàn)ssh功能

4.1 測(cè)試環(huán)境

　　服務(wù)端： 172.16.200.49，監(jiān)聽端口（'0.0.0.0', 8888），Linux系統(tǒng)

　　客戶端：本機(jī)win10

4.2 測(cè)試代碼

　　服務(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("進(jìn)入等待時(shí)間....")
  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é)果存儲(chǔ)返回
    # 返回結(jié)果
    conn.send(res.encode('utf-8'))

　　客戶端：

# -*- coding: UTF-8 -*-
import socket
# 客戶端
# 聲明協(xié)議類型,同時(shí)生成socket對(duì)象
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ā)送太大，客戶端接收不完，就會(huì)先存儲(chǔ)在緩存當(dāng)中。但是下一次命令，客戶端接收的還是上次命令沒有發(fā)完的數(shù)據(jù)。

四、模擬FTP上傳文件

　　本例當(dāng)服務(wù)端在Linux環(huán)境中，struck模塊有些問(wèn)題....，目前在win10中可以正常運(yùn)行

　　服務(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()刪除打包時(shí)附加的多余空字符
          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):
  # 定義一個(gè)FtpClien類
  def __init__(self, host, port):
    self.host = host
    self.port = port
  def client_push(self):
    # 聲明協(xié)議類型,同時(shí)生成socket對(duì)象
    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通過(guò)socket實(shí)現(xiàn)多個(gè)連接并實(shí)現(xiàn)ssh功能詳解的全部?jī)?nèi)容，希望對(duì)大家有所幫助。感興趣的朋友可以繼續(xù)參閱本站：Python入門之三角函數(shù)全解【收藏】、Python基礎(chǔ)練習(xí)之用戶登錄實(shí)現(xiàn)代碼分享、python好玩的項(xiàng)目—色情圖片識(shí)別代碼分享等，有什么問(wèn)題可以隨時(shí)留言，小編會(huì)及時(shí)回復(fù)大家的。感謝朋友們對(duì)本站的支持！

您可能感興趣的文章: