| #!/usr/bin/env python |
| |
| # Copyright (c) 2013 Google Inc. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """ |
| Make sure binary is relinked when manifest settings are changed. |
| """ |
| |
| import TestGyp |
| |
| import os |
| import sys |
| |
| if sys.platform == 'win32': |
| import pywintypes |
| import win32api |
| import winerror |
| |
| RT_MANIFEST = 24 |
| |
| class LoadLibrary(object): |
| """Context manager for loading and releasing binaries in Windows. |
| Yields the handle of the binary loaded.""" |
| def __init__(self, path): |
| self._path = path |
| self._handle = None |
| |
| def __enter__(self): |
| self._handle = win32api.LoadLibrary(self._path) |
| return self._handle |
| |
| def __exit__(self, type, value, traceback): |
| win32api.FreeLibrary(self._handle) |
| |
| def extract_manifest(path, resource_name): |
| """Reads manifest from |path| and returns it as a string. |
| Returns None is there is no such manifest.""" |
| with LoadLibrary(path) as handle: |
| try: |
| return win32api.LoadResource(handle, RT_MANIFEST, resource_name) |
| except pywintypes.error as error: |
| if error.args[0] == winerror.ERROR_RESOURCE_DATA_NOT_FOUND: |
| return None |
| else: |
| raise |
| |
| test = TestGyp.TestGyp(formats=['msvs', 'ninja']) |
| |
| CHDIR = 'linker-flags' |
| |
| gyp_template = ''' |
| { |
| 'targets': [ |
| { |
| 'target_name': 'test_update_manifest', |
| 'type': 'executable', |
| 'sources': ['hello.cc'], |
| 'msvs_settings': { |
| 'VCLinkerTool': { |
| 'EnableUAC': 'true', |
| 'UACExecutionLevel': '%(uac_execution_level)d', |
| }, |
| 'VCManifestTool': { |
| 'EmbedManifest': 'true', |
| 'AdditionalManifestFiles': '%(additional_manifest_files)s', |
| }, |
| }, |
| }, |
| ], |
| } |
| ''' |
| |
| gypfile = 'update-manifest.gyp' |
| |
| def WriteAndUpdate(uac_execution_level, additional_manifest_files, do_build): |
| with open(os.path.join(CHDIR, gypfile), 'wb') as f: |
| f.write(gyp_template % { |
| 'uac_execution_level': uac_execution_level, |
| 'additional_manifest_files': additional_manifest_files, |
| }) |
| test.run_gyp(gypfile, chdir=CHDIR) |
| if do_build: |
| test.build(gypfile, chdir=CHDIR) |
| exe_file = test.built_file_path('test_update_manifest.exe', chdir=CHDIR) |
| return extract_manifest(exe_file, 1) |
| |
| manifest = WriteAndUpdate(0, '', True) |
| test.fail_test('asInvoker' not in manifest) |
| test.fail_test('35138b9a-5d96-4fbd-8e2d-a2440225f93a' in manifest) |
| |
| # Make sure that updating .gyp and regenerating doesn't cause a rebuild. |
| WriteAndUpdate(0, '', False) |
| test.up_to_date(gypfile, test.ALL, chdir=CHDIR) |
| |
| # But make sure that changing a manifest property does cause a relink. |
| manifest = WriteAndUpdate(2, '', True) |
| test.fail_test('requireAdministrator' not in manifest) |
| |
| # Adding a manifest causes a rebuild. |
| manifest = WriteAndUpdate(2, 'extra.manifest', True) |
| test.fail_test('35138b9a-5d96-4fbd-8e2d-a2440225f93a' not in manifest) |