coolaj86
/
gitroast
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Work #475 and #458

This commit is contained in:
Unknown 2014-09-16 13:34:09 -04:00
parent 62f21ff3ed
commit ebb4f1b78c
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
models/publickey.go
View File

@ -14,7 +14,6 @@ import (
"os/exec" "os/exec"
"path" "path"
"path/filepath" "path/filepath"
"runtime"
"strings" "strings"
"sync" "sync"
"time" "time"
@ -23,6 +22,7 @@ import (
"github.com/gogits/gogs/modules/log" "github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/process" "github.com/gogits/gogs/modules/process"
"github.com/gogits/gogs/modules/setting"
) )
const ( const (
@ -120,23 +120,30 @@ func CheckPublicKeyString(content string) (bool, error) {
tmpFile.WriteString(content) tmpFile.WriteString(content)
tmpFile.Close() tmpFile.Close()
// … see if ssh-keygen recognizes its contents // Check if ssh-keygen recognizes its contents.
stdout, stderr, err := process.Exec("CheckPublicKeyString", "ssh-keygen", "-l", "-f", tmpPath) stdout, stderr, err := process.Exec("CheckPublicKeyString", "ssh-keygen", "-l", "-f", tmpPath)
if err != nil { if err != nil {
return false, errors.New("ssh-keygen -l -f: " + stderr) return false, errors.New("ssh-keygen -l -f: " + stderr)
} else if len(stdout) < 2 { } else if len(stdout) < 2 {
return false, errors.New("ssh-keygen returned not enough output to evaluate the key") return false, errors.New("ssh-keygen returned not enough output to evaluate the key")
} }
// The ssh-keygen in Windows does not print key type, so no need go further.
if setting.IsWindows {
return true, nil
}
sshKeygenOutput := strings.Split(stdout, " ") sshKeygenOutput := strings.Split(stdout, " ")
if len(sshKeygenOutput) < 4 { if len(sshKeygenOutput) < 4 {
return false, errors.New("Not enough fields returned by ssh-keygen -l -f") return false, errors.New("Not enough fields returned by ssh-keygen -l -f")
} }
// Check if key type and key size match.
keySize, err := com.StrTo(sshKeygenOutput[0]).Int() keySize, err := com.StrTo(sshKeygenOutput[0]).Int()
if err != nil { if err != nil {
return false, errors.New("Cannot get key size of the given key") return false, errors.New("Cannot get key size of the given key")
} }
keyType := strings.TrimSpace(sshKeygenOutput[len(sshKeygenOutput)-1]) keyType := strings.TrimSpace(sshKeygenOutput[len(sshKeygenOutput)-1])
if minimumKeySize := MinimumKeySize[keyType]; minimumKeySize == 0 { if minimumKeySize := MinimumKeySize[keyType]; minimumKeySize == 0 {
return false, errors.New("Sorry, unrecognized public key type") return false, errors.New("Sorry, unrecognized public key type")
} else if keySize < minimumKeySize { } else if keySize < minimumKeySize {
@ -163,7 +170,7 @@ func saveAuthorizedKeyFile(key *PublicKey) error {
} }
// FIXME: following command does not support in Windows. // FIXME: following command does not support in Windows.
if runtime.GOOS != "windows" { if !setting.IsWindows {
if finfo.Mode().Perm() > 0600 { if finfo.Mode().Perm() > 0600 {
log.Error(4, "authorized_keys file has unusual permission flags: %s - setting to -rw-------", finfo.Mode().Perm().String()) log.Error(4, "authorized_keys file has unusual permission flags: %s - setting to -rw-------", finfo.Mode().Perm().String())
if err = f.Chmod(0600); err != nil { if err = f.Chmod(0600); err != nil {

3
modules/setting/setting.go
View File

@ -10,6 +10,7 @@ import (
"os/exec" "os/exec"
"path" "path"
"path/filepath" "path/filepath"
"runtime"
"strings" "strings"
"time" "time"
@ -98,12 +99,14 @@ var (
CustomPath string // Custom directory path. CustomPath string // Custom directory path.
ProdMode bool ProdMode bool
RunUser string RunUser string
IsWindows bool
// I18n settings. // I18n settings.
Langs, Names []string Langs, Names []string
) )
func init() { func init() {
IsWindows = runtime.GOOS == "windows"
log.NewLogger(0, "console", `{"level": 0}`) log.NewLogger(0, "console", `{"level": 0}`)
} }