luna_scanner 0.0.3

data/lib/luna_scanner/update_firmware.sh

@@ -0,0 +1,273 @@
+#!/bin/bash
+
+NOW=`date '+%Y-%m-%d %H:%M'`
+
+echo "Start at ${NOW}"
+
+UPDATE_HOST=http://10.0.4.48
+LIMIT_SPEED=900k
+
+if [ $# -ge 1 ]; then
+  echo "Use update host $1"
+  UPDATE_HOST=$1
+else
+  echo "Use default update host ${UPDATE_HOST}"
+fi
+
+if [ $# -ge 2 ]; then
+  LIMIT_SPEED=$2
+  echo "Use download speed ${LIMIT_SPEED}"
+else
+  echo "Use default download speed ${LIMIT_SPEED}"
+fi
+
+if [ ! -d /root ]; then
+  echo "/root directory not exist! Exit"
+  exit 1
+fi
+
+if [ ! -f /proc/itc_sn/sn ]; then
+  echo "/proc/itc_sn/sn file not exist! Exit"
+  exit 1
+fi
+
+if [ ! -f /boot/uboot/uEnv.txt ]; then
+  echo "/boot/uboot/uEnv.txt file not exist! Exit"
+  exit 1
+fi
+
+which wget
+if [ ! $? -eq 0 ]; then
+    echo "wget tool not installed. Exit"
+    exit 1
+fi
+
+which md5sum
+if [ ! $? -eq 0 ]; then
+    echo "md5sum tool not installed. Exit"
+    exit 1
+fi
+
+which awk
+if [ ! $? -eq 0 ]; then
+    echo "awk tool not installed. Exit"
+    exit 1
+fi
+
+SN=`cat /proc/itc_sn/sn`
+echo "Serial NO. ${SN}"
+
+
+echo "Start download firmware.md5 file."
+# tell server I have started.
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=1 &> /dev/null
+sleep 1
+wget -O /root/firmware.md5 ${UPDATE_HOST}/static/firmwares/firmware.md5 --limit-rate=${LIMIT_SPEED}
+ret=$?
+if [ ! $ret -eq 0 ]; then
+    echo "firmware.md5 file downlaod failed! ${ret}"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=401\&reason=firmware.md5_download_failed &> /dev/null
+    exit 1
+fi
+
+echo "Start download package 1 -> vardir ..."
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=2 &> /dev/null
+sleep 1
+wget -O /root/vardir.tar.xz ${UPDATE_HOST}/static/firmwares/vardir.tar.xz --limit-rate=${LIMIT_SPEED}
+ret=$?
+if [ ! $ret -eq 0 ]; then
+    echo "Package vardir.tar.xz download failed! ${ret}"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=402\&reason=vardir.tar.xz_download_failed &> /dev/null
+    exit 1
+fi
+
+echo "Start download package 2 -> rootfs.tar.xz  ..."
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=3 &> /dev/null
+sleep 1
+wget -O /root/rootfs.tar.xz ${UPDATE_HOST}/static/firmwares/rootfs.tar.xz --limit-rate=${LIMIT_SPEED}
+ret=$?
+if [ ! $ret -eq 0 ]; then
+    echo "Package rootfs.tar.xz download failed! ${ret}"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=403\&reason=rootfs.tar.xz_download_failed &> /dev/null
+    exit 1
+fi
+
+echo "Start download uImage ..."
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=4 &> /dev/null
+sleep 1
+wget -O /root/uImage ${UPDATE_HOST}/static/firmwares/uImage --limit-rate=${LIMIT_SPEED}
+ret=$?
+if [ ! $ret -eq 0 ]; then
+    echo "Package uImage download failed! ${ret}"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=404\&reason=uImage_download_failed &> /dev/null
+    exit 1
+fi
+
+echo "Finished download. Start validate firmware."
+
+if [ ! -f /root/firmware.md5 ]; then
+  echo "/root/firmware.md5 not exist. Exit"
+  wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=401\&reason=firmware.md5_not_exist &> /dev/null
+  exit 1
+fi
+
+if [ ! -f /root/vardir.tar.xz ]; then
+  echo "/root/vardir.tar.xz not exist. Exit"
+  wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=402\&reason=vardir.tar.xz_not_exist &> /dev/null
+  exit 1
+fi
+
+if [ ! -f /root/rootfs.tar.xz ]; then
+  echo "/root/rootfs.tar.xz not exist. Exit"
+  wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=403\&reason=rootfs.tar.xz_not_exist &> /dev/null
+  exit 1
+fi
+
+if [ ! -f /root/uImage ]; then
+  echo "/root/uImage not exist. Exit"
+  wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=404\&reason=uImage_not_exist &> /dev/null
+  exit 1
+fi
+
+cd /root
+
+Vardir_md5=`md5sum vardir.tar.xz | awk -F' ' '{print $1}'`
+Vardir_md5_download=`grep vardir firmware.md5 | awk -F' ' '{print $1}'`
+
+if test "${Vardir_md5}" != "${Vardir_md5_download}" ; then
+    echo "vardir.tar.xz's md5 isn't matched"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=422\&reason=vardir.tar.xz_md5_not_matched &> /dev/null
+    exit 2
+fi
+
+Root_md5=`md5sum rootfs.tar.xz | awk -F' ' '{print $1}'`
+Root_md5_download=`grep rootfs firmware.md5 | awk -F' ' '{print $1}'`
+
+if test "${Root_md5}" != "${Root_md5_download}" ; then
+    echo "rootfs.tar.xz's md5 isn't matched"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=423\&reason=rootfs.tar.xz_md5_not_matched &> /dev/null
+    exit 2
+fi
+
+UImage_md5=`md5sum uImage | awk -F' ' '{print $1}'`
+UImage_md5_download=`grep uImage firmware.md5 | awk -F' ' '{print $1}'`
+
+if test "${UImage_md5}" != "${UImage_md5_download}" ; then
+    echo "uImage's md5 isn't matched"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=424\&reason=uImage_md5_not_matched &> /dev/null
+    exit 2
+fi
+
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=13 &> /dev/null
+
+echo "Validation passed. Start setup ..."
+sleep 1
+
+echo "Updating kernel ..."
+sleep 1
+cp -rf /root/uImage /boot/uboot/uImage
+ret=$?
+if [ ! $ret -eq 0 ]; then
+    echo "Updating uImage failed: #{ret}"
+    wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=443\&reason=uImage_copy_failed &> /dev/null
+    exit 1
+fi
+echo "kernel updated success"
+
+cat - <<'EOF' > /boot/uboot/update_rootfs.sh
+#!/bin/bash
+
+set -xe
+
+DIR=`pwd`
+MMC_DEV=/dev/mmcblk0
+ROOTFS_PART=${MMC_DEV}p2
+ROOTDIR_PART=${MMC_DEV}p5
+VARDIR_PART=${MMC_DEV}p6
+
+ROOTFS_PATH=/rootfs
+ROOTDIR_PATH=/rootdir
+VARDIR_PATH=/vardir
+
+#ROOTFS_URL="http://192.168.8.80/rootfs-release.tar.xz"
+#ROOTDIR_URL=
+#VARDIR_URL=
+
+mount_parts(){
+  ## make dir if missing
+  mkdir -p ${ROOTFS_PATH}
+  mkdir -p ${ROOTDIR_PATH} #not updating this at the moment.
+  mkdir -p ${VARDIR_PATH}
+
+  ## mount to correct dir
+  mount ${ROOTFS_PART} ${ROOTFS_PATH}
+  mount ${ROOTDIR_PART} ${ROOTDIR_PATH}
+  mount ${VARDIR_PART} ${VARDIR_PATH}
+}
+
+update_parts(){
+  ##get parts
+  if [ -f ${ROOTDIR_PATH}/rootfs.tar.xz ]; then
+    echo "Already dl'ed."
+  else
+    echo "No update packages, skipping."
+ #   wget -O ${ROOTDIR_PATH}/rootfs.tar.xz ${ROOTFS_URL}
+ #   wget -O ${ROOTDIR_PATH}/rootdir.tar.xz ${ROOTDIR_URL}
+ #   wget -O ${ROOTDIR_PATH}/vardir.tar.xz ${VARDIR_URL}
+  fi
+
+  #flash parts
+  cd ${ROOTFS_PATH}
+  cp ${ROOTFS_PATH}/etc/network/interfaces /root/interfaces
+  rm -rf *
+  xzcat ${ROOTDIR_PATH}/rootfs.tar.xz | tar xv
+
+  #cd ${ROOTDIR_PATH}
+  #rm -rf *
+  #xzcat ${ROOTDIR_PATH}/rootdir.tar.xz | tar xv
+
+  cd ${VARDIR_PATH}
+  cp ${VARDIR_PATH}/lib/luna/luna_client.sqlite3 /root/ || true
+  rm -rf *
+  xzcat ${ROOTDIR_PATH}/vardir.tar.xz | tar xv
+  mv /root/luna_client.sqlite3 ${VARDIR_PATH}/lib/luna/ || true
+  # make sure to copy authorized_keys
+  mkdir -p ${ROOTDIR_PATH}/.ssh || true
+  cp -f ${ROOTFS_PATH}/root/.ssh/authorized_keys ${ROOTDIR_PATH}/.ssh/ || true
+  mv /root/interfaces ${ROOTFS_PATH}/etc/network/interfaces || true
+}
+
+cleaning(){
+  rm -rf ${ROOTDIR_PATH}/rootfs.tar.xz   || true
+  rm -rf ${ROOTDIR_PATH}/rootdir.tar.xz  || true
+  rm -rf ${ROOTDIR_PATH}/vardir.tar.xz   || true
+}
+
+verify(){
+  sed -i "s/mmcblk0p3/mmcblk0p2/g" ${BOOTFS_PATH}/uEnv.txt
+}
+
+umount_parts(){
+  cd ${DIR}
+  umount ${ROOTFS_PART} || true
+  umount ${ROOTDIR_PART} || true
+  umount ${VARDIR_PART} || true
+}
+
+
+#do the job
+umount_parts
+mount_parts
+update_parts
+#verify
+cleaning
+umount_parts
+EOF
+
+chmod 770 /boot/uboot/update_rootfs.sh
+
+sed -i "s/mmcblk0p2/mmcblk0p3/g" /boot/uboot/uEnv.txt
+echo "Finished. Restarting device."
+
+wget -qO- ${UPDATE_HOST}/v3/update/report_progress?sn=${SN}\&status=20 &> /dev/null
+reboot

data/lib/luna_scanner/util.rb

@@ -0,0 +1,64 @@
+module LunaScanner
+  class Util
+    class << self
+
+      def ip_valid?(ip)
+        ip_array = ip.to_s.split(".")
+        return false if ip_array.size != 4
+
+        return false if ip_array[3].to_i <= 0 || ip_array[3].to_i > 255
+
+        true
+      end
+
+      def begin_ip(ip)
+        ip_array = ip.to_s.split(".")
+        raise "IP Error." if ip_array.size != 4
+
+        "#{ip_array[0]}.#{ip_array[1]}.#{ip_array[2]}.1"
+      end
+
+      def end_ip(ip)
+        ip_array = ip.to_s.split(".")
+        raise "IP Error." if ip_array.size != 4
+
+        "#{ip_array[0]}.#{ip_array[1]}.#{ip_array[2]}.255"
+      end
+
+      def ip_range(begin_ip, end_ip)
+        start_ip_array = begin_ip.split(".")
+        end_ip_array = end_ip.split(".")
+        raise "IP not valid." if start_ip_array.size != 4 || end_ip_array.size != 4
+        raise "IP not valid." if start_ip_array[3].to_i > 255 || end_ip_array[3].to_i > 255
+        raise "Start ip and end ip must be in same range." if start_ip_array[0] != end_ip_array[0] || start_ip_array[1] != end_ip_array[1]
+        raise "IP range not valid" if start_ip_array[2].to_i > end_ip_array[2].to_i
+        raise "IP range not valid" if (start_ip_array[2].to_i == end_ip_array[2].to_i) && start_ip_array[3].to_i > end_ip_array[3].to_i
+
+        ip_array = []
+        if start_ip_array[2].to_i == end_ip_array[2].to_i # 192.168.1.1 ~ 192.168.1.10
+          (start_ip_array[3].to_i..end_ip_array[3].to_i).step do |i|
+            ip_array << start_ip_array[0] + "." + start_ip_array[1] + "." + start_ip_array[2] + ".#{i}"
+          end
+        else # 192.168.1.1 ~ 192.168.3.10
+          (start_ip_array[2].to_i..end_ip_array[2].to_i).step do |i|
+            if start_ip_array[2].to_i == i
+              (start_ip_array[3].to_i..255).step do |j|
+                ip_array << start_ip_array[0] + "." + start_ip_array[1] + "." + start_ip_array[2] + ".#{j}"
+              end
+            elsif end_ip_array[2].to_i == i
+              (1..end_ip_array[3].to_i).step do |j|
+                ip_array << start_ip_array[0] + "." + start_ip_array[1] + "." + end_ip_array[2] + ".#{j}"
+              end
+            else
+              (1..255).step do |j|
+                ip_array << start_ip_array[0] + "." + start_ip_array[1] + ".#{i}.#{j}"
+              end
+            end
+          end
+        end
+
+        ip_array
+      end
+    end
+  end
+end

data/lib/luna_scanner/version.rb

@@ -0,0 +1,3 @@
+module LunaScanner
+  VERSION = "0.0.3"
+end

data/lib/luna_scanner/web.rb

@@ -0,0 +1,98 @@
+require "sinatra/base"
+
+module LunaScanner
+  class Web < Sinatra::Base
+    $scan_hosts = Array.new
+
+    enable :inline_templates
+
+    not_found do
+      "Sir, I don't understand what you mean."
+    end
+
+    before do
+      @local_ip ||= LunaScanner.local_ip
+    end
+
+    get '/' do
+      @start_ip = Util.begin_ip @local_ip if params[:start_ip].to_s.length == 0
+      @end_ip = Util.end_ip @local_ip if params[:end_ip].to_s.length == 0
+      erb :index
+    end
+
+    post '/scan' do
+      redirect "/" if !Util.ip_valid?(params[:start_ip]) || !Util.ip_valid?(params[:end_ip])
+
+      @start_ip = params[:start_ip]
+      @end_ip = params[:end_ip]
+
+      @ip_range = Util.ip_range(params[:start_ip], params[:end_ip])
+
+      scanner = LunaScanner::Scanner.new(100, params[:start_ip], params[:end_ip])
+
+
+      100.times do
+
+      end
+      erb :index
+    end
+
+  end
+end
+
+__END__
+
+@@ layout
+<html>
+<head>
+<title>Luna scanner</title>
+<style type='text/css'>
+.table td {
+border:1px solid #BDBDBD;
+}
+</style>
+</head>
+<body>
+<h1>Luna scanner</h1>
+Local IP: <%= @local_ip %>
+<%= yield %>
+<footer></footer>
+</body>
+</html>
+
+@@ index
+<div id='scan_form'>
+<form action='/scan' method='POST'>
+From <input type='text' name='start_ip' value="<%= @start_ip %>" /> to <input type='text' name='end_ip' value="<%= @end_ip %>" />
+<input type='submit' class='btn'/>
+</form>
+</div>
+
+<div id='scan_result'>
+<% if @ip_range && @ip_range.size > 0 %>
+<table class='table'>
+<thead>
+<tr>
+  <th>Select</th>
+  <th>Sequence</th>
+  <th>IP</th>
+  <th>Status</th>
+  <th>Operation</th>
+</tr>
+</thead>
+<tbody>
+<% @ip_range.each_with_index do |ip, index| %>
+<tr>
+<td style='text-align:center'>
+  <input type='checkbox' />
+</td>
+<td style='text-align:center'><%= index + 1 %></td>
+<td><%= ip %></td>
+<td></td>
+<td></td>
+</tr>
+<% end %>
+</tbody>
+</table>
+<% end %>
+</div>

data/luna_scanner.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'luna_scanner/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "luna_scanner"
+  spec.version       = LunaScanner::VERSION
+  spec.authors       = ["qichunren"]
+  spec.email         = ["whyruby@gmail.com"]
+  spec.description   = %q{Discover luna-client devices in LAN}
+  spec.summary       = %q{Discover luna-clinet devices in LAN, batch config them.}
+  spec.homepage      = "https://github.com/qichunren/luna-scanner"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.bindir = 'bin'
+
+  spec.add_runtime_dependency "net-ssh", "~> 2.7.0"
+  spec.add_runtime_dependency "net-scp", "~> 1.1.2"
+  spec.add_runtime_dependency "sinatra", "~> 1.4.4"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+end