gottdad/main.go

241 lines
6.6 KiB
Go
Raw Normal View History

2020-06-14 16:27:57 +02:00
package main
import (
2020-06-14 16:32:21 +02:00
"bufio"
2020-06-14 22:39:17 +02:00
"encoding/binary"
2020-06-14 16:27:57 +02:00
"net"
2020-06-21 14:57:44 +02:00
"regexp"
2020-06-14 16:27:57 +02:00
)
2020-06-21 16:10:45 +02:00
type Client struct {
ClientID uint32
Name string
Address string
CompanyID uint8
}
2020-06-14 16:27:57 +02:00
func main() {
2020-06-15 15:04:55 +02:00
var (
2020-06-21 16:10:45 +02:00
clients map[uint32]Client
2020-06-21 15:20:36 +02:00
paused bool = true
forcePaused bool = true
2020-06-15 15:04:55 +02:00
)
2020-06-14 16:32:21 +02:00
conn, err := net.Dial("tcp", "poop.siteop.biz:3977")
2020-06-14 16:27:57 +02:00
failError(err, "net.Dial")
logInfoDebug("Connected to poop.siteop.biz:3977")
2020-06-14 16:44:40 +02:00
//send auth
p := PacketAdminJoin{
2020-06-14 16:48:00 +02:00
Packet: Packet{PType: AdminPacketAdminJoin},
2020-06-14 17:13:25 +02:00
Password: "plop",
AppName: "gottdad",
AppVersion: "alpha",
2020-06-14 16:44:40 +02:00
}
_, err = conn.Write(p.Bytes())
failError(err, "conn.Write")
2020-06-14 17:38:25 +02:00
logInfoDebug("Authentication sent (%v)", p.Bytes())
2020-06-14 22:16:39 +02:00
r := bufio.NewReader(conn)
2020-06-14 22:39:17 +02:00
b := make([]byte, 0xFFFF)
read := 0
n := 0
2020-06-14 16:44:40 +02:00
2020-06-14 22:39:17 +02:00
for {
p := Packet{}
2020-06-14 22:05:45 +02:00
for {
2020-06-14 22:39:17 +02:00
if read >= 3 {
2020-06-15 12:48:31 +02:00
//logInfoDebug("Packet read")
2020-06-14 22:05:45 +02:00
break
}
2020-06-14 22:46:23 +02:00
n, err = r.Read(b[read:])
logErrorDebug(err, "r.Read")
read += n
2020-06-15 12:48:31 +02:00
//logInfoDebug("Waiting for packet, read %d bytes.", read)
2020-06-14 22:46:23 +02:00
2020-06-14 22:39:17 +02:00
}
p.PLength = binary.LittleEndian.Uint16(b[0:])
p.PType = b[2]
if p.PLength <= 3 {
logInfoAlert("Wrong packet length")
break
}
2020-06-14 22:20:19 +02:00
2020-06-15 12:48:31 +02:00
//logInfoDebug("Waiting for packet data : len : %d / type : %d", p.PLength, p.PType)
2020-06-14 22:44:39 +02:00
2020-06-14 22:39:17 +02:00
for {
2020-06-14 22:45:52 +02:00
if read >= int(p.PLength) {
2020-06-15 12:48:44 +02:00
//logInfoDebug("Data read")
2020-06-14 22:45:52 +02:00
break
}
2020-06-14 22:39:17 +02:00
n, err = r.Read(b[read:])
logErrorDebug(err, "r.Read")
read += n
2020-06-15 12:48:31 +02:00
//logInfoDebug("Waiting for data, read %d/%d bytes.", read, p.PLength)
2020-06-14 22:45:52 +02:00
2020-06-14 22:20:19 +02:00
}
2020-06-14 22:39:17 +02:00
2020-06-15 12:42:02 +02:00
switch p.PType {
case AdminPacketServerProtocol:
2020-06-15 15:04:55 +02:00
sp := PacketServerProtocol{
2020-06-15 12:42:02 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerProtocol :\n- ProtocolVersion: %v\n- FurtherData: %v\n- UpdatePacketType: %v\n- FrequenciesAllowed: %b", sp.ProtocolVersion, sp.FurtherData, sp.UpdatePacketType, sp.FrequenciesAllowed)
case AdminPacketServerWelcome:
2020-06-15 15:04:55 +02:00
sp := PacketServerWelcome{
2020-06-15 12:42:02 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerWelcome :\n- ServerName: %v\n- OpenTTDVersion: %v\n- Dedicated: %v\n- MapSeed: %x\n- MapLandscape: %v\n- MapStartDate: %v\n- Size: %v x %v", sp.ServerName, sp.OpenTTDVersion, sp.Dedicated, sp.MapSeed, sp.MapLandscape, sp.MapStartDate, sp.MapX, sp.MapY)
px := PacketAdminUpdateFrequency{
Packet: Packet{PType: AdminPacketAdminUpdateFrequency},
UpdateType: AdminUpdateDate,
UpdateFrequency: AdminFrequencyDaily,
}
2020-06-15 12:42:27 +02:00
_, err = conn.Write(px.Bytes())
2020-06-15 12:50:09 +02:00
2020-06-15 12:50:20 +02:00
px = PacketAdminUpdateFrequency{
2020-06-15 12:50:09 +02:00
Packet: Packet{PType: AdminPacketAdminUpdateFrequency},
UpdateType: AdminUpdateClientInfo,
UpdateFrequency: AdminFrequencyAutomatic,
}
_, err = conn.Write(px.Bytes())
2020-06-15 14:28:42 +02:00
px = PacketAdminUpdateFrequency{
Packet: Packet{PType: AdminPacketAdminUpdateFrequency},
UpdateType: AdminUpdateChat,
UpdateFrequency: AdminFrequencyAutomatic,
}
_, err = conn.Write(px.Bytes())
px = PacketAdminUpdateFrequency{
Packet: Packet{PType: AdminPacketAdminUpdateFrequency},
UpdateType: AdminUpdateConsole,
UpdateFrequency: AdminFrequencyAutomatic,
}
_, err = conn.Write(px.Bytes())
2020-06-15 12:47:05 +02:00
case AdminPacketServerDate:
2020-06-15 15:04:55 +02:00
sp := PacketServerDate{
2020-06-15 12:46:43 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerDate :\n- Date: %d", sp.Date)
2020-06-15 15:04:55 +02:00
paused = false
2020-06-15 14:14:24 +02:00
case AdminPacketServerClientJoin:
2020-06-15 15:04:55 +02:00
sp := PacketServerClientJoin{
2020-06-15 14:14:24 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerClientJoin :\n- ClientID: %d", sp.ClientID)
case AdminPacketServerClientInfo:
2020-06-15 15:04:55 +02:00
sp := PacketServerClientInfo{
2020-06-15 14:14:24 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
2020-06-15 14:16:43 +02:00
logInfoDebug("AdminPacketServerClientInfo :\n- ClientID: %d\n- Address: %s\n- Name: %s\n- Lang: %d\n- Date: %d\n- CompanyID: %d", sp.ClientID, sp.Address, sp.Name, sp.Lang, sp.Date, sp.CompanyID)
2020-06-21 16:10:45 +02:00
clt := Client{
ClientID: sp.ClientID,
Name: sp.Name,
Address: sp.Address,
CompanyID: sp.CompanyID,
}
clients[sp.ClientID] = clt
logInfoDebug("Clients : %v", clients)
2020-06-15 14:14:24 +02:00
case AdminPacketServerClientError:
2020-06-15 15:04:55 +02:00
sp := PacketServerClientError{
2020-06-15 14:14:24 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerClientError :\n- ClientID: %d\n- ErrorID: %d", sp.ClientID, sp.ErrorID)
2020-06-15 14:28:42 +02:00
case AdminPacketServerClientQuit:
2020-06-15 15:04:55 +02:00
sp := PacketServerClientQuit{
2020-06-15 14:28:42 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
2020-06-15 14:44:52 +02:00
logInfoDebug("AdminPacketServerClientQuit :\n- ClientID: %d", sp.ClientID)
2020-06-21 16:10:45 +02:00
delete(clients, sp.ClientID)
2020-06-15 14:44:52 +02:00
case AdminPacketServerChat:
2020-06-15 15:04:55 +02:00
sp := PacketServerChat{
2020-06-15 14:44:52 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerChat :\n- ActionID: %d\n- DestinationID: %d\n- ClientID: %d\n- Message: %s\n- Amount: %d", sp.ActionID, sp.DestinationID, sp.ClientID, sp.Message, sp.Amount)
2020-06-21 15:33:53 +02:00
if sp.Message == "!unpause" {
logInfoDebug("AdminPacketServerChat : Unpausing")
forcePaused = false
}
2020-06-21 15:34:07 +02:00
if sp.Message == "!pause" {
2020-06-21 15:33:53 +02:00
logInfoDebug("AdminPacketServerChat : Pausing")
forcePaused = true
}
2020-06-15 14:44:52 +02:00
case AdminPacketServerConsole:
2020-06-15 15:04:55 +02:00
sp := PacketServerConsole{
2020-06-15 14:44:52 +02:00
Packet: p,
}
sp.Read(b[:p.PLength])
2020-06-21 15:26:56 +02:00
ok, err := regexp.MatchString("\\[udp\\] queried from .*", sp.Text)
logErrorDebug(err, "queried from")
2020-06-21 15:20:36 +02:00
if sp.Origin != "net" || ok == false {
2020-06-21 15:22:09 +02:00
logInfoDebug("AdminPacketServerConsole :\n- Origin: %q\n- Text: %s", sp.Origin, sp.Text)
2020-06-21 14:57:15 +02:00
}
2020-06-21 15:47:37 +02:00
case AdminPacketServerRCon:
sp := PacketServerRCon{
Packet: p,
}
sp.Read(b[:p.PLength])
2020-06-21 15:48:01 +02:00
logInfoDebug("AdminPacketServerRCon :\n- ColorID: %d\n- Output: %s", sp.ColorID, sp.Output)
2020-06-15 15:07:52 +02:00
case AdminPacketServerRConEnd:
sp := PacketServerRConEnd{
Packet: p,
}
sp.Read(b[:p.PLength])
logInfoDebug("AdminPacketServerRConEnd :\n- Command: %s", sp.Command)
2020-06-15 12:42:02 +02:00
default:
logInfoDebug("Packet fully read : len : %d / type : %d", p.PLength, p.PType)
}
2020-06-14 23:31:31 +02:00
2020-06-14 22:39:17 +02:00
c := make([]byte, 0xFFFF)
copy(c, b[p.PLength:])
b = c
2020-06-14 22:42:34 +02:00
read -= int(p.PLength)
2020-06-15 15:04:55 +02:00
2020-06-21 16:10:45 +02:00
if len(clients) == 0 {
2020-06-21 15:37:17 +02:00
px := PacketAdminRCon{
Packet: Packet{PType: AdminPacketAdminRCon},
Command: "clients",
}
_, err = conn.Write(px.Bytes())
}
2020-06-21 15:20:36 +02:00
if !paused && forcePaused {
paused = false
px := PacketAdminRCon{
Packet: Packet{PType: AdminPacketAdminRCon},
Command: "pause",
}
_, err = conn.Write(px.Bytes())
}
2020-06-21 16:10:45 +02:00
if paused && !forcePaused && len(clients) > 1 { // server is client #1
2020-06-15 15:04:55 +02:00
paused = false
2020-06-15 15:05:12 +02:00
px := PacketAdminRCon{
2020-06-15 15:04:55 +02:00
Packet: Packet{PType: AdminPacketAdminRCon},
Command: "unpause",
}
_, err = conn.Write(px.Bytes())
}
2020-06-21 16:10:45 +02:00
if !paused && len(clients) == 1 { // server is client #1
2020-06-15 15:04:55 +02:00
paused = true
2020-06-15 15:05:12 +02:00
px := PacketAdminRCon{
2020-06-15 15:04:55 +02:00
Packet: Packet{PType: AdminPacketAdminRCon},
Command: "pause",
}
_, err = conn.Write(px.Bytes())
}
2020-06-14 22:39:17 +02:00
}
2020-06-14 16:27:57 +02:00
}