luna_scanner 0.0.3

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    ZDVhMmFmOGU4NzgyY2I3NTIzMzNjNWYzOWJiNDFhNTVkMDM2ZDFiNg==
+  data.tar.gz: !binary |-
+    NTE5YzNhOGJjNjY1ZDM4NzdmMzJiODNhMTFlZTY0ZDlhYTk0N2RmYg==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    OTY4ZDNhZWUwYjgwYzRlZGQxMWQ5YTQzYjc4ZTJmYzZmM2FmYjBjMDhmYjlm
+    M2RlYmM2NWI1MWYwOWFiZDA3NzU5NWJmMjE4YjgwMmQ1MTBkMGQ1YzIyN2Fl
+    ZjVmNGU5NTFkMTUwYjBkY2UzNzYyMjU2MzVmZmQxMjlhN2VjOWY=
+  data.tar.gz: !binary |-
+    MmMwNTI5OWE1Nzc4YzFmY2QxOTBiNWYwYjMxMGVhOWMwNjE0YTg0ZmM2OTc3
+    ZmQ1YTIzMTNhZDIyNmIwZmI1NzUyNzhhZWU4ZDU1NDg4MDc0YzkyMjM2ZTYx
+    ZDkyZTcyYTJkYTg3YjViMmQ0YmI0YThhMDBkNmMzNDM1YzE2MTA=

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.idea

data/Changelog.txt

@@ -0,0 +1,10 @@
+## Change log
+
+v0.0.1
+    Scan current LAN devices.
+
+v0.0.2
+    Better scan result display.
+
+v0.0.3
+    Add [luna_scanner reboot] action.

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in luna_scanner.gemspec
+gemspec

data/INSTALL

@@ -0,0 +1,21 @@
+#!/bin/sh
+
+gem build luna_scanner.gemspec
+
+cd vendor/cache
+gem install --no-ri --no-rdoc  --local net-ssh-2.7.0.gem
+gem install --no-ri --no-rdoc  --local net-scp-1.1.2.gem
+gem install --no-ri --no-rdoc  --local rack-1.5.2.gem
+gem install --no-ri --no-rdoc  --local rack-protection-1.5.1.gem
+gem install --no-ri --no-rdoc  --local rake-10.1.0.gem
+gem install --no-ri --no-rdoc  --local sinatra-1.4.4.gem
+gem install --no-ri --no-rdoc  --local tilt-1.4.1.gem
+gem install --no-ri --no-rdoc  --local tilt-1.4.1.gem
+
+echo "Installed dependency."
+
+cd ../..
+gem install --no-ri --no-rdoc --local luna_scanner-0.0.2.gem
+rm -f luna_scanner-0.0.2.gem
+
+echo "\n  Installed success."

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 qichunren
+
+MIT License
+
+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.md

