423 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			423 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # vim: set sts=2 ts=8 sw=2 tw=99 et ft=python:
 | |
| import os
 | |
| 
 | |
| def ResolveEnvPath(env, folder):
 | |
|   if env in os.environ:
 | |
|     path = os.environ[env]
 | |
|     if os.path.isdir(path):
 | |
|       return path
 | |
|     return None
 | |
| 
 | |
|   head = os.getcwd()
 | |
|   oldhead = None
 | |
|   while head != None and head != oldhead:
 | |
|     path = os.path.join(head, folder)
 | |
|     if os.path.isdir(path):
 | |
|       return path
 | |
|     oldhead = head
 | |
|     head, tail = os.path.split(head)
 | |
| 
 | |
|   return None
 | |
| 
 | |
| def Normalize(path):
 | |
|   return os.path.abspath(os.path.normpath(path))
 | |
|   
 | |
| def SetArchFlags(compiler):
 | |
|   if compiler.behavior == 'gcc':
 | |
|     if compiler.target.arch == 'x86_64':
 | |
|       compiler.cflags += ['-fPIC']
 | |
|   elif compiler.like('msvc'):
 | |
|     if compiler.target.arch == 'x86_64':
 | |
|       compiler.defines += ['WIN64']
 | |
| 
 | |
| hl2sdk_manifests_path = None
 | |
| 
 | |
| # First we check if the manifest exists in the current path
 | |
| hl2sdk_manifests_path = os.path.join(builder.sourcePath, 'hl2sdk-manifests/SdkHelpers.ambuild')
 | |
| 
 | |
| if not os.path.exists(hl2sdk_manifests_path):
 | |
|   # manifests does not exists in the project file, use the path from --hl2sdk-manifest-path
 | |
|   if not builder.options.hl2sdk_manifest:
 | |
|     raise Exception('HL2SDK Manifest root path not set! (--hl2sdk-manifest-path)')
 | |
|   else:
 | |
|     hl2sdk_manifests_path = os.path.join(builder.options.hl2sdk_manifest, 'SdkHelpers.ambuild')
 | |
| 
 | |
|     if not os.path.exists(hl2sdk_manifests_path):
 | |
|       raise Exception('Could not find SdkHelpers.ambuild in the given HL2SDK Manifest path!')
 | |
| 
 | |
| 
 | |
| SdkHelpers = builder.Eval(hl2sdk_manifests_path, {
 | |
|     'Project': 'sm-extension'
 | |
| })
 | |
| 
 | |
| class ExtensionConfig(object):
 | |
|   def __init__(self):
 | |
|     self.sdk_manifests = []
 | |
|     self.sdks = {}
 | |
|     self.sdk_targets = []
 | |
|     self.extensions = []
 | |
|     self.generated_headers = None
 | |
|     self.mms_root = None
 | |
|     self.sm_root = None
 | |
|     self.all_targets = []
 | |
|     self.target_archs = set()
 | |
|     self.libsafetyhook = {}
 | |
| 
 | |
|     if builder.options.targets:
 | |
|       target_archs = builder.options.targets.split(',')
 | |
|     else:
 | |
|       target_archs = ['x86', 'x86_64']
 | |
| 
 | |
|     for arch in target_archs:
 | |
|         try:
 | |
|             cxx = builder.DetectCxx(target_arch = arch)
 | |
|             self.target_archs.add(cxx.target.arch)
 | |
|         except Exception as e:
 | |
|             # Error if archs were manually overridden.
 | |
|             if builder.options.targets:
 | |
|                 raise
 | |
|             print('Skipping target {}: {}'.format(arch, e))
 | |
|             continue
 | |
|         self.all_targets.append(cxx)
 | |
| 
 | |
|     if not self.all_targets:
 | |
|         raise Exception('No suitable C/C++ compiler was found.')
 | |
| 
 | |
|   def use_auto_versioning(self):
 | |
|     if builder.backend != 'amb2':
 | |
|       return False
 | |
|     return not getattr(builder.options, 'disable_auto_versioning', False)
 | |
| 
 | |
|   @property
 | |
|   def tag(self):
 | |
