[CodeGen] Match LLVM machine debug stripping

1 files changed, 34 insertions, 0 deletions

diff --git a/backend/lib/CodeGen/MachineStripDebug.cpp b/backend/lib/CodeGen/MachineStripDebug.cpp
new file mode 100644
index 0000000..4699ef0
--- /dev/null
+++ b/backend/lib/CodeGen/MachineStripDebug.cpp
@@ -0,0 +1,34 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "rcc/CodeGen/MachineStripDebug.h"
+
+#include "rcc/CodeGen/MachineBasicBlock.h"
+#include "rcc/CodeGen/MachineFunction.h"
+#include "rcc/CodeGen/MachineInstr.h"
+
+namespace rcc {
+
+Expected<bool> stripDebugMachineFunction(MachineFunction &Function) {
+  bool Changed = false;
+  for (MachineBasicBlock *Block = Function.firstBlock(); Block != nullptr; Block = Block->next()) {
+    for (MachineInstr *Instruction = Block->firstInstruction(); Instruction != nullptr;) {
+      MachineInstr *Next = Instruction->next();
+      if (Instruction->isDebugInstr() && Instruction->operandCount() > 1) {
+        auto Removed = Block->removeInstruction(*Instruction);
+        if (!Removed)
+          return std::unexpected(Removed.error());
+        Changed = true;
+        Instruction = Next;
+        continue;
+      }
+      if (Instruction->debugLocation()) {
+        Instruction->clearDebugLocation();
+        Changed = true;
+      }
+      Instruction = Next;
+    }
+  }
+  return Changed;
+}
+
+} // namespace rcc