protocol/v0v1/server: Add FlushIO

1 files changed, 9 insertions, 0 deletions

diff --git a/protocol/v0v1/server/session.go b/protocol/v0v1/server/session.go
index 75884806..7e107f53 100644
--- a/protocol/v0v1/server/session.go
+++ b/protocol/v0v1/server/session.go
@@ -87,6 +87,15 @@ func (session *Session) WriteFlush() error {
 	return session.enc.WriteFlush()
 }
 
+// FlushIO flushes buffered transport output without emitting pkt-line frames.
+func (session *Session) FlushIO() error {
+	if session.useSideBand {
+		return session.sideband.FlushIO()
+	}
+
+	return session.enc.FlushIO()
+}
+
 // ProgressWriter returns one chunking writer for sideband progress output.
 //
 // When side-band-64k was not negotiated, writes are discarded.