import re
from functools import reduce
from pathlib import PurePosixPath
STDINT_SIZES = [
"16",
"32",
"64",
"8",
"least16",
"least32",
"least64",
"least8",
"max",
"ptr",
]
COMPILER_HEADER_TYPES = {
"bool": "<stdbool.h>",
"va_list": "<stdarg.h>",
}
COMPILER_HEADER_TYPES.update({f"int{size}_t": "<stdint.h>" for size in STDINT_SIZES})
COMPILER_HEADER_TYPES.update({f"uint{size}_t": "<stdint.h>" for size in STDINT_SIZES})
NONIDENTIFIER = re.compile("[^a-zA-Z0-9_]+")
COMMON_HEADER = PurePosixPath("__llvm-libc-common.h")
COMMON_ATTRIBUTES = {
"_Noreturn",
"_Returns_twice",
}
LIBRARY_DESCRIPTIONS = {
"stdc": "Standard C",
"posix": "POSIX",
"bsd": "BSD",
"gnu": "GNU",
"linux": "Linux",
"uefi": "UEFI",
"svid": "SVID",
}
HEADER_TEMPLATE = """\
//===-- {library} header <{header}> --===//
//
{license_lines}
//
//===---------------------------------------------------------------------===//
#ifndef {guard}
#define {guard}
%%public_api()
#endif // {guard}
"""
LLVM_LICENSE_TEXT = [
"Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.",
"See https://llvm.org/LICENSE.txt for license information.",
"SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception",
]
PROXY_TEMPLATE = """\
//===-- Implementation proxy header for <{header}> --===//
//
{license_lines}
//
//===---------------------------------------------------------------------===//
#ifndef {guard}
#define {guard}
#ifdef LIBC_FULL_BUILD
{include_lines}
{macro_lines}
#else // Overlay mode
#include <{header}>
#endif // LLVM_LIBC_FULL_BUILD
#endif // {guard}
"""
class HeaderFile:
def __init__(self, name):
self.template_file = None
self.name = name
self.macros = []
self.types = []
self.enumerations = []
self.objects = []
self.functions = []
self.extra_standards = {}
self.standards = []
self.merge_yaml_files = []
self.license_text = []
def add_macro(self, macro):
self.macros.append(macro)
def add_type(self, type_):
self.types.append(type_)
def add_enumeration(self, enumeration):
self.enumerations.append(enumeration)
def add_object(self, object):
self.objects.append(object)
def add_function(self, function):
self.functions.append(function)
def merge(self, other):
self.macros = sorted(set(self.macros) | set(other.macros))
self.types = sorted(set(self.types) | set(other.types))
self.enumerations = sorted(set(self.enumerations) | set(other.enumerations))
self.objects = sorted(set(self.objects) | set(other.objects))
self.functions = sorted(set(self.functions) | set(other.functions))
self.extra_standards |= other.extra_standards
if self.license_text:
assert not other.license_text, "only one `license_text` allowed"
else:
self.license_text = other.license_text
def all_types(self):
return reduce(
lambda a, b: a | b,
[f.signature_types() for f in self.functions],
set(self.types),
)
def all_attributes(self):
return reduce(
lambda a, b: a | b,
[set(f.attributes) for f in self.functions],
set(),
)
def all_standards(self):
return set(self.standards).union(
*(filter(None, (f.standards for f in self.functions)))
)
def includes(self):
return (
{
PurePosixPath("llvm-libc-macros") / macro.header
for macro in self.macros
if macro.header is not None
}
| {
COMPILER_HEADER_TYPES.get(
typ.name,
PurePosixPath("llvm-libc-types") / f"{typ.name}.h",
)
for typ in self.all_types()
}
| {
PurePosixPath("llvm-libc-macros") / f"{attr}.h"
for attr in self.all_attributes() - COMMON_ATTRIBUTES
}
)
def header_guard(self, proxy=False):
words = [word.upper() for word in NONIDENTIFIER.split(self.name) if word]
if proxy:
return "LLVM_LIBC_HDR_" + "_".join(words[:-1]) + "_PROXY_H"
return "_LLVM_LIBC_" + "_".join(words)
def library_description(self):
descriptions = LIBRARY_DESCRIPTIONS | self.extra_standards
if "stdc" in self.standards:
return descriptions["stdc"]
if "posix" in self.standards:
return descriptions["posix"]
standards = self.all_standards()
return " / ".join(
sorted(
descriptions[standard]
for standard in standards
if standard not in {"stdc", "posix"}
)
)
def license_lines(self):
lines = self.license_text or LLVM_LICENSE_TEXT
return "\n".join([f"// {line}" for line in lines])
def template(self, dir, files_read):
if self.template_file is not None:
template_path = dir / self.template_file
files_read.add(template_path)
return template_path.read_text()
return HEADER_TEMPLATE.format(
library=self.library_description(),
header=self.name,
guard=self.header_guard(),
license_lines=self.license_lines(),
)
def include_lines(self, with_common=False):
path_prefix = PurePosixPath("../" * (len(PurePosixPath(self.name).parents) - 1))
def relpath(file):
return path_prefix / file
return [
f"#include {file}"
for file in ([f'"{relpath(COMMON_HEADER)!s}"'] if with_common else [])
+ sorted(
file if isinstance(file, str) else f'"{relpath(file)!s}"'
for file in self.includes()
)
]
def macro_lines(self):
content = []
for macro in sorted(self.macros):
if str(macro):
content.extend(["", f"{macro}"])
return content
def enum_lines(self):
content = []
if self.enumerations:
combined_enum_content = ",\n ".join(
str(enum) for enum in self.enumerations
)
content.append(f"\nenum {{\n {combined_enum_content},\n}};")
return content
def proxy_contents(self):
return PROXY_TEMPLATE.format(
header=self.name,
guard=self.header_guard(proxy=True),
license_lines=self.license_lines(),
include_lines="\n".join(self.include_lines()),
macro_lines="\n".join(self.macro_lines()),
)
def public_api(self):
content = (
self.include_lines(self.template_file is None)
+ self.macro_lines()
+ self.enum_lines()
+ ["\n__BEGIN_C_DECLS\n"]
)
current_guard = None
last_name = None
for function in sorted(self.functions):
if last_name == function.name_without_underscores():
content.pop()
if function.guard == None and current_guard == None:
content.append(str(function) + " __NOEXCEPT;")
content.append("")
else:
if current_guard == None:
current_guard = function.guard
content.append(f"#ifdef {current_guard}")
content.append(str(function) + " __NOEXCEPT;")
content.append("")
elif current_guard == function.guard:
content.append(str(function) + " __NOEXCEPT;")
content.append("")
else:
content.pop()
content.append(f"#endif // {current_guard}")
content.append("")
current_guard = function.guard
if current_guard is not None:
content.append(f"#ifdef {current_guard}")
content.append(str(function) + " __NOEXCEPT;")
content.append("")
last_name = function.name_without_underscores()
if current_guard != None:
content.pop()
content.append(f"#endif // {current_guard}")
content.append("")
content.extend(str(object) for object in self.objects)
if self.objects:
content.append("")
content.append("__END_C_DECLS")
return "\n".join(content)
def json_data(self):
return {
"name": self.name,
"standards": self.standards,
"includes": sorted(str(file) for file in {COMMON_HEADER} | self.includes()),
}