X-Git-Url: https://notaz.gp2x.de/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=tools%2Fasmproc.c;h=956ffa249245e7c8148a82a3e3d6add8cb037360;hb=HEAD;hp=3ff983b24b65608a451cab6b6c5d27c2849fb8b9;hpb=368deb2385c2ee4971964c3c08688c4ad36cb56e;p=ia32rtools.git diff --git a/tools/asmproc.c b/tools/asmproc.c index 3ff983b..956ffa2 100644 --- a/tools/asmproc.c +++ b/tools/asmproc.c @@ -1,45 +1,24 @@ +/* + * ia32rtools + * (C) notaz, 2013,2014 + * + * This work is licensed under the terms of 3-clause BSD license. + * See COPYING file in the top-level directory. + */ + #include #include #include #include "my_assert.h" - -static int my_isblank(char c) -{ - return c == '\t' || c == ' ' || c == '\r' || c == '\n'; -} - -static char *sskip(char *s) -{ - while (my_isblank(*s)) - s++; - - return s; -} - -static char *next_word(char *w, size_t wsize, char *s) -{ - size_t i; - - s = sskip(s); - - for (i = 0; i < wsize - 1; i++) { - if (*s == 0 || my_isblank(s[i])) - break; - w[i] = s[i]; - } - w[i] = 0; - - if (*s != 0 && !my_isblank(s[i])) - printf("warning: '%s' truncated\n", w); - - return s + i; -} +#include "my_str.h" +#include "common.h" struct sl_item { char *name; unsigned int callsites:1; unsigned int found:1; + unsigned int ignore_missing:1; }; static int cmp_sym(const void *p1_, const void *p2_) @@ -48,16 +27,21 @@ static int cmp_sym(const void *p1_, const void *p2_) const char *s1 = p1->name, *s2 = p2->name; int i; + if (*s1 == '_') + s1++; + if (*s2 == '_') + s2++; + 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]; @@ -81,21 +65,21 @@ static int cmp_sym_sort(const void *p1_, const void *p2_) } void read_list(struct sl_item **sl_in, int *cnt, int *alloc, - FILE *f, int callsites) + FILE *f, int callsites, int ignore_missing) { 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].callsites = callsites; + sl[c].ignore_missing = ignore_missing; sl[c].found = 0; c++; @@ -111,13 +95,13 @@ void read_list(struct sl_item **sl_in, int *cnt, int *alloc, *cnt = c; } -const char *sym_use(const struct sl_item *sym) +const char *sym_use(const struct sl_item *sym, int is_rm) { static char buf[256+3]; int ret; ret = snprintf(buf, sizeof(buf), "%s%s", - sym->callsites ? "" : "rm_", sym->name); + is_rm ? "rm_" : "", sym->name); if (ret >= sizeof(buf)) { printf("truncation detected: '%s'\n", buf); exit(1); @@ -126,10 +110,14 @@ const char *sym_use(const struct sl_item *sym) return buf; } +#define IS_OR2(w, x, y) (IS(w, x) || IS(w, y)) +#define IS_OR3(w, x, y, z) (IS(w, x) || IS(w, y) || IS(w, z)) + int main(int argc, char *argv[]) { struct sl_item *symlist, *sym, ssym = { NULL, }; int patch_callsites = 0; + int ignore_missing = 0; FILE *fout, *fin, *f; int symlist_alloc; int symlist_cnt; @@ -138,13 +126,17 @@ int main(int argc, char *argv[]) char word2[256]; char word3[256]; char word4[256]; - char *p; + char word5[256]; + char word6[256]; + char func[256]; + char *p, *p2; int i; if (argc < 4) { - // -c - patch callsites - printf("usage:\n%s [[-c] ]*>\n", + printf("usage:\n%s [[-c][-i] ]*>\n", argv[0]); + printf(" -c - patch callsites\n" + " -i - ignore missing syms\n"); return 1; } @@ -154,18 +146,23 @@ int main(int argc, char *argv[]) my_assert_not(symlist, NULL); for (i = 3; i < argc; i++) { - if (!strcmp(argv[i], "-c")) { + if (strcmp(argv[i], "-c") == 0) { patch_callsites = 1; continue; } + if (strcmp(argv[i], "-i") == 0) { + ignore_missing = 1; + continue; + } f = fopen(argv[i], "r"); my_assert_not(f, NULL); read_list(&symlist, &symlist_cnt, &symlist_alloc, - f, patch_callsites); + f, patch_callsites, ignore_missing); fclose(f); patch_callsites = 0; + ignore_missing = 0; } qsort(symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym_sort); @@ -194,24 +191,39 @@ int main(int argc, char *argv[]) p = next_word(word2, sizeof(word2), p); - if (!strcasecmp(word2, "proc") || !strcasecmp(word2, "endp")) { + if (IS_OR2(word2, "proc", "endp")) { + if (IS(word2, "proc")) + strcpy(func, word); + else + func[0] = 0; + ssym.name = word; 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); + fprintf(fout, "%s\t%s%s", sym_use(sym, 1), word2, p); continue; } } - if (!strcasecmp(word, "call") || !strcasecmp(word, "jmp")) { + if (IS_OR2(word, "call", "jmp")) { ssym.name = word2; sym = bsearch(&ssym, symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym); - if (sym != NULL && sym->callsites) { + if (sym != NULL) { fprintf(fout, "\t\t%s\t%s%s", word, - sym_use(sym), p); + sym_use(sym, sym->callsites || IS(word2, func)), p); + continue; + } + } + + if (IS(word, "public")) { + ssym.name = word2; + sym = bsearch(&ssym, symlist, symlist_cnt, + sizeof(symlist[0]), cmp_sym); + if (sym != NULL) { + fprintf(fout, "\t\tpublic %s%s", sym_use(sym, 1), p); continue; } } @@ -222,40 +234,135 @@ int main(int argc, char *argv[]) p = next_word(word3, sizeof(word3), p); - if (!strcasecmp(word, "dd") && !strcasecmp(word2, "offset")) { + // push offset + if (IS(word, "push") && IS(word2, "offset")) { + ssym.name = word3; + sym = bsearch(&ssym, symlist, symlist_cnt, + sizeof(symlist[0]), cmp_sym); + if (sym != NULL) { + fprintf(fout, "\t\t%s %s %s%s", word, word2, + sym_use(sym, sym->callsites || IS(word3, func)), p); + continue; + } + } + + // jcc short + if (word[0] == 'j' && IS(word2, "short") && !IS(word3, "exit")) { ssym.name = word3; sym = bsearch(&ssym, symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym); - if (sym != NULL && sym->callsites) { - fprintf(fout, "\t\tdd offset %s%s", - sym_use(sym), p); + if (sym != NULL) { + fprintf(fout, "\t\t%s ", word); + // for conditional "call", don't print 'short' + if (IS(word3, func)) + fprintf(fout, "short "); + fprintf(fout, "%s%s", + sym_use(sym, sym->callsites || IS(word3, func)), p); continue; } } + // dd offset + if (IS(word, "dd") + && (IS(word2, "offset") || strstr(p, "offset"))) + { + fprintf(fout, "\t\tdd"); + p = next_word(word, sizeof(word), line); + goto offset_loop; + } + p = sskip(p); if (*p == 0 || *p == ';') goto pass; // need at least 4 words p = next_word(word4, sizeof(word4), p); - if (!strcasecmp(word2, "dd") && !strcasecmp(word3, "offset")) { + // dd offset + if (IS(word2, "dd") + && (IS(word3, "offset") || strstr(p, "offset"))) + { + fprintf(fout, "%s\tdd", word); + p = next_word(word, sizeof(word), line); + p = next_word(word, sizeof(word), p); + goto offset_loop; + } + + // mov , offset + // jcc ptr + if ( (IS(word, "mov") && IS(word3, "offset")) + || (word[0] == 'j' && IS(word3, "ptr")) ) { ssym.name = word4; sym = bsearch(&ssym, symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym); - if (sym != NULL && sym->callsites) { - fprintf(fout, "%s\tdd offset %s%s", word, - sym_use(sym), p); + if (sym != NULL) { + fprintf(fout, "\t\t%s\t%s %s %s%s", + word, word2, word3, + sym_use(sym, sym->callsites), p); + continue; + } + } + + p = sskip(p); + if (*p == 0 || *p == ';') + goto pass; // need at least 5 words + + p = next_word(word5, sizeof(word5), p); + + p = sskip(p); + if (*p == 0 || *p == ';') + goto pass; // need at least 6 words + + p = next_word(word6, sizeof(word6), p); + + // dword ptr , offset + if ( IS(word2, "dword") && IS(word3, "ptr") + && IS(word5, "offset") ) { + ssym.name = word6; + sym = bsearch(&ssym, symlist, symlist_cnt, + sizeof(symlist[0]), cmp_sym); + if (sym != NULL) { + fprintf(fout, "\t\t%s\tdword ptr %s offset %s%s", + word, word4, sym_use(sym, sym->callsites), p); continue; } } pass: fwrite(line, 1, strlen(line), fout); + continue; + +offset_loop: + while (1) { + p2 = next_word(word, sizeof(word), p); + if (word[0] == 0 || word[0] == ';') { + break; + } + if (!IS(word, "offset")) { + // pass through + p2 = strstr(p, "offset"); + if (p2 == NULL) + break; + fwrite(p, 1, p2 - p, fout); + p2 = next_word(word, sizeof(word), p2); + } + p = next_word(word, sizeof(word), p2); + p2 = strchr(word, ','); + if (p2) + *p2 = 0; + + ssym.name = word; + sym = bsearch(&ssym, symlist, symlist_cnt, + sizeof(symlist[0]), cmp_sym); + fprintf(fout, " offset %s%s", + (sym != NULL) ? sym_use(sym, sym->callsites) : word, + p2 ? "," : ""); + } + fprintf(fout, "%s", p); + continue; } for (i = 0; i < symlist_cnt; i++) { - if (!symlist[i].found) + if (!symlist[i].found && !symlist[i].ignore_missing) printf("warning: sym '%s' not found\n", symlist[i].name); } @@ -264,3 +371,5 @@ pass: return 0; } + +// vim:ts=2:shiftwidth=2