Coverage Report

Created: 2023-12-08 06:05

/src/capstonev5/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
58.9k
{
277
58.9k
  uint32_t data = 0;
278
58.9k
  int i;
279
280
75.1k
  for(i = 0;; i++) {
281
75.1k
    if (code_len < i + 1) {
282
431
      *leng = -1;
283
431
      return 0;
284
431
    }
285
286
287
74.6k
    if (i > 4 || (i == 4 && (code[i] & 0x7f) > 0x0f)) {
288
701
      *leng = -1;
289
701
      return 0;
290
701
    }
291
292
73.9k
    data = data + (((uint32_t) code[i] & 0x7f) << (i * 7));
293
73.9k
    if (code[i] >> 7 == 0) {
294
57.8k
      break;
295
57.8k
    }
296
73.9k
  }
297
298
57.8k
  *leng = i + 1;
299
300
57.8k
  return data;
301
58.9k
}
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
812
{
309
812
  uint64_t data;
310
812
  int i;
311
312
812
  data = 0;
313
2.15k
  for(i = 0;; i++){
314
2.15k
    if (code_len < i + 1) {
315
16
      *leng = -1;
316
16
      return 0;
317
16
    }
318
319
2.13k
    if (i > 9 || (i == 9 && (code[i] & 0x7f) > 0x01)) {
320
6
      *leng = -1;
321
6
      return 0;
322
6
    }
323
324
2.13k
    data = data + (((uint64_t) code[i] & 0x7f) << (i * 7));
325
2.13k
    if (code[i] >> 7 == 0) {
326
790
      break;
327
790
    }
328
2.13k
  }
329
330
790
  *leng = i + 1;
331
332
790
  return data;
333
812
}
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
916
{
340
916
  memcpy(dest, code, 4);
341
916
}
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
1.17k
{
348
1.17k
  memcpy(dest, code, 8);
349
1.17k
}
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
2.09k
{
357
2.09k
  int8_t data;
358
359
2.09k
  if (code_len < 1) {
360
0
    *leng = -1;
361
0
    return -1;
362
0
  }
363
364
2.09k
  *leng = 1;
365
366
2.09k
  if (code[0] == 0x40) {
367
263
    return -1;
368
263
  }
369
370
1.83k
  data = code[0] & 0x7f;
371
372
1.83k
  return data;
373
2.09k
}
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
6.27k
{
382
6.27k
  size_t len = 0;
383
6.27k
  uint32_t data;
384
385
6.27k
  data = get_varuint32(code, code_len, &len);
386
6.27k
  if (len == -1) {
387
44
    return false;
388
44
  }
389
390
6.22k
  if (MI->flat_insn->detail) {
391
6.22k
    MI->flat_insn->detail->wasm.op_count = 1;
392
6.22k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
393
6.22k
    MI->flat_insn->detail->wasm.operands[0].size= len;
394
6.22k
    MI->flat_insn->detail->wasm.operands[0].varuint32= data;
395
6.22k
  }
396
397
6.22k
  MI->wasm_data.size = len;
398
6.22k
  MI->wasm_data.type = WASM_OP_VARUINT32;
399
6.22k
  MI->wasm_data.uint32 = data;
400
6.22k
  *param_size = len;
401
402
6.22k
  return true;
403
6.27k
}
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
812
{
412
812
  size_t len = 0;
413
812
  uint64_t data;
414
415
812
  data = get_varuint64(code, code_len, &len);
416
812
  if (len == -1) {
417
22
    return false;
418
22
  }
419
420
790
  if (MI->flat_insn->detail) {
421
790
    MI->flat_insn->detail->wasm.op_count = 1;
422
790
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT64;
423
790
    MI->flat_insn->detail->wasm.operands[0].size = len;
424
790
    MI->flat_insn->detail->wasm.operands[0].varuint64 = data;
425
790
  }
426
427
790
  MI->wasm_data.size = len;
428
790
  MI->wasm_data.type = WASM_OP_VARUINT64;
429
790
  MI->wasm_data.uint64 = data;
430
790
  *param_size = len;
431
432
790
  return true;
433
812
}
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
15.8k
{
442
15.8k
  size_t tmp, len = 0;
443
15.8k
  uint32_t data[2];
444
445
15.8k
  if (MI->flat_insn->detail) {
446
15.8k
    MI->flat_insn->detail->wasm.op_count = 2;
447
15.8k
  }
448
449
15.8k
  data[0] = get_varuint32(code, code_len, &tmp);
450
15.8k
  if (tmp == -1) {
451
114
    return false;
452
114
  }
453
454
15.7k
  if (MI->flat_insn->detail) {
455
15.7k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_VARUINT32;
456
15.7k
    MI->flat_insn->detail->wasm.operands[0].size = tmp;
457
15.7k
    MI->flat_insn->detail->wasm.operands[0].varuint32 = data[0];
458
15.7k
  }
459
460
15.7k
  len = tmp;
461
15.7k
  data[1] = get_varuint32(&code[len], code_len - len, &tmp);
462
15.7k
  if (len == -1) {
463
0
    return false;
464
0
  }
465
466
15.7k
  if (MI->flat_insn->detail) {
467
15.7k
    MI->flat_insn->detail->wasm.operands[1].type = WASM_OP_VARUINT32;
468
15.7k
    MI->flat_insn->detail->wasm.operands[1].size = tmp;
469
15.7k
    MI->flat_insn->detail->wasm.operands[1].varuint32 = data[1];
470
15.7k
  }
471
472
15.7k
  len += tmp;
473
15.7k
  MI->wasm_data.size = len;
474
15.7k
  MI->wasm_data.type = WASM_OP_IMM;
475
15.7k
  MI->wasm_data.immediate[0] = data[0];
476
15.7k
  MI->wasm_data.immediate[1] = data[1];
477
15.7k
  *param_size = len;
478
479
15.7k
  return true;
480
15.7k
}
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
466
{
489
466
  if (code_len < 4) {
490
8
    return false;
491
8
  }
492
493
458
  get_uint32(code, &(MI->wasm_data.uint32));
494
495
458
  if (MI->flat_insn->detail) {
496
458
    MI->flat_insn->detail->wasm.op_count = 1;
497
458
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT32;
498
458
    MI->flat_insn->detail->wasm.operands[0].size = 4;
499
458
    get_uint32(code, &(MI->flat_insn->detail->wasm.operands[0].uint32));
500
458
  }
501
502
458
  MI->wasm_data.size = 4;
503
458
  MI->wasm_data.type = WASM_OP_UINT32;
504
458
  *param_size = 4;
505
506
458
  return true;
507
466
}
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
604
{
516
604
  if (code_len < 8) {
517
16
    return false;
518
16
  }
519
520
588
  get_uint64(code, &(MI->wasm_data.uint64));
521
522
588
  if (MI->flat_insn->detail) {
523
588
    MI->flat_insn->detail->wasm.op_count = 1;
524
588
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_UINT64;
525
588
    MI->flat_insn->detail->wasm.operands[0].size = 8;
526
588
    get_uint64(code, &(MI->flat_insn->detail->wasm.operands[0].uint64));
527
588
  }
528
529
588
  MI->wasm_data.size = 8;
530
588
  MI->wasm_data.type = WASM_OP_UINT64;
531
588
  *param_size = 8;
532
533
588
  return true;
534
604
}
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.91k
{
543
1.91k
  uint32_t length, default_target;
544
1.91k
  int tmp_len = 0, i;
545
1.91k
  size_t var_len;
546
547
  // read length
548
1.91k
  length = get_varuint32(code, code_len, &var_len);
549
1.91k
  if (var_len == -1) {
550
22
    return false;
551
22
  }
552
553
1.89k
  tmp_len += var_len;
554
1.89k
  MI->wasm_data.brtable.length = length;
555
1.89k
  if (length >= UINT32_MAX - tmp_len) {
556
    // integer overflow check
557
1
    return false;
558
1
  }
559
1.89k
  if (code_len < tmp_len + length) {
560
    // safety check that we have minimum enough data to read
561
103
    return false;
562
103
  }
563
  // base address + 1 byte opcode + tmp_len for number of cases = start of targets
564
1.79k
  MI->wasm_data.brtable.address = MI->address + 1 + tmp_len;
565
566
1.79k
  if (MI->flat_insn->detail) {
567
1.79k
    MI->flat_insn->detail->wasm.op_count = 1;
568
1.79k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_BRTABLE;
569
1.79k
    MI->flat_insn->detail->wasm.operands[0].brtable.length = MI->wasm_data.brtable.length;
570
1.79k
    MI->flat_insn->detail->wasm.operands[0].brtable.address = MI->wasm_data.brtable.address;
571
1.79k
  }
572
573
  // read data
574
19.1k
  for(i = 0; i < length; i++){
575
17.4k
    if (code_len < tmp_len) {
576
0
      return false;
577
0
    }
578
579
17.4k
    get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
580
17.4k
    if (var_len == -1) {
581
42
      return false;
582
42
    }
583
584
17.3k
    tmp_len += var_len;
585
17.3k
  }
586
587
  // read default target
588
1.75k
  default_target = get_varuint32(code + tmp_len, code_len - tmp_len, &var_len);
589
1.75k
  if (var_len == -1) {
590
26
    return false;
591
26
  }
592
593
1.72k
  MI->wasm_data.brtable.default_target = default_target;
594
1.72k
  MI->wasm_data.type = WASM_OP_BRTABLE;
595
1.72k
  *param_size = tmp_len + var_len;
596
597
1.72k
  if (MI->flat_insn->detail) {
598
1.72k
    MI->flat_insn->detail->wasm.operands[0].size = *param_size;
599
1.72k
    MI->flat_insn->detail->wasm.operands[0].brtable.default_target = MI->wasm_data.brtable.default_target;
600
1.72k
  }
601
602
1.72k
  return true;
603
1.75k
}
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
2.09k
{
612
2.09k
  size_t len = 0;
613
614
2.09k
  MI->wasm_data.type = WASM_OP_INT7;
615
2.09k
  MI->wasm_data.int7 = get_varint7(code, code_len, &len);
616
2.09k
  if (len == -1) {
617
0
    return false;
618
0
  }
619
620
2.09k
  if (MI->flat_insn->detail) {
621
2.09k
    MI->flat_insn->detail->wasm.op_count = 1;
622
2.09k
    MI->flat_insn->detail->wasm.operands[0].type = WASM_OP_INT7;
623
2.09k
    MI->flat_insn->detail->wasm.operands[0].size = 1;
624
2.09k
    MI->flat_insn->detail->wasm.operands[0].int7 = MI->wasm_data.int7;
625
2.09k
  }
626
627
2.09k
  *param_size = len;
628
629
2.09k
  return true;
630
2.09k
}
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
144k
{
635
144k
  unsigned char opcode;
636
144k
  uint16_t param_size;
637
638
144k
  if (code_len == 0)
639
0
    return false;
640
641
144k
  opcode = code[0];
642
144k
  if (opcodes[opcode] == -1) {
643
    // invalid opcode
644
460
    return false;
645
460
  }
646
647
  // valid opcode
648
144k
  MI->address = address;
649
144k
  MI->OpcodePub = MI->Opcode = opcode;
650
651
144k
  if (MI->flat_insn->detail) {
652
144k
    memset(MI->flat_insn->detail, 0, offsetof(cs_detail, wasm)+sizeof(cs_wasm));
653
144k
    WASM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
654
144k
  }
655
656
  // setup groups
657
144k
  switch(opcode) {
658
0
    default:
659
0
      return false;
660
661
606
    case WASM_INS_I32_CONST:
662
606
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
663
13
        return false;
664
13
      }
665
666
593
      if (MI->flat_insn->detail) {
667
593
        MI->flat_insn->detail->wasm.op_count = 1;
668
593
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
669
593
        MI->flat_insn->detail->groups_count++;
670
593
      }
671
672
593
      *size = param_size + 1;
673
674
593
      break;
675
676
816
    case WASM_INS_I64_CONST:
677
816
      if (code_len == 1 || !read_varuint64(&code[1], code_len - 1, &param_size, MI)) {
678
26
        return false;
679
26
      }
680
681
790
      if (MI->flat_insn->detail) {
682
790
        MI->flat_insn->detail->wasm.op_count = 1;
683
790
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
684
790
        MI->flat_insn->detail->groups_count++;
685
790
      }
686
687
790
      *size = param_size + 1;
688
689
790
      break;
690
691
471
    case WASM_INS_F32_CONST:
692
471
      if (code_len == 1 || !read_uint32(&code[1], code_len - 1, &param_size, MI)) {
693
13
        return false;
694
13
      }
695
696
458
      if (MI->flat_insn->detail) {
697
458
        MI->flat_insn->detail->wasm.op_count = 1;
698
458
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
699
458
        MI->flat_insn->detail->groups_count++;
700
458
      }
701
702
458
      *size = param_size + 1;
703
704
458
      break;
705
706
611
    case WASM_INS_F64_CONST:
707
611
      if (code_len == 1 || !read_uint64(&code[1], code_len - 1, &param_size, MI)) {
708
23
        return false;
709
23
      }
710
711
588
      if (MI->flat_insn->detail) {
712
588
        MI->flat_insn->detail->wasm.op_count = 1;
713
588
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
714
588
        MI->flat_insn->detail->groups_count++;
715
588
      }
716
717
588
      *size = param_size + 1;
718
719
588
      break;
720
721
486
    case WASM_INS_I32_EQZ:
722
4.99k
    case WASM_INS_I32_EQ:
723
5.42k
    case WASM_INS_I32_NE:
724
5.95k
    case WASM_INS_I32_LT_S:
725
7.12k
    case WASM_INS_I32_LT_U:
726
7.67k
    case WASM_INS_I32_GT_S:
727
8.10k
    case WASM_INS_I32_GT_U:
728
8.73k
    case WASM_INS_I32_LE_S:
729
9.42k
    case WASM_INS_I32_LE_U:
730
9.88k
    case WASM_INS_I32_GE_S:
731
10.4k
    case WASM_INS_I32_GE_U:
732
11.0k
    case WASM_INS_I64_EQZ:
733
11.5k
    case WASM_INS_I64_EQ:
734
12.3k
    case WASM_INS_I64_NE:
735
13.3k
    case WASM_INS_I64_LT_S:
736
14.1k
    case WASM_INS_I64_LT_U:
737
14.7k
    case WASN_INS_I64_GT_S:
738
15.2k
    case WASM_INS_I64_GT_U:
739
15.7k
    case WASM_INS_I64_LE_S:
740
16.2k
    case WASM_INS_I64_LE_U:
741
16.7k
    case WASM_INS_I64_GE_S:
742
17.2k
    case WASM_INS_I64_GE_U:
743
17.7k
    case WASM_INS_F32_EQ:
744
19.4k
    case WASM_INS_F32_NE:
745
20.1k
    case WASM_INS_F32_LT:
746
20.6k
    case WASM_INS_F32_GT:
747
21.1k
    case WASM_INS_F32_LE:
748
21.7k
    case WASM_INS_F32_GE:
749
22.4k
    case WASM_INS_F64_EQ:
750
23.5k
    case WASM_INS_F64_NE:
751
28.6k
    case WASM_INS_F64_LT:
752
29.2k
    case WASM_INS_F64_GT:
753
29.9k
    case WASM_INS_F64_LE:
754
30.5k
    case WASM_INS_F64_GE:
755
30.9k
    case WASM_INS_I32_CLZ:
756
32.0k
    case WASM_INS_I32_CTZ:
757
32.9k
    case WASM_INS_I32_POPCNT:
758
33.8k
    case WASM_INS_I32_ADD:
759
34.3k
    case WASM_INS_I32_SUB:
760
34.8k
    case WASM_INS_I32_MUL:
761
35.5k
    case WASM_INS_I32_DIV_S:
762
36.1k
    case WASM_INS_I32_DIV_U:
763
36.7k
    case WASM_INS_I32_REM_S:
764
37.3k
    case WASM_INS_I32_REM_U:
765
37.9k
    case WASM_INS_I32_AND:
766
38.7k
    case WASM_INS_I32_OR:
767
39.5k
    case WASM_INS_I32_XOR:
768
40.1k
    case WASM_INS_I32_SHL:
769
40.7k
    case WASM_INS_I32_SHR_S:
770
41.3k
    case WASM_INS_I32_SHR_U:
771
41.9k
    case WASM_INS_I32_ROTL:
772
43.1k
    case WASM_INS_I32_ROTR:
773
43.6k
    case WASM_INS_I64_CLZ:
774
44.3k
    case WASM_INS_I64_CTZ:
775
46.9k
    case WASM_INS_I64_POPCNT:
776
47.4k
    case WASM_INS_I64_ADD:
777
48.0k
    case WASM_INS_I64_SUB:
778
48.8k
    case WASM_INS_I64_MUL:
779
49.4k
    case WASM_INS_I64_DIV_S:
780
50.9k
    case WASM_INS_I64_DIV_U:
781
54.8k
    case WASM_INS_I64_REM_S:
782
55.3k
    case WASM_INS_I64_REM_U:
783
56.0k
    case WASM_INS_I64_AND:
784
56.4k
    case WASM_INS_I64_OR:
785
56.9k
    case WASM_INS_I64_XOR:
786
57.5k
    case WASM_INS_I64_SHL:
787
58.0k
    case WASM_INS_I64_SHR_S:
788
58.8k
    case WASM_INS_I64_SHR_U:
789
59.3k
    case WASM_INS_I64_ROTL:
790
60.0k
    case WASM_INS_I64_ROTR:
791
60.5k
    case WASM_INS_F32_ABS:
792
62.5k
    case WASM_INS_F32_NEG:
793
63.1k
    case WASM_INS_F32_CEIL:
794
63.7k
    case WASM_INS_F32_FLOOR:
795
64.2k
    case WASM_INS_F32_TRUNC:
796
64.6k
    case WASM_INS_F32_NEAREST:
797
65.4k
    case WASM_INS_F32_SQRT:
798
66.0k
    case WASM_INS_F32_ADD:
799
68.6k
    case WASM_INS_F32_SUB:
800
69.3k
    case WASM_INS_F32_MUL:
801
69.7k
    case WASM_INS_F32_DIV:
802
70.8k
    case WASM_INS_F32_MIN:
803
71.3k
    case WASM_INS_F32_MAX:
804
71.8k
    case WASM_INS_F32_COPYSIGN:
805
72.7k
    case WASM_INS_F64_ABS:
806
73.3k
    case WASM_INS_F64_NEG:
807
74.2k
    case WASM_INS_F64_CEIL:
808
74.8k
    case WASM_INS_F64_FLOOR:
809
75.3k
    case WASM_INS_F64_TRUNC:
810
76.2k
    case WASM_INS_F64_NEAREST:
811
76.7k
    case WASM_INS_F64_SQRT:
812
77.3k
    case WASM_INS_F64_ADD:
813
77.9k
    case WASM_INS_F64_SUB:
814
78.5k
    case WASM_INS_F64_MUL:
815
79.0k
    case WASM_INS_F64_DIV:
816
79.9k
    case WASM_INS_F64_MIN:
817
80.8k
    case WASM_INS_F64_MAX:
818
81.2k
    case WASM_INS_F64_COPYSIGN:
819
81.7k
    case WASM_INS_I32_WARP_I64:
820
82.4k
    case WASP_INS_I32_TRUNC_S_F32:
821
82.9k
    case WASM_INS_I32_TRUNC_U_F32:
822
83.7k
    case WASM_INS_I32_TRUNC_S_F64:
823
84.4k
    case WASM_INS_I32_TRUNC_U_F64:
824
85.0k
    case WASM_INS_I64_EXTEND_S_I32:
825
85.5k
    case WASM_INS_I64_EXTEND_U_I32:
826
86.0k
    case WASM_INS_I64_TRUNC_S_F32:
827
86.4k
    case WASM_INS_I64_TRUNC_U_F32:
828
87.0k
    case WASM_INS_I64_TRUNC_S_F64:
829
87.5k
    case WASM_INS_I64_TRUNC_U_F64:
830
88.0k
    case WASM_INS_F32_CONVERT_S_I32:
831
88.5k
    case WASM_INS_F32_CONVERT_U_I32:
832
89.0k
    case WASM_INS_F32_CONVERT_S_I64:
833
89.7k
    case WASM_INS_F32_CONVERT_U_I64:
834
90.4k
    case WASM_INS_F32_DEMOTE_F64:
835
91.0k
    case WASM_INS_F64_CONVERT_S_I32:
836
91.6k
    case WASM_INS_F64_CONVERT_U_I32:
837
92.4k
    case WASM_INS_F64_CONVERT_S_I64:
838
92.8k
    case WASM_INS_F64_CONVERT_U_I64:
839
93.4k
    case WASM_INS_F64_PROMOTE_F32:
840
93.9k
    case WASM_INS_I32_REINTERPRET_F32:
841
94.6k
    case WASM_INS_I64_REINTERPRET_F64:
842
95.8k
    case WASM_INS_F32_REINTERPRET_I32:
843
96.6k
    case WASM_INS_F64_REINTERPRET_I64:
844
96.6k
      MI->wasm_data.type = WASM_OP_NONE;
845
846
96.6k
      if (MI->flat_insn->detail) {
847
96.6k
        MI->flat_insn->detail->wasm.op_count = 0;
848
96.6k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_NUMBERIC;
849
96.6k
        MI->flat_insn->detail->groups_count++;
850
96.6k
      }
851
852
96.6k
      *size = 1;
853
854
96.6k
      break;
855
856
972
    case WASM_INS_DROP:
857
1.82k
    case WASM_INS_SELECT:
858
1.82k
      MI->wasm_data.type = WASM_OP_NONE;
859
860
1.82k
      if (MI->flat_insn->detail) {
861
1.82k
        MI->flat_insn->detail->wasm.op_count = 0;
862
1.82k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_PARAMETRIC;
863
1.82k
        MI->flat_insn->detail->groups_count++;
864
1.82k
      }
865
866
1.82k
      *size = 1;
867
868
1.82k
      break;
869
870
934
    case WASM_INS_GET_LOCAL:
871
1.57k
    case WASM_INS_SET_LOCAL:
872
2.07k
    case WASM_INS_TEE_LOCAL:
873
2.67k
    case WASM_INS_GET_GLOBAL:
874
3.78k
    case WASM_INS_SET_GLOBAL:
875
3.78k
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
876
88
        return false;
877
88
      }
