Subversion Repositories freemyipod

#!/usr/bin/env python

#

#

#    Copyright 2010 TheSeven

#

#

#    This file is part of emBIOS.

#

#    emBIOS is free software: you can redistribute it and/or

#    modify it under the terms of the GNU General Public License as

#    published by the Free Software Foundation, either version 2 of the

#    License, or (at your option) any later version.

#

#    emBIOS is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

#    See the GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with emBIOS.  If not, see <http://www.gnu.org/licenses/>.

#

#

import sys

import libembiosbootcfg

from optparse import *

parser = OptionParser("usage: %prog [options] <infile> <outfile>")

filegroup = OptionGroup(parser, "Booting from a file",

                        "Use these options to configure emBIOS to try "

		        "booting from a file on a FAT32 partition")

filegroup.add_option("--file", help = "Boot from FILE")

filegroup.add_option("--file-compressed", action = "store_true", default = False,

                     help = "Specify this if FILE is compressed")

filegroup.add_option("--file-run-from", type = "int", metavar = "ADDR",

                     help = "Make sure FILE is executed from memory address ADDR")

parser.add_option_group(filegroup)

flashgroup = OptionGroup(parser, "Booting from a boot flash image",

                         "Use these options to configure emBIOS to try "

                         "booting from an image located in the boot flash")

flashgroup.add_option("--flash", metavar = "NAME", help = "Boot from flash image NAME")

flashgroup.add_option("--flash-compressed", action = "store_true", default = False,

                      help = "Specify this if the image is compressed")

flashgroup.add_option("--flash-run-from", type = "int", metavar = "ADDR",

                      help = "Make sure the image is executed from memory address ADDR")

parser.add_option_group(flashgroup)

mmapgroup = OptionGroup(parser, "Booting from a memory region",

                        "Use these options to configure emBIOS to try "

                        "booting from a memory location, such as an embedded "

			"app or an app located on a memory-mapped flash")

mmapgroup.add_option("--mmap-addr", metavar = "ADDR", help = "Boot from memory location ADDR")

mmapgroup.add_option("--mmap-size", metavar = "SIZE",

                     help = "Specifies the size of the executable at ADDR in bytes")

mmapgroup.add_option("--mmap-compressed", action = "store_true", default = False,

                     help = "Specify this if the executable is compressed")

mmapgroup.add_option("--mmap-run-from", type = "int", metavar = "COPYADDR",

                     help = "Copy the executable to COPYADDR before executing it")

parser.add_option_group(mmapgroup)

(options, args) = parser.parse_args()

if len(args) != 2: parser.error("incorrect number of arguments")

if (options.mmap_addr and not options.mmap_size) or \

   (not options.mmap_addr and options.mmap_size):

    parser.error("either none or both of --mmap-addr and --map-size must be specified")

file = open(args[0], "rb")

data = file.read()

file.close()

config = {"reset": True}

if options.file:

    config["tryfile"] = True

    config["filename"] = options.file

    if options.file_compressed: config["filecomp"] = True

    if options.file_run_from:

        config["filecopy"] = True

        config["filedest"] = options.file_run_from

if options.flash:

    config["tryflash"] = True

    config["flashname"] = options.flash

    if options.flash_compressed: config["flashcomp"] = True

    if options.flash_run_from:

        config["flashcopy"] = True

        config["flashdest"] = options.flash_run_from

if options.mmap_addr:

    config["trymmap"] = True

    config["mmapaddr"] = options.mmap_addr

    config["mmapsize"] = options.mmap_size

    if options.mmap_compressed: config["mmapcomp"] = True

    if options.mmap_run_from:

        config["mmapcopy"] = True

        config["mmapdest"] = options.mmap_run_from

data = libembiosbootcfg.configure(data, **config)

file = open(args[1], "wb")

file.write(data)

file.close()