AlkantarClanX12
Current Path : /opt/cpanel/ea-ruby27/root/usr/share/passenger/phusion_passenger/ |
Current File : //opt/cpanel/ea-ruby27/root/usr/share/passenger/phusion_passenger/message_client.rb |
# encoding: binary # Phusion Passenger - https://www.phusionpassenger.com/ # Copyright (c) 2010-2017 Phusion Holding B.V. # # "Passenger", "Phusion Passenger" and "Union Station" are registered # trademarks of Phusion Holding B.V. # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in # all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. require 'socket' PhusionPassenger.require_passenger_lib 'message_channel' PhusionPassenger.require_passenger_lib 'utils' module PhusionPassenger # A convenience class for communicating with MessageServer servers, # for example the ApplicationPool server. class MessageClient include Utils class ProtocolError < StandardError end # Connect to the given server. By default it connects to the current # generation's core. def initialize(username, password, address) @socket = connect_to_server(address) begin @channel = MessageChannel.new(@socket) result = @channel.read if result.nil? raise EOFError elsif result.size != 2 || result[0] != "version" raise IOError, "The message server didn't sent a valid version identifier" elsif result[1] != "1" raise IOError, "Unsupported message server protocol version #{result[1]}" end @channel.write_scalar(username) @channel.write_scalar(password) result = @channel.read if result.nil? raise EOFError elsif result[0] != "status" raise ProtocolError, "Invalid authentication response: expected \"status\", got #{result[0].inspect}" elsif result[1] == "ok" # Do nothing elsif result[1] == "error" if result[2] raise SecurityError, "Authentication error: #{result[2]}" else raise SecurityError, "Authentication error (no server message given)" end else raise ProtocolError, "Invalid authentication response: #{result.inspect}" end rescue Exception @socket.close raise end end def close @socket.close if @socket @channel = @socket = nil end def connected? return !!@channel end ### Low level I/O methods ### def read return @channel.read rescue auto_disconnect raise end def write(*args) @channel.write(*args) rescue auto_disconnect raise end def write_scalar(*args) @channel.write_scalar(*args) rescue auto_disconnect raise end def read_scalar return @channel.read_scalar rescue auto_disconnect raise end def recv_io(klass = IO, negotiate = true) return @channel.recv_io(klass, negotiate) rescue auto_disconnect raise end private def auto_disconnect if @channel @socket.close rescue nil @socket = @channel = nil end end end end # module PhusionPassenger