sourcepawn/jit/x86/opcode_helpers.cpp
author David Anderson <dvander@alliedmods.net>
Thu Jan 25 22:32:29 2007 +0000 (2007-01-25)
changeset 367 16458628caa0
parent 350 1e4e30f7d6ca
child 368 a36eb0a9ab02
permissions -rw-r--r--
Added header license info
dvander@367
     1
**
dvander@367
     2
 * ===============================================================
dvander@367
     3
 * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@367
     4
 * ===============================================================
dvander@367
     5
 *
dvander@367
     6
 * This file is not open source and may not be copied without explicit
dvander@367
     7
 * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@367
     8
 * in whole or significant part.
dvander@367
     9
 * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@367
    10
 *
dvander@367
    11
 * Version: $Id$
dvander@367
    12
 */
dvander@367
    13
dvander@157
    14
#include <limits.h>
dvander@157
    15
#include <string.h>
dvander@157
    16
#include <malloc.h>
dvander@157
    17
#include "jit_x86.h"
dvander@157
    18
#include "opcode_helpers.h"
dvander@157
    19
#include "x86_macros.h"
dvander@157
    20
dvander@157
    21
#define NUM_INFO_PARAMS	5
dvander@157
    22
dvander@157
    23
jitoffs_t Write_Execute_Function(JitWriter *jit)
dvander@157
    24
{
dvander@157
    25
	/** 
dvander@157
    26
	 * The variables we're passed in:
dvander@157
    27
	 *  sp_context_t *ctx, uint32_t code_idx, cell_t *result
dvander@157
    28
	 */
dvander@157
    29
dvander@157
    30
	/**
dvander@157
    31
	 * !NOTE!
dvander@157
    32
	 * Currently, we do not accept ctx->frm as the new frame pointer.
dvander@157
    33
	 * Instead, we copy the frame from the stack pointer.
dvander@157
    34
	 * This is because we do not support resuming or sleeping!
dvander@157
    35
	 */
dvander@157
    36
dvander@157
    37
	//push ebp
dvander@157
    38
	//mov ebp, esp
dvander@157
    39
	IA32_Push_Reg(jit, REG_EBP);
dvander@157
    40
	IA32_Mov_Reg_Rm(jit, REG_EBP, REG_ESP, MOD_REG);
dvander@157
    41
dvander@157
    42
	//push esi
dvander@157
    43
	//push edi
dvander@157
    44
	//push ebx
dvander@157
    45
	IA32_Push_Reg(jit, REG_ESI);
dvander@157
    46
	IA32_Push_Reg(jit, REG_EDI);
dvander@157
    47
	IA32_Push_Reg(jit, REG_EBX);
dvander@157
    48
dvander@157
    49
	//sub esp, 4*n	- reserve info array
dvander@157
    50
	//mov esi, esp	- save info pointer
dvander@157
    51
	IA32_Sub_Rm_Imm8(jit, REG_ESP, 4*NUM_INFO_PARAMS, MOD_REG);
dvander@157
    52
	IA32_Mov_Reg_Rm(jit, AMX_REG_INFO, REG_ESP, MOD_REG);
dvander@157
    53
dvander@157
    54
	/* Initial memory setup */
dvander@157
    55
	//mov eax, [ebp+16]		- get result pointer
dvander@157
    56
	//mov [esi+8], eax		- store into info pointer
dvander@157
    57
	//mov eax, [ebp+8]		- get context
dvander@157
    58
	//mov [esi+12], eax		- store context into info pointer
dvander@157
    59
	//mov ecx, [eax+<offs>]	- get heap pointer
dvander@157
    60
	//mov [esi+4], ecx		- store heap into info pointer
dvander@157
    61
	//mov ebp, [eax+<offs>]	- get data pointer
dvander@157
    62
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, REG_EBP, 16);
dvander@157
    63
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_INFO, REG_EAX, AMX_INFO_RETVAL);
dvander@157
    64
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, REG_EBP, 8);
dvander@157
    65
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_INFO, REG_EAX, AMX_INFO_CONTEXT);
dvander@157
    66
	IA32_Mov_Reg_Rm_Disp8(jit, REG_ECX, REG_EAX, offsetof(sp_context_t, hp));
dvander@157
    67
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_INFO, REG_ECX, AMX_INFO_HEAP);
dvander@157
    68
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_DAT, REG_EAX, offsetof(sp_context_t, memory));
dvander@157
    69
dvander@157
    70
	/* Frame setup */
dvander@157
    71
	//mov edi, [eax+<offs>]	- get stack pointer
dvander@157
    72
	//add edi, ebp			- relocate to data section
dvander@157
    73
	//mov ebx, edi			- copy sp to frm
dvander@157
    74
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_STK, REG_EAX, offsetof(sp_context_t, sp));
dvander@157
    75
	IA32_Add_Rm_Reg(jit, AMX_REG_STK, AMX_REG_DAT, MOD_REG);
dvander@157
    76
	IA32_Mov_Reg_Rm(jit, AMX_REG_FRM, AMX_REG_STK, MOD_REG);
dvander@157
    77
dvander@157
    78
	/* Info memory setup */
dvander@157
    79
	//mov ecx, [eax+<offs>] - copy memsize to temp var
dvander@157
    80
	//add ecx, ebp			- relocate
dvander@157
    81
	//mov [esi+x], ecx		- store relocated
dvander@157
    82
	IA32_Mov_Reg_Rm_Disp8(jit, REG_ECX, REG_EAX, offsetof(sp_context_t, memory));
dvander@157
    83
	IA32_Add_Reg_Rm(jit, AMX_REG_TMP, AMX_REG_DAT, MOD_REG);
dvander@157
    84
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_INFO, REG_ECX, AMX_INFO_STACKTOP);
dvander@157
    85
dvander@157
    86
	/* Remaining needed vars */
dvander@157
    87
	//mov ecx, [esp+(4*(NUM_INFO_PARAMS+3))+12] - get code index (normally esp+12, but we have another array on the stack)
dvander@157
    88
	//add ecx, [eax+<offs>]   - add code base to index
dvander@157
    89
	IA32_Mov_Reg_Esp_Disp8(jit, REG_ECX, 12+(4*(NUM_INFO_PARAMS+3)));
dvander@157
    90
	IA32_Add_Reg_Rm_Disp8(jit, REG_ECX, REG_EAX, offsetof(sp_context_t, codebase));
dvander@157
    91
dvander@157
    92
	/* by now, everything is set up, so we can call into the plugin */
dvander@157
    93
	//call ecx
