kurchatov 0.0.1

data/lib/kurchatov/responders/riemann.rb

@@ -0,0 +1,46 @@
+# encoding: utf-8
+
+require "kurchatov/riemann/client"
+require "kurchatov/mixin/queue"
+
+module Kurchatov
+  module Responders
+    class Riemann < Kurchatov::Plugin
+
+      include Kurchatov::Mixin::Queue
+
+      FLUSH_INTERVAL = 0.5
+
+      def initialize(conn)
+        @hosts = conn
+        @riemanns = Array.new
+      end
+
+      def run
+        make_clients
+        loop { flush; sleep FLUSH_INTERVAL }
+      end
+
+      private
+
+      def make_clients
+        @riemanns.clear
+        @hosts.each do |host|
+          riemann, port = host.split(':')
+          @riemanns << Kurchatov::Riemann::Client.new(:host => riemann, :port => port)
+          @name = @riemanns.map { |c| "riemann client [#{c.host}:#{c.port}]" }.join(' , ')
+        end
+      end
+
+      def flush
+        @events_to_send ||= events.all
+        unless @events_to_send.empty?
+          @riemanns.each {|riemann| riemann << @events_to_send }
+          Log.debug("Sended events via #{@name.inspect}: #{@events_to_send}")
+        end
+        @events_to_send = nil
+      end
+
+    end
+  end
+end

data/lib/kurchatov/responders/udp.rb

@@ -0,0 +1,32 @@
+module Kurchatov
+  module Responders
+    class Udp < Kurchatov::Plugin
+
+      include Kurchatov::Mixin::Event
+      include Kurchatov::Mixin::Queue
+
+      def initialize(conn)
+        @host, @port = conn.split(':')
+        @name = "udp responder #{@host}:#{@port}"
+      end
+
+
+      def run
+        Socket.udp_server_loop(@host, @port) do |data, src|
+          process(data, src)
+        end
+      end
+
+      def process(data,src)
+        begin
+          event << JSON.parse(data)
+          src.reply "sended\n\n"
+        rescue => e
+          src.reply "failed to send: #{data.inspect}\n"
+          Log.error("Failed parse #{data.inspect}, #{e.class}: #{e}\n #{e.backtrace.join("\n")}")
+        end
+      end
+
+    end
+  end
+end

data/lib/kurchatov/riemann/client.rb

@@ -0,0 +1,49 @@
+require "timeout"
+require "socket"
+require "beefcake"
+require "kurchatov/riemann/event"
+require "kurchatov/riemann/message"
+
+module Kurchatov
+  module Riemann
+    class Client
+
+      attr_accessor :host, :port
+
+      CONNECT_TIMEOUT = 5
+      SEND_TIMEOUT = 5
+      RIEMANN_PORT = 5555
+
+      def initialize(opts = {})
+        @host = opts[:host]
+        @port = opts[:port] || RIEMANN_PORT
+        @mutex = Mutex.new
+      end
+
+      def <<(events)
+       events = events.map {|e| Event.new(e) }
+       message = Message.new(:events => events)
+       with_connection do |socket|
+        x = message.encode_with_length
+        Timeout::timeout(SEND_TIMEOUT) {
+          socket.write(x)
+          socket.flush
+        }
+       end
+      end
+
+      def with_connection
+        @mutex.synchronize do
+          yield(@socket || connect)
+        end
+      end
+
+      def connect
+        Timeout::timeout(CONNECT_TIMEOUT) {
+          @socket ||= TCPSocket.new(@host, @port)
+        }
+      end
+
+    end
+  end
+end

data/lib/kurchatov/riemann/event.rb

@@ -0,0 +1,42 @@
+module Kurchatov
+  module Riemann
+    class Event
+      include Beefcake::Message
+      optional :time, :int64, 1
+      optional :state, :string,  2
+      optional :service, :string, 3
+      optional :host, :string, 4
+      optional :description, :string, 5
+      repeated :tags, :string, 7
+      optional :ttl, :float, 8
+      optional :metric_sint64, :sint64, 13
+      optional :metric_d, :double, 14
+      optional :metric_f, :float, 15
+
+      def initialize(hash = nil)
+        if hash
+          super(hash)
+          self.metric = hash[:metric] if hash[:metric]
+        else
+          super
+        end
+        @time ||= Time.now.to_i
+      end
+
+      def metric
+        metric_d || metric_sint64 || metric_f
+      end
+
+      def metric=(m)
+        if Integer === m and (-(2**63)...2**63) === m
+          self.metric_sint64 = m
+          self.metric_f = m.to_f
+        else
+          self.metric_d = m.to_f
+          self.metric_f = m.to_f
+        end
+      end
+
+    end
+  end
+end

