зеркало из https://github.com/microsoft/docker.git
Trim quotes from TLS flags.
Signed-off-by: Daniel Nephin <dnephin@docker.com>
This commit is contained in:
Родитель
e4c1f07729
Коммит
abe32de6b4
|
@ -59,11 +59,15 @@ func (commonOpts *CommonOptions) InstallFlags(flags *pflag.FlagSet) {
|
||||||
|
|
||||||
// TODO use flag flags.String("identity"}, "i", "", "Path to libtrust key file")
|
// TODO use flag flags.String("identity"}, "i", "", "Path to libtrust key file")
|
||||||
|
|
||||||
commonOpts.TLSOptions = &tlsconfig.Options{}
|
commonOpts.TLSOptions = &tlsconfig.Options{
|
||||||
|
CAFile: filepath.Join(dockerCertPath, DefaultCaFile),
|
||||||
|
CertFile: filepath.Join(dockerCertPath, DefaultCertFile),
|
||||||
|
KeyFile: filepath.Join(dockerCertPath, DefaultKeyFile),
|
||||||
|
}
|
||||||
tlsOptions := commonOpts.TLSOptions
|
tlsOptions := commonOpts.TLSOptions
|
||||||
flags.StringVar(&tlsOptions.CAFile, "tlscacert", filepath.Join(dockerCertPath, DefaultCaFile), "Trust certs signed only by this CA")
|
flags.Var(opts.NewQuotedString(&tlsOptions.CAFile), "tlscacert", "Trust certs signed only by this CA")
|
||||||
flags.StringVar(&tlsOptions.CertFile, "tlscert", filepath.Join(dockerCertPath, DefaultCertFile), "Path to TLS certificate file")
|
flags.Var(opts.NewQuotedString(&tlsOptions.CertFile), "tlscert", "Path to TLS certificate file")
|
||||||
flags.StringVar(&tlsOptions.KeyFile, "tlskey", filepath.Join(dockerCertPath, DefaultKeyFile), "Path to TLS key file")
|
flags.Var(opts.NewQuotedString(&tlsOptions.KeyFile), "tlskey", "Path to TLS key file")
|
||||||
|
|
||||||
hostOpt := opts.NewNamedListOptsRef("hosts", &commonOpts.Hosts, opts.ValidateHost)
|
hostOpt := opts.NewNamedListOptsRef("hosts", &commonOpts.Hosts, opts.ValidateHost)
|
||||||
flags.VarP(hostOpt, "host", "H", "Daemon socket(s) to connect to")
|
flags.VarP(hostOpt, "host", "H", "Daemon socket(s) to connect to")
|
||||||
|
|
|
@ -0,0 +1,42 @@
|
||||||
|
package flags
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
cliconfig "github.com/docker/docker/cli/config"
|
||||||
|
"github.com/docker/docker/pkg/testutil/assert"
|
||||||
|
"github.com/spf13/pflag"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCommonOptionsInstallFlags(t *testing.T) {
|
||||||
|
flags := pflag.NewFlagSet("testing", pflag.ContinueOnError)
|
||||||
|
opts := NewCommonOptions()
|
||||||
|
opts.InstallFlags(flags)
|
||||||
|
|
||||||
|
err := flags.Parse([]string{
|
||||||
|
"--tlscacert=\"/foo/cafile\"",
|
||||||
|
"--tlscert=\"/foo/cert\"",
|
||||||
|
"--tlskey=\"/foo/key\"",
|
||||||
|
})
|
||||||
|
assert.NilError(t, err)
|
||||||
|
assert.Equal(t, opts.TLSOptions.CAFile, "/foo/cafile")
|
||||||
|
assert.Equal(t, opts.TLSOptions.CertFile, "/foo/cert")
|
||||||
|
assert.Equal(t, opts.TLSOptions.KeyFile, "/foo/key")
|
||||||
|
}
|
||||||
|
|
||||||
|
func defaultPath(filename string) string {
|
||||||
|
return filepath.Join(cliconfig.Dir(), filename)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCommonOptionsInstallFlagsWithDefaults(t *testing.T) {
|
||||||
|
flags := pflag.NewFlagSet("testing", pflag.ContinueOnError)
|
||||||
|
opts := NewCommonOptions()
|
||||||
|
opts.InstallFlags(flags)
|
||||||
|
|
||||||
|
err := flags.Parse([]string{})
|
||||||
|
assert.NilError(t, err)
|
||||||
|
assert.Equal(t, opts.TLSOptions.CAFile, defaultPath("ca.pem"))
|
||||||
|
assert.Equal(t, opts.TLSOptions.CertFile, defaultPath("cert.pem"))
|
||||||
|
assert.Equal(t, opts.TLSOptions.KeyFile, defaultPath("key.pem"))
|
||||||
|
}
|
|
@ -2,11 +2,13 @@ package opts
|
||||||
|
|
||||||
// QuotedString is a string that may have extra quotes around the value. The
|
// QuotedString is a string that may have extra quotes around the value. The
|
||||||
// quotes are stripped from the value.
|
// quotes are stripped from the value.
|
||||||
type QuotedString string
|
type QuotedString struct {
|
||||||
|
value *string
|
||||||
|
}
|
||||||
|
|
||||||
// Set sets a new value
|
// Set sets a new value
|
||||||
func (s *QuotedString) Set(val string) error {
|
func (s *QuotedString) Set(val string) error {
|
||||||
*s = QuotedString(trimQuotes(val))
|
*s.value = trimQuotes(val)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -16,7 +18,7 @@ func (s *QuotedString) Type() string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *QuotedString) String() string {
|
func (s *QuotedString) String() string {
|
||||||
return string(*s)
|
return string(*s.value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func trimQuotes(value string) string {
|
func trimQuotes(value string) string {
|
||||||
|
@ -28,3 +30,8 @@ func trimQuotes(value string) string {
|
||||||
}
|
}
|
||||||
return value
|
return value
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// NewQuotedString returns a new quoted string option
|
||||||
|
func NewQuotedString(value *string) *QuotedString {
|
||||||
|
return &QuotedString{value: value}
|
||||||
|
}
|
||||||
|
|
|
@ -6,19 +6,23 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestQuotedStringSetWithQuotes(t *testing.T) {
|
func TestQuotedStringSetWithQuotes(t *testing.T) {
|
||||||
qs := QuotedString("")
|
value := ""
|
||||||
|
qs := NewQuotedString(&value)
|
||||||
assert.NilError(t, qs.Set("\"something\""))
|
assert.NilError(t, qs.Set("\"something\""))
|
||||||
assert.Equal(t, qs.String(), "something")
|
assert.Equal(t, qs.String(), "something")
|
||||||
|
assert.Equal(t, value, "something")
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestQuotedStringSetWithMismatchedQuotes(t *testing.T) {
|
func TestQuotedStringSetWithMismatchedQuotes(t *testing.T) {
|
||||||
qs := QuotedString("")
|
value := ""
|
||||||
|
qs := NewQuotedString(&value)
|
||||||
assert.NilError(t, qs.Set("\"something'"))
|
assert.NilError(t, qs.Set("\"something'"))
|
||||||
assert.Equal(t, qs.String(), "\"something'")
|
assert.Equal(t, qs.String(), "\"something'")
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestQuotedStringSetWithNoQuotes(t *testing.T) {
|
func TestQuotedStringSetWithNoQuotes(t *testing.T) {
|
||||||
qs := QuotedString("")
|
value := ""
|
||||||
|
qs := NewQuotedString(&value)
|
||||||
assert.NilError(t, qs.Set("something"))
|
assert.NilError(t, qs.Set("something"))
|
||||||
assert.Equal(t, qs.String(), "something")
|
assert.Equal(t, qs.String(), "something")
|
||||||
}
|
}
|
||||||
|
|
Загрузка…
Ссылка в новой задаче