# sock.py python中的网络编程-socket篇

# 要创建套接字，必须使用socket.socket(socket_family, socket_type, protocol=0)
# socket_family 地址家族 AF_UNIX或AF_INET 
# socket_type 套接字类型 SOCK_STREAM或SOCK_DGRAM
# protocol通常省略，默认为0

# 为了创建TCP/IP套接字
# tcpSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 为了创建UDP/IP套接字
# udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 套接字对象(内置)方法
# 服务器套接字方法
# s.bind()        将地址(主机名，端口号对)绑定到套接字上
# s.listen()    设置并启动TCP监听器
# s.accept()    被动接受TCP客户端链接，一直等待直到连接到达(阻塞)

# 客户端套接字方法
# s.connect()    主动发起TCP服务器连接
# s.connect_ex()扩展版本，以错误码形式返回问题，而不是抛出一个异常

# 普通的套接字方法
# s.recv()        接受TCP信息
# s.recv_into()    接受TCP信息到指定的缓冲区
# s.send()         发送TCP信息
# s.sendall()    完整地发送TCP信息

# 创建TCP服务器
# ss = socket()             创建服务器套接字
# ss.bind()                    套接字与地址绑定
# ss.listen()                监听连接
# inf_loop:                    服务器无限循环
#     cs = ss.accept()        接受客户端连接(返回客户端套接字)
#     comm_loop:                通信循环
#         cs.recv()/cs.send() 对话(接受/发送)
#     cs.close()                关闭客户端套接字
# ss.close()                关闭服务器套接字

# 创建TCP客户端
# cs = socket()                创建客户端套接字
# cs.connect()                尝试连接服务器
# comm_loop:                通信循环
#     cs.send()/cs.recv()        对话(发送/接受)
# cs.close()                关闭客户端套接字

# 创建UDP服务器
# ss = socket()                创建服务器套接字
# ss.bind()                    绑定服务器套接字
# inf_loop():                服务器无线循环
#     cs = ss.recvfrom()/ss.sendto()    # 关闭(接受/发送)
# ss.close()

# 创建UDP客户端
# cs = socket()                创建客户端套接字
# comm_loop:                通信循环
#     cs.sendto()/cs.recvfrom()    对话(发送/接受)
# cs.close()                关闭客户端套接字

# tcpServer.py TCP服务端
#!/usr/bin/env python

from socket import *
from time import ctime

HOST = ''              # 主机名
PORT = 31416        # 端口
BUFSIZ = 1024        # 缓冲大小
ADDR = (HOST, PORT) # 套接字地址(主机名，端口)

tcpSerSock = socket(AF_INET, SOCK_STREAM)  # 创建套接字
tcpSerSock.bind(ADDR)  # 套接字与地址绑定
tcpSerSock.listen(5)  # 监听

try:
    while True:
        print('waiting for connection...')
        tcpCliSock, addr = tcpSerSock.accept()  # 阻塞，等待连接，返回客户端套接字与套接字地址(主机名，端口)
        print('...connected from:', addr)

        while True:
            data = tcpCliSock.recv(BUFSIZ)
            if not data:
                break
            reply = '[%s] %s' % (ctime(), data.decode('utf-8'))
            tcpCliSock.send(bytes(reply, 'utf-8'))
        tcpCliSock.close()
    tcpSerSock.close()
except KeyboardInterrupt:
    print('Bye!')
  

# tcpClient.py TCP客户端
#!/usr/bin/env python

from socket import *

HOST = 'localhost' # or 'localhost'
PORT = 31416
BUFSIZ = 1024
ADDR = (HOST, PORT)

tcpCliSock = socket(AF_INET, SOCK_STREAM)
tcpCliSock.connect(ADDR)

while True:
    data = input('> ')
    if not data:
        break
    tcpCliSock.send(bytes(data, 'utf-8'))
    data = tcpCliSock.recv(BUFSIZ)
    if not data:
        break
    print(data.decode('utf-8'))

tcpCliSock.close()

# udpServer.py UDP服务端
#!/usr/bin/env python

from socket import *
from time import ctime

HOST = ''
PORT = 31416
BUFSIZ = 1024
ADDR = (HOST, PORT)

udpSerSock = socket(AF_INET, SOCK_DGRAM)
udpSerSock.bind(ADDR)

try:
    while True:
        print('waiting for massage...')
        data, addr = udpSerSock.recvfrom(BUFSIZ)
        reply = '[%s]: %s' % (ctime(), data.decode('utf-8'))
        udpSerSock.sendto(bytes(reply, 'utf-8'), addr)
        print('...received from and returned to:', addr)
except KeyboardInterrupt:
    print('Bye!')

udpSerSock.close()

# udpClient.py UDP客户端
#!/usr/bin/env python

from socket import *

HOST = 'localhost'
PORT = 31416
BUFSIZ = 1024
ADDR = (HOST, PORT)

udpCliSock = socket(AF_INET, SOCK_DGRAM)

while True:
    data = input('> ')
    if not data:
        break
    udpCliSock.sendto(bytes(data, 'utf-8'), ADDR)
    data, ADDR = udpCliSock.recvfrom(BUFSIZ)
    if not data:
        break
    print(data.decode('utf-8'))

udpCliSock.close()