mirror of https://github.com/status-im/consul.git
115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
|
package sprawl
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
|
||
|
|
"github.com/hashicorp/consul/testing/deployer/topology"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
consulUID = "100"
|
||
|
|
consulGID = "1000"
|
||
|
|
consulUserArg = consulUID + ":" + consulGID
|
||
|
|
)
|
||
|
|
|
||
|
|
func tlsPrefixFromNode(node *topology.Node) string {
|
||
|
|
switch node.Kind {
|
||
|
|
case topology.NodeKindServer:
|
||
|
|
return node.Partition + "." + node.Name + ".server"
|
||
|
|
case topology.NodeKindClient:
|
||
|
|
return node.Partition + "." + node.Name + ".client"
|
||
|
|
default:
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func tlsCertCreateCommand(node *topology.Node) string {
|
||
|
|
if node.IsServer() {
|
||
|
|
return fmt.Sprintf(`consul tls cert create -server -dc=%s -node=%s`, node.Datacenter, node.PodName())
|
||
|
|
} else {
|
||
|
|
return fmt.Sprintf(`consul tls cert create -client -dc=%s`, node.Datacenter)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Sprawl) initTLS(ctx context.Context) error {
|
||
|
|
for _, cluster := range s.topology.Clusters {
|
||
|
|
|
||
|
|
var buf bytes.Buffer
|
||
|
|
|
||
|
|
// Create the CA if not already done, and proceed to do all of the
|
||
|
|
// consul CLI calls inside of a throwaway temp directory.
|
||
|
|
buf.WriteString(`
|
||
|
|
if [[ ! -f consul-agent-ca-key.pem || ! -f consul-agent-ca.pem ]]; then
|
||
|
|
consul tls ca create
|
||
|
|
fi
|
||
|
|
rm -rf tmp
|
||
|
|
mkdir -p tmp
|
||
|
|
cp -a consul-agent-ca-key.pem consul-agent-ca.pem tmp
|
||
|
|
cd tmp
|
||
|
|
`)
|
||
|
|
|
||
|
|
for _, node := range cluster.Nodes {
|
||
|
|
if !node.IsAgent() || node.Disabled {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
node.TLSCertPrefix = tlsPrefixFromNode(node)
|
||
|
|
if node.TLSCertPrefix == "" {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
expectPrefix := cluster.Datacenter + "-" + string(node.Kind) + "-consul-0"
|
||
|
|
|
||
|
|
// Conditionally generate these in isolation and rename them to
|
||
|
|
// not rely upon the numerical indexing.
|
||
|
|
buf.WriteString(fmt.Sprintf(`
|
||
|
|
if [[ ! -f %[1]s || ! -f %[2]s ]]; then
|
||
|
|
rm -f %[3]s %[4]s
|
||
|
|
%[5]s
|
||
|
|
mv -f %[3]s %[1]s
|
||
|
|
mv -f %[4]s %[2]s
|
||
|
|
fi
|
||
|
|
`,
|
||
|
|
"../"+node.TLSCertPrefix+"-key.pem", "../"+node.TLSCertPrefix+".pem",
|
||
|
|
expectPrefix+"-key.pem", expectPrefix+".pem",
|
||
|
|
tlsCertCreateCommand(node),
|
||
|
|
))
|
||
|
|
}
|
||
|
|
|
||
|
|
err := s.runner.DockerExec(ctx, []string{
|
||
|
|
"run",
|
||
|
|
"--rm",
|
||
|
|
"-i",
|
||
|
|
"--net=none",
|
||
|
|
"-v", cluster.TLSVolumeName + ":/data",
|
||
|
|
"busybox:latest",
|
||
|
|
"sh", "-c",
|
||
|
|
// Need this so the permissions stick; docker seems to treat unused volumes differently.
|
||
|
|
`touch /data/VOLUME_PLACEHOLDER && chown -R ` + consulUserArg + ` /data`,
|
||
|
|
}, io.Discard, nil)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not initialize docker volume for cert data %q: %w", cluster.TLSVolumeName, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = s.runner.DockerExec(ctx, []string{"run",
|
||
|
|
"--rm",
|
||
|
|
"-i",
|
||
|
|
"--net=none",
|
||
|
|
"-u", consulUserArg,
|
||
|
|
"-v", cluster.TLSVolumeName + ":/data",
|
||
|
|
"-w", "/data",
|
||
|
|
"--entrypoint", "",
|
||
|
|
cluster.Images.Consul,
|
||
|
|
"/bin/sh", "-ec", buf.String(),
|
||
|
|
}, io.Discard, nil)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not create all necessary TLS certificates in docker volume: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|