kanamei-keystone 0.0.23 → 0.0.24

data/lib/keystone.rb

@@ -11,7 +11,7 @@ autoload :Moji          , 'moji'
 
 module Keystone
 
-  VERSION = '0.0.23'
+  VERSION = '0.0.24'
 
   autoload :StringUtil , 'keystone/string_util'
   autoload :Batch      , 'keystone/batch'

data/lib/keystone/batch/base.rb

@@ -134,7 +134,7 @@ module Keystone::Batch
           debug pid_file
           if File.exists?(pid_file)
             pid = File.open(pid_file).read.chomp
-            pid_list = `ps -e | awk '{print $1}'`
+            pid_list = `ps -ax | awk '{print $1}'`
             if (pid != nil && pid != "" ) && pid_list =~ /#{pid}/
               warn "pid:#{pid} still running"
               double_process_check_worked = true

data/lib/keystone/core_ext/ipaddr.rb

@@ -0,0 +1,19 @@
+
+require 'ipaddr'
+
+class IPAddr
+  def to_a
+    begin_addr = (@addr & @mask_addr)
+    case @family
+    when Socket::AF_INET
+      end_addr = (@addr | (IN4MASK ^ @mask_addr))
+    when Socket::AF_INET6
+      end_addr = (@addr | (IN6MASK ^ @mask_addr))
+    else
+      raise "unsupported address family"
+    end
+    ret = []
+    begin_addr.upto(end_addr){|addr| ret << IPAddr.new(addr,Socket::AF_INET)}
+    return ret
+  end
+end

data/util/bin/command_broadcast.rb

@@ -0,0 +1,135 @@
+
+# TODO 
+# みしあげでおま
+
+require 'rubygems'
+require 'keystone'
+require 'ipaddr'
+require 'net/telnet'
+require 'net/ssh'
+require 'pp'
+
+open("/tmp/command_broadcast.log.#{Time.now.usec}","w") do |f|
+  f.puts "test"
+end
+
+INPUT_MESSAGE_CMD = "input command you broadcast:"
+INPUT_MESSAGE_IP = %|input ipaddress you broadcast(ex"192.168.1.13","192.168.1.0/24"):|
+
+# TODO lvm 絡みも入れるべきかも
+# ホワイトリストにすべき？
+CANCEL_CMDS = %w|rm mv cp iptables fdisk dd| 
+
+def ppp(message)
+  print message
+  STDOUT.flush
+end
+
+ppp INPUT_MESSAGE_IP
+ipaddr = nil
+STDIN.each_line do |line|
+  line.strip!
+  begin
+    ipaddr = IPAddr.new line
+    break
+  rescue
+    ppp INPUT_MESSAGE_IP
+    next
+  end
+end
+
+ppp INPUT_MESSAGE_CMD
+cmd = nil
+STDIN.each_line do |line|
+  line.strip!
+  if line.blank?
+    ppp INPUT_MESSAGE_CMD
+    next
+  end
+  cmd = line
+  break
+end
+
+CANCEL_CMDS.each do |cancel_cmd|
+  if cmd =~ /\A#{cancel_cmd}/
+    puts "we can not run this command!!"
+    exit
+  end
+end
+
+puts "!! command you want to execute is below !!"
+puts "#{cmd}"
+ppp "?? are you sure to execute this??(y/N):"
+STDIN.each_line do |line|
+  line.chomp!
+  break if line.downcase == 'y'
+  puts "exit"
+  exit
+end
+
+alive_servers = []
+ipaddr.to_a.each do |ip|
+  ping = "ping -c 1 -t 2 #{ip.to_s}"
+  ret = `#{ping}`
+  if ret =~ /100% packet loss/
+    puts "#{ip.to_s} does not exits"
+  else
+    # OK な場合
+    puts "#{ip.to_s} exits"
+    alive_servers << ip
+  end
+end
+
+results = []
+
+alive_servers.each do |server|
+  ret = ""
+  begin
+    ip = server.to_s
+    pp ip
+    puts "++ telnet ++"
+    telnet = Net::Telnet.new("Host" => ip) {|c|} #print c}
+    telnet.login("webadmin", "inu00neko") {|c|} #print c}
+    telnet.cmd(cmd) {|c| ret = c}
+    STDOUT.flush # <- これがないとここまで処理が来てることがわかりにくい
+    # ログインセッションの終了
+    telnet.cmd("exit") {|c|} #print c}
+    telnet.close
+    telnet_ok = true
+#  rescue TimeoutError => e
+  rescue TimeoutError => e
+    puts e.message
+  rescue Errno::ECONNREFUSED => e
+    puts e.message
+  rescue Errno::ENETUNREACH => e
+    puts e.message
+  rescue Errno::EACCES => e
+    puts e.message
+  end
+
+  unless telnet_ok
+    begin
+      puts "++ ssh ++"
+      Net::SSH.start(ip, 'webadmin', :password => 'inu00neko') do |ssh|
+        ret = (ssh.exec!(cmd))
+      end    
+    rescue Exception => e
+    end
+  end
+  
+  results << [ip,ret]
+  puts ret
+  sleep 1
+end
+
+pp results
+
+open("/tmp/command_broadcast.log.#{Time.now.usec}","w") do |f|
+  f.puts results.inspect
+end
+
+__END__
+find /etc/lognotify/ -name "lognotify*.conf" -type f -exec cat {} \;
+
+
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: kanamei-keystone
 version: !ruby/object:Gem::Version 
-  version: 0.0.23
+  version: 0.0.24
 platform: ruby
 authors: 
 - kanamei
@@ -23,6 +23,7 @@ extra_rdoc_files: []
 
 files: 
 - util/bin
+- util/bin/command_broadcast.rb
 - util/bin/disk_size_check.rb
 - util/bin/notice
 - util/README
@@ -40,6 +41,7 @@ files:
 - lib/keystone/core_ext/array.rb
 - lib/keystone/core_ext/blank.rb
 - lib/keystone/core_ext/dir.rb
+- lib/keystone/core_ext/ipaddr.rb
 - lib/keystone/core_ext/object.rb
 - lib/keystone/core_ext/tempfile.rb
 - lib/keystone/core_ext/uri.rb