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
29.5k
{
277
29.5k
  uint32_t data = 0;
278
29.5k
  int i;
279
280
43.6k
  for (i = 0;; i++) {
281
43.6k
    if (code_len < i + 1) {
282
280
      *leng = -1;
283
280
      return 0;
284
280
    }
285
286
43.3k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
287
556
      *leng = -1;
288
556
      return 0;
289
556
    }
290
291
42.7k
    data = data + (((uint32_t)code[i] & 0x7f) << (i * 7));
292
42.7k
    if (code[i] >> 7 == 0) {
293
28.6k
      break;
294
28.6k
    }
295
42.7k
  }
296
297
28.6k
  *leng = i + 1;
298
299
28.6k
  return data;
300
29.5k
}
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
309
{
309
309
  uint64_t data;
310
309
  int i;
311
312
309
  data = 0;
313
1.21k
  for (i = 0;; i++) {
314
1.21k
    if (code_len < i + 1) {
315
6
      *leng = -1;
316
6
      return 0;
317
6
    }
318
319
1.20k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
2
      *leng = -1;
321
2
      return 0;
322
2
    }
323
324
1.20k
    data = data + (((uint64_t)code[i] & 0x7f) << (i * 7));
325
1.20k
    if (code[i] >> 7 == 0) {
326
301
      break;
327
301
    }
328
1.20k
  }
329
330
301
  *leng = i + 1;
331
332
301
  return data;
