myaxes 0.1.0

data/bin/myaxes

@@ -0,0 +1,38 @@
+#!/usr/bin/env ruby
+
+require "myaxes"
+
+AXES_VER = "0.1.0 alpha"
+AXES_AUTHORS = "InViZz"
+
+################################################################################
+puts "------------------------------------------------------------------------------------ "
+puts "                          @             @                                            "
+puts "                         @@             @@                                           "
+puts "                       @@@@             @@@@                                         "
+puts "                      @@@@@             @@@@@                                        "
+puts "                     @@@@@@     @@@     @@@@@@                                       "
+puts "                    @@@@@@@     @@@     @@@@@@@                                      "
+puts "                    @@@@@@@@@@@@@@@@@@@@@@@@@@@                                      "
+puts "                    @@@@@@@@@@@@@@@@@@@@@@@@@@@                                      "
+puts "                    @@@@@@@     @@@     @@@@@@@                                      "
+puts "                     @@@@@@     @@@     @@@@@@                                       "
+puts "@@@@@@@@@@   @@@ @@@  @@@@@     @@@     @@@@@  @@@@@@   @@@  @@@  @@@@@@@@   @@@@@@  "
+puts "@@@@@@@@@@@  @@@ @@@   @@@@     @@@     @@@@  @@@@@@@@  @@@  @@@  @@@@@@@@  @@@@@@@  "
+puts "@@! @@! @@!  @@! !@@     @@     @@@     @@    @@!  @@@  @@!  !@@  @@!       !@@      "
+puts "!@! !@! !@!  !@! @!!      @     @@@     @     !@!  @!@  !@!  @!!  !@!       !@!      "
+puts "@!! !!@ @!@   !@!@!             @@@           @!@!@!@!   !@@!@!   @!!!:!    !!@@!!   "
+puts "!@!   ! !@!    @!!!             @@@           !!!@!!!!    @!!!    !!!!!:     !!@!!!  "
+puts "!!:     !!:    !!:              @@@           !!:  !!!   !: :!!   !!:            !:! "
+puts ":!:     :!:    :!:              @@@           :!:  !:!  :!:  !:!  :!:           !:!  "
+puts ":::     ::      ::              @@@           ::   :::   ::  :::   :: ::::  :::: ::  "
+puts " :      :       :               @@@            :   : :   :   ::   : :: ::   :: : :   "
+puts "                               @@@@@                                                 "
+puts "#{AXES_VER} by #{AXES_AUTHORS}          @@@@@                                                 "
+puts "                               @@@@@                                                 "
+puts "------------------------------------------------------------------------------------"
+################################################################################
+$DEBUG = false
+
+myaxes = MyAxes.new('/etc/myaxes.yml')
+myaxes.start

data/lib/myaxes.rb

@@ -0,0 +1,145 @@
+require "rubygems"
+require "net/ssh"
+require "net/ssh/gateway"
+require "log4r"
+require 'myaxes/axeconfig'
+require 'myaxes/options'
+
+class MyAxes
+
+  include Log4r
+
+  def initialize(config='~/.myaxes.yml')
+    @logger = Logger.new('MyAxes')
+    @logger.outputters = Outputter.stdout
+
+    @options = Options.new.parse
+    @logger.debug "test #{@options}" if $DEBUG
+
+    @conf = AxeConfig.new(@options)
+    @config = @conf.read(config)
+    @targets = @conf.targets
+    @threads = []
+
+    @ssh_options ={
+      :port => @config['Global']['ssh_port'],
+    	:verbose => @config['Global']['debug_level'].to_sym,
+    	:auth_methods => %w(publickey password keyboard-interactive),
+    	:keys => @config['Global']['ssh_keys'],
+    	:password => @config['Global']['password']
+    }
+    @commands_proc = Proc.new { |session, hostname|
+      @targets[hostname].each do |query|
+        @logger.debug "Query: #{query}" if $DEBUG
+        name = hostname.chomp.split(".")[0]
+        cmd = "mysql -u #{@config['Targets'][name]['login']} -e '#{query}' -p"
+        output = self.exec(session,cmd,name)
+        puts "\033[0;32m[*] #{hostname}\033[0m: #{output}"
+      end
+    }
+
+  end
+
+  def start
+    @targets.each_key do |hostname|
+      if self.use_gw?
+        @threads << Thread.new {
+          self.via_gw do |jump_server|
+            begin
+              jump_server.ssh(hostname, @config['Global']['login'], @ssh_options) do |session|
+                @commands_proc.call(session,hostname)
+              end
+            rescue Net::SSH::Disconnect => errmsg
+              warn "#{hostname} : #{errmsg}"
+            rescue Net::SSH::AuthenticationFailed => errmsg
+              warn "#{hostname} : #{errmsg}"
+            rescue Errno::ETIMEDOUT => errmsg
+              warn "#{hostname} : #{errmsg}"
+            rescue Errno::ECONNREFUSED => errmsg
+              warn "#{hostname} : #{errmsg}"
+            end
+          end
+        }
+      else
+        @threads << Thread.new {
+          begin
+            Net::SSH.start(hostname, @config['Global']['login'], @ssh_options) do |session|
+              @commands_proc.call(session,hostname)
+       		  end
+       		rescue Net::SSH::Disconnect => errmsg
+            warn "#{hostname} : #{errmsg}"
+          rescue Net::SSH::AuthenticationFailed => errmsg
+            warn "#{hostname} : #{errmsg}"
+          rescue Errno::ETIMEDOUT => errmsg
+            warn "#{hostname} : #{errmsg}"
+          rescue Errno::ECONNREFUSED => errmsg
+            warn "#{hostname} : #{errmsg}"
+          end
+       	}
+      end
+    end
+
+    @threads.each {|thread|
+      thread.join
+    }
+  end
+
+  def via_gw
+    begin
+      jump_server = Net::SSH::Gateway.new(@config['Global']['jump_server'], @config['Global']['login'], @ssh_options)
+
+	    @logger.debug "port forwarding ok" if $DEBUG
+
+	    yield jump_server
+
+	  rescue Net::SSH::Disconnect => errmsg
+    	warn "Gateway : #{errmsg}"
+    rescue Net::SSH::AuthenticationFailed => errmsg
+    	warn "Gateway : #{errmsg}"
+    rescue Errno::ETIMEDOUT => errmsg
+    	warn "Gateway : #{errmsg}"
+    rescue Errno::ECONNREFUSED => errmsg
+    	warn "Gateway : #{errmsg}"
+    end
+
+  end
+
+  def use_gw?
+    @config['Global']['use_jump']
+  end
+
+  def exec(session,cmd,name)
+    channel = session.open_channel do |channel|
+      channel.request_pty do |ch, success|
+        raise "Could not obtain pty (i.e. an interactive ssh session)" if !success
+      end
+      channel.exec(cmd) do |ch, success|
+        die "could not execute command" unless success
+          channel.on_data do |ch, data|
+            if data == "Enter password: "
+              @logger.debug "DEBUG: Password request" if $DEBUG
+              channel.send_data "#{@config['Targets'][name]['password']}\n"
+            else
+              channel[:result] ||= ""
+              channel[:result] << data
+            end
+          end
+
+          channel.on_extended_data do |ch, type, data|
+            raise "SSH command returned on stderr: #{data}"
+          end
+        end
+      end
+
+          # Nothing has actually happened yet. Everything above will respond to the
+          # server after each execution of the ssh loop until it has nothing left
+          # to process. For example, if the above recieved a password challenge from
+          # the server, ssh's exec loop would execute twice - once for the password,
+          # then again after clearing the password (or twice more and exit if the
+          # password was bad)
+      channel.wait
+
+      return channel[:result] # it returns with \r\n at the end
+  end
+
+end