@@ -0,0 +1,99 @@
+# LunaScanner
+
+LunaScanner是一个局域网中设备扫描的命令行工具。它可以批量局域网中设备，批量执行远程命令。
+
+## 使用场景
+
+ 我们在做的一套系统中包括服务器和许多客户端机器，它们都是Linux机器。这些客户端机器的系统（固件）都是统一刷机部署的。日常管理客户端机器是通
+ 过SSH加上一个private key登录进入终端，执行相应的SHELL命令来完成操作的。在平时开发和测试的过程中，经常需要统一地批量修改当前网络中客户端
+ 系统中的某一个文件或者重启等各种操作，于是我就创建了这个工具。
+
+## 安装
+
+```
+$ gem install luna_scanner
+```
+
+## 使用方法
+
+`luna_scanner`
+直接输入这个命令是扫描当前网段中的客户端Linux机器，如当前机器的IP是192.168.1.23，那么扫描的网段是192.168.1.1到192.168.1.255
+
+参数:
+
+--ip_range 是用于设置扫描的网段，如--ip_range=192.168.0.1,192.168.4.255
+   
+-t  是用于设置扫描的并发线程数，如-t 180，这个数值默认是120，建议不要设置得过高
+   
+-r  是用于设置扫描结果的文件，默认是/tmp/lunascan.txt
+
+luna_scanner reboot
+
+luna_scanner --ip_range=192.168.0.1,192.168.4.255 -r 192_ip.txt
+
+luna_scanner change_ip -i 10_ip.txt
+
+luna_scanner upload --source_file update_firmware.sh --target_file /usr/local/luna-client/script/update_firmware.sh -i to_be_update_devices.txt -c 'chmod a+x /usr/local/luna-client/script/update_firmware.sh'
+
+luna_scanner upload --source_file update_firmware.sh --target_file /tmp/no_use.sh -i to_be_update_devices.txt -c '/usr/local/luna-client/script/update_firmware.sh http://192.168.3.100 900k'
+
+luna_scanner upload --source_file /Users/qichunren/Downloads/update_firmware.sh --target_file /usr/local/luna-client/script/update_firmware.sh -c 'chmod a+x /usr/local/luna-client/script/update_firmware.sh && /usr/local/luna-client/script/update_firmware.sh http://10.0.4.48 900k' -i 10_ip.txt
+
+###　批量更新终端设备
+
+1. 批量扫描当前网段的设置
+2. 修改扫描结果的文件，只留下需要更新的设备的文本行
+3. 远程更新命令批量执行
+
+
+    luna_scanner --ip_range=10.0.0.1,10.0.15.150 -t 150 -r 10_ip.txt
+
+<pre>
+------IP-----------SN---------MODEL-----VERSION------
+      10.0.6.27 00000389    T-7203D  20131112.0506
+     10.0.6.221 00000208    T-7203D  20131112.0506
+    10.0.10.227 00000032    T-7203D  20131209.2239
+     10.0.11.40 00000307    T-7203D  20131112.0506
+    10.0.13.220 00000262    T-7203D  20131112.0506
+
+5 devices found.
+Write scan result to /Users/qichunren/RubymineProjects/luna_scanner/10_ip.txt
+</pre>
+
+    vim 10_ip.txt # Remote device which ip is '10.0.10.227', it is latest version, so I remove this text line.
+    # Because new version update_firmware.sh file is not compatible with old clone, so before we update firmware, we should upload the new one to replace old one.
+    # Then invoke update command to upgrade firmware from remote luna-server.
+    luna_scanner upload --source_file /Users/qichunren/Downloads/update_firmware.sh --target_file /usr/local/luna-client/script/update_firmware.sh -c 'chmod a+x /usr/local/luna-client/script/update_firmware.sh && /usr/local/luna-client/script/update_firmware.sh http://10.0.4.48 1300k' -i 10_ip.txt
+
+    ## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+
+scp ~/Downloads/update_firmware.sh root@10.0.6.12:/usr/local/luna-client/script/update_firmware.sh
+scp ~/Downloads/update_firmware.sh root@10.0.7.156:/usr/local/luna-client/script/update_firmware.sh
+
+
+### Work Step
+
+1. Scan
+    luna_scanner --ip_range 192.168.0.1,192.168.3.254 -t 150
+    luna_scanner --ip_range 8.128.2.0,8.128.3.254 -t 150 -r /tmp/lunascan8.txt
+    luna_scanner --ip_range 10.0.0.0,10.0.15.254 -t 150 -r /tmp/lunascan8.txt
+
+    luna_scanner --ip_range 8.128.0.0,8.128.1.254 -t 150 -r /tmp/lunascan0.txt
+
+
+    cat /tmp/lunascan8.txt  | grep U > /tmp/lunascan.txt
+
+2. Update
+    luna_scanner update  # /tmp/lunascan.txt
+
+    luna_scanner change_ip --reboot
+
+
+

data/Rakefile

@@ -0,0 +1,14 @@
+require "bundler/gem_tasks"
+
+### Gem local install.
+#
+# bundle
+# bundle package --all
+# cd vendor/cache
+# gem install --no-ri --no-rdoc  --local net-ssh-2.7.0.gem
+# gem install --no-ri --no-rdoc  --local net-scp-1.1.2.gem
+# gem install --no-ri --no-rdoc  --local rack-1.5.2.gem
+# gem install --no-ri --no-rdoc  --local rack-protection-1.5.1.gem
+# gem install --no-ri --no-rdoc  --local rake-10.1.0.gem
+# gem install --no-ri --no-rdoc  --local sinatra-1.4.4.gem
+# gem install --no-ri --no-rdoc  --local tilt-1.4.1.gem

