diff --git a/bake.c b/bake.c index 60928f0..d133736 100644 --- a/bake.c +++ b/bake.c @@ -3,7 +3,7 @@ * * Licensed under the GNU Public License version 3 only, see LICENSE. * - * @BAKE cc -std=c89 -O2 -I. $@ -o $* $+ # @STOP + * @BAKE cc -std=c99 -O2 $@ -o $* $+ # @STOP */ #define _POSIX_C_SOURCE 200809L @@ -20,13 +20,11 @@ #include #include #include +#include #include #include "config.h" -#define START "@BAKE" -#define STOP "@STOP" - #define HELP \ BOLD "[option] target-file" RESET " [" GREEN "arguments" RESET " ...]\n" \ "Use the format `" BOLD "@BAKE" RESET " cmd ...' within the target-file, this will execute the\n" \ @@ -40,8 +38,21 @@ "\t" YELLOW "$*" RESET " returns target-file without suffix (^-> abc.x)\n" \ "\t" YELLOW "$+" RESET " returns " GREEN "arguments" RESET "\n" +typedef struct { + size_t len; + char * buf; +} string_t; + +typedef string_t map_t; + +static string_t START = { 5, "@BAKE" }; +static string_t STOP = { 5, "@STOP" }; + /*** Utility functions ***/ +#define strneq(a,b,n) (!strncmp(a,b,n)) +#define streq(a,b) (!strcmp(a,b)) + static void swap(char * a, char * b) { *a ^= *b; @@ -50,74 +61,72 @@ swap(char * a, char * b) { } static char * -find(char * x, char * buf, char * end) { - size_t len = strlen(x); - for (; (buf < end) && len < (size_t)(end - buf); ++buf) { - if (!strncmp(buf, x, len)) +find(string_t x, char * buf, char * end) { + for (; (buf < end) && x.len < (size_t)(end - buf); ++buf) { + if (!strncmp(buf, x.buf, x.len)) { return buf; } } return NULL; } -static char * -insert(char * new, char * str, size_t offset, size_t shift) { - size_t len, max; - len = strlen(new); - max = (strlen(str) + 1 - offset - shift); - memmove(str + offset + len, str + offset + shift, max); - memcpy(str + offset, new, len); - return str; +static void +insert(char * str, size_t slen, char * new, size_t len, size_t shift) { + memmove(str + len, str + shift, slen - len - shift); + memcpy(str, new, len); } /*** g_short, g_all Functions ***/ -static char * g_filename, * g_short, * g_all; +#define GLOBALS_COUNT 3 -static char * -shorten(char * fn) { - size_t i, last = 0, len; - char * sh; - len = strlen(fn); - sh = malloc(len + 1); - if (!sh) { return NULL; } - for (i = 0; i < len; ++i) { - if (fn[i] == '.') { last = i; } +static string_t globals[GLOBALS_COUNT]; + +#define g_filename globals[0] +#define g_short globals[1] +#define g_all globals[2] + +static string_t +shorten(string_t s) { + size_t i, last = 0, len = s.len; + char * sh, * fn = s.buf; + sh = malloc(len); + if (sh) { + for (i = 0; i < len; ++i) { + if (fn[i] == '.') { last = i; } + } + last = last ? last : i; + strncpy(sh, fn, last); + sh[last] = '\0'; } - last = last ? last : i; - strncpy(sh, fn, last); - sh[last] = '\0'; - return sh; + return (string_t) { last, sh ? sh : calloc(0,0) }; } -static char * -all_args(size_t argc, char ** argv) { - char * all = NULL; +static string_t +all_args(int argc, char ** argv) { + string_t s = (string_t) { 0, NULL }; if (argc > 2) { - size_t i, len = argc; - - for (i = 2; i < argc; ++i) - { len += strlen(argv[i]); } - - all = malloc(len + 1); - if (!all) { return NULL; } - - all[len] = '\0'; - for (len = 0, i = 2; i < argc; ++i) { - strcpy(all + len, argv[i]); - len += strlen(argv[i]) + 1; - if (i + 1 < argc) { all[len - 1] = ' '; } + size_t i, len = 0; + for (i = 2; i < (size_t) argc; ++i) { + len += strlen(argv[i]); + } + s.buf = malloc(len); + s.len = len; + if (s.buf) { + for (len = 0, i = 2; i < (size_t) argc; ++i) { + strcpy(s.buf + len, argv[i]); + len += strlen(argv[i]); + if (i + 1 < argc) { + s.buf[len - 1] = ' '; + len++; + } + } } } - return all; + return s; } /*** Map ***/ -typedef struct { - char * str; - size_t len; -} map_t; - static map_t map(char * fn) { struct stat s; @@ -129,7 +138,7 @@ map(char * fn) { && s.st_mode & S_IFREG && s.st_size) { m.len = (size_t) s.st_size; - m.str = (char *) mmap(NULL, m.len, PROT_READ, MAP_SHARED, fd, 0); + m.buf = (char *) mmap(NULL, m.len, PROT_READ, MAP_SHARED, fd, 0); } close(fd); } @@ -138,24 +147,25 @@ map(char * fn) { /*** Important Functions ***/ -static char * -find_region(map_t m) { +static string_t +find_region(map_t m, string_t startsym, string_t stopsym) { extern char * strndup(const char * s, size_t n); /* for splint */ char * buf = NULL, * start, * stop; + size_t len; - start = find(START, m.str, m.str + m.len); + start = find(startsym, m.buf, m.buf + m.len); if (start) { - start += strlen(START); + start += startsym.len; #ifdef REQUIRE_SPACE if (!isspace(*start)) { - fprintf(stderr, RED "%s" RESET ": Found start without suffix spacing.\n", g_filename); - return buf; + fprintf(stderr, RED "%s" RESET ": Found start without suffix spacing.\n", g_filename.buf); + return (string_t) { 0 , buf }; } #endif /* REQUIRE_SPACE */ - stop = find(STOP, start, start + m.len - (start - m.str)); + stop = find(stopsym, start, start + m.len - (start - m.buf)); if (!stop) { stop = start; @@ -167,16 +177,30 @@ find_region(map_t m) { } if (stop) - { buf = strndup(start, (size_t) (stop - m.str) - (start - m.str)); } + { + len = (size_t) (stop - m.buf) - (start - m.buf); + buf = strndup(start, len); + } } - return buf; + return (string_t) { len, buf }; +} + +static string_t +file_find_region(char * fn, string_t start, string_t stop) { + string_t s = { 0, NULL }; + map_t m = map(fn); + if (m.buf) { + s = find_region(m, start, stop); + munmap(m.buf, m.len); + } + return s; } static int -root(char ** rootp) { +root(string_t s) { char x[1] = {'\0'}; - char * root = *rootp; - size_t len = strlen(root); + char * root = s.buf; + size_t len = s.len; int ret; while (len && root[len] != '/') @@ -188,139 +212,131 @@ root(char ** rootp) { ret = chdir(root); swap(root + len, x); - *rootp += len + 1; +/* *rootp += len + 1; */ return ret; } -static size_t -expand_size(char * buf, int argc, char ** argv) { - size_t i, len, max; +#define ARRLEN(x) (sizeof(x) / sizeof(x[0])) - len = max = strlen(buf) + 1; +static string_t +expand(string_t s) { + enum { + MACRO_FILENAME = 0, + MACRO_SHORT = 1, + MACRO_ARGS = 2, + }; - for (i = 0; i < len; ++i) { - if (buf[i] == '\\') { - i += 2; - continue; - } else if (buf[i] == '$') { - switch (buf[++i]) { - case '@': - max += strlen(g_filename); - break; - case '*': - if (!g_short) - { g_short = shorten(g_filename); } - max += g_short ? strlen(g_short) : 0; - break; - case '+': - if (!g_all) - { g_all = all_args((size_t) argc, argv); } - max += g_all ? strlen(g_all) : 0; - break; + string_t macro[] = { + [MACRO_FILENAME] = { 2, "$@" }, + [MACRO_SHORT ] = { 2, "$*" }, + [MACRO_ARGS ] = { 2, "$+" }, + }; + + size_t i, f; + + { + size_t max = s.len; + for (i = 0; i < s.len; ++i) { + for (f = 0; f < ARRLEN(macro); ++f) { + if (!strncmp(s.buf + i, macro[f].buf, macro[f].len)) { + max += globals[f].len; + } + } + } + s.buf = realloc(s.buf, max + 27); /* I don't know, man */ + if (!s.buf) { return (string_t) { 0, NULL}; } + memset(s.buf + s.len, 0, max - s.len); + s.len = max; + } + for (i = 0; i < s.len; ++i) { + for (f = 0; f < ARRLEN(macro); ++f) { + if (!strncmp(s.buf + i, macro[f].buf, macro[f].len)) { + insert(s.buf + i, s.len - i, globals[f].buf, globals[f].len, 2); } } } - return max; -} - -static char * -expand(char * buf) { - size_t i; - char * ptr = NULL; - - for (i = 0; buf[i]; ++i) { - if (buf[i] == '\\') { - i += 2; - continue; - } else if (buf[i] == '$') { - switch (buf[++i]) { - case '@': - ptr = g_filename; - break; - case '*': - ptr = g_short; - break; - case '+': - ptr = g_all ? g_all : ""; - break; - default: continue; - } - buf = insert(ptr, buf, i - 1, 2); - } - } - free(g_short); free(g_all); - return buf; + return s; } /* Strips all prefixing and leading whitespace. * Except if the last character beforehand is a newline. */ static size_t -strip(char * buf) { - size_t i = strlen(buf); +strip(string_t s) { + size_t i = s.len; if (!i) { return 0; } - while (isspace(buf[i - 1])) + while (isspace(s.buf[i])) { --i; } - buf[i] = '\0'; - for (i = 0; isspace(buf[i]); ++i); + s.buf[i] = '\0'; + for (i = 0; isspace(s.buf[i]); ++i); - return i - (buf[i - 1] == '\n'); + return i - (s.buf[i - 1] == '\n'); } static int run(char * buf) { - fputs(GREEN "output" RESET ":\n", stderr); - return system(buf); + int ret = 127; + fputs(BOLD GREEN "output" RESET ":\n", stderr); + pid_t pid = fork(); + if (!pid) { + execl("/bin/sh", "sh", "-c", buf, NULL); + } else { + int status; + waitpid(pid, &status, 0); + if (!WIFEXITED(status)) { ret = 126; } + else { ret = WEXITSTATUS(status); } + } + return ret; } int main(int argc, char ** argv) { int ret = 0; - char * buf = NULL; + string_t s = { 0, NULL }; if (argc < 2 - || !strcmp(argv[1], "-h") - || !strcmp(argv[1], "--help")) + || streq(argv[1], "-h") + || streq(argv[1], "--help")) { goto help; } - g_filename = argv[1]; + g_filename = (string_t) { strlen(argv[1]), argv[1] }; - if (!strcmp(argv[1], "-n") - || !strcmp(argv[1], "--dry-run")) { - if (argc > 2) { ret = 1; g_filename = argv[2]; } + if (streq(argv[1], "-n") + || streq(argv[1], "--dry-run")) { + if (argc > 2) { + ret = 1; + g_filename = (string_t) { strlen(argv[2]), argv[2] }; + } else { goto help; } } - { map_t m = map(g_filename); - if (m.str) { - buf = find_region(m); - munmap(m.str, m.len); - } - } + s = file_find_region(g_filename.buf, START, STOP); - if (!buf) { - if (errno) { fprintf(stderr, BOLD RED "%s" RESET ": %s\n", g_filename, strerror(errno)); } - else { fprintf(stderr, BOLD RED "%s" RESET ": File unrecognized.\n", g_filename); } + if (!s.buf) { + if (errno) + { fprintf(stderr, BOLD RED "%s" RESET ": '" BOLD "%s" RESET "' %s\n", argv[0], g_filename.buf, strerror(errno)); } + else + { fprintf(stderr, BOLD RED "%s" RESET ": '" BOLD "%s" RESET "' File unrecognized.\n", argv[0], g_filename.buf); } return 1; } - root(&g_filename); - { char * buf2 = buf; - buf = realloc(buf, expand_size(buf, argc, argv)); - if (!buf) - { free(buf2); free(g_short); free(g_all); return 1; } - } - buf = expand(buf); + g_all = all_args(argc, argv); + g_short = shorten(g_filename); + root(g_filename); - fprintf(stderr, GREEN "%s" RESET ": %s\n", argv[0], buf + strip(buf)); - ret = ret ? 0 : run(buf); + s = expand(s); + free(g_short.buf); free(g_all.buf); + if (!s.buf) { return 1; } + + fprintf(stderr, BOLD GREEN "%s" RESET ": %s\n", argv[0], s.buf + strip(s)); + ret = ret ? 0 : run(s.buf); if (ret) - { fprintf(stderr, RED "result" RESET ": " BOLD "%d\n" RESET, ret); } + { fprintf(stderr, BOLD RED "result" RESET ": " BOLD "%d\n" RESET, ret); } - free(buf); + free(s.buf); return ret; help: fprintf(stderr, YELLOW "%s" RESET ": %s", argv[0], HELP DESC); diff --git a/config.h b/config.h index ac0d3fb..ae744d3 100644 --- a/config.h +++ b/config.h @@ -4,12 +4,12 @@ #define ENABLE_COLOR #ifdef ENABLE_COLOR -# define RED "\e[91m" -# define GREEN "\e[92m" -# define YELLOW "\e[93m" -# define DIM "\e[2m" -# define BOLD "\e[1m" -# define RESET "\e[0m" +# define RED "\033[91m" +# define GREEN "\033[92m" +# define YELLOW "\033[93m" +# define DIM "\033[2m" +# define BOLD "\033[1m" +# define RESET "\033[0m" #else # define RED # define GREEN