Coverage Report

Created: 2025-12-14 06:36

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/capstonenext/arch/WASM/WASMDisassembler.c
Line
Count
Source
1
/* Capstone Disassembly Engine */
2
/* By Spike, xwings 2019 */
3
4
#include <string.h>
5
#include <stddef.h>
6
7
#include "WASMDisassembler.h"
8
#include "WASMMapping.h"
9
#include "../../cs_priv.h"
10
11
static const short opcodes[256] = {
12
  WASM_INS_UNREACHABLE,
13
  WASM_INS_NOP,
14
  WASM_INS_BLOCK,
15
  WASM_INS_LOOP,
16
  WASM_INS_IF,
17
  WASM_INS_ELSE,
18
  -1,
19
  -1,
20
  -1,
21
  -1,
22
  -1,
23
  WASM_INS_END,
24
  WASM_INS_BR,
25
  WASM_INS_BR_IF,
26
  WASM_INS_BR_TABLE,
27
  WASM_INS_RETURN,
28
  WASM_INS_CALL,
29
  WASM_INS_CALL_INDIRECT,
30
  -1,
31
  -1,
32
  -1,
33
  -1,
34
  -1,
35
  -1,
36
  -1,
37
  -1,
38
  WASM_INS_DROP,
39
  WASM_INS_SELECT,
40
  -1,
41
  -1,
42
  -1,
43
  -1,
44
  WASM_INS_GET_LOCAL,
45
  WASM_INS_SET_LOCAL,
46
  WASM_INS_TEE_LOCAL,
47
  WASM_INS_GET_GLOBAL,
48
  WASM_INS_SET_GLOBAL,
49
  -1,
50
  -1,
51
  -1,
52
  WASM_INS_I32_LOAD,
53
  WASM_INS_I64_LOAD,
54
  WASM_INS_F32_LOAD,
55
  WASM_INS_F64_LOAD,
56
  WASM_INS_I32_LOAD8_S,
57
  WASM_INS_I32_LOAD8_U,
58
  WASM_INS_I32_LOAD16_S,
59
  WASM_INS_I32_LOAD16_U,
60
  WASM_INS_I64_LOAD8_S,
61
  WASM_INS_I64_LOAD8_U,
62
  WASM_INS_I64_LOAD16_S,
63
  WASM_INS_I64_LOAD16_U,
64
  WASM_INS_I64_LOAD32_S,
65
  WASM_INS_I64_LOAD32_U,
66
  WASM_INS_I32_STORE,
67
  WASM_INS_I64_STORE,
68
  WASM_INS_F32_STORE,
69
  WASM_INS_F64_STORE,
70
  WASM_INS_I32_STORE8,
71
  WASM_INS_I32_STORE16,
72
  WASM_INS_I64_STORE8,
73
  WASM_INS_I64_STORE16,
74
  WASM_INS_I64_STORE32,
75
  WASM_INS_CURRENT_MEMORY,
76
  WASM_INS_GROW_MEMORY,
77
  WASM_INS_I32_CONST,
78
  WASM_INS_I64_CONST,
79
  WASM_INS_F32_CONST,
80
  WASM_INS_F64_CONST,
81
  WASM_INS_I32_EQZ,
82
  WASM_INS_I32_EQ,
83
  WASM_INS_I32_NE,
84
  WASM_INS_I32_LT_S,
85
  WASM_INS_I32_LT_U,
86
  WASM_INS_I32_GT_S,
87
  WASM_INS_I32_GT_U,
88
  WASM_INS_I32_LE_S,
89
  WASM_INS_I32_LE_U,
90
  WASM_INS_I32_GE_S,
91
  WASM_INS_I32_GE_U,
92
  WASM_INS_I64_EQZ,
93
  WASM_INS_I64_EQ,
94
  WASM_INS_I64_NE,
95
  WASM_INS_I64_LT_S,
96
  WASM_INS_I64_LT_U,
97
  WASN_INS_I64_GT_S,
98
  WASM_INS_I64_GT_U,
99
  WASM_INS_I64_LE_S,
100
  WASM_INS_I64_LE_U,
101
  WASM_INS_I64_GE_S,
102
  WASM_INS_I64_GE_U,
103
  WASM_INS_F32_EQ,
104
  WASM_INS_F32_NE,
105
  WASM_INS_F32_LT,
106
  WASM_INS_F32_GT,
107
  WASM_INS_F32_LE,
108
  WASM_INS_F32_GE,
109
  WASM_INS_F64_EQ,
110
  WASM_INS_F64_NE,
111
  WASM_INS_F64_LT,
112
  WASM_INS_F64_GT,
113
  WASM_INS_F64_LE,
114
  WASM_INS_F64_GE,
115
  WASM_INS_I32_CLZ,
116
  WASM_INS_I32_CTZ,
117
  WASM_INS_I32_POPCNT,
118
  WASM_INS_I32_ADD,
119
  WASM_INS_I32_SUB,
120
  WASM_INS_I32_MUL,
121
  WASM_INS_I32_DIV_S,
122
  WASM_INS_I32_DIV_U,
123
  WASM_INS_I32_REM_S,
124
  WASM_INS_I32_REM_U,
125
  WASM_INS_I32_AND,
126
  WASM_INS_I32_OR,
127
  WASM_INS_I32_XOR,
128
  WASM_INS_I32_SHL,
129
  WASM_INS_I32_SHR_S,
130
  WASM_INS_I32_SHR_U,
131
  WASM_INS_I32_ROTL,
132
  WASM_INS_I32_ROTR,
133
  WASM_INS_I64_CLZ,
134
  WASM_INS_I64_CTZ,
135
  WASM_INS_I64_POPCNT,
136
  WASM_INS_I64_ADD,
137
  WASM_INS_I64_SUB,
138
  WASM_INS_I64_MUL,
139
  WASM_INS_I64_DIV_S,
140
  WASM_INS_I64_DIV_U,
141
  WASM_INS_I64_REM_S,
142
  WASM_INS_I64_REM_U,
143
  WASM_INS_I64_AND,
144
  WASM_INS_I64_OR,
145
  WASM_INS_I64_XOR,
146
  WASM_INS_I64_SHL,
147
  WASM_INS_I64_SHR_S,
148
  WASM_INS_I64_SHR_U,
149
  WASM_INS_I64_ROTL,
150
  WASM_INS_I64_ROTR,
151
  WASM_INS_F32_ABS,
152
  WASM_INS_F32_NEG,
153
  WASM_INS_F32_CEIL,
154
  WASM_INS_F32_FLOOR,
155
  WASM_INS_F32_TRUNC,
156
  WASM_INS_F32_NEAREST,
157
  WASM_INS_F32_SQRT,
158
  WASM_INS_F32_ADD,
159
  WASM_INS_F32_SUB,
160
  WASM_INS_F32_MUL,
161
  WASM_INS_F32_DIV,
162
  WASM_INS_F32_MIN,
163
  WASM_INS_F32_MAX,
164
  WASM_INS_F32_COPYSIGN,
165
  WASM_INS_F64_ABS,
166
  WASM_INS_F64_NEG,
167
  WASM_INS_F64_CEIL,
168
  WASM_INS_F64_FLOOR,
169
  WASM_INS_F64_TRUNC,
170
  WASM_INS_F64_NEAREST,
171
  WASM_INS_F64_SQRT,
172
  WASM_INS_F64_ADD,
173
  WASM_INS_F64_SUB,
174
  WASM_INS_F64_MUL,
175
  WASM_INS_F64_DIV,
176
  WASM_INS_F64_MIN,
177
  WASM_INS_F64_MAX,
178
  WASM_INS_F64_COPYSIGN,
179
  WASM_INS_I32_WARP_I64,
180
  WASP_INS_I32_TRUNC_S_F32,
181
  WASM_INS_I32_TRUNC_U_F32,
182
  WASM_INS_I32_TRUNC_S_F64,
183
  WASM_INS_I32_TRUNC_U_F64,
184
  WASM_INS_I64_EXTEND_S_I32,
185
  WASM_INS_I64_EXTEND_U_I32,
186
  WASM_INS_I64_TRUNC_S_F32,
187
  WASM_INS_I64_TRUNC_U_F32,
188
  WASM_INS_I64_TRUNC_S_F64,
189
  WASM_INS_I64_TRUNC_U_F64,
190
  WASM_INS_F32_CONVERT_S_I32,
191
  WASM_INS_F32_CONVERT_U_I32,
192
  WASM_INS_F32_CONVERT_S_I64,
193
  WASM_INS_F32_CONVERT_U_I64,
194
  WASM_INS_F32_DEMOTE_F64,
195
  WASM_INS_F64_CONVERT_S_I32,
196
  WASM_INS_F64_CONVERT_U_I32,
197
  WASM_INS_F64_CONVERT_S_I64,
198
  WASM_INS_F64_CONVERT_U_I64,
199
  WASM_INS_F64_PROMOTE_F32,
200
  WASM_INS_I32_REINTERPRET_F32,
201
  WASM_INS_I64_REINTERPRET_F64,
202
  WASM_INS_F32_REINTERPRET_I32,
203
  WASM_INS_F64_REINTERPRET_I64,
204
  -1,
205
  -1,
206
  -1,
207
  -1,
208
  -1,
209
  -1,
210
  -1,
211
  -1,
212
  -1,
213
  -1,
214
  -1,
215
  -1,
216
  -1,
217
  -1,
218
  -1,
219
  -1,
220
  -1,
221
  -1,
222
  -1,
223
  -1,
224
  -1,
225
  -1,
226
  -1,
227
  -1,
228
  -1,
229
  -1,
230
  -1,
231
  -1,
232
  -1,
233
  -1,
234
  -1,
235
  -1,
236
  -1,
237
  -1,
238
  -1,
239
  -1,
240
  -1,
241
  -1,
242
  -1,
243
  -1,
244
  -1,
245
  -1,
246
  -1,
247
  -1,
248
  -1,
249
  -1,
250
  -1,
251
  -1,
252
  -1,
253
  -1,
254
  -1,
255
  -1,
256
  -1,
257
  -1,
258
  -1,
259
  -1,
260
  -1,
261
  -1,
262
  -1,
263
  -1,
264
  -1,
265
  -1,
266
  -1,
267
  -1,
268
};
269
270
// input  | code: code pointer start from varuint32
271
//        | code_len: real code len count from varint
272
//        | leng: return value, means length of varint. -1 means error
273
// return | varint
274
static uint32_t get_varuint32(const uint8_t *code, size_t code_len,
275
            size_t *leng)