data/bin/luna_scanner

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+require 'luna_scanner'
+
+Signal.trap("INT") { exit 1 }
+
+LunaScanner::CLI.start

data/lib/luna_scanner.rb

@@ -0,0 +1,66 @@
+require 'socket'
+require 'net/ssh'
+require "luna_scanner/version"
+
+module LunaScanner
+
+  class SSHKeyNotFoundError < StandardError; end
+
+  class << self
+    def root
+      @root_path ||= File.expand_path("../", __FILE__)
+    end
+
+    def pwd
+      @pwd ||= Dir.pwd
+    end
+
+    def local_ip
+      @local_ip ||= begin
+        orig = Socket.do_not_reverse_lookup
+        Socket.do_not_reverse_lookup =true # turn off reverse DNS resolution temporarily
+        UDPSocket.open do |s|
+          s.connect '64.233.187.99', 1 #google
+          s.addr.last
+        end
+      rescue
+
+      ensure
+        Socket.do_not_reverse_lookup = orig
+      end
+    end
+
+    def ssh_key
+      @ssh_key ||= Dir.home.to_s + "/yu_pri"
+    end
+
+    def check_ssh_key!
+      raise SSHKeyNotFoundError.new("Please provide ssh key file: #{ssh_key}") if not File.exist?(ssh_key)
+
+      ssh_key
+    end
+
+    def start_ssh(ip, &block)
+      Net::SSH.start(
+          "#{ip}", 'root',
+          :auth_methods => ["publickey"],
+          :user_known_hosts_file => "/dev/null",
+          :timeout => 8,
+          :verbose => :error,
+          :keys => [ ssh_key ]  # Fix key permission: chmod g-wr ./yu_pri  chmod o-wr ./yu_pri  chmod u-w ./yu_pri
+      ) do |session|
+        block.call(session)
+      end
+    end
+
+
+  end
+end
+
+LunaScanner.autoload :Logger,   "luna_scanner/logger"
+LunaScanner.autoload :Device,   "luna_scanner/device"
+LunaScanner.autoload :Util,     "luna_scanner/util"
+LunaScanner.autoload :CLI,      "luna_scanner/cli"
+LunaScanner.autoload :Rcommand, "luna_scanner/rcommand"
+LunaScanner.autoload :Scanner,  "luna_scanner/scanner"
+LunaScanner.autoload :Web,      "luna_scanner/web"

data/lib/luna_scanner/cli.rb

