#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <netinet/ip.h>
#include <netinet/ip6.h>
#include <netinet/tcp.h>
#include <netinet/ether.h>
#include <pcap/sll.h>
#include <pcap/pcap.h>
#include "gtest/gtest.h"

#include "stellar/stratum_decoder.h"

#include <stellar/session.h>


#define LINUX_COOKED_CAPTURE 20

struct pcap_loop_arg
{
	pcap_t *pcap_handle;
	void *ctx;
};

extern struct stratum_field *stratum_data_process(struct stratum_decoder_info *stratum_decoder_info, const char *tcpdata, size_t datalen);
extern void free_stratum_filed(struct stratum_field *stratum_field);
extern "C" void *stratum_decoder_init(struct stellar *st);

struct stratum_decoder_info *stratum_decoder_info = NULL;

static void pcap_handle_cb(pcap_loop_arg *userarg, const struct pcap_pkthdr *pkthdr, const u_char *packet)
{
	struct pcap_loop_arg *arg = (struct pcap_loop_arg *)userarg;
	int payload_len = 0;
	char *payload = NULL;
	unsigned short eth_proto_type;
	unsigned char *ip_header;

	int data_link_type = pcap_datalink(arg->pcap_handle);
	switch (data_link_type) {
		case DLT_EN10MB:
			eth_proto_type = ntohs(*(unsigned short *)(packet + 12));
			ip_header = (unsigned char *)(packet + sizeof(struct ethhdr));
			break;
		case 276://DLT_LINUX_SLL2
			eth_proto_type = ntohs(*(unsigned short *)packet);
			ip_header = (unsigned char *)(packet + LINUX_COOKED_CAPTURE);
			break;
		default:
			return;
	}

	if (eth_proto_type == ETH_P_IP) {
		int l4_proto = *(unsigned char *)(ip_header + 9);
		if (l4_proto == IPPROTO_TCP) {
			int ip_total_len = ntohs(*(unsigned short *)(ip_header + 2));
			int ip_header_len = (*(unsigned char *)ip_header & 0x0f) * 4;
			int tcp_header_len = 4 * ((*(unsigned char *)(ip_header + sizeof(iphdr) + 12) & 0xf0) >> 4);
			payload_len = ip_total_len - ip_header_len - tcp_header_len;
			payload = (char *)ip_header + ip_header_len + tcp_header_len;
		} else {
			return;
		}
		
	} else if (eth_proto_type == ETH_P_IPV6) {
		int l4_proto = *(unsigned char *)(ip_header + 6);
		if (l4_proto == IPPROTO_TCP) {
			int tcp_header_len = 4 * ((*(unsigned char *)(ip_header + sizeof(struct ip6_hdr) + 12) & 0xf0) >> 4);
			payload_len = pkthdr->caplen - (ip_header - packet) - sizeof(struct ip6_hdr) - tcp_header_len;
			payload = (char *)ip_header + sizeof(struct ip6_hdr) + tcp_header_len;
		} else {
			return;
		}
	} else {
		return;
	}
	
    struct stratum_field *stratum = stratum_data_process(stratum_decoder_info, payload, payload_len);
    if (stratum != NULL)
    {
        arg->ctx = stratum;
        pcap_breakloop(arg->pcap_handle);
    }
}

TEST(stratum_decoder, parse01)
{
    char error[100];
    struct pcap_loop_arg arg;
    pcap_t *handle = pcap_open_offline("pcap/01-bch_f2pool_cpuminer_1.pcapng", error);
    ASSERT_NE(handle, nullptr);

    memset(&arg, 0, sizeof(arg));
    arg.pcap_handle = handle;
    pcap_loop(handle, -1, (pcap_handler)pcap_handle_cb, (u_char *)&arg);

    struct stratum_field *stratum = (struct stratum_field *)arg.ctx;
    EXPECT_EQ(stratum->type, OTHER);
    EXPECT_STREQ((const char *)stratum->mining_program.iov_base, "cpuminer/2.5.1");
    EXPECT_EQ(stratum->mining_pools.iov_len, 0);
    EXPECT_STREQ((const char *)stratum->mining_subscribe.iov_base, "{\"id\": 1, \"method\": \"mining.subscribe\", \"params\": [\"cpuminer/2.5.1\"]}\n");

    pcap_close(handle);
    free_stratum_filed(stratum);
}