data/lib/myaxes/axeconfig.rb

@@ -0,0 +1,40 @@
+require "yaml"
+
+class AxeConfig < Hash
+
+  attr_reader :targets, :config
+
+  def initialize(options)
+    @config = Hash.new
+    @options = options
+    @targets = Hash.new
+  end
+
+  def read(config)
+    user = @options['user']
+    ip = @options['ip']
+    #password = @options['password']
+    group = @options['group']
+
+    puts "Configfile (#{config}) not found" unless File.readable?(config)
+
+    begin
+      @config = YAML.load_file(config)
+    rescue Exception => errmsg
+      puts "Configfile format error: #{errmsg}"
+    end
+
+    @config['Targets'].each_key do |target|
+      if @config['Targets'][target]['query'].include? group
+        @config['Targets'][target]['query'][group].each do |query|
+          query.to_s.gsub!('{ip}', ip).gsub!('{user}', user)
+        end
+        @targets["#{@config['Targets'][target]['hostname']}"] = @config['Targets'][target]['query'][group]
+      end
+    end
+
+    self.config
+  end
+
+
+end

data/lib/myaxes/options.rb

@@ -0,0 +1,40 @@
+require "optparse"
+
+class Options
+
+  attr_reader :options
+
+  def initialize
+    @options = Hash.new
+
+    @options['mode'] = 'log'
+    @options['log'] = $stdout
+
+    end
+
+  def parse
+    OptionParser.new do |opt|
+      opt.separator ""
+      opt.separator "Usage: myaxes -g <group> -a <ip> -u <username> -p <password>[-H <hostnames>]"
+      opt.separator ""
+      opt.separator "shell> myaxes -g web -a 192.168.0.101 -u web301 -p qwerty"
+      opt.separator ""
+      opt.separator ""
+      opt.separator "Options:"
+      opt.separator "-------------"
+      opt.on('-g', '--group GROUPNAME', 'Group from config') { |val| @options['group'] = val }
+      opt.on('-a', '--address IP-ADDRESS', 'Ip address') { |val| @options['ip'] = val }
+      opt.on('-u', '--user USERNAME', 'Username') { |val| @options['user'] = val }
+      opt.on('-p', '--password PASSWORD', 'Password') { |val| @options['password'] = val }
+      opt.on('-H', '--hosts HOSTNAME_X,HOSTNAME_Y,HOSTNAME_Z', Array, 'Other Hosts(not from config)')  { |val| @options['hosts'] = val }
+      opt.separator ""
+      opt.on_tail('-h', '--help')           { puts opt; exit }
+
+      opt.parse!
+    end
+
+    self.options
+
+  end
+
+end

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: myaxes
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- InViZz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-08 00:00:00.000000000 Z
+dependencies: []
+description: Simple console app for remote execute mysql queries
+email: morion.estariol@gmail.com
+executables:
+- myaxes
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/myaxes.rb
+- lib/myaxes/axeconfig.rb
+- lib/myaxes/options.rb
+- bin/myaxes
+homepage: https://github.com/InViZz/MyAxes
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: MyAxes
+test_files: []