X-Git-Url: https://notaz.gp2x.de/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=tools%2Fasmproc.c;h=956ffa249245e7c8148a82a3e3d6add8cb037360;hb=HEAD;hp=911fe5827a06b671875da6c76fa766ba8c4cd9a3;hpb=ede51255344cb1cd5859fb24cae2e87692ca0887;p=ia32rtools.git diff --git a/tools/asmproc.c b/tools/asmproc.c index 911fe58..956ffa2 100644 --- a/tools/asmproc.c +++ b/tools/asmproc.c @@ -1,44 +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; - int is_replace; + unsigned int callsites:1; + unsigned int found:1; + unsigned int ignore_missing:1; }; static int cmp_sym(const void *p1_, const void *p2_) @@ -47,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]; @@ -79,27 +64,30 @@ 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, 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].is_replace = is_repl; + sl[c].callsites = callsites; + sl[c].ignore_missing = ignore_missing; + 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 +95,29 @@ 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, int is_rm) +{ + static char buf[256+3]; + int ret; + + ret = snprintf(buf, sizeof(buf), "%s%s", + is_rm ? "rm_" : "", sym->name); + if (ret >= sizeof(buf)) { + printf("truncation detected: '%s'\n", buf); + exit(1); + } + + 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; + 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; @@ -118,37 +126,52 @@ 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 != 5) { - // rmlist - prefix func with 'rm_', callsites with '_' - // renlist - prefix func and callsites with 'rm_' - printf("usage:\n%s \n", + if (argc < 4) { + printf("usage:\n%s [[-c][-i] ]*>\n", argv[0]); + printf(" -c - patch callsites\n" + " -i - ignore missing syms\n"); 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; + } + if (strcmp(argv[i], "-i") == 0) { + ignore_missing = 1; + continue; + } - f = fopen(argv[4], "r"); - my_assert_not(f, NULL); - read_list(&symlist, &symlist_cnt, &symlist_alloc, f, 0); - fclose(f); + f = fopen(argv[i], "r"); + my_assert_not(f, NULL); + read_list(&symlist, &symlist_cnt, &symlist_alloc, + f, patch_callsites, ignore_missing); + fclose(f); + + patch_callsites = 0; + ignore_missing = 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); @@ -168,23 +191,39 @@ for (i = 0; i < symlist_cnt; i++) 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) { - fprintf(fout, "rm_%s\t%s%s", word, word2, p); + sym->found = 1; + fprintf(fout, "%s\t%s%s", sym_use(sym, 1), word2, p); + continue; + } + } + + if (IS_OR2(word, "call", "jmp")) { + 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", word, + sym_use(sym, sym->callsites || IS(word2, func)), p); continue; } } - if (!strcasecmp(word, "call") || !strcasecmp(word, "jmp")) { + if (IS(word, "public")) { 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); + fprintf(fout, "\t\tpublic %s%s", sym_use(sym, 1), p); continue; } } @@ -195,35 +234,136 @@ for (i = 0; i < symlist_cnt; i++) 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) { - fprintf(fout, "\t\tdd offset %s%s", word3, p); + 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) { - fprintf(fout, "%s\tdd offset %s%s%s", word, - sym->is_replace ? "_" : "rm_", word4, p); + 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 && !symlist[i].ignore_missing) + printf("warning: sym '%s' not found\n", symlist[i].name); } fclose(fin); @@ -231,3 +371,5 @@ pass: return 0; } + +// vim:ts=2:shiftwidth=2