script/vz-start

   1 #!/usr/bin/env ruby
   2
   3 SCRIPT_DIR = File.dirname(File.expand_path(__FILE__))
   4 SCRIPT_NAME = $0.split('/')[-1]
   5
   6 SSH           = '/usr/bin/ssh'      # TODO: pipe to ssh istead of using
   7                                     # the vzctl wrappers
   8 VZLIST        = '/usr/sbin/vzlist'
   9 VZCTL         = '/usr/sbin/vzctl'
  10
  11
  12 require "/etc/mwamko/available_ip_addresses.range"
  13
  14 if ARGV.size != 1 and ARGV[1] != '--kill-daemon'
  15
  16   STDERR.puts "SYNOPSIS:"
  17   STDERR.puts "  #{SCRIPT_NAME} \e[4mVEID\e[24m"
  18   STDERR.puts "  #{SCRIPT_NAME} \e[4mVEID\e[24m --kill-daemon"
  19   STDERR.puts ""
  20   STDERR.puts "DESCRIPTION:"
  21   STDERR.puts "  --kill_daemon "
  22   STDERR.puts "      Kills the daemon and removes related PID files from /tmp"
  23
  24   exit
  25 end
  26
  27 # Check ARGV
  28 raise 'Unsafe VEID' if !(ARGV[0] =~ /^[0-9]*$/)
  29
  30 $VEID = ARGV[0].to_i
  31
  32 if ARGV[1] != '--kill-daemon'
  33   # Check against vzlist
  34   out = `#{VZLIST} -a --no-header -o veid,status`
  35
  36   ve_found = false
  37   current_list = out.collect do |line|
  38
  39     ve, status = line.strip.split(' ')
  40     ve = ve.to_i
  41     status = status.strip
  42
  43     if $VEID == ve
  44       ve_found = true
  45       if status != 'stopped'
  46         STDERR.puts "VE #{$VEID} is already #{status}"
  47         exit
  48       end
  49     end
  50   end
  51
  52   if !ve_found
  53     STDERR.puts "VE #{$VEID} does not exist on the system"
  54     exit
  55   end
  56 end
  57
  58
  59 # All checks passed. Time to define and start the daemon
  60
  61 require "#{SCRIPT_DIR}/../lib/ciconia_daemonizer"
  62
  63 class CiconiaDaemonizer < Daemon::Base
  64   def self.start
  65     raise 'no $VEID' if $VEID == nil
  66
  67     run_and_record("echo > /dev/null")
  68
  69     begin
  70       require "#{SCRIPT_DIR}/../lib/database_for_scripts"
  71     rescue => detail
  72       @@ruby_errors =
  73         detail.message + "\n" +
  74         detail.backtrace.join("\n")
  75       run_and_record("echo > /dev/null")
  76     end
  77
  78     local_ips = []
  79     begin
  80       ve = VirtualEnvironment.find_by_veid($VEID.to_i)
  81       ve.grab_available_ip
  82       OpenVZ.set_ip_in_ve(ve)
  83
  84       ve.save!
  85
  86     rescue => detail
  87       @@ruby_errors =
  88         detail.message + "\n" +
  89         detail.backtrace.join("\n")
  90       run_and_record("echo > /dev/null")
  91     end
  92
  93     run_and_record("#{VZCTL} start #{$VEID}")
  94
  95 #    if !@@has_stderr and !@@has_rubyerr
  96 #      begin
  97 #        local_ips =
  98 #          LocalIP.find_all_by_veid($VEID.to_i).collect {|lip| lip.ip}
  99 #
 100 #        ip_list =
 101 #          local_ips.collect { |lip| "--ipadd #{lip}" }.join(' ')
 102 #
 103 #        unless ip_list.empty?
 104 #          run_and_record("#{VZCTL} set #{$VEID} #{ip_list} --save")
 105 #        end
 106 #      rescue => detail
 107 #        @@ruby_errors =
 108 #          detail.message + "\n" +
 109 #          detail.backtrace.join("\n")
 110 #        run_and_record("echo > /dev/null")
 111 #      end
 112 #
 113 #    end
 114
 115
 116     p @@std_errors_txt
 117     # The part in () is a hack - it ignores vzqouta errors
 118     if (@@std_errors_txt =~ /vzquota/ or (!@@has_stderr)) and !@@has_rubyerr
 119
 120       # Wait for SSHD to come up
 121       loop do
 122         if `#{VZCTL} exec #{$VEID} "ps | grep sshd"`.empty?
 123           sleep 2
 124         else
 125           break
 126         end
 127       end
 128
 129       begin
 130         ve = VirtualEnvironment.find_by_veid($VEID.to_i)
 131         ve.status = 'started'
 132         ve.updated_on = Time.now
 133
 134         if ve.setup_needed
 135           require "#{SCRIPT_DIR}/../lib/password_gen"
 136
 137           upass = PasswordGen.generate_pronounceable
 138           uname = User.find(ve.uid).name
 139
 140           OpenVZ.set_ve_userpassw(ve.veid, uname, upass)
 141           OpenVZ.add_user_to_ve_sudoers(uname, ve.veid)
 142           OpenVZ.ve_DNS_config(ve.veid)
 143           OpenVZ.ssh_keygen(ve.veid)
 144
 145           ve.setup_needed = false
 146         end
 147
 148         ve.save!
 149
 150       rescue => detail
 151         @@ruby_errors =
 152           detail.message + "\n" +
 153           detail.backtrace.join("\n")
 154         run_and_record("echo > /dev/null")
 155       end
 156
 157     end
 158
 159     if @@has_rubyerr
 160       `mv #{@@pid_file} #{@@pid_file[0..-5]}.rubyerr`
 161     elsif @@has_stderr
 162       `mv #{@@pid_file} #{@@pid_file[0..-5]}.stderr`
 163     elsif @@has_stdout
 164       `mv #{@@pid_file} #{@@pid_file[0..-5]}.out`
 165     else
 166       `rm #{@@pid_file}`
 167     end
 168
 169
 170   end
 171
 172   def self.stop
 173     `kill #{@child_pid}`
 174   end
 175 end
 176
 177
 178 # The daemon is ready.
 179 # Now it is time to run it, respecting the Queueing policy
 180
 181 task_identifier = "#{SCRIPT_NAME} #{$VEID}"
 182
 183 if ARGV[1] != '--kill-daemon'
 184   CiconiaDaemonizer.daemonize(task_identifier, 0)
 185   exit
 186 end
 187
 188
 189 # Enter the Queue
 190 # * If it is your turn then it is time to daemonize and run
 191 # * Else check if the queue is running
 192 #  * If it is, then put yourself in the queue and exit this script
 193 #  * If not, then safe start the Queue
 194 ##require "#{SCRIPT_DIR}/../lib/queue"
 195 ##q = Queue.new(SCRIPT_NAME, task_identifier)
 196 ##q.enter
 197
 198 ##CiconiaDaemonizer.daemonize(task_identifier, 0) rescue nil
 199
 200 # Figure out the next script in the queue
 201 #  * If there is one then run it
 202 ##q.exit
 203
 204