opscode-pushy-client 2.3.0

data/lib/pushy_client/version.rb

@@ -0,0 +1,23 @@
+# @copyright Copyright 2014-2017, Chef Software Inc. All Rights Reserved.
+#
+# This file is provided to you under the Apache License,
+# Version 2.0 (the "License"); you may not use this file
+# except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+
+# Note: the version must also be updated in
+# omnibus/config/projects/push-jobs-client.rb
+class PushyClient
+  VERSION = "2.3.0"
+  PROTOCOL_VERSION = "2.0"
+end

data/lib/pushy_client/whitelist.rb

@@ -0,0 +1,66 @@
+# @copyright Copyright 2014 Chef Software, Inc. All Rights Reserved.
+#
+# This file is provided to you under the Apache License,
+# Version 2.0 (the "License"); you may not use this file
+# except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+
+require "chef/log"
+
+class PushyClient
+  class Whitelist
+
+    attr_accessor :whitelist
+
+    def initialize(whitelist)
+      @whitelist = whitelist
+    end
+
+    def [](argument)
+      command = process(argument)
+      node_name = "UNKNOWN"
+      job_id = "UNKNOWN"
+      Chef::Log.info("[#{node_name}] Job #{job_id}: whitelist '#{argument}' to '#{command}'")
+      command
+    end
+
+    def process(argument)
+      # If we have an exact match, use it
+      if whitelist.has_key?(argument)
+        return whitelist[argument]
+      else
+        whitelist.keys.each do |key|
+          if key.kind_of?(Regexp)
+            if key.match(argument)
+              value = whitelist[key]
+              if value.kind_of?(Hash)
+                # Need a deep copy, don't want to change the global value
+                new_value = Marshal.load(Marshal.dump(value))
+                new_value[:command_line] = argument.gsub(key, value[:command_line])
+                return new_value
+              else
+                return argument.gsub(key, value)
+              end
+            end
+          end
+        end
+      end
+
+      nil
+    end
+
+    def method_missing(method, *args, &block)
+      @whitelist.send(method, *args, &block)
+    end
+  end
+end

data/lib/pushy_client/win32.rb

@@ -0,0 +1,27 @@
+#
+# Author:: John Keiser (<jkeiser@opscode.com>)
+# Copyright:: Copyright (c) 2012 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'win32/process'
+
+module Process
+  extend FFI::Library
+  # Override WaitForSingleObject with :blocking => true, so that Process.wait
+  # (from win32-process) will not block the Ruby interpreter while waiting for
+  # the process to complete.
+  ffi_lib :kernel32
+  attach_function :WaitForSingleObject, [:ulong, :ulong], :ulong, :blocking => true
+end

data/lib/pushy_client/windows_service.rb

