yieldbot/sensu-yieldbot-plugins

#! /usr/bin/env ruby
#
# check-ssh
#
# DESCRIPTION:
#   Check the status and version of SSH
#
# OUTPUT:
#   plain-text
#
# PLATFORMS:
#   Linux
#
# DEPENDENCIES:
#   gem: socket
#   gem: sensu-plugin
#
# USAGE:
#   check-ssh.rb -h <hostname> -p port
#
# NOTES:
#
# LICENSE:
#   Copyright 2014 Yieldbot, Inc  <devops@yieldbot.com>
#   Released under the same terms as Sensu (the MIT license); see LICENSE
#   for details.
#

require 'rubygems' if RUBY_VERSION < '1.9.0'
require 'sensu-plugin/check/cli'
require 'socket'

#
# == Check SSH
#
class CheckSSH < Sensu::Plugin::Check::CLI
  option :hostname,
         description: 'The host you wish to connect to',
         short:       '-h HOSTNAME',
         long:        '--hostname HOSTNAME',
         default:     'localhost'

  option :port,
         description: 'The port you wish to connect to',
         short:       '-p PORT',
         long:        '--PORT PORT',
         default:     '22'

  # Open a socket to the host and port specified and return back the
  # header.  If the header doesn't contain the sting then crit, else ok
  #
  def run
    s = TCPSocket.open(config[:hostname], config[:port]).gets.chop
    critical unless s.include?('SSH')
    ok(s)
  end
end