dvander@157
    94
	IA32_Call_Reg(jit, REG_ECX);
dvander@157
    95
dvander@157
    96
	/* if the code flow gets to here, there was a normal return */
dvander@157
    97
	//mov ecx, [esi+8]		- get retval pointer
dvander@157
    98
	//mov [ecx], eax		- store retval from PRI
dvander@157
    99
	//mov eax, SP_ERROR_NONE	- set no error
dvander@157
   100
	IA32_Mov_Reg_Rm_Disp8(jit, REG_ECX, AMX_REG_INFO, AMX_INFO_RETVAL);
dvander@157
   101
	IA32_Mov_Rm_Reg(jit, REG_ECX, AMX_REG_PRI, MOD_MEM_REG);
dvander@157
   102
	IA32_Mov_Reg_Imm32(jit, REG_EAX, SP_ERROR_NONE);
dvander@157
   103
dvander@157
   104
	/* save where error checking/halting functions should go to */
dvander@157
   105
	jitoffs_t offs_return = jit->get_outputpos();
dvander@157
   106
	//mov esp, esi			- restore stack pointer
dvander@157
   107
	IA32_Mov_Reg_Rm(jit, REG_ESP, REG_ESI, MOD_REG);
dvander@157
   108
dvander@157
   109
	/* _FOR NOW_ ...
dvander@157
   110
	 * We are going to restore SP, HP, and FRM for now.  This is for 
dvander@157
   111
     * debugging only, to check for alignment errors.  As such:
dvander@157
   112
	 * :TODO: probably remove this.
dvander@157
   113
	 */
dvander@157
   114
	//mov ecx, [esi+context]
dvander@157
   115
	//sub edi, ebp
dvander@157
   116
	//mov edx, [esi+heap]
dvander@157
   117
	//mov [ecx+sp], edi
dvander@157
   118
	//mov [ecx+hp], edx
dvander@157
   119
	IA32_Mov_Reg_Rm_Disp8(jit, REG_ECX, REG_ESI, AMX_INFO_CONTEXT);
dvander@157
   120
	IA32_Sub_Reg_Rm(jit, REG_EDI, REG_EBP, MOD_REG);
dvander@157
   121
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EDX, REG_ESI, AMX_INFO_HEAP);
dvander@157
   122
	IA32_Mov_Rm_Reg_Disp8(jit, REG_ECX, REG_EDI, offsetof(sp_context_t, sp));
dvander@157
   123
	IA32_Mov_Rm_Reg_Disp8(jit, REG_ECX, REG_EDX, offsetof(sp_context_t, hp));
dvander@157
   124
dvander@157
   125
	//add esp, 4*NUM_INFO_PARAMS
dvander@157
   126
	//pop ebx
dvander@157
   127
	//pop edi
dvander@157
   128
	//pop esi
dvander@157
   129
	//pop ebp
dvander@157
   130
	//ret
dvander@157
   131
	IA32_Add_Rm_Imm8(jit, REG_ESP, 4*NUM_INFO_PARAMS, MOD_REG);
dvander@157
   132
	IA32_Pop_Reg(jit, REG_EBX);
dvander@157
   133
	IA32_Pop_Reg(jit, REG_EDI);
dvander@157
   134
	IA32_Pop_Reg(jit, REG_ESI);
dvander@157
   135
	IA32_Pop_Reg(jit, REG_EBP);
dvander@157
   136
	IA32_Return(jit);
dvander@157
   137
dvander@157
   138
	return offs_return;
dvander@157
   139
}
dvander@157
   140
dvander@157
   141
void Write_BreakDebug(JitWriter *jit)
dvander@157
   142
{
dvander@241
   143
	//push edi
dvander@241
   144
	//mov edi, ecx
dvander@157
   145
	//mov ecx, [esi+ctx]
dvander@157
   146
	//cmp [ecx+dbreak], 0
dvander@157
   147
	//jnz :nocall
dvander@241
   148
	IA32_Push_Reg(jit, REG_EDI);
dvander@241
   149
	IA32_Mov_Reg_Rm(jit, REG_EDI, REG_ECX, MOD_REG);
dvander@157
   150
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@157
   151
	IA32_Cmp_Rm_Disp8_Imm8(jit, AMX_REG_TMP, offsetof(sp_context_t, dbreak), 0);
dvander@241
   152
	jitoffs_t jmp = IA32_Jump_Cond_Imm8(jit, CC_Z, 0);
dvander@157
   153
dvander@241
   154
	/* NOTE, Hack! PUSHAD pushes EDI last which still has the CIP */
dvander@157
   155
	//pushad
dvander@157
   156
	//push [esi+frm]
faluco@238
   157
	//push ctx
dvander@157
   158
	//mov ecx, [ecx+dbreak]
dvander@157
   159
	//call ecx
dvander@157
   160
	//add esp, 8
dvander@157
   161
	//popad
dvander@241
   162
	IA32_Pushad(jit);
dvander@157
   163
	IA32_Push_Rm_Disp8(jit, AMX_REG_INFO, AMX_INFO_FRAME); //:TODO: move to regs and push? and dont disp for 0
dvander@237
   164
	IA32_Push_Reg(jit, AMX_REG_TMP);
dvander@157
   165
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_TMP, offsetof(sp_context_t, dbreak));
dvander@157
   166
	IA32_Call_Reg(jit, AMX_REG_TMP);
dvander@157
   167
	IA32_Add_Rm_Imm8(jit, REG_ESP, 4*2, MOD_REG);
dvander@157
   168
	IA32_Popad(jit);
dvander@157
   169
dvander@157
   170
	//:nocall
dvander@241
   171
	//pop edi
dvander@241
   172
	//ret
dvander@241
   173
	IA32_Pop_Reg(jit, REG_EDI);
dvander@157
   174
	IA32_Send_Jump8_Here(jit, jmp);
dvander@157
   175
	IA32_Return(jit);
dvander@157
   176
}
dvander@157
   177
dvander@157
   178
void Write_GetError(JitWriter *jit)
dvander@157
   179
{
dvander@157
   180
	CompData *data = (CompData *)jit->data;
dvander@157
   181
dvander@157
   182
	//mov eax, [esi+info.context]
dvander@157
   183
	//mov eax, [eax+ctx.error]
dvander@157
   184
	//jmp [jit_return]
dvander@157
   185
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@237
   186
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, REG_EAX, offsetof(sp_context_t, n_err));
dvander@157
   187
	IA32_Jump_Imm32_Abs(jit, data->jit_return);
dvander@157
   188
}
dvander@157
   189
