From 52a716a5e0dbb35d8e4fff55517898df5c3855f4 Mon Sep 17 00:00:00 2001
From: Igor Milavec <igor.milavec@gmail.com>
Date: Mon, 14 Feb 2022 10:28:38 +0100
Subject: [PATCH 1/2] Fix SocketAsyncEventArgsAwaitable calling continuation
 multiple times

---
 src/Renci.SshNet/Abstractions/SocketExtensions.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Renci.SshNet/Abstractions/SocketExtensions.cs b/src/Renci.SshNet/Abstractions/SocketExtensions.cs
index d763e1a34..d112e5339 100644
--- a/src/Renci.SshNet/Abstractions/SocketExtensions.cs
+++ b/src/Renci.SshNet/Abstractions/SocketExtensions.cs
@@ -36,7 +36,7 @@ public SocketAsyncEventArgsAwaitable ExecuteAsync(Func<SocketAsyncEventArgs, boo
             public void SetCompleted()
             {
                 IsCompleted = true;
-                var continuation = continuationAction ?? Interlocked.CompareExchange(ref continuationAction, SENTINEL, null);
+                var continuation = Interlocked.CompareExchange(ref continuationAction, SENTINEL, continuationAction);
                 if (continuation != null)
                 {
                     continuation();

From df812986fef77067ad951bdf019579d14af418d6 Mon Sep 17 00:00:00 2001
From: Rob Hague <rob.hague00@gmail.com>
Date: Sat, 27 Apr 2024 21:20:11 +0200
Subject: [PATCH 2/2] Use Interlocked.Exchange

---
 src/Renci.SshNet/Abstractions/SocketExtensions.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Renci.SshNet/Abstractions/SocketExtensions.cs b/src/Renci.SshNet/Abstractions/SocketExtensions.cs
index 42edadf33..eef7c1f6b 100644
--- a/src/Renci.SshNet/Abstractions/SocketExtensions.cs
+++ b/src/Renci.SshNet/Abstractions/SocketExtensions.cs
@@ -33,10 +33,10 @@ public AwaitableSocketAsyncEventArgs ExecuteAsync(Func<SocketAsyncEventArgs, boo
                 return this;
             }
 
-            public void SetCompleted()
+            private void SetCompleted()
             {
                 IsCompleted = true;
-                var continuation = Interlocked.CompareExchange(ref _continuationAction, SENTINEL, _continuationAction);
+                var continuation = Interlocked.Exchange(ref _continuationAction, SENTINEL);
                 if (continuation is not null)
                 {
                     continuation();