opennebula_nagios_probe 1.0.0

data/lib/probe/occi/nocci/network.rb

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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.
+###########################################################################
+
+module Occi
+  class Network < Resource
+  end
+end

data/lib/probe/occi/nocci/resource.rb

@@ -0,0 +1,96 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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 'httparty'
+
+module Occi
+# OCCI Resource class.
+  class Resource
+    include HTTParty
+
+    def initialize(connection)
+      self.class.base_uri "#{connection[:endpoint]}"
+      # nebula OCCI format
+      self.class.basic_auth "#{connection[:auth][:username]}", Digest::SHA1.hexdigest(connection[:auth][:password])
+
+      # Low-level debugging
+      # self.class.debug_output
+    end
+
+    # Callback invoked whenever a subclass is created. This method dynamically defines virtual @endpoint
+    # attribute located in child instance, which contains backslash + name of inheriting class. It is used
+    # for request building.
+    def self.inherited(childclass)
+      super(childclass)
+
+      path = childclass.to_s.split('::').last.downcase
+
+      childclass.send(:define_method, :endpoint) { "/#{path}" }
+    end
+
+    def entity(id)
+      "#{endpoint}/#{id}"
+    end
+
+    # Returns the contents of the pool.
+    # 200 OK: An XML representation of the pool in the http body.
+    # This means query the point /network, /storage etc.
+    def all
+      begin
+        response = self.class.get(endpoint)
+      rescue => e
+        raise e.class, 'Could not initiate basic endpoint connectivity query, maybe HTTP/SSL server problem?'
+      ensure
+        if !response.nil?
+          fail HTTPResponseError, "Basic pool availibility request failed! #{response.body}" unless response.code.between?(200, 300)
+          response.body
+        else
+          fail HTTPResponseError, 'Basic pool availibility request failed!'
+        end
+      end
+    end
+
+    # Returns the representation of specific resource identified by +id+.
+    # 200 OK: An XML representation of the pool in the http body.
+    def find(id)
+      begin
+        response = self.class.get(entity(id))
+      rescue => e
+        raise e.class, 'Could not initiate specific resource query, maybe HTTP/SSL server problem?'
+      ensure
+        if !response.nil?
+          fail HTTPResponseError, "Specific resource request failed! #{response.body}" unless response.code.between?(200, 300)
+          response.body
+        else
+          fail HTTPResponseError, 'Specific resource request failed!'
+        end
+      end
+    end
+  end
+
+# HTTPResponseError class.
+# Slightly modified HTTParty::ResponseError
+# for better cooperation with existing code
+
+  class HTTPResponseError < HTTParty::ResponseError
+    attr_reader :message
+
+    def initialize(m)
+      super(m)
+      @message = m
+    end
+  end
+end

data/lib/probe/occi/nocci/storage.rb

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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.
+###########################################################################
+module Occi
+# OCCI Storage class.
+
+  class Storage < Resource
+  end
+end

data/lib/probe/occi/rocci/compute.rb

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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 'occi-cli'
+require 'occi-core'
+require 'occi-api'
+require 'timeout'
+
+module Rocci
+  # OCCI Compute class.
+  class Compute < Resource
+    # Create, check and destroy resource
+    def create_check_destroy
+      # Build resource
+      res = @client.get_resource('compute')
+      res.model = model
+      res.title = @opts.vmname
+      res.hostname = res.title
+
+      # Fill resource mixin
+      if @opts.template.include?('http')
+        orig_mxn = model.get_by_id(@opts.template)
+      else
+        orig_mxn = @client.get_mixin(@opts.template, 'os_tpl', true)
+      end
+
+      if orig_mxn.nil?
+        fail Occi::Api::Client::Errors::AmbiguousNameError, 'Invalid, non-existing or ambiguous mixin (template) name'
+      end
+
+      res.mixins << orig_mxn
+
+      # Create and commit resource
+      response = create(res)
+      new_vm = response.gsub!(@opts.endpoint.chomp('/'), '')
+
+      # Following block checks out for sucessfull VM deployment
+      # and clean up then
+      begin
+        timeout(@opts.timeout) do
+          loop do
+            d = describe(new_vm).first
+            if d.attributes.occi.compute.state == 'active'
+              # puts "OK, resource did enter 'active' state in time"
+              break
+            end
+            sleep @opts.timeout / 5
+          end
+        end
+      rescue Timeout::Error
+        raise Timeout::Error, "Resource did not enter 'active' state in time!"
+      ensure
+        delete new_vm
+      end
+    end
+  end
+end