dvander@157
   190
void Write_SetError(JitWriter *jit, int error)
dvander@157
   191
{
dvander@157
   192
	CompData *data = (CompData *)jit->data;
dvander@157
   193
dvander@157
   194
	//mov eax, <error>
dvander@157
   195
	//jmp [jit_return]
dvander@157
   196
	IA32_Mov_Reg_Imm32(jit, REG_EAX, error);
dvander@157
   197
	IA32_Jump_Imm32_Abs(jit, data->jit_return);
dvander@157
   198
}
dvander@157
   199
dvander@157
   200
void Write_Check_DivZero(JitWriter *jit, jit_uint8_t reg)
dvander@157
   201
{
dvander@157
   202
	//test reg, reg
dvander@157
   203
	//jz :error
dvander@157
   204
	IA32_Test_Rm_Reg(jit, reg, reg, MOD_REG);
dvander@157
   205
	IA32_Jump_Cond_Imm32_Abs(jit, CC_Z, ((CompData *)jit->data)->jit_error_divzero);
dvander@157
   206
}
dvander@157
   207
dvander@157
   208
void Write_CheckHeap_Min(JitWriter *jit)
dvander@157
   209
{
dvander@157
   210
	/* Check if the stack went beyond the heap low.
dvander@157
   211
	 * This usually means there was a compiler error.
dvander@157
   212
	 * NOTE: Special optimization here.
dvander@157
   213
	 * The heap low is always known ahead of time! :)
dvander@157
   214
	 */
dvander@157
   215
	CompData *data = (CompData *)jit->data;
dvander@157
   216
	//cmp [esi+info.heap], <heaplow>
dvander@157
   217
	//jb :error
dvander@157
   218
	IA32_Cmp_Rm_Imm32_Disp8(jit, AMX_REG_INFO, AMX_INFO_HEAP, data->plugin->data_size);
dvander@157
   219
	IA32_Jump_Cond_Imm32_Abs(jit, CC_B, data->jit_error_heapmin);
dvander@157
   220
}
dvander@157
   221
dvander@157
   222
void Write_CheckHeap_Low(JitWriter *jit)
dvander@157
   223
{
dvander@157
   224
	/* Check if the heap is trying to grow beyond the stack.
dvander@157
   225
	 */
dvander@157
   226
	//mov ecx, [esi+info.heap]
dvander@157
   227
	//lea ecx, [ebp+ecx+STACK_MARGIN]
dvander@157
   228
	//cmp ecx, edi
dvander@157
   229
	//ja :error		; I think this is right
dvander@157
   230
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   231
	IA32_Lea_Reg_DispRegMultImm8(jit, AMX_REG_TMP, AMX_REG_DAT, AMX_REG_TMP, NOSCALE, STACK_MARGIN);
dvander@157
   232
	IA32_Cmp_Reg_Rm(jit, AMX_REG_TMP, AMX_REG_STK, MOD_REG);
dvander@157
   233
	IA32_Jump_Cond_Imm32_Abs(jit, CC_A, ((CompData *)jit->data)->jit_error_heaplow);
dvander@157
   234
}
dvander@157
   235
dvander@157
   236
void Write_CheckStack_Min(JitWriter *jit)
dvander@157
   237
{
dvander@157
   238
	/* Check if the stack went beyond the stack top
dvander@157
   239
	 * This usually means there was a compiler error.
dvander@157
   240
	 */
dvander@157
   241
	//cmp edi, [esi+info.stacktop]
dvander@157
   242
	//jae :error
dvander@157
   243
	IA32_Cmp_Reg_Rm_Disp8(jit, AMX_REG_STK, AMX_REG_INFO, AMX_INFO_STACKTOP);
dvander@157
   244
	IA32_Jump_Cond_Imm32_Abs(jit, CC_AE, ((CompData *)jit->data)->jit_error_stackmin);
dvander@157
   245
}
dvander@157
   246
dvander@157
   247
void Write_CheckStack_Low(JitWriter *jit)
dvander@157
   248
{
dvander@157
   249
	/* Check if the stack went beyond the heap boundary.
dvander@157
   250
	 * Unfortunately this one isn't as quick as the other check.
dvander@157
   251
	 * The stack margin check is important for sysreq.n having space.
dvander@157
   252
	 */
dvander@157
   253
	//mov ecx, [esi+info.heap]
dvander@157
   254
	//lea ecx, [ebp+ecx+STACK_MARGIN]
dvander@157
   255
	//cmp edi, ecx
dvander@157
   256
	//jb :error		; I think this is right
dvander@157
   257
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   258
	IA32_Lea_Reg_DispRegMultImm8(jit, AMX_REG_TMP, AMX_REG_DAT, AMX_REG_TMP, NOSCALE, STACK_MARGIN);
dvander@157
   259
	IA32_Cmp_Reg_Rm(jit, AMX_REG_STK, AMX_REG_TMP, MOD_REG);
dvander@157
   260
	IA32_Jump_Cond_Imm32_Abs(jit, CC_B, ((CompData *)jit->data)->jit_error_stacklow);
dvander@157
   261
}
dvander@157
   262
dvander@157
   263
