fumiama/simple-crypto
1
0
Fork 0
mirror of https://github.com/fumiama/simple-crypto.git synced 2026-06-12 06:50:24 +08:00
Code Activity

优化tea

Browse Source
This commit is contained in:
源文雨
2022-04-18 15:22:59 +08:00
parent 20353357b5
commit 2e833f3ab8
3 changed files with 123 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
README.md
View File

@@ -59,10 +59,10 @@ make install
```c ```c
uint8_t* md5(const uint8_t *data, size_t data_len, uint8_t digest[16]); uint8_t* md5(const uint8_t *data, size_t data_len, uint8_t digest[16]);
TEADAT* tea_encrypt_qq(const TEA t[4], const TEADAT* src); int64_t tea_encrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out);
TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt_qq(const TEA t[4], const TEADAT* src); int64_t tea_decrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
``` ```

12
simplecrypto.h
View File

@@ -21,12 +21,12 @@ struct TEADAT {
}; };
typedef struct TEADAT TEADAT; typedef struct TEADAT TEADAT;
TEADAT* tea_encrypt_qq(const TEA t[4], const TEADAT* src); int64_t tea_encrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out);
TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt_qq(const TEA t[4], const TEADAT* src); int64_t tea_decrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
TEADAT* tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src); int64_t tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out);
// ---------------TEA area--------------- // ---------------TEA area---------------

188
tea.c
View File

