首页 > 代码库 > python运维开发之第八天（socket）

python运维开发之第八天（socket）

2024-08-13 05:20:22   219人阅读

什么是 Socket?

Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。

socket()函数

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

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

参数

  • family: 套接字家族可以使AF_UNIX或者AF_INET 地址簇
  • type: 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM　即tcp/ip和udp　
  • protocol: 一般不填默认为0.

Socket 对象(内建)方法

函数描述
服务器端套接字
s.bind()绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
s.listen()开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
s.accept()被动接受TCP客户端连接,(阻塞式)等待连接的到来
客户端套接字
s.connect()主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
s.connect_ex()connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
公共用途的套接字函数
s.recv()接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
s.send()发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
s.sendall()完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
s.recvform()接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
s.sendto()发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
s.close()关闭套接字
s.getpeername()返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
s.getsockname()返回套接字自己的地址。通常是一个元组(ipaddr,port)
s.setsockopt(level,optname,value)设置给定套接字选项的值。
s.getsockopt(level,optname[.buflen])返回套接字选项的值。
s.settimeout(timeout)设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
s.gettimeout()返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。
s.fileno()返回套接字的文件描述符。
s.setblocking(flag)如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。
s.makefile()创建一个与该套接字相关连的文件

简单实例

服务端

我们使用 socket 模块的 socket 函数来创建一个 socket 对象。socket 对象可以通过调用其他函数来设置一个 socket 服务。

现在我们可以通过调用 bind(hostname, port) 函数来指定服务的 port(端口)。

接着，我们调用 socket 对象的 accept 方法。该方法等待客户端的连接，并返回 connection 对象，表示已连接到客户端。

完整代码如下：

#!/usr/bin/env python# -*- coding: utf-8 -*-# @Author  : Willpower-chen# @blog: http://www.cnblogs.com/willpower-chen/import socket,os,time    #导入socket和os模块#实例化，AF_INET地址簇ipv4协议，SOCK_STREAM这是TCP/IP协议server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)server.bind((‘0.0.0.0‘,9999))    #绑定地址及服务端口server.listen(5)                        #限制连接数while True:                        #当一个client端断开时，接收下一个客户端    # 每个客户端发送过来的数据，生成相应的实例conn，client端连接地址addr    conn,addr = server.accept()    print(‘新的连接：‘,addr)    while True:        print(‘等待新的指令‘)        data = conn.recv(1024)    #接收client端发来的数据        if not data:            print(‘客户端已断开‘)            break        #当数据为空时，跳出循环等待下一个连接        print(‘执行指令‘,data.decode())    #打印要执行的命令        cmd_res = os.popen(data.decode()).read()#返回client发过来的cmd命令，在server端的执行结果        #由于每次接收数据的大小为1024，为了确定client需要接收几次，所以需要先把总数据大小发给client        print(‘开始发送‘,len(cmd_res))        if len(cmd_res) == 0 :     #如果cmd执行结果为空时，server端不会发送数据            cmd_res = ‘cmd_res is not info‘    #为了防止为空时不发数据，自定义一个数据        print(‘执行命令结果大小：‘,len(cmd_res))#打印命令执行结果数据大小        conn.send(str(len(cmd_res.encode(‘utf-8‘))).encode(‘utf-8‘))    #给client发送数据总大小信息        #time.sleep(0.5)        client_ack = conn.recv(1024)    #等待client端确认ack，防止粘包        print(‘ack from client‘,client_ack.decode(‘utf-8‘))    #打印ack确认信息        conn.send(cmd_res.encode(‘utf-8‘))    #给client发送命令执行结果数据        print(‘发送完毕‘)server.close()

客户端

接下来我们写一个简单的客户端实例连接到以上创建的服务。端口号为 9999。

socket.connect(hosname, port ) 方法打开一个 TCP 连接到主机为 hostname 端口为 port 的服务商。连接后我们就可以从服务端后期数据，记住，操作完成后需要关闭连接。

完整代码如下：

#!/usr/bin/env python# -*- coding: utf-8 -*-# @Author  : Willpower-chen# @blog: http://www.cnblogs.com/willpower-chen/import socketclient = socket.socket() #默认地址簇是ipv4，传输协议是TCP/IPclient.connect((‘localhost‘,9999))#与server建立连接while True:    cmd = input(‘>>:‘).strip()  # 用户输入cmd命令    if len(cmd) == 0: continue #如果用户输入为空，循环输入    client.send(cmd.encode(‘utf-8‘))#发送cmd命令，encode是把字符串转为byte(python3.0传输只能是二进制)    cmd_res_size = client.recv(1024)#确认接收命令结果的长度    print(‘命令结果大小‘,cmd_res_size.decode())    #打印命令结果的长度    client.send(‘准备好接收了，loser可以发了‘.encode(‘utf-8‘))#ack确认，防止粘包，注意这里不能用b‘‘因为加中文了    receive_size = 0    #已经接收的数据，默认先设置为0    receive_data = ‘‘.encode()    while receive_size < int(cmd_res_size.decode()): #已接收数据小于总数据时，循环接收数据        data = client.recv(1024)        receive_size += len(data) #每次收到的有可能小于1024，所以用len（data）计算        receive_data += data    else:        print(‘cmd res receive done...‘,receive_size)        print(receive_data.decode(‘utf-8‘))client.close()

Python Internet 模块

以下列出了 Python 网络编程的一些重要模块：

协议功能用处端口号Python 模块
HTTP网页访问80httplib, urllib, xmlrpclib
NNTP阅读和张贴新闻文章，俗称为"帖子"119nntplib
FTP文件传输20ftplib, urllib
SMTP发送邮件25smtplib
POP3接收邮件110poplib
IMAP4获取邮件143imaplib
Telnet命令行23telnetlib
Gopher信息查找70gopherlib, urllib


参考网址：http://www.runoob.com/python/python-socket.html

 

python运维开发之第八天（socket）


联系
我们