void Write_Check_VerifyAddr(JitWriter *jit, jit_uint8_t reg)
dvander@157
   264
{
dvander@157
   265
	CompData *data = (CompData *)jit->data;
dvander@157
   266
dvander@157
   267
	/* :TODO: Should this be checking for below heaplow?
dvander@157
   268
	 * The old JIT did not.
dvander@157
   269
	 */
dvander@157
   270
dvander@157
   271
	bool call = false;
dvander@157
   272
	if (!(data->inline_level & JIT_INLINE_ERRORCHECKS))
dvander@157
   273
	{
dvander@157
   274
		/* If we're not in the initial generation phase,
dvander@157
   275
		 * Write a call to the actual routine instead.
dvander@157
   276
		 */
dvander@157
   277
		if ((reg == REG_EAX) && data->jit_verify_addr_eax)
dvander@157
   278
		{
dvander@157
   279
			jitoffs_t call = IA32_Call_Imm32(jit, 0);
dvander@157
   280
			IA32_Write_Jump32(jit, call, data->jit_verify_addr_eax);
dvander@157
   281
			return;
dvander@157
   282
		} else if ((reg == REG_EDX) && data->jit_verify_addr_edx) {
dvander@157
   283
			jitoffs_t call = IA32_Call_Imm32(jit, 0);
dvander@157
   284
			IA32_Write_Jump32(jit, call, data->jit_verify_addr_edx);
dvander@157
   285
			return;
dvander@157
   286
		}
dvander@157
   287
		call = true;
dvander@157
   288
	}
dvander@157
   289
dvander@157
   290
	/** 
dvander@157
   291
	 * :TODO: If we can't find a nicer way of doing this,
dvander@157
   292
	 *  then scrap it on high optimizations.  The second portion is not needed at all!
dvander@157
   293
	 */
dvander@157
   294
dvander@157
   295
	/* Part 1: Check if we're in the memory bounds */
dvander@157
   296
	//cmp <reg>, <stpu>
dvander@157
   297
	//jae :error
dvander@157
   298
	IA32_Cmp_Rm_Imm32(jit, MOD_REG, reg, ((CompData *)jit->data)->plugin->memory);
dvander@157
   299
	IA32_Jump_Cond_Imm32_Abs(jit, CC_AE, ((CompData *)jit->data)->jit_error_memaccess);
dvander@157
   300
dvander@157
   301
	/* Part 2: Check if we're in the invalid region between HP and SP */
dvander@157
   302
	jitoffs_t jmp;
dvander@157
   303
	//cmp <reg>, [esi+info.heap]
dvander@157
   304
	//jb :continue
dvander@157
   305
	//lea ecx, [ebp+<reg>]
dvander@157
   306
	//cmp edi, ecx
dvander@157
   307
	//jb :error
dvander@157
   308
	//:continue
dvander@157
   309
	IA32_Cmp_Reg_Rm_Disp8(jit, reg, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   310
	jmp = IA32_Jump_Cond_Imm8(jit, CC_B, 0);
dvander@157
   311
	IA32_Lea_Reg_DispRegMultImm8(jit, AMX_REG_TMP, AMX_REG_DAT, reg, NOSCALE, 0);
dvander@157
   312
	IA32_Cmp_Reg_Rm(jit, AMX_REG_TMP, AMX_REG_STK, MOD_REG);
dvander@157
   313
	IA32_Jump_Cond_Imm32_Abs(jit, CC_B, ((CompData *)jit->data)->jit_error_memaccess);
dvander@157
   314
	IA32_Send_Jump8_Here(jit, jmp);
dvander@157
   315
dvander@157
   316
	if (call)
dvander@157
   317
	{
dvander@157
   318
		IA32_Return(jit);
dvander@157
   319
	}
dvander@157
   320
}
dvander@157
   321
dvander@157
   322
void Macro_PushN_Addr(JitWriter *jit, int i)
dvander@157
   323
{
dvander@157
   324
	//push eax
dvander@157
   325
	//mov eax, [esi+frm]
dvander@157
   326
	//loop i times:
dvander@157
   327
	// lea ecx, [eax+<val>]
dvander@157
   328
	// mov [edi-4*i], ecx
dvander@157
   329
	//sub edi, 4*N
dvander@157
   330
	//pop eax
dvander@157
   331
dvander@157
   332
	cell_t val;
dvander@157
   333
	int n = 1;
dvander@157
   334
	IA32_Push_Reg(jit, AMX_REG_PRI);
dvander@157
   335
	IA32_Mov_Reg_Rm(jit, AMX_REG_PRI, AMX_REG_INFO, MOD_MEM_REG);
dvander@157
   336
	do
dvander@157
   337
	{
dvander@157
   338
		val = jit->read_cell();
dvander@157
   339
		if (val < SCHAR_MAX && val > SCHAR_MIN)
dvander@157
   340
			IA32_Lea_DispRegImm8(jit, AMX_REG_TMP, AMX_REG_PRI, (jit_int8_t)val);
dvander@157
   341
		else
dvander@157
   342
			IA32_Lea_DispRegImm32(jit, AMX_REG_TMP, AMX_REG_PRI, val);
dvander@157
   343
		IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_STK, AMX_REG_TMP, -4*n);
dvander@157
   344
	} while (n++ < i);
dvander@157
   345
	IA32_Sub_Rm_Imm8(jit, AMX_REG_STK, 4*i, MOD_REG);
dvander@157
   346
	IA32_Pop_Reg(jit, AMX_REG_PRI);
dvander@157
   347
}
dvander@157
   348
dvander@157
   349
void Macro_PushN_S(JitWriter *jit, int i)
dvander@157
   350
{
dvander@157
   351
	//loop i times:
dvander@157
   352
	// mov ecx, [ebx+<val>]
dvander@157
   353
	// mov [edi-4*i], ecx
dvander@157
   354
	//sub edi, 4*N
dvander@157
   355
dvander@157
   356
	cell_t val;
dvander@157
   357
	int n = 1;
dvander@157
   358
	do 
dvander@157
   359
	{
dvander@157
   360
		val = jit->read_cell();
dvander@157
   361
		if (val < SCHAR_MAX && val > SCHAR_MIN)
dvander@157
   362
			IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_FRM, (jit_int8_t)val);
dvander@157
   363
		else
dvander@157
   364
			IA32_Mov_Reg_Rm_Disp32(jit, AMX_REG_TMP, AMX_REG_FRM, val);
dvander@157
   365
		IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_STK, AMX_REG_TMP, -4*n);
dvander@157
   366
	} while (n++ < i);
dvander@157
   367
	IA32_Sub_Rm_Imm8(jit, AMX_REG_STK, 4*i, MOD_REG);
dvander@157
   368
}
dvander@157
   369
dvander@157
   370
void Macro_PushN_C(JitWriter *jit, int i)
dvander@157
   371
{
dvander@157
   372
	//loop i times:
dvander@157
   373
	// mov [edi-4*i], <val>
dvander@157
   374
	//sub edi, 4*N
dvander@157
   375
dvander@157
   376
	int n = 1;
dvander@157
   377
	do 
dvander@157
   378
	{
dvander@157
   379
		IA32_Mov_Rm_Imm32_Disp8(jit, AMX_REG_STK, jit->read_cell(), -4*n);
dvander@157
   380
	} while (n++ < i);
dvander@157
   381
	IA32_Sub_Rm_Imm8(jit, AMX_REG_STK, 4*i, MOD_REG);
dvander@157
   382
}
dvander@157
   383
dvander@157
   384