878
879
3.70k
      if (MI->flat_insn->detail) {
880
3.70k
        MI->flat_insn->detail->wasm.op_count = 1;
881
3.70k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_VARIABLE;
882
3.70k
        MI->flat_insn->detail->groups_count++;
883
3.70k
      }
884
885
3.70k
      *size = param_size + 1;
886
887
3.70k
      break;
888
889
352
    case WASM_INS_I32_LOAD:
890
1.74k
    case WASM_INS_I64_LOAD:
891
2.44k
    case WASM_INS_F32_LOAD:
892
2.97k
    case WASM_INS_F64_LOAD:
893
3.75k
    case WASM_INS_I32_LOAD8_S:
894
4.66k
    case WASM_INS_I32_LOAD8_U:
895
5.18k
    case WASM_INS_I32_LOAD16_S:
896
5.70k
    case WASM_INS_I32_LOAD16_U:
897
6.70k
    case WASM_INS_I64_LOAD8_S:
898
7.85k
    case WASM_INS_I64_LOAD8_U:
899
8.42k
    case WASM_INS_I64_LOAD16_S:
900
9.12k
    case WASM_INS_I64_LOAD16_U:
901
9.71k
    case WASM_INS_I64_LOAD32_S:
902
10.4k
    case WASM_INS_I64_LOAD32_U:
903
11.0k
    case WASM_INS_I32_STORE:
904
11.5k
    case WASM_INS_I64_STORE:
905
12.7k
    case WASM_INS_F32_STORE:
906
13.1k
    case WASM_INS_F64_STORE:
907
13.9k
    case WASM_INS_I32_STORE8:
908
14.7k
    case WASM_INS_I32_STORE16:
909
15.1k
    case WASM_INS_I64_STORE8:
910
15.5k
    case WASM_INS_I64_STORE16:
911
16.1k
    case WASM_INS_I64_STORE32:
912
16.1k
      if (code_len == 1 || !read_memoryimmediate(&code[1], code_len - 1, &param_size, MI)) {
913
409
        return false;
914
409
      }
915
916
15.7k
      if (MI->flat_insn->detail) {
917
15.7k
        MI->flat_insn->detail->wasm.op_count = 2;
918
15.7k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
919
15.7k
        MI->flat_insn->detail->groups_count++;
920
15.7k
      }
921
922
15.7k
      *size = param_size + 1;
923
924
15.7k
      break;
925
926
657
    case WASM_INS_CURRENT_MEMORY:
927
1.48k
    case WASM_INS_GROW_MEMORY:
928
1.48k
      MI->wasm_data.type = WASM_OP_NONE;
