1//===- AMDGPUMetadataVerifier.h - MsgPack Types -----------------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9/// \file
10/// This is a verifier for AMDGPU HSA metadata, which can verify both
11/// well-typed metadata and untyped metadata. When verifying in the non-strict
12/// mode, untyped metadata is coerced into the correct type if possible.
13//
14//===----------------------------------------------------------------------===//
15
16#ifndef LLVM_BINARYFORMAT_AMDGPUMETADATAVERIFIER_H
17#define LLVM_BINARYFORMAT_AMDGPUMETADATAVERIFIER_H
18
19#include "llvm/BinaryFormat/MsgPackDocument.h"
20
21namespace llvm {
22namespace AMDGPU {
23namespace HSAMD {
24namespace V3 {
25
26/// Verifier for AMDGPU HSA metadata.
27///
28/// Operates in two modes:
29///
30/// In strict mode, metadata must already be well-typed.
31///
32/// In non-strict mode, metadata is coerced into expected types when possible.
33class MetadataVerifier {
34  bool Strict;
35
36  bool verifyScalar(msgpack::DocNode &Node, msgpack::Type SKind,
37                    function_ref<bool(msgpack::DocNode &)> verifyValue = {});
38  bool verifyInteger(msgpack::DocNode &Node);
39  bool verifyArray(msgpack::DocNode &Node,
40                   function_ref<bool(msgpack::DocNode &)> verifyNode,
41                   Optional<size_t> Size = None);
42  bool verifyEntry(msgpack::MapDocNode &MapNode, StringRef Key, bool Required,
43                   function_ref<bool(msgpack::DocNode &)> verifyNode);
44  bool
45  verifyScalarEntry(msgpack::MapDocNode &MapNode, StringRef Key, bool Required,
46                    msgpack::Type SKind,
47                    function_ref<bool(msgpack::DocNode &)> verifyValue = {});
48  bool verifyIntegerEntry(msgpack::MapDocNode &MapNode, StringRef Key,
49                          bool Required);
50  bool verifyKernelArgs(msgpack::DocNode &Node);
51  bool verifyKernel(msgpack::DocNode &Node);
52
53public:
54  /// Construct a MetadataVerifier, specifying whether it will operate in \p
55  /// Strict mode.
56  MetadataVerifier(bool Strict) : Strict(Strict) {}
57
58  /// Verify given HSA metadata.
59  ///
60  /// \returns True when successful, false when metadata is invalid.
61  bool verify(msgpack::DocNode &HSAMetadataRoot);
62};
63
64} // end namespace V3
65} // end namespace HSAMD
66} // end namespace AMDGPU
67} // end namespace llvm
68
69#endif // LLVM_BINARYFORMAT_AMDGPUMETADATAVERIFIER_H
70