/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */ /* If you are missing that file, acquire a complete release at teeworlds.com. */ #include #include #include #include #include #include #include #include #include #include // std::get_time #include // std::size #include // std::istringstream #include #include "lock.h" #include "logger.h" #include "system.h" #include #if defined(CONF_WEBSOCKETS) #include #endif #if defined(CONF_FAMILY_UNIX) #include #include #include #include #include #include #include /* unix net includes */ #include #include #include #include #include #include #include #include #if defined(CONF_PLATFORM_MACOS) // some lock and pthread functions are already defined in headers // included from Carbon.h // this prevents having duplicate definitions of those #define _lock_set_user_ #define _task_user_ #include #include #include #include #if defined(__MAC_10_10) && __MAC_OS_X_VERSION_MIN_REQUIRED >= __MAC_10_10 #include #endif #endif #elif defined(CONF_FAMILY_WINDOWS) #include #include #include #include #include #include #include #include #include #include #include // SHChangeNotify #include #include #else #error NOT IMPLEMENTED #endif #if defined(CONF_PLATFORM_SOLARIS) #include #endif IOHANDLE io_stdin() { return stdin; } IOHANDLE io_stdout() { return stdout; } IOHANDLE io_stderr() { return stderr; } IOHANDLE io_current_exe() { // From https://stackoverflow.com/a/1024937. #if defined(CONF_FAMILY_WINDOWS) wchar_t wide_path[IO_MAX_PATH_LENGTH]; if(GetModuleFileNameW(NULL, wide_path, std::size(wide_path)) == 0 || GetLastError() != ERROR_SUCCESS) { return 0; } const std::optional path = windows_wide_to_utf8(wide_path); return path.has_value() ? io_open(path.value().c_str(), IOFLAG_READ) : 0; #elif defined(CONF_PLATFORM_MACOS) char path[IO_MAX_PATH_LENGTH]; uint32_t path_size = sizeof(path); if(_NSGetExecutablePath(path, &path_size)) { return 0; } return io_open(path, IOFLAG_READ); #else static const char *NAMES[] = { "/proc/self/exe", // Linux, Android "/proc/curproc/exe", // NetBSD "/proc/curproc/file", // DragonFly }; for(auto &name : NAMES) { IOHANDLE result = io_open(name, IOFLAG_READ); if(result) { return result; } } return 0; #endif } static NETSTATS network_stats = {0}; #define VLEN 128 #define PACKETSIZE 1400 typedef struct { #ifdef CONF_PLATFORM_LINUX int pos; int size; struct mmsghdr msgs[VLEN]; struct iovec iovecs[VLEN]; char bufs[VLEN][PACKETSIZE]; char sockaddrs[VLEN][128]; #else char buf[PACKETSIZE]; #endif } NETSOCKET_BUFFER; void net_buffer_init(NETSOCKET_BUFFER *buffer); void net_buffer_reinit(NETSOCKET_BUFFER *buffer); void net_buffer_simple(NETSOCKET_BUFFER *buffer, char **buf, int *size); struct NETSOCKET_INTERNAL { int type; int ipv4sock; int ipv6sock; int web_ipv4sock; NETSOCKET_BUFFER buffer; }; static NETSOCKET_INTERNAL invalid_socket = {NETTYPE_INVALID, -1, -1, -1}; #define AF_WEBSOCKET_INET (0xee) std::atomic_bool dbg_assert_failing = false; DBG_ASSERT_HANDLER dbg_assert_handler; bool dbg_assert_has_failed() { return dbg_assert_failing.load(std::memory_order_acquire); } void dbg_assert_imp(const char *filename, int line, bool test, const char *msg) { if(!test) { const bool already_failing = dbg_assert_has_failed(); dbg_assert_failing.store(true, std::memory_order_release); char error[512]; str_format(error, sizeof(error), "%s(%d): %s", filename, line, msg); dbg_msg("assert", "%s", error); if(!already_failing) { DBG_ASSERT_HANDLER handler = dbg_assert_handler; if(handler) handler(error); } log_global_logger_finish(); dbg_break(); } } void dbg_break() { #ifdef __GNUC__ __builtin_trap(); #else abort(); #endif } void dbg_assert_set_handler(DBG_ASSERT_HANDLER handler) { dbg_assert_handler = std::move(handler); } void dbg_msg(const char *sys, const char *fmt, ...) { va_list args; va_start(args, fmt); log_log_v(LEVEL_INFO, sys, fmt, args); va_end(args); } /* */ void mem_copy(void *dest, const void *source, size_t size) { memcpy(dest, source, size); } void mem_move(void *dest, const void *source, size_t size) { memmove(dest, source, size); } int mem_comp(const void *a, const void *b, size_t size) { return memcmp(a, b, size); } bool mem_has_null(const void *block, size_t size) { const unsigned char *bytes = (const unsigned char *)block; for(size_t i = 0; i < size; i++) { if(bytes[i] == 0) { return true; } } return false; } IOHANDLE io_open_impl(const char *filename, int flags) { dbg_assert(flags == (IOFLAG_READ | IOFLAG_SKIP_BOM) || flags == IOFLAG_READ || flags == IOFLAG_WRITE || flags == IOFLAG_APPEND, "flags must be read, read+skipbom, write or append"); #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_filename = windows_utf8_to_wide(filename); DWORD desired_access; DWORD creation_disposition; const char *open_mode; if((flags & IOFLAG_READ) != 0) { desired_access = FILE_READ_DATA; creation_disposition = OPEN_EXISTING; open_mode = "rb"; } else if(flags == IOFLAG_WRITE) { desired_access = FILE_WRITE_DATA; creation_disposition = CREATE_ALWAYS; open_mode = "wb"; } else if(flags == IOFLAG_APPEND) { desired_access = FILE_APPEND_DATA; creation_disposition = OPEN_ALWAYS; open_mode = "ab"; } else { dbg_assert(false, "logic error"); return nullptr; } HANDLE handle = CreateFileW(wide_filename.c_str(), desired_access, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, nullptr, creation_disposition, FILE_ATTRIBUTE_NORMAL, nullptr); if(handle == INVALID_HANDLE_VALUE) return nullptr; const int file_descriptor = _open_osfhandle((intptr_t)handle, 0); dbg_assert(file_descriptor != -1, "_open_osfhandle failure"); FILE *file_stream = _fdopen(file_descriptor, open_mode); dbg_assert(file_stream != nullptr, "_fdopen failure"); return file_stream; #else const char *open_mode; if((flags & IOFLAG_READ) != 0) { open_mode = "rb"; } else if(flags == IOFLAG_WRITE) { open_mode = "wb"; } else if(flags == IOFLAG_APPEND) { open_mode = "ab"; } else { dbg_assert(false, "logic error"); return nullptr; } return fopen(filename, open_mode); #endif } IOHANDLE io_open(const char *filename, int flags) { IOHANDLE result = io_open_impl(filename, flags); unsigned char buf[3]; if((flags & IOFLAG_SKIP_BOM) == 0 || !result) { return result; } if(io_read(result, buf, sizeof(buf)) != 3 || buf[0] != 0xef || buf[1] != 0xbb || buf[2] != 0xbf) { io_seek(result, 0, IOSEEK_START); } return result; } unsigned io_read(IOHANDLE io, void *buffer, unsigned size) { return fread(buffer, 1, size, (FILE *)io); } void io_read_all(IOHANDLE io, void **result, unsigned *result_len) { long signed_len = io_length(io); unsigned len = signed_len < 0 ? 1024 : (unsigned)signed_len; // use default initial size if we couldn't get the length char *buffer = (char *)malloc(len + 1); unsigned read = io_read(io, buffer, len + 1); // +1 to check if the file size is larger than expected if(read < len) { buffer = (char *)realloc(buffer, read + 1); len = read; } else if(read > len) { unsigned cap = 2 * read; len = read; buffer = (char *)realloc(buffer, cap); while((read = io_read(io, buffer + len, cap - len)) != 0) { len += read; if(len == cap) { cap *= 2; buffer = (char *)realloc(buffer, cap); } } buffer = (char *)realloc(buffer, len + 1); } buffer[len] = 0; *result = buffer; *result_len = len; } char *io_read_all_str(IOHANDLE io) { void *buffer; unsigned len; io_read_all(io, &buffer, &len); if(mem_has_null(buffer, len)) { free(buffer); return nullptr; } return (char *)buffer; } int io_skip(IOHANDLE io, int size) { return io_seek(io, size, IOSEEK_CUR); } 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; break; default: dbg_assert(false, "origin invalid"); return -1; } return fseek((FILE *)io, offset, real_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; } int io_error(IOHANDLE io) { return ferror((FILE *)io); } unsigned io_write(IOHANDLE io, const void *buffer, unsigned size) { return fwrite(buffer, 1, size, (FILE *)io); } bool io_write_newline(IOHANDLE io) { #if defined(CONF_FAMILY_WINDOWS) return io_write(io, "\r\n", 2) == 2; #else return io_write(io, "\n", 1) == 1; #endif } int io_close(IOHANDLE io) { return fclose((FILE *)io) != 0; } int io_flush(IOHANDLE io) { return fflush((FILE *)io); } int io_sync(IOHANDLE io) { if(io_flush(io)) { return 1; } #if defined(CONF_FAMILY_WINDOWS) return FlushFileBuffers((HANDLE)_get_osfhandle(_fileno((FILE *)io))) == FALSE; #else return fsync(fileno((FILE *)io)) != 0; #endif } #define ASYNC_BUFSIZE (8 * 1024) #define ASYNC_LOCAL_BUFSIZE (64 * 1024) struct ASYNCIO { CLock lock; IOHANDLE io; SEMAPHORE sphore; void *thread; unsigned char *buffer; unsigned int buffer_size; unsigned int read_pos; unsigned int write_pos; int error; unsigned char finish; unsigned char refcount; }; enum { ASYNCIO_RUNNING, ASYNCIO_CLOSE, ASYNCIO_EXIT, }; struct BUFFERS { unsigned char *buf1; unsigned int len1; unsigned char *buf2; unsigned int len2; }; static void buffer_ptrs(ASYNCIO *aio, struct BUFFERS *buffers) { mem_zero(buffers, sizeof(*buffers)); if(aio->read_pos < aio->write_pos) { buffers->buf1 = aio->buffer + aio->read_pos; buffers->len1 = aio->write_pos - aio->read_pos; } else if(aio->read_pos > aio->write_pos) { buffers->buf1 = aio->buffer + aio->read_pos; buffers->len1 = aio->buffer_size - aio->read_pos; buffers->buf2 = aio->buffer; buffers->len2 = aio->write_pos; } } static void aio_handle_free_and_unlock(ASYNCIO *aio) RELEASE(aio->lock) { int do_free; aio->refcount--; do_free = aio->refcount == 0; aio->lock.unlock(); if(do_free) { free(aio->buffer); sphore_destroy(&aio->sphore); delete aio; } } static void aio_thread(void *user) { ASYNCIO *aio = (ASYNCIO *)user; aio->lock.lock(); while(true) { struct BUFFERS buffers; int result_io_error; unsigned char local_buffer[ASYNC_LOCAL_BUFSIZE]; unsigned int local_buffer_len = 0; if(aio->read_pos == aio->write_pos) { if(aio->finish != ASYNCIO_RUNNING) { if(aio->finish == ASYNCIO_CLOSE) { io_close(aio->io); } aio_handle_free_and_unlock(aio); break; } aio->lock.unlock(); sphore_wait(&aio->sphore); aio->lock.lock(); continue; } buffer_ptrs(aio, &buffers); if(buffers.buf1) { if(buffers.len1 > sizeof(local_buffer) - local_buffer_len) { buffers.len1 = sizeof(local_buffer) - local_buffer_len; } mem_copy(local_buffer + local_buffer_len, buffers.buf1, buffers.len1); local_buffer_len += buffers.len1; if(buffers.buf2) { if(buffers.len2 > sizeof(local_buffer) - local_buffer_len) { buffers.len2 = sizeof(local_buffer) - local_buffer_len; } mem_copy(local_buffer + local_buffer_len, buffers.buf2, buffers.len2); local_buffer_len += buffers.len2; } } aio->read_pos = (aio->read_pos + buffers.len1 + buffers.len2) % aio->buffer_size; aio->lock.unlock(); io_write(aio->io, local_buffer, local_buffer_len); io_flush(aio->io); result_io_error = io_error(aio->io); aio->lock.lock(); aio->error = result_io_error; } } ASYNCIO *aio_new(IOHANDLE io) { ASYNCIO *aio = new ASYNCIO; if(!aio) { return 0; } aio->io = io; sphore_init(&aio->sphore); aio->thread = 0; aio->buffer = (unsigned char *)malloc(ASYNC_BUFSIZE); if(!aio->buffer) { sphore_destroy(&aio->sphore); delete aio; return 0; } aio->buffer_size = ASYNC_BUFSIZE; aio->read_pos = 0; aio->write_pos = 0; aio->error = 0; aio->finish = ASYNCIO_RUNNING; aio->refcount = 2; aio->thread = thread_init(aio_thread, aio, "aio"); if(!aio->thread) { free(aio->buffer); sphore_destroy(&aio->sphore); delete aio; return 0; } return aio; } static unsigned int buffer_len(ASYNCIO *aio) { if(aio->write_pos >= aio->read_pos) { return aio->write_pos - aio->read_pos; } else { return aio->buffer_size + aio->write_pos - aio->read_pos; } } static unsigned int next_buffer_size(unsigned int cur_size, unsigned int need_size) { while(cur_size < need_size) { cur_size *= 2; } return cur_size; } void aio_lock(ASYNCIO *aio) ACQUIRE(aio->lock) { aio->lock.lock(); } void aio_unlock(ASYNCIO *aio) RELEASE(aio->lock) { aio->lock.unlock(); sphore_signal(&aio->sphore); } void aio_write_unlocked(ASYNCIO *aio, const void *buffer, unsigned size) { unsigned int remaining; remaining = aio->buffer_size - buffer_len(aio); // Don't allow full queue to distinguish between empty and full queue. if(size < remaining) { unsigned int remaining_contiguous = aio->buffer_size - aio->write_pos; if(size > remaining_contiguous) { mem_copy(aio->buffer + aio->write_pos, buffer, remaining_contiguous); size -= remaining_contiguous; buffer = ((unsigned char *)buffer) + remaining_contiguous; aio->write_pos = 0; } mem_copy(aio->buffer + aio->write_pos, buffer, size); aio->write_pos = (aio->write_pos + size) % aio->buffer_size; } else { // Add 1 so the new buffer isn't completely filled. unsigned int new_written = buffer_len(aio) + size + 1; unsigned int next_size = next_buffer_size(aio->buffer_size, new_written); unsigned int next_len = 0; unsigned char *next_buffer = (unsigned char *)malloc(next_size); struct BUFFERS buffers; buffer_ptrs(aio, &buffers); if(buffers.buf1) { mem_copy(next_buffer + next_len, buffers.buf1, buffers.len1); next_len += buffers.len1; if(buffers.buf2) { mem_copy(next_buffer + next_len, buffers.buf2, buffers.len2); next_len += buffers.len2; } } mem_copy(next_buffer + next_len, buffer, size); next_len += size; free(aio->buffer); aio->buffer = next_buffer; aio->buffer_size = next_size; aio->read_pos = 0; aio->write_pos = next_len; } } void aio_write(ASYNCIO *aio, const void *buffer, unsigned size) { aio_lock(aio); aio_write_unlocked(aio, buffer, size); aio_unlock(aio); } void aio_write_newline_unlocked(ASYNCIO *aio) { #if defined(CONF_FAMILY_WINDOWS) aio_write_unlocked(aio, "\r\n", 2); #else aio_write_unlocked(aio, "\n", 1); #endif } void aio_write_newline(ASYNCIO *aio) { aio_lock(aio); aio_write_newline_unlocked(aio); aio_unlock(aio); } int aio_error(ASYNCIO *aio) { CLockScope ls(aio->lock); return aio->error; } void aio_free(ASYNCIO *aio) { aio->lock.lock(); if(aio->thread) { thread_detach(aio->thread); aio->thread = 0; } aio_handle_free_and_unlock(aio); } void aio_close(ASYNCIO *aio) { { CLockScope ls(aio->lock); aio->finish = ASYNCIO_CLOSE; } sphore_signal(&aio->sphore); } void aio_wait(ASYNCIO *aio) { void *thread; { CLockScope ls(aio->lock); thread = aio->thread; aio->thread = 0; if(aio->finish == ASYNCIO_RUNNING) { aio->finish = ASYNCIO_EXIT; } } sphore_signal(&aio->sphore); thread_wait(thread); } struct THREAD_RUN { void (*threadfunc)(void *); void *u; }; #if defined(CONF_FAMILY_UNIX) static void *thread_run(void *user) #elif defined(CONF_FAMILY_WINDOWS) static unsigned long __stdcall thread_run(void *user) #else #error not implemented #endif { #if defined(CONF_FAMILY_WINDOWS) CWindowsComLifecycle WindowsComLifecycle(false); #endif struct THREAD_RUN *data = (THREAD_RUN *)user; void (*threadfunc)(void *) = data->threadfunc; void *u = data->u; free(data); threadfunc(u); return 0; } void *thread_init(void (*threadfunc)(void *), void *u, const char *name) { struct THREAD_RUN *data = (THREAD_RUN *)malloc(sizeof(*data)); data->threadfunc = threadfunc; data->u = u; #if defined(CONF_FAMILY_UNIX) { pthread_attr_t attr; dbg_assert(pthread_attr_init(&attr) == 0, "pthread_attr_init failure"); #if defined(CONF_PLATFORM_MACOS) && defined(__MAC_10_10) && __MAC_OS_X_VERSION_MIN_REQUIRED >= __MAC_10_10 dbg_assert(pthread_attr_set_qos_class_np(&attr, QOS_CLASS_USER_INTERACTIVE, 0) == 0, "pthread_attr_set_qos_class_np failure"); #endif pthread_t id; dbg_assert(pthread_create(&id, &attr, thread_run, data) == 0, "pthread_create failure"); return (void *)id; } #elif defined(CONF_FAMILY_WINDOWS) HANDLE thread = CreateThread(nullptr, 0, thread_run, data, 0, nullptr); dbg_assert(thread != nullptr, "CreateThread failure"); // TODO: Set thread name using SetThreadDescription (would require minimum Windows 10 version 1607) return thread; #else #error not implemented #endif } void thread_wait(void *thread) { #if defined(CONF_FAMILY_UNIX) dbg_assert(pthread_join((pthread_t)thread, nullptr) == 0, "pthread_join failure"); #elif defined(CONF_FAMILY_WINDOWS) dbg_assert(WaitForSingleObject((HANDLE)thread, INFINITE) == WAIT_OBJECT_0, "WaitForSingleObject failure"); dbg_assert(CloseHandle(thread), "CloseHandle failure"); #else #error not implemented #endif } void thread_yield() { #if defined(CONF_FAMILY_UNIX) dbg_assert(sched_yield() == 0, "sched_yield failure"); #elif defined(CONF_FAMILY_WINDOWS) Sleep(0); #else #error not implemented #endif } void thread_detach(void *thread) { #if defined(CONF_FAMILY_UNIX) dbg_assert(pthread_detach((pthread_t)thread) == 0, "pthread_detach failure"); #elif defined(CONF_FAMILY_WINDOWS) dbg_assert(CloseHandle(thread), "CloseHandle failure"); #else #error not implemented #endif } void thread_init_and_detach(void (*threadfunc)(void *), void *u, const char *name) { void *thread = thread_init(threadfunc, u, name); thread_detach(thread); } #if defined(CONF_FAMILY_WINDOWS) void sphore_init(SEMAPHORE *sem) { *sem = CreateSemaphoreW(nullptr, 0, std::numeric_limits::max(), nullptr); dbg_assert(*sem != nullptr, "CreateSemaphoreW failure"); } void sphore_wait(SEMAPHORE *sem) { dbg_assert(WaitForSingleObject((HANDLE)*sem, INFINITE) == WAIT_OBJECT_0, "WaitForSingleObject failure"); } void sphore_signal(SEMAPHORE *sem) { dbg_assert(ReleaseSemaphore((HANDLE)*sem, 1, nullptr), "ReleaseSemaphore failure"); } void sphore_destroy(SEMAPHORE *sem) { dbg_assert(CloseHandle((HANDLE)*sem), "CloseHandle failure"); } #elif defined(CONF_PLATFORM_MACOS) void sphore_init(SEMAPHORE *sem) { char aBuf[32]; str_format(aBuf, sizeof(aBuf), "%p", (void *)sem); *sem = sem_open(aBuf, O_CREAT | O_EXCL, S_IRWXU | S_IRWXG, 0); dbg_assert(*sem != SEM_FAILED, "sem_open failure"); } void sphore_wait(SEMAPHORE *sem) { while(true) { if(sem_wait(*sem) == 0) break; dbg_assert(errno == EINTR, "sem_wait failure"); } } void sphore_signal(SEMAPHORE *sem) { dbg_assert(sem_post(*sem) == 0, "sem_post failure"); } void sphore_destroy(SEMAPHORE *sem) { dbg_assert(sem_close(*sem) == 0, "sem_close failure"); char aBuf[32]; str_format(aBuf, sizeof(aBuf), "%p", (void *)sem); dbg_assert(sem_unlink(aBuf) == 0, "sem_unlink failure"); } #elif defined(CONF_FAMILY_UNIX) void sphore_init(SEMAPHORE *sem) { dbg_assert(sem_init(sem, 0, 0) == 0, "sem_init failure"); } void sphore_wait(SEMAPHORE *sem) { while(true) { if(sem_wait(sem) == 0) break; dbg_assert(errno == EINTR, "sem_wait failure"); } } void sphore_signal(SEMAPHORE *sem) { dbg_assert(sem_post(sem) == 0, "sem_post failure"); } void sphore_destroy(SEMAPHORE *sem) { dbg_assert(sem_destroy(sem) == 0, "sem_destroy failure"); } #endif static int new_tick = -1; void set_new_tick() { new_tick = 1; } /* ----- time ----- */ static_assert(std::chrono::steady_clock::is_steady, "Compiler does not support steady clocks, it might be out of date."); static_assert(std::chrono::steady_clock::period::den / std::chrono::steady_clock::period::num >= 1000000000, "Compiler has a bad timer precision and might be out of date."); static const std::chrono::time_point tw_start_time = std::chrono::steady_clock::now(); int64_t time_get_impl() { return std::chrono::duration_cast(std::chrono::steady_clock::now() - tw_start_time).count(); } int64_t time_get() { static int64_t last = 0; if(new_tick == 0) return last; if(new_tick != -1) new_tick = 0; last = time_get_impl(); return last; } int64_t time_freq() { using namespace std::chrono_literals; return std::chrono::nanoseconds(1s).count(); } /* ----- network ----- */ const NETADDR NETADDR_ZEROED = {NETTYPE_INVALID, {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, 0}; static void netaddr_to_sockaddr_in(const NETADDR *src, struct sockaddr_in *dest) { mem_zero(dest, sizeof(struct sockaddr_in)); if(src->type != NETTYPE_IPV4 && src->type != NETTYPE_WEBSOCKET_IPV4) { dbg_msg("system", "couldn't convert NETADDR of type %d to ipv4", src->type); return; } dest->sin_family = AF_INET; dest->sin_port = htons(src->port); mem_copy(&dest->sin_addr.s_addr, src->ip, 4); } static void netaddr_to_sockaddr_in6(const NETADDR *src, struct sockaddr_in6 *dest) { mem_zero(dest, sizeof(struct sockaddr_in6)); if(src->type != NETTYPE_IPV6) { dbg_msg("system", "couldn't not convert NETADDR of type %d to ipv6", src->type); return; } dest->sin6_family = AF_INET6; dest->sin6_port = htons(src->port); mem_copy(&dest->sin6_addr.s6_addr, src->ip, 16); } static void sockaddr_to_netaddr(const struct sockaddr *src, NETADDR *dst) { if(src->sa_family == AF_INET) { mem_zero(dst, sizeof(NETADDR)); dst->type = NETTYPE_IPV4; dst->port = htons(((struct sockaddr_in *)src)->sin_port); mem_copy(dst->ip, &((struct sockaddr_in *)src)->sin_addr.s_addr, 4); } else if(src->sa_family == AF_WEBSOCKET_INET) { mem_zero(dst, sizeof(NETADDR)); dst->type = NETTYPE_WEBSOCKET_IPV4; dst->port = htons(((struct sockaddr_in *)src)->sin_port); mem_copy(dst->ip, &((struct sockaddr_in *)src)->sin_addr.s_addr, 4); } else if(src->sa_family == AF_INET6) { mem_zero(dst, sizeof(NETADDR)); dst->type = NETTYPE_IPV6; dst->port = htons(((struct sockaddr_in6 *)src)->sin6_port); mem_copy(dst->ip, &((struct sockaddr_in6 *)src)->sin6_addr.s6_addr, 16); } else { mem_zero(dst, sizeof(struct sockaddr)); dbg_msg("system", "couldn't convert sockaddr of family %d", src->sa_family); } } int net_addr_comp(const NETADDR *a, const NETADDR *b) { return mem_comp(a, b, sizeof(NETADDR)); } bool NETADDR::operator==(const NETADDR &other) const { return net_addr_comp(this, &other) == 0; } int net_addr_comp_noport(const NETADDR *a, const NETADDR *b) { NETADDR ta = *a, tb = *b; ta.port = tb.port = 0; return net_addr_comp(&ta, &tb); } void net_addr_str_v6(const unsigned short ip[8], int port, char *buffer, int buffer_size) { int longest_seq_len = 0; int longest_seq_start = -1; int w = 0; int i; { int seq_len = 0; int seq_start = -1; // Determine longest sequence of zeros. for(i = 0; i < 8 + 1; i++) { if(seq_start != -1) { if(i == 8 || ip[i] != 0) { if(longest_seq_len < seq_len) { longest_seq_len = seq_len; longest_seq_start = seq_start; } seq_len = 0; seq_start = -1; } else { seq_len += 1; } } else { if(i != 8 && ip[i] == 0) { seq_start = i; seq_len = 1; } } } } if(longest_seq_len <= 1) { longest_seq_len = 0; longest_seq_start = -1; } w += str_copy(buffer + w, "[", buffer_size - w); for(i = 0; i < 8; i++) { if(longest_seq_start <= i && i < longest_seq_start + longest_seq_len) { if(i == longest_seq_start) { w += str_copy(buffer + w, "::", buffer_size - w); } } else { const char *colon = (i == 0 || i == longest_seq_start + longest_seq_len) ? "" : ":"; w += str_format(buffer + w, buffer_size - w, "%s%x", colon, ip[i]); } } w += str_copy(buffer + w, "]", buffer_size - w); if(port >= 0) { str_format(buffer + w, buffer_size - w, ":%d", port); } } void net_addr_str(const NETADDR *addr, char *string, int max_length, int add_port) { if(addr->type == NETTYPE_IPV4 || addr->type == NETTYPE_WEBSOCKET_IPV4) { if(add_port != 0) 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 str_format(string, max_length, "%d.%d.%d.%d", addr->ip[0], addr->ip[1], addr->ip[2], addr->ip[3]); } else if(addr->type == NETTYPE_IPV6) { int port = -1; unsigned short ip[8]; int i; if(add_port) { port = addr->port; } for(i = 0; i < 8; i++) { ip[i] = (addr->ip[i * 2] << 8) | (addr->ip[i * 2 + 1]); } net_addr_str_v6(ip, port, string, max_length); } else str_format(string, max_length, "unknown type %d", addr->type); } static int priv_net_extract(const char *hostname, char *host, int max_host, int *port) { int i; *port = 0; host[0] = 0; if(hostname[0] == '[') { // ipv6 mode for(i = 1; i < max_host && hostname[i] && hostname[i] != ']'; i++) host[i - 1] = hostname[i]; host[i - 1] = 0; if(hostname[i] != ']') // malformatted return -1; i++; if(hostname[i] == ':') *port = str_toint(hostname + i + 1); } else { // generic mode (ipv4, hostname etc) for(i = 0; i < max_host - 1 && hostname[i] && hostname[i] != ':'; i++) host[i] = hostname[i]; host[i] = 0; if(hostname[i] == ':') *port = str_toint(hostname + i + 1); } return 0; } int net_host_lookup_impl(const char *hostname, NETADDR *addr, int types) { struct addrinfo hints; struct addrinfo *result = NULL; int e; char host[256]; int port = 0; if(priv_net_extract(hostname, host, sizeof(host), &port)) return -1; dbg_msg("host_lookup", "host='%s' port=%d %d", host, port, types); mem_zero(&hints, sizeof(hints)); hints.ai_family = AF_UNSPEC; if(types == NETTYPE_IPV4) hints.ai_family = AF_INET; else if(types == NETTYPE_IPV6) hints.ai_family = AF_INET6; e = getaddrinfo(host, NULL, &hints, &result); if(!result) return -1; if(e != 0) { freeaddrinfo(result); return -1; } sockaddr_to_netaddr(result->ai_addr, addr); addr->port = port; freeaddrinfo(result); return 0; } int net_host_lookup(const char *hostname, NETADDR *addr, int types) { const char *ws_hostname = str_startswith(hostname, "ws://"); if(ws_hostname) { if((types & NETTYPE_WEBSOCKET_IPV4) == 0) { return -1; } int result = net_host_lookup_impl(ws_hostname, addr, NETTYPE_IPV4); if(result == 0 && addr->type == NETTYPE_IPV4) { addr->type = NETTYPE_WEBSOCKET_IPV4; } return result; } return net_host_lookup_impl(hostname, addr, types & ~NETTYPE_WEBSOCKET_IPV4); } static int parse_int(int *out, const char **str) { int i = 0; *out = 0; if(!str_isnum(**str)) return -1; i = **str - '0'; (*str)++; while(true) { if(!str_isnum(**str)) { *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_url(NETADDR *addr, const char *string, char *host_buf, size_t host_buf_size) { const char *str = str_startswith(string, "tw-0.6+udp://"); if(!str) return 1; mem_zero(addr, sizeof(*addr)); int length = str_length(str); int start = 0; int end = length; for(int i = 0; i < length; i++) { if(str[i] == '@') { if(start != 0) { // Two at signs. return true; } start = i + 1; } else if(str[i] == '/' || str[i] == '?' || str[i] == '#') { end = i; break; } } char host[128]; str_truncate(host, sizeof(host), str + start, end - start); if(host_buf) str_copy(host_buf, host, host_buf_size); return net_addr_from_str(addr, host); } int net_addr_from_str(NETADDR *addr, const char *string) { const char *str = string; mem_zero(addr, sizeof(NETADDR)); if(str[0] == '[') { /* ipv6 */ struct sockaddr_in6 sa6; char buf[128]; int i; str++; for(i = 0; i < 127 && str[i] && str[i] != ']'; i++) buf[i] = str[i]; buf[i] = 0; str += i; #if defined(CONF_FAMILY_WINDOWS) { int size; sa6.sin6_family = AF_INET6; size = (int)sizeof(sa6); if(WSAStringToAddressA(buf, AF_INET6, NULL, (struct sockaddr *)&sa6, &size) != 0) return -1; } #else sa6.sin6_family = AF_INET6; if(inet_pton(AF_INET6, buf, &sa6.sin6_addr) != 1) return -1; #endif sockaddr_to_netaddr((struct sockaddr *)&sa6, addr); if(*str == ']') { str++; if(*str == ':') { str++; if(parse_uint16(&addr->port, &str)) return -1; } else { addr->port = 0; } } else return -1; return 0; } 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; } if(*str != '\0') return -1; addr->type = NETTYPE_IPV4; } return 0; } static void priv_net_close_socket(int sock) { #if defined(CONF_FAMILY_WINDOWS) closesocket(sock); #else if(close(sock) != 0) dbg_msg("socket", "close failed: %d", errno); #endif } static int priv_net_close_all_sockets(NETSOCKET sock) { /* close down ipv4 */ if(sock->ipv4sock >= 0) { priv_net_close_socket(sock->ipv4sock); sock->ipv4sock = -1; sock->type &= ~NETTYPE_IPV4; } #if defined(CONF_WEBSOCKETS) /* close down websocket_ipv4 */ if(sock->web_ipv4sock >= 0) { websocket_destroy(sock->web_ipv4sock); sock->web_ipv4sock = -1; sock->type &= ~NETTYPE_WEBSOCKET_IPV4; } #endif /* close down ipv6 */ if(sock->ipv6sock >= 0) { priv_net_close_socket(sock->ipv6sock); sock->ipv6sock = -1; sock->type &= ~NETTYPE_IPV6; } free(sock); return 0; } #if defined(CONF_FAMILY_WINDOWS) std::string windows_format_system_message(unsigned long error) { WCHAR *wide_message; const DWORD flags = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_MAX_WIDTH_MASK; if(FormatMessageW(flags, NULL, error, 0, (LPWSTR)&wide_message, 0, NULL) == 0) return "unknown error"; std::optional message = windows_wide_to_utf8(wide_message); LocalFree(wide_message); return message.value_or("(invalid UTF-16 in error message)"); } #endif static int priv_net_create_socket(int domain, int type, struct sockaddr *addr, int sockaddrlen) { int sock, e; /* create socket */ sock = socket(domain, type, 0); if(sock < 0) { #if defined(CONF_FAMILY_WINDOWS) int error = WSAGetLastError(); const std::string message = windows_format_system_message(error); dbg_msg("net", "failed to create socket with domain %d and type %d (%d '%s')", domain, type, error, message.c_str()); #else dbg_msg("net", "failed to create socket with domain %d and type %d (%d '%s')", domain, type, errno, strerror(errno)); #endif return -1; } #if defined(CONF_FAMILY_UNIX) /* on tcp sockets set SO_REUSEADDR to fix port rebind on restart */ if(domain == AF_INET && type == SOCK_STREAM) { int option = 1; if(setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(option)) != 0) dbg_msg("socket", "Setting SO_REUSEADDR failed: %d", errno); } #endif /* set to IPv6 only if that's what we are creating */ #if defined(IPV6_V6ONLY) /* windows sdk 6.1 and higher */ if(domain == AF_INET6) { int ipv6only = 1; if(setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, (const char *)&ipv6only, sizeof(ipv6only)) != 0) dbg_msg("socket", "Setting V6ONLY failed: %d", errno); } #endif /* bind the socket */ e = bind(sock, addr, sockaddrlen); if(e != 0) { #if defined(CONF_FAMILY_WINDOWS) int error = WSAGetLastError(); const std::string message = windows_format_system_message(error); dbg_msg("net", "failed to bind socket with domain %d and type %d (%d '%s')", domain, type, error, message.c_str()); #else dbg_msg("net", "failed to bind socket with domain %d and type %d (%d '%s')", domain, type, errno, strerror(errno)); #endif priv_net_close_socket(sock); return -1; } /* return the newly created socket */ return sock; } int net_socket_type(NETSOCKET sock) { return sock->type; } NETSOCKET net_udp_create(NETADDR bindaddr) { NETSOCKET sock = (NETSOCKET_INTERNAL *)malloc(sizeof(*sock)); *sock = invalid_socket; NETADDR tmpbindaddr = bindaddr; int broadcast = 1; int socket = -1; if(bindaddr.type & NETTYPE_IPV4) { struct sockaddr_in addr; /* bind, we should check for error */ tmpbindaddr.type = NETTYPE_IPV4; netaddr_to_sockaddr_in(&tmpbindaddr, &addr); socket = priv_net_create_socket(AF_INET, SOCK_DGRAM, (struct sockaddr *)&addr, sizeof(addr)); if(socket >= 0) { sock->type |= NETTYPE_IPV4; sock->ipv4sock = socket; /* set broadcast */ if(setsockopt(socket, SOL_SOCKET, SO_BROADCAST, (const char *)&broadcast, sizeof(broadcast)) != 0) { dbg_msg("socket", "Setting BROADCAST on ipv4 failed: %d", net_errno()); } { /* set DSCP/TOS */ int iptos = 0x10 /* IPTOS_LOWDELAY */; if(setsockopt(socket, IPPROTO_IP, IP_TOS, (char *)&iptos, sizeof(iptos)) != 0) { dbg_msg("socket", "Setting TOS on ipv4 failed: %d", net_errno()); } } } } #if defined(CONF_WEBSOCKETS) if(bindaddr.type & NETTYPE_WEBSOCKET_IPV4) { char addr_str[NETADDR_MAXSTRSIZE]; /* bind, we should check for error */ tmpbindaddr.type = NETTYPE_WEBSOCKET_IPV4; net_addr_str(&tmpbindaddr, addr_str, sizeof(addr_str), 0); socket = websocket_create(addr_str, tmpbindaddr.port); if(socket >= 0) { sock->type |= NETTYPE_WEBSOCKET_IPV4; sock->web_ipv4sock = socket; } } #endif if(bindaddr.type & NETTYPE_IPV6) { struct sockaddr_in6 addr; /* bind, we should check for error */ tmpbindaddr.type = NETTYPE_IPV6; netaddr_to_sockaddr_in6(&tmpbindaddr, &addr); socket = priv_net_create_socket(AF_INET6, SOCK_DGRAM, (struct sockaddr *)&addr, sizeof(addr)); if(socket >= 0) { sock->type |= NETTYPE_IPV6; sock->ipv6sock = socket; /* set broadcast */ if(setsockopt(socket, SOL_SOCKET, SO_BROADCAST, (const char *)&broadcast, sizeof(broadcast)) != 0) { dbg_msg("socket", "Setting BROADCAST on ipv6 failed: %d", net_errno()); } // TODO: setting IP_TOS on ipv6 with setsockopt is not supported on Windows, see https://github.com/ddnet/ddnet/issues/7605 #if !defined(CONF_FAMILY_WINDOWS) { /* set DSCP/TOS */ int iptos = 0x10 /* IPTOS_LOWDELAY */; if(setsockopt(socket, IPPROTO_IP, IP_TOS, (char *)&iptos, sizeof(iptos)) != 0) { dbg_msg("socket", "Setting TOS on ipv6 failed: %d", net_errno()); } } #endif } } if(socket < 0) { free(sock); sock = nullptr; } else { /* set non-blocking */ net_set_non_blocking(sock); net_buffer_init(&sock->buffer); } /* return */ return sock; } int net_udp_send(NETSOCKET sock, const NETADDR *addr, const void *data, int size) { int d = -1; if(addr->type & NETTYPE_IPV4) { if(sock->ipv4sock >= 0) { struct sockaddr_in sa; if(addr->type & NETTYPE_LINK_BROADCAST) { mem_zero(&sa, sizeof(sa)); sa.sin_port = htons(addr->port); sa.sin_family = AF_INET; sa.sin_addr.s_addr = INADDR_BROADCAST; } else netaddr_to_sockaddr_in(addr, &sa); d = sendto((int)sock->ipv4sock, (const char *)data, size, 0, (struct sockaddr *)&sa, sizeof(sa)); } else dbg_msg("net", "can't send ipv4 traffic to this socket"); } #if defined(CONF_WEBSOCKETS) if(addr->type & NETTYPE_WEBSOCKET_IPV4) { if(sock->web_ipv4sock >= 0) { char addr_str[NETADDR_MAXSTRSIZE]; str_format(addr_str, sizeof(addr_str), "%d.%d.%d.%d", addr->ip[0], addr->ip[1], addr->ip[2], addr->ip[3]); d = websocket_send(sock->web_ipv4sock, (const unsigned char *)data, size, addr_str, addr->port); } else dbg_msg("net", "can't send websocket_ipv4 traffic to this socket"); } #endif if(addr->type & NETTYPE_IPV6) { if(sock->ipv6sock >= 0) { struct sockaddr_in6 sa; if(addr->type & NETTYPE_LINK_BROADCAST) { mem_zero(&sa, sizeof(sa)); sa.sin6_port = htons(addr->port); sa.sin6_family = AF_INET6; sa.sin6_addr.s6_addr[0] = 0xff; /* multicast */ sa.sin6_addr.s6_addr[1] = 0x02; /* link local scope */ sa.sin6_addr.s6_addr[15] = 1; /* all nodes */ } else netaddr_to_sockaddr_in6(addr, &sa); d = sendto((int)sock->ipv6sock, (const char *)data, size, 0, (struct sockaddr *)&sa, sizeof(sa)); } else dbg_msg("net", "can't send ipv6 traffic to this socket"); } /* else dbg_msg("net", "can't send to network of type %d", addr->type); */ /*if(d < 0) { char addrstr[256]; net_addr_str(addr, addrstr, sizeof(addrstr)); dbg_msg("net", "sendto error (%d '%s')", errno, strerror(errno)); 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; } void net_buffer_init(NETSOCKET_BUFFER *buffer) { #if defined(CONF_PLATFORM_LINUX) int i; buffer->pos = 0; buffer->size = 0; mem_zero(buffer->msgs, sizeof(buffer->msgs)); mem_zero(buffer->iovecs, sizeof(buffer->iovecs)); mem_zero(buffer->sockaddrs, sizeof(buffer->sockaddrs)); for(i = 0; i < VLEN; ++i) { buffer->iovecs[i].iov_base = buffer->bufs[i]; buffer->iovecs[i].iov_len = PACKETSIZE; buffer->msgs[i].msg_hdr.msg_iov = &(buffer->iovecs[i]); buffer->msgs[i].msg_hdr.msg_iovlen = 1; buffer->msgs[i].msg_hdr.msg_name = &(buffer->sockaddrs[i]); buffer->msgs[i].msg_hdr.msg_namelen = sizeof(buffer->sockaddrs[i]); } #endif } void net_buffer_reinit(NETSOCKET_BUFFER *buffer) { #if defined(CONF_PLATFORM_LINUX) for(int i = 0; i < VLEN; i++) { buffer->msgs[i].msg_hdr.msg_namelen = sizeof(buffer->sockaddrs[i]); } #endif } void net_buffer_simple(NETSOCKET_BUFFER *buffer, char **buf, int *size) { #if defined(CONF_PLATFORM_LINUX) *buf = buffer->bufs[0]; *size = sizeof(buffer->bufs[0]); #else *buf = buffer->buf; *size = sizeof(buffer->buf); #endif } int net_udp_recv(NETSOCKET sock, NETADDR *addr, unsigned char **data) { char sockaddrbuf[128]; int bytes = 0; #if defined(CONF_PLATFORM_LINUX) if(sock->ipv4sock >= 0) { if(sock->buffer.pos >= sock->buffer.size) { net_buffer_reinit(&sock->buffer); sock->buffer.size = recvmmsg(sock->ipv4sock, sock->buffer.msgs, VLEN, 0, NULL); sock->buffer.pos = 0; } } if(sock->ipv6sock >= 0) { if(sock->buffer.pos >= sock->buffer.size) { net_buffer_reinit(&sock->buffer); sock->buffer.size = recvmmsg(sock->ipv6sock, sock->buffer.msgs, VLEN, 0, NULL); sock->buffer.pos = 0; } } if(sock->buffer.pos < sock->buffer.size) { sockaddr_to_netaddr((struct sockaddr *)&(sock->buffer.sockaddrs[sock->buffer.pos]), addr); bytes = sock->buffer.msgs[sock->buffer.pos].msg_len; *data = (unsigned char *)sock->buffer.bufs[sock->buffer.pos]; sock->buffer.pos++; network_stats.recv_bytes += bytes; network_stats.recv_packets++; return bytes; } #else if(sock->ipv4sock >= 0) { socklen_t fromlen = sizeof(struct sockaddr_in); bytes = recvfrom(sock->ipv4sock, sock->buffer.buf, sizeof(sock->buffer.buf), 0, (struct sockaddr *)&sockaddrbuf, &fromlen); *data = (unsigned char *)sock->buffer.buf; } if(bytes <= 0 && sock->ipv6sock >= 0) { socklen_t fromlen = sizeof(struct sockaddr_in6); bytes = recvfrom(sock->ipv6sock, sock->buffer.buf, sizeof(sock->buffer.buf), 0, (struct sockaddr *)&sockaddrbuf, &fromlen); *data = (unsigned char *)sock->buffer.buf; } #endif #if defined(CONF_WEBSOCKETS) if(bytes <= 0 && sock->web_ipv4sock >= 0) { char *buf; int size; net_buffer_simple(&sock->buffer, &buf, &size); socklen_t fromlen = sizeof(struct sockaddr); struct sockaddr_in *sockaddrbuf_in = (struct sockaddr_in *)&sockaddrbuf; bytes = websocket_recv(sock->web_ipv4sock, (unsigned char *)buf, size, sockaddrbuf_in, fromlen); *data = (unsigned char *)buf; sockaddrbuf_in->sin_family = AF_WEBSOCKET_INET; } #endif if(bytes > 0) { sockaddr_to_netaddr((struct sockaddr *)&sockaddrbuf, 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) { return priv_net_close_all_sockets(sock); } NETSOCKET net_tcp_create(NETADDR bindaddr) { NETSOCKET sock = (NETSOCKET_INTERNAL *)malloc(sizeof(*sock)); *sock = invalid_socket; NETADDR tmpbindaddr = bindaddr; int socket4 = -1; if(bindaddr.type & NETTYPE_IPV4) { struct sockaddr_in addr; /* bind, we should check for error */ tmpbindaddr.type = NETTYPE_IPV4; netaddr_to_sockaddr_in(&tmpbindaddr, &addr); socket4 = priv_net_create_socket(AF_INET, SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr)); if(socket4 >= 0) { sock->type |= NETTYPE_IPV4; sock->ipv4sock = socket4; } } int socket6 = -1; if(bindaddr.type & NETTYPE_IPV6) { struct sockaddr_in6 addr; /* bind, we should check for error */ tmpbindaddr.type = NETTYPE_IPV6; netaddr_to_sockaddr_in6(&tmpbindaddr, &addr); socket6 = priv_net_create_socket(AF_INET6, SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr)); if(socket6 >= 0) { sock->type |= NETTYPE_IPV6; sock->ipv6sock = socket6; } } if(socket4 < 0 && socket6 < 0) { free(sock); sock = nullptr; } /* return */ return sock; } static int net_set_blocking_impl(NETSOCKET sock, bool blocking) { unsigned long mode = blocking ? 0 : 1; const char *mode_str = blocking ? "blocking" : "non-blocking"; int sockets[] = {sock->ipv4sock, sock->ipv6sock}; const char *socket_str[] = {"ipv4", "ipv6"}; for(size_t i = 0; i < std::size(sockets); ++i) { if(sockets[i] >= 0) { #if defined(CONF_FAMILY_WINDOWS) int result = ioctlsocket(sockets[i], FIONBIO, (unsigned long *)&mode); if(result != NO_ERROR) dbg_msg("socket", "setting %s %s failed: %d", socket_str[i], mode_str, result); #else if(ioctl(sockets[i], FIONBIO, (unsigned long *)&mode) == -1) dbg_msg("socket", "setting %s %s failed: %d", socket_str[i], mode_str, errno); #endif } } return 0; } int net_set_non_blocking(NETSOCKET sock) { return net_set_blocking_impl(sock, false); } int net_set_blocking(NETSOCKET sock) { return net_set_blocking_impl(sock, true); } int net_tcp_listen(NETSOCKET sock, int backlog) { int err = -1; if(sock->ipv4sock >= 0) err = listen(sock->ipv4sock, backlog); if(sock->ipv6sock >= 0) err = listen(sock->ipv6sock, backlog); return err; } int net_tcp_accept(NETSOCKET sock, NETSOCKET *new_sock, NETADDR *a) { int s; socklen_t sockaddr_len; *new_sock = nullptr; if(sock->ipv4sock >= 0) { struct sockaddr_in addr; sockaddr_len = sizeof(addr); s = accept(sock->ipv4sock, (struct sockaddr *)&addr, &sockaddr_len); if(s != -1) { sockaddr_to_netaddr((const struct sockaddr *)&addr, a); *new_sock = (NETSOCKET_INTERNAL *)malloc(sizeof(**new_sock)); **new_sock = invalid_socket; (*new_sock)->type = NETTYPE_IPV4; (*new_sock)->ipv4sock = s; return s; } } if(sock->ipv6sock >= 0) { struct sockaddr_in6 addr; sockaddr_len = sizeof(addr); s = accept(sock->ipv6sock, (struct sockaddr *)&addr, &sockaddr_len); if(s != -1) { *new_sock = (NETSOCKET_INTERNAL *)malloc(sizeof(**new_sock)); **new_sock = invalid_socket; sockaddr_to_netaddr((const struct sockaddr *)&addr, a); (*new_sock)->type = NETTYPE_IPV6; (*new_sock)->ipv6sock = s; return s; } } return -1; } int net_tcp_connect(NETSOCKET sock, const NETADDR *a) { if(a->type & NETTYPE_IPV4) { struct sockaddr_in addr; netaddr_to_sockaddr_in(a, &addr); if(sock->ipv4sock < 0) return -2; return connect(sock->ipv4sock, (struct sockaddr *)&addr, sizeof(addr)); } if(a->type & NETTYPE_IPV6) { struct sockaddr_in6 addr; netaddr_to_sockaddr_in6(a, &addr); if(sock->ipv6sock < 0) return -2; return connect(sock->ipv6sock, (struct sockaddr *)&addr, sizeof(addr)); } return -1; } int net_tcp_connect_non_blocking(NETSOCKET sock, NETADDR bindaddr) { int res = 0; net_set_non_blocking(sock); res = net_tcp_connect(sock, &bindaddr); net_set_blocking(sock); return res; } int net_tcp_send(NETSOCKET sock, const void *data, int size) { int bytes = -1; if(sock->ipv4sock >= 0) bytes = send((int)sock->ipv4sock, (const char *)data, size, 0); if(sock->ipv6sock >= 0) bytes = send((int)sock->ipv6sock, (const char *)data, size, 0); return bytes; } int net_tcp_recv(NETSOCKET sock, void *data, int maxsize) { int bytes = -1; if(sock->ipv4sock >= 0) bytes = recv((int)sock->ipv4sock, (char *)data, maxsize, 0); if(sock->ipv6sock >= 0) bytes = recv((int)sock->ipv6sock, (char *)data, maxsize, 0); return bytes; } int net_tcp_close(NETSOCKET sock) { return priv_net_close_all_sockets(sock); } int net_errno() { #if defined(CONF_FAMILY_WINDOWS) return WSAGetLastError(); #else return errno; #endif } int net_would_block() { #if defined(CONF_FAMILY_WINDOWS) return net_errno() == WSAEWOULDBLOCK; #else return net_errno() == EWOULDBLOCK; #endif } void net_init() { #if defined(CONF_FAMILY_WINDOWS) WSADATA wsa_data; dbg_assert(WSAStartup(MAKEWORD(1, 1), &wsa_data) == 0, "network initialization failed."); #endif } #if defined(CONF_FAMILY_UNIX) UNIXSOCKET net_unix_create_unnamed() { return socket(AF_UNIX, SOCK_DGRAM, 0); } int net_unix_send(UNIXSOCKET sock, UNIXSOCKETADDR *addr, void *data, int size) { return sendto(sock, data, size, 0, (struct sockaddr *)addr, sizeof(struct sockaddr_un)); } void net_unix_set_addr(UNIXSOCKETADDR *addr, const char *path) { mem_zero(addr, sizeof(*addr)); addr->sun_family = AF_UNIX; str_copy(addr->sun_path, path); } void net_unix_close(UNIXSOCKET sock) { close(sock); } #endif #if defined(CONF_FAMILY_WINDOWS) static inline time_t filetime_to_unixtime(LPFILETIME filetime) { time_t t; ULARGE_INTEGER li; li.LowPart = filetime->dwLowDateTime; li.HighPart = filetime->dwHighDateTime; li.QuadPart /= 10000000; // 100ns to 1s li.QuadPart -= 11644473600LL; // Windows epoch is in the past t = li.QuadPart; return t == (time_t)li.QuadPart ? t : (time_t)-1; } #endif void fs_listdir(const char *dir, FS_LISTDIR_CALLBACK cb, int type, void *user) { #if defined(CONF_FAMILY_WINDOWS) char buffer[IO_MAX_PATH_LENGTH]; str_format(buffer, sizeof(buffer), "%s/*", dir); const std::wstring wide_buffer = windows_utf8_to_wide(buffer); WIN32_FIND_DATAW finddata; HANDLE handle = FindFirstFileW(wide_buffer.c_str(), &finddata); if(handle == INVALID_HANDLE_VALUE) return; do { const std::optional current_entry = windows_wide_to_utf8(finddata.cFileName); if(!current_entry.has_value()) { log_error("filesystem", "ERROR: file/folder name containing invalid UTF-16 found in folder '%s'", dir); continue; } if(cb(current_entry.value().c_str(), (finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0, type, user)) break; } while(FindNextFileW(handle, &finddata)); FindClose(handle); #else DIR *dir_handle = opendir(dir); if(dir_handle == nullptr) return; char buffer[IO_MAX_PATH_LENGTH]; str_format(buffer, sizeof(buffer), "%s/", dir); size_t length = str_length(buffer); while(true) { struct dirent *entry = readdir(dir_handle); if(entry == nullptr) break; if(!str_utf8_check(entry->d_name)) { log_error("filesystem", "ERROR: file/folder name containing invalid UTF-8 found in folder '%s'", dir); continue; } str_copy(buffer + length, entry->d_name, sizeof(buffer) - length); if(cb(entry->d_name, fs_is_dir(buffer), type, user)) break; } closedir(dir_handle); #endif } void fs_listdir_fileinfo(const char *dir, FS_LISTDIR_CALLBACK_FILEINFO cb, int type, void *user) { #if defined(CONF_FAMILY_WINDOWS) char buffer[IO_MAX_PATH_LENGTH]; str_format(buffer, sizeof(buffer), "%s/*", dir); const std::wstring wide_buffer = windows_utf8_to_wide(buffer); WIN32_FIND_DATAW finddata; HANDLE handle = FindFirstFileW(wide_buffer.c_str(), &finddata); if(handle == INVALID_HANDLE_VALUE) return; do { const std::optional current_entry = windows_wide_to_utf8(finddata.cFileName); if(!current_entry.has_value()) { log_error("filesystem", "ERROR: file/folder name containing invalid UTF-16 found in folder '%s'", dir); continue; } CFsFileInfo info; info.m_pName = current_entry.value().c_str(); info.m_TimeCreated = filetime_to_unixtime(&finddata.ftCreationTime); info.m_TimeModified = filetime_to_unixtime(&finddata.ftLastWriteTime); if(cb(&info, (finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0, type, user)) break; } while(FindNextFileW(handle, &finddata)); FindClose(handle); #else DIR *dir_handle = opendir(dir); if(dir_handle == nullptr) return; char buffer[IO_MAX_PATH_LENGTH]; str_format(buffer, sizeof(buffer), "%s/", dir); size_t length = str_length(buffer); while(true) { struct dirent *entry = readdir(dir_handle); if(entry == nullptr) break; if(!str_utf8_check(entry->d_name)) { log_error("filesystem", "ERROR: file/folder name containing invalid UTF-8 found in folder '%s'", dir); continue; } str_copy(buffer + length, entry->d_name, sizeof(buffer) - length); time_t created = -1, modified = -1; fs_file_time(buffer, &created, &modified); CFsFileInfo info; info.m_pName = entry->d_name; info.m_TimeCreated = created; info.m_TimeModified = modified; if(cb(&info, fs_is_dir(buffer), type, user)) break; } closedir(dir_handle); #endif } int fs_storage_path(const char *appname, char *path, int max) { #if defined(CONF_FAMILY_WINDOWS) WCHAR *wide_home = _wgetenv(L"APPDATA"); if(!wide_home) { path[0] = '\0'; return -1; } const std::optional home = windows_wide_to_utf8(wide_home); if(!home.has_value()) { log_error("filesystem", "ERROR: the APPDATA environment variable contains invalid UTF-16"); path[0] = '\0'; return -1; } str_format(path, max, "%s/%s", home.value().c_str(), appname); return 0; #else char *home = getenv("HOME"); if(!home) { path[0] = '\0'; return -1; } if(!str_utf8_check(home)) { log_error("filesystem", "ERROR: the HOME environment variable contains invalid UTF-8"); path[0] = '\0'; return -1; } #if defined(CONF_PLATFORM_HAIKU) str_format(path, max, "%s/config/settings/%s", home, appname); #elif defined(CONF_PLATFORM_MACOS) str_format(path, max, "%s/Library/Application Support/%s", home, appname); #else if(str_comp(appname, "Teeworlds") == 0) { // fallback for old directory for Teeworlds compatibility str_format(path, max, "%s/.%s", home, appname); } else { char *data_home = getenv("XDG_DATA_HOME"); if(data_home) { if(!str_utf8_check(data_home)) { log_error("filesystem", "ERROR: the XDG_DATA_HOME environment variable contains invalid UTF-8"); path[0] = '\0'; return -1; } str_format(path, max, "%s/%s", data_home, appname); } else str_format(path, max, "%s/.local/share/%s", home, appname); } for(int i = str_length(path) - str_length(appname); path[i]; i++) path[i] = tolower((unsigned char)path[i]); #endif return 0; #endif } int fs_makedir_rec_for(const char *path) { char buffer[IO_MAX_PATH_LENGTH]; str_copy(buffer, path); for(int index = 1; buffer[index] != '\0'; ++index) { // Do not try to create folder for drive letters on Windows, // as this is not necessary and may fail for system drives. if((buffer[index] == '/' || buffer[index] == '\\') && buffer[index + 1] != '\0' && buffer[index - 1] != ':') { buffer[index] = '\0'; if(fs_makedir(buffer) < 0) { return -1; } buffer[index] = '/'; } } return 0; } int fs_makedir(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); if(CreateDirectoryW(wide_path.c_str(), NULL) != 0) return 0; if(GetLastError() == ERROR_ALREADY_EXISTS) return 0; return -1; #else #ifdef CONF_PLATFORM_HAIKU struct stat st; if(stat(path, &st) == 0) return 0; #endif if(mkdir(path, 0755) == 0) return 0; if(errno == EEXIST) return 0; return -1; #endif } int fs_removedir(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); if(RemoveDirectoryW(wide_path.c_str()) != 0) return 0; return -1; #else if(rmdir(path) == 0) return 0; return -1; #endif } int fs_is_file(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); DWORD attributes = GetFileAttributesW(wide_path.c_str()); return attributes != INVALID_FILE_ATTRIBUTES && !(attributes & FILE_ATTRIBUTE_DIRECTORY) ? 1 : 0; #else struct stat sb; if(stat(path, &sb) == -1) return 0; return S_ISREG(sb.st_mode) ? 1 : 0; #endif } int fs_is_dir(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); DWORD attributes = GetFileAttributesW(wide_path.c_str()); return attributes != INVALID_FILE_ATTRIBUTES && (attributes & FILE_ATTRIBUTE_DIRECTORY) ? 1 : 0; #else struct stat sb; if(stat(path, &sb) == -1) return 0; return S_ISDIR(sb.st_mode) ? 1 : 0; #endif } int fs_is_relative_path(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); return PathIsRelativeW(wide_path.c_str()) ? 1 : 0; #else return path[0] == '/' ? 0 : 1; // yes, it's that simple #endif } int fs_chdir(const char *path) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_path = windows_utf8_to_wide(path); return SetCurrentDirectoryW(wide_path.c_str()) != 0 ? 0 : 1; #else return chdir(path) ? 1 : 0; #endif } char *fs_getcwd(char *buffer, int buffer_size) { #if defined(CONF_FAMILY_WINDOWS) const DWORD size_needed = GetCurrentDirectoryW(0, nullptr); std::wstring wide_current_dir(size_needed, L'0'); dbg_assert(GetCurrentDirectoryW(size_needed, wide_current_dir.data()) == size_needed - 1, "GetCurrentDirectoryW failure"); const std::optional current_dir = windows_wide_to_utf8(wide_current_dir.c_str()); if(!current_dir.has_value()) { buffer[0] = '\0'; return nullptr; } str_copy(buffer, current_dir.value().c_str(), buffer_size); return buffer; #else char *result = getcwd(buffer, buffer_size); if(result == nullptr || !str_utf8_check(result)) { buffer[0] = '\0'; return nullptr; } return result; #endif } const char *fs_filename(const char *path) { for(const char *filename = path + str_length(path); filename >= path; --filename) { if(filename[0] == '/' || filename[0] == '\\') return filename + 1; } return path; } void fs_split_file_extension(const char *filename, char *name, size_t name_size, char *extension, size_t extension_size) { dbg_assert(name != nullptr || extension != nullptr, "name or extension parameter required"); dbg_assert(name == nullptr || name_size > 0, "name_size invalid"); dbg_assert(extension == nullptr || extension_size > 0, "extension_size invalid"); const char *last_dot = str_rchr(filename, '.'); if(last_dot == nullptr || last_dot == filename) { if(extension != nullptr) extension[0] = '\0'; if(name != nullptr) str_copy(name, filename, name_size); } else { if(extension != nullptr) str_copy(extension, last_dot + 1, extension_size); if(name != nullptr) str_truncate(name, name_size, filename, last_dot - filename); } } int fs_parent_dir(char *path) { char *parent = 0; for(; *path; ++path) { if(*path == '/' || *path == '\\') parent = path; } if(parent) { *parent = 0; return 0; } return 1; } int fs_remove(const char *filename) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_filename = windows_utf8_to_wide(filename); return DeleteFileW(wide_filename.c_str()) == 0; #else return unlink(filename) != 0; #endif } int fs_rename(const char *oldname, const char *newname) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_oldname = windows_utf8_to_wide(oldname); const std::wstring wide_newname = windows_utf8_to_wide(newname); if(MoveFileExW(wide_oldname.c_str(), wide_newname.c_str(), MOVEFILE_REPLACE_EXISTING | MOVEFILE_COPY_ALLOWED) == 0) return 1; #else if(rename(oldname, newname) != 0) return 1; #endif return 0; } int fs_file_time(const char *name, time_t *created, time_t *modified) { #if defined(CONF_FAMILY_WINDOWS) WIN32_FIND_DATAW finddata; const std::wstring wide_name = windows_utf8_to_wide(name); HANDLE handle = FindFirstFileW(wide_name.c_str(), &finddata); if(handle == INVALID_HANDLE_VALUE) return 1; *created = filetime_to_unixtime(&finddata.ftCreationTime); *modified = filetime_to_unixtime(&finddata.ftLastWriteTime); FindClose(handle); #elif defined(CONF_FAMILY_UNIX) struct stat sb; if(stat(name, &sb)) return 1; *created = sb.st_ctime; *modified = sb.st_mtime; #else #error not implemented #endif return 0; } 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; int sockid; tv.tv_sec = time / 1000000; tv.tv_usec = time % 1000000; sockid = 0; FD_ZERO(&readfds); if(sock->ipv4sock >= 0) { FD_SET(sock->ipv4sock, &readfds); sockid = sock->ipv4sock; } if(sock->ipv6sock >= 0) { FD_SET(sock->ipv6sock, &readfds); if(sock->ipv6sock > sockid) sockid = sock->ipv6sock; } #if defined(CONF_WEBSOCKETS) if(sock->web_ipv4sock >= 0) { int maxfd = websocket_fd_set(sock->web_ipv4sock, &readfds); if(maxfd > sockid) { sockid = maxfd; FD_SET(sockid, &readfds); } } #endif /* don't care about writefds and exceptfds */ if(time < 0) select(sockid + 1, &readfds, NULL, NULL, NULL); else select(sockid + 1, &readfds, NULL, NULL, &tv); if(sock->ipv4sock >= 0 && FD_ISSET(sock->ipv4sock, &readfds)) return 1; #if defined(CONF_WEBSOCKETS) if(sock->web_ipv4sock >= 0 && FD_ISSET(sockid, &readfds)) return 1; #endif if(sock->ipv6sock >= 0 && FD_ISSET(sock->ipv6sock, &readfds)) return 1; return 0; } int64_t time_timestamp() { return time(0); } static struct tm *time_localtime_threadlocal(time_t *time_data) { #if defined(CONF_FAMILY_WINDOWS) // The result of localtime is thread-local on Windows // https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/localtime-localtime32-localtime64 return localtime(time_data); #else // Thread-local buffer for the result of localtime_r thread_local struct tm time_info_buf; return localtime_r(time_data, &time_info_buf); #endif } int time_houroftheday() { time_t time_data; time(&time_data); struct tm *time_info = time_localtime_threadlocal(&time_data); return time_info->tm_hour; } static bool time_iseasterday(time_t time_data, struct tm *time_info) { // compute Easter day (Sunday) using https://en.wikipedia.org/w/index.php?title=Computus&oldid=890710285#Anonymous_Gregorian_algorithm int Y = time_info->tm_year + 1900; int a = Y % 19; int b = Y / 100; int c = Y % 100; int d = b / 4; int e = b % 4; int f = (b + 8) / 25; int g = (b - f + 1) / 3; int h = (19 * a + b - d - g + 15) % 30; int i = c / 4; int k = c % 4; int L = (32 + 2 * e + 2 * i - h - k) % 7; int m = (a + 11 * h + 22 * L) / 451; int month = (h + L - 7 * m + 114) / 31; int day = ((h + L - 7 * m + 114) % 31) + 1; // (now-1d ≤ easter ≤ now+2d) <=> (easter-2d ≤ now ≤ easter+1d) <=> (Good Friday ≤ now ≤ Easter Monday) for(int day_offset = -1; day_offset <= 2; day_offset++) { time_data = time_data + day_offset * 60 * 60 * 24; time_info = time_localtime_threadlocal(&time_data); if(time_info->tm_mon == month - 1 && time_info->tm_mday == day) return true; } return false; } ETimeSeason time_season() { time_t time_data; time(&time_data); struct tm *time_info = time_localtime_threadlocal(&time_data); if((time_info->tm_mon == 11 && time_info->tm_mday == 31) || (time_info->tm_mon == 0 && time_info->tm_mday == 1)) { return SEASON_NEWYEAR; } else if(time_info->tm_mon == 11 && time_info->tm_mday >= 24 && time_info->tm_mday <= 26) { return SEASON_XMAS; } else if((time_info->tm_mon == 9 && time_info->tm_mday == 31) || (time_info->tm_mon == 10 && time_info->tm_mday == 1)) { return SEASON_HALLOWEEN; } else if(time_iseasterday(time_data, time_info)) { return SEASON_EASTER; } switch(time_info->tm_mon) { case 11: case 0: case 1: return SEASON_WINTER; case 2: case 3: case 4: return SEASON_SPRING; case 5: case 6: case 7: return SEASON_SUMMER; case 8: case 9: case 10: return SEASON_AUTUMN; default: dbg_assert(false, "Invalid month"); dbg_break(); } } void str_append(char *dst, const char *src, int dst_size) { int s = str_length(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 */ str_utf8_fix_truncation(dst); } int str_copy(char *dst, const char *src, int dst_size) { dst[0] = '\0'; strncat(dst, src, dst_size - 1); return str_utf8_fix_truncation(dst); } void str_utf8_truncate(char *dst, int dst_size, const char *src, int truncation_len) { int size = -1; const char *cursor = src; int pos = 0; while(pos <= truncation_len && cursor - src < dst_size && size != cursor - src) { size = cursor - src; if(str_utf8_decode(&cursor) == 0) { break; } pos++; } str_copy(dst, src, size + 1); } void str_truncate(char *dst, int dst_size, const char *src, int truncation_len) { int size = dst_size; if(truncation_len < size) { size = truncation_len + 1; } str_copy(dst, src, size); } int str_length(const char *str) { return (int)strlen(str); } int str_format_v(char *buffer, int buffer_size, const char *format, va_list args) { #if defined(CONF_FAMILY_WINDOWS) _vsprintf_p(buffer, buffer_size, format, args); buffer[buffer_size - 1] = 0; /* assure null termination */ #else vsnprintf(buffer, buffer_size, format, args); /* null termination is assured by definition of vsnprintf */ #endif return str_utf8_fix_truncation(buffer); } int str_format_int(char *buffer, size_t buffer_size, int value) { buffer[0] = '\0'; // Fix false positive clang-analyzer-core.UndefinedBinaryOperatorResult when using result auto result = std::to_chars(buffer, buffer + buffer_size - 1, value); result.ptr[0] = '\0'; return result.ptr - buffer; } #undef str_format int str_format(char *buffer, int buffer_size, const char *format, ...) { va_list args; va_start(args, format); int length = str_format_v(buffer, buffer_size, format, args); va_end(args); return length; } #if !defined(CONF_DEBUG) #define str_format str_format_opt #endif const char *str_trim_words(const char *str, int words) { while(*str && str_isspace(*str)) str++; while(words && *str) { if(str_isspace(*str) && !str_isspace(*(str + 1))) words--; str++; } return str; } bool str_has_cc(const char *str) { unsigned char *s = (unsigned char *)str; while(*s) { if(*s < 32) { return true; } s++; } return false; } /* makes sure that the string only contains the characters between 32 and 255 */ void str_sanitize_cc(char *str_in) { unsigned char *str = (unsigned char *)str_in; while(*str) { if(*str < 32) *str = ' '; 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++; } } void str_sanitize_filename(char *str_in) { unsigned char *str = (unsigned char *)str_in; while(*str) { if(*str < 32 || *str == '\\' || *str == '/' || *str == '|' || *str == ':' || *str == '*' || *str == '?' || *str == '<' || *str == '>' || *str == '"') *str = ' '; str++; } } /* removes leading and trailing spaces and limits the use of multiple spaces */ void str_clean_whitespaces(char *str_in) { char *read = str_in; char *write = str_in; /* skip initial whitespace */ while(*read == ' ') read++; /* end of read string is detected in the loop */ while(true) { /* skip whitespace */ int found_whitespace = 0; for(; *read == ' '; read++) found_whitespace = 1; /* if not at the end of the string, put a found whitespace here */ if(*read) { if(found_whitespace) *write++ = ' '; *write++ = *read++; } else { *write = 0; break; } } } char *str_skip_to_whitespace(char *str) { while(*str && !str_isspace(*str)) str++; return str; } const char *str_skip_to_whitespace_const(const char *str) { while(*str && !str_isspace(*str)) str++; return str; } char *str_skip_whitespaces(char *str) { while(*str && str_isspace(*str)) str++; return str; } const char *str_skip_whitespaces_const(const char *str) { while(*str && str_isspace(*str)) str++; return 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 } int str_comp_nocase_num(const char *a, const char *b, int num) { #if defined(CONF_FAMILY_WINDOWS) return _strnicmp(a, b, num); #else return strncasecmp(a, b, num); #endif } int str_comp(const char *a, const char *b) { return strcmp(a, b); } int str_comp_num(const char *a, const char *b, int num) { return strncmp(a, b, num); } int str_comp_filenames(const char *a, const char *b) { int result; for(; *a && *b; ++a, ++b) { if(str_isnum(*a) && str_isnum(*b)) { result = 0; do { if(!result) result = *a - *b; ++a; ++b; } while(str_isnum(*a) && str_isnum(*b)); if(str_isnum(*a)) return 1; else if(str_isnum(*b)) return -1; else if(result || *a == '\0' || *b == '\0') return result; } result = tolower(*a) - tolower(*b); if(result) return result; } return *a - *b; } const char *str_startswith_nocase(const char *str, const char *prefix) { int prefixl = str_length(prefix); if(str_comp_nocase_num(str, prefix, prefixl) == 0) { return str + prefixl; } else { return 0; } } const char *str_startswith(const char *str, const char *prefix) { int prefixl = str_length(prefix); if(str_comp_num(str, prefix, prefixl) == 0) { return str + prefixl; } else { return 0; } } const char *str_endswith_nocase(const char *str, const char *suffix) { int strl = str_length(str); int suffixl = str_length(suffix); const char *strsuffix; if(strl < suffixl) { return 0; } strsuffix = str + strl - suffixl; if(str_comp_nocase(strsuffix, suffix) == 0) { return strsuffix; } else { return 0; } } const char *str_endswith(const char *str, const char *suffix) { int strl = str_length(str); int suffixl = str_length(suffix); const char *strsuffix; if(strl < suffixl) { return 0; } strsuffix = str + strl - suffixl; if(str_comp(strsuffix, suffix) == 0) { return strsuffix; } else { return 0; } } static int min3(int a, int b, int c) { int min = a; if(b < min) min = b; if(c < min) min = c; return min; } int str_utf8_dist(const char *a, const char *b) { int buf_len = 2 * (str_length(a) + 1 + str_length(b) + 1); int *buf = (int *)calloc(buf_len, sizeof(*buf)); int result = str_utf8_dist_buffer(a, b, buf, buf_len); free(buf); return result; } static int str_to_utf32_unchecked(const char *str, int **out) { int out_len = 0; while((**out = str_utf8_decode(&str))) { (*out)++; out_len++; } return out_len; } int str_utf32_dist_buffer(const int *a, int a_len, const int *b, int b_len, int *buf, int buf_len) { int i, j; dbg_assert(buf_len >= (a_len + 1) + (b_len + 1), "buffer too small"); if(a_len > b_len) { int tmp1 = a_len; const int *tmp2 = a; a_len = b_len; a = b; b_len = tmp1; b = tmp2; } #define B(i, j) buf[((j)&1) * (a_len + 1) + (i)] for(i = 0; i <= a_len; i++) { B(i, 0) = i; } for(j = 1; j <= b_len; j++) { B(0, j) = j; for(i = 1; i <= a_len; i++) { int subst = (a[i - 1] != b[j - 1]); B(i, j) = min3( B(i - 1, j) + 1, B(i, j - 1) + 1, B(i - 1, j - 1) + subst); } } return B(a_len, b_len); #undef B } int str_utf8_dist_buffer(const char *a_utf8, const char *b_utf8, int *buf, int buf_len) { int a_utf8_len = str_length(a_utf8); int b_utf8_len = str_length(b_utf8); int *a, *b; // UTF-32 int a_len, b_len; // UTF-32 length dbg_assert(buf_len >= 2 * (a_utf8_len + 1 + b_utf8_len + 1), "buffer too small"); if(a_utf8_len > b_utf8_len) { const char *tmp2 = a_utf8; a_utf8 = b_utf8; b_utf8 = tmp2; } a = buf; a_len = str_to_utf32_unchecked(a_utf8, &buf); b = buf; b_len = str_to_utf32_unchecked(b_utf8, &buf); return str_utf32_dist_buffer(a, a_len, b, b_len, buf, buf_len - b_len - a_len); } 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((unsigned char)*a) == tolower((unsigned char)*b)) { a++; b++; } if(!(*b)) return haystack; haystack++; } return 0; } const char *str_find(const char *haystack, const char *needle) { while(*haystack) /* native implementation */ { const char *a = haystack; const char *b = needle; while(*a && *b && *a == *b) { a++; b++; } if(!(*b)) return haystack; haystack++; } return 0; } const char *str_rchr(const char *haystack, char needle) { return strrchr(haystack, needle); } int str_countchr(const char *haystack, char needle) { int count = 0; while(*haystack) { if(*haystack == needle) count++; haystack++; } return count; } void str_hex(char *dst, int dst_size, const void *data, int data_size) { static const char hex[] = "0123456789ABCDEF"; int data_index; int dst_index; for(data_index = 0, dst_index = 0; data_index < data_size && dst_index < dst_size - 3; data_index++) { dst[data_index * 3] = hex[((const unsigned char *)data)[data_index] >> 4]; dst[data_index * 3 + 1] = hex[((const unsigned char *)data)[data_index] & 0xf]; dst[data_index * 3 + 2] = ' '; dst_index += 3; } dst[dst_index] = '\0'; } void str_hex_cstyle(char *dst, int dst_size, const void *data, int data_size, int bytes_per_line) { static const char hex[] = "0123456789ABCDEF"; int data_index; int dst_index; int remaining_bytes_per_line = bytes_per_line; for(data_index = 0, dst_index = 0; data_index < data_size && dst_index < dst_size - 6; data_index++) { --remaining_bytes_per_line; dst[data_index * 6] = '0'; dst[data_index * 6 + 1] = 'x'; dst[data_index * 6 + 2] = hex[((const unsigned char *)data)[data_index] >> 4]; dst[data_index * 6 + 3] = hex[((const unsigned char *)data)[data_index] & 0xf]; dst[data_index * 6 + 4] = ','; if(remaining_bytes_per_line == 0) { dst[data_index * 6 + 5] = '\n'; remaining_bytes_per_line = bytes_per_line; } else { dst[data_index * 6 + 5] = ' '; } dst_index += 6; } dst[dst_index] = '\0'; // Remove trailing comma and space/newline if(dst_index >= 1) dst[dst_index - 1] = '\0'; if(dst_index >= 2) dst[dst_index - 2] = '\0'; } static int hexval(char x) { switch(x) { case '0': return 0; case '1': return 1; case '2': return 2; case '3': return 3; case '4': return 4; case '5': return 5; case '6': return 6; case '7': return 7; case '8': return 8; case '9': return 9; case 'a': case 'A': return 10; case 'b': case 'B': return 11; case 'c': case 'C': return 12; case 'd': case 'D': return 13; case 'e': case 'E': return 14; case 'f': case 'F': return 15; default: return -1; } } static int byteval(const char *hex, unsigned char *dst) { int v1 = hexval(hex[0]); int v2 = hexval(hex[1]); if(v1 < 0 || v2 < 0) return 1; *dst = v1 * 16 + v2; return 0; } int str_hex_decode(void *dst, int dst_size, const char *src) { unsigned char *cdst = (unsigned char *)dst; int slen = str_length(src); int len = slen / 2; int i; if(slen != dst_size * 2) return 2; for(i = 0; i < len && dst_size; i++, dst_size--) { if(byteval(src + i * 2, cdst++)) return 1; } return 0; } void str_base64(char *dst, int dst_size, const void *data_raw, int data_size) { static const char DIGITS[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; const unsigned char *data = (const unsigned char *)data_raw; unsigned value = 0; int num_bits = 0; int i = 0; int o = 0; dst_size -= 1; dst[dst_size] = 0; while(true) { if(num_bits < 6 && i < data_size) { value = (value << 8) | data[i]; num_bits += 8; i += 1; } if(o == dst_size) { return; } if(num_bits > 0) { unsigned padded; if(num_bits >= 6) { padded = (value >> (num_bits - 6)) & 0x3f; } else { padded = (value << (6 - num_bits)) & 0x3f; } dst[o] = DIGITS[padded]; num_bits -= 6; o += 1; } else if(o % 4 != 0) { dst[o] = '='; o += 1; } else { dst[o] = 0; return; } } } static int base64_digit_value(char digit) { if('A' <= digit && digit <= 'Z') { return digit - 'A'; } else if('a' <= digit && digit <= 'z') { return digit - 'a' + 26; } else if('0' <= digit && digit <= '9') { return digit - '0' + 52; } else if(digit == '+') { return 62; } else if(digit == '/') { return 63; } return -1; } int str_base64_decode(void *dst_raw, int dst_size, const char *data) { unsigned char *dst = (unsigned char *)dst_raw; int data_len = str_length(data); int i; int o = 0; if(data_len % 4 != 0) { return -3; } if(data_len / 4 * 3 > dst_size) { // Output buffer too small. return -2; } for(i = 0; i < data_len; i += 4) { int num_output_bytes = 3; char copy[4]; int d[4]; int value; int b; mem_copy(copy, data + i, sizeof(copy)); if(i == data_len - 4) { if(copy[3] == '=') { copy[3] = 'A'; num_output_bytes = 2; if(copy[2] == '=') { copy[2] = 'A'; num_output_bytes = 1; } } } d[0] = base64_digit_value(copy[0]); d[1] = base64_digit_value(copy[1]); d[2] = base64_digit_value(copy[2]); d[3] = base64_digit_value(copy[3]); if(d[0] == -1 || d[1] == -1 || d[2] == -1 || d[3] == -1) { // Invalid digit. return -1; } value = (d[0] << 18) | (d[1] << 12) | (d[2] << 6) | d[3]; for(b = 0; b < 3; b++) { unsigned char byte_value = (value >> (16 - 8 * b)) & 0xff; if(b < num_output_bytes) { dst[o] = byte_value; o += 1; } else { if(byte_value != 0) { // Padding not zeroed. return -2; } } } } return o; } #ifdef __GNUC__ #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wformat-nonliteral" #endif void str_timestamp_ex(time_t time_data, char *buffer, int buffer_size, const char *format) { struct tm *time_info = time_localtime_threadlocal(&time_data); strftime(buffer, buffer_size, format, time_info); buffer[buffer_size - 1] = 0; /* assure null termination */ } void str_timestamp_format(char *buffer, int buffer_size, const char *format) { time_t time_data; time(&time_data); str_timestamp_ex(time_data, buffer, buffer_size, format); } void str_timestamp(char *buffer, int buffer_size) { str_timestamp_format(buffer, buffer_size, FORMAT_NOSPACE); } bool timestamp_from_str(const char *string, const char *format, time_t *timestamp) { std::tm tm{}; std::istringstream ss(string); ss >> std::get_time(&tm, format); if(ss.fail() || !ss.eof()) return false; time_t result = mktime(&tm); if(result < 0) return false; *timestamp = result; return true; } #ifdef __GNUC__ #pragma GCC diagnostic pop #endif int str_time(int64_t centisecs, int format, char *buffer, int buffer_size) { const int sec = 100; const int min = 60 * sec; const int hour = 60 * min; const int day = 24 * hour; if(buffer_size <= 0) return -1; if(centisecs < 0) centisecs = 0; buffer[0] = 0; switch(format) { case TIME_DAYS: if(centisecs >= day) return str_format(buffer, buffer_size, "%" PRId64 "d %02" PRId64 ":%02" PRId64 ":%02" PRId64, centisecs / day, (centisecs % day) / hour, (centisecs % hour) / min, (centisecs % min) / sec); [[fallthrough]]; case TIME_HOURS: if(centisecs >= hour) return str_format(buffer, buffer_size, "%02" PRId64 ":%02" PRId64 ":%02" PRId64, centisecs / hour, (centisecs % hour) / min, (centisecs % min) / sec); [[fallthrough]]; case TIME_MINS: return str_format(buffer, buffer_size, "%02" PRId64 ":%02" PRId64, centisecs / min, (centisecs % min) / sec); case TIME_HOURS_CENTISECS: if(centisecs >= hour) return str_format(buffer, buffer_size, "%02" PRId64 ":%02" PRId64 ":%02" PRId64 ".%02" PRId64, centisecs / hour, (centisecs % hour) / min, (centisecs % min) / sec, centisecs % sec); [[fallthrough]]; case TIME_MINS_CENTISECS: if(centisecs >= min) return str_format(buffer, buffer_size, "%02" PRId64 ":%02" PRId64 ".%02" PRId64, centisecs / min, (centisecs % min) / sec, centisecs % sec); [[fallthrough]]; case TIME_SECS_CENTISECS: return str_format(buffer, buffer_size, "%02" PRId64 ".%02" PRId64, (centisecs % min) / sec, centisecs % sec); } return -1; } int str_time_float(float secs, int format, char *buffer, int buffer_size) { return str_time(llroundf(secs * 1000) / 10, format, buffer, buffer_size); } void str_escape(char **dst, const char *src, const char *end) { while(*src && *dst + 1 < end) { if(*src == '"' || *src == '\\') // escape \ and " { if(*dst + 2 < end) *(*dst)++ = '\\'; else break; } *(*dst)++ = *src++; } **dst = 0; } void net_stats(NETSTATS *stats_inout) { *stats_inout = network_stats; } int str_isspace(char c) { return c == ' ' || c == '\n' || c == '\r' || c == '\t'; } char str_uppercase(char c) { if(c >= 'a' && c <= 'z') return 'A' + (c - 'a'); return c; } bool str_isnum(char c) { return c >= '0' && c <= '9'; } int str_isallnum(const char *str) { while(*str) { if(!str_isnum(*str)) return 0; str++; } return 1; } int str_isallnum_hex(const char *str) { while(*str) { if(!str_isnum(*str) && !(*str >= 'a' && *str <= 'f') && !(*str >= 'A' && *str <= 'F')) return 0; str++; } return 1; } int str_toint(const char *str) { return str_toint_base(str, 10); } bool str_toint(const char *str, int *out) { // returns true if conversion was successful char *end; int value = strtol(str, &end, 10); if(*end != '\0') return false; if(out != nullptr) *out = value; return true; } int str_toint_base(const char *str, int base) { return strtol(str, nullptr, base); } unsigned long str_toulong_base(const char *str, int base) { return strtoul(str, nullptr, base); } int64_t str_toint64_base(const char *str, int base) { return strtoll(str, nullptr, base); } float str_tofloat(const char *str) { return strtod(str, nullptr); } bool str_tofloat(const char *str, float *out) { // returns true if conversion was successful char *end; float value = strtod(str, &end); if(*end != '\0') return false; if(out != nullptr) *out = value; return true; } int str_utf8_comp_nocase(const char *a, const char *b) { int code_a; int code_b; while(*a && *b) { code_a = str_utf8_tolower(str_utf8_decode(&a)); code_b = str_utf8_tolower(str_utf8_decode(&b)); if(code_a != code_b) return code_a - code_b; } return (unsigned char)*a - (unsigned char)*b; } int str_utf8_comp_nocase_num(const char *a, const char *b, int num) { int code_a; int code_b; const char *old_a = a; if(num <= 0) return 0; while(*a && *b) { code_a = str_utf8_tolower(str_utf8_decode(&a)); code_b = str_utf8_tolower(str_utf8_decode(&b)); if(code_a != code_b) return code_a - code_b; if(a - old_a >= num) return 0; } return (unsigned char)*a - (unsigned char)*b; } const char *str_utf8_find_nocase(const char *haystack, const char *needle, const char **end) { while(*haystack) /* native implementation */ { const char *a = haystack; const char *b = needle; const char *a_next = a; const char *b_next = b; while(*a && *b && str_utf8_tolower(str_utf8_decode(&a_next)) == str_utf8_tolower(str_utf8_decode(&b_next))) { a = a_next; b = b_next; } if(!(*b)) { if(end != nullptr) *end = a_next; return haystack; } str_utf8_decode(&haystack); } if(end != nullptr) *end = nullptr; return nullptr; } int str_utf8_isspace(int code) { return code <= 0x0020 || code == 0x0085 || code == 0x00A0 || code == 0x034F || code == 0x115F || code == 0x1160 || code == 0x1680 || code == 0x180E || (code >= 0x2000 && code <= 0x200F) || (code >= 0x2028 && code <= 0x202F) || (code >= 0x205F && code <= 0x2064) || (code >= 0x206A && code <= 0x206F) || code == 0x2800 || code == 0x3000 || code == 0x3164 || (code >= 0xFE00 && code <= 0xFE0F) || code == 0xFEFF || code == 0xFFA0 || (code >= 0xFFF9 && code <= 0xFFFC); } const char *str_utf8_skip_whitespaces(const char *str) { const char *str_old; int code; while(*str) { str_old = str; code = str_utf8_decode(&str); // check if unicode is not empty if(!str_utf8_isspace(code)) { return str_old; } } return str; } void str_utf8_trim_right(char *param) { const char *str = param; char *end = 0; while(*str) { char *str_old = (char *)str; int code = str_utf8_decode(&str); // check if unicode is not empty if(!str_utf8_isspace(code)) { end = 0; } else if(!end) { end = str_old; } } if(end) { *end = 0; } } int str_utf8_isstart(char c) { if((c & 0xC0) == 0x80) /* 10xxxxxx */ return 0; return 1; } int str_utf8_rewind(const char *str, int cursor) { while(cursor) { cursor--; if(str_utf8_isstart(*(str + cursor))) break; } return cursor; } int str_utf8_fix_truncation(char *str) { int len = str_length(str); if(len > 0) { int last_char_index = str_utf8_rewind(str, len); const char *last_char = str + last_char_index; // Fix truncated UTF-8. if(str_utf8_decode(&last_char) == -1) { str[last_char_index] = 0; return last_char_index; } } return len; } int str_utf8_forward(const char *str, int cursor) { const char *ptr = str + cursor; if(str_utf8_decode(&ptr) == 0) { return cursor; } return ptr - str; } int str_utf8_encode(char *ptr, int chr) { /* encode */ if(chr <= 0x7F) { ptr[0] = (char)chr; return 1; } else if(chr <= 0x7FF) { ptr[0] = 0xC0 | ((chr >> 6) & 0x1F); ptr[1] = 0x80 | (chr & 0x3F); return 2; } else if(chr <= 0xFFFF) { ptr[0] = 0xE0 | ((chr >> 12) & 0x0F); ptr[1] = 0x80 | ((chr >> 6) & 0x3F); ptr[2] = 0x80 | (chr & 0x3F); return 3; } else if(chr <= 0x10FFFF) { ptr[0] = 0xF0 | ((chr >> 18) & 0x07); ptr[1] = 0x80 | ((chr >> 12) & 0x3F); ptr[2] = 0x80 | ((chr >> 6) & 0x3F); ptr[3] = 0x80 | (chr & 0x3F); return 4; } return 0; } static unsigned char str_byte_next(const char **ptr) { unsigned char byte_value = **ptr; (*ptr)++; return byte_value; } static void str_byte_rewind(const char **ptr) { (*ptr)--; } int str_utf8_decode(const char **ptr) { // As per https://encoding.spec.whatwg.org/#utf-8-decoder. unsigned char utf8_lower_boundary = 0x80; unsigned char utf8_upper_boundary = 0xBF; int utf8_code_point = 0; int utf8_bytes_seen = 0; int utf8_bytes_needed = 0; while(true) { unsigned char byte_value = str_byte_next(ptr); if(utf8_bytes_needed == 0) { if(byte_value <= 0x7F) { return byte_value; } else if(0xC2 <= byte_value && byte_value <= 0xDF) { utf8_bytes_needed = 1; utf8_code_point = byte_value - 0xC0; } else if(0xE0 <= byte_value && byte_value <= 0xEF) { if(byte_value == 0xE0) utf8_lower_boundary = 0xA0; if(byte_value == 0xED) utf8_upper_boundary = 0x9F; utf8_bytes_needed = 2; utf8_code_point = byte_value - 0xE0; } else if(0xF0 <= byte_value && byte_value <= 0xF4) { if(byte_value == 0xF0) utf8_lower_boundary = 0x90; if(byte_value == 0xF4) utf8_upper_boundary = 0x8F; utf8_bytes_needed = 3; utf8_code_point = byte_value - 0xF0; } else { return -1; // Error. } utf8_code_point = utf8_code_point << (6 * utf8_bytes_needed); continue; } if(!(utf8_lower_boundary <= byte_value && byte_value <= utf8_upper_boundary)) { // Resetting variables not necessary, will be done when // the function is called again. str_byte_rewind(ptr); return -1; } utf8_lower_boundary = 0x80; utf8_upper_boundary = 0xBF; utf8_bytes_seen += 1; utf8_code_point = utf8_code_point + ((byte_value - 0x80) << (6 * (utf8_bytes_needed - utf8_bytes_seen))); if(utf8_bytes_seen != utf8_bytes_needed) { continue; } // Resetting variables not necessary, see above. return utf8_code_point; } } int str_utf8_check(const char *str) { int codepoint; while((codepoint = str_utf8_decode(&str))) { if(codepoint == -1) { return 0; } } return 1; } void str_utf8_stats(const char *str, size_t max_size, size_t max_count, size_t *size, size_t *count) { const char *cursor = str; *size = 0; *count = 0; while(*size < max_size && *count < max_count) { if(str_utf8_decode(&cursor) == 0) { break; } if((size_t)(cursor - str) >= max_size) { break; } *size = cursor - str; ++(*count); } } size_t str_utf8_offset_bytes_to_chars(const char *str, size_t byte_offset) { size_t char_offset = 0; size_t current_offset = 0; while(current_offset < byte_offset) { const size_t prev_byte_offset = current_offset; current_offset = str_utf8_forward(str, current_offset); if(current_offset == prev_byte_offset) break; char_offset++; } return char_offset; } size_t str_utf8_offset_chars_to_bytes(const char *str, size_t char_offset) { size_t byte_offset = 0; for(size_t i = 0; i < char_offset; i++) { const size_t prev_byte_offset = byte_offset; byte_offset = str_utf8_forward(str, byte_offset); if(byte_offset == prev_byte_offset) break; } return byte_offset; } unsigned str_quickhash(const char *str) { unsigned hash = 5381; for(; *str; str++) hash = ((hash << 5) + hash) + (*str); /* hash * 33 + c */ return hash; } static const char *str_token_get(const char *str, const char *delim, int *length) { size_t len = strspn(str, delim); if(len > 1) str++; else str += len; if(!*str) return NULL; *length = strcspn(str, delim); return str; } int str_in_list(const char *list, const char *delim, const char *needle) { const char *tok = list; int len = 0, notfound = 1, needlelen = str_length(needle); while(notfound && (tok = str_token_get(tok, delim, &len))) { notfound = needlelen != len || str_comp_num(tok, needle, len); tok = tok + len; } return !notfound; } const char *str_next_token(const char *str, const char *delim, char *buffer, int buffer_size) { int len = 0; const char *tok = str_token_get(str, delim, &len); if(len < 0 || tok == NULL) { buffer[0] = '\0'; return NULL; } len = buffer_size > len ? len : buffer_size - 1; mem_copy(buffer, tok, len); buffer[len] = '\0'; return tok + len; } static_assert(sizeof(unsigned) == 4, "unsigned must be 4 bytes in size"); static_assert(sizeof(unsigned) == sizeof(int), "unsigned and int must have the same size"); unsigned bytes_be_to_uint(const unsigned char *bytes) { return ((bytes[0] & 0xffu) << 24u) | ((bytes[1] & 0xffu) << 16u) | ((bytes[2] & 0xffu) << 8u) | (bytes[3] & 0xffu); } void uint_to_bytes_be(unsigned char *bytes, unsigned value) { bytes[0] = (value >> 24u) & 0xffu; bytes[1] = (value >> 16u) & 0xffu; bytes[2] = (value >> 8u) & 0xffu; bytes[3] = value & 0xffu; } int pid() { #if defined(CONF_FAMILY_WINDOWS) return _getpid(); #else return getpid(); #endif } void cmdline_fix(int *argc, const char ***argv) { #if defined(CONF_FAMILY_WINDOWS) int wide_argc = 0; WCHAR **wide_argv = CommandLineToArgvW(GetCommandLineW(), &wide_argc); dbg_assert(wide_argv != NULL, "CommandLineToArgvW failure"); dbg_assert(wide_argc > 0, "Invalid argc value"); int total_size = 0; for(int i = 0; i < wide_argc; i++) { int size = WideCharToMultiByte(CP_UTF8, 0, wide_argv[i], -1, NULL, 0, NULL, NULL); dbg_assert(size != 0, "WideCharToMultiByte failure"); total_size += size; } char **new_argv = (char **)malloc((wide_argc + 1) * sizeof(*new_argv)); new_argv[0] = (char *)malloc(total_size); mem_zero(new_argv[0], total_size); int remaining_size = total_size; for(int i = 0; i < wide_argc; i++) { int size = WideCharToMultiByte(CP_UTF8, 0, wide_argv[i], -1, new_argv[i], remaining_size, NULL, NULL); dbg_assert(size != 0, "WideCharToMultiByte failure"); remaining_size -= size; new_argv[i + 1] = new_argv[i] + size; } LocalFree(wide_argv); new_argv[wide_argc] = 0; *argc = wide_argc; *argv = (const char **)new_argv; #endif } void cmdline_free(int argc, const char **argv) { #if defined(CONF_FAMILY_WINDOWS) free((void *)*argv); free((char **)argv); #endif } PROCESS shell_execute(const char *file, EShellExecuteWindowState window_state) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_file = windows_utf8_to_wide(file); SHELLEXECUTEINFOW info; mem_zero(&info, sizeof(SHELLEXECUTEINFOW)); info.cbSize = sizeof(SHELLEXECUTEINFOW); info.lpVerb = L"open"; info.lpFile = wide_file.c_str(); switch(window_state) { case EShellExecuteWindowState::FOREGROUND: info.nShow = SW_SHOW; break; case EShellExecuteWindowState::BACKGROUND: info.nShow = SW_SHOWMINNOACTIVE; break; default: dbg_assert(false, "window_state invalid"); dbg_break(); } info.fMask = SEE_MASK_NOCLOSEPROCESS; // Save and restore the FPU control word because ShellExecute might change it fenv_t floating_point_environment; int fegetenv_result = fegetenv(&floating_point_environment); ShellExecuteExW(&info); if(fegetenv_result == 0) fesetenv(&floating_point_environment); return info.hProcess; #elif defined(CONF_FAMILY_UNIX) char *argv[2]; pid_t pid; argv[0] = (char *)file; argv[1] = NULL; pid = fork(); if(pid == -1) { return 0; } if(pid == 0) { execvp(file, argv); _exit(1); } return pid; #endif } int kill_process(PROCESS process) { #if defined(CONF_FAMILY_WINDOWS) BOOL success = TerminateProcess(process, 0); BOOL is_alive = is_process_alive(process); if(success || !is_alive) { CloseHandle(process); return true; } return false; #elif defined(CONF_FAMILY_UNIX) if(!is_process_alive(process)) return true; int status; kill(process, SIGTERM); return waitpid(process, &status, 0) != -1; #endif } bool is_process_alive(PROCESS process) { if(process == INVALID_PROCESS) return false; #if defined(CONF_FAMILY_WINDOWS) DWORD exit_code; GetExitCodeProcess(process, &exit_code); return exit_code == STILL_ACTIVE; #else return waitpid(process, nullptr, WNOHANG) == 0; #endif } int open_link(const char *link) { #if defined(CONF_FAMILY_WINDOWS) const std::wstring wide_link = windows_utf8_to_wide(link); SHELLEXECUTEINFOW info; mem_zero(&info, sizeof(SHELLEXECUTEINFOW)); info.cbSize = sizeof(SHELLEXECUTEINFOW); info.lpVerb = NULL; // NULL to use the default verb, as "open" may not be available info.lpFile = wide_link.c_str(); info.nShow = SW_SHOWNORMAL; // The SEE_MASK_NOASYNC flag ensures that the ShellExecuteEx function // finishes its DDE conversation before it returns, so it's not necessary // to pump messages in the calling thread. // The SEE_MASK_FLAG_NO_UI flag suppresses error messages that would pop up // when the link cannot be opened, e.g. when a folder does not exist. // The SEE_MASK_ASYNCOK flag is not used. It would allow the call to // ShellExecuteEx to return earlier, but it also prevents us from doing // our own error handling, as the function would always return TRUE. info.fMask = SEE_MASK_NOASYNC | SEE_MASK_FLAG_NO_UI; // Save and restore the FPU control word because ShellExecute might change it fenv_t floating_point_environment; int fegetenv_result = fegetenv(&floating_point_environment); BOOL success = ShellExecuteExW(&info); if(fegetenv_result == 0) fesetenv(&floating_point_environment); return success; #elif defined(CONF_PLATFORM_LINUX) const int pid = fork(); if(pid == 0) execlp("xdg-open", "xdg-open", link, nullptr); return pid > 0; #elif defined(CONF_FAMILY_UNIX) const int pid = fork(); if(pid == 0) execlp("open", "open", link, nullptr); return pid > 0; #endif } int open_file(const char *path) { #if defined(CONF_PLATFORM_MACOS) return open_link(path); #else // Create a file link so the path can contain forward and // backward slashes. But the file link must be absolute. char buf[512]; char workingDir[IO_MAX_PATH_LENGTH]; if(fs_is_relative_path(path)) { if(!fs_getcwd(workingDir, sizeof(workingDir))) return 0; str_append(workingDir, "/"); } else workingDir[0] = '\0'; str_format(buf, sizeof(buf), "file://%s%s", workingDir, path); return open_link(buf); #endif } struct SECURE_RANDOM_DATA { int initialized; #if defined(CONF_FAMILY_WINDOWS) HCRYPTPROV provider; #else IOHANDLE urandom; #endif }; static struct SECURE_RANDOM_DATA secure_random_data = {0}; int secure_random_init() { if(secure_random_data.initialized) { return 0; } #if defined(CONF_FAMILY_WINDOWS) if(CryptAcquireContext(&secure_random_data.provider, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) { secure_random_data.initialized = 1; return 0; } else { return 1; } #else secure_random_data.urandom = io_open("/dev/urandom", IOFLAG_READ); if(secure_random_data.urandom) { secure_random_data.initialized = 1; return 0; } else { return 1; } #endif } int secure_random_uninit() { if(!secure_random_data.initialized) { return 0; } #if defined(CONF_FAMILY_WINDOWS) if(CryptReleaseContext(secure_random_data.provider, 0)) { secure_random_data.initialized = 0; return 0; } else { return 1; } #else if(!io_close(secure_random_data.urandom)) { secure_random_data.initialized = 0; return 0; } else { return 1; } #endif } void generate_password(char *buffer, unsigned length, const unsigned short *random, unsigned random_length) { static const char VALUES[] = "ABCDEFGHKLMNPRSTUVWXYZabcdefghjkmnopqt23456789"; static const size_t NUM_VALUES = sizeof(VALUES) - 1; // Disregard the '\0'. unsigned i; dbg_assert(length >= random_length * 2 + 1, "too small buffer"); dbg_assert(NUM_VALUES * NUM_VALUES >= 2048, "need at least 2048 possibilities for 2-character sequences"); buffer[random_length * 2] = 0; for(i = 0; i < random_length; i++) { unsigned short random_number = random[i] % 2048; buffer[2 * i + 0] = VALUES[random_number / NUM_VALUES]; buffer[2 * i + 1] = VALUES[random_number % NUM_VALUES]; } } #define MAX_PASSWORD_LENGTH 128 void secure_random_password(char *buffer, unsigned length, unsigned pw_length) { unsigned short random[MAX_PASSWORD_LENGTH / 2]; // With 6 characters, we get a password entropy of log(2048) * 6/2 = 33bit. dbg_assert(length >= pw_length + 1, "too small buffer"); dbg_assert(pw_length >= 6, "too small password length"); dbg_assert(pw_length % 2 == 0, "need an even password length"); dbg_assert(pw_length <= MAX_PASSWORD_LENGTH, "too large password length"); secure_random_fill(random, pw_length); generate_password(buffer, length, random, pw_length / 2); } #undef MAX_PASSWORD_LENGTH void secure_random_fill(void *bytes, unsigned length) { if(!secure_random_data.initialized) { dbg_msg("secure", "called secure_random_fill before secure_random_init"); dbg_break(); } #if defined(CONF_FAMILY_WINDOWS) if(!CryptGenRandom(secure_random_data.provider, length, (unsigned char *)bytes)) { const DWORD LastError = GetLastError(); const std::string ErrorMsg = windows_format_system_message(LastError); dbg_msg("secure", "CryptGenRandom failed: %ld %s", LastError, ErrorMsg.c_str()); dbg_break(); } #else if(length != io_read(secure_random_data.urandom, bytes, length)) { dbg_msg("secure", "io_read returned with a short read"); dbg_break(); } #endif } int secure_rand() { unsigned int i; secure_random_fill(&i, sizeof(i)); return (int)(i % RAND_MAX); } // From https://graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2. static unsigned int find_next_power_of_two_minus_one(unsigned int n) { n--; n |= n >> 1; n |= n >> 2; n |= n >> 4; n |= n >> 4; n |= n >> 16; return n; } int secure_rand_below(int below) { unsigned int mask = find_next_power_of_two_minus_one(below); dbg_assert(below > 0, "below must be positive"); while(true) { unsigned int n; secure_random_fill(&n, sizeof(n)); n &= mask; if((int)n < below) { return n; } } } bool os_version_str(char *version, size_t length) { #if defined(CONF_FAMILY_WINDOWS) const WCHAR *module_path = L"kernel32.dll"; DWORD handle; DWORD size = GetFileVersionInfoSizeW(module_path, &handle); if(!size) { return false; } void *data = malloc(size); if(!GetFileVersionInfoW(module_path, handle, size, data)) { free(data); return false; } VS_FIXEDFILEINFO *fileinfo; UINT unused; if(!VerQueryValueW(data, L"\\", (void **)&fileinfo, &unused)) { free(data); return false; } str_format(version, length, "Windows %hu.%hu.%hu.%hu", HIWORD(fileinfo->dwProductVersionMS), LOWORD(fileinfo->dwProductVersionMS), HIWORD(fileinfo->dwProductVersionLS), LOWORD(fileinfo->dwProductVersionLS)); free(data); return true; #else struct utsname u; if(uname(&u)) { return false; } char extra[128]; extra[0] = 0; do { IOHANDLE os_release = io_open("/etc/os-release", IOFLAG_READ); char buf[4096]; int read; int offset; char *newline; if(!os_release) { break; } read = io_read(os_release, buf, sizeof(buf) - 1); io_close(os_release); buf[read] = 0; if(str_startswith(buf, "PRETTY_NAME=")) { offset = 0; } else { const char *found = str_find(buf, "\nPRETTY_NAME="); if(!found) { break; } offset = found - buf + 1; } newline = (char *)str_find(buf + offset, "\n"); if(newline) { *newline = 0; } str_format(extra, sizeof(extra), "; %s", buf + offset + 12); } while(false); str_format(version, length, "%s %s (%s, %s)%s", u.sysname, u.release, u.machine, u.version, extra); return true; #endif } void os_locale_str(char *locale, size_t length) { #if defined(CONF_FAMILY_WINDOWS) wchar_t wide_buffer[LOCALE_NAME_MAX_LENGTH]; dbg_assert(GetUserDefaultLocaleName(wide_buffer, std::size(wide_buffer)) > 0, "GetUserDefaultLocaleName failure"); const std::optional buffer = windows_wide_to_utf8(wide_buffer); dbg_assert(buffer.has_value(), "GetUserDefaultLocaleName returned invalid UTF-16"); str_copy(locale, buffer.value().c_str(), length); #elif defined(CONF_PLATFORM_MACOS) CFLocaleRef locale_ref = CFLocaleCopyCurrent(); CFStringRef locale_identifier_ref = static_cast(CFLocaleGetValue(locale_ref, kCFLocaleIdentifier)); // Count number of UTF16 codepoints, +1 for zero-termination. // Assume maximum possible length for encoding as UTF-8. CFIndex locale_identifier_size = (UTF8_BYTE_LENGTH * CFStringGetLength(locale_identifier_ref) + 1) * sizeof(char); char *locale_identifier = (char *)malloc(locale_identifier_size); dbg_assert(CFStringGetCString(locale_identifier_ref, locale_identifier, locale_identifier_size, kCFStringEncodingUTF8), "CFStringGetCString failure"); str_copy(locale, locale_identifier, length); free(locale_identifier); CFRelease(locale_ref); #else static const char *ENV_VARIABLES[] = { "LC_ALL", "LC_MESSAGES", "LANG", }; locale[0] = '\0'; for(const char *env_variable : ENV_VARIABLES) { const char *env_value = getenv(env_variable); if(env_value) { str_copy(locale, env_value, length); break; } } #endif // Ensure RFC 3066 format: // - use hyphens instead of underscores // - truncate locale string after first non-standard letter for(int i = 0; i < str_length(locale); ++i) { if(locale[i] == '_') { locale[i] = '-'; } else if(locale[i] != '-' && !(locale[i] >= 'a' && locale[i] <= 'z') && !(locale[i] >= 'A' && locale[i] <= 'Z') && !(str_isnum(locale[i]))) { locale[i] = '\0'; break; } } // Use default if we could not determine the locale, // i.e. if only the C or POSIX locale is available. if(locale[0] == '\0' || str_comp(locale, "C") == 0 || str_comp(locale, "POSIX") == 0) str_copy(locale, "en-US", length); } #if defined(CONF_EXCEPTION_HANDLING) #if defined(CONF_FAMILY_WINDOWS) static HMODULE exception_handling_module = nullptr; #endif void init_exception_handler() { #if defined(CONF_FAMILY_WINDOWS) const char *module_name = "exchndl.dll"; exception_handling_module = LoadLibraryA(module_name); if(exception_handling_module == nullptr) { const DWORD LastError = GetLastError(); const std::string ErrorMsg = windows_format_system_message(LastError); dbg_msg("exception_handling", "failed to load exception handling library '%s' (error %ld %s)", module_name, LastError, ErrorMsg.c_str()); } #else #error exception handling not implemented #endif } void set_exception_handler_log_file(const char *log_file_path) { #if defined(CONF_FAMILY_WINDOWS) if(exception_handling_module != nullptr) { const std::wstring wide_log_file_path = windows_utf8_to_wide(log_file_path); // Intentional #ifdef __MINGW32__ #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wcast-function-type" #endif const char *function_name = "ExcHndlSetLogFileNameW"; auto exception_log_file_path_func = (BOOL(APIENTRY *)(const WCHAR *))(GetProcAddress(exception_handling_module, function_name)); #ifdef __MINGW32__ #pragma GCC diagnostic pop #endif if(exception_log_file_path_func == nullptr) { const DWORD LastError = GetLastError(); const std::string ErrorMsg = windows_format_system_message(LastError); dbg_msg("exception_handling", "could not find function '%s' in exception handling library (error %ld %s)", function_name, LastError, ErrorMsg.c_str()); } else exception_log_file_path_func(wide_log_file_path.c_str()); } #else #error exception handling not implemented #endif } #endif std::chrono::nanoseconds time_get_nanoseconds() { return std::chrono::nanoseconds(time_get_impl()); } int net_socket_read_wait(NETSOCKET sock, std::chrono::nanoseconds nanoseconds) { using namespace std::chrono_literals; return ::net_socket_read_wait(sock, (nanoseconds / std::chrono::nanoseconds(1us).count()).count()); } #if defined(CONF_FAMILY_WINDOWS) std::wstring windows_utf8_to_wide(const char *str) { const int orig_length = str_length(str); if(orig_length == 0) return L""; const int size_needed = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str, orig_length, nullptr, 0); dbg_assert(size_needed > 0, "Invalid UTF-8 passed to windows_utf8_to_wide"); std::wstring wide_string(size_needed, L'\0'); dbg_assert(MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str, orig_length, wide_string.data(), size_needed) == size_needed, "MultiByteToWideChar failure"); return wide_string; } std::optional windows_wide_to_utf8(const wchar_t *wide_str) { const int orig_length = wcslen(wide_str); if(orig_length == 0) return ""; const int size_needed = WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, wide_str, orig_length, nullptr, 0, nullptr, nullptr); if(size_needed == 0) return {}; std::string string(size_needed, '\0'); dbg_assert(WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, wide_str, orig_length, string.data(), size_needed, nullptr, nullptr) == size_needed, "WideCharToMultiByte failure"); return string; } // See https://learn.microsoft.com/en-us/windows/win32/learnwin32/initializing-the-com-library CWindowsComLifecycle::CWindowsComLifecycle(bool HasWindow) { HRESULT result = CoInitializeEx(NULL, (HasWindow ? COINIT_APARTMENTTHREADED : COINIT_MULTITHREADED) | COINIT_DISABLE_OLE1DDE); dbg_assert(result != S_FALSE, "COM library already initialized on this thread"); dbg_assert(result == S_OK, "COM library initialization failed"); } CWindowsComLifecycle::~CWindowsComLifecycle() { CoUninitialize(); } static void windows_print_error(const char *system, const char *prefix, HRESULT error) { const std::string message = windows_format_system_message(error); dbg_msg(system, "%s: %s", prefix, message.c_str()); } static std::wstring filename_from_path(const std::wstring &path) { const size_t pos = path.find_last_of(L"/\\"); return pos == std::wstring::npos ? path : path.substr(pos + 1); } bool shell_register_protocol(const char *protocol_name, const char *executable, bool *updated) { const std::wstring protocol_name_wide = windows_utf8_to_wide(protocol_name); const std::wstring executable_wide = windows_utf8_to_wide(executable); // Open registry key for protocol associations of the current user HKEY handle_subkey_classes; const LRESULT result_subkey_classes = RegOpenKeyExW(HKEY_CURRENT_USER, L"SOFTWARE\\Classes", 0, KEY_ALL_ACCESS, &handle_subkey_classes); if(result_subkey_classes != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error opening registry key", result_subkey_classes); return false; } // Create the protocol key HKEY handle_subkey_protocol; const LRESULT result_subkey_protocol = RegCreateKeyExW(handle_subkey_classes, protocol_name_wide.c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_protocol, NULL); RegCloseKey(handle_subkey_classes); if(result_subkey_protocol != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error creating registry key", result_subkey_protocol); return false; } // Set the default value for the key, which specifies the name of the display name of the protocol const std::wstring value_protocol = L"URL:" + protocol_name_wide + L" Protocol"; const LRESULT result_value_protocol = RegSetValueExW(handle_subkey_protocol, L"", 0, REG_SZ, (BYTE *)value_protocol.c_str(), (value_protocol.length() + 1) * sizeof(wchar_t)); if(result_value_protocol != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error setting registry value", result_value_protocol); RegCloseKey(handle_subkey_protocol); return false; } // Set the "URL Protocol" value, to specify that this key describes a URL protocol const LRESULT result_value_empty = RegSetValueEx(handle_subkey_protocol, L"URL Protocol", 0, REG_SZ, (BYTE *)L"", sizeof(wchar_t)); if(result_value_empty != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error setting registry value", result_value_empty); RegCloseKey(handle_subkey_protocol); return false; } // Create the "DefaultIcon" subkey HKEY handle_subkey_icon; const LRESULT result_subkey_icon = RegCreateKeyExW(handle_subkey_protocol, L"DefaultIcon", 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_icon, NULL); if(result_subkey_icon != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error creating registry key", result_subkey_icon); RegCloseKey(handle_subkey_protocol); return false; } // Set the default value for the key, which specifies the icon associated with the protocol const std::wstring value_icon = L"\"" + executable_wide + L"\",0"; const LRESULT result_value_icon = RegSetValueExW(handle_subkey_icon, L"", 0, REG_SZ, (BYTE *)value_icon.c_str(), (value_icon.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_icon); if(result_value_icon != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error setting registry value", result_value_icon); RegCloseKey(handle_subkey_protocol); return false; } // Create the "shell\open\command" subkeys HKEY handle_subkey_shell_open_command; const LRESULT result_subkey_shell_open_command = RegCreateKeyExW(handle_subkey_protocol, L"shell\\open\\command", 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_shell_open_command, NULL); RegCloseKey(handle_subkey_protocol); if(result_subkey_shell_open_command != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error creating registry key", result_subkey_shell_open_command); return false; } // Get the previous default value for the key, so we can determine if it changed wchar_t old_value_executable[MAX_PATH + 16]; DWORD old_size_executable = sizeof(old_value_executable); const LRESULT result_old_value_executable = RegGetValueW(handle_subkey_shell_open_command, NULL, L"", RRF_RT_REG_SZ, NULL, (BYTE *)old_value_executable, &old_size_executable); const std::wstring value_executable = L"\"" + executable_wide + L"\" \"%1\""; if(result_old_value_executable != ERROR_SUCCESS || wcscmp(old_value_executable, value_executable.c_str()) != 0) { // Set the default value for the key, which specifies the executable command associated with the protocol const LRESULT result_value_executable = RegSetValueExW(handle_subkey_shell_open_command, L"", 0, REG_SZ, (BYTE *)value_executable.c_str(), (value_executable.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_shell_open_command); if(result_value_executable != ERROR_SUCCESS) { windows_print_error("shell_register_protocol", "Error setting registry value", result_value_executable); return false; } *updated = true; } else { RegCloseKey(handle_subkey_shell_open_command); } return true; } bool shell_register_extension(const char *extension, const char *description, const char *executable_name, const char *executable, bool *updated) { const std::wstring extension_wide = windows_utf8_to_wide(extension); const std::wstring executable_name_wide = windows_utf8_to_wide(executable_name); const std::wstring description_wide = executable_name_wide + L" " + windows_utf8_to_wide(description); const std::wstring program_id_wide = executable_name_wide + extension_wide; const std::wstring executable_wide = windows_utf8_to_wide(executable); // Open registry key for file associations of the current user HKEY handle_subkey_classes; const LRESULT result_subkey_classes = RegOpenKeyExW(HKEY_CURRENT_USER, L"SOFTWARE\\Classes", 0, KEY_ALL_ACCESS, &handle_subkey_classes); if(result_subkey_classes != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error opening registry key", result_subkey_classes); return false; } // Create the program ID key HKEY handle_subkey_program_id; const LRESULT result_subkey_program_id = RegCreateKeyExW(handle_subkey_classes, program_id_wide.c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_program_id, NULL); if(result_subkey_program_id != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error creating registry key", result_subkey_program_id); RegCloseKey(handle_subkey_classes); return false; } // Set the default value for the key, which specifies the file type description for legacy applications const LRESULT result_description_default = RegSetValueExW(handle_subkey_program_id, L"", 0, REG_SZ, (BYTE *)description_wide.c_str(), (description_wide.length() + 1) * sizeof(wchar_t)); if(result_description_default != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error setting registry value", result_description_default); RegCloseKey(handle_subkey_program_id); RegCloseKey(handle_subkey_classes); return false; } // Set the "FriendlyTypeName" value, which specifies the file type description for modern applications const LRESULT result_description_friendly = RegSetValueExW(handle_subkey_program_id, L"FriendlyTypeName", 0, REG_SZ, (BYTE *)description_wide.c_str(), (description_wide.length() + 1) * sizeof(wchar_t)); if(result_description_friendly != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error setting registry value", result_description_friendly); RegCloseKey(handle_subkey_program_id); RegCloseKey(handle_subkey_classes); return false; } // Create the "DefaultIcon" subkey HKEY handle_subkey_icon; const LRESULT result_subkey_icon = RegCreateKeyExW(handle_subkey_program_id, L"DefaultIcon", 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_icon, NULL); if(result_subkey_icon != ERROR_SUCCESS) { windows_print_error("register_protocol", "Error creating registry key", result_subkey_icon); RegCloseKey(handle_subkey_program_id); RegCloseKey(handle_subkey_classes); return false; } // Set the default value for the key, which specifies the icon associated with the program ID const std::wstring value_icon = L"\"" + executable_wide + L"\",0"; const LRESULT result_value_icon = RegSetValueExW(handle_subkey_icon, L"", 0, REG_SZ, (BYTE *)value_icon.c_str(), (value_icon.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_icon); if(result_value_icon != ERROR_SUCCESS) { windows_print_error("register_protocol", "Error setting registry value", result_value_icon); RegCloseKey(handle_subkey_program_id); RegCloseKey(handle_subkey_classes); return false; } // Create the "shell\open\command" subkeys HKEY handle_subkey_shell_open_command; const LRESULT result_subkey_shell_open_command = RegCreateKeyExW(handle_subkey_program_id, L"shell\\open\\command", 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_shell_open_command, NULL); RegCloseKey(handle_subkey_program_id); if(result_subkey_shell_open_command != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error creating registry key", result_subkey_shell_open_command); RegCloseKey(handle_subkey_classes); return false; } // Get the previous default value for the key, so we can determine if it changed wchar_t old_value_executable[MAX_PATH + 16]; DWORD old_size_executable = sizeof(old_value_executable); const LRESULT result_old_value_executable = RegGetValueW(handle_subkey_shell_open_command, NULL, L"", RRF_RT_REG_SZ, NULL, (BYTE *)old_value_executable, &old_size_executable); const std::wstring value_executable = L"\"" + executable_wide + L"\" \"%1\""; if(result_old_value_executable != ERROR_SUCCESS || wcscmp(old_value_executable, value_executable.c_str()) != 0) { // Set the default value for the key, which specifies the executable command associated with the application const LRESULT result_value_executable = RegSetValueExW(handle_subkey_shell_open_command, L"", 0, REG_SZ, (BYTE *)value_executable.c_str(), (value_executable.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_shell_open_command); if(result_value_executable != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error setting registry value", result_value_executable); RegCloseKey(handle_subkey_classes); return false; } *updated = true; } // Create the file extension key HKEY handle_subkey_extension; const LRESULT result_subkey_extension = RegCreateKeyExW(handle_subkey_classes, extension_wide.c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_extension, NULL); RegCloseKey(handle_subkey_classes); if(result_subkey_extension != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error creating registry key", result_subkey_extension); return false; } // Get the previous default value for the key, so we can determine if it changed wchar_t old_value_application[128]; DWORD old_size_application = sizeof(old_value_application); const LRESULT result_old_value_application = RegGetValueW(handle_subkey_extension, NULL, L"", RRF_RT_REG_SZ, NULL, (BYTE *)old_value_application, &old_size_application); if(result_old_value_application != ERROR_SUCCESS || wcscmp(old_value_application, program_id_wide.c_str()) != 0) { // Set the default value for the key, which associates the file extension with the program ID const LRESULT result_value_application = RegSetValueExW(handle_subkey_extension, L"", 0, REG_SZ, (BYTE *)program_id_wide.c_str(), (program_id_wide.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_extension); if(result_value_application != ERROR_SUCCESS) { windows_print_error("shell_register_extension", "Error setting registry value", result_value_application); return false; } *updated = true; } else { RegCloseKey(handle_subkey_extension); } return true; } bool shell_register_application(const char *name, const char *executable, bool *updated) { const std::wstring name_wide = windows_utf8_to_wide(name); const std::wstring executable_filename = filename_from_path(windows_utf8_to_wide(executable)); // Open registry key for application registrations HKEY handle_subkey_applications; const LRESULT result_subkey_applications = RegOpenKeyExW(HKEY_CURRENT_USER, L"SOFTWARE\\Classes\\Applications", 0, KEY_ALL_ACCESS, &handle_subkey_applications); if(result_subkey_applications != ERROR_SUCCESS) { windows_print_error("shell_register_application", "Error opening registry key", result_subkey_applications); return false; } // Create the program key HKEY handle_subkey_program; const LRESULT result_subkey_program = RegCreateKeyExW(handle_subkey_applications, executable_filename.c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &handle_subkey_program, NULL); RegCloseKey(handle_subkey_applications); if(result_subkey_program != ERROR_SUCCESS) { windows_print_error("shell_register_application", "Error creating registry key", result_subkey_program); return false; } // Get the previous default value for the key, so we can determine if it changed wchar_t old_value_executable[MAX_PATH]; DWORD old_size_executable = sizeof(old_value_executable); const LRESULT result_old_value_executable = RegGetValueW(handle_subkey_program, NULL, L"FriendlyAppName", RRF_RT_REG_SZ, NULL, (BYTE *)old_value_executable, &old_size_executable); if(result_old_value_executable != ERROR_SUCCESS || wcscmp(old_value_executable, name_wide.c_str()) != 0) { // Set the "FriendlyAppName" value, which specifies the displayed name of the application const LRESULT result_program_name = RegSetValueExW(handle_subkey_program, L"FriendlyAppName", 0, REG_SZ, (BYTE *)name_wide.c_str(), (name_wide.length() + 1) * sizeof(wchar_t)); RegCloseKey(handle_subkey_program); if(result_program_name != ERROR_SUCCESS) { windows_print_error("shell_register_application", "Error setting registry value", result_program_name); return false; } *updated = true; } else { RegCloseKey(handle_subkey_program); } return true; } bool shell_unregister_class(const char *shell_class, bool *updated) { const std::wstring class_wide = windows_utf8_to_wide(shell_class); // Open registry key for protocol and file associations of the current user HKEY handle_subkey_classes; const LRESULT result_subkey_classes = RegOpenKeyExW(HKEY_CURRENT_USER, L"SOFTWARE\\Classes", 0, KEY_ALL_ACCESS, &handle_subkey_classes); if(result_subkey_classes != ERROR_SUCCESS) { windows_print_error("shell_unregister_class", "Error opening registry key", result_subkey_classes); return false; } // Delete the registry keys for the shell class (protocol or program ID) LRESULT result_delete = RegDeleteTreeW(handle_subkey_classes, class_wide.c_str()); RegCloseKey(handle_subkey_classes); if(result_delete == ERROR_SUCCESS) { *updated = true; } else if(result_delete != ERROR_FILE_NOT_FOUND) { windows_print_error("shell_unregister_class", "Error deleting registry key", result_delete); return false; } return true; } bool shell_unregister_application(const char *executable, bool *updated) { const std::wstring executable_filename = filename_from_path(windows_utf8_to_wide(executable)); // Open registry key for application registrations HKEY handle_subkey_applications; const LRESULT result_subkey_applications = RegOpenKeyExW(HKEY_CURRENT_USER, L"SOFTWARE\\Classes\\Applications", 0, KEY_ALL_ACCESS, &handle_subkey_applications); if(result_subkey_applications != ERROR_SUCCESS) { windows_print_error("shell_unregister_application", "Error opening registry key", result_subkey_applications); return false; } // Delete the registry keys for the application description LRESULT result_delete = RegDeleteTreeW(handle_subkey_applications, executable_filename.c_str()); RegCloseKey(handle_subkey_applications); if(result_delete == ERROR_SUCCESS) { *updated = true; } else if(result_delete != ERROR_FILE_NOT_FOUND) { windows_print_error("shell_unregister_application", "Error deleting registry key", result_delete); return false; } return true; } void shell_update() { SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL); } #endif size_t std::hash::operator()(const NETADDR &Addr) const noexcept { return std::hash{}(std::string_view((const char *)&Addr, sizeof(Addr))); }