v0.0.4

Author: Snshadow

.gitignore

@@ -6,6 +6,7 @@
 *.exe~
 *.dll
 *.so
+*.syso
 *.dylib
 
 # Test binary, built with `go test -c`

README.md

@@ -102,7 +102,7 @@ import (
 )
 
 func main() {
-    targetPath := "C:\\test\\test.txt"
+	targetPath := "C:\\test\\test.txt"
 
 	eaList, err := ntfs_ea.QueryEaFile(targetPath)
 	if err != nil {

cmd/query_file_ea/query_file_ea.go

@@ -17,15 +17,17 @@ import (
 
 func main() {
 	var targetPath, queryName string
-	var dump, extract bool
+	var dump, extract, stdout bool
 
 	flag.StringVar(&targetPath, "target-path", "", "path of the file to query EA")
 	flag.StringVar(&queryName, "query-name", "", "names of EA to query, split by comma, if it is not given, query all EA from the file")
+
 	flag.BoolVar(&dump, "dump", false, "dump EA to console, this is enabled by default if no action is given")
 	flag.BoolVar(&extract, "extract", false, "extract EA to file(s) with according EaName")
+	flag.BoolVar(&stdout, "stdout", false, "extract EA into stdout")
 
 	flag.Usage = func() {
-		fmt.Fprintf(flag.CommandLine.Output(), "%s queries EA(Extended Attribute) from a file in NTFS(New Technology File System).\nUsage: %s -query-name [eaName1],[eaName2],... -extract [target path]\n or  %s -target-path [target path] -query-name [eaName1],[eaName2],... -dump -extract\nThis program is supposed to work only in Windows.\n\n", os.Args[0], os.Args[0], os.Args[0])
+		fmt.Fprintf(flag.CommandLine.Output(), "%s queries EA(Extended Attribute) from a file in NTFS(New Technology File System).\nUsage: %s -query-name [eaName1],[eaName2],... -extract [target path]\n or %s -target-path [target path] -query-name [eaName1],[eaName2],... -dump -extract\nWrite EA value to stdout(for piping output): %s -stdout -target-path [target path] -query-name [eaName] | (process output)\n\n", os.Args[0], os.Args[0], os.Args[0], os.Args[0])
 		flag.PrintDefaults()
 	}
 
@@ -66,11 +68,15 @@ func main() {
 		}
 
 		if extract {
-			err := os.WriteFile(ea.EaName, ea.EaValue, 0777)
-			if err != nil {
-				fmt.Fprintf(os.Stderr, "Failed to write EA for %s into file: %v\n", ea.EaName, err)
+			if stdout {
+				os.Stdout.Write(ea.EaValue)
 			} else {
-				fmt.Printf("Extracted EaValue in \"%s\"", ea.EaName)
+				err := os.WriteFile(ea.EaName, ea.EaValue, 0777)
+				if err != nil {
+					fmt.Fprintf(os.Stderr, "Failed to write EA for %s into file: %v\n", ea.EaName, err)
+				} else {
+					fmt.Printf("Extracted EaValue in \"%s\"", ea.EaName)
+				}
 			}
 		}
 	}

cmd/query_file_ea/query_file_ea.json

@@ -29,7 +29,7 @@
         "OriginalFilename": "",
         "PrivateBuild": "",
         "ProductName": "query_file_ea.exe",
-        "ProductVersion": "v0.0.3",
+        "ProductVersion": "v0.0.4",
         "SpecialBuild": ""
     },
     "VarFileInfo": {

cmd/write_file_ea/write_file_ea.go

@@ -15,17 +15,18 @@ import (
 
 func main() {
 	var srcPath, targetPath, eaName string
-	var needEa, removeEa bool
+	var needEa, removeEa, stdin bool
 
 	flag.StringVar(&targetPath, "target-path", "", "path of target file to write EA")
 	flag.StringVar(&srcPath, "source-path", "", "path of source file to be used as content for EA")
 	flag.StringVar(&eaName, "ea-name", "", "name of the EA")
 
 	flag.BoolVar(&needEa, "need-ea", false, "set flag if file needs to be interpreted with EA")
 	flag.BoolVar(&removeEa, "remove-ea", false, "remove the EA with the given name")
+	flag.BoolVar(&stdin, "stdin", false, "use stdin as content for EA")
 
 	flag.Usage = func() {
-		fmt.Fprintf(flag.CommandLine.Output(), "%s writes EA(Extended Attribute) info a file in NTFS(New Technology File System) with the content of a given source file, if the source file is empty the EA with EaName is removed if exists.\nUsage: %s [target path] [source path] [EA name]\n or\n %s -target-path [target path] -source-path [source path] -ea-name [EA name] -need-ea\nTo remove EA with specific name, use: %s -remove-ea [target path] [EA name]\nThis program is supposed to work only in Windows.\n\n", os.Args[0], os.Args[0], os.Args[0], os.Args[0])
+		fmt.Fprintf(flag.CommandLine.Output(), "%s writes EA(Extended Attribute) info a file in NTFS(New Technology File System) with the content of a given source file, if the source file is empty the EA with EaName is removed if exists.\nUsage: %s [target path] [source path] [EA name]\n or\n %s -target-path [target path] -source-path [source path] -ea-name [EA name] -need-ea\nWrite EA from stdin: echo \"[content for ea] | %s -stdin -target-path [target path] -ea-name [EA name]\nTo remove EA with specific name, use: %s -remove-ea [target path] [EA name]\n\n", os.Args[0], os.Args[0], os.Args[0], os.Args[0], os.Args[0])
 		flag.PrintDefaults()
 	}
 
@@ -40,7 +41,7 @@ func main() {
 		srcPath = flag.Arg(1)
 	}
 	if eaName == "" {
-		if removeEa {
+		if removeEa || stdin {
 			eaName = flag.Arg(1)
 			requiredArg = 2
 		} else {
@@ -53,7 +54,7 @@ func main() {
 		flags |= ntfs_ea.NeedEa
 	}
 
-	if !(targetPath != "" && srcPath != "" && eaName != "") && !(removeEa && targetPath != "" && eaName != "") && flag.NArg() < requiredArg {
+	if !(targetPath != "" && srcPath != "" && eaName != "") && !((removeEa || stdin) && targetPath != "" && eaName != "") && flag.NArg() < requiredArg {
 		flag.Usage()
 		os.Exit(1)
 	}
@@ -73,9 +74,40 @@ func main() {
 
 		return
 	}
+
+	if stdin {
+		eaBuf := make([]byte, 65536)
+
+		n, err := os.Stdin.Read(eaBuf)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Failed to read from stdin: %v", err)
+			os.Exit(2)
+		}
+
+		if n < len(eaBuf) {
+			eaBuf = eaBuf[:n]
+		}
+
+		eaToWrite := ntfs_ea.EaInfo{
+			Flags: flags,
+			EaName: eaName,
+			EaValue: eaBuf,
+		}
+
+		err = ntfs_ea.EaWriteFile(targetPath, eaToWrite)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Failed to write EA into file: %v\n", err)
+			os.Exit(2)
+		}
+
+		fmt.Printf("Written EA info file \"%s\" from stdin with ea name \"%s\"\n", targetPath, eaName)
+
+		return
+	}
+
 	err := ntfs_ea.WriteEaWithFile(targetPath, srcPath, eaName, flags)
 	if err != nil {
-		fmt.Fprintf(os.Stderr, "failed to write EA into file: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Failed to write EA into file: %v\n", err)
 		os.Exit(2)
 	}
 

cmd/write_file_ea/write_file_ea.json

@@ -29,7 +29,7 @@
         "OriginalFilename": "",
         "PrivateBuild": "",
         "ProductName": "write_file_ea.exe",
-        "ProductVersion": "v0.0.3",
+        "ProductVersion": "v0.0.4",
         "SpecialBuild": ""
     },
     "VarFileInfo": {