socket_proxy 1.0.0

data/README

@@ -0,0 +1,15 @@
+socket_proxy - Creates I/O pipes for TCP socket tunneling.
+
+
+== Author
+
+Name:: Hiroshi Nakamura
+E-mail:: nahi@ruby-lang.org
+Project web site:: http://github.com/nahi/socket_proxy
+
+
+== License
+
+This program is copyrighted free software by Hiroshi Nakamura.  You can
+redistribute it and/or modify it under the same terms of Ruby's license;
+either the dual license version in 2003, or any later version.

data/bin/socket_proxy

@@ -0,0 +1,63 @@
+#!/usr/bin/env ruby
+
+# socket_proxy.rb -- Creates I/O pipes for TCP socket tunneling.
+# Copyright (C) 1999-2001, 2003 NAKAMURA, Hiroshi
+
+# This application is copyrighted free software by NAKAMURA, Hiroshi.
+# You can redistribute it and/or modify it under the same term as Ruby.
+
+require 'socket_proxy'
+require 'getoptlong'
+
+def main
+  opts = GetoptLong.new(['--debug', '-d', GetoptLong::NO_ARGUMENT], ['--daemon', '-s', GetoptLong::NO_ARGUMENT])
+  opts.each do |name, _|
+    case name
+    when '--debug'
+      debug = true
+    when '--daemon'
+      daemon = true
+    end
+  end
+
+  destname = ARGV.shift
+  portpairs = []
+  while srcport = ARGV.shift
+    destport = ARGV.shift or raise ArgumentError.new("Port must be given as a pair of src and dest.")
+    portpairs << [srcport, destport]
+  end
+  usage if portpairs.empty? or !destname
+
+  # To run as a daemon...
+  if daemon
+    exit! if fork
+    Process.setsid
+    exit! if fork
+    STDIN.close
+    STDOUT.close
+    STDERR.close
+  end
+
+  app = SocketProxy.new(destname, portpairs)
+  app.dump_response = true if debug
+  app.start
+end
+
+def usage
+  STDERR.print <<EOM
+Usage: #{$0} [OPTIONS] destname srcport destport [[srcport destport]...]
+
+    Creates I/O pipes for TCP socket tunneling.
+
+    destname ... hostname of a destination(name or ip-addr).
+    srcport .... source TCP port# or UNIX domain socket name of localhost.
+    destport ... destination port# of the destination host.
+
+  OPTIONS:
+    -d ......... dumps data from destination port(not dumped by default).
+    -s ......... run as a daemon.
+EOM
+  exit 1
+end
+
+main

data/lib/socket_proxy.rb

