moob 0.1.1 → 0.2.0

data/bin/moob

@@ -5,45 +5,95 @@ require 'moob'
 
 options = {
     :action => :jnlp,
-    :type => :idrac6,
-    :machines => []
+    :type => :auto,
+    :machines => [],
+    :password => ENV['password']
 }
 
 OptionParser.new do |opts|
-    opts.on '-l', '--list',
-    'List LOM types' do
-        Moob::TYPES.each do |k, v|
-            puts "#{k} (#{v.name})"
-        end
-        exit 0
-    end
     opts.on '-t', '--type t',
-    'LOM type (idrac6 by default)' do |t|
+    'LOM type, \'auto\' for autodetection, \'list\' to list',
+    'Defaults to auto' do |t|
+        if t == 'list'
+            Moob::TYPES.each do |sym, klass|
+                puts "#{sym} (#{klass.name})"
+            end
+            exit 0
+        else
             options[:type] = t.to_sym
+        end
+    end
+
+    opts.on '-a', '--action t',
+    'Action to perform, \'list\' to list',
+    'Defaults to jnlp' do |a|
+        if a == 'list'
+            Moob::TYPES.each do |sym, klass|
+                if klass.actions
+                    puts "#{sym}:"
+                    klass.actions.each do |name, descr|
+                        puts "  #{name}: #{descr}"
+                    end
+                else
+                    puts "#{sym} doesn't have any actions"
+                end
+            end
+            exit 0
+        else
+            options[:action] = a.to_sym
+        end
     end
+
     opts.on '-u', '--username u',
-    'LOM username' do |u|
-            options[:username] = u
+    'LOM username',
+    'Defaults to the model\'s default if known' do |u|
+        options[:username] = u
     end
+
     opts.on '-p', '--password p',
-    'LOM password' do |p|
-            options[:password] = p
+    'LOM password',
+    'Defaults to the model\'s default if known',
+    'Use the environment variable PASSWORD instead!' do |p|
+        options[:password] = p
     end
 
     opts.on '-m', '--machines a,b,c', Array,
-    'Comma-separated list of machines' do |m|
-            options[:machines] = m
+    'Comma-separated list of LOM hostnames' do |m|
+        options[:machines] = m
+    end
+
+    opts.on '-v', '--verbose' do
+        $VERBOSE = true
     end
 end.parse!
 
 if options[:machines].empty?
-    $stderr.puts 'No machines selected.'
+    $stderr.puts 'No LOMs selected.'
     exit 1
 end
 
-options[:machines].each do |h|
-    case options[:action]
-    when :jnlp
-        Moob.start_jnlp options[:type], h, options
+begin
+    options[:machines].each do |h|
+        lom = Moob.lom options[:type], h, options
+
+        puts "Trying to authenticate to #{h}..." if $VERBOSE
+        lom.authenticate
+        puts "Authenticated on #{h}." if $VERBOSE
+
+        case options[:action]
+        when :jnlp
+            Moob.start_jnlp lom
+        else
+            res = lom.send options[:action]
+            puts res if res
+        end
     end
+    puts 'There might be a delay before Java Web Start shows up...' if options[:action] == :jnlp
+
+rescue Exception => e
+    $stderr.puts "\033[31mFailure (#{e.class}): #{e}\033[0m\n\n" \
+        "Backtrace to provide in any support request:\033[2;34m"
+    $stderr.puts e.backtrace
+    $stderr.puts "\033[0m"
+    exit 1
 end

data/lib/moob.rb

@@ -1,9 +1,14 @@
-require 'socket'
-require 'cgi'
-require 'patron'
-
 module Moob
-    VERSION = [0,1,1]
+    VERSION = [0,2,0]
+
+    class ResponseError < Exception
+        def initialize response
+            @response = response
+        end
+        def to_s
+            "#{@response.url} failed (status #{@response.status})"
+        end
+    end
 
     autoload :BaseLom,    'moob/baselom.rb'
     autoload :Idrac6,     'moob/idrac6.rb'
