#ifndef SYSTEM_INCLUDED
#define SYSTEM_INCLUDED

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>

#ifdef _WIN32

#define WIN32_LEAN_AND_MEAN

#include <direct.h> // _mkdir
#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>

#else

#include <poll.h>
#include <time.h>
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/file.h>
#include <sys/socket.h>
#include <sys/random.h>
#include <arpa/inet.h>

#define SOCKET int
#define INVALID_SOCKET ((SOCKET) -1)

#endif

#ifdef BUILD_TEST

void startup_simulation(uint64_t seed_);
int  spawn_simulated_process(char *args);
void update_simulation(void);
void cleanup_simulation(void);
uint64_t simulation_random_number(void);

void*  mock_malloc(size_t len);
void*  mock_realloc(void *ptr, size_t len);
void   mock_free(void *ptr);
int    mock_remove(char *path);
int    mock_rename(char *oldpath, char *newpath);
SOCKET mock_socket(int domain, int type, int protocol);
int    mock_bind(SOCKET fd, void *addr, size_t addr_len);
int    mock_listen(SOCKET fd, int backlog);
SOCKET mock_accept(SOCKET fd, void *addr, socklen_t *addr_len);
int    mock_getsockopt(SOCKET fd, int level, int optname, void *optval, socklen_t *optlen);
int    mock_recv(SOCKET fd, void *dst, int len, int flags);
int    mock_send(SOCKET fd, void *src, int len, int flags);
int    mock_connect(SOCKET fd, void *addr, size_t addr_len);

#ifdef _WIN32
int      mock_closesocket(SOCKET fd);
int      mock_ioctlsocket(SOCKET fd, long cmd, u_long *argp);
HANDLE   mock_CreateFileW(WCHAR *lpFileName, DWORD dwDesiredAccess, DWORD dwShareMode, LPSECURITY_ATTRIBUTES lpSecurityAttributes, DWORD dwCreationDisposition, DWORD dwFlagsAndAttributes, HANDLE hTemplateFile);
BOOL     mock_CloseHandle(HANDLE handle);
BOOL     mock_LockFile(HANDLE hFile, DWORD dwFileOffsetLow, DWORD dwFileOffsetHigh, DWORD nNumberOfBytesToLockLow, DWORD nNumberOfBytesToLockHigh);
BOOL     mock_UnlockFile(HANDLE hFile, DWORD dwFileOffsetLow, DWORD dwFileOffsetHigh, DWORD nNumberOfBytesToUnlockLow, DWORD nNumberOfBytesToUnlockHigh);
BOOL     mock_FlushFileBuffers(HANDLE handle);
BOOL     mock_ReadFile(HANDLE handle, char *dst, DWORD len, DWORD *num, OVERLAPPED *ov);
BOOL     mock_WriteFile(HANDLE handle, char *src, DWORD len, DWORD *num, OVERLAPPED *ov);
BOOL     mock_GetFileSizeEx(HANDLE handle, LARGE_INTEGER *buf);
BOOL     mock_QueryPerformanceCounter(LARGE_INTEGER *lpPerformanceCount);
BOOL     mock_QueryPerformanceFrequency(LARGE_INTEGER *lpFrequency);
char*    mock__fullpath(char *path, char *dst, int cap);
int      mock__mkdir(char *path);
HANDLE   mock_FindFirstFileA(char *lpFileName, WIN32_FIND_DATAA *lpFindFileData);
BOOL     mock_FindNextFileA(HANDLE hFindFile, WIN32_FIND_DATAA *lpFindFileData);
BOOL     mock_FindClose(HANDLE hFindFile);
#else
int     mock_clock_gettime(clockid_t clockid, struct timespec *tp);
int     mock_open(char *path, int flags, int mode);
int     mock_close(int fd);
int     mock_flock(int fd, int op);
int     mock_fsync(int fd);
int     mock_read(int fd, char *dst, int len);
int     mock_write(int fd, char *src, int len);
int     mock_fstat(int fd, struct stat *buf);
int     mock_mkstemp(char *path);
char*   mock_realpath(char *path, char *dst);
int     mock_mkdir(char *path, mode_t mode);
int     mock_fcntl(int fd, int cmd, ...);
DIR*    mock_opendir(char *name);
struct dirent* mock_readdir(DIR *dirp);
int     mock_closedir(DIR *dirp);
#endif

