mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +00:00
eb953286ec
* [NOD-641] Upgrade to github.com/pkg/errors v0.9.1 and use errors.As where needed * [NOD-641] Fix find and replace error * [NOD-641] Use errors.As for error type checking * [NOD-641] Fix errors.As for pointer types * [NOD-641] Use errors.As where needed * [NOD-641] Rename rErr->ruleErr * [NOD-641] Rename derr->dbErr * [NOD-641] e->flagsErr where necessary * [NOD-641] change jerr to more appropriate name * [NOD-641] Rename cerr->bdRuleErr * [NOD-641] Rename serr->scriptErr * [NOD-641] Use errors.Is instead of testutil.AreErrorsEqual in TestNewHashFromStr * [NOD-641] Rename bdRuleErr->dagRuleErr * [NOD-641] Rename mErr->msgErr * [NOD-641] Rename dErr->deserializeErr
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
// Copyright (c) 2013-2014 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/pkg/errors"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
flags "github.com/jessevdk/go-flags"
|
|
"github.com/kaspanet/kaspad/util"
|
|
)
|
|
|
|
type config struct {
|
|
Directory string `short:"d" long:"directory" description:"Directory to write certificate pair"`
|
|
Years int `short:"y" long:"years" description:"How many years a certificate is valid for"`
|
|
Organization string `short:"o" long:"org" description:"Organization in certificate"`
|
|
ExtraHosts []string `short:"H" long:"host" description:"Additional hosts/IPs to create certificate for"`
|
|
Force bool `short:"f" long:"force" description:"Force overwriting of any old certs and keys"`
|
|
}
|
|
|
|
func main() {
|
|
cfg := config{
|
|
Years: 10,
|
|
Organization: "gencerts",
|
|
}
|
|
parser := flags.NewParser(&cfg, flags.Default)
|
|
_, err := parser.Parse()
|
|
if err != nil {
|
|
var flagsErr *flags.Error
|
|
if ok := errors.As(err, &flagsErr); !ok || flagsErr.Type != flags.ErrHelp {
|
|
parser.WriteHelp(os.Stderr)
|
|
}
|
|
return
|
|
}
|
|
|
|
if cfg.Directory == "" {
|
|
var err error
|
|
cfg.Directory, err = os.Getwd()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "no directory specified and cannot get working directory\n")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
cfg.Directory = cleanAndExpandPath(cfg.Directory)
|
|
certFile := filepath.Join(cfg.Directory, "rpc.cert")
|
|
keyFile := filepath.Join(cfg.Directory, "rpc.key")
|
|
|
|
if !cfg.Force {
|
|
if fileExists(certFile) || fileExists(keyFile) {
|
|
fmt.Fprintf(os.Stderr, "%s: certificate and/or key files exist; use -f to force\n", cfg.Directory)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
validUntil := time.Now().Add(time.Duration(cfg.Years) * 365 * 24 * time.Hour)
|
|
cert, key, err := util.NewTLSCertPair(cfg.Organization, validUntil, cfg.ExtraHosts)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "cannot generate certificate pair: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Write cert and key files.
|
|
if err = ioutil.WriteFile(certFile, cert, 0666); err != nil {
|
|
fmt.Fprintf(os.Stderr, "cannot write cert: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
if err = ioutil.WriteFile(keyFile, key, 0600); err != nil {
|
|
os.Remove(certFile)
|
|
fmt.Fprintf(os.Stderr, "cannot write key: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// cleanAndExpandPath expands environement variables and leading ~ in the
|
|
// passed path, cleans the result, and returns it.
|
|
func cleanAndExpandPath(path string) string {
|
|
// Expand initial ~ to OS specific home directory.
|
|
if strings.HasPrefix(path, "~") {
|
|
appHomeDir := util.AppDataDir("gencerts", false)
|
|
homeDir := filepath.Dir(appHomeDir)
|
|
path = strings.Replace(path, "~", homeDir, 1)
|
|
}
|
|
|
|
// NOTE: The os.ExpandEnv doesn't work with Windows-style %VARIABLE%,
|
|
// but they variables can still be expanded via POSIX-style $VARIABLE.
|
|
return filepath.Clean(os.ExpandEnv(path))
|
|
}
|
|
|
|
// filesExists reports whether the named file or directory exists.
|
|
func fileExists(name string) bool {
|
|
if _, err := os.Stat(name); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|