mirror of https://github.com/status-im/consul.git
116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
package create
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/hashicorp/consul/command/flags"
|
|
"github.com/hashicorp/consul/command/tls"
|
|
"github.com/hashicorp/consul/tlsutil"
|
|
"github.com/mitchellh/cli"
|
|
)
|
|
|
|
func New(ui cli.Ui) *cmd {
|
|
c := &cmd{UI: ui}
|
|
c.init()
|
|
return c
|
|
}
|
|
|
|
type cmd struct {
|
|
UI cli.Ui
|
|
flags *flag.FlagSet
|
|
help string
|
|
days int
|
|
domain string
|
|
constraint bool
|
|
additionalConstraints flags.AppendSliceValue
|
|
}
|
|
|
|
func (c *cmd) init() {
|
|
c.flags = flag.NewFlagSet("", flag.ContinueOnError)
|
|
// TODO: perhaps add a -years arg to better capture user intent given that leap years are a thing
|
|
c.flags.IntVar(&c.days, "days", 1825, "Provide number of days the CA is valid for from now on. Defaults to 5 years.")
|
|
c.flags.BoolVar(&c.constraint, "name-constraint", false, "Add name constraints for the CA. Results in rejecting "+
|
|
"certificates for other DNS than specified. If turned on localhost and -domain will be added to the allowed "+
|
|
"DNS. If the UI is going to be served over HTTPS its DNS has to be added with -additional-constraint. It is not "+
|
|
"possible to add that after the fact! Defaults to false.")
|
|
c.flags.StringVar(&c.domain, "domain", "consul", "Domain of consul cluster. Only used in combination with -name-constraint. Defaults to consul.")
|
|
c.flags.Var(&c.additionalConstraints, "additional-name-constraint", "Add name constraints for the CA. Results in rejecting certificates "+
|
|
"for other DNS than specified. Can be used multiple times. Only used in combination with -name-constraint.")
|
|
c.help = flags.Usage(help, c.flags)
|
|
}
|
|
|
|
func (c *cmd) Run(args []string) int {
|
|
if err := c.flags.Parse(args); err != nil {
|
|
if err == flag.ErrHelp {
|
|
return 0
|
|
}
|
|
c.UI.Error(fmt.Sprintf("Failed to parse args: %v", err))
|
|
return 1
|
|
}
|
|
|
|
certFileName := fmt.Sprintf("%s-agent-ca.pem", c.domain)
|
|
pkFileName := fmt.Sprintf("%s-agent-ca-key.pem", c.domain)
|
|
|
|
if !(tls.FileDoesNotExist(certFileName)) {
|
|
c.UI.Error(certFileName + " already exists.")
|
|
return 1
|
|
}
|
|
if !(tls.FileDoesNotExist(pkFileName)) {
|
|
c.UI.Error(pkFileName + " already exists.")
|
|
return 1
|
|
}
|
|
|
|
sn, err := tlsutil.GenerateSerialNumber()
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
s, pk, err := tlsutil.GeneratePrivateKey()
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
}
|
|
constraints := []string{}
|
|
if c.constraint {
|
|
constraints = append(c.additionalConstraints, []string{c.domain, "localhost"}...)
|
|
}
|
|
ca, err := tlsutil.GenerateCA(s, sn, c.days, constraints)
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
}
|
|
caFile, err := os.Create(certFileName)
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
}
|
|
caFile.WriteString(ca)
|
|
c.UI.Output("==> Saved " + certFileName)
|
|
pkFile, err := os.Create(pkFileName)
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
}
|
|
pkFile.WriteString(pk)
|
|
c.UI.Output("==> Saved " + pkFileName)
|
|
|
|
return 0
|
|
}
|
|
|
|
func (c *cmd) Synopsis() string {
|
|
return synopsis
|
|
}
|
|
|
|
func (c *cmd) Help() string {
|
|
return c.help
|
|
}
|
|
|
|
const synopsis = "Create a new consul CA"
|
|
const help = `
|
|
Usage: consul tls ca create [options]
|
|
|
|
Create a new consul CA:
|
|
|
|
$ consul tls ca create
|
|
==> saved consul-agent-ca.pem
|
|
==> saved consul-agent-ca-key.pem
|
|
`
|