@@ -16,33 +21,36 @@ module Moob
         :sun        => SunILom
     }
 
-    class ResponseError < Exception
-        def initialize response
-            @response = response
-        end
-        def to_s
-            "#{@response.url} failed (status #{@response.status})"
+    def self.lom type, hostname, options = {}
+        case type
+        when :auto
+            TYPES.find do |sym, klass|
+                puts "Trying type #{sym}..." if $VERBOSE
+                lom = klass.new hostname, options
+                if lom.detect
+                    puts "Type #{sym} detected." if $VERBOSE
+                    return lom
+                end
+                false
+            end
+            raise RuntimeError.new "Couldn't detect a known LOM type"
+        else
+            return TYPES[type].new hostname, options
         end
     end
 
-    def self.start_jnlp type, hostname, options = {}
-
-        lom = TYPES[type].new hostname, options
-        jnlp = lom.authenticate.jnlp
+    def self.start_jnlp lom
+        jnlp = lom.jnlp
 
         unless jnlp[/<\/jnlp>/]
             raise RuntimeError.new "Invalid JNLP file (\"#{jnlp}\")"
         end
 
-        filepath = "/tmp/#{hostname}.jnlp"
+        filepath = "/tmp/#{lom.hostname}.jnlp"
         File.open filepath, 'w' do |f|
             f.write jnlp
         end
 
-        system "javaws #{filepath}"
-    end
-
-    def self.cmdline
-        start_jnlp :megatrends, 'brenda.lom.sto.spotify.net'
+        raise Exception.new "javaws failed" unless system "javaws #{filepath} &"
     end
 end

data/lib/moob/baselom.rb

@@ -1,5 +1,10 @@
+require 'socket'
+require 'cgi'
+require 'patron'
+require 'set'
+
 class Moob::BaseLom
-    @@name = 'Unknown'
+    @name = "Unknown"
 
     def initialize hostname, options = {}
         @hostname = hostname
@@ -13,22 +18,24 @@ class Moob::BaseLom
         @session.connect_timeout = 10_000
         @session.timeout = 10_000
         @session.insecure = true
-        @session.ignore_content_length = true
     end
 
-    def jnlp
-        raise new NoMethodError
+    attr_reader :hostname, :username
+
+    def detect
+        false
     end
 
     def self.name
-        @@name
+        @name
+    end
+
+    def self.actions
+        @actions
     end
 
-    protected
-    # Get rid of security checks.
-    # Might break some features,
-    # please let me know if that's the case!
-    def desecurize_jnlp jnlp
-        jnlp.sub /<security>.*<\/security>/m, ''
+    def self.action sym, descr
+        @actions ||= {}
+        @actions[sym] = descr
     end
-end
+end

data/lib/moob/idrac6.rb

@@ -1,10 +1,75 @@
-class Moob::Idrac6 < Moob::BaseLom
-    @@name = 'Dell iDrac 6'
+module Moob
+class Idrac6 < BaseLom
+    @name = 'Dell iDrac 6'
+
+    INFO_FIELDS = %w{
+        sysDesc
+        biosVer
+        svcTag
+        expSvcCode
+        hostName
+        osName
+        osVersion
+        sysRev
+        LCCfwVersion
+        recoveryAction
+        initCountdown
+        presentCountdown
+        datetime
+        fwVersion
+        fwUpdated
+        hwVersion
+        macAddr
+        v4Enabled
+        v4IPAddr
+        v4Gateway
+        v4NetMask
+        v4DHCPEnabled
+        v4DHCPServers
+        v4DNS1
+        v4DNS2
+        v6Enabled
+        v6Addr
+        v6Gateway
+        v6DHCPEnabled
+        v6LinkLocal
+        v6Prefix
+        v6SiteLocal
+        v6SiteLocal3
+        v6SiteLocal4
+        v6SiteLocal5
+        v6SiteLocal6
+        v6SiteLocal7
+        v6SiteLocal8
+        v6SiteLocal9
+        v6SiteLocal10
+        v6SiteLocal11
+        v6SiteLocal12
+        v6SiteLocal13
+        v6SiteLocal14
+        v6SiteLocal15
+        racName
+        v6DHCPServers
+        v6DNS1
+        v6DNS2
+        NicEtherMac1
+        NicEtherMac2
+        NicEtherMac3
+        NicEtherMac4
+        NiciSCSIMac1
+        NiciSCSIMac2
+        NiciSCSIMac3
+        NiciSCSIMac4
+        NicEtherVMac1
+        NicEtherVMac2
+        NicEtherVMac3
+        NicEtherVMac4
+    }
 
     def initialize hostname, options = {}
         super hostname, options
