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
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package client
import (
"bytes"
"encoding/json"
"fmt"
"io"
)
// InternalSnapctlCmdNeedsStdin returns true if the given snapctl command
// needs data from stdin
func InternalSnapctlCmdNeedsStdin(name string) bool {
switch name {
case "fde-setup-result":
return true
default:
return false
}
}
// SnapCtlOptions holds the various options with which snapctl is invoked.
type SnapCtlOptions struct {
// ContextID is a string used to determine the context of this call (e.g.
// which context and handler should be used, etc.)
ContextID string `json:"context-id"`
// Args contains a list of parameters to use for this invocation.
Args []string `json:"args"`
}
// SnapCtlPostData is the data posted to the daemon /v2/snapctl endpoint
// TODO: this can be removed again once we no longer need to pass stdin data
// but instead use a real stdin stream
type SnapCtlPostData struct {
SnapCtlOptions
Stdin []byte `json:"stdin,omitempty"`
}
type snapctlOutput struct {
Stdout string `json:"stdout"`
Stderr string `json:"stderr"`
}
// protect against too much data via stdin
var stdinReadLimit = int64(4 * 1000 * 1000)
// RunSnapctl requests a snapctl run for the given options.
func (client *Client) RunSnapctl(options *SnapCtlOptions, stdin io.Reader) (stdout, stderr []byte, err error) {
// TODO: instead of reading all of stdin here we need to forward it to
// the daemon eventually
var stdinData []byte
if stdin != nil {
limitedStdin := &io.LimitedReader{R: stdin, N: stdinReadLimit + 1}
stdinData, err = io.ReadAll(limitedStdin)
if err != nil {
return nil, nil, fmt.Errorf("cannot read stdin: %v", err)
}
if limitedStdin.N <= 0 {
return nil, nil, fmt.Errorf("cannot read more than %v bytes of data from stdin", stdinReadLimit)
}
}
b, err := json.Marshal(SnapCtlPostData{
SnapCtlOptions: *options,
Stdin: stdinData,
})
if err != nil {
return nil, nil, fmt.Errorf("cannot marshal options: %s", err)
}
var output snapctlOutput
_, err = client.doSync("POST", "/v2/snapctl", nil, nil, bytes.NewReader(b), &output)
if err != nil {
return nil, nil, err
}
return []byte(output.Stdout), []byte(output.Stderr), nil
}
|
