UDP 套接字是可以使用 connect 系統(tǒng)調(diào)用連接到指定的地址的。從此以后,這個(gè)套接字只會(huì)接收來(lái)自這個(gè)地址的數(shù)據(jù),而且可以使用 send 系統(tǒng)調(diào)用直接發(fā)數(shù)據(jù)而不用指定地址。可以再次調(diào)用 connect 來(lái)連接到別的地方。但是在 Python 里,一旦調(diào)用 connect 之后,就再也回不到最初的能夠接收從任意地址來(lái)的數(shù)據(jù)的狀態(tài)了!
這是 Python 的 API 限制,沒(méi)辦法給 connect 方法傳遞到 AF_UNSPEC 地址簇(在 C 代碼里寫死了的)。C 里邊就可以做到的(代碼來(lái)自這里):
int disconnect_udp_sock(int fd) {
struct sockaddr_in sin;
memset((char *)&sin, 0, sizeof(sin));
sin.sin_family = AF_UNSPEC;
return (connect(fd, (struct sockaddr *)&sin, sizeof(sin)));
}
不過(guò)既然是 Python 的限制,拿 ctypes 就可以繞過(guò)了嘛,有些麻煩就是了:
from ctypes import CDLL, create_string_buffer
def disconnect(sock):
libc = CDLL("libc.so.6")
buf = create_string_buffer(16) # sizeof struct sockaddr_in
libc.connect(sock.fileno(), buf, 16)
AF_UNSPEC 的值是 0,所以把一個(gè)和 struct sockaddr_in 一樣長(zhǎng)的全零緩沖區(qū)傳給 connect 就可以了 :-)