// Common
#define sys_malloc           mock_malloc
#define sys_realloc          mock_realloc
#define sys_free             mock_free
#define sys_remove           mock_remove
#define sys_rename           mock_rename
#define sys_socket           mock_socket
#define sys_bind             mock_bind
#define sys_listen           mock_listen
#define sys_accept           mock_accept
#define sys_getsockopt       mock_getsockopt
#define sys_recv             mock_recv
#define sys_send             mock_send
#define sys_connect          mock_connect

// Windows
#define sys__mkdir           mock__mkdir
#define sys_closesocket      mock_closesocket
#define sys_ioctlsocket      mock_ioctlsocket
#define sys_CreateFileW      mock_CreateFileW
#define sys_CloseHandle      mock_CloseHandle
#define sys_LockFile         mock_LockFile
#define sys_UnlockFile       mock_UnlockFile
#define sys_FlushFileBuffers mock_FlushFileBuffers
#define sys_ReadFile         mock_ReadFile
#define sys_WriteFile        mock_WriteFile
#define sys_GetFileSizeEx    mock_GetFileSizeEx
#define sys__fullpath        mock__fullpath
#define sys_QueryPerformanceCounter   mock_QueryPerformanceCounter
#define sys_QueryPerformanceFrequency mock_QueryPerformanceFrequency
#define sys_FindFirstFileA   mock_FindFirstFileA
#define sys_FindNextFileA    mock_FindNextFileA
#define sys_FindClose        mock_FindClose

// Linux
#define sys_mkdir            mock_mkdir
#define sys_open             mock_open
#define sys_close            mock_close
#define sys_flock            mock_flock
#define sys_fsync            mock_fsync
#define sys_read             mock_read
#define sys_write            mock_write
#define sys_fstat            mock_fstat
#define sys_mkstemp          mock_mkstemp
#define sys_realpath         mock_realpath
#define sys_clock_gettime    mock_clock_gettime
#define sys_fcntl            mock_fcntl
#define sys_getrandom        mock_getrandom
#define sys_opendir          mock_opendir
#define sys_readdir          mock_readdir
#define sys_closedir         mock_closedir

#else

// Common
#define sys_malloc           malloc
#define sys_realloc          realloc
#define sys_free             free
#define sys_remove           remove
#define sys_rename           rename
#define sys_socket           socket
#define sys_bind             bind
#define sys_listen           listen
#define sys_accept           accept
#define sys_getsockopt       getsockopt
#define sys_recv             recv
#define sys_send             send
#define sys_connect          connect

// Windows
#define sys__mkdir           _mkdir
#define sys_closesocket      closesocket
#define sys_ioctlsocket      ioctlsocket
#define sys_CreateFileW      CreateFileW
#define sys_CloseHandle      CloseHandle
#define sys_LockFile         LockFile
#define sys_UnlockFile       UnlockFile
#define sys_FlushFileBuffers FlushFileBuffers
#define sys_ReadFile         ReadFile
#define sys_WriteFile        WriteFile
#define sys_GetFileSizeEx    GetFileSizeEx
#define sys__fullpath        _fullpath
#define sys_QueryPerformanceCounter   QueryPerformanceCounter
#define sys_QueryPerformanceFrequency QueryPerformanceFrequency
#define sys_FindFirstFileA   FindFirstFileA
#define sys_FindNextFileA    FindNextFileA
#define sys_FindClose        FindClose

// Linux
#define sys_mkdir            mkdir
#define sys_open             open
#define sys_close            close
#define sys_flock            flock
#define sys_fsync            fsync
#define sys_read             read
#define sys_write            write
#define sys_fstat            fstat
#define sys_mkstemp          mkstemp
#define sys_realpath         realpath
#define sys_clock_gettime    clock_gettime
#define sys_fcntl            fcntl
#define sys_opendir          opendir
#define sys_readdir          readdir
#define sys_closedir         closedir

#endif
#endif // SYSTEM_INCLUDED