data/lib/probe/occi/rocci/network.rb

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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.
+###########################################################################
+
+module Rocci
+  class Network < Resource
+  end
+end

data/lib/probe/occi/rocci/resource.rb

@@ -0,0 +1,62 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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 'occi-api'
+
+module Rocci
+  # rOCCI Resource class.
+  class Resource
+    include Occi::Api::Dsl
+
+    def initialize(opts)
+      @opts = opts
+      connect(:http, opts)
+    end
+
+    # Callback invoked whenever a subclass is created. This method dynamically defines virtual @endpoint
+    # attribute located in child instance, which contains backslash + name of inheriting class. It is used
+    # for request building.
+    def self.inherited(childclass)
+      super(childclass)
+
+      path = childclass.to_s.split('::').last.downcase
+
+      childclass.send(:define_method, :resource_uri) { "#{path}" }
+    end
+
+    def entity(id)
+      "/#{resource_uri}/#{id}"
+    end
+
+    # Returns the contents of the pool.
+    # 200 OK: An XML representation of the pool in the http body.
+    # This means query the point "network", "storage" etc.
+    # Please read Occi::Api documentation here https://github.com/arax/rOCCI-api.
+    def all
+      describe(resource_uri)
+    end
+
+    # Returns the representation of specific resource identified by +id+.
+    # 200 OK: An XML representation of the pool in the http body.
+    def find(id)
+      describe(entity(id))
+    end
+
+    def create_check_destroy
+      # Overriden in sibling (compute)
+    end
+  end
+end

data/lib/probe/occi/rocci/storage.rb

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+###########################################################################
+## 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.
+###########################################################################
+module Rocci
+# OCCI Storage class.
+
+  class Storage < Resource
+  end
+end

data/lib/probe/opennebula_econe_probe.rb

@@ -0,0 +1,108 @@
+# encoding: UTF-8
+###########################################################################
+## 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 'opennebula_probe'
+require 'AWS'
+require 'digest/sha1'
+
+# OpenNebulaEconeProbe - Econe client query service implementation.
+
+class OpenNebulaEconeProbe < OpennebulaProbe
+  def initialize(opts)
+    super(opts)
+
+    @connection = AWS::EC2::Base.new(
+        access_key_id:      @opts.username,
+        secret_access_key:  Digest::SHA1.hexdigest(@opts.password),
+        server:             @opts.hostname,
+        port:               @opts.port,
+        path:               @opts.path,
+        use_ssl:            @opts.protocol == :https
+    )
+  end
+
+  def check_crit
+    @logger.info "Checking for basic connectivity at #{@endpoint}"
+    begin
+      @connection.describe_images
+      @connection.describe_instances
+    rescue StandardError => e
+      @logger.error "Failed to check connectivity: #{e.message}"
+      @logger.debug "#{e.backtrace.join("\n")}"
+      return true
+    end
+
+    false
+  end
+
+  def check_resources(resources)
+    if resources.map { |x| x[:resource] }.reduce(true) { |product, resource| product && resource.nil? }
+      @logger.info 'There are no resources to check, for details on how to specify resources see --help'
+      return false
+    end
+
+    resources.each do |resource_hash|
+      resource = resource_hash[:resource]
+
+      next unless resource
+
+      @logger.info "Looking for #{resource_hash[:resource_string]}s: #{resource.inspect}"
+      if resource_hash[:resource_type] == :image
+        result = @connection.describe_images
+        set    = 'imagesSet'
+        id     = 'imageId'
+      elsif resource_hash[:resource_type] == :compute
+        result = @connection.describe_instances
+        result = result['reservationSet']['item'][0] if result['reservationSet'] && result['reservationSet']['item']
+        set    = 'instancesSet'
+        id     = 'amiLaunchIndex'
+      else
+        fail 'Wrong resource definition'
+      end
+
+      @logger.debug result
+
+      fail "No #{resource_hash[:resource_string].capitalize} found" unless result && result[set]
+
+      resource.each do |resource_to_look_for|
+        found = false
+
+        result[set]['item'].each { |resource_found| found = true if resource_to_look_for == resource_found[id] }
+
+        fail "#{resource_hash[:resource_string].capitalize} #{resource_to_look_for} not found" unless found
+      end
+    end
+
+    false
+  end
+
+  def check_warn
+    @logger.info "Checking for resource availability at #{@endpoint}"
+
+    # iterate over given resources
+    @logger.info "Not looking for networks, since it is not supported by OpenNebula's ECONE server'"  if @opts.network
+
+    resources = []
+    resources << { resource_type: :image, resource: @opts.storage, resource_string: 'image' }
+    resources << { resource_type: :compute, resource: @opts.compute, resource_string: 'compute instance' }
+
+    check_resources(resources)
+
+  rescue StandardError => e
+    @logger.error "Failed to check resource availability: #{e.message}"
+    @logger.debug "#{e.backtrace.join("\n")}"
+    return true
+  end
+end

