##################################################################################### # # Copyright (c) Microsoft Corporation. # # This source code is subject to terms and conditions of the Microsoft Public # License. A copy of the license can be found in the License.html file at the # root of this distribution. If you cannot locate the Microsoft Public # License, please send an email to dlr@microsoft.com. By using this source # code in any fashion, you are agreeing to be bound by the terms of the # Microsoft Public License. # # You must not remove this notice, or any other, from this software. # ##################################################################################### # # test socket # from lib.assert_util import * import sys #workaround - _socket does not appear to be in $PYTHONPATH for CPython #only when run from the old test suite. try: import socket except: print "Unable to import socket (_socket) from CPython" sys.exit(0) #----------------------- #--GLOBALS AF_DICT = {"AF_APPLETALK" : 5, "AF_DECnet" : 12, "AF_INET" : 2, "AF_INET6" : 10, "AF_IPX" : 4, "AF_IRDA" : 23, "AF_SNA" : 22, "AF_UNSPEC" : 0, } ST_DICT = {"SOCK_DGRAM" : 2, "SOCK_RAW" : 3, "SOCK_RDM" : 4, "SOCK_SEQPACKET" : 5, "SOCK_STREAM" : 1, } IPPROTO_DICT = { "IPPROTO_AH" : 51, "IPPROTO_DSTOPTS" : 60, "IPPROTO_ESP" : 50, "IPPROTO_FRAGMENT" : 44, "IPPROTO_HOPOPTS" : 0, "IPPROTO_ICMP" : 1, "IPPROTO_ICMPV6" : 58, "IPPROTO_IDP" : 22, "IPPROTO_IGMP" : 2, "IPPROTO_IP" : 0, "IPPROTO_IPV6" : 41, "IPPROTO_NONE" : 59, "IPPROTO_PUP" : 12, "IPPROTO_RAW" : 255, "IPPROTO_ROUTING" : 43, "IPPROTO_TCP" : 6, "IPPROTO_UDP" : 17, } OTHER_GLOBALS = {"AI_ADDRCONFIG" : 32, "AI_ALL" : 16, "AI_CANONNAME" : 2, "AI_NUMERICHOST" : 4, "AI_PASSIVE" : 1, "AI_V4MAPPED" : 8, "EAI_ADDRFAMILY" : -9, "EAI_AGAIN" : -3, "EAI_BADFLAGS" : -1, "EAI_FAIL" : -4, "EAI_FAMILY" : -6, "EAI_MEMORY" : -10, "EAI_NODATA" : -5, "EAI_NONAME" : -2, "EAI_SERVICE" : -8, "EAI_SOCKTYPE" : -7, "EAI_SYSTEM" : -11, "INADDR_ALLHOSTS_GROUP" : -536870911, "INADDR_ANY" : 0, "INADDR_BROADCAST" : -1, "INADDR_LOOPBACK" : 2130706433, "INADDR_MAX_LOCAL_GROUP" : -536870657, "INADDR_NONE" : -1, "INADDR_UNSPEC_GROUP" : -536870912, "IPPORT_RESERVED" : 1024, "IPPORT_USERRESERVED" : 5000, "IPV6_CHECKSUM" : 7, "IPV6_DSTOPTS" : 4, "IPV6_HOPLIMIT" : 8, "IPV6_HOPOPTS" : 3, "IPV6_JOIN_GROUP" : 20, "IPV6_LEAVE_GROUP" : 21, "IPV6_MULTICAST_HOPS" : 18, "IPV6_MULTICAST_IF" : 17, "IPV6_MULTICAST_LOOP" : 19, "IPV6_NEXTHOP" : 9, "IPV6_PKTINFO" : 2, "IPV6_RTHDR" : 5, "IPV6_RTHDR_TYPE_0" : 0, "IPV6_UNICAST_HOPS" : 16, "IPV6_V6ONLY" : 26, "IP_ADD_MEMBERSHIP" : 35, "IP_DEFAULT_MULTICAST_LOOP" : 1, "IP_DEFAULT_MULTICAST_TTL" : 1, "IP_DROP_MEMBERSHIP" : 36, "IP_HDRINCL" : 3, "IP_MAX_MEMBERSHIPS" : 20, "IP_MULTICAST_IF" : 32, "IP_MULTICAST_LOOP" : 34, "IP_MULTICAST_TTL" : 33, "IP_OPTIONS" : 4, "IP_RECVOPTS" : 6, "IP_RECVRETOPTS" : 7, "IP_RETOPTS" : 7, "IP_TOS" : 1, "IP_TTL" : 2, "MSG_CTRUNC" : 8, "MSG_DONTROUTE" : 4, "MSG_DONTWAIT" : 64, "MSG_EOR" : 128, "MSG_OOB" : 1, "MSG_PEEK" : 2, "MSG_TRUNC" : 32, "MSG_WAITALL" : 256, "NI_DGRAM" : 16, "NI_MAXHOST" : 1025, "NI_MAXSERV" : 32, "NI_NAMEREQD" : 8, "NI_NOFQDN" : 4, "NI_NUMERICHOST" : 1, "NI_NUMERICSERV" : 2, "PACKET_BROADCAST" : 1, "PACKET_FASTROUTE" : 6, "PACKET_HOST" : 0, "PACKET_LOOPBACK" : 5, "PACKET_MULTICAST" : 2, "PACKET_OTHERHOST" : 3, "PACKET_OUTGOING" : 4, "PF_PACKET" : 17, "SHUT_RD" : 0, "SHUT_RDWR" : 2, "SHUT_WR" : 1, "SOL_IP" : 0, "SOL_SOCKET" : 1, "SOL_TCP" : 6, "SOL_UDP" : 17, "SOMAXCONN" : 128, "SO_ACCEPTCONN" : 30, "SO_BROADCAST" : 6, "SO_DEBUG" : 1, "SO_DONTROUTE" : 5, "SO_ERROR" : 4, "SO_KEEPALIVE" : 9, "SO_LINGER" : 13, "SO_OOBINLINE" : 10, "SO_RCVBUF" : 8, "SO_RCVLOWAT" : 18, "SO_RCVTIMEO" : 20, "SO_REUSEADDR" : 2, "SO_SNDBUF" : 7, "SO_SNDLOWAT" : 19, "SO_SNDTIMEO" : 21, "SO_TYPE" : 3, "SSL_ERROR_EOF" : 8, "SSL_ERROR_INVALID_ERROR_CODE" : 9, "SSL_ERROR_SSL" : 1, "SSL_ERROR_SYSCALL" : 5, "SSL_ERROR_WANT_CONNECT" : 7, "SSL_ERROR_WANT_READ" : 2, "SSL_ERROR_WANT_WRITE" : 3, "SSL_ERROR_WANT_X509_LOOKUP" : 4, "SSL_ERROR_ZERO_RETURN" : 6, "TCP_CORK" : 3, "TCP_DEFER_ACCEPT" : 9, "TCP_INFO" : 11, "TCP_KEEPCNT" : 6, "TCP_KEEPIDLE" : 4, "TCP_KEEPINTVL" : 5, "TCP_LINGER2" : 8, "TCP_MAXSEG" : 2, "TCP_NODELAY" : 1, "TCP_QUICKACK" : 12, "TCP_SYNCNT" : 7, "TCP_WINDOW_CLAMP" : 10} @skip("win32") def test_HandleToSocket(): import clr try: s = socket.socket() system_socket = socket.socket.HandleToSocket(s.fileno()) AreEqual(s.fileno(), system_socket.Handle.ToInt64()) finally: s.close() def test_getprotobyname(): ''' Tests socket.getprotobyname ''' #IP and CPython proto_map = { "ggp": socket.IPPROTO_GGP, "icmp": socket.IPPROTO_ICMP, "ip": socket.IPPROTO_IP, "pup": socket.IPPROTO_PUP, "tcp": socket.IPPROTO_TCP, "udp": socket.IPPROTO_UDP, } #supported only by IP iponly_map = {"dstopts": socket.IPPROTO_DSTOPTS, "none": socket.IPPROTO_NONE, "raw": socket.IPPROTO_RAW, "ipv4": socket.IPPROTO_IPV4, "ipv6": socket.IPPROTO_IPV6, "esp": socket.IPPROTO_ESP, "fragment": socket.IPPROTO_FRAGMENT, "nd": socket.IPPROTO_ND, "icmpv6": socket.IPPROTO_ICMPV6, "routing": socket.IPPROTO_ROUTING, } if is_cli: proto_map.update(iponly_map) for proto_name, good_val in proto_map.iteritems(): temp_val = socket.getprotobyname(proto_name) AreEqual(temp_val, good_val) #negative cases bad_list = ["", "blah", "i"] for name in bad_list: AssertError(socket.error, socket.getprotobyname, name) def test_getaddrinfo(): ''' Tests socket.getaddrinfo ''' joe = { ("127.0.0.1", 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 1) : "[(2, 0, 0, '', ('127.0.0.1', 1))]", ("127.0.0.1", 0, 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 0, 0, 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 0, 0, 0, 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 0, 0, 0, 0, 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 0, 0, 0, 0, 0) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", ("127.0.0.1", 0, 0, 0, 0, 1) : "[(2, 0, 0, '', ('127.0.0.1', 0))]", } tmp = socket.getaddrinfo("127.0.0.1", 0, 0, 0, -100000, 0) tmp = socket.getaddrinfo("127.0.0.1", 0, 0, 0, 100000, 0) tmp = socket.getaddrinfo("127.0.0.1", 0, 0, 0, 0, 0) #just try them as-is for params,value in joe.iteritems(): addrinfo = socket.getaddrinfo(*params) AreEqual(repr(addrinfo), value) #change the address family for addr_fam in ["AF_INET", "AF_UNSPEC"]: addrinfo = socket.getaddrinfo("127.0.0.1", 0, eval("socket." + addr_fam), 0, 0, 0) AreEqual(repr(addrinfo), "[(2, 0, 0, '', ('127.0.0.1', 0))]") #change the socket type for socktype in ["SOCK_DGRAM", "SOCK_RAW", "SOCK_STREAM"]: socktype = eval("socket." + socktype) addrinfo = socket.getaddrinfo("127.0.0.1", 0, 0, socktype, 0, 0) AreEqual(repr(addrinfo), "[(2, " + str(socktype) + ", 0, '', ('127.0.0.1', 0))]") #change the protocol for proto in IPPROTO_DICT.keys():#["SOCK_DGRAM", "SOCK_RAW", "SOCK_STREAM"]: try: proto = eval("socket." + proto) except: print proto continue addrinfo = socket.getaddrinfo("127.0.0.1", 0, 0, 0, proto, 0) AreEqual(repr(addrinfo), "[(2, 0, " + str(proto) + ", '', ('127.0.0.1', 0))]") #negative cases AssertError(socket.gaierror, socket.getaddrinfo, "should never work.dfkdfjkkjdfkkdfjkdjf", 0) #CodePlex Work Item 5445 #AssertError(socket.gaierror, socket.getaddrinfo, "1", 0) AssertError(socket.gaierror, socket.getaddrinfo, ".", 0) AssertError(socket.error, socket.getaddrinfo, "127.0.0.1", 3.14, 0, 0, 0, 0) AssertError(socket.error, socket.getaddrinfo, "127.0.0.1", 0, -1, 0, 0, 0) AssertError(socket.error, socket.getaddrinfo, "127.0.0.1", 0, 0, -1, 0, 0) socket.getaddrinfo("127.0.0.1", 0, 0, 0, 1000000, 0) socket.getaddrinfo("127.0.0.1", 0, 0, 0, -1000000, 0) socket.getaddrinfo("127.0.0.1", 0, 0, 0, 0, 0) def test_getnameinfo(): ''' Tests socket.getnameinfo() ''' #sanity #CodePlex Work Item 5447 #socket.getnameinfo(("127.0.0.1", 80), 8) #socket.getnameinfo(("127.0.0.1", 80), 9) #host, service = socket.getnameinfo( ("127.0.0.1", 80), 8) #AreEqual(service, '80') if is_cli: AssertError(NotImplementedError, socket.getnameinfo, ("127.0.0.1", 80), 0) #IP gives a TypeError #AssertError(SystemError, socket.getnameinfo, ("127.0.0.1"), 8) #AssertError(SystemError, socket.getnameinfo, (321), 8) AssertError(TypeError, socket.getnameinfo, ("127.0.0.1"), '0') AssertError(TypeError, socket.getnameinfo, ("127.0.0.1", 80, 0, 0, 0), 8) AssertError(socket.gaierror, socket.getnameinfo, ('no such host will ever exist', 80), 8) def test_gethostbyaddr(): ''' Tests socket.gethostbyaddr ''' socket.gethostbyaddr("localhost") socket.gethostbyaddr("127.0.0.1") if is_cli: socket.gethostbyaddr("") def test_gethostbyname(): ''' Tests socket.gethostbyname ''' #sanity AreEqual(socket.gethostbyname("localhost"), "127.0.0.1") AreEqual(socket.gethostbyname("127.0.0.1"), "127.0.0.1") AreEqual(socket.gethostbyname(""), "255.255.255.255") #negative AssertError(socket.gaierror, socket.gethostbyname, "should never work") def test_gethostbyname_ex(): ''' Tests socket.gethostbyname_ex ''' #sanity joe = socket.gethostbyname_ex("localhost")[2] Assert(joe.count("127.0.0.1")==1) joe = socket.gethostbyname_ex("127.0.0.1")[2] Assert(joe.count("127.0.0.1")==1) #negative AssertError(socket.gaierror, socket.gethostbyname_ex, "should never work") def test_getservbyport(): if is_cli: AssertError(NotImplementedError, socket.getservbyport, 80) def test_getservbyname(): if is_cli: AssertError(NotImplementedError, socket.getservbyname, "http") def test_inet_ntop(): ''' Tests socket.inet_ntop ''' if not is_cli: return #negative AssertError(socket.error, socket.inet_ntop, socket.AF_INET, "garbage dkfjdkfjdkfj") def test_inet_pton(): ''' Tests socket.inet_pton ''' if not is_cli: return #sanity socket.inet_pton(socket.AF_INET, "127.0.0.1") #negative AssertError(socket.error, socket.inet_pton, socket.AF_INET, "garbage dkfjdkfjdkfj") def test_getfqdn(): ''' Tests socket.getfqdn ''' #TODO run_test(__name__)