rping 0.1.0

data/README

@@ -0,0 +1,38 @@
+= rping
+
+== Description
+
+rping is a ruby implementation of ping.
+
+== Source Code
+
+https://bitbucket.org/winebarrel/rping
+
+
+== Example
+=== Command
+
+  shell> rping
+  Usage: rping [options]
+      -c COUNT
+      -i INTERVAL
+      -w TIMEOUT
+      -d DESTINATION
+  shell> rping -c 3 -d localhost
+  PING localhost
+  36 bytes from localhost (127.0.0.1): icmp_seq=1 ttl=128 time=1.0 ms
+  36 bytes from localhost (127.0.0.1): icmp_seq=2 ttl=128 time=0.0 ms
+  36 bytes from localhost (127.0.0.1): icmp_seq=3 ttl=128 time=0.0 ms
+
+== Library
+
+  require 'lib/rping'
+  
+  p RPing.ping('localhost')
+  #=> [{:time=>0.0, :dest=>"127.0.0.1", :src=>"127.0.0.1", :ttl=>128, :size=>36, :seq=>1}]
+  
+  RPing.ping('127.0.0.1', :count => 3) do |reply|
+    p reply
+    #=> {:time=>0.0, :src=>"127.0.0.1", :dest=>"127.0.0.1", :ttl=>128, :size=>36, :seq=>1}
+  end
+

data/bin/rping

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+$LOAD_PATH << File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'rubygems'
+require 'lib/rping'
+require 'optparse'
+
+$stdout.sync = true
+
+options = {}
+dest = nil
+
+ARGV.options do |opt|
+  opt.on('-c COUNT')       {|v| options[:count]    = v.to_i }
+  opt.on('-i INTERVAL')    {|v| options[:interval] = v.to_i }
+  opt.on('-w TIMEOUT')     {|v| options[:timeout]  = v.to_i }
+  opt.on('-d DESTINATION') {|v| dest = v }
+
+  opt.parse!
+
+  unless dest
+    puts opt.help
+    exit 1
+  end
+end
+
+puts "PING #{dest}"
+
+RPing.ping(dest, options) do |r|
+  if r
+    puts <<-EOS
+#{r[:size]} bytes from #{dest} (#{r[:dest]}): icmp_seq=#{r[:seq]} ttl=#{r[:ttl]} time=#{'%.1f' % r[:time]} ms
+    EOS
+  else
+    puts 'packet loss'
+  end
+end

data/lib/rping.rb

@@ -0,0 +1,154 @@
+require 'socket'
+require 'timeout'
+
+class RPing
+  MAX_LEN = 64 * 1024
+
+  def initialize(options = {})
+    @count    = options.fetch(:count, 1)
+    @interval = options.fetch(:interval, 1)
+    @timeout  = options.fetch(:timeout, 4)
+    @icmp_id  = Process.pid & 0xffff
+    @seq_num  = 1
+  end
+
+  def self.ping(dest, options = {}, &block)
+    self.new(options).ping(dest, &block)
+  end
+
+  def ping(addr, &block)
+    unless addr =~ /\A\d{3}\.\d{3}\.\d{3}\.\d{3}\Z/
+      addr = IPSocket.getaddress(addr)
+    end
+
+    sock = nil
+    buf = []
+
+    begin
+      sock = make_socket
+      th = ping_recv(sock, buf, &block)
+      ping_send(sock, addr)
+      th.join
+    ensure
+      sock.close if sock
+    end
+
+    return buf
+  end
+
+  def make_socket
+    Socket.new(Socket::AF_INET, Socket::SOCK_RAW, Socket::IPPROTO_ICMP)
+  end
+
+  def ping_send(sock, addr)
+    sockaddr = Socket.pack_sockaddr_in(0, addr)
+
+    @count.times do |i|
+      sleep @interval unless i.zero?
+      req = make_echo_request
+      sock.send(req, 0, sockaddr)
+    end
+  end
+
+  def ping_recv(sock, buf = nil, &block)
+    Thread.start do
+      @count.times do
+        reply = ping_recv0(sock)
+        yield(reply) if block
+        buf << reply if buf
+      end
+    end
+  end
+
+  private
+  def ping_recv0(sock)
+    reply = nil
+
+    begin
+      timeout(@timeout) do
+        if select([sock], nil, nil, @timeout)
+          msg = sock.recv(MAX_LEN)
+          recv_time = Time.now.to_f
+          ip, icmp = unpack_echo_reply(msg)
+
+          # icmp[0] == 0: Type == Echo Reply
+          if icmp[0] == 0 and icmp[3] == @icmp_id
+            reply = {
+              :dest => ip[8].bytes.to_a.join('.'),
+              :src  => ip[9].bytes.to_a.join('.'),
+              :size => msg.length,
+              :ttl  => ip[5],
+              :seq  => icmp[4],
+              :time => (recv_time - icmp[5]) * 1000,
+            }
+          end
+        end
+      end
+    rescue Timeout::Error
+    end
+
+    return reply
+  end
+
+  def make_echo_request
+    data = Time.now.to_f
+
+    req = [
+      [       8, :C], # Type = Echo
+      [       0, :C], # Code = 0
+      [       0, :n], # Checksum
+      [@icmp_id, :n], # Identification
+      [@seq_num, :n], # Sequence Number
+      [    data, :d], # Data
+    ]
+
+    @seq_num += 1
+    req = req.map {|i| i[0] }.pack(req.map {|i| i[1] }.join)
+    update_checksum(req)
+
+    return req
+  end
+
+  def update_checksum(buf)
+    sum = 0
+    buf.unpack('n*').each {|i| sum += i }
+
+    unless (buf.length % 2).zero?
+      sum += buf[-1].unpack('n')[0]
+    end
+
+    sum = (sum & 0xffff) + (sum >> 16);
+    sum = (sum & 0xffff) + (sum >> 16);
+
+    buf[2, 2] = [~sum & 0xffff].pack('n')
+  end
+
+  def unpack_echo_reply(buf)
+    tmpl = [
+      # IP
+      :C,  # Version, IHL
+      :C,  # Type of Service
+      :n,  # Total Length
+      :n,  # Identification 
+      :n,  # Flags, Fragment Offset
+      :C,  # Time to Live,
+      :C,  # Protocol
+      :n,  # Header Checksum
+      :a4, # Source Address
+      :a4, # Destination Address
+      # ICMP
+      :C,  # Type
+      :C,  # Code
+      :n,  # Checksum
+      :n,  # Identification
+      :n,  # Sequence Number
+      :d,  # Data
+    ]
+
+    reply = buf.unpack(tmpl.join)
+    ip = reply.slice(0, 19)
+    icmp = reply.slice(10, 6)
+
+    return [ip, icmp]
+  end
+end

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification 
+name: rping
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- winebarrel
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-01-21 00:00:00 Z
+dependencies: []
+
+description: 
+email: sgwr_dts@yahoo.co.jp
+executables: 
+- rping
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README
+- bin/rping
+- lib/rping.rb
+homepage: https://bitbucket.org/winebarrel/rping
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: rping is a ruby implementation of ping.
+test_files: []
+