From 72abc9bf07a9e1a67671edf49e0f69734ad08151 Mon Sep 17 00:00:00 2001
From: Jianzhou Zhao <jianzhouzh@google.com>
Date: Thu, 4 Mar 2021 16:48:06 +0000
Subject: [PATCH] [dfsan] add a missing zero origin at atomic commands

---
 llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp b/llvm/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp
index 21a7163efdb55..ab057bb9b85ad 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) {