from celery import shared_task from distutils.dir_util import copy_tree from .views import makewificonfig from .hostname import hostname import os import time import glob import tempfile import crypt import subprocess import datetime @shared_task def mkfirmware(device, path): BEFORE = os.getcwd() DEVICEDIR = "/opt/vpnmanager/device-config/%i/" % device.model.id SRCDIR = "/opt/vpnmanager/imagebuilder/%i/" % device.model.id if glob.glob(SRCDIR + "/.kumilock"): return False with open(SRCDIR + "/.kumilock", "w") as lock: lock.write("") tempdir = tempfile.TemporaryDirectory() copy_tree(DEVICEDIR, tempdir.name) # Write OpenVPN config with open(tempdir.name + "/etc/openvpn/client.conf", "w") as vpnconf: vpnconf.write(device.vpnconfig) # Write secret with open(tempdir.name + "/etc/vpnsecret", "w") as secret: secret.write('SECRET="%s"\nHOSTNAME="%s"' % (device.secret, hostname)) # Write password with open(tempdir.name + "/etc/shadow", "r") as shadow: password = crypt.crypt(device.password, crypt.mksalt(crypt.METHOD_MD5)) shadowin = shadow.read() with open(tempdir.name + "/etc/shadow", "w") as shadowout: shadowout.write(shadowin.replace("$PASSWORD", password)) # Write Wireless config with open(tempdir.name + "/etc/config/wireless", "r") as wireless: wirein = wireless.read() with open(tempdir.name + "/etc/config/wireless", "w") as wireout: wire = wirein.replace("$SSID", device.ssid or device.serial) wire = wire.replace("$KEY", device.key or "") wire = wire.replace("$ENC", "psk2" if device.key else "none") wireout.write(wire + "\n" + makewificonfig(device)) # Create compilation environment os.system("rm -rf " + SRCDIR + "/files/") os.mkdir(SRCDIR + "/files/") os.system("cp -r " + tempdir.name + "/* " + SRCDIR + "/files/") tempdir.cleanup() os.system("rm " + SRCDIR + "/bin/targets/ar71xx/generic/*") # Build image os.chdir(SRCDIR) try: subprocess.run(["/usr/bin/make"], shell=True) os.chdir(BEFORE) os.rename(glob.glob(SRCDIR + "/bin/targets/ar71xx/generic/*squashfs-sysupgrade.bin")[0], "%s/%s.bin" % (path, device.id)) try: os.remove(SRCDIR + "/.kumilock") except: pass os.system("rm -rf " + SRCDIR + "/files/") os.system("rm " + SRCDIR + "/bin/targets/ar71xx/generic/*") return True except Exception as e: os.remove(SRCDIR + "/.kumilock") os.chdir(BEFORE) return False