data/lib/kurchatov/riemann/message.rb

@@ -0,0 +1,18 @@
+module Kurchatov
+  module Riemann
+    class Message
+      include Beefcake::Message
+
+      optional :ok, :bool, 2
+      optional :error, :string, 3
+      repeated :events, Event, 6
+
+      def encode_with_length
+        buffer = ''
+        encoded = encode buffer
+        "#{[encoded.length].pack('N')}#{encoded}"
+      end
+
+    end
+  end
+end

data/lib/kurchatov/version.rb

@@ -0,0 +1,3 @@
+module Kurchatov
+  VERSION = "0.0.1"
+end

data/lib/kurchatov.rb

@@ -0,0 +1,3 @@
+require "kurchatov/version"
+
+module Kurchatov; end

data/lib/ohai/plugins/darwin/hostname.rb

@@ -0,0 +1,22 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "fqdn", "hostname"
+
+hostname from("hostname -s")
+fqdn from("hostname")

data/lib/ohai/plugins/darwin/platform.rb

@@ -0,0 +1,38 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "platform", "platform_version", "platform_build", "platform_family"
+
+popen4("/usr/bin/sw_vers") do |pid, stdin, stdout, stderr|
+  stdin.close
+  stdout.each do |line|
+    case line
+    when /^ProductName:\s+(.+)$/
+      macname = $1
+      macname.downcase!
+      macname.gsub!(" ", "_")
+      platform macname
+    when /^ProductVersion:\s+(.+)$/
+      platform_version $1
+    when /^BuildVersion:\s+(.+)$/
+      platform_build $1
+    end
+  end
+end
+
+platform_family "mac_os_x"

data/lib/ohai/plugins/hostname.rb

@@ -0,0 +1,27 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "fqdn", "domain"
+
+require_plugin "#{os}::hostname"
+
+# Domain is everything after the first dot
+if fqdn
+  fqdn =~ /.+?\.(.*)/
+  domain $1 
+end

data/lib/ohai/plugins/linux/hostname.rb

@@ -0,0 +1,26 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "hostname", "fqdn"
+
+hostname from("hostname -s")
+begin
+  fqdn from("hostname --fqdn")
+rescue
+  Ohai::Log.debug("hostname -f returned an error, probably no domain is set")
+end

data/lib/ohai/plugins/linux/platform.rb

