Add type guard to prevent crash

slackhq · Nov 13, 2024 · 16275ce · 16275ce
1 parent 0edc8bd
commit 16275ce
Showing 1 changed file with 15 additions and 13 deletions.
diff --git a/src/analyzer/expr/fetch/array_fetch_analyzer.rs b/src/analyzer/expr/fetch/array_fetch_analyzer.rs
@@ -434,31 +434,33 @@ pub(crate) fn get_array_access_type_given_offset(
             } => match *name {
                 StrId::KEYED_CONTAINER | StrId::ANY_ARRAY => {
                     if let Some(type_params) = type_params {
-                        if let Some(existing_type) = stmt_type {
-                            stmt_type = Some(add_union_type(
-                                existing_type,
-                                type_params.get(1).unwrap(),
-                                codebase,
-                                false,
-                            ));
-                        } else {
-                            stmt_type = Some(type_params.get(1).unwrap().clone());
+                        if type_params.len() > 1 {
+                            if let Some(existing_type) = stmt_type {
+                                stmt_type = Some(add_union_type(
+                                    existing_type,
+                                    &type_params[1],
+                                    codebase,
+                                    false,
+                                ));
+                            } else {
+                                stmt_type = Some(type_params[1].clone());
+                            }
+
+                            has_valid_expected_offset = true;
                         }
-
-                        has_valid_expected_offset = true;
                     }
                 }
                 StrId::CONTAINER => {
                     if let Some(type_params) = type_params {
                         if let Some(existing_type) = stmt_type {
                             stmt_type = Some(add_union_type(
                                 existing_type,
-                                type_params.first().unwrap(),
+                                &type_params[0],
                                 codebase,
                                 false,
                             ));
                         } else {
-                            stmt_type = Some(type_params.first().unwrap().clone());
+                            stmt_type = Some(type_params[0].clone());
                         }
 
                         has_valid_expected_offset = true;