1 files changed, 24 insertions, 13 deletions

diff --git a/sux.h b/sux.h
index 4ba4862..d267755 100644
--- a/sux.h
+++ b/sux.h
@@ -533,14 +533,20 @@ static /*inline*/ uint64_t get_ortho_addr(struct sux *cpu, uint8_t prefix, uint6
 				}
 				op[i].scale = (inst_size == 2) ? tmp.u8[0] : 0;
 			}
-			if (addr_size != 0xFF || (rs && op[i].id == MEM_IMM)) {
-				inst_size = (addr_size != 0xFF) ? addr_size+1 : rs;
+			if (addr_size != 0xFF) {
+				inst_size = addr_size+1;
 				op[i].value = read_value(cpu, 0, address, inst_size-1, inc_clk, 0);
 				value[i] = op[i].value;
 				if (inc_pc) {
 					address += inst_size;
 				}
 			}
+			if (rs && op[i].id == MEM_IMM) {
+				value[i] = address;
+				if (inc_pc) {
+					address += rs;
+				}
+			}
 			if (is_rind) {
 				for (int j = 0; j < 2 && op[i].rind[j] != 0xFF; j++) {
 					uint64_t reg;
@@ -900,7 +906,7 @@ static /*inline*/ void exec_ortho_inst(struct sux *cpu, uint8_t opcode, uint8_t
 	operand op[2];
 	cpu->pc = get_ortho_addr(cpu, prefix, cpu->pc, op, address, op_type, op_id, 1, 1, thread);
 	if (op[1].type) {
-		src = read_value(cpu, 0, address[1], size, 1, 1);
+		src = read_value(cpu, 0, (op[1].id != MEM_IMM) ? op[1].value : address[1], size, 1, 1);
 	} else {
 		switch (op[1].id) {
 			case REG_A  : src = cpu->a;	break;
@@ -922,7 +928,9 @@ static /*inline*/ void exec_ortho_inst(struct sux *cpu, uint8_t opcode, uint8_t
 		}
 	}
 	if (op[0].type) {
-		dst = read_value(cpu, 0, address[0], size, 1, 1);
+		if (op[0].id != MEM_IMM) {
+			dst = read_value(cpu, 0, op[0].value, size, 1, 1);
+		}
 	} else {
 		switch (op[0].id) {
 			case REG_A  : dst = cpu->a;	break;
@@ -1005,23 +1013,23 @@ static /*inline*/ void exec_ortho_inst(struct sux *cpu, uint8_t opcode, uint8_t
 		case ORTHO_2OP(IDV): dst = idiv(cpu, dst, src, &rem, (op[0].type) ? size+1 : 8, thread); isdiv = 1; break;
 		case ORTHO_2OP(LEA):
 			do {
-				uint64_t addr;
+				uint64_t address;
 				uint64_t mask;
 				if (op[1].type) {
 					uint8_t addr_size = get_ortho_addrsize(prefix, op[1].id);
 					size = (!size) ? addr_size : size;
-					addr = address[1];
+					address = op[1].value;
 				} else {
-					addr = src;
+					address = src;
 				}
 				mask = (-(uint64_t)1 >> ((7 - size) * 8));
-				dst = (addr & mask);
+				dst = (address & mask);
 			} while (0);
 			break;
 		case ORTHO_1OP(PEA):
 			do {
-				uint64_t addr = (op[0].type) ? address[0] : dst;
-				push(cpu, addr, 7, thread);
+				uint64_t address = (op[0].type) ? op[0].value : dst;
+				push(cpu, address, 7, thread);
 			} while (0);
 			break;
 		case ORTHO_1OP(INC): dst = inc_dec(cpu, dst, size, 1, thread); break;
@@ -1043,9 +1051,10 @@ static /*inline*/ void exec_ortho_inst(struct sux *cpu, uint8_t opcode, uint8_t
 		case ORTHO_1CC(SET, VC): dst = set(cpu, !getflag(V), thread); break;
 
 	}
-
 	if (op[0].type) {
-		write_value(cpu, dst, address[0], size, 1, 1);
+		if (op[0].id != MEM_IMM) {
+			write_value(cpu, dst, op[0].value, size, 1, 1);
+		}
 	} else {
 		switch (op[0].id) {
 			case REG_A  : cpu->a	= dst;	break;
@@ -1068,7 +1077,9 @@ static /*inline*/ void exec_ortho_inst(struct sux *cpu, uint8_t opcode, uint8_t
 	}
 	if (isdiv) {
 		if (op[1].type) {
-			write_value(cpu, rem, address[1], size, 1, 1);
+			if (op[0].id != MEM_IMM) {
+				write_value(cpu, dst, op[1].value, size, 1, 1);
+			}
 		} else {
 			switch (op[1].id) {
 				case REG_A  : cpu->a	= rem;	break;