#!/usr/bin/env python
#
# SPDX-FileCopyrightText: 2018-2022 Espressif Systems (Shanghai) CO LTD
# SPDX-License-Identifier: Apache-2.0

# Utility for testing the web server. Test cases:
# Assume the device supports 'n' simultaneous open sockets
#
# HTTP Server Tests
#
# 0. Firmware Settings:
# - Create a dormant thread whose sole job is to call httpd_stop() when instructed
# - Measure the following before httpd_start() is called:
#     - current free memory
#     - current free sockets
# - Measure the same whenever httpd_stop is called
# - Register maximum possible URI handlers: should be successful
# - Register one more URI handler: should fail
# - Deregister on URI handler: should be successful
# - Register on more URI handler: should succeed
# - Register separate handlers for /hello, /hello/type_html. Also
#   ensure that /hello/type_html is registered BEFORE  /hello. (tests
#   that largest matching URI is picked properly)
# - Create URI handler /adder. Make sure it uses a custom free_ctx
#   structure to free it up

# 1. Using Standard Python HTTP Client
# - simple GET on /hello (returns Hello World. Ensures that basic
#   firmware tests are complete, or returns error)
# - POST on /hello (should fail)
# - PUT on /hello (should fail)
# - simple POST on /echo (returns whatever the POST data)
# - simple PUT on /echo (returns whatever the PUT data)
# - GET on /echo (should fail)
# - simple GET on /hello/type_html (returns Content type as text/html)
# - simple GET on /hello/status_500 (returns HTTP status 500)
# - simple GET on /false_uri (returns HTTP status 404)
# - largest matching URI handler is picked is already verified because
#   of /hello and /hello/type_html tests
#
#
# 2. Session Tests
# - Sessions + Pipelining basics:
#    - Create max supported sessions
#    - On session i,
#          - send 3 back-to-back POST requests with data i on /adder
#          - read back 3 responses. They should be i, 2i and 3i
#    - Tests that
#          - pipelining works
#          - per-session context is maintained for all supported
#            sessions
#    - Close all sessions
#
# - Cleanup leftover data: Tests that the web server properly cleans
#   up leftover data
#    - Create a session
#    - POST on /leftover_data with 52 bytes of data (data includes
#      \r\n)(the handler only
#      reads first 10 bytes and returns them, leaving the rest of the
#      bytes unread)
#    - GET on /hello (should return 'Hello World')
#    - POST on /false_uri with 52  bytes of data (data includes \r\n)
#      (should return HTTP 404)
#    - GET on /hello (should return 'Hello World')
#
# - Test HTTPd Asynchronous response
#   - Create a session
#   - GET on /async_data
#   - returns 'Hello World!' as a response
#   - the handler schedules an async response, which generates a second
#     response 'Hello Double World!'
#
# - Spillover test
#    - Create max supported sessions with the web server
#    - GET /hello on all the sessions (should return Hello World)
#    - Create one more session, this should fail
#    - GET /hello on all the sessions (should return Hello World)
#
# - Timeout test
#    - Create a session and only Send 'GE' on the same (simulates a
#      client that left the network halfway through a request)
#    - Wait for recv-wait-timeout
#    - Server should automatically close the socket


# ############ TODO TESTS #############

# 3. Stress Tests
#
# - httperf
#     - Run the following httperf command:
#     httperf --server=10.31.130.126 --wsess=8,50,0.5 --rate 8 --burst-length 2
#
#     - The above implies that the test suite will open
#           - 8 simultaneous connections with the server
#           - the rate of opening the sessions will be 8 per sec. So in our
#             case, a new connection will be opened every 0.2 seconds for 1 second
#           - The burst length 2 indicates that 2 requests will be sent
#             simultaneously on the same connection in a single go
#           - 0.5 seconds is the time between sending out 2 bursts
#           - 50 is the total number of requests that will be sent out
#
#     - So in the above example, the test suite will open 8
#       connections, each separated by 0.2 seconds. On each connection
#       it will send 2 requests in a single burst. The bursts on a
#       single connection will be separated by 0.5 seconds. A total of
#       25 bursts (25 x 2 = 50) will be sent out

