Sha256: d4b32774ce1cd8b1b26496ceec239472dba5d01fdcf69a48c385283d9d9cd002

Contents?: true

Size: 1.54 KB

Versions: 11

Compression:

Stored size: 1.54 KB

Contents

# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

module Contrast
  module Api
    module Communication
      # Behavior common to all sockets used
      # to communicate with the Contrast Service.
      module Socket
        SOCKET_MONITOR = Monitor.new

        # Send a message across the socket and read back the response.
        # @param marshaled [String] some marshaled form of data to be sent across
        #   the socket. Typically an encoded form of Contrast::Api::Dtm::Message.
        # @return [String] some marshalled form of data returned by the socket.
        #   Typically an encoded form of Contrast::Api::Settings::AgentSettings.
        def send_marshaled marshaled
          SOCKET_MONITOR.synchronize do
            socket = new_socket

            # write message length
            len = marshaled.length
            socket.write([len].pack(Contrast::Api::ENCODING_STRING))

            # write the entire message
            socket.write(marshaled)
            socket.close_write

            # read the response length
            len = socket.read(4).unpack1(Contrast::Api::ENCODING_STRING)

            # read expected response to end
            socket.read(len)
          end
        end

        # Override this method to return a socket.
        # Should be interface compatible with TCPSocket, UNIXSocket, etc.
        def new_socket
          raise NoMethodError, 'This is abstract, override it.'
        end
      end
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
contrast-agent-4.13.1 lib/contrast/api/communication/socket.rb
contrast-agent-4.13.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.12.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.11.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.10.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.9.1 lib/contrast/api/communication/socket.rb
contrast-agent-4.9.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.8.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.7.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.6.0 lib/contrast/api/communication/socket.rb
contrast-agent-4.5.0 lib/contrast/api/communication/socket.rb