mirror of
https://github.com/DrKLO/Telegram.git
synced 2024-12-22 22:45:18 +01:00
241 lines
8.8 KiB
C++
Executable file
241 lines
8.8 KiB
C++
Executable file
#include "sqlite/sqlite3.h"
|
|
#include "tgnet/NativeByteBuffer.h"
|
|
#include "tgnet/BuffersStorage.h"
|
|
|
|
void throw_sqlite3_exception(JNIEnv *env, sqlite3 *handle, int errcode) {
|
|
if (SQLITE_OK == errcode) {
|
|
errcode = sqlite3_errcode(handle);
|
|
}
|
|
const char *errmsg = sqlite3_errmsg(handle);
|
|
jclass exClass = env->FindClass("org/telegram/SQLite/SQLiteException");
|
|
env->ThrowNew(exClass, errmsg);
|
|
}
|
|
|
|
extern "C" {
|
|
|
|
int Java_org_telegram_SQLite_SQLitePreparedStatement_step(JNIEnv *env, jobject object, int statementHandle) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_step(handle);
|
|
if (errcode == SQLITE_ROW) {
|
|
return 0;
|
|
} else if(errcode == SQLITE_DONE) {
|
|
return 1;
|
|
} else if(errcode == SQLITE_BUSY) {
|
|
return -1;
|
|
}
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLitePreparedStatement_prepare(JNIEnv *env, jobject object, int sqliteHandle, jstring sql) {
|
|
sqlite3 *handle = (sqlite3 *) sqliteHandle;
|
|
|
|
char const *sqlStr = env->GetStringUTFChars(sql, 0);
|
|
|
|
sqlite3_stmt *stmt_handle;
|
|
|
|
int errcode = sqlite3_prepare_v2(handle, sqlStr, -1, &stmt_handle, 0);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, handle, errcode);
|
|
}
|
|
|
|
if (sqlStr != 0) {
|
|
env->ReleaseStringUTFChars(sql, sqlStr);
|
|
}
|
|
|
|
return (int) stmt_handle;
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_reset(JNIEnv *env, jobject object, int statementHandle) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_reset(handle);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_finalize(JNIEnv *env, jobject object, int statementHandle) {
|
|
sqlite3_finalize((sqlite3_stmt *) statementHandle);
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindByteBuffer(JNIEnv *env, jobject object, int statementHandle, int index, jobject value, int length) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
void *buf = env->GetDirectBufferAddress(value);
|
|
|
|
int errcode = sqlite3_bind_blob(handle, index, buf, length, SQLITE_STATIC);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindString(JNIEnv *env, jobject object, int statementHandle, int index, jstring value) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
char const *valueStr = env->GetStringUTFChars(value, 0);
|
|
|
|
int errcode = sqlite3_bind_text(handle, index, valueStr, -1, SQLITE_TRANSIENT);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
|
|
if (valueStr != 0) {
|
|
env->ReleaseStringUTFChars(value, valueStr);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindInt(JNIEnv *env, jobject object, int statementHandle, int index, int value) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_bind_int(handle, index, value);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindLong(JNIEnv *env, jobject object, int statementHandle, int index, long long value) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_bind_int64(handle, index, value);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindDouble(JNIEnv *env, jobject object, int statementHandle, int index, double value) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_bind_double(handle, index, value);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLitePreparedStatement_bindNull(JNIEnv *env, jobject object, int statementHandle, int index) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
|
|
int errcode = sqlite3_bind_null(handle, index);
|
|
if (SQLITE_OK != errcode) {
|
|
throw_sqlite3_exception(env, sqlite3_db_handle(handle), errcode);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLiteDatabase_closedb(JNIEnv *env, jobject object, int sqliteHandle) {
|
|
sqlite3 *handle = (sqlite3 *)sqliteHandle;
|
|
int err = sqlite3_close(handle);
|
|
if (SQLITE_OK != err) {
|
|
throw_sqlite3_exception(env, handle, err);
|
|
}
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLiteDatabase_beginTransaction(JNIEnv *env, jobject object, int sqliteHandle) {
|
|
sqlite3 *handle = (sqlite3 *)sqliteHandle;
|
|
sqlite3_exec(handle, "BEGIN", 0, 0, 0);
|
|
}
|
|
|
|
void Java_org_telegram_SQLite_SQLiteDatabase_commitTransaction(JNIEnv *env, jobject object, int sqliteHandle) {
|
|
sqlite3 *handle = (sqlite3 *)sqliteHandle;
|
|
sqlite3_exec(handle, "COMMIT", 0, 0, 0);
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLiteDatabase_opendb(JNIEnv *env, jobject object, jstring fileName, jstring tempDir) {
|
|
char const *fileNameStr = env->GetStringUTFChars(fileName, 0);
|
|
char const *tempDirStr = env->GetStringUTFChars(tempDir, 0);
|
|
|
|
if (sqlite3_temp_directory != 0) {
|
|
sqlite3_free(sqlite3_temp_directory);
|
|
}
|
|
sqlite3_temp_directory = sqlite3_mprintf("%s", tempDirStr);
|
|
|
|
sqlite3 *handle = 0;
|
|
int err = sqlite3_open(fileNameStr, &handle);
|
|
if (SQLITE_OK != err) {
|
|
throw_sqlite3_exception(env, handle, err);
|
|
}
|
|
if (fileNameStr != 0) {
|
|
env->ReleaseStringUTFChars(fileName, fileNameStr);
|
|
}
|
|
if (tempDirStr != 0) {
|
|
env->ReleaseStringUTFChars(tempDir, tempDirStr);
|
|
}
|
|
return (int)handle;
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLiteCursor_columnType(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
return sqlite3_column_type(handle, columnIndex);
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLiteCursor_columnIsNull(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
int valType = sqlite3_column_type(handle, columnIndex);
|
|
return SQLITE_NULL == valType;
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLiteCursor_columnIntValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
int valType = sqlite3_column_type(handle, columnIndex);
|
|
if (SQLITE_NULL == valType) {
|
|
return 0;
|
|
}
|
|
return sqlite3_column_int(handle, columnIndex);
|
|
}
|
|
|
|
long long Java_org_telegram_SQLite_SQLiteCursor_columnLongValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
int valType = sqlite3_column_type(handle, columnIndex);
|
|
if (SQLITE_NULL == valType) {
|
|
return 0;
|
|
}
|
|
return sqlite3_column_int64(handle, columnIndex);
|
|
}
|
|
|
|
double Java_org_telegram_SQLite_SQLiteCursor_columnDoubleValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
int valType = sqlite3_column_type(handle, columnIndex);
|
|
if (SQLITE_NULL == valType) {
|
|
return 0;
|
|
}
|
|
return sqlite3_column_double(handle, columnIndex);
|
|
}
|
|
|
|
jstring Java_org_telegram_SQLite_SQLiteCursor_columnStringValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
const char *str = (const char *) sqlite3_column_text(handle, columnIndex);
|
|
if (str != 0) {
|
|
return env->NewStringUTF(str);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
jbyteArray Java_org_telegram_SQLite_SQLiteCursor_columnByteArrayValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *)statementHandle;
|
|
const jbyte *buf = (const jbyte *) sqlite3_column_blob(handle, columnIndex);
|
|
int length = sqlite3_column_bytes(handle, columnIndex);
|
|
if (buf != nullptr && length > 0) {
|
|
jbyteArray result = env->NewByteArray(length);
|
|
env->SetByteArrayRegion(result, 0, length, buf);
|
|
return result;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
int Java_org_telegram_SQLite_SQLiteCursor_columnByteBufferValue(JNIEnv *env, jobject object, int statementHandle, int columnIndex) {
|
|
sqlite3_stmt *handle = (sqlite3_stmt *) statementHandle;
|
|
int length = sqlite3_column_bytes(handle, columnIndex);
|
|
if (length <= 0) {
|
|
return 0;
|
|
}
|
|
NativeByteBuffer *buffer = BuffersStorage::getInstance().getFreeBuffer(length);
|
|
if (buffer == nullptr) {
|
|
return 0;
|
|
}
|
|
const char *buf = (const char *) sqlite3_column_blob(handle, columnIndex);
|
|
if (buf == nullptr) {
|
|
return 0;
|
|
}
|
|
memcpy(buffer->bytes(), buf, length);
|
|
return (int) buffer;
|
|
}
|
|
|
|
}
|