right_agent 0.5.1
Sign up to get free protection for your applications and to get access to all the features.
- data/LICENSE +20 -0
- data/README.rdoc +78 -0
- data/Rakefile +86 -0
- data/lib/right_agent.rb +66 -0
- data/lib/right_agent/actor.rb +163 -0
- data/lib/right_agent/actor_registry.rb +76 -0
- data/lib/right_agent/actors/agent_manager.rb +189 -0
- data/lib/right_agent/agent.rb +735 -0
- data/lib/right_agent/agent_config.rb +403 -0
- data/lib/right_agent/agent_identity.rb +209 -0
- data/lib/right_agent/agent_tags_manager.rb +213 -0
- data/lib/right_agent/audit_formatter.rb +107 -0
- data/lib/right_agent/broker_client.rb +683 -0
- data/lib/right_agent/command.rb +30 -0
- data/lib/right_agent/command/agent_manager_commands.rb +134 -0
- data/lib/right_agent/command/command_client.rb +136 -0
- data/lib/right_agent/command/command_constants.rb +42 -0
- data/lib/right_agent/command/command_io.rb +128 -0
- data/lib/right_agent/command/command_parser.rb +87 -0
- data/lib/right_agent/command/command_runner.rb +105 -0
- data/lib/right_agent/command/command_serializer.rb +63 -0
- data/lib/right_agent/console.rb +65 -0
- data/lib/right_agent/core_payload_types.rb +42 -0
- data/lib/right_agent/core_payload_types/cookbook.rb +61 -0
- data/lib/right_agent/core_payload_types/cookbook_position.rb +46 -0
- data/lib/right_agent/core_payload_types/cookbook_repository.rb +116 -0
- data/lib/right_agent/core_payload_types/cookbook_sequence.rb +70 -0
- data/lib/right_agent/core_payload_types/dev_repositories.rb +90 -0
- data/lib/right_agent/core_payload_types/event_categories.rb +38 -0
- data/lib/right_agent/core_payload_types/executable_bundle.rb +138 -0
- data/lib/right_agent/core_payload_types/login_policy.rb +72 -0
- data/lib/right_agent/core_payload_types/login_user.rb +62 -0
- data/lib/right_agent/core_payload_types/planned_volume.rb +94 -0
- data/lib/right_agent/core_payload_types/recipe_instantiation.rb +60 -0
- data/lib/right_agent/core_payload_types/repositories_bundle.rb +50 -0
- data/lib/right_agent/core_payload_types/right_script_attachment.rb +95 -0
- data/lib/right_agent/core_payload_types/right_script_instantiation.rb +73 -0
- data/lib/right_agent/core_payload_types/secure_document.rb +66 -0
- data/lib/right_agent/core_payload_types/secure_document_location.rb +63 -0
- data/lib/right_agent/core_payload_types/software_repository_instantiation.rb +61 -0
- data/lib/right_agent/daemonize.rb +35 -0
- data/lib/right_agent/dispatcher.rb +348 -0
- data/lib/right_agent/enrollment_result.rb +217 -0
- data/lib/right_agent/exceptions.rb +30 -0
- data/lib/right_agent/ha_broker_client.rb +1278 -0
- data/lib/right_agent/idempotent_request.rb +140 -0
- data/lib/right_agent/log.rb +418 -0
- data/lib/right_agent/monkey_patches.rb +29 -0
- data/lib/right_agent/monkey_patches/amqp_patch.rb +274 -0
- data/lib/right_agent/monkey_patches/ruby_patch.rb +49 -0
- data/lib/right_agent/monkey_patches/ruby_patch/array_patch.rb +29 -0
- data/lib/right_agent/monkey_patches/ruby_patch/darwin_patch.rb +24 -0
- data/lib/right_agent/monkey_patches/ruby_patch/linux_patch.rb +24 -0
- data/lib/right_agent/monkey_patches/ruby_patch/linux_patch/file_patch.rb +30 -0
- data/lib/right_agent/monkey_patches/ruby_patch/object_patch.rb +49 -0
- data/lib/right_agent/monkey_patches/ruby_patch/singleton_patch.rb +46 -0
- data/lib/right_agent/monkey_patches/ruby_patch/string_patch.rb +107 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch.rb +32 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch/file_patch.rb +90 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch/process_patch.rb +63 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch/stdio_patch.rb +27 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch/time_patch.rb +55 -0
- data/lib/right_agent/monkey_patches/ruby_patch/windows_patch/win32ole_patch.rb +34 -0
- data/lib/right_agent/multiplexer.rb +91 -0
- data/lib/right_agent/operation_result.rb +270 -0
- data/lib/right_agent/packets.rb +637 -0
- data/lib/right_agent/payload_formatter.rb +104 -0
- data/lib/right_agent/pid_file.rb +159 -0
- data/lib/right_agent/platform.rb +319 -0
- data/lib/right_agent/platform/darwin.rb +227 -0
- data/lib/right_agent/platform/linux.rb +268 -0
- data/lib/right_agent/platform/windows.rb +1204 -0
- data/lib/right_agent/scripts/agent_controller.rb +522 -0
- data/lib/right_agent/scripts/agent_deployer.rb +379 -0
- data/lib/right_agent/scripts/common_parser.rb +153 -0
- data/lib/right_agent/scripts/log_level_manager.rb +193 -0
- data/lib/right_agent/scripts/stats_manager.rb +256 -0
- data/lib/right_agent/scripts/usage.rb +58 -0
- data/lib/right_agent/secure_identity.rb +92 -0
- data/lib/right_agent/security.rb +32 -0
- data/lib/right_agent/security/cached_certificate_store_proxy.rb +63 -0
- data/lib/right_agent/security/certificate.rb +102 -0
- data/lib/right_agent/security/certificate_cache.rb +89 -0
- data/lib/right_agent/security/distinguished_name.rb +56 -0
- data/lib/right_agent/security/encrypted_document.rb +84 -0
- data/lib/right_agent/security/rsa_key_pair.rb +76 -0
- data/lib/right_agent/security/signature.rb +86 -0
- data/lib/right_agent/security/static_certificate_store.rb +69 -0
- data/lib/right_agent/sender.rb +937 -0
- data/lib/right_agent/serialize.rb +29 -0
- data/lib/right_agent/serialize/message_pack.rb +102 -0
- data/lib/right_agent/serialize/secure_serializer.rb +131 -0
- data/lib/right_agent/serialize/secure_serializer_initializer.rb +47 -0
- data/lib/right_agent/serialize/serializable.rb +135 -0
- data/lib/right_agent/serialize/serializer.rb +149 -0
- data/lib/right_agent/stats_helper.rb +731 -0
- data/lib/right_agent/subprocess.rb +38 -0
- data/lib/right_agent/tracer.rb +124 -0
- data/right_agent.gemspec +60 -0
- data/spec/actor_registry_spec.rb +81 -0
- data/spec/actor_spec.rb +99 -0
- data/spec/agent_config_spec.rb +226 -0
- data/spec/agent_identity_spec.rb +75 -0
- data/spec/agent_spec.rb +571 -0
- data/spec/broker_client_spec.rb +961 -0
- data/spec/command/agent_manager_commands_spec.rb +51 -0
- data/spec/command/command_io_spec.rb +93 -0
- data/spec/command/command_parser_spec.rb +79 -0
- data/spec/command/command_runner_spec.rb +72 -0
- data/spec/command/command_serializer_spec.rb +51 -0
- data/spec/core_payload_types/dev_repositories_spec.rb +64 -0
- data/spec/core_payload_types/executable_bundle_spec.rb +59 -0
- data/spec/core_payload_types/login_user_spec.rb +98 -0
- data/spec/core_payload_types/right_script_attachment_spec.rb +65 -0
- data/spec/core_payload_types/spec_helper.rb +23 -0
- data/spec/dispatcher_spec.rb +372 -0
- data/spec/enrollment_result_spec.rb +53 -0
- data/spec/ha_broker_client_spec.rb +1673 -0
- data/spec/idempotent_request_spec.rb +136 -0
- data/spec/log_spec.rb +177 -0
- data/spec/monkey_patches/amqp_patch_spec.rb +100 -0
- data/spec/monkey_patches/eventmachine_spec.rb +62 -0
- data/spec/monkey_patches/string_patch_spec.rb +99 -0
- data/spec/multiplexer_spec.rb +48 -0
- data/spec/operation_result_spec.rb +171 -0
- data/spec/packets_spec.rb +418 -0
- data/spec/platform/platform_spec.rb +60 -0
- data/spec/results_mock.rb +45 -0
- data/spec/secure_identity_spec.rb +50 -0
- data/spec/security/cached_certificate_store_proxy_spec.rb +56 -0
- data/spec/security/certificate_cache_spec.rb +71 -0
- data/spec/security/certificate_spec.rb +49 -0
- data/spec/security/distinguished_name_spec.rb +46 -0
- data/spec/security/encrypted_document_spec.rb +55 -0
- data/spec/security/rsa_key_pair_spec.rb +55 -0
- data/spec/security/signature_spec.rb +66 -0
- data/spec/security/static_certificate_store_spec.rb +52 -0
- data/spec/sender_spec.rb +887 -0
- data/spec/serialize/message_pack_spec.rb +131 -0
- data/spec/serialize/secure_serializer_spec.rb +102 -0
- data/spec/serialize/serializable_spec.rb +90 -0
- data/spec/serialize/serializer_spec.rb +174 -0
- data/spec/spec.opts +2 -0
- data/spec/spec_helper.rb +77 -0
- data/spec/stats_helper_spec.rb +681 -0
- data/spec/tracer_spec.rb +114 -0
- metadata +320 -0
@@ -0,0 +1,49 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
class Object
|
24
|
+
unless defined? instance_exec # 1.9 and 1.8.7
|
25
|
+
module InstanceExecHelper; end
|
26
|
+
include InstanceExecHelper
|
27
|
+
|
28
|
+
# Evaluate the block with the given arguments within the context of
|
29
|
+
# this object, so self is set to the method receiver.
|
30
|
+
#
|
31
|
+
# From Mauricio's http://eigenclass.org/hiki/bounded+space+instance_exec
|
32
|
+
def instance_exec(*args, &block)
|
33
|
+
begin
|
34
|
+
old_critical, Thread.critical = Thread.critical, true
|
35
|
+
n = 0
|
36
|
+
n += 1 while respond_to?(method_name = "__instance_exec#{n}")
|
37
|
+
InstanceExecMethods.module_eval { define_method(method_name, &block) }
|
38
|
+
ensure
|
39
|
+
Thread.critical = old_critical
|
40
|
+
end
|
41
|
+
|
42
|
+
begin
|
43
|
+
send(method_name, *args)
|
44
|
+
ensure
|
45
|
+
InstanceExecMethods.module_eval { remove_method(method_name) } rescue nil
|
46
|
+
end
|
47
|
+
end
|
48
|
+
end
|
49
|
+
end
|
@@ -0,0 +1,46 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
require 'singleton'
|
24
|
+
|
25
|
+
module RightScale
|
26
|
+
|
27
|
+
module Singleton
|
28
|
+
|
29
|
+
module ClassMethods
|
30
|
+
|
31
|
+
# Redirect class methods to singleton instance
|
32
|
+
def method_missing(meth, *args, &blk)
|
33
|
+
self.instance.__send__(meth, *args, &blk)
|
34
|
+
end
|
35
|
+
|
36
|
+
end
|
37
|
+
|
38
|
+
# Upon inclusion, also include standard Singleton mixin
|
39
|
+
def self.included(base)
|
40
|
+
base.__send__(:include, ::Singleton)
|
41
|
+
base.__send__(:extend, ClassMethods)
|
42
|
+
end
|
43
|
+
|
44
|
+
end
|
45
|
+
|
46
|
+
end
|
@@ -0,0 +1,107 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
class String
|
24
|
+
##
|
25
|
+
# Convert to snake case.
|
26
|
+
#
|
27
|
+
# "FooBar".snake_case #=> "foo_bar"
|
28
|
+
# "HeadlineCNNNews".snake_case #=> "headline_cnn_news"
|
29
|
+
# "CNN".snake_case #=> "cnn"
|
30
|
+
#
|
31
|
+
# @return [String] Receiver converted to snake case.
|
32
|
+
#
|
33
|
+
# @api public
|
34
|
+
def snake_case
|
35
|
+
return downcase if match(/\A[A-Z]+\z/)
|
36
|
+
gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
|
37
|
+
gsub(/([a-z])([A-Z])/, '\1_\2').
|
38
|
+
downcase
|
39
|
+
end
|
40
|
+
|
41
|
+
##
|
42
|
+
# Convert a constant name to a path, assuming a conventional structure.
|
43
|
+
#
|
44
|
+
# "FooBar::Baz".to_const_path # => "foo_bar/baz"
|
45
|
+
#
|
46
|
+
# @return [String] Path to the file containing the constant named by receiver
|
47
|
+
# (constantized string), assuming a conventional structure.
|
48
|
+
#
|
49
|
+
# @api public
|
50
|
+
def to_const_path
|
51
|
+
snake_case.gsub(/::/, "/")
|
52
|
+
end
|
53
|
+
|
54
|
+
# Convert constant name to constant
|
55
|
+
#
|
56
|
+
# "FooBar::Baz".to_const => FooBar::Baz
|
57
|
+
#
|
58
|
+
# @return [Constant] Constant corresponding to given name or nil if no
|
59
|
+
# constant with that name exists
|
60
|
+
#
|
61
|
+
# @api public
|
62
|
+
def to_const
|
63
|
+
names = split('::')
|
64
|
+
names.shift if names.empty? || names.first.empty?
|
65
|
+
|
66
|
+
constant = Object
|
67
|
+
names.each do |name|
|
68
|
+
# modified to return nil instead of raising an const_missing error
|
69
|
+
constant = constant && constant.const_defined?(name) ? constant.const_get(name) : nil
|
70
|
+
end
|
71
|
+
constant
|
72
|
+
end
|
73
|
+
|
74
|
+
# Reverse operation of snake case:
|
75
|
+
#
|
76
|
+
# "some_string/some_other_string" => "SomeString::SomeOtherString"
|
77
|
+
#
|
78
|
+
# @return [String] Camelized string
|
79
|
+
#
|
80
|
+
# @api public
|
81
|
+
if !String.public_method_defined?(:camelize)
|
82
|
+
def camelize(first_letter = :upper)
|
83
|
+
case first_letter
|
84
|
+
when :upper then gsub(/\/(.?)/) { "::" + $1.upcase }.gsub(/(^|_)(.)/) { $2.upcase }
|
85
|
+
when :lower then first + camelize(self)[1..-1]
|
86
|
+
end
|
87
|
+
end
|
88
|
+
end
|
89
|
+
|
90
|
+
# Add ability to output colored text to console
|
91
|
+
# e.g.: puts "Hello".red
|
92
|
+
def bold; colorize("\e[1m\e[29m"); end
|
93
|
+
def grey; colorize("\e[30m"); end
|
94
|
+
def red; colorize("\e[1m\e[31m"); end
|
95
|
+
def dark_red; colorize("\e[31m"); end
|
96
|
+
def green; colorize("\e[1m\e[32m"); end
|
97
|
+
def dark_green; colorize("\e[32m"); end
|
98
|
+
def yellow; colorize("\e[1m\e[33m"); end
|
99
|
+
def blue; colorize("\e[1m\e[34m"); end
|
100
|
+
def dark_blue; colorize("\e[34m"); end
|
101
|
+
def pur; colorize("\e[1m\e[35m"); end
|
102
|
+
def colorize(color_code)
|
103
|
+
# Doesn't work with the Windows prompt...
|
104
|
+
(RightScale::Platform.windows? || !$stdout.isatty) ? to_s : "#{color_code}#{to_s}\e[0m"
|
105
|
+
end
|
106
|
+
|
107
|
+
end
|
@@ -0,0 +1,32 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
# load File monkey-patch first to enable use of File.normalize_path
|
24
|
+
require File.expand_path(File.join(File.dirname(__FILE__), 'windows_patch', 'file_patch'))
|
25
|
+
|
26
|
+
# hereafter used normalize_path to ensure files are only required once.
|
27
|
+
WINDOWS_MONKEY_PATCHES_BASE_DIR = File.normalize_path(File.join(File.dirname(__FILE__), 'windows_patch'))
|
28
|
+
|
29
|
+
require File.join(WINDOWS_MONKEY_PATCHES_BASE_DIR, 'process_patch')
|
30
|
+
require File.join(WINDOWS_MONKEY_PATCHES_BASE_DIR, 'stdio_patch')
|
31
|
+
require File.join(WINDOWS_MONKEY_PATCHES_BASE_DIR, 'time_patch')
|
32
|
+
require File.join(WINDOWS_MONKEY_PATCHES_BASE_DIR, 'win32ole_patch')
|
@@ -0,0 +1,90 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
require 'rubygems'
|
24
|
+
|
25
|
+
begin
|
26
|
+
require 'windows/file'
|
27
|
+
|
28
|
+
class File
|
29
|
+
|
30
|
+
# converts a long path to a short path. in windows terms, this means
|
31
|
+
# taking any file/folder name over 8 characters in length and truncating
|
32
|
+
# it to 6 characters with ~1..~n appended depending on how many similar
|
33
|
+
# names exist in the same directory. file extensions are simply chopped
|
34
|
+
# at three letters. the short name is equivalent for all API calls to
|
35
|
+
# the long path but requires no special quoting, etc. the path must
|
36
|
+
# exist at least partially for the API call to succeed.
|
37
|
+
#
|
38
|
+
# === Parameters
|
39
|
+
# long_path(String):: fully or partially existing long path to be
|
40
|
+
# converted to its short path equivalent.
|
41
|
+
#
|
42
|
+
# === Return
|
43
|
+
# short_path(String):: short path equivalent or same path if non-existent
|
44
|
+
def self.long_path_to_short_path(long_path)
|
45
|
+
if File.exists?(long_path)
|
46
|
+
length = 260
|
47
|
+
while true
|
48
|
+
buffer = 0.chr * length
|
49
|
+
length = ::Windows::File::GetShortPathName.call(long_path, buffer, buffer.length)
|
50
|
+
if length < buffer.length
|
51
|
+
break
|
52
|
+
end
|
53
|
+
end
|
54
|
+
return buffer.unpack('A*').first.gsub("\\", "/")
|
55
|
+
else
|
56
|
+
# must get short path for any existing ancestor since child doesn't
|
57
|
+
# (currently) exist.
|
58
|
+
child_name = File.basename(long_path)
|
59
|
+
long_parent_path = File.dirname(long_path)
|
60
|
+
|
61
|
+
# note that root dirname is root itself (at least in windows)
|
62
|
+
return long_path if long_path == long_parent_path
|
63
|
+
|
64
|
+
# recursion
|
65
|
+
short_parent_path = long_path_to_short_path(File.dirname(long_path))
|
66
|
+
return File.join(short_parent_path, child_name)
|
67
|
+
end
|
68
|
+
end
|
69
|
+
|
70
|
+
# First expand the path then shorten the directory.
|
71
|
+
# Only shorten the directory and not the file name because 'gem' wants
|
72
|
+
# long file names
|
73
|
+
def self.normalize_path(file_name, *dir_string)
|
74
|
+
path = File.expand_path(file_name, *dir_string)
|
75
|
+
dir = self.long_path_to_short_path(File.dirname(path))
|
76
|
+
File.join(dir, File.basename(path))
|
77
|
+
end
|
78
|
+
|
79
|
+
end
|
80
|
+
|
81
|
+
rescue LoadError
|
82
|
+
# use the simple definition of normalize_path to avoid breaking code which
|
83
|
+
# depends on having this definition.
|
84
|
+
class File
|
85
|
+
def self.normalize_path(file_name, *dir_string)
|
86
|
+
File.expand_path(file_name, *dir_string)
|
87
|
+
end
|
88
|
+
end
|
89
|
+
|
90
|
+
end
|
@@ -0,0 +1,63 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
require 'rubygems'
|
24
|
+
|
25
|
+
begin
|
26
|
+
# win32/process monkey-patches the Process class but drops support for any kill
|
27
|
+
# signals which are not directly portable. some signals are acceptable, if not
|
28
|
+
# strictly portable. the 'TERM' signal used to be supported in Ruby v1.8.6 but
|
29
|
+
# raises an exception in Ruby v1.8.7. we will monkey-patch the monkey-patch to
|
30
|
+
# get the best possible implementation of signals.
|
31
|
+
WIN32_PROCESS_LOAD_ORDER_ERROR = "Must require process_patch before win32/process"
|
32
|
+
|
33
|
+
module Process
|
34
|
+
unless defined?(@@ruby_c_kill)
|
35
|
+
@@ruby_c_kill = method(:kill)
|
36
|
+
|
37
|
+
# check to ensure this is the first time 'win32/process' has been required
|
38
|
+
raise LoadError, WIN32_PROCESS_LOAD_ORDER_ERROR unless require 'win32/process'
|
39
|
+
|
40
|
+
@@win32_kill = method(:kill)
|
41
|
+
|
42
|
+
def self.kill(sig, *pids)
|
43
|
+
sig = 1 if 'TERM' == sig # Signals 1 and 4-8 kill the process in a nice manner.
|
44
|
+
@@win32_kill.call(sig, *pids)
|
45
|
+
end
|
46
|
+
|
47
|
+
# implements getpgid() for Windws
|
48
|
+
def self.getpgid(pid)
|
49
|
+
# FIX: we currently only use this to check if the process is running.
|
50
|
+
# it is possible to get the parent process id for a process in Windows if
|
51
|
+
# we actually need this info.
|
52
|
+
return Process.kill(0, pid).contains?(pid) ? 0 : -1
|
53
|
+
rescue
|
54
|
+
raise Errno::ESRCH
|
55
|
+
end
|
56
|
+
end
|
57
|
+
end
|
58
|
+
|
59
|
+
rescue LoadError => e
|
60
|
+
# ignore load error and skip monkey-patch if gems are not yet installed.
|
61
|
+
# re-raise if the error condition we explicitly check for.
|
62
|
+
raise if WIN32_PROCESS_LOAD_ORDER_ERROR == e.message
|
63
|
+
end
|
@@ -0,0 +1,27 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
# it's useful to auto-flush console output on Windows where it would otherwise
|
24
|
+
# be unavailable from a ruby process which was hanging (our code would never
|
25
|
+
# hang, of course). this is consistent with Powershell's auto-flush behavior.
|
26
|
+
STDOUT.sync = true
|
27
|
+
STDERR.sync = true
|
@@ -0,0 +1,55 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2011 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
|
23
|
+
require 'rubygems'
|
24
|
+
|
25
|
+
begin
|
26
|
+
require 'windows/time'
|
27
|
+
|
28
|
+
# monkey patch Time.now because Windows Ruby interpreters used the wrong API
|
29
|
+
# and queried local time instead of UTC time prior to Ruby v1.9.1. This
|
30
|
+
# made the Ruby 1.8.x interpreters vulnerable to external changes to
|
31
|
+
# timezone which cause Time.now to return times which are offset from from the
|
32
|
+
# correct value. This implementation is borrowed from the C source for Ruby
|
33
|
+
# v1.9.1 from www.ruby-lang.org ("win32/win32.c").
|
34
|
+
if RUBY_VERSION < "1.9.1"
|
35
|
+
class Time
|
36
|
+
def self.now
|
37
|
+
# query UTC time as a 64-bit ularge value.
|
38
|
+
filetime = 0.chr * 8
|
39
|
+
::Windows::Time::GetSystemTimeAsFileTime.call(filetime)
|
40
|
+
low_date_time = filetime[0,4].unpack('V')[0]
|
41
|
+
high_date_time = filetime[4,4].unpack('V')[0]
|
42
|
+
value = high_date_time * 0x100000000 + low_date_time
|
43
|
+
|
44
|
+
# value is now 100-nanosec intervals since 1601/01/01 00:00:00 UTC,
|
45
|
+
# convert it into UNIX time (since 1970/01/01 00:00:00 UTC).
|
46
|
+
value /= 10 # 100-nanoseconds to microseconds
|
47
|
+
microseconds = 1000 * 1000
|
48
|
+
value -= ((1970 - 1601) * 365.2425).to_i * 24 * 60 * 60 * microseconds
|
49
|
+
return Time.at(value / microseconds, value % microseconds)
|
50
|
+
end
|
51
|
+
end
|
52
|
+
end
|
53
|
+
rescue LoadError
|
54
|
+
# ignore load error and skip monkey-patch if gems are not yet installed.
|
55
|
+
end
|