# Miori Lorenzo Dec 2012
#
# Samsung YP-F3, YP-U7, YP-W1 Firmware Extractor
#
# ~~~~~ Some fields are still unknown ~~~~~~
# 

import struct, os, sys

romFile = sys.argv[1]

outFolder = "%s_extracted/" % romFile

os.mkdir(outFolder)
f = open(romFile, "rb")
f.seek(0, 2)
original_size = f.tell()
f.seek(0)
def getValue(bytes):
    
    return struct.unpack("I", bytes)[0]

files_no = getValue(f.read(4))
total_size = 0
print "Total files: %i"%files_no
f.seek(0x14)

for i in xrange(files_no):
    
    fn = f.read(52).replace("\x00", "")
    
    print "File name: %s"%fn
    
    fsize = getValue(f.read(4))
    total_size += fsize
    
    print "File size: 0x%x bytes"%fsize
    
    fpos = getValue(f.read(4))
    
    print "File pos: 0x%x"%fpos
    
    print "Unknown field: 0x%x bytes"%(getValue(f.read(4)))
    
    pos2continue = f.tell()

    f.seek(fpos)
    
    fout = open(outFolder + fn, "wb", 1024)
    
    fout.write(f.read(fsize))
    
    fout.close()
    
    f.seek(pos2continue)

f.close()

print "Original file size: %i"%original_size
print "Total extracted size: %i"%total_size
print "Header & Stuff %i"%abs(original_size-total_size)