packages/hyprspace: HyprspaceRPC.Route: validate argument count

This commit is contained in:
Max Headroom 2023-01-23 20:28:55 +01:00
parent c5c3441aa4
commit fa79991e93

View file

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