Coverage Report

Created: 2025-11-24 06:12

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
26.0k
{
277
26.0k
  uint32_t data = 0;
278
26.0k
  int i;
279
280
36.8k
  for (i = 0;; i++) {
281
36.8k
    if (code_len < i + 1) {
282
255
      *leng = -1;
283
255
      return 0;
284
255
    }
285
286
36.6k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
287
508
      *leng = -1;
288
508
      return 0;
289
508
    }
290
291
36.1k
    data = data + (((uint32_t)code[i] & 0x7f) << (i * 7));
292
36.1k
    if (code[i] >> 7 == 0) {
293
25.2k
      break;
294
25.2k
    }
295
36.1k
  }
296
297
25.2k
  *leng = i + 1;
298
299
25.2k
  return data;
300
26.0k
}
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
308
{
309
308
  uint64_t data;
310
308
  int i;
311
312
308
  data = 0;
313
1.20k
  for (i = 0;; i++) {
314
1.20k
    if (code_len < i + 1) {
315
6
      *leng = -1;
316
6
      return 0;
317
6
    }
318
319
1.19k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
1
      *leng = -1;
321
1
      return 0;
322
1
    }
323
324
1.19k
    data = data + (((uint64_t)code[i] & 0x7f) << (i * 7));
325
1.19k
    if (code[i] >> 7 == 0) {
326
301
      break;
327
301
    }
328
1.19k
  }
329
330
301
  *leng = i + 1;
331
332
301
  return data;
