From 1dcce2148d855af67f845319414a94db5601be3e Mon Sep 17 00:00:00 2001
From: Anton Korobeynikov <asl@math.spbu.ru>
Date: Sat, 22 Mar 2008 21:04:01 +0000
Subject: [PATCH] Add information about callee-saved registers on Win64

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@48692 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Target/X86/X86RegisterInfo.cpp | 30 ++++++++++++++++++++++++------
 lib/Target/X86/X86RegisterInfo.h   |  4 ++++
 2 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/lib/Target/X86/X86RegisterInfo.cpp b/lib/Target/X86/X86RegisterInfo.cpp
index f905ffa4523..234c24b6afc 100644
--- a/lib/Target/X86/X86RegisterInfo.cpp
+++ b/lib/Target/X86/X86RegisterInfo.cpp
@@ -45,6 +45,7 @@ X86RegisterInfo::X86RegisterInfo(X86TargetMachine &tm,
   // Cache some information.
   const X86Subtarget *Subtarget = &TM.getSubtarget<X86Subtarget>();
   Is64Bit = Subtarget->is64Bit();
+  IsWin64 = Subtarget->isTargetWin64();
   StackAlign = TM.getFrameInfo()->getStackAlignment();
   if (Is64Bit) {
     SlotSize = 8;
@@ -196,9 +197,17 @@ X86RegisterInfo::getCalleeSavedRegs(const MachineFunction *MF) const {
     X86::RBX, X86::R12, X86::R13, X86::R14, X86::R15, X86::RBP, 0
   };
 
-  if (Is64Bit)
-    return CalleeSavedRegs64Bit;
-  else {
+  static const unsigned CalleeSavedRegsWin64[] = {
+    X86::RBX, X86::RBP, X86::RDI, X86::RSI,
+    X86::R12, X86::R13, X86::R14, X86::R15, 0
+  };
+
+  if (Is64Bit) {
+    if (IsWin64)
+      return CalleeSavedRegsWin64;
+    else
+      return CalleeSavedRegs64Bit;
+  } else {
     if (MF) {
         MachineFrameInfo *MFI = MF->getFrameInfo();
         MachineModuleInfo *MMI = MFI->getMachineModuleInfo();
@@ -225,10 +234,19 @@ X86RegisterInfo::getCalleeSavedRegClasses(const MachineFunction *MF) const {
     &X86::GR64RegClass, &X86::GR64RegClass,
     &X86::GR64RegClass, &X86::GR64RegClass, 0
   };
+  static const TargetRegisterClass * const CalleeSavedRegClassesWin64[] = {
+    &X86::GR64RegClass, &X86::GR64RegClass,
+    &X86::GR64RegClass, &X86::GR64RegClass,
+    &X86::GR64RegClass, &X86::GR64RegClass,
+    &X86::GR64RegClass, &X86::GR64RegClass, 0
+  };
 
-  if (Is64Bit)
-    return CalleeSavedRegClasses64Bit;
-  else {
+  if (Is64Bit) {
+    if (IsWin64)
+      return CalleeSavedRegClassesWin64;
+    else
+      return CalleeSavedRegClasses64Bit;
+  } else {
     if (MF) {
         MachineFrameInfo *MFI = MF->getFrameInfo();
         MachineModuleInfo *MMI = MFI->getMachineModuleInfo();
diff --git a/lib/Target/X86/X86RegisterInfo.h b/lib/Target/X86/X86RegisterInfo.h
index 1c8f5e28b99..cf6e703d6bf 100644
--- a/lib/Target/X86/X86RegisterInfo.h
+++ b/lib/Target/X86/X86RegisterInfo.h
@@ -59,6 +59,10 @@ private:
   ///
   bool Is64Bit;
 
+  /// IsWin64 - Is the target on of win64 flavours
+  ///
+  bool IsWin64;
+
   /// SlotSize - Stack slot size in bytes.
   ///
   unsigned SlotSize;