#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright 2012 Matt Martz
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import csv
import datetime
import errno
import math
import os
import platform
import re
import signal
import socket
import sys
import threading
import time
import timeit
import xml.parsers.expat

try:
    import gzip
    GZIP_BASE = gzip.GzipFile
except ImportError:
    gzip = None
    GZIP_BASE = object

try:
    import psutil
    PSUTIL_AVAILABLE = True
except ImportError:
    psutil = None
    PSUTIL_AVAILABLE = False

try:
    import iwlib
    IWLIB_AVAILABLE = True
except ImportError:
    iwlib = None
    IWLIB_AVAILABLE = False

try:
    import httpx
    import asyncio
    HTTPX_AVAILABLE = True
except ImportError:
    httpx = None
    asyncio = None
    HTTPX_AVAILABLE = False

try:
    import requests
    REQUESTS_AVAILABLE = True
except ImportError:
    requests = None
    REQUESTS_AVAILABLE = False

__version__ = '2.1.4b1'


class FakeShutdownEvent(object):
    """Class to fake a threading.Event.isSet so that users of this module
    are not required to register their own threading.Event()
    """

    @staticmethod
    def isSet():
        "Dummy method to always return false"""
        return False

    is_set = isSet


# Some global variables we use
DEBUG = False
_GLOBAL_DEFAULT_TIMEOUT = object()
PY25PLUS = sys.version_info[:2] >= (2, 5)
PY26PLUS = sys.version_info[:2] >= (2, 6)
PY32PLUS = sys.version_info[:2] >= (3, 2)
PY310PLUS = sys.version_info[:2] >= (3, 10)

# Begin import game to handle Python 2 and Python 3
try:
    import json
except ImportError:
    try:
        import simplejson as json
    except ImportError:
        json = None

try:
    import xml.etree.ElementTree as ET
    try:
        from xml.etree.ElementTree import _Element as ET_Element
    except ImportError:
        pass
except ImportError:
    from xml.dom import minidom as DOM
    from xml.parsers.expat import ExpatError
    ET = None

try:
    from urllib2 import (urlopen, Request, HTTPError, URLError,
                         AbstractHTTPHandler, ProxyHandler,
                         HTTPDefaultErrorHandler, HTTPRedirectHandler,
                         HTTPErrorProcessor, OpenerDirector)
except ImportError:
    from urllib.request import (urlopen, Request, HTTPError, URLError,
                                AbstractHTTPHandler, ProxyHandler,
                                HTTPDefaultErrorHandler, HTTPRedirectHandler,
                                HTTPErrorProcessor, OpenerDirector)

try:
    from httplib import HTTPConnection, BadStatusLine
except ImportError:
    from http.client import HTTPConnection, BadStatusLine

try:
    from httplib import HTTPSConnection
except ImportError:
    try:
        from http.client import HTTPSConnection
    except ImportError:
        HTTPSConnection = None

try:
    from httplib import FakeSocket
except ImportError:
    FakeSocket = None

try:
    from Queue import Queue
except ImportError:
    from queue import Queue

try:
    from urlparse import urlparse
except ImportError:
    from urllib.parse import urlparse

try:
    from urlparse import parse_qs
except ImportError:
    try:
        from urllib.parse import parse_qs
    except ImportError:
        from cgi import parse_qs

try:
    from hashlib import md5
except ImportError:
    from md5 import md5

try:
    from argparse import ArgumentParser as ArgParser
    from argparse import SUPPRESS as ARG_SUPPRESS
    PARSER_TYPE_INT = int
    PARSER_TYPE_STR = str
    PARSER_TYPE_FLOAT = float
except ImportError:
    from optparse import OptionParser as ArgParser
    from optparse import SUPPRESS_HELP as ARG_SUPPRESS
    PARSER_TYPE_INT = 'int'
    PARSER_TYPE_STR = 'string'
    PARSER_TYPE_FLOAT = 'float'

try:
    from cStringIO import StringIO
    BytesIO = None
except ImportError:
    try:
        from StringIO import StringIO
        BytesIO = None
    except ImportError:
        from io import StringIO, BytesIO

try:
    import __builtin__
except ImportError:
    import builtins
    from io import TextIOWrapper, FileIO

    class _Py3Utf8Output(TextIOWrapper):
        """UTF-8 encoded wrapper around stdout for py3, to override
        ASCII stdout
        """
        def __init__(self, f, **kwargs):
            buf = FileIO(f.fileno(), 'w')
            super(_Py3Utf8Output, self).__init__(
                buf,
                encoding='utf8',
                errors='strict'
            )

        def write(self, s):
            super(_Py3Utf8Output, self).write(s)
            self.flush()

    _py3_print = getattr(builtins, 'print')
    try:
        _py3_utf8_stdout = _Py3Utf8Output(sys.stdout)
        _py3_utf8_stderr = _Py3Utf8Output(sys.stderr)
    except OSError:
        # sys.stdout/sys.stderr is not a compatible stdout/stderr object
        # just use it and hope things go ok
        _py3_utf8_stdout = sys.stdout
        _py3_utf8_stderr = sys.stderr

    def to_utf8(v):
        """No-op encode to utf-8 for py3"""
        return v

    def print_(*args, **kwargs):
        """Wrapper function for py3 to print, with a utf-8 encoded stdout"""
        if kwargs.get('file') == sys.stderr:
            kwargs['file'] = _py3_utf8_stderr
        else:
            kwargs['file'] = kwargs.get('file', _py3_utf8_stdout)
        _py3_print(*args, **kwargs)
else:
    del __builtin__

    def to_utf8(v):
        """Encode value to utf-8 if possible for py2"""
        try:
            return v.encode('utf8', 'strict')
        except AttributeError:
            return v

    def print_(*args, **kwargs):
        """The new-style print function for Python 2.4 and 2.5.

        Taken from https://pypi.python.org/pypi/six/

        Modified to set encoding to UTF-8 always, and to flush after write
        """
        fp = kwargs.pop("file", sys.stdout)
        if fp is None:
            return

        def write(data):
            if not isinstance(data, basestring):
                data = str(data)
            # If the file has an encoding, encode unicode with it.
            encoding = 'utf8'  # Always trust UTF-8 for output
            if (isinstance(fp, file) and
                    isinstance(data, unicode) and
                    encoding is not None):
                errors = getattr(fp, "errors", None)
                if errors is None:
                    errors = "strict"
                data = data.encode(encoding, errors)
            fp.write(data)
            fp.flush()
        want_unicode = False
        sep = kwargs.pop("sep", None)
        if sep is not None:
            if isinstance(sep, unicode):
                want_unicode = True
            elif not isinstance(sep, str):
                raise TypeError("sep must be None or a string")
        end = kwargs.pop("end", None)
        if end is not None:
            if isinstance(end, unicode):
                want_unicode = True
            elif not isinstance(end, str):
                raise TypeError("end must be None or a string")
        if kwargs:
            raise TypeError("invalid keyword arguments to print()")
        if not want_unicode:
            for arg in args:
                if isinstance(arg, unicode):
                    want_unicode = True
                    break
        if want_unicode:
            newline = unicode("\n")
            space = unicode(" ")
        else:
            newline = "\n"
            space = " "
        if sep is None:
            sep = space
        if end is None:
            end = newline
        for i, arg in enumerate(args):
            if i:
                write(sep)
            write(arg)
        write(end)

# Exception "constants" to support Python 2 through Python 3
try:
    import ssl
    try:
        CERT_ERROR = (ssl.CertificateError,)
    except AttributeError:
        CERT_ERROR = tuple()

    HTTP_ERRORS = (
        (HTTPError, URLError, socket.error, ssl.SSLError, BadStatusLine) +
        CERT_ERROR
    )
except ImportError:
    ssl = None
    HTTP_ERRORS = (HTTPError, URLError, socket.error, BadStatusLine)

if PY32PLUS:
    etree_iter = ET.Element.iter
elif PY25PLUS:
    etree_iter = ET_Element.getiterator

if PY26PLUS:
    thread_is_alive = threading.Thread.is_alive
else:
    thread_is_alive = threading.Thread.isAlive


def event_is_set(event):
    try:
        return event.is_set()
    except AttributeError:
        return event.isSet()


class SpeedtestException(Exception):
    """Base exception for this module"""


class SpeedtestCLIError(SpeedtestException):
    """Generic exception for raising errors during CLI operation"""


class SpeedtestHTTPError(SpeedtestException):
    """Base HTTP exception for this module"""


class SpeedtestConfigError(SpeedtestException):
    """Configuration XML is invalid"""


class SpeedtestServersError(SpeedtestException):
    """Servers XML is invalid"""


class ConfigRetrievalError(SpeedtestHTTPError):
    """Could not retrieve config.php"""


class ServersRetrievalError(SpeedtestHTTPError):
    """Could not retrieve speedtest-servers.php"""


class InvalidServerIDType(SpeedtestException):
    """Server ID used for filtering was not an integer"""


class NoMatchedServers(SpeedtestException):
    """No servers matched when filtering"""


class SpeedtestMiniConnectFailure(SpeedtestException):
    """Could not connect to the provided speedtest mini server"""


class InvalidSpeedtestMiniServer(SpeedtestException):
    """Server provided as a speedtest mini server does not actually appear
    to be a speedtest mini server
    """


class ShareResultsConnectFailure(SpeedtestException):
    """Could not connect to speedtest.net API to POST results"""


class ShareResultsSubmitFailure(SpeedtestException):
    """Unable to successfully POST results to speedtest.net API after
    connection
    """


class SpeedtestUploadTimeout(SpeedtestException):
    """testlength configuration reached during upload
    Used to ensure the upload halts when no additional data should be sent
    """


class SpeedtestBestServerFailure(SpeedtestException):
    """Unable to determine best server"""


class SpeedtestMissingBestServer(SpeedtestException):
    """get_best_server not called or not able to determine best server"""


