Add handling of structure returns and parameters in aarch64 calling convention

1 files changed, 39 insertions, 0 deletions

diff --git a/arch/arm64/arch_arm64.cpp b/arch/arm64/arch_arm64.cpp
index fd4942fd..20b00e8a 100644
--- a/arch/arm64/arch_arm64.cpp
+++ b/arch/arm64/arch_arm64.cpp
@@ -2646,9 +2646,42 @@ class Arm64CallingConvention : public CallingConvention
 
 
 	virtual uint32_t GetIntegerReturnValueRegister() override { return REG_X0; }
+	virtual uint32_t GetHighIntegerReturnValueRegister() override { return REG_X1; }
 
 
 	virtual uint32_t GetFloatReturnValueRegister() override { return REG_V0; }
+
+
+	bool IsReturnTypeRegisterCompatible(BinaryView*, Type* type) override
+	{
+		if (!type)
+			return false;
+		if (type->IsFloat())
+			return true;
+		return type->GetWidth() <= 16;
+	}
+
+
+	Variable GetIndirectReturnValueLocation() override
+	{
+		return Variable::Register(REG_X8);
+	}
+
+
+	bool IsArgumentTypeRegisterCompatible(BinaryView*, Type* type) override
+	{
+		if (!type)
+			return false;
+		if (type->IsFloat())
+			return true;
+		return type->GetWidth() <= 16;
+	}
+
+
+	bool IsNonRegisterArgumentIndirect(BinaryView*, Type*) override
+	{
+		return true;
+	}
 };
 
 
@@ -2664,6 +2697,12 @@ public:
 	{
 		return false;
 	}
+
+
+	virtual bool AreStackArgumentsNaturallyAligned() override
+	{
+		return true;
+	}
 };