@@ -0,0 +1,253 @@
+#
+# Author:: Christopher Maier (<maier@lambda.local>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/application'
+require 'chef/config'
+require 'chef/log'
+require 'chef/rest'
+require 'mixlib/cli'
+require 'win32/daemon'
+require_relative '../pushy_client'
+require_relative '../pushy_client/cli'
+require_relative '../pushy_client/version'
+
+class PushyClient
+  class WindowsService < ::Win32::Daemon
+    include Mixlib::CLI
+
+    option :config_file,
+      :short => "-c CONFIG",
+      :long => "--config CONFIG",
+      :default => PushyClient::CLI.find_default_config,
+      :description => "The configuration file to use"
+
+    option :log_location,
+      :short        => "-L LOGLOCATION",
+      :long         => "--logfile LOGLOCATION",
+      :description  => "Set the log file location",
+      :default => "#{ENV['SYSTEMDRIVE']}/chef/push-client.log"
+
+    option :allow_unencrypted,
+      :long        => "--allow_unencrypted",
+      :boolean     => true,
+      :description => "Allow unencrypted connections to 1.x servers"
+
+    def service_init
+      @service_action_mutex = Mutex.new
+      @service_signal = ConditionVariable.new
+
+      reconfigure
+      Chef::Log.info("Chef Push Jobs Client Service initialized")
+    end
+
+    def service_main(*startup_parameters)
+      begin
+        @service_action_mutex.synchronize do
+          Chef::Log.info("Push Client version: #{::PushyClient::VERSION}")
+          Chef::Log.info("Push Client started as service with parameters: #{startup_parameters}")
+          Chef::Log.info("Push Client passed: #{ARGV.join(', ')}")
+          reconfigure(startup_parameters)
+
+          # Lifted from PushyClient::CLI
+          ohai = Ohai::System.new
+          ohai.load_plugins
+          ohai.run_plugins(true, ['os', 'hostname'])
+
+          @client = PushyClient.new(
+                                    :chef_server_url => Chef::Config[:chef_server_url],
+                                    :client_key      => Chef::Config[:client_key],
+                                    :node_name       => Chef::Config[:node_name] || ohai[:fqdn] || ohai[:hostname],
+                                    :whitelist       => Chef::Config[:whitelist] || { 'chef-client' => 'chef-client' },
+                                    :hostname        => ohai[:hostname],
+                                    :allow_unencrypted => Chef::Config[:allow_unencrypted]
+                                    )
+
+          @client.start
+          Chef::Log.info("pushy-client is started...")
+
+          # Wait until we get service exit signal
+          @service_signal.wait(@service_action_mutex)
+          Chef::Log.debug("Stopping pushy-client...")
+          @client.stop
+        end
+
+      rescue Exception => e
+        Chef::Log.error("#{e.class}: #{e}")
+        Chef::Log.error("Terminating pushy-client service...")
+        Chef::Application.debug_stacktrace(e)
+      end
+
+      Chef::Log.debug("Exiting service...")
+    end
+
+    ################################################################################
+    # Control Signal Callback Methods
+    ################################################################################
+
+    def service_stop
+      Chef::Log.info("STOP request from operating system.")
+      while !@service_action_mutex.try_lock do
+        Chef::Log.info("Pushy is being initialized waiting for initialization to complete.")
+        sleep(1)
+      end
+
+      @service_signal.signal
+      @service_action_mutex.unlock
+    end
+
+    def service_pause
+      Chef::Log.info("PAUSE request from operating system.")
+      Chef::Log.info("Pushy Client Service doesn't support PAUSE.")
+      Chef::Log.info("Pushy Client Service is still running.")
+    end
+
+    def service_resume
+      Chef::Log.info("RESUME signal received from the OS.")
+    end
+
+    def service_shutdown
+      Chef::Log.info("SHUTDOWN signal received from the OS.")
+
+      # Treat shutdown similar to stop.
+
+      service_stop
+    end
+
+    ################################################################################
+    # Internal Methods
+    ################################################################################
+
+    private
+
+    def apply_config(config_file_path)
+      Chef::Config.from_file(config_file_path)
+      Chef::Config.merge!(config)
+    end
+
+    # Lifted from Chef::Application, with addition of optional startup parameters
+    # for playing nicely with Windows Services and logic from PushyClient::CLI
+    def reconfigure(startup_parameters=[])
+      Chef::Config[:force_logger] = true
+
+      configure_chef startup_parameters
+      configure_logging
+    end
+
+    # Lifted from application.rb
+    # See application.rb for related comments.
+
+    def configure_logging
+      Chef::Log.init(Chef::Config[:log_location])
+      if want_additional_logger?
+        configure_stdout_logger
+      end
+      Chef::Log.level = resolve_log_level
+    end
+
+    def configure_stdout_logger
+      stdout_logger = Logger.new(STDOUT)
+      STDOUT.sync = true
+      stdout_logger.formatter = Chef::Log.logger.formatter
+      Chef::Log.loggers <<  stdout_logger
+    end
+
+    # Based on config and whether or not STDOUT is a tty, should we setup a
+    # secondary logger for stdout?
+    def want_additional_logger?
+      ( Chef::Config[:log_location] != STDOUT ) && STDOUT.tty? && (!Chef::Config[:daemonize]) && (Chef::Config[:force_logger])
+    end
+
+    # Use of output formatters is assumed if `force_formatter` is set or if
+    # `force_logger` is not set and STDOUT is to a console (tty)
+    def using_output_formatter?
+      Chef::Config[:force_formatter] || (!Chef::Config[:force_logger] && STDOUT.tty?)
+    end
+
+    def auto_log_level?
+      Chef::Config[:log_level] == :auto
+    end
+
+    # if log_level is `:auto`, convert it to :warn (when using output formatter)
+    # or :info (no output formatter). See also +using_output_formatter?+
+    def resolve_log_level
+      if auto_log_level?
+        if using_output_formatter?
+          :warn
+        else
+          :info
+        end
+      else
+        Chef::Config[:log_level]
+      end
+    end
+
+    def configure_chef(startup_parameters)
+      # Bit of a hack ahead:
+      # It is possible to specify a service's binary_path_name with arguments, like "foo.exe -x argX".
+      # It is also possible to specify startup parameters separately, either via the the Services manager
+      # or by using the registry (I think).
+
+      # In order to accommodate all possible sources of parameterization, we first parse any command line
+      # arguments.  We then parse any startup parameters.  This works, because Mixlib::CLI reuses its internal
+      # 'config' hash; thus, anything in startup parameters will override any command line parameters that
+      # might be set via the service's binary_path_name
+      #
+      # All these parameters then get layered on top of those from Chef::Config
+
+      parse_options # Operates on ARGV by default
+      parse_options startup_parameters
+
+      Chef::Log.info("Push Client using default config file path: '#{config[:config_file]}'")
+
+      begin
+        case config[:config_file]
+        when /^(http|https):\/\//
+          begin
+            # First we will try Chef::HTTP::SimpleJSON as preference to Chef::REST
+            require 'chef/http/simple_json'
+            Chef::HTTP::SimpleJSON.new("").streaming_request(config[:config_file]) { |f| apply_config(f.path) }
+          rescue LoadError
+            require 'chef/rest'
+            Chef::REST.new("", nil, nil).fetch(config[:config_file]) { |f| apply_config(f.path) }
+          end
+        else
+          ::File::open(config[:config_file]) { |f| apply_config(f.path) }
+        end
+      rescue Errno::ENOENT => error
+        Chef::Log.warn("*****************************************")
+        Chef::Log.warn("Did not find config file: #{config[:config_file]}, using command line options.")
+        Chef::Log.warn("*****************************************")
+
+        Chef::Config.merge!(config)
+      rescue SocketError => error
+        Chef::Application.fatal!("Error getting config file #{Chef::Config[:config_file]}", 2)
+      rescue Chef::Exceptions::ConfigurationError => error
+        Chef::Application.fatal!("Error processing config file #{Chef::Config[:config_file]} with error #{error.message}", 2)
+      rescue Exception => error
+        Chef::Application.fatal!("Unknown error processing config file #{Chef::Config[:config_file]} with error #{error.message}", 2)
+      end
+    end
+
+  end
+end
+
+# To run this file as a service, it must be called as a script from within
+# the Windows Service framework.  In that case, kick off the main loop!
+if __FILE__ == $0
+    PushyClient::WindowsService.mainloop
+end