void Macro_PushN(JitWriter *jit, int i)
dvander@157
   385
{
dvander@157
   386
	//loop i times:
dvander@157
   387
	// mov ecx, [ebp+<val>]
dvander@157
   388
	// mov [edi-4*i], ecx
dvander@157
   389
	//sub edi, 4*N
dvander@157
   390
dvander@157
   391
	cell_t val;
dvander@157
   392
	int n = 1;
dvander@157
   393
	do 
dvander@157
   394
	{
dvander@157
   395
		val = jit->read_cell();
dvander@157
   396
		if (val < SCHAR_MAX && val > SCHAR_MIN)
dvander@157
   397
			IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_DAT, (jit_int8_t)val);
dvander@157
   398
		else
dvander@157
   399
			IA32_Mov_Reg_Rm_Disp32(jit, AMX_REG_TMP, AMX_REG_DAT, val);
dvander@157
   400
		IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_STK, AMX_REG_TMP, -4*n);
dvander@157
   401
	} while (n++ < i);
dvander@157
   402
	IA32_Sub_Rm_Imm8(jit, AMX_REG_STK, 4*i, MOD_REG);
dvander@157
   403
}
dvander@157
   404
dvander@157
   405
void WriteOp_Sysreq_C_Function(JitWriter *jit)
dvander@157
   406
{
dvander@157
   407
	/* The small daddy of the big daddy of opcodes.
dvander@157
   408
	 * ecx - native index
dvander@157
   409
	 */
dvander@157
   410
	CompData *data = (CompData *)jit->data;
dvander@157
   411
dvander@157
   412
	/* save registers we will need */
dvander@157
   413
	//push edx
dvander@157
   414
	IA32_Push_Reg(jit, AMX_REG_ALT);
dvander@157
   415
dvander@157
   416
	/* push some callback stuff */
dvander@157
   417
	//push edi		; stack
dvander@157
   418
	//push ecx		; native index
dvander@157
   419
	IA32_Push_Reg(jit, AMX_REG_STK);
dvander@157
   420
	IA32_Push_Reg(jit, REG_ECX);
dvander@157
   421
dvander@157
   422
	/* Relocate stack, heap, frm information, then store back */
dvander@157
   423
	//mov eax, [esi+context]
dvander@157
   424
	//mov ecx, [esi+hea]
dvander@157
   425
	//sub edi, ebp
dvander@157
   426
	//mov [eax+hp], ecx
dvander@157
   427
	//mov ecx, [esi]
dvander@157
   428
	//mov [eax+sp], edi
dvander@157
   429
	//mov [eax+frm], ecx
dvander@157
   430
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@157
   431
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   432
	IA32_Sub_Reg_Rm(jit, AMX_REG_STK, AMX_REG_DAT, MOD_REG);
dvander@157
   433
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_TMP, offsetof(sp_context_t, hp));
dvander@157
   434
	IA32_Mov_Reg_Rm(jit, AMX_REG_TMP, AMX_INFO_FRM, MOD_MEM_REG);
dvander@157
   435
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_STK, offsetof(sp_context_t, sp));
dvander@157
   436
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_TMP, offsetof(sp_context_t, frm));
dvander@157
   437
dvander@157
   438
	/* finally, push the last parameter and make the call */
dvander@157
   439
	//push eax		; context
dvander@157
   440
	//call NativeCallback
dvander@157
   441
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   442
	jitoffs_t call = IA32_Call_Imm32(jit, 0);
dvander@157
   443
	if (!data->debug)
dvander@157
   444
	{
dvander@350
   445
		IA32_Write_Jump32_Abs(jit, call, (void *)NativeCallback);
dvander@157
   446
	} else {
dvander@350
   447
		IA32_Write_Jump32_Abs(jit, call, (void *)NativeCallback_Debug);
dvander@157
   448
	}
dvander@157
   449
dvander@157
   450
	/* Test for error */
dvander@157
   451
	//mov ecx, [esi+context]
dvander@157
   452
	//cmp [ecx+err], 0
dvander@157
   453
	//jnz :error
dvander@157
   454
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@237
   455
	IA32_Cmp_Rm_Disp8_Imm8(jit, AMX_REG_TMP, offsetof(sp_context_t, n_err), 0);
dvander@157
   456
	IA32_Jump_Cond_Imm32_Abs(jit, CC_NZ, data->jit_extern_error);
dvander@157
   457
dvander@157
   458
	/* restore what we damaged */
dvander@157
   459
	//add esp, 4*3
dvander@157
   460
	//add edi, ebp
dvander@157
   461
	//pop edx
dvander@157
   462
	IA32_Add_Rm_Imm8(jit, REG_ESP, 4*3, MOD_REG);
dvander@157
   463
	IA32_Add_Reg_Rm(jit, AMX_REG_STK, AMX_REG_DAT, MOD_REG);
dvander@157
   464
	IA32_Pop_Reg(jit, AMX_REG_ALT);
dvander@157
   465
dvander@157
   466
	//ret
dvander@157
   467
	IA32_Return(jit);
dvander@157
   468
}
dvander@157
   469
dvander@157
   470
void GenerateArrayIndirectionVectors(cell_t *arraybase, cell_t dims[], ucell_t _dimcount, bool autozero)
dvander@157
   471
{
dvander@157
   472
	cell_t vectors = 1;				/* we need one vector to start off with */
dvander@157
   473
	cell_t cur_offs = 0;
dvander@157
   474
	cell_t cur_write = 0;
dvander@157
   475
	cell_t dimcount = _dimcount;
dvander@157
   476
dvander@157
   477
	/* Initialize rotation */
dvander@157
   478
	cur_write = dims[dimcount-1];
dvander@157
   479
	
dvander@157
   480
	while (--dimcount >= 1)
dvander@157
   481
	{
dvander@157
   482
		cell_t cur_dim = dims[dimcount];
dvander@157
   483
		cell_t sub_dim = dims[dimcount-1];
dvander@157
   484
		for (cell_t i=0; i<vectors; i++)
dvander@157
   485
		{
dvander@157
   486
			for (cell_t j=0; j<cur_dim; j++)
dvander@157
   487
			{
dvander@157
   488
				arraybase[cur_offs] = (cur_write - cur_offs)*sizeof(cell_t);
dvander@157
   489
				cur_offs++;
dvander@157
   490
				cur_write += sub_dim;
dvander@157
   491
			}
dvander@157
   492
		}
dvander@157
   493
		vectors = cur_dim;
dvander@157
   494
	}
dvander@157
   495
dvander@157
   496
	/* everything after cur_offs can be zeroed */
dvander@157
   497
	if (autozero)
dvander@157
   498
	{
dvander@157
   499
		size_t size = 1;
dvander@157
   500
		for (ucell_t i=0; i<_dimcount; i++)
dvander@157
   501
		{
dvander@157
   502
			size *= dims[i];
dvander@157
   503
		}
dvander@157
   504
		memset(&arraybase[cur_offs], 0, size*sizeof(cell_t));
dvander@157
   505
	}
dvander@157
   506
dvander@157
   507
	return;
dvander@157
   508
}
dvander@157
   509
