6543
c063329e9a
All checks were successful
continuous-integration/drone/push Build is passing
Unexport generateToken() move CreateLogin into task Create func config.SetDefaultLogin() Unexport loadConfig() & saveConfig unexport config var make SetDefaultLogin() case insensitive update func descriptions move FindSSHKey to task module Reviewed-on: #288 Reviewed-by: Norwin <noerw@noreply.gitea.io> Reviewed-by: Andrew Thornton <art27@cantab.net> Co-Authored-By: 6543 <6543@obermui.de> Co-Committed-By: 6543 <6543@obermui.de>
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package task
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"code.gitea.io/tea/modules/utils"
|
|
|
|
"code.gitea.io/sdk/gitea"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
// findSSHKey retrieves the ssh keys registered in gitea, and tries to find
|
|
// a matching private key in ~/.ssh/. If no match is found, path is empty.
|
|
func findSSHKey(client *gitea.Client) (string, error) {
|
|
// get keys registered on gitea instance
|
|
keys, _, err := client.ListMyPublicKeys(gitea.ListPublicKeysOptions{})
|
|
if err != nil || len(keys) == 0 {
|
|
return "", err
|
|
}
|
|
|
|
// enumerate ~/.ssh/*.pub files
|
|
glob, err := utils.AbsPathWithExpansion("~/.ssh/*.pub")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
localPubkeyPaths, err := filepath.Glob(glob)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// parse each local key with present privkey & compare fingerprints to online keys
|
|
for _, pubkeyPath := range localPubkeyPaths {
|
|
var pubkeyFile []byte
|
|
pubkeyFile, err = ioutil.ReadFile(pubkeyPath)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
fields := strings.Split(string(pubkeyFile), " ")
|
|
if len(fields) < 2 { // first word is key type, second word is key material
|
|
continue
|
|
}
|
|
|
|
var keymaterial []byte
|
|
keymaterial, err = base64.StdEncoding.DecodeString(fields[1])
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
var pubkey ssh.PublicKey
|
|
pubkey, err = ssh.ParsePublicKey(keymaterial)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
privkeyPath := strings.TrimSuffix(pubkeyPath, ".pub")
|
|
var exists bool
|
|
exists, err = utils.FileExist(privkeyPath)
|
|
if err != nil || !exists {
|
|
continue
|
|
}
|
|
|
|
// if pubkey fingerprints match, return path to corresponding privkey.
|
|
fingerprint := ssh.FingerprintSHA256(pubkey)
|
|
for _, key := range keys {
|
|
if fingerprint == key.Fingerprint {
|
|
return privkeyPath, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
return "", err
|
|
}
|