mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
28855b4987
Detail: userdata and vm metadata take a long time to program on KVM routers. This does it all in one go, processed on the router. BUG-ID: CLOUDSTACK-3163 Tested-by: Wido Signed-off-by: Marcus Sorensen <marcus@betterservers.com> 1374695897 -0600
141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
#!/usr/bin/python
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import sys, getopt, json, os, base64
|
|
|
|
def main(argv):
|
|
fpath = ''
|
|
b64data = ''
|
|
|
|
try:
|
|
opts, args = getopt.getopt(argv,"f:d:")
|
|
except getopt.GetoptError:
|
|
print 'params: -f <filename> -d <b64jsondata>'
|
|
sys.exit(2)
|
|
for opt, arg in opts:
|
|
if opt == '-f':
|
|
fpath = arg
|
|
elif opt == '-d':
|
|
b64data = arg
|
|
|
|
json_data = ''
|
|
if fpath != '':
|
|
fh = open(fpath, 'r')
|
|
json_data = json.loads(fh.read())
|
|
elif b64data != '':
|
|
json_data = json.loads(base64.b64decode(b64data))
|
|
else:
|
|
print '-f <filename> or -d <b64jsondata> required'
|
|
sys.exit(2)
|
|
|
|
for ip in json_data:
|
|
for item in json_data[ip]:
|
|
folder = item[0]
|
|
file = item[1]
|
|
data = item[2]
|
|
|
|
# process only valid data
|
|
if folder != "userdata" and folder != "metadata":
|
|
continue
|
|
|
|
if file == "":
|
|
continue
|
|
|
|
htaccess(ip, folder, file)
|
|
|
|
if data == "":
|
|
deletefile(ip, folder, file)
|
|
else:
|
|
createfile(ip, folder, file, data)
|
|
|
|
if fpath != '':
|
|
fh.close()
|
|
os.remove(fpath)
|
|
|
|
def deletefile(ip, folder, file):
|
|
datafile = "/var/www/html/" + folder + "/" + ip + "/" + file
|
|
|
|
if os.path.exists(datafile):
|
|
os.remove(datafile)
|
|
|
|
def createfile(ip, folder, file, data):
|
|
dest = "/var/www/html/" + folder + "/" + ip + "/" + file
|
|
metamanifestdir = "/var/www/html/" + folder + "/" + ip
|
|
metamanifest = metamanifestdir + "/meta-data"
|
|
|
|
# base64 decode userdata
|
|
if folder == "userdata" or folder == "user-data":
|
|
if data is not None:
|
|
data = base64.b64decode(data)
|
|
|
|
if data is not None:
|
|
open(dest, "w").write(data)
|
|
else:
|
|
open(dest, "w").write("")
|
|
os.chmod(dest, 0644)
|
|
|
|
if folder == "metadata" or folder == "meta-data":
|
|
if not os.path.exists(metamanifestdir):
|
|
os.makedirs(metamanifestdir, 0755)
|
|
if os.path.exists(metamanifest):
|
|
if not file in open(metamanifest).read():
|
|
open(metamanifest, "a").write(file + '\n')
|
|
else:
|
|
open(metamanifest, "w").write(file + '\n')
|
|
|
|
if os.path.exists(metamanifest):
|
|
os.chmod(metamanifest, 0644)
|
|
|
|
def htaccess(ip, folder, file):
|
|
entry = "RewriteRule ^" + file + "$ ../" + folder + "/%{REMOTE_ADDR}/" + file + " [L,NC,QSA]"
|
|
htaccessFolder = "/var/www/html/latest"
|
|
htaccessFile = htaccessFolder + "/.htaccess"
|
|
|
|
if not os.path.exists(htaccessFolder):
|
|
os.mkdir(htaccessFolder,0755)
|
|
|
|
if os.path.exists(htaccessFile):
|
|
if not entry in open(htaccessFile).read():
|
|
open(htaccessFile, "a").write(entry + '\n')
|
|
|
|
entry="Options -Indexes\nOrder Deny,Allow\nDeny from all\nAllow from " + ip
|
|
htaccessFolder = "/var/www/html/" + folder + "/" + ip
|
|
htaccessFile = htaccessFolder+"/.htaccess"
|
|
|
|
if not os.path.exists(htaccessFolder):
|
|
os.makedirs(htaccessFolder,0755)
|
|
|
|
open(htaccessFile, "w").write(entry + '\n')
|
|
|
|
if folder == "metadata" or folder == "meta-data":
|
|
entry="RewriteRule ^meta-data/(.+)$ ../" + folder + "/%{REMOTE_ADDR}/$1 [L,NC,QSA]"
|
|
htaccessFolder = "/var/www/html/latest"
|
|
htaccessFile = htaccessFolder + "/.htaccess"
|
|
|
|
if not entry in open(htaccessFile).read():
|
|
open(htaccessFile, "a").write(entry + '\n')
|
|
|
|
entry="RewriteRule ^meta-data/$ ../" + folder + "/%{REMOTE_ADDR}/meta-data [L,NC,QSA]"
|
|
|
|
if not entry in open(htaccessFile).read():
|
|
open(htaccessFile, "a").write(entry + '\n')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|