1 files changed, 5 insertions, 4 deletions
diff --git a/cmd/btrfs-fsck/main.go b/cmd/btrfs-fsck/main.go
index 4a230c6..fd9eb1b 100644
--- a/cmd/btrfs-fsck/main.go
+++ b/cmd/btrfs-fsck/main.go
@@ -5,25 +5,26 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"os"
 )
 
 func main() {
-	if err := Main(os.Args[1:]...); err != nil {
+	if err := Main(context.Background(), os.Args[1:]...); err != nil {
 		fmt.Fprintf(os.Stderr, "%v: error: %v\n", os.Args[0], err)
 		os.Exit(1)
 	}
 }
 
-func Main(imgfilenames ...string) (err error) {
+func Main(ctx context.Context, imgfilenames ...string) (err error) {
 	maybeSetErr := func(_err error) {
 		if _err != nil && err == nil {
 			err = _err
 		}
 	}
 
-	fs, sb, err := pass0(imgfilenames...)
+	fs, sb, err := pass0(ctx, imgfilenames...)
 	if err != nil {
 		return err
 	}
@@ -31,7 +32,7 @@ func Main(imgfilenames ...string) (err error) {
 		maybeSetErr(fs.Close())
 	}()
 
-	foundNodes, err := pass1(fs, sb)
+	foundNodes, err := pass1(ctx, fs, sb)
 	if err != nil {
 		return err
 	}