333
308
}
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
406
{
340
406
  memcpy(dest, code, 4);
341
406
}
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
450
{
348
450
  memcpy(dest, code, 8);
349
450
}
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
785
{
357
785
  int8_t data;
358
359
785
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
785
  *leng = 1;
365
366
785
  if (code[0] == 0x40) {
367
67
    return -1;
368
67
  }
369
370
718
  data = code[0] & 0x7f;
371
372
718
  return data;
373
785
}
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.88k
{
383
2.88k
  size_t len = 0;
384
2.88k
  uint32_t data;
385
386
2.88k
  data = get_varuint32(code, code_len, &len);
387
2.88k
  if (len == -1) {
388
17
    return false;
389
17
  }
390
391
2.86k
  if (MI->flat_insn->detail) {
392
2.86k
    MI->flat_insn->detail->wasm.op_count = 1;
393
2.86k
    MI->flat_insn->detail->wasm.operands[0].type =
394
2.86k
      WASM_OP_VARUINT32;
395
2.86k
    MI->flat_insn->detail->wasm.operands[0].size = len;
396
2.86k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data;
397
2.86k
  }
398
399
2.86k
  MI->wasm_data.size = len;
400
2.86k
  MI->wasm_data.type = WASM_OP_VARUINT32;
401
2.86k
  MI->wasm_data.uint32 = data;
402
2.86k
  *param_size = len;
403
404
2.86k
  return true;
405
2.88k
}
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
308
{
415
308
  size_t len = 0;
416
308
  uint64_t data;
417
418
308
  data = get_varuint64(code, code_len, &len);
419
308
  if (len == -1) {
420
7
    return false;
421
7
  }
422
423
301
  if (MI->flat_insn->detail) {
424
301
    MI->flat_insn->detail->wasm.op_count = 1;
425
301
    MI->flat_insn->detail->wasm.operands[0].type =
426
301
      WASM_OP_VARUINT64;
427
301
    MI->flat_insn->detail->wasm.operands[0].size = len;
428
301
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
429
301
  }
430
431
301
  MI->wasm_data.size = len;
432
301
  MI->wasm_data.type = WASM_OP_VARUINT64;
433
301
  MI->wasm_data.uint64 = data;
434
301
  *param_size = len;
435
436
301
  return true;
437
308
}
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.21k
{
447
6.21k
  size_t tmp, len = 0;
448
6.21k
  uint32_t data[2];
449
450
6.21k
  if (MI->flat_insn->detail) {
451
6.21k
    MI->flat_insn->detail->wasm.op_count = 2;
452
6.21k
  }
453
454
6.21k
  data[0] = get_varuint32(code, code_len, &tmp);
455
6.21k
  if (tmp == -1) {
456
35
    return false;
457
35
  }
458
459
6.18k
  if (MI->flat_insn->detail) {
460
6.18k
    MI->flat_insn->detail->wasm.operands[0].type =
461
6.18k
      WASM_OP_VARUINT32;
462
6.18k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
463
6.18k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
464
6.18k
  }
465
466
6.18k
  len = tmp;
467
6.18k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
468
469
6.18k
  if (MI->flat_insn->detail) {
470
6.18k
    MI->flat_insn->detail->wasm.operands[1].type =
471
6.18k
      WASM_OP_VARUINT32;
472
6.18k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
473
6.18k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
474
6.18k
  }
475
476
6.18k
  len += tmp;
477
6.18k
  MI->wasm_data.size = len;
478
6.18k
  MI->wasm_data.type = WASM_OP_IMM;
479
6.18k
  MI->wasm_data.immediate[0] = data[0];
480
6.18k
  MI->wasm_data.immediate[1] = data[1];
481
6.18k
  *param_size = len;
482
483
6.18k
  return true;
484
6.21k
}
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
209
{
494
209
  if (code_len < 4) {
495
6
    return false;
496
6
  }
497
498
203
  get_uint32(code, &(MI->wasm_data.uint32));
499
500
203
  if (MI->flat_insn->detail) {
501
203
    MI->flat_insn->detail->wasm.op_count = 1;
502
203
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
503
203
    MI->flat_insn->detail->wasm.operands[0].size = 4;
504
203
    get_uint32(code,
505
203
         &(MI->flat_insn->detail->wasm.operands[0].uint32));
506
203
  }
507
508
203
  MI->wasm_data.size = 4;
509
203
  MI->wasm_data.type = WASM_OP_UINT32;
510
203
  *param_size = 4;
511
512
203
  return true;
513
209
}
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
234
{
523
234
  if (code_len < 8) {
524
9
    return false;
525
9
  }
526
527
225
  get_uint64(code, &(MI->wasm_data.uint64));
528
529
225
  if (MI->flat_insn->detail) {
530
225
    MI->flat_insn->detail->wasm.op_count = 1;
531
225
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
532
225
    MI->flat_insn->detail->wasm.operands[0].size = 8;
533
225
    get_uint64(code,
534
225
         &(MI->flat_insn->detail->wasm.operands[0].uint64));
535
225
  }
536
537
225
  MI->wasm_data.size = 8;
538
225
  MI->wasm_data.type = WASM_OP_UINT64;
539
225
  *param_size = 8;
540
541
225
  return true;
542
234
}
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.35k
{
552
1.35k
  uint32_t length, default_target;
553
1.35k
  int tmp_len = 0, i;
554
1.35k
  size_t var_len;
555
556
  // read length
557
1.35k
  length = get_varuint32(code, code_len, &var_len);
558
1.35k
  if (var_len == -1) {
559
11
    return false;
560
11
  }
561
562
1.34k
  tmp_len += var_len;
563
1.34k
  MI->wasm_data.brtable.length = length;
564
1.34k
  if (length >= UINT32_MAX - tmp_len) {
565
    // integer overflow check
566
1
    return false;
567
1
  }
568
1.34k
  if (code_len < tmp_len + length) {
569
    // safety check that we have minimum enough data to read
570
53
    return false;
571
53
  }
572
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
573
1.29k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
574
575
1.29k
  if (MI->flat_insn->detail) {
576
1.29k
    MI->flat_insn->detail->wasm.op_count = 1;
577
1.29k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
578
1.29k
    MI->flat_insn->detail->wasm.operands[0].brtable.length =
579
1.29k
      MI->wasm_data.brtable.length;
580
1.29k
    MI->flat_insn->detail->wasm.operands[0].brtable.address =
581
1.29k
      MI->wasm_data.brtable.address;
582
1.29k
  }
583
584
  // read data
585
9.39k
  for (i = 0; i < length; i++) {
586
8.13k
    if (code_len < tmp_len) {
587
0
      return false;
588
0
    }
589
590
8.13k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
591
8.13k
    if (var_len == -1) {
592
24
      return false;
593
24
    }
594
595
8.10k
    tmp_len += var_len;
596
8.10k
  }
597
598
  // read default target
599
1.26k
  default_target =
600
1.26k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
601
1.26k
  if (var_len == -1) {
602
18
    return false;
603
18
  }
604
605
1.24k
  MI->wasm_data.brtable.default_target = default_target;
606
1.24k
  MI->wasm_data.type = WASM_OP_BRTABLE;
607
1.24k
  *param_size = tmp_len + var_len;
608
609
1.24k
  if (MI->flat_insn->detail) {
610
1.24k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
611
1.24k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target =
612
1.24k
      MI->wasm_data.brtable.default_target;
613
1.24k
  }
614
615
1.24k
  return true;
616
1.26k
}
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
785
{
626
785
  size_t len = 0;
627
628
785
  MI->wasm_data.type = WASM_OP_INT7;
629
785
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
630
785
  if (len == -1) {
631
0
    return false;
632
0
  }
633
634
785
  if (MI->flat_insn->detail) {
635
785
    MI->flat_insn->detail->wasm.op_count = 1;
636
785
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
637
785
    MI->flat_insn->detail->wasm.operands[0].size = 1;
638
785
    MI->flat_insn->detail->wasm.operands[0].int7 =
639
785
      MI->wasm_data.int7;
640
785
  }
641
642
785
  *param_size = len;
643
644
785
  return true;
645
785
}
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
62.9k
{
651
62.9k
  unsigned char opcode;
652
62.9k
  uint16_t param_size;
653
654
62.9k
  if (code_len == 0)
655
0
    return false;
656
657
62.9k
  opcode = code[0];
658
62.9k
  if (opcodes[opcode] == -1) {
659
    // invalid opcode
660
100
    return false;
661
100
  }
662
663
  // valid opcode
664
62.8k
  MI->address = address;
665
62.8k
  MI->OpcodePub = MI->Opcode = opcode;
666
667
62.8k
  if (MI->flat_insn->detail) {
668
62.8k
    memset(MI->flat_insn->detail, 0,
669
62.8k
           offsetof(cs_detail, wasm) + sizeof(cs_wasm));
670
62.8k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
671
62.8k
  }
672
673
  // setup groups
674
62.8k
  switch (opcode) {
675
0
  default:
676
0
    return false;
677
678
212
  case WASM_INS_I32_CONST:
679
212
    if (code_len == 1 ||
680
209
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
681
8
      return false;
682
8
    }
683
684
204
    if (MI->flat_insn->detail) {
685
204
      MI->flat_insn->detail->wasm.op_count = 1;
686
204
      MI->flat_insn->detail
687
204
        ->groups[MI->flat_insn->detail->groups_count] =
688
204
        WASM_GRP_NUMBERIC;
689
204
      MI->flat_insn->detail->groups_count++;
690
204
    }
691
692
204
    *size = param_size + 1;
693
694
204
    break;
695
696
311
  case WASM_INS_I64_CONST:
697
311
    if (code_len == 1 ||
698
308
        !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
699
10
      return false;
700
10
    }
701
702
301
    if (MI->flat_insn->detail) {
703
301
      MI->flat_insn->detail->wasm.op_count = 1;
704
301
      MI->flat_insn->detail
705
301
        ->groups[MI->flat_insn->detail->groups_count] =
706
301
        WASM_GRP_NUMBERIC;
707
301
      MI->flat_insn->detail->groups_count++;
708
301
    }
709
710
301
    *size = param_size + 1;
711
712
301
    break;
713
714
210
  case WASM_INS_F32_CONST:
715
210
    if (code_len == 1 ||
716
209
        !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
717
7
      return false;
718
7
    }
719
720
203
    if (MI->flat_insn->detail) {
721
203
      MI->flat_insn->detail->wasm.op_count = 1;
722
203
      MI->flat_insn->detail
723
203
        ->groups[MI->flat_insn->detail->groups_count] =
724
203
        WASM_GRP_NUMBERIC;
725
203
      MI->flat_insn->detail->groups_count++;
726
203
    }
727
728
203
    *size = param_size + 1;
729
730
203
    break;
731
732
237
  case WASM_INS_F64_CONST:
733
237
    if (code_len == 1 ||
734
234
        !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
735
12
      return false;
736
12
    }
737
738
225
    if (MI->flat_insn->detail) {
739
225
      MI->flat_insn->detail->wasm.op_count = 1;
740
225
      MI->flat_insn->detail
741
225
        ->groups[MI->flat_insn->detail->groups_count] =
742
225
        WASM_GRP_NUMBERIC;
743
225
      MI->flat_insn->detail->groups_count++;
744
225
    }
745
746
225
    *size = param_size + 1;
747
748
225
    break;
749
750
226
  case WASM_INS_I32_EQZ:
751
441
  case WASM_INS_I32_EQ:
752
645
  case WASM_INS_I32_NE:
753
848
  case WASM_INS_I32_LT_S:
754
1.05k
  case WASM_INS_I32_LT_U:
755
1.26k
  case WASM_INS_I32_GT_S:
756
1.47k
  case WASM_INS_I32_GT_U:
757
1.90k
  case WASM_INS_I32_LE_S:
758
2.10k
  case WASM_INS_I32_LE_U:
759
2.29k
  case WASM_INS_I32_GE_S:
760
2.51k
  case WASM_INS_I32_GE_U:
761
2.84k
  case WASM_INS_I64_EQZ:
762
3.05k
  case WASM_INS_I64_EQ:
763
3.31k
  case WASM_INS_I64_NE:
764
3.51k
  case WASM_INS_I64_LT_S:
765
3.70k
  case WASM_INS_I64_LT_U:
766
3.91k
  case WASN_INS_I64_GT_S:
767
4.13k
  case WASM_INS_I64_GT_U:
768
4.82k
  case WASM_INS_I64_LE_S:
769
5.66k
  case WASM_INS_I64_LE_U:
770
6.36k
  case WASM_INS_I64_GE_S:
771
6.66k
  case WASM_INS_I64_GE_U:
772
6.88k
  case WASM_INS_F32_EQ:
773
7.24k
  case WASM_INS_F32_NE:
774
7.56k
  case WASM_INS_F32_LT:
775
7.81k
  case WASM_INS_F32_GT:
776
8.18k
  case WASM_INS_F32_LE:
777
8.39k
  case WASM_INS_F32_GE:
778
8.61k
  case WASM_INS_F64_EQ:
779
8.91k
  case WASM_INS_F64_NE:
780
9.34k
  case WASM_INS_F64_LT:
781
9.71k
  case WASM_INS_F64_GT:
782
13.7k
  case WASM_INS_F64_LE:
783
13.9k
  case WASM_INS_F64_GE:
784
14.1k
  case WASM_INS_I32_CLZ:
785
14.7k
  case WASM_INS_I32_CTZ:
786
14.9k
  case WASM_INS_I32_POPCNT:
787
15.1k
  case WASM_INS_I32_ADD:
788
15.4k
  case WASM_INS_I32_SUB:
789
15.7k
  case WASM_INS_I32_MUL:
790
15.9k
  case WASM_INS_I32_DIV_S:
791
16.3k
  case WASM_INS_I32_DIV_U:
792
16.6k
  case WASM_INS_I32_REM_S:
793
16.8k
  case WASM_INS_I32_REM_U:
794
17.0k
  case WASM_INS_I32_AND:
795
17.3k
  case WASM_INS_I32_OR:
796
17.5k
  case WASM_INS_I32_XOR:
797
17.9k
  case WASM_INS_I32_SHL:
798
18.3k
  case WASM_INS_I32_SHR_S:
799
18.7k
  case WASM_INS_I32_SHR_U:
800
18.9k
  case WASM_INS_I32_ROTL:
801
19.5k
  case WASM_INS_I32_ROTR:
802
19.7k
  case WASM_INS_I64_CLZ:
803
20.0k
  case WASM_INS_I64_CTZ:
804
20.2k
  case WASM_INS_I64_POPCNT:
805
20.4k
  case WASM_INS_I64_ADD:
806
20.6k
  case WASM_INS_I64_SUB:
807
20.9k
  case WASM_INS_I64_MUL:
808
21.1k
  case WASM_INS_I64_DIV_S:
809
23.0k
  case WASM_INS_I64_DIV_U:
810
24.8k
  case WASM_INS_I64_REM_S:
811
25.4k
  case WASM_INS_I64_REM_U:
812
25.6k
  case WASM_INS_I64_AND:
813
25.8k
  case WASM_INS_I64_OR:
814
26.0k
  case WASM_INS_I64_XOR:
815
26.2k
  case WASM_INS_I64_SHL:
816
27.0k
  case WASM_INS_I64_SHR_S:
817
27.2k
  case WASM_INS_I64_SHR_U:
818
27.4k
  case WASM_INS_I64_ROTL:
819
27.6k
  case WASM_INS_I64_ROTR:
820
27.8k
  case WASM_INS_F32_ABS:
821
28.0k
  case WASM_INS_F32_NEG:
822
28.2k
  case WASM_INS_F32_CEIL:
823
28.4k
  case WASM_INS_F32_FLOOR:
824
28.7k
  case WASM_INS_F32_TRUNC:
825
28.9k
  case WASM_INS_F32_NEAREST:
826
29.1k
  case WASM_INS_F32_SQRT:
827
30.3k
  case WASM_INS_F32_ADD:
828
30.5k
  case WASM_INS_F32_SUB:
829
30.7k
  case WASM_INS_F32_MUL:
830
31.0k
  case WASM_INS_F32_DIV:
831
31.2k
  case WASM_INS_F32_MIN:
832
31.4k
  case WASM_INS_F32_MAX:
833
31.6k
  case WASM_INS_F32_COPYSIGN:
834
32.1k
  case WASM_INS_F64_ABS:
835
32.6k
  case WASM_INS_F64_NEG:
836
33.0k
  case WASM_INS_F64_CEIL:
837
33.4k
  case WASM_INS_F64_FLOOR:
838
33.7k
  case WASM_INS_F64_TRUNC:
839
33.9k
  case WASM_INS_F64_NEAREST:
840
34.1k
  case WASM_INS_F64_SQRT:
841
34.4k
  case WASM_INS_F64_ADD:
842
34.6k
  case WASM_INS_F64_SUB:
843
34.9k
  case WASM_INS_F64_MUL:
844
35.1k
  case WASM_INS_F64_DIV:
845
35.3k
  case WASM_INS_F64_MIN:
846
35.5k
  case WASM_INS_F64_MAX:
847
36.6k
  case WASM_INS_F64_COPYSIGN:
848
37.1k
  case WASM_INS_I32_WARP_I64:
849
37.3k
  case WASP_INS_I32_TRUNC_S_F32:
850
37.7k
  case WASM_INS_I32_TRUNC_U_F32:
851
37.9k
  case WASM_INS_I32_TRUNC_S_F64:
852
39.3k
  case WASM_INS_I32_TRUNC_U_F64:
853
39.8k
  case WASM_INS_I64_EXTEND_S_I32:
854
41.0k
  case WASM_INS_I64_EXTEND_U_I32:
855
41.3k
  case WASM_INS_I64_TRUNC_S_F32:
856
41.9k
  case WASM_INS_I64_TRUNC_U_F32:
857
42.1k
  case WASM_INS_I64_TRUNC_S_F64:
858
42.3k
  case WASM_INS_I64_TRUNC_U_F64:
859
42.5k
  case WASM_INS_F32_CONVERT_S_I32:
860
42.7k
  case WASM_INS_F32_CONVERT_U_I32:
861
42.9k
  case WASM_INS_F32_CONVERT_S_I64:
862
43.2k
  case WASM_INS_F32_CONVERT_U_I64:
863
43.5k
  case WASM_INS_F32_DEMOTE_F64:
864
43.7k
  case WASM_INS_F64_CONVERT_S_I32:
865
43.9k
  case WASM_INS_F64_CONVERT_U_I32:
866
44.1k
  case WASM_INS_F64_CONVERT_S_I64:
867
44.4k
  case WASM_INS_F64_CONVERT_U_I64:
868
44.6k
  case WASM_INS_F64_PROMOTE_F32:
869
44.8k
  case WASM_INS_I32_REINTERPRET_F32:
870
45.0k
  case WASM_INS_I64_REINTERPRET_F64:
871
45.3k
  case WASM_INS_F32_REINTERPRET_I32:
872
45.6k
  case WASM_INS_F64_REINTERPRET_I64:
873
45.6k
    MI->wasm_data.type = WASM_OP_NONE;
874
875
45.6k
    if (MI->flat_insn->detail) {
876
45.6k
      MI->flat_insn->detail->wasm.op_count = 0;
877
45.6k
      MI->flat_insn->detail
878
45.6k
        ->groups[MI->flat_insn->detail->groups_count] =
879
45.6k
        WASM_GRP_NUMBERIC;
880
45.6k
      MI->flat_insn->detail->groups_count++;
881
45.6k
    }
882
883
45.6k
    *size = 1;
884
885
45.6k
    break;
886
887
201
  case WASM_INS_DROP:
888
454
  case WASM_INS_SELECT:
889
454
    MI->wasm_data.type = WASM_OP_NONE;
890
891
454
    if (MI->flat_insn->detail) {
892
454
      MI->flat_insn->detail->wasm.op_count = 0;
893
454
      MI->flat_insn->detail
894
454
        ->groups[MI->flat_insn->detail->groups_count] =
895
454
        WASM_GRP_PARAMETRIC;
896
454
      MI->flat_insn->detail->groups_count++;
897
454
    }
898
899
454
    *size = 1;
900
901
454
    break;
902
903
279
  case WASM_INS_GET_LOCAL:
904
507
  case WASM_INS_SET_LOCAL:
905
712
  case WASM_INS_TEE_LOCAL:
906
1.17k
  case WASM_INS_GET_GLOBAL:
907
1.57k
  case WASM_INS_SET_GLOBAL:
908
1.57k
    if (code_len == 1 ||
909
1.53k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
910
44
      return false;
911
44
    }
912
913
1.52k
    if (MI->flat_insn->detail) {
914
1.52k
      MI->flat_insn->detail->wasm.op_count = 1;
915
1.52k
      MI->flat_insn->detail
916
1.52k
        ->groups[MI->flat_insn->detail->groups_count] =
917
1.52k
        WASM_GRP_VARIABLE;
918
1.52k
      MI->flat_insn->detail->groups_count++;
919
1.52k
    }
920
921
1.52k
    *size = param_size + 1;
922
923
1.52k
    break;
924
925
487
  case WASM_INS_I32_LOAD:
926
696
  case WASM_INS_I64_LOAD:
927
942
  case WASM_INS_F32_LOAD:
928
1.37k
  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.02k
  case WASM_INS_I32_LOAD16_S:
932
2.29k
  case WASM_INS_I32_LOAD16_U:
933
2.74k
  case WASM_INS_I64_LOAD8_S:
934
3.05k
  case WASM_INS_I64_LOAD8_U:
935
3.27k
  case WASM_INS_I64_LOAD16_S:
936
3.57k
  case WASM_INS_I64_LOAD16_U:
937
3.79k
  case WASM_INS_I64_LOAD32_S:
938
4.24k
  case WASM_INS_I64_LOAD32_U:
939
4.51k
  case WASM_INS_I32_STORE:
940
4.72k
  case WASM_INS_I64_STORE:
941
4.98k
  case WASM_INS_F32_STORE:
942
5.20k
  case WASM_INS_F64_STORE:
943
5.41k
  case WASM_INS_I32_STORE8:
944
5.62k
  case WASM_INS_I32_STORE16:
945
5.82k
  case WASM_INS_I64_STORE8:
946
6.11k
  case WASM_INS_I64_STORE16:
947
6.39k
  case WASM_INS_I64_STORE32:
948
6.39k
    if (code_len == 1 ||
949
6.21k
        !read_memoryimmediate(&code[1], code_len - 1, &param_size,
950
6.21k
            MI)) {
951
211
      return false;
952
211
    }
953
954
6.18k
    if (MI->flat_insn->detail) {
955
6.18k
      MI->flat_insn->detail->wasm.op_count = 2;
956
6.18k
      MI->flat_insn->detail
957
6.18k
        ->groups[MI->flat_insn->detail->groups_count] =
958
6.18k
        WASM_GRP_MEMORY;
959
6.18k
      MI->flat_insn->detail->groups_count++;
960
6.18k
    }
961
962
6.18k
    *size = param_size + 1;
963
964
6.18k
    break;
965
966
236
  case WASM_INS_CURRENT_MEMORY:
967
504
  case WASM_INS_GROW_MEMORY:
968
504
    MI->wasm_data.type = WASM_OP_NONE;
969
970
504
    if (MI->flat_insn->detail) {
971
504
      MI->flat_insn->detail->wasm.op_count = 0;
972
504
      MI->flat_insn->detail
973
504
        ->groups[MI->flat_insn->detail->groups_count] =
974
504
        WASM_GRP_MEMORY;
975
504
      MI->flat_insn->detail->groups_count++;
976
504
    }
977
978
504
    *size = 1;
979
980
504
    break;
981
982
3.05k
  case WASM_INS_UNREACHABLE:
983
3.34k
  case WASM_INS_NOP:
984
3.56k
  case WASM_INS_ELSE:
985
3.81k
  case WASM_INS_END:
986
4.02k
  case WASM_INS_RETURN:
987
4.02k
    MI->wasm_data.type = WASM_OP_NONE;
988
989
4.02k
    if (MI->flat_insn->detail) {
990
4.02k
      MI->flat_insn->detail->wasm.op_count = 0;
991
4.02k
      MI->flat_insn->detail
992
4.02k
        ->groups[MI->flat_insn->detail->groups_count] =
993
4.02k
        WASM_GRP_CONTROL;
994
4.02k
      MI->flat_insn->detail->groups_count++;
995
4.02k
    }
996
997
4.02k
    *size = 1;
998
999
4.02k
    break;
1000
1001
220
  case WASM_INS_BLOCK:
1002
430
  case WASM_INS_LOOP:
1003
808
  case WASM_INS_IF:
1004
808
    if (code_len == 1 ||
1005
785
        !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
1006
23
      return false;
1007
23
    }
1008
1009
785
    if (MI->flat_insn->detail) {
1010
785
      MI->flat_insn->detail->wasm.op_count = 1;
1011
785
      MI->flat_insn->detail
1012
785
        ->groups[MI->flat_insn->detail->groups_count] =
1013
785
        WASM_GRP_CONTROL;
1014
785
      MI->flat_insn->detail->groups_count++;
1015
785
    }
1016
1017
785
    *size = param_size + 1;
1018
1019
785
    break;
1020
1021
230
  case WASM_INS_BR:
1022
692
  case WASM_INS_BR_IF:
1023
934
  case WASM_INS_CALL:
1024
1.16k
  case WASM_INS_CALL_INDIRECT:
1025
1.16k
    if (code_len == 1 ||
1026
1.13k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
1027
31
      return false;
1028
31
    }
1029
1030
1.13k
    if (MI->flat_insn->detail) {
1031
1.13k
      MI->flat_insn->detail->wasm.op_count = 1;
1032
1.13k
      MI->flat_insn->detail
1033
1.13k
        ->groups[MI->flat_insn->detail->groups_count] =
1034
1.13k
        WASM_GRP_CONTROL;
1035
1.13k
      MI->flat_insn->detail->groups_count++;
1036
1.13k
    }
1037
1038
1.13k
    *size = param_size + 1;
1039
1040
1.13k
    break;
1041
1042
1.36k
  case WASM_INS_BR_TABLE:
1043
1.36k
    if (code_len == 1 ||
1044
1.35k
        !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
1045
112
      return false;
1046
112
    }
1047
1048
1.24k
    if (MI->flat_insn->detail) {
1049
1.24k
      MI->flat_insn->detail->wasm.op_count = 1;
1050
1.24k
      MI->flat_insn->detail
1051
1.24k
        ->groups[MI->flat_insn->detail->groups_count] =
1052
1.24k
        WASM_GRP_CONTROL;
1053
1.24k
      MI->flat_insn->detail->groups_count++;
1054
1.24k
    }
1055
1056
1.24k
    *size = param_size + 1;
1057
1058
1.24k
    break;
1059
62.8k
  }
1060
1061
62.3k
  return true;
1062
62.8k
}