10 unsigned int callsites:1;
14 static int cmp_sym(const void *p1_, const void *p2_)
16 const struct sl_item *p1 = p1_, *p2 = p2_;
17 const char *s1 = p1->name, *s2 = p2->name;
21 if ((s1[i] | s2[i]) == 0)
27 if (s1[i] == 0 && s2[i] == '@')
29 if (s1[i] == '@' && s2[i] == 0)
38 static int cmp_sym_sort(const void *p1_, const void *p2_)
40 const struct sl_item *p1 = p1_, *p2 = p2_;
41 const char *s1 = p1->name, *s2 = p2->name;
44 ret = cmp_sym(p1_, p2_);
46 printf("%s: dupe sym: '%s' '%s'\n", __func__, s1, s2);
52 void read_list(struct sl_item **sl_in, int *cnt, int *alloc,
53 FILE *f, int callsites)
55 struct sl_item *sl = *sl_in;
60 while (fgets(line, sizeof(line), f) != NULL) {
61 next_word(word, sizeof(word), line);
62 if (word[0] == 0 || word[0] == ';' || word[0] == '#')
65 sl[c].name = strdup(word);
66 sl[c].callsites = callsites;
72 sl = realloc(sl, *alloc * sizeof(sl[0]));
73 my_assert_not(sl, NULL);
74 memset(sl + c, 0, (*alloc - c) * sizeof(sl[0]));
82 const char *sym_use(const struct sl_item *sym)
84 static char buf[256+3];
87 ret = snprintf(buf, sizeof(buf), "rm_%s", sym->name);
88 if (ret >= sizeof(buf)) {
89 printf("truncation detected: '%s'\n", buf);
96 #define IS(w, y) !strcasecmp(w, y)
97 #define IS_OR2(w, x, y) (IS(w, x) || IS(w, y))
98 #define IS_OR3(w, x, y, z) (IS(w, x) || IS(w, y) || IS(w, z))
100 int main(int argc, char *argv[])
102 struct sl_item *symlist, *sym, ssym = { NULL, };
103 int patch_callsites = 0;
104 FILE *fout, *fin, *f;
118 // -c - patch callsites
119 printf("usage:\n%s <asmf_out> <asmf_in> [[-c] <listf>]*>\n",
126 symlist = calloc(symlist_alloc, sizeof(symlist[0]));
127 my_assert_not(symlist, NULL);
129 for (i = 3; i < argc; i++) {
130 if (strcmp(argv[i], "-c") == 0) {
135 f = fopen(argv[i], "r");
136 my_assert_not(f, NULL);
137 read_list(&symlist, &symlist_cnt, &symlist_alloc,
144 qsort(symlist, symlist_cnt, sizeof(symlist[0]), cmp_sym_sort);
147 printf("symlist:\n");
148 for (i = 0; i < symlist_cnt; i++)
149 printf("%d '%s'\n", symlist[i].callsites, symlist[i].name);
152 fin = fopen(argv[2], "r");
153 my_assert_not(fin, NULL);
155 fout = fopen(argv[1], "w");
156 my_assert_not(fout, NULL);
158 while (fgets(line, sizeof(line), fin))
161 if (*p == 0 || *p == ';')
164 p = sskip(next_word(word, sizeof(word), p));
165 if (*p == 0 || *p == ';')
166 goto pass; // need at least 2 words
168 p = next_word(word2, sizeof(word2), p);
170 if (IS_OR2(word2, "proc", "endp")) {
172 sym = bsearch(&ssym, symlist, symlist_cnt,
173 sizeof(symlist[0]), cmp_sym);
176 fprintf(fout, "rm_%s\t%s%s", word, word2, p);
181 if (IS_OR3(word, "call", "jmp", "public")) {
183 sym = bsearch(&ssym, symlist, symlist_cnt,
184 sizeof(symlist[0]), cmp_sym);
185 if (sym != NULL && sym->callsites) {
186 fprintf(fout, "\t\t%s\t%s%s", word,
193 if (*p == 0 || *p == ';')
194 goto pass; // need at least 3 words
196 p = next_word(word3, sizeof(word3), p);
201 if ( (IS_OR2(word, "dd", "push") && IS(word2, "offset"))
202 || (word[0] == 'j' && IS(word2, "short")) ) {
204 sym = bsearch(&ssym, symlist, symlist_cnt,
205 sizeof(symlist[0]), cmp_sym);
206 if (sym != NULL && sym->callsites) {
207 fprintf(fout, "\t\t%s %s %s%s",
208 word, word2, sym_use(sym), p);
214 if (*p == 0 || *p == ';')
215 goto pass; // need at least 4 words
217 p = next_word(word4, sizeof(word4), p);
219 // <name> dd offset <sym>
220 if (IS(word2, "dd") && IS(word3, "offset")) {
222 sym = bsearch(&ssym, symlist, symlist_cnt,
223 sizeof(symlist[0]), cmp_sym);
224 if (sym != NULL && sym->callsites) {
225 fprintf(fout, "%s\tdd offset %s%s", word,
231 // mov <something>, offset <sym>
232 // jcc <some> ptr <sym>
233 if ( (IS(word, "mov") && IS(word3, "offset"))
234 || (word[0] == 'j' && IS(word3, "ptr")) ) {
236 sym = bsearch(&ssym, symlist, symlist_cnt,
237 sizeof(symlist[0]), cmp_sym);
238 if (sym != NULL && sym->callsites) {
239 fprintf(fout, "\t\t%s\t%s %s %s%s",
247 if (*p == 0 || *p == ';')
248 goto pass; // need at least 5 words
250 p = next_word(word5, sizeof(word5), p);
253 if (*p == 0 || *p == ';')
254 goto pass; // need at least 6 words
256 p = next_word(word6, sizeof(word6), p);
258 // <op> dword ptr <something>, offset <sym>
259 if ( IS(word2, "dword") && IS(word3, "ptr")
260 && IS(word5, "offset") ) {
262 sym = bsearch(&ssym, symlist, symlist_cnt,
263 sizeof(symlist[0]), cmp_sym);
264 if (sym != NULL && sym->callsites) {
265 fprintf(fout, "\t\t%s\tdword ptr %s offset %s%s",
266 word, word4, sym_use(sym), p);
272 fwrite(line, 1, strlen(line), fout);
275 for (i = 0; i < symlist_cnt; i++) {
276 if (!symlist[i].found)
277 printf("warning: sym '%s' not found\n", symlist[i].name);