You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
236 lines
9.5 KiB
236 lines
9.5 KiB
/*
|
|
* utils.h - Misc utilities
|
|
*
|
|
* Copyright (C) 2013 - 2018, Max Lv <max.c.lv@gmail.com>
|
|
*
|
|
* 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
|
|
* <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef _UTILS_H
|
|
#define _UTILS_H
|
|
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
|
|
#define PORTSTRLEN 16
|
|
#define SS_ADDRSTRLEN (INET6_ADDRSTRLEN + PORTSTRLEN + 1)
|
|
|
|
#ifdef __ANDROID__
|
|
|
|
#include <android/log.h>
|
|
#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__); \
|
|
} \
|
|
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(stdout, " %s ERROR: ", timestr); \
|
|
ss_color_reset(); \
|
|
fprintf(stdout, format "\n", ## __VA_ARGS__); \
|
|
} \
|
|
while (0)
|
|
|
|
|
|
#else // not __MINGW32__
|
|
|
|
#include <syslog.h>
|
|
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__); \
|
|
} else { \
|
|
fprintf(stdout, " %s INFO: " format "\n", timestr, \
|
|
## __VA_ARGS__); \
|
|
} \
|
|
} \
|
|
} \
|
|
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__); \
|
|
} else { \
|
|
fprintf(stderr, " %s ERROR: " format "\n", timestr, \
|
|
## __VA_ARGS__); \
|
|
} \
|
|
} } \
|
|
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_align(size_t size);
|
|
void *ss_realloc(void *ptr, size_t new_size);
|
|
|
|
#define ss_free(ptr) \
|
|
do { \
|
|
free(ptr); \
|
|
ptr = NULL; \
|
|
} while (0)
|
|
|
|
char *get_default_conf(void);
|
|
|
|
#endif // _UTILS_H
|