From 64a17f3e0683a0a492a3fed9c4c17a4335d1f421 Mon Sep 17 00:00:00 2001
From: Loic Guegan <manzerbredes@mailbox.org>
Date: Sun, 2 May 2021 19:10:15 +0200
Subject: Simplify ds segment register management

---
 src/core/int.S | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/core/int.S')

diff --git a/src/core/int.S b/src/core/int.S
index 2897698..c8612dc 100644
--- a/src/core/int.S
+++ b/src/core/int.S
@@ -26,6 +26,9 @@
     push %rbp
     push %rsi
     push %rdi
+    mov %ds, %rax
+    push %rax
+    mov 56(%rsp), %rax # Restore %rax
     push %rax
     mov $0x10, %ax
     mov %ax, %ds
@@ -33,6 +36,8 @@
 .endm
 
 .macro RESTORE_REGS
+    pop %rax
+    mov %rax,%ds
     pop %rdi    
     pop %rsi    
     pop %rbp    
-- 
cgit v1.2.3