feat(protbot): add headless MHF protocol bot as cmd/protbot

Copy MHBridge into the Erupe module as cmd/protbot/ so it can be built, tested, and maintained alongside the server. The bot implements the full sign → entrance → channel login flow and supports lobby entry, chat, session setup, and quest enumeration. The conn/ package keeps its own Blowfish crypto primitives to avoid importing erupe-ce/config (which requires a config file at init).
2026-03-27 01:53:19 +01:00 · 2026-02-20 02:49:23 +01:00
parent 754b5a3bff
commit 0e84377e21
19 changed files with 2106 additions and 0 deletions
--- a/cmd/protbot/main.go
+++ b/cmd/protbot/main.go
@@ -0,0 +1,154 @@
+// protbot is a headless MHF protocol bot for testing Erupe server instances.
+//
+// Usage:
+//
+//	protbot --sign-addr 127.0.0.1:53312 --user test --pass test --action login
+//	protbot --sign-addr 127.0.0.1:53312 --user test --pass test --action lobby
+//	protbot --sign-addr 127.0.0.1:53312 --user test --pass test --action session
+//	protbot --sign-addr 127.0.0.1:53312 --user test --pass test --action chat --message "Hello"
+//	protbot --sign-addr 127.0.0.1:53312 --user test --pass test --action quests
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"erupe-ce/cmd/protbot/scenario"
+)
+
+func main() {
+	signAddr := flag.String("sign-addr", "127.0.0.1:53312", "Sign server address (host:port)")
+	user := flag.String("user", "", "Username")
+	pass := flag.String("pass", "", "Password")
+	action := flag.String("action", "login", "Action to perform: login, lobby, session, chat, quests")
+	message := flag.String("message", "", "Chat message to send (used with --action chat)")
+	flag.Parse()
+
+	if *user == "" || *pass == "" {
+		fmt.Fprintln(os.Stderr, "error: --user and --pass are required")
+		flag.Usage()
+		os.Exit(1)
+	}
+
+	switch *action {
+	case "login":
+		result, err := scenario.Login(*signAddr, *user, *pass)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "login failed: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Println("[done] Login successful!")
+		result.Channel.Close()
+
+	case "lobby":
+		result, err := scenario.Login(*signAddr, *user, *pass)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "login failed: %v\n", err)
+			os.Exit(1)
+		}
+		if err := scenario.EnterLobby(result.Channel); err != nil {
+			fmt.Fprintf(os.Stderr, "enter lobby failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+		fmt.Println("[done] Lobby entry successful!")
+		result.Channel.Close()
+
+	case "session":
+		result, err := scenario.Login(*signAddr, *user, *pass)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "login failed: %v\n", err)
+			os.Exit(1)
+		}
+		charID := result.Sign.CharIDs[0]
+		if _, err := scenario.SetupSession(result.Channel, charID); err != nil {
+			fmt.Fprintf(os.Stderr, "session setup failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+		if err := scenario.EnterLobby(result.Channel); err != nil {
+			fmt.Fprintf(os.Stderr, "enter lobby failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+		fmt.Println("[session] Connected. Press Ctrl+C to disconnect.")
+		waitForSignal()
+		scenario.Logout(result.Channel)
+
+	case "chat":
+		result, err := scenario.Login(*signAddr, *user, *pass)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "login failed: %v\n", err)
+			os.Exit(1)
+		}
+		charID := result.Sign.CharIDs[0]
+		if _, err := scenario.SetupSession(result.Channel, charID); err != nil {
+			fmt.Fprintf(os.Stderr, "session setup failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+		if err := scenario.EnterLobby(result.Channel); err != nil {
+			fmt.Fprintf(os.Stderr, "enter lobby failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+
+		// Register chat listener.
+		scenario.ListenChat(result.Channel, func(msg scenario.ChatMessage) {
+			fmt.Printf("[chat] <%s> (type=%d): %s\n", msg.SenderName, msg.ChatType, msg.Message)
+		})
+
+		// Send a message if provided.
+		if *message != "" {
+			if err := scenario.SendChat(result.Channel, 0x03, 1, *message, *user); err != nil {
+				fmt.Fprintf(os.Stderr, "send chat failed: %v\n", err)
+			}
+		}
+
+		fmt.Println("[chat] Listening for chat messages. Press Ctrl+C to disconnect.")
+		waitForSignal()
+		scenario.Logout(result.Channel)
+
+	case "quests":
+		result, err := scenario.Login(*signAddr, *user, *pass)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "login failed: %v\n", err)
+			os.Exit(1)
+		}
+		charID := result.Sign.CharIDs[0]
+		if _, err := scenario.SetupSession(result.Channel, charID); err != nil {
+			fmt.Fprintf(os.Stderr, "session setup failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+		if err := scenario.EnterLobby(result.Channel); err != nil {
+			fmt.Fprintf(os.Stderr, "enter lobby failed: %v\n", err)
+			result.Channel.Close()
+			os.Exit(1)
+		}
+
+		data, err := scenario.EnumerateQuests(result.Channel, 0, 0)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "enumerate quests failed: %v\n", err)
+			scenario.Logout(result.Channel)
+			os.Exit(1)
+		}
+		fmt.Printf("[quests] Received %d bytes of quest data\n", len(data))
+		scenario.Logout(result.Channel)
+
+	default:
+		fmt.Fprintf(os.Stderr, "unknown action: %s (supported: login, lobby, session, chat, quests)\n", *action)
+		os.Exit(1)
+	}
+}
+
+// waitForSignal blocks until SIGINT or SIGTERM is received.
+func waitForSignal() {
+	sig := make(chan os.Signal, 1)
+	signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
+	<-sig
+	fmt.Println("\n[signal] Shutting down...")
+}