276
22.2k
{
277
22.2k
  uint32_t data = 0;
278
22.2k
  int i;
279
280
33.2k
  for (i = 0;; i++) {
281
33.2k
    if (code_len < i + 1) {
282
254
      *leng = -1;
283
254
      return 0;
284
254
    }
285
286
33.0k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
287
488
      *leng = -1;
288
488
      return 0;
289
488
    }
290
291
32.5k
    data = data + (((uint32_t)code[i] & 0x7f) << (i * 7));
292
32.5k
    if (code[i] >> 7 == 0) {
293
21.4k
      break;
294
21.4k
    }
295
32.5k
  }
296
297
21.4k
  *leng = i + 1;
298
299
21.4k
  return data;
300
22.2k
}
301
302
// input  | code : code pointer start from varuint64
303
//        | code_len : real code len count from varint
304
//        | leng: return value, means length of varint. -1 means error
305
// return   | varint
306
static uint64_t get_varuint64(const uint8_t *code, size_t code_len,
307
            size_t *leng)
308
370
{
309
370
  uint64_t data;
310
370
  int i;
311
312
370
  data = 0;
313
1.52k
  for (i = 0;; i++) {
314
1.52k
    if (code_len < i + 1) {
315
8
      *leng = -1;
316
8
      return 0;
317
8
    }
318
319
1.51k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
1
      *leng = -1;
321
1
      return 0;
322
1
    }
323
324
1.51k
    data = data + (((uint64_t)code[i] & 0x7f) << (i * 7));
325
1.51k
    if (code[i] >> 7 == 0) {
326
361
      break;
327
361
    }
328
1.51k
  }
