mirror of
https://github.com/CPunch/gopenfusion.git
synced 2026-01-28 15:40:05 +00:00
can now respond to packets
This commit is contained in:
91
server/client.go
Normal file
91
server/client.go
Normal file
@@ -0,0 +1,91 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
}
|
||||
109
server/server.go
109
server/server.go
@@ -1,7 +1,6 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
@@ -9,93 +8,49 @@ import (
|
||||
"github.com/CPunch/GoFusion/protocol"
|
||||
)
|
||||
|
||||
const (
|
||||
E_KEY = "m@rQn~W#"
|
||||
KEY_LENGTH = 8
|
||||
)
|
||||
|
||||
func encrypt_byte_change_A(ERSize int, data []byte) int {
|
||||
var num, num2, num3 int
|
||||
|
||||
for num+ERSize <= len(data) {
|
||||
num4 := num + num3
|
||||
num5 := num + (ERSize - 1 - num3)
|
||||
|
||||
b := data[num4]
|
||||
data[num4] = data[num5]
|
||||
data[num5] = b
|
||||
num += ERSize
|
||||
num3++
|
||||
if num3 > ERSize/2 {
|
||||
num3 = 0
|
||||
}
|
||||
}
|
||||
|
||||
num2 = ERSize - (num + ERSize - len(data))
|
||||
return num + num2
|
||||
type Server struct {
|
||||
listener net.Listener
|
||||
clients map[*Client]bool
|
||||
unregister chan *Client
|
||||
}
|
||||
|
||||
func xorData(buff, key []byte, size int) {
|
||||
for i := 0; i < size; i++ {
|
||||
buff[i] ^= key[i%KEY_LENGTH]
|
||||
func newServer() *Server {
|
||||
listener, err := net.Listen("tcp", ":23000")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
return &Server{
|
||||
listener: listener,
|
||||
clients: make(map[*Client]bool),
|
||||
unregister: make(chan *Client),
|
||||
}
|
||||
}
|
||||
|
||||
func decryptData(buff, key []byte) {
|
||||
ERSize := len(buff)%(KEY_LENGTH/2+1)*2 + KEY_LENGTH
|
||||
size := encrypt_byte_change_A(ERSize, buff)
|
||||
xorData(buff, key, size)
|
||||
}
|
||||
func (server *Server) Start() {
|
||||
log.Print("Server hosted on 127.0.0.1:23000")
|
||||
|
||||
func handle(conn net.Conn) {
|
||||
defer func() {
|
||||
conn.Close()
|
||||
}()
|
||||
|
||||
tmp := make([]byte, 4)
|
||||
for {
|
||||
// grab packet size
|
||||
if _, err := conn.Read(tmp); err != nil {
|
||||
panic(fmt.Errorf("[FATAL] failed to read packet size! %v", err))
|
||||
}
|
||||
|
||||
sz := int(binary.LittleEndian.Uint32(tmp))
|
||||
pkt := protocol.NewPacket(sz)
|
||||
|
||||
if _, err := conn.Read(pkt.Buf); err != nil {
|
||||
panic(fmt.Errorf("[FATAL] failed to read packet body! %v", err))
|
||||
}
|
||||
|
||||
decryptData(pkt.Buf, []byte(E_KEY))
|
||||
typeID := int(binary.LittleEndian.Uint32(pkt.Read(4)))
|
||||
sizeof := len(pkt.Buf)
|
||||
log.Printf("Got packet ID: %x\n", typeID)
|
||||
|
||||
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, with a sizeof: %d", loginPkt, sizeof)
|
||||
select {
|
||||
case client := <-server.unregister:
|
||||
delete(server.clients, client)
|
||||
fmt.Printf("Client %p disconnected\n", client)
|
||||
default:
|
||||
log.Printf("[WARN] unsupported packet ID: %x\n", typeID)
|
||||
conn, err := server.listener.Accept()
|
||||
if err != nil {
|
||||
log.Println("Connection error: ", err)
|
||||
return
|
||||
}
|
||||
|
||||
client := newClient(server, conn, []byte(protocol.E_KEY))
|
||||
server.clients[client] = true
|
||||
go client.ClientHandler()
|
||||
fmt.Printf("Client %p connected\n", client)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
server, err := net.Listen("tcp", ":23000")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
log.Print("Server hosted on 127.0.0.1:23000")
|
||||
for {
|
||||
conn, err := server.Accept()
|
||||
if err != nil {
|
||||
log.Println("Connection error: ", err)
|
||||
return
|
||||
}
|
||||
go handle(conn)
|
||||
}
|
||||
server := newServer()
|
||||
server.Start()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user