Subversion Repositories c0or1k

#! /usr/bin/env python2.6
# -*- mode: python; coding: utf-8; -*-
#
#  Codezero -- a microkernel for embedded systems.
#
#  Copyright © 2009  B Labs Ltd
#
import os, sys, shelve
from os.path import join
 
PROJRELROOT = '../../'
 
SCRIPTROOT = os.path.abspath(os.path.dirname("."))
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), PROJRELROOT)))
 
from config.projpaths import *
from config.configuration import *
 
containers_assembler_body = \
'''
.align 4
.section .cont.%d
.incbin "%s"
'''
 
containers_lds_start = \
'''/*
 * Autogenerated linker script that packs all containers
 * in a single image.
 *
 * Copyright (C) 2009 B Labs Ltd.
 */
 
SECTIONS
{'''
 
containers_lds_body = \
'''
	.cont.%d : { *(.cont.%d) }'''
 
containers_lds_end = \
'''
}
'''
 
class AllContainerPacker:
    def __init__(self, image_list, container_list):
        self.cont_images_in = image_list
        self.cont_images_in.sort()
        self.containers = container_list
 
        self.CONTAINERS_BUILDDIR = join(PROJROOT, 'build/conts')
        self.containers_lds_out = join(self.CONTAINERS_BUILDDIR, \
                                      'containers.lds')
        self.containers_S_out = join(self.CONTAINERS_BUILDDIR, 'containers.S')
        self.containers_elf_out = join(self.CONTAINERS_BUILDDIR, \
                                      'containers.elf')
 
    def generate_container_S(self, target_path):
        with open(target_path, 'w+') as f:
            file_body = ""
            img_i = 0
            for img in self.cont_images_in:
                file_body += containers_assembler_body % (img_i, img)
                img_i += 1
 
            f.write(file_body)
 
    def generate_container_lds(self, target_path):
        with open(target_path, 'w+') as f:
            img_i = 0
            file_body = containers_lds_start
            for img in self.cont_images_in:
                file_body += containers_lds_body % (img_i, img_i)
                img_i += 1
            file_body += containers_lds_end
            f.write(file_body)
 
    def pack_all(self, config):
        self.generate_container_lds(self.containers_lds_out)
        self.generate_container_S(self.containers_S_out)
        os.system(config.toolchain_kernel + "gcc " + "-nostdlib -o %s -T%s %s" \
                  % (self.containers_elf_out, self.containers_lds_out, \
                     self.containers_S_out))
 
        # Return the final image to calling script
        return self.containers_elf_out
 
    def clean(self):
        if os.path.exists(self.containers_elf_out):
            shutil.rmtree(self.containers_elf_out)
        if os.path.exists(self.containers_lds_out):
            shutil.rmtree(self.containers_lds_out)
        if os.path.exists(self.containers_S_out):
            shutil.rmtree(self.containers_S_out)
 
if __name__ == "__main__":
    all_cont_packer = AllContainerPacker([], [])