diff options
| author | Tamir Duberstein <tamird@google.com> | 2021-07-14 13:54:14 -0700 |
|---|---|---|
| committer | gVisor bot <gvisor-bot@google.com> | 2021-07-14 13:57:15 -0700 |
| commit | e963657e7c22b3b0be78c84ec7bec28dd0fc269b (patch) | |
| tree | cf8c29e030ac9d69a0b91c7ca7345b9bb47a600a | |
| parent | a1044cb88192ad3891500c867c22f5cb388054e5 (diff) | |
Set tcp endpoint state atomically
PiperOrigin-RevId: 384776517
| -rw-r--r-- | pkg/tcpip/transport/tcp/endpoint.go | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/pkg/tcpip/transport/tcp/endpoint.go b/pkg/tcpip/transport/tcp/endpoint.go index 9945fdd6b..ebc88d6c3 100644 --- a/pkg/tcpip/transport/tcp/endpoint.go +++ b/pkg/tcpip/transport/tcp/endpoint.go @@ -754,7 +754,7 @@ func (e *endpoint) ResumeWork() { // // Precondition: e.mu must be held to call this method. func (e *endpoint) setEndpointState(state EndpointState) { - oldstate := EndpointState(atomic.LoadUint32(&e.state)) + oldstate := EndpointState(atomic.SwapUint32(&e.state, uint32(state))) switch state { case StateEstablished: e.stack.Stats().TCP.CurrentEstablished.Increment() @@ -771,7 +771,6 @@ func (e *endpoint) setEndpointState(state EndpointState) { e.stack.Stats().TCP.CurrentEstablished.Decrement() } } - atomic.StoreUint32(&e.state, uint32(state)) } // EndpointState returns the current state of the endpoint. |