From 42409ad6003de6cc2f5a2fa02e9e6937efc9d055 Mon Sep 17 00:00:00 2001
From: Francisco Javier Honduvilla Coto <javierhonduco@gmail.com>
Date: Thu, 28 Apr 2022 11:34:00 +0100
Subject: [PATCH] Add strict mode API

See https://github.com/libbpf/libbpf/wiki/Libbpf-1.0-migration-guide for context.
Related to https://github.com/aquasecurity/libbpfgo/issues/159
---
 libbpfgo.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/libbpfgo.go b/libbpfgo.go
index e5117390..0b309ecc 100644
--- a/libbpfgo.go
+++ b/libbpfgo.go
@@ -328,6 +328,24 @@ func NewModuleFromFile(bpfObjPath string) (*Module, error) {
 	})
 }
 
+func SetStrictMode(mode string) error {
+	var chosenMode uint32
+	switch mode {
+	case "all":
+		chosenMode = C.LIBBPF_STRICT_ALL
+	case "none":
+		chosenMode = C.LIBBPF_STRICT_NONE
+	case "clean_ptrs":
+		chosenMode = C.LIBBPF_STRICT_CLEAN_PTRS
+	case "direct_errors":
+		chosenMode = C.LIBBPF_STRICT_DIRECT_ERRS
+	default:
+		return fmt.Errorf("%s is not a valid strict mode", mode)
+	}
+	C.libbpf_set_strict_mode(chosenMode)
+	return nil
+}
+
 func NewModuleFromFileArgs(args NewModuleArgs) (*Module, error) {
 	C.set_print_fn()
 	if err := bumpMemlockRlimit(); err != nil {