From 66feea732ecb17230e298a7cc887a3f0fe4298cd Mon Sep 17 00:00:00 2001
From: Paul Gardner-Stephen <paul@servalproject.org>
Date: Thu, 18 Apr 2019 21:14:45 +0930
Subject: [PATCH] more work on preprocessor to get routine sizes for packing

---
 src/preprocess.c | 82 ++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 79 insertions(+), 3 deletions(-)

diff --git a/src/preprocess.c b/src/preprocess.c
index 81bc0e8f..d44295fc 100644
--- a/src/preprocess.c
+++ b/src/preprocess.c
@@ -17,6 +17,11 @@
 typedef struct source_file {
   char *file;
   int address;
+
+  // For getting routine sizes from first run of ophis
+  int low_addr;
+  int high_addr;
+  int size;
 } source_file;
 
 source_file source_files[MAX_FILES];
@@ -32,6 +37,25 @@ static int compare_source(const void* a, const void* b)
   return strcmp(aa->file,bb->file);
 }
 
+int dump_file(FILE *out,const char *name)
+{
+  int retVal=0;
+  FILE *f=NULL;
+  
+  do {
+    f=fopen(name,"r");
+    if (!f) DO_ERROR("Failed to read file for input");
+    char line[8192];
+    while(!feof(f)) {
+      line[0]=0; fgets(line,8192,f);
+      fprintf(out,"%s",line);
+    }
+  } while(0);
+
+  if (f) fclose(f);
+  return retVal;
+}
+
 void usage(void)
 {
   fprintf(stderr,"usage: preprocess [-d <directory>]\n");
@@ -98,10 +122,62 @@ int main(int argc,char **argv)
     fprintf(stderr,"Sorted.\n");
     
     for(int i=0;i<source_count;i++) {
-      printf("$%04X : %s\n",
-	     source_files[i].address!=0x7ffffff?source_files[i].address:0,
-	     source_files[i].file);
+      fprintf(stderr,"$%04X : %s\n",
+	      source_files[i].address!=0x7ffffff?source_files[i].address:0,
+	      source_files[i].file);
     }
+
+    // First pass is to work out the size of each routine, so that we can pack them optimally
+    // around the routines that are located at fixed addresses.
+    unlink("temp.map");
+    unlink("temp.s");
+    FILE *out=fopen("temp.s","w");
+    if (!out) DO_ERROR("Could not write to temp.s");
+    fprintf(out,"\t.org 0\n");
+    for(int i=0;i<source_count;i++) {
+      fprintf(out,"; Source file '%s'\n",source_files[i].file);
+      fprintf(out,"__routine_start_%s:\n",source_files[i].file);
+      char path[8192];
+      snprintf(path,8192,"%s/%s",directory,source_files[i].file);
+      if (dump_file(out,path)) DO_ERROR("Could not read source file");
+      fprintf(out,"__routine_end_%s:\n",source_files[i].file);
+    }
+    fclose(out);
+    char cmd[8192];
+    snprintf(cmd,8192,"Ophis/bin/ophis temp.s -m temp.map");
+    fprintf(stderr,"Running first pass to get size of routines...\n");
+    system(cmd);
+
+    // Parse temp.map contents to get size of every routine
+    fprintf(stderr,"Reading sizes of routines...\n");
+    out=fopen("temp.map","r");
+    if (!out) DO_ERROR("Could not open temp.map for reading. Did Ophis fail due to an error in one of the assembly files?");
+    while(!feof(out)) {
+      int addr;
+      char routine[8192];
+      char line[8192];
+      line[0]=0; fgets(line,8192,out);
+      if (sscanf(line,"$%x __routine_start_%[^ \n]",&addr,routine)==2) {
+	for(int i=0;i<source_count;i++)
+	  if (!strcmp(routine,source_files[i].file))
+	    { source_files[i].low_addr=addr;
+	      source_files[i].size=source_files[i].high_addr-source_files[i].low_addr;
+	      break; }
+      }
+      if (sscanf(line,"$%x __routine_end_%[^ \n]",&addr,routine)==2) {
+	for(int i=0;i<source_count;i++)
+	  if (!strcmp(routine,source_files[i].file))
+	    { source_files[i].high_addr=addr;
+	      source_files[i].size=source_files[i].high_addr-source_files[i].low_addr;
+	      break; }
+      }
+    }
+    fclose(out);
+
+    for(int i=0;i<source_count;i++)
+      fprintf(stderr,"%s : %d bytes\n",source_files[i].file,source_files[i].size);
+    
+    if (retVal) break;
     
   } while(0);