2023-09-20 10:45:18 +03:00
|
|
|
// x-run: tcc -Wextra -Wall -run cbt.c
|
2023-09-19 11:00:57 +03:00
|
|
|
#ifndef _CBT_H_
|
|
|
|
#define _CBT_H_
|
|
|
|
|
2023-09-20 10:45:18 +03:00
|
|
|
#include <stdbool.h>
|
|
|
|
#include <stddef.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
enum cbt_loglevel {
|
|
|
|
CBT_LOG_FATAL,
|
|
|
|
CBT_LOG_ERROR,
|
|
|
|
CBT_LOG_WARNING,
|
|
|
|
CBT_LOG_INFO,
|
|
|
|
CBT_LOG_DEBUG,
|
|
|
|
CBT_LOG_ALL
|
|
|
|
};
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_strarr {
|
2023-11-19 13:50:44 +03:00
|
|
|
char **items;
|
|
|
|
size_t size, cap;
|
|
|
|
};
|
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
extern long cbt_start_time;
|
|
|
|
extern enum cbt_loglevel cbt_verbosity;
|
2023-11-19 13:50:44 +03:00
|
|
|
extern char *cbt_cc, *cbt_cache_dir;
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
void cbt__init(int argc, char **argv, const char *source_file);
|
|
|
|
void cbt_cleanup(void);
|
|
|
|
#define CBT_INIT(ARGC, ARGV) cbt__init(ARGC, ARGV, __FILE__);
|
|
|
|
|
|
|
|
extern const char *cbt_log__colors[CBT_LOG_ALL + 1];
|
|
|
|
extern const char *cbt_log__typecolors[8];
|
2023-09-20 10:45:18 +03:00
|
|
|
|
|
|
|
void cbt_log(enum cbt_loglevel lvl, const char *fmt, ...);
|
2023-11-19 13:50:44 +03:00
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
long cbt_get_time(void);
|
2023-11-19 13:50:44 +03:00
|
|
|
|
2023-09-20 10:45:18 +03:00
|
|
|
unsigned long cbt_get_modtime(const char *filename);
|
|
|
|
bool cbt_needs_recompilation(const char *input, const char *output);
|
2023-11-19 13:50:44 +03:00
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
const char *cbt_escape_shell(char *arg);
|
|
|
|
const char *cbt_escape_args(char **args);
|
2023-11-25 19:21:10 +03:00
|
|
|
const char *cbt_escape_argsd(struct cbt_strarr arr);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
#ifndef CBT_FMT_NBUFS
|
|
|
|
#define CBT_FMT_NBUFS 64
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef CBT_FMT_BUFSIZE
|
|
|
|
#define CBT_FMT_BUFSIZE 1024
|
|
|
|
#endif
|
|
|
|
|
|
|
|
const char *cbt_fmt(const char *fmt, ...);
|
2023-09-20 10:45:18 +03:00
|
|
|
|
|
|
|
struct cbt_proc {
|
|
|
|
FILE *fp_stdin, *fp_stdout, *fp_stderr;
|
|
|
|
int pid;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct cbt_procgroup {
|
|
|
|
struct cbt_proc *items;
|
|
|
|
size_t size, cap;
|
2023-11-19 13:50:44 +03:00
|
|
|
bool fixed_size;
|
2023-09-20 10:45:18 +03:00
|
|
|
};
|
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
enum cbt_proc_mode {
|
2023-11-19 13:50:44 +03:00
|
|
|
CBT_PROC_AUTO = 0,
|
2023-11-15 10:24:27 +03:00
|
|
|
CBT_PROC_R = 1,
|
|
|
|
CBT_PROC_W = 2,
|
|
|
|
CBT_PROC_RW = 3,
|
|
|
|
};
|
|
|
|
|
|
|
|
#define cbt_proc_new(...) _cbt_proc_new(__VA_ARGS__, NULL)
|
2023-11-19 13:50:44 +03:00
|
|
|
#define cbt_shell(...) cbt_proc_wait(_cbt_proc_new(__VA_ARGS__, NULL))
|
2023-11-15 10:24:27 +03:00
|
|
|
struct cbt_proc _cbt_proc_new(enum cbt_proc_mode mode, ...);
|
|
|
|
struct cbt_proc cbt_proc_newv(enum cbt_proc_mode mode, va_list args);
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_proc cbt_proc_newd(enum cbt_proc_mode mode, struct cbt_strarr args);
|
2023-11-15 10:24:27 +03:00
|
|
|
int cbt_proc_wait(struct cbt_proc proc);
|
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
enum cbt_libtype { CBT_LIB_STATIC, CBT_LIB_SHARED };
|
|
|
|
|
|
|
|
struct cbt_lib {
|
|
|
|
enum cbt_libtype type;
|
|
|
|
const char *libname;
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_strarr sources, cflags, ldflags;
|
2023-11-19 13:50:44 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
#define cbt_lib_src(LIB, ...) _cbt_lib_src(LIB, __VA_ARGS__, NULL)
|
|
|
|
#define cbt_lib_cflags(LIB, ...) _cbt_lib_cflags(LIB, __VA_ARGS__, NULL)
|
|
|
|
#define cbt_lib_ldflags(LIB, ...) _cbt_lib_ldflags(LIB, __VA_ARGS__, NULL)
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_lib cbt_lib(const char *libname);
|
2023-11-19 13:50:44 +03:00
|
|
|
void _cbt_lib_src(struct cbt_lib *lib, ...);
|
|
|
|
void _cbt_lib_cflags(struct cbt_lib *lib, ...);
|
|
|
|
void _cbt_lib_ldflags(struct cbt_lib *lib, ...);
|
|
|
|
bool cbt_lib_build(struct cbt_lib lib, const char *cc);
|
|
|
|
void cbt_lib_free(struct cbt_lib lib);
|
|
|
|
|
|
|
|
struct cbt_binary {
|
|
|
|
const char *out_path;
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_strarr sources, cflags, ldflags;
|
2023-11-19 13:50:44 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
#define cbt_binary_add_src(B, ...) _cbt_binary_add_src(B, __VA_ARGS__, NULL)
|
|
|
|
#define cbt_binary_build(B, CC, ...) _cbt_binary_build(B, CC, __VA_ARGS__, NULL)
|
|
|
|
struct cbt_binary cbt_binary(const char *out_path);
|
|
|
|
void _cbt_binary_add_src(struct cbt_binary bin, ...);
|
|
|
|
bool _cbt_binary_build(struct cbt_binary bin, const char *compiler, ...);
|
|
|
|
void cbt_binary_free(struct cbt_binary bin);
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
#define CBT_ARRLEN(ARR) (sizeof(ARR) / sizeof(ARR[0]))
|
2023-09-20 10:45:18 +03:00
|
|
|
|
|
|
|
#ifndef CBT_REALLOC
|
|
|
|
#define CBT_REALLOC realloc
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define CBT_DA_INIT_SIZE 256
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
#define cbt_da_add(ARR, ITEM) \
|
|
|
|
{ \
|
|
|
|
if (ARR.size >= ARR.cap) { \
|
|
|
|
size_t size = ARR.size == 0 ? CBT_DA_INIT_SIZE : ARR.size * 2; \
|
|
|
|
ARR.items = CBT_REALLOC(ARR.items, size * sizeof(ARR.items[0])); \
|
|
|
|
} \
|
|
|
|
ARR.items[ARR.size] = (ITEM); \
|
|
|
|
ARR.size++; \
|
|
|
|
}
|
2023-09-19 11:00:57 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
#define cbt_da_remove(ARR, I) \
|
|
|
|
if (I >= 0 && I < ARR.size) { \
|
|
|
|
memmove(&ARR.items[I], &ARR.items[I + 1], \
|
|
|
|
(ARR.size - 1) * sizeof((ARR).items[0])); \
|
|
|
|
ARR.size--; \
|
|
|
|
}
|
2023-09-20 10:45:18 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
#define cbt_da_merge(OUT, IN) \
|
|
|
|
{ \
|
|
|
|
OUT.cap = OUT.size + IN.size; \
|
|
|
|
OUT.items = realloc(OUT.items, sizeof(OUT.items[0]) * OUT.cap); \
|
|
|
|
size_t offset = OUT.size; \
|
|
|
|
OUT.size = OUT.size + IN.size; \
|
|
|
|
memmove(&OUT.items[offset], IN.items, sizeof(OUT.items[0]) * IN.size); \
|
|
|
|
}
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
#define cbt_da_copy(OUT, IN, T) \
|
|
|
|
{ \
|
|
|
|
OUT.size = IN.size; \
|
|
|
|
OUT.cap = IN.size; \
|
|
|
|
OUT.items = malloc(sizeof(IN.items[0]) * IN.size); \
|
|
|
|
memcpy(OUT.items, IN.items, sizeof(IN.items[0]) * IN.size); \
|
|
|
|
}
|
2023-11-19 13:50:44 +03:00
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
#ifdef CBT_IMPLEMENTATION
|
2023-11-18 14:02:12 +03:00
|
|
|
//-*- begin cbt_impl.c 5
|
2023-11-15 10:24:27 +03:00
|
|
|
#define __USE_GNU
|
|
|
|
#include <ctype.h>
|
2023-11-18 14:02:12 +03:00
|
|
|
#include <errno.h>
|
2023-11-19 13:50:44 +03:00
|
|
|
#include <glob.h>
|
|
|
|
#include <libgen.h>
|
2023-11-25 19:21:10 +03:00
|
|
|
#ifdef __APPLE__
|
2023-11-19 13:50:44 +03:00
|
|
|
#include <linux/limits.h>
|
2023-11-25 19:21:10 +03:00
|
|
|
#endif
|
2023-11-18 14:02:12 +03:00
|
|
|
#include <poll.h>
|
|
|
|
#include <pthread.h>
|
|
|
|
#include <signal.h>
|
2023-11-15 10:24:27 +03:00
|
|
|
#include <stdarg.h>
|
2023-11-18 14:02:12 +03:00
|
|
|
#include <stdbool.h>
|
|
|
|
#include <stdlib.h>
|
2023-11-15 10:24:27 +03:00
|
|
|
#include <string.h>
|
|
|
|
#include <sys/stat.h>
|
2023-11-18 14:02:12 +03:00
|
|
|
#include <sys/types.h>
|
2023-11-15 10:24:27 +03:00
|
|
|
#include <sys/wait.h>
|
2023-11-18 14:02:12 +03:00
|
|
|
#include <time.h>
|
|
|
|
#include <unistd.h>
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
#define CBT_FAIL(COND) \
|
|
|
|
if (COND) { \
|
|
|
|
cbt_log(CBT_LOG_FATAL, "condition %s failed, %s", #COND, strerror(errno)); \
|
|
|
|
abort(); \
|
|
|
|
};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
#define CBT_TODO(NAME) \
|
|
|
|
{ \
|
|
|
|
cbt_log(CBT_LOG_FATAL, "%s: Not implemented! %s:%d", NAME, __FILE__, \
|
|
|
|
__LINE__); \
|
|
|
|
abort(); \
|
|
|
|
}
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
|
|
|
|
#ifndef PATH_MAX
|
|
|
|
#define PATH_MAX 1024
|
|
|
|
#endif
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
enum cbt_loglevel cbt_verbosity = CBT_LOG_INFO;
|
2023-09-20 10:45:18 +03:00
|
|
|
char *cbt_cc = NULL;
|
2023-11-19 13:50:44 +03:00
|
|
|
char cbt_location[PATH_MAX] = {0};
|
|
|
|
char *cbt_cache_dir = NULL;
|
2023-11-15 10:24:27 +03:00
|
|
|
long cbt_start_time = 0;
|
2023-11-16 14:36:47 +03:00
|
|
|
bool cbt_running = false;
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
struct _cbt__autoproc_fdset {
|
|
|
|
struct pollfd *items;
|
|
|
|
size_t size, cap;
|
2023-11-18 14:02:12 +03:00
|
|
|
} cbt__autoproc_fdset = {0};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
struct _cbt__autoproc_set {
|
|
|
|
struct _cbt__autoproc_set__item {
|
|
|
|
FILE *fp;
|
|
|
|
bool is_err;
|
|
|
|
} *items;
|
|
|
|
size_t size, cap;
|
2023-11-18 14:02:12 +03:00
|
|
|
} cbt__autoproc_set = {0};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
struct cbt_procgroup cbt__proc_pool = {0, .cap = 4, .fixed_size = 1};
|
2023-11-16 14:36:47 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
pthread_t cbt__autoproc_thread;
|
2023-11-15 10:24:27 +03:00
|
|
|
pthread_mutex_t _cbt__autoproc_mut = PTHREAD_MUTEX_INITIALIZER;
|
|
|
|
|
2023-09-20 10:45:18 +03:00
|
|
|
const char *cbt_log__colors[CBT_LOG_ALL + 1] = {
|
2023-11-18 14:02:12 +03:00
|
|
|
"\x1b[1;31m", "\x1b[31m", "\x1b[33m", "\x1b[32m", "\x1b[34m", "\x1b[35m"};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
const char *cbt_log__text[CBT_LOG_ALL + 1] = {"FATAL", "ERROR", "WARN ",
|
|
|
|
"INFO ", "DEBUG", "TRACE"};
|
2023-11-15 10:24:27 +03:00
|
|
|
void *cbt__line_processor(void *);
|
|
|
|
|
|
|
|
void cbt__init(int argc, char **argv, const char *source_file) {
|
|
|
|
cbt_start_time = cbt_get_time();
|
|
|
|
cbt_running = true;
|
|
|
|
(void)argc;
|
2023-09-20 10:45:18 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
cbt_log(CBT_LOG_INFO, "Running CBT build %s %s", __DATE__, __TIME__);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
if (!cbt_cc)
|
|
|
|
cbt_cc = getenv("CC");
|
|
|
|
if (!cbt_cc)
|
|
|
|
cbt_cc = "cc";
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
cbt_log(CBT_LOG_INFO, "Found C Compiler: %s", cbt_cc);
|
2023-11-19 13:50:44 +03:00
|
|
|
|
|
|
|
CBT_FAIL(readlink("/proc/self/exe", cbt_location, PATH_MAX) == -1);
|
|
|
|
cbt_log(CBT_LOG_INFO, "Location: %s", cbt_location);
|
|
|
|
|
|
|
|
char *cbt_cache_dir = strdup(cbt_location);
|
|
|
|
cbt_cache_dir = strdup(cbt_fmt("%s/%s", dirname(cbt_cache_dir), "cache"));
|
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "Args: %s", cbt_escape_args(argv));
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "%s", cbt_fmt("format test: %s", "ok"));
|
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
cbt_log(CBT_LOG_INFO, "Cache dir: %s", cbt_cache_dir);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
cbt__proc_pool.items = calloc(cbt__proc_pool.cap, sizeof(struct cbt_proc));
|
2023-11-18 14:14:01 +03:00
|
|
|
|
2023-11-19 13:50:44 +03:00
|
|
|
{
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "Starting line processor thread");
|
|
|
|
int err;
|
|
|
|
if ((err = pthread_create(&cbt__autoproc_thread, NULL, cbt__line_processor,
|
|
|
|
NULL)) != 0) {
|
|
|
|
cbt_log(CBT_LOG_ERROR, "pthread_create() failed for line processor: %d",
|
|
|
|
err);
|
|
|
|
exit(1);
|
|
|
|
}
|
2023-11-18 14:14:01 +03:00
|
|
|
}
|
2023-11-16 14:36:47 +03:00
|
|
|
|
|
|
|
if (cbt_needs_recompilation(source_file, argv[0])) {
|
|
|
|
cbt_log(CBT_LOG_INFO, "Recompiling...");
|
2023-11-19 13:50:44 +03:00
|
|
|
struct cbt_proc cc = cbt_proc_new(CBT_PROC_AUTO, cbt_cc, source_file, "-o",
|
|
|
|
argv[0], "-Wall", "-Wextra", NULL);
|
2023-11-16 14:36:47 +03:00
|
|
|
int status = cbt_proc_wait(cc);
|
|
|
|
cbt_log(CBT_LOG_INFO, "CC returned %d", status);
|
|
|
|
if (status == 0) {
|
|
|
|
cbt_cleanup();
|
|
|
|
cbt_log(CBT_LOG_INFO, "Restarting CBT", cbt_escape_args(argv));
|
|
|
|
execv(argv[0], argv);
|
|
|
|
exit(0);
|
|
|
|
}
|
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
void cbt_cleanup(void) {
|
|
|
|
usleep(50000);
|
|
|
|
pthread_mutex_lock(&_cbt__autoproc_mut);
|
|
|
|
cbt_running = false;
|
|
|
|
cbt_log(CBT_LOG_INFO, "cbt shutting down");
|
2023-11-19 13:50:44 +03:00
|
|
|
pthread_join(cbt__autoproc_thread, NULL);
|
2023-11-15 10:24:27 +03:00
|
|
|
pthread_mutex_unlock(&_cbt__autoproc_mut);
|
2023-11-16 14:36:47 +03:00
|
|
|
cbt_log(CBT_LOG_INFO, "shutdown complete");
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
const char *cbt_fmt(const char *fmt, ...) {
|
|
|
|
static char cbt_fmt_buffers[CBT_FMT_NBUFS][CBT_FMT_BUFSIZE];
|
|
|
|
static int cbt_fmt_buffer = 0;
|
|
|
|
char *buffer = cbt_fmt_buffers[cbt_fmt_buffer++ % CBT_FMT_NBUFS];
|
|
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
|
|
vsnprintf(buffer, CBT_FMT_BUFSIZE - 1, fmt, args);
|
|
|
|
va_end(args);
|
|
|
|
|
|
|
|
return buffer;
|
2023-09-20 10:45:18 +03:00
|
|
|
}
|
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
long cbt_get_time(void) {
|
|
|
|
struct timespec now;
|
|
|
|
clock_gettime(CLOCK_REALTIME, &now);
|
|
|
|
return (long)(now.tv_sec * 1000000L) + (long)(now.tv_nsec / 1000L);
|
|
|
|
}
|
2023-09-20 10:45:18 +03:00
|
|
|
|
|
|
|
void cbt_log(enum cbt_loglevel lvl, const char *fmt, ...) {
|
2023-11-18 14:02:12 +03:00
|
|
|
if (lvl > cbt_verbosity)
|
|
|
|
return;
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
double time_delta = (double)(cbt_get_time() - cbt_start_time) / 1000.0L;
|
2023-11-18 14:02:12 +03:00
|
|
|
printf("[%10.3lf] [%s%s\x1b[0m] ", time_delta, cbt_log__colors[lvl],
|
|
|
|
cbt_log__text[lvl]);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
|
|
vprintf(fmt, args);
|
|
|
|
va_end(args);
|
|
|
|
fputc('\n', stdout);
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: find a better option
|
|
|
|
const char *cbt_escape_shell(char *arg) {
|
|
|
|
int n = strlen(arg);
|
|
|
|
|
|
|
|
char *esc = (char *)cbt_fmt(""); // just take a buffer
|
|
|
|
|
|
|
|
esc[0] = '\'';
|
|
|
|
int j = 1, i;
|
|
|
|
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
if (arg[i] == '\'') {
|
2023-11-18 14:02:12 +03:00
|
|
|
if (j + 4 >= CBT_FMT_BUFSIZE) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
esc[j] = '\'';
|
2023-11-18 14:02:12 +03:00
|
|
|
esc[j + 1] = '\\';
|
|
|
|
esc[j + 2] = '\'';
|
|
|
|
esc[j + 3] = '\'';
|
2023-11-15 10:24:27 +03:00
|
|
|
j += 4;
|
2023-09-20 10:45:18 +03:00
|
|
|
} else {
|
2023-11-18 14:02:12 +03:00
|
|
|
if (j >= CBT_FMT_BUFSIZE) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
esc[j++] = arg[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
if (j >= CBT_FMT_BUFSIZE) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
esc[j] = '\'';
|
|
|
|
|
|
|
|
return esc;
|
|
|
|
}
|
|
|
|
|
|
|
|
const char *cbt_escape_args(char **args) {
|
2023-11-18 14:02:12 +03:00
|
|
|
if (args == NULL || args[0] == NULL)
|
|
|
|
return cbt_fmt("");
|
2023-11-15 10:24:27 +03:00
|
|
|
char *out = (char *)cbt_escape_shell(args[0]);
|
|
|
|
for (int i = 1; args[i] != NULL; i++) {
|
|
|
|
out = (char *)cbt_fmt("%s %s", out, cbt_escape_shell(args[i]));
|
|
|
|
}
|
|
|
|
return out;
|
|
|
|
}
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
const char *cbt_escape_argsd(struct cbt_strarr arr) {
|
2023-11-19 13:50:44 +03:00
|
|
|
if (arr.size == 0 || arr.items[0] == NULL)
|
|
|
|
return cbt_fmt("");
|
|
|
|
char *out = (char *)cbt_escape_shell(arr.items[0]);
|
|
|
|
for (size_t i = 1; i < arr.size; i++) {
|
|
|
|
if (arr.items[i]) {
|
|
|
|
out = (char *)cbt_fmt("%s %s", out, cbt_escape_shell(arr.items[i]));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return out;
|
|
|
|
}
|
|
|
|
|
2023-11-15 10:24:27 +03:00
|
|
|
unsigned long cbt_get_modtime(const char *filename) {
|
|
|
|
struct stat st;
|
|
|
|
int ret = lstat(filename, &st);
|
2023-11-18 14:02:12 +03:00
|
|
|
if (ret == -1)
|
|
|
|
return -1;
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "stat(%s) -> %ld", filename, st.st_mtime);
|
|
|
|
return st.st_mtime;
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
bool cbt_needs_recompilation(const char *input, const char *output) {
|
|
|
|
unsigned long mod_in = cbt_get_modtime(input),
|
|
|
|
mod_out = cbt_get_modtime(output);
|
|
|
|
if (mod_in == (unsigned long)-1) {
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(CBT_LOG_WARNING, "cbt_needs_recompilation(): %s does not exist",
|
|
|
|
input);
|
2023-11-15 10:24:27 +03:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return mod_out == (unsigned long)-1 || mod_in > mod_out;
|
|
|
|
}
|
|
|
|
|
|
|
|
void cbt__add_fd(int fd, bool is_err) {
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "cbt__add_fd() adding %d", fd);
|
|
|
|
pthread_mutex_lock(&_cbt__autoproc_mut);
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "cbt__add_fd() locked mutex for fd=%d", fd);
|
|
|
|
|
|
|
|
struct pollfd pfd;
|
|
|
|
pfd.fd = fd;
|
|
|
|
pfd.events = POLLIN | POLLOUT;
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
struct _cbt__autoproc_set__item info = {.is_err = is_err};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
cbt_da_add(cbt__autoproc_fdset, pfd);
|
|
|
|
cbt_da_add(cbt__autoproc_set, info);
|
|
|
|
pthread_mutex_unlock(&_cbt__autoproc_mut);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cbt_proc _cbt_proc_new(enum cbt_proc_mode mode, ...) {
|
|
|
|
va_list args;
|
|
|
|
va_start(args, mode);
|
|
|
|
struct cbt_proc process = cbt_proc_newv(mode, args);
|
|
|
|
va_end(args);
|
|
|
|
return process;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cbt_proc cbt_proc_newv(enum cbt_proc_mode mode, va_list args) {
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_strarr args_da = {0};
|
2023-11-15 10:24:27 +03:00
|
|
|
while (true) {
|
|
|
|
const char *arg = va_arg(args, const char *);
|
2023-11-18 14:02:12 +03:00
|
|
|
if (arg == NULL)
|
|
|
|
break;
|
2023-11-15 10:24:27 +03:00
|
|
|
cbt_da_add(args_da, strdup(arg));
|
|
|
|
}
|
2023-11-16 14:36:47 +03:00
|
|
|
cbt_da_add(args_da, NULL);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
return cbt_proc_newd(mode, args_da);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cbt_proc cbt_proc_newd(enum cbt_proc_mode mode, struct cbt_strarr args) {
|
|
|
|
struct cbt_proc procinfo = {0};
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "Spawning process with mode %d and args %s", mode,
|
2023-11-25 19:21:10 +03:00
|
|
|
cbt_escape_args(args.items));
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
int pipe_fds_sout[2] = {0}, pipe_fds_serr[2] = {0}, pipe_fds_sin[2] = {0};
|
|
|
|
if (mode & CBT_PROC_W)
|
|
|
|
CBT_FAIL(pipe(pipe_fds_sin) == -1);
|
2023-11-15 10:24:27 +03:00
|
|
|
CBT_FAIL(pipe(pipe_fds_sout) == -1);
|
|
|
|
CBT_FAIL(pipe(pipe_fds_serr) == -1);
|
|
|
|
|
|
|
|
procinfo.pid = fork();
|
|
|
|
if (procinfo.pid < 0) {
|
|
|
|
cbt_log(CBT_LOG_ERROR, "fork() failed: %s", strerror(errno));
|
|
|
|
goto cbt_proc_new_cleanup;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (procinfo.pid == 0) { // child
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
if (mode & CBT_PROC_W)
|
|
|
|
CBT_FAIL(dup2(pipe_fds_sin[1], STDIN_FILENO) == -1);
|
2023-11-15 10:24:27 +03:00
|
|
|
CBT_FAIL(dup2(pipe_fds_sout[1], STDOUT_FILENO) == -1);
|
|
|
|
CBT_FAIL(dup2(pipe_fds_serr[1], STDERR_FILENO) == -1);
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
execvp(args.items[0], (char *const *)args.items);
|
2023-11-15 10:24:27 +03:00
|
|
|
__builtin_unreachable();
|
|
|
|
} else {
|
2023-11-18 14:02:12 +03:00
|
|
|
if (mode & CBT_PROC_W)
|
|
|
|
procinfo.fp_stdin = fdopen(pipe_fds_sin[0], "w");
|
|
|
|
if (mode & CBT_PROC_R)
|
|
|
|
procinfo.fp_stdout = fdopen(pipe_fds_sout[0], "r");
|
|
|
|
if (mode & CBT_PROC_R)
|
|
|
|
procinfo.fp_stderr = fdopen(pipe_fds_serr[0], "r");
|
|
|
|
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "stdin: %p (fd=%d)", procinfo.fp_stdin,
|
|
|
|
pipe_fds_sin[0]);
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "stdout: %p (fd=%d)", procinfo.fp_stdout,
|
|
|
|
pipe_fds_sout[0]);
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "stderr: %p (fd=%d)", procinfo.fp_stderr,
|
|
|
|
pipe_fds_serr[0]);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
if (!(mode & CBT_PROC_R)) {
|
|
|
|
cbt__add_fd(pipe_fds_sout[0], false);
|
|
|
|
cbt__add_fd(pipe_fds_serr[0], true);
|
2023-09-20 10:45:18 +03:00
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_proc_new_cleanup:
|
2023-11-15 10:24:27 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "freeing up args");
|
2023-11-25 19:21:10 +03:00
|
|
|
for (size_t i = 0; i < args.size; i++) {
|
|
|
|
free(args.items[i]);
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
2023-11-25 19:21:10 +03:00
|
|
|
free(args.items);
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
return procinfo;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
int cbt_proc_wait(struct cbt_proc proc) {
|
|
|
|
int wstatus;
|
|
|
|
|
|
|
|
do {
|
|
|
|
pid_t w = waitpid(proc.pid, &wstatus, WUNTRACED | WCONTINUED);
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "waitpid(%d) -> %d, status = %d", proc.pid, w,
|
|
|
|
wstatus);
|
2023-11-15 10:24:27 +03:00
|
|
|
if (w == -1) {
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(CBT_LOG_WARNING, "waitpid(%d) -> %d (%s)", proc.pid, w,
|
|
|
|
strerror(errno));
|
2023-11-15 10:24:27 +03:00
|
|
|
return -2000;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (WIFEXITED(wstatus)) {
|
|
|
|
return WEXITSTATUS(wstatus);
|
|
|
|
} else if (WIFSIGNALED(wstatus)) {
|
|
|
|
return 1000 + WTERMSIG(wstatus);
|
|
|
|
}
|
|
|
|
} while (!WIFEXITED(wstatus) && !WIFSIGNALED(wstatus));
|
|
|
|
return -1000;
|
|
|
|
}
|
|
|
|
|
2023-11-18 14:02:12 +03:00
|
|
|
static void cbt__line_processor_sig_handler(int signal) { (void)signal; }
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
void *cbt__line_processor(void *arg) {
|
|
|
|
(void)arg;
|
2023-11-18 14:02:12 +03:00
|
|
|
struct timespec timeout = {.tv_sec = 0, .tv_nsec = 100};
|
2023-11-15 10:24:27 +03:00
|
|
|
|
|
|
|
sigset_t sigmask;
|
|
|
|
sigemptyset(&sigmask);
|
|
|
|
sigaddset(&sigmask, SIGUSR1);
|
|
|
|
|
|
|
|
struct sigaction sa;
|
|
|
|
sa.sa_handler = cbt__line_processor_sig_handler;
|
|
|
|
sigemptyset(&sa.sa_mask);
|
|
|
|
sa.sa_flags = SA_RESTART;
|
|
|
|
|
|
|
|
CBT_FAIL(sigaction(SIGUSR1, &sa, NULL) == -1);
|
|
|
|
|
|
|
|
while (cbt_running) {
|
|
|
|
pthread_mutex_lock(&_cbt__autoproc_mut);
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
#if 1 || defined(__APPLE__)
|
2023-11-18 14:02:12 +03:00
|
|
|
int res = ppoll(cbt__autoproc_fdset.items, cbt__autoproc_fdset.size,
|
|
|
|
&timeout, &sigmask);
|
2023-11-15 10:24:27 +03:00
|
|
|
if (res < 0) {
|
|
|
|
cbt_log(CBT_LOG_ERROR, "ppoll() -> %d (%s)", res, strerror(errno));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
for (int i = 0; i < cbt__autoproc_set.size; i++) {
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(CBT_LOG_DEBUG, "fdset[%d] = { %d, %d }", i,
|
|
|
|
cbt__autoproc_set.items[i].fd, cbt__autoproc_set.items[i].events);
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
|
|
|
cbt_log(CBT_LOG_DEBUG, "poll(_, %d, _) ...", cbt__autoproc_set.size);
|
|
|
|
int res = poll(cbt__autoproc_set.items, cbt__autoproc_set.size, 1000);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
char buf[1024];
|
|
|
|
for (size_t i = 0; i < cbt__autoproc_fdset.size; i++) {
|
|
|
|
struct pollfd pfd = cbt__autoproc_fdset.items[i];
|
|
|
|
struct _cbt__autoproc_set__item item = cbt__autoproc_set.items[i];
|
|
|
|
if (pfd.revents != 0) {
|
|
|
|
if (pfd.revents & POLLIN) {
|
|
|
|
ssize_t s = read(pfd.fd, buf, sizeof(buf));
|
2023-11-18 14:02:12 +03:00
|
|
|
cbt_log(item.is_err ? CBT_LOG_ERROR : CBT_LOG_INFO, "fd(%d): %.*s",
|
|
|
|
pfd.fd, (int)s - 1, buf);
|
2023-11-15 10:24:27 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pthread_mutex_unlock(&_cbt__autoproc_mut);
|
|
|
|
usleep(1000);
|
2023-09-20 10:45:18 +03:00
|
|
|
}
|
2023-11-15 10:24:27 +03:00
|
|
|
return NULL;
|
2023-09-20 10:45:18 +03:00
|
|
|
}
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
struct cbt_lib cbt_lib(const char *libname) {
|
2023-11-19 13:50:44 +03:00
|
|
|
return (struct cbt_lib){.type = CBT_LIB_STATIC,
|
|
|
|
.libname = libname,
|
|
|
|
.cflags = {0},
|
|
|
|
.ldflags = {0},
|
|
|
|
.sources = {0}};
|
|
|
|
}
|
|
|
|
|
|
|
|
int _cbt_globerr(const char *epath, int eerrno) {
|
|
|
|
cbt_log(CBT_LOG_WARNING, "glob() failed at %s: %s", epath, strerror(eerrno));
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void _cbt_lib_src(struct cbt_lib *lib, ...) {
|
|
|
|
va_list args;
|
|
|
|
va_start(args, lib);
|
|
|
|
|
|
|
|
glob_t globbuf;
|
|
|
|
|
|
|
|
for (char *arg = va_arg(args, char *); arg; arg = va_arg(args, char *)) {
|
2023-11-25 19:21:10 +03:00
|
|
|
glob(arg, GLOB_APPEND | GLOB_BRACE, _cbt_globerr, &globbuf);
|
2023-11-19 13:50:44 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < globbuf.gl_pathc; i++) {
|
|
|
|
char *path = strdup(globbuf.gl_pathv[i]);
|
|
|
|
cbt_da_add(lib->sources, path);
|
|
|
|
}
|
|
|
|
|
|
|
|
globfree(&globbuf);
|
|
|
|
va_end(args);
|
|
|
|
}
|
|
|
|
|
|
|
|
void _cbt_lib_cflags(struct cbt_lib *lib, ...) {
|
|
|
|
va_list args;
|
|
|
|
va_start(args, lib);
|
|
|
|
for (char *arg = va_arg(args, char *); arg; arg = va_arg(args, char *)) {
|
|
|
|
cbt_da_add(lib->cflags, arg);
|
|
|
|
}
|
|
|
|
va_end(args);
|
|
|
|
}
|
|
|
|
|
|
|
|
void _cbt_lib_ldflags(struct cbt_lib *lib, ...) {
|
|
|
|
va_list args;
|
|
|
|
va_start(args, lib);
|
|
|
|
for (char *arg = va_arg(args, char *); arg; arg = va_arg(args, char *)) {
|
|
|
|
cbt_da_add(lib->ldflags, arg);
|
|
|
|
}
|
|
|
|
va_end(args);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool cbt_lib_build(struct cbt_lib lib, const char *compiler) {
|
2023-11-25 19:21:10 +03:00
|
|
|
cbt_log(CBT_LOG_INFO, "library: %s", lib.libname);
|
2023-11-19 13:50:44 +03:00
|
|
|
cbt_log(CBT_LOG_INFO, " cflags: %s", cbt_escape_argsd(lib.cflags));
|
|
|
|
cbt_log(CBT_LOG_INFO, " ldflags: %s", cbt_escape_argsd(lib.ldflags));
|
|
|
|
cbt_log(CBT_LOG_INFO, " sources: %s", cbt_escape_argsd(lib.sources));
|
|
|
|
cbt_log(CBT_LOG_INFO, " cc: %s", compiler ? compiler : cbt_cc);
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
if (lib.sources.size == 0) {
|
|
|
|
cbt_log(CBT_LOG_WARNING, "cbt_lib_build(): no sources for %s", lib.libname);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cbt_strarr args = {0};
|
2023-11-19 13:50:44 +03:00
|
|
|
cbt_da_add(args, (char *)(compiler ? compiler : cbt_cc));
|
|
|
|
|
|
|
|
cbt_da_merge(args, lib.cflags);
|
|
|
|
|
|
|
|
cbt_log(CBT_LOG_INFO, "args.size: %d", args.size);
|
|
|
|
cbt_log(CBT_LOG_INFO, "CMD: %s", cbt_escape_argsd(args));
|
|
|
|
|
2023-11-25 19:21:10 +03:00
|
|
|
cbt_da_add(args, NULL);
|
|
|
|
cbt_da_add(args, "-o");
|
|
|
|
cbt_da_add(args, NULL);
|
|
|
|
|
|
|
|
for (size_t i = 0; i < lib.sources.size; i++) {
|
|
|
|
args.items[args.size - 3] = lib.sources.items[i];
|
|
|
|
args.items[args.size - 1] = ".o"; // TODO
|
|
|
|
struct cbt_proc proc = cbt_proc_newd(CBT_PROC_AUTO, args);
|
|
|
|
if (cbt_proc_wait(proc) != 0) {
|
|
|
|
cbt_log(CBT_LOG_ERROR, "build failed: %s", cbt_escape_argsd(args));
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
char **argp = &args.items[args.size - 1];
|
|
|
|
cbt_da_merge(args, lib.ldflags);
|
|
|
|
|
|
|
|
*argp = (char *)cbt_fmt("lib%s.so", lib.libname);
|
|
|
|
|
|
|
|
// todo: link
|
|
|
|
|
|
|
|
return false;
|
2023-11-19 13:50:44 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
void cbt_lib_free(struct cbt_lib lib) {
|
|
|
|
free(lib.cflags.items);
|
|
|
|
free(lib.ldflags.items);
|
|
|
|
for (size_t i = 0; i < lib.sources.size; i++) {
|
|
|
|
free(lib.sources.items[i]);
|
|
|
|
}
|
|
|
|
free(lib.sources.items);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cbt_binary cbt_binary(const char *out_path);
|
|
|
|
void _cbt_binary_add_src(struct cbt_binary bin, ...);
|
|
|
|
bool _cbt_binary_build(struct cbt_binary bin, const char *compiler, ...);
|
|
|
|
void cbt_binary_free(struct cbt_binary bin);
|
|
|
|
|
|
|
|
#undef CBT_TODO
|
2023-11-15 10:24:27 +03:00
|
|
|
#undef CBT_FAIL
|
2023-11-18 14:02:12 +03:00
|
|
|
//-*- end
|
2023-11-15 10:24:27 +03:00
|
|
|
#endif
|
|
|
|
|
2023-09-19 11:00:57 +03:00
|
|
|
#endif
|