socket模块

Posted 2020-11-10 星空的衣角

1、TCP和UDP

TCP可靠性的实现：

• 校验码
• 接收方反馈
• 信息包附带序号

UDP：

• 快 不需要花费时间建立和关闭连接
• 快 偶尔丢失一两个消息包无所谓，但是TCP会严格检查
• 快 UDP的限制是一个信息包不超过64KB的数据

TCP和UDP区别就是UDP不建立连接，只保证数据的完整性，数据传输快，但是不保证数据是否真的被收到，也不保证数据是够只接收一次，也不保证次序。

服务端是用来给一个或者多个客户端提供服务的，当客户端发起请求，开始等待服务端的返回结果，服务端接受完请求以后，根据自己的逻辑进行处理请求，并返回给客户端，客户端接收到返回结果以后，关闭和服务端的连接。

最常用个客户端和服务端有两种模式C/S（mysql）模式和B/S模式（网站）

实现图：

2、socket常用函数

①、创建套接字
s = socket.socket(address_family, socket_type)
address family：
socket.AF_INET           默认ipv4
Socket.AF_INET             ipv6
Socket.AF_UNIX       只能用于单一的unix系统进行间通信
socket type：
socket.SOCK_STREAM    流式socket，TCP
socket.SOCK_DGRAM    数据报是socket   UDP

TCP:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 绑定地址：   bind(address)
address = (‘0.0.0.0’, 8009)
s.bind(address) 或者s.bind((‘0.0.0.0’, 8009))
address 必须是一个元组，容易错误，address = （host，port）
host：服务端ip，字符串类型， 如果为0.0.0.0  代表本机的任意一个ip
port：服务端提供的端口， 整型， 0-1024为系统保留

监听消息：  
s.listen(badklog)
backlog代表可以同时接受多少个socket连接

接受连接：
conn, addr = s.accept()
接受连接并返回元组(conn, addr), 其中conn是新的套接字对象，每一个新的连接就创建一个新的对象。可以用来接受和发送数据，addr是客户端的地址。包含host和port

发送数据:
s.send(string)             发送字符串到连接的套接字，可能未将指定内容全部发送
s.sendall(string)      内部递归调用send，将所有内容发送出去，建议使用。

接收数据：
data = s.recv(bufsize)
接收套接字数据，数据以字符串形式返回，bufsize指定最多接收的数据量，可以使用1024， 2048
如果不知道接收的数量有多少，可能几个字节，可能几M，一般通过循环接收

UDP:
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.sendto(string)
data, address = s.recvfrom(bufsize)

客户端：
客户端首先也要创建socket套接字
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
客户端连接服务端函数：
s.connect(address)                连接到address的套接字
result = connect_ex(address)        成功返回0，失败返回错误码（推荐使用）

通用：
s.close()                关闭socket套接字
s.getsocketname()                   获取套接字的名字
s.settimeout(timeout)        设置套接字超时时间，timeout为float类型，单位为秒。
s.gettimeout()             获得套接字超时时间

通用：
s.setblocking(flag)
flage为bool值
setblocking(True) is equivalent to settimeout(None); 相当于不设置超时时间，一直阻塞在那里
setblocking(False) is equivalent to settimeout(0.0). 相当于设置超时时间为0, 如果设置False，那么accept和recv时一旦无数据，则报错。

s.fileno()
返回套接字的文件描述符（一个小整数）。这对于select.select()是有用的。

3、实现ssh链接功能

server端01：

import socket
import paramiko
from threading import Thread
import sys


class SSH(Thread):
    def __init__(self, ip, port, user, password):
        # super().__init__()
        Thread.__init__(self)
        self.ip = ip
        self.prot = port
        self.user = user
        self.password = password

    def ssh(self, cmd):
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        try:
            ssh.connect(hostname=self.ip, username=self.user, password=self.password, timeout=1)
        except:
            print("Timeout or not parmission")
            return 1
        stdin, stdout, stderr = ssh.exec_command(cmd)  # 输出会分别保存到三个文件里
        stdout = stdout.read()  # [:-1]指的是去掉最后一个字符，因为这个最后输出会有个换行符
        stderr = stderr.read()[:-1]
        ssh.close()
        if stdout:
            return stdout.decode(‘utf-8‘) or "ok"
        else:
            return stderr

    def run(self):
        HOST = ‘‘
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.bind((HOST, self.prot))
        s.listen(1)
        conn, addr = s.accept()
        print("Connected by: {0}".format(addr))
        while True:
            try:
                data = conn.recv(1024)
                if not data:
                    break
                conn.sendall(self.ssh(data))
            except:
                print("close connection")
                break
        s.close()

if __name__ == ‘__main__‘:
    ip = ‘127.0.0.1‘
    prot = 1234
    user = ‘root‘
    password = ‘123456‘
    ssh = SSH(ip, prot, user, password)
    ssh.start()

server端02：

import socket
import commands

HOST = ‘‘                 # Symbolic name meaning all available interfaces
PORT = 1234              # Arbitrary non-privileged port
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((HOST, PORT))
s.listen(1)
conn, addr = s.accept()
print(‘Connected by‘, addr)
while True:
    data = conn.recv(1024)
    if not data:
        break
    result = commands.getoutput(data)
    conn.sendall(result)
s.close()

client端：

import socket
import sys

HOST = ‘192.168.137.130‘    # The remote host
PORT = 1234              # The same port as used by the server
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.connect((HOST, PORT))
    while 1:
        world = input("[[email protected] socket]#")
        if world:
            s.sendall(world.encode(‘utf-8‘))
            data = s.recv(1024)
            if data:
                # print(‘Received‘, repr(data.decode(‘utf-8‘)).upper())
                print("{0}".format(data.decode(‘utf-8‘)))

        else:
            print("input nothing!")
            sys.exit(1)