mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-21 23:22:34 +01:00
refactor(channelserver): remove Channels fallbacks, use Registry as sole cross-channel API
main.go always sets both Channels and Registry together, making the Channels fallback paths dead code. This removes: - Server.Channels field from the Server struct - 3 if/else fallback blocks in handlers_session.go (replaced with Registry.FindChannelForStage, SearchSessions, SearchStages) - 1 if/else fallback block in handlers_guild_ops.go (replaced with Registry.NotifyMailToCharID) - 3 method fallbacks in sys_channel_server.go (WorldcastMHF, FindSessionByCharID, DisconnectUser now delegate directly) Updates anti-patterns.md #6 to "accepted design" — Session struct is appropriate for this game server's handler pattern, and cross-channel coupling is now fully routed through the ChannelRegistry interface.
This commit is contained in:
@@ -572,7 +572,7 @@ func TestHandleMsgSysAuthTerminal(t *testing.T) {
|
||||
func TestHandleMsgSysLockGlobalSema_NoMatch(t *testing.T) {
|
||||
server := createMockServer()
|
||||
server.GlobalID = "test-server"
|
||||
server.Channels = []*Server{}
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{})
|
||||
session := createMockSession(1, server)
|
||||
|
||||
pkt := &mhfpacket.MsgSysLockGlobalSema{
|
||||
@@ -602,7 +602,7 @@ func TestHandleMsgSysLockGlobalSema_WithChannel(t *testing.T) {
|
||||
GlobalID: "other-server",
|
||||
}
|
||||
channel.stages.Store("stage_user123", NewStage("stage_user123"))
|
||||
server.Channels = []*Server{channel}
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{channel})
|
||||
|
||||
session := createMockSession(1, server)
|
||||
|
||||
@@ -633,7 +633,7 @@ func TestHandleMsgSysLockGlobalSema_SameServer(t *testing.T) {
|
||||
GlobalID: "test-server",
|
||||
}
|
||||
channel.stages.Store("stage_user456", NewStage("stage_user456"))
|
||||
server.Channels = []*Server{channel}
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{channel})
|
||||
|
||||
session := createMockSession(1, server)
|
||||
|
||||
|
||||
@@ -858,7 +858,7 @@ func TestHandleMsgSysUnlockGlobalSema_Coverage3(t *testing.T) {
|
||||
|
||||
func TestHandleMsgSysLockGlobalSema(t *testing.T) {
|
||||
server := createMockServer()
|
||||
server.Channels = make([]*Server, 0)
|
||||
server.Registry = NewLocalChannelRegistry(make([]*Server, 0))
|
||||
|
||||
t.Run("no_channels_returns_response", func(t *testing.T) {
|
||||
session := createMockSession(1, server)
|
||||
|
||||
@@ -318,28 +318,7 @@ func handleMsgMhfOperateGuildMember(s *Session, p mhfpacket.MHFPacket) {
|
||||
if err := s.server.mailRepo.SendMail(mail.SenderID, mail.RecipientID, mail.Subject, mail.Body, 0, 0, false, true); err != nil {
|
||||
s.logger.Warn("Failed to send guild member operation mail", zap.Error(err))
|
||||
}
|
||||
if s.server.Registry != nil {
|
||||
s.server.Registry.NotifyMailToCharID(pkt.CharID, s, &mail)
|
||||
} else {
|
||||
// Fallback: find the target session under lock, then notify outside the lock.
|
||||
var targetSession *Session
|
||||
for _, channel := range s.server.Channels {
|
||||
channel.Lock()
|
||||
for _, session := range channel.sessions {
|
||||
if session.charID == pkt.CharID {
|
||||
targetSession = session
|
||||
break
|
||||
}
|
||||
}
|
||||
channel.Unlock()
|
||||
if targetSession != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
if targetSession != nil {
|
||||
SendMailNotification(s, &mail, targetSession)
|
||||
}
|
||||
}
|
||||
s.server.Registry.NotifyMailToCharID(pkt.CharID, s, &mail)
|
||||
doAckSimpleSucceed(s, pkt.AckHandle, make([]byte, 4))
|
||||
}
|
||||
}
|
||||
|
||||
@@ -12,7 +12,6 @@ import (
|
||||
"erupe-ce/network/mhfpacket"
|
||||
"fmt"
|
||||
"io"
|
||||
"net"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
@@ -442,19 +441,7 @@ func handleMsgSysEcho(s *Session, p mhfpacket.MHFPacket) {}
|
||||
|
||||
func handleMsgSysLockGlobalSema(s *Session, p mhfpacket.MHFPacket) {
|
||||
pkt := p.(*mhfpacket.MsgSysLockGlobalSema)
|
||||
var sgid string
|
||||
if s.server.Registry != nil {
|
||||
sgid = s.server.Registry.FindChannelForStage(pkt.UserIDString)
|
||||
} else {
|
||||
for _, channel := range s.server.Channels {
|
||||
channel.stages.Range(func(id string, _ *Stage) bool {
|
||||
if strings.HasSuffix(id, pkt.UserIDString) {
|
||||
sgid = channel.GlobalID
|
||||
}
|
||||
return true
|
||||
})
|
||||
}
|
||||
}
|
||||
sgid := s.server.Registry.FindChannelForStage(pkt.UserIDString)
|
||||
bf := byteframe.NewByteFrame()
|
||||
if len(sgid) > 0 && sgid != s.server.GlobalID {
|
||||
bf.WriteUint8(0)
|
||||
@@ -517,59 +504,33 @@ func handleMsgMhfTransitMessage(s *Session, p mhfpacket.MHFPacket) {
|
||||
resp.WriteUint16(0)
|
||||
switch pkt.SearchType {
|
||||
case 1, 2, 3: // usersearchidx, usersearchname, lobbysearchname
|
||||
// Snapshot matching sessions under lock, then build response outside locks.
|
||||
type sessionResult struct {
|
||||
charID uint32
|
||||
name []byte
|
||||
stageID []byte
|
||||
ip net.IP
|
||||
port uint16
|
||||
userBin3 []byte
|
||||
}
|
||||
var results []sessionResult
|
||||
|
||||
for _, c := range s.server.Channels {
|
||||
if count == maxResults {
|
||||
break
|
||||
predicate := func(snap SessionSnapshot) bool {
|
||||
switch pkt.SearchType {
|
||||
case 1:
|
||||
return snap.CharID == cid
|
||||
case 2:
|
||||
return strings.Contains(snap.Name, term)
|
||||
case 3:
|
||||
return snap.ServerIP.String() == ip && snap.ServerPort == port && snap.StageID == term
|
||||
}
|
||||
c.Lock()
|
||||
for _, session := range c.sessions {
|
||||
if count == maxResults {
|
||||
break
|
||||
}
|
||||
if pkt.SearchType == 1 && session.charID != cid {
|
||||
continue
|
||||
}
|
||||
if pkt.SearchType == 2 && !strings.Contains(session.Name, term) {
|
||||
continue
|
||||
}
|
||||
if pkt.SearchType == 3 && session.server.IP != ip && session.server.Port != port && session.stage.id != term {
|
||||
continue
|
||||
}
|
||||
count++
|
||||
results = append(results, sessionResult{
|
||||
charID: session.charID,
|
||||
name: stringsupport.UTF8ToSJIS(session.Name),
|
||||
stageID: stringsupport.UTF8ToSJIS(session.stage.id),
|
||||
ip: net.ParseIP(c.IP).To4(),
|
||||
port: c.Port,
|
||||
userBin3: c.userBinary.GetCopy(session.charID, 3),
|
||||
})
|
||||
}
|
||||
c.Unlock()
|
||||
return false
|
||||
}
|
||||
snapshots := s.server.Registry.SearchSessions(predicate, int(maxResults))
|
||||
count = uint16(len(snapshots))
|
||||
|
||||
for _, r := range results {
|
||||
for _, snap := range snapshots {
|
||||
if !local {
|
||||
resp.WriteUint32(binary.LittleEndian.Uint32(r.ip))
|
||||
resp.WriteUint32(binary.LittleEndian.Uint32(snap.ServerIP))
|
||||
} else {
|
||||
resp.WriteUint32(localhostAddrLE)
|
||||
}
|
||||
resp.WriteUint16(r.port)
|
||||
resp.WriteUint32(r.charID)
|
||||
resp.WriteUint8(uint8(len(r.stageID) + 1))
|
||||
resp.WriteUint8(uint8(len(r.name) + 1))
|
||||
resp.WriteUint16(uint16(len(r.userBin3)))
|
||||
resp.WriteUint16(snap.ServerPort)
|
||||
resp.WriteUint32(snap.CharID)
|
||||
sjisStageID := stringsupport.UTF8ToSJIS(snap.StageID)
|
||||
sjisName := stringsupport.UTF8ToSJIS(snap.Name)
|
||||
resp.WriteUint8(uint8(len(sjisStageID) + 1))
|
||||
resp.WriteUint8(uint8(len(sjisName) + 1))
|
||||
resp.WriteUint16(uint16(len(snap.UserBinary3)))
|
||||
|
||||
// TODO: This case might be <=G2
|
||||
if s.server.erupeConfig.RealClientMode <= cfg.G1 {
|
||||
@@ -579,9 +540,9 @@ func handleMsgMhfTransitMessage(s *Session, p mhfpacket.MHFPacket) {
|
||||
}
|
||||
resp.WriteBytes(make([]byte, 8))
|
||||
|
||||
resp.WriteNullTerminatedBytes(r.stageID)
|
||||
resp.WriteNullTerminatedBytes(r.name)
|
||||
resp.WriteBytes(r.userBin3)
|
||||
resp.WriteNullTerminatedBytes(sjisStageID)
|
||||
resp.WriteNullTerminatedBytes(sjisName)
|
||||
resp.WriteBytes(snap.UserBinary3)
|
||||
}
|
||||
case 4: // lobbysearch
|
||||
type FindPartyParams struct {
|
||||
@@ -668,119 +629,81 @@ func handleMsgMhfTransitMessage(s *Session, p mhfpacket.MHFPacket) {
|
||||
}
|
||||
}
|
||||
}
|
||||
// Snapshot matching stages under lock, then build response outside locks.
|
||||
type stageResult struct {
|
||||
ip net.IP
|
||||
port uint16
|
||||
clientCount int
|
||||
reserved int
|
||||
maxPlayers uint16
|
||||
stageID string
|
||||
stageData []int16
|
||||
rawBinData0 []byte
|
||||
rawBinData1 []byte
|
||||
allStages := s.server.Registry.SearchStages(findPartyParams.StagePrefix, int(maxResults))
|
||||
|
||||
// Post-fetch filtering on snapshots (rank restriction, targets)
|
||||
type filteredStage struct {
|
||||
StageSnapshot
|
||||
stageData []int16
|
||||
}
|
||||
var stageResults []stageResult
|
||||
var stageResults []filteredStage
|
||||
for _, snap := range allStages {
|
||||
sb3 := byteframe.NewByteFrameFromBytes(snap.RawBinData3)
|
||||
_, _ = sb3.Seek(4, 0)
|
||||
|
||||
for _, c := range s.server.Channels {
|
||||
if count == maxResults {
|
||||
break
|
||||
stageDataParams := 7
|
||||
if s.server.erupeConfig.RealClientMode <= cfg.G10 {
|
||||
stageDataParams = 4
|
||||
} else if s.server.erupeConfig.RealClientMode <= cfg.Z1 {
|
||||
stageDataParams = 6
|
||||
}
|
||||
cIP := net.ParseIP(c.IP).To4()
|
||||
cPort := c.Port
|
||||
c.stages.Range(func(_ string, stage *Stage) bool {
|
||||
if count == maxResults {
|
||||
return false
|
||||
|
||||
var stageData []int16
|
||||
for i := 0; i < stageDataParams; i++ {
|
||||
if s.server.erupeConfig.RealClientMode >= cfg.Z1 {
|
||||
stageData = append(stageData, sb3.ReadInt16())
|
||||
} else {
|
||||
stageData = append(stageData, int16(sb3.ReadInt8()))
|
||||
}
|
||||
if strings.HasPrefix(stage.id, findPartyParams.StagePrefix) {
|
||||
stage.RLock()
|
||||
sb3 := byteframe.NewByteFrameFromBytes(stage.rawBinaryData[stageBinaryKey{1, 3}])
|
||||
_, _ = sb3.Seek(4, 0)
|
||||
}
|
||||
|
||||
stageDataParams := 7
|
||||
if s.server.erupeConfig.RealClientMode <= cfg.G10 {
|
||||
stageDataParams = 4
|
||||
} else if s.server.erupeConfig.RealClientMode <= cfg.Z1 {
|
||||
stageDataParams = 6
|
||||
}
|
||||
|
||||
var stageData []int16
|
||||
for i := 0; i < stageDataParams; i++ {
|
||||
if s.server.erupeConfig.RealClientMode >= cfg.Z1 {
|
||||
stageData = append(stageData, sb3.ReadInt16())
|
||||
} else {
|
||||
stageData = append(stageData, int16(sb3.ReadInt8()))
|
||||
}
|
||||
}
|
||||
|
||||
if findPartyParams.RankRestriction >= 0 {
|
||||
if stageData[0] > findPartyParams.RankRestriction {
|
||||
stage.RUnlock()
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
var hasTarget bool
|
||||
if len(findPartyParams.Targets) > 0 {
|
||||
for _, target := range findPartyParams.Targets {
|
||||
if target == stageData[1] {
|
||||
hasTarget = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !hasTarget {
|
||||
stage.RUnlock()
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
// Copy binary data under lock
|
||||
bin0 := stage.rawBinaryData[stageBinaryKey{1, 0}]
|
||||
bin0Copy := make([]byte, len(bin0))
|
||||
copy(bin0Copy, bin0)
|
||||
bin1 := stage.rawBinaryData[stageBinaryKey{1, 1}]
|
||||
bin1Copy := make([]byte, len(bin1))
|
||||
copy(bin1Copy, bin1)
|
||||
|
||||
count++
|
||||
stageResults = append(stageResults, stageResult{
|
||||
ip: cIP,
|
||||
port: cPort,
|
||||
clientCount: len(stage.clients) + len(stage.reservedClientSlots),
|
||||
reserved: len(stage.reservedClientSlots),
|
||||
maxPlayers: stage.maxPlayers,
|
||||
stageID: stage.id,
|
||||
stageData: stageData,
|
||||
rawBinData0: bin0Copy,
|
||||
rawBinData1: bin1Copy,
|
||||
})
|
||||
stage.RUnlock()
|
||||
if findPartyParams.RankRestriction >= 0 {
|
||||
if stageData[0] > findPartyParams.RankRestriction {
|
||||
continue
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
if len(findPartyParams.Targets) > 0 {
|
||||
var hasTarget bool
|
||||
for _, target := range findPartyParams.Targets {
|
||||
if target == stageData[1] {
|
||||
hasTarget = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !hasTarget {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
stageResults = append(stageResults, filteredStage{
|
||||
StageSnapshot: snap,
|
||||
stageData: stageData,
|
||||
})
|
||||
}
|
||||
count = uint16(len(stageResults))
|
||||
|
||||
for _, sr := range stageResults {
|
||||
if !local {
|
||||
resp.WriteUint32(binary.LittleEndian.Uint32(sr.ip))
|
||||
resp.WriteUint32(binary.LittleEndian.Uint32(sr.ServerIP))
|
||||
} else {
|
||||
resp.WriteUint32(localhostAddrLE)
|
||||
}
|
||||
resp.WriteUint16(sr.port)
|
||||
resp.WriteUint16(sr.ServerPort)
|
||||
|
||||
resp.WriteUint16(0) // Static?
|
||||
resp.WriteUint16(0) // Unk, [0 1 2]
|
||||
resp.WriteUint16(uint16(sr.clientCount))
|
||||
resp.WriteUint16(sr.maxPlayers)
|
||||
resp.WriteUint16(uint16(sr.ClientCount))
|
||||
resp.WriteUint16(sr.MaxPlayers)
|
||||
// TODO: Retail returned the number of clients in quests, not workshop/my series
|
||||
resp.WriteUint16(uint16(sr.reserved))
|
||||
resp.WriteUint16(uint16(sr.Reserved))
|
||||
|
||||
resp.WriteUint8(0) // Static?
|
||||
resp.WriteUint8(uint8(sr.maxPlayers))
|
||||
resp.WriteUint8(uint8(sr.MaxPlayers))
|
||||
resp.WriteUint8(1) // Static?
|
||||
resp.WriteUint8(uint8(len(sr.stageID) + 1))
|
||||
resp.WriteUint8(uint8(len(sr.rawBinData0)))
|
||||
resp.WriteUint8(uint8(len(sr.rawBinData1)))
|
||||
resp.WriteUint8(uint8(len(sr.StageID) + 1))
|
||||
resp.WriteUint8(uint8(len(sr.RawBinData0)))
|
||||
resp.WriteUint8(uint8(len(sr.RawBinData1)))
|
||||
|
||||
for i := range sr.stageData {
|
||||
if s.server.erupeConfig.RealClientMode >= cfg.Z1 {
|
||||
@@ -792,9 +715,9 @@ func handleMsgMhfTransitMessage(s *Session, p mhfpacket.MHFPacket) {
|
||||
resp.WriteUint8(0) // Unk
|
||||
resp.WriteUint8(0) // Unk
|
||||
|
||||
resp.WriteNullTerminatedBytes([]byte(sr.stageID))
|
||||
resp.WriteBytes(sr.rawBinData0)
|
||||
resp.WriteBytes(sr.rawBinData1)
|
||||
resp.WriteNullTerminatedBytes([]byte(sr.StageID))
|
||||
resp.WriteBytes(sr.RawBinData0)
|
||||
resp.WriteBytes(sr.RawBinData1)
|
||||
}
|
||||
}
|
||||
_, _ = resp.Seek(0, io.SeekStart)
|
||||
|
||||
@@ -281,7 +281,7 @@ func TestHandleMsgSysLockGlobalSema_RemoteMatch(t *testing.T) {
|
||||
clients: make(map[*Session]uint32),
|
||||
reservedClientSlots: make(map[uint32]bool),
|
||||
})
|
||||
server.Channels = []*Server{server, otherChannel}
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{server, otherChannel})
|
||||
|
||||
session := createMockSession(1, server)
|
||||
|
||||
|
||||
@@ -43,7 +43,6 @@ type Config struct {
|
||||
// own locks internally and may be acquired at any point.
|
||||
type Server struct {
|
||||
sync.Mutex
|
||||
Channels []*Server
|
||||
Registry ChannelRegistry
|
||||
ID uint16
|
||||
GlobalID string
|
||||
@@ -332,16 +331,7 @@ func (s *Server) BroadcastMHF(pkt mhfpacket.MHFPacket, ignoredSession *Session)
|
||||
|
||||
// WorldcastMHF broadcasts a packet to all sessions across all channel servers.
|
||||
func (s *Server) WorldcastMHF(pkt mhfpacket.MHFPacket, ignoredSession *Session, ignoredChannel *Server) {
|
||||
if s.Registry != nil {
|
||||
s.Registry.Worldcast(pkt, ignoredSession, ignoredChannel)
|
||||
return
|
||||
}
|
||||
for _, c := range s.Channels {
|
||||
if c == ignoredChannel {
|
||||
continue
|
||||
}
|
||||
c.BroadcastMHF(pkt, ignoredSession)
|
||||
}
|
||||
s.Registry.Worldcast(pkt, ignoredSession, ignoredChannel)
|
||||
}
|
||||
|
||||
// BroadcastChatMessage broadcasts a simple chat message to all the sessions.
|
||||
@@ -382,20 +372,7 @@ func (s *Server) DiscordScreenShotSend(charName string, title string, descriptio
|
||||
|
||||
// FindSessionByCharID looks up a session by character ID across all channels.
|
||||
func (s *Server) FindSessionByCharID(charID uint32) *Session {
|
||||
if s.Registry != nil {
|
||||
return s.Registry.FindSessionByCharID(charID)
|
||||
}
|
||||
for _, c := range s.Channels {
|
||||
c.Lock()
|
||||
for _, session := range c.sessions {
|
||||
if session.charID == charID {
|
||||
c.Unlock()
|
||||
return session
|
||||
}
|
||||
}
|
||||
c.Unlock()
|
||||
}
|
||||
return nil
|
||||
return s.Registry.FindSessionByCharID(charID)
|
||||
}
|
||||
|
||||
// DisconnectUser disconnects all sessions belonging to the given user ID.
|
||||
@@ -404,22 +381,7 @@ func (s *Server) DisconnectUser(uid uint32) {
|
||||
if err != nil {
|
||||
s.logger.Error("Failed to query characters for disconnect", zap.Error(err))
|
||||
}
|
||||
if s.Registry != nil {
|
||||
s.Registry.DisconnectUser(cids)
|
||||
return
|
||||
}
|
||||
for _, c := range s.Channels {
|
||||
c.Lock()
|
||||
for _, session := range c.sessions {
|
||||
for _, cid := range cids {
|
||||
if session.charID == cid {
|
||||
_ = session.rawConn.Close()
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
c.Unlock()
|
||||
}
|
||||
s.Registry.DisconnectUser(cids)
|
||||
}
|
||||
|
||||
// FindObjectByChar finds a stage object owned by the given character ID.
|
||||
|
||||
@@ -52,7 +52,7 @@ func (m *mockConn) WasClosed() bool {
|
||||
// createTestServer creates a test server instance
|
||||
func createTestServer() *Server {
|
||||
logger, _ := zap.NewDevelopment()
|
||||
return &Server{
|
||||
s := &Server{
|
||||
ID: 1,
|
||||
logger: logger,
|
||||
sessions: make(map[net.Conn]*Session),
|
||||
@@ -71,6 +71,8 @@ func createTestServer() *Server {
|
||||
support: make([]uint32, 30),
|
||||
},
|
||||
}
|
||||
s.Registry = NewLocalChannelRegistry([]*Server{s})
|
||||
return s
|
||||
}
|
||||
|
||||
// createTestSessionForServer creates a session for a specific server
|
||||
@@ -296,7 +298,7 @@ func TestBroadcastMHFAllSessions(t *testing.T) {
|
||||
// TestFindSessionByCharID tests finding sessions by character ID
|
||||
func TestFindSessionByCharID(t *testing.T) {
|
||||
server := createTestServer()
|
||||
server.Channels = []*Server{server} // Add itself as a channel
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{server})
|
||||
|
||||
// Create sessions with different char IDs
|
||||
charIDs := []uint32{100, 200, 300}
|
||||
|
||||
@@ -55,17 +55,19 @@ func createTestSession(mock network.Conn) *Session {
|
||||
// Create a production logger for testing (will output to stderr)
|
||||
logger, _ := zap.NewProduction()
|
||||
|
||||
server := &Server{
|
||||
erupeConfig: &cfg.Config{
|
||||
DebugOptions: cfg.DebugOptions{
|
||||
LogOutboundMessages: false,
|
||||
},
|
||||
},
|
||||
}
|
||||
server.Registry = NewLocalChannelRegistry([]*Server{server})
|
||||
s := &Session{
|
||||
logger: logger,
|
||||
sendPackets: make(chan packet, 20),
|
||||
cryptConn: mock,
|
||||
server: &Server{
|
||||
erupeConfig: &cfg.Config{
|
||||
DebugOptions: cfg.DebugOptions{
|
||||
LogOutboundMessages: false,
|
||||
},
|
||||
},
|
||||
},
|
||||
server: server,
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
@@ -50,6 +50,7 @@ func createMockServer() *Server {
|
||||
},
|
||||
}
|
||||
s.i18n = getLangStrings(s)
|
||||
s.Registry = NewLocalChannelRegistry([]*Server{s})
|
||||
return s
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user