929
930
1.48k
      if (MI->flat_insn->detail) {
931
1.48k
        MI->flat_insn->detail->wasm.op_count = 0;
932
1.48k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_MEMORY;
933
1.48k
        MI->flat_insn->detail->groups_count++;
934
1.48k
      }
935
936
1.48k
      *size = 1;
937
938
1.48k
      break;
939
940
11.9k
    case WASM_INS_UNREACHABLE:
941
13.0k
    case WASM_INS_NOP:
942
13.7k
    case WASM_INS_ELSE:
943
15.5k
    case WASM_INS_END:
944
16.0k
    case WASM_INS_RETURN:
945
16.0k
      MI->wasm_data.type = WASM_OP_NONE;
946
947
16.0k
      if (MI->flat_insn->detail) {
948
16.0k
        MI->flat_insn->detail->wasm.op_count = 0;
949
16.0k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
950
16.0k
        MI->flat_insn->detail->groups_count++;
951
16.0k
      }
952
953
16.0k
      *size = 1;
954
955
16.0k
      break;
956
957
826
    case WASM_INS_BLOCK:
958
1.34k
    case WASM_INS_LOOP:
959
2.13k
    case WASM_INS_IF:
960
2.13k
      if (code_len == 1 || !read_varint7(&code[1], code_len - 1, &param_size, MI)) {
961
41
        return false;
962
41
      }