@@ -0,0 +1,159 @@
+#encoding: utf-8
+require 'optparse'
+
+module LunaScanner
+  class CLI
+
+    def initialize()
+      @options = {
+          :thread_size => 120,
+          :reboot => false,
+          :result => "/tmp/lunascan.txt",
+          :start_ip => nil,
+          :end_ip => nil,
+          :source_file => nil,
+          :target_file => nil,
+          :input_ip => "/tmp/lunascan.txt",
+          :command => ""
+      }
+      option_parser = OptionParser.new do |opts|
+        opts.banner = 'Luna Scanner toolkit.'
+
+        opts.on('-t', '--thread_size COUNT', 'Set luna_scanner scan thread size.') do |thread_size|
+          @options[:thread_size] = thread_size.to_i
+        end
+
+        # TODO include extra ip, exclude extra ip from --ip_range option
+        opts.on('--ip_range START_IP,END_IP', 'Set luna_scanner scan ip range.') do |ip_param|
+          ip_range = ip_param.split(",")
+          @options[:start_ip] = ip_range[0]
+          @options[:end_ip] = ip_range[1]
+        end
+
+        opts.on('-i', '--input INPUT_IP', 'Scan from given INPUT_IP file.') do |file|
+          #TODO fixed path string, such as ~ or ./
+          if file && file.start_with?("/")
+            @options[:input_ip] = file
+          else
+            @options[:input_ip] = LunaScanner.pwd + "/" + file
+          end
+        end
+
+        opts.on('-r', '--result RESULT_FILE', 'Store scan result to file.') do |result_file|
+          #TODO fixed path string, such as ~ or ./
+          if result_file && result_file.start_with?("/")
+            @options[:result] = result_file
+          else
+            @options[:result] = LunaScanner.pwd + "/" + result_file
+          end
+        end
+
+        opts.on('--reboot', 'Reboot devices.') do
+          @options[:reboot] = true
+        end
+
+        opts.on('--source_file SOURCE_FILE', 'Source file to upload to remote. Only be used for [upload] action') do |file|
+          #TODO fixed path string, such as ~ or ./
+          if file && file.start_with?("/")
+            @options[:source_file] = file
+          else
+            @options[:source_file] = LunaScanner.pwd + "/" + file
+          end
+        end
+
+        opts.on('--target_file TARGET_FILE', 'File to upload to target place. Only be used for [upload] action') do |file|
+          if file && file.start_with?("/")
+            @options[:target_file] = file
+          else
+            puts "--target_file option value must start with / absolute path"
+            exit 1
+          end
+        end
+
+        opts.on('-c', '--command SHELL_COMMAND', 'Shell command execute on remote devices. Only be used for [upload] action') do |command_string|
+          @options[:command] = command_string
+        end
+
+        opts.on_tail("-h", "--help", "luna_scanner usage.") do
+          puts "Luna Scanner usage:"
+          puts "  luna_scanner [action] [option]"
+          puts "    luna_scanner                                                 -> Scan currnet LAN devices with default configuration"
+          puts "    luna_scanner -t 60                                           -> Set scan thread size to 60"
+          puts "    luna_scanner change_ip                                       -> Batch change LAN devices ip configuration"
+          puts "    luna_scanner upload --source_file file1 --target_file file2  -> Upload file1 to file2 on LAN devices"
+          puts "    luna_scanner web                                             -> Start luna_scanner web ui on 4567 port (To be done)"
+          puts "    luna_scanner -v                                              -> Display luna_scanner version"
+          puts "\n"
+          puts opts
+          exit 1
+        end
+
+        opts.on('-v', '--version', 'Display luna_scanner version.') do
+          puts "luna_scanner v#{LunaScanner::VERSION}"
+          exit 1
+        end
+
+      end.parse!
+
+    end
+
+    def execute
+      if ARGV[0].to_s == ''
+        LunaScanner.check_ssh_key!
+        LunaScanner::Scanner.scan!(@options)
+      elsif ARGV[0].to_s == 'reboot'
+        LunaScanner.check_ssh_key!
+        @options[:reboot] = true
+        LunaScanner::Scanner.scan!(@options)
+      elsif ARGV[0].to_s == 'upload'
+        LunaScanner.check_ssh_key!
+        source_file = @options.delete(:source_file)
+        target_file = @options.delete(:target_file)
+        if source_file.nil? || target_file.nil?
+          puts "--source_file SOURCE_FILE or --target_file TARGET_FILE options missing for upload action."
+          exit 1
+        end
+        if !File.exist?(source_file)
+          puts "Source file #{source_file} not exist."
+          exit 2
+        end
+        if @options[:input_ip] && !File.exist?(@options[:input_ip])
+          puts "Input ip file #{@options[:input_ip]} not exist."
+          exit 3
+        end
+
+        LunaScanner::Scanner.upload!(source_file, target_file, @options) do |shell|
+          shell.exec!(@options[:command]) if @options[:command].length > 0
+        end
+      elsif ARGV[0].to_s == 'web'
+        LunaScanner::Web.run!
+      elsif ARGV[0].to_s == 'update'
+        LunaScanner.check_ssh_key!
+
+        if @options[:input_ip].nil? || !File.exist?(@options[:input_ip])
+          puts "Input ip file #{@options[:input_ip]} not exist."
+          exit 3
+        end
+
+        LunaScanner::Rcommand.batch_update(@options)
+      elsif ARGV[0].to_s == 'change_ip'
+        LunaScanner.check_ssh_key!
+
+        if @options[:input_ip].nil? || !File.exist?(@options[:input_ip])
+          puts "Input ip file #{@options[:input_ip]} not exist."
+          exit 3
+        end
+
+        LunaScanner::Rcommand.batch_change_ip(@options)
+
+      else
+        puts "Invalid action / options"
+        exit 1
+      end
+    end
+
+    def self.start
+      self.new.execute
+    end
+  end
+end