|     if builder.options.debug == '1':
 | |
|       return 'Debug'
 | |
|     return 'Release'
 | |
| 
 | |
|   def findSdkPath(self, sdk_name):
 | |
|     dir_name = 'hl2sdk-{}'.format(sdk_name)
 | |
|     if builder.options.hl2sdk_root:
 | |
|       sdk_path = os.path.join(builder.options.hl2sdk_root, dir_name)
 | |
|       if os.path.exists(sdk_path):
 | |
|         return sdk_path
 | |
|     return ResolveEnvPath('HL2SDK{}'.format(sdk_name.upper()), dir_name)
 | |
| 
 | |
|   def shouldIncludeSdk(self, sdk):
 | |
|     return not sdk.get('source2', False)
 | |
| 
 | |
|   def detectSDKs(self):
 | |
|     sdk_list = [s for s in builder.options.sdks.split(',') if s]
 | |
|     SdkHelpers.sdk_filter = self.shouldIncludeSdk
 | |
|     SdkHelpers.find_sdk_path = self.findSdkPath
 | |
|     SdkHelpers.findSdks(builder, self.all_targets, sdk_list)
 | |
| 
 | |
|     self.sdks = SdkHelpers.sdks
 | |
|     self.sdk_manifests = SdkHelpers.sdk_manifests
 | |
|     self.sdk_targets = SdkHelpers.sdk_targets
 | |
| 
 | |
|     if builder.options.mms_path:
 | |
|       self.mms_root = builder.options.mms_path
 | |
|     else:
 | |
|       self.mms_root = ResolveEnvPath('MMSOURCE112', 'mmsource-1.12')
 | |
|       if not self.mms_root:
 | |
|         self.mms_root = ResolveEnvPath('MMSOURCE', 'metamod-source')
 | |
|       if not self.mms_root:
 | |
|         self.mms_root = ResolveEnvPath('MMSOURCE_DEV', 'metamod-source')
 | |
|       if not self.mms_root:
 | |
|         self.mms_root = ResolveEnvPath('MMSOURCE_DEV', 'mmsource-central')
 | |
| 
 | |
|     if not self.mms_root or not os.path.isdir(self.mms_root):
 | |
|       raise Exception('Could not find a source copy of Metamod:Source')
 | |
|     self.mms_root = Normalize(self.mms_root)
 | |
| 
 | |
|     if builder.options.sm_path:
 | |
|       self.sm_root = builder.options.sm_path
 | |
|     else:
 | |
|       self.sm_root = ResolveEnvPath('SOURCEMOD112', 'sourcemod-1.12')
 | |
|       if not self.sm_root:
 | |
|         self.sm_root = ResolveEnvPath('SOURCEMOD', 'sourcemod')
 | |
|       if not self.sm_root:
 | |
|         self.sm_root = ResolveEnvPath('SOURCEMOD_DEV', 'sourcemod')
 | |
|       if not self.sm_root:
 | |
|         self.sm_root = ResolveEnvPath('SOURCEMOD_DEV', 'sourcemod-central')
 | |
| 
 | |
|     if not self.sm_root or not os.path.isdir(self.sm_root):
 | |
|       raise Exception('Could not find a source copy of SourceMod')
 | |
|     self.sm_root = Normalize(self.sm_root)
 | |
| 
 | |
|   def configure(self):
 | |
|     
 | |
|     allowed_archs = ['x86','x86_64']
 | |
| 
 | |
|     if not set(self.target_archs).issubset(allowed_archs):
 | |
|       raise Exception('Unknown target architecture: {0}'.format(self.target_archs))
 | |
| 
 | |
|     for cxx in self.all_targets:
 | |
|         self.configure_cxx(cxx)
 | |
| 
 | |
|   def configure_cxx(self, cxx):
 | |
|     if cxx.family == 'msvc':
 | |
|       if cxx.version < 1914 and builder.options.generator != 'vs':
 | |
|         raise Exception(f'Only MSVC 2017 15.7 and later are supported, full C++17 support is required. ({str(cxx.version)} < 1914)')
 | |
|     elif cxx.family == 'gcc':
 | |
