From f9af813d490072f075531d51d5a333ecf24bc5cc Mon Sep 17 00:00:00 2001 From: Erik Date: Tue, 18 Apr 2023 15:28:55 +0200 Subject: [PATCH] Add create command structure and parsing --- pcsdk/pcsdk.go | 24 ++++++++++++++++++++++++ pcsdk/pcsdk_test.go | 29 +++++++++++++++++++++++++++++ 2 files changed, 53 insertions(+) create mode 100644 pcsdk/pcsdk.go create mode 100644 pcsdk/pcsdk_test.go diff --git a/pcsdk/pcsdk.go b/pcsdk/pcsdk.go new file mode 100644 index 0000000..4b9610c --- /dev/null +++ b/pcsdk/pcsdk.go @@ -0,0 +1,24 @@ +package pcsdk + +import ( + "net/netip" + + "github.com/google/uuid" + "github.com/thefeli73/polemos/state" +) + +type ProxyCommandCreate struct { + Command CommandCreate `json:"create"` + // signature Signature +} + +type CommandCreate struct { + IncomingPort uint16 `json:"incoming_port"` + DestinationPort uint16 `json:"destination_port"` + DestinationIP netip.Addr `json:"destination_ip"` + Id string `json:"id"` +} + +func NewCommandCreate(iport uint16, oport uint16, oip netip.Addr, id state.CustomUUID) CommandCreate { + return CommandCreate{iport, oport, oip, uuid.UUID.String(uuid.UUID(id))} +} diff --git a/pcsdk/pcsdk_test.go b/pcsdk/pcsdk_test.go new file mode 100644 index 0000000..b14462e --- /dev/null +++ b/pcsdk/pcsdk_test.go @@ -0,0 +1,29 @@ +package pcsdk + +import ( + "encoding/json" + "net/netip" + "testing" + + "github.com/google/uuid" + "github.com/thefeli73/polemos/state" +) + +func TestCommandCreateJsonParse(t *testing.T) { + ip, _ := netip.ParseAddr("127.0.0.99") + uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1") + c := NewCommandCreate(5555, 6666, ip, state.CustomUUID(uuid)) + m := ProxyCommandCreate{c} + msg, err := json.Marshal(m) + if err != nil { + t.Fatalf(`%q`, err) + } + + expected := "{\"create\":{\"incoming_port\":5555,\"destination_port\":6666,\"destination_ip\":\"127.0.0.99\",\"id\":\"87e79cbc-6df6-4462-8412-85d6c473e3b1\"}}" + if string(msg) != expected { + t.Fatalf( + "\nExpected:\t %q\nGot:\t\t %q\n", expected, msg) + } +} + +