123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254 |
- #include <stdio.h>
- #include <string.h>
- #include <jni.h>
- #include <sys/types.h>
- #include <inttypes.h>
- #include <stdlib.h>
- #include <openssl/aes.h>
- #include <openssl/evp.h>
- #include <unistd.h>
- #include <dirent.h>
- #include <sys/stat.h>
- int registerNativeTgNetFunctions(JavaVM *vm, JNIEnv *env);
- int videoOnJNILoad(JavaVM *vm, JNIEnv *env);
- int imageOnJNILoad(JavaVM *vm, JNIEnv *env);
- int tgvoipOnJNILoad(JavaVM *vm, JNIEnv *env);
- jint JNI_OnLoad(JavaVM *vm, void *reserved) {
- JNIEnv *env = 0;
- srand(time(NULL));
-
- if ((*vm)->GetEnv(vm, (void **) &env, JNI_VERSION_1_6) != JNI_OK) {
- return -1;
- }
- if (imageOnJNILoad(vm, env) != JNI_TRUE) {
- return -1;
- }
- if (videoOnJNILoad(vm, env) != JNI_TRUE) {
- return -1;
- }
- if (registerNativeTgNetFunctions(vm, env) != JNI_TRUE) {
- return -1;
- }
- tgvoipOnJNILoad(vm, env);
- return JNI_VERSION_1_6;
- }
- void JNI_OnUnload(JavaVM *vm, void *reserved) {
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesIgeEncryption(JNIEnv *env, jclass class, jobject buffer, jbyteArray key, jbyteArray iv, jboolean encrypt, jint offset, jint length) {
- jbyte *what = (*env)->GetDirectBufferAddress(env, buffer) + offset;
- unsigned char *keyBuff = (unsigned char *)(*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *)(*env)->GetByteArrayElements(env, iv, NULL);
-
- AES_KEY akey;
- if (!encrypt) {
- AES_set_decrypt_key(keyBuff, 32 * 8, &akey);
- AES_ige_encrypt(what, what, length, &akey, ivBuff, AES_DECRYPT);
- } else {
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- AES_ige_encrypt(what, what, length, &akey, ivBuff, AES_ENCRYPT);
- }
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, 0);
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesIgeEncryptionByteArray(JNIEnv *env, jclass class, jbyteArray buffer, jbyteArray key, jbyteArray iv, jboolean encrypt, jint offset, jint length) {
- unsigned char *bufferBuff = (unsigned char *) (*env)->GetByteArrayElements(env, buffer, NULL);
- unsigned char *keyBuff = (unsigned char *) (*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *) (*env)->GetByteArrayElements(env, iv, NULL);
- AES_KEY akey;
- if (!encrypt) {
- AES_set_decrypt_key(keyBuff, 32 * 8, &akey);
- AES_ige_encrypt(bufferBuff, bufferBuff, length, &akey, ivBuff, AES_DECRYPT);
- } else {
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- AES_ige_encrypt(bufferBuff, bufferBuff, length, &akey, ivBuff, AES_ENCRYPT);
- }
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, 0);
- (*env)->ReleaseByteArrayElements(env, buffer, bufferBuff, 0);
- }
- JNIEXPORT jint Java_org_telegram_messenger_Utilities_pbkdf2(JNIEnv *env, jclass class, jbyteArray password, jbyteArray salt, jbyteArray dst, jint iterations) {
- jbyte *passwordBuff = (*env)->GetByteArrayElements(env, password, NULL);
- size_t passwordLength = (size_t) (*env)->GetArrayLength(env, password);
- jbyte *saltBuff = (*env)->GetByteArrayElements(env, salt, NULL);
- size_t saltLength = (size_t) (*env)->GetArrayLength(env, salt);
- jbyte *dstBuff = (*env)->GetByteArrayElements(env, dst, NULL);
- size_t dstLength = (size_t) (*env)->GetArrayLength(env, dst);
- int result = PKCS5_PBKDF2_HMAC((char *) passwordBuff, passwordLength, (uint8_t *) saltBuff, saltLength, (unsigned int) iterations, EVP_sha512(), dstLength, (uint8_t *) dstBuff);
- (*env)->ReleaseByteArrayElements(env, password, passwordBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, salt, saltBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, dst, dstBuff, 0);
- return result;
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesCtrDecryption(JNIEnv *env, jclass class, jobject buffer, jbyteArray key, jbyteArray iv, jint offset, jint length) {
- jbyte *what = (*env)->GetDirectBufferAddress(env, buffer) + offset;
- unsigned char *keyBuff = (unsigned char *)(*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *)(*env)->GetByteArrayElements(env, iv, NULL);
- AES_KEY akey;
- unsigned int num = 0;
- uint8_t count[16];
- memset(count, 0, 16);
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- AES_ctr128_encrypt(what, what, length, &akey, ivBuff, count, &num);
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, JNI_ABORT);
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesCtrDecryptionByteArray(JNIEnv *env, jclass class, jbyteArray buffer, jbyteArray key, jbyteArray iv, jint offset, jint length, jint fileOffset) {
- unsigned char *bufferBuff = (unsigned char *) (*env)->GetByteArrayElements(env, buffer, NULL);
- unsigned char *keyBuff = (unsigned char *) (*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *) (*env)->GetByteArrayElements(env, iv, NULL);
- AES_KEY akey;
- uint8_t count[16];
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- unsigned int num = (unsigned int) (fileOffset % 16);
- int o = fileOffset / 16;
- ivBuff[15] = (uint8_t) (o & 0xff);
- ivBuff[14] = (uint8_t) ((o >> 8) & 0xff);
- ivBuff[13] = (uint8_t) ((o >> 16) & 0xff);
- ivBuff[12] = (uint8_t) ((o >> 24) & 0xff);
- AES_encrypt(ivBuff, count, &akey);
- o = (fileOffset + 15) / 16;
- ivBuff[15] = (uint8_t) (o & 0xff);
- ivBuff[14] = (uint8_t) ((o >> 8) & 0xff);
- ivBuff[13] = (uint8_t) ((o >> 16) & 0xff);
- ivBuff[12] = (uint8_t) ((o >> 24) & 0xff);
- AES_ctr128_encrypt(bufferBuff + offset, bufferBuff + offset, length, &akey, ivBuff, count, &num);
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, buffer, bufferBuff, 0);
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesCbcEncryptionByteArray(JNIEnv *env, jclass class, jbyteArray buffer, jbyteArray key, jbyteArray iv, jint offset, jint length, jint fileOffset, jint encrypt) {
- unsigned char *bufferBuff = (unsigned char *) (*env)->GetByteArrayElements(env, buffer, NULL);
- unsigned char *keyBuff = (unsigned char *) (*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *) (*env)->GetByteArrayElements(env, iv, NULL);
- AES_KEY akey;
- if (encrypt) {
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- } else {
- AES_set_decrypt_key(keyBuff, 32 * 8, &akey);
- if (fileOffset != 0) {
- int o = (fileOffset + 15) / 16;
- ivBuff[15] = (uint8_t) (o & 0xff);
- ivBuff[14] = (uint8_t) ((o >> 8) & 0xff);
- ivBuff[13] = (uint8_t) ((o >> 16) & 0xff);
- ivBuff[12] = (uint8_t) ((o >> 24) & 0xff);
- }
- }
- AES_cbc_encrypt(bufferBuff, bufferBuff, length, &akey, ivBuff, encrypt);
- (*env)->ReleaseByteArrayElements(env, buffer, bufferBuff, 0);
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, JNI_ABORT);
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_aesCbcEncryption(JNIEnv *env, jclass class, jobject buffer, jbyteArray key, jbyteArray iv, jint offset, jint length, jint encrypt) {
- unsigned char *bufferBuff = (*env)->GetDirectBufferAddress(env, buffer) + offset;
- unsigned char *keyBuff = (unsigned char *) (*env)->GetByteArrayElements(env, key, NULL);
- unsigned char *ivBuff = (unsigned char *) (*env)->GetByteArrayElements(env, iv, NULL);
- AES_KEY akey;
- if (encrypt) {
- AES_set_encrypt_key(keyBuff, 32 * 8, &akey);
- } else {
- AES_set_decrypt_key(keyBuff, 32 * 8, &akey);
- }
- AES_cbc_encrypt(bufferBuff + offset, bufferBuff + offset, length, &akey, ivBuff, encrypt);
- (*env)->ReleaseByteArrayElements(env, key, keyBuff, JNI_ABORT);
- (*env)->ReleaseByteArrayElements(env, iv, ivBuff, JNI_ABORT);
- }
- int64_t listdir(const char *fileName, int32_t mode, int32_t docType, int64_t time, uint8_t subdirs) {
- int64_t value = 0;
- DIR *dir;
- struct stat attrib;
- if ((dir = opendir(fileName)) != NULL) {
- char buff[4096];
- struct dirent *entry;
- while ((entry = readdir(dir)) != NULL) {
- char *name = entry->d_name;
- size_t len = strlen(name);
- if (name[0] == '.') {
- continue;
- }
- if ((docType == 1 || docType == 2) && len > 4) {
- if (name[len - 4] == '.' && (
- ((name[len - 3] == 'm' || name[len - 3] == 'M') && (name[len - 2] == 'p' || name[len - 2] == 'P') && (name[len - 1] == '3')) ||
- ((name[len - 3] == 'm' || name[len - 3] == 'M') && (name[len - 2] == '4') && (name[len - 1] == 'a' || name[len - 1] == 'A'))
- )) {
- if (docType == 1) {
- continue;
- }
- } else if (docType == 2) {
- continue;
- }
- }
- strncpy(buff, fileName, 4095);
- strncat(buff, "/", 4095);
- strncat(buff, entry->d_name, 4095);
- if (entry->d_type == DT_DIR) {
- if (subdirs) {
- value += listdir(buff, mode, docType, time, subdirs);
- }
- } else {
- stat(buff, &attrib);
- if (mode == 0) {
- value += 512 * attrib.st_blocks;
- } else if (mode == 1) {
- if (attrib.st_atim.tv_sec != 0) {
- if (attrib.st_atim.tv_sec < time) {
- remove(buff);
- }
- } else {
- if (attrib.st_mtim.tv_sec < time) {
- remove(buff);
- }
- }
- }
- }
- }
- closedir(dir);
- }
- return value;
- }
- JNIEXPORT jlong Java_org_telegram_messenger_Utilities_getDirSize(JNIEnv *env, jclass class, jstring path, jint docType, jboolean subdirs) {
- const char *fileName = (*env)->GetStringUTFChars(env, path, NULL);
- jlong value = listdir(fileName, 0, docType, 0, subdirs);
- (*env)->ReleaseStringUTFChars(env, path, fileName);
- return value;
- }
- JNIEXPORT void Java_org_telegram_messenger_Utilities_clearDir(JNIEnv *env, jclass class, jstring path, jint docType, jlong time, jboolean subdirs) {
- const char *fileName = (*env)->GetStringUTFChars(env, path, NULL);
- listdir(fileName, 1, docType, time, subdirs);
- (*env)->ReleaseStringUTFChars(env, path, fileName);
- }
|