Index: build/mac/make_heap_non_executable.py |
=================================================================== |
--- build/mac/make_heap_non_executable.py (revision 97969) |
+++ build/mac/make_heap_non_executable.py (working copy) |
@@ -1,231 +0,0 @@ |
-#!/usr/bin/python |
- |
-# Copyright (c) 2011 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. |
- |
-# Usage: make_heap_non_executable.py <executable_path> |
-# |
-# Arranges for the executable at |executable_path| to have its data (heap) |
-# pages protected to prevent execution on Mac OS X 10.7 ("Lion"). |
-# |
-# Traditionally in Mac OS X, 32-bit processes did not have data pages set to |
-# prohibit execution. Although user programs could call mprotect and |
-# mach_vm_protect to deny execution of code in data pages, the kernel would |
-# silently ignore such requests without updating the page tables, and the |
-# hardware would happily execute code on such pages. 64-bit processes were |
-# always given proper hardware protection of data pages. This behavior was |
-# controllable on a system-wide level via the vm.allow_data_exec sysctl, which |
-# is set by default to 1. The bit with value 1 (set by default) allows code |
-# execution on data pages for 32-bit processes, and the bit with value 2 |
-# (clear by default) does the same for 64-bit processes. |
-# |
-# In Mac OS X 10.7, executables can "opt in" to having hardware protection |
-# against code execution on data pages applied. This is done by setting a new |
-# bit in the |flags| field of an executable's |mach_header|. When |
-# MH_NO_HEAP_EXECUTION is set, proper protections will be applied, regardless |
-# of the setting of vm.allow_data_exec. See xnu-1699.22.73/osfmk/vm/vm_map.c |
-# override_nx and xnu-1699.22.73/bsd/kern/mach_loader.c load_machfile. |
-# |
-# The Apple toolchain has been revised to set the MH_NO_HEAP_EXECUTION when |
-# producing executables, provided that -allow_heap_execute is not specified |
-# at link time. Only linkers shipping with Xcode 4.0 and later (ld64-123.2 and |
-# later) have this ability. See ld64-123.2.1/src/ld/Options.cpp |
-# Options::reconfigureDefaults() and |
-# ld64-123.2.1/src/ld/HeaderAndLoadCommands.hpp |
-# HeaderAndLoadCommandsAtom<A>::flags(). |
-# |
-# This script sets the MH_NO_HEAP_EXECUTION bit on Mach-O executables. It is |
-# intended for use with executables produced by a linker that predates Apple's |
-# modifications to set this bit itself. It is also useful for setting this bit |
-# for non-i386 executables, including x86_64 executables. Apple's linker only |
-# sets it for 32-bit i386 executables, presumably under the assumption that |
-# the value of vm.allow_data_exec is set in stone. However, if someone were to |
-# change vm.allow_data_exec to 2 or 3, 64-bit x86_64 executables would run |
-# without hardware protection against code execution on data pages. This |
-# script can set the bit for x86_64 executables, guaranteeing that they run |
-# with appropriate protection even when vm.allow_data_exec has been tampered |
-# with. |
-# |
-# This script is able to operate on thin (single-architecture) Mach-O files |
-# and fat (universal, multi-architecture) files. When operating on fat files, |
-# it will set the MH_NO_HEAP_EXECUTION bit for each architecture contained |
-# therein. |
- |
- |
-import os |
-import struct |
-import sys |
- |
- |
-# <mach-o/fat.h> |
-FAT_MAGIC = 0xcafebabe |
-FAT_CIGAM = 0xbebafeca |
- |
-# <mach-o/loader.h> |
-MH_MAGIC = 0xfeedface |
-MH_CIGAM = 0xcefaedfe |
-MH_MAGIC_64 = 0xfeedfacf |
-MH_CIGAM_64 = 0xcffaedfe |
-MH_EXECUTE = 0x2 |
-MH_NO_HEAP_EXECUTION = 0x1000000 |
- |
- |
-class MachOError(Exception): |
- """A class for exceptions thrown by this module.""" |
- |
- pass |
- |
- |
-def CheckedSeek(file, offset): |
- """Seeks the file-like object at |file| to offset |offset| and raises a |
- MachOError if anything funny happens.""" |
- |
- file.seek(offset, os.SEEK_SET) |
- new_offset = file.tell() |
- if new_offset != offset: |
- raise MachOError, \ |
- 'seek: expected offset %d, observed %d' % (offset, new_offset) |
- |
- |
-def CheckedRead(file, count): |
- """Reads |count| bytes from the file-like |file| object, raising a |
- MachOError if any other number of bytes is read.""" |
- |
- bytes = file.read(count) |
- if len(bytes) != count: |
- raise MachOError, \ |
- 'read: expected length %d, observed %d' % (count, len(bytes)) |
- |
- return bytes |
- |
- |
-def ReadUInt32(file, endian): |
- """Reads an unsinged 32-bit integer from the file-like |file| object, |
- treating it as having endianness specified by |endian| (per the |struct| |
- module), and returns it as a number. Raises a MachOError if the proper |
- length of data can't be read from |file|.""" |
- |
- bytes = CheckedRead(file, 4) |
- |
- (uint32,) = struct.unpack(endian + 'I', bytes) |
- return uint32 |
- |
- |
-def ReadMachHeader(file, endian): |
- """Reads an entire |mach_header| structure (<mach-o/loader.h>) from the |
- file-like |file| object, treating it as having endianness specified by |
- |endian| (per the |struct| module), and returns a 7-tuple of its members |
- as numbers. Raises a MachOError if the proper length of data can't be read |
- from |file|.""" |
- |
- bytes = CheckedRead(file, 28) |
- |
- magic, cputype, cpusubtype, filetype, ncmds, sizeofcmds, flags = \ |
- struct.unpack(endian + '7I', bytes) |
- return magic, cputype, cpusubtype, filetype, ncmds, sizeofcmds, flags |
- |
- |
-def ReadFatArch(file): |
- """Reads an entire |fat_arch| structure (<mach-o/fat.h>) from the file-like |
- |file| object, treating it as having endianness specified by |endian| |
- (per the |struct| module), and returns a 5-tuple of its members as numbers. |
- Raises a MachOError if the proper length of data can't be read from |
- |file|.""" |
- |
- bytes = CheckedRead(file, 20) |
- |
- cputype, cpusubtype, offset, size, align = struct.unpack('>5I', bytes) |
- return cputype, cpusubtype, offset, size, align |
- |
- |
-def WriteUInt32(file, uint32, endian): |
- """Writes |uint32| as an unsinged 32-bit integer to the file-like |file| |
- object, treating it as having endianness specified by |endian| (per the |
- |struct| module).""" |
- |
- bytes = struct.pack(endian + 'I', uint32) |
- assert len(bytes) == 4 |
- |
- file.write(bytes) |
- |
- |
-def HandleMachOFile(file, offset=0): |
- """Seeks the file-like |file| object to |offset|, reads its |mach_header|, |
- and rewrites the header's |flags| field if appropriate. The header's |
- endianness is detected. Both 32-bit and 64-bit Mach-O headers are supported |
- (mach_header and mach_header_64). Raises MachOError if used on a header that |
- does not have a known magic number or is not of type MH_EXECUTE. The |
- MH_NO_HEAP_EXECUTION is set in the |flags| field and written to |file| if |
- not already set. If already set, nothing is written.""" |
- |
- CheckedSeek(file, offset) |
- magic = ReadUInt32(file, '<') |
- if magic == MH_MAGIC or magic == MH_MAGIC_64: |
- endian = '<' |
- elif magic == MH_CIGAM or magic == MH_CIGAM_64: |
- endian = '>' |
- else: |
- raise MachOError, \ |
- 'Mach-O file at offset %d has illusion of magic' % offset |
- |
- CheckedSeek(file, offset) |
- magic, cputype, cpusubtype, filetype, ncmds, sizeofcmds, flags = \ |
- ReadMachHeader(file, endian) |
- assert magic == MH_MAGIC or magic == MH_MAGIC_64 |
- if filetype != MH_EXECUTE: |
- raise MachOError, \ |
- 'Mach-O file at offset %d is type 0x%x, expected MH_EXECUTE' % \ |
- (offset, filetype) |
- |
- if not flags & MH_NO_HEAP_EXECUTION: |
- flags |= MH_NO_HEAP_EXECUTION |
- CheckedSeek(file, offset + 24) |
- WriteUInt32(file, flags, endian) |
- |
- |
-def HandleFatFile(file, fat_offset=0): |
- """Seeks the file-like |file| object to |offset| and loops over its |
- |fat_header| entries, calling HandleMachOFile for each.""" |
- |
- CheckedSeek(file, fat_offset) |
- magic = ReadUInt32(file, '>') |
- assert magic == FAT_MAGIC |
- |
- nfat_arch = ReadUInt32(file, '>') |
- |
- for index in xrange(0, nfat_arch): |
- cputype, cpusubtype, offset, size, align = ReadFatArch(file) |
- assert size >= 28 |
- |
- # HandleMachOFile will seek around. Come back here after calling it, in |
- # case it sought. |
- fat_arch_offset = file.tell() |
- HandleMachOFile(file, offset) |
- CheckedSeek(file, fat_arch_offset) |
- |
- |
-def main(me, args): |
- if len(args) != 1: |
- print >>sys.stderr, 'usage: %s <executable_path>' % me |
- return 1 |
- |
- executable_path = args[0] |
- executable_file = open(executable_path, 'rb+') |
- |
- magic = ReadUInt32(executable_file, '<') |
- if magic == FAT_CIGAM: |
- # Check FAT_CIGAM and not FAT_MAGIC because the read was little-endian. |
- HandleFatFile(executable_file) |
- elif magic == MH_MAGIC or magic == MH_CIGAM or \ |
- magic == MH_MAGIC_64 or magic == MH_CIGAM_64: |
- HandleMachOFile(executable_file) |
- else: |
- raise MachOError, '%s is not a Mach-O or fat file' % executable_file |
- |
- executable_file.close() |
- |
- return 0 |
- |
-if __name__ == '__main__': |
- sys.exit(main(sys.argv[0], sys.argv[1:])) |