Coverage Report

Created: 2026-03-07 06:35

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/wasm3/source/m3_compile.c
Line
Count
Source
1
//
2
//  m3_compile.c
3
//
4
//  Created by Steven Massey on 4/17/19.
5
//  Copyright © 2019 Steven Massey. All rights reserved.
6
//
7
8
// Allow using opcodes for compilation process
9
#define M3_COMPILE_OPCODES
10
11
#include "m3_env.h"
12
#include "m3_compile.h"
13
#include "m3_exec.h"
14
#include "m3_exception.h"
15
#include "m3_info.h"
16
17
//----- EMIT --------------------------------------------------------------------------------------------------------------
18
19
static inline
20
pc_t GetPC (IM3Compilation o)
21
223
{
22
223
    return GetPagePC (o->page);
23
223
}
24
25
static M3_NOINLINE
26
M3Result  EnsureCodePageNumLines  (IM3Compilation o, u32 i_numLines)
27
37.0k
{
28
37.0k
    M3Result result = m3Err_none;
29
30
37.0k
    i_numLines += 2; // room for Bridge
31
32
37.0k
    if (NumFreeLines (o->page) < i_numLines)
33
26
    {
34
26
        IM3CodePage page = AcquireCodePageWithCapacity (o->runtime, i_numLines);
35
36
26
        if (page)
37
26
        {
38
26
            m3log (emit, "bridging new code page from: %d %p (free slots: %d) to: %d", o->page->info.sequence, GetPC (o), NumFreeLines (o->page), page->info.sequence);
39
26
            d_m3Assert (NumFreeLines (o->page) >= 2);
40
41
26
            EmitWord (o->page, op_Branch);
42
26
            EmitWord (o->page, GetPagePC (page));
43
44
26
            ReleaseCodePage (o->runtime, o->page);
45
46
26
            o->page = page;
47
26
        }
48
0
        else result = m3Err_mallocFailedCodePage;
49
26
    }
50
51
37.0k
    return result;
52
37.0k
}
53
54
static M3_NOINLINE
55
M3Result  EmitOp  (IM3Compilation o, IM3Operation i_operation)
56
37.7k
{
57
37.7k
    M3Result result = m3Err_none;                                 d_m3Assert (i_operation or IsStackPolymorphic (o));
58
59
    // it's OK for page to be null; when compile-walking the bytecode without emitting
60
37.7k
    if (o->page)
61
36.9k
    {
62
# if d_m3EnableOpTracing
63
        if (i_operation != op_DumpStack)
64
            o->numEmits++;
65
# endif
66
67
        // have execution jump to a new page if slots are critically low
68
36.9k
        result = EnsureCodePageNumLines (o, d_m3CodePageFreeLinesThreshold);
69
70
36.9k
        if (not result)
71
36.9k
        {                                                           if (d_m3LogEmit) log_emit (o, i_operation);
72
# if d_m3RecordBacktraces
73
            EmitMappingEntry (o->page, o->lastOpcodeStart - o->module->wasmStart);
74
# endif // d_m3RecordBacktraces
75
36.9k
            EmitWord (o->page, i_operation);
76
36.9k
        }
77
36.9k
    }
78
79
37.7k
    return result;
80
37.7k
}
81
82
// Push an immediate constant into the M3 codestream
83
static M3_NOINLINE
84
void  EmitConstant32  (IM3Compilation o, const u32 i_immediate)
85
54
{
86
54
    if (o->page)
87
54
        EmitWord32 (o->page, i_immediate);
88
54
}
89
90
static M3_NOINLINE
91
void  EmitSlotOffset  (IM3Compilation o, const i32 i_offset)
92
65.3k
{
93
65.3k
    if (o->page)
94
64.9k
        EmitWord32 (o->page, i_offset);
95
65.3k
}
96
97
static M3_NOINLINE
98
pc_t  EmitPointer  (IM3Compilation o, const void * const i_pointer)
99
1.59k
{
100
1.59k
    pc_t ptr = GetPagePC (o->page);
101
102
1.59k
    if (o->page)
103
1.12k
        EmitWord (o->page, i_pointer);
104
105
1.59k
    return ptr;
106
1.59k
}
107
108
static M3_NOINLINE
109
void * ReservePointer (IM3Compilation o)
110
52
{
111
52
    pc_t ptr = GetPagePC (o->page);
112
52
    EmitPointer (o, NULL);
113
52
    return (void *) ptr;
114
52
}
115
116
117
//-------------------------------------------------------------------------------------------------------------------------
118
119
#define d_indent "     | %s"
120
121
// just want less letters and numbers to stare at down the way in the compiler table
122
#define i_32    c_m3Type_i32
123
#define i_64    c_m3Type_i64
124
#define f_32    c_m3Type_f32
125
#define f_64    c_m3Type_f64
126
#define none    c_m3Type_none
127
#define any     (u8)-1
128
129
#if d_m3HasFloat
130
#   define FPOP(x) x
131
#else
132
#   define FPOP(x) NULL
133
#endif
134
135
static const IM3Operation c_preserveSetSlot [] = { NULL, op_PreserveSetSlot_i32,       op_PreserveSetSlot_i64,
136
                                                    FPOP(op_PreserveSetSlot_f32), FPOP(op_PreserveSetSlot_f64) };
137
static const IM3Operation c_setSetOps [] =       { NULL, op_SetSlot_i32,               op_SetSlot_i64,
138
                                                    FPOP(op_SetSlot_f32),         FPOP(op_SetSlot_f64) };
139
static const IM3Operation c_setGlobalOps [] =    { NULL, op_SetGlobal_i32,             op_SetGlobal_i64,
140
                                                    FPOP(op_SetGlobal_f32),       FPOP(op_SetGlobal_f64) };
141
static const IM3Operation c_setRegisterOps [] =  { NULL, op_SetRegister_i32,           op_SetRegister_i64,
142
                                                    FPOP(op_SetRegister_f32),     FPOP(op_SetRegister_f64) };
143
144
static const IM3Operation c_intSelectOps [2] [4] =      { { op_Select_i32_rss, op_Select_i32_srs, op_Select_i32_ssr, op_Select_i32_sss },
145
                                                          { op_Select_i64_rss, op_Select_i64_srs, op_Select_i64_ssr, op_Select_i64_sss } };
146
#if d_m3HasFloat
147
static const IM3Operation c_fpSelectOps [2] [2] [3] = { { { op_Select_f32_sss, op_Select_f32_srs, op_Select_f32_ssr },        // selector in slot
148
                                                          { op_Select_f32_rss, op_Select_f32_rrs, op_Select_f32_rsr } },      // selector in reg
149
                                                        { { op_Select_f64_sss, op_Select_f64_srs, op_Select_f64_ssr },        // selector in slot
150
                                                          { op_Select_f64_rss, op_Select_f64_rrs, op_Select_f64_rsr } } };    // selector in reg
