From d50c2746d6f2dd3c2c4f75b683f0affde0824827 Mon Sep 17 00:00:00 2001
From: Him188 <Him188@mamoe.net>
Date: Wed, 2 Oct 2019 20:51:11 +0800
Subject: [PATCH] Little updates

---
 .../network/protocol/tim/handler/ActionPacketHandler.kt  | 8 +++++++-
 .../mirai/network/protocol/tim/packet/ClientPacket.kt    | 4 ++--
 .../mamoe/mirai/network/protocol/tim/packet/Packet.kt    | 4 +---
 .../network/protocol/tim/packet/{ => login}/SKey.kt      | 3 ++-
 .../network/protocol/tim/packet/{ => login}/Session.kt   | 3 ++-
 .../network/protocol/tim/packet/{ => login}/Touch.kt     | 4 ++--
 .../protocol/tim/packet/{ => login}/VerificationCode.kt  | 9 +++++----
 7 files changed, 21 insertions(+), 14 deletions(-)
 rename mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/{ => login}/SKey.kt (94%)
 rename mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/{ => login}/Session.kt (97%)
 rename mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/{ => login}/Touch.kt (96%)
 rename mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/{ => login}/VerificationCode.kt (96%)

diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/handler/ActionPacketHandler.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/handler/ActionPacketHandler.kt
index a010cf8b5..8b06d7d56 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/handler/ActionPacketHandler.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/handler/ActionPacketHandler.kt
@@ -2,7 +2,10 @@ package net.mamoe.mirai.network.protocol.tim.handler
 
 import kotlinx.coroutines.runBlocking
 import net.mamoe.mirai.network.LoginSession
-import net.mamoe.mirai.network.protocol.tim.packet.*
+import net.mamoe.mirai.network.protocol.tim.packet.ClientAccountInfoRequestPacket
+import net.mamoe.mirai.network.protocol.tim.packet.ServerAccountInfoResponsePacket
+import net.mamoe.mirai.network.protocol.tim.packet.ServerEventPacket
+import net.mamoe.mirai.network.protocol.tim.packet.ServerPacket
 import net.mamoe.mirai.network.protocol.tim.packet.action.AddFriendResult
 import net.mamoe.mirai.network.protocol.tim.packet.action.ClientAddFriendPacket
 import net.mamoe.mirai.network.protocol.tim.packet.action.ClientCanAddFriendPacket
@@ -10,6 +13,9 @@ import net.mamoe.mirai.network.protocol.tim.packet.action.ServerCanAddFriendResp
 import net.mamoe.mirai.network.protocol.tim.packet.image.ServerTryGetImageIDFailedPacket
 import net.mamoe.mirai.network.protocol.tim.packet.image.ServerTryGetImageIDResponsePacket
 import net.mamoe.mirai.network.protocol.tim.packet.image.ServerTryGetImageIDSuccessPacket