def create_connection(address, timeout=_GLOBAL_DEFAULT_TIMEOUT,
                      source_address=None):
    """Connect to *address* and return the socket object.

    Convenience function.  Connect to *address* (a 2-tuple ``(host,
    port)``) and return the socket object.  Passing the optional
    *timeout* parameter will set the timeout on the socket instance
    before attempting to connect.  If no *timeout* is supplied, the
    global default timeout setting returned by :func:`getdefaulttimeout`
    is used.  If *source_address* is set it must be a tuple of (host, port)
    for the socket to bind as a source address before making the connection.
    An host of '' or port 0 tells the OS to use the default.

    Largely vendored from Python 2.7, modified to work with Python 2.4
    """

    host, port = address
    err = None
    for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
        af, socktype, proto, canonname, sa = res
        sock = None
        try:
            sock = socket.socket(af, socktype, proto)
            if timeout is not _GLOBAL_DEFAULT_TIMEOUT:
                sock.settimeout(float(timeout))
            if source_address:
                sock.bind(source_address)
            sock.connect(sa)
            return sock

        except socket.error:
            err = get_exception()
            if sock is not None:
                sock.close()

    if err is not None:
        raise err
    else:
        raise socket.error("getaddrinfo returns an empty list")


class SpeedtestHTTPConnection(HTTPConnection):
    """Custom HTTPConnection to support source_address across
    Python 2.4 - Python 3
    """
    def __init__(self, *args, **kwargs):
        source_address = kwargs.pop('source_address', None)
        timeout = kwargs.pop('timeout', 10)

        self._tunnel_host = None

        HTTPConnection.__init__(self, *args, **kwargs)

        self.source_address = source_address
        self.timeout = timeout

    def connect(self):
        """Connect to the host and port specified in __init__."""
        try:
            self.sock = socket.create_connection(
                (self.host, self.port),
                self.timeout,
                self.source_address
            )
        except (AttributeError, TypeError):
            self.sock = create_connection(
                (self.host, self.port),
                self.timeout,
                self.source_address
            )

        if self._tunnel_host:
            self._tunnel()


if HTTPSConnection:
    class SpeedtestHTTPSConnection(HTTPSConnection):
        """Custom HTTPSConnection to support source_address across
        Python 2.4 - Python 3
        """
        default_port = 443

        def __init__(self, *args, **kwargs):
            source_address = kwargs.pop('source_address', None)
            timeout = kwargs.pop('timeout', 10)

            self._tunnel_host = None

            HTTPSConnection.__init__(self, *args, **kwargs)

            self.timeout = timeout
            self.source_address = source_address

        def connect(self):
            "Connect to a host on a given (SSL) port."
            try:
                self.sock = socket.create_connection(
                    (self.host, self.port),
                    self.timeout,
                    self.source_address
                )
            except (AttributeError, TypeError):
                self.sock = create_connection(
                    (self.host, self.port),
                    self.timeout,
                    self.source_address
                )

            if self._tunnel_host:
                self._tunnel()

            if ssl:
                try:
                    kwargs = {}
                    if hasattr(ssl, 'SSLContext'):
                        if self._tunnel_host:
                            kwargs['server_hostname'] = self._tunnel_host
                        else:
                            kwargs['server_hostname'] = self.host
                    self.sock = self._context.wrap_socket(self.sock, **kwargs)
                except AttributeError:
                    self.sock = ssl.wrap_socket(self.sock)
                    try:
                        self.sock.server_hostname = self.host
                    except AttributeError:
                        pass
            elif FakeSocket:
                # Python 2.4/2.5 support
                try:
                    self.sock = FakeSocket(self.sock, socket.ssl(self.sock))
                except AttributeError:
                    raise SpeedtestException(
                        'This version of Python does not support HTTPS/SSL '
                        'functionality'
                    )
            else:
                raise SpeedtestException(
                    'This version of Python does not support HTTPS/SSL '
                    'functionality'
                )


def _build_connection(connection, source_address, timeout, context=None):
    """Cross Python 2.4 - Python 3 callable to build an ``HTTPConnection`` or
    ``HTTPSConnection`` with the args we need

    Called from ``http(s)_open`` methods of ``SpeedtestHTTPHandler`` or
    ``SpeedtestHTTPSHandler``
    """
    def inner(host, **kwargs):
        kwargs.update({
            'source_address': source_address,
            'timeout': timeout
        })
        if context:
            kwargs['context'] = context
        return connection(host, **kwargs)
    return inner


class SpeedtestHTTPHandler(AbstractHTTPHandler):
    """Custom ``HTTPHandler`` that can build a ``HTTPConnection`` with the
    args we need for ``source_address`` and ``timeout``
    """
    def __init__(self, debuglevel=0, source_address=None, timeout=10):
        AbstractHTTPHandler.__init__(self, debuglevel)
        self.source_address = source_address
        self.timeout = timeout

    def http_open(self, req):
        return self.do_open(
            _build_connection(
                SpeedtestHTTPConnection,
                self.source_address,
                self.timeout
            ),
            req
        )

    http_request = AbstractHTTPHandler.do_request_


class SpeedtestHTTPSHandler(AbstractHTTPHandler):
    """Custom ``HTTPSHandler`` that can build a ``HTTPSConnection`` with the
    args we need for ``source_address`` and ``timeout``
    """
    def __init__(self, debuglevel=0, context=None, source_address=None,
                 timeout=10):
        AbstractHTTPHandler.__init__(self, debuglevel)
        self._context = context
        self.source_address = source_address
        self.timeout = timeout

    def https_open(self, req):
        return self.do_open(
            _build_connection(
                SpeedtestHTTPSConnection,
                self.source_address,
                self.timeout,
                context=self._context,
            ),
            req
        )

    https_request = AbstractHTTPHandler.do_request_


