PRESUBMIT_VERSION = '2.0.0'
_PARTITION_ALLOC_BASE_PATH = 'base/allocator/partition_allocator/src/'
_SOURCE_FILE_PATTERN = r'.*\.(h|hpp|c|cc|cpp)$'
_BUILD_FILE_PATTERN = r'.*\.(gn|gni)$'
def CheckForIncludeGuards(input_api, output_api):
def guard_for_file(file):
local_path = file.LocalPath()
if input_api.is_windows:
local_path = local_path.replace('\\', '/')
assert local_path.startswith(_PARTITION_ALLOC_BASE_PATH)
guard = input_api.os_path.normpath(
local_path[len(_PARTITION_ALLOC_BASE_PATH):])
guard = guard + '_'
guard = guard.upper()
guard = input_api.re.sub(r'[+\\/.-]', '_', guard)
return guard
def is_partition_alloc_header_file(f):
return f.LocalPath().endswith('.h')
errors = []
for f in input_api.AffectedSourceFiles(is_partition_alloc_header_file):
expected_guard = guard_for_file(f)
guard_name_pattern = input_api.re.escape(expected_guard)
guard_pattern = input_api.re.compile(r'#ifndef\s+(' +
guard_name_pattern + ')')
guard_name = None
guard_line_number = None
seen_guard_end = False
for line_number, line in enumerate(f.NewContents()):
if guard_name is None:
match = guard_pattern.match(line)
if match:
guard_name = match.group(1)
guard_line_number = line_number
continue
if line_number == guard_line_number + 1:
expected_line = '#define %s' % guard_name
if line != expected_line:
errors.append(
output_api.PresubmitPromptWarning(
'Missing "%s" for include guard' % expected_line,
['%s:%d' % (f.LocalPath(), line_number + 1)],
'Expected: %r\nGot: %r' % (expected_line, line)))
if not seen_guard_end and line == '#endif // %s' % guard_name:
seen_guard_end = True
continue
if seen_guard_end:
if line.strip() != '':
errors.append(
output_api.PresubmitPromptWarning(
'Include guard %s not covering the whole file' %
(guard_name), [f.LocalPath()]))
break
if guard_name is None:
errors.append(
output_api.PresubmitPromptWarning(
'Missing include guard in %s\n'
'Recommended name: %s\n' %
(f.LocalPath(), expected_guard)))
return errors
def CheckNoExternalImportInGn(input_api, output_api):
import_re = input_api.re.compile(r'^ *import\("([^"]+)"\)')
sources = lambda affected_file: input_api.FilterSourceFile(
affected_file,
files_to_skip=[],
files_to_check=[_BUILD_FILE_PATTERN])
errors = []
for f in input_api.AffectedSourceFiles(sources):
for line_number, line in f.ChangedContents():
match = import_re.search(line)
if not match:
continue
import_path = match.group(1)
if import_path.startswith('//build_overrides/'):
continue
if not import_path.startswith('//'):
continue;
errors.append(output_api.PresubmitError(
'%s:%d\nPartitionAlloc disallow external import: %s' %
(f.LocalPath(), line_number + 1, import_path)))
return errors;
def CheckNoCpp23(input_api, output_api):
RAW_PATTERNS = [
(r'\bif\s+consteval\b', 'if consteval'),
(r'^\s*#\s*elifdef\b', '#elifdef'),
(r'^\s*#\s*elifndef\b', '#elifndef'),
(r'^\s*#\s*warning\b', '#warning'),
(r'\bimport\s+std\b', 'Standard Library Modules'),
(r'\bimport\s+std\.compat\b', 'Standard Library Modules'),
(r'\bthis\s+auto\b', 'Deducing this (explicit object parameter)'),
(r'\b\d+(?:u|U)?(?:z|Z)(?:u|U)?\b', 'size_t literal suffix (z/Z)'),
(r'\[\[assume\(.*\)\]\]', '[[assume(...)]] attribute'),
]
LIBRARY_SYMBOLS = [
'std::unreachable',
'std::to_underlying',
'std::byteswap',
'std::forward_like',
'std::move_only_function',
'std::invoke_r',
'std::start_lifetime_as',
'std::start_lifetime_as_array',
'std::out_ptr',
'std::inout_ptr',
'std::allocate_at_least',
'std::ranges::to',
'std::ranges::fold_left',
'std::ranges::fold_right',
'std::ranges::fold_left_first',
'std::ranges::fold_right_last',
'std::ranges::fold_left_with_iter',
'std::ranges::contains',
'views::zip',
'views::zip_transform',
'views::enumerate',
'views::chunk',
'views::chunk_by',
'views::slide',
'views::stride',
'views::join_with',
'views::adjacent',
'views::adjacent_transform',
'views::cartesian_product',
'views::as_rvalue',
'views::as_const',
'views::repeat',
]
compiled_checks = []
for pattern, feature_name in RAW_PATTERNS:
compiled_checks.append((input_api.re.compile(pattern), feature_name))
for symbol in LIBRARY_SYMBOLS:
pattern = r'\b' + input_api.re.escape(symbol) + r'\b'
description = f'C++23 library symbol: {symbol}'
compiled_checks.append((input_api.re.compile(pattern), description))
sources = lambda affected_file: input_api.FilterSourceFile(
affected_file,
files_to_skip=[],
files_to_check=[_SOURCE_FILE_PATTERN])
errors = []
for f in input_api.AffectedSourceFiles(sources):
for line_number, line in enumerate(f.NewContents()):
line_no_comment = line.split('//')[0]
for matcher, error_desc in compiled_checks:
if matcher.search(line_no_comment):
errors.append(
output_api.PresubmitError(
'%s:%d\nPartitionAlloc disallows C++23 feature: %s'
% (f.LocalPath(), line_number + 1, error_desc)))
return errors
def CheckNoCpp23Features(input_api, output_api):
CPP_23_PATTERNS = (
r'#include <(expected|flat_map|flat_set|generator|mdspan|print|'
r'spanstream|stacktrace|stdatomic.h|stdfloat)>',
r'if !?consteval',
r'^#elifn?def',
r'std::byteswap',
r'#warning',
r'static.*operator(\(\)|\[\])',
r'std::from_range',
r'\[\[assume[^[]*\]\]',
r'std::move_only_function',
r'std::unreachable',
r'std::(in)?out_ptr',
r'std::start_lifetime_as',
r'std::ranges::(contains|contains_subrange|starts_with|ends_with|'
r'find_last|find_last_if|find_last_if_not|iota|shift_left|'
r'shift_right|fold_left|fold_left_first|fold_right|fold_right_last|'
r'fold_left_with_iter|fold_left_first_with_iter)',
)
sources = lambda affected_file: input_api.FilterSourceFile(
affected_file,
files_to_skip=[
r'.*partition_alloc_base/augmentations/compiler_specific\.h'
],
files_to_check=[_SOURCE_FILE_PATTERN])
errors = []
for f in input_api.AffectedSourceFiles(sources):
for line_number, line in enumerate(f.NewContents()):
for pattern in CPP_23_PATTERNS:
match = input_api.re.search(pattern, line)
if not match:
continue
errors.append(
output_api.PresubmitError(
'%s:%d\nPartitionAlloc disallows C++23 features: `%s`'
% (f.LocalPath(), line_number + 1, match.group(0))))
return errors
def CheckNoNDebug(input_api, output_api):
sources = lambda affected_file: input_api.FilterSourceFile(
affected_file,
files_to_skip=[],
files_to_check=[_SOURCE_FILE_PATTERN])
errors = []
for f in input_api.AffectedSourceFiles(sources):
for line_number, line in f.ChangedContents():
if 'NDEBUG' in line:
errors.append(output_api.PresubmitError('%s:%d\nPartitionAlloc'
% (f.LocalPath(), line_number + 1)
+ 'disallows NDEBUG, use PA_BUILDFLAG(IS_DEBUG) instead'))
return errors