Python http server IPv6

今天準備測試一下IPv6皿伺，習慣性地使用Python的http.server模塊寝贡，意外的發(fā)現(xiàn)它不支持IPv6悼吱。在StackOverflow上有人說沸停，把HttpServer的address_family改成AF_INET6即可堰酿。于是瀏覽了一下/usr/lib/python3.6/http/server.py代碼疾宏，說干就干。

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--cgi', action='store_true',
                       help='Run as CGI Server')
    parser.add_argument('--ipv6', '-6', action='store_true',
                        help='Set address family to IPv6')
    parser.add_argument('--bind', '-b', default='', metavar='ADDRESS',
                        help='Specify alternate bind address '
                             '[default: all interfaces]')
    parser.add_argument('port', action='store',
                        default=8000, type=int,
                        nargs='?',
                        help='Specify alternate port [default: 8000]')
    args = parser.parse_args()
    if args.cgi:
        handler_class = CGIHTTPRequestHandler
    else:
        handler_class = SimpleHTTPRequestHandler
    server_class=HTTPServer

    if args.ipv6:
        server_class = type("HTTPServerV6", (HTTPServer,),
                            dict(address_family=socket.AF_INET6))
    else:
        server_class = HTTPServer

    test(HandlerClass=handler_class, ServerClass=server_class, port=args.port, bind=args.bind)

加了一行parser.add_argument('--ipv6', '-6', action='store_true', help='Set address family to IPv6')來解析IPv6參數(shù)触创，許多命令行工具都用了這種方式坎藐，例如ping -6, curl -6, telnet -6等。
使用type函數(shù)哼绑，動態(tài)創(chuàng)建一個新的類HTTPServerV6岩馍，其address_family設置成AF_INET6。這種寫法可以不用顯式繼承而得到一個新的派生類抖韩，不會修改原類的屬性蛀恩。
修改完代碼并保存，執(zhí)行python -m http.server -6 -b ::帽蝶，成功監(jiān)聽IPv6地址赦肋，使用netstat -lpt可以查看。不加-6參數(shù)程序還是原來的用法励稳，堪稱完美佃乘。

tcp6       0      0 [::]:8000               [::]:*                  LISTEN      7252/python3

興致勃勃準備去Github給Python提PR，發(fā)現(xiàn)master上的代碼是這樣的驹尼，原來Python 3.8開始趣避，http.server模塊就已經(jīng)支持IPv6了。

if __name__ == '__main__':
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument('--cgi', action='store_true',
                       help='Run as CGI Server')
    parser.add_argument('--bind', '-b', metavar='ADDRESS',
                        help='Specify alternate bind address '
                             '[default: all interfaces]')
    parser.add_argument('--directory', '-d', default=os.getcwd(),
                        help='Specify alternative directory '
                        '[default:current directory]')
    parser.add_argument('port', action='store',
                        default=8000, type=int,
                        nargs='?',
                        help='Specify alternate port [default: 8000]')
    args = parser.parse_args()
    if args.cgi:
        handler_class = CGIHTTPRequestHandler
    else:
        handler_class = partial(SimpleHTTPRequestHandler,
                                directory=args.directory)

    # ensure dual-stack is not disabled; ref #38907
    class DualStackServer(ThreadingHTTPServer):
        def server_bind(self):
            # suppress exception when protocol is IPv4
            with contextlib.suppress(Exception):
                self.socket.setsockopt(
                    socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, 0)
            return super().server_bind()

    test(
        HandlerClass=handler_class,
        ServerClass=DualStackServer,
        port=args.port,
        bind=args.bind,
    )

它使用了雙棧套接字技術新翎，在Linux上默認啟用雙棧套接字程帕，只需要創(chuàng)建一個基于IPv6的TCP套接字，即可同時支持IPv4和IPv6地啰。如果系統(tǒng)設置了IPV6_V6ONLY選項愁拭，即/proc/sys/net/ipv6/bindv6only不為0，則需要使用setsockopt取消這個選項才能啟用雙棧套接字亏吝。

sock = socket(AF_INET6, SOCK_STREAM, 0);
//enable dual stack if disabled
//int opt = 0;
//setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, &opt, sizeof(opt));

?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者