Python UDP socket 与 TCP socket

UDP socket 与 TCP socket

一、UDPsocket
UDP属于无连接协议，在编程时不需要建立连接，而是直接向接收方发送信息。UDP不提供应答重传机制，无法保证数据一定能够到达目的地。UDP编程经常用到socket模块方法如下所示：

1.socket(family[,type[,proto]])：创建一个Socket对象，family为socket.AF_INET表示使用IPV4，socket.AF_INET6表示使用IPV6；type为SOCK_STREAM表示使用TCP，SOCK_DGRAM表示使用UDP。
2.sendto(string,address)：把string指定的内容发送给address指定的地址，其中address是一个包含接收方主机IP地址和应用进程端口号的元组，格式为(IP地址，端口号)。
3.recvfrom(bufsize[,flags])：接收数据。

UDP协议优点：效率高、首部中只包含双方地址与校验和等很少的字段，额外开销小。

UDPServerSocket

import socket

s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
s.bind(('',8001))
while True:
    data,addr=s.recvfrom(1024)
    print(data.decode())
    if data.decode().lower()=='exit':
        break
s.close()

UDPClientSocket

import socket
s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
while True:
   s.sendto(input('输入：').encode(),('127.0.0.1',8001))
s.close()

或者

sys.argv[1].encode() 替换 input('输入：').encode()

sys.argv[1]是在命令行执行python文件时，在文件后面输入内容；
input(‘输入：’).encode()是执行文件后，进入文件再输入内容；

二、TCP socket
TCP一般用于可靠数据传输的场合， UDP编程经常用到socket模块方法如下所示：
1.socket(family[,type[,proto]])：创建一个Socket对象，family为socket.AF_INET表示使用IPV4，socket.AF_INET6表示使用IPV6；type为SOCK_STREAM表示使用TCP，SOCK_DGRAM表示使用UDP。
2.connect(address)：连接远程主机；
3.send(bytes[,flags])：发送数据；
4.recv(bufsize[,flags])：接收数据；
5.bind(address)：绑定地址；
6.listen(backlog)：开始监听，等待客户端连接，blacklog排队数，backlog+1表示允许的最大连接数；
7.accept()：响应客户端的请求，接收一个连接；

TCPserversocket

import socket
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind(('',8002))
s.listen(1)
conn,addr=s.accept()#接受TCP连接并返回（conn,addr）,其中conn是新的套接字对象，可以用来接收和发送数据。addr是连接客户端的地址。
print('连接',addr)
while True:
    print('接收：',conn.recv(1024).decode())
    c = input('输入：')
    conn.sendall(c.encode())
    if c.lower()=='bye':
        break
conn.close()
s.close()

TCPclientsocket

import socket
HOST = '127.0.0.1'
POST = 8002
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
    s.connect((HOST,POST))
except Exception as e:
    print('server not found!')
while True:
    c = input('输入:')
    s.sendall(c.encode())
    print('接收:',s.recv(1024).decode())
    if c.lower()=='bye':
        break
s.close()

三、socket基础知识拓展：
sendall是用在TCP的；sendto是用在UDP的；send直接调用系统的接口，不知道是否发送数据完成，一般不建议用。



四、Socket安全运维拓展
Gethostname()：获取本地主机名；
Gethostbyname()：根据主机名获取ip地址，或根据域名获取IP地址；
Gethostbyaddr()：根据ip地址获取主机名；只能获取本机的。
Getservbyport()：根据端口号获取服务名；
Getservbyname()：根据服务名获取对应的端口号；

import socket
hostname = socket.gethostname()
ip = socket.gethostbyname(hostname)
addr=socket.gethostbyname('www.xxxx.cn')
host = socket.gethostbyaddr(ip)
servername = socket.getservbyport(3389,'tcp')
port = socket.getservbyname('https')

print("主机名：",hostname)
print("IP地址：",ip)
print("IP地址：",addr)
print("主机名：",host)
print("服务名：",servername)
print("端口号：",port)