summaryrefslogtreecommitdiff
path: root/android/sqlite3_android.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'android/sqlite3_android.cpp')
-rw-r--r--android/sqlite3_android.cpp396
1 files changed, 0 insertions, 396 deletions
diff --git a/android/sqlite3_android.cpp b/android/sqlite3_android.cpp
deleted file mode 100644
index 1723c1e..0000000
--- a/android/sqlite3_android.cpp
+++ /dev/null
@@ -1,396 +0,0 @@
-/*
- * Copyright (C) 2007 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#define LOG_TAG "sqlite3_android"
-
-#include <ctype.h>
-#include <stdlib.h>
-#include <string.h>
-#include <unistd.h>
-
-#include <unicode/ucol.h>
-#include <unicode/ustring.h>
-#include <cutils/log.h>
-
-#include "sqlite3_android.h"
-#include "PhoneNumberUtils.h"
-
-#define ENABLE_ANDROID_LOG 0
-
-static int collate16(void *p, int n1, const void *v1, int n2, const void *v2)
-{
- UCollator *coll = (UCollator *) p;
- UCollationResult result = ucol_strcoll(coll, (const UChar *) v1, n1,
- (const UChar *) v2, n2);
-
- if (result == UCOL_LESS) {
- return -1;
- } else if (result == UCOL_GREATER) {
- return 1;
- } else {
- return 0;
- }
-}
-
-static int collate8(void *p, int n1, const void *v1, int n2, const void *v2)
-{
- UCollator *coll = (UCollator *) p;
- UCharIterator i1, i2;
- UErrorCode status = U_ZERO_ERROR;
-
- uiter_setUTF8(&i1, (const char *) v1, n1);
- uiter_setUTF8(&i2, (const char *) v2, n2);
-
- UCollationResult result = ucol_strcollIter(coll, &i1, &i2, &status);
-
- if (U_FAILURE(status)) {
-// LOGE("Collation iterator error: %d\n", status);
- }
-
- if (result == UCOL_LESS) {
- return -1;
- } else if (result == UCOL_GREATER) {
- return 1;
- } else {
- return 0;
- }
-}
-
-static void phone_numbers_equal(sqlite3_context * context, int argc, sqlite3_value ** argv)
-{
- if (argc != 2) {
- sqlite3_result_int(context, 0);
- return;
- }
-
- char const * num1 = (char const *)sqlite3_value_text(argv[0]);
- char const * num2 = (char const *)sqlite3_value_text(argv[1]);
-
- if (num1 == NULL || num2 == NULL) {
- sqlite3_result_null(context);
- return;
- }
-
- bool equal = android::phone_number_compare(num1, num2);
-
- if (equal) {
- sqlite3_result_int(context, 1);
- } else {
- sqlite3_result_int(context, 0);
- }
-}
-
-#if ENABLE_ANDROID_LOG
-static void android_log(sqlite3_context * context, int argc, sqlite3_value ** argv)
-{
- char const * tag = "sqlite_trigger";
- char const * msg = "";
- int msgIndex = 0;
-
- switch (argc) {
- case 2:
- tag = (char const *)sqlite3_value_text(argv[0]);
- if (tag == NULL) {
- tag = "sqlite_trigger";
- }
- msgIndex = 1;
- case 1:
- msg = (char const *)sqlite3_value_text(argv[msgIndex]);
- if (msg == NULL) {
- msg = "";
- }
- LOG(LOG_INFO, tag, msg);
- sqlite3_result_int(context, 1);
- return;
-
- default:
- sqlite3_result_int(context, 0);
- return;
- }
-}
-#endif
-
-static void delete_file(sqlite3_context * context, int argc, sqlite3_value ** argv)
-{
- if (argc != 1) {
- sqlite3_result_int(context, 0);
- return;
- }
-
- char const * path = (char const *)sqlite3_value_text(argv[0]);
- if (path == NULL) {
- sqlite3_result_null(context);
- return;
- }
-
- if (strncmp("/sdcard/", path, 8) != 0) {
- sqlite3_result_null(context);
- return;
- }
-
- int err = unlink(path);
- if (err != -1) {
- // No error occured, return true
- sqlite3_result_int(context, 1);
- } else {
- // An error occured, return false
- sqlite3_result_int(context, 0);
- }
-}
-
-static void tokenize_auxdata_delete(void * data)
-{
- sqlite3_stmt * statement = (sqlite3_stmt *)data;
- sqlite3_finalize(statement);
-}
-
-static void base16Encode(char* dest, const char* src, uint32_t size)
-{
- static const char * BASE16_TABLE = "0123456789abcdef";
- for (uint32_t i = 0; i < size; i++) {
- char ch = *src++;
- *dest++ = BASE16_TABLE[ (ch & 0xf0) >> 4 ];
- *dest++ = BASE16_TABLE[ (ch & 0x0f) ];
- }
-}
-
-struct SqliteUserData {
- sqlite3 * handle;
- UCollator* collator;
-};
-
-/**
- * This function is invoked as:
- *
- * _TOKENIZE('<token_table>', <data_row_id>, <data>, <delimiter>)
- *
- * It will then split data on each instance of delimiter and insert each token
- * into token_table's 'token' column with data_row_id in the 'source' column.
- * The function returns the number of tokens generated.
- */
-static void tokenize(sqlite3_context * context, int argc, sqlite3_value ** argv)
-{
- //LOGD("enter tokenize");
- int err;
-
- if (argc != 4) {
- LOGE("Tokenize requires 4 arguments");
- sqlite3_result_null(context);
- return;
- }
-
- sqlite3 * handle = sqlite3_context_db_handle(context);
- UCollator* collator = (UCollator*)sqlite3_user_data(context);
- char const * tokenTable = (char const *)sqlite3_value_text(argv[0]);
- if (tokenTable == NULL) {
- LOGE("tokenTable null");
- sqlite3_result_null(context);
- return;
- }
-
- // Get or create the prepared statement for the insertions
- sqlite3_stmt * statement = (sqlite3_stmt *)sqlite3_get_auxdata(context, 0);
- if (!statement) {
- char * sql = sqlite3_mprintf("INSERT INTO %s (token, source) VALUES (?, ?);", tokenTable);
- err = sqlite3_prepare_v2(handle, sql, -1, &statement, NULL);
- sqlite3_free(sql);
- if (err) {
- LOGE("prepare failed");
- sqlite3_result_null(context);
- return;
- }
- // This binds the statement to the table it was compiled against, which is argv[0].
- // If this function is ever called with a different table the finalizer will be called
- // and sqlite3_get_auxdata() will return null above, forcing a recompile for the new table.
- sqlite3_set_auxdata(context, 0, statement, tokenize_auxdata_delete);
- } else {
- // Reset the cached statement so that binding the row ID will work properly
- sqlite3_reset(statement);
- }
-
- // Bind the row ID of the source row
- int64_t rowID = sqlite3_value_int64(argv[1]);
- err = sqlite3_bind_int64(statement, 2, rowID);
- if (err != SQLITE_OK) {
- LOGE("bind failed");
- sqlite3_result_null(context);
- return;
- }
-
- // Get the raw bytes for the string to tokenize
- // the string will be modified by following code
- // however, sqlite did not reuse the string, so it is safe to not dup it
- UChar * origData = (UChar *)sqlite3_value_text16(argv[2]);
- if (origData == NULL) {
- sqlite3_result_null(context);
- return;
- }
-
- // Get the raw bytes for the delimiter
- const UChar * delim = (const UChar *)sqlite3_value_text16(argv[3]);
- if (delim == NULL) {
- LOGE("can't get delimiter");
- sqlite3_result_null(context);
- return;
- }
-
- UChar * token = NULL;
- UChar *state;
- int numTokens = 0;
-
- do {
- if (numTokens == 0) {
- token = origData;
- }
-
- // Reset the program so we can use it to perform the insert
- sqlite3_reset(statement);
- UErrorCode status = U_ZERO_ERROR;
- char keybuf[1024];
- uint32_t result = ucol_getSortKey(collator, token, -1, (uint8_t*)keybuf, sizeof(keybuf)-1);
- if (result > sizeof(keybuf)) {
- // TODO allocate memory for this super big string
- LOGE("ucol_getSortKey needs bigger buffer %d", result);
- break;
- }
- uint32_t keysize = result-1;
- uint32_t base16Size = keysize*2;
- char *base16buf = (char*)malloc(base16Size);
- base16Encode(base16buf, keybuf, keysize);
- err = sqlite3_bind_text(statement, 1, base16buf, base16Size, SQLITE_STATIC);
-
- if (err != SQLITE_OK) {
- LOGE(" sqlite3_bind_text16 error %d", err);
- free(base16buf);
- break;
- }
-
- err = sqlite3_step(statement);
- free(base16buf);
-
- if (err != SQLITE_DONE) {
- LOGE(" sqlite3_step error %d", err);
- break;
- }
- numTokens++;
- if (numTokens == 1) {
- // first call
- u_strtok_r(origData, delim, &state);
- }
- } while ((token = u_strtok_r(NULL, delim, &state)) != NULL);
- sqlite3_result_int(context, numTokens);
-}
-
-static void localized_collator_dtor(UCollator* collator)
-{
- ucol_close(collator);
-}
-
-#define LOCALIZED_COLLATOR_NAME "LOCALIZED"
-
-extern "C" int register_localized_collators(sqlite3* handle, const char* systemLocale, int utf16Storage)
-{
- int err;
- UErrorCode status = U_ZERO_ERROR;
- void* icudata;
-
- UCollator* collator = ucol_open(systemLocale, &status);
- if (U_FAILURE(status)) {
- return -1;
- }
-
- ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_PRIMARY, &status);
- if (U_FAILURE(status)) {
- return -1;
- }
-
- status = U_ZERO_ERROR;
- char buf[1024];
- int n = ucol_getShortDefinitionString(collator, NULL, buf, 1024, &status);
-
- if (utf16Storage) {
- err = sqlite3_create_collation_v2(handle, LOCALIZED_COLLATOR_NAME, SQLITE_UTF16, collator,
- collate16, (void(*)(void*))localized_collator_dtor);
- } else {
- err = sqlite3_create_collation_v2(handle, LOCALIZED_COLLATOR_NAME, SQLITE_UTF8, collator,
- collate8, (void(*)(void*))localized_collator_dtor);
- }
- if (err != SQLITE_OK) {
- return err;
- }
-
- // Register the _TOKENIZE function
- err = sqlite3_create_function(handle, "_TOKENIZE", 4, SQLITE_UTF16, collator, tokenize, NULL, NULL);
- if (err != SQLITE_OK) {
- return err;
- }
-
- return SQLITE_OK;
-}
-
-
-extern "C" int register_android_functions(sqlite3 * handle, int utf16Storage)
-{
- int err;
- UErrorCode status = U_ZERO_ERROR;
-
- UCollator * collator = ucol_open(NULL, &status);
- if (U_FAILURE(status)) {
- return -1;
- }
-
- if (utf16Storage) {
- // Note that text should be stored as UTF-16
- err = sqlite3_exec(handle, "PRAGMA encoding = 'UTF-16'", 0, 0, 0);
- if (err != SQLITE_OK) {
- return err;
- }
-
- // Register the UNICODE collation
- err = sqlite3_create_collation_v2(handle, "UNICODE", SQLITE_UTF16, collator, collate16,
- (void(*)(void*))localized_collator_dtor);
- } else {
- err = sqlite3_create_collation_v2(handle, "UNICODE", SQLITE_UTF8, collator, collate8,
- (void(*)(void*))localized_collator_dtor);
- }
-
- if (err != SQLITE_OK) {
- return err;
- }
-
- // Register the PHONE_NUM_EQUALS function
- err = sqlite3_create_function(handle, "PHONE_NUMBERS_EQUAL", 2, SQLITE_UTF8, NULL, phone_numbers_equal, NULL, NULL);
- if (err != SQLITE_OK) {
- return err;
- }
-
- // Register the _DELETE_FILE function
- err = sqlite3_create_function(handle, "_DELETE_FILE", 1, SQLITE_UTF8, NULL, delete_file, NULL, NULL);
- if (err != SQLITE_OK) {
- return err;
- }
-
-#if ENABLE_ANDROID_LOG
- // Register the _LOG function
- err = sqlite3_create_function(handle, "_LOG", 1, SQLITE_UTF8, NULL, android_log, NULL, NULL);
- if (err != SQLITE_OK) {
- return err;
- }
-#endif
-
- return SQLITE_OK;
-}
-