Enable -fcas-emit-casid-file to llvm backend.

This change enables -fcas-emit-casid-file to the llvm backend and allows
emission of a .casid file when -fcas-backend and
-fcas-backend-mode=native or -fcas-backend-mode=verify is used

Author: rastogishubham

clang/test/CAS/cas-emit-casid.c

@@ -0,0 +1,31 @@
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=native -Xclang -fcas-emit-casid-file %s -o %t/test.o 
+// RUN: cat %t/test.o.casid | FileCheck %s --check-prefix=NATIVE_FILENAME
+// NATIVE_FILENAME: CASID:Jllvmcas://{{.*}}
+//
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=verify -Xclang -fcas-emit-casid-file %s -o %t/test.o 
+// RUN: cat %t/test.o.casid | FileCheck %s --check-prefix=VERIFY_FILENAME
+// VERIFY_FILENAME: CASID:Jllvmcas://{{.*}}
+//
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=casid -Xclang -fcas-emit-casid-file %s -o %t/test.o 
+// RUN: not cat %t/test.o.casid
+//
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=native -Xclang -fcas-emit-casid-file %s -o -
+// RUN: not cat %t/test.o.casid
+//
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=verify -Xclang -fcas-emit-casid-file %s -o -
+// RUN: not cat %t/test.o.casid
+//
+// RUN: rm -rf %t && mkdir -p %t
+// RUN: %clang -c -Xclang -fcas-backend -Xclang -fcas-path -Xclang %t/cas -Xclang -fcas-backend-mode=casid -Xclang -fcas-emit-casid-file %s -o -
+// RUN: not cat %t/test.o.casid
+
+void test(void) {}
+
+int test1(void) {
+  return 0;
+}

llvm/include/llvm/MC/MCAsmBackend.h

@@ -90,7 +90,8 @@ class MCAsmBackend {
           const llvm::MCAsmLayout &, cas::ObjectStore &, raw_ostream *)>
           CreateFromMcAssembler,
       std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
-          SerializeObjectFile) const;
+          SerializeObjectFile,
+      raw_pwrite_stream *CasIDOS = nullptr) const;
   // END MCCAS
 
 

llvm/include/llvm/MC/MCMachOCASWriter.h

@@ -52,7 +52,8 @@ class MachOCASWriter : public MCObjectWriter {
           CreateFromMcAssembler,
       std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
           SerializeObjectFile,
-      std::optional<MCTargetOptions::ResultCallBackTy> CallBack);
+      std::optional<MCTargetOptions::ResultCallBackTy> CallBack,
+      raw_pwrite_stream *CasIDOS = nullptr);
 
   void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout,
                         const MCFragment *Fragment, const MCFixup &Fixup,
@@ -122,6 +123,7 @@ class MachOCASWriter : public MCObjectWriter {
 
 private:
   raw_pwrite_stream &OS;
+  raw_pwrite_stream *CasIDOS;
 
   SmallString<512> InternalBuffer;
   raw_svector_ostream InternalOS;
@@ -157,7 +159,8 @@ std::unique_ptr<MCObjectWriter> createMachOCASWriter(
         CreateFromMcAssembler,
     std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
         SerializeObjectFile,
-    std::optional<MCTargetOptions::ResultCallBackTy> CallBack = std::nullopt);
+    std::optional<MCTargetOptions::ResultCallBackTy> CallBack = std::nullopt,
+    raw_pwrite_stream *CasIDOS = nullptr);
 
 } // end namespace llvm
 

llvm/include/llvm/Target/TargetMachine.h

