🎨 Fix #258

Gwion · Feb 27, 2023 · 6ae331b · 6ae331b · elfring · Feb 27, 2023
1 parent 97920d7
commit 6ae331b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/src/lib/string.c b/src/lib/string.c
@@ -381,8 +381,13 @@ static SFUN(string_load) {
   fseek(f, 0, SEEK_END);
   const size_t sz = ftell(f);
   char c[sz + 1];
+  c[sz] = '\0';
   rewind(f);
-  (void)fread(c, 1, sz, f);
+  const size_t ret = fread(c, 1, sz, f);
+  if(ret != sz) {
+    xfun_handle(shred, "StringLoadException");
+    return;
+  }
   fclose(f);
   *(M_Object*)RETURN = new_string(shred->info->vm->gwion, c);
 }

diff --git a/tests/plug/compile_file.c b/tests/plug/compile_file.c
@@ -14,10 +14,12 @@
 #include <string.h>
 GWION_IMPORT(compile_file) {
   DECL_OB(FILE *, file, = fopen("rm_me.gw", "w+"));
-  fprintf(file, "1;");
-  rewind(file);
-  const m_bool ret =
+  if(fprintf(file, "1;") >= 0) {
+    rewind(file);
+    const m_bool ret =
       compile_file(gwi->gwion, __FILE__, file) ? GW_OK : GW_ERROR;
-  fclose(file);
-  return ret;
+    fclose(file);
+    return ret;
+  }
+  return GW_ERROR;
 }