use go 1.18 atomic operations

Author: espadolini

Diff for: ssh/server_test.go

@@ -71,10 +71,10 @@ func TestNewServerConnValidationErrors(t *testing.T) {
 	if err == nil {
 		t.Fatal("NewServerConn with invalid public key auth algorithms succeeded")
 	}
-	if !c.closed.Load() {
+	if !c.isClosed() {
 		t.Fatal("NewServerConn with invalid public key auth algorithms left connection open")
 	}
-	if c.used.Load() {
+	if c.isUsed() {
 		t.Fatal("NewServerConn with invalid public key auth algorithms used connection")
 	}
 
@@ -88,36 +88,44 @@ func TestNewServerConnValidationErrors(t *testing.T) {
 	if err == nil {
 		t.Fatal("NewServerConn with unsupported key exchange succeeded")
 	}
-	if !c.closed.Load() {
+	if !c.isClosed() {
 		t.Fatal("NewServerConn with unsupported key exchange left connection open")
 	}
-	if c.used.Load() {
+	if c.isUsed() {
 		t.Fatal("NewServerConn with unsupported key exchange used connection")
 	}
 }
 
 type markerConn struct {
-	closed atomic.Bool
-	used   atomic.Bool
+	closed uint32
+	used   uint32
+}
+
+func (c *markerConn) isClosed() bool {
+	return atomic.LoadUint32(&c.closed) != 0
+}
+
+func (c *markerConn) isUsed() bool {
+	return atomic.LoadUint32(&c.used) != 0
 }
 
 func (c *markerConn) Close() error {
-	c.closed.Store(true)
+	atomic.StoreUint32(&c.closed, 1)
 	return nil
 }
 
 func (c *markerConn) Read(b []byte) (n int, err error) {
-	c.used.Store(true)
-	if c.closed.Load() {
+	atomic.StoreUint32(&c.used, 1)
+	if atomic.LoadUint32(&c.closed) != 0 {
 		return 0, net.ErrClosed
 	} else {
 		return 0, io.EOF
 	}
 }
 
 func (c *markerConn) Write(b []byte) (n int, err error) {
-	c.used.Store(true)
-	if c.closed.Load() {
+	atomic.StoreUint32(&c.used, 1)
+	if atomic.LoadUint32(&c.closed) != 0 {
 		return 0, net.ErrClosed
 	} else {
 		return 0, io.ErrClosedPipe