/* * utils.h - Misc utilities * * Copyright (C) 2013 - 2019, Max Lv * * This file is part of the shadowsocks-libev. * * shadowsocks-libev is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * shadowsocks-libev is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with shadowsocks-libev; see the file COPYING. If not, see * . */ #ifndef _UTILS_H #define _UTILS_H #include #include #include #include #include #define PORTSTRLEN 16 #define SS_ADDRSTRLEN (INET6_ADDRSTRLEN + PORTSTRLEN + 1) #ifdef __ANDROID__ #include #define USE_TTY() #define USE_SYSLOG(ident, _cond) #define LOGI(...) \ ((void)__android_log_print(ANDROID_LOG_DEBUG, "shadowsocks", \ __VA_ARGS__)) #define LOGE(...) \ ((void)__android_log_print(ANDROID_LOG_ERROR, "shadowsocks", \ __VA_ARGS__)) #else // not __ANDROID__ #define STR(x) # x #define TOSTR(x) STR(x) #ifdef LIB_ONLY extern FILE *logfile; #define TIME_FORMAT "%Y-%m-%d %H:%M:%S" #define USE_TTY() #define USE_SYSLOG(ident, _cond) #define USE_LOGFILE(ident) \ do { \ if (ident != NULL) { logfile = fopen(ident, "w+"); } } \ while (0) #define CLOSE_LOGFILE \ do { \ if (logfile != NULL) { fclose(logfile); } } \ while (0) #define LOGI(format, ...) \ do { \ if (logfile != NULL) { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ fprintf(logfile, " %s INFO: " format "\n", timestr, ## __VA_ARGS__); \ fflush(logfile); } \ } \ while (0) #define LOGE(format, ...) \ do { \ if (logfile != NULL) { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ fprintf(logfile, " %s ERROR: " format "\n", timestr, \ ## __VA_ARGS__); \ fflush(logfile); } \ } \ while (0) #else // not LIB_ONLY #ifdef __MINGW32__ #define USE_TTY() #define USE_SYSLOG(ident, _cond) #define USE_LOGFILE(ident) #define TIME_FORMAT "%Y-%m-%d %H:%M:%S" #define LOGI(format, ...) \ do { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ ss_color_info(); \ fprintf(stdout, " %s INFO: ", timestr); \ ss_color_reset(); \ fprintf(stdout, format "\n", ## __VA_ARGS__); \ fflush(stdout); \ } \ while (0) #define LOGE(format, ...) \ do { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ ss_color_error(); \ fprintf(stderr, " %s ERROR: ", timestr); \ ss_color_reset(); \ fprintf(stderr, format "\n", ## __VA_ARGS__); \ fflush(stderr); \ } \ while (0) #else // not __MINGW32__ #include extern int use_tty; extern int use_syslog; #define HAS_SYSLOG #define TIME_FORMAT "%F %T" #define USE_TTY() \ do { \ use_tty = isatty(STDERR_FILENO); \ } while (0) #define USE_SYSLOG(_ident, _cond) \ do { \ if (!use_syslog && (_cond)) { \ use_syslog = 1; \ } \ if (use_syslog) { \ openlog((_ident), LOG_CONS | LOG_PID, LOG_DAEMON); \ } \ } while (0) #define LOGI(format, ...) \ do { \ if (use_syslog) { \ syslog(LOG_INFO, format, ## __VA_ARGS__); \ } else { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ if (use_tty) { \ fprintf(stdout, "\e[01;32m %s INFO: \e[0m" format "\n", timestr, \ ## __VA_ARGS__); \ fflush(stdout); \ } else { \ fprintf(stdout, " %s INFO: " format "\n", timestr, \ ## __VA_ARGS__); \ fflush(stdout); \ } \ } \ } \ while (0) #define LOGE(format, ...) \ do { \ if (use_syslog) { \ syslog(LOG_ERR, format, ## __VA_ARGS__); \ } else { \ time_t now = time(NULL); \ char timestr[20]; \ strftime(timestr, 20, TIME_FORMAT, localtime(&now)); \ if (use_tty) { \ fprintf(stderr, "\e[01;35m %s ERROR: \e[0m" format "\n", timestr, \ ## __VA_ARGS__); \ fflush(stderr); \ } else { \ fprintf(stderr, " %s ERROR: " format "\n", timestr, \ ## __VA_ARGS__); \ fflush(stderr); \ } \ } } \ while (0) #endif // if __MINGW32__ #endif // if LIB_ONLY #endif // if __ANDROID__ // Workaround for "%z" in Windows printf #ifdef __MINGW32__ #define SSIZE_FMT "%Id" #define SIZE_FMT "%Iu" #else #define SSIZE_FMT "%zd" #define SIZE_FMT "%zu" #endif #ifdef __MINGW32__ // Override Windows built-in functions #ifdef ERROR #undef ERROR #endif #define ERROR(s) ss_error(s) // Implemented in winsock.c void ss_error(const char *s); void ss_color_info(void); void ss_color_error(void); void ss_color_reset(void); #else void ERROR(const char *s); #endif char *ss_itoa(int i); int ss_isnumeric(const char *s); int run_as(const char *user); void FATAL(const char *msg); void usage(void); void daemonize(const char *path); char *ss_strndup(const char *s, size_t n); #ifdef HAVE_SETRLIMIT int set_nofile(int nofile); #endif void *ss_malloc(size_t size); void *ss_aligned_malloc(size_t size); void *ss_realloc(void *ptr, size_t new_size); #define ss_free(ptr) \ { \ free(ptr); \ ptr = NULL; \ } #ifdef __MINGW32__ #define ss_aligned_free(ptr) \ { \ _aligned_free(ptr); \ ptr = NULL; \ } #else #define ss_aligned_free(ptr) ss_free(ptr) #endif int ss_is_ipv6addr(const char *addr); char *get_default_conf(void); uint16_t load16_be(const void *s); #endif // _UTILS_H