qmsk/net/transport/endpoint.py
author Tero Marttila <terom@fixme.fi>
Sun, 23 Aug 2009 22:39:19 +0300
changeset 35 50dc1517f797
parent 32 2168a9ffaef7
child 37 14db3fe42b6c
permissions -rw-r--r--
fix endpoint.InetAddr.resolve
32
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     1
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     2
from qmsk.net.socket import address, constants, af_unix
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     3
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     4
class AddrInfo (address.addrinfo) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     5
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     6
        Represents a sockaddr along with related address-family etc. information
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     7
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     8
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
     9
class Endpoint (object) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    10
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    11
        Abstract network address interface, 
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    12
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    13
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    14
    def resolve (self, family, socktype, protocol = 0, passive = True) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    15
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    16
            Translate this Endpoint into a sequence of AddrInfo objects for the given family/socktype.
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    17
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    18
        
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    19
        raise NotImplemented()
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    20
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    21
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    22
class InetAddr (Endpoint, address.endpoint) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    23
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    24
        An internet address, either IPv4 or IPv6.
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    25
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    26
           hostname     - [str] literal address or DNS hostname
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    27
           service      - [str] port number or service name
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    28
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    29
    
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    30
    def resolve (self, family, socktype, protocol = 0, passive = True) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    31
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    32
            Resolve using getaddrinfo
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    33
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    34
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    35
        flags = 0
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    36
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    37
        if passive :
35
50dc1517f797 fix endpoint.InetAddr.resolve
Tero Marttila <terom@fixme.fi>
parents: 32
diff changeset
    38
            flags |= constants.AI_PASSIVE
32
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    39
35
50dc1517f797 fix endpoint.InetAddr.resolve
Tero Marttila <terom@fixme.fi>
parents: 32
diff changeset
    40
        return self.getaddrinfo(family, socktype, protocol, flags)
32
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    41
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    42
class UnixAddr (Endpoint) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    43
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    44
        A local AF_UNIX address, as a path.
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    45
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    46
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    47
    def __init__ (self, path) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    48
        self.path = path
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    49
        self.addr = af_unix.sockaddr_un(path)
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    50
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    51
    def resolve (self, family, socktype, protocol = 0, passive = True) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    52
        if family != constants.AF_UNIX :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    53
            raise ValueError("Address family mismatch: %s" % (family, ))
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    54
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    55
        if not socktype :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    56
            raise ValueError("Unknown socktype: %s" % (socktype, ))
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    57
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    58
        return [AddrInfo(0, constants.AF_UNIX, socktype, protocol, self.addr, self.path)]
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    59
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    60
class SockAddr (Endpoint) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    61
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    62
        A specific address, suitable both as an endpoint.
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    63
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    64
            addr        - low-level sockaddr object
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    65
            family      - AF_*
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    66
            socktype    - SOCK_*
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    67
    """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    68
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    69
    def __init__ (self, addr, family = constants.AF_UNSPEC, socktype = 0) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    70
        if not family :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    71
            family = addr.family
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    72
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    73
        self.addr = addr
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    74
        self.family = family
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    75
        self.socktype = socktype
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    76
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    77
    def resolve (self, family, socktype, protocol = 0, passive = True) :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    78
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    79
            Returns a single AddrInfo object representing this address 
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    80
        """
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    81
    
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    82
        if family and family != self.family :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    83
            raise ValueError("Address family mismatch: %s should be %s" % (family, self.family))
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    84
        
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    85
        elif not family :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    86
            family = self.family
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    87
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    88
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    89
        if socktype and self.socktype and socktype != self.socktype :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    90
            raise ValueError("Socket type mismatch: %s should be %s" % (socktype, self.socktype))
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    91
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    92
        if not socktype :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    93
            if self.socktype :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    94
                socktype = self.socktype
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    95
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    96
            else :
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    97
                raise ValueError("Socket type unknown")
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    98
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
    99
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
   100
        return [AddrInfo(0, family, socktype, protocol, self.addr, None)]
2168a9ffaef7 transport.endpoint stuff
Tero Marttila <terom@fixme.fi>
parents:
diff changeset
   101