Preliminary debug state

mtproto/wrappers/abridged.go

@@ -31,15 +31,18 @@ func (a *AbridgedReadWriteCloserWithAddr) Read(p []byte) (int, error) {
 		buf := &bytes.Buffer{}
 		buf.Grow(3)
 
+		q := make([]byte, 1)
+
 		if _, err := io.CopyN(buf, a.conn, 1); err != nil {
 			return errors.Annotate(err, "Cannot read message length")
 		}
 		msgLength := uint8(buf.Bytes()[0])
+		q[0] = msgLength
 		buf.Reset()
 
 		if msgLength >= abridgedQuickAckLength {
 			a.opts.ReadHacks.QuickAck = true
-			msgLength -= 0x80
+			msgLength -= abridgedQuickAckLength
 		}
 
 		msgLength32 := uint32(msgLength)
@@ -49,6 +52,7 @@ func (a *AbridgedReadWriteCloserWithAddr) Read(p []byte) (int, error) {
 			}
 			number := uint24{}
 			copy(number[:], buf.Bytes())
+			q = append(q, buf.Bytes()...)
 			msgLength32 = fromUint24(number)
 		}
 		msgLength32 *= 4
@@ -59,6 +63,7 @@ func (a *AbridgedReadWriteCloserWithAddr) Read(p []byte) (int, error) {
 		if _, err := io.CopyN(buf, a.conn, int64(msgLength32)); err != nil {
 			return errors.Annotate(err, "Cannot read message")
 		}
+		q = append(q, buf.Bytes()...)
 		a.Buffer.Write(buf.Bytes())
 
 		return nil
@@ -87,10 +92,9 @@ func (a *AbridgedReadWriteCloserWithAddr) Write(p []byte) (int, error) {
 		buf.Write(length24[:])
 		buf.Write(p)
 		return a.conn.Write(buf.Bytes())
-
-	default:
-		return 0, errors.Errorf("Packet is too big %d", len(p))
 	}
+
+	return 0, errors.Errorf("Packet is too big %d", len(p))
 }
 
 func (a *AbridgedReadWriteCloserWithAddr) Close() error {
@@ -105,6 +109,10 @@ func (a *AbridgedReadWriteCloserWithAddr) RemoteAddr() *net.TCPAddr {
 	return a.conn.RemoteAddr()
 }
 
+func (a *AbridgedReadWriteCloserWithAddr) SocketID() string {
+	return a.conn.SocketID()
+}
+
 func toUint24(number uint32) uint24 {
 	return uint24{byte(number), byte(number >> 8), byte(number >> 16)}
 }

mtproto/wrappers/frame.go

@@ -120,6 +120,10 @@ func (f *FrameRWC) RemoteAddr() *net.TCPAddr {
 	return f.conn.RemoteAddr()
 }
 
+func (f *FrameRWC) SocketID() string {
+	return f.conn.SocketID()
+}
+
 func NewFrameRWC(conn wrappers.ReadWriteCloserWithAddr, seqNo int32) wrappers.ReadWriteCloserWithAddr {
 	return &FrameRWC{
 		BufferedReader: wrappers.NewBufferedReader(),

mtproto/wrappers/intermediate.go

@@ -77,6 +77,10 @@ func (i *IntermediateReadWriteCloserWithAddr) RemoteAddr() *net.TCPAddr {
 	return i.conn.RemoteAddr()
 }
 
+func (i *IntermediateReadWriteCloserWithAddr) SocketID() string {
+	return i.conn.SocketID()
+}
+
 func NewIntermediateRWC(conn wrappers.ReadWriteCloserWithAddr, connOpts *mtproto.ConnectionOpts) wrappers.ReadWriteCloserWithAddr {
 	return &IntermediateReadWriteCloserWithAddr{
 		BufferedReader: wrappers.NewBufferedReader(),

mtproto/wrappers/proxy_request.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/9seconds/mtg/mtproto"
 	"github.com/9seconds/mtg/mtproto/rpc"
+	"github.com/9seconds/mtg/utils"
 	"github.com/9seconds/mtg/wrappers"
 )
 
@@ -29,7 +30,7 @@ func (p *ProxyRequestReadWriteCloserWithAddr) Read(buf []byte) (int, error) {
 
 		switch {
 		case bytes.Equal(ans, rpc.TagProxyAns):
-			return p.readProxyAns(buf)
+			return p.readProxyAns()
 		case bytes.Equal(ans, rpc.TagSimpleAck):
 			return p.readSimpleAck()
 		case bytes.Equal(ans, rpc.TagCloseExt):
@@ -44,21 +45,16 @@ func (p *ProxyRequestReadWriteCloserWithAddr) readCloseExt() error {
 	return errors.New("Connection has been closed remotely")
 }
 
-func (p *ProxyRequestReadWriteCloserWithAddr) readProxyAns(buf []byte) (err error) {
+func (p *ProxyRequestReadWriteCloserWithAddr) readProxyAns() (err error) {
 	if _, err = io.CopyN(ioutil.Discard, p.conn, 8+4); err != nil {
 		return errors.Annotate(err, "Cannot skip flags and connid")
 	}
 
-	n := len(buf)
-	preBuffer := &bytes.Buffer{}
-	for n == len(buf) {
-		n, err = p.conn.Read(buf)
-		if err != nil {
-			return errors.Annotate(err, "Cannot read proxy answer")
-		}
-		preBuffer.Write(buf[:n])
+	buf, err := utils.ReadCurrentData(p.conn)
+	if err != nil {
+		return errors.Annotate(err, "Cannot read proxy answer")
 	}
-	p.Buffer.Write(preBuffer.Bytes())
+	p.Buffer.Write(buf)
 
 	return nil
 }
@@ -97,6 +93,10 @@ func (p *ProxyRequestReadWriteCloserWithAddr) RemoteAddr() *net.TCPAddr {
 	return p.conn.RemoteAddr()
 }
 
+func (p *ProxyRequestReadWriteCloserWithAddr) SocketID() string {
+	return p.conn.SocketID()
+}
+
 func NewProxyRequestRWC(conn wrappers.ReadWriteCloserWithAddr, connOpts *mtproto.ConnectionOpts, adTag []byte) (wrappers.ReadWriteCloserWithAddr, error) {
 	req, err := rpc.NewProxyRequest(connOpts.ClientAddr, conn.LocalAddr(), connOpts, adTag)
 	if err != nil {