mirror of
https://github.com/cbeuw/Cloak.git
synced 2024-11-19 09:25:41 +00:00
124 lines
2.9 KiB
Go
124 lines
2.9 KiB
Go
package server
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
mux "github.com/cbeuw/Cloak/internal/multiplex"
|
|
"github.com/cbeuw/Cloak/internal/server/usermanager"
|
|
"github.com/cbeuw/Cloak/internal/util"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func getSeshConfig(unordered bool) *mux.SessionConfig {
|
|
sessionKey := make([]byte, 32)
|
|
rand.Read(sessionKey)
|
|
obfuscator, _ := mux.GenerateObfs(0x00, sessionKey, true)
|
|
|
|
seshConfig := &mux.SessionConfig{
|
|
Obfuscator: obfuscator,
|
|
Valve: nil,
|
|
UnitRead: util.ReadTLS,
|
|
Unordered: unordered,
|
|
}
|
|
return seshConfig
|
|
}
|
|
|
|
func TestActiveUser_Bypass(t *testing.T) {
|
|
manager, err := usermanager.MakeLocalManager(MOCK_DB_NAME)
|
|
if err != nil {
|
|
t.Fatal("failed to make local manager", err)
|
|
}
|
|
panel := MakeUserPanel(manager)
|
|
UID, _ := base64.StdEncoding.DecodeString("u97xvcc5YoQA8obCyt9q/w==")
|
|
user, _ := panel.GetBypassUser(UID)
|
|
var sesh0 *mux.Session
|
|
var existing bool
|
|
var sesh1 *mux.Session
|
|
|
|
// get first session
|
|
sesh0, existing, err = user.GetSession(0, getSeshConfig(false))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if existing {
|
|
t.Fatal("get first session: first session returned as existing")
|
|
}
|
|
if sesh0 == nil {
|
|
t.Fatal("get first session: no session returned")
|
|
}
|
|
|
|
// get first session again
|
|
seshx, existing, err := user.GetSession(0, &mux.SessionConfig{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !existing {
|
|
t.Fatal("get first session again: first session get again returned as not existing")
|
|
}
|
|
if seshx == nil {
|
|
t.Fatal("get first session again: no session returned")
|
|
}
|
|
if seshx != sesh0 {
|
|
t.Fatal("returned a different instance")
|
|
}
|
|
|
|
// get second session
|
|
sesh1, existing, err = user.GetSession(1, getSeshConfig(false))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if existing {
|
|
t.Fatal("get second session: second session returned as existing")
|
|
}
|
|
if sesh1 == nil {
|
|
t.Fatal("get second session: no session returned")
|
|
}
|
|
|
|
if user.NumSession() != 2 {
|
|
t.Fatal("number of session is not 2")
|
|
}
|
|
|
|
user.CloseSession(0, "")
|
|
if user.NumSession() != 1 {
|
|
t.Fatal("number of session is not 1 after deleting one")
|
|
}
|
|
if !sesh0.IsClosed() {
|
|
t.Fatal("session not closed after deletion")
|
|
}
|
|
|
|
user.closeAllSessions("")
|
|
if !sesh1.IsClosed() {
|
|
t.Fatal("session not closed after user termination")
|
|
}
|
|
|
|
// get session again after termination
|
|
seshy, existing, err := user.GetSession(0, getSeshConfig(false))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if existing {
|
|
t.Fatal("get session again after termination: session returned as existing")
|
|
}
|
|
if seshy == nil {
|
|
t.Fatal("get session again after termination: no session returned")
|
|
}
|
|
if seshy == sesh0 || seshy == sesh1 {
|
|
t.Fatal("get session after termination returned the same instance")
|
|
}
|
|
|
|
user.CloseSession(0, "")
|
|
if panel.isActive(user.arrUID[:]) {
|
|
t.Fatal("user still active after last session deleted")
|
|
}
|
|
|
|
err = manager.Close()
|
|
if err != nil {
|
|
t.Fatal("failed to close localmanager", err)
|
|
}
|
|
err = os.Remove(MOCK_DB_NAME)
|
|
if err != nil {
|
|
t.Fatal("failed to delete mockdb", err)
|
|
}
|
|
}
|