Coverage Report

Created: 2023-12-08 06:05

/src/capstonenext/arch/WASM/WASMDisassembler.c
Line
Count
Source (jump to first uncovered line)
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
34.6k
{
277
34.6k
  uint32_t data = 0;
278
34.6k
  int i;
279
280
42.6k
  for(i = 0;; i++) {
281
42.6k
    if (code_len < i + 1) {
282
188
      *leng = -1;
283
188
      return 0;
284
188
    }
285
286
287
42.4k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
288
394
      *leng = -1;
289
394
      return 0;
290
394
    }
291
292
42.0k
    data = data + (((uint32_t) code[i] & 0x7f) << (i * 7));
293
42.0k
    if (code[i] >> 7 == 0) {
294
34.0k
      break;
295
34.0k
    }
296
42.0k
  }
297
298
34.0k
  *leng = i + 1;
299
300
34.0k
  return data;
301
34.6k
}
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
457
{
309
457
  uint64_t data;
310
457
  int i;
311
312
457
  data = 0;
313
816
  for(i = 0;; i++){
314
816
    if (code_len < i + 1) {
315
8
      *leng = -1;
316
8
      return 0;
317
8
    }
318
319
808
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
4
      *leng = -1;
321
4
      return 0;
322
4
    }
323
324
804
    data = data + (((uint64_t) code[i] & 0x7f) << (i * 7));
325
804
    if (code[i] >> 7 == 0) {
326
445
      break;
327
445
    }
328
804
  }
329
330
445
  *leng = i + 1;
331
332
445
  return data;
333
457
}
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
438
{
340
438
  memcpy(dest, code, 4);
341
438
}
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
624
{
348
624
  memcpy(dest, code, 8);
349
624
}
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.19k
{
357
1.19k
  int8_t data;
358
359
1.19k
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
1.19k
  *leng = 1;
365
366
1.19k
  if (code[0] == 0x40) {
367
68
    return -1;
368
68
  }
369
370
1.12k
  data = code[0] & 0x7f;
371
372
1.12k
  return data;
373
1.19k
}
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
3.74k
{
382
3.74k
  size_t len = 0;
383
3.74k
  uint32_t data;
384
385
3.74k
  data = get_varuint32(code, code_len, &len);
386
3.74k
  if (len == -1) {
387
25
    return false;
388
25
  }
389
390
3.71k
  if (MI->flat_insn->detail) {
391
3.71k
    MI->flat_insn->detail->wasm.op_count = 1;
392
3.71k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
393
3.71k
    MI->flat_insn->detail->wasm.operands[0].size= len;
394
3.71k
    MI->flat_insn->detail->wasm.operands[0].varuint32= data;
395
3.71k
  }
396
397
3.71k
  MI->wasm_data.size = len;
398
3.71k
  MI->wasm_data.type = WASM_OP_VARUINT32;
399
3.71k
  MI->wasm_data.uint32 = data;
400
3.71k
  *param_size = len;
401
402
3.71k
  return true;
403
3.74k
}
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
457
{
412
457
  size_t len = 0;
413
457
  uint64_t data;
414
415
457
  data = get_varuint64(code, code_len, &len);
416
457
  if (len == -1) {
417
12
    return false;
418
12
  }
419
420
445
  if (MI->flat_insn->detail) {
421
445
    MI->flat_insn->detail->wasm.op_count = 1;
422
445
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT64;
423
445
    MI->flat_insn->detail->wasm.operands[0].size = len;
424
445
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
425
445
  }
426
427
445
  MI->wasm_data.size = len;
428
445
  MI->wasm_data.type = WASM_OP_VARUINT64;
429
445
  MI->wasm_data.uint64 = data;
430
445
  *param_size = len;
431
432
445
  return true;
433
457
}
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
9.89k
{
442
9.89k
  size_t tmp, len = 0;
443
9.89k
  uint32_t data[2];
444
445
9.89k
  if (MI->flat_insn->detail) {
446
9.89k
    MI->flat_insn->detail->wasm.op_count = 2;
447
9.89k
  }
448
449
9.89k
  data[0] = get_varuint32(code, code_len, &tmp);
450
9.89k
  if (tmp == -1) {
451
60
    return false;
452
60
  }
453
454
9.83k
  if (MI->flat_insn->detail) {
455
9.83k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
456
9.83k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
457
9.83k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
458
9.83k
  }
459
460
9.83k
  len = tmp;
461
9.83k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
462
9.83k
  if (len == -1) {
463
0
    return false;
464
0
  }
465
466
9.83k
  if (MI->flat_insn->detail) {
467
9.83k
    MI->flat_insn->detail->wasm.operands[1].type = WASM_OP_VARUINT32;
468
9.83k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
469
9.83k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
470
9.83k
  }
471
472
9.83k
  len += tmp;
473
9.83k
  MI->wasm_data.size = len;
474
9.83k
  MI->wasm_data.type = WASM_OP_IMM;
475
9.83k
  MI->wasm_data.immediate[0] = data[0];
476
9.83k
  MI->wasm_data.immediate[1] = data[1];
477
9.83k
  *param_size = len;
478
479
9.83k
  return true;
480
9.83k
}
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
3
    return false;
