gopenfusion/server/client.go

package main

import (
	"encoding/binary"
	"fmt"
	"log"
	"net"

	"github.com/CPunch/GoFusion/protocol"
)

type Client struct {
	server *Server
	conn   net.Conn
	key    []byte
}

func newClient(server *Server, conn net.Conn, key []byte) *Client {
	return &Client{
		server: server,
		conn:   conn,
		key:    key,
	}
}

func (client *Client) Send(data interface{}, typeID uint32) {
	tmp := make([]byte, 4)

	// encode
	pkt := protocol.NewPacket(0)
	pkt.Encode(data)

	// write packet size
	log.Printf("Sending %#v, sizeof: %d", data, len(pkt.Buf))
	binary.LittleEndian.PutUint32(tmp, uint32(len(pkt.Buf)+4))
	if _, err := client.conn.Write(tmp); err != nil {
		panic(fmt.Errorf("[FATAL] failed to write packet size! %v", err))
	}

	// prepend the typeID to the packet body
	binary.LittleEndian.PutUint32(tmp, uint32(typeID))
	pkt.Buf = append(tmp, pkt.Buf...)

	// encrypt body
	protocol.EncryptData(pkt.Buf[:], client.key)

	// write packet body
	if _, err := client.conn.Write(pkt.Buf); err != nil {
		panic(fmt.Errorf("[FATAL] failed to write packet body! %v", err))
	}
	log.Printf("sent!")
}

func (client *Client) ClientHandler() {
	defer func() {
		client.conn.Close()
		client.server.unregister <- client
	}()

	tmp := make([]byte, 4)
	for {
		// read packet size
		if _, err := client.conn.Read(tmp); err != nil {
			panic(fmt.Errorf("[FATAL] failed to read packet size! %v", err))
		}
		sz := int(binary.LittleEndian.Uint32(tmp))

		// read packet body
		pkt := protocol.NewPacket(sz)
		if _, err := client.conn.Read(pkt.Buf); err != nil {
			panic(fmt.Errorf("[FATAL] failed to read packet body! %v", err))
		}

		// decrypt && grab typeID
		protocol.DecryptData(pkt.Buf[:], client.key)
		typeID := int(binary.LittleEndian.Uint32(pkt.Read(4)))

		log.Printf("Got packet ID: %x, with a sizeof: %d\n", typeID, len(pkt.Buf))
		pkt.ResetCursor()
		switch typeID {
		case protocol.P_CL2LS_REQ_LOGIN:
			var loginPkt protocol.SP_CL2LS_REQ_LOGIN
			pkt.Decode(&loginPkt)
			log.Printf("Got packet: %#v", loginPkt)

			client.Send(&protocol.SP_LS2CL_REP_LOGIN_FAIL{ErrorCode: protocol.LOGIN_FAIL_VERSION_ERROR, ID: loginPkt.ID}, protocol.P_LS2CL_REP_LOGIN_FAIL)
		default:
			log.Printf("[WARN] unsupported packet ID: %x\n", typeID)
		}
	}
}