puppetlabs/vmpooler
2
0
Fork 0
mirror of https://github.com/puppetlabs/vmpooler.git synced 2026-01-26 10:08:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Validate data payload before operating on it

Browse source
This commit is contained in:
Scott Schneider 2015-03-20 11:16:13 -07:00
parent 99a18c781a
commit d48d487de0
1 changed files with 29 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

43
lib/vmpooler/api/v1.rb
View file

@ -494,6 +494,8 @@ module Vmpooler
put "#{api_prefix}/vm/:hostname/?" do
content_type :json
failure = false
result = {}
status 404
@ -504,28 +506,41 @@ module Vmpooler
if $redis.exists('vmpooler__vm__' + params[:hostname])
jdata = JSON.parse(request.body.read)
# Validate data payload
jdata.each do |param, arg|
case param
when 'lifetime'
arg = arg.to_i
if arg > 0
$redis.hset('vmpooler__vm__' + params[:hostname], param, arg)
status 200
result['ok'] = true
unless arg.to_i > 0
failure = true
end
when 'tags'
if arg.is_a?(Hash)
arg.keys.each do |tag|
$redis.hset('vmpooler__vm__' + params[:hostname], 'tag:' + tag, arg[tag])
end
status 200
result['ok'] = true
unless arg.is_a?(Hash)
failure = true
end
else
failure = true
end
end
if failure
status 400
else
jdata.each do |param, arg|
case param
when 'lifetime'
arg = arg.to_i
$redis.hset('vmpooler__vm__' + params[:hostname], param, arg)
when 'tags'
arg.keys.each do |tag|
$redis.hset('vmpooler__vm__' + params[:hostname], 'tag:' + tag, arg[tag])
end
end
end
status 200
result['ok'] = true
end
end
JSON.pretty_generate(result)