sock/addr.pyx
changeset 0 975801b28d85
child 1 0ca9278146d7
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sock/addr.pyx	Sun Aug 16 02:58:32 2009 +0300
@@ -0,0 +1,220 @@
+"""
+    Socket addresses
+"""
+
+cimport libc
+
+cdef class sockaddr :
+    """
+        A network-level socket address
+
+        >>> sockaddr().family
+        0
+        >>> sockaddr().port
+        Traceback (most recent call last):
+          ...
+        NotImplementedError
+        >>> sockaddr().getnameinfo()
+        Traceback (most recent call last):
+          ...
+        NotImplementedError
+    """
+    
+    # address family
+    cdef readonly libc.sa_family_t family
+
+    cdef void _init_family (self, libc.sa_family_t family=libc.AF_UNSPEC) :
+        self.family = family
+
+    cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
+        """
+            Get the sockaddr pointer and sockaddr length for this address
+        """ 
+
+        raise NotImplementedError()
+
+    def getnameinfo (self) :
+        """
+            Returns a (host, serv) tuple for this address à la getnameinfo
+        """
+
+        cdef libc.sockaddr *sa
+        cdef libc.socklen_t sa_len
+
+        # XXX: take as args?
+        cdef int flags = libc.NI_NUMERICHOST | libc.NI_NUMERICSERV
+        
+        # get our abstract sockaddr
+        self._get_sockaddr(&sa, &sa_len)
+
+        # get nice text format
+        return libc.getnameinfo(sa, sa_len, flags)
+
+    property addr :
+        """
+            The ASCII literal network address
+        """
+
+        def __get__ (self) :
+            """
+                Default implmentation using getnameinfo()
+            """
+
+            addr, port = self.getnameinfo()
+
+            return addr
+
+    property port :
+        """
+            The integer port number
+        """
+
+        def __get__ (self) :
+            """
+                Default implementation using getnameinfo() and int()
+            """
+
+            addr, port = self.getnameinfo()
+
+            return int(port)
+
+cdef class sockaddr_in (sockaddr) :
+    """
+        AF_INET struct sockaddr_in
+
+        >>> sa = sockaddr_in("127.0.0.1", 80)
+        >>> sa.addr
+        '127.0.0.1'
+        >>> sa.port
+        80
+        >>> str(sa)
+        '127.0.0.1:80'
+        >>> str(sockaddr_in())
+        '0.0.0.0:0'
+    """
+
+    # the struct sockaddr_in
+    cdef libc.sockaddr_in sockaddr
+
+    def __init__ (self, object addr=None, libc.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)
+        """
+
+        # store our family
+        # XXX: this should be a class attribute...
+        self._init_family(libc.AF_INET)
+
+        # constant af
+        self.sockaddr.sin_family = self.family
+        
+        # set the sin_port
+        self.sockaddr.sin_port = libc.htons(port)
+        
+        if addr :
+            # set the sin_addr
+            # this automatically converts the addr from str -> char *
+            libc.inet_pton(self.family, addr, &self.sockaddr.sin_addr)
+
+        else :
+            # set as INADDR_ANY
+            self.sockaddr.sin_addr.s_addr = libc.INADDR_ANY
+    
+    cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
+        if sa_ptr :
+            sa_ptr[0] = <libc.sockaddr *> &self.sockaddr
+
+        if sa_len :
+            sa_len[0] = sizeof(self.sockaddr)
+
+        return 0
+
+    property port :
+        """
+            The integer port number
+        """
+
+        def __get__ (self) :
+            return libc.ntohs(self.sockaddr.sin_port)
+
+    def __str__ (self) :
+        """
+            Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string
+        """
+        
+        # format
+        return "%s:%s" % self.getnameinfo()
+
+cdef class sockaddr_in6 (sockaddr) :
+    """
+        AF_INET6 struct sockaddr_in6
+
+        >>> sa6 = sockaddr_in6("::1", 80)
+        >>> sa6.addr
+        '::1'
+        >>> sa6.port
+        80
+        >>> str(sa6)
+        '[::1]:80'
+        >>> str(sockaddr_in6())
+        '[::]:0'
+    """
+
+    cdef libc.sockaddr_in6 sockaddr
+
+    def __init__ (self, object addr=None, libc.in_port_t port=0) :
+        """
+            Construct using given literal IPv6 address and TCP/UDP port
+
+                addr        - IPv6 address, defaults to in6addr_any (::)
+                port        - TCP/UDP port, defaults to 0 (ephemeral)
+        """
+
+        # store our family
+        # XXX: this should be a class attribute...
+        self._init_family(libc.AF_INET6)
+
+        # constant af
+        self.sockaddr.sin6_family = self.family
+        
+        # set the sin_port
+        self.sockaddr.sin6_port = libc.htons(port)
+        
+        if addr :
+            # set the sin_addr
+            # this automatically converts the addr from str -> char *
+            libc.inet_pton(self.family, addr, &self.sockaddr.sin6_addr)
+
+        else :
+            # set as INADDR_ANY
+            self.sockaddr.sin6_addr = libc.in6addr_any
+
+    cdef int _get_sockaddr (self, libc.sockaddr **sa_ptr, libc.socklen_t *sa_len) except -1 :
+        if sa_ptr :
+            sa_ptr[0] = <libc.sockaddr *> &self.sockaddr
+
+        if sa_len :
+            sa_len[0] = sizeof(self.sockaddr)
+
+        return 0
+
+    property port :
+        """
+            The integer port number
+        """
+
+        def __get__ (self) :
+            return libc.ntohs(self.sockaddr.sin6_port)
+
+
+    def __str__ (self) :
+        """
+            Return the literal ASCII representation for this sockaddr as a '[<addr>]:<port> string
+        """
+        
+        # format
+        return "[%s]:%s" % self.getnameinfo()
+