pcp-client 0.4.0 → 0.5.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 2ccd94b628705ee8a3f3abcd2a0a91014fe40cd2
+  data.tar.gz: a0243198f1ea56931fe5fdc55d7c487cbba6dbe7
+SHA512:
+  metadata.gz: d77ed2f71cb68af0c14f589bd5e328ff2f193d245e6d8ce933ea07aae718683e78b500628d21ccc2026a614c81644ed44259010b9919e9982fc4e2e3ca2fc4da
+  data.tar.gz: 09655b7b2defada82e89e3da8063e2edf6898a60733fc76f6761859b1024b44efb7966270ad6a367aa966236431599631effd67e219a641c47680782bea0e1b1

data/bin/pcp-ping

@@ -0,0 +1,112 @@
+#!/usr/bin/env ruby
+require 'pcp/client'
+require 'optparse'
+
+ssl_key = '../pcp-broker/test-resources/ssl/private_keys/client01.example.com.pem'
+ssl_cert = '../pcp-broker/test-resources/ssl/certs/client01.example.com.pem'
+ssl_ca_cert = '../pcp-broker/test-resources/ssl/ca/ca_crt.pem'
+debug = false
+server = 'wss://localhost:8142/pcp'
+
+OptionParser.new do |opts|
+  opts.on("-d",  "--debug", "Run noisily") { |v| debug = v }
+  opts.on("--ssl_key FILE", "Use this ssl key") { |v| ssl_key = v }
+  opts.on("--ssl_cert FILE", "Use this ssl cert") { |v| ssl_cert = v }
+  opts.on("--ssl_ca_cert FILE", "Use this ssl ca cert") { |v| ssl_ca_cert = v }
+  opts.on("--server URL", "Server to connect to") { |v| server = v }
+end.parse!
+
+start_time = Time.now.to_f
+
+Thread.new { EM.run }
+Thread.pass until EM.reactor_running?
+
+client = PCP::Client.new({:ssl_key => ssl_key,
+                          :ssl_cert => ssl_cert,
+                          :ssl_ca_cert => ssl_ca_cert,
+                          :loglevel => debug ? Logger::DEBUG : Logger::WARN,
+                          :server => server,
+                         })
+
+# Record the expected set of destinations from the destination_report,
+# then all the responses from the rpc_blocking_response
+mutex = Mutex.new
+have_responses = ConditionVariable.new
+responses = {}
+expected = nil
+ping_time = nil
+
+client.on_message = proc do |message|
+  mutex.synchronize do
+    case message[:message_type]
+    when 'http://puppetlabs.com/destination_report'
+      expected = JSON.load(message.data)['targets']
+    when 'http://puppetlabs.com/rpc_blocking_response'
+      sender = message[:sender]
+      time = Time.now.to_f - ping_time
+      responses[sender] = time
+      puts '%-55s %0.2f seconds' % [sender, time]
+    else
+      p [:unexpected_message, message]
+    end
+
+    have_responses.signal
+  end
+end
+
+client.connect
+
+if !client.associated?
+  puts "Didn't connect to broker."
+  exit 1
+end
+
+connect_time = Time.now.to_f - start_time
+puts "Connected to broker in %0.2f seconds." % [connect_time]
+puts
+
+ping = ::PCP::Message.new(:message_type => 'http://puppetlabs.com/rpc_blocking_request',
+                          :targets => ['pcp://*/agent'],
+                          :destination_report => true)
+
+ping.data = {:transaction_id => ping[:id],
+             :module => 'echo',
+             :action => 'echo',
+             :params => {:argument => 'echo'}}.to_json
+
+ping.expires(3)
+
+ping_time = Time.now.to_f
+client.send(ping)
+
+begin
+  Timeout::timeout(3) do
+    done = false
+    loop do
+      mutex.synchronize do
+        have_responses.wait(mutex)
+        if expected && expected.all? { |from| responses.include?(from) }
+          # Have all responses we expect, bail
+          run_time = Time.now.to_f - start_time
+          puts
+          puts "Received all responses in %0.2f seconds." % [run_time]
+          done = true
+        end
+      end
+      break if done
+    end
+  end
+rescue Timeout::Error
+  puts
+  if expected
+    puts "Received %d/%d responses.  Following nodes were missing" % [responses.keys.size, expected.size]
+    puts
+    expected.select { |from| !responses.include?(from) }.each do |missing|
+      puts "    #{missing}"
+    end
+  else
+    puts "Didn't get destination report, not sure how many responses to expect.  We got %d." % [responses.keys.size]
+  end
+end
+
+EM.stop

