From 8d2ff19bc3fde1bc997e5ec6c521ee6670dd613f Mon Sep 17 00:00:00 2001
From: Xia Lifang <lifang_xia@linux.alibaba.com>
Date: Wed, 12 Oct 2022 19:36:02 +0800
Subject: [PATCH] Add riscv binary recognition in gdb script

Signed-off-by: Xia Lifang <lifang_xia@linux.alibaba.com>
Signed-off-by: Mao Han <han_mao@linux.alibaba.com>
Change-Id: I61793c3c1e32828ac4196a20f85b4fe2832c4992
---
 python-packages/gdbrunner/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python-packages/gdbrunner/__init__.py b/python-packages/gdbrunner/__init__.py
index 2eabf99d2..35d09b8c6 100644
--- a/python-packages/gdbrunner/__init__.py
+++ b/python-packages/gdbrunner/__init__.py
@@ -337,6 +337,9 @@ def get_binary_arch(binary_file):
     elif e_machine == 0x3E:
         assert ei_class == 2
         return "x86_64"
+    elif e_machine == 0xF3:
+        assert ei_class == 2
+        return "riscv64"
     else:
         raise RuntimeError("unknown architecture: 0x{:x}".format(e_machine))