2

cimport qmsk.net.py as py

     3

     4

cdef class sockaddr_in (sockaddr) :

     5

     6

    def __init__ (self, object addr=None, platform.in_port_t port=0) :

     7

        """

     8

            Construct using given literal IPv4 address and TCP/UDP port

     9

    10

                addr        - IPv4 address, defaults to INADDR_ANY (0.0.0.0)

    11

                port        - TCP/UDP port, defaults to 0 (ephemeral)

    12

        """

    13

    14

        # zero

    15

        libc.memset(&self.sockaddr, 0, sizeof(self.sockaddr))

    16

    17

        # store our family

    18

        # XXX: this should be a class attribute...

    19

        self._init_family(platform.AF_INET)

    20

    21

        # constant af

    22

        self.sockaddr.sin_family = self.family

    23

    24

        # set the sin_port

    25

        self.sockaddr.sin_port = platform.htons(port)

    26

    27

        if addr :

    28

            # set the sin_addr

    29

            # this automatically converts the addr from str -> char *

    30

            platform.inet_pton(self.family, addr, &self.sockaddr.sin_addr)

    31

    32

        else :

    33

            # set as INADDR_ANY

    34

            self.sockaddr.sin_addr.s_addr = platform.INADDR_ANY

    35

    36

    cdef int _get_sockaddr (self, platform.sockaddr **sa_ptr, platform.socklen_t *sa_len) except -1 :

    37

        if sa_ptr :

    38

            sa_ptr[0] = <platform.sockaddr *> &self.sockaddr

    39

    40

        if sa_len :

    41

            sa_len[0] = sizeof(self.sockaddr)

    42

    43

        return 0

    44

    45

    cdef int _set_sockaddr (self, platform.sockaddr *sa, size_t sa_len) except -1 :

    46

        assert sa_len == sizeof(self.sockaddr)

    47

    48

        libc.memcpy(&self.sockaddr, sa, sa_len)

    49

    50

    property port :

    51

        """

    52

            The integer port number

    53

    54

            >>> sockaddr_in(port=1234).port

    55

            1234

    56

        """

    57

    58

        def __get__ (self) :

    59

            return platform.ntohs(self.sockaddr.sin_port)

    60

    61

    def __cmp__ (self, other_obj) :

    62

        """

    63

            A sockaddr_in is equal to any other sockaddr_in which has the same addr and port

    64

    65

            >>> assert sockaddr_in() == sockaddr_in()

    66

            >>> assert sockaddr_in('127.0.0.1', 80) == sockaddr_in('127.0.0.1', 80)

    67

            >>> addr = sockaddr_in(); assert addr == addr

    68

        """

    69

    70

        if not isinstance(other_obj, sockaddr_in) :

    71

            return <object> py.Py_NotImplemented

    72

    73

        cdef sockaddr_in other = other_obj

    74

        cdef platform.sockaddr_in *sa1 = &self.sockaddr, *sa2 = &other.sockaddr

    75

    76

        if other is self :

    77

            return 0

    78

    79

        return (

    80

                libc.memcmp(<void *> &sa1.sin_port, <void *> &sa2.sin_port, sizeof(sa1.sin_port))

    81

            or  libc.memcmp(<void *> &sa1.sin_addr, <void *> &sa2.sin_addr, sizeof(sa1.sin_addr))

    82

        )

    83

    84

    def __str__ (self) :

    85

        """

    86

            Return the literal ASCII representation for this sockaddr as an '<addr>:<port> string

    87

    88

            >>> str(sockaddr_in())

    89

            '0.0.0.0:0'

    90

        """

    91

    92

        # format

    93

        return "%s:%s" % self.getnameinfo()

    94

    95