mslink.py
上传用户:market2
上传日期:2018-11-18
资源大小:18786k
文件大小:10k
源码类别:

外挂编程

开发平台:

Windows_Unix

  1. """SCons.Tool.mslink
  2. Tool-specific initialization for the Microsoft linker.
  3. There normally shouldn't be any need to import this module directly.
  4. It will usually be imported through the generic SCons.Tool.Tool()
  5. selection method.
  6. """
  7. #
  8. # Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008 The SCons Foundation
  9. #
  10. # Permission is hereby granted, free of charge, to any person obtaining
  11. # a copy of this software and associated documentation files (the
  12. # "Software"), to deal in the Software without restriction, including
  13. # without limitation the rights to use, copy, modify, merge, publish,
  14. # distribute, sublicense, and/or sell copies of the Software, and to
  15. # permit persons to whom the Software is furnished to do so, subject to
  16. # the following conditions:
  17. #
  18. # The above copyright notice and this permission notice shall be included
  19. # in all copies or substantial portions of the Software.
  20. #
  21. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  22. # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  23. # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  24. # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  25. # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  26. # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  27. # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  28. #
  29. __revision__ = "src/engine/SCons/Tool/mslink.py 3057 2008/06/09 22:21:00 knight"
  30. import os.path
  31. import SCons.Action
  32. import SCons.Defaults
  33. import SCons.Errors
  34. import SCons.Platform.win32
  35. import SCons.Tool
  36. import SCons.Tool.msvc
  37. import SCons.Tool.msvs
  38. import SCons.Util
  39. def pdbGenerator(env, target, source, for_signature):
  40.     try:
  41.         return ['/PDB:%s' % target[0].attributes.pdb, '/DEBUG']
  42.     except (AttributeError, IndexError):
  43.         return None
  44. def windowsShlinkTargets(target, source, env, for_signature):
  45.     listCmd = []
  46.     dll = env.FindIxes(target, 'SHLIBPREFIX', 'SHLIBSUFFIX')
  47.     if dll: listCmd.append("/out:%s"%dll.get_string(for_signature))
  48.     implib = env.FindIxes(target, 'LIBPREFIX', 'LIBSUFFIX')
  49.     if implib: listCmd.append("/implib:%s"%implib.get_string(for_signature))
  50.     return listCmd
  51. def windowsShlinkSources(target, source, env, for_signature):
  52.     listCmd = []
  53.     deffile = env.FindIxes(source, "WINDOWSDEFPREFIX", "WINDOWSDEFSUFFIX")
  54.     for src in source:
  55.         if src == deffile:
  56.             # Treat this source as a .def file.
  57.             listCmd.append("/def:%s" % src.get_string(for_signature))
  58.         else:
  59.             # Just treat it as a generic source file.
  60.             listCmd.append(src)
  61.     return listCmd
  62. def windowsLibEmitter(target, source, env):
  63.     SCons.Tool.msvc.validate_vars(env)
  64.     extratargets = []
  65.     extrasources = []
  66.     dll = env.FindIxes(target, "SHLIBPREFIX", "SHLIBSUFFIX")
  67.     no_import_lib = env.get('no_import_lib', 0)
  68.     if not dll:
  69.         raise SCons.Errors.UserError, "A shared library should have exactly one target with the suffix: %s" % env.subst("$SHLIBSUFFIX")
  70.     insert_def = env.subst("$WINDOWS_INSERT_DEF")
  71.     if not insert_def in ['', '0', 0] and 
  72.        not env.FindIxes(source, "WINDOWSDEFPREFIX", "WINDOWSDEFSUFFIX"):
  73.         # append a def file to the list of sources
  74.         extrasources.append(
  75.             env.ReplaceIxes(dll,
  76.                             "SHLIBPREFIX", "SHLIBSUFFIX",
  77.                             "WINDOWSDEFPREFIX", "WINDOWSDEFSUFFIX"))
  78.     version_num, suite = SCons.Tool.msvs.msvs_parse_version(env.get('MSVS_VERSION', '6.0'))
  79.     if version_num >= 8.0 and env.get('WINDOWS_INSERT_MANIFEST', 0):
  80.         # MSVC 8 automatically generates .manifest files that must be installed
  81.         extratargets.append(
  82.             env.ReplaceIxes(dll,
  83.                             "SHLIBPREFIX", "SHLIBSUFFIX",
  84.                             "WINDOWSSHLIBMANIFESTPREFIX", "WINDOWSSHLIBMANIFESTSUFFIX"))
  85.     if env.has_key('PDB') and env['PDB']:
  86.         pdb = env.arg2nodes('$PDB', target=target, source=source)[0]
  87.         extratargets.append(pdb)
  88.         target[0].attributes.pdb = pdb
  89.     if not no_import_lib and 
  90.        not env.FindIxes(target, "LIBPREFIX", "LIBSUFFIX"):
  91.         # Append an import library to the list of targets.
  92.         extratargets.append(
  93.             env.ReplaceIxes(dll,
  94.                             "SHLIBPREFIX", "SHLIBSUFFIX",
  95.                             "LIBPREFIX", "LIBSUFFIX"))
  96.         # and .exp file is created if there are exports from a DLL
  97.         extratargets.append(
  98.             env.ReplaceIxes(dll,
  99.                             "SHLIBPREFIX", "SHLIBSUFFIX",
  100.                             "WINDOWSEXPPREFIX", "WINDOWSEXPSUFFIX"))
  101.     return (target+extratargets, source+extrasources)
  102. def prog_emitter(target, source, env):
  103.     SCons.Tool.msvc.validate_vars(env)
  104.     extratargets = []
  105.     exe = env.FindIxes(target, "PROGPREFIX", "PROGSUFFIX")
  106.     if not exe:
  107.         raise SCons.Errors.UserError, "An executable should have exactly one target with the suffix: %s" % env.subst("$PROGSUFFIX")
  108.     version_num, suite = SCons.Tool.msvs.msvs_parse_version(env.get('MSVS_VERSION', '6.0'))
  109.     if version_num >= 8.0 and env.get('WINDOWS_INSERT_MANIFEST', 0):
  110.         # MSVC 8 automatically generates .manifest files that have to be installed
  111.         extratargets.append(
  112.             env.ReplaceIxes(exe,
  113.                             "PROGPREFIX", "PROGSUFFIX",
  114.                             "WINDOWSPROGMANIFESTPREFIX", "WINDOWSPROGMANIFESTSUFFIX"))
  115.     if env.has_key('PDB') and env['PDB']:
  116.         pdb = env.arg2nodes('$PDB', target=target, source=source)[0]
  117.         extratargets.append(pdb)
  118.         target[0].attributes.pdb = pdb
  119.     return (target+extratargets,source)
  120. def RegServerFunc(target, source, env):
  121.     if env.has_key('register') and env['register']:
  122.         ret = regServerAction([target[0]], [source[0]], env)
  123.         if ret:
  124.             raise SCons.Errors.UserError, "Unable to register %s" % target[0]
  125.         else:
  126.             print "Registered %s sucessfully" % target[0]
  127.         return ret
  128.     return 0
  129. regServerAction = SCons.Action.Action("$REGSVRCOM", "$REGSVRCOMSTR")
  130. regServerCheck = SCons.Action.Action(RegServerFunc, None)
  131. shlibLinkAction = SCons.Action.Action('${TEMPFILE("$SHLINK $SHLINKFLAGS $_SHLINK_TARGETS $( $_LIBDIRFLAGS $) $_LIBFLAGS $_PDB $_SHLINK_SOURCES")}')
  132. compositeLinkAction = shlibLinkAction + regServerCheck
  133. def generate(env):
  134.     """Add Builders and construction variables for ar to an Environment."""
  135.     SCons.Tool.createSharedLibBuilder(env)
  136.     SCons.Tool.createProgBuilder(env)
  137.     env['SHLINK']      = '$LINK'
  138.     env['SHLINKFLAGS'] = SCons.Util.CLVar('$LINKFLAGS /dll')
  139.     env['_SHLINK_TARGETS'] = windowsShlinkTargets
  140.     env['_SHLINK_SOURCES'] = windowsShlinkSources
  141.     env['SHLINKCOM']   =  compositeLinkAction
  142.     env.Append(SHLIBEMITTER = [windowsLibEmitter])
  143.     env['LINK']        = 'link'
  144.     env['LINKFLAGS']   = SCons.Util.CLVar('/nologo')
  145.     env['_PDB'] = pdbGenerator
  146.     env['LINKCOM'] = '${TEMPFILE("$LINK $LINKFLAGS /OUT:$TARGET.windows $( $_LIBDIRFLAGS $) $_LIBFLAGS $_PDB $SOURCES.windows")}'
  147.     env.Append(PROGEMITTER = [prog_emitter])
  148.     env['LIBDIRPREFIX']='/LIBPATH:'
  149.     env['LIBDIRSUFFIX']=''
  150.     env['LIBLINKPREFIX']=''
  151.     env['LIBLINKSUFFIX']='$LIBSUFFIX'
  152.     env['WIN32DEFPREFIX']        = ''
  153.     env['WIN32DEFSUFFIX']        = '.def'
  154.     env['WIN32_INSERT_DEF']      = 0
  155.     env['WINDOWSDEFPREFIX']      = '${WIN32DEFPREFIX}'
  156.     env['WINDOWSDEFSUFFIX']      = '${WIN32DEFSUFFIX}'
  157.     env['WINDOWS_INSERT_DEF']    = '${WIN32_INSERT_DEF}'
  158.     env['WIN32EXPPREFIX']        = ''
  159.     env['WIN32EXPSUFFIX']        = '.exp'
  160.     env['WINDOWSEXPPREFIX']      = '${WIN32EXPPREFIX}'
  161.     env['WINDOWSEXPSUFFIX']      = '${WIN32EXPSUFFIX}'
  162.     env['WINDOWSSHLIBMANIFESTPREFIX'] = ''
  163.     env['WINDOWSSHLIBMANIFESTSUFFIX'] = '${SHLIBSUFFIX}.manifest'
  164.     env['WINDOWSPROGMANIFESTPREFIX']  = ''
  165.     env['WINDOWSPROGMANIFESTSUFFIX']  = '${PROGSUFFIX}.manifest'
  166.     env['REGSVRACTION'] = regServerCheck
  167.     env['REGSVR'] = os.path.join(SCons.Platform.win32.get_system_root(),'System32','regsvr32')
  168.     env['REGSVRFLAGS'] = '/s '
  169.     env['REGSVRCOM'] = '$REGSVR $REGSVRFLAGS ${TARGET.windows}'
  170.     try:
  171.         version = SCons.Tool.msvs.get_default_visualstudio_version(env)
  172.         if env.has_key('MSVS_IGNORE_IDE_PATHS') and env['MSVS_IGNORE_IDE_PATHS']:
  173.             include_path, lib_path, exe_path = SCons.Tool.msvc.get_msvc_default_paths(env,version)
  174.         else:
  175.             include_path, lib_path, exe_path = SCons.Tool.msvc.get_msvc_paths(env,version)
  176.         # since other tools can set these, we just make sure that the
  177.         # relevant stuff from MSVS is in there somewhere.
  178.         env.PrependENVPath('INCLUDE', include_path)
  179.         env.PrependENVPath('LIB', lib_path)
  180.         env.PrependENVPath('PATH', exe_path)
  181.     except (SCons.Util.RegError, SCons.Errors.InternalError):
  182.         pass
  183.     # For most platforms, a loadable module is the same as a shared
  184.     # library.  Platforms which are different can override these, but
  185.     # setting them the same means that LoadableModule works everywhere.
  186.     SCons.Tool.createLoadableModuleBuilder(env)
  187.     env['LDMODULE'] = '$SHLINK'
  188.     env['LDMODULEPREFIX'] = '$SHLIBPREFIX'
  189.     env['LDMODULESUFFIX'] = '$SHLIBSUFFIX'
  190.     env['LDMODULEFLAGS'] = '$SHLINKFLAGS'
  191.     # We can't use '$SHLINKCOM' here because that will stringify the
  192.     # action list on expansion, and will then try to execute expanded
  193.     # strings, with the upshot that it would try to execute RegServerFunc
  194.     # as a command.
  195.     env['LDMODULECOM'] = compositeLinkAction
  196. def exists(env):
  197.     platform = env.get('PLATFORM', '')
  198.     if SCons.Tool.msvs.is_msvs_installed():
  199.         # there's at least one version of MSVS installed.
  200.         return 1
  201.     elif platform in ('win32', 'cygwin'):
  202.         # Only explicitly search for a 'link' executable on Windows
  203.         # systems.  Some other systems (e.g. Ubuntu Linux) have an
  204.         # executable named 'link' and we don't want that to make SCons
  205.         # think Visual Studio is installed.
  206.         return env.Detect('link')
  207.     return None