5 * This work is licensed under the terms of 3-clause BSD license.
6 * See COPYING file in the top-level directory.
14 #include "my_assert.h"
17 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
18 #define IS(w, y) !strcmp(w, y)
19 #define IS_START(w, y) !strncmp(w, y, strlen(y))
21 #include "protoparse.h"
23 static const char *asmfn;
26 static const struct parsed_proto *g_func_sym_pp;
27 static char g_comment[256];
28 static int g_warn_cnt;
29 static int g_cconv_novalidate;
30 static int g_arm_mode;
32 // note: must be in ascending order
42 #define anote(fmt, ...) \
43 printf("%s:%d: note: " fmt, asmfn, asmln, ##__VA_ARGS__)
44 #define awarn(fmt, ...) do { \
45 printf("%s:%d: warning: " fmt, asmfn, asmln, ##__VA_ARGS__); \
46 if (++g_warn_cnt == 10) { \
51 #define aerr(fmt, ...) do { \
52 printf("%s:%d: error: " fmt, asmfn, asmln, ##__VA_ARGS__); \
57 #include "masm_tools.h"
59 static char *next_word_s(char *w, size_t wsize, char *s)
66 for (i = 0; i < wsize - 1; i++) {
69 if (s[i] == 0 || (!quote && (my_isblank(s[i]) || s[i] == ',')))
75 if (s[i] != 0 && !my_isblank(s[i]) && s[i] != ',')
76 printf("warning: '%s' truncated\n", w);
81 static void next_section(FILE *fasm, char *name)
90 while (fgets(line, sizeof(line), fasm))
100 while (strlen(line) == sizeof(line) - 1) {
101 // one of those long comment lines..
102 if (!fgets(line, sizeof(line), fasm))
108 for (wordc = 0; wordc < ARRAY_SIZE(words); wordc++) {
109 p = sskip(next_word(words[wordc], sizeof(words[0]), p));
110 if (*p == 0 || *p == ';') {
119 if (!IS(words[1], "segment"))
122 strcpy(name, words[0]);
127 static enum dx_type parse_dx_directive(const char *name)
143 static const char *type_name(enum dx_type type)
162 static const char *type_name_float(enum dx_type type)
174 return "<bad_float>";
177 static int type_size(enum dx_type type)
196 static char *escape_string(char *s)
201 for (; *s != 0; s++) {
215 return strcpy(s, buf);
218 static void sprint_pp_short(const struct parsed_proto *pp, char *buf,
225 if (pp->ret_type.is_ptr)
227 else if (IS(pp->ret_type.name, "void"))
234 for (i = 0; i < pp->argc; i++) {
235 if (pp->arg[i].reg != NULL)
236 snprintf(buf + l, buf_size - l, "%s%s",
237 i == 0 ? "" : ",", pp->arg[i].reg);
239 snprintf(buf + l, buf_size - l, "%sa%d",
240 i == 0 ? "" : ",", i + 1);
243 snprintf(buf + l, buf_size - l, ")");
246 static const struct parsed_proto *check_var(FILE *fhdr,
247 const char *sym, const char *varname)
249 const struct parsed_proto *pp, *pp_sym;
250 char fp_sym[256], fp_var[256];
252 pp = proto_parse(fhdr, varname, 1);
254 if (IS_START(varname, "sub_"))
255 awarn("sub_ sym missing proto: '%s'\n", varname);
259 if (!pp->is_func && !pp->is_fptr)
262 pp_print(fp_var, sizeof(fp_var), pp);
264 if (pp->argc_reg == 0)
266 if (pp->argc_reg == 1 && pp->argc_stack == 0
267 && IS(pp->arg[0].reg, "ecx"))
271 if (!g_cconv_novalidate
272 && (pp->argc_reg != 2
273 || !IS(pp->arg[0].reg, "ecx")
274 || !IS(pp->arg[1].reg, "edx")))
276 awarn("unhandled reg call: %s\n", fp_var);
280 sprint_pp_short(pp, g_comment, sizeof(g_comment));
283 g_func_sym_pp = NULL;
284 pp_sym = proto_parse(fhdr, sym, 1);
287 if (!pp_sym->is_fptr)
288 aerr("func ptr data, but label '%s' !is_fptr\n", pp_sym->name);
289 g_func_sym_pp = pp_sym;
292 pp_sym = g_func_sym_pp;
297 if (pp_cmp_func(pp, pp_sym)) {
298 pp_print(fp_sym, sizeof(fp_sym), pp_sym);
299 anote("var: %s\n", fp_var);
300 anote("sym: %s\n", fp_sym);
301 awarn("^ mismatch\n");
307 static void output_decorated_pp(FILE *fout,
308 const struct parsed_proto *pp)
310 if (pp->name[0] != '_')
311 fprintf(fout, pp->is_fastcall ? "@" : "_");
312 fprintf(fout, "%s", pp->name);
313 if (pp->is_stdcall && pp->argc > 0)
314 fprintf(fout, "@%d", pp->argc * 4);
317 static int align_value(int src_val)
320 awarn("bad align: %d\n", src_val);
326 return __builtin_ffs(src_val) - 1;
329 static int cmpstringp(const void *p1, const void *p2)
331 return strcmp(*(char * const *)p1, *(char * const *)p2);
334 /* XXX: maybe move to external file? */
335 static const char *unwanted_syms[] = {
361 static int is_unwanted_sym(const char *sym)
363 return bsearch(&sym, unwanted_syms, ARRAY_SIZE(unwanted_syms),
364 sizeof(unwanted_syms[0]), cmpstringp) != NULL;
367 int main(int argc, char *argv[])
369 FILE *fout, *fasm, *fhdr, *frlist;
370 const struct parsed_proto *pp;
371 int no_decorations = 0;
372 char comment_char = '#';
399 // -nd: no symbol decorations
400 printf("usage:\n%s [-nd] [-i] [-a] <.s> <.asm> <hdrf> [rlist]*\n",
405 for (arg = 1; arg < argc; arg++) {
406 if (IS(argv[arg], "-nd"))
408 else if (IS(argv[arg], "-i"))
409 g_cconv_novalidate = 1;
410 else if (IS(argv[arg], "-a")) {
421 fasm = fopen(asmfn, "r");
422 my_assert_not(fasm, NULL);
425 fhdr = fopen(hdrfn, "r");
426 my_assert_not(fhdr, NULL);
428 fout = fopen(argv[arg_out], "w");
429 my_assert_not(fout, NULL);
432 pub_syms = malloc(pub_sym_alloc * sizeof(pub_syms[0]));
433 my_assert_not(pub_syms, NULL);
436 rlist = malloc(rlist_alloc * sizeof(rlist[0]));
437 my_assert_not(rlist, NULL);
439 for (; arg < argc; arg++) {
440 frlist = fopen(argv[arg], "r");
441 my_assert_not(frlist, NULL);
443 while (fgets(line, sizeof(line), frlist)) {
445 if (*p == 0 || *p == ';')
448 p = next_word(words[0], sizeof(words[0]), p);
449 if (words[0][0] == 0)
452 if (rlist_cnt >= rlist_alloc) {
453 rlist_alloc = rlist_alloc * 2 + 64;
454 rlist = realloc(rlist, rlist_alloc * sizeof(rlist[0]));
455 my_assert_not(rlist, NULL);
457 rlist[rlist_cnt++] = strdup(words[0]);
465 qsort(rlist, rlist_cnt, sizeof(rlist[0]), cmpstringp);
467 qsort(unwanted_syms, ARRAY_SIZE(unwanted_syms),
468 sizeof(unwanted_syms[0]), cmpstringp);
473 next_section(fasm, line);
476 if (IS(line + 1, "text"))
479 if (IS(line + 1, "rdata"))
480 fprintf(fout, "\n.section .rodata\n");
481 else if (IS(line + 1, "data"))
482 fprintf(fout, "\n.data\n");
484 aerr("unhandled section: '%s'\n", line);
486 fprintf(fout, ".align %d\n", align_value(4));
488 while (fgets(line, sizeof(line), fasm))
498 if (IS_START(p, ";org") && sscanf(p + 5, "%Xh", &i) == 1) {
499 // ;org is only seen at section start, so assume . addr 0
502 fprintf(fout, "\t\t .skip 0x%x\n", i);
507 for (wordc = 0; wordc < ARRAY_SIZE(words); wordc++) {
508 p = sskip(next_word_s(words[wordc], sizeof(words[0]), p));
509 if (*p == 0 || *p == ';') {
520 if (IS_START(p, "sctclrtype"))
521 g_func_sym_pp = NULL;
524 if (wordc == 2 && IS(words[1], "ends"))
526 if (wordc <= 2 && IS(words[0], "end"))
529 aerr("unhandled: '%s'\n", words[0]);
532 if (IS(words[0], "assume"))
535 if (IS(words[0], "align")) {
536 val = parse_number(words[1]);
537 fprintf(fout, "\t\t .align %d", align_value(val));
542 type = parse_dx_directive(words[0]);
543 if (type == DXT_UNSPEC) {
544 type = parse_dx_directive(words[1]);
548 if (type == DXT_UNSPEC)
549 aerr("unhandled decl: '%s %s'\n", words[0], words[1]);
552 snprintf(last_sym, sizeof(last_sym), "%s", sym);
554 pp = proto_parse(fhdr, sym, 1);
556 g_func_sym_pp = NULL;
558 // public/global name
559 if (pub_sym_cnt >= pub_sym_alloc) {
561 pub_syms = realloc(pub_syms, pub_sym_alloc * sizeof(pub_syms[0]));
562 my_assert_not(pub_syms, NULL);
564 pub_syms[pub_sym_cnt++] = strdup(sym);
568 fprintf(fout, "%s%s:", no_decorations ? "" : "_", sym);
581 fprintf(fout, "\t\t ");
584 // fill out some unwanted strings with zeroes..
585 if (type == DXT_BYTE && words[w][0] == '\''
586 && is_unwanted_sym(last_sym))
589 for (; w < wordc; w++) {
590 if (words[w][0] == '\'') {
592 for (; *p && *p != '\''; p++)
596 // assume encoded byte
600 fprintf(fout, ".skip %d", len);
603 else if (type == DXT_BYTE
604 && (words[w][0] == '\''
605 || (w + 1 < wordc && words[w + 1][0] == '\'')))
607 // string; use asciz for most common case
608 if (w == wordc - 2 && IS(words[w + 1], "0")) {
609 fprintf(fout, ".asciz \"");
613 fprintf(fout, ".ascii \"");
615 for (; w < wordc; w++) {
616 if (words[w][0] == '\'') {
618 p2 = strchr(p, '\'');
620 aerr("unterminated string? '%s'\n", p);
621 memcpy(word, p, p2 - p);
623 fprintf(fout, "%s", escape_string(word));
626 val = parse_number(words[w]);
628 aerr("bad string trailing byte?\n");
629 fprintf(fout, "\\x%02lx", val);
636 if (w == wordc - 2) {
637 if (IS_START(words[w + 1], "dup(")) {
638 cnt = parse_number(words[w]);
639 p = words[w + 1] + 4;
643 memmove(word, p, p2 - p);
648 val = parse_number(word);
650 fprintf(fout, ".fill 0x%02lx,%d,0x%02lx",
651 cnt, type_size(type), val);
656 if (type == DXT_DWORD && words[w][0] == '\''
657 && words[w][5] == '\'' && strlen(words[w]) == 6)
663 val = (p[1] << 24) | (p[2] << 16) | (p[3] << 8) | p[4];
664 fprintf(fout, ".long 0x%lx", val);
665 snprintf(g_comment, sizeof(g_comment), "%s", words[w]);
669 if (type >= DXT_DWORD && strchr(words[w], '.'))
674 if (g_arm_mode && type == DXT_TEN) {
675 fprintf(fout, ".fill 10");
676 snprintf(g_comment, sizeof(g_comment), "%s %s",
677 type_name_float(type), words[w]);
680 fprintf(fout, "%s %s", type_name_float(type), words[w]);
685 fprintf(fout, "%s ", type_name(type));
686 for (; w < wordc; w++)
691 is_label = is_bss = 0;
692 if (w <= wordc - 2 && IS(words[w], "offset")) {
696 else if (IS(words[w], "?")) {
699 else if (type == DXT_DWORD
700 && !('0' <= words[w][0] && words[w][0] <= '9'))
711 if (IS_START(p, "loc_") || IS_START(p, "__imp")
712 || strchr(p, '?') || strchr(p, '@')
713 || bsearch(&p, rlist, rlist_cnt, sizeof(rlist[0]),
717 snprintf(g_comment, sizeof(g_comment), "%s", p);
720 pp = check_var(fhdr, sym, p);
722 fprintf(fout, "%s%s",
723 (no_decorations || p[0] == '_') ? "" : "_", p);
727 fprintf(fout, "%s", pp->name);
729 output_decorated_pp(fout, pp);
734 val = parse_number(words[w]);
736 fprintf(fout, "%ld", val);
738 fprintf(fout, "0x%lx", val);
745 if (g_comment[0] != 0) {
746 fprintf(fout, "\t\t%c %s", comment_char, g_comment);
756 for (i = 0; i < pub_sym_cnt; i++)
757 fprintf(fout, ".global %s%s\n",
758 no_decorations ? "" : "_", pub_syms[i]);
767 // vim:ts=2:shiftwidth=2:expandtab