mirror of
https://github.com/github/codeql.git
synced 2025-12-17 17:23:36 +01:00
773ef62406
This checks in the trapgen script generating trap entries in C++. The codegen suite has been slightly reorganized, moving the templates directory up one level and chopping everything into smaller bazel packages. Running tests is now done via ``` bazel run //swift/codegen/test ``` With respect to the PoC, the nested `codeql::trap` namespace has been dropped in favour of a `Trap` prefix (or suffix in case of entries) within the `codeql` namespace. Also, generated C++ code is not checked in in git any more, and generated during build. Finally, labels get printed in hex in the trap file. `TrapLabel` is for the moment only default-constructible, so only one single label is possible. `TrapArena`, that is responsible for creating disjoint labels will come in a later commit.
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
""" generator options, categorized by tags """
|
|
|
|
import argparse
|
|
import collections
|
|
import pathlib
|
|
from typing import Set
|
|
|
|
from . import paths
|
|
|
|
|
|
def _init_options():
|
|
Option("--verbose", "-v", action="store_true")
|
|
Option("--schema", tags=["schema"], type=_abspath, default=paths.swift_dir / "codegen/schema.yml")
|
|
Option("--dbscheme", tags=["dbscheme"], type=_abspath, default=paths.swift_dir / "ql/lib/swift.dbscheme")
|
|
Option("--ql-output", tags=["ql"], type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/generated")
|
|
Option("--ql-stub-output", tags=["ql"], type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/elements")
|
|
Option("--codeql-binary", tags=["ql"], default="codeql")
|
|
Option("--trap-output", tags=["trap"], type=_abspath, required=True)
|
|
|
|
|
|
def _abspath(x):
|
|
return pathlib.Path(x).resolve()
|
|
|
|
|
|
_options = collections.defaultdict(list)
|
|
|
|
|
|
class Option:
|
|
def __init__(self, *args, tags=None, **kwargs):
|
|
tags = tags or []
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
if tags:
|
|
for t in tags:
|
|
_options[t].append(self)
|
|
else:
|
|
_options["*"].append(self)
|
|
|
|
def add_to(self, parser: argparse.ArgumentParser):
|
|
parser.add_argument(*self.args, **self.kwargs)
|
|
|
|
|
|
_init_options()
|
|
|
|
|
|
def get(tags: Set[str]):
|
|
""" get options marked by `tags`
|
|
|
|
Options tagged by wildcard '*' are always returned
|
|
"""
|
|
# use specifically tagged options + those tagged with wildcard *
|
|
return (o for tag in ('*',) + tuple(tags) for o in _options[tag])
|