[ia32rtools.git] / tools / protoparse.h


struct parsed_proto;

struct parsed_type {
	char *name;
	unsigned int is_array:1;
	unsigned int is_ptr:1;
};

struct parsed_proto_arg {
	char *reg;
	struct parsed_type type;
	struct parsed_proto *fptr;
	void *datap;
};

struct parsed_proto {
	char name[256];
	union {
		struct parsed_type ret_type;
		struct parsed_type type;
	};
	struct parsed_proto_arg arg[16];
	int argc;
	int argc_stack;
	int argc_reg;
	unsigned int is_func:1;
	unsigned int is_stdcall:1;
	unsigned int is_vararg:1;
	unsigned int is_fptr:1;
};

static const char *hdrfn;
static int hdrfline = 0;

static int find_protostr(char *dst, size_t dlen, FILE *fhdr,
	const char *fname, const char *sym_)
{
	const char *sym = sym_;
	const char *finc_name;
	FILE *finc;
	int symlen;
	int line = 0;
	int ret;
	char *p;

	if (sym[0] == '_' && strncmp(fname, "stdc", 4) == 0)
		sym++;
	symlen = strlen(sym);

	rewind(fhdr);

	while (fgets(dst, dlen, fhdr))
	{
		line++;
		if (strncmp(dst, "//#include ", 11) == 0) {
			finc_name = dst + 11;
			p = strpbrk(finc_name, "\r\n ");
			if (p != NULL)
				*p = 0;

			finc = fopen(finc_name, "r");
			if (finc == NULL) {
				printf("%s:%d: can't open '%s'\n",
					fname, line, finc_name);
				continue;
			}
			ret = find_protostr(dst, dlen, finc,
				finc_name, sym_);
			fclose(finc);
			if (ret == 0)
				break;
			continue;
		}
		if (strncmp(sskip(dst), "//", 2) == 0)
			continue;

		p = strstr(dst, sym);
		if (p != NULL && p > dst
		   && (my_isblank(p[-1]) || my_issep(p[-1]))
		   && (my_isblank(p[symlen]) || my_issep(p[symlen])))
			break;
	}
	hdrfline = line;

	if (feof(fhdr))
		return -1;

	p = dst + strlen(dst);
	for (p--; p > dst && my_isblank(*p); --p)
		*p = 0;

	return 0;
}

static int get_regparm(char *dst, size_t dlen, char *p)
{
	int i, o;

	if (*p != '<')
		return 0;

	for (o = 0, i = 1; o < dlen; i++) {
		if (p[i] == 0)
			return 0;
		if (p[i] == '>')
			break;
		dst[o++] = p[i];
	}
	dst[o] = 0;
	return i + 1;
}

// hmh..
static const char *known_type_mod[] = {
	"const",
	"signed",
	"unsigned",
	"struct",
	"enum",
};

static const char *known_ptr_types[] = {
	"HANDLE",
	"HMODULE",
	"HINSTANCE",
	"HWND",
	"HDC",
	"HGDIOBJ",
	"PLONG",
	"PDWORD",
	"PVOID",
	"PCVOID",
};

static const char *ignored_keywords[] = {
	"extern",
	"WINBASEAPI",
	"WINUSERAPI",
};

// returns ptr to char after type ends
static int typecmp(const char *n, const char *t)
{
	for (; *t != 0; n++, t++) {
		while (n[0] == ' ' && (n[1] == ' ' || n[1] == '*'))
			n++;
		while (t[0] == ' ' && (t[1] == ' ' || t[1] == '*'))
			t++;
		if (*n != *t)
			return *n - *t;
	}

	return 0;
}

static const char *skip_type_mod(const char *n)
{
	int len;
	int i;

	for (i = 0; i < ARRAY_SIZE(known_type_mod); i++) {
		len = strlen(known_type_mod[i]);
		if (strncmp(n, known_type_mod[i], len) != 0)
			continue;

		n += len;
		while (my_isblank(*n))
			n++;
		i = 0;
	}

	return n;
}

