2017-05-02 23:33:51 +00:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
2017-11-02 23:43:11 +00:00
|
|
|
"crypto/rand"
|
2017-05-02 23:33:51 +00:00
|
|
|
"crypto/rsa"
|
|
|
|
"crypto/x509"
|
|
|
|
"encoding/pem"
|
2017-08-14 15:43:33 +00:00
|
|
|
"errors"
|
2018-05-09 19:40:07 +00:00
|
|
|
"github.com/yawning/bulb/utils/pkcs1"
|
2017-11-02 23:43:11 +00:00
|
|
|
"io/ioutil"
|
2018-01-12 18:02:15 +00:00
|
|
|
"math"
|
|
|
|
"math/big"
|
2017-05-02 23:33:51 +00:00
|
|
|
)
|
|
|
|
|
2017-07-04 18:29:11 +00:00
|
|
|
const (
|
2018-01-08 19:01:23 +00:00
|
|
|
// InvalidPrivateKeyFileError is a library error, thrown when the given key file fials to load
|
2017-07-04 18:29:11 +00:00
|
|
|
InvalidPrivateKeyFileError = Error("InvalidPrivateKeyFileError")
|
2018-01-08 19:01:23 +00:00
|
|
|
|
|
|
|
// RicochetKeySize - tor onion services currently use rsa key sizes of 1024 bits
|
|
|
|
RicochetKeySize = 1024
|
2017-07-04 18:29:11 +00:00
|
|
|
)
|
|
|
|
|
2018-05-09 19:40:07 +00:00
|
|
|
// GetRandNumber is a helper function which returns a random integer, this is
|
2018-01-17 18:18:46 +00:00
|
|
|
// currently mostly used to generate messageids
|
2018-01-12 18:02:15 +00:00
|
|
|
func GetRandNumber() *big.Int {
|
2018-01-12 18:04:20 +00:00
|
|
|
num, err := rand.Int(rand.Reader, big.NewInt(math.MaxUint32))
|
|
|
|
// If we can't generate random numbers then panicking is probably
|
|
|
|
// the best option.
|
|
|
|
CheckError(err)
|
|
|
|
return num
|
2018-01-12 18:02:15 +00:00
|
|
|
}
|
|
|
|
|
2018-01-08 19:01:23 +00:00
|
|
|
// GeneratePrivateKey generates a new private key for use
|
2017-08-14 15:43:33 +00:00
|
|
|
func GeneratePrivateKey() (*rsa.PrivateKey, error) {
|
2018-01-08 19:01:23 +00:00
|
|
|
privateKey, err := rsa.GenerateKey(rand.Reader, RicochetKeySize)
|
2017-08-14 15:43:33 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, errors.New("Could not generate key: " + err.Error())
|
|
|
|
}
|
|
|
|
privateKeyDer := x509.MarshalPKCS1PrivateKey(privateKey)
|
|
|
|
return x509.ParsePKCS1PrivateKey(privateKeyDer)
|
|
|
|
}
|
|
|
|
|
2017-05-02 23:33:51 +00:00
|
|
|
// LoadPrivateKeyFromFile loads a private key from a file...
|
|
|
|
func LoadPrivateKeyFromFile(filename string) (*rsa.PrivateKey, error) {
|
|
|
|
pemData, err := ioutil.ReadFile(filename)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2017-08-14 15:43:33 +00:00
|
|
|
return ParsePrivateKey(pemData)
|
|
|
|
}
|
|
|
|
|
2018-01-08 19:01:23 +00:00
|
|
|
// ParsePrivateKey Convert a private key string to a usable private key
|
2017-08-14 15:43:33 +00:00
|
|
|
func ParsePrivateKey(pemData []byte) (*rsa.PrivateKey, error) {
|
2017-05-02 23:33:51 +00:00
|
|
|
block, _ := pem.Decode(pemData)
|
|
|
|
if block == nil || block.Type != "RSA PRIVATE KEY" {
|
2017-07-04 18:29:11 +00:00
|
|
|
return nil, InvalidPrivateKeyFileError
|
2017-05-02 23:33:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return x509.ParsePKCS1PrivateKey(block.Bytes)
|
|
|
|
}
|
2017-08-14 15:43:33 +00:00
|
|
|
|
2018-01-08 19:01:23 +00:00
|
|
|
// PrivateKeyToString turns a private key into storable string
|
2017-08-14 15:43:33 +00:00
|
|
|
func PrivateKeyToString(privateKey *rsa.PrivateKey) string {
|
|
|
|
privateKeyBlock := pem.Block{
|
2017-11-02 23:43:11 +00:00
|
|
|
Type: "RSA PRIVATE KEY",
|
2017-08-14 15:43:33 +00:00
|
|
|
Headers: nil,
|
2017-11-02 23:43:11 +00:00
|
|
|
Bytes: x509.MarshalPKCS1PrivateKey(privateKey),
|
2017-08-14 15:43:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return string(pem.EncodeToMemory(&privateKeyBlock))
|
|
|
|
}
|
2018-05-01 01:03:06 +00:00
|
|
|
|
|
|
|
// return an onion address from a private key
|
|
|
|
func GetOnionAddress(privateKey *rsa.PrivateKey) (string, error) {
|
|
|
|
addr, err := pkcs1.OnionAddr(&privateKey.PublicKey)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
} else if addr == "" {
|
|
|
|
return "", OnionAddressGenerationError
|
|
|
|
}
|
|
|
|
return addr, nil
|
2018-05-09 19:40:07 +00:00
|
|
|
}
|