|       if cxx.version < 'gcc-8':
 | |
|         raise Exception('Only GCC versions 8 or later are supported, full C++17 support is required.')
 | |
|     elif cxx.family == 'clang':
 | |
|       if cxx.version < 'clang-5':
 | |
|         raise Exception('Only clang versions 5 or later are supported, full C++17 support is required.')
 | |
| 
 | |
|     if cxx.like('gcc'):
 | |
|       self.configure_gcc(cxx)
 | |
|     elif cxx.family == 'msvc':
 | |
|       self.configure_msvc(cxx)
 | |
| 
 | |
|     # Optimization
 | |
|     if builder.options.opt == '1':
 | |
|       cxx.defines += ['NDEBUG']
 | |
| 
 | |
|     # Debugging
 | |
|     if builder.options.debug == '1':
 | |
|       cxx.defines += ['DEBUG', '_DEBUG']
 | |
| 
 | |
|     # Platform-specifics
 | |
|     if cxx.target.platform == 'linux':
 | |
|       self.configure_linux(cxx)
 | |
|     elif cxx.target.platform == 'mac':
 | |
|       self.configure_mac(cxx)
 | |
|     elif cxx.target.platform == 'windows':
 | |
|       self.configure_windows(cxx)
 | |
| 
 | |
|   def configure_gcc(self, cxx):
 | |
|     cxx.defines += [
 | |
|       'stricmp=strcasecmp',
 | |
|       '_stricmp=strcasecmp',
 | |
|       '_snprintf=snprintf',
 | |
|       '_vsnprintf=vsnprintf',
 | |
|       'HAVE_STDINT_H',
 | |
|       'GNUC',
 | |
|     ]
 | |
|     cxx.cflags += [
 | |
|       '-pipe',
 | |
|       '-fno-strict-aliasing',
 | |
|       '-Wall',
 | |
|       '-Werror',
 | |
|       '-Wno-unused',
 | |
|       '-Wno-switch',
 | |
|       '-Wno-array-bounds',
 | |
|       '-fvisibility=hidden',
 | |
|     ]
 | |
| 
 | |
|     if cxx.target.arch in ['x86', 'x86_64']:
 | |
|       cxx.cflags += ['-msse']
 | |
| 
 | |
|     cxx.cxxflags += [
 | |
|       '-fno-threadsafe-statics',
 | |
|       '-Wno-non-virtual-dtor',
 | |
|       '-Wno-overloaded-virtual',
 | |
|       '-Wno-register',
 | |
|       '-fvisibility-inlines-hidden',
 | |
|       '-std=c++17',
 | |
|     ]
 | |
| 
 | |
|     
 | |
|     have_gcc = cxx.family == 'gcc'
 | |
|     have_clang = cxx.family == 'clang'
 | |
| 
 | |
|     # Work around errors from smsdk_ext.cpp
 | |
|     if have_clang:
 | |
|       cxx.cxxflags += ['-Wno-implicit-exception-spec-mismatch']
 | |
| 
 | |
|     # Work around SDK warnings.
 | |
|     if cxx.version >= 'clang-10.0' or cxx.version >= 'apple-clang-12.0':
 | |
|         cxx.cflags += [
 | |
|             '-Wno-implicit-int-float-conversion',
 | |
|             '-Wno-tautological-overlap-compare',
 | |
|         ]
 | |
| 
 | |
|     if have_gcc:
 | |
|       cxx.cflags += ['-mfpmath=sse']
 | |
|       cxx.cflags += ['-Wno-maybe-uninitialized']
 | |
| 
 | |
|     if builder.options.opt == '1':
 | |
|         cxx.cflags += ['-O3']
 | |
| 
 | |
|     # Don't omit the frame pointer.
 | |
|     cxx.cflags += ['-fno-omit-frame-pointer']
 | |
| 
 | |
|   def configure_msvc(self, cxx):
 | |
| 
 | |
|     if builder.options.debug == '1':
 | |
|       cxx.cflags += ['/MTd']
 | |
|       cxx.linkflags += ['/NODEFAULTLIB:libcmt']
 | |
|     else:
 | |
|       cxx.cflags += ['/MT']
 | |