491
3
  }
492
493
219
  get_uint32(code, &(MI->wasm_data.uint32));
494
495
219
  if (MI->flat_insn->detail) {
496
219
    MI->flat_insn->detail->wasm.op_count = 1;
497
219
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
498
219
    MI->flat_insn->detail->wasm.operands[0].size = 4;
499
219
    get_uint32(code, &(MI->flat_insn->detail->wasm.operands[0].uint32));
500
219
  }
501
502
219
  MI->wasm_data.size = 4;
503
219
  MI->wasm_data.type = WASM_OP_UINT32;
504
219
  *param_size = 4;
505
506
219
  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
320
{
516
320
  if (code_len < 8) {
517
8
    return false;
518
8
  }
519
520
312
  get_uint64(code, &(MI->wasm_data.uint64));
521
522
312
  if (MI->flat_insn->detail) {
523
312
    MI->flat_insn->detail->wasm.op_count = 1;
524
312
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
525
312
    MI->flat_insn->detail->wasm.operands[0].size = 8;
526
312
    get_uint64(code, &(MI->flat_insn->detail->wasm.operands[0].uint64));
527
312
  }
528
529
312
  MI->wasm_data.size = 8;
530
312
  MI->wasm_data.type = WASM_OP_UINT64;
531
312
  *param_size = 8;
532
533
312
  return true;
534
320
}
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
812
{
543
812
  uint32_t length, default_target;
544
812
  int tmp_len = 0, i;
545
812
  size_t var_len;
546
547
  // read length
548
812
  length = get_varuint32(code, code_len, &var_len);
549
812
  if (var_len == -1) {
550
10
    return false;
551
10
  }
552
553
802
  tmp_len += var_len;
554
802
  MI->wasm_data.brtable.length = length;
555
802
  if (length >= UINT32_MAX - tmp_len) {
556
    // integer overflow check
557
0
    return false;
558
0
  }
559
802
  if (code_len < tmp_len + length) {
560
    // safety check that we have minimum enough data to read
561
54
    return false;
562
54
  }
563
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
564
748
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
565
566
748
  if (MI->flat_insn->detail) {
567
748
    MI->flat_insn->detail->wasm.op_count = 1;
568
748
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
569
748
    MI->flat_insn->detail->wasm.operands[0].brtable.length = MI->wasm_data.brtable.length;
570
748
    MI->flat_insn->detail->wasm.operands[0].brtable.address = MI->wasm_data.brtable.address;
571
748
  }
572
573
  // read data
574
10.3k
  for(i = 0; i < length; i++){
575
9.60k
    if (code_len < tmp_len) {
576
0
      return false;
577
0
    }
578
579
9.60k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
580
9.60k
    if (var_len == -1) {
581
16
      return false;
582
16
    }
583
584
9.58k
    tmp_len += var_len;
585
9.58k
  }
586
587
  // read default target
588
732
  default_target = get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
589
732
  if (var_len == -1) {
590
11
    return false;
591
11
  }
592
593
721
  MI->wasm_data.brtable.default_target = default_target;
594
721
  MI->wasm_data.type = WASM_OP_BRTABLE;
595
721
  *param_size = tmp_len + var_len;
596
597
721
  if (MI->flat_insn->detail) {
598
721
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
599
721
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target = MI->wasm_data.brtable.default_target;
600
721
  }
601
602
721
  return true;
603
732
}
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.19k
{
612
1.19k
  size_t len = 0;
613
614
1.19k
  MI->wasm_data.type = WASM_OP_INT7;
615
1.19k
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
616
1.19k
  if (len == -1) {
617
0
    return false;
618
0
  }
619
620
1.19k
  if (MI->flat_insn->detail) {
621
1.19k
    MI->flat_insn->detail->wasm.op_count = 1;
622
1.19k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
623
1.19k
    MI->flat_insn->detail->wasm.operands[0].size = 1;
624
1.19k
    MI->flat_insn->detail->wasm.operands[0].int7 = MI->wasm_data.int7;
625
1.19k
  }
626
627
1.19k
  *param_size = len;
628
629
1.19k
  return true;
630
1.19k
}
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
75.8k
{
635
75.8k
  unsigned char opcode;
636
75.8k
  uint16_t param_size;
637
638
75.8k
  if (code_len == 0)
639
0
    return false;
640
641
75.8k
  opcode = code[0];
642
75.8k
  if (opcodes[opcode] == -1) {
643
    // invalid opcode
644
332
    return false;
645
332
  }
646
647
  // valid opcode
648
75.5k
  MI->address = address;
649
75.5k
  MI->OpcodePub = MI->Opcode = opcode;
650
651
75.5k
  if (MI->flat_insn->detail) {
652
75.5k
    memset(MI->flat_insn->detail, 0, offsetof(cs_detail, wasm)+sizeof(cs_wasm));
653
75.5k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
654
75.5k
  }
655
656
  // setup groups
657
75.5k
  switch(opcode) {
658
0
    default:
659
0
      return false;
660
661
384
    case WASM_INS_I32_CONST:
662
384
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
663
8
        return false;
664
8
      }
665
666
376
      if (MI->flat_insn->detail) {
667
376
        MI->flat_insn->detail->wasm.op_count = 1;
668
376
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
669
376
        MI->flat_insn->detail->groups_count++;
670
376
      }
671
672
376
      *size = param_size + 1;
673
674
376
      break;
675
676
460
    case WASM_INS_I64_CONST:
677
460
      if (code_len == 1 || !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
678
15
        return false;
679
15
      }
680
681
445
      if (MI->flat_insn->detail) {
682
445
        MI->flat_insn->detail->wasm.op_count = 1;
683
445
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
684
445
        MI->flat_insn->detail->groups_count++;
685
445
      }
686
687
445
      *size = param_size + 1;
688
689
445
      break;
690
691
224
    case WASM_INS_F32_CONST:
692
224
      if (code_len == 1 || !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
693
5
        return false;
694
5
      }
695
696
219
      if (MI->flat_insn->detail) {
697
219
        MI->flat_insn->detail->wasm.op_count = 1;
698
219
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
699
219
        MI->flat_insn->detail->groups_count++;
700
219
      }
701
702
219
      *size = param_size + 1;
703
704
219
      break;
705
706
324
    case WASM_INS_F64_CONST:
707
324
      if (code_len == 1 || !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
708
12
        return false;
709
12
      }
710
711
312
      if (MI->flat_insn->detail) {
712
312
        MI->flat_insn->detail->wasm.op_count = 1;
713
312
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
714
312
        MI->flat_insn->detail->groups_count++;
715
312
      }
716
717
312
      *size = param_size + 1;
718
719
312
      break;
720
721
278
    case WASM_INS_I32_EQZ:
722
509
    case WASM_INS_I32_EQ:
723
733
    case WASM_INS_I32_NE:
724
1.06k
    case WASM_INS_I32_LT_S:
725
1.87k
    case WASM_INS_I32_LT_U:
726
2.22k
    case WASM_INS_I32_GT_S:
727
2.45k
    case WASM_INS_I32_GT_U:
728
2.85k
    case WASM_INS_I32_LE_S:
729
3.19k
    case WASM_INS_I32_LE_U:
730
3.45k
    case WASM_INS_I32_GE_S:
731
3.70k
    case WASM_INS_I32_GE_U:
732
4.09k
    case WASM_INS_I64_EQZ:
733
4.42k
    case WASM_INS_I64_EQ:
734
4.68k
    case WASM_INS_I64_NE:
735
5.42k
    case WASM_INS_I64_LT_S:
736
5.65k
    case WASM_INS_I64_LT_U:
737
5.89k
    case WASN_INS_I64_GT_S:
738
6.15k
    case WASM_INS_I64_GT_U:
739
6.39k
    case WASM_INS_I64_LE_S:
740
6.66k
    case WASM_INS_I64_LE_U:
741
6.90k
    case WASM_INS_I64_GE_S:
742
7.16k
    case WASM_INS_I64_GE_U:
743
7.42k
    case WASM_INS_F32_EQ:
744
7.69k
    case WASM_INS_F32_NE:
745
8.05k
    case WASM_INS_F32_LT:
746
8.27k
    case WASM_INS_F32_GT:
747
8.53k
    case WASM_INS_F32_LE:
748
8.91k
    case WASM_INS_F32_GE:
749
9.38k
    case WASM_INS_F64_EQ:
750
10.0k
    case WASM_INS_F64_NE:
751
10.9k
    case WASM_INS_F64_LT:
752
11.2k
    case WASM_INS_F64_GT:
753
11.7k
    case WASM_INS_F64_LE:
754
12.0k
    case WASM_INS_F64_GE:
755
12.2k
    case WASM_INS_I32_CLZ:
756
13.0k
    case WASM_INS_I32_CTZ:
757
13.5k
    case WASM_INS_I32_POPCNT:
758
13.8k
    case WASM_INS_I32_ADD:
759
14.0k
    case WASM_INS_I32_SUB:
760
14.3k
    case WASM_INS_I32_MUL:
761
14.7k
    case WASM_INS_I32_DIV_S:
762
15.1k
    case WASM_INS_I32_DIV_U:
763
15.5k
    case WASM_INS_I32_REM_S:
764
15.8k
    case WASM_INS_I32_REM_U:
765
16.1k
    case WASM_INS_I32_AND:
766
16.5k
    case WASM_INS_I32_OR:
767
16.9k
    case WASM_INS_I32_XOR:
768
17.2k
    case WASM_INS_I32_SHL:
769
17.5k
    case WASM_INS_I32_SHR_S:
770
17.8k
    case WASM_INS_I32_SHR_U:
771
18.1k
    case WASM_INS_I32_ROTL:
772
18.8k
    case WASM_INS_I32_ROTR:
773
19.1k
    case WASM_INS_I64_CLZ:
774
19.4k
    case WASM_INS_I64_CTZ:
775
19.8k
    case WASM_INS_I64_POPCNT:
776
20.1k
    case WASM_INS_I64_ADD:
777
20.4k
    case WASM_INS_I64_SUB:
778
20.8k
    case WASM_INS_I64_MUL:
779
21.2k
    case WASM_INS_I64_DIV_S:
780
22.3k
    case WASM_INS_I64_DIV_U:
781
24.0k
    case WASM_INS_I64_REM_S:
782
24.2k
    case WASM_INS_I64_REM_U:
783
24.6k
    case WASM_INS_I64_AND:
784
24.8k
    case WASM_INS_I64_OR:
785
25.1k
    case WASM_INS_I64_XOR:
786
25.4k
    case WASM_INS_I64_SHL:
787
25.7k
    case WASM_INS_I64_SHR_S:
788
25.9k
    case WASM_INS_I64_SHR_U:
789
26.2k
    case WASM_INS_I64_ROTL:
790
26.6k
    case WASM_INS_I64_ROTR:
791
27.0k
    case WASM_INS_F32_ABS:
792
27.4k
    case WASM_INS_F32_NEG:
793
27.8k
    case WASM_INS_F32_CEIL:
794
28.2k
    case WASM_INS_F32_FLOOR:
795
28.4k
    case WASM_INS_F32_TRUNC:
796
28.7k
    case WASM_INS_F32_NEAREST:
797
28.9k
    case WASM_INS_F32_SQRT:
798
29.3k
    case WASM_INS_F32_ADD:
799
31.8k
    case WASM_INS_F32_SUB:
800
32.1k
    case WASM_INS_F32_MUL:
801
32.3k
    case WASM_INS_F32_DIV:
802
32.8k
    case WASM_INS_F32_MIN:
803
33.0k
    case WASM_INS_F32_MAX:
804
33.3k
    case WASM_INS_F32_COPYSIGN:
805
33.9k
    case WASM_INS_F64_ABS:
806
34.2k
    case WASM_INS_F64_NEG:
807
34.9k
    case WASM_INS_F64_CEIL:
808
35.3k
    case WASM_INS_F64_FLOOR:
809
35.6k
    case WASM_INS_F64_TRUNC:
810
35.9k
    case WASM_INS_F64_NEAREST:
811
36.3k
    case WASM_INS_F64_SQRT:
812
36.6k
    case WASM_INS_F64_ADD:
813
37.0k
    case WASM_INS_F64_SUB:
814
37.3k
    case WASM_INS_F64_MUL:
815
37.6k
    case WASM_INS_F64_DIV:
816
38.3k
    case WASM_INS_F64_MIN:
817
38.9k
    case WASM_INS_F64_MAX:
818
39.2k
    case WASM_INS_F64_COPYSIGN:
819
39.5k
    case WASM_INS_I32_WARP_I64:
820
39.9k
    case WASP_INS_I32_TRUNC_S_F32:
821
40.1k
    case WASM_INS_I32_TRUNC_U_F32:
822
40.4k
    case WASM_INS_I32_TRUNC_S_F64:
823
40.8k
    case WASM_INS_I32_TRUNC_U_F64:
824
41.1k
    case WASM_INS_I64_EXTEND_S_I32:
825
41.4k
    case WASM_INS_I64_EXTEND_U_I32:
826
41.7k
    case WASM_INS_I64_TRUNC_S_F32:
827
41.9k
    case WASM_INS_I64_TRUNC_U_F32:
828
42.3k
    case WASM_INS_I64_TRUNC_S_F64:
829
42.5k
    case WASM_INS_I64_TRUNC_U_F64:
830
42.8k
    case WASM_INS_F32_CONVERT_S_I32:
831
43.1k
    case WASM_INS_F32_CONVERT_U_I32:
832
43.4k
    case WASM_INS_F32_CONVERT_S_I64:
833
43.9k
    case WASM_INS_F32_CONVERT_U_I64:
834
44.2k
    case WASM_INS_F32_DEMOTE_F64:
835
44.7k
    case WASM_INS_F64_CONVERT_S_I32:
836
45.0k
    case WASM_INS_F64_CONVERT_U_I32:
837
45.3k
    case WASM_INS_F64_CONVERT_S_I64:
838
45.5k
    case WASM_INS_F64_CONVERT_U_I64:
839
45.8k
    case WASM_INS_F64_PROMOTE_F32:
840
46.1k
    case WASM_INS_I32_REINTERPRET_F32:
841
46.5k
    case WASM_INS_I64_REINTERPRET_F64:
842
46.8k
    case WASM_INS_F32_REINTERPRET_I32:
843
47.2k
    case WASM_INS_F64_REINTERPRET_I64:
844
47.2k
      MI->wasm_data.type = WASM_OP_NONE;
845
846
47.2k
      if (MI->flat_insn->detail) {
847
47.2k
        MI->flat_insn->detail->wasm.op_count = 0;
848
47.2k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
849
47.2k
        MI->flat_insn->detail->groups_count++;
850
47.2k
      }
851
852
47.2k
      *size = 1;
853
854
47.2k
      break;
855
856
587
    case WASM_INS_DROP:
857
1.17k
    case WASM_INS_SELECT:
858
1.17k
      MI->wasm_data.type = WASM_OP_NONE;
859
860
1.17k
      if (MI->flat_insn->detail) {
861
1.17k
        MI->flat_insn->detail->wasm.op_count = 0;
862
1.17k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_PARAMETRIC;
863
1.17k
        MI->flat_insn->detail->groups_count++;
864
1.17k
      }
865
866
1.17k
      *size = 1;
867
868
1.17k
      break;
869
870
560
    case WASM_INS_GET_LOCAL:
871
925
    case WASM_INS_SET_LOCAL:
872
1.16k
    case WASM_INS_TEE_LOCAL:
873
1.56k
    case WASM_INS_GET_GLOBAL:
874
2.33k
    case WASM_INS_SET_GLOBAL:
875
2.33k
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
876
48
        return false;
877
48
      }
