Coverage Report

Created: 2025-11-09 07:00

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/capstonev5/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> // offsetof macro
6
// alternatively #include "../../utils.h" like everyone else
7
8
#include "WASMDisassembler.h"
9
#include "WASMMapping.h"
10
#include "../../cs_priv.h"
11
12
static const short opcodes[256] = {
13
  WASM_INS_UNREACHABLE,
14
  WASM_INS_NOP,
15
  WASM_INS_BLOCK,
16
  WASM_INS_LOOP,
17
  WASM_INS_IF,
18
  WASM_INS_ELSE,
19
  -1,
20
  -1,
21
  -1,
22
  -1,
23
  -1,
24
  WASM_INS_END,
25
  WASM_INS_BR,
26
  WASM_INS_BR_IF,
27
  WASM_INS_BR_TABLE,
28
  WASM_INS_RETURN,
29
  WASM_INS_CALL,
30
  WASM_INS_CALL_INDIRECT,
31
  -1,
32
  -1,
33
  -1,
34
  -1,
35
  -1,
36
  -1,
37
  -1,
38
  -1,
39
  WASM_INS_DROP,
40
  WASM_INS_SELECT,
41
  -1,
42
  -1,
43
  -1,
44
  -1,
45
  WASM_INS_GET_LOCAL,
46
  WASM_INS_SET_LOCAL,
47
  WASM_INS_TEE_LOCAL,
48
  WASM_INS_GET_GLOBAL,
49
  WASM_INS_SET_GLOBAL,
50
  -1,
51
  -1,
52
  -1,
53
  WASM_INS_I32_LOAD,
54
  WASM_INS_I64_LOAD,
55
  WASM_INS_F32_LOAD,
56
  WASM_INS_F64_LOAD,
57
  WASM_INS_I32_LOAD8_S,
58
  WASM_INS_I32_LOAD8_U,
59
  WASM_INS_I32_LOAD16_S,
60
  WASM_INS_I32_LOAD16_U,
61
  WASM_INS_I64_LOAD8_S,
62
  WASM_INS_I64_LOAD8_U,
63
  WASM_INS_I64_LOAD16_S,
64
  WASM_INS_I64_LOAD16_U,
65
  WASM_INS_I64_LOAD32_S,
66
  WASM_INS_I64_LOAD32_U,
67
  WASM_INS_I32_STORE,
68
  WASM_INS_I64_STORE,
69
  WASM_INS_F32_STORE,
70
  WASM_INS_F64_STORE,
71
  WASM_INS_I32_STORE8,
72
  WASM_INS_I32_STORE16,
73
  WASM_INS_I64_STORE8,
74
  WASM_INS_I64_STORE16,
75
  WASM_INS_I64_STORE32,
76
  WASM_INS_CURRENT_MEMORY,
77
  WASM_INS_GROW_MEMORY,
78
  WASM_INS_I32_CONST,
79
  WASM_INS_I64_CONST,
80
  WASM_INS_F32_CONST,
81
  WASM_INS_F64_CONST,
82
  WASM_INS_I32_EQZ,
83
  WASM_INS_I32_EQ,
84
  WASM_INS_I32_NE,
85
  WASM_INS_I32_LT_S,
86
  WASM_INS_I32_LT_U,
87
  WASM_INS_I32_GT_S,
88
  WASM_INS_I32_GT_U,
89
  WASM_INS_I32_LE_S,
90
  WASM_INS_I32_LE_U,
91
  WASM_INS_I32_GE_S,
92
  WASM_INS_I32_GE_U,
93
  WASM_INS_I64_EQZ,
94
  WASM_INS_I64_EQ,
95
  WASM_INS_I64_NE,
96
  WASM_INS_I64_LT_S,
97
  WASM_INS_I64_LT_U,
98
  WASN_INS_I64_GT_S,
99
  WASM_INS_I64_GT_U,
100
  WASM_INS_I64_LE_S,
101
  WASM_INS_I64_LE_U,
102
  WASM_INS_I64_GE_S,
103
  WASM_INS_I64_GE_U,
104
  WASM_INS_F32_EQ,
105
  WASM_INS_F32_NE,
106
  WASM_INS_F32_LT,
107
  WASM_INS_F32_GT,
108
  WASM_INS_F32_LE,
109
  WASM_INS_F32_GE,
110
  WASM_INS_F64_EQ,
111
  WASM_INS_F64_NE,
112
  WASM_INS_F64_LT,
113
  WASM_INS_F64_GT,
114
  WASM_INS_F64_LE,
115
  WASM_INS_F64_GE,
116
  WASM_INS_I32_CLZ,
117
  WASM_INS_I32_CTZ,
118
  WASM_INS_I32_POPCNT,
119
  WASM_INS_I32_ADD,
120
  WASM_INS_I32_SUB,
121
  WASM_INS_I32_MUL,
122
  WASM_INS_I32_DIV_S,
123
  WASM_INS_I32_DIV_U,
124
  WASM_INS_I32_REM_S,
125
  WASM_INS_I32_REM_U,
126
  WASM_INS_I32_AND,
127
  WASM_INS_I32_OR,
128
  WASM_INS_I32_XOR,
129
  WASM_INS_I32_SHL,
130
  WASM_INS_I32_SHR_S,
131
  WASM_INS_I32_SHR_U,
132
  WASM_INS_I32_ROTL,
133
  WASM_INS_I32_ROTR,
134
  WASM_INS_I64_CLZ,
135
  WASM_INS_I64_CTZ,
136
  WASM_INS_I64_POPCNT,
137
  WASM_INS_I64_ADD,
138
  WASM_INS_I64_SUB,
139
  WASM_INS_I64_MUL,
140
  WASM_INS_I64_DIV_S,
141
  WASM_INS_I64_DIV_U,
142
  WASM_INS_I64_REM_S,
143
  WASM_INS_I64_REM_U,
144
  WASM_INS_I64_AND,
145
  WASM_INS_I64_OR,
146
  WASM_INS_I64_XOR,
147
  WASM_INS_I64_SHL,
148
  WASM_INS_I64_SHR_S,
149
  WASM_INS_I64_SHR_U,
150
  WASM_INS_I64_ROTL,
151
  WASM_INS_I64_ROTR,
152
  WASM_INS_F32_ABS,
153
  WASM_INS_F32_NEG,
154
  WASM_INS_F32_CEIL,
155
  WASM_INS_F32_FLOOR,
156
  WASM_INS_F32_TRUNC,
157
  WASM_INS_F32_NEAREST,
158
  WASM_INS_F32_SQRT,
159
  WASM_INS_F32_ADD,
160
  WASM_INS_F32_SUB,
161
  WASM_INS_F32_MUL,
162
  WASM_INS_F32_DIV,
163
  WASM_INS_F32_MIN,
164
  WASM_INS_F32_MAX,
165
  WASM_INS_F32_COPYSIGN,
166
  WASM_INS_F64_ABS,
167
  WASM_INS_F64_NEG,
168
  WASM_INS_F64_CEIL,
169
  WASM_INS_F64_FLOOR,
170
  WASM_INS_F64_TRUNC,
171
  WASM_INS_F64_NEAREST,
172
  WASM_INS_F64_SQRT,
173
  WASM_INS_F64_ADD,
174
  WASM_INS_F64_SUB,
175
  WASM_INS_F64_MUL,
176
  WASM_INS_F64_DIV,
177
  WASM_INS_F64_MIN,
178
  WASM_INS_F64_MAX,
179
  WASM_INS_F64_COPYSIGN,
180
  WASM_INS_I32_WARP_I64,
181
  WASP_INS_I32_TRUNC_S_F32,
182
  WASM_INS_I32_TRUNC_U_F32,
183
  WASM_INS_I32_TRUNC_S_F64,
184
  WASM_INS_I32_TRUNC_U_F64,
185
  WASM_INS_I64_EXTEND_S_I32,
186
  WASM_INS_I64_EXTEND_U_I32,
187
  WASM_INS_I64_TRUNC_S_F32,
188
  WASM_INS_I64_TRUNC_U_F32,
189
  WASM_INS_I64_TRUNC_S_F64,
190
  WASM_INS_I64_TRUNC_U_F64,
191
  WASM_INS_F32_CONVERT_S_I32,
192
  WASM_INS_F32_CONVERT_U_I32,
193
  WASM_INS_F32_CONVERT_S_I64,
194
  WASM_INS_F32_CONVERT_U_I64,
195
  WASM_INS_F32_DEMOTE_F64,
196
  WASM_INS_F64_CONVERT_S_I32,
197
  WASM_INS_F64_CONVERT_U_I32,
198
  WASM_INS_F64_CONVERT_S_I64,
199
  WASM_INS_F64_CONVERT_U_I64,
200
  WASM_INS_F64_PROMOTE_F32,
201
  WASM_INS_I32_REINTERPRET_F32,
202
  WASM_INS_I64_REINTERPRET_F64,
203
  WASM_INS_F32_REINTERPRET_I32,
204
  WASM_INS_F64_REINTERPRET_I64,
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
  -1,
269
};
270
271
// input  | code: code pointer start from varuint32
272
//        | code_len: real code len count from varint
273
//        | leng: return value, means length of varint. -1 means error
274
// return | varint
275
static uint32_t get_varuint32(const uint8_t *code, size_t code_len, size_t *leng)
276
22.2k
{
277
22.2k
  uint32_t data = 0;
278
22.2k
  int i;
279
280
33.0k
  for(i = 0;; i++) {
281
33.0k
    if (code_len < i + 1) {
282
247
      *leng = -1;
283
247
      return 0;
284
247
    }
285
286
287
32.7k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
288
533
      *leng = -1;
289
533
      return 0;
290
533
    }
291
292
32.2k
    data = data + (((uint32_t) code[i] & 0x7f) << (i * 7));
293
32.2k
    if (code[i] >> 7 == 0) {
294
21.4k
      break;
295
21.4k
    }
296
32.2k
  }
