Ticket #1713: test_socket_server.py

#!/usr/bin/env python
"""Test socket connections on Ubuntu.

Show specific behaviour when using ``socket.getaddrinfo`` with host argument
set to ``None``.

In this case, the address info for IPV6 will be returned first, so when the
server listen socket is created based on this info, it will be IPV6, and when a
client connects, its address will be a IPV6 address.

Example client code::

    import socket
    host = 'localhost'
    port = 8080
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect((host, port))
    s.close()
"""

import socket
import sys

def main(args):
    sock = None
    flags = 0

    try:
        host = args.pop(0)
        port = args.pop(0)
    except:
        host = ''
        # Uncomment following line and comment out line above for testing
        # differing behaviour related to IPV6.
        #host = '0.0.0.0'
        port = 8080

    if host == '':
        host = None
        flags = socket.AI_PASSIVE
    info = socket.getaddrinfo(host, port, socket.AF_UNSPEC, socket.SOCK_STREAM,
        0, flags)
    print info

    af, socktype, proto, canonname, sa = info[0]
    try:
        sock = socket.socket(af, socktype, proto)
        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        sock.bind((host or '', port))
    except socket.error, msg:
        if sock:
            sock.close()
        print "Error: could not open socket '%s' port %s'" % (host, port)
        return 1
    else:
        sock.listen(1)
        print "Listening on '%s' port %s'" % (host, port)

    while True:
        try:
            s, addr = sock.accept()
            print "Remote address: %s" % (addr,)
        except KeyboardInterrupt:
            break
    return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))