tux/trok
1
0
Fork 0
mirror of https://github.com/tuxdotrs/trok.git synced 2025-07-06 11:56:34 +05:30
Code Issues Packages Projects Releases Wiki Activity

feat: setup addr arg

Browse Source
This commit is contained in:
tux
2025-02-27 22:17:13 +05:30
parent e64319616f
commit b5756cf429
4 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/server.go
View File

@ -15,16 +15,17 @@ var serverCmd = &cobra.Command{
Short: "Initiates the remote proxy server",
Long: "Initiates the remote proxy server",
Run: func(cmd *cobra.Command, args []string) {
port, err := cmd.Flags().GetUint16("port")
addr, err := cmd.Flags().GetString("addr")
if err != nil {
panic(err)
}
server.Start(port)
server.Start(addr)
},
}
func init() {
rootCmd.AddCommand(serverCmd)
serverCmd.Flags().Uint16P("port", "p", 1421, "Port for the server to listen on")
serverCmd.Flags().StringP("addr", "a", "0.0.0.0:1337", "Addr for the server to listen on")
clientCmd.MarkFlagRequired("addr")
}

4
internal/server/init.go
View File

@ -11,9 +11,9 @@ import (
"github.com/rs/zerolog/log"
)
func Start(port uint16) {
func Start(addr string) {
var trok Trok
if err := trok.Init(port); err != nil {
if err := trok.Init(addr); err != nil {
log.Fatal().Msgf("failed to init trok %v", err)
}

18
internal/server/tcp.go
View File

@ -5,7 +5,6 @@ Copyright © 2024 tux <0xtux@pm.me>
package server
import (
"fmt"
"net"
)
@ -14,13 +13,8 @@ type TCPServer struct {
listener net.Listener
}
func (s *TCPServer) Init(port uint16, title string) error {
address := ":"
if port > 0 {
address = fmt.Sprintf(":%d", port)
}
ln, err := net.Listen("tcp", address)
func (s *TCPServer) Init(addr, title string) error {
ln, err := net.Listen("tcp", addr)
if err != nil {
return err
}
@ -44,6 +38,14 @@ func (s *TCPServer) Stop() error {
return s.listener.Close()
}
func (s *TCPServer) Addr() string {
return s.listener.Addr().String()
}
func (c *TCPServer) Host() string {
return c.listener.Addr().(*net.TCPAddr).IP.String()
}
func (s *TCPServer) Port() uint16 {
return uint16(s.listener.Addr().(*net.TCPAddr).Port)
}

10
internal/server/trok.go
View File

@ -27,20 +27,20 @@ type Trok struct {
mutex sync.Mutex
}
func (t *Trok) Init(port uint16) error {
func (t *Trok) Init(addr string) error {
t.publicConns = make(map[string]Conn)
err := t.controlServer.Init(port, "Controller")
err := t.controlServer.Init(addr, "Controller")
return err
}
func (t *Trok) Start() {
go t.controlServer.Start(t.ControlConnHandler)
log.Info().Msgf("started Trok server on port %d", t.controlServer.Port())
log.Info().Msgf("started Trok server on %s", t.controlServer.Addr())
}
func (t *Trok) Stop() {
t.controlServer.Stop()
log.Info().Msgf("stopped Trok server on port %d", t.controlServer.Port())
log.Info().Msgf("stopped Trok server on %s", t.controlServer.Addr())
}
func (t *Trok) ControlConnHandler(conn net.Conn) {
@ -75,7 +75,7 @@ func (t *Trok) handleCMDHELO(p *lib.ProtocolHandler, m *lib.Message) {
log.Info().Msgf("[CMD] %s [ARG] %s", m.CMD, m.ARG)
var s TCPServer
err := s.Init(0, "Handler")
err := s.Init(":", "Handler")
if err != nil {
log.Error().Msgf("error handling HELO cmd: %v", err)
return