Forming sanitary shell commands or system calls in Ruby

后端 未结 5 1768
长发绾君心
长发绾君心 2020-12-08 11:20

I\'m building a daemon that will help me manage my server(s). Webmin works fine, as does just opening a shell to the server, but I\'d prefer to be able to control server ope

相关标签:
5条回答
  • 2020-12-08 12:00

    It doesn't look like you need a shell for what you're doing. See the documentation for system here: http://ruby-doc.org/core/classes/Kernel.html#M001441

    You should use the second form of system. Your example above would become:

    system 'usermod', '-p', @options['shadow'], @options['username']
    

    A nicer (IMO) way to write this is:

    system *%W(usermod -p #{@options['shadow']} #{@options['username']})
    

    The arguments this way are passed directly into the execve call, so you don't have to worry about sneaky shell tricks.

    0 讨论(0)
  • 2020-12-08 12:00

    This is an old question, but since it's pretty much the only real answer you'll find when googling I thought I'd add a caveat. The multi argument version of system seems reasonably safe on Linux, but it is NOT on Windows.

    Try system "dir", "&", "echo", "hi!" on a Windows system. Both dir and echo will be run. Echo could of course just as well be something far less innocuous.

    0 讨论(0)
  • 2020-12-08 12:08

    If you need not just the exit status but also the result you probably want to use Open3.popen3:

    require 'open3'
    stdin, stdout, stderr = Open3.popen3('usermod', '-p', @options['shadow'], @options['username'])
    stdout.gets
    sterr.gets
    

    More information here: Getting output of system() calls in Ruby

    0 讨论(0)
  • 2020-12-08 12:16

    I'd suggest looking into the 'shellwords' module. This script:

    require 'shellwords'
    parts = ['echo', "'hello world'; !%& some stuff", 'and another argument']
    command = Shellwords.shelljoin( parts )
    puts command
    output = `#{ command }`
    puts output
    

    outputs the escaped text and the expected output:

    echo \'hello\ world\'\;\ \!\%\&\ some\ stuff and\ another\ argument
    'hello world'; !%& some stuff and another argument
    
    0 讨论(0)
  • 2020-12-08 12:19

    I know this is an old thread, but there is another option that was lightly touched on by Simon Hürlimann.

    There is not a lot of information about this topic and I think this might help others in need.

    For this example we'll use Open3 which gives you the ability to run commands synchronously or asynchronously, and provides stdout, stderr, exit codes, and PID.

    Open3 grants you access to stdout, stderr, exit codes and a thread to wait for the child process when running another program. You can specify various attributes, redirections, current directory, etc., of the program in the same way as for Process.spawn. (Source: Open3 Docs)

    I chose to format the output as a CommandStatus object. This contains our stdout, stderr, pid (Of the worker thread) and exitstatus.

    class Command
      require 'open3'
    
      class CommandStatus
        @stdout     = nil
        @stderr     = nil
        @pid        = nil
        @exitstatus = nil
    
        def initialize(stdout, stderr, process)
          @stdout     = stdout
          @stderr     = stderr
          @pid        = process.pid
          @exitstatus = process.exitstatus
        end
    
        def stdout
          @stdout
        end
    
        def stderr
          @stderr
        end
    
        def exit_status
          @exitstatus
        end
    
        def pid
          @pid
        end
      end
    
      def self.execute(command)
        command_stdout = nil
        command_stderr = nil
        process = Open3.popen3(ENV, command + ';') do |stdin, stdout, stderr, thread|
          stdin.close
          stdout_buffer   = stdout.read
          stderr_buffer   = stderr.read
          command_stdout  = stdout_buffer if stdout_buffer.length > 0
          command_stderr  = stderr_buffer if stderr_buffer.length > 0
          thread.value # Wait for Process::Status object to be returned
        end
        return CommandStatus.new(command_stdout, command_stderr, process)
      end
    end
    
    
    cmd = Command::execute("echo {1..10}")
    
    puts "STDOUT: #{cmd.stdout}"
    puts "STDERR: #{cmd.stderr}"
    puts "EXIT: #{cmd.exit_status}"
    

    While reading the STDOUT/ERR buffers, I use command_stdout = stdout_buffer if stdout_buffer.length > 0 to control whether the command_stdout variable is assigned or not. You should pass nil instead of "" when no data is present. It's more clear when handing data later on.

    You probably noticed me using command + ';'. The reason for this is based on the documentation from Kernel.exec (Which is what popen3 uses):

    If the string from the first form (exec("command")) follows these simple rules:

    • no meta characters
    • no shell reserved word and no special built-in
    • Ruby invokes the command directly without shell

    You can force shell invocation by adding ";" to the string (because ";" is a meta character)

    This simply prevents a Ruby from throwing a 'spawn': No such file or directory error if you pass a malformed command. Instead it will pass it straight to the kernel where the error will be resolved gracefully and appear as STDERR instead of an uncaught exception.

    0 讨论(0)
提交回复
热议问题