@@ -377,7 +377,8 @@ class TargetMachine {
   addPassesToEmitFile(PassManagerBase &, raw_pwrite_stream &,
                       raw_pwrite_stream *, CodeGenFileType,
                       bool /*DisableVerify*/ = true,
-                      MachineModuleInfoWrapperPass *MMIWP = nullptr) {
+                      MachineModuleInfoWrapperPass *MMIWP = nullptr,
+                      raw_pwrite_stream *CasIDOS = nullptr) {
     return true;
   }
 
@@ -443,11 +444,11 @@ class LLVMTargetMachine : public TargetMachine {
   /// emitted.  Typically this will involve several steps of code generation.
   /// \p MMIWP is an optional parameter that, if set to non-nullptr,
   /// will be used to set the MachineModuloInfo for this PM.
-  bool
-  addPassesToEmitFile(PassManagerBase &PM, raw_pwrite_stream &Out,
-                      raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
-                      bool DisableVerify = true,
-                      MachineModuleInfoWrapperPass *MMIWP = nullptr) override;
+  bool addPassesToEmitFile(PassManagerBase &PM, raw_pwrite_stream &Out,
+                           raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
+                           bool DisableVerify = true,
+                           MachineModuleInfoWrapperPass *MMIWP = nullptr,
+                           raw_pwrite_stream *CasIDOS = nullptr) override;
 
   virtual Error buildCodeGenPipeline(ModulePassManager &,
                                      MachineFunctionPassManager &,
@@ -482,11 +483,12 @@ class LLVMTargetMachine : public TargetMachine {
   /// machine code from the MI representation.
   bool addAsmPrinter(PassManagerBase &PM, raw_pwrite_stream &Out,
                      raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
-                     MCContext &Context);
+                     MCContext &Context, raw_pwrite_stream *CasIDOS = nullptr);
 
   Expected<std::unique_ptr<MCStreamer>>
   createMCStreamer(raw_pwrite_stream &Out, raw_pwrite_stream *DwoOut,
-                   CodeGenFileType FileType, MCContext &Ctx);
+                   CodeGenFileType FileType, MCContext &Ctx,
+                   raw_pwrite_stream *CasIDOS = nullptr);
 
   /// True if the target uses physical regs (as nearly all targets do). False
   /// for stack machines such as WebAssembly and other virtual-register

llvm/lib/CodeGen/LLVMTargetMachine.cpp

@@ -124,13 +124,11 @@ addPassesToGenerateCode(LLVMTargetMachine &TM, PassManagerBase &PM,
   return PassConfig;
 }
 
-bool LLVMTargetMachine::addAsmPrinter(PassManagerBase &PM,
-                                      raw_pwrite_stream &Out,
-                                      raw_pwrite_stream *DwoOut,
-                                      CodeGenFileType FileType,
-                                      MCContext &Context) {
+bool LLVMTargetMachine::addAsmPrinter(
+    PassManagerBase &PM, raw_pwrite_stream &Out, raw_pwrite_stream *DwoOut,
+    CodeGenFileType FileType, MCContext &Context, raw_pwrite_stream *CasIDOS) {
   Expected<std::unique_ptr<MCStreamer>> MCStreamerOrErr =
-      createMCStreamer(Out, DwoOut, FileType, Context);
+      createMCStreamer(Out, DwoOut, FileType, Context, CasIDOS);
   if (auto Err = MCStreamerOrErr.takeError())
     return true;
 
@@ -146,7 +144,7 @@ bool LLVMTargetMachine::addAsmPrinter(PassManagerBase &PM,
 
 Expected<std::unique_ptr<MCStreamer>> LLVMTargetMachine::createMCStreamer(
     raw_pwrite_stream &Out, raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
-    MCContext &Context) {
+    MCContext &Context, raw_pwrite_stream *CasIDOS) {
   if (Options.MCOptions.MCSaveTempLabels)
     Context.setAllowTemporaryLabels(false);
 
@@ -228,7 +226,7 @@ Expected<std::unique_ptr<MCStreamer>> LLVMTargetMachine::createMCStreamer(
       CASBackendWriter = MAB->createCASObjectWriter(
           Out, getTargetTriple(), *Options.MCOptions.CAS, Options.MCOptions,
           Options.MCOptions.CASObjMode, CreateFromMcAssembler,
-          SerializeObjectFile);
+          SerializeObjectFile, CasIDOS);
     }
     // END MCCAS
     AsmStreamer.reset(getTarget().createMCObjectStreamer(
@@ -254,7 +252,7 @@ Expected<std::unique_ptr<MCStreamer>> LLVMTargetMachine::createMCStreamer(
 bool LLVMTargetMachine::addPassesToEmitFile(
     PassManagerBase &PM, raw_pwrite_stream &Out, raw_pwrite_stream *DwoOut,
     CodeGenFileType FileType, bool DisableVerify,
-    MachineModuleInfoWrapperPass *MMIWP) {
+    MachineModuleInfoWrapperPass *MMIWP, raw_pwrite_stream *CasIDOS) {
   // Add common CodeGen passes.
   if (!MMIWP)
     MMIWP = new MachineModuleInfoWrapperPass(this);
@@ -264,7 +262,8 @@ bool LLVMTargetMachine::addPassesToEmitFile(
     return true;
 
   if (TargetPassConfig::willCompleteCodeGenPipeline()) {
-    if (addAsmPrinter(PM, Out, DwoOut, FileType, MMIWP->getMMI().getContext()))
+    if (addAsmPrinter(PM, Out, DwoOut, FileType, MMIWP->getMMI().getContext(),
+                      CasIDOS))
       return true;
   } else {
     // MIR printing is redundant with -filetype=null.

llvm/lib/MC/MCAsmBackend.cpp

@@ -69,14 +69,15 @@ std::unique_ptr<MCObjectWriter> MCAsmBackend::createCASObjectWriter(
         cas::ObjectStore &, raw_ostream *)>
         CreateFromMcAssembler,
     std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
-        SerializeObjectFile) const {
+        SerializeObjectFile,
+    raw_pwrite_stream *CasIDOS) const {
   auto TW = createObjectTargetWriter();
   switch (TW->getFormat()) {
   case Triple::MachO:
     return createMachOCASWriter(cast<MCMachObjectTargetWriter>(std::move(TW)),
                                 TT, CAS, Mode, OS, Endian == support::little,
                                 CreateFromMcAssembler, SerializeObjectFile,
-                                MCOpts.ResultCallBack);
+                                MCOpts.ResultCallBack, CasIDOS);
   default:
     llvm_unreachable("unexpected object format");
   }

llvm/lib/MC/MachOCASWriter.cpp

@@ -52,9 +52,10 @@ MachOCASWriter::MachOCASWriter(
         CreateFromMcAssembler,
     std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
         SerializeObjectFile,
-    std::optional<MCTargetOptions::ResultCallBackTy> ResultCallBack)
+    std::optional<MCTargetOptions::ResultCallBackTy> ResultCallBack,
+    raw_pwrite_stream *CasIDOS)
     : Target(TT), CAS(CAS), Mode(Mode), ResultCallBack(ResultCallBack), OS(OS),
-      InternalOS(InternalBuffer),
+      CasIDOS(CasIDOS), InternalOS(InternalBuffer),
       MOW(std::move(MOTW), InternalOS, IsLittleEndian),
       CreateFromMcAssembler(CreateFromMcAssembler),
       SerializeObjectFile(SerializeObjectFile) {
@@ -114,6 +115,9 @@ uint64_t MachOCASWriter::writeObject(MCAssembler &Asm,
   }
   }
 
+  if (CasIDOS)
+    writeCASIDBuffer(CASObj.getID(), *CasIDOS);
+
   return OS.tell() - StartOffset;
 }
 
@@ -127,8 +131,9 @@ std::unique_ptr<MCObjectWriter> llvm::createMachOCASWriter(
         CreateFromMcAssembler,
     std::function<Error(cas::ObjectProxy, cas::ObjectStore &, raw_ostream &)>
         SerializeObjectFile,
-    std::optional<MCTargetOptions::ResultCallBackTy> ResultCallBack) {
-  return std::make_unique<MachOCASWriter>(std::move(MOTW), TT, CAS, Mode, OS,
-                                          IsLittleEndian, CreateFromMcAssembler,
-                                          SerializeObjectFile, ResultCallBack);
+    std::optional<MCTargetOptions::ResultCallBackTy> ResultCallBack,
+    raw_pwrite_stream *CasIDOS) {
+  return std::make_unique<MachOCASWriter>(
+      std::move(MOTW), TT, CAS, Mode, OS, IsLittleEndian, CreateFromMcAssembler,
+      SerializeObjectFile, ResultCallBack, CasIDOS);
 }