Class: DRb::DRbMessage
- Inherits:
-
Object
- Object
- DRb::DRbMessage
- Defined in:
- lib/drb/drb.rb
Overview
Handler for sending and receiving drb messages.
This takes care of the low-level marshalling and unmarshalling of drb requests and responses sent over the wire between server and client. This relieves the implementor of a new drb protocol layer with having to deal with these details.
The user does not have to directly deal with this object in normal use.
Instance Method Summary collapse
-
#dump(obj, error = false) ⇒ Object
:nodoc:.
-
#initialize(config) ⇒ DRbMessage
constructor
:nodoc:.
-
#load(soc) ⇒ Object
:nodoc:.
-
#recv_reply(stream) ⇒ Object
:nodoc:.
-
#recv_request(stream) ⇒ Object
:nodoc:.
-
#send_reply(stream, succ, result) ⇒ Object
:nodoc:.
-
#send_request(stream, ref, msg_id, arg, b) ⇒ Object
:nodoc:.
Constructor Details
#initialize(config) ⇒ DRbMessage
:nodoc:
554 555 556 557 |
# File 'lib/drb/drb.rb', line 554 def initialize(config) # :nodoc: @load_limit = config[:load_limit] @argc_limit = config[:argc_limit] end |
Instance Method Details
#dump(obj, error = false) ⇒ Object
:nodoc:
559 560 561 562 563 564 565 566 567 |
# File 'lib/drb/drb.rb', line 559 def dump(obj, error=false) # :nodoc: obj = make_proxy(obj, error) if obj.kind_of? DRbUndumped begin str = Marshal::dump(obj) rescue str = Marshal::dump(make_proxy(obj, error)) end [str.size].pack('N') + str end |
#load(soc) ⇒ Object
:nodoc:
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 |
# File 'lib/drb/drb.rb', line 569 def load(soc) # :nodoc: begin sz = soc.read(4) # sizeof (N) rescue raise(DRbConnError, $!., $!.backtrace) end raise(DRbConnError, 'connection closed') if sz.nil? raise(DRbConnError, 'premature header') if sz.size < 4 sz = sz.unpack('N')[0] raise(DRbConnError, "too large packet #{sz}") if @load_limit < sz begin str = soc.read(sz) rescue raise(DRbConnError, $!., $!.backtrace) end raise(DRbConnError, 'connection closed') if str.nil? raise(DRbConnError, 'premature marshal format(can\'t read)') if str.size < sz DRb.mutex.synchronize do begin save = Thread.current[:drb_untaint] Thread.current[:drb_untaint] = [] Marshal::load(str) rescue NameError, ArgumentError DRbUnknown.new($!, str) ensure Thread.current[:drb_untaint].each do |x| x.untaint end Thread.current[:drb_untaint] = save end end end |
#recv_reply(stream) ⇒ Object
:nodoc:
636 637 638 639 640 |
# File 'lib/drb/drb.rb', line 636 def recv_reply(stream) # :nodoc: succ = load(stream) result = load(stream) [succ, result] end |
#recv_request(stream) ⇒ Object
:nodoc:
616 617 618 619 620 621 622 623 624 625 626 627 628 |
# File 'lib/drb/drb.rb', line 616 def recv_request(stream) # :nodoc: ref = load(stream) ro = DRb.to_obj(ref) msg = load(stream) argc = load(stream) raise(DRbConnError, "too many arguments") if @argc_limit < argc argv = Array.new(argc, nil) argc.times do |n| argv[n] = load(stream) end block = load(stream) return ro, msg, argv, block end |
#send_reply(stream, succ, result) ⇒ Object
:nodoc:
630 631 632 633 634 |
# File 'lib/drb/drb.rb', line 630 def send_reply(stream, succ, result) # :nodoc: stream.write(dump(succ) + dump(result, !succ)) rescue raise(DRbConnError, $!., $!.backtrace) end |
#send_request(stream, ref, msg_id, arg, b) ⇒ Object
:nodoc:
602 603 604 605 606 607 608 609 610 611 612 613 614 |
# File 'lib/drb/drb.rb', line 602 def send_request(stream, ref, msg_id, arg, b) # :nodoc: ary = [] ary.push(dump(ref.__drbref)) ary.push(dump(msg_id.id2name)) ary.push(dump(arg.length)) arg.each do |e| ary.push(dump(e)) end ary.push(dump(b)) stream.write(ary.join('')) rescue raise(DRbConnError, $!., $!.backtrace) end |