# 4. Leak Tests
# - Simple Leak test
#    - Simple GET on /hello/restart (returns success, stop web server, measures leaks, restarts webserver)
#    - Simple GET on /hello/restart_results (returns the leak results)
# - Leak test with open sockets
#    - Open 8 sessions
#    - Simple GET on /hello/restart (returns success, stop web server,
#      measures leaks, restarts webserver)
#    - All sockets should get closed
#    - Simple GET on /hello/restart_results (returns the leak results)


from __future__ import division, print_function

import argparse
import http.client
import logging
import random
import socket
import string
import sys
import threading
import time
from builtins import object, range, str

_verbose_ = False


class Session(object):
    def __init__(self, addr, port, timeout=15):
        self.client = socket.create_connection((addr, int(port)), timeout=timeout)
        self.target = addr
        self.status = 0
        self.encoding = ''
        self.content_type = ''
        self.content_len = 0

    def send_err_check(self, request, data=None):
        rval = True
        try:
            self.client.sendall(request.encode())
            if data:
                self.client.sendall(data.encode())
        except socket.error as err:
            self.client.close()
            logging.info('Socket Error in send :{}'.format(err))
            rval = False
        return rval

    def send_get(self, path, headers=None):
        request = 'GET ' + path + ' HTTP/1.1\r\nHost: ' + self.target
        if headers:
            for field, value in headers.items():
                request += '\r\n' + field + ': ' + value
        request += '\r\n\r\n'
        return self.send_err_check(request)

    def send_put(self, path, data, headers=None):
        request = 'PUT ' + path + ' HTTP/1.1\r\nHost: ' + self.target
        if headers:
            for field, value in headers.items():
                request += '\r\n' + field + ': ' + value
        request += '\r\nContent-Length: ' + str(len(data)) + '\r\n\r\n'
        return self.send_err_check(request, data)

    def send_post(self, path, data, headers=None):
        request = 'POST ' + path + ' HTTP/1.1\r\nHost: ' + self.target
        if headers:
            for field, value in headers.items():
                request += '\r\n' + field + ': ' + value
        request += '\r\nContent-Length: ' + str(len(data)) + '\r\n\r\n'
        return self.send_err_check(request, data)

    def read_resp_hdrs(self):
        try:
            state = 'nothing'
            resp_read = ''
            while True:
                char = self.client.recv(1).decode()
                if char == '\r' and state == 'nothing':
                    state = 'first_cr'
                elif char == '\n' and state == 'first_cr':
                    state = 'first_lf'
                elif char == '\r' and state == 'first_lf':
                    state = 'second_cr'
                elif char == '\n' and state == 'second_cr':
                    state = 'second_lf'
                else:
                    state = 'nothing'
                resp_read += char
                if state == 'second_lf':
                    break
            # Handle first line
            line_hdrs = resp_read.splitlines()
            line_comp = line_hdrs[0].split()
            self.status = line_comp[1]
            del line_hdrs[0]
            self.encoding = ''
            self.content_type = ''
            headers = dict()
            # Process other headers
            for h in range(len(line_hdrs)):
                line_comp = line_hdrs[h].split(':')
                if line_comp[0] == 'Content-Length':
                    self.content_len = int(line_comp[1])
                if line_comp[0] == 'Content-Type':
                    self.content_type = line_comp[1].lstrip()
                if line_comp[0] == 'Transfer-Encoding':
                    self.encoding = line_comp[1].lstrip()
                if len(line_comp) == 2:
                    headers[line_comp[0]] = line_comp[1].lstrip()
            return headers
        except socket.error as err:
            self.client.close()
            logging.info('Socket Error in recv :{}'.format(err))
            return None

    def read_resp_data(self):
        try:
            read_data = ''
            if self.encoding != 'chunked':
                while len(read_data) != self.content_len:
                    read_data += self.client.recv(self.content_len).decode()
            else:
                chunk_data_buf = ''
                while (True):
                    # Read one character into temp  buffer
                    read_ch = self.client.recv(1)
                    # Check CRLF
                    if (read_ch == '\r'):
                        read_ch = self.client.recv(1).decode()
                        if (read_ch == '\n'):
                            # If CRLF decode length of chunk
                            chunk_len = int(chunk_data_buf, 16)
                            # Keep adding to contents
                            self.content_len += chunk_len
                            rem_len = chunk_len
                            while (rem_len):
                                new_data = self.client.recv(rem_len)
                                read_data += new_data
                                rem_len -= len(new_data)
                            chunk_data_buf = ''
                            # Fetch remaining CRLF
                            if self.client.recv(2) != '\r\n':
                                # Error in packet
                                logging.info('Error in chunked data')
                                return None
                            if not chunk_len:
                                # If last chunk
                                break
                            continue
                        chunk_data_buf += '\r'
                    # If not CRLF continue appending
                    # character to chunked data buffer
                    chunk_data_buf += read_ch
            return read_data
        except socket.error as err:
            self.client.close()
            logging.info('Socket Error in recv :{}'.format(err))
            return None

    def close(self):
        self.client.close()


