cgit.sxemacs.org Git - riece/blob - lisp/server.rb

   1 # A simple IPC server executing Ruby programs.
   2
   3 require 'thread'
   4
   5 class Server
   6   module B
   7   end
   8
   9   def initialize
  10     @buf = ''
  11     @que = Queue.new
  12     @thr = Hash.new
  13     @cnt = 0
  14   end
  15
  16   def dispatch(line)
  17     case line.chomp
  18     when /\AD /
  19       @buf << $'
  20     when /\A(\S+)\s*/
  21       c = $1
  22       r = $'
  23       d = "dispatch_#{c.downcase}"
  24       if respond_to?(d, true)
  25         Thread.start do
  26           self.send(d, c, r)
  27         end
  28       else
  29         puts("ERR 103 Unknown command\r\n")
  30       end
  31     end
  32   end
  33
  34   def dispatch_cancel(c, r)
  35     puts("ERR 100 Not implemented\r\n")
  36   end
  37
  38   def dispatch_bye(c, r)
  39     puts("ERR 100 Not implemented\r\n")
  40   end
  41
  42   def dispatch_auth(c, r)
  43     puts("ERR 100 Not implemented\r\n")
  44   end
  45
  46   def dispatch_reset(c, r)
  47     puts("ERR 100 Not implemented\r\n")
  48   end
  49
  50   def dispatch_end(c, r)
  51     enq_data
  52   end
  53
  54   def dispatch_help(c, r)
  55     puts("ERR 100 Not implemented\r\n")
  56   end
  57
  58   def dispatch_quit(c, r)
  59     puts("ERR 100 Not implemented\r\n")
  60   end
  61
  62   def dispatch_eval(c, r)
  63     r = deq_data if r.empty?
  64     name = nil
  65     Thread.exclusive do
  66       while @thr.include?(name = @cnt.to_s)
  67         @cnt += 1
  68       end
  69       @thr[name] = Thread.current
  70     end
  71     puts("S name #{name}\r\n")
  72     puts("OK\r\n")
  73     Thread.current[:rubyserv_name] = name
  74     begin
  75       Thread.current[:rubyserv_error] = false
  76       Thread.current[:rubyserv_response] = eval(r, B.module_eval('binding()'))
  77     rescue Exception => e
  78       Thread.current[:rubyserv_error] = true
  79       Thread.current[:rubyserv_response] = e
  80     end
  81     puts("# exit #{name}\r\n")
  82   end
  83
  84   def dispatch_poll(c, r)
  85     thr = @thr[r]
  86     if !thr
  87       puts("ERR 105 Parameter error: no such name \"#{r}\"\r\n")
  88     elsif thr.alive?
  89       puts("S running #{r}\r\n")
  90       puts("OK\r\n")
  91     else
  92       if thr[:rubyserv_error]
  93         puts("S exited #{r}\r\n")
  94       else
  95         puts("S finished #{r}\r\n")
  96       end
  97       if d = thr[:rubyserv_response]
  98         send_data(d.to_s)
  99       end
 100       puts("OK\r\n")
 101     end
 102   end
 103
 104   def dispatch_exit(c, r)
 105     thr = @thr[r]
 106     if !thr
 107       puts("ERR 105 Parameter error: no such name \"#{r}\"\r\n")
 108       return
 109     end
 110     thr.kill if thr.alive?
 111     @thr.delete(r)
 112     puts("OK\r\n")
 113   end
 114
 115   def escape(s)
 116     s.gsub(/[%\r\n]/) {|m| '%%%02X' % m[0]}
 117   end
 118
 119   def unescape(s)
 120     s.gsub(/%([0-9A-Z][0-9A-Z])/, ['\1'].pack('H*'))
 121   end
 122
 123   def output(s)
 124     puts("# output #{Thread.current[:rubyserv_name]} #{s}\r\n")
 125   end
 126
 127   def send_data(d)
 128     d = escape(d)
 129     begin
 130       len = [d.length, 998].min   # 998 = 1000 - "D "
 131       puts("D #{d[0 ... len]}\r\n")
 132       d = d[len .. -1]
 133     end until d.empty?
 134   end
 135
 136   def enq_data
 137     d = unescape(@buf)
 138     @buf = ''
 139     @que.enq(d)
 140   end
 141
 142   def deq_data
 143     @que.deq
 144   end
 145 end
 146
 147 if $0 == __FILE__
 148   server = Server.new
 149   while gets
 150     server.dispatch($_)
 151   end
 152 end