mirror of
https://github.com/drakkan/sftpgo.git
synced 2024-11-21 23:20:24 +00:00
85 lines
3.3 KiB
Go
85 lines
3.3 KiB
Go
// Copyright (C) 2019-2022 Nicola Murino
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published
|
|
// by the Free Software Foundation, version 3.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package common
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBasicAuth(t *testing.T) {
|
|
httpAuth, err := NewBasicAuthProvider("")
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.IsEnabled())
|
|
|
|
_, err = NewBasicAuthProvider("missing path")
|
|
require.Error(t, err)
|
|
|
|
authUserFile := filepath.Join(os.TempDir(), "http_users.txt")
|
|
authUserData := []byte("test1:$2y$05$bcHSED7aO1cfLto6ZdDBOOKzlwftslVhtpIkRhAtSa4GuLmk5mola\n")
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
|
|
httpAuth, err = NewBasicAuthProvider(authUserFile)
|
|
require.NoError(t, err)
|
|
require.True(t, httpAuth.IsEnabled())
|
|
require.False(t, httpAuth.ValidateCredentials("test1", "wrong1"))
|
|
require.False(t, httpAuth.ValidateCredentials("test2", "password2"))
|
|
require.True(t, httpAuth.ValidateCredentials("test1", "password1"))
|
|
|
|
authUserData = append(authUserData, []byte("test2:$1$OtSSTL8b$bmaCqEksI1e7rnZSjsIDR1\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test2", "wrong2"))
|
|
require.True(t, httpAuth.ValidateCredentials("test2", "password2"))
|
|
|
|
authUserData = append(authUserData, []byte("test2:$apr1$gLnIkRIf$Xr/6aJfmIrihP4b2N2tcs/\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test2", "wrong2"))
|
|
require.True(t, httpAuth.ValidateCredentials("test2", "password2"))
|
|
|
|
authUserData = append(authUserData, []byte("test3:$apr1$gLnIkRIf$Xr/6aJfmIrihP4b2N2tcs/\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test3", "password3"))
|
|
|
|
authUserData = append(authUserData, []byte("test4:$invalid$gLnIkRIf$Xr/6$aJfmIr$ihP4b2N2tcs/\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test4", "password3"))
|
|
|
|
if runtime.GOOS != "windows" {
|
|
authUserData = append(authUserData, []byte("test5:$apr1$gLnIkRIf$Xr/6aJfmIrihP4b2N2tcs/\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
err = os.Chmod(authUserFile, 0001)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test5", "password2"))
|
|
err = os.Chmod(authUserFile, os.ModePerm)
|
|
require.NoError(t, err)
|
|
}
|
|
authUserData = append(authUserData, []byte("\"foo\"bar\"\r\n")...)
|
|
err = os.WriteFile(authUserFile, authUserData, os.ModePerm)
|
|
require.NoError(t, err)
|
|
require.False(t, httpAuth.ValidateCredentials("test2", "password2"))
|
|
|
|
err = os.Remove(authUserFile)
|
|
require.NoError(t, err)
|
|
}
|