ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-16 11:58:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

clang-format pip

Browse Source
This commit is contained in:
Maciej Domanski
2022-12-28 13:05:44 +01:00
parent 342081d5ee
commit 1bcc9bfa91
1 changed files with 58 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
.custom-format.py
View File

@ -3,7 +3,7 @@
# american fuzzy lop++ - custom code formatter # american fuzzy lop++ - custom code formatter
# -------------------------------------------- # --------------------------------------------
# #
# Written and maintaned by Andrea Fioraldi <andreafioraldi@gmail.com> # Written and maintained by Andrea Fioraldi <andreafioraldi@gmail.com>
# #
# Copyright 2015, 2016, 2017 Google Inc. All rights reserved. # Copyright 2015, 2016, 2017 Google Inc. All rights reserved.
# Copyright 2019-2022 AFLplusplus Project. All rights reserved. # Copyright 2019-2022 AFLplusplus Project. All rights reserved.
@ -18,24 +18,57 @@
import subprocess import subprocess
import sys import sys
import os import os
import re # import re # TODO: for future use
import shutil import shutil
import importlib.metadata
# string_re = re.compile('(\\"(\\\\.|[^"\\\\])*\\")') # future use # string_re = re.compile('(\\"(\\\\.|[^"\\\\])*\\")') # TODO: for future use
with open(".clang-format") as f:
fmt = f.read()
CURRENT_LLVM = os.getenv('LLVM_VERSION', 14) CURRENT_LLVM = os.getenv('LLVM_VERSION', 14)
CLANG_FORMAT_BIN = os.getenv("CLANG_FORMAT_BIN", "") CLANG_FORMAT_BIN = os.getenv("CLANG_FORMAT_BIN", "")
def check_clang_format_pip_version():
"""
Check if the correct version of clang-format is installed via pip.
Returns:
bool: True if the correct version of clang-format is installed,
False otherwise.
"""
# Check if clang-format is installed
if importlib.util.find_spec('clang_format'):
# Check if the installed version is the expected LLVM version
if importlib.metadata.version('clang-format')\
.startswith(CURRENT_LLVM+'.'):
return True
else:
# Return False, because the clang-format version does not match
return False
else:
# If the 'clang_format' package isn't installed, return False
return False
with open(".clang-format") as f:
fmt = f.read()
CLANG_FORMAT_PIP = check_clang_format_pip_version()
if shutil.which(CLANG_FORMAT_BIN) is None: if shutil.which(CLANG_FORMAT_BIN) is None:
CLANG_FORMAT_BIN = f"clang-format-{CURRENT_LLVM}" CLANG_FORMAT_BIN = f"clang-format-{CURRENT_LLVM}"
if shutil.which(CLANG_FORMAT_BIN) is None: if shutil.which(CLANG_FORMAT_BIN) is None \
and CLANG_FORMAT_PIP is False:
print(f"[!] clang-format-{CURRENT_LLVM} is needed. Aborted.") print(f"[!] clang-format-{CURRENT_LLVM} is needed. Aborted.")
print(f"Run `pip3 install \"clang-format=={CURRENT_LLVM}.*\"` \
to install via pip.")
exit(1) exit(1)
if CLANG_FORMAT_PIP:
CLANG_FORMAT_BIN = shutil.which("clang-format")
COLUMN_LIMIT = 80 COLUMN_LIMIT = 80
for line in fmt.split("\n"): for line in fmt.split("\n"):
line = line.split(":") line = line.split(":")
@ -54,7 +87,7 @@ def custom_format(filename):
for line in src.split("\n"): for line in src.split("\n"):
if line.lstrip().startswith("#"): if line.lstrip().startswith("#"):
if line[line.find("#") + 1 :].lstrip().startswith("define"): if line[line.find("#") + 1:].lstrip().startswith("define"):
in_define = True in_define = True
if ( if (
@ -73,7 +106,7 @@ def custom_format(filename):
define_padding = 0 define_padding = 0
if last_line is not None and in_define and last_line.endswith("\\"): if last_line is not None and in_define and last_line.endswith("\\"):
last_line = last_line[:-1] last_line = last_line[:-1]
define_padding = max(0, len(last_line[last_line.rfind("\n") + 1 :])) define_padding = max(0, len(last_line[last_line.rfind("\n") + 1:]))
if ( if (
last_line is not None last_line is not None