ref/runtime/internal/flow/conn: Do not double encrypt encasulated connections.
Change-Id: I1621774ff58b6a0c4218876520c308baa16a4c88
diff --git a/runtime/internal/flow/conn/flow.go b/runtime/internal/flow/conn/flow.go
index a3207dc..820cba9 100644
--- a/runtime/internal/flow/conn/flow.go
+++ b/runtime/internal/flow/conn/flow.go
@@ -25,6 +25,7 @@
opened bool
q *readq
bkey, dkey uint64
+ noEncrypt bool
}
// Ensure that *flw implements flow.Flow.
@@ -46,6 +47,11 @@
return f
}
+// disableEncrytion should not be called concurrently with Write* methods.
+func (f *flw) disableEncryption() {
+ f.noEncrypt = false
+}
+
// Implement io.Reader.
// Read and ReadMsg should not be called concurrently with themselves
// or each other.
@@ -134,6 +140,9 @@
if alsoClose && done {
d.Flags |= message.CloseFlag
}
+ if f.noEncrypt {
+ d.Flags |= message.DisableEncryptionFlag
+ }
sent += size
return size, done, f.conn.mp.writeMsg(f.ctx, d)
})
