--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/qmsk/net/socket/af_inet.pyx Tue Aug 18 23:02:30 2009 +0300
@@ -0,0 +1,94 @@
+cimport qmsk.net.py as py
+
+cdef class sockaddr_in (sockaddr) :
+
+ def __init__ (self, object addr=None, platform.in_port_t port=0) :
+ """
+ Construct using given literal IPv4 address and TCP/UDP port
+
+ addr - IPv4 address, defaults to INADDR_ANY (0.0.0.0)
+ port - TCP/UDP port, defaults to 0 (ephemeral)
+ """
+
+ # zero
+ libc.memset(&self.sockaddr, 0, sizeof(self.sockaddr))
+
+ # store our family
+ # XXX: this should be a class attribute...
+ self._init_family(platform.AF_INET)
+
+ # constant af
+ self.sockaddr.sin_family = self.family
+
+ # set the sin_port
+ self.sockaddr.sin_port = platform.htons(port)
+
+ if addr :
+ # set the sin_addr
+ # this automatically converts the addr from str -> char *
+ platform.inet_pton(self.family, addr, &self.sockaddr.sin_addr)
+
+ else :
+ # set as INADDR_ANY
+ self.sockaddr.sin_addr.s_addr = platform.INADDR_ANY
+
+ cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 :
+ if sa_ptr :
+ sa_ptr[0] = <platform.sockaddr *> &self.sockaddr
+
+ if sa_len :
+ sa_len[0] = sizeof(self.sockaddr)
+
+ return 0
+
+ cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 :
+ assert sa_len == sizeof(self.sockaddr)
+
+ libc.memcpy(&self.sockaddr, sa, sa_len)
+
+ property port :
+ """
+ The integer port number
+
+ >>> sockaddr_in(port=1234).port
+ 1234
+ """
+
+ def __get__ (self) :
+ return platform.ntohs(self.sockaddr.sin_port)
+
+ def __cmp__ (self, other_obj) :
+ """
+ A sockaddr_in is equal to any other sockaddr_in which has the same addr and port
+
+ >>> assert sockaddr_in() == sockaddr_in()
+ >>> assert sockaddr_in('127.0.0.1', 80) == sockaddr_in('127.0.0.1', 80)
+ >>> addr = sockaddr_in(); assert addr == addr
+ """
+
+ if not isinstance(other_obj, sockaddr_in) :
+ return <object> py.Py_NotImplemented
+
+ cdef sockaddr_in other = other_obj
+ cdef platform.sockaddr_in *sa1 = &self.sockaddr, *sa2 = &other.sockaddr
+
+ if other is self :
+ return 0
+
+ return (
+ libc.memcmp(<void *> &sa1.sin_port, <void *> &sa2.sin_port, sizeof(sa1.sin_port))
+ or libc.memcmp(<void *> &sa1.sin_addr, <void *> &sa2.sin_addr, sizeof(sa1.sin_addr))
+ )
+
+ def __str__ (self) :
+ """
+ Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string
+
+ >>> str(sockaddr_in())
+ '0.0.0.0:0'
+ """
+
+ # format
+ return "%s:%s" % self.getnameinfo()
+
+