-
Notifications
You must be signed in to change notification settings - Fork 1
/
secret_entry.go
86 lines (71 loc) · 1.97 KB
/
secret_entry.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
// Copyright 2015 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.
package client
import (
"fmt"
"github.com/keybase/client/go/libkb"
"github.com/keybase/client/go/pinentry"
keybase1 "github.com/keybase/client/go/protocol"
"io"
)
type SecretEntry struct {
libkb.Contextified
pinentry *pinentry.Pinentry
terminal *Terminal
initRes *error
tty string
}
func NewSecretEntry(g *libkb.GlobalContext, t *Terminal, tty string) *SecretEntry {
return &SecretEntry{
Contextified: libkb.NewContextified(g),
terminal: t,
tty: tty,
}
}
func (se *SecretEntry) Init() (err error) {
se.G().Log.Debug("+ SecretEntry.Init()")
if se.initRes != nil {
se.G().Log.Debug("- SecretEntry.Init() -> cached %s", libkb.ErrToOk(*se.initRes))
return *se.initRes
}
if se.G().Env.GetNoPinentry() {
se.G().Log.Debug("| Pinentry skipped due to config")
} else {
pe := pinentry.New(se.G().Env.GetPinentry(), se.G().Log, se.tty)
if e2, fatalerr := pe.Init(); fatalerr != nil {
err = fatalerr
} else if e2 != nil {
se.G().Log.Debug("| Pinentry initialization failed: %s", e2)
} else {
se.pinentry = pe
se.G().Log.Debug("| Pinentry initialized")
}
}
if err == nil {
if se.pinentry == nil && se.terminal == nil {
err = fmt.Errorf("No terminal and pinentry init; cannot input secrets")
}
}
se.initRes = &err
se.G().Log.Debug("- SecretEntry.Init() -> %s", libkb.ErrToOk(err))
return err
}
func (se *SecretEntry) Get(arg keybase1.SecretEntryArg, termArg *keybase1.SecretEntryArg, w io.Writer) (res *keybase1.SecretEntryRes, err error) {
if err = se.Init(); err != nil {
return
}
if pe := se.pinentry; pe != nil {
if len(arg.Reason) > 0 {
fmt.Fprintf(w, "Collecting your passphrase for %s.\n", arg.Reason)
}
res, err = pe.Get(arg)
} else if se.terminal == nil {
err = NoTerminalError{}
} else {
if termArg == nil {
termArg = &arg
}
res, err = se.terminal.GetSecret(termArg)
}
return
}