root/hodgestar/PythonCode/NokiaArcExtractor/arc-extract.py

"""Nokia arc file extractor.

   Copyright 2008 Simon Cross <hodgestar+arc@gmail.com>
   Copyright 2008 Mark Ferry <gnomeza+arc@gmail.com>

   With some help from:
   http://www.opensource.or.ke/index.php?option=com_content&task=view&id=81&Itemid=1

   Notes:
     - <Gnome> Hodgestar: did you try running /usr/bin/strings with --encoding=l ?
       <Gnome> produces good lists of file paths
   """

import zlib
import os
import sys

def dump_header(data, recovery_folder, max_offset=256):
    """Dump header."""
    offset = 0
    while offset <= max_offset:
        try:
            decoder = zlib.decompressobj()
            msg = decoder.decompress(data[offset:])
        except zlib.error:
            offset += 1
        else:
            break
    else:
        raise RuntimeError("Unable to find end of header.")

    hdr_len = offset
    print "Found header of %d bytes." % (hdr_len,)

    fname = os.path.join(recovery_folder, "header.dat")
    f = file(fname, "wb")
    try:
        f.write(data[:hdr_len])
    finally:
        f.close()

    return data[hdr_len:]

def unpack_filelist(data, recovery_folder):
    """Unpack list of files."""
    n_files = ord(data[0])
    data = data[21:]

    print "Unpacking %d files ..." % (n_files,)

    fname = os.path.join(recovery_folder, "filelist.txt")
    f = file(fname, "wb")

    for i in range(n_files):
        for j, s in enumerate(data):
            if s in ("\x00", "\x80"):
                f.write(data[:j])
                f.write("\n")
                data = data[j+56+1:]
                break

    f.close()
    return data

def dump_contacts(data, recovery_folder):
    """Dump contacts (assumed to be everything after the file list)."""
    fname = os.path.join(recovery_folder, "contacts.db")
    f = file(fname, "wb")
    f.write(data)
    f.close()

def unpack(arc_file, recovery_folder):
    data = file(arc_file, "rb").read()
    len_file = len(data)

    data = dump_header(data, recovery_folder)
    print "Dumped header."

    i = 0
    while True:
        pos = len_file - len(data)

        try:
            decoder = zlib.decompressobj()
            msg = decoder.decompress(data)
        except zlib.error:
            # end of zlib blocks
            break

        data = decoder.unused_data
        old_pos = pos
        pos = len_file - len(data)

        print "Decoded data from %d to %d as file%d.rec." % (old_pos, pos, i)

        fname = os.path.join(recovery_folder, "file_%d.rec" % (i,))
        f = file(fname, "wb")
        try:
            f.write(msg)
        finally:
            f.close()

        i += 1

    # process file list
    data = unpack_filelist(data, recovery_folder)
    old_pos = pos
    pos = len_file - len(data)
    print "Files unpacked from %d to %d" % (old_pos, pos)

    # dump contacts
    dump_contacts(data, recovery_folder)
    print "Contacts db dumped from %d to %d" % (pos, len_file)


if __name__ == "__main__":
    if len(sys.argv) != 3:
        print "Usage: %s <arc file> <recovery folder>" % (sys.argv[0],)
        sys.exit(1)

    arc_file = sys.argv[1]
    recovery_folder = sys.argv[2]

    unpack(arc_file, recovery_folder)