From c2feee4ed1b35cc590e2beaa595d710b09e1427c Mon Sep 17 00:00:00 2001
From: Akira Moroo <retrage01@gmail.com>
Date: Fri, 12 Nov 2021 06:01:19 +0000
Subject: Add platform check for `-A` CoreSight mode

REF:
https://github.com/AFLplusplus/AFLplusplus/pull/1156#discussion_r747454306

Signed-off-by: Akira Moroo <retrage01@gmail.com>
---
 src/afl-analyze.c | 6 ++++++
 src/afl-fuzz.c    | 7 +++++++
 src/afl-showmap.c | 6 ++++++
 src/afl-tmin.c    | 6 ++++++
 4 files changed, 25 insertions(+)

diff --git a/src/afl-analyze.c b/src/afl-analyze.c
index 4872c60d..6e1a9e7b 100644
--- a/src/afl-analyze.c
+++ b/src/afl-analyze.c
@@ -997,6 +997,12 @@ int main(int argc, char **argv_orig, char **envp) {
 
         if (cs_mode) { FATAL("Multiple -A options not supported"); }
 
+        if (!(__aarch64__ && __linux__)) {
+
+          FATAL("-A option is not supported on this platform");
+
+        }
+
         cs_mode = 1;
         fsrv.cs_mode = cs_mode;
         break;
diff --git a/src/afl-fuzz.c b/src/afl-fuzz.c
index 6538e0a0..e6d9508b 100644
--- a/src/afl-fuzz.c
+++ b/src/afl-fuzz.c
@@ -830,6 +830,13 @@ int main(int argc, char **argv_orig, char **envp) {
       case 'A':                                           /* CoreSight mode */
 
         if (afl->fsrv.cs_mode) { FATAL("Multiple -A options not supported"); }
+
+        if (!(__aarch64__ && __linux__)) {
+
+          FATAL("-A option is not supported on this platform");
+
+        }
+
         afl->fsrv.cs_mode = 1;
 
         break;
diff --git a/src/afl-showmap.c b/src/afl-showmap.c
index 57737562..0e7d9df6 100644
--- a/src/afl-showmap.c
+++ b/src/afl-showmap.c
@@ -1069,6 +1069,12 @@ int main(int argc, char **argv_orig, char **envp) {
 
         if (fsrv->cs_mode) { FATAL("Multiple -P options not supported"); }
 
+        if (!(__aarch64__ && __linux__)) {
+
+          FATAL("-P option is not supported on this platform");
+
+        }
+
         fsrv->cs_mode = true;
         break;
 
diff --git a/src/afl-tmin.c b/src/afl-tmin.c
index 42883404..110beed5 100644
--- a/src/afl-tmin.c
+++ b/src/afl-tmin.c
@@ -1040,6 +1040,12 @@ int main(int argc, char **argv_orig, char **envp) {
 
         if (fsrv->cs_mode) { FATAL("Multiple -A options not supported"); }
 
+        if (!(__aarch64__ && __linux__)) {
+
+          FATAL("-A option is not supported on this platform");
+
+        }
+
         fsrv->cs_mode = 1;
         break;
 
-- 
cgit 1.4.1