From 3fdc8106de9814e3836c9513ecd8501c2e8d872c Mon Sep 17 00:00:00 2001
From: Max <max@privatevoid.net>
Date: Mon, 23 Jan 2023 20:28:55 +0100
Subject: [PATCH] HyprspaceRPC.Route: validate argument count

---
 rpc/server.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/rpc/server.go b/rpc/server.go
index edb3129..714b377 100644
--- a/rpc/server.go
+++ b/rpc/server.go
@@ -75,6 +75,9 @@ func (hsr *HyprspaceRPC) Route(args *RouteArgs, reply *RouteReply) error {
 			Routes: routes,
 		}
 	case Relay:
+		if len(args.Args) != 2 {
+			return errors.New("expected exactly 2 arguments")
+		}
 		var networks []net.IPNet
 		if args.Args[0] == "all" {
 			for _, r := range hsr.config.Routes {
@@ -99,6 +102,9 @@ func (hsr *HyprspaceRPC) Route(args *RouteArgs, reply *RouteReply) error {
 			p2p.AddReroute(n, p)
 		}
 	case Reset:
+		if len(args.Args) != 1 {
+			return errors.New("expected exactly 1 argument")
+		}
 		var networks []net.IPNet
 		if args.Args[0] == "all" {
 			for _, r := range hsr.config.Routes {