14fccf43aSdan 29b1c62d4Sdrh #if defined(SQLITE_ENABLE_SESSION) && defined(SQLITE_ENABLE_PREUPDATE_HOOK) 34fccf43aSdan #include "sqlite3session.h" 44fccf43aSdan #include <assert.h> 54fccf43aSdan #include <string.h> 64fccf43aSdan 75d8a2984Sdan #ifndef SQLITE_AMALGAMATION 84fccf43aSdan # include "sqliteInt.h" 94fccf43aSdan # include "vdbeInt.h" 105d8a2984Sdan #endif 114fccf43aSdan 124fccf43aSdan typedef struct SessionTable SessionTable; 134fccf43aSdan typedef struct SessionChange SessionChange; 14296c7658Sdan typedef struct SessionBuffer SessionBuffer; 15ef7a6304Sdan typedef struct SessionInput SessionInput; 16ef7a6304Sdan 17ef7a6304Sdan /* 18ef7a6304Sdan ** Minimum chunk size used by streaming versions of functions. 19ef7a6304Sdan */ 20f1a08ad8Sdrh #ifndef SESSIONS_STRM_CHUNK_SIZE 214757c658Sdan # ifdef SQLITE_TEST 22f1a08ad8Sdrh # define SESSIONS_STRM_CHUNK_SIZE 64 234757c658Sdan # else 24f1a08ad8Sdrh # define SESSIONS_STRM_CHUNK_SIZE 1024 25f1a08ad8Sdrh # endif 264757c658Sdan #endif 274fccf43aSdan 28cf8e9144Sdan typedef struct SessionHook SessionHook; 29cf8e9144Sdan struct SessionHook { 30cf8e9144Sdan void *pCtx; 31cf8e9144Sdan int (*xOld)(void*,int,sqlite3_value**); 32cf8e9144Sdan int (*xNew)(void*,int,sqlite3_value**); 33cf8e9144Sdan int (*xCount)(void*); 34cf8e9144Sdan int (*xDepth)(void*); 35cf8e9144Sdan }; 36cf8e9144Sdan 37296c7658Sdan /* 38296c7658Sdan ** Session handle structure. 39296c7658Sdan */ 404fccf43aSdan struct sqlite3_session { 414fccf43aSdan sqlite3 *db; /* Database handle session is attached to */ 424fccf43aSdan char *zDb; /* Name of database session is attached to */ 43296c7658Sdan int bEnable; /* True if currently recording */ 44b4480e94Sdan int bIndirect; /* True if all changes are indirect */ 45ff4d0f41Sdan int bAutoAttach; /* True to auto-attach tables */ 464fccf43aSdan int rc; /* Non-zero if an error has occurred */ 477531a5a3Sdan void *pFilterCtx; /* First argument to pass to xTableFilter */ 487531a5a3Sdan int (*xTableFilter)(void *pCtx, const char *zTab); 491611e5a3Sdan sqlite3_value *pZeroBlob; /* Value containing X'' */ 504fccf43aSdan sqlite3_session *pNext; /* Next session object on same db. */ 514fccf43aSdan SessionTable *pTable; /* List of attached tables */ 52cf8e9144Sdan SessionHook hook; /* APIs to grab new and old data with */ 534fccf43aSdan }; 544fccf43aSdan 554fccf43aSdan /* 56ef7a6304Sdan ** Instances of this structure are used to build strings or binary records. 57ef7a6304Sdan */ 58ef7a6304Sdan struct SessionBuffer { 59ef7a6304Sdan u8 *aBuf; /* Pointer to changeset buffer */ 60ef7a6304Sdan int nBuf; /* Size of buffer aBuf */ 61ef7a6304Sdan int nAlloc; /* Size of allocation containing aBuf */ 62ef7a6304Sdan }; 63ef7a6304Sdan 64ef7a6304Sdan /* 6516228167Sdan ** An object of this type is used internally as an abstraction for 6616228167Sdan ** input data. Input data may be supplied either as a single large buffer 6716228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g. 68f1a08ad8Sdrh ** sqlite3changeset_start_strm()). 69ef7a6304Sdan */ 70ef7a6304Sdan struct SessionInput { 71d9151526Sdan int bNoDiscard; /* If true, discard no data */ 72d9151526Sdan int iCurrent; /* Offset in aData[] of current change */ 734757c658Sdan int iNext; /* Offset in aData[] of next change */ 744757c658Sdan u8 *aData; /* Pointer to buffer containing changeset */ 754757c658Sdan int nData; /* Number of bytes in aData */ 764757c658Sdan 77ef7a6304Sdan SessionBuffer buf; /* Current read buffer */ 78ef7a6304Sdan int (*xInput)(void*, void*, int*); /* Input stream call (or NULL) */ 79ef7a6304Sdan void *pIn; /* First argument to xInput */ 80ef7a6304Sdan int bEof; /* Set to true after xInput finished */ 81ef7a6304Sdan }; 82ef7a6304Sdan 83ef7a6304Sdan /* 84296c7658Sdan ** Structure for changeset iterators. 85296c7658Sdan */ 86296c7658Sdan struct sqlite3_changeset_iter { 87ef7a6304Sdan SessionInput in; /* Input buffer or stream */ 88ef7a6304Sdan SessionBuffer tblhdr; /* Buffer to hold apValue/zTab/abPK/ */ 8973b3c055Sdan int bPatchset; /* True if this is a patchset */ 90296c7658Sdan int rc; /* Iterator error code */ 91296c7658Sdan sqlite3_stmt *pConflict; /* Points to conflicting row, if any */ 92296c7658Sdan char *zTab; /* Current table */ 93296c7658Sdan int nCol; /* Number of columns in zTab */ 94296c7658Sdan int op; /* Current operation */ 95b4480e94Sdan int bIndirect; /* True if current change was indirect */ 96244593c8Sdan u8 *abPK; /* Primary key array */ 97296c7658Sdan sqlite3_value **apValue; /* old.* and new.* values */ 98296c7658Sdan }; 99296c7658Sdan 100296c7658Sdan /* 1014fccf43aSdan ** Each session object maintains a set of the following structures, one 1024fccf43aSdan ** for each table the session object is monitoring. The structures are 1034fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable. 1044fccf43aSdan ** 1054fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have 1064fccf43aSdan ** been modified in any way since the session object was attached to the 1074fccf43aSdan ** table. 1084fccf43aSdan ** 1094fccf43aSdan ** The data associated with each hash-table entry is a structure containing 1104fccf43aSdan ** a subset of the initial values that the modified row contained at the 1114fccf43aSdan ** start of the session. Or no initial values if the row was inserted. 1124fccf43aSdan */ 1134fccf43aSdan struct SessionTable { 1144fccf43aSdan SessionTable *pNext; 1154fccf43aSdan char *zName; /* Local name of table */ 1164fccf43aSdan int nCol; /* Number of columns in table zName */ 1173739f298Sdan int bStat1; /* True if this is sqlite_stat1 */ 118e8d5648eSdan const char **azCol; /* Column names */ 119e8d5648eSdan u8 *abPK; /* Array of primary key flags */ 120296c7658Sdan int nEntry; /* Total number of entries in hash table */ 1214fccf43aSdan int nChange; /* Size of apChange[] array */ 1224fccf43aSdan SessionChange **apChange; /* Hash table buckets */ 1234fccf43aSdan }; 1244fccf43aSdan 1254fccf43aSdan /* 1264fccf43aSdan ** RECORD FORMAT: 1274fccf43aSdan ** 1284fccf43aSdan ** The following record format is similar to (but not compatible with) that 1294fccf43aSdan ** used in SQLite database files. This format is used as part of the 1304fccf43aSdan ** change-set binary format, and so must be architecture independent. 1314fccf43aSdan ** 1324fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained - 1334fccf43aSdan ** there is no separation of header and data. Each field begins with a 1344fccf43aSdan ** single byte describing its type, as follows: 1354fccf43aSdan ** 1364fccf43aSdan ** 0x00: Undefined value. 1374fccf43aSdan ** 0x01: Integer value. 1384fccf43aSdan ** 0x02: Real value. 1394fccf43aSdan ** 0x03: Text value. 1404fccf43aSdan ** 0x04: Blob value. 1414fccf43aSdan ** 0x05: SQL NULL value. 1424fccf43aSdan ** 1434fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT 1444fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists 1454fccf43aSdan ** only of the single type byte. For other types of values, the type byte 1464fccf43aSdan ** is followed by: 1474fccf43aSdan ** 1484fccf43aSdan ** Text values: 1494fccf43aSdan ** A varint containing the number of bytes in the value (encoded using 1504fccf43aSdan ** UTF-8). Followed by a buffer containing the UTF-8 representation 1514fccf43aSdan ** of the text value. There is no nul terminator. 1524fccf43aSdan ** 1534fccf43aSdan ** Blob values: 1544fccf43aSdan ** A varint containing the number of bytes in the value, followed by 1554fccf43aSdan ** a buffer containing the value itself. 1564fccf43aSdan ** 1574fccf43aSdan ** Integer values: 1584fccf43aSdan ** An 8-byte big-endian integer value. 1594fccf43aSdan ** 1604fccf43aSdan ** Real values: 1614fccf43aSdan ** An 8-byte big-endian IEEE 754-2008 real value. 1624fccf43aSdan ** 1634fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite 1644fccf43aSdan ** record format. 1654fccf43aSdan ** 1664fccf43aSdan ** CHANGESET FORMAT: 1674fccf43aSdan ** 1684fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on 1694fccf43aSdan ** one or more tables. Operations on a single table are grouped together, 1704fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all 1714fccf43aSdan ** mixed together). 1724fccf43aSdan ** 1734fccf43aSdan ** Each group of changes begins with a table header: 1744fccf43aSdan ** 1754fccf43aSdan ** 1 byte: Constant 0x54 (capital 'T') 176730bb805Sdan ** Varint: Number of columns in the table. 17773b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 1784fccf43aSdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 1794fccf43aSdan ** 1804fccf43aSdan ** Followed by one or more changes to the table. 1814fccf43aSdan ** 182c8be6437Sdrh ** 1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09). 1835d607a6eSdan ** 1 byte: The "indirect-change" flag. 1844fccf43aSdan ** old.* record: (delete and update only) 1854fccf43aSdan ** new.* record: (insert and update only) 18673b3c055Sdan ** 187730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the 188730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of 189730bb805Sdan ** columns in the table. The i'th field of each record is associated with 190730bb805Sdan ** the i'th column of the table, counting from left to right in the order 191730bb805Sdan ** in which columns were declared in the CREATE TABLE statement. 192730bb805Sdan ** 193730bb805Sdan ** The new.* record that is part of each INSERT change contains the values 194730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each 195730bb805Sdan ** DELETE change contains the values that made up the row that was deleted 196730bb805Sdan ** from the database. In the changeset format, the records that are part 197730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00) 198730bb805Sdan ** fields. 199730bb805Sdan ** 200730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields 201730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are 202730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields 203730bb805Sdan ** are set to the values that made up the row before the UPDATE that the 204730bb805Sdan ** change records took place. Within the new.* record, fields associated 205730bb805Sdan ** with table columns modified by the UPDATE change contain the new 206730bb805Sdan ** values. Fields associated with table columns that are not modified 207730bb805Sdan ** are set to "undefined". 208730bb805Sdan ** 20973b3c055Sdan ** PATCHSET FORMAT: 21073b3c055Sdan ** 21173b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset, 212730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict 213730bb805Sdan ** resolution is required when applying the changeset. 21473b3c055Sdan ** 21573b3c055Sdan ** Each group of changes begins with a table header: 21673b3c055Sdan ** 21773b3c055Sdan ** 1 byte: Constant 0x50 (capital 'P') 218730bb805Sdan ** Varint: Number of columns in the table. 21973b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 22073b3c055Sdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 22173b3c055Sdan ** 22273b3c055Sdan ** Followed by one or more changes to the table. 22373b3c055Sdan ** 224c8be6437Sdrh ** 1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09). 22573b3c055Sdan ** 1 byte: The "indirect-change" flag. 226730bb805Sdan ** single record: (PK fields for DELETE, PK and modified fields for UPDATE, 227730bb805Sdan ** full record for INSERT). 228730bb805Sdan ** 229730bb805Sdan ** As in the changeset format, each field of the single record that is part 230730bb805Sdan ** of a patchset change is associated with the correspondingly positioned 231730bb805Sdan ** table column, counting from left to right within the CREATE TABLE 232730bb805Sdan ** statement. 233730bb805Sdan ** 234730bb805Sdan ** For a DELETE change, all fields within the record except those associated 235730bb805Sdan ** with PRIMARY KEY columns are set to "undefined". The PRIMARY KEY fields 236730bb805Sdan ** contain the values identifying the row to delete. 237730bb805Sdan ** 238730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY 239730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined". 240730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update, 241730bb805Sdan ** and fields associated with modified columns contain the new column values. 242730bb805Sdan ** 243730bb805Sdan ** The records associated with INSERT changes are in the same format as for 244730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT 245730bb805Sdan ** change to contain a field set to "undefined". 2464fccf43aSdan */ 2474fccf43aSdan 2484fccf43aSdan /* 2494fccf43aSdan ** For each row modified during a session, there exists a single instance of 2504fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table. 2514fccf43aSdan */ 2524fccf43aSdan struct SessionChange { 253798693b2Sdan int op; /* One of UPDATE, DELETE, INSERT */ 254b4480e94Sdan int bIndirect; /* True if this change is "indirect" */ 2554fccf43aSdan int nRecord; /* Number of bytes in buffer aRecord[] */ 2564fccf43aSdan u8 *aRecord; /* Buffer containing old.* record */ 2574fccf43aSdan SessionChange *pNext; /* For hash-table collisions */ 2584fccf43aSdan }; 2594fccf43aSdan 260296c7658Sdan /* 261296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the 262296c7658Sdan ** number of bytes written. 263296c7658Sdan */ 264296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){ 265296c7658Sdan return putVarint32(aBuf, iVal); 2664fccf43aSdan } 2674fccf43aSdan 268296c7658Sdan /* 269296c7658Sdan ** Return the number of bytes required to store value iVal as a varint. 270296c7658Sdan */ 271296c7658Sdan static int sessionVarintLen(int iVal){ 272296c7658Sdan return sqlite3VarintLen(iVal); 273296c7658Sdan } 274296c7658Sdan 275296c7658Sdan /* 276296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of 277296c7658Sdan ** bytes read. 278296c7658Sdan */ 2794fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){ 280296c7658Sdan return getVarint32(aBuf, *piVal); 2814fccf43aSdan } 2824fccf43aSdan 28348cd59a5Sdrh /* Load an unaligned and unsigned 32-bit integer */ 28448cd59a5Sdrh #define SESSION_UINT32(x) (((u32)(x)[0]<<24)|((x)[1]<<16)|((x)[2]<<8)|(x)[3]) 28548cd59a5Sdrh 286296c7658Sdan /* 287296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return 288296c7658Sdan ** the value read. 289296c7658Sdan */ 2904fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){ 29148cd59a5Sdrh u64 x = SESSION_UINT32(aRec); 29248cd59a5Sdrh u32 y = SESSION_UINT32(aRec+4); 29348cd59a5Sdrh x = (x<<32) + y; 29448cd59a5Sdrh return (sqlite3_int64)x; 2954fccf43aSdan } 2964fccf43aSdan 2974fccf43aSdan /* 298296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[]. 299296c7658Sdan */ 300296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){ 301296c7658Sdan aBuf[0] = (i>>56) & 0xFF; 302296c7658Sdan aBuf[1] = (i>>48) & 0xFF; 303296c7658Sdan aBuf[2] = (i>>40) & 0xFF; 304296c7658Sdan aBuf[3] = (i>>32) & 0xFF; 305296c7658Sdan aBuf[4] = (i>>24) & 0xFF; 306296c7658Sdan aBuf[5] = (i>>16) & 0xFF; 307296c7658Sdan aBuf[6] = (i>> 8) & 0xFF; 308296c7658Sdan aBuf[7] = (i>> 0) & 0xFF; 309296c7658Sdan } 310296c7658Sdan 311296c7658Sdan /* 3124fccf43aSdan ** This function is used to serialize the contents of value pValue (see 3134fccf43aSdan ** comment titled "RECORD FORMAT" above). 3144fccf43aSdan ** 3154fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to 3164fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before 3174fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is 3184fccf43aSdan ** set *pnWrite. 3194fccf43aSdan ** 3204fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs 3214fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16)) 3224fccf43aSdan ** SQLITE_NOMEM is returned. 3234fccf43aSdan */ 3244fccf43aSdan static int sessionSerializeValue( 3254fccf43aSdan u8 *aBuf, /* If non-NULL, write serialized value here */ 3264fccf43aSdan sqlite3_value *pValue, /* Value to serialize */ 3274fccf43aSdan int *pnWrite /* IN/OUT: Increment by bytes written */ 3284fccf43aSdan ){ 329296c7658Sdan int nByte; /* Size of serialized value in bytes */ 3304fccf43aSdan 33180fe2d93Sdan if( pValue ){ 33280fe2d93Sdan int eType; /* Value type (SQLITE_NULL, TEXT etc.) */ 33380fe2d93Sdan 3344fccf43aSdan eType = sqlite3_value_type(pValue); 3354fccf43aSdan if( aBuf ) aBuf[0] = eType; 3364fccf43aSdan 3374fccf43aSdan switch( eType ){ 3384fccf43aSdan case SQLITE_NULL: 3394fccf43aSdan nByte = 1; 3404fccf43aSdan break; 3414fccf43aSdan 3424fccf43aSdan case SQLITE_INTEGER: 3434fccf43aSdan case SQLITE_FLOAT: 3444fccf43aSdan if( aBuf ){ 3454fccf43aSdan /* TODO: SQLite does something special to deal with mixed-endian 3464fccf43aSdan ** floating point values (e.g. ARM7). This code probably should 3474fccf43aSdan ** too. */ 3484fccf43aSdan u64 i; 3494fccf43aSdan if( eType==SQLITE_INTEGER ){ 3504fccf43aSdan i = (u64)sqlite3_value_int64(pValue); 3514fccf43aSdan }else{ 3524fccf43aSdan double r; 3534fccf43aSdan assert( sizeof(double)==8 && sizeof(u64)==8 ); 3544fccf43aSdan r = sqlite3_value_double(pValue); 3554fccf43aSdan memcpy(&i, &r, 8); 3564fccf43aSdan } 357296c7658Sdan sessionPutI64(&aBuf[1], i); 3584fccf43aSdan } 3594fccf43aSdan nByte = 9; 3604fccf43aSdan break; 3614fccf43aSdan 3624e895da1Sdan default: { 36380fe2d93Sdan u8 *z; 36480fe2d93Sdan int n; 36580fe2d93Sdan int nVarint; 36680fe2d93Sdan 3674e895da1Sdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 36880fe2d93Sdan if( eType==SQLITE_TEXT ){ 36980fe2d93Sdan z = (u8 *)sqlite3_value_text(pValue); 37080fe2d93Sdan }else{ 37180fe2d93Sdan z = (u8 *)sqlite3_value_blob(pValue); 37280fe2d93Sdan } 37380fe2d93Sdan n = sqlite3_value_bytes(pValue); 3743cc89d95Sdan if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 37580fe2d93Sdan nVarint = sessionVarintLen(n); 37680fe2d93Sdan 3774fccf43aSdan if( aBuf ){ 3784fccf43aSdan sessionVarintPut(&aBuf[1], n); 379895decf6Sdan if( n ) memcpy(&aBuf[nVarint + 1], z, n); 3804fccf43aSdan } 3814fccf43aSdan 3824fccf43aSdan nByte = 1 + nVarint + n; 3834fccf43aSdan break; 3844fccf43aSdan } 3854fccf43aSdan } 38680fe2d93Sdan }else{ 38780fe2d93Sdan nByte = 1; 38880fe2d93Sdan if( aBuf ) aBuf[0] = '\0'; 38980fe2d93Sdan } 3904fccf43aSdan 391fa122adaSdan if( pnWrite ) *pnWrite += nByte; 3924fccf43aSdan return SQLITE_OK; 3934fccf43aSdan } 3944fccf43aSdan 395fa122adaSdan 396798693b2Sdan /* 397798693b2Sdan ** This macro is used to calculate hash key values for data structures. In 398798693b2Sdan ** order to use this macro, the entire data structure must be represented 399798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value 400798693b2Sdan ** for a data structure represented as three such integers, the macro may 401798693b2Sdan ** then be used as follows: 402798693b2Sdan ** 403798693b2Sdan ** int hash_key_value; 404798693b2Sdan ** hash_key_value = HASH_APPEND(0, <value 1>); 405798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 2>); 406798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 3>); 407798693b2Sdan ** 408798693b2Sdan ** In practice, the data structures this macro is used for are the primary 409798693b2Sdan ** key values of modified rows. 410798693b2Sdan */ 4114131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add) 412798693b2Sdan 413798693b2Sdan /* 414798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the 415798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 416798693b2Sdan */ 4174131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){ 418e8d5648eSdan h = HASH_APPEND(h, i & 0xFFFFFFFF); 419e8d5648eSdan return HASH_APPEND(h, (i>>32)&0xFFFFFFFF); 420e8d5648eSdan } 421798693b2Sdan 422798693b2Sdan /* 423798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to 424798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value. 425798693b2Sdan */ 4264131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){ 427e8d5648eSdan int i; 428e8d5648eSdan for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]); 429e8d5648eSdan return h; 430e8d5648eSdan } 431e8d5648eSdan 4324fccf43aSdan /* 433798693b2Sdan ** Append the hash of the data type passed as the second argument to the 434798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 435798693b2Sdan */ 436798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){ 437798693b2Sdan return HASH_APPEND(h, eType); 438798693b2Sdan } 439798693b2Sdan 440798693b2Sdan /* 4414131639cSdan ** This function may only be called from within a pre-update callback. 4424131639cSdan ** It calculates a hash based on the primary key values of the old.* or 443798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to 444798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL 445798693b2Sdan ** values, *pbNullPK is set to true before returning. 446798693b2Sdan ** 447798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values 448798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned 449798693b2Sdan ** and the output variables are set as described above. 4504fccf43aSdan */ 451798693b2Sdan static int sessionPreupdateHash( 452cf8e9144Sdan sqlite3_session *pSession, /* Session object that owns pTab */ 453e8d5648eSdan SessionTable *pTab, /* Session table handle */ 454e8d5648eSdan int bNew, /* True to hash the new.* PK */ 45527453faeSdan int *piHash, /* OUT: Hash value */ 456798693b2Sdan int *pbNullPK /* OUT: True if there are NULL values in PK */ 457e8d5648eSdan ){ 4584131639cSdan unsigned int h = 0; /* Hash value to return */ 4594131639cSdan int i; /* Used to iterate through columns */ 460e8d5648eSdan 46127453faeSdan assert( *pbNullPK==0 ); 462cf8e9144Sdan assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) ); 463e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 464e8d5648eSdan if( pTab->abPK[i] ){ 465e8d5648eSdan int rc; 466e8d5648eSdan int eType; 467e8d5648eSdan sqlite3_value *pVal; 468e8d5648eSdan 469e8d5648eSdan if( bNew ){ 470cf8e9144Sdan rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal); 471e8d5648eSdan }else{ 472cf8e9144Sdan rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal); 473e8d5648eSdan } 47412ca0b56Sdan if( rc!=SQLITE_OK ) return rc; 475e8d5648eSdan 476e8d5648eSdan eType = sqlite3_value_type(pVal); 477798693b2Sdan h = sessionHashAppendType(h, eType); 4786734007dSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 479e8d5648eSdan i64 iVal; 480e8d5648eSdan if( eType==SQLITE_INTEGER ){ 481e8d5648eSdan iVal = sqlite3_value_int64(pVal); 482e8d5648eSdan }else{ 483e8d5648eSdan double rVal = sqlite3_value_double(pVal); 484e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 485e8d5648eSdan memcpy(&iVal, &rVal, 8); 486e8d5648eSdan } 487e8d5648eSdan h = sessionHashAppendI64(h, iVal); 4886734007dSdan }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 4896734007dSdan const u8 *z; 4903cc89d95Sdan int n; 4916734007dSdan if( eType==SQLITE_TEXT ){ 4926734007dSdan z = (const u8 *)sqlite3_value_text(pVal); 4936734007dSdan }else{ 4946734007dSdan z = (const u8 *)sqlite3_value_blob(pVal); 495e8d5648eSdan } 4963cc89d95Sdan n = sqlite3_value_bytes(pVal); 4973cc89d95Sdan if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 4983cc89d95Sdan h = sessionHashAppendBlob(h, n, z); 4996734007dSdan }else{ 50027453faeSdan assert( eType==SQLITE_NULL ); 5011611e5a3Sdan assert( pTab->bStat1==0 || i!=1 ); 50227453faeSdan *pbNullPK = 1; 503e8d5648eSdan } 504e8d5648eSdan } 505e8d5648eSdan } 506e8d5648eSdan 507e8d5648eSdan *piHash = (h % pTab->nChange); 508e8d5648eSdan return SQLITE_OK; 509e8d5648eSdan } 510e8d5648eSdan 5114131639cSdan /* 5126cda207fSdan ** The buffer that the argument points to contains a serialized SQL value. 5136cda207fSdan ** Return the number of bytes of space occupied by the value (including 5146cda207fSdan ** the type byte). 5156cda207fSdan */ 5166cda207fSdan static int sessionSerialLen(u8 *a){ 5176cda207fSdan int e = *a; 5186cda207fSdan int n; 5196cda207fSdan if( e==0 ) return 1; 5206cda207fSdan if( e==SQLITE_NULL ) return 1; 5216cda207fSdan if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9; 5226cda207fSdan return sessionVarintGet(&a[1], &n) + 1 + n; 5236cda207fSdan } 5246cda207fSdan 5256cda207fSdan /* 5265d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a 527798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys 5284131639cSdan ** calculated by this function are compatible with those calculated by 5294131639cSdan ** sessionPreupdateHash(). 53064277f4aSdan ** 53164277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from 53264277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely. 5334131639cSdan */ 5344131639cSdan static unsigned int sessionChangeHash( 5354131639cSdan SessionTable *pTab, /* Table handle */ 53664277f4aSdan int bPkOnly, /* Record consists of PK fields only */ 5375d607a6eSdan u8 *aRecord, /* Change record */ 5384131639cSdan int nBucket /* Assume this many buckets in hash table */ 539e8d5648eSdan ){ 5404131639cSdan unsigned int h = 0; /* Value to return */ 5414131639cSdan int i; /* Used to iterate through columns */ 5425d607a6eSdan u8 *a = aRecord; /* Used to iterate through change record */ 543e8d5648eSdan 544e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 5456cda207fSdan int eType = *a; 546e8d5648eSdan int isPK = pTab->abPK[i]; 54764277f4aSdan if( bPkOnly && isPK==0 ) continue; 548e8d5648eSdan 54927453faeSdan /* It is not possible for eType to be SQLITE_NULL here. The session 55027453faeSdan ** module does not record changes for rows with NULL values stored in 55127453faeSdan ** primary key columns. */ 55227453faeSdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 55327453faeSdan || eType==SQLITE_TEXT || eType==SQLITE_BLOB 5546cda207fSdan || eType==SQLITE_NULL || eType==0 55527453faeSdan ); 5561611e5a3Sdan assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) ); 55727453faeSdan 5586cda207fSdan if( isPK ){ 5596cda207fSdan a++; 560798693b2Sdan h = sessionHashAppendType(h, eType); 56127453faeSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 5626734007dSdan h = sessionHashAppendI64(h, sessionGetI64(a)); 563e8d5648eSdan a += 8; 5641611e5a3Sdan }else{ 565e8d5648eSdan int n; 566e8d5648eSdan a += sessionVarintGet(a, &n); 5676734007dSdan h = sessionHashAppendBlob(h, n, a); 568e8d5648eSdan a += n; 569e8d5648eSdan } 5706cda207fSdan }else{ 5716cda207fSdan a += sessionSerialLen(a); 5726cda207fSdan } 573e8d5648eSdan } 574e8d5648eSdan return (h % nBucket); 575e8d5648eSdan } 576e8d5648eSdan 577798693b2Sdan /* 578798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab. 579798693b2Sdan ** This function returns true if the two records apply to the same row (i.e. 580798693b2Sdan ** have the same values stored in the primary key columns), or false 581798693b2Sdan ** otherwise. 582798693b2Sdan */ 5835d607a6eSdan static int sessionChangeEqual( 584798693b2Sdan SessionTable *pTab, /* Table used for PK definition */ 585a71d2371Sdan int bLeftPkOnly, /* True if aLeft[] contains PK fields only */ 5865d607a6eSdan u8 *aLeft, /* Change record */ 587a71d2371Sdan int bRightPkOnly, /* True if aRight[] contains PK fields only */ 5885d607a6eSdan u8 *aRight /* Change record */ 5895d607a6eSdan ){ 590798693b2Sdan u8 *a1 = aLeft; /* Cursor to iterate through aLeft */ 591798693b2Sdan u8 *a2 = aRight; /* Cursor to iterate through aRight */ 592798693b2Sdan int iCol; /* Used to iterate through table columns */ 5935d607a6eSdan 594798693b2Sdan for(iCol=0; iCol<pTab->nCol; iCol++){ 59514faa061Sdan if( pTab->abPK[iCol] ){ 5965d607a6eSdan int n1 = sessionSerialLen(a1); 5975d607a6eSdan int n2 = sessionSerialLen(a2); 5985d607a6eSdan 599798693b2Sdan if( pTab->abPK[iCol] && (n1!=n2 || memcmp(a1, a2, n1)) ){ 6005d607a6eSdan return 0; 6015d607a6eSdan } 60214faa061Sdan a1 += n1; 60314faa061Sdan a2 += n2; 60414faa061Sdan }else{ 60514faa061Sdan if( bLeftPkOnly==0 ) a1 += sessionSerialLen(a1); 60614faa061Sdan if( bRightPkOnly==0 ) a2 += sessionSerialLen(a2); 60714faa061Sdan } 6085d607a6eSdan } 6095d607a6eSdan 6105d607a6eSdan return 1; 6115d607a6eSdan } 6125d607a6eSdan 613798693b2Sdan /* 614798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change 615798693b2Sdan ** records with nCol columns. This function "merges" the two records into 616798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is 617798693b2Sdan ** then set to point to one byte after the last byte written before 618798693b2Sdan ** returning. 619798693b2Sdan ** 620798693b2Sdan ** The merging of records is done as follows: For each column, if the 621798693b2Sdan ** aRight record contains a value for the column, copy the value from 622798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither 623798693b2Sdan ** record contains a value for a given column, then neither does the 624798693b2Sdan ** output record. 625798693b2Sdan */ 6265d607a6eSdan static void sessionMergeRecord( 6275d607a6eSdan u8 **paOut, 628798693b2Sdan int nCol, 6295d607a6eSdan u8 *aLeft, 6305d607a6eSdan u8 *aRight 6315d607a6eSdan ){ 632798693b2Sdan u8 *a1 = aLeft; /* Cursor used to iterate through aLeft */ 633798693b2Sdan u8 *a2 = aRight; /* Cursor used to iterate through aRight */ 634798693b2Sdan u8 *aOut = *paOut; /* Output cursor */ 635798693b2Sdan int iCol; /* Used to iterate from 0 to nCol */ 6365d607a6eSdan 637798693b2Sdan for(iCol=0; iCol<nCol; iCol++){ 6385d607a6eSdan int n1 = sessionSerialLen(a1); 6395d607a6eSdan int n2 = sessionSerialLen(a2); 6405d607a6eSdan if( *a2 ){ 6415d607a6eSdan memcpy(aOut, a2, n2); 6425d607a6eSdan aOut += n2; 6435d607a6eSdan }else{ 6445d607a6eSdan memcpy(aOut, a1, n1); 6455d607a6eSdan aOut += n1; 6465d607a6eSdan } 6475d607a6eSdan a1 += n1; 6485d607a6eSdan a2 += n2; 6495d607a6eSdan } 6505d607a6eSdan 6515d607a6eSdan *paOut = aOut; 6525d607a6eSdan } 6535d607a6eSdan 654798693b2Sdan /* 655798693b2Sdan ** This is a helper function used by sessionMergeUpdate(). 656798693b2Sdan ** 657798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value 658798693b2Sdan ** within a change record. Before it returns, both have been advanced so 659798693b2Sdan ** as to point to the next value in the record. 660798693b2Sdan ** 661798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e. 662fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo 663798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the 664798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal 665798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points 666fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words: 667fa29ecc4Sdan ** 668fa29ecc4Sdan ** if( *paTwo is valid ) return *paTwo; 669fa29ecc4Sdan ** return *paOne; 670fa29ecc4Sdan ** 671798693b2Sdan */ 6725d607a6eSdan static u8 *sessionMergeValue( 673798693b2Sdan u8 **paOne, /* IN/OUT: Left-hand buffer pointer */ 674798693b2Sdan u8 **paTwo, /* IN/OUT: Right-hand buffer pointer */ 675798693b2Sdan int *pnVal /* OUT: Bytes in returned value */ 6765d607a6eSdan ){ 6775d607a6eSdan u8 *a1 = *paOne; 6785d607a6eSdan u8 *a2 = *paTwo; 6795d607a6eSdan u8 *pRet = 0; 6805d607a6eSdan int n1; 6815d607a6eSdan 6825d607a6eSdan assert( a1 ); 6835d607a6eSdan if( a2 ){ 6845d607a6eSdan int n2 = sessionSerialLen(a2); 6855d607a6eSdan if( *a2 ){ 6865d607a6eSdan *pnVal = n2; 6875d607a6eSdan pRet = a2; 6885d607a6eSdan } 6895d607a6eSdan *paTwo = &a2[n2]; 6905d607a6eSdan } 6915d607a6eSdan 6925d607a6eSdan n1 = sessionSerialLen(a1); 6935d607a6eSdan if( pRet==0 ){ 6945d607a6eSdan *pnVal = n1; 6955d607a6eSdan pRet = a1; 6965d607a6eSdan } 6975d607a6eSdan *paOne = &a1[n1]; 6985d607a6eSdan 6995d607a6eSdan return pRet; 7005d607a6eSdan } 7015d607a6eSdan 702798693b2Sdan /* 703798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes 704798693b2Sdan ** on the same row. 705798693b2Sdan */ 7065d607a6eSdan static int sessionMergeUpdate( 707798693b2Sdan u8 **paOut, /* IN/OUT: Pointer to output buffer */ 708798693b2Sdan SessionTable *pTab, /* Table change pertains to */ 709a71d2371Sdan int bPatchset, /* True if records are patchset records */ 710798693b2Sdan u8 *aOldRecord1, /* old.* record for first change */ 711798693b2Sdan u8 *aOldRecord2, /* old.* record for second change */ 712798693b2Sdan u8 *aNewRecord1, /* new.* record for first change */ 713798693b2Sdan u8 *aNewRecord2 /* new.* record for second change */ 7145d607a6eSdan ){ 7155d607a6eSdan u8 *aOld1 = aOldRecord1; 7165d607a6eSdan u8 *aOld2 = aOldRecord2; 7175d607a6eSdan u8 *aNew1 = aNewRecord1; 7185d607a6eSdan u8 *aNew2 = aNewRecord2; 7195d607a6eSdan 7205d607a6eSdan u8 *aOut = *paOut; 7215d607a6eSdan int i; 72264277f4aSdan 72364277f4aSdan if( bPatchset==0 ){ 7245d607a6eSdan int bRequired = 0; 7255d607a6eSdan 7265d607a6eSdan assert( aOldRecord1 && aNewRecord1 ); 7275d607a6eSdan 7285d607a6eSdan /* Write the old.* vector first. */ 7295d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7305d607a6eSdan int nOld; 7315d607a6eSdan u8 *aOld; 7325d607a6eSdan int nNew; 7335d607a6eSdan u8 *aNew; 7345d607a6eSdan 7355d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 7365d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 7375d607a6eSdan if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){ 7385d607a6eSdan if( pTab->abPK[i]==0 ) bRequired = 1; 7395d607a6eSdan memcpy(aOut, aOld, nOld); 7405d607a6eSdan aOut += nOld; 7415d607a6eSdan }else{ 7425d607a6eSdan *(aOut++) = '\0'; 7435d607a6eSdan } 7445d607a6eSdan } 7455d607a6eSdan 7465d607a6eSdan if( !bRequired ) return 0; 74764277f4aSdan } 7485d607a6eSdan 7495d607a6eSdan /* Write the new.* vector */ 7505d607a6eSdan aOld1 = aOldRecord1; 7515d607a6eSdan aOld2 = aOldRecord2; 7525d607a6eSdan aNew1 = aNewRecord1; 7535d607a6eSdan aNew2 = aNewRecord2; 7545d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7555d607a6eSdan int nOld; 7565d607a6eSdan u8 *aOld; 7575d607a6eSdan int nNew; 7585d607a6eSdan u8 *aNew; 7595d607a6eSdan 7605d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 7615d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 76264277f4aSdan if( bPatchset==0 76364277f4aSdan && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew))) 76464277f4aSdan ){ 7655d607a6eSdan *(aOut++) = '\0'; 7665d607a6eSdan }else{ 7675d607a6eSdan memcpy(aOut, aNew, nNew); 7685d607a6eSdan aOut += nNew; 7695d607a6eSdan } 7705d607a6eSdan } 7715d607a6eSdan 7725d607a6eSdan *paOut = aOut; 7735d607a6eSdan return 1; 7745d607a6eSdan } 7755d607a6eSdan 77677fc1d5bSdan /* 77777fc1d5bSdan ** This function is only called from within a pre-update-hook callback. 77877fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row 77977fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise 78077fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns 78177fc1d5bSdan ** false. 78277fc1d5bSdan */ 78377fc1d5bSdan static int sessionPreupdateEqual( 784cf8e9144Sdan sqlite3_session *pSession, /* Session object that owns SessionTable */ 78577fc1d5bSdan SessionTable *pTab, /* Table associated with change */ 78677fc1d5bSdan SessionChange *pChange, /* Change to compare to */ 78777fc1d5bSdan int op /* Current pre-update operation */ 788e8d5648eSdan ){ 78977fc1d5bSdan int iCol; /* Used to iterate through columns */ 79077fc1d5bSdan u8 *a = pChange->aRecord; /* Cursor used to scan change record */ 791e8d5648eSdan 79277fc1d5bSdan assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE ); 79377fc1d5bSdan for(iCol=0; iCol<pTab->nCol; iCol++){ 79477fc1d5bSdan if( !pTab->abPK[iCol] ){ 795798693b2Sdan a += sessionSerialLen(a); 796e8d5648eSdan }else{ 7976734007dSdan sqlite3_value *pVal; /* Value returned by preupdate_new/old */ 7986734007dSdan int rc; /* Error code from preupdate_new/old */ 799798693b2Sdan int eType = *a++; /* Type of value from change record */ 8006734007dSdan 8016734007dSdan /* The following calls to preupdate_new() and preupdate_old() can not 8026734007dSdan ** fail. This is because they cache their return values, and by the 8036734007dSdan ** time control flows to here they have already been called once from 8046734007dSdan ** within sessionPreupdateHash(). The first two asserts below verify 8056734007dSdan ** this (that the method has already been called). */ 80677fc1d5bSdan if( op==SQLITE_INSERT ){ 807cf8e9144Sdan /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */ 808cf8e9144Sdan rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal); 809e8d5648eSdan }else{ 810cf8e9144Sdan /* assert( db->pPreUpdate->pUnpacked ); */ 811cf8e9144Sdan rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal); 812e8d5648eSdan } 8136734007dSdan assert( rc==SQLITE_OK ); 8141611e5a3Sdan if( sqlite3_value_type(pVal)!=eType ) return 0; 815e8d5648eSdan 81612ca0b56Sdan /* A SessionChange object never has a NULL value in a PK column */ 81712ca0b56Sdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 81812ca0b56Sdan || eType==SQLITE_BLOB || eType==SQLITE_TEXT 81912ca0b56Sdan ); 82012ca0b56Sdan 82112ca0b56Sdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 822e8d5648eSdan i64 iVal = sessionGetI64(a); 823e8d5648eSdan a += 8; 824e8d5648eSdan if( eType==SQLITE_INTEGER ){ 82577fc1d5bSdan if( sqlite3_value_int64(pVal)!=iVal ) return 0; 826e8d5648eSdan }else{ 827e8d5648eSdan double rVal; 828e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 829e8d5648eSdan memcpy(&rVal, &iVal, 8); 83077fc1d5bSdan if( sqlite3_value_double(pVal)!=rVal ) return 0; 831e8d5648eSdan } 83212ca0b56Sdan }else{ 833e8d5648eSdan int n; 834e8d5648eSdan const u8 *z; 835e8d5648eSdan a += sessionVarintGet(a, &n); 83677fc1d5bSdan if( sqlite3_value_bytes(pVal)!=n ) return 0; 83712ca0b56Sdan if( eType==SQLITE_TEXT ){ 83812ca0b56Sdan z = sqlite3_value_text(pVal); 83912ca0b56Sdan }else{ 84012ca0b56Sdan z = sqlite3_value_blob(pVal); 84112ca0b56Sdan } 84277fc1d5bSdan if( memcmp(a, z, n) ) return 0; 843e8d5648eSdan a += n; 844e8d5648eSdan } 845e8d5648eSdan } 846e8d5648eSdan } 847e8d5648eSdan 84877fc1d5bSdan return 1; 8494fccf43aSdan } 8504fccf43aSdan 8514fccf43aSdan /* 8524fccf43aSdan ** If required, grow the hash table used to store changes on table pTab 8534fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the 8544fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return 8554fccf43aSdan ** SQLITE_OK. 8564fccf43aSdan ** 8574fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In 8584fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway. 8594fccf43aSdan ** Growing the hash table in this case is a performance optimization only, 8604fccf43aSdan ** it is not required for correct operation. 8614fccf43aSdan */ 86264277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){ 8634fccf43aSdan if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){ 8644fccf43aSdan int i; 8654fccf43aSdan SessionChange **apNew; 8664fccf43aSdan int nNew = (pTab->nChange ? pTab->nChange : 128) * 2; 8674fccf43aSdan 8684fccf43aSdan apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew); 8694fccf43aSdan if( apNew==0 ){ 8704fccf43aSdan if( pTab->nChange==0 ){ 8714fccf43aSdan return SQLITE_ERROR; 8724fccf43aSdan } 8734fccf43aSdan return SQLITE_OK; 8744fccf43aSdan } 8754fccf43aSdan memset(apNew, 0, sizeof(SessionChange *) * nNew); 8764fccf43aSdan 8774fccf43aSdan for(i=0; i<pTab->nChange; i++){ 8784fccf43aSdan SessionChange *p; 8794fccf43aSdan SessionChange *pNext; 8804fccf43aSdan for(p=pTab->apChange[i]; p; p=pNext){ 88164277f4aSdan int bPkOnly = (p->op==SQLITE_DELETE && bPatchset); 88264277f4aSdan int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew); 8834fccf43aSdan pNext = p->pNext; 8844fccf43aSdan p->pNext = apNew[iHash]; 8854fccf43aSdan apNew[iHash] = p; 8864fccf43aSdan } 8874fccf43aSdan } 8884fccf43aSdan 8894fccf43aSdan sqlite3_free(pTab->apChange); 8904fccf43aSdan pTab->nChange = nNew; 8914fccf43aSdan pTab->apChange = apNew; 8924fccf43aSdan } 8934fccf43aSdan 8944fccf43aSdan return SQLITE_OK; 8954fccf43aSdan } 8964fccf43aSdan 897296c7658Sdan /* 898e8d5648eSdan ** This function queries the database for the names of the columns of table 8993f975373Sdrh ** zThis, in schema zDb. 900e8d5648eSdan ** 90177fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number 90277fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a 903e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to 904e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not 905e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding 906e8d5648eSdan ** column is part of the primary key. 907e8d5648eSdan ** 908e8d5648eSdan ** For example, if the table is declared as: 909e8d5648eSdan ** 910e8d5648eSdan ** CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z)); 911e8d5648eSdan ** 91277fc1d5bSdan ** Then the four output variables are populated as follows: 913e8d5648eSdan ** 91477fc1d5bSdan ** *pnCol = 4 915e8d5648eSdan ** *pzTab = "tbl1" 916e8d5648eSdan ** *pazCol = {"w", "x", "y", "z"} 917e8d5648eSdan ** *pabPK = {1, 0, 0, 1} 918e8d5648eSdan ** 919e8d5648eSdan ** All returned buffers are part of the same single allocation, which must 9203f975373Sdrh ** be freed using sqlite3_free() by the caller 921e8d5648eSdan */ 922e8d5648eSdan static int sessionTableInfo( 923e8d5648eSdan sqlite3 *db, /* Database connection */ 924e8d5648eSdan const char *zDb, /* Name of attached database (e.g. "main") */ 925e8d5648eSdan const char *zThis, /* Table name */ 926ca62ad57Sdan int *pnCol, /* OUT: number of columns */ 927e8d5648eSdan const char **pzTab, /* OUT: Copy of zThis */ 928e8d5648eSdan const char ***pazCol, /* OUT: Array of column names for table */ 929e8d5648eSdan u8 **pabPK /* OUT: Array of booleans - true for PK col */ 930e8d5648eSdan ){ 931e8d5648eSdan char *zPragma; 932e8d5648eSdan sqlite3_stmt *pStmt; 933e8d5648eSdan int rc; 934e8d5648eSdan int nByte; 935e8d5648eSdan int nDbCol = 0; 936e8d5648eSdan int nThis; 937e8d5648eSdan int i; 93874f598b6Smistachkin u8 *pAlloc = 0; 939db04571cSdan char **azCol = 0; 94074f598b6Smistachkin u8 *abPK = 0; 941e8d5648eSdan 942db04571cSdan assert( pazCol && pabPK ); 943e8d5648eSdan 944cfdbde21Sdrh nThis = sqlite3Strlen30(zThis); 945614efe2bSdan if( nThis==12 && 0==sqlite3_stricmp("sqlite_stat1", zThis) ){ 94658713184Sdan rc = sqlite3_table_column_metadata(db, zDb, zThis, 0, 0, 0, 0, 0, 0); 94758713184Sdan if( rc==SQLITE_OK ){ 948614efe2bSdan /* For sqlite_stat1, pretend that (tbl,idx) is the PRIMARY KEY. */ 949614efe2bSdan zPragma = sqlite3_mprintf( 950614efe2bSdan "SELECT 0, 'tbl', '', 0, '', 1 UNION ALL " 951614efe2bSdan "SELECT 1, 'idx', '', 0, '', 2 UNION ALL " 952614efe2bSdan "SELECT 2, 'stat', '', 0, '', 0" 953614efe2bSdan ); 95458713184Sdan }else if( rc==SQLITE_ERROR ){ 95558713184Sdan zPragma = sqlite3_mprintf(""); 95658713184Sdan }else{ 95758713184Sdan return rc; 95858713184Sdan } 959614efe2bSdan }else{ 960e8d5648eSdan zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis); 961614efe2bSdan } 962e8d5648eSdan if( !zPragma ) return SQLITE_NOMEM; 963e8d5648eSdan 964e8d5648eSdan rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0); 965e8d5648eSdan sqlite3_free(zPragma); 966e8d5648eSdan if( rc!=SQLITE_OK ) return rc; 967e8d5648eSdan 968e8d5648eSdan nByte = nThis + 1; 969e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 970e8d5648eSdan nByte += sqlite3_column_bytes(pStmt, 1); 971e8d5648eSdan nDbCol++; 972e8d5648eSdan } 973e8d5648eSdan rc = sqlite3_reset(pStmt); 974e8d5648eSdan 975e8d5648eSdan if( rc==SQLITE_OK ){ 976e8d5648eSdan nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1); 977e8d5648eSdan pAlloc = sqlite3_malloc(nByte); 978e8d5648eSdan if( pAlloc==0 ){ 979e8d5648eSdan rc = SQLITE_NOMEM; 980e8d5648eSdan } 981e8d5648eSdan } 982e8d5648eSdan if( rc==SQLITE_OK ){ 983e8d5648eSdan azCol = (char **)pAlloc; 984ca62ad57Sdan pAlloc = (u8 *)&azCol[nDbCol]; 985e8d5648eSdan abPK = (u8 *)pAlloc; 986ca62ad57Sdan pAlloc = &abPK[nDbCol]; 987e8d5648eSdan if( pzTab ){ 988e8d5648eSdan memcpy(pAlloc, zThis, nThis+1); 989e8d5648eSdan *pzTab = (char *)pAlloc; 990e8d5648eSdan pAlloc += nThis+1; 991e8d5648eSdan } 992e8d5648eSdan 993e8d5648eSdan i = 0; 994e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 995e8d5648eSdan int nName = sqlite3_column_bytes(pStmt, 1); 996e8d5648eSdan const unsigned char *zName = sqlite3_column_text(pStmt, 1); 997e8d5648eSdan if( zName==0 ) break; 998e8d5648eSdan memcpy(pAlloc, zName, nName+1); 999e8d5648eSdan azCol[i] = (char *)pAlloc; 1000e8d5648eSdan pAlloc += nName+1; 1001db04571cSdan abPK[i] = sqlite3_column_int(pStmt, 5); 1002e8d5648eSdan i++; 1003e8d5648eSdan } 1004e8d5648eSdan rc = sqlite3_reset(pStmt); 1005e8d5648eSdan 1006e8d5648eSdan } 1007e8d5648eSdan 1008e8d5648eSdan /* If successful, populate the output variables. Otherwise, zero them and 1009e8d5648eSdan ** free any allocation made. An error code will be returned in this case. 1010e8d5648eSdan */ 1011e8d5648eSdan if( rc==SQLITE_OK ){ 1012db04571cSdan *pazCol = (const char **)azCol; 1013db04571cSdan *pabPK = abPK; 1014ca62ad57Sdan *pnCol = nDbCol; 1015e8d5648eSdan }else{ 1016db04571cSdan *pazCol = 0; 1017db04571cSdan *pabPK = 0; 1018ca62ad57Sdan *pnCol = 0; 1019e8d5648eSdan if( pzTab ) *pzTab = 0; 1020db04571cSdan sqlite3_free(azCol); 1021e8d5648eSdan } 1022e8d5648eSdan sqlite3_finalize(pStmt); 1023e8d5648eSdan return rc; 1024e8d5648eSdan } 1025e8d5648eSdan 1026e8d5648eSdan /* 1027296c7658Sdan ** This function is only called from within a pre-update handler for a 1028296c7658Sdan ** write to table pTab, part of session pSession. If this is the first 10296dc29e60Sdan ** write to this table, initalize the SessionTable.nCol, azCol[] and 10306dc29e60Sdan ** abPK[] arrays accordingly. 1031296c7658Sdan ** 10326dc29e60Sdan ** If an error occurs, an error code is stored in sqlite3_session.rc and 10336dc29e60Sdan ** non-zero returned. Or, if no error occurs but the table has no primary 10346dc29e60Sdan ** key, sqlite3_session.rc is left set to SQLITE_OK and non-zero returned to 10356dc29e60Sdan ** indicate that updates on this table should be ignored. SessionTable.abPK 10366dc29e60Sdan ** is set to NULL in this case. 1037296c7658Sdan */ 10384fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){ 10394fccf43aSdan if( pTab->nCol==0 ){ 10406dc29e60Sdan u8 *abPK; 1041e8d5648eSdan assert( pTab->azCol==0 || pTab->abPK==0 ); 1042e8d5648eSdan pSession->rc = sessionTableInfo(pSession->db, pSession->zDb, 10436dc29e60Sdan pTab->zName, &pTab->nCol, 0, &pTab->azCol, &abPK 1044e8d5648eSdan ); 10456dc29e60Sdan if( pSession->rc==SQLITE_OK ){ 10466dc29e60Sdan int i; 10476dc29e60Sdan for(i=0; i<pTab->nCol; i++){ 10486dc29e60Sdan if( abPK[i] ){ 10496dc29e60Sdan pTab->abPK = abPK; 10506dc29e60Sdan break; 1051ca62ad57Sdan } 10524fccf43aSdan } 10533739f298Sdan if( 0==sqlite3_stricmp("sqlite_stat1", pTab->zName) ){ 10543739f298Sdan pTab->bStat1 = 1; 10553739f298Sdan } 10566dc29e60Sdan } 10576dc29e60Sdan } 10586dc29e60Sdan return (pSession->rc || pTab->abPK==0); 1059e8d5648eSdan } 1060e8d5648eSdan 106177fc1d5bSdan /* 10621611e5a3Sdan ** Versions of the four methods in object SessionHook for use with the 10631611e5a3Sdan ** sqlite_stat1 table. The purpose of this is to substitute a zero-length 10641611e5a3Sdan ** blob each time a NULL value is read from the "idx" column of the 10651611e5a3Sdan ** sqlite_stat1 table. 10661611e5a3Sdan */ 10671611e5a3Sdan typedef struct SessionStat1Ctx SessionStat1Ctx; 10681611e5a3Sdan struct SessionStat1Ctx { 10691611e5a3Sdan SessionHook hook; 10701611e5a3Sdan sqlite3_session *pSession; 10711611e5a3Sdan }; 10721611e5a3Sdan static int sessionStat1Old(void *pCtx, int iCol, sqlite3_value **ppVal){ 10731611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 10741611e5a3Sdan sqlite3_value *pVal = 0; 10751611e5a3Sdan int rc = p->hook.xOld(p->hook.pCtx, iCol, &pVal); 10761611e5a3Sdan if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){ 10771611e5a3Sdan pVal = p->pSession->pZeroBlob; 10781611e5a3Sdan } 10791611e5a3Sdan *ppVal = pVal; 10801611e5a3Sdan return rc; 10811611e5a3Sdan } 10821611e5a3Sdan static int sessionStat1New(void *pCtx, int iCol, sqlite3_value **ppVal){ 10831611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 10841611e5a3Sdan sqlite3_value *pVal = 0; 10851611e5a3Sdan int rc = p->hook.xNew(p->hook.pCtx, iCol, &pVal); 10861611e5a3Sdan if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){ 10871611e5a3Sdan pVal = p->pSession->pZeroBlob; 10881611e5a3Sdan } 10891611e5a3Sdan *ppVal = pVal; 10901611e5a3Sdan return rc; 10911611e5a3Sdan } 10921611e5a3Sdan static int sessionStat1Count(void *pCtx){ 10931611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 10941611e5a3Sdan return p->hook.xCount(p->hook.pCtx); 10951611e5a3Sdan } 10961611e5a3Sdan static int sessionStat1Depth(void *pCtx){ 10971611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 10981611e5a3Sdan return p->hook.xDepth(p->hook.pCtx); 10991611e5a3Sdan } 11001611e5a3Sdan 11011611e5a3Sdan 11021611e5a3Sdan /* 110377fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a 110477fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change 110577fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument. 110677fc1d5bSdan ** 110777fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added 110877fc1d5bSdan ** to the changed-rows hash table associated with table pTab. 110977fc1d5bSdan */ 1110e8d5648eSdan static void sessionPreupdateOneChange( 111177fc1d5bSdan int op, /* One of SQLITE_UPDATE, INSERT, DELETE */ 111277fc1d5bSdan sqlite3_session *pSession, /* Session object pTab is attached to */ 111377fc1d5bSdan SessionTable *pTab /* Table that change applies to */ 1114e8d5648eSdan ){ 1115e8d5648eSdan int iHash; 1116cf8e9144Sdan int bNull = 0; 1117e8d5648eSdan int rc = SQLITE_OK; 1118*fa3d7855Smistachkin SessionStat1Ctx stat1 = {0}; 1119e8d5648eSdan 1120e8d5648eSdan if( pSession->rc ) return; 1121e8d5648eSdan 1122e8d5648eSdan /* Load table details if required */ 1123e8d5648eSdan if( sessionInitTable(pSession, pTab) ) return; 1124e8d5648eSdan 11256dc29e60Sdan /* Check the number of columns in this xPreUpdate call matches the 11266dc29e60Sdan ** number of columns in the table. */ 11276dc29e60Sdan if( pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx) ){ 11286dc29e60Sdan pSession->rc = SQLITE_SCHEMA; 11296dc29e60Sdan return; 11306dc29e60Sdan } 11316dc29e60Sdan 1132e8d5648eSdan /* Grow the hash table if required */ 113364277f4aSdan if( sessionGrowHash(0, pTab) ){ 11345d607a6eSdan pSession->rc = SQLITE_NOMEM; 11355d607a6eSdan return; 11365d607a6eSdan } 1137e8d5648eSdan 11381611e5a3Sdan if( pTab->bStat1 ){ 11391611e5a3Sdan stat1.hook = pSession->hook; 11401611e5a3Sdan stat1.pSession = pSession; 11411611e5a3Sdan pSession->hook.pCtx = (void*)&stat1; 11421611e5a3Sdan pSession->hook.xNew = sessionStat1New; 11431611e5a3Sdan pSession->hook.xOld = sessionStat1Old; 11441611e5a3Sdan pSession->hook.xCount = sessionStat1Count; 11451611e5a3Sdan pSession->hook.xDepth = sessionStat1Depth; 11461611e5a3Sdan if( pSession->pZeroBlob==0 ){ 11471611e5a3Sdan sqlite3_value *p = sqlite3ValueNew(0); 11481611e5a3Sdan if( p==0 ){ 11491611e5a3Sdan rc = SQLITE_NOMEM; 11501611e5a3Sdan goto error_out; 11511611e5a3Sdan } 11521611e5a3Sdan sqlite3ValueSetStr(p, 0, "", 0, SQLITE_STATIC); 11531611e5a3Sdan pSession->pZeroBlob = p; 11541611e5a3Sdan } 11551611e5a3Sdan } 11561611e5a3Sdan 115780fe2d93Sdan /* Calculate the hash-key for this change. If the primary key of the row 115880fe2d93Sdan ** includes a NULL value, exit early. Such changes are ignored by the 115980fe2d93Sdan ** session module. */ 1160cf8e9144Sdan rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull); 116180fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 116280fe2d93Sdan 11631611e5a3Sdan if( bNull==0 ){ 116480fe2d93Sdan /* Search the hash table for an existing record for this row. */ 1165b4480e94Sdan SessionChange *pC; 11666734007dSdan for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){ 1167cf8e9144Sdan if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break; 1168e8d5648eSdan } 116980fe2d93Sdan 1170e8d5648eSdan if( pC==0 ){ 1171e8d5648eSdan /* Create a new change object containing all the old values (if 1172e8d5648eSdan ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK 1173e8d5648eSdan ** values (if this is an INSERT). */ 1174b4480e94Sdan SessionChange *pChange; /* New change object */ 1175e8d5648eSdan int nByte; /* Number of bytes to allocate */ 1176e8d5648eSdan int i; /* Used to iterate through columns */ 1177e8d5648eSdan 1178b4480e94Sdan assert( rc==SQLITE_OK ); 1179e8d5648eSdan pTab->nEntry++; 1180e8d5648eSdan 1181e8d5648eSdan /* Figure out how large an allocation is required */ 1182e8d5648eSdan nByte = sizeof(SessionChange); 118380fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1184e8d5648eSdan sqlite3_value *p = 0; 1185e8d5648eSdan if( op!=SQLITE_INSERT ){ 1186cf8e9144Sdan TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p); 118780fe2d93Sdan assert( trc==SQLITE_OK ); 118880fe2d93Sdan }else if( pTab->abPK[i] ){ 1189cf8e9144Sdan TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p); 119080fe2d93Sdan assert( trc==SQLITE_OK ); 1191e8d5648eSdan } 119280fe2d93Sdan 119380fe2d93Sdan /* This may fail if SQLite value p contains a utf-16 string that must 119480fe2d93Sdan ** be converted to utf-8 and an OOM error occurs while doing so. */ 1195e8d5648eSdan rc = sessionSerializeValue(0, p, &nByte); 119680fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 1197e8d5648eSdan } 1198e8d5648eSdan 1199e8d5648eSdan /* Allocate the change object */ 1200e8d5648eSdan pChange = (SessionChange *)sqlite3_malloc(nByte); 1201e8d5648eSdan if( !pChange ){ 1202e8d5648eSdan rc = SQLITE_NOMEM; 120380fe2d93Sdan goto error_out; 1204e8d5648eSdan }else{ 1205e8d5648eSdan memset(pChange, 0, sizeof(SessionChange)); 1206e8d5648eSdan pChange->aRecord = (u8 *)&pChange[1]; 1207e8d5648eSdan } 1208e8d5648eSdan 120980fe2d93Sdan /* Populate the change object. None of the preupdate_old(), 121080fe2d93Sdan ** preupdate_new() or SerializeValue() calls below may fail as all 121180fe2d93Sdan ** required values and encodings have already been cached in memory. 121280fe2d93Sdan ** It is not possible for an OOM to occur in this block. */ 1213e8d5648eSdan nByte = 0; 121480fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1215e8d5648eSdan sqlite3_value *p = 0; 1216e8d5648eSdan if( op!=SQLITE_INSERT ){ 1217cf8e9144Sdan pSession->hook.xOld(pSession->hook.pCtx, i, &p); 121880fe2d93Sdan }else if( pTab->abPK[i] ){ 1219cf8e9144Sdan pSession->hook.xNew(pSession->hook.pCtx, i, &p); 1220e8d5648eSdan } 122180fe2d93Sdan sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte); 1222e8d5648eSdan } 122380fe2d93Sdan 122480fe2d93Sdan /* Add the change to the hash-table */ 1225cf8e9144Sdan if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){ 1226b4480e94Sdan pChange->bIndirect = 1; 1227b4480e94Sdan } 122812ca0b56Sdan pChange->nRecord = nByte; 1229798693b2Sdan pChange->op = op; 1230e8d5648eSdan pChange->pNext = pTab->apChange[iHash]; 1231e8d5648eSdan pTab->apChange[iHash] = pChange; 123280fe2d93Sdan 123380fe2d93Sdan }else if( pC->bIndirect ){ 1234b4480e94Sdan /* If the existing change is considered "indirect", but this current 1235b4480e94Sdan ** change is "direct", mark the change object as direct. */ 1236cf8e9144Sdan if( pSession->hook.xDepth(pSession->hook.pCtx)==0 1237cf8e9144Sdan && pSession->bIndirect==0 1238cf8e9144Sdan ){ 1239b4480e94Sdan pC->bIndirect = 0; 1240b4480e94Sdan } 1241e8d5648eSdan } 12424fccf43aSdan } 124312ca0b56Sdan 124412ca0b56Sdan /* If an error has occurred, mark the session object as failed. */ 124580fe2d93Sdan error_out: 12461611e5a3Sdan if( pTab->bStat1 ){ 12471611e5a3Sdan pSession->hook = stat1.hook; 12481611e5a3Sdan } 124912ca0b56Sdan if( rc!=SQLITE_OK ){ 125012ca0b56Sdan pSession->rc = rc; 125112ca0b56Sdan } 125227453faeSdan } 12534fccf43aSdan 1254cf8e9144Sdan static int sessionFindTable( 1255cf8e9144Sdan sqlite3_session *pSession, 1256cf8e9144Sdan const char *zName, 1257cf8e9144Sdan SessionTable **ppTab 1258cf8e9144Sdan ){ 1259cf8e9144Sdan int rc = SQLITE_OK; 1260cf8e9144Sdan int nName = sqlite3Strlen30(zName); 1261cf8e9144Sdan SessionTable *pRet; 1262cf8e9144Sdan 1263cf8e9144Sdan /* Search for an existing table */ 1264cf8e9144Sdan for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){ 1265cf8e9144Sdan if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break; 1266cf8e9144Sdan } 1267cf8e9144Sdan 1268cf8e9144Sdan if( pRet==0 && pSession->bAutoAttach ){ 1269cf8e9144Sdan /* If there is a table-filter configured, invoke it. If it returns 0, 1270cf8e9144Sdan ** do not automatically add the new table. */ 1271cf8e9144Sdan if( pSession->xTableFilter==0 1272cf8e9144Sdan || pSession->xTableFilter(pSession->pFilterCtx, zName) 1273cf8e9144Sdan ){ 1274cf8e9144Sdan rc = sqlite3session_attach(pSession, zName); 1275cf8e9144Sdan if( rc==SQLITE_OK ){ 12766c39e6a8Sdan for(pRet=pSession->pTable; pRet->pNext; pRet=pRet->pNext); 1277cf8e9144Sdan assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ); 1278cf8e9144Sdan } 1279cf8e9144Sdan } 1280cf8e9144Sdan } 1281cf8e9144Sdan 1282cf8e9144Sdan assert( rc==SQLITE_OK || pRet==0 ); 1283cf8e9144Sdan *ppTab = pRet; 1284cf8e9144Sdan return rc; 1285cf8e9144Sdan } 1286cf8e9144Sdan 12874fccf43aSdan /* 12884fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases. 12894fccf43aSdan */ 12904fccf43aSdan static void xPreUpdate( 12914fccf43aSdan void *pCtx, /* Copy of third arg to preupdate_hook() */ 12924fccf43aSdan sqlite3 *db, /* Database handle */ 12934fccf43aSdan int op, /* SQLITE_UPDATE, DELETE or INSERT */ 12944fccf43aSdan char const *zDb, /* Database name */ 12954fccf43aSdan char const *zName, /* Table name */ 12964fccf43aSdan sqlite3_int64 iKey1, /* Rowid of row about to be deleted/updated */ 12974fccf43aSdan sqlite3_int64 iKey2 /* New rowid value (for a rowid UPDATE) */ 12984fccf43aSdan ){ 12994fccf43aSdan sqlite3_session *pSession; 1300cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); 13014fccf43aSdan 13024c220252Sdan assert( sqlite3_mutex_held(db->mutex) ); 13034c220252Sdan 13044fccf43aSdan for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){ 13054fccf43aSdan SessionTable *pTab; 1306296c7658Sdan 1307e8d5648eSdan /* If this session is attached to a different database ("main", "temp" 1308e8d5648eSdan ** etc.), or if it is not currently enabled, there is nothing to do. Skip 1309e8d5648eSdan ** to the next session object attached to this database. */ 1310296c7658Sdan if( pSession->bEnable==0 ) continue; 13114fccf43aSdan if( pSession->rc ) continue; 13124fccf43aSdan if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue; 1313296c7658Sdan 1314cf8e9144Sdan pSession->rc = sessionFindTable(pSession, zName, &pTab); 1315cf8e9144Sdan if( pTab ){ 1316cf8e9144Sdan assert( pSession->rc==SQLITE_OK ); 1317e8d5648eSdan sessionPreupdateOneChange(op, pSession, pTab); 1318e8d5648eSdan if( op==SQLITE_UPDATE ){ 1319e8d5648eSdan sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab); 13204fccf43aSdan } 13214fccf43aSdan } 13224fccf43aSdan } 13234fccf43aSdan } 1324cf8e9144Sdan 1325cf8e9144Sdan /* 1326cf8e9144Sdan ** The pre-update hook implementations. 1327cf8e9144Sdan */ 1328cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){ 1329cf8e9144Sdan return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal); 1330cf8e9144Sdan } 1331cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){ 1332cf8e9144Sdan return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal); 1333cf8e9144Sdan } 1334cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){ 1335cf8e9144Sdan return sqlite3_preupdate_count((sqlite3*)pCtx); 1336cf8e9144Sdan } 1337cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){ 1338cf8e9144Sdan return sqlite3_preupdate_depth((sqlite3*)pCtx); 1339cf8e9144Sdan } 1340cf8e9144Sdan 1341cf8e9144Sdan /* 1342cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only 1343cf8e9144Sdan ** argument. 1344cf8e9144Sdan */ 1345cf8e9144Sdan static void sessionPreupdateHooks( 1346cf8e9144Sdan sqlite3_session *pSession 1347cf8e9144Sdan ){ 1348cf8e9144Sdan pSession->hook.pCtx = (void*)pSession->db; 1349cf8e9144Sdan pSession->hook.xOld = sessionPreupdateOld; 1350cf8e9144Sdan pSession->hook.xNew = sessionPreupdateNew; 1351cf8e9144Sdan pSession->hook.xCount = sessionPreupdateCount; 1352cf8e9144Sdan pSession->hook.xDepth = sessionPreupdateDepth; 1353cf8e9144Sdan } 1354cf8e9144Sdan 1355cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx; 1356cf8e9144Sdan struct SessionDiffCtx { 1357cf8e9144Sdan sqlite3_stmt *pStmt; 1358cf8e9144Sdan int nOldOff; 1359cf8e9144Sdan }; 1360cf8e9144Sdan 1361cf8e9144Sdan /* 1362cf8e9144Sdan ** The diff hook implementations. 1363cf8e9144Sdan */ 1364cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){ 1365cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1366cf8e9144Sdan *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff); 1367cf8e9144Sdan return SQLITE_OK; 1368cf8e9144Sdan } 1369cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){ 1370cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1371cf8e9144Sdan *ppVal = sqlite3_column_value(p->pStmt, iVal); 1372cf8e9144Sdan return SQLITE_OK; 1373cf8e9144Sdan } 1374cf8e9144Sdan static int sessionDiffCount(void *pCtx){ 1375cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1376cf8e9144Sdan return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt); 1377cf8e9144Sdan } 1378cf8e9144Sdan static int sessionDiffDepth(void *pCtx){ 1379cf8e9144Sdan return 0; 1380cf8e9144Sdan } 1381cf8e9144Sdan 1382cf8e9144Sdan /* 1383cf8e9144Sdan ** Install the diff hooks on the session object passed as the only 1384cf8e9144Sdan ** argument. 1385cf8e9144Sdan */ 1386cf8e9144Sdan static void sessionDiffHooks( 1387cf8e9144Sdan sqlite3_session *pSession, 1388cf8e9144Sdan SessionDiffCtx *pDiffCtx 1389cf8e9144Sdan ){ 1390cf8e9144Sdan pSession->hook.pCtx = (void*)pDiffCtx; 1391cf8e9144Sdan pSession->hook.xOld = sessionDiffOld; 1392cf8e9144Sdan pSession->hook.xNew = sessionDiffNew; 1393cf8e9144Sdan pSession->hook.xCount = sessionDiffCount; 1394cf8e9144Sdan pSession->hook.xDepth = sessionDiffDepth; 1395cf8e9144Sdan } 1396cf8e9144Sdan 1397cf8e9144Sdan static char *sessionExprComparePK( 1398cf8e9144Sdan int nCol, 1399cf8e9144Sdan const char *zDb1, const char *zDb2, 1400cf8e9144Sdan const char *zTab, 1401cf8e9144Sdan const char **azCol, u8 *abPK 1402cf8e9144Sdan ){ 1403cf8e9144Sdan int i; 1404cf8e9144Sdan const char *zSep = ""; 1405cf8e9144Sdan char *zRet = 0; 1406cf8e9144Sdan 1407cf8e9144Sdan for(i=0; i<nCol; i++){ 1408cf8e9144Sdan if( abPK[i] ){ 1409cf8e9144Sdan zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"", 1410cf8e9144Sdan zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i] 1411cf8e9144Sdan ); 1412cf8e9144Sdan zSep = " AND "; 1413cf8e9144Sdan if( zRet==0 ) break; 1414cf8e9144Sdan } 1415cf8e9144Sdan } 1416cf8e9144Sdan 1417cf8e9144Sdan return zRet; 1418cf8e9144Sdan } 1419cf8e9144Sdan 1420cf8e9144Sdan static char *sessionExprCompareOther( 1421cf8e9144Sdan int nCol, 1422cf8e9144Sdan const char *zDb1, const char *zDb2, 1423cf8e9144Sdan const char *zTab, 1424cf8e9144Sdan const char **azCol, u8 *abPK 1425cf8e9144Sdan ){ 1426cf8e9144Sdan int i; 1427cf8e9144Sdan const char *zSep = ""; 1428cf8e9144Sdan char *zRet = 0; 1429cf8e9144Sdan int bHave = 0; 1430cf8e9144Sdan 1431cf8e9144Sdan for(i=0; i<nCol; i++){ 1432cf8e9144Sdan if( abPK[i]==0 ){ 1433cf8e9144Sdan bHave = 1; 1434cf8e9144Sdan zRet = sqlite3_mprintf( 1435cf8e9144Sdan "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"", 1436cf8e9144Sdan zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i] 1437cf8e9144Sdan ); 1438cf8e9144Sdan zSep = " OR "; 1439cf8e9144Sdan if( zRet==0 ) break; 1440cf8e9144Sdan } 1441cf8e9144Sdan } 1442cf8e9144Sdan 1443cf8e9144Sdan if( bHave==0 ){ 1444cf8e9144Sdan assert( zRet==0 ); 1445cf8e9144Sdan zRet = sqlite3_mprintf("0"); 1446cf8e9144Sdan } 1447cf8e9144Sdan 1448cf8e9144Sdan return zRet; 1449cf8e9144Sdan } 1450cf8e9144Sdan 1451cf8e9144Sdan static char *sessionSelectFindNew( 1452cf8e9144Sdan int nCol, 1453cf8e9144Sdan const char *zDb1, /* Pick rows in this db only */ 1454cf8e9144Sdan const char *zDb2, /* But not in this one */ 1455cf8e9144Sdan const char *zTbl, /* Table name */ 1456cf8e9144Sdan const char *zExpr 1457cf8e9144Sdan ){ 1458cf8e9144Sdan char *zRet = sqlite3_mprintf( 1459cf8e9144Sdan "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS (" 1460cf8e9144Sdan " SELECT 1 FROM \"%w\".\"%w\" WHERE %s" 1461cf8e9144Sdan ")", 1462cf8e9144Sdan zDb1, zTbl, zDb2, zTbl, zExpr 1463cf8e9144Sdan ); 1464cf8e9144Sdan return zRet; 1465cf8e9144Sdan } 1466cf8e9144Sdan 1467cf8e9144Sdan static int sessionDiffFindNew( 1468cf8e9144Sdan int op, 1469cf8e9144Sdan sqlite3_session *pSession, 1470cf8e9144Sdan SessionTable *pTab, 1471cf8e9144Sdan const char *zDb1, 1472cf8e9144Sdan const char *zDb2, 1473cf8e9144Sdan char *zExpr 1474cf8e9144Sdan ){ 1475cf8e9144Sdan int rc = SQLITE_OK; 1476cf8e9144Sdan char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr); 1477cf8e9144Sdan 1478cf8e9144Sdan if( zStmt==0 ){ 1479cf8e9144Sdan rc = SQLITE_NOMEM; 1480cf8e9144Sdan }else{ 1481cf8e9144Sdan sqlite3_stmt *pStmt; 1482cf8e9144Sdan rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0); 1483cf8e9144Sdan if( rc==SQLITE_OK ){ 1484cf8e9144Sdan SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx; 1485cf8e9144Sdan pDiffCtx->pStmt = pStmt; 1486cf8e9144Sdan pDiffCtx->nOldOff = 0; 1487cf8e9144Sdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1488cf8e9144Sdan sessionPreupdateOneChange(op, pSession, pTab); 1489cf8e9144Sdan } 1490cf8e9144Sdan rc = sqlite3_finalize(pStmt); 1491cf8e9144Sdan } 1492cf8e9144Sdan sqlite3_free(zStmt); 1493cf8e9144Sdan } 1494cf8e9144Sdan 1495cf8e9144Sdan return rc; 1496cf8e9144Sdan } 1497cf8e9144Sdan 1498cf8e9144Sdan static int sessionDiffFindModified( 1499cf8e9144Sdan sqlite3_session *pSession, 1500cf8e9144Sdan SessionTable *pTab, 1501cf8e9144Sdan const char *zFrom, 1502cf8e9144Sdan const char *zExpr 1503cf8e9144Sdan ){ 1504cf8e9144Sdan int rc = SQLITE_OK; 1505cf8e9144Sdan 1506cf8e9144Sdan char *zExpr2 = sessionExprCompareOther(pTab->nCol, 1507cf8e9144Sdan pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK 1508cf8e9144Sdan ); 1509cf8e9144Sdan if( zExpr2==0 ){ 1510cf8e9144Sdan rc = SQLITE_NOMEM; 1511cf8e9144Sdan }else{ 1512cf8e9144Sdan char *zStmt = sqlite3_mprintf( 1513dd009f83Sdan "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)", 1514cf8e9144Sdan pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2 1515cf8e9144Sdan ); 1516cf8e9144Sdan if( zStmt==0 ){ 1517cf8e9144Sdan rc = SQLITE_NOMEM; 1518cf8e9144Sdan }else{ 1519cf8e9144Sdan sqlite3_stmt *pStmt; 1520cf8e9144Sdan rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0); 1521cf8e9144Sdan 1522cf8e9144Sdan if( rc==SQLITE_OK ){ 1523cf8e9144Sdan SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx; 1524cf8e9144Sdan pDiffCtx->pStmt = pStmt; 1525cf8e9144Sdan pDiffCtx->nOldOff = pTab->nCol; 1526cf8e9144Sdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1527cf8e9144Sdan sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab); 1528cf8e9144Sdan } 1529cf8e9144Sdan rc = sqlite3_finalize(pStmt); 1530cf8e9144Sdan } 1531cf8e9144Sdan sqlite3_free(zStmt); 1532cf8e9144Sdan } 1533cf8e9144Sdan } 1534cf8e9144Sdan 1535cf8e9144Sdan return rc; 1536cf8e9144Sdan } 1537cf8e9144Sdan 1538cf8e9144Sdan int sqlite3session_diff( 1539cf8e9144Sdan sqlite3_session *pSession, 1540cf8e9144Sdan const char *zFrom, 1541cf8e9144Sdan const char *zTbl, 1542cf8e9144Sdan char **pzErrMsg 1543cf8e9144Sdan ){ 1544cf8e9144Sdan const char *zDb = pSession->zDb; 1545cf8e9144Sdan int rc = pSession->rc; 1546cf8e9144Sdan SessionDiffCtx d; 1547cf8e9144Sdan 1548cf8e9144Sdan memset(&d, 0, sizeof(d)); 1549cf8e9144Sdan sessionDiffHooks(pSession, &d); 1550cf8e9144Sdan 155110dc553cSdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1552cf8e9144Sdan if( pzErrMsg ) *pzErrMsg = 0; 1553cf8e9144Sdan if( rc==SQLITE_OK ){ 1554cf8e9144Sdan char *zExpr = 0; 1555cf8e9144Sdan sqlite3 *db = pSession->db; 1556cf8e9144Sdan SessionTable *pTo; /* Table zTbl */ 1557cf8e9144Sdan 1558cf8e9144Sdan /* Locate and if necessary initialize the target table object */ 1559cf8e9144Sdan rc = sessionFindTable(pSession, zTbl, &pTo); 1560cf8e9144Sdan if( pTo==0 ) goto diff_out; 15616dc29e60Sdan if( sessionInitTable(pSession, pTo) ){ 15626dc29e60Sdan rc = pSession->rc; 15636dc29e60Sdan goto diff_out; 1564cf8e9144Sdan } 1565cf8e9144Sdan 1566cf8e9144Sdan /* Check the table schemas match */ 1567cf8e9144Sdan if( rc==SQLITE_OK ){ 15684cc923e3Sdan int bHasPk = 0; 1569b9db9099Sdan int bMismatch = 0; 1570cf8e9144Sdan int nCol; /* Columns in zFrom.zTbl */ 1571cf8e9144Sdan u8 *abPK; 1572cf8e9144Sdan const char **azCol = 0; 1573cf8e9144Sdan rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK); 1574cf8e9144Sdan if( rc==SQLITE_OK ){ 15754cc923e3Sdan if( pTo->nCol!=nCol ){ 1576cf8e9144Sdan bMismatch = 1; 1577cf8e9144Sdan }else{ 1578cf8e9144Sdan int i; 1579cf8e9144Sdan for(i=0; i<nCol; i++){ 15804cc923e3Sdan if( pTo->abPK[i]!=abPK[i] ) bMismatch = 1; 1581cf8e9144Sdan if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1; 15824cc923e3Sdan if( abPK[i] ) bHasPk = 1; 1583cf8e9144Sdan } 1584cf8e9144Sdan } 1585cf8e9144Sdan } 1586dbbd8160Sdrh sqlite3_free((char*)azCol); 1587b9db9099Sdan if( bMismatch ){ 1588b9db9099Sdan *pzErrMsg = sqlite3_mprintf("table schemas do not match"); 1589b9db9099Sdan rc = SQLITE_SCHEMA; 1590b9db9099Sdan } 1591b9db9099Sdan if( bHasPk==0 ){ 1592b9db9099Sdan /* Ignore tables with no primary keys */ 1593b9db9099Sdan goto diff_out; 1594b9db9099Sdan } 1595cf8e9144Sdan } 1596cf8e9144Sdan 1597cf8e9144Sdan if( rc==SQLITE_OK ){ 1598cf8e9144Sdan zExpr = sessionExprComparePK(pTo->nCol, 1599cf8e9144Sdan zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK 1600cf8e9144Sdan ); 1601cf8e9144Sdan } 1602cf8e9144Sdan 1603cf8e9144Sdan /* Find new rows */ 1604cf8e9144Sdan if( rc==SQLITE_OK ){ 1605cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr); 1606cf8e9144Sdan } 1607cf8e9144Sdan 1608cf8e9144Sdan /* Find old rows */ 1609cf8e9144Sdan if( rc==SQLITE_OK ){ 1610cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr); 1611cf8e9144Sdan } 1612cf8e9144Sdan 1613cf8e9144Sdan /* Find modified rows */ 1614cf8e9144Sdan if( rc==SQLITE_OK ){ 1615cf8e9144Sdan rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr); 1616cf8e9144Sdan } 1617cf8e9144Sdan 1618cf8e9144Sdan sqlite3_free(zExpr); 1619cf8e9144Sdan } 1620cf8e9144Sdan 1621cf8e9144Sdan diff_out: 1622cf8e9144Sdan sessionPreupdateHooks(pSession); 162310dc553cSdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 1624cf8e9144Sdan return rc; 1625296c7658Sdan } 16264fccf43aSdan 16274fccf43aSdan /* 16284fccf43aSdan ** Create a session object. This session object will record changes to 16294fccf43aSdan ** database zDb attached to connection db. 16304fccf43aSdan */ 16314fccf43aSdan int sqlite3session_create( 16324fccf43aSdan sqlite3 *db, /* Database handle */ 16334fccf43aSdan const char *zDb, /* Name of db (e.g. "main") */ 16344fccf43aSdan sqlite3_session **ppSession /* OUT: New session object */ 16354fccf43aSdan ){ 1636296c7658Sdan sqlite3_session *pNew; /* Newly allocated session object */ 1637296c7658Sdan sqlite3_session *pOld; /* Session object already attached to db */ 1638cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */ 16394fccf43aSdan 1640296c7658Sdan /* Zero the output value in case an error occurs. */ 16414fccf43aSdan *ppSession = 0; 16424fccf43aSdan 16434fccf43aSdan /* Allocate and populate the new session object. */ 16444fccf43aSdan pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1); 16454fccf43aSdan if( !pNew ) return SQLITE_NOMEM; 16464fccf43aSdan memset(pNew, 0, sizeof(sqlite3_session)); 16474fccf43aSdan pNew->db = db; 16484fccf43aSdan pNew->zDb = (char *)&pNew[1]; 1649296c7658Sdan pNew->bEnable = 1; 16504fccf43aSdan memcpy(pNew->zDb, zDb, nDb+1); 1651cf8e9144Sdan sessionPreupdateHooks(pNew); 16524fccf43aSdan 16534fccf43aSdan /* Add the new session object to the linked list of session objects 16544fccf43aSdan ** attached to database handle $db. Do this under the cover of the db 16554fccf43aSdan ** handle mutex. */ 16564fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 16574fccf43aSdan pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew); 16584fccf43aSdan pNew->pNext = pOld; 16594fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 16604fccf43aSdan 16614fccf43aSdan *ppSession = pNew; 16624fccf43aSdan return SQLITE_OK; 16634fccf43aSdan } 16644fccf43aSdan 166577fc1d5bSdan /* 166677fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents 166777fc1d5bSdan ** of the changed-rows hash tables are also deleted. 166877fc1d5bSdan */ 16691ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){ 16705d607a6eSdan SessionTable *pNext; 16715d607a6eSdan SessionTable *pTab; 16725d607a6eSdan 16735d607a6eSdan for(pTab=pList; pTab; pTab=pNext){ 16745d607a6eSdan int i; 16755d607a6eSdan pNext = pTab->pNext; 16765d607a6eSdan for(i=0; i<pTab->nChange; i++){ 16775d607a6eSdan SessionChange *p; 167802d436b1Smistachkin SessionChange *pNextChange; 167902d436b1Smistachkin for(p=pTab->apChange[i]; p; p=pNextChange){ 168002d436b1Smistachkin pNextChange = p->pNext; 16815d607a6eSdan sqlite3_free(p); 16825d607a6eSdan } 16835d607a6eSdan } 16845d607a6eSdan sqlite3_free((char*)pTab->azCol); /* cast works around VC++ bug */ 16855d607a6eSdan sqlite3_free(pTab->apChange); 16865d607a6eSdan sqlite3_free(pTab); 16875d607a6eSdan } 16885d607a6eSdan } 16895d607a6eSdan 16904fccf43aSdan /* 16914fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create(). 16924fccf43aSdan */ 16934fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){ 16944fccf43aSdan sqlite3 *db = pSession->db; 16954fccf43aSdan sqlite3_session *pHead; 16964fccf43aSdan sqlite3_session **pp; 16974fccf43aSdan 1698296c7658Sdan /* Unlink the session from the linked list of sessions attached to the 1699296c7658Sdan ** database handle. Hold the db mutex while doing so. */ 17004fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 17014fccf43aSdan pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0); 170250d348b1Sdrh for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){ 170350d348b1Sdrh if( (*pp)==pSession ){ 17044fccf43aSdan *pp = (*pp)->pNext; 17054fccf43aSdan if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead); 170650d348b1Sdrh break; 170750d348b1Sdrh } 170850d348b1Sdrh } 17094fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 17101611e5a3Sdan sqlite3ValueFree(pSession->pZeroBlob); 17114fccf43aSdan 1712296c7658Sdan /* Delete all attached table objects. And the contents of their 1713296c7658Sdan ** associated hash-tables. */ 17145d607a6eSdan sessionDeleteTable(pSession->pTable); 17154fccf43aSdan 1716296c7658Sdan /* Free the session object itself. */ 17174fccf43aSdan sqlite3_free(pSession); 17184fccf43aSdan } 17194fccf43aSdan 17204fccf43aSdan /* 17217531a5a3Sdan ** Set a table filter on a Session Object. 17227531a5a3Sdan */ 17237531a5a3Sdan void sqlite3session_table_filter( 17247531a5a3Sdan sqlite3_session *pSession, 17257531a5a3Sdan int(*xFilter)(void*, const char*), 17267531a5a3Sdan void *pCtx /* First argument passed to xFilter */ 17277531a5a3Sdan ){ 17287531a5a3Sdan pSession->bAutoAttach = 1; 17297531a5a3Sdan pSession->pFilterCtx = pCtx; 17307531a5a3Sdan pSession->xTableFilter = xFilter; 17317531a5a3Sdan } 17327531a5a3Sdan 17337531a5a3Sdan /* 17344fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table 17354fccf43aSdan ** while the session object is enabled will be recorded. 17364fccf43aSdan ** 17374fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does 17384fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias) 17394fccf43aSdan ** or not. 17404fccf43aSdan */ 17414fccf43aSdan int sqlite3session_attach( 17424fccf43aSdan sqlite3_session *pSession, /* Session object */ 17434fccf43aSdan const char *zName /* Table name */ 17444fccf43aSdan ){ 1745ff4d0f41Sdan int rc = SQLITE_OK; 1746ff4d0f41Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1747ff4d0f41Sdan 1748ff4d0f41Sdan if( !zName ){ 1749ff4d0f41Sdan pSession->bAutoAttach = 1; 1750ff4d0f41Sdan }else{ 1751296c7658Sdan SessionTable *pTab; /* New table object (if required) */ 1752296c7658Sdan int nName; /* Number of bytes in string zName */ 17534fccf43aSdan 17544fccf43aSdan /* First search for an existing entry. If one is found, this call is 17554fccf43aSdan ** a no-op. Return early. */ 1756cfdbde21Sdrh nName = sqlite3Strlen30(zName); 17574fccf43aSdan for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){ 17584c220252Sdan if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break; 17594fccf43aSdan } 17604fccf43aSdan 17614c220252Sdan if( !pTab ){ 17624fccf43aSdan /* Allocate new SessionTable object. */ 17634fccf43aSdan pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1); 17644c220252Sdan if( !pTab ){ 17654c220252Sdan rc = SQLITE_NOMEM; 17664c220252Sdan }else{ 17676c39e6a8Sdan /* Populate the new SessionTable object and link it into the list. 17686c39e6a8Sdan ** The new object must be linked onto the end of the list, not 17696c39e6a8Sdan ** simply added to the start of it in order to ensure that tables 17706c39e6a8Sdan ** appear in the correct order when a changeset or patchset is 17716c39e6a8Sdan ** eventually generated. */ 17726c39e6a8Sdan SessionTable **ppTab; 17734fccf43aSdan memset(pTab, 0, sizeof(SessionTable)); 17744fccf43aSdan pTab->zName = (char *)&pTab[1]; 17754fccf43aSdan memcpy(pTab->zName, zName, nName+1); 17766c39e6a8Sdan for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext); 17776c39e6a8Sdan *ppTab = pTab; 17784c220252Sdan } 17794c220252Sdan } 1780ff4d0f41Sdan } 17814fccf43aSdan 17824c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 17834c220252Sdan return rc; 17844fccf43aSdan } 17854fccf43aSdan 1786296c7658Sdan /* 1787296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data. 1788296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is. 1789296c7658Sdan ** 1790296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered, 1791296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero. 1792296c7658Sdan */ 17934fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){ 179480fe2d93Sdan if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){ 17954fccf43aSdan u8 *aNew; 17964fccf43aSdan int nNew = p->nAlloc ? p->nAlloc : 128; 17974fccf43aSdan do { 17984fccf43aSdan nNew = nNew*2; 1799ef7a6304Sdan }while( nNew<(p->nBuf+nByte) ); 18004fccf43aSdan 18014fccf43aSdan aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew); 18024fccf43aSdan if( 0==aNew ){ 18034fccf43aSdan *pRc = SQLITE_NOMEM; 180480fe2d93Sdan }else{ 18054fccf43aSdan p->aBuf = aNew; 18064fccf43aSdan p->nAlloc = nNew; 18074fccf43aSdan } 180880fe2d93Sdan } 180980fe2d93Sdan return (*pRc!=SQLITE_OK); 18104fccf43aSdan } 18114fccf43aSdan 1812296c7658Sdan /* 1813fa122adaSdan ** Append the value passed as the second argument to the buffer passed 1814fa122adaSdan ** as the first. 1815fa122adaSdan ** 1816fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called. 1817fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code 1818fa122adaSdan ** before returning. 1819fa122adaSdan */ 1820fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){ 1821fa122adaSdan int rc = *pRc; 1822fa122adaSdan if( rc==SQLITE_OK ){ 1823fa122adaSdan int nByte = 0; 1824e8fa8c96Sdan rc = sessionSerializeValue(0, pVal, &nByte); 1825fa122adaSdan sessionBufferGrow(p, nByte, &rc); 1826fa122adaSdan if( rc==SQLITE_OK ){ 1827fa122adaSdan rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0); 1828fa122adaSdan p->nBuf += nByte; 1829fa122adaSdan }else{ 1830fa122adaSdan *pRc = rc; 1831fa122adaSdan } 1832fa122adaSdan } 1833fa122adaSdan } 1834fa122adaSdan 1835fa122adaSdan /* 1836296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1837296c7658Sdan ** called. Otherwise, append a single byte to the buffer. 1838296c7658Sdan ** 1839296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1840296c7658Sdan ** returning. 1841296c7658Sdan */ 18424fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){ 184380fe2d93Sdan if( 0==sessionBufferGrow(p, 1, pRc) ){ 18444fccf43aSdan p->aBuf[p->nBuf++] = v; 18454fccf43aSdan } 18464fccf43aSdan } 18474fccf43aSdan 1848296c7658Sdan /* 1849296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1850296c7658Sdan ** called. Otherwise, append a single varint to the buffer. 1851296c7658Sdan ** 1852296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1853296c7658Sdan ** returning. 1854296c7658Sdan */ 1855cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){ 185680fe2d93Sdan if( 0==sessionBufferGrow(p, 9, pRc) ){ 18574fccf43aSdan p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v); 18584fccf43aSdan } 18594fccf43aSdan } 18604fccf43aSdan 1861296c7658Sdan /* 1862296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1863296c7658Sdan ** called. Otherwise, append a blob of data to the buffer. 1864296c7658Sdan ** 1865296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1866296c7658Sdan ** returning. 1867296c7658Sdan */ 18684fccf43aSdan static void sessionAppendBlob( 18694fccf43aSdan SessionBuffer *p, 18704fccf43aSdan const u8 *aBlob, 18714fccf43aSdan int nBlob, 18724fccf43aSdan int *pRc 18734fccf43aSdan ){ 1874895decf6Sdan if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){ 18754fccf43aSdan memcpy(&p->aBuf[p->nBuf], aBlob, nBlob); 18764fccf43aSdan p->nBuf += nBlob; 18774fccf43aSdan } 18784fccf43aSdan } 18794fccf43aSdan 1880296c7658Sdan /* 1881296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1882296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string 1883296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer. 1884296c7658Sdan ** 1885296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1886296c7658Sdan ** returning. 1887296c7658Sdan */ 1888d5f0767cSdan static void sessionAppendStr( 1889d5f0767cSdan SessionBuffer *p, 1890d5f0767cSdan const char *zStr, 1891d5f0767cSdan int *pRc 1892d5f0767cSdan ){ 1893cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr); 189480fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1895d5f0767cSdan memcpy(&p->aBuf[p->nBuf], zStr, nStr); 1896d5f0767cSdan p->nBuf += nStr; 1897d5f0767cSdan } 1898d5f0767cSdan } 1899d5f0767cSdan 1900296c7658Sdan /* 1901296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1902296c7658Sdan ** called. Otherwise, append the string representation of integer iVal 1903296c7658Sdan ** to the buffer. No nul-terminator is written. 1904296c7658Sdan ** 1905296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1906296c7658Sdan ** returning. 1907296c7658Sdan */ 1908d5f0767cSdan static void sessionAppendInteger( 1909296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1910296c7658Sdan int iVal, /* Value to write the string rep. of */ 1911296c7658Sdan int *pRc /* IN/OUT: Error code */ 1912d5f0767cSdan ){ 1913d5f0767cSdan char aBuf[24]; 1914d5f0767cSdan sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal); 1915d5f0767cSdan sessionAppendStr(p, aBuf, pRc); 1916d5f0767cSdan } 1917d5f0767cSdan 1918296c7658Sdan /* 1919296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1920296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and 1921296c7658Sdan ** with any embedded quote characters escaped to the buffer. No 1922296c7658Sdan ** nul-terminator byte is written. 1923296c7658Sdan ** 1924296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1925296c7658Sdan ** returning. 1926296c7658Sdan */ 1927d5f0767cSdan static void sessionAppendIdent( 1928296c7658Sdan SessionBuffer *p, /* Buffer to a append to */ 1929296c7658Sdan const char *zStr, /* String to quote, escape and append */ 1930296c7658Sdan int *pRc /* IN/OUT: Error code */ 1931d5f0767cSdan ){ 1932cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1; 193380fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1934d5f0767cSdan char *zOut = (char *)&p->aBuf[p->nBuf]; 1935d5f0767cSdan const char *zIn = zStr; 1936d5f0767cSdan *zOut++ = '"'; 1937d5f0767cSdan while( *zIn ){ 1938d5f0767cSdan if( *zIn=='"' ) *zOut++ = '"'; 1939d5f0767cSdan *zOut++ = *(zIn++); 1940d5f0767cSdan } 1941d5f0767cSdan *zOut++ = '"'; 1942cfdbde21Sdrh p->nBuf = (int)((u8 *)zOut - p->aBuf); 1943d5f0767cSdan } 1944d5f0767cSdan } 1945d5f0767cSdan 1946296c7658Sdan /* 1947296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1948296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored 1949296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points 1950296c7658Sdan ** to to the buffer. 1951296c7658Sdan */ 19524fccf43aSdan static void sessionAppendCol( 1953296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1954296c7658Sdan sqlite3_stmt *pStmt, /* Handle pointing to row containing value */ 1955296c7658Sdan int iCol, /* Column to read value from */ 1956296c7658Sdan int *pRc /* IN/OUT: Error code */ 19574fccf43aSdan ){ 19584fccf43aSdan if( *pRc==SQLITE_OK ){ 19594fccf43aSdan int eType = sqlite3_column_type(pStmt, iCol); 19604fccf43aSdan sessionAppendByte(p, (u8)eType, pRc); 19614fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 19624fccf43aSdan sqlite3_int64 i; 19634fccf43aSdan u8 aBuf[8]; 19644fccf43aSdan if( eType==SQLITE_INTEGER ){ 19654fccf43aSdan i = sqlite3_column_int64(pStmt, iCol); 19664fccf43aSdan }else{ 19674fccf43aSdan double r = sqlite3_column_double(pStmt, iCol); 19684fccf43aSdan memcpy(&i, &r, 8); 19694fccf43aSdan } 1970296c7658Sdan sessionPutI64(aBuf, i); 19714fccf43aSdan sessionAppendBlob(p, aBuf, 8, pRc); 19724fccf43aSdan } 19734fccf43aSdan if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){ 19746734007dSdan u8 *z; 19753cc89d95Sdan int nByte; 19766734007dSdan if( eType==SQLITE_BLOB ){ 19776734007dSdan z = (u8 *)sqlite3_column_blob(pStmt, iCol); 19786734007dSdan }else{ 19796734007dSdan z = (u8 *)sqlite3_column_text(pStmt, iCol); 19806734007dSdan } 19813cc89d95Sdan nByte = sqlite3_column_bytes(pStmt, iCol); 19823cc89d95Sdan if( z || (eType==SQLITE_BLOB && nByte==0) ){ 19834fccf43aSdan sessionAppendVarint(p, nByte, pRc); 19846734007dSdan sessionAppendBlob(p, z, nByte, pRc); 19856734007dSdan }else{ 19866734007dSdan *pRc = SQLITE_NOMEM; 19876734007dSdan } 19884fccf43aSdan } 19894fccf43aSdan } 19904fccf43aSdan } 19914fccf43aSdan 1992296c7658Sdan /* 1993296c7658Sdan ** 199480fe2d93Sdan ** This function appends an update change to the buffer (see the comments 199580fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change 199680fe2d93Sdan ** consists of: 1997296c7658Sdan ** 1998296c7658Sdan ** 1 byte: SQLITE_UPDATE (0x17) 1999296c7658Sdan ** n bytes: old.* record (see RECORD FORMAT) 2000296c7658Sdan ** m bytes: new.* record (see RECORD FORMAT) 2001296c7658Sdan ** 2002296c7658Sdan ** The SessionChange object passed as the third argument contains the 2003296c7658Sdan ** values that were stored in the row when the session began (the old.* 2004296c7658Sdan ** values). The statement handle passed as the second argument points 2005296c7658Sdan ** at the current version of the row (the new.* values). 2006296c7658Sdan ** 2007296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value 2008296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer. 2009296c7658Sdan ** 2010296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the 2011296c7658Sdan ** original values of any fields that have been modified. The new.* record 2012296c7658Sdan ** contains the new values of only those fields that have been modified. 2013296c7658Sdan */ 201480fe2d93Sdan static int sessionAppendUpdate( 2015296c7658Sdan SessionBuffer *pBuf, /* Buffer to append to */ 201673b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 2017296c7658Sdan sqlite3_stmt *pStmt, /* Statement handle pointing at new row */ 2018296c7658Sdan SessionChange *p, /* Object containing old values */ 201980fe2d93Sdan u8 *abPK /* Boolean array - true for PK columns */ 20204fccf43aSdan ){ 202180fe2d93Sdan int rc = SQLITE_OK; 2022296c7658Sdan SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */ 2023296c7658Sdan int bNoop = 1; /* Set to zero if any values are modified */ 20241f34f8ccSdan int nRewind = pBuf->nBuf; /* Set to zero if any values are modified */ 2025296c7658Sdan int i; /* Used to iterate through columns */ 2026296c7658Sdan u8 *pCsr = p->aRecord; /* Used to iterate through old.* values */ 2027296c7658Sdan 202880fe2d93Sdan sessionAppendByte(pBuf, SQLITE_UPDATE, &rc); 202980fe2d93Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 20304fccf43aSdan for(i=0; i<sqlite3_column_count(pStmt); i++){ 203137f133ecSdan int bChanged = 0; 20324fccf43aSdan int nAdvance; 20334fccf43aSdan int eType = *pCsr; 20344fccf43aSdan switch( eType ){ 20354fccf43aSdan case SQLITE_NULL: 20364fccf43aSdan nAdvance = 1; 20374fccf43aSdan if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){ 203837f133ecSdan bChanged = 1; 20394fccf43aSdan } 20404fccf43aSdan break; 20414fccf43aSdan 20424fccf43aSdan case SQLITE_FLOAT: 20434fccf43aSdan case SQLITE_INTEGER: { 20444fccf43aSdan nAdvance = 9; 20454fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) ){ 20464fccf43aSdan sqlite3_int64 iVal = sessionGetI64(&pCsr[1]); 20474fccf43aSdan if( eType==SQLITE_INTEGER ){ 20484fccf43aSdan if( iVal==sqlite3_column_int64(pStmt, i) ) break; 20494fccf43aSdan }else{ 20504fccf43aSdan double dVal; 20514fccf43aSdan memcpy(&dVal, &iVal, 8); 20524fccf43aSdan if( dVal==sqlite3_column_double(pStmt, i) ) break; 20534fccf43aSdan } 20544fccf43aSdan } 205537f133ecSdan bChanged = 1; 20564fccf43aSdan break; 20574fccf43aSdan } 20584fccf43aSdan 2059e5754eecSdan default: { 2060895decf6Sdan int n; 2061895decf6Sdan int nHdr = 1 + sessionVarintGet(&pCsr[1], &n); 2062e5754eecSdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 2063895decf6Sdan nAdvance = nHdr + n; 20644fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) 2065895decf6Sdan && n==sqlite3_column_bytes(pStmt, i) 2066895decf6Sdan && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n)) 20674fccf43aSdan ){ 20684fccf43aSdan break; 20694fccf43aSdan } 207037f133ecSdan bChanged = 1; 20714fccf43aSdan } 20724fccf43aSdan } 20734fccf43aSdan 207473b3c055Sdan /* If at least one field has been modified, this is not a no-op. */ 207573b3c055Sdan if( bChanged ) bNoop = 0; 207673b3c055Sdan 207773b3c055Sdan /* Add a field to the old.* record. This is omitted if this modules is 207873b3c055Sdan ** currently generating a patchset. */ 207973b3c055Sdan if( bPatchset==0 ){ 208037f133ecSdan if( bChanged || abPK[i] ){ 208180fe2d93Sdan sessionAppendBlob(pBuf, pCsr, nAdvance, &rc); 20824fccf43aSdan }else{ 208380fe2d93Sdan sessionAppendByte(pBuf, 0, &rc); 208437f133ecSdan } 208573b3c055Sdan } 208637f133ecSdan 208773b3c055Sdan /* Add a field to the new.* record. Or the only record if currently 208873b3c055Sdan ** generating a patchset. */ 208973b3c055Sdan if( bChanged || (bPatchset && abPK[i]) ){ 209080fe2d93Sdan sessionAppendCol(&buf2, pStmt, i, &rc); 209137f133ecSdan }else{ 209280fe2d93Sdan sessionAppendByte(&buf2, 0, &rc); 20934fccf43aSdan } 209437f133ecSdan 20954fccf43aSdan pCsr += nAdvance; 20964fccf43aSdan } 20974fccf43aSdan 20984fccf43aSdan if( bNoop ){ 20991f34f8ccSdan pBuf->nBuf = nRewind; 21004fccf43aSdan }else{ 210180fe2d93Sdan sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc); 21024fccf43aSdan } 21031f34f8ccSdan sqlite3_free(buf2.aBuf); 210480fe2d93Sdan 210580fe2d93Sdan return rc; 2106d5f0767cSdan } 21074fccf43aSdan 2108a71d2371Sdan /* 2109a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use 2110a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset 2111a71d2371Sdan ** format otherwise. 2112a71d2371Sdan */ 211373b3c055Sdan static int sessionAppendDelete( 211473b3c055Sdan SessionBuffer *pBuf, /* Buffer to append to */ 211573b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 211673b3c055Sdan SessionChange *p, /* Object containing old values */ 2117a71d2371Sdan int nCol, /* Number of columns in table */ 211873b3c055Sdan u8 *abPK /* Boolean array - true for PK columns */ 211973b3c055Sdan ){ 212073b3c055Sdan int rc = SQLITE_OK; 212173b3c055Sdan 212273b3c055Sdan sessionAppendByte(pBuf, SQLITE_DELETE, &rc); 212373b3c055Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 212473b3c055Sdan 212573b3c055Sdan if( bPatchset==0 ){ 212673b3c055Sdan sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc); 212773b3c055Sdan }else{ 212873b3c055Sdan int i; 212973b3c055Sdan u8 *a = p->aRecord; 213073b3c055Sdan for(i=0; i<nCol; i++){ 213173b3c055Sdan u8 *pStart = a; 213273b3c055Sdan int eType = *a++; 213373b3c055Sdan 213473b3c055Sdan switch( eType ){ 213573b3c055Sdan case 0: 213673b3c055Sdan case SQLITE_NULL: 213773b3c055Sdan assert( abPK[i]==0 ); 213873b3c055Sdan break; 213973b3c055Sdan 214073b3c055Sdan case SQLITE_FLOAT: 214173b3c055Sdan case SQLITE_INTEGER: 214273b3c055Sdan a += 8; 214373b3c055Sdan break; 214473b3c055Sdan 214573b3c055Sdan default: { 214673b3c055Sdan int n; 214773b3c055Sdan a += sessionVarintGet(a, &n); 214873b3c055Sdan a += n; 214973b3c055Sdan break; 215073b3c055Sdan } 215173b3c055Sdan } 215273b3c055Sdan if( abPK[i] ){ 2153f5ab08c7Sdrh sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc); 215473b3c055Sdan } 215573b3c055Sdan } 215673b3c055Sdan assert( (a - p->aRecord)==p->nRecord ); 215773b3c055Sdan } 215873b3c055Sdan 215973b3c055Sdan return rc; 216073b3c055Sdan } 216173b3c055Sdan 216277fc1d5bSdan /* 216377fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table 216477fc1d5bSdan ** zTab in database zDb based on its primary key. i.e. 216577fc1d5bSdan ** 216677fc1d5bSdan ** SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ... 216777fc1d5bSdan */ 2168e8d5648eSdan static int sessionSelectStmt( 2169e8d5648eSdan sqlite3 *db, /* Database handle */ 2170d7fb7d24Sdan const char *zDb, /* Database name */ 2171e8d5648eSdan const char *zTab, /* Table name */ 217277fc1d5bSdan int nCol, /* Number of columns in table */ 217377fc1d5bSdan const char **azCol, /* Names of table columns */ 217477fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 217577fc1d5bSdan sqlite3_stmt **ppStmt /* OUT: Prepared SELECT statement */ 2176d5f0767cSdan ){ 2177e8d5648eSdan int rc = SQLITE_OK; 21783739f298Sdan char *zSql = 0; 21793739f298Sdan int nSql = -1; 21803739f298Sdan 21813739f298Sdan if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){ 21823739f298Sdan zSql = sqlite3_mprintf( 21833739f298Sdan "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND " 21843739f298Sdan "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb 21853739f298Sdan ); 21863739f298Sdan }else{ 2187d5f0767cSdan int i; 2188e8d5648eSdan const char *zSep = ""; 2189e8d5648eSdan SessionBuffer buf = {0, 0, 0}; 2190d5f0767cSdan 2191e8d5648eSdan sessionAppendStr(&buf, "SELECT * FROM ", &rc); 2192d7fb7d24Sdan sessionAppendIdent(&buf, zDb, &rc); 2193d7fb7d24Sdan sessionAppendStr(&buf, ".", &rc); 2194e8d5648eSdan sessionAppendIdent(&buf, zTab, &rc); 2195e8d5648eSdan sessionAppendStr(&buf, " WHERE ", &rc); 2196e8d5648eSdan for(i=0; i<nCol; i++){ 2197e8d5648eSdan if( abPK[i] ){ 2198e8d5648eSdan sessionAppendStr(&buf, zSep, &rc); 2199e8d5648eSdan sessionAppendIdent(&buf, azCol[i], &rc); 22003739f298Sdan sessionAppendStr(&buf, " IS ?", &rc); 2201e8d5648eSdan sessionAppendInteger(&buf, i+1, &rc); 2202e8d5648eSdan zSep = " AND "; 2203d5f0767cSdan } 2204d5f0767cSdan } 22053739f298Sdan zSql = (char*)buf.aBuf; 22063739f298Sdan nSql = buf.nBuf; 2207d5f0767cSdan } 22083739f298Sdan 22093739f298Sdan if( rc==SQLITE_OK ){ 22103739f298Sdan rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0); 22113739f298Sdan } 22123739f298Sdan sqlite3_free(zSql); 2213e8d5648eSdan return rc; 2214d5f0767cSdan } 2215d5f0767cSdan 221677fc1d5bSdan /* 221777fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange 221877fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement 221977fc1d5bSdan ** is as prepared by function sessionSelectStmt(). 222077fc1d5bSdan ** 222177fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite 222277fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise. 222377fc1d5bSdan */ 2224e8d5648eSdan static int sessionSelectBind( 222577fc1d5bSdan sqlite3_stmt *pSelect, /* SELECT from sessionSelectStmt() */ 222677fc1d5bSdan int nCol, /* Number of columns in table */ 222777fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 222877fc1d5bSdan SessionChange *pChange /* Change structure */ 2229e8d5648eSdan ){ 2230e8d5648eSdan int i; 2231e8d5648eSdan int rc = SQLITE_OK; 2232e5754eecSdan u8 *a = pChange->aRecord; 2233d5f0767cSdan 2234e8d5648eSdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2235e8d5648eSdan int eType = *a++; 2236e8d5648eSdan 2237e8d5648eSdan switch( eType ){ 223880fe2d93Sdan case 0: 2239e8d5648eSdan case SQLITE_NULL: 22401611e5a3Sdan assert( abPK[i]==0 ); 2241e8d5648eSdan break; 2242e8d5648eSdan 2243e8d5648eSdan case SQLITE_INTEGER: { 2244e8d5648eSdan if( abPK[i] ){ 2245e8d5648eSdan i64 iVal = sessionGetI64(a); 2246e8d5648eSdan rc = sqlite3_bind_int64(pSelect, i+1, iVal); 2247e8d5648eSdan } 2248e8d5648eSdan a += 8; 2249e8d5648eSdan break; 2250d5f0767cSdan } 2251296c7658Sdan 2252e8d5648eSdan case SQLITE_FLOAT: { 2253e8d5648eSdan if( abPK[i] ){ 2254e8d5648eSdan double rVal; 2255e8d5648eSdan i64 iVal = sessionGetI64(a); 2256e8d5648eSdan memcpy(&rVal, &iVal, 8); 22574e895da1Sdan rc = sqlite3_bind_double(pSelect, i+1, rVal); 2258d5f0767cSdan } 2259e8d5648eSdan a += 8; 2260e8d5648eSdan break; 2261e8d5648eSdan } 2262e8d5648eSdan 2263e8d5648eSdan case SQLITE_TEXT: { 2264e8d5648eSdan int n; 2265e8d5648eSdan a += sessionVarintGet(a, &n); 2266e8d5648eSdan if( abPK[i] ){ 2267e8d5648eSdan rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT); 2268e8d5648eSdan } 2269e8d5648eSdan a += n; 2270e8d5648eSdan break; 2271e8d5648eSdan } 2272e8d5648eSdan 2273e5754eecSdan default: { 2274e8d5648eSdan int n; 2275e5754eecSdan assert( eType==SQLITE_BLOB ); 2276e8d5648eSdan a += sessionVarintGet(a, &n); 2277e8d5648eSdan if( abPK[i] ){ 2278e8d5648eSdan rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT); 2279e8d5648eSdan } 2280e8d5648eSdan a += n; 2281e8d5648eSdan break; 2282e8d5648eSdan } 2283e8d5648eSdan } 2284e8d5648eSdan } 2285e8d5648eSdan 2286d5f0767cSdan return rc; 22874fccf43aSdan } 22884fccf43aSdan 228977fc1d5bSdan /* 229077fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it 229177fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary 229277fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an 229377fc1d5bSdan ** SQLite error code before returning. 229477fc1d5bSdan */ 22955d607a6eSdan static void sessionAppendTableHdr( 2296a71d2371Sdan SessionBuffer *pBuf, /* Append header to this buffer */ 2297a71d2371Sdan int bPatchset, /* Use the patchset format if true */ 2298a71d2371Sdan SessionTable *pTab, /* Table object to append header for */ 2299a71d2371Sdan int *pRc /* IN/OUT: Error code */ 23005d607a6eSdan ){ 23015d607a6eSdan /* Write a table header */ 230273b3c055Sdan sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc); 23035d607a6eSdan sessionAppendVarint(pBuf, pTab->nCol, pRc); 23045d607a6eSdan sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc); 23054f528042Sdan sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc); 23065d607a6eSdan } 23075d607a6eSdan 2308a71d2371Sdan /* 2309a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset 2310a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object 2311a71d2371Sdan ** passed as the first argument. 2312a71d2371Sdan ** 2313a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset 2314a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error 2315a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set 2316a71d2371Sdan ** to 0. 2317a71d2371Sdan */ 2318adf3bf58Sdrh static int sessionGenerateChangeset( 23194fccf43aSdan sqlite3_session *pSession, /* Session object */ 232073b3c055Sdan int bPatchset, /* True for patchset, false for changeset */ 2321ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2322ef7a6304Sdan void *pOut, /* First argument for xOutput */ 23234fccf43aSdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 23244fccf43aSdan void **ppChangeset /* OUT: Buffer containing changeset */ 23254fccf43aSdan ){ 2326296c7658Sdan sqlite3 *db = pSession->db; /* Source database handle */ 2327296c7658Sdan SessionTable *pTab; /* Used to iterate through attached tables */ 2328296c7658Sdan SessionBuffer buf = {0,0,0}; /* Buffer in which to accumlate changeset */ 2329296c7658Sdan int rc; /* Return code */ 23304fccf43aSdan 2331ef7a6304Sdan assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) ); 2332ef7a6304Sdan 2333296c7658Sdan /* Zero the output variables in case an error occurs. If this session 2334296c7658Sdan ** object is already in the error state (sqlite3_session.rc != SQLITE_OK), 2335296c7658Sdan ** this call will be a no-op. */ 2336ef7a6304Sdan if( xOutput==0 ){ 23374fccf43aSdan *pnChangeset = 0; 23384fccf43aSdan *ppChangeset = 0; 2339ef7a6304Sdan } 2340e5754eecSdan 2341e5754eecSdan if( pSession->rc ) return pSession->rc; 2342e5754eecSdan rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0); 2343e5754eecSdan if( rc!=SQLITE_OK ) return rc; 2344e5754eecSdan 2345e5754eecSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 23464fccf43aSdan 23474fccf43aSdan for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 23484fccf43aSdan if( pTab->nEntry ){ 2349d7fb7d24Sdan const char *zName = pTab->zName; 2350a9605b91Sdan int nCol; /* Number of columns in table */ 2351a9605b91Sdan u8 *abPK; /* Primary key array */ 2352a9605b91Sdan const char **azCol = 0; /* Table columns */ 23531f34f8ccSdan int i; /* Used to iterate through hash buckets */ 23541f34f8ccSdan sqlite3_stmt *pSel = 0; /* SELECT statement to query table pTab */ 23551f34f8ccSdan int nRewind = buf.nBuf; /* Initial size of write buffer */ 23561f34f8ccSdan int nNoop; /* Size of buffer after writing tbl header */ 23574fccf43aSdan 2358a9605b91Sdan /* Check the table schema is still Ok. */ 2359a9605b91Sdan rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK); 2360a9605b91Sdan if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){ 2361a9605b91Sdan rc = SQLITE_SCHEMA; 2362a9605b91Sdan } 2363a9605b91Sdan 23644fccf43aSdan /* Write a table header */ 236573b3c055Sdan sessionAppendTableHdr(&buf, bPatchset, pTab, &rc); 23664fccf43aSdan 23674fccf43aSdan /* Build and compile a statement to execute: */ 23684fccf43aSdan if( rc==SQLITE_OK ){ 2369d7fb7d24Sdan rc = sessionSelectStmt( 2370a9605b91Sdan db, pSession->zDb, zName, nCol, azCol, abPK, &pSel); 23714fccf43aSdan } 23724fccf43aSdan 23731f34f8ccSdan nNoop = buf.nBuf; 237412ca0b56Sdan for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){ 2375e8d5648eSdan SessionChange *p; /* Used to iterate through changes */ 2376e8d5648eSdan 23774fccf43aSdan for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){ 2378e5754eecSdan rc = sessionSelectBind(pSel, nCol, abPK, p); 237980fe2d93Sdan if( rc!=SQLITE_OK ) continue; 23801f34f8ccSdan if( sqlite3_step(pSel)==SQLITE_ROW ){ 2381798693b2Sdan if( p->op==SQLITE_INSERT ){ 23824fccf43aSdan int iCol; 23834fccf43aSdan sessionAppendByte(&buf, SQLITE_INSERT, &rc); 2384b4480e94Sdan sessionAppendByte(&buf, p->bIndirect, &rc); 2385e8d5648eSdan for(iCol=0; iCol<nCol; iCol++){ 23861f34f8ccSdan sessionAppendCol(&buf, pSel, iCol, &rc); 23874fccf43aSdan } 2388e8d5648eSdan }else{ 238973b3c055Sdan rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK); 23904fccf43aSdan } 2391798693b2Sdan }else if( p->op!=SQLITE_INSERT ){ 2392a71d2371Sdan rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK); 23934fccf43aSdan } 239412ca0b56Sdan if( rc==SQLITE_OK ){ 23951f34f8ccSdan rc = sqlite3_reset(pSel); 23964fccf43aSdan } 2397ef7a6304Sdan 2398f1a08ad8Sdrh /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass 2399ef7a6304Sdan ** its contents to the xOutput() callback. */ 2400ef7a6304Sdan if( xOutput 2401ef7a6304Sdan && rc==SQLITE_OK 2402ef7a6304Sdan && buf.nBuf>nNoop 2403f1a08ad8Sdrh && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE 2404ef7a6304Sdan ){ 2405ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2406ef7a6304Sdan nNoop = -1; 2407ef7a6304Sdan buf.nBuf = 0; 2408ef7a6304Sdan } 2409ef7a6304Sdan 24104fccf43aSdan } 2411e8d5648eSdan } 24124fccf43aSdan 24131f34f8ccSdan sqlite3_finalize(pSel); 24141f34f8ccSdan if( buf.nBuf==nNoop ){ 24154fccf43aSdan buf.nBuf = nRewind; 24164fccf43aSdan } 2417cfdbde21Sdrh sqlite3_free((char*)azCol); /* cast works around VC++ bug */ 24184fccf43aSdan } 24194fccf43aSdan } 24204fccf43aSdan 24214fccf43aSdan if( rc==SQLITE_OK ){ 2422ef7a6304Sdan if( xOutput==0 ){ 24234fccf43aSdan *pnChangeset = buf.nBuf; 24244fccf43aSdan *ppChangeset = buf.aBuf; 2425ef7a6304Sdan buf.aBuf = 0; 2426ef7a6304Sdan }else if( buf.nBuf>0 ){ 2427ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2428ef7a6304Sdan } 24294fccf43aSdan } 24304c220252Sdan 2431ef7a6304Sdan sqlite3_free(buf.aBuf); 2432e5754eecSdan sqlite3_exec(db, "RELEASE changeset", 0, 0, 0); 24334c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 24344fccf43aSdan return rc; 24354fccf43aSdan } 24364fccf43aSdan 2437296c7658Sdan /* 243873b3c055Sdan ** Obtain a changeset object containing all changes recorded by the 243973b3c055Sdan ** session object passed as the first argument. 244073b3c055Sdan ** 244173b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 244273b3c055Sdan ** using sqlite3_free(). 244373b3c055Sdan */ 244473b3c055Sdan int sqlite3session_changeset( 244573b3c055Sdan sqlite3_session *pSession, /* Session object */ 244673b3c055Sdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 244773b3c055Sdan void **ppChangeset /* OUT: Buffer containing changeset */ 244873b3c055Sdan ){ 2449ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset); 2450ef7a6304Sdan } 2451ef7a6304Sdan 2452ef7a6304Sdan /* 2453ef7a6304Sdan ** Streaming version of sqlite3session_changeset(). 2454ef7a6304Sdan */ 2455f1a08ad8Sdrh int sqlite3session_changeset_strm( 2456ef7a6304Sdan sqlite3_session *pSession, 2457ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2458ef7a6304Sdan void *pOut 2459ef7a6304Sdan ){ 2460ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0); 2461ef7a6304Sdan } 2462ef7a6304Sdan 2463ef7a6304Sdan /* 2464ef7a6304Sdan ** Streaming version of sqlite3session_patchset(). 2465ef7a6304Sdan */ 2466f1a08ad8Sdrh int sqlite3session_patchset_strm( 2467ef7a6304Sdan sqlite3_session *pSession, 2468ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2469ef7a6304Sdan void *pOut 2470ef7a6304Sdan ){ 2471ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0); 247273b3c055Sdan } 247373b3c055Sdan 247473b3c055Sdan /* 247573b3c055Sdan ** Obtain a patchset object containing all changes recorded by the 247673b3c055Sdan ** session object passed as the first argument. 247773b3c055Sdan ** 247873b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 247973b3c055Sdan ** using sqlite3_free(). 248073b3c055Sdan */ 248173b3c055Sdan int sqlite3session_patchset( 248273b3c055Sdan sqlite3_session *pSession, /* Session object */ 248373b3c055Sdan int *pnPatchset, /* OUT: Size of buffer at *ppChangeset */ 248473b3c055Sdan void **ppPatchset /* OUT: Buffer containing changeset */ 248573b3c055Sdan ){ 2486ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset); 248773b3c055Sdan } 248873b3c055Sdan 248973b3c055Sdan /* 2490296c7658Sdan ** Enable or disable the session object passed as the first argument. 2491296c7658Sdan */ 24924fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){ 24934c220252Sdan int ret; 24944c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2495296c7658Sdan if( bEnable>=0 ){ 2496296c7658Sdan pSession->bEnable = bEnable; 24974fccf43aSdan } 24984c220252Sdan ret = pSession->bEnable; 24994c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 25004c220252Sdan return ret; 2501296c7658Sdan } 25024fccf43aSdan 25034fccf43aSdan /* 2504b4480e94Sdan ** Enable or disable the session object passed as the first argument. 2505b4480e94Sdan */ 2506b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){ 2507b4480e94Sdan int ret; 2508b4480e94Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2509b4480e94Sdan if( bIndirect>=0 ){ 2510b4480e94Sdan pSession->bIndirect = bIndirect; 2511b4480e94Sdan } 2512b4480e94Sdan ret = pSession->bIndirect; 2513b4480e94Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2514b4480e94Sdan return ret; 2515b4480e94Sdan } 2516b4480e94Sdan 2517b4480e94Sdan /* 2518b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded 2519b69ec348Sdan ** by the session object passed as the only argument. 2520b69ec348Sdan */ 2521b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){ 2522b69ec348Sdan int ret = 0; 2523b69ec348Sdan SessionTable *pTab; 2524b69ec348Sdan 2525b69ec348Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2526b69ec348Sdan for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){ 2527b69ec348Sdan ret = (pTab->nEntry>0); 2528b69ec348Sdan } 2529b69ec348Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2530b69ec348Sdan 2531ff530326Sdan return (ret==0); 2532b69ec348Sdan } 2533b69ec348Sdan 2534b69ec348Sdan /* 2535f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm(). 25364fccf43aSdan */ 2537adf3bf58Sdrh static int sessionChangesetStart( 2538296c7658Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2539ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2540ef7a6304Sdan void *pIn, 2541296c7658Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2542296c7658Sdan void *pChangeset /* Pointer to buffer containing changeset */ 25434fccf43aSdan ){ 25444fccf43aSdan sqlite3_changeset_iter *pRet; /* Iterator to return */ 25454fccf43aSdan int nByte; /* Number of bytes to allocate for iterator */ 25464fccf43aSdan 2547ef7a6304Sdan assert( xInput==0 || (pChangeset==0 && nChangeset==0) ); 2548ef7a6304Sdan 2549296c7658Sdan /* Zero the output variable in case an error occurs. */ 2550296c7658Sdan *pp = 0; 25514fccf43aSdan 2552296c7658Sdan /* Allocate and initialize the iterator structure. */ 25534fccf43aSdan nByte = sizeof(sqlite3_changeset_iter); 25544fccf43aSdan pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte); 25554fccf43aSdan if( !pRet ) return SQLITE_NOMEM; 25564fccf43aSdan memset(pRet, 0, sizeof(sqlite3_changeset_iter)); 25574757c658Sdan pRet->in.aData = (u8 *)pChangeset; 25584757c658Sdan pRet->in.nData = nChangeset; 2559ef7a6304Sdan pRet->in.xInput = xInput; 2560ef7a6304Sdan pRet->in.pIn = pIn; 2561ef7a6304Sdan pRet->in.bEof = (xInput ? 0 : 1); 25624fccf43aSdan 2563296c7658Sdan /* Populate the output variable and return success. */ 2564296c7658Sdan *pp = pRet; 25654fccf43aSdan return SQLITE_OK; 25664fccf43aSdan } 25674fccf43aSdan 2568296c7658Sdan /* 2569ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset. 2570ef7a6304Sdan */ 2571ef7a6304Sdan int sqlite3changeset_start( 2572ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2573ef7a6304Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2574ef7a6304Sdan void *pChangeset /* Pointer to buffer containing changeset */ 2575ef7a6304Sdan ){ 2576ef7a6304Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset); 2577ef7a6304Sdan } 2578ef7a6304Sdan 2579ef7a6304Sdan /* 2580ef7a6304Sdan ** Streaming version of sqlite3changeset_start(). 2581ef7a6304Sdan */ 2582f1a08ad8Sdrh int sqlite3changeset_start_strm( 2583ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2584ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2585ef7a6304Sdan void *pIn 2586ef7a6304Sdan ){ 2587ef7a6304Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0); 2588ef7a6304Sdan } 2589ef7a6304Sdan 2590ef7a6304Sdan /* 2591d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming 2592d9151526Sdan ** object and the buffer is full, discard some data to free up space. 2593d9151526Sdan */ 2594d9151526Sdan static void sessionDiscardData(SessionInput *pIn){ 2595d9151526Sdan if( pIn->bEof && pIn->xInput && pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){ 2596d9151526Sdan int nMove = pIn->buf.nBuf - pIn->iNext; 2597d9151526Sdan assert( nMove>=0 ); 2598d9151526Sdan if( nMove>0 ){ 2599d9151526Sdan memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove); 2600d9151526Sdan } 2601d9151526Sdan pIn->buf.nBuf -= pIn->iNext; 2602d9151526Sdan pIn->iNext = 0; 2603d9151526Sdan pIn->nData = pIn->buf.nBuf; 2604d9151526Sdan } 2605d9151526Sdan } 2606d9151526Sdan 2607d9151526Sdan /* 2608ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or, 2609ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available 2610ef7a6304Sdan ** data is in the buffer. 2611ef7a6304Sdan ** 2612ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise. 2613ef7a6304Sdan */ 26144757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){ 2615ef7a6304Sdan int rc = SQLITE_OK; 26164757c658Sdan if( pIn->xInput ){ 26174757c658Sdan while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){ 2618f1a08ad8Sdrh int nNew = SESSIONS_STRM_CHUNK_SIZE; 26194757c658Sdan 2620d9151526Sdan if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn); 26214757c658Sdan if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){ 26224757c658Sdan rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew); 26234757c658Sdan if( nNew==0 ){ 26244757c658Sdan pIn->bEof = 1; 26254757c658Sdan }else{ 26264757c658Sdan pIn->buf.nBuf += nNew; 26274757c658Sdan } 26284757c658Sdan } 26294757c658Sdan 26304757c658Sdan pIn->aData = pIn->buf.aBuf; 26314757c658Sdan pIn->nData = pIn->buf.nBuf; 26324757c658Sdan } 2633ef7a6304Sdan } 2634ef7a6304Sdan return rc; 2635ef7a6304Sdan } 2636ef7a6304Sdan 2637ef7a6304Sdan /* 2638ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record 2639ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec 2640ef7a6304Sdan ** until it points to the byte immediately following that record. 2641ef7a6304Sdan */ 2642ef7a6304Sdan static void sessionSkipRecord( 2643ef7a6304Sdan u8 **ppRec, /* IN/OUT: Record pointer */ 2644ef7a6304Sdan int nCol /* Number of values in record */ 2645ef7a6304Sdan ){ 2646ef7a6304Sdan u8 *aRec = *ppRec; 2647ef7a6304Sdan int i; 2648ef7a6304Sdan for(i=0; i<nCol; i++){ 2649ef7a6304Sdan int eType = *aRec++; 2650ef7a6304Sdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2651ef7a6304Sdan int nByte; 2652ef7a6304Sdan aRec += sessionVarintGet((u8*)aRec, &nByte); 2653ef7a6304Sdan aRec += nByte; 2654ef7a6304Sdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2655ef7a6304Sdan aRec += 8; 2656ef7a6304Sdan } 2657ef7a6304Sdan } 2658ef7a6304Sdan 2659ef7a6304Sdan *ppRec = aRec; 2660ef7a6304Sdan } 2661ef7a6304Sdan 2662ef7a6304Sdan /* 26634757c658Sdan ** This function sets the value of the sqlite3_value object passed as the 26644757c658Sdan ** first argument to a copy of the string or blob held in the aData[] 26654757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM 26664757c658Sdan ** error occurs. 26674757c658Sdan */ 26684757c658Sdan static int sessionValueSetStr( 26694757c658Sdan sqlite3_value *pVal, /* Set the value of this object */ 26704757c658Sdan u8 *aData, /* Buffer containing string or blob data */ 26714757c658Sdan int nData, /* Size of buffer aData[] in bytes */ 26724757c658Sdan u8 enc /* String encoding (0 for blobs) */ 26734757c658Sdan ){ 267416228167Sdan /* In theory this code could just pass SQLITE_TRANSIENT as the final 267516228167Sdan ** argument to sqlite3ValueSetStr() and have the copy created 267616228167Sdan ** automatically. But doing so makes it difficult to detect any OOM 267716228167Sdan ** error. Hence the code to create the copy externally. */ 26783cc89d95Sdan u8 *aCopy = sqlite3_malloc(nData+1); 26794757c658Sdan if( aCopy==0 ) return SQLITE_NOMEM; 26804757c658Sdan memcpy(aCopy, aData, nData); 26814757c658Sdan sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free); 26824757c658Sdan return SQLITE_OK; 26834757c658Sdan } 26844757c658Sdan 26854757c658Sdan /* 2686296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT" 2687296c7658Sdan ** for details. 2688296c7658Sdan ** 2689296c7658Sdan ** When this function is called, *paChange points to the start of the record 2690296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to 2691296c7658Sdan ** one byte after the end of the same record before this function returns. 2692a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or, 2693a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields 2694a71d2371Sdan ** (in other words, it is a patchset DELETE record). 2695296c7658Sdan ** 2696296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller) 2697296c7658Sdan ** is set to point to an sqlite3_value object containing the value read 2698296c7658Sdan ** from the corresponding position in the record. If that value is not 2699296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change 2700296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is 2701296c7658Sdan ** set to NULL. 2702296c7658Sdan ** 2703296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures 2704296c7658Sdan ** using sqlite3_free(). 2705296c7658Sdan ** 2706296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned. 2707296c7658Sdan ** The apOut[] array may have been partially populated in this case. 2708296c7658Sdan */ 27094fccf43aSdan static int sessionReadRecord( 2710ef7a6304Sdan SessionInput *pIn, /* Input data */ 27114fccf43aSdan int nCol, /* Number of values in record */ 271273b3c055Sdan u8 *abPK, /* Array of primary key flags, or NULL */ 27134fccf43aSdan sqlite3_value **apOut /* Write values to this array */ 27144fccf43aSdan ){ 2715296c7658Sdan int i; /* Used to iterate through columns */ 2716ef7a6304Sdan int rc = SQLITE_OK; 27174fccf43aSdan 2718ef7a6304Sdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2719ef7a6304Sdan int eType = 0; /* Type of value (SQLITE_NULL, TEXT etc.) */ 272073b3c055Sdan if( abPK && abPK[i]==0 ) continue; 2721ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2722ef7a6304Sdan if( rc==SQLITE_OK ){ 27234757c658Sdan eType = pIn->aData[pIn->iNext++]; 2724ef7a6304Sdan } 2725ef7a6304Sdan 2726e8fa8c96Sdan assert( apOut[i]==0 ); 27274fccf43aSdan if( eType ){ 27284fccf43aSdan apOut[i] = sqlite3ValueNew(0); 2729ef7a6304Sdan if( !apOut[i] ) rc = SQLITE_NOMEM; 2730ef7a6304Sdan } 27314fccf43aSdan 2732ef7a6304Sdan if( rc==SQLITE_OK ){ 27334757c658Sdan u8 *aVal = &pIn->aData[pIn->iNext]; 27344fccf43aSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 27354fccf43aSdan int nByte; 2736ef7a6304Sdan pIn->iNext += sessionVarintGet(aVal, &nByte); 2737ef7a6304Sdan rc = sessionInputBuffer(pIn, nByte); 2738e8fa8c96Sdan if( rc==SQLITE_OK ){ 27396734007dSdan u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0); 27404757c658Sdan rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc); 274191ddd559Sdan } 2742ef7a6304Sdan pIn->iNext += nByte; 27434fccf43aSdan } 27444fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2745ef7a6304Sdan sqlite3_int64 v = sessionGetI64(aVal); 27464fccf43aSdan if( eType==SQLITE_INTEGER ){ 27474fccf43aSdan sqlite3VdbeMemSetInt64(apOut[i], v); 27484fccf43aSdan }else{ 27494fccf43aSdan double d; 27504e895da1Sdan memcpy(&d, &v, 8); 27514fccf43aSdan sqlite3VdbeMemSetDouble(apOut[i], d); 27524fccf43aSdan } 2753ef7a6304Sdan pIn->iNext += 8; 275491ddd559Sdan } 27554fccf43aSdan } 27564fccf43aSdan } 27574fccf43aSdan 2758ef7a6304Sdan return rc; 2759ef7a6304Sdan } 2760ef7a6304Sdan 2761ef7a6304Sdan /* 2762ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2763ef7a6304Sdan ** Specifically, to the following: 2764ef7a6304Sdan ** 2765ef7a6304Sdan ** + number of columns in table (varint) 2766ef7a6304Sdan ** + array of PK flags (1 byte per column), 2767ef7a6304Sdan ** + table name (nul terminated). 2768ef7a6304Sdan ** 2769ef7a6304Sdan ** This function ensures that all of the above is present in the input 2770ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()). 2771ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code. 2772ef7a6304Sdan ** The input pointer is not moved. 2773ef7a6304Sdan */ 2774ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){ 2775ef7a6304Sdan int rc = SQLITE_OK; 2776ef7a6304Sdan int nCol = 0; 27774757c658Sdan int nRead = 0; 2778ef7a6304Sdan 2779ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2780ef7a6304Sdan if( rc==SQLITE_OK ){ 27814757c658Sdan nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol); 27824757c658Sdan rc = sessionInputBuffer(pIn, nRead+nCol+100); 27834757c658Sdan nRead += nCol; 2784ef7a6304Sdan } 27854757c658Sdan 2786ef7a6304Sdan while( rc==SQLITE_OK ){ 27874757c658Sdan while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){ 27884757c658Sdan nRead++; 2789ef7a6304Sdan } 2790e8fa8c96Sdan if( (pIn->iNext + nRead)<pIn->nData ) break; 27914757c658Sdan rc = sessionInputBuffer(pIn, nRead + 100); 27924757c658Sdan } 2793e8fa8c96Sdan *pnByte = nRead+1; 2794ef7a6304Sdan return rc; 2795ef7a6304Sdan } 2796ef7a6304Sdan 2797ef7a6304Sdan /* 2798fa122adaSdan ** The input pointer currently points to the first byte of the first field 2799fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire 280016228167Sdan ** record is buffered. It does not move the input pointer. 280116228167Sdan ** 280216228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of 280316228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The 280416228167Sdan ** final value of *pnByte is undefined in this case. 2805fa122adaSdan */ 2806fa122adaSdan static int sessionChangesetBufferRecord( 280716228167Sdan SessionInput *pIn, /* Input data */ 280816228167Sdan int nCol, /* Number of columns in record */ 280916228167Sdan int *pnByte /* OUT: Size of record in bytes */ 2810fa122adaSdan ){ 2811fa122adaSdan int rc = SQLITE_OK; 2812fa122adaSdan int nByte = 0; 2813fa122adaSdan int i; 2814fa122adaSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 2815fa122adaSdan int eType; 2816fa122adaSdan rc = sessionInputBuffer(pIn, nByte + 10); 2817fa122adaSdan if( rc==SQLITE_OK ){ 2818fa122adaSdan eType = pIn->aData[pIn->iNext + nByte++]; 2819fa122adaSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2820fa122adaSdan int n; 2821fa122adaSdan nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n); 2822fa122adaSdan nByte += n; 2823fa122adaSdan rc = sessionInputBuffer(pIn, nByte); 2824fa122adaSdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2825fa122adaSdan nByte += 8; 2826fa122adaSdan } 2827fa122adaSdan } 2828fa122adaSdan } 2829fa122adaSdan *pnByte = nByte; 2830fa122adaSdan return rc; 2831fa122adaSdan } 2832fa122adaSdan 2833fa122adaSdan /* 2834ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2835ef7a6304Sdan ** Specifically, to the following: 2836ef7a6304Sdan ** 2837ef7a6304Sdan ** + number of columns in table (varint) 2838ef7a6304Sdan ** + array of PK flags (1 byte per column), 2839ef7a6304Sdan ** + table name (nul terminated). 284016228167Sdan ** 284116228167Sdan ** This function decodes the table-header and populates the p->nCol, 284216228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is 284316228167Sdan ** also allocated or resized according to the new value of p->nCol. The 284416228167Sdan ** input pointer is left pointing to the byte following the table header. 284516228167Sdan ** 284616228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code 284716228167Sdan ** is returned and the final values of the various fields enumerated above 284816228167Sdan ** are undefined. 2849ef7a6304Sdan */ 2850ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){ 2851ef7a6304Sdan int rc; 2852ef7a6304Sdan int nCopy; 2853ef7a6304Sdan assert( p->rc==SQLITE_OK ); 2854ef7a6304Sdan 2855ef7a6304Sdan rc = sessionChangesetBufferTblhdr(&p->in, &nCopy); 2856ef7a6304Sdan if( rc==SQLITE_OK ){ 2857ef7a6304Sdan int nByte; 2858ef7a6304Sdan int nVarint; 28594757c658Sdan nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol); 2860ef7a6304Sdan nCopy -= nVarint; 2861ef7a6304Sdan p->in.iNext += nVarint; 2862ef7a6304Sdan nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy; 2863ef7a6304Sdan p->tblhdr.nBuf = 0; 2864ef7a6304Sdan sessionBufferGrow(&p->tblhdr, nByte, &rc); 2865ef7a6304Sdan } 2866ef7a6304Sdan 2867ef7a6304Sdan if( rc==SQLITE_OK ){ 2868ef7a6304Sdan int iPK = sizeof(sqlite3_value*)*p->nCol*2; 2869ef7a6304Sdan memset(p->tblhdr.aBuf, 0, iPK); 28704757c658Sdan memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy); 2871ef7a6304Sdan p->in.iNext += nCopy; 2872ef7a6304Sdan } 2873ef7a6304Sdan 2874ef7a6304Sdan p->apValue = (sqlite3_value**)p->tblhdr.aBuf; 2875ef7a6304Sdan p->abPK = (u8*)&p->apValue[p->nCol*2]; 2876ef7a6304Sdan p->zTab = (char*)&p->abPK[p->nCol]; 2877ef7a6304Sdan return (p->rc = rc); 28784fccf43aSdan } 28794fccf43aSdan 288077fc1d5bSdan /* 288177fc1d5bSdan ** Advance the changeset iterator to the next change. 288277fc1d5bSdan ** 288377fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public 288477fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the 288577fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values. 288677fc1d5bSdan ** 288777fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change 288877fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in 288977fc1d5bSdan ** the record to *pnRec. 289077fc1d5bSdan ** 289177fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is 289277fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite 289377fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further 289477fc1d5bSdan ** changes in the changeset. 289577fc1d5bSdan */ 28965d607a6eSdan static int sessionChangesetNext( 289777fc1d5bSdan sqlite3_changeset_iter *p, /* Changeset iterator */ 289877fc1d5bSdan u8 **paRec, /* If non-NULL, store record pointer here */ 289977fc1d5bSdan int *pnRec /* If non-NULL, store size of record here */ 29005d607a6eSdan ){ 29014fccf43aSdan int i; 2902ef7a6304Sdan u8 op; 29034fccf43aSdan 29045d607a6eSdan assert( (paRec==0 && pnRec==0) || (paRec && pnRec) ); 29055d607a6eSdan 2906296c7658Sdan /* If the iterator is in the error-state, return immediately. */ 29074fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29084fccf43aSdan 29095d607a6eSdan /* Free the current contents of p->apValue[], if any. */ 29104fccf43aSdan if( p->apValue ){ 29114fccf43aSdan for(i=0; i<p->nCol*2; i++){ 29124fccf43aSdan sqlite3ValueFree(p->apValue[i]); 29134fccf43aSdan } 29144fccf43aSdan memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2); 29154fccf43aSdan } 29164fccf43aSdan 2917ef7a6304Sdan /* Make sure the buffer contains at least 10 bytes of input data, or all 2918ef7a6304Sdan ** remaining data if there are less than 10 bytes available. This is 2919ef7a6304Sdan ** sufficient either for the 'T' or 'P' byte and the varint that follows 2920ef7a6304Sdan ** it, or for the two single byte values otherwise. */ 2921ef7a6304Sdan p->rc = sessionInputBuffer(&p->in, 2); 2922ef7a6304Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2923ef7a6304Sdan 29244fccf43aSdan /* If the iterator is already at the end of the changeset, return DONE. */ 29254757c658Sdan if( p->in.iNext>=p->in.nData ){ 29264fccf43aSdan return SQLITE_DONE; 29274fccf43aSdan } 29284fccf43aSdan 2929d9151526Sdan sessionDiscardData(&p->in); 2930d9151526Sdan p->in.iCurrent = p->in.iNext; 2931d9151526Sdan 29324757c658Sdan op = p->in.aData[p->in.iNext++]; 293307d0f15eSdan while( op=='T' || op=='P' ){ 2934ef7a6304Sdan p->bPatchset = (op=='P'); 2935ef7a6304Sdan if( sessionChangesetReadTblhdr(p) ) return p->rc; 2936ef7a6304Sdan if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc; 2937d9151526Sdan p->in.iCurrent = p->in.iNext; 293807d0f15eSdan if( p->in.iNext>=p->in.nData ) return SQLITE_DONE; 29394757c658Sdan op = p->in.aData[p->in.iNext++]; 29405d607a6eSdan } 29415d607a6eSdan 2942ef7a6304Sdan p->op = op; 29434757c658Sdan p->bIndirect = p->in.aData[p->in.iNext++]; 29444fccf43aSdan if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){ 29454757c658Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 29464fccf43aSdan } 29474fccf43aSdan 2948cbf6d2d2Sdan if( paRec ){ 2949cbf6d2d2Sdan int nVal; /* Number of values to buffer */ 2950cbf6d2d2Sdan if( p->bPatchset==0 && op==SQLITE_UPDATE ){ 2951cbf6d2d2Sdan nVal = p->nCol * 2; 2952cbf6d2d2Sdan }else if( p->bPatchset && op==SQLITE_DELETE ){ 2953cbf6d2d2Sdan nVal = 0; 2954cbf6d2d2Sdan for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++; 2955cbf6d2d2Sdan }else{ 2956cbf6d2d2Sdan nVal = p->nCol; 2957cbf6d2d2Sdan } 2958cbf6d2d2Sdan p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec); 2959cbf6d2d2Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2960cbf6d2d2Sdan *paRec = &p->in.aData[p->in.iNext]; 2961cbf6d2d2Sdan p->in.iNext += *pnRec; 2962cbf6d2d2Sdan }else{ 29635d607a6eSdan 29644fccf43aSdan /* If this is an UPDATE or DELETE, read the old.* record. */ 296573b3c055Sdan if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){ 296673b3c055Sdan u8 *abPK = p->bPatchset ? p->abPK : 0; 2967cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue); 29684fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29694fccf43aSdan } 29704fccf43aSdan 29714fccf43aSdan /* If this is an INSERT or UPDATE, read the new.* record. */ 29724fccf43aSdan if( p->op!=SQLITE_DELETE ){ 2973cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]); 29744fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29754fccf43aSdan } 29764fccf43aSdan 2977cbf6d2d2Sdan if( p->bPatchset && p->op==SQLITE_UPDATE ){ 297873b3c055Sdan /* If this is an UPDATE that is part of a patchset, then all PK and 297973b3c055Sdan ** modified fields are present in the new.* record. The old.* record 298073b3c055Sdan ** is currently completely empty. This block shifts the PK fields from 298173b3c055Sdan ** new.* to old.*, to accommodate the code that reads these arrays. */ 298273b3c055Sdan for(i=0; i<p->nCol; i++){ 298373b3c055Sdan assert( p->apValue[i]==0 ); 298473b3c055Sdan assert( p->abPK[i]==0 || p->apValue[i+p->nCol] ); 298573b3c055Sdan if( p->abPK[i] ){ 298673b3c055Sdan p->apValue[i] = p->apValue[i+p->nCol]; 298773b3c055Sdan p->apValue[i+p->nCol] = 0; 298873b3c055Sdan } 298973b3c055Sdan } 299073b3c055Sdan } 2991cbf6d2d2Sdan } 2992ef7a6304Sdan 29934fccf43aSdan return SQLITE_ROW; 29944fccf43aSdan } 29954fccf43aSdan 29964fccf43aSdan /* 29975d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next 29985d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE 29995d607a6eSdan ** or SQLITE_CORRUPT. 30005d607a6eSdan ** 30015d607a6eSdan ** This function may not be called on iterators passed to a conflict handler 30025d607a6eSdan ** callback by changeset_apply(). 30035d607a6eSdan */ 30045d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){ 30055d607a6eSdan return sessionChangesetNext(p, 0, 0); 30065d607a6eSdan } 30075d607a6eSdan 30085d607a6eSdan /* 3009244593c8Sdan ** The following function extracts information on the current change 301077fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next() 30114fccf43aSdan ** has returned SQLITE_ROW. 30124fccf43aSdan */ 30134fccf43aSdan int sqlite3changeset_op( 3014296c7658Sdan sqlite3_changeset_iter *pIter, /* Iterator handle */ 30154fccf43aSdan const char **pzTab, /* OUT: Pointer to table name */ 30164fccf43aSdan int *pnCol, /* OUT: Number of columns in table */ 3017b4480e94Sdan int *pOp, /* OUT: SQLITE_INSERT, DELETE or UPDATE */ 3018b4480e94Sdan int *pbIndirect /* OUT: True if change is indirect */ 30194fccf43aSdan ){ 30204fccf43aSdan *pOp = pIter->op; 30214fccf43aSdan *pnCol = pIter->nCol; 30224fccf43aSdan *pzTab = pIter->zTab; 3023b4480e94Sdan if( pbIndirect ) *pbIndirect = pIter->bIndirect; 30244fccf43aSdan return SQLITE_OK; 30254fccf43aSdan } 30264fccf43aSdan 302777fc1d5bSdan /* 302877fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in 302977fc1d5bSdan ** the database table affected by the change that pIter currently points 303077fc1d5bSdan ** to. This function may only be called after changeset_next() returns 303177fc1d5bSdan ** SQLITE_ROW. 303277fc1d5bSdan */ 3033244593c8Sdan int sqlite3changeset_pk( 3034244593c8Sdan sqlite3_changeset_iter *pIter, /* Iterator object */ 3035244593c8Sdan unsigned char **pabPK, /* OUT: Array of boolean - true for PK cols */ 3036244593c8Sdan int *pnCol /* OUT: Number of entries in output array */ 3037244593c8Sdan ){ 3038244593c8Sdan *pabPK = pIter->abPK; 3039244593c8Sdan if( pnCol ) *pnCol = pIter->nCol; 3040244593c8Sdan return SQLITE_OK; 3041244593c8Sdan } 3042244593c8Sdan 3043296c7658Sdan /* 3044296c7658Sdan ** This function may only be called while the iterator is pointing to an 3045296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()). 3046296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3047296c7658Sdan ** 3048296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3049296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not 3050296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3051296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL. 3052296c7658Sdan ** 3053296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3054296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3055296c7658Sdan */ 30564fccf43aSdan int sqlite3changeset_old( 3057296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3058296c7658Sdan int iVal, /* Index of old.* value to retrieve */ 30594fccf43aSdan sqlite3_value **ppValue /* OUT: Old value (or NULL pointer) */ 30604fccf43aSdan ){ 3061d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){ 3062d5f0767cSdan return SQLITE_MISUSE; 3063d5f0767cSdan } 30644fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 30654fccf43aSdan return SQLITE_RANGE; 30664fccf43aSdan } 30674fccf43aSdan *ppValue = pIter->apValue[iVal]; 30684fccf43aSdan return SQLITE_OK; 30694fccf43aSdan } 30704fccf43aSdan 3071296c7658Sdan /* 3072296c7658Sdan ** This function may only be called while the iterator is pointing to an 3073296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()). 3074296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3075296c7658Sdan ** 3076296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3077296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not 3078296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3079296c7658Sdan ** was not modified), set *ppValue to NULL. 3080296c7658Sdan ** 3081296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3082296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3083296c7658Sdan */ 30844fccf43aSdan int sqlite3changeset_new( 3085296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3086296c7658Sdan int iVal, /* Index of new.* value to retrieve */ 30874fccf43aSdan sqlite3_value **ppValue /* OUT: New value (or NULL pointer) */ 30884fccf43aSdan ){ 3089d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){ 3090d5f0767cSdan return SQLITE_MISUSE; 3091d5f0767cSdan } 30924fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 30934fccf43aSdan return SQLITE_RANGE; 30944fccf43aSdan } 30954fccf43aSdan *ppValue = pIter->apValue[pIter->nCol+iVal]; 30964fccf43aSdan return SQLITE_OK; 30974fccf43aSdan } 30984fccf43aSdan 3099296c7658Sdan /* 31007aa469cdSdan ** The following two macros are used internally. They are similar to the 31017aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that 31027aa469cdSdan ** they omit all error checking and return a pointer to the requested value. 31037aa469cdSdan */ 31047aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)] 31057aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)] 31067aa469cdSdan 31077aa469cdSdan /* 3108296c7658Sdan ** This function may only be called with a changeset iterator that has been 3109296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT 3110296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned. 3111296c7658Sdan ** 3112296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure 3113296c7658Sdan ** containing the iVal'th value of the conflicting record. 3114296c7658Sdan ** 3115296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error 3116296c7658Sdan ** code is returned. Otherwise, SQLITE_OK. 3117296c7658Sdan */ 3118d5f0767cSdan int sqlite3changeset_conflict( 3119296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3120296c7658Sdan int iVal, /* Index of conflict record value to fetch */ 3121d5f0767cSdan sqlite3_value **ppValue /* OUT: Value from conflicting row */ 3122d5f0767cSdan ){ 3123d5f0767cSdan if( !pIter->pConflict ){ 3124d5f0767cSdan return SQLITE_MISUSE; 3125d5f0767cSdan } 3126ff677b20Sdan if( iVal<0 || iVal>=pIter->nCol ){ 3127d5f0767cSdan return SQLITE_RANGE; 3128d5f0767cSdan } 3129d5f0767cSdan *ppValue = sqlite3_column_value(pIter->pConflict, iVal); 3130d5f0767cSdan return SQLITE_OK; 3131d5f0767cSdan } 3132d5f0767cSdan 31334fccf43aSdan /* 3134cb3e4b79Sdan ** This function may only be called with an iterator passed to an 3135cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case 3136cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key 3137cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK. 3138cb3e4b79Sdan ** 3139cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE. 3140cb3e4b79Sdan */ 3141cb3e4b79Sdan int sqlite3changeset_fk_conflicts( 3142cb3e4b79Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3143cb3e4b79Sdan int *pnOut /* OUT: Number of FK violations */ 3144cb3e4b79Sdan ){ 3145cb3e4b79Sdan if( pIter->pConflict || pIter->apValue ){ 3146cb3e4b79Sdan return SQLITE_MISUSE; 3147cb3e4b79Sdan } 3148cb3e4b79Sdan *pnOut = pIter->nCol; 3149cb3e4b79Sdan return SQLITE_OK; 3150cb3e4b79Sdan } 3151cb3e4b79Sdan 3152cb3e4b79Sdan 3153cb3e4b79Sdan /* 31544fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start(). 31554fccf43aSdan ** 31564fccf43aSdan ** This function may not be called on iterators passed to a conflict handler 31574fccf43aSdan ** callback by changeset_apply(). 31584fccf43aSdan */ 31594fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){ 3160cbf6d2d2Sdan int rc = SQLITE_OK; 3161cbf6d2d2Sdan if( p ){ 3162296c7658Sdan int i; /* Used to iterate through p->apValue[] */ 3163cbf6d2d2Sdan rc = p->rc; 316412ca0b56Sdan if( p->apValue ){ 31654fccf43aSdan for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]); 316612ca0b56Sdan } 3167ef7a6304Sdan sqlite3_free(p->tblhdr.aBuf); 31684757c658Sdan sqlite3_free(p->in.buf.aBuf); 31694fccf43aSdan sqlite3_free(p); 3170cbf6d2d2Sdan } 31714fccf43aSdan return rc; 31724fccf43aSdan } 31734fccf43aSdan 3174fa122adaSdan static int sessionChangesetInvert( 3175fa122adaSdan SessionInput *pInput, /* Input changeset */ 3176fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3177fa122adaSdan void *pOut, 317891ddd559Sdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 317991ddd559Sdan void **ppInverted /* OUT: Inverse of pChangeset */ 318091ddd559Sdan ){ 3181cfec7eeeSdan int rc = SQLITE_OK; /* Return value */ 3182fa122adaSdan SessionBuffer sOut; /* Output buffer */ 3183cfec7eeeSdan int nCol = 0; /* Number of cols in current table */ 3184cfec7eeeSdan u8 *abPK = 0; /* PK array for current table */ 3185cfec7eeeSdan sqlite3_value **apVal = 0; /* Space for values for UPDATE inversion */ 3186ef7a6304Sdan SessionBuffer sPK = {0, 0, 0}; /* PK array for current table */ 318791ddd559Sdan 3188fa122adaSdan /* Initialize the output buffer */ 3189fa122adaSdan memset(&sOut, 0, sizeof(SessionBuffer)); 3190fa122adaSdan 319191ddd559Sdan /* Zero the output variables in case an error occurs. */ 3192fa122adaSdan if( ppInverted ){ 319391ddd559Sdan *ppInverted = 0; 319491ddd559Sdan *pnInverted = 0; 3195fa122adaSdan } 319691ddd559Sdan 3197fa122adaSdan while( 1 ){ 3198ef7a6304Sdan u8 eType; 3199fa122adaSdan 3200fa122adaSdan /* Test for EOF. */ 3201fa122adaSdan if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert; 3202fa122adaSdan if( pInput->iNext>=pInput->nData ) break; 3203fa122adaSdan eType = pInput->aData[pInput->iNext]; 3204fa122adaSdan 320591ddd559Sdan switch( eType ){ 320691ddd559Sdan case 'T': { 3207244593c8Sdan /* A 'table' record consists of: 3208244593c8Sdan ** 3209244593c8Sdan ** * A constant 'T' character, 3210244593c8Sdan ** * Number of columns in said table (a varint), 3211ef7a6304Sdan ** * An array of nCol bytes (sPK), 3212244593c8Sdan ** * A nul-terminated table name. 3213244593c8Sdan */ 3214ef7a6304Sdan int nByte; 3215fa122adaSdan int nVar; 3216fa122adaSdan pInput->iNext++; 3217fa122adaSdan if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){ 3218ef7a6304Sdan goto finished_invert; 3219ef7a6304Sdan } 3220fa122adaSdan nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol); 3221ef7a6304Sdan sPK.nBuf = 0; 3222fa122adaSdan sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc); 3223fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3224fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3225ef7a6304Sdan if( rc ) goto finished_invert; 3226fa122adaSdan 3227fa122adaSdan pInput->iNext += nByte; 3228cfec7eeeSdan sqlite3_free(apVal); 3229cfec7eeeSdan apVal = 0; 3230ef7a6304Sdan abPK = sPK.aBuf; 323191ddd559Sdan break; 323291ddd559Sdan } 323391ddd559Sdan 323491ddd559Sdan case SQLITE_INSERT: 323591ddd559Sdan case SQLITE_DELETE: { 323691ddd559Sdan int nByte; 3237fa122adaSdan int bIndirect = pInput->aData[pInput->iNext+1]; 3238fa122adaSdan int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE); 3239fa122adaSdan pInput->iNext += 2; 3240fa122adaSdan assert( rc==SQLITE_OK ); 3241fa122adaSdan rc = sessionChangesetBufferRecord(pInput, nCol, &nByte); 3242fa122adaSdan sessionAppendByte(&sOut, eType2, &rc); 3243fa122adaSdan sessionAppendByte(&sOut, bIndirect, &rc); 3244fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3245fa122adaSdan pInput->iNext += nByte; 3246fa122adaSdan if( rc ) goto finished_invert; 324791ddd559Sdan break; 324891ddd559Sdan } 324991ddd559Sdan 325091ddd559Sdan case SQLITE_UPDATE: { 3251cfec7eeeSdan int iCol; 325291ddd559Sdan 3253cfec7eeeSdan if( 0==apVal ){ 3254cfec7eeeSdan apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2); 3255cfec7eeeSdan if( 0==apVal ){ 3256cfec7eeeSdan rc = SQLITE_NOMEM; 3257cfec7eeeSdan goto finished_invert; 3258cfec7eeeSdan } 3259cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3260cfec7eeeSdan } 326191ddd559Sdan 3262cfec7eeeSdan /* Write the header for the new UPDATE change. Same as the original. */ 3263fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3264fa122adaSdan sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc); 326591ddd559Sdan 3266ef7a6304Sdan /* Read the old.* and new.* records for the update change. */ 3267fa122adaSdan pInput->iNext += 2; 3268fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]); 3269ef7a6304Sdan if( rc==SQLITE_OK ){ 3270fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]); 3271ef7a6304Sdan } 3272ef7a6304Sdan 3273cfec7eeeSdan /* Write the new old.* record. Consists of the PK columns from the 3274cfec7eeeSdan ** original old.* record, and the other values from the original 3275cfec7eeeSdan ** new.* record. */ 3276e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3277cfec7eeeSdan sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)]; 3278fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3279cfec7eeeSdan } 3280cfec7eeeSdan 3281cfec7eeeSdan /* Write the new new.* record. Consists of a copy of all values 3282cfec7eeeSdan ** from the original old.* record, except for the PK columns, which 3283cfec7eeeSdan ** are set to "undefined". */ 3284e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3285cfec7eeeSdan sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]); 3286fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3287cfec7eeeSdan } 3288cfec7eeeSdan 3289cfec7eeeSdan for(iCol=0; iCol<nCol*2; iCol++){ 3290cfec7eeeSdan sqlite3ValueFree(apVal[iCol]); 3291cfec7eeeSdan } 3292cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3293cfec7eeeSdan if( rc!=SQLITE_OK ){ 3294cfec7eeeSdan goto finished_invert; 3295cfec7eeeSdan } 3296cfec7eeeSdan 329791ddd559Sdan break; 329891ddd559Sdan } 329991ddd559Sdan 330091ddd559Sdan default: 33014757c658Sdan rc = SQLITE_CORRUPT_BKPT; 3302cfec7eeeSdan goto finished_invert; 330391ddd559Sdan } 3304fa122adaSdan 3305fa122adaSdan assert( rc==SQLITE_OK ); 3306f1a08ad8Sdrh if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 3307fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3308fa122adaSdan sOut.nBuf = 0; 3309fa122adaSdan if( rc!=SQLITE_OK ) goto finished_invert; 3310fa122adaSdan } 331191ddd559Sdan } 331291ddd559Sdan 3313cfec7eeeSdan assert( rc==SQLITE_OK ); 3314fa122adaSdan if( pnInverted ){ 3315fa122adaSdan *pnInverted = sOut.nBuf; 3316fa122adaSdan *ppInverted = sOut.aBuf; 3317fa122adaSdan sOut.aBuf = 0; 3318fa122adaSdan }else if( sOut.nBuf>0 ){ 3319fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3320fa122adaSdan } 3321cfec7eeeSdan 3322cfec7eeeSdan finished_invert: 3323fa122adaSdan sqlite3_free(sOut.aBuf); 3324cfec7eeeSdan sqlite3_free(apVal); 3325ef7a6304Sdan sqlite3_free(sPK.aBuf); 3326cfec7eeeSdan return rc; 332791ddd559Sdan } 332891ddd559Sdan 3329fa122adaSdan 3330fa122adaSdan /* 3331fa122adaSdan ** Invert a changeset object. 3332fa122adaSdan */ 3333fa122adaSdan int sqlite3changeset_invert( 3334fa122adaSdan int nChangeset, /* Number of bytes in input */ 3335fa122adaSdan const void *pChangeset, /* Input changeset */ 3336fa122adaSdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 3337fa122adaSdan void **ppInverted /* OUT: Inverse of pChangeset */ 3338fa122adaSdan ){ 3339fa122adaSdan SessionInput sInput; 3340fa122adaSdan 3341fa122adaSdan /* Set up the input stream */ 3342fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3343fa122adaSdan sInput.nData = nChangeset; 3344fa122adaSdan sInput.aData = (u8*)pChangeset; 3345fa122adaSdan 3346fa122adaSdan return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted); 3347fa122adaSdan } 3348fa122adaSdan 3349fa122adaSdan /* 3350fa122adaSdan ** Streaming version of sqlite3changeset_invert(). 3351fa122adaSdan */ 3352f1a08ad8Sdrh int sqlite3changeset_invert_strm( 3353fa122adaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 3354fa122adaSdan void *pIn, 3355fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3356fa122adaSdan void *pOut 3357fa122adaSdan ){ 3358fa122adaSdan SessionInput sInput; 3359fa122adaSdan int rc; 3360fa122adaSdan 3361fa122adaSdan /* Set up the input stream */ 3362fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3363fa122adaSdan sInput.xInput = xInput; 3364fa122adaSdan sInput.pIn = pIn; 3365fa122adaSdan 3366fa122adaSdan rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0); 3367fa122adaSdan sqlite3_free(sInput.buf.aBuf); 3368fa122adaSdan return rc; 3369fa122adaSdan } 3370fa122adaSdan 33710c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx; 33720c698471Sdan struct SessionApplyCtx { 33730c698471Sdan sqlite3 *db; 33740c698471Sdan sqlite3_stmt *pDelete; /* DELETE statement */ 3375cfec7eeeSdan sqlite3_stmt *pUpdate; /* UPDATE statement */ 33760c698471Sdan sqlite3_stmt *pInsert; /* INSERT statement */ 33770c698471Sdan sqlite3_stmt *pSelect; /* SELECT statement */ 33780c698471Sdan int nCol; /* Size of azCol[] and abPK[] arrays */ 33790c698471Sdan const char **azCol; /* Array of column names */ 33800c698471Sdan u8 *abPK; /* Boolean array - true if column is in PK */ 3381d1cccf19Sdan int bStat1; /* True if table is sqlite_stat1 */ 3382d9151526Sdan int bDeferConstraints; /* True to defer constraints */ 3383d9151526Sdan SessionBuffer constraints; /* Deferred constraints are stored here */ 33840c698471Sdan }; 33850c698471Sdan 3386d5f0767cSdan /* 3387d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table 3388d5f0767cSdan ** structure like this: 3389d5f0767cSdan ** 3390d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3391d5f0767cSdan ** 3392d5f0767cSdan ** The DELETE statement looks like this: 3393d5f0767cSdan ** 3394db04571cSdan ** DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4) 3395d5f0767cSdan ** 3396d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require 3397d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the 3398d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE. 3399296c7658Sdan ** 3400296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left 3401296c7658Sdan ** pointing to the prepared version of the SQL statement. 3402d5f0767cSdan */ 3403d5f0767cSdan static int sessionDeleteRow( 3404d5f0767cSdan sqlite3 *db, /* Database handle */ 3405d5f0767cSdan const char *zTab, /* Table name */ 34060c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3407d5f0767cSdan ){ 3408296c7658Sdan int i; 3409296c7658Sdan const char *zSep = ""; 3410d5f0767cSdan int rc = SQLITE_OK; 3411d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 34127cf7df7dSdan int nPk = 0; 3413d5f0767cSdan 3414d5f0767cSdan sessionAppendStr(&buf, "DELETE FROM ", &rc); 3415d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3416296c7658Sdan sessionAppendStr(&buf, " WHERE ", &rc); 3417296c7658Sdan 3418296c7658Sdan for(i=0; i<p->nCol; i++){ 3419296c7658Sdan if( p->abPK[i] ){ 34207cf7df7dSdan nPk++; 3421296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3422296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3423296c7658Sdan sessionAppendStr(&buf, " = ?", &rc); 3424296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3425296c7658Sdan zSep = " AND "; 3426296c7658Sdan } 3427296c7658Sdan } 3428296c7658Sdan 34297cf7df7dSdan if( nPk<p->nCol ){ 3430296c7658Sdan sessionAppendStr(&buf, " AND (?", &rc); 3431296c7658Sdan sessionAppendInteger(&buf, p->nCol+1, &rc); 3432296c7658Sdan sessionAppendStr(&buf, " OR ", &rc); 3433296c7658Sdan 3434296c7658Sdan zSep = ""; 3435296c7658Sdan for(i=0; i<p->nCol; i++){ 3436296c7658Sdan if( !p->abPK[i] ){ 3437296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3438296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3439296c7658Sdan sessionAppendStr(&buf, " IS ?", &rc); 3440296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3441296c7658Sdan zSep = "AND "; 3442296c7658Sdan } 3443296c7658Sdan } 3444296c7658Sdan sessionAppendStr(&buf, ")", &rc); 34457cf7df7dSdan } 3446d5f0767cSdan 3447d5f0767cSdan if( rc==SQLITE_OK ){ 34480c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0); 3449d5f0767cSdan } 3450d5f0767cSdan sqlite3_free(buf.aBuf); 3451d5f0767cSdan 3452d5f0767cSdan return rc; 3453d5f0767cSdan } 3454d5f0767cSdan 3455d5f0767cSdan /* 3456d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db. 3457d5f0767cSdan ** Assuming a table structure like this: 3458d5f0767cSdan ** 3459d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3460d5f0767cSdan ** 3461d5f0767cSdan ** The UPDATE statement looks like this: 3462d5f0767cSdan ** 3463d5f0767cSdan ** UPDATE x SET 3464d5f0767cSdan ** a = CASE WHEN ?2 THEN ?3 ELSE a END, 3465964cbd46Sdan ** b = CASE WHEN ?5 THEN ?6 ELSE b END, 3466964cbd46Sdan ** c = CASE WHEN ?8 THEN ?9 ELSE c END, 3467964cbd46Sdan ** d = CASE WHEN ?11 THEN ?12 ELSE d END 3468d5f0767cSdan ** WHERE a = ?1 AND c = ?7 AND (?13 OR 3469964cbd46Sdan ** (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND 3470d5f0767cSdan ** ) 3471d5f0767cSdan ** 3472d5f0767cSdan ** For each column in the table, there are three variables to bind: 3473d5f0767cSdan ** 3474d5f0767cSdan ** ?(i*3+1) The old.* value of the column, if any. 3475d5f0767cSdan ** ?(i*3+2) A boolean flag indicating that the value is being modified. 3476d5f0767cSdan ** ?(i*3+3) The new.* value of the column, if any. 3477d5f0767cSdan ** 3478d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update 3479d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the 3480d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns 3481d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)". 3482d5f0767cSdan ** 3483296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left 3484296c7658Sdan ** pointing to the prepared version of the SQL statement. 3485d5f0767cSdan */ 3486d5f0767cSdan static int sessionUpdateRow( 3487d5f0767cSdan sqlite3 *db, /* Database handle */ 3488d5f0767cSdan const char *zTab, /* Table name */ 34890c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3490d5f0767cSdan ){ 3491d5f0767cSdan int rc = SQLITE_OK; 3492d5f0767cSdan int i; 3493d5f0767cSdan const char *zSep = ""; 3494d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 3495d5f0767cSdan 3496d5f0767cSdan /* Append "UPDATE tbl SET " */ 3497d5f0767cSdan sessionAppendStr(&buf, "UPDATE ", &rc); 3498d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3499d5f0767cSdan sessionAppendStr(&buf, " SET ", &rc); 3500d5f0767cSdan 3501d5f0767cSdan /* Append the assignments */ 35020c698471Sdan for(i=0; i<p->nCol; i++){ 3503d5f0767cSdan sessionAppendStr(&buf, zSep, &rc); 35040c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3505d5f0767cSdan sessionAppendStr(&buf, " = CASE WHEN ?", &rc); 3506d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3507d5f0767cSdan sessionAppendStr(&buf, " THEN ?", &rc); 3508d5f0767cSdan sessionAppendInteger(&buf, i*3+3, &rc); 3509d5f0767cSdan sessionAppendStr(&buf, " ELSE ", &rc); 35100c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3511d5f0767cSdan sessionAppendStr(&buf, " END", &rc); 3512d5f0767cSdan zSep = ", "; 3513d5f0767cSdan } 3514d5f0767cSdan 3515d5f0767cSdan /* Append the PK part of the WHERE clause */ 3516d5f0767cSdan sessionAppendStr(&buf, " WHERE ", &rc); 35170c698471Sdan for(i=0; i<p->nCol; i++){ 35180c698471Sdan if( p->abPK[i] ){ 35190c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3520d5f0767cSdan sessionAppendStr(&buf, " = ?", &rc); 3521d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3522d5f0767cSdan sessionAppendStr(&buf, " AND ", &rc); 3523d5f0767cSdan } 3524d5f0767cSdan } 3525d5f0767cSdan 3526d5f0767cSdan /* Append the non-PK part of the WHERE clause */ 3527d5f0767cSdan sessionAppendStr(&buf, " (?", &rc); 35280c698471Sdan sessionAppendInteger(&buf, p->nCol*3+1, &rc); 3529d5f0767cSdan sessionAppendStr(&buf, " OR 1", &rc); 35300c698471Sdan for(i=0; i<p->nCol; i++){ 35310c698471Sdan if( !p->abPK[i] ){ 3532d5f0767cSdan sessionAppendStr(&buf, " AND (?", &rc); 3533d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3534d5f0767cSdan sessionAppendStr(&buf, "=0 OR ", &rc); 35350c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3536d5f0767cSdan sessionAppendStr(&buf, " IS ?", &rc); 3537d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3538d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3539d5f0767cSdan } 3540d5f0767cSdan } 3541d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3542d5f0767cSdan 3543d5f0767cSdan if( rc==SQLITE_OK ){ 35440c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0); 3545d5f0767cSdan } 3546d5f0767cSdan sqlite3_free(buf.aBuf); 3547d5f0767cSdan 3548d5f0767cSdan return rc; 3549d5f0767cSdan } 3550d5f0767cSdan 35513739f298Sdan 3552296c7658Sdan /* 3553296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary 3554296c7658Sdan ** key. Assuming the following table structure: 3555296c7658Sdan ** 3556296c7658Sdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3557296c7658Sdan ** 3558296c7658Sdan ** The SELECT statement looks like this: 3559296c7658Sdan ** 3560296c7658Sdan ** SELECT * FROM x WHERE a = ?1 AND c = ?3 3561296c7658Sdan ** 3562296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left 3563296c7658Sdan ** pointing to the prepared version of the SQL statement. 3564296c7658Sdan */ 3565d5f0767cSdan static int sessionSelectRow( 3566d5f0767cSdan sqlite3 *db, /* Database handle */ 3567d5f0767cSdan const char *zTab, /* Table name */ 35680c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3569d5f0767cSdan ){ 3570d7fb7d24Sdan return sessionSelectStmt( 3571d7fb7d24Sdan db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect); 3572d5f0767cSdan } 3573d5f0767cSdan 3574296c7658Sdan /* 3575296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab. 3576296c7658Sdan ** For example: 3577296c7658Sdan ** 3578296c7658Sdan ** INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...); 3579296c7658Sdan ** 3580296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left 3581296c7658Sdan ** pointing to the prepared version of the SQL statement. 3582296c7658Sdan */ 35830c698471Sdan static int sessionInsertRow( 35840c698471Sdan sqlite3 *db, /* Database handle */ 35850c698471Sdan const char *zTab, /* Table name */ 35860c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 35870c698471Sdan ){ 35880c698471Sdan int rc = SQLITE_OK; 35890c698471Sdan int i; 35900c698471Sdan SessionBuffer buf = {0, 0, 0}; 35910c698471Sdan 35920c698471Sdan sessionAppendStr(&buf, "INSERT INTO main.", &rc); 35930c698471Sdan sessionAppendIdent(&buf, zTab, &rc); 3594ff677b20Sdan sessionAppendStr(&buf, "(", &rc); 3595ff677b20Sdan for(i=0; i<p->nCol; i++){ 3596ff677b20Sdan if( i!=0 ) sessionAppendStr(&buf, ", ", &rc); 3597ff677b20Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3598ff677b20Sdan } 3599ff677b20Sdan 3600ff677b20Sdan sessionAppendStr(&buf, ") VALUES(?", &rc); 36010c698471Sdan for(i=1; i<p->nCol; i++){ 36020c698471Sdan sessionAppendStr(&buf, ", ?", &rc); 36030c698471Sdan } 36040c698471Sdan sessionAppendStr(&buf, ")", &rc); 36050c698471Sdan 36060c698471Sdan if( rc==SQLITE_OK ){ 36070c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0); 36080c698471Sdan } 36090c698471Sdan sqlite3_free(buf.aBuf); 36100c698471Sdan return rc; 36110c698471Sdan } 36120c698471Sdan 36133739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){ 36143739f298Sdan return sqlite3_prepare_v2(db, zSql, -1, pp, 0); 36153739f298Sdan } 36163739f298Sdan 36173739f298Sdan /* 36183739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table. 36193739f298Sdan ** These are similar to those created by sessionSelectRow(), 36203739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for 36213739f298Sdan ** other tables. 36223739f298Sdan */ 36233739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){ 36243739f298Sdan int rc = sessionSelectRow(db, "sqlite_stat1", p); 36253739f298Sdan if( rc==SQLITE_OK ){ 36263739f298Sdan rc = sessionPrepare(db, &p->pInsert, 36273739f298Sdan "INSERT INTO main.sqlite_stat1 VALUES(?1, " 36283739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, " 36293739f298Sdan "?3)" 36303739f298Sdan ); 36313739f298Sdan } 36323739f298Sdan if( rc==SQLITE_OK ){ 36333739f298Sdan rc = sessionPrepare(db, &p->pUpdate, 36343739f298Sdan "UPDATE main.sqlite_stat1 SET " 36353739f298Sdan "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, " 36363739f298Sdan "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, " 36373739f298Sdan "stat = CASE WHEN ?8 THEN ?9 ELSE stat END " 36383739f298Sdan "WHERE tbl=?1 AND idx IS " 36393739f298Sdan "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END " 36403739f298Sdan "AND (?10 OR ?8=0 OR stat IS ?7)" 36413739f298Sdan ); 36423739f298Sdan } 36433739f298Sdan if( rc==SQLITE_OK ){ 36443739f298Sdan rc = sessionPrepare(db, &p->pDelete, 36453739f298Sdan "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS " 36463739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END " 36473739f298Sdan "AND (?4 OR stat IS ?3)" 36483739f298Sdan ); 36493739f298Sdan } 36503739f298Sdan assert( rc==SQLITE_OK ); 36513739f298Sdan return rc; 36523739f298Sdan } 36533739f298Sdan 3654296c7658Sdan /* 36557aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem. 36567aa469cdSdan ** See comments in the body of this function for details. 36577aa469cdSdan */ 36587aa469cdSdan static int sessionBindValue( 36597aa469cdSdan sqlite3_stmt *pStmt, /* Statement to bind value to */ 36607aa469cdSdan int i, /* Parameter number to bind to */ 36617aa469cdSdan sqlite3_value *pVal /* Value to bind */ 36627aa469cdSdan ){ 36635671ef69Sdrh int eType = sqlite3_value_type(pVal); 3664082c96dfSdan /* COVERAGE: The (pVal->z==0) branch is never true using current versions 3665082c96dfSdan ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either 3666082c96dfSdan ** the (pVal->z) variable remains as it was or the type of the value is 3667082c96dfSdan ** set to SQLITE_NULL. */ 36685671ef69Sdrh if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){ 36697aa469cdSdan /* This condition occurs when an earlier OOM in a call to 36707aa469cdSdan ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within 3671082c96dfSdan ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */ 36727aa469cdSdan return SQLITE_NOMEM; 36737aa469cdSdan } 36747aa469cdSdan return sqlite3_bind_value(pStmt, i, pVal); 36757aa469cdSdan } 36767aa469cdSdan 36777aa469cdSdan /* 3678db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function 3679db04571cSdan ** transfers new.* values from the current iterator entry to statement 3680db04571cSdan ** pStmt. The table being inserted into has nCol columns. 3681db04571cSdan ** 3682d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of 3683db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1) 3684db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points 3685db04571cSdan ** to an array nCol elements in size. In this case only those values for 3686db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the 3687db04571cSdan ** statement. 3688db04571cSdan ** 3689db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK. 3690db04571cSdan */ 36917aa469cdSdan static int sessionBindRow( 3692db04571cSdan sqlite3_changeset_iter *pIter, /* Iterator to read values from */ 36937aa469cdSdan int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **), 3694db04571cSdan int nCol, /* Number of columns */ 3695db04571cSdan u8 *abPK, /* If not NULL, bind only if true */ 3696db04571cSdan sqlite3_stmt *pStmt /* Bind values to this statement */ 3697db04571cSdan ){ 3698db04571cSdan int i; 3699db04571cSdan int rc = SQLITE_OK; 37007aa469cdSdan 37017aa469cdSdan /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the 37027aa469cdSdan ** argument iterator points to a suitable entry. Make sure that xValue 37037aa469cdSdan ** is one of these to guarantee that it is safe to ignore the return 37047aa469cdSdan ** in the code below. */ 37057aa469cdSdan assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new ); 37067aa469cdSdan 3707db04571cSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 3708db04571cSdan if( !abPK || abPK[i] ){ 3709db04571cSdan sqlite3_value *pVal; 37107aa469cdSdan (void)xValue(pIter, i, &pVal); 37117aa469cdSdan rc = sessionBindValue(pStmt, i+1, pVal); 3712db04571cSdan } 3713db04571cSdan } 3714db04571cSdan return rc; 3715db04571cSdan } 3716db04571cSdan 3717db04571cSdan /* 3718296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function. 3719296c7658Sdan ** This function binds the primary key values from the change that changeset 3720296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table 3721296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row 3722296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error 3723296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an 37247aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned. 37257aa469cdSdan ** 37267aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset() 37277aa469cdSdan ** statement pSelect. If any other value is returned, the statement does 37287aa469cdSdan ** not require a reset(). 3729296c7658Sdan ** 3730296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the 3731db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or 3732db04571cSdan ** UPDATE, bind values from the old.* record. 3733296c7658Sdan */ 37340c698471Sdan static int sessionSeekToRow( 373537f133ecSdan sqlite3 *db, /* Database handle */ 373637f133ecSdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 373737f133ecSdan u8 *abPK, /* Primary key flags array */ 37380c698471Sdan sqlite3_stmt *pSelect /* SELECT statement from sessionSelectRow() */ 373937f133ecSdan ){ 37407aa469cdSdan int rc; /* Return code */ 3741296c7658Sdan int nCol; /* Number of columns in table */ 3742296c7658Sdan int op; /* Changset operation (SQLITE_UPDATE etc.) */ 3743296c7658Sdan const char *zDummy; /* Unused */ 374437f133ecSdan 3745b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 37467aa469cdSdan rc = sessionBindRow(pIter, 3747db04571cSdan op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old, 3748db04571cSdan nCol, abPK, pSelect 3749db04571cSdan ); 37500c698471Sdan 37510c698471Sdan if( rc==SQLITE_OK ){ 37520c698471Sdan rc = sqlite3_step(pSelect); 37530c698471Sdan if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect); 37540c698471Sdan } 37550c698471Sdan 37560c698471Sdan return rc; 37570c698471Sdan } 37580c698471Sdan 3759296c7658Sdan /* 3760296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator 3761296c7658Sdan ** currently points to. 3762296c7658Sdan ** 3763296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT. 3764296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked 3765296c7658Sdan ** depends solely on eType, as follows: 3766296c7658Sdan ** 3767296c7658Sdan ** eType value Value passed to xConflict 3768296c7658Sdan ** ------------------------------------------------- 3769296c7658Sdan ** CHANGESET_DATA CHANGESET_NOTFOUND 3770296c7658Sdan ** CHANGESET_CONFLICT CHANGESET_CONSTRAINT 3771296c7658Sdan ** 3772296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing 3773296c7658Sdan ** record with the same primary key as the record about to be deleted, updated 3774296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict 3775296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict 3776296c7658Sdan ** handler invoked is as follows: 3777296c7658Sdan ** 3778296c7658Sdan ** eType value PK Record found? Value passed to xConflict 3779296c7658Sdan ** ---------------------------------------------------------------- 3780296c7658Sdan ** CHANGESET_DATA Yes CHANGESET_DATA 3781296c7658Sdan ** CHANGESET_DATA No CHANGESET_NOTFOUND 3782296c7658Sdan ** CHANGESET_CONFLICT Yes CHANGESET_CONFLICT 3783296c7658Sdan ** CHANGESET_CONFLICT No CHANGESET_CONSTRAINT 3784296c7658Sdan ** 3785296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and 3786296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace 3787296c7658Sdan ** is set to non-zero before returning SQLITE_OK. 3788296c7658Sdan ** 3789296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is 3790296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value, 3791296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT, 3792296c7658Sdan ** this function returns SQLITE_OK. 3793296c7658Sdan */ 37940c698471Sdan static int sessionConflictHandler( 3795296c7658Sdan int eType, /* Either CHANGESET_DATA or CONFLICT */ 3796296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 37970c698471Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 37980c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter*), 3799296c7658Sdan void *pCtx, /* First argument for conflict handler */ 3800296c7658Sdan int *pbReplace /* OUT: Set to true if PK row is found */ 38010c698471Sdan ){ 380274f598b6Smistachkin int res = 0; /* Value returned by conflict handler */ 38030c698471Sdan int rc; 38040c698471Sdan int nCol; 38050c698471Sdan int op; 38060c698471Sdan const char *zDummy; 38070c698471Sdan 3808b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 38090c698471Sdan 38100c698471Sdan assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA ); 38110c698471Sdan assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT ); 38120c698471Sdan assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND ); 381337f133ecSdan 381437f133ecSdan /* Bind the new.* PRIMARY KEY values to the SELECT statement. */ 38150c698471Sdan if( pbReplace ){ 38160c698471Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 38170c698471Sdan }else{ 3818db04571cSdan rc = SQLITE_OK; 38190c698471Sdan } 38200c698471Sdan 38210c698471Sdan if( rc==SQLITE_ROW ){ 38220c698471Sdan /* There exists another row with the new.* primary key. */ 38230c698471Sdan pIter->pConflict = p->pSelect; 38240c698471Sdan res = xConflict(pCtx, eType, pIter); 38250c698471Sdan pIter->pConflict = 0; 38260c698471Sdan rc = sqlite3_reset(p->pSelect); 3827db04571cSdan }else if( rc==SQLITE_OK ){ 3828d9151526Sdan if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){ 3829d9151526Sdan /* Instead of invoking the conflict handler, append the change blob 3830d9151526Sdan ** to the SessionApplyCtx.constraints buffer. */ 3831d9151526Sdan u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent]; 3832d9151526Sdan int nBlob = pIter->in.iNext - pIter->in.iCurrent; 3833d9151526Sdan sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc); 3834d9151526Sdan res = SQLITE_CHANGESET_OMIT; 3835d9151526Sdan }else{ 38360c698471Sdan /* No other row with the new.* primary key. */ 38370c698471Sdan res = xConflict(pCtx, eType+1, pIter); 38380c698471Sdan if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE; 383937f133ecSdan } 3840d9151526Sdan } 384137f133ecSdan 384237f133ecSdan if( rc==SQLITE_OK ){ 38430c698471Sdan switch( res ){ 38440c698471Sdan case SQLITE_CHANGESET_REPLACE: 3845f51e5f6cSdan assert( pbReplace ); 3846f51e5f6cSdan *pbReplace = 1; 38470c698471Sdan break; 38480c698471Sdan 38490c698471Sdan case SQLITE_CHANGESET_OMIT: 38500c698471Sdan break; 38510c698471Sdan 38520c698471Sdan case SQLITE_CHANGESET_ABORT: 38530c698471Sdan rc = SQLITE_ABORT; 38540c698471Sdan break; 38550c698471Sdan 38560c698471Sdan default: 38570c698471Sdan rc = SQLITE_MISUSE; 38580c698471Sdan break; 38590c698471Sdan } 38600c698471Sdan } 38610c698471Sdan 38620c698471Sdan return rc; 38630c698471Sdan } 38640c698471Sdan 3865296c7658Sdan /* 3866296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument 3867296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke 3868296c7658Sdan ** the conflict handler callback. 3869296c7658Sdan ** 3870296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If 3871296c7658Sdan ** one is encountered, update or delete the row with the matching primary key 3872296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs, 3873296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry 3874296c7658Sdan ** to true before returning. In this case the caller will invoke this function 3875296c7658Sdan ** again, this time with pbRetry set to NULL. 3876296c7658Sdan ** 3877296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is 3878296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead. 3879296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such 3880296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true 3881296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting 3882296c7658Sdan ** row before invoking this function again, this time with pbReplace set 3883296c7658Sdan ** to NULL. 3884296c7658Sdan ** 3885296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function 3886296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is 3887296c7658Sdan ** returned. 3888296c7658Sdan */ 38890c698471Sdan static int sessionApplyOneOp( 3890296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3891296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 38920c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter *), 3893296c7658Sdan void *pCtx, /* First argument for the conflict handler */ 3894296c7658Sdan int *pbReplace, /* OUT: True to remove PK row and retry */ 3895296c7658Sdan int *pbRetry /* OUT: True to retry. */ 38960c698471Sdan ){ 38970c698471Sdan const char *zDummy; 38980c698471Sdan int op; 38990c698471Sdan int nCol; 39000c698471Sdan int rc = SQLITE_OK; 39010c698471Sdan 39020c698471Sdan assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect ); 39030c698471Sdan assert( p->azCol && p->abPK ); 39040c698471Sdan assert( !pbReplace || *pbReplace==0 ); 39050c698471Sdan 3906b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 39070c698471Sdan 39080c698471Sdan if( op==SQLITE_DELETE ){ 39090c698471Sdan 391073b3c055Sdan /* Bind values to the DELETE statement. If conflict handling is required, 391173b3c055Sdan ** bind values for all columns and set bound variable (nCol+1) to true. 391273b3c055Sdan ** Or, if conflict handling is not required, bind just the PK column 391373b3c055Sdan ** values and, if it exists, set (nCol+1) to false. Conflict handling 391473b3c055Sdan ** is not required if: 391573b3c055Sdan ** 391673b3c055Sdan ** * this is a patchset, or 391773b3c055Sdan ** * (pbRetry==0), or 391873b3c055Sdan ** * all columns of the table are PK columns (in this case there is 391973b3c055Sdan ** no (nCol+1) variable to bind to). 392073b3c055Sdan */ 392173b3c055Sdan u8 *abPK = (pIter->bPatchset ? p->abPK : 0); 392273b3c055Sdan rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete); 39237cf7df7dSdan if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){ 392473b3c055Sdan rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK)); 39257cf7df7dSdan } 39260c698471Sdan if( rc!=SQLITE_OK ) return rc; 39270c698471Sdan 39280c698471Sdan sqlite3_step(p->pDelete); 39290c698471Sdan rc = sqlite3_reset(p->pDelete); 39300c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 39310c698471Sdan rc = sessionConflictHandler( 39320c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 39330c698471Sdan ); 393435e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 39350c698471Sdan rc = sessionConflictHandler( 39360c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 39370c698471Sdan ); 39380c698471Sdan } 39390c698471Sdan 39400c698471Sdan }else if( op==SQLITE_UPDATE ){ 39410c698471Sdan int i; 39420c698471Sdan 39430c698471Sdan /* Bind values to the UPDATE statement. */ 39440c698471Sdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 39457aa469cdSdan sqlite3_value *pOld = sessionChangesetOld(pIter, i); 39467aa469cdSdan sqlite3_value *pNew = sessionChangesetNew(pIter, i); 39477aa469cdSdan 39480c698471Sdan sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew); 39497aa469cdSdan if( pOld ){ 39507aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+1, pOld); 39517aa469cdSdan } 39527aa469cdSdan if( rc==SQLITE_OK && pNew ){ 39537aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+3, pNew); 39540c698471Sdan } 39550c698471Sdan } 395673b3c055Sdan if( rc==SQLITE_OK ){ 395773b3c055Sdan sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset); 395873b3c055Sdan } 39590c698471Sdan if( rc!=SQLITE_OK ) return rc; 39600c698471Sdan 39610c698471Sdan /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict, 39620c698471Sdan ** the result will be SQLITE_OK with 0 rows modified. */ 39630c698471Sdan sqlite3_step(p->pUpdate); 39640c698471Sdan rc = sqlite3_reset(p->pUpdate); 39650c698471Sdan 39660c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 39670c698471Sdan /* A NOTFOUND or DATA error. Search the table to see if it contains 39680c698471Sdan ** a row with a matching primary key. If so, this is a DATA conflict. 39690c698471Sdan ** Otherwise, if there is no primary key match, it is a NOTFOUND. */ 39700c698471Sdan 39710c698471Sdan rc = sessionConflictHandler( 39720c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 39730c698471Sdan ); 39740c698471Sdan 397535e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 3976db04571cSdan /* This is always a CONSTRAINT conflict. */ 3977db04571cSdan rc = sessionConflictHandler( 3978db04571cSdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 39790c698471Sdan ); 39800c698471Sdan } 39810c698471Sdan 39820c698471Sdan }else{ 39830c698471Sdan assert( op==SQLITE_INSERT ); 3984d1cccf19Sdan if( p->bStat1 ){ 3985d1cccf19Sdan /* Check if there is a conflicting row. For sqlite_stat1, this needs 3986d1cccf19Sdan ** to be done using a SELECT, as there is no PRIMARY KEY in the 3987d1cccf19Sdan ** database schema to throw an exception if a duplicate is inserted. */ 3988d1cccf19Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 3989d1cccf19Sdan if( rc==SQLITE_ROW ){ 3990d1cccf19Sdan rc = SQLITE_CONSTRAINT; 3991d1cccf19Sdan sqlite3_reset(p->pSelect); 3992d1cccf19Sdan } 3993d1cccf19Sdan } 3994d1cccf19Sdan 3995d1cccf19Sdan if( rc==SQLITE_OK ){ 39967aa469cdSdan rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert); 39970c698471Sdan if( rc!=SQLITE_OK ) return rc; 39980c698471Sdan 39990c698471Sdan sqlite3_step(p->pInsert); 40000c698471Sdan rc = sqlite3_reset(p->pInsert); 4001d1cccf19Sdan } 4002d1cccf19Sdan 400335e2858eSdan if( (rc&0xff)==SQLITE_CONSTRAINT ){ 40040c698471Sdan rc = sessionConflictHandler( 40050c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace 40060c698471Sdan ); 400737f133ecSdan } 400837f133ecSdan } 400937f133ecSdan 401037f133ecSdan return rc; 401137f133ecSdan } 401237f133ecSdan 40135f5663dcSdan /* 40145f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument 40155f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke 40165f5663dcSdan ** the conflict handler callback. 40175f5663dcSdan ** 40185f5663dcSdan ** The difference between this function and sessionApplyOne() is that this 40195f5663dcSdan ** function handles the case where the conflict-handler is invoked and 40205f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be 40215f5663dcSdan ** retried in some manner. 40225f5663dcSdan */ 4023d9151526Sdan static int sessionApplyOneWithRetry( 4024d9151526Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4025d9151526Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator to read change from */ 4026d9151526Sdan SessionApplyCtx *pApply, /* Apply context */ 4027d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4028d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4029d9151526Sdan ){ 4030d9151526Sdan int bReplace = 0; 4031d9151526Sdan int bRetry = 0; 4032d9151526Sdan int rc; 4033d9151526Sdan 4034d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry); 40355f5663dcSdan assert( rc==SQLITE_OK || (bRetry==0 && bReplace==0) ); 4036d9151526Sdan 40375f5663dcSdan /* If the bRetry flag is set, the change has not been applied due to an 40385f5663dcSdan ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and 40395f5663dcSdan ** a row with the correct PK is present in the db, but one or more other 40405f5663dcSdan ** fields do not contain the expected values) and the conflict handler 40415f5663dcSdan ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation, 40425f5663dcSdan ** but pass NULL as the final argument so that sessionApplyOneOp() ignores 40435f5663dcSdan ** the SQLITE_CHANGESET_DATA problem. */ 40445f5663dcSdan if( bRetry ){ 40455f5663dcSdan assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE ); 40465f5663dcSdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4047d9151526Sdan } 4048d9151526Sdan 40495f5663dcSdan /* If the bReplace flag is set, the change is an INSERT that has not 40505f5663dcSdan ** been performed because the database already contains a row with the 40515f5663dcSdan ** specified primary key and the conflict handler returned 40525f5663dcSdan ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row 40535f5663dcSdan ** before reattempting the INSERT. */ 40545f5663dcSdan else if( bReplace ){ 4055d9151526Sdan assert( pIter->op==SQLITE_INSERT ); 4056d9151526Sdan rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0); 4057d9151526Sdan if( rc==SQLITE_OK ){ 4058d9151526Sdan rc = sessionBindRow(pIter, 4059d9151526Sdan sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete); 4060d9151526Sdan sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1); 4061d9151526Sdan } 4062d9151526Sdan if( rc==SQLITE_OK ){ 4063d9151526Sdan sqlite3_step(pApply->pDelete); 4064d9151526Sdan rc = sqlite3_reset(pApply->pDelete); 4065d9151526Sdan } 4066d9151526Sdan if( rc==SQLITE_OK ){ 4067d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4068d9151526Sdan } 4069d9151526Sdan if( rc==SQLITE_OK ){ 4070d9151526Sdan rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0); 4071d9151526Sdan } 4072d9151526Sdan } 4073d9151526Sdan 4074d9151526Sdan return rc; 4075d9151526Sdan } 4076d9151526Sdan 4077d9151526Sdan /* 4078d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer. 4079d9151526Sdan */ 4080d9151526Sdan static int sessionRetryConstraints( 4081d9151526Sdan sqlite3 *db, 4082d9151526Sdan int bPatchset, 4083d9151526Sdan const char *zTab, 4084d9151526Sdan SessionApplyCtx *pApply, 4085d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4086d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4087d9151526Sdan ){ 4088d9151526Sdan int rc = SQLITE_OK; 4089d9151526Sdan 4090d9151526Sdan while( pApply->constraints.nBuf ){ 4091d9151526Sdan sqlite3_changeset_iter *pIter2 = 0; 4092d9151526Sdan SessionBuffer cons = pApply->constraints; 4093d9151526Sdan memset(&pApply->constraints, 0, sizeof(SessionBuffer)); 4094d9151526Sdan 4095d9151526Sdan rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf); 4096d9151526Sdan if( rc==SQLITE_OK ){ 4097d9151526Sdan int nByte = 2*pApply->nCol*sizeof(sqlite3_value*); 4098d9151526Sdan int rc2; 4099d9151526Sdan pIter2->bPatchset = bPatchset; 4100d9151526Sdan pIter2->zTab = (char*)zTab; 4101d9151526Sdan pIter2->nCol = pApply->nCol; 4102d9151526Sdan pIter2->abPK = pApply->abPK; 4103d9151526Sdan sessionBufferGrow(&pIter2->tblhdr, nByte, &rc); 4104d9151526Sdan pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf; 4105d9151526Sdan if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte); 4106d9151526Sdan 4107d9151526Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){ 4108d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx); 4109d9151526Sdan } 4110d9151526Sdan 4111d9151526Sdan rc2 = sqlite3changeset_finalize(pIter2); 41127e0765a9Sdrh if( rc==SQLITE_OK ) rc = rc2; 4113d9151526Sdan } 4114d9151526Sdan assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 ); 4115d9151526Sdan 4116d9151526Sdan sqlite3_free(cons.aBuf); 4117d9151526Sdan if( rc!=SQLITE_OK ) break; 4118d9151526Sdan if( pApply->constraints.nBuf>=cons.nBuf ){ 4119d9151526Sdan /* No progress was made on the last round. */ 4120d9151526Sdan pApply->bDeferConstraints = 0; 4121d9151526Sdan } 4122d9151526Sdan } 4123d9151526Sdan 4124d9151526Sdan return rc; 4125d9151526Sdan } 4126d9151526Sdan 4127296c7658Sdan /* 41284757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but 41294757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the 41304757c658Sdan ** changeset to the main database attached to handle "db". The supplied 41314757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered 41324757c658Sdan ** while applying the change. 4133296c7658Sdan */ 41344757c658Sdan static int sessionChangesetApply( 4135296c7658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 41364757c658Sdan sqlite3_changeset_iter *pIter, /* Changeset to apply */ 413740368988Sdan int(*xFilter)( 413840368988Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 413940368988Sdan const char *zTab /* Table name */ 414040368988Sdan ), 4141d5f0767cSdan int(*xConflict)( 4142d5f0767cSdan void *pCtx, /* Copy of fifth arg to _apply() */ 4143d5f0767cSdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4144d5f0767cSdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4145d5f0767cSdan ), 4146296c7658Sdan void *pCtx /* First argument passed to xConflict */ 4147d5f0767cSdan ){ 4148ca62ad57Sdan int schemaMismatch = 0; 4149296c7658Sdan int rc; /* Return code */ 4150d5f0767cSdan const char *zTab = 0; /* Name of current table */ 4151cfdbde21Sdrh int nTab = 0; /* Result of sqlite3Strlen30(zTab) */ 4152296c7658Sdan SessionApplyCtx sApply; /* changeset_apply() context object */ 41535f5663dcSdan int bPatchset; 4154d5f0767cSdan 4155082c96dfSdan assert( xConflict!=0 ); 4156082c96dfSdan 4157d9151526Sdan pIter->in.bNoDiscard = 1; 41580c698471Sdan memset(&sApply, 0, sizeof(sApply)); 41594c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 41600c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0); 4161cb3e4b79Sdan if( rc==SQLITE_OK ){ 4162cb3e4b79Sdan rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0); 4163cb3e4b79Sdan } 41640c698471Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){ 41650c698471Sdan int nCol; 4166d5f0767cSdan int op; 41670c698471Sdan const char *zNew; 4168ca62ad57Sdan 4169b4480e94Sdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0); 4170d5f0767cSdan 41710c698471Sdan if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){ 4172ca62ad57Sdan u8 *abPK; 4173ca62ad57Sdan 4174d9151526Sdan rc = sessionRetryConstraints( 4175d9151526Sdan db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx 4176d9151526Sdan ); 4177d9151526Sdan if( rc!=SQLITE_OK ) break; 4178d9151526Sdan 4179cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 41800c698471Sdan sqlite3_finalize(sApply.pDelete); 41810c698471Sdan sqlite3_finalize(sApply.pUpdate); 41820c698471Sdan sqlite3_finalize(sApply.pInsert); 41830c698471Sdan sqlite3_finalize(sApply.pSelect); 41840c698471Sdan memset(&sApply, 0, sizeof(sApply)); 41850c698471Sdan sApply.db = db; 4186d9151526Sdan sApply.bDeferConstraints = 1; 418737f133ecSdan 418840368988Sdan /* If an xFilter() callback was specified, invoke it now. If the 418940368988Sdan ** xFilter callback returns zero, skip this table. If it returns 419040368988Sdan ** non-zero, proceed. */ 419140368988Sdan schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew))); 419240368988Sdan if( schemaMismatch ){ 419340368988Sdan zTab = sqlite3_mprintf("%s", zNew); 4194f05ac112Sdan if( zTab==0 ){ 4195f05ac112Sdan rc = SQLITE_NOMEM; 4196f05ac112Sdan break; 4197f05ac112Sdan } 41984f528042Sdan nTab = (int)strlen(zTab); 419940368988Sdan sApply.azCol = (const char **)zTab; 420040368988Sdan }else{ 4201ff677b20Sdan int nMinCol = 0; 4202ff677b20Sdan int i; 4203ff677b20Sdan 4204ca62ad57Sdan sqlite3changeset_pk(pIter, &abPK, 0); 4205296c7658Sdan rc = sessionTableInfo( 4206ca62ad57Sdan db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK 4207ca62ad57Sdan ); 4208ca62ad57Sdan if( rc!=SQLITE_OK ) break; 4209ff677b20Sdan for(i=0; i<sApply.nCol; i++){ 4210ff677b20Sdan if( sApply.abPK[i] ) nMinCol = i+1; 4211ff677b20Sdan } 42120c698471Sdan 4213ca62ad57Sdan if( sApply.nCol==0 ){ 4214ca62ad57Sdan schemaMismatch = 1; 4215ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4216ca62ad57Sdan "sqlite3changeset_apply(): no such table: %s", zTab 4217ca62ad57Sdan ); 4218ca62ad57Sdan } 4219ff677b20Sdan else if( sApply.nCol<nCol ){ 4220ca62ad57Sdan schemaMismatch = 1; 4221ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4222ff677b20Sdan "sqlite3changeset_apply(): table %s has %d columns, " 4223ff677b20Sdan "expected %d or more", 4224ca62ad57Sdan zTab, sApply.nCol, nCol 4225ca62ad57Sdan ); 4226ca62ad57Sdan } 4227ff677b20Sdan else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){ 4228ca62ad57Sdan schemaMismatch = 1; 422940368988Sdan sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): " 423040368988Sdan "primary key mismatch for table %s", zTab 4231ca62ad57Sdan ); 4232ca62ad57Sdan } 4233ff677b20Sdan else{ 4234ff677b20Sdan sApply.nCol = nCol; 42353739f298Sdan if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){ 42363739f298Sdan if( (rc = sessionStat1Sql(db, &sApply) ) ){ 42373739f298Sdan break; 42383739f298Sdan } 4239d1cccf19Sdan sApply.bStat1 = 1; 4240d1cccf19Sdan }else{ 4241ff677b20Sdan if((rc = sessionSelectRow(db, zTab, &sApply)) 42420c698471Sdan || (rc = sessionUpdateRow(db, zTab, &sApply)) 42430c698471Sdan || (rc = sessionDeleteRow(db, zTab, &sApply)) 42440c698471Sdan || (rc = sessionInsertRow(db, zTab, &sApply)) 424537f133ecSdan ){ 424637f133ecSdan break; 424737f133ecSdan } 4248d1cccf19Sdan sApply.bStat1 = 0; 4249d1cccf19Sdan } 4250ff677b20Sdan } 4251cfdbde21Sdrh nTab = sqlite3Strlen30(zTab); 4252d5f0767cSdan } 425340368988Sdan } 4254d5f0767cSdan 4255ca62ad57Sdan /* If there is a schema mismatch on the current table, proceed to the 4256ca62ad57Sdan ** next change. A log message has already been issued. */ 4257ca62ad57Sdan if( schemaMismatch ) continue; 4258ca62ad57Sdan 4259d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx); 42600c698471Sdan } 42610c698471Sdan 42625f5663dcSdan bPatchset = pIter->bPatchset; 4263296c7658Sdan if( rc==SQLITE_OK ){ 4264296c7658Sdan rc = sqlite3changeset_finalize(pIter); 4265296c7658Sdan }else{ 4266296c7658Sdan sqlite3changeset_finalize(pIter); 4267296c7658Sdan } 4268d5f0767cSdan 4269d5f0767cSdan if( rc==SQLITE_OK ){ 42705f5663dcSdan rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx); 42715f5663dcSdan } 42725f5663dcSdan 42735f5663dcSdan if( rc==SQLITE_OK ){ 427407001c45Sdrh int nFk, notUsed; 427507001c45Sdrh sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, ¬Used, 0); 427607001c45Sdrh if( nFk!=0 ){ 4277cb3e4b79Sdan int res = SQLITE_CHANGESET_ABORT; 4278cb3e4b79Sdan sqlite3_changeset_iter sIter; 4279cb3e4b79Sdan memset(&sIter, 0, sizeof(sIter)); 4280cb3e4b79Sdan sIter.nCol = nFk; 4281cb3e4b79Sdan res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter); 4282cb3e4b79Sdan if( res!=SQLITE_CHANGESET_OMIT ){ 4283cb3e4b79Sdan rc = SQLITE_CONSTRAINT; 4284cb3e4b79Sdan } 4285cb3e4b79Sdan } 4286cb3e4b79Sdan } 4287cb3e4b79Sdan sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0); 4288cb3e4b79Sdan 4289cb3e4b79Sdan if( rc==SQLITE_OK ){ 4290d5f0767cSdan rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4291d5f0767cSdan }else{ 4292d5f0767cSdan sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0); 4293d5f0767cSdan sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4294d5f0767cSdan } 4295d5f0767cSdan 42960c698471Sdan sqlite3_finalize(sApply.pInsert); 42970c698471Sdan sqlite3_finalize(sApply.pDelete); 42980c698471Sdan sqlite3_finalize(sApply.pUpdate); 42990c698471Sdan sqlite3_finalize(sApply.pSelect); 4300cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 4301d9151526Sdan sqlite3_free((char*)sApply.constraints.aBuf); 43024c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 4303d5f0767cSdan return rc; 4304d5f0767cSdan } 430591ddd559Sdan 430677fc1d5bSdan /* 43074757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database 43084757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 43094757c658Sdan ** to resolve any conflicts encountered while applying the change. 43104757c658Sdan */ 43114757c658Sdan int sqlite3changeset_apply( 43124757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 43134757c658Sdan int nChangeset, /* Size of changeset in bytes */ 43144757c658Sdan void *pChangeset, /* Changeset blob */ 43154757c658Sdan int(*xFilter)( 43164757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 43174757c658Sdan const char *zTab /* Table name */ 43184757c658Sdan ), 43194757c658Sdan int(*xConflict)( 43204757c658Sdan void *pCtx, /* Copy of fifth arg to _apply() */ 43214757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 43224757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 43234757c658Sdan ), 43244757c658Sdan void *pCtx /* First argument passed to xConflict */ 43254757c658Sdan ){ 43264757c658Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 43274757c658Sdan int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset); 43284757c658Sdan if( rc==SQLITE_OK ){ 43294757c658Sdan rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx); 43304757c658Sdan } 43314757c658Sdan return rc; 43324757c658Sdan } 43334757c658Sdan 43344757c658Sdan /* 43354757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database 43364757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 43374757c658Sdan ** to resolve any conflicts encountered while applying the change. 43384757c658Sdan */ 4339f1a08ad8Sdrh int sqlite3changeset_apply_strm( 43404757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 43414757c658Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 43424757c658Sdan void *pIn, /* First arg for xInput */ 43434757c658Sdan int(*xFilter)( 43444757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 43454757c658Sdan const char *zTab /* Table name */ 43464757c658Sdan ), 43474757c658Sdan int(*xConflict)( 43484757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 43494757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 43504757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 43514757c658Sdan ), 43524757c658Sdan void *pCtx /* First argument passed to xConflict */ 43534757c658Sdan ){ 43544757c658Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 4355f1a08ad8Sdrh int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 43564757c658Sdan if( rc==SQLITE_OK ){ 43574757c658Sdan rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx); 43584757c658Sdan } 43594757c658Sdan return rc; 43604757c658Sdan } 43614757c658Sdan 43624757c658Sdan /* 43635898ad69Sdan ** sqlite3_changegroup handle. 43645898ad69Sdan */ 43655898ad69Sdan struct sqlite3_changegroup { 43665898ad69Sdan int rc; /* Error code */ 43675898ad69Sdan int bPatch; /* True to accumulate patchsets */ 43685898ad69Sdan SessionTable *pList; /* List of tables in current patch */ 43695898ad69Sdan }; 43705898ad69Sdan 43715898ad69Sdan /* 437277fc1d5bSdan ** This function is called to merge two changes to the same row together as 437377fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is 437477fc1d5bSdan ** allocated and a pointer to it stored in *ppNew. 437577fc1d5bSdan */ 43765d607a6eSdan static int sessionChangeMerge( 437777fc1d5bSdan SessionTable *pTab, /* Table structure */ 437864277f4aSdan int bPatchset, /* True for patchsets */ 437977fc1d5bSdan SessionChange *pExist, /* Existing change */ 438077fc1d5bSdan int op2, /* Second change operation */ 438177fc1d5bSdan int bIndirect, /* True if second change is indirect */ 438277fc1d5bSdan u8 *aRec, /* Second change record */ 438377fc1d5bSdan int nRec, /* Number of bytes in aRec */ 438477fc1d5bSdan SessionChange **ppNew /* OUT: Merged change */ 43855d607a6eSdan ){ 43865d607a6eSdan SessionChange *pNew = 0; 43875d607a6eSdan 43885d607a6eSdan if( !pExist ){ 4389cbf6d2d2Sdan pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec); 43905d607a6eSdan if( !pNew ){ 43915d607a6eSdan return SQLITE_NOMEM; 43925d607a6eSdan } 43935d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 4394798693b2Sdan pNew->op = op2; 43955d607a6eSdan pNew->bIndirect = bIndirect; 43965d607a6eSdan pNew->nRecord = nRec; 4397cbf6d2d2Sdan pNew->aRecord = (u8*)&pNew[1]; 4398cbf6d2d2Sdan memcpy(pNew->aRecord, aRec, nRec); 43995d607a6eSdan }else{ 4400798693b2Sdan int op1 = pExist->op; 44015d607a6eSdan 44025d607a6eSdan /* 44035d607a6eSdan ** op1=INSERT, op2=INSERT -> Unsupported. Discard op2. 44045d607a6eSdan ** op1=INSERT, op2=UPDATE -> INSERT. 44055d607a6eSdan ** op1=INSERT, op2=DELETE -> (none) 44065d607a6eSdan ** 44075d607a6eSdan ** op1=UPDATE, op2=INSERT -> Unsupported. Discard op2. 44085d607a6eSdan ** op1=UPDATE, op2=UPDATE -> UPDATE. 44095d607a6eSdan ** op1=UPDATE, op2=DELETE -> DELETE. 44105d607a6eSdan ** 44115d607a6eSdan ** op1=DELETE, op2=INSERT -> UPDATE. 44125d607a6eSdan ** op1=DELETE, op2=UPDATE -> Unsupported. Discard op2. 44135d607a6eSdan ** op1=DELETE, op2=DELETE -> Unsupported. Discard op2. 44145d607a6eSdan */ 44155d607a6eSdan if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT) 44165d607a6eSdan || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT) 44175d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE) 44185d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_DELETE) 44195d607a6eSdan ){ 44205d607a6eSdan pNew = pExist; 44215d607a6eSdan }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){ 44225d607a6eSdan sqlite3_free(pExist); 44235d607a6eSdan assert( pNew==0 ); 44245d607a6eSdan }else{ 442564277f4aSdan u8 *aExist = pExist->aRecord; 44265d607a6eSdan int nByte; 44275d607a6eSdan u8 *aCsr; 44285d607a6eSdan 442964277f4aSdan /* Allocate a new SessionChange object. Ensure that the aRecord[] 443064277f4aSdan ** buffer of the new object is large enough to hold any record that 443164277f4aSdan ** may be generated by combining the input records. */ 44325d607a6eSdan nByte = sizeof(SessionChange) + pExist->nRecord + nRec; 44335d607a6eSdan pNew = (SessionChange *)sqlite3_malloc(nByte); 44345d607a6eSdan if( !pNew ){ 44351756ae10Sdan sqlite3_free(pExist); 44365d607a6eSdan return SQLITE_NOMEM; 44375d607a6eSdan } 44385d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 44395d607a6eSdan pNew->bIndirect = (bIndirect && pExist->bIndirect); 44405d607a6eSdan aCsr = pNew->aRecord = (u8 *)&pNew[1]; 44415d607a6eSdan 4442b08a1efaSdan if( op1==SQLITE_INSERT ){ /* INSERT + UPDATE */ 44435d607a6eSdan u8 *a1 = aRec; 4444b08a1efaSdan assert( op2==SQLITE_UPDATE ); 4445798693b2Sdan pNew->op = SQLITE_INSERT; 4446ef7a6304Sdan if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol); 444764277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1); 4448b08a1efaSdan }else if( op1==SQLITE_DELETE ){ /* DELETE + INSERT */ 4449b08a1efaSdan assert( op2==SQLITE_INSERT ); 4450798693b2Sdan pNew->op = SQLITE_UPDATE; 4451fa29ecc4Sdan if( bPatchset ){ 4452fa29ecc4Sdan memcpy(aCsr, aRec, nRec); 4453fa29ecc4Sdan aCsr += nRec; 4454fa29ecc4Sdan }else{ 445564277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){ 4456b08a1efaSdan sqlite3_free(pNew); 4457b08a1efaSdan pNew = 0; 44585d607a6eSdan } 4459fa29ecc4Sdan } 4460b08a1efaSdan }else if( op2==SQLITE_UPDATE ){ /* UPDATE + UPDATE */ 446164277f4aSdan u8 *a1 = aExist; 44625d607a6eSdan u8 *a2 = aRec; 4463cfec7eeeSdan assert( op1==SQLITE_UPDATE ); 446464277f4aSdan if( bPatchset==0 ){ 4465ef7a6304Sdan sessionSkipRecord(&a1, pTab->nCol); 4466ef7a6304Sdan sessionSkipRecord(&a2, pTab->nCol); 446764277f4aSdan } 4468798693b2Sdan pNew->op = SQLITE_UPDATE; 446964277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){ 44705d607a6eSdan sqlite3_free(pNew); 44715d607a6eSdan pNew = 0; 44725d607a6eSdan } 4473b08a1efaSdan }else{ /* UPDATE + DELETE */ 4474b08a1efaSdan assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE ); 4475798693b2Sdan pNew->op = SQLITE_DELETE; 447664277f4aSdan if( bPatchset ){ 447764277f4aSdan memcpy(aCsr, aRec, nRec); 447864277f4aSdan aCsr += nRec; 447964277f4aSdan }else{ 448064277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist); 448164277f4aSdan } 44825d607a6eSdan } 44835d607a6eSdan 44845d607a6eSdan if( pNew ){ 44854f528042Sdan pNew->nRecord = (int)(aCsr - pNew->aRecord); 44865d607a6eSdan } 44875d607a6eSdan sqlite3_free(pExist); 44885d607a6eSdan } 44895d607a6eSdan } 44905d607a6eSdan 44915d607a6eSdan *ppNew = pNew; 44925d607a6eSdan return SQLITE_OK; 44935d607a6eSdan } 44945d607a6eSdan 449577fc1d5bSdan /* 44965898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as 44975898ad69Sdan ** the first argument to the changegroup hash tables. 449877fc1d5bSdan */ 449916228167Sdan static int sessionChangesetToHash( 4500cbf6d2d2Sdan sqlite3_changeset_iter *pIter, /* Iterator to read from */ 45015898ad69Sdan sqlite3_changegroup *pGrp /* Changegroup object to add changeset to */ 45025d607a6eSdan ){ 45035d607a6eSdan u8 *aRec; 45045d607a6eSdan int nRec; 4505cbf6d2d2Sdan int rc = SQLITE_OK; 45065d607a6eSdan SessionTable *pTab = 0; 45075d607a6eSdan 45085898ad69Sdan 45095d607a6eSdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec) ){ 45105d607a6eSdan const char *zNew; 45115d607a6eSdan int nCol; 45125d607a6eSdan int op; 45135d607a6eSdan int iHash; 45145d607a6eSdan int bIndirect; 45155d607a6eSdan SessionChange *pChange; 45165d607a6eSdan SessionChange *pExist = 0; 45175d607a6eSdan SessionChange **pp; 45185d607a6eSdan 45195898ad69Sdan if( pGrp->pList==0 ){ 45205898ad69Sdan pGrp->bPatch = pIter->bPatchset; 45215898ad69Sdan }else if( pIter->bPatchset!=pGrp->bPatch ){ 45225898ad69Sdan rc = SQLITE_ERROR; 45235898ad69Sdan break; 45245898ad69Sdan } 45255898ad69Sdan 45265d607a6eSdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect); 4527ef7a6304Sdan if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){ 45285d607a6eSdan /* Search the list for a matching table */ 45294f528042Sdan int nNew = (int)strlen(zNew); 4530f29123b5Sdan u8 *abPK; 4531f29123b5Sdan 4532f29123b5Sdan sqlite3changeset_pk(pIter, &abPK, 0); 45335898ad69Sdan for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){ 45345d607a6eSdan if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break; 45355d607a6eSdan } 45365d607a6eSdan if( !pTab ){ 45376c39e6a8Sdan SessionTable **ppTab; 45386c39e6a8Sdan 4539ef7a6304Sdan pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1); 4540f29123b5Sdan if( !pTab ){ 4541f29123b5Sdan rc = SQLITE_NOMEM; 4542f29123b5Sdan break; 4543f29123b5Sdan } 45445d607a6eSdan memset(pTab, 0, sizeof(SessionTable)); 4545f29123b5Sdan pTab->nCol = nCol; 4546ef7a6304Sdan pTab->abPK = (u8*)&pTab[1]; 4547ef7a6304Sdan memcpy(pTab->abPK, abPK, nCol); 4548ef7a6304Sdan pTab->zName = (char*)&pTab->abPK[nCol]; 4549ef7a6304Sdan memcpy(pTab->zName, zNew, nNew+1); 45506c39e6a8Sdan 45516c39e6a8Sdan /* The new object must be linked on to the end of the list, not 45526c39e6a8Sdan ** simply added to the start of it. This is to ensure that the 45536c39e6a8Sdan ** tables within the output of sqlite3changegroup_output() are in 45546c39e6a8Sdan ** the right order. */ 45556c39e6a8Sdan for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext); 45566c39e6a8Sdan *ppTab = pTab; 4557f29123b5Sdan }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){ 4558f29123b5Sdan rc = SQLITE_SCHEMA; 4559f29123b5Sdan break; 45605d607a6eSdan } 45615d607a6eSdan } 45625d607a6eSdan 4563cbf6d2d2Sdan if( sessionGrowHash(pIter->bPatchset, pTab) ){ 45641756ae10Sdan rc = SQLITE_NOMEM; 45651756ae10Sdan break; 45661756ae10Sdan } 456764277f4aSdan iHash = sessionChangeHash( 4568cbf6d2d2Sdan pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange 456964277f4aSdan ); 45705d607a6eSdan 45715d607a6eSdan /* Search for existing entry. If found, remove it from the hash table. 45725d607a6eSdan ** Code below may link it back in. 45735d607a6eSdan */ 45745d607a6eSdan for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){ 457564277f4aSdan int bPkOnly1 = 0; 457664277f4aSdan int bPkOnly2 = 0; 4577cbf6d2d2Sdan if( pIter->bPatchset ){ 457864277f4aSdan bPkOnly1 = (*pp)->op==SQLITE_DELETE; 457964277f4aSdan bPkOnly2 = op==SQLITE_DELETE; 458064277f4aSdan } 458164277f4aSdan if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){ 45825d607a6eSdan pExist = *pp; 45835d607a6eSdan *pp = (*pp)->pNext; 45845d607a6eSdan pTab->nEntry--; 45855d607a6eSdan break; 45865d607a6eSdan } 45875d607a6eSdan } 45885d607a6eSdan 458964277f4aSdan rc = sessionChangeMerge(pTab, 4590cbf6d2d2Sdan pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange 459164277f4aSdan ); 45925d607a6eSdan if( rc ) break; 45935d607a6eSdan if( pChange ){ 45945d607a6eSdan pChange->pNext = pTab->apChange[iHash]; 45955d607a6eSdan pTab->apChange[iHash] = pChange; 45965d607a6eSdan pTab->nEntry++; 45975d607a6eSdan } 45985d607a6eSdan } 45995d607a6eSdan 4600cbf6d2d2Sdan if( rc==SQLITE_OK ) rc = pIter->rc; 46015d607a6eSdan return rc; 46025d607a6eSdan } 46035d607a6eSdan 46045d607a6eSdan /* 46055898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets) 46065898ad69Sdan ** added to the changegroup object passed as the first argument. 46075d607a6eSdan ** 46085898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the 46095898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming 46105898ad69Sdan ** interfaces. 46115d607a6eSdan ** 46125898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a 46135898ad69Sdan ** buffer containing the output changeset before this function returns. In 46145898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It 46155898ad69Sdan ** is the responsibility of the caller to free the output buffer using 46165898ad69Sdan ** sqlite3_free() when it is no longer required. 46175898ad69Sdan ** 46185898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite 46195898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut) 46205898ad69Sdan ** are both set to 0 before returning. 46215d607a6eSdan */ 46225898ad69Sdan static int sessionChangegroupOutput( 46235898ad69Sdan sqlite3_changegroup *pGrp, 4624cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4625cbf6d2d2Sdan void *pOut, 4626cbf6d2d2Sdan int *pnOut, 4627cbf6d2d2Sdan void **ppOut 46285d607a6eSdan ){ 46295898ad69Sdan int rc = SQLITE_OK; 4630e8fa8c96Sdan SessionBuffer buf = {0, 0, 0}; 46315898ad69Sdan SessionTable *pTab; 4632cbf6d2d2Sdan assert( xOutput==0 || (ppOut==0 && pnOut==0) ); 46335d607a6eSdan 46345d607a6eSdan /* Create the serialized output changeset based on the contents of the 46355898ad69Sdan ** hash tables attached to the SessionTable objects in list p->pList. 46365d607a6eSdan */ 46375898ad69Sdan for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 46385d607a6eSdan int i; 46395d607a6eSdan if( pTab->nEntry==0 ) continue; 46405d607a6eSdan 46415898ad69Sdan sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc); 46425d607a6eSdan for(i=0; i<pTab->nChange; i++){ 46435d607a6eSdan SessionChange *p; 46445d607a6eSdan for(p=pTab->apChange[i]; p; p=p->pNext){ 4645798693b2Sdan sessionAppendByte(&buf, p->op, &rc); 46465d607a6eSdan sessionAppendByte(&buf, p->bIndirect, &rc); 46475d607a6eSdan sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc); 46485d607a6eSdan } 46495d607a6eSdan } 4650cbf6d2d2Sdan 4651f1a08ad8Sdrh if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 4652cbf6d2d2Sdan rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4653cbf6d2d2Sdan buf.nBuf = 0; 4654cbf6d2d2Sdan } 46555d607a6eSdan } 46565d607a6eSdan 46575d607a6eSdan if( rc==SQLITE_OK ){ 4658cbf6d2d2Sdan if( xOutput ){ 4659cbf6d2d2Sdan if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4660cbf6d2d2Sdan }else{ 46615d607a6eSdan *ppOut = buf.aBuf; 46625d607a6eSdan *pnOut = buf.nBuf; 4663cbf6d2d2Sdan buf.aBuf = 0; 46645d607a6eSdan } 46655d607a6eSdan } 4666cbf6d2d2Sdan sqlite3_free(buf.aBuf); 46675d607a6eSdan 46685d607a6eSdan return rc; 46695d607a6eSdan } 46705d607a6eSdan 4671cbf6d2d2Sdan /* 46725898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup. 46735898ad69Sdan */ 46745898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){ 46755898ad69Sdan int rc = SQLITE_OK; /* Return code */ 46765898ad69Sdan sqlite3_changegroup *p; /* New object */ 46775898ad69Sdan p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup)); 46785898ad69Sdan if( p==0 ){ 46795898ad69Sdan rc = SQLITE_NOMEM; 46805898ad69Sdan }else{ 46815898ad69Sdan memset(p, 0, sizeof(sqlite3_changegroup)); 46825898ad69Sdan } 46835898ad69Sdan *pp = p; 46845898ad69Sdan return rc; 46855898ad69Sdan } 46865898ad69Sdan 46875898ad69Sdan /* 46885898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes, 46895898ad69Sdan ** to changeset-group p. 46905898ad69Sdan */ 46915898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){ 46925898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 46935898ad69Sdan int rc; /* Return code */ 46945898ad69Sdan 46955898ad69Sdan rc = sqlite3changeset_start(&pIter, nData, pData); 46965898ad69Sdan if( rc==SQLITE_OK ){ 46975898ad69Sdan rc = sessionChangesetToHash(pIter, pGrp); 46985898ad69Sdan } 46995898ad69Sdan sqlite3changeset_finalize(pIter); 47005898ad69Sdan return rc; 47015898ad69Sdan } 47025898ad69Sdan 47035898ad69Sdan /* 47045898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation 47055898ad69Sdan ** of all changesets added to the group so far. 47065898ad69Sdan */ 47075898ad69Sdan int sqlite3changegroup_output( 47085898ad69Sdan sqlite3_changegroup *pGrp, 47095898ad69Sdan int *pnData, 47105898ad69Sdan void **ppData 47115898ad69Sdan ){ 47125898ad69Sdan return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData); 47135898ad69Sdan } 47145898ad69Sdan 47155898ad69Sdan /* 47165898ad69Sdan ** Streaming versions of changegroup_add(). 47175898ad69Sdan */ 47185898ad69Sdan int sqlite3changegroup_add_strm( 47195898ad69Sdan sqlite3_changegroup *pGrp, 47205898ad69Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 47215898ad69Sdan void *pIn 47225898ad69Sdan ){ 47235898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 47245898ad69Sdan int rc; /* Return code */ 47255898ad69Sdan 47265898ad69Sdan rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 47275898ad69Sdan if( rc==SQLITE_OK ){ 47285898ad69Sdan rc = sessionChangesetToHash(pIter, pGrp); 47295898ad69Sdan } 47305898ad69Sdan sqlite3changeset_finalize(pIter); 47315898ad69Sdan return rc; 47325898ad69Sdan } 47335898ad69Sdan 47345898ad69Sdan /* 47355898ad69Sdan ** Streaming versions of changegroup_output(). 47365898ad69Sdan */ 47375898ad69Sdan int sqlite3changegroup_output_strm( 47385898ad69Sdan sqlite3_changegroup *pGrp, 47395898ad69Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 47405898ad69Sdan void *pOut 47415898ad69Sdan ){ 47425898ad69Sdan return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0); 47435898ad69Sdan } 47445898ad69Sdan 47455898ad69Sdan /* 47465898ad69Sdan ** Delete a changegroup object. 47475898ad69Sdan */ 47485898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){ 47495898ad69Sdan if( pGrp ){ 47505898ad69Sdan sessionDeleteTable(pGrp->pList); 47515898ad69Sdan sqlite3_free(pGrp); 47525898ad69Sdan } 47535898ad69Sdan } 47545898ad69Sdan 47555898ad69Sdan /* 4756cbf6d2d2Sdan ** Combine two changesets together. 4757cbf6d2d2Sdan */ 4758cbf6d2d2Sdan int sqlite3changeset_concat( 4759cbf6d2d2Sdan int nLeft, /* Number of bytes in lhs input */ 4760cbf6d2d2Sdan void *pLeft, /* Lhs input changeset */ 4761cbf6d2d2Sdan int nRight /* Number of bytes in rhs input */, 4762cbf6d2d2Sdan void *pRight, /* Rhs input changeset */ 4763cbf6d2d2Sdan int *pnOut, /* OUT: Number of bytes in output changeset */ 4764cbf6d2d2Sdan void **ppOut /* OUT: changeset (left <concat> right) */ 4765cbf6d2d2Sdan ){ 47665898ad69Sdan sqlite3_changegroup *pGrp; 4767cbf6d2d2Sdan int rc; 4768cbf6d2d2Sdan 47695898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 4770cbf6d2d2Sdan if( rc==SQLITE_OK ){ 47715898ad69Sdan rc = sqlite3changegroup_add(pGrp, nLeft, pLeft); 4772cbf6d2d2Sdan } 4773cbf6d2d2Sdan if( rc==SQLITE_OK ){ 47745898ad69Sdan rc = sqlite3changegroup_add(pGrp, nRight, pRight); 4775cbf6d2d2Sdan } 47765898ad69Sdan if( rc==SQLITE_OK ){ 47775898ad69Sdan rc = sqlite3changegroup_output(pGrp, pnOut, ppOut); 47785898ad69Sdan } 47795898ad69Sdan sqlite3changegroup_delete(pGrp); 4780cbf6d2d2Sdan 4781cbf6d2d2Sdan return rc; 4782cbf6d2d2Sdan } 4783cbf6d2d2Sdan 4784cbf6d2d2Sdan /* 4785cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat(). 4786cbf6d2d2Sdan */ 4787f1a08ad8Sdrh int sqlite3changeset_concat_strm( 4788cbf6d2d2Sdan int (*xInputA)(void *pIn, void *pData, int *pnData), 4789cbf6d2d2Sdan void *pInA, 4790cbf6d2d2Sdan int (*xInputB)(void *pIn, void *pData, int *pnData), 4791cbf6d2d2Sdan void *pInB, 4792cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4793cbf6d2d2Sdan void *pOut 4794cbf6d2d2Sdan ){ 47955898ad69Sdan sqlite3_changegroup *pGrp; 4796cbf6d2d2Sdan int rc; 4797cbf6d2d2Sdan 47985898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 4799cbf6d2d2Sdan if( rc==SQLITE_OK ){ 48005898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA); 4801cbf6d2d2Sdan } 4802cbf6d2d2Sdan if( rc==SQLITE_OK ){ 48035898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB); 4804cbf6d2d2Sdan } 48055898ad69Sdan if( rc==SQLITE_OK ){ 48065898ad69Sdan rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut); 48075898ad69Sdan } 48085898ad69Sdan sqlite3changegroup_delete(pGrp); 4809cbf6d2d2Sdan 4810cbf6d2d2Sdan return rc; 4811cbf6d2d2Sdan } 4812cbf6d2d2Sdan 48139b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */ 4814