python 之socket

socket 编程的基本流程

socket编程的基本参数和相关函数

基本参数

Socket函数使用的格式为：socket(family,type[,protocol])

• 参数一：family 指定应用程序使用的通信协议的协议族，对于TCP/IP协议族，该参数为AF_INET，以下为该参数的一些常用选项

Family参数        
socket.AF_UNIX      #只能够用于单一的Unix系统进程间通信
socket.AF_INET      #服务器之间网络通信
socket.AF_INET6     #IPv6

• 参数二：type 是要创建套接字的类型

Type参数
socket.SOCK_STREAM      #流式socket , 当使用TCP时选择此参数
socket.SOCK_DGRAM       #数据报式socket ,当使用UDP时选择此参数
socket.SOCK_RAW
#原始套接字，普通的套接字无法处理ICMP、IGMP等网络报文，而SOCK_RAW可以；其次，SOCK_RAW也可以处理特殊的IPv4报文；此外，利用原始套接字，可以通过IP_HDRINCL套接字选项由用户构造IP头。

• 参数三：protocol 指明所要接收的协议类型，通常为0或者不填。

Type参数
socket.IPPROTO_RAW
#相当于protocol=255，此时socket只能用来发送IP包，而不能接收任何的数据。发送的数据需要自己填充IP包头，并且自己计算校验和
socket.IPPROTO_IP
#相当于protocol=0，此时用于接收任何的IP数据包。其中的校验和和协议分析由程序自己完成。
使用例子：
   创建TCP Socket：s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
   创建UDP Socket：s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

socket相关的函数

• s.bind(address)
  将套接字绑定到地址, 在AF_INET下,以元组（host,port）的形式表示地址.
  服务端函数
• s.listen(backlog)
  开始监听TCP传入连接。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
• s.accept()
  接受TCP连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。
• s.connect(address)
  连接到address处的套接字。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
  客户端函数
• s.connect_ex(adddress)
  功能与connect(address)相同，但是成功返回0，失败返回error的值。
• s.recv(bufsize[,flag])
  接受TCP套接字的数据。数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
  公共函数
• s.send(string[,flag])
  发送TCP数据。将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
• s.sendall(string[,flag])
  完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
• s.recvfrom(bufsize[.flag])
  接受UDP套接字的数据。与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
• s.sendto(string[,flag],address)
  发送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()
  创建一个与该套接字相关连的文件

一个简单的sockket编程实例(默认TCP)

#server.py文件
import socket
sk = socket.socket()
IpPort = ('127.0.0.1',8080)
sk.bind(IpPort)
sk.listen(5)
while True:
    #conn,address= sk.accept()
    result = sk.accept()
    #这里就相当于：conn = result[0],address = result[1]
    print result
    print type(result)
#    conn.send('hello')
#    conn.close()

#client.py文件
#!/usr/bin/env python
#coding:utf-8
import socket
client = socket.socket()
IpPort = ('127.0.0.1',8080)
client.connect(IpPort)
data = client.recv(1024)
print data
#通过上面我们返回的结果我们可以看到返回是一个元组类型的：

#server.py输出的结果：
(<socket._socketobject object at 0x0000000002B43AD8>, ('127.0.0.1', 57326))
<type 'tuple'>

更新上面的程序，使其有更好的交互性

#client.py
#!/usr/bin/env python
#coding:utf-8
import socket
client = socket.socket()
IpPort = ('127.0.0.1',8080)
client.connect(IpPort)
while True:
    data = client.recv(1024)
    print data
    inp = raw_input('client:')
    if inp == 'exit':
        break
    client.send(inp)

#server.py
#!/usr/bin/env python
#coding:utf-8
import socket
sk = socket.socket()
IpPort = ('127.0.0.1',8080)
sk.bind(IpPort)
sk.listen(5)
while True:
    conn,address= sk.accept()
    conn.send('hello')
    flag = True
    while True:
        data = conn.recv(1024)
        conn.send('sb')
        if data == 'exit':
            flag = False
        print data
    conn.close()

使用SocketServer实现异步多线程服务端

#socket_server.py文件
#!/usr/bin/env python
#coding:utf-8
import SocketServer
class MyServer(SocketServer.BaseRequestHandler):
    def setup(self):
        pass
    def finish(self):
        pass
    def handle(self):
        #print self.request,self.client_address,self.server
        conn = self.request
        conn.send('hello')
        flag = True
        while flag:
            data = conn.recv(1024)
            print data
            if data == 'exit':
                flag = False
            conn.send('sb')
        conn.close()  
if __name__ == '__main__':
    server = SocketServer.ThreadingTCPServer(('127.0.0.1',9999),MyServer)
    server.serve_forever()

#socket_client.py文件
#!/usr/bin/env python
#coding:utf-8
import socket
client = socket.socket()
IpPort = ('127.0.0.1',8080)
client.connect(IpPort)
while True:
    data = client.recv(1024)
    print data
    inp = raw_input('client')
    if inp == 'exit':
        break
    client.send(inp)