210 lines
7.7 KiB
C
210 lines
7.7 KiB
C
#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);
|
|
DWORD mock_SetFilePointer(HANDLE hFile, LONG lDistanceToMove, PLONG lpDistanceToMoveHigh, DWORD dwMoveMethod);
|
|
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);
|
|
BOOL mock_MoveFileExW(WCHAR *lpExistingFileName, WCHAR *lpNewFileName, DWORD dwFlags);
|
|
#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);
|
|
off_t mock_lseek(int fd, off_t offset, int whence);
|
|
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_SetFilePointer mock_SetFilePointer
|
|
#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
|
|
#define sys_MoveFileExW mock_MoveFileExW
|
|
|
|
// 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_lseek mock_lseek
|
|
#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_SetFilePointer SetFilePointer
|
|
#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
|
|
#define sys_MoveFileExW MoveFileExW
|
|
|
|
// 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_lseek lseek
|
|
#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
|