go-http-peering/cmd/keygen/util.go

88 lines
1.8 KiB
Go
Raw Permalink Normal View History

2019-02-22 17:35:49 +01:00
package main
import (
"bytes"
"crypto/rsa"
"fmt"
"io/ioutil"
"os"
"strings"
2019-02-22 17:35:49 +01:00
"syscall"
"forge.cadoles.com/Cadoles/go-http-peering/crypto"
2022-09-12 17:46:59 +02:00
"github.com/pkg/errors"
2019-02-22 17:35:49 +01:00
"golang.org/x/crypto/ssh/terminal"
)
func getPassphrase() ([]byte, error) {
passphrase, exists := os.LookupEnv("KEY_PASSPHRASE")
if exists {
2019-02-22 17:35:49 +01:00
return []byte(passphrase), nil
}
return askPassphrase()
}
func askPassphrase() ([]byte, error) {
fmt.Print("Passphrase: ")
passphrase, err := terminal.ReadPassword(syscall.Stdin)
if err != nil {
2022-09-12 17:46:59 +02:00
return nil, errors.WithStack(err)
2019-02-22 17:35:49 +01:00
}
fmt.Println()
fmt.Print("Confirm passphrase: ")
passphraseConfirmation, err := terminal.ReadPassword(syscall.Stdin)
if err != nil {
2022-09-12 17:46:59 +02:00
return nil, errors.WithStack(err)
2019-02-22 17:35:49 +01:00
}
fmt.Println()
if !bytes.Equal(passphrase, passphraseConfirmation) {
return nil, errors.New("passphrases does not match")
}
return passphrase, nil
}
func loadPrivateKey() (*rsa.PrivateKey, error) {
if keyFile == "" {
return nil, errors.New("you must specify a key file to load")
}
pem, err := ioutil.ReadFile(keyFile)
if err != nil {
2022-09-12 17:46:59 +02:00
return nil, errors.WithStack(err)
2019-02-22 17:35:49 +01:00
}
passphrase, err := getPassphrase()
if err != nil {
2022-09-12 17:46:59 +02:00
return nil, errors.WithStack(err)
2019-02-22 17:35:49 +01:00
}
privateKey, err := crypto.DecodePEMEncryptedPrivateKey(pem, passphrase)
if err != nil {
2022-09-12 17:46:59 +02:00
return nil, errors.WithStack(err)
2019-02-22 17:35:49 +01:00
}
return privateKey, nil
}
func loadToken() (string, error) {
if tokenFile == "" {
return "", errors.New("you must specify a token file to load")
}
token, err := os.ReadFile(tokenFile)
if err != nil {
return "", errors.WithStack(err)
}
return strings.TrimSpace(string(token)), nil
}
2019-02-22 17:35:49 +01:00
func handleError(err error) {
if !debug {
2022-09-12 17:46:59 +02:00
fmt.Printf("%+v\n", errors.WithStack(err))
2019-02-22 17:35:49 +01:00
} else {
2022-09-12 17:46:59 +02:00
panic(fmt.Sprintf("%+v", errors.WithStack(err)))
2019-02-22 17:35:49 +01:00
}
os.Exit(1)
}