/*
 * Copyright (C) 2017 - 2022 Xilinx, Inc.
 * Copyright (C) 2022 - 2023 Advanced Micro Devices, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include "lwip/inet.h"
#include "lwip/init.h"
#include "lwip/udp.h"
#include "lwip/tcpip.h"
#include "lwip/timeouts.h"
#include "netif/xadapter.h"
#include "xil_printf.h"
#include "xparameters.h"
#include <sleep.h>
#include <FreeRTOS.h>
#include <task.h>

#define PLATFORM_EMAC_BASEADDR 0xE000B000

#if LWIP_DHCP == 1
#include "lwip/dhcp.h"
extern volatile int dhcp_timoutcntr;
err_t dhcp_start(struct netif *netif);
#endif

#ifdef XPS_BOARD_ZCU102
#if defined(XPAR_XIICPS_0_DEVICE_ID) || defined(XPAR_XIICPS_0_BASEADDR)
int IicPhyReset(void);
#endif
#endif

void print_app_header();
void start_application();

#define THREAD_STACKSIZE 1024

#define DEFAULT_IP_ADDRESS "192.168.1.10"
#define DEFAULT_IP_MASK "255.255.255.0"
#define DEFAULT_GW_ADDRESS "192.168.1.1"

struct netif server_netif;

// u32_t sys_now(void) { return xTaskGetTickCount() * portTICK_PERIOD_MS; }

static void print_ip(char *msg, ip_addr_t *ip) {
  xil_printf(msg);
  xil_printf("%d.%d.%d.%d\n\r", ip4_addr1(ip), ip4_addr2(ip), ip4_addr3(ip),
             ip4_addr4(ip));
}

static void print_ip_settings(ip_addr_t *ip, ip_addr_t *mask, ip_addr_t *gw) {
  print_ip("Board IP:       ", ip);
  print_ip("Netmask :       ", mask);
  print_ip("Gateway :       ", gw);
}

static void assign_default_ip(ip_addr_t *ip, ip_addr_t *mask, ip_addr_t *gw) {
  int err;

  xil_printf("Configuring default IP %s \r\n", DEFAULT_IP_ADDRESS);

  err = inet_aton(DEFAULT_IP_ADDRESS, ip);
  if (!err)
    xil_printf("Invalid default IP address: %d\r\n", err);

  err = inet_aton(DEFAULT_IP_MASK, mask);
  if (!err)
    xil_printf("Invalid default IP MASK: %d\r\n", err);

  err = inet_aton(DEFAULT_GW_ADDRESS, gw);
  if (!err)
    xil_printf("Invalid default gateway address: %d\r\n", err);
}

void network_thread() {
#if LWIP_DHCP == 1
  int mscnt = 0;
#endif
  /* the mac address of the board. this should be unique per board */
  u8_t mac_ethernet_address[] = {0x00, 0x0a, 0x35, 0x00, 0x01, 0x02};

  /* Add network interface to the netif_list, and set it as default */
  if (!xemac_add(&server_netif, NULL, NULL, NULL, mac_ethernet_address,
                 PLATFORM_EMAC_BASEADDR)) {
    xil_printf("Error adding N/W interface\r\n");
    return;
  }

  netif_set_default(&server_netif);

  /* specify that the network if is up */
  netif_set_up(&server_netif);

  /* start packet receive thread - required for lwIP operation */
  sys_thread_new("xemacif_input_thread",
  		(void(*)(void*))xemacif_input_thread, &server_netif,
  		THREAD_STACKSIZE, DEFAULT_THREAD_PRIO);

#if LWIP_DHCP == 1
  dhcp_start(&server_netif);
  while (1) {
    vTaskDelay(DHCP_FINE_TIMER_MSECS / portTICK_RATE_MS);
    dhcp_fine_tmr();
    mscnt += DHCP_FINE_TIMER_MSECS;
    if (mscnt >= DHCP_COARSE_TIMER_SECS * 1000) {
      dhcp_coarse_tmr();
      mscnt = 0;
    }
  }
#else
  // vTaskDelete(NULL);
#endif
}

void udp_receiver(void *arg, struct udp_pcb *pcb, struct pbuf *p,
    const ip_addr_t *addr, u16_t port){
      u32_t now = sys_now();
      xil_printf("%i received len: %i tot_len: %i data: ", now, p->len, p->tot_len);
      char * pointer = p->payload;
      for(uint8_t i = 0; i < p->len; i++) {
        xil_printf(" %02x", *(pointer + i));
      }
      xil_printf("\n");
}

int main_thread(void * _) {

#if LWIP_DHCP == 1
  int mscnt = 0;
#endif

#ifdef XPS_BOARD_ZCU102
  IicPhyReset();
#endif
  xil_printf("\n\r\n\r");
  xil_printf("-----lwIP Socket Mode UDP Client Application------\r\n");

  /* initialize lwIP before calling sys_thread_new */
  //lwip_init();
  tcpip_init(NULL, NULL);
  //give init some time
  vTaskDelay(1000 * portTICK_PERIOD_MS);

  /* any thread using lwIP should be created using sys_thread_new */
  network_thread();

#if LWIP_DHCP == 1
  while (1) {
    vTaskDelay(DHCP_FINE_TIMER_MSECS / portTICK_RATE_MS);
    if (server_netif.ip_addr.addr) {
      xil_printf("DHCP request success\r\n");
      break;
    }
    mscnt += DHCP_FINE_TIMER_MSECS;
    if (mscnt >= 10000) {
      xil_printf("ERROR: DHCP request timed out\r\n");
      assign_default_ip(&(server_netif.ip_addr), &(server_netif.netmask),
                        &(server_netif.gw));
      break;
    }
  }

#else
  assign_default_ip(&(server_netif.ip_addr), &(server_netif.netmask),
                    &(server_netif.gw));
#endif

  print_ip_settings(&(server_netif.ip_addr), &(server_netif.netmask),
                    &(server_netif.gw));
  xil_printf("\r\n");

  /* print all application headers */
  // print_app_header();
  xil_printf("\r\n");


  uint8_t data[] = {'1','2','3','4','5'};

  struct pbuf* tx = pbuf_alloc_reference(data, sizeof(data), PBUF_REF);

  ip_addr_t addr  = IPADDR4_INIT_BYTES(192,168,1,5);

  struct udp_pcb *udpecho_raw_pcb = udp_new();

  u32_t now = sys_now();
  xil_printf("%i pcb: %p\r\n", now, udpecho_raw_pcb);

  udp_sendto(udpecho_raw_pcb, tx, &addr, 8100);
  udp_recv(udpecho_raw_pcb,udp_receiver,NULL);

  while (1) {
    sys_check_timeouts();
    
    vTaskDelay(200 * portTICK_PERIOD_MS);
  }

  pbuf_free(tx);

  /* start the application*/
  // start_application();
  return 0;
}

int testEth(){
  sys_thread_new("main_thread", (void(*)(void*))main_thread, 0,
			THREAD_STACKSIZE, DEFAULT_THREAD_PRIO);
	vTaskStartScheduler();
}