Autoupdate: check uid before restart service.

lesovsky · May 5, 2021 · 19cf058 · 19cf058
1 parent 449f4c6
commit 19cf058
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/packaging/autoupdate/autoupdate.go b/internal/packaging/autoupdate/autoupdate.go
@@ -410,6 +410,10 @@ func updateBinary(sourceFile string, destFile string) error {
 
 // restartSystemdService checks privileges and restarts pgscv service.
 func restartSystemdService() error {
+	if os.Geteuid() != 0 {
+		return fmt.Errorf("root privileges required")
+	}
+
 	cmd := exec.Command("systemctl", "restart", "pgscv.service")
 	// after cmd.Start execution of this code could be interrupted, end even err might not be handled.
 	err := cmd.Start()