mirror of
https://github.com/apache/cloudstack.git
synced 2025-11-04 00:02:37 +01:00
94 lines
2.4 KiB
Python
94 lines
2.4 KiB
Python
#!/usr/bin/python
|
|
|
|
|
|
|
|
#
|
|
# Copyright (C) 2010 Cloud.com, Inc. All rights reserved.
|
|
#
|
|
# This software is licensed under the GNU General Public License v3 or later.
|
|
#
|
|
# It is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or any later version.
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
|
|
|
|
import sys
|
|
import base64
|
|
import string
|
|
import os
|
|
import tempfile
|
|
from subprocess import call
|
|
|
|
def vm_data(args):
|
|
|
|
router_ip = args.pop('routerIP')
|
|
vm_ip = args.pop('vmIP')
|
|
|
|
for pair in args:
|
|
pairList = pair.split(',')
|
|
vmDataFolder = pairList[0]
|
|
vmDataFile = pairList[1]
|
|
vmDataValue = args[pair]
|
|
cmd = ["/bin/bash", "/root/userdata.sh", "-v", vm_ip, "-F", vmDataFolder, "-f", vmDataFile]
|
|
|
|
fd = None
|
|
tmp_path = None
|
|
if (vmDataValue != "none"):
|
|
try:
|
|
fd,tmp_path = tempfile.mkstemp()
|
|
tmpfile = open(tmp_path, 'w')
|
|
|
|
if (vmDataFolder == "userdata"):
|
|
vmDataValue = base64.urlsafe_b64decode(vmDataValue)
|
|
|
|
tmpfile.write(vmDataValue)
|
|
tmpfile.close()
|
|
cmd.append("-d")
|
|
cmd.append(tmp_path)
|
|
except:
|
|
if (fd !=None):
|
|
os.close(fd)
|
|
os.remove(tmp_path)
|
|
return ''
|
|
|
|
try:
|
|
call(cmd)
|
|
txt = 'success'
|
|
except:
|
|
txt = ''
|
|
|
|
if (fd != None):
|
|
os.close(fd)
|
|
os.remove(tmp_path)
|
|
|
|
return txt
|
|
|
|
def parseFileData(fileName):
|
|
args = {}
|
|
fd = open(fileName)
|
|
|
|
line = fd.readline()
|
|
while (line != ""):
|
|
key=string.strip(line[:], '\n')
|
|
if (key == ""):
|
|
break
|
|
|
|
line=fd.readline()
|
|
val=string.strip(line[:], '\n')
|
|
args[key]=val
|
|
line=fd.readline()
|
|
return args
|
|
|
|
if __name__ == "__main__":
|
|
vm_data(parseFileData("/tmp/" + sys.argv[1]))
|
|
|