fixes, start removing mscrt

[ia32rtools.git] / tools / asmproc.c

diff --git a/tools/asmproc.c b/tools/asmproc.c
index 911fe58..e8e9efb 100644 (file)
--- a/tools/asmproc.c
+++ b/tools/asmproc.c
@@ -38,7 +38,8 @@ static char *next_word(char *w, size_t wsize, char *s)
 
 struct sl_item {
        char *name;
-       int is_replace;
+       unsigned int callsites:1;
+       unsigned int found:1;
 };
 
 static int cmp_sym(const void *p1_, const void *p2_)
@@ -48,15 +49,15 @@ static int cmp_sym(const void *p1_, const void *p2_)
        int i;
 
        for (i = 0; ; i++) {
+               if ((s1[i] | s2[i]) == 0)
+                       break;
+
                if (s1[i] == s2[i])
                        continue;
 
-               if (s1[i] == 0) {
-                       if (s2[i] == 0 || s2[i] == '@')
-                               break;
-               }
-
-               if (s2[i] == 0 || s1[i] == '@')
+               if (s1[i] ==  0  && s2[i] == '@')
+                       break;
+               if (s1[i] == '@' && s2[i] ==  0)
                        break;
 
                return s1[i] - s2[i];
@@ -79,27 +80,29 @@ static int cmp_sym_sort(const void *p1_, const void *p2_)
        return ret;
 }
 