class HTTP2Downloader(threading.Thread):
    """Thread class for retrieving a URL using HTTP/2 with TCP monitoring"""

    def __init__(self, i, url, start, timeout, shutdown_event=None, monitor_tcp=False):
        threading.Thread.__init__(self)
        self.url = url
        self.result = [0]
        self.starttime = start
        self.timeout = timeout
        self.i = i
        self.monitor_tcp = monitor_tcp

        # TCP monitoring data
        self.timing_data = []  # List of (timestamp, bytes_received) tuples
        self.tcp_stats = {}
        self.connection_events = []  # List of (timestamp, event_type, details) tuples
        self.pauses = []  # List of pause periods
        self.fastest_periods = []  # List of fastest transfer periods

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

    def _get_tcp_stats(self, sock):
        """Get TCP statistics from socket if available"""
        try:
            # Get TCP_INFO if available (Linux)
            import struct
            TCP_INFO = 11  # SOL_TCP = 6, TCP_INFO = 11
            info = sock.getsockopt(socket.IPPROTO_TCP, TCP_INFO, 128)
            if len(info) >= 32:  # Basic TCP_INFO structure
                # Parse basic TCP info (this is platform-specific)
                # For now, we'll just return basic socket info
                return {
                    'bytes_sent': sock.getsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF),
                    'bytes_received': sock.getsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF),
                }
        except (OSError, AttributeError, struct.error):
            pass
        return {}

    def _analyze_timing(self):
        """Analyze timing data to find fastest periods and pauses"""
        if len(self.timing_data) < 2:
            return

        # Calculate transfer rates between consecutive measurements
        rates = []
        for i in range(1, len(self.timing_data)):
            prev_time, prev_bytes = self.timing_data[i-1]
            curr_time, curr_bytes = self.timing_data[i]

            time_diff = curr_time - prev_time
            bytes_diff = curr_bytes - prev_bytes

            if time_diff > 0:
                rate = bytes_diff / time_diff  # bytes per second
                rates.append((prev_time, curr_time, rate, bytes_diff))

        if not rates:
            return

        # Find fastest periods (top 25% of rates)
        rates.sort(key=lambda x: x[2], reverse=True)
        top_25_percent = max(1, len(rates) // 4)
        self.fastest_periods = rates[:top_25_percent]

        # Find pauses (periods with very low or zero transfer rates)
        avg_rate = sum(r[2] for r in rates) / len(rates)
        pause_threshold = avg_rate * 0.1  # 10% of average rate

        for start_time, end_time, rate, bytes_diff in rates:
            if rate < pause_threshold and (end_time - start_time) > 0.1:  # Pause > 100ms
                self.pauses.append({
                    'start_time': start_time,
                    'end_time': end_time,
                    'duration': end_time - start_time,
                    'bytes_transferred': bytes_diff,
                    'rate': rate
                })

    def run(self):
        try:
            if (timeit.default_timer() - self.starttime) <= self.timeout:
                # Use httpx with HTTP/2 support
                with httpx.Client(http2=True, timeout=self.timeout) as client:
                    # Record connection start
                    if self.monitor_tcp:
                        self.connection_events.append((
                            timeit.default_timer(),
                            'connection_start',
                            {'url': self.url}
                        ))

                    with client.stream('GET', self.url) as response:
                        response.raise_for_status()

                        # Record response start
                        if self.monitor_tcp:
                            self.connection_events.append((
                                timeit.default_timer(),
                                'response_start',
                                {'status_code': response.status_code}
                            ))

                        last_time = timeit.default_timer()
                        last_bytes = 0

                        for chunk in response.iter_bytes(10240):
                            if (event_is_set(self._shutdown_event) or
                                (timeit.default_timer() - self.starttime) > self.timeout):
                                break
                            if not chunk:
                                break

                            current_time = timeit.default_timer()
                            current_bytes = last_bytes + len(chunk)

                            # Record timing data
                            if self.monitor_tcp:
                                self.timing_data.append((current_time, current_bytes))

                                # Check for potential pauses (no data for > 100ms)
                                if current_time - last_time > 0.1:
                                    self.connection_events.append((
                                        current_time,
                                        'potential_pause',
                                        {
                                            'duration': current_time - last_time,
                                            'bytes_since_last': len(chunk)
                                        }
                                    ))

                            self.result.append(len(chunk))
                            last_time = current_time
                            last_bytes = current_bytes

                        # Record connection end
                        if self.monitor_tcp:
                            self.connection_events.append((
                                timeit.default_timer(),
                                'connection_end',
                                {'total_bytes': sum(self.result)}
                            ))

                            # Analyze timing data
                            self._analyze_timing()

        except Exception as e:
            if self.monitor_tcp:
                self.connection_events.append((
                    timeit.default_timer(),
                    'connection_error',
                    {'error': str(e)}
                ))
            pass


class HTTP2Uploader(threading.Thread):
    """Thread class for uploading data using HTTP/2 with TCP monitoring"""

    def __init__(self, i, url, data, start, size, timeout, shutdown_event=None, monitor_tcp=False):
        threading.Thread.__init__(self)
        self.url = url
        self.data = data
        self.data.start = self.starttime = start  # Set the start time for the data object
        self.size = size
        self.timeout = timeout
        self.i = i
        self.result = 0
        self.monitor_tcp = monitor_tcp

        # TCP monitoring data
        self.timing_data = []  # List of (timestamp, bytes_sent) tuples
        self.tcp_stats = {}
        self.connection_events = []  # List of (timestamp, event_type, details) tuples
        self.pauses = []  # List of pause periods
        self.fastest_periods = []  # List of fastest transfer periods

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

    def _analyze_timing(self):
        """Analyze timing data to find fastest periods and pauses"""
        if len(self.timing_data) < 2:
            return

        # Calculate transfer rates between consecutive measurements
        rates = []
        for i in range(1, len(self.timing_data)):
            prev_time, prev_bytes = self.timing_data[i-1]
            curr_time, curr_bytes = self.timing_data[i]

            time_diff = curr_time - prev_time
            bytes_diff = curr_bytes - prev_bytes

            if time_diff > 0:
                rate = bytes_diff / time_diff  # bytes per second
                rates.append((prev_time, curr_time, rate, bytes_diff))

        if not rates:
            return

        # Find fastest periods (top 25% of rates)
        rates.sort(key=lambda x: x[2], reverse=True)
        top_25_percent = max(1, len(rates) // 4)
        self.fastest_periods = rates[:top_25_percent]

        # Find pauses (periods with very low or zero transfer rates)
        avg_rate = sum(r[2] for r in rates) / len(rates)
        pause_threshold = avg_rate * 0.1  # 10% of average rate

        for start_time, end_time, rate, bytes_diff in rates:
            if rate < pause_threshold and (end_time - start_time) > 0.1:  # Pause > 100ms
                self.pauses.append({
                    'start_time': start_time,
                    'end_time': end_time,
                    'duration': end_time - start_time,
                    'bytes_transferred': bytes_diff,
                    'rate': rate
                })

    def run(self):
        try:
            if ((timeit.default_timer() - self.starttime) <= self.timeout and
                    not event_is_set(self._shutdown_event)):
                # Use httpx with HTTP/2 support
                with httpx.Client(http2=True, timeout=self.timeout) as client:
                    # Record connection start
                    if self.monitor_tcp:
                        self.connection_events.append((
                            timeit.default_timer(),
                            'upload_start',
                            {'url': self.url, 'size': self.size}
                        ))

                    # Read data in chunks like the original HTTPUploader
                    data_chunks = []
                    remaining = self.size
                    bytes_sent = 0
                    last_time = timeit.default_timer()

                    while remaining > 0 and not event_is_set(self._shutdown_event):
                        chunk_size = min(10240, remaining)
                        chunk = self.data.read(chunk_size)
                        if not chunk:
                            break
                        data_chunks.append(chunk)
                        remaining -= len(chunk)
                        bytes_sent += len(chunk)

                        # Record timing data
                        if self.monitor_tcp:
                            current_time = timeit.default_timer()
                            self.timing_data.append((current_time, bytes_sent))

                            # Check for potential pauses
                            if current_time - last_time > 0.1:
                                self.connection_events.append((
                                    current_time,
                                    'potential_pause',
                                    {
                                        'duration': current_time - last_time,
                                        'bytes_since_last': len(chunk)
                                    }
                                ))
                            last_time = current_time

                    # Combine all chunks
                    upload_data = b''.join(data_chunks)

                    # Record upload start
                    if self.monitor_tcp:
                        self.connection_events.append((
                            timeit.default_timer(),
                            'upload_transmit_start',
                            {'data_size': len(upload_data)}
                        ))

                    response = client.post(self.url, content=upload_data)
                    response.raise_for_status()

                    # Record upload completion
                    if self.monitor_tcp:
                        self.connection_events.append((
                            timeit.default_timer(),
                            'upload_complete',
                            {'status_code': response.status_code}
                        ))

                        # Analyze timing data
                        self._analyze_timing()

                    self.result = sum(self.data.total)
        except (IOError, SpeedtestUploadTimeout):
            self.result = sum(self.data.total)
        except Exception as e:
            if self.monitor_tcp:
                self.connection_events.append((
                    timeit.default_timer(),
                    'upload_error',
                    {'error': str(e)}
                ))
            self.result = 0


def http_request(url, headers=None, timeout=10, http2=False):
    """Make an HTTP request using the best available library

    Priority: httpx (with HTTP/2) > requests > urllib
    """
    if not headers:
        headers = {}

    # Try httpx first (with HTTP/2 support if requested)
    if HTTPX_AVAILABLE:
        try:
            with httpx.Client(http2=http2, timeout=timeout) as client:
                response = client.get(url, headers=headers)
                response.raise_for_status()
                return response.content, response.status_code, None
        except Exception as e:
            if not REQUESTS_AVAILABLE:
                raise e  # If requests not available, re-raise httpx error

    # Try requests as fallback
    if REQUESTS_AVAILABLE:
        try:
            response = requests.get(url, headers=headers, timeout=timeout)
            response.raise_for_status()
            return response.content, response.status_code, None
        except Exception as e:
            if not HTTPX_AVAILABLE:
                raise e  # If httpx not available, re-raise requests error

    # Fallback to urllib (original behavior)
    request = build_request(url, headers=headers)
    uh, e = catch_request(request)
    if e:
        raise e

    try:
        stream = get_response_stream(uh)
        content = stream.read()
        stream.close()
        uh.close()
        return content, int(uh.code), None
    except Exception as e:
        uh.close()
        raise e


def build_opener(source_address=None, timeout=10):
    """Function similar to ``urllib2.build_opener`` that will build
    an ``OpenerDirector`` with the explicit handlers we want,
    ``source_address`` for binding, ``timeout`` and our custom
    `User-Agent`
    """

    printer('Timeout set to %d' % timeout, debug=True)

    if source_address:
        source_address_tuple = (source_address, 0)
        printer('Binding to source address: %r' % (source_address_tuple,),
                debug=True)
    else:
        source_address_tuple = None

    handlers = [
        ProxyHandler(),
        SpeedtestHTTPHandler(source_address=source_address_tuple,
                             timeout=timeout),
        SpeedtestHTTPSHandler(source_address=source_address_tuple,
                              timeout=timeout),
        HTTPDefaultErrorHandler(),
        HTTPRedirectHandler(),
        HTTPErrorProcessor()
    ]

    opener = OpenerDirector()
    opener.addheaders = [('User-agent', build_user_agent())]

    for handler in handlers:
        opener.add_handler(handler)

    return opener


class GzipDecodedResponse(GZIP_BASE):
    """A file-like object to decode a response encoded with the gzip
    method, as described in RFC 1952.

    Largely copied from ``xmlrpclib``/``xmlrpc.client`` and modified
    to work for py2.4-py3
    """
    def __init__(self, response):
        # response doesn't support tell() and read(), required by
        # GzipFile
        if not gzip:
            raise SpeedtestHTTPError('HTTP response body is gzip encoded, '
                                     'but gzip support is not available')
        IO = BytesIO or StringIO
        self.io = IO()
        while 1:
            chunk = response.read(1024)
            if len(chunk) == 0:
                break
            self.io.write(chunk)
        self.io.seek(0)
        gzip.GzipFile.__init__(self, mode='rb', fileobj=self.io)

    def close(self):
        try:
            gzip.GzipFile.close(self)
        finally:
            self.io.close()


def get_exception():
    """Helper function to work with py2.4-py3 for getting the current
    exception in a try/except block
    """
    return sys.exc_info()[1]


def distance(origin, destination):
    """Determine distance between 2 sets of [lat,lon] in km"""

    lat1, lon1 = origin
    lat2, lon2 = destination
    radius = 6371  # km

    dlat = math.radians(lat2 - lat1)
    dlon = math.radians(lon2 - lon1)
    a = (math.sin(dlat / 2) * math.sin(dlat / 2) +
         math.cos(math.radians(lat1)) *
         math.cos(math.radians(lat2)) * math.sin(dlon / 2) *
         math.sin(dlon / 2))
    c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))
    d = radius * c

    return d


def get_wifi_speed_from_iwlib(interface):
    """Get Wi-Fi speed from iwlib library (preferred method)

    Returns a dictionary with tx_rate and rx_rate in Mbps if available, otherwise None.
    """
    if not IWLIB_AVAILABLE:
        return None

    try:
        wifi_info = iwlib.get_iwconfig(interface)
        if 'BitRate' in wifi_info:
            # iwlib only provides one bitrate, typically TX rate
            bitrate_str = wifi_info['BitRate'].decode() if isinstance(wifi_info['BitRate'], bytes) else wifi_info['BitRate']
            import re
            match = re.search(r'([0-9.]+)\s*Mb/s', bitrate_str)
            if match:
                tx_rate = float(match.group(1))
                # iwlib doesn't provide separate RX rate, so we'll use TX rate for both
                return {'tx_rate': tx_rate, 'rx_rate': tx_rate}
        return None
    except Exception:
        return None


def get_wifi_speed_from_sysfs(interface):
    """Get Wi-Fi speed from /sys filesystem

    Returns the speed in Mbps if available, otherwise None.
    """
    try:
        # Try the speed file first
        speed_file = f'/sys/class/net/{interface}/speed'
        with open(speed_file, 'r') as f:
            speed = int(f.read().strip())
            if speed > 0:
                return speed
    except (FileNotFoundError, ValueError, OSError):
        pass

    # Try to get speed from wireless phy if available
    try:
        phy_link = f'/sys/class/net/{interface}/phy80211'
        if os.path.exists(phy_link):
            phy_path = os.path.realpath(phy_link)
            # Look for speed information in the phy directory
            # This is more complex and may not be available on all systems
            pass
    except Exception:
        pass

    return None