-         @username ||= 'root'
-         @password ||= 'calvin'
+        @username ||= 'root'
+        @password ||= 'calvin'
     end
 
     def authenticate
@@ -16,34 +81,96 @@ class Moob::Idrac6 < Moob::BaseLom
             "user=#{@username}&password=#{@password}"
         raise ResponseError.new auth unless auth.status == 200
 
-        result_match = auth.body.match /<authResult>([^<]+)<\/authResult>/
-        raise Exception.new 'Cannot find auth result' unless result_match
-        result_code = result_match[1]
-        unless result_code == "0"
-            raise Exception.new "Auth failed with: \"#{auth.body}\""
-        end
+        auth.body =~ /<authResult>([^<]+)<\/authResult>/
+        raise Exception.new 'Cannot find auth result' unless $&
+        raise Exception.new "Auth failed with: \"#{auth.body}\"" unless $1 == "0"
         return self
     end
 
+    def detect
+        begin
+            home = @session.get 'login.html'
+            home.body =~ /Integrated Dell Remote Access Controller 6/
+        rescue
+            false
+        end
+    end
+
+    action :jnlp, 'Remote control'
     def jnlp
         idx = @session.get 'index.html'
         raise ResponseError.new idx unless idx.status == 200
 
-        name_match = idx.body.match /var DnsName += +"([^"]+)"/
-        raise Exception.new 'Couldn\'t find the DNS name' unless name_match
-        dns_name = name_match[1]
+        idx.body =~ /var DnsName += +"([^"]+)"/
+        raise Exception.new "Couldn't find the DNS name" unless $&
+        dns_name = $1
 
-        sys_match = idx.body.match /var sysNameStr += +"([^"]+)"/
-        raise Exception.new 'Couldn\'t find the system name' unless name_match
-        sys_name = sys_match[1] # eg PowerEdge R610
+        idx.body =~ /var sysNameStr += +"([^"]+)"/
+        raise Exception.new "Couldn't find the system name" unless $&
+        sys_name = $1 # eg PowerEdge R610
 
-        # eg escaped "idrac-D4MHZ4J, PowerEdge R610, User:root"
+        # eg escaped "idrac-A1BCD2E, PowerEdge R610, User:root"
         title = CGI::escape "#{dns_name}, #{sys_name}, User:#{@username}"
-        path = "viewer.jnlp(#{@hostname}@0@#{title}@#{Time.now.to_i * 1000})"
 
-        viewer = @session.get path
+        viewer = @session.get "viewer.jnlp(#{@hostname}@0@#{title}@#{Time.now.to_i * 1000})"
         raise ResponseError.new viewer unless viewer.status == 200
 
