From 409e5199f68ac4aafd4b1e9aeae0bb79e464b94a Mon Sep 17 00:00:00 2001
From: Hadi Ravanbakhsh <ravanbakhsh@google.com>
Date: Thu, 20 Feb 2025 11:26:58 -0800
Subject: [PATCH] Ignore unexpected feature ids.

PiperOrigin-RevId: 729186301
---
 centipede/BUILD         | 1 +
 centipede/feature_set.h | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/centipede/BUILD b/centipede/BUILD
index 840735e6..4d62b68f 100644
--- a/centipede/BUILD
+++ b/centipede/BUILD
@@ -584,6 +584,7 @@ cc_library(
         ":control_flow",
         ":feature",
         ":util",
+        "@com_google_absl//absl/log",
         "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/strings",
         "@com_google_fuzztest//common:logging",
diff --git a/centipede/feature_set.h b/centipede/feature_set.h
index de28f3b3..81ba5878 100644
--- a/centipede/feature_set.h
+++ b/centipede/feature_set.h
@@ -22,6 +22,7 @@
 #include <ostream>
 #include <string>
 
+#include "absl/log/log.h"
 #include "./centipede/control_flow.h"
 #include "./centipede/feature.h"
 #include "./centipede/util.h"
@@ -108,6 +109,11 @@ class FeatureSet {
   // feature.
   bool ShouldDiscardFeature(feature_t feature) const {
     size_t domain_id = feature_domains::Domain::FeatureToDomainId(feature);
+    // TODO(b/385774476): Remove this check once the root cause is fixed.
+    if (domain_id >= feature_domains::kNumDomains) {
+      LOG(ERROR) << "Unexpected feature with id: " << feature;
+      return true;
+    }
     return should_discard_domain_.test(domain_id);
   }