dvander@1203: /*
dvander@1203: ** 2005 May 23 
dvander@1203: **
dvander@1203: ** The author disclaims copyright to this source code.  In place of
dvander@1203: ** a legal notice, here is a blessing:
dvander@1203: **
dvander@1203: **    May you do good and not evil.
dvander@1203: **    May you find forgiveness for yourself and forgive others.
dvander@1203: **    May you share freely, never taking more than you give.
dvander@1203: **
dvander@1203: *************************************************************************
dvander@1203: **
dvander@1203: ** This file contains functions used to access the internal hash tables
dvander@1203: ** of user defined functions and collation sequences.
dvander@1203: **
dvander@1203: ** $Id$
dvander@1203: */
dvander@1203: 
dvander@1203: #include "sqliteInt.h"
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Invoke the 'collation needed' callback to request a collation sequence
dvander@1203: ** in the database text encoding of name zName, length nName.
dvander@1203: ** If the collation sequence
dvander@1203: */
dvander@1203: static void callCollNeeded(sqlite3 *db, const char *zName, int nName){
dvander@1203:   assert( !db->xCollNeeded || !db->xCollNeeded16 );
dvander@1203:   if( nName<0 ) nName = strlen(zName);
dvander@1203:   if( db->xCollNeeded ){
dvander@1645:     char *zExternal = sqlite3DbStrNDup(db, zName, nName);
dvander@1203:     if( !zExternal ) return;
dvander@1203:     db->xCollNeeded(db->pCollNeededArg, db, (int)ENC(db), zExternal);
dvander@1645:     sqlite3_free(zExternal);
dvander@1203:   }
dvander@1203: #ifndef SQLITE_OMIT_UTF16
dvander@1203:   if( db->xCollNeeded16 ){
dvander@1203:     char const *zExternal;
dvander@1645:     sqlite3_value *pTmp = sqlite3ValueNew(db);
dvander@1203:     sqlite3ValueSetStr(pTmp, nName, zName, SQLITE_UTF8, SQLITE_STATIC);
dvander@1203:     zExternal = sqlite3ValueText(pTmp, SQLITE_UTF16NATIVE);
dvander@1203:     if( zExternal ){
dvander@1203:       db->xCollNeeded16(db->pCollNeededArg, db, (int)ENC(db), zExternal);
dvander@1203:     }
dvander@1203:     sqlite3ValueFree(pTmp);
dvander@1203:   }
dvander@1203: #endif
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** This routine is called if the collation factory fails to deliver a
dvander@1203: ** collation function in the best encoding but there may be other versions
dvander@1203: ** of this collation function (for other text encodings) available. Use one
dvander@1203: ** of these instead if they exist. Avoid a UTF-8 <-> UTF-16 conversion if
dvander@1203: ** possible.
dvander@1203: */
dvander@1203: static int synthCollSeq(sqlite3 *db, CollSeq *pColl){
dvander@1203:   CollSeq *pColl2;
dvander@1203:   char *z = pColl->zName;
dvander@1203:   int n = strlen(z);
dvander@1203:   int i;
dvander@1203:   static const u8 aEnc[] = { SQLITE_UTF16BE, SQLITE_UTF16LE, SQLITE_UTF8 };
dvander@1203:   for(i=0; i<3; i++){
dvander@1203:     pColl2 = sqlite3FindCollSeq(db, aEnc[i], z, n, 0);
dvander@1203:     if( pColl2->xCmp!=0 ){
dvander@1203:       memcpy(pColl, pColl2, sizeof(CollSeq));
dvander@1203:       pColl->xDel = 0;         /* Do not copy the destructor */
dvander@1203:       return SQLITE_OK;
dvander@1203:     }
dvander@1203:   }
dvander@1203:   return SQLITE_ERROR;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** This function is responsible for invoking the collation factory callback
dvander@1203: ** or substituting a collation sequence of a different encoding when the
dvander@1203: ** requested collation sequence is not available in the database native
dvander@1203: ** encoding.
dvander@1203: ** 
dvander@1203: ** If it is not NULL, then pColl must point to the database native encoding 
dvander@1203: ** collation sequence with name zName, length nName.
dvander@1203: **
dvander@1203: ** The return value is either the collation sequence to be used in database
dvander@1203: ** db for collation type name zName, length nName, or NULL, if no collation
dvander@1203: ** sequence can be found.
dvander@1203: */
dvander@1203: CollSeq *sqlite3GetCollSeq(
dvander@1203:   sqlite3* db, 
dvander@1203:   CollSeq *pColl, 
dvander@1203:   const char *zName, 
dvander@1203:   int nName
dvander@1203: ){
dvander@1203:   CollSeq *p;
dvander@1203: 
dvander@1203:   p = pColl;
dvander@1203:   if( !p ){
dvander@1203:     p = sqlite3FindCollSeq(db, ENC(db), zName, nName, 0);
dvander@1203:   }
dvander@1203:   if( !p || !p->xCmp ){
dvander@1203:     /* No collation sequence of this type for this encoding is registered.
dvander@1203:     ** Call the collation factory to see if it can supply us with one.
dvander@1203:     */
dvander@1203:     callCollNeeded(db, zName, nName);
dvander@1203:     p = sqlite3FindCollSeq(db, ENC(db), zName, nName, 0);
dvander@1203:   }
dvander@1203:   if( p && !p->xCmp && synthCollSeq(db, p) ){
dvander@1203:     p = 0;
dvander@1203:   }
dvander@1203:   assert( !p || p->xCmp );
dvander@1203:   return p;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** This routine is called on a collation sequence before it is used to
dvander@1203: ** check that it is defined. An undefined collation sequence exists when
dvander@1203: ** a database is loaded that contains references to collation sequences
dvander@1203: ** that have not been defined by sqlite3_create_collation() etc.
dvander@1203: **
dvander@1203: ** If required, this routine calls the 'collation needed' callback to
dvander@1203: ** request a definition of the collating sequence. If this doesn't work, 
dvander@1203: ** an equivalent collating sequence that uses a text encoding different
dvander@1203: ** from the main database is substituted, if one is available.
dvander@1203: */
dvander@1203: int sqlite3CheckCollSeq(Parse *pParse, CollSeq *pColl){
dvander@1203:   if( pColl ){
dvander@1203:     const char *zName = pColl->zName;
dvander@1203:     CollSeq *p = sqlite3GetCollSeq(pParse->db, pColl, zName, -1);
dvander@1203:     if( !p ){
dvander@1203:       if( pParse->nErr==0 ){
dvander@1203:         sqlite3ErrorMsg(pParse, "no such collation sequence: %s", zName);
dvander@1203:       }
dvander@1203:       pParse->nErr++;
dvander@1203:       return SQLITE_ERROR;
dvander@1203:     }
dvander@1203:     assert( p==pColl );
dvander@1203:   }
dvander@1203:   return SQLITE_OK;
dvander@1203: }
dvander@1203: 
dvander@1203: 
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Locate and return an entry from the db.aCollSeq hash table. If the entry
dvander@1203: ** specified by zName and nName is not found and parameter 'create' is
dvander@1203: ** true, then create a new entry. Otherwise return NULL.
dvander@1203: **
dvander@1203: ** Each pointer stored in the sqlite3.aCollSeq hash table contains an
dvander@1203: ** array of three CollSeq structures. The first is the collation sequence
dvander@1203: ** prefferred for UTF-8, the second UTF-16le, and the third UTF-16be.
dvander@1203: **
dvander@1203: ** Stored immediately after the three collation sequences is a copy of
dvander@1203: ** the collation sequence name. A pointer to this string is stored in
dvander@1203: ** each collation sequence structure.
dvander@1203: */
dvander@1203: static CollSeq *findCollSeqEntry(
dvander@1203:   sqlite3 *db,
dvander@1203:   const char *zName,
dvander@1203:   int nName,
dvander@1203:   int create
dvander@1203: ){
dvander@1203:   CollSeq *pColl;
dvander@1203:   if( nName<0 ) nName = strlen(zName);
dvander@1203:   pColl = sqlite3HashFind(&db->aCollSeq, zName, nName);
dvander@1203: 
dvander@1203:   if( 0==pColl && create ){
dvander@1645:     pColl = sqlite3DbMallocZero(db, 3*sizeof(*pColl) + nName + 1 );
dvander@1203:     if( pColl ){
dvander@1203:       CollSeq *pDel = 0;
dvander@1203:       pColl[0].zName = (char*)&pColl[3];
dvander@1203:       pColl[0].enc = SQLITE_UTF8;
dvander@1203:       pColl[1].zName = (char*)&pColl[3];
dvander@1203:       pColl[1].enc = SQLITE_UTF16LE;
dvander@1203:       pColl[2].zName = (char*)&pColl[3];
dvander@1203:       pColl[2].enc = SQLITE_UTF16BE;
dvander@1203:       memcpy(pColl[0].zName, zName, nName);
dvander@1203:       pColl[0].zName[nName] = 0;
dvander@1203:       pDel = sqlite3HashInsert(&db->aCollSeq, pColl[0].zName, nName, pColl);
dvander@1203: 
dvander@1203:       /* If a malloc() failure occured in sqlite3HashInsert(), it will 
dvander@1203:       ** return the pColl pointer to be deleted (because it wasn't added
dvander@1203:       ** to the hash table).
dvander@1203:       */
dvander@1645:       assert( pDel==0 || pDel==pColl );
dvander@1645:       if( pDel!=0 ){
dvander@1645:         db->mallocFailed = 1;
dvander@1645:         sqlite3_free(pDel);
dvander@1203:         pColl = 0;
dvander@1203:       }
dvander@1203:     }
dvander@1203:   }
dvander@1203:   return pColl;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Parameter zName points to a UTF-8 encoded string nName bytes long.
dvander@1203: ** Return the CollSeq* pointer for the collation sequence named zName
dvander@1203: ** for the encoding 'enc' from the database 'db'.
dvander@1203: **
dvander@1203: ** If the entry specified is not found and 'create' is true, then create a
dvander@1203: ** new entry.  Otherwise return NULL.
dvander@1203: **
dvander@1203: ** A separate function sqlite3LocateCollSeq() is a wrapper around
dvander@1203: ** this routine.  sqlite3LocateCollSeq() invokes the collation factory
dvander@1203: ** if necessary and generates an error message if the collating sequence
dvander@1203: ** cannot be found.
dvander@1203: */
dvander@1203: CollSeq *sqlite3FindCollSeq(
dvander@1203:   sqlite3 *db,
dvander@1203:   u8 enc,
dvander@1203:   const char *zName,
dvander@1203:   int nName,
dvander@1203:   int create
dvander@1203: ){
dvander@1203:   CollSeq *pColl;
dvander@1203:   if( zName ){
dvander@1203:     pColl = findCollSeqEntry(db, zName, nName, create);
dvander@1203:   }else{
dvander@1203:     pColl = db->pDfltColl;
dvander@1203:   }
dvander@1203:   assert( SQLITE_UTF8==1 && SQLITE_UTF16LE==2 && SQLITE_UTF16BE==3 );
dvander@1203:   assert( enc>=SQLITE_UTF8 && enc<=SQLITE_UTF16BE );
dvander@1203:   if( pColl ) pColl += enc-1;
dvander@1203:   return pColl;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Locate a user function given a name, a number of arguments and a flag
dvander@1203: ** indicating whether the function prefers UTF-16 over UTF-8.  Return a
dvander@1203: ** pointer to the FuncDef structure that defines that function, or return
dvander@1203: ** NULL if the function does not exist.
dvander@1203: **
dvander@1203: ** If the createFlag argument is true, then a new (blank) FuncDef
dvander@1203: ** structure is created and liked into the "db" structure if a
dvander@1203: ** no matching function previously existed.  When createFlag is true
dvander@1203: ** and the nArg parameter is -1, then only a function that accepts
dvander@1203: ** any number of arguments will be returned.
dvander@1203: **
dvander@1203: ** If createFlag is false and nArg is -1, then the first valid
dvander@1203: ** function found is returned.  A function is valid if either xFunc
dvander@1203: ** or xStep is non-zero.
dvander@1203: **
dvander@1203: ** If createFlag is false, then a function with the required name and
dvander@1203: ** number of arguments may be returned even if the eTextRep flag does not
dvander@1203: ** match that requested.
dvander@1203: */
dvander@1203: FuncDef *sqlite3FindFunction(
dvander@1203:   sqlite3 *db,       /* An open database */
dvander@1203:   const char *zName, /* Name of the function.  Not null-terminated */
dvander@1203:   int nName,         /* Number of characters in the name */
dvander@1203:   int nArg,          /* Number of arguments.  -1 means any number */
dvander@1203:   u8 enc,            /* Preferred text encoding */
dvander@1203:   int createFlag     /* Create new entry if true and does not otherwise exist */
dvander@1203: ){
dvander@1203:   FuncDef *p;         /* Iterator variable */
dvander@1203:   FuncDef *pFirst;    /* First function with this name */
dvander@1203:   FuncDef *pBest = 0; /* Best match found so far */
dvander@1203:   int bestmatch = 0;  
dvander@1203: 
dvander@1203: 
dvander@1203:   assert( enc==SQLITE_UTF8 || enc==SQLITE_UTF16LE || enc==SQLITE_UTF16BE );
dvander@1203:   if( nArg<-1 ) nArg = -1;
dvander@1203: 
dvander@1203:   pFirst = (FuncDef*)sqlite3HashFind(&db->aFunc, zName, nName);
dvander@1203:   for(p=pFirst; p; p=p->pNext){
dvander@1203:     /* During the search for the best function definition, bestmatch is set
dvander@1203:     ** as follows to indicate the quality of the match with the definition
dvander@1203:     ** pointed to by pBest:
dvander@1203:     **
dvander@1203:     ** 0: pBest is NULL. No match has been found.
dvander@1203:     ** 1: A variable arguments function that prefers UTF-8 when a UTF-16
dvander@1203:     **    encoding is requested, or vice versa.
dvander@1203:     ** 2: A variable arguments function that uses UTF-16BE when UTF-16LE is
dvander@1203:     **    requested, or vice versa.
dvander@1203:     ** 3: A variable arguments function using the same text encoding.
dvander@1203:     ** 4: A function with the exact number of arguments requested that
dvander@1203:     **    prefers UTF-8 when a UTF-16 encoding is requested, or vice versa.
dvander@1203:     ** 5: A function with the exact number of arguments requested that
dvander@1203:     **    prefers UTF-16LE when UTF-16BE is requested, or vice versa.
dvander@1203:     ** 6: An exact match.
dvander@1203:     **
dvander@1203:     ** A larger value of 'matchqual' indicates a more desirable match.
dvander@1203:     */
dvander@1203:     if( p->nArg==-1 || p->nArg==nArg || nArg==-1 ){
dvander@1203:       int match = 1;          /* Quality of this match */
dvander@1203:       if( p->nArg==nArg || nArg==-1 ){
dvander@1203:         match = 4;
dvander@1203:       }
dvander@1203:       if( enc==p->iPrefEnc ){
dvander@1203:         match += 2;
dvander@1203:       }
dvander@1203:       else if( (enc==SQLITE_UTF16LE && p->iPrefEnc==SQLITE_UTF16BE) ||
dvander@1203:                (enc==SQLITE_UTF16BE && p->iPrefEnc==SQLITE_UTF16LE) ){
dvander@1203:         match += 1;
dvander@1203:       }
dvander@1203: 
dvander@1203:       if( match>bestmatch ){
dvander@1203:         pBest = p;
dvander@1203:         bestmatch = match;
dvander@1203:       }
dvander@1203:     }
dvander@1203:   }
dvander@1203: 
dvander@1203:   /* If the createFlag parameter is true, and the seach did not reveal an
dvander@1203:   ** exact match for the name, number of arguments and encoding, then add a
dvander@1203:   ** new entry to the hash table and return it.
dvander@1203:   */
dvander@1203:   if( createFlag && bestmatch<6 && 
dvander@1645:       (pBest = sqlite3DbMallocZero(db, sizeof(*pBest)+nName))!=0 ){
dvander@1203:     pBest->nArg = nArg;
dvander@1203:     pBest->pNext = pFirst;
dvander@1203:     pBest->iPrefEnc = enc;
dvander@1203:     memcpy(pBest->zName, zName, nName);
dvander@1203:     pBest->zName[nName] = 0;
dvander@1203:     if( pBest==sqlite3HashInsert(&db->aFunc,pBest->zName,nName,(void*)pBest) ){
dvander@1645:       db->mallocFailed = 1;
dvander@1645:       sqlite3_free(pBest);
dvander@1203:       return 0;
dvander@1203:     }
dvander@1203:   }
dvander@1203: 
dvander@1203:   if( pBest && (pBest->xStep || pBest->xFunc || createFlag) ){
dvander@1203:     return pBest;
dvander@1203:   }
dvander@1203:   return 0;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Free all resources held by the schema structure. The void* argument points
dvander@1645: ** at a Schema struct. This function does not call sqlite3_free() on the 
dvander@1203: ** pointer itself, it just cleans up subsiduary resources (i.e. the contents
dvander@1203: ** of the schema hash tables).
dvander@1203: */
dvander@1203: void sqlite3SchemaFree(void *p){
dvander@1203:   Hash temp1;
dvander@1203:   Hash temp2;
dvander@1203:   HashElem *pElem;
dvander@1203:   Schema *pSchema = (Schema *)p;
dvander@1203: 
dvander@1203:   temp1 = pSchema->tblHash;
dvander@1203:   temp2 = pSchema->trigHash;
dvander@1203:   sqlite3HashInit(&pSchema->trigHash, SQLITE_HASH_STRING, 0);
dvander@1203:   sqlite3HashClear(&pSchema->aFKey);
dvander@1203:   sqlite3HashClear(&pSchema->idxHash);
dvander@1203:   for(pElem=sqliteHashFirst(&temp2); pElem; pElem=sqliteHashNext(pElem)){
dvander@1203:     sqlite3DeleteTrigger((Trigger*)sqliteHashData(pElem));
dvander@1203:   }
dvander@1203:   sqlite3HashClear(&temp2);
dvander@1203:   sqlite3HashInit(&pSchema->tblHash, SQLITE_HASH_STRING, 0);
dvander@1203:   for(pElem=sqliteHashFirst(&temp1); pElem; pElem=sqliteHashNext(pElem)){
dvander@1203:     Table *pTab = sqliteHashData(pElem);
dvander@1203:     sqlite3DeleteTable(pTab);
dvander@1203:   }
dvander@1203:   sqlite3HashClear(&temp1);
dvander@1203:   pSchema->pSeqTab = 0;
dvander@1203:   pSchema->flags &= ~DB_SchemaLoaded;
dvander@1203: }
dvander@1203: 
dvander@1203: /*
dvander@1203: ** Find and return the schema associated with a BTree.  Create
dvander@1203: ** a new one if necessary.
dvander@1203: */
dvander@1645: Schema *sqlite3SchemaGet(sqlite3 *db, Btree *pBt){
dvander@1203:   Schema * p;
dvander@1203:   if( pBt ){
dvander@1645:     p = (Schema *)sqlite3BtreeSchema(pBt, sizeof(Schema), sqlite3SchemaFree);
dvander@1203:   }else{
dvander@1645:     p = (Schema *)sqlite3MallocZero(sizeof(Schema));
dvander@1203:   }
dvander@1645:   if( !p ){
dvander@1645:     db->mallocFailed = 1;
dvander@1645:   }else if ( 0==p->file_format ){
dvander@1203:     sqlite3HashInit(&p->tblHash, SQLITE_HASH_STRING, 0);
dvander@1203:     sqlite3HashInit(&p->idxHash, SQLITE_HASH_STRING, 0);
dvander@1203:     sqlite3HashInit(&p->trigHash, SQLITE_HASH_STRING, 0);
dvander@1203:     sqlite3HashInit(&p->aFKey, SQLITE_HASH_STRING, 1);
dvander@1203:     p->enc = SQLITE_UTF8;
dvander@1203:   }
dvander@1203:   return p;
dvander@1203: }