From b12240c9210d8e6ebb5aa9125b64736f2fd07c73 Mon Sep 17 00:00:00 2001
From: Joseph Ligier <joseph.ligier@accenture.com>
Date: Thu, 13 Nov 2025 17:12:34 +0100
Subject: [PATCH] Add Tracepoint in main.go

---
 Taskfile.yml                |  9 ++++++-
 {{project-name}}-go/main.go | 53 ++++++++++++++++++++++++++++++-------
 2 files changed, 51 insertions(+), 11 deletions(-)

diff --git a/Taskfile.yml b/Taskfile.yml
index 96abc85..a878bc9 100644
--- a/Taskfile.yml
+++ b/Taskfile.yml
@@ -9,12 +9,19 @@ tasks:
   install-ebpf:
     desc: "Install the eBPF program"
     cmds:
-      - go run main.go {{ "{{ .IFACE }}" }}
+      - go run main.go {{ "{{ .HOOK }}" }}
     dir: {{project-name}}-go
     deps:
       - build-ebpf
 
+  fmt:
+    desc: "format go code"
+    cmds:
+      - go fmt main.go
+    dir: {{project-name}}-go
+
   default:
     desc: "Build and install the eBPF program"
     cmds:
+      - task: fmt
       - task: install-ebpf
diff --git a/{{project-name}}-go/main.go b/{{project-name}}-go/main.go
index 04cb90b..cf8c6fa 100644
--- a/{{project-name}}-go/main.go
+++ b/{{project-name}}-go/main.go
@@ -5,9 +5,16 @@ import (
 	"context"
 	"fmt"
 	"log"
+
+{% if program_type == "xdp" %}
 	"net"
+{% endif %}
+
 	"os"
 	"os/signal"
+{% if program_type == "tracepoint" %}
+	"strings"
+{% endif %}
 	"syscall"
 
 	"github.com/cilium/ebpf"
@@ -19,6 +26,14 @@ import (
 
 const progName = "{{crate_name}}"
 
+{%- case program_type -%}
+{%- when "tracepoint" %}
+const defaultCategory = "{{tracepoint_category}}"
+const defaultName = "{{tracepoint_name}}"
+{%- when "xdp" %}
+const defaultIface = "{{default_iface}}"
+{%- endcase %}
+
 //go:embed .ebpf/{{project-name}}
 var ebpfBytes []byte
 
@@ -45,13 +60,6 @@ func extractPrintableStrings(raw []byte) []string {
 }
 
 func main() {
-	defaultIface := "{{default_iface}}"
-	ifaceName := defaultIface
-
-	if len(os.Args) > 1 {
-		ifaceName = os.Args[1]
-	}
-
 	spec, err := ebpf.LoadCollectionSpecFromReader(bytes.NewReader(ebpfBytes))
 	if err != nil {
 		log.Fatalf("LoadCollectionSpec failed: %v", err)
@@ -68,11 +76,35 @@ func main() {
 		log.Fatalf("Program %s not found", progName)
 	}
 
-	iface, err := net.InterfaceByName(ifaceName)
+    {%- case program_type -%}
+        {%- when "tracepoint" %}
+        category := defaultCategory
+        name := defaultName
+	attachment := fmt.Sprintf("%s:%s", category, name)
+
+        if len(os.Args) > 1 {
+            attachment = os.Args[1]
+            parts := strings.SplitN(attachment, ":", 2)
+            if len(parts) != 2 {
+                log.Fatalf("invalid attachment format: %s, expected category:name", attachment)
+            }
+            category, name = parts[0], parts[1]
+        }
+
+	tp, err := link.Tracepoint(category, name, prog, nil)
+	if err != nil {
+		log.Fatalf("opening tracepoint: %s", err)
+	}
+	defer tp.Close()
+        {%- when "xdp" %}
+	attachment := defaultIface
+	if len(os.Args) > 1 {
+		attachment = os.Args[1]
+	}
+	iface, err := net.InterfaceByName(attachment)
 	if err != nil {
 		log.Fatalf("Interface not found: %v", err)
 	}
-
 	l, err := link.AttachXDP(link.XDPOptions{
 		Program:   prog,
 		Interface: iface.Index,
@@ -81,7 +113,8 @@ func main() {
 		log.Fatalf("AttachXDP failed: %v", err)
 	}
 	defer l.Close()
-	fmt.Printf("✅ Program '%s' attached to %s\n", progName, ifaceName)
+    {%- endcase %}
+	fmt.Printf("✅ Program '%s' attached to %s\n", progName, attachment)
 
 	logMap, ok := coll.Maps["AYA_LOGS"]
 	if !ok {