def get_wifi_speed_from_iw(interface):
    """Get Wi-Fi speed from iw command

    Returns a dictionary with tx_rate and rx_rate in Mbps if available, otherwise None.
    """
    try:
        import subprocess
        result = subprocess.run(['iw', 'dev', interface, 'link'],
                              capture_output=True, text=True, timeout=5)
        if result.returncode == 0:
            tx_rate = None
            rx_rate = None

            for line in result.stdout.split('\n'):
                # Look for TX bitrate
                if 'tx bitrate:' in line:
                    import re
                    match = re.search(r'tx bitrate:\s*([0-9.]+)\s*MBit/s', line)
                    if match:
                        tx_rate = float(match.group(1))

                # Look for RX bitrate
                elif 'rx bitrate:' in line:
                    import re
                    match = re.search(r'rx bitrate:\s*([0-9.]+)\s*MBit/s', line)
                    if match:
                        rx_rate = float(match.group(1))

            # Return both rates if we found at least one
            if tx_rate is not None or rx_rate is not None:
                return {'tx_rate': tx_rate, 'rx_rate': rx_rate}
        return None
    except Exception:
        return None


def get_wifi_speed_from_iwconfig(interface):
    """Get Wi-Fi speed from iwconfig command (fallback)

    Returns a dictionary with tx_rate and rx_rate in Mbps if available, otherwise None.
    """
    try:
        import subprocess
        result = subprocess.run(['iwconfig', interface],
                              capture_output=True, text=True, timeout=5)
        if result.returncode == 0:
            # Look for "Bit Rate=" pattern in the output
            for line in result.stdout.split('\n'):
                if 'Bit Rate=' in line:
                    # Extract the bit rate value
                    import re
                    match = re.search(r'Bit Rate=([0-9.]+)\s*Mb/s', line)
                    if match:
                        rate = float(match.group(1))
                        # iwconfig typically shows TX rate, use same for both
                        return {'tx_rate': rate, 'rx_rate': rate}
        return None
    except Exception:
        return None


def get_wifi_speed(interface):
    """Get Wi-Fi speed using the best available method

    Tries multiple methods in order of preference:
    1. iw command (most detailed, provides separate TX/RX rates)
    2. iwlib library (preferred, no subprocess, most reliable)
    3. /sys filesystem (fastest, no subprocess)
    4. iwconfig command (fallback, older systems)

    Returns a dictionary with tx_rate and rx_rate in Mbps if available, otherwise None.
    """
    # Try iw command first (most detailed, provides separate TX/RX rates)
    speeds = get_wifi_speed_from_iw(interface)
    if speeds is not None:
        return speeds

    # Try iwlib library (preferred method, no subprocess)
    speeds = get_wifi_speed_from_iwlib(interface)
    if speeds is not None:
        return speeds

    # Try /sys filesystem (fastest, no subprocess)
    speed = get_wifi_speed_from_sysfs(interface)
    if speed is not None and speed > 0:
        return {'tx_rate': speed, 'rx_rate': speed}

    # Fallback to iwconfig (older systems)
    speeds = get_wifi_speed_from_iwconfig(interface)
    if speeds is not None:
        return speeds

    return None


def get_network_interface_info():
    """Get information about the active network interface

    Returns a dictionary with interface information if psutil is available,
    otherwise returns None.
    """
    if not PSUTIL_AVAILABLE:
        return None

    try:
        # Get network interface statistics and addresses
        net_stats = psutil.net_if_stats()
        net_addrs = psutil.net_if_addrs()

        # Find the active interface (up and with IPv4 address)
        # Prefer non-loopback interfaces
        candidates = []

        for interface, stats in net_stats.items():
            if stats.isup:
                # Check if interface has an IPv4 address
                iface_addrs = net_addrs.get(interface, [])
                has_ipv4 = any(addr.family == socket.AF_INET for addr in iface_addrs)

                if has_ipv4:
                    # Determine interface type based on naming conventions
                    interface_type = determine_interface_type(interface)

                    # Skip loopback interfaces unless no other option
                    is_loopback = interface_type == 'Loopback'

                    # Get speed - try enhanced methods for Wi-Fi interfaces if psutil reports 0
                    speed = stats.speed
                    tx_rate = None
                    rx_rate = None

                    if interface_type == 'Wi-Fi' and speed == 0:
                        wifi_speeds = get_wifi_speed(interface)
                        if wifi_speeds is not None:
                            tx_rate = wifi_speeds.get('tx_rate')
                            rx_rate = wifi_speeds.get('rx_rate')
                            # Use TX rate as the primary speed for compatibility
                            speed = tx_rate if tx_rate is not None else rx_rate

                    candidates.append({
                        'interface': interface,
                        'type': interface_type,
                        'speed': speed,
                        'tx_rate': tx_rate,
                        'rx_rate': rx_rate,
                        'duplex': stats.duplex,
                        'mtu': stats.mtu,
                        'is_up': stats.isup,
                        'is_loopback': is_loopback
                    })

        if not candidates:
            return None

        # Prefer non-loopback interfaces, then by speed (higher is better)
        candidates.sort(key=lambda x: (x['is_loopback'], -x['speed']))

        # Return the best candidate, removing the temporary is_loopback key
        result = candidates[0]
        del result['is_loopback']
        return result

    except Exception:
        # If any error occurs, return None to gracefully handle it
        return None


def determine_interface_type(interface_name):
    """Determine if an interface is Ethernet or Wi-Fi based on naming conventions"""
    if not interface_name:
        return 'Unknown'

    interface_lower = interface_name.lower()

    # Common Wi-Fi interface name patterns
    wifi_patterns = ['wlan', 'wifi', 'wi-fi', 'wireless', 'ath', 'ra']
    for pattern in wifi_patterns:
        if pattern in interface_lower:
            return 'Wi-Fi'

    # Common Ethernet interface name patterns
    ethernet_patterns = ['eth', 'en', 'ethernet', 'em', 'p', 'bond']
    for pattern in ethernet_patterns:
        if interface_lower.startswith(pattern):
            return 'Ethernet'

    # Check for loopback
    if interface_lower in ['lo', 'loopback']:
        return 'Loopback'

    return 'Unknown'


def build_user_agent():
    """Build a Mozilla/5.0 compatible User-Agent string"""

    ua_tuple = (
        'Mozilla/5.0',
        '(%s; U; %s; en-us)' % (platform.platform(),
                                platform.architecture()[0]),
        'Python/%s' % platform.python_version(),
        '(KHTML, like Gecko)',
        'speedtest-cli/%s' % __version__
    )
    user_agent = ' '.join(ua_tuple)
    printer('User-Agent: %s' % user_agent, debug=True)
    return user_agent


def build_request(url, data=None, headers=None, bump='0', secure=False):
    """Build a urllib2 request object

    This function automatically adds a User-Agent header to all requests

    """

    if not headers:
        headers = {}

    if url[0] == ':':
        scheme = ('http', 'https')[bool(secure)]
        schemed_url = '%s%s' % (scheme, url)
    else:
        schemed_url = url

    if '?' in url:
        delim = '&'
    else:
        delim = '?'

    # WHO YOU GONNA CALL? CACHE BUSTERS!
    final_url = '%s%sx=%s.%s' % (schemed_url, delim,
                                 int(timeit.time.time() * 1000),
                                 bump)

    headers.update({
        'Cache-Control': 'no-cache',
    })

    printer('%s %s' % (('GET', 'POST')[bool(data)], final_url),
            debug=True)

    return Request(final_url, data=data, headers=headers)


def catch_request(request, opener=None):
    """Helper function to catch common exceptions encountered when
    establishing a connection with a HTTP/HTTPS request

    """

    if opener:
        _open = opener.open
    else:
        _open = urlopen

    try:
        uh = _open(request)
        if request.get_full_url() != uh.geturl():
            printer('Redirected to %s' % uh.geturl(), debug=True)
        return uh, False
    except HTTP_ERRORS:
        e = get_exception()
        return None, e


def get_response_stream(response):
    """Helper function to return either a Gzip reader if
    ``Content-Encoding`` is ``gzip`` otherwise the response itself

    """

    try:
        getheader = response.headers.getheader
    except AttributeError:
        getheader = response.getheader

    if getheader('content-encoding') == 'gzip':
        return GzipDecodedResponse(response)

    return response


def get_attributes_by_tag_name(dom, tag_name):
    """Retrieve an attribute from an XML document and return it in a
    consistent format

    Only used with xml.dom.minidom, which is likely only to be used
    with python versions older than 2.5
    """
    elem = dom.getElementsByTagName(tag_name)[0]
    return dict(list(elem.attributes.items()))


def print_dots(shutdown_event):
    """Built in callback function used by Thread classes for printing
    status
    """
    def inner(current, total, start=False, end=False):
        if event_is_set(shutdown_event):
            return

        sys.stdout.write('.')
        if current + 1 == total and end is True:
            sys.stdout.write('\n')
        sys.stdout.flush()
    return inner


def do_nothing(*args, **kwargs):
    pass


class HTTPDownloader(threading.Thread):
    """Thread class for retrieving a URL"""

    def __init__(self, i, request, start, timeout, opener=None,
                 shutdown_event=None):
        threading.Thread.__init__(self)
        self.request = request
        self.result = [0]
        self.starttime = start
        self.timeout = timeout
        self.i = i
        if opener:
            self._opener = opener.open
        else:
            self._opener = urlopen

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

    def run(self):
        try:
            if (timeit.default_timer() - self.starttime) <= self.timeout:
                f = self._opener(self.request)
                while (not event_is_set(self._shutdown_event) and
                        (timeit.default_timer() - self.starttime) <=
                        self.timeout):
                    self.result.append(len(f.read(10240)))
                    if self.result[-1] == 0:
                        break
                f.close()
        except IOError:
            pass
        except HTTP_ERRORS:
            pass


class HTTPUploaderData(object):
    """File like object to improve cutting off the upload once the timeout
    has been reached
    """

    def __init__(self, length, start, timeout, shutdown_event=None):
        self.length = length
        self.start = start
        self.timeout = timeout

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

        self._data = None

        self.total = [0]

    def pre_allocate(self):
        chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'
        multiplier = int(round(int(self.length) / 36.0))
        IO = BytesIO or StringIO
        try:
            self._data = IO(
                ('content1=%s' %
                 (chars * multiplier)[0:int(self.length) - 9]
                 ).encode()
            )
        except MemoryError:
            raise SpeedtestCLIError(
                'Insufficient memory to pre-allocate upload data. Please '
                'use --no-pre-allocate'
            )

    @property
    def data(self):
        if not self._data:
            self.pre_allocate()
        return self._data

    def read(self, n=10240):
        if ((timeit.default_timer() - self.start) <= self.timeout and
                not event_is_set(self._shutdown_event)):
            chunk = self.data.read(n)
            self.total.append(len(chunk))
            return chunk
        else:
            raise SpeedtestUploadTimeout()

    def __len__(self):
        return self.length