329
330
361
  *leng = i + 1;
331
332
361
  return data;
333
370
}
334
335
// input  | code : code pointer start from uint32
336
//      | dest : the pointer where we store the uint32
337
// return | None
338
static void get_uint32(const uint8_t *code, uint32_t *dest)
339
420
{
340
420
  memcpy(dest, code, 4);
341
420
}
342
343
// input  | code : code pointer start from uint32
344
//      | dest : the pointer where we store the uint64
345
// return   | None
346
static void get_uint64(const uint8_t *code, uint64_t *dest)
347
446
{
348
446
  memcpy(dest, code, 8);
349
446
}
350
351
// input  | code : code pointer start from varint7
352
//      | code_len : start from the code pointer to the end, how long is it
353
//      | leng : length of the param , -1 means error
354
// return   | data of varint7
355
static int8_t get_varint7(const uint8_t *code, size_t code_len, size_t *leng)
356
892
{
357
892
  int8_t data;
358
359
892
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
892
  *leng = 1;
365
366
892
  if (code[0] == 0x40) {
367
197
    return -1;
368
197
  }
369
370
695
  data = code[0] & 0x7f;
371
372
695
  return data;
373
892
}
374
375
// input  | code : code pointer start from varuint32
376
//      | code_len : start from the code pointer to the end, how long is it
377
//      | param_size : pointer of the param size
378
//      | MI : Mcinst handler in this round of disasm
379
// return   | true/false if the function successfully finished
380
static bool read_varuint32(const uint8_t *code, size_t code_len,
381
         uint16_t *param_size, MCInst *MI)
382
2.53k
{
383
2.53k
  size_t len = 0;
384
2.53k
  uint32_t data;
385
386
2.53k
  data = get_varuint32(code, code_len, &len);
387
2.53k
  if (len == -1) {
388
24
    return false;
389
24
  }
390
391
2.50k
  if (MI->flat_insn->detail) {
392
2.50k
    MI->flat_insn->detail->wasm.op_count = 1;
393
2.50k
    MI->flat_insn->detail->wasm.operands[0].type =
394
2.50k
      WASM_OP_VARUINT32;
395
2.50k
    MI->flat_insn->detail->wasm.operands[0].size = len;
396
2.50k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data;
397
2.50k
  }
398
399
2.50k
  MI->wasm_data.size = len;
400
2.50k
  MI->wasm_data.type = WASM_OP_VARUINT32;
401
2.50k
  MI->wasm_data.uint32 = data;
402
2.50k
  *param_size = len;
403
404
2.50k
  return true;
405
2.53k
}
406
407
// input  | code : code pointer start from varuint64
408
//      | code_len : start from the code pointer to the end, how long is it
409
//      | param_size : pointer of the param size
410
//      | MI : Mcinst handler in this round of disasm
411
// return   | true/false if the function successfully finished
412
static bool read_varuint64(const uint8_t *code, size_t code_len,
413
         uint16_t *param_size, MCInst *MI)
414
370
{
415
370
  size_t len = 0;
416
370
  uint64_t data;
417
418
370
  data = get_varuint64(code, code_len, &len);
419
370
  if (len == -1) {
420
9
    return false;
421
9
  }
422
423
361
  if (MI->flat_insn->detail) {
424
361
    MI->flat_insn->detail->wasm.op_count = 1;
425
361
    MI->flat_insn->detail->wasm.operands[0].type =
426
361
      WASM_OP_VARUINT64;
427
361
    MI->flat_insn->detail->wasm.operands[0].size = len;
428
361
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
429
361
  }
430
431
361
  MI->wasm_data.size = len;
432
361
  MI->wasm_data.type = WASM_OP_VARUINT64;
433
361
  MI->wasm_data.uint64 = data;
434
361
  *param_size = len;
435
436
361
  return true;
437
370
}
438
439
// input  | code : code pointer start from memoryimmediate
440
//      | code_len : start from the code pointer to the end, how long is it
441
//      | param_size : pointer of the param size (sum of two params)
442
//      | MI : Mcinst handler in this round of disasm
443
// return   | true/false if the function successfully finished
444
static bool read_memoryimmediate(const uint8_t *code, size_t code_len,
445
         uint16_t *param_size, MCInst *MI)
