初步使用新封包

2026-06-12 14:10:50 +08:00 · 2021-12-11 23:50:46 +08:00
parent a16926a673
commit 5abe175e90
6 changed files with 405 additions and 151 deletions
--- a/crypto.c
+++ b/crypto.c
@@ -0,0 +1,128 @@
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include "crypto.h"
+
+// TEA encoding sumtable
+static const uint32_t sumtable[0x10] = {
+	0x9e3579b9,
+	0x3c6ef172,
+	0xd2a66d2b,
+	0x78dd36e4,
+	0x17e5609d,
+	0xb54fda56,
+	0x5384560f,
+	0xf1bb77c8,
+	0x8ff24781,
+	0x2e4ac13a,
+	0xcc653af3,
+	0x6a9964ac,
+	0x08d12965,
+	0xa708081e,
+	0x451221d7,
+	0xe37793d0,
+};
+
+static uint8_t seqs[THREADCNT]; // 消息序号
+
+static inline int is_md5_equal(uint8_t* digest, uint8_t* digest2) {
+    #ifdef CPUBIT64
+        return (digest[0] == digest2[0]) &&
+                (digest[1] == digest2[1]);
+    #else
+        return (digest[0] == digest2[0]) &&
+                (digest[1] == digest2[1]) &&
+                (digest[2] == digest2[2]) &&
+                (digest[3] == digest2[3]);
+    #endif
+}
+
+void init_crypto() {
+    srand(time(NULL));
+}
+
+void reset_seq(int index) {
+    seqs[index] = 0;
+}
+
+char* raw_encrypt(const char* buf, off_t* len, int index, const char pwd[64]) {
+    TEADAT tin = {*len, (uint8_t*)buf};
+    TEA tea[4];
+
+    ((uint64_t*)tea)[0] = ((uint64_t*)pwd)[0];
+    ((uint64_t*)tea)[1] = ((uint64_t*)pwd)[1];
+    ((uint8_t*)tea)[15] = seqs[index]++;
+    TEADAT* tout = tea_encrypt_native_endian(tea, sumtable, &tin);
+
+    *len = tout->len;
+    char* encbuf = (char*)malloc(*len);
+    memcpy(encbuf, tout->data, *len);
+    free(tout->ptr);
+    free(tout);
+
+    return encbuf;
+}
+
+char* raw_decrypt(const char* buf, off_t* len, int index, const char pwd[64]) {
+    TEADAT tin = {*len, (uint8_t*)buf};
+    TEA tea[4];
+
+    ((uint64_t*)tea)[0] = ((uint64_t*)pwd)[0];
+    ((uint64_t*)tea)[1] = ((uint64_t*)pwd)[1];
+    ((uint8_t*)tea)[15] = seqs[index]++;
+    TEADAT* tout = tea_decrypt_native_endian(tea, sumtable, &tin);
+
+    *len = tout->len;
+    char* decbuf = (char*)malloc(*len);
+    memcpy(decbuf, tout->data, *len);
+    free(tout->ptr);
+    free(tout);
+
+    return decbuf;
+}
+
+void cmdpacket_encrypt(CMDPACKET* p, int index, const char pwd[64]) {
+    TEADAT tin = {p->datalen, p->data};
+    TEA tea[4];
+
+    ((uint64_t*)tea)[0] = ((uint64_t*)pwd)[0];
+    ((uint64_t*)tea)[1] = ((uint64_t*)pwd)[1];
+    ((uint8_t*)tea)[15] = seqs[index]++;
+    TEADAT* tout = tea_encrypt_native_endian(tea, sumtable, &tin);
+
+    uint8_t* datamd5 = md5(p->data, p->datalen);
+    memcpy(p->md5, datamd5, 16);
+    free(datamd5);
+
+    p->datalen = tout->len;
+    memcpy(p->data, tout->data, p->datalen);
+    free(tout->ptr);
+    free(tout);
+
+    return;
+}
+
+int cmdpacket_decrypt(CMDPACKET* p, int index, const char pwd[64]) {
+    TEADAT tin = {p->datalen, p->data};
+    TEA tea[4];
+
+    ((uint64_t*)tea)[0] = ((uint64_t*)pwd)[0];
+    ((uint64_t*)tea)[1] = ((uint64_t*)pwd)[1];
+    ((uint8_t*)tea)[15] = seqs[index];
+    TEADAT* tout = tea_decrypt_native_endian(tea, sumtable, &tin);
+
+    uint8_t* datamd5 = md5(tout->data, tout->len);
+    if(is_md5_equal(datamd5, p->md5)) {
+        seqs[index]++;
+        p->datalen = tout->len;
+        memcpy(p->data, tout->data, p->datalen);
+        free(datamd5);
+        free(tout->ptr);
+        free(tout);
+        return 1;
+    }
+    free(datamd5);
+    free(tout->ptr);
+    free(tout);
+    return 0;
+}