/* radare - LGPL - Copyright 2010-2024 pancake */

#include <r_debug.h>

R_API ut64 r_debug_arg_get(RDebug *dbg, const char *cc, int num) {
	R_RETURN_VAL_IF_FAIL (dbg, UT64_MAX);
	if (!cc) {
		cc = r_anal_syscc_default (dbg->anal);
	}
	if (dbg->anal && !R_STR_ISEMPTY (cc)) {
		if (!strcmp (cc, "stdcall") || !strcmp (cc, "pascal")) {
			ut64 sp = r_debug_reg_get (dbg, "SP");
			if (dbg->bits == 64) {
				ut64 n64;
				sp += 8; // skip return address, assume we are inside the call
				sp += 8 * num;
				if (dbg->iob.read_at (dbg->iob.io, sp, (ut8*)&n64, sizeof (ut64)) != sizeof (ut64)) {
					return 0;
				}
				// TODO: honor endianness of platform
				return (ut64)n64;
			} else {
				sp += 4; // skip return address, assume we are inside the call
				sp += 4 * num;
				ut32 n32;
				if (dbg->iob.read_at (dbg->iob.io, sp, (ut8*)&n32, sizeof (ut32)) != sizeof (ut32)) {
					return 0;
				}
				// TODO: honor endianness of platform
				return (ut64)n32;
			}
		}
		const char *rn = r_anal_cc_argloc (dbg->anal, cc, num, 0, -1);
		if (rn) {
			return r_debug_reg_get (dbg, rn);
		}
	}
	r_strf_var (reg, 32, "A%d", num);
	return r_debug_reg_get (dbg, reg);
}

R_API bool r_debug_arg_set(RDebug *dbg, const char *cc, int num, ut64 val) {
	R_RETURN_VAL_IF_FAIL (dbg, false);
	if (!R_STR_ISEMPTY (cc)) {
		cc = r_anal_syscc_default (dbg->anal);
	}
	const char *rn = r_anal_cc_argloc (dbg->anal, cc, num, 0, -1);
	if (rn) {
		r_debug_reg_set (dbg, rn, val);
		return true;
	}
	char reg[32];
	snprintf (reg, sizeof (reg) - 1, "A%d", num);
	r_debug_reg_set (dbg, reg, val);
	return true;
}