// SPDX-License-Identifier: Apache-2.0
// Copyright 2015-2021 Espressif Systems (Shanghai) PTE LTD
//
// 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.
//

/** Includes **/

#include "stats.h"
#include "esp_hosted_config.h"
#if TEST_RAW_TP
#include "os_wrapper.h"
#include "transport_drv.h"
#endif
#include "esp_log.h"

// use mempool and zero copy for Tx
#include "mempool.h"

#if ESP_PKT_STATS
struct pkt_stats_t pkt_stats;
void *pkt_stats_thread = NULL;
extern volatile uint8_t wifi_tx_throttling;
#endif

#if ESP_PKT_STATS || TEST_RAW_TP
DEFINE_LOG_TAG(stats);
#endif

/** Constants/Macros **/
#define RAW_TP_TX_TASK_STACK_SIZE        2048

/** Exported variables **/

/** Function declaration **/

/** Exported Functions **/

#if TEST_RAW_TP
static int test_raw_tp = 0;
static uint8_t log_raw_tp_stats_timer_running = 0;
static uint32_t raw_tp_timer_count = 0;
void *hosted_timer_handler = NULL;
static void * raw_tp_tx_task_id = 0;
static uint64_t test_raw_tx_len = 0;
static uint64_t test_raw_rx_len = 0;

static struct mempool * buf_mp_g = NULL;

void stats_mempool_free(void* ptr)
{
	mempool_free(buf_mp_g, ptr);
}

void test_raw_tp_cleanup(void)
{
	int ret = 0;

	if (log_raw_tp_stats_timer_running) {
		ret = g_h.funcs->_h_timer_stop(hosted_timer_handler);
		if (!ret) {
			log_raw_tp_stats_timer_running = 0;
		}
		raw_tp_timer_count = 0;
	}

	if (raw_tp_tx_task_id) {
		ret = g_h.funcs->_h_thread_cancel(raw_tp_tx_task_id);
		raw_tp_tx_task_id = 0;
	}
}

void raw_tp_timer_func(void * arg)
{
#if USE_FLOATING_POINT
	double actual_bandwidth_tx = 0;
	double actual_bandwidth_rx = 0;
#else
	uint64_t actual_bandwidth_tx = 0;
	uint64_t actual_bandwidth_rx = 0;
#endif
	int32_t div = 1024;


	actual_bandwidth_tx = (test_raw_tx_len*8)/TEST_RAW_TP__TIMEOUT;
	actual_bandwidth_rx = (test_raw_rx_len*8)/TEST_RAW_TP__TIMEOUT;
#if USE_FLOATING_POINT
	ESP_LOGI(TAG, "%lu-%lu sec Tx:%.2f Rx:%.2f kbps\n\r", raw_tp_timer_count, raw_tp_timer_count + TEST_RAW_TP__TIMEOUT, actual_bandwidth_tx/div, actual_bandwidth_rx/div);
#else
	ESP_LOGI(TAG, "%lu-%lu sec Tx:%lu Rx:%lu Kbps", raw_tp_timer_count, raw_tp_timer_count + TEST_RAW_TP__TIMEOUT, (unsigned long)actual_bandwidth_tx/div, (unsigned long)actual_bandwidth_rx/div);
#endif
	raw_tp_timer_count+=TEST_RAW_TP__TIMEOUT;
	test_raw_tx_len = test_raw_rx_len = 0;
}

