/
cmd_exec.go
120 lines (106 loc) · 2.75 KB
/
cmd_exec.go
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
* Copyright (c) 2021 Gilles Chehade <gilles@poolp.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
package main
import (
"flag"
"io"
"io/ioutil"
"log"
"os"
"os/exec"
"github.com/PlakarLabs/plakar/encryption"
"github.com/PlakarLabs/plakar/logger"
"github.com/PlakarLabs/plakar/storage"
)
func init() {
registerCommand("exec", cmd_exec)
}
func cmd_exec(ctx Plakar, repository *storage.Repository, args []string) int {
flags := flag.NewFlagSet("exec", flag.ExitOnError)
flags.Parse(args)
if flags.NArg() == 0 {
logger.Error("%s: at least one parameters is required", flags.Name())
return 1
}
snapshots, err := getSnapshots(repository, []string{flags.Args()[0]})
if err != nil {
log.Fatal(err)
}
if len(snapshots) != 1 {
return 0
}
snapshot := snapshots[0]
_, pathname := parseSnapshotID(flags.Args()[0])
hasher := encryption.GetHasher(repository.Configuration().Hashing)
hasher.Write([]byte(pathname))
pathnameChecksum := hasher.Sum(nil)
key := [32]byte{}
copy(key[:], pathnameChecksum)
object := snapshot.Index.LookupObjectForPathnameChecksum(key)
if object == nil {
return 0
}
file, err := ioutil.TempFile(os.TempDir(), "plakar")
if err != nil {
log.Fatal(err)
}
defer os.Remove(file.Name())
file.Chmod(0500)
errors := 0
for _, chunkChecksum := range object.Chunks {
data, err := snapshot.GetChunk(chunkChecksum)
if err != nil {
logger.Error("%s: could not obtain chunk '%s'", flags.Name(), chunkChecksum)
errors++
break
}
file.Write(data)
}
file.Close()
if errors != 0 {
return 1
}
cmd := exec.Command(file.Name(), flags.Args()[1:]...)
stdin, err := cmd.StdinPipe()
if err != nil {
log.Fatal(err)
}
go func() {
defer stdin.Close()
io.Copy(stdin, os.Stdin)
}()
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal(err)
}
go func() {
defer stdout.Close()
io.Copy(os.Stdout, stdout)
}()
stderr, err := cmd.StderrPipe()
if err != nil {
log.Fatal(err)
}
go func() {
defer stdout.Close()
io.Copy(os.Stderr, stderr)
}()
if cmd.Start() == nil {
cmd.Wait()
return cmd.ProcessState.ExitCode()
}
return 1
}