Fix test

Velocidex · Jul 26, 2024 · 24cc351 · 24cc351
1 parent 8fd6d33
commit 24cc351
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/bin/usn.go b/bin/usn.go
@@ -68,7 +68,11 @@ func doUSN() {
 	filename_filter, err := regexp.Compile(*usn_command_filename_filter)
 	kingpin.FatalIfError(err, "Filename filter")
 
-	for record := range parser.ParseUSN(context.Background(), ntfs_ctx, 0) {
+	usn_stream, err := parser.OpenUSNStream(ntfs_ctx)
+	kingpin.FatalIfError(err, "OpenUSNStream")
+
+	for record := range parser.ParseUSN(context.Background(),
+		ntfs_ctx, usn_stream, 0) {
 		mft_id := record.FileReferenceNumberID()
 		mft_seq := uint16(record.FileReferenceNumberSequence())
 
@@ -92,7 +96,8 @@ func doUSN() {
 			})
 	}
 
-	for record := range parser.ParseUSN(context.Background(), ntfs_ctx, 0) {
+	for record := range parser.ParseUSN(
+		context.Background(), ntfs_ctx, usn_stream, 0) {
 		filename := record.Filename()
 
 		if !filename_filter.MatchString(filename) {