dvander@157
   510
/**
dvander@157
   511
 * A few notes about this function.
dvander@157
   512
 * I was more concerned about efficient use of registers here, rather than 
dvander@157
   513
 *  fine-tuned optimization.  The reason is that the code is already complicated,
dvander@157
   514
 *  and it is very easy to mess up.
dvander@157
   515
 */
dvander@157
   516
void WriteIntrinsic_GenArray(JitWriter *jit)
dvander@157
   517
{
dvander@157
   518
	/**
dvander@157
   519
	 * save important values
dvander@157
   520
	 */
dvander@157
   521
	//push ebx
dvander@157
   522
	//push eax
dvander@157
   523
	//push edx
dvander@157
   524
	//push ecx				;value is referenced on stack
dvander@157
   525
	IA32_Push_Reg(jit, REG_EBX);
dvander@157
   526
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   527
	IA32_Push_Reg(jit, REG_EDX);
dvander@157
   528
	IA32_Push_Reg(jit, REG_ECX);
dvander@157
   529
dvander@157
   530
	/**
dvander@157
   531
	 * Calculate how many cells will be needed.
dvander@157
   532
	 */
dvander@157
   533
	//mov edx, [edi]		;get last dimension's count
dvander@157
   534
	//mov eax, 1			;position at second to last dimension
dvander@157
   535
	//:loop
dvander@157
   536
	//cmp eax, [esp]		;compare to # of params
dvander@157
   537
	//jae :done				;end loop if done
dvander@157
   538
	//mov ecx, [edi+eax*4]	;get dimension size
dvander@157
   539
	//imul edx, ecx			;multiply by size
dvander@157
   540
	//add eax, 1			;increment
dvander@157
   541
	//add edx, ecx			;add size (indirection vector)
dvander@157
   542
	//jmp :loop				;jump back
dvander@157
   543
	//:done
dvander@157
   544
	IA32_Mov_Reg_Rm(jit, REG_EDX, AMX_REG_STK, MOD_MEM_REG);
dvander@157
   545
	IA32_Mov_Reg_Imm32(jit, REG_EAX, 1);
dvander@157
   546
	jitoffs_t loop1 = jit->get_outputpos();
dvander@157
   547
	IA32_Cmp_Reg_Rm_ESP(jit, REG_EAX);
dvander@157
   548
	jitoffs_t done1 = IA32_Jump_Cond_Imm8(jit, CC_AE, 0);
dvander@157
   549
	IA32_Mov_Reg_Rm_Disp_Reg(jit, REG_ECX, AMX_REG_STK, REG_EAX, SCALE4);
dvander@157
   550
	IA32_IMul_Reg_Rm(jit, REG_EDX, REG_ECX, MOD_REG);
dvander@157
   551
	IA32_Add_Rm_Imm8(jit, REG_EAX, 1, MOD_REG);
dvander@157
   552
	IA32_Add_Reg_Rm(jit, REG_EDX, REG_ECX, MOD_REG);
dvander@157
   553
	IA32_Write_Jump8(jit, IA32_Jump_Imm8(jit, loop1), loop1);
dvander@157
   554
	IA32_Send_Jump8_Here(jit, done1);
dvander@157
   555
dvander@157
   556
	/* Test if we have heap space for this */
dvander@157
   557
	//mov eax, [esi+info.heap]	;get heap pointer
dvander@157
   558
	//lea eax, [eax+edx*4]		;new heap pointer
dvander@157
   559
	//cmp eax, <hlw>			;compare to heap low
dvander@157
   560
	//jbe :error				;die if we hit this (it should always be >)
dvander@157
   561
	//add eax, ebp				;relocate to stack
dvander@157
   562
	//cmp eax, edi				;die if above the stack pointer
dvander@157
   563
	//jae :error
dvander@157
   564
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   565
	IA32_Lea_Reg_DispRegMult(jit, REG_EAX, REG_EAX, REG_EDX, SCALE4);
dvander@157
   566
	IA32_Cmp_Rm_Imm32(jit, MOD_REG, REG_EAX, ((CompData *)jit->data)->plugin->data_size);
dvander@157
   567
	IA32_Jump_Cond_Imm32_Abs(jit, CC_BE, ((CompData *)jit->data)->jit_error_array_too_big);
dvander@157
   568
	IA32_Add_Reg_Rm(jit, REG_EAX, AMX_REG_DAT, MOD_REG);
dvander@157
   569
	IA32_Cmp_Reg_Rm(jit, REG_EAX, AMX_REG_STK, MOD_REG);
dvander@157
   570
	IA32_Jump_Cond_Imm32_Abs(jit, CC_AE, ((CompData *)jit->data)->jit_error_array_too_big);
dvander@157
   571
	
dvander@157
   572
	/* Prepare for indirection iteration */
dvander@157
   573
	//mov eax, [esi+info.heap]	;get heap pointer
dvander@157
   574
	//lea ebx, [eax+edx*4]		;new heap pointer
dvander@157
   575
	//mov [esi+info.heap], ebx	;store back
dvander@157
   576
	//push eax					;save heap pointer - we need it
dvander@157
   577
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   578
	IA32_Lea_Reg_DispRegMult(jit, REG_EBX, REG_EAX, REG_EDX, SCALE4);
dvander@157
   579
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_INFO, REG_EBX, AMX_INFO_HEAP);
dvander@157
   580
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   581
dvander@157
   582
	WriteOp_Tracker_Push_Reg(jit, REG_EDX);
dvander@157
   583
	
dvander@157
   584
	/* This part is too messy to do in straight assembly.
dvander@157
   585
	 * I'm letting the compiler handle it and thus it's in C.
dvander@157
   586
	 */
dvander@157
   587
	//lea ebx, [ebp+eax]		;get base pointer
dvander@157
   588
	//push dword [esp-8]		;push autozero
dvander@157
   589
	//push dword [esp-8]		;push dimension count
dvander@157
   590
	//push edi					;push dim array
dvander@157
   591
	//push ebx
dvander@157
   592
	//call GenerateArrayIndirectionVectors
dvander@157
   593
	//add esp, 4*4
dvander@157
   594
	IA32_Lea_Reg_DispRegMult(jit, REG_EBX, REG_EAX, REG_EBP, NOSCALE);
dvander@157
   595
	IA32_Push_Rm_Disp8_ESP(jit, 8);
dvander@157
   596
	IA32_Push_Rm_Disp8_ESP(jit, 8);
