Merge pull request #68 from 0xc0170/dev_gcc_sizes

gcc - print section sizes

Author: sg-

tools/hooks.py

@@ -11,7 +11,7 @@
 _running_hooks = {}
 
 # Available hook types
-_hook_types = ["binary", "compile", "link", "assemble"]
+_hook_types = ["binary", "compile", "link", "assemble", "static_sizes"]
 
 # Available hook steps
 _hook_steps = ["pre", "replace", "post"]
@@ -103,6 +103,9 @@ def hook_cmdline_assembler(self, function):
     def hook_cmdline_binary(self, function):
         return self._hook_cmdline("binary", function)
 
+    def hook_cmdline_static_sizes(self, function):
+        return self._hook_cmdline("static_sizes", function)
+
     # Return the command line after applying the hook
     def _get_cmdline(self, hook_type, cmdline):
         if self._cmdline_hooks.has_key(hook_type):

tools/toolchains/__init__.py

@@ -25,6 +25,8 @@
 from os.path import join, splitext, exists, relpath, dirname, basename, split
 from inspect import getmro
 
+from elftools.elf.elffile import ELFFile
+
 from multiprocessing import Pool, cpu_count
 from tools.utils import run_cmd, mkdir, rel_path, ToolException, NotSupportedException, split_path
 from tools.settings import BUILD_OPTIONS, MBED_ORG_USER
@@ -701,6 +703,11 @@ def link_program(self, r, tmp_path, name):
 
             self.binary(r, elf, bin)
 
+        self.info("Memory sections sizes:")
+        size_dict = self.static_sizes(elf)
+        for section, size in size_dict.iteritems():
+            print("{:20} {}".format(section, size))
+
         self.var("compile_succeded", True)
         self.var("binary", filename)
 
@@ -757,6 +764,41 @@ def tool_error(self, message):
     def var(self, key, value):
         self.notify({'type': 'var', 'key': key, 'val': value})
 
+    def static_sizes(self, elf):
+        """Accepts elf, returns a dict sizes per section (text, data, bss)"""
+        section_sizes = {}
+
+        SHF_WRITE = 0x1
+        SHF_ALLOC = 0x2
+        SHF_EXECINSTR = 0x4
+        SHT_PROGBITS = "SHT_PROGBITS"
+        SHT_NOBITS = "SHT_NOBITS"
+
+        text = 0
+        data = 0
+        bss = 0
+        with open(elf, 'rb') as f:
+            elffile = ELFFile(f)
+            for section in elffile.iter_sections():
+                flags = section['sh_flags']
+                size = section['sh_size']
+                if (flags & SHF_ALLOC) == 0:
+                    # Section has no relevant data so ignore it
+                    continue
+                if (flags & SHF_EXECINSTR) or not (flags & SHF_WRITE):
+                    # Executable code or read only data
+                    text += size
+                elif section['sh_type'] != SHT_NOBITS:
+                    # Non-zero read/write data
+                    data += size
+                else:
+                    # Zero init read/write data
+                    bss += size
+        section_sizes["text"] = text
+        section_sizes["data"] = data
+        section_sizes["bss"] = bss
+        return section_sizes
+
 from tools.settings import ARM_BIN
 from tools.settings import GCC_ARM_PATH, GCC_CR_PATH
 from tools.settings import IAR_PATH