Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 1 | // Copyright (c) 2015 The Khronos Group Inc. |
| 2 | // |
| 3 | // Permission is hereby granted, free of charge, to any person obtaining a |
| 4 | // copy of this software and/or associated documentation files (the |
| 5 | // "Materials"), to deal in the Materials without restriction, including |
| 6 | // without limitation the rights to use, copy, modify, merge, publish, |
| 7 | // distribute, sublicense, and/or sell copies of the Materials, and to |
| 8 | // permit persons to whom the Materials are furnished to do so, subject to |
| 9 | // the following conditions: |
| 10 | // |
| 11 | // The above copyright notice and this permission notice shall be included |
| 12 | // in all copies or substantial portions of the Materials. |
| 13 | // |
| 14 | // MODIFICATIONS TO THIS FILE MAY MEAN IT NO LONGER ACCURATELY REFLECTS |
| 15 | // KHRONOS STANDARDS. THE UNMODIFIED, NORMATIVE VERSIONS OF KHRONOS |
| 16 | // SPECIFICATIONS AND HEADER INFORMATION ARE LOCATED AT |
| 17 | // https://www.khronos.org/registry/ |
| 18 | // |
| 19 | // THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
| 20 | // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| 21 | // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
| 22 | // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY |
| 23 | // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, |
| 24 | // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE |
| 25 | // MATERIALS OR THE USE OR OTHER DEALINGS IN THE MATERIALS. |
| 26 | |
| 27 | #include <libspirv/libspirv.h> |
| 28 | #include "binary.h" |
| 29 | #include "diagnostic.h" |
| 30 | #include "ext_inst.h" |
| 31 | #include "opcode.h" |
| 32 | #include "operand.h" |
| 33 | |
| 34 | #include <assert.h> |
| 35 | #include <string.h> |
| 36 | |
| 37 | #include <sstream> |
| 38 | |
| 39 | // Binary API |
| 40 | |
| 41 | enum { |
| 42 | I32_ENDIAN_LITTLE = 0x03020100ul, |
| 43 | I32_ENDIAN_BIG = 0x00010203ul, |
| 44 | }; |
| 45 | |
| 46 | static const union { |
| 47 | unsigned char bytes[4]; |
| 48 | uint32_t value; |
| 49 | } o32_host_order = {{0, 1, 2, 3}}; |
| 50 | |
| 51 | #define I32_ENDIAN_HOST (o32_host_order.value) |
| 52 | |
| 53 | spv_result_t spvBinaryEndianness(const spv_binary binary, |
| 54 | spv_endianness_t *pEndian) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 55 | if (!binary->code || !binary->wordCount) return SPV_ERROR_INVALID_BINARY; |
| 56 | if (!pEndian) return SPV_ERROR_INVALID_POINTER; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 57 | |
| 58 | uint8_t bytes[4]; |
| 59 | memcpy(bytes, binary->code, sizeof(uint32_t)); |
| 60 | |
| 61 | if (0x03 == bytes[0] && 0x02 == bytes[1] && 0x23 == bytes[2] && |
| 62 | 0x07 == bytes[3]) { |
| 63 | *pEndian = SPV_ENDIANNESS_LITTLE; |
| 64 | return SPV_SUCCESS; |
| 65 | } |
| 66 | |
| 67 | if (0x07 == bytes[0] && 0x23 == bytes[1] && 0x02 == bytes[2] && |
| 68 | 0x03 == bytes[3]) { |
| 69 | *pEndian = SPV_ENDIANNESS_BIG; |
| 70 | return SPV_SUCCESS; |
| 71 | } |
| 72 | |
| 73 | return SPV_ERROR_INVALID_BINARY; |
| 74 | } |
| 75 | |
| 76 | uint32_t spvFixWord(const uint32_t word, const spv_endianness_t endian) { |
| 77 | if ((SPV_ENDIANNESS_LITTLE == endian && I32_ENDIAN_HOST == I32_ENDIAN_BIG) || |
| 78 | (SPV_ENDIANNESS_BIG == endian && I32_ENDIAN_HOST == I32_ENDIAN_LITTLE)) { |
| 79 | return (word & 0x000000ff) << 24 | (word & 0x0000ff00) << 8 | |
| 80 | (word & 0x00ff0000) >> 8 | (word & 0xff000000) >> 24; |
| 81 | } |
| 82 | |
| 83 | return word; |
| 84 | } |
| 85 | |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 86 | uint64_t spvFixDoubleWord(const uint32_t low, const uint32_t high, |
| 87 | const spv_endianness_t endian) { |
| 88 | return (uint64_t(spvFixWord(high, endian)) << 32) | spvFixWord(low, endian); |
| 89 | } |
| 90 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 91 | spv_result_t spvBinaryHeaderGet(const spv_binary binary, |
| 92 | const spv_endianness_t endian, |
| 93 | spv_header_t *pHeader) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 94 | if (!binary->code || !binary->wordCount) return SPV_ERROR_INVALID_BINARY; |
| 95 | if (!pHeader) return SPV_ERROR_INVALID_POINTER; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 96 | |
| 97 | // TODO: Validation checking? |
| 98 | pHeader->magic = spvFixWord(binary->code[SPV_INDEX_MAGIC_NUMBER], endian); |
| 99 | pHeader->version = spvFixWord(binary->code[SPV_INDEX_VERSION_NUMBER], endian); |
| 100 | pHeader->generator = |
| 101 | spvFixWord(binary->code[SPV_INDEX_GENERATOR_NUMBER], endian); |
| 102 | pHeader->bound = spvFixWord(binary->code[SPV_INDEX_BOUND], endian); |
| 103 | pHeader->schema = spvFixWord(binary->code[SPV_INDEX_SCHEMA], endian); |
| 104 | pHeader->instructions = &binary->code[SPV_INDEX_INSTRUCTION]; |
| 105 | |
| 106 | return SPV_SUCCESS; |
| 107 | } |
| 108 | |
| 109 | spv_result_t spvBinaryHeaderSet(spv_binary_t *binary, const uint32_t bound) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 110 | if (!binary) return SPV_ERROR_INVALID_BINARY; |
| 111 | if (!binary->code || !binary->wordCount) return SPV_ERROR_INVALID_BINARY; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 112 | |
| 113 | binary->code[SPV_INDEX_MAGIC_NUMBER] = SPV_MAGIC_NUMBER; |
| 114 | binary->code[SPV_INDEX_VERSION_NUMBER] = SPV_VERSION_NUMBER; |
Kenneth Benzie (Benie) | 81d7d49 | 2015-06-01 09:50:46 -0700 | [diff] [blame] | 115 | binary->code[SPV_INDEX_GENERATOR_NUMBER] = SPV_GENERATOR_KHRONOS; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 116 | binary->code[SPV_INDEX_BOUND] = bound; |
| 117 | binary->code[SPV_INDEX_SCHEMA] = 0; // NOTE: Reserved |
| 118 | |
| 119 | return SPV_SUCCESS; |
| 120 | } |
| 121 | |
| 122 | spv_result_t spvBinaryEncodeU32(const uint32_t value, spv_instruction_t *pInst, |
| 123 | const spv_position position, |
| 124 | spv_diagnostic *pDiagnostic) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 125 | if (pInst->wordCount + 1 > SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX) { |
| 126 | DIAGNOSTIC << "Instruction word count '" |
| 127 | << SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX << "' exceeded."; |
| 128 | return SPV_ERROR_INVALID_TEXT; |
| 129 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 130 | |
| 131 | pInst->words[pInst->wordCount++] = (uint32_t)value; |
| 132 | return SPV_SUCCESS; |
| 133 | } |
| 134 | |
| 135 | spv_result_t spvBinaryEncodeU64(const uint64_t value, spv_instruction_t *pInst, |
| 136 | const spv_position position, |
| 137 | spv_diagnostic *pDiagnostic) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 138 | if (pInst->wordCount + 2 > SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX) { |
| 139 | DIAGNOSTIC << "Instruction word count '" |
| 140 | << SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX << "' exceeded."; |
| 141 | return SPV_ERROR_INVALID_TEXT; |
| 142 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 143 | |
| 144 | uint32_t low = (uint32_t)(0x00000000ffffffff & value); |
| 145 | uint32_t high = (uint32_t)((0xffffffff00000000 & value) >> 32); |
| 146 | pInst->words[pInst->wordCount++] = low; |
| 147 | pInst->words[pInst->wordCount++] = high; |
| 148 | return SPV_SUCCESS; |
| 149 | } |
| 150 | |
| 151 | spv_result_t spvBinaryEncodeString(const char *str, spv_instruction_t *pInst, |
| 152 | const spv_position position, |
| 153 | spv_diagnostic *pDiagnostic) { |
| 154 | size_t length = strlen(str); |
| 155 | size_t wordCount = (length / 4) + 1; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 156 | if ((sizeof(uint32_t) * pInst->wordCount) + length > |
| 157 | sizeof(uint32_t) * SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX) { |
| 158 | DIAGNOSTIC << "Instruction word count '" |
| 159 | << SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX << "'exceeded."; |
| 160 | return SPV_ERROR_INVALID_TEXT; |
| 161 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 162 | |
| 163 | char *dest = (char *)&pInst->words[pInst->wordCount]; |
| 164 | strncpy(dest, str, length); |
| 165 | pInst->wordCount += (uint16_t)wordCount; |
| 166 | |
| 167 | return SPV_SUCCESS; |
| 168 | } |
| 169 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 170 | // TODO(dneto): This API is not powerful enough in the case that the |
| 171 | // number and type of operands are not known until partway through parsing |
| 172 | // the operation. This happens when enum operands might have different number |
| 173 | // of operands, or with extended instructions. |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 174 | spv_operand_type_t spvBinaryOperandInfo(const uint32_t word, |
| 175 | const uint16_t operandIndex, |
| 176 | const spv_opcode_desc opcodeEntry, |
| 177 | const spv_operand_table operandTable, |
| 178 | spv_operand_desc *pOperandEntry) { |
| 179 | spv_operand_type_t type; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 180 | if (operandIndex < opcodeEntry->numTypes) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 181 | // NOTE: Do operand table lookup to set operandEntry if successful |
| 182 | uint16_t index = operandIndex - 1; |
| 183 | type = opcodeEntry->operandTypes[index]; |
| 184 | spv_operand_desc entry = nullptr; |
| 185 | if (!spvOperandTableValueLookup(operandTable, type, word, &entry)) { |
| 186 | if (SPV_OPERAND_TYPE_NONE != entry->operandTypes[0]) { |
| 187 | *pOperandEntry = entry; |
| 188 | } |
| 189 | } |
| 190 | } else if (*pOperandEntry) { |
| 191 | // NOTE: Use specified operand entry operand type for this word |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 192 | uint16_t index = operandIndex - opcodeEntry->numTypes; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 193 | type = (*pOperandEntry)->operandTypes[index]; |
| 194 | } else if (OpSwitch == opcodeEntry->opcode) { |
| 195 | // NOTE: OpSwitch is a special case which expects a list of paired extra |
| 196 | // operands |
| 197 | assert(0 && |
| 198 | "This case is previously untested, remove this assert and ensure it " |
| 199 | "is behaving correctly!"); |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 200 | uint16_t lastIndex = opcodeEntry->numTypes - 1; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 201 | uint16_t index = lastIndex + ((operandIndex - lastIndex) % 2); |
| 202 | type = opcodeEntry->operandTypes[index]; |
| 203 | } else { |
| 204 | // NOTE: Default to last operand type in opcode entry |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 205 | uint16_t index = opcodeEntry->numTypes - 1; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 206 | type = opcodeEntry->operandTypes[index]; |
| 207 | } |
| 208 | return type; |
| 209 | } |
| 210 | |
| 211 | spv_result_t spvBinaryDecodeOperand( |
| 212 | const Op opcode, const spv_operand_type_t type, const uint32_t *words, |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 213 | uint16_t numWords, const spv_endianness_t endian, const uint32_t options, |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 214 | const spv_operand_table operandTable, const spv_ext_inst_table extInstTable, |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 215 | spv_operand_pattern_t *pExpectedOperands, spv_ext_inst_type_t *pExtInstType, |
| 216 | out_stream &stream, spv_position position, spv_diagnostic *pDiagnostic) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 217 | if (!words || !position) return SPV_ERROR_INVALID_POINTER; |
| 218 | if (!pDiagnostic) return SPV_ERROR_INVALID_DIAGNOSTIC; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 219 | |
| 220 | bool print = spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_PRINT, options); |
| 221 | bool color = |
| 222 | print && spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_COLOR, options); |
| 223 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 224 | switch (type) { |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 225 | case SPV_OPERAND_TYPE_ID: |
| 226 | case SPV_OPERAND_TYPE_RESULT_ID: |
| 227 | case SPV_OPERAND_TYPE_OPTIONAL_ID: |
| 228 | case SPV_OPERAND_TYPE_ID_IN_OPTIONAL_TUPLE: { |
| 229 | if (color) { |
Pyry Haulos | 26b3b00 | 2015-09-09 13:35:53 -0700 | [diff] [blame] | 230 | if (type == SPV_OPERAND_TYPE_RESULT_ID) { |
| 231 | stream.get() << clr::blue(); |
| 232 | } else { |
| 233 | stream.get() << clr::yellow(); |
| 234 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 235 | } |
Lei Zhang | 97afd5c | 2015-09-14 15:26:12 -0400 | [diff] [blame] | 236 | stream.get() << "%" << spvFixWord(words[0], endian); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 237 | stream.get() << ((color) ? clr::reset() : ""); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 238 | position->index++; |
| 239 | } break; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 240 | case SPV_OPERAND_TYPE_LITERAL_NUMBER: { |
| 241 | // NOTE: Special case for extended instruction use |
| 242 | if (OpExtInst == opcode) { |
| 243 | spv_ext_inst_desc extInst; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 244 | if (spvExtInstTableValueLookup(extInstTable, *pExtInstType, words[0], |
| 245 | &extInst)) { |
| 246 | DIAGNOSTIC << "Invalid extended instruction '" << words[0] << "'."; |
| 247 | return SPV_ERROR_INVALID_BINARY; |
| 248 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 249 | spvPrependOperandTypes(extInst->operandTypes, pExpectedOperands); |
Andrew Woloszyn | 0d350b5 | 2015-08-21 14:23:42 -0400 | [diff] [blame] | 250 | stream.get() << (color ? clr::red() : ""); |
| 251 | stream.get() << extInst->name; |
| 252 | stream.get() << (color ? clr::reset() : ""); |
Lei Zhang | 41bf073 | 2015-09-14 12:26:15 -0400 | [diff] [blame] | 253 | position->index++; |
| 254 | break; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 255 | } |
Lei Zhang | 41bf073 | 2015-09-14 12:26:15 -0400 | [diff] [blame] | 256 | } // Fall through for the general case. |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 257 | case SPV_OPERAND_TYPE_MULTIWORD_LITERAL_NUMBER: |
Lei Zhang | 41bf073 | 2015-09-14 12:26:15 -0400 | [diff] [blame] | 258 | case SPV_OPERAND_TYPE_LITERAL: |
| 259 | case SPV_OPERAND_TYPE_OPTIONAL_LITERAL: |
| 260 | case SPV_OPERAND_TYPE_LITERAL_IN_OPTIONAL_TUPLE: { |
| 261 | // TODO: Need to support multiple word literals |
| 262 | stream.get() << (color ? clr::red() : ""); |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 263 | if (numWords > 2) { |
| 264 | DIAGNOSTIC << "Literal numbers larger than 64-bit not supported yet."; |
| 265 | return SPV_UNSUPPORTED; |
| 266 | } else if (numWords == 2) { |
| 267 | stream.get() << spvFixDoubleWord(words[0], words[1], endian); |
| 268 | position->index += 2; |
| 269 | } else { |
| 270 | stream.get() << spvFixWord(words[0], endian); |
| 271 | position->index++; |
| 272 | } |
Lei Zhang | 41bf073 | 2015-09-14 12:26:15 -0400 | [diff] [blame] | 273 | stream.get() << (color ? clr::reset() : ""); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 274 | } break; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 275 | case SPV_OPERAND_TYPE_LITERAL_STRING: |
| 276 | case SPV_OPERAND_TYPE_OPTIONAL_LITERAL_STRING: { |
Lei Zhang | 97afd5c | 2015-09-14 15:26:12 -0400 | [diff] [blame] | 277 | const char *string = (const char *)words; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 278 | uint64_t stringOperandCount = (strlen(string) / 4) + 1; |
| 279 | |
| 280 | // NOTE: Special case for extended instruction import |
| 281 | if (OpExtInstImport == opcode) { |
| 282 | *pExtInstType = spvExtInstImportTypeGet(string); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 283 | if (SPV_EXT_INST_TYPE_NONE == *pExtInstType) { |
| 284 | DIAGNOSTIC << "Invalid extended instruction import'" << string |
| 285 | << "'."; |
| 286 | return SPV_ERROR_INVALID_BINARY; |
| 287 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 288 | } |
| 289 | |
| 290 | stream.get() << "\""; |
| 291 | stream.get() << (color ? clr::green() : ""); |
| 292 | stream.get() << string; |
| 293 | stream.get() << (color ? clr::reset() : ""); |
| 294 | stream.get() << "\""; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 295 | position->index += stringOperandCount; |
| 296 | } break; |
| 297 | case SPV_OPERAND_TYPE_CAPABILITY: |
| 298 | case SPV_OPERAND_TYPE_SOURCE_LANGUAGE: |
| 299 | case SPV_OPERAND_TYPE_EXECUTION_MODEL: |
| 300 | case SPV_OPERAND_TYPE_ADDRESSING_MODEL: |
| 301 | case SPV_OPERAND_TYPE_MEMORY_MODEL: |
| 302 | case SPV_OPERAND_TYPE_EXECUTION_MODE: |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 303 | case SPV_OPERAND_TYPE_OPTIONAL_EXECUTION_MODE: |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 304 | case SPV_OPERAND_TYPE_STORAGE_CLASS: |
| 305 | case SPV_OPERAND_TYPE_DIMENSIONALITY: |
| 306 | case SPV_OPERAND_TYPE_SAMPLER_ADDRESSING_MODE: |
| 307 | case SPV_OPERAND_TYPE_SAMPLER_FILTER_MODE: |
| 308 | case SPV_OPERAND_TYPE_FP_FAST_MATH_MODE: |
| 309 | case SPV_OPERAND_TYPE_FP_ROUNDING_MODE: |
| 310 | case SPV_OPERAND_TYPE_LINKAGE_TYPE: |
| 311 | case SPV_OPERAND_TYPE_ACCESS_QUALIFIER: |
| 312 | case SPV_OPERAND_TYPE_FUNCTION_PARAMETER_ATTRIBUTE: |
| 313 | case SPV_OPERAND_TYPE_DECORATION: |
| 314 | case SPV_OPERAND_TYPE_BUILT_IN: |
| 315 | case SPV_OPERAND_TYPE_SELECTION_CONTROL: |
| 316 | case SPV_OPERAND_TYPE_LOOP_CONTROL: |
| 317 | case SPV_OPERAND_TYPE_FUNCTION_CONTROL: |
| 318 | case SPV_OPERAND_TYPE_MEMORY_SEMANTICS: |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 319 | case SPV_OPERAND_TYPE_OPTIONAL_MEMORY_ACCESS: |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 320 | case SPV_OPERAND_TYPE_EXECUTION_SCOPE: |
| 321 | case SPV_OPERAND_TYPE_GROUP_OPERATION: |
| 322 | case SPV_OPERAND_TYPE_KERNEL_ENQ_FLAGS: |
David Neto | 4799482 | 2015-08-27 13:11:01 -0400 | [diff] [blame] | 323 | case SPV_OPERAND_TYPE_KERNEL_PROFILING_INFO: { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 324 | spv_operand_desc entry; |
Lei Zhang | 97afd5c | 2015-09-14 15:26:12 -0400 | [diff] [blame] | 325 | if (spvOperandTableValueLookup(operandTable, type, |
| 326 | spvFixWord(words[0], endian), &entry)) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 327 | DIAGNOSTIC << "Invalid " << spvOperandTypeStr(type) << " operand '" |
Lei Zhang | 97afd5c | 2015-09-14 15:26:12 -0400 | [diff] [blame] | 328 | << words[0] << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 329 | return SPV_ERROR_INVALID_TEXT; |
| 330 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 331 | stream.get() << entry->name; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 332 | // Prepare to accept operands to this operand, if needed. |
| 333 | spvPrependOperandTypes(entry->operandTypes, pExpectedOperands); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 334 | position->index++; |
| 335 | } break; |
| 336 | default: { |
| 337 | DIAGNOSTIC << "Invalid binary operand '" << type << "'"; |
| 338 | return SPV_ERROR_INVALID_BINARY; |
| 339 | } |
| 340 | } |
| 341 | |
| 342 | return SPV_SUCCESS; |
| 343 | } |
| 344 | |
| 345 | spv_result_t spvBinaryDecodeOpcode( |
| 346 | spv_instruction_t *pInst, const spv_endianness_t endian, |
| 347 | const uint32_t options, const spv_opcode_table opcodeTable, |
| 348 | const spv_operand_table operandTable, const spv_ext_inst_table extInstTable, |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 349 | spv_assembly_syntax_format_t format, out_stream &stream, |
| 350 | spv_position position, spv_diagnostic *pDiagnostic) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 351 | if (!pInst || !position) return SPV_ERROR_INVALID_POINTER; |
| 352 | if (!opcodeTable || !operandTable || !extInstTable) |
| 353 | return SPV_ERROR_INVALID_TABLE; |
| 354 | if (!pDiagnostic) return SPV_ERROR_INVALID_DIAGNOSTIC; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 355 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 356 | spv_position_t instructionStart = *position; |
| 357 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 358 | uint16_t wordCount; |
| 359 | Op opcode; |
| 360 | spvOpcodeSplit(spvFixWord(pInst->words[0], endian), &wordCount, &opcode); |
| 361 | |
| 362 | spv_opcode_desc opcodeEntry; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 363 | if (spvOpcodeTableValueLookup(opcodeTable, opcode, &opcodeEntry)) { |
| 364 | DIAGNOSTIC << "Invalid Opcode '" << opcode << "'."; |
| 365 | return SPV_ERROR_INVALID_BINARY; |
| 366 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 367 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 368 | // See if there are enough required words. |
| 369 | // Some operands in the operand types are optional or could be zero length. |
| 370 | // The optional and zero length opeands must be at the end of the list. |
| 371 | if (opcodeEntry->numTypes > wordCount && |
| 372 | !spvOperandIsOptional(opcodeEntry->operandTypes[wordCount])) { |
| 373 | uint16_t numRequired; |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 374 | for (numRequired = 0; |
| 375 | numRequired < opcodeEntry->numTypes && |
| 376 | !spvOperandIsOptional(opcodeEntry->operandTypes[numRequired]); |
| 377 | numRequired++) |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 378 | ; |
| 379 | DIAGNOSTIC << "Invalid instruction Op" << opcodeEntry->name |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 380 | << " word count '" << wordCount << "', expected at least '" |
| 381 | << numRequired << "'."; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 382 | return SPV_ERROR_INVALID_BINARY; |
| 383 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 384 | |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 385 | const bool isAssigmentFormat = |
| 386 | SPV_ASSEMBLY_SYNTAX_FORMAT_ASSIGNMENT == format; |
| 387 | |
| 388 | // For Canonical Assembly Format, all words are written to stream in order. |
| 389 | // For Assignment Assembly Format, <result-id> and the equal sign are written |
| 390 | // to stream first, while the rest are written to no_result_id_stream. After |
| 391 | // processing all words, all words in no_result_id_stream are transcribed to |
| 392 | // stream. |
| 393 | |
Lei Zhang | 8a37520 | 2015-08-24 15:52:26 -0400 | [diff] [blame] | 394 | std::stringstream no_result_id_strstream; |
| 395 | out_stream no_result_id_stream(no_result_id_strstream); |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 396 | (isAssigmentFormat ? no_result_id_stream.get() : stream.get()) |
| 397 | << "Op" << opcodeEntry->name; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 398 | |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 399 | const int16_t result_id_index = spvOpcodeResultIdIndex(opcodeEntry); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 400 | position->index++; |
| 401 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 402 | // Maintains the ordered list of expected operand types. |
| 403 | // For many instructions we only need the {numTypes, operandTypes} |
| 404 | // entries in opcodeEntry. However, sometimes we need to modify |
| 405 | // the list as we parse the operands. This occurs when an operand |
| 406 | // has its own logical operands (such as the LocalSize operand for |
| 407 | // ExecutionMode), or for extended instructions that may have their |
| 408 | // own operands depending on the selected extended instruction. |
| 409 | spv_operand_pattern_t expectedOperands( |
| 410 | opcodeEntry->operandTypes, |
| 411 | opcodeEntry->operandTypes + opcodeEntry->numTypes); |
| 412 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 413 | for (uint16_t index = 1; index < wordCount; ++index) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 414 | const uint64_t currentPosIndex = position->index; |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 415 | const bool currentIsResultId = result_id_index == index - 1; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 416 | |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 417 | if (expectedOperands.empty()) { |
| 418 | DIAGNOSTIC << "Invalid instruction Op" << opcodeEntry->name |
| 419 | << " starting at word " << instructionStart.index |
| 420 | << ": expected no more operands after " << index |
| 421 | << " words, but word count is " << wordCount << "."; |
| 422 | return SPV_ERROR_INVALID_BINARY; |
| 423 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 424 | |
| 425 | spv_operand_type_t type = spvTakeFirstMatchableOperand(&expectedOperands); |
| 426 | |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 427 | if (isAssigmentFormat) { |
| 428 | if (!currentIsResultId) no_result_id_stream.get() << " "; |
| 429 | } else { |
| 430 | stream.get() << " "; |
| 431 | } |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 432 | |
| 433 | uint16_t numWords = 1; |
| 434 | if (type == SPV_OPERAND_TYPE_MULTIWORD_LITERAL_NUMBER) { |
| 435 | // Make sure this is the last operand for this instruction. |
| 436 | if (expectedOperands.empty()) { |
| 437 | numWords = wordCount - index; |
| 438 | } else { |
| 439 | // TODO(antiagainst): This may not be an error. The exact design has not |
| 440 | // been settled yet. |
| 441 | DIAGNOSTIC << "Multiple word literal numbers can only appear as the " |
| 442 | "last operand of an instruction."; |
| 443 | return SPV_ERROR_INVALID_BINARY; |
| 444 | } |
| 445 | } |
| 446 | |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 447 | if (spvBinaryDecodeOperand( |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 448 | opcodeEntry->opcode, type, pInst->words + index, numWords, endian, |
| 449 | options, operandTable, extInstTable, &expectedOperands, |
| 450 | &pInst->extInstType, |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 451 | (isAssigmentFormat && !currentIsResultId ? no_result_id_stream |
| 452 | : stream), |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 453 | position, pDiagnostic)) { |
| 454 | DIAGNOSTIC << "UNEXPLAINED ERROR"; |
| 455 | return SPV_ERROR_INVALID_BINARY; |
| 456 | } |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 457 | if (isAssigmentFormat && currentIsResultId) stream.get() << " = "; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 458 | index += (uint16_t)(position->index - currentPosIndex - 1); |
| 459 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 460 | // TODO(dneto): There's an opportunity for a more informative message. |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 461 | if (!expectedOperands.empty() && |
| 462 | !spvOperandIsOptional(expectedOperands.front())) { |
| 463 | DIAGNOSTIC << "Invalid instruction Op" << opcodeEntry->name |
| 464 | << " starting at word " << instructionStart.index |
| 465 | << ": expected more operands after " << wordCount << " words."; |
| 466 | return SPV_ERROR_INVALID_BINARY; |
| 467 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 468 | |
Lei Zhang | 8a37520 | 2015-08-24 15:52:26 -0400 | [diff] [blame] | 469 | stream.get() << no_result_id_strstream.str(); |
| 470 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 471 | return SPV_SUCCESS; |
| 472 | } |
| 473 | |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 474 | spv_result_t spvBinaryToText(uint32_t *code, const uint64_t wordCount, |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 475 | const uint32_t options, |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 476 | const spv_opcode_table opcodeTable, |
| 477 | const spv_operand_table operandTable, |
| 478 | const spv_ext_inst_table extInstTable, |
| 479 | spv_text *pText, spv_diagnostic *pDiagnostic) { |
Lei Zhang | 29e667e | 2015-09-11 11:01:59 -0400 | [diff] [blame] | 480 | return spvBinaryToTextWithFormat( |
| 481 | code, wordCount, options, opcodeTable, operandTable, extInstTable, |
| 482 | SPV_ASSEMBLY_SYNTAX_FORMAT_DEFAULT, pText, pDiagnostic); |
| 483 | } |
| 484 | |
| 485 | spv_result_t spvBinaryToTextWithFormat( |
| 486 | uint32_t *code, const uint64_t wordCount, const uint32_t options, |
| 487 | const spv_opcode_table opcodeTable, const spv_operand_table operandTable, |
| 488 | const spv_ext_inst_table extInstTable, spv_assembly_syntax_format_t format, |
| 489 | spv_text *pText, spv_diagnostic *pDiagnostic) { |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 490 | spv_binary_t binary = {code, wordCount}; |
| 491 | |
Andrew Woloszyn | 4b4acde | 2015-09-10 10:28:22 -0400 | [diff] [blame] | 492 | spv_position_t position = {}; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 493 | if (!binary.code || !binary.wordCount) { |
| 494 | DIAGNOSTIC << "Binary stream is empty."; |
| 495 | return SPV_ERROR_INVALID_BINARY; |
| 496 | } |
| 497 | if (!opcodeTable || !operandTable || !extInstTable) |
| 498 | return SPV_ERROR_INVALID_TABLE; |
| 499 | if (pText && spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_PRINT, options)) |
| 500 | return SPV_ERROR_INVALID_POINTER; |
| 501 | if (!pText && !spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_PRINT, options)) |
| 502 | return SPV_ERROR_INVALID_POINTER; |
| 503 | if (!pDiagnostic) return SPV_ERROR_INVALID_DIAGNOSTIC; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 504 | |
| 505 | spv_endianness_t endian; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 506 | if (spvBinaryEndianness(&binary, &endian)) { |
| 507 | DIAGNOSTIC << "Invalid SPIR-V magic number '" << std::hex << binary.code[0] |
| 508 | << "'."; |
| 509 | return SPV_ERROR_INVALID_BINARY; |
| 510 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 511 | |
| 512 | spv_header_t header; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 513 | if (spvBinaryHeaderGet(&binary, endian, &header)) { |
| 514 | DIAGNOSTIC << "Invalid SPIR-V header."; |
| 515 | return SPV_ERROR_INVALID_BINARY; |
| 516 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 517 | |
| 518 | bool print = spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_PRINT, options); |
| 519 | bool color = |
| 520 | print && spvIsInBitfield(SPV_BINARY_TO_TEXT_OPTION_COLOR, options); |
| 521 | |
| 522 | std::stringstream sstream; |
| 523 | out_stream stream(sstream); |
| 524 | if (print) { |
| 525 | stream = out_stream(); |
| 526 | } |
| 527 | |
| 528 | if (color) { |
| 529 | stream.get() << clr::grey(); |
| 530 | } |
| 531 | stream.get() << "; SPIR-V\n" |
| 532 | << "; Version: " << header.version << "\n" |
| 533 | << "; Generator: " << spvGeneratorStr(header.generator) << "\n" |
| 534 | << "; Bound: " << header.bound << "\n" |
| 535 | << "; Schema: " << header.schema << "\n"; |
| 536 | if (color) { |
| 537 | stream.get() << clr::reset(); |
| 538 | } |
| 539 | |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 540 | const uint32_t *words = binary.code; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 541 | position.index = SPV_INDEX_INSTRUCTION; |
| 542 | spv_ext_inst_type_t extInstType = SPV_EXT_INST_TYPE_NONE; |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 543 | while (position.index < binary.wordCount) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 544 | uint64_t index = position.index; |
| 545 | uint16_t wordCount; |
| 546 | Op opcode; |
| 547 | spvOpcodeSplit(spvFixWord(words[position.index], endian), &wordCount, |
| 548 | &opcode); |
| 549 | |
| 550 | spv_instruction_t inst = {}; |
| 551 | inst.extInstType = extInstType; |
| 552 | spvInstructionCopy(&words[position.index], opcode, wordCount, endian, |
| 553 | &inst); |
| 554 | |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 555 | if (spvBinaryDecodeOpcode(&inst, endian, options, opcodeTable, operandTable, |
| 556 | extInstTable, format, stream, &position, |
| 557 | pDiagnostic)) |
| 558 | return SPV_ERROR_INVALID_BINARY; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 559 | extInstType = inst.extInstType; |
| 560 | |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 561 | if ((index + wordCount) != position.index) { |
| 562 | DIAGNOSTIC << "Invalid word count."; |
| 563 | return SPV_ERROR_INVALID_BINARY; |
| 564 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 565 | |
| 566 | stream.get() << "\n"; |
| 567 | } |
| 568 | |
| 569 | if (!print) { |
| 570 | size_t length = sstream.str().size(); |
| 571 | char *str = new char[length + 1]; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 572 | if (!str) return SPV_ERROR_OUT_OF_MEMORY; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 573 | strncpy(str, sstream.str().c_str(), length + 1); |
| 574 | spv_text text = new spv_text_t(); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 575 | if (!text) return SPV_ERROR_OUT_OF_MEMORY; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 576 | text->str = str; |
| 577 | text->length = length; |
| 578 | *pText = text; |
| 579 | } |
| 580 | |
| 581 | return SPV_SUCCESS; |
| 582 | } |
| 583 | |
| 584 | void spvBinaryDestroy(spv_binary binary) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 585 | if (!binary) return; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 586 | if (binary->code) { |
| 587 | delete[] binary->code; |
| 588 | } |
| 589 | delete binary; |
| 590 | } |