rha 0.1.0

data/MIT-LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2011 Joel Bryan Juliano <joelbryan.juliano@gmail.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README

@@ -0,0 +1,73 @@
+README for Rha
+==================
+
+Rha is a gem providing a ruby interface to Heartbeat High Availability Cluster.
+
+To install, type 'gem install rha'
+
+Usage:
+
+      require 'rubygems'
+      require 'rha'
+      
+      ha_cf = RhaConfig.new
+      ha_cf.use_logd = "on"
+      ha_cf.mcast = "eth0 239.0.0.42 694 1 0"
+      ha_cf.bcast = "eth1"
+      ha_cf.node = "alice bob"
+      ha_cf.keepalive = "1000ms"
+      ha_cf.deadtime = "30000ms"
+      ha_cf.realtime = true
+      ha_cf.conn_logd_time = "60"
+      ha_cf.crm = "false"
+      ha_cf.compression = "zlib"
+      ha_cf.compression_threshold = 2
+
+      authkeys = RhaAuthkeys.new
+      authkeys.auth(1)
+      authkeys.add_md5(1, 1, "secret")
+      authkeys.add_sha1(1, 2, "secret")
+      authkeys.add_crc(1, 3)
+      
+      haresources = RhaResources.new
+      haresources.preferred_node = "alice"
+      haresources.resource_name = "192.168.1.10"
+      haresources.script_name = "sshd nginx mysql"
+
+      ha_cf_file = File.new("ha.cf", "w+")
+      ha_cf_file.puts(ha_cf.config)
+
+      authkeys_file = File.new("authkeys", "w+")
+      authkeys_file.puts(authkeys.config)
+      
+      haresources_file = File.new("haresources", "w+")
+      haresources_file.puts(haresources.config)
+
+      ha_cf_file.close      
+      authkeys_file.close
+      haresources_file.close
+
+ha.cf:
+
+      use_logd on 
+      bcast eth1 
+      compression zlib 
+      compression_threshold 2 
+      conn_logd_time 60 
+      crm false 
+      deadtime 30000ms 
+      keepalive 1000ms 
+      mcast eth0 239.0.0.42 694 1 0 
+      node alice bob 
+      realtime true 
+
+authkeys:
+
+      auth 1
+      3 crc 
+      2 sha1 secret
+      1 md5 secret
+
+haresources:
+
+      alice 192.168.1.10 sshd nginx mysql

data/lib/rha/authkeys.rb

@@ -0,0 +1,80 @@
+  # = rha - A Heartbeat gem for Ruby
+  #
+  # Homepage::  http://github.com/jjuliano/rha
+  # Author::    Joel Bryan Juliano
+  # Copyright:: (cc) 2011 Joel Bryan Juliano
+  # License::   MIT
+
+  #
+  # class RhaAuthkeys.new( array, str, array)
+  #
+
+  #
+  # RhaAuthkeys are authentication configurations for heartbeat.
+  #
+  class RhaAuthkeys
+    #
+    # Returns a new RhaConfig Object
+    #
+    def initialize()
+      @num ||= []
+    end
+
+    # 
+    # auth num selects the currently active authentication method and secret.
+    #
+    # num is a numerical identifier, between 1 and 15 inclusive.
+    # It must be unique within the file.
+    #
+    def auth(value)
+      @num.push("auth #{value}")
+    end
+
+    #
+    # MD5 hash method. This method requires a shared secret.
+    #
+    def add_md5(auth_num, method_num, secret)
+      begin
+        num_index = @num.index("auth #{auth_num}")
+        @num.insert(num_index + 1, method_num.to_s + " md5 " + secret.to_s + "\n")
+      rescue
+        puts "no such auth num #{auth_num}"
+        return false
+      end
+    end
+
+    #
+    # SHA-1 hash method. This method requires a shared secret.
+    #
+    def add_sha1(auth_num, method_num, secret)
+      begin
+        num_index = @num.index("auth #{auth_num}")
+        @num.insert(num_index + 1, method_num.to_s + " sha1 " + secret.to_s + "\n")
+      rescue
+        puts "no such auth num #{auth_num}"
+        return false
+      end
+    end
+
+    #
+    # Cyclic Redundancy Check hash method. This method does not require
+    # a shared secret and is insecure; it's use is strongly discouraged.
+    #
+    def add_crc(auth_num, method_num)
+      begin
+        num_index = @num.index("auth #{auth_num}")
+        @num.insert(num_index + 1, method_num.to_s + " crc " + "\n")
+      rescue
+        puts "no such auth num #{auth_num}"
+        return false
+      end
+    end
+
+    #
+    # Return the RhaAuthkeys configuration
+    #
+    def config
+      return @num
+    end
+    
+  end

data/lib/rha/daemon.rb

@@ -0,0 +1,227 @@
+  # = rha - A Heartbeat gem for Ruby
+  #
+  # Homepage::  http://github.com/jjuliano/rha
+  # Author::    Joel Bryan Juliano
+  # Copyright:: (cc) 2011 Joel Bryan Juliano
+  # License::   MIT
+
+  #
+  # class RhaDaemon.new( array, str, array)
+  #
+
+  #
+  # RhaDaemon controls the heartbeat daemon
+  #
+  class RhaDaemon
+
+    #
+    # Increment debugging level. Higher levels are more verbose.
+    #
+    attr_accessor :debug_level
+
+    #
+    # Alternate path to heartbeat. If this is not set, environment path
+    # is used.
+    #
+    attr_accessor :path_to_heartbeat
+
+    #
+    # Set heartbeat daemon option; This will set an arbitrary
+    # heartbeat option.
+    #
+    attr_accessor :option
+
+    #
+    # Returns a new RhaDaemon Object
+    #
+    def initialize()
+    end
+
+    #
+    # Reload heartbeat. This option is functionally identical to
+    # sending a running heartbeat process a HUP signal. If the
+    # configuration has not changed, then this option is essentially
+    # a no-op. If ha.cf  or authkeys has changed, then heartbeat
+    # will re-read these files and update its configuration.
+    #
+    def reload
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-r " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+
+    #
+    # Kill (stop) heartbeat.
+    #
+    def reload
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-k " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+
+    #
+    # Report heartbeat status.
+    #
+    def report
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-k " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+    
+    #
+    # Heartbeat restart exec flag (internal use only).
+    #
+    def restart
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-R " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+
+    #
+    # Report heartbeat status.
+    #
+    def status
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-s " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+
+    #
+    # Heartbeat current resource state for restart (internal use only).
+    #
+    def restart_with_current_resources
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-R -C " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+    
+    #
+    # Print out heartbeat version.
+    #
+    def version
+      tmp = Tempfile.new('tmp')
+      command = option_string() + "-V " + " 2> " + tmp.path
+      success = system(command)
+      if success
+        begin
+          while (line = tmp.readline)
+            line.chomp
+            selected_string = line
+          end
+        rescue EOFError
+          tmp.close
+        end
+        return selected_string
+      else
+        tmp.close!
+        return success
+      end
+    end
+
+    private
+
+    def option_string()
+
+      unless @path_to_heartbeat
+        ostring = "heartbeat "
+      else
+        ostring = @path_to_heartbeat + " "
+      end
+
+      if @option
+        ostring += @option + " "
+      end
+
+      if @debug_level
+        ostring += "-d " + @debug_level
+      end
+
+      return ostring
+
+    end
+  end
+