Admin Else
1001c809e5
- RegisterUntilLow now panics if startIndex > endIndex (programmer must address version incompatibility explicitly, not silently skip) - Replace per-version RegisterCritical handlers with dual-range RegisterUntil: modern range (v1_21_8->latest) + legacy range (v1_8->1.12.2) - Remove v1_12_2 handler duplicates (covered by upward range from v1_8) - Move legacy chat handling into chat.Module.SendMessage (protocol<764 uses v1_8.PlayToServerPacketChat instead of v1_21_11) - Skip chat signing init for pre-1.19 versions
115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/admin-else/strom/mc/event"
|
|
"github.com/admin-else/strom/mc/proto"
|
|
"github.com/admin-else/strom/mc/proto_base"
|
|
"github.com/admin-else/strom/mc/proto_generated/v1_21_8"
|
|
"github.com/admin-else/strom/mc/proto_generated/v1_8"
|
|
"github.com/admin-else/strom/mc/server"
|
|
)
|
|
|
|
type StatusClient struct {
|
|
*proto.Conn
|
|
Status string
|
|
PingSendTime, PingReceiveTime time.Time
|
|
DoPingRoundTripTime bool
|
|
}
|
|
|
|
func (s *StatusClient) OnStatus(p *v1_21_8.StatusToClientPacketServerInfo) (err error) {
|
|
s.Status = p.Response
|
|
s.PingSendTime = time.Now()
|
|
if s.DoPingRoundTripTime {
|
|
err = s.Send(&v1_21_8.StatusToServerPacketPing{Time: s.PingSendTime.UnixMilli()})
|
|
} else {
|
|
err = event.HandlerDoneErr{}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *StatusClient) OnPong(p *v1_21_8.StatusToClientPacketPing) (err error) {
|
|
s.PingReceiveTime = time.Now()
|
|
err = event.HandlerDoneErr{}
|
|
return
|
|
}
|
|
|
|
func (s *StatusClient) OnStatusV1_8(p *v1_8.StatusToClientPacketServerInfo) (err error) {
|
|
s.Status = p.Response
|
|
s.PingSendTime = time.Now()
|
|
if s.DoPingRoundTripTime {
|
|
err = s.Send(&v1_21_8.StatusToServerPacketPing{Time: s.PingSendTime.UnixMilli()})
|
|
} else {
|
|
err = event.HandlerDoneErr{}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *StatusClient) OnPongV1_8(p *v1_8.StatusToClientPacketPing) (err error) {
|
|
s.PingReceiveTime = time.Now()
|
|
err = event.HandlerDoneErr{}
|
|
return
|
|
}
|
|
|
|
// StatusRaw returns a StatusClient that is not connected to a server.
|
|
// So ignore the resource leak warning. And maybe attach a warn ignore comment.
|
|
// It does not resolve SRV records.
|
|
func StatusRaw(ctx context.Context, addr string) (s *StatusClient, err error) {
|
|
return StatusRawWithVersion(ctx, addr, "1.21.8")
|
|
}
|
|
|
|
// StatusRawWithVersion is like StatusRaw but uses the given Minecraft version for the handshake.
|
|
func StatusRawWithVersion(ctx context.Context, addr, version string) (s *StatusClient, err error) {
|
|
c, err := ConnectVersionLess(ctx, addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer c.Close()
|
|
err = c.SetVersion(version)
|
|
if err != nil {
|
|
return
|
|
}
|
|
s = &StatusClient{
|
|
Conn: c,
|
|
}
|
|
s.RegisterUntil("26.2", s.OnStatus, s.OnPong)
|
|
s.RegisterUntil("1.12.2", s.OnStatusV1_8, s.OnPongV1_8)
|
|
|
|
p, err := MakeHandshakePacketAddr(s.Conn, proto_base.Status, addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = s.Send(p)
|
|
if err != nil {
|
|
return
|
|
}
|
|
s.SetState(proto_base.Status)
|
|
err = s.Send(&v1_21_8.StatusToServerPacketPingStart{})
|
|
|
|
err = s.StartConn()
|
|
return
|
|
}
|
|
|
|
// StatusNoDns is like Status but does not resolve SRV records.
|
|
func StatusNoDns(ctx context.Context, addr string) (status server.StatusResponse, err error) {
|
|
s, err := StatusRaw(ctx, addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer s.Close()
|
|
err = json.Unmarshal([]byte(s.Status), &status)
|
|
return
|
|
}
|
|
|
|
// Status resolves the status of the minecraft server at the given address.
|
|
func Status(ctx context.Context, addr string) (status server.StatusResponse, err error) {
|
|
addr, _, _, err = DoDNSChecked(ctx, addr, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return StatusNoDns(ctx, addr)
|
|
}
|