// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
// MATERIALS OR THE USE OR OTHER DEALINGS IN THE MATERIALS.
+#include "UnitSPIRV.h"
+
+#include "TestFixture.h"
+#include "gmock/gmock.h"
+
+// Returns true if two spv_parsed_operand_t values are equal.
+// To use this operator, this definition must appear in the same namespace
+// as spv_parsed_operand_t.
+static bool operator==(const spv_parsed_operand_t& a,
+ const spv_parsed_operand_t& b) {
+ return a.offset == b.offset && a.num_words == b.num_words &&
+ a.type == b.type && a.number_kind == b.number_kind &&
+ a.number_bit_width == b.number_bit_width;
+}
+
namespace {
+using ::spvtest::Concatenate;
+using ::spvtest::MakeInstruction;
+using ::spvtest::MakeVector;
+using ::testing::_;
+using ::testing::AnyOf;
+using ::testing::ElementsAre;
+using ::testing::Eq;
+
+// An easily-constructible and comparable object for the contents of an
+// spv_parsed_instruction_t.
+struct ParsedInstruction {
+ ParsedInstruction(const spv_parsed_instruction_t& inst)
+ : words(inst.words, inst.words + inst.num_words),
+ opcode(inst.opcode),
+ ext_inst_type(inst.ext_inst_type),
+ type_id(inst.type_id),
+ result_id(inst.result_id),
+ operands(inst.operands, inst.operands + inst.num_operands) {}
+
+ std::vector<uint32_t> words;
+ SpvOp opcode;
+ spv_ext_inst_type_t ext_inst_type;
+ uint32_t type_id;
+ uint32_t result_id;
+ std::vector<spv_parsed_operand_t> operands;
+
+ bool operator==(const ParsedInstruction& b) const {
+ return words == b.words && opcode == b.opcode &&
+ ext_inst_type == b.ext_inst_type && type_id == b.type_id &&
+ result_id == b.result_id && operands == b.operands;
+ }
+};
+
+// Prints a ParsedInstruction object to the given output stream, and returns
+// the stream.
+std::ostream& operator<<(std::ostream& os, const ParsedInstruction& inst) {
+ os << "\nParsedInstruction( {";
+ spvtest::PrintTo(spvtest::WordVector(inst.words), &os);
+ os << "}, opcode: " << int(inst.opcode)
+ << " ext_inst_type: " << int(inst.ext_inst_type)
+ << " type_id: " << inst.type_id << " result_id: " << inst.result_id;
+ for (const auto& operand : inst.operands) {
+ os << " { offset: " << operand.offset << " num_words: " << operand.num_words
+ << " type: " << int(operand.type)
+ << " number_kind: " << int(operand.number_kind)
+ << " number_bit_width: " << int(operand.number_bit_width) << "}";
+ }
+ return os;
+}
+
+// Sanity check for the equality operator on ParsedInstruction.
+TEST(ParsedInstruction, ZeroInitializedAreEqual) {
+ spv_parsed_instruction_t pi = {};
+ ParsedInstruction a(pi);
+ ParsedInstruction b(pi);
+ EXPECT_THAT(a, ::testing::TypedEq<ParsedInstruction>(b));
+}
+
+// A template class with static member functions that forward to non-static
+// member functions corresponding to the header and instruction callbacks
+// for spvBinaryParse.
+template <typename Client, typename T>
+class BinaryParseClient : public T {
+ public:
+ static spv_result_t Header(void* user_data, spv_endianness_t endian,
+ uint32_t magic, uint32_t version,
+ uint32_t generator, uint32_t id_bound,
+ uint32_t reserved) {
+ return static_cast<Client*>(user_data)
+ ->HandleHeader(endian, magic, version, generator, id_bound, reserved);
+ }
+
+ static spv_result_t Instruction(
+ void* user_data, const spv_parsed_instruction_t* parsed_instruction) {
+ return static_cast<Client*>(user_data)
+ ->HandleInstruction(parsed_instruction);
+ }
+};
+
+using Words = std::vector<uint32_t>;
+using Endians = std::vector<spv_endianness_t>;
+using Sentences = std::vector<Words>; // Maybe this is too cute?
+using Instructions = std::vector<ParsedInstruction>;
+
+// A binary parse client that captures the results of parsing a binary,
+// and whose callbacks can be made to succeed for a specified number of
+// times, and then always fail with a given failure code.
+template <typename T>
+class CaptureParseResults
+ : public BinaryParseClient<CaptureParseResults<T>, T> {
+ public:
+ // Capture the header from the parser callback.
+ // If the number of callback successes has not yet been exhausted, then
+ // returns SPV_SUCCESS, which itself counts as a callback success.
+ // Otherwise returns the stored failure code.
+ virtual spv_result_t HandleHeader(spv_endianness_t endian, uint32_t magic,
+ uint32_t version, uint32_t generator,
+ uint32_t id_bound, uint32_t reserved) {
+ endians_.push_back(endian);
+ headers_.push_back({magic, version, generator, id_bound, reserved});
+ return ComputeResultCode();
+ }
+
+ // Capture the parsed instruction data from the parser callback.
+ // If the number of callback successes has not yet been exhausted, then
+ // returns SPV_SUCCESS, which itself counts as a callback success.
+ // Otherwise returns the stored failure code.
+ virtual spv_result_t HandleInstruction(
+ const spv_parsed_instruction_t* parsed_instruction) {
+ EXPECT_NE(nullptr, parsed_instruction);
+ instructions_.emplace_back(*parsed_instruction);
+ return ComputeResultCode();
+ }
+
+ // Getters
+ const Endians& endians() const { return endians_; }
+ const Sentences& headers() const { return headers_; }
+ const Instructions& instructions() const { return instructions_; }
+
+ protected:
+ // Returns the appropriate result code based on whether we still have more
+ // successes to return. Decrements the number of successes still remaining,
+ // if needed.
+ spv_result_t ComputeResultCode() {
+ if (num_passing_callbacks_ < 0) return SPV_SUCCESS;
+ if (num_passing_callbacks_ == 0) return fail_code_;
+ num_passing_callbacks_--;
+ return SPV_SUCCESS;
+ }
+
+ // How many callbacks should succeed before they start failing?
+ // If this is negative, then all callbacks should pass.
+ int num_passing_callbacks_ = -1; // By default, never fail.
+ // The result code to use on callback failure.
+ spv_result_t fail_code_ = SPV_ERROR_INVALID_BINARY;
+
+ // Accumulated results for calls to HandleHeader.
+ Endians endians_;
+ Sentences headers_;
+
+ // Accumulated results for calls to HandleHeader.
+ Instructions instructions_;
+};
+
+// Returns the expected SPIR-V module header words for the Khronos
+// Assembler generator, and with a given Id bound.
+Words ExpectedHeaderForBound(uint32_t bound) {
+ return {SpvMagicNumber, SpvVersion,
+ SPV_GENERATOR_WORD(SPV_GENERATOR_KHRONOS_ASSEMBLER, 0), bound, 0};
+}
+
+// Returns a parsed operand for a non-number value at the given word offset
+// within an instruction.
+spv_parsed_operand_t MakeSimpleOperand(uint16_t offset,
+ spv_operand_type_t type) {
+ return {offset, 1, type, SPV_NUMBER_NONE, 0};
+}
+
+// Returns a parsed operand for a literal unsigned integer value at the given
+// word offset within an instruction.
+spv_parsed_operand_t MakeLiteralNumberOperand(uint16_t offset) {
+ return {offset, 1, SPV_OPERAND_TYPE_LITERAL_INTEGER, SPV_NUMBER_UNSIGNED_INT,
+ 32};
+}
+
+// Returns a parsed operand for a literal string value at the given
+// word offset within an instruction.
+spv_parsed_operand_t MakeLiteralStringOperand(uint16_t offset,
+ uint16_t length) {
+ return {offset, length, SPV_OPERAND_TYPE_LITERAL_STRING, SPV_NUMBER_NONE, 0};
+}
+
+// Returns a ParsedInstruction for an OpTypeVoid instruction that would
+// generate the given result Id.
+ParsedInstruction MakeParsedVoidTypeInstruction(uint32_t result_id) {
+ const auto void_inst = MakeInstruction(SpvOpTypeVoid, {result_id});
+ const auto void_operands = std::vector<spv_parsed_operand_t>{
+ MakeSimpleOperand(1, SPV_OPERAND_TYPE_RESULT_ID)};
+ const spv_parsed_instruction_t parsed_void_inst = {
+ void_inst.data(),
+ uint16_t(void_inst.size()),
+ SpvOpTypeVoid,
+ SPV_EXT_INST_TYPE_NONE,
+ 0, // type id
+ result_id,
+ void_operands.data(),
+ uint16_t(void_operands.size())};
+ return ParsedInstruction(parsed_void_inst);
+}
+
+// Returns a ParsedInstruction for an OpTypeInt instruction that generates
+// the given result Id for a 32-bit signed integer scalar type.
+ParsedInstruction MakeParsedInt32TypeInstruction(uint32_t result_id) {
+ const auto i32_inst = MakeInstruction(SpvOpTypeInt, {result_id, 32, 1});
+ const auto i32_operands = std::vector<spv_parsed_operand_t>{
+ MakeSimpleOperand(1, SPV_OPERAND_TYPE_RESULT_ID),
+ MakeLiteralNumberOperand(2), MakeLiteralNumberOperand(3)};
+ spv_parsed_instruction_t parsed_i32_inst = {i32_inst.data(),
+ uint16_t(i32_inst.size()),
+ SpvOpTypeInt,
+ SPV_EXT_INST_TYPE_NONE,
+ 0, // type id
+ result_id,
+ i32_operands.data(),
+ uint16_t(i32_operands.size())};
+ return ParsedInstruction(parsed_i32_inst);
+}
+
+using BinaryParseTest =
+ spvtest::TextToBinaryTestBase<CaptureParseResults<::testing::Test>>;
+
+TEST_F(BinaryParseTest, EmptyModuleHasValidHeaderAndNoInstructionCallbacks) {
+ const auto binary = CompileSuccessfully("");
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_SUCCESS,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ EXPECT_EQ(nullptr, diagnostic);
+ EXPECT_THAT(endians(), AnyOf(Eq(Endians{SPV_ENDIANNESS_LITTLE}),
+ Eq(Endians{SPV_ENDIANNESS_BIG})));
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(1)}));
+ EXPECT_THAT(instructions(), Eq(Instructions{}));
+}
+
+// Check the result of multiple instruction callbacks.
+//
+// This test exercises non-default values for the following members of the
+// spv_parsed_instruction_t struct: words, num_words, opcode, result_id,
+// operands, num_operands.
+TEST_F(BinaryParseTest, TwoScalarTypesGenerateTwoInstructionCallbacks) {
+ const auto binary = CompileSuccessfully(
+ "%1 = OpTypeVoid "
+ "%2 = OpTypeInt 32 1");
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_SUCCESS,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ EXPECT_EQ(nullptr, diagnostic);
+
+ // The Id bound must be computed correctly. The module has two generated Ids,
+ // so the bound is 3.
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(3)}));
+
+ // The instructions callbacks must have the correct data.
+ EXPECT_THAT(instructions(), Eq(Instructions{
+ MakeParsedVoidTypeInstruction(1),
+ MakeParsedInt32TypeInstruction(2),
+ }));
+}
+
+TEST_F(BinaryParseTest, EarlyReturnWithZeroPassingCallbacks) {
+ const auto binary = CompileSuccessfully(
+ "%1 = OpTypeVoid "
+ "%2 = OpTypeInt 32 1");
+
+ num_passing_callbacks_ = 0;
+
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_ERROR_INVALID_BINARY,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ // On error, the binary parser doesn't generate its own diagnostics.
+ EXPECT_EQ(nullptr, diagnostic);
+
+ // Early termination is registered after we have saved the header result.
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(3)}));
+ // The instruction callbacks are never called.
+ EXPECT_THAT(instructions(), Eq(Instructions{}));
+}
+
+TEST_F(BinaryParseTest,
+ EarlyReturnWithZeroPassingCallbacksAndSpecifiedResultCode) {
+ const auto binary = CompileSuccessfully(
+ "%1 = OpTypeVoid "
+ "%2 = OpTypeInt 32 1");
+
+ num_passing_callbacks_ = 0;
+ fail_code_ = SPV_REQUESTED_TERMINATION;
+
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_REQUESTED_TERMINATION,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ // On early termination, the binary parser doesn't generate its own
+ // diagnostics.
+ EXPECT_EQ(nullptr, diagnostic);
+ // Early exit is registered after we have saved the header result.
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(3)}));
+ // The instruction callbacks are never called.
+ EXPECT_THAT(instructions(), Eq(Instructions{}));
+}
+
+TEST_F(BinaryParseTest, EarlyReturnWithOnePassingCallback) {
+ const auto binary = CompileSuccessfully(
+ "%1 = OpTypeVoid "
+ "%2 = OpTypeInt 32 1 "
+ "%3 = OpTypeFloat 32");
+
+ num_passing_callbacks_ = 1;
+ fail_code_ = SPV_REQUESTED_TERMINATION;
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_REQUESTED_TERMINATION,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ // On early termination, the binary parser doesn't generate its own
+ // diagnostics.
+ EXPECT_EQ(nullptr, diagnostic);
+
+ // Early termination is registered after we have saved the header result.
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(4)}));
+ // The header callback succeeded. Then the instruction callback was called
+ // once (on the first instruction), and then it requested termination,
+ // preventing further instruction callbacks.
+ EXPECT_THAT(instructions(), Eq(Instructions{
+ MakeParsedVoidTypeInstruction(1),
+ }));
+}
+
+TEST_F(BinaryParseTest, EarlyReturnWithTwoPassingCallbacks) {
+ const auto binary = CompileSuccessfully(
+ "%1 = OpTypeVoid "
+ "%2 = OpTypeInt 32 1 "
+ "%3 = OpTypeFloat 32");
+
+ num_passing_callbacks_ = 2;
+ fail_code_ = SPV_REQUESTED_TERMINATION;
+ spv_diagnostic diagnostic = nullptr;
+ EXPECT_EQ(SPV_REQUESTED_TERMINATION,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, &diagnostic));
+ // On early termination, the binary parser doesn't generate its own
+ // diagnostics.
+ EXPECT_EQ(nullptr, diagnostic);
+
+ // Early termination is registered after we have saved the header result.
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(4)}));
+ // The header callback succeeded. Then the instruction callback was
+ // called twice and then it requested termination, preventing further
+ // instruction callbacks.
+ EXPECT_THAT(instructions(), Eq(Instructions{
+ MakeParsedVoidTypeInstruction(1),
+ MakeParsedInt32TypeInstruction(2),
+ }));
+}
+
+TEST_F(BinaryParseTest, InstructionWithStringOperand) {
+ const std::string str =
+ "the future is already here, it's just not evenly distributed";
+ const auto str_words = MakeVector(str);
+ const auto instruction = MakeInstruction(SpvOpName, {99}, str_words);
+ const auto binary = Concatenate({ExpectedHeaderForBound(100), instruction});
+
+ EXPECT_EQ(SPV_SUCCESS,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, nullptr));
+
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(100)}));
+
+ const auto operands = std::vector<spv_parsed_operand_t>{
+ MakeSimpleOperand(1, SPV_OPERAND_TYPE_ID),
+ MakeLiteralStringOperand(2, uint16_t(str_words.size()))};
+ const spv_parsed_instruction_t parsed_inst = {instruction.data(),
+ uint16_t(instruction.size()),
+ SpvOpName,
+ SPV_EXT_INST_TYPE_NONE,
+ 0, // type id
+ 0, // No result id for OpName
+ operands.data(),
+ uint16_t(operands.size())};
+ EXPECT_THAT(instructions(), Eq(Instructions{ParsedInstruction(parsed_inst)}));
+}
+
+// Checks for non-zero values for the result_id and ext_inst_type members
+// spv_parsed_instruction_t.
+TEST_F(BinaryParseTest, ExtendedInstruction) {
+ const auto binary = CompileSuccessfully(
+ "%extcl = OpExtInstImport \"OpenCL.std\" "
+ "%result = OpExtInst %float %extcl sqrt %x");
+
+ EXPECT_EQ(SPV_SUCCESS,
+ spvBinaryParse(context, this, binary.data(), binary.size(), Header,
+ Instruction, nullptr));
+
+ EXPECT_THAT(headers(), Eq(Sentences{ExpectedHeaderForBound(5)}));
+
+ const auto operands = std::vector<spv_parsed_operand_t>{
+ MakeSimpleOperand(1, SPV_OPERAND_TYPE_TYPE_ID),
+ MakeSimpleOperand(2, SPV_OPERAND_TYPE_RESULT_ID),
+ MakeSimpleOperand(3, SPV_OPERAND_TYPE_ID), // Extended instruction set Id
+ MakeSimpleOperand(4, SPV_OPERAND_TYPE_EXTENSION_INSTRUCTION_NUMBER),
+ MakeSimpleOperand(5, SPV_OPERAND_TYPE_ID), // Id of the argument
+ };
+ const auto instruction = MakeInstruction(
+ SpvOpExtInst, {2, 3, 1, uint32_t(OpenCLLIB::Entrypoints::Sqrt), 4});
+ const spv_parsed_instruction_t parsed_inst = {instruction.data(),
+ uint16_t(instruction.size()),
+ SpvOpExtInst,
+ SPV_EXT_INST_TYPE_OPENCL_STD,
+ 2, // type id
+ 3, // result id
+ operands.data(),
+ uint16_t(operands.size())};
+ EXPECT_THAT(instructions(), ElementsAre(_, ParsedInstruction(parsed_inst)));
+}
+
// TODO(dneto): Add tests for spvBinaryParse:
-// - early return via callback error values.
// - test each diagnostic in binary.cpp
-// - test the data sent via the header callback.
-// - test the data sent via the instruction callback.
} // anonymous namespace