333
309
}
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
692
{
340
692
  memcpy(dest, code, 4);
341
692
}
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
452
{
348
452
  memcpy(dest, code, 8);
349
452
}
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
974
{
357
974
  int8_t data;
358
359
974
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
974
  *leng = 1;
365
366
974
  if (code[0] == 0x40) {
367
196
    return -1;
368
196
  }
369
370
778
  data = code[0] & 0x7f;
371
372
778
  return data;
373
974
}
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
3.23k
{
383
3.23k
  size_t len = 0;
384
3.23k
  uint32_t data;
385
386
3.23k
  data = get_varuint32(code, code_len, &len);
387
3.23k
  if (len == -1) {
388
21
    return false;
389
21
  }
390
391
3.21k
  if (MI->flat_insn->detail) {
392
3.21k
    MI->flat_insn->detail->wasm.op_count = 1;
393
3.21k
    MI->flat_insn->detail->wasm.operands[0].type =
394
3.21k
      WASM_OP_VARUINT32;
395
3.21k
    MI->flat_insn->detail->wasm.operands[0].size = len;
396
3.21k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data;
397
3.21k
  }
398
399
3.21k
  MI->wasm_data.size = len;
400
3.21k
  MI->wasm_data.type = WASM_OP_VARUINT32;
401
3.21k
  MI->wasm_data.uint32 = data;
402
3.21k
  *param_size = len;
403
404
3.21k
  return true;
405
3.23k
}
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
309
{
415
309
  size_t len = 0;
416
309
  uint64_t data;
417
418
309
  data = get_varuint64(code, code_len, &len);
419
309
  if (len == -1) {
420
8
    return false;
421
8
  }
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
309
}
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
352
{
494
352
  if (code_len < 4) {
495
6
    return false;
496
6
  }
497
498
346
  get_uint32(code, &(MI->wasm_data.uint32));
499
500
346
  if (MI->flat_insn->detail) {
501
346
    MI->flat_insn->detail->wasm.op_count = 1;
502
346
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
503
346
    MI->flat_insn->detail->wasm.operands[0].size = 4;
504
346
    get_uint32(code,
505
346
         &(MI->flat_insn->detail->wasm.operands[0].uint32));
506
346
  }
507
508
346
  MI->wasm_data.size = 4;
509
346
  MI->wasm_data.type = WASM_OP_UINT32;
510
346
  *param_size = 4;
511
512
346
  return true;
513
352
}
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
236
{
523
236
  if (code_len < 8) {
524
10
    return false;
525
10
  }
526
527
226
  get_uint64(code, &(MI->wasm_data.uint64));
528
529
226
  if (MI->flat_insn->detail) {
530
226
    MI->flat_insn->detail->wasm.op_count = 1;
531
226
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
532
226
    MI->flat_insn->detail->wasm.operands[0].size = 8;
533
226
    get_uint64(code,
534
226
         &(MI->flat_insn->detail->wasm.operands[0].uint64));
535
226
  }
536
537
226
  MI->wasm_data.size = 8;
538
226
  MI->wasm_data.type = WASM_OP_UINT64;
539
226
  *param_size = 8;
540
541
226
  return true;
542
236
}
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.69k
{
552
1.69k
  uint32_t length, default_target;
553
1.69k
  int tmp_len = 0, i;
554
1.69k
  size_t var_len;
555
556
  // read length
557
1.69k
  length = get_varuint32(code, code_len, &var_len);
558
1.69k
  if (var_len == -1) {
559
12
    return false;
560
12
  }
561
562
1.68k
  tmp_len += var_len;
563
1.68k
  MI->wasm_data.brtable.length = length;
564
1.68k
  if (length >= UINT32_MAX - tmp_len) {
565
    // integer overflow check
566
1
    return false;
567
1
  }
568
1.68k
  if (code_len < tmp_len + length) {
569
    // safety check that we have minimum enough data to read
570
62
    return false;
571
62
  }
572
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
573
1.62k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
574
575
1.62k
  if (MI->flat_insn->detail) {
576
1.62k
    MI->flat_insn->detail->wasm.op_count = 1;
577
1.62k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
578
1.62k
    MI->flat_insn->detail->wasm.operands[0].brtable.length =
579
1.62k
      MI->wasm_data.brtable.length;
580
1.62k
    MI->flat_insn->detail->wasm.operands[0].brtable.address =
581
1.62k
      MI->wasm_data.brtable.address;
582
1.62k
  }
583
584
  // read data
585
10.6k
  for (i = 0; i < length; i++) {
586
9.05k
    if (code_len < tmp_len) {
587
0
      return false;
588
0
    }
589
590
9.05k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
591
9.05k
    if (var_len == -1) {
592
29
      return false;
593
29
    }
594
595
9.02k
    tmp_len += var_len;
596
9.02k
  }
597
598
  // read default target
599
1.59k
  default_target =
600
1.59k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
601
1.59k
  if (var_len == -1) {
602
24
    return false;
603
24
  }
604
605
1.56k
  MI->wasm_data.brtable.default_target = default_target;
606
1.56k
  MI->wasm_data.type = WASM_OP_BRTABLE;
607
1.56k
  *param_size = tmp_len + var_len;
608
609
1.56k
  if (MI->flat_insn->detail) {
610
1.56k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
611
1.56k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target =
612
1.56k
      MI->wasm_data.brtable.default_target;
613
1.56k
  }
614
615
1.56k
  return true;
616
1.59k
}
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
974
{
626
974
  size_t len = 0;
627
628
974
  MI->wasm_data.type = WASM_OP_INT7;
629
974
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
630
974
  if (len == -1) {
631
0
    return false;
632
0
  }
633
634
974
  if (MI->flat_insn->detail) {
635
974
    MI->flat_insn->detail->wasm.op_count = 1;
636
974
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
637
974
    MI->flat_insn->detail->wasm.operands[0].size = 1;
638
974
    MI->flat_insn->detail->wasm.operands[0].int7 =
639
974
      MI->wasm_data.int7;
640
974
  }
641
642
974
  *param_size = len;
643
644
974
  return true;
645
974
}
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
68.5k
{
651
68.5k
  unsigned char opcode;
652
68.5k
  uint16_t param_size;
653
654
68.5k
  if (code_len == 0)
655
0
    return false;
656
657
68.5k
  opcode = code[0];
658
68.5k
  if (opcodes[opcode] == -1) {
659
    // invalid opcode
660
107
    return false;
661
107
  }
662
663
  // valid opcode
664
68.4k
  MI->address = address;
665
68.4k
  MI->OpcodePub = MI->Opcode = opcode;
666
667
68.4k
  if (MI->flat_insn->detail) {
668
68.4k
    memset(MI->flat_insn->detail, 0,
669
68.4k
           offsetof(cs_detail, wasm) + sizeof(cs_wasm));
670
68.4k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
671
68.4k
  }
672
673
  // setup groups
674
68.4k
  switch (opcode) {
675
0
  default:
676
0
    return false;
677
678
277
  case WASM_INS_I32_CONST:
679
277
    if (code_len == 1 ||
680
274
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
681
9
      return false;
682
9
    }
683
684
268
    if (MI->flat_insn->detail) {
685
268
      MI->flat_insn->detail->wasm.op_count = 1;
686
268
      MI->flat_insn->detail
687
268
        ->groups[MI->flat_insn->detail->groups_count] =
688
268
        WASM_GRP_NUMBERIC;
689
268
      MI->flat_insn->detail->groups_count++;
690
268
    }
691
692
268
    *size = param_size + 1;
693
694
268
    break;
695
696
312
  case WASM_INS_I64_CONST:
697
312
    if (code_len == 1 ||
698
309
        !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
699
11
      return false;
700
11
    }
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
353
  case WASM_INS_F32_CONST:
715
353
    if (code_len == 1 ||
716
352
        !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
717
7
      return false;
718
7
    }
719
720
346
    if (MI->flat_insn->detail) {
721
346
      MI->flat_insn->detail->wasm.op_count = 1;
722
346
      MI->flat_insn->detail
723
346
        ->groups[MI->flat_insn->detail->groups_count] =
724
346
        WASM_GRP_NUMBERIC;
725
346
      MI->flat_insn->detail->groups_count++;
726
346
    }
727
728
346
    *size = param_size + 1;
729
730
346
    break;
731
732
239
  case WASM_INS_F64_CONST:
733
239
    if (code_len == 1 ||
734
236
        !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
735
13
      return false;
736
13
    }
737
738
226
    if (MI->flat_insn->detail) {
739
226
      MI->flat_insn->detail->wasm.op_count = 1;
740
226
      MI->flat_insn->detail
741
226
        ->groups[MI->flat_insn->detail->groups_count] =
742
226
        WASM_GRP_NUMBERIC;
743
226
      MI->flat_insn->detail->groups_count++;
744
226
    }
745
746
226
    *size = param_size + 1;
747
748
226
    break;
749
750
234
  case WASM_INS_I32_EQZ:
751
453
  case WASM_INS_I32_EQ:
752
737
  case WASM_INS_I32_NE:
753
940
  case WASM_INS_I32_LT_S:
754
1.15k
  case WASM_INS_I32_LT_U:
755
1.38k
  case WASM_INS_I32_GT_S:
756
1.59k
  case WASM_INS_I32_GT_U:
757
2.01k
  case WASM_INS_I32_LE_S:
758
2.21k
  case WASM_INS_I32_LE_U:
759
2.54k
  case WASM_INS_I32_GE_S:
760
2.79k
  case WASM_INS_I32_GE_U:
761
3.13k
  case WASM_INS_I64_EQZ:
762
3.34k
  case WASM_INS_I64_EQ:
763
3.62k
  case WASM_INS_I64_NE:
764
3.83k
  case WASM_INS_I64_LT_S:
765
4.03k
  case WASM_INS_I64_LT_U:
766
4.24k
  case WASN_INS_I64_GT_S:
767
4.45k
  case WASM_INS_I64_GT_U:
768
5.15k
  case WASM_INS_I64_LE_S:
769
5.99k
  case WASM_INS_I64_LE_U:
770
6.70k
  case WASM_INS_I64_GE_S:
771
7.00k
  case WASM_INS_I64_GE_U:
772
7.23k
  case WASM_INS_F32_EQ:
773
7.58k
  case WASM_INS_F32_NE:
774
7.91k
  case WASM_INS_F32_LT:
775
8.17k
  case WASM_INS_F32_GT:
776
8.56k
  case WASM_INS_F32_LE:
777
8.80k
  case WASM_INS_F32_GE:
778
9.02k
  case WASM_INS_F64_EQ:
779
9.32k
  case WASM_INS_F64_NE:
780
9.78k
  case WASM_INS_F64_LT:
781
10.1k
  case WASM_INS_F64_GT:
782
14.2k
  case WASM_INS_F64_LE:
783
14.4k
  case WASM_INS_F64_GE:
784
14.6k
  case WASM_INS_I32_CLZ:
785
15.2k
  case WASM_INS_I32_CTZ:
786
15.5k
  case WASM_INS_I32_POPCNT:
787
15.7k
  case WASM_INS_I32_ADD:
788
16.1k
  case WASM_INS_I32_SUB:
789
16.4k
  case WASM_INS_I32_MUL:
790
16.6k
  case WASM_INS_I32_DIV_S:
791
16.9k
  case WASM_INS_I32_DIV_U:
792
17.3k
  case WASM_INS_I32_REM_S:
793
17.6k
  case WASM_INS_I32_REM_U:
794
17.9k
  case WASM_INS_I32_AND:
795
18.2k
  case WASM_INS_I32_OR:
796
18.4k
  case WASM_INS_I32_XOR:
797
18.8k
  case WASM_INS_I32_SHL:
798
19.2k
  case WASM_INS_I32_SHR_S:
799
19.6k
  case WASM_INS_I32_SHR_U:
800
19.8k
  case WASM_INS_I32_ROTL:
801
20.5k
  case WASM_INS_I32_ROTR:
802
20.7k
  case WASM_INS_I64_CLZ:
803
20.9k
  case WASM_INS_I64_CTZ:
804
21.1k
  case WASM_INS_I64_POPCNT:
805
21.4k
  case WASM_INS_I64_ADD:
806
21.7k
  case WASM_INS_I64_SUB:
807
22.0k
  case WASM_INS_I64_MUL:
808
22.2k
  case WASM_INS_I64_DIV_S:
809
24.4k
  case WASM_INS_I64_DIV_U:
810
26.3k
  case WASM_INS_I64_REM_S:
811
26.8k
  case WASM_INS_I64_REM_U:
812
27.1k
  case WASM_INS_I64_AND:
813
27.3k
  case WASM_INS_I64_OR:
814
27.5k
  case WASM_INS_I64_XOR:
815
27.8k
  case WASM_INS_I64_SHL:
816
28.5k
  case WASM_INS_I64_SHR_S:
817
28.8k
  case WASM_INS_I64_SHR_U:
818
29.0k
  case WASM_INS_I64_ROTL:
819
29.2k
  case WASM_INS_I64_ROTR:
820
29.4k
  case WASM_INS_F32_ABS:
821
29.7k
  case WASM_INS_F32_NEG:
822
30.1k
  case WASM_INS_F32_CEIL:
823
30.3k
  case WASM_INS_F32_FLOOR:
824
30.5k
  case WASM_INS_F32_TRUNC:
825
30.7k
  case WASM_INS_F32_NEAREST:
826
31.1k
  case WASM_INS_F32_SQRT:
827
32.2k
  case WASM_INS_F32_ADD:
828
32.5k
  case WASM_INS_F32_SUB:
829
32.7k
  case WASM_INS_F32_MUL:
830
32.9k
  case WASM_INS_F32_DIV:
831
33.1k
  case WASM_INS_F32_MIN:
832
33.3k
  case WASM_INS_F32_MAX:
833
33.5k
  case WASM_INS_F32_COPYSIGN:
834
34.0k
  case WASM_INS_F64_ABS:
835
34.5k
  case WASM_INS_F64_NEG:
836
35.0k
  case WASM_INS_F64_CEIL:
837
35.3k
  case WASM_INS_F64_FLOOR:
838
35.7k
  case WASM_INS_F64_TRUNC:
839
35.9k
  case WASM_INS_F64_NEAREST:
840
36.1k
  case WASM_INS_F64_SQRT:
841
36.3k
  case WASM_INS_F64_ADD:
842
36.7k
  case WASM_INS_F64_SUB:
843
37.0k
  case WASM_INS_F64_MUL:
844
37.3k
  case WASM_INS_F64_DIV:
845
37.6k
  case WASM_INS_F64_MIN:
846
37.9k
  case WASM_INS_F64_MAX:
847
39.0k
  case WASM_INS_F64_COPYSIGN:
848
39.4k
  case WASM_INS_I32_WARP_I64:
849
39.6k
  case WASP_INS_I32_TRUNC_S_F32:
850
40.1k
  case WASM_INS_I32_TRUNC_U_F32:
851
40.3k
  case WASM_INS_I32_TRUNC_S_F64:
852
41.7k
  case WASM_INS_I32_TRUNC_U_F64:
853
42.2k
  case WASM_INS_I64_EXTEND_S_I32:
854
43.4k
  case WASM_INS_I64_EXTEND_U_I32:
855
43.8k
  case WASM_INS_I64_TRUNC_S_F32:
856
44.4k
  case WASM_INS_I64_TRUNC_U_F32:
857
44.8k
  case WASM_INS_I64_TRUNC_S_F64:
858
45.2k
  case WASM_INS_I64_TRUNC_U_F64:
859
45.4k
  case WASM_INS_F32_CONVERT_S_I32:
860
45.7k
  case WASM_INS_F32_CONVERT_U_I32:
861
45.9k
  case WASM_INS_F32_CONVERT_S_I64:
862
46.1k
  case WASM_INS_F32_CONVERT_U_I64:
863
46.4k
  case WASM_INS_F32_DEMOTE_F64:
864
46.6k
  case WASM_INS_F64_CONVERT_S_I32:
865
47.0k
  case WASM_INS_F64_CONVERT_U_I32:
866
47.2k
  case WASM_INS_F64_CONVERT_S_I64:
867
47.5k
  case WASM_INS_F64_CONVERT_U_I64:
868
47.7k
  case WASM_INS_F64_PROMOTE_F32:
869
47.9k
  case WASM_INS_I32_REINTERPRET_F32:
870
48.1k
  case WASM_INS_I64_REINTERPRET_F64:
871
48.5k
  case WASM_INS_F32_REINTERPRET_I32:
872
48.7k
  case WASM_INS_F64_REINTERPRET_I64:
873
48.7k
    MI->wasm_data.type = WASM_OP_NONE;
874
875
48.7k
    if (MI->flat_insn->detail) {
876
48.7k
      MI->flat_insn->detail->wasm.op_count = 0;
877
48.7k
      MI->flat_insn->detail
878
48.7k
        ->groups[MI->flat_insn->detail->groups_count] =
879
48.7k
        WASM_GRP_NUMBERIC;
880
48.7k
      MI->flat_insn->detail->groups_count++;
881
48.7k
    }
882
883
48.7k
    *size = 1;
884
885
48.7k
    break;
886
887
205
  case WASM_INS_DROP:
888
458
  case WASM_INS_SELECT:
889
458
    MI->wasm_data.type = WASM_OP_NONE;
890
891
458
    if (MI->flat_insn->detail) {
892
458
      MI->flat_insn->detail->wasm.op_count = 0;
893
458
      MI->flat_insn->detail
894
458
        ->groups[MI->flat_insn->detail->groups_count] =
895
458
        WASM_GRP_PARAMETRIC;
896
458
      MI->flat_insn->detail->groups_count++;
897
458
    }
898
899
458
    *size = 1;
900
901
458
    break;
902
903
401
  case WASM_INS_GET_LOCAL:
904
637
  case WASM_INS_SET_LOCAL:
905
847
  case WASM_INS_TEE_LOCAL:
906
1.30k
  case WASM_INS_GET_GLOBAL:
907
1.83k
  case WASM_INS_SET_GLOBAL:
908
1.83k
    if (code_len == 1 ||
909
1.79k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
910
49
      return false;
911
49
    }
912
913
1.79k
    if (MI->flat_insn->detail) {
914
1.79k
      MI->flat_insn->detail->wasm.op_count = 1;
915
1.79k
      MI->flat_insn->detail
916
1.79k
        ->groups[MI->flat_insn->detail->groups_count] =
917
1.79k
        WASM_GRP_VARIABLE;
918
1.79k
      MI->flat_insn->detail->groups_count++;
919
1.79k
    }
920
921
1.79k
    *size = param_size + 1;
922
923
1.79k
    break;
924
925
508
  case WASM_INS_I32_LOAD:
926
718
  case WASM_INS_I64_LOAD:
927
964
  case WASM_INS_F32_LOAD:
928
1.44k
  case WASM_INS_F64_LOAD:
929
1.81k
  case WASM_INS_I32_LOAD8_S:
930
2.03k
  case WASM_INS_I32_LOAD8_U:
931
2.25k
  case WASM_INS_I32_LOAD16_S:
932
2.55k
  case WASM_INS_I32_LOAD16_U:
933
3.10k
  case WASM_INS_I64_LOAD8_S:
934
3.44k
  case WASM_INS_I64_LOAD8_U:
935
3.67k
  case WASM_INS_I64_LOAD16_S:
936
3.97k
  case WASM_INS_I64_LOAD16_U:
937
4.32k
  case WASM_INS_I64_LOAD32_S:
938
4.85k
  case WASM_INS_I64_LOAD32_U:
939
5.13k
  case WASM_INS_I32_STORE:
940
5.35k
  case WASM_INS_I64_STORE:
941
5.62k
  case WASM_INS_F32_STORE:
942
5.87k
  case WASM_INS_F64_STORE:
943
6.08k
  case WASM_INS_I32_STORE8:
944
6.34k
  case WASM_INS_I32_STORE16:
945
6.58k
  case WASM_INS_I64_STORE8:
946
6.91k
  case WASM_INS_I64_STORE16:
947
7.18k
  case WASM_INS_I64_STORE32:
948
7.18k
    if (code_len == 1 ||
949
6.99k
        !read_memoryimmediate(&code[1], code_len - 1, &param_size,
950
6.99k
            MI)) {
951
234
      return false;
952
234
    }
953
954
6.95k
    if (MI->flat_insn->detail) {
955
6.95k
      MI->flat_insn->detail->wasm.op_count = 2;
956
6.95k
      MI->flat_insn->detail
957
6.95k
        ->groups[MI->flat_insn->detail->groups_count] =
958
6.95k
        WASM_GRP_MEMORY;
959
6.95k
      MI->flat_insn->detail->groups_count++;
960
6.95k
    }
961
962
6.95k
    *size = param_size + 1;
963
964
6.95k
    break;
965
966
245
  case WASM_INS_CURRENT_MEMORY:
967
548
  case WASM_INS_GROW_MEMORY:
968
548
    MI->wasm_data.type = WASM_OP_NONE;
969
970
548
    if (MI->flat_insn->detail) {
971
548
      MI->flat_insn->detail->wasm.op_count = 0;
972
548
      MI->flat_insn->detail
973
548
        ->groups[MI->flat_insn->detail->groups_count] =
974
548
        WASM_GRP_MEMORY;
975
548
      MI->flat_insn->detail->groups_count++;
976
548
    }
977
978
548
    *size = 1;
979
980
548
    break;
981
982
3.44k
  case WASM_INS_UNREACHABLE:
983
3.74k
  case WASM_INS_NOP:
984
3.97k
  case WASM_INS_ELSE:
985
4.22k
  case WASM_INS_END:
986
4.61k
  case WASM_INS_RETURN:
987
4.61k
    MI->wasm_data.type = WASM_OP_NONE;
988
989
4.61k
    if (MI->flat_insn->detail) {
990
4.61k
      MI->flat_insn->detail->wasm.op_count = 0;
991
4.61k
      MI->flat_insn->detail
992
4.61k
        ->groups[MI->flat_insn->detail->groups_count] =
993
4.61k
        WASM_GRP_CONTROL;
994
4.61k
      MI->flat_insn->detail->groups_count++;
995
4.61k
    }
996
997
4.61k
    *size = 1;
998
999
4.61k
    break;
1000
1001
358
  case WASM_INS_BLOCK:
1002
574
  case WASM_INS_LOOP:
1003
1.00k
  case WASM_INS_IF:
1004
1.00k
    if (code_len == 1 ||
1005
974
        !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
1006
27
      return false;
1007
27
    }
1008
1009
974
    if (MI->flat_insn->detail) {
1010
974
      MI->flat_insn->detail->wasm.op_count = 1;
1011
974
      MI->flat_insn->detail
1012
974
        ->groups[MI->flat_insn->detail->groups_count] =
1013
974
        WASM_GRP_CONTROL;
1014
974
      MI->flat_insn->detail->groups_count++;
1015
974
    }
1016
1017
974
    *size = param_size + 1;
1018
1019
974
    break;
1020
1021
243
  case WASM_INS_BR:
1022
710
  case WASM_INS_BR_IF:
1023
958
  case WASM_INS_CALL:
1024
1.19k
  case WASM_INS_CALL_INDIRECT:
1025
1.19k
    if (code_len == 1 ||
1026
1.16k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
1027
34
      return false;
1028
34
    }
1029
1030
1.15k
    if (MI->flat_insn->detail) {
1031
1.15k
      MI->flat_insn->detail->wasm.op_count = 1;
1032
1.15k
      MI->flat_insn->detail
1033
1.15k
        ->groups[MI->flat_insn->detail->groups_count] =
1034
1.15k
        WASM_GRP_CONTROL;
1035
1.15k
      MI->flat_insn->detail->groups_count++;
1036
1.15k
    }
1037
1038
1.15k
    *size = param_size + 1;
1039
1040
1.15k
    break;
1041
1042
1.70k
  case WASM_INS_BR_TABLE:
1043
1.70k
    if (code_len == 1 ||
1044
1.69k
        !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
1045
134
      return false;
1046
134
    }
1047
1048
1.56k
    if (MI->flat_insn->detail) {
1049
1.56k
      MI->flat_insn->detail->wasm.op_count = 1;
1050
1.56k
      MI->flat_insn->detail
1051
1.56k
        ->groups[MI->flat_insn->detail->groups_count] =
1052
1.56k
        WASM_GRP_CONTROL;
1053
1.56k
      MI->flat_insn->detail->groups_count++;
1054
1.56k
    }
1055
1056
1.56k
    *size = param_size + 1;
1057
1058
1.56k
    break;
1059
68.4k
  }
1060
1061
67.9k
  return true;
1062
68.4k
}