878
879
2.28k
      if (MI->flat_insn->detail) {
880
2.28k
        MI->flat_insn->detail->wasm.op_count = 1;
881
2.28k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_VARIABLE;
882
2.28k
        MI->flat_insn->detail->groups_count++;
883
2.28k
      }
884
885
2.28k
      *size = param_size + 1;
886
887
2.28k
      break;
888
889
133
    case WASM_INS_I32_LOAD:
890
1.19k
    case WASM_INS_I64_LOAD:
891
1.54k
    case WASM_INS_F32_LOAD:
892
1.85k
    case WASM_INS_F64_LOAD:
893
2.41k
    case WASM_INS_I32_LOAD8_S:
894
3.00k
    case WASM_INS_I32_LOAD8_U:
895
3.27k
    case WASM_INS_I32_LOAD16_S:
896
3.57k
    case WASM_INS_I32_LOAD16_U:
897
4.19k
    case WASM_INS_I64_LOAD8_S:
898
5.10k
    case WASM_INS_I64_LOAD8_U:
899
5.42k
    case WASM_INS_I64_LOAD16_S:
900
5.74k
    case WASM_INS_I64_LOAD16_U:
901
6.08k
    case WASM_INS_I64_LOAD32_S:
902
6.49k
    case WASM_INS_I64_LOAD32_U:
