d2b09d854f
Implement a new error handling library based on pkg/errors. It provides stack saving on wrapping and exports some function to add stack saving also to external errors. It also implements custom zerolog error formatting without adding too much verbosity by just printing the chain error file:line without a full stack trace of every error. * Add a --detailed-errors options to print error with they full chain * Wrap all error returns. Use errors.WithStack to wrap without adding a new messsage and error.Wrap[f] to add a message. * Add golangci-lint wrapcheck to check that external packages errors are wrapped. This won't check that internal packages error are wrapped. But we want also to ensure this case so we'll have to find something else to check also these.
81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
// Copyright 2019 Sorint.lab
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package util
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
|
|
"agola.io/agola/internal/errors"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
// GenSSHKeyPair generate an ssh keypair in rsa format, returning the private
|
|
// key (in pem encoding) and the public key (in the OpenSSH base64 format)
|
|
func GenSSHKeyPair(bits int) ([]byte, []byte, error) {
|
|
privateKey, err := rsa.GenerateKey(rand.Reader, bits)
|
|
if err != nil {
|
|
return nil, nil, errors.WithStack(err)
|
|
}
|
|
|
|
err = privateKey.Validate()
|
|
if err != nil {
|
|
return nil, nil, errors.WithStack(err)
|
|
}
|
|
privateKeyPEM := &pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(privateKey)}
|
|
|
|
var privateBuf bytes.Buffer
|
|
if err := pem.Encode(&privateBuf, privateKeyPEM); err != nil {
|
|
return nil, nil, errors.New("failed to pem encode private key")
|
|
}
|
|
|
|
pub, err := ssh.NewPublicKey(&privateKey.PublicKey)
|
|
if err != nil {
|
|
return nil, nil, errors.New("failed to generate public key")
|
|
}
|
|
|
|
// remove trailing \n returned by ssh.MarshalAuthorizedKey
|
|
return privateBuf.Bytes(), bytes.TrimSuffix(ssh.MarshalAuthorizedKey(pub), []byte("\n")), nil
|
|
}
|
|
|
|
// ExtraxtPublicKey extracts the public key from a ssh private key in pem format
|
|
func ExtractPublicKey(privateKeyPEM []byte) ([]byte, error) {
|
|
block, _ := pem.Decode(privateKeyPEM)
|
|
if block == nil || block.Type != "RSA PRIVATE KEY" {
|
|
return nil, errors.New("failed to decode PEM block containing rsa private key")
|
|
}
|
|
|
|
privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
|
|
if err != nil {
|
|
return nil, errors.New("failed to parse rsa private key")
|
|
}
|
|
|
|
err = privateKey.Validate()
|
|
if err != nil {
|
|
return nil, errors.New("failed to validate rsa private key")
|
|
}
|
|
|
|
pub, err := ssh.NewPublicKey(&privateKey.PublicKey)
|
|
if err != nil {
|
|
return nil, errors.New("failed to generate public key")
|
|
}
|
|
|
|
// remove trailing \n returned by ssh.MarshalAuthorizedKey
|
|
return bytes.TrimSuffix(ssh.MarshalAuthorizedKey(pub), []byte("\n")), nil
|
|
}
|