class HTTPUploader(threading.Thread):
    """Thread class for putting a URL"""

    def __init__(self, i, request, start, size, timeout, opener=None,
                 shutdown_event=None):
        threading.Thread.__init__(self)
        self.request = request
        self.request.data.start = self.starttime = start
        self.size = size
        self.result = 0
        self.timeout = timeout
        self.i = i

        if opener:
            self._opener = opener.open
        else:
            self._opener = urlopen

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

    def run(self):
        request = self.request
        try:
            if ((timeit.default_timer() - self.starttime) <= self.timeout and
                    not event_is_set(self._shutdown_event)):
                try:
                    f = self._opener(request)
                except TypeError:
                    # PY24 expects a string or buffer
                    # This also causes issues with Ctrl-C, but we will concede
                    # for the moment that Ctrl-C on PY24 isn't immediate
                    request = build_request(self.request.get_full_url(),
                                            data=request.data.read(self.size))
                    f = self._opener(request)
                f.read(11)
                f.close()
                self.result = sum(self.request.data.total)
            else:
                self.result = 0
        except (IOError, SpeedtestUploadTimeout):
            self.result = sum(self.request.data.total)
        except HTTP_ERRORS:
            self.result = 0


class SpeedtestResults(object):
    """Class for holding the results of a speedtest, including:

    Download speed
    Upload speed
    Ping/Latency to test server
    Data about server that the test was run against

    Additionally this class can return a result data as a dictionary or CSV,
    as well as submit a POST of the result data to the speedtest.net API
    to get a share results image link.
    """

    def __init__(self, download=0, upload=0, ping=0, server=None, client=None,
                 opener=None, secure=False):
        self.download = download
        self.upload = upload
        self.ping = ping
        if server is None:
            self.server = {}
        else:
            self.server = server
        self.client = client or {}

        self._share = None
        self.timestamp = '%sZ' % datetime.datetime.now(datetime.UTC).isoformat()
        self.bytes_received = 0
        self.bytes_sent = 0

        if opener:
            self._opener = opener
        else:
            self._opener = build_opener()

        self._secure = secure

        # Get network interface information if available
        self.interface_info = get_network_interface_info()

    def __repr__(self):
        return repr(self.dict())

    def share(self):
        """POST data to the speedtest.net API to obtain a share results
        link
        """

        if self._share:
            return self._share

        download = int(round(self.download / 1000.0, 0))
        ping = int(round(self.ping, 0))
        upload = int(round(self.upload / 1000.0, 0))

        # Build the request to send results back to speedtest.net
        # We use a list instead of a dict because the API expects parameters
        # in a certain order
        api_data = [
            'recommendedserverid=%s' % self.server['id'],
            'ping=%s' % ping,
            'screenresolution=',
            'promo=',
            'download=%s' % download,
            'screendpi=',
            'upload=%s' % upload,
            'testmethod=http',
            'hash=%s' % md5(('%s-%s-%s-%s' %
                             (ping, upload, download, '297aae72'))
                            .encode()).hexdigest(),
            'touchscreen=none',
            'startmode=pingselect',
            'accuracy=1',
            'bytesreceived=%s' % self.bytes_received,
            'bytessent=%s' % self.bytes_sent,
            'serverid=%s' % self.server['id'],
        ]

        headers = {'Referer': 'http://c.speedtest.net/flash/speedtest.swf'}
        request = build_request('://www.speedtest.net/api/api.php',
                                data='&'.join(api_data).encode(),
                                headers=headers, secure=self._secure)
        f, e = catch_request(request, opener=self._opener)
        if e:
            raise ShareResultsConnectFailure(e)

        response = f.read()
        code = f.code
        f.close()

        if int(code) != 200:
            raise ShareResultsSubmitFailure('Could not submit results to '
                                            'speedtest.net')

        qsargs = parse_qs(response.decode())
        resultid = qsargs.get('resultid')
        if not resultid or len(resultid) != 1:
            raise ShareResultsSubmitFailure('Could not submit results to '
                                            'speedtest.net')

        self._share = 'http://www.speedtest.net/result/%s.png' % resultid[0]

        return self._share

    def dict(self):
        """Return dictionary of result data"""

        result = {
            'download': self.download,
            'upload': self.upload,
            'ping': self.ping,
            'server': self.server,
            'timestamp': self.timestamp,
            'bytes_sent': self.bytes_sent,
            'bytes_received': self.bytes_received,
            'share': self._share,
            'client': self.client,
        }

        # Add interface information if available
        if self.interface_info:
            result['interface'] = self.interface_info

        return result

    @staticmethod
    def csv_header(delimiter=','):
        """Return CSV Headers"""

        row = ['Server ID', 'Sponsor', 'Server Name', 'Timestamp', 'Distance',
               'Ping', 'Download', 'Upload', 'Share', 'IP Address', 'Interface',
               'Interface Type', 'Interface Speed', 'TX Rate', 'RX Rate']
        out = StringIO()
        writer = csv.writer(out, delimiter=delimiter, lineterminator='')
        writer.writerow([to_utf8(v) for v in row])
        return out.getvalue()

    def csv(self, delimiter=','):
        """Return data in CSV format"""

        data = self.dict()
        out = StringIO()
        writer = csv.writer(out, delimiter=delimiter, lineterminator='')

        # Extract interface information
        interface_name = ''
        interface_type = ''
        interface_speed = ''
        tx_rate = ''
        rx_rate = ''
        if self.interface_info:
            interface_name = self.interface_info.get('interface', '')
            interface_type = self.interface_info.get('type', '')
            interface_speed = self.interface_info.get('speed', '')
            tx_rate = self.interface_info.get('tx_rate', '')
            rx_rate = self.interface_info.get('rx_rate', '')

        row = [data['server']['id'], data['server']['sponsor'],
               data['server']['name'], data['timestamp'],
               data['server']['d'], data['ping'], data['download'],
               data['upload'], self._share or '', self.client['ip'],
               interface_name, interface_type, interface_speed, tx_rate, rx_rate]
        writer.writerow([to_utf8(v) for v in row])
        return out.getvalue()

    def json(self, pretty=False):
        """Return data in JSON format"""

        kwargs = {}
        if pretty:
            kwargs.update({
                'indent': 4,
                'sort_keys': True
            })
        return json.dumps(self.dict(), **kwargs)


