#include <sys/socket.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <arpa/inet.h> 
#include <signal.h>
#include <sys/mman.h>
#include <fcntl.h>

#include "Devices//tmsgheaders.h"
#include "Devices//lmx2594.h"
#include "Devices//max2870.h"
#include "Devices//lmk04821.h"
#include "Devices//ad9912.h"
#include "Devices/pe43711.h"

#include "command.h"

#define REQUESTED_MEMORY_SIZE  0x1000
#define SERVER_PORT 5025
#define BACKLOG 10

volatile int conn_fd = 0;
volatile int pci_fd = 0;
int listen_fd = 0;

void *bar1;

//Обработчик ошибок
void error(const char *msg)
{
	perror(msg);

	if (listen_fd != 0) 
	{
		close(listen_fd);
	}
    if (pci_fd != 0)
    {
        close(pci_fd);
    }
    munmap(bar1, REQUESTED_MEMORY_SIZE);
	exit(1);
} 

//Обработчик сигнала SIGINT завершения программы при нажатии Ctrl+C
void handleCloseSignal(int signal)
{
	if (signal == SIGINT) 
	{
		printf("\nCaught signal %d, closing socket and exiting...\n", signal);
		if (conn_fd != 0) 
		{
			close(conn_fd);
		}
		if (listen_fd != 0) 
		{
			close(listen_fd);
		}
        if (pci_fd != 0)
        {
            close(pci_fd);
        }
        munmap(bar1, REQUESTED_MEMORY_SIZE);
		exit(0);
	}
}

int main(int argc, char *argv[])
{
	char *filename = "/dev/MyDmaModule";

	// Длина структуры адреса
	socklen_t client_len;
	// Переменная для количества байт принятых из сокета и для хранения ошибок
	ssize_t n;

	struct sockaddr_in serv_addr, client_addr;
	
	// Приёмный буффер для сокета
	char recvBuff[1024];
	// Разделители команд
	const char charSeparator[] = {"\n"};
	// Указатель команды
	char *ptrLexeme = NULL;
	// Указатель для хранения контекста токенизации
	char *savePtr = NULL;

    pci_fd = open(filename, O_RDWR | O_SYNC);
    if (pci_fd == -1) {
        int error = errno;
        fprintf(stderr, "Cannot open PCIe device file: %s\n", strerror(error));
        return 1;
    }

    bar1 = mmap(NULL, REQUESTED_MEMORY_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, pci_fd, 0);
    if (bar1 == MAP_FAILED) {
        perror("mmap");
        close(pci_fd);
        return 1;
    }

    lmk04821_a_init(bar1);
    usleep(500);
    lmk04821_b_init(bar1);
	pe43711_att_1_init(bar1);
	usleep(1);
	pe43711_att_2_init(bar1);
    rst_for_fpga(bar1);
    shift_reg(bar1);
    ad9912_init(bar1);
    lmx2594_init(bar1);
	max2870_init(bar1);

	// Установка обработчика сигналов
	signal(SIGINT, handleCloseSignal);

	listen_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (listen_fd < 0)
	{
		error("Error : Could not create socket!");
	} 

	memset(&serv_addr, 0, sizeof(serv_addr));

	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	serv_addr.sin_port = htons(SERVER_PORT); 

	if (setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &(int){1}, sizeof(int)) < 0)
    {
		error("setsockopt(SO_REUSEADDR) failed");
	}

	if (bind(listen_fd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0) 
	{
        error("Error : Bind failed");
    }

	if (listen(listen_fd, BACKLOG) < 0) 
	{
        error("Error : Listen failed");
    }
	printf("Server is listening on port %d...\n", SERVER_PORT);
	
	while(1) {
		client_len = sizeof(client_addr);
		conn_fd = accept(listen_fd, (struct sockaddr*)&client_addr, &client_len);
		if (conn_fd < 0) 
		{
			perror("Error : Accept failed");
			continue;
		}
		printf("Connection established with client\n");

		//Очистка буфера
		memset(recvBuff, 0, sizeof(recvBuff));

		while ((n = recv(conn_fd, recvBuff, sizeof(recvBuff) - 1, 0)) > 0)
		{
			recvBuff[n] = 0;
		
			if(fputs(recvBuff, stdout) == EOF)
			{
				printf("\n Error : Fputs error\n");
				break;
			}
		
			// Указатель команды
			ptrLexeme = NULL;
			// Указатель для хранения контекста токенизации
			savePtr = NULL;

			// Инициализируем функцию и ищем команду в строке
			ptrLexeme = strtok_r(recvBuff, charSeparator, &savePtr);
			
			// Выполняем команды, пока не дойдём до конца приёмного буффера
			while (ptrLexeme) {
				// Запуск парсера команд
				processCommand(ptrLexeme);

				// Ищем команды разделенные разделителем
				ptrLexeme = strtok_r(NULL, charSeparator, &savePtr);
			}
		} 

		if (n == 0)
		{
			printf("\n Client closed the connection\n");
		}
		else if (n < 0)
		{
			perror("Read error");
		} 

		close(conn_fd);
		printf("Waiting for new client connection...\n");
	}

	return 0;
}