def test_val(text, expected, received):
    if expected != received:
        logging.info(' Fail!')
        logging.info('  [reason] {}:'.format(text))
        logging.info('        expected: {}'.format(expected))
        logging.info('        received: {}'.format(received))
        return False
    return True


class adder_thread (threading.Thread):
    def __init__(self, id, dut, port):
        threading.Thread.__init__(self)
        self.id = id
        self.dut = dut
        self.depth = 3
        self.session = Session(dut, port)

    def run(self):
        self.response = []

        # Pipeline 3 requests
        if (_verbose_):
            logging.info('   Thread: Using adder start {}'.format(self.id))

        for _ in range(self.depth):
            self.session.send_post('/adder', str(self.id))
            time.sleep(2)

        for _ in range(self.depth):
            self.session.read_resp_hdrs()
            self.response.append(self.session.read_resp_data())

    def adder_result(self):
        if len(self.response) != self.depth:
            logging.info('Error : missing response packets')
            return False
        for i in range(len(self.response)):
            if not test_val('Thread' + str(self.id) + ' response[' + str(i) + ']',
                            str(self.id * (i + 1)), str(self.response[i])):
                return False
        return True

    def close(self):
        self.session.close()


def get_hello(dut, port):
    # GET /hello should return 'Hello World!'
    logging.info("[test] GET /hello returns 'Hello World!' =>")
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('GET', '/hello')
    resp = conn.getresponse()
    if not test_val('status_code', 200, resp.status):
        conn.close()
        return False
    if not test_val('data', 'Hello World!', resp.read().decode()):
        conn.close()
        return False
    if not test_val('data', 'text/html', resp.getheader('Content-Type')):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def put_hello(dut, port):
    # PUT /hello returns 405'
    logging.info('[test] PUT /hello returns 405 =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('PUT', '/hello', 'Hello')
    resp = conn.getresponse()
    if not test_val('status_code', 405, resp.status):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def post_hello(dut, port):
    # POST /hello returns 405'
    logging.info('[test] POST /hello returns 405 =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('POST', '/hello', 'Hello')
    resp = conn.getresponse()
    if not test_val('status_code', 405, resp.status):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def post_echo(dut, port):
    # POST /echo echoes data'
    logging.info('[test] POST /echo echoes data =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('POST', '/echo', 'Hello')
    resp = conn.getresponse()
    if not test_val('status_code', 200, resp.status):
        conn.close()
        return False
    if not test_val('data', 'Hello', resp.read().decode()):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def put_echo(dut, port):
    # PUT /echo echoes data'
    logging.info('[test] PUT /echo echoes data =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('PUT', '/echo', 'Hello')
    resp = conn.getresponse()
    if not test_val('status_code', 200, resp.status):
        conn.close()
        return False
    if not test_val('data', 'Hello', resp.read().decode()):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def get_echo(dut, port):
    # GET /echo returns 404'
    logging.info('[test] GET /echo returns 405 =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('GET', '/echo')
    resp = conn.getresponse()
    if not test_val('status_code', 405, resp.status):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def get_test_headers(dut, port):
    # GET /test_header returns data of Header2'
    logging.info('[test] GET /test_header =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    custom_header = {'Header1': 'Value1', 'Header3': 'Value3'}
    header2_values = ['', '  ', 'Value2', '   Value2', 'Value2  ', '  Value2  ']
    for val in header2_values:
        custom_header['Header2'] = val
        conn.request('GET', '/test_header', headers=custom_header)
        resp = conn.getresponse()
        if not test_val('status_code', 200, resp.status):
            conn.close()
            return False
        hdr_val_start_idx = val.find('Value2')
        if hdr_val_start_idx == -1:
            if not test_val('header: Header2', '', resp.read().decode()):
                conn.close()
                return False
        else:
            if not test_val('header: Header2', val[hdr_val_start_idx:], resp.read().decode()):
                conn.close()
                return False
        resp.read()
    logging.info('Success')
    conn.close()
    return True


def get_hello_type(dut, port):
    # GET /hello/type_html returns text/html as Content-Type'
    logging.info('[test] GET /hello/type_html has Content-Type of text/html =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('GET', '/hello/type_html')
    resp = conn.getresponse()
    if not test_val('status_code', 200, resp.status):
        conn.close()
        return False
    if not test_val('data', 'Hello World!', resp.read().decode()):
        conn.close()
        return False
    if not test_val('data', 'text/html', resp.getheader('Content-Type')):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def get_hello_status(dut, port):
    # GET /hello/status_500 returns status 500'
    logging.info('[test] GET /hello/status_500 returns status 500 =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('GET', '/hello/status_500')
    resp = conn.getresponse()
    if not test_val('status_code', 500, resp.status):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def get_false_uri(dut, port):
    # GET /false_uri returns status 404'
    logging.info('[test] GET /false_uri returns status 404 =>')
    conn = http.client.HTTPConnection(dut, int(port), timeout=15)
    conn.request('GET', '/false_uri')
    resp = conn.getresponse()
    if not test_val('status_code', 404, resp.status):
        conn.close()
        return False
    logging.info('Success')
    conn.close()
    return True


def parallel_sessions_adder(dut, port, max_sessions):
    # POSTs on /adder in parallel sessions
    logging.info('[test] POST {pipelined} on /adder in ' + str(max_sessions) + ' sessions =>')
    t = []
    # Create all sessions
    for i in range(max_sessions):
        t.append(adder_thread(i, dut, port))

    for i in range(len(t)):
        t[i].start()

    for i in range(len(t)):
        t[i].join()

    res = True
    for i in range(len(t)):
        if not test_val('Thread' + str(i) + ' Failed', t[i].adder_result(), True):
            res = False
        t[i].close()
    if (res):
        logging.info('Success')
    return res


def async_response_test(dut, port):
    # Test that an asynchronous work is executed in the HTTPD's context
    # This is tested by reading two responses over the same session
    logging.info('[test] Test HTTPD Work Queue (Async response) =>')
    s = Session(dut, port)

    s.send_get('/async_data')
    s.read_resp_hdrs()
    if not test_val('First Response', 'Hello World!', s.read_resp_data()):
        s.close()
        return False
    s.read_resp_hdrs()
    if not test_val('Second Response', 'Hello Double World!', s.read_resp_data()):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def leftover_data_test(dut, port):
    # Leftover data in POST is purged (valid and invalid URIs)
    logging.info('[test] Leftover data in POST is purged (valid and invalid URIs) =>')
    s = http.client.HTTPConnection(dut + ':' + port, timeout=15)

    s.request('POST', url='/leftover_data', body='abcdefghijklmnopqrstuvwxyz\r\nabcdefghijklmnopqrstuvwxyz')
    resp = s.getresponse()
    if not test_val('Partial data', 'abcdefghij', resp.read().decode()):
        s.close()
        return False

    s.request('GET', url='/hello')
    resp = s.getresponse()
    if not test_val('Hello World Data', 'Hello World!', resp.read().decode()):
        s.close()
        return False

    s.request('POST', url='/false_uri', body='abcdefghijklmnopqrstuvwxyz\r\nabcdefghijklmnopqrstuvwxyz')
    resp = s.getresponse()
    if not test_val('False URI Status', str(404), str(resp.status)):
        s.close()
        return False
    # socket would have been closed by server due to error
    s.close()

    s = http.client.HTTPConnection(dut + ':' + port, timeout=15)
    s.request('GET', url='/hello')
    resp = s.getresponse()
    if not test_val('Hello World Data', 'Hello World!', resp.read().decode()):
        s.close()
        return False

    s.close()
    logging.info('Success')
    return True


def spillover_session(dut, port, max_sess):
    # Session max_sess_sessions + 1 is rejected
    logging.info('[test] Session max_sess_sessions ({}) + 1 is rejected =>'.format(max_sess))
    s = []
    _verbose_ = True
    for i in range(max_sess + 1):
        if (_verbose_):
            logging.info('Executing {}'.format(i))
        try:
            a = http.client.HTTPConnection(dut + ':' + port, timeout=15)
            a.request('GET', url='/hello')
            resp = a.getresponse()
            if not test_val('Connection ' + str(i), 'Hello World!', resp.read().decode()):
                a.close()
                break
            s.append(a)
        except Exception:
            if (_verbose_):
                logging.info('Connection {} rejected'.format(i))
            a.close()
            break

    # Close open connections
    for a in s:
        a.close()

    # Check if number of connections is equal to max_sess
    logging.info(['Fail','Success'][len(s) == max_sess])
    return (len(s) == max_sess)


def recv_timeout_test(dut, port):
    logging.info('[test] Timeout occurs if partial packet sent =>')
    s = Session(dut, port)
    s.client.sendall(b'GE')
    s.read_resp_hdrs()
    resp = s.read_resp_data()
    if not test_val('Request Timeout', 'Server closed this connection', resp):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def packet_size_limit_test(dut, port, test_size):
    logging.info('[test] send size limit test =>')
    retry = 5
    while (retry):
        retry -= 1
        logging.info('data size = {}'.format(test_size))
        s = http.client.HTTPConnection(dut + ':' + port, timeout=15)
        random_data = ''.join(string.printable[random.randint(0,len(string.printable)) - 1] for _ in list(range(test_size)))
        path = '/echo'
        s.request('POST', url=path, body=random_data)
        resp = s.getresponse()
        if not test_val('Error', '200', str(resp.status)):
            if test_val('Error', '500', str(resp.status)):
                logging.info('Data too large to be allocated')
                test_size = test_size // 10
            else:
                logging.info('Unexpected error')
            s.close()
            logging.info('Retry...')
            continue
        resp = resp.read().decode()
        result = (resp == random_data)
        if not result:
            test_val('Data size', str(len(random_data)), str(len(resp)))
            s.close()
            logging.info('Retry...')
            continue
        s.close()
        logging.info('Success')
        return True
    logging.info('Failed')
    return False


def arbitrary_termination_test(dut, port):
    logging.info('[test] Arbitrary termination test =>')
    cases = [
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nCustom: SomeValue\r\n\r\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\nHost: ' + dut + '\r\nCustom: SomeValue\r\n\r\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\nCustom: SomeValue\r\n\r\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nCustom: SomeValue\n\r\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nCustom: SomeValue\r\n\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\nHost: ' + dut + '\nCustom: SomeValue\n\n',
            'code': '200',
            'header': 'SomeValue'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: 5\n\r\nABCDE',
            'code': '200',
            'body': 'ABCDE'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: 5\r\n\nABCDE',
            'code': '200',
            'body': 'ABCDE'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: 5\n\nABCDE',
            'code': '200',
            'body': 'ABCDE'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: 5\n\n\rABCD',
            'code': '200',
            'body': '\rABCD'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\r\nCustom: SomeValue\r\r\n\r\r\n',
            'code': '400'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\r\nHost: ' + dut + '\r\n\r\n',
            'code': '400'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\n\rHost: ' + dut + '\r\n\r\n',
            'code': '400'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\rCustom: SomeValue\r\n',
            'code': '400'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nCustom: Some\rValue\r\n',
            'code': '400'
        },
        {
            'request': 'POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nCustom- SomeValue\r\n\r\n',
            'code': '400'
        }
    ]
    for case in cases:
        s = Session(dut, port)
        s.client.sendall((case['request']).encode())
        resp_hdrs = s.read_resp_hdrs()
        resp_body = s.read_resp_data()
        s.close()
        if not test_val('Response Code', case['code'], s.status):
            return False
        if 'header' in case.keys():
            resp_hdr_val = None
            if 'Custom' in resp_hdrs.keys():
                resp_hdr_val = resp_hdrs['Custom']
            if not test_val('Response Header', case['header'], resp_hdr_val):
                return False
        if 'body' in case.keys():
            if not test_val('Response Body', case['body'], resp_body):
                return False
    logging.info('Success')
    return True


def code_500_server_error_test(dut, port):
    logging.info('[test] 500 Server Error test =>')
    s = Session(dut, port)
    # Sending a very large content length will cause malloc to fail
    content_len = 2**30
    s.client.sendall(('POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: ' + str(content_len) + '\r\n\r\nABCD').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Server Error', '500', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_501_method_not_impl(dut, port):
    logging.info('[test] 501 Method Not Implemented =>')
    s = Session(dut, port)
    path = '/hello'
    s.client.sendall(('ABC ' + path + ' HTTP/1.1\r\nHost: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    # Presently server sends back 400 Bad Request
    # if not test_val("Server Error", "501", s.status):
    #     s.close()
    #     return False
    if not test_val('Server Error', '400', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_505_version_not_supported(dut, port):
    logging.info('[test] 505 Version Not Supported =>')
    s = Session(dut, port)
    path = '/hello'
    s.client.sendall(('GET ' + path + ' HTTP/2.0\r\nHost: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Server Error', '505', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_400_bad_request(dut, port):
    logging.info('[test] 400 Bad Request =>')
    s = Session(dut, port)
    path = '/hello'
    s.client.sendall(('XYZ ' + path + ' HTTP/1.1\r\nHost: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Client Error', '400', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_404_not_found(dut, port):
    logging.info('[test] 404 Not Found =>')
    s = Session(dut, port)
    path = '/dummy'
    s.client.sendall(('GET ' + path + ' HTTP/1.1\r\nHost: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Client Error', '404', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_405_method_not_allowed(dut, port):
    logging.info('[test] 405 Method Not Allowed =>')
    s = Session(dut, port)
    path = '/hello'
    s.client.sendall(('POST ' + path + ' HTTP/1.1\r\nHost: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Client Error', '405', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_408_req_timeout(dut, port):
    logging.info('[test] 408 Request Timeout =>')
    s = Session(dut, port)
    s.client.sendall(('POST /echo HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Length: 10\r\n\r\nABCD').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Client Error', '408', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def code_411_length_required(dut, port):
    logging.info('[test] 411 Length Required =>')
    s = Session(dut, port)
    path = '/echo'
    s.client.sendall(('POST ' + path + ' HTTP/1.1\r\nHost: ' + dut + '\r\nContent-Type: text/plain\r\nTransfer-Encoding: chunked\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    # Presently server sends back 400 Bad Request
    # if not test_val("Client Error", "411", s.status):
    #    s.close()
    #    return False
    if not test_val('Client Error', '400', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


def send_getx_uri_len(dut, port, length):
    s = Session(dut, port)
    method = 'GET '
    version = ' HTTP/1.1\r\n'
    path = '/' + 'x' * (length - len(method) - len(version) - len('/'))
    s.client.sendall(method.encode())
    time.sleep(1)
    s.client.sendall(path.encode())
    time.sleep(1)
    s.client.sendall((version + 'Host: ' + dut + '\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    s.close()
    return s.status


def code_414_uri_too_long(dut, port, max_uri_len):
    logging.info('[test] 414 URI Too Long =>')
    status = send_getx_uri_len(dut, port, max_uri_len)
    if not test_val('Client Error', '404', status):
        return False
    status = send_getx_uri_len(dut, port, max_uri_len + 1)
    if not test_val('Client Error', '414', status):
        return False
    logging.info('Success')
    return True


def send_postx_hdr_len(dut, port, length):
    s = Session(dut, port)
    path = '/echo'
    host = 'Host: ' + dut
    custom_hdr_field = '\r\nCustom: '
    custom_hdr_val = 'x' * (length - len(host) - len(custom_hdr_field) - len('\r\n\r\n') + len('0'))
    request = ('POST ' + path + ' HTTP/1.1\r\n' + host + custom_hdr_field + custom_hdr_val + '\r\n\r\n').encode()
    s.client.sendall(request[:length // 2])
    time.sleep(1)
    s.client.sendall(request[length // 2:])
    hdr = s.read_resp_hdrs()
    resp = s.read_resp_data()
    s.close()
    if hdr and ('Custom' in hdr):
        return (hdr['Custom'] == custom_hdr_val), resp
    return False, s.status


def code_431_hdr_too_long(dut, port, max_hdr_len):
    logging.info('[test] 431 Header Too Long =>')
    res, status = send_postx_hdr_len(dut, port, max_hdr_len)
    if not res:
        return False
    res, status = send_postx_hdr_len(dut, port, max_hdr_len + 1)
    if not test_val('Client Error', '431', status):
        return False
    logging.info('Success')
    return True


def test_upgrade_not_supported(dut, port):
    logging.info('[test] Upgrade Not Supported =>')
    s = Session(dut, port)
    # path = "/hello"
    s.client.sendall(('OPTIONS * HTTP/1.1\r\nHost:' + dut + '\r\nUpgrade: TLS/1.0\r\nConnection: Upgrade\r\n\r\n').encode())
    s.read_resp_hdrs()
    s.read_resp_data()
    if not test_val('Client Error', '400', s.status):
        s.close()
        return False
    s.close()
    logging.info('Success')
    return True


if __name__ == '__main__':
    # Execution begins here...
    # Configuration
    # Max number of threads/sessions
    max_sessions = 7
    max_uri_len = 512
    max_hdr_len = 512

    parser = argparse.ArgumentParser(description='Run HTTPD Test')
    parser.add_argument('-4','--ipv4', help='IPv4 address')
    parser.add_argument('-6','--ipv6', help='IPv6 address')
    parser.add_argument('-p','--port', help='Port')
    args = vars(parser.parse_args())

    dut4 = args['ipv4']
    dut6 = args['ipv6']
    port = args['port']
    dut  = dut4

    _verbose_ = True

    logging.info('### Basic HTTP Client Tests')
    get_hello(dut, port)
    post_hello(dut, port)
    put_hello(dut, port)
    post_echo(dut, port)
    get_echo(dut, port)
    put_echo(dut, port)
    get_hello_type(dut, port)
    get_hello_status(dut, port)
    get_false_uri(dut, port)
    get_test_headers(dut, port)

    logging.info('### Error code tests')
    code_500_server_error_test(dut, port)
    code_501_method_not_impl(dut, port)
    code_505_version_not_supported(dut, port)
    code_400_bad_request(dut, port)
    code_404_not_found(dut, port)
    code_405_method_not_allowed(dut, port)
    code_408_req_timeout(dut, port)
    code_414_uri_too_long(dut, port, max_uri_len)
    code_431_hdr_too_long(dut, port, max_hdr_len)
    test_upgrade_not_supported(dut, port)

    # Not supported yet (Error on chunked request)
    # code_411_length_required(dut, port)

    logging.info('### Sessions and Context Tests')
    parallel_sessions_adder(dut, port, max_sessions)
    leftover_data_test(dut, port)
    async_response_test(dut, port)
    spillover_session(dut, port, max_sessions)
    recv_timeout_test(dut, port)
    packet_size_limit_test(dut, port, 50 * 1024)
    arbitrary_termination_test(dut, port)
    get_hello(dut, port)

    sys.exit()