forked from TrueCloudLab/certificates
114 lines
3.7 KiB
Go
114 lines
3.7 KiB
Go
package pki
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/smallstep/certificates/templates"
|
|
"github.com/smallstep/cli/config"
|
|
"github.com/smallstep/cli/errs"
|
|
"github.com/smallstep/cli/utils"
|
|
)
|
|
|
|
// SSHTemplates contains the configuration of default templates used on ssh.
|
|
// Relative paths are relative to the StepPath.
|
|
var SSHTemplates = &templates.SSHTemplates{
|
|
User: []templates.Template{
|
|
{Name: "include.tpl", Type: templates.Snippet, TemplatePath: "templates/ssh/include.tpl", Path: "~/.ssh/config", Comment: "#"},
|
|
{Name: "config.tpl", Type: templates.File, TemplatePath: "templates/ssh/config.tpl", Path: "ssh/config", Comment: "#"},
|
|
{Name: "known_hosts.tpl", Type: templates.File, TemplatePath: "templates/ssh/known_hosts.tpl", Path: "ssh/known_hosts", Comment: "#"},
|
|
},
|
|
Host: []templates.Template{
|
|
{Name: "sshd_config.tpl", Type: templates.Snippet, TemplatePath: "templates/ssh/sshd_config.tpl", Path: "/etc/ssh/sshd_config", Comment: "#"},
|
|
{Name: "ca.tpl", Type: templates.Snippet, TemplatePath: "templates/ssh/ca.tpl", Path: "/etc/ssh/ca.pub", Comment: "#"},
|
|
},
|
|
}
|
|
|
|
// SSHTemplateData contains the data of the default templates used on ssh.
|
|
var SSHTemplateData = map[string]string{
|
|
// include.tpl adds the step ssh config file
|
|
"include.tpl": `Host *
|
|
Include {{.User.StepPath}}/ssh/config`,
|
|
|
|
// config.tpl is the step ssh config file, it includes the Match rule
|
|
// and references the step known_hosts file
|
|
"config.tpl": `Match exec "step ssh check-host %h"
|
|
ForwardAgent yes
|
|
{{- if .User.User }}
|
|
User {{.User.User}}
|
|
{{- end }}
|
|
UserKnownHostsFile {{.User.StepPath}}/ssh/known_hosts
|
|
ProxyCommand step ssh proxycommand {{- if .User.RegistryUsername}} --username {{.User.RegistryUsername}}{{end}}{{- if .User.RegistryPassword}} --password {{.User.RegistryPassword}}{{end}} %r %h %p`,
|
|
|
|
// known_hosts.tpl authorizes the ssh hosts key
|
|
"known_hosts.tpl": `@cert-authority * {{.Step.SSH.HostKey.Type}} {{.Step.SSH.HostKey.Marshal | toString | b64enc}}
|
|
{{- range .Step.SSH.HostFederatedKeys}}
|
|
@cert-authority * {{.Type}} {{.Marshal | toString | b64enc}}
|
|
{{- end}}
|
|
`,
|
|
|
|
// sshd_config.tpl adds the configuration to support certificates
|
|
"sshd_config.tpl": `TrustedUserCAKeys /etc/ssh/ca.pub
|
|
HostCertificate /etc/ssh/{{.User.Certificate}}
|
|
HostKey /etc/ssh/{{.User.Key}}`,
|
|
|
|
// ca.tpl contains the public key used to authorized clients
|
|
"ca.tpl": `{{.Step.SSH.UserKey.Type}} {{.Step.SSH.UserKey.Marshal | toString | b64enc}}
|
|
{{- range .Step.SSH.UserFederatedKeys}}
|
|
{{.Type}} {{.Marshal | toString | b64enc}}
|
|
{{- end}}
|
|
`,
|
|
}
|
|
|
|
// getTemplates returns all the templates enabled
|
|
func (p *PKI) getTemplates() *templates.Templates {
|
|
if !p.enableSSH {
|
|
return nil
|
|
}
|
|
|
|
return &templates.Templates{
|
|
SSH: SSHTemplates,
|
|
Data: map[string]interface{}{},
|
|
}
|
|
}
|
|
|
|
// generateTemplates generates given templates.
|
|
func generateTemplates(t *templates.Templates) error {
|
|
if t == nil {
|
|
return nil
|
|
}
|
|
|
|
base := GetTemplatesPath()
|
|
// Generate SSH templates
|
|
if t.SSH != nil {
|
|
// all ssh templates are under ssh:
|
|
sshDir := filepath.Join(base, "ssh")
|
|
if _, err := os.Stat(sshDir); os.IsNotExist(err) {
|
|
if err = os.MkdirAll(sshDir, 0700); err != nil {
|
|
return errs.FileError(err, sshDir)
|
|
}
|
|
}
|
|
// Create all templates
|
|
for _, t := range t.SSH.User {
|
|
data, ok := SSHTemplateData[t.Name]
|
|
if !ok {
|
|
return errors.Errorf("template %s does not exists", t.Name)
|
|
}
|
|
if err := utils.WriteFile(config.StepAbs(t.TemplatePath), []byte(data), 0644); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
for _, t := range t.SSH.Host {
|
|
data, ok := SSHTemplateData[t.Name]
|
|
if !ok {
|
|
return errors.Errorf("template %s does not exists", t.Name)
|
|
}
|
|
if err := utils.WriteFile(config.StepAbs(t.TemplatePath), []byte(data), 0644); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|