151
#endif
152
153
// all args & returns are 64-bit aligned, so use 2 slots for a d_m3Use32BitSlots=1 build
154
static const u16 c_ioSlotCount = sizeof (u64) / sizeof (m3slot_t);
155
156
static
157
M3Result  AcquireCompilationCodePage  (IM3Compilation o, IM3CodePage * o_codePage)
158
286
{
159
286
    M3Result result = m3Err_none;
160
161
286
    IM3CodePage page = AcquireCodePage (o->runtime);
162
163
286
    if (page)
164
286
    {
165
#       if (d_m3EnableCodePageRefCounting)
166
        {
167
            if (o->function)
168
            {
169
                IM3Function func = o->function;
170
                page->info.usageCount++;
171
172
                u32 index = func->numCodePageRefs++;
173
_               (m3ReallocArray (& func->codePageRefs, IM3CodePage, func->numCodePageRefs, index));
174
                func->codePageRefs [index] = page;
175
            }
176
        }
177
#       endif
178
286
    }
179
286
    else _throw (m3Err_mallocFailedCodePage);
180
181
286
    _catch:
182
183
286
    * o_codePage = page;
184
185
286
    return result;
186
286
}
187
188
static inline
189
void  ReleaseCompilationCodePage  (IM3Compilation o)
190
280
{
191
280
    ReleaseCodePage (o->runtime, o->page);
192
280
}
193
194
static inline
195
u16 GetTypeNumSlots (u8 i_type)
196
290k
{
197
290k
#   if d_m3Use32BitSlots
198
290k
        return Is64BitType (i_type) ? 2 : 1;
199
#   else
200
        return 1;
201
#   endif
202
290k
}
203
204
static inline
205
void  AlignSlotToType  (u16 * io_slot, u8 i_type)
206
37.6k
{
207
    // align 64-bit words to even slots (if d_m3Use32BitSlots)
208
37.6k
    u16 numSlots = GetTypeNumSlots (i_type);
209
210
37.6k
    u16 mask = numSlots - 1;
211
37.6k
    * io_slot = (* io_slot + mask) & ~mask;
212
37.6k
}
213
214
static inline
215
i16  GetStackTopIndex  (IM3Compilation o)
216
3.83k
{                                                           d_m3Assert (o->stackIndex > o->stackFirstDynamicIndex or IsStackPolymorphic (o));
217
3.83k
    return o->stackIndex - 1;
218
3.83k
}
219
220
221
// Items in the static portion of the stack (args/locals) are hidden from GetStackTypeFromTop ()
222
// In other words, only "real" Wasm stack items can be inspected.  This is important when
223
// returning values, etc. and you need an accurate wasm-view of the stack.
224
static
225
u8  GetStackTypeFromTop  (IM3Compilation o, u16 i_offset)
226
16.2k
{
227
16.2k
    u8 type = c_m3Type_none;
228
229
16.2k
    ++i_offset;
230
16.2k
    if (o->stackIndex >= i_offset)
231
16.1k
    {
232
16.1k
        u16 index = o->stackIndex - i_offset;
233
234
16.1k
        if (index >= o->stackFirstDynamicIndex)
235
15.5k
            type = o->typeStack [index];
236
16.1k
    }
237
238
16.2k
    return type;
239
16.2k
}
240
241
static inline
242
u8  GetStackTopType  (IM3Compilation o)
243
16.0k
{
244
16.0k
    return GetStackTypeFromTop (o, 0);
245
16.0k
}
246
247
static inline
248
u8  GetStackTypeFromBottom  (IM3Compilation o, u16 i_offset)
249
118k
{
250
118k
    u8 type = c_m3Type_none;
251
252
118k
    if (i_offset < o->stackIndex)
253
118k
        type = o->typeStack [i_offset];
254
255
118k
    return type;
256
118k
}
257
258
259
11.5k
static inline bool  IsConstantSlot    (IM3Compilation o, u16 i_slot)  { return (i_slot >= o->slotFirstConstIndex and i_slot < o->slotMaxConstIndex); }
260
3.68k
static inline bool  IsSlotAllocated   (IM3Compilation o, u16 i_slot)  { return o->m3Slots [i_slot]; }
261
262
static inline
263
bool  IsStackIndexInRegister  (IM3Compilation o, i32 i_stackIndex)
264
30.5k
{                                                                           d_m3Assert (i_stackIndex < o->stackIndex or IsStackPolymorphic (o));
265
30.5k
    if (i_stackIndex >= 0 and i_stackIndex < o->stackIndex)
266
30.4k
        return (o->wasmStack [i_stackIndex] >= d_m3Reg0SlotAlias);
267
62
    else
268
62
        return false;
269
30.5k
}
270
271
679
static inline u16   GetNumBlockValuesOnStack      (IM3Compilation o)  { return o->stackIndex - o->block.blockStackIndex; }
272
273
1.32k
static inline bool  IsStackTopInRegister          (IM3Compilation o)  { return IsStackIndexInRegister (o, (i32) GetStackTopIndex (o));       }
274
249
static inline bool  IsStackTopMinus1InRegister    (IM3Compilation o)  { return IsStackIndexInRegister (o, (i32) GetStackTopIndex (o) - 1);   }
275
15
static inline bool  IsStackTopMinus2InRegister    (IM3Compilation o)  { return IsStackIndexInRegister (o, (i32) GetStackTopIndex (o) - 2);   }
276
277
879
static inline bool  IsStackTopInSlot              (IM3Compilation o)  { return not IsStackTopInRegister (o); }
278
279
63
static inline bool  IsValidSlot                   (u16 i_slot)        { return (i_slot < d_m3MaxFunctionSlots); }
280
281
static inline
282
u16  GetStackTopSlotNumber  (IM3Compilation o)
283
729
{
284
729
    i16 i = GetStackTopIndex (o);
285
286
729
    u16 slot = c_slotUnused;
287
288
729
    if (i >= 0)
289
698
        slot = o->wasmStack [i];
290
291
729
    return slot;
292
729
}
293
294
295
// from bottom
296
static inline
297
u16  GetSlotForStackIndex  (IM3Compilation o, u16 i_stackIndex)
298
229k
{                                                                   d_m3Assert (i_stackIndex < o->stackIndex or IsStackPolymorphic (o));
299
229k
    u16 slot = c_slotUnused;
300
301
229k
    if (i_stackIndex < o->stackIndex)
302
229k
        slot = o->wasmStack [i_stackIndex];
303
304
229k
    return slot;
305
229k
}
306
307
static inline
308
u16  GetExtraSlotForStackIndex  (IM3Compilation o, u16 i_stackIndex)
309
60.7k
{
310
60.7k
    u16 baseSlot = GetSlotForStackIndex (o, i_stackIndex);
311
312
60.7k
    if (baseSlot != c_slotUnused)
313
60.7k
    {
314
60.7k
        u16 extraSlot = GetTypeNumSlots (GetStackTypeFromBottom (o, i_stackIndex)) - 1;
315
60.7k
        baseSlot += extraSlot;
316
60.7k
    }
317
318
60.7k
    return baseSlot;
319
60.7k
}
320
321
322
static inline
323
void  TouchSlot  (IM3Compilation o, u16 i_slot)
324
96.5k
{
325
    // op_Entry uses this value to track and detect stack overflow
326
96.5k
    o->maxStackSlots = M3_MAX (o->maxStackSlots, i_slot + 1);
327
96.5k
}
328
329
static inline
330
void  MarkSlotAllocated  (IM3Compilation o, u16 i_slot)
331
84.1k
{                                                                   d_m3Assert (o->m3Slots [i_slot] == 0); // shouldn't be already allocated
332
84.1k
    o->m3Slots [i_slot] = 1;
333
334
84.1k
    o->slotMaxAllocatedIndexPlusOne = M3_MAX (o->slotMaxAllocatedIndexPlusOne, i_slot + 1);
335
336
84.1k
    TouchSlot (o, i_slot);
337
84.1k
}
338
339
static inline
340
void  MarkSlotsAllocated  (IM3Compilation o, u16 i_slot, u16 i_numSlots)
341
57.2k
{
342
137k
    while (i_numSlots--)
343
79.9k
        MarkSlotAllocated (o, i_slot++);
344
57.2k
}
345
346
static inline
347
void  MarkSlotsAllocatedByType  (IM3Compilation o, u16 i_slot, u8 i_type)
348
25.6k
{
349
25.6k
    u16 numSlots = GetTypeNumSlots (i_type);
350
25.6k
    MarkSlotsAllocated (o, i_slot, numSlots);
351
25.6k
}
352
353
354
static
355
M3Result  AllocateSlotsWithinRange  (IM3Compilation o, u16 * o_slot, u8 i_type, u16 i_startSlot, u16 i_endSlot)
356
37.2k
{
357
37.2k
    M3Result result = m3Err_functionStackOverflow;
358
359
37.2k
    u16 numSlots = GetTypeNumSlots (i_type);
360
37.2k
    u16 searchOffset = numSlots - 1;
361
362
37.2k
    AlignSlotToType (& i_startSlot, i_type);
363
364
    // search for 1 or 2 consecutive slots in the execution stack
365
37.2k
    u16 i = i_startSlot;
366
3.83M
    while (i + searchOffset < i_endSlot)
367
3.82M
    {
368
3.82M
        if (o->m3Slots [i] == 0 and o->m3Slots [i + searchOffset] == 0)
369
31.5k
        {
370
31.5k
            MarkSlotsAllocated (o, i, numSlots);
371
372
31.5k
            * o_slot = i;
373
31.5k
            result = m3Err_none;
374
31.5k
            break;
375
31.5k
        }
376
377
        // keep 2-slot allocations even-aligned
378
3.79M
        i += numSlots;
379
3.79M
    }
380
381
37.2k
    return result;
382
37.2k
}
383
384
static inline
385
M3Result  AllocateSlots  (IM3Compilation o, u16 * o_slot, u8 i_type)
386
31.5k
{
387
31.5k
    return AllocateSlotsWithinRange (o, o_slot, i_type, o->slotFirstDynamicIndex, d_m3MaxFunctionSlots);
388
31.5k
}
389
390
static inline
391
M3Result  AllocateConstantSlots  (IM3Compilation o, u16 * o_slot, u8 i_type)
392
5.65k
{
393
5.65k
    u16 maxTableIndex = o->slotFirstConstIndex + d_m3MaxConstantTableSize;
394
5.65k
    return AllocateSlotsWithinRange (o, o_slot, i_type, o->slotFirstConstIndex, M3_MIN(o->slotFirstDynamicIndex, maxTableIndex));
395
5.65k
}
396
397
398
// TOQUE: this usage count system could be eliminated. real world code doesn't frequently trigger it.  just copy to multiple
399
// unique slots.
400
static inline
401
M3Result  IncrementSlotUsageCount  (IM3Compilation o, u16 i_slot)
402
0
{                                                                                       d_m3Assert (i_slot < d_m3MaxFunctionSlots);
403
0
    M3Result result = m3Err_none;                                                       d_m3Assert (o->m3Slots [i_slot] > 0);
404
405
    // OPTZ (memory): 'm3Slots' could still be fused with 'typeStack' if 4 bits were used to indicate: [0,1,2,many]. The many-case
406
    // would scan 'wasmStack' to determine the actual usage count
407
0
    if (o->m3Slots [i_slot] < 0xFF)
408
0
    {
409
0
        o->m3Slots [i_slot]++;
410
0
    }
411
0
    else result = "slot usage count overflow";
412
413
0
    return result;
414
0
}
415
416
static inline
417
void DeallocateSlot (IM3Compilation o, i16 i_slot, u8 i_type)
418
40.8k
{                                                                                       d_m3Assert (i_slot >= o->slotFirstDynamicIndex);
419
40.8k
                                                                                        d_m3Assert (i_slot < o->slotMaxAllocatedIndexPlusOne);
420
94.8k
    for (u16 i = 0; i < GetTypeNumSlots (i_type); ++i, ++i_slot)
421
53.9k
    {                                                                                   d_m3Assert (o->m3Slots [i_slot]);
422
53.9k
        -- o->m3Slots [i_slot];
423
53.9k
    }
424
40.8k
}
425
426
427
static inline
428
bool  IsRegisterTypeAllocated  (IM3Compilation o, u8 i_type)
429
9
{
430
9
    return IsRegisterAllocated (o, IsFpType (i_type));
431
9
}
432
433
static inline
434
void  AllocateRegister  (IM3Compilation o, u32 i_register, u16 i_stackIndex)
435
684
{                                                                                       d_m3Assert (not IsRegisterAllocated (o, i_register));
436
684
    o->regStackIndexPlusOne [i_register] = i_stackIndex + 1;
437
684
}
438
439
static inline
440
void  DeallocateRegister  (IM3Compilation o, u32 i_register)
441
670
{                                                                                       d_m3Assert (IsRegisterAllocated (o, i_register));
442
670
    o->regStackIndexPlusOne [i_register] = c_m3RegisterUnallocated;
443
670
}
444
445
static inline
446
u16  GetRegisterStackIndex  (IM3Compilation o, u32 i_register)
447
111
{                                                                                       d_m3Assert (IsRegisterAllocated (o, i_register));
448
111
    return o->regStackIndexPlusOne [i_register] - 1;
449
111
}
450
451
u16  GetMaxUsedSlotPlusOne  (IM3Compilation o)
452
195
{
453
3.74k
    while (o->slotMaxAllocatedIndexPlusOne > o->slotFirstDynamicIndex)
454
3.68k
    {
455
3.68k
        if (IsSlotAllocated (o, o->slotMaxAllocatedIndexPlusOne - 1))
456
138
            break;
457
458
3.54k
        o->slotMaxAllocatedIndexPlusOne--;
459
3.54k
    }
460
461
#   ifdef DEBUG
462
        u16 maxSlot = o->slotMaxAllocatedIndexPlusOne;
463
        while (maxSlot < d_m3MaxFunctionSlots)
464
        {
465
            d_m3Assert (o->m3Slots [maxSlot] == 0);
466
            maxSlot++;
467
        }
468
#   endif
469
470
195
    return o->slotMaxAllocatedIndexPlusOne;
471
195
}
472
473
static
474
M3Result  PreserveRegisterIfOccupied  (IM3Compilation o, u8 i_registerType)
475
1.49k
{
476
1.49k
    M3Result result = m3Err_none;
477
478
1.49k
    u32 regSelect = IsFpType (i_registerType);
479
480
1.49k
    if (IsRegisterAllocated (o, regSelect))
481
98
    {
482
98
        u16 stackIndex = GetRegisterStackIndex (o, regSelect);
483
98
        DeallocateRegister (o, regSelect);
484
485
98
        u8 type = GetStackTypeFromBottom (o, stackIndex);
486
487
        // and point to a exec slot
488
98
        u16 slot = c_slotUnused;
489
98
_       (AllocateSlots (o, & slot, type));
490
98
        o->wasmStack [stackIndex] = slot;
491
492
98
_       (EmitOp (o, c_setSetOps [type]));
493
98
        EmitSlotOffset (o, slot);
494
98
    }
495
496
1.49k
    _catch: return result;
497
1.49k
}
498
499
500
// all values must be in slots before entering loop, if, and else blocks
501
// otherwise they'd end up preserve-copied in the block to probably different locations (if/else)
502
static inline
503
M3Result  PreserveRegisters  (IM3Compilation o)
504
477
{
505
477
    M3Result result;
506
507
477
_   (PreserveRegisterIfOccupied (o, c_m3Type_f64));
508
477
_   (PreserveRegisterIfOccupied (o, c_m3Type_i64));
509
510
477
    _catch: return result;
511
477
}
512
513
static
514
M3Result  PreserveNonTopRegisters  (IM3Compilation o)
515
44
{
516
44
    M3Result result = m3Err_none;
517
518
44
    i16 stackTop = GetStackTopIndex (o);
519
520
44
    if (stackTop >= 0)
521
43
    {
522
43
        if (IsRegisterAllocated (o, 0))     // r0
523
8
        {
524
8
            if (GetRegisterStackIndex (o, 0) != stackTop)
525
8
_               (PreserveRegisterIfOccupied (o, c_m3Type_i64));
526
8
        }
527
528
43
        if (IsRegisterAllocated (o, 1))     // fp0
529
5
        {
530
5
            if (GetRegisterStackIndex (o, 1) != stackTop)
531
5
_               (PreserveRegisterIfOccupied (o, c_m3Type_f64));
532
5
        }
533
43
    }
534
535
44
    _catch: return result;
536
44
}
537
538
539
//----------------------------------------------------------------------------------------------------------------------
540
541
static
542
M3Result  Push  (IM3Compilation o, u8 i_type, u16 i_slot)
543
58.5k
{
544
58.5k
    M3Result result = m3Err_none;
545
546
#if !d_m3HasFloat
547
    if (i_type == c_m3Type_f32 || i_type == c_m3Type_f64) {
548
        return m3Err_unknownOpcode;
549
    }
550
#endif
551
552
58.5k
    u16 stackIndex = o->stackIndex++;                                       // printf ("push: %d\n", (i32) i);
553
554
58.5k
    if (stackIndex < d_m3MaxFunctionStackHeight)
555
56.4k
    {
556
56.4k
        o->wasmStack        [stackIndex] = i_slot;
557
56.4k
        o->typeStack        [stackIndex] = i_type;
558
559
56.4k
        if (IsRegisterSlotAlias (i_slot))
560
684
        {
561
684
            u32 regSelect = IsFpRegisterSlotAlias (i_slot);
562
684
            AllocateRegister (o, regSelect, stackIndex);
563
684
        }
564
565
56.4k
        if (d_m3LogWasmStack) dump_type_stack (o);
566
56.4k
    }
567
2.11k
    else result = m3Err_functionStackOverflow;
568
569
58.5k
    return result;
570
58.5k
}
571
572
static inline
573
M3Result  PushRegister  (IM3Compilation o, u8 i_type)
574
587
{
575
587
    M3Result result = m3Err_none;                                                       d_m3Assert ((u16) d_m3Reg0SlotAlias > (u16) d_m3MaxFunctionSlots);
576
587
    u16 slot = IsFpType (i_type) ? d_m3Fp0SlotAlias : d_m3Reg0SlotAlias;                d_m3Assert (i_type or IsStackPolymorphic (o));
577
578
587
_   (Push (o, i_type, slot));
579
580
587
    _catch: return result;
581
580
}
582
583
static
584
M3Result  Pop  (IM3Compilation o)
585
47.0k
{
586
47.0k
    M3Result result = m3Err_none;
587
588
47.0k
    if (o->stackIndex > o->block.blockStackIndex)
589
43.2k
    {
590
43.2k
        o->stackIndex--;                                                //  printf ("pop: %d\n", (i32) o->stackIndex);
591
592
43.2k
        u16 slot = o->wasmStack [o->stackIndex];
593
43.2k
        u8 type = o->typeStack [o->stackIndex];
594
595
43.2k
        if (IsRegisterSlotAlias (slot))
596
572
        {
597
572
            u32 regSelect = IsFpRegisterSlotAlias (slot);
598
572
            DeallocateRegister (o, regSelect);
599
572
        }
600
42.6k
        else if (slot >= o->slotFirstDynamicIndex)
601
40.8k
        {
602
40.8k
            DeallocateSlot (o, slot, type);
603
40.8k
        }
604
43.2k
    }
605
3.85k
    else if (not IsStackPolymorphic (o))
606
0
        result = m3Err_functionStackUnderrun;
607
608
47.0k
    return result;
609
47.0k
}
610
611
static
612
M3Result  PopType  (IM3Compilation o, u8 i_type)
613
15.2k
{
614
15.2k
    M3Result result = m3Err_none;
615
616
15.2k
    u8 topType = GetStackTopType (o);
617
618
15.2k
    if (i_type == topType or o->block.isPolymorphic)
619
15.2k
    {
620
15.2k
_       (Pop (o));
621
15.2k
    }
622
15.2k
    else _throw (m3Err_typeMismatch);
623
624
15.2k
    _catch:
625
15.2k
    return result;
626
15.2k
}
627
628
static
629
M3Result  _PushAllocatedSlotAndEmit  (IM3Compilation o, u8 i_type, bool i_doEmit)
630
31.4k
{
631
31.4k
    M3Result result = m3Err_none;
632
633
31.4k
    u16 slot = c_slotUnused;
634
635
31.4k
_   (AllocateSlots (o, & slot, i_type));
636
31.4k
_   (Push (o, i_type, slot));
637
638
31.4k
    if (i_doEmit)
639
6.51k
        EmitSlotOffset (o, slot);
640
641
//    printf ("push: %d\n", (u32) slot);
642
643
31.4k
    _catch: return result;
644
31.4k
}
645
646
static inline
647
M3Result  PushAllocatedSlotAndEmit  (IM3Compilation o, u8 i_type)
648
6.51k
{
649
6.51k
    return _PushAllocatedSlotAndEmit (o, i_type, true);
650
6.51k
}
651
652
static inline
653
M3Result  PushAllocatedSlot  (IM3Compilation o, u8 i_type)
654
24.9k
{
655
24.9k
    return _PushAllocatedSlotAndEmit (o, i_type, false);
656
24.9k
}
657
658
static
659
M3Result  PushConst  (IM3Compilation o, u64 i_word, u8 i_type)
660
20.5k
{
661
20.5k
    M3Result result = m3Err_none;
662
663
    // Early-exit if we're not emitting
664
20.5k
    if (!o->page) return result;
665
666
20.5k
    bool matchFound = false;
667
5.65k
    bool is64BitType = Is64BitType (i_type);
668
669
5.65k
    u16 numRequiredSlots = GetTypeNumSlots (i_type);
670
5.65k
    u16 numUsedConstSlots = o->slotMaxConstIndex - o->slotFirstConstIndex;
671
672
    // search for duplicate matching constant slot to reuse
673
5.65k
    if (numRequiredSlots == 2 and numUsedConstSlots >= 2)
674
0
    {
675
0
        u16 firstConstSlot = o->slotFirstConstIndex;
676
0
        AlignSlotToType (& firstConstSlot, c_m3Type_i64);
677
678
0
        for (u16 slot = firstConstSlot; slot < o->slotMaxConstIndex - 1; slot += 2)
679
0
        {
680
0
            if (IsSlotAllocated (o, slot) and IsSlotAllocated (o, slot + 1))
681
0
            {
682
0
                u64 constant;
683
0
                memcpy (&constant, &o->constants [slot - o->slotFirstConstIndex], sizeof(constant));
684
685
0
                if (constant == i_word)
686
0
                {
687
0
                    matchFound = true;
688
0
_                   (Push (o, i_type, slot));
689
0
                    break;
690
0
                }
691
0
            }
692
0
        }
693
0
    }
694
5.65k
    else if (numRequiredSlots == 1)
695
772
    {
696
772
        for (u16 i = 0; i < numUsedConstSlots; ++i)
697
0
        {
698
0
            u16 slot = o->slotFirstConstIndex + i;
699
700
0
            if (IsSlotAllocated (o, slot))
701
0
            {
702
0
                bool matches;
703
0
                if (is64BitType) {
704
0
                    u64 constant;
705
0
                    memcpy (&constant, &o->constants [i], sizeof(constant));
706
0
                    matches = (constant == i_word);
707
0
                } else {
708
0
                    u32 constant;
709
0
                    memcpy (&constant, &o->constants [i], sizeof(constant));
710
0
                    matches = (constant == i_word);
711
0
                }
712
0
                if (matches)
713
0
                {
714
0
                    matchFound = true;
715
0
_                   (Push (o, i_type, slot));
716
0
                    break;
717
0
                }
718
0
            }
719
0
        }
720
772
    }
721
722
5.65k
    if (not matchFound)
723
5.65k
    {
724
5.65k
        u16 slot = c_slotUnused;
725
5.65k
        result = AllocateConstantSlots (o, & slot, i_type);
726
727
5.65k
        if (result || slot == c_slotUnused) // no more constant table space; use inline constants
728
5.65k
        {
729
5.65k
            result = m3Err_none;
730
731
5.65k
            if (is64BitType) {
732
4.88k
_               (EmitOp (o, op_Const64));
733
4.88k
                EmitWord64 (o->page, i_word);
734
4.88k
            } else {
735
771
_               (EmitOp (o, op_Const32));
736
771
                EmitWord32 (o->page, (u32) i_word);
737
771
            }
738
739
5.65k
_           (PushAllocatedSlotAndEmit (o, i_type));
740
5.65k
        }
741
3
        else
742
3
        {
743
3
            u16 constTableIndex = slot - o->slotFirstConstIndex;
744
745
3
            d_m3Assert(constTableIndex < d_m3MaxConstantTableSize);
746
747
3
            if (is64BitType) {
748
2
                memcpy (& o->constants [constTableIndex], &i_word, sizeof(i_word));
749
2
            } else {
750
1
                u32 word32 = i_word;
751
1
                memcpy (& o->constants [constTableIndex], &word32, sizeof(word32));
752
1
            }
753
754
3
_           (Push (o, i_type, slot));
755
756
3
            o->slotMaxConstIndex = M3_MAX (slot + numRequiredSlots, o->slotMaxConstIndex);
757
3
        }
758
5.65k
    }
759
760
5.65k
    _catch: return result;
761
5.65k
}
762
763
static inline
764
M3Result  EmitSlotNumOfStackTopAndPop  (IM3Compilation o)
765
613
{
766
    // no emit if value is in register
767
613
    if (IsStackTopInSlot (o))
768
407
        EmitSlotOffset (o, GetStackTopSlotNumber (o));
769
770
613
    return Pop (o);
771
613
}
772
773
774
// Or, maybe: EmitTrappingOp
775
M3Result  AddTrapRecord  (IM3Compilation o)
776
161
{
777
161
    M3Result result = m3Err_none;
778
779
161
    if (o->function)
780
161
    {
781
161
    }
782
783
161
    return result;
784
161
}
785
786
static
787
M3Result  UnwindBlockStack  (IM3Compilation o)
788
184
{
789
184
    M3Result result = m3Err_none;
790
791
184
    u32 popCount = 0;
792
6.15k
    while (o->stackIndex > o->block.blockStackIndex)
793
5.97k
    {
794
5.97k
_       (Pop (o));
795
5.97k
        ++popCount;
796
5.97k
    }
797
798
184
    if (popCount)
799
115
    {
800
115
        m3log (compile, "unwound stack top: %d", popCount);
801
115
    }
802
803
184
    _catch: return result;
804
184
}
805
806
static inline
807
M3Result  SetStackPolymorphic  (IM3Compilation o)
808
161
{
809
161
    o->block.isPolymorphic = true;                              m3log (compile, "stack set polymorphic");
810
161
    return UnwindBlockStack (o);
811
161
}
812
813
static
814
void  PatchBranches  (IM3Compilation o)
815
215
{
816
215
    pc_t pc = GetPC (o);
817
818
215
    pc_t patches = o->block.patches;
819
215
    o->block.patches = NULL;
820
821
215
    while (patches)
822
0
    {                                                           m3log (compile, "patching location: %p to pc: %p", patches, pc);
823
0
        pc_t next = * (pc_t *) patches;
824
0
        * (pc_t *) patches = pc;
825
0
        patches = next;
826
0
    }
827
215
}
828
829
//-------------------------------------------------------------------------------------------------------------------------
830
831
static
832
M3Result  CopyStackIndexToSlot  (IM3Compilation o, u16 i_destSlot, u16 i_stackIndex)  // NoPushPop
833
28.9k
{
834
28.9k
    M3Result result = m3Err_none;
835
836
28.9k
    IM3Operation op;
837
838
28.9k
    u8 type = GetStackTypeFromBottom (o, i_stackIndex);
839
28.9k
    bool inRegister = IsStackIndexInRegister (o, i_stackIndex);
840
841
28.9k
    if (inRegister)
842
6
    {
843
6
        op = c_setSetOps [type];
844
6
    }
845
28.9k
    else op = Is64BitType (type) ? op_CopySlot_64 : op_CopySlot_32;
846
847
28.9k
_   (EmitOp (o, op));
848
28.9k
    EmitSlotOffset (o, i_destSlot);
849
850
28.9k
    if (not inRegister)
851
28.9k
    {
852
28.9k
        u16 srcSlot = GetSlotForStackIndex (o, i_stackIndex);
853
28.9k
        EmitSlotOffset (o, srcSlot);
854
28.9k
    }
855
856
28.9k
    _catch: return result;
857
28.9k
}
858
859
static
860
M3Result  CopyStackTopToSlot  (IM3Compilation o, u16 i_destSlot)  // NoPushPop
861
865
{
862
865
    M3Result result;
863
864
865
    i16 stackTop = GetStackTopIndex (o);
865
865
_   (CopyStackIndexToSlot (o, i_destSlot, (u16) stackTop));
866
867
865
    _catch: return result;
868
865
}
869
870
871
// a copy-on-write strategy is used with locals. when a get local occurs, it's not copied anywhere. the stack
872
// entry just has a index pointer to that local memory slot.
873
// then, when a previously referenced local is set, the current value needs to be preserved for those references
874
875
// TODO: consider getting rid of these specialized operations: PreserveSetSlot & PreserveCopySlot.
876
// They likely just take up space (which seems to reduce performance) without improving performance.
877
static
878
M3Result  PreservedCopyTopSlot  (IM3Compilation o, u16 i_destSlot, u16 i_preserveSlot)
879
0
{
880
0
    M3Result result = m3Err_none;             d_m3Assert (i_destSlot != i_preserveSlot);
881
882
0
    IM3Operation op;
883
884
0
    u8 type = GetStackTopType (o);
885
886
0
    if (IsStackTopInRegister (o))
887
0
    {
888
0
        op = c_preserveSetSlot [type];
889
0
    }
890
0
    else op = Is64BitType (type) ? op_PreserveCopySlot_64 : op_PreserveCopySlot_32;
891
892
0
_   (EmitOp (o, op));
893
0
    EmitSlotOffset (o, i_destSlot);
894
895
0
    if (IsStackTopInSlot (o))
896
0
        EmitSlotOffset (o, GetStackTopSlotNumber (o));
897
898
0
    EmitSlotOffset (o, i_preserveSlot);
899
900
0
    _catch: return result;
901
0
}
902
903
static
904
M3Result  CopyStackTopToRegister  (IM3Compilation o, bool i_updateStack)
905
257
{
906
257
    M3Result result = m3Err_none;
907
908
257
    if (IsStackTopInSlot (o))
909
257
    {
910
257
        u8 type = GetStackTopType (o);
911
912
257
_       (PreserveRegisterIfOccupied (o, type));
913
914
257
        IM3Operation op = c_setRegisterOps [type];
915
916
257
_       (EmitOp (o, op));
917
257
        EmitSlotOffset (o, GetStackTopSlotNumber (o));
918
919
257
        if (i_updateStack)
920
0
        {
921
0
_           (PopType (o, type));
922
0
_           (PushRegister (o, type));
923
0
        }
924
257
    }
925
926
257
    _catch: return result;
927
257
}
928
929
930
// if local is unreferenced, o_preservedSlotNumber will be equal to localIndex on return
931
static
932
M3Result  FindReferencedLocalWithinCurrentBlock  (IM3Compilation o, u16 * o_preservedSlotNumber, u32 i_localSlot)
933
19.1k
{
934
19.1k
    M3Result result = m3Err_none;
935
936
19.1k
    IM3CompilationScope scope = & o->block;
937
19.1k
    u16 startIndex = scope->blockStackIndex;
938
939
19.6k
    while (scope->opcode == c_waOp_block)
940
556
    {
941
556
        scope = scope->outer;
942
556
        if (not scope)
943
0
            break;
944
945
556
        startIndex = scope->blockStackIndex;
946
556
    }
947
948
19.1k
    * o_preservedSlotNumber = (u16) i_localSlot;
949
950
593k
    for (u32 i = startIndex; i < o->stackIndex; ++i)
951
573k
    {
952
573k
        if (o->wasmStack [i] == i_localSlot)
953
0
        {
954
0
            if (* o_preservedSlotNumber == i_localSlot)
955
0
            {
956
0
                u8 type = GetStackTypeFromBottom (o, i);                    d_m3Assert (type != c_m3Type_none)
957
958
0
_               (AllocateSlots (o, o_preservedSlotNumber, type));
959
0
            }
960
0
            else
961
0
_               (IncrementSlotUsageCount (o, * o_preservedSlotNumber));
962
963
0
            o->wasmStack [i] = * o_preservedSlotNumber;
964
0
        }
965
573k
    }
966
967
19.1k
    _catch: return result;
968
19.1k
}
969
970
static
971
M3Result  GetBlockScope  (IM3Compilation o, IM3CompilationScope * o_scope, u32 i_depth)
972
260
{
973
260
    M3Result result = m3Err_none;
974
975
260
    IM3CompilationScope scope = & o->block;
976
977
262
    while (i_depth--)
978
5
    {
979
5
        scope = scope->outer;
980
5
        _throwif ("invalid block depth", not scope);
981
2
    }
982
983
257
    * o_scope = scope;
984
985
260
    _catch:
986
260
    return result;
987
257
}
988
989
static
990
M3Result  CopyStackSlotsR  (IM3Compilation o, u16 i_targetSlotStackIndex, u16 i_stackIndex, u16 i_endStackIndex, u16 i_tempSlot)
991
17.0k
{
992
17.0k
    M3Result result = m3Err_none;
993
994
17.0k
    if (i_stackIndex < i_endStackIndex)
995
16.7k
    {
996
16.7k
        u16 srcSlot = GetSlotForStackIndex (o, i_stackIndex);
997
998
16.7k
        u8 type = GetStackTypeFromBottom (o, i_stackIndex);
999
16.7k
        u16 numSlots = GetTypeNumSlots (type);
1000
16.7k
        u16 extraSlot = numSlots - 1;
1001
1002
16.7k
        u16 targetSlot = GetSlotForStackIndex (o, i_targetSlotStackIndex);
1003
1004
16.7k
        u16 preserveIndex = i_stackIndex;
1005
16.7k
        u16 collisionSlot = srcSlot;
1006
1007
16.7k
        if (targetSlot != srcSlot)
1008
15.6k
        {
1009
            // search for collisions
1010
15.6k
            u16 checkIndex = i_stackIndex + 1;
1011
64.0k
            while (checkIndex < i_endStackIndex)
1012
60.7k
            {
1013
60.7k
                u16 otherSlot1 = GetSlotForStackIndex (o, checkIndex);
1014
60.7k
                u16 otherSlot2 = GetExtraSlotForStackIndex (o, checkIndex);
1015
1016
60.7k
                if (targetSlot == otherSlot1 or
1017
48.4k
                    targetSlot == otherSlot2 or
1018
48.4k
                    targetSlot + extraSlot == otherSlot1)
1019
12.3k
                {
1020
12.3k
                    _throwif (m3Err_functionStackOverflow, i_tempSlot >= d_m3MaxFunctionSlots);
1021
1022
12.3k
_                   (CopyStackIndexToSlot (o, i_tempSlot, checkIndex));
1023
12.3k
                    o->wasmStack [checkIndex] = i_tempSlot;
1024
12.3k
                    i_tempSlot += GetTypeNumSlots (c_m3Type_i64);
1025
12.3k
                    TouchSlot (o, i_tempSlot - 1);
1026
1027
                    // restore this on the way back down
1028
12.3k
                    preserveIndex = checkIndex;
1029
12.3k
                    collisionSlot = otherSlot1;
1030
1031
12.3k
                    break;
1032
12.3k
                }
1033
1034
48.3k
                ++checkIndex;
1035
48.3k
            }
1036
1037
15.6k
_           (CopyStackIndexToSlot (o, targetSlot, i_stackIndex));                                               m3log (compile, " copying slot: %d to slot: %d", srcSlot, targetSlot);
1038
15.6k
            o->wasmStack [i_stackIndex] = targetSlot;
1039
1040
15.6k
        }
1041
1042
16.7k
_       (CopyStackSlotsR (o, i_targetSlotStackIndex + 1, i_stackIndex + 1, i_endStackIndex, i_tempSlot));
1043
1044
        // restore the stack state
1045
16.6k
        o->wasmStack [i_stackIndex] = srcSlot;
1046
16.6k
        o->wasmStack [preserveIndex] = collisionSlot;
1047
16.6k
    }
1048
1049
17.0k
    _catch:
1050
17.0k
    return result;
1051
17.0k
}
1052
1053
static
1054
M3Result  ResolveBlockResults  (IM3Compilation o, IM3CompilationScope i_targetBlock, bool i_isBranch)
1055
248
{
1056
248
    M3Result result = m3Err_none;                                   if (d_m3LogWasmStack) dump_type_stack (o);
1057
1058
248
    bool isLoop = (i_targetBlock->opcode == c_waOp_loop and i_isBranch);
1059
1060
248
    u16 numParams = GetFuncTypeNumParams (i_targetBlock->type);
1061
248
    u16 numResults = GetFuncTypeNumResults (i_targetBlock->type);
1062
1063
248
    u16 slotRecords = i_targetBlock->exitStackIndex;
1064
1065
248
    u16 numValues;
1066
1067
248
    if (not isLoop)
1068
248
    {
1069
248
        numValues = numResults;
1070
248
        slotRecords += numParams;
1071
248
    }
1072
0
    else numValues = numParams;
1073
1074
248
    u16 blockHeight = GetNumBlockValuesOnStack (o);
1075
1076
248
    _throwif (m3Err_typeCountMismatch, i_isBranch ? (blockHeight < numValues) : (blockHeight != numValues));
1077
1078
248
    if (numValues)
1079
248
    {
1080
248
        u16 endIndex = GetStackTopIndex (o) + 1;
1081
248
        u16 numRemValues = numValues;
1082
1083
        // The last result is taken from _fp0. See PushBlockResults.
1084
248
        if (not isLoop and IsFpType (GetStackTopType (o)))
1085
248
        {
1086
248
_           (CopyStackTopToRegister (o, false));
1087
248
            --endIndex;
1088
248
            --numRemValues;
1089
248
        }
1090
1091
        // TODO: tempslot affects maxStackSlots, so can grow unnecess each time.
1092
248
        u16 tempSlot = o->maxStackSlots;// GetMaxUsedSlotPlusOne (o); doesn't work cause can collide with slotRecords
1093
248
        AlignSlotToType (& tempSlot, c_m3Type_i64);
1094
1095
248
_       (CopyStackSlotsR (o, slotRecords, endIndex - numRemValues, endIndex, tempSlot));
1096
1097
242
        if (d_m3LogWasmStack) dump_type_stack (o);
1098
242
    }
1099
1100
248
    _catch: return result;
1101
248
}
1102
1103
1104
static
1105
M3Result  ReturnValues  (IM3Compilation o, IM3CompilationScope i_functionBlock, bool i_isBranch)
1106
3
{
1107
3
    M3Result result = m3Err_none;                                               if (d_m3LogWasmStack) dump_type_stack (o);
1108
1109
3
    u16 numReturns = GetFuncTypeNumResults (i_functionBlock->type);     // could just o->function too...
1110
3
    u16 blockHeight = GetNumBlockValuesOnStack (o);
1111
1112
3
    if (not IsStackPolymorphic (o))
1113
3
        _throwif (m3Err_typeCountMismatch, i_isBranch ? (blockHeight < numReturns) : (blockHeight != numReturns));
1114
1115
3
    if (numReturns)
1116
3
    {
1117
        // return slots like args are 64-bit aligned
1118
3
        u16 returnSlot = numReturns * c_ioSlotCount;
1119
3
        u16 stackTop = GetStackTopIndex (o);
1120
1121
152
        for (u16 i = 0; i < numReturns; ++i)
1122
150
        {
1123
150
            u8 returnType = GetFuncTypeResultType (i_functionBlock->type, numReturns - 1 - i);
1124
1125
150
            u8 stackType = GetStackTypeFromTop (o, i);  // using FromTop so that only dynamic items are checked
1126
1127
150
            if (IsStackPolymorphic (o) and stackType == c_m3Type_none)
1128
149
                stackType = returnType;
1129
1130
150
            _throwif (m3Err_typeMismatch, returnType != stackType);
1131
1132
149
            if (not IsStackPolymorphic (o))
1133
0
            {
1134
0
                returnSlot -= c_ioSlotCount;
1135
0
_               (CopyStackIndexToSlot (o, returnSlot, stackTop--));
1136
0
            }
1137
149
        }
1138
1139
2
        if (not i_isBranch)
1140
2
        {
1141
151
            while (numReturns--)
1142
149
_               (Pop (o));
1143
2
        }
1144
2
    }
1145
1146
3
    _catch: return result;
1147
3
}
1148
1149
1150
//-------------------------------------------------------------------------------------------------------------------------
1151
1152
static
1153
M3Result  Compile_Const_i32  (IM3Compilation o, m3opcode_t i_opcode)
1154
6.17k
{
1155
6.17k
    M3Result result;
1156
1157
6.17k
    i32 value;
1158
6.17k
_   (ReadLEB_i32 (& value, & o->wasm, o->wasmEnd));
1159
6.17k
_   (PushConst (o, value, c_m3Type_i32));                       m3log (compile, d_indent " (const i32 = %" PRIi32 ")", get_indention_string (o), value);
1160
6.17k
    _catch: return result;
1161
6.17k
}
1162
1163
static
1164
M3Result  Compile_Const_i64  (IM3Compilation o, m3opcode_t i_opcode)
1165
13.6k
{
1166
13.6k
    M3Result result;
1167
1168
13.6k
    i64 value;
1169
13.6k
_   (ReadLEB_i64 (& value, & o->wasm, o->wasmEnd));
1170
13.6k
_   (PushConst (o, value, c_m3Type_i64));                       m3log (compile, d_indent " (const i64 = %" PRIi64 ")", get_indention_string (o), value);
1171
13.6k
    _catch: return result;
1172
13.6k
}
1173
1174
1175
#if d_m3ImplementFloat
1176
static
1177
M3Result  Compile_Const_f32  (IM3Compilation o, m3opcode_t i_opcode)
1178
589
{
1179
589
    M3Result result;
1180
1181
589
    union { u32 u; f32 f; } value = { 0 };
1182
1183
589
_   (Read_f32 (& value.f, & o->wasm, o->wasmEnd));              m3log (compile, d_indent " (const f32 = %" PRIf32 ")", get_indention_string (o), value.f);
1184
589
_   (PushConst (o, value.u, c_m3Type_f32));
1185
1186
589
    _catch: return result;
1187
589
}
1188
1189
static
1190
M3Result  Compile_Const_f64  (IM3Compilation o, m3opcode_t i_opcode)
1191
103
{
1192
103
    M3Result result;
1193
1194
103
    union { u64 u; f64 f; } value = { 0 };
1195
1196
103
_   (Read_f64 (& value.f, & o->wasm, o->wasmEnd));              m3log (compile, d_indent " (const f64 = %" PRIf64 ")", get_indention_string (o), value.f);
1197
103
_   (PushConst (o, value.u, c_m3Type_f64));
1198
1199
103
    _catch: return result;
1200
103
}
1201
#endif
1202
1203
#if d_m3CascadedOpcodes
1204
1205
static
1206
M3Result  Compile_ExtendedOpcode  (IM3Compilation o, m3opcode_t i_opcode)
1207
1
{
1208
1
_try {
1209
1
    u8 opcode;
1210
1
_   (Read_u8 (& opcode, & o->wasm, o->wasmEnd));             m3log (compile, d_indent " (FC: %" PRIi32 ")", get_indention_string (o), opcode);
1211
1212
1
    i_opcode = (i_opcode << 8) | opcode;
1213
1214
    //printf("Extended opcode: 0x%x\n", i_opcode);
1215
1216
1
    IM3OpInfo opInfo = GetOpInfo (i_opcode);
1217
1
    _throwif (m3Err_unknownOpcode, not opInfo);
1218
1219
1
    M3Compiler compiler = opInfo->compiler;
1220
1
    _throwif (m3Err_noCompiler, not compiler);
1221
1222
1
_   ((* compiler) (o, i_opcode));
1223
1224
1
    o->previousOpcode = i_opcode;
1225
1226
1
    } _catch: return result;
1227
1
}
1228
#endif
1229
1230
static
1231
M3Result  Compile_Return  (IM3Compilation o, m3opcode_t i_opcode)
1232
19
{
1233
19
    M3Result result = m3Err_none;
1234
1235
19
    if (not IsStackPolymorphic (o))
1236
0
    {
1237
0
        IM3CompilationScope functionScope;
1238
0
_       (GetBlockScope (o, & functionScope, o->block.depth));
1239
1240
0
_       (ReturnValues (o, functionScope, true));
1241
1242
0
_       (EmitOp (o, op_Return));
1243
1244
0
_       (SetStackPolymorphic (o));
1245
0
    }
1246
1247
19
    _catch: return result;
1248
19
}
1249
1250
static
1251
M3Result  ValidateBlockEnd  (IM3Compilation o)
1252
687
{
1253
687
    M3Result result = m3Err_none;
1254
/*
1255
    u16 numResults = GetFuncTypeNumResults (o->block.type);
1256
    u16 blockHeight = GetNumBlockValuesOnStack (o);
1257
1258
    if (IsStackPolymorphic (o))
1259
    {
1260
    }
1261
    else
1262
    {
1263
    }
1264
1265
687
    _catch: */ return result;
1266
687
}
1267
1268
static
1269
M3Result  Compile_End  (IM3Compilation o, m3opcode_t i_opcode)
1270
493
{
1271
493
    M3Result result = m3Err_none;                   //dump_type_stack (o);
1272
1273
    // function end:
1274
493
    if (o->block.depth == 0)
1275
471
    {
1276
471
        ValidateBlockEnd (o);
1277
1278
//      if (not IsStackPolymorphic (o))
1279
471
        {
1280
471
            if (o->function)
1281
3
            {
1282
3
_               (ReturnValues (o, & o->block, false));
1283
2
            }
1284
1285
470
_           (EmitOp (o, op_Return));
1286
470
        }
1287
470
    }
1288
1289
493
    _catch: return result;
1290
493
}
1291
1292
1293
static
1294
M3Result  Compile_SetLocal  (IM3Compilation o, m3opcode_t i_opcode)
1295
1
{
1296
1
    M3Result result;
1297
1298
1
    u32 localIndex;
1299
1
_   (ReadLEB_u32 (& localIndex, & o->wasm, o->wasmEnd));             //  printf ("--- set local: %d \n", localSlot);
1300
1301
1
    if (localIndex < GetFunctionNumArgsAndLocals (o->function))
1302
1
    {
1303
1
        u16 localSlot = GetSlotForStackIndex (o, localIndex);
1304
1305
1
        u16 preserveSlot;
1306
1
_       (FindReferencedLocalWithinCurrentBlock (o, & preserveSlot, localSlot));  // preserve will be different than local, if referenced
1307
1308
1
        if (preserveSlot == localSlot)
1309
1
_           (CopyStackTopToSlot (o, localSlot))
1310
0
        else
1311
0
_           (PreservedCopyTopSlot (o, localSlot, preserveSlot))
1312
1313
1
        if (i_opcode != c_waOp_teeLocal)
1314
1
_           (Pop (o));
1315
1
    }
1316
1
    else _throw ("local index out of bounds");
1317
1318
1
    _catch: return result;
1319
1
}
1320
1321
static
1322
M3Result  Compile_GetLocal  (IM3Compilation o, m3opcode_t i_opcode)
1323
2
{
1324
2
_try {
1325
1326
2
    u32 localIndex;
1327
2
_   (ReadLEB_u32 (& localIndex, & o->wasm, o->wasmEnd));
1328
1329
2
    if (localIndex >= GetFunctionNumArgsAndLocals (o->function))
1330
2
        _throw ("local index out of bounds");
1331
1332
2
    u8 type = GetStackTypeFromBottom (o, localIndex);
1333
2
    u16 slot = GetSlotForStackIndex (o, localIndex);
1334
1335
2
_   (Push (o, type, slot));
1336
1337
2
    } _catch: return result;
1338
2
}
1339
1340
static
1341
M3Result  Compile_GetGlobal  (IM3Compilation o, M3Global * i_global)
1342
859
{
1343
859
    M3Result result;
1344
1345
859
    IM3Operation op = Is64BitType (i_global->type) ? op_GetGlobal_s64 : op_GetGlobal_s32;
1346
859
_   (EmitOp (o, op));
1347
859
    EmitPointer (o, & i_global->i64Value);
1348
859
_   (PushAllocatedSlotAndEmit (o, i_global->type));
1349
1350
859
    _catch: return result;
1351
859
}
1352
1353
static
1354
M3Result  Compile_SetGlobal  (IM3Compilation o, M3Global * i_global)
1355
0
{
1356
0
    M3Result result = m3Err_none;
1357
1358
0
    if (i_global->isMutable)
1359
0
    {
1360
0
        IM3Operation op;
1361
0
        u8 type = GetStackTopType (o);
1362
1363
0
        if (IsStackTopInRegister (o))
1364
0
        {
1365
0
            op = c_setGlobalOps [type];
1366
0
        }
1367
0
        else op = Is64BitType (type) ? op_SetGlobal_s64 : op_SetGlobal_s32;
1368
1369
0
_      (EmitOp (o, op));
1370
0
        EmitPointer (o, & i_global->i64Value);
1371
1372
0
        if (IsStackTopInSlot (o))
1373
0
            EmitSlotOffset (o, GetStackTopSlotNumber (o));
1374
1375
0
_      (Pop (o));
1376
0
    }
1377
0
    else _throw (m3Err_settingImmutableGlobal);
1378
1379
0
    _catch: return result;
1380
0
}
1381
1382
static
1383
M3Result  Compile_GetSetGlobal  (IM3Compilation o, m3opcode_t i_opcode)
1384
860
{
1385
860
    M3Result result = m3Err_none;
1386
1387
860
    u32 globalIndex;
1388
860
_   (ReadLEB_u32 (& globalIndex, & o->wasm, o->wasmEnd));
1389
1390
860
    if (globalIndex < o->module->numGlobals)
1391
859
    {
1392
859
        if (o->module->globals)
1393
859
        {
1394
859
            M3Global * global = & o->module->globals [globalIndex];
1395
1396
859
_           ((i_opcode == c_waOp_getGlobal) ? Compile_GetGlobal (o, global) : Compile_SetGlobal (o, global));
1397
859
        }
1398
859
        else _throw (ErrorCompile (m3Err_globalMemoryNotAllocated, o, "module '%s' is missing global memory", o->module->name));
1399
859
    }
1400
859
    else _throw (m3Err_globaIndexOutOfBounds);
1401
1402
860
    _catch: return result;
1403
859
}
1404
1405
static
1406
void  EmitPatchingBranchPointer  (IM3Compilation o, IM3CompilationScope i_scope)
1407
242
{
1408
242
    pc_t patch = EmitPointer (o, i_scope->patches);                     m3log (compile, "branch patch required at: %p", patch);
1409
242
    i_scope->patches = patch;
1410
242
}
1411
1412
static
1413
M3Result  EmitPatchingBranch  (IM3Compilation o, IM3CompilationScope i_scope)
1414
242
{
1415
242
    M3Result result = m3Err_none;
1416
1417
242
_   (EmitOp (o, op_Branch));
1418
242
    EmitPatchingBranchPointer (o, i_scope);
1419
1420
242
    _catch: return result;
1421
242
}
1422
1423
static
1424
M3Result  Compile_Branch  (IM3Compilation o, m3opcode_t i_opcode)
1425
9
{
1426
9
    M3Result result;
1427
1428
9
    u32 depth;
1429
9
_   (ReadLEB_u32 (& depth, & o->wasm, o->wasmEnd));
1430
1431
9
    IM3CompilationScope scope;
1432
9
_   (GetBlockScope (o, & scope, depth));
1433
1434
    // branch target is a loop (continue)
1435
9
    if (scope->opcode == c_waOp_loop)
1436
1
    {
1437
1
        if (i_opcode == c_waOp_branchIf)
1438
0
        {
1439
0
            if (GetFuncTypeNumParams (scope->type))
1440
0
            {
1441
0
                IM3Operation op = IsStackTopInRegister (o) ? op_BranchIfPrologue_r : op_BranchIfPrologue_s;
1442
1443
0
_               (EmitOp (o, op));
1444
0
_               (EmitSlotNumOfStackTopAndPop (o));
1445
1446
0
                pc_t * jumpTo = (pc_t *) ReservePointer (o);
1447
1448
0
_               (ResolveBlockResults (o, scope, /* isBranch: */ true));
1449
1450
0
_               (EmitOp (o, op_ContinueLoop));
1451
0
                EmitPointer (o, scope->pc);
1452
1453
0
                * jumpTo = GetPC (o);
1454
0
            }
1455
0
            else
1456
0
            {
1457
                // move the condition to a register
1458
0
_               (CopyStackTopToRegister (o, false));
1459
0
_               (PopType (o, c_m3Type_i32));
1460
1461
0
_               (EmitOp (o, op_ContinueLoopIf));
1462
0
                EmitPointer (o, scope->pc);
1463
0
            }
1464
1465
//          dump_type_stack(o);
1466
0
        }
1467
1
        else // is c_waOp_branch
1468
1
        {
1469
1
    _       (EmitOp (o, op_ContinueLoop));
1470
1
            EmitPointer (o, scope->pc);
1471
1
            o->block.isPolymorphic = true;
1472
1
        }
1473
1
    }
1474
8
    else // forward branch
1475
8
    {
1476
8
        pc_t * jumpTo = NULL;
1477
1478
8
        bool isReturn = (scope->depth == 0);
1479
8
        bool targetHasResults = GetFuncTypeNumResults (scope->type);
1480
1481
8
        if (i_opcode == c_waOp_branchIf)
1482
8
        {
1483
8
            if (targetHasResults or isReturn)
1484
8
            {
1485
8
                IM3Operation op = IsStackTopInRegister (o) ? op_BranchIfPrologue_r : op_BranchIfPrologue_s;
1486
1487
8
    _           (EmitOp (o, op));
1488
8
    _           (EmitSlotNumOfStackTopAndPop (o)); // condition
1489
1490
                // this is continuation point, if the branch isn't taken
1491
8
                jumpTo = (pc_t *) ReservePointer (o);
1492
8
            }
1493
0
            else
1494
0
            {
1495
0
                IM3Operation op = IsStackTopInRegister (o) ? op_BranchIf_r : op_BranchIf_s;
1496
1497
0
    _           (EmitOp (o, op));
1498
0
    _           (EmitSlotNumOfStackTopAndPop (o)); // condition
1499
1500
0
                EmitPatchingBranchPointer (o, scope);
1501
0
                goto _catch;
1502
0
            }
1503
8
        }
1504
1505
8
        if (not IsStackPolymorphic (o))
1506
0
        {
1507
0
            if (isReturn)
1508
0
            {
1509
0
_               (ReturnValues (o, scope, true));
1510
0
_               (EmitOp (o, op_Return));
1511
0
            }
1512
0
            else
1513
0
            {
1514
0
_               (ResolveBlockResults (o, scope, true));
1515
0
_               (EmitPatchingBranch (o, scope));
1516
0
            }
1517
0
        }
1518
1519
8
        if (jumpTo)
1520
8
        {
1521
8
            * jumpTo = GetPC (o);
1522
8
        }
1523
1524
8
        if (i_opcode == c_waOp_branch)
1525
8
_           (SetStackPolymorphic (o));
1526
8
    }
1527
1528
9
    _catch: return result;
1529
9
}
1530
1531
static
1532
M3Result  Compile_BranchTable  (IM3Compilation o, m3opcode_t i_opcode)
1533
9
{
1534
9
_try {
1535
9
    u32 targetCount;
1536
9
_   (ReadLEB_u32 (& targetCount, & o->wasm, o->wasmEnd));
1537
1538
9
_   (PreserveRegisterIfOccupied (o, c_m3Type_i64));         // move branch operand to a slot
1539
9
    u16 slot = GetStackTopSlotNumber (o);
1540
9
_   (Pop (o));
1541
1542
    // OPTZ: according to spec: "forward branches that target a control instruction with a non-empty
1543
    // result type consume matching operands first and push them back on the operand stack after unwinding"
1544
    // So, this move-to-reg is only necessary if the target scopes have a type.
1545
1546
9
    u32 numCodeLines = targetCount + 4; // 3 => IM3Operation + slot + target_count + default_target
1547
9
_   (EnsureCodePageNumLines (o, numCodeLines));
1548
1549
9
_   (EmitOp (o, op_BranchTable));
1550
9
    EmitSlotOffset (o, slot);
1551
9
    EmitConstant32 (o, targetCount);
1552
1553
9
    IM3CodePage continueOpPage = NULL;
1554
1555
9
    ++targetCount; // include default
1556
251
    for (u32 i = 0; i < targetCount; ++i)
1557
251
    {
1558
251
        u32 target;
1559
251
_       (ReadLEB_u32 (& target, & o->wasm, o->wasmEnd));
1560
1561
251
        IM3CompilationScope scope;
1562
251
_       (GetBlockScope (o, & scope, target));
1563
1564
        // TODO: don't need codepage rigmarole for
1565
        // no-param forward-branch targets
1566
1567
248
_       (AcquireCompilationCodePage (o, & continueOpPage));
1568
1569
248
        pc_t startPC = GetPagePC (continueOpPage);
1570
248
        IM3CodePage savedPage = o->page;
1571
248
        o->page = continueOpPage;
1572
1573
248
        if (scope->opcode == c_waOp_loop)
1574
0
        {
1575
0
_           (ResolveBlockResults (o, scope, true));
1576
1577
0
_           (EmitOp (o, op_ContinueLoop));
1578
0
            EmitPointer (o, scope->pc);
1579
0
        }
1580
248
        else
1581
248
        {
1582
            // TODO: this could be fused with equivalent targets
1583
248
            if (not IsStackPolymorphic (o))
1584
248
            {
1585
248
                if (scope->depth == 0)
1586
0
                {
1587
0
_                   (ReturnValues (o, scope, true));
1588
0
_                   (EmitOp (o, op_Return));
1589
0
                }
1590
248
                else
1591
248
                {
1592
248
_                   (ResolveBlockResults (o, scope, true));
1593
1594
242
_                   (EmitPatchingBranch (o, scope));
1595
242
                }
1596
248
            }
1597
248
        }
1598
1599
242
        ReleaseCompilationCodePage (o);     // FIX: continueOpPage can get lost if thrown
1600
242
        o->page = savedPage;
1601
1602
242
        EmitPointer (o, startPC);
1603
242
    }
1604
1605
0
_   (SetStackPolymorphic (o));
1606
1607
0
    }
1608
1609
9
    _catch: return result;
1610
0
}
1611
1612
static
1613
M3Result  CompileCallArgsAndReturn  (IM3Compilation o, u16 * o_stackOffset, IM3FuncType i_type, bool i_isIndirect)
1614
160
{
1615
160
_try {
1616
1617
160
    u16 topSlot = GetMaxUsedSlotPlusOne (o);
1618
1619
    // force use of at least one stack slot; this is to help ensure
1620
    // the m3 stack overflows (and traps) before the native stack can overflow.
1621
    // e.g. see Wasm spec test 'runaway' in call.wast
1622
160
    topSlot = M3_MAX (1, topSlot);
1623
1624
    // stack frame is 64-bit aligned
1625
160
    AlignSlotToType (& topSlot, c_m3Type_i64);
1626
1627
160
    * o_stackOffset = topSlot;
1628
1629
    // wait to pop this here so that topSlot search is correct
1630
160
    if (i_isIndirect)
1631
160
_       (Pop (o));
1632
1633
160
    u16 numArgs = GetFuncTypeNumParams (i_type);
1634
160
    u16 numRets = GetFuncTypeNumResults (i_type);
1635
1636
160
    u16 argTop = topSlot + (numArgs + numRets) * c_ioSlotCount;
1637
1638
1.02k
    while (numArgs--)
1639
864
    {
1640
864
_       (CopyStackTopToSlot (o, argTop -= c_ioSlotCount));
1641
864
_       (Pop (o));
1642
864
    }
1643
1644
160
    u16 i = 0;
1645
11.4k
    while (numRets--)
1646
11.2k
    {
1647
11.2k
        u8 type = GetFuncTypeResultType (i_type, i++);
1648
1649
11.2k
_       (Push (o, type, topSlot));
1650
11.2k
        MarkSlotsAllocatedByType (o, topSlot, type);
1651
1652
11.2k
        topSlot += c_ioSlotCount;
1653
11.2k
    }
1654
1655
160
    } _catch: return result;
1656
160
}
1657
1658
static
1659
M3Result  Compile_Call  (IM3Compilation o, m3opcode_t i_opcode)
1660
153
{
1661
153
_try {
1662
153
    u32 functionIndex;
1663
153
_   (ReadLEB_u32 (& functionIndex, & o->wasm, o->wasmEnd));
1664
1665
153
    IM3Function function = Module_GetFunction (o->module, functionIndex);
1666
1667
153
    if (function)
1668
152
    {                                                                   m3log (compile, d_indent " (func= [%d] '%s'; args= %d)",
1669
152
                                                                                get_indention_string (o), functionIndex, m3_GetFunctionName (function), function->funcType->numArgs);
1670
152
        if (function->module)
1671
152
        {
1672
152
            u16 slotTop;
1673
152
_           (CompileCallArgsAndReturn (o, & slotTop, function->funcType, false));
1674
1675
152
            IM3Operation op;
1676
152
            const void * operand;
1677
1678
152
            if (function->compiled)
1679
0
            {
1680
0
                op = op_Call;
1681
0
                operand = function->compiled;
1682
0
            }
1683
152
            else
1684
152
            {
1685
152
                op = op_Compile;
1686
152
                operand = function;
1687
152
            }
1688
1689
152
_           (EmitOp     (o, op));
1690
152
            EmitPointer (o, operand);
1691
152
            EmitSlotOffset  (o, slotTop);
1692
152
        }
1693
0
        else
1694
0
        {
1695
0
            _throw (ErrorCompile (m3Err_functionImportMissing, o, "'%s.%s'", GetFunctionImportModuleName (function), m3_GetFunctionName (function)));
1696
0
        }
1697
152
    }
1698
152
    else _throw (m3Err_functionLookupFailed);
1699
1700
153
    } _catch: return result;
1701
152
}
1702
1703
static
1704
M3Result  Compile_CallIndirect  (IM3Compilation o, m3opcode_t i_opcode)
1705
9
{
1706
9
_try {
1707
9
    u32 typeIndex;
1708
9
_   (ReadLEB_u32 (& typeIndex, & o->wasm, o->wasmEnd));
1709
1710
9
    u32 tableIndex;
1711
9
_   (ReadLEB_u32 (& tableIndex, & o->wasm, o->wasmEnd));
1712
1713
9
    _throwif ("function call type index out of range", typeIndex >= o->module->numFuncTypes);
1714
1715
8
    if (IsStackTopInRegister (o))
1716
8
_       (PreserveRegisterIfOccupied (o, c_m3Type_i32));
1717
1718
8
    u16 tableIndexSlot = GetStackTopSlotNumber (o);
1719
1720
8
    u16 execTop;
1721
8
    IM3FuncType type = o->module->funcTypes [typeIndex];
1722
8
_   (CompileCallArgsAndReturn (o, & execTop, type, true));
1723
1724
7
_   (EmitOp         (o, op_CallIndirect));
1725
7
    EmitSlotOffset  (o, tableIndexSlot);
1726
7
    EmitPointer     (o, o->module);
1727
7
    EmitPointer     (o, type);              // TODO: unify all types in M3Environment
1728
7
    EmitSlotOffset  (o, execTop);
1729
1730
9
} _catch:
1731
9
    return result;
1732
7
}
1733
1734
static
1735
M3Result  Compile_Memory_Size  (IM3Compilation o, m3opcode_t i_opcode)
1736
1
{
1737
1
    M3Result result;
1738
1739
1
    i8 reserved;
1740
1
_   (ReadLEB_i7 (& reserved, & o->wasm, o->wasmEnd));
1741
1742
0
_   (PreserveRegisterIfOccupied (o, c_m3Type_i32));
1743
1744
0
_   (EmitOp     (o, op_MemSize));
1745
1746
0
_   (PushRegister (o, c_m3Type_i32));
1747
1748
1
    _catch: return result;
1749
0
}
1750
1751
static
1752
M3Result  Compile_Memory_Grow  (IM3Compilation o, m3opcode_t i_opcode)
1753
8
{
1754
8
    M3Result result;
1755
1756
8
    i8 reserved;
1757
8
_   (ReadLEB_i7 (& reserved, & o->wasm, o->wasmEnd));
1758
1759
8
_   (CopyStackTopToRegister (o, false));
1760
8
_   (PopType (o, c_m3Type_i32));
1761
1762
8
_   (EmitOp     (o, op_MemGrow));
1763
1764
8
_   (PushRegister (o, c_m3Type_i32));
1765
1766
8
    _catch: return result;
1767
7
}
1768
1769
static
1770
M3Result  Compile_Memory_CopyFill  (IM3Compilation o, m3opcode_t i_opcode)
1771
1
{
1772
1
    M3Result result = m3Err_none;
1773
1774
1
    u32 sourceMemoryIdx, targetMemoryIdx;
1775
1
    IM3Operation op;
1776
1
    if (i_opcode == c_waOp_memoryCopy)
1777
1
    {
1778
1
_       (ReadLEB_u32 (& sourceMemoryIdx, & o->wasm, o->wasmEnd));
1779
1
        op = op_MemCopy;
1780
1
    }
1781
0
    else op = op_MemFill;
1782
1783
1
_   (ReadLEB_u32 (& targetMemoryIdx, & o->wasm, o->wasmEnd));
1784
1785
1
_   (CopyStackTopToRegister (o, false));
1786
1787
1
_   (EmitOp  (o, op));
1788
1
_   (PopType (o, c_m3Type_i32));
1789
1
_   (EmitSlotNumOfStackTopAndPop (o));
1790
1
_   (EmitSlotNumOfStackTopAndPop (o));
1791
1792
1
    _catch: return result;
1793
1
}
1794
1795
1796
static
1797
M3Result  ReadBlockType  (IM3Compilation o, IM3FuncType * o_blockType)
1798
521
{
1799
521
    M3Result result;
1800
1801
521
    i64 type;
1802
521
_   (ReadLebSigned (& type, 33, & o->wasm, o->wasmEnd));
1803
1804
517
    if (type < 0)
1805
7
    {
1806
7
        u8 valueType;
1807
7
_       (NormalizeType (&valueType, type));                                m3log (compile, d_indent " (type: %s)", get_indention_string (o), c_waTypes [valueType]);
1808
7
        *o_blockType = o->module->environment->retFuncTypes[valueType];
1809
7
    }
1810
510
    else
1811
510
    {
1812
510
        _throwif("func type out of bounds", type >= o->module->numFuncTypes);
1813
505
        *o_blockType = o->module->funcTypes[type];                         m3log (compile, d_indent " (type: %s)", get_indention_string (o), SPrintFuncTypeSignature (*o_blockType));
1814
505
    }
1815
521
    _catch: return result;
1816
517
}
1817
1818
static
1819
M3Result  PreserveArgsAndLocals  (IM3Compilation o)
1820
521
{
1821
521
    M3Result result = m3Err_none;
1822
1823
521
    if (o->stackIndex > o->stackFirstDynamicIndex)
1824
508
    {
1825
508
        u32 numArgsAndLocals = GetFunctionNumArgsAndLocals (o->function);
1826
1827
19.6k
        for (u32 i = 0; i < numArgsAndLocals; ++i)
1828
19.1k
        {
1829
19.1k
            u16 slot = GetSlotForStackIndex (o, i);
1830
1831
19.1k
            u16 preservedSlotNumber;
1832
19.1k
_           (FindReferencedLocalWithinCurrentBlock (o, & preservedSlotNumber, slot));
1833
1834
19.1k
            if (preservedSlotNumber != slot)
1835
0
            {
1836
0
                u8 type = GetStackTypeFromBottom (o, i);                    d_m3Assert (type != c_m3Type_none)
1837
0
                IM3Operation op = Is64BitType (type) ? op_CopySlot_64 : op_CopySlot_32;
1838
1839
0
                EmitOp          (o, op);
1840
0
                EmitSlotOffset  (o, preservedSlotNumber);
1841
0
                EmitSlotOffset  (o, slot);
1842
0
            }
1843
19.1k
        }
1844
508
    }
1845
1846
521
    _catch:
1847
521
    return result;
1848
521
}
1849
1850
static
1851
M3Result  Compile_LoopOrBlock  (IM3Compilation o, m3opcode_t i_opcode)
1852
477
{
1853
477
    M3Result result;
1854
1855
    // TODO: these shouldn't be necessary for non-loop blocks?
1856
477
_   (PreserveRegisters (o));
1857
477
_   (PreserveArgsAndLocals (o));
1858
1859
477
    IM3FuncType blockType;
1860
477
_   (ReadBlockType (o, & blockType));
1861
1862
468
    if (i_opcode == c_waOp_loop)
1863
446
    {
1864
446
        u16 numParams = GetFuncTypeNumParams (blockType);
1865
446
        if (numParams)
1866
428
        {
1867
            // instantiate constants
1868
428
            u16 numValues = GetNumBlockValuesOnStack (o);                   // CompileBlock enforces this at comptime
1869
428
                                                                            d_m3Assert (numValues >= numParams);
1870
428
            if (numValues >= numParams)
1871
361
            {
1872
361
                u16 stackTop = GetStackTopIndex (o) + 1;
1873
1874
11.9k
                for (u16 i = stackTop - numParams; i < stackTop; ++i)
1875
11.5k
                {
1876
11.5k
                    u16 slot = GetSlotForStackIndex (o, i);
1877
11.5k
                    u8 type = GetStackTypeFromBottom (o, i);
1878
1879
11.5k
                    if (IsConstantSlot (o, slot))
1880
0
                    {
1881
0
                        u16 newSlot = c_slotUnused;
1882
0
_                       (AllocateSlots (o, & newSlot, type));
1883
0
_                       (CopyStackIndexToSlot (o, newSlot, i));
1884
0
                        o->wasmStack [i] = newSlot;
1885
0
                    }
1886
11.5k
                }
1887
361
            }
1888
428
        }
1889
1890
446
_       (EmitOp (o, op_Loop));
1891
446
    }
1892
22
    else
1893
22
    {
1894
22
    }
1895
1896
468
_   (CompileBlock (o, blockType, i_opcode));
1897
1898
477
    _catch: return result;
1899
17
}
1900
1901
static
1902
M3Result  CompileElseBlock  (IM3Compilation o, pc_t * o_startPC, IM3FuncType i_blockType)
1903
3
{
1904
3
    IM3CodePage savedPage = o->page;
1905
3
_try {
1906
1907
3
    IM3CodePage elsePage;
1908
3
_   (AcquireCompilationCodePage (o, & elsePage));
1909
1910
3
    * o_startPC = GetPagePC (elsePage);
1911
1912
3
    o->page = elsePage;
1913
1914
3
_   (CompileBlock (o, i_blockType, c_waOp_else));
1915
1916
2
_   (EmitOp (o, op_Branch));
1917
2
    EmitPointer (o, GetPagePC (savedPage));
1918
3
} _catch:
1919
3
    if(o->page != savedPage) {
1920
3
        ReleaseCompilationCodePage (o);
1921
3
    }
1922
3
    o->page = savedPage;
1923
3
    return result;
1924
2
}
1925
1926
static
1927
M3Result  Compile_If  (IM3Compilation o, m3opcode_t i_opcode)
1928
44
{
1929
    /*      [   op_If   ]
1930
            [ <else-pc> ]   ---->   [ ..else..  ]
1931
            [  ..if..   ]           [ ..block.. ]
1932
            [ ..block.. ]           [ op_Branch ]
1933
            [    end    ]  <-----   [  <end-pc> ]       */
1934
1935
44
_try {
1936
1937
44
_   (PreserveNonTopRegisters (o));
1938
44
_   (PreserveArgsAndLocals (o));
1939
1940
44
    IM3Operation op = IsStackTopInRegister (o) ? op_If_r : op_If_s;
1941
1942
44
_   (EmitOp (o, op));
1943
44
_   (EmitSlotNumOfStackTopAndPop (o));
1944
1945
44
    pc_t * pc = (pc_t *) ReservePointer (o);
1946
1947
44
    IM3FuncType blockType;
1948
44
_   (ReadBlockType (o, & blockType));
1949
1950
//  dump_type_stack (o);
1951
1952
44
    u16 stackIndex = o->stackIndex;
1953
1954
44
_   (CompileBlock (o, blockType, i_opcode));
1955
1956
3
    if (o->previousOpcode == c_waOp_else)
1957
1
    {
1958
1
        o->stackIndex = stackIndex;
1959
1
_       (CompileElseBlock (o, pc, blockType));
1960
0
    }
1961
2
    else
1962
2
    {
1963
        // if block produces values and there isn't a defined else
1964
        // case, then we need to make one up so that the pass-through
1965
        // results end up in the right place
1966
2
        if (GetFuncTypeNumResults (blockType))
1967
2
        {
1968
            // rewind to the if's end to create a fake else block
1969
2
            o->wasm--;
1970
2
            o->stackIndex = stackIndex;
1971
1972
//          dump_type_stack (o);
1973
1974
2
_           (CompileElseBlock (o, pc, blockType));
1975
2
        }
1976
0
        else * pc = GetPC (o);
1977
2
    }
1978
1979
44
    } _catch: return result;
1980
3
}
1981
1982
static
1983
M3Result  Compile_Select  (IM3Compilation o, m3opcode_t i_opcode)
1984
21
{
1985
21
    M3Result result = m3Err_none;
1986
1987
21
    u16 slots [3] = { c_slotUnused, c_slotUnused, c_slotUnused };
1988
1989
21
    u8 type = GetStackTypeFromTop (o, 1); // get type of selection
1990
1991
21
    IM3Operation op = NULL;
1992
1993
21
    if (IsFpType (type))
1994
11
    {
1995
11
#   if d_m3HasFloat
1996
        // not consuming a fp reg, so preserve
1997
11
        if (not IsStackTopMinus1InRegister (o) and
1998
11
            not IsStackTopMinus2InRegister (o))
1999
10
        {
2000
10
_           (PreserveRegisterIfOccupied (o, type));
2001
10
        }
2002
2003
11
        bool selectorInReg = IsStackTopInRegister (o);
2004
11
        slots [0] = GetStackTopSlotNumber (o);
2005
11
_       (Pop (o));
2006
2007
11
        u32 opIndex = 0;
2008
2009
33
        for (u32 i = 1; i <= 2; ++i)
2010
22
        {
2011
22
            if (IsStackTopInRegister (o))
2012
1
                opIndex = i;
2013
21
            else
2014
21
                slots [i] = GetStackTopSlotNumber (o);
2015
2016
22
_          (Pop (o));
2017
22
        }
2018
2019
11
        op = c_fpSelectOps [type - c_m3Type_f32] [selectorInReg] [opIndex];
2020
#   else
2021
        _throw (m3Err_unknownOpcode);
2022
#   endif
2023
11
    }
2024
10
    else if (IsIntType (type))
2025
7
    {
2026
        // 'sss' operation doesn't consume a register, so might have to protected its contents
2027
7
        if (not IsStackTopInRegister (o) and
2028
4
            not IsStackTopMinus1InRegister (o) and
2029
4
            not IsStackTopMinus2InRegister (o))
2030
4
        {
2031
4
_           (PreserveRegisterIfOccupied (o, type));
2032
4
        }
2033
2034
7
        u32 opIndex = 3;  // op_Select_*_sss
2035
2036
28
        for (u32 i = 0; i < 3; ++i)
2037
21
        {
2038
21
            if (IsStackTopInRegister (o))
2039
5
                opIndex = i;
2040
16
            else
2041
16
                slots [i] = GetStackTopSlotNumber (o);
2042
2043
21
_          (Pop (o));
2044
21
        }
2045
2046
7
        op = c_intSelectOps [type - c_m3Type_i32] [opIndex];
2047
7
    }
2048
3
    else if (not IsStackPolymorphic (o))
2049
21
        _throw (m3Err_functionStackUnderrun);
2050
2051
21
    EmitOp (o, op);
2052
84
    for (u32 i = 0; i < 3; i++)
2053
63
    {
2054
63
        if (IsValidSlot (slots [i]))
2055
48
            EmitSlotOffset (o, slots [i]);
2056
63
    }
2057
21
_   (PushRegister (o, type));
2058
2059
21
    _catch: return result;
2060
20
}
2061
2062
static
2063
M3Result  Compile_Drop  (IM3Compilation o, m3opcode_t i_opcode)
2064
14
{
2065
14
    M3Result result = Pop (o);                                              if (d_m3LogWasmStack) dump_type_stack (o);
2066
14
    return result;
2067
14
}
2068
2069
static
2070
M3Result  Compile_Nop  (IM3Compilation o, m3opcode_t i_opcode)
2071
10
{
2072
10
    return m3Err_none;
2073
10
}
2074
2075
static
2076
M3Result  Compile_Unreachable  (IM3Compilation o, m3opcode_t i_opcode)
2077
161
{
2078
161
    M3Result result;
2079
2080
161
_   (AddTrapRecord (o));
2081
2082
161
_   (EmitOp (o, op_Unreachable));
2083
161
_   (SetStackPolymorphic (o));
2084
2085
161
    _catch:
2086
161
    return result;
2087
161
}
2088
2089
2090
// OPTZ: currently all stack slot indices take up a full word, but
2091
// dual stack source operands could be packed together
2092
static
2093
M3Result  Compile_Operator  (IM3Compilation o, m3opcode_t i_opcode)
2094
320
{
2095
320
    M3Result result;
2096
2097
320
    IM3OpInfo opInfo = GetOpInfo (i_opcode);
2098
320
    _throwif (m3Err_unknownOpcode, not opInfo);
2099
2100
320
    IM3Operation op;
2101
2102
    // This preserve is for for FP compare operations.
2103
    // either need additional slot destination operations or the
2104
    // easy fix, move _r0 out of the way.
2105
    // moving out the way might be the optimal solution most often?
2106
    // otherwise, the _r0 reg can get buried down in the stack
2107
    // and be idle & wasted for a moment.
2108
320
    if (IsFpType (GetStackTopType (o)) and IsIntType (opInfo->type))
2109
109
    {
2110
109
_       (PreserveRegisterIfOccupied (o, opInfo->type));
2111
109
    }
2112
2113
320
    if (opInfo->stackOffset == 0)
2114
86
    {
2115
86
        if (IsStackTopInRegister (o))
2116
33
        {
2117
33
            op = opInfo->operations [0]; // _s
2118
33
        }
2119
53
        else
2120
53
        {
2121
53
_           (PreserveRegisterIfOccupied (o, opInfo->type));
2122
53
            op = opInfo->operations [1]; // _r
2123
53
        }
2124
86
    }
2125
234
    else
2126
234
    {
2127
234
        if (IsStackTopInRegister (o))
2128
132
        {
2129
132
            op = opInfo->operations [0];  // _rs
2130
2131
132
            if (IsStackTopMinus1InRegister (o))
2132
2
            {                                       d_m3Assert (i_opcode == c_waOp_store_f32 or i_opcode == c_waOp_store_f64);
2133
2
                op = opInfo->operations [3]; // _rr for fp.store
2134
2
            }
2135
132
        }
2136
102
        else if (IsStackTopMinus1InRegister (o))
2137
0
        {
2138
0
            op = opInfo->operations [1]; // _sr
2139
2140
0
            if (not op)  // must be commutative, then
2141
0
                op = opInfo->operations [0];
2142
0
        }
2143
102
        else
2144
102
        {
2145
102
_           (PreserveRegisterIfOccupied (o, opInfo->type));     // _ss
2146
102
            op = opInfo->operations [2];
2147
102
        }
2148
234
    }
2149
2150
320
    if (op)
2151
318
    {
2152
318
_       (EmitOp (o, op));
2153
2154
318
_       (EmitSlotNumOfStackTopAndPop (o));
2155
2156
318
        if (opInfo->stackOffset < 0)
2157
318
_           (EmitSlotNumOfStackTopAndPop (o));
2158
2159
318
        if (opInfo->type != c_m3Type_none)
2160
315
_           (PushRegister (o, opInfo->type));
2161
315
    }
2162
2
    else
2163
2
    {
2164
#       ifdef DEBUG
2165
            result = ErrorCompile ("no operation found for opcode", o, "'%s'", opInfo->name);
2166
#       else
2167
2
            result = ErrorCompile ("no operation found for opcode", o, "%x", i_opcode);
2168
2
#       endif
2169
2
        _throw (result);
2170
0
    }
2171
2172
320
    _catch: return result;
2173
320
}
2174
2175
static
2176
M3Result  Compile_Convert  (IM3Compilation o, m3opcode_t i_opcode)
2177
9
{
2178
9
_try {
2179
9
    IM3OpInfo opInfo = GetOpInfo (i_opcode);
2180
9
    _throwif (m3Err_unknownOpcode, not opInfo);
2181
2182
9
    bool destInSlot = IsRegisterTypeAllocated (o, opInfo->type);
2183
9
    bool sourceInSlot = IsStackTopInSlot (o);
2184
2185
9
    IM3Operation op = opInfo->operations [destInSlot * 2 + sourceInSlot];
2186
2187
9
_   (EmitOp (o, op));
2188
9
_   (EmitSlotNumOfStackTopAndPop (o));
2189
2190
9
    if (destInSlot)
2191
2
_       (PushAllocatedSlotAndEmit (o, opInfo->type))
2192
7
    else
2193
7
_       (PushRegister (o, opInfo->type))
2194
2195
9
}
2196
9
    _catch: return result;
2197
9
}
2198
2199
static
2200
M3Result  Compile_Load_Store  (IM3Compilation o, m3opcode_t i_opcode)
2201
45
{
2202
45
_try {
2203
45
    u32 alignHint, memoryOffset;
2204
2205
45
_   (ReadLEB_u32 (& alignHint, & o->wasm, o->wasmEnd));
2206
45
_   (ReadLEB_u32 (& memoryOffset, & o->wasm, o->wasmEnd));
2207
45
                                                                        m3log (compile, d_indent " (offset = %d)", get_indention_string (o), memoryOffset);
2208
45
    IM3OpInfo opInfo = GetOpInfo (i_opcode);
2209
45
    _throwif (m3Err_unknownOpcode, not opInfo);
2210
2211
45
    if (IsFpType (opInfo->type))
2212
45
_       (PreserveRegisterIfOccupied (o, c_m3Type_f64));
2213
2214
45
_   (Compile_Operator (o, i_opcode));
2215
2216
45
    EmitConstant32 (o, memoryOffset);
2217
45
}
2218
45
    _catch: return result;
2219
45
}
2220
2221
2222
M3Result  CompileRawFunction  (IM3Module io_module,  IM3Function io_function, const void * i_function, const void * i_userdata)
2223
0
{
2224
0
    d_m3Assert (io_module->runtime);
2225
2226
0
    IM3CodePage page = AcquireCodePageWithCapacity (io_module->runtime, 4);
2227
2228
0
    if (page)
2229
0
    {
2230
0
        io_function->compiled = GetPagePC (page);
2231
0
        io_function->module = io_module;
2232
2233
0
        EmitWord (page, op_CallRawFunction);
2234
0
        EmitWord (page, i_function);
2235
0
        EmitWord (page, io_function);
2236
0
        EmitWord (page, i_userdata);
2237
2238
0
        ReleaseCodePage (io_module->runtime, page);
2239
0
        return m3Err_none;
2240
0
    }
2241
0
    else {
2242
0
        return m3Err_mallocFailedCodePage;
2243
0
    }
2244
0
}
2245
2246
2247
2248
// d_logOp, d_logOp2 macros aren't actually used by the compiler, just codepage decoding (d_m3LogCodePages = 1)
2249
#define d_logOp(OP)                         { op_##OP,                  NULL,                       NULL,                       NULL }
2250
#define d_logOp2(OP1,OP2)                   { op_##OP1,                 op_##OP2,                   NULL,                       NULL }
2251
2252
#define d_emptyOpList                       { NULL,                     NULL,                       NULL,                       NULL }
2253
#define d_unaryOpList(TYPE, NAME)           { op_##TYPE##_##NAME##_r,   op_##TYPE##_##NAME##_s,     NULL,                       NULL }
2254
#define d_binOpList(TYPE, NAME)             { op_##TYPE##_##NAME##_rs,  op_##TYPE##_##NAME##_sr,    op_##TYPE##_##NAME##_ss,    NULL }
2255
#define d_storeFpOpList(TYPE, NAME)         { op_##TYPE##_##NAME##_rs,  op_##TYPE##_##NAME##_sr,    op_##TYPE##_##NAME##_ss,    op_##TYPE##_##NAME##_rr }
2256
#define d_commutativeBinOpList(TYPE, NAME)  { op_##TYPE##_##NAME##_rs,  NULL,                       op_##TYPE##_##NAME##_ss,    NULL }
2257
#define d_convertOpList(OP)                 { op_##OP##_r_r,            op_##OP##_r_s,              op_##OP##_s_r,              op_##OP##_s_s }
2258
2259
2260
const M3OpInfo c_operations [] =
2261
{
2262
    M3OP( "unreachable",         0, none,   d_logOp (Unreachable),              Compile_Unreachable ),  // 0x00
2263
    M3OP( "nop",                 0, none,   d_emptyOpList,                      Compile_Nop ),          // 0x01 .
2264
    M3OP( "block",               0, none,   d_emptyOpList,                      Compile_LoopOrBlock ),  // 0x02
2265
    M3OP( "loop",                0, none,   d_logOp (Loop),                     Compile_LoopOrBlock ),  // 0x03
2266
    M3OP( "if",                 -1, none,   d_emptyOpList,                      Compile_If ),           // 0x04
2267
    M3OP( "else",                0, none,   d_emptyOpList,                      Compile_Nop ),          // 0x05
2268
2269
    M3OP_RESERVED, M3OP_RESERVED, M3OP_RESERVED, M3OP_RESERVED, M3OP_RESERVED,                          // 0x06...0x0a
2270
2271
    M3OP( "end",                 0, none,   d_emptyOpList,                      Compile_End ),          // 0x0b
2272
    M3OP( "br",                  0, none,   d_logOp (Branch),                   Compile_Branch ),       // 0x0c
2273
    M3OP( "br_if",              -1, none,   d_logOp2 (BranchIf_r, BranchIf_s),  Compile_Branch ),       // 0x0d
2274
    M3OP( "br_table",           -1, none,   d_logOp (BranchTable),              Compile_BranchTable ),  // 0x0e
2275
    M3OP( "return",              0, any,    d_logOp (Return),                   Compile_Return ),       // 0x0f
2276
    M3OP( "call",                0, any,    d_logOp (Call),                     Compile_Call ),         // 0x10
2277
    M3OP( "call_indirect",       0, any,    d_logOp (CallIndirect),             Compile_CallIndirect ), // 0x11
2278
    M3OP( "return_call",         0, any,    d_emptyOpList,                      Compile_Call ),         // 0x12 TODO: Optimize
2279
    M3OP( "return_call_indirect",0, any,    d_emptyOpList,                      Compile_CallIndirect ), // 0x13
2280
2281
    M3OP_RESERVED,  M3OP_RESERVED,                                                                      // 0x14...
2282
    M3OP_RESERVED,  M3OP_RESERVED, M3OP_RESERVED, M3OP_RESERVED,                                        // ...0x19
2283
2284
    M3OP( "drop",               -1, none,   d_emptyOpList,                      Compile_Drop ),         // 0x1a
2285
    M3OP( "select",             -2, any,    d_emptyOpList,                      Compile_Select  ),      // 0x1b
2286
2287
    M3OP_RESERVED,  M3OP_RESERVED, M3OP_RESERVED, M3OP_RESERVED,                                        // 0x1c...0x1f
2288
2289
    M3OP( "local.get",          1,  any,    d_emptyOpList,                      Compile_GetLocal ),     // 0x20
2290
    M3OP( "local.set",          1,  none,   d_emptyOpList,                      Compile_SetLocal ),     // 0x21
2291
    M3OP( "local.tee",          0,  any,    d_emptyOpList,                      Compile_SetLocal ),     // 0x22
2292
    M3OP( "global.get",         1,  none,   d_emptyOpList,                      Compile_GetSetGlobal ), // 0x23
2293
    M3OP( "global.set",         1,  none,   d_emptyOpList,                      Compile_GetSetGlobal ), // 0x24
2294
2295
    M3OP_RESERVED,  M3OP_RESERVED, M3OP_RESERVED,                                                       // 0x25...0x27
2296
2297
    M3OP( "i32.load",           0,  i_32,   d_unaryOpList (i32, Load_i32),      Compile_Load_Store ),   // 0x28
2298
    M3OP( "i64.load",           0,  i_64,   d_unaryOpList (i64, Load_i64),      Compile_Load_Store ),   // 0x29
2299
    M3OP_F( "f32.load",         0,  f_32,   d_unaryOpList (f32, Load_f32),      Compile_Load_Store ),   // 0x2a
2300
    M3OP_F( "f64.load",         0,  f_64,   d_unaryOpList (f64, Load_f64),      Compile_Load_Store ),   // 0x2b
2301
2302
    M3OP( "i32.load8_s",        0,  i_32,   d_unaryOpList (i32, Load_i8),       Compile_Load_Store ),   // 0x2c
2303
    M3OP( "i32.load8_u",        0,  i_32,   d_unaryOpList (i32, Load_u8),       Compile_Load_Store ),   // 0x2d
2304
    M3OP( "i32.load16_s",       0,  i_32,   d_unaryOpList (i32, Load_i16),      Compile_Load_Store ),   // 0x2e
2305
    M3OP( "i32.load16_u",       0,  i_32,   d_unaryOpList (i32, Load_u16),      Compile_Load_Store ),   // 0x2f
2306
2307
    M3OP( "i64.load8_s",        0,  i_64,   d_unaryOpList (i64, Load_i8),       Compile_Load_Store ),   // 0x30
2308
    M3OP( "i64.load8_u",        0,  i_64,   d_unaryOpList (i64, Load_u8),       Compile_Load_Store ),   // 0x31
2309
    M3OP( "i64.load16_s",       0,  i_64,   d_unaryOpList (i64, Load_i16),      Compile_Load_Store ),   // 0x32
2310
    M3OP( "i64.load16_u",       0,  i_64,   d_unaryOpList (i64, Load_u16),      Compile_Load_Store ),   // 0x33
2311
    M3OP( "i64.load32_s",       0,  i_64,   d_unaryOpList (i64, Load_i32),      Compile_Load_Store ),   // 0x34
2312
    M3OP( "i64.load32_u",       0,  i_64,   d_unaryOpList (i64, Load_u32),      Compile_Load_Store ),   // 0x35
2313
2314
    M3OP( "i32.store",          -2, none,   d_binOpList (i32, Store_i32),       Compile_Load_Store ),   // 0x36
2315
    M3OP( "i64.store",          -2, none,   d_binOpList (i64, Store_i64),       Compile_Load_Store ),   // 0x37
2316
    M3OP_F( "f32.store",        -2, none,   d_storeFpOpList (f32, Store_f32),   Compile_Load_Store ),   // 0x38
2317
    M3OP_F( "f64.store",        -2, none,   d_storeFpOpList (f64, Store_f64),   Compile_Load_Store ),   // 0x39
2318
2319
    M3OP( "i32.store8",         -2, none,   d_binOpList (i32, Store_u8),        Compile_Load_Store ),   // 0x3a
2320
    M3OP( "i32.store16",        -2, none,   d_binOpList (i32, Store_i16),       Compile_Load_Store ),   // 0x3b
2321
2322
    M3OP( "i64.store8",         -2, none,   d_binOpList (i64, Store_u8),        Compile_Load_Store ),   // 0x3c
2323
    M3OP( "i64.store16",        -2, none,   d_binOpList (i64, Store_i16),       Compile_Load_Store ),   // 0x3d
2324
    M3OP( "i64.store32",        -2, none,   d_binOpList (i64, Store_i32),       Compile_Load_Store ),   // 0x3e
2325
2326
    M3OP( "memory.size",        1,  i_32,   d_logOp (MemSize),                  Compile_Memory_Size ),  // 0x3f
2327
    M3OP( "memory.grow",        1,  i_32,   d_logOp (MemGrow),                  Compile_Memory_Grow ),  // 0x40
2328
2329
    M3OP( "i32.const",          1,  i_32,   d_logOp (Const32),                  Compile_Const_i32 ),    // 0x41
2330
    M3OP( "i64.const",          1,  i_64,   d_logOp (Const64),                  Compile_Const_i64 ),    // 0x42
2331
    M3OP_F( "f32.const",        1,  f_32,   d_emptyOpList,                      Compile_Const_f32 ),    // 0x43
2332
    M3OP_F( "f64.const",        1,  f_64,   d_emptyOpList,                      Compile_Const_f64 ),    // 0x44
2333
2334
    M3OP( "i32.eqz",            0,  i_32,   d_unaryOpList (i32, EqualToZero)        , NULL  ),          // 0x45
2335
    M3OP( "i32.eq",             -1, i_32,   d_commutativeBinOpList (i32, Equal)     , NULL  ),          // 0x46
2336
    M3OP( "i32.ne",             -1, i_32,   d_commutativeBinOpList (i32, NotEqual)  , NULL  ),          // 0x47
2337
    M3OP( "i32.lt_s",           -1, i_32,   d_binOpList (i32, LessThan)             , NULL  ),          // 0x48
2338
    M3OP( "i32.lt_u",           -1, i_32,   d_binOpList (u32, LessThan)             , NULL  ),          // 0x49
2339
    M3OP( "i32.gt_s",           -1, i_32,   d_binOpList (i32, GreaterThan)          , NULL  ),          // 0x4a
2340
    M3OP( "i32.gt_u",           -1, i_32,   d_binOpList (u32, GreaterThan)          , NULL  ),          // 0x4b
2341
    M3OP( "i32.le_s",           -1, i_32,   d_binOpList (i32, LessThanOrEqual)      , NULL  ),          // 0x4c
2342
    M3OP( "i32.le_u",           -1, i_32,   d_binOpList (u32, LessThanOrEqual)      , NULL  ),          // 0x4d
2343
    M3OP( "i32.ge_s",           -1, i_32,   d_binOpList (i32, GreaterThanOrEqual)   , NULL  ),          // 0x4e
2344
    M3OP( "i32.ge_u",           -1, i_32,   d_binOpList (u32, GreaterThanOrEqual)   , NULL  ),          // 0x4f
2345
2346
    M3OP( "i64.eqz",            0,  i_32,   d_unaryOpList (i64, EqualToZero)        , NULL  ),          // 0x50
2347
    M3OP( "i64.eq",             -1, i_32,   d_commutativeBinOpList (i64, Equal)     , NULL  ),          // 0x51
2348
    M3OP( "i64.ne",             -1, i_32,   d_commutativeBinOpList (i64, NotEqual)  , NULL  ),          // 0x52
2349
    M3OP( "i64.lt_s",           -1, i_32,   d_binOpList (i64, LessThan)             , NULL  ),          // 0x53
2350
    M3OP( "i64.lt_u",           -1, i_32,   d_binOpList (u64, LessThan)             , NULL  ),          // 0x54
2351
    M3OP( "i64.gt_s",           -1, i_32,   d_binOpList (i64, GreaterThan)          , NULL  ),          // 0x55
2352
    M3OP( "i64.gt_u",           -1, i_32,   d_binOpList (u64, GreaterThan)          , NULL  ),          // 0x56
2353
    M3OP( "i64.le_s",           -1, i_32,   d_binOpList (i64, LessThanOrEqual)      , NULL  ),          // 0x57
2354
    M3OP( "i64.le_u",           -1, i_32,   d_binOpList (u64, LessThanOrEqual)      , NULL  ),          // 0x58
2355
    M3OP( "i64.ge_s",           -1, i_32,   d_binOpList (i64, GreaterThanOrEqual)   , NULL  ),          // 0x59
2356
    M3OP( "i64.ge_u",           -1, i_32,   d_binOpList (u64, GreaterThanOrEqual)   , NULL  ),          // 0x5a
2357
2358
    M3OP_F( "f32.eq",           -1, i_32,   d_commutativeBinOpList (f32, Equal)     , NULL  ),          // 0x5b
2359
    M3OP_F( "f32.ne",           -1, i_32,   d_commutativeBinOpList (f32, NotEqual)  , NULL  ),          // 0x5c
2360
    M3OP_F( "f32.lt",           -1, i_32,   d_binOpList (f32, LessThan)             , NULL  ),          // 0x5d
2361
    M3OP_F( "f32.gt",           -1, i_32,   d_binOpList (f32, GreaterThan)          , NULL  ),          // 0x5e
2362
    M3OP_F( "f32.le",           -1, i_32,   d_binOpList (f32, LessThanOrEqual)      , NULL  ),          // 0x5f
2363
    M3OP_F( "f32.ge",           -1, i_32,   d_binOpList (f32, GreaterThanOrEqual)   , NULL  ),          // 0x60
2364
2365
    M3OP_F( "f64.eq",           -1, i_32,   d_commutativeBinOpList (f64, Equal)     , NULL  ),          // 0x61
2366
    M3OP_F( "f64.ne",           -1, i_32,   d_commutativeBinOpList (f64, NotEqual)  , NULL  ),          // 0x62
2367
    M3OP_F( "f64.lt",           -1, i_32,   d_binOpList (f64, LessThan)             , NULL  ),          // 0x63
2368
    M3OP_F( "f64.gt",           -1, i_32,   d_binOpList (f64, GreaterThan)          , NULL  ),          // 0x64
2369
    M3OP_F( "f64.le",           -1, i_32,   d_binOpList (f64, LessThanOrEqual)      , NULL  ),          // 0x65
2370
    M3OP_F( "f64.ge",           -1, i_32,   d_binOpList (f64, GreaterThanOrEqual)   , NULL  ),          // 0x66
2371
2372
    M3OP( "i32.clz",            0,  i_32,   d_unaryOpList (u32, Clz)                , NULL  ),          // 0x67
2373
    M3OP( "i32.ctz",            0,  i_32,   d_unaryOpList (u32, Ctz)                , NULL  ),          // 0x68
2374
    M3OP( "i32.popcnt",         0,  i_32,   d_unaryOpList (u32, Popcnt)             , NULL  ),          // 0x69
2375
2376
    M3OP( "i32.add",            -1, i_32,   d_commutativeBinOpList (i32, Add)       , NULL  ),          // 0x6a
2377
    M3OP( "i32.sub",            -1, i_32,   d_binOpList (i32, Subtract)             , NULL  ),          // 0x6b
2378
    M3OP( "i32.mul",            -1, i_32,   d_commutativeBinOpList (i32, Multiply)  , NULL  ),          // 0x6c
2379
    M3OP( "i32.div_s",          -1, i_32,   d_binOpList (i32, Divide)               , NULL  ),          // 0x6d
2380
    M3OP( "i32.div_u",          -1, i_32,   d_binOpList (u32, Divide)               , NULL  ),          // 0x6e
2381
    M3OP( "i32.rem_s",          -1, i_32,   d_binOpList (i32, Remainder)            , NULL  ),          // 0x6f
2382
    M3OP( "i32.rem_u",          -1, i_32,   d_binOpList (u32, Remainder)            , NULL  ),          // 0x70
2383
    M3OP( "i32.and",            -1, i_32,   d_commutativeBinOpList (u32, And)       , NULL  ),          // 0x71
2384
    M3OP( "i32.or",             -1, i_32,   d_commutativeBinOpList (u32, Or)        , NULL  ),          // 0x72
2385
    M3OP( "i32.xor",            -1, i_32,   d_commutativeBinOpList (u32, Xor)       , NULL  ),          // 0x73
2386
    M3OP( "i32.shl",            -1, i_32,   d_binOpList (u32, ShiftLeft)            , NULL  ),          // 0x74
2387
    M3OP( "i32.shr_s",          -1, i_32,   d_binOpList (i32, ShiftRight)           , NULL  ),          // 0x75
2388
    M3OP( "i32.shr_u",          -1, i_32,   d_binOpList (u32, ShiftRight)           , NULL  ),          // 0x76
2389
    M3OP( "i32.rotl",           -1, i_32,   d_binOpList (u32, Rotl)                 , NULL  ),          // 0x77
2390
    M3OP( "i32.rotr",           -1, i_32,   d_binOpList (u32, Rotr)                 , NULL  ),          // 0x78
2391
2392
    M3OP( "i64.clz",            0,  i_64,   d_unaryOpList (u64, Clz)                , NULL  ),          // 0x79
2393
    M3OP( "i64.ctz",            0,  i_64,   d_unaryOpList (u64, Ctz)                , NULL  ),          // 0x7a
2394
    M3OP( "i64.popcnt",         0,  i_64,   d_unaryOpList (u64, Popcnt)             , NULL  ),          // 0x7b
2395
2396
    M3OP( "i64.add",            -1, i_64,   d_commutativeBinOpList (i64, Add)       , NULL  ),          // 0x7c
2397
    M3OP( "i64.sub",            -1, i_64,   d_binOpList (i64, Subtract)             , NULL  ),          // 0x7d
2398
    M3OP( "i64.mul",            -1, i_64,   d_commutativeBinOpList (i64, Multiply)  , NULL  ),          // 0x7e
2399
    M3OP( "i64.div_s",          -1, i_64,   d_binOpList (i64, Divide)               , NULL  ),          // 0x7f
2400
    M3OP( "i64.div_u",          -1, i_64,   d_binOpList (u64, Divide)               , NULL  ),          // 0x80
2401
    M3OP( "i64.rem_s",          -1, i_64,   d_binOpList (i64, Remainder)            , NULL  ),          // 0x81
2402
    M3OP( "i64.rem_u",          -1, i_64,   d_binOpList (u64, Remainder)            , NULL  ),          // 0x82
2403
    M3OP( "i64.and",            -1, i_64,   d_commutativeBinOpList (u64, And)       , NULL  ),          // 0x83
2404
    M3OP( "i64.or",             -1, i_64,   d_commutativeBinOpList (u64, Or)        , NULL  ),          // 0x84
2405
    M3OP( "i64.xor",            -1, i_64,   d_commutativeBinOpList (u64, Xor)       , NULL  ),          // 0x85
2406
    M3OP( "i64.shl",            -1, i_64,   d_binOpList (u64, ShiftLeft)            , NULL  ),          // 0x86
2407
    M3OP( "i64.shr_s",          -1, i_64,   d_binOpList (i64, ShiftRight)           , NULL  ),          // 0x87
2408
    M3OP( "i64.shr_u",          -1, i_64,   d_binOpList (u64, ShiftRight)           , NULL  ),          // 0x88
2409
    M3OP( "i64.rotl",           -1, i_64,   d_binOpList (u64, Rotl)                 , NULL  ),          // 0x89
2410
    M3OP( "i64.rotr",           -1, i_64,   d_binOpList (u64, Rotr)                 , NULL  ),          // 0x8a
2411
2412
    M3OP_F( "f32.abs",          0,  f_32,   d_unaryOpList(f32, Abs)                 , NULL  ),          // 0x8b
2413
    M3OP_F( "f32.neg",          0,  f_32,   d_unaryOpList(f32, Negate)              , NULL  ),          // 0x8c
2414
    M3OP_F( "f32.ceil",         0,  f_32,   d_unaryOpList(f32, Ceil)                , NULL  ),          // 0x8d
2415
    M3OP_F( "f32.floor",        0,  f_32,   d_unaryOpList(f32, Floor)               , NULL  ),          // 0x8e
2416
    M3OP_F( "f32.trunc",        0,  f_32,   d_unaryOpList(f32, Trunc)               , NULL  ),          // 0x8f
2417
    M3OP_F( "f32.nearest",      0,  f_32,   d_unaryOpList(f32, Nearest)             , NULL  ),          // 0x90
2418
    M3OP_F( "f32.sqrt",         0,  f_32,   d_unaryOpList(f32, Sqrt)                , NULL  ),          // 0x91
2419
2420
    M3OP_F( "f32.add",          -1, f_32,   d_commutativeBinOpList (f32, Add)       , NULL  ),          // 0x92
2421
    M3OP_F( "f32.sub",          -1, f_32,   d_binOpList (f32, Subtract)             , NULL  ),          // 0x93
2422
    M3OP_F( "f32.mul",          -1, f_32,   d_commutativeBinOpList (f32, Multiply)  , NULL  ),          // 0x94
2423
    M3OP_F( "f32.div",          -1, f_32,   d_binOpList (f32, Divide)               , NULL  ),          // 0x95
2424
    M3OP_F( "f32.min",          -1, f_32,   d_commutativeBinOpList (f32, Min)       , NULL  ),          // 0x96
2425
    M3OP_F( "f32.max",          -1, f_32,   d_commutativeBinOpList (f32, Max)       , NULL  ),          // 0x97
2426
    M3OP_F( "f32.copysign",     -1, f_32,   d_binOpList (f32, CopySign)             , NULL  ),          // 0x98
2427
2428
    M3OP_F( "f64.abs",          0,  f_64,   d_unaryOpList(f64, Abs)                 , NULL  ),          // 0x99
2429
    M3OP_F( "f64.neg",          0,  f_64,   d_unaryOpList(f64, Negate)              , NULL  ),          // 0x9a
2430
    M3OP_F( "f64.ceil",         0,  f_64,   d_unaryOpList(f64, Ceil)                , NULL  ),          // 0x9b
2431
    M3OP_F( "f64.floor",        0,  f_64,   d_unaryOpList(f64, Floor)               , NULL  ),          // 0x9c
2432
    M3OP_F( "f64.trunc",        0,  f_64,   d_unaryOpList(f64, Trunc)               , NULL  ),          // 0x9d
2433
    M3OP_F( "f64.nearest",      0,  f_64,   d_unaryOpList(f64, Nearest)             , NULL  ),          // 0x9e
2434
    M3OP_F( "f64.sqrt",         0,  f_64,   d_unaryOpList(f64, Sqrt)                , NULL  ),          // 0x9f
2435
2436
    M3OP_F( "f64.add",          -1, f_64,   d_commutativeBinOpList (f64, Add)       , NULL  ),          // 0xa0
2437
    M3OP_F( "f64.sub",          -1, f_64,   d_binOpList (f64, Subtract)             , NULL  ),          // 0xa1
2438
    M3OP_F( "f64.mul",          -1, f_64,   d_commutativeBinOpList (f64, Multiply)  , NULL  ),          // 0xa2
2439
    M3OP_F( "f64.div",          -1, f_64,   d_binOpList (f64, Divide)               , NULL  ),          // 0xa3
2440
    M3OP_F( "f64.min",          -1, f_64,   d_commutativeBinOpList (f64, Min)       , NULL  ),          // 0xa4
2441
    M3OP_F( "f64.max",          -1, f_64,   d_commutativeBinOpList (f64, Max)       , NULL  ),          // 0xa5
2442
    M3OP_F( "f64.copysign",     -1, f_64,   d_binOpList (f64, CopySign)             , NULL  ),          // 0xa6
2443
2444
    M3OP( "i32.wrap/i64",       0,  i_32,   d_unaryOpList (i32, Wrap_i64),          NULL    ),          // 0xa7
2445
    M3OP_F( "i32.trunc_s/f32",  0,  i_32,   d_convertOpList (i32_Trunc_f32),        Compile_Convert ),  // 0xa8
2446
    M3OP_F( "i32.trunc_u/f32",  0,  i_32,   d_convertOpList (u32_Trunc_f32),        Compile_Convert ),  // 0xa9
2447
    M3OP_F( "i32.trunc_s/f64",  0,  i_32,   d_convertOpList (i32_Trunc_f64),        Compile_Convert ),  // 0xaa
2448
    M3OP_F( "i32.trunc_u/f64",  0,  i_32,   d_convertOpList (u32_Trunc_f64),        Compile_Convert ),  // 0xab
2449
2450
    M3OP( "i64.extend_s/i32",   0,  i_64,   d_unaryOpList (i64, Extend_i32),        NULL    ),          // 0xac
2451
    M3OP( "i64.extend_u/i32",   0,  i_64,   d_unaryOpList (i64, Extend_u32),        NULL    ),          // 0xad
2452
2453
    M3OP_F( "i64.trunc_s/f32",  0,  i_64,   d_convertOpList (i64_Trunc_f32),        Compile_Convert ),  // 0xae
2454
    M3OP_F( "i64.trunc_u/f32",  0,  i_64,   d_convertOpList (u64_Trunc_f32),        Compile_Convert ),  // 0xaf
2455
    M3OP_F( "i64.trunc_s/f64",  0,  i_64,   d_convertOpList (i64_Trunc_f64),        Compile_Convert ),  // 0xb0
2456
    M3OP_F( "i64.trunc_u/f64",  0,  i_64,   d_convertOpList (u64_Trunc_f64),        Compile_Convert ),  // 0xb1
2457
2458
    M3OP_F( "f32.convert_s/i32",0,  f_32,   d_convertOpList (f32_Convert_i32),      Compile_Convert ),  // 0xb2
2459
    M3OP_F( "f32.convert_u/i32",0,  f_32,   d_convertOpList (f32_Convert_u32),      Compile_Convert ),  // 0xb3
2460
    M3OP_F( "f32.convert_s/i64",0,  f_32,   d_convertOpList (f32_Convert_i64),      Compile_Convert ),  // 0xb4
2461
    M3OP_F( "f32.convert_u/i64",0,  f_32,   d_convertOpList (f32_Convert_u64),      Compile_Convert ),  // 0xb5
2462
2463
    M3OP_F( "f32.demote/f64",   0,  f_32,   d_unaryOpList (f32, Demote_f64),        NULL    ),          // 0xb6
2464
2465
    M3OP_F( "f64.convert_s/i32",0,  f_64,   d_convertOpList (f64_Convert_i32),      Compile_Convert ),  // 0xb7
2466
    M3OP_F( "f64.convert_u/i32",0,  f_64,   d_convertOpList (f64_Convert_u32),      Compile_Convert ),  // 0xb8
2467
    M3OP_F( "f64.convert_s/i64",0,  f_64,   d_convertOpList (f64_Convert_i64),      Compile_Convert ),  // 0xb9
2468
    M3OP_F( "f64.convert_u/i64",0,  f_64,   d_convertOpList (f64_Convert_u64),      Compile_Convert ),  // 0xba
2469
2470
    M3OP_F( "f64.promote/f32",  0,  f_64,   d_unaryOpList (f64, Promote_f32),       NULL    ),          // 0xbb
2471
2472
    M3OP_F( "i32.reinterpret/f32",0,i_32,   d_convertOpList (i32_Reinterpret_f32),  Compile_Convert ),  // 0xbc
2473
    M3OP_F( "i64.reinterpret/f64",0,i_64,   d_convertOpList (i64_Reinterpret_f64),  Compile_Convert ),  // 0xbd
2474
    M3OP_F( "f32.reinterpret/i32",0,f_32,   d_convertOpList (f32_Reinterpret_i32),  Compile_Convert ),  // 0xbe
2475
    M3OP_F( "f64.reinterpret/i64",0,f_64,   d_convertOpList (f64_Reinterpret_i64),  Compile_Convert ),  // 0xbf
2476
2477
    M3OP( "i32.extend8_s",       0,  i_32,   d_unaryOpList (i32, Extend8_s),        NULL    ),          // 0xc0
2478
    M3OP( "i32.extend16_s",      0,  i_32,   d_unaryOpList (i32, Extend16_s),       NULL    ),          // 0xc1
2479
    M3OP( "i64.extend8_s",       0,  i_64,   d_unaryOpList (i64, Extend8_s),        NULL    ),          // 0xc2
2480
    M3OP( "i64.extend16_s",      0,  i_64,   d_unaryOpList (i64, Extend16_s),       NULL    ),          // 0xc3
2481
    M3OP( "i64.extend32_s",      0,  i_64,   d_unaryOpList (i64, Extend32_s),       NULL    ),          // 0xc4
2482
2483
# ifdef DEBUG // for codepage logging. the order doesn't matter:
2484
#   define d_m3DebugOp(OP) M3OP (#OP, 0, none, { op_##OP })
2485
2486
# if d_m3HasFloat
2487
#   define d_m3DebugTypedOp(OP) M3OP (#OP, 0, none, { op_##OP##_i32, op_##OP##_i64, op_##OP##_f32, op_##OP##_f64, })
2488
# else
2489
#   define d_m3DebugTypedOp(OP) M3OP (#OP, 0, none, { op_##OP##_i32, op_##OP##_i64 })
2490
# endif
2491
2492
    d_m3DebugOp (Compile),          d_m3DebugOp (Entry),            d_m3DebugOp (End),
2493
    d_m3DebugOp (Unsupported),      d_m3DebugOp (CallRawFunction),
2494
2495
    d_m3DebugOp (GetGlobal_s32),    d_m3DebugOp (GetGlobal_s64),    d_m3DebugOp (ContinueLoop),     d_m3DebugOp (ContinueLoopIf),
2496
2497
    d_m3DebugOp (CopySlot_32),      d_m3DebugOp (PreserveCopySlot_32), d_m3DebugOp (If_s),          d_m3DebugOp (BranchIfPrologue_s),
2498
    d_m3DebugOp (CopySlot_64),      d_m3DebugOp (PreserveCopySlot_64), d_m3DebugOp (If_r),          d_m3DebugOp (BranchIfPrologue_r),
2499
2500
    d_m3DebugOp (Select_i32_rss),   d_m3DebugOp (Select_i32_srs),   d_m3DebugOp (Select_i32_ssr),   d_m3DebugOp (Select_i32_sss),
2501
    d_m3DebugOp (Select_i64_rss),   d_m3DebugOp (Select_i64_srs),   d_m3DebugOp (Select_i64_ssr),   d_m3DebugOp (Select_i64_sss),
2502
2503
# if d_m3HasFloat
2504
    d_m3DebugOp (Select_f32_sss),   d_m3DebugOp (Select_f32_srs),   d_m3DebugOp (Select_f32_ssr),
2505
    d_m3DebugOp (Select_f32_rss),   d_m3DebugOp (Select_f32_rrs),   d_m3DebugOp (Select_f32_rsr),
2506
2507
    d_m3DebugOp (Select_f64_sss),   d_m3DebugOp (Select_f64_srs),   d_m3DebugOp (Select_f64_ssr),
2508
    d_m3DebugOp (Select_f64_rss),   d_m3DebugOp (Select_f64_rrs),   d_m3DebugOp (Select_f64_rsr),
2509
# endif
2510
2511
    d_m3DebugOp (MemFill),          d_m3DebugOp (MemCopy),
2512
2513
    d_m3DebugTypedOp (SetGlobal),   d_m3DebugOp (SetGlobal_s32),    d_m3DebugOp (SetGlobal_s64),
2514
2515
    d_m3DebugTypedOp (SetRegister), d_m3DebugTypedOp (SetSlot),     d_m3DebugTypedOp (PreserveSetSlot),
2516
# endif
2517
2518
# if d_m3CascadedOpcodes
2519
    [c_waOp_extended] = M3OP( "0xFC", 0, c_m3Type_unknown,   d_emptyOpList,  Compile_ExtendedOpcode ),
2520
# endif
2521
2522
# ifdef DEBUG
2523
    M3OP( "termination", 0, c_m3Type_unknown ) // for find_operation_info
2524
# endif
2525
};
2526
2527
const M3OpInfo c_operationsFC [] =
2528
{
2529
    M3OP_F( "i32.trunc_s:sat/f32",0,  i_32,   d_convertOpList (i32_TruncSat_f32),        Compile_Convert ),  // 0x00
2530
    M3OP_F( "i32.trunc_u:sat/f32",0,  i_32,   d_convertOpList (u32_TruncSat_f32),        Compile_Convert ),  // 0x01
2531
    M3OP_F( "i32.trunc_s:sat/f64",0,  i_32,   d_convertOpList (i32_TruncSat_f64),        Compile_Convert ),  // 0x02
2532
    M3OP_F( "i32.trunc_u:sat/f64",0,  i_32,   d_convertOpList (u32_TruncSat_f64),        Compile_Convert ),  // 0x03
2533
    M3OP_F( "i64.trunc_s:sat/f32",0,  i_64,   d_convertOpList (i64_TruncSat_f32),        Compile_Convert ),  // 0x04
2534
    M3OP_F( "i64.trunc_u:sat/f32",0,  i_64,   d_convertOpList (u64_TruncSat_f32),        Compile_Convert ),  // 0x05
2535
    M3OP_F( "i64.trunc_s:sat/f64",0,  i_64,   d_convertOpList (i64_TruncSat_f64),        Compile_Convert ),  // 0x06
2536
    M3OP_F( "i64.trunc_u:sat/f64",0,  i_64,   d_convertOpList (u64_TruncSat_f64),        Compile_Convert ),  // 0x07
2537
2538
    M3OP_RESERVED, M3OP_RESERVED,
2539
2540
    M3OP( "memory.copy",            0,  none,   d_emptyOpList,                           Compile_Memory_CopyFill ), // 0x0a
2541
    M3OP( "memory.fill",            0,  none,   d_emptyOpList,                           Compile_Memory_CopyFill ), // 0x0b
2542
2543
2544
# ifdef DEBUG
2545
    M3OP( "termination", 0, c_m3Type_unknown ) // for find_operation_info
2546
# endif
2547
};
2548
2549
2550
IM3OpInfo  GetOpInfo  (m3opcode_t opcode)
2551
23.5k
{
2552
23.5k
    switch (opcode >> 8) {
2553
23.5k
    case 0x00:
2554
23.5k
        if (M3_LIKELY(opcode < M3_COUNT_OF(c_operations))) {
2555
23.5k
            return &c_operations[opcode];
2556
23.5k
        }
2557
0
        break;
2558
1
    case c_waOp_extended:
2559
1
        opcode &= 0xFF;
2560
1
        if (M3_LIKELY(opcode < M3_COUNT_OF(c_operationsFC))) {
2561
1
            return &c_operationsFC[opcode];
2562
1
        }
2563
0
        break;
2564
23.5k
    }
2565
0
    return NULL;
2566
23.5k
}
2567
2568
M3Result  CompileBlockStatements  (IM3Compilation o)
2569
1.03k
{
2570
1.03k
    M3Result result = m3Err_none;
2571
1.03k
    bool validEnd = false;
2572
2573
23.1k
    while (o->wasm < o->wasmEnd)
2574
23.1k
    {
2575
# if d_m3EnableOpTracing
2576
        if (o->numEmits)
2577
        {
2578
            EmitOp          (o, op_DumpStack);
2579
            EmitConstant32  (o, o->numOpcodes);
2580
            EmitConstant32  (o, GetMaxUsedSlotPlusOne(o));
2581
            EmitPointer     (o, o->function);
2582
2583
            o->numEmits = 0;
2584
        }
2585
# endif
2586
23.1k
        m3opcode_t opcode;
2587
23.1k
        o->lastOpcodeStart = o->wasm;
2588
23.1k
_       (Read_opcode (& opcode, & o->wasm, o->wasmEnd));                log_opcode (o, opcode);
2589
2590
        // Restrict opcodes when evaluating expressions
2591
23.1k
        if (not o->function) {
2592
21.8k
            switch (opcode) {
2593
19.8k
            case c_waOp_i32_const: case c_waOp_i64_const:
2594
20.5k
            case c_waOp_f32_const: case c_waOp_f64_const:
2595
21.8k
            case c_waOp_getGlobal: case c_waOp_end:
2596
21.8k
                break;
2597
12
            default:
2598
12
                _throw(m3Err_restrictedOpcode);
2599
21.8k
            }
2600
21.8k
        }
2601
2602
23.1k
        IM3OpInfo opinfo = GetOpInfo (opcode);
2603
2604
23.1k
        if (opinfo == NULL)
2605
23.1k
            _throw (ErrorCompile (m3Err_unknownOpcode, o, "opcode '%x' not available", opcode));
2606
2607
23.1k
        if (opinfo->compiler) {
2608
22.8k
_           ((* opinfo->compiler) (o, opcode))
2609
22.8k
        } else {
2610
275
_           (Compile_Operator (o, opcode));
2611
270
        }
2612
2613
22.6k
        o->previousOpcode = opcode;
2614
2615
22.6k
        if (opcode == c_waOp_else)
2616
1
        {
2617
1
            _throwif (m3Err_wasmMalformed, o->block.opcode != c_waOp_if);
2618
1
            validEnd = true;
2619
1
            break;
2620
1
        }
2621
22.6k
        else if (opcode == c_waOp_end)
2622
492
        {
2623
492
            validEnd = true;
2624
492
            break;
2625
492
        }
2626
22.6k
    }
2627
497
    _throwif(m3Err_wasmMalformed, !(validEnd));
2628
2629
1.03k
_catch:
2630
1.03k
    return result;
2631
493
}
2632
2633
static
2634
M3Result  PushBlockResults  (IM3Compilation o)
2635
731
{
2636
731
    M3Result result = m3Err_none;
2637
2638
731
    u16 numResults = GetFuncTypeNumResults (o->block.type);
2639
2640
23.3k
    for (u16 i = 0; i < numResults; ++i)
2641
22.6k
    {
2642
22.6k
        u8 type = GetFuncTypeResultType (o->block.type, i);
2643
2644
22.6k
        if (i == numResults - 1 and IsFpType (type))
2645
248
        {
2646
248
_           (PushRegister (o, type));
2647
247
        }
2648
22.3k
        else
2649
22.5k
_           (PushAllocatedSlot (o, type));
2650
22.5k
    }
2651
2652
731
    _catch: return result;
2653
731
}
2654
2655
2656
M3Result  CompileBlock  (IM3Compilation o, IM3FuncType i_blockType, m3opcode_t i_blockOpcode)
2657
711
{
2658
711
                                                                                        d_m3Assert (not IsRegisterAllocated (o, 0));
2659
711
                                                                                        d_m3Assert (not IsRegisterAllocated (o, 1));
2660
711
    M3CompilationScope outerScope = o->block;
2661
711
    M3CompilationScope * block = & o->block;
2662
2663
711
    block->outer            = & outerScope;
2664
711
    block->pc               = GetPagePC (o->page);
2665
711
    block->patches          = NULL;
2666
711
    block->type             = i_blockType;
2667
711
    block->depth            ++;
2668
711
    block->opcode           = i_blockOpcode;
2669
2670
    /*
2671
     The block stack frame is a little strange but for good reasons.  Because blocks need to be restarted to
2672
     compile different pathways (if/else), the incoming params must be saved.  The parameters are popped
2673
     and validated.  But, then the stack top is readjusted so they aren't subsequently overwritten.
2674
     Next, the result are preallocated to find destination slots.  But again these are immediately popped
2675
     (deallocated) and the stack top is readjusted to keep these records in pace. This allows branch instructions
2676
     to find their result landing pads.  Finally, the params are copied from the "dead" records and pushed back
2677
     onto the stack as active stack items for the CompileBlockStatements () call.
2678
2679
    [     block      ]
2680
    [     params     ]
2681
    ------------------
2682
    [     result     ]  <---- blockStackIndex
2683
    [      slots     ]
2684
    ------------------
2685
    [   saved param  ]
2686
    [     records    ]
2687
                        <----- exitStackIndex
2688
    */
2689
2690
711
_try {
2691
    // validate and dealloc params ----------------------------
2692
2693
711
    u16 stackIndex = o->stackIndex;
2694
2695
711
    u16 numParams = GetFuncTypeNumParams (i_blockType);
2696
2697
711
    if (i_blockOpcode != c_waOp_else)
2698
708
    {
2699
15.9k
        for (u16 i = 0; i < numParams; ++i)
2700
15.2k
        {
2701
15.2k
            u8 type = GetFuncTypeParamType (i_blockType, numParams - 1 - i);
2702
15.2k
_           (PopType (o, type));
2703
15.2k
        }
2704
708
    }
2705
3
    else {
2706
3
        if (IsStackPolymorphic (o) && o->block.blockStackIndex + numParams > o->stackIndex) {
2707
1
            o->stackIndex = o->block.blockStackIndex;
2708
2
        } else {
2709
2
            o->stackIndex -= numParams;
2710
2
        }
2711
3
    }
2712
2713
711
    u16 paramIndex = o->stackIndex;
2714
711
    block->exitStackIndex = paramIndex; // consume the params at block exit
2715
2716
    // keep copies of param slots in the stack
2717
711
    o->stackIndex = stackIndex;
2718
2719
    // find slots for the results ----------------------------
2720
711
    PushBlockResults (o);
2721
2722
711
    stackIndex = o->stackIndex;
2723
2724
    // dealloc but keep record of the result slots in the stack
2725
711
    u16 numResults = GetFuncTypeNumResults (i_blockType);
2726
24.9k
    while (numResults--)
2727
24.1k
        Pop (o);
2728
2729
711
    block->blockStackIndex = o->stackIndex = stackIndex;
2730
2731
    // push the params back onto the stack -------------------
2732
15.9k
    for (u16 i = 0; i < numParams; ++i)
2733
15.2k
    {
2734
15.2k
        u8 type = GetFuncTypeParamType (i_blockType, i);
2735
2736
15.2k
        u16 slot = GetSlotForStackIndex (o, paramIndex + i);
2737
15.2k
        Push (o, type, slot);
2738
2739
15.2k
        if (slot >= o->slotFirstDynamicIndex && slot != c_slotUnused)
2740
14.3k
            MarkSlotsAllocatedByType (o, slot, type);
2741
15.2k
    }
2742
2743
    //--------------------------------------------------------
2744
2745
711
_   (CompileBlockStatements (o));
2746
2747
216
_   (ValidateBlockEnd (o));
2748
2749
216
    if (o->function)    // skip for expressions
2750
23
    {
2751
23
        if (not IsStackPolymorphic (o))
2752
23
_           (ResolveBlockResults (o, & o->block, /* isBranch: */ false));
2753
2754
23
_       (UnwindBlockStack (o))
2755
2756
23
        if (not ((i_blockOpcode == c_waOp_if and numResults) or o->previousOpcode == c_waOp_else))
2757
20
        {
2758
20
            o->stackIndex = o->block.exitStackIndex;
2759
20
_           (PushBlockResults (o));
2760
19
        }
2761
23
    }
2762
2763
215
    PatchBranches (o);
2764
2765
215
    o->block = outerScope;
2766
2767
711
}   _catch: return result;
2768
215
}
2769
2770
static
2771
M3Result  CompileLocals  (IM3Compilation o)
2772
35
{
2773
35
    M3Result result;
2774
2775
35
    u32 numLocals = 0;
2776
35
    u32 numLocalBlocks;
2777
35
_   (ReadLEB_u32 (& numLocalBlocks, & o->wasm, o->wasmEnd));
2778
2779
44
    for (u32 l = 0; l < numLocalBlocks; ++l)
2780
9
    {
2781
9
        u32 varCount;
2782
9
        i8 waType;
2783
9
        u8 localType;
2784
2785
9
_       (ReadLEB_u32 (& varCount, & o->wasm, o->wasmEnd));
2786
9
_       (ReadLEB_i7 (& waType, & o->wasm, o->wasmEnd));
2787
9
_       (NormalizeType (& localType, waType));
2788
9
        numLocals += varCount;                                                          m3log (compile, "pushing locals. count: %d; type: %s", varCount, c_waTypes [localType]);
2789
2.03k
        while (varCount--)
2790
2.02k
_           (PushAllocatedSlot (o, localType));
2791
9
    }
2792
2793
35
    if (o->function)
2794
35
        o->function->numLocals = numLocals;
2795
2796
35
    _catch: return result;
2797
35
}
2798
2799
static
2800
M3Result  ReserveConstants  (IM3Compilation o)
2801
35
{
2802
35
    M3Result result = m3Err_none;
2803
2804
    // in the interest of speed, this blindly scans the Wasm code looking for any byte
2805
    // that looks like an const opcode.
2806
35
    u16 numConstantSlots = 0;
2807
2808
35
    bytes_t wa = o->wasm;
2809
2.75k
    while (wa < o->wasmEnd)
2810
2.72k
    {
2811
2.72k
        u8 code = * wa++;
2812
2.72k
        u16 addSlots = 0;
2813
2814
2.72k
        if (code == c_waOp_i32_const or code == c_waOp_f32_const)
2815
2
            addSlots = 1;
2816
2.72k
        else if (code == c_waOp_i64_const or code == c_waOp_f64_const)
2817
7
            addSlots = GetTypeNumSlots (c_m3Type_i64);
2818
2819
2.72k
        if (numConstantSlots + addSlots >= d_m3MaxConstantTableSize)
2820
0
            break;
2821
2822
2.72k
        numConstantSlots += addSlots;
2823
2.72k
    }
2824
2825
    // if constants overflow their reserved stack space, the compiler simply emits op_Const
2826
    // operations as needed. Compiled expressions (global inits) don't pass through this
2827
    // ReserveConstants function and thus always produce inline constants.
2828
2829
35
    AlignSlotToType (& numConstantSlots, c_m3Type_i64);                                         m3log (compile, "reserved constant slots: %d", numConstantSlots);
2830
2831
35
    o->slotFirstDynamicIndex = o->slotFirstConstIndex + numConstantSlots;
2832
2833
35
    if (o->slotFirstDynamicIndex >= d_m3MaxFunctionSlots)
2834
35
        _throw (m3Err_functionStackOverflow);
2835
2836
35
    _catch:
2837
35
    return result;
2838
35
}
2839
2840
2841
M3Result  CompileFunction  (IM3Function io_function)
2842
35
{
2843
35
    if (!io_function->wasm) return "function body is missing";
2844
2845
35
    IM3FuncType funcType = io_function->funcType;                   m3log (compile, "compiling: [%d] %s %s; wasm-size: %d",
2846
35
                                                                        io_function->index, m3_GetFunctionName (io_function), SPrintFuncTypeSignature (funcType), (u32) (io_function->wasmEnd - io_function->wasm));
2847
35
    IM3Runtime runtime = io_function->module->runtime;
2848
2849
35
    IM3Compilation o = & runtime->compilation;                      d_m3Assert (d_m3MaxFunctionSlots >= d_m3MaxFunctionStackHeight * (d_m3Use32BitSlots + 1))  // need twice as many slots in 32-bit mode
2850
35
    memset (o, 0x0, sizeof (M3Compilation));
2851
2852
35
    o->runtime  = runtime;
2853
35
    o->module   = io_function->module;
2854
35
    o->function = io_function;
2855
35
    o->wasm     = io_function->wasm;
2856
35
    o->wasmEnd  = io_function->wasmEnd;
2857
35
    o->block.type = funcType;
2858
2859
35
_try {
2860
    // skip over code size. the end was already calculated during parse phase
2861
35
    u32 size;
2862
35
_   (ReadLEB_u32 (& size, & o->wasm, o->wasmEnd));                  d_m3Assert (size == (o->wasmEnd - o->wasm))
2863
2864
35
_   (AcquireCompilationCodePage (o, & o->page));
2865
2866
35
    pc_t pc = GetPagePC (o->page);
2867
2868
35
    u16 numRetSlots = GetFunctionNumReturns (o->function) * c_ioSlotCount;
2869
2870
4.18k
    for (u16 i = 0; i < numRetSlots; ++i)
2871
4.14k
        MarkSlotAllocated (o, i);
2872
2873
35
    o->function->numRetSlots = o->slotFirstDynamicIndex = numRetSlots;
2874
2875
35
    u16 numArgs = GetFunctionNumArgs (o->function);
2876
2877
    // push the arg types to the type stack
2878
583
    for (u16 i = 0; i < numArgs; ++i)
2879
548
    {
2880
548
        u8 type = GetFunctionArgType (o->function, i);
2881
548
_       (PushAllocatedSlot (o, type));
2882
2883
        // prevent allocator fill-in
2884
548
        o->slotFirstDynamicIndex += c_ioSlotCount;
2885
548
    }
2886
2887
35
    o->slotMaxAllocatedIndexPlusOne = o->function->numRetAndArgSlots = o->slotFirstLocalIndex = o->slotFirstDynamicIndex;
2888
2889
35
_   (CompileLocals (o));
2890
2891
35
    u16 maxSlot = GetMaxUsedSlotPlusOne (o);
2892
2893
35
    o->function->numLocalBytes = (maxSlot - o->slotFirstLocalIndex) * sizeof (m3slot_t);
2894
2895
35
    o->slotFirstConstIndex = o->slotMaxConstIndex = maxSlot;
2896
2897
    // ReserveConstants initializes o->firstDynamicSlotNumber
2898
35
_   (ReserveConstants (o));
2899
2900
    // start tracking the max stack used (Push() also updates this value) so that op_Entry can precisely detect stack overflow
2901
35
    o->maxStackSlots = o->slotMaxAllocatedIndexPlusOne = o->slotFirstDynamicIndex;
2902
2903
35
    o->block.blockStackIndex = o->stackFirstDynamicIndex = o->stackIndex;                           m3log (compile, "start stack index: %d",
2904
35
                                                                                                          (u32) o->stackFirstDynamicIndex);
2905
35
_   (EmitOp (o, op_Entry));
2906
35
    EmitPointer (o, io_function);
2907
2908
35
_   (CompileBlockStatements (o));
2909
2910
    // TODO: validate opcode sequences
2911
2
    _throwif(m3Err_wasmMalformed, o->previousOpcode != c_waOp_end);
2912
2913
2
    io_function->compiled = pc;
2914
2
    io_function->maxStackSlots = o->maxStackSlots;
2915
2916
2
    u16 numConstantSlots = o->slotMaxConstIndex - o->slotFirstConstIndex;                           m3log (compile, "unique constant slots: %d; unused slots: %d",
2917
2
                                                                                                           numConstantSlots, o->slotFirstDynamicIndex - o->slotMaxConstIndex);
2918
2
    io_function->numConstantBytes = numConstantSlots * sizeof (m3slot_t);
2919
2920
2
    if (numConstantSlots)
2921
1
    {
2922
1
        io_function->constants = m3_CopyMem (o->constants, io_function->numConstantBytes);
2923
1
        _throwifnull(io_function->constants);
2924
1
    }
2925
2926
35
} _catch:
2927
2928
35
    ReleaseCompilationCodePage (o);
2929
2930
35
    return result;
2931
2
}