static int check_type(const char *name, struct parsed_type *type)
{
	const char *n, *n1;
	int ret = -1;
	int i;

	n = skip_type_mod(name);

	for (i = 0; i < ARRAY_SIZE(known_ptr_types); i++) {
		if (typecmp(n, known_ptr_types[i]))
			continue;

		type->is_ptr = 1;
		break;
	}

	if (n[0] == 'L' && n[1] == 'P')
		type->is_ptr = 1;

	// assume single word
	while (!my_isblank(*n) && !my_issep(*n))
		n++;

	while (1) {
		n1 = n;
		while (my_isblank(*n))
			n++;
		if (*n == '*') {
			type->is_ptr = 1;
			n++;
			continue;
		}
		break;
	}

	ret = n1 - name;
	type->name = strndup(name, ret);
	return ret;
}

/* args are always expanded to 32bit */
static const char *map_reg(const char *reg)
{
	const char *regs_f[] = { "eax", "ebx", "ecx", "edx", "esi", "edi" };
	const char *regs_w[] = { "ax",  "bx",  "cx",  "dx",  "si",  "di" };
	const char *regs_b[] = { "al",  "bl",  "cl",  "dl" };
	int i;

	for (i = 0; i < ARRAY_SIZE(regs_w); i++)
		if (IS(reg, regs_w[i]))
			return regs_f[i];

	for (i = 0; i < ARRAY_SIZE(regs_b); i++)
		if (IS(reg, regs_b[i]))
			return regs_f[i];

	return reg;
}

