*
* 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
#include <string.h>
#include <sys/mman.h>
#include <sys/stat.h>
+#include <sys/wait.h>
#include <unistd.h>
#include "config.h"
char * buf;
} string_t;
-const static string_t START = { 5, "@BAKE" };
-const static string_t STOP = { 5, "@STOP" };
+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;
return NULL;
}
-static char *
-insert(char * new, size_t len, char * str, size_t slen, size_t offset, size_t shift) {
- size_t max;
- max = (slen + 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 ***/
#define GLOBALS_COUNT 3
-static char * globals[GLOBALS_COUNT];
+static string_t globals[GLOBALS_COUNT];
#define g_filename globals[0]
#define g_short globals[1]
#define g_all globals[2]
-/* doing a ? strlen(a) : "" really bothered me and this could be optimized out */
-static size_t *
-get_globals_length(void) {
- static size_t len[GLOBALS_COUNT];
- size_t i;
- for (i = 0; i < GLOBALS_COUNT; ++i) {
- len[i] = strlen(globals[i]);
- }
- return len;
-}
-
-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
+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 ? sh : calloc(1,1);
+ return (string_t) { last, sh ? sh : calloc(0,0) };
}
-static char *
-all_args(size_t argc, char ** argv) {
+static string_t
+all_args(int argc, char ** argv) {
char * all = NULL;
+ size_t len = 0;
if (argc > 2) {
- size_t i, len = argc;
+ int i;
+ len = (size_t) argc;
for (i = 2; i < argc; ++i)
{ len += strlen(argv[i]); }
- all = malloc(len + 1);
- if (!all) { return NULL; }
+ all = malloc(len);
+ if (all) {
+ all[len - 1] = '\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] = ' '; }
+ }
+ }
+ }
+ return (string_t) { len, all ? all : calloc(0,0) };
+}
- 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] = ' '; }
+#if 0
+static string_t
+all_args(int argc, char ** argv) {
+ string_t s = (string_t) { 0, NULL };
+ if (argc > 2) {
+ size_t i;
+ for (i = (size_t) argc; i > 2; --i) {
+ s.len += strlen(argv[i]) + 1;
+ }
+ s.buf = malloc(s.len + (size_t) argc);
+ if (s.buf) {
+ size_t len;
+ s.buf[s.len - 1] = '\0';
+ for (len = 0, i = (size_t) argc; i > 2; --i) {
+ strcpy(s.buf + len, argv[i]);
+ len += strlen(argv[i]) + 1;
+ s.buf[len - 1] = ' ';
+ }
}
}
- return all ? all : calloc(1,1);
+ return s;
}
+#endif
/*** Map ***/
-typedef struct {
- char * str;
- size_t len;
-} map_t;
-
static map_t
map(char * fn) {
struct stat s;
&& 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);
}
/*** Important Functions ***/
-static char *
-find_region(map_t m, size_t * retlen) {
+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 += START.len;
+ 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;
if (stop)
{
- *retlen = (size_t) (stop - m.str) - (start - m.str);
- buf = strndup(start, *retlen);
+ 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] != '/')
ret = chdir(root);
swap(root + len, x);
- *rootp += len + 1;
+/* *rootp += len + 1; */
return ret;
}
#define ARRLEN(x) (sizeof(x) / sizeof(x[0]))
-static char *
-expand(char * buf, size_t len) {
+static string_t
+expand(string_t s) {
enum {
MACRO_FILENAME = 0,
MACRO_SHORT = 1,
MACRO_ARGS = 2,
};
- char *macro[] = {
- [MACRO_FILENAME] = "$@",
- [MACRO_SHORT ] = "$*",
- [MACRO_ARGS ] = "$+",
- };
-
- size_t macro_length[] = {
- [MACRO_FILENAME] = 2,
- [MACRO_SHORT ] = 2,
- [MACRO_ARGS ] = 2,
+ string_t macro[] = {
+ [MACRO_FILENAME] = { 2, "$@" },
+ [MACRO_SHORT ] = { 2, "$*" },
+ [MACRO_ARGS ] = { 2, "$+" },
};
- size_t * globals_length = get_globals_length();
-
- char * ptr;
size_t i, f;
{
- size_t max = len + 1;
- for (i = 0; i < len; ++i) {
+ size_t max = s.len;
+ for (i = 0; i < s.len; ++i) {
for (f = 0; f < ARRLEN(macro); ++f) {
- if (!strncmp(buf + i, macro[f], macro_length[f])) {
- max += globals_length[f];
+ if (!strncmp(s.buf + i, macro[f].buf, macro[f].len)) {
+ max += globals[f].len;
}
}
}
- buf = realloc(buf, max + 10);
- len = max - 1;
+ 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 < len; ++i) {
+ for (i = 0; i < s.len; ++i) {
for (f = 0; f < ARRLEN(macro); ++f) {
- if (!strncmp(buf + i, macro[f], macro_length[f])) {
- buf = insert(globals[f], globals_length[f], buf, len, i, 2);
+ 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 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) {
+ int ret = 127;
fputs(GREEN "output" RESET ":\n", stderr);
- return system(buf);
+ pid_t pid = fork();
+ if (!pid) {
+ execl("/bin/sh", "-c", "'ls bake.c'", 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;
- size_t len;
+ 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, &len);
- munmap(m.str, m.len);
- }
- }
+ s = file_find_region(g_filename.buf, START, STOP);
- if (!buf) {
+ if (!s.buf) {
if (errno)
- { fprintf(stderr, BOLD RED "%s" RESET ": %s\n", g_filename, strerror(errno)); }
+ { fprintf(stderr, BOLD RED "%s" RESET ": %s\n", g_filename.buf, strerror(errno)); }
else
- { fprintf(stderr, BOLD RED "%s" RESET ": File unrecognized.\n", g_filename); }
+ { fprintf(stderr, BOLD RED "%s" RESET ": File unrecognized.\n", g_filename.buf); }
return 1;
}
- g_all = all_args((size_t) argc, argv);
+ g_all = all_args(argc, argv);
g_short = shorten(g_filename);
- root(&g_filename);
- buf = expand(buf, len);
- free(g_short); free(g_all);
- if (!buf) { return 1; }
+ root(g_filename);
+
+ s = expand(s);
+ free(g_short.buf); free(g_all.buf);
+ if (!s.buf) { return 1; }
- fprintf(stderr, GREEN "%s" RESET ": %s\n", argv[0], buf + strip(buf));
- ret = ret ? 0 : run(buf);
+ fprintf(stderr, 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, RED "result" RESET ": " BOLD "%d, %s\n" RESET, errno, strerror(errno)); }
- free(buf);
+ free(s.buf);
return ret;
help:
fprintf(stderr, YELLOW "%s" RESET ": %s", argv[0], HELP DESC);