feat: add AttachStructOps() (#476)

Add AttachStructOps() and selftest.
aquasecurity · Mar 6, 2025 · d3930cf · d3930cf
1 parent 12c2216
commit d3930cf
Show file tree

Hide file tree

Showing 11 changed files with 860 additions and 0 deletions.
diff --git a/link.go b/link.go
@@ -33,6 +33,7 @@ const (
 	CgroupLegacy
 	Netns
 	Iter
+	StructOps
 )
 
 //
@@ -47,6 +48,7 @@ type bpfLinkLegacy struct {
 type BPFLink struct {
 	link      *C.struct_bpf_link
 	prog      *BPFProg
+	m         *BPFMap
 	linkType  LinkType
 	eventName string
 	legacy    *bpfLinkLegacy // if set, this is a fake BPFLink

diff --git a/map.go b/map.go
@@ -73,6 +73,19 @@ func (m *BPFMap) ReuseFD(fd int) error {
 	return nil
 }
 
+func (m *BPFMap) AttachStructOps() (*BPFLink, error) {
+	linkC, errno := C.bpf_map__attach_struct_ops(m.bpfMap)
+	if linkC == nil {
+		return nil, fmt.Errorf("failed to attach struct_ops: %w", errno)
+	}
+	return &BPFLink{
+		link:      linkC,
+		m:         m,
+		linkType:  StructOps,
+		eventName: fmt.Sprintf("structOps-%s", m.Name()),
+	}, nil
+}
+
 func (m *BPFMap) Name() string {
 	return C.GoString(C.bpf_map__name(m.bpfMap))
 }

diff --git a/selftest/common/.gitignore b/selftest/common/.gitignore
@@ -0,0 +1 @@
+scx