297
298
21.4k
  *leng = i + 1;
299
300
21.4k
  return data;
301
22.2k
}
302
303
// input  | code : code pointer start from varuint64
304
//        | code_len : real code len count from varint
305
//        | leng: return value, means length of varint. -1 means error
306
// return   | varint
307
static uint64_t get_varuint64(const uint8_t *code, size_t code_len, size_t *leng)
308
306
{
309
306
  uint64_t data;
310
306
  int i;
311
312
306
  data = 0;
313
1.15k
  for(i = 0;; i++){
314
1.15k
    if (code_len < i + 1) {
315
5
      *leng = -1;
316
5
      return 0;
317
5
    }
318
319
1.15k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
2
      *leng = -1;
321
2
      return 0;
322
2
    }
323
324
1.14k
    data = data + (((uint64_t) code[i] & 0x7f) << (i * 7));
325
1.14k
    if (code[i] >> 7 == 0) {
326
299
      break;
327
299
    }
328
1.14k
  }
329
330
299
  *leng = i + 1;
331
332
299
  return data;
333
306
}
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
430
{
340
430
  memcpy(dest, code, 4);
341
430
}
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
844
{
348
844
  memcpy(dest, code, 8);
349
844
}
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
1.10k
{
357
1.10k
  int8_t data;
358
359
1.10k
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
1.10k
  *leng = 1;
365
366
1.10k
  if (code[0] == 0x40) {
367
200
    return -1;
368
200
  }
369
370
909
  data = code[0] & 0x7f;
371
372
909
  return data;
373
1.10k
}
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, uint16_t *param_size, MCInst *MI)
381
2.35k
{
382
2.35k
  size_t len = 0;
383
2.35k
  uint32_t data;
384
385
2.35k
  data = get_varuint32(code, code_len, &len);
386
2.35k
  if (len == -1) {
387
17
    return false;
388
17
  }
389
390
2.33k
  if (MI->flat_insn->detail) {
391
2.33k
    MI->flat_insn->detail->wasm.op_count = 1;
392
2.33k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
393
2.33k
    MI->flat_insn->detail->wasm.operands[0].size= len;
394
2.33k
    MI->flat_insn->detail->wasm.operands[0].varuint32= data;
395
2.33k
  }
396
397
2.33k
  MI->wasm_data.size = len;
398
2.33k
  MI->wasm_data.type = WASM_OP_VARUINT32;
399
2.33k
  MI->wasm_data.uint32 = data;
400
2.33k
  *param_size = len;
401
402
2.33k
  return true;
403
2.35k
}
404
405
// input  | code : code pointer start from varuint64
406
//      | code_len : start from the code pointer to the end, how long is it
407
//      | param_size : pointer of the param size
408
//      | MI : Mcinst handler in this round of disasm
409
// return   | true/false if the function successfully finished 
410
static bool read_varuint64(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
411
306
{
412
306
  size_t len = 0;
413
306
  uint64_t data;
414
415
306
  data = get_varuint64(code, code_len, &len);
416
306
  if (len == -1) {
417
7
    return false;
418
7
  }
419
420
299
  if (MI->flat_insn->detail) {
421
299
    MI->flat_insn->detail->wasm.op_count = 1;
422
299
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT64;
423
299
    MI->flat_insn->detail->wasm.operands[0].size = len;
424
299
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
425
299
  }
426
427
299
  MI->wasm_data.size = len;
428
299
  MI->wasm_data.type = WASM_OP_VARUINT64;
429
299
  MI->wasm_data.uint64 = data;
430
299
  *param_size = len;
431
432
299
  return true;
433
306
}
434
435
// input  | code : code pointer start from memoryimmediate
436
//      | code_len : start from the code pointer to the end, how long is it
437
//      | param_size : pointer of the param size (sum of two params)
438
//      | MI : Mcinst handler in this round of disasm
439
// return   | true/false if the function successfully finished 
440
static bool read_memoryimmediate(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
441
5.95k
{
442
5.95k
  size_t tmp, len = 0;
443
5.95k
  uint32_t data[2];
444
445
5.95k
  if (MI->flat_insn->detail) {
446
5.95k
    MI->flat_insn->detail->wasm.op_count = 2;
447
5.95k
  }
448
449
5.95k
  data[0] = get_varuint32(code, code_len, &tmp);
450
5.95k
  if (tmp == -1) {
451
39
    return false;
452
39
  }
453
454
5.91k
  if (MI->flat_insn->detail) {
455
5.91k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
456
5.91k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
457
5.91k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
458
5.91k
  }
459
460
5.91k
  len = tmp;
461
5.91k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
462
5.91k
  if (len == -1) {
463
0
    return false;
464
0
  }
465
466
5.91k
  if (MI->flat_insn->detail) {
467
5.91k
    MI->flat_insn->detail->wasm.operands[1].type = WASM_OP_VARUINT32;
468
5.91k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
469
5.91k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
470
5.91k
  }
471
472
5.91k
  len += tmp;
473
5.91k
  MI->wasm_data.size = len;
474
5.91k
  MI->wasm_data.type = WASM_OP_IMM;
475
5.91k
  MI->wasm_data.immediate[0] = data[0];
476
5.91k
  MI->wasm_data.immediate[1] = data[1];
477
5.91k
  *param_size = len;
478
479
5.91k
  return true;
480
5.91k
}
481
482
// input  | code : code pointer start from uint32
483
//      | code_len : start from the code pointer to the end, how long is it
484
//      | param_size : pointer of the param size
485
//      | MI : Mcinst handler in this round of disasm
486
// return   | true/false if the function successfully finished 
487
static bool read_uint32(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
488
222
{
489
222
  if (code_len < 4) {
490
7
    return false;
491
7
  }
492
493
215
  get_uint32(code, &(MI->wasm_data.uint32));
494
495
215
  if (MI->flat_insn->detail) {
496
215
    MI->flat_insn->detail->wasm.op_count = 1;
497
215
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
498
215
    MI->flat_insn->detail->wasm.operands[0].size = 4;
499
215
    get_uint32(code, &(MI->flat_insn->detail->wasm.operands[0].uint32));
500
215
  }
501
502
215
  MI->wasm_data.size = 4;
503
215
  MI->wasm_data.type = WASM_OP_UINT32;
504
215
  *param_size = 4;
505
506
215
  return true;
507
222
}
508
509
// input  | code : code pointer start from uint64
510
//      | code_len : start from the code pointer to the end, how long is it
511
//      | param_size : pointer of the param size
512
//      | MI : Mcinst handler in this round of disasm
513
// return   | true/false if the function successfully finished 
514
static bool read_uint64(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
515
429
{
516
429
  if (code_len < 8) {
517
7
    return false;
518
7
  }
519
520
422
  get_uint64(code, &(MI->wasm_data.uint64));
521
522
422
  if (MI->flat_insn->detail) {
523
422
    MI->flat_insn->detail->wasm.op_count = 1;
524
422
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
525
422
    MI->flat_insn->detail->wasm.operands[0].size = 8;
526
422
    get_uint64(code, &(MI->flat_insn->detail->wasm.operands[0].uint64));
527
422
  }
528
529
422
  MI->wasm_data.size = 8;
530
422
  MI->wasm_data.type = WASM_OP_UINT64;
531
422
  *param_size = 8;
532
533
422
  return true;
534
429
}
535
536
// input  | code : code pointer start from brtable
537
//      | code_len : start from the code pointer to the end, how long is it
538
//      | param_size : pointer of the param size (sum of all param)
539
//      | MI : Mcinst handler in this round of disasm
540
// return   | true/false if the function successfully finished 
541
static bool read_brtable(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
542
1.43k
{
543
1.43k
  uint32_t length, default_target;
544
1.43k
  int tmp_len = 0, i;
545
1.43k
  size_t var_len;
546
547
  // read length
548
1.43k
  length = get_varuint32(code, code_len, &var_len);
549
1.43k
  if (var_len == -1) {
550
12
    return false;
551
12
  }
552
553
1.42k
  tmp_len += var_len;
554
1.42k
  MI->wasm_data.brtable.length = length;
555
1.42k
  if (length >= UINT32_MAX - tmp_len) {
556
    // integer overflow check
557
1
    return false;
558
1
  }
559
1.42k
  if (code_len < tmp_len + length) {
560
    // safety check that we have minimum enough data to read
561
47
    return false;
562
47
  }
563
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
564
1.37k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
565
566
1.37k
  if (MI->flat_insn->detail) {
567
1.37k
    MI->flat_insn->detail->wasm.op_count = 1;
568
1.37k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
569
1.37k
    MI->flat_insn->detail->wasm.operands[0].brtable.length = MI->wasm_data.brtable.length;
570
1.37k
    MI->flat_insn->detail->wasm.operands[0].brtable.address = MI->wasm_data.brtable.address;
571
1.37k
  }
572
573
  // read data
574
6.61k
  for(i = 0; i < length; i++){
575
5.25k
    if (code_len < tmp_len) {
576
0
      return false;
577
0
    }
578
579
5.25k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
580
5.25k
    if (var_len == -1) {
581
24
      return false;
582
24
    }
583
584
5.23k
    tmp_len += var_len;
585
5.23k
  }
586
587
  // read default target
588
1.35k
  default_target = get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
589
1.35k
  if (var_len == -1) {
590
21
    return false;
591
21
  }
592
593
1.33k
  MI->wasm_data.brtable.default_target = default_target;
594
1.33k
  MI->wasm_data.type = WASM_OP_BRTABLE;
595
1.33k
  *param_size = tmp_len + var_len;
596
597
1.33k
  if (MI->flat_insn->detail) {
598
1.33k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
599
1.33k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target = MI->wasm_data.brtable.default_target;
600
1.33k
  }
601
602
1.33k
  return true;
603
1.35k
}
604
605
// input  | code : code pointer start from varint7
606
//      | code_len : start from the code pointer to the end, how long is it
607
//      | param_size : pointer of the param size
608
//      | MI : Mcinst handler in this round of disasm
609
// return   | true/false if the function successfully finished 
610
static bool read_varint7(const uint8_t *code, size_t code_len, uint16_t *param_size, MCInst *MI)
611
1.10k
{
612
1.10k
  size_t len = 0;
613
614
1.10k
  MI->wasm_data.type = WASM_OP_INT7;
615
1.10k
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
616
1.10k
  if (len == -1) {
617
0
    return false;
618
0
  }
619
620
1.10k
  if (MI->flat_insn->detail) {
621
1.10k
    MI->flat_insn->detail->wasm.op_count = 1;
622
1.10k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
623
1.10k
    MI->flat_insn->detail->wasm.operands[0].size = 1;
624
1.10k
    MI->flat_insn->detail->wasm.operands[0].int7 = MI->wasm_data.int7;
625
1.10k
  }
626
627
1.10k
  *param_size = len;
628
629
1.10k
  return true;
630
1.10k
}
631
632
bool WASM_getInstruction(csh ud, const uint8_t *code, size_t code_len,
633
    MCInst *MI, uint16_t *size, uint64_t address, void *inst_info)
634
56.5k
{
635
56.5k
  unsigned char opcode;
636
56.5k
  uint16_t param_size;
637
638
56.5k
  if (code_len == 0)
639
0
    return false;
640
641
56.5k
  opcode = code[0];
642
56.5k
  if (opcodes[opcode] == -1) {
643
    // invalid opcode
644
90
    return false;
645
90
  }
646
647
  // valid opcode
648
56.4k
  MI->address = address;
649
56.4k
  MI->OpcodePub = MI->Opcode = opcode;
650
651
56.4k
  if (MI->flat_insn->detail) {
652
56.4k
    memset(MI->flat_insn->detail, 0, offsetof(cs_detail, wasm)+sizeof(cs_wasm));
653
56.4k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
654
56.4k
  }
655
656
  // setup groups
657
56.4k
  switch(opcode) {
658
0
    default:
659
0
      return false;
660
661
358
    case WASM_INS_I32_CONST:
662
358
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
663
10
        return false;
664
10
      }
665
666
348
      if (MI->flat_insn->detail) {
667
348
        MI->flat_insn->detail->wasm.op_count = 1;
668
348
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
669
348
        MI->flat_insn->detail->groups_count++;
670
348
      }
671
672
348
      *size = param_size + 1;
673
674
348
      break;
675
676
309
    case WASM_INS_I64_CONST:
677
309
      if (code_len == 1 || !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
678
10
        return false;
679
10
      }
680
681
299
      if (MI->flat_insn->detail) {
682
299
        MI->flat_insn->detail->wasm.op_count = 1;
683
299
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
684
299
        MI->flat_insn->detail->groups_count++;
685
299
      }
686
687
299
      *size = param_size + 1;
688
689
299
      break;
690
691
223
    case WASM_INS_F32_CONST:
692
223
      if (code_len == 1 || !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
693
8
        return false;
694
8
      }
695
696
215
      if (MI->flat_insn->detail) {
697
215
        MI->flat_insn->detail->wasm.op_count = 1;
698
215
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
699
215
        MI->flat_insn->detail->groups_count++;
700
215
      }
701
702
215
      *size = param_size + 1;
703
704
215
      break;
705
706
431
    case WASM_INS_F64_CONST:
707
431
      if (code_len == 1 || !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
708
9
        return false;
709
9
      }
710
711
422
      if (MI->flat_insn->detail) {
712
422
        MI->flat_insn->detail->wasm.op_count = 1;
713
422
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
714
422
        MI->flat_insn->detail->groups_count++;
715
422
      }
716
717
422
      *size = param_size + 1;
718
719
422
      break;
720
721
206
    case WASM_INS_I32_EQZ:
722
423
    case WASM_INS_I32_EQ:
723
647
    case WASM_INS_I32_NE:
724
955
    case WASM_INS_I32_LT_S:
725
1.20k
    case WASM_INS_I32_LT_U:
726
1.40k
    case WASM_INS_I32_GT_S:
727
1.60k
    case WASM_INS_I32_GT_U:
728
1.81k
    case WASM_INS_I32_LE_S:
729
2.01k
    case WASM_INS_I32_LE_U:
730
2.36k
    case WASM_INS_I32_GE_S:
731
2.57k
    case WASM_INS_I32_GE_U:
732
2.78k
    case WASM_INS_I64_EQZ:
733
3.25k
    case WASM_INS_I64_EQ:
734
3.59k
    case WASM_INS_I64_NE:
735
3.90k
    case WASM_INS_I64_LT_S:
736
4.10k
    case WASM_INS_I64_LT_U:
737
4.32k
    case WASN_INS_I64_GT_S:
738
4.51k
    case WASM_INS_I64_GT_U:
739
4.72k
    case WASM_INS_I64_LE_S:
740
4.91k
    case WASM_INS_I64_LE_U:
741
6.63k
    case WASM_INS_I64_GE_S:
742
6.87k
    case WASM_INS_I64_GE_U:
743
7.08k
    case WASM_INS_F32_EQ:
744
7.28k
    case WASM_INS_F32_NE:
745
7.52k
    case WASM_INS_F32_LT:
746
7.72k
    case WASM_INS_F32_GT:
747
7.93k
    case WASM_INS_F32_LE:
748
8.15k
    case WASM_INS_F32_GE:
749
12.4k
    case WASM_INS_F64_EQ:
750
12.6k
    case WASM_INS_F64_NE:
751
13.0k
    case WASM_INS_F64_LT:
752
13.5k
    case WASM_INS_F64_GT:
753
13.7k
    case WASM_INS_F64_LE:
754
13.9k
    case WASM_INS_F64_GE:
755
14.2k
    case WASM_INS_I32_CLZ:
756
14.6k
    case WASM_INS_I32_CTZ:
757
14.8k
    case WASM_INS_I32_POPCNT:
758
15.0k
    case WASM_INS_I32_ADD:
759
15.2k
    case WASM_INS_I32_SUB:
760
15.4k
    case WASM_INS_I32_MUL:
761
15.7k
    case WASM_INS_I32_DIV_S:
762
15.9k
    case WASM_INS_I32_DIV_U:
763
16.1k
    case WASM_INS_I32_REM_S:
764
16.3k
    case WASM_INS_I32_REM_U:
765
16.7k
    case WASM_INS_I32_AND:
766
17.0k
    case WASM_INS_I32_OR:
767
17.3k
    case WASM_INS_I32_XOR:
768
17.5k
    case WASM_INS_I32_SHL:
769
17.7k
    case WASM_INS_I32_SHR_S:
770
18.0k
    case WASM_INS_I32_SHR_U:
771
18.2k
    case WASM_INS_I32_ROTL:
772
18.4k
    case WASM_INS_I32_ROTR:
773
18.6k
    case WASM_INS_I64_CLZ:
774
18.9k
    case WASM_INS_I64_CTZ:
775
19.1k
    case WASM_INS_I64_POPCNT:
776
19.3k
    case WASM_INS_I64_ADD:
777
19.5k
    case WASM_INS_I64_SUB:
778
19.8k
    case WASM_INS_I64_MUL:
779
20.0k
    case WASM_INS_I64_DIV_S:
780
20.8k
    case WASM_INS_I64_DIV_U:
781
21.0k
    case WASM_INS_I64_REM_S:
782
21.3k
    case WASM_INS_I64_REM_U:
783
21.5k
    case WASM_INS_I64_AND:
784
21.8k
    case WASM_INS_I64_OR:
785
22.0k
    case WASM_INS_I64_XOR:
786
22.5k
    case WASM_INS_I64_SHL:
787
22.8k
    case WASM_INS_I64_SHR_S:
788
23.1k
    case WASM_INS_I64_SHR_U:
789
23.3k
    case WASM_INS_I64_ROTL:
790
23.5k
    case WASM_INS_I64_ROTR:
791
23.7k
    case WASM_INS_F32_ABS:
792
23.9k
    case WASM_INS_F32_NEG:
793
24.5k
    case WASM_INS_F32_CEIL:
794
24.9k
    case WASM_INS_F32_FLOOR:
795
26.3k
    case WASM_INS_F32_TRUNC:
796
26.6k
    case WASM_INS_F32_NEAREST:
797
26.8k
    case WASM_INS_F32_SQRT:
798
27.3k
    case WASM_INS_F32_ADD:
799
27.6k
    case WASM_INS_F32_SUB:
800
27.8k
    case WASM_INS_F32_MUL:
801
28.0k
    case WASM_INS_F32_DIV:
802
28.2k
    case WASM_INS_F32_MIN:
803
28.4k
    case WASM_INS_F32_MAX:
804
28.6k
    case WASM_INS_F32_COPYSIGN:
805
28.8k
    case WASM_INS_F64_ABS:
806
29.0k
    case WASM_INS_F64_NEG:
807
29.3k
    case WASM_INS_F64_CEIL:
808
29.5k
    case WASM_INS_F64_FLOOR:
809
29.9k
    case WASM_INS_F64_TRUNC:
810
30.2k
    case WASM_INS_F64_NEAREST:
811
30.4k
    case WASM_INS_F64_SQRT:
812
30.6k
    case WASM_INS_F64_ADD:
813
30.8k
    case WASM_INS_F64_SUB:
814
31.0k
    case WASM_INS_F64_MUL:
815
31.2k
    case WASM_INS_F64_DIV:
816
31.7k
    case WASM_INS_F64_MIN:
817
31.9k
    case WASM_INS_F64_MAX:
818
32.1k
    case WASM_INS_F64_COPYSIGN:
819
32.3k
    case WASM_INS_I32_WARP_I64:
820
32.5k
    case WASP_INS_I32_TRUNC_S_F32:
821
32.7k
    case WASM_INS_I32_TRUNC_U_F32:
822
32.9k
    case WASM_INS_I32_TRUNC_S_F64:
823
33.1k
    case WASM_INS_I32_TRUNC_U_F64:
824
33.3k
    case WASM_INS_I64_EXTEND_S_I32:
825
33.5k
    case WASM_INS_I64_EXTEND_U_I32:
826
33.7k
    case WASM_INS_I64_TRUNC_S_F32:
827
33.9k
    case WASM_INS_I64_TRUNC_U_F32:
828
34.2k
    case WASM_INS_I64_TRUNC_S_F64:
829
34.4k
    case WASM_INS_I64_TRUNC_U_F64:
830
34.6k
    case WASM_INS_F32_CONVERT_S_I32:
831
34.8k
    case WASM_INS_F32_CONVERT_U_I32:
832
35.0k
    case WASM_INS_F32_CONVERT_S_I64:
833
35.2k
    case WASM_INS_F32_CONVERT_U_I64:
834
35.4k
    case WASM_INS_F32_DEMOTE_F64:
835
35.7k
    case WASM_INS_F64_CONVERT_S_I32:
836
35.9k
    case WASM_INS_F64_CONVERT_U_I32:
837
36.2k
    case WASM_INS_F64_CONVERT_S_I64:
838
36.4k
    case WASM_INS_F64_CONVERT_U_I64:
839
36.6k
    case WASM_INS_F64_PROMOTE_F32:
840
36.8k
    case WASM_INS_I32_REINTERPRET_F32:
841
37.0k
    case WASM_INS_I64_REINTERPRET_F64:
842
38.8k
    case WASM_INS_F32_REINTERPRET_I32:
843
39.0k
    case WASM_INS_F64_REINTERPRET_I64:
844
39.0k
      MI->wasm_data.type = WASM_OP_NONE;
845
846
39.0k
      if (MI->flat_insn->detail) {
847
39.0k
        MI->flat_insn->detail->wasm.op_count = 0;
848
39.0k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
849
39.0k
        MI->flat_insn->detail->groups_count++;
850
39.0k
      }
851
852
39.0k
      *size = 1;
853
854
39.0k
      break;
855
856
226
    case WASM_INS_DROP:
857
454
    case WASM_INS_SELECT:
858
454
      MI->wasm_data.type = WASM_OP_NONE;
859
860
454
      if (MI->flat_insn->detail) {
861
454
        MI->flat_insn->detail->wasm.op_count = 0;
862
454
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_PARAMETRIC;
863
454
        MI->flat_insn->detail->groups_count++;
864
454
      }
865
866
454
      *size = 1;
867
868
454
      break;
869
870
301
    case WASM_INS_GET_LOCAL:
871
529
    case WASM_INS_SET_LOCAL:
872
737
    case WASM_INS_TEE_LOCAL:
873
942
    case WASM_INS_GET_GLOBAL:
874
1.16k
    case WASM_INS_SET_GLOBAL:
875
1.16k
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
876
33
        return false;
877
33
      }
878
879
1.13k
      if (MI->flat_insn->detail) {
880
1.13k
        MI->flat_insn->detail->wasm.op_count = 1;
881
1.13k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_VARIABLE;
882
1.13k
        MI->flat_insn->detail->groups_count++;
883
1.13k
      }
884
885
1.13k
      *size = param_size + 1;
886
887
1.13k
      break;
888
889
463
    case WASM_INS_I32_LOAD:
890
689
    case WASM_INS_I64_LOAD:
891
888
    case WASM_INS_F32_LOAD:
892
1.16k
    case WASM_INS_F64_LOAD:
893
1.37k
    case WASM_INS_I32_LOAD8_S:
894
1.77k
    case WASM_INS_I32_LOAD8_U:
895
2.00k
    case WASM_INS_I32_LOAD16_S:
896
2.33k
    case WASM_INS_I32_LOAD16_U:
897
3.00k
    case WASM_INS_I64_LOAD8_S:
898
3.30k
    case WASM_INS_I64_LOAD8_U:
899
3.54k
    case WASM_INS_I64_LOAD16_S:
900
3.74k
    case WASM_INS_I64_LOAD16_U:
901
3.95k
    case WASM_INS_I64_LOAD32_S:
902
4.21k
    case WASM_INS_I64_LOAD32_U:
903
4.42k
    case WASM_INS_I32_STORE:
904
4.64k
    case WASM_INS_I64_STORE:
905
4.84k
    case WASM_INS_F32_STORE:
906
5.05k
    case WASM_INS_F64_STORE:
907
5.28k
    case WASM_INS_I32_STORE8:
908
5.49k
    case WASM_INS_I32_STORE16:
909
5.70k
    case WASM_INS_I64_STORE8:
910
5.92k
    case WASM_INS_I64_STORE16:
911
6.12k
    case WASM_INS_I64_STORE32:
912
6.12k
      if (code_len == 1 || !read_memoryimmediate(&code[1], code_len - 1, &param_size, MI)) {
913
208
        return false;
914
208
      }
915
916
5.91k
      if (MI->flat_insn->detail) {
917
5.91k
        MI->flat_insn->detail->wasm.op_count = 2;
918
5.91k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
919
5.91k
        MI->flat_insn->detail->groups_count++;
920
5.91k
      }
921
922
5.91k
      *size = param_size + 1;
923
924
5.91k
      break;
925
926
224
    case WASM_INS_CURRENT_MEMORY:
927
613
    case WASM_INS_GROW_MEMORY:
928
613
      MI->wasm_data.type = WASM_OP_NONE;
929
930
613
      if (MI->flat_insn->detail) {
931
613
        MI->flat_insn->detail->wasm.op_count = 0;
932
613
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
933
613
        MI->flat_insn->detail->groups_count++;
934
613
      }
935
936
613
      *size = 1;
937
938
613
      break;
939
940
3.04k
    case WASM_INS_UNREACHABLE:
941
3.29k
    case WASM_INS_NOP:
942
3.54k
    case WASM_INS_ELSE:
943
3.73k
    case WASM_INS_END:
944
4.21k
    case WASM_INS_RETURN:
945
4.21k
      MI->wasm_data.type = WASM_OP_NONE;
946
947
4.21k
      if (MI->flat_insn->detail) {
948
4.21k
        MI->flat_insn->detail->wasm.op_count = 0;
949
4.21k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
950
4.21k
        MI->flat_insn->detail->groups_count++;
951
4.21k
      }
952
953
4.21k
      *size = 1;
954
955
4.21k
      break;
956
957
428
    case WASM_INS_BLOCK:
958
649
    case WASM_INS_LOOP:
959
1.13k
    case WASM_INS_IF:
960
1.13k
      if (code_len == 1 || !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
961
28
        return false;
962
28
      }
963
964
1.10k
      if (MI->flat_insn->detail) {
965
1.10k
        MI->flat_insn->detail->wasm.op_count = 1;
966
1.10k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
967
1.10k
        MI->flat_insn->detail->groups_count++;
968
1.10k
      }
969
970
1.10k
      *size = param_size + 1;
971
972
1.10k
      break;
973
974
230
    case WASM_INS_BR:
975
458
    case WASM_INS_BR_IF:
976
676
    case WASM_INS_CALL:
977
880
    case WASM_INS_CALL_INDIRECT:
978
880
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
979
30
        return false;
980
30
      }
981
982
850
      if (MI->flat_insn->detail) {
983
850
        MI->flat_insn->detail->wasm.op_count = 1;
984
850
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
985
850
        MI->flat_insn->detail->groups_count++;
986
850
      }
987
988
850
      *size = param_size + 1;
989
990
850
      break;
991
992
1.44k
    case WASM_INS_BR_TABLE:
993
1.44k
      if (code_len == 1 || !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
994
113
        return false;
995
113
      }
996
997
1.33k
      if (MI->flat_insn->detail) {
998
1.33k
        MI->flat_insn->detail->wasm.op_count = 1;
999
1.33k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
1000
1.33k
        MI->flat_insn->detail->groups_count++;
1001
1.33k
      }
1002
1003
1.33k
      *size = param_size + 1;
1004
1005
1.33k
      break;
1006
56.4k
  }
1007
1008
55.9k
  return true;
1009
56.4k
}