qmsk.net: comparison sock/addr.pyx

equal deleted inserted replaced

-:664a1dfe08ac
+:59bed837c265
 cdef class sockaddr :
 cdef void _init_family (self, libc.sa_family_t family=libc.AF_UNSPEC) :
 self.family = family
+# XXX:use size_t
 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
 """
 cdef libc.socklen_t sa_len
 self._get_sockaddr(&sa, &sa_len)
 return sa_len
+cdef int _set_sockaddr (self, libc.sockaddr *sa, size_t sa_len) except -1 :
+"""
+Set the sockaddr value for this address; sa_len must match!
+"""
+raise NotImplementedError()
 def getnameinfo (self) :
 """
 Returns a (host, serv) tuple for this address à la getnameinfo
 """
 if sa_len :
 sa_len[0] = sizeof(self.sockaddr)
 return 0
+cdef int _set_sockaddr (self, libc.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
 """
 if sa_len :
 sa_len[0] = sizeof(self.sockaddr)
 return 0
+cdef int _set_sockaddr (self, libc.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
 """
 """
 # format
 return "[%s]:%s" % self.getnameinfo()
+# mapping of AF -> sockaddr, user-modifyable
+SOCKADDR_BY_FAMILY = {
+libc.AF_INET:   sockaddr_in,
+libc.AF_INET6:  sockaddr_in6,
+}
+# build a sockaddr from the given sockaddr struct, based on sa_family
+cdef sockaddr build_sockaddr (libc.sockaddr *sa, size_t sa_len) :
+# lookup correct class to use
+addr_type = SOCKADDR_BY_FAMILY[sa.sa_family]
+# construct with defaults
+cdef sockaddr addr = addr_type()
+# store
+addr._set_sockaddr(sa, sa_len)
+return addr
+cdef class addrinfo :
+cdef _init_addrinfo (self, libc.addrinfo *ai) :
+#ai.flags = c_ai.ai_flags
+self.family = ai.ai_family
+self.socktype = ai.ai_socktype
+self.protocol = ai.ai_protocol
+self.addr = build_sockaddr(ai.ai_addr, ai.ai_addrlen)
+self.canonname = ai.ai_canonname if ai.ai_canonname else None
+def __str__ (self) :
+return "family=%d, socktype=%d, protocol=%d, addr=%s, canonname=%s" % (self.family, self.socktype, self.protocol, self.addr, self.canonname)
+cdef addrinfo build_addrinfo (libc.addrinfo *c_ai) :
+cdef addrinfo ai = addrinfo()
+ai._init_addrinfo(c_ai)
+return ai
+cdef class endpoint :
+def __init__ (self, hostname=None, service=None) :
+"""
+Construct with the given hostname/service, either of which may be None.
+A hostname of None implies all valid local addresses (with AI_PASSIVE), and a service of None implies an
+ephemeral port.
+hostname        - the literal address or DNS hostname or anything else that GAI supports
+service         - the numeric port or service name
+"""
+self.hostname = str(hostname)
+self.service = str(service)
+cpdef getaddrinfo (self, int family, int socktype, int protocol = 0, int flags = libc.AI_PASSIVE) :
+"""
+Look up our hostname/service using the given socket parameters, and return a sequence of addrinfo objects.
+"""
+# XXX: Cython doesn't support proper compound value literals...
+cdef libc.addrinfo hints
+libc.memset(&hints, 0, sizeof(hints))
+hints.ai_flags          = flags
+hints.ai_family         = family
+hints.ai_socktype       = socktype
+hints.ai_protocol       = protocol
+cdef libc.addrinfo *res, *r
+cdef int err
+cdef object ret = []
+cdef char *hostname = NULL
+cdef char *service = NULL
+if self.hostname is not None :
+hostname = self.hostname
+if self.service is not None :
+service = self.service
+# operate!
+err = libc.c_getaddrinfo(hostname, service, &hints, &res)
+try :
+if err :
+# XXX: raise a GAIError
+raise Exception(libc.gai_strerror(err))
+# gather results
+r = res
+while r :
+ret.append(build_addrinfo(r))
+r = r.ai_next
+# ok
+return ret
+finally :
+libc.c_freeaddrinfo(res)
+def __str__ (self) :
+return "hostname=%s, service=%s" % (self.hostname, self.service)

changeset 5	59bed837c265
parent 2	171e77f8d675