|
| 1 | +from collections import deque |
| 2 | + |
| 3 | +class Sa2SeedKey: |
| 4 | + register: int |
| 5 | + carry_flag: int = 0 |
| 6 | + instruction_tape: bytearray |
| 7 | + instruction_pointer: int = 0 |
| 8 | + for_pointers: deque = deque() |
| 9 | + for_iterations: deque = deque() |
| 10 | + |
| 11 | + def __init__(self, instruction_tape, seed): |
| 12 | + self.instruction_tape = instruction_tape |
| 13 | + self.register = seed |
| 14 | + |
| 15 | + def rsl(self): |
| 16 | + self.carry_flag = self.register & 0x80000000 |
| 17 | + self.register = self.register << 1 |
| 18 | + if(self.carry_flag): |
| 19 | + self.register = self.register | 0x1 |
| 20 | + self.register = self.register & 0xFFFFFFFF |
| 21 | + self.instruction_pointer += 1 |
| 22 | + |
| 23 | + def rsr(self): |
| 24 | + self.carry_flag = self.register & 0x1 |
| 25 | + self.register = self.register >> 1 |
| 26 | + if(self.carry_flag): |
| 27 | + self.register = self.register | 0x80000000 |
| 28 | + self.instruction_pointer += 1 |
| 29 | + |
| 30 | + def add(self): |
| 31 | + self.carry_flag = 0 |
| 32 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 5] |
| 33 | + add_int = operands[0] << 24 | operands[1] << 16 | operands[2] << 8 | operands[3] |
| 34 | + output_register = self.register + add_int |
| 35 | + if (output_register > 0xffffffff): |
| 36 | + self.carry_flag = 1 |
| 37 | + output_register = output_register & 0xffffffff |
| 38 | + self.register = output_register |
| 39 | + self.instruction_pointer += 5 |
| 40 | + |
| 41 | + def sub(self): |
| 42 | + self.carry_flag = 0 |
| 43 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 5] |
| 44 | + sub_int = operands[0] << 24 | operands[1] << 16 | operands[2] << 8 | operands[3] |
| 45 | + output_register = self.register - sub_int |
| 46 | + if (output_register < 0): |
| 47 | + self.carry_flag = 1 |
| 48 | + output_register = output_register & 0xffffffff |
| 49 | + self.register = output_register |
| 50 | + self.instruction_pointer += 5 |
| 51 | + |
| 52 | + def eor(self): |
| 53 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 5] |
| 54 | + xor_int = operands[0] << 24 | operands[1] << 16 | operands[2] << 8 | operands[3] |
| 55 | + self.register = self.register ^ xor_int |
| 56 | + self.instruction_pointer += 5 |
| 57 | + |
| 58 | + def for_loop(self): |
| 59 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 2] |
| 60 | + self.for_iterations.appendleft(operands[0] - 1) |
| 61 | + self.instruction_pointer += 2 |
| 62 | + self.for_pointers.appendleft(self.instruction_pointer) |
| 63 | + |
| 64 | + def next_loop(self): |
| 65 | + if(self.for_iterations[0] > 0): |
| 66 | + self.for_iterations[0] -= 1 |
| 67 | + self.instruction_pointer = self.for_pointers[0] |
| 68 | + else: |
| 69 | + self.for_iterations.popleft() |
| 70 | + self.for_pointers.popleft() |
| 71 | + self.instruction_pointer += 1 |
| 72 | + |
| 73 | + # bcc = branch conditional |
| 74 | + def bcc(self): |
| 75 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 2] |
| 76 | + skip_count = operands[0] + 2 |
| 77 | + if(self.carry_flag == 0): |
| 78 | + self.instruction_pointer += skip_count |
| 79 | + else: |
| 80 | + self.instruction_pointer += 2 |
| 81 | + |
| 82 | + # bra = branch unconditional |
| 83 | + def bra(self): |
| 84 | + operands = self.instruction_tape[self.instruction_pointer + 1:self.instruction_pointer + 2] |
| 85 | + skip_count = operands[0] + 2 |
| 86 | + self.instruction_pointer += skip_count |
| 87 | + |
| 88 | + def finish(self): |
| 89 | + self.instruction_pointer += 1 |
| 90 | + |
| 91 | + def execute(self): |
| 92 | + instruction_set = { |
| 93 | + 0x81 : self.rsl, |
| 94 | + 0x82 : self.rsr, |
| 95 | + 0x93 : self.add, |
| 96 | + 0x84 : self.sub, |
| 97 | + 0x87 : self.eor, |
| 98 | + 0x68 : self.for_loop, |
| 99 | + 0x49 : self.next_loop, |
| 100 | + 0x4A : self.bcc, |
| 101 | + 0x6B : self.bra, |
| 102 | + 0x4C : self.finish |
| 103 | + } |
| 104 | + while(self.instruction_pointer < len(self.instruction_tape)): |
| 105 | + instruction_set[self.instruction_tape[self.instruction_pointer]]() |
| 106 | + return self.register |
0 commit comments