tecnh-chef 0.9.9.vpc
Sign up to get free protection for your applications and to get access to all the features.
- data/LICENSE +201 -0
- data/README.rdoc +172 -0
- data/bin/chef-client +26 -0
- data/bin/chef-solo +25 -0
- data/bin/knife +26 -0
- data/bin/shef +34 -0
- data/distro/README +2 -0
- data/distro/common/man/man1/chef-indexer.1 +42 -0
- data/distro/common/man/man1/chef-server-webui.1 +106 -0
- data/distro/common/man/man1/chef-server.1 +107 -0
- data/distro/common/man/man1/chef-solr-indexer.1 +55 -0
- data/distro/common/man/man1/chef-solr.1 +55 -0
- data/distro/common/man/man8/chef-client.8 +63 -0
- data/distro/common/man/man8/chef-solo.8 +57 -0
- data/distro/common/man/man8/chef-solr-rebuild.8 +37 -0
- data/distro/common/man/man8/knife.8 +1339 -0
- data/distro/common/man/man8/shef.8 +45 -0
- data/distro/common/markdown/README +3 -0
- data/distro/common/markdown/knife.mkd +825 -0
- data/distro/debian/etc/default/chef-client +4 -0
- data/distro/debian/etc/default/chef-server +6 -0
- data/distro/debian/etc/default/chef-server-webui +6 -0
- data/distro/debian/etc/default/chef-solr +4 -0
- data/distro/debian/etc/default/chef-solr-indexer +4 -0
- data/distro/debian/etc/init.d/chef-client +175 -0
- data/distro/debian/etc/init.d/chef-server +120 -0
- data/distro/debian/etc/init.d/chef-server-webui +121 -0
- data/distro/debian/etc/init.d/chef-solr +177 -0
- data/distro/debian/etc/init.d/chef-solr-indexer +176 -0
- data/distro/debian/etc/init/chef-client.conf +17 -0
- data/distro/debian/etc/init/chef-server-webui.conf +17 -0
- data/distro/debian/etc/init/chef-server.conf +17 -0
- data/distro/debian/etc/init/chef-solr-indexer.conf +17 -0
- data/distro/debian/etc/init/chef-solr.conf +17 -0
- data/distro/redhat/etc/init.d/chef-client +106 -0
- data/distro/redhat/etc/init.d/chef-server +112 -0
- data/distro/redhat/etc/init.d/chef-server-webui +112 -0
- data/distro/redhat/etc/init.d/chef-solr +104 -0
- data/distro/redhat/etc/init.d/chef-solr-indexer +104 -0
- data/distro/redhat/etc/logrotate.d/chef-client +8 -0
- data/distro/redhat/etc/logrotate.d/chef-server +8 -0
- data/distro/redhat/etc/logrotate.d/chef-server-webui +8 -0
- data/distro/redhat/etc/logrotate.d/chef-solr +8 -0
- data/distro/redhat/etc/logrotate.d/chef-solr-indexer +8 -0
- data/distro/redhat/etc/sysconfig/chef-client +15 -0
- data/distro/redhat/etc/sysconfig/chef-server +14 -0
- data/distro/redhat/etc/sysconfig/chef-server-webui +14 -0
- data/distro/redhat/etc/sysconfig/chef-solr +8 -0
- data/distro/redhat/etc/sysconfig/chef-solr-indexer +7 -0
- data/lib/chef.rb +39 -0
- data/lib/chef/api_client.rb +264 -0
- data/lib/chef/application.rb +125 -0
- data/lib/chef/application/agent.rb +18 -0
- data/lib/chef/application/client.rb +237 -0
- data/lib/chef/application/knife.rb +160 -0
- data/lib/chef/application/solo.rb +216 -0
- data/lib/chef/applications.rb +4 -0
- data/lib/chef/cache.rb +61 -0
- data/lib/chef/cache/checksum.rb +91 -0
- data/lib/chef/certificate.rb +194 -0
- data/lib/chef/checksum.rb +118 -0
- data/lib/chef/client.rb +340 -0
- data/lib/chef/config.rb +240 -0
- data/lib/chef/cookbook/cookbook_collection.rb +44 -0
- data/lib/chef/cookbook/file_system_file_vendor.rb +54 -0
- data/lib/chef/cookbook/file_vendor.rb +48 -0
- data/lib/chef/cookbook/metadata.rb +500 -0
- data/lib/chef/cookbook/metadata/version.rb +87 -0
- data/lib/chef/cookbook/remote_file_vendor.rb +80 -0
- data/lib/chef/cookbook/syntax_check.rb +136 -0
- data/lib/chef/cookbook_loader.rb +225 -0
- data/lib/chef/cookbook_site_streaming_uploader.rb +215 -0
- data/lib/chef/cookbook_uploader.rb +102 -0
- data/lib/chef/cookbook_version.rb +877 -0
- data/lib/chef/couchdb.rb +239 -0
- data/lib/chef/daemon.rb +172 -0
- data/lib/chef/data_bag.rb +215 -0
- data/lib/chef/data_bag_item.rb +228 -0
- data/lib/chef/exceptions.rb +61 -0
- data/lib/chef/file_access_control.rb +140 -0
- data/lib/chef/file_cache.rb +218 -0
- data/lib/chef/handler.rb +163 -0
- data/lib/chef/handler/json_file.rb +58 -0
- data/lib/chef/index_queue.rb +29 -0
- data/lib/chef/index_queue/amqp_client.rb +113 -0
- data/lib/chef/index_queue/consumer.rb +76 -0
- data/lib/chef/index_queue/indexable.rb +76 -0
- data/lib/chef/knife.rb +496 -0
- data/lib/chef/knife/bootstrap.rb +167 -0
- data/lib/chef/knife/bootstrap/centos5-gems.erb +50 -0
- data/lib/chef/knife/bootstrap/fedora13-gems.erb +47 -0
- data/lib/chef/knife/bootstrap/ubuntu10.04-apt.erb +41 -0
- data/lib/chef/knife/bootstrap/ubuntu10.04-gems.erb +53 -0
- data/lib/chef/knife/client_bulk_delete.rb +40 -0
- data/lib/chef/knife/client_create.rb +70 -0
- data/lib/chef/knife/client_delete.rb +45 -0
- data/lib/chef/knife/client_edit.rb +45 -0
- data/lib/chef/knife/client_list.rb +40 -0
- data/lib/chef/knife/client_reregister.rb +56 -0
- data/lib/chef/knife/client_show.rb +50 -0
- data/lib/chef/knife/configure.rb +140 -0
- data/lib/chef/knife/configure_client.rb +50 -0
- data/lib/chef/knife/cookbook_bulk_delete.rb +58 -0
- data/lib/chef/knife/cookbook_create.rb +209 -0
- data/lib/chef/knife/cookbook_delete.rb +143 -0
- data/lib/chef/knife/cookbook_download.rb +130 -0
- data/lib/chef/knife/cookbook_list.rb +41 -0
- data/lib/chef/knife/cookbook_metadata.rb +82 -0
- data/lib/chef/knife/cookbook_metadata_from_file.rb +40 -0
- data/lib/chef/knife/cookbook_show.rb +98 -0
- data/lib/chef/knife/cookbook_site_download.rb +58 -0
- data/lib/chef/knife/cookbook_site_list.rb +56 -0
- data/lib/chef/knife/cookbook_site_search.rb +51 -0
- data/lib/chef/knife/cookbook_site_share.rb +108 -0
- data/lib/chef/knife/cookbook_site_show.rb +57 -0
- data/lib/chef/knife/cookbook_site_unshare.rb +52 -0
- data/lib/chef/knife/cookbook_site_vendor.rb +127 -0
- data/lib/chef/knife/cookbook_test.rb +82 -0
- data/lib/chef/knife/cookbook_upload.rb +95 -0
- data/lib/chef/knife/data_bag_create.rb +59 -0
- data/lib/chef/knife/data_bag_delete.rb +48 -0
- data/lib/chef/knife/data_bag_edit.rb +50 -0
- data/lib/chef/knife/data_bag_from_file.rb +44 -0
- data/lib/chef/knife/data_bag_list.rb +43 -0
- data/lib/chef/knife/data_bag_show.rb +41 -0
- data/lib/chef/knife/ec2_instance_data.rb +46 -0
- data/lib/chef/knife/ec2_server_create.rb +211 -0
- data/lib/chef/knife/ec2_server_delete.rb +90 -0
- data/lib/chef/knife/ec2_server_list.rb +88 -0
- data/lib/chef/knife/index_rebuild.rb +51 -0
- data/lib/chef/knife/node_bulk_delete.rb +43 -0
- data/lib/chef/knife/node_create.rb +47 -0
- data/lib/chef/knife/node_delete.rb +44 -0
- data/lib/chef/knife/node_edit.rb +44 -0
- data/lib/chef/knife/node_from_file.rb +42 -0
- data/lib/chef/knife/node_list.rb +41 -0
- data/lib/chef/knife/node_run_list_add.rb +64 -0
- data/lib/chef/knife/node_run_list_remove.rb +45 -0
- data/lib/chef/knife/node_show.rb +54 -0
- data/lib/chef/knife/rackspace_server_create.rb +156 -0
- data/lib/chef/knife/rackspace_server_delete.rb +57 -0
- data/lib/chef/knife/rackspace_server_list.rb +59 -0
- data/lib/chef/knife/recipe_list.rb +32 -0
- data/lib/chef/knife/role_bulk_delete.rb +44 -0
- data/lib/chef/knife/role_create.rb +52 -0
- data/lib/chef/knife/role_delete.rb +44 -0
- data/lib/chef/knife/role_edit.rb +45 -0
- data/lib/chef/knife/role_from_file.rb +46 -0
- data/lib/chef/knife/role_list.rb +40 -0
- data/lib/chef/knife/role_show.rb +51 -0
- data/lib/chef/knife/search.rb +94 -0
- data/lib/chef/knife/slicehost_images_list.rb +53 -0
- data/lib/chef/knife/slicehost_server_create.rb +103 -0
- data/lib/chef/knife/slicehost_server_delete.rb +61 -0
- data/lib/chef/knife/slicehost_server_list.rb +64 -0
- data/lib/chef/knife/ssh.rb +311 -0
- data/lib/chef/knife/status.rb +70 -0
- data/lib/chef/knife/terremark_server_create.rb +152 -0
- data/lib/chef/knife/terremark_server_delete.rb +87 -0
- data/lib/chef/knife/terremark_server_list.rb +77 -0
- data/lib/chef/log.rb +61 -0
- data/lib/chef/mixin/check_helper.rb +31 -0
- data/lib/chef/mixin/checksum.rb +32 -0
- data/lib/chef/mixin/command.rb +220 -0
- data/lib/chef/mixin/command/unix.rb +215 -0
- data/lib/chef/mixin/command/windows.rb +72 -0
- data/lib/chef/mixin/convert_to_class_name.rb +63 -0
- data/lib/chef/mixin/create_path.rb +56 -0
- data/lib/chef/mixin/deep_merge.rb +225 -0
- data/lib/chef/mixin/deprecation.rb +65 -0
- data/lib/chef/mixin/find_preferred_file.rb +92 -0
- data/lib/chef/mixin/from_file.rb +50 -0
- data/lib/chef/mixin/language.rb +156 -0
- data/lib/chef/mixin/language_include_attribute.rb +61 -0
- data/lib/chef/mixin/language_include_recipe.rb +52 -0
- data/lib/chef/mixin/params_validate.rb +225 -0
- data/lib/chef/mixin/recipe_definition_dsl_core.rb +81 -0
- data/lib/chef/mixin/shell_out.rb +38 -0
- data/lib/chef/mixin/template.rb +95 -0
- data/lib/chef/mixin/xml_escape.rb +140 -0
- data/lib/chef/mixins.rb +16 -0
- data/lib/chef/monkey_patches/dir.rb +36 -0
- data/lib/chef/monkey_patches/string.rb +28 -0
- data/lib/chef/monkey_patches/tempfile.rb +64 -0
- data/lib/chef/node.rb +605 -0
- data/lib/chef/node/attribute.rb +450 -0
- data/lib/chef/openid_registration.rb +187 -0
- data/lib/chef/platform.rb +354 -0
- data/lib/chef/provider.rb +124 -0
- data/lib/chef/provider/breakpoint.rb +36 -0
- data/lib/chef/provider/cookbook_file.rb +96 -0
- data/lib/chef/provider/cron.rb +186 -0
- data/lib/chef/provider/deploy.rb +319 -0
- data/lib/chef/provider/deploy/revision.rb +73 -0
- data/lib/chef/provider/deploy/timestamped.rb +33 -0
- data/lib/chef/provider/directory.rb +72 -0
- data/lib/chef/provider/env.rb +152 -0
- data/lib/chef/provider/env/windows.rb +75 -0
- data/lib/chef/provider/erl_call.rb +72 -0
- data/lib/chef/provider/execute.rb +58 -0
- data/lib/chef/provider/file.rb +212 -0
- data/lib/chef/provider/git.rb +203 -0
- data/lib/chef/provider/group.rb +133 -0
- data/lib/chef/provider/group/dscl.rb +121 -0
- data/lib/chef/provider/group/gpasswd.rb +53 -0
- data/lib/chef/provider/group/groupadd.rb +78 -0
- data/lib/chef/provider/group/pw.rb +84 -0
- data/lib/chef/provider/group/usermod.rb +57 -0
- data/lib/chef/provider/group/windows.rb +79 -0
- data/lib/chef/provider/http_request.rb +106 -0
- data/lib/chef/provider/ifconfig.rb +132 -0
- data/lib/chef/provider/link.rb +156 -0
- data/lib/chef/provider/log.rb +53 -0
- data/lib/chef/provider/mdadm.rb +91 -0
- data/lib/chef/provider/mount.rb +117 -0
- data/lib/chef/provider/mount/mount.rb +227 -0
- data/lib/chef/provider/mount/windows.rb +80 -0
- data/lib/chef/provider/package.rb +160 -0
- data/lib/chef/provider/package/apt.rb +110 -0
- data/lib/chef/provider/package/dpkg.rb +112 -0
- data/lib/chef/provider/package/easy_install.rb +114 -0
- data/lib/chef/provider/package/freebsd.rb +123 -0
- data/lib/chef/provider/package/macports.rb +105 -0
- data/lib/chef/provider/package/pacman.rb +101 -0
- data/lib/chef/provider/package/portage.rb +124 -0
- data/lib/chef/provider/package/rpm.rb +101 -0
- data/lib/chef/provider/package/rubygems.rb +462 -0
- data/lib/chef/provider/package/solaris.rb +127 -0
- data/lib/chef/provider/package/yum-dump.py +125 -0
- data/lib/chef/provider/package/yum.rb +222 -0
- data/lib/chef/provider/package/zypper.rb +133 -0
- data/lib/chef/provider/remote_directory.rb +134 -0
- data/lib/chef/provider/remote_file.rb +120 -0
- data/lib/chef/provider/route.rb +195 -0
- data/lib/chef/provider/ruby_block.rb +33 -0
- data/lib/chef/provider/script.rb +55 -0
- data/lib/chef/provider/service.rb +128 -0
- data/lib/chef/provider/service/arch.rb +109 -0
- data/lib/chef/provider/service/debian.rb +105 -0
- data/lib/chef/provider/service/freebsd.rb +156 -0
- data/lib/chef/provider/service/gentoo.rb +54 -0
- data/lib/chef/provider/service/init.rb +71 -0
- data/lib/chef/provider/service/redhat.rb +60 -0
- data/lib/chef/provider/service/simple.rb +118 -0
- data/lib/chef/provider/service/solaris.rb +85 -0
- data/lib/chef/provider/service/upstart.rb +192 -0
- data/lib/chef/provider/service/windows.rb +129 -0
- data/lib/chef/provider/subversion.rb +156 -0
- data/lib/chef/provider/template.rb +105 -0
- data/lib/chef/provider/user.rb +187 -0
- data/lib/chef/provider/user/dscl.rb +280 -0
- data/lib/chef/provider/user/pw.rb +113 -0
- data/lib/chef/provider/user/useradd.rb +109 -0
- data/lib/chef/provider/user/windows.rb +124 -0
- data/lib/chef/providers.rb +89 -0
- data/lib/chef/recipe.rb +130 -0
- data/lib/chef/resource.rb +417 -0
- data/lib/chef/resource/apt_package.rb +34 -0
- data/lib/chef/resource/bash.rb +33 -0
- data/lib/chef/resource/breakpoint.rb +35 -0
- data/lib/chef/resource/cookbook_file.rb +45 -0
- data/lib/chef/resource/cron.rb +188 -0
- data/lib/chef/resource/csh.rb +33 -0
- data/lib/chef/resource/deploy.rb +371 -0
- data/lib/chef/resource/deploy_revision.rb +35 -0
- data/lib/chef/resource/directory.rb +76 -0
- data/lib/chef/resource/dpkg_package.rb +34 -0
- data/lib/chef/resource/easy_install_package.rb +41 -0
- data/lib/chef/resource/env.rb +58 -0
- data/lib/chef/resource/erl_call.rb +83 -0
- data/lib/chef/resource/execute.rb +127 -0
- data/lib/chef/resource/file.rb +92 -0
- data/lib/chef/resource/freebsd_package.rb +35 -0
- data/lib/chef/resource/gem_package.rb +49 -0
- data/lib/chef/resource/git.rb +36 -0
- data/lib/chef/resource/group.rb +70 -0
- data/lib/chef/resource/http_request.rb +52 -0
- data/lib/chef/resource/ifconfig.rb +134 -0
- data/lib/chef/resource/link.rb +78 -0
- data/lib/chef/resource/log.rb +62 -0
- data/lib/chef/resource/macports_package.rb +29 -0
- data/lib/chef/resource/mdadm.rb +82 -0
- data/lib/chef/resource/mount.rb +135 -0
- data/lib/chef/resource/package.rb +80 -0
- data/lib/chef/resource/pacman_package.rb +33 -0
- data/lib/chef/resource/perl.rb +33 -0
- data/lib/chef/resource/portage_package.rb +33 -0
- data/lib/chef/resource/python.rb +33 -0
- data/lib/chef/resource/remote_directory.rb +100 -0
- data/lib/chef/resource/remote_file.rb +83 -0
- data/lib/chef/resource/route.rb +135 -0
- data/lib/chef/resource/ruby.rb +33 -0
- data/lib/chef/resource/ruby_block.rb +40 -0
- data/lib/chef/resource/scm.rb +146 -0
- data/lib/chef/resource/script.rb +51 -0
- data/lib/chef/resource/service.rb +160 -0
- data/lib/chef/resource/solaris_package.rb +36 -0
- data/lib/chef/resource/subversion.rb +36 -0
- data/lib/chef/resource/template.rb +69 -0
- data/lib/chef/resource/timestamped_deploy.rb +31 -0
- data/lib/chef/resource/user.rb +130 -0
- data/lib/chef/resource/yum_package.rb +43 -0
- data/lib/chef/resource_collection.rb +216 -0
- data/lib/chef/resource_collection/stepable_iterator.rb +124 -0
- data/lib/chef/resource_definition.rb +67 -0
- data/lib/chef/resource_definition_list.rb +38 -0
- data/lib/chef/resources.rb +62 -0
- data/lib/chef/rest.rb +388 -0
- data/lib/chef/rest/auth_credentials.rb +78 -0
- data/lib/chef/rest/cookie_jar.rb +31 -0
- data/lib/chef/rest/rest_request.rb +186 -0
- data/lib/chef/role.rb +287 -0
- data/lib/chef/run_context.rb +110 -0
- data/lib/chef/run_list.rb +150 -0
- data/lib/chef/run_list/run_list_expansion.rb +172 -0
- data/lib/chef/run_list/run_list_item.rb +78 -0
- data/lib/chef/run_status.rb +121 -0
- data/lib/chef/runner.rb +131 -0
- data/lib/chef/sandbox.rb +153 -0
- data/lib/chef/search/query.rb +60 -0
- data/lib/chef/shef.rb +325 -0
- data/lib/chef/shef/ext.rb +565 -0
- data/lib/chef/shef/model_wrapper.rb +120 -0
- data/lib/chef/shef/shef_rest.rb +28 -0
- data/lib/chef/shef/shef_session.rb +271 -0
- data/lib/chef/shell_out.rb +413 -0
- data/lib/chef/streaming_cookbook_uploader.rb +201 -0
- data/lib/chef/tasks/chef_repo.rake +256 -0
- data/lib/chef/util/file_edit.rb +122 -0
- data/lib/chef/util/windows.rb +56 -0
- data/lib/chef/util/windows/net_group.rb +101 -0
- data/lib/chef/util/windows/net_use.rb +121 -0
- data/lib/chef/util/windows/net_user.rb +198 -0
- data/lib/chef/util/windows/volume.rb +59 -0
- data/lib/chef/version.rb +21 -0
- data/lib/chef/webui_user.rb +232 -0
- metadata +614 -0
@@ -0,0 +1,120 @@
|
|
1
|
+
#--
|
2
|
+
# Author:: Daniel DeLeo (<dan@opscode.com>)
|
3
|
+
# Copyright:: Copyright (c) 2010 Opscode, Inc.
|
4
|
+
# License:: Apache License, Version 2.0
|
5
|
+
#
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
7
|
+
# you may not use this file except in compliance with the License.
|
8
|
+
# You may obtain a copy of the License at
|
9
|
+
#
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
11
|
+
#
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
15
|
+
# See the License for the specific language governing permissions and
|
16
|
+
# limitations under the License.
|
17
|
+
#
|
18
|
+
|
19
|
+
require 'chef/mixin/convert_to_class_name'
|
20
|
+
require 'chef/mixin/language'
|
21
|
+
|
22
|
+
module Shef
|
23
|
+
class ModelWrapper
|
24
|
+
|
25
|
+
include Chef::Mixin::ConvertToClassName
|
26
|
+
|
27
|
+
attr_reader :model_symbol
|
28
|
+
|
29
|
+
def initialize(model_class, symbol=nil)
|
30
|
+
@model_class = model_class
|
31
|
+
@model_symbol = symbol || convert_to_snake_case(model_class.name, "Chef").to_sym
|
32
|
+
end
|
33
|
+
|
34
|
+
def search(query)
|
35
|
+
return all if query.to_s == "all"
|
36
|
+
results = []
|
37
|
+
Chef::Search::Query.new.search(@model_symbol, format_query(query)) do |obj|
|
38
|
+
if block_given?
|
39
|
+
results << yield(obj)
|
40
|
+
else
|
41
|
+
results << obj
|
42
|
+
end
|
43
|
+
end
|
44
|
+
results
|
45
|
+
end
|
46
|
+
|
47
|
+
alias :find :search
|
48
|
+
|
49
|
+
def all(&block)
|
50
|
+
all_objects = list_objects
|
51
|
+
block_given? ? all_objects.map(&block) : all_objects
|
52
|
+
end
|
53
|
+
|
54
|
+
alias :list :all
|
55
|
+
|
56
|
+
def show(obj_id)
|
57
|
+
@model_class.load(obj_id)
|
58
|
+
end
|
59
|
+
|
60
|
+
alias :load :show
|
61
|
+
|
62
|
+
def transform(what_to_transform, &block)
|
63
|
+
if what_to_transform == :all
|
64
|
+
objects_to_transform = list_objects
|
65
|
+
else
|
66
|
+
objects_to_transform = search(what_to_transform)
|
67
|
+
end
|
68
|
+
objects_to_transform.each do |obj|
|
69
|
+
if result = yield(obj)
|
70
|
+
obj.save
|
71
|
+
end
|
72
|
+
end
|
73
|
+
end
|
74
|
+
|
75
|
+
alias :bulk_edit :transform
|
76
|
+
|
77
|
+
private
|
78
|
+
|
79
|
+
# paper over inconsistencies in the model classes APIs, and return the objects
|
80
|
+
# the user wanted instead of the URI=>object stuff
|
81
|
+
def list_objects
|
82
|
+
objects = @model_class.method(:list).arity == 0? @model_class.list : @model_class.list(true)
|
83
|
+
objects.map { |obj| Array(obj).find {|o| o.kind_of?(@model_class)} }
|
84
|
+
end
|
85
|
+
|
86
|
+
def format_query(query)
|
87
|
+
if query.respond_to?(:keys)
|
88
|
+
query.map { |key, value| "#{key}:#{value}" }.join(" AND ")
|
89
|
+
else
|
90
|
+
query
|
91
|
+
end
|
92
|
+
end
|
93
|
+
end
|
94
|
+
|
95
|
+
class NamedDataBagWrapper < ModelWrapper
|
96
|
+
|
97
|
+
def initialize(databag_name)
|
98
|
+
@model_symbol = @databag_name = databag_name
|
99
|
+
end
|
100
|
+
|
101
|
+
|
102
|
+
alias :list :all
|
103
|
+
|
104
|
+
def show(item)
|
105
|
+
Chef::DataBagItem.load(@databag_name, item)
|
106
|
+
end
|
107
|
+
|
108
|
+
private
|
109
|
+
|
110
|
+
def list_objects
|
111
|
+
all_items = []
|
112
|
+
Chef::Search::Query.new.search(@databag_name) do |item|
|
113
|
+
all_items << item
|
114
|
+
end
|
115
|
+
all_items
|
116
|
+
end
|
117
|
+
|
118
|
+
end
|
119
|
+
|
120
|
+
end
|
@@ -0,0 +1,28 @@
|
|
1
|
+
#--
|
2
|
+
# Author:: Daniel DeLeo (<dan@opscode.com>)
|
3
|
+
# Copyright:: Copyright (c) 2010 Opscode, Inc.
|
4
|
+
# License:: Apache License, Version 2.0
|
5
|
+
#
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
7
|
+
# you may not use this file except in compliance with the License.
|
8
|
+
# You may obtain a copy of the License at
|
9
|
+
#
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
11
|
+
#
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
15
|
+
# See the License for the specific language governing permissions and
|
16
|
+
# limitations under the License.
|
17
|
+
#
|
18
|
+
|
19
|
+
module Shef
|
20
|
+
class ShefREST < Chef::REST
|
21
|
+
|
22
|
+
alias :get :get_rest
|
23
|
+
alias :put :put_rest
|
24
|
+
alias :post :post_rest
|
25
|
+
alias :delete :delete_rest
|
26
|
+
|
27
|
+
end
|
28
|
+
end
|
@@ -0,0 +1,271 @@
|
|
1
|
+
#--
|
2
|
+
# Author:: Daniel DeLeo (<dan@kallistec.com>)
|
3
|
+
# Copyright:: Copyright (c) 2009 Daniel DeLeo
|
4
|
+
# License:: Apache License, Version 2.0
|
5
|
+
#
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
7
|
+
# you may not use this file except in compliance with the License.
|
8
|
+
# You may obtain a copy of the License at
|
9
|
+
#
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
11
|
+
#
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
15
|
+
# See the License for the specific language governing permissions and
|
16
|
+
# limitations under the License.
|
17
|
+
#
|
18
|
+
|
19
|
+
require 'chef/recipe'
|
20
|
+
require 'chef/run_context'
|
21
|
+
require 'chef/config'
|
22
|
+
require 'chef/client'
|
23
|
+
require 'chef/cookbook/cookbook_collection'
|
24
|
+
require 'chef/cookbook_loader'
|
25
|
+
|
26
|
+
module Shef
|
27
|
+
class ShefSession
|
28
|
+
include Singleton
|
29
|
+
|
30
|
+
def self.session_type(type=nil)
|
31
|
+
@session_type = type if type
|
32
|
+
@session_type
|
33
|
+
end
|
34
|
+
|
35
|
+
attr_accessor :node, :compile, :recipe, :run_context
|
36
|
+
attr_reader :node_attributes, :client
|
37
|
+
def initialize
|
38
|
+
@node_built = false
|
39
|
+
end
|
40
|
+
|
41
|
+
def node_built?
|
42
|
+
!!@node_built
|
43
|
+
end
|
44
|
+
|
45
|
+
def reset!
|
46
|
+
loading do
|
47
|
+
rebuild_node
|
48
|
+
@node = client.node
|
49
|
+
shorten_node_inspect
|
50
|
+
Shef::Extensions.extend_context_node(@node)
|
51
|
+
rebuild_context
|
52
|
+
node.consume_attributes(node_attributes) if node_attributes
|
53
|
+
@recipe = Chef::Recipe.new(nil, nil, run_context)
|
54
|
+
Shef::Extensions.extend_context_recipe(@recipe)
|
55
|
+
@node_built = true
|
56
|
+
end
|
57
|
+
end
|
58
|
+
|
59
|
+
def node_attributes=(attrs)
|
60
|
+
@node_attributes = attrs
|
61
|
+
@node.consume_attributes(@node_attributes)
|
62
|
+
end
|
63
|
+
|
64
|
+
def resource_collection
|
65
|
+
run_context.resource_collection
|
66
|
+
end
|
67
|
+
|
68
|
+
def run_context
|
69
|
+
@run_context || rebuild_context
|
70
|
+
end
|
71
|
+
|
72
|
+
def definitions
|
73
|
+
nil
|
74
|
+
end
|
75
|
+
|
76
|
+
def cookbook_loader
|
77
|
+
nil
|
78
|
+
end
|
79
|
+
|
80
|
+
def save_node
|
81
|
+
raise "Not Supported! #{self.class.name} doesn't support #save_node, maybe you need to run shef in client mode?"
|
82
|
+
end
|
83
|
+
|
84
|
+
def rebuild_context
|
85
|
+
raise "Not Implemented! :rebuild_collection should be implemented by subclasses"
|
86
|
+
end
|
87
|
+
|
88
|
+
private
|
89
|
+
|
90
|
+
def loading
|
91
|
+
show_loading_progress
|
92
|
+
begin
|
93
|
+
yield
|
94
|
+
rescue => e
|
95
|
+
loading_complete(false)
|
96
|
+
raise e
|
97
|
+
else
|
98
|
+
loading_complete(true)
|
99
|
+
end
|
100
|
+
end
|
101
|
+
|
102
|
+
def show_loading_progress
|
103
|
+
print "Loading"
|
104
|
+
@loading = true
|
105
|
+
@dot_printer = Thread.new do
|
106
|
+
while @loading
|
107
|
+
print "."
|
108
|
+
sleep 0.5
|
109
|
+
end
|
110
|
+
end
|
111
|
+
end
|
112
|
+
|
113
|
+
def loading_complete(success)
|
114
|
+
@loading = false
|
115
|
+
@dot_printer.join
|
116
|
+
msg = success ? "done.\n\n" : "epic fail!\n\n"
|
117
|
+
print msg
|
118
|
+
end
|
119
|
+
|
120
|
+
def shorten_node_inspect
|
121
|
+
def @node.inspect
|
122
|
+
"<Chef::Node:0x#{self.object_id.to_s(16)} @name=\"#{self.name}\">"
|
123
|
+
end
|
124
|
+
end
|
125
|
+
|
126
|
+
def rebuild_node
|
127
|
+
raise "Not Implemented! :rebuild_node should be implemented by subclasses"
|
128
|
+
end
|
129
|
+
|
130
|
+
end
|
131
|
+
|
132
|
+
class StandAloneSession < ShefSession
|
133
|
+
|
134
|
+
session_type :standalone
|
135
|
+
|
136
|
+
def rebuild_context
|
137
|
+
@run_context = Chef::RunContext.new(@node, {}) # no recipes
|
138
|
+
end
|
139
|
+
|
140
|
+
private
|
141
|
+
|
142
|
+
def rebuild_node
|
143
|
+
Chef::Config[:solo] = true
|
144
|
+
@client = Chef::Client.new
|
145
|
+
@client.run_ohai
|
146
|
+
@client.build_node
|
147
|
+
end
|
148
|
+
|
149
|
+
end
|
150
|
+
|
151
|
+
class SoloSession < ShefSession
|
152
|
+
|
153
|
+
session_type :solo
|
154
|
+
|
155
|
+
def definitions
|
156
|
+
@run_context.definitions
|
157
|
+
end
|
158
|
+
|
159
|
+
def rebuild_context
|
160
|
+
@run_context = Chef::RunContext.new(@node, Chef::CookbookCollection.new(Chef::CookbookLoader.new))
|
161
|
+
end
|
162
|
+
|
163
|
+
private
|
164
|
+
|
165
|
+
def rebuild_node
|
166
|
+
# Tell the client we're chef solo so it won't try to contact the server
|
167
|
+
Chef::Config[:solo] = true
|
168
|
+
@client = Chef::Client.new
|
169
|
+
@client.run_ohai
|
170
|
+
@client.build_node
|
171
|
+
end
|
172
|
+
|
173
|
+
end
|
174
|
+
|
175
|
+
class ClientSession < SoloSession
|
176
|
+
|
177
|
+
session_type :solo
|
178
|
+
|
179
|
+
def save_node
|
180
|
+
@client.save_node
|
181
|
+
end
|
182
|
+
|
183
|
+
private
|
184
|
+
|
185
|
+
def rebuild_node
|
186
|
+
# Make sure the client knows this is not chef solo
|
187
|
+
Chef::Config[:solo] = false
|
188
|
+
@client = Chef::Client.new
|
189
|
+
@client.run_ohai
|
190
|
+
@client.register
|
191
|
+
@client.build_node
|
192
|
+
|
193
|
+
@client.sync_cookbooks({})
|
194
|
+
end
|
195
|
+
|
196
|
+
end
|
197
|
+
|
198
|
+
class DoppelGangerClient < Chef::Client
|
199
|
+
|
200
|
+
attr_reader :node_name
|
201
|
+
|
202
|
+
def initialize(node_name)
|
203
|
+
@node_name = node_name
|
204
|
+
@ohai = Ohai::System.new
|
205
|
+
end
|
206
|
+
|
207
|
+
# Run the very smallest amount of ohai we can get away with and still
|
208
|
+
# hope to have things work. Otherwise we're not very good doppelgangers
|
209
|
+
def run_ohai
|
210
|
+
@ohai.require_plugin('os')
|
211
|
+
end
|
212
|
+
|
213
|
+
# DoppelGanger implementation of build_node. preserves as many of the node's
|
214
|
+
# attributes, and does not save updates to the server
|
215
|
+
def build_node
|
216
|
+
Chef::Log.debug("Building node object for #{@node_name}")
|
217
|
+
|
218
|
+
@node = Chef::Node.find_or_create(node_name)
|
219
|
+
|
220
|
+
ohai_data = @ohai.data.merge(@node.automatic_attrs)
|
221
|
+
|
222
|
+
@node.process_external_attrs(ohai_data,nil)
|
223
|
+
@node.reset_defaults_and_overrides
|
224
|
+
|
225
|
+
@node
|
226
|
+
end
|
227
|
+
|
228
|
+
def register
|
229
|
+
@rest = Chef::REST.new(Chef::Config[:chef_server_url], Chef::Config[:node_name], Chef::Config[:client_key])
|
230
|
+
end
|
231
|
+
|
232
|
+
end
|
233
|
+
|
234
|
+
class DoppelGangerSession < ClientSession
|
235
|
+
|
236
|
+
session_type "doppelganger client"
|
237
|
+
|
238
|
+
def save_node
|
239
|
+
puts "A doppelganger should think twice before saving the node"
|
240
|
+
end
|
241
|
+
|
242
|
+
def assume_identity(node_name)
|
243
|
+
Chef::Config[:doppelganger] = @node_name = node_name
|
244
|
+
reset!
|
245
|
+
rescue Exception => e
|
246
|
+
puts "#{e.class.name}: #{e.message}"
|
247
|
+
puts Array(e.backtrace).join("\n")
|
248
|
+
puts
|
249
|
+
puts "* " * 40
|
250
|
+
puts "failed to assume the identity of node '#{node_name}', resetting"
|
251
|
+
puts "* " * 40
|
252
|
+
puts
|
253
|
+
Chef::Config[:doppelganger] = false
|
254
|
+
@node_built = false
|
255
|
+
Shef.session
|
256
|
+
end
|
257
|
+
|
258
|
+
def rebuild_node
|
259
|
+
# Make sure the client knows this is not chef solo
|
260
|
+
Chef::Config[:solo] = false
|
261
|
+
@client = DoppelGangerClient.new(@node_name)
|
262
|
+
@client.run_ohai
|
263
|
+
@client.register
|
264
|
+
@client.build_node
|
265
|
+
|
266
|
+
@client.sync_cookbooks({})
|
267
|
+
end
|
268
|
+
|
269
|
+
end
|
270
|
+
|
271
|
+
end
|
@@ -0,0 +1,413 @@
|
|
1
|
+
#--
|
2
|
+
# Author:: Daniel DeLeo (<dan@opscode.com>)
|
3
|
+
# Copyright:: Copyright (c) 2010 Opscode, Inc.
|
4
|
+
# License:: Apache License, Version 2.0
|
5
|
+
#
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
7
|
+
# you may not use this file except in compliance with the License.
|
8
|
+
# You may obtain a copy of the License at
|
9
|
+
#
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
11
|
+
#
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
15
|
+
# See the License for the specific language governing permissions and
|
16
|
+
# limitations under the License.
|
17
|
+
#
|
18
|
+
|
19
|
+
require 'etc'
|
20
|
+
require 'tmpdir'
|
21
|
+
require 'chef/log'
|
22
|
+
require 'fcntl'
|
23
|
+
require 'chef/exceptions'
|
24
|
+
|
25
|
+
class Chef
|
26
|
+
|
27
|
+
# == Chef::ShellOut
|
28
|
+
# Provides a simplified interface to shelling out yet still collecting both
|
29
|
+
# standard out and standard error and providing full control over environment,
|
30
|
+
# working directory, uid, gid, etc.
|
31
|
+
#
|
32
|
+
# No means for passing input to the subprocess is provided, nor is there any
|
33
|
+
# way to inspect the output of the command as it is being read. If you need
|
34
|
+
# to do that, you have to use popen4 (in Chef::Mixin::Command)
|
35
|
+
#
|
36
|
+
# === Platform Support
|
37
|
+
# Chef::ShellOut uses Kernel.fork() and is therefore unsuitable for Windows
|
38
|
+
# or jruby.
|
39
|
+
class ShellOut
|
40
|
+
READ_WAIT_TIME = 0.01
|
41
|
+
READ_SIZE = 4096
|
42
|
+
DEFAULT_READ_TIMEOUT = 60
|
43
|
+
DEFAULT_ENVIRONMENT = {'LC_ALL' => 'C'}
|
44
|
+
|
45
|
+
attr_accessor :user, :group, :cwd, :valid_exit_codes
|
46
|
+
attr_reader :command, :umask, :environment
|
47
|
+
attr_writer :timeout
|
48
|
+
attr_reader :execution_time
|
49
|
+
|
50
|
+
attr_reader :stdout, :stderr, :status
|
51
|
+
|
52
|
+
attr_reader :stdin_pipe, :stdout_pipe, :stderr_pipe, :process_status_pipe
|
53
|
+
|
54
|
+
# === Arguments:
|
55
|
+
# Takes a single command, or a list of command fragments. These are used
|
56
|
+
# as arguments to Kernel.exec. See the Kernel.exec documentation for more
|
57
|
+
# explanation of how arguments are evaluated. The last argument can be an
|
58
|
+
# options Hash.
|
59
|
+
# === Options:
|
60
|
+
# If the last argument is a Hash, it is removed from the list of args passed
|
61
|
+
# to exec and used as an options hash. The following options are available:
|
62
|
+
# * +user+: the user the commmand should run as. if an integer is given, it is
|
63
|
+
# used as a uid. A string is treated as a username and resolved to a uid
|
64
|
+
# with Etc.getpwnam
|
65
|
+
# * +group+: the group the command should run as. works similarly to +user+
|
66
|
+
# * +cwd+: the directory to chdir to before running the command
|
67
|
+
# * +umask+: a umask to set before running the command. If given as an Integer,
|
68
|
+
# be sure to use two leading zeros so it's parsed as Octal. A string will
|
69
|
+
# be treated as an octal integer
|
70
|
+
# * +returns+: one or more Integer values to use as valid exit codes for the
|
71
|
+
# subprocess. This only has an effect if you call +error!+ after
|
72
|
+
# +run_command+.
|
73
|
+
# * +environment+: a Hash of environment variables to set before the command
|
74
|
+
# is run. By default, the environment will *always* be set to 'LC_ALL' => 'C'
|
75
|
+
# to prevent issues with multibyte characters in Ruby 1.8. To avoid this,
|
76
|
+
# use :environment => nil for *no* extra environment settings, or
|
77
|
+
# :environment => {'LC_ALL'=>nil, ...} to set other environment settings
|
78
|
+
# without changing the locale.
|
79
|
+
# * +timeout+: a Numeric value for the number of seconds to wait on the
|
80
|
+
# child process before raising an Exception. This is calculated as the
|
81
|
+
# total amount of time that ShellOut waited on the child process without
|
82
|
+
# receiving any output (i.e., IO.select returned nil). Default is 60
|
83
|
+
# seconds. Note: the stdlib Timeout library is not used.
|
84
|
+
# === Examples:
|
85
|
+
# Invoke find(1) to search for .rb files:
|
86
|
+
# find = Chef::ShellOut.new("find . -name '*.rb'")
|
87
|
+
# find.run_command
|
88
|
+
# # If all went well, the results are on +stdout+
|
89
|
+
# puts find.stdout
|
90
|
+
# # find(1) prints diagnostic info to STDERR:
|
91
|
+
# puts "error messages" + find.stderr
|
92
|
+
# # Raise an exception if it didn't exit with 0
|
93
|
+
# find.error!
|
94
|
+
# Run a command as the +www+ user with no extra ENV settings from +/tmp+
|
95
|
+
# cmd = Chef::ShellOut.new("apachectl", "start", :user => 'www', :env => nil, :cwd => '/tmp')
|
96
|
+
# cmd.run_command # etc.
|
97
|
+
def initialize(*command_args)
|
98
|
+
@stdout, @stderr = '', ''
|
99
|
+
@environment = DEFAULT_ENVIRONMENT
|
100
|
+
@cwd = Dir.tmpdir
|
101
|
+
@valid_exit_codes = [0]
|
102
|
+
|
103
|
+
if command_args.last.is_a?(Hash)
|
104
|
+
parse_options(command_args.pop)
|
105
|
+
end
|
106
|
+
|
107
|
+
@command = command_args.size == 1 ? command_args.first : command_args
|
108
|
+
end
|
109
|
+
|
110
|
+
def umask=(new_umask)
|
111
|
+
@umask = (new_umask.respond_to?(:oct) ? new_umask.oct : new_umask.to_i) & 007777
|
112
|
+
end
|
113
|
+
|
114
|
+
def uid
|
115
|
+
return nil unless user
|
116
|
+
user.kind_of?(Integer) ? user : Etc.getpwnam(user.to_s).uid
|
117
|
+
end
|
118
|
+
|
119
|
+
def gid
|
120
|
+
return nil unless group
|
121
|
+
group.kind_of?(Integer) ? group : Etc.getgrnam(group.to_s).gid
|
122
|
+
end
|
123
|
+
|
124
|
+
def timeout
|
125
|
+
@timeout || DEFAULT_READ_TIMEOUT
|
126
|
+
end
|
127
|
+
|
128
|
+
# Creates a String showing the output of the command, including a banner
|
129
|
+
# showing the exact command executed. Used by +invalid!+ to show command
|
130
|
+
# results when the command exited with an unexpected status.
|
131
|
+
def format_for_exception
|
132
|
+
msg = ""
|
133
|
+
msg << "---- Begin output of #{command} ----\n"
|
134
|
+
msg << "STDOUT: #{stdout.strip}\n"
|
135
|
+
msg << "STDERR: #{stderr.strip}\n"
|
136
|
+
msg << "---- End output of #{command} ----\n"
|
137
|
+
msg << "Ran #{command} returned #{status.exitstatus}" if status
|
138
|
+
msg
|
139
|
+
end
|
140
|
+
|
141
|
+
def exitstatus
|
142
|
+
@status && @status.exitstatus
|
143
|
+
end
|
144
|
+
|
145
|
+
# Run the command, writing the command's standard out and standard error
|
146
|
+
# to +stdout+ and +stderr+, and saving its exit status object to +status+
|
147
|
+
# === Returns
|
148
|
+
# returns +self+; +stdout+, +stderr+, +status+, and +exitstatus+ will be
|
149
|
+
# populated with results of the command
|
150
|
+
# === Raises
|
151
|
+
# * Errno::EACCES when you are not privileged to execute the command
|
152
|
+
# * Errno::ENOENT when the command is not available on the system (or not
|
153
|
+
# in the current $PATH)
|
154
|
+
# * Chef::Exceptions::CommandTimeout when the command does not complete
|
155
|
+
# within +timeout+ seconds (default: 60s)
|
156
|
+
def run_command
|
157
|
+
Chef::Log.debug("sh(#{@command})")
|
158
|
+
@child_pid = fork_subprocess
|
159
|
+
|
160
|
+
configure_parent_process_file_descriptors
|
161
|
+
propagate_pre_exec_failure
|
162
|
+
|
163
|
+
@result = nil
|
164
|
+
@execution_time = 0
|
165
|
+
|
166
|
+
# Ruby 1.8.7 and 1.8.6 from mid 2009 try to allocate objects during GC
|
167
|
+
# when calling IO.select and IO#read. Some OS Vendors are not interested
|
168
|
+
# in updating their ruby packages (Apple, *cough*) and we *have to*
|
169
|
+
# make it work. So I give you this epic hack:
|
170
|
+
GC.disable
|
171
|
+
until @status
|
172
|
+
ready = IO.select(open_pipes, nil, nil, READ_WAIT_TIME)
|
173
|
+
unless ready
|
174
|
+
@execution_time += READ_WAIT_TIME
|
175
|
+
if @execution_time >= timeout && !@result
|
176
|
+
raise Chef::Exceptions::CommandTimeout, "command timed out:\n#{format_for_exception}"
|
177
|
+
end
|
178
|
+
end
|
179
|
+
|
180
|
+
if ready && ready.first.include?(child_stdout)
|
181
|
+
read_stdout_to_buffer
|
182
|
+
end
|
183
|
+
if ready && ready.first.include?(child_stderr)
|
184
|
+
read_stderr_to_buffer
|
185
|
+
end
|
186
|
+
|
187
|
+
unless @status
|
188
|
+
# make one more pass to get the last of the output after the
|
189
|
+
# child process dies
|
190
|
+
if results = Process.waitpid2(@child_pid, Process::WNOHANG)
|
191
|
+
@status = results.last
|
192
|
+
redo
|
193
|
+
end
|
194
|
+
end
|
195
|
+
end
|
196
|
+
self
|
197
|
+
rescue Exception
|
198
|
+
# do our best to kill zombies
|
199
|
+
Process.waitpid2(@child_pid, Process::WNOHANG) rescue nil
|
200
|
+
raise
|
201
|
+
ensure
|
202
|
+
# no matter what happens, turn the GC back on, and hope whatever busted
|
203
|
+
# version of ruby we're on doesn't allocate some objects during the next
|
204
|
+
# GC run.
|
205
|
+
GC.enable
|
206
|
+
close_all_pipes
|
207
|
+
end
|
208
|
+
|
209
|
+
# Checks the +exitstatus+ against the set of +valid_exit_codes+. If
|
210
|
+
# +exitstatus+ is not in the list of +valid_exit_codes+, calls +invalid!+,
|
211
|
+
# which raises an Exception.
|
212
|
+
# === Returns
|
213
|
+
# nil::: always returns nil when it does not raise
|
214
|
+
# === Raises
|
215
|
+
# Chef::Exceptions::ShellCommandFailed::: via +invalid!+
|
216
|
+
def error!
|
217
|
+
unless Array(valid_exit_codes).include?(exitstatus)
|
218
|
+
invalid!("Expected process to exit 0, but it exited with #{exitstatus}")
|
219
|
+
end
|
220
|
+
end
|
221
|
+
|
222
|
+
# Raises a Chef::Exceptions::ShellCommandFailed exception, appending the
|
223
|
+
# command's stdout, stderr, and exitstatus to the exception message.
|
224
|
+
# === Arguments
|
225
|
+
# +msg+: A String to use as the basis of the exception message. The
|
226
|
+
# default explanation is very generic, providing a more informative message
|
227
|
+
# is highly encouraged.
|
228
|
+
# === Raises
|
229
|
+
# Chef::Exceptions::ShellCommandFailed always
|
230
|
+
def invalid!(msg=nil)
|
231
|
+
msg ||= "Command produced unexpected results"
|
232
|
+
raise Chef::Exceptions::ShellCommandFailed, msg + "\n" + format_for_exception
|
233
|
+
end
|
234
|
+
|
235
|
+
def inspect
|
236
|
+
"<#{self.class.name}##{object_id}: command: '#@command' process_status: #{@status.inspect} " +
|
237
|
+
"stdout: '#{stdout.strip}' stderr: '#{stderr.strip}' child_pid: #{@child_pid.inspect} " +
|
238
|
+
"environment: #{@environment.inspect} timeout: #{timeout} user: #@user group: #@group working_dir: #@cwd >"
|
239
|
+
end
|
240
|
+
|
241
|
+
private
|
242
|
+
|
243
|
+
def parse_options(opts)
|
244
|
+
opts.each do |option, setting|
|
245
|
+
case option.to_s
|
246
|
+
when 'cwd'
|
247
|
+
self.cwd = setting
|
248
|
+
when 'user'
|
249
|
+
self.user = setting
|
250
|
+
when 'group'
|
251
|
+
self.group = setting
|
252
|
+
when 'umask'
|
253
|
+
self.umask = setting
|
254
|
+
when 'timeout'
|
255
|
+
self.timeout = setting
|
256
|
+
when 'returns'
|
257
|
+
self.valid_exit_codes = Array(setting)
|
258
|
+
when 'environment', 'env'
|
259
|
+
# passing :environment => nil means don't set any new ENV vars
|
260
|
+
setting.nil? ? @environment = {} : @environment.merge!(setting)
|
261
|
+
else
|
262
|
+
raise Chef::Exceptions::InvalidCommandOption, "option '#{option.inspect}' is not a valid option for #{self.class.name}"
|
263
|
+
end
|
264
|
+
end
|
265
|
+
end
|
266
|
+
|
267
|
+
def set_user
|
268
|
+
if user
|
269
|
+
Process.euid = uid
|
270
|
+
Process.uid = uid
|
271
|
+
end
|
272
|
+
end
|
273
|
+
|
274
|
+
def set_group
|
275
|
+
if group
|
276
|
+
Process.egid = gid
|
277
|
+
Process.gid = gid
|
278
|
+
end
|
279
|
+
end
|
280
|
+
|
281
|
+
def set_environment
|
282
|
+
environment.each do |env_var,value|
|
283
|
+
ENV[env_var] = value
|
284
|
+
end
|
285
|
+
end
|
286
|
+
|
287
|
+
def set_umask
|
288
|
+
File.umask(umask) if umask
|
289
|
+
end
|
290
|
+
|
291
|
+
def initialize_ipc
|
292
|
+
@stdout_pipe, @stderr_pipe, @process_status_pipe = IO.pipe, IO.pipe, IO.pipe
|
293
|
+
@process_status_pipe.last.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC)
|
294
|
+
end
|
295
|
+
|
296
|
+
def child_stdout
|
297
|
+
@stdout_pipe[0]
|
298
|
+
end
|
299
|
+
|
300
|
+
def child_stderr
|
301
|
+
@stderr_pipe[0]
|
302
|
+
end
|
303
|
+
|
304
|
+
def child_process_status
|
305
|
+
@process_status_pipe[0]
|
306
|
+
end
|
307
|
+
|
308
|
+
def close_all_pipes
|
309
|
+
child_stdout.close unless child_stdout.closed?
|
310
|
+
child_stderr.close unless child_stderr.closed?
|
311
|
+
child_process_status.close unless child_process_status.closed?
|
312
|
+
end
|
313
|
+
|
314
|
+
# replace stdout, and stderr with pipes to the parent, and close the
|
315
|
+
# reader side of the error marshaling side channel. Close STDIN so when we
|
316
|
+
# exec, the new program will know it's never getting input ever.
|
317
|
+
def configure_subprocess_file_descriptors
|
318
|
+
process_status_pipe.first.close
|
319
|
+
|
320
|
+
# HACK: for some reason, just STDIN.close isn't good enough when running
|
321
|
+
# under ruby 1.9.2, so make it good enough:
|
322
|
+
stdin_reader, stdin_writer = IO.pipe
|
323
|
+
stdin_writer.close
|
324
|
+
STDIN.reopen stdin_reader
|
325
|
+
stdin_reader.close
|
326
|
+
|
327
|
+
stdout_pipe.first.close
|
328
|
+
STDOUT.reopen stdout_pipe.last
|
329
|
+
stdout_pipe.last.close
|
330
|
+
|
331
|
+
stderr_pipe.first.close
|
332
|
+
STDERR.reopen stderr_pipe.last
|
333
|
+
stderr_pipe.last.close
|
334
|
+
|
335
|
+
STDOUT.sync = STDERR.sync = true
|
336
|
+
end
|
337
|
+
|
338
|
+
def configure_parent_process_file_descriptors
|
339
|
+
# Close the sides of the pipes we don't care about
|
340
|
+
stdout_pipe.last.close
|
341
|
+
stderr_pipe.last.close
|
342
|
+
process_status_pipe.last.close
|
343
|
+
# Get output as it happens rather than buffered
|
344
|
+
child_stdout.sync = true
|
345
|
+
child_stderr.sync = true
|
346
|
+
|
347
|
+
true
|
348
|
+
end
|
349
|
+
|
350
|
+
# Some patch levels of ruby in wide use (in particular the ruby 1.8.6 on OSX)
|
351
|
+
# segfault when you IO.select a pipe that's reached eof. Weak sauce.
|
352
|
+
def open_pipes
|
353
|
+
@open_pipes ||= [child_stdout, child_stderr]
|
354
|
+
end
|
355
|
+
|
356
|
+
def read_stdout_to_buffer
|
357
|
+
while chunk = child_stdout.read_nonblock(READ_SIZE)
|
358
|
+
@stdout << chunk
|
359
|
+
end
|
360
|
+
rescue Errno::EAGAIN
|
361
|
+
rescue EOFError
|
362
|
+
open_pipes.delete_at(0)
|
363
|
+
end
|
364
|
+
|
365
|
+
def read_stderr_to_buffer
|
366
|
+
while chunk = child_stderr.read_nonblock(READ_SIZE)
|
367
|
+
@stderr << chunk
|
368
|
+
end
|
369
|
+
rescue Errno::EAGAIN
|
370
|
+
rescue EOFError
|
371
|
+
open_pipes.delete_at(1)
|
372
|
+
end
|
373
|
+
|
374
|
+
def fork_subprocess
|
375
|
+
initialize_ipc
|
376
|
+
|
377
|
+
fork do
|
378
|
+
configure_subprocess_file_descriptors
|
379
|
+
|
380
|
+
set_user
|
381
|
+
set_group
|
382
|
+
set_environment
|
383
|
+
set_umask
|
384
|
+
|
385
|
+
begin
|
386
|
+
command.kind_of?(Array) ? exec(*command) : exec(command)
|
387
|
+
|
388
|
+
raise 'forty-two' # Should never get here
|
389
|
+
rescue Exception => e
|
390
|
+
Marshal.dump(e, process_status_pipe.last)
|
391
|
+
process_status_pipe.last.flush
|
392
|
+
end
|
393
|
+
process_status_pipe.last.close unless (process_status_pipe.last.closed?)
|
394
|
+
exit!
|
395
|
+
end
|
396
|
+
end
|
397
|
+
|
398
|
+
# Attempt to get a Marshaled error from the side-channel.
|
399
|
+
# If it's there, un-marshal it and raise. If it's not there,
|
400
|
+
# assume everything went well.
|
401
|
+
def propagate_pre_exec_failure
|
402
|
+
begin
|
403
|
+
e = Marshal.load child_process_status
|
404
|
+
raise(Exception === e ? e : "unknown failure: #{e.inspect}")
|
405
|
+
rescue EOFError # If we get an EOF error, then the exec was successful
|
406
|
+
true
|
407
|
+
ensure
|
408
|
+
child_process_status.close
|
409
|
+
end
|
410
|
+
end
|
411
|
+
|
412
|
+
end
|
413
|
+
end
|