446
6.05k
{
447
6.05k
  size_t tmp, len = 0;
448
6.05k
  uint32_t data[2];
449
450
6.05k
  if (MI->flat_insn->detail) {
451
6.05k
    MI->flat_insn->detail->wasm.op_count = 2;
452
6.05k
  }
453
454
6.05k
  data[0] = get_varuint32(code, code_len, &tmp);
455
6.05k
  if (tmp == -1) {
456
32
    return false;
457
32
  }
458
459
6.02k
  if (MI->flat_insn->detail) {
460
6.02k
    MI->flat_insn->detail->wasm.operands[0].type =
461
6.02k
      WASM_OP_VARUINT32;
462
6.02k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
463
6.02k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
464
6.02k
  }
465
466
6.02k
  len = tmp;
467
6.02k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
468
469
6.02k
  if (MI->flat_insn->detail) {
470
6.02k
    MI->flat_insn->detail->wasm.operands[1].type =
471
6.02k
      WASM_OP_VARUINT32;
472
6.02k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
473
6.02k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
474
6.02k
  }
475
476
6.02k
  len += tmp;
477
6.02k
  MI->wasm_data.size = len;
478
6.02k
  MI->wasm_data.type = WASM_OP_IMM;
479
6.02k
  MI->wasm_data.immediate[0] = data[0];
480
6.02k
  MI->wasm_data.immediate[1] = data[1];
481
6.02k
  *param_size = len;
482
483
6.02k
  return true;
484
6.05k
}
485
486
// input  | code : code pointer start from uint32
487
//      | code_len : start from the code pointer to the end, how long is it
488
//      | param_size : pointer of the param size
489
//      | MI : Mcinst handler in this round of disasm
490
// return   | true/false if the function successfully finished
491
static bool read_uint32(const uint8_t *code, size_t code_len,
492
      uint16_t *param_size, MCInst *MI)
493
215
{
494
215
  if (code_len < 4) {
495
5
    return false;
496
5
  }
497
498
210
  get_uint32(code, &(MI->wasm_data.uint32));
499
500
210
  if (MI->flat_insn->detail) {
501
210
    MI->flat_insn->detail->wasm.op_count = 1;
502
210
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
503
210
    MI->flat_insn->detail->wasm.operands[0].size = 4;
504
210
    get_uint32(code,
505
210
         &(MI->flat_insn->detail->wasm.operands[0].uint32));
506
210
  }
507
508
210
  MI->wasm_data.size = 4;
509
210
  MI->wasm_data.type = WASM_OP_UINT32;
510
210
  *param_size = 4;
511
512
210
  return true;
513
215
}
514
515
// input  | code : code pointer start from uint64
516
//      | code_len : start from the code pointer to the end, how long is it
517
//      | param_size : pointer of the param size
518
//      | MI : Mcinst handler in this round of disasm
519
// return   | true/false if the function successfully finished
520
static bool read_uint64(const uint8_t *code, size_t code_len,
521
      uint16_t *param_size, MCInst *MI)
522
231
{
523
231
  if (code_len < 8) {
524
8
    return false;
525
8
  }
526
527
223
  get_uint64(code, &(MI->wasm_data.uint64));
528
529
223
  if (MI->flat_insn->detail) {
530
223
    MI->flat_insn->detail->wasm.op_count = 1;
531
223
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
532
223
    MI->flat_insn->detail->wasm.operands[0].size = 8;
533
223
    get_uint64(code,
534
223
         &(MI->flat_insn->detail->wasm.operands[0].uint64));
535
223
  }
536
537
223
  MI->wasm_data.size = 8;
538
223
  MI->wasm_data.type = WASM_OP_UINT64;
539
223
  *param_size = 8;
540
541
223
  return true;
542
231
}
543
544
// input  | code : code pointer start from brtable
545
//      | code_len : start from the code pointer to the end, how long is it
546
//      | param_size : pointer of the param size (sum of all param)
547
//      | MI : Mcinst handler in this round of disasm
548
// return   | true/false if the function successfully finished
549
static bool read_brtable(const uint8_t *code, size_t code_len,
550
       uint16_t *param_size, MCInst *MI)
