fmt & refactor

This commit is contained in:
Mylloon 2024-01-02 02:06:59 +01:00
parent f86829dd3a
commit d6e7a526ee
Signed by: Anri
GPG key ID: A82D63DFF8D1317F

View file

@ -479,8 +479,10 @@ module InstructionSelector : InstructionSelector = struct
let div ~dst ~srcl ~srcr =
(* Division *)
let rax = `Reg X86_64_Architecture.RAX in
(* Dividende dans RAX et diviseur dans R15 *)
mov ~src:srcl ~dst:rax
@ mov ~src:srcr ~dst:r15
(* Division avec signe *)
@ [ Instruction cqto; Instruction (idivq ~src:r15) ]
@ mov ~dst ~src:rax
;;
@ -531,67 +533,83 @@ module FrameManager (IS : InstructionSelector) : FrameManager = struct
space after the function prologue. *)
let stack_usage_after_prologue fd =
Mint.size_in_bytes
+ ((if empty_frame fd then 0 else 1) * Mint.size_in_bytes)
(* Cas où il n'y a 0 arguments et variable locales *)
+ (if empty_frame fd then 0 else Mint.size_in_bytes)
(* Taille variables locales *)
+ fd.locals_space
;;
(* Ajoute les éléments de liste sur la pile *)
(** Ajoute les éléments de liste sur la pile *)
let add_stack stack list start op_addr =
fst
(List.fold_left
(fun (stack, addr) id -> S.IdMap.add id (Mint.of_int addr) stack, op_addr addr)
(stack, start)
list)
;;
let frame_descriptor ~params ~locals =
(*Ajout des variables locals d'abord. ON commence à la position -8 dans la pile vide, on décroit l'adresse de 8 à chaque nouvelle valeur*)
(* Ajout des variables locals d'abord.
On commence à la position -8 dans la pile vide,
on décroit l'adresse de 8 à chaque nouvelle valeur *)
let stack = add_stack S.IdMap.empty locals (-8) (fun x -> x - 8) in
(* Ajout des variables global ensuite. ON reprend la même pile, on commence à l'adresse 16 et on ajoute l'adresse qui suit de 8 *)
(* Ajout des variables global ensuite.
On reprend la même pile, on commence à l'adresse 16
et on ajoute l'adresse qui suit de 8 *)
let stack = add_stack stack params 16 (fun x -> x + 8) in
(* ON créé le frame descriptor ensuite*)
{param_count = List.length params; locals_space = (List.length locals) * 8; stack_map = stack; }
(* On créé le frame descriptor ensuite *)
{ param_count = List.length params
; locals_space = List.length locals * 8
; stack_map = stack
}
;;
let location_of fd id : T.address =
match S.IdMap.find_opt id fd.stack_map with
(* Si on retrouve id dans la stack map, alors c'est une variable locale *)
| Some o -> {offset = Some (Lit o); base = Some (RBP); idx = None; scale = `One;}
| Some o -> { offset = Some (Lit o); base = Some RBP; idx = None; scale = `One }
(* Sinon, variable globale *)
| None -> {offset = Some (Lab (data_label_of_global id)); base = Some (RIP); idx = None; scale = `One;}
| None ->
{ offset = Some (Lab (data_label_of_global id))
; base = Some RIP
; idx = None
; scale = `One
}
;;
open T (* Necessaire pour pouvoir créer des instructions*)
let function_prologue fd =
(*Instruction 1 : On pousse rbp sur la pile, pour sauvegarder la valeur actuelle de la base de la pile, quand on dépilera à la fin*)
(*Instruction 2 : On initialise la base de la pile en mettant rsp au niveau de rbp*)
[Instruction (pushq ~src:rbp);Instruction (movq ~src:rsp ~dst:rbp)]
(*On ajoute de l'espace pour les futurs variables locales. On regarde combien il y en a dans le frame descriptor*)
@ if fd.locals_space <> 0 then
(*On va faire des subq (descendre la pile vers le bas) pour pouvoir y mettre les variables locales.*)
[Instruction (subq ~src:(liti fd.locals_space) ~dst:rsp)]
(* On pousse rbp sur la pile,
pour sauvegarder la valeur actuelle de la base de la pile,
permet de revenir on en était quand on dépilera à la fin *)
[ T.Instruction (T.pushq ~src:rbp)
; (* On initialise la base de la pile en mettant rsp au niveau de rbp *)
T.Instruction (T.movq ~src:rsp ~dst:rbp)
]
@
(* On ajoute de l'espace pour les futurs variables locales.
On regarde combien il y en a dans le frame descriptor *)
if fd.locals_space <> 0
then
(* On va faire des subq (descendre la pile vers le bas)
pour pouvoir y mettre les variables locales. *)
[ T.Instruction (T.subq ~src:(T.liti fd.locals_space) ~dst:rsp) ]
else []
;;
let function_epilogue fd =
let remontage =
(*Si de l'espace a été alloué pour les variables locales, on ajoute des instructiosn pour remonter la pile (vers le haut) avec addq*)
if fd.locals_space <> 0 then
[Instruction (addq ~src:(liti fd.locals_space) ~dst:rsp)]
else []
in
remontage @ [Instruction (popq ~dst:rbp)] (*Pour restaurer l'état de la pile, on effectue un popq ("l'inverse" de pushq du prologue)*)
(* Si de l'espace a été alloué pour les variables locales,
on ajoute des instructions pour remonter la pile (vers le haut)
avec addq ("l'inverse" du subq du prologue) *)
(if fd.locals_space <> 0
then [ T.Instruction (T.addq ~src:(T.liti fd.locals_space) ~dst:rsp) ]
else [])
@ (* Pour restaurer l'état de la pile,
on effectue un popq ("l'inverse" de pushq du prologue) *)
[ T.Instruction (T.popq ~dst:rbp) ]
;;
let call fd ~kind ~f ~args =
(* TODO *)
(* TODO : Il faut empiler les arguments ? *)
(* Appelle la fonction *)
[ T.Instruction (T.calldi ~tgt:f) ]
;;