lib/shuttle/runner.rb in shuttle-deploy-0.2.0.beta7 vs lib/shuttle/runner.rb in shuttle-deploy-0.2.0.beta9

@@ -153,9 +153,12 @@
       else
         raise ConfigError, "Invalid command: #{command}"
       end
 
       ssh.close
+    rescue SocketError => err
+      STDERR.puts "Socket error: #{err.message}"
+      exit 1
     rescue Net::SSH::AuthenticationFailed
       STDERR.puts "SSH Authentication failed"
       exit 1
     end
   end
\ No newline at end of file