mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-07-02 12:05:56 +00:00
117 lines
2.8 KiB
Go
117 lines
2.8 KiB
Go
package util
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"unicode"
|
|
|
|
"code.google.com/p/go.crypto/ssh"
|
|
"code.google.com/p/go.text/unicode/norm"
|
|
)
|
|
|
|
// standard characters allowed in token string.
|
|
var chars = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")
|
|
|
|
// default token length
|
|
var length = 40
|
|
|
|
var (
|
|
lat = []*unicode.RangeTable{unicode.Letter, unicode.Number}
|
|
nop = []*unicode.RangeTable{unicode.Mark, unicode.Sk, unicode.Lm}
|
|
)
|
|
|
|
// helper function to create a Gravatar Hash
|
|
// for the given Email address.
|
|
func CreateGravatar(email string) string {
|
|
email = strings.ToLower(strings.TrimSpace(email))
|
|
hash := md5.New()
|
|
hash.Write([]byte(email))
|
|
return fmt.Sprintf("%x", hash.Sum(nil))
|
|
}
|
|
|
|
// helper function to create a Slug for the
|
|
// given string of text.
|
|
func CreateSlug(s string) string {
|
|
buf := make([]rune, 0, len(s))
|
|
dash := false
|
|
for _, r := range norm.NFKD.String(s) {
|
|
switch {
|
|
// unicode 'letters' like mandarin characters pass through
|
|
case unicode.IsOneOf(lat, r):
|
|
buf = append(buf, unicode.ToLower(r))
|
|
dash = true
|
|
case unicode.IsOneOf(nop, r):
|
|
// skip
|
|
case dash:
|
|
buf = append(buf, '-')
|
|
dash = false
|
|
}
|
|
}
|
|
if i := len(buf) - 1; i >= 0 && buf[i] == '-' {
|
|
buf = buf[:i]
|
|
}
|
|
return string(buf)
|
|
}
|
|
|
|
// GenerateToken generates random strings good for use in URIs to
|
|
// identify unique objects.
|
|
func GenerateToken() string {
|
|
b := make([]byte, length)
|
|
r := make([]byte, length+(length/4)) // storage for random bytes.
|
|
clen := byte(len(chars))
|
|
maxrb := byte(256 - (256 % len(chars)))
|
|
i := 0
|
|
for {
|
|
io.ReadFull(rand.Reader, r)
|
|
for _, c := range r {
|
|
if c >= maxrb {
|
|
// Skip this number to avoid modulo bias.
|
|
continue
|
|
}
|
|
b[i] = chars[c%clen]
|
|
i++
|
|
if i == length {
|
|
return string(b)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// SSH Functions
|
|
|
|
const (
|
|
RSA_BITS = 2048 // Default number of bits in an RSA key
|
|
RSA_BITS_MIN = 768 // Minimum number of bits in an RSA key
|
|
)
|
|
|
|
// helper function to generate an RSA Private Key.
|
|
func GeneratePrivateKey() (*rsa.PrivateKey, error) {
|
|
return rsa.GenerateKey(rand.Reader, RSA_BITS)
|
|
}
|
|
|
|
// helper function that marshalls an RSA Public Key to an SSH
|
|
// .authorized_keys format
|
|
func MarshalPublicKey(pubkey *rsa.PublicKey) string {
|
|
pk, err := ssh.NewPublicKey(pubkey)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return string(ssh.MarshalAuthorizedKey(pk))
|
|
}
|
|
|
|
// helper function that marshalls an RSA Private Key to
|
|
// a PEM encoded file.
|
|
func MarshalPrivateKey(privkey *rsa.PrivateKey) string {
|
|
privateKeyMarshaled := x509.MarshalPKCS1PrivateKey(privkey)
|
|
privateKeyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Headers: nil, Bytes: privateKeyMarshaled})
|
|
return string(privateKeyPEM)
|
|
}
|