Violent Ruby: Banner Grabber - Basic Socket Usage with Rescue

basic_socket_usage_with_rescue.rb

# coding: utf-8
# Basic socket usage to grab a banner. 
# @author Kent 'picat' Gruber

require 'socket'

# Grab the banner of a given +ip+ address and +port+
# to attempt to connect to.
#
# @param ip   [String]  Target IP address.
# @param port [Integer] Target port.
#
# @return [String]
def grab_banner(ip, port)
  TCPSocket.new(ip, port).recv(1024)
rescue
  "Unable to establish a connection with #{ip}:#{port}"
end

# Print the result of the method to STDOUT.
puts grab_banner('localhost', 2222)