socket使用

本节导读：

• 什么是socket
• socket通信讨论
• socket套接字方法
• socket服务端客户端的创建

一 什么是scoket

Socket是应用层与TCP/IP协议族通信的中间软件抽象层，它是一组接口。在设计模式中，Socket其实就是一个门面模式，它把复杂的TCP/IP协议族隐藏在Socket接口后面，对用户来说，一组简单的接口就是全部。

socket起源于Unix，而Unix/Linux 基本哲学之一就是“一切皆文件”，都可以用“打开open –> 读写write/read –> 关闭close”模式 来操作。Socket就是该模式的一个实现，socket即是一种特殊的文件，一些socket函数就是对其进行的操作（读/写IO、打开、关闭）

你想给另一台计算机发消息，你知道他的IP地址，他的机器上同时运行着qq、迅雷、word、浏览器等程序，你想给他的qq发消息，那想一下，你现在只能通过ip找到他的机器，但如果让这台机器知道把消息发给qq程序呢？答案就是通过port,一个机器上可以有0-65535个端口，你的程序想从网络上收发数据，就必须绑定一个端口，这样，远程发到这个端口上的数据，就全会转给这个程序

 

二 socket的工作流程

下面我们举个打电话的小例子来说明一下

如果你要给你的一个朋友打电话，先拨号，朋友听到电话铃声后提起电话，这时你和你的朋友就建立起了连接，就可以讲话了。等交流结束，挂断电话结束此次交谈。 生活中的场景就解释了这工作原理。

（如果你去一家餐馆吃饭，假设哪里的老板就是服务端，而你自己就是客户端，当你去吃饭的时候，你肯定的知道那个餐馆，也就是服务端的地址吧，但是对于你自己来说，餐馆的老板不需要知道你的地址吧）

 

三 socket套接字方法

socket参数

family(socket家族)

• socket.AF_UNIX：用于本机进程间通讯，为了保证程序安全，两个独立的程序(进程)间是不能互相访问彼此的内存的，但为了实现进程间的通讯，可以通过创建一个本地的socket来完成
• socket.AF_INET:(还有AF_INET6被用于ipv6，还有一些其他的地址家族，不过，他们要么是只用于某个平台，要么就是已经被废弃，或者是很少被使用，或者是根本没有实现，所有地址家族中，AF_INET是使用最广泛的一个，python支持很多种地址家族，但是由于我们只关心网络编程，所以大部分时候我么只使用AF_INET)

socket type类型

• socket.SOCK_STREAM #for tcp
• socket.SOCK_DGRAM #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 #废弃了

(Only SOCK_STREAM and SOCK_DGRAM appear to be generally useful.)

proto=0 请忽略，特殊用途

fileno=None 请忽略，特殊用途

 

socket函数

phone.bind('主机ip地址',端口号)  #绑定到（主机，端口号）套接字
phone.listen() #开始TCP监听
phone.accept() #被动接受TCP客户的连接，等待连接的到来

s.connect()  #主动初始化TCP服务器连接
s.connect_ex()   #connect()函数的扩展版本,出错时返回出错码,而不是抛出异常

s.recv()  #接收数据
s.send()# 发送数据(send在待发送数据量大于己端缓存区剩余空间时,数据丢失,不会发完，可后面通过实例解释)
s.sendall()  #发送完整的TCP数据(本质就是循环调用send,sendall在待发送数据量大于己端缓存区剩余空间时,数据不丢失,循环调用send直到发完)
s.recvfrom()  #Receive data from the socket. The return value is a pair (bytes, address)
s.getpeername()  #连接到当前套接字的远端的地址
s.sendto() #发送UDP数据
s.getsockname() #返回指定套接字的参数
s.setsockopt() #设置指定套接字的参数
s.close()  # 关闭套接字
socket.setblocking(flag) #True or False,设置socket为非阻塞模式，以后讲io异步时会用
socket.getaddrinfo(host, port, family=0, type=0, proto=0, flags=0)   #返回远程主机的地址信息，例子 socket.getaddrinfo('luffycity.com',80)
socket.getfqdn()# 拿到本机的主机名
socket.gethostbyname()  # 通过域名解析ip地址

phone.setblocking()  #设置套接字的阻塞与非阻塞模式
phone.settimeout()  #设置阻塞套接字操作的超时时间
phone.gettimeout()  #得到阻塞套接字操作的超时时间

phone.fileno()  # 套接字的文件描述符
phone.makefile() #创建一个与该套接字相关的文件

四 socket 服务端，客户端的创建

socket服务端创建

import socket
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)#买手机
phone.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) #可以多次启动
#执行多次的时候会报错，那么怎么办呢、？就在绑卡前面加上上面那句setsockopt方法就ok了
phone.bind(('192.168.20.44',8080))#绑定手机卡(ip,端口)
# 端口号在1024以前的是系统用的，1024以后的都是你自己写的程序去定义的端口

print('starting run......')
phone.listen(5) #开机   5代表的是最多挂起5个，也可以好多个
while True: #链接循环
    coon,client_addr=phone.accept()#等待接电话,（coon是建立的链接，客户端的ip和端口号组成的元组）
    print(coon,client_addr)

    #收发消息
    while True:  #通信循环
        try:  #如果不加try...except ,就会报错，因为它不知道你什么时候断开链接的，服务器还以为你在运行
            data = coon.recv(1024) #收了1024个字节的消息
            print('client data 收到消息:%s'%data.decode('utf-8'))
            coon.send(data.upper())  #发消息
        except Exception:  #因为你不知道客户端什么时候断开链接，
            break
    coon.close() #挂电话
phone.close() #关机


# 处理逻辑错误的两种方式：
    # if 判断
    # try...except 异常处理
# 异常处理
# 当你知道直接错误的条件时就用if判断了
# 当程序错误一定发生，但是你又预知不了它出错的条件是什么的时候，就用try...except

服务端

socket客户端创建

import socket
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)#买手机
phone.connect(('192.168.20.44',8080))  #直接连接服务端的ip和端口

# 发收消息
while True:
    msg = input('>>:').strip()  #用户输入
    if not msg:continue  #如果为空就继续输
    phone.send(msg.encode('utf-8'))  #  发送你输入的消息
    # phone.send('hello'.encode('utf-8'))
    data = phone.recv(1024)  #在接收一下
    print('server back res服务端返回结果:>>%s'%data.decode('utf-8'))

phone.close()

注意：

如果你在重启服务端的时候可能遇到这样的问题：

这个是由于你的服务端仍然存在四次挥手的time_wait状态在占用地址（如果不懂，请深入研究1.tcp三次握手，四次挥手 2.syn洪水攻击 3.服务器高并发情况下会有大量的time_wait状态的优化方法）。那么怎么解决呢？你也可以这样的

1 #加入一条socket配置，重用ip和端口
2 
3 phone=socket(AF_INET,SOCK_STREAM)
4 phone.setsockopt(SOL_SOCKET,SO_REUSEADDR,1) #就是它，在bind前加
5 phone.bind(('127.0.0.1',8080))

基于tcp协议模拟ssh远程执行命令

import socket
import subprocess
phone=socket.socket(socket.AF_INET,socket.SOCK_STREAM)#买手机
phone.bind(('192.168.20.44',8081))#绑定手机卡
phone.listen(5)#阻塞的最大个数
print('starting....')
while True:
    conn,addr=phone.accept()#等待连接
    print(addr,conn)
    while True:
        cmd=conn.recv(10240)#接收的最大值
        # if not cmd :break
        print('接收的是：%s'%cmd.decode('utf-8'))
        #处理过程
        res=subprocess.Popen(cmd.decode('utf-8'),shell=True,   #Popen是执行命令的方法
                             stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE )
        stdout=res.stdout.read()
        stuerr=res.stderr.read()
        conn.send(stdout+stuerr)
    conn.close()
phone.close()

基于udp协议的socket

from socket import *
udp_server = socket(AF_INET,SOCK_DGRAM)
udp_server.bind(('127.0.0.1',8080)) #绑定
while True:#通讯循环
    msg,client_addr= udp_server.recvfrom(1024)
    print('收到的消息是：%s'%msg.decode('utf-8'))
    udp_server.sendto(msg.upper(),client_addr)
udp_server.close()