static int parse_protostr(char *protostr, struct parsed_proto *pp)
{
	struct parsed_proto_arg *arg;
	char regparm[16];
	char buf[256];
	char cconv[32];
	int xarg = 0;
	char *p, *p1;
	int i, l;
	int ret;

	p = sskip(protostr);
	if (p[0] == '/' && p[1] == '/') {
		printf("%s:%d: commented out?\n", hdrfn, hdrfline);
		p = sskip(p + 2);
	}

	// strip unneeded stuff
	for (p1 = p; p1[0] != 0 && p1[1] != 0; p1++) {
		if ((p1[0] == '/' && p1[1] == '*')
		 || (p1[0] == '*' && p1[1] == '/'))
			p1[0] = p1[1] = ' ';
	}

	for (i = 0; i < ARRAY_SIZE(ignored_keywords); i++) {
		l = strlen(ignored_keywords[i]);
		if (!strncmp(p, ignored_keywords[i], l) && my_isblank(p[l]))
			p = sskip(p + l + 1);
	}

	ret = check_type(p, &pp->ret_type);
	if (ret <= 0) {
		printf("%s:%d:%zd: unhandled return in '%s'\n",
			hdrfn, hdrfline, (p - protostr) + 1, protostr);
		return -1;
	}
	p = sskip(p + ret);

	if (!strchr(p, ')')) {
		p = next_idt(buf, sizeof(buf), p);
		p = sskip(p);
		if (buf[0] == 0) {
			printf("%s:%d:%zd: var name missing\n",
				hdrfn, hdrfline, (p - protostr) + 1);
			return -1;
		}
		strcpy(pp->name, buf);

		p1 = strchr(p, ']');
		if (p1 != NULL) {
			p = p1 + 1;
			pp->ret_type.is_array = 1;
		}
		return p - protostr;
	}

	pp->is_func = 1;

	if (*p == '(') {
		pp->is_fptr = 1;
		p = sskip(p + 1);
	}

	p = next_word(cconv, sizeof(cconv), p);
	p = sskip(p);
	if (cconv[0] == 0) {
		printf("%s:%d:%zd: cconv missing\n",
			hdrfn, hdrfline, (p - protostr) + 1);
		return -1;
	}
	if      (IS(cconv, "__cdecl"))
		pp->is_stdcall = 0;
	else if (IS(cconv, "__stdcall"))
		pp->is_stdcall = 1;
	else if (IS(cconv, "__fastcall"))
		pp->is_stdcall = 1;
	else if (IS(cconv, "__thiscall"))
		pp->is_stdcall = 1;
	else if (IS(cconv, "__userpurge"))
		pp->is_stdcall = 1; // in all cases seen..
	else if (IS(cconv, "__usercall"))
		pp->is_stdcall = 0; // ..or is it?
	else if (IS(cconv, "WINAPI"))
		pp->is_stdcall = 1;
	else {
		printf("%s:%d:%zd: unhandled cconv: '%s'\n",
			hdrfn, hdrfline, (p - protostr) + 1, cconv);
		return -1;
	}

	if (pp->is_fptr) {
		if (*p != '*') {
			printf("%s:%d:%zd: '*' expected\n",
				hdrfn, hdrfline, (p - protostr) + 1);
			return -1;
		}
		p = sskip(p + 1);
	}

	p = next_idt(buf, sizeof(buf), p);
	p = sskip(p);
	if (buf[0] == 0) {
		printf("%s:%d:%zd: func name missing\n",
			hdrfn, hdrfline, (p - protostr) + 1);
		return -1;
	}
	strcpy(pp->name, buf);

	ret = get_regparm(regparm, sizeof(regparm), p);
	if (ret > 0) {
		if (!IS(regparm, "eax") && !IS(regparm, "ax")
		 && !IS(regparm, "al"))
		{
			printf("%s:%d:%zd: bad regparm: %s\n",
				hdrfn, hdrfline, (p - protostr) + 1, regparm);
			return -1;
		}
		p += ret;
		p = sskip(p);
	}

	if (pp->is_fptr) {
		if (*p != ')') {
			printf("%s:%d:%zd: ')' expected\n",
				hdrfn, hdrfline, (p - protostr) + 1);
			return -1;
		}
		p = sskip(p + 1);
	}

	if (*p != '(') {
		printf("%s:%d:%zd: '(' expected, got '%c'\n",
				hdrfn, hdrfline, (p - protostr) + 1, *p);
		return -1;
	}
	p++;

	// check for x(void)
	p = sskip(p);
	if ((!strncmp(p, "void", 4) || !strncmp(p, "VOID", 4))
	   && *sskip(p + 4) == ')')
		p += 4;

	while (1) {
		p = sskip(p);
		if (*p == ')') {
			p++;
			break;
		}
		if (*p == ',')
			p = sskip(p + 1);

		if (!strncmp(p, "...", 3)) {
			pp->is_vararg = 1;
			p = sskip(p + 3);
			if (*p == ')') {
				p++;
				break;
			}
			printf("%s:%d:%zd: ')' expected\n",
				hdrfn, hdrfline, (p - protostr) + 1);
			return -1;
		}

		arg = &pp->arg[xarg];
		xarg++;

		p1 = p;
		ret = check_type(p, &arg->type);
		if (ret <= 0) {
			printf("%s:%d:%zd: unhandled type for arg%d\n",
				hdrfn, hdrfline, (p - protostr) + 1, xarg);
			return -1;
		}
		p = sskip(p + ret);

		if (*p == '(') {
			// func ptr
			arg->fptr = calloc(1, sizeof(*arg->fptr));
			ret = parse_protostr(p1, arg->fptr);
			if (ret < 0) {
				printf("%s:%d:%zd: funcarg parse failed\n",
					hdrfn, hdrfline, p1 - protostr);
				return -1;
			}
			// we'll treat it as void * for non-calls
			arg->type.name = "void *";
			arg->type.is_ptr = 1;

			p = p1 + ret;
		}

		p = next_idt(buf, sizeof(buf), p);
		p = sskip(p);
#if 0
		if (buf[0] == 0) {
			printf("%s:%d:%zd: idt missing for arg%d\n",
				hdrfn, hdrfline, (p - protostr) + 1, xarg);
			return -1;
		}
#endif
		arg->reg = NULL;

		ret = get_regparm(regparm, sizeof(regparm), p);
		if (ret > 0) {
			p += ret;
			p = sskip(p);

			arg->reg = strdup(map_reg(regparm));
		}
	}

	if (xarg > 0 && (IS(cconv, "__fastcall") || IS(cconv, "__thiscall"))) {
		if (pp->arg[0].reg != NULL) {
			printf("%s:%d: %s with arg1 spec %s?\n",
				hdrfn, hdrfline, cconv, pp->arg[0].reg);
		}
		pp->arg[0].reg = strdup("ecx");
	}

	if (xarg > 1 && IS(cconv, "__fastcall")) {
		if (pp->arg[1].reg != NULL) {
			printf("%s:%d: %s with arg2 spec %s?\n",
				hdrfn, hdrfline, cconv, pp->arg[1].reg);
		}
		pp->arg[1].reg = strdup("edx");
	}

	if (pp->is_vararg && pp->is_stdcall) {
		printf("%s:%d: vararg stdcall?\n", hdrfn, hdrfline);
		return -1;
	}

	pp->argc = xarg;

	for (i = 0; i < pp->argc; i++) {
		if (pp->arg[i].reg == NULL)
			pp->argc_stack++;
		else
			pp->argc_reg++;
	}

	return p - protostr;
}