-        return desecurize_jnlp viewer.body
+        return viewer.body
+    end
+
+    def power_action action
+        req = @session.post "data?set=pwState:#{action}", {}
+        raise ResponseError.new req unless req.status == 200
+        raise Exception.new 'The answer looks wrong' unless req.body =~ /<status>ok<\/status>/
+        return nil
+    end
+
+    action :poff,    'Power Off System'
+    action :pon,     'Power On System'
+    action :pcycle,  'Power Cycle System (cold boot)'
+    action :reboot,  'Reset System (warm boot)'
+    action :nmi,     'NMI (Non-Masking Interrupt)'
+    action :shudown, 'Graceful Shutdown'
+
+    def poff;     power_action 0; end
+    def pon;      power_action 1; end
+    def pcycle;   power_action 2; end
+    def reboot;   power_action 3; end
+    def nmi;      power_action 4; end
+    def shutdown; power_action 5; end
+
+    def get_infos keys
+        infos = @session.post "data?get=#{keys.join(',')}", {}
+
+        raise ResponseError.new infos unless infos.status == 200
+        raise Exception.new "The status isn't OK" unless infos.body =~ /<status>ok<\/status>/
+
+        return Hash[keys.collect do |k|
+            if infos.body =~ /<#{k}>(.*?)<\/#{k}>/
+                [k, $1]
+            else
+                [k, nil]
+            end
+        end]
+    end
+
+    action :pstatus, 'Power status'
+    def pstatus
+        case get_infos(['pwState'])['pwState']
+        when '0'
+            return :off
+        when '1'
+            return :on
+        else
+            return nil
+        end
+    end
+
+    action :infos, 'Get system information'
+    def infos
+        return get_infos(INFO_FIELDS).collect do |k,v|
+            "#{k}: #{v}"
+        end.join "\n"
     end
-end
+end
+end

data/lib/moob/megatrends.rb

@@ -1,12 +1,13 @@
-class Moob::Megatrends < Moob::BaseLom
-    @@name = 'American Megatrends'
+module Moob
+class Megatrends < BaseLom
+    @name = 'American Megatrends'
 
     def initialize hostname, options = {}
         super hostname, options
         @username ||= 'root'
         @password ||= 'superuser'
         begin
-            @ip = Socket.getaddrinfo(hostname, nil)[0][2]
+            @ip = Socket.getaddrinfo(hostname, nil)[0][3]
         rescue
             raise Exception.new "Couldn't resolve \"#{hostname}\""
         end
@@ -19,19 +20,65 @@ class Moob::Megatrends < Moob::BaseLom
 
         raise ResponseError.new auth unless auth.status == 200
 
-        cookie_match = auth.body.match /'SESSION_COOKIE' *: *'([^']+)'/
-        unless cookie_match
-            raise Exception.new "Couldn't find auth cookie in \"#{auth.body}\""
-        end
+        auth.body =~ /'SESSION_COOKIE' *: *'([^']+)'/
+        raise Exception.new "Couldn't find auth cookie in \"#{auth.body}\"" unless $&
 
-        @cookie = "test=1; path=/; SessionCookie=#{cookie_match[1]}"
+        @cookie = "test=1; path=/; SessionCookie=#{$1}"
         return self
     end
 
+    def detect
+        begin
+            home = @session.get 'page/login.html'
+            home.body =~ /\.\.\/res\/banner_right.png/
+        rescue
+            false
+        end
+    end
+
+    action :jnlp, 'Remote control'
     def jnlp
+        @session.ignore_content_length = true
         viewer = @session.get 'Java/jviewer.jnlp', { 'Cookie' => @cookie }
         raise ResponseError.new viewer unless viewer.status == 200
 
-        return desecurize_jnlp viewer.body
+        return viewer.body
+    end
+
+    def power_action action
+        req = @session.post 'rpc/hostctl.asp',
+            { 'WEBVAR_POWER_CMD' => action },
+            { 'Cookie' => @cookie }
+        raise ResponseError.new req unless req.status == 200
+        unless req.body =~ /WEBVAR_STRUCTNAME_HL_POWERSTATUS/
+            raise Exception.new 'The answer looks wrong'
+        end
+        return nil
     end