551
1.29k
{
552
1.29k
  uint32_t length, default_target;
553
1.29k
  int tmp_len = 0, i;
554
1.29k
  size_t var_len;
555
556
  // read length
557
1.29k
  length = get_varuint32(code, code_len, &var_len);
558
1.29k
  if (var_len == -1) {
559
10
    return false;
560
10
  }
561
562
1.28k
  tmp_len += var_len;
563
1.28k
  MI->wasm_data.brtable.length = length;
564
1.28k
  if (length >= UINT32_MAX - tmp_len) {
565
    // integer overflow check
566
1
    return false;
567
1
  }
568
1.28k
  if (code_len < tmp_len + length) {
569
    // safety check that we have minimum enough data to read
570
55
    return false;
571
55
  }
572
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
573
1.22k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
574
575
1.22k
  if (MI->flat_insn->detail) {
576
1.22k
    MI->flat_insn->detail->wasm.op_count = 1;
577
1.22k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
578
1.22k
    MI->flat_insn->detail->wasm.operands[0].brtable.length =
579
1.22k
      MI->wasm_data.brtable.length;
580
1.22k
    MI->flat_insn->detail->wasm.operands[0].brtable.address =
581
1.22k
      MI->wasm_data.brtable.address;
582
1.22k
  }
583
584
  // read data
585
6.31k
  for (i = 0; i < length; i++) {
586
5.11k
    if (code_len < tmp_len) {
587
0
      return false;
588
0
    }
589
590
5.11k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
591
5.11k
    if (var_len == -1) {
592
24
      return false;
593
24
    }
594
595
5.08k
    tmp_len += var_len;
596
5.08k
  }
597
598
  // read default target
599
1.20k
  default_target =
600
1.20k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
601
1.20k
  if (var_len == -1) {
602
22
    return false;
603
22
  }
604
605
1.18k
  MI->wasm_data.brtable.default_target = default_target;
606
1.18k
  MI->wasm_data.type = WASM_OP_BRTABLE;
607
1.18k
  *param_size = tmp_len + var_len;
608
609
1.18k
  if (MI->flat_insn->detail) {
610
1.18k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
611
1.18k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target =
612
1.18k
      MI->wasm_data.brtable.default_target;
613
1.18k
  }
614
615
1.18k
  return true;
616
1.20k
}
617
618
// input  | code : code pointer start from varint7
619
//      | code_len : start from the code pointer to the end, how long is it
620
//      | param_size : pointer of the param size
621
//      | MI : Mcinst handler in this round of disasm
622
// return   | true/false if the function successfully finished
623
static bool read_varint7(const uint8_t *code, size_t code_len,
624
       uint16_t *param_size, MCInst *MI)
625
892
{
626
892
  size_t len = 0;
627
628
892
  MI->wasm_data.type = WASM_OP_INT7;
629
892
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
630
892
  if (len == -1) {
631
0
    return false;
632
0
  }
633
634
892
  if (MI->flat_insn->detail) {
635
892
    MI->flat_insn->detail->wasm.op_count = 1;
636
892
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
637
892
    MI->flat_insn->detail->wasm.operands[0].size = 1;
638
892
    MI->flat_insn->detail->wasm.operands[0].int7 =
639
892
      MI->wasm_data.int7;
640
892
  }
641
642
892
  *param_size = len;
643
644
892
  return true;
645
892
}
646
647
bool WASM_getInstruction(csh ud, const uint8_t *code, size_t code_len,
648
       MCInst *MI, uint16_t *size, uint64_t address,
649
       void *inst_info)