|     cxx.defines += [
 | |
|       '_CRT_SECURE_NO_DEPRECATE',
 | |
|       '_CRT_SECURE_NO_WARNINGS',
 | |
|       '_CRT_NONSTDC_NO_DEPRECATE',
 | |
|       '_ITERATOR_DEBUG_LEVEL=0',
 | |
|     ]
 | |
|     cxx.cflags += [
 | |
|       '/W3',
 | |
|     ]
 | |
|     cxx.cxxflags += [
 | |
|       '/EHsc',
 | |
|       '/GR-',
 | |
|       '/TP',
 | |
|       '/std:c++17',
 | |
|     ]
 | |
|     cxx.linkflags += [
 | |
|       'kernel32.lib',
 | |
|       'user32.lib',
 | |
|       'gdi32.lib',
 | |
|       'winspool.lib',
 | |
|       'comdlg32.lib',
 | |
|       'advapi32.lib',
 | |
|       'shell32.lib',
 | |
|       'ole32.lib',
 | |
|       'oleaut32.lib',
 | |
|       'uuid.lib',
 | |
|       'odbc32.lib',
 | |
|       'odbccp32.lib',
 | |
|     ]
 | |
| 
 | |
|     if builder.options.opt == '1':
 | |
|       cxx.cflags += ['/Ox', '/Zo']
 | |
|       cxx.linkflags += ['/OPT:ICF', '/OPT:REF']
 | |
| 
 | |
|     if builder.options.debug == '1':
 | |
|       cxx.cflags += ['/Od', '/RTC1']
 | |
| 
 | |
|     # This needs to be after our optimization flags which could otherwise disable it.
 | |
|     # Don't omit the frame pointer.
 | |
|     cxx.cflags += ['/Oy-']
 | |
| 
 | |
|   def configure_linux(self, cxx):
 | |
|     cxx.defines += ['LINUX', '_LINUX', 'POSIX', '_FILE_OFFSET_BITS=64']
 | |
|     cxx.linkflags += ['-lm']
 | |
|     if cxx.family == 'gcc':
 | |
|       cxx.linkflags += ['-static-libgcc']
 | |
|     elif cxx.family == 'clang':
 | |
|       cxx.linkflags += ['-lgcc_eh']
 | |
|     cxx.linkflags += ['-static-libstdc++']
 | |
| 
 | |
|   def configure_mac(self, cxx):
 | |
|     cxx.defines += ['OSX', '_OSX', 'POSIX', 'KE_ABSOLUTELY_NO_STL']
 | |
|     cxx.cflags += ['-mmacosx-version-min=10.15']
 | |
|     cxx.linkflags += [
 | |
|       '-mmacosx-version-min=10.15',
 | |
|       '-stdlib=libc++',
 | |
|       '-lc++',
 | |
|     ]
 | |
|     cxx.cxxflags += ['-stdlib=libc++']
 | |
| 
 | |
|   def configure_windows(self, cxx):
 | |
|     cxx.defines += ['WIN32', '_WINDOWS']
 | |
|     
 | |
|   def LibraryBuilder(self, compiler, name):
 | |
|     binary = compiler.Library(name)
 | |
|     self.AddVersioning(binary)
 | |
|     if binary.compiler.like('msvc'):
 | |
|       binary.compiler.linkflags += ['/SUBSYSTEM:WINDOWS']
 | |
|     self.AddCxxCompat(binary)
 | |
|     return binary
 | |
| 
 | |
|   def Library(self, context, compiler, name):
 | |
|     compiler = compiler.clone()
 | |
|     SetArchFlags(compiler)
 | |
|     return self.LibraryBuilder(compiler, name)
 | |
| 
 | |
|   def ConfigureForExtension(self, context, compiler):
 | |
|     compiler.cxxincludes += [
 | |
|       os.path.join(context.currentSourcePath),
 | |
|       os.path.join(context.currentSourcePath, 'sdk'),
 | |
|       os.path.join(self.sm_root, 'public'),
 | |
|       os.path.join(self.sm_root, 'public', 'extensions'),
 | |
|       os.path.join(self.sm_root, 'sourcepawn', 'include'),
 | |
|       os.path.join(self.sm_root, 'public', 'amtl', 'amtl'),
 | |
|       os.path.join(self.sm_root, 'public', 'amtl'),
 | |
|     ]
 | |