903
6.84k
    case WASM_INS_I32_STORE:
904
7.11k
    case WASM_INS_I64_STORE:
905
8.05k
    case WASM_INS_F32_STORE:
906
8.31k
    case WASM_INS_F64_STORE:
907
8.84k
    case WASM_INS_I32_STORE8:
908
9.43k
    case WASM_INS_I32_STORE16:
909
9.56k
    case WASM_INS_I64_STORE8:
910
9.68k
    case WASM_INS_I64_STORE16:
911
10.0k
    case WASM_INS_I64_STORE32:
912
10.0k
      if (code_len == 1 || !read_memoryimmediate(&code[1], code_len - 1, &param_size, MI)) {
913
188
        return false;
914
188
      }
915
916
9.83k
      if (MI->flat_insn->detail) {
917
9.83k
        MI->flat_insn->detail->wasm.op_count = 2;
918
9.83k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
919
9.83k
        MI->flat_insn->detail->groups_count++;
920
9.83k
      }
921
922
9.83k
      *size = param_size + 1;
923
924
9.83k
      break;
925
926
319
    case WASM_INS_CURRENT_MEMORY:
927
770
    case WASM_INS_GROW_MEMORY:
928
770
      MI->wasm_data.type = WASM_OP_NONE;
929
930
770
      if (MI->flat_insn->detail) {
931
770
        MI->flat_insn->detail->wasm.op_count = 0;
932
770
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
933
770
        MI->flat_insn->detail->groups_count++;
934
770
      }