-void read_list(struct sl_item **sl_in, int *cnt, int *alloc, FILE *f, int is_repl)
+void read_list(struct sl_item **sl_in, int *cnt, int *alloc,
+       FILE *f, int callsites)
 {
        struct sl_item *sl = *sl_in;
        int c = *cnt;
        char line[256];
        char word[256];
-       char *r;
 
        while (fgets(line, sizeof(line), f) != NULL) {
-               r = next_word(word, sizeof(word), line);
-               if (r == word)
+               next_word(word, sizeof(word), line);
+               if (word[0] == 0 || word[0] == ';' || word[0] == '#')
                        continue;
 
                sl[c].name = strdup(word);
-               sl[c].is_replace = is_repl;
+               sl[c].callsites = callsites;
+               sl[c].found = 0;
                c++;
 
                if (c >= *alloc) {
                        *alloc *= 2;
                        sl = realloc(sl, *alloc * sizeof(sl[0]));
                        my_assert_not(sl, NULL);
+                       memset(sl + c, 0, (*alloc - c) * sizeof(sl[0]));
                }
        }
 
@@ -107,9 +110,24 @@ void read_list(struct sl_item **sl_in, int *cnt, int *alloc, FILE *f, int is_rep
        *cnt = c;
 }
 
+const char *sym_use(const struct sl_item *sym)
+{
+       static char buf[256+3];
+       int ret;
+
+       ret = snprintf(buf, sizeof(buf), "rm_%s", sym->name);
+       if (ret >= sizeof(buf)) {
+               printf("truncation detected: '%s'\n", buf);
+               exit(1);
+       }
+
+       return buf;
+}
+
 int main(int argc, char *argv[])
 {
-       struct sl_item *symlist, *sym, ssym;
+       struct sl_item *symlist, *sym, ssym = { NULL, };
+       int patch_callsites = 0;
        FILE *fout, *fin, *f;
        int symlist_alloc;
        int symlist_cnt;
@@ -119,36 +137,42 @@ int main(int argc, char *argv[])
        char word3[256];
        char word4[256];
        char *p;
+       int i;
 
-       if (argc != 5) {
-               // rmlist - prefix func with 'rm_', callsites with '_'
-               // renlist - prefix func and callsites with 'rm_'
-               printf("usage:\n%s <asmf_out> <asmf_in> <rmlist> <renlist>\n",
+       if (argc < 4) {
+               // -c - patch callsites
+               printf("usage:\n%s <asmf_out> <asmf_in> [[-c] <listf>]*>\n",
                        argv[0]);
                return 1;
        }
 
        symlist_alloc = 16;
        symlist_cnt = 0;
-       symlist = malloc(symlist_alloc * sizeof(symlist[0]));
+       symlist = calloc(symlist_alloc, sizeof(symlist[0]));
        my_assert_not(symlist, NULL);
 
-       f = fopen(argv[3], "r");
-       my_assert_not(f, NULL);
-       read_list(&symlist, &symlist_cnt, &symlist_alloc, f, 1);
-       fclose(f);
+       for (i = 3; i < argc; i++) {
+               if (strcmp(argv[i], "-c") == 0) {
+                       patch_callsites = 1;
+                       continue;
+               }
+
+               f = fopen(argv[i], "r");
+               my_assert_not(f, NULL);
+               read_list(&symlist, &symlist_cnt, &symlist_alloc,
+                       f, patch_callsites);
+               fclose(f);
 
-       f = fopen(argv[4], "r");
-       my_assert_not(f, NULL);
-       read_list(&symlist, &symlist_cnt, &symlist_alloc, f, 0);
-       fclose(f);
+               patch_callsites = 0;
+       }
 
        qsort(symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym_sort);
 
-printf("symlist:\n");
-int i;
-for (i = 0; i < symlist_cnt; i++)
- printf("%d '%s'\n", symlist[i].is_replace, symlist[i].name);
+#if 0
+       printf("symlist:\n");
+       for (i = 0; i < symlist_cnt; i++)
+               printf("%d '%s'\n", symlist[i].callsites, symlist[i].name);
+#endif
 
        fin = fopen(argv[2], "r");
        my_assert_not(fin, NULL);
@@ -173,6 +197,7 @@ for (i = 0; i < symlist_cnt; i++)
                        sym = bsearch(&ssym, symlist, symlist_cnt,
                                sizeof(symlist[0]), cmp_sym);
                        if (sym != NULL) {
+                               sym->found = 1;
                                fprintf(fout, "rm_%s\t%s%s", word, word2, p);
                                continue;
                        }
@@ -182,9 +207,9 @@ for (i = 0; i < symlist_cnt; i++)
                        ssym.name = word2;
                        sym = bsearch(&ssym, symlist, symlist_cnt,
                                sizeof(symlist[0]), cmp_sym);
-                       if (sym != NULL) {
-                               fprintf(fout, "\t\t%s\t%s%s%s", word,
-                                       sym->is_replace ? "_" : "rm_", word2, p);
+                       if (sym != NULL && sym->callsites) {
+                               fprintf(fout, "\t\t%s\t%s%s", word,
+                                       sym_use(sym), p);
                                continue;
                        }
                }
@@ -199,8 +224,9 @@ for (i = 0; i < symlist_cnt; i++)
                        ssym.name = word3;
                        sym = bsearch(&ssym, symlist, symlist_cnt,
                                sizeof(symlist[0]), cmp_sym);
-                       if (sym != NULL) {
-                               fprintf(fout, "\t\tdd offset %s%s", word3, p);
+                       if (sym != NULL && sym->callsites) {
+                               fprintf(fout, "\t\tdd offset %s%s",
+                                       sym_use(sym), p);
                                continue;
                        }
                }
@@ -215,9 +241,9 @@ for (i = 0; i < symlist_cnt; i++)
                        ssym.name = word4;
                        sym = bsearch(&ssym, symlist, symlist_cnt,
                                sizeof(symlist[0]), cmp_sym);
-                       if (sym != NULL) {
-                               fprintf(fout, "%s\tdd offset %s%s%s", word,
-                                       sym->is_replace ? "_" : "rm_", word4, p);
+                       if (sym != NULL && sym->callsites) {
+                               fprintf(fout, "%s\tdd offset %s%s", word,
+                                       sym_use(sym), p);
                                continue;
                        }
                }
@@ -226,6 +252,11 @@ pass:
                fwrite(line, 1, strlen(line), fout);
        }
 
+       for (i = 0; i < symlist_cnt; i++) {
+               if (!symlist[i].found)
+                       printf("warning: sym '%s' not found\n", symlist[i].name);
+       }
+
        fclose(fin);
        fclose(fout);