Class: DRb::DRbSSHServerConn

Inherits:
Object
  • Object
show all
Defined in:
lib/drbssh.rb

Overview

DRbSSH protocol server per-connection class. Handles client-to-client communications by utilizing thread-safe Queue’s for the input and output, and by adding a small ‘rep’ or ‘req’ packet before sending, so we can have two-way DRb duplexed over a single pair of filedescriptors.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(uri, config, client) ⇒ DRbSSHServerConn

Create a new server-connection for the specified client.



208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
 
# File 'lib/drbssh.rb', line 208

def initialize(uri, config, client)
  @uri = uri
  @client = client
  @srv_requestq = Queue.new

  msg = DRbMessage.new(config)

  # Read-thread
  Thread.new do
    # Read from client, and delegate request/reply to the correct place.
    begin
      loop do
        type = msg.load(client.read_fd)
        if type == 'req'
          @srv_requestq.push(msg.recv_request(client.read_fd))
        else
          client.receiveq.push(msg.recv_reply(client.read_fd))
        end
      end
    rescue
      client.receiveq.push($!)
      @srv_requestq.push($!)
    end
  end

  # Write-thread
  Thread.new do
    # Wait for outgoing data on send queue, and add header-packet before
    # writing.
    begin
      loop do
        type, data = client.sendq.pop

        client.write_fd.write(msg.dump(type))

        if type == 'req'
          msg.send_request(client.write_fd, *data)
        else
          msg.send_reply(client.write_fd, *data)
        end
      end
    rescue
      client.receiveq.push($!)
      @srv_requestq.push($!)
    end
  end
end

Instance Attribute Details

#uri ⇒ Object (readonly)

Returns the value of attribute uri.



205
206
207
 
# File 'lib/drbssh.rb', line 205

def uri
  @uri
end

Instance Method Details

#close ⇒ Object

Delegate shutdown to client.



257
258
259
260
261
262
263
264
265
 
# File 'lib/drbssh.rb', line 257

def close
  return unless @client.alive?

  Timeout::timeout(15) do
    sleep 0.1 until @client.sendq.empty?
  end rescue nil

  @client.close
end

#recv_request ⇒ Object

Wait for a request to appear on the request-queue



268
269
270
271
272
273
274
275
276
 
# File 'lib/drbssh.rb', line 268

def recv_request
  reply = @srv_requestq.pop
  if reply.is_a? Exception
    self.close
    raise reply
  else
    reply
  end
end

#send_reply(succ, result) ⇒ Object

Queue client-reply



279
280
281
 
# File 'lib/drbssh.rb', line 279

def send_reply(succ, result)
  @client.sendq.push(['rep', [succ, result]])
end