/* * Copyright (c) 2013 Hugh Bailey * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include "obsconfig.h" #if !defined(__APPLE__) #include #include #include #ifdef __FreeBSD__ #include #include #include #include #include #include #include #else #include #endif #include #endif #include "darray.h" #include "dstr.h" #include "platform.h" #include "threading.h" void *os_dlopen(const char *path) { struct dstr dylib_name; if (!path) return NULL; dstr_init_copy(&dylib_name, path); #ifdef __APPLE__ if (!dstr_find(&dylib_name, ".so") && !dstr_find(&dylib_name, ".dylib")) #else if (!dstr_find(&dylib_name, ".so")) #endif dstr_cat(&dylib_name, ".so"); #ifdef __APPLE__ void *res = dlopen(dylib_name.array, RTLD_LAZY | RTLD_FIRST); #else void *res = dlopen(dylib_name.array, RTLD_LAZY); #endif if (!res) blog(LOG_ERROR, "os_dlopen(%s->%s): %s\n", path, dylib_name.array, dlerror()); dstr_free(&dylib_name); return res; } void *os_dlsym(void *module, const char *func) { return dlsym(module, func); } void os_dlclose(void *module) { if (module) dlclose(module); } #if !defined(__APPLE__) struct os_cpu_usage_info { clock_t last_cpu_time, last_sys_time, last_user_time; int core_count; }; os_cpu_usage_info_t *os_cpu_usage_info_start(void) { struct os_cpu_usage_info *info = bmalloc(sizeof(*info)); struct tms time_sample; info->last_cpu_time = times(&time_sample); info->last_sys_time = time_sample.tms_stime; info->last_user_time = time_sample.tms_utime; info->core_count = sysconf(_SC_NPROCESSORS_ONLN); return info; } double os_cpu_usage_info_query(os_cpu_usage_info_t *info) { struct tms time_sample; clock_t cur_cpu_time; double percent; if (!info) return 0.0; cur_cpu_time = times(&time_sample); if (cur_cpu_time <= info->last_cpu_time || time_sample.tms_stime < info->last_sys_time || time_sample.tms_utime < info->last_user_time) return 0.0; percent = (double)(time_sample.tms_stime - info->last_sys_time + (time_sample.tms_utime - info->last_user_time)); percent /= (double)(cur_cpu_time - info->last_cpu_time); percent /= (double)info->core_count; info->last_cpu_time = cur_cpu_time; info->last_sys_time = time_sample.tms_stime; info->last_user_time = time_sample.tms_utime; return percent * 100.0; } void os_cpu_usage_info_destroy(os_cpu_usage_info_t *info) { if (info) bfree(info); } #endif bool os_sleepto_ns(uint64_t time_target) { uint64_t current = os_gettime_ns(); if (time_target < current) return false; time_target -= current; struct timespec req, remain; memset(&req, 0, sizeof(req)); memset(&remain, 0, sizeof(remain)); req.tv_sec = time_target / 1000000000; req.tv_nsec = time_target % 1000000000; while (nanosleep(&req, &remain)) { req = remain; memset(&remain, 0, sizeof(remain)); } return true; } void os_sleep_ms(uint32_t duration) { usleep(duration * 1000); } #if !defined(__APPLE__) uint64_t os_gettime_ns(void) { struct timespec ts; clock_gettime(CLOCK_MONOTONIC, &ts); return ((uint64_t)ts.tv_sec * 1000000000ULL + (uint64_t)ts.tv_nsec); } /* should return $HOME/.[name], or when using XDG, * should return $HOME/.config/[name] as default */ int os_get_config_path(char *dst, size_t size, const char *name) { #ifdef USE_XDG char *xdg_ptr = getenv("XDG_CONFIG_HOME"); // If XDG_CONFIG_HOME is unset, // we use the default $HOME/.config/[name] instead if (xdg_ptr == NULL) { char *home_ptr = getenv("HOME"); if (home_ptr == NULL) bcrash("Could not get $HOME\n"); if (!name || !*name) { return snprintf(dst, size, "%s/.config", home_ptr); } else { return snprintf(dst, size, "%s/.config/%s", home_ptr, name); } } else { if (!name || !*name) return snprintf(dst, size, "%s", xdg_ptr); else return snprintf(dst, size, "%s/%s", xdg_ptr, name); } #else char *path_ptr = getenv("HOME"); if (path_ptr == NULL) bcrash("Could not get $HOME\n"); if (!name || !*name) return snprintf(dst, size, "%s", path_ptr); else return snprintf(dst, size, "%s/.%s", path_ptr, name); #endif } /* should return $HOME/.[name], or when using XDG, * should return $HOME/.config/[name] as default */ char *os_get_config_path_ptr(const char *name) { #ifdef USE_XDG struct dstr path; char *xdg_ptr = getenv("XDG_CONFIG_HOME"); /* If XDG_CONFIG_HOME is unset, * we use the default $HOME/.config/[name] instead */ if (xdg_ptr == NULL) { char *home_ptr = getenv("HOME"); if (home_ptr == NULL) bcrash("Could not get $HOME\n"); dstr_init_copy(&path, home_ptr); dstr_cat(&path, "/.config/"); dstr_cat(&path, name); } else { dstr_init_copy(&path, xdg_ptr); dstr_cat(&path, "/"); dstr_cat(&path, name); } return path.array; #else char *path_ptr = getenv("HOME"); if (path_ptr == NULL) bcrash("Could not get $HOME\n"); struct dstr path; dstr_init_copy(&path, path_ptr); dstr_cat(&path, "/."); dstr_cat(&path, name); return path.array; #endif } int os_get_program_data_path(char *dst, size_t size, const char *name) { return snprintf(dst, size, "/usr/local/share/%s", !!name ? name : ""); } char *os_get_program_data_path_ptr(const char *name) { size_t len = snprintf(NULL, 0, "/usr/local/share/%s", !!name ? name : ""); char *str = bmalloc(len + 1); snprintf(str, len + 1, "/usr/local/share/%s", !!name ? name : ""); str[len] = 0; return str; } char *os_get_executable_path_ptr(const char *name) { char exe[PATH_MAX]; #if defined(__FreeBSD__) || defined(__DragonFly__) int sysctlname[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1}; size_t pathlen = PATH_MAX; ssize_t count; if (sysctl(sysctlname, nitems(sysctlname), exe, &pathlen, NULL, 0) == -1) { blog(LOG_ERROR, "sysctl(KERN_PROC_PATHNAME) failed, errno %d", errno); return NULL; } count = pathlen; #else ssize_t count = readlink("/proc/self/exe", exe, PATH_MAX - 1); if (count >= 0) { exe[count] = '\0'; } #endif const char *path_out = NULL; struct dstr path; if (count == -1) { return NULL; } path_out = dirname(exe); if (!path_out) { return NULL; } dstr_init_copy(&path, path_out); dstr_cat(&path, "/"); if (name && *name) { dstr_cat(&path, name); } return path.array; } #endif bool os_file_exists(const char *path) { return access(path, F_OK) == 0; } size_t os_get_abs_path(const char *path, char *abspath, size_t size) { size_t min_size = size < PATH_MAX ? size : PATH_MAX; char newpath[PATH_MAX]; int ret; if (!abspath) return 0; if (!realpath(path, newpath)) return 0; ret = snprintf(abspath, min_size, "%s", newpath); return ret >= 0 ? ret : 0; } char *os_get_abs_path_ptr(const char *path) { char *ptr = bmalloc(512); if (!os_get_abs_path(path, ptr, 512)) { bfree(ptr); ptr = NULL; } return ptr; } struct os_dir { const char *path; DIR *dir; struct dirent *cur_dirent; struct os_dirent out; }; os_dir_t *os_opendir(const char *path) { struct os_dir *dir; DIR *dir_val; dir_val = opendir(path); if (!dir_val) return NULL; dir = bzalloc(sizeof(struct os_dir)); dir->dir = dir_val; dir->path = path; return dir; } static inline bool is_dir(const char *path) { struct stat stat_info; if (stat(path, &stat_info) == 0) return !!S_ISDIR(stat_info.st_mode); blog(LOG_DEBUG, "is_dir: stat for %s failed, errno: %d", path, errno); return false; } struct os_dirent *os_readdir(os_dir_t *dir) { struct dstr file_path = {0}; if (!dir) return NULL; dir->cur_dirent = readdir(dir->dir); if (!dir->cur_dirent) return NULL; strncpy(dir->out.d_name, dir->cur_dirent->d_name, 255); dstr_copy(&file_path, dir->path); dstr_cat(&file_path, "/"); dstr_cat(&file_path, dir->out.d_name); dir->out.directory = is_dir(file_path.array); dstr_free(&file_path); return &dir->out; } void os_closedir(os_dir_t *dir) { if (dir) { closedir(dir->dir); bfree(dir); } } int64_t os_get_free_space(const char *path) { struct statvfs info; int64_t ret = (int64_t)statvfs(path, &info); if (ret == 0) ret = (int64_t)info.f_bsize * (int64_t)info.f_bfree; return ret; } struct posix_glob_info { struct os_glob_info base; glob_t gl; }; int os_glob(const char *pattern, int flags, os_glob_t **pglob) { struct posix_glob_info pgi; int ret = glob(pattern, 0, NULL, &pgi.gl); if (ret == 0) { DARRAY(struct os_globent) list; da_init(list); for (size_t i = 0; i < pgi.gl.gl_pathc; i++) { struct os_globent ent = {0}; ent.path = pgi.gl.gl_pathv[i]; ent.directory = is_dir(ent.path); da_push_back(list, &ent); } pgi.base.gl_pathc = list.num; pgi.base.gl_pathv = list.array; *pglob = bmemdup(&pgi, sizeof(pgi)); } else { *pglob = NULL; } UNUSED_PARAMETER(flags); return ret; } void os_globfree(os_glob_t *pglob) { if (pglob) { struct posix_glob_info *pgi = (struct posix_glob_info *)pglob; globfree(&pgi->gl); bfree(pgi->base.gl_pathv); bfree(pgi); } } int os_unlink(const char *path) { return unlink(path); } int os_rmdir(const char *path) { return rmdir(path); } int os_mkdir(const char *path) { if (mkdir(path, 0755) == 0) return MKDIR_SUCCESS; return (errno == EEXIST) ? MKDIR_EXISTS : MKDIR_ERROR; } int os_rename(const char *old_path, const char *new_path) { return rename(old_path, new_path); } int os_safe_replace(const char *target, const char *from, const char *backup) { if (backup && os_file_exists(target) && rename(target, backup) != 0) return -1; return rename(from, target); } #if !defined(__APPLE__) os_performance_token_t *os_request_high_performance(const char *reason) { UNUSED_PARAMETER(reason); return NULL; } void os_end_high_performance(os_performance_token_t *token) { UNUSED_PARAMETER(token); } #endif int os_copyfile(const char *file_path_in, const char *file_path_out) { FILE *file_out = NULL; FILE *file_in = NULL; uint8_t data[4096]; int ret = -1; size_t size; if (os_file_exists(file_path_out)) return -1; file_in = fopen(file_path_in, "rb"); if (!file_in) return -1; file_out = fopen(file_path_out, "ab+"); if (!file_out) goto error; do { size = fread(data, 1, sizeof(data), file_in); if (size) size = fwrite(data, 1, size, file_out); } while (size == sizeof(data)); ret = feof(file_in) ? 0 : -1; error: if (file_out) fclose(file_out); fclose(file_in); return ret; } char *os_getcwd(char *path, size_t size) { return getcwd(path, size); } int os_chdir(const char *path) { return chdir(path); } #if !defined(__APPLE__) #if HAVE_DBUS struct dbus_sleep_info; extern struct dbus_sleep_info *dbus_sleep_info_create(void); extern void dbus_inhibit_sleep(struct dbus_sleep_info *dbus, const char *sleep, bool active); extern void dbus_sleep_info_destroy(struct dbus_sleep_info *dbus); #endif struct os_inhibit_info { #if HAVE_DBUS struct dbus_sleep_info *dbus; #endif pthread_t screensaver_thread; os_event_t *stop_event; char *reason; posix_spawnattr_t attr; bool active; }; os_inhibit_t *os_inhibit_sleep_create(const char *reason) { struct os_inhibit_info *info = bzalloc(sizeof(*info)); sigset_t set; #if HAVE_DBUS info->dbus = dbus_sleep_info_create(); #endif os_event_init(&info->stop_event, OS_EVENT_TYPE_AUTO); posix_spawnattr_init(&info->attr); sigemptyset(&set); posix_spawnattr_setsigmask(&info->attr, &set); sigaddset(&set, SIGPIPE); posix_spawnattr_setsigdefault(&info->attr, &set); posix_spawnattr_setflags(&info->attr, POSIX_SPAWN_SETSIGDEF | POSIX_SPAWN_SETSIGMASK); info->reason = bstrdup(reason); return info; } extern char **environ; static void reset_screensaver(os_inhibit_t *info) { char *argv[3] = {(char *)"xdg-screensaver", (char *)"reset", NULL}; pid_t pid; int err = posix_spawnp(&pid, "xdg-screensaver", NULL, &info->attr, argv, environ); if (err == 0) { int status; while (waitpid(pid, &status, 0) == -1) ; } else { blog(LOG_WARNING, "Failed to create xdg-screensaver: %d", err); } } static void *screensaver_thread(void *param) { os_inhibit_t *info = param; while (os_event_timedwait(info->stop_event, 30000) == ETIMEDOUT) reset_screensaver(info); return NULL; } bool os_inhibit_sleep_set_active(os_inhibit_t *info, bool active) { int ret; if (!info) return false; if (info->active == active) return false; #if HAVE_DBUS if (info->dbus) dbus_inhibit_sleep(info->dbus, info->reason, active); #endif if (!info->stop_event) return true; if (active) { ret = pthread_create(&info->screensaver_thread, NULL, &screensaver_thread, info); if (ret < 0) { blog(LOG_ERROR, "Failed to create screensaver " "inhibitor thread"); return false; } } else { os_event_signal(info->stop_event); pthread_join(info->screensaver_thread, NULL); } info->active = active; return true; } void os_inhibit_sleep_destroy(os_inhibit_t *info) { if (info) { os_inhibit_sleep_set_active(info, false); #if HAVE_DBUS dbus_sleep_info_destroy(info->dbus); #endif os_event_destroy(info->stop_event); posix_spawnattr_destroy(&info->attr); bfree(info->reason); bfree(info); } } #endif void os_breakpoint() { raise(SIGTRAP); } #ifndef __APPLE__ static int physical_cores = 0; static int logical_cores = 0; static bool core_count_initialized = false; static void os_get_cores_internal(void) { if (core_count_initialized) return; core_count_initialized = true; logical_cores = sysconf(_SC_NPROCESSORS_ONLN); #if defined(__linux__) int physical_id = -1; int last_physical_id = -1; int core_count = 0; char *line = NULL; size_t linecap = 0; FILE *fp; struct dstr proc_phys_id; struct dstr proc_phys_ids; fp = fopen("/proc/cpuinfo", "r"); if (!fp) return; dstr_init(&proc_phys_id); dstr_init(&proc_phys_ids); while (getline(&line, &linecap, fp) != -1) { if (!strncmp(line, "physical id", 11)) { char *start = strchr(line, ':'); if (!start || *(++start) == '\0') continue; physical_id = atoi(start); dstr_free(&proc_phys_id); dstr_init(&proc_phys_id); dstr_catf(&proc_phys_id, "%d", physical_id); } if (!strncmp(line, "cpu cores", 9)) { char *start = strchr(line, ':'); if (!start || *(++start) == '\0') continue; if (dstr_is_empty(&proc_phys_ids) || (!dstr_is_empty(&proc_phys_ids) && !dstr_find(&proc_phys_ids, proc_phys_id.array))) { dstr_cat_dstr(&proc_phys_ids, &proc_phys_id); dstr_cat(&proc_phys_ids, " "); core_count += atoi(start); } } if (*line == '\n' && physical_id != last_physical_id) { last_physical_id = physical_id; } } if (core_count == 0) physical_cores = logical_cores; else physical_cores = core_count; fclose(fp); dstr_free(&proc_phys_ids); dstr_free(&proc_phys_id); free(line); #elif defined(__FreeBSD__) char *text = os_quick_read_utf8_file("/var/run/dmesg.boot"); char *core_count = text; int packages = 0; int cores = 0; struct dstr proc_packages; struct dstr proc_cores; dstr_init(&proc_packages); dstr_init(&proc_cores); if (!text || !*text) { physical_cores = logical_cores; return; } core_count = strstr(core_count, "\nFreeBSD/SMP: "); if (!core_count) goto FreeBSD_cores_cleanup; core_count++; core_count = strstr(core_count, "\nFreeBSD/SMP: "); if (!core_count) goto FreeBSD_cores_cleanup; core_count = strstr(core_count, ": "); core_count += 2; size_t len = strcspn(core_count, " "); dstr_ncopy(&proc_packages, core_count, len); core_count = strstr(core_count, "package(s) x "); if (!core_count) goto FreeBSD_cores_cleanup; core_count += 13; len = strcspn(core_count, " "); dstr_ncopy(&proc_cores, core_count, len); FreeBSD_cores_cleanup: if (!dstr_is_empty(&proc_packages)) packages = atoi(proc_packages.array); if (!dstr_is_empty(&proc_cores)) cores = atoi(proc_cores.array); if (packages == 0) physical_cores = logical_cores; else if (cores == 0) physical_cores = packages; else physical_cores = packages * cores; dstr_free(&proc_cores); dstr_free(&proc_packages); bfree(text); #else physical_cores = logical_cores; #endif } int os_get_physical_cores(void) { if (!core_count_initialized) os_get_cores_internal(); return physical_cores; } int os_get_logical_cores(void) { if (!core_count_initialized) os_get_cores_internal(); return logical_cores; } #ifdef __FreeBSD__ uint64_t os_get_sys_free_size(void) { uint64_t mem_free = 0; size_t length = sizeof(mem_free); if (sysctlbyname("vm.stats.vm.v_free_count", &mem_free, &length, NULL, 0) < 0) return 0; return mem_free; } static inline bool os_get_proc_memory_usage_internal(struct kinfo_proc *kinfo) { int mib[] = {CTL_KERN, KERN_PROC, KERN_PROC_PID, getpid()}; size_t length = sizeof(*kinfo); if (sysctl(mib, sizeof(mib) / sizeof(mib[0]), kinfo, &length, NULL, 0) < 0) return false; return true; } bool os_get_proc_memory_usage(os_proc_memory_usage_t *usage) { struct kinfo_proc kinfo; if (!os_get_proc_memory_usage_internal(&kinfo)) return false; usage->resident_size = (uint64_t)kinfo.ki_rssize * sysconf(_SC_PAGESIZE); usage->virtual_size = (uint64_t)kinfo.ki_size; return true; } uint64_t os_get_proc_resident_size(void) { struct kinfo_proc kinfo; if (!os_get_proc_memory_usage_internal(&kinfo)) return 0; return (uint64_t)kinfo.ki_rssize * sysconf(_SC_PAGESIZE); } uint64_t os_get_proc_virtual_size(void) { struct kinfo_proc kinfo; if (!os_get_proc_memory_usage_internal(&kinfo)) return 0; return (uint64_t)kinfo.ki_size; } #else uint64_t os_get_sys_free_size(void) { return 0; } typedef struct { unsigned long virtual_size; unsigned long resident_size; unsigned long share_pages; unsigned long text; unsigned long library; unsigned long data; unsigned long dirty_pages; } statm_t; static inline bool os_get_proc_memory_usage_internal(statm_t *statm) { const char *statm_path = "/proc/self/statm"; FILE *f = fopen(statm_path, "r"); if (!f) return false; if (fscanf(f, "%lu %lu %lu %lu %lu %lu %lu", &statm->virtual_size, &statm->resident_size, &statm->share_pages, &statm->text, &statm->library, &statm->data, &statm->dirty_pages) != 7) { fclose(f); return false; } fclose(f); return true; } bool os_get_proc_memory_usage(os_proc_memory_usage_t *usage) { statm_t statm = {}; if (!os_get_proc_memory_usage_internal(&statm)) return false; usage->resident_size = statm.resident_size; usage->virtual_size = statm.virtual_size; return true; } uint64_t os_get_proc_resident_size(void) { statm_t statm = {}; if (!os_get_proc_memory_usage_internal(&statm)) return 0; return (uint64_t)statm.resident_size; } uint64_t os_get_proc_virtual_size(void) { statm_t statm = {}; if (!os_get_proc_memory_usage_internal(&statm)) return 0; return (uint64_t)statm.virtual_size; } #endif #endif uint64_t os_get_free_disk_space(const char *dir) { struct statvfs info; if (statvfs(dir, &info) != 0) return 0; return (uint64_t)info.f_frsize * (uint64_t)info.f_bavail; }