putchunks: add signal handler

named-data · Jan 3, 2025 · de4ebcf · de4ebcf
1 parent f3ec652
commit de4ebcf
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tools/putchunks.go b/tools/putchunks.go
@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"os/signal"
+	"syscall"
 
 	enc "github.com/named-data/ndnd/std/encoding"
 	"github.com/named-data/ndnd/std/engine"
@@ -92,6 +94,7 @@ func (pc *PutChunks) run() {
 	}
 
 	// wait forever
-	// TODO: quit on SIGTERM, SIGINT or face failure
-	select {}
+	sigchan := make(chan os.Signal, 1)
+	signal.Notify(sigchan, os.Interrupt, syscall.SIGTERM)
+	<-sigchan
 }