From 232f6875e185e81716acaef2d18522908835204c Mon Sep 17 00:00:00 2001
From: Andreea Popescu <andreea.popescu@reef.pl>
Date: Fri, 28 Jun 2024 14:55:33 +0100
Subject: [PATCH] add bounds check

---
 pallets/subtensor/src/uids.rs | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/pallets/subtensor/src/uids.rs b/pallets/subtensor/src/uids.rs
index 5f96c36c2..3bad000af 100644
--- a/pallets/subtensor/src/uids.rs
+++ b/pallets/subtensor/src/uids.rs
@@ -10,12 +10,23 @@ impl<T: Config> Pallet<T> {
     }
 
     /// Resets the trust, emission, consensus, incentive, dividends of the neuron to default
+    fn clear_element_at<N>(position: u16) -> impl Fn(&mut Vec<N>)
+    where
+        N: From<u8>,
+    {
+        move |vec: &mut Vec<N>| {
+            if vec.len() > position as usize {
+                vec[position as usize] = N::from(0);
+            };
+        }
+    }
+
     pub fn clear_neuron(netuid: u16, neuron_uid: u16) {
-        Emission::<T>::mutate(netuid, |v| v[neuron_uid as usize] = 0);
-        Trust::<T>::mutate(netuid, |v| v[neuron_uid as usize] = 0);
-        Consensus::<T>::mutate(netuid, |v| v[neuron_uid as usize] = 0);
-        Incentive::<T>::mutate(netuid, |v| v[neuron_uid as usize] = 0);
-        Dividends::<T>::mutate(netuid, |v| v[neuron_uid as usize] = 0);
+        Emission::<T>::mutate(netuid, Self::clear_element_at(neuron_uid));
+        Trust::<T>::mutate(netuid, Self::clear_element_at(neuron_uid));
+        Consensus::<T>::mutate(netuid, Self::clear_element_at(neuron_uid));
+        Incentive::<T>::mutate(netuid, Self::clear_element_at(neuron_uid));
+        Dividends::<T>::mutate(netuid, Self::clear_element_at(neuron_uid));
     }
 
     /// Replace the neuron under this uid.