160 lines
7.7 KiB
Python
160 lines
7.7 KiB
Python
#!/usr/bin/python3
|
|
|
|
#/**
|
|
# configureOs
|
|
#@brief Script para realizar la configuracion del sistema operativo restaurado.
|
|
#@param 1 disco
|
|
#@param 2 particion
|
|
#@return
|
|
#@TODO comprobar que el tipo de particion corresponde con el sistema de archivos.
|
|
#@exception OG_ERR_FORMAT # 1 formato incorrecto.
|
|
#*/ ##
|
|
|
|
import sys
|
|
import os
|
|
import os.path
|
|
import subprocess
|
|
|
|
import ogGlobals
|
|
import SystemLib
|
|
import DiskLib
|
|
import FileSystemLib
|
|
import NetLib
|
|
import InventoryLib
|
|
import BootLib
|
|
import UEFILib
|
|
import PostConfLib
|
|
import FileLib
|
|
|
|
## el código bash original hace:
|
|
## [ -f $DEVICECFG ] && source $DEVICECFG
|
|
## pero luego no utiliza ninguna de las variables definidas en el archivo...
|
|
|
|
disk = sys.argv[1]
|
|
par = sys.argv[2]
|
|
|
|
# Si el sistema de archivos no esta extendido, ampliarlo al tamaño de su partición.
|
|
partsize = DiskLib.ogGetPartitionSize (disk, par)
|
|
if not partsize: sys.exit (1)
|
|
fssize = FileSystemLib.ogGetFsSize (disk, par)
|
|
if fssize < partsize:
|
|
print ('Extender sistema de archivos.')
|
|
FileSystemLib.ogExtendFs (disk, par)
|
|
|
|
# Si no existe partición activa, activar este sistema.
|
|
flagactive = DiskLib.ogGetPartitionActive (disk)
|
|
if not flagactive: DiskLib.ogSetPartitionActive (disk, par)
|
|
|
|
# Si el sistema de archivos es de solo lectura, no hacer la post-configuración.
|
|
mntdir = FileSystemLib.ogMount (disk, par)
|
|
if not FileSystemLib.ogIsWritable (disk, par):
|
|
print ('AVISO: sistema de archivos de solo lectura, no se ejecuta postconfiguración.')
|
|
sys.exit (0)
|
|
|
|
# Nombre del cliente.
|
|
host = NetLib.ogGetHostname()
|
|
|
|
# Post-configuración personalizada para cada tipo de sistema operativo.
|
|
ostype = InventoryLib.ogGetOsType (disk, par)
|
|
if 'Windows' == ostype:
|
|
if not host: host = 'pc'
|
|
BootLib.ogSetWindowsName (disk, par, host) # Cambiar nombre en sistemas Windows.
|
|
if InventoryLib.ogIsEfiActive(): # Si es UEFI copio el cargador de arranque a la partición EFI e instalo Grub.
|
|
UEFILib.ogRestoreEfiBootLoader (disk, par)
|
|
esp = DiskLib.ogGetEsp()
|
|
if not esp:
|
|
SystemLib.ogRaiseError ([], ogGlobals.OG_ERR_NOTFOUND, 'ESP')
|
|
sys.exit (1)
|
|
efidisk, efipart = esp.split()
|
|
BootLib.ogGrubInstallMbr (efidisk, efipart, 'TRUE')
|
|
else:
|
|
BootLib.ogFixBootSector (disk, par) # Configurar el boot sector de la partición Windows.
|
|
BootLib.ogWindowsBootParameters (disk, par) # Configurar el gestor de arranque de Windows XP/Vista/7.
|
|
BootLib.ogWindowsRegisterPartition (disk, par, 'C', disk, par) # Registrar en Windows que la partición indicada es su nueva unidad C:\
|
|
PostConfLib.ogConfigureOgagent (disk, par) # Configurar nuevo agente OGAgent.
|
|
path1 = FileLib.ogGetPath (file=f'{mntdir}/windows/ogAdmWinClient.exe')
|
|
path2 = FileLib.ogGetPath (file=f'{mntdir}/winnt/ogAdmWinClient.exe')
|
|
if path1 or path2: # Eliminar el antiguo cliente de Windows.
|
|
PostConfLib.ogInstallMiniSetup (disk, par, 'postconf.cmd')
|
|
PostConfLib.ogUninstallWindowsClient (disk, par, 'postconf.cmd')
|
|
|
|
elif 'Linux' == ostype:
|
|
BootLib.ogConfigureFstab (disk, par) # Configuro fstab: particion de Swap y si es UEFI además la partición EFI.
|
|
if InventoryLib.ogIsEfiActive(): # Si es UEFI instalo Grub en la partición EFI
|
|
esp = DiskLib.ogGetEsp()
|
|
efidisk, efipart = esp.split()
|
|
BootLib.ogGrubInstallMbr (efidisk, efipart, 'TRUE')
|
|
BootLib.ogGrubInstallPartition (disk, par) ## Instala (no configura) el codigo de arranque del Grub en la partición (no lo configura, se mantiene el original de la imagen)
|
|
find_out = subprocess.run (['find', f'{mntdir}/usr/sbin', f'{mntdir}/sbin', f'{mntdir}/usr/local/sbin', '-name', 'ogAdmLnxClient', '-print'], capture_output=True, text=True).stdout
|
|
if find_out:
|
|
PostConfLib.ogUninstallLinuxClient (disk, par) # Eliminar el antiguo cliente de Linux.
|
|
PostConfLib.ogConfigureOgagent (disk, par) # Configurar nuevo agente OGAgent.
|
|
## Modificar el nombre del equipo
|
|
print (f'Asignar nombre Linux "{host}".')
|
|
etc = FileLib.ogGetPath (src=f'{disk} {par}', file='/etc')
|
|
if os.path.isdir (etc):
|
|
with open (f'{etc}/hostname', 'w') as fd:
|
|
fd.write (f'{host}\n')
|
|
|
|
elif 'MacOS' == ostype:
|
|
open (f'{mntdir}/osxpostconf', 'a').close() ## touch, Fichero indicador de activación de postconfiguración.
|
|
|
|
# Crear fichero de configuración del servicio de arranque.
|
|
with open (f'{mntdir}/Library/LaunchDaemons/es.opengnsys.postconfd.plist', 'w') as fd:
|
|
fd.write ('<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">\n')
|
|
fd.write ('<plist version="1.0">\n')
|
|
fd.write ('\t<dict>\n')
|
|
fd.write ('\t\t<key>Label</key>\n')
|
|
fd.write ('\t\t<string>es.opengnsys.postconfd.sh</string>\n')
|
|
fd.write ('\t\t<key>ProgramArguments</key>\n')
|
|
fd.write ('\t\t<array>\n')
|
|
fd.write ('\t\t\t<string>/var/root/postconfd.sh</string>\n')
|
|
fd.write ('\t\t</array>\n')
|
|
fd.write ('\t\t<key>RunAtLoad</key>\n')
|
|
fd.write ('\t\t<true/>\n')
|
|
fd.write ('\t\t<key>StandardOutPath</key>\n')
|
|
fd.write ('\t\t<string>/var/log/postconfd.log</string>\n')
|
|
fd.write ('\t\t<key>StandardErrorPath</key>\n')
|
|
fd.write ('\t\t<string>/var/log/postconfd.err</string>\n')
|
|
fd.write ('\t\t<key>Debug</key>\n')
|
|
fd.write ('\t\t<true/>\n')
|
|
fd.write ('\t</dict>\n')
|
|
fd.write ('</plist>\n')
|
|
|
|
# Programa de inicio que será ejecutado en el arranque de Mac OS X.
|
|
with open (f'{mntdir}/var/root/postconfd.sh', 'w') as fd:
|
|
fd.write ('#!/bin/bash\n')
|
|
fd.write ('# postconfd - ejecución de scripts de inicio.\n')
|
|
fd.write ('\n')
|
|
fd.write ('# Ejecutar postconfiguración si existe el fichero indicador.\n')
|
|
fd.write ('if [ -e /osxpostconf ]; then\n')
|
|
fd.write ('\t# Tomar nombre del equipo.\n')
|
|
fd.write (f'\tHOST="{host}"\n')
|
|
fd.write ('\tif [ -z "$HOST" ]; then\n')
|
|
fd.write ('\t\t# Si no hay nombre asociado, activar la red para obtener datos del DHCP.\n')
|
|
fd.write ('\t\tsource /etc/rc.common\n')
|
|
fd.write ('\t\tCheckForNetwork\n')
|
|
fd.write ('\t\twhile [ "$NETWORKUP" != "-YES-" ]; do\n')
|
|
fd.write ('\t\t\tsleep 5\n')
|
|
fd.write ('\t\t\tNETWORKUP=\n')
|
|
fd.write ('\t\t\tCheckForNetwork\n')
|
|
fd.write ('\t\tdone\n')
|
|
fd.write ('\t\t# Componer nombre del equipo a partir de datos del DHCP.\n')
|
|
fd.write ('\t\tIP=$(ifconfig en0 inet | awk \'{if ($1=="inet") print $2}\')\n')
|
|
fd.write ('\t\tHOST="mac-$(echo ${IP//./-} | cut -f3-4 -d-)"\n')
|
|
fd.write ('\tfi\n')
|
|
fd.write ('\t# Asignar nombre del equipo.\n')
|
|
fd.write ('\tscutil --set ComputerName "$HOST"\n')
|
|
fd.write ('\tscutil --set LocalHostName "$HOST"\n')
|
|
fd.write ('\tscutil --set HostName "$HOST"\n')
|
|
fd.write ('\thostname "$HOST"\n')
|
|
fd.write ('\t# Descromprimir ficheros de versión para obtener inventario de aplicaciones.\n')
|
|
fd.write ('\tfind /Applications -type d -name "*.app" -prune -exec \\n')
|
|
fd.write ('\t ditto --nopreserveHFSCompression "{}/Contents/version.plist" "{}/Contents/version.plist.uncompress"\n')
|
|
fd.write ('\trm -f /osxpostconf # Borrar fichero indicador de psotconfiguración\n')
|
|
fd.write ('fi\n')
|
|
os.chmod (f'{mntdir}/var/root/postconfd.sh', 0o700) # Dar permiso de ejecución.
|
|
PostConfLib.ogConfigureOgagent (disk, par) # Configurar nuevo agente OGAgent de sistema operativo.
|
|
|
|
sys.exit (0)
|