author | Tero Marttila <terom@fixme.fi> |
Mon, 17 Aug 2009 20:24:12 +0300 | |
changeset 22 | f6e8d5e37998 |
parent 20 | 0e4933d5862e |
child 23 | 15d8bb96b8d4 |
permissions | -rw-r--r-- |
11
7ae92c2b433f
move libc+py back to qmsk.net.x
Tero Marttila <terom@fixme.fi>
parents:
10
diff
changeset
|
1 |
cimport qmsk.net.libc as libc |
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
2 |
cimport qmsk.net.py as py |
6 | 3 |
|
20
0e4933d5862e
rename qmsk.net.socket.addr -> qmsk.net.socket.address
Tero Marttila <terom@fixme.fi>
parents:
17
diff
changeset
|
4 |
from qmsk.net.socket.address cimport * |
10 | 5 |
|
6 |
cimport qmsk.net.socket.platform as platform |
|
6 | 7 |
|
8 |
cdef class sockaddr : |
|
10 | 9 |
cdef void _init_family (self, platform.sa_family_t family=platform.AF_UNSPEC) : |
6 | 10 |
self.family = family |
11 |
||
12 |
# XXX:use size_t |
|
10 | 13 |
cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 : |
6 | 14 |
""" |
15 |
Get the sockaddr pointer and sockaddr length for this address |
|
16 |
""" |
|
17 |
||
18 |
raise NotImplementedError() |
|
19 |
||
10 | 20 |
cdef platform.sockaddr* _get_sockaddr_ptr (self) except NULL : |
6 | 21 |
""" |
22 |
Get the sockaddr pointer |
|
23 |
""" |
|
24 |
||
10 | 25 |
cdef platform.sockaddr *sa |
26 |
cdef platform.socklen_t sa_len |
|
6 | 27 |
|
28 |
self._get_sockaddr(&sa, &sa_len) |
|
29 |
||
30 |
return sa |
|
31 |
||
10 | 32 |
cdef platform.socklen_t _get_sockaddr_len (self) except -1 : |
6 | 33 |
""" |
34 |
Get the sockaddr len |
|
35 |
""" |
|
36 |
||
10 | 37 |
cdef platform.sockaddr *sa |
38 |
cdef platform.socklen_t sa_len |
|
6 | 39 |
|
40 |
self._get_sockaddr(&sa, &sa_len) |
|
41 |
||
42 |
return sa_len |
|
43 |
||
10 | 44 |
cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 : |
6 | 45 |
""" |
46 |
Set the sockaddr value for this address; sa_len must match! |
|
47 |
""" |
|
48 |
||
49 |
raise NotImplementedError() |
|
50 |
||
51 |
def getnameinfo (self) : |
|
52 |
""" |
|
53 |
Returns a (host, serv) tuple for this address à la getnameinfo |
|
54 |
""" |
|
55 |
||
10 | 56 |
cdef platform.sockaddr *sa |
57 |
cdef platform.socklen_t sa_len |
|
6 | 58 |
|
59 |
# XXX: take as args? |
|
10 | 60 |
cdef int flags = platform.NI_NUMERICHOST | platform.NI_NUMERICSERV |
6 | 61 |
|
62 |
# get our abstract sockaddr |
|
63 |
self._get_sockaddr(&sa, &sa_len) |
|
64 |
||
65 |
# get nice text format |
|
10 | 66 |
return platform.getnameinfo(sa, sa_len, flags) |
6 | 67 |
|
68 |
property addr : |
|
69 |
""" |
|
70 |
The ASCII literal network address |
|
71 |
""" |
|
72 |
||
73 |
def __get__ (self) : |
|
74 |
""" |
|
75 |
Default implmentation using getnameinfo() |
|
76 |
""" |
|
77 |
||
78 |
addr, port = self.getnameinfo() |
|
79 |
||
80 |
return addr |
|
81 |
||
82 |
property port : |
|
83 |
""" |
|
84 |
The integer port number |
|
85 |
""" |
|
86 |
||
87 |
def __get__ (self) : |
|
88 |
""" |
|
89 |
Default implementation using getnameinfo() and int() |
|
90 |
""" |
|
91 |
||
92 |
addr, port = self.getnameinfo() |
|
93 |
||
94 |
return int(port) |
|
17
5f4077a530b0
add repr for sockaddr, and implment read for sockets, plus a recvmsg wrapper called readv...
Tero Marttila <terom@fixme.fi>
parents:
11
diff
changeset
|
95 |
|
5f4077a530b0
add repr for sockaddr, and implment read for sockets, plus a recvmsg wrapper called readv...
Tero Marttila <terom@fixme.fi>
parents:
11
diff
changeset
|
96 |
def __repr__ (self) : |
5f4077a530b0
add repr for sockaddr, and implment read for sockets, plus a recvmsg wrapper called readv...
Tero Marttila <terom@fixme.fi>
parents:
11
diff
changeset
|
97 |
return "sockaddr(%d, %s, %d)" % (self.family, self.addr, self.port) |
6 | 98 |
|
99 |
cdef class sockaddr_in (sockaddr) : |
|
100 |
""" |
|
101 |
AF_INET struct sockaddr_in |
|
102 |
||
103 |
>>> sa = sockaddr_in("127.0.0.1", 80) |
|
104 |
>>> sa.addr |
|
105 |
'127.0.0.1' |
|
106 |
>>> sa.port |
|
107 |
80 |
|
108 |
>>> str(sa) |
|
109 |
'127.0.0.1:80' |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
110 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
111 |
>>> sockaddr_in('2001::5') |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
112 |
Traceback (most recent call last): |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
113 |
... |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
114 |
NameError: Invalid network address for specified address family: '2001::5' |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
115 |
|
6 | 116 |
""" |
117 |
||
118 |
# the struct sockaddr_in |
|
10 | 119 |
cdef platform.sockaddr_in sockaddr |
6 | 120 |
|
10 | 121 |
def __init__ (self, object addr=None, platform.in_port_t port=0) : |
6 | 122 |
""" |
123 |
Construct using given literal IPv4 address and TCP/UDP port |
|
124 |
||
125 |
addr - IPv4 address, defaults to INADDR_ANY (0.0.0.0) |
|
126 |
port - TCP/UDP port, defaults to 0 (ephemeral) |
|
127 |
""" |
|
128 |
||
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
129 |
# zero |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
130 |
libc.memset(&self.sockaddr, 0, sizeof(self.sockaddr)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
131 |
|
6 | 132 |
# store our family |
133 |
# XXX: this should be a class attribute... |
|
10 | 134 |
self._init_family(platform.AF_INET) |
6 | 135 |
|
136 |
# constant af |
|
137 |
self.sockaddr.sin_family = self.family |
|
138 |
||
139 |
# set the sin_port |
|
10 | 140 |
self.sockaddr.sin_port = platform.htons(port) |
6 | 141 |
|
142 |
if addr : |
|
143 |
# set the sin_addr |
|
144 |
# this automatically converts the addr from str -> char * |
|
10 | 145 |
platform.inet_pton(self.family, addr, &self.sockaddr.sin_addr) |
6 | 146 |
|
147 |
else : |
|
148 |
# set as INADDR_ANY |
|
10 | 149 |
self.sockaddr.sin_addr.s_addr = platform.INADDR_ANY |
6 | 150 |
|
10 | 151 |
cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 : |
6 | 152 |
if sa_ptr : |
10 | 153 |
sa_ptr[0] = <platform.sockaddr *> &self.sockaddr |
6 | 154 |
|
155 |
if sa_len : |
|
156 |
sa_len[0] = sizeof(self.sockaddr) |
|
157 |
||
158 |
return 0 |
|
159 |
||
10 | 160 |
cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 : |
6 | 161 |
assert sa_len == sizeof(self.sockaddr) |
162 |
||
163 |
libc.memcpy(&self.sockaddr, sa, sa_len) |
|
164 |
||
165 |
property port : |
|
166 |
""" |
|
167 |
The integer port number |
|
168 |
""" |
|
169 |
||
170 |
def __get__ (self) : |
|
10 | 171 |
return platform.ntohs(self.sockaddr.sin_port) |
6 | 172 |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
173 |
def __cmp__ (self, other_obj) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
174 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
175 |
A sockaddr_in is equal to any other sockaddr_in which has the same addr and port |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
176 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
177 |
>>> assert sockaddr_in() == sockaddr_in() |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
178 |
>>> assert sockaddr_in('127.0.0.1', 80) == sockaddr_in('127.0.0.1', 80) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
179 |
>>> addr = sockaddr_in(); assert addr == addr |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
180 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
181 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
182 |
if not isinstance(other_obj, sockaddr_in) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
183 |
return <object> py.Py_NotImplemented |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
184 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
185 |
cdef sockaddr_in other = other_obj |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
186 |
cdef platform.sockaddr_in *sa1 = &self.sockaddr, *sa2 = &other.sockaddr |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
187 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
188 |
if other is self : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
189 |
return 0 |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
190 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
191 |
return ( |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
192 |
libc.memcmp(<void *> &sa1.sin_port, <void *> &sa2.sin_port, sizeof(sa1.sin_port)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
193 |
or libc.memcmp(<void *> &sa1.sin_addr, <void *> &sa2.sin_addr, sizeof(sa1.sin_addr)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
194 |
) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
195 |
|
6 | 196 |
def __str__ (self) : |
197 |
""" |
|
198 |
Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
199 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
200 |
>>> str(sockaddr_in()) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
201 |
'0.0.0.0:0' |
6 | 202 |
""" |
203 |
||
204 |
# format |
|
205 |
return "%s:%s" % self.getnameinfo() |
|
206 |
||
207 |
cdef class sockaddr_in6 (sockaddr) : |
|
208 |
""" |
|
209 |
AF_INET6 struct sockaddr_in6 |
|
210 |
||
211 |
>>> sa6 = sockaddr_in6("::1", 80) |
|
212 |
>>> sa6.addr |
|
213 |
'::1' |
|
214 |
>>> sa6.port |
|
215 |
80 |
|
216 |
>>> str(sa6) |
|
217 |
'[::1]:80' |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
218 |
|
6 | 219 |
""" |
220 |
||
10 | 221 |
cdef platform.sockaddr_in6 sockaddr |
6 | 222 |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
223 |
def __init__ (self, object addr=None, platform.in_port_t port=0, unsigned int scope_id = 0) : |
6 | 224 |
""" |
225 |
Construct using given literal IPv6 address and TCP/UDP port |
|
226 |
||
10 | 227 |
addr - IPv6 address, defaults to platform.in6addr_any (::) |
6 | 228 |
port - TCP/UDP port, defaults to 0 (ephemeral) |
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
229 |
scope_id - (optional) scope ID representing interface index for link-local addresses |
6 | 230 |
""" |
231 |
||
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
232 |
# zero |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
233 |
libc.memset(&self.sockaddr, 0, sizeof(self.sockaddr)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
234 |
|
6 | 235 |
# store our family |
236 |
# XXX: this should be a class attribute... |
|
10 | 237 |
self._init_family(platform.AF_INET6) |
6 | 238 |
|
239 |
# constant af |
|
240 |
self.sockaddr.sin6_family = self.family |
|
241 |
||
242 |
# set the sin_port |
|
10 | 243 |
self.sockaddr.sin6_port = platform.htons(port) |
6 | 244 |
|
245 |
if addr : |
|
246 |
# set the sin_addr |
|
247 |
# this automatically converts the addr from str -> char * |
|
10 | 248 |
platform.inet_pton(self.family, addr, &self.sockaddr.sin6_addr) |
6 | 249 |
|
250 |
else : |
|
251 |
# set as INADDR_ANY |
|
10 | 252 |
self.sockaddr.sin6_addr = platform.in6addr_any |
6 | 253 |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
254 |
# scope ID |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
255 |
self.sockaddr.sin6_scope_id = scope_id |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
256 |
|
10 | 257 |
cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 : |
6 | 258 |
if sa_ptr : |
10 | 259 |
sa_ptr[0] = <platform.sockaddr *> &self.sockaddr |
6 | 260 |
|
261 |
if sa_len : |
|
262 |
sa_len[0] = sizeof(self.sockaddr) |
|
263 |
||
264 |
return 0 |
|
265 |
||
10 | 266 |
cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 : |
6 | 267 |
assert sa_len == sizeof(self.sockaddr) |
268 |
||
269 |
libc.memcpy(&self.sockaddr, sa, sa_len) |
|
270 |
||
271 |
property port : |
|
272 |
""" |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
273 |
The integer port number. |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
274 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
275 |
This will represent it correctly in host byte order. |
6 | 276 |
""" |
277 |
||
278 |
def __get__ (self) : |
|
10 | 279 |
return platform.ntohs(self.sockaddr.sin6_port) |
6 | 280 |
|
281 |
||
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
282 |
property flowinfo : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
283 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
284 |
The integer flowinfo |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
285 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
286 |
XXX: byteorder? |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
287 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
288 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
289 |
def __get__ (self) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
290 |
return self.sockaddr.sin6_flowinfo |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
291 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
292 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
293 |
property scope_id : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
294 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
295 |
The scope ID - corresponds to an interface index for link-scope addresses. |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
296 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
297 |
This should be in host byte order... |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
298 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
299 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
300 |
def __get__ (self) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
301 |
return self.sockaddr.sin6_scope_id |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
302 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
303 |
def __cmp__ (self, other_obj) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
304 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
305 |
A sockaddr_in6 is equal to any other sockaddr_in6 which has the same addr, port and scope ID. |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
306 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
307 |
XXX: flowinfo? |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
308 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
309 |
XXX: A sockaddr_in6 is also equal to a sockaddr_in if the sockaddr_in6 represents the given v4-mapped address. |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
310 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
311 |
>>> assert sockaddr_in6() == sockaddr_in6() |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
312 |
>>> assert sockaddr_in6('0:0:0::1', 80) == sockaddr_in6('::1', 80) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
313 |
>>> assert sockaddr_in6('::127.0.0.1') == sockaddr_in('127.0.0.1') |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
314 |
""" |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
315 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
316 |
if not isinstance(other_obj, sockaddr_in6) : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
317 |
return <object> py.Py_NotImplemented |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
318 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
319 |
cdef sockaddr_in6 other = other_obj |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
320 |
cdef platform.sockaddr_in6 *sa1 = &self.sockaddr, *sa2 = &other.sockaddr |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
321 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
322 |
if other is self : |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
323 |
return 0 |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
324 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
325 |
return ( |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
326 |
libc.memcmp(<void *> &sa1.sin6_port, <void *> &sa2.sin6_port, sizeof(sa1.sin6_port)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
327 |
or libc.memcmp(<void *> &sa1.sin6_addr, <void *> &sa2.sin6_addr, sizeof(sa1.sin6_addr)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
328 |
or libc.memcmp(<void *> &sa1.sin6_scope_id, <void *> &sa2.sin6_scope_id, sizeof(sa1.sin6_scope_id)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
329 |
) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
330 |
|
6 | 331 |
def __str__ (self) : |
332 |
""" |
|
333 |
Return the literal ASCII representation for this sockaddr as a '[<addr>]:<port> string |
|
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
334 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
335 |
>>> str(sockaddr_in6()) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
336 |
'[::]:0' |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
337 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
338 |
>>> str(sockaddr_in6('2001:0::05:1')) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
339 |
'[2001::5:1]' |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
340 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
341 |
>>> str(sockaddr_in6('fe80::abcd', scope_id=5)) |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
342 |
'[fe80::abcd%5]' |
6 | 343 |
""" |
22
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
344 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
345 |
addr, port = self.getnameinfo() |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
346 |
scope_id = self.scope_id |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
347 |
|
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
348 |
# format with scope_id |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
349 |
return "[%s%s]:%s" % ( |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
350 |
addr, |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
351 |
"%%%d" % scope_id if scope_id else "", |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
352 |
port |
f6e8d5e37998
some __cmp__ and doctest love for socket.address
Tero Marttila <terom@fixme.fi>
parents:
20
diff
changeset
|
353 |
) |
6 | 354 |
|
355 |
# mapping of AF -> sockaddr, user-modifyable |
|
356 |
SOCKADDR_BY_FAMILY = { |
|
10 | 357 |
platform.AF_INET: sockaddr_in, |
358 |
platform.AF_INET6: sockaddr_in6, |
|
6 | 359 |
} |
360 |
||
361 |
# build a sockaddr from the given sockaddr struct, based on sa_family |
|
10 | 362 |
cdef sockaddr build_sockaddr (platform.sockaddr *sa, size_t sa_len) : |
6 | 363 |
# lookup correct class to use |
364 |
addr_type = SOCKADDR_BY_FAMILY[sa.sa_family] |
|
365 |
||
366 |
# construct with defaults |
|
367 |
cdef sockaddr addr = addr_type() |
|
368 |
||
369 |
# store |
|
370 |
addr._set_sockaddr(sa, sa_len) |
|
371 |
||
372 |
return addr |
|
373 |
||
374 |
cdef class addrinfo : |
|
375 |
||
10 | 376 |
cdef _init_addrinfo (self, platform.addrinfo *ai) : |
6 | 377 |
#ai.flags = c_ai.ai_flags |
378 |
self.family = ai.ai_family |
|
379 |
self.socktype = ai.ai_socktype |
|
380 |
self.protocol = ai.ai_protocol |
|
381 |
self.addr = build_sockaddr(ai.ai_addr, ai.ai_addrlen) |
|
382 |
self.canonname = ai.ai_canonname if ai.ai_canonname else None |
|
383 |
||
384 |
def __str__ (self) : |
|
385 |
return "family=%d, socktype=%d, protocol=%d, addr=%s, canonname=%s" % (self.family, self.socktype, self.protocol, self.addr, self.canonname) |
|
386 |
||
10 | 387 |
cdef addrinfo build_addrinfo (platform.addrinfo *c_ai) : |
6 | 388 |
cdef addrinfo ai = addrinfo() |
389 |
||
390 |
ai._init_addrinfo(c_ai) |
|
391 |
||
392 |
return ai |
|
393 |
||
394 |
cdef class endpoint : |
|
395 |
||
396 |
def __init__ (self, hostname=None, service=None) : |
|
397 |
""" |
|
398 |
Construct with the given hostname/service, either of which may be None. |
|
399 |
||
400 |
A hostname of None implies all valid local addresses (with AI_PASSIVE), and a service of None implies an |
|
401 |
ephemeral port. |
|
402 |
||
403 |
hostname - the literal address or DNS hostname or anything else that GAI supports |
|
404 |
service - the numeric port or service name |
|
405 |
""" |
|
406 |
||
407 |
self.hostname = str(hostname) |
|
408 |
self.service = str(service) |
|
409 |
||
10 | 410 |
cpdef getaddrinfo (self, int family, int socktype, int protocol = 0, int flags = platform.AI_PASSIVE) : |
6 | 411 |
""" |
412 |
Look up our hostname/service using the given socket parameters, and return a sequence of addrinfo objects. |
|
413 |
""" |
|
414 |
||
415 |
# XXX: Cython doesn't support proper compound value literals... |
|
10 | 416 |
cdef platform.addrinfo hints |
6 | 417 |
|
418 |
libc.memset(&hints, 0, sizeof(hints)) |
|
419 |
hints.ai_flags = flags |
|
420 |
hints.ai_family = family |
|
421 |
hints.ai_socktype = socktype |
|
422 |
hints.ai_protocol = protocol |
|
423 |
||
10 | 424 |
cdef platform.addrinfo *res, *r |
6 | 425 |
cdef int err |
426 |
cdef object ret = [] |
|
427 |
||
428 |
cdef char *hostname = NULL |
|
429 |
cdef char *service = NULL |
|
430 |
||
431 |
if self.hostname is not None : |
|
432 |
hostname = self.hostname |
|
433 |
||
434 |
if self.service is not None : |
|
435 |
service = self.service |
|
436 |
||
437 |
# operate! |
|
10 | 438 |
err = platform.c_getaddrinfo(hostname, service, &hints, &res) |
6 | 439 |
|
440 |
try : |
|
441 |
if err : |
|
442 |
# XXX: raise a GAIError |
|
10 | 443 |
raise Exception(platform.gai_strerror(err)) |
6 | 444 |
|
445 |
# gather results |
|
446 |
r = res |
|
447 |
||
448 |
while r : |
|
449 |
ret.append(build_addrinfo(r)) |
|
450 |
||
451 |
r = r.ai_next |
|
452 |
||
453 |
# ok |
|
454 |
return ret |
|
455 |
||
456 |
finally : |
|
10 | 457 |
platform.c_freeaddrinfo(res) |
6 | 458 |
|
459 |
def __str__ (self) : |
|
460 |
return "hostname=%s, service=%s" % (self.hostname, self.service) |
|
461 |