+/*
+ * 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 <stdio.h>
#include <stdlib.h>
#include <string.h>
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;
*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), "rm_%s", sym->name);
+ ret = snprintf(buf, sizeof(buf), "%s%s",
+ is_rm ? "rm_" : "", sym->name);
if (ret >= sizeof(buf)) {
printf("truncation detected: '%s'\n", buf);
exit(1);
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;
}
}
ssym.name = word2;
sym = bsearch(&ssym, symlist, symlist_cnt,
sizeof(symlist[0]), cmp_sym);
- if (sym != NULL
- && (sym->callsites || IS(word2, func)))
- {
+ 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;
}
}
sym = bsearch(&ssym, symlist, symlist_cnt,
sizeof(symlist[0]), cmp_sym);
if (sym != NULL) {
- fprintf(fout, "\t\tpublic %s%s", sym_use(sym), p);
+ fprintf(fout, "\t\tpublic %s%s", sym_use(sym, 1), p);
continue;
}
}
p = next_word(word3, sizeof(word3), p);
// push offset <sym>
+ 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 <sym>
- if ( (IS(word, "push") && IS(word2, "offset"))
- || (word[0] == 'j' && IS(word2, "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 || IS(word3, func)))
- {
- fprintf(fout, "\t\t%s %s %s%s",
- word, word2, 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 <sym>
- if (IS(word, "dd") && IS(word2, "offset")) {
+ if (IS(word, "dd")
+ && (IS(word2, "offset") || strstr(p, "offset")))
+ {
fprintf(fout, "\t\tdd");
- strcpy(word, word3);
+ p = next_word(word, sizeof(word), line);
goto offset_loop;
}
p = next_word(word4, sizeof(word4), p);
// <name> dd offset <sym>
- if (IS(word2, "dd") && IS(word3, "offset")) {
+ if (IS(word2, "dd")
+ && (IS(word3, "offset") || strstr(p, "offset")))
+ {
fprintf(fout, "%s\tdd", word);
- strcpy(word, word4);
+ p = next_word(word, sizeof(word), line);
+ p = next_word(word, sizeof(word), p);
goto offset_loop;
}
ssym.name = word4;
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 %s%s",
word, word2, word3,
- sym_use(sym), p);
+ sym_use(sym, sym->callsites), p);
continue;
}
}
ssym.name = word6;
sym = bsearch(&ssym, symlist, symlist_cnt,
sizeof(symlist[0]), cmp_sym);
- if (sym != NULL && sym->callsites) {
+ if (sym != NULL) {
fprintf(fout, "\t\t%s\tdword ptr %s offset %s%s",
- word, word4, sym_use(sym), p);
+ word, word4, sym_use(sym, sym->callsites), p);
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;
sym = bsearch(&ssym, symlist, symlist_cnt,
sizeof(symlist[0]), cmp_sym);
fprintf(fout, " offset %s%s",
- (sym != NULL && sym->callsites) ? sym_use(sym) : word,
+ (sym != NULL) ? sym_use(sym, sym->callsites) : word,
p2 ? "," : "");
-
- p2 = next_word(word, sizeof(word), p);
- if (word[0] == 0 || word[0] == ';') {
- break;
- }
- if (!IS(word, "offset")) {
- printf("could not handle offset array\n");
- break;
- }
- p = next_word(word, sizeof(word), p2);
}
fprintf(fout, "%s", p);
continue;