data/omnibus/Berksfile

@@ -0,0 +1,12 @@
+source 'https://supermarket.chef.io'
+
+cookbook 'omnibus'
+
+# Uncomment to use the latest version of the Omnibus cookbook from GitHub
+# cookbook 'omnibus', github: 'opscode-cookbooks/omnibus'
+
+group :integration do
+  cookbook 'apt',      '~> 2.3'
+  cookbook 'freebsd',  '~> 0.1'
+  cookbook 'yum-epel', '~> 0.3'
+end

data/omnibus/Gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+gem 'omnibus', git: 'https://github.com/chef/omnibus.git'
+gem 'omnibus-software', git: 'https://github.com/chef/omnibus-software.git'
+
+# This development group is installed by default when you run `bundle install`,
+# but if you are using Omnibus in a CI-based infrastructure, you do not need
+# the Test Kitchen-based build lab. You can skip these unnecessary dependencies
+# by running `bundle install --without development` to speed up build times.
+group :development do
+  gem 'berkshelf'
+  gem 'test-kitchen'
+  gem 'kitchen-vagrant'
+  gem 'winrm-fs'
+end

data/omnibus/Gemfile.lock

@@ -0,0 +1,232 @@
+GIT
+  remote: https://github.com/chef/omnibus-software.git
+  revision: 69e7fd212b293795c9a1abf7e2aed3be89ac6d06
+  specs:
+    omnibus-software (4.0.0)
+      chef-sugar (>= 3.4.0)
+      omnibus (>= 5.5.0)
+
+GIT
+  remote: https://github.com/chef/omnibus.git
+  revision: 433220e0b7c434dbc4a36daaa1fecbdb1bf7231d
+  specs:
+    omnibus (5.5.0)
+      aws-sdk (~> 2)
+      chef-sugar (~> 3.3)
+      cleanroom (~> 1.0)
+      ffi-yajl (~> 2.2)
+      license_scout
+      mixlib-shellout (~> 2.0)
+      mixlib-versioning
+      ohai (~> 8.0)
+      pedump
+      ruby-progressbar (~> 1.7)
+      thor (~> 0.18)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.5.0)
+      public_suffix (~> 2.0, >= 2.0.2)
+    artifactory (2.8.1)
+    awesome_print (1.7.0)
+    aws-sdk (2.8.14)
+      aws-sdk-resources (= 2.8.14)
+    aws-sdk-core (2.8.14)
+      aws-sigv4 (~> 1.0)
+      jmespath (~> 1.0)
+    aws-sdk-resources (2.8.14)
+      aws-sdk-core (= 2.8.14)
+    aws-sigv4 (1.0.0)
+    berkshelf (5.6.4)
+      addressable (~> 2.3, >= 2.3.4)
+      berkshelf-api-client (>= 2.0.2, < 4.0)
+      buff-config (~> 2.0)
+      buff-extensions (~> 2.0)
+      buff-shell_out (~> 1.0)
+      cleanroom (~> 1.0)
+      faraday (~> 0.9)
+      httpclient (~> 2.7)
+      minitar (~> 0.5, >= 0.5.4)
+      mixlib-archive (~> 0.4)
+      octokit (~> 4.0)
+      retryable (~> 2.0)
+      ridley (~> 5.0)
+      solve (> 2.0, < 4.0)
+      thor (~> 0.19, < 0.19.2)
+    berkshelf-api-client (3.0.0)
+      faraday (~> 0.9)
+      httpclient (~> 2.7)
+      ridley (>= 4.5, < 6.0)
+    buff-config (2.0.0)
+      buff-extensions (~> 2.0)
+      varia_model (~> 0.6)
+    buff-extensions (2.0.0)
+    buff-ignore (1.2.0)
+    buff-ruby_engine (1.0.0)
+    buff-shell_out (1.1.0)
+      buff-ruby_engine (~> 1.0)
+    builder (3.2.3)
+    celluloid (0.16.0)
+      timers (~> 4.0.0)
+    celluloid-io (0.16.2)
+      celluloid (>= 0.16.0)
+      nio4r (>= 1.1.0)
+    chef-config (12.19.36)
+      addressable
+      fuzzyurl
+      mixlib-config (~> 2.0)
+      mixlib-shellout (~> 2.0)
+    chef-sugar (3.4.0)
+    cleanroom (1.0.0)
+    erubis (2.7.0)
+    faraday (0.9.2)
+      multipart-post (>= 1.2, < 3)
+    ffi (1.9.18)
+    ffi-yajl (2.3.0)
+      libyajl2 (~> 1.2)
+    fuzzyurl (0.9.0)
+    gssapi (1.2.0)
+      ffi (>= 1.0.1)
+    gyoku (1.3.1)
+      builder (>= 2.1.2)
+    hashie (3.5.5)
+    hitimes (1.2.4)
+    httpclient (2.8.3)
+    iostruct (0.0.4)
+    ipaddress (0.8.3)
+    jmespath (1.3.1)
+    json (2.0.3)
+    kitchen-vagrant (1.0.2)
+      test-kitchen (~> 1.4)
+    libyajl2 (1.2.0)
+    license_scout (0.1.3)
+      ffi-yajl (~> 2.2)
+      mixlib-shellout (~> 2.2)
+    little-plugger (1.1.4)
+    logging (2.2.0)
+      little-plugger (~> 1.1)
+      multi_json (~> 1.10)
+    minitar (0.6.1)
+    mixlib-archive (0.4.1)
+      mixlib-log
+    mixlib-authentication (1.4.1)
+      mixlib-log
+    mixlib-cli (1.7.0)
+    mixlib-config (2.2.4)
+    mixlib-install (2.1.12)
+      artifactory
+      mixlib-shellout
+      mixlib-versioning
+      thor
+    mixlib-log (1.7.1)
+    mixlib-shellout (2.2.7)
+    mixlib-versioning (1.1.0)
+    molinillo (0.5.7)
+    multi_json (1.12.1)
+    multipart-post (2.0.0)
+    net-scp (1.2.1)
+      net-ssh (>= 2.6.5)
+    net-ssh (4.1.0)
+    net-ssh-gateway (1.3.0)
+      net-ssh (>= 2.6.5)
+    nio4r (2.0.0)
+    nori (2.6.0)
+    octokit (4.6.2)
+      sawyer (~> 0.8.0, >= 0.5.3)
+    ohai (8.23.0)
+      chef-config (>= 12.5.0.alpha.1, < 13)
+      ffi (~> 1.9)
+      ffi-yajl (~> 2.2)
+      ipaddress
+      mixlib-cli
+      mixlib-config (~> 2.0)
+      mixlib-log (>= 1.7.1, < 2.0)
+      mixlib-shellout (~> 2.0)
+      plist (~> 3.1)
+      systemu (~> 2.6.4)
+      wmi-lite (~> 1.0)
+    pedump (0.5.2)
+      awesome_print
+      iostruct (>= 0.0.4)
+      multipart-post (~> 2.0.0)
+      progressbar
+      zhexdump (>= 0.0.2)
+    plist (3.2.0)
+    progressbar (1.8.2)
+    public_suffix (2.0.5)
+    retryable (2.0.4)
+    ridley (5.1.0)
+      addressable
+      buff-config (~> 2.0)
+      buff-extensions (~> 2.0)
+      buff-ignore (~> 1.2)
+      buff-shell_out (~> 1.0)
+      celluloid (~> 0.16.0)
+      celluloid-io (~> 0.16.1)
+      chef-config (>= 12.5.0)
+      erubis
+      faraday (~> 0.9.0)
+      hashie (>= 2.0.2, < 4.0.0)
+      httpclient (~> 2.7)
+      json (>= 1.7.7)
+      mixlib-authentication (>= 1.3.0)
+      retryable (~> 2.0)
+      semverse (~> 2.0)
+      varia_model (~> 0.6)
+    ruby-progressbar (1.8.1)
+    rubyntlm (0.6.1)
+    rubyzip (1.2.1)
+    safe_yaml (1.0.4)
+    sawyer (0.8.1)
+      addressable (>= 2.3.5, < 2.6)
+      faraday (~> 0.8, < 1.0)
+    semverse (2.0.0)
+    solve (3.1.0)
+      molinillo (>= 0.5)
+      semverse (>= 1.1, < 3.0)
+    systemu (2.6.5)
+    test-kitchen (1.16.0)
+      mixlib-install (>= 1.2, < 3.0)
+      mixlib-shellout (>= 1.2, < 3.0)
+      net-scp (~> 1.1)
+      net-ssh (>= 2.9, < 5.0)
+      net-ssh-gateway (~> 1.2)
+      safe_yaml (~> 1.0)
+      thor (~> 0.19, < 0.19.2)
+    thor (0.19.1)
+    timers (4.0.4)
+      hitimes
+    varia_model (0.6.0)
+      buff-extensions (~> 2.0)
+      hashie (>= 2.0.2, < 4.0.0)
+    winrm (2.1.3)
+      builder (>= 2.1.2)
+      erubis (~> 2.7)
+      gssapi (~> 1.2)
+      gyoku (~> 1.0)
+      httpclient (~> 2.2, >= 2.2.0.2)
+      logging (>= 1.6.1, < 3.0)
+      nori (~> 2.0)
+      rubyntlm (~> 0.6.0, >= 0.6.1)
+    winrm-fs (1.0.1)
+      erubis (~> 2.7)
+      logging (>= 1.6.1, < 3.0)
+      rubyzip (~> 1.1)
+      winrm (~> 2.0)
+    wmi-lite (1.0.0)
+    zhexdump (0.0.2)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  berkshelf
+  kitchen-vagrant
+  omnibus!
+  omnibus-software!
+  test-kitchen
+  winrm-fs
+
+BUNDLED WITH
+   1.14.6