root/morphix/trunk/morph-scripts/isomorph.pl

#!/usr/bin/perl

# 
# isomorph
#

# Description:
# 
# lets you morph a currently downloaded iso
# by adding a new mini- or mainmodule, debian package, shell script, copy-on-boot file,
# removing these, getting, or listing them
#

###
#
# See isomorph -h for usage
#
###

use File::Temp qw/ tempdir /;
use File::Copy;

$mode = "";
$type = "";
$file = "";
$path = "";

$xml = 0;

$oldmountpoint = "";
$newmountpoint = "";

$oldiso = "";
$newiso = "";

sub type_valid {
    $t = @_[0];
    if ($t eq "main" ||
        $t eq "mini" ||
        $t eq "deb" ||
        $t eq "exec" ||
        $t eq "base" ||
        $t eq "copy" ||
        $t eq "boot") {
        return 1;
    }
    return 0;
}

###
#
# Option handling stuff
#
###

$option = $ARGV[0];
$type = $ARGV[1];

if ($option eq "--add") {
    if ($ARGV[4] ne "" && type_valid($type) == 1) {
        $mode = "add";
        $file = $ARGV[2];
        if ($type eq "copy") {
            $path = $ARGV[3];
            $oldiso = $ARGV[4];
            $newiso = $ARGV[5];
        }
        else {
            $oldiso = $ARGV[3];
            $newiso = $ARGV[4];
        }
    }
    else {
        $mode = "";
    }
}

if ($option eq "--get") {
    if ($ARGV[4] ne "" && type_valid($type) == 1) {
        $mode = "get";
        $oldiso = $ARGV[3];
        $path = $ARGV[4];
    }
    else {
        if ($ARGV[3] ne "") {
            $mode = "get";
            $type = "all";
            $oldiso = $ARGV[2];
            $path = $ARGV[3];
        }
        else {
            $mode = "";
        }
    }
}

if ($option eq "--del" || $option eq "--del-all") {
    $mode = "del";

    if ($ARGV[2] eq "" && type_valid($type) == 0) {
        $mode = "";
    } 
    elsif ($ARGV[4] ne "" && type_valid($type) == 1) {
        $file = $ARGV[2];
        $oldiso = $ARGV[3];
        $newiso = $ARGV[4];
    } 
    elsif ($ARGV[3] ne "" && type_valid($type) == 1) {
        $mode = "del-all";
        $oldiso = $ARGV[2];
        $newiso = $ARGV[3];
    }
    else {
        $mode = "del-all";
        $oldiso = $ARGV[1];
        $newiso = $ARGV[2];
        $type = "all";
    }
}

if ($option eq "--list" || $option eq "--list-xml") {
    if ($option eq "--list-xml") {
        $xml = 1;
    }
    if ($ARGV[2] ne "" && type_valid($type) == 1) {
        $mode = "list";
        $oldiso = $ARGV[2];
    }
    else { # if only one argument, type is the iso
        if ($ARGV[1] ne "") {
            $mode = "list";
            $type = "all";
            $oldiso = $ARGV[1];
        }
        else {
            $mode = "";
        }
    }
}

###
#
# Mode handling stuff
#
###

