|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** 2003 April 6
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** The author disclaims copyright to this source code. In place of
|
|
Packit |
87b942 |
** a legal notice, here is a blessing:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** May you do good and not evil.
|
|
Packit |
87b942 |
** May you find forgiveness for yourself and forgive others.
|
|
Packit |
87b942 |
** May you share freely, never taking more than you give.
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
*************************************************************************
|
|
Packit |
87b942 |
** This file contains code used to implement the ATTACH and DETACH commands.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
#include "sqliteInt.h"
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
#ifndef SQLITE_OMIT_ATTACH
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** Resolve an expression that was part of an ATTACH or DETACH statement. This
|
|
Packit |
87b942 |
** is slightly different from resolving a normal SQL expression, because simple
|
|
Packit |
87b942 |
** identifiers are treated as strings, not possible column names or aliases.
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** i.e. if the parser sees:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** ATTACH DATABASE abc AS def
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** it treats the two expressions as literal strings 'abc' and 'def' instead of
|
|
Packit |
87b942 |
** looking for columns of the same name.
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** This only applies to the root node of pExpr, so the statement:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** ATTACH DATABASE abc||def AS 'db2'
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** will fail because neither abc or def can be resolved.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
static int resolveAttachExpr(NameContext *pName, Expr *pExpr)
|
|
Packit |
87b942 |
{
|
|
Packit |
87b942 |
int rc = SQLITE_OK;
|
|
Packit |
87b942 |
if( pExpr ){
|
|
Packit |
87b942 |
if( pExpr->op!=TK_ID ){
|
|
Packit |
87b942 |
rc = sqlite3ResolveExprNames(pName, pExpr);
|
|
Packit |
87b942 |
}else{
|
|
Packit |
87b942 |
pExpr->op = TK_STRING;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return rc;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** An SQL user-function registered to do the work of an ATTACH statement. The
|
|
Packit |
87b942 |
** three arguments to the function come directly from an attach statement:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** ATTACH DATABASE x AS y KEY z
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** SELECT sqlite_attach(x, y, z)
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** If the optional "KEY z" syntax is omitted, an SQL NULL is passed as the
|
|
Packit |
87b942 |
** third argument.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
static void attachFunc(
|
|
Packit |
87b942 |
sqlite3_context *context,
|
|
Packit |
87b942 |
int NotUsed,
|
|
Packit |
87b942 |
sqlite3_value **argv
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
int i;
|
|
Packit |
87b942 |
int rc = 0;
|
|
Packit |
87b942 |
sqlite3 *db = sqlite3_context_db_handle(context);
|
|
Packit |
87b942 |
const char *zName;
|
|
Packit |
87b942 |
const char *zFile;
|
|
Packit |
87b942 |
char *zPath = 0;
|
|
Packit |
87b942 |
char *zErr = 0;
|
|
Packit |
87b942 |
unsigned int flags;
|
|
Packit |
87b942 |
Db *aNew; /* New array of Db pointers */
|
|
Packit |
87b942 |
Db *pNew; /* Db object for the newly attached database */
|
|
Packit |
87b942 |
char *zErrDyn = 0;
|
|
Packit |
87b942 |
sqlite3_vfs *pVfs;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
UNUSED_PARAMETER(NotUsed);
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
zFile = (const char *)sqlite3_value_text(argv[0]);
|
|
Packit |
87b942 |
zName = (const char *)sqlite3_value_text(argv[1]);
|
|
Packit |
87b942 |
if( zFile==0 ) zFile = "";
|
|
Packit |
87b942 |
if( zName==0 ) zName = "";
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/* Check for the following errors:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** * Too many attached databases,
|
|
Packit |
87b942 |
** * Transaction currently open
|
|
Packit |
87b942 |
** * Specified database name already being used.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
if( db->nDb>=db->aLimit[SQLITE_LIMIT_ATTACHED]+2 ){
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db, "too many attached databases - max %d",
|
|
Packit |
87b942 |
db->aLimit[SQLITE_LIMIT_ATTACHED]
|
|
Packit |
87b942 |
);
|
|
Packit |
87b942 |
goto attach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
for(i=0; i<db->nDb; i++){
|
|
Packit |
87b942 |
char *z = db->aDb[i].zDbSName;
|
|
Packit |
87b942 |
assert( z && zName );
|
|
Packit |
87b942 |
if( sqlite3StrICmp(z, zName)==0 ){
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db, "database %s is already in use", zName);
|
|
Packit |
87b942 |
goto attach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/* Allocate the new entry in the db->aDb[] array and initialize the schema
|
|
Packit |
87b942 |
** hash tables.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
if( db->aDb==db->aDbStatic ){
|
|
Packit |
87b942 |
aNew = sqlite3DbMallocRawNN(db, sizeof(db->aDb[0])*3 );
|
|
Packit |
87b942 |
if( aNew==0 ) return;
|
|
Packit |
87b942 |
memcpy(aNew, db->aDb, sizeof(db->aDb[0])*2);
|
|
Packit |
87b942 |
}else{
|
|
Packit |
87b942 |
aNew = sqlite3DbRealloc(db, db->aDb, sizeof(db->aDb[0])*(db->nDb+1) );
|
|
Packit |
87b942 |
if( aNew==0 ) return;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
db->aDb = aNew;
|
|
Packit |
87b942 |
pNew = &db->aDb[db->nDb];
|
|
Packit |
87b942 |
memset(pNew, 0, sizeof(*pNew));
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/* Open the database file. If the btree is successfully opened, use
|
|
Packit |
87b942 |
** it to obtain the database schema. At this point the schema may
|
|
Packit |
87b942 |
** or may not be initialized.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
flags = db->openFlags;
|
|
Packit |
87b942 |
rc = sqlite3ParseUri(db->pVfs->zName, zFile, &flags, &pVfs, &zPath, &zErr);
|
|
Packit |
87b942 |
if( rc!=SQLITE_OK ){
|
|
Packit |
87b942 |
if( rc==SQLITE_NOMEM ) sqlite3OomFault(db);
|
|
Packit |
87b942 |
sqlite3_result_error(context, zErr, -1);
|
|
Packit |
87b942 |
sqlite3_free(zErr);
|
|
Packit |
87b942 |
return;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
assert( pVfs );
|
|
Packit |
87b942 |
flags |= SQLITE_OPEN_MAIN_DB;
|
|
Packit |
87b942 |
rc = sqlite3BtreeOpen(pVfs, zPath, db, &pNew->pBt, 0, flags);
|
|
Packit |
87b942 |
sqlite3_free( zPath );
|
|
Packit |
87b942 |
db->nDb++;
|
|
Packit |
87b942 |
db->skipBtreeMutex = 0;
|
|
Packit |
87b942 |
if( rc==SQLITE_CONSTRAINT ){
|
|
Packit |
87b942 |
rc = SQLITE_ERROR;
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db, "database is already attached");
|
|
Packit |
87b942 |
}else if( rc==SQLITE_OK ){
|
|
Packit |
87b942 |
Pager *pPager;
|
|
Packit |
87b942 |
pNew->pSchema = sqlite3SchemaGet(db, pNew->pBt);
|
|
Packit |
87b942 |
if( !pNew->pSchema ){
|
|
Packit |
87b942 |
rc = SQLITE_NOMEM_BKPT;
|
|
Packit |
87b942 |
}else if( pNew->pSchema->file_format && pNew->pSchema->enc!=ENC(db) ){
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db,
|
|
Packit |
87b942 |
"attached databases must use the same text encoding as main database");
|
|
Packit |
87b942 |
rc = SQLITE_ERROR;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
sqlite3BtreeEnter(pNew->pBt);
|
|
Packit |
87b942 |
pPager = sqlite3BtreePager(pNew->pBt);
|
|
Packit |
87b942 |
sqlite3PagerLockingMode(pPager, db->dfltLockMode);
|
|
Packit |
87b942 |
sqlite3BtreeSecureDelete(pNew->pBt,
|
|
Packit |
87b942 |
sqlite3BtreeSecureDelete(db->aDb[0].pBt,-1) );
|
|
Packit |
87b942 |
#ifndef SQLITE_OMIT_PAGER_PRAGMAS
|
|
Packit |
87b942 |
sqlite3BtreeSetPagerFlags(pNew->pBt,
|
|
Packit |
87b942 |
PAGER_SYNCHRONOUS_FULL | (db->flags & PAGER_FLAGS_MASK));
|
|
Packit |
87b942 |
#endif
|
|
Packit |
87b942 |
sqlite3BtreeLeave(pNew->pBt);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
pNew->safety_level = SQLITE_DEFAULT_SYNCHRONOUS+1;
|
|
Packit |
87b942 |
pNew->zDbSName = sqlite3DbStrDup(db, zName);
|
|
Packit |
87b942 |
if( rc==SQLITE_OK && pNew->zDbSName==0 ){
|
|
Packit |
87b942 |
rc = SQLITE_NOMEM_BKPT;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
#ifdef SQLITE_HAS_CODEC
|
|
Packit |
87b942 |
if( rc==SQLITE_OK ){
|
|
Packit |
87b942 |
extern int sqlite3CodecAttach(sqlite3*, int, const void*, int);
|
|
Packit |
87b942 |
extern void sqlite3CodecGetKey(sqlite3*, int, void**, int*);
|
|
Packit |
87b942 |
int nKey;
|
|
Packit |
87b942 |
char *zKey;
|
|
Packit |
87b942 |
int t = sqlite3_value_type(argv[2]);
|
|
Packit |
87b942 |
switch( t ){
|
|
Packit |
87b942 |
case SQLITE_INTEGER:
|
|
Packit |
87b942 |
case SQLITE_FLOAT:
|
|
Packit |
87b942 |
zErrDyn = sqlite3DbStrDup(db, "Invalid key value");
|
|
Packit |
87b942 |
rc = SQLITE_ERROR;
|
|
Packit |
87b942 |
break;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
case SQLITE_TEXT:
|
|
Packit |
87b942 |
case SQLITE_BLOB:
|
|
Packit |
87b942 |
nKey = sqlite3_value_bytes(argv[2]);
|
|
Packit |
87b942 |
zKey = (char *)sqlite3_value_blob(argv[2]);
|
|
Packit |
87b942 |
rc = sqlite3CodecAttach(db, db->nDb-1, zKey, nKey);
|
|
Packit |
87b942 |
break;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
case SQLITE_NULL:
|
|
Packit |
87b942 |
/* No key specified. Use the key from the main database */
|
|
Packit |
87b942 |
sqlite3CodecGetKey(db, 0, (void**)&zKey, &nKey);
|
|
Packit |
87b942 |
if( nKey || sqlite3BtreeGetOptimalReserve(db->aDb[0].pBt)>0 ){
|
|
Packit |
87b942 |
rc = sqlite3CodecAttach(db, db->nDb-1, zKey, nKey);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
break;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/* If the file was opened successfully, read the schema for the new database.
|
|
Packit |
87b942 |
** If this fails, or if opening the file failed, then close the file and
|
|
Packit |
87b942 |
** remove the entry from the db->aDb[] array. i.e. put everything back the way
|
|
Packit |
87b942 |
** we found it.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
if( rc==SQLITE_OK ){
|
|
Packit |
87b942 |
sqlite3BtreeEnterAll(db);
|
|
Packit |
87b942 |
rc = sqlite3Init(db, &zErrDyn);
|
|
Packit |
87b942 |
sqlite3BtreeLeaveAll(db);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#ifdef SQLITE_USER_AUTHENTICATION
|
|
Packit |
87b942 |
if( rc==SQLITE_OK ){
|
|
Packit |
87b942 |
u8 newAuth = 0;
|
|
Packit |
87b942 |
rc = sqlite3UserAuthCheckLogin(db, zName, &newAuth);
|
|
Packit |
87b942 |
if( newAuth<db->auth.authLevel ){
|
|
Packit |
87b942 |
rc = SQLITE_AUTH_USER;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif
|
|
Packit |
87b942 |
if( rc ){
|
|
Packit |
87b942 |
int iDb = db->nDb - 1;
|
|
Packit |
87b942 |
assert( iDb>=2 );
|
|
Packit |
87b942 |
if( db->aDb[iDb].pBt ){
|
|
Packit |
87b942 |
sqlite3BtreeClose(db->aDb[iDb].pBt);
|
|
Packit |
87b942 |
db->aDb[iDb].pBt = 0;
|
|
Packit |
87b942 |
db->aDb[iDb].pSchema = 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
sqlite3ResetAllSchemasOfConnection(db);
|
|
Packit |
87b942 |
db->nDb = iDb;
|
|
Packit |
87b942 |
if( rc==SQLITE_NOMEM || rc==SQLITE_IOERR_NOMEM ){
|
|
Packit |
87b942 |
sqlite3OomFault(db);
|
|
Packit |
87b942 |
sqlite3DbFree(db, zErrDyn);
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db, "out of memory");
|
|
Packit |
87b942 |
}else if( zErrDyn==0 ){
|
|
Packit |
87b942 |
zErrDyn = sqlite3MPrintf(db, "unable to open database: %s", zFile);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
goto attach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
return;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
attach_error:
|
|
Packit |
87b942 |
/* Return an error if we get here */
|
|
Packit |
87b942 |
if( zErrDyn ){
|
|
Packit |
87b942 |
sqlite3_result_error(context, zErrDyn, -1);
|
|
Packit |
87b942 |
sqlite3DbFree(db, zErrDyn);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( rc ) sqlite3_result_error_code(context, rc);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** An SQL user-function registered to do the work of an DETACH statement. The
|
|
Packit |
87b942 |
** three arguments to the function come directly from a detach statement:
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** DETACH DATABASE x
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** SELECT sqlite_detach(x)
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
static void detachFunc(
|
|
Packit |
87b942 |
sqlite3_context *context,
|
|
Packit |
87b942 |
int NotUsed,
|
|
Packit |
87b942 |
sqlite3_value **argv
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
const char *zName = (const char *)sqlite3_value_text(argv[0]);
|
|
Packit |
87b942 |
sqlite3 *db = sqlite3_context_db_handle(context);
|
|
Packit |
87b942 |
int i;
|
|
Packit |
87b942 |
Db *pDb = 0;
|
|
Packit |
87b942 |
char zErr[128];
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
UNUSED_PARAMETER(NotUsed);
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
if( zName==0 ) zName = "";
|
|
Packit |
87b942 |
for(i=0; i<db->nDb; i++){
|
|
Packit |
87b942 |
pDb = &db->aDb[i];
|
|
Packit |
87b942 |
if( pDb->pBt==0 ) continue;
|
|
Packit |
87b942 |
if( sqlite3StrICmp(pDb->zDbSName, zName)==0 ) break;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
if( i>=db->nDb ){
|
|
Packit |
87b942 |
sqlite3_snprintf(sizeof(zErr),zErr, "no such database: %s", zName);
|
|
Packit |
87b942 |
goto detach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( i<2 ){
|
|
Packit |
87b942 |
sqlite3_snprintf(sizeof(zErr),zErr, "cannot detach database %s", zName);
|
|
Packit |
87b942 |
goto detach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3BtreeIsInReadTrans(pDb->pBt) || sqlite3BtreeIsInBackup(pDb->pBt) ){
|
|
Packit |
87b942 |
sqlite3_snprintf(sizeof(zErr),zErr, "database %s is locked", zName);
|
|
Packit |
87b942 |
goto detach_error;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
sqlite3BtreeClose(pDb->pBt);
|
|
Packit |
87b942 |
pDb->pBt = 0;
|
|
Packit |
87b942 |
pDb->pSchema = 0;
|
|
Packit |
87b942 |
sqlite3CollapseDatabaseArray(db);
|
|
Packit |
87b942 |
return;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
detach_error:
|
|
Packit |
87b942 |
sqlite3_result_error(context, zErr, -1);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** This procedure generates VDBE code for a single invocation of either the
|
|
Packit |
87b942 |
** sqlite_detach() or sqlite_attach() SQL user functions.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
static void codeAttach(
|
|
Packit |
87b942 |
Parse *pParse, /* The parser context */
|
|
Packit |
87b942 |
int type, /* Either SQLITE_ATTACH or SQLITE_DETACH */
|
|
Packit |
87b942 |
FuncDef const *pFunc,/* FuncDef wrapper for detachFunc() or attachFunc() */
|
|
Packit |
87b942 |
Expr *pAuthArg, /* Expression to pass to authorization callback */
|
|
Packit |
87b942 |
Expr *pFilename, /* Name of database file */
|
|
Packit |
87b942 |
Expr *pDbname, /* Name of the database to use internally */
|
|
Packit |
87b942 |
Expr *pKey /* Database key for encryption extension */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
int rc;
|
|
Packit |
87b942 |
NameContext sName;
|
|
Packit |
87b942 |
Vdbe *v;
|
|
Packit |
87b942 |
sqlite3* db = pParse->db;
|
|
Packit |
87b942 |
int regArgs;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
if( pParse->nErr ) goto attach_end;
|
|
Packit |
87b942 |
memset(&sName, 0, sizeof(NameContext));
|
|
Packit |
87b942 |
sName.pParse = pParse;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
if(
|
|
Packit |
87b942 |
SQLITE_OK!=(rc = resolveAttachExpr(&sName, pFilename)) ||
|
|
Packit |
87b942 |
SQLITE_OK!=(rc = resolveAttachExpr(&sName, pDbname)) ||
|
|
Packit |
87b942 |
SQLITE_OK!=(rc = resolveAttachExpr(&sName, pKey))
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
goto attach_end;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
#ifndef SQLITE_OMIT_AUTHORIZATION
|
|
Packit |
87b942 |
if( pAuthArg ){
|
|
Packit |
87b942 |
char *zAuthArg;
|
|
Packit |
87b942 |
if( pAuthArg->op==TK_STRING ){
|
|
Packit |
87b942 |
zAuthArg = pAuthArg->u.zToken;
|
|
Packit |
87b942 |
}else{
|
|
Packit |
87b942 |
zAuthArg = 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
rc = sqlite3AuthCheck(pParse, type, zAuthArg, 0, 0);
|
|
Packit |
87b942 |
if(rc!=SQLITE_OK ){
|
|
Packit |
87b942 |
goto attach_end;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif /* SQLITE_OMIT_AUTHORIZATION */
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
v = sqlite3GetVdbe(pParse);
|
|
Packit |
87b942 |
regArgs = sqlite3GetTempRange(pParse, 4);
|
|
Packit |
87b942 |
sqlite3ExprCode(pParse, pFilename, regArgs);
|
|
Packit |
87b942 |
sqlite3ExprCode(pParse, pDbname, regArgs+1);
|
|
Packit |
87b942 |
sqlite3ExprCode(pParse, pKey, regArgs+2);
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
assert( v || db->mallocFailed );
|
|
Packit |
87b942 |
if( v ){
|
|
Packit |
87b942 |
sqlite3VdbeAddOp4(v, OP_Function0, 0, regArgs+3-pFunc->nArg, regArgs+3,
|
|
Packit |
87b942 |
(char *)pFunc, P4_FUNCDEF);
|
|
Packit |
87b942 |
assert( pFunc->nArg==-1 || (pFunc->nArg&0xff)==pFunc->nArg );
|
|
Packit |
87b942 |
sqlite3VdbeChangeP5(v, (u8)(pFunc->nArg));
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/* Code an OP_Expire. For an ATTACH statement, set P1 to true (expire this
|
|
Packit |
87b942 |
** statement only). For DETACH, set it to false (expire all existing
|
|
Packit |
87b942 |
** statements).
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
sqlite3VdbeAddOp1(v, OP_Expire, (type==SQLITE_ATTACH));
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
attach_end:
|
|
Packit |
87b942 |
sqlite3ExprDelete(db, pFilename);
|
|
Packit |
87b942 |
sqlite3ExprDelete(db, pDbname);
|
|
Packit |
87b942 |
sqlite3ExprDelete(db, pKey);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** Called by the parser to compile a DETACH statement.
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** DETACH pDbname
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
void sqlite3Detach(Parse *pParse, Expr *pDbname){
|
|
Packit |
87b942 |
static const FuncDef detach_func = {
|
|
Packit |
87b942 |
1, /* nArg */
|
|
Packit |
87b942 |
SQLITE_UTF8, /* funcFlags */
|
|
Packit |
87b942 |
0, /* pUserData */
|
|
Packit |
87b942 |
0, /* pNext */
|
|
Packit |
87b942 |
detachFunc, /* xSFunc */
|
|
Packit |
87b942 |
0, /* xFinalize */
|
|
Packit |
87b942 |
"sqlite_detach", /* zName */
|
|
Packit |
87b942 |
{0}
|
|
Packit |
87b942 |
};
|
|
Packit |
87b942 |
codeAttach(pParse, SQLITE_DETACH, &detach_func, pDbname, 0, 0, pDbname);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** Called by the parser to compile an ATTACH statement.
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** ATTACH p AS pDbname KEY pKey
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
void sqlite3Attach(Parse *pParse, Expr *p, Expr *pDbname, Expr *pKey){
|
|
Packit |
87b942 |
static const FuncDef attach_func = {
|
|
Packit |
87b942 |
3, /* nArg */
|
|
Packit |
87b942 |
SQLITE_UTF8, /* funcFlags */
|
|
Packit |
87b942 |
0, /* pUserData */
|
|
Packit |
87b942 |
0, /* pNext */
|
|
Packit |
87b942 |
attachFunc, /* xSFunc */
|
|
Packit |
87b942 |
0, /* xFinalize */
|
|
Packit |
87b942 |
"sqlite_attach", /* zName */
|
|
Packit |
87b942 |
{0}
|
|
Packit |
87b942 |
};
|
|
Packit |
87b942 |
codeAttach(pParse, SQLITE_ATTACH, &attach_func, p, p, pDbname, pKey);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif /* SQLITE_OMIT_ATTACH */
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** Initialize a DbFixer structure. This routine must be called prior
|
|
Packit |
87b942 |
** to passing the structure to one of the sqliteFixAAAA() routines below.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
void sqlite3FixInit(
|
|
Packit |
87b942 |
DbFixer *pFix, /* The fixer to be initialized */
|
|
Packit |
87b942 |
Parse *pParse, /* Error messages will be written here */
|
|
Packit |
87b942 |
int iDb, /* This is the database that must be used */
|
|
Packit |
87b942 |
const char *zType, /* "view", "trigger", or "index" */
|
|
Packit |
87b942 |
const Token *pName /* Name of the view, trigger, or index */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
sqlite3 *db;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
db = pParse->db;
|
|
Packit |
87b942 |
assert( db->nDb>iDb );
|
|
Packit |
87b942 |
pFix->pParse = pParse;
|
|
Packit |
87b942 |
pFix->zDb = db->aDb[iDb].zDbSName;
|
|
Packit |
87b942 |
pFix->pSchema = db->aDb[iDb].pSchema;
|
|
Packit |
87b942 |
pFix->zType = zType;
|
|
Packit |
87b942 |
pFix->pName = pName;
|
|
Packit |
87b942 |
pFix->bVarOnly = (iDb==1);
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
/*
|
|
Packit |
87b942 |
** The following set of routines walk through the parse tree and assign
|
|
Packit |
87b942 |
** a specific database to all table references where the database name
|
|
Packit |
87b942 |
** was left unspecified in the original SQL statement. The pFix structure
|
|
Packit |
87b942 |
** must have been initialized by a prior call to sqlite3FixInit().
|
|
Packit |
87b942 |
**
|
|
Packit |
87b942 |
** These routines are used to make sure that an index, trigger, or
|
|
Packit |
87b942 |
** view in one database does not refer to objects in a different database.
|
|
Packit |
87b942 |
** (Exception: indices, triggers, and views in the TEMP database are
|
|
Packit |
87b942 |
** allowed to refer to anything.) If a reference is explicitly made
|
|
Packit |
87b942 |
** to an object in a different database, an error message is added to
|
|
Packit |
87b942 |
** pParse->zErrMsg and these routines return non-zero. If everything
|
|
Packit |
87b942 |
** checks out, these routines return 0.
|
|
Packit |
87b942 |
*/
|
|
Packit |
87b942 |
int sqlite3FixSrcList(
|
|
Packit |
87b942 |
DbFixer *pFix, /* Context of the fixation */
|
|
Packit |
87b942 |
SrcList *pList /* The Source list to check and modify */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
int i;
|
|
Packit |
87b942 |
const char *zDb;
|
|
Packit |
87b942 |
struct SrcList_item *pItem;
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
if( NEVER(pList==0) ) return 0;
|
|
Packit |
87b942 |
zDb = pFix->zDb;
|
|
Packit |
87b942 |
for(i=0, pItem=pList->a; i<pList->nSrc; i++, pItem++){
|
|
Packit |
87b942 |
if( pFix->bVarOnly==0 ){
|
|
Packit |
87b942 |
if( pItem->zDatabase && sqlite3StrICmp(pItem->zDatabase, zDb) ){
|
|
Packit |
87b942 |
sqlite3ErrorMsg(pFix->pParse,
|
|
Packit |
87b942 |
"%s %T cannot reference objects in database %s",
|
|
Packit |
87b942 |
pFix->zType, pFix->pName, pItem->zDatabase);
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
sqlite3DbFree(pFix->pParse->db, pItem->zDatabase);
|
|
Packit |
87b942 |
pItem->zDatabase = 0;
|
|
Packit |
87b942 |
pItem->pSchema = pFix->pSchema;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#if !defined(SQLITE_OMIT_VIEW) || !defined(SQLITE_OMIT_TRIGGER)
|
|
Packit |
87b942 |
if( sqlite3FixSelect(pFix, pItem->pSelect) ) return 1;
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pItem->pOn) ) return 1;
|
|
Packit |
87b942 |
#endif
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#if !defined(SQLITE_OMIT_VIEW) || !defined(SQLITE_OMIT_TRIGGER)
|
|
Packit |
87b942 |
int sqlite3FixSelect(
|
|
Packit |
87b942 |
DbFixer *pFix, /* Context of the fixation */
|
|
Packit |
87b942 |
Select *pSelect /* The SELECT statement to be fixed to one database */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
while( pSelect ){
|
|
Packit |
87b942 |
if( sqlite3FixExprList(pFix, pSelect->pEList) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixSrcList(pFix, pSelect->pSrc) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pSelect->pWhere) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExprList(pFix, pSelect->pGroupBy) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pSelect->pHaving) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExprList(pFix, pSelect->pOrderBy) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pSelect->pLimit) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
pSelect = pSelect->pPrior;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
int sqlite3FixExpr(
|
|
Packit |
87b942 |
DbFixer *pFix, /* Context of the fixation */
|
|
Packit |
87b942 |
Expr *pExpr /* The expression to be fixed to one database */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
while( pExpr ){
|
|
Packit |
87b942 |
if( pExpr->op==TK_VARIABLE ){
|
|
Packit |
87b942 |
if( pFix->pParse->db->init.busy ){
|
|
Packit |
87b942 |
pExpr->op = TK_NULL;
|
|
Packit |
87b942 |
}else{
|
|
Packit |
87b942 |
sqlite3ErrorMsg(pFix->pParse, "%s cannot use variables", pFix->zType);
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( ExprHasProperty(pExpr, EP_TokenOnly|EP_Leaf) ) break;
|
|
Packit |
87b942 |
if( ExprHasProperty(pExpr, EP_xIsSelect) ){
|
|
Packit |
87b942 |
if( sqlite3FixSelect(pFix, pExpr->x.pSelect) ) return 1;
|
|
Packit |
87b942 |
}else{
|
|
Packit |
87b942 |
if( sqlite3FixExprList(pFix, pExpr->x.pList) ) return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pExpr->pRight) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
pExpr = pExpr->pLeft;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
int sqlite3FixExprList(
|
|
Packit |
87b942 |
DbFixer *pFix, /* Context of the fixation */
|
|
Packit |
87b942 |
ExprList *pList /* The expression to be fixed to one database */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
int i;
|
|
Packit |
87b942 |
struct ExprList_item *pItem;
|
|
Packit |
87b942 |
if( pList==0 ) return 0;
|
|
Packit |
87b942 |
for(i=0, pItem=pList->a; i<pList->nExpr; i++, pItem++){
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pItem->pExpr) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif
|
|
Packit |
87b942 |
|
|
Packit |
87b942 |
#ifndef SQLITE_OMIT_TRIGGER
|
|
Packit |
87b942 |
int sqlite3FixTriggerStep(
|
|
Packit |
87b942 |
DbFixer *pFix, /* Context of the fixation */
|
|
Packit |
87b942 |
TriggerStep *pStep /* The trigger step be fixed to one database */
|
|
Packit |
87b942 |
){
|
|
Packit |
87b942 |
while( pStep ){
|
|
Packit |
87b942 |
if( sqlite3FixSelect(pFix, pStep->pSelect) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExpr(pFix, pStep->pWhere) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
if( sqlite3FixExprList(pFix, pStep->pExprList) ){
|
|
Packit |
87b942 |
return 1;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
pStep = pStep->pNext;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
return 0;
|
|
Packit |
87b942 |
}
|
|
Packit |
87b942 |
#endif
|