init

net/conn.go

@@ -0,0 +1,57 @@
+package net
+
+import (
+	"bufio"
+	"crypto/cipher"
+	"io"
+	"net"
+
+	pk "github.com/Tnze/go-mc/net/packet"
+)
+
+type Conn struct {
+	socket net.Conn
+	io.ByteReader
+	io.Writer
+
+	threshold int
+}
+
+func DialMC(addr string) (conn *Conn, err error) {
+	conn = new(Conn)
+	conn.socket, err = net.Dial("tcp", addr)
+	if err != nil {
+		return
+	}
+
+	conn.ByteReader = bufio.NewReader(conn.socket)
+	conn.Writer = conn.socket
+
+	return
+}
+
+func (c *Conn) ReadPacket() (pk.Packet, error) {
+	pk, err := pk.RecvPacket(c.ByteReader, c.threshold > 0)
+	return *pk, err
+}
+
+func (c *Conn) WritePacket(p pk.Packet) error {
+	_, err := c.Write(p.Pack(c.threshold))
+	return err
+}
+
+func (c *Conn) SetCipher(encoStream, decoStream cipher.Stream) {
+	//加密连接
+	c.ByteReader = bufio.NewReader(cipher.StreamReader{ //Set reciver for AES
+		S: decoStream,
+		R: c.socket,
+	})
+	c.Writer = cipher.StreamWriter{
+		S: encoStream,
+		W: c.socket,
+	}
+}
+
+func (c *Conn) SetThreshold(t int) {
+	c.threshold = t
+}