dvander@157
   597
	IA32_Push_Reg(jit, REG_EDI);
dvander@157
   598
	IA32_Push_Reg(jit, REG_EBX);
dvander@157
   599
	IA32_Write_Jump32_Abs(jit, IA32_Call_Imm32(jit, 0), (void *)&GenerateArrayIndirectionVectors);
dvander@157
   600
	IA32_Add_Rm_Imm8(jit, REG_ESP, 4*4, MOD_REG);
dvander@157
   601
dvander@157
   602
	/* Store the heap pointer back into the stack */
dvander@157
   603
	//pop eax					;restore heap pointer
dvander@157
   604
	//pop ecx					;restore param count
dvander@157
   605
	//lea edi, [edi+ecx*4-4]	;pop params-4 off the stack
dvander@157
   606
	//mov [edi], eax			;store back the heap pointer
dvander@157
   607
	IA32_Pop_Reg(jit, REG_EAX);
dvander@157
   608
	IA32_Pop_Reg(jit, REG_ECX);
dvander@157
   609
	IA32_Lea_Reg_DispRegMultImm8(jit, AMX_REG_STK, AMX_REG_STK, REG_ECX, SCALE4, -4);
dvander@157
   610
	IA32_Mov_Rm_Reg(jit, AMX_REG_STK, REG_EAX, MOD_MEM_REG);
dvander@157
   611
dvander@157
   612
	/* Return to caller */
dvander@157
   613
	//pop edx
dvander@157
   614
	//pop eax
dvander@157
   615
	//pop ebx
dvander@157
   616
	//ret
dvander@157
   617
	IA32_Pop_Reg(jit, REG_ECX);
dvander@157
   618
	IA32_Pop_Reg(jit, REG_EAX);
dvander@157
   619
	IA32_Pop_Reg(jit, REG_EBX);
dvander@157
   620
	IA32_Return(jit);
dvander@157
   621
}
dvander@157
   622
dvander@157
   623
void WriteOp_Sysreq_N_Function(JitWriter *jit)
dvander@157
   624
{
dvander@157
   625
	/* The big daddy of opcodes.
dvander@157
   626
	 * eax - num_params
dvander@157
   627
	 * ecx - native index
dvander@157
   628
	 */
dvander@157
   629
	CompData *data = (CompData *)jit->data;
dvander@157
   630
dvander@157
   631
	/* store the number of parameters on the stack */
dvander@157
   632
	//mov [edi-4], eax
dvander@157
   633
	//sub edi, 4
dvander@157
   634
	IA32_Mov_Rm_Reg_Disp8(jit, AMX_REG_STK, REG_EAX, -4);
dvander@157
   635
	IA32_Sub_Rm_Imm8(jit, AMX_REG_STK, 4, MOD_REG);
dvander@157
   636
dvander@157
   637
	/* save registers we will need */
dvander@157
   638
	//push eax	; num_params for stack popping
dvander@157
   639
	//push edx
dvander@157
   640
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   641
	IA32_Push_Reg(jit, AMX_REG_ALT);
dvander@157
   642
dvander@157
   643
	/* push some callback stuff */
dvander@157
   644
	//push edi		; stack
dvander@157
   645
	//push ecx		; native index
dvander@157
   646
	IA32_Push_Reg(jit, AMX_REG_STK);
dvander@157
   647
	IA32_Push_Reg(jit, REG_ECX);
dvander@157
   648
dvander@157
   649
	/* Relocate stack, heap, frm information, then store back */
dvander@157
   650
	//mov eax, [esi+context]
dvander@157
   651
	//mov ecx, [esi+hea]
dvander@157
   652
	//sub edi, ebp
dvander@157
   653
	//mov [eax+hp], ecx
dvander@157
   654
	//mov ecx, [esi]
dvander@157
   655
	//mov [eax+sp], edi
dvander@157
   656
	//mov [eax+frm], ecx
dvander@157
   657
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@157
   658
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_HEAP);
dvander@157
   659
	IA32_Sub_Reg_Rm(jit, AMX_REG_STK, AMX_REG_DAT, MOD_REG);
dvander@157
   660
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_TMP, offsetof(sp_context_t, hp));
dvander@157
   661
	IA32_Mov_Reg_Rm(jit, AMX_REG_TMP, AMX_INFO_FRM, MOD_MEM_REG);
dvander@157
   662
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_STK, offsetof(sp_context_t, sp));
dvander@157
   663
	IA32_Mov_Rm_Reg_Disp8(jit, REG_EAX, AMX_REG_TMP, offsetof(sp_context_t, frm));
dvander@157
   664
dvander@157
   665
	/* finally, push the last parameter and make the call */
dvander@157
   666
	//push eax		; context
dvander@157
   667
	//call NativeCallback
dvander@157
   668
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   669
	jitoffs_t call = IA32_Call_Imm32(jit, 0);
dvander@157
   670
	if (!data->debug)
dvander@157
   671
	{
dvander@350
   672
		IA32_Write_Jump32_Abs(jit, call, (void *)NativeCallback);
dvander@157
   673
	} else {
dvander@350
   674
		IA32_Write_Jump32_Abs(jit, call, (void *)NativeCallback_Debug);
dvander@157
   675
	}
dvander@157
   676
dvander@157
   677
	/* Test for error */
dvander@157
   678
	//mov ecx, [esi+context]
dvander@157
   679
	//cmp [ecx+err], 0
dvander@157
   680
	//jnz :error
dvander@157
   681
	IA32_Mov_Reg_Rm_Disp8(jit, AMX_REG_TMP, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@237
   682
	IA32_Cmp_Rm_Disp8_Imm8(jit, AMX_REG_TMP, offsetof(sp_context_t, n_err), 0);
dvander@157
   683
	IA32_Jump_Cond_Imm32_Abs(jit, CC_NZ, data->jit_extern_error);
dvander@157
   684
dvander@157
   685
	/* restore what we damaged */
dvander@157
   686
	//add esp, 4*3
dvander@157
   687
	//add edi, ebp
dvander@157
   688
	//pop edx
dvander@157
   689
	//pop ecx	; num_params
dvander@157
   690
	IA32_Add_Rm_Imm8(jit, REG_ESP, 4*3, MOD_REG);
dvander@157
   691
	IA32_Add_Reg_Rm(jit, AMX_REG_STK, AMX_REG_DAT, MOD_REG);
dvander@157
   692
	IA32_Pop_Reg(jit, AMX_REG_ALT);
dvander@157
   693
	IA32_Pop_Reg(jit, REG_ECX);
dvander@157
   694
dvander@157
   695
	/* pop the AMX stack.  do not check the margins.
dvander@157
   696
	 * Note that this is not a true macro - we don't bother to
dvander@157
   697
	 * set ALT here because nothing will be using it.
dvander@157
   698
	 */
dvander@157
   699
	//lea edi, [edi+ecx*4+4]
dvander@157
   700
	IA32_Lea_Reg_DispRegMultImm8(jit, AMX_REG_STK, AMX_REG_STK, REG_ECX, SCALE4, 4);
dvander@157
   701
dvander@157
   702
	//ret
dvander@157
   703
	IA32_Return(jit);
dvander@157
   704
}
dvander@157
   705
dvander@157
   706
void WriteOp_Tracker_Push_Reg(JitWriter *jit, uint8_t reg)
dvander@157
   707
{
dvander@157
   708
	CompData *data = (CompData *)jit->data;
dvander@157
   709
dvander@157
   710
	/* Save registers that may be damaged by the call */
dvander@157
   711
	//push eax
dvander@157
   712
	//push ecx
dvander@157
   713
	//push edi
dvander@157
   714
	//lea edi, [<reg>*4]		; we want the count in bytes not in cells
dvander@157
   715
	IA32_Push_Reg(jit, AMX_REG_PRI);
dvander@157
   716
	if (reg == REG_ECX)
dvander@157
   717
	{
dvander@157
   718
		IA32_Push_Reg(jit, AMX_REG_TMP);
dvander@157
   719
	}
dvander@157
   720
	IA32_Push_Reg(jit, AMX_REG_STK);
dvander@157
   721
	IA32_Lea_Reg_RegMultImm32(jit, REG_EDI, reg, SCALE4, 0);
dvander@157
   722
dvander@157
   723
	/* Get the context ptr, push it and call the check */
dvander@157
   724
	//mov eax, [esi+context]
dvander@157
   725
	//push eax
dvander@157
   726
	//call JIT_VerifyOrAllocateTracker
dvander@157
   727
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, AMX_REG_INFO, AMX_INFO_CONTEXT);
dvander@157
   728
	IA32_Push_Reg(jit, REG_EAX);
dvander@157
   729
	jitoffs_t call = IA32_Call_Imm32(jit, 0);
dvander@350
   730
	IA32_Write_Jump32_Abs(jit, call, (void *)JIT_VerifyOrAllocateTracker);
dvander@157
   731
dvander@157
   732
	/* Check for errors */
dvander@237
   733
	//cmp eax, 0
dvander@157
   734
	//jnz :error
faluco@238
   735
	IA32_Cmp_Rm_Imm8(jit, MOD_REG, REG_EAX, 0);
faluco@238
   736
	IA32_Jump_Cond_Imm32_Abs(jit, CC_NZ, data->jit_return);
dvander@157
   737
dvander@237
   738
	/* Restore */
dvander@237
   739
	//pop eax
dvander@237
   740
	IA32_Pop_Reg(jit, REG_EAX);
dvander@237
   741
	
dvander@157
   742
	/* Push the register into the stack and increment pCur */
dvander@157
   743
	//mov edx, [eax+vm[]]
dvander@157
   744
	//mov eax, [edx+pcur]
dvander@157
   745
	//add [edx+pcur], 4
dvander@157
   746
	//mov [eax], edi
dvander@157
   747
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EDX, REG_EAX, offsetof(sp_context_t, vm[JITVARS_TRACKER]));
dvander@157
   748
	IA32_Mov_Reg_Rm_Disp8(jit, REG_EAX, REG_EDX, offsetof(tracker_t, pCur));
dvander@157
   749
	IA32_Add_Rm_Imm8_Disp8(jit, REG_EDX, 4, offsetof(tracker_t, pCur));
dvander@157
   750
	IA32_Mov_Rm_Reg(jit, REG_EAX, REG_EDI, MOD_MEM_REG);
dvander@157
   751
dvander@157
   752
	/* Restore PRI, ALT and STK */
dvander@157
   753
	//pop edi
dvander@157
   754
	//pop ecx
dvander@157
   755
	//pop eax
dvander@157
   756
	IA32_Pop_Reg(jit, AMX_REG_STK);
dvander@157
   757
	if (reg == REG_ECX)
dvander@157
   758
	{
dvander@157
   759
		IA32_Pop_Reg(jit, AMX_REG_TMP);
dvander@157
   760
	}
dvander@157
   761
	IA32_Pop_Reg(jit, AMX_REG_PRI);
dvander@157
   762
}
dvander@157
   763
dvander@237
   764
int JIT_VerifyOrAllocateTracker(sp_context_t *ctx)
dvander@157
   765
{
dvander@157
   766
	tracker_t *trk = (tracker_t *)(ctx->vm[JITVARS_TRACKER]);
dvander@157
   767
dvander@157
   768
	if ((size_t)(trk->pCur - trk->pBase) >= trk->size)
dvander@157
   769
	{
dvander@237
   770
		return SP_ERROR_TRACKER_BOUNDS;
dvander@157
   771
	}
dvander@157
   772
dvander@157
   773
	if (trk->pCur+1 - (trk->pBase + trk->size) == 0)
dvander@157
   774
	{
dvander@157
   775
		size_t disp = trk->size - 1;
dvander@157
   776
		trk->size *= 2;
dvander@157
   777
		trk->pBase = (ucell_t *)realloc(trk->pBase, trk->size * sizeof(cell_t));
dvander@157
   778
dvander@157
   779
		if (!trk->pBase)
dvander@157
   780
		{
dvander@237
   781
			return SP_ERROR_TRACKER_BOUNDS;
dvander@157
   782
		}
dvander@157
   783
dvander@157
   784
		trk->pCur = trk->pBase + disp;
dvander@157
   785
	}
dvander@237
   786
dvander@237
   787
	return SP_ERROR_NONE;
dvander@157
   788
}
dvander@157
   789
dvander@237
   790
int JIT_VerifyLowBoundTracker(sp_context_t *ctx)
dvander@157
   791
{
dvander@157
   792
	tracker_t *trk = (tracker_t *)(ctx->vm[JITVARS_TRACKER]);
dvander@157
   793
dvander@157
   794
	if (trk->pCur <= trk->pBase)
dvander@157
   795
	{
dvander@237
   796
		return SP_ERROR_TRACKER_BOUNDS;
dvander@157
   797
	}
dvander@237
   798
dvander@237
   799
	return SP_ERROR_NONE;
dvander@237
   800
}