gyptest-cl-buffer-security-check.py

Enable keyboard shortcuts

#!/usr/bin/env python

# Copyright (c) 2012 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 buffer security check setting is extracted properly.

"""

import TestGyp

import sys

if sys.platform == 'win32':

  test = TestGyp.TestGyp(formats=['msvs', 'ninja'])

  CHDIR = 'compiler-flags'

  test.run_gyp('buffer-security-check.gyp', chdir=CHDIR)

  test.build('buffer-security-check.gyp', chdir=CHDIR)

  def GetDisassemblyOfMain(exe):

    # The standard library uses buffer security checks independent of our

    # buffer security settings, so we extract just our code (i.e. main()) to

    # check against.

    full_path = test.built_file_path(exe, chdir=CHDIR)

    output = test.run_dumpbin('/disasm', full_path)

    result = []

    in_main = False

    for line in output.splitlines():

      if line == '_main:':

        in_main = True

      elif in_main:

        # Disassembly of next function starts.

        if line.startswith('_'):

          break

        result.append(line)

    return '\n'.join(result)

  # Buffer security checks are on by default, make sure security_cookie

  # appears in the disassembly of our code.

  if 'security_cookie' not in GetDisassemblyOfMain('test_bsc_unset.exe'):

    test.fail_test()

  # Explicitly on.

  if 'security_cookie' not in GetDisassemblyOfMain('test_bsc_on.exe'):

    test.fail_test()

  # Explicitly off, shouldn't be a reference to the security cookie.

  if 'security_cookie' in GetDisassemblyOfMain('test_bsc_off.exe'):

    test.fail_test()

  test.pass_test()