[llvm] 72abc9b - [dfsan] add a missing zero origin at atomic commands

[Jianzhou Zhao via llvm-commits]

Author: Jianzhou Zhao
Date: 2021-03-04T16:50:05Z
New Revision: 72abc9bf07a9e1a67671edf49e0f69734ad08151

URL: https://github.com/llvm/llvm-project/commit/72abc9bf07a9e1a67671edf49e0f69734ad08151
DIFF: https://github.com/llvm/llvm-project/commit/72abc9bf07a9e1a67671edf49e0f69734ad08151.diff

LOG: [dfsan] add a missing zero origin at atomic commands

Added: 
    

Modified: 
    llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp b/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp
index 21a7163efdb5..ab057bb9b85a 100644
--- a/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp
+++ b/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp
@@ -2346,6 +2346,7 @@ void DFSanVisitor::visitCASOrRMW(Align InstAlignment, Instruction &I) {
   const Align ShadowAlign = DFSF.getShadowAlign(InstAlignment);
   DFSF.storeZeroPrimitiveShadow(Addr, Size, ShadowAlign, &I);
   DFSF.setShadow(&I, DFSF.DFS.getZeroShadow(&I));
+  DFSF.setOrigin(&I, DFSF.DFS.ZeroOrigin);
 }
 
 void DFSanVisitor::visitAtomicRMWInst(AtomicRMWInst &I) {