aboutsummaryrefslogtreecommitdiff
path: root/gocryptfs-xray/paths_ctlsock.go
blob: 5a19d327cd03a0c3d6b9af7a2671ee3c48104e3a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package main

import (
	"bufio"
	"fmt"
	"os"

	"github.com/rfjakob/gocryptfs/v2/ctlsock"
)

func decryptPaths(socketPath string, sep0 bool) {
	var req ctlsock.RequestStruct
	transformPaths(socketPath, &req, &req.DecryptPath, sep0)
}

func encryptPaths(socketPath string, sep0 bool) {
	var req ctlsock.RequestStruct
	transformPaths(socketPath, &req, &req.EncryptPath, sep0)
}

func transformPaths(socketPath string, req *ctlsock.RequestStruct, in *string, sep0 bool) {
	errorCount := 0
	c, err := ctlsock.New(socketPath)
	if err != nil {
		fmt.Fprintf(os.Stderr, "fatal: %v\n", err)
		os.Exit(1)
	}
	line := 1
	var separator byte = '\n'
	if sep0 {
		separator = '\000'
	}
	r := bufio.NewReader(os.Stdin)
	for eof := false; !eof; line++ {
		val, err := r.ReadBytes(separator)
		if len(val) == 0 {
			break
		}
		if err != nil {
			// break the loop after we have processed the last val
			eof = true
		} else {
			// drop trailing separator
			val = val[:len(val)-1]
		}
		*in = string(val)
		resp, err := c.Query(req)
		if err != nil {
			fmt.Fprintf(os.Stderr, "error at input line %d %q: %v\n", line, *in, err)
			errorCount++
			continue
		}
		if resp.WarnText != "" {
			fmt.Fprintf(os.Stderr, "warning at input line %d %q: %v\n", line, *in, resp.WarnText)
		}
		fmt.Printf("%s%c", resp.Result, separator)
	}
	if errorCount == 0 {
		os.Exit(0)
	}
	os.Exit(1)
}