@@ -0,0 +1,113 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+def get_redhatish_platform(contents)
+  contents[/^Red Hat/i] ? "redhat" : contents[/(\w+)/i, 1].downcase
+end
+
+def get_redhatish_version(contents)
+  contents[/Rawhide/i] ? contents[/((\d+) \(Rawhide\))/i, 1].downcase : contents[/release ([\d\.]+)/, 1]
+end
+
+provides "platform", "platform_version", "platform_family"
+
+require_plugin 'linux::lsb'
+
+# platform [ and platform_version ? ] should be lower case to avoid dealing with RedHat/Redhat/redhat matching 
+if File.exists?("/etc/oracle-release")
+  contents = File.read("/etc/oracle-release").chomp
+  platform "oracle"
+  platform_version get_redhatish_version(contents)
+elsif File.exists?("/etc/enterprise-release")
+  contents = File.read("/etc/enterprise-release").chomp
+  platform "oracle"
+  platform_version get_redhatish_version(contents)
+elsif File.exists?("/etc/debian_version")
+  # Ubuntu and Debian both have /etc/debian_version
+  # Ubuntu should always have a working lsb, debian does not by default
+  if lsb[:id] =~ /Ubuntu/i
+    platform "ubuntu"
+    platform_version lsb[:release]
+  elsif lsb[:id] =~ /LinuxMint/i
+    platform "linuxmint"
+    platform_version lsb[:release]
+  else 
+    if File.exists?("/usr/bin/raspi-config")
+      platform "raspbian"
+    else
+      platform "debian"
+    end
+    platform_version File.read("/etc/debian_version").chomp
+  end
+elsif File.exists?("/etc/redhat-release")
+  contents = File.read("/etc/redhat-release").chomp
+  platform get_redhatish_platform(contents)
+  platform_version get_redhatish_version(contents)
+elsif File.exists?("/etc/system-release")
+  contents = File.read("/etc/system-release").chomp
+  platform get_redhatish_platform(contents)
+  platform_version get_redhatish_version(contents)
+elsif File.exists?('/etc/gentoo-release')
+  platform "gentoo"
+  platform_version File.read('/etc/gentoo-release').scan(/(\d+|\.+)/).join
+elsif File.exists?('/etc/SuSE-release')
+  platform "suse"
+  suse_release = File.read("/etc/SuSE-release")
+  platform_version suse_release.scan(/VERSION = (\d+)\nPATCHLEVEL = (\d+)/).flatten.join(".")
+  platform_version suse_release.scan(/VERSION = ([\d\.]{2,})/).flatten.join(".") if platform_version == ""
+elsif File.exists?('/etc/slackware-version')
+  platform "slackware"
+  platform_version File.read("/etc/slackware-version").scan(/(\d+|\.+)/).join
+elsif File.exists?('/etc/arch-release')
+  platform "arch"
+  # no way to determine platform_version in a rolling release distribution
+  # kernel release will be used - ex. 2.6.32-ARCH
+elsif lsb[:id] =~ /RedHat/i
+  platform "redhat"
+  platform_version lsb[:release]
+elsif lsb[:id] =~ /Amazon/i
+  platform "amazon"
+  platform_version lsb[:release]
+elsif lsb[:id] =~ /ScientificSL/i
+  platform "scientific"
+  platform_version lsb[:release]
+elsif lsb[:id] =~ /XenServer/i
+  platform "xenserver"
+  platform_version lsb[:release]
+elsif lsb[:id] # LSB can provide odd data that changes between releases, so we currently fall back on it rather than dealing with its subtleties 
+  platform lsb[:id].downcase
+  platform_version lsb[:release]
+end
+
+
+case platform
+  when /debian/, /ubuntu/, /linuxmint/, /raspbian/
+    platform_family "debian"
+  when /fedora/
+    platform_family "fedora"
+  when /oracle/, /centos/, /redhat/, /scientific/, /enterpriseenterprise/, /amazon/, /xenserver/ # Note that 'enterpriseenterprise' is oracle's LSB "distributor ID"
+    platform_family "rhel"
+  when /suse/
+    platform_family "suse"
+  when /gentoo/
+    platform_family "gentoo"
+  when /slackware/
+    platform_family "slackware"
+  when /arch/ 
+    platform_family "arch" 
+end

data/lib/ohai/plugins/linux/virtualization.rb

