add transformPlayerInfo and go fmt

DarkNeos/transform.go

@@ -8,7 +8,7 @@ import (
 	"google.golang.org/protobuf/proto"
 )
 
-const FILLING_TOKEN uint16 = 0xcccc;
+const FILLING_TOKEN uint16 = 0xcccc
 const UTF16_BUFFER_MAX_LEN int = 20
 
 const (
@@ -26,7 +26,9 @@ func Transform(src []byte, tranformType int) ([]byte, error) {
 
 		switch message.Msg.(type) {
 		case *(ygopropb.YgoCtosMsg_CtosPlayerInfo):
-        return TransformPlayerInfo(message.GetCtosPlayerInfo()), nil
+			return transformPlayerInfo(message.GetCtosPlayerInfo()), nil
+		case *(ygopropb.YgoCtosMsg_CtosJoinGame):
+			return transformJoinGame(message.GetCtosJoinGame()), nil
 		default:
 			return nil, errors.New("Unhandled YgoCtosMsg type")
 		}
@@ -37,13 +39,26 @@ func Transform(src []byte, tranformType int) ([]byte, error) {
 	}
 }
 
+func transformPlayerInfo(pb *ygopropb.CtosPlayerInfo) []byte {
+	buf := strToUtf16Buffer(pb.Name)
 
-func TransformPlayerInfo(pb *ygopropb.CtosPlayerInfo) []byte {
-  info := strToUtf16Buffer(pb.Name)
-
-  return uint16BufToByteBuf(info)
+	return uint16BufToByteBuf(buf)
 }
 
+func transformJoinGame(pb *ygopropb.CtosJoinGame) []byte {
+	buf := make([]byte, 0)
+
+	version := uint16(pb.Version)
+	buf = append(buf, byte(version), byte(pb.Version>>8), 0, 0)
+
+	buf = append(buf, byte(pb.Gameid), byte(pb.Gameid>>8), byte(pb.Gameid>>16), byte(pb.Gameid>>24))
+
+	for _, v := range uint16BufToByteBuf(strToUtf16Buffer(pb.Passwd)) {
+		buf = append(buf, v)
+	}
+
+	return buf
+}
 
 func strToUtf16Buffer(s string) []uint16 {
 	b := make([]uint16, UTF16_BUFFER_MAX_LEN, UTF16_BUFFER_MAX_LEN)
@@ -58,20 +73,23 @@ func strToUtf16Buffer(s string) []uint16 {
 		if i < UTF16_BUFFER_MAX_LEN {
 			b[i] = v
 
-      if i == len(s_utf16) - 1 && i < len(b) - 1 {
-        b[i + 1] = 0
+			if i == len(s_utf16)-1 && i < len(b)-1 {
+				b[i+1] = 0
+			}
+		} else {
+			break
 		}
-    } else { break }
 	}
 
 	return b
 }
 
 func uint16BufToByteBuf(u16_b []uint16) []byte {
-  b := make([]byte, 0, len(u16_b) * 2)
+	b := make([]byte, 0, len(u16_b)*2)
 
 	for _, v := range u16_b {
-    b = append(b, byte(v >> 8), byte(v))
+		// little endian
+		b = append(b, byte(v), byte(v>>8))
 	}
 
 	return b