mirrors/mostlymatter
1
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

Increase entropy for MFA secret (#14290)

Просмотр исходного кода 
Co-authored-by: mattermod <mattermod@users.noreply.github.com>
Этот коммит содержится в:
Claudio Costa
2020-04-22 18:46:16 +02:00
коммит произвёл GitHub
родитель 50821d1a34
Коммит ad68af10df
4 изменённых файлов: 40 добавлений и 26 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

4
services/mfa/mfa.go
Просмотреть файл

@@ -4,7 +4,6 @@
package mfa
import (
b32 "encoding/base32"
"fmt"
"net/http"
"net/url"
@@ -18,6 +17,7 @@ import (
)
const (
// This will result in 160 bits of entropy (base32 encoded), as recommended by rfc4226.
MFA_SECRET_SIZE = 20
)
@@ -58,7 +58,7 @@ func (m *Mfa) GenerateSecret(user *model.User) (string, []byte, *model.AppError)
issuer := getIssuerFromUrl(*m.ConfigService.Config().ServiceSettings.SiteURL)
secret := b32.StdEncoding.EncodeToString([]byte(model.NewRandomString(MFA_SECRET_SIZE)))
secret := model.NewRandomBase32String(MFA_SECRET_SIZE)
authLink := fmt.Sprintf("otpauth://totp/%s:%s?secret=%s&issuer=%s", issuer, user.Email, secret, issuer)

28
services/mfa/mfa_test.go
Просмотреть файл

@@ -4,7 +4,6 @@
package mfa
import (
b32 "encoding/base32"
"fmt"
"net/http"
"net/url"
@@ -37,7 +36,7 @@ func TestGenerateSecret(t *testing.T) {
mfa := New(wrongConfigService, nil)
_, _, err := mfa.GenerateSecret(user)
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.mfa_disabled.app_error")
require.Equal(t, "mfa.mfa_disabled.app_error", err.Id)
})
t.Run("fail on store action fail", func(t *testing.T) {
@@ -51,7 +50,7 @@ func TestGenerateSecret(t *testing.T) {
mfa := New(configService, &storeMock)
_, _, err := mfa.GenerateSecret(user)
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.generate_qr_code.save_secret.app_error")
require.Equal(t, "mfa.generate_qr_code.save_secret.app_error", err.Id)
})
t.Run("Successful generate secret", func(t *testing.T) {
@@ -94,7 +93,8 @@ func TestGetIssuerFromUrl(t *testing.T) {
func TestActivate(t *testing.T) {
user := &model.User{Id: model.NewId(), Roles: "system_user"}
user.MfaSecret = b32.StdEncoding.EncodeToString([]byte(model.NewRandomString(MFA_SECRET_SIZE)))
user.MfaSecret = model.NewRandomBase32String(MFA_SECRET_SIZE)
token := dgoogauth.ComputeCode(user.MfaSecret, time.Now().UTC().Unix()/30)
config := model.Config{}
@@ -110,21 +110,21 @@ func TestActivate(t *testing.T) {
mfa := New(wrongConfigService, nil)
err := mfa.Activate(user, "not-important")
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.mfa_disabled.app_error")
require.Equal(t, "mfa.mfa_disabled.app_error", err.Id)
})
t.Run("fail on wrongly formatted token", func(t *testing.T) {
mfa := New(configService, nil)
err := mfa.Activate(user, "invalid-token")
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.activate.authenticate.app_error")
require.Equal(t, "mfa.activate.authenticate.app_error", err.Id)
})
t.Run("fail on invalid token", func(t *testing.T) {
mfa := New(configService, nil)
err := mfa.Activate(user, "000000")
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.activate.bad_token.app_error")
require.Equal(t, "mfa.activate.bad_token.app_error", err.Id)
})
t.Run("fail on store action fail", func(t *testing.T) {
@@ -138,7 +138,7 @@ func TestActivate(t *testing.T) {
mfa := New(configService, &storeMock)
err := mfa.Activate(user, fmt.Sprintf("%06d", token))
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.activate.save_active.app_error")
require.Equal(t, "mfa.activate.save_active.app_error", err.Id)
})
t.Run("Successful activate", func(t *testing.T) {
@@ -171,7 +171,7 @@ func TestDeactivate(t *testing.T) {
mfa := New(wrongConfigService, nil)
err := mfa.Deactivate(user.Id)
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.mfa_disabled.app_error")
require.Equal(t, "mfa.mfa_disabled.app_error", err.Id)
})
t.Run("fail on store UpdateMfaActive action fail", func(t *testing.T) {
@@ -188,7 +188,7 @@ func TestDeactivate(t *testing.T) {
mfa := New(configService, &storeMock)
err := mfa.Deactivate(user.Id)
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.deactivate.save_active.app_error")
require.Equal(t, "mfa.deactivate.save_active.app_error", err.Id)
})
t.Run("fail on store UpdateMfaSecret action fail", func(t *testing.T) {
@@ -205,7 +205,7 @@ func TestDeactivate(t *testing.T) {
mfa := New(configService, &storeMock)
err := mfa.Deactivate(user.Id)
require.NotNil(t, err)
require.Equal(t, err.Id, "mfa.deactivate.save_secret.app_error")
require.Equal(t, "mfa.deactivate.save_secret.app_error", err.Id)
})
t.Run("Successful deactivate", func(t *testing.T) {
@@ -226,7 +226,7 @@ func TestDeactivate(t *testing.T) {
}
func TestValidateToken(t *testing.T) {
secret := b32.StdEncoding.EncodeToString([]byte(model.NewRandomString(MFA_SECRET_SIZE)))
secret := model.NewRandomBase32String(MFA_SECRET_SIZE)
token := dgoogauth.ComputeCode(secret, time.Now().UTC().Unix()/30)
config := model.Config{}
@@ -243,7 +243,7 @@ func TestValidateToken(t *testing.T) {
ok, err := mfa.ValidateToken(secret, fmt.Sprintf("%06d", token))
require.NotNil(t, err)
require.False(t, ok)
require.Equal(t, err.Id, "mfa.mfa_disabled.app_error")
require.Equal(t, "mfa.mfa_disabled.app_error", err.Id)
})
t.Run("fail on wrongly formatted token", func(t *testing.T) {
@@ -251,7 +251,7 @@ func TestValidateToken(t *testing.T) {
ok, err := mfa.ValidateToken(secret, "invalid-token")
require.NotNil(t, err)
require.False(t, ok)
require.Equal(t, err.Id, "mfa.validate_token.authenticate.app_error")
require.Equal(t, "mfa.validate_token.authenticate.app_error", err.Id)
})
t.Run("fail on invalid token", func(t *testing.T) {