From 53ef81a4fc52a63614202367913c8c1acd534b24 Mon Sep 17 00:00:00 2001
From: 9seconds <nineseconds@yandex.ru>
Date: Thu, 5 Jul 2018 17:10:49 +0300
Subject: [PATCH] Use the same protocol as client

---
 client/middle.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/client/middle.go b/client/middle.go
index d5dc4d3..2197cec 100644
--- a/client/middle.go
+++ b/client/middle.go
@@ -21,5 +21,10 @@ func MiddleInit(conn net.Conn, conf *config.Config) (*mtproto.ConnectionOpts, wr
 		newConn = mtwrappers.NewIntermediateRWC(newConn, opts)
 	}
 
+	opts.ConnectionProto = mtproto.ConnectionProtocolIPv4
+	if conn.LocalAddr().(*net.TCPAddr).IP.To4() == nil {
+		opts.ConnectionProto = mtproto.ConnectionProtocolIPv6
+	}
+
 	return opts, newConn, nil
 }