+import net.mamoe.mirai.network.protocol.tim.packet.login.ClientSKeyRefreshmentRequestPacket
+import net.mamoe.mirai.network.protocol.tim.packet.login.ClientSKeyRequestPacket
+import net.mamoe.mirai.network.protocol.tim.packet.login.ServerSKeyResponsePacket
 import net.mamoe.mirai.task.MiraiThreadPool
 import net.mamoe.mirai.utils.getGTK
 import java.awt.image.BufferedImage
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/ClientPacket.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/ClientPacket.kt
index 3408a4fa8..d673fc26e 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/ClientPacket.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/ClientPacket.kt
@@ -192,9 +192,9 @@ fun UInt.toByteArray(): ByteArray = byteArrayOf(
 
 fun Int.toUHexString(separator: String = " "): String = this.toByteArray().toUHexString(separator)
 
-internal fun md5(str: String): ByteArray = MessageDigest.getInstance("MD5").digest(str.toByteArray())
+fun md5(str: String): ByteArray = MessageDigest.getInstance("MD5").digest(str.toByteArray())
 
-internal fun md5(byteArray: ByteArray): ByteArray = MessageDigest.getInstance("MD5").digest(byteArray)
+fun md5(byteArray: ByteArray): ByteArray = MessageDigest.getInstance("MD5").digest(byteArray)
 
 
 @Throws(IOException::class)
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Packet.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Packet.kt
index 67d62c971..7a0d114c1 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Packet.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Packet.kt
@@ -3,9 +3,7 @@ package net.mamoe.mirai.network.protocol.tim.packet
 /**
  * @author Him188moe
  */
-interface Packet {
-
-}
+interface Packet
 
 
 object PacketNameFormatter {
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/SKey.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/SKey.kt
similarity index 94%
rename from mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/SKey.kt
rename to mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/SKey.kt
index e883afad3..90a2c44e5 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/SKey.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/SKey.kt
@@ -1,6 +1,7 @@
-package net.mamoe.mirai.network.protocol.tim.packet
+package net.mamoe.mirai.network.protocol.tim.packet.login
 
 import net.mamoe.mirai.network.protocol.tim.TIMProtocol
+import net.mamoe.mirai.network.protocol.tim.packet.*
 import net.mamoe.mirai.utils.TEA
 import java.io.DataInputStream
 
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Session.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Session.kt
similarity index 97%
rename from mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Session.kt
rename to mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Session.kt
index f596b7b11..682397356 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Session.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Session.kt
@@ -1,6 +1,7 @@
-package net.mamoe.mirai.network.protocol.tim.packet
+package net.mamoe.mirai.network.protocol.tim.packet.login
 
 import net.mamoe.mirai.network.protocol.tim.TIMProtocol
+import net.mamoe.mirai.network.protocol.tim.packet.*
 import net.mamoe.mirai.utils.TEA
 import net.mamoe.mirai.utils.lazyEncode
 import java.io.DataInputStream
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Touch.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Touch.kt
similarity index 96%
rename from mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Touch.kt
rename to mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Touch.kt
index 1f828288c..dd00b17c6 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/Touch.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/Touch.kt
@@ -1,7 +1,7 @@
-package net.mamoe.mirai.network.protocol.tim.packet
+package net.mamoe.mirai.network.protocol.tim.packet.login
 
 import net.mamoe.mirai.network.protocol.tim.TIMProtocol
-import net.mamoe.mirai.network.protocol.tim.packet.login.ClientPasswordSubmissionPacket
+import net.mamoe.mirai.network.protocol.tim.packet.*
 import net.mamoe.mirai.utils.ByteArrayDataOutputStream
 import net.mamoe.mirai.utils.TEA
 import net.mamoe.mirai.utils.hexToBytes
diff --git a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/VerificationCode.kt b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/VerificationCode.kt
similarity index 96%
rename from mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/VerificationCode.kt
rename to mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/VerificationCode.kt
index c55e91bfd..f27977b45 100644
--- a/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/VerificationCode.kt
+++ b/mirai-core/src/main/java/net/mamoe/mirai/network/protocol/tim/packet/login/VerificationCode.kt
@@ -1,6 +1,7 @@
-package net.mamoe.mirai.network.protocol.tim.packet
+package net.mamoe.mirai.network.protocol.tim.packet.login
 
 import net.mamoe.mirai.network.protocol.tim.TIMProtocol
+import net.mamoe.mirai.network.protocol.tim.packet.*
 import net.mamoe.mirai.utils.TEA
 import net.mamoe.mirai.utils.Tested
 import net.mamoe.mirai.utils.hexToBytes
@@ -51,11 +52,11 @@ class ClientVerificationCodeSubmitPacket(
         private val packetIdLast: Int,
         private val qq: Long,
         private val token0825: ByteArray,
-        private val verificationCode: String,
+        private val captcha: String,
         private val verificationToken: ByteArray
 ) : ClientPacket() {
     init {
-        require(verificationCode.length == 4) { "verificationCode.length must == 4" }
+        require(captcha.length == 4) { "captcha.length must == 4" }
     }
 
     override fun encode() {
@@ -75,7 +76,7 @@ class ClientVerificationCodeSubmitPacket(
             it.writeHex(TIMProtocol.publicKey)//25
 
             it.writeHex("14 00 05 00 00 00 00 00 04")
-            it.write(verificationCode.toUpperCase().toByteArray())
+            it.write(captcha.toUpperCase().toByteArray())
             it.writeHex("00 38")
             it.write(verificationToken)