mirror of
https://github.com/CPunch/gopenfusion.git
synced 2024-11-14 03:50:05 +00:00
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
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)
|
|
}
|
|
}
|
|
}
|