Coverage Report

Created: 2025-11-16 06:38

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
30.4k
{
277
30.4k
  uint32_t data = 0;
278
30.4k
  int i;
279
280
41.6k
  for (i = 0;; i++) {
281
41.6k
    if (code_len < i + 1) {
282
254
      *leng = -1;
283
254
      return 0;
284
254
    }
285
286
41.3k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
287
467
      *leng = -1;
288
467
      return 0;
289
467
    }
290
291
40.9k
    data = data + (((uint32_t)code[i] & 0x7f) << (i * 7));
292
40.9k
    if (code[i] >> 7 == 0) {
293
29.6k
      break;
294
29.6k
    }
295
40.9k
  }
296
297
29.6k
  *leng = i + 1;
298
299
29.6k
  return data;
300
30.4k
}
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
326
{
309
326
  uint64_t data;
310
326
  int i;
311
312
326
  data = 0;
313
1.24k
  for (i = 0;; i++) {
314
1.24k
    if (code_len < i + 1) {
315
3
      *leng = -1;
316
3
      return 0;
317
3
    }
318
319
1.23k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
1
      *leng = -1;
321
1
      return 0;
322
1
    }
323
324
1.23k
    data = data + (((uint64_t)code[i] & 0x7f) << (i * 7));
325
1.23k
    if (code[i] >> 7 == 0) {
326
322
      break;
327
322
    }
328
1.23k
  }
329
330
322
  *leng = i + 1;
331
332
322
  return data;