|     return compiler
 | |
| 	
 | |
|   def AddCDetour(self, binary):
 | |
|     public_path = os.path.join(self.sm_root, 'public')
 | |
|     binary.sources += [ 
 | |
|       os.path.join(public_path, 'CDetour', 'detours.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'allocator.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'easy.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'inline_hook.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'mid_hook.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'os.linux.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'utility.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'src', 'vmt_hook.cpp'),
 | |
|       os.path.join(public_path, 'safetyhook', 'zydis', 'Zydis.c')
 | |
|     ]
 | |
| 	
 | |
|     binary.compiler.includes += [
 | |
|       os.path.join(public_path, 'safetyhook', 'include'),
 | |
|       os.path.join(public_path, 'safetyhook', 'zydis'),
 | |
|     ]
 | |
| 
 | |
|   def ExtLibrary(self, context, compiler, name):
 | |
|     binary = self.Library(context, compiler, name)
 | |
|     SetArchFlags(compiler)
 | |
|     self.ConfigureForExtension(context, binary.compiler)
 | |
|     return binary
 | |
| 
 | |
|   def AddCxxCompat(self, binary):
 | |
|     if binary.compiler.target.platform == 'linux':
 | |
|       binary.sources += [
 | |
|         os.path.join(self.sm_root, 'public', 'amtl', 'compat', 'stdcxx.cpp'),
 | |
|       ]
 | |
| 
 | |
|   def ConfigureForHL2(self, context, binary, sdk):
 | |
|     compiler = binary.compiler
 | |
|     SetArchFlags(compiler)
 | |
| 
 | |
|     compiler.cxxincludes += [
 | |
|       os.path.join(self.mms_root, 'core'),
 | |
|       os.path.join(self.mms_root, 'core', 'sourcehook'),
 | |
|     ]
 | |
| 
 | |
|     for other_sdk in self.sdk_manifests:
 | |
|       compiler.defines += ['SE_{}={}'.format(other_sdk['define'], other_sdk['code'])]
 | |
| 
 | |
|     SdkHelpers.configureCxx(context, binary, sdk)
 | |
| 
 | |
|     return binary
 | |
| 
 | |
|   def HL2Library(self, context, compiler, name, sdk):
 | |
|     binary = self.Library(context, compiler, name)
 | |
|     self.ConfigureForExtension(context, binary.compiler)
 | |
|     return self.ConfigureForHL2(context, binary, sdk)
 | |
| 
 | |
|   def HL2Config(self, project, context, compiler, name, sdk):
 | |
|     binary = project.Configure(compiler, name,
 | |
|                                '{0} - {1} {2}'.format(self.tag, sdk['name'], compiler.target.arch))
 | |
|     self.AddCxxCompat(binary)
 | |
|     return self.ConfigureForHL2(context, binary, sdk)
 | |
| 
 | |
|   def HL2ExtConfig(self, project, context, compiler, name, sdk):
 | |
|     binary = project.Configure(compiler, name,
 | |
|                                '{0} - {1} {2}'.format(self.tag, sdk['name'], compiler.target.arch))
 | |
|     self.AddCxxCompat(binary)
 | |
|     self.ConfigureForHL2(context, binary, sdk)
 | |
|     self.ConfigureForExtension(context, binary.compiler)
 | |
|     return binary
 | |
| 
 | |
| Extension = ExtensionConfig()
 | |
| Extension.detectSDKs()
 | |
| Extension.configure()
 | |
| 
 | |
| # This will clone the list and each cxx object as we recurse, preventing child
 | |
| # scripts from messing up global state.
 | |
| builder.targets = builder.CloneableList(Extension.all_targets)
 | |
| 
 | |
| BuildScripts = [
 | |
|   'AMBuilder',
 | |
|   'PackageScript',
 | |
| ]
 | |
| 
 | |
| builder.Build(BuildScripts, { 'Extension': Extension })
 | |
| 
 |