Add UnaryOperator visitor to DataFlowSanitizer

[cameron.mcinally]

Summary

I'm not familiar with this sanitizer at all, so I would appreciate a thorough review...

[cameron.mcinally]

Ping.

[cameron.mcinally]

Ping*2.

[pcc]

LGTM

llvm/trunk/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp

public:
 
   DFSanVisitor(DFSanFunction &DFSF) : DFSF(DFSF) {}
 
   const DataLayout &getDataLayout() const {
     return DFSF.F->getParent()->getDataLayout();
   }
 
   void visitOperandShadowInst(Instruction &I);
+  void visitUnaryOperator(UnaryOperator &UO);
   void visitBinaryOperator(BinaryOperator &BO);
   void visitCastInst(CastInst &CI);
   void visitCmpInst(CmpInst &CI);
   void visitGetElementPtrInst(GetElementPtrInst &GEPI);
   void visitLoadInst(LoadInst &LI);
   void visitStoreInst(StoreInst &SI);
   void visitReturnInst(ReturnInst &RI);
   void visitCallSite(CallSite CS);
void DFSanVisitor::visitStoreInst(StoreInst &SI) {
   Value* Shadow = DFSF.getShadow(SI.getValueOperand());
   if (ClCombinePointerLabelsOnStore) {
     Value *PtrShadow = DFSF.getShadow(SI.getPointerOperand());
     Shadow = DFSF.combineShadows(Shadow, PtrShadow, &SI);
   }
   DFSF.storeShadow(SI.getPointerOperand(), Size, Align, Shadow, &SI);
 }
 
+void DFSanVisitor::visitUnaryOperator(UnaryOperator &UO) {
+  visitOperandShadowInst(UO);
+}
+
 void DFSanVisitor::visitBinaryOperator(BinaryOperator &BO) {
   visitOperandShadowInst(BO);
 }
 
 void DFSanVisitor::visitCastInst(CastInst &CI) { visitOperandShadowInst(CI); }
 
 void DFSanVisitor::visitCmpInst(CmpInst &CI) { visitOperandShadowInst(CI); }
 

llvm/trunk/test/Instrumentation/DataFlowSanitizer/arith.ll

define i8 @udiv(i8 %a, i8 %b) {
   ; CHECK: load{{.*}}__dfsan_arg_tls
   ; CHECK: call{{.*}}__dfsan_union
   ; CHECK: udiv i8
   ; CHECK: store{{.*}}__dfsan_retval_tls
   ; CHECK: ret i8
   %c = udiv i8 %a, %b
   ret i8 %c
 }
+
+define double @fneg(double %a) {
+  ; CHECK: @"dfs$fneg"
+  ; CHECK: load{{.*}}__dfsan_arg_tls
+  ; CHECK: fneg double
+  ; CHECK: store{{.*}}__dfsan_retval_tls
+  ; CHECK: ret double
+  %c = fneg double %a
+  ret double %c
+}