Commit 63d0c0b9 authored by Dmitriy Zaporozhets's avatar Dmitriy Zaporozhets

Merge branch 'gitlab_popen_array' into 'master'

Change Gitlab::Popen to only accept arrays as commands
parents 6c1af645 36c2c354
...@@ -53,7 +53,7 @@ class Key < ActiveRecord::Base ...@@ -53,7 +53,7 @@ class Key < ActiveRecord::Base
Tempfile.open('gitlab_key_file') do |file| Tempfile.open('gitlab_key_file') do |file|
file.puts key file.puts key
file.rewind file.rewind
cmd_output, cmd_status = popen("ssh-keygen -lf #{file.path}", '/tmp') cmd_output, cmd_status = popen(%W(ssh-keygen -lf #{file.path}), '/tmp')
end end
if cmd_status.zero? if cmd_status.zero?
......
require 'fileutils' require 'fileutils'
require 'open3'
module Gitlab module Gitlab
module Popen module Popen
def popen(cmd, path) extend self
def popen(cmd, path=nil)
unless cmd.is_a?(Array)
raise "System commands must be given as an array of strings"
end
path ||= Dir.pwd
vars = { "PWD" => path } vars = { "PWD" => path }
options = { chdir: path } options = { chdir: path }
...@@ -12,10 +20,10 @@ module Gitlab ...@@ -12,10 +20,10 @@ module Gitlab
@cmd_output = "" @cmd_output = ""
@cmd_status = 0 @cmd_status = 0
Open3.popen3(vars, cmd, options) do |stdin, stdout, stderr, wait_thr| Open3.popen3(vars, *cmd, options) do |stdin, stdout, stderr, wait_thr|
@cmd_status = wait_thr.value.exitstatus
@cmd_output << stdout.read @cmd_output << stdout.read
@cmd_output << stderr.read @cmd_output << stderr.read
@cmd_status = wait_thr.value.exitstatus
end end
return @cmd_output, @cmd_status return @cmd_output, @cmd_status
......
...@@ -33,7 +33,7 @@ module Gitlab ...@@ -33,7 +33,7 @@ module Gitlab
end end
def create def create
output, status = popen("git clone #{project.repository.path_to_repo} #{path}", output, status = popen(%W(git clone -- #{project.repository.path_to_repo} #{path}),
Gitlab.config.satellites.path) Gitlab.config.satellites.path)
log("PID: #{project.id}: git clone #{project.repository.path_to_repo} #{path}") log("PID: #{project.id}: git clone #{project.repository.path_to_repo} #{path}")
......
...@@ -10,7 +10,7 @@ describe 'Gitlab::Popen', no_db: true do ...@@ -10,7 +10,7 @@ describe 'Gitlab::Popen', no_db: true do
context 'zero status' do context 'zero status' do
before do before do
@output, @status = @klass.new.popen('ls', path) @output, @status = @klass.new.popen(%W(ls), path)
end end
it { @status.should be_zero } it { @status.should be_zero }
...@@ -19,11 +19,27 @@ describe 'Gitlab::Popen', no_db: true do ...@@ -19,11 +19,27 @@ describe 'Gitlab::Popen', no_db: true do
context 'non-zero status' do context 'non-zero status' do
before do before do
@output, @status = @klass.new.popen('cat NOTHING', path) @output, @status = @klass.new.popen(%W(cat NOTHING), path)
end end
it { @status.should == 1 } it { @status.should == 1 }
it { @output.should include('No such file or directory') } it { @output.should include('No such file or directory') }
end end
context 'unsafe string command' do
it 'raises an error when it gets called with a string argument' do
expect { @klass.new.popen('ls', path) }.to raise_error
end
end
context 'without a directory argument' do
before do
@output, @status = @klass.new.popen(%W(ls))
end
it { @status.should be_zero }
it { @output.should include('spec') }
end
end end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment