169 lines
4 KiB
C
169 lines
4 KiB
C
#include "config.h"
|
|
#include "packet.h"
|
|
#include "util.h"
|
|
|
|
/*
|
|
* will include error message from function if errno isn't 0
|
|
* end program is fatal is 1
|
|
*/
|
|
void error(int fatal, const char *fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
|
|
/* to preserve errno */
|
|
int errsv = errno;
|
|
|
|
/* Determine the length of the formatted error message */
|
|
va_list args_copy;
|
|
va_copy(args_copy, args);
|
|
size_t error_len = vsnprintf(NULL, 0, fmt, args_copy);
|
|
va_end(args_copy);
|
|
|
|
/* 7 for [zsm], space and null */
|
|
char errorstr[error_len + 1];
|
|
vsnprintf(errorstr, error_len + 1, fmt, args);
|
|
fprintf(stderr, "[zsm] ");
|
|
|
|
if (errsv != 0 && errsv != EEXIST) {
|
|
perror(errorstr);
|
|
errno = 0;
|
|
} else {
|
|
fprintf(stderr, "%s\n", errorstr);
|
|
}
|
|
|
|
va_end(args);
|
|
if (fatal) exit(1);
|
|
}
|
|
|
|
void *memalloc(size_t size)
|
|
{
|
|
void *ptr = malloc(size);
|
|
if (!ptr) {
|
|
write_log(LOG_ERROR, "Error allocating memory\n");
|
|
return NULL;
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
void *estrdup(void *str)
|
|
{
|
|
void *modstr = strdup(str);
|
|
if (modstr == NULL) {
|
|
write_log(LOG_ERROR, "Error allocating memory\n");
|
|
return NULL;
|
|
}
|
|
return modstr;
|
|
}
|
|
|
|
/*
|
|
* Takes heap-allocated str and replace ~ with home path
|
|
* Returns heap-allocated newstr
|
|
*/
|
|
char *replace_home(char *str)
|
|
{
|
|
char *home = getenv("HOME");
|
|
if (home == NULL) {
|
|
write_log(LOG_ERROR, "$HOME not defined\n");
|
|
return str;
|
|
}
|
|
char *newstr = memalloc(strlen(str) + strlen(home));
|
|
/* replace ~ with home */
|
|
snprintf(newstr, strlen(str) + strlen(home), "%s%s", home, str + 1);
|
|
free(str);
|
|
return newstr;
|
|
}
|
|
|
|
/*
|
|
* Recursively create directory by creating each subdirectory
|
|
* like mkdir -p
|
|
*/
|
|
void mkdir_p(const char *destdir)
|
|
{
|
|
char *path = memalloc(PATH_MAX);
|
|
char dir_path[PATH_MAX];
|
|
|
|
if (destdir[0] == '~') {
|
|
char *home = getenv("HOME");
|
|
if (home == NULL) {
|
|
write_log(LOG_ERROR, "$HOME not defined\n");
|
|
return;
|
|
}
|
|
/* replace ~ with home */
|
|
snprintf(path, PATH_MAX, "%s%s", home, destdir + 1);
|
|
} else {
|
|
strcpy(path, destdir);
|
|
}
|
|
|
|
/* fix first / not appearing in the string */
|
|
if (path[0] == '/')
|
|
strcat(dir_path, "/");
|
|
|
|
char *token = strtok(path, "/");
|
|
while (token != NULL) {
|
|
strcat(dir_path, token);
|
|
strcat(dir_path, "/");
|
|
|
|
if (mkdir(dir_path, 0755) == -1) {
|
|
struct stat st;
|
|
if (stat(dir_path, &st) == 0 && S_ISDIR(st.st_mode)) {
|
|
/* Directory already exists, continue to the next dir */
|
|
token = strtok(NULL, "/");
|
|
continue;
|
|
}
|
|
|
|
write_log(LOG_ERROR, "mkdir failed: %s\n", strerror(errno));
|
|
free(path);
|
|
return;
|
|
}
|
|
token = strtok(NULL, "/");
|
|
}
|
|
|
|
free(path);
|
|
return;
|
|
}
|
|
|
|
void create_data_dir()
|
|
{
|
|
char *client_data_dir = estrdup(CLIENT_DATA_DIR);
|
|
mkdir_p(client_data_dir);
|
|
client_data_dir = replace_home(client_data_dir);
|
|
free(client_data_dir);
|
|
}
|
|
|
|
void write_log(int type, const char *fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
char *client_log = memalloc(PATH_MAX);
|
|
snprintf(client_log, PATH_MAX, "%s/%s", CLIENT_DATA_DIR, "zen.log");
|
|
if (access(client_log, W_OK) != 0) {
|
|
/* If log file doesn't exist, most likely data dir won't exist too */
|
|
create_data_dir();
|
|
}
|
|
FILE *log = fopen(client_log, "a");
|
|
if (log != NULL) {
|
|
time_t now = time(NULL);
|
|
struct tm *t = localtime(&now);
|
|
/* either info or error */
|
|
int type_len = type == LOG_INFO ? 4 : 5;
|
|
char logtype[4 + type_len];
|
|
snprintf(logtype, 4 + type_len, "[%s] ", type == LOG_INFO ? "INFO" : "ERROR");
|
|
char time[21];
|
|
strftime(time, 22, "%Y-%m-%d %H:%M:%S ", t);
|
|
char details[2 + type_len + 22];
|
|
snprintf(details, 2 + type_len + 22, "%s%s", logtype, time);
|
|
fprintf(log, details);
|
|
vfprintf(log, fmt, args);
|
|
}
|
|
fclose(log);
|
|
va_end(args);
|
|
}
|
|
|
|
void print_bin(const uint8_t *ptr, size_t length)
|
|
{
|
|
for (size_t i = 0; i < length; i++) {
|
|
printf("%02x ", ptr[i]);
|
|
}
|
|
printf("\n");
|
|
}
|