labeler: fix codeql issues

Signed-off-by: Tuomas Katila <tuomas.katila@intel.com>

cmd/internal/labeler/labeler.go

@@ -16,6 +16,7 @@ package labeler
 
 import (
 	"fmt"
+	"math"
 	"os"
 	"os/signal"
 	"path"
@@ -205,12 +206,18 @@ func (l *labeler) getNumaNode(gpuName string) int {
 		return -1
 	}
 
-	numa, err := strconv.ParseInt(strings.TrimSpace(string(data)), 10, 64)
+	numa, err := strconv.ParseInt(strings.TrimSpace(string(data)), 10, 32)
 	if err != nil {
 		klog.Warning("Can't convert numa_node: ", err)
 		return -1
 	}
 
+	if numa > math.MaxInt16 {
+		klog.Warning("Too large numa: ", numa)
+
+		return -1
+	}
+
 	return int(numa)
 }
 
@@ -305,7 +312,9 @@ func (l *labeler) createLabels() error {
 			numaMapping[numaNode] = numaList
 		}
 
-		l.labels.addNumericLabel(labelNamespace+"memory.max", int64(memoryAmount))
+		if memoryAmount < math.MaxInt64 {
+			l.labels.addNumericLabel(labelNamespace+"memory.max", int64(memoryAmount))
+		}
 	}
 
 	gpuCount := len(gpuNumList)

cmd/internal/labeler/labeler_test.go

@@ -520,6 +520,48 @@ func getTestCases() []testcase {
 				"gpu.intel.com/tiles":         "27",
 			},
 		},
+		{
+			sysfsdirs: []string{
+				"card1/device/drm/card1",
+				"card1/gt/gt0",
+			},
+			sysfsfiles: map[string][]byte{
+				"card1/device/vendor":    []byte("0x8086"),
+				"card1/lmem_total_bytes": []byte("8000"),
+				"card1/device/numa_node": []byte("2147483648"), // max int32 + 1
+			},
+			name:           "too large numa node",
+			memoryOverride: 16000000000,
+			expectedRetval: nil,
+			expectedLabels: labelMap{
+				"gpu.intel.com/millicores":  "1000",
+				"gpu.intel.com/memory.max":  "8000",
+				"gpu.intel.com/gpu-numbers": "1",
+				"gpu.intel.com/cards":       "card1",
+				"gpu.intel.com/tiles":       "1",
+			},
+		},
+		{
+			sysfsdirs: []string{
+				"card1/device/drm/card1",
+				"card1/gt/gt0",
+			},
+			sysfsfiles: map[string][]byte{
+				"card1/device/vendor":    []byte("0x8086"),
+				"card1/lmem_total_bytes": []byte("8000"),
+				"card1/device/numa_node": []byte("32768"), // max int16 + 1
+			},
+			name:           "too large numa node",
+			memoryOverride: 16000000000,
+			expectedRetval: nil,
+			expectedLabels: labelMap{
+				"gpu.intel.com/millicores":  "1000",
+				"gpu.intel.com/memory.max":  "8000",
+				"gpu.intel.com/gpu-numbers": "1",
+				"gpu.intel.com/cards":       "card1",
+				"gpu.intel.com/tiles":       "1",
+			},
+		},
 	}
 }