class Speedtest(object):
    """Class for performing standard speedtest.net testing operations"""

    def __init__(self, config=None, source_address=None, timeout=10,
                 secure=False, shutdown_event=None, prefer_country=True, http2=False, monitor_tcp=False):
        self.config = {}

        self._source_address = source_address
        self._timeout = timeout
        self._opener = build_opener(source_address, timeout)

        self._secure = secure
        self._prefer_country = prefer_country
        self._http2 = http2
        self._monitor_tcp = monitor_tcp

        if shutdown_event:
            self._shutdown_event = shutdown_event
        else:
            self._shutdown_event = FakeShutdownEvent()

        self.get_config()
        if config is not None:
            self.config.update(config)

        self.servers = {}
        self.closest = []
        self._best = {}

        self.results = SpeedtestResults(
            client=self.config['client'],
            opener=self._opener,
            secure=secure,
        )

    @property
    def best(self):
        if not self._best:
            self.get_best_server()
        return self._best

    def get_config(self):
        """Download the speedtest.net configuration and return only the data
        we are interested in
        """

        headers = {}
        if gzip:
            headers['Accept-Encoding'] = 'gzip'

        # Use the new http_request function with HTTP/2 support if available
        url = 'https://www.speedtest.net/speedtest-config.php' if self._secure else 'http://www.speedtest.net/speedtest-config.php'

        try:
            configxml, status_code, error = http_request(
                url,
                headers=headers,
                timeout=self._timeout,
                http2=self._http2 and HTTPX_AVAILABLE
            )

            if status_code != 200:
                return None

        except Exception as e:
            raise ConfigRetrievalError(e)

        printer('Config XML:\n%s' % configxml, debug=True)

        try:
            try:
                root = ET.fromstring(configxml)
            except ET.ParseError:
                e = get_exception()
                raise SpeedtestConfigError(
                    'Malformed speedtest.net configuration: %s' % e
                )
            server_config = root.find('server-config').attrib
            download = root.find('download').attrib
            upload = root.find('upload').attrib
            # times = root.find('times').attrib
            client = root.find('client').attrib

        except AttributeError:
            try:
                root = DOM.parseString(configxml)
            except ExpatError:
                e = get_exception()
                raise SpeedtestConfigError(
                    'Malformed speedtest.net configuration: %s' % e
                )
            server_config = get_attributes_by_tag_name(root, 'server-config')
            download = get_attributes_by_tag_name(root, 'download')
            upload = get_attributes_by_tag_name(root, 'upload')
            # times = get_attributes_by_tag_name(root, 'times')
            client = get_attributes_by_tag_name(root, 'client')

        ignore_servers = [
            int(i) for i in server_config['ignoreids'].split(',') if i
        ]

        ratio = int(upload['ratio'])
        upload_max = int(upload['maxchunkcount'])
        up_sizes = [32768, 65536, 131072, 262144, 524288, 1048576, 7340032]
        sizes = {
            'upload': up_sizes[ratio - 1:],
            'download': [350, 500, 750, 1000, 1500, 2000, 2500,
                         3000, 3500, 4000]
        }

        size_count = len(sizes['upload'])

        upload_count = int(math.ceil(upload_max / size_count))

        counts = {
            'upload': upload_count,
            'download': int(download['threadsperurl'])
        }

        threads = {
            'upload': int(upload['threads']),
            'download': int(server_config['threadcount']) * 2
        }

        length = {
            'upload': int(upload['testlength']),
            'download': int(download['testlength'])
        }

        self.config.update({
            'client': client,
            'ignore_servers': ignore_servers,
            'sizes': sizes,
            'counts': counts,
            'threads': threads,
            'length': length,
            'upload_max': upload_count * size_count
        })

        try:
            self.lat_lon = (float(client['lat']), float(client['lon']))
        except ValueError:
            raise SpeedtestConfigError(
                'Unknown location: lat=%r lon=%r' %
                (client.get('lat'), client.get('lon'))
            )

        printer('Config:\n%r' % self.config, debug=True)

        return self.config

    def get_servers(self, servers=None, exclude=None):
        """Retrieve a the list of speedtest.net servers, optionally filtered
        to servers matching those specified in the ``servers`` argument
        """
        if servers is None:
            servers = []

        if exclude is None:
            exclude = []

        self.servers.clear()

        for server_list in (servers, exclude):
            for i, s in enumerate(server_list):
                try:
                    server_list[i] = int(s)
                except ValueError:
                    raise InvalidServerIDType(
                        '%s is an invalid server type, must be int' % s
                    )

        urls = [
            'https://www.speedtest.net/speedtest-servers-static.php',
            'https://c.speedtest.net/speedtest-servers-static.php',
            'https://www.speedtest.net/speedtest-servers.php',
            'https://c.speedtest.net/speedtest-servers.php',
            'http://www.speedtest.net/speedtest-servers-static.php',
            'http://c.speedtest.net/speedtest-servers-static.php',
            'http://www.speedtest.net/speedtest-servers.php',
            'http://c.speedtest.net/speedtest-servers.php',
        ]

        headers = {}
        if gzip:
            headers['Accept-Encoding'] = 'gzip'

        errors = []
        for url in urls:
            try:
                # Use the new http_request function with HTTP/2 support if available
                full_url = '%s?threads=%s' % (url, self.config['threads']['download'])

                try:
                    serversxml, status_code, error = http_request(
                        full_url,
                    headers=headers,
                        timeout=self._timeout,
                        http2=self._http2 and HTTPX_AVAILABLE
                )

                    if status_code != 200:
                        raise ServersRetrievalError()

                except Exception as e:
                    errors.append('%s: %s' % (url, e))
                    raise ServersRetrievalError()

                printer('Servers XML:\n%s' % serversxml, debug=True)

                try:
                    try:
                        try:
                            root = ET.fromstring(serversxml)
                        except ET.ParseError:
                            e = get_exception()
                            raise SpeedtestServersError(
                                'Malformed speedtest.net server list: %s' % e
                            )
                        elements = etree_iter(root, 'server')
                    except AttributeError:
                        try:
                            root = DOM.parseString(serversxml)
                        except ExpatError:
                            e = get_exception()
                            raise SpeedtestServersError(
                                'Malformed speedtest.net server list: %s' % e
                            )
                        elements = root.getElementsByTagName('server')
                except (SyntaxError, xml.parsers.expat.ExpatError):
                    raise ServersRetrievalError()

                for server in elements:
                    try:
                        attrib = server.attrib
                    except AttributeError:
                        attrib = dict(list(server.attributes.items()))

                    if servers and int(attrib.get('id')) not in servers:
                        continue

                    if (int(attrib.get('id')) in self.config['ignore_servers']
                            or int(attrib.get('id')) in exclude):
                        continue

                    try:
                        d = distance(self.lat_lon,
                                     (float(attrib.get('lat')),
                                      float(attrib.get('lon'))))
                    except Exception:
                        continue

                    attrib['d'] = d

                    try:
                        self.servers[d].append(attrib)
                    except KeyError:
                        self.servers[d] = [attrib]

                break

            except ServersRetrievalError:
                continue

        if (servers or exclude) and not self.servers:
            raise NoMatchedServers()

        return self.servers

    def set_mini_server(self, server):
        """Instead of querying for a list of servers, set a link to a
        speedtest mini server
        """

        urlparts = urlparse(server)

        name, ext = os.path.splitext(urlparts[2])
        if ext:
            url = os.path.dirname(server)
        else:
            url = server

        request = build_request(url)
        uh, e = catch_request(request, opener=self._opener)
        if e:
            raise SpeedtestMiniConnectFailure('Failed to connect to %s' %
                                              server)
        else:
            text = uh.read()
            uh.close()

        extension = re.findall('upload_?[Ee]xtension: "([^"]+)"',
                               text.decode())
        if not extension:
            for ext in ['php', 'asp', 'aspx', 'jsp']:
                try:
                    f = self._opener.open(
                        '%s/speedtest/upload.%s' % (url, ext)
                    )
                except Exception:
                    pass
                else:
                    data = f.read().strip().decode()
                    if (f.code == 200 and
                            len(data.splitlines()) == 1 and
                            re.match('size=[0-9]', data)):
                        extension = [ext]
                        break
        if not urlparts or not extension:
            raise InvalidSpeedtestMiniServer('Invalid Speedtest Mini Server: '
                                             '%s' % server)

        self.servers = [{
            'sponsor': 'Speedtest Mini',
            'name': urlparts[1],
            'd': 0,
            'url': '%s/speedtest/upload.%s' % (url.rstrip('/'), extension[0]),
            'latency': 0,
            'id': 0
        }]

        return self.servers

    def get_closest_servers(self, limit=5):
        """Limit servers to the closest speedtest.net servers based on
        geographic distance
        """

        if not self.servers:
            self.get_servers()

        for d in sorted(self.servers.keys()):
            for s in self.servers[d]:
                self.closest.append(s)
                if len(self.closest) == limit:
                    break
            else:
                continue
            break

        printer('Closest Servers:\n%r' % self.closest, debug=True)
        return self.closest

    def prefer_same_country_servers(self, servers=None):
        """Filter servers to prefer those in the same country as the client"""
        if servers is None:
            servers = self.closest

        client_country = self.config.get('client', {}).get('country', '')
        if not client_country:
            return servers

        same_country = []
        other_country = []

        for server in servers:
            if server.get('country', '') == client_country:
                same_country.append(server)
            else:
                other_country.append(server)

        # Return same-country servers first, then others
        return same_country + other_country

    def get_best_server(self, servers=None):
        """Perform a speedtest.net "ping" to determine which speedtest.net
        server has the lowest latency
        """

        if not servers:
            if not self.closest:
                servers = self.get_closest_servers()
            servers = self.closest

            # Apply country preference if enabled
            if self._prefer_country:
                servers = self.prefer_same_country_servers(servers)

        if self._source_address:
            source_address_tuple = (self._source_address, 0)
        else:
            source_address_tuple = None

        user_agent = build_user_agent()

        results = {}
        for server in servers:
            cum = []
            url = os.path.dirname(server['url'])
            stamp = int(timeit.time.time() * 1000)
            latency_url = '%s/latency.txt?x=%s' % (url, stamp)
            for i in range(0, 3):
                this_latency_url = '%s.%s' % (latency_url, i)
                printer('%s %s' % ('GET', this_latency_url),
                        debug=True)
                urlparts = urlparse(latency_url)
                try:
                    if urlparts[0] == 'https':
                        h = SpeedtestHTTPSConnection(
                            urlparts[1],
                            source_address=source_address_tuple
                        )
                    else:
                        h = SpeedtestHTTPConnection(
                            urlparts[1],
                            source_address=source_address_tuple
                        )
                    headers = {'User-Agent': user_agent}
                    path = '%s?%s' % (urlparts[2], urlparts[4])
                    start = timeit.default_timer()
                    h.request("GET", path, headers=headers)
                    r = h.getresponse()
                    total = (timeit.default_timer() - start)
                except HTTP_ERRORS:
                    e = get_exception()
                    printer('ERROR: %r' % e, debug=True)
                    cum.append(3600)
                    continue

                text = r.read(9)
                if int(r.status) == 200 and text == 'test=test'.encode():
                    cum.append(total)
                else:
                    cum.append(3600)
                h.close()

            avg = round((sum(cum) / 6) * 1000.0, 3)
            results[avg] = server

        try:
            fastest = sorted(results.keys())[0]
        except IndexError:
            raise SpeedtestBestServerFailure('Unable to connect to servers to '
                                             'test latency.')
        best = results[fastest]
        best['latency'] = fastest

        self.results.ping = fastest
        self.results.server = best

        self._best.update(best)
        printer('Best Server:\n%r' % best, debug=True)
        return best

    def download(self, callback=do_nothing, threads=None):
        """Test download speed against speedtest.net

        A ``threads`` value of ``None`` will fall back to those dictated
        by the speedtest.net configuration
        """

        urls = []
        for size in self.config['sizes']['download']:
            for _ in range(0, self.config['counts']['download']):
                urls.append('%s/random%sx%s.jpg' %
                            (os.path.dirname(self.best['url']), size, size))

        request_count = len(urls)
        requests = []
        for i, url in enumerate(urls):
            requests.append(
                build_request(url, bump=i, secure=self._secure)
            )

        max_threads = threads or self.config['threads']['download']
        in_flight = {'threads': 0}

        def producer(q, requests, request_count):
            for i, request in enumerate(requests):
                if self._http2 and HTTPX_AVAILABLE:
                    # Use HTTP/2 downloader
                    thread = HTTP2Downloader(
                        i,
                        request.get_full_url(),
                        start,
                        self.config['length']['download'],
                        shutdown_event=self._shutdown_event,
                        monitor_tcp=self._monitor_tcp
                    )
                else:
                    # Use HTTP/1.1 downloader
                    thread = HTTPDownloader(
                    i,
                    request,
                    start,
                    self.config['length']['download'],
                    opener=self._opener,
                    shutdown_event=self._shutdown_event
                )
                while in_flight['threads'] >= max_threads:
                    timeit.time.sleep(0.001)
                thread.start()
                q.put(thread, True)
                in_flight['threads'] += 1
                callback(i, request_count, start=True)

        finished = []

        def consumer(q, request_count):
            _is_alive = thread_is_alive
            while len(finished) < request_count:
                thread = q.get(True)
                while _is_alive(thread):
                    thread.join(timeout=0.001)
                in_flight['threads'] -= 1
                finished.append(sum(thread.result))
                callback(thread.i, request_count, end=True)

        q = Queue(max_threads)
        prod_thread = threading.Thread(target=producer,
                                       args=(q, requests, request_count))
        cons_thread = threading.Thread(target=consumer,
                                       args=(q, request_count))
        start = timeit.default_timer()
        prod_thread.start()
        cons_thread.start()
        _is_alive = thread_is_alive
        while _is_alive(prod_thread):
            prod_thread.join(timeout=0.001)
        while _is_alive(cons_thread):
            cons_thread.join(timeout=0.001)

        stop = timeit.default_timer()

        # Ensure minimum test duration of 3 seconds for accurate results on fast networks
        min_duration = 3.0
        if (stop - start) < min_duration:
            printer('Download test completed too quickly, extending to minimum duration...', debug=True)
            # Wait for remaining time to ensure minimum duration
            remaining_time = min_duration - (stop - start)
            time.sleep(remaining_time)
            stop = timeit.default_timer()

        self.results.bytes_received = sum(finished)
        self.results.download = (
            (self.results.bytes_received / (stop - start)) * 8.0
        )

        # Collect TCP monitoring data if enabled
        if self._monitor_tcp:
            self._collect_tcp_monitoring_data(finished, start, stop, 'download')

        if self.results.download > 100000:
            self.config['threads']['upload'] = 8
        return self.results.download

    def upload(self, callback=do_nothing, pre_allocate=True, threads=None):
        """Test upload speed against speedtest.net

        A ``threads`` value of ``None`` will fall back to those dictated
        by the speedtest.net configuration
        """

        sizes = []

        for size in self.config['sizes']['upload']:
            for _ in range(0, self.config['counts']['upload']):
                sizes.append(size)

        # request_count = len(sizes)
        request_count = self.config['upload_max']

        requests = []
        for i, size in enumerate(sizes):
            # We set ``0`` for ``start`` and handle setting the actual
            # ``start`` in ``HTTPUploader`` to get better measurements
            data = HTTPUploaderData(
                size,
                0,
                self.config['length']['upload'],
                shutdown_event=self._shutdown_event
            )
            if pre_allocate:
                data.pre_allocate()

            headers = {'Content-length': size}
            requests.append(
                (
                    build_request(self.best['url'], data, secure=self._secure,
                                  headers=headers),
                    size
                )
            )

        max_threads = threads or self.config['threads']['upload']
        in_flight = {'threads': 0}

        def producer(q, requests, request_count):
            for i, request in enumerate(requests[:request_count]):
                if self._http2 and HTTPX_AVAILABLE:
                    # Use HTTP/2 uploader
                    thread = HTTP2Uploader(
                        i,
                        request[0].get_full_url(),
                        request[0].data,
                        start,
                        request[1],
                        self.config['length']['upload'],
                        shutdown_event=self._shutdown_event,
                        monitor_tcp=self._monitor_tcp
                    )
                else:
                    # Use HTTP/1.1 uploader
                    thread = HTTPUploader(
                    i,
                    request[0],
                    start,
                    request[1],
                    self.config['length']['upload'],
                    opener=self._opener,
                    shutdown_event=self._shutdown_event
                )
                while in_flight['threads'] >= max_threads:
                    timeit.time.sleep(0.001)
                thread.start()
                q.put(thread, True)
                in_flight['threads'] += 1
                callback(i, request_count, start=True)

        finished = []

        def consumer(q, request_count):
            _is_alive = thread_is_alive
            while len(finished) < request_count:
                thread = q.get(True)
                while _is_alive(thread):
                    thread.join(timeout=0.001)
                in_flight['threads'] -= 1
                finished.append(thread.result)
                callback(thread.i, request_count, end=True)

        q = Queue(threads or self.config['threads']['upload'])
        prod_thread = threading.Thread(target=producer,
                                       args=(q, requests, request_count))
        cons_thread = threading.Thread(target=consumer,
                                       args=(q, request_count))
        start = timeit.default_timer()
        prod_thread.start()
        cons_thread.start()
        _is_alive = thread_is_alive
        while _is_alive(prod_thread):
            prod_thread.join(timeout=0.1)
        while _is_alive(cons_thread):
            cons_thread.join(timeout=0.1)

        stop = timeit.default_timer()

        # Ensure minimum test duration of 3 seconds for accurate results on fast networks
        min_duration = 3.0
        if (stop - start) < min_duration:
            printer('Upload test completed too quickly, extending to minimum duration...', debug=True)
            # Wait for remaining time to ensure minimum duration
            remaining_time = min_duration - (stop - start)
            time.sleep(remaining_time)
            stop = timeit.default_timer()

        self.results.bytes_sent = sum(finished)
        self.results.upload = (
            (self.results.bytes_sent / (stop - start)) * 8.0
        )

        # Collect TCP monitoring data if enabled
        if self._monitor_tcp:
            self._collect_tcp_monitoring_data(finished, start, stop, 'upload')

        return self.results.upload

    def _collect_tcp_monitoring_data(self, finished, start, stop, test_type):
        """Collect and analyze TCP monitoring data from completed threads"""
        if not hasattr(self, '_tcp_monitoring_data'):
            self._tcp_monitoring_data = {
                'download': {'threads': [], 'events': [], 'pauses': [], 'fastest_periods': []},
                'upload': {'threads': [], 'events': [], 'pauses': [], 'fastest_periods': []}
            }

        # Store basic timing information
        self._tcp_monitoring_data[test_type]['test_duration'] = stop - start
        self._tcp_monitoring_data[test_type]['total_bytes'] = sum(finished)
        self._tcp_monitoring_data[test_type]['average_rate'] = sum(finished) / (stop - start)

        # Note: Detailed thread-level monitoring data would require access to the actual
        # thread objects, which would need to be passed to this method. For now, we
        # provide basic timing analysis. The HTTP/2 classes are already collecting
        # detailed timing data internally.

    def get_tcp_monitoring_report(self, units=('bit', 1)):
        """Generate a detailed TCP monitoring report"""
        if not hasattr(self, '_tcp_monitoring_data'):
            return "TCP monitoring not enabled"

        report = []
        report.append("=== TCP Performance Analysis ===")

        # Determine unit conversion based on speedtest units
        unit_name, unit_multiplier = units
        if unit_name == 'byte':
            # Display in MB/s (megabytes per second)
            rate_divisor = 1000.0 * 1000.0  # Convert to MB/s
            rate_unit = "MB/s"
        else:
            # Display in Mbit/s (megabits per second) - default
            rate_divisor = (1000.0 * 1000.0) / 8.0  # Convert to Mbit/s
            rate_unit = "Mbit/s"

        for test_type in ['download', 'upload']:
            if test_type not in self._tcp_monitoring_data:
                continue

            data = self._tcp_monitoring_data[test_type]
            report.append(f"\n{test_type.capitalize()} Test:")
            report.append(f"  Duration: {data.get('test_duration', 0):.2f} seconds")

            # Convert bytes/sec to appropriate units for consistency with speedtest output
            bytes_per_sec = data.get('average_rate', 0)
            converted_rate = bytes_per_sec / rate_divisor
            report.append(f"  Average rate: {converted_rate:.2f} {rate_unit}")

            if 'pauses' in data and data['pauses']:
                report.append(f"  Pauses detected: {len(data['pauses'])}")
                for i, pause in enumerate(data['pauses'][:5]):  # Show first 5 pauses
                    report.append(f"    Pause {i+1}: {pause.get('duration', 0):.3f}s at {pause.get('start_time', 0):.2f}s")

            if 'fastest_periods' in data and data['fastest_periods']:
                report.append(f"  Fastest periods: {len(data['fastest_periods'])}")
                for i, period in enumerate(data['fastest_periods'][:3]):  # Show top 3
                    # Convert bytes/sec to appropriate units
                    period_rate_converted = period[2] / rate_divisor
                    report.append(f"    Period {i+1}: {period_rate_converted:.2f} {rate_unit} from {period[0]:.2f}s to {period[1]:.2f}s")

        return "\n".join(report)


