added sanity check to prevent length of a multi-dimensional array. added more tests with extended paths

Author: fernantho

encoding/ssz/query/generalized_index.go

@@ -61,6 +61,9 @@ func GetGeneralizedIndexFromPath(info *SszInfo, path []PathElement) (uint64, err
 
 		// Check if a path element is a length field
 		if element.Length {
+			if element.Index != nil {
+				return 0, fmt.Errorf("len() is not supported for indexed elements (multi-dimensional arrays)")
+			}
 			// Length field is only valid for List and Bitlist types
 			if fieldSsz.sszType != List && fieldSsz.sszType != Bitlist {
 				return 0, fmt.Errorf("len() is only supported for List and Bitlist types, got %s", fieldSsz.sszType)
@@ -74,6 +77,7 @@ func GetGeneralizedIndexFromPath(info *SszInfo, path []PathElement) (uint64, err
 		if element.Index == nil {
 			continue
 		}
+
 		switch fieldSsz.sszType {
 		case List:
 			li, err := fieldSsz.ListInfo()

encoding/ssz/query/generalized_index_test.go

@@ -190,6 +190,43 @@ func TestGetIndicesFromPath_VariableTestContainer(t *testing.T) {
 			expectedIndex: 5186,
 			expectError:   false,
 		},
+		{
+			name:          "variable_container_list",
+			path:          "variable_container_list",
+			expectedIndex: 21,
+			expectError:   false,
+		},
+		{
+			name:          "len(variable_container_list)",
+			path:          "len(variable_container_list)",
+			expectedIndex: 43,
+			expectError:   false,
+		},
+		{
+			name:          "variable_container_list[0]",
+			path:          "variable_container_list[0]",
+			expectedIndex: 672,
+			expectError:   false,
+		},
+		{
+			name:          "variable_container_list[0].inner_1",
+			path:          "variable_container_list[0].inner_1",
+			expectedIndex: 1344,
+			expectError:   false,
+		},
+		{
+			name:          "variable_container_list[0].inner_1.field_list_uint64[1]",
+			path:          "variable_container_list[0].inner_1.field_list_uint64[1]",
+			expectedIndex: 344128,
+			expectError:   false,
+		},
+		{
+			name:          "variable_container_list[0].inner_1.len(nested_list_field[3])",
+			path:          "variable_container_list[0].inner_1.len(nested_list_field[3])",
+			expectedIndex: 1376775,
+			expectError:   true,
+			errorMessage:  "len() is not supported for indexed elements (multi-dimensional arrays)",
+		},
 	}
 
 	for _, tc := range testCases {