chore(pkg/driver): cleanup eBPF probe symlink in Cleanup method.

Signed-off-by: Federico Di Pierro <[email protected]>
falcosecurity · Dec 11, 2023 · 287da4e · 287da4e
1 parent 2b56ef4
commit 287da4e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/driver/type/bpf.go b/pkg/driver/type/bpf.go
@@ -43,15 +43,17 @@ func (b *bpf) String() string {
 	return TypeBpf
 }
 
-func (b *bpf) Cleanup(_ *output.Printer, _ string) error {
+func (b *bpf) Cleanup(printer *output.Printer, driverName string) error {
+	symlinkPath := filepath.Join(homedir.Get(), ".falco", fmt.Sprintf("%s-bpf.o", driverName))
+	printer.Logger.Info("Removing eBPF probe symlink", printer.Logger.Args("path", symlinkPath))
+	_ = os.Remove(symlinkPath)
 	return nil
 }
 
 func (b *bpf) Load(printer *output.Printer, src, driverName string, fallback bool) error {
 	if !fallback {
 		symlinkPath := filepath.Join(homedir.Get(), ".falco", fmt.Sprintf("%s-bpf.o", driverName))
 		printer.Logger.Info("Symlinking eBPF probe", printer.Logger.Args("src", src, "dest", symlinkPath))
-		_ = os.Remove(symlinkPath)
 		err := os.Symlink(src, symlinkPath)
 		if err == nil {
 			printer.Logger.Info("eBPF probe symlinked")