Coverage Report

Created: 2025-12-05 06:11

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
21.8k
{
277
21.8k
  uint32_t data = 0;
278
21.8k
  int i;
279
280
30.0k
  for (i = 0;; i++) {
281
30.0k
    if (code_len < i + 1) {
282
230
      *leng = -1;
283
230
      return 0;
284
230
    }
285
286
29.8k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
287
166
      *leng = -1;
288
166
      return 0;
289
166
    }
290
291
29.7k
    data = data + (((uint32_t)code[i] & 0x7f) << (i * 7));
292
29.7k
    if (code[i] >> 7 == 0) {
293
21.4k
      break;
294
21.4k
    }
295
29.7k
  }
296
297
21.4k
  *leng = i + 1;
298
299
21.4k
  return data;
300
21.8k
}
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
414
{
309
414
  uint64_t data;
310
414
  int i;
311
312
414
  data = 0;
313
1.68k
  for (i = 0;; i++) {
314
1.68k
    if (code_len < i + 1) {
315
10
      *leng = -1;
316
10
      return 0;
317
10
    }
318
319
1.67k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
2
      *leng = -1;
321
2
      return 0;
322
2
    }
323
324
1.67k
    data = data + (((uint64_t)code[i] & 0x7f) << (i * 7));
325
1.67k
    if (code[i] >> 7 == 0) {
326
402
      break;
327
402
    }
328
1.67k
  }
329
330
402
  *leng = i + 1;
331
332
402
  return data;
333
414
}
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
498
{
340
498
  memcpy(dest, code, 4);
341
498
}
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
464
{
348
464
  memcpy(dest, code, 8);
349
464
}
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
885
{
357
885
  int8_t data;
358
359
885
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
885
  *leng = 1;
365
366
885
  if (code[0] == 0x40) {
367
196
    return -1;
368
196
  }
369
370
689
  data = code[0] & 0x7f;
371
372
689
  return data;
373
885
}
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.71k
{
383
2.71k
  size_t len = 0;
384
2.71k
  uint32_t data;
385
386
2.71k
  data = get_varuint32(code, code_len, &len);
387
2.71k
  if (len == -1) {
388
23
    return false;
389
23
  }
390
391
2.69k
  if (MI->flat_insn->detail) {
392
2.69k
    MI->flat_insn->detail->wasm.op_count = 1;
393
2.69k
    MI->flat_insn->detail->wasm.operands[0].type =
394
2.69k
      WASM_OP_VARUINT32;
395
2.69k
    MI->flat_insn->detail->wasm.operands[0].size = len;
396
2.69k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data;
397
2.69k
  }
398
399
2.69k
  MI->wasm_data.size = len;
400
2.69k
  MI->wasm_data.type = WASM_OP_VARUINT32;
401
2.69k
  MI->wasm_data.uint32 = data;
402
2.69k
  *param_size = len;
403
404
2.69k
  return true;
405
2.71k
}
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
414
{
415
414
  size_t len = 0;
416
414
  uint64_t data;
417
418
414
  data = get_varuint64(code, code_len, &len);
419
414
  if (len == -1) {
420
12
    return false;
421
12
  }
422
423
402
  if (MI->flat_insn->detail) {
424
402
    MI->flat_insn->detail->wasm.op_count = 1;
425
402
    MI->flat_insn->detail->wasm.operands[0].type =
426
402
      WASM_OP_VARUINT64;
427
402
    MI->flat_insn->detail->wasm.operands[0].size = len;
428
402
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
429
402
  }
430
431
402
  MI->wasm_data.size = len;
432
402
  MI->wasm_data.type = WASM_OP_VARUINT64;
433
402
  MI->wasm_data.uint64 = data;
434
402
  *param_size = len;
435
436
402
  return true;
437
414
}
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
5.58k
{
447
5.58k
  size_t tmp, len = 0;
448
5.58k
  uint32_t data[2];
449
450
5.58k
  if (MI->flat_insn->detail) {
451
5.58k
    MI->flat_insn->detail->wasm.op_count = 2;
452
5.58k
  }
453
454
5.58k
  data[0] = get_varuint32(code, code_len, &tmp);
455
5.58k
  if (tmp == -1) {
456
33
    return false;
457
33
  }
458
459
5.55k
  if (MI->flat_insn->detail) {
460
5.55k
    MI->flat_insn->detail->wasm.operands[0].type =
461
5.55k
      WASM_OP_VARUINT32;
462
5.55k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
463
5.55k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
464
5.55k
  }
465
466
5.55k
  len = tmp;
467
5.55k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
468
469
5.55k
  if (MI->flat_insn->detail) {
470
5.55k
    MI->flat_insn->detail->wasm.operands[1].type =
471
5.55k
      WASM_OP_VARUINT32;
472
5.55k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
473
5.55k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
474
5.55k
  }
475
476
5.55k
  len += tmp;
477
5.55k
  MI->wasm_data.size = len;
478
5.55k
  MI->wasm_data.type = WASM_OP_IMM;
479
5.55k
  MI->wasm_data.immediate[0] = data[0];
480
5.55k
  MI->wasm_data.immediate[1] = data[1];
481
5.55k
  *param_size = len;
482
483
5.55k
  return true;
484
5.58k
}
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
253
{
494
253
  if (code_len < 4) {
495
4
    return false;
496
4
  }
497
498
249
  get_uint32(code, &(MI->wasm_data.uint32));
499
500
249
  if (MI->flat_insn->detail) {
501
249
    MI->flat_insn->detail->wasm.op_count = 1;
502
249
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
503
249
    MI->flat_insn->detail->wasm.operands[0].size = 4;
504
249
    get_uint32(code,
505
249
         &(MI->flat_insn->detail->wasm.operands[0].uint32));
506
249
  }
507
508
249
  MI->wasm_data.size = 4;
509
249
  MI->wasm_data.type = WASM_OP_UINT32;
510
249
  *param_size = 4;
511
512
249
  return true;
513
253
}
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
239
{
523
239
  if (code_len < 8) {
524
7
    return false;
525
7
  }
526
527
232
  get_uint64(code, &(MI->wasm_data.uint64));
528
529
232
  if (MI->flat_insn->detail) {
530
232
    MI->flat_insn->detail->wasm.op_count = 1;
531
232
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
532
232
    MI->flat_insn->detail->wasm.operands[0].size = 8;
533
232
    get_uint64(code,
534
232
         &(MI->flat_insn->detail->wasm.operands[0].uint64));
535
232
  }
536
537
232
  MI->wasm_data.size = 8;
538
232
  MI->wasm_data.type = WASM_OP_UINT64;
539
232
  *param_size = 8;
540
541
232
  return true;
542
239
}
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.44k
{
552
1.44k
  uint32_t length, default_target;
553
1.44k
  int tmp_len = 0, i;
554
1.44k
  size_t var_len;
555
556
  // read length
557
1.44k
  length = get_varuint32(code, code_len, &var_len);
558
1.44k
  if (var_len == -1) {
559
8
    return false;
560
8
  }
561
562
1.43k
  tmp_len += var_len;
563
1.43k
  MI->wasm_data.brtable.length = length;
564
1.43k
  if (length >= UINT32_MAX - tmp_len) {
565
    // integer overflow check
566
1
    return false;
567
1
  }
568
1.43k
  if (code_len < tmp_len + length) {
569
    // safety check that we have minimum enough data to read
570
56
    return false;
571
56
  }
572
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
573
1.37k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
574
575
1.37k
  if (MI->flat_insn->detail) {
576
1.37k
    MI->flat_insn->detail->wasm.op_count = 1;
577
1.37k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
578
1.37k
    MI->flat_insn->detail->wasm.operands[0].brtable.length =
579
1.37k
      MI->wasm_data.brtable.length;
580
1.37k
    MI->flat_insn->detail->wasm.operands[0].brtable.address =
581
1.37k
      MI->wasm_data.brtable.address;
582
1.37k
  }
583
584
  // read data
585
6.52k
  for (i = 0; i < length; i++) {
586
5.17k
    if (code_len < tmp_len) {
587
0
      return false;
588
0
    }
589
590
5.17k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
591
5.17k
    if (var_len == -1) {
592
21
      return false;
593
21
    }
594
595
5.15k
    tmp_len += var_len;
596
5.15k
  }
597
598
  // read default target
599
1.35k
  default_target =
600
1.35k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
601
1.35k
  if (var_len == -1) {
602
19
    return false;
603
19
  }
604
605
1.33k
  MI->wasm_data.brtable.default_target = default_target;
606
1.33k
  MI->wasm_data.type = WASM_OP_BRTABLE;
607
1.33k
  *param_size = tmp_len + var_len;
608
609
1.33k
  if (MI->flat_insn->detail) {
610
1.33k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
611
1.33k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target =
612
1.33k
      MI->wasm_data.brtable.default_target;
613
1.33k
  }
614
615
1.33k
  return true;
616
1.35k
}
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
885
{
626
885
  size_t len = 0;
627
628
885
  MI->wasm_data.type = WASM_OP_INT7;
629
885
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
630
885
  if (len == -1) {
631
0
    return false;
632
0
  }
633
634
885
  if (MI->flat_insn->detail) {
635
885
    MI->flat_insn->detail->wasm.op_count = 1;
636
885
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
637
885
    MI->flat_insn->detail->wasm.operands[0].size = 1;
638
885
    MI->flat_insn->detail->wasm.operands[0].int7 =
639
885
      MI->wasm_data.int7;
640
885
  }
641
642
885
  *param_size = len;
643
644
885
  return true;
645
885
}
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
60.9k
{
651
60.9k
  unsigned char opcode;
652
60.9k
  uint16_t param_size;
653
654
60.9k
  if (code_len == 0)
655
0
    return false;
656
657
60.9k
  opcode = code[0];
658
60.9k
  if (opcodes[opcode] == -1) {
659
    // invalid opcode
660
170
    return false;
661
170
  }
662
663
  // valid opcode
664
60.8k
  MI->address = address;
665
60.8k
  MI->OpcodePub = MI->Opcode = opcode;
666
667
60.8k
  if (MI->flat_insn->detail) {
668
60.8k
    memset(MI->flat_insn->detail, 0,
669
60.8k
           offsetof(cs_detail, wasm) + sizeof(cs_wasm));
670
60.8k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
671
60.8k
  }
672
673
  // setup groups
674
60.8k
  switch (opcode) {
675
0
  default:
676
0
    return false;
677
678
257
  case WASM_INS_I32_CONST:
679
257
    if (code_len == 1 ||
680
252
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
681
11
      return false;
682
11
    }
683
684
246
    if (MI->flat_insn->detail) {
685
246
      MI->flat_insn->detail->wasm.op_count = 1;
686
246
      MI->flat_insn->detail
687
246
        ->groups[MI->flat_insn->detail->groups_count] =
688
246
        WASM_GRP_NUMBERIC;
689
246
      MI->flat_insn->detail->groups_count++;
690
246
    }
691
692
246
    *size = param_size + 1;
693
694
246
    break;
695
696
416
  case WASM_INS_I64_CONST:
697
416
    if (code_len == 1 ||
698
414
        !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
699
14
      return false;
700
14
    }
701
702
402
    if (MI->flat_insn->detail) {
703
402
      MI->flat_insn->detail->wasm.op_count = 1;
704
402
      MI->flat_insn->detail
705
402
        ->groups[MI->flat_insn->detail->groups_count] =
706
402
        WASM_GRP_NUMBERIC;
707
402
      MI->flat_insn->detail->groups_count++;
708
402
    }
709
710
402
    *size = param_size + 1;
711
712
402
    break;
713
714
255
  case WASM_INS_F32_CONST:
715
255
    if (code_len == 1 ||
716
253
        !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
717
6
      return false;
718
6
    }
719
720
249
    if (MI->flat_insn->detail) {
721
249
      MI->flat_insn->detail->wasm.op_count = 1;
722
249
      MI->flat_insn->detail
723
249
        ->groups[MI->flat_insn->detail->groups_count] =
724
249
        WASM_GRP_NUMBERIC;
725
249
      MI->flat_insn->detail->groups_count++;
726
249
    }
727
728
249
    *size = param_size + 1;
729
730
249
    break;
731
732
243
  case WASM_INS_F64_CONST:
733
243
    if (code_len == 1 ||
734
239
        !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
735
11
      return false;
736
11
    }
737
738
232
    if (MI->flat_insn->detail) {
739
232
      MI->flat_insn->detail->wasm.op_count = 1;
740
232
      MI->flat_insn->detail
741
232
        ->groups[MI->flat_insn->detail->groups_count] =
742
232
        WASM_GRP_NUMBERIC;
743
232
      MI->flat_insn->detail->groups_count++;
744
232
    }
745
746
232
    *size = param_size + 1;
747
748
232
    break;
749
750
228
  case WASM_INS_I32_EQZ:
751
455
  case WASM_INS_I32_EQ:
752
675
  case WASM_INS_I32_NE:
753
886
  case WASM_INS_I32_LT_S:
754
1.09k
  case WASM_INS_I32_LT_U:
755
1.31k
  case WASM_INS_I32_GT_S:
756
1.68k
  case WASM_INS_I32_GT_U:
757
1.90k
  case WASM_INS_I32_LE_S:
758
2.11k
  case WASM_INS_I32_LE_U:
759
2.32k
  case WASM_INS_I32_GE_S:
760
2.52k
  case WASM_INS_I32_GE_U:
761
2.74k
  case WASM_INS_I64_EQZ:
762
2.95k
  case WASM_INS_I64_EQ:
763
3.67k
  case WASM_INS_I64_NE:
764
4.03k
  case WASM_INS_I64_LT_S:
765
4.25k
  case WASM_INS_I64_LT_U:
766
4.49k
  case WASN_INS_I64_GT_S:
767
4.70k
  case WASM_INS_I64_GT_U:
768
5.11k
  case WASM_INS_I64_LE_S:
769
5.34k
  case WASM_INS_I64_LE_U:
770
5.55k
  case WASM_INS_I64_GE_S:
771
5.84k
  case WASM_INS_I64_GE_U:
772
6.15k
  case WASM_INS_F32_EQ:
773
6.58k
  case WASM_INS_F32_NE:
774
6.83k
  case WASM_INS_F32_LT:
775
7.02k
  case WASM_INS_F32_GT:
776
7.26k
  case WASM_INS_F32_LE:
777
7.54k
  case WASM_INS_F32_GE:
778
7.98k
  case WASM_INS_F64_EQ:
779
8.28k
  case WASM_INS_F64_NE:
780
8.55k
  case WASM_INS_F64_LT:
781
9.02k
  case WASM_INS_F64_GT:
782
9.59k
  case WASM_INS_F64_LE:
783
9.93k
  case WASM_INS_F64_GE:
784
10.1k
  case WASM_INS_I32_CLZ:
785
10.3k
  case WASM_INS_I32_CTZ:
786
10.6k
  case WASM_INS_I32_POPCNT:
787
10.8k
  case WASM_INS_I32_ADD:
788
11.0k
  case WASM_INS_I32_SUB:
789
11.3k
  case WASM_INS_I32_MUL:
790
11.6k
  case WASM_INS_I32_DIV_S:
791
12.1k
  case WASM_INS_I32_DIV_U:
792
12.3k
  case WASM_INS_I32_REM_S:
793
12.6k
  case WASM_INS_I32_REM_U:
794
12.8k
  case WASM_INS_I32_AND:
795
13.1k
  case WASM_INS_I32_OR:
796
13.4k
  case WASM_INS_I32_XOR:
797
13.8k
  case WASM_INS_I32_SHL:
798
14.0k
  case WASM_INS_I32_SHR_S:
799
14.3k
  case WASM_INS_I32_SHR_U:
800
14.6k
  case WASM_INS_I32_ROTL:
801
15.0k
  case WASM_INS_I32_ROTR:
802
15.3k
  case WASM_INS_I64_CLZ:
803
15.5k
  case WASM_INS_I64_CTZ:
804
15.8k
  case WASM_INS_I64_POPCNT:
805
16.0k
  case WASM_INS_I64_ADD:
806
16.2k
  case WASM_INS_I64_SUB:
807
16.5k
  case WASM_INS_I64_MUL:
808
16.7k
  case WASM_INS_I64_DIV_S:
809
18.6k
  case WASM_INS_I64_DIV_U:
810
22.6k
  case WASM_INS_I64_REM_S:
811
22.8k
  case WASM_INS_I64_REM_U:
812
23.0k
  case WASM_INS_I64_AND:
813
23.2k
  case WASM_INS_I64_OR:
814
23.5k
  case WASM_INS_I64_XOR:
815
23.8k
  case WASM_INS_I64_SHL:
816
24.1k
  case WASM_INS_I64_SHR_S:
817
24.3k
  case WASM_INS_I64_SHR_U:
818
24.5k
  case WASM_INS_I64_ROTL:
819
24.8k
  case WASM_INS_I64_ROTR:
820
25.4k
  case WASM_INS_F32_ABS:
821
25.6k
  case WASM_INS_F32_NEG:
822
25.8k
  case WASM_INS_F32_CEIL:
823
26.0k
  case WASM_INS_F32_FLOOR:
824
26.3k
  case WASM_INS_F32_TRUNC:
825
26.5k
  case WASM_INS_F32_NEAREST:
826
26.8k
  case WASM_INS_F32_SQRT:
827
27.1k
  case WASM_INS_F32_ADD:
828
27.3k
  case WASM_INS_F32_SUB:
829
27.5k
  case WASM_INS_F32_MUL:
830
27.7k
  case WASM_INS_F32_DIV:
831
27.9k
  case WASM_INS_F32_MIN:
832
28.2k
  case WASM_INS_F32_MAX:
833
28.4k
  case WASM_INS_F32_COPYSIGN:
834
28.6k
  case WASM_INS_F64_ABS:
835
28.9k
  case WASM_INS_F64_NEG:
836
29.1k
  case WASM_INS_F64_CEIL:
837
29.3k
  case WASM_INS_F64_FLOOR:
838
29.7k
  case WASM_INS_F64_TRUNC:
839
29.9k
  case WASM_INS_F64_NEAREST:
840
30.1k
  case WASM_INS_F64_SQRT:
841
30.3k
  case WASM_INS_F64_ADD:
842
30.5k
  case WASM_INS_F64_SUB:
843
30.8k
  case WASM_INS_F64_MUL:
844
31.0k
  case WASM_INS_F64_DIV:
845
31.2k
  case WASM_INS_F64_MIN:
846
31.5k
  case WASM_INS_F64_MAX:
847
31.7k
  case WASM_INS_F64_COPYSIGN:
848
31.9k
  case WASM_INS_I32_WARP_I64:
849
32.2k
  case WASP_INS_I32_TRUNC_S_F32:
850
33.2k
  case WASM_INS_I32_TRUNC_U_F32:
851
33.5k
  case WASM_INS_I32_TRUNC_S_F64:
852
36.9k
  case WASM_INS_I32_TRUNC_U_F64:
853
37.4k
  case WASM_INS_I64_EXTEND_S_I32:
854
37.7k
  case WASM_INS_I64_EXTEND_U_I32:
855
37.9k
  case WASM_INS_I64_TRUNC_S_F32:
856
38.5k
  case WASM_INS_I64_TRUNC_U_F32:
857
38.7k
  case WASM_INS_I64_TRUNC_S_F64:
858
38.9k
  case WASM_INS_I64_TRUNC_U_F64:
859
39.1k
  case WASM_INS_F32_CONVERT_S_I32:
860
39.4k
  case WASM_INS_F32_CONVERT_U_I32:
861
39.6k
  case WASM_INS_F32_CONVERT_S_I64:
862
39.8k
  case WASM_INS_F32_CONVERT_U_I64:
863
40.0k
  case WASM_INS_F32_DEMOTE_F64:
864
40.2k
  case WASM_INS_F64_CONVERT_S_I32:
865
40.4k
  case WASM_INS_F64_CONVERT_U_I32:
866
40.7k
  case WASM_INS_F64_CONVERT_S_I64:
867
41.0k
  case WASM_INS_F64_CONVERT_U_I64:
868
41.2k
  case WASM_INS_F64_PROMOTE_F32:
869
41.4k
  case WASM_INS_I32_REINTERPRET_F32:
870
41.6k
  case WASM_INS_I64_REINTERPRET_F64:
871
41.9k
  case WASM_INS_F32_REINTERPRET_I32:
872
42.2k
  case WASM_INS_F64_REINTERPRET_I64:
873
42.2k
    MI->wasm_data.type = WASM_OP_NONE;
874
875
42.2k
    if (MI->flat_insn->detail) {
876
42.2k
      MI->flat_insn->detail->wasm.op_count = 0;
877
42.2k
      MI->flat_insn->detail
878
42.2k
        ->groups[MI->flat_insn->detail->groups_count] =
879
42.2k
        WASM_GRP_NUMBERIC;
880
42.2k
      MI->flat_insn->detail->groups_count++;
881
42.2k
    }
882
883
42.2k
    *size = 1;
884
885
42.2k
    break;
886
887
216
  case WASM_INS_DROP:
888
427
  case WASM_INS_SELECT:
889
427
    MI->wasm_data.type = WASM_OP_NONE;
890
891
427
    if (MI->flat_insn->detail) {
892
427
      MI->flat_insn->detail->wasm.op_count = 0;
893
427
      MI->flat_insn->detail
894
427
        ->groups[MI->flat_insn->detail->groups_count] =
895
427
        WASM_GRP_PARAMETRIC;
896
427
      MI->flat_insn->detail->groups_count++;
897
427
    }
898
899
427
    *size = 1;
900
901
427
    break;
902
903
319
  case WASM_INS_GET_LOCAL:
904
580
  case WASM_INS_SET_LOCAL:
905
815
  case WASM_INS_TEE_LOCAL:
906
1.04k
  case WASM_INS_GET_GLOBAL:
907
1.26k
  case WASM_INS_SET_GLOBAL:
908
1.26k
    if (code_len == 1 ||
909
1.23k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
910
42
      return false;
911
42
    }
912
913
1.22k
    if (MI->flat_insn->detail) {
914
1.22k
      MI->flat_insn->detail->wasm.op_count = 1;
915
1.22k
      MI->flat_insn->detail
916
1.22k
        ->groups[MI->flat_insn->detail->groups_count] =
917
1.22k
        WASM_GRP_VARIABLE;
918
1.22k
      MI->flat_insn->detail->groups_count++;
919
1.22k
    }
920
921
1.22k
    *size = param_size + 1;
922
923
1.22k
    break;
924
925
220
  case WASM_INS_I32_LOAD:
926
437
  case WASM_INS_I64_LOAD:
927
651
  case WASM_INS_F32_LOAD:
928
871
  case WASM_INS_F64_LOAD:
929
1.15k
  case WASM_INS_I32_LOAD8_S:
930
1.38k
  case WASM_INS_I32_LOAD8_U:
931
1.60k
  case WASM_INS_I32_LOAD16_S:
932
1.83k
  case WASM_INS_I32_LOAD16_U:
933
2.28k
  case WASM_INS_I64_LOAD8_S:
934
2.55k
  case WASM_INS_I64_LOAD8_U:
935
2.79k
  case WASM_INS_I64_LOAD16_S:
936
3.05k
  case WASM_INS_I64_LOAD16_U:
937
3.29k
  case WASM_INS_I64_LOAD32_S:
938
3.67k
  case WASM_INS_I64_LOAD32_U:
939
3.94k
  case WASM_INS_I32_STORE:
940
4.15k
  case WASM_INS_I64_STORE:
941
4.42k
  case WASM_INS_F32_STORE:
942
4.63k
  case WASM_INS_F64_STORE:
943
4.85k
  case WASM_INS_I32_STORE8:
944
5.06k
  case WASM_INS_I32_STORE16:
945
5.28k
  case WASM_INS_I64_STORE8:
946
5.54k
  case WASM_INS_I64_STORE16:
947
5.74k
  case WASM_INS_I64_STORE32:
948
5.74k
    if (code_len == 1 ||
949
5.58k
        !read_memoryimmediate(&code[1], code_len - 1, &param_size,
950
5.58k
            MI)) {
951
186
      return false;
952
186
    }
953
954
5.55k
    if (MI->flat_insn->detail) {
955
5.55k
      MI->flat_insn->detail->wasm.op_count = 2;
956
5.55k
      MI->flat_insn->detail
957
5.55k
        ->groups[MI->flat_insn->detail->groups_count] =
958
5.55k
        WASM_GRP_MEMORY;
959
5.55k
      MI->flat_insn->detail->groups_count++;
960
5.55k
    }
961
962
5.55k
    *size = param_size + 1;
963
964
5.55k
    break;
965
966
279
  case WASM_INS_CURRENT_MEMORY:
967
578
  case WASM_INS_GROW_MEMORY:
968
578
    MI->wasm_data.type = WASM_OP_NONE;
969
970
578
    if (MI->flat_insn->detail) {
971
578
      MI->flat_insn->detail->wasm.op_count = 0;
972
578
      MI->flat_insn->detail
973
578
        ->groups[MI->flat_insn->detail->groups_count] =
974
578
        WASM_GRP_MEMORY;
975
578
      MI->flat_insn->detail->groups_count++;
976
578
    }
977
978
578
    *size = 1;
979
980
578
    break;
981
982
4.66k
  case WASM_INS_UNREACHABLE:
983
4.94k
  case WASM_INS_NOP:
984
5.31k
  case WASM_INS_ELSE:
985
5.52k
  case WASM_INS_END:
986
5.77k
  case WASM_INS_RETURN:
987
5.77k
    MI->wasm_data.type = WASM_OP_NONE;
988
989
5.77k
    if (MI->flat_insn->detail) {
990
5.77k
      MI->flat_insn->detail->wasm.op_count = 0;
991
5.77k
      MI->flat_insn->detail
992
5.77k
        ->groups[MI->flat_insn->detail->groups_count] =
993
5.77k
        WASM_GRP_CONTROL;
994
5.77k
      MI->flat_insn->detail->groups_count++;
995
5.77k
    }
996
997
5.77k
    *size = 1;
998
999
5.77k
    break;
1000
1001
235
  case WASM_INS_BLOCK:
1002
452
  case WASM_INS_LOOP:
1003
908
  case WASM_INS_IF:
1004
908
    if (code_len == 1 ||
1005
885
        !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
1006
23
      return false;
1007
23
    }
1008
1009
885
    if (MI->flat_insn->detail) {
1010
885
      MI->flat_insn->detail->wasm.op_count = 1;
1011
885
      MI->flat_insn->detail
1012
885
        ->groups[MI->flat_insn->detail->groups_count] =
1013
885
        WASM_GRP_CONTROL;
1014
885
      MI->flat_insn->detail->groups_count++;
1015
885
    }
1016
1017
885
    *size = param_size + 1;
1018
1019
885
    break;
1020
1021
215
  case WASM_INS_BR:
1022
644
  case WASM_INS_BR_IF:
1023
928
  case WASM_INS_CALL:
1024
1.25k
  case WASM_INS_CALL_INDIRECT:
1025
1.25k
    if (code_len == 1 ||
1026
1.22k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
1027
29
      return false;
1028
29
    }
1029
1030
1.22k
    if (MI->flat_insn->detail) {
1031
1.22k
      MI->flat_insn->detail->wasm.op_count = 1;
1032
1.22k
      MI->flat_insn->detail
1033
1.22k
        ->groups[MI->flat_insn->detail->groups_count] =
1034
1.22k
        WASM_GRP_CONTROL;
1035
1.22k
      MI->flat_insn->detail->groups_count++;
1036
1.22k
    }
1037
1038
1.22k
    *size = param_size + 1;
1039
1040
1.22k
    break;
1041
1042
1.44k
  case WASM_INS_BR_TABLE:
1043
1.44k
    if (code_len == 1 ||
1044
1.44k
        !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
1045
107
      return false;
1046
107
    }
1047
1048
1.33k
    if (MI->flat_insn->detail) {
1049
1.33k
      MI->flat_insn->detail->wasm.op_count = 1;
1050
1.33k
      MI->flat_insn->detail
1051
1.33k
        ->groups[MI->flat_insn->detail->groups_count] =
1052
1.33k
        WASM_GRP_CONTROL;
1053
1.33k
      MI->flat_insn->detail->groups_count++;
1054
1.33k
    }
1055
1056
1.33k
    *size = param_size + 1;
1057
1058
1.33k
    break;
1059
60.8k
  }
1060
1061
60.3k
  return true;
1062
60.8k
}