0
|
1 |
"""
|
|
2 |
Socket addresses
|
|
3 |
"""
|
|
4 |
|
|
5 |
cimport libc
|
|
6 |
|
|
7 |
cdef class sockaddr :
|
|
8 |
"""
|
|
9 |
A network-level socket address
|
|
10 |
|
|
11 |
>>> sockaddr().family
|
|
12 |
0
|
|
13 |
>>> sockaddr().port
|
|
14 |
Traceback (most recent call last):
|
|
15 |
...
|
|
16 |
NotImplementedError
|
|
17 |
>>> sockaddr().getnameinfo()
|
|
18 |
Traceback (most recent call last):
|
|
19 |
...
|
|
20 |
NotImplementedError
|
|
21 |
"""
|
|
22 |
|
|
23 |
# address family
|
|
24 |
cdef readonly libc.sa_family_t family
|
|
25 |
|
|
26 |
cdef void _init_family (self, libc.sa_family_t family=libc.AF_UNSPEC) :
|
|
27 |
self.family = family
|
|
28 |
|
|
29 |
cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
|
|
30 |
"""
|
|
31 |
Get the sockaddr pointer and sockaddr length for this address
|
|
32 |
"""
|
|
33 |
|
|
34 |
raise NotImplementedError()
|
|
35 |
|
|
36 |
def getnameinfo (self) :
|
|
37 |
"""
|
|
38 |
Returns a (host, serv) tuple for this address à la getnameinfo
|
|
39 |
"""
|
|
40 |
|
|
41 |
cdef libc.sockaddr *sa
|
|
42 |
cdef libc.socklen_t sa_len
|
|
43 |
|
|
44 |
# XXX: take as args?
|
|
45 |
cdef int flags = libc.NI_NUMERICHOST | libc.NI_NUMERICSERV
|
|
46 |
|
|
47 |
# get our abstract sockaddr
|
|
48 |
self._get_sockaddr(&sa, &sa_len)
|
|
49 |
|
|
50 |
# get nice text format
|
|
51 |
return libc.getnameinfo(sa, sa_len, flags)
|
|
52 |
|
|
53 |
property addr :
|
|
54 |
"""
|
|
55 |
The ASCII literal network address
|
|
56 |
"""
|
|
57 |
|
|
58 |
def __get__ (self) :
|
|
59 |
"""
|
|
60 |
Default implmentation using getnameinfo()
|
|
61 |
"""
|
|
62 |
|
|
63 |
addr, port = self.getnameinfo()
|
|
64 |
|
|
65 |
return addr
|
|
66 |
|
|
67 |
property port :
|
|
68 |
"""
|
|
69 |
The integer port number
|
|
70 |
"""
|
|
71 |
|
|
72 |
def __get__ (self) :
|
|
73 |
"""
|
|
74 |
Default implementation using getnameinfo() and int()
|
|
75 |
"""
|
|
76 |
|
|
77 |
addr, port = self.getnameinfo()
|
|
78 |
|
|
79 |
return int(port)
|
|
80 |
|
|
81 |
cdef class sockaddr_in (sockaddr) :
|
|
82 |
"""
|
|
83 |
AF_INET struct sockaddr_in
|
|
84 |
|
|
85 |
>>> sa = sockaddr_in("127.0.0.1", 80)
|
|
86 |
>>> sa.addr
|
|
87 |
'127.0.0.1'
|
|
88 |
>>> sa.port
|
|
89 |
80
|
|
90 |
>>> str(sa)
|
|
91 |
'127.0.0.1:80'
|
|
92 |
>>> str(sockaddr_in())
|
|
93 |
'0.0.0.0:0'
|
|
94 |
"""
|
|
95 |
|
|
96 |
# the struct sockaddr_in
|
|
97 |
cdef libc.sockaddr_in sockaddr
|
|
98 |
|
|
99 |
def __init__ (self, object addr=None, libc.in_port_t port=0) :
|
|
100 |
"""
|
|
101 |
Construct using given literal IPv4 address and TCP/UDP port
|
|
102 |
|
|
103 |
addr - IPv4 address, defaults to INADDR_ANY (0.0.0.0)
|
|
104 |
port - TCP/UDP port, defaults to 0 (ephemeral)
|
|
105 |
"""
|
|
106 |
|
|
107 |
# store our family
|
|
108 |
# XXX: this should be a class attribute...
|
|
109 |
self._init_family(libc.AF_INET)
|
|
110 |
|
|
111 |
# constant af
|
|
112 |
self.sockaddr.sin_family = self.family
|
|
113 |
|
|
114 |
# set the sin_port
|
|
115 |
self.sockaddr.sin_port = libc.htons(port)
|
|
116 |
|
|
117 |
if addr :
|
|
118 |
# set the sin_addr
|
|
119 |
# this automatically converts the addr from str -> char *
|
|
120 |
libc.inet_pton(self.family, addr, &self.sockaddr.sin_addr)
|
|
121 |
|
|
122 |
else :
|
|
123 |
# set as INADDR_ANY
|
|
124 |
self.sockaddr.sin_addr.s_addr = libc.INADDR_ANY
|
|
125 |
|
|
126 |
cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
|
|
127 |
if sa_ptr :
|
|
128 |
sa_ptr[0] = <libc.sockaddr *> &self.sockaddr
|
|
129 |
|
|
130 |
if sa_len :
|
|
131 |
sa_len[0] = sizeof(self.sockaddr)
|
|
132 |
|
|
133 |
return 0
|
|
134 |
|
|
135 |
property port :
|
|
136 |
"""
|
|
137 |
The integer port number
|
|
138 |
"""
|
|
139 |
|
|
140 |
def __get__ (self) :
|
|
141 |
return libc.ntohs(self.sockaddr.sin_port)
|
|
142 |
|
|
143 |
def __str__ (self) :
|
|
144 |
"""
|
|
145 |
Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string
|
|
146 |
"""
|
|
147 |
|
|
148 |
# format
|
|
149 |
return "%s:%s" % self.getnameinfo()
|
|
150 |
|
|
151 |
cdef class sockaddr_in6 (sockaddr) :
|
|
152 |
"""
|
|
153 |
AF_INET6 struct sockaddr_in6
|
|
154 |
|
|
155 |
>>> sa6 = sockaddr_in6("::1", 80)
|
|
156 |
>>> sa6.addr
|
|
157 |
'::1'
|
|
158 |
>>> sa6.port
|
|
159 |
80
|
|
160 |
>>> str(sa6)
|
|
161 |
'[::1]:80'
|
|
162 |
>>> str(sockaddr_in6())
|
|
163 |
'[::]:0'
|
|
164 |
"""
|
|
165 |
|
|
166 |
cdef libc.sockaddr_in6 sockaddr
|
|
167 |
|
|
168 |
def __init__ (self, object addr=None, libc.in_port_t port=0) :
|
|
169 |
"""
|
|
170 |
Construct using given literal IPv6 address and TCP/UDP port
|
|
171 |
|
|
172 |
addr - IPv6 address, defaults to in6addr_any (::)
|
|
173 |
port - TCP/UDP port, defaults to 0 (ephemeral)
|
|
174 |
"""
|
|
175 |
|
|
176 |
# store our family
|
|
177 |
# XXX: this should be a class attribute...
|
|
178 |
self._init_family(libc.AF_INET6)
|
|
179 |
|
|
180 |
# constant af
|
|
181 |
self.sockaddr.sin6_family = self.family
|
|
182 |
|
|
183 |
# set the sin_port
|
|
184 |
self.sockaddr.sin6_port = libc.htons(port)
|
|
185 |
|
|
186 |
if addr :
|
|
187 |
# set the sin_addr
|
|
188 |
# this automatically converts the addr from str -> char *
|
|
189 |
libc.inet_pton(self.family, addr, &self.sockaddr.sin6_addr)
|
|
190 |
|
|
191 |
else :
|
|
192 |
# set as INADDR_ANY
|
|
193 |
self.sockaddr.sin6_addr = libc.in6addr_any
|
|
194 |
|
|
195 |
cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
|
|
196 |
if sa_ptr :
|
|
197 |
sa_ptr[0] = <libc.sockaddr *> &self.sockaddr
|
|
198 |
|
|
199 |
if sa_len :
|
|
200 |
sa_len[0] = sizeof(self.sockaddr)
|
|
201 |
|
|
202 |
return 0
|
|
203 |
|
|
204 |
property port :
|
|
205 |
"""
|
|
206 |
The integer port number
|
|
207 |
"""
|
|
208 |
|
|
209 |
def __get__ (self) :
|
|
210 |
return libc.ntohs(self.sockaddr.sin6_port)
|
|
211 |
|
|
212 |
|
|
213 |
def __str__ (self) :
|
|
214 |
"""
|
|
215 |
Return the literal ASCII representation for this sockaddr as a '[<addr>]:<port> string
|
|
216 |
"""
|
|
217 |
|
|
218 |
# format
|
|
219 |
return "[%s]:%s" % self.getnameinfo()
|
|
220 |
|