Implement readquad, writequad, fetchins and crack.

Signed-off-by: jmug <u.g.a.mariano@gmail.com>
This commit is contained in:
Mariano Uvalle 2025-01-31 22:22:53 -08:00
parent 55089bde91
commit 4c0d93398f
2 changed files with 44 additions and 6 deletions

View file

@ -210,13 +210,20 @@ let map_addr_segfault (addr : quad) : int =
are glued together.
*)
let readquad (m : mach) (addr : quad) : quad = failwith "readquad not implemented"
let readquad (m : mach) (addr : quad) : quad =
int64_of_sbytes @@ Array.to_list @@ Array.sub m.mem (map_addr_segfault addr) 8
let writequad (m : mach) (addr : quad) (w : quad) : unit =
failwith "writequad not implemented"
;;
Array.blit (Array.of_list @@ sbytes_of_int64 w) 0 m.mem (map_addr_segfault addr) 8
exception InvalidInstructinAlignment
let fetchins (m : mach) (addr : quad) : ins =
let ins_byte = m.mem.(map_addr_segfault addr) in
match ins_byte with
| InsB0 i -> i
| _ -> raise InvalidInstructinAlignment
let fetchins (m : mach) (addr : quad) : ins = failwith "fetchins not implemented"
(* Compute the instruction result.
* NOTE: See int64_overflow.ml for the definition of the return type
@ -239,11 +246,16 @@ let interp_operands (m : mach) : ins -> int64 list =
;;
let validate_operands : ins -> unit = function
| _ -> failwith "validate_operands not implemented"
(* TODO: Add validations, leaq can only take indirect operands for example *)
| _ -> ()
;;
let crack : ins -> ins list = function
| _ -> failwith "crack not implemented"
| Pushq, [ op ] -> [Subq, [Imm (Lit 8L); Reg Rsp]; Movq, [op; Ind2 Rsp]]
| Popq, [ op ] -> [Movq, [Ind2 Rsp; op]; Addq, [Imm (Lit 8L); Reg Rsp]]
| Callq, [ op ] -> [Subq, [Imm (Lit 8L); Reg Rsp]; Movq, [Reg Rip; Ind2 Rsp]; Jmp, [op]]
| Retq, [] -> [Movq, [Ind2 Rsp; Reg Rip]; Addq, [Imm (Lit 8L); Reg Rsp]]
| i -> [ i ]
;;
(* TODO: double check against spec *)

View file

@ -44,12 +44,38 @@ let mov_ri =
InsFrag;
]
let crack_tests =
let open Asm in
let pu = (Pushq, [~$42]) in
let po = (Popq, [~%Rax]) in
let ca = (Callq, [Ind2 Rax]) in
let re = (Retq, []) in
[
("crack_pushq", assert_eqf (fun() ->
crack pu) ([ Subq, [Imm (Lit 8L); Reg Rsp]
; Movq, [~$42; Ind2 Rsp] ]));
("crack_pop", assert_eqf (fun() ->
crack po) ([ Movq, [Ind2 Rsp; Reg Rax]
; Addq, [Imm (Lit 8L); Reg Rsp] ]));
("crack_call", assert_eqf (fun() ->
crack ca) ([ Subq, [Imm (Lit 8L); Reg Rsp]
; Movq, [Reg Rip; Ind2 Rsp]
; Jmp, [Ind2 Rax] ]));
("crack retq", assert_eqf (fun() ->
crack re) ([ Movq, [Ind2 Rsp; Reg Rip]
; Addq, [Imm (Lit 8L); Reg Rsp] ]));
]
let provided_tests : suite = [
Test ("My Tests", [
("assert", test_my)
]);
Test ("Student provided crack tests", crack_tests);
Test ("Student-Provided Big Test for Part III: Score recorded as PartIIITestCase", [
]);