mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-06 09:35:20 +00:00
141 lines
4.3 KiB
Python
Executable File
141 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2014-present, The osquery authors
|
|
#
|
|
# This source code is licensed as defined by the LICENSE file found in the
|
|
# root directory of this source tree.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR GPL-2.0-only)
|
|
|
|
import argparse
|
|
import json
|
|
import logging
|
|
import os
|
|
import shutil
|
|
|
|
logging_format = '[%(levelname)s] %(message)s'
|
|
logging.basicConfig(level=logging.INFO, format=logging_format)
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
REPO_ROOT_DIR = os.path.realpath(os.path.join(SCRIPT_DIR, "../.."))
|
|
|
|
|
|
def get_files_to_compile(json_data):
|
|
files_to_compile = []
|
|
for element in json_data:
|
|
filename = element["file"]
|
|
if not filename.endswith("tests.cpp") and \
|
|
not filename.endswith("benchmarks.cpp") and \
|
|
"example" not in filename and \
|
|
"generated/gen" not in filename and \
|
|
"test_util" not in filename:
|
|
base = filename.rfind("osquery/")
|
|
filename = filename[base + len("osquery/"):]
|
|
base_generated = filename.rfind("generated/")
|
|
if base_generated >= 0:
|
|
filename = filename[base_generated:]
|
|
files_to_compile.append(filename)
|
|
return files_to_compile
|
|
|
|
TARGETS_PREAMBLE = """\
|
|
# DO NOT EDIT
|
|
# Automatically generated: make sync
|
|
|
|
thrift_library(
|
|
name = "if",
|
|
languages = [
|
|
"cpp2",
|
|
"py",
|
|
],
|
|
py_base_module = "osquery",
|
|
thrift_cpp2_options = "stack_arguments",
|
|
thrift_srcs = {
|
|
"extensions.thrift": [
|
|
"Extension",
|
|
"ExtensionManager",
|
|
],
|
|
},
|
|
)
|
|
|
|
cpp_library(
|
|
name = "osquery_sdk",
|
|
srcs = [
|
|
"""
|
|
|
|
TARGETS_POSTSCRIPT = """ ],
|
|
headers = AutoHeaders.RECURSIVE_GLOB,
|
|
compiler_flags = [
|
|
"-Wno-unused-function",
|
|
"-Wno-non-virtual-dtor",
|
|
"-Wno-address",
|
|
"-Wno-overloaded-virtual",
|
|
"-Wno-unknown-pragmas",
|
|
],
|
|
link_whole = True,
|
|
propagated_pp_flags = [
|
|
"-DOSQUERY_BUILD_VERSION=%s-fb",
|
|
"-DOSQUERY_BUILD_SDK_VERSION=%s-fb",
|
|
"-DOSQUERY_BUILD_PLATFORM=centos7",
|
|
"-DOSQUERY_BUILD_DISTRO=centos7",
|
|
"-DOSQUERY_PLATFORM_MASK=9",
|
|
"-DFBTHRIFT",
|
|
"-DRAPIDJSON_HAS_STDSTRING=1",
|
|
],
|
|
deps = [
|
|
":if-cpp2",
|
|
"//folly/init:init",
|
|
"//rocksdb:rocksdb",
|
|
"//thrift/lib/cpp2:server",
|
|
"//thrift/lib/cpp2:thrift_base",
|
|
],
|
|
external_deps = [
|
|
("boost", None, "boost_filesystem"),
|
|
("boost", None, "boost_thread"),
|
|
"boost",
|
|
"glog",
|
|
"gflags",
|
|
("googletest", None, "gtest"),
|
|
("util-linux", None, "uuid"),
|
|
("rapidjson"),
|
|
],
|
|
)
|
|
"""
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description=(
|
|
"Generate a TARGETS files from CMake metadata"
|
|
))
|
|
parser.add_argument("--input", "-i", required=True)
|
|
parser.add_argument("--version", "-v", required=True)
|
|
parser.add_argument("--output", "-o", required=True)
|
|
parser.add_argument("--sources", "-s", required=True)
|
|
parser.add_argument("--sdk", required=True)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
with open(os.path.join(args.output, "TARGETS"), "w") as out:
|
|
with open(args.input, "r") as f:
|
|
try:
|
|
json_data = json.loads(f.read())
|
|
except ValueError:
|
|
logging.critical("Error: %s is not valid JSON" % args.input)
|
|
|
|
source_files = get_files_to_compile(json_data)
|
|
source_files.sort()
|
|
out.write(TARGETS_PREAMBLE)
|
|
for source_file in source_files:
|
|
if source_file == "extensions/impl_thrift.cpp":
|
|
source_file = "extensions/impl_fbthrift.cpp"
|
|
out.write(" \"%s\",\n" % source_file)
|
|
p = os.path.join(args.output, source_file)
|
|
if p.find("generated") < 0:
|
|
try:
|
|
os.makedirs(os.path.dirname(p), 0755)
|
|
except:
|
|
pass
|
|
shutil.copyfile(
|
|
os.path.join(args.sources, source_file), p)
|
|
out.write(TARGETS_POSTSCRIPT % (args.version, args.sdk))
|
|
|
|
except IOError as e:
|
|
logging.critical("Error: %s doesn't exist: %s" % (args.input, str(e)))
|