#!/usr/bin/python """ Written by Igor Pashev The author has placed this work in the Public Domain, thereby relinquishing all copyrights. Everyone is free to use, modify, republish, sell or give away this work without prior consent from anybody. """ from __future__ import print_function from DysonInstaller.hdd import HDD from DysonInstaller.snack import * from snack import * from subprocess import Popen, PIPE, call from pprint import pprint from tempfile import mkstemp, mkdtemp from urllib2 import urlopen, URLError, HTTPError from time import sleep from fnmatch import fnmatch from os.path import basename, exists import os import re import sys import math import shutil import traceback import apt # Snack screen screen = None # Physical network interfaces physlinks = [] # List of disks installed on the system: hdds = [] # List of zpools found on system: zpools = [] # List of imported zpools, we should export them after installation: imported_zpools = [] # Name of the root ZFS pool: either existing or newly created rpool = None # Name of the root slice, where ZFS root pool will be created (e. g. c0t0d0s0) # Used for installing GRUB rslice = None # Object, describing solaris partition solaris_partition = None # XXX rslice and solaris_partition aare known only # on new clean installations, so we allow installing GRUB # only on new installations # Created boot environment, e. g. rpoot/ROOT/osdy-0 bootenv = None # Install root dir, mount point for bootenv rootdir = None # Version to install: codename = 'lacaille' # Arch to install: # XXX Use dpkg-architecture when we support more ;-) archname = 'illumos-amd64' # Dyson mirrors: mirror = None mirrors = [ ('http://apt.osdyson.org', 'Japan'), ('http://ftp.nluug.nl/os/illumos/dyson/apt/', 'Netherlands'), ('http://mirror-us.osdyson.org/apt/', 'USA'), ('http://mirror-ru.osdyson.org/apt/', 'Russia'), ] apt_cache = None # APT cache, used to check if package exists, updated after 'apt-get update' profile = None profiles = [ { # This must be the first 'name': 'Minimal', 'desc' : 'Minimal package set with minimal configuration', }, { 'name': 'Basic', 'desc' : 'Basic system without X', # Some packages may be absent, they will be skipped: 'packages' : '''locales vim bash-completion mc sudo lynx mutt wget curl openssh-server '''.split(), }, { 'name': 'Desktop', 'desc' : 'Install X, XFCE and some graphical applications', 'packages' : '''mc locales bash-completion vim-gtk synaptic abiword dillo gnumeric sylpheed geany lightdm pidgin evince sudo gimp midori stardict mirage xorg xfce4 xfce4-terminal dbus curl wget transmission-gtk xarchiver xfce4-screenshooter xfce4-clipman-plugin gksu libgtk2-perl xfce4-xkb-plugin '''.split(), }, ] # Top of /rpool/boot/grub/menu.lst, if creating new file: grub_top = r''' foreground 343434 background F7FbFF default 0 timeout 5 ''' # Entry in /rpool/boot/grub/menu.lst for newly created boot environment # This string must be formatted. # 0 is the partition number, # a is the slice number; both are not used grub_entry = r''' title Dyson at {bootenv} findroot (pool_{rpool},{partition},{slice}) bootfs {bootenv} kernel$ /platform/i86pc/kernel/amd64/unix -B $ZFS-BOOTFS module$ /platform/i86pc/amd64/boot_archive ''' # Stages of debootstraping in order. Debootstrap gives # progress for each stage, but we need entire progress. # Format: mile : (a, z), where: # - mile is a stage name, e. g. DOWNDEBS; # - a, z - the start and the end position on progress bar, # e. i. a=0 z=100 - entire progress bar (100%); miles = { 'START' : (0, 0), # Dummy 'DOWNREL' : (0, 1), # Downloading the 'Release' file 'DOWNPKGS' : (1, 5), # Downloading the 'Packages' file 'SIZEDEBS' : (5, 10), # Finding packags sizes 'DOWNDEBS' : (10, 25), # Downloading packages (*.deb) 'EXTRACTPKGS' : (25, 45), # Extracting core packages 'INSTCORE' : (45, 50), # Installing core packages 'UNPACKREQ' : (50, 60), # Unpacking required packages 'CONFREQ' : (60, 70), # Configuring required packages 'UNPACKBASE' : (70, 85), # Unpacking the base system 'CONFBASE' : (85, 100), # Configuring the base system } # FS to be created in boot environment. # Order is important. befs = [ {'name':'/usr', 'options':{}}, {'name':'/usr/local', 'options':{ 'compression':'on', }}, {'name':'/usr/src', 'options':{ 'compression':'on', }}, {'name':'/var', 'options':{}}, {'name':'/var/cache', 'options':{}}, {'name':'/var/lib', 'options':{}}, {'name':'/var/lib/dpkg', 'options':{ 'compression' : 'on', 'setuid' : 'off', 'devices' : 'off', }}, {'name':'/var/log', 'options':{ 'compression' : 'on', 'setuid' : 'off', 'devices' : 'off', }}, {'name':'/var/mail', 'options':{ 'compression' : 'on', 'setuid' : 'off', 'devices' : 'off', }}, {'name':'/var/spool', 'options':{ 'compression' : 'on', 'setuid' : 'off', 'devices' : 'off', }}, {'name':'/var/tmp', 'options':{ 'compression' : 'on', 'setuid' : 'off', 'devices' : 'off', }}, ] # XXX Not yet. befs = [] class Abort(Exception): ''' User clicked "Cancel" ''' pass class NoDisks(Exception): ''' No Hard Drives found ''' pass def welcome(): welcome = ButtonChoiceWindow(screen, "Welcome to the Dyson installer", ''' Dyson is an operating system derived from Debian and based on Illumos core. \ It uses Illumos unix kernel and libc, ZFS file system and SMF to manage system startup. \ To learn more visit http://osdyson.org This installer will guide you through disk paritioning, filesystem creation, \ installation of base system and minimal configuration. Please note, this system IS VERY EXPERIMENTAL. \ You can LOOSE ALL YOUR DATA which this system can reach ;-) Whould you like to continue?''', buttons=[('Install Dyson', True), ('Exit the installer', False)], width=70) if not welcome: raise Abort('Installation canceled') def get_memory_size(): return os.sysconf("SC_PHYS_PAGES") * os.sysconf("SC_PAGE_SIZE") def get_zpool_slice(pool_name): result = [] ''' Returns slices/disks (e. g. c0t4d0s0) on which zpool exists ''' zpool_cmd = Popen(['zpool', 'status', pool_name], stderr=PIPE, stdout=PIPE) # Skip till the header: re_header = re.compile(r'NAME\s+STATE\s+READ\s+WRITE\s+CKSUM') for line in zpool_cmd.stdout: if re_header.search(line): break re_disk = re.compile(r'(\w+)\s+ONLINE(\s+\d+){3}') for line in zpool_cmd.stdout: m = re_disk.search(line) if m and not m.group(1) in [pool_name, 'mirror']: result.append(m.group(1)) return result def find_zpools(): global zpools progress = ProgressMessage(screen, title='Please wait', text='Searching for ZFS pools ...') zpools = [] zpool_cmd = Popen(['zpool', 'list', '-H', '-o', 'name'], stdout=PIPE, stderr=PIPE) for line in zpool_cmd.stdout: zpools.append(line.rstrip()) zpool_cmd = Popen("zpool import | awk '/pool:/ {print $2}'", shell=True, stdout=PIPE, stderr=PIPE) for line in zpool_cmd.stdout: zpools.append(line.rstrip()) def find_hdds(): global hdds progress = ProgressMessage(screen, title='Please wait', text='Searching for disks ...') hdds = [] pat = re.compile('\d+\.\s+(\S+)\s+<(.+?)(\s+cyl.*)?>') format_cmd = Popen('format 0: rpool = choose_rpool() if not rpool: hdd = choose_hdd(len(zpools)) if hdd: rpool = create_rpool(hdd) if rpool: if not pool_is_imported(rpool): progress = ProgressMessage(screen, title='Please wait', text='Importing ZFS pool "{}" ...'.format(rpool)) zpool_cmd = Popen(['zpool', 'import', '-R', '/mnt/'+rpool, '-f', rpool], stderr=PIPE, stdout=PIPE) out, err = zpool_cmd.communicate() if 0 != zpool_cmd.returncode: ButtonChoiceWindow(screen, title='Importing of ZFS pool "{}" failed'.format(rpool), text=err, buttons=['Ok'], width=70) rpool = None imported_zpools.append(rpool) def zfs_exists(fs): fs_exists = call(['zfs', 'list', '-H', fs], stdout=PIPE, stderr=PIPE) return fs_exists == 0 def zfs_create(fs, zvol=None, options={}): args = ['zfs', 'create'] if zvol: args += ['-V', zvol] for o in options: args += ['-o', o + '=' + options[o]] args.append(fs) zfs_cmd = Popen(args, stderr=PIPE, stdout=PIPE) out, err = zfs_cmd.communicate() if 0 != zfs_cmd.returncode: ButtonChoiceWindow(screen, title='Creating of ZFS filesystem failed', text=err, buttons = ['Ok'], width=70) return False return True def configure_zfs(): global bootenv global rootdir progress = ProgressBar(screen, title='Creating ZFS filesystems', top = 4 + len(befs), # 4 for ROOT, /swap, /home, and BE width = 70, ) root = rpool + '/ROOT' if not zfs_exists(root): progress.text = 'Creating ' + root zfs_create(root, options={'canmount':'off', 'mountpoint':'none'}) progress.advance() swap = rpool + '/swap' if not zfs_exists(swap): progress.text = 'Creating ' + swap zfs_create(swap, zvol='2G') progress.advance() home = rpool + '/home' if not zfs_exists(home): progress.text = 'Creating ' + home zfs_create(home, options={'mountpoint':'/home', 'compression':'on', 'devices':'off', 'setuid':'off'}) progress.advance() # Make sure we are creating new boot environment for be_no in range(50): bootenv = root + '/osdy-' + str(be_no) if not zfs_exists(bootenv): break else: bootenv = None rootdir = mkdtemp(prefix='install-') progress.text = 'Creating ' + bootenv zfs_create(bootenv, options={'mountpoint':'legacy'}) call(['mount', '-F', 'zfs', bootenv, rootdir]) progress.advance() for fs in befs: p = bootenv + fs['name'] progress.text = 'Creating ' + p zfs_create(p, options=fs['options']) progress.advance() # Will rollback if debootstrap fails call(['zfs', 'snapshot', '-r', bootenv + '@empty']) def choose_hdd(number_of_zpools=0): hdd_items = map(lambda x: '{}: {} {}'.format(x.name, x.capacity, x.description), hdds) choice = None buttons = [] buttons.append(('Use selected disk', 'ok')) if number_of_zpools > 0: buttons.append(('Back', 'zpool')) buttons.append(('Cancel', 'cancel')) while not choice in ['ok', 'zpool', 'cancel']: (choice, hdd_no) = ListboxChoiceWindow(screen, title='Choose hard disk drive', text='Choose a disk for the root ZFS pool', items=hdd_items, buttons=buttons, width=76, scroll=1, default=0, help=None ) if choice in ['ok', None]: choice = configure_partitions(hdds[hdd_no], len(hdds)) if choice == 'another': choice = None if choice == 'ok': return hdds[hdd_no] if choice == 'zpool': return None raise Abort('HDD choosing') def configure_partitions(hdd, number_of_disks=1): global solaris_partition choice = None top_text = 'For the root ZFS pool you need a disk containing a Solaris partition (id 0xbf). \ If this disk includes such a partition and you are happy with it, use this disk. \ Otherwise you have to change partitioning. Note that only the first Solaris partition \ will be used.' while not choice in ['ok', 'another', 'cancel']: part_info = top_text part_info += '\n\n' have_partitions = len(hdd.partitions) > 0 have_gpt = False solaris_partition = None buttons = [] if have_partitions: part_info += 'Partitions on {name} {desc} {size}:\n\n'.format( name=hdd.name, desc=hdd.description, size=hdd.capacity) part_info += ' TYPE ID NAME SIZE\n' n = 0 for p in hdd.partitions: n = n + 1 if p.id == 0xBF and solaris_partition == None: solaris_partition = p if p.id == 0xEE: have_gpt = True part_info += '#{n:<2} {primary} {id:#04x} {system:25} {capacity:10}\n'.format( n=n, capacity=p.capacity, system=p.system, id=p.id, primary=['primary', 'logical'][p.logical]) else: part_info += 'This disk is not partitioned.\n' if solaris_partition != None: buttons.append(('Use this disk', 'ok')) if have_gpt: part_info += '\nThis disk has GUID Partition Table (GPT) and is unsupported.\n' # XXX gparted else: buttons.append(('Edit partitions', 'fdisk')) if number_of_disks > 1: buttons.append(('Back', 'another')) buttons.append(('Cancel', 'cancel')) choice = ButtonChoiceWindow(screen, title='Partitions of {}'.format(hdd.name), text=part_info, buttons=buttons, width=76, help=None ) if choice == 'fdisk': screen.suspend() call(['cfdisk', hdd.raw_device]) hdd.reread_partitions() screen.resume() # end while if choice == 'cancel': raise Abort('Partitioning') return choice def get_mirror(): if not hasattr(get_mirror, "m"): get_mirror.m='' while True: (choice, entry) = EntryWindow(screen, title='Enter APT repository URL', text='If you want to have line ' '"deb ftp://192.168.1.1 bok main" in /etc/apt/sources.list ' 'enter "ftp://192.168.1.1" here.\n' 'Remember to add URL protocol.\n', prompts=[('URL', get_mirror.m)], buttons = [ ('Ok', 'ok'), ('Back', 'back'), ('Cancel', 'cancel')], width=70, entryWidth=50, ) if choice == 'back': return None if choice in ['ok', None]: get_mirror.m = entry[0].strip() if valid_mirror(get_mirror.m): return get_mirror.m if choice == 'cancel': raise Abort('Entering an APT mirror') def choose_physlink(): pass def configure_network(): global physlinks if not physlinks: physlinks = dladm_show_phys() if not physlinks: choice = ButtonChoiceWindow(screen, title='Network is unreachable', text='No network interfaces found on this system. ' 'This mean that public APT repositories cannot be used ' 'to install Dyson.', buttons=['Ok']) return while True: if len(physlinks) > 1: link = choose_physlink() else: link = physlinks[0] break def valid_mirror(mirror): progress = ProgressMessage(screen, title='Please wait...', text='Checking APT mirror: ' + mirror) try: o = urlopen('{mirror}/dists/{codename}/Release'.format( mirror=mirror, codename=codename), timeout=15) have_code_name = False have_arch = False re_arch = re.compile(r'^Architectures:.+\b{archname}\b.*$'.format(archname=archname)) re_code = re.compile(r'^Codename: +\b{codename}\b.*$'.format(codename=codename)) for line in o.read(400).split('\n'): if re_code.match(line): have_code_name = True if re_arch.match(line): have_arch = True if have_code_name and have_arch: return True raise URLError('Not a valid Dyson APT repository') except HTTPError as e: ButtonChoiceWindow(screen, title='Error', text='{code} {reason}.\n\nMirror {mirror} is not usable.'.format( mirror=mirror, code=e.code, reason=e.reason), buttons = ['Ok'], width=70) except URLError as e: ButtonChoiceWindow(screen, title='Error', text='{reason}.\n\nMirror {mirror} is not usable.'.format( reason=e.reason,mirror=mirror), buttons = ['Ok'], width=70) except ValueError: ButtonChoiceWindow(screen, title='Failed to check mirror', text='Invalid URL: ' + mirror, buttons = ['Ok'], width=70) except Exception: ButtonChoiceWindow(screen, title='Failed to check mirror', text='Unknown error: ' + mirror, buttons = ['Ok'], width=70) return False def configure_mirror(): global mirror global mirrors maxlen = 0 for m in mirrors: if len(m[0]) > maxlen: maxlen = len(m[0]) items = map(lambda x: '{url: <{0}} - {info}'.format(maxlen, url=x[0], info=x[1]), mirrors) items.append(('Enter another mirror', None)) mirror = None while not mirror: (choice, m) = ListboxChoiceWindow(screen, title='Choose APT mirror', text='Choose APT repository from which Dyson will be installed', items=items, buttons=[('Ok','ok'), ('Cancel', 'cancel')], width=76, scroll=1, default=0, ) if choice in ['ok', None]: if m == None: mirror = get_mirror() else: if valid_mirror(mirrors[m][0]): mirror = mirrors[m][0] if choice == 'cancel': raise Abort('Choosing APT mirror') def choose_profile(): global profile global profiles maxlen = 0 for p in profiles: if len(p['name']) > maxlen: maxlen = len(p['name']) items = map(lambda x: '{profile: <{0}} - {desc}'.format(maxlen, profile=x['name'], desc=x['desc']), profiles) while not profile: (choice, p) = ListboxChoiceWindow(screen, title='Install software', text='Choose software profile. Additional software will be installed depending on your choice.', items=items, buttons=[('Ok','ok'), ('Cancel', 'cancel')], width=76, scroll=1, default=1, # default is "basic" ) if choice == 'cancel': raise Abort('Choosing software profile') if choice in ['ok', None]: profile = p def debootstrap(): progress = ProgressBar(screen, title='Installing base system, please wait...', width=70) progress.text = ' ' read, write = os.pipe() os.makedirs(rootdir + '/root') log = os.open(rootdir + '/root/debootstrap.log', os.O_WRONLY + os.O_CREAT) pid = os.fork() if pid == 0: os.close(read) os.dup2(write, 3) os.dup2(log, 1) os.dup2(log, 2) os.close(log) os.close(write) try: os.execl('/usr/sbin/debootstrap', 'debootstrap', '--debian-installer', '--exclude=gawk,aptitude,aptitude-common,libboost-iostreams1.48.0,libboost-iostreams1.49.0,libcwidget3', '--include=illumos-grub,illumos-kernel', codename, rootdir, mirror) except EnvironmentError as e: sys._exit(e.errno) else: os.close(write) di = os.fdopen(read, 'r', 1) info_re = re.compile(' +') m = 'START' IA = [] # Arguments for IF, cleaned after each IF PA = [] # Arguments for PF, cleaned after each PF EA = [] # Arguments for EF, cleaned after each EF WA = [] # Arguments for WF, cleaned after each WF while True: line = di.readline() if not line: break try: colon = line.find(':') if colon == -1: continue cmd = line[:colon] info = line[colon+1:].strip() text = None if cmd == 'P': # "P: 23 454 XYZ" or "P: 123 3445" p = info_re.split(info) if len(p) == 3: m = p[2].strip() progress.progress = math.ceil(miles[m][0] + float(p[0])*(miles[m][1] - miles[m][0]) / float(p[1])) elif cmd == 'IA': IA.append(info) elif cmd == 'PA': PA.append(info) elif cmd == 'EA': EA.append(info) elif cmd == 'WA': WA.append(info) elif cmd == 'IF': text = (info % tuple(IA)) IA = [] elif cmd == 'PF': text = (info % tuple(PA)) PA = [] elif cmd == 'EF': text = (info % tuple(EA)) EA = [] elif cmd == 'WF': text = (info % tuple(WA)) WA = [] if text: progress.text = text os.write(log, text) os.write(log, '\n') except: pass status = os.wait()[1] os.close(log) return os.WEXITSTATUS(status) def install_minimal(): while True: configure_mirror() code = debootstrap() if code == 0: return while True: choice = ButtonChoiceWindow(screen, title='Installation failed', text='Debootstrap failed.', buttons=[('View log', 'log'), ('Try again', 'again'), ('Cancel', 'cancel')], width=40) if choice == 'log': screen.suspend() call(['less', rootdir + '/root/debootstrap.log']) screen.resume() continue if choice == 'again': break if choice == 'cancel': raise Abort('debootstrap failed') p = ProgressMessage(screen, title='Please wait', text='Undoing previous try ...') umount_in_bootenv() # debootstrap mounts required FS for us call(['zfs', 'rollback', '-r', bootenv + '@empty']) def apt_get_update(log): global apt_cache progress = ProgressBar(screen, title='Updating APT cache', text='Please wait ...') read, write = os.pipe() pid = os.fork() if pid == 0: os.close(read) os.dup2(log, 1) os.dup2(log, 2) try: os.execl('/usr/sbin/chroot', 'chroot apt-get update', rootdir, '/usr/bin/apt-get', '-o', 'APT::Status-Fd=' + str(write), 'update') except EnvironmentError as e: os.write(log, str(e)) sys._exit(e.errno) else: # Parent os.close(write) pr = os.fdopen(read, 'r', 1) while True: line = pr.readline() if not line: break try: (status, f, percent, text) = line.split(':') except: os.write(log, 'Installer failed to parse line "{}"'.format(line)) continue progress.progress = float(percent) progress.text = text status = os.wait()[1] if 0 == status: progress.text = "Reading files ..." apt_cache = apt.Cache(rootdir=rootdir) return os.WEXITSTATUS(status) def apt_get_install(log): global profile global profiles progress = ProgressBar(screen, title='Installing packages', text='Please wait ...') packages = [pkg for pkg in profiles[profile]["packages"] if pkg in apt_cache] missed = [pkg for pkg in profiles[profile]["packages"] if not pkg in apt_cache] for p in missed: os.write(log, "WARNING: missing package `{}'\n".format(p)) read, write = os.pipe() pid = os.fork() if pid == 0: os.close(read) os.dup2(log, 1) os.dup2(log, 2) try: os.execl('/usr/sbin/chroot', 'chroot apt-get install', rootdir, '/usr/bin/apt-get', '-y', '-o', 'APT::Status-Fd=' + str(write), 'install', *packages) except EnvironmentError as e: os.write(log, str(e)) sys._exit(e.errno) else: # Parent os.close(write) pr = os.fdopen(read, 'r', 1) while True: line = pr.readline() if not line: break try: (status, f, percent, text) = line.split(':') percent = float(percent) except: os.write(log, 'Installer failed to parse line "{}"'.format(line)) continue if status == 'dlstatus': percent = percent / 2.0 elif status == 'pmstatus': percent = 50.0 + percent / 2.0 else: continue progress.progress = math.ceil(percent) progress.text = text status = os.wait()[1] # apt-get sometimes reports less than 100% at the end progress.progress = 100 return os.WEXITSTATUS(status) def install_profile(): global profile while True: if not profile: return logfile = rootdir + '/root/dyson-install.log' log = os.open(logfile, os.O_WRONLY + os.O_CREAT) rc = apt_get_update(log) if 0 == rc: rc = apt_get_install(log) os.close(log) if 0 == rc: return while True: choice = ButtonChoiceWindow(screen, title='Installation failed', text='Something went wrong. This is not fatal, because minimal system' ' is installed. You can search log for failures, fix them' ' (if it was, for example, a network failure) and try again.' ' Or give up and stay with minimal system.', buttons=[('View log', 'log'), ('Try again', 'again'), ('Give up', 'skip')], width=50) if choice == 'log': screen.suspend() call(['less', logfile]) screen.resume() continue if choice == 'again': break if choice == 'skip': profile = None break def umount_in_bootenv(): '''unmount all FS mounted in BE on final cleanup. These FS also may be left after interruption or debootstrap failure''' global bootenv global rootdir if not bootenv: return fslist = ['/dev/fd', '/proc', '/devices', '/home'] for fs in fslist: call(['umount', rootdir + fs], stdout=PIPE, stderr=PIPE) def in_bootenv(cmd): chroot = ['chroot', rootdir] chroot += cmd pobject = Popen(chroot, stderr=PIPE, stdout=PIPE) out, err = pobject.communicate() return (pobject.returncode, out, err) def write_vfstab(): vfstab=''' #device device mount FS fsck mount mount #to mount to fsck point type pass at boot options # fd - /dev/fd fd - no - swap - /tmp tmpfs - yes - /dev/zvol/dsk/{rpool}/swap - - swap - no - '''.format(rpool=rpool) try: f = open(rootdir + '/etc/vfstab', 'w') print(vfstab, file=f) f.close() except: pass # http://stackoverflow.com/questions/2532053/validate-a-hostname-string def isValidNodename(hostname): if len(hostname) > 63: return False return re.match(r'^(?!-)[a-z0-9\-]+(? a, c0t4d0s3 -> d''' try: n = s[s.rfind('s')+1:] return chr(int(n) + ord('a')) except: # random return 'a' def configure_grub(): global rslice # write grub menu for references: try: menu = open(rootdir + '/boot/grub/menu.lst', 'w') print('# This file is just for references. It is not used by GRUB', file=menu) print('# Actual menu.lst used by GRUB is {}/boot/grub/menu.lst'.format(rpool), file=menu) print('# where {} is the name of root ZFS pool'.format(rpool), file=menu) print(grub_top, file=menu) print(grub_entry.format(rpool=rpool, bootenv=bootenv, partition=1, slice='a'), file=menu) menu.close() except: pass # Zpool was not created during installation, trying to guess slice name: if not rslice: slices = get_zpool_slice(rpool) if len(slices) == 1 and re.match(r'^\w+s\d+$', slices[0]): rslice = slices[0] items = [] default = None text = '' if rslice: text += 'Installing GRUB on the master boot sector (MBR) ' text += 'overrides any boot manager currently installed: ' text += 'the system will always boot the GRUB in the solaris partition' items.append(('Install GRUB to MBR', 'mbr')) items.append(('Install GRUB to a partition only', 'partition')) items.append(('Only update GRUB menu', 'menu')) if rpool in zpools: # installing on existing pool default = 'menu' else: default = 'mbr' else: text += 'It looks like root ZFS pool "{rpool}" was not created during this installation, '.format(rpool=rpool) text += 'so installing GRUB is not supported. ' text += 'Laterly you can consult {bootenv}/boot/grub/menu.lst for details.'.format(bootenv=bootenv) default = 'skip' items.append(('Do nothing', 'skip')) button, install = ListboxChoiceWindow(screen, title='Configure GRUB', text=text, items=items, buttons=['Ok'], width=50, default=default) if install == 'skip': return progress = ProgressMessage(screen, title='Configuring GRUB') installgrub_cmd = ['/usr/sbin/installgrub'] if install == 'mbr': progress.text = 'Installing GRUB to the master boot record ...' installgrub_cmd += ['-f', '-m'] elif install == 'menu': # dry run, just to get partition number progress.text = 'Updating GRUB menu ...' installgrub_cmd += ['-n'] else: progress.text = 'Installing GRUB to partition ...' installgrub_cmd += [rootdir+'/boot/grub/stage1', rootdir+'/boot/grub/stage2', '/dev/rdsk/' + rslice] installgrub = Popen(installgrub_cmd, stdout=PIPE, stderr=PIPE) out, err = installgrub.communicate() if installgrub.returncode != 0: ButtonChoiceWindow(screen, title='Error', text='Installing of GRUB failed: '+err, buttons=['Ok'], width=60) m = re.search(r'stage1 written to partition (\d)', out) if m: partition = m.group(1) else: partition = 0 # Random rpool_path = '/mnt/{rpool}/{rpool}'.format(rpool=rpool) try: if not exists(rpool_path + '/boot'): os.mkdir(rpool_path + '/boot') if not exists(rpool_path + '/boot/grub'): os.mkdir(rpool_path + '/boot/grub') if not exists(rpool_path + '/boot/grub/bootsign'): os.mkdir(rpool_path + '/boot/grub/bootsign') if not exists(rpool_path + '/boot/grub/bootsign/pool_'+rpool): open(rpool_path + '/boot/grub/bootsign/pool_'+rpool, 'w').close() if not exists(rpool_path + '/boot/grub/menu.lst'): menu = open(rpool_path + '/boot/grub/menu.lst', 'w') print(grub_top, file=menu) menu.close() re_bootfs = re.compile(r'^\s*bootfs\s+' + bootenv) have_this_bootenv = False menu = open(rpool_path + '/boot/grub/menu.lst', 'r+') for line in menu: if re_bootfs.match(line): have_this_bootenv = True break if not have_this_bootenv: print(grub_entry.format(rpool=rpool, bootenv=bootenv, partition=partition, slice=slice4grub(rslice)), file=menu) menu.close() except EnvironmentError as e: text = 'Failed to write GRUB configuration: ' if e.filename: text += e.filename + ': ' text += e.strerror ButtonChoiceWindow(screen, title='Error', text=text, buttons=['Ok'], width=60) def cleanup(destroy_bootenv=False): global bootenv global rootdir progress = ProgressMessage(screen, title='Cleaning up') if bootenv: umount_in_bootenv() call(['umount', rootdir], stdout=PIPE, stderr=PIPE) if destroy_bootenv: progress.text='Destroying {}, please wait ...'.format(bootenv) call(['zfs', 'destroy', '-r', bootenv], stdout=PIPE, stderr=PIPE) else: progress.text = 'Adjusting boot environment "{}" ...'.format(bootenv) call(['zfs', 'destroy', '-r', bootenv + '@empty'], stdout=PIPE, stderr=PIPE) call(['zfs', 'set', 'canmount=noauto', bootenv], stdout=PIPE, stderr=PIPE) call(['zfs', 'set', 'mountpoint=/', bootenv], stdout=PIPE, stderr=PIPE) try: os.rmdir(rootdir) except: pass bootenv = None progress.text = 'Exporting ZFS pools ...' for pool in imported_zpools: call(['zpool', 'export', '-f', pool], stdout=PIPE, stderr=PIPE) def goodbye(): ButtonChoiceWindow(screen, title='Success', text='The Dyson system is successfully installed and configured. ' 'Hopefully, it will boot :-)', buttons=['Reboot']) screen = SnackScreen() while True: try: screen.pushHelpLine('F2 - switch to console, F1 - switch back to the installer') welcome() screen.pushHelpLine(' ') find_hdds() configure_rpool() configure_zfs() choose_profile() install_minimal() configure_bootenv() configure_grub() goodbye() cleanup() break except Abort as e: choice = ButtonChoiceWindow(screen, title='Cancel installation', text='Installation is canceled. Would do you like to start it again or reboot?', buttons=[('Restart', 'restart'), ('Reboot', 'reboot')], width=50) cleanup(destroy_bootenv=True) if choice == 'reboot': break except NoDisks as e: ButtonChoiceWindow(screen, title='Error', text='No disks found on the system. ' 'Installation of Dyson is not possible.', buttons=['Reboot']) break except: ButtonChoiceWindow(screen, title='FATAL ERROR', text='The installer has badly failed:\n'+traceback.format_exc(), width=70, buttons=['Restart the installer']) cleanup(destroy_bootenv=True) screen.finish() sys.exit(0)