@@ -0,0 +1,301 @@
+# socket_proxy.rb -- Creates I/O pipes for TCP socket tunneling.
+# Copyright (C) 1999-2001, 2003 NAKAMURA, Hiroshi
+
+# This application is copyrighted free software by NAKAMURA, Hiroshi.
+# You can redistribute it and/or modify it under the same term as Ruby.
+
+# Ruby bundled library
+require 'socket'
+require 'logger'
+
+class SocketProxy < Logger::Application
+  module Dump
+    # Written by Arai-san and published at [ruby-list:31987].
+    # http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-list/31987
+    def hexdump(str)
+      offset = 0
+      result = []
+      while raw = str.slice(offset, 16) and raw.length > 0
+        # data field
+        data = ''
+        for v in raw.unpack('N* a*')
+          if v.kind_of? Integer
+            data << sprintf("%08x ", v)
+          else
+            v.each_byte {|c| data << sprintf("%02x", c) }
+          end
+        end
+        # text field
+        text = raw.tr("\000-\037\177-\377", ".")
+        result << sprintf("%08x  %-36s  %s", offset, data, text)
+        offset += 16
+        # omit duplicate line
+        if /^(#{ Regexp.quote(raw) })+/n =~ str[offset .. -1]
+          result << sprintf("%08x  ...", offset)
+          offset += $&.length
+          # should print at the end
+          if offset == str.length
+            result << sprintf("%08x  %-36s  %s", offset-16, data, text)
+          end
+        end
+      end
+      result
+    end
+    module_function :hexdump
+  end
+
+  include Logger::Severity
+  include Socket::Constants
+
+  attr_accessor :dump_request
+  attr_accessor :dump_response
+
+private
+
+  Timeout = 100			# [sec]
+  ReadBlockSize = 10 * 1024	# [byte]
+
+  class SessionPool
+    def initialize
+      @pool = []
+      @sockets = []
+    end
+
+    def sockets
+      @sockets
+    end
+
+    def each
+      @pool.each do |i|
+	yield i
+      end
+    end
+
+    def add(svrsock, clntsock)
+      @pool.push(Session.new(svrsock, clntsock))
+      @sockets.push(svrsock, clntsock)
+    end
+
+    def del(session)
+      @pool.delete(session)
+      @sockets.delete(session.server)
+      @sockets.delete(session.client)
+    end
+
+  private
+
+    class Session
+      attr(:server)
+      attr(:client)
+
+      def closed?
+        @server.closed? and @client.closed?
+      end
+
+      private
+
+      def initialize(server = nil, client = nil)
+      	@server = server
+      	@client = client
+      end
+    end
+  end
+
+  class ListenSocketHash < Hash
+    def sockets
+      keys
+    end
+  end
+
+  AppName = File.basename(__FILE__)
+  ShiftAge = 0
+  ShiftSize = 0
+
+  def initialize(destname, portpairs)
+    super(AppName)
+    set_log(AppName + '.log', ShiftAge, ShiftSize)
+    @log.level = Logger::INFO
+    @destname = destname
+    @portpairs = portpairs
+    @dump_request = true
+    @dump_response = false
+    @sessionpool = SessionPool.new
+    @waitsockets = nil
+  end
+
+  def init_waitsockets
+    @waitsockets = ListenSocketHash.new
+    @portpairs.each do |srcport, destport|
+      wait = if is_for_tcp(srcport)
+	  TCPServer.new(srcport.to_i)
+	else
+	  UNIXServer.new(srcport)
+	end
+      @waitsockets[wait] = [srcport, destport]
+      dump_start(srcport, destport)
+    end
+  end
+
+  def terminate_waitsockets
+    @waitsockets.each do |sock, portpair|
+      sock.close
+      srcport, destport = portpair
+      File.unlink(srcport) unless is_for_tcp(srcport)
+      dump_end(srcport, destport)
+    end
+  end
+
+  def is_for_tcp(srcport)
+    srcport.to_i != 0
+  end
+
+  def run
+    begin
+      init_waitsockets
+      while true
+	readwait = @sessionpool.sockets + @waitsockets.sockets
+        readready, = IO.select(readwait, nil, nil, Timeout)
+        next unless readready
+        readready.each do |sock|
+	  if (portpair = @waitsockets[sock])
+	    newsock = sock.accept
+	    dump_accept(newsock.peeraddr[2], portpair)
+	    if !add_session(newsock, portpair)
+      	      log(WARN) { 'Closing server socket...' }
+	      newsock.close
+	    end
+	  else
+	    @sessionpool.each do |session|
+	      if sock.equal?(session.server)
+		transfer(session, true)
+		next
+	      elsif sock.equal?(session.client)
+		transfer(session, false)
+		next
+	      end
+	    end
+	  end
+	end
+      end
+    ensure
+      terminate_waitsockets
+    end
+  end
+
+  def transfer(session, is_server)
+    readsock = nil
+    writesock = nil
+    if is_server
+      readsock = session.server
+      writesock = session.client
+    else
+      readsock = session.client
+      writesock = session.server
+    end
+
+    readbuf = nil
+    begin
+      readbuf = readsock.sysread(ReadBlockSize)
+    rescue IOError
+      # not opend for reading
+      close_session(session, readsock, writesock)
+      return
+    rescue EOFError
+      close_session(session, readsock, writesock)
+      return
+    rescue Errno::ECONNRESET
+      log(INFO) { "#{$!} while reading." }
+      close_session(session, readsock, writesock)
+      return
+    rescue
+      log(WARN) { "Detected an exception. Stopping ..." }
+      log(WARN) { $! }
+      log(WARN) { $@ }
+      close_session(session, readsock, writesock)
+      return
+    end
+
+    if is_server
+      dump_transfer_data(true, readbuf) if @dump_request
+    else
+      dump_transfer_data(false, readbuf) if @dump_response
+    end
+
+    writesize = 0
+    while (writesize < readbuf.size)
+      begin
+      	writesize += writesock.syswrite(readbuf[writesize..-1])
+      rescue Errno::ECONNRESET
+      	log(INFO) { "#{$!} while writing." }
+	log(INFO) { $@ }
+        close_session(session, readsock, writesock)
+      	return
+      rescue
+	log(WARN) { "Detected an exception. Stopping ..." }
+	log(WARN) { $@ }
+        close_session(session, readsock, writesock)
+	return
+      end
+    end
+  end
+
+  def add_session(svrsock, portpair)
+    srcport, destport = portpair
+    begin
+      clntsock = TCPSocket.new(@destname, destport)
+    rescue
+      log(ERROR) { 'Create client socket failed.' }
+      return
+    end
+    @sessionpool.add(svrsock, clntsock)
+    dump_add_session(portpair)
+  end
+
+  def close_session(session, readsock, writesock)
+    readsock.close_read
+    writesock.close_write
+    if session.closed?
+      @sessionpool.del(session)
+      dump_close_session
+    end
+  end
+
+  def dump_start(srcport, destport)
+    log(INFO) { 'Started ... src=%s, dest=%s@%s' % [srcport, destport, @destname] }
+  end
+
+  def dump_accept(from, portpair)
+    log(INFO) {
+      srcport, destport = portpair
+      "Accepted ... src=%s from %s" % [srcport, from]
+    }
+  end
+
+  def dump_add_session(portpair)
+    log(INFO) {
+      srcport, destport = portpair
+      'Connection established ... src=%s dest=%s@%s' % [srcport, destport, @destname]
+    }
+  end
+
+  def dump_transfer_data(is_src2dest, data)
+    if is_src2dest
+      log(INFO) { 'Transfer data ... [src] -> [dest]' }
+    else
+      log(INFO) { 'Transfer data ... [src] <- [dest]' }
+    end
+    dump_data(data)
+  end
+
+  def dump_data(data)
+    log(INFO) { "Transferred data;\n" << Dump.hexdump(data).join("\n") }
+  end
+
+  def dump_close_session
+    log(INFO) { 'Connection closed.' }
+  end
+
+  def dump_end(srcport, destport)
+    log(INFO) { 'Stopped ... src=%s, dest=%s@%s' % [srcport, destport, @destname] }
+  end
+end

data/sample/monitor.rb

@@ -0,0 +1,137 @@
+#!/home/achilles/nakahiro/bin/ruby
+#
+# TCP Tunnel
+# Copyright (c) 2001 by Michael Neumann (neumann@s-direktnet.de)
+#
+# $Id: monitor.rb,v 1.2 2001/07/13 04:24:29 nakahiro Exp $
+# 
+# Modified to use TCPSocketPipe and some tk view.
+# Copyright (c) 2001 NAKAMURA Hiroshi.
+
+# This application is copyrighted free software by Michael Neumann and
+# NAKAMURA, Hiroshi.  You can redistribute it and/or modify it under
+# the same term as Ruby or under BSD License.
+
+require 'TCPSocketPipe'
+require 'tk'
+
+
+unless ARGV.size == 3
+  puts "USAGE: #$0 srcPort destName destPort"
+  puts "  e.g. #$0 8070 localhost 8080"
+  exit 1
+end
+
+LISTENHOST = 'localhost'
+LISTENPORT = ARGV.shift
+TUNNELHOST = ARGV.shift
+TUNNELPORT = ARGV.shift
+
+WIDTH  = 50
+HEIGHT = 35
+
+
+root = TkRoot.new { title "TCP Tunnel/Monitor: Tunneling #{LISTENHOST}:#{LISTENPORT} to #{TUNNELHOST}:#{TUNNELPORT}" }
+
+top = TkFrame.new(root) {
+  pack( 'side' => 'top', 'fill' => 'x' )
+}
+
+bottom2 = TkFrame.new(root) {
+  pack( 'side' => 'bottom', 'fill' => 'both' )
+}
+
+bottom3 = TkFrame.new(bottom2) {
+  pack 'side' => 'bottom', 'fill' => 'x'  
+}
+
+bottom  = TkFrame.new(bottom2) {
+  pack( 'side' => 'top', 'fill' => 'both' )
+}
+
+bot_label = TkLabel.new(bottom3) {
+  text "Listening for connections on port #{LISTENPORT} for host #{LISTENHOST}"
+  pack
+}
+
+llabel = TkLabel.new(top) {
+  text "From #{LISTENHOST}:#{LISTENPORT}"
+  pack 'side' => 'right'
+}
+rlabel = TkLabel.new(top) {
+  text "From #{TUNNELHOST}:#{TUNNELPORT}  "
+  pack 'side' => 'left'
+}
+
+$ltext  = TkText.new(bottom, 'width' => WIDTH, 'height' => HEIGHT) {
+  pack( 'side' => 'left', 'fill' => 'y' )
+}
+$rtext  = TkText.new(bottom, 'width' => WIDTH, 'height' => HEIGHT) {
+  pack( 'side' => 'right', 'fill' => 'y' )
+}
+
+scroll = TkScrollbar.new(bottom) {
+  command proc { |arg|
+    $ltext.yview *arg
+    $rtext.yview *arg
+  }
+  pack( 'side' => 'right', 'fill' => 'y' )
+}
+
+$ltext.configure( 'yscrollcommand' => proc { |arg| scroll.set *arg } )
+$ltext.yscrollcommand( proc { |arg| scroll.set *arg } )
+$rtext.configure( 'yscrollcommand' => proc { |arg| scroll.set *arg } )
+$rtext.yscrollcommand( proc { |arg| scroll.set *arg } )
+
+$sessionCount = 0
+$sessionResetP = false
+TkButton.new(top) {
+  text "Clear"
+  command {
+    $ltext.value = ""
+    $rtext.value = "" 
+    $sessionResetP = true
+  }
+  pack
+}
+
+
+class TCPSocketPipe < Application
+  def dumpTransferData( isFromSrcToDestP, data )
+    if isFromSrcToDestP
+      log( SEV_INFO, 'Transfer data ... [src] -> [dest]' )
+      $ltext.insert( 'end', data ) unless $sessionResetP
+    else
+      log( SEV_INFO, 'Transfer data ... [src] <- [dest]' )
+      $rtext.insert( 'end', data ) unless $sessionResetP
+    end
+    dumpData( data )
+  end
+
+  def dumpAddSession
+    $sessionCount += 1
+    str = "--<open: ##{ $sessionCount }>--"
+    $ltext.insert( 'end', str + '-' * ( WIDTH - str.size ) << "\n" )
+    $rtext.insert( 'end', str + '-' * ( WIDTH - str.size ) << "\n" )
+  end
+
+  def dumpCloseSession
+    if $sessionResetP
+      $sessionCount = 0
+      $sessionResetP = false
+      return
+    end
+
+    str = "--<close: ##{ $sessionCount }>--"
+    $ltext.insert( 'end', "\n" << str << '-' * ( WIDTH - str.size ) << "\n" )
+    $rtext.insert( 'end', "\n" << str << '-' * ( WIDTH - str.size ) << "\n" )
+  end
+end
+
+Thread.new {
+  app = TCPSocketPipe.new( LISTENPORT, TUNNELHOST, TUNNELPORT )
+  app.dumpResponse = true
+  app.start()
+}
+
+Tk.mainloop

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: socket_proxy
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Hiroshi Nakamura
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-22 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: nahi@ruby-lang.org
+executables:
+- socket_proxy
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/socket_proxy.rb
+- bin/socket_proxy
+- sample/monitor.rb
+- README
+homepage: http://github.com/nahi/socket_proxy
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Creates I/O pipes for TCP socket tunneling.
+test_files: []