[sqlite] Backport icuCaseFunc16 patch from SQLite.

third_party/sqlite/src/ext/icu/icu.c

 /*
 ** 2007 May 6
 **
 ** The author disclaims copyright to this source code.  In place of
 ** a legal notice, here is a blessing:
 **
 **    May you do good and not evil.
 **    May you find forgiveness for yourself and forgive others.
 **    May you share freely, never taking more than you give.
 **
@@ skipping 307 matching lines @@
 ** as the second argument is the same as invoking the 1 argument version
 ** of upper() or lower().
 **
 **     lower('I', 'en_us') -> 'i'
 **     lower('I', 'tr_tr') -> 'ı' (small dotless i)
 **
 ** http://www.icu-project.org/userguide/posix.html#case_mappings
 */
 static void icuCaseFunc16(sqlite3_context *p, int nArg, sqlite3_value **apArg){
   const UChar *zInput;
-  UChar *zOutput;
+  UChar *zOutput = 0;
   int nInput;
-  int nOutput;
-
-  UErrorCode status = U_ZERO_ERROR;
+  int nOut;
+  int cnt;
+  UErrorCode status;
   const char *zLocale = 0;
 
   assert(nArg==1 || nArg==2);
   if( nArg==2 ){
     zLocale = (const char *)sqlite3_value_text(apArg[1]);
   }
 
   zInput = sqlite3_value_text16(apArg[0]);
   if( !zInput ){
     return;
   }
-  nOutput = nInput = sqlite3_value_bytes16(apArg[0]);
-
-  zOutput = sqlite3_malloc(nOutput);
-  if( !zOutput ){
+  nOut = nInput = sqlite3_value_bytes16(apArg[0]);
+  if( nOut==0 ){
+    sqlite3_result_text16(p, "", 0, SQLITE_STATIC);
     return;
   }
 
-  if( sqlite3_user_data(p) ){
-    nOutput = u_strToUpper(
-        zOutput, nOutput/2, zInput, nInput/2, zLocale, &status) * 2;
-  }else{
-    nOutput = u_strToLower(
-        zOutput, nOutput/2, zInput, nInput/2, zLocale, &status) * 2;
-  }
-
-  if ( status == U_BUFFER_OVERFLOW_ERROR ) {
-    UChar* newOutput = sqlite3_realloc(zOutput, nOutput);
-    if( !newOutput ){
+  for(cnt=0; cnt<2; cnt++){
+    UChar *zNew = sqlite3_realloc(zOutput, nOut);
+    if( zNew==0 ){
       sqlite3_free(zOutput);
+      sqlite3_result_error_nomem(p);
       return;
     }
-    zOutput = newOutput;
+    zOutput = zNew;
     status = U_ZERO_ERROR;
     if( sqlite3_user_data(p) ){
-      nOutput = u_strToUpper(
-          zOutput, nOutput/2, zInput, nInput/2, zLocale, &status) * 2;
+      nOut = 2*u_strToUpper(zOutput,nOut/2,zInput,nInput/2,zLocale,&status);
     }else{
-      nOutput = u_strToLower(
-          zOutput, nOutput/2, zInput, nInput/2, zLocale, &status) * 2;
+      nOut = 2*u_strToLower(zOutput,nOut/2,zInput,nInput/2,zLocale,&status);
+    }
+    if( !U_SUCCESS(status) ){
+      if( status==U_BUFFER_OVERFLOW_ERROR ) continue;
+      icuFunctionError(p,
+          sqlite3_user_data(p) ? "u_strToUpper()" : "u_strToLower", status);
+      return;
     }
   }
-
-  if( U_FAILURE(status) ){
-    icuFunctionError(p, "u_strToLower()/u_strToUpper", status);
-    sqlite3_free(zOutput);
-    return;
-  }
-
-  sqlite3_result_text16(p, zOutput, nOutput, xFree);
+  sqlite3_result_text16(p, zOutput, nOut, xFree);
 }
 
 /*
 ** Collation sequence destructor function. The pCtx argument points to
 ** a UCollator structure previously allocated using ucol_open().
 */
 static void icuCollationDel(void *pCtx){
   UCollator *p = (UCollator *)pCtx;
   ucol_close(p);
 }
@@ skipping 121 matching lines @@
   sqlite3 *db, 
   char **pzErrMsg,
   const sqlite3_api_routines *pApi
 ){
   SQLITE_EXTENSION_INIT2(pApi)
   return sqlite3IcuInit(db);
 }
 #endif
 
 #endif