[Filebeat] [Netflow] fix field name conversion to snake case (#10950) (…

…#11019)

Original field name conversion was buggy.

(cherry picked from commit 85e470e)

x-pack/filebeat/input/netflow/case.go

@@ -14,7 +14,12 @@ import (
 )
 
 var fieldNameConverter = caseConverter{
-	conversion: make(map[string]string),
+	conversion: map[string]string{
+		// Special handled fields
+
+		// VRFname should be VRFName
+		"VRFname": "vrf_name",
+	},
 }
 
 type caseConverter struct {
@@ -50,24 +55,24 @@ func (c *caseConverter) ToSnakeCase(orig record.Map) common.MapStr {
 // format. This function is tailored to some specifics of NetFlow field names.
 // Don't reuse it.
 func CamelCaseToSnakeCase(in string) string {
-	// Lowercase those few fields that are already snake-cased
+	// skip those few fields that are already snake-cased
 	if strings.ContainsRune(in, '_') {
 		return strings.ToLower(in)
 	}
 
 	out := make([]rune, 0, len(in)+4)
 	runes := []rune(in)
-	upperStrike := 1
-	for pos, r := range runes {
+	upperCount := 1
+	for _, r := range runes {
 		lr := unicode.ToLower(r)
 		isUpper := lr != r
 		if isUpper {
-			if upperStrike == 0 {
+			if upperCount == 0 {
 				out = append(out, '_')
 			}
-			upperStrike++
+			upperCount++
 		} else {
-			if upperStrike > 2 {
+			if upperCount > 2 {
 				// Some magic here:
 				// NetFlow usually lowercases all but the first letter of an
 				// acronym (Icmp) Except when it is 2 characters long: (IP).
@@ -77,9 +82,10 @@ func CamelCaseToSnakeCase(in string) string {
 				// postNATSourceIPv4Address     : post_nat_source_ipv4_address
 				// selectorIDTotalFlowsObserved : selector_id_total_flows_...
 				out = append(out, '_')
-				out[pos], out[pos-1] = out[pos-1], out[pos]
+				n := len(out) - 1
+				out[n], out[n-1] = out[n-1], out[n]
 			}
-			upperStrike = 0
+			upperCount = 0
 		}
 		out = append(out, lr)
 	}

x-pack/filebeat/input/netflow/case_test.go

@@ -0,0 +1,36 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License;
+// you may not use this file except in compliance with the Elastic License.
+
+package netflow
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCamelCaseToSnakeCase(t *testing.T) {
+	for _, testCase := range [][2]string{
+		{"aBCDe", "a_bc_de"},
+		{"postNATSourceIPv4Address", "post_nat_source_ipv4_address"},
+		{"selectorIDTotalFlowsObserved", "selector_id_total_flows_observed"},
+		{"engineId", "engine_id"},
+		{"samplerRandomInterval", "sampler_random_interval"},
+		{"dot1qVlanId", "dot1q_vlan_id"},
+		{"messageMD5Checksum", "message_md5_checksum"},
+		{"hashIPPayloadSize", "hash_ip_payload_size"},
+		{"upperCILimit", "upper_ci_limit"},
+		{"virtualStationUUID", "virtual_station_uuid"},
+		{"selectorIDTotalFlowsObserved", "selector_id_total_flows_observed"},
+		{"postMCastLayer2OctetDeltaCount", "post_mcast_layer2_octet_delta_count"},
+		{"IPSecSPI", "ip_sec_spi"},
+		{"VRFname", "vrf_name"},
+	} {
+		s, found := fieldNameConverter.conversion[testCase[0]]
+		if !found {
+			s = CamelCaseToSnakeCase(testCase[0])
+		}
+		assert.Equal(t, testCase[1], s)
+	}
+}