@@ -28,11 +28,7 @@ const static uint32_t qqsumtable[0x10] = {
0xe3779b90, 0xe3779b90,
}; };
TEADAT* tea_encrypt_qq(const TEA t[4], const TEADAT* src) { int64_t tea_encrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out) {
return tea_encrypt(t, qqsumtable, src);
}
TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src) {
int64_t lens = src->len; int64_t lens = src->len;
int64_t fill = 10 - (lens+1)%8; int64_t fill = 10 - (lens+1)%8;
int64_t dstlen = fill+lens+7; int64_t dstlen = fill+lens+7;
@@ -42,7 +38,6 @@ TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT*
((uint32_t*)dstdat)[2] = rand(); ((uint32_t*)dstdat)[2] = rand();
dstdat[0] = (fill-3)|0xF8; // 存储pad长度 dstdat[0] = (fill-3)|0xF8; // 存储pad长度
memcpy(dstdat+fill, src->data, lens); memcpy(dstdat+fill, src->data, lens);
uint64_t iv1 = 0, iv2 = 0, holder; uint64_t iv1 = 0, iv2 = 0, holder;
for(int64_t i = 0; i < dstlen/8; i++) { for(int64_t i = 0; i < dstlen/8; i++) {
#ifdef WORDS_BIGENDIAN #ifdef WORDS_BIGENDIAN
@@ -51,17 +46,15 @@ TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT*
uint64_t block = __builtin_bswap64(((uint64_t*)dstdat)[i]); uint64_t block = __builtin_bswap64(((uint64_t*)dstdat)[i]);
#endif #endif
holder = block ^ iv1; holder = block ^ iv1;
iv1 = holder; iv1 = holder;
uint32_t v1 = holder; uint32_t v1 = holder;
iv1 >>= 32; iv1 >>= 32;
uint32_t v0 = iv1; uint32_t v0 = iv1;
for (int i = 0; i < 0x10; i++) { for (int i = 0; i < 0x10; i++) {
v0 += (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]); v0 += (v1 + qqsumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]);
v1 += (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]); v1 += (v0 + qqsumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]);
} }
iv1 = ((uint64_t)v0<<32) | (uint64_t)v1; iv1 = ((uint64_t)v0<<32) | (uint64_t)v1;
iv1 = iv1 ^ iv2; iv1 = iv1 ^ iv2;
iv2 = holder; iv2 = holder;
#ifdef WORDS_BIGENDIAN #ifdef WORDS_BIGENDIAN
@@ -70,15 +63,13 @@ TEADAT* tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT*
((uint64_t*)dstdat)[i] = __builtin_bswap64(iv1); ((uint64_t*)dstdat)[i] = __builtin_bswap64(iv1);
#endif #endif
} }
out->len = dstlen;
TEADAT* dst = (TEADAT*)malloc(sizeof(TEADAT)); out->data = dstdat;
dst->len = dstlen; out->ptr = dstdat;
dst->data = dstdat; return dstlen;
dst->ptr = dstdat;
return dst;
} }
TEADAT* tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src) { int64_t tea_encrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out) {
int64_t lens = src->len; int64_t lens = src->len;
int64_t fill = 10 - (lens+1)%8; int64_t fill = 10 - (lens+1)%8;
int64_t dstlen = fill+lens+7; int64_t dstlen = fill+lens+7;
@@ -88,12 +79,14 @@ TEADAT* tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10],
((uint32_t*)dstdat)[2] = rand(); ((uint32_t*)dstdat)[2] = rand();
dstdat[0] = (fill-3)|0xF8; // 存储pad长度 dstdat[0] = (fill-3)|0xF8; // 存储pad长度
memcpy(dstdat+fill, src->data, lens); memcpy(dstdat+fill, src->data, lens);
uint64_t iv1 = 0, iv2 = 0, holder; uint64_t iv1 = 0, iv2 = 0, holder;
for(int64_t i = 0; i < dstlen/8; i++) { for(int64_t i = 0; i < dstlen/8; i++) {
uint64_t block = ((uint64_t*)dstdat)[i]; #ifdef WORDS_BIGENDIAN
uint64_t block = ((uint64_t*)dstdat)[i];
#else
uint64_t block = __builtin_bswap64(((uint64_t*)dstdat)[i]);
#endif
holder = block ^ iv1; holder = block ^ iv1;
iv1 = holder; iv1 = holder;
uint32_t v1 = holder; uint32_t v1 = holder;
iv1 >>= 32; iv1 >>= 32;
@@ -103,29 +96,56 @@ TEADAT* tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10],
v1 += (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]); v1 += (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]);
} }
iv1 = ((uint64_t)v0<<32) | (uint64_t)v1; iv1 = ((uint64_t)v0<<32) | (uint64_t)v1;
iv1 = iv1 ^ iv2;
iv2 = holder;
#ifdef WORDS_BIGENDIAN
((uint64_t*)dstdat)[i] = iv1;
#else
((uint64_t*)dstdat)[i] = __builtin_bswap64(iv1);
#endif
}
out->len = dstlen;
out->data = dstdat;
out->ptr = dstdat;
return dstlen;
}
int64_t tea_encrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out) {
int64_t lens = src->len;
int64_t fill = 10 - (lens+1)%8;
int64_t dstlen = fill+lens+7;
uint8_t* dstdat = (uint8_t*)malloc(dstlen);
((uint32_t*)dstdat)[0] = rand();
((uint32_t*)dstdat)[1] = rand();
((uint32_t*)dstdat)[2] = rand();
dstdat[0] = (fill-3)|0xF8; // 存储pad长度
memcpy(dstdat+fill, src->data, lens);
uint64_t iv1 = 0, iv2 = 0, holder;
for(int64_t i = 0; i < dstlen/8; i++) {
uint64_t block = ((uint64_t*)dstdat)[i];
holder = block ^ iv1;
iv1 = holder;
uint32_t v1 = holder;
iv1 >>= 32;
uint32_t v0 = iv1;
for (int i = 0; i < 0x10; i++) {
v0 += (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]);
v1 += (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]);
}
iv1 = ((uint64_t)v0<<32) | (uint64_t)v1;
iv1 = iv1 ^ iv2; iv1 = iv1 ^ iv2;
iv2 = holder; iv2 = holder;
((uint64_t*)dstdat)[i] = iv1; ((uint64_t*)dstdat)[i] = iv1;
} }
out->len = dstlen;
TEADAT* dst = (TEADAT*)malloc(sizeof(TEADAT)); out->data = dstdat;
dst->len = dstlen; out->ptr = dstdat;
dst->data = dstdat; return dstlen;
dst->ptr = dstdat;
return dst;
} }
TEADAT* tea_decrypt_qq(const TEA t[4], const TEADAT* src) { int64_t tea_decrypt_qq(const TEA t[4], const TEADAT* src, TEADAT* out) {
return tea_decrypt(t, qqsumtable, src); if (src->len < 16 || (src->len)%8 != 0) return 0;
}
TEADAT* tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src) {
if (src->len < 16 || (src->len)%8 != 0) {
return NULL;
}
uint8_t* dstdat = (uint8_t*)malloc(src->len); uint8_t* dstdat = (uint8_t*)malloc(src->len);
uint64_t iv1, iv2 = 0, holder = 0; uint64_t iv1, iv2 = 0, holder = 0;
for(int64_t i = 0; i < src->len/8; i++) { for(int64_t i = 0; i < src->len/8; i++) {
#ifdef WORDS_BIGENDIAN #ifdef WORDS_BIGENDIAN
@@ -133,47 +153,40 @@ TEADAT* tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT*
#else #else
iv1 = __builtin_bswap64(((uint64_t*)(src->data))[i]); iv1 = __builtin_bswap64(((uint64_t*)(src->data))[i]);
#endif #endif
iv2 ^= iv1; iv2 ^= iv1;
uint32_t v1 = iv2; uint32_t v1 = iv2;
iv2 >>= 32; iv2 >>= 32;
uint32_t v0 = iv2; uint32_t v0 = iv2;
for (int i = 0x0f; i >= 0; i--) { for (int i = 0x0f; i >= 0; i--) {
v1 -= (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]); v1 -= (v0 + qqsumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]);
v0 -= (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]); v0 -= (v1 + qqsumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]);
} }
iv2 = ((uint64_t)v0<<32) | (uint64_t)v1; iv2 = ((uint64_t)v0<<32) | (uint64_t)v1;
#ifdef WORDS_BIGENDIAN #ifdef WORDS_BIGENDIAN
((uint64_t*)dstdat)[i] = iv2^holder; ((uint64_t*)dstdat)[i] = iv2^holder;
#else #else
((uint64_t*)dstdat)[i] = __builtin_bswap64(iv2^holder); ((uint64_t*)dstdat)[i] = __builtin_bswap64(iv2^holder);
#endif #endif
holder = iv1; holder = iv1;
} }
TEADAT* dst = (TEADAT*)malloc(sizeof(TEADAT));
int start = (dstdat[0]&7)+3; int start = (dstdat[0]&7)+3;
dst->len = src->len-7-start; out->len = src->len-7-start;
dst->data = dstdat+start; out->data = dstdat+start;
dst->ptr = dstdat; out->ptr = dstdat;
return dst; return out->len;
} }
TEADAT* tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src) { int64_t tea_decrypt(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out) {
if (src->len < 16 || (src->len)%8 != 0) { if (src->len < 16 || (src->len)%8 != 0) return 0;
return NULL;
}
uint8_t* dstdat = (uint8_t*)malloc(src->len); uint8_t* dstdat = (uint8_t*)malloc(src->len);
uint64_t iv1, iv2 = 0, holder = 0; uint64_t iv1, iv2 = 0, holder = 0;
for(int64_t i = 0; i < src->len/8; i++) { for(int64_t i = 0; i < src->len/8; i++) {
iv1 = ((uint64_t*)(src->data))[i]; #ifdef WORDS_BIGENDIAN
iv1 = ((uint64_t*)(src->data))[i];
#else
iv1 = __builtin_bswap64(((uint64_t*)(src->data))[i]);
#endif
iv2 ^= iv1; iv2 ^= iv1;
uint32_t v1 = iv2; uint32_t v1 = iv2;
iv2 >>= 32; iv2 >>= 32;
uint32_t v0 = iv2; uint32_t v0 = iv2;
@@ -182,24 +195,50 @@ TEADAT* tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10],
v0 -= (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]); v0 -= (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]);
} }
iv2 = ((uint64_t)v0<<32) | (uint64_t)v1; iv2 = ((uint64_t)v0<<32) | (uint64_t)v1;
#ifdef WORDS_BIGENDIAN
((uint64_t*)dstdat)[i] = iv2^holder; ((uint64_t*)dstdat)[i] = iv2^holder;
#else
((uint64_t*)dstdat)[i] = __builtin_bswap64(iv2^holder);
#endif
holder = iv1; holder = iv1;
} }
TEADAT* dst = (TEADAT*)malloc(sizeof(TEADAT));
int start = (dstdat[0]&7)+3; int start = (dstdat[0]&7)+3;
dst->len = src->len-7-start; out->len = src->len-7-start;
dst->data = dstdat+start; out->data = dstdat+start;
dst->ptr = dstdat; out->ptr = dstdat;
return dst; return out->len;
}
int64_t tea_decrypt_native_endian(const TEA t[4], const uint32_t sumtable[0x10], const TEADAT* src, TEADAT* out) {
if (src->len < 16 || (src->len)%8 != 0) return 0;
uint8_t* dstdat = (uint8_t*)malloc(src->len);
uint64_t iv1, iv2 = 0, holder = 0;
for(int64_t i = 0; i < src->len/8; i++) {
iv1 = ((uint64_t*)(src->data))[i];
iv2 ^= iv1;
uint32_t v1 = iv2;
iv2 >>= 32;
uint32_t v0 = iv2;
for (int i = 0x0f; i >= 0; i--) {
v1 -= (v0 + sumtable[i]) ^ ((v0 << 4) + t[2]) ^ ((v0 >> 5) + t[3]);
v0 -= (v1 + sumtable[i]) ^ ((v1 << 4) + t[0]) ^ ((v1 >> 5) + t[1]);
}
iv2 = ((uint64_t)v0<<32) | (uint64_t)v1;
((uint64_t*)dstdat)[i] = iv2^holder;
holder = iv1;
}
int start = (dstdat[0]&7)+3;
out->len = src->len-7-start;
out->data = dstdat+start;
out->ptr = dstdat;
return out->len;
} }
#ifdef TEST_SIMPLE_CRYPTO #ifdef TEST_SIMPLE_CRYPTO
int main(int argc, char **argv) { int main(int argc, char **argv) {
TEADAT* td = (TEADAT*)malloc(sizeof(TEADAT)); TEADAT* td = (TEADAT*)malloc(sizeof(TEADAT));
TEA* t = (TEA*)"32107654BA98FEDC"; TEA* t = (TEA*)"32107654BA98FEDC";
TEADAT out;
if (argc != 3) { if (argc != 3) {
printf("usage: %s -[e|d] 'string'\n", argv[0]); printf("usage: %s -[e|d] 'string'\n", argv[0]);
@@ -209,12 +248,11 @@ int main(int argc, char **argv) {
case 'e': case 'e':
td->data = (uint8_t*)(argv[2]); td->data = (uint8_t*)(argv[2]);
td->len = strlen(argv[2]); td->len = strlen(argv[2]);
TEADAT* tde = tea_encrypt_qq(t, td); tea_encrypt_qq(t, td, &out);
// display result // display result
for (int i = 0; i < tde->len; i++) printf("%02x", ((uint8_t*)(tde->data))[i]); for (int i = 0; i < out.len; i++) printf("%02x", ((uint8_t*)(out.data))[i]);
putchar('\n'); putchar('\n');
free(tde->ptr); free(out.ptr);
free(tde);
break; break;
case 'd': case 'd':
td->len = strlen(argv[2])/2; td->len = strlen(argv[2])/2;
@@ -227,22 +265,18 @@ int main(int argc, char **argv) {
td->data[i] = x; td->data[i] = x;
argv[2][i*2] = 0; argv[2][i*2] = 0;
} }
TEADAT* tdd = tea_decrypt_qq(t, td); if (tea_decrypt_qq(t, td, &out)) {
free(td->data); out.data[out.len] = 0;
if (tdd) {
tdd->data[tdd->len] = 0;
// printf("decode output len: %lld\n", tdd->len); // printf("decode output len: %lld\n", tdd->len);
for (int i = 0; i < tdd->len; i++) putchar(tdd->data[i]); for (int i = 0; i < out.len; i++) putchar(out.data[i]);
putchar('\n'); putchar('\n');
free(tdd->ptr); free(out.ptr);
free(tdd);
} else puts("decode error!"); } else puts("decode error!");
free(td->data);
break; break;
default: break; default: break;
} }
free(td); free(td);
return 0; return 0;
} }
#endif #endif