Revert 191173 "Check in a custom-built import library for user32..."

trunk/src/build/win/importlibs/create_importlib_win.py

-#!/usr/bin/env python
-# Copyright (c) 2012 The Chromium Authors. All rights reserved.
-# Use of this source code is governed by a BSD-style license that can be
-# found in the LICENSE file.
-#
-"""Creates an import library from an import description file."""
-import ast
-import logging
-import optparse
-import os
-import os.path
-import shutil
-import subprocess
-import sys
-import tempfile
-
-
-_USAGE = """\
-Usage: %prog [options] [imports-file]
-
-Creates an import library from imports-file.
-
-Note: this script uses the microsoft assembler (ml.exe) and the library tool
-    (lib.exe), both of which must be in path.
-"""
-
-
-_ASM_STUB_HEADER = """\
-; This file is autogenerated by create_importlib_win.py, do not edit.
-.386
-.MODEL FLAT, C
-.CODE
-
-; Stubs to provide mangled names to lib.exe for the
-; correct generation of import libs.
-"""
-
-
-_DEF_STUB_HEADER = """\
-; This file is autogenerated by create_importlib_win.py, do not edit.
-
-; Export declarations for generating import libs.
-"""
-
-
-_LOGGER = logging.getLogger()
-
-
-
-class _Error(Exception):
-  pass
-
-
-class _ImportLibraryGenerator(object):
-  def __init__(self, temp_dir):
-    self._temp_dir = temp_dir
-
-  def _Shell(self, cmd, **kw):
-    ret = subprocess.call(cmd, **kw)
-    _LOGGER.info('Running "%s" returned %d.', cmd, ret)
-    if ret != 0:
-      raise _Error('Command "%s" returned %d.' % (cmd, ret))
-
-  def _ReadImportsFile(self, imports_file):
-    # Slurp the imports file.
-    return ast.literal_eval(open(imports_file).read())
-
-  def _WriteStubsFile(self, import_names, output_file):
-    output_file.write(_ASM_STUB_HEADER)
-
-    for name in import_names:
-      output_file.write('%s PROC\n' % name)
-      output_file.write('%s ENDP\n' % name)
-
-    output_file.write('END\n')
-
-  def _WriteDefFile(self, dll_name, import_names, output_file):
-    output_file.write(_DEF_STUB_HEADER)
-    output_file.write('NAME %s\n' % dll_name)
-    output_file.write('EXPORTS\n')
-    for name in import_names:
-      name = name.split('@')[0]
-      output_file.write('  %s\n' % name)
-
-  def _CreateObj(self, dll_name, imports):
-    """Writes an assembly file containing empty declarations.
-
-    For each imported function of the form:
-
-    AddClipboardFormatListener@4 PROC
-    AddClipboardFormatListener@4 ENDP
-
-    The resulting object file is then supplied to lib.exe with a .def file
-    declaring the corresponding non-adorned exports as they appear on the
-    exporting DLL, e.g.
-
-    EXPORTS
-      AddClipboardFormatListener
-
-    In combination, the .def file and the .obj file cause lib.exe to generate
-    an x86 import lib with public symbols named like
-    "__imp__AddClipboardFormatListener@4", binding to exports named like
-    "AddClipboardFormatListener".
-
-    All of this is perpetrated in a temporary directory, as the intermediate
-    artifacts are quick and easy to produce, and of no interest to anyone
-    after the fact."""
-
-    # Create an .asm file to provide stdcall-like stub names to lib.exe.
-    asm_name = dll_name + '.asm'
-    _LOGGER.info('Writing asm file "%s".', asm_name)
-    with open(os.path.join(self._temp_dir, asm_name), 'wb') as stubs_file:
-      self._WriteStubsFile(imports, stubs_file)
-
-    # Invoke on the assembler to compile it to .obj.
-    obj_name = dll_name + '.obj'
-    cmdline = ['ml.exe', '/nologo', '/c', asm_name, '/Fo', obj_name]
-    self._Shell(cmdline, cwd=self._temp_dir)
-
-    return obj_name
-
-  def _CreateImportLib(self, dll_name, imports, architecture, output_file):
-    """Creates an import lib binding imports to dll_name for architecture.
-
-    On success, writes the import library to output file.
-    """
-    obj_file = None
-
-    # For x86 architecture we have to provide an object file for correct
-    # name mangling between the import stubs and the exported functions.
-    if architecture == 'x86':
-      obj_file = self._CreateObj(dll_name, imports)
-
-    # Create the corresponding .def file. This file has the non stdcall-adorned
-    # names, as exported by the destination DLL.
-    def_name = dll_name + '.def'
-    _LOGGER.info('Writing def file "%s".', def_name)
-    with open(os.path.join(self._temp_dir, def_name), 'wb') as def_file:
-      self._WriteDefFile(dll_name, imports, def_file)
-
-    # Invoke on lib.exe to create the import library.
-    # We generate everything into the temporary directory, as the .exp export
-    # files will be generated at the same path as the import library, and we
-    # don't want those files potentially gunking the works.
-    dll_base_name, ext = os.path.splitext(dll_name)
-    lib_name = dll_base_name + '.lib'
-    cmdline = ['lib.exe',
-               '/machine:%s' % architecture,
-               '/def:%s' % def_name,
-               '/out:%s' % lib_name]
-    if obj_file:
-      cmdline.append(obj_file)
-
-    self._Shell(cmdline, cwd=self._temp_dir)
-
-    # Copy the .lib file to the output directory.
-    shutil.copyfile(os.path.join(self._temp_dir, lib_name), output_file)
-    _LOGGER.info('Created "%s".', output_file)
-
-  def CreateImportLib(self, imports_file, output_file):
-    # Read the imports file.
-    imports = self._ReadImportsFile(imports_file)
-
-    # Creates the requested import library in the output directory.
-    self._CreateImportLib(imports['dll_name'],
-                          imports['imports'],
-                          imports.get('architecture', 'x86'),
-                          output_file)
-
-
-def main():
-  parser = optparse.OptionParser(usage=_USAGE)
-  parser.add_option('-o', '--output-file',
-                    help='Specifies the output file path.')
-  parser.add_option('-k', '--keep-temp-dir',
-                    action='store_true',
-                    help='Keep the temporary directory.')
-  parser.add_option('-v', '--verbose',
-                    action='store_true',
-                    help='Verbose logging.')
-
-  options, args = parser.parse_args()
-
-  print args
-  if len(args) != 1:
-    parser.error('You must provide an imports file.')
-
-  if not options.output_file:
-    parser.error('You must provide an output file.')
-
-  options.output_file = os.path.abspath(options.output_file)
-
-  if options.verbose:
-    logging.basicConfig(level=logging.INFO)
-  else:
-    logging.basicConfig(level=logging.WARN)
-
-
-  temp_dir = tempfile.mkdtemp()
-  _LOGGER.info('Created temporary directory "%s."', temp_dir)
-  try:
-    # Create a generator and create the import lib.
-    generator = _ImportLibraryGenerator(temp_dir)
-
-    ret = generator.CreateImportLib(args[0], options.output_file)
-  except Exception, e:
-    _LOGGER.exception('Failed to create imports.')
-    ret = 1
-  finally:
-    if not options.keep_temp_dir:
-      shutil.rmtree(temp_dir)
-      _LOGGER.info('Deleted temporary directory "%s."', temp_dir)
-
-  return ret
-
-
-if __name__ == '__main__':
-  sys.exit(main())