Files
swift-mirror/utils/check_freestanding_size.py
Daniel Duan 3dfc40898c [NFC] Remove Python 2 imports from __future__ (#42086)
The `__future__` we relied on is now,  where the 3 specific things are
all included [since Python 3.0](https://docs.python.org/3/library/__future__.html):

* absolute_import
* print_function
* unicode_literals
* division

These import statements are no-ops and are no longer necessary.
2022-04-13 14:01:30 -07:00

68 lines
2.2 KiB
Python
Executable File

#!/usr/bin/env python3
#
# This source file is part of the Swift.org open source project
#
# Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
# Licensed under Apache License v2.0 with Runtime Library Exception
#
# See https://swift.org/LICENSE.txt for license information
# See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
import argparse
import subprocess
import sys
parser = argparse.ArgumentParser()
parser.add_argument("--path", help="path to a binary to check")
parser.add_argument("--triple", help="target triple of the freestanding stdlib")
parser.add_argument("--size-path", help="path to llvm-size binary to use")
args = parser.parse_args()
################################################################################
#
# EXPECTED/ALLOWED TEXT SEGMENT SIZE
#
# Before bumping the maximum text segment size please consult with:
# @kubamracek, @compnerd
#
# The 'freestanding' build of the Swift runtime and standard library is
# intended to be lean and allow dead-code elimination from the standard library
# based on client code usage.
#
################################################################################
# As of 2022-01-05: TEXT segment is 640 KiB on x86_64 Darwin, stdlib without
# asserts. Let's allow for some minor increases (~ 15%) and cap at 740 kiB:
max_text_size_x86_64_apple_macos = 740 * 1024
################################################################################
if args.triple == "x86_64-apple-macos":
max_text_size = max_text_size_x86_64_apple_macos
actual_text_size = 0
nm = args.size_path
lines = subprocess.check_output(
[nm, "--format=darwin", args.path]) \
.decode("utf-8").strip().splitlines()
for line in lines:
if line.startswith("Segment __TEXT: "):
actual_text_size = int(line[len("Segment __TEXT: "):])
if actual_text_size == 0:
print("cannot determine TEXT segment size")
sys.exit(1)
else:
print("triple {} not handled yet".format(args.triple))
sys.exit(1)
print("max_text_size: %d" % max_text_size)
print("actual_text_size: %d" % actual_text_size)
fail = actual_text_size > max_text_size
sys.exit(1 if fail else 0)