static void raw_tp_tx_task(void const* pvParameters)
{
	int ret;
	static uint16_t seq_num = 0;
	uint8_t *raw_tp_tx_buf = NULL;
	uint32_t *ptr = NULL;
	uint32_t i = 0;
	g_h.funcs->_h_sleep(5);

	buf_mp_g = mempool_create(MAX_TRANSPORT_BUFFER_SIZE);
#ifdef CONFIG_ESP_CACHE_MALLOC
	assert(channel->memp);
#endif

	while (1) {

#if CONFIG_H_LOWER_MEMCOPY
		raw_tp_tx_buf = (uint8_t*)g_h.funcs->_h_calloc(1, MAX_TRANSPORT_BUFFER_SIZE);

		ptr = (uint32_t*) raw_tp_tx_buf;
		for (i=0; i<(TEST_RAW_TP__BUF_SIZE/4-1); i++, ptr++)
			*ptr = 0xBAADF00D;

		ret = esp_hosted_tx(ESP_TEST_IF, 0, raw_tp_tx_buf, TEST_RAW_TP__BUF_SIZE, H_BUFF_ZEROCOPY, H_DEFLT_FREE_FUNC);

#else
		raw_tp_tx_buf = mempool_alloc(buf_mp_g, MAX_TRANSPORT_BUFFER_SIZE, true);

		ptr = (uint32_t*) (raw_tp_tx_buf + H_ESP_PAYLOAD_HEADER_OFFSET);
		for (i=0; i<(TEST_RAW_TP__BUF_SIZE/4-1); i++, ptr++)
			*ptr = 0xBAADF00D;

		ret = esp_hosted_tx(ESP_TEST_IF, 0, raw_tp_tx_buf, TEST_RAW_TP__BUF_SIZE, H_BUFF_ZEROCOPY, stats_mempool_free);
#endif
		if (ret != STM_OK) {
			ESP_LOGE(TAG, "Failed to send to queue\n");
			continue;
		}
#if CONFIG_H_LOWER_MEMCOPY
		g_h.funcs->_h_free(raw_tp_tx_buf);
#endif
		test_raw_tx_len += (TEST_RAW_TP__BUF_SIZE);
		seq_num++;
	}
}

static void process_raw_tp_flags(uint8_t cap)
{
	test_raw_tp_cleanup();

	if (test_raw_tp) {
		hosted_timer_handler = g_h.funcs->_h_timer_start(TEST_RAW_TP__TIMEOUT,
				RPC__TIMER_PERIODIC, raw_tp_timer_func, NULL);
		if (!hosted_timer_handler) {
			ESP_LOGE(TAG, "Failed to create timer\n\r");
			return;
		}
		log_raw_tp_stats_timer_running = 1;

		ESP_LOGD(TAG, "capabilities: %d", cap);
		if ((cap & ESP_TEST_RAW_TP__HOST_TO_ESP) ||
			(cap & ESP_TEST_RAW_TP__BIDIRECTIONAL)) {
			raw_tp_tx_task_id = g_h.funcs->_h_thread_create("raw_tp_tx", DFLT_TASK_PRIO,
				RAW_TP_TX_TASK_STACK_SIZE, raw_tp_tx_task, NULL);
			assert(raw_tp_tx_task_id);
		}
	}
}

static void start_test_raw_tp(void)
{
	test_raw_tp = 1;
}

static void stop_test_raw_tp(void)
{
	test_raw_tp = 0;
}

void process_test_capabilities(uint8_t cap)
{
	ESP_LOGI(TAG, "ESP peripheral capabilities: 0x%x", cap);
	if ((cap & ESP_TEST_RAW_TP) == ESP_TEST_RAW_TP) {
		start_test_raw_tp();
		ESP_LOGI(TAG, "***** Host Raw throughput Testing (report per %u sec) *****\n\r",TEST_RAW_TP__TIMEOUT);
	} else {
		ESP_LOGW(TAG, "Raw Throughput testing not enabled on slave. Stopping test.");
		stop_test_raw_tp();
	}
	process_raw_tp_flags(H_TEST_RAW_TP_DIR);
}

void update_test_raw_tp_rx_len(uint16_t len)
{
	test_raw_rx_len+=(len);
}

#endif
#if H_MEM_STATS
struct mem_stats h_stats_g;
#endif

#if ESP_PKT_STATS
void stats_timer_func(void * arg)
{
	ESP_LOGI(TAG, "slave: sta_rx_in: %lu sta_rx_out: %lu sta_tx_in [pass: %lu drop: %lu] sta_tx_out: %lu ",
			pkt_stats.sta_rx_in,pkt_stats.sta_rx_out,
			pkt_stats.sta_tx_in_pass, pkt_stats.sta_tx_in_drop, pkt_stats.sta_tx_out);
	ESP_LOGI(TAG, "wifi_tx_throttling %u", wifi_tx_throttling);
}
#endif

void create_debugging_tasks(void)
{
#if ESP_PKT_STATS
	if (ESP_PKT_STATS_REPORT_INTERVAL) {
		ESP_LOGI(TAG, "Start Pkt_stats reporting thread [timer: %u sec]", ESP_PKT_STATS_REPORT_INTERVAL);
		pkt_stats_thread = g_h.funcs->_h_timer_start(ESP_PKT_STATS_REPORT_INTERVAL,
				RPC__TIMER_PERIODIC, stats_timer_func, NULL);
		assert(pkt_stats_thread);
	}
#endif
}