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.
146 lines
2.9 KiB
146 lines
2.9 KiB
/* Copyright 2016 Alex 'AdUser' Z (ad_user@runbox.com) |
|
* |
|
* This program is free software; you can redistribute it and/or modify |
|
* it under the terms of the GNU General Public License version 2 as |
|
* published by the Free Software Foundation. |
|
*/ |
|
#include "source.h" |
|
|
|
#include <limits.h> |
|
#include <sys/file.h> |
|
#include <sys/stat.h> |
|
|
|
#include <glob.h> |
|
|
|
typedef struct f2b_file_t { |
|
struct f2b_file_t *next; |
|
bool opened; |
|
char path[PATH_MAX]; |
|
FILE *fd; |
|
struct stat st; |
|
} f2b_file_t; |
|
|
|
struct _config { |
|
char error[256]; |
|
f2b_file_t *files; |
|
}; |
|
|
|
static bool |
|
file_open(f2b_file_t *file, const char *path) { |
|
struct stat st; |
|
char buf[PATH_MAX] = ""; |
|
|
|
assert(file != NULL); |
|
assert(path != NULL || file->path[0] != '\0'); |
|
|
|
strlcpy(buf, path ? path : file->path, sizeof(buf)); |
|
|
|
memset(file, 0x0, sizeof(f2b_file_t)); |
|
|
|
if (stat(buf, &st) != 0) |
|
return false; |
|
|
|
if (!(S_ISREG(st.st_mode) || S_ISFIFO(st.st_mode))) |
|
return false; |
|
|
|
if ((file->fd = fopen(buf, "r")) == NULL) |
|
return false; |
|
|
|
if (S_ISREG(st.st_mode) && fseek(file->fd, 0, SEEK_END) < 0) |
|
return false; |
|
|
|
memcpy(&file->st, &st, sizeof(st)); |
|
strlcpy(file->path, buf, sizeof(file->path)); |
|
file->opened = true; |
|
|
|
return true; |
|
} |
|
|
|
static void |
|
file_close(f2b_file_t *file) { |
|
assert(file != NULL); |
|
|
|
if (file->fd) |
|
fclose(file->fd); |
|
|
|
file->opened = false; |
|
file->fd = NULL; |
|
} |
|
|
|
static bool |
|
file_rotated(const f2b_file_t *file) { |
|
struct stat st; |
|
|
|
assert(file != NULL); |
|
|
|
if (!file->opened) |
|
return true; |
|
|
|
if (stat(file->path, &st) != 0) |
|
return true; |
|
|
|
if (file->st.st_dev != st.st_dev || |
|
file->st.st_ino != st.st_ino || |
|
file->st.st_size > st.st_size) |
|
return true; |
|
|
|
return false; |
|
} |
|
|
|
static bool |
|
file_getline(const f2b_file_t *file, char *buf, size_t bufsize) { |
|
assert(file != NULL); |
|
assert(buf != NULL); |
|
|
|
if (feof(file->fd)) |
|
clearerr(file->fd); |
|
/* fread()+EOF set is implementation defined */ |
|
if (fgets(buf, bufsize, file->fd) != NULL) |
|
return true; |
|
|
|
return false; |
|
} |
|
|
|
static f2b_file_t * |
|
list_from_glob(const char *pattern) { |
|
f2b_file_t *file = NULL; |
|
f2b_file_t *files = NULL; |
|
glob_t globbuf; |
|
|
|
assert(pattern != NULL); |
|
|
|
if (glob(pattern, GLOB_MARK | GLOB_NOESCAPE, NULL, &globbuf) != 0) |
|
return NULL; |
|
|
|
for (size_t i = 0; i < globbuf.gl_pathc; i++) { |
|
if ((file = calloc(1, sizeof(f2b_file_t))) == NULL) |
|
continue; |
|
if (file_open(file, globbuf.gl_pathv[i]) == false) { |
|
/* f2b_log_msg(log_error, "can't open file: %s: %s", globbuf.gl_pathv[i], strerror(errno)); */ |
|
free(file); |
|
continue; |
|
} |
|
if (files != NULL) { |
|
file->next = files; |
|
files = file; |
|
} else { |
|
files = file; |
|
} |
|
} |
|
|
|
globfree(&globbuf); |
|
return files; |
|
} |
|
|
|
bool |
|
stop(cfg_t *cfg) { |
|
f2b_file_t *next = NULL; |
|
|
|
for (; cfg->files != NULL; cfg->files = next) { |
|
next = cfg->files->next; |
|
file_close(cfg->files); |
|
free(cfg->files); |
|
} |
|
|
|
return true; |
|
}
|
|
|