static int proto_parse(FILE *fhdr, const char *sym, struct parsed_proto *pp)
{
	char protostr[256];
	int ret;

	memset(pp, 0, sizeof(*pp));

	ret = find_protostr(protostr, sizeof(protostr), fhdr, hdrfn, sym);
	if (ret != 0) {
		printf("%s: sym '%s' is missing\n", hdrfn, sym);
		return ret;
	}

	return parse_protostr(protostr, pp) < 0 ? -1 : 0;
}

static void proto_release(struct parsed_proto *pp)
{
	int i;

	for (i = 0; i < pp->argc; i++) {
		if (pp->arg[i].reg != NULL)
			free(pp->arg[i].reg);
		if (pp->arg[i].type.name != NULL)
			free(pp->arg[i].type.name);
		if (pp->arg[i].fptr != NULL)
			free(pp->arg[i].fptr);
	}
	if (pp->ret_type.name != NULL)
		free(pp->ret_type.name);
}
Commit	Line	Data
c36e914d	1
39b168b8	2	struct parsed_proto;
39b168b8	3
3ebea2cf	4	struct parsed_type {
	5	char *name;
	6	unsigned int is_array:1;
	7	unsigned int is_ptr:1;
	8	};
	9
39b168b8	10	struct parsed_proto_arg {
39b168b8	11	char *reg;
3ebea2cf	12	struct parsed_type type;
39b168b8	13	struct parsed_proto *fptr;
	14	void *datap;
	15	};
	16
c36e914d	17	struct parsed_proto {
39b168b8	18	char name[256];
3ebea2cf	19	union {
	20	struct parsed_type ret_type;
	21	struct parsed_type type;
	22	};
39b168b8	23	struct parsed_proto_arg arg[16];
c36e914d	24	int argc;
	25	int argc_stack;
	26	int argc_reg;
06c5d854	27	unsigned int is_func:1;
39b168b8	28	unsigned int is_stdcall:1;
7ba45c34	29	unsigned int is_vararg:1;
39b168b8	30	unsigned int is_fptr:1;
c36e914d	31	};
	32
	33	static const char *hdrfn;
	34	static int hdrfline = 0;
	35
39b168b8	36	static int find_protostr(char dst, size_t dlen, FILE fhdr,
39b168b8	37	const char fname, const char sym_)
c36e914d	38	{
39b168b8	39	const char *sym = sym_;
06c5d854	40	const char *finc_name;
39b168b8	41	FILE *finc;
39b168b8	42	int symlen;
c36e914d	43	int line = 0;
39b168b8	44	int ret;
c36e914d	45	char *p;
c36e914d	46
39b168b8	47	if (sym[0] == '_' && strncmp(fname, "stdc", 4) == 0)
	48	sym++;
	49	symlen = strlen(sym);
	50
c36e914d	51	rewind(fhdr);
	52
	53	while (fgets(dst, dlen, fhdr))
	54	{
	55	line++;
06c5d854	56	if (strncmp(dst, "//#include ", 11) == 0) {
	57	finc_name = dst + 11;
	58	p = strpbrk(finc_name, "\r\n ");
39b168b8	59	if (p != NULL)
	60	*p = 0;
	61
06c5d854	62	finc = fopen(finc_name, "r");
39b168b8	63	if (finc == NULL) {
39b168b8	64	printf("%s:%d: can't open '%s'\n",
06c5d854	65	fname, line, finc_name);
39b168b8	66	continue;
	67	}
	68	ret = find_protostr(dst, dlen, finc,
06c5d854	69	finc_name, sym_);
39b168b8	70	fclose(finc);
	71	if (ret == 0)
	72	break;
	73	continue;
	74	}
06c5d854	75	if (strncmp(sskip(dst), "//", 2) == 0)
06c5d854	76	continue;
39b168b8	77
39b168b8	78	p = strstr(dst, sym);
06c5d854	79	if (p != NULL && p > dst
06c5d854	80	&& (my_isblank(p[-1]) \|\| my_issep(p[-1]))
39b168b8	81	&& (my_isblank(p[symlen]) \|\| my_issep(p[symlen])))
c36e914d	82	break;
	83	}
	84	hdrfline = line;
	85
	86	if (feof(fhdr))
	87	return -1;
	88
	89	p = dst + strlen(dst);
	90	for (p--; p > dst && my_isblank(*p); --p)
	91	*p = 0;
	92
	93	return 0;
	94	}
	95
	96	static int get_regparm(char dst, size_t dlen, char p)
	97	{
	98	int i, o;
	99
	100	if (*p != '<')
	101	return 0;
	102
	103	for (o = 0, i = 1; o < dlen; i++) {
	104	if (p[i] == 0)
	105	return 0;
	106	if (p[i] == '>')
	107	break;
	108	dst[o++] = p[i];
	109	}
	110	dst[o] = 0;
	111	return i + 1;
	112	}
	113
	114	// hmh..
