forked from pentestify/lab
-
Notifications
You must be signed in to change notification settings - Fork 0
/
labmon.rb
56 lines (47 loc) · 1.07 KB
/
labmon.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
require 'sinatra'
require 'lab'
before do
## Basic blacklisting of metacharacters
redirect to "/exception" if request.path_info =~ /\;|\|/
@controller = Lab::Controllers::VmController.new
if File.exist? "config.txt"
@controller.from_file(File.open("config.txt").read.strip)
end
end
get '/' do
redirect to "/list"
end
get '/exception' do
"sorry, that's not allowed, request contains bad data"
end
get '/list' do
erb :list
end
get '/show' do
redirect to '/list'
end
get '/show/:hostname' do
# Get the watcher
hostname = params[:hostname]
@vm = @controller[hostname]
erb :show
end
get '/start/:hostname' do
hostname = params[:hostname]
@vm = @controller[hostname]
@vm.start
redirect to "/show/#{hostname}"
end
get '/stop/:hostname' do
hostname = params[:hostname]
@vm = @controller[hostname]
@vm.stop
redirect to "/show/#{hostname}"
end
get '/revert_snapshot/:hostname' do
hostname = params[:hostname]
snapshot = params[:snapshot] || "clean"
@vm = @controller[hostname]
@vm.revert_snapshot snapshot
redirect to "/show/#{hostname}"
end