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 os
import struct

bitmaps = ["sidepane.ucl", "warning.ucl", "installing.ucl", \
           "formatting.ucl", "copying.ucl", "flashing.ucl"]

flashfiles = [("ildrcfg ", 2, 0, 0, "iloader.cfg.ucl"), \
              ("iloader ", 2, 0, 0, "iloader.embiosapp.ucl"), \
              ("umsboot ", 2, 0, 0, "umsboot-ipodclassic.ucl"), \
              ("embiosldr", 12, 8, 0, "embiosldr-ipodclassic.bin"), \
              ("embios  ", 2, 0, 0, "embios-ipodclassic.ucl")]

firstinstfiles = [(2, "/iLoader/iLoader.cfg", "../iloader/themes/ipodclassic-default/iLoader/iloader.cfg", 1), \
                  (2, "/iLoader/theme.ucl", "../iloader/themes/ipodclassic-default/iLoader/theme.ucl", 2)]

commonfiles = [(2, "/iLoader/NORFlash.bak", -2, 10)]

if len(sys.argv) > 4 and sys.argv[4] != "-":
    pathlen = len(sys.argv[4])
    for d in os.walk(sys.argv[4]):
        prefix = d[0].replace("\\", "/")[pathlen:] + "/"
        for dir in d[1]:
            if dir != ".svn":
                firstinstfiles.append((1, prefix + dir, 1))
        for f in d[2]:
            if not prefix.find("/.svn/") > -1:
                firstinstfiles.append((2, prefix + f, d[0] + "/" + f, os.path.getsize(d[0] + "/" + f) / 100000 + 1))

file = open(sys.argv[1], "rb")
installer = file.read()
file.close()
installer = installer.ljust((len(installer) + 3) & ~3)

for f in bitmaps:
  file = open("build/" + f, "rb")
  fdata = file.read()
  file.close()
  fdata = fdata.ljust((len(fdata) + 3) & ~3)
  installer = installer + struct.pack("<I", len(fdata)) + fdata

statusfirst = 0
statuscommon = 0
scriptsize = 20

for f in flashfiles:
  scriptsize = scriptsize + 4
  if (f[1] & 4) == 0: scriptsize = scriptsize + 8
  if f[3] == 0: scriptsize = scriptsize + 8

for f in firstinstfiles:
  scriptsize = scriptsize + 12
  if f[0] > 1: scriptsize = scriptsize + 8

for f in commonfiles:
  scriptsize = scriptsize + 12
  if f[0] > 1: scriptsize = scriptsize + 8

scriptsize = ((len(installer) + scriptsize + 15) & ~15) - len(installer)

filedata = ""
flash = ""
for f in flashfiles:
  flash = flash + struct.pack("BBBB", f[3], f[1], f[2], 1)
  if f[3] == 0:
    file = open("flashfiles/" + f[4], "rb")
    fdata = file.read()
    file.close()
    flash = flash + struct.pack("<II", scriptsize + len(filedata), len(fdata))
    filedata = filedata + fdata.ljust((len(fdata) + 15) & ~15)
  if (f[1] & 4) == 0: flash = flash + f[0]

firstinstall = ""
for f in firstinstfiles:
  size = 0
  nameptr = scriptsize + len(filedata)
  filedata = filedata + (f[1] + "\0").ljust((len(f[1]) + 16) & ~15)
  if f[0] == 1:
    firstinstall = firstinstall + struct.pack("<III", f[0], nameptr, f[2])
    statusfirst = statusfirst + f[2]
  else:
    if type(f[2]) == str:
      file = open(f[2], "rb")
      fdata = file.read()
      file.close()
      ptr = scriptsize + len(filedata)
      size = len(fdata)
      filedata = filedata + fdata.ljust((len(fdata) + 15) & ~15)
    else:
      ptr = f[2]
    firstinstall = firstinstall + struct.pack("<IIiII", f[0], nameptr, ptr, size, f[3])
    statusfirst = statusfirst + f[3]

common = ""
for f in commonfiles:
  size = 0
  nameptr = scriptsize + len(filedata)
  filedata = filedata + (f[1] + "\0").ljust((len(f[1]) + 16) & ~15)
  if f[0] == 1:
    common = common + struct.pack("<III", f[0], nameptr, f[2])
    statuscommon = statuscommon + f[2]
  else:
    if type(f[2]) == str:
      file = open("../iloader/themes/default/iloader/" + f[2], "rb")
      fdata = file.read()
      file.close()
      ptr = scriptsize + len(filedata)
      size = len(fdata)
      filedata = filedata + fdata.ljust((len(fdata) + 15) & ~15)
    else:
      ptr = f[2]
    common = common + struct.pack("<IIiII", f[0], nameptr, ptr, size, f[3])
    statuscommon = statuscommon + f[3]

script = flash + struct.pack("<IIII", 0, len(flash) + 16 + len(firstinstall), \
                             statusfirst, statusfirst + statuscommon) \
       + firstinstall + common + struct.pack("<I", 0)
file = open(sys.argv[2], "wb")
file.write(installer + script.ljust(scriptsize) + filedata)
file.close()