|
| 1 | +From 012fcbff096e259de26aba5f9e88b05a56444faf Mon Sep 17 00:00:00 2001 |
| 2 | +From: Kevin Lockwood <v-klockwood@microsoft.com> |
| 3 | +Date: Tue, 18 Mar 2025 14:07:29 -0700 |
| 4 | +Subject: [PATCH] [Medium] patch clang16 for CVE-2023-29935 |
| 5 | + |
| 6 | +Link: https://github.com/llvm/llvm-project/commit/e7833c20d835d0f358acf7708a72bc23b1d87973.patch |
| 7 | +--- |
| 8 | + .../Conversion/MemRefToLLVM/MemRefToLLVM.cpp | 31 ++---------- |
| 9 | + .../MemRefToLLVM/memref-to-llvm.mlir | 48 +++++++++++++++---- |
| 10 | + 2 files changed, 45 insertions(+), 34 deletions(-) |
| 11 | + |
| 12 | +diff --git a/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp b/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp |
| 13 | +index 7132560ad..a225553f5 100644 |
| 14 | +--- a/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp |
| 15 | ++++ b/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp |
| 16 | +@@ -534,15 +534,12 @@ struct GenericAtomicRMWOpLowering |
| 17 | + |
| 18 | + // Split the block into initial, loop, and ending parts. |
| 19 | + auto *initBlock = rewriter.getInsertionBlock(); |
| 20 | +- auto *loopBlock = rewriter.createBlock( |
| 21 | +- initBlock->getParent(), std::next(Region::iterator(initBlock)), |
| 22 | +- valueType, loc); |
| 23 | +- auto *endBlock = rewriter.createBlock( |
| 24 | +- loopBlock->getParent(), std::next(Region::iterator(loopBlock))); |
| 25 | ++ auto *loopBlock = rewriter.splitBlock(initBlock, Block::iterator(atomicOp)); |
| 26 | ++ loopBlock->addArgument(valueType, loc); |
| 27 | + |
| 28 | +- // Operations range to be moved to `endBlock`. |
| 29 | +- auto opsToMoveStart = atomicOp->getIterator(); |
| 30 | +- auto opsToMoveEnd = initBlock->back().getIterator(); |
| 31 | ++ |
| 32 | ++ auto *endBlock = |
| 33 | ++ rewriter.splitBlock(loopBlock, Block::iterator(atomicOp)++); |
| 34 | + |
| 35 | + // Compute the loaded value and branch to the loop block. |
| 36 | + rewriter.setInsertionPointToEnd(initBlock); |
| 37 | +@@ -585,30 +582,12 @@ struct GenericAtomicRMWOpLowering |
| 38 | + loopBlock, newLoaded); |
| 39 | + |
| 40 | + rewriter.setInsertionPointToEnd(endBlock); |
| 41 | +- moveOpsRange(atomicOp.getResult(), newLoaded, std::next(opsToMoveStart), |
| 42 | +- std::next(opsToMoveEnd), rewriter); |
| 43 | + |
| 44 | + // The 'result' of the atomic_rmw op is the newly loaded value. |
| 45 | + rewriter.replaceOp(atomicOp, {newLoaded}); |
| 46 | + |
| 47 | + return success(); |
| 48 | + } |
| 49 | +- |
| 50 | +-private: |
| 51 | +- // Clones a segment of ops [start, end) and erases the original. |
| 52 | +- void moveOpsRange(ValueRange oldResult, ValueRange newResult, |
| 53 | +- Block::iterator start, Block::iterator end, |
| 54 | +- ConversionPatternRewriter &rewriter) const { |
| 55 | +- IRMapping mapping; |
| 56 | +- mapping.map(oldResult, newResult); |
| 57 | +- SmallVector<Operation *, 2> opsToErase; |
| 58 | +- for (auto it = start; it != end; ++it) { |
| 59 | +- rewriter.clone(*it, mapping); |
| 60 | +- opsToErase.push_back(&*it); |
| 61 | +- } |
| 62 | +- for (auto *it : opsToErase) |
| 63 | +- rewriter.eraseOp(it); |
| 64 | +- } |
| 65 | + }; |
| 66 | + |
| 67 | + /// Returns the LLVM type of the global variable given the memref type `type`. |
| 68 | +diff --git a/mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir b/mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir |
| 69 | +index 1a8a75d1e..ba2d75fc6 100644 |
| 70 | +--- a/mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir |
| 71 | ++++ b/mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir |
| 72 | +@@ -366,16 +366,48 @@ func.func @generic_atomic_rmw(%I : memref<10xi32>, %i : index) { |
| 73 | + ^bb0(%old_value : i32): |
| 74 | + memref.atomic_yield %old_value : i32 |
| 75 | + } |
| 76 | +- // CHECK: [[init:%.*]] = llvm.load %{{.*}} : !llvm.ptr<i32> |
| 77 | +- // CHECK-NEXT: llvm.br ^bb1([[init]] : i32) |
| 78 | +- // CHECK-NEXT: ^bb1([[loaded:%.*]]: i32): |
| 79 | +- // CHECK-NEXT: [[pair:%.*]] = llvm.cmpxchg %{{.*}}, [[loaded]], [[loaded]] |
| 80 | +- // CHECK-SAME: acq_rel monotonic : i32 |
| 81 | +- // CHECK-NEXT: [[new:%.*]] = llvm.extractvalue [[pair]][0] |
| 82 | +- // CHECK-NEXT: [[ok:%.*]] = llvm.extractvalue [[pair]][1] |
| 83 | +- // CHECK-NEXT: llvm.cond_br [[ok]], ^bb2, ^bb1([[new]] : i32) |
| 84 | + llvm.return |
| 85 | + } |
| 86 | ++// CHECK: %[[INIT:.*]] = llvm.load %{{.*}} : !llvm.ptr -> i32 |
| 87 | ++// CHECK-NEXT: llvm.br ^bb1(%[[INIT]] : i32) |
| 88 | ++// CHECK-NEXT: ^bb1(%[[LOADED:.*]]: i32): |
| 89 | ++// CHECK-NEXT: %[[PAIR:.*]] = llvm.cmpxchg %{{.*}}, %[[LOADED]], %[[LOADED]] |
| 90 | ++// CHECK-SAME: acq_rel monotonic : !llvm.ptr, i32 |
| 91 | ++// CHECK-NEXT: %[[NEW:.*]] = llvm.extractvalue %[[PAIR]][0] |
| 92 | ++// CHECK-NEXT: %[[OK:.*]] = llvm.extractvalue %[[PAIR]][1] |
| 93 | ++// CHECK-NEXT: llvm.cond_br %[[OK]], ^bb2, ^bb1(%[[NEW]] : i32) |
| 94 | ++ |
| 95 | ++// ----- |
| 96 | ++ |
| 97 | ++// CHECK-LABEL: func @generic_atomic_rmw_in_alloca_scope |
| 98 | ++func.func @generic_atomic_rmw_in_alloca_scope(){ |
| 99 | ++ %c1 = arith.constant 1 : index |
| 100 | ++ %alloc = memref.alloc() : memref<2x3xi32> |
| 101 | ++ memref.alloca_scope { |
| 102 | ++ %0 = memref.generic_atomic_rmw %alloc[%c1, %c1] : memref<2x3xi32> { |
| 103 | ++ ^bb0(%arg0: i32): |
| 104 | ++ memref.atomic_yield %arg0 : i32 |
| 105 | ++ } |
| 106 | ++ } |
| 107 | ++ return |
| 108 | ++} |
| 109 | ++// CHECK: %[[STACK_SAVE:.*]] = llvm.intr.stacksave : !llvm.ptr |
| 110 | ++// CHECK-NEXT: llvm.br ^bb1 |
| 111 | ++// CHECK: ^bb1: |
| 112 | ++// CHECK: %[[INIT:.*]] = llvm.load %[[BUF:.*]] : !llvm.ptr -> i32 |
| 113 | ++// CHECK-NEXT: llvm.br ^bb2(%[[INIT]] : i32) |
| 114 | ++// CHECK-NEXT: ^bb2(%[[LOADED:.*]]: i32): |
| 115 | ++// CHECK-NEXT: %[[PAIR:.*]] = llvm.cmpxchg %[[BUF]], %[[LOADED]], %[[LOADED]] |
| 116 | ++// CHECK-SAME: acq_rel monotonic : !llvm.ptr, i32 |
| 117 | ++// CHECK-NEXT: %[[NEW:.*]] = llvm.extractvalue %[[PAIR]][0] |
| 118 | ++// CHECK-NEXT: %[[OK:.*]] = llvm.extractvalue %[[PAIR]][1] |
| 119 | ++// CHECK-NEXT: llvm.cond_br %[[OK]], ^bb3, ^bb2(%[[NEW]] : i32) |
| 120 | ++// CHECK-NEXT: ^bb3: |
| 121 | ++// CHECK-NEXT: llvm.intr.stackrestore %[[STACK_SAVE]] : !llvm.ptr |
| 122 | ++// CHECK-NEXT: llvm.br ^bb4 |
| 123 | ++// CHECK-NEXT: ^bb4: |
| 124 | ++// CHECK-NEXT: return |
| 125 | ++ |
| 126 | + |
| 127 | + // ----- |
| 128 | + |
| 129 | +-- |
| 130 | +2.34.1 |
| 131 | + |
0 commit comments