python socket编程

 

什么是 Socket?

Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。
TCP可靠性的实现：
（1）校验码
（2）接收方反馈
（3）信息包附带序号
UDP：（1）快 不需要花费时间建立和关闭连接
（2）快 偶尔丢失一两个消息包无所谓，但是TCP会严格检查
（3）快 UDP的限制是一个信息包不超过64KB的数据
TCP和UDP区别就是UDP不建立连接，只保证数据的完整性，数据传输快，但是不保证数据是否真的被收到，也不保证数据是够只接收一次，也不保证次序。

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

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

 

socket()函数

Python 中，我们用 socket（）函数来创建套接字，语法格式如下：

socket.socket([family[, type[, proto]]])

参数

family: 套接字家族可以使AF_UNIX或者AF_INET
type: 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM
protocol: 一般不填默认为0.

 

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()是有用的。

 

简单例子

客户端向服务端发送一个消息。

服务端

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time  : 2018-6-1 17:52
# @Author: yangjian
# @File  : Client.py


'''
python3 socket收发消息都是只能使用bytes类型，而python2.7则可以使用str类型
'''

'''
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM,0)
参数一：地址簇
　　socket.AF_INET IPv4（默认）
　　socket.AF_INET6 IPv6
　　socket.AF_UNIX 只能够用于单一的Unix系统进程间通信
参数二：类型
　　socket.SOCK_STREAM　　流式socket , for TCP （默认）
　　socket.SOCK_DGRAM　　 数据报式socket , for UDP
    socket.SOCK_RAW 原始套接字，普通的套接字无法处理ICMP、IGMP等网络报文，而SOCK_RAW可以；其次，SOCK_RAW也可以处理
                    特殊的IPv4报文；此外，利用原始套接字，可以通过IP_HDRINCL套接字选项由用户构造IP头。
　　socket.SOCK_RDM 是一种可靠的UDP形式，即保证交付数据报但不保证顺序。SOCK_RAM用来提供对原始协议的低级访问，在需要
                    执行某些特殊操作时使用，如发送ICMP报文。SOCK_RAM通常仅限于高级用户或管理员运行的程序使用。
　　socket.SOCK_SEQPACKET 可靠的连续数据包服务
参数三：协议
　（默认）与特定的地址家族相关的协议,如果是 0 ，则系统就会根据地址格式和套接类别,自动选择一个合适的协议
'''

import socket

#创建socket对象
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM,0)  #创建socket对象


#绑定
ip_port = ('127.0.0.1',8087)  #定义ip、端口号
s.bind(ip_port)  #socket对象绑定ip、端口号（ip、端口号一定是元组形式）

s.getsockname()  #服务端使用，返回服务端自身的套接字地址（ip、端口号）

#监听，并设置最大挂起连接数
s.listen(5)  #监听，除了正在连接的客户端，最多还能能挂起5个连接，超过之后就拒绝连接


#获取客户端连接
conn, addr = s.accept()   #获取客户端的scoket对象conn和客户端的地址address(ip、端口号)


#收发消息
recv_data = conn.recv(1024)   #socket获取的客户端对象（conn）用来接收消息
print(str(recv_data,encoding='utf-8'))
send_data = '你好'
conn.send(bytes(send_data,encoding='utf-8'))   #socket获取的客户端对象（conn）用来发送消息-----python3只能发送字节类型（bytes），python2.7可以直接发送字符串

#关闭客户端对象
conn.close()

 

客户端

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time  : 2018-6-1 17:51
# @Author: yangjian
# @File  : Server.py

'''
python3 socket收发消息都是只能使用bytes类型，而python2.7则可以使用str类型
'''

import socket

#创建socket对象
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  #创建socket对象

s.settimeout(5)   #设置超时时间
#s.getpeername()  #客户端使用，返回服务端的套接字地址（ip、端口号）

#连接服务端
#ip_port = ('127.0.0.1',9999)  #定义ip、端口号
s.connect(('127.0.0.1',8087))  #socket对象连接服务端。（ip、端口号一定是元组形式） ----连接失败会报错
#s.connect_ex(ip_port)   #功能同上，不过如果连接失败会返回一个错误编码（不报错）


#发消息
send_data = '你好'
s.send(bytes(send_data,encoding='utf-8'))    #发送消息，成功返回发送的字节数-----python3只能发送字节类型（bytes），python2.7可以直接发送字符串
#s.sendall(bytes(send_data,encoding='utf-8'))   #发送所有数据，成功返回None，失败报错

#收消息
recv_data = s.recv(1024)


#关闭
s.close()

 

ssh交互

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time  : 2018-6-1 18:15
# @Author: yangjian
# @File  : ssh_test.py

import paramiko


client = paramiko.SSHClient()
# 允许链接不在know_host文件中的主机
client.set_missing_host_key_policy(paramiko.AutoAddPolicy)

client.connect('100.106.106.220', 22, username='root', password='123456', timeout=5)
while 1:
    cmd = input('[root@localhost ~]#')
    stdin, stdout, stderr = client.exec_command(cmd)
    for std in stdout.readlines():
        print(std)
client.close()

结果：
D:\pycharmproject\venv\Scripts\python.exe D:/pycharmproject/0531/ssh_test.py
[root@localhost ~]#ls -ll
total 8
-rw-r--r-- 1 root root 1361 May 31 10:11 yangjian.his
-rw-r--r-- 1 root root  924 May 31 10:11 yangjian.time