if ($mode eq "") {
        print("Usage: isomorph [OPTION]
All isomorph options are of 4 types: list, add, get and delete
'list' and 'get' are actions on a single iso, 
'add' and 'delete' output a modified iso and leave the old iso intact.
With these 4 options, many types of actions are possible:

list options:
[--list ISO]\t\t\t\tlist all morphix files on iso
[--list-xml ISO]\t\t\tlist all morphix files in XML format
[--list main ISO]\t\t\tlist mainmodules
[--list mini ISO]\t\t\tlist minimodules
[--list deb ISO]\t\t\tlist debian packages
[--list exec ISO]\t\t\tlist scripts
[--list copy ISO]\t\t\tlist copy-on-boot files

add options:
[--add base BASEMODULE OLDISO NEWISO]\t\treplace basemodule
[--add main MAINMODULE OLDISO NEWISO]\t\tadd mainmodule to oldiso
[--add mini MINIMODULE OLDISO NEWISO]\t\tadd minimodule to oldiso
[--add deb  DEBPACKAGE OLDISO NEWISO]\t\tadd package to oldiso
[--add exec SHELLSCRIPT OLDISO NEWISO]\t\tadd script to oldiso
[--add copy FILE PATH OLDISO NEWISO]\t\tadd file to oldiso in path
[--add boot \"Boot Option\" OLDISO NEWISO]\tadd boot option to oldiso

get options:
[--get ISO DIRECTORY]\t\t\tcopy all morphix files from iso to dir
[--get main ISO DIRECTORY]\t\tcopy mainmodules from iso to directory
[--get mini ISO DIRECTORY]\t\tcopy minimodules
[--get deb ISO DIRECTORY]\t\tcopy debian packages
[--get exec ISO DIRECTORY]\t\tcopy scripts
[--get copy ISO DIRECTORY]\t\tcopy copy-on-boot files

delete options:
[--del-all OLDISO NEWISO]\t\t\tdelete all morphix files from ISO
[--del-all main OLDISO NEWISO]\t\t\tdelete all mainmodules
[--del-all mini OLDISO NEWISO]\t\t\tdelete all minimodules
[--del-all deb OLDISO NEWISO]
[--del-all exec OLDISO NEWISO]
[--del-all copy OLDISO NEWISO]
[--del main MAINMODULE OLDISO NEWISO]\t\tdelete mainmodule from oldiso
[--del mini MINIMODULE OLDISO NEWISO]\t\tdelete minimodule from oldiso
[--del deb DEBPACKAGE OLDISO NEWISO]\t\tdelete package from oldiso
[--del exec SHELLSCRIPT OLDISO NEWISO]
[--del copy FILE OLDISO NEWISO]
[--del boot \"Boot Option\" OLDISO NEWISO]\tdelete boot option from oldiso
\n");
        exit(1);
}

sub makenewiso {
    $mode = @_[0];
    if ($mode eq "add" || $mode eq "del" || $mode eq "del-all") {
        return 1;
    }
    return 0;
}

sub dostuff {
    if ($mode eq "list") {
        if ($xml == 1) {
            listxmlheader();
        }
        if ($type eq "main" || $type eq "all") {
            domain();
        }
        if ($type eq "mini" || $type eq "all") {
            domini();
        }
        if ($type eq "deb" || $type eq "all") {
            dodeb();
        }
        if ($type eq "exec" || $type eq "all") {
            doexec();
        }
        if ($type eq "copy" || $type eq "all") {
            docopy();
        }
        if ($mode eq "list" && $xml == 1) {
            listxmlfooter();
        }
    }
    if ($mode eq "get") {
        if ($type eq "main" || $type eq "all") {
            domain();
        }
        if ($type eq "mini" || $type eq "all") {
            domini();
        }
        if ($type eq "deb" || $type eq "all") {
            dodeb();
        }
        if ($type eq "exec" || $type eq "all") {
            doexec();
        }
        if ($type eq "copy" || $type eq "all") {
            docopy();
        }
    }
            

    if (makenewiso($mode)) {
        print $newmountpoint;
        print $oldmountpoint;
        print ("Copying old iso...\n");
        system("cp -a $oldmountpoint/* $newmountpoint");
        print ("Busy making new iso...\n");
    }
    if ($mode eq "add") {
        if ($type eq "base") {
            if ( -e $file) {
                system("cp $file $newmountpoint/base/morphix");
            } else {
                isodie("Basemodule $file not found!");
            }
        }
        if ($type eq "mini") {
            if ( -e $file) {
                system("cp $file $newmountpoint/minimod/");
            } else { 
                isodie("Minimodule $file not found!"); 
            }
        }
        if ($type eq "main") {
            if ( -e $file) {
                system("cp $file $newmountpoint/mainmod/");
            } else { 
                isodie("Mainmodule $fie not found!"); 
            }
        }
        if ($type eq "deb") {
            if ( -e $file) {
                system("cp $file $newmountpoint/deb/");
            } else { 
                isodie("Debian package $file not found!"); 
            }
        }
        if ($type eq "exec") {
            if ( -e $file) {
                system("cp $file $newmountpoint/exec/");
            } else { 
                isodie("Shell script $file not found!"); 
            }
        }
        if ($type eq "copy") {
            if ( -e $file) {
                system("mkdir -p $newmountpoint/copy/$path");
                if (-e "$newmountpoint/copy/$path") {
                    system("cp $file $newmountpoint/copy/$path");
                }
                else {
                    isodie("Can't make directory $newmountpoint/copy/$path");
                }
            } else { 
                isodie("File $file to be placed in /copy not found!"); 
            }
        }
        if ($type eq "boot") {
            if (-e "$newmountpoint/base/boot.img") {
                isodie("Old boot.img not yet supported for adding removing boot options"); 
            }
            elsif (-e "$newmountpoint/boot/grub/iso9660_stage1_5" || -e "$newmountpoint/boot/grub/stage2_eltorito") {
                $dir = "$oldmountpoint/boot/grub";
                opendir(DIR, $dir);
                @file2 = grep !/^\.\.?$/, readdir DIR;
                foreach $item (@file2) {
                    if ($item =~ m".lst") {
                            system ("sed '/kernel/s/\$/ $file/' $oldmountpoint/boot/grub/$item >$newmountpoint/boot/grub/$item");
                        }
                    }
                closedir(DIR);  
                }
            else {
                isodie("Can't add boot option unable to find boot option");
                }
            }
    }
    if ($mode eq "del") {
        if ($type eq "mini") {
            if (! -e "$newmountpoint/minimod/$file") {
                isodie("Minimodule $file not found!");
            }
            system("rm $newmountpoint/minimod/$file");
            if (-e "$newmountpoint/minimod/$file") {
                isodie("Minimodule $file not deleteable: $!");
            }
        }
        if ($type eq "main") {
            if (! -e "$newmountpoint/mainmod/$file") {
                isodie("Mainmodule $file not found!");
            }
            system("rm $newmountpoint/mainmod/$file");
            if (-e "$newmountpoint/mainmod/$file") {
                isodie("Mainmodule $file not deleteable: $!");
            }
        }
        if ($type eq "deb") {
            if (! -e "$newmountpoint/deb/$file") {
                isodie("Debian package $file not found!");
            }
            system("rm $newmountpoint/deb/$file");
            if (-e "$newmountpoint/deb/$file") {
                isodie("Debian package $file not deleteable: $!");
            }
        }
        if ($type eq "exec") {
            if (! -e "$newmountpoint/exec/$file") {
                isodie("Script $file not found!");
            }
            system("rm $newmountpoint/exec/$file");
            if (-e "$newmountpoint/exec/$file") {
                isodie("Script $file not deleteable: $!");
            }
        }
        if ($type eq "copy") {
            if (! -e "$newmountpoint/copy/$file") {
                isodie("File $file not found!");
            }
            system("rm $newmountpoint/copy/$file");
            if (-e "$newmountpoint/copy/$file") {
                isodie("File $file not deleteable: $!");
            }
        }
        if ($type eq "boot") {
            if (-e "$newmountpoint/base/boot.img") {
                isodie("Old boot.img not yet supported for adding removing boot options"); 
            }
            elsif (-e "$newmountpoint/boot/grub/iso9660_stage1_5" || -e "$newmountpoint/boot/grub/stage2_eltorito") {
                $dir = "$oldmountpoint/boot/grub";
                opendir(DIR, $dir);
                @file2 = grep !/^\.\.?$/, readdir DIR;
                foreach $item (@file2) {
                    if ($item =~ m".lst") {
                            system ("sed '/kernel/s/$file//g' $oldmountpoint/boot/grub/$item >$newmountpoint/boot/grub/$item");
                            system ("cat $newmountpoint/boot/grub/$item");
                        }
                    }
                closedir(DIR);  
                }
            else {
                isodie("Can't remove boot option unable to find boot option");
                }
            }   
    }
    if ($mode eq "del-all") {
        if ($type eq "mini" || $type eq "all") {
            system("rm -r $newmountpoint/minimod");
            if (-e "$newmountpoint/minimod") {
                isodie("Minimodule dir not deleteable: $!");
            }
            mkdir("$newmountpoint/minimod");
        }
        if ($type eq "main" || $type eq "all") {
            system("rm -r $newmountpoint/mainmod");
            if (-e "$newmountpoint/mainmod") {
                isodie("Mainmodule dir not deleteable: $!");
            }
            mkdir("$newmountpoint/mainmod");
        }
        if ($type eq "deb" || $type eq "all") {
            system("rm -r $newmountpoint/deb");
            if (-e "$newmountpoint/deb") {
                isodie("Debian package dir not deleteable: $!");
            }
            mkdir("$newmountpoint/deb");
        }
        if ($type eq "exec" || $type eq "all") {
            system("rm -r $newmountpoint/exec");
            if (-e "$newmountpoint/exec") {
                isodie("Exec dir not deleteable: $!");
            }
            mkdir("$newmountpoint/exec");
        }
        if ($type eq "copy" || $type eq "all") {
            system("rm -r $newmountpoint/copy");
            if (-e "$newmountpoint/copy") {
                isodie("Copy dir not deleteable: $!");
            }
            mkdir("$newmountpoint/copy");
        }
        
    }
    if (makenewiso($mode)) {    
        # do those md5sums
        system("find $newmountpoint/ -type f -print0 | xargs -0 md5sum > $newmountpoint/md5sums.txt");
        
        # finally make the damn thing
        # We currently have used 2 types of CD bootloaders: 
        # syslinux or grub with floppy emulation, of which we only have
        # to boot using the /base/boot.img file
        # and grub no-floppy-emulation booting, of which there are two
        # methods for booting: using the iso9660 stage 1.5 grub image or
        # the stage2 eltorito image.
        # Floppy emulation is deprecated in favour of the use of the
        # iso9660 stage 1.5 grub image, but is as of now the most common

        if (-e "$newmountpoint/base/boot.img") {
            system("mkisofs -pad -l -r -J -v -V \"Morphix Live CD\" -b base/boot.img -c base/boot.cat -hide-rr-moved -o $newiso $newmountpoint");
        }
        elsif (-e "$newmountpoint/boot/grub/iso9660_stage1_5") {
            system("mkisofs -pad -l -r -J -v -V \"Morphix LiveCD\" -b boot/grub/iso9660_stage1_5 -c base/boot.cat -no-emul-boot -boot-load-size 4 -boot-info-table -hide -rr -moved -o $newiso $newmountpoint");
        }
        elsif (-e "$newmountpoint/boot/grub/stage2_eltorito") {
            system("mkisofs -pad -l -r -J -v -V \"Morphix LiveCD\" -b boot/grub/stage2_eltorito -c base/boot.cat -no-emul-boot -boot-load-size 4 -boot-info-table -hide -rr -moved -o $newiso $newmountpoint");
        }
    }
}

sub checkiso {
    if (-e "@_[0]/mainmod" && -e "@_[0]/mainmod" && -e "@_[0]/base") {
        return 1;
    }
    return 0;
}

sub isodie {
    umountoldiso();
    removetempdirs();
    die "Error: @_[0]";
}

sub listxmlheader {
    print("<iso>\n");
}

sub listxmlfooter {
    print("</iso>\n");
}

sub domain {
    $dir = "$oldmountpoint/mainmod";
    opendir(DIR, $dir);
    if (DIR && $xml == 1) {
        @file2 = grep !/^\.\.?$/, readdir DIR;
        foreach $item (@file2) {
            print "\t<main>$item</main>\n";
        }
        closedir(DIR);
    }
    elsif (DIR) {
        if ($mode eq "list") {
            
            print("Mainmodules in $oldiso:\n");
            @file2 = grep !/^\.\.?$/, readdir DIR;
            print "@file2\n";
            closedir(DIR);
        }
        else {
            @file2 = grep !/^\.\.?$/, readdir DIR;
            foreach $file3 (@file2) {
                system("cp $dir/$file3 $path/");
            }
            closedir(DIR);
        }
    }
}

sub domini {
    $dir = "$oldmountpoint/minimod";
    opendir(DIR, $dir);
    if (DIR && $xml == 1) {
        @file2 = grep !/^\.\.?$/, readdir DIR;
        foreach $item (@file2) {
            print "\t<mini>$item</mini>\n";
        }
        closedir(DIR);
    }
    elsif (DIR) {
        if ($mode eq "list") {
            print("Minimodules in $oldiso:\n");
            @file2 = grep !/^\.\.?$/, readdir DIR;
            print "@file2\n";
            closedir(DIR);
        }
        else {
            @file2 = grep !/^\.\.?$/, readdir DIR;
            foreach $file3 (@file2) {
                system("cp $dir/$file3 $path/");
            }
            closedir(DIR);
        }
    }
}

sub dodeb {
    $dir = "$oldmountpoint/deb";
    opendir(DIR, $dir);
    if (DIR && $xml == 1) {
        @file2 = grep !/^\.\.?$/, readdir DIR;
        foreach $item (@file2) {
            print "\t<deb>$item</deb>\n";
        }
        closedir(DIR);
    }
    elsif (DIR) {
        if ($mode eq "list") {
            print("Debian packages in $oldiso:\n");
            @file2 = grep !/^\.\.?$/, readdir DIR;
            print "@file2\n";
            closedir(DIR);
        }
        else {
            @file2 = grep !/^\.\.?$/, readdir DIR;
            foreach $file3 (@file2) {
                system("cp $dir/$file3 $path/");
            }
            closedir(DIR);
        }
    }
}

sub doexec {
    $dir = "$oldmountpoint/exec";
    opendir(DIR, $dir);
    if (DIR && $xml == 1) {
        @file2 = grep !/^\.\.?$/, readdir DIR;
        foreach $item (@file2) {
            print "\t<exec>$item</exec>\n";
        }
        closedir(DIR);
    }
    elsif (DIR) {
        if ($mode eq "list") {
            print("Shell scripts in $oldiso:\n");
            @file2 = grep !/^\.\.?$/, readdir DIR;
            print "@file2\n";</