From d1bfbd97b15a604118efc6e3ed85ee073395a1ab Mon Sep 17 00:00:00 2001
From: Emil <emilwilliams@tuta.io>
Date: Thu, 28 Sep 2023 05:02:34 +0000
Subject: [PATCH] Memory optimization

---
 baked.c | 107 +++++++++++++++++++++++++++++++++++++++++++++-------------------
 1 file changed, 75 insertions(+), 32 deletions(-)

diff --git a/baked.c b/baked.c
index f945d91..e3c78de 100644
--- a/baked.c
+++ b/baked.c
@@ -28,6 +28,10 @@
   "\t$*  returns target-file without suffix\n"                        \
   "\t$+  returns arguments\n"
 
+#define local_assert(expr, ret) do { assert(expr); if (!expr) { return ret; }} while (0)
+
+static char * g_filename, * g_short, * g_all;
+
 static char *
 find(char * buf, const char * token)
 {
@@ -47,13 +51,15 @@ load(const char * fn)
   if (fp)
   {
     struct stat s;
-    size_t len;
+    off_t len;
     if (! stat(fn,&s)
-    &&    s.st_mode & S_IFREG
-    &&   (len = s.st_size)
-    &&   (buf = malloc(len + 1))
-    &&    fread(buf, 1, len, fp) > strlen(STOP) + strlen(START))
-    { buf[len] = '\0'; }
+    &&    s.st_mode & S_IFREG)
+    {
+      len = s.st_size;
+      buf = malloc(len + 1);
+      fread(buf, 1, len, fp);
+      buf[len] = '\0';
+    }
     fclose(fp);
   }
   return buf;
@@ -66,16 +72,16 @@ find_region(const char * fn)
   char * buf, * start, * stop;
 
   if (!(buf = load(fn)))
-  { fprintf(stderr, errno ? "cannot access '%s':" : "'%s': file too short", fn); return NULL; }
+  { if (errno) { fprintf(stderr, "cannot access '%s': ", fn); } return NULL; }
 
   if (!(start = find(buf,   START))
   ||  !(stop  = find(start, STOP)))
-  { fprintf(stderr, "No usable format located in '%s'", fn); return NULL; }
+  { fprintf(stderr, "No usable format located in '%s'", fn); free(buf); return NULL; }
 
   len = stop - start - strlen(STOP);
   memmove(buf, start, len);
-  buf = realloc(buf, len + 1);
   buf[len] = '\0';
+
   return buf;
 }
 
@@ -90,8 +96,8 @@ swap(char * a, char * b)
 static int
 root(char * root)
 {
-  char x[1] = "\0";
   int ret;
+  char x[1] = "\0";
   size_t len = strlen(root);
   while (len && root[len] != '/')
   { --len; }
@@ -106,9 +112,11 @@ root(char * root)
 static char *
 insert(const char * new, char * str, size_t offset, size_t shift)
 {
-  size_t len = strlen(new);
-  size_t max = strlen(str) + 1;
-  /* str = realloc(str, max + len); */
+  size_t len, max;
+  local_assert(new, str);
+  local_assert(str, NULL);
+  len = strlen(new);
+  max = strlen(str) + 1;
   memmove(str + offset + len, str + offset + shift, max - offset - shift);
   memcpy(str + offset, new, len);
   return str;
@@ -117,8 +125,12 @@ insert(const char * new, char * str, size_t offset, size_t shift)
 static char *
 shorten(char * fn)
 {
-  size_t i, last = 0, len = strlen(fn);
-  char * sh = malloc(len + 1);
+  size_t i, last = 0, len;
+  char * sh;
+  local_assert(fn, NULL);
+  len = strlen(fn);
+  sh = malloc(len + 1);
+  local_assert(sh, NULL);
   for (i = 0; i < len; ++i)
   {
     if (fn[i] == '.')
@@ -151,12 +163,11 @@ all_args(size_t argc, char ** argv)
   return all;
 }
 
-static char *
-expand(char * buf, int argc, char ** argv)
+static size_t
+expand_size(char * buf, size_t len, int argc, char ** argv)
 {
-  size_t i, len = strlen(buf);
-  char * str[3] = {0}, * ptr;
-  buf = realloc(buf, 500);
+  size_t i, max = len;
+  g_filename = argv[1];
   for (i = 0; i < len; ++i)
   {
     if (buf[i] == '\\')
@@ -166,19 +177,47 @@ expand(char * buf, int argc, char ** argv)
       switch (buf[++i])
       {
       case '@':
-        if (!str[0])
-        { str[0] = argv[1]; }
-        ptr = str[0];
+        max += strlen(g_filename);
         break;
       case '*':
-        if (!str[1])
-        { str[1] = shorten(argv[1]); }
-        ptr = str[1];
+        if (!g_short)
+        { g_short = shorten(argv[1]); }
+        max += strlen(g_short);
         break;
       case '+':
-        if (!str[2])
-        { str[2] = all_args((size_t) argc, argv); }
-        ptr = str[2] ? str[2] : "";
+        if (!g_all)
+        { g_all = all_args((size_t) argc, argv); }
+        max += g_all ? strlen(g_all) : 0;
+        break;
+      }
+    }
+  }
+  return max;
+}
+
+static char *
+expand(char * buf, size_t len)
+{
+  size_t i;
+  char * ptr = NULL;
+  buf = realloc(buf, len);
+  local_assert(buf, NULL);
+  for (i = 0; i < len; ++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;
       }
@@ -186,7 +225,7 @@ expand(char * buf, int argc, char ** argv)
       len = strlen(buf);
     }
   }
-  free(str[1]); free(str[2]);
+  free(g_short); free(g_all);
   return buf;
 }
 
@@ -194,12 +233,16 @@ int
 main(int argc, char ** argv)
 {
   int ret;
+  size_t len;
   char * buf;
   if (argc < 2)
   { fprintf(stderr, "%s: %s", argv[0], HELP DESC); return 1; }
   buf = find_region(argv[1]);
-  root(argv[1]);
-  buf = expand(buf, argc, argv);
+  if (!buf
+  ||   root(argv[1]))
+  { if (errno) { perror(NULL); } return 1; }
+  len = expand_size(buf, strlen(buf), argc, argv) + 1;
+  buf = expand(buf, len);
   fprintf(stderr, "Exec: %s\nOutput:\n", buf);
   fprintf(stderr, "Result: %d\n", (ret = system(buf)));
   free(buf);