Merge pull request #366 from ripienaar/347.6

(#347) Bolt Task AAA

.rspec

@@ -1,2 +1 @@
 --color
---format doc

lib/mcollective/agent/bolt_task.rb

@@ -87,6 +87,27 @@ class Bolt_task < RPC::Agent
         end
       end
 
+      # Performs an additional authorization and audit line using the task name as action
+      def before_processing_hook(msg, connection)
+        if respond_to?("authorization_hook")
+          original_action = request.action
+          task = request[:task]
+
+          begin
+            Log.debug(request)
+            if ["run_and_wait", "run_no_wait"].include?(original_action) && task
+              request.action = task
+              authorization_hook(request)
+              audit_request(request, connection)
+            end
+          rescue
+            raise(RPCAborted, "You are not authorized to run Bolt Task %s" % task)
+          ensure
+            request.action = original_action
+          end
+        end
+      end
+
       def reply_task_status(status)
         reply[:exitcode] = status["exitcode"]
         reply[:stdout] = status["stdout"]

lib/mcollective/util/tasks_support.rb

@@ -457,6 +457,8 @@ def file_sha256(file_path)
       # @raise [StandardError] when the file does not exist
       def file_size(file_path)
         File.stat(file_path).size
+      rescue
+        -1
       end
 
       # Validates a task cache file