64c59faf	115	static const char *known_type_mod[] = {
	116	"const",
	117	"signed",
	118	"unsigned",
3ebea2cf	119	"struct",
3ebea2cf	120	"enum",
64c59faf	121	};
64c59faf	122
3ebea2cf	123	static const char *known_ptr_types[] = {
c36e914d	124	"HANDLE",
3ebea2cf	125	"HMODULE",
3ebea2cf	126	"HINSTANCE",
c36e914d	127	"HWND",
3ebea2cf	128	"HDC",
	129	"HGDIOBJ",
	130	"PLONG",
	131	"PDWORD",
	132	"PVOID",
	133	"PCVOID",
	134	};
	135
	136	static const char *ignored_keywords[] = {
	137	"extern",
	138	"WINBASEAPI",
	139	"WINUSERAPI",
c36e914d	140	};
c36e914d	141
64c59faf	142	// returns ptr to char after type ends
3ebea2cf	143	static int typecmp(const char n, const char t)
39b168b8	144	{
	145	for (; *t != 0; n++, t++) {
	146	while (n[0] == ' ' && (n[1] == ' ' \|\| n[1] == '*'))
	147	n++;
	148	while (t[0] == ' ' && (t[1] == ' ' \|\| t[1] == '*'))
	149	t++;
	150	if (n != t)
3ebea2cf	151	return n - t;
39b168b8	152	}
39b168b8	153
3ebea2cf	154	return 0;
39b168b8	155	}
39b168b8	156
3ebea2cf	157	static const char skip_type_mod(const char n)
c36e914d	158	{
64c59faf	159	int len;
39b168b8	160	int i;
c36e914d	161
64c59faf	162	for (i = 0; i < ARRAY_SIZE(known_type_mod); i++) {
	163	len = strlen(known_type_mod[i]);
	164	if (strncmp(n, known_type_mod[i], len) != 0)
	165	continue;
	166
	167	n += len;
	168	while (my_isblank(*n))
	169	n++;
	170	i = 0;
	171	}
	172
3ebea2cf	173	return n;
	174	}
	175
	176	static int check_type(const char name, struct parsed_type type)
	177	{
	178	const char n, n1;
	179	int ret = -1;
	180	int i;
	181
	182	n = skip_type_mod(name);
	183
	184	for (i = 0; i < ARRAY_SIZE(known_ptr_types); i++) {
	185	if (typecmp(n, known_ptr_types[i]))
64c59faf	186	continue;
64c59faf	187
3ebea2cf	188	type->is_ptr = 1;
3ebea2cf	189	break;
c36e914d	190	}
c36e914d	191
3ebea2cf	192	if (n[0] == 'L' && n[1] == 'P')
	193	type->is_ptr = 1;
	194
	195	// assume single word
	196	while (!my_isblank(n) && !my_issep(n))
	197	n++;
	198
	199	while (1) {
	200	n1 = n;
	201	while (my_isblank(*n))
	202	n++;
	203	if (n == '') {
	204	type->is_ptr = 1;
	205	n++;
	206	continue;
	207	}
	208	break;
	209	}
	210
	211	ret = n1 - name;
	212	type->name = strndup(name, ret);
	213	return ret;
c36e914d	214	}
	215
	216	/* args are always expanded to 32bit */
	217	static const char map_reg(const char reg)
	218	{
	219	const char *regs_f[] = { "eax", "ebx", "ecx", "edx", "esi", "edi" };
	220	const char *regs_w[] = { "ax", "bx", "cx", "dx", "si", "di" };
	221	const char *regs_b[] = { "al", "bl", "cl", "dl" };
	222	int i;
	223
	224	for (i = 0; i < ARRAY_SIZE(regs_w); i++)
	225	if (IS(reg, regs_w[i]))
	226	return regs_f[i];
	227
	228	for (i = 0; i < ARRAY_SIZE(regs_b); i++)
	229	if (IS(reg, regs_b[i]))
	230	return regs_f[i];
	231
	232	return reg;
	233	}
	234
	235	static int parse_protostr(char protostr, struct parsed_proto pp)
	236	{
39b168b8	237	struct parsed_proto_arg *arg;
c36e914d	238	char regparm[16];
	239	char buf[256];
	240	char cconv[32];
c36e914d	241	int xarg = 0;
39b168b8	242	char p, p1;
3ebea2cf	243	int i, l;
c36e914d	244	int ret;
c36e914d	245
06c5d854	246	p = sskip(protostr);
c36e914d	247	if (p[0] == '/' && p[1] == '/') {
06c5d854	248	printf("%s:%d: commented out?\n", hdrfn, hdrfline);
c36e914d	249	p = sskip(p + 2);
	250	}
	251
06c5d854	252	// strip unneeded stuff
	253	for (p1 = p; p1[0] != 0 && p1[1] != 0; p1++) {
	254	if ((p1[0] == '/' && p1[1] == '*')
	255	\|\| (p1[0] == '*' && p1[1] == '/'))
	256	p1[0] = p1[1] = ' ';
	257	}
	258
3ebea2cf	259	for (i = 0; i < ARRAY_SIZE(ignored_keywords); i++) {
	260	l = strlen(ignored_keywords[i]);
	261	if (!strncmp(p, ignored_keywords[i], l) && my_isblank(p[l]))
	262	p = sskip(p + l + 1);
	263	}
06c5d854	264
3ebea2cf	265	ret = check_type(p, &pp->ret_type);
3ebea2cf	266	if (ret <= 0) {
63df67be	267	printf("%s:%d:%zd: unhandled return in '%s'\n",
c36e914d	268	hdrfn, hdrfline, (p - protostr) + 1, protostr);
39b168b8	269	return -1;
c36e914d	270	}
64c59faf	271	p = sskip(p + ret);
c36e914d	272
06c5d854	273	if (!strchr(p, ')')) {
	274	p = next_idt(buf, sizeof(buf), p);
	275	p = sskip(p);
	276	if (buf[0] == 0) {
63df67be	277	printf("%s:%d:%zd: var name missing\n",
06c5d854	278	hdrfn, hdrfline, (p - protostr) + 1);
	279	return -1;
	280	}
	281	strcpy(pp->name, buf);
	282
	283	p1 = strchr(p, ']');
	284	if (p1 != NULL) {
	285	p = p1 + 1;
3ebea2cf	286	pp->ret_type.is_array = 1;
06c5d854	287	}
	288	return p - protostr;
	289	}
	290
	291	pp->is_func = 1;
	292
39b168b8	293	if (*p == '(') {
	294	pp->is_fptr = 1;
	295	p = sskip(p + 1);
	296	}
	297
c36e914d	298	p = next_word(cconv, sizeof(cconv), p);
	299	p = sskip(p);
	300	if (cconv[0] == 0) {
63df67be	301	printf("%s:%d:%zd: cconv missing\n",
c36e914d	302	hdrfn, hdrfline, (p - protostr) + 1);
39b168b8	303	return -1;
c36e914d	304	}
	305	if (IS(cconv, "__cdecl"))
	306	pp->is_stdcall = 0;
	307	else if (IS(cconv, "__stdcall"))
	308	pp->is_stdcall = 1;
	309	else if (IS(cconv, "__fastcall"))
	310	pp->is_stdcall = 1;
	311	else if (IS(cconv, "__thiscall"))
	312	pp->is_stdcall = 1;
	313	else if (IS(cconv, "__userpurge"))
	314	pp->is_stdcall = 1; // in all cases seen..
	315	else if (IS(cconv, "__usercall"))
	316	pp->is_stdcall = 0; // ..or is it?
64c59faf	317	else if (IS(cconv, "WINAPI"))
64c59faf	318	pp->is_stdcall = 1;
c36e914d	319	else {
63df67be	320	printf("%s:%d:%zd: unhandled cconv: '%s'\n",
c36e914d	321	hdrfn, hdrfline, (p - protostr) + 1, cconv);
39b168b8	322	return -1;
	323	}
	324
	325	if (pp->is_fptr) {
	326	if (p != '') {
63df67be	327	printf("%s:%d:%zd: '*' expected\n",
39b168b8	328	hdrfn, hdrfline, (p - protostr) + 1);
	329	return -1;
	330	}
	331	p = sskip(p + 1);
c36e914d	332	}
	333
	334	p = next_idt(buf, sizeof(buf), p);
	335	p = sskip(p);
	336	if (buf[0] == 0) {
63df67be	337	printf("%s:%d:%zd: func name missing\n",
39b168b8	338	hdrfn, hdrfline, (p - protostr) + 1);
39b168b8	339	return -1;
c36e914d	340	}
39b168b8	341	strcpy(pp->name, buf);
c36e914d	342
	343	ret = get_regparm(regparm, sizeof(regparm), p);
	344	if (ret > 0) {
	345	if (!IS(regparm, "eax") && !IS(regparm, "ax")
	346	&& !IS(regparm, "al"))
	347	{
63df67be	348	printf("%s:%d:%zd: bad regparm: %s\n",
c36e914d	349	hdrfn, hdrfline, (p - protostr) + 1, regparm);
39b168b8	350	return -1;
c36e914d	351	}
	352	p += ret;
	353	p = sskip(p);
	354	}
	355
39b168b8	356	if (pp->is_fptr) {
39b168b8	357	if (*p != ')') {
63df67be	358	printf("%s:%d:%zd: ')' expected\n",
39b168b8	359	hdrfn, hdrfline, (p - protostr) + 1);
	360	return -1;
	361	}
	362	p = sskip(p + 1);
	363	}
	364
c36e914d	365	if (*p != '(') {
63df67be	366	printf("%s:%d:%zd: '(' expected, got '%c'\n",
c36e914d	367	hdrfn, hdrfline, (p - protostr) + 1, *p);
39b168b8	368	return -1;
c36e914d	369	}
	370	p++;
	371
39b168b8	372	// check for x(void)
39b168b8	373	p = sskip(p);
06c5d854	374	if ((!strncmp(p, "void", 4) \|\| !strncmp(p, "VOID", 4))
06c5d854	375	&& *sskip(p + 4) == ')')
39b168b8	376	p += 4;
39b168b8	377
c36e914d	378	while (1) {
c36e914d	379	p = sskip(p);
39b168b8	380	if (*p == ')') {
39b168b8	381	p++;
c36e914d	382	break;
39b168b8	383	}
c36e914d	384	if (*p == ',')
	385	p = sskip(p + 1);
	386
7ba45c34	387	if (!strncmp(p, "...", 3)) {
	388	pp->is_vararg = 1;
	389	p = sskip(p + 3);
	390	if (*p == ')') {
	391	p++;
	392	break;
	393	}
63df67be	394	printf("%s:%d:%zd: ')' expected\n",
7ba45c34	395	hdrfn, hdrfline, (p - protostr) + 1);
	396	return -1;
	397	}
	398
39b168b8	399	arg = &pp->arg[xarg];
c36e914d	400	xarg++;
c36e914d	401
39b168b8	402	p1 = p;
3ebea2cf	403	ret = check_type(p, &arg->type);
3ebea2cf	404	if (ret <= 0) {
63df67be	405	printf("%s:%d:%zd: unhandled type for arg%d\n",
c36e914d	406	hdrfn, hdrfline, (p - protostr) + 1, xarg);
39b168b8	407	return -1;
c36e914d	408	}
64c59faf	409	p = sskip(p + ret);
c36e914d	410
39b168b8	411	if (*p == '(') {
	412	// func ptr
	413	arg->fptr = calloc(1, sizeof(*arg->fptr));
	414	ret = parse_protostr(p1, arg->fptr);
	415	if (ret < 0) {
63df67be	416	printf("%s:%d:%zd: funcarg parse failed\n",
39b168b8	417	hdrfn, hdrfline, p1 - protostr);
	418	return -1;
	419	}
	420	// we'll treat it as void * for non-calls
3ebea2cf	421	arg->type.name = "void *";
3ebea2cf	422	arg->type.is_ptr = 1;
39b168b8	423
	424	p = p1 + ret;
	425	}
	426
c36e914d	427	p = next_idt(buf, sizeof(buf), p);
	428	p = sskip(p);
	429	#if 0
	430	if (buf[0] == 0) {
63df67be	431	printf("%s:%d:%zd: idt missing for arg%d\n",
c36e914d	432	hdrfn, hdrfline, (p - protostr) + 1, xarg);
39b168b8	433	return -1;
c36e914d	434	}
c36e914d	435	#endif
39b168b8	436	arg->reg = NULL;
c36e914d	437
	438	ret = get_regparm(regparm, sizeof(regparm), p);
	439	if (ret > 0) {
	440	p += ret;
	441	p = sskip(p);
	442
39b168b8	443	arg->reg = strdup(map_reg(regparm));
c36e914d	444	}
	445	}
	446
	447	if (xarg > 0 && (IS(cconv, "__fastcall") \|\| IS(cconv, "__thiscall"))) {
	448	if (pp->arg[0].reg != NULL) {
	449	printf("%s:%d: %s with arg1 spec %s?\n",
	450	hdrfn, hdrfline, cconv, pp->arg[0].reg);
	451	}
	452	pp->arg[0].reg = strdup("ecx");
	453	}
	454
	455	if (xarg > 1 && IS(cconv, "__fastcall")) {
	456	if (pp->arg[1].reg != NULL) {
	457	printf("%s:%d: %s with arg2 spec %s?\n",
	458	hdrfn, hdrfline, cconv, pp->arg[1].reg);
	459	}
	460	pp->arg[1].reg = strdup("edx");
	461	}
	462
7ba45c34	463	if (pp->is_vararg && pp->is_stdcall) {
	464	printf("%s:%d: vararg stdcall?\n", hdrfn, hdrfline);
	465	return -1;
	466	}
	467
c36e914d	468	pp->argc = xarg;
	469
	470	for (i = 0; i < pp->argc; i++) {
	471	if (pp->arg[i].reg == NULL)
	472	pp->argc_stack++;
	473	else
	474	pp->argc_reg++;
	475	}
	476
39b168b8	477	return p - protostr;
c36e914d	478	}
	479
	480	static int proto_parse(FILE fhdr, const char sym, struct parsed_proto *pp)
	481	{
	482	char protostr[256];
	483	int ret;
	484
	485	memset(pp, 0, sizeof(*pp));
	486
39b168b8	487	ret = find_protostr(protostr, sizeof(protostr), fhdr, hdrfn, sym);
c36e914d	488	if (ret != 0) {
	489	printf("%s: sym '%s' is missing\n", hdrfn, sym);
	490	return ret;
	491	}
	492
39b168b8	493	return parse_protostr(protostr, pp) < 0 ? -1 : 0;
c36e914d	494	}
	495
	496	static void proto_release(struct parsed_proto *pp)
	497	{
	498	int i;
	499
	500	for (i = 0; i < pp->argc; i++) {
39b168b8	501	if (pp->arg[i].reg != NULL)
c36e914d	502	free(pp->arg[i].reg);
3ebea2cf	503	if (pp->arg[i].type.name != NULL)
3ebea2cf	504	free(pp->arg[i].type.name);
39b168b8	505	if (pp->arg[i].fptr != NULL)
39b168b8	506	free(pp->arg[i].fptr);
c36e914d	507	}
3ebea2cf	508	if (pp->ret_type.name != NULL)
3ebea2cf	509	free(pp->ret_type.name);
c36e914d	510	}