Skip to content

Commit

Permalink
Merge pull request #7611 from Poohblah/master
Browse files Browse the repository at this point in the history
fix for #7610 (incorrect permissions on ~/.ssh/authorized_keys causes authentication failure after insecure keypair replacement)
  • Loading branch information
sethvargo authored Jul 20, 2016
2 parents 9ab0d4b + 4106320 commit 2418cb1
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 1 deletion.
3 changes: 2 additions & 1 deletion plugins/guests/linux/cap/public_key.rb
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def self.remove_public_key(machine, contents)
contents = contents.strip << "\n"

remote_path = "/tmp/vagrant-remove-pubkey-#{Time.now.to_i}"
Tempfile.open("vagrant-bsd-remove-public-key") do |f|
Tempfile.open("vagrant-linux-remove-public-key") do |f|
f.binmode
f.write(contents)
f.fsync
Expand All @@ -54,6 +54,7 @@ def self.remove_public_key(machine, contents)
if test -f ~/.ssh/authorized_keys; then
grep -v -x -f '#{remote_path}' ~/.ssh/authorized_keys > ~/.ssh/authorized_keys.tmp
mv ~/.ssh/authorized_keys.tmp ~/.ssh/authorized_keys
chmod 0600 ~/.ssh/authorized_keys
fi
rm -f '#{remote_path}'
Expand Down
32 changes: 32 additions & 0 deletions test/unit/plugins/guests/linux/cap/remove_public_key_test.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
require_relative "../../../../base"

describe "VagrantPlugins::GuestLinux::Cap::RemovePublicKey" do
let(:caps) do
VagrantPlugins::GuestLinux::Plugin
.components
.guest_capabilities[:linux]
end

let(:machine) { double("machine") }
let(:comm) { VagrantTests::DummyCommunicator::Communicator.new(machine) }

before do
allow(machine).to receive(:communicate).and_return(comm)
end

after do
comm.verify_expectations!
end

describe ".remove_public_key" do
let(:cap) { caps.get(:remove_public_key) }

it "removes the public key" do
cap.remove_public_key(machine, "ssh-rsa ...")
expect(comm.received_commands[0]).to match(/grep -v -x -f '\/tmp\/vagrant-(.+)' ~\/\.ssh\/authorized_keys > ~\/.ssh\/authorized_keys\.tmp/)
expect(comm.received_commands[0]).to match(/mv ~\/.ssh\/authorized_keys\.tmp ~\/.ssh\/authorized_keys/)
expect(comm.received_commands[0]).to match(/chmod 0600 ~\/.ssh\/authorized_keys/)
expect(comm.received_commands[0]).to match(/rm -f '\/tmp\/vagrant-(.+)'/)
end
end
end

0 comments on commit 2418cb1

Please sign in to comment.