socket编程初学

socket相关方法和参数介绍

sk.bind(address)

　　#s.bind(address) 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址。

sk.listen(backlog)

　　#开始监听传入连接。backlog指定在拒绝连接之前，可以挂起的最大连接数量。

      #backlog等于5，表示内核已经接到了连接请求，但服务器还没有调用accept进行处理的连接个数最大为5
      #这个值不能无限大，因为要在内核中维护连接队列

sk.setblocking(bool)

　　#是否阻塞（默认True），如果设置False，那么accept和recv时一旦无数据，则报错。

sk.accept()

　　#接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。

　　#接收TCP 客户的连接（阻塞式）等待连接的到来

sk.connect(address)

　　#连接到address处的套接字。一般，address的格式为元组（hostname,port）,如果连接出错，返回socket.error错误。

sk.connect_ex(address)

　　#同上，只不过会有返回值，连接成功时返回 0 ，连接失败时候返回编码，例如：10061

sk.close()

　　#关闭套接字

sk.recv(bufsize[,flag])

　　#接受套接字的数据。数据以字符串形式返回，bufsize指定最多可以接收的数量。flag提供有关消息的其他信息，通常可以忽略。

sk.recvfrom(bufsize[.flag])

　　#与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。

sk.send(string[,flag])

　　#将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。即：可能未将指定内容全部发送。

sk.sendall(string[,flag])

　　#将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。

      #内部通过递归调用send，将所有内容发送出去。

sk.sendto(string[,flag],address)

　　#将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。

sk.settimeout(timeout)

　　#设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如 client 连接最多等待5s ）

sk.getpeername()

　　#返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。

sk.getsockname()

　　#返回套接字自己的地址。通常是一个元组(ipaddr,port)

sk.fileno()

　　#套接字的文件描述符

例子：实现相互发消息

server端

import socker

sk = socket.socket()
addr = ('127.0.0.1',9090)
sk.bind(addr)
sk.listen(3)

while True:
    conn,addr = sk.accept()
    print(addr)
    while True:
        try: #捕捉客户端异常关闭（ctrl+c）
            data = conn.recv(1024)
        except Exception:
            break
        if not data:  #客户端如果退出，服务端将收到空消息，退出
            #conn.close()
            break
        print(str(data,'utf8'))
        inp = input('>>>>')
        conn.send(bytes(inp,'utf8'))




client端

import socket

sk = socket.socket()
addr = ('127.0.0.1',9090)
sk.connect(addr)
while True:
    inp = input('>>>')
    if inp == 'q':
        break
    sk.send(bytes(inp,'utf8'))
    data = sk.recv(1024)
    print(str(data,'utf8'))
sk.close()

例子：远程执行命令获取命令结果

server端

import socket
import subprocess

#远程执行命令返回结果
sk = socket.socket()
addr = ('127.0.0.1',9090)
sk.bind(addr)
sk.listen(3)

while True:
    conn,addr = sk.accept()
    print(addr)
    while True:
        try: #捕捉客户端异常关闭（ctrl+c）
            data = conn.recv(1024)
        except Exception:
            break
        if not data:  #客户端如果退出，服务端将收到空消息，退出
            #conn.close()
            break
        print('执行命令为:',str(data,'utf8'))
        obj=subprocess.Popen(str(data,'utf8'),shell=True,stdout=subprocess.PIPE)
        cmd_result=obj.stdout.read()
        result_len=len(cmd_result)#结果的长度客户端用来判断发送的是否全部接收
        result_len=bytes(str(result_len),'utf8')
        print(result_len)
        #print(cmd_result)
        conn.sendall(result_len)
        conn.sendall(cmd_result)


client端

import socket

#远程执行命令返回结果
sk = socket.socket()
addr = ('127.0.0.1',9090)
sk.connect(addr)
while True:
    inp = input('>>>')
    if inp == 'q':
        break
    sk.send(bytes(inp,'utf8'))
    res_len=sk.recv(1024)#命令结果的长度客户端用来判断发送的是否全部接收
    print(res_len)
    data=bytes()
    while len(data)!=int(str(res_len,'utf8')):
        recv=sk.recv(1024)
        data+=recv
    print(str(data,'gbk'))
sk.close()

例子：上传文件 

server端

import socket
import re,os,sys

sk = socket.socket()
addr = ('127.0.0.1',9091)
sk.bind(addr)
sk.listen(3)

BASE_DIR=os.path.dirname(os.path.abspath(__file__))

while True:
    conn,addr = sk.accept()
    print(addr)
    while True:
        info = conn.recv(1024)
        file_info = str(info,'utf8')
        cmd,file_name,file_size=file_info.split('|')
        path=os.path.join(BASE_DIR,'pic',file_name)
        f = open(path, "wb")
        file_size=int(file_size)
        has_recv =0
        while has_recv != file_size: #判断文件是否接收完
            data=conn.recv(1024)
            f.write(data)
            has_recv+=len(data)
        f.close()
        print('写入成功')


client端

import socket
import re,os,sys

sk=socket.socket()
addr=('127.0.0.1',9091)
sk.connect(addr)

BASE_DIR=os.path.dirname(os.path.abspath(__file__))

while True:
    inp = input('>>>>:').strip() #put|111.png
    cmd,path = inp.split("|")
    path = os.path.join(BASE_DIR,path)
    file_name = os.path.basename(path) #获取文件名字
    file_size = os.stat(path).st_size #获取文件大小
    file_info = '%s|%s|%s' %(cmd,file_name,file_size) #发送文件的属性信息
    sk.sendall(bytes(file_info,'utf8'))

    f=open(path,'rb')
    has_sent=0
    while has_sent !=file_size: #判断文件是否发送完
        data = f.read(1024)
        #print(type(data))
        sk.sendall(data)
        has_sent+=len(data)
    f.close()
    print('上传成功')