2021-01-05 11:34:35 -08:00
|
|
|
package multiserver
|
|
|
|
|
|
|
|
import (
|
2021-01-09 03:26:30 -08:00
|
|
|
"errors"
|
|
|
|
"net"
|
|
|
|
"sync"
|
2021-01-18 13:08:22 -08:00
|
|
|
|
|
|
|
"github.com/anon55555/mt/rudp"
|
2021-01-05 11:34:35 -08:00
|
|
|
)
|
|
|
|
|
2021-01-06 05:42:55 -08:00
|
|
|
var ErrPlayerLimitReached = errors.New("player limit reached")
|
|
|
|
|
2021-01-05 11:34:35 -08:00
|
|
|
type Listener struct {
|
2021-01-18 13:08:22 -08:00
|
|
|
*rudp.Listener
|
|
|
|
mu sync.RWMutex
|
|
|
|
peers map[*Peer]struct{}
|
2021-01-05 11:34:35 -08:00
|
|
|
}
|
|
|
|
|
2021-01-06 14:39:54 -08:00
|
|
|
var listener *Listener
|
|
|
|
|
2021-01-05 11:34:35 -08:00
|
|
|
func Listen(conn net.PacketConn) *Listener {
|
2021-01-18 13:08:22 -08:00
|
|
|
return &Listener{Listener: rudp.Listen(conn)}
|
2021-01-05 11:34:35 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Accept waits for and returns a connecting Peer
|
|
|
|
// You should keep calling this until it returns ErrClosed
|
|
|
|
// so it doesn't leak a goroutine
|
|
|
|
func (l *Listener) Accept() (*Peer, error) {
|
2021-01-19 08:15:28 -08:00
|
|
|
rp, err := l.Listener.Accept()
|
2021-01-18 13:08:22 -08:00
|
|
|
if err != nil {
|
2021-01-05 11:34:35 -08:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2021-01-19 08:15:28 -08:00
|
|
|
clt := &Peer{Peer: rp}
|
|
|
|
|
2021-01-05 11:34:35 -08:00
|
|
|
l.mu.Lock()
|
2021-01-18 13:08:22 -08:00
|
|
|
l.peers[clt] = struct{}{}
|
|
|
|
l.mu.Unlock()
|
|
|
|
go func() {
|
|
|
|
<-clt.Disco()
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
l.mu.Lock()
|
|
|
|
delete(l.peers, clt)
|
|
|
|
l.mu.Unlock()
|
|
|
|
}()
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
clt.forward = true
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
clt.aoIDs = make(map[uint16]bool)
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
connectedPeers++
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
maxPeers, ok := GetConfKey("player_limit").(int)
|
|
|
|
if !ok {
|
|
|
|
maxPeers = -1
|
|
|
|
}
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
if GetPeerCount() >= maxPeers && maxPeers > -1 {
|
|
|
|
data := []byte{
|
|
|
|
uint8(0x00), uint8(ToClientAccessDenied),
|
|
|
|
uint8(AccessDeniedTooManyUsers), uint8(0x00), uint8(0x00), uint8(0x00), uint8(0x00),
|
2021-01-06 05:42:55 -08:00
|
|
|
}
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
_, err := clt.Send(rudp.Pkt{Data: data})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2021-01-06 05:42:55 -08:00
|
|
|
}
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
return nil, ErrPlayerLimitReached
|
2021-01-05 11:34:35 -08:00
|
|
|
}
|
2021-01-09 03:26:30 -08:00
|
|
|
|
2021-01-18 13:08:22 -08:00
|
|
|
return clt, nil
|
2021-01-05 11:34:35 -08:00
|
|
|
}
|
2021-01-06 14:39:54 -08:00
|
|
|
|
2021-01-17 12:43:23 -08:00
|
|
|
// GetPeerByUsername returns the Peer that is using name for
|
|
|
|
// authentication
|
|
|
|
func (l *Listener) GetPeerByUsername(name string) *Peer {
|
2021-01-18 13:08:22 -08:00
|
|
|
l.mu.RLock()
|
|
|
|
defer l.mu.RUnlock()
|
|
|
|
|
|
|
|
for peer := range l.peers {
|
|
|
|
if string(peer.username) == name {
|
|
|
|
return peer
|
2021-01-14 10:06:40 -08:00
|
|
|
}
|
|
|
|
}
|
2021-01-18 13:08:22 -08:00
|
|
|
|
2021-01-14 10:06:40 -08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-01-17 12:43:23 -08:00
|
|
|
// GetPeers returns an array containing all connected client Peers
|
2021-01-14 10:06:40 -08:00
|
|
|
func (l *Listener) GetPeers() []*Peer {
|
2021-01-18 13:08:22 -08:00
|
|
|
l.mu.RLock()
|
|
|
|
defer l.mu.RUnlock()
|
2021-01-14 10:06:40 -08:00
|
|
|
|
|
|
|
var r []*Peer
|
2021-01-18 13:08:22 -08:00
|
|
|
for p := range l.peers {
|
|
|
|
r = append(r, p)
|
2021-01-14 10:06:40 -08:00
|
|
|
}
|
|
|
|
return r
|
2021-01-06 14:39:54 -08:00
|
|
|
}
|
|
|
|
|
2021-01-17 12:43:23 -08:00
|
|
|
// SetListener is used to make a listener available globally
|
|
|
|
// This can only be done once
|
2021-01-06 14:39:54 -08:00
|
|
|
func SetListener(l *Listener) {
|
2021-01-17 12:43:23 -08:00
|
|
|
if listener == nil {
|
|
|
|
listener = l
|
|
|
|
}
|
2021-01-06 14:39:54 -08:00
|
|
|
}
|
|
|
|
|
2021-01-17 12:43:23 -08:00
|
|
|
// GetListener returns the global listener
|
2021-01-06 14:39:54 -08:00
|
|
|
func GetListener() *Listener {
|
|
|
|
return listener
|
|
|
|
}
|