cmd_unix.go 1.0 KB

123456789101112131415161718192021222324252627282930313233
  1. // Copyright (C) 2019 Nicola Murino
  2. //
  3. // This program is free software: you can redistribute it and/or modify
  4. // it under the terms of the GNU Affero General Public License as published
  5. // by the Free Software Foundation, version 3.
  6. //
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU Affero General Public License for more details.
  11. //
  12. // You should have received a copy of the GNU Affero General Public License
  13. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  14. //go:build !windows
  15. // +build !windows
  16. package sftpd
  17. import (
  18. "os"
  19. "os/exec"
  20. "syscall"
  21. )
  22. func wrapCmd(cmd *exec.Cmd, uid, gid int) *exec.Cmd {
  23. isCurrentUser := os.Getuid() == uid && os.Getgid() == gid
  24. if (uid > 0 || gid > 0) && !isCurrentUser {
  25. cmd.SysProcAttr = &syscall.SysProcAttr{}
  26. cmd.SysProcAttr.Credential = &syscall.Credential{Uid: uint32(uid), Gid: uint32(gid)}
  27. }
  28. return cmd
  29. }