data/lib/pcp-client/version.rb

@@ -0,0 +1,4 @@
+class PCPClient
+  VERSION = '0.5.1'.freeze
+end
+

data/lib/pcp/client.rb

@@ -2,6 +2,7 @@ require 'eventmachine'
 require 'faye/websocket'
 require 'pcp/message'
 require 'logger'
+require 'openssl'
 
 # So EventMachine when you specify :verify_peer => true in the TLS
 # options decides what that means is it should just fire off a
@@ -80,6 +81,7 @@ module PCP
       type = params[:type] || "ruby-pcp-client-#{$$}"
       @identity = make_identity(@ssl_cert, type)
       @on_message = params[:on_message]
+      @max_message_size = params[:max_message_size] || 64*1024*1024
       @associated = false
     end
 
@@ -114,7 +116,9 @@ module PCP
           :xxx_hostname => URI.parse(@server).host,
         }
 
-        @connection = Faye::WebSocket::Client.new(@server, nil, {:tls => start_tls_options})
+        @connection = Faye::WebSocket::Client.new(@server, nil, {:tls => start_tls_options,
+                                                                 :ping => 30,
+                                                                 :max_length => @max_message_size})
 
         @connection.on :open do |event|
           begin
@@ -205,6 +209,9 @@ module PCP
     def send(message)
       EM.next_tick do
         @logger.debug { [:send, message] }
+        if message[:expires].nil?
+          message.expires(3)
+        end
         message[:sender] = identity
         @connection.send(message.encode)
       end

data/lib/pcp/message.rb

@@ -111,11 +111,19 @@ module PCP
         chunks << frame_chunk(i + 2, @chunks[i])
       end
 
-      RSchema.validate!(PCP::Protocol::Envelope, envelope)
+      validate
 
       [1, frame_chunk(1, envelope.to_json), chunks].flatten
     end
 
+    # Validate the data in message against schema
+    #
+    # @api public
+    # @return ignore
+    def validate
+      RSchema.validate!(PCP::Protocol::Envelope, envelope)
+    end
+
     private
 
     # Decodes an array of bytes into the message
@@ -144,7 +152,7 @@ module PCP
           parsed.each do |k,v|
             @envelope[k.to_sym] = v
           end
-          RSchema.validate!(PCP::Protocol::Envelope, @envelope)
+          validate
         else
           @chunks[type - 2] = body
         end

metadata

@@ -1,98 +1,93 @@
 --- !ruby/object:Gem::Specification
 name: pcp-client
 version: !ruby/object:Gem::Version
-  version: 0.4.0
-  prerelease: 
+  version: 0.5.1
 platform: ruby
 authors:
 - Puppet Labs
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-04-18 00:00:00.000000000 Z
+date: 2018-09-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: eventmachine
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.2'
 - !ruby/object:Gem::Dependency
   name: faye-websocket
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '0.10'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '0.10'
 - !ruby/object:Gem::Dependency
   name: rschema
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.3'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.3'
 description: See https://github.com/puppetlabs/pcp-specifications
 email: puppet@puppetlabs.com
-executables: []
+executables:
+- pcp-ping
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/pcp-ping
+- lib/pcp-client/version.rb
+- lib/pcp.rb
 - lib/pcp/client.rb
 - lib/pcp/message.rb
 - lib/pcp/protocol.rb
 - lib/pcp/simple_logger.rb
-- lib/pcp.rb
 homepage: https://github.com/puppetlabs/ruby-pcp-client
 licenses:
 - ASL 2.0
+metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.23.2
+rubygems_version: 2.5.1
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: Client library for PCP
 test_files: []