Skip to content

Commit 617e524

Browse files
committed
clean up windows socket attrs
1 parent 141ff2b commit 617e524

File tree

1 file changed

+16
-15
lines changed

1 file changed

+16
-15
lines changed

stdlib/src/socket.rs

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -50,17 +50,18 @@ mod _socket {
5050
#[pyattr(name = "has_ipv6")]
5151
const HAS_IPV6: bool = true;
5252
#[pyattr]
53-
const AF_UNSPEC: i32 = 0;
54-
#[pyattr]
55-
const SOL_TCP: u32 = 6;
56-
#[pyattr]
5753
use c::{
58-
AF_INET, AF_INET6, IPPROTO_IP, IPPROTO_IP as IPPROTO_IPIP, IPPROTO_IPV6, IPPROTO_TCP,
59-
IPPROTO_UDP, MSG_OOB, MSG_PEEK, MSG_WAITALL, NI_NAMEREQD, NI_NOFQDN, NI_NUMERICHOST,
60-
NI_NUMERICSERV, SHUT_RD, SHUT_RDWR, SHUT_WR, SOCK_DGRAM, SOCK_STREAM, SOL_SOCKET,
61-
SO_BROADCAST, SO_ERROR, SO_LINGER, SO_OOBINLINE, SO_REUSEADDR, SO_TYPE, TCP_NODELAY,
54+
AF_INET, AF_INET6, AF_UNSPEC, IPPROTO_IP, IPPROTO_IP as IPPROTO_IPIP, IPPROTO_IPV6,
55+
IPPROTO_TCP, IPPROTO_TCP as SOL_TCP, IPPROTO_UDP, MSG_OOB, MSG_PEEK, MSG_WAITALL,
56+
NI_NAMEREQD, NI_NOFQDN, NI_NUMERICHOST, NI_NUMERICSERV, SHUT_RD, SHUT_RDWR, SHUT_WR,
57+
SOCK_DGRAM, SOCK_STREAM, SOL_SOCKET, SO_BROADCAST, SO_ERROR, SO_LINGER, SO_OOBINLINE,
58+
SO_REUSEADDR, SO_TYPE, TCP_NODELAY,
6259
};
6360

61+
#[cfg(unix)]
62+
#[pyattr]
63+
use c::{AF_UNIX, SO_REUSEPORT};
64+
6465
#[cfg(not(target_os = "freebsd"))]
6566
#[pyattr]
6667
use c::{AI_ADDRCONFIG, AI_ALL, AI_NUMERICHOST, AI_NUMERICSERV, AI_PASSIVE};
@@ -69,6 +70,13 @@ mod _socket {
6970
#[pyattr]
7071
use c::{SOCK_RAW, SOCK_RDM, SOCK_SEQPACKET};
7172

73+
#[cfg(windows)]
74+
#[pyattr]
75+
use winapi::shared::ws2def::{
76+
IPPROTO_CBT, IPPROTO_ICLFXBM, IPPROTO_IGP, IPPROTO_L2TP, IPPROTO_PGM, IPPROTO_RDP,
77+
IPPROTO_SCTP, IPPROTO_ST,
78+
};
79+
7280
#[pyattr]
7381
fn error(vm: &VirtualMachine) -> PyTypeRef {
7482
vm.ctx.exceptions.os_error.clone()
@@ -120,13 +128,6 @@ mod _socket {
120128
.clone()
121129
}
122130

123-
#[cfg(windows)]
124-
#[pyattr]
125-
use c::{
126-
IPPROTO_CBT, IPPROTO_ICLFXBM, IPPROTO_IGP, IPPROTO_L2TP, IPPROTO_PGM, IPPROTO_RDP,
127-
IPPROTO_SCTP, IPPROTO_ST,
128-
};
129-
130131
#[pyfunction]
131132
fn htonl(x: u32) -> u32 {
132133
u32::to_be(x)

0 commit comments

Comments
 (0)