@@ -0,0 +1,125 @@
+#
+# Author:: Thom May (<thom@clearairturbulence.org>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "virtualization"
+
+virtualization Mash.new
+
+# if it is possible to detect paravirt vs hardware virt, it should be put in
+# virtualization[:mechanism]
+
+## Xen
+# /proc/xen is an empty dir for EL6 + Linode Guests
+if File.exists?("/proc/xen")
+  virtualization[:system] = "xen"
+  # Assume guest
+  virtualization[:role] = "guest"
+
+  # This file should exist on most Xen systems, normally empty for guests
+  if File.exists?("/proc/xen/capabilities")
+    if File.read("/proc/xen/capabilities") =~ /control_d/i
+      virtualization[:role] = "host"
+    end
+  end
+end
+
+# Xen Notes:
+# - cpuid of guests, if we could get it, would also be a clue
+# - may be able to determine if under paravirt from /dev/xen/evtchn (See OHAI-253)
+# - EL6 guests carry a 'hypervisor' cpu flag
+# - Additional edge cases likely should not change the above assumptions
+#   but rather be additive - btm
+
+# Detect from kernel module
+if File.exists?("/proc/modules")
+  modules = File.read("/proc/modules")
+  if modules =~ /^kvm/
+    virtualization[:system] = "kvm"
+    virtualization[:role] = "host"
+  elsif modules =~ /^vboxdrv/
+    virtualization[:system] = "vbox"
+    virtualization[:role] = "host"
+  elsif modules =~ /^vboxguest/
+    virtualization[:system] = "vbox"
+    virtualization[:role] = "guest"
+  end
+end
+
+# Detect KVM/QEMU from cpuinfo, report as KVM
+# We could pick KVM from 'Booting paravirtualized kernel on KVM' in dmesg
+# 2.6.27-9-server (intrepid) has this / 2.6.18-6-amd64 (etch) does not
+# It would be great if we could read pv_info in the kernel
+# Wait for reply to: http://article.gmane.org/gmane.comp.emulators.kvm.devel/27885
+if File.exists?("/proc/cpuinfo")
+  if File.read("/proc/cpuinfo") =~ /QEMU Virtual CPU/
+    virtualization[:system] = "kvm"
+    virtualization[:role] = "guest"
+  end
+end
+
+# Detect OpenVZ / Virtuozzo.
+# http://wiki.openvz.org/BC_proc_entries
+if File.exists?("/proc/bc/0")
+  virtualization[:system] = "openvz"
+  virtualization[:role] = "host"
+elsif File.exists?("/proc/vz")
+  virtualization[:system] = "openvz"
+  virtualization[:role] = "guest"
+end
+
+# http://www.dmo.ca/blog/detecting-virtualization-on-linux
+if File.exists?("/usr/sbin/dmidecode")
+  popen4("dmidecode") do |pid, stdin, stdout, stderr|
+    stdin.close
+    dmi_info = stdout.read
+    case dmi_info
+    when /Manufacturer: Microsoft/
+      if dmi_info =~ /Product Name: Virtual Machine/
+        virtualization[:system] = "virtualpc"
+        virtualization[:role] = "guest"
+      end
+    when /Manufacturer: VMware/
+      if dmi_info =~ /Product Name: VMware Virtual Platform/
+        virtualization[:system] = "vmware"
+        virtualization[:role] = "guest"
+      end
+    when /Manufacturer: Xen/
+      if dmi_info =~ /Product Name: HVM domU/
+        virtualization[:system] = "xen"
+        virtualization[:role] = "guest"
+      end
+    else
+      nil
+    end
+
+  end
+end
+
+# Detect Linux-VServer
+if File.exists?("/proc/self/status")
+  proc_self_status = File.read("/proc/self/status")
+  vxid = proc_self_status.match(/^(s_context|VxID): (\d+)$/)
+  if vxid and vxid[2]
+    virtualization[:system] = "linux-vserver"
+    if vxid[2] == "0"
+      virtualization[:role] = "host"
+    else
+      virtualization[:role] = "guest"
+     end
+  end
+end

data/lib/ohai/plugins/os.rb

@@ -0,0 +1,53 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "os", "os_version"
+
+require 'rbconfig'
+
+require_plugin 'kernel'
+
+case ::RbConfig::CONFIG['host_os']
+when /aix(.+)$/
+  os "aix"
+when /darwin(.+)$/
+  os "darwin"
+when /hpux(.+)$/
+  os "hpux"
+when /linux/
+  os "linux"
+when /freebsd(.+)$/
+  os "freebsd"
+when /openbsd(.+)$/
+  os "openbsd"
+when /netbsd(.*)$/
+  os "netbsd"
+when /solaris2/
+  os "solaris2"
+when /mswin|mingw32|windows/
+  # After long discussion in IRC the "powers that be" have come to a concensus
+  # that there is no other Windows platforms exist that were not based on the
+  # Windows_NT kernel, so we herby decree that "windows" will refer to all
+  # platforms built upon the Windows_NT kernel and have access to win32 or win64
+  # subsystems.
+  os "windows"
+else
+  os ::RbConfig::CONFIG['host_os']
+end
+
+os_version kernel[:release]

data/lib/ohai/plugins/platform.rb

@@ -0,0 +1,28 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+provides "platform", "platform_version", "platform_family"
+
+require_plugin "#{os}::platform"
+
+platform os unless attribute?("platform")
+
+platform_version os_version unless attribute?("platform_version")
+
+platform_family platform unless attribute?("platform_family")
+