sudo: add the sudo_user arg to SystemCommand.

This commit is contained in:
Ilya Kulakov 2023-02-17 14:48:13 -08:00
parent 563387a7b4
commit d470661b37
4 changed files with 65 additions and 29 deletions

View File

@ -106,11 +106,13 @@ module Cask
all_services.each do |service| all_services.each do |service|
ohai "Removing launchctl service #{service}" ohai "Removing launchctl service #{service}"
booleans.each do |with_sudo| booleans.each do |with_sudo|
sudo_user = with_sudo ? "root" : "" sudo_user = with_sudo ? "root" : nil
plist_status = command.run( plist_status = command.run(
"/bin/launchctl", "/bin/launchctl",
args: ["list", service], args: ["list", service],
sudo: sudo_user, print_stderr: false sudo: with_sudo,
sudo_user: sudo_user,
print_stderr: false,
).stdout ).stdout
if plist_status.start_with?("{") if plist_status.start_with?("{")
command.run!("/bin/launchctl", args: ["remove", service], sudo: sudo_user) command.run!("/bin/launchctl", args: ["remove", service], sudo: sudo_user)
@ -123,13 +125,23 @@ module Cask
paths.each { |elt| elt.prepend(Dir.home).freeze } unless with_sudo paths.each { |elt| elt.prepend(Dir.home).freeze } unless with_sudo
paths = paths.map { |elt| Pathname(elt) }.select(&:exist?) paths = paths.map { |elt| Pathname(elt) }.select(&:exist?)
paths.each do |path| paths.each do |path|
command.run!("/bin/rm", args: ["-f", "--", path], sudo: sudo_user) command.run!("/bin/rm", args: ["-f", "--", path], sudo: with_sudo, sudo_user: sudo_user)
end end
# undocumented and untested: pass a path to uninstall :launchctl # undocumented and untested: pass a path to uninstall :launchctl
next unless Pathname(service).exist? next unless Pathname(service).exist?
command.run!("/bin/launchctl", args: ["unload", "-w", "--", service], sudo: sudo_user) command.run!(
command.run!("/bin/rm", args: ["-f", "--", service], sudo: sudo_user) "/bin/launchctl",
args: ["unload", "-w", "--", service],
sudo: with_sudo,
sudo_user: sudo_user,
)
command.run!(
"/bin/rm",
args: ["-f", "--", service],
sudo: with_sudo,
sudo_user: sudo_user,
)
sleep 1 sleep 1
end end
end end
@ -302,15 +314,35 @@ module Cask
def uninstall_kext(*kexts, command: nil, **_) def uninstall_kext(*kexts, command: nil, **_)
kexts.each do |kext| kexts.each do |kext|
ohai "Unloading kernel extension #{kext}" ohai "Unloading kernel extension #{kext}"
is_loaded = system_command!("/usr/sbin/kextstat", args: ["-l", "-b", kext], sudo: "root").stdout is_loaded = system_command!(
"/usr/sbin/kextstat",
args: ["-l", "-b", kext],
sudo: true,
sudo_user: "root",
).stdout
if is_loaded.length > 1 if is_loaded.length > 1
system_command!("/sbin/kextunload", args: ["-b", kext], sudo: "root") system_command!(
"/sbin/kextunload",
args: ["-b", kext],
sudo: true,
sudo_user: "root",
)
sleep 1 sleep 1
end end
kexts = system_command!("/usr/sbin/kextfind", args: ["-b", kext], sudo: "root").stdout kexts = system_command!(
"/usr/sbin/kextfind",
args: ["-b", kext],
sudo: true,
sudo_user: "root",
).stdout
kexts.chomp.lines.each do |kext_path| kexts.chomp.lines.each do |kext_path|
ohai "Removing kernel extension #{kext_path}" ohai "Removing kernel extension #{kext_path}"
system_command!("/bin/rm", args: ["-rf", kext_path], sudo: "root") system_command!(
"/bin/rm",
args: ["-rf", kext_path],
sudo: true,
sudo_user: "root",
)
end end
end end
end end

View File

@ -62,7 +62,7 @@ module Cask
"USER" => User.current, "USER" => User.current,
"USERNAME" => User.current, "USERNAME" => User.current,
} }
command.run!("/usr/sbin/installer", sudo: "root", args: args, print_stdout: true, env: env) command.run!("/usr/sbin/installer", sudo: true, sudo_user: "root", args: args, print_stdout: true, env: env)
end end
end end

View File

@ -32,7 +32,8 @@ module Cask
"/usr/bin/xargs", "/usr/bin/xargs",
args: ["-0", "--", "/bin/rm", "--"], args: ["-0", "--", "/bin/rm", "--"],
input: pkgutil_bom_files.join("\0"), input: pkgutil_bom_files.join("\0"),
sudo: "root", sudo: true,
sudo_user: "root",
) )
end end
@ -42,7 +43,8 @@ module Cask
"/usr/bin/xargs", "/usr/bin/xargs",
args: ["-0", "--", "/bin/rm", "--"], args: ["-0", "--", "/bin/rm", "--"],
input: pkgutil_bom_specials.join("\0"), input: pkgutil_bom_specials.join("\0"),
sudo: "root", sudo: true,
sudo_user: "root",
) )
end end
@ -59,7 +61,7 @@ module Cask
sig { void } sig { void }
def forget def forget
odebug "Unregistering pkg receipt (aka forgetting)" odebug "Unregistering pkg receipt (aka forgetting)"
@command.run!("/usr/sbin/pkgutil", args: ["--forget", package_id], sudo: "root") @command.run!("/usr/sbin/pkgutil", args: ["--forget", package_id], sudo: true, sudo_user: "root")
end end
sig { returns(T::Array[Pathname]) } sig { returns(T::Array[Pathname]) }
@ -114,7 +116,8 @@ module Cask
"/usr/bin/xargs", "/usr/bin/xargs",
args: ["-0", "--", RMDIR_SH.to_s], args: ["-0", "--", RMDIR_SH.to_s],
input: Array(path).join("\0"), input: Array(path).join("\0"),
sudo: "root", sudo: true,
sudo_user: "root",
) )
end end

View File

@ -64,7 +64,8 @@ class SystemCommand
params( params(
executable: T.any(String, Pathname), executable: T.any(String, Pathname),
args: T::Array[T.any(String, Integer, Float, URI::Generic)], args: T::Array[T.any(String, Integer, Float, URI::Generic)],
sudo: T.any(T::Boolean, String), sudo: T::Boolean,
sudo_user: T.nilable(String),
env: T::Hash[String, String], env: T::Hash[String, String],
input: T.any(String, T::Array[String]), input: T.any(String, T::Array[String]),
must_succeed: T::Boolean, must_succeed: T::Boolean,
@ -81,6 +82,7 @@ class SystemCommand
executable, executable,
args: [], args: [],
sudo: false, sudo: false,
sudo_user: nil,
env: {}, env: {},
input: [], input: [],
must_succeed: false, must_succeed: false,
@ -95,7 +97,11 @@ class SystemCommand
require "extend/ENV" require "extend/ENV"
@executable = executable @executable = executable
@args = args @args = args
raise ArgumentError, "sudo_user cannot be set if sudo is false" if !sudo && !sudo_user.nil?
@sudo = sudo @sudo = sudo
@sudo_user = sudo_user
env.each_key do |name| env.each_key do |name|
next if /^[\w&&\D]\w*$/.match?(name) next if /^[\w&&\D]\w*$/.match?(name)
@ -122,12 +128,7 @@ class SystemCommand
attr_reader :executable, :args, :input, :chdir, :env attr_reader :executable, :args, :input, :chdir, :env
attr_predicate :print_stdout?, :print_stderr?, :must_succeed? attr_predicate :sudo?, :print_stdout?, :print_stderr?, :must_succeed?
sig { returns(T::Boolean) }
def sudo?
@sudo != false && @sudo != ""
end
sig { returns(T::Boolean) } sig { returns(T::Boolean) }
def debug? def debug?
@ -158,7 +159,7 @@ class SystemCommand
sig { returns(T::Array[String]) } sig { returns(T::Array[String]) }
def sudo_prefix def sudo_prefix
user_flags = @sudo.is_a?(String) ? ["-u", @sudo] : [] user_flags = @sudo_user.nil? ? [] : ["-u", @sudo_user]
askpass_flags = ENV.key?("SUDO_ASKPASS") ? ["-A"] : [] askpass_flags = ENV.key?("SUDO_ASKPASS") ? ["-A"] : []
["/usr/bin/sudo", *user_flags, *askpass_flags, "-E", *env_args, "--"] ["/usr/bin/sudo", *user_flags, *askpass_flags, "-E", *env_args, "--"]
end end