author | Tero Marttila <terom@fixme.fi> |
Fri, 25 Sep 2009 21:34:04 +0300 | |
changeset 49 | e2f79e68418a |
parent 24 | f18b5787c46c |
permissions | -rw-r--r-- |
49
e2f79e68418a
fix up circular cimports related to the sockaddr type, and touch up endpoint/getaddrinfo a bit
Tero Marttila <terom@fixme.fi>
parents:
24
diff
changeset
|
1 |
cimport qmsk.net.libc as libc |
24
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
2 |
cimport qmsk.net.py as py |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
3 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
4 |
cdef class sockaddr_in (sockaddr) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
5 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
6 |
def __init__ (self, object addr=None, platform.in_port_t port=0) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
7 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
8 |
Construct using given literal IPv4 address and TCP/UDP port |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
9 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
10 |
addr - IPv4 address, defaults to INADDR_ANY (0.0.0.0) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
11 |
port - TCP/UDP port, defaults to 0 (ephemeral) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
12 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
13 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
14 |
# zero |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
15 |
libc.memset(&self.sockaddr, 0, sizeof(self.sockaddr)) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
16 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
17 |
# store our family |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
18 |
# XXX: this should be a class attribute... |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
19 |
self._init_family(platform.AF_INET) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
20 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
21 |
# constant af |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
22 |
self.sockaddr.sin_family = self.family |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
23 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
24 |
# set the sin_port |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
25 |
self.sockaddr.sin_port = platform.htons(port) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
26 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
27 |
if addr : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
28 |
# set the sin_addr |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
29 |
# this automatically converts the addr from str -> char * |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
30 |
platform.inet_pton(self.family, addr, &self.sockaddr.sin_addr) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
31 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
32 |
else : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
33 |
# set as INADDR_ANY |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
34 |
self.sockaddr.sin_addr.s_addr = platform.INADDR_ANY |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
35 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
36 |
cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
37 |
if sa_ptr : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
38 |
sa_ptr[0] = <platform.sockaddr *> &self.sockaddr |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
39 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
40 |
if sa_len : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
41 |
sa_len[0] = sizeof(self.sockaddr) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
42 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
43 |
return 0 |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
44 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
45 |
cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
46 |
assert sa_len == sizeof(self.sockaddr) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
47 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
48 |
libc.memcpy(&self.sockaddr, sa, sa_len) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
49 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
50 |
property port : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
51 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
52 |
The integer port number |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
53 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
54 |
>>> sockaddr_in(port=1234).port |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
55 |
1234 |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
56 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
57 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
58 |
def __get__ (self) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
59 |
return platform.ntohs(self.sockaddr.sin_port) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
60 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
61 |
def __cmp__ (self, other_obj) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
62 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
63 |
A sockaddr_in is equal to any other sockaddr_in which has the same addr and port |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
64 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
65 |
>>> assert sockaddr_in() == sockaddr_in() |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
66 |
>>> assert sockaddr_in('127.0.0.1', 80) == sockaddr_in('127.0.0.1', 80) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
67 |
>>> addr = sockaddr_in(); assert addr == addr |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
68 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
69 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
70 |
if not isinstance(other_obj, sockaddr_in) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
71 |
return <object> py.Py_NotImplemented |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
72 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
73 |
cdef sockaddr_in other = other_obj |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
74 |
cdef platform.sockaddr_in *sa1 = &self.sockaddr, *sa2 = &other.sockaddr |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
75 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
76 |
if other is self : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
77 |
return 0 |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
78 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
79 |
return ( |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
80 |
libc.memcmp(<void *> &sa1.sin_port, <void *> &sa2.sin_port, sizeof(sa1.sin_port)) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
81 |
or libc.memcmp(<void *> &sa1.sin_addr, <void *> &sa2.sin_addr, sizeof(sa1.sin_addr)) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
82 |
) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
83 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
84 |
def __str__ (self) : |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
85 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
86 |
Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
87 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
88 |
>>> str(sockaddr_in()) |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
89 |
'0.0.0.0:0' |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
90 |
""" |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
91 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
92 |
# format |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
93 |
return "%s:%s" % self.getnameinfo() |
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
94 |
|
f18b5787c46c
reorganize socket, adding af_inet/af_inet6 modules, and moving sctp in
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
95 |