yess

2025-12-19 17:26:58 +01:00
parent cb38382a07
commit d7df6b25ef
20 changed files with 401 additions and 167 deletions
@@ -0,0 +1,23 @@
+package client
+
+import (
+	"net"
+
+	"github.com/admin-else/strom/proto"
+	"github.com/admin-else/strom/proto_base"
+)
+
+func Connect(addr string) (ret *proto.Conn, err error) {
+	ret = &proto.Conn{}
+	ret.Version = "1.21.8"
+	ret.State = proto_base.Handshaking
+	ret.CompressionThreshold = -1
+	ret.Actor = proto_base.Client
+	ret.Conn, err = net.Dial("tcp", addr)
+	if err != nil {
+		return
+	}
+	ret.R = ret.Conn
+	ret.W = ret.Conn
+	return
+}
@@ -0,0 +1,52 @@
+package modules
+
+import (
+	"errors"
+	"fmt"
+
+	"github.com/admin-else/strom/event"
+	"github.com/admin-else/strom/proto"
+	"github.com/admin-else/strom/proto_base"
+	"github.com/admin-else/strom/proto_generated/v1_21_8"
+)
+
+type ConfigIgnorer struct {
+	*proto.Conn
+}
+
+func (c *ConfigIgnorer) Default(event any) (err error) {
+	fmt.Printf("%#v\n", event)
+	return
+}
+
+func (c *ConfigIgnorer) OnKnownPacks(packet v1_21_8.PacketCommonSelectKnownPacks) (err error) {
+	return c.Send(packet)
+}
+
+func (c *ConfigIgnorer) OnFinish(_ v1_21_8.ConfigurationToClientPacketFinishConfiguration) (err error) {
+	err = c.Send(v1_21_8.ConfigurationToServerPacketFinishConfiguration{})
+	if err != nil {
+		return
+	}
+	c.State = proto_base.Play
+	err = event.HandlerDone
+	return
+}
+
+func (c *ConfigIgnorer) OnPing(packet v1_21_8.ConfigurationToClientPacketPing) (err error) {
+	err = c.Send(v1_21_8.ConfigurationToServerPacketPong(packet))
+	return
+}
+
+func (c *ConfigIgnorer) OnKeepAlive(packet v1_21_8.ConfigurationToClientPacketKeepAlive) (err error) {
+	err = c.Send(v1_21_8.ConfigurationToServerPacketKeepAlive(packet))
+	return
+}
+
+func IgnoreConfig(c *proto.Conn) (err error) {
+	err = c.Start(&ConfigIgnorer{c})
+	if err != nil {
+		err = errors.Join(err, errors.New("failed to ignore config"))
+	}
+	return
+}
@@ -0,0 +1,164 @@
+package modules
+
+import (
+	"crypto/aes"
+	"crypto/cipher"
+	"crypto/rand"
+	"crypto/rsa"
+	"crypto/x509"
+	"errors"
+	"fmt"
+	"net"
+	"strconv"
+
+	"github.com/admin-else/strom/api"
+	"github.com/admin-else/strom/client"
+	"github.com/admin-else/strom/crypto"
+	"github.com/admin-else/strom/event"
+	"github.com/admin-else/strom/proto"
+	"github.com/admin-else/strom/proto_base"
+
+	"github.com/admin-else/strom/data"
+	"github.com/admin-else/strom/proto_generated/v1_21_8"
+)
+
+type LoginClient struct {
+	*proto.Conn
+	Account *api.Account
+
+	GivenAccount v1_21_8.LoginToClientPacketSuccess
+}
+
+func (s *LoginClient) Default(event any) (err error) {
+	err = fmt.Errorf("unexpected event: %#v", event)
+	return
+}
+
+func (s *LoginClient) OnCycle(_ event.OnLoopCycle) (err error) {
+	return
+}
+
+func (s *LoginClient) OnStart(_ event.OnStart) (err error) {
+	host, portStr, err := net.SplitHostPort(s.RemoteAddr().String())
+	if err != nil {
+		return
+	}
+	versionData, err := data.LookUpProtocolVersionByName(s.Version)
+	if err != nil {
+		return
+	}
+	port, err := strconv.Atoi(portStr)
+	if err != nil {
+		return
+	}
+	err = s.Send(v1_21_8.HandshakingToServerPacketSetProtocol{
+		ProtocolVersion: proto_base.VarInt(versionData.Version),
+		ServerHost:      host,
+		ServerPort:      uint16(port),
+		NextState:       proto_base.VarInt(proto_base.Login),
+	})
+	if err != nil {
+		return
+	}
+	s.State = proto_base.Login
+	err = s.Send(v1_21_8.LoginToServerPacketLoginStart{Username: s.Account.Name, PlayerUUID: s.Account.Uuid})
+	return
+}
+
+func (s *LoginClient) OnCompress(compress v1_21_8.LoginToClientPacketCompress) (err error) {
+	s.CompressionThreshold = int32(compress.Threshold)
+	return
+}
+
+func (s *LoginClient) OnEncrypt(packet v1_21_8.LoginToClientPacketEncryptionBegin) (err error) {
+	sharedSecret := make([]byte, 16)
+	_, _ = rand.Read(sharedSecret) //never fails
+
+	if packet.ShouldAuthenticate {
+		if s.Account.Ygg == "" {
+			err = fmt.Errorf("the account has no token so we cant join online servers")
+			return
+		}
+		serverId := crypto.AuthDigest([]byte(packet.ServerId), sharedSecret, packet.PublicKey)
+		err = s.Account.JoinServer(serverId)
+		if err != nil {
+			return
+		}
+	}
+
+	pubAny, err := x509.ParsePKIXPublicKey(packet.PublicKey)
+	if err != nil {
+		return
+	}
+	pub, ok := pubAny.(*rsa.PublicKey)
+	if !ok {
+		err = fmt.Errorf("public key is not rsa")
+		return
+	}
+	verifyTokenEnc, err := rsa.EncryptPKCS1v15(rand.Reader, pub, packet.VerifyToken)
+	if err != nil {
+		return
+	}
+
+	sharedSecretEnc, err := rsa.EncryptPKCS1v15(rand.Reader, pub, sharedSecret)
+	if err != nil {
+		return
+	}
+	err = s.Send(v1_21_8.LoginToServerPacketEncryptionBegin{SharedSecret: sharedSecretEnc, VerifyToken: verifyTokenEnc})
+	if err != nil {
+		return
+	}
+
+	var b cipher.Block
+	b, err = aes.NewCipher(sharedSecret)
+	if err != nil {
+		return
+	}
+	s.R = cipher.StreamReader{
+		S: crypto.NewCFB8Decrypt(b, sharedSecret),
+		R: s.Conn,
+	}
+	s.W = cipher.StreamWriter{
+		S: crypto.NewCFB8Encrypt(b, sharedSecret),
+		W: s.Conn,
+	}
+	return
+
+}
+
+func (s *LoginClient) OnSuccess(success v1_21_8.LoginToClientPacketSuccess) (err error) {
+	s.GivenAccount = success
+	err = s.Send(v1_21_8.LoginToServerPacketLoginAcknowledged{})
+	if err != nil {
+		return
+	}
+	s.State = proto_base.Configuration
+	err = event.HandlerDone
+	return
+}
+
+func Login(c *proto.Conn, account *api.Account) (err error) {
+	err = c.Start(&LoginClient{
+		Conn:    c,
+		Account: account,
+	})
+	if err != nil {
+		err = errors.Join(err, errors.New("failed to login"))
+	}
+	return
+}
+
+func ConnectAndLogin(connectTo string, account *api.Account) (c *proto.Conn, err error) {
+	c, err = client.Connect(connectTo)
+	if err != nil {
+		return
+	}
+	err = c.Start(&LoginClient{
+		Conn:    c,
+		Account: account,
+	})
+	if err != nil {
+		err = errors.Join(err, errors.New("failed to login"))
+	}
+	return
+}
@@ -0,0 +1,54 @@
+package modules
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/admin-else/strom/event"
+	"github.com/admin-else/strom/proto"
+	"github.com/admin-else/strom/proto_base"
+
+	"github.com/admin-else/strom/data"
+	"github.com/admin-else/strom/proto_generated/v1_21_8"
+)
+
+type StatusClient struct {
+	*proto.Conn
+	Status string
+}
+
+func (s *StatusClient) Default(event any) (err error) {
+	err = fmt.Errorf("unexpected event: %v", event)
+	return
+}
+
+func (s *StatusClient) OnStart(_ event.OnStart) (err error) {
+	parts := strings.Split(s.RemoteAddr().String(), ":")
+	versionData, err := data.LookUpProtocolVersionByName(s.Version)
+	if err != nil {
+		return
+	}
+	port, err := strconv.Atoi(parts[1])
+	if err != nil {
+		return
+	}
+	err = s.Send(v1_21_8.HandshakingToServerPacketSetProtocol{
+		ProtocolVersion: proto_base.VarInt(versionData.Version),
+		ServerHost:      parts[0],
+		ServerPort:      uint16(port),
+		NextState:       proto_base.VarInt(proto_base.Status),
+	})
+	if err != nil {
+		return
+	}
+	s.State = proto_base.Status
+	err = s.Send(v1_21_8.StatusToServerPacketPingStart{})
+	return
+}
+
+func (s *StatusClient) OnStatus(p v1_21_8.StatusToClientPacketServerInfo) (err error) {
+	s.Status = p.Response
+	err = event.HandlerDone
+	return
+}
@@ -0,0 +1,63 @@
+package modules
+
+import "time"
+
+type IntervalTask struct {
+	F           func() error
+	LastTrigger time.Time
+	Interval    time.Duration
+}
+
+type OnceTask struct {
+	F       func() error
+	Trigger time.Time
+}
+
+type Timer struct {
+	Once     []OnceTask
+	Interval []IntervalTask
+	Active   bool
+}
+
+func (t *Timer) Tick() (err error) {
+	if !t.Active {
+		return
+	}
+	for i, task := range t.Interval {
+		if time.Since(task.LastTrigger) >= task.Interval {
+			task.LastTrigger = time.Now()
+			err = task.F()
+			if err != nil {
+				return
+			}
+		}
+		t.Interval[i] = task
+	}
+	var triggerLater []OnceTask
+	for _, task := range t.Once {
+		if task.Trigger.After(time.Now()) {
+			err = task.F()
+			if err != nil {
+				return
+			}
+		} else {
+			triggerLater = append(triggerLater, task)
+		}
+	}
+	t.Once = triggerLater
+	return
+}
+
+func (t *Timer) Every(interval time.Duration, f func() error) {
+	t.Interval = append(t.Interval, IntervalTask{
+		F:        f,
+		Interval: interval,
+	})
+}
+
+func (t *Timer) In(interval time.Duration, f func() error) {
+	t.Once = append(t.Once, OnceTask{
+		F:       f,
+		Trigger: time.Now().Add(interval),
+	})
+}