963
964
2.09k
      if (MI->flat_insn->detail) {
965
2.09k
        MI->flat_insn->detail->wasm.op_count = 1;
966
2.09k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
967
2.09k
        MI->flat_insn->detail->groups_count++;
968
2.09k
      }
969
970
2.09k
      *size = param_size + 1;
971
972
2.09k
      break;
973
974
481
    case WASM_INS_BR:
975
936
    case WASM_INS_BR_IF:
976
1.51k
    case WASM_INS_CALL:
977
2.00k
    case WASM_INS_CALL_INDIRECT:
978
2.00k
      if (code_len == 1 || !read_varuint32(&code[1], code_len - 1, &param_size, MI)) {
979
67
        return false;
980
67
      }
981
982
1.93k
      if (MI->flat_insn->detail) {
983
1.93k
        MI->flat_insn->detail->wasm.op_count = 1;
984
1.93k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
985
1.93k
        MI->flat_insn->detail->groups_count++;
986
1.93k
      }
987
988
1.93k
      *size = param_size + 1;
989
990
1.93k
      break;
991
992
1.93k
    case WASM_INS_BR_TABLE:
993
1.93k
      if (code_len == 1 || !read_brtable(&code[1], code_len - 1, &param_size, MI)) {
994
206
        return false;
995
206
      }
996
997
1.72k
      if (MI->flat_insn->detail) {
998
1.72k
        MI->flat_insn->detail->wasm.op_count = 1;
999
1.72k
        MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = WASM_GRP_CONTROL;
1000
1.72k
        MI->flat_insn->detail->groups_count++;
1001
1.72k
      }
1002
1003
1.72k
      *size = param_size + 1;
1004
1005
1.72k
      break;
1006
144k
  }
1007
1008
143k
  return true;
1009
144k
}