+
+    action :poff,      'Power Off'
+    action :pon,       'Power On'
+    action :pcycle,    'Power Cycle'
+    action :preset,    'Power Reset'
+    action :soft_poff, 'Soft Power Off'
+    def poff;      power_action 0; end
+    def pon;       power_action 1; end
+    def pcycle;    power_action 2; end
+    def preset;    power_action 3; end
+    def soft_poff; power_action 5; end
+
+    action :pstatus, 'Power status'
+    def pstatus
+        status = @session.get 'rpc/hoststatus.asp',
+            { 'Cookie' => @cookie }
+        raise ResponseError.new status unless status.status == 200
+        raise Exception.new 'Couldn\'t read the state' unless status.body =~ /'JF_STATE' : (.),/
+        case $1
+        when '0'
+            return :off
+        when '1'
+            return :on
+        end
+    end
+end
 end

data/lib/moob/sunilom.rb

@@ -1,5 +1,6 @@
-class Moob::SunILom < Moob::BaseLom
-    @@name = 'Sun Integrated Lights Out Manager'
+module Moob
+class SunILom < BaseLom
+    @name = 'Sun Integrated Lights Out Manager'
 
     def initialize hostname, options = {}
         super hostname, options
@@ -15,19 +16,38 @@ class Moob::SunILom < Moob::BaseLom
 
         raise ResponseError.new auth unless auth.status == 200
 
-        cookie_match = auth.body.match /SetWebSessionString\("([^"]+)","([^"]+)"\);/
-        unless cookie_match
-            raise Exception.new "Couldn't find session cookie in \"#{auth.body}\""
+        if auth.body =~ /\/iPages\/i_login.asp\?msg=([^"])+"/
+            error = "code #{$1}"
+            error_page = @session.get "iPages/i_login.asp?msg=#{$1}"
+            if error_page.body =~ /<div class="AlrtErrTxt">(.*?)<\/div>/
+                error = "\"#{$1.gsub /<[^>]+>/, ''}\""
+            end
+
+            raise Exception.new "Auth failed (#{error})"
         end
 
-        @cookie = "#{cookie_match[1]}=#{cookie_match[2]}; langsetting=EN"
+        auth.body =~ /SetWebSessionString\("([^"]+)","([^"]+)"\);/
+        raise Exception.new "Couldn't find session cookie in \"#{auth.body}\"" unless $&
+
+        @cookie = "#{$1}=#{$2}; langsetting=EN"
         return self
     end
 
+    action :jnlp, 'Remote control'
     def jnlp
         viewer = @session.get 'cgi-bin/jnlpgenerator-8', { 'Cookie' => @cookie }
         raise ResponseError.new viewer unless viewer.status == 200
 
         return viewer.body
     end
+
+    def detect
+        begin
+            home = @session.get 'iPages/i_login.asp'
+            home.body =~ /Sun\(TM\) Sun Integrated Lights Out Manager/
+        rescue
+            false
+        end
+    end
+end
 end

metadata

@@ -1,12 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: moob
 version: !ruby/object:Gem::Version 
-  prerelease: false
+  hash: 23
+  prerelease: 
   segments: 
   - 0
-  - 1
-  - 1
-  version: 0.1.1
+  - 2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Pierre Carrier
@@ -14,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-07-25 00:00:00 +02:00
+date: 2011-07-30 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -25,11 +26,12 @@ dependencies:
     requirements: 
     - - ~>
       - !ruby/object:Gem::Version 
+        hash: 19
         segments: 
         - 0
         - 4
-        - 13
-        version: 0.4.13
+        - 14
+        version: 0.4.14
   type: :runtime
   version_requirements: *id001
 description: Control systems using Web-based out-of-band managers without a browser
@@ -51,8 +53,8 @@ files:
 - COPYING
 has_rdoc: true
 homepage: http://github.com/pcarrier/moob
-licenses: []
-
+licenses: 
+- Public-domain-like
 post_install_message: 
 rdoc_options: []
 
@@ -63,6 +65,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      hash: 3
       segments: 
       - 0
       version: "0"
@@ -71,6 +74,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      hash: 31
       segments: 
       - 1
       - 2
@@ -79,7 +83,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: moob
-rubygems_version: 1.3.7
+rubygems_version: 1.6.2
 signing_key: 
 specification_version: 3
 summary: Manage Out-Of-Band!