/* copyright (c) 2007 magnus auvinen, see licence.txt for more info */ #include #include #include #include #include #include /*#include "detect.h"*/ #include "system.h" /*#include "e_console.h"*/ #if defined(CONF_FAMILY_UNIX) #include #include /* unix net includes */ #include #include #include #include #include #include #include #include #include #include #elif defined(CONF_FAMILY_WINDOWS) #define WIN32_LEAN_AND_MEAN #define _WIN32_WINNT 0x0501 /* required for mingw to get getaddrinfo to work */ #include #include #include #include #include #include #define EWOULDBLOCK WSAEWOULDBLOCK #else #error NOT IMPLEMENTED #endif #if defined(__cplusplus) extern "C" { #endif IOHANDLE io_stdin() { return (IOHANDLE)stdin; } IOHANDLE io_stdout() { return (IOHANDLE)stdout; } IOHANDLE io_stderr() { return (IOHANDLE)stderr; } static DBG_LOGGER loggers[16]; static int num_loggers = 0; static NETSTATS network_stats = {0}; void dbg_logger(DBG_LOGGER logger) { loggers[num_loggers++] = logger; } void dbg_assert_imp(const char *filename, int line, int test, const char *msg) { if(!test) { dbg_msg("assert", "%s(%d): %s", filename, line, msg); dbg_break(); } } void dbg_break() { *((unsigned*)0) = 0x0; } void dbg_msg(const char *sys, const char *fmt, ...) { va_list args; char str[1024*4]; char *msg; int i, len; str_format(str, sizeof(str), "[%08x][%s]: ", (int)time(0), sys); len = strlen(str); msg = (char *)str + len; va_start(args, fmt); #if defined(CONF_FAMILY_WINDOWS) _vsnprintf(msg, sizeof(str)-len, fmt, args); #else vsnprintf(msg, sizeof(str)-len, fmt, args); #endif va_end(args); for(i = 0; i < num_loggers; i++) loggers[i](str); } static void logger_stdout(const char *line) { printf("%s\n", line); fflush(stdout); } static void logger_debugger(const char *line) { #if defined(CONF_FAMILY_WINDOWS) OutputDebugString(line); OutputDebugString("\n"); #endif } IOHANDLE logfile = 0; static void logger_file(const char *line) { io_write(logfile, line, strlen(line)); io_write(logfile, "\n", 1); io_flush(logfile); } void dbg_logger_stdout() { dbg_logger(logger_stdout); } void dbg_logger_debugger() { dbg_logger(logger_debugger); } void dbg_logger_file(const char *filename) { logfile = io_open(filename, IOFLAG_WRITE); if(logfile) dbg_logger(logger_file); else dbg_msg("dbg/logger", "failed to open '%s' for logging", filename); } /* */ int memory_alloced = 0; struct memheader { const char *filename; int line; int size; struct memheader *prev; struct memheader *next; }; struct memtail { int guard; }; static struct memheader *first = 0; int mem_allocated() { return memory_alloced; } void *mem_alloc_debug(const char *filename, int line, unsigned size, unsigned alignment) { /* TODO: fix alignment */ /* TODO: add debugging */ struct memheader *header = (struct memheader *)malloc(size+sizeof(struct memheader)+sizeof(struct memtail)); struct memtail *tail = (struct memtail *)(((char*)(header+1))+size); header->size = size; header->filename = filename; header->line = line; memory_alloced += header->size; tail->guard = 0xbaadc0de; header->prev = (struct memheader *)0; header->next = first; if(first) first->prev = header; first = header; /*dbg_msg("mem", "++ %p", header+1); */ return header+1; } void mem_free(void *p) { if(p) { struct memheader *header = (struct memheader *)p - 1; struct memtail *tail = (struct memtail *)(((char*)(header+1))+header->size); if(tail->guard != 0xbaadc0de) dbg_msg("mem", "!! %p", p); /* dbg_msg("mem", "-- %p", p); */ memory_alloced -= header->size; if(header->prev) header->prev->next = header->next; else first = header->next; if(header->next) header->next->prev = header->prev; free(header); } } void mem_debug_dump() { char buf[1024]; struct memheader *header = first; IOHANDLE f = io_open("memory.txt", IOFLAG_WRITE); while(header) { str_format(buf, sizeof(buf), "%s(%d): %d\n", header->filename, header->line, header->size); io_write(f, buf, strlen(buf)); header = header->next; } io_close(f); } void mem_copy(void *dest, const void *source, unsigned size) { memcpy(dest, source, size); } void mem_move(void *dest, const void *source, unsigned size) { memmove(dest, source, size); } void mem_zero(void *block,unsigned size) { memset(block, 0, size); } IOHANDLE io_open(const char *filename, int flags) { if(flags == IOFLAG_READ) return (IOHANDLE)fopen(filename, "rb"); if(flags == IOFLAG_WRITE) return (IOHANDLE)fopen(filename, "wb"); return 0x0; } unsigned io_read(IOHANDLE io, void *buffer, unsigned size) { return fread(buffer, 1, size, (FILE*)io); } unsigned io_skip(IOHANDLE io, unsigned size) { fseek((FILE*)io, size, SEEK_CUR); return size; } int io_seek(IOHANDLE io, int offset, int origin) { int real_origin; switch(origin) { case IOSEEK_START: real_origin = SEEK_SET; break; case IOSEEK_CUR: real_origin = SEEK_CUR; break; case IOSEEK_END: real_origin = SEEK_END; } return fseek((FILE*)io, offset, origin); } long int io_tell(IOHANDLE io) { return ftell((FILE*)io); } long int io_length(IOHANDLE io) { long int length; io_seek(io, 0, IOSEEK_END); length = io_tell(io); io_seek(io, 0, IOSEEK_START); return length; } unsigned io_write(IOHANDLE io, const void *buffer, unsigned size) { return fwrite(buffer, 1, size, (FILE*)io); } int io_close(IOHANDLE io) { fclose((FILE*)io); return 1; } int io_flush(IOHANDLE io) { fflush((FILE*)io); return 0; } void *thread_create(void (*threadfunc)(void *), void *u) { #if defined(CONF_FAMILY_UNIX) pthread_t id; pthread_create(&id, NULL, (void *(*)(void*))threadfunc, u); return (void*)id; #elif defined(CONF_FAMILY_WINDOWS) return CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)threadfunc, u, 0, NULL); #else #error not implemented #endif } void thread_wait(void *thread) { #if defined(CONF_FAMILY_UNIX) pthread_join((pthread_t)thread, NULL); #elif defined(CONF_FAMILY_WINDOWS) WaitForSingleObject((HANDLE)thread, INFINITE); #else #error not implemented #endif } void thread_destroy(void *thread) { #if defined(CONF_FAMILY_UNIX) void *r = 0; pthread_join((pthread_t)thread, &r); #else /*#error not implemented*/ #endif } void thread_yield() { #if defined(CONF_FAMILY_UNIX) sched_yield(); #elif defined(CONF_FAMILY_WINDOWS) Sleep(0); #else #error not implemented #endif } void thread_sleep(int milliseconds) { #if defined(CONF_FAMILY_UNIX) usleep(milliseconds*1000); #elif defined(CONF_FAMILY_WINDOWS) Sleep(milliseconds); #else #error not implemented #endif } #if defined(CONF_FAMILY_UNIX) typedef pthread_mutex_t LOCKINTERNAL; #elif defined(CONF_FAMILY_WINDOWS) typedef CRITICAL_SECTION LOCKINTERNAL; #else #error not implemented on this platform #endif LOCK lock_create() { LOCKINTERNAL *lock = (LOCKINTERNAL*)mem_alloc(sizeof(LOCKINTERNAL), 4); #if defined(CONF_FAMILY_UNIX) pthread_mutex_init(lock, 0x0); #elif defined(CONF_FAMILY_WINDOWS) InitializeCriticalSection((LPCRITICAL_SECTION)lock); #else #error not implemented on this platform #endif return (LOCK)lock; } void lock_destroy(LOCK lock) { #if defined(CONF_FAMILY_UNIX) pthread_mutex_destroy((LOCKINTERNAL *)lock); #elif defined(CONF_FAMILY_WINDOWS) DeleteCriticalSection((LPCRITICAL_SECTION)lock); #else #error not implemented on this platform #endif mem_free(lock); } int lock_try(LOCK lock) { #if defined(CONF_FAMILY_UNIX) return pthread_mutex_trylock((LOCKINTERNAL *)lock); #elif defined(CONF_FAMILY_WINDOWS) return TryEnterCriticalSection((LPCRITICAL_SECTION)lock); #else #error not implemented on this platform #endif } void lock_wait(LOCK lock) { #if defined(CONF_FAMILY_UNIX) pthread_mutex_lock((LOCKINTERNAL *)lock); #elif defined(CONF_FAMILY_WINDOWS) EnterCriticalSection((LPCRITICAL_SECTION)lock); #else #error not implemented on this platform #endif } void lock_release(LOCK lock) { #if defined(CONF_FAMILY_UNIX) pthread_mutex_unlock((LOCKINTERNAL *)lock); #elif defined(CONF_FAMILY_WINDOWS) LeaveCriticalSection((LPCRITICAL_SECTION)lock); #else #error not implemented on this platform #endif } /* ----- time ----- */ int64 time_get() { #if defined(CONF_FAMILY_UNIX) struct timeval val; gettimeofday(&val, NULL); return (int64)val.tv_sec*(int64)1000000+(int64)val.tv_usec; #elif defined(CONF_FAMILY_WINDOWS) static int64 last = 0; int64 t; QueryPerformanceCounter((PLARGE_INTEGER)&t); if(tsin_family = AF_INET; p->sin_port = htons(src->port); p->sin_addr.s_addr = htonl(src->ip[0]<<24|src->ip[1]<<16|src->ip[2]<<8|src->ip[3]); } static void sockaddr_to_netaddr(const struct sockaddr *src, NETADDR *dst) { /* TODO: IPv6 support */ unsigned int ip = htonl(((struct sockaddr_in*)src)->sin_addr.s_addr); mem_zero(dst, sizeof(NETADDR)); dst->type = NETTYPE_IPV4; dst->port = htons(((struct sockaddr_in*)src)->sin_port); dst->ip[0] = (unsigned char)((ip>>24)&0xFF); dst->ip[1] = (unsigned char)((ip>>16)&0xFF); dst->ip[2] = (unsigned char)((ip>>8)&0xFF); dst->ip[3] = (unsigned char)(ip&0xFF); } int net_addr_comp(const NETADDR *a, const NETADDR *b) { return mem_comp(a, b, sizeof(NETADDR)); } void net_addr_str(const NETADDR *addr, char *string, int max_length) { if(addr->type == NETTYPE_IPV4) str_format(string, max_length, "%d.%d.%d.%d:%d", addr->ip[0], addr->ip[1], addr->ip[2], addr->ip[3], addr->port); else if(addr->type == NETTYPE_IPV6) { str_format(string, max_length, "[%x:%x:%x:%x:%x:%x:%x:%x]:%d", (addr->ip[0]<<8)|addr->ip[1], (addr->ip[2]<<8)|addr->ip[3], (addr->ip[4]<<8)|addr->ip[5], (addr->ip[6]<<8)|addr->ip[7], (addr->ip[8]<<8)|addr->ip[9], (addr->ip[10]<<8)|addr->ip[11], (addr->ip[12]<<8)|addr->ip[13], (addr->ip[14]<<8)|addr->ip[15], addr->port); } else str_format(string, max_length, "unknown type %d", addr->type); } int net_host_lookup(const char *hostname, NETADDR *addr, int types) { /* TODO: IPv6 support */ struct addrinfo hints; struct addrinfo *result; int e; mem_zero(&hints, sizeof(hints)); hints.ai_family = AF_INET; e = getaddrinfo(hostname, NULL, &hints, &result); if(e != 0 || !result) return -1; sockaddr_to_netaddr(result->ai_addr, addr); freeaddrinfo(result); addr->port = 0; return 0; } static int parse_int(int *out, const char **str) { int i = 0; *out = 0; if(**str < '0' || **str > '9') return -1; i = **str - '0'; (*str)++; while(1) { if(**str < '0' || **str > '9') { *out = i; return 0; } i = (i*10) + (**str - '0'); (*str)++; } return 0; } static int parse_char(char c, const char **str) { if(**str != c) return -1; (*str)++; return 0; } static int parse_uint8(unsigned char *out, const char **str) { int i; if(parse_int(&i, str) != 0) return -1; if(i < 0 || i > 0xff) return -1; *out = i; return 0; } static int parse_uint16(unsigned short *out, const char **str) { int i; if(parse_int(&i, str) != 0) return -1; if(i < 0 || i > 0xffff) return -1; *out = i; return 0; } int net_addr_from_str(NETADDR *addr, const char *string) { const char *str = string; mem_zero(addr, sizeof(NETADDR)); if(str[0] == '[') { /* TODO: ipv6 */ } else { /* ipv4 */ if(parse_uint8(&addr->ip[0], &str)) return -1; if(parse_char('.', &str)) return -1; if(parse_uint8(&addr->ip[1], &str)) return -1; if(parse_char('.', &str)) return -1; if(parse_uint8(&addr->ip[2], &str)) return -1; if(parse_char('.', &str)) return -1; if(parse_uint8(&addr->ip[3], &str)) return -1; if(*str == ':') { str++; if(parse_uint16(&addr->port, &str)) return -1; } addr->type = NETTYPE_IPV4; } return 0; } NETSOCKET net_udp_create(NETADDR bindaddr) { /* TODO: IPv6 support */ struct sockaddr addr; unsigned int mode = 1; int broadcast = 1; /* create socket */ int sock = socket(AF_INET, SOCK_DGRAM, 0); if(sock < 0) return NETSOCKET_INVALID; /* bind, we should check for error */ netaddr_to_sockaddr(&bindaddr, &addr); if(bind(sock, &addr, sizeof(addr)) != 0) { net_udp_close(sock); return NETSOCKET_INVALID; } /* set non-blocking */ #if defined(CONF_FAMILY_WINDOWS) ioctlsocket(sock, FIONBIO, (unsigned long *)&mode); #else ioctl(sock, FIONBIO, (unsigned long *)&mode); #endif /* set boardcast */ setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (const char*)&broadcast, sizeof(broadcast)); /* return */ return sock; } int net_udp_send(NETSOCKET sock, const NETADDR *addr, const void *data, int size) { struct sockaddr sa; int d; mem_zero(&sa, sizeof(sa)); netaddr_to_sockaddr(addr, &sa); d = sendto((int)sock, (const char*)data, size, 0, &sa, sizeof(sa)); if(d < 0) { char addrstr[256]; net_addr_str(addr, addrstr, sizeof(addrstr)); dbg_msg("net", "sendto error %d %x", d, d); dbg_msg("net", "\tsock = %d %x", sock, sock); dbg_msg("net", "\tsize = %d %x", size, size); dbg_msg("net", "\taddr = %s", addrstr); } network_stats.sent_bytes += size; network_stats.sent_packets++; return d; } int net_udp_recv(NETSOCKET sock, NETADDR *addr, void *data, int maxsize) { struct sockaddr from; int bytes; socklen_t fromlen = sizeof(struct sockaddr); bytes = recvfrom(sock, (char*)data, maxsize, 0, &from, &fromlen); if(bytes > 0) { sockaddr_to_netaddr(&from, addr); network_stats.recv_bytes += bytes; network_stats.recv_packets++; return bytes; } else if(bytes == 0) return 0; return -1; /* error */ } int net_udp_close(NETSOCKET sock) { #if defined(CONF_FAMILY_WINDOWS) closesocket(sock); #else close((int)sock); #endif return 0; } NETSOCKET net_tcp_create(const NETADDR *a) { /* TODO: IPv6 support */ struct sockaddr addr; /* create socket */ int sock = socket(AF_INET, SOCK_STREAM, 0); if(sock < 0) return NETSOCKET_INVALID; /* bind, we should check for error */ netaddr_to_sockaddr(a, &addr); bind(sock, &addr, sizeof(addr)); /* return */ return sock; } int net_tcp_set_non_blocking(NETSOCKET sock) { unsigned int mode = 1; #if defined(CONF_FAMILY_WINDOWS) return ioctlsocket(sock, FIONBIO, (unsigned long *)&mode); #else return ioctl(sock, FIONBIO, (unsigned long *)&mode); #endif } int net_tcp_set_blocking(NETSOCKET sock) { unsigned int mode = 0; #if defined(CONF_FAMILY_WINDOWS) return ioctlsocket(sock, FIONBIO, (unsigned long *)&mode); #else return ioctl(sock, FIONBIO, (unsigned long *)&mode); #endif } int net_tcp_listen(NETSOCKET sock, int backlog) { return listen(sock, backlog); } int net_tcp_accept(NETSOCKET sock, NETSOCKET *new_sock, NETADDR *a) { int s; socklen_t sockaddr_len; struct sockaddr addr; sockaddr_len = sizeof(addr); s = accept(sock, &addr, &sockaddr_len); if (s != -1) { sockaddr_to_netaddr(&addr, a); *new_sock = s; } return s; } int net_tcp_connect(NETSOCKET sock, const NETADDR *a) { struct sockaddr addr; netaddr_to_sockaddr(a, &addr); return connect(sock, &addr, sizeof(addr)); } int net_tcp_connect_non_blocking(NETSOCKET sock, const NETADDR *a) { struct sockaddr addr; int res; netaddr_to_sockaddr(a, &addr); net_tcp_set_non_blocking(sock); res = connect(sock, &addr, sizeof(addr)); net_tcp_set_blocking(sock); return res; } int net_tcp_send(NETSOCKET sock, const void *data, int size) { int d; d = send((int)sock, (const char*)data, size, 0); return d; } int net_tcp_recv(NETSOCKET sock, void *data, int maxsize) { int bytes; bytes = recv((int)sock, (char*)data, maxsize, 0); return bytes; } int net_tcp_close(NETSOCKET sock) { #if defined(CONF_FAMILY_WINDOWS) closesocket(sock); #else close((int)sock); #endif return 0; } int net_errno() { return errno; } int net_would_block() { return net_errno() == EWOULDBLOCK; } int net_init() { #if defined(CONF_FAMILY_WINDOWS) WSADATA wsaData; int err = WSAStartup(MAKEWORD(1, 1), &wsaData); dbg_assert(err == 0, "network initialization failed."); return err==0?0:1; #endif return 0; } int fs_listdir(const char *dir, fs_listdir_callback cb, void *user) { #if defined(CONF_FAMILY_WINDOWS) WIN32_FIND_DATA finddata; HANDLE handle; char buffer[1024*2]; str_format(buffer, sizeof(buffer), "%s/*", dir); handle = FindFirstFileA(buffer, &finddata); if (handle == INVALID_HANDLE_VALUE) return 0; /* add all the entries */ do { if(finddata.cFileName[0] != '.') cb(finddata.cFileName, 0, user); } while (FindNextFileA(handle, &finddata)); FindClose(handle); return 0; #else struct dirent *entry; DIR *d = opendir(dir); if(!d) return 0; while((entry = readdir(d)) != NULL) cb(entry->d_name, 0, user); /* close the directory and return */ closedir(d); return 0; #endif } int fs_storage_path(const char *appname, char *path, int max) { #if defined(CONF_FAMILY_WINDOWS) HRESULT r; char *home = getenv("APPDATA"); if(!home) return -1; _snprintf(path, max, "%s/%s", home, appname); return 0; #else char *home = getenv("HOME"); int i; if(!home) return -1; #if defined(CONF_PLATFORM_MACOSX) snprintf(path, max, "%s/Library/Application Support/%s", home, appname); #else snprintf(path, max, "%s/.%s", home, appname); for(i = strlen(home)+2; path[i]; i++) path[i] = tolower(path[i]); #endif return 0; #endif } int fs_makedir(const char *path) { #if defined(CONF_FAMILY_WINDOWS) if(_mkdir(path) == 0) return 0; if(errno == EEXIST) return 0; return -1; #else if(mkdir(path, 0755) == 0) return 0; if(errno == EEXIST) return 0; return -1; #endif } void swap_endian(void *data, unsigned elem_size, unsigned num) { char *src = (char*) data; char *dst = src + (elem_size - 1); while(num) { unsigned n = elem_size>>1; char tmp; while(n) { tmp = *src; *src = *dst; *dst = tmp; src++; dst--; n--; } src = src + (elem_size>>1); dst = src + (elem_size - 1); num--; } } int net_socket_read_wait(NETSOCKET sock, int time) { struct timeval tv; fd_set readfds; tv.tv_sec = 0; tv.tv_usec = 1000*time; FD_ZERO(&readfds); FD_SET(sock, &readfds); /* don't care about writefds and exceptfds */ select(sock+1, &readfds, NULL, NULL, &tv); if(FD_ISSET(sock, &readfds)) return 1; return 0; } unsigned time_timestamp() { return time(0); } void str_append(char *dst, const char *src, int dst_size) { int s = strlen(dst); int i = 0; while(s < dst_size) { dst[s] = src[i]; if(!src[i]) /* check for null termination */ break; s++; i++; } dst[dst_size-1] = 0; /* assure null termination */ } void str_copy(char *dst, const char *src, int dst_size) { strncpy(dst, src, dst_size); dst[dst_size-1] = 0; /* assure null termination */ } void str_format(char *buffer, int buffer_size, const char *format, ...) { #if defined(CONF_FAMILY_WINDOWS) va_list ap; va_start(ap, format); _vsnprintf(buffer, buffer_size, format, ap); va_end(ap); #else va_list ap; va_start(ap, format); vsnprintf(buffer, buffer_size, format, ap); va_end(ap); #endif buffer[buffer_size-1] = 0; /* assure null termination */ } /* makes sure that the string only contains the characters between 32 and 127 */ void str_sanitize_strong(char *str_in) { unsigned char *str = (unsigned char *)str_in; while(*str) { *str &= 0x7f; if(*str < 32) *str = 32; str++; } } /* makes sure that the string only contains the characters between 32 and 255 + \r\n\t */ void str_sanitize(char *str_in) { unsigned char *str = (unsigned char *)str_in; while(*str) { if(*str < 32 && !(*str == '\r') && !(*str == '\n') && !(*str == '\t')) *str = ' '; str++; } } /* case */ int str_comp_nocase(const char *a, const char *b) { #if defined(CONF_FAMILY_WINDOWS) return _stricmp(a,b); #else return strcasecmp(a,b); #endif } const char *str_find_nocase(const char *haystack, const char *needle) { while(*haystack) /* native implementation */ { const char *a = haystack; const char *b = needle; while(*a && *b && tolower(*a) == tolower(*b)) { a++; b++; } if(!(*b)) return haystack; haystack++; } return 0; } void str_hex(char *dst, int dst_size, const void *data, int data_size) { static const char hex[] = "0123456789ABCDEF"; int b; for(b = 0; b < data_size && b < dst_size/4-4; b++) { dst[b*3] = hex[((const unsigned char *)data)[b]>>4]; dst[b*3+1] = hex[((const unsigned char *)data)[b]&0xf]; dst[b*3+2] = ' '; dst[b*3+3] = 0; } } int mem_comp(const void *a, const void *b, int size) { return memcmp(a,b,size); } void net_stats(NETSTATS *stats_inout) { *stats_inout = network_stats; } #if defined(__cplusplus) } #endif