Merge branch 'main' into tests/cancel-invoice

xWerk · Jul 22, 2024 · 1c8fc2a · 1c8fc2a
2 parents 6879df9 + ddce56a
commit 1c8fc2a
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/modules/invoice-module/sablier-v2/StreamManager.sol b/src/modules/invoice-module/sablier-v2/StreamManager.sol
@@ -70,6 +70,13 @@ abstract contract StreamManager is IStreamManager {
         _;
     }
 
+    /// @notice Reverts if the `msg.sender` is not the initial stream sender (creator of the stream)
+    modifier onlyInitialStreamSender(uint256 streamId) {
+        address initialSender = _initialStreamSender[streamId];
+        if (msg.sender != initialSender) revert Errors.OnlyInitialStreamSender(initialSender);
+        _;
+    }
+
     /*//////////////////////////////////////////////////////////////////////////
                                 NON-CONSTANT FUNCTIONS
     //////////////////////////////////////////////////////////////////////////*/
@@ -244,7 +251,12 @@ abstract contract StreamManager is IStreamManager {
     }
 
     /// @dev Withdraws from either a linear or tranched stream
-    function _withdrawStream(ISablierV2Lockup sablier, uint256 streamId, address to, uint128 amount) internal {
+    function _withdrawStream(
+        ISablierV2Lockup sablier,
+        uint256 streamId,
+        address to,
+        uint128 amount
+    ) internal onlyInitialStreamSender(streamId) {
         sablier.withdraw(streamId, to, amount);
     }