TEST(stratum_decoder, parse02)
{
    char error[100];
    struct pcap_loop_arg arg;
    pcap_t *handle = pcap_open_offline("pcap/02-eth-antpool.pcapng", error);
    ASSERT_NE(handle, nullptr);

    memset(&arg, 0, sizeof(arg));
    arg.pcap_handle = handle;
    pcap_loop(handle, -1, (pcap_handler)pcap_handle_cb, (u_char *)&arg);

    struct stratum_field *stratum = (struct stratum_field *)arg.ctx;
    EXPECT_EQ(stratum->type, ETH);
    EXPECT_STREQ((const char *)stratum->mining_program.iov_base, "lolMinerWorker");
    EXPECT_EQ(stratum->mining_pools.iov_len, 0);
    EXPECT_STREQ((const char *)stratum->mining_subscribe.iov_base, "{\"id\":1,\"jsonrpc\":\"2.0\",\"method\":\"eth_submitLogin\",\"worker\":\"lolMinerWorker\",\"params\":[\"0xa0279Ad2aA6BDb440f041Aa0947178431d9Bd253.lolMinerWorker\", \"x\"]} \n");

    pcap_close(handle);
    free_stratum_filed(stratum);
}

TEST(stratum_decoder, parse03)
{
    char error[100];
    struct pcap_loop_arg arg;
    pcap_t *handle = pcap_open_offline("pcap/03-xmr_f2pool_nanominer_1.pcapng", error);
    ASSERT_NE(handle, nullptr);

    memset(&arg, 0, sizeof(arg));
    arg.pcap_handle = handle;
    pcap_loop(handle, -1, (pcap_handler)pcap_handle_cb, (u_char *)&arg);

    struct stratum_field *stratum = (struct stratum_field *)arg.ctx;
    EXPECT_EQ(stratum->type, OTHER);
    EXPECT_STREQ((const char *)stratum->mining_program.iov_base, "nanominer/3.3.5-cuda11");
    EXPECT_EQ(stratum->mining_pools.iov_len, 0);
    EXPECT_STREQ((const char *)stratum->mining_subscribe.iov_base, "{\"id\": 1, \"method\": \"login\", \"params\": { \"login\":\"yhzzhk.001.001\",\"pass\" : \"x\", \"agent\" : \"nanominer/3.3.5-cuda11\"}}\n");

    pcap_close(handle);
    free_stratum_filed(stratum);
}

TEST(stratum_decoder, parse04)
{
    char error[100];
    struct pcap_loop_arg arg;
    pcap_t *handle = pcap_open_offline("pcap/04-zec-antpool.pcapng", error);
    ASSERT_NE(handle, nullptr);

    memset(&arg, 0, sizeof(arg));
    arg.pcap_handle = handle;
    pcap_loop(handle, -1, (pcap_handler)pcap_handle_cb, (u_char *)&arg);

    struct stratum_field *stratum = (struct stratum_field *)arg.ctx;
    EXPECT_EQ(stratum->type, OTHER);
    EXPECT_STREQ((const char *)stratum->mining_program.iov_base, "nheqminer/0.5c");
    EXPECT_STREQ((const char*)stratum->mining_pools.iov_base, "stratum-zec.antpool.com");
    EXPECT_STREQ((const char *)stratum->mining_subscribe.iov_base, "{\"id\":1,\"method\":\"mining.subscribe\",\"params\":[\"nheqminer/0.5c\", null,\"stratum-zec.antpool.com\",\"8899\"]}\n");

    pcap_close(handle);
    free_stratum_filed(stratum);
}

int main(int argc, char **argv)
{
	testing::InitGoogleTest(&argc, argv);
	
    stratum_decoder_info = (struct stratum_decoder_info*)stratum_decoder_init(NULL);

	int result = RUN_ALL_TESTS();

	return result;
}