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 235f01d3a7eSdan ** with PRIMARY KEY columns are omitted. The PRIMARY KEY fields contain the 236f01d3a7eSdan ** 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; 51924a0c453Sdan if( e==0 || e==0xFF ) 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 599bd45374cSdan if( 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 } 842*3f2cebb6Sdan if( n>0 && 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; 1118fa3d7855Smistachkin 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 ); 2186bd45374cSdan if( zSql==0 ) rc = SQLITE_NOMEM; 21873739f298Sdan }else{ 2188d5f0767cSdan int i; 2189e8d5648eSdan const char *zSep = ""; 2190e8d5648eSdan SessionBuffer buf = {0, 0, 0}; 2191d5f0767cSdan 2192e8d5648eSdan sessionAppendStr(&buf, "SELECT * FROM ", &rc); 2193d7fb7d24Sdan sessionAppendIdent(&buf, zDb, &rc); 2194d7fb7d24Sdan sessionAppendStr(&buf, ".", &rc); 2195e8d5648eSdan sessionAppendIdent(&buf, zTab, &rc); 2196e8d5648eSdan sessionAppendStr(&buf, " WHERE ", &rc); 2197e8d5648eSdan for(i=0; i<nCol; i++){ 2198e8d5648eSdan if( abPK[i] ){ 2199e8d5648eSdan sessionAppendStr(&buf, zSep, &rc); 2200e8d5648eSdan sessionAppendIdent(&buf, azCol[i], &rc); 22013739f298Sdan sessionAppendStr(&buf, " IS ?", &rc); 2202e8d5648eSdan sessionAppendInteger(&buf, i+1, &rc); 2203e8d5648eSdan zSep = " AND "; 2204d5f0767cSdan } 2205d5f0767cSdan } 22063739f298Sdan zSql = (char*)buf.aBuf; 22073739f298Sdan nSql = buf.nBuf; 2208d5f0767cSdan } 22093739f298Sdan 22103739f298Sdan if( rc==SQLITE_OK ){ 22113739f298Sdan rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0); 22123739f298Sdan } 22133739f298Sdan sqlite3_free(zSql); 2214e8d5648eSdan return rc; 2215d5f0767cSdan } 2216d5f0767cSdan 221777fc1d5bSdan /* 221877fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange 221977fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement 222077fc1d5bSdan ** is as prepared by function sessionSelectStmt(). 222177fc1d5bSdan ** 222277fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite 222377fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise. 222477fc1d5bSdan */ 2225e8d5648eSdan static int sessionSelectBind( 222677fc1d5bSdan sqlite3_stmt *pSelect, /* SELECT from sessionSelectStmt() */ 222777fc1d5bSdan int nCol, /* Number of columns in table */ 222877fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 222977fc1d5bSdan SessionChange *pChange /* Change structure */ 2230e8d5648eSdan ){ 2231e8d5648eSdan int i; 2232e8d5648eSdan int rc = SQLITE_OK; 2233e5754eecSdan u8 *a = pChange->aRecord; 2234d5f0767cSdan 2235e8d5648eSdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2236e8d5648eSdan int eType = *a++; 2237e8d5648eSdan 2238e8d5648eSdan switch( eType ){ 223980fe2d93Sdan case 0: 2240e8d5648eSdan case SQLITE_NULL: 22411611e5a3Sdan assert( abPK[i]==0 ); 2242e8d5648eSdan break; 2243e8d5648eSdan 2244e8d5648eSdan case SQLITE_INTEGER: { 2245e8d5648eSdan if( abPK[i] ){ 2246e8d5648eSdan i64 iVal = sessionGetI64(a); 2247e8d5648eSdan rc = sqlite3_bind_int64(pSelect, i+1, iVal); 2248e8d5648eSdan } 2249e8d5648eSdan a += 8; 2250e8d5648eSdan break; 2251d5f0767cSdan } 2252296c7658Sdan 2253e8d5648eSdan case SQLITE_FLOAT: { 2254e8d5648eSdan if( abPK[i] ){ 2255e8d5648eSdan double rVal; 2256e8d5648eSdan i64 iVal = sessionGetI64(a); 2257e8d5648eSdan memcpy(&rVal, &iVal, 8); 22584e895da1Sdan rc = sqlite3_bind_double(pSelect, i+1, rVal); 2259d5f0767cSdan } 2260e8d5648eSdan a += 8; 2261e8d5648eSdan break; 2262e8d5648eSdan } 2263e8d5648eSdan 2264e8d5648eSdan case SQLITE_TEXT: { 2265e8d5648eSdan int n; 2266e8d5648eSdan a += sessionVarintGet(a, &n); 2267e8d5648eSdan if( abPK[i] ){ 2268e8d5648eSdan rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT); 2269e8d5648eSdan } 2270e8d5648eSdan a += n; 2271e8d5648eSdan break; 2272e8d5648eSdan } 2273e8d5648eSdan 2274e5754eecSdan default: { 2275e8d5648eSdan int n; 2276e5754eecSdan assert( eType==SQLITE_BLOB ); 2277e8d5648eSdan a += sessionVarintGet(a, &n); 2278e8d5648eSdan if( abPK[i] ){ 2279e8d5648eSdan rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT); 2280e8d5648eSdan } 2281e8d5648eSdan a += n; 2282e8d5648eSdan break; 2283e8d5648eSdan } 2284e8d5648eSdan } 2285e8d5648eSdan } 2286e8d5648eSdan 2287d5f0767cSdan return rc; 22884fccf43aSdan } 22894fccf43aSdan 229077fc1d5bSdan /* 229177fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it 229277fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary 229377fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an 229477fc1d5bSdan ** SQLite error code before returning. 229577fc1d5bSdan */ 22965d607a6eSdan static void sessionAppendTableHdr( 2297a71d2371Sdan SessionBuffer *pBuf, /* Append header to this buffer */ 2298a71d2371Sdan int bPatchset, /* Use the patchset format if true */ 2299a71d2371Sdan SessionTable *pTab, /* Table object to append header for */ 2300a71d2371Sdan int *pRc /* IN/OUT: Error code */ 23015d607a6eSdan ){ 23025d607a6eSdan /* Write a table header */ 230373b3c055Sdan sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc); 23045d607a6eSdan sessionAppendVarint(pBuf, pTab->nCol, pRc); 23055d607a6eSdan sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc); 23064f528042Sdan sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc); 23075d607a6eSdan } 23085d607a6eSdan 2309a71d2371Sdan /* 2310a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset 2311a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object 2312a71d2371Sdan ** passed as the first argument. 2313a71d2371Sdan ** 2314a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset 2315a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error 2316a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set 2317a71d2371Sdan ** to 0. 2318a71d2371Sdan */ 2319adf3bf58Sdrh static int sessionGenerateChangeset( 23204fccf43aSdan sqlite3_session *pSession, /* Session object */ 232173b3c055Sdan int bPatchset, /* True for patchset, false for changeset */ 2322ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2323ef7a6304Sdan void *pOut, /* First argument for xOutput */ 23244fccf43aSdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 23254fccf43aSdan void **ppChangeset /* OUT: Buffer containing changeset */ 23264fccf43aSdan ){ 2327296c7658Sdan sqlite3 *db = pSession->db; /* Source database handle */ 2328296c7658Sdan SessionTable *pTab; /* Used to iterate through attached tables */ 2329296c7658Sdan SessionBuffer buf = {0,0,0}; /* Buffer in which to accumlate changeset */ 2330296c7658Sdan int rc; /* Return code */ 23314fccf43aSdan 2332ef7a6304Sdan assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) ); 2333ef7a6304Sdan 2334296c7658Sdan /* Zero the output variables in case an error occurs. If this session 2335296c7658Sdan ** object is already in the error state (sqlite3_session.rc != SQLITE_OK), 2336296c7658Sdan ** this call will be a no-op. */ 2337ef7a6304Sdan if( xOutput==0 ){ 23384fccf43aSdan *pnChangeset = 0; 23394fccf43aSdan *ppChangeset = 0; 2340ef7a6304Sdan } 2341e5754eecSdan 2342e5754eecSdan if( pSession->rc ) return pSession->rc; 2343e5754eecSdan rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0); 2344e5754eecSdan if( rc!=SQLITE_OK ) return rc; 2345e5754eecSdan 2346e5754eecSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 23474fccf43aSdan 23484fccf43aSdan for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 23494fccf43aSdan if( pTab->nEntry ){ 2350d7fb7d24Sdan const char *zName = pTab->zName; 2351a9605b91Sdan int nCol; /* Number of columns in table */ 2352a9605b91Sdan u8 *abPK; /* Primary key array */ 2353a9605b91Sdan const char **azCol = 0; /* Table columns */ 23541f34f8ccSdan int i; /* Used to iterate through hash buckets */ 23551f34f8ccSdan sqlite3_stmt *pSel = 0; /* SELECT statement to query table pTab */ 23561f34f8ccSdan int nRewind = buf.nBuf; /* Initial size of write buffer */ 23571f34f8ccSdan int nNoop; /* Size of buffer after writing tbl header */ 23584fccf43aSdan 2359a9605b91Sdan /* Check the table schema is still Ok. */ 2360a9605b91Sdan rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK); 2361a9605b91Sdan if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){ 2362a9605b91Sdan rc = SQLITE_SCHEMA; 2363a9605b91Sdan } 2364a9605b91Sdan 23654fccf43aSdan /* Write a table header */ 236673b3c055Sdan sessionAppendTableHdr(&buf, bPatchset, pTab, &rc); 23674fccf43aSdan 23684fccf43aSdan /* Build and compile a statement to execute: */ 23694fccf43aSdan if( rc==SQLITE_OK ){ 2370d7fb7d24Sdan rc = sessionSelectStmt( 2371a9605b91Sdan db, pSession->zDb, zName, nCol, azCol, abPK, &pSel); 23724fccf43aSdan } 23734fccf43aSdan 23741f34f8ccSdan nNoop = buf.nBuf; 237512ca0b56Sdan for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){ 2376e8d5648eSdan SessionChange *p; /* Used to iterate through changes */ 2377e8d5648eSdan 23784fccf43aSdan for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){ 2379e5754eecSdan rc = sessionSelectBind(pSel, nCol, abPK, p); 238080fe2d93Sdan if( rc!=SQLITE_OK ) continue; 23811f34f8ccSdan if( sqlite3_step(pSel)==SQLITE_ROW ){ 2382798693b2Sdan if( p->op==SQLITE_INSERT ){ 23834fccf43aSdan int iCol; 23844fccf43aSdan sessionAppendByte(&buf, SQLITE_INSERT, &rc); 2385b4480e94Sdan sessionAppendByte(&buf, p->bIndirect, &rc); 2386e8d5648eSdan for(iCol=0; iCol<nCol; iCol++){ 23871f34f8ccSdan sessionAppendCol(&buf, pSel, iCol, &rc); 23884fccf43aSdan } 2389e8d5648eSdan }else{ 239073b3c055Sdan rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK); 23914fccf43aSdan } 2392798693b2Sdan }else if( p->op!=SQLITE_INSERT ){ 2393a71d2371Sdan rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK); 23944fccf43aSdan } 239512ca0b56Sdan if( rc==SQLITE_OK ){ 23961f34f8ccSdan rc = sqlite3_reset(pSel); 23974fccf43aSdan } 2398ef7a6304Sdan 2399f1a08ad8Sdrh /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass 2400ef7a6304Sdan ** its contents to the xOutput() callback. */ 2401ef7a6304Sdan if( xOutput 2402ef7a6304Sdan && rc==SQLITE_OK 2403ef7a6304Sdan && buf.nBuf>nNoop 2404f1a08ad8Sdrh && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE 2405ef7a6304Sdan ){ 2406ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2407ef7a6304Sdan nNoop = -1; 2408ef7a6304Sdan buf.nBuf = 0; 2409ef7a6304Sdan } 2410ef7a6304Sdan 24114fccf43aSdan } 2412e8d5648eSdan } 24134fccf43aSdan 24141f34f8ccSdan sqlite3_finalize(pSel); 24151f34f8ccSdan if( buf.nBuf==nNoop ){ 24164fccf43aSdan buf.nBuf = nRewind; 24174fccf43aSdan } 2418cfdbde21Sdrh sqlite3_free((char*)azCol); /* cast works around VC++ bug */ 24194fccf43aSdan } 24204fccf43aSdan } 24214fccf43aSdan 24224fccf43aSdan if( rc==SQLITE_OK ){ 2423ef7a6304Sdan if( xOutput==0 ){ 24244fccf43aSdan *pnChangeset = buf.nBuf; 24254fccf43aSdan *ppChangeset = buf.aBuf; 2426ef7a6304Sdan buf.aBuf = 0; 2427ef7a6304Sdan }else if( buf.nBuf>0 ){ 2428ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2429ef7a6304Sdan } 24304fccf43aSdan } 24314c220252Sdan 2432ef7a6304Sdan sqlite3_free(buf.aBuf); 2433e5754eecSdan sqlite3_exec(db, "RELEASE changeset", 0, 0, 0); 24344c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 24354fccf43aSdan return rc; 24364fccf43aSdan } 24374fccf43aSdan 2438296c7658Sdan /* 243973b3c055Sdan ** Obtain a changeset object containing all changes recorded by the 244073b3c055Sdan ** session object passed as the first argument. 244173b3c055Sdan ** 244273b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 244373b3c055Sdan ** using sqlite3_free(). 244473b3c055Sdan */ 244573b3c055Sdan int sqlite3session_changeset( 244673b3c055Sdan sqlite3_session *pSession, /* Session object */ 244773b3c055Sdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 244873b3c055Sdan void **ppChangeset /* OUT: Buffer containing changeset */ 244973b3c055Sdan ){ 2450ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset); 2451ef7a6304Sdan } 2452ef7a6304Sdan 2453ef7a6304Sdan /* 2454ef7a6304Sdan ** Streaming version of sqlite3session_changeset(). 2455ef7a6304Sdan */ 2456f1a08ad8Sdrh int sqlite3session_changeset_strm( 2457ef7a6304Sdan sqlite3_session *pSession, 2458ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2459ef7a6304Sdan void *pOut 2460ef7a6304Sdan ){ 2461ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0); 2462ef7a6304Sdan } 2463ef7a6304Sdan 2464ef7a6304Sdan /* 2465ef7a6304Sdan ** Streaming version of sqlite3session_patchset(). 2466ef7a6304Sdan */ 2467f1a08ad8Sdrh int sqlite3session_patchset_strm( 2468ef7a6304Sdan sqlite3_session *pSession, 2469ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2470ef7a6304Sdan void *pOut 2471ef7a6304Sdan ){ 2472ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0); 247373b3c055Sdan } 247473b3c055Sdan 247573b3c055Sdan /* 247673b3c055Sdan ** Obtain a patchset object containing all changes recorded by the 247773b3c055Sdan ** session object passed as the first argument. 247873b3c055Sdan ** 247973b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 248073b3c055Sdan ** using sqlite3_free(). 248173b3c055Sdan */ 248273b3c055Sdan int sqlite3session_patchset( 248373b3c055Sdan sqlite3_session *pSession, /* Session object */ 248473b3c055Sdan int *pnPatchset, /* OUT: Size of buffer at *ppChangeset */ 248573b3c055Sdan void **ppPatchset /* OUT: Buffer containing changeset */ 248673b3c055Sdan ){ 2487ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset); 248873b3c055Sdan } 248973b3c055Sdan 249073b3c055Sdan /* 2491296c7658Sdan ** Enable or disable the session object passed as the first argument. 2492296c7658Sdan */ 24934fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){ 24944c220252Sdan int ret; 24954c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2496296c7658Sdan if( bEnable>=0 ){ 2497296c7658Sdan pSession->bEnable = bEnable; 24984fccf43aSdan } 24994c220252Sdan ret = pSession->bEnable; 25004c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 25014c220252Sdan return ret; 2502296c7658Sdan } 25034fccf43aSdan 25044fccf43aSdan /* 2505b4480e94Sdan ** Enable or disable the session object passed as the first argument. 2506b4480e94Sdan */ 2507b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){ 2508b4480e94Sdan int ret; 2509b4480e94Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2510b4480e94Sdan if( bIndirect>=0 ){ 2511b4480e94Sdan pSession->bIndirect = bIndirect; 2512b4480e94Sdan } 2513b4480e94Sdan ret = pSession->bIndirect; 2514b4480e94Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2515b4480e94Sdan return ret; 2516b4480e94Sdan } 2517b4480e94Sdan 2518b4480e94Sdan /* 2519b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded 2520b69ec348Sdan ** by the session object passed as the only argument. 2521b69ec348Sdan */ 2522b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){ 2523b69ec348Sdan int ret = 0; 2524b69ec348Sdan SessionTable *pTab; 2525b69ec348Sdan 2526b69ec348Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2527b69ec348Sdan for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){ 2528b69ec348Sdan ret = (pTab->nEntry>0); 2529b69ec348Sdan } 2530b69ec348Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2531b69ec348Sdan 2532ff530326Sdan return (ret==0); 2533b69ec348Sdan } 2534b69ec348Sdan 2535b69ec348Sdan /* 2536f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm(). 25374fccf43aSdan */ 2538adf3bf58Sdrh static int sessionChangesetStart( 2539296c7658Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2540ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2541ef7a6304Sdan void *pIn, 2542296c7658Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2543296c7658Sdan void *pChangeset /* Pointer to buffer containing changeset */ 25444fccf43aSdan ){ 25454fccf43aSdan sqlite3_changeset_iter *pRet; /* Iterator to return */ 25464fccf43aSdan int nByte; /* Number of bytes to allocate for iterator */ 25474fccf43aSdan 2548ef7a6304Sdan assert( xInput==0 || (pChangeset==0 && nChangeset==0) ); 2549ef7a6304Sdan 2550296c7658Sdan /* Zero the output variable in case an error occurs. */ 2551296c7658Sdan *pp = 0; 25524fccf43aSdan 2553296c7658Sdan /* Allocate and initialize the iterator structure. */ 25544fccf43aSdan nByte = sizeof(sqlite3_changeset_iter); 25554fccf43aSdan pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte); 25564fccf43aSdan if( !pRet ) return SQLITE_NOMEM; 25574fccf43aSdan memset(pRet, 0, sizeof(sqlite3_changeset_iter)); 25584757c658Sdan pRet->in.aData = (u8 *)pChangeset; 25594757c658Sdan pRet->in.nData = nChangeset; 2560ef7a6304Sdan pRet->in.xInput = xInput; 2561ef7a6304Sdan pRet->in.pIn = pIn; 2562ef7a6304Sdan pRet->in.bEof = (xInput ? 0 : 1); 25634fccf43aSdan 2564296c7658Sdan /* Populate the output variable and return success. */ 2565296c7658Sdan *pp = pRet; 25664fccf43aSdan return SQLITE_OK; 25674fccf43aSdan } 25684fccf43aSdan 2569296c7658Sdan /* 2570ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset. 2571ef7a6304Sdan */ 2572ef7a6304Sdan int sqlite3changeset_start( 2573ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2574ef7a6304Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2575ef7a6304Sdan void *pChangeset /* Pointer to buffer containing changeset */ 2576ef7a6304Sdan ){ 2577ef7a6304Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset); 2578ef7a6304Sdan } 2579ef7a6304Sdan 2580ef7a6304Sdan /* 2581ef7a6304Sdan ** Streaming version of sqlite3changeset_start(). 2582ef7a6304Sdan */ 2583f1a08ad8Sdrh int sqlite3changeset_start_strm( 2584ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2585ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2586ef7a6304Sdan void *pIn 2587ef7a6304Sdan ){ 2588ef7a6304Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0); 2589ef7a6304Sdan } 2590ef7a6304Sdan 2591ef7a6304Sdan /* 2592d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming 2593d9151526Sdan ** object and the buffer is full, discard some data to free up space. 2594d9151526Sdan */ 2595d9151526Sdan static void sessionDiscardData(SessionInput *pIn){ 2596d9151526Sdan if( pIn->bEof && pIn->xInput && pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){ 2597d9151526Sdan int nMove = pIn->buf.nBuf - pIn->iNext; 2598d9151526Sdan assert( nMove>=0 ); 2599d9151526Sdan if( nMove>0 ){ 2600d9151526Sdan memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove); 2601d9151526Sdan } 2602d9151526Sdan pIn->buf.nBuf -= pIn->iNext; 2603d9151526Sdan pIn->iNext = 0; 2604d9151526Sdan pIn->nData = pIn->buf.nBuf; 2605d9151526Sdan } 2606d9151526Sdan } 2607d9151526Sdan 2608d9151526Sdan /* 2609ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or, 2610ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available 2611ef7a6304Sdan ** data is in the buffer. 2612ef7a6304Sdan ** 2613ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise. 2614ef7a6304Sdan */ 26154757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){ 2616ef7a6304Sdan int rc = SQLITE_OK; 26174757c658Sdan if( pIn->xInput ){ 26184757c658Sdan while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){ 2619f1a08ad8Sdrh int nNew = SESSIONS_STRM_CHUNK_SIZE; 26204757c658Sdan 2621d9151526Sdan if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn); 26224757c658Sdan if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){ 26234757c658Sdan rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew); 26244757c658Sdan if( nNew==0 ){ 26254757c658Sdan pIn->bEof = 1; 26264757c658Sdan }else{ 26274757c658Sdan pIn->buf.nBuf += nNew; 26284757c658Sdan } 26294757c658Sdan } 26304757c658Sdan 26314757c658Sdan pIn->aData = pIn->buf.aBuf; 26324757c658Sdan pIn->nData = pIn->buf.nBuf; 26334757c658Sdan } 2634ef7a6304Sdan } 2635ef7a6304Sdan return rc; 2636ef7a6304Sdan } 2637ef7a6304Sdan 2638ef7a6304Sdan /* 2639ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record 2640ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec 2641ef7a6304Sdan ** until it points to the byte immediately following that record. 2642ef7a6304Sdan */ 2643ef7a6304Sdan static void sessionSkipRecord( 2644ef7a6304Sdan u8 **ppRec, /* IN/OUT: Record pointer */ 2645ef7a6304Sdan int nCol /* Number of values in record */ 2646ef7a6304Sdan ){ 2647ef7a6304Sdan u8 *aRec = *ppRec; 2648ef7a6304Sdan int i; 2649ef7a6304Sdan for(i=0; i<nCol; i++){ 2650ef7a6304Sdan int eType = *aRec++; 2651ef7a6304Sdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2652ef7a6304Sdan int nByte; 2653ef7a6304Sdan aRec += sessionVarintGet((u8*)aRec, &nByte); 2654ef7a6304Sdan aRec += nByte; 2655ef7a6304Sdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2656ef7a6304Sdan aRec += 8; 2657ef7a6304Sdan } 2658ef7a6304Sdan } 2659ef7a6304Sdan 2660ef7a6304Sdan *ppRec = aRec; 2661ef7a6304Sdan } 2662ef7a6304Sdan 2663ef7a6304Sdan /* 26644757c658Sdan ** This function sets the value of the sqlite3_value object passed as the 26654757c658Sdan ** first argument to a copy of the string or blob held in the aData[] 26664757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM 26674757c658Sdan ** error occurs. 26684757c658Sdan */ 26694757c658Sdan static int sessionValueSetStr( 26704757c658Sdan sqlite3_value *pVal, /* Set the value of this object */ 26714757c658Sdan u8 *aData, /* Buffer containing string or blob data */ 26724757c658Sdan int nData, /* Size of buffer aData[] in bytes */ 26734757c658Sdan u8 enc /* String encoding (0 for blobs) */ 26744757c658Sdan ){ 267516228167Sdan /* In theory this code could just pass SQLITE_TRANSIENT as the final 267616228167Sdan ** argument to sqlite3ValueSetStr() and have the copy created 267716228167Sdan ** automatically. But doing so makes it difficult to detect any OOM 267816228167Sdan ** error. Hence the code to create the copy externally. */ 26793cc89d95Sdan u8 *aCopy = sqlite3_malloc(nData+1); 26804757c658Sdan if( aCopy==0 ) return SQLITE_NOMEM; 26814757c658Sdan memcpy(aCopy, aData, nData); 26824757c658Sdan sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free); 26834757c658Sdan return SQLITE_OK; 26844757c658Sdan } 26854757c658Sdan 26864757c658Sdan /* 2687296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT" 2688296c7658Sdan ** for details. 2689296c7658Sdan ** 2690296c7658Sdan ** When this function is called, *paChange points to the start of the record 2691296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to 2692296c7658Sdan ** one byte after the end of the same record before this function returns. 2693a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or, 2694a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields 2695a71d2371Sdan ** (in other words, it is a patchset DELETE record). 2696296c7658Sdan ** 2697296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller) 2698296c7658Sdan ** is set to point to an sqlite3_value object containing the value read 2699296c7658Sdan ** from the corresponding position in the record. If that value is not 2700296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change 2701296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is 2702296c7658Sdan ** set to NULL. 2703296c7658Sdan ** 2704296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures 2705296c7658Sdan ** using sqlite3_free(). 2706296c7658Sdan ** 2707296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned. 2708296c7658Sdan ** The apOut[] array may have been partially populated in this case. 2709296c7658Sdan */ 27104fccf43aSdan static int sessionReadRecord( 2711ef7a6304Sdan SessionInput *pIn, /* Input data */ 27124fccf43aSdan int nCol, /* Number of values in record */ 271373b3c055Sdan u8 *abPK, /* Array of primary key flags, or NULL */ 27144fccf43aSdan sqlite3_value **apOut /* Write values to this array */ 27154fccf43aSdan ){ 2716296c7658Sdan int i; /* Used to iterate through columns */ 2717ef7a6304Sdan int rc = SQLITE_OK; 27184fccf43aSdan 2719ef7a6304Sdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2720ef7a6304Sdan int eType = 0; /* Type of value (SQLITE_NULL, TEXT etc.) */ 272173b3c055Sdan if( abPK && abPK[i]==0 ) continue; 2722ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2723ef7a6304Sdan if( rc==SQLITE_OK ){ 2724e341ec69Sdan if( pIn->iNext>=pIn->nData ){ 2725e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2726e341ec69Sdan }else{ 27274757c658Sdan eType = pIn->aData[pIn->iNext++]; 2728e8fa8c96Sdan assert( apOut[i]==0 ); 27294fccf43aSdan if( eType ){ 27304fccf43aSdan apOut[i] = sqlite3ValueNew(0); 2731ef7a6304Sdan if( !apOut[i] ) rc = SQLITE_NOMEM; 2732ef7a6304Sdan } 2733dd8a4af8Sdan } 2734e341ec69Sdan } 27354fccf43aSdan 2736ef7a6304Sdan if( rc==SQLITE_OK ){ 27374757c658Sdan u8 *aVal = &pIn->aData[pIn->iNext]; 27384fccf43aSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 27394fccf43aSdan int nByte; 2740ef7a6304Sdan pIn->iNext += sessionVarintGet(aVal, &nByte); 2741ef7a6304Sdan rc = sessionInputBuffer(pIn, nByte); 2742e8fa8c96Sdan if( rc==SQLITE_OK ){ 2743e341ec69Sdan if( nByte<0 || nByte>pIn->nData-pIn->iNext ){ 2744e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2745e341ec69Sdan }else{ 27466734007dSdan u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0); 27474757c658Sdan rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc); 2748ef7a6304Sdan pIn->iNext += nByte; 27494fccf43aSdan } 2750e341ec69Sdan } 2751e341ec69Sdan } 27524fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2753ef7a6304Sdan sqlite3_int64 v = sessionGetI64(aVal); 27544fccf43aSdan if( eType==SQLITE_INTEGER ){ 27554fccf43aSdan sqlite3VdbeMemSetInt64(apOut[i], v); 27564fccf43aSdan }else{ 27574fccf43aSdan double d; 27584e895da1Sdan memcpy(&d, &v, 8); 27594fccf43aSdan sqlite3VdbeMemSetDouble(apOut[i], d); 27604fccf43aSdan } 2761ef7a6304Sdan pIn->iNext += 8; 276291ddd559Sdan } 27634fccf43aSdan } 27644fccf43aSdan } 27654fccf43aSdan 2766ef7a6304Sdan return rc; 2767ef7a6304Sdan } 2768ef7a6304Sdan 2769ef7a6304Sdan /* 2770ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2771ef7a6304Sdan ** Specifically, to the following: 2772ef7a6304Sdan ** 2773ef7a6304Sdan ** + number of columns in table (varint) 2774ef7a6304Sdan ** + array of PK flags (1 byte per column), 2775ef7a6304Sdan ** + table name (nul terminated). 2776ef7a6304Sdan ** 2777ef7a6304Sdan ** This function ensures that all of the above is present in the input 2778ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()). 2779ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code. 2780ef7a6304Sdan ** The input pointer is not moved. 2781ef7a6304Sdan */ 2782ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){ 2783ef7a6304Sdan int rc = SQLITE_OK; 2784ef7a6304Sdan int nCol = 0; 27854757c658Sdan int nRead = 0; 2786ef7a6304Sdan 2787ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2788ef7a6304Sdan if( rc==SQLITE_OK ){ 27894757c658Sdan nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol); 27906344eddaSdan /* The hard upper limit for the number of columns in an SQLite 27916344eddaSdan ** database table is, according to sqliteLimit.h, 32676. So 27926344eddaSdan ** consider any table-header that purports to have more than 65536 27936344eddaSdan ** columns to be corrupt. This is convenient because otherwise, 27946344eddaSdan ** if the (nCol>65536) condition below were omitted, a sufficiently 27956344eddaSdan ** large value for nCol may cause nRead to wrap around and become 27966344eddaSdan ** negative. Leading to a crash. */ 27976344eddaSdan if( nCol<0 || nCol>65536 ){ 2798e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2799e341ec69Sdan }else{ 28004757c658Sdan rc = sessionInputBuffer(pIn, nRead+nCol+100); 28014757c658Sdan nRead += nCol; 2802ef7a6304Sdan } 2803e341ec69Sdan } 28044757c658Sdan 2805ef7a6304Sdan while( rc==SQLITE_OK ){ 28064757c658Sdan while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){ 28074757c658Sdan nRead++; 2808ef7a6304Sdan } 2809e8fa8c96Sdan if( (pIn->iNext + nRead)<pIn->nData ) break; 28104757c658Sdan rc = sessionInputBuffer(pIn, nRead + 100); 28114757c658Sdan } 2812e8fa8c96Sdan *pnByte = nRead+1; 2813ef7a6304Sdan return rc; 2814ef7a6304Sdan } 2815ef7a6304Sdan 2816ef7a6304Sdan /* 2817fa122adaSdan ** The input pointer currently points to the first byte of the first field 2818fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire 281916228167Sdan ** record is buffered. It does not move the input pointer. 282016228167Sdan ** 282116228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of 282216228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The 282316228167Sdan ** final value of *pnByte is undefined in this case. 2824fa122adaSdan */ 2825fa122adaSdan static int sessionChangesetBufferRecord( 282616228167Sdan SessionInput *pIn, /* Input data */ 282716228167Sdan int nCol, /* Number of columns in record */ 282816228167Sdan int *pnByte /* OUT: Size of record in bytes */ 2829fa122adaSdan ){ 2830fa122adaSdan int rc = SQLITE_OK; 2831fa122adaSdan int nByte = 0; 2832fa122adaSdan int i; 2833fa122adaSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 2834fa122adaSdan int eType; 2835fa122adaSdan rc = sessionInputBuffer(pIn, nByte + 10); 2836fa122adaSdan if( rc==SQLITE_OK ){ 2837fa122adaSdan eType = pIn->aData[pIn->iNext + nByte++]; 2838fa122adaSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2839fa122adaSdan int n; 2840fa122adaSdan nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n); 2841fa122adaSdan nByte += n; 2842fa122adaSdan rc = sessionInputBuffer(pIn, nByte); 2843fa122adaSdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2844fa122adaSdan nByte += 8; 2845fa122adaSdan } 2846fa122adaSdan } 2847fa122adaSdan } 2848fa122adaSdan *pnByte = nByte; 2849fa122adaSdan return rc; 2850fa122adaSdan } 2851fa122adaSdan 2852fa122adaSdan /* 2853ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2854ef7a6304Sdan ** Specifically, to the following: 2855ef7a6304Sdan ** 2856ef7a6304Sdan ** + number of columns in table (varint) 2857ef7a6304Sdan ** + array of PK flags (1 byte per column), 2858ef7a6304Sdan ** + table name (nul terminated). 285916228167Sdan ** 286016228167Sdan ** This function decodes the table-header and populates the p->nCol, 286116228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is 286216228167Sdan ** also allocated or resized according to the new value of p->nCol. The 286316228167Sdan ** input pointer is left pointing to the byte following the table header. 286416228167Sdan ** 286516228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code 286616228167Sdan ** is returned and the final values of the various fields enumerated above 286716228167Sdan ** are undefined. 2868ef7a6304Sdan */ 2869ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){ 2870ef7a6304Sdan int rc; 2871ef7a6304Sdan int nCopy; 2872ef7a6304Sdan assert( p->rc==SQLITE_OK ); 2873ef7a6304Sdan 2874ef7a6304Sdan rc = sessionChangesetBufferTblhdr(&p->in, &nCopy); 2875ef7a6304Sdan if( rc==SQLITE_OK ){ 2876ef7a6304Sdan int nByte; 2877ef7a6304Sdan int nVarint; 28784757c658Sdan nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol); 2879dd8a4af8Sdan if( p->nCol>0 ){ 2880ef7a6304Sdan nCopy -= nVarint; 2881ef7a6304Sdan p->in.iNext += nVarint; 2882ef7a6304Sdan nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy; 2883ef7a6304Sdan p->tblhdr.nBuf = 0; 2884ef7a6304Sdan sessionBufferGrow(&p->tblhdr, nByte, &rc); 2885dd8a4af8Sdan }else{ 2886e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2887dd8a4af8Sdan } 2888ef7a6304Sdan } 2889ef7a6304Sdan 2890ef7a6304Sdan if( rc==SQLITE_OK ){ 2891ef7a6304Sdan int iPK = sizeof(sqlite3_value*)*p->nCol*2; 2892ef7a6304Sdan memset(p->tblhdr.aBuf, 0, iPK); 28934757c658Sdan memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy); 2894ef7a6304Sdan p->in.iNext += nCopy; 2895ef7a6304Sdan } 2896ef7a6304Sdan 2897ef7a6304Sdan p->apValue = (sqlite3_value**)p->tblhdr.aBuf; 2898ef7a6304Sdan p->abPK = (u8*)&p->apValue[p->nCol*2]; 2899ef7a6304Sdan p->zTab = (char*)&p->abPK[p->nCol]; 2900ef7a6304Sdan return (p->rc = rc); 29014fccf43aSdan } 29024fccf43aSdan 290377fc1d5bSdan /* 290477fc1d5bSdan ** Advance the changeset iterator to the next change. 290577fc1d5bSdan ** 290677fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public 290777fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the 290877fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values. 290977fc1d5bSdan ** 291077fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change 291177fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in 291277fc1d5bSdan ** the record to *pnRec. 291377fc1d5bSdan ** 291477fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is 291577fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite 291677fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further 291777fc1d5bSdan ** changes in the changeset. 291877fc1d5bSdan */ 29195d607a6eSdan static int sessionChangesetNext( 292077fc1d5bSdan sqlite3_changeset_iter *p, /* Changeset iterator */ 292177fc1d5bSdan u8 **paRec, /* If non-NULL, store record pointer here */ 2922c0a499eaSdan int *pnRec, /* If non-NULL, store size of record here */ 2923c0a499eaSdan int *pbNew /* If non-NULL, true if new table */ 29245d607a6eSdan ){ 29254fccf43aSdan int i; 2926ef7a6304Sdan u8 op; 29274fccf43aSdan 29285d607a6eSdan assert( (paRec==0 && pnRec==0) || (paRec && pnRec) ); 29295d607a6eSdan 2930296c7658Sdan /* If the iterator is in the error-state, return immediately. */ 29314fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29324fccf43aSdan 29335d607a6eSdan /* Free the current contents of p->apValue[], if any. */ 29344fccf43aSdan if( p->apValue ){ 29354fccf43aSdan for(i=0; i<p->nCol*2; i++){ 29364fccf43aSdan sqlite3ValueFree(p->apValue[i]); 29374fccf43aSdan } 29384fccf43aSdan memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2); 29394fccf43aSdan } 29404fccf43aSdan 2941ef7a6304Sdan /* Make sure the buffer contains at least 10 bytes of input data, or all 2942ef7a6304Sdan ** remaining data if there are less than 10 bytes available. This is 2943ef7a6304Sdan ** sufficient either for the 'T' or 'P' byte and the varint that follows 2944ef7a6304Sdan ** it, or for the two single byte values otherwise. */ 2945ef7a6304Sdan p->rc = sessionInputBuffer(&p->in, 2); 2946ef7a6304Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2947ef7a6304Sdan 29484fccf43aSdan /* If the iterator is already at the end of the changeset, return DONE. */ 29494757c658Sdan if( p->in.iNext>=p->in.nData ){ 29504fccf43aSdan return SQLITE_DONE; 29514fccf43aSdan } 29524fccf43aSdan 2953d9151526Sdan sessionDiscardData(&p->in); 2954d9151526Sdan p->in.iCurrent = p->in.iNext; 2955d9151526Sdan 29564757c658Sdan op = p->in.aData[p->in.iNext++]; 295707d0f15eSdan while( op=='T' || op=='P' ){ 2958c0a499eaSdan if( pbNew ) *pbNew = 1; 2959ef7a6304Sdan p->bPatchset = (op=='P'); 2960ef7a6304Sdan if( sessionChangesetReadTblhdr(p) ) return p->rc; 2961ef7a6304Sdan if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc; 2962d9151526Sdan p->in.iCurrent = p->in.iNext; 296307d0f15eSdan if( p->in.iNext>=p->in.nData ) return SQLITE_DONE; 29644757c658Sdan op = p->in.aData[p->in.iNext++]; 29655d607a6eSdan } 29665d607a6eSdan 2967dd8a4af8Sdan if( p->zTab==0 ){ 2968dd8a4af8Sdan /* The first record in the changeset is not a table header. Must be a 2969dd8a4af8Sdan ** corrupt changeset. */ 2970dd8a4af8Sdan assert( p->in.iNext==1 ); 2971dd8a4af8Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 2972dd8a4af8Sdan } 2973dd8a4af8Sdan 2974ef7a6304Sdan p->op = op; 29754757c658Sdan p->bIndirect = p->in.aData[p->in.iNext++]; 29764fccf43aSdan if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){ 29774757c658Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 29784fccf43aSdan } 29794fccf43aSdan 2980cbf6d2d2Sdan if( paRec ){ 2981cbf6d2d2Sdan int nVal; /* Number of values to buffer */ 2982cbf6d2d2Sdan if( p->bPatchset==0 && op==SQLITE_UPDATE ){ 2983cbf6d2d2Sdan nVal = p->nCol * 2; 2984cbf6d2d2Sdan }else if( p->bPatchset && op==SQLITE_DELETE ){ 2985cbf6d2d2Sdan nVal = 0; 2986cbf6d2d2Sdan for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++; 2987cbf6d2d2Sdan }else{ 2988cbf6d2d2Sdan nVal = p->nCol; 2989cbf6d2d2Sdan } 2990cbf6d2d2Sdan p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec); 2991cbf6d2d2Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2992cbf6d2d2Sdan *paRec = &p->in.aData[p->in.iNext]; 2993cbf6d2d2Sdan p->in.iNext += *pnRec; 2994cbf6d2d2Sdan }else{ 29955d607a6eSdan 29964fccf43aSdan /* If this is an UPDATE or DELETE, read the old.* record. */ 299773b3c055Sdan if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){ 299873b3c055Sdan u8 *abPK = p->bPatchset ? p->abPK : 0; 2999cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue); 30004fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30014fccf43aSdan } 30024fccf43aSdan 30034fccf43aSdan /* If this is an INSERT or UPDATE, read the new.* record. */ 30044fccf43aSdan if( p->op!=SQLITE_DELETE ){ 3005cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]); 30064fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30074fccf43aSdan } 30084fccf43aSdan 3009cbf6d2d2Sdan if( p->bPatchset && p->op==SQLITE_UPDATE ){ 301073b3c055Sdan /* If this is an UPDATE that is part of a patchset, then all PK and 301173b3c055Sdan ** modified fields are present in the new.* record. The old.* record 301273b3c055Sdan ** is currently completely empty. This block shifts the PK fields from 301373b3c055Sdan ** new.* to old.*, to accommodate the code that reads these arrays. */ 301473b3c055Sdan for(i=0; i<p->nCol; i++){ 301573b3c055Sdan assert( p->apValue[i]==0 ); 301673b3c055Sdan if( p->abPK[i] ){ 301773b3c055Sdan p->apValue[i] = p->apValue[i+p->nCol]; 3018e341ec69Sdan if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT); 301973b3c055Sdan p->apValue[i+p->nCol] = 0; 302073b3c055Sdan } 302173b3c055Sdan } 302273b3c055Sdan } 3023cbf6d2d2Sdan } 3024ef7a6304Sdan 30254fccf43aSdan return SQLITE_ROW; 30264fccf43aSdan } 30274fccf43aSdan 30284fccf43aSdan /* 30295d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next 30305d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE 30315d607a6eSdan ** or SQLITE_CORRUPT. 30325d607a6eSdan ** 30335d607a6eSdan ** This function may not be called on iterators passed to a conflict handler 30345d607a6eSdan ** callback by changeset_apply(). 30355d607a6eSdan */ 30365d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){ 3037c0a499eaSdan return sessionChangesetNext(p, 0, 0, 0); 30385d607a6eSdan } 30395d607a6eSdan 30405d607a6eSdan /* 3041244593c8Sdan ** The following function extracts information on the current change 304277fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next() 30434fccf43aSdan ** has returned SQLITE_ROW. 30444fccf43aSdan */ 30454fccf43aSdan int sqlite3changeset_op( 3046296c7658Sdan sqlite3_changeset_iter *pIter, /* Iterator handle */ 30474fccf43aSdan const char **pzTab, /* OUT: Pointer to table name */ 30484fccf43aSdan int *pnCol, /* OUT: Number of columns in table */ 3049b4480e94Sdan int *pOp, /* OUT: SQLITE_INSERT, DELETE or UPDATE */ 3050b4480e94Sdan int *pbIndirect /* OUT: True if change is indirect */ 30514fccf43aSdan ){ 30524fccf43aSdan *pOp = pIter->op; 30534fccf43aSdan *pnCol = pIter->nCol; 30544fccf43aSdan *pzTab = pIter->zTab; 3055b4480e94Sdan if( pbIndirect ) *pbIndirect = pIter->bIndirect; 30564fccf43aSdan return SQLITE_OK; 30574fccf43aSdan } 30584fccf43aSdan 305977fc1d5bSdan /* 306077fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in 306177fc1d5bSdan ** the database table affected by the change that pIter currently points 306277fc1d5bSdan ** to. This function may only be called after changeset_next() returns 306377fc1d5bSdan ** SQLITE_ROW. 306477fc1d5bSdan */ 3065244593c8Sdan int sqlite3changeset_pk( 3066244593c8Sdan sqlite3_changeset_iter *pIter, /* Iterator object */ 3067244593c8Sdan unsigned char **pabPK, /* OUT: Array of boolean - true for PK cols */ 3068244593c8Sdan int *pnCol /* OUT: Number of entries in output array */ 3069244593c8Sdan ){ 3070244593c8Sdan *pabPK = pIter->abPK; 3071244593c8Sdan if( pnCol ) *pnCol = pIter->nCol; 3072244593c8Sdan return SQLITE_OK; 3073244593c8Sdan } 3074244593c8Sdan 3075296c7658Sdan /* 3076296c7658Sdan ** This function may only be called while the iterator is pointing to an 3077296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()). 3078296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3079296c7658Sdan ** 3080296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3081296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not 3082296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3083296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL. 3084296c7658Sdan ** 3085296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3086296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3087296c7658Sdan */ 30884fccf43aSdan int sqlite3changeset_old( 3089296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3090296c7658Sdan int iVal, /* Index of old.* value to retrieve */ 30914fccf43aSdan sqlite3_value **ppValue /* OUT: Old value (or NULL pointer) */ 30924fccf43aSdan ){ 3093d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){ 3094d5f0767cSdan return SQLITE_MISUSE; 3095d5f0767cSdan } 30964fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 30974fccf43aSdan return SQLITE_RANGE; 30984fccf43aSdan } 30994fccf43aSdan *ppValue = pIter->apValue[iVal]; 31004fccf43aSdan return SQLITE_OK; 31014fccf43aSdan } 31024fccf43aSdan 3103296c7658Sdan /* 3104296c7658Sdan ** This function may only be called while the iterator is pointing to an 3105296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()). 3106296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3107296c7658Sdan ** 3108296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3109296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not 3110296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3111296c7658Sdan ** was not modified), set *ppValue to NULL. 3112296c7658Sdan ** 3113296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3114296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3115296c7658Sdan */ 31164fccf43aSdan int sqlite3changeset_new( 3117296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3118296c7658Sdan int iVal, /* Index of new.* value to retrieve */ 31194fccf43aSdan sqlite3_value **ppValue /* OUT: New value (or NULL pointer) */ 31204fccf43aSdan ){ 3121d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){ 3122d5f0767cSdan return SQLITE_MISUSE; 3123d5f0767cSdan } 31244fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 31254fccf43aSdan return SQLITE_RANGE; 31264fccf43aSdan } 31274fccf43aSdan *ppValue = pIter->apValue[pIter->nCol+iVal]; 31284fccf43aSdan return SQLITE_OK; 31294fccf43aSdan } 31304fccf43aSdan 3131296c7658Sdan /* 31327aa469cdSdan ** The following two macros are used internally. They are similar to the 31337aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that 31347aa469cdSdan ** they omit all error checking and return a pointer to the requested value. 31357aa469cdSdan */ 31367aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)] 31377aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)] 31387aa469cdSdan 31397aa469cdSdan /* 3140296c7658Sdan ** This function may only be called with a changeset iterator that has been 3141296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT 3142296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned. 3143296c7658Sdan ** 3144296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure 3145296c7658Sdan ** containing the iVal'th value of the conflicting record. 3146296c7658Sdan ** 3147296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error 3148296c7658Sdan ** code is returned. Otherwise, SQLITE_OK. 3149296c7658Sdan */ 3150d5f0767cSdan int sqlite3changeset_conflict( 3151296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3152296c7658Sdan int iVal, /* Index of conflict record value to fetch */ 3153d5f0767cSdan sqlite3_value **ppValue /* OUT: Value from conflicting row */ 3154d5f0767cSdan ){ 3155d5f0767cSdan if( !pIter->pConflict ){ 3156d5f0767cSdan return SQLITE_MISUSE; 3157d5f0767cSdan } 3158ff677b20Sdan if( iVal<0 || iVal>=pIter->nCol ){ 3159d5f0767cSdan return SQLITE_RANGE; 3160d5f0767cSdan } 3161d5f0767cSdan *ppValue = sqlite3_column_value(pIter->pConflict, iVal); 3162d5f0767cSdan return SQLITE_OK; 3163d5f0767cSdan } 3164d5f0767cSdan 31654fccf43aSdan /* 3166cb3e4b79Sdan ** This function may only be called with an iterator passed to an 3167cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case 3168cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key 3169cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK. 3170cb3e4b79Sdan ** 3171cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE. 3172cb3e4b79Sdan */ 3173cb3e4b79Sdan int sqlite3changeset_fk_conflicts( 3174cb3e4b79Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3175cb3e4b79Sdan int *pnOut /* OUT: Number of FK violations */ 3176cb3e4b79Sdan ){ 3177cb3e4b79Sdan if( pIter->pConflict || pIter->apValue ){ 3178cb3e4b79Sdan return SQLITE_MISUSE; 3179cb3e4b79Sdan } 3180cb3e4b79Sdan *pnOut = pIter->nCol; 3181cb3e4b79Sdan return SQLITE_OK; 3182cb3e4b79Sdan } 3183cb3e4b79Sdan 3184cb3e4b79Sdan 3185cb3e4b79Sdan /* 31864fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start(). 31874fccf43aSdan ** 31884fccf43aSdan ** This function may not be called on iterators passed to a conflict handler 31894fccf43aSdan ** callback by changeset_apply(). 31904fccf43aSdan */ 31914fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){ 3192cbf6d2d2Sdan int rc = SQLITE_OK; 3193cbf6d2d2Sdan if( p ){ 3194296c7658Sdan int i; /* Used to iterate through p->apValue[] */ 3195cbf6d2d2Sdan rc = p->rc; 319612ca0b56Sdan if( p->apValue ){ 31974fccf43aSdan for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]); 319812ca0b56Sdan } 3199ef7a6304Sdan sqlite3_free(p->tblhdr.aBuf); 32004757c658Sdan sqlite3_free(p->in.buf.aBuf); 32014fccf43aSdan sqlite3_free(p); 3202cbf6d2d2Sdan } 32034fccf43aSdan return rc; 32044fccf43aSdan } 32054fccf43aSdan 3206fa122adaSdan static int sessionChangesetInvert( 3207fa122adaSdan SessionInput *pInput, /* Input changeset */ 3208fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3209fa122adaSdan void *pOut, 321091ddd559Sdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 321191ddd559Sdan void **ppInverted /* OUT: Inverse of pChangeset */ 321291ddd559Sdan ){ 3213cfec7eeeSdan int rc = SQLITE_OK; /* Return value */ 3214fa122adaSdan SessionBuffer sOut; /* Output buffer */ 3215cfec7eeeSdan int nCol = 0; /* Number of cols in current table */ 3216cfec7eeeSdan u8 *abPK = 0; /* PK array for current table */ 3217cfec7eeeSdan sqlite3_value **apVal = 0; /* Space for values for UPDATE inversion */ 3218ef7a6304Sdan SessionBuffer sPK = {0, 0, 0}; /* PK array for current table */ 321991ddd559Sdan 3220fa122adaSdan /* Initialize the output buffer */ 3221fa122adaSdan memset(&sOut, 0, sizeof(SessionBuffer)); 3222fa122adaSdan 322391ddd559Sdan /* Zero the output variables in case an error occurs. */ 3224fa122adaSdan if( ppInverted ){ 322591ddd559Sdan *ppInverted = 0; 322691ddd559Sdan *pnInverted = 0; 3227fa122adaSdan } 322891ddd559Sdan 3229fa122adaSdan while( 1 ){ 3230ef7a6304Sdan u8 eType; 3231fa122adaSdan 3232fa122adaSdan /* Test for EOF. */ 3233fa122adaSdan if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert; 3234fa122adaSdan if( pInput->iNext>=pInput->nData ) break; 3235fa122adaSdan eType = pInput->aData[pInput->iNext]; 3236fa122adaSdan 323791ddd559Sdan switch( eType ){ 323891ddd559Sdan case 'T': { 3239244593c8Sdan /* A 'table' record consists of: 3240244593c8Sdan ** 3241244593c8Sdan ** * A constant 'T' character, 3242244593c8Sdan ** * Number of columns in said table (a varint), 3243ef7a6304Sdan ** * An array of nCol bytes (sPK), 3244244593c8Sdan ** * A nul-terminated table name. 3245244593c8Sdan */ 3246ef7a6304Sdan int nByte; 3247fa122adaSdan int nVar; 3248fa122adaSdan pInput->iNext++; 3249fa122adaSdan if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){ 3250ef7a6304Sdan goto finished_invert; 3251ef7a6304Sdan } 3252fa122adaSdan nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol); 3253ef7a6304Sdan sPK.nBuf = 0; 3254fa122adaSdan sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc); 3255fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3256fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3257ef7a6304Sdan if( rc ) goto finished_invert; 3258fa122adaSdan 3259fa122adaSdan pInput->iNext += nByte; 3260cfec7eeeSdan sqlite3_free(apVal); 3261cfec7eeeSdan apVal = 0; 3262ef7a6304Sdan abPK = sPK.aBuf; 326391ddd559Sdan break; 326491ddd559Sdan } 326591ddd559Sdan 326691ddd559Sdan case SQLITE_INSERT: 326791ddd559Sdan case SQLITE_DELETE: { 326891ddd559Sdan int nByte; 3269fa122adaSdan int bIndirect = pInput->aData[pInput->iNext+1]; 3270fa122adaSdan int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE); 3271fa122adaSdan pInput->iNext += 2; 3272fa122adaSdan assert( rc==SQLITE_OK ); 3273fa122adaSdan rc = sessionChangesetBufferRecord(pInput, nCol, &nByte); 3274fa122adaSdan sessionAppendByte(&sOut, eType2, &rc); 3275fa122adaSdan sessionAppendByte(&sOut, bIndirect, &rc); 3276fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3277fa122adaSdan pInput->iNext += nByte; 3278fa122adaSdan if( rc ) goto finished_invert; 327991ddd559Sdan break; 328091ddd559Sdan } 328191ddd559Sdan 328291ddd559Sdan case SQLITE_UPDATE: { 3283cfec7eeeSdan int iCol; 328491ddd559Sdan 3285cfec7eeeSdan if( 0==apVal ){ 3286cfec7eeeSdan apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2); 3287cfec7eeeSdan if( 0==apVal ){ 3288cfec7eeeSdan rc = SQLITE_NOMEM; 3289cfec7eeeSdan goto finished_invert; 3290cfec7eeeSdan } 3291cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3292cfec7eeeSdan } 329391ddd559Sdan 3294cfec7eeeSdan /* Write the header for the new UPDATE change. Same as the original. */ 3295fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3296fa122adaSdan sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc); 329791ddd559Sdan 3298ef7a6304Sdan /* Read the old.* and new.* records for the update change. */ 3299fa122adaSdan pInput->iNext += 2; 3300fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]); 3301ef7a6304Sdan if( rc==SQLITE_OK ){ 3302fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]); 3303ef7a6304Sdan } 3304ef7a6304Sdan 3305cfec7eeeSdan /* Write the new old.* record. Consists of the PK columns from the 3306cfec7eeeSdan ** original old.* record, and the other values from the original 3307cfec7eeeSdan ** new.* record. */ 3308e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3309cfec7eeeSdan sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)]; 3310fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3311cfec7eeeSdan } 3312cfec7eeeSdan 3313cfec7eeeSdan /* Write the new new.* record. Consists of a copy of all values 3314cfec7eeeSdan ** from the original old.* record, except for the PK columns, which 3315cfec7eeeSdan ** are set to "undefined". */ 3316e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3317cfec7eeeSdan sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]); 3318fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3319cfec7eeeSdan } 3320cfec7eeeSdan 3321cfec7eeeSdan for(iCol=0; iCol<nCol*2; iCol++){ 3322cfec7eeeSdan sqlite3ValueFree(apVal[iCol]); 3323cfec7eeeSdan } 3324cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3325cfec7eeeSdan if( rc!=SQLITE_OK ){ 3326cfec7eeeSdan goto finished_invert; 3327cfec7eeeSdan } 3328cfec7eeeSdan 332991ddd559Sdan break; 333091ddd559Sdan } 333191ddd559Sdan 333291ddd559Sdan default: 33334757c658Sdan rc = SQLITE_CORRUPT_BKPT; 3334cfec7eeeSdan goto finished_invert; 333591ddd559Sdan } 3336fa122adaSdan 3337fa122adaSdan assert( rc==SQLITE_OK ); 3338f1a08ad8Sdrh if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 3339fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3340fa122adaSdan sOut.nBuf = 0; 3341fa122adaSdan if( rc!=SQLITE_OK ) goto finished_invert; 3342fa122adaSdan } 334391ddd559Sdan } 334491ddd559Sdan 3345cfec7eeeSdan assert( rc==SQLITE_OK ); 3346fa122adaSdan if( pnInverted ){ 3347fa122adaSdan *pnInverted = sOut.nBuf; 3348fa122adaSdan *ppInverted = sOut.aBuf; 3349fa122adaSdan sOut.aBuf = 0; 3350fa122adaSdan }else if( sOut.nBuf>0 ){ 3351fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3352fa122adaSdan } 3353cfec7eeeSdan 3354cfec7eeeSdan finished_invert: 3355fa122adaSdan sqlite3_free(sOut.aBuf); 3356cfec7eeeSdan sqlite3_free(apVal); 3357ef7a6304Sdan sqlite3_free(sPK.aBuf); 3358cfec7eeeSdan return rc; 335991ddd559Sdan } 336091ddd559Sdan 3361fa122adaSdan 3362fa122adaSdan /* 3363fa122adaSdan ** Invert a changeset object. 3364fa122adaSdan */ 3365fa122adaSdan int sqlite3changeset_invert( 3366fa122adaSdan int nChangeset, /* Number of bytes in input */ 3367fa122adaSdan const void *pChangeset, /* Input changeset */ 3368fa122adaSdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 3369fa122adaSdan void **ppInverted /* OUT: Inverse of pChangeset */ 3370fa122adaSdan ){ 3371fa122adaSdan SessionInput sInput; 3372fa122adaSdan 3373fa122adaSdan /* Set up the input stream */ 3374fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3375fa122adaSdan sInput.nData = nChangeset; 3376fa122adaSdan sInput.aData = (u8*)pChangeset; 3377fa122adaSdan 3378fa122adaSdan return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted); 3379fa122adaSdan } 3380fa122adaSdan 3381fa122adaSdan /* 3382fa122adaSdan ** Streaming version of sqlite3changeset_invert(). 3383fa122adaSdan */ 3384f1a08ad8Sdrh int sqlite3changeset_invert_strm( 3385fa122adaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 3386fa122adaSdan void *pIn, 3387fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3388fa122adaSdan void *pOut 3389fa122adaSdan ){ 3390fa122adaSdan SessionInput sInput; 3391fa122adaSdan int rc; 3392fa122adaSdan 3393fa122adaSdan /* Set up the input stream */ 3394fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3395fa122adaSdan sInput.xInput = xInput; 3396fa122adaSdan sInput.pIn = pIn; 3397fa122adaSdan 3398fa122adaSdan rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0); 3399fa122adaSdan sqlite3_free(sInput.buf.aBuf); 3400fa122adaSdan return rc; 3401fa122adaSdan } 3402fa122adaSdan 34030c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx; 34040c698471Sdan struct SessionApplyCtx { 34050c698471Sdan sqlite3 *db; 34060c698471Sdan sqlite3_stmt *pDelete; /* DELETE statement */ 3407cfec7eeeSdan sqlite3_stmt *pUpdate; /* UPDATE statement */ 34080c698471Sdan sqlite3_stmt *pInsert; /* INSERT statement */ 34090c698471Sdan sqlite3_stmt *pSelect; /* SELECT statement */ 34100c698471Sdan int nCol; /* Size of azCol[] and abPK[] arrays */ 34110c698471Sdan const char **azCol; /* Array of column names */ 34120c698471Sdan u8 *abPK; /* Boolean array - true if column is in PK */ 3413d1cccf19Sdan int bStat1; /* True if table is sqlite_stat1 */ 3414d9151526Sdan int bDeferConstraints; /* True to defer constraints */ 3415d9151526Sdan SessionBuffer constraints; /* Deferred constraints are stored here */ 3416a38e6c57Sdan SessionBuffer rebase; /* Rebase information (if any) here */ 3417a38e6c57Sdan int bRebaseStarted; /* If table header is already in rebase */ 34180c698471Sdan }; 34190c698471Sdan 3420d5f0767cSdan /* 3421d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table 3422d5f0767cSdan ** structure like this: 3423d5f0767cSdan ** 3424d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3425d5f0767cSdan ** 3426d5f0767cSdan ** The DELETE statement looks like this: 3427d5f0767cSdan ** 3428db04571cSdan ** DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4) 3429d5f0767cSdan ** 3430d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require 3431d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the 3432d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE. 3433296c7658Sdan ** 3434296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left 3435296c7658Sdan ** pointing to the prepared version of the SQL statement. 3436d5f0767cSdan */ 3437d5f0767cSdan static int sessionDeleteRow( 3438d5f0767cSdan sqlite3 *db, /* Database handle */ 3439d5f0767cSdan const char *zTab, /* Table name */ 34400c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3441d5f0767cSdan ){ 3442296c7658Sdan int i; 3443296c7658Sdan const char *zSep = ""; 3444d5f0767cSdan int rc = SQLITE_OK; 3445d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 34467cf7df7dSdan int nPk = 0; 3447d5f0767cSdan 3448d5f0767cSdan sessionAppendStr(&buf, "DELETE FROM ", &rc); 3449d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3450296c7658Sdan sessionAppendStr(&buf, " WHERE ", &rc); 3451296c7658Sdan 3452296c7658Sdan for(i=0; i<p->nCol; i++){ 3453296c7658Sdan if( p->abPK[i] ){ 34547cf7df7dSdan nPk++; 3455296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3456296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3457296c7658Sdan sessionAppendStr(&buf, " = ?", &rc); 3458296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3459296c7658Sdan zSep = " AND "; 3460296c7658Sdan } 3461296c7658Sdan } 3462296c7658Sdan 34637cf7df7dSdan if( nPk<p->nCol ){ 3464296c7658Sdan sessionAppendStr(&buf, " AND (?", &rc); 3465296c7658Sdan sessionAppendInteger(&buf, p->nCol+1, &rc); 3466296c7658Sdan sessionAppendStr(&buf, " OR ", &rc); 3467296c7658Sdan 3468296c7658Sdan zSep = ""; 3469296c7658Sdan for(i=0; i<p->nCol; i++){ 3470296c7658Sdan if( !p->abPK[i] ){ 3471296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3472296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3473296c7658Sdan sessionAppendStr(&buf, " IS ?", &rc); 3474296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3475296c7658Sdan zSep = "AND "; 3476296c7658Sdan } 3477296c7658Sdan } 3478296c7658Sdan sessionAppendStr(&buf, ")", &rc); 34797cf7df7dSdan } 3480d5f0767cSdan 3481d5f0767cSdan if( rc==SQLITE_OK ){ 34820c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0); 3483d5f0767cSdan } 3484d5f0767cSdan sqlite3_free(buf.aBuf); 3485d5f0767cSdan 3486d5f0767cSdan return rc; 3487d5f0767cSdan } 3488d5f0767cSdan 3489d5f0767cSdan /* 3490d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db. 3491d5f0767cSdan ** Assuming a table structure like this: 3492d5f0767cSdan ** 3493d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3494d5f0767cSdan ** 3495d5f0767cSdan ** The UPDATE statement looks like this: 3496d5f0767cSdan ** 3497d5f0767cSdan ** UPDATE x SET 3498d5f0767cSdan ** a = CASE WHEN ?2 THEN ?3 ELSE a END, 3499964cbd46Sdan ** b = CASE WHEN ?5 THEN ?6 ELSE b END, 3500964cbd46Sdan ** c = CASE WHEN ?8 THEN ?9 ELSE c END, 3501964cbd46Sdan ** d = CASE WHEN ?11 THEN ?12 ELSE d END 3502d5f0767cSdan ** WHERE a = ?1 AND c = ?7 AND (?13 OR 3503964cbd46Sdan ** (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND 3504d5f0767cSdan ** ) 3505d5f0767cSdan ** 3506d5f0767cSdan ** For each column in the table, there are three variables to bind: 3507d5f0767cSdan ** 3508d5f0767cSdan ** ?(i*3+1) The old.* value of the column, if any. 3509d5f0767cSdan ** ?(i*3+2) A boolean flag indicating that the value is being modified. 3510d5f0767cSdan ** ?(i*3+3) The new.* value of the column, if any. 3511d5f0767cSdan ** 3512d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update 3513d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the 3514d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns 3515d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)". 3516d5f0767cSdan ** 3517296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left 3518296c7658Sdan ** pointing to the prepared version of the SQL statement. 3519d5f0767cSdan */ 3520d5f0767cSdan static int sessionUpdateRow( 3521d5f0767cSdan sqlite3 *db, /* Database handle */ 3522d5f0767cSdan const char *zTab, /* Table name */ 35230c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3524d5f0767cSdan ){ 3525d5f0767cSdan int rc = SQLITE_OK; 3526d5f0767cSdan int i; 3527d5f0767cSdan const char *zSep = ""; 3528d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 3529d5f0767cSdan 3530d5f0767cSdan /* Append "UPDATE tbl SET " */ 3531d5f0767cSdan sessionAppendStr(&buf, "UPDATE ", &rc); 3532d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3533d5f0767cSdan sessionAppendStr(&buf, " SET ", &rc); 3534d5f0767cSdan 3535d5f0767cSdan /* Append the assignments */ 35360c698471Sdan for(i=0; i<p->nCol; i++){ 3537d5f0767cSdan sessionAppendStr(&buf, zSep, &rc); 35380c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3539d5f0767cSdan sessionAppendStr(&buf, " = CASE WHEN ?", &rc); 3540d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3541d5f0767cSdan sessionAppendStr(&buf, " THEN ?", &rc); 3542d5f0767cSdan sessionAppendInteger(&buf, i*3+3, &rc); 3543d5f0767cSdan sessionAppendStr(&buf, " ELSE ", &rc); 35440c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3545d5f0767cSdan sessionAppendStr(&buf, " END", &rc); 3546d5f0767cSdan zSep = ", "; 3547d5f0767cSdan } 3548d5f0767cSdan 3549d5f0767cSdan /* Append the PK part of the WHERE clause */ 3550d5f0767cSdan sessionAppendStr(&buf, " WHERE ", &rc); 35510c698471Sdan for(i=0; i<p->nCol; i++){ 35520c698471Sdan if( p->abPK[i] ){ 35530c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3554d5f0767cSdan sessionAppendStr(&buf, " = ?", &rc); 3555d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3556d5f0767cSdan sessionAppendStr(&buf, " AND ", &rc); 3557d5f0767cSdan } 3558d5f0767cSdan } 3559d5f0767cSdan 3560d5f0767cSdan /* Append the non-PK part of the WHERE clause */ 3561d5f0767cSdan sessionAppendStr(&buf, " (?", &rc); 35620c698471Sdan sessionAppendInteger(&buf, p->nCol*3+1, &rc); 3563d5f0767cSdan sessionAppendStr(&buf, " OR 1", &rc); 35640c698471Sdan for(i=0; i<p->nCol; i++){ 35650c698471Sdan if( !p->abPK[i] ){ 3566d5f0767cSdan sessionAppendStr(&buf, " AND (?", &rc); 3567d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3568d5f0767cSdan sessionAppendStr(&buf, "=0 OR ", &rc); 35690c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3570d5f0767cSdan sessionAppendStr(&buf, " IS ?", &rc); 3571d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3572d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3573d5f0767cSdan } 3574d5f0767cSdan } 3575d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3576d5f0767cSdan 3577d5f0767cSdan if( rc==SQLITE_OK ){ 35780c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0); 3579d5f0767cSdan } 3580d5f0767cSdan sqlite3_free(buf.aBuf); 3581d5f0767cSdan 3582d5f0767cSdan return rc; 3583d5f0767cSdan } 3584d5f0767cSdan 35853739f298Sdan 3586296c7658Sdan /* 3587296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary 3588296c7658Sdan ** key. Assuming the following table structure: 3589296c7658Sdan ** 3590296c7658Sdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3591296c7658Sdan ** 3592296c7658Sdan ** The SELECT statement looks like this: 3593296c7658Sdan ** 3594296c7658Sdan ** SELECT * FROM x WHERE a = ?1 AND c = ?3 3595296c7658Sdan ** 3596296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left 3597296c7658Sdan ** pointing to the prepared version of the SQL statement. 3598296c7658Sdan */ 3599d5f0767cSdan static int sessionSelectRow( 3600d5f0767cSdan sqlite3 *db, /* Database handle */ 3601d5f0767cSdan const char *zTab, /* Table name */ 36020c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3603d5f0767cSdan ){ 3604d7fb7d24Sdan return sessionSelectStmt( 3605d7fb7d24Sdan db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect); 3606d5f0767cSdan } 3607d5f0767cSdan 3608296c7658Sdan /* 3609296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab. 3610296c7658Sdan ** For example: 3611296c7658Sdan ** 3612296c7658Sdan ** INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...); 3613296c7658Sdan ** 3614296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left 3615296c7658Sdan ** pointing to the prepared version of the SQL statement. 3616296c7658Sdan */ 36170c698471Sdan static int sessionInsertRow( 36180c698471Sdan sqlite3 *db, /* Database handle */ 36190c698471Sdan const char *zTab, /* Table name */ 36200c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 36210c698471Sdan ){ 36220c698471Sdan int rc = SQLITE_OK; 36230c698471Sdan int i; 36240c698471Sdan SessionBuffer buf = {0, 0, 0}; 36250c698471Sdan 36260c698471Sdan sessionAppendStr(&buf, "INSERT INTO main.", &rc); 36270c698471Sdan sessionAppendIdent(&buf, zTab, &rc); 3628ff677b20Sdan sessionAppendStr(&buf, "(", &rc); 3629ff677b20Sdan for(i=0; i<p->nCol; i++){ 3630ff677b20Sdan if( i!=0 ) sessionAppendStr(&buf, ", ", &rc); 3631ff677b20Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3632ff677b20Sdan } 3633ff677b20Sdan 3634ff677b20Sdan sessionAppendStr(&buf, ") VALUES(?", &rc); 36350c698471Sdan for(i=1; i<p->nCol; i++){ 36360c698471Sdan sessionAppendStr(&buf, ", ?", &rc); 36370c698471Sdan } 36380c698471Sdan sessionAppendStr(&buf, ")", &rc); 36390c698471Sdan 36400c698471Sdan if( rc==SQLITE_OK ){ 36410c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0); 36420c698471Sdan } 36430c698471Sdan sqlite3_free(buf.aBuf); 36440c698471Sdan return rc; 36450c698471Sdan } 36460c698471Sdan 36473739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){ 36483739f298Sdan return sqlite3_prepare_v2(db, zSql, -1, pp, 0); 36493739f298Sdan } 36503739f298Sdan 36513739f298Sdan /* 36523739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table. 36533739f298Sdan ** These are similar to those created by sessionSelectRow(), 36543739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for 36553739f298Sdan ** other tables. 36563739f298Sdan */ 36573739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){ 36583739f298Sdan int rc = sessionSelectRow(db, "sqlite_stat1", p); 36593739f298Sdan if( rc==SQLITE_OK ){ 36603739f298Sdan rc = sessionPrepare(db, &p->pInsert, 36613739f298Sdan "INSERT INTO main.sqlite_stat1 VALUES(?1, " 36623739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, " 36633739f298Sdan "?3)" 36643739f298Sdan ); 36653739f298Sdan } 36663739f298Sdan if( rc==SQLITE_OK ){ 36673739f298Sdan rc = sessionPrepare(db, &p->pUpdate, 36683739f298Sdan "UPDATE main.sqlite_stat1 SET " 36693739f298Sdan "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, " 36703739f298Sdan "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, " 36713739f298Sdan "stat = CASE WHEN ?8 THEN ?9 ELSE stat END " 36723739f298Sdan "WHERE tbl=?1 AND idx IS " 36733739f298Sdan "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END " 36743739f298Sdan "AND (?10 OR ?8=0 OR stat IS ?7)" 36753739f298Sdan ); 36763739f298Sdan } 36773739f298Sdan if( rc==SQLITE_OK ){ 36783739f298Sdan rc = sessionPrepare(db, &p->pDelete, 36793739f298Sdan "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS " 36803739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END " 36813739f298Sdan "AND (?4 OR stat IS ?3)" 36823739f298Sdan ); 36833739f298Sdan } 36843739f298Sdan return rc; 36853739f298Sdan } 36863739f298Sdan 3687296c7658Sdan /* 36887aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem. 36897aa469cdSdan ** See comments in the body of this function for details. 36907aa469cdSdan */ 36917aa469cdSdan static int sessionBindValue( 36927aa469cdSdan sqlite3_stmt *pStmt, /* Statement to bind value to */ 36937aa469cdSdan int i, /* Parameter number to bind to */ 36947aa469cdSdan sqlite3_value *pVal /* Value to bind */ 36957aa469cdSdan ){ 36965671ef69Sdrh int eType = sqlite3_value_type(pVal); 3697082c96dfSdan /* COVERAGE: The (pVal->z==0) branch is never true using current versions 3698082c96dfSdan ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either 3699082c96dfSdan ** the (pVal->z) variable remains as it was or the type of the value is 3700082c96dfSdan ** set to SQLITE_NULL. */ 37015671ef69Sdrh if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){ 37027aa469cdSdan /* This condition occurs when an earlier OOM in a call to 37037aa469cdSdan ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within 3704082c96dfSdan ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */ 37057aa469cdSdan return SQLITE_NOMEM; 37067aa469cdSdan } 37077aa469cdSdan return sqlite3_bind_value(pStmt, i, pVal); 37087aa469cdSdan } 37097aa469cdSdan 37107aa469cdSdan /* 3711db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function 3712db04571cSdan ** transfers new.* values from the current iterator entry to statement 3713db04571cSdan ** pStmt. The table being inserted into has nCol columns. 3714db04571cSdan ** 3715d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of 3716db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1) 3717db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points 3718db04571cSdan ** to an array nCol elements in size. In this case only those values for 3719db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the 3720db04571cSdan ** statement. 3721db04571cSdan ** 3722db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK. 3723db04571cSdan */ 37247aa469cdSdan static int sessionBindRow( 3725db04571cSdan sqlite3_changeset_iter *pIter, /* Iterator to read values from */ 37267aa469cdSdan int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **), 3727db04571cSdan int nCol, /* Number of columns */ 3728db04571cSdan u8 *abPK, /* If not NULL, bind only if true */ 3729db04571cSdan sqlite3_stmt *pStmt /* Bind values to this statement */ 3730db04571cSdan ){ 3731db04571cSdan int i; 3732db04571cSdan int rc = SQLITE_OK; 37337aa469cdSdan 37347aa469cdSdan /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the 37357aa469cdSdan ** argument iterator points to a suitable entry. Make sure that xValue 37367aa469cdSdan ** is one of these to guarantee that it is safe to ignore the return 37377aa469cdSdan ** in the code below. */ 37387aa469cdSdan assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new ); 37397aa469cdSdan 3740db04571cSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 3741db04571cSdan if( !abPK || abPK[i] ){ 3742db04571cSdan sqlite3_value *pVal; 37437aa469cdSdan (void)xValue(pIter, i, &pVal); 3744dd8a4af8Sdan if( pVal==0 ){ 3745dd8a4af8Sdan /* The value in the changeset was "undefined". This indicates a 3746dd8a4af8Sdan ** corrupt changeset blob. */ 3747e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 3748dd8a4af8Sdan }else{ 37497aa469cdSdan rc = sessionBindValue(pStmt, i+1, pVal); 3750db04571cSdan } 3751db04571cSdan } 3752dd8a4af8Sdan } 3753db04571cSdan return rc; 3754db04571cSdan } 3755db04571cSdan 3756db04571cSdan /* 3757296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function. 3758296c7658Sdan ** This function binds the primary key values from the change that changeset 3759296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table 3760296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row 3761296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error 3762296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an 37637aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned. 37647aa469cdSdan ** 37657aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset() 37667aa469cdSdan ** statement pSelect. If any other value is returned, the statement does 37677aa469cdSdan ** not require a reset(). 3768296c7658Sdan ** 3769296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the 3770db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or 3771db04571cSdan ** UPDATE, bind values from the old.* record. 3772296c7658Sdan */ 37730c698471Sdan static int sessionSeekToRow( 377437f133ecSdan sqlite3 *db, /* Database handle */ 377537f133ecSdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 377637f133ecSdan u8 *abPK, /* Primary key flags array */ 37770c698471Sdan sqlite3_stmt *pSelect /* SELECT statement from sessionSelectRow() */ 377837f133ecSdan ){ 37797aa469cdSdan int rc; /* Return code */ 3780296c7658Sdan int nCol; /* Number of columns in table */ 3781296c7658Sdan int op; /* Changset operation (SQLITE_UPDATE etc.) */ 3782296c7658Sdan const char *zDummy; /* Unused */ 378337f133ecSdan 3784b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 37857aa469cdSdan rc = sessionBindRow(pIter, 3786db04571cSdan op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old, 3787db04571cSdan nCol, abPK, pSelect 3788db04571cSdan ); 37890c698471Sdan 37900c698471Sdan if( rc==SQLITE_OK ){ 37910c698471Sdan rc = sqlite3_step(pSelect); 37920c698471Sdan if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect); 37930c698471Sdan } 37940c698471Sdan 37950c698471Sdan return rc; 37960c698471Sdan } 37970c698471Sdan 3798b880a7b1Sdan /* 3799b880a7b1Sdan ** This function is called from within sqlite3changset_apply_v2() when 3800b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution 3801b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE).. 3802b880a7b1Sdan ** It adds a conflict resolution record to the buffer in 3803b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller 3804b880a7b1Sdan ** of apply_v2() as the "rebase" buffer. 3805b880a7b1Sdan ** 3806b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise. 3807b880a7b1Sdan */ 3808a38e6c57Sdan static int sessionRebaseAdd( 3809b880a7b1Sdan SessionApplyCtx *p, /* Apply context */ 3810b880a7b1Sdan int eType, /* Conflict resolution (OMIT or REPLACE) */ 3811b880a7b1Sdan sqlite3_changeset_iter *pIter /* Iterator pointing at current change */ 3812a38e6c57Sdan ){ 3813a38e6c57Sdan int rc = SQLITE_OK; 3814a38e6c57Sdan int i; 3815a38e6c57Sdan int eOp = pIter->op; 3816a38e6c57Sdan if( p->bRebaseStarted==0 ){ 3817a38e6c57Sdan /* Append a table-header to the rebase buffer */ 3818a38e6c57Sdan const char *zTab = pIter->zTab; 3819a38e6c57Sdan sessionAppendByte(&p->rebase, 'T', &rc); 3820a38e6c57Sdan sessionAppendVarint(&p->rebase, p->nCol, &rc); 3821a38e6c57Sdan sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc); 3822a38e6c57Sdan sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc); 3823a38e6c57Sdan p->bRebaseStarted = 1; 3824a38e6c57Sdan } 3825a38e6c57Sdan 3826a38e6c57Sdan assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT ); 3827a38e6c57Sdan assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE ); 3828a38e6c57Sdan 3829f01d3a7eSdan sessionAppendByte(&p->rebase, 3830f01d3a7eSdan (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc 3831f01d3a7eSdan ); 3832f01d3a7eSdan sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc); 3833a38e6c57Sdan for(i=0; i<p->nCol; i++){ 3834a38e6c57Sdan sqlite3_value *pVal = 0; 3835f01d3a7eSdan if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){ 3836a38e6c57Sdan sqlite3changeset_old(pIter, i, &pVal); 3837a38e6c57Sdan }else{ 3838a38e6c57Sdan sqlite3changeset_new(pIter, i, &pVal); 3839a38e6c57Sdan } 3840a38e6c57Sdan sessionAppendValue(&p->rebase, pVal, &rc); 3841a38e6c57Sdan } 3842a38e6c57Sdan 3843a38e6c57Sdan return rc; 3844a38e6c57Sdan } 3845a38e6c57Sdan 3846296c7658Sdan /* 3847296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator 3848296c7658Sdan ** currently points to. 3849296c7658Sdan ** 3850296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT. 3851296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked 3852296c7658Sdan ** depends solely on eType, as follows: 3853296c7658Sdan ** 3854296c7658Sdan ** eType value Value passed to xConflict 3855296c7658Sdan ** ------------------------------------------------- 3856296c7658Sdan ** CHANGESET_DATA CHANGESET_NOTFOUND 3857296c7658Sdan ** CHANGESET_CONFLICT CHANGESET_CONSTRAINT 3858296c7658Sdan ** 3859296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing 3860296c7658Sdan ** record with the same primary key as the record about to be deleted, updated 3861296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict 3862296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict 3863296c7658Sdan ** handler invoked is as follows: 3864296c7658Sdan ** 3865296c7658Sdan ** eType value PK Record found? Value passed to xConflict 3866296c7658Sdan ** ---------------------------------------------------------------- 3867296c7658Sdan ** CHANGESET_DATA Yes CHANGESET_DATA 3868296c7658Sdan ** CHANGESET_DATA No CHANGESET_NOTFOUND 3869296c7658Sdan ** CHANGESET_CONFLICT Yes CHANGESET_CONFLICT 3870296c7658Sdan ** CHANGESET_CONFLICT No CHANGESET_CONSTRAINT 3871296c7658Sdan ** 3872296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and 3873296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace 3874296c7658Sdan ** is set to non-zero before returning SQLITE_OK. 3875296c7658Sdan ** 3876296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is 3877296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value, 3878296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT, 3879296c7658Sdan ** this function returns SQLITE_OK. 3880296c7658Sdan */ 38810c698471Sdan static int sessionConflictHandler( 3882296c7658Sdan int eType, /* Either CHANGESET_DATA or CONFLICT */ 3883296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 38840c698471Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 38850c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter*), 3886296c7658Sdan void *pCtx, /* First argument for conflict handler */ 3887296c7658Sdan int *pbReplace /* OUT: Set to true if PK row is found */ 38880c698471Sdan ){ 388974f598b6Smistachkin int res = 0; /* Value returned by conflict handler */ 38900c698471Sdan int rc; 38910c698471Sdan int nCol; 38920c698471Sdan int op; 38930c698471Sdan const char *zDummy; 38940c698471Sdan 3895b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 38960c698471Sdan 38970c698471Sdan assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA ); 38980c698471Sdan assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT ); 38990c698471Sdan assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND ); 390037f133ecSdan 390137f133ecSdan /* Bind the new.* PRIMARY KEY values to the SELECT statement. */ 39020c698471Sdan if( pbReplace ){ 39030c698471Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 39040c698471Sdan }else{ 3905db04571cSdan rc = SQLITE_OK; 39060c698471Sdan } 39070c698471Sdan 39080c698471Sdan if( rc==SQLITE_ROW ){ 39090c698471Sdan /* There exists another row with the new.* primary key. */ 39100c698471Sdan pIter->pConflict = p->pSelect; 39110c698471Sdan res = xConflict(pCtx, eType, pIter); 39120c698471Sdan pIter->pConflict = 0; 39130c698471Sdan rc = sqlite3_reset(p->pSelect); 3914db04571cSdan }else if( rc==SQLITE_OK ){ 3915d9151526Sdan if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){ 3916d9151526Sdan /* Instead of invoking the conflict handler, append the change blob 3917d9151526Sdan ** to the SessionApplyCtx.constraints buffer. */ 3918d9151526Sdan u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent]; 3919d9151526Sdan int nBlob = pIter->in.iNext - pIter->in.iCurrent; 3920d9151526Sdan sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc); 3921a38e6c57Sdan return SQLITE_OK; 3922d9151526Sdan }else{ 39230c698471Sdan /* No other row with the new.* primary key. */ 39240c698471Sdan res = xConflict(pCtx, eType+1, pIter); 39250c698471Sdan if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE; 392637f133ecSdan } 3927d9151526Sdan } 392837f133ecSdan 392937f133ecSdan if( rc==SQLITE_OK ){ 39300c698471Sdan switch( res ){ 39310c698471Sdan case SQLITE_CHANGESET_REPLACE: 3932f51e5f6cSdan assert( pbReplace ); 3933f51e5f6cSdan *pbReplace = 1; 39340c698471Sdan break; 39350c698471Sdan 39360c698471Sdan case SQLITE_CHANGESET_OMIT: 39370c698471Sdan break; 39380c698471Sdan 39390c698471Sdan case SQLITE_CHANGESET_ABORT: 39400c698471Sdan rc = SQLITE_ABORT; 39410c698471Sdan break; 39420c698471Sdan 39430c698471Sdan default: 39440c698471Sdan rc = SQLITE_MISUSE; 39450c698471Sdan break; 39460c698471Sdan } 3947a38e6c57Sdan if( rc==SQLITE_OK ){ 3948a38e6c57Sdan rc = sessionRebaseAdd(p, res, pIter); 3949a38e6c57Sdan } 39500c698471Sdan } 39510c698471Sdan 39520c698471Sdan return rc; 39530c698471Sdan } 39540c698471Sdan 3955296c7658Sdan /* 3956296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument 3957296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke 3958296c7658Sdan ** the conflict handler callback. 3959296c7658Sdan ** 3960296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If 3961296c7658Sdan ** one is encountered, update or delete the row with the matching primary key 3962296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs, 3963296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry 3964296c7658Sdan ** to true before returning. In this case the caller will invoke this function 3965296c7658Sdan ** again, this time with pbRetry set to NULL. 3966296c7658Sdan ** 3967296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is 3968296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead. 3969296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such 3970296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true 3971296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting 3972296c7658Sdan ** row before invoking this function again, this time with pbReplace set 3973296c7658Sdan ** to NULL. 3974296c7658Sdan ** 3975296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function 3976296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is 3977296c7658Sdan ** returned. 3978296c7658Sdan */ 39790c698471Sdan static int sessionApplyOneOp( 3980296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3981296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 39820c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter *), 3983296c7658Sdan void *pCtx, /* First argument for the conflict handler */ 3984296c7658Sdan int *pbReplace, /* OUT: True to remove PK row and retry */ 3985296c7658Sdan int *pbRetry /* OUT: True to retry. */ 39860c698471Sdan ){ 39870c698471Sdan const char *zDummy; 39880c698471Sdan int op; 39890c698471Sdan int nCol; 39900c698471Sdan int rc = SQLITE_OK; 39910c698471Sdan 39920c698471Sdan assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect ); 39930c698471Sdan assert( p->azCol && p->abPK ); 39940c698471Sdan assert( !pbReplace || *pbReplace==0 ); 39950c698471Sdan 3996b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 39970c698471Sdan 39980c698471Sdan if( op==SQLITE_DELETE ){ 39990c698471Sdan 400073b3c055Sdan /* Bind values to the DELETE statement. If conflict handling is required, 400173b3c055Sdan ** bind values for all columns and set bound variable (nCol+1) to true. 400273b3c055Sdan ** Or, if conflict handling is not required, bind just the PK column 400373b3c055Sdan ** values and, if it exists, set (nCol+1) to false. Conflict handling 400473b3c055Sdan ** is not required if: 400573b3c055Sdan ** 400673b3c055Sdan ** * this is a patchset, or 400773b3c055Sdan ** * (pbRetry==0), or 400873b3c055Sdan ** * all columns of the table are PK columns (in this case there is 400973b3c055Sdan ** no (nCol+1) variable to bind to). 401073b3c055Sdan */ 401173b3c055Sdan u8 *abPK = (pIter->bPatchset ? p->abPK : 0); 401273b3c055Sdan rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete); 40137cf7df7dSdan if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){ 401473b3c055Sdan rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK)); 40157cf7df7dSdan } 40160c698471Sdan if( rc!=SQLITE_OK ) return rc; 40170c698471Sdan 40180c698471Sdan sqlite3_step(p->pDelete); 40190c698471Sdan rc = sqlite3_reset(p->pDelete); 40200c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 40210c698471Sdan rc = sessionConflictHandler( 40220c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 40230c698471Sdan ); 402435e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 40250c698471Sdan rc = sessionConflictHandler( 40260c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 40270c698471Sdan ); 40280c698471Sdan } 40290c698471Sdan 40300c698471Sdan }else if( op==SQLITE_UPDATE ){ 40310c698471Sdan int i; 40320c698471Sdan 40330c698471Sdan /* Bind values to the UPDATE statement. */ 40340c698471Sdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 40357aa469cdSdan sqlite3_value *pOld = sessionChangesetOld(pIter, i); 40367aa469cdSdan sqlite3_value *pNew = sessionChangesetNew(pIter, i); 40377aa469cdSdan 40380c698471Sdan sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew); 40397aa469cdSdan if( pOld ){ 40407aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+1, pOld); 40417aa469cdSdan } 40427aa469cdSdan if( rc==SQLITE_OK && pNew ){ 40437aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+3, pNew); 40440c698471Sdan } 40450c698471Sdan } 404673b3c055Sdan if( rc==SQLITE_OK ){ 404773b3c055Sdan sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset); 404873b3c055Sdan } 40490c698471Sdan if( rc!=SQLITE_OK ) return rc; 40500c698471Sdan 40510c698471Sdan /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict, 40520c698471Sdan ** the result will be SQLITE_OK with 0 rows modified. */ 40530c698471Sdan sqlite3_step(p->pUpdate); 40540c698471Sdan rc = sqlite3_reset(p->pUpdate); 40550c698471Sdan 40560c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 40570c698471Sdan /* A NOTFOUND or DATA error. Search the table to see if it contains 40580c698471Sdan ** a row with a matching primary key. If so, this is a DATA conflict. 40590c698471Sdan ** Otherwise, if there is no primary key match, it is a NOTFOUND. */ 40600c698471Sdan 40610c698471Sdan rc = sessionConflictHandler( 40620c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 40630c698471Sdan ); 40640c698471Sdan 406535e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 4066db04571cSdan /* This is always a CONSTRAINT conflict. */ 4067db04571cSdan rc = sessionConflictHandler( 4068db04571cSdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 40690c698471Sdan ); 40700c698471Sdan } 40710c698471Sdan 40720c698471Sdan }else{ 40730c698471Sdan assert( op==SQLITE_INSERT ); 4074d1cccf19Sdan if( p->bStat1 ){ 4075d1cccf19Sdan /* Check if there is a conflicting row. For sqlite_stat1, this needs 4076d1cccf19Sdan ** to be done using a SELECT, as there is no PRIMARY KEY in the 4077d1cccf19Sdan ** database schema to throw an exception if a duplicate is inserted. */ 4078d1cccf19Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 4079d1cccf19Sdan if( rc==SQLITE_ROW ){ 4080d1cccf19Sdan rc = SQLITE_CONSTRAINT; 4081d1cccf19Sdan sqlite3_reset(p->pSelect); 4082d1cccf19Sdan } 4083d1cccf19Sdan } 4084d1cccf19Sdan 4085d1cccf19Sdan if( rc==SQLITE_OK ){ 40867aa469cdSdan rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert); 40870c698471Sdan if( rc!=SQLITE_OK ) return rc; 40880c698471Sdan 40890c698471Sdan sqlite3_step(p->pInsert); 40900c698471Sdan rc = sqlite3_reset(p->pInsert); 4091d1cccf19Sdan } 4092d1cccf19Sdan 409335e2858eSdan if( (rc&0xff)==SQLITE_CONSTRAINT ){ 40940c698471Sdan rc = sessionConflictHandler( 40950c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace 40960c698471Sdan ); 409737f133ecSdan } 409837f133ecSdan } 409937f133ecSdan 410037f133ecSdan return rc; 410137f133ecSdan } 410237f133ecSdan 41035f5663dcSdan /* 41045f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument 41055f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke 41065f5663dcSdan ** the conflict handler callback. 41075f5663dcSdan ** 41085f5663dcSdan ** The difference between this function and sessionApplyOne() is that this 41095f5663dcSdan ** function handles the case where the conflict-handler is invoked and 41105f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be 41115f5663dcSdan ** retried in some manner. 41125f5663dcSdan */ 4113d9151526Sdan static int sessionApplyOneWithRetry( 4114d9151526Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4115d9151526Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator to read change from */ 4116d9151526Sdan SessionApplyCtx *pApply, /* Apply context */ 4117d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4118d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4119d9151526Sdan ){ 4120d9151526Sdan int bReplace = 0; 4121d9151526Sdan int bRetry = 0; 4122d9151526Sdan int rc; 4123d9151526Sdan 4124d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry); 4125a38e6c57Sdan if( rc==SQLITE_OK ){ 41265f5663dcSdan /* If the bRetry flag is set, the change has not been applied due to an 41275f5663dcSdan ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and 41285f5663dcSdan ** a row with the correct PK is present in the db, but one or more other 41295f5663dcSdan ** fields do not contain the expected values) and the conflict handler 41305f5663dcSdan ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation, 41315f5663dcSdan ** but pass NULL as the final argument so that sessionApplyOneOp() ignores 41325f5663dcSdan ** the SQLITE_CHANGESET_DATA problem. */ 41335f5663dcSdan if( bRetry ){ 41345f5663dcSdan assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE ); 41355f5663dcSdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4136d9151526Sdan } 4137d9151526Sdan 41385f5663dcSdan /* If the bReplace flag is set, the change is an INSERT that has not 41395f5663dcSdan ** been performed because the database already contains a row with the 41405f5663dcSdan ** specified primary key and the conflict handler returned 41415f5663dcSdan ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row 41425f5663dcSdan ** before reattempting the INSERT. */ 41435f5663dcSdan else if( bReplace ){ 4144d9151526Sdan assert( pIter->op==SQLITE_INSERT ); 4145d9151526Sdan rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0); 4146d9151526Sdan if( rc==SQLITE_OK ){ 4147d9151526Sdan rc = sessionBindRow(pIter, 4148d9151526Sdan sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete); 4149d9151526Sdan sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1); 4150d9151526Sdan } 4151d9151526Sdan if( rc==SQLITE_OK ){ 4152d9151526Sdan sqlite3_step(pApply->pDelete); 4153d9151526Sdan rc = sqlite3_reset(pApply->pDelete); 4154d9151526Sdan } 4155d9151526Sdan if( rc==SQLITE_OK ){ 4156d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4157d9151526Sdan } 4158d9151526Sdan if( rc==SQLITE_OK ){ 4159d9151526Sdan rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0); 4160d9151526Sdan } 4161d9151526Sdan } 4162a38e6c57Sdan } 4163d9151526Sdan 4164d9151526Sdan return rc; 4165d9151526Sdan } 4166d9151526Sdan 4167d9151526Sdan /* 4168d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer. 4169d9151526Sdan */ 4170d9151526Sdan static int sessionRetryConstraints( 4171d9151526Sdan sqlite3 *db, 4172d9151526Sdan int bPatchset, 4173d9151526Sdan const char *zTab, 4174d9151526Sdan SessionApplyCtx *pApply, 4175d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4176d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4177d9151526Sdan ){ 4178d9151526Sdan int rc = SQLITE_OK; 4179d9151526Sdan 4180d9151526Sdan while( pApply->constraints.nBuf ){ 4181d9151526Sdan sqlite3_changeset_iter *pIter2 = 0; 4182d9151526Sdan SessionBuffer cons = pApply->constraints; 4183d9151526Sdan memset(&pApply->constraints, 0, sizeof(SessionBuffer)); 4184d9151526Sdan 4185d9151526Sdan rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf); 4186d9151526Sdan if( rc==SQLITE_OK ){ 4187d9151526Sdan int nByte = 2*pApply->nCol*sizeof(sqlite3_value*); 4188d9151526Sdan int rc2; 4189d9151526Sdan pIter2->bPatchset = bPatchset; 4190d9151526Sdan pIter2->zTab = (char*)zTab; 4191d9151526Sdan pIter2->nCol = pApply->nCol; 4192d9151526Sdan pIter2->abPK = pApply->abPK; 4193d9151526Sdan sessionBufferGrow(&pIter2->tblhdr, nByte, &rc); 4194d9151526Sdan pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf; 4195d9151526Sdan if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte); 4196d9151526Sdan 4197d9151526Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){ 4198d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx); 4199d9151526Sdan } 4200d9151526Sdan 4201d9151526Sdan rc2 = sqlite3changeset_finalize(pIter2); 42027e0765a9Sdrh if( rc==SQLITE_OK ) rc = rc2; 4203d9151526Sdan } 4204d9151526Sdan assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 ); 4205d9151526Sdan 4206d9151526Sdan sqlite3_free(cons.aBuf); 4207d9151526Sdan if( rc!=SQLITE_OK ) break; 4208d9151526Sdan if( pApply->constraints.nBuf>=cons.nBuf ){ 4209d9151526Sdan /* No progress was made on the last round. */ 4210d9151526Sdan pApply->bDeferConstraints = 0; 4211d9151526Sdan } 4212d9151526Sdan } 4213d9151526Sdan 4214d9151526Sdan return rc; 4215d9151526Sdan } 4216d9151526Sdan 4217296c7658Sdan /* 42184757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but 42194757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the 42204757c658Sdan ** changeset to the main database attached to handle "db". The supplied 42214757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered 42224757c658Sdan ** while applying the change. 4223296c7658Sdan */ 42244757c658Sdan static int sessionChangesetApply( 4225296c7658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 42264757c658Sdan sqlite3_changeset_iter *pIter, /* Changeset to apply */ 422740368988Sdan int(*xFilter)( 422840368988Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 422940368988Sdan const char *zTab /* Table name */ 423040368988Sdan ), 4231d5f0767cSdan int(*xConflict)( 4232d5f0767cSdan void *pCtx, /* Copy of fifth arg to _apply() */ 4233d5f0767cSdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4234d5f0767cSdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4235d5f0767cSdan ), 4236a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4237a38e6c57Sdan void **ppRebase, int *pnRebase /* OUT: Rebase information */ 4238d5f0767cSdan ){ 4239ca62ad57Sdan int schemaMismatch = 0; 4240296c7658Sdan int rc; /* Return code */ 4241d5f0767cSdan const char *zTab = 0; /* Name of current table */ 4242cfdbde21Sdrh int nTab = 0; /* Result of sqlite3Strlen30(zTab) */ 4243296c7658Sdan SessionApplyCtx sApply; /* changeset_apply() context object */ 42445f5663dcSdan int bPatchset; 4245d5f0767cSdan 4246082c96dfSdan assert( xConflict!=0 ); 4247082c96dfSdan 4248d9151526Sdan pIter->in.bNoDiscard = 1; 42490c698471Sdan memset(&sApply, 0, sizeof(sApply)); 42504c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 42510c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0); 4252cb3e4b79Sdan if( rc==SQLITE_OK ){ 4253cb3e4b79Sdan rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0); 4254cb3e4b79Sdan } 42550c698471Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){ 42560c698471Sdan int nCol; 4257d5f0767cSdan int op; 42580c698471Sdan const char *zNew; 4259ca62ad57Sdan 4260b4480e94Sdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0); 4261d5f0767cSdan 42620c698471Sdan if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){ 4263ca62ad57Sdan u8 *abPK; 4264ca62ad57Sdan 4265d9151526Sdan rc = sessionRetryConstraints( 4266d9151526Sdan db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx 4267d9151526Sdan ); 4268d9151526Sdan if( rc!=SQLITE_OK ) break; 4269d9151526Sdan 4270cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 42710c698471Sdan sqlite3_finalize(sApply.pDelete); 42720c698471Sdan sqlite3_finalize(sApply.pUpdate); 42730c698471Sdan sqlite3_finalize(sApply.pInsert); 42740c698471Sdan sqlite3_finalize(sApply.pSelect); 42750c698471Sdan sApply.db = db; 4276f1b40e83Sdan sApply.pDelete = 0; 4277f1b40e83Sdan sApply.pUpdate = 0; 4278f1b40e83Sdan sApply.pInsert = 0; 4279f1b40e83Sdan sApply.pSelect = 0; 4280f1b40e83Sdan sApply.nCol = 0; 4281f1b40e83Sdan sApply.azCol = 0; 4282f1b40e83Sdan sApply.abPK = 0; 4283f1b40e83Sdan sApply.bStat1 = 0; 4284d9151526Sdan sApply.bDeferConstraints = 1; 4285a38e6c57Sdan sApply.bRebaseStarted = 0; 4286f1b40e83Sdan memset(&sApply.constraints, 0, sizeof(SessionBuffer)); 428737f133ecSdan 428840368988Sdan /* If an xFilter() callback was specified, invoke it now. If the 428940368988Sdan ** xFilter callback returns zero, skip this table. If it returns 429040368988Sdan ** non-zero, proceed. */ 429140368988Sdan schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew))); 429240368988Sdan if( schemaMismatch ){ 429340368988Sdan zTab = sqlite3_mprintf("%s", zNew); 4294f05ac112Sdan if( zTab==0 ){ 4295f05ac112Sdan rc = SQLITE_NOMEM; 4296f05ac112Sdan break; 4297f05ac112Sdan } 42984f528042Sdan nTab = (int)strlen(zTab); 429940368988Sdan sApply.azCol = (const char **)zTab; 430040368988Sdan }else{ 4301ff677b20Sdan int nMinCol = 0; 4302ff677b20Sdan int i; 4303ff677b20Sdan 4304ca62ad57Sdan sqlite3changeset_pk(pIter, &abPK, 0); 4305296c7658Sdan rc = sessionTableInfo( 4306ca62ad57Sdan db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK 4307ca62ad57Sdan ); 4308ca62ad57Sdan if( rc!=SQLITE_OK ) break; 4309ff677b20Sdan for(i=0; i<sApply.nCol; i++){ 4310ff677b20Sdan if( sApply.abPK[i] ) nMinCol = i+1; 4311ff677b20Sdan } 43120c698471Sdan 4313ca62ad57Sdan if( sApply.nCol==0 ){ 4314ca62ad57Sdan schemaMismatch = 1; 4315ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4316ca62ad57Sdan "sqlite3changeset_apply(): no such table: %s", zTab 4317ca62ad57Sdan ); 4318ca62ad57Sdan } 4319ff677b20Sdan else if( sApply.nCol<nCol ){ 4320ca62ad57Sdan schemaMismatch = 1; 4321ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4322ff677b20Sdan "sqlite3changeset_apply(): table %s has %d columns, " 4323ff677b20Sdan "expected %d or more", 4324ca62ad57Sdan zTab, sApply.nCol, nCol 4325ca62ad57Sdan ); 4326ca62ad57Sdan } 4327ff677b20Sdan else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){ 4328ca62ad57Sdan schemaMismatch = 1; 432940368988Sdan sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): " 433040368988Sdan "primary key mismatch for table %s", zTab 4331ca62ad57Sdan ); 4332ca62ad57Sdan } 4333ff677b20Sdan else{ 4334ff677b20Sdan sApply.nCol = nCol; 43353739f298Sdan if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){ 43363739f298Sdan if( (rc = sessionStat1Sql(db, &sApply) ) ){ 43373739f298Sdan break; 43383739f298Sdan } 4339d1cccf19Sdan sApply.bStat1 = 1; 4340d1cccf19Sdan }else{ 4341ff677b20Sdan if((rc = sessionSelectRow(db, zTab, &sApply)) 43420c698471Sdan || (rc = sessionUpdateRow(db, zTab, &sApply)) 43430c698471Sdan || (rc = sessionDeleteRow(db, zTab, &sApply)) 43440c698471Sdan || (rc = sessionInsertRow(db, zTab, &sApply)) 434537f133ecSdan ){ 434637f133ecSdan break; 434737f133ecSdan } 4348d1cccf19Sdan sApply.bStat1 = 0; 4349d1cccf19Sdan } 4350ff677b20Sdan } 4351cfdbde21Sdrh nTab = sqlite3Strlen30(zTab); 4352d5f0767cSdan } 435340368988Sdan } 4354d5f0767cSdan 4355ca62ad57Sdan /* If there is a schema mismatch on the current table, proceed to the 4356ca62ad57Sdan ** next change. A log message has already been issued. */ 4357ca62ad57Sdan if( schemaMismatch ) continue; 4358ca62ad57Sdan 4359d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx); 43600c698471Sdan } 43610c698471Sdan 43625f5663dcSdan bPatchset = pIter->bPatchset; 4363296c7658Sdan if( rc==SQLITE_OK ){ 4364296c7658Sdan rc = sqlite3changeset_finalize(pIter); 4365296c7658Sdan }else{ 4366296c7658Sdan sqlite3changeset_finalize(pIter); 4367296c7658Sdan } 4368d5f0767cSdan 4369d5f0767cSdan if( rc==SQLITE_OK ){ 43705f5663dcSdan rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx); 43715f5663dcSdan } 43725f5663dcSdan 43735f5663dcSdan if( rc==SQLITE_OK ){ 437407001c45Sdrh int nFk, notUsed; 437507001c45Sdrh sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, ¬Used, 0); 437607001c45Sdrh if( nFk!=0 ){ 4377cb3e4b79Sdan int res = SQLITE_CHANGESET_ABORT; 4378cb3e4b79Sdan sqlite3_changeset_iter sIter; 4379cb3e4b79Sdan memset(&sIter, 0, sizeof(sIter)); 4380cb3e4b79Sdan sIter.nCol = nFk; 4381cb3e4b79Sdan res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter); 4382cb3e4b79Sdan if( res!=SQLITE_CHANGESET_OMIT ){ 4383cb3e4b79Sdan rc = SQLITE_CONSTRAINT; 4384cb3e4b79Sdan } 4385cb3e4b79Sdan } 4386cb3e4b79Sdan } 4387cb3e4b79Sdan sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0); 4388cb3e4b79Sdan 4389cb3e4b79Sdan if( rc==SQLITE_OK ){ 4390d5f0767cSdan rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4391d5f0767cSdan }else{ 4392d5f0767cSdan sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0); 4393d5f0767cSdan sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4394d5f0767cSdan } 4395d5f0767cSdan 439695ccb6dcSdan if( rc==SQLITE_OK && bPatchset==0 && ppRebase && pnRebase ){ 4397a38e6c57Sdan *ppRebase = (void*)sApply.rebase.aBuf; 4398a38e6c57Sdan *pnRebase = sApply.rebase.nBuf; 4399a38e6c57Sdan sApply.rebase.aBuf = 0; 4400a38e6c57Sdan } 44010c698471Sdan sqlite3_finalize(sApply.pInsert); 44020c698471Sdan sqlite3_finalize(sApply.pDelete); 44030c698471Sdan sqlite3_finalize(sApply.pUpdate); 44040c698471Sdan sqlite3_finalize(sApply.pSelect); 4405cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 4406d9151526Sdan sqlite3_free((char*)sApply.constraints.aBuf); 4407a38e6c57Sdan sqlite3_free((char*)sApply.rebase.aBuf); 44084c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 4409d5f0767cSdan return rc; 4410d5f0767cSdan } 441191ddd559Sdan 4412b880a7b1Sdan /* 4413b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main 4414b880a7b1Sdan ** database attached to handle "db". 4415b880a7b1Sdan */ 4416a38e6c57Sdan int sqlite3changeset_apply_v2( 4417a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4418a38e6c57Sdan int nChangeset, /* Size of changeset in bytes */ 4419a38e6c57Sdan void *pChangeset, /* Changeset blob */ 4420a38e6c57Sdan int(*xFilter)( 4421a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4422a38e6c57Sdan const char *zTab /* Table name */ 4423a38e6c57Sdan ), 4424a38e6c57Sdan int(*xConflict)( 4425a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4426a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4427a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4428a38e6c57Sdan ), 4429a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4430a38e6c57Sdan void **ppRebase, int *pnRebase 4431a38e6c57Sdan ){ 4432a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 4433a38e6c57Sdan int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset); 4434a38e6c57Sdan if( rc==SQLITE_OK ){ 4435a38e6c57Sdan rc = sessionChangesetApply( 4436a38e6c57Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase 4437a38e6c57Sdan ); 4438a38e6c57Sdan } 4439a38e6c57Sdan return rc; 4440a38e6c57Sdan } 4441a38e6c57Sdan 444277fc1d5bSdan /* 44434757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database 44444757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 44454757c658Sdan ** to resolve any conflicts encountered while applying the change. 44464757c658Sdan */ 44474757c658Sdan int sqlite3changeset_apply( 44484757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 44494757c658Sdan int nChangeset, /* Size of changeset in bytes */ 44504757c658Sdan void *pChangeset, /* Changeset blob */ 44514757c658Sdan int(*xFilter)( 44524757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 44534757c658Sdan const char *zTab /* Table name */ 44544757c658Sdan ), 44554757c658Sdan int(*xConflict)( 44564757c658Sdan void *pCtx, /* Copy of fifth arg to _apply() */ 44574757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 44584757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 44594757c658Sdan ), 44604757c658Sdan void *pCtx /* First argument passed to xConflict */ 44614757c658Sdan ){ 4462a38e6c57Sdan return sqlite3changeset_apply_v2( 4463a38e6c57Sdan db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0 4464a38e6c57Sdan ); 44654757c658Sdan } 44664757c658Sdan 44674757c658Sdan /* 44684757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database 44694757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 44704757c658Sdan ** to resolve any conflicts encountered while applying the change. 44714757c658Sdan */ 4472a38e6c57Sdan int sqlite3changeset_apply_v2_strm( 4473a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4474a38e6c57Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 4475a38e6c57Sdan void *pIn, /* First arg for xInput */ 4476a38e6c57Sdan int(*xFilter)( 4477a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4478a38e6c57Sdan const char *zTab /* Table name */ 4479a38e6c57Sdan ), 4480a38e6c57Sdan int(*xConflict)( 4481a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4482a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4483a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4484a38e6c57Sdan ), 4485a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4486a38e6c57Sdan void **ppRebase, int *pnRebase 4487a38e6c57Sdan ){ 4488a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 4489a38e6c57Sdan int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 4490a38e6c57Sdan if( rc==SQLITE_OK ){ 4491a38e6c57Sdan rc = sessionChangesetApply( 4492a38e6c57Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase 4493a38e6c57Sdan ); 4494a38e6c57Sdan } 4495a38e6c57Sdan return rc; 4496a38e6c57Sdan } 4497f1a08ad8Sdrh int sqlite3changeset_apply_strm( 44984757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 44994757c658Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 45004757c658Sdan void *pIn, /* First arg for xInput */ 45014757c658Sdan int(*xFilter)( 45024757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45034757c658Sdan const char *zTab /* Table name */ 45044757c658Sdan ), 45054757c658Sdan int(*xConflict)( 45064757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45074757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 45084757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 45094757c658Sdan ), 45104757c658Sdan void *pCtx /* First argument passed to xConflict */ 45114757c658Sdan ){ 4512a38e6c57Sdan return sqlite3changeset_apply_v2_strm( 4513a38e6c57Sdan db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0 4514a38e6c57Sdan ); 45154757c658Sdan } 45164757c658Sdan 45174757c658Sdan /* 45185898ad69Sdan ** sqlite3_changegroup handle. 45195898ad69Sdan */ 45205898ad69Sdan struct sqlite3_changegroup { 45215898ad69Sdan int rc; /* Error code */ 45225898ad69Sdan int bPatch; /* True to accumulate patchsets */ 45235898ad69Sdan SessionTable *pList; /* List of tables in current patch */ 45245898ad69Sdan }; 45255898ad69Sdan 45265898ad69Sdan /* 452777fc1d5bSdan ** This function is called to merge two changes to the same row together as 452877fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is 452977fc1d5bSdan ** allocated and a pointer to it stored in *ppNew. 453077fc1d5bSdan */ 45315d607a6eSdan static int sessionChangeMerge( 453277fc1d5bSdan SessionTable *pTab, /* Table structure */ 4533c0a499eaSdan int bRebase, /* True for a rebase hash-table */ 453464277f4aSdan int bPatchset, /* True for patchsets */ 453577fc1d5bSdan SessionChange *pExist, /* Existing change */ 453677fc1d5bSdan int op2, /* Second change operation */ 453777fc1d5bSdan int bIndirect, /* True if second change is indirect */ 453877fc1d5bSdan u8 *aRec, /* Second change record */ 453977fc1d5bSdan int nRec, /* Number of bytes in aRec */ 454077fc1d5bSdan SessionChange **ppNew /* OUT: Merged change */ 45415d607a6eSdan ){ 45425d607a6eSdan SessionChange *pNew = 0; 454324a0c453Sdan int rc = SQLITE_OK; 45445d607a6eSdan 45455d607a6eSdan if( !pExist ){ 4546cbf6d2d2Sdan pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec); 45475d607a6eSdan if( !pNew ){ 45485d607a6eSdan return SQLITE_NOMEM; 45495d607a6eSdan } 45505d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 4551798693b2Sdan pNew->op = op2; 45525d607a6eSdan pNew->bIndirect = bIndirect; 4553cbf6d2d2Sdan pNew->aRecord = (u8*)&pNew[1]; 455424a0c453Sdan if( bIndirect==0 || bRebase==0 ){ 455524a0c453Sdan pNew->nRecord = nRec; 4556cbf6d2d2Sdan memcpy(pNew->aRecord, aRec, nRec); 455724a0c453Sdan }else{ 455824a0c453Sdan int i; 455924a0c453Sdan u8 *pIn = aRec; 456024a0c453Sdan u8 *pOut = pNew->aRecord; 456124a0c453Sdan for(i=0; i<pTab->nCol; i++){ 456224a0c453Sdan int nIn = sessionSerialLen(pIn); 456324a0c453Sdan if( *pIn==0 ){ 456424a0c453Sdan *pOut++ = 0; 456524a0c453Sdan }else if( pTab->abPK[i]==0 ){ 456624a0c453Sdan *pOut++ = 0xFF; 456724a0c453Sdan }else{ 456824a0c453Sdan memcpy(pOut, pIn, nIn); 456924a0c453Sdan pOut += nIn; 457024a0c453Sdan } 457124a0c453Sdan pIn += nIn; 457224a0c453Sdan } 457324a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 457424a0c453Sdan } 4575c0a499eaSdan }else if( bRebase ){ 457624a0c453Sdan if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){ 457724a0c453Sdan *ppNew = pExist; 457824a0c453Sdan }else{ 457924a0c453Sdan int nByte = nRec + pExist->nRecord + sizeof(SessionChange); 458024a0c453Sdan pNew = (SessionChange*)sqlite3_malloc(nByte); 458124a0c453Sdan if( pNew==0 ){ 458224a0c453Sdan rc = SQLITE_NOMEM; 458324a0c453Sdan }else{ 458424a0c453Sdan int i; 458524a0c453Sdan u8 *a1 = pExist->aRecord; 458624a0c453Sdan u8 *a2 = aRec; 458724a0c453Sdan u8 *pOut; 458824a0c453Sdan 458924a0c453Sdan memset(pNew, 0, nByte); 459024a0c453Sdan pNew->bIndirect = bIndirect || pExist->bIndirect; 459124a0c453Sdan pNew->op = op2; 459224a0c453Sdan pOut = pNew->aRecord = (u8*)&pNew[1]; 459324a0c453Sdan 459424a0c453Sdan for(i=0; i<pTab->nCol; i++){ 459524a0c453Sdan int n1 = sessionSerialLen(a1); 459624a0c453Sdan int n2 = sessionSerialLen(a2); 4597f231e18cSdan if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){ 459824a0c453Sdan *pOut++ = 0xFF; 459924a0c453Sdan }else if( *a2==0 ){ 460024a0c453Sdan memcpy(pOut, a1, n1); 460124a0c453Sdan pOut += n1; 460224a0c453Sdan }else{ 460324a0c453Sdan memcpy(pOut, a2, n2); 460424a0c453Sdan pOut += n2; 460524a0c453Sdan } 460624a0c453Sdan a1 += n1; 460724a0c453Sdan a2 += n2; 460824a0c453Sdan } 460924a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 461024a0c453Sdan } 461124a0c453Sdan sqlite3_free(pExist); 461224a0c453Sdan } 46135d607a6eSdan }else{ 4614798693b2Sdan int op1 = pExist->op; 46155d607a6eSdan 46165d607a6eSdan /* 46175d607a6eSdan ** op1=INSERT, op2=INSERT -> Unsupported. Discard op2. 46185d607a6eSdan ** op1=INSERT, op2=UPDATE -> INSERT. 46195d607a6eSdan ** op1=INSERT, op2=DELETE -> (none) 46205d607a6eSdan ** 46215d607a6eSdan ** op1=UPDATE, op2=INSERT -> Unsupported. Discard op2. 46225d607a6eSdan ** op1=UPDATE, op2=UPDATE -> UPDATE. 46235d607a6eSdan ** op1=UPDATE, op2=DELETE -> DELETE. 46245d607a6eSdan ** 46255d607a6eSdan ** op1=DELETE, op2=INSERT -> UPDATE. 46265d607a6eSdan ** op1=DELETE, op2=UPDATE -> Unsupported. Discard op2. 46275d607a6eSdan ** op1=DELETE, op2=DELETE -> Unsupported. Discard op2. 46285d607a6eSdan */ 46295d607a6eSdan if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT) 46305d607a6eSdan || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT) 46315d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE) 46325d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_DELETE) 46335d607a6eSdan ){ 46345d607a6eSdan pNew = pExist; 46355d607a6eSdan }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){ 46365d607a6eSdan sqlite3_free(pExist); 46375d607a6eSdan assert( pNew==0 ); 46385d607a6eSdan }else{ 463964277f4aSdan u8 *aExist = pExist->aRecord; 46405d607a6eSdan int nByte; 46415d607a6eSdan u8 *aCsr; 46425d607a6eSdan 464364277f4aSdan /* Allocate a new SessionChange object. Ensure that the aRecord[] 464464277f4aSdan ** buffer of the new object is large enough to hold any record that 464564277f4aSdan ** may be generated by combining the input records. */ 46465d607a6eSdan nByte = sizeof(SessionChange) + pExist->nRecord + nRec; 46475d607a6eSdan pNew = (SessionChange *)sqlite3_malloc(nByte); 46485d607a6eSdan if( !pNew ){ 46491756ae10Sdan sqlite3_free(pExist); 46505d607a6eSdan return SQLITE_NOMEM; 46515d607a6eSdan } 46525d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 46535d607a6eSdan pNew->bIndirect = (bIndirect && pExist->bIndirect); 46545d607a6eSdan aCsr = pNew->aRecord = (u8 *)&pNew[1]; 46555d607a6eSdan 4656b08a1efaSdan if( op1==SQLITE_INSERT ){ /* INSERT + UPDATE */ 46575d607a6eSdan u8 *a1 = aRec; 4658b08a1efaSdan assert( op2==SQLITE_UPDATE ); 4659798693b2Sdan pNew->op = SQLITE_INSERT; 4660ef7a6304Sdan if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol); 466164277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1); 4662b08a1efaSdan }else if( op1==SQLITE_DELETE ){ /* DELETE + INSERT */ 4663b08a1efaSdan assert( op2==SQLITE_INSERT ); 4664798693b2Sdan pNew->op = SQLITE_UPDATE; 4665fa29ecc4Sdan if( bPatchset ){ 4666fa29ecc4Sdan memcpy(aCsr, aRec, nRec); 4667fa29ecc4Sdan aCsr += nRec; 4668fa29ecc4Sdan }else{ 466964277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){ 4670b08a1efaSdan sqlite3_free(pNew); 4671b08a1efaSdan pNew = 0; 46725d607a6eSdan } 4673fa29ecc4Sdan } 4674b08a1efaSdan }else if( op2==SQLITE_UPDATE ){ /* UPDATE + UPDATE */ 467564277f4aSdan u8 *a1 = aExist; 46765d607a6eSdan u8 *a2 = aRec; 4677cfec7eeeSdan assert( op1==SQLITE_UPDATE ); 467864277f4aSdan if( bPatchset==0 ){ 4679ef7a6304Sdan sessionSkipRecord(&a1, pTab->nCol); 4680ef7a6304Sdan sessionSkipRecord(&a2, pTab->nCol); 468164277f4aSdan } 4682798693b2Sdan pNew->op = SQLITE_UPDATE; 468364277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){ 46845d607a6eSdan sqlite3_free(pNew); 46855d607a6eSdan pNew = 0; 46865d607a6eSdan } 4687b08a1efaSdan }else{ /* UPDATE + DELETE */ 4688b08a1efaSdan assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE ); 4689798693b2Sdan pNew->op = SQLITE_DELETE; 469064277f4aSdan if( bPatchset ){ 469164277f4aSdan memcpy(aCsr, aRec, nRec); 469264277f4aSdan aCsr += nRec; 469364277f4aSdan }else{ 469464277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist); 469564277f4aSdan } 46965d607a6eSdan } 46975d607a6eSdan 46985d607a6eSdan if( pNew ){ 46994f528042Sdan pNew->nRecord = (int)(aCsr - pNew->aRecord); 47005d607a6eSdan } 47015d607a6eSdan sqlite3_free(pExist); 47025d607a6eSdan } 47035d607a6eSdan } 47045d607a6eSdan 47055d607a6eSdan *ppNew = pNew; 470624a0c453Sdan return rc; 47075d607a6eSdan } 47085d607a6eSdan 470977fc1d5bSdan /* 47105898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as 47115898ad69Sdan ** the first argument to the changegroup hash tables. 471277fc1d5bSdan */ 471316228167Sdan static int sessionChangesetToHash( 4714cbf6d2d2Sdan sqlite3_changeset_iter *pIter, /* Iterator to read from */ 4715c0a499eaSdan sqlite3_changegroup *pGrp, /* Changegroup object to add changeset to */ 4716c0a499eaSdan int bRebase /* True if hash table is for rebasing */ 47175d607a6eSdan ){ 47185d607a6eSdan u8 *aRec; 47195d607a6eSdan int nRec; 4720cbf6d2d2Sdan int rc = SQLITE_OK; 47215d607a6eSdan SessionTable *pTab = 0; 47225d607a6eSdan 4723c0a499eaSdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){ 47245d607a6eSdan const char *zNew; 47255d607a6eSdan int nCol; 47265d607a6eSdan int op; 47275d607a6eSdan int iHash; 47285d607a6eSdan int bIndirect; 47295d607a6eSdan SessionChange *pChange; 47305d607a6eSdan SessionChange *pExist = 0; 47315d607a6eSdan SessionChange **pp; 47325d607a6eSdan 47335898ad69Sdan if( pGrp->pList==0 ){ 47345898ad69Sdan pGrp->bPatch = pIter->bPatchset; 47355898ad69Sdan }else if( pIter->bPatchset!=pGrp->bPatch ){ 47365898ad69Sdan rc = SQLITE_ERROR; 47375898ad69Sdan break; 47385898ad69Sdan } 47395898ad69Sdan 47405d607a6eSdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect); 4741ef7a6304Sdan if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){ 47425d607a6eSdan /* Search the list for a matching table */ 47434f528042Sdan int nNew = (int)strlen(zNew); 4744f29123b5Sdan u8 *abPK; 4745f29123b5Sdan 4746f29123b5Sdan sqlite3changeset_pk(pIter, &abPK, 0); 47475898ad69Sdan for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){ 47485d607a6eSdan if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break; 47495d607a6eSdan } 47505d607a6eSdan if( !pTab ){ 47516c39e6a8Sdan SessionTable **ppTab; 47526c39e6a8Sdan 4753ef7a6304Sdan pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1); 4754f29123b5Sdan if( !pTab ){ 4755f29123b5Sdan rc = SQLITE_NOMEM; 4756f29123b5Sdan break; 4757f29123b5Sdan } 47585d607a6eSdan memset(pTab, 0, sizeof(SessionTable)); 4759f29123b5Sdan pTab->nCol = nCol; 4760ef7a6304Sdan pTab->abPK = (u8*)&pTab[1]; 4761ef7a6304Sdan memcpy(pTab->abPK, abPK, nCol); 4762ef7a6304Sdan pTab->zName = (char*)&pTab->abPK[nCol]; 4763ef7a6304Sdan memcpy(pTab->zName, zNew, nNew+1); 47646c39e6a8Sdan 47656c39e6a8Sdan /* The new object must be linked on to the end of the list, not 47666c39e6a8Sdan ** simply added to the start of it. This is to ensure that the 47676c39e6a8Sdan ** tables within the output of sqlite3changegroup_output() are in 47686c39e6a8Sdan ** the right order. */ 47696c39e6a8Sdan for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext); 47706c39e6a8Sdan *ppTab = pTab; 4771f29123b5Sdan }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){ 4772f29123b5Sdan rc = SQLITE_SCHEMA; 4773f29123b5Sdan break; 47745d607a6eSdan } 47755d607a6eSdan } 47765d607a6eSdan 4777cbf6d2d2Sdan if( sessionGrowHash(pIter->bPatchset, pTab) ){ 47781756ae10Sdan rc = SQLITE_NOMEM; 47791756ae10Sdan break; 47801756ae10Sdan } 478164277f4aSdan iHash = sessionChangeHash( 4782cbf6d2d2Sdan pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange 478364277f4aSdan ); 47845d607a6eSdan 47855d607a6eSdan /* Search for existing entry. If found, remove it from the hash table. 47865d607a6eSdan ** Code below may link it back in. 47875d607a6eSdan */ 47885d607a6eSdan for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){ 478964277f4aSdan int bPkOnly1 = 0; 479064277f4aSdan int bPkOnly2 = 0; 4791cbf6d2d2Sdan if( pIter->bPatchset ){ 479264277f4aSdan bPkOnly1 = (*pp)->op==SQLITE_DELETE; 479364277f4aSdan bPkOnly2 = op==SQLITE_DELETE; 479464277f4aSdan } 479564277f4aSdan if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){ 47965d607a6eSdan pExist = *pp; 47975d607a6eSdan *pp = (*pp)->pNext; 47985d607a6eSdan pTab->nEntry--; 47995d607a6eSdan break; 48005d607a6eSdan } 48015d607a6eSdan } 48025d607a6eSdan 4803c0a499eaSdan rc = sessionChangeMerge(pTab, bRebase, 4804cbf6d2d2Sdan pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange 480564277f4aSdan ); 48065d607a6eSdan if( rc ) break; 48075d607a6eSdan if( pChange ){ 48085d607a6eSdan pChange->pNext = pTab->apChange[iHash]; 48095d607a6eSdan pTab->apChange[iHash] = pChange; 48105d607a6eSdan pTab->nEntry++; 48115d607a6eSdan } 48125d607a6eSdan } 48135d607a6eSdan 4814cbf6d2d2Sdan if( rc==SQLITE_OK ) rc = pIter->rc; 48155d607a6eSdan return rc; 48165d607a6eSdan } 48175d607a6eSdan 48185d607a6eSdan /* 48195898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets) 48205898ad69Sdan ** added to the changegroup object passed as the first argument. 48215d607a6eSdan ** 48225898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the 48235898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming 48245898ad69Sdan ** interfaces. 48255d607a6eSdan ** 48265898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a 48275898ad69Sdan ** buffer containing the output changeset before this function returns. In 48285898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It 48295898ad69Sdan ** is the responsibility of the caller to free the output buffer using 48305898ad69Sdan ** sqlite3_free() when it is no longer required. 48315898ad69Sdan ** 48325898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite 48335898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut) 48345898ad69Sdan ** are both set to 0 before returning. 48355d607a6eSdan */ 48365898ad69Sdan static int sessionChangegroupOutput( 48375898ad69Sdan sqlite3_changegroup *pGrp, 4838cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4839cbf6d2d2Sdan void *pOut, 4840cbf6d2d2Sdan int *pnOut, 4841cbf6d2d2Sdan void **ppOut 48425d607a6eSdan ){ 48435898ad69Sdan int rc = SQLITE_OK; 4844e8fa8c96Sdan SessionBuffer buf = {0, 0, 0}; 48455898ad69Sdan SessionTable *pTab; 4846cbf6d2d2Sdan assert( xOutput==0 || (ppOut==0 && pnOut==0) ); 48475d607a6eSdan 48485d607a6eSdan /* Create the serialized output changeset based on the contents of the 48495898ad69Sdan ** hash tables attached to the SessionTable objects in list p->pList. 48505d607a6eSdan */ 48515898ad69Sdan for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 48525d607a6eSdan int i; 48535d607a6eSdan if( pTab->nEntry==0 ) continue; 48545d607a6eSdan 48555898ad69Sdan sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc); 48565d607a6eSdan for(i=0; i<pTab->nChange; i++){ 48575d607a6eSdan SessionChange *p; 48585d607a6eSdan for(p=pTab->apChange[i]; p; p=p->pNext){ 4859798693b2Sdan sessionAppendByte(&buf, p->op, &rc); 48605d607a6eSdan sessionAppendByte(&buf, p->bIndirect, &rc); 48615d607a6eSdan sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc); 48625d607a6eSdan } 48635d607a6eSdan } 4864cbf6d2d2Sdan 4865f1a08ad8Sdrh if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 4866cbf6d2d2Sdan rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4867cbf6d2d2Sdan buf.nBuf = 0; 4868cbf6d2d2Sdan } 48695d607a6eSdan } 48705d607a6eSdan 48715d607a6eSdan if( rc==SQLITE_OK ){ 4872cbf6d2d2Sdan if( xOutput ){ 4873cbf6d2d2Sdan if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4874cbf6d2d2Sdan }else{ 48755d607a6eSdan *ppOut = buf.aBuf; 48765d607a6eSdan *pnOut = buf.nBuf; 4877cbf6d2d2Sdan buf.aBuf = 0; 48785d607a6eSdan } 48795d607a6eSdan } 4880cbf6d2d2Sdan sqlite3_free(buf.aBuf); 48815d607a6eSdan 48825d607a6eSdan return rc; 48835d607a6eSdan } 48845d607a6eSdan 4885cbf6d2d2Sdan /* 48865898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup. 48875898ad69Sdan */ 48885898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){ 48895898ad69Sdan int rc = SQLITE_OK; /* Return code */ 48905898ad69Sdan sqlite3_changegroup *p; /* New object */ 48915898ad69Sdan p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup)); 48925898ad69Sdan if( p==0 ){ 48935898ad69Sdan rc = SQLITE_NOMEM; 48945898ad69Sdan }else{ 48955898ad69Sdan memset(p, 0, sizeof(sqlite3_changegroup)); 48965898ad69Sdan } 48975898ad69Sdan *pp = p; 48985898ad69Sdan return rc; 48995898ad69Sdan } 49005898ad69Sdan 49015898ad69Sdan /* 49025898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes, 49035898ad69Sdan ** to changeset-group p. 49045898ad69Sdan */ 49055898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){ 49065898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 49075898ad69Sdan int rc; /* Return code */ 49085898ad69Sdan 49095898ad69Sdan rc = sqlite3changeset_start(&pIter, nData, pData); 49105898ad69Sdan if( rc==SQLITE_OK ){ 4911c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 49125898ad69Sdan } 49135898ad69Sdan sqlite3changeset_finalize(pIter); 49145898ad69Sdan return rc; 49155898ad69Sdan } 49165898ad69Sdan 49175898ad69Sdan /* 49185898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation 49195898ad69Sdan ** of all changesets added to the group so far. 49205898ad69Sdan */ 49215898ad69Sdan int sqlite3changegroup_output( 49225898ad69Sdan sqlite3_changegroup *pGrp, 49235898ad69Sdan int *pnData, 49245898ad69Sdan void **ppData 49255898ad69Sdan ){ 49265898ad69Sdan return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData); 49275898ad69Sdan } 49285898ad69Sdan 49295898ad69Sdan /* 49305898ad69Sdan ** Streaming versions of changegroup_add(). 49315898ad69Sdan */ 49325898ad69Sdan int sqlite3changegroup_add_strm( 49335898ad69Sdan sqlite3_changegroup *pGrp, 49345898ad69Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 49355898ad69Sdan void *pIn 49365898ad69Sdan ){ 49375898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 49385898ad69Sdan int rc; /* Return code */ 49395898ad69Sdan 49405898ad69Sdan rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 49415898ad69Sdan if( rc==SQLITE_OK ){ 4942c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 49435898ad69Sdan } 49445898ad69Sdan sqlite3changeset_finalize(pIter); 49455898ad69Sdan return rc; 49465898ad69Sdan } 49475898ad69Sdan 49485898ad69Sdan /* 49495898ad69Sdan ** Streaming versions of changegroup_output(). 49505898ad69Sdan */ 49515898ad69Sdan int sqlite3changegroup_output_strm( 49525898ad69Sdan sqlite3_changegroup *pGrp, 49535898ad69Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 49545898ad69Sdan void *pOut 49555898ad69Sdan ){ 49565898ad69Sdan return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0); 49575898ad69Sdan } 49585898ad69Sdan 49595898ad69Sdan /* 49605898ad69Sdan ** Delete a changegroup object. 49615898ad69Sdan */ 49625898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){ 49635898ad69Sdan if( pGrp ){ 49645898ad69Sdan sessionDeleteTable(pGrp->pList); 49655898ad69Sdan sqlite3_free(pGrp); 49665898ad69Sdan } 49675898ad69Sdan } 49685898ad69Sdan 49695898ad69Sdan /* 4970cbf6d2d2Sdan ** Combine two changesets together. 4971cbf6d2d2Sdan */ 4972cbf6d2d2Sdan int sqlite3changeset_concat( 4973cbf6d2d2Sdan int nLeft, /* Number of bytes in lhs input */ 4974cbf6d2d2Sdan void *pLeft, /* Lhs input changeset */ 4975cbf6d2d2Sdan int nRight /* Number of bytes in rhs input */, 4976cbf6d2d2Sdan void *pRight, /* Rhs input changeset */ 4977cbf6d2d2Sdan int *pnOut, /* OUT: Number of bytes in output changeset */ 4978cbf6d2d2Sdan void **ppOut /* OUT: changeset (left <concat> right) */ 4979cbf6d2d2Sdan ){ 49805898ad69Sdan sqlite3_changegroup *pGrp; 4981cbf6d2d2Sdan int rc; 4982cbf6d2d2Sdan 49835898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 4984cbf6d2d2Sdan if( rc==SQLITE_OK ){ 49855898ad69Sdan rc = sqlite3changegroup_add(pGrp, nLeft, pLeft); 4986cbf6d2d2Sdan } 4987cbf6d2d2Sdan if( rc==SQLITE_OK ){ 49885898ad69Sdan rc = sqlite3changegroup_add(pGrp, nRight, pRight); 4989cbf6d2d2Sdan } 49905898ad69Sdan if( rc==SQLITE_OK ){ 49915898ad69Sdan rc = sqlite3changegroup_output(pGrp, pnOut, ppOut); 49925898ad69Sdan } 49935898ad69Sdan sqlite3changegroup_delete(pGrp); 4994cbf6d2d2Sdan 4995cbf6d2d2Sdan return rc; 4996cbf6d2d2Sdan } 4997cbf6d2d2Sdan 4998cbf6d2d2Sdan /* 4999cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat(). 5000cbf6d2d2Sdan */ 5001f1a08ad8Sdrh int sqlite3changeset_concat_strm( 5002cbf6d2d2Sdan int (*xInputA)(void *pIn, void *pData, int *pnData), 5003cbf6d2d2Sdan void *pInA, 5004cbf6d2d2Sdan int (*xInputB)(void *pIn, void *pData, int *pnData), 5005cbf6d2d2Sdan void *pInB, 5006cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 5007cbf6d2d2Sdan void *pOut 5008cbf6d2d2Sdan ){ 50095898ad69Sdan sqlite3_changegroup *pGrp; 5010cbf6d2d2Sdan int rc; 5011cbf6d2d2Sdan 50125898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 5013cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50145898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA); 5015cbf6d2d2Sdan } 5016cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50175898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB); 5018cbf6d2d2Sdan } 50195898ad69Sdan if( rc==SQLITE_OK ){ 50205898ad69Sdan rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut); 50215898ad69Sdan } 50225898ad69Sdan sqlite3changegroup_delete(pGrp); 5023cbf6d2d2Sdan 5024cbf6d2d2Sdan return rc; 5025cbf6d2d2Sdan } 5026cbf6d2d2Sdan 50273fa5463cSdan /* 50283fa5463cSdan ** Changeset rebaser handle. 50293fa5463cSdan */ 5030c0a499eaSdan struct sqlite3_rebaser { 5031c0a499eaSdan sqlite3_changegroup grp; /* Hash table */ 5032c0a499eaSdan }; 5033c0a499eaSdan 5034c0a499eaSdan /* 5035c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol 5036c0a499eaSdan ** fields in size. This function appends an nCol sessions module 5037b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for 5038b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[]. 5039c0a499eaSdan */ 5040c0a499eaSdan static void sessionAppendRecordMerge( 5041b880a7b1Sdan SessionBuffer *pBuf, /* Buffer to append to */ 5042b880a7b1Sdan int nCol, /* Number of columns in each record */ 5043b880a7b1Sdan u8 *a1, int n1, /* Record 1 */ 5044b880a7b1Sdan u8 *a2, int n2, /* Record 2 */ 5045b880a7b1Sdan int *pRc /* IN/OUT: error code */ 5046c0a499eaSdan ){ 5047c0a499eaSdan sessionBufferGrow(pBuf, n1+n2, pRc); 5048c0a499eaSdan if( *pRc==SQLITE_OK ){ 5049c0a499eaSdan int i; 5050c0a499eaSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5051c0a499eaSdan for(i=0; i<nCol; i++){ 5052c0a499eaSdan int nn1 = sessionSerialLen(a1); 5053c0a499eaSdan int nn2 = sessionSerialLen(a2); 505424a0c453Sdan if( *a1==0 || *a1==0xFF ){ 5055c0a499eaSdan memcpy(pOut, a2, nn2); 5056c0a499eaSdan pOut += nn2; 5057c0a499eaSdan }else{ 5058c0a499eaSdan memcpy(pOut, a1, nn1); 5059c0a499eaSdan pOut += nn1; 5060c0a499eaSdan } 5061f1b40e83Sdan a1 += nn1; 5062f1b40e83Sdan a2 += nn2; 5063c0a499eaSdan } 5064f1b40e83Sdan 5065f1b40e83Sdan pBuf->nBuf = pOut-pBuf->aBuf; 5066f1b40e83Sdan assert( pBuf->nBuf<=pBuf->nAlloc ); 5067c0a499eaSdan } 5068c0a499eaSdan } 5069c0a499eaSdan 50703fa5463cSdan /* 50713fa5463cSdan ** This function is called when rebasing a local UPDATE change against one 50723fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current 50733fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single 50743fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer 50753fa5463cSdan ** pBuf. 50763fa5463cSdan ** 50773fa5463cSdan ** Rebasing the UPDATE involves: 50783fa5463cSdan ** 50793fa5463cSdan ** * Removing any changes to fields for which the corresponding field 50803fa5463cSdan ** in the rebase buffer is set to "replaced" (type 0xFF). If this 50813fa5463cSdan ** means the UPDATE change updates no fields, nothing is appended 50823fa5463cSdan ** to the output buffer. 50833fa5463cSdan ** 50843fa5463cSdan ** * For each field modified by the local change for which the 50853fa5463cSdan ** corresponding field in the rebase buffer is not "undefined" (0x00) 50863fa5463cSdan ** or "replaced" (0xFF), the old.* value is replaced by the value 50873fa5463cSdan ** in the rebase buffer. 50883fa5463cSdan */ 5089f01d3a7eSdan static void sessionAppendPartialUpdate( 509024a0c453Sdan SessionBuffer *pBuf, /* Append record here */ 509124a0c453Sdan sqlite3_changeset_iter *pIter, /* Iterator pointed at local change */ 509224a0c453Sdan u8 *aRec, int nRec, /* Local change */ 509324a0c453Sdan u8 *aChange, int nChange, /* Record to rebase against */ 509424a0c453Sdan int *pRc /* IN/OUT: Return Code */ 5095f01d3a7eSdan ){ 5096f01d3a7eSdan sessionBufferGrow(pBuf, 2+nRec+nChange, pRc); 5097f01d3a7eSdan if( *pRc==SQLITE_OK ){ 5098f01d3a7eSdan int bData = 0; 5099f01d3a7eSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5100f01d3a7eSdan int i; 5101f01d3a7eSdan u8 *a1 = aRec; 5102f01d3a7eSdan u8 *a2 = aChange; 5103f01d3a7eSdan 5104f01d3a7eSdan *pOut++ = SQLITE_UPDATE; 5105f01d3a7eSdan *pOut++ = pIter->bIndirect; 5106f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5107f01d3a7eSdan int n1 = sessionSerialLen(a1); 5108f01d3a7eSdan int n2 = sessionSerialLen(a2); 5109f01d3a7eSdan if( pIter->abPK[i] || a2[0]==0 ){ 5110f01d3a7eSdan if( !pIter->abPK[i] ) bData = 1; 5111f01d3a7eSdan memcpy(pOut, a1, n1); 5112f01d3a7eSdan pOut += n1; 511324a0c453Sdan }else if( a2[0]!=0xFF ){ 511424a0c453Sdan bData = 1; 511524a0c453Sdan memcpy(pOut, a2, n2); 511624a0c453Sdan pOut += n2; 5117f01d3a7eSdan }else{ 5118f01d3a7eSdan *pOut++ = '\0'; 5119f01d3a7eSdan } 5120f01d3a7eSdan a1 += n1; 5121f01d3a7eSdan a2 += n2; 5122f01d3a7eSdan } 5123f01d3a7eSdan if( bData ){ 5124f01d3a7eSdan a2 = aChange; 5125f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5126f01d3a7eSdan int n1 = sessionSerialLen(a1); 5127f01d3a7eSdan int n2 = sessionSerialLen(a2); 512824a0c453Sdan if( pIter->abPK[i] || a2[0]!=0xFF ){ 5129f01d3a7eSdan memcpy(pOut, a1, n1); 5130f01d3a7eSdan pOut += n1; 5131f01d3a7eSdan }else{ 5132f01d3a7eSdan *pOut++ = '\0'; 5133f01d3a7eSdan } 5134f01d3a7eSdan a1 += n1; 5135f01d3a7eSdan a2 += n2; 5136f01d3a7eSdan } 5137f01d3a7eSdan pBuf->nBuf = (pOut - pBuf->aBuf); 5138f01d3a7eSdan } 5139f01d3a7eSdan } 5140f01d3a7eSdan } 5141f01d3a7eSdan 51423fa5463cSdan /* 51433fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases 51443fa5463cSdan ** that changeset according to the current configuration of the rebaser 51453fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput 51463fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking 51473fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL, 51483fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset 51493fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of 51503fa5463cSdan ** the buffer in bytes. It is the responsibility of the caller to eventually 51513fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free(). 51523fa5463cSdan ** 51533fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and 51543fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before 51553fa5463cSdan ** returning. 51563fa5463cSdan */ 5157c0a499eaSdan static int sessionRebase( 5158c0a499eaSdan sqlite3_rebaser *p, /* Rebaser hash table */ 5159c0a499eaSdan sqlite3_changeset_iter *pIter, /* Input data */ 5160c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5161c0a499eaSdan void *pOut, /* Context for xOutput callback */ 5162c0a499eaSdan int *pnOut, /* OUT: Number of bytes in output changeset */ 5163c0a499eaSdan void **ppOut /* OUT: Inverse of pChangeset */ 5164c0a499eaSdan ){ 5165c0a499eaSdan int rc = SQLITE_OK; 5166c0a499eaSdan u8 *aRec = 0; 5167c0a499eaSdan int nRec = 0; 5168c0a499eaSdan int bNew = 0; 5169c0a499eaSdan SessionTable *pTab = 0; 5170c0a499eaSdan SessionBuffer sOut = {0,0,0}; 5171c0a499eaSdan 5172f1b40e83Sdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){ 5173c0a499eaSdan SessionChange *pChange = 0; 5174f01d3a7eSdan int bDone = 0; 5175c0a499eaSdan 5176c0a499eaSdan if( bNew ){ 5177c0a499eaSdan const char *zTab = pIter->zTab; 5178c0a499eaSdan for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){ 5179c0a499eaSdan if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break; 5180c0a499eaSdan } 5181c0a499eaSdan bNew = 0; 5182c0a499eaSdan 518395ccb6dcSdan /* A patchset may not be rebased */ 518495ccb6dcSdan if( pIter->bPatchset ){ 518595ccb6dcSdan rc = SQLITE_ERROR; 518695ccb6dcSdan } 518795ccb6dcSdan 5188c0a499eaSdan /* Append a table header to the output for this new table */ 5189c0a499eaSdan sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc); 5190c0a499eaSdan sessionAppendVarint(&sOut, pIter->nCol, &rc); 5191c0a499eaSdan sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc); 5192c0a499eaSdan sessionAppendBlob(&sOut, (u8*)pIter->zTab, strlen(pIter->zTab)+1, &rc); 5193c0a499eaSdan } 5194c0a499eaSdan 51958cb83beeSdan if( pTab && rc==SQLITE_OK ){ 51968cb83beeSdan int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange); 5197c0a499eaSdan 5198c0a499eaSdan for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){ 51998cb83beeSdan if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){ 5200c0a499eaSdan break; 5201c0a499eaSdan } 5202c0a499eaSdan } 5203c0a499eaSdan } 5204c0a499eaSdan 5205c0a499eaSdan if( pChange ){ 5206c0a499eaSdan assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT ); 5207f01d3a7eSdan switch( pIter->op ){ 5208f01d3a7eSdan case SQLITE_INSERT: 5209f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5210f01d3a7eSdan bDone = 1; 5211f01d3a7eSdan if( pChange->bIndirect==0 ){ 5212f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_UPDATE, &rc); 5213f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5214f01d3a7eSdan sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc); 5215f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5216f01d3a7eSdan } 5217f01d3a7eSdan } 5218f01d3a7eSdan break; 5219f01d3a7eSdan 5220f01d3a7eSdan case SQLITE_UPDATE: 5221f01d3a7eSdan bDone = 1; 5222f01d3a7eSdan if( pChange->op==SQLITE_DELETE ){ 5223f01d3a7eSdan if( pChange->bIndirect==0 ){ 5224f01d3a7eSdan u8 *pCsr = aRec; 5225f01d3a7eSdan sessionSkipRecord(&pCsr, pIter->nCol); 5226f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_INSERT, &rc); 5227f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5228b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5229f01d3a7eSdan pCsr, nRec-(pCsr-aRec), 5230f01d3a7eSdan pChange->aRecord, pChange->nRecord, &rc 5231f01d3a7eSdan ); 5232f01d3a7eSdan } 5233f01d3a7eSdan }else{ 5234f01d3a7eSdan sessionAppendPartialUpdate(&sOut, pIter, 5235f01d3a7eSdan aRec, nRec, pChange->aRecord, pChange->nRecord, &rc 5236f01d3a7eSdan ); 5237f01d3a7eSdan } 5238f01d3a7eSdan break; 5239f01d3a7eSdan 5240f01d3a7eSdan default: 5241f01d3a7eSdan assert( pIter->op==SQLITE_DELETE ); 5242f01d3a7eSdan bDone = 1; 5243f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5244f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_DELETE, &rc); 5245f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5246b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5247f01d3a7eSdan pChange->aRecord, pChange->nRecord, aRec, nRec, &rc 5248f01d3a7eSdan ); 5249f01d3a7eSdan } 5250f01d3a7eSdan break; 5251f01d3a7eSdan } 5252f01d3a7eSdan } 5253f01d3a7eSdan 5254f01d3a7eSdan if( bDone==0 ){ 5255f01d3a7eSdan sessionAppendByte(&sOut, pIter->op, &rc); 5256f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5257f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5258f01d3a7eSdan } 5259c0a499eaSdan if( rc==SQLITE_OK && xOutput && sOut.nBuf>SESSIONS_STRM_CHUNK_SIZE ){ 5260c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5261c0a499eaSdan sOut.nBuf = 0; 5262c0a499eaSdan } 5263c0a499eaSdan if( rc ) break; 5264c0a499eaSdan } 5265c0a499eaSdan 5266c0a499eaSdan if( rc!=SQLITE_OK ){ 5267c0a499eaSdan sqlite3_free(sOut.aBuf); 5268c0a499eaSdan memset(&sOut, 0, sizeof(sOut)); 5269c0a499eaSdan } 5270c0a499eaSdan 5271c0a499eaSdan if( rc==SQLITE_OK ){ 5272c0a499eaSdan if( xOutput ){ 5273c0a499eaSdan if( sOut.nBuf>0 ){ 5274c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5275c0a499eaSdan } 5276c0a499eaSdan }else{ 5277c0a499eaSdan *ppOut = (void*)sOut.aBuf; 5278c0a499eaSdan *pnOut = sOut.nBuf; 5279c0a499eaSdan sOut.aBuf = 0; 5280c0a499eaSdan } 5281c0a499eaSdan } 5282c0a499eaSdan sqlite3_free(sOut.aBuf); 5283c0a499eaSdan return rc; 5284c0a499eaSdan } 5285c0a499eaSdan 5286c0a499eaSdan /* 5287c0a499eaSdan ** Create a new rebaser object. 5288c0a499eaSdan */ 5289c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){ 5290c0a499eaSdan int rc = SQLITE_OK; 5291c0a499eaSdan sqlite3_rebaser *pNew; 5292c0a499eaSdan 5293c0a499eaSdan pNew = sqlite3_malloc(sizeof(sqlite3_rebaser)); 5294c0a499eaSdan if( pNew==0 ){ 5295c0a499eaSdan rc = SQLITE_NOMEM; 5296f1b40e83Sdan }else{ 5297f1b40e83Sdan memset(pNew, 0, sizeof(sqlite3_rebaser)); 5298c0a499eaSdan } 5299c0a499eaSdan *ppNew = pNew; 5300c0a499eaSdan return rc; 5301c0a499eaSdan } 5302c0a499eaSdan 5303c0a499eaSdan /* 5304c0a499eaSdan ** Call this one or more times to configure a rebaser. 5305c0a499eaSdan */ 5306c0a499eaSdan int sqlite3rebaser_configure( 5307c0a499eaSdan sqlite3_rebaser *p, 5308c0a499eaSdan int nRebase, const void *pRebase 5309c0a499eaSdan ){ 5310c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator opened on pData/nData */ 5311c0a499eaSdan int rc; /* Return code */ 5312c0a499eaSdan rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase); 5313c0a499eaSdan if( rc==SQLITE_OK ){ 5314c0a499eaSdan rc = sessionChangesetToHash(pIter, &p->grp, 1); 5315c0a499eaSdan } 5316c0a499eaSdan sqlite3changeset_finalize(pIter); 5317c0a499eaSdan return rc; 5318c0a499eaSdan } 5319c0a499eaSdan 5320c0a499eaSdan /* 5321c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5322c0a499eaSdan */ 5323c0a499eaSdan int sqlite3rebaser_rebase( 5324c0a499eaSdan sqlite3_rebaser *p, 5325c0a499eaSdan int nIn, const void *pIn, 5326c0a499eaSdan int *pnOut, void **ppOut 5327c0a499eaSdan ){ 5328c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5329c0a499eaSdan int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn); 5330c0a499eaSdan 5331c0a499eaSdan if( rc==SQLITE_OK ){ 5332c0a499eaSdan rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut); 5333c0a499eaSdan sqlite3changeset_finalize(pIter); 5334c0a499eaSdan } 5335c0a499eaSdan 5336c0a499eaSdan return rc; 5337c0a499eaSdan } 5338c0a499eaSdan 5339c0a499eaSdan /* 5340c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5341c0a499eaSdan */ 5342c0a499eaSdan int sqlite3rebaser_rebase_strm( 5343c0a499eaSdan sqlite3_rebaser *p, 5344c0a499eaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 5345c0a499eaSdan void *pIn, 5346c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5347c0a499eaSdan void *pOut 5348c0a499eaSdan ){ 5349c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5350c0a499eaSdan int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 5351c0a499eaSdan 5352c0a499eaSdan if( rc==SQLITE_OK ){ 5353c0a499eaSdan rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0); 5354c0a499eaSdan sqlite3changeset_finalize(pIter); 5355c0a499eaSdan } 5356c0a499eaSdan 5357c0a499eaSdan return rc; 5358c0a499eaSdan } 5359c0a499eaSdan 5360c0a499eaSdan /* 5361c0a499eaSdan ** Destroy a rebaser object 5362c0a499eaSdan */ 5363f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){ 5364c0a499eaSdan if( p ){ 5365c0a499eaSdan sessionDeleteTable(p->grp.pList); 5366c0a499eaSdan sqlite3_free(p); 5367c0a499eaSdan } 5368c0a499eaSdan } 5369c0a499eaSdan 53709b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */ 5371