added initial typing of forwarders, now only tcp
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user