935
936
770
      *size = 1;
937
938
770
      break;
939
940
6.57k
    case WASM_INS_UNREACHABLE:
941
7.20k
    case WASM_INS_NOP:
942
7.62k
    case WASM_INS_ELSE:
943
9.20k
    case WASM_INS_END:
944
9.43k
    case WASM_INS_RETURN:
945
9.43k
      MI->wasm_data.type = WASM_OP_NONE;
946
947
9.43k
      if (MI->flat_insn->detail) {
948
9.43k
        MI->flat_insn->detail->wasm.op_count = 0;
949
9.43k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
950
9.43k
        MI->flat_insn->detail->groups_count++;
951
9.43k
      }
952
953
9.43k
      *size = 1;
954
955
9.43k
      break;
956
957
472
    case WASM_INS_BLOCK:
958
766
    case WASM_INS_LOOP:
959
1.20k
    case WASM_INS_IF:
960
1.20k
      if (code_len == 1 || !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
961
13
        return false;
962
13
      }
963
964
1.19k
      if (MI->flat_insn->detail) {
965
1.19k
        MI->flat_insn->detail->wasm.op_count = 1;
966
1.19k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
967
1.19k
        MI->flat_insn->detail->groups_count++;
968
1.19k
      }
969
970
1.19k
      *size = param_size + 1;
971
972
1.19k
      break;
973
974
262
    case WASM_INS_BR:
975
506
    case WASM_INS_BR_IF:
976
819
    case WASM_INS_CALL:
977
1.08k
    case WASM_INS_CALL_INDIRECT:
978
1.08k
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
979
31
        return false;
980
31
      }
981
982
1.05k
      if (MI->flat_insn->detail) {
983
1.05k
        MI->flat_insn->detail->wasm.op_count = 1;
984
1.05k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
985
1.05k
        MI->flat_insn->detail->groups_count++;
986
1.05k
      }
987
988
1.05k
      *size = param_size + 1;
989
990
1.05k
      break;
991
992
818
    case WASM_INS_BR_TABLE:
993
818
      if (code_len == 1 || !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
994
97
        return false;
995
97
      }
996
997
721
      if (MI->flat_insn->detail) {
998
721
        MI->flat_insn->detail->wasm.op_count = 1;
999
721
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
1000
721
        MI->flat_insn->detail->groups_count++;
1001
721
      }
1002
1003
721
      *size = param_size + 1;
1004
1005
721
      break;
1006
75.5k
  }
1007
1008
75.1k
  return true;
1009
75.5k
}