data/lib/probe/opennebula_occi_probe.rb

@@ -0,0 +1,126 @@
+# encoding: UTF-8
+###########################################################################
+## 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 'opennebula_probe'
+require 'occi/client'
+
+# OpenNebulaOcciProbe - OCCI client query service implementation.
+
+class OpenNebulaOcciProbe < OpennebulaProbe
+  def initialize(opts)
+    super(opts)
+
+    if @opts.user_cred
+      creds = {
+          type:               'x509',
+          user_cert:          @opts.user_cred,
+          user_cert_password: @opts.password,
+          ca_path:            @opts.ca_path,
+          ca_file:            @opts.ca_file,
+          voms:               @opts.voms
+      }
+    else
+      creds = {
+          username: @opts.username,
+          password: @opts.password,
+          type:     'basic'
+      }
+    end
+
+    @client = OcciClient.new(
+        endpoint: @endpoint,
+        auth:     creds,
+        occi:     @opts.service,
+        template: @opts.template_uuid,
+        vmname:   @opts.vmname,
+        timeout:  @opts.timeout
+    )
+  end
+
+  def check_crit
+    @logger.info "Checking for basic connectivity at #{@endpoint}"
+
+    begin
+      # make a few simple queries just to be sure that the service is running
+      @client.network.all
+      # Not supported yet
+      @client.compute.all unless @opts.service == 'rocci'
+      @client.storage.all
+    rescue StandardError => e
+      @logger.error "Failed to check connectivity: #{e}"
+      @logger.debug "#{e.backtrace.join("\n")}"
+      return true
+    end
+
+    false
+  end
+
+  def check_resources(resources)
+    # extract key ":resource" from hashes to new array and determine, if any of them are other than nil
+    if resources.map { |x| x[:resource] }.reduce(true) { |product, resource| product && resource.nil? }
+      @logger.info 'There are no resources to check, for details on how to specify resources see --help'
+      return false
+    end
+
+    resources.each do |resource_hash|
+      resource = resource_hash[:resource]
+      next unless resource
+
+      begin
+        @logger.info "Looking for #{resource_hash[:resource_string]}s: #{resource.inspect}"
+        result = resource.map { |id| resource_hash[:resource_connection].find id }
+        @logger.debug result
+      end
+    end
+
+    false
+  end
+
+  def check_warn
+    @logger.info "Checking for resource availability at #{@endpoint}"
+
+    resources = []
+
+    # Not supported yet
+    unless @opts.service == 'rocci'
+      resources << { resource:            @opts.storage,
+                     resource_string:     'image',
+                     resource_connection: @client.storage
+                   }
+    end
+    resources   << { resource:            @opts.compute,
+                     resource_string:     'compute instance',
+                     resource_connection: @client.compute
+                   }
+    resources   << { resource:            @opts.network,
+                     resource_string:     'network',
+                     resource_connection: @client.network
+                   }
+
+    # Additionally create VM from template when using rOCCI if needed
+    if !@opts.template_uuid.nil?
+      @client.compute.create_check_destroy
+    else
+      check_resources(resources)
+    end
+
+    false
+
+  rescue StandardError => e
+    @logger.error "Failed to check resource availability: #{e.message}"
+    @logger.debug "#{e.backtrace.join("\n")}"
+    return true
+  end
+end