Set a 'Host' header on unix sockets

Workaround for a bug in the HTTPUNIX client.

2 files changed, 33 insertions, 14 deletions

diff --git a/lib/gitlab_net.rb b/lib/gitlab_net.rb
index 8488adc..e06557c 100644
--- a/lib/gitlab_net.rb
+++ b/lib/gitlab_net.rb
@@ -159,6 +159,12 @@ class GitlabNet
 
     request.set_form_data(params.merge(secret_token: secret_token))
 
+    if uri.is_a?(URI::HTTPUNIX)
+      # The HTTPUNIX HTTP client does not set a correct Host header. This can
+      # lead to 400 Bad Request responses.
+      request['Host'] = 'localhost'
+    end
+
     request
   end
 
diff --git a/spec/gitlab_net_spec.rb b/spec/gitlab_net_spec.rb
index 3d38231..e3ab16c 100644
--- a/spec/gitlab_net_spec.rb
+++ b/spec/gitlab_net_spec.rb
@@ -248,24 +248,37 @@ describe GitlabNet, vcr: true do
   end
 
   describe :http_request_for do
-    let(:get) do
-      double(Net::HTTP::Get).tap do |get|
-        Net::HTTP::Get.stub(:new) { get }
+    context 'with stub' do
+      let(:get) do
+        double(Net::HTTP::Get).tap do |get|
+          Net::HTTP::Get.stub(:new) { get }
+        end
+      end
+      let(:user) { 'user' }
+      let(:password) { 'password' }
+      let(:url) { URI 'http://localhost/' }
+      subject { gitlab_net.send :http_request_for, :get, url }
+  
+      before do
+        gitlab_net.send(:config).http_settings.stub(:[]).with('user') { user }
+        gitlab_net.send(:config).http_settings.stub(:[]).with('password') { password }
+        get.should_receive(:basic_auth).with(user, password).once
+        get.should_receive(:set_form_data).with(hash_including(secret_token: 'a123')).once
       end
+  
+      it { should_not be_nil }
     end
-    let(:user) { 'user' }
-    let(:password) { 'password' }
-    let(:url) { URI 'http://localhost/' }
-    subject { gitlab_net.send :http_request_for, :get, url }
 
-    before do
-      gitlab_net.send(:config).http_settings.stub(:[]).with('user') { user }
-      gitlab_net.send(:config).http_settings.stub(:[]).with('password') { password }
-      get.should_receive(:basic_auth).with(user, password).once
-      get.should_receive(:set_form_data).with(hash_including(secret_token: 'a123')).once
-    end
+    context 'Unix socket' do
+      it 'sets the Host header to "localhost"' do
+        gitlab_net = described_class.new
+        gitlab_net.should_receive(:secret_token).and_return('a123')
 
-    it { should_not be_nil }
+        request = gitlab_net.send(:http_request_for, :get, URI('http+unix://%2Ffoo'))
+
+        expect(request['Host']).to eq('localhost')
+      end
+    end
   end
 
   describe :cert_store do