def ctrl_c(shutdown_event):
    """Catch Ctrl-C key sequence and set a SHUTDOWN_EVENT for our threaded
    operations
    """
    def inner(signum, frame):
        shutdown_event.set()
        printer('\nCancelling...', error=True)
        sys.exit(0)
    return inner


def version():
    """Print the version"""

    printer('speedtest-cli %s' % __version__)
    printer('Python %s' % sys.version.replace('\n', ''))
    sys.exit(0)


def csv_header(delimiter=','):
    """Print the CSV Headers"""

    printer(SpeedtestResults.csv_header(delimiter=delimiter))
    sys.exit(0)


def parse_args():
    """Function to handle building and parsing of command line arguments"""
    description = (
        'Command line interface for testing internet bandwidth using '
        'speedtest.net.\n'
        '------------------------------------------------------------'
        '--------------\n'
        'https://github.com/sivel/speedtest-cli')

    parser = ArgParser(description=description)
    # Give optparse.OptionParser an `add_argument` method for
    # compatibility with argparse.ArgumentParser
    try:
        parser.add_argument = parser.add_option
    except AttributeError:
        pass
    parser.add_argument('--no-download', dest='download', default=True,
                        action='store_const', const=False,
                        help='Do not perform download test')
    parser.add_argument('--no-upload', dest='upload', default=True,
                        action='store_const', const=False,
                        help='Do not perform upload test')
    parser.add_argument('--single', default=False, action='store_true',
                        help='Only use a single connection instead of '
                             'multiple. This simulates a typical file '
                             'transfer.')
    parser.add_argument('--bytes', dest='units', action='store_const',
                        const=('byte', 8), default=('bit', 1),
                        help='Display values in bytes instead of bits. Does '
                             'not affect the image generated by --share, nor '
                             'output from --json or --csv')
    parser.add_argument('--share', action='store_true',
                        help='Generate and provide a URL to the speedtest.net '
                             'share results image, not displayed with --csv')
    parser.add_argument('--simple', action='store_true', default=False,
                        help='Suppress verbose output, only show basic '
                             'information')
    parser.add_argument('--csv', action='store_true', default=False,
                        help='Suppress verbose output, only show basic '
                             'information in CSV format. Speeds listed in '
                             'bit/s and not affected by --bytes')
    parser.add_argument('--csv-delimiter', default=',', type=PARSER_TYPE_STR,
                        help='Single character delimiter to use in CSV '
                             'output. Default ","')
    parser.add_argument('--csv-header', action='store_true', default=False,
                        help='Print CSV headers')
    parser.add_argument('--json', action='store_true', default=False,
                        help='Suppress verbose output, only show basic '
                             'information in JSON format. Speeds listed in '
                             'bit/s and not affected by --bytes')
    parser.add_argument('--list', action='store_true',
                        help='Display a list of speedtest.net servers '
                             'sorted by distance')
    parser.add_argument('--server', type=PARSER_TYPE_INT, action='append',
                        help='Specify a server ID to test against. Can be '
                             'supplied multiple times')
    parser.add_argument('--exclude', type=PARSER_TYPE_INT, action='append',
                        help='Exclude a server from selection. Can be '
                             'supplied multiple times')
    parser.add_argument('--mini', help='URL of the Speedtest Mini server')
    parser.add_argument('--source', help='Source IP address to bind to')
    parser.add_argument('--timeout', default=10, type=PARSER_TYPE_FLOAT,
                        help='HTTP timeout in seconds. Default 10')
    parser.add_argument('--secure', action='store_true',
                        help='Use HTTPS instead of HTTP when communicating '
                             'with speedtest.net operated servers')
    parser.add_argument('--no-pre-allocate', dest='pre_allocate',
                        action='store_const', default=True, const=False,
                        help='Do not pre allocate upload data. Pre allocation '
                             'is enabled by default to improve upload '
                             'performance. To support systems with '
                             'insufficient memory, use this option to avoid a '
                             'MemoryError')
    parser.add_argument('--prefer-country', action='store_true', default=True,
                        help='Prefer servers in the same country as the client '
                             '(default: enabled)')
    parser.add_argument('--no-prefer-country', dest='prefer_country',
                        action='store_false',
                        help='Do not prefer servers in the same country, '
                             'use only distance-based selection')
    parser.add_argument('--http2', action='store_true',
                        help='Use HTTP/2 for connections (requires httpx)')
    parser.add_argument('--monitor-tcp', action='store_true',
                        help='Enable detailed TCP performance monitoring (requires HTTP/2)')
    parser.add_argument('--version', action='store_true',
                        help='Show the version number and exit')
    parser.add_argument('--debug', action='store_true',
                        help=ARG_SUPPRESS, default=ARG_SUPPRESS)

    options = parser.parse_args()
    if isinstance(options, tuple):
        args = options[0]
    else:
        args = options
    return args


def validate_optional_args(args):
    """Check if an argument was provided that depends on a module that may
    not be part of the Python standard library.

    If such an argument is supplied, and the module does not exist, exit
    with an error stating which module is missing.
    """
    optional_args = {
        'json': ('json/simplejson python module', json),
        'secure': ('SSL support', HTTPSConnection),
        'http2': ('httpx python module for HTTP/2 support', httpx),
        'requests': ('requests python module for HTTP requests', requests),
    }

    for arg, info in optional_args.items():
        if getattr(args, arg, False) and info[1] is None:
            raise SystemExit('%s is not installed. --%s is '
                             'unavailable' % (info[0], arg))


def printer(string, quiet=False, debug=False, error=False, **kwargs):
    """Helper function print a string with various features"""

    if debug and not DEBUG:
        return

    if debug:
        if sys.stdout.isatty():
            out = '\033[1;30mDEBUG: %s\033[0m' % string
        else:
            out = 'DEBUG: %s' % string
    else:
        out = string

    if error:
        kwargs['file'] = sys.stderr

    if not quiet:
        print_(out, **kwargs)


def shell():
    """Run the full speedtest.net test"""

    global DEBUG
    shutdown_event = threading.Event()

    signal.signal(signal.SIGINT, ctrl_c(shutdown_event))

    args = parse_args()

    # Print the version and exit
    if args.version:
        version()

    if not args.download and not args.upload:
        raise SpeedtestCLIError('Cannot supply both --no-download and '
                                '--no-upload')

    if len(args.csv_delimiter) != 1:
        raise SpeedtestCLIError('--csv-delimiter must be a single character')

    if args.csv_header:
        csv_header(args.csv_delimiter)

    validate_optional_args(args)

    debug = getattr(args, 'debug', False)
    if debug == 'SUPPRESSHELP':
        debug = False
    if debug:
        DEBUG = True

    if args.simple or args.csv or args.json:
        quiet = True
    else:
        quiet = False

    if args.csv or args.json:
        machine_format = True
    else:
        machine_format = False

    # Don't set a callback if we are running quietly
    if quiet or debug:
        callback = do_nothing
    else:
        callback = print_dots(shutdown_event)

    printer('Retrieving speedtest.net configuration...', quiet)
    if args.http2 and HTTPX_AVAILABLE:
        printer('Using HTTP/2 for connections', quiet)
    elif args.http2 and not HTTPX_AVAILABLE:
        printer('HTTP/2 requested but httpx not available, falling back to HTTP/1.1', error=True)
    try:
        # Validate that TCP monitoring requires HTTP/2
        monitor_tcp = getattr(args, 'monitor_tcp', False)
        if monitor_tcp and not args.http2:
            printer('TCP monitoring requires HTTP/2. Enabling HTTP/2...', quiet)
            args.http2 = True

        speedtest = Speedtest(
            source_address=args.source,
            timeout=args.timeout,
            secure=args.secure,
            prefer_country=args.prefer_country,
            http2=args.http2,
            monitor_tcp=monitor_tcp
        )
    except (ConfigRetrievalError,) + HTTP_ERRORS:
        printer('Cannot retrieve speedtest configuration', error=True)
        raise SpeedtestCLIError(get_exception())

    if args.list:
        try:
            speedtest.get_servers()
        except (ServersRetrievalError,) + HTTP_ERRORS:
            printer('Cannot retrieve speedtest server list', error=True)
            raise SpeedtestCLIError(get_exception())

        for _, servers in sorted(speedtest.servers.items()):
            for server in servers:
                line = ('%(id)5s) %(sponsor)s (%(name)s, %(country)s) '
                        '[%(d)0.2f km]' % server)
                try:
                    printer(line)
                except IOError:
                    e = get_exception()
                    if e.errno != errno.EPIPE:
                        raise
        sys.exit(0)

    printer('Testing from %(isp)s (%(ip)s)...' % speedtest.config['client'],
            quiet)

    # Display interface information if available
    if speedtest.results.interface_info:
        interface = speedtest.results.interface_info
        if interface['type'] == 'Wi-Fi' and interface.get('tx_rate') is not None and interface.get('rx_rate') is not None:
            printer('Network interface: %s (%s) - TX: %s Mbps, RX: %s Mbps' %
                    (interface['interface'], interface['type'],
                     interface['tx_rate'], interface['rx_rate']),
                    quiet)
        else:
            printer('Network interface: %s (%s) - %s Mbps' %
                    (interface['interface'], interface['type'], interface['speed']),
            quiet)

    if not args.mini:
        printer('Retrieving speedtest.net server list...', quiet)
        try:
            speedtest.get_servers(servers=args.server, exclude=args.exclude)
        except NoMatchedServers:
            raise SpeedtestCLIError(
                'No matched servers: %s' %
                ', '.join('%s' % s for s in args.server)
            )
        except (ServersRetrievalError,) + HTTP_ERRORS:
            printer('Cannot retrieve speedtest server list', error=True)
            raise SpeedtestCLIError(get_exception())
        except InvalidServerIDType:
            raise SpeedtestCLIError(
                '%s is an invalid server type, must '
                'be an int' % ', '.join('%s' % s for s in args.server)
            )

        if args.server and len(args.server) == 1:
            printer('Retrieving information for the selected server...', quiet)
        else:
            printer('Selecting best server based on ping...', quiet)
        speedtest.get_best_server()
    elif args.mini:
        speedtest.get_best_server(speedtest.set_mini_server(args.mini))

    results = speedtest.results

    printer('Hosted by %(sponsor)s (%(name)s) [%(d)0.2f km]: '
            '%(latency)s ms' % results.server, quiet)

    if args.download:
        printer('Testing download speed', quiet,
                end=('', '\n')[bool(debug)])
        speedtest.download(
            callback=callback,
            threads=(None, 1)[args.single]
        )
        printer('Download: %0.2f M%s/s' %
                ((results.download / 1000.0 / 1000.0) / args.units[1],
                 args.units[0]),
                quiet)
    else:
        printer('Skipping download test', quiet)

    if args.upload:
        printer('Testing upload speed', quiet,
                end=('', '\n')[bool(debug)])
        speedtest.upload(
            callback=callback,
            pre_allocate=args.pre_allocate,
            threads=(None, 1)[args.single]
        )
        printer('Upload: %0.2f M%s/s' %
                ((results.upload / 1000.0 / 1000.0) / args.units[1],
                 args.units[0]),
                quiet)
    else:
        printer('Skipping upload test', quiet)

    # Display TCP monitoring report if enabled
    if getattr(args, 'monitor_tcp', False):
        tcp_report = speedtest.get_tcp_monitoring_report(args.units)
        printer(tcp_report, quiet)

    printer('Results:\n%r' % results.dict(), debug=True)

    if not args.simple and args.share:
        results.share()

    if args.simple:
        output = 'Ping: %s ms\nDownload: %0.2f M%s/s\nUpload: %0.2f M%s/s' % (
            results.ping,
                 (results.download / 1000.0 / 1000.0) / args.units[1],
                 args.units[0],
                 (results.upload / 1000.0 / 1000.0) / args.units[1],
            args.units[0]
        )

        # Add interface information if available
        if results.interface_info:
            interface = results.interface_info
            if interface['type'] == 'Wi-Fi' and interface.get('tx_rate') is not None and interface.get('rx_rate') is not None:
                output += '\nInterface: %s (%s) - TX: %s Mbps, RX: %s Mbps' % (
                    interface['interface'], interface['type'],
                    interface['tx_rate'], interface['rx_rate']
                )
            else:
                output += '\nInterface: %s (%s) - %s Mbps' % (
                    interface['interface'], interface['type'], interface['speed']
                )

        printer(output)
    elif args.csv:
        printer(results.csv(delimiter=args.csv_delimiter))
    elif args.json:
        printer(results.json())

    if args.share and not machine_format:
        printer('Share results: %s' % results.share())


def main():
    try:
        shell()
    except KeyboardInterrupt:
        printer('\nCancelling...', error=True)
    except (SpeedtestException, SystemExit):
        e = get_exception()
        # Ignore a successful exit, or argparse exit
        if getattr(e, 'code', 1) not in (0, 2):
            msg = '%s' % e
            if not msg:
                msg = '%r' % e
            raise SystemExit('ERROR: %s' % msg)


if __name__ == '__main__':
    main()