summaryrefslogtreecommitdiffstats
path: root/models
diff options
context:
space:
mode:
authormrsdizzie <info@mrsdizzie.com>2019-11-03 06:08:18 -0500
committerLunny Xiao <xiaolunwen@gmail.com>2019-11-03 19:08:18 +0800
commitdce22efbee6a04b8c19348dac831cd88b18ac07c (patch)
tree9e40551545635048d891528686d3361e43bf55eb /models
parent022d2d8beb6297016ed26b0090c6a4a4ac404437 (diff)
downloadgitea-dce22efbee6a04b8c19348dac831cd88b18ac07c.tar.gz
gitea-dce22efbee6a04b8c19348dac831cd88b18ac07c.zip
Fix SSH2 conditonal in key parsing code (#8806)
Avoid out of bounds error by using strings.HasPrefix to check for starting SSH2 text rather than assuming user input has at least 31 characters. Add tests for bad input as well. Fixes #8800
Diffstat (limited to 'models')
-rw-r--r--models/ssh_key.go2
-rw-r--r--models/ssh_key_test.go13
2 files changed, 14 insertions, 1 deletions
diff --git a/models/ssh_key.go b/models/ssh_key.go
index 69699f24c1..f441c3e42c 100644
--- a/models/ssh_key.go
+++ b/models/ssh_key.go
@@ -107,7 +107,7 @@ func parseKeyString(content string) (string, error) {
var keyType, keyContent, keyComment string
- if content[:len(ssh2keyStart)] == ssh2keyStart {
+ if strings.HasPrefix(content, ssh2keyStart) {
// Parse SSH2 file format.
// Transform all legal line endings to a single "\n".
diff --git a/models/ssh_key_test.go b/models/ssh_key_test.go
index 4bb612a671..95cd4eeb1a 100644
--- a/models/ssh_key_test.go
+++ b/models/ssh_key_test.go
@@ -131,6 +131,19 @@ AAAAC3NzaC1lZDI1NTE5AAAAICV0MGX/W9IvLA4FXpIuUcdDcbj5KX4syHgsTy7soVgf
_, err := CheckPublicKeyString(test.content)
assert.NoError(t, err)
}
+
+ for _, invalidKeys := range []struct {
+ content string
+ }{
+ {"test"},
+ {"---- NOT A REAL KEY ----"},
+ {"bad\nkey"},
+ {"\t\t:)\t\r\n"},
+ {"\r\ntest \r\ngitea\r\n\r\n"},
+ } {
+ _, err := CheckPublicKeyString(invalidKeys.content)
+ assert.Error(t, err)
+ }
}
func Test_calcFingerprint(t *testing.T) {