added initial typing of forwarders, now only tcp

This commit is contained in:
2026-03-26 17:17:18 +02:00
parent e05d610882
commit 230022f856
8 changed files with 115 additions and 68 deletions

View File

@@ -12,6 +12,7 @@ import (
type CreateForwarderParams struct {
Hostname string
Type string
Lport uint32
Dport uint32
Addresses []string
@@ -39,6 +40,7 @@ func (tool *Tool) createForwarder(params *CreateForwarderParams) (*CreateForward
defer conn.Close()
ctx, _ := context.WithTimeout(context.Background(), 1*time.Second)
opReq := &mlbctl.CreateForwarderParams{
Type: params.Type,
Lport: params.Lport,
Dport: params.Dport,
Destinations: params.Addresses,

View File

@@ -12,6 +12,7 @@ import (
type DeleteForwarderParams struct {
Hostname string
Type string
Lport uint32
}
type DeleteForwarderResult struct{}
@@ -37,6 +38,7 @@ func (tool *Tool) deleteForwarder(params *DeleteForwarderParams) (*DeleteForward
defer conn.Close()
ctx, _ := context.WithTimeout(context.Background(), 1*time.Second)
opReq := &mlbctl.DeleteForwarderParams{
Type: params.Type,
Lport: params.Lport,
}
_, err = cli.DeleteForwarder(ctx, opReq)

View File

@@ -35,9 +35,11 @@ func NewTool() *Tool {
Short: "Create forwarder",
Run: tool.CreateForwarder,
}
createForwarderCmd.Flags().StringVarP(&tool.createForwarderParams.Type, "type", "T", "tcp", "Type of forwarding: tcp")
createForwarderCmd.Flags().Uint32VarP(&tool.createForwarderParams.Lport, "lport", "L", 0, "Listening port")
createForwarderCmd.Flags().Uint32VarP(&tool.createForwarderParams.Dport, "dport", "D", 0, "Destination port")
createForwarderCmd.Flags().StringArrayVarP(&tool.createForwarderParams.Addresses, "dests", "A", []string{}, "Destination address list")
createForwarderCmd.MarkFlagRequired("type")
createForwarderCmd.MarkFlagRequired("lport")
createForwarderCmd.MarkFlagRequired("dport")
createForwarderCmd.MarkFlagRequired("dests")
@@ -49,8 +51,10 @@ func NewTool() *Tool {
Short: "Delete forwarder",
Run: tool.DeleteForwarder,
}
deleteForwarderCmd.Flags().StringVarP(&tool.deleteForwarderParams.Type, "type", "T", "tcp", "Forwarder type: tcp")
deleteForwarderCmd.Flags().Uint32VarP(&tool.deleteForwarderParams.Lport, "lport", "L", 0, "Listening port")
deleteForwarderCmd.MarkFlagRequired("lport")
deleteForwarderCmd.MarkFlagRequired("type")
tool.cmd.AddCommand(deleteForwarderCmd)
return tool