File: aproxy.rb

package info (click to toggle)
xemacs21-packages 2009.02.17.dfsg.1-1
  • links: PTS
  • area: main
  • in suites: squeeze
  • size: 116,928 kB
  • ctags: 88,975
  • sloc: lisp: 1,232,060; ansic: 16,570; java: 13,514; xml: 6,477; sh: 4,611; makefile: 4,036; asm: 3,007; perl: 839; cpp: 500; ruby: 257; csh: 96; haskell: 93; awk: 49; python: 47
file content (107 lines) | stat: -rw-r--r-- 2,852 bytes parent folder | download | duplicates (15)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# aproxy.rb --- async proxy between an IRC server and a client
# Copyright (C) 1998-2005 Daiki Ueno

# Author: Daiki Ueno <ueno@unixuser.org>
# Created: 1998-09-28
# Keywords: IRC, riece

# This file is part of Riece.

# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.	 See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with GNU Emacs; see the file COPYING.  If not, write to the
# Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.

require 'io/nonblock'
require 'socket'

class AProxy
  def initialize(host, port, size, back)
    @host = host
    @port = port
    @size = size
    @back = File.new(back, 'a') if back
  end

  def start
    $stdout.nonblock = true
    trap('STOP', 'IGNORE')
    trap('TSTP', 'IGNORE')
    socket = TCPSocket.new(@host, @port)
    $stdout.write("NOTICE CONNECTED #{$$}\r\n")
    wfds_in = []
    buf = ''
    loop do
      rfds, wfds, = select([socket, $stdin], wfds_in)
      unless wfds.empty?
        if buf.length > @size
          s = buf.slice!(0 ... @size)
          @back.write(s) if @back
        end
        begin
          until buf.empty?
            len = $stdout.syswrite(buf)
            buf.slice!(0 .. len)
          end
          wfds_in.clear
        rescue Errno::EAGAIN
        end
      end
      if rfds.include?(socket)
        line = socket.gets("\r\n")
        break unless line
        if line =~ /\A(?::\S+\s+)?PING\s+(.*)\r\n/i
          socket.write("PONG #{$1}\r\n")
        else
          wfds_in = [$stdout]
          buf << line
        end
      end
      if rfds.include?($stdin)
        line = $stdin.gets("\r\n")
        break unless line
        socket.write(line)
      end
    end
    socket.close
  end
end

if $0 == __FILE__
  require 'optparse'

  opt_size, opt_back = nil, nil
  opts = OptionParser.new do |opts|
    opts.banner = <<"End"
Usage: #{$0} [OPTIONS] host port
End
    opts.on('-s', '--size SIZE', 'Size of buffer.') do |size|
      opt_size = size.to_i
    end
    opts.on('-b', '--back BACK', 'Send outdated messages to BACK.') do |back|
      opt_back = back
    end
    opts.on_tail('--help', '-h', 'Show this message.') do
      $stdout.print(opts.to_s)
      exit(0)
    end
  end
  begin
    opts.parse!(ARGV)
  rescue OptionParser::ParseError
    $stderr.print(opts.to_s)
    exit(1)
  end

  AProxy.new(ARGV.shift, ARGV.shift, opt_size, opt_back).start
end