333
326
}
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
422
{
340
422
  memcpy(dest, code, 4);
341
422
}
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
510
{
348
510
  memcpy(dest, code, 8);
349
510
}
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
899
{
357
899
  int8_t data;
358
359
899
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
899
  *leng = 1;
365
366
899
  if (code[0] == 0x40) {
367
73
    return -1;
368
73
  }
369
370
826
  data = code[0] & 0x7f;
371
372
826
  return data;
373
899
}
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.86k
{
383
2.86k
  size_t len = 0;
384
2.86k
  uint32_t data;
385
386
2.86k
  data = get_varuint32(code, code_len, &len);
387
2.86k
  if (len == -1) {
388
24
    return false;
389
24
  }
390
391
2.84k
  if (MI->flat_insn->detail) {
392
2.84k
    MI->flat_insn->detail->wasm.op_count = 1;
393
2.84k
    MI->flat_insn->detail->wasm.operands[0].type =
394
2.84k
      WASM_OP_VARUINT32;
395
2.84k
    MI->flat_insn->detail->wasm.operands[0].size = len;
396
2.84k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data;
397
2.84k
  }
398
399
2.84k
  MI->wasm_data.size = len;
400
2.84k
  MI->wasm_data.type = WASM_OP_VARUINT32;
401
2.84k
  MI->wasm_data.uint32 = data;
402
2.84k
  *param_size = len;
403
404
2.84k
  return true;
405
2.86k
}
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
326
{
415
326
  size_t len = 0;
416
326
  uint64_t data;
417
418
326
  data = get_varuint64(code, code_len, &len);
419
326
  if (len == -1) {
420
4
    return false;
421
4
  }
422
423
322
  if (MI->flat_insn->detail) {
424
322
    MI->flat_insn->detail->wasm.op_count = 1;
425
322
    MI->flat_insn->detail->wasm.operands[0].type =
426
322
      WASM_OP_VARUINT64;
427
322
    MI->flat_insn->detail->wasm.operands[0].size = len;
428
322
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
429
322
  }
430
431
322
  MI->wasm_data.size = len;
432
322
  MI->wasm_data.type = WASM_OP_VARUINT64;
433
322
  MI->wasm_data.uint64 = data;
434
322
  *param_size = len;
435
436
322
  return true;
437
326
}
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.70k
{
447
6.70k
  size_t tmp, len = 0;
448
6.70k
  uint32_t data[2];
449
450
6.70k
  if (MI->flat_insn->detail) {
451
6.70k
    MI->flat_insn->detail->wasm.op_count = 2;
452
6.70k
  }
453
454
6.70k
  data[0] = get_varuint32(code, code_len, &tmp);
455
6.70k
  if (tmp == -1) {
456
46
    return false;
457
46
  }
458
459
6.65k
  if (MI->flat_insn->detail) {
460
6.65k
    MI->flat_insn->detail->wasm.operands[0].type =
461
6.65k
      WASM_OP_VARUINT32;
462
6.65k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
463
6.65k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
464
6.65k
  }
465
466
6.65k
  len = tmp;
467
6.65k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
468
469
6.65k
  if (MI->flat_insn->detail) {
470
6.65k
    MI->flat_insn->detail->wasm.operands[1].type =
471
6.65k
      WASM_OP_VARUINT32;
472
6.65k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
473
6.65k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
474
6.65k
  }
475
476
6.65k
  len += tmp;
477
6.65k
  MI->wasm_data.size = len;
478
6.65k
  MI->wasm_data.type = WASM_OP_IMM;
479
6.65k
  MI->wasm_data.immediate[0] = data[0];
480
6.65k
  MI->wasm_data.immediate[1] = data[1];
481
6.65k
  *param_size = len;
482
483
6.65k
  return true;
484
6.70k
}
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
217
{
494
217
  if (code_len < 4) {
495
6
    return false;
496
6
  }
497
498
211
  get_uint32(code, &(MI->wasm_data.uint32));
499
500
211
  if (MI->flat_insn->detail) {
501
211
    MI->flat_insn->detail->wasm.op_count = 1;
502
211
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
503
211
    MI->flat_insn->detail->wasm.operands[0].size = 4;
504
211
    get_uint32(code,
505
211
         &(MI->flat_insn->detail->wasm.operands[0].uint32));
506
211
  }
507
508
211
  MI->wasm_data.size = 4;
509
211
  MI->wasm_data.type = WASM_OP_UINT32;
510
211
  *param_size = 4;
511
512
211
  return true;
513
217
}
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
267
{
523
267
  if (code_len < 8) {
524
12
    return false;
525
12
  }
526
527
255
  get_uint64(code, &(MI->wasm_data.uint64));
528
529
255
  if (MI->flat_insn->detail) {
530
255
    MI->flat_insn->detail->wasm.op_count = 1;
531
255
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
532
255
    MI->flat_insn->detail->wasm.operands[0].size = 8;
533
255
    get_uint64(code,
534
255
         &(MI->flat_insn->detail->wasm.operands[0].uint64));
535
255
  }
536
537
255
  MI->wasm_data.size = 8;
538
255
  MI->wasm_data.type = WASM_OP_UINT64;
539
255
  *param_size = 8;
540
541
255
  return true;
542
267
}
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.73k
{
552
1.73k
  uint32_t length, default_target;
553
1.73k
  int tmp_len = 0, i;
554
1.73k
  size_t var_len;
555
556
  // read length
557
1.73k
  length = get_varuint32(code, code_len, &var_len);
558
1.73k
  if (var_len == -1) {
559
9
    return false;
560
9
  }
561
562
1.72k
  tmp_len += var_len;
563
1.72k
  MI->wasm_data.brtable.length = length;
564
1.72k
  if (length >= UINT32_MAX - tmp_len) {
565
    // integer overflow check
566
1
    return false;
567
1
  }
568
1.72k
  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.66k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
574
575
1.66k
  if (MI->flat_insn->detail) {
576
1.66k
    MI->flat_insn->detail->wasm.op_count = 1;
577
1.66k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
578
1.66k
    MI->flat_insn->detail->wasm.operands[0].brtable.length =
579
1.66k
      MI->wasm_data.brtable.length;
580
1.66k
    MI->flat_insn->detail->wasm.operands[0].brtable.address =
581
1.66k
      MI->wasm_data.brtable.address;
582
1.66k
  }
583
584
  // read data
585
12.4k
  for (i = 0; i < length; i++) {
586
10.8k
    if (code_len < tmp_len) {
587
0
      return false;
588
0
    }
589
590
10.8k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
591
10.8k
    if (var_len == -1) {
592
20
      return false;
593
20
    }
594
595
10.7k
    tmp_len += var_len;
596
10.7k
  }
597
598
  // read default target
599
1.64k
  default_target =
600
1.64k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
601
1.64k
  if (var_len == -1) {
602
17
    return false;
603
17
  }
604
605
1.62k
  MI->wasm_data.brtable.default_target = default_target;
606
1.62k
  MI->wasm_data.type = WASM_OP_BRTABLE;
607
1.62k
  *param_size = tmp_len + var_len;
608
609
1.62k
  if (MI->flat_insn->detail) {
610
1.62k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
611
1.62k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target =
612
1.62k
      MI->wasm_data.brtable.default_target;
613
1.62k
  }
614
615
1.62k
  return true;
616
1.64k
}
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
899
{
626
899
  size_t len = 0;
627
628
899
  MI->wasm_data.type = WASM_OP_INT7;
629
899
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
630
899
  if (len == -1) {
631
0
    return false;
632
0
  }
633
634
899
  if (MI->flat_insn->detail) {
635
899
    MI->flat_insn->detail->wasm.op_count = 1;
636
899
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
637
899
    MI->flat_insn->detail->wasm.operands[0].size = 1;
638
899
    MI->flat_insn->detail->wasm.operands[0].int7 =
639
899
      MI->wasm_data.int7;
640
899
  }
641
642
899
  *param_size = len;
643
644
899
  return true;
645
899
}
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
64.4k
{
651
64.4k
  unsigned char opcode;
652
64.4k
  uint16_t param_size;
653
654
64.4k
  if (code_len == 0)
655
0
    return false;
656
657
64.4k
  opcode = code[0];
658
64.4k
  if (opcodes[opcode] == -1) {
659
    // invalid opcode
660
176
    return false;
661
176
  }
662
663
  // valid opcode
664
64.2k
  MI->address = address;
665
64.2k
  MI->OpcodePub = MI->Opcode = opcode;
666
667
64.2k
  if (MI->flat_insn->detail) {
668
64.2k
    memset(MI->flat_insn->detail, 0,
669
64.2k
           offsetof(cs_detail, wasm) + sizeof(cs_wasm));
670
64.2k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
671
64.2k
  }
672
673
  // setup groups
674
64.2k
  switch (opcode) {
675
0
  default:
676
0
    return false;
677
678
219
  case WASM_INS_I32_CONST:
679
219
    if (code_len == 1 ||
680
215
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
681
7
      return false;
682
7
    }
683
684
212
    if (MI->flat_insn->detail) {
685
212
      MI->flat_insn->detail->wasm.op_count = 1;
686
212
      MI->flat_insn->detail
687
212
        ->groups[MI->flat_insn->detail->groups_count] =
688
212
        WASM_GRP_NUMBERIC;
689
212
      MI->flat_insn->detail->groups_count++;
690
212
    }
691
692
212
    *size = param_size + 1;
693
694
212
    break;
695
696
332
  case WASM_INS_I64_CONST:
697
332
    if (code_len == 1 ||
698
326
        !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
699
10
      return false;
700
10
    }
701
702
322
    if (MI->flat_insn->detail) {
703
322
      MI->flat_insn->detail->wasm.op_count = 1;
704
322
      MI->flat_insn->detail
705
322
        ->groups[MI->flat_insn->detail->groups_count] =
706
322
        WASM_GRP_NUMBERIC;
707
322
      MI->flat_insn->detail->groups_count++;
708
322
    }
709
710
322
    *size = param_size + 1;
711
712
322
    break;
713
714
222
  case WASM_INS_F32_CONST:
715
222
    if (code_len == 1 ||
716
217
        !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
717
11
      return false;
718
11
    }
719
720
211
    if (MI->flat_insn->detail) {
721
211
      MI->flat_insn->detail->wasm.op_count = 1;
722
211
      MI->flat_insn->detail
723
211
        ->groups[MI->flat_insn->detail->groups_count] =
724
211
        WASM_GRP_NUMBERIC;
725
211
      MI->flat_insn->detail->groups_count++;
726
211
    }
727
728
211
    *size = param_size + 1;
729
730
211
    break;
731
732
272
  case WASM_INS_F64_CONST:
733
272
    if (code_len == 1 ||
734
267
        !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
735
17
      return false;
736
17
    }
737
738
255
    if (MI->flat_insn->detail) {
739
255
      MI->flat_insn->detail->wasm.op_count = 1;
740
255
      MI->flat_insn->detail
741
255
        ->groups[MI->flat_insn->detail->groups_count] =
742
255
        WASM_GRP_NUMBERIC;
743
255
      MI->flat_insn->detail->groups_count++;
744
255
    }
745
746
255
    *size = param_size + 1;
747
748
255
    break;
749
750
292
  case WASM_INS_I32_EQZ:
751
521
  case WASM_INS_I32_EQ:
752
894
  case WASM_INS_I32_NE:
753
1.15k
  case WASM_INS_I32_LT_S:
754
1.35k
  case WASM_INS_I32_LT_U:
755
1.61k
  case WASM_INS_I32_GT_S:
756
1.82k
  case WASM_INS_I32_GT_U:
757
2.24k
  case WASM_INS_I32_LE_S:
758
2.44k
  case WASM_INS_I32_LE_U:
759
2.79k
  case WASM_INS_I32_GE_S:
760
3.00k
  case WASM_INS_I32_GE_U:
761
3.22k
  case WASM_INS_I64_EQZ:
762
3.43k
  case WASM_INS_I64_EQ:
763
4.80k
  case WASM_INS_I64_NE:
764
5.03k
  case WASM_INS_I64_LT_S:
765
5.24k
  case WASM_INS_I64_LT_U:
766
5.47k
  case WASN_INS_I64_GT_S:
767
5.67k
  case WASM_INS_I64_GT_U:
768
6.39k
  case WASM_INS_I64_LE_S:
769
6.67k
  case WASM_INS_I64_LE_U:
770
7.01k
  case WASM_INS_I64_GE_S:
771
7.25k
  case WASM_INS_I64_GE_U:
772
7.50k
  case WASM_INS_F32_EQ:
773
7.71k
  case WASM_INS_F32_NE:
774
7.98k
  case WASM_INS_F32_LT:
775
8.23k
  case WASM_INS_F32_GT:
776
8.47k
  case WASM_INS_F32_LE:
777
8.69k
  case WASM_INS_F32_GE:
778
9.05k
  case WASM_INS_F64_EQ:
779
9.42k
  case WASM_INS_F64_NE:
780
9.88k
  case WASM_INS_F64_LT:
781
10.4k
  case WASM_INS_F64_GT:
782
14.5k
  case WASM_INS_F64_LE:
783
14.9k
  case WASM_INS_F64_GE:
784
15.1k
  case WASM_INS_I32_CLZ:
785
15.4k
  case WASM_INS_I32_CTZ:
786
15.8k
  case WASM_INS_I32_POPCNT:
787
16.0k
  case WASM_INS_I32_ADD:
788
16.3k
  case WASM_INS_I32_SUB:
789
16.5k
  case WASM_INS_I32_MUL:
790
16.8k
  case WASM_INS_I32_DIV_S:
791
17.2k
  case WASM_INS_I32_DIV_U:
792
17.7k
  case WASM_INS_I32_REM_S:
793
17.9k
  case WASM_INS_I32_REM_U:
794
18.2k
  case WASM_INS_I32_AND:
795
18.4k
  case WASM_INS_I32_OR:
796
18.7k
  case WASM_INS_I32_XOR:
797
19.2k
  case WASM_INS_I32_SHL:
798
19.7k
  case WASM_INS_I32_SHR_S:
799
19.9k
  case WASM_INS_I32_SHR_U:
800
20.1k
  case WASM_INS_I32_ROTL:
801
20.9k
  case WASM_INS_I32_ROTR:
802
21.1k
  case WASM_INS_I64_CLZ:
803
21.4k
  case WASM_INS_I64_CTZ:
804
21.6k
  case WASM_INS_I64_POPCNT:
805
21.9k
  case WASM_INS_I64_ADD:
806
22.1k
  case WASM_INS_I64_SUB:
807
22.4k
  case WASM_INS_I64_MUL:
808
22.7k
  case WASM_INS_I64_DIV_S:
809
25.3k
  case WASM_INS_I64_DIV_U:
810
27.2k
  case WASM_INS_I64_REM_S:
811
27.9k
  case WASM_INS_I64_REM_U:
812
28.1k
  case WASM_INS_I64_AND:
813
28.3k
  case WASM_INS_I64_OR:
814
28.5k
  case WASM_INS_I64_XOR:
815
28.7k
  case WASM_INS_I64_SHL:
816
29.0k
  case WASM_INS_I64_SHR_S:
817
29.3k
  case WASM_INS_I64_SHR_U:
818
29.5k
  case WASM_INS_I64_ROTL:
819
29.7k
  case WASM_INS_I64_ROTR:
820
29.9k
  case WASM_INS_F32_ABS:
821
30.2k
  case WASM_INS_F32_NEG:
822
30.4k
  case WASM_INS_F32_CEIL:
823
30.6k
  case WASM_INS_F32_FLOOR:
824
30.8k
  case WASM_INS_F32_TRUNC:
825
31.1k
  case WASM_INS_F32_NEAREST:
826
31.4k
  case WASM_INS_F32_SQRT:
827
32.0k
  case WASM_INS_F32_ADD:
828
32.2k
  case WASM_INS_F32_SUB:
829
32.4k
  case WASM_INS_F32_MUL:
830
32.6k
  case WASM_INS_F32_DIV:
831
32.8k
  case WASM_INS_F32_MIN:
832
33.1k
  case WASM_INS_F32_MAX:
833
33.5k
  case WASM_INS_F32_COPYSIGN:
834
34.1k
  case WASM_INS_F64_ABS:
835
34.6k
  case WASM_INS_F64_NEG:
836
35.0k
  case WASM_INS_F64_CEIL:
837
35.2k
  case WASM_INS_F64_FLOOR:
838
35.6k
  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.6k
  case WASM_INS_F64_SUB:
843
36.8k
  case WASM_INS_F64_MUL:
844
37.1k
  case WASM_INS_F64_DIV:
845
37.4k
  case WASM_INS_F64_MIN:
846
37.6k
  case WASM_INS_F64_MAX:
847
37.8k
  case WASM_INS_F64_COPYSIGN:
848
38.0k
  case WASM_INS_I32_WARP_I64:
849
38.2k
  case WASP_INS_I32_TRUNC_S_F32:
850
38.6k
  case WASM_INS_I32_TRUNC_U_F32:
851
38.8k
  case WASM_INS_I32_TRUNC_S_F64:
852
40.3k
  case WASM_INS_I32_TRUNC_U_F64:
853
40.8k
  case WASM_INS_I64_EXTEND_S_I32:
854
41.0k
  case WASM_INS_I64_EXTEND_U_I32:
855
41.4k
  case WASM_INS_I64_TRUNC_S_F32:
856
42.0k
  case WASM_INS_I64_TRUNC_U_F32:
857
42.2k
  case WASM_INS_I64_TRUNC_S_F64:
858
42.4k
  case WASM_INS_I64_TRUNC_U_F64:
859
42.6k
  case WASM_INS_F32_CONVERT_S_I32:
860
42.9k
  case WASM_INS_F32_CONVERT_U_I32:
861
43.1k
  case WASM_INS_F32_CONVERT_S_I64:
862
43.3k
  case WASM_INS_F32_CONVERT_U_I64:
863
43.6k
  case WASM_INS_F32_DEMOTE_F64:
864
43.8k
  case WASM_INS_F64_CONVERT_S_I32:
865
44.0k
  case WASM_INS_F64_CONVERT_U_I32:
866
44.3k
  case WASM_INS_F64_CONVERT_S_I64:
867
44.5k
  case WASM_INS_F64_CONVERT_U_I64:
868
44.7k
  case WASM_INS_F64_PROMOTE_F32:
869
44.9k
  case WASM_INS_I32_REINTERPRET_F32:
870
45.2k
  case WASM_INS_I64_REINTERPRET_F64:
871
45.4k
  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
212
  case WASM_INS_DROP:
888
422
  case WASM_INS_SELECT:
889
422
    MI->wasm_data.type = WASM_OP_NONE;
890
891
422
    if (MI->flat_insn->detail) {
892
422
      MI->flat_insn->detail->wasm.op_count = 0;
893
422
      MI->flat_insn->detail
894
422
        ->groups[MI->flat_insn->detail->groups_count] =
895
422
        WASM_GRP_PARAMETRIC;
896
422
      MI->flat_insn->detail->groups_count++;
897
422
    }
898
899
422
    *size = 1;
900
901
422
    break;
902
903
332
  case WASM_INS_GET_LOCAL:
904
564
  case WASM_INS_SET_LOCAL:
905
818
  case WASM_INS_TEE_LOCAL:
906
1.16k
  case WASM_INS_GET_GLOBAL:
907
1.51k
  case WASM_INS_SET_GLOBAL:
908
1.51k
    if (code_len == 1 ||
909
1.47k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
910
50
      return false;
911
50
    }
912
913
1.46k
    if (MI->flat_insn->detail) {
914
1.46k
      MI->flat_insn->detail->wasm.op_count = 1;
915
1.46k
      MI->flat_insn->detail
916
1.46k
        ->groups[MI->flat_insn->detail->groups_count] =
917
1.46k
        WASM_GRP_VARIABLE;
918
1.46k
      MI->flat_insn->detail->groups_count++;
919
1.46k
    }
920
921
1.46k
    *size = param_size + 1;
922
923
1.46k
    break;
924
925
461
  case WASM_INS_I32_LOAD:
926
691
  case WASM_INS_I64_LOAD:
927
952
  case WASM_INS_F32_LOAD:
928
1.45k
  case WASM_INS_F64_LOAD:
929
1.69k
  case WASM_INS_I32_LOAD8_S:
930
1.92k
  case WASM_INS_I32_LOAD8_U:
931
2.14k
  case WASM_INS_I32_LOAD16_S:
932
2.38k
  case WASM_INS_I32_LOAD16_U:
933
2.84k
  case WASM_INS_I64_LOAD8_S:
934
3.10k
  case WASM_INS_I64_LOAD8_U:
935
3.34k
  case WASM_INS_I64_LOAD16_S:
936
3.67k
  case WASM_INS_I64_LOAD16_U:
937
3.91k
  case WASM_INS_I64_LOAD32_S:
938
4.39k
  case WASM_INS_I64_LOAD32_U:
939
4.87k
  case WASM_INS_I32_STORE:
940
5.09k
  case WASM_INS_I64_STORE:
941
5.34k
  case WASM_INS_F32_STORE:
942
5.57k
  case WASM_INS_F64_STORE:
943
5.82k
  case WASM_INS_I32_STORE8:
944
6.05k
  case WASM_INS_I32_STORE16:
945
6.26k
  case WASM_INS_I64_STORE8:
946
6.57k
  case WASM_INS_I64_STORE16:
947
6.84k
  case WASM_INS_I64_STORE32:
948
6.84k
    if (code_len == 1 ||
949
6.70k
        !read_memoryimmediate(&code[1], code_len - 1, &param_size,
950
6.70k
            MI)) {
951
187
      return false;
952
187
    }
953
954
6.65k
    if (MI->flat_insn->detail) {
955
6.65k
      MI->flat_insn->detail->wasm.op_count = 2;
956
6.65k
      MI->flat_insn->detail
957
6.65k
        ->groups[MI->flat_insn->detail->groups_count] =
958
6.65k
        WASM_GRP_MEMORY;
959
6.65k
      MI->flat_insn->detail->groups_count++;
960
6.65k
    }
961
962
6.65k
    *size = param_size + 1;
963
964
6.65k
    break;
965
966
253
  case WASM_INS_CURRENT_MEMORY:
967
554
  case WASM_INS_GROW_MEMORY:
968
554
    MI->wasm_data.type = WASM_OP_NONE;
969
970
554
    if (MI->flat_insn->detail) {
971
554
      MI->flat_insn->detail->wasm.op_count = 0;
972
554
      MI->flat_insn->detail
973
554
        ->groups[MI->flat_insn->detail->groups_count] =
974
554
        WASM_GRP_MEMORY;
975
554
      MI->flat_insn->detail->groups_count++;
976
554
    }
977
978
554
    *size = 1;
979
980
554
    break;
981
982
3.40k
  case WASM_INS_UNREACHABLE:
983
3.70k
  case WASM_INS_NOP:
984
3.94k
  case WASM_INS_ELSE:
985
4.16k
  case WASM_INS_END:
986
4.38k
  case WASM_INS_RETURN:
987
4.38k
    MI->wasm_data.type = WASM_OP_NONE;
988
989
4.38k
    if (MI->flat_insn->detail) {
990
4.38k
      MI->flat_insn->detail->wasm.op_count = 0;
991
4.38k
      MI->flat_insn->detail
992
4.38k
        ->groups[MI->flat_insn->detail->groups_count] =
993
4.38k
        WASM_GRP_CONTROL;
994
4.38k
      MI->flat_insn->detail->groups_count++;
995
4.38k
    }
996
997
4.38k
    *size = 1;
998
999
4.38k
    break;
1000
1001
255
  case WASM_INS_BLOCK:
1002
505
  case WASM_INS_LOOP:
1003
918
  case WASM_INS_IF:
1004
918
    if (code_len == 1 ||
1005
899
        !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
1006
19
      return false;
1007
19
    }
1008
1009
899
    if (MI->flat_insn->detail) {
1010
899
      MI->flat_insn->detail->wasm.op_count = 1;
1011
899
      MI->flat_insn->detail
1012
899
        ->groups[MI->flat_insn->detail->groups_count] =
1013
899
        WASM_GRP_CONTROL;
1014
899
      MI->flat_insn->detail->groups_count++;
1015
899
    }
1016
1017
899
    *size = param_size + 1;
1018
1019
899
    break;
1020
1021
233
  case WASM_INS_BR:
1022
660
  case WASM_INS_BR_IF:
1023
908
  case WASM_INS_CALL:
1024
1.20k
  case WASM_INS_CALL_INDIRECT:
1025
1.20k
    if (code_len == 1 ||
1026
1.17k
        !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
1027
33
      return false;
1028
33
    }
1029
1030
1.16k
    if (MI->flat_insn->detail) {
1031
1.16k
      MI->flat_insn->detail->wasm.op_count = 1;
1032
1.16k
      MI->flat_insn->detail
1033
1.16k
        ->groups[MI->flat_insn->detail->groups_count] =
1034
1.16k
        WASM_GRP_CONTROL;
1035
1.16k
      MI->flat_insn->detail->groups_count++;
1036
1.16k
    }
1037
1038
1.16k
    *size = param_size + 1;
1039
1040
1.16k
    break;
1041
1042
1.73k
  case WASM_INS_BR_TABLE:
1043
1.73k
    if (code_len == 1 ||
1044
1.73k
        !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
1045
107
      return false;
1046
107
    }
1047
1048
1.62k
    if (MI->flat_insn->detail) {
1049
1.62k
      MI->flat_insn->detail->wasm.op_count = 1;
1050
1.62k
      MI->flat_insn->detail
1051
1.62k
        ->groups[MI->flat_insn->detail->groups_count] =
1052
1.62k
        WASM_GRP_CONTROL;
1053
1.62k
      MI->flat_insn->detail->groups_count++;
1054
1.62k
    }
1055
1056
1.62k
    *size = param_size + 1;
1057
1058
1.62k
    break;
1059
64.2k
  }
1060
1061
63.8k
  return true;
1062
64.2k
}