#include "my_assert.h"
#include "my_str.h"
-static int find_protostr(char *dst, size_t dlen, FILE *fhdr,
- const char *sym, int *pline)
-{
- int line = 0;
- char *p;
+#define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
+#define IS(w, y) !strcmp(w, y)
- while (fgets(dst, dlen, fhdr))
- {
- line++;
- if (strstr(dst, sym) != NULL)
- break;
- }
- *pline = line;
+#include "protoparse.h"
- if (feof(fhdr))
- return -1;
+static int is_x86_reg_saved(const char *reg)
+{
+ static const char *nosave_regs[] = { "eax", "edx", "ecx" };
+ int nosave = 0;
+ int r;
- p = dst + strlen(dst);
- for (p--; p > dst && my_isblank(*p); --p)
- *p = 0;
+ for (r = 0; r < ARRAY_SIZE(nosave_regs); r++)
+ if (strcmp(reg, nosave_regs[r]) == 0)
+ nosave = 1;
- return 0;
+ return !nosave;
}
-static int get_regparm(char *dst, size_t dlen, char *p)
+static void out_toasm_x86(FILE *f, char *sym, struct parsed_proto *pp)
{
- int i, o;
+ int must_save = 0;
+ int sarg_ofs = 1; // stack offset to args, in DWORDs
+ int args_repushed = 0;
+ int i;
+
+ for (i = 0; i < pp->argc; i++) {
+ if (pp->arg[i].reg != NULL)
+ must_save |= is_x86_reg_saved(pp->arg[i].reg);
+ }
+
+ fprintf(f, ".global _%s\n", sym);
+ fprintf(f, "_%s:\n", sym);
- if (*p != '<')
- return 0;
+ if (pp->argc_reg == 0 && !pp->is_stdcall) {
+ fprintf(f, "\tjmp %s\n\n", sym);
+ return;
+ }
+
+ if (pp->argc_stack == 0 && !must_save && !pp->is_stdcall) {
+ // load arg regs
+ for (i = 0; i < pp->argc; i++) {
+ fprintf(f, "\tmovl %d(%%esp), %%%s\n",
+ (i + sarg_ofs) * 4, pp->arg[i].reg);
+ }
+ fprintf(f, "\tjmp %s\n\n", sym);
+ return;
+ }
- for (o = 0, i = 1; o < dlen; i++) {
- if (p[i] == 0)
- return 0;
- if (p[i] == '>')
- break;
- dst[o++] = p[i];
+ // save the regs
+ for (i = 0; i < pp->argc; i++) {
+ if (pp->arg[i].reg != NULL && is_x86_reg_saved(pp->arg[i].reg)) {
+ fprintf(f, "\tpushl %%%s\n", pp->arg[i].reg);
+ sarg_ofs++;
+ }
}
- dst[o] = 0;
- return i + 1;
+
+ // reconstruct arg stack
+ for (i = pp->argc - 1; i >= 0; i--) {
+ if (pp->arg[i].reg == NULL) {
+ fprintf(f, "\tmovl %d(%%esp), %%eax\n",
+ (i + sarg_ofs) * 4);
+ fprintf(f, "\tpushl %%eax\n");
+ sarg_ofs++;
+ args_repushed++;
+ }
+ }
+ my_assert(args_repushed, pp->argc_stack);
+
+ // load arg regs
+ for (i = 0; i < pp->argc; i++) {
+ if (pp->arg[i].reg != NULL) {
+ fprintf(f, "\tmovl %d(%%esp), %%%s\n",
+ (i + sarg_ofs) * 4, pp->arg[i].reg);
+ }
+ }
+
+ fprintf(f, "\n\t# %s\n", pp->is_stdcall ? "__stdcall" : "__cdecl");
+ fprintf(f, "\tcall %s\n\n", sym);
+
+ if (args_repushed && !pp->is_stdcall)
+ fprintf(f, "\tadd $%d,%%esp\n", args_repushed * 4);
+
+ // restore regs
+ for (i = pp->argc - 1; i >= 0; i--) {
+ if (pp->arg[i].reg != NULL && is_x86_reg_saved(pp->arg[i].reg))
+ fprintf(f, "\tpopl %%%s\n", pp->arg[i].reg);
+ }
+
+ fprintf(f, "\tret\n\n");
}
-static const char *known_types[] = {
- "unsigned int",
- "signed int",
- "int",
- "void",
- "DWORD",
- "HMODULE",
- "HANDLE",
- "HWND",
-};
-
-static int check_type(const char *name)
+static void out_fromasm_x86(FILE *f, char *sym, struct parsed_proto *pp)
{
- int i, l;
+ int sarg_ofs = 1; // stack offset to args, in DWORDs
+ int stack_args;
+ int i;
+
+ fprintf(f, "# %s\n", pp->is_stdcall ? "__stdcall" : "__cdecl");
+ fprintf(f, ".global %s\n", sym);
+ fprintf(f, "%s:\n", sym);
- for (i = 0; i < sizeof(known_types) / sizeof(known_types[0]); i++) {
- l = strlen(known_types[i]);
- if (strncmp(known_types[i], name, l) == 0)
- return l;
+ if (pp->argc_reg == 0 && !pp->is_stdcall) {
+ fprintf(f, "\tjmp _%s\n\n", sym);
+ return;
}
- return 0;
+ fprintf(f, "\tpushl %%edx\n"); // just in case..
+ sarg_ofs++;
+
+ // construct arg stack
+ stack_args = pp->argc_stack;
+ for (i = pp->argc - 1; i >= 0; i--) {
+ if (pp->arg[i].reg == NULL) {
+ fprintf(f, "\tmovl %d(%%esp), %%edx\n",
+ (sarg_ofs + stack_args - 1) * 4);
+ fprintf(f, "\tpushl %%edx\n");
+ stack_args--;
+ }
+ else {
+ fprintf(f, "\tpushl %%%s\n", pp->arg[i].reg);
+ }
+ sarg_ofs++;
+ }
+
+ // no worries about calling conventions - always __cdecl
+ fprintf(f, "\n\tcall _%s\n\n", sym);
+
+ if (sarg_ofs > 2)
+ fprintf(f, "\tadd $%d,%%esp\n", (sarg_ofs - 2) * 4);
+
+ fprintf(f, "\tpopl %%edx\n");
+
+ if (pp->is_stdcall && pp->argc_stack)
+ fprintf(f, "\tret $%d\n\n", pp->argc_stack * 4);
+ else
+ fprintf(f, "\tret\n\n");
}
int main(int argc, char *argv[])
{
- FILE *fout, *fsyms, *fhdr;
- const char *hdrfn;
- char protostr[256];
+ FILE *fout, *fsyms_to, *fsyms_from, *fhdr;
+ struct parsed_proto pp;
char line[256];
char sym[256];
- char buf[256];
- char regparm[16];
- char *p;
- int first_regparm = 0;
- int have_regparm;
- int pline = 0;
- int xarg;
int ret;
- if (argc != 4) {
- // -c - patch callsites
- printf("usage:\n%s <bridge.s> <symf> <hdrf>\n",
+ if (argc != 5) {
+ printf("usage:\n%s <bridge.s> <toasm_symf> <fromasm_symf> <hdrf>\n",
argv[0]);
return 1;
}
- hdrfn = argv[3];
+ hdrfn = argv[4];
fhdr = fopen(hdrfn, "r");
my_assert_not(fhdr, NULL);
- fsyms = fopen(argv[2], "r");
- my_assert_not(fsyms, NULL);
+ fsyms_from = fopen(argv[3], "r");
+ my_assert_not(fsyms_from, NULL);
+
+ fsyms_to = fopen(argv[2], "r");
+ my_assert_not(fsyms_to, NULL);
fout = fopen(argv[1], "w");
my_assert_not(fout, NULL);
fprintf(fout, ".text\n\n");
+ fprintf(fout, "# to asm\n\n");
- while (fgets(line, sizeof(line), fsyms))
+ while (fgets(line, sizeof(line), fsyms_to))
{
next_word(sym, sizeof(sym), line);
if (sym[0] == 0 || sym[0] == ';' || sym[0] == '#')
continue;
- ret = find_protostr(protostr, sizeof(protostr), fhdr,
- sym, &pline);
- if (ret != 0) {
- printf("%s: sym '%s' is missing\n",
- hdrfn, sym);
- return 1;
- }
-
- p = protostr;
- if (p[0] == '/' && p[1] == '/') {
- printf("warning: decl for sym '%s' is commented out\n", sym);
- p = sskip(p + 2);
- }
+ ret = proto_parse(fhdr, sym, &pp);
+ if (ret)
+ goto out;
- ret = check_type(p);
- if (ret <= 0) {
- printf("%s:%d:%ld: unhandled return in '%s'\n",
- hdrfn, pline, (p - protostr) + 1, protostr);
- return 1;
- }
- p += ret;
- p = sskip(p);
-
- // ignore calling convention specifier, for now
- p = next_word(buf, sizeof(buf), p);
- p = sskip(p);
- if (buf[0] == 0) {
- printf("%s:%d:%ld: cconv missing\n",
- hdrfn, pline, (p - protostr) + 1);
- return 1;
- }
+ out_toasm_x86(fout, sym, &pp);
+ proto_release(&pp);
+ }
- p = next_idt(buf, sizeof(buf), p);
- p = sskip(p);
- if (buf[0] == 0) {
- printf("%s:%d:%ld: func name missing\n",
- hdrfn, pline, (p - protostr) + 1);
- return 1;
- }
+ fprintf(fout, "# from asm\n\n");
- ret = get_regparm(regparm, sizeof(regparm), p);
- if (ret > 0) {
- if (strcmp(regparm, "eax") && strcmp(regparm, "ax")) {
- printf("%s:%d:%ld: bad regparm: %s\n",
- hdrfn, pline, (p - protostr) + 1, regparm);
- return 1;
- }
- p += ret;
- p = sskip(p);
- }
+ while (fgets(line, sizeof(line), fsyms_from))
+ {
+ next_word(sym, sizeof(sym), line);
+ if (sym[0] == 0 || sym[0] == ';' || sym[0] == '#')
+ continue;
- if (*p != '(') {
- printf("%s:%d:%ld: '(' expected, got '%c'\n",
- hdrfn, pline, (p - protostr) + 1, *p);
- return 1;
- }
- p++;
-
- fprintf(fout, ".global _asm_%s\n", sym);
- fprintf(fout, "_asm_%s:\n", sym);
-
- xarg = 1;
- while (1) {
- p = sskip(p);
- if (*p == ')')
- break;
- if (*p == ',')
- p = sskip(p + 1);
-
- ret = check_type(p);
- if (ret <= 0) {
- printf("%s:%d:%ld: unhandled type for arg%d\n",
- hdrfn, pline, (p - protostr) + 1, xarg);
- return 1;
- }
- p += ret;
- p = sskip(p);
-
- p = next_idt(buf, sizeof(buf), p);
- p = sskip(p);
- if (buf[0] == 0) {
- printf("%s:%d:%ld: idt missing for arg%d\n",
- hdrfn, pline, (p - protostr) + 1, xarg);
- return 1;
- }
-
- have_regparm = 0;
- ret = get_regparm(regparm, sizeof(regparm), p);
- if (ret > 0) {
- p += ret;
- p = sskip(p);
-
- have_regparm = 1;
- fprintf(fout, "\t movl %d(%%esp), %%%s\n",
- xarg * 4, regparm);
- }
- if (xarg == 1)
- first_regparm = have_regparm;
- else if (have_regparm != first_regparm) {
- printf("%s:%d:%ld: mixed regparm is unhandled\n",
- hdrfn, pline, (p - protostr) + 1);
- return 1;
- }
- }
+ ret = proto_parse(fhdr, sym, &pp);
+ if (ret)
+ goto out;
- fprintf(fout, "\t jmp %s\n\n", sym);
+ out_fromasm_x86(fout, sym, &pp);
+ proto_release(&pp);
}
+ ret = 0;
+out:
fclose(fout);
- return 0;
+ fclose(fsyms_to);
+ fclose(fsyms_from);
+ fclose(fhdr);
+ if (ret)
+ remove(argv[1]);
+
+ return ret;
}