650
63.6k
{
651
63.6k
  unsigned char opcode;
652
63.6k
  uint16_t param_size;
653
654
63.6k
  if (code_len == 0)
655
0
    return false;
656
657
63.6k
  opcode = code[0];
658
63.6k
  if (opcodes[opcode] == -1) {
659
    // invalid opcode
660
93
    return false;
661
93
  }
662
663
  // valid opcode
664
63.5k
  MI->address = address;
665
63.5k
  MI->OpcodePub = MI->Opcode = opcode;
666
667
63.5k
  if (MI->flat_insn->detail) {
668
63.5k
    memset(MI->flat_insn->detail, 0,
669
63.5k
           offsetof(cs_detail, wasm) + sizeof(cs_wasm));
670
63.5k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
671
63.5k
  }
672
673
  // setup groups
674
63.5k
  switch (opcode) {
675
0
  default:
676
0
    return false;
677
678
213
  case WASM_INS_I32_CONST:
679
213
    if (code_len == 1 ||
680
209
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
681
8
      return false;
682
8
    }
683
684
205
    if (MI->flat_insn->detail) {
685
205
      MI->flat_insn->detail->wasm.op_count = 1;
686
205
      MI->flat_insn->detail
687
205
        ->groups[MI->flat_insn->detail->groups_count] =
688
205
        WASM_GRP_NUMBERIC;
689
205
      MI->flat_insn->detail->groups_count++;
690
205
    }
691
692
205
    *size = param_size + 1;
693
694
205
    break;
695
696
373
  case WASM_INS_I64_CONST:
697
373
    if (code_len == 1 ||
698
370
        !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
699
12
      return false;
700
12
    }
701
702
361
    if (MI->flat_insn->detail) {
703
361
      MI->flat_insn->detail->wasm.op_count = 1;
704
361
      MI->flat_insn->detail
705
361
        ->groups[MI->flat_insn->detail->groups_count] =
706
361
        WASM_GRP_NUMBERIC;
707
361
      MI->flat_insn->detail->groups_count++;
708
361
    }
709
710
361
    *size = param_size + 1;
711
712
361
    break;
713
714
217
  case WASM_INS_F32_CONST:
715
217
    if (code_len == 1 ||
716
215
        !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
717
7
      return false;
718
7
    }
719
720
210
    if (MI->flat_insn->detail) {
721
210
      MI->flat_insn->detail->wasm.op_count = 1;
722
210
      MI->flat_insn->detail
723
210
        ->groups[MI->flat_insn->detail->groups_count] =
724
210
        WASM_GRP_NUMBERIC;
725
210
      MI->flat_insn->detail->groups_count++;
726
210
    }
727
728
210
    *size = param_size + 1;
729
730
210
    break;
731
732
233
  case WASM_INS_F64_CONST:
733
233
    if (code_len == 1 ||
734
231
        !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
735
10
      return false;
736
10
    }
737
738
223
    if (MI->flat_insn->detail) {
739
223
      MI->flat_insn->detail->wasm.op_count = 1;
740
223
      MI->flat_insn->detail
741
223
        ->groups[MI->flat_insn->detail->groups_count] =
742
223
        WASM_GRP_NUMBERIC;
743
223
      MI->flat_insn->detail->groups_count++;
744
223
    }
745
746
223
    *size = param_size + 1;
747
748
223
    break;
749
750
199
  case WASM_INS_I32_EQZ:
751
414
  case WASM_INS_I32_EQ:
752
682
  case WASM_INS_I32_NE:
753
883
  case WASM_INS_I32_LT_S:
754
1.08k
  case WASM_INS_I32_LT_U:
755
1.28k
  case WASM_INS_I32_GT_S:
756
1.54k
  case WASM_INS_I32_GT_U:
757
1.74k
  case WASM_INS_I32_LE_S:
758
1.94k
  case WASM_INS_I32_LE_U:
759
2.15k
  case WASM_INS_I32_GE_S:
760
2.35k
  case WASM_INS_I32_GE_U:
761
3.37k
  case WASM_INS_I64_EQZ:
762
3.58k
  case WASM_INS_I64_EQ:
763
4.23k
  case WASM_INS_I64_NE:
764
4.51k
  case WASM_INS_I64_LT_S:
765
4.71k
  case WASM_INS_I64_LT_U:
766
4.91k
  case WASN_INS_I64_GT_S:
767
5.11k
  case WASM_INS_I64_GT_U:
768
5.58k
  case WASM_INS_I64_LE_S:
769
5.79k
  case WASM_INS_I64_LE_U:
770
6.03k
  case WASM_INS_I64_GE_S:
771
6.42k
  case WASM_INS_I64_GE_U:
772
6.68k
  case WASM_INS_F32_EQ:
773
6.90k
  case WASM_INS_F32_NE:
774
7.12k
  case WASM_INS_F32_LT:
775
7.31k
  case WASM_INS_F32_GT:
776
7.52k
  case WASM_INS_F32_LE:
777
7.72k
  case WASM_INS_F32_GE:
778
13.3k
  case WASM_INS_F64_EQ:
779
13.6k
  case WASM_INS_F64_NE:
780
14.0k
  case WASM_INS_F64_LT:
781
14.4k
  case WASM_INS_F64_GT:
782
15.2k
  case WASM_INS_F64_LE:
783
15.5k
  case WASM_INS_F64_GE:
784
15.7k
  case WASM_INS_I32_CLZ:
785
15.9k
  case WASM_INS_I32_CTZ:
786
16.1k
  case WASM_INS_I32_POPCNT:
787
16.6k
  case WASM_INS_I32_ADD:
788
16.8k
  case WASM_INS_I32_SUB:
789
17.1k
  case WASM_INS_I32_MUL:
790
17.3k
  case WASM_INS_I32_DIV_S:
791
17.8k
  case WASM_INS_I32_DIV_U:
792
18.0k
  case WASM_INS_I32_REM_S:
793
18.3k
  case WASM_INS_I32_REM_U:
794
18.5k
  case WASM_INS_I32_AND:
795
18.7k
  case WASM_INS_I32_OR:
796
21.6k
  case WASM_INS_I32_XOR:
797
21.9k
  case WASM_INS_I32_SHL:
798
22.2k
  case WASM_INS_I32_SHR_S:
799
22.4k
  case WASM_INS_I32_SHR_U:
800
22.6k
  case WASM_INS_I32_ROTL:
801
23.0k
  case WASM_INS_I32_ROTR:
802
23.2k
  case WASM_INS_I64_CLZ:
803
23.6k
  case WASM_INS_I64_CTZ:
804
24.0k
  case WASM_INS_I64_POPCNT:
805
24.2k
  case WASM_INS_I64_ADD:
806
24.4k
  case WASM_INS_I64_SUB:
807
24.6k
  case WASM_INS_I64_MUL:
808
24.8k
  case WASM_INS_I64_DIV_S:
809
28.6k
  case WASM_INS_I64_DIV_U:
810
30.5k
  case WASM_INS_I64_REM_S:
811
30.7k
  case WASM_INS_I64_REM_U:
812
30.9k
  case WASM_INS_I64_AND:
813
31.1k
  case WASM_INS_I64_OR:
814
31.5k
  case WASM_INS_I64_XOR:
815
31.7k
  case WASM_INS_I64_SHL:
816
32.0k
  case WASM_INS_I64_SHR_S:
817
32.3k
  case WASM_INS_I64_SHR_U:
818
32.5k
  case WASM_INS_I64_ROTL:
819
32.7k
  case WASM_INS_I64_ROTR:
820
32.9k
  case WASM_INS_F32_ABS:
821
33.1k
  case WASM_INS_F32_NEG:
822
33.4k
  case WASM_INS_F32_CEIL:
823
33.6k
  case WASM_INS_F32_FLOOR:
824
33.9k
  case WASM_INS_F32_TRUNC:
825
34.1k
  case WASM_INS_F32_NEAREST:
826
34.3k
  case WASM_INS_F32_SQRT:
827
34.6k
  case WASM_INS_F32_ADD:
828
34.8k
  case WASM_INS_F32_SUB:
829
35.0k
  case WASM_INS_F32_MUL:
830
35.3k
  case WASM_INS_F32_DIV:
831
35.5k
  case WASM_INS_F32_MIN:
832
35.8k
  case WASM_INS_F32_MAX:
833
36.0k
  case WASM_INS_F32_COPYSIGN:
834
36.2k
  case WASM_INS_F64_ABS:
835
36.5k
  case WASM_INS_F64_NEG:
836
36.7k
  case WASM_INS_F64_CEIL:
837
36.9k
  case WASM_INS_F64_FLOOR:
838
37.3k
  case WASM_INS_F64_TRUNC:
839
37.5k
  case WASM_INS_F64_NEAREST:
840
37.7k
  case WASM_INS_F64_SQRT:
841
38.0k
  case WASM_INS_F64_ADD:
842
38.2k
  case WASM_INS_F64_SUB:
843
38.5k
  case WASM_INS_F64_MUL:
844
38.7k
  case WASM_INS_F64_DIV:
845
39.1k
  case WASM_INS_F64_MIN:
846
39.3k
  case WASM_INS_F64_MAX:
847
39.5k
  case WASM_INS_F64_COPYSIGN:
848
39.7k
  case WASM_INS_I32_WARP_I64:
849
39.9k
  case WASP_INS_I32_TRUNC_S_F32:
850
40.2k
  case WASM_INS_I32_TRUNC_U_F32:
851
40.4k
  case WASM_INS_I32_TRUNC_S_F64:
852
41.8k
  case WASM_INS_I32_TRUNC_U_F64:
853
42.2k
  case WASM_INS_I64_EXTEND_S_I32:
854
42.4k
  case WASM_INS_I64_EXTEND_U_I32:
855
42.7k
  case WASM_INS_I64_TRUNC_S_F32:
856
43.4k
  case WASM_INS_I64_TRUNC_U_F32:
857
43.6k
  case WASM_INS_I64_TRUNC_S_F64:
858
43.8k
  case WASM_INS_I64_TRUNC_U_F64:
859
44.0k
  case WASM_INS_F32_CONVERT_S_I32:
860
44.2k
  case WASM_INS_F32_CONVERT_U_I32:
861
44.4k
  case WASM_INS_F32_CONVERT_S_I64:
862
44.6k
  case WASM_INS_F32_CONVERT_U_I64:
863
44.8k
  case WASM_INS_F32_DEMOTE_F64:
864
45.0k
  case WASM_INS_F64_CONVERT_S_I32:
865
45.2k
  case WASM_INS_F64_CONVERT_U_I32:
866
45.4k
  case WASM_INS_F64_CONVERT_S_I64:
867
45.7k
  case WASM_INS_F64_CONVERT_U_I64:
868
45.9k
  case WASM_INS_F64_PROMOTE_F32:
869
46.4k
  case WASM_INS_I32_REINTERPRET_F32:
870
46.6k
  case WASM_INS_I64_REINTERPRET_F64:
871
46.7k
  case WASM_INS_F32_REINTERPRET_I32:
872
47.0k
  case WASM_INS_F64_REINTERPRET_I64:
873
47.0k
    MI->wasm_data.type = WASM_OP_NONE;
874
875
47.0k
    if (MI->flat_insn->detail) {
876
47.0k
      MI->flat_insn->detail->wasm.op_count = 0;
877
47.0k
      MI->flat_insn->detail
878
47.0k
        ->groups[MI->flat_insn->detail->groups_count] =
879
47.0k
        WASM_GRP_NUMBERIC;
880
47.0k
      MI->flat_insn->detail->groups_count++;
881
47.0k
    }
882
883
47.0k
    *size = 1;
884
885
47.0k
    break;
886
887
200
  case WASM_INS_DROP:
888
432
  case WASM_INS_SELECT:
889
432
    MI->wasm_data.type = WASM_OP_NONE;
890
891
432
    if (MI->flat_insn->detail) {
892
432
      MI->flat_insn->detail->wasm.op_count = 0;
893
432
      MI->flat_insn->detail
894
432
        ->groups[MI->flat_insn->detail->groups_count] =
895
432
        WASM_GRP_PARAMETRIC;
896
432
      MI->flat_insn->detail->groups_count++;
897
432
    }
898
899
432
    *size = 1;
900
901
432
    break;
902
903
246
  case WASM_INS_GET_LOCAL:
904
512
  case WASM_INS_SET_LOCAL:
905
733
  case WASM_INS_TEE_LOCAL:
906
943
  case WASM_INS_GET_GLOBAL:
907
1.15k
  case WASM_INS_SET_GLOBAL:
908
1.15k
    if (code_len == 1 ||
909
1.12k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
910
41
      return false;
911
41
    }
912
913
1.11k
    if (MI->flat_insn->detail) {
914
1.11k
      MI->flat_insn->detail->wasm.op_count = 1;
915
1.11k
      MI->flat_insn->detail
916
1.11k
        ->groups[MI->flat_insn->detail->groups_count] =
917
1.11k
        WASM_GRP_VARIABLE;
918
1.11k
      MI->flat_insn->detail->groups_count++;
919
1.11k
    }
920
921
1.11k
    *size = param_size + 1;
922
923
1.11k
    break;
924
925
449
  case WASM_INS_I32_LOAD:
926
660
  case WASM_INS_I64_LOAD:
927
1.14k
  case WASM_INS_F32_LOAD:
928
1.34k
  case WASM_INS_F64_LOAD:
929
1.59k
  case WASM_INS_I32_LOAD8_S:
930
1.81k
  case WASM_INS_I32_LOAD8_U:
931
2.03k
  case WASM_INS_I32_LOAD16_S:
932
2.31k
  case WASM_INS_I32_LOAD16_U:
933
2.69k
  case WASM_INS_I64_LOAD8_S:
934
2.90k
  case WASM_INS_I64_LOAD8_U:
935
3.12k
  case WASM_INS_I64_LOAD16_S:
936
3.41k
  case WASM_INS_I64_LOAD16_U:
937
3.62k
  case WASM_INS_I64_LOAD32_S:
938
4.08k
  case WASM_INS_I64_LOAD32_U:
939
4.35k
  case WASM_INS_I32_STORE:
940
4.55k
  case WASM_INS_I64_STORE:
941
4.76k
  case WASM_INS_F32_STORE:
942
4.96k
  case WASM_INS_F64_STORE:
943
5.19k
  case WASM_INS_I32_STORE8:
944
5.41k
  case WASM_INS_I32_STORE16:
945
5.65k
  case WASM_INS_I64_STORE8:
946
6.02k
  case WASM_INS_I64_STORE16:
947
6.22k
  case WASM_INS_I64_STORE32:
948
6.22k
    if (code_len == 1 ||
949
6.05k
        !read_memoryimmediate(&code[1], code_len - 1, &param_size,
950
6.05k
            MI)) {
951
196
      return false;
952
196
    }
953
954
6.02k
    if (MI->flat_insn->detail) {
955
6.02k
      MI->flat_insn->detail->wasm.op_count = 2;
956
6.02k
      MI->flat_insn->detail
957
6.02k
        ->groups[MI->flat_insn->detail->groups_count] =
958
6.02k
        WASM_GRP_MEMORY;
959
6.02k
      MI->flat_insn->detail->groups_count++;
960
6.02k
    }
961
962
6.02k
    *size = param_size + 1;
963
964
6.02k
    break;
965
966
244
  case WASM_INS_CURRENT_MEMORY:
967
533
  case WASM_INS_GROW_MEMORY:
968
533
    MI->wasm_data.type = WASM_OP_NONE;
969
970
533
    if (MI->flat_insn->detail) {
971
533
      MI->flat_insn->detail->wasm.op_count = 0;
972
533
      MI->flat_insn->detail
973
533
        ->groups[MI->flat_insn->detail->groups_count] =
974
533
        WASM_GRP_MEMORY;
975
533
      MI->flat_insn->detail->groups_count++;
976
533
    }
977
978
533
    *size = 1;
979
980
533
    break;
981
982
2.70k
  case WASM_INS_UNREACHABLE:
983
2.98k
  case WASM_INS_NOP:
984
3.18k
  case WASM_INS_ELSE:
985
3.39k
  case WASM_INS_END:
986
3.63k
  case WASM_INS_RETURN:
987
3.63k
    MI->wasm_data.type = WASM_OP_NONE;
988
989
3.63k
    if (MI->flat_insn->detail) {
990
3.63k
      MI->flat_insn->detail->wasm.op_count = 0;
991
3.63k
      MI->flat_insn->detail
992
3.63k
        ->groups[MI->flat_insn->detail->groups_count] =
993
3.63k
        WASM_GRP_CONTROL;
994
3.63k
      MI->flat_insn->detail->groups_count++;
995
3.63k
    }
996
997
3.63k
    *size = 1;
998
999
3.63k
    break;
1000
1001
255
  case WASM_INS_BLOCK:
1002
486
  case WASM_INS_LOOP:
1003
915
  case WASM_INS_IF:
1004
915
    if (code_len == 1 ||
1005
892
        !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
1006
23
      return false;
1007
23
    }
1008
1009
892
    if (MI->flat_insn->detail) {
1010
892
      MI->flat_insn->detail->wasm.op_count = 1;
1011
892
      MI->flat_insn->detail
1012
892
        ->groups[MI->flat_insn->detail->groups_count] =
1013
892
        WASM_GRP_CONTROL;
1014
892
      MI->flat_insn->detail->groups_count++;
1015
892
    }
1016
1017
892
    *size = param_size + 1;
1018
1019
892
    break;
1020
1021
204
  case WASM_INS_BR:
1022
668
  case WASM_INS_BR_IF:
1023
963
  case WASM_INS_CALL:
1024
1.22k
  case WASM_INS_CALL_INDIRECT:
1025
1.22k
    if (code_len == 1 ||
1026
1.19k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
1027
32
      return false;
1028
32
    }
1029
1030
1.19k
    if (MI->flat_insn->detail) {
1031
1.19k
      MI->flat_insn->detail->wasm.op_count = 1;
1032
1.19k
      MI->flat_insn->detail
1033
1.19k
        ->groups[MI->flat_insn->detail->groups_count] =
1034
1.19k
        WASM_GRP_CONTROL;
1035
1.19k
      MI->flat_insn->detail->groups_count++;
1036
1.19k
    }
1037
1038
1.19k
    *size = param_size + 1;
1039
1040
1.19k
    break;
1041
1042
1.29k
  case WASM_INS_BR_TABLE:
1043
1.29k
    if (code_len == 1 ||
1044
1.29k
        !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
1045
113
      return false;
1046
113
    }
1047
1048
1.18k
    if (MI->flat_insn->detail) {
1049
1.18k
      MI->flat_insn->detail->wasm.op_count = 1;
1050
1.18k
      MI->flat_insn->detail
1051
1.18k
        ->groups[MI->flat_insn->detail->groups_count] =
1052
1.18k
        WASM_GRP_CONTROL;
1053
1.18k
      MI->flat_insn->detail->groups_count++;
1054
1.18k
    }
1055
1056
1.18k
    *size = param_size + 1;
1057
1058
1.18k
    break;
1059
63.5k
  }
1060
1061
63.0k
  return true;
1062
63.5k
}