1 files changed, 9 insertions, 4 deletions
diff --git a/arm9/lib/include/OS_system.h b/arm9/lib/include/OS_system.h
index c7f121bd..20228cbf 100644
--- a/arm9/lib/include/OS_system.h
+++ b/arm9/lib/include/OS_system.h
@@ -17,17 +17,22 @@ typedef enum {
     OS_PROCMODE_SYS=31
 } OSProcMode;
 
+#pragma enumsalwaysint on
 typedef enum {
-    OS_INTRMODE_DISABLE = HW_PSR_IRQ_DISABLE,
+    OS_INTRMODE_DISABLE_IRQ = HW_PSR_DISABLE_IRQ,
+    OS_INTRMODE_DISABLE_FIQ = HW_PSR_DISABLE_FIQ,
     OS_INTRMODE_ENABLE = 0
 } OSIntrMode;
+#pragma enumsalwaysint reset
 
 OSIntrMode OS_EnableInterrupts();
 OSIntrMode OS_DisableInterrupts();
-OSIntrMode OS_RestoreInterrupts(OSIntrMode state);
+OSIntrMode OS_RestoreInterrupts(register OSIntrMode state);
 OSIntrMode OS_DisableInterrupts_IrqAndFiq();
-OSIntrMode OS_RestoreInterrupts_IrqAndFiq(OSIntrMode state);
+OSIntrMode OS_RestoreInterrupts_IrqAndFiq(register OSIntrMode state);
+OSIntrMode OS_GetCpsrIrq();
 OSProcMode OS_GetProcMode();
-
+void OS_SpinWait();
+void OS_WaitVBlankIntr();
 
 #endif //POKEDIAMOND_OS_SYSTEM_H