Skip to content
Snippets Groups Projects
check_static_binaries.py 5.82 KiB
Newer Older
  • Learn to ignore specific revisions
  • #!/usr/bin/env python3
    
    # ***** BEGIN GPL LICENSE BLOCK *****
    #
    # This program 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.
    #
    # This program 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 this program; if not, write to the Free Software Foundation,
    # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
    #
    # Contributor(s): Sergey Sharybin
    #
    # #**** END GPL LICENSE BLOCK #****
    
    # <pep8 compliant>
    
    import os
    from pathlib import Path
    import re
    import subprocess
    import sys
    import unittest
    
    from check_utils import (sliceCommandLineArguments,
                             parseArguments)
    
    
    ALLOWED_LIBS = (
        # Core C/C++ libraries
        "ld-linux.so",
        "ld-linux-x86-64.so",
        "libc.so",
        "libm.so",
        "libstdc++.so",
        "libdl.so",
        "libpthread.so",
        "libgcc_s.so",
        "librt.so",
        "libutil.so",
    
        # X11 libraries we don't link statically,
        "libX11.so",
        "libXext.so",
        "libXrender.so",
        "libXxf86vm.so",
        "libXi.so",
    
        # OpenGL libraries.
        "libGL.so",
        "libGLU.so",
    
        # Own dependencies we don't link statically.
        "libfreetype.so",
    )
    
    IGNORE_FILES = ("blender-softwaregl", )
    IGNORE_EXTENSION = (".sh", ".py", )
    
    
    # Library dependencies.
    
    def getNeededLibrariesLDD(binary_filepath):
        """
        This function uses ldd to collect libraries which binary depends on.
    
        Not totally safe since ldd might actually execute the binary to get it's
        symbols and will also collect indirect dependnecies which might not be
        desired.
    
        Has advantage of telling that some dependnecy library is not found.
        """
        ldd_command = ("ldd", str(binary_filepath))
        ldd_output = subprocess.check_output(ldd_command, stderr=subprocess.STDOUT)
        lines = ldd_output.decode().split("\n")
        libraries = []
        for line in lines:
            line = line.strip()
            if not line:
                continue
            lib_name = line.split("=>")[0]
            lib_name = lib_name.split(" (")[0].strip()
            lib_file_name = os.path.basename(lib_name)
            libraries.append(lib_file_name)
        return libraries
    
    
    def getNeededLibrariesOBJDUMP(binary_filepath):
        """
        This function uses objdump to get direct dependencies of a given binary.
    
        Totally safe, but will require manual check over libraries which are not
        found on the system.
        """
        objdump_command = ("objdump", "-p", str(binary_filepath))
        objdump_output = subprocess.check_output(objdump_command,
                                                 stderr=subprocess.STDOUT)
        lines = objdump_output.decode().split("\n")
        libraries = []
        for line in lines:
            line = line.strip()
            if not line:
                continue
            if not line.startswith("NEEDED"):
                continue
            lib_name = line[6:].strip()
            libraries.append(lib_name)
        return libraries
    
    
    def getNeededLibraries(binary_filepath):
        """
        Get all libraries given binary depends on.
        """
        if False:
            return getNeededLibrariesLDD(binary_filepath)
        else:
            return getNeededLibrariesOBJDUMP(binary_filepath)
    
    
    def stripLibraryABI(lib_name):
        """
        Strip ABI suffix from .so file
    
        Example; libexample.so.1.0 => libexample.so
        """
        lib_name_no_abi = lib_name
        # TOOD(sergey): Optimize this!
        while True:
            no_abi = re.sub(r"\.[0-9]+$", "", lib_name_no_abi)
            if lib_name_no_abi == no_abi:
                break
            lib_name_no_abi = no_abi
        return lib_name_no_abi
    
    
    class UnitTesting(unittest.TestCase):
        def checkBinary(self, binary_filepath):
            """
            Check given binary file to be a proper static self-sufficient.
            """
    
            libraries = getNeededLibraries(binary_filepath)
            for lib_name in libraries:
                lib_name_no_abi = stripLibraryABI(lib_name)
                self.assertTrue(lib_name_no_abi in ALLOWED_LIBS,
                                "Error detected in {}: library used {}" . format(
                                    binary_filepath, lib_name))
    
        def checkDirectory(self, directory):
            """
            Recursively traverse directory and check every binary in in.
            """
    
            for path in Path(directory).rglob("*"):
                # Ignore any checks on directory.
                if path.is_dir():
                    continue
                # Ignore script files.
                if path.name in IGNORE_FILES:
                    continue
                if path.suffix in IGNORE_EXTENSION:
                    continue
                # Check any executable binary,
                if path.stat().st_mode & 0o111 != 0:
                    self.checkBinary(path)
                # Check all dynamic libraries.
                elif path.suffix == ".so":
                    self.checkBinary(path)
    
        def test_directoryIsStatic(self):
            # Parse arguments which are not handled by unit testing framework.
            args = parseArguments()
            # Do some sanity checks first.
            self.assertTrue(os.path.exists(args.directory),
                            "Given directory does not exist: {}" .
                            format(args.directory))
            self.assertTrue(os.path.isdir(args.directory),
                            "Given path is not a directory: {}" .
                            format(args.directory))
            # Perform actual test,
            self.checkDirectory(args.directory)
    
    
    def main():
        # Slice command line arguments by '--'
        unittest_args, parser_args = sliceCommandLineArguments()
        # Construct and run unit tests.
        unittest.main(argv=unittest_args)
    
    
    if __name__ == "__main__":
        main()