mirror of
https://github.com/onyx-and-iris/q3rcon.git
synced 2026-03-03 01:19:16 +00:00
Compare commits
No commits in common. "main" and "v0.4.0" have entirely different histories.
1
.gitignore
vendored
1
.gitignore
vendored
@ -24,7 +24,6 @@ go.work.sum
|
|||||||
|
|
||||||
# env file
|
# env file
|
||||||
.env
|
.env
|
||||||
.envrc
|
|
||||||
|
|
||||||
# Added by goreleaser init:
|
# Added by goreleaser init:
|
||||||
dist/
|
dist/
|
||||||
|
|||||||
21
CHANGELOG.md
21
CHANGELOG.md
@ -11,26 +11,7 @@ Before any major/minor/patch bump all unit tests will be run to verify they pass
|
|||||||
|
|
||||||
- [x]
|
- [x]
|
||||||
|
|
||||||
# [0.5.3] - 2026-02-26
|
# [0.4.0] - 2026-02-15
|
||||||
|
|
||||||
### Added
|
|
||||||
|
|
||||||
- Spinner for long running CLI commands.
|
|
||||||
|
|
||||||
# [0.5.1] - 2026-02-18
|
|
||||||
|
|
||||||
### Added
|
|
||||||
|
|
||||||
- CLI configuration can be managed through env vars, see [Environment Variables](https://github.com/onyx-and-iris/q3rcon?tab=readme-ov-file#environment-variables) under Configuration in README.
|
|
||||||
|
|
||||||
### Changed
|
|
||||||
|
|
||||||
- The CLI now supports `--long` and `-short` style flags. Several examples in README.
|
|
||||||
- `--help` output has been improved.
|
|
||||||
- Colour codes have been removed from CLI output. This makes the responses easier to read.
|
|
||||||
|
|
||||||
|
|
||||||
# [0.4.1] - 2026-02-15
|
|
||||||
|
|
||||||
### Added
|
### Added
|
||||||
|
|
||||||
|
|||||||
58
README.md
58
README.md
@ -17,15 +17,9 @@ Quake3 Rcon works by firing UDP packets to the game server port, responses may b
|
|||||||
|
|
||||||
Rcon itself is insecure and each packet includes the password so I don't suggest using it remotely. If you have direct access to the server then SSH in first, then use this tool locally.
|
Rcon itself is insecure and each packet includes the password so I don't suggest using it remotely. If you have direct access to the server then SSH in first, then use this tool locally.
|
||||||
|
|
||||||
---
|
## Use
|
||||||
|
|
||||||
## Package
|
`go get github.com/onyx-and-iris/q3rcon`
|
||||||
|
|
||||||
#### Use
|
|
||||||
|
|
||||||
```console
|
|
||||||
go get github.com/onyx-and-iris/q3rcon
|
|
||||||
```
|
|
||||||
|
|
||||||
```go
|
```go
|
||||||
package main
|
package main
|
||||||
@ -97,24 +91,12 @@ rcon, err := q3rcon.New(
|
|||||||
q3rcon.WithTimeouts(timeouts))
|
q3rcon.WithTimeouts(timeouts))
|
||||||
```
|
```
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## Command line
|
## Command line
|
||||||
|
|
||||||
### Install
|
Pass `host`, `port` and `rconpass` as flags, for example:
|
||||||
|
|
||||||
```console
|
|
||||||
go install github.com/onyx-and-iris/q3rcon/cmd/q3rcon@latest
|
|
||||||
```
|
```
|
||||||
|
q3rcon -h=localhost -p=30000 -r="rconpassword" "mapname"
|
||||||
### Configuration
|
|
||||||
|
|
||||||
#### Flags
|
|
||||||
|
|
||||||
Pass `--host`, `--port` and `--rconpass` as flags, for example:
|
|
||||||
|
|
||||||
```console
|
|
||||||
q3rcon --host=localhost --port=30000 --rconpass="rconpassword" "mapname"
|
|
||||||
```
|
```
|
||||||
|
|
||||||
- `host` defaults to "localhost"
|
- `host` defaults to "localhost"
|
||||||
@ -123,41 +105,25 @@ q3rcon --host=localhost --port=30000 --rconpass="rconpassword" "mapname"
|
|||||||
|
|
||||||
Arguments following the flags will be sent as rcon commands. You may send multiple arguments.
|
Arguments following the flags will be sent as rcon commands. You may send multiple arguments.
|
||||||
|
|
||||||
#### Environment Variables
|
#### Interactive mode
|
||||||
|
|
||||||
example .envrc:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
export Q3RCON_HOST="localhost"
|
|
||||||
export Q3RCON_PORT=28960
|
|
||||||
export Q3RCON_RCONPASS="rconpassword"
|
|
||||||
```
|
|
||||||
|
|
||||||
### Interactive mode
|
|
||||||
|
|
||||||
Pass `interactive (-i shorthand)` flag to enable interactive mode, for example:
|
Pass `interactive (-i shorthand)` flag to enable interactive mode, for example:
|
||||||
|
|
||||||
```console
|
```bash
|
||||||
q3rcon -H=localhost -p=30000 -r="rconpassword" -i
|
q3rcon -h=localhost -p=30000 -r="rconpassword" -i
|
||||||
```
|
```
|
||||||
|
|
||||||
If interactive mode is enabled, any arguments sent on the command line will be ignored.
|
If interactive mode is enabled, any arguments sent on the command line will be ignored.
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## Your own implementation
|
## Your own implementation
|
||||||
|
|
||||||
The included CLI is a generic implementation, while it can be used out of the box you may find that some requests result in fragmented responses. The solution is to implement your own version, adjusting the timings with the functional options as detailed above.
|
The included CLI is a generic implementation, while it can be used out of the box you may find that some requests result in fragmented responses. The solution is to implement your own version, adjusting the timings with the functional options as detailed above. I could have increased the default timeouts but that would add unnecessary delay for most requests, so I decided to leave those details to the users of the package.
|
||||||
|
|
||||||
Since you can include the q3rcon package into your own CLI/package you can easily make your own modifications, for example, I added [colour to the terminal][status] and [tabulated some of the responses][mapname].
|
Since you can include the q3rcon package into your own package you can easily make your own modifications, for example, I added [colour to the terminal][status] and [reformatted some of the responses][mapname].
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## Logging
|
## Logging
|
||||||
|
|
||||||
The `--loglevel` flag allows you to control the verbosity of the application's logging output.
|
The `-loglevel` flag allows you to control the verbosity of the application's logging output.
|
||||||
|
|
||||||
Acceptable values for this flag are:
|
Acceptable values for this flag are:
|
||||||
|
|
||||||
@ -171,8 +137,8 @@ Acceptable values for this flag are:
|
|||||||
|
|
||||||
For example, to set the log level to `debug`, you can use:
|
For example, to set the log level to `debug`, you can use:
|
||||||
|
|
||||||
```console
|
```bash
|
||||||
q3rcon -H=localhost -p=28960 -r="rconpassword" -l=debug -i
|
q3rcon -p=28960 -r="rconpassword" -loglevel=debug -i
|
||||||
```
|
```
|
||||||
|
|
||||||
The default log level is `warn` if the flag is not specified.
|
The default log level is `warn` if the flag is not specified.
|
||||||
|
|||||||
10
Taskfile.yml
10
Taskfile.yml
@ -8,8 +8,8 @@ vars:
|
|||||||
WINDOWS: '{{.BIN_DIR}}/{{.PROGRAM}}_windows_amd64.exe'
|
WINDOWS: '{{.BIN_DIR}}/{{.PROGRAM}}_windows_amd64.exe'
|
||||||
LINUX: '{{.BIN_DIR}}/{{.PROGRAM}}_linux_amd64'
|
LINUX: '{{.BIN_DIR}}/{{.PROGRAM}}_linux_amd64'
|
||||||
MACOS: '{{.BIN_DIR}}/{{.PROGRAM}}_darwin_amd64'
|
MACOS: '{{.BIN_DIR}}/{{.PROGRAM}}_darwin_amd64'
|
||||||
VERSION:
|
GIT_COMMIT:
|
||||||
sh: 'git describe --tags $(git rev-list --tags --max-count=1)'
|
sh: git log -n 1 --format=%h
|
||||||
|
|
||||||
tasks:
|
tasks:
|
||||||
default:
|
default:
|
||||||
@ -39,19 +39,19 @@ tasks:
|
|||||||
build-windows:
|
build-windows:
|
||||||
desc: Build the q3rcon project for Windows
|
desc: Build the q3rcon project for Windows
|
||||||
cmds:
|
cmds:
|
||||||
- GOOS=windows GOARCH=amd64 go build -o {{.WINDOWS}} -ldflags="-X main.version={{.VERSION}}" ./cmd/{{.PROGRAM}}/
|
- GOOS=windows GOARCH=amd64 go build -o {{.WINDOWS}} -ldflags="-X main.Version={{.GIT_COMMIT}}" ./cmd/{{.PROGRAM}}/
|
||||||
internal: true
|
internal: true
|
||||||
|
|
||||||
build-linux:
|
build-linux:
|
||||||
desc: Build the q3rcon project for Linux
|
desc: Build the q3rcon project for Linux
|
||||||
cmds:
|
cmds:
|
||||||
- GOOS=linux GOARCH=amd64 go build -o {{.LINUX}} -ldflags="-X main.version={{.VERSION}}" ./cmd/{{.PROGRAM}}/
|
- GOOS=linux GOARCH=amd64 go build -o {{.LINUX}} -ldflags="-X main.Version={{.GIT_COMMIT}}" ./cmd/{{.PROGRAM}}/
|
||||||
internal: true
|
internal: true
|
||||||
|
|
||||||
build-macos:
|
build-macos:
|
||||||
desc: Build the q3rcon project for macOS
|
desc: Build the q3rcon project for macOS
|
||||||
cmds:
|
cmds:
|
||||||
- GOOS=darwin GOARCH=amd64 go build -o {{.MACOS}} -ldflags="-X main.version={{.VERSION}}" ./cmd/{{.PROGRAM}}/
|
- GOOS=darwin GOARCH=amd64 go build -o {{.MACOS}} -ldflags="-X main.Version={{.GIT_COMMIT}}" ./cmd/{{.PROGRAM}}/
|
||||||
internal: true
|
internal: true
|
||||||
|
|
||||||
test:
|
test:
|
||||||
|
|||||||
@ -2,54 +2,18 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"errors"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"regexp"
|
|
||||||
"runtime/debug"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/charmbracelet/log"
|
log "github.com/sirupsen/logrus"
|
||||||
"github.com/chelnak/ysmrr"
|
|
||||||
"github.com/peterbourgon/ff/v4"
|
|
||||||
"github.com/peterbourgon/ff/v4/ffhelp"
|
|
||||||
|
|
||||||
"github.com/onyx-and-iris/q3rcon"
|
"github.com/onyx-and-iris/q3rcon"
|
||||||
)
|
)
|
||||||
|
|
||||||
var version string // Version will be set at build time
|
|
||||||
|
|
||||||
type Flags struct {
|
|
||||||
Host string
|
|
||||||
Port int
|
|
||||||
Rconpass string
|
|
||||||
Interactive bool
|
|
||||||
LogLevel string
|
|
||||||
Version bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f Flags) Validate() error {
|
|
||||||
if f.Port < 1024 || f.Port > 65535 {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"invalid port value, got: (%d) expected: in range 1024-65535",
|
|
||||||
f.Port,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(f.Rconpass) < 8 {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"invalid rcon password, got: (%s) expected: at least 8 characters",
|
|
||||||
f.Rconpass,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var reColourCodes = regexp.MustCompile(`\^[0-9]`)
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
var exitCode int
|
var exitCode int
|
||||||
|
|
||||||
@ -70,131 +34,88 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type context struct {
|
|
||||||
client *q3rcon.Rcon
|
|
||||||
timeouts map[string]time.Duration
|
|
||||||
in io.Reader
|
|
||||||
sm ysmrr.SpinnerManager
|
|
||||||
}
|
|
||||||
|
|
||||||
// run executes the main logic of the application and returns a cleanup function and an error if any.
|
// run executes the main logic of the application and returns a cleanup function and an error if any.
|
||||||
func run() (func(), error) {
|
func run() (func(), error) {
|
||||||
var flags Flags
|
var (
|
||||||
|
host string
|
||||||
|
port int
|
||||||
|
rconpass string
|
||||||
|
interactive bool
|
||||||
|
loglevel string
|
||||||
|
)
|
||||||
|
|
||||||
fs := ff.NewFlagSet("q3rcon - A command-line RCON client for Q3 Rcon compatible game servers")
|
flag.StringVar(&host, "host", "localhost", "hostname of the gameserver")
|
||||||
fs.StringVar(&flags.Host, 'H', "host", "localhost", "hostname of the gameserver")
|
flag.StringVar(&host, "h", "localhost", "hostname of the gameserver (shorthand)")
|
||||||
fs.IntVar(
|
flag.IntVar(&port, "port", 28960, "port on which the gameserver resides, default is 28960")
|
||||||
&flags.Port,
|
flag.IntVar(
|
||||||
'p',
|
&port,
|
||||||
"port",
|
"p",
|
||||||
28960,
|
28960,
|
||||||
"port on which the gameserver resides, default is 28960",
|
"port on which the gameserver resides, default is 28960 (shorthand)",
|
||||||
)
|
)
|
||||||
fs.StringVar(
|
flag.StringVar(&rconpass, "rconpass", os.Getenv("RCON_PASS"), "rcon password of the gameserver")
|
||||||
&flags.Rconpass,
|
flag.StringVar(
|
||||||
'r',
|
&rconpass,
|
||||||
"rconpass",
|
"r",
|
||||||
"",
|
os.Getenv("RCON_PASS"),
|
||||||
"rcon password of the gameserver",
|
"rcon password of the gameserver (shorthand)",
|
||||||
)
|
)
|
||||||
|
|
||||||
fs.BoolVar(&flags.Interactive, 'i', "interactive", "run in interactive mode")
|
flag.BoolVar(&interactive, "interactive", false, "run in interactive mode")
|
||||||
fs.StringVar(
|
flag.BoolVar(&interactive, "i", false, "run in interactive mode")
|
||||||
&flags.LogLevel,
|
|
||||||
'l',
|
|
||||||
"loglevel",
|
|
||||||
"info",
|
|
||||||
"Log level (debug, info, warn, error, fatal, panic)",
|
|
||||||
)
|
|
||||||
fs.BoolVar(&flags.Version, 'v', "version", "print version information and exit")
|
|
||||||
|
|
||||||
err := ff.Parse(fs, os.Args[1:],
|
flag.StringVar(&loglevel, "loglevel", "warn", "log level")
|
||||||
ff.WithEnvVarPrefix("Q3RCON"),
|
flag.StringVar(&loglevel, "l", "warn", "log level (shorthand)")
|
||||||
)
|
|
||||||
switch {
|
|
||||||
case errors.Is(err, ff.ErrHelp):
|
|
||||||
fmt.Fprintf(os.Stderr, "%s\n", ffhelp.Flags(fs, "q3rcon [flags] <rcon commands>"))
|
|
||||||
return nil, nil
|
|
||||||
case err != nil:
|
|
||||||
return nil, fmt.Errorf("failed to parse flags: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if flags.Version {
|
flag.Parse()
|
||||||
fmt.Printf("q3rcon version: %s\n", versionFromBuild())
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := flags.Validate(); err != nil {
|
level, err := log.ParseLevel(loglevel)
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
level, err := log.ParseLevel(flags.LogLevel)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("invalid log level: %s", flags.LogLevel)
|
return nil, fmt.Errorf("invalid log level: %s", loglevel)
|
||||||
}
|
}
|
||||||
log.SetLevel(level)
|
log.SetLevel(level)
|
||||||
|
|
||||||
timeouts := map[string]time.Duration{
|
if port < 1024 || port > 65535 {
|
||||||
"map": time.Second,
|
return nil, fmt.Errorf("invalid port value, got: (%d) expected: in range 1024-65535", port)
|
||||||
"map_rotate": time.Second,
|
|
||||||
"map_restart": time.Second,
|
|
||||||
}
|
}
|
||||||
log.Debugf("using timeouts: %v", timeouts)
|
|
||||||
|
|
||||||
client, closer, err := connectRcon(flags.Host, flags.Port, flags.Rconpass, timeouts)
|
if len(rconpass) < 8 {
|
||||||
|
return nil, fmt.Errorf(
|
||||||
|
"invalid rcon password, got: (%s) expected: at least 8 characters",
|
||||||
|
rconpass,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
client, closer, err := connectRcon(host, port, rconpass)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to connect to rcon: %w", err)
|
return nil, fmt.Errorf("failed to connect to rcon: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
sm := ysmrr.NewSpinnerManager()
|
if interactive {
|
||||||
sm.AddSpinner("")
|
|
||||||
|
|
||||||
ctx := &context{
|
|
||||||
client: client,
|
|
||||||
timeouts: timeouts,
|
|
||||||
in: os.Stdin,
|
|
||||||
sm: sm,
|
|
||||||
}
|
|
||||||
|
|
||||||
if flags.Interactive {
|
|
||||||
fmt.Printf("Enter 'Q' to exit.\n>> ")
|
fmt.Printf("Enter 'Q' to exit.\n>> ")
|
||||||
err := interactiveMode(ctx)
|
err := interactiveMode(client, os.Stdin)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return closer, fmt.Errorf("interactive mode error: %w", err)
|
return closer, fmt.Errorf("interactive mode error: %w", err)
|
||||||
}
|
}
|
||||||
return closer, nil
|
return closer, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
commands := fs.GetArgs()
|
commands := flag.Args()
|
||||||
if len(commands) == 0 {
|
if len(commands) == 0 {
|
||||||
log.Debug("no commands provided, defaulting to 'status'")
|
log.Debug("no commands provided, defaulting to 'status'")
|
||||||
commands = append(commands, "status")
|
commands = append(commands, "status")
|
||||||
}
|
}
|
||||||
runCommands(ctx, commands)
|
runCommands(client, commands)
|
||||||
|
|
||||||
return closer, nil
|
return closer, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// versionFromBuild retrieves the version information from the build metadata.
|
func connectRcon(host string, port int, password string) (*q3rcon.Rcon, func(), error) {
|
||||||
func versionFromBuild() string {
|
client, err := q3rcon.New(host, port, password, q3rcon.WithTimeouts(map[string]time.Duration{
|
||||||
if version != "" {
|
"map": time.Second,
|
||||||
return version
|
"map_rotate": time.Second,
|
||||||
}
|
"map_restart": time.Second,
|
||||||
|
}))
|
||||||
info, ok := debug.ReadBuildInfo()
|
|
||||||
if !ok {
|
|
||||||
return "(unable to read version)"
|
|
||||||
}
|
|
||||||
return strings.Split(info.Main.Version, "-")[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
func connectRcon(
|
|
||||||
host string,
|
|
||||||
port int,
|
|
||||||
password string,
|
|
||||||
timeouts map[string]time.Duration,
|
|
||||||
) (*q3rcon.Rcon, func(), error) {
|
|
||||||
client, err := q3rcon.New(host, port, password, q3rcon.WithTimeouts(timeouts))
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
@ -208,19 +129,34 @@ func connectRcon(
|
|||||||
return client, closer, nil
|
return client, closer, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// runCommands runs the commands given in the flag.Args slice.
|
||||||
|
// If no commands are given, it defaults to running the "status" command.
|
||||||
|
func runCommands(client *q3rcon.Rcon, commands []string) {
|
||||||
|
for _, cmd := range commands {
|
||||||
|
resp, err := client.Send(cmd)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
fmt.Print(resp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// interactiveMode continuously reads from input until a quit signal is given.
|
// interactiveMode continuously reads from input until a quit signal is given.
|
||||||
func interactiveMode(ctx *context) error {
|
func interactiveMode(client *q3rcon.Rcon, input io.Reader) error {
|
||||||
scanner := bufio.NewScanner(ctx.in)
|
scanner := bufio.NewScanner(input)
|
||||||
for scanner.Scan() {
|
for scanner.Scan() {
|
||||||
cmd := scanner.Text()
|
cmd := scanner.Text()
|
||||||
if strings.EqualFold(cmd, "Q") {
|
if strings.EqualFold(cmd, "Q") {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := runCommand(ctx, cmd); err != nil {
|
resp, err := client.Send(cmd)
|
||||||
fmt.Printf("Error: %v\n", err)
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
fmt.Printf(">> ")
|
fmt.Printf("%s>> ", resp)
|
||||||
}
|
}
|
||||||
|
|
||||||
if scanner.Err() != nil {
|
if scanner.Err() != nil {
|
||||||
@ -228,32 +164,3 @@ func interactiveMode(ctx *context) error {
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// runCommands executes a list of commands sequentially and prints any errors encountered.
|
|
||||||
func runCommands(ctx *context, commands []string) {
|
|
||||||
for _, cmd := range commands {
|
|
||||||
if err := runCommand(ctx, cmd); err != nil {
|
|
||||||
fmt.Printf("Error: %v\n", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// runCommand sends a command to the RCON client and prints the response.
|
|
||||||
// If the command is in the timeouts map, it starts a spinner until the response is received.
|
|
||||||
func runCommand(ctx *context, cmd string) error {
|
|
||||||
before, _, _ := strings.Cut(cmd, " ")
|
|
||||||
_, ok := ctx.timeouts[before]
|
|
||||||
if ok {
|
|
||||||
ctx.sm.Start()
|
|
||||||
}
|
|
||||||
|
|
||||||
resp, err := ctx.client.Send(cmd)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to run command '%s': %w", cmd, err)
|
|
||||||
}
|
|
||||||
if ctx.sm.Running() {
|
|
||||||
ctx.sm.Stop()
|
|
||||||
}
|
|
||||||
fmt.Print(reColourCodes.ReplaceAllString(resp, ""))
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|||||||
29
go.mod
29
go.mod
@ -1,30 +1,7 @@
|
|||||||
module github.com/onyx-and-iris/q3rcon
|
module github.com/onyx-and-iris/q3rcon
|
||||||
|
|
||||||
go 1.25.0
|
go 1.23.0
|
||||||
|
|
||||||
require (
|
require github.com/sirupsen/logrus v1.9.3
|
||||||
github.com/charmbracelet/log v0.4.2
|
|
||||||
github.com/chelnak/ysmrr v0.6.0
|
|
||||||
github.com/peterbourgon/ff/v4 v4.0.0-beta.1
|
|
||||||
)
|
|
||||||
|
|
||||||
require (
|
require golang.org/x/sys v0.34.0 // indirect
|
||||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
|
|
||||||
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc // indirect
|
|
||||||
github.com/charmbracelet/lipgloss v1.1.0 // indirect
|
|
||||||
github.com/charmbracelet/x/ansi v0.8.0 // indirect
|
|
||||||
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd // indirect
|
|
||||||
github.com/charmbracelet/x/term v0.2.1 // indirect
|
|
||||||
github.com/fatih/color v1.18.0 // indirect
|
|
||||||
github.com/go-logfmt/logfmt v0.6.0 // indirect
|
|
||||||
github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
|
|
||||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
|
||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
|
||||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
|
||||||
github.com/muesli/termenv v0.16.0 // indirect
|
|
||||||
github.com/rivo/uniseg v0.4.7 // indirect
|
|
||||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
|
|
||||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d // indirect
|
|
||||||
golang.org/x/sys v0.41.0 // indirect
|
|
||||||
golang.org/x/term v0.40.0 // indirect
|
|
||||||
)
|
|
||||||
|
|||||||
64
go.sum
64
go.sum
@ -1,56 +1,16 @@
|
|||||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
|
|
||||||
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc h1:4pZI35227imm7yK2bGPcfpFEmuY1gc2YSTShr4iJBfs=
|
|
||||||
github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc/go.mod h1:X4/0JoqgTIPSFcRA/P6INZzIuyqdFY5rm8tb41s9okk=
|
|
||||||
github.com/charmbracelet/lipgloss v1.1.0 h1:vYXsiLHVkK7fp74RkV7b2kq9+zDLoEU4MZoFqR/noCY=
|
|
||||||
github.com/charmbracelet/lipgloss v1.1.0/go.mod h1:/6Q8FR2o+kj8rz4Dq0zQc3vYf7X+B0binUUBwA0aL30=
|
|
||||||
github.com/charmbracelet/log v0.4.2 h1:hYt8Qj6a8yLnvR+h7MwsJv/XvmBJXiueUcI3cIxsyig=
|
|
||||||
github.com/charmbracelet/log v0.4.2/go.mod h1:qifHGX/tc7eluv2R6pWIpyHDDrrb/AG71Pf2ysQu5nw=
|
|
||||||
github.com/charmbracelet/x/ansi v0.8.0 h1:9GTq3xq9caJW8ZrBTe0LIe2fvfLR/bYXKTx2llXn7xE=
|
|
||||||
github.com/charmbracelet/x/ansi v0.8.0/go.mod h1:wdYl/ONOLHLIVmQaxbIYEC/cRKOQyjTkowiI4blgS9Q=
|
|
||||||
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd h1:vy0GVL4jeHEwG5YOXDmi86oYw2yuYUGqz6a8sLwg0X8=
|
|
||||||
github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd/go.mod h1:xe0nKWGd3eJgtqZRaN9RjMtK7xUYchjzPr7q6kcvCCs=
|
|
||||||
github.com/charmbracelet/x/term v0.2.1 h1:AQeHeLZ1OqSXhrAWpYUtZyX1T3zVxfpZuEQMIQaGIAQ=
|
|
||||||
github.com/charmbracelet/x/term v0.2.1/go.mod h1:oQ4enTYFV7QN4m0i9mzHrViD7TQKvNEEkHUMCmsxdUg=
|
|
||||||
github.com/chelnak/ysmrr v0.6.0 h1:kMhO0oI02tl/9szvxrOE0yeImtrK4KQhER0oXu1K/iM=
|
|
||||||
github.com/chelnak/ysmrr v0.6.0/go.mod h1:56JSrmQgb7/7xoMvuD87h3PE/qW6K1+BQcrgWtVLTUo=
|
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
|
||||||
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
|
||||||
github.com/go-logfmt/logfmt v0.6.0 h1:wGYYu3uicYdqXVgoYbvnkrPVXkuLM1p1ifugDMEdRi4=
|
|
||||||
github.com/go-logfmt/logfmt v0.6.0/go.mod h1:WYhtIu8zTZfxdn5+rREduYbwxfcBr/Vr6KEVveWlfTs=
|
|
||||||
github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
|
|
||||||
github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
|
|
||||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
|
||||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
|
||||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
|
||||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
|
||||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
|
||||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
|
||||||
github.com/muesli/termenv v0.16.0 h1:S5AlUN9dENB57rsbnkPyfdGuWIlkmzJjbFf0Tf5FWUc=
|
|
||||||
github.com/muesli/termenv v0.16.0/go.mod h1:ZRfOIKPFDYQoDFF4Olj7/QJbW60Ol/kL1pU3VfY/Cnk=
|
|
||||||
github.com/pelletier/go-toml/v2 v2.0.9 h1:uH2qQXheeefCCkuBBSLi7jCiSmj3VRh2+Goq2N7Xxu0=
|
|
||||||
github.com/pelletier/go-toml/v2 v2.0.9/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc=
|
|
||||||
github.com/peterbourgon/ff/v4 v4.0.0-beta.1 h1:hV8qRu3V7YfiSMsBSfPfdcznAvPQd3jI5zDddSrDoUc=
|
|
||||||
github.com/peterbourgon/ff/v4 v4.0.0-beta.1/go.mod h1:onQJUKipvCyFmZ1rIYwFAh1BhPOvftb1uhvSI7krNLc=
|
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
|
||||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
|
github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
|
||||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
||||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
|
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
|
golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=
|
||||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
|
golang.org/x/sys v0.34.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
||||||
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
|
||||||
golang.org/x/term v0.40.0 h1:36e4zGLqU4yhjlmxEaagx2KuYbJq3EwY8K943ZsHcvg=
|
|
||||||
golang.org/x/term v0.40.0/go.mod h1:w2P8uVp06p2iyKKuvXIm7N/y0UCRt3UfJTfZ7oOpglM=
|
|
||||||
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
|
||||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
|
||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
|
||||||
|
|||||||
@ -1,33 +1,34 @@
|
|||||||
package q3rcon
|
package conn
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/charmbracelet/log"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type UDPConn struct {
|
type UDPConn struct {
|
||||||
conn *net.UDPConn
|
conn *net.UDPConn
|
||||||
}
|
}
|
||||||
|
|
||||||
func newUDPConn(host string, port int) (*UDPConn, error) {
|
func New(host string, port int) (UDPConn, error) {
|
||||||
udpAddr, err := net.ResolveUDPAddr("udp4", net.JoinHostPort(host, fmt.Sprintf("%d", port)))
|
udpAddr, err := net.ResolveUDPAddr("udp4", fmt.Sprintf("%s:%d", host, port))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return UDPConn{}, err
|
||||||
}
|
}
|
||||||
conn, err := net.DialUDP("udp4", nil, udpAddr)
|
conn, err := net.DialUDP("udp4", nil, udpAddr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return UDPConn{}, err
|
||||||
}
|
}
|
||||||
log.Infof("Outgoing address %s", conn.RemoteAddr())
|
log.Infof("Outgoing address %s", conn.RemoteAddr())
|
||||||
|
|
||||||
return &UDPConn{
|
return UDPConn{
|
||||||
conn: conn,
|
conn: conn,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConn) Write(buf []byte) (int, error) {
|
func (c UDPConn) Write(buf []byte) (int, error) {
|
||||||
n, err := c.conn.Write(buf)
|
n, err := c.conn.Write(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
@ -36,7 +37,8 @@ func (c *UDPConn) Write(buf []byte) (int, error) {
|
|||||||
return n, nil
|
return n, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConn) Read(buf []byte) (int, error) {
|
func (c UDPConn) ReadUntil(timeout time.Time, buf []byte) (int, error) {
|
||||||
|
c.conn.SetReadDeadline(timeout)
|
||||||
rlen, _, err := c.conn.ReadFromUDP(buf)
|
rlen, _, err := c.conn.ReadFromUDP(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
@ -44,7 +46,7 @@ func (c *UDPConn) Read(buf []byte) (int, error) {
|
|||||||
return rlen, nil
|
return rlen, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConn) Close() error {
|
func (c UDPConn) Close() error {
|
||||||
err := c.conn.Close()
|
err := c.conn.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
36
internal/packet/request.go
Normal file
36
internal/packet/request.go
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
const bufSz = 512
|
||||||
|
|
||||||
|
type Request struct {
|
||||||
|
magic []byte
|
||||||
|
password string
|
||||||
|
buf *bytes.Buffer
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRequest(password string) Request {
|
||||||
|
return Request{
|
||||||
|
magic: []byte{'\xff', '\xff', '\xff', '\xff'},
|
||||||
|
password: password,
|
||||||
|
buf: bytes.NewBuffer(make([]byte, bufSz)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Request) Header() []byte {
|
||||||
|
return append(r.magic, []byte("rcon")...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Request) Encode(cmd string) []byte {
|
||||||
|
r.buf.Reset()
|
||||||
|
r.buf.Write(r.Header())
|
||||||
|
r.buf.WriteString(fmt.Sprintf(" %s %s", r.password, cmd))
|
||||||
|
log.Tracef("Encoded request: %s", r.buf.String())
|
||||||
|
return r.buf.Bytes()
|
||||||
|
}
|
||||||
13
internal/packet/response.go
Normal file
13
internal/packet/response.go
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
type Response struct {
|
||||||
|
magic []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewResponse() Response {
|
||||||
|
return Response{magic: []byte{'\xff', '\xff', '\xff', '\xff'}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Response) Header() []byte {
|
||||||
|
return append(r.magic, []byte("print\n")...)
|
||||||
|
}
|
||||||
2
makefile
2
makefile
@ -6,7 +6,7 @@ BIN_DIR := bin
|
|||||||
WINDOWS=$(BIN_DIR)/$(PROGRAM)_windows_amd64.exe
|
WINDOWS=$(BIN_DIR)/$(PROGRAM)_windows_amd64.exe
|
||||||
LINUX=$(BIN_DIR)/$(PROGRAM)_linux_amd64
|
LINUX=$(BIN_DIR)/$(PROGRAM)_linux_amd64
|
||||||
MACOS=$(BIN_DIR)/$(PROGRAM)_darwin_amd64
|
MACOS=$(BIN_DIR)/$(PROGRAM)_darwin_amd64
|
||||||
VERSION=$(shell git describe --tags $(shell git rev-list --tags --max-count=1))
|
VERSION=$(shell git log -n 1 --format=%h)
|
||||||
|
|
||||||
.DEFAULT_GOAL := build
|
.DEFAULT_GOAL := build
|
||||||
|
|
||||||
|
|||||||
62
q3rcon.go
62
q3rcon.go
@ -1,31 +1,24 @@
|
|||||||
package q3rcon
|
package q3rcon
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/charmbracelet/log"
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
|
"github.com/onyx-and-iris/q3rcon/internal/conn"
|
||||||
|
"github.com/onyx-and-iris/q3rcon/internal/packet"
|
||||||
)
|
)
|
||||||
|
|
||||||
const respBufSiz = 2048
|
const respBufSiz = 2048
|
||||||
|
|
||||||
type encoder interface {
|
|
||||||
encode(cmd string) ([]byte, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
type decoder interface {
|
|
||||||
isValid(buf []byte) bool
|
|
||||||
decode(buf []byte) string
|
|
||||||
}
|
|
||||||
|
|
||||||
type Rcon struct {
|
type Rcon struct {
|
||||||
conn io.ReadWriteCloser
|
conn conn.UDPConn
|
||||||
request encoder
|
request packet.Request
|
||||||
response decoder
|
response packet.Response
|
||||||
|
|
||||||
loginTimeout time.Duration
|
loginTimeout time.Duration
|
||||||
defaultTimeout time.Duration
|
defaultTimeout time.Duration
|
||||||
@ -37,15 +30,15 @@ func New(host string, port int, password string, options ...Option) (*Rcon, erro
|
|||||||
return nil, errors.New("no password provided")
|
return nil, errors.New("no password provided")
|
||||||
}
|
}
|
||||||
|
|
||||||
conn, err := newUDPConn(host, port)
|
conn, err := conn.New(host, port)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("error creating UDP connection: %w", err)
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
r := &Rcon{
|
r := &Rcon{
|
||||||
conn: conn,
|
conn: conn,
|
||||||
request: newRequest(password),
|
request: packet.NewRequest(password),
|
||||||
response: newResponse(),
|
response: packet.NewResponse(),
|
||||||
|
|
||||||
loginTimeout: 5 * time.Second,
|
loginTimeout: 5 * time.Second,
|
||||||
defaultTimeout: 20 * time.Millisecond,
|
defaultTimeout: 20 * time.Millisecond,
|
||||||
@ -57,7 +50,7 @@ func New(host string, port int, password string, options ...Option) (*Rcon, erro
|
|||||||
}
|
}
|
||||||
|
|
||||||
if err = r.login(); err != nil {
|
if err = r.login(); err != nil {
|
||||||
return nil, fmt.Errorf("error logging in: %w", err)
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return r, nil
|
return r, nil
|
||||||
@ -72,7 +65,7 @@ func (r Rcon) login() error {
|
|||||||
default:
|
default:
|
||||||
resp, err := r.Send("login")
|
resp, err := r.Send("login")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error sending login command: %w", err)
|
return err
|
||||||
}
|
}
|
||||||
if resp == "" {
|
if resp == "" {
|
||||||
continue
|
continue
|
||||||
@ -101,14 +94,9 @@ func (r Rcon) Send(cmdWithArgs string) (string, error) {
|
|||||||
|
|
||||||
go r.listen(timeout, respChan, errChan)
|
go r.listen(timeout, respChan, errChan)
|
||||||
|
|
||||||
encodedCmd, err := r.request.encode(cmdWithArgs)
|
_, err := r.conn.Write(r.request.Encode(cmdWithArgs))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("error encoding command: %w", err)
|
return "", err
|
||||||
}
|
|
||||||
|
|
||||||
_, err = r.conn.Write(encodedCmd)
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("error writing command to connection: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
select {
|
select {
|
||||||
@ -130,17 +118,7 @@ func (r Rcon) listen(timeout time.Duration, respChan chan<- string, errChan chan
|
|||||||
respChan <- sb.String()
|
respChan <- sb.String()
|
||||||
return
|
return
|
||||||
default:
|
default:
|
||||||
c, ok := r.conn.(*UDPConn)
|
rlen, err := r.conn.ReadUntil(time.Now().Add(timeout), respBuf)
|
||||||
if !ok {
|
|
||||||
errChan <- errors.New("connection is not a UDPConn")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
err := c.conn.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
if err != nil {
|
|
||||||
errChan <- fmt.Errorf("error setting read deadline: %w", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
rlen, err := r.conn.Read(respBuf)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
e, ok := err.(net.Error)
|
e, ok := err.(net.Error)
|
||||||
if ok {
|
if ok {
|
||||||
@ -153,8 +131,10 @@ func (r Rcon) listen(timeout time.Duration, respChan chan<- string, errChan chan
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if r.response.isValid(respBuf[:rlen]) {
|
if rlen > len(r.response.Header()) {
|
||||||
sb.WriteString(r.response.decode(respBuf[:rlen]))
|
if bytes.HasPrefix(respBuf, r.response.Header()) {
|
||||||
|
sb.Write(respBuf[len(r.response.Header()):rlen])
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
35
request.go
35
request.go
@ -1,35 +0,0 @@
|
|||||||
package q3rcon
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
bufSz = 1024
|
|
||||||
requestHeader = "\xff\xff\xff\xffrcon"
|
|
||||||
)
|
|
||||||
|
|
||||||
type request struct {
|
|
||||||
password string
|
|
||||||
buf *bytes.Buffer
|
|
||||||
}
|
|
||||||
|
|
||||||
func newRequest(password string) request {
|
|
||||||
return request{
|
|
||||||
password: password,
|
|
||||||
buf: bytes.NewBuffer(make([]byte, 0, bufSz)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r request) encode(cmd string) ([]byte, error) {
|
|
||||||
if cmd == "" {
|
|
||||||
return nil, errors.New("command cannot be empty")
|
|
||||||
}
|
|
||||||
|
|
||||||
r.buf.Reset()
|
|
||||||
r.buf.WriteString(requestHeader)
|
|
||||||
r.buf.WriteString(fmt.Sprintf(" %s %s", r.password, cmd))
|
|
||||||
return r.buf.Bytes(), nil
|
|
||||||
}
|
|
||||||
21
response.go
21
response.go
@ -1,21 +0,0 @@
|
|||||||
package q3rcon
|
|
||||||
|
|
||||||
import "bytes"
|
|
||||||
|
|
||||||
const (
|
|
||||||
responseHeader = "\xff\xff\xff\xffprint\n"
|
|
||||||
)
|
|
||||||
|
|
||||||
type response struct{}
|
|
||||||
|
|
||||||
func newResponse() response {
|
|
||||||
return response{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r response) isValid(buf []byte) bool {
|
|
||||||
return len(buf) > len(responseHeader) && bytes.HasPrefix(buf, []byte(responseHeader))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r response) decode(buf []byte) string {
|
|
||||||
return string(buf[len(responseHeader):])
|
|
||||||
}
|
|
||||||
Loading…
x
Reference in New Issue
Block a user