mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-22 15:43:49 +01:00
156b5c53f7
Add handlePacket dispatch tests for all switch cases (DSGN, SIGN, DLTSKEYSIGN, PS4SGN, PS3SGN, VITASGN, WIIUSGN, COGLNK, VITACOGLNK, DELETE). Add makeSignResponse branch tests covering PSN client PSNID field, CapLink key/host paths, MezFes minigame switch, non-localhost remote addr, and PSN token registration. Add startSignCapture enabled-path tests with temp dir and default output dir.
1091 lines
30 KiB
Go
1091 lines
30 KiB
Go
package signserver
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"erupe-ce/common/byteframe"
|
|
cfg "erupe-ce/config"
|
|
|
|
"go.uber.org/zap"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
// spyConn implements network.Conn and records plaintext packets sent.
|
|
type spyConn struct {
|
|
mu sync.Mutex
|
|
sent [][]byte // plaintext packets captured from SendPacket
|
|
readData []byte // data to return from ReadPacket (unused in handler tests)
|
|
}
|
|
|
|
func (s *spyConn) ReadPacket() ([]byte, error) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
if len(s.readData) == 0 {
|
|
return nil, fmt.Errorf("no data")
|
|
}
|
|
data := s.readData
|
|
s.readData = nil
|
|
return data, nil
|
|
}
|
|
|
|
func (s *spyConn) SendPacket(data []byte) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
cp := make([]byte, len(data))
|
|
copy(cp, data)
|
|
s.sent = append(s.sent, cp)
|
|
return nil
|
|
}
|
|
|
|
// lastSent returns the last packet sent, or nil if none.
|
|
func (s *spyConn) lastSent() []byte {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
if len(s.sent) == 0 {
|
|
return nil
|
|
}
|
|
return s.sent[len(s.sent)-1]
|
|
}
|
|
|
|
// newHandlerSession creates a Session with a spyConn for handler tests.
|
|
func newHandlerSession(userRepo SignUserRepo, charRepo SignCharacterRepo, sessionRepo SignSessionRepo, erupeConfig *cfg.Config) (*Session, *spyConn) {
|
|
logger := zap.NewNop()
|
|
mc := newMockConn() // still needed for rawConn (used by makeSignResponse for RemoteAddr)
|
|
spy := &spyConn{}
|
|
server := &Server{
|
|
logger: logger,
|
|
erupeConfig: erupeConfig,
|
|
userRepo: userRepo,
|
|
charRepo: charRepo,
|
|
sessionRepo: sessionRepo,
|
|
}
|
|
session := &Session{
|
|
logger: logger,
|
|
server: server,
|
|
rawConn: mc,
|
|
cryptConn: spy,
|
|
}
|
|
return session, spy
|
|
}
|
|
|
|
// defaultConfig returns a minimal config suitable for most handler tests.
|
|
func defaultConfig() *cfg.Config {
|
|
return &cfg.Config{
|
|
RealClientMode: cfg.ZZ,
|
|
}
|
|
}
|
|
|
|
// hashPassword creates a bcrypt hash for testing.
|
|
func hashPassword(t *testing.T, password string) string {
|
|
t.Helper()
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.MinCost)
|
|
if err != nil {
|
|
t.Fatal("failed to hash password:", err)
|
|
}
|
|
return string(hash)
|
|
}
|
|
|
|
// --- sendCode ---
|
|
|
|
func TestSendCode(t *testing.T) {
|
|
codes := []RespID{
|
|
SIGN_SUCCESS,
|
|
SIGN_EABORT,
|
|
SIGN_ECOGLINK,
|
|
SIGN_EPSI,
|
|
SIGN_EMBID,
|
|
SIGN_EELIMINATE,
|
|
SIGN_ESUSPEND,
|
|
}
|
|
|
|
for _, code := range codes {
|
|
t.Run(fmt.Sprintf("code_%d", code), func(t *testing.T) {
|
|
session, spy := newHandlerSession(nil, nil, nil, defaultConfig())
|
|
session.sendCode(code)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("sendCode() sent no packet")
|
|
}
|
|
if len(pkt) != 1 {
|
|
t.Fatalf("sendCode() packet len = %d, want 1", len(pkt))
|
|
}
|
|
if RespID(pkt[0]) != code {
|
|
t.Errorf("sendCode() = %d, want %d", pkt[0], code)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// --- authenticate ---
|
|
|
|
func TestAuthenticate_Success(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
characters: []character{{ID: 1, Name: "TestChar"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDTokenID: 100,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.authenticate("testuser", pass)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() sent no packet")
|
|
}
|
|
if len(pkt) < 1 {
|
|
t.Fatal("authenticate() packet too short")
|
|
}
|
|
if RespID(pkt[0]) != SIGN_SUCCESS {
|
|
t.Errorf("authenticate() first byte = %d, want SIGN_SUCCESS(%d)", pkt[0], SIGN_SUCCESS)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_NewCharaRequest(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
newCharCount: 0,
|
|
characters: []character{{ID: 1, Name: "TestChar"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDTokenID: 100,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.authenticate("testuser+", pass)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() sent no packet for new chara request")
|
|
}
|
|
if !charRepo.createCalled {
|
|
t.Error("authenticate() with '+' suffix should call CreateCharacter")
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_LoginFailed(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: sql.ErrNoRows,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.authenticate("unknownuser", "pass")
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() sent no packet for failed login")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EAUTH {
|
|
t.Errorf("authenticate() failed login = %v, want [%d]", pkt, SIGN_EAUTH)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_WithDebugLogging(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: sql.ErrNoRows,
|
|
}
|
|
config := defaultConfig()
|
|
config.DebugOptions.LogOutboundMessages = true
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, config)
|
|
session.authenticate("user", "pass")
|
|
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("authenticate() with debug logging sent no packet")
|
|
}
|
|
}
|
|
|
|
// --- handleDSGN ---
|
|
|
|
func TestHandleDSGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: sql.ErrNoRows,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("testuser"))
|
|
bf.WriteNullTerminatedBytes([]byte("testpass"))
|
|
bf.WriteNullTerminatedBytes([]byte("unk"))
|
|
|
|
session.handleDSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handleDSGN() sent no packet")
|
|
}
|
|
}
|
|
|
|
// --- handleWIIUSGN ---
|
|
|
|
func TestHandleWIIUSGN_Success(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
wiiuUID: 10,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
characters: []character{{ID: 1, Name: "WiiUChar"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDTokenID: 200,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.client = WIIU
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteBytes(make([]byte, 1))
|
|
key := make([]byte, 64)
|
|
copy(key, []byte("wiiu-test-key"))
|
|
bf.WriteBytes(key)
|
|
|
|
session.handleWIIUSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handleWIIUSGN() sent no packet")
|
|
}
|
|
if RespID(pkt[0]) != SIGN_SUCCESS {
|
|
t.Errorf("handleWIIUSGN() = %d, want SIGN_SUCCESS(%d)", pkt[0], SIGN_SUCCESS)
|
|
}
|
|
}
|
|
|
|
func TestHandleWIIUSGN_UnlinkedKey(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
wiiuErr: sql.ErrNoRows,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.client = WIIU
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteBytes(make([]byte, 1))
|
|
bf.WriteBytes(make([]byte, 64))
|
|
|
|
session.handleWIIUSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handleWIIUSGN() sent no packet for unlinked key")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handleWIIUSGN() unlinked = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
func TestHandleWIIUSGN_DBError(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
wiiuErr: errMockDB,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.client = WIIU
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteBytes(make([]byte, 1))
|
|
bf.WriteBytes(make([]byte, 64))
|
|
|
|
session.handleWIIUSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handleWIIUSGN() sent no packet for DB error")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handleWIIUSGN() DB error = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
// --- handlePSSGN ---
|
|
|
|
func TestHandlePSSGN_PS4_Success(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
psnUID: 20,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
characters: []character{{ID: 1, Name: "PS4Char"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDTokenID: 300,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.client = PS4
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("ps4_user_psn"))
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN(PS4) sent no packet")
|
|
}
|
|
if RespID(pkt[0]) != SIGN_SUCCESS {
|
|
t.Errorf("handlePSSGN(PS4) = %d, want SIGN_SUCCESS(%d)", pkt[0], SIGN_SUCCESS)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSSGN_PS3_Success(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
psnUID: 21,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
characters: []character{{ID: 1, Name: "PS3Char"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDTokenID: 301,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.client = PS3
|
|
|
|
// PS3: needs ≥128 bytes remaining after current position
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("0000000255"))
|
|
bf.WriteBytes([]byte("! ")) // 2 bytes
|
|
bf.WriteBytes(make([]byte, 82)) // 82 bytes padding
|
|
bf.WriteNullTerminatedBytes([]byte("ps3_user"))
|
|
for len(bf.Data()) < 140 {
|
|
bf.WriteUint8(0)
|
|
}
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN(PS3) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePSSGN_MalformedShortBuffer(t *testing.T) {
|
|
session, spy := newHandlerSession(nil, nil, nil, defaultConfig())
|
|
session.client = PS3
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteBytes(make([]byte, 10))
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN() short buffer sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handlePSSGN() short buffer = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSSGN_UnknownPSN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
psnErr: sql.ErrNoRows,
|
|
}
|
|
charRepo := &mockSignCharacterRepo{}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerPSNTokenID: 400,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.client = PS4
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("unknown_psn"))
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN() unknown PSN sent no packet")
|
|
}
|
|
// Unknown PSN calls makeSignResponse(0) which should produce SIGN_SUCCESS
|
|
if RespID(pkt[0]) != SIGN_SUCCESS {
|
|
t.Errorf("handlePSSGN() unknown PSN first byte = %d, want SIGN_SUCCESS(%d)", pkt[0], SIGN_SUCCESS)
|
|
}
|
|
if session.psn != "unknown_psn" {
|
|
t.Errorf("session.psn = %q, want %q", session.psn, "unknown_psn")
|
|
}
|
|
}
|
|
|
|
func TestHandlePSSGN_DBError(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
psnErr: errMockDB,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.client = PS4
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("some_psn"))
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN() DB error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handlePSSGN() DB error = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
// --- handlePSNLink ---
|
|
|
|
func TestHandlePSNLink_Success(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCount: 0,
|
|
psnIDForUsername: "",
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "linked_psn_id",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() success sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_SUCCESS {
|
|
t.Errorf("handlePSNLink() success = %v, want [%d]", pkt, SIGN_SUCCESS)
|
|
}
|
|
if !userRepo.setPSNIDCalled {
|
|
t.Error("handlePSNLink() should call SetPSNID on success")
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_InvalidCredentials(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: sql.ErrNoRows,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("baduser\nbadpass"))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() invalid creds sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePSNLink() invalid creds = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_TokenLookupError(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByTokenErr: errMockDB,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("badtoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() token error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePSNLink() token error = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_PSNAlreadyLinked(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCount: 1,
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "already_linked_psn",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() PSN already linked sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EPSI {
|
|
t.Errorf("handlePSNLink() PSN already linked = %v, want [%d]", pkt, SIGN_EPSI)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_AccountAlreadyLinked(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCount: 0,
|
|
psnIDForUsername: "existing_psn",
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "new_psn",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() account already linked sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EMBID {
|
|
t.Errorf("handlePSNLink() account already linked = %v, want [%d]", pkt, SIGN_EMBID)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_SetPSNIDError(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCount: 0,
|
|
psnIDForUsername: "",
|
|
setPSNIDErr: errMockDB,
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "psn_to_link",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() SetPSNID error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePSNLink() SetPSNID error = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_CountByPSNIDError(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCountErr: errMockDB,
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "psn_id",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() CountByPSNID error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePSNLink() CountByPSNID error = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
func TestHandlePSNLink_GetPSNIDForUsernameError(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
psnCount: 0,
|
|
psnIDForUsernameErr: errMockDB,
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
psnIDByToken: "psn_id",
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("testuser\n" + pass))
|
|
bf.WriteNullTerminatedBytes([]byte("sometoken"))
|
|
|
|
session.handlePSNLink(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSNLink() GetPSNIDForUsername error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePSNLink() GetPSNIDForUsername error = %v, want [%d]", pkt, SIGN_ECOGLINK)
|
|
}
|
|
}
|
|
|
|
// --- VITA client path ---
|
|
|
|
func TestHandlePSSGN_VITA_MalformedShortBuffer(t *testing.T) {
|
|
session, spy := newHandlerSession(nil, nil, nil, defaultConfig())
|
|
session.client = VITA
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteBytes(make([]byte, 50))
|
|
|
|
session.handlePSSGN(byteframe.NewByteFrameFromBytes(bf.Data()))
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePSSGN(VITA) short buffer sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handlePSSGN(VITA) short buffer = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
// --- authenticate error paths ---
|
|
|
|
func TestAuthenticate_WrongPassword(t *testing.T) {
|
|
hash := hashPassword(t, "correctpassword")
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.authenticate("testuser", "wrongpassword")
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() wrong password sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EPASS {
|
|
t.Errorf("authenticate() wrong password = %v, want [%d]", pkt, SIGN_EPASS)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_PermanentBan(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
permanentBans: 1,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.authenticate("banneduser", pass)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() permanent ban sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EELIMINATE {
|
|
t.Errorf("authenticate() permanent ban = %v, want [%d]", pkt, SIGN_EELIMINATE)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_ActiveBan(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
activeBans: 1,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.authenticate("suspendeduser", pass)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() active ban sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ESUSPEND {
|
|
t.Errorf("authenticate() active ban = %v, want [%d]", pkt, SIGN_ESUSPEND)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_DBError(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: errMockDB,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
session.authenticate("user", "pass")
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() DB error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("authenticate() DB error = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_AutoCreateError(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{
|
|
credErr: sql.ErrNoRows,
|
|
registerErr: errMockDB,
|
|
}
|
|
config := defaultConfig()
|
|
config.AutoCreateAccount = true
|
|
|
|
session, spy := newHandlerSession(userRepo, nil, nil, config)
|
|
session.authenticate("newuser", "pass")
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() auto-create error sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("authenticate() auto-create error = %v, want [%d]", pkt, SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticate_RegisterTokenError(t *testing.T) {
|
|
pass := "hunter2"
|
|
hash := hashPassword(t, pass)
|
|
|
|
userRepo := &mockSignUserRepo{
|
|
credUID: 42,
|
|
credPassword: hash,
|
|
lastLogin: time.Now(),
|
|
returnExpiry: time.Now().Add(time.Hour * 24 * 30),
|
|
}
|
|
charRepo := &mockSignCharacterRepo{
|
|
characters: []character{{ID: 1, Name: "Char"}},
|
|
}
|
|
sessionRepo := &mockSignSessionRepo{
|
|
registerUIDErr: errMockDB,
|
|
}
|
|
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
session.authenticate("user", pass)
|
|
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("authenticate() token error sent no packet")
|
|
}
|
|
// When registerUidToken fails, makeSignResponse returns SIGN_EABORT as first byte
|
|
if RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("authenticate() token error first byte = %d, want SIGN_EABORT(%d)", pkt[0], SIGN_EABORT)
|
|
}
|
|
}
|
|
|
|
// --- handlePacket dispatch ---
|
|
|
|
func TestHandlePacket_DSGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{credErr: sql.ErrNoRows}
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("DSGN:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("user"))
|
|
bf.WriteNullTerminatedBytes([]byte("pass"))
|
|
bf.WriteNullTerminatedBytes([]byte("unk"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(DSGN) error: %v", err)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(DSGN) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_SIGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{credErr: sql.ErrNoRows}
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("SIGN:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("user"))
|
|
bf.WriteNullTerminatedBytes([]byte("pass"))
|
|
bf.WriteNullTerminatedBytes([]byte("unk"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(SIGN) error: %v", err)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(SIGN) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_DLTSKEYSIGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{credErr: sql.ErrNoRows}
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("DLTSKEYSIGN:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("user"))
|
|
bf.WriteNullTerminatedBytes([]byte("pass"))
|
|
bf.WriteNullTerminatedBytes([]byte("unk"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(DLTSKEYSIGN) error: %v", err)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(DLTSKEYSIGN) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_PS4SGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{psnUID: 10}
|
|
charRepo := &mockSignCharacterRepo{characters: []character{{ID: 1, Name: "PS4Char"}}}
|
|
sessionRepo := &mockSignSessionRepo{registerUIDTokenID: 100}
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("PS4SGN:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("ps4_psn_id"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(PS4SGN) error: %v", err)
|
|
}
|
|
if session.client != PS4 {
|
|
t.Errorf("handlePacket(PS4SGN) client = %d, want PS4(%d)", session.client, PS4)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(PS4SGN) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_PS3SGN(t *testing.T) {
|
|
// PS3 with short buffer should abort
|
|
session, spy := newHandlerSession(nil, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("PS3SGN:100"))
|
|
bf.WriteBytes(make([]byte, 10)) // too short for PS3
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(PS3SGN) error: %v", err)
|
|
}
|
|
if session.client != PS3 {
|
|
t.Errorf("handlePacket(PS3SGN) client = %d, want PS3(%d)", session.client, PS3)
|
|
}
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePacket(PS3SGN) sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handlePacket(PS3SGN) short buffer = %v, want SIGN_EABORT", pkt)
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_VITASGN(t *testing.T) {
|
|
// VITA with short buffer should abort
|
|
session, spy := newHandlerSession(nil, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("VITASGN:100"))
|
|
bf.WriteBytes(make([]byte, 10)) // too short
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(VITASGN) error: %v", err)
|
|
}
|
|
if session.client != VITA {
|
|
t.Errorf("handlePacket(VITASGN) client = %d, want VITA(%d)", session.client, VITA)
|
|
}
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePacket(VITASGN) sent no packet")
|
|
}
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_EABORT {
|
|
t.Errorf("handlePacket(VITASGN) short buffer = %v, want SIGN_EABORT", pkt)
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_WIIUSGN(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{wiiuUID: 10}
|
|
charRepo := &mockSignCharacterRepo{characters: []character{{ID: 1, Name: "WiiUChar"}}}
|
|
sessionRepo := &mockSignSessionRepo{registerUIDTokenID: 200}
|
|
session, spy := newHandlerSession(userRepo, charRepo, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("WIIUSGN:100"))
|
|
bf.WriteBytes(make([]byte, 1)) // skip byte
|
|
bf.WriteBytes(make([]byte, 64)) // wiiuKey
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(WIIUSGN) error: %v", err)
|
|
}
|
|
if session.client != WIIU {
|
|
t.Errorf("handlePacket(WIIUSGN) client = %d, want WIIU(%d)", session.client, WIIU)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(WIIUSGN) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_COGLNK(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{credErr: sql.ErrNoRows}
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("COGLNK:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("baduser\nbadpass"))
|
|
bf.WriteNullTerminatedBytes([]byte("token"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(COGLNK) error: %v", err)
|
|
}
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePacket(COGLNK) sent no packet")
|
|
}
|
|
// Invalid creds → SIGN_ECOGLINK
|
|
if len(pkt) != 1 || RespID(pkt[0]) != SIGN_ECOGLINK {
|
|
t.Errorf("handlePacket(COGLNK) = %v, want SIGN_ECOGLINK", pkt)
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_VITACOGLNK(t *testing.T) {
|
|
userRepo := &mockSignUserRepo{credErr: sql.ErrNoRows}
|
|
session, spy := newHandlerSession(userRepo, nil, nil, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("VITACOGLNK:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("client_id"))
|
|
bf.WriteNullTerminatedBytes([]byte("user\npass"))
|
|
bf.WriteNullTerminatedBytes([]byte("token"))
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(VITACOGLNK) error: %v", err)
|
|
}
|
|
if spy.lastSent() == nil {
|
|
t.Fatal("handlePacket(VITACOGLNK) sent no packet")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_DELETE(t *testing.T) {
|
|
charRepo := &mockSignCharacterRepo{isNew: true}
|
|
sessionRepo := &mockSignSessionRepo{validateResult: true}
|
|
session, spy := newHandlerSession(nil, charRepo, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("DELETE:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("sesstoken"))
|
|
bf.WriteUint32(42) // characterID
|
|
bf.WriteUint32(1) // tokenID
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(DELETE) error: %v", err)
|
|
}
|
|
pkt := spy.lastSent()
|
|
if pkt == nil {
|
|
t.Fatal("handlePacket(DELETE) sent no packet")
|
|
}
|
|
if pkt[0] != 0x01 {
|
|
t.Errorf("handlePacket(DELETE) = %x, want 0x01 (DEL_SUCCESS)", pkt[0])
|
|
}
|
|
if !charRepo.hardDeleteCalled {
|
|
t.Error("handlePacket(DELETE) should call HardDelete for new character")
|
|
}
|
|
}
|
|
|
|
func TestHandlePacket_DELETE_InvalidToken(t *testing.T) {
|
|
sessionRepo := &mockSignSessionRepo{validateResult: false}
|
|
session, spy := newHandlerSession(nil, nil, sessionRepo, defaultConfig())
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteNullTerminatedBytes([]byte("DELETE:100"))
|
|
bf.WriteNullTerminatedBytes([]byte("badtoken"))
|
|
bf.WriteUint32(42) // characterID
|
|
bf.WriteUint32(1) // tokenID
|
|
|
|
err := session.handlePacket(bf.Data())
|
|
if err != nil {
|
|
t.Fatalf("handlePacket(DELETE) error: %v", err)
|
|
}
|
|
// Invalid token → deleteCharacter returns error → no packet sent
|
|
if spy.lastSent() != nil {
|
|
t.Error("handlePacket(DELETE) with invalid token should not send DEL_SUCCESS")
|
|
}
|
|
}
|