@@ -444,23 +444,23 @@ func TestLoadTracepointSensor(t *testing.T) {
444444 }
445445
446446 var sensorMaps = []tus.SensorMap {
447+ // generic_tracepoint_output
448+ {Name : "tcpmon_map" , Progs : []uint {5 }},
449+ }
450+
451+ if config .EnableLargeProgs () {
447452 // all programs
448- {Name : "process_call_heap" , Progs : []uint {0 , 1 , 2 , 3 , 4 , 5 }},
453+ sensorMaps = append ( sensorMaps , tus. SensorMap {Name : "process_call_heap" , Progs : []uint {0 , 1 , 2 , 3 , 4 , 5 }})
449454
450455 // all but generic_tracepoint_output
451- {Name : "tp_calls" , Progs : []uint {0 , 1 , 2 , 3 , 4 }},
456+ sensorMaps = append ( sensorMaps , tus. SensorMap {Name : "tp_calls" , Progs : []uint {0 , 1 , 2 , 3 , 4 }})
452457
453458 // all but generic_tracepoint_event,generic_tracepoint_filter
454- {Name : "retprobe_map" , Progs : []uint {1 , 2 }},
455-
456- // generic_tracepoint_output
457- {Name : "tcpmon_map" , Progs : []uint {5 }},
459+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "retprobe_map" , Progs : []uint {1 , 2 }})
458460
459461 // all kprobe but generic_tracepoint_filter
460- {Name : "config_map" , Progs : []uint {0 , 2 }},
461- }
462+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "config_map" , Progs : []uint {0 , 2 }})
462463
463- if config .EnableLargeProgs () {
464464 // shared with base sensor
465465 sensorMaps = append (sensorMaps , tus.SensorMap {Name : "execve_map" , Progs : []uint {3 , 4 , 5 }})
466466
@@ -474,11 +474,25 @@ func TestLoadTracepointSensor(t *testing.T) {
474474 sensorMaps = append (sensorMaps , tus.SensorMap {Name : "tg_conf_map" , Progs : []uint {0 }})
475475 }
476476 } else {
477+ sensorProgs = append (sensorProgs , tus.SensorProg {Name : "generic_tracepoint_process_event_2" , Type : ebpf .TracePoint })
478+
479+ // all programs
480+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "process_call_heap" , Progs : []uint {0 , 1 , 2 , 3 , 4 , 5 , 6 }})
481+
482+ // all but generic_tracepoint_output
483+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "tp_calls" , Progs : []uint {0 , 1 , 2 , 3 , 4 , 6 }})
484+
485+ // all but generic_tracepoint_event,generic_tracepoint_filter
486+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "retprobe_map" , Progs : []uint {1 , 2 , 6 }})
487+
488+ // all kprobe but generic_tracepoint_filter
489+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "config_map" , Progs : []uint {0 , 2 , 6 }})
490+
477491 // shared with base sensor
478492 sensorMaps = append (sensorMaps , tus.SensorMap {Name : "execve_map" , Progs : []uint {3 }})
479493
480494 // only generic_tracepoint_event*
481- sensorMaps = append (sensorMaps , tus.SensorMap {Name : "buffer_heap_map" , Progs : []uint {2 }})
495+ sensorMaps = append (sensorMaps , tus.SensorMap {Name : "buffer_heap_map" , Progs : []uint {2 , 6 }})
482496
483497 sensorMaps = append (sensorMaps , tus.SensorMap {Name : "tg_conf_map" , Progs : []uint {0 }})
484498 }
0 commit comments