通過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文件位置處能看到上傳的文件