1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384 |
- // gcc -lutil build.c -o hacer && ./hacer
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <ctype.h>
- #include <errno.h>
- #include <stdarg.h>
- #include <signal.h>
- #include <unistd.h>
- #include <sys/prctl.h>
- #include <pty.h>
- #include <fcntl.h>
- #include <fnmatch.h>
- #include <dirent.h>
- #include <libgen.h>
- #include <sys/stat.h>
- #include <sys/types.h>
- #include <sys/wait.h>
- #include <sys/sysinfo.h>
- // link with -lutil
- // list all <header> files in use
- //egrep -ro '^\s*#\s*include\s+<(.*)>' src/ | sed -e 's/^.*<\(.*\)>.*$/\1/'
- char** g_gcc_opts_list;
- char* g_gcc_opts_flat;
- char* g_gcc_include;
- char* g_gcc_libs;
- int g_nprocs;
- struct child_process_info {
- int pid;
- char state; // 'q'ueued, 'r'unning, 'd'ead
- int exit_status;
- int pty;
- int child_stdin;
- int child_stdout;
- int child_stderr;
- FILE* f_stdin;
- FILE* f_stdout;
- FILE* f_stderr;
-
- char* output_buffer;
- size_t buf_alloc;
- size_t buf_len;
- };
- typedef struct rglob_entry {
- char type;
- char* full_path;
- char* file_name;
- // char* dir_name;
- } rglob_entry;
- typedef struct rglob {
- char* pattern;
- int len;
- int alloc;
- rglob_entry* entries;
-
- } rglob;
- typedef struct strlist {
- int len;
- int alloc;
- char** entries;
- } strlist;
- void free_strpp(char** l) {
- for(char** s = l; *s; s++) free(*s);
- free(l);
- }
- #define PP_ARG_N(_1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18, _19, _20, _21, _22, _23, _24, _25, _26, _27, _28, _29, _30, _31, _32, _33, _34, _35, _36, _37, _38, _39, _40, _41, _42, _43, _44, _45, _46, _47, _48, _49, _50, _51, _52, _53, _54, _55, _56, _57, _58, _59, _60, _61, _62, _63, N, ...) N
- #define PP_RSEQ_N() 63, 62, 61, 60, 59, 58, 57, 56, 55, 54, 53, 52, 51, 50, 49, 48, 47, 46, 45, 44, 43, 42, 41, 40, 39, 38, 37, 36, 35, 34, 33, 32, 31, 30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0
- #define PP_NARG_(...) PP_ARG_N(__VA_ARGS__)
- #define PP_NARG(...) PP_NARG_(__VA_ARGS__, PP_RSEQ_N())
- #define check_alloc(x) \
- if((x)->len + 1 >= (x)->alloc) { \
- (x)->alloc *= 2; \
- (x)->entries = realloc((x)->entries, (x)->alloc * sizeof(*(x)->entries)); \
- }
- void strlist_init(strlist* sl) {
- sl->len = 0;
- sl->alloc = 32;
- sl->entries = malloc(sl->alloc * sizeof(*sl->entries));
- }
- strlist* strlist_new() {
- strlist* sl = malloc(sizeof(*sl));
- strlist_init(sl);
- return sl;
- }
- void strlist_push(strlist* sl, char* e) {
- check_alloc(sl);
- sl->entries[sl->len++] = e;
- sl->entries[sl->len] = 0;
- }
- typedef unsigned long hash_t;
- hash_t strhash(char* str) {
- unsigned long h = 0;
- int c;
- while(c = *str++) {
- h = c + (h << 6) + (h << 16) - h;
- }
- return h;
- }
- hash_t strnhash(char* str, size_t n) {
- unsigned long h = 0;
- int c;
- while((c = *str++) && n--) {
- h = c + (h << 6) + (h << 16) - h;
- }
-
- return h;
- }
- typedef struct {
- hash_t hash;
- char* str;
- unsigned int len;
- int refs;
- } string_cache_bucket;
- typedef struct {
- int fill;
- int alloc;
- string_cache_bucket* buckets;
- } string_cache_t;
- string_cache_t string_cache;
- void string_cache_init(int alloc) {
- string_cache.fill = 0;
- string_cache.alloc = alloc ? alloc : 1024;
- string_cache.buckets = calloc(1, string_cache.alloc * sizeof(*string_cache.buckets));
- }
- int string_cache_find_bucket(string_cache_t* ht, hash_t hash, char* key) {
- int b = hash % ht->alloc;
-
- for(int i = 0; i < ht->alloc; i++) {
- // empty bucket
- if(ht->buckets[b].str == NULL) return b;
-
- // full bucket
- if(ht->buckets[b].hash == hash) {
- if(0 == strcmp(key, ht->buckets[b].str)) {
- return b;
- }
- }
-
- // probe forward on collisions
- b = (b + 1) % ht->alloc;
- }
-
- // should never reach here
- printf("oops. -1 bucket \n");
- return -1;
- }
- void string_cache_expand(string_cache_t* ht) {
- int old_alloc = ht->alloc;
- ht->alloc *= 2;
-
- string_cache_bucket* old = ht->buckets;
- ht->buckets = calloc(1, ht->alloc * sizeof(*ht->buckets));
-
- for(int i = 0, f = 0; i < old_alloc && f < ht->fill; i++) {
- if(!old[i].str) continue;
-
- int b = string_cache_find_bucket(ht, old[i].hash, old[i].str);
-
- ht->buckets[b] = old[i];
- f++;
- }
-
- free(old);
- }
- char* strcache(char* in) {
- hash_t hash = strhash(in);
- int b = string_cache_find_bucket(&string_cache, hash, in);
-
- if(!string_cache.buckets[b].str) {
- if(string_cache.fill > string_cache.alloc * .80) {
- string_cache_expand(&string_cache);
-
- // the bucket location has changed
- b = string_cache_find_bucket(&string_cache, hash, in);
- }
-
- string_cache.fill++;
-
- string_cache.buckets[b].str = strdup(in);
- string_cache.buckets[b].hash = hash;
- string_cache.buckets[b].refs = 1;
- string_cache.buckets[b].len = strlen(in);
- }
- else {
- string_cache.buckets[b].refs++;
- }
-
- return string_cache.buckets[b].str;
- }
- char* strncache(char* in, size_t n) {
- hash_t hash = strnhash(in, n);
- int b = string_cache_find_bucket(&string_cache, hash, in);
-
- if(!string_cache.buckets[b].str) {
- if(string_cache.fill > string_cache.alloc * .80) {
- string_cache_expand(&string_cache);
-
- // the bucket location has changed
- b = string_cache_find_bucket(&string_cache, hash, in);
- }
-
- string_cache.fill++;
-
- string_cache.buckets[b].str = strndup(in, n);
- string_cache.buckets[b].hash = hash;
- string_cache.buckets[b].refs = 1;
- string_cache.buckets[b].len = n;
- }
- else {
- string_cache.buckets[b].refs++;
- }
-
- return string_cache.buckets[b].str;
- }
- void struncache(char* in) {
- hash_t hash = strhash(in);
- int b = string_cache_find_bucket(&string_cache, hash, in);
-
- if(!string_cache.buckets[b].str) {
- // normal string, free it
- free(in);
- return;
- }
-
- string_cache.buckets[b].refs--;
- if(string_cache.buckets[b].refs == 0) {
- // just do nothing for now. deletion is a pain.
- }
- }
- typedef struct {
- hash_t hash;
- char* key;
- void* value;
- } hashbucket;
- typedef struct hashtable {
- int fill;
- int alloc;
- hashbucket* buckets;
- } hashtable;
- void hash_init(hashtable* ht, int alloc) {
- ht->fill = 0;
- ht->alloc = alloc ? alloc : 128;
- ht->buckets = calloc(1, ht->alloc * sizeof(*ht->buckets));
- }
- hashtable* hash_new(int alloc) {
- hashtable* ht = malloc(sizeof(*ht));
- hash_init(ht, alloc);
- return ht;
- }
- int hash_find_bucket(hashtable* ht, hash_t hash, char* key) {
- int b = hash % ht->alloc;
-
- for(int i = 0; i < ht->alloc; i++) {
- // empty bucket
- if(ht->buckets[b].key == NULL) return b;
-
- // full bucket
- if(ht->buckets[b].hash == hash) {
- if(0 == strcmp(key, ht->buckets[b].key)) {
- return b;
- }
- }
-
- // probe forward on collisions
- b = (b + 1) % ht->alloc;
- }
-
- // should never reach here
- printf("oops. -1 bucket \n");
- return -1;
- }
- void hash_expand(hashtable* ht) {
- int old_alloc = ht->alloc;
- ht->alloc *= 2;
-
- hashbucket* old = ht->buckets;
- ht->buckets = calloc(1, ht->alloc * sizeof(*ht->buckets));
-
- for(int i = 0, f = 0; i < old_alloc && f < ht->fill; i++) {
- if(!old[i].key) continue;
-
- int b = hash_find_bucket(ht, old[i].hash, old[i].key);
-
- ht->buckets[b] = old[i];
- f++;
- }
-
- free(old);
- }
- void hash_insert(hashtable* ht, char* key, void* value) {
-
- hash_t hash = strhash(key);
- int b = hash_find_bucket(ht, hash, key);
-
- if(!ht->buckets[b].key) {
- if(ht->fill > ht->alloc * .80) {
- hash_expand(ht);
-
- // the bucket location has changed
- b = hash_find_bucket(ht, hash, key);
- }
-
- ht->fill++;
- }
-
- ht->buckets[b].key = strcache(key);
- ht->buckets[b].hash = hash;
- ht->buckets[b].value = value;
- }
- void* hash_find(hashtable* ht, char* key) {
- hash_t hash = strhash(key);
- int b = hash_find_bucket(ht, hash, key);
-
- return ht->buckets[b].value;
- }
- // splits on whitespace
- char** strsplit(char* splitters, char* in) {
- char* e;
- int alloc = 32;
- int len = 0;
- char** list = malloc(alloc * sizeof(*list));
-
- for(char* s = in; *s;) {
- e = strpbrk(s, splitters);
- if(!e) e = s + strlen(s);
-
- if(len >= alloc - 1) {
- alloc *= 2;
- list = realloc(list, alloc* sizeof(*list));
- }
-
- list[len++] = strndup(s, e - s);
-
- e += strspn(e, splitters);
- s = e;
- }
-
- list[len] = NULL;
-
- return list;
- }
- struct child_process_info* exec_cmdline_pipe(char* cmdline);
- struct child_process_info* exec_process_pipe(char* exec_path, char* args[]);
- int mkdirp(char* path, mode_t mode);
- char* resolve_path(char* in, time_t* mtime_out);
- #define path_join(...) path_join_(PP_NARG(__VA_ARGS__), __VA_ARGS__)
- char* path_join_(size_t nargs, ...);
- #define concat_lists(...) concat_lists_(PP_NARG(__VA_ARGS__), __VA_ARGS__)
- char** concat_lists_(int nargs, ...);
- #define strjoin(j, ...) strjoin_(j, PP_NARG(__VA_ARGS__), __VA_ARGS__)
- char* strjoin_(char* joiner, size_t nargs, ...);
- #define strcatdup(...) strcatdup_(PP_NARG(__VA_ARGS__), __VA_ARGS__)
- char* strcatdup_(size_t nargs, ...);
- void recursive_glob(char* base_path, char* pattern, int flags, rglob* results);
- char* dir_name(char* path);
- char* base_name(char* path);
- size_t list_len(char** list);
- char* join_str_list(char* list[], char* joiner);
- char* sprintfdup(char* fmt, ...);
- char* read_whole_file(char* path, size_t* srcLen);
- static inline char* strskip(char* s, char* skip) {
- return s + strspn(s, skip);
- }
- #define FSU_EXCLUDE_HIDDEN (1<<0)
- #define FSU_NO_FOLLOW_SYMLINKS (1<<1)
- #define FSU_INCLUDE_DIRS (1<<2)
- #define FSU_EXCLUDE_FILES (1<<3)
- char* pkg_config(char** packages, char* opts) {
- char* tmp;
-
- char* pkgs = join_str_list(packages, " ");
-
- for(char* c = opts; *c; c++) {
- switch(*c) {
- case 'c':
- case 'C':
- case 'i':
- case 'I':
- tmp = strjoin(" ", "--cflags", pkgs);
- free(pkgs);
- pkgs = tmp;
- break;
-
- case 'l':
- case 'L':
- tmp = strjoin(" ", "--libs", pkgs);
- free(pkgs);
- pkgs = tmp;
- break;
- }
- }
-
- tmp = strjoin(" ", "pkg-config", pkgs);
- free(pkgs);
-
- FILE* f = popen(tmp, "r");
- free(tmp);
- if(!f) {
- fprintf(stderr, "Could not run command '%s'\n", tmp);
- exit(1);
- return NULL;
- }
-
- int len = 2048;
- int fill = 0;
- char* buffer = malloc(len * sizeof(*buffer));
- while(!feof(f)) {
- if(fill + 1 >= len) {
- len *= 2;
- buffer = realloc(buffer, len * sizeof(*buffer));
- }
- fill += fread(buffer + fill, 1, len - fill - 1, f);
- }
-
- buffer[fill] = 0;
- pclose(f);
-
- // strip out newlines and other garbage
- for(char* c = buffer; *c; c++) if(isspace(*c)) *c = ' ';
-
- return buffer;
- }
- // return 0 to continue, nonzero to stop all directory scanning
- typedef int (*readDirCallbackFn)(char* /*fullPath*/, char* /*fileName*/, unsigned char /*type*/, void* /*data*/);
- // returns negative on error, nonzero if scanning was halted by the callback
- int recurse_dirs(
- char* path,
- readDirCallbackFn fn,
- void* data,
- int depth,
- unsigned int flags
- );
- typedef struct realname_entry {
- char* realname;
- time_t mtime;
- } realname_entry;
- struct {
- /*
- struct {
- int len;
- int alloc;
- struct {
- char* fake_name;
- realname_entry* entry;
- }* entries;
- } lookup; */
- hashtable names;
- int len;
- int alloc;
- struct realname_entry* entries;
- } realname_cache;
- void realname_cache_init();
- time_t realname_cache_add(char* fake_name, char* real_name);
- realname_entry* realname_cache_search_real(char* real_name);
- realname_entry* realname_cache_search(char* fake_name);
- char* realname_cache_find(char* fake_name);
- hashtable mkdir_cache;
- strlist compile_cache;
- void mkdirp_cached(char* path, mode_t mode) {
- void* there = hash_find(&mkdir_cache, path);
- if(!there) {
- hash_insert(&mkdir_cache, path, NULL);
- mkdirp(path, mode);
- }
- }
- void read_cpi(struct child_process_info* cpi) {
- while(1) {
- if(cpi->buf_len > cpi->buf_alloc - 128) {
- cpi->buf_alloc *= 2;
- cpi->output_buffer = realloc(cpi->output_buffer, cpi->buf_alloc * sizeof(*cpi->output_buffer));
- }
-
- int ret = read(cpi->pty, cpi->output_buffer + cpi->buf_len, cpi->buf_alloc - cpi->buf_len - 1);
- if(ret > 0) cpi->buf_len += ret;
- else return;
- }
-
- return ;
- }
- char* printpct(float f) {
- static char buf[32];
- if(f != 100) sprintf(buf, "\e[1;33m %2.0f%%\e[0m", f);
- else sprintf(buf, "\e[1;32m100%%\e[0m", f);
-
- return buf;
- }
- int execute_mt(strlist* cmds, int threads, char* fmt, struct child_process_info*** cpis) {
- int ret = 0;
- int running = 0;
-
- struct child_process_info** procs = calloc(1, cmds->len * sizeof(*procs));
- if(cpis) *cpis = procs;
-
- if(fmt) printf(fmt, printpct(0)), fflush(stdout);
-
- int waiting = cmds->len;
- while(waiting > 0) {
- for(int i = 0; i < cmds->len; i++) {
-
- // keep the cores full
- if(!procs[i] && running < threads) {
- procs[i] = exec_cmdline_pipe(cmds->entries[i]);
- procs[i]->state = 'r';
- running++;
- }
-
- if(!procs[i] || procs[i]->state == 'd') continue;
-
-
- read_cpi(procs[i]);
-
- int status;
- // returns 0 if nothing happened, -1 on error, childpid if it exited
- int pid = waitpid(procs[i]->pid, &status, WNOHANG);
- if(pid != 0) {
- procs[i]->state = 'd';
- waiting--;
- running--;
-
- if(fmt) printf("\r"), printf(fmt, printpct((cmds->len - waiting) * 100.0f / (float)cmds->len)), fflush(stdout);
-
- read_cpi(procs[i]);
-
- procs[i]->output_buffer[procs[i]->buf_len] = 0;
- procs[i]->exit_status = WEXITSTATUS(status);
-
- if(!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
- // printf("error on pid %d/%d = %d\n", pid, procs[i]->pid, WEXITSTATUS(status));
- // printf("%s\n", procs[i]->output_buffer);
- ret = 1;
- }
- }
-
-
- }
-
-
- usleep(100);
- }
-
- if(fmt) {
- printf("\r");
- if(ret) printf(fmt, "\e[1;31mFAIL\e[0m");
- else printf(fmt, "\e[32mDONE\e[0m");
- fflush(stdout);
- }
-
- printf("\n");
-
-
- return ret;
- }
- int compile_cache_execute() {
- int ret = 0;
- struct child_process_info** cpis;
- // printf("compile cache length %d", compile_cache.len);
- ret = execute_mt(&compile_cache, g_nprocs, "Compiling... %s", &cpis);
-
- if(ret) {
- for(int i = 0; i < compile_cache.len; i++ ) {
- if(cpis[i]->exit_status) {
- printf("%.*s\n", (int)cpis[i]->buf_len, cpis[i]->output_buffer);
- }
- }
-
- }
-
- // TODO free compile cache
- // TODO free cpis
- return ret;
- }
- size_t span_path(char* s) {
- size_t n = 0;
- for(; *s; s++, n++) {
- if(isspace(*s)) break;
- if(*s == '\\') {
- s++;
- n++;
- }
- }
- return n;
- }
- strlist* parse_gcc_dep_file(char* dep_file_path, time_t* newest_mtime) {
- size_t dep_src_len = 0;
- strlist* dep_list;
- time_t newest = 0;
-
- char* dep_src = read_whole_file(dep_file_path, &dep_src_len);
- if(!dep_src) return NULL;
-
- dep_list = strlist_new();
-
- // skip the first filename junk
- char* s = strchr(dep_src, ':');
- s++;
-
- int ret = 0;
-
- // gather dep strings, ignoring line continuations
- while(*s) {
- do {
- s = strskip(s, " \t\r\n");
- if(*s == '\\') {
- if(s[1] == '\r') s++;
- if(s[1] == '\n') s++;
- }
- } while(isspace(*s));
-
- int dlen = span_path(s);
- if(dlen == 0) break;
-
- time_t dep_mtime;
- char* dep_fake = strncache(s, dlen);
- char* dep_real = resolve_path(dep_fake, &dep_mtime);
- if(dep_mtime > newest) newest = dep_mtime;
-
- strlist_push(dep_list, dep_real);
-
- struncache(dep_fake);
- struncache(dep_real);
-
- s += dlen;
- }
-
- free(dep_src);
-
- if(newest_mtime) *newest_mtime = newest;
- return dep_list;
- }
- int gen_deps(char* src_path, char* dep_path, time_t src_mtime, time_t obj_mtime) {
- time_t dep_mtime = 0;
- time_t newest_mtime = 0;
-
- char* real_dep_path = resolve_path(dep_path, &dep_mtime);
- if(dep_mtime < src_mtime) {
- //gcc -MM -MG -MT $1 -MF "build/$1.d" $1 $CFLAGS $LDADD
- // printf(" generating deps\n");
- char* cmd = sprintfdup("gcc -MM -MG -MT '' -MF %s %s %s", dep_path, src_path, g_gcc_opts_flat);
- system(cmd);
- free(cmd);
- }
-
- strlist* deps = parse_gcc_dep_file(real_dep_path, &newest_mtime);
-
- // free or process deps
-
- return newest_mtime > obj_mtime;
- FAIL:
- return 0;
- }
- size_t list_len(char** list) {
- size_t total = 0;
- for(; *list; list++) total++;
- return total;
- }
- char** concat_lists_(int nargs, ...) {
- size_t total = 0;
- char** out, **end;
- if(nargs == 0) return NULL;
- // calculate total list length
- va_list va;
- va_start(va, nargs);
- for(size_t i = 0; i < nargs; i++) {
- char** s = va_arg(va, char**);
- if(s) total += list_len(s);
- }
- va_end(va);
- out = malloc((total + 1) * sizeof(char**));
- end = out;
- va_start(va, nargs);
-
- // concat lists
- for(size_t i = 0; i < nargs; i++) {
- char** s = va_arg(va, char**);
- size_t l = list_len(s);
-
- if(s) {
- memcpy(end, s, l * sizeof(*s));
- end += l;
- }
- }
- va_end(va);
- *end = 0;
- return out;
- }
- char* join_str_list(char* list[], char* joiner) {
- size_t list_len = 0;
- size_t total = 0;
- size_t jlen = strlen(joiner);
-
- // calculate total length
- for(int i = 0; list[i]; i++) {
- list_len++;
- total += strlen(list[i]);
- }
-
- if(total == 0) return strdup("");
-
- total += (list_len - 1) * jlen;
- char* out = malloc((total + 1) * sizeof(*out));
-
- char* end = out;
- for(int i = 0; list[i]; i++) {
- char* s = list[i];
- size_t l = strlen(s);
-
- if(i > 0) {
- memcpy(end, joiner, jlen);
- end += jlen;
- }
-
- if(s) {
- memcpy(end, s, l);
- end += l;
- }
-
- total += strlen(list[i]);
- }
-
- *end = 0;
-
- return out;
- }
- // concatenate all argument strings together in a new buffer
- char* strcatdup_(size_t nargs, ...) {
- size_t total = 0;
- char* out, *end;
-
- if(nargs == 0) return NULL;
-
- // calculate total buffer len
- va_list va;
- va_start(va, nargs);
-
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- if(s) total += strlen(s);
- }
-
- va_end(va);
-
- out = malloc((total + 1) * sizeof(char*));
- end = out;
-
- va_start(va, nargs);
-
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- if(s) {
- strcpy(end, s); // not exactly the ost efficient, but maybe faster than
- end += strlen(s); // a C version. TODO: test the speed
- };
- }
-
- va_end(va);
-
- *end = 0;
-
- return out;
- }
- // concatenate all argument strings together in a new buffer,
- // with the given joining string between them
- char* strjoin_(char* joiner, size_t nargs, ...) {
- size_t total = 0;
- char* out, *end;
- size_t j_len;
-
- if(nargs == 0) return NULL;
-
- // calculate total buffer len
- va_list va;
- va_start(va, nargs);
-
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- if(s) total += strlen(s);
- }
-
- va_end(va);
-
- j_len = strlen(joiner);
- total += j_len * (nargs - 1);
-
- out = malloc((total + 1) * sizeof(char*));
- end = out;
-
- va_start(va, nargs);
-
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- if(s) {
- if(i > 0) {
- strcpy(end, joiner);
- end += j_len;
- }
-
- strcpy(end, s); // not exactly the ost efficient, but maybe faster than
- end += strlen(s); // a C version. TODO: test the speed
- };
- }
-
- va_end(va);
-
- *end = 0;
-
- return out;
- }
- // allocates a new buffer and calls sprintf with it
- // why isn't this a standard function?
- char* sprintfdup(char* fmt, ...) {
- va_list va;
-
- va_start(va, fmt);
- size_t n = vsnprintf(NULL, 0, fmt, va);
- char* buf = malloc(n + 1);
- va_end(va);
-
- va_start(va, fmt);
- vsnprintf(buf, n + 1, fmt, va);
- va_end(va);
-
- return buf;
- }
- char* dir_name(char* path) {
- char* n = strdup(path);
- char* o = dirname(n);
- return strcache(o);
- }
- char* base_name(char* path) {
- char* n = strdup(path);
- char* o = basename(n);
- return strcache(o);
- }
- int rglob_fn(char* full_path, char* file_name, unsigned char type, void* _results) {
- rglob* res = (rglob*)_results;
-
- if(0 == fnmatch(res->pattern, file_name, 0)) {
- check_alloc(res);
-
- res->entries[res->len].type = type;
- res->entries[res->len].full_path = strcache(full_path);
- res->entries[res->len].file_name = strcache(file_name);
- res->len++;
- }
-
- return 0;
- }
- void recursive_glob(char* base_path, char* pattern, int flags, rglob* results) {
-
- // to pass into recurse_dirs()
- results->pattern = pattern;
- results->len = 0;
- results->alloc = 32;
- results->entries = malloc(sizeof(*results->entries) * results->alloc);
-
- recurse_dirs(base_path, rglob_fn, results, -1, flags);
- }
- // does not handle escaped slashes
- int mkdirp(char* path, mode_t mode) {
-
- char* clean_path = strdup(path);
-
- // inch along the path creating each directory in line
- for(char* p = clean_path; *p; p++) {
- if(*p == '/') {
- *p = 0;
-
- if(mkdir(clean_path, mode)) {
- if(errno != EEXIST) goto FAIL;
- }
-
- *p = '/';
- }
- }
-
- // mop up the last dir
- if(mkdir(clean_path, mode)) {
- if(errno != EEXIST) goto FAIL;
- }
-
- free(clean_path);
- return 0;
-
- FAIL:
- free(clean_path);
- return -1;
- }
- void realname_cache_init() {
- realname_cache.len = 0;
- realname_cache.alloc = 1024;
- realname_cache.entries = malloc(realname_cache.alloc * sizeof(*realname_cache.entries));
- hash_init(&realname_cache.names, 1024);
- }
- time_t realname_cache_add(char* fake_name, char* real_name) {
- realname_entry* e = hash_find(&realname_cache.names, fake_name);
- if(e) return e->mtime;
-
- e = hash_find(&realname_cache.names, real_name);
- if(!e) {
- struct stat st;
- lstat(real_name, &st);
-
- e = &realname_cache.entries[realname_cache.len];
- e->realname = strcache(real_name);
- e->mtime = st.st_mtim.tv_sec;
- realname_cache.len++;
-
- hash_insert(&realname_cache.names, real_name, e);
- }
-
- hash_insert(&realname_cache.names, fake_name, e);
-
- return e->mtime;
- }
- realname_entry* realname_cache_search_real(char* real_name) {
- for(int i = 0; i < realname_cache.len; i++) {
- if(0 == strcmp(real_name, realname_cache.entries[i].realname)) {
- return &realname_cache.entries[i];
- }
- }
-
- return NULL;
- }
- realname_entry* realname_cache_search(char* fake_name) {
- return hash_find(&realname_cache.names, fake_name);
- }
- char* realname_cache_find(char* fake_name) {
- realname_entry* r = realname_cache_search(fake_name);
- return r ? r->realname : NULL;
- }
- // works like realpath(), except also handles ~/
- char* resolve_path(char* in, time_t* mtime_out) {
- int tmp_was_malloced = 0;
- char* out, *tmp;
-
- if(!in) return NULL;
-
- realname_entry* e = realname_cache_search(in);
- if(e) {
- if(mtime_out) *mtime_out = e->mtime;
- return strcache(e->realname);
- }
-
- // skip leading whitespace
- while(isspace(*in)) in++;
-
- // handle home dir shorthand
- if(in[0] == '~') {
- char* home = getenv("HOME");
-
- tmp_was_malloced = 1;
- tmp = malloc(sizeof(*tmp) * (strlen(home) + strlen(in) + 2));
-
- strcpy(tmp, home);
- strcat(tmp, "/"); // just in case
- strcat(tmp, in + 1);
- }
- else tmp = in;
-
- out = realpath(tmp, NULL);
-
- if(tmp_was_malloced) free(tmp);
-
- time_t t = 0;
- if(out) {
- // put it in the cache
- t = realname_cache_add(in, out);
- }
- else {
- // temporary
- struct stat st;
- if(!lstat(in, &st))
- t = st.st_mtim.tv_sec;
- }
-
- if(mtime_out) *mtime_out = t;
- return out ? out : in;
- }
- char* path_join_(size_t nargs, ...) {
- size_t total = 0;
- char* out, *end;
- size_t j_len;
- char* joiner = "/";
- int escape;
- if(nargs == 0) return NULL;
- // calculate total buffer length
- va_list va;
- va_start(va, nargs);
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- if(s) total += strlen(s);
- }
- va_end(va);
- j_len = strlen(joiner);
- total += j_len * (nargs - 1);
- out = malloc((total + 1) * sizeof(char*));
- end = out;
- va_start(va, nargs);
- for(size_t i = 0; i < nargs; i++) {
- char* s = va_arg(va, char*);
- size_t l = strlen(s);
-
- if(s) {
- if(l > 1) {
- escape = s[l-2] == '\\' ? 1 : 0;
- }
- if(i > 0 && (s[0] == joiner[0])) {
- s++;
- l--;
- }
- if(i > 0 && i != nargs-1 && !escape && (s[l-1] == joiner[0])) {
- l--;
- }
- if(i > 0) {
- strcpy(end, joiner);
- end += j_len;
- }
- // should be strncpy, but GCC is so fucking stupid that it
- // has a warning about using strncpy to do exactly what
- // strncpy does if you read the fucking man page.
- // fortunately, we are already terminating our strings
- // manually so memcpy is a drop-in replacement here.
- memcpy(end, s, l);
- end += l;
- }
- }
- va_end(va);
- *end = 0;
- return out;
- }
- // returns negative on error, nonzero if scanning was halted by the callback
- int recurse_dirs(
- char* path,
- readDirCallbackFn fn,
- void* data,
- int depth,
- unsigned int flags
- ) {
-
- DIR* derp;
- struct dirent* result;
- int stop = 0;
-
- if(fn == NULL) {
- fprintf(stderr, "Error: readAllDir called with null function pointer.\n");
- return -1;
- }
-
- derp = opendir(path);
- if(derp == NULL) {
- fprintf(stderr, "Error opening directory '%s': %s\n", path, strerror(errno));
- return -1;
- }
-
-
- while((result = readdir(derp)) && !stop) {
- char* n = result->d_name;
- unsigned char type = DT_UNKNOWN;
- char* fullPath;
-
- // skip self and parent dir entries
- if(n[0] == '.') {
- if(n[1] == '.' && n[2] == 0) continue;
- if(n[1] == 0) continue;
-
- if(flags & FSU_EXCLUDE_HIDDEN) continue;
- }
- #ifdef _DIRENT_HAVE_D_TYPE
- type = result->d_type; // the way life should be
- #else
- // do some slow extra bullshit to get the type
- fullPath = path_join(path, n);
-
- struct stat upgrade_your_fs;
-
- lstat(fullPath, &upgrade_your_fs);
-
- if(S_ISREG(upgrade_your_fs.st_mode)) type = DT_REG;
- else if(S_ISDIR(upgrade_your_fs.st_mode)) type = DT_DIR;
- else if(S_ISLNK(upgrade_your_fs.st_mode)) type = DT_LNK;
- #endif
-
- if(flags & FSU_NO_FOLLOW_SYMLINKS && type == DT_LNK) {
- continue;
- }
-
- #ifdef _DIRENT_HAVE_D_TYPE
- fullPath = path_join(path, n);
- #endif
-
- if(type == DT_DIR) {
- if(flags & FSU_INCLUDE_DIRS) {
- stop = fn(fullPath, n, type, data);
- }
- if(depth != 0) {
- stop |= recurse_dirs(fullPath, fn, data, depth - 1, flags);
- }
- }
- else if(type == DT_REG) {
- if(!(flags & FSU_EXCLUDE_FILES)) {
- stop = fn(fullPath, n, type, data);
- }
- }
-
- free(fullPath);
- }
-
-
- closedir(derp);
-
- return stop;
- }
- struct child_process_info* exec_cmdline_pipe(char* cmdline) {
- // kinda janky. it doesn't handle quotes
- char** args = strsplit(" ", cmdline);
- struct child_process_info* cpi = exec_process_pipe(args[0], args);
- // printf("executing '%s'\n", cmdline);
- // for(char** s = args; *s; s++) printf("%d - '%s'\n", (s-args), *s);
- free_strpp(args);
-
- return cpi;
- }
- // effectively a better, asynchronous version of system()
- // redirects and captures the child process i/o
- struct child_process_info* exec_process_pipe(char* exec_path, char* args[]) {
-
-
- int master, slave; // pty
-
-
- errno = 0;
- if(openpty(&master, &slave, NULL, NULL, NULL) < 0) {
- fprintf(stderr, "Error opening new pty for '%s' [errno=%d]\n", exec_path, errno);
- return NULL;
- }
-
- errno = 0;
-
- int childPID = fork();
- if(childPID == -1) {
-
- fprintf(stderr, "failed to fork trying to execute '%s'\n", exec_path);
- perror(strerror(errno));
- return NULL;
- }
- else if(childPID == 0) { // child process
-
- setsid();
-
- // redirect standard fd's to the pipe fd's
- if(dup2(slave, fileno(stdin)) == -1) {
- printf("failed 1\n");
- exit(errno);
- }
- if(dup2(slave, fileno(stdout)) == -1) {
- printf("failed 2\n");
- exit(errno);
- }
- if(dup2(slave, fileno(stderr)) == -1) {
- printf("failed 3\n");
- exit(errno);
- }
-
- if(ioctl(slave, TIOCSCTTY, NULL) < 0) {
- fprintf(stderr, "ioctl TIOCSCTTY failed: %s, %d\n", exec_path, errno);
- }
-
- // close original fd's
- close(master);
- close(slave);
-
- // die when the parent does (linux only)
- prctl(PR_SET_PDEATHSIG, SIGHUP);
-
- // swap for the desired program
- execvp(exec_path, args); // never returns if successful
-
- fprintf(stderr, "failed to execute '%s'\n", exec_path);
- exit(1); // kill the forked process
- }
- else { // parent process
-
- // close the child-end of the pipes
- struct child_process_info* cpi;
- cpi = calloc(1, sizeof(*cpi));
- cpi->pid = childPID;
- cpi->pty = master;
-
- cpi->state = 'q';
- cpi->buf_alloc = 4096;
- cpi->output_buffer = malloc(cpi->buf_alloc * sizeof(*cpi->output_buffer));
-
- // set to non-blocking
- fcntl(master, F_SETFL, fcntl(master, F_GETFL) | FNDELAY | O_NONBLOCK);
-
- close(slave);
-
- // tcsetattr(STDIN_FILENO, TCSANOW, &master);
- // fcntl(master, F_SETFL, FNDELAY);
-
- // int status;
- // returns 0 if nothing happened, -1 on error
- // pid = waitpid(childPID, &status, WNOHANG);
-
- return cpi;
- }
-
- return NULL; // shouldn't reach here
- }
- char* read_whole_file(char* path, size_t* srcLen) {
- size_t fsize, total_read = 0, bytes_read;
- char* contents;
- FILE* f;
-
-
- f = fopen(path, "rb");
- if(!f) {
- fprintf(stderr, "Could not open file \"%s\"\n", path);
- return NULL;
- }
-
- fseek(f, 0, SEEK_END);
- fsize = ftell(f);
- rewind(f);
-
- contents = malloc(fsize + 1);
-
- while(total_read < fsize) {
- bytes_read = fread(contents + total_read, sizeof(char), fsize - total_read, f);
- total_read += bytes_read;
- }
-
- contents[fsize] = 0;
-
- fclose(f);
-
- if(srcLen) *srcLen = fsize;
-
- return contents;
- }
|