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 281f48e67dSdan static int sessions_strm_chunk_size = SESSIONS_STRM_CHUNK_SIZE; 291f48e67dSdan 30cf8e9144Sdan typedef struct SessionHook SessionHook; 31cf8e9144Sdan struct SessionHook { 32cf8e9144Sdan void *pCtx; 33cf8e9144Sdan int (*xOld)(void*,int,sqlite3_value**); 34cf8e9144Sdan int (*xNew)(void*,int,sqlite3_value**); 35cf8e9144Sdan int (*xCount)(void*); 36cf8e9144Sdan int (*xDepth)(void*); 37cf8e9144Sdan }; 38cf8e9144Sdan 39296c7658Sdan /* 40296c7658Sdan ** Session handle structure. 41296c7658Sdan */ 424fccf43aSdan struct sqlite3_session { 434fccf43aSdan sqlite3 *db; /* Database handle session is attached to */ 444fccf43aSdan char *zDb; /* Name of database session is attached to */ 45296c7658Sdan int bEnable; /* True if currently recording */ 46b4480e94Sdan int bIndirect; /* True if all changes are indirect */ 47ff4d0f41Sdan int bAutoAttach; /* True to auto-attach tables */ 484fccf43aSdan int rc; /* Non-zero if an error has occurred */ 497531a5a3Sdan void *pFilterCtx; /* First argument to pass to xTableFilter */ 507531a5a3Sdan int (*xTableFilter)(void *pCtx, const char *zTab); 511611e5a3Sdan sqlite3_value *pZeroBlob; /* Value containing X'' */ 524fccf43aSdan sqlite3_session *pNext; /* Next session object on same db. */ 534fccf43aSdan SessionTable *pTable; /* List of attached tables */ 54cf8e9144Sdan SessionHook hook; /* APIs to grab new and old data with */ 554fccf43aSdan }; 564fccf43aSdan 574fccf43aSdan /* 58ef7a6304Sdan ** Instances of this structure are used to build strings or binary records. 59ef7a6304Sdan */ 60ef7a6304Sdan struct SessionBuffer { 61ef7a6304Sdan u8 *aBuf; /* Pointer to changeset buffer */ 62ef7a6304Sdan int nBuf; /* Size of buffer aBuf */ 63ef7a6304Sdan int nAlloc; /* Size of allocation containing aBuf */ 64ef7a6304Sdan }; 65ef7a6304Sdan 66ef7a6304Sdan /* 6716228167Sdan ** An object of this type is used internally as an abstraction for 6816228167Sdan ** input data. Input data may be supplied either as a single large buffer 6916228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g. 70f1a08ad8Sdrh ** sqlite3changeset_start_strm()). 71ef7a6304Sdan */ 72ef7a6304Sdan struct SessionInput { 733e259bcdSdan int bNoDiscard; /* If true, do not discard in InputBuffer() */ 74d9151526Sdan int iCurrent; /* Offset in aData[] of current change */ 754757c658Sdan int iNext; /* Offset in aData[] of next change */ 764757c658Sdan u8 *aData; /* Pointer to buffer containing changeset */ 774757c658Sdan int nData; /* Number of bytes in aData */ 784757c658Sdan 79ef7a6304Sdan SessionBuffer buf; /* Current read buffer */ 80ef7a6304Sdan int (*xInput)(void*, void*, int*); /* Input stream call (or NULL) */ 81ef7a6304Sdan void *pIn; /* First argument to xInput */ 82ef7a6304Sdan int bEof; /* Set to true after xInput finished */ 83ef7a6304Sdan }; 84ef7a6304Sdan 85ef7a6304Sdan /* 86296c7658Sdan ** Structure for changeset iterators. 87296c7658Sdan */ 88296c7658Sdan struct sqlite3_changeset_iter { 89ef7a6304Sdan SessionInput in; /* Input buffer or stream */ 90ef7a6304Sdan SessionBuffer tblhdr; /* Buffer to hold apValue/zTab/abPK/ */ 9173b3c055Sdan int bPatchset; /* True if this is a patchset */ 9244748f27Sdan int bInvert; /* True to invert changeset */ 93296c7658Sdan int rc; /* Iterator error code */ 94296c7658Sdan sqlite3_stmt *pConflict; /* Points to conflicting row, if any */ 95296c7658Sdan char *zTab; /* Current table */ 96296c7658Sdan int nCol; /* Number of columns in zTab */ 97296c7658Sdan int op; /* Current operation */ 98b4480e94Sdan int bIndirect; /* True if current change was indirect */ 99244593c8Sdan u8 *abPK; /* Primary key array */ 100296c7658Sdan sqlite3_value **apValue; /* old.* and new.* values */ 101296c7658Sdan }; 102296c7658Sdan 103296c7658Sdan /* 1044fccf43aSdan ** Each session object maintains a set of the following structures, one 1054fccf43aSdan ** for each table the session object is monitoring. The structures are 1064fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable. 1074fccf43aSdan ** 1084fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have 1094fccf43aSdan ** been modified in any way since the session object was attached to the 1104fccf43aSdan ** table. 1114fccf43aSdan ** 1124fccf43aSdan ** The data associated with each hash-table entry is a structure containing 1134fccf43aSdan ** a subset of the initial values that the modified row contained at the 1144fccf43aSdan ** start of the session. Or no initial values if the row was inserted. 1154fccf43aSdan */ 1164fccf43aSdan struct SessionTable { 1174fccf43aSdan SessionTable *pNext; 1184fccf43aSdan char *zName; /* Local name of table */ 1194fccf43aSdan int nCol; /* Number of columns in table zName */ 1203739f298Sdan int bStat1; /* True if this is sqlite_stat1 */ 121e8d5648eSdan const char **azCol; /* Column names */ 122e8d5648eSdan u8 *abPK; /* Array of primary key flags */ 123296c7658Sdan int nEntry; /* Total number of entries in hash table */ 1244fccf43aSdan int nChange; /* Size of apChange[] array */ 1254fccf43aSdan SessionChange **apChange; /* Hash table buckets */ 1264fccf43aSdan }; 1274fccf43aSdan 1284fccf43aSdan /* 1294fccf43aSdan ** RECORD FORMAT: 1304fccf43aSdan ** 1314fccf43aSdan ** The following record format is similar to (but not compatible with) that 1324fccf43aSdan ** used in SQLite database files. This format is used as part of the 1334fccf43aSdan ** change-set binary format, and so must be architecture independent. 1344fccf43aSdan ** 1354fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained - 1364fccf43aSdan ** there is no separation of header and data. Each field begins with a 1374fccf43aSdan ** single byte describing its type, as follows: 1384fccf43aSdan ** 1394fccf43aSdan ** 0x00: Undefined value. 1404fccf43aSdan ** 0x01: Integer value. 1414fccf43aSdan ** 0x02: Real value. 1424fccf43aSdan ** 0x03: Text value. 1434fccf43aSdan ** 0x04: Blob value. 1444fccf43aSdan ** 0x05: SQL NULL value. 1454fccf43aSdan ** 1464fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT 1474fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists 1484fccf43aSdan ** only of the single type byte. For other types of values, the type byte 1494fccf43aSdan ** is followed by: 1504fccf43aSdan ** 1514fccf43aSdan ** Text values: 1524fccf43aSdan ** A varint containing the number of bytes in the value (encoded using 1534fccf43aSdan ** UTF-8). Followed by a buffer containing the UTF-8 representation 1544fccf43aSdan ** of the text value. There is no nul terminator. 1554fccf43aSdan ** 1564fccf43aSdan ** Blob values: 1574fccf43aSdan ** A varint containing the number of bytes in the value, followed by 1584fccf43aSdan ** a buffer containing the value itself. 1594fccf43aSdan ** 1604fccf43aSdan ** Integer values: 1614fccf43aSdan ** An 8-byte big-endian integer value. 1624fccf43aSdan ** 1634fccf43aSdan ** Real values: 1644fccf43aSdan ** An 8-byte big-endian IEEE 754-2008 real value. 1654fccf43aSdan ** 1664fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite 1674fccf43aSdan ** record format. 1684fccf43aSdan ** 1694fccf43aSdan ** CHANGESET FORMAT: 1704fccf43aSdan ** 1714fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on 1724fccf43aSdan ** one or more tables. Operations on a single table are grouped together, 1734fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all 1744fccf43aSdan ** mixed together). 1754fccf43aSdan ** 1764fccf43aSdan ** Each group of changes begins with a table header: 1774fccf43aSdan ** 1784fccf43aSdan ** 1 byte: Constant 0x54 (capital 'T') 179730bb805Sdan ** Varint: Number of columns in the table. 18073b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 1814fccf43aSdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 1824fccf43aSdan ** 1834fccf43aSdan ** Followed by one or more changes to the table. 1844fccf43aSdan ** 185c8be6437Sdrh ** 1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09). 1865d607a6eSdan ** 1 byte: The "indirect-change" flag. 1874fccf43aSdan ** old.* record: (delete and update only) 1884fccf43aSdan ** new.* record: (insert and update only) 18973b3c055Sdan ** 190730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the 191730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of 192730bb805Sdan ** columns in the table. The i'th field of each record is associated with 193730bb805Sdan ** the i'th column of the table, counting from left to right in the order 194730bb805Sdan ** in which columns were declared in the CREATE TABLE statement. 195730bb805Sdan ** 196730bb805Sdan ** The new.* record that is part of each INSERT change contains the values 197730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each 198730bb805Sdan ** DELETE change contains the values that made up the row that was deleted 199730bb805Sdan ** from the database. In the changeset format, the records that are part 200730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00) 201730bb805Sdan ** fields. 202730bb805Sdan ** 203730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields 204730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are 205730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields 206730bb805Sdan ** are set to the values that made up the row before the UPDATE that the 207730bb805Sdan ** change records took place. Within the new.* record, fields associated 208730bb805Sdan ** with table columns modified by the UPDATE change contain the new 209730bb805Sdan ** values. Fields associated with table columns that are not modified 210730bb805Sdan ** are set to "undefined". 211730bb805Sdan ** 21273b3c055Sdan ** PATCHSET FORMAT: 21373b3c055Sdan ** 21473b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset, 215730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict 216730bb805Sdan ** resolution is required when applying the changeset. 21773b3c055Sdan ** 21873b3c055Sdan ** Each group of changes begins with a table header: 21973b3c055Sdan ** 22073b3c055Sdan ** 1 byte: Constant 0x50 (capital 'P') 221730bb805Sdan ** Varint: Number of columns in the table. 22273b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 22373b3c055Sdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 22473b3c055Sdan ** 22573b3c055Sdan ** Followed by one or more changes to the table. 22673b3c055Sdan ** 227c8be6437Sdrh ** 1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09). 22873b3c055Sdan ** 1 byte: The "indirect-change" flag. 229730bb805Sdan ** single record: (PK fields for DELETE, PK and modified fields for UPDATE, 230730bb805Sdan ** full record for INSERT). 231730bb805Sdan ** 232730bb805Sdan ** As in the changeset format, each field of the single record that is part 233730bb805Sdan ** of a patchset change is associated with the correspondingly positioned 234730bb805Sdan ** table column, counting from left to right within the CREATE TABLE 235730bb805Sdan ** statement. 236730bb805Sdan ** 237730bb805Sdan ** For a DELETE change, all fields within the record except those associated 238f01d3a7eSdan ** with PRIMARY KEY columns are omitted. The PRIMARY KEY fields contain the 239f01d3a7eSdan ** values identifying the row to delete. 240730bb805Sdan ** 241730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY 242730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined". 243730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update, 244730bb805Sdan ** and fields associated with modified columns contain the new column values. 245730bb805Sdan ** 246730bb805Sdan ** The records associated with INSERT changes are in the same format as for 247730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT 248730bb805Sdan ** change to contain a field set to "undefined". 249b74cf4b6Sdan ** 250b74cf4b6Sdan ** REBASE BLOB FORMAT: 251b74cf4b6Sdan ** 252b74cf4b6Sdan ** A rebase blob may be output by sqlite3changeset_apply_v2() and its 253b74cf4b6Sdan ** streaming equivalent for use with the sqlite3_rebaser APIs to rebase 254b74cf4b6Sdan ** existing changesets. A rebase blob contains one entry for each conflict 255b74cf4b6Sdan ** resolved using either the OMIT or REPLACE strategies within the apply_v2() 256b74cf4b6Sdan ** call. 257b74cf4b6Sdan ** 258b74cf4b6Sdan ** The format used for a rebase blob is very similar to that used for 259b74cf4b6Sdan ** changesets. All entries related to a single table are grouped together. 260b74cf4b6Sdan ** 261b74cf4b6Sdan ** Each group of entries begins with a table header in changeset format: 262b74cf4b6Sdan ** 263b74cf4b6Sdan ** 1 byte: Constant 0x54 (capital 'T') 264b74cf4b6Sdan ** Varint: Number of columns in the table. 265b74cf4b6Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 266b74cf4b6Sdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 267b74cf4b6Sdan ** 268b74cf4b6Sdan ** Followed by one or more entries associated with the table. 269b74cf4b6Sdan ** 270b74cf4b6Sdan ** 1 byte: Either SQLITE_INSERT (0x12), DELETE (0x09). 271b74cf4b6Sdan ** 1 byte: Flag. 0x01 for REPLACE, 0x00 for OMIT. 272b74cf4b6Sdan ** record: (in the record format defined above). 273b74cf4b6Sdan ** 274b74cf4b6Sdan ** In a rebase blob, the first field is set to SQLITE_INSERT if the change 275b74cf4b6Sdan ** that caused the conflict was an INSERT or UPDATE, or to SQLITE_DELETE if 276b74cf4b6Sdan ** it was a DELETE. The second field is set to 0x01 if the conflict 277b74cf4b6Sdan ** resolution strategy was REPLACE, or 0x00 if it was OMIT. 278b74cf4b6Sdan ** 279b74cf4b6Sdan ** If the change that caused the conflict was a DELETE, then the single 280b74cf4b6Sdan ** record is a copy of the old.* record from the original changeset. If it 281b74cf4b6Sdan ** was an INSERT, then the single record is a copy of the new.* record. If 282b74cf4b6Sdan ** the conflicting change was an UPDATE, then the single record is a copy 283b74cf4b6Sdan ** of the new.* record with the PK fields filled in based on the original 284b74cf4b6Sdan ** old.* record. 2854fccf43aSdan */ 2864fccf43aSdan 2874fccf43aSdan /* 2884fccf43aSdan ** For each row modified during a session, there exists a single instance of 2894fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table. 2904fccf43aSdan */ 2914fccf43aSdan struct SessionChange { 292798693b2Sdan int op; /* One of UPDATE, DELETE, INSERT */ 293b4480e94Sdan int bIndirect; /* True if this change is "indirect" */ 2944fccf43aSdan int nRecord; /* Number of bytes in buffer aRecord[] */ 2954fccf43aSdan u8 *aRecord; /* Buffer containing old.* record */ 2964fccf43aSdan SessionChange *pNext; /* For hash-table collisions */ 2974fccf43aSdan }; 2984fccf43aSdan 299296c7658Sdan /* 300296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the 301296c7658Sdan ** number of bytes written. 302296c7658Sdan */ 303296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){ 304296c7658Sdan return putVarint32(aBuf, iVal); 3054fccf43aSdan } 3064fccf43aSdan 307296c7658Sdan /* 308296c7658Sdan ** Return the number of bytes required to store value iVal as a varint. 309296c7658Sdan */ 310296c7658Sdan static int sessionVarintLen(int iVal){ 311296c7658Sdan return sqlite3VarintLen(iVal); 312296c7658Sdan } 313296c7658Sdan 314296c7658Sdan /* 315296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of 316296c7658Sdan ** bytes read. 317296c7658Sdan */ 3184fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){ 319296c7658Sdan return getVarint32(aBuf, *piVal); 3204fccf43aSdan } 3214fccf43aSdan 32248cd59a5Sdrh /* Load an unaligned and unsigned 32-bit integer */ 32348cd59a5Sdrh #define SESSION_UINT32(x) (((u32)(x)[0]<<24)|((x)[1]<<16)|((x)[2]<<8)|(x)[3]) 32448cd59a5Sdrh 325296c7658Sdan /* 326296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return 327296c7658Sdan ** the value read. 328296c7658Sdan */ 3294fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){ 33048cd59a5Sdrh u64 x = SESSION_UINT32(aRec); 33148cd59a5Sdrh u32 y = SESSION_UINT32(aRec+4); 33248cd59a5Sdrh x = (x<<32) + y; 33348cd59a5Sdrh return (sqlite3_int64)x; 3344fccf43aSdan } 3354fccf43aSdan 3364fccf43aSdan /* 337296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[]. 338296c7658Sdan */ 339296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){ 340296c7658Sdan aBuf[0] = (i>>56) & 0xFF; 341296c7658Sdan aBuf[1] = (i>>48) & 0xFF; 342296c7658Sdan aBuf[2] = (i>>40) & 0xFF; 343296c7658Sdan aBuf[3] = (i>>32) & 0xFF; 344296c7658Sdan aBuf[4] = (i>>24) & 0xFF; 345296c7658Sdan aBuf[5] = (i>>16) & 0xFF; 346296c7658Sdan aBuf[6] = (i>> 8) & 0xFF; 347296c7658Sdan aBuf[7] = (i>> 0) & 0xFF; 348296c7658Sdan } 349296c7658Sdan 350296c7658Sdan /* 3514fccf43aSdan ** This function is used to serialize the contents of value pValue (see 3524fccf43aSdan ** comment titled "RECORD FORMAT" above). 3534fccf43aSdan ** 3544fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to 3554fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before 3564fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is 3574fccf43aSdan ** set *pnWrite. 3584fccf43aSdan ** 3594fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs 3604fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16)) 3614fccf43aSdan ** SQLITE_NOMEM is returned. 3624fccf43aSdan */ 3634fccf43aSdan static int sessionSerializeValue( 3644fccf43aSdan u8 *aBuf, /* If non-NULL, write serialized value here */ 3654fccf43aSdan sqlite3_value *pValue, /* Value to serialize */ 3662d77d80aSdrh sqlite3_int64 *pnWrite /* IN/OUT: Increment by bytes written */ 3674fccf43aSdan ){ 368296c7658Sdan int nByte; /* Size of serialized value in bytes */ 3694fccf43aSdan 37080fe2d93Sdan if( pValue ){ 37180fe2d93Sdan int eType; /* Value type (SQLITE_NULL, TEXT etc.) */ 37280fe2d93Sdan 3734fccf43aSdan eType = sqlite3_value_type(pValue); 3744fccf43aSdan if( aBuf ) aBuf[0] = eType; 3754fccf43aSdan 3764fccf43aSdan switch( eType ){ 3774fccf43aSdan case SQLITE_NULL: 3784fccf43aSdan nByte = 1; 3794fccf43aSdan break; 3804fccf43aSdan 3814fccf43aSdan case SQLITE_INTEGER: 3824fccf43aSdan case SQLITE_FLOAT: 3834fccf43aSdan if( aBuf ){ 3844fccf43aSdan /* TODO: SQLite does something special to deal with mixed-endian 3854fccf43aSdan ** floating point values (e.g. ARM7). This code probably should 3864fccf43aSdan ** too. */ 3874fccf43aSdan u64 i; 3884fccf43aSdan if( eType==SQLITE_INTEGER ){ 3894fccf43aSdan i = (u64)sqlite3_value_int64(pValue); 3904fccf43aSdan }else{ 3914fccf43aSdan double r; 3924fccf43aSdan assert( sizeof(double)==8 && sizeof(u64)==8 ); 3934fccf43aSdan r = sqlite3_value_double(pValue); 3944fccf43aSdan memcpy(&i, &r, 8); 3954fccf43aSdan } 396296c7658Sdan sessionPutI64(&aBuf[1], i); 3974fccf43aSdan } 3984fccf43aSdan nByte = 9; 3994fccf43aSdan break; 4004fccf43aSdan 4014e895da1Sdan default: { 40280fe2d93Sdan u8 *z; 40380fe2d93Sdan int n; 40480fe2d93Sdan int nVarint; 40580fe2d93Sdan 4064e895da1Sdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 40780fe2d93Sdan if( eType==SQLITE_TEXT ){ 40880fe2d93Sdan z = (u8 *)sqlite3_value_text(pValue); 40980fe2d93Sdan }else{ 41080fe2d93Sdan z = (u8 *)sqlite3_value_blob(pValue); 41180fe2d93Sdan } 41280fe2d93Sdan n = sqlite3_value_bytes(pValue); 4133cc89d95Sdan if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 41480fe2d93Sdan nVarint = sessionVarintLen(n); 41580fe2d93Sdan 4164fccf43aSdan if( aBuf ){ 4174fccf43aSdan sessionVarintPut(&aBuf[1], n); 418895decf6Sdan if( n ) memcpy(&aBuf[nVarint + 1], z, n); 4194fccf43aSdan } 4204fccf43aSdan 4214fccf43aSdan nByte = 1 + nVarint + n; 4224fccf43aSdan break; 4234fccf43aSdan } 4244fccf43aSdan } 42580fe2d93Sdan }else{ 42680fe2d93Sdan nByte = 1; 42780fe2d93Sdan if( aBuf ) aBuf[0] = '\0'; 42880fe2d93Sdan } 4294fccf43aSdan 430fa122adaSdan if( pnWrite ) *pnWrite += nByte; 4314fccf43aSdan return SQLITE_OK; 4324fccf43aSdan } 4334fccf43aSdan 434fa122adaSdan 435798693b2Sdan /* 436798693b2Sdan ** This macro is used to calculate hash key values for data structures. In 437798693b2Sdan ** order to use this macro, the entire data structure must be represented 438798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value 439798693b2Sdan ** for a data structure represented as three such integers, the macro may 440798693b2Sdan ** then be used as follows: 441798693b2Sdan ** 442798693b2Sdan ** int hash_key_value; 443798693b2Sdan ** hash_key_value = HASH_APPEND(0, <value 1>); 444798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 2>); 445798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 3>); 446798693b2Sdan ** 447798693b2Sdan ** In practice, the data structures this macro is used for are the primary 448798693b2Sdan ** key values of modified rows. 449798693b2Sdan */ 4504131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add) 451798693b2Sdan 452798693b2Sdan /* 453798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the 454798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 455798693b2Sdan */ 4564131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){ 457e8d5648eSdan h = HASH_APPEND(h, i & 0xFFFFFFFF); 458e8d5648eSdan return HASH_APPEND(h, (i>>32)&0xFFFFFFFF); 459e8d5648eSdan } 460798693b2Sdan 461798693b2Sdan /* 462798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to 463798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value. 464798693b2Sdan */ 4654131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){ 466e8d5648eSdan int i; 467e8d5648eSdan for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]); 468e8d5648eSdan return h; 469e8d5648eSdan } 470e8d5648eSdan 4714fccf43aSdan /* 472798693b2Sdan ** Append the hash of the data type passed as the second argument to the 473798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 474798693b2Sdan */ 475798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){ 476798693b2Sdan return HASH_APPEND(h, eType); 477798693b2Sdan } 478798693b2Sdan 479798693b2Sdan /* 4804131639cSdan ** This function may only be called from within a pre-update callback. 4814131639cSdan ** It calculates a hash based on the primary key values of the old.* or 482798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to 483798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL 484798693b2Sdan ** values, *pbNullPK is set to true before returning. 485798693b2Sdan ** 486798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values 487798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned 488798693b2Sdan ** and the output variables are set as described above. 4894fccf43aSdan */ 490798693b2Sdan static int sessionPreupdateHash( 491cf8e9144Sdan sqlite3_session *pSession, /* Session object that owns pTab */ 492e8d5648eSdan SessionTable *pTab, /* Session table handle */ 493e8d5648eSdan int bNew, /* True to hash the new.* PK */ 49427453faeSdan int *piHash, /* OUT: Hash value */ 495798693b2Sdan int *pbNullPK /* OUT: True if there are NULL values in PK */ 496e8d5648eSdan ){ 4974131639cSdan unsigned int h = 0; /* Hash value to return */ 4984131639cSdan int i; /* Used to iterate through columns */ 499e8d5648eSdan 50027453faeSdan assert( *pbNullPK==0 ); 501cf8e9144Sdan assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) ); 502e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 503e8d5648eSdan if( pTab->abPK[i] ){ 504e8d5648eSdan int rc; 505e8d5648eSdan int eType; 506e8d5648eSdan sqlite3_value *pVal; 507e8d5648eSdan 508e8d5648eSdan if( bNew ){ 509cf8e9144Sdan rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal); 510e8d5648eSdan }else{ 511cf8e9144Sdan rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal); 512e8d5648eSdan } 51312ca0b56Sdan if( rc!=SQLITE_OK ) return rc; 514e8d5648eSdan 515e8d5648eSdan eType = sqlite3_value_type(pVal); 516798693b2Sdan h = sessionHashAppendType(h, eType); 5176734007dSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 518e8d5648eSdan i64 iVal; 519e8d5648eSdan if( eType==SQLITE_INTEGER ){ 520e8d5648eSdan iVal = sqlite3_value_int64(pVal); 521e8d5648eSdan }else{ 522e8d5648eSdan double rVal = sqlite3_value_double(pVal); 523e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 524e8d5648eSdan memcpy(&iVal, &rVal, 8); 525e8d5648eSdan } 526e8d5648eSdan h = sessionHashAppendI64(h, iVal); 5276734007dSdan }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 5286734007dSdan const u8 *z; 5293cc89d95Sdan int n; 5306734007dSdan if( eType==SQLITE_TEXT ){ 5316734007dSdan z = (const u8 *)sqlite3_value_text(pVal); 5326734007dSdan }else{ 5336734007dSdan z = (const u8 *)sqlite3_value_blob(pVal); 534e8d5648eSdan } 5353cc89d95Sdan n = sqlite3_value_bytes(pVal); 5363cc89d95Sdan if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 5373cc89d95Sdan h = sessionHashAppendBlob(h, n, z); 5386734007dSdan }else{ 53927453faeSdan assert( eType==SQLITE_NULL ); 5401611e5a3Sdan assert( pTab->bStat1==0 || i!=1 ); 54127453faeSdan *pbNullPK = 1; 542e8d5648eSdan } 543e8d5648eSdan } 544e8d5648eSdan } 545e8d5648eSdan 546e8d5648eSdan *piHash = (h % pTab->nChange); 547e8d5648eSdan return SQLITE_OK; 548e8d5648eSdan } 549e8d5648eSdan 5504131639cSdan /* 5516cda207fSdan ** The buffer that the argument points to contains a serialized SQL value. 5526cda207fSdan ** Return the number of bytes of space occupied by the value (including 5536cda207fSdan ** the type byte). 5546cda207fSdan */ 5556cda207fSdan static int sessionSerialLen(u8 *a){ 5566cda207fSdan int e = *a; 5576cda207fSdan int n; 55824a0c453Sdan if( e==0 || e==0xFF ) return 1; 5596cda207fSdan if( e==SQLITE_NULL ) return 1; 5606cda207fSdan if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9; 5616cda207fSdan return sessionVarintGet(&a[1], &n) + 1 + n; 5626cda207fSdan } 5636cda207fSdan 5646cda207fSdan /* 5655d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a 566798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys 5674131639cSdan ** calculated by this function are compatible with those calculated by 5684131639cSdan ** sessionPreupdateHash(). 56964277f4aSdan ** 57064277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from 57164277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely. 5724131639cSdan */ 5734131639cSdan static unsigned int sessionChangeHash( 5744131639cSdan SessionTable *pTab, /* Table handle */ 57564277f4aSdan int bPkOnly, /* Record consists of PK fields only */ 5765d607a6eSdan u8 *aRecord, /* Change record */ 5774131639cSdan int nBucket /* Assume this many buckets in hash table */ 578e8d5648eSdan ){ 5794131639cSdan unsigned int h = 0; /* Value to return */ 5804131639cSdan int i; /* Used to iterate through columns */ 5815d607a6eSdan u8 *a = aRecord; /* Used to iterate through change record */ 582e8d5648eSdan 583e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 5846cda207fSdan int eType = *a; 585e8d5648eSdan int isPK = pTab->abPK[i]; 58664277f4aSdan if( bPkOnly && isPK==0 ) continue; 587e8d5648eSdan 58827453faeSdan /* It is not possible for eType to be SQLITE_NULL here. The session 58927453faeSdan ** module does not record changes for rows with NULL values stored in 59027453faeSdan ** primary key columns. */ 59127453faeSdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 59227453faeSdan || eType==SQLITE_TEXT || eType==SQLITE_BLOB 5936cda207fSdan || eType==SQLITE_NULL || eType==0 59427453faeSdan ); 5951611e5a3Sdan assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) ); 59627453faeSdan 5976cda207fSdan if( isPK ){ 5986cda207fSdan a++; 599798693b2Sdan h = sessionHashAppendType(h, eType); 60027453faeSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 6016734007dSdan h = sessionHashAppendI64(h, sessionGetI64(a)); 602e8d5648eSdan a += 8; 6031611e5a3Sdan }else{ 604e8d5648eSdan int n; 605e8d5648eSdan a += sessionVarintGet(a, &n); 6066734007dSdan h = sessionHashAppendBlob(h, n, a); 607e8d5648eSdan a += n; 608e8d5648eSdan } 6096cda207fSdan }else{ 6106cda207fSdan a += sessionSerialLen(a); 6116cda207fSdan } 612e8d5648eSdan } 613e8d5648eSdan return (h % nBucket); 614e8d5648eSdan } 615e8d5648eSdan 616798693b2Sdan /* 617798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab. 618798693b2Sdan ** This function returns true if the two records apply to the same row (i.e. 619798693b2Sdan ** have the same values stored in the primary key columns), or false 620798693b2Sdan ** otherwise. 621798693b2Sdan */ 6225d607a6eSdan static int sessionChangeEqual( 623798693b2Sdan SessionTable *pTab, /* Table used for PK definition */ 624a71d2371Sdan int bLeftPkOnly, /* True if aLeft[] contains PK fields only */ 6255d607a6eSdan u8 *aLeft, /* Change record */ 626a71d2371Sdan int bRightPkOnly, /* True if aRight[] contains PK fields only */ 6275d607a6eSdan u8 *aRight /* Change record */ 6285d607a6eSdan ){ 629798693b2Sdan u8 *a1 = aLeft; /* Cursor to iterate through aLeft */ 630798693b2Sdan u8 *a2 = aRight; /* Cursor to iterate through aRight */ 631798693b2Sdan int iCol; /* Used to iterate through table columns */ 6325d607a6eSdan 633798693b2Sdan for(iCol=0; iCol<pTab->nCol; iCol++){ 63414faa061Sdan if( pTab->abPK[iCol] ){ 6355d607a6eSdan int n1 = sessionSerialLen(a1); 6365d607a6eSdan int n2 = sessionSerialLen(a2); 6375d607a6eSdan 638bd45374cSdan if( n1!=n2 || memcmp(a1, a2, n1) ){ 6395d607a6eSdan return 0; 6405d607a6eSdan } 64114faa061Sdan a1 += n1; 64214faa061Sdan a2 += n2; 64314faa061Sdan }else{ 64414faa061Sdan if( bLeftPkOnly==0 ) a1 += sessionSerialLen(a1); 64514faa061Sdan if( bRightPkOnly==0 ) a2 += sessionSerialLen(a2); 64614faa061Sdan } 6475d607a6eSdan } 6485d607a6eSdan 6495d607a6eSdan return 1; 6505d607a6eSdan } 6515d607a6eSdan 652798693b2Sdan /* 653798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change 654798693b2Sdan ** records with nCol columns. This function "merges" the two records into 655798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is 656798693b2Sdan ** then set to point to one byte after the last byte written before 657798693b2Sdan ** returning. 658798693b2Sdan ** 659798693b2Sdan ** The merging of records is done as follows: For each column, if the 660798693b2Sdan ** aRight record contains a value for the column, copy the value from 661798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither 662798693b2Sdan ** record contains a value for a given column, then neither does the 663798693b2Sdan ** output record. 664798693b2Sdan */ 6655d607a6eSdan static void sessionMergeRecord( 6665d607a6eSdan u8 **paOut, 667798693b2Sdan int nCol, 6685d607a6eSdan u8 *aLeft, 6695d607a6eSdan u8 *aRight 6705d607a6eSdan ){ 671798693b2Sdan u8 *a1 = aLeft; /* Cursor used to iterate through aLeft */ 672798693b2Sdan u8 *a2 = aRight; /* Cursor used to iterate through aRight */ 673798693b2Sdan u8 *aOut = *paOut; /* Output cursor */ 674798693b2Sdan int iCol; /* Used to iterate from 0 to nCol */ 6755d607a6eSdan 676798693b2Sdan for(iCol=0; iCol<nCol; iCol++){ 6775d607a6eSdan int n1 = sessionSerialLen(a1); 6785d607a6eSdan int n2 = sessionSerialLen(a2); 6795d607a6eSdan if( *a2 ){ 6805d607a6eSdan memcpy(aOut, a2, n2); 6815d607a6eSdan aOut += n2; 6825d607a6eSdan }else{ 6835d607a6eSdan memcpy(aOut, a1, n1); 6845d607a6eSdan aOut += n1; 6855d607a6eSdan } 6865d607a6eSdan a1 += n1; 6875d607a6eSdan a2 += n2; 6885d607a6eSdan } 6895d607a6eSdan 6905d607a6eSdan *paOut = aOut; 6915d607a6eSdan } 6925d607a6eSdan 693798693b2Sdan /* 694798693b2Sdan ** This is a helper function used by sessionMergeUpdate(). 695798693b2Sdan ** 696798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value 697798693b2Sdan ** within a change record. Before it returns, both have been advanced so 698798693b2Sdan ** as to point to the next value in the record. 699798693b2Sdan ** 700798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e. 701fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo 702798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the 703798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal 704798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points 705fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words: 706fa29ecc4Sdan ** 707fa29ecc4Sdan ** if( *paTwo is valid ) return *paTwo; 708fa29ecc4Sdan ** return *paOne; 709fa29ecc4Sdan ** 710798693b2Sdan */ 7115d607a6eSdan static u8 *sessionMergeValue( 712798693b2Sdan u8 **paOne, /* IN/OUT: Left-hand buffer pointer */ 713798693b2Sdan u8 **paTwo, /* IN/OUT: Right-hand buffer pointer */ 714798693b2Sdan int *pnVal /* OUT: Bytes in returned value */ 7155d607a6eSdan ){ 7165d607a6eSdan u8 *a1 = *paOne; 7175d607a6eSdan u8 *a2 = *paTwo; 7185d607a6eSdan u8 *pRet = 0; 7195d607a6eSdan int n1; 7205d607a6eSdan 7215d607a6eSdan assert( a1 ); 7225d607a6eSdan if( a2 ){ 7235d607a6eSdan int n2 = sessionSerialLen(a2); 7245d607a6eSdan if( *a2 ){ 7255d607a6eSdan *pnVal = n2; 7265d607a6eSdan pRet = a2; 7275d607a6eSdan } 7285d607a6eSdan *paTwo = &a2[n2]; 7295d607a6eSdan } 7305d607a6eSdan 7315d607a6eSdan n1 = sessionSerialLen(a1); 7325d607a6eSdan if( pRet==0 ){ 7335d607a6eSdan *pnVal = n1; 7345d607a6eSdan pRet = a1; 7355d607a6eSdan } 7365d607a6eSdan *paOne = &a1[n1]; 7375d607a6eSdan 7385d607a6eSdan return pRet; 7395d607a6eSdan } 7405d607a6eSdan 741798693b2Sdan /* 742798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes 743798693b2Sdan ** on the same row. 744798693b2Sdan */ 7455d607a6eSdan static int sessionMergeUpdate( 746798693b2Sdan u8 **paOut, /* IN/OUT: Pointer to output buffer */ 747798693b2Sdan SessionTable *pTab, /* Table change pertains to */ 748a71d2371Sdan int bPatchset, /* True if records are patchset records */ 749798693b2Sdan u8 *aOldRecord1, /* old.* record for first change */ 750798693b2Sdan u8 *aOldRecord2, /* old.* record for second change */ 751798693b2Sdan u8 *aNewRecord1, /* new.* record for first change */ 752798693b2Sdan u8 *aNewRecord2 /* new.* record for second change */ 7535d607a6eSdan ){ 7545d607a6eSdan u8 *aOld1 = aOldRecord1; 7555d607a6eSdan u8 *aOld2 = aOldRecord2; 7565d607a6eSdan u8 *aNew1 = aNewRecord1; 7575d607a6eSdan u8 *aNew2 = aNewRecord2; 7585d607a6eSdan 7595d607a6eSdan u8 *aOut = *paOut; 7605d607a6eSdan int i; 76164277f4aSdan 76264277f4aSdan if( bPatchset==0 ){ 7635d607a6eSdan int bRequired = 0; 7645d607a6eSdan 7655d607a6eSdan assert( aOldRecord1 && aNewRecord1 ); 7665d607a6eSdan 7675d607a6eSdan /* Write the old.* vector first. */ 7685d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7695d607a6eSdan int nOld; 7705d607a6eSdan u8 *aOld; 7715d607a6eSdan int nNew; 7725d607a6eSdan u8 *aNew; 7735d607a6eSdan 7745d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 7755d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 7765d607a6eSdan if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){ 7775d607a6eSdan if( pTab->abPK[i]==0 ) bRequired = 1; 7785d607a6eSdan memcpy(aOut, aOld, nOld); 7795d607a6eSdan aOut += nOld; 7805d607a6eSdan }else{ 7815d607a6eSdan *(aOut++) = '\0'; 7825d607a6eSdan } 7835d607a6eSdan } 7845d607a6eSdan 7855d607a6eSdan if( !bRequired ) return 0; 78664277f4aSdan } 7875d607a6eSdan 7885d607a6eSdan /* Write the new.* vector */ 7895d607a6eSdan aOld1 = aOldRecord1; 7905d607a6eSdan aOld2 = aOldRecord2; 7915d607a6eSdan aNew1 = aNewRecord1; 7925d607a6eSdan aNew2 = aNewRecord2; 7935d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7945d607a6eSdan int nOld; 7955d607a6eSdan u8 *aOld; 7965d607a6eSdan int nNew; 7975d607a6eSdan u8 *aNew; 7985d607a6eSdan 7995d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 8005d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 80164277f4aSdan if( bPatchset==0 80264277f4aSdan && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew))) 80364277f4aSdan ){ 8045d607a6eSdan *(aOut++) = '\0'; 8055d607a6eSdan }else{ 8065d607a6eSdan memcpy(aOut, aNew, nNew); 8075d607a6eSdan aOut += nNew; 8085d607a6eSdan } 8095d607a6eSdan } 8105d607a6eSdan 8115d607a6eSdan *paOut = aOut; 8125d607a6eSdan return 1; 8135d607a6eSdan } 8145d607a6eSdan 81577fc1d5bSdan /* 81677fc1d5bSdan ** This function is only called from within a pre-update-hook callback. 81777fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row 81877fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise 81977fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns 82077fc1d5bSdan ** false. 82177fc1d5bSdan */ 82277fc1d5bSdan static int sessionPreupdateEqual( 823cf8e9144Sdan sqlite3_session *pSession, /* Session object that owns SessionTable */ 82477fc1d5bSdan SessionTable *pTab, /* Table associated with change */ 82577fc1d5bSdan SessionChange *pChange, /* Change to compare to */ 82677fc1d5bSdan int op /* Current pre-update operation */ 827e8d5648eSdan ){ 82877fc1d5bSdan int iCol; /* Used to iterate through columns */ 82977fc1d5bSdan u8 *a = pChange->aRecord; /* Cursor used to scan change record */ 830e8d5648eSdan 83177fc1d5bSdan assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE ); 83277fc1d5bSdan for(iCol=0; iCol<pTab->nCol; iCol++){ 83377fc1d5bSdan if( !pTab->abPK[iCol] ){ 834798693b2Sdan a += sessionSerialLen(a); 835e8d5648eSdan }else{ 8366734007dSdan sqlite3_value *pVal; /* Value returned by preupdate_new/old */ 8376734007dSdan int rc; /* Error code from preupdate_new/old */ 838798693b2Sdan int eType = *a++; /* Type of value from change record */ 8396734007dSdan 8406734007dSdan /* The following calls to preupdate_new() and preupdate_old() can not 8416734007dSdan ** fail. This is because they cache their return values, and by the 8426734007dSdan ** time control flows to here they have already been called once from 8436734007dSdan ** within sessionPreupdateHash(). The first two asserts below verify 8446734007dSdan ** this (that the method has already been called). */ 84577fc1d5bSdan if( op==SQLITE_INSERT ){ 846cf8e9144Sdan /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */ 847cf8e9144Sdan rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal); 848e8d5648eSdan }else{ 849cf8e9144Sdan /* assert( db->pPreUpdate->pUnpacked ); */ 850cf8e9144Sdan rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal); 851e8d5648eSdan } 8526734007dSdan assert( rc==SQLITE_OK ); 8531611e5a3Sdan if( sqlite3_value_type(pVal)!=eType ) return 0; 854e8d5648eSdan 85512ca0b56Sdan /* A SessionChange object never has a NULL value in a PK column */ 85612ca0b56Sdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 85712ca0b56Sdan || eType==SQLITE_BLOB || eType==SQLITE_TEXT 85812ca0b56Sdan ); 85912ca0b56Sdan 86012ca0b56Sdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 861e8d5648eSdan i64 iVal = sessionGetI64(a); 862e8d5648eSdan a += 8; 863e8d5648eSdan if( eType==SQLITE_INTEGER ){ 86477fc1d5bSdan if( sqlite3_value_int64(pVal)!=iVal ) return 0; 865e8d5648eSdan }else{ 866e8d5648eSdan double rVal; 867e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 868e8d5648eSdan memcpy(&rVal, &iVal, 8); 86977fc1d5bSdan if( sqlite3_value_double(pVal)!=rVal ) return 0; 870e8d5648eSdan } 87112ca0b56Sdan }else{ 872e8d5648eSdan int n; 873e8d5648eSdan const u8 *z; 874e8d5648eSdan a += sessionVarintGet(a, &n); 87577fc1d5bSdan if( sqlite3_value_bytes(pVal)!=n ) return 0; 87612ca0b56Sdan if( eType==SQLITE_TEXT ){ 87712ca0b56Sdan z = sqlite3_value_text(pVal); 87812ca0b56Sdan }else{ 87912ca0b56Sdan z = sqlite3_value_blob(pVal); 88012ca0b56Sdan } 8813f2cebb6Sdan if( n>0 && memcmp(a, z, n) ) return 0; 882e8d5648eSdan a += n; 883e8d5648eSdan } 884e8d5648eSdan } 885e8d5648eSdan } 886e8d5648eSdan 88777fc1d5bSdan return 1; 8884fccf43aSdan } 8894fccf43aSdan 8904fccf43aSdan /* 8914fccf43aSdan ** If required, grow the hash table used to store changes on table pTab 8924fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the 8934fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return 8944fccf43aSdan ** SQLITE_OK. 8954fccf43aSdan ** 8964fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In 8974fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway. 8984fccf43aSdan ** Growing the hash table in this case is a performance optimization only, 8994fccf43aSdan ** it is not required for correct operation. 9004fccf43aSdan */ 90164277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){ 9024fccf43aSdan if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){ 9034fccf43aSdan int i; 9044fccf43aSdan SessionChange **apNew; 905f6ad201aSdrh sqlite3_int64 nNew = 2*(sqlite3_int64)(pTab->nChange ? pTab->nChange : 128); 9064fccf43aSdan 9072d77d80aSdrh apNew = (SessionChange **)sqlite3_malloc64(sizeof(SessionChange *) * nNew); 9084fccf43aSdan if( apNew==0 ){ 9094fccf43aSdan if( pTab->nChange==0 ){ 9104fccf43aSdan return SQLITE_ERROR; 9114fccf43aSdan } 9124fccf43aSdan return SQLITE_OK; 9134fccf43aSdan } 9144fccf43aSdan memset(apNew, 0, sizeof(SessionChange *) * nNew); 9154fccf43aSdan 9164fccf43aSdan for(i=0; i<pTab->nChange; i++){ 9174fccf43aSdan SessionChange *p; 9184fccf43aSdan SessionChange *pNext; 9194fccf43aSdan for(p=pTab->apChange[i]; p; p=pNext){ 92064277f4aSdan int bPkOnly = (p->op==SQLITE_DELETE && bPatchset); 92164277f4aSdan int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew); 9224fccf43aSdan pNext = p->pNext; 9234fccf43aSdan p->pNext = apNew[iHash]; 9244fccf43aSdan apNew[iHash] = p; 9254fccf43aSdan } 9264fccf43aSdan } 9274fccf43aSdan 9284fccf43aSdan sqlite3_free(pTab->apChange); 9294fccf43aSdan pTab->nChange = nNew; 9304fccf43aSdan pTab->apChange = apNew; 9314fccf43aSdan } 9324fccf43aSdan 9334fccf43aSdan return SQLITE_OK; 9344fccf43aSdan } 9354fccf43aSdan 936296c7658Sdan /* 937e8d5648eSdan ** This function queries the database for the names of the columns of table 9383f975373Sdrh ** zThis, in schema zDb. 939e8d5648eSdan ** 94077fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number 94177fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a 942e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to 943e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not 944e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding 945e8d5648eSdan ** column is part of the primary key. 946e8d5648eSdan ** 947e8d5648eSdan ** For example, if the table is declared as: 948e8d5648eSdan ** 949e8d5648eSdan ** CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z)); 950e8d5648eSdan ** 95177fc1d5bSdan ** Then the four output variables are populated as follows: 952e8d5648eSdan ** 95377fc1d5bSdan ** *pnCol = 4 954e8d5648eSdan ** *pzTab = "tbl1" 955e8d5648eSdan ** *pazCol = {"w", "x", "y", "z"} 956e8d5648eSdan ** *pabPK = {1, 0, 0, 1} 957e8d5648eSdan ** 958e8d5648eSdan ** All returned buffers are part of the same single allocation, which must 9593f975373Sdrh ** be freed using sqlite3_free() by the caller 960e8d5648eSdan */ 961e8d5648eSdan static int sessionTableInfo( 962e8d5648eSdan sqlite3 *db, /* Database connection */ 963e8d5648eSdan const char *zDb, /* Name of attached database (e.g. "main") */ 964e8d5648eSdan const char *zThis, /* Table name */ 965ca62ad57Sdan int *pnCol, /* OUT: number of columns */ 966e8d5648eSdan const char **pzTab, /* OUT: Copy of zThis */ 967e8d5648eSdan const char ***pazCol, /* OUT: Array of column names for table */ 968e8d5648eSdan u8 **pabPK /* OUT: Array of booleans - true for PK col */ 969e8d5648eSdan ){ 970e8d5648eSdan char *zPragma; 971e8d5648eSdan sqlite3_stmt *pStmt; 972e8d5648eSdan int rc; 9732d77d80aSdrh sqlite3_int64 nByte; 974e8d5648eSdan int nDbCol = 0; 975e8d5648eSdan int nThis; 976e8d5648eSdan int i; 97774f598b6Smistachkin u8 *pAlloc = 0; 978db04571cSdan char **azCol = 0; 97974f598b6Smistachkin u8 *abPK = 0; 980e8d5648eSdan 981db04571cSdan assert( pazCol && pabPK ); 982e8d5648eSdan 983cfdbde21Sdrh nThis = sqlite3Strlen30(zThis); 984614efe2bSdan if( nThis==12 && 0==sqlite3_stricmp("sqlite_stat1", zThis) ){ 98558713184Sdan rc = sqlite3_table_column_metadata(db, zDb, zThis, 0, 0, 0, 0, 0, 0); 98658713184Sdan if( rc==SQLITE_OK ){ 987614efe2bSdan /* For sqlite_stat1, pretend that (tbl,idx) is the PRIMARY KEY. */ 988614efe2bSdan zPragma = sqlite3_mprintf( 989614efe2bSdan "SELECT 0, 'tbl', '', 0, '', 1 UNION ALL " 990614efe2bSdan "SELECT 1, 'idx', '', 0, '', 2 UNION ALL " 991614efe2bSdan "SELECT 2, 'stat', '', 0, '', 0" 992614efe2bSdan ); 99358713184Sdan }else if( rc==SQLITE_ERROR ){ 99458713184Sdan zPragma = sqlite3_mprintf(""); 99558713184Sdan }else{ 99658713184Sdan return rc; 99758713184Sdan } 998614efe2bSdan }else{ 999e8d5648eSdan zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis); 1000614efe2bSdan } 1001e8d5648eSdan if( !zPragma ) return SQLITE_NOMEM; 1002e8d5648eSdan 1003e8d5648eSdan rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0); 1004e8d5648eSdan sqlite3_free(zPragma); 1005e8d5648eSdan if( rc!=SQLITE_OK ) return rc; 1006e8d5648eSdan 1007e8d5648eSdan nByte = nThis + 1; 1008e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1009e8d5648eSdan nByte += sqlite3_column_bytes(pStmt, 1); 1010e8d5648eSdan nDbCol++; 1011e8d5648eSdan } 1012e8d5648eSdan rc = sqlite3_reset(pStmt); 1013e8d5648eSdan 1014e8d5648eSdan if( rc==SQLITE_OK ){ 1015e8d5648eSdan nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1); 10162d77d80aSdrh pAlloc = sqlite3_malloc64(nByte); 1017e8d5648eSdan if( pAlloc==0 ){ 1018e8d5648eSdan rc = SQLITE_NOMEM; 1019e8d5648eSdan } 1020e8d5648eSdan } 1021e8d5648eSdan if( rc==SQLITE_OK ){ 1022e8d5648eSdan azCol = (char **)pAlloc; 1023ca62ad57Sdan pAlloc = (u8 *)&azCol[nDbCol]; 1024e8d5648eSdan abPK = (u8 *)pAlloc; 1025ca62ad57Sdan pAlloc = &abPK[nDbCol]; 1026e8d5648eSdan if( pzTab ){ 1027e8d5648eSdan memcpy(pAlloc, zThis, nThis+1); 1028e8d5648eSdan *pzTab = (char *)pAlloc; 1029e8d5648eSdan pAlloc += nThis+1; 1030e8d5648eSdan } 1031e8d5648eSdan 1032e8d5648eSdan i = 0; 1033e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1034e8d5648eSdan int nName = sqlite3_column_bytes(pStmt, 1); 1035e8d5648eSdan const unsigned char *zName = sqlite3_column_text(pStmt, 1); 1036e8d5648eSdan if( zName==0 ) break; 1037e8d5648eSdan memcpy(pAlloc, zName, nName+1); 1038e8d5648eSdan azCol[i] = (char *)pAlloc; 1039e8d5648eSdan pAlloc += nName+1; 1040db04571cSdan abPK[i] = sqlite3_column_int(pStmt, 5); 1041e8d5648eSdan i++; 1042e8d5648eSdan } 1043e8d5648eSdan rc = sqlite3_reset(pStmt); 1044e8d5648eSdan 1045e8d5648eSdan } 1046e8d5648eSdan 1047e8d5648eSdan /* If successful, populate the output variables. Otherwise, zero them and 1048e8d5648eSdan ** free any allocation made. An error code will be returned in this case. 1049e8d5648eSdan */ 1050e8d5648eSdan if( rc==SQLITE_OK ){ 1051db04571cSdan *pazCol = (const char **)azCol; 1052db04571cSdan *pabPK = abPK; 1053ca62ad57Sdan *pnCol = nDbCol; 1054e8d5648eSdan }else{ 1055db04571cSdan *pazCol = 0; 1056db04571cSdan *pabPK = 0; 1057ca62ad57Sdan *pnCol = 0; 1058e8d5648eSdan if( pzTab ) *pzTab = 0; 1059db04571cSdan sqlite3_free(azCol); 1060e8d5648eSdan } 1061e8d5648eSdan sqlite3_finalize(pStmt); 1062e8d5648eSdan return rc; 1063e8d5648eSdan } 1064e8d5648eSdan 1065e8d5648eSdan /* 1066296c7658Sdan ** This function is only called from within a pre-update handler for a 1067296c7658Sdan ** write to table pTab, part of session pSession. If this is the first 10686dc29e60Sdan ** write to this table, initalize the SessionTable.nCol, azCol[] and 10696dc29e60Sdan ** abPK[] arrays accordingly. 1070296c7658Sdan ** 10716dc29e60Sdan ** If an error occurs, an error code is stored in sqlite3_session.rc and 10726dc29e60Sdan ** non-zero returned. Or, if no error occurs but the table has no primary 10736dc29e60Sdan ** key, sqlite3_session.rc is left set to SQLITE_OK and non-zero returned to 10746dc29e60Sdan ** indicate that updates on this table should be ignored. SessionTable.abPK 10756dc29e60Sdan ** is set to NULL in this case. 1076296c7658Sdan */ 10774fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){ 10784fccf43aSdan if( pTab->nCol==0 ){ 10796dc29e60Sdan u8 *abPK; 1080e8d5648eSdan assert( pTab->azCol==0 || pTab->abPK==0 ); 1081e8d5648eSdan pSession->rc = sessionTableInfo(pSession->db, pSession->zDb, 10826dc29e60Sdan pTab->zName, &pTab->nCol, 0, &pTab->azCol, &abPK 1083e8d5648eSdan ); 10846dc29e60Sdan if( pSession->rc==SQLITE_OK ){ 10856dc29e60Sdan int i; 10866dc29e60Sdan for(i=0; i<pTab->nCol; i++){ 10876dc29e60Sdan if( abPK[i] ){ 10886dc29e60Sdan pTab->abPK = abPK; 10896dc29e60Sdan break; 1090ca62ad57Sdan } 10914fccf43aSdan } 10923739f298Sdan if( 0==sqlite3_stricmp("sqlite_stat1", pTab->zName) ){ 10933739f298Sdan pTab->bStat1 = 1; 10943739f298Sdan } 10956dc29e60Sdan } 10966dc29e60Sdan } 10976dc29e60Sdan return (pSession->rc || pTab->abPK==0); 1098e8d5648eSdan } 1099e8d5648eSdan 110077fc1d5bSdan /* 11011611e5a3Sdan ** Versions of the four methods in object SessionHook for use with the 11021611e5a3Sdan ** sqlite_stat1 table. The purpose of this is to substitute a zero-length 11031611e5a3Sdan ** blob each time a NULL value is read from the "idx" column of the 11041611e5a3Sdan ** sqlite_stat1 table. 11051611e5a3Sdan */ 11061611e5a3Sdan typedef struct SessionStat1Ctx SessionStat1Ctx; 11071611e5a3Sdan struct SessionStat1Ctx { 11081611e5a3Sdan SessionHook hook; 11091611e5a3Sdan sqlite3_session *pSession; 11101611e5a3Sdan }; 11111611e5a3Sdan static int sessionStat1Old(void *pCtx, int iCol, sqlite3_value **ppVal){ 11121611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 11131611e5a3Sdan sqlite3_value *pVal = 0; 11141611e5a3Sdan int rc = p->hook.xOld(p->hook.pCtx, iCol, &pVal); 11151611e5a3Sdan if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){ 11161611e5a3Sdan pVal = p->pSession->pZeroBlob; 11171611e5a3Sdan } 11181611e5a3Sdan *ppVal = pVal; 11191611e5a3Sdan return rc; 11201611e5a3Sdan } 11211611e5a3Sdan static int sessionStat1New(void *pCtx, int iCol, sqlite3_value **ppVal){ 11221611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 11231611e5a3Sdan sqlite3_value *pVal = 0; 11241611e5a3Sdan int rc = p->hook.xNew(p->hook.pCtx, iCol, &pVal); 11251611e5a3Sdan if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){ 11261611e5a3Sdan pVal = p->pSession->pZeroBlob; 11271611e5a3Sdan } 11281611e5a3Sdan *ppVal = pVal; 11291611e5a3Sdan return rc; 11301611e5a3Sdan } 11311611e5a3Sdan static int sessionStat1Count(void *pCtx){ 11321611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 11331611e5a3Sdan return p->hook.xCount(p->hook.pCtx); 11341611e5a3Sdan } 11351611e5a3Sdan static int sessionStat1Depth(void *pCtx){ 11361611e5a3Sdan SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx; 11371611e5a3Sdan return p->hook.xDepth(p->hook.pCtx); 11381611e5a3Sdan } 11391611e5a3Sdan 11401611e5a3Sdan 11411611e5a3Sdan /* 114277fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a 114377fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change 114477fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument. 114577fc1d5bSdan ** 114677fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added 114777fc1d5bSdan ** to the changed-rows hash table associated with table pTab. 114877fc1d5bSdan */ 1149e8d5648eSdan static void sessionPreupdateOneChange( 115077fc1d5bSdan int op, /* One of SQLITE_UPDATE, INSERT, DELETE */ 115177fc1d5bSdan sqlite3_session *pSession, /* Session object pTab is attached to */ 115277fc1d5bSdan SessionTable *pTab /* Table that change applies to */ 1153e8d5648eSdan ){ 1154e8d5648eSdan int iHash; 1155cf8e9144Sdan int bNull = 0; 1156e8d5648eSdan int rc = SQLITE_OK; 1157fa94d492Sdrh SessionStat1Ctx stat1 = {{0,0,0,0,0},0}; 1158e8d5648eSdan 1159e8d5648eSdan if( pSession->rc ) return; 1160e8d5648eSdan 1161e8d5648eSdan /* Load table details if required */ 1162e8d5648eSdan if( sessionInitTable(pSession, pTab) ) return; 1163e8d5648eSdan 11646dc29e60Sdan /* Check the number of columns in this xPreUpdate call matches the 11656dc29e60Sdan ** number of columns in the table. */ 11666dc29e60Sdan if( pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx) ){ 11676dc29e60Sdan pSession->rc = SQLITE_SCHEMA; 11686dc29e60Sdan return; 11696dc29e60Sdan } 11706dc29e60Sdan 1171e8d5648eSdan /* Grow the hash table if required */ 117264277f4aSdan if( sessionGrowHash(0, pTab) ){ 11735d607a6eSdan pSession->rc = SQLITE_NOMEM; 11745d607a6eSdan return; 11755d607a6eSdan } 1176e8d5648eSdan 11771611e5a3Sdan if( pTab->bStat1 ){ 11781611e5a3Sdan stat1.hook = pSession->hook; 11791611e5a3Sdan stat1.pSession = pSession; 11801611e5a3Sdan pSession->hook.pCtx = (void*)&stat1; 11811611e5a3Sdan pSession->hook.xNew = sessionStat1New; 11821611e5a3Sdan pSession->hook.xOld = sessionStat1Old; 11831611e5a3Sdan pSession->hook.xCount = sessionStat1Count; 11841611e5a3Sdan pSession->hook.xDepth = sessionStat1Depth; 11851611e5a3Sdan if( pSession->pZeroBlob==0 ){ 11861611e5a3Sdan sqlite3_value *p = sqlite3ValueNew(0); 11871611e5a3Sdan if( p==0 ){ 11881611e5a3Sdan rc = SQLITE_NOMEM; 11891611e5a3Sdan goto error_out; 11901611e5a3Sdan } 11911611e5a3Sdan sqlite3ValueSetStr(p, 0, "", 0, SQLITE_STATIC); 11921611e5a3Sdan pSession->pZeroBlob = p; 11931611e5a3Sdan } 11941611e5a3Sdan } 11951611e5a3Sdan 119680fe2d93Sdan /* Calculate the hash-key for this change. If the primary key of the row 119780fe2d93Sdan ** includes a NULL value, exit early. Such changes are ignored by the 119880fe2d93Sdan ** session module. */ 1199cf8e9144Sdan rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull); 120080fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 120180fe2d93Sdan 12021611e5a3Sdan if( bNull==0 ){ 120380fe2d93Sdan /* Search the hash table for an existing record for this row. */ 1204b4480e94Sdan SessionChange *pC; 12056734007dSdan for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){ 1206cf8e9144Sdan if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break; 1207e8d5648eSdan } 120880fe2d93Sdan 1209e8d5648eSdan if( pC==0 ){ 1210e8d5648eSdan /* Create a new change object containing all the old values (if 1211e8d5648eSdan ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK 1212e8d5648eSdan ** values (if this is an INSERT). */ 1213b4480e94Sdan SessionChange *pChange; /* New change object */ 12142d77d80aSdrh sqlite3_int64 nByte; /* Number of bytes to allocate */ 1215e8d5648eSdan int i; /* Used to iterate through columns */ 1216e8d5648eSdan 1217b4480e94Sdan assert( rc==SQLITE_OK ); 1218e8d5648eSdan pTab->nEntry++; 1219e8d5648eSdan 1220e8d5648eSdan /* Figure out how large an allocation is required */ 1221e8d5648eSdan nByte = sizeof(SessionChange); 122280fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1223e8d5648eSdan sqlite3_value *p = 0; 1224e8d5648eSdan if( op!=SQLITE_INSERT ){ 1225cf8e9144Sdan TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p); 122680fe2d93Sdan assert( trc==SQLITE_OK ); 122780fe2d93Sdan }else if( pTab->abPK[i] ){ 1228cf8e9144Sdan TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p); 122980fe2d93Sdan assert( trc==SQLITE_OK ); 1230e8d5648eSdan } 123180fe2d93Sdan 123280fe2d93Sdan /* This may fail if SQLite value p contains a utf-16 string that must 123380fe2d93Sdan ** be converted to utf-8 and an OOM error occurs while doing so. */ 1234e8d5648eSdan rc = sessionSerializeValue(0, p, &nByte); 123580fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 1236e8d5648eSdan } 1237e8d5648eSdan 1238e8d5648eSdan /* Allocate the change object */ 12392d77d80aSdrh pChange = (SessionChange *)sqlite3_malloc64(nByte); 1240e8d5648eSdan if( !pChange ){ 1241e8d5648eSdan rc = SQLITE_NOMEM; 124280fe2d93Sdan goto error_out; 1243e8d5648eSdan }else{ 1244e8d5648eSdan memset(pChange, 0, sizeof(SessionChange)); 1245e8d5648eSdan pChange->aRecord = (u8 *)&pChange[1]; 1246e8d5648eSdan } 1247e8d5648eSdan 124880fe2d93Sdan /* Populate the change object. None of the preupdate_old(), 124980fe2d93Sdan ** preupdate_new() or SerializeValue() calls below may fail as all 125080fe2d93Sdan ** required values and encodings have already been cached in memory. 125180fe2d93Sdan ** It is not possible for an OOM to occur in this block. */ 1252e8d5648eSdan nByte = 0; 125380fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1254e8d5648eSdan sqlite3_value *p = 0; 1255e8d5648eSdan if( op!=SQLITE_INSERT ){ 1256cf8e9144Sdan pSession->hook.xOld(pSession->hook.pCtx, i, &p); 125780fe2d93Sdan }else if( pTab->abPK[i] ){ 1258cf8e9144Sdan pSession->hook.xNew(pSession->hook.pCtx, i, &p); 1259e8d5648eSdan } 126080fe2d93Sdan sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte); 1261e8d5648eSdan } 126280fe2d93Sdan 126380fe2d93Sdan /* Add the change to the hash-table */ 1264cf8e9144Sdan if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){ 1265b4480e94Sdan pChange->bIndirect = 1; 1266b4480e94Sdan } 126712ca0b56Sdan pChange->nRecord = nByte; 1268798693b2Sdan pChange->op = op; 1269e8d5648eSdan pChange->pNext = pTab->apChange[iHash]; 1270e8d5648eSdan pTab->apChange[iHash] = pChange; 127180fe2d93Sdan 127280fe2d93Sdan }else if( pC->bIndirect ){ 1273b4480e94Sdan /* If the existing change is considered "indirect", but this current 1274b4480e94Sdan ** change is "direct", mark the change object as direct. */ 1275cf8e9144Sdan if( pSession->hook.xDepth(pSession->hook.pCtx)==0 1276cf8e9144Sdan && pSession->bIndirect==0 1277cf8e9144Sdan ){ 1278b4480e94Sdan pC->bIndirect = 0; 1279b4480e94Sdan } 1280e8d5648eSdan } 12814fccf43aSdan } 128212ca0b56Sdan 128312ca0b56Sdan /* If an error has occurred, mark the session object as failed. */ 128480fe2d93Sdan error_out: 12851611e5a3Sdan if( pTab->bStat1 ){ 12861611e5a3Sdan pSession->hook = stat1.hook; 12871611e5a3Sdan } 128812ca0b56Sdan if( rc!=SQLITE_OK ){ 128912ca0b56Sdan pSession->rc = rc; 129012ca0b56Sdan } 129127453faeSdan } 12924fccf43aSdan 1293cf8e9144Sdan static int sessionFindTable( 1294cf8e9144Sdan sqlite3_session *pSession, 1295cf8e9144Sdan const char *zName, 1296cf8e9144Sdan SessionTable **ppTab 1297cf8e9144Sdan ){ 1298cf8e9144Sdan int rc = SQLITE_OK; 1299cf8e9144Sdan int nName = sqlite3Strlen30(zName); 1300cf8e9144Sdan SessionTable *pRet; 1301cf8e9144Sdan 1302cf8e9144Sdan /* Search for an existing table */ 1303cf8e9144Sdan for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){ 1304cf8e9144Sdan if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break; 1305cf8e9144Sdan } 1306cf8e9144Sdan 1307cf8e9144Sdan if( pRet==0 && pSession->bAutoAttach ){ 1308cf8e9144Sdan /* If there is a table-filter configured, invoke it. If it returns 0, 1309cf8e9144Sdan ** do not automatically add the new table. */ 1310cf8e9144Sdan if( pSession->xTableFilter==0 1311cf8e9144Sdan || pSession->xTableFilter(pSession->pFilterCtx, zName) 1312cf8e9144Sdan ){ 1313cf8e9144Sdan rc = sqlite3session_attach(pSession, zName); 1314cf8e9144Sdan if( rc==SQLITE_OK ){ 13156c39e6a8Sdan for(pRet=pSession->pTable; pRet->pNext; pRet=pRet->pNext); 1316cf8e9144Sdan assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ); 1317cf8e9144Sdan } 1318cf8e9144Sdan } 1319cf8e9144Sdan } 1320cf8e9144Sdan 1321cf8e9144Sdan assert( rc==SQLITE_OK || pRet==0 ); 1322cf8e9144Sdan *ppTab = pRet; 1323cf8e9144Sdan return rc; 1324cf8e9144Sdan } 1325cf8e9144Sdan 13264fccf43aSdan /* 13274fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases. 13284fccf43aSdan */ 13294fccf43aSdan static void xPreUpdate( 13304fccf43aSdan void *pCtx, /* Copy of third arg to preupdate_hook() */ 13314fccf43aSdan sqlite3 *db, /* Database handle */ 13324fccf43aSdan int op, /* SQLITE_UPDATE, DELETE or INSERT */ 13334fccf43aSdan char const *zDb, /* Database name */ 13344fccf43aSdan char const *zName, /* Table name */ 13354fccf43aSdan sqlite3_int64 iKey1, /* Rowid of row about to be deleted/updated */ 13364fccf43aSdan sqlite3_int64 iKey2 /* New rowid value (for a rowid UPDATE) */ 13374fccf43aSdan ){ 13384fccf43aSdan sqlite3_session *pSession; 1339cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); 13404fccf43aSdan 13414c220252Sdan assert( sqlite3_mutex_held(db->mutex) ); 13424c220252Sdan 13434fccf43aSdan for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){ 13444fccf43aSdan SessionTable *pTab; 1345296c7658Sdan 1346e8d5648eSdan /* If this session is attached to a different database ("main", "temp" 1347e8d5648eSdan ** etc.), or if it is not currently enabled, there is nothing to do. Skip 1348e8d5648eSdan ** to the next session object attached to this database. */ 1349296c7658Sdan if( pSession->bEnable==0 ) continue; 13504fccf43aSdan if( pSession->rc ) continue; 13514fccf43aSdan if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue; 1352296c7658Sdan 1353cf8e9144Sdan pSession->rc = sessionFindTable(pSession, zName, &pTab); 1354cf8e9144Sdan if( pTab ){ 1355cf8e9144Sdan assert( pSession->rc==SQLITE_OK ); 1356e8d5648eSdan sessionPreupdateOneChange(op, pSession, pTab); 1357e8d5648eSdan if( op==SQLITE_UPDATE ){ 1358e8d5648eSdan sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab); 13594fccf43aSdan } 13604fccf43aSdan } 13614fccf43aSdan } 13624fccf43aSdan } 1363cf8e9144Sdan 1364cf8e9144Sdan /* 1365cf8e9144Sdan ** The pre-update hook implementations. 1366cf8e9144Sdan */ 1367cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){ 1368cf8e9144Sdan return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal); 1369cf8e9144Sdan } 1370cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){ 1371cf8e9144Sdan return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal); 1372cf8e9144Sdan } 1373cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){ 1374cf8e9144Sdan return sqlite3_preupdate_count((sqlite3*)pCtx); 1375cf8e9144Sdan } 1376cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){ 1377cf8e9144Sdan return sqlite3_preupdate_depth((sqlite3*)pCtx); 1378cf8e9144Sdan } 1379cf8e9144Sdan 1380cf8e9144Sdan /* 1381cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only 1382cf8e9144Sdan ** argument. 1383cf8e9144Sdan */ 1384cf8e9144Sdan static void sessionPreupdateHooks( 1385cf8e9144Sdan sqlite3_session *pSession 1386cf8e9144Sdan ){ 1387cf8e9144Sdan pSession->hook.pCtx = (void*)pSession->db; 1388cf8e9144Sdan pSession->hook.xOld = sessionPreupdateOld; 1389cf8e9144Sdan pSession->hook.xNew = sessionPreupdateNew; 1390cf8e9144Sdan pSession->hook.xCount = sessionPreupdateCount; 1391cf8e9144Sdan pSession->hook.xDepth = sessionPreupdateDepth; 1392cf8e9144Sdan } 1393cf8e9144Sdan 1394cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx; 1395cf8e9144Sdan struct SessionDiffCtx { 1396cf8e9144Sdan sqlite3_stmt *pStmt; 1397cf8e9144Sdan int nOldOff; 1398cf8e9144Sdan }; 1399cf8e9144Sdan 1400cf8e9144Sdan /* 1401cf8e9144Sdan ** The diff hook implementations. 1402cf8e9144Sdan */ 1403cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){ 1404cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1405cf8e9144Sdan *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff); 1406cf8e9144Sdan return SQLITE_OK; 1407cf8e9144Sdan } 1408cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){ 1409cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1410cf8e9144Sdan *ppVal = sqlite3_column_value(p->pStmt, iVal); 1411cf8e9144Sdan return SQLITE_OK; 1412cf8e9144Sdan } 1413cf8e9144Sdan static int sessionDiffCount(void *pCtx){ 1414cf8e9144Sdan SessionDiffCtx *p = (SessionDiffCtx*)pCtx; 1415cf8e9144Sdan return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt); 1416cf8e9144Sdan } 1417cf8e9144Sdan static int sessionDiffDepth(void *pCtx){ 1418cf8e9144Sdan return 0; 1419cf8e9144Sdan } 1420cf8e9144Sdan 1421cf8e9144Sdan /* 1422cf8e9144Sdan ** Install the diff hooks on the session object passed as the only 1423cf8e9144Sdan ** argument. 1424cf8e9144Sdan */ 1425cf8e9144Sdan static void sessionDiffHooks( 1426cf8e9144Sdan sqlite3_session *pSession, 1427cf8e9144Sdan SessionDiffCtx *pDiffCtx 1428cf8e9144Sdan ){ 1429cf8e9144Sdan pSession->hook.pCtx = (void*)pDiffCtx; 1430cf8e9144Sdan pSession->hook.xOld = sessionDiffOld; 1431cf8e9144Sdan pSession->hook.xNew = sessionDiffNew; 1432cf8e9144Sdan pSession->hook.xCount = sessionDiffCount; 1433cf8e9144Sdan pSession->hook.xDepth = sessionDiffDepth; 1434cf8e9144Sdan } 1435cf8e9144Sdan 1436cf8e9144Sdan static char *sessionExprComparePK( 1437cf8e9144Sdan int nCol, 1438cf8e9144Sdan const char *zDb1, const char *zDb2, 1439cf8e9144Sdan const char *zTab, 1440cf8e9144Sdan const char **azCol, u8 *abPK 1441cf8e9144Sdan ){ 1442cf8e9144Sdan int i; 1443cf8e9144Sdan const char *zSep = ""; 1444cf8e9144Sdan char *zRet = 0; 1445cf8e9144Sdan 1446cf8e9144Sdan for(i=0; i<nCol; i++){ 1447cf8e9144Sdan if( abPK[i] ){ 1448cf8e9144Sdan zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"", 1449cf8e9144Sdan zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i] 1450cf8e9144Sdan ); 1451cf8e9144Sdan zSep = " AND "; 1452cf8e9144Sdan if( zRet==0 ) break; 1453cf8e9144Sdan } 1454cf8e9144Sdan } 1455cf8e9144Sdan 1456cf8e9144Sdan return zRet; 1457cf8e9144Sdan } 1458cf8e9144Sdan 1459cf8e9144Sdan static char *sessionExprCompareOther( 1460cf8e9144Sdan int nCol, 1461cf8e9144Sdan const char *zDb1, const char *zDb2, 1462cf8e9144Sdan const char *zTab, 1463cf8e9144Sdan const char **azCol, u8 *abPK 1464cf8e9144Sdan ){ 1465cf8e9144Sdan int i; 1466cf8e9144Sdan const char *zSep = ""; 1467cf8e9144Sdan char *zRet = 0; 1468cf8e9144Sdan int bHave = 0; 1469cf8e9144Sdan 1470cf8e9144Sdan for(i=0; i<nCol; i++){ 1471cf8e9144Sdan if( abPK[i]==0 ){ 1472cf8e9144Sdan bHave = 1; 1473cf8e9144Sdan zRet = sqlite3_mprintf( 1474cf8e9144Sdan "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"", 1475cf8e9144Sdan zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i] 1476cf8e9144Sdan ); 1477cf8e9144Sdan zSep = " OR "; 1478cf8e9144Sdan if( zRet==0 ) break; 1479cf8e9144Sdan } 1480cf8e9144Sdan } 1481cf8e9144Sdan 1482cf8e9144Sdan if( bHave==0 ){ 1483cf8e9144Sdan assert( zRet==0 ); 1484cf8e9144Sdan zRet = sqlite3_mprintf("0"); 1485cf8e9144Sdan } 1486cf8e9144Sdan 1487cf8e9144Sdan return zRet; 1488cf8e9144Sdan } 1489cf8e9144Sdan 1490cf8e9144Sdan static char *sessionSelectFindNew( 1491cf8e9144Sdan int nCol, 1492cf8e9144Sdan const char *zDb1, /* Pick rows in this db only */ 1493cf8e9144Sdan const char *zDb2, /* But not in this one */ 1494cf8e9144Sdan const char *zTbl, /* Table name */ 1495cf8e9144Sdan const char *zExpr 1496cf8e9144Sdan ){ 1497cf8e9144Sdan char *zRet = sqlite3_mprintf( 1498cf8e9144Sdan "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS (" 1499cf8e9144Sdan " SELECT 1 FROM \"%w\".\"%w\" WHERE %s" 1500cf8e9144Sdan ")", 1501cf8e9144Sdan zDb1, zTbl, zDb2, zTbl, zExpr 1502cf8e9144Sdan ); 1503cf8e9144Sdan return zRet; 1504cf8e9144Sdan } 1505cf8e9144Sdan 1506cf8e9144Sdan static int sessionDiffFindNew( 1507cf8e9144Sdan int op, 1508cf8e9144Sdan sqlite3_session *pSession, 1509cf8e9144Sdan SessionTable *pTab, 1510cf8e9144Sdan const char *zDb1, 1511cf8e9144Sdan const char *zDb2, 1512cf8e9144Sdan char *zExpr 1513cf8e9144Sdan ){ 1514cf8e9144Sdan int rc = SQLITE_OK; 1515cf8e9144Sdan char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr); 1516cf8e9144Sdan 1517cf8e9144Sdan if( zStmt==0 ){ 1518cf8e9144Sdan rc = SQLITE_NOMEM; 1519cf8e9144Sdan }else{ 1520cf8e9144Sdan sqlite3_stmt *pStmt; 1521cf8e9144Sdan rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0); 1522cf8e9144Sdan if( rc==SQLITE_OK ){ 1523cf8e9144Sdan SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx; 1524cf8e9144Sdan pDiffCtx->pStmt = pStmt; 1525cf8e9144Sdan pDiffCtx->nOldOff = 0; 1526cf8e9144Sdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1527cf8e9144Sdan sessionPreupdateOneChange(op, pSession, pTab); 1528cf8e9144Sdan } 1529cf8e9144Sdan rc = sqlite3_finalize(pStmt); 1530cf8e9144Sdan } 1531cf8e9144Sdan sqlite3_free(zStmt); 1532cf8e9144Sdan } 1533cf8e9144Sdan 1534cf8e9144Sdan return rc; 1535cf8e9144Sdan } 1536cf8e9144Sdan 1537cf8e9144Sdan static int sessionDiffFindModified( 1538cf8e9144Sdan sqlite3_session *pSession, 1539cf8e9144Sdan SessionTable *pTab, 1540cf8e9144Sdan const char *zFrom, 1541cf8e9144Sdan const char *zExpr 1542cf8e9144Sdan ){ 1543cf8e9144Sdan int rc = SQLITE_OK; 1544cf8e9144Sdan 1545cf8e9144Sdan char *zExpr2 = sessionExprCompareOther(pTab->nCol, 1546cf8e9144Sdan pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK 1547cf8e9144Sdan ); 1548cf8e9144Sdan if( zExpr2==0 ){ 1549cf8e9144Sdan rc = SQLITE_NOMEM; 1550cf8e9144Sdan }else{ 1551cf8e9144Sdan char *zStmt = sqlite3_mprintf( 1552dd009f83Sdan "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)", 1553cf8e9144Sdan pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2 1554cf8e9144Sdan ); 1555cf8e9144Sdan if( zStmt==0 ){ 1556cf8e9144Sdan rc = SQLITE_NOMEM; 1557cf8e9144Sdan }else{ 1558cf8e9144Sdan sqlite3_stmt *pStmt; 1559cf8e9144Sdan rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0); 1560cf8e9144Sdan 1561cf8e9144Sdan if( rc==SQLITE_OK ){ 1562cf8e9144Sdan SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx; 1563cf8e9144Sdan pDiffCtx->pStmt = pStmt; 1564cf8e9144Sdan pDiffCtx->nOldOff = pTab->nCol; 1565cf8e9144Sdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 1566cf8e9144Sdan sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab); 1567cf8e9144Sdan } 1568cf8e9144Sdan rc = sqlite3_finalize(pStmt); 1569cf8e9144Sdan } 1570cf8e9144Sdan sqlite3_free(zStmt); 1571cf8e9144Sdan } 1572cf8e9144Sdan } 1573cf8e9144Sdan 1574cf8e9144Sdan return rc; 1575cf8e9144Sdan } 1576cf8e9144Sdan 1577cf8e9144Sdan int sqlite3session_diff( 1578cf8e9144Sdan sqlite3_session *pSession, 1579cf8e9144Sdan const char *zFrom, 1580cf8e9144Sdan const char *zTbl, 1581cf8e9144Sdan char **pzErrMsg 1582cf8e9144Sdan ){ 1583cf8e9144Sdan const char *zDb = pSession->zDb; 1584cf8e9144Sdan int rc = pSession->rc; 1585cf8e9144Sdan SessionDiffCtx d; 1586cf8e9144Sdan 1587cf8e9144Sdan memset(&d, 0, sizeof(d)); 1588cf8e9144Sdan sessionDiffHooks(pSession, &d); 1589cf8e9144Sdan 159010dc553cSdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1591cf8e9144Sdan if( pzErrMsg ) *pzErrMsg = 0; 1592cf8e9144Sdan if( rc==SQLITE_OK ){ 1593cf8e9144Sdan char *zExpr = 0; 1594cf8e9144Sdan sqlite3 *db = pSession->db; 1595cf8e9144Sdan SessionTable *pTo; /* Table zTbl */ 1596cf8e9144Sdan 1597cf8e9144Sdan /* Locate and if necessary initialize the target table object */ 1598cf8e9144Sdan rc = sessionFindTable(pSession, zTbl, &pTo); 1599cf8e9144Sdan if( pTo==0 ) goto diff_out; 16006dc29e60Sdan if( sessionInitTable(pSession, pTo) ){ 16016dc29e60Sdan rc = pSession->rc; 16026dc29e60Sdan goto diff_out; 1603cf8e9144Sdan } 1604cf8e9144Sdan 1605cf8e9144Sdan /* Check the table schemas match */ 1606cf8e9144Sdan if( rc==SQLITE_OK ){ 16074cc923e3Sdan int bHasPk = 0; 1608b9db9099Sdan int bMismatch = 0; 1609cf8e9144Sdan int nCol; /* Columns in zFrom.zTbl */ 1610cf8e9144Sdan u8 *abPK; 1611cf8e9144Sdan const char **azCol = 0; 1612cf8e9144Sdan rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK); 1613cf8e9144Sdan if( rc==SQLITE_OK ){ 16144cc923e3Sdan if( pTo->nCol!=nCol ){ 1615cf8e9144Sdan bMismatch = 1; 1616cf8e9144Sdan }else{ 1617cf8e9144Sdan int i; 1618cf8e9144Sdan for(i=0; i<nCol; i++){ 16194cc923e3Sdan if( pTo->abPK[i]!=abPK[i] ) bMismatch = 1; 1620cf8e9144Sdan if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1; 16214cc923e3Sdan if( abPK[i] ) bHasPk = 1; 1622cf8e9144Sdan } 1623cf8e9144Sdan } 1624cf8e9144Sdan } 1625dbbd8160Sdrh sqlite3_free((char*)azCol); 1626b9db9099Sdan if( bMismatch ){ 1627b9db9099Sdan *pzErrMsg = sqlite3_mprintf("table schemas do not match"); 1628b9db9099Sdan rc = SQLITE_SCHEMA; 1629b9db9099Sdan } 1630b9db9099Sdan if( bHasPk==0 ){ 1631b9db9099Sdan /* Ignore tables with no primary keys */ 1632b9db9099Sdan goto diff_out; 1633b9db9099Sdan } 1634cf8e9144Sdan } 1635cf8e9144Sdan 1636cf8e9144Sdan if( rc==SQLITE_OK ){ 1637cf8e9144Sdan zExpr = sessionExprComparePK(pTo->nCol, 1638cf8e9144Sdan zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK 1639cf8e9144Sdan ); 1640cf8e9144Sdan } 1641cf8e9144Sdan 1642cf8e9144Sdan /* Find new rows */ 1643cf8e9144Sdan if( rc==SQLITE_OK ){ 1644cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr); 1645cf8e9144Sdan } 1646cf8e9144Sdan 1647cf8e9144Sdan /* Find old rows */ 1648cf8e9144Sdan if( rc==SQLITE_OK ){ 1649cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr); 1650cf8e9144Sdan } 1651cf8e9144Sdan 1652cf8e9144Sdan /* Find modified rows */ 1653cf8e9144Sdan if( rc==SQLITE_OK ){ 1654cf8e9144Sdan rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr); 1655cf8e9144Sdan } 1656cf8e9144Sdan 1657cf8e9144Sdan sqlite3_free(zExpr); 1658cf8e9144Sdan } 1659cf8e9144Sdan 1660cf8e9144Sdan diff_out: 1661cf8e9144Sdan sessionPreupdateHooks(pSession); 166210dc553cSdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 1663cf8e9144Sdan return rc; 1664296c7658Sdan } 16654fccf43aSdan 16664fccf43aSdan /* 16674fccf43aSdan ** Create a session object. This session object will record changes to 16684fccf43aSdan ** database zDb attached to connection db. 16694fccf43aSdan */ 16704fccf43aSdan int sqlite3session_create( 16714fccf43aSdan sqlite3 *db, /* Database handle */ 16724fccf43aSdan const char *zDb, /* Name of db (e.g. "main") */ 16734fccf43aSdan sqlite3_session **ppSession /* OUT: New session object */ 16744fccf43aSdan ){ 1675296c7658Sdan sqlite3_session *pNew; /* Newly allocated session object */ 1676296c7658Sdan sqlite3_session *pOld; /* Session object already attached to db */ 1677cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */ 16784fccf43aSdan 1679296c7658Sdan /* Zero the output value in case an error occurs. */ 16804fccf43aSdan *ppSession = 0; 16814fccf43aSdan 16824fccf43aSdan /* Allocate and populate the new session object. */ 16832d77d80aSdrh pNew = (sqlite3_session *)sqlite3_malloc64(sizeof(sqlite3_session) + nDb + 1); 16844fccf43aSdan if( !pNew ) return SQLITE_NOMEM; 16854fccf43aSdan memset(pNew, 0, sizeof(sqlite3_session)); 16864fccf43aSdan pNew->db = db; 16874fccf43aSdan pNew->zDb = (char *)&pNew[1]; 1688296c7658Sdan pNew->bEnable = 1; 16894fccf43aSdan memcpy(pNew->zDb, zDb, nDb+1); 1690cf8e9144Sdan sessionPreupdateHooks(pNew); 16914fccf43aSdan 16924fccf43aSdan /* Add the new session object to the linked list of session objects 16934fccf43aSdan ** attached to database handle $db. Do this under the cover of the db 16944fccf43aSdan ** handle mutex. */ 16954fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 16964fccf43aSdan pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew); 16974fccf43aSdan pNew->pNext = pOld; 16984fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 16994fccf43aSdan 17004fccf43aSdan *ppSession = pNew; 17014fccf43aSdan return SQLITE_OK; 17024fccf43aSdan } 17034fccf43aSdan 170477fc1d5bSdan /* 170577fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents 170677fc1d5bSdan ** of the changed-rows hash tables are also deleted. 170777fc1d5bSdan */ 17081ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){ 17095d607a6eSdan SessionTable *pNext; 17105d607a6eSdan SessionTable *pTab; 17115d607a6eSdan 17125d607a6eSdan for(pTab=pList; pTab; pTab=pNext){ 17135d607a6eSdan int i; 17145d607a6eSdan pNext = pTab->pNext; 17155d607a6eSdan for(i=0; i<pTab->nChange; i++){ 17165d607a6eSdan SessionChange *p; 171702d436b1Smistachkin SessionChange *pNextChange; 171802d436b1Smistachkin for(p=pTab->apChange[i]; p; p=pNextChange){ 171902d436b1Smistachkin pNextChange = p->pNext; 17205d607a6eSdan sqlite3_free(p); 17215d607a6eSdan } 17225d607a6eSdan } 17235d607a6eSdan sqlite3_free((char*)pTab->azCol); /* cast works around VC++ bug */ 17245d607a6eSdan sqlite3_free(pTab->apChange); 17255d607a6eSdan sqlite3_free(pTab); 17265d607a6eSdan } 17275d607a6eSdan } 17285d607a6eSdan 17294fccf43aSdan /* 17304fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create(). 17314fccf43aSdan */ 17324fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){ 17334fccf43aSdan sqlite3 *db = pSession->db; 17344fccf43aSdan sqlite3_session *pHead; 17354fccf43aSdan sqlite3_session **pp; 17364fccf43aSdan 1737296c7658Sdan /* Unlink the session from the linked list of sessions attached to the 1738296c7658Sdan ** database handle. Hold the db mutex while doing so. */ 17394fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 17404fccf43aSdan pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0); 174150d348b1Sdrh for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){ 174250d348b1Sdrh if( (*pp)==pSession ){ 17434fccf43aSdan *pp = (*pp)->pNext; 17444fccf43aSdan if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead); 174550d348b1Sdrh break; 174650d348b1Sdrh } 174750d348b1Sdrh } 17484fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 17491611e5a3Sdan sqlite3ValueFree(pSession->pZeroBlob); 17504fccf43aSdan 1751296c7658Sdan /* Delete all attached table objects. And the contents of their 1752296c7658Sdan ** associated hash-tables. */ 17535d607a6eSdan sessionDeleteTable(pSession->pTable); 17544fccf43aSdan 1755296c7658Sdan /* Free the session object itself. */ 17564fccf43aSdan sqlite3_free(pSession); 17574fccf43aSdan } 17584fccf43aSdan 17594fccf43aSdan /* 17607531a5a3Sdan ** Set a table filter on a Session Object. 17617531a5a3Sdan */ 17627531a5a3Sdan void sqlite3session_table_filter( 17637531a5a3Sdan sqlite3_session *pSession, 17647531a5a3Sdan int(*xFilter)(void*, const char*), 17657531a5a3Sdan void *pCtx /* First argument passed to xFilter */ 17667531a5a3Sdan ){ 17677531a5a3Sdan pSession->bAutoAttach = 1; 17687531a5a3Sdan pSession->pFilterCtx = pCtx; 17697531a5a3Sdan pSession->xTableFilter = xFilter; 17707531a5a3Sdan } 17717531a5a3Sdan 17727531a5a3Sdan /* 17734fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table 17744fccf43aSdan ** while the session object is enabled will be recorded. 17754fccf43aSdan ** 17764fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does 17774fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias) 17784fccf43aSdan ** or not. 17794fccf43aSdan */ 17804fccf43aSdan int sqlite3session_attach( 17814fccf43aSdan sqlite3_session *pSession, /* Session object */ 17824fccf43aSdan const char *zName /* Table name */ 17834fccf43aSdan ){ 1784ff4d0f41Sdan int rc = SQLITE_OK; 1785ff4d0f41Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1786ff4d0f41Sdan 1787ff4d0f41Sdan if( !zName ){ 1788ff4d0f41Sdan pSession->bAutoAttach = 1; 1789ff4d0f41Sdan }else{ 1790296c7658Sdan SessionTable *pTab; /* New table object (if required) */ 1791296c7658Sdan int nName; /* Number of bytes in string zName */ 17924fccf43aSdan 17934fccf43aSdan /* First search for an existing entry. If one is found, this call is 17944fccf43aSdan ** a no-op. Return early. */ 1795cfdbde21Sdrh nName = sqlite3Strlen30(zName); 17964fccf43aSdan for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){ 17974c220252Sdan if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break; 17984fccf43aSdan } 17994fccf43aSdan 18004c220252Sdan if( !pTab ){ 18014fccf43aSdan /* Allocate new SessionTable object. */ 18022d77d80aSdrh pTab = (SessionTable *)sqlite3_malloc64(sizeof(SessionTable) + nName + 1); 18034c220252Sdan if( !pTab ){ 18044c220252Sdan rc = SQLITE_NOMEM; 18054c220252Sdan }else{ 18066c39e6a8Sdan /* Populate the new SessionTable object and link it into the list. 18076c39e6a8Sdan ** The new object must be linked onto the end of the list, not 18086c39e6a8Sdan ** simply added to the start of it in order to ensure that tables 18096c39e6a8Sdan ** appear in the correct order when a changeset or patchset is 18106c39e6a8Sdan ** eventually generated. */ 18116c39e6a8Sdan SessionTable **ppTab; 18124fccf43aSdan memset(pTab, 0, sizeof(SessionTable)); 18134fccf43aSdan pTab->zName = (char *)&pTab[1]; 18144fccf43aSdan memcpy(pTab->zName, zName, nName+1); 18156c39e6a8Sdan for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext); 18166c39e6a8Sdan *ppTab = pTab; 18174c220252Sdan } 18184c220252Sdan } 1819ff4d0f41Sdan } 18204fccf43aSdan 18214c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 18224c220252Sdan return rc; 18234fccf43aSdan } 18244fccf43aSdan 1825296c7658Sdan /* 1826296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data. 1827296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is. 1828296c7658Sdan ** 1829296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered, 1830296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero. 1831296c7658Sdan */ 1832f6ad201aSdrh static int sessionBufferGrow(SessionBuffer *p, size_t nByte, int *pRc){ 1833*3196abb7Smistachkin if( *pRc==SQLITE_OK && (size_t)(p->nAlloc-p->nBuf)<nByte ){ 18344fccf43aSdan u8 *aNew; 18359c18ef09Sdan i64 nNew = p->nAlloc ? p->nAlloc : 128; 18364fccf43aSdan do { 18374fccf43aSdan nNew = nNew*2; 18389c18ef09Sdan }while( (nNew-p->nBuf)<nByte ); 18394fccf43aSdan 18409c18ef09Sdan aNew = (u8 *)sqlite3_realloc64(p->aBuf, nNew); 18414fccf43aSdan if( 0==aNew ){ 18424fccf43aSdan *pRc = SQLITE_NOMEM; 184380fe2d93Sdan }else{ 18444fccf43aSdan p->aBuf = aNew; 18454fccf43aSdan p->nAlloc = nNew; 18464fccf43aSdan } 184780fe2d93Sdan } 184880fe2d93Sdan return (*pRc!=SQLITE_OK); 18494fccf43aSdan } 18504fccf43aSdan 1851296c7658Sdan /* 1852fa122adaSdan ** Append the value passed as the second argument to the buffer passed 1853fa122adaSdan ** as the first. 1854fa122adaSdan ** 1855fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called. 1856fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code 1857fa122adaSdan ** before returning. 1858fa122adaSdan */ 1859fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){ 1860fa122adaSdan int rc = *pRc; 1861fa122adaSdan if( rc==SQLITE_OK ){ 18622d77d80aSdrh sqlite3_int64 nByte = 0; 1863e8fa8c96Sdan rc = sessionSerializeValue(0, pVal, &nByte); 1864fa122adaSdan sessionBufferGrow(p, nByte, &rc); 1865fa122adaSdan if( rc==SQLITE_OK ){ 1866fa122adaSdan rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0); 1867fa122adaSdan p->nBuf += nByte; 1868fa122adaSdan }else{ 1869fa122adaSdan *pRc = rc; 1870fa122adaSdan } 1871fa122adaSdan } 1872fa122adaSdan } 1873fa122adaSdan 1874fa122adaSdan /* 1875296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1876296c7658Sdan ** called. Otherwise, append a single byte to the buffer. 1877296c7658Sdan ** 1878296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1879296c7658Sdan ** returning. 1880296c7658Sdan */ 18814fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){ 188280fe2d93Sdan if( 0==sessionBufferGrow(p, 1, pRc) ){ 18834fccf43aSdan p->aBuf[p->nBuf++] = v; 18844fccf43aSdan } 18854fccf43aSdan } 18864fccf43aSdan 1887296c7658Sdan /* 1888296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1889296c7658Sdan ** called. Otherwise, append a single varint to the buffer. 1890296c7658Sdan ** 1891296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1892296c7658Sdan ** returning. 1893296c7658Sdan */ 1894cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){ 189580fe2d93Sdan if( 0==sessionBufferGrow(p, 9, pRc) ){ 18964fccf43aSdan p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v); 18974fccf43aSdan } 18984fccf43aSdan } 18994fccf43aSdan 1900296c7658Sdan /* 1901296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1902296c7658Sdan ** called. Otherwise, append a blob of data to the buffer. 1903296c7658Sdan ** 1904296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1905296c7658Sdan ** returning. 1906296c7658Sdan */ 19074fccf43aSdan static void sessionAppendBlob( 19084fccf43aSdan SessionBuffer *p, 19094fccf43aSdan const u8 *aBlob, 19104fccf43aSdan int nBlob, 19114fccf43aSdan int *pRc 19124fccf43aSdan ){ 1913895decf6Sdan if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){ 19144fccf43aSdan memcpy(&p->aBuf[p->nBuf], aBlob, nBlob); 19154fccf43aSdan p->nBuf += nBlob; 19164fccf43aSdan } 19174fccf43aSdan } 19184fccf43aSdan 1919296c7658Sdan /* 1920296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1921296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string 1922296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer. 1923296c7658Sdan ** 1924296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1925296c7658Sdan ** returning. 1926296c7658Sdan */ 1927d5f0767cSdan static void sessionAppendStr( 1928d5f0767cSdan SessionBuffer *p, 1929d5f0767cSdan const char *zStr, 1930d5f0767cSdan int *pRc 1931d5f0767cSdan ){ 1932cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr); 193380fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1934d5f0767cSdan memcpy(&p->aBuf[p->nBuf], zStr, nStr); 1935d5f0767cSdan p->nBuf += nStr; 1936d5f0767cSdan } 1937d5f0767cSdan } 1938d5f0767cSdan 1939296c7658Sdan /* 1940296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1941296c7658Sdan ** called. Otherwise, append the string representation of integer iVal 1942296c7658Sdan ** to the buffer. No nul-terminator is written. 1943296c7658Sdan ** 1944296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1945296c7658Sdan ** returning. 1946296c7658Sdan */ 1947d5f0767cSdan static void sessionAppendInteger( 1948296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1949296c7658Sdan int iVal, /* Value to write the string rep. of */ 1950296c7658Sdan int *pRc /* IN/OUT: Error code */ 1951d5f0767cSdan ){ 1952d5f0767cSdan char aBuf[24]; 1953d5f0767cSdan sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal); 1954d5f0767cSdan sessionAppendStr(p, aBuf, pRc); 1955d5f0767cSdan } 1956d5f0767cSdan 1957296c7658Sdan /* 1958296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1959296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and 1960296c7658Sdan ** with any embedded quote characters escaped to the buffer. No 1961296c7658Sdan ** nul-terminator byte is written. 1962296c7658Sdan ** 1963296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1964296c7658Sdan ** returning. 1965296c7658Sdan */ 1966d5f0767cSdan static void sessionAppendIdent( 1967296c7658Sdan SessionBuffer *p, /* Buffer to a append to */ 1968296c7658Sdan const char *zStr, /* String to quote, escape and append */ 1969296c7658Sdan int *pRc /* IN/OUT: Error code */ 1970d5f0767cSdan ){ 1971cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1; 197280fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1973d5f0767cSdan char *zOut = (char *)&p->aBuf[p->nBuf]; 1974d5f0767cSdan const char *zIn = zStr; 1975d5f0767cSdan *zOut++ = '"'; 1976d5f0767cSdan while( *zIn ){ 1977d5f0767cSdan if( *zIn=='"' ) *zOut++ = '"'; 1978d5f0767cSdan *zOut++ = *(zIn++); 1979d5f0767cSdan } 1980d5f0767cSdan *zOut++ = '"'; 1981cfdbde21Sdrh p->nBuf = (int)((u8 *)zOut - p->aBuf); 1982d5f0767cSdan } 1983d5f0767cSdan } 1984d5f0767cSdan 1985296c7658Sdan /* 1986296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1987296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored 1988296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points 1989296c7658Sdan ** to to the buffer. 1990296c7658Sdan */ 19914fccf43aSdan static void sessionAppendCol( 1992296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1993296c7658Sdan sqlite3_stmt *pStmt, /* Handle pointing to row containing value */ 1994296c7658Sdan int iCol, /* Column to read value from */ 1995296c7658Sdan int *pRc /* IN/OUT: Error code */ 19964fccf43aSdan ){ 19974fccf43aSdan if( *pRc==SQLITE_OK ){ 19984fccf43aSdan int eType = sqlite3_column_type(pStmt, iCol); 19994fccf43aSdan sessionAppendByte(p, (u8)eType, pRc); 20004fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 20014fccf43aSdan sqlite3_int64 i; 20024fccf43aSdan u8 aBuf[8]; 20034fccf43aSdan if( eType==SQLITE_INTEGER ){ 20044fccf43aSdan i = sqlite3_column_int64(pStmt, iCol); 20054fccf43aSdan }else{ 20064fccf43aSdan double r = sqlite3_column_double(pStmt, iCol); 20074fccf43aSdan memcpy(&i, &r, 8); 20084fccf43aSdan } 2009296c7658Sdan sessionPutI64(aBuf, i); 20104fccf43aSdan sessionAppendBlob(p, aBuf, 8, pRc); 20114fccf43aSdan } 20124fccf43aSdan if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){ 20136734007dSdan u8 *z; 20143cc89d95Sdan int nByte; 20156734007dSdan if( eType==SQLITE_BLOB ){ 20166734007dSdan z = (u8 *)sqlite3_column_blob(pStmt, iCol); 20176734007dSdan }else{ 20186734007dSdan z = (u8 *)sqlite3_column_text(pStmt, iCol); 20196734007dSdan } 20203cc89d95Sdan nByte = sqlite3_column_bytes(pStmt, iCol); 20213cc89d95Sdan if( z || (eType==SQLITE_BLOB && nByte==0) ){ 20224fccf43aSdan sessionAppendVarint(p, nByte, pRc); 20236734007dSdan sessionAppendBlob(p, z, nByte, pRc); 20246734007dSdan }else{ 20256734007dSdan *pRc = SQLITE_NOMEM; 20266734007dSdan } 20274fccf43aSdan } 20284fccf43aSdan } 20294fccf43aSdan } 20304fccf43aSdan 2031296c7658Sdan /* 2032296c7658Sdan ** 203380fe2d93Sdan ** This function appends an update change to the buffer (see the comments 203480fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change 203580fe2d93Sdan ** consists of: 2036296c7658Sdan ** 2037296c7658Sdan ** 1 byte: SQLITE_UPDATE (0x17) 2038296c7658Sdan ** n bytes: old.* record (see RECORD FORMAT) 2039296c7658Sdan ** m bytes: new.* record (see RECORD FORMAT) 2040296c7658Sdan ** 2041296c7658Sdan ** The SessionChange object passed as the third argument contains the 2042296c7658Sdan ** values that were stored in the row when the session began (the old.* 2043296c7658Sdan ** values). The statement handle passed as the second argument points 2044296c7658Sdan ** at the current version of the row (the new.* values). 2045296c7658Sdan ** 2046296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value 2047296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer. 2048296c7658Sdan ** 2049296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the 2050296c7658Sdan ** original values of any fields that have been modified. The new.* record 2051296c7658Sdan ** contains the new values of only those fields that have been modified. 2052296c7658Sdan */ 205380fe2d93Sdan static int sessionAppendUpdate( 2054296c7658Sdan SessionBuffer *pBuf, /* Buffer to append to */ 205573b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 2056296c7658Sdan sqlite3_stmt *pStmt, /* Statement handle pointing at new row */ 2057296c7658Sdan SessionChange *p, /* Object containing old values */ 205880fe2d93Sdan u8 *abPK /* Boolean array - true for PK columns */ 20594fccf43aSdan ){ 206080fe2d93Sdan int rc = SQLITE_OK; 2061296c7658Sdan SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */ 2062296c7658Sdan int bNoop = 1; /* Set to zero if any values are modified */ 20631f34f8ccSdan int nRewind = pBuf->nBuf; /* Set to zero if any values are modified */ 2064296c7658Sdan int i; /* Used to iterate through columns */ 2065296c7658Sdan u8 *pCsr = p->aRecord; /* Used to iterate through old.* values */ 2066296c7658Sdan 206780fe2d93Sdan sessionAppendByte(pBuf, SQLITE_UPDATE, &rc); 206880fe2d93Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 20694fccf43aSdan for(i=0; i<sqlite3_column_count(pStmt); i++){ 207037f133ecSdan int bChanged = 0; 20714fccf43aSdan int nAdvance; 20724fccf43aSdan int eType = *pCsr; 20734fccf43aSdan switch( eType ){ 20744fccf43aSdan case SQLITE_NULL: 20754fccf43aSdan nAdvance = 1; 20764fccf43aSdan if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){ 207737f133ecSdan bChanged = 1; 20784fccf43aSdan } 20794fccf43aSdan break; 20804fccf43aSdan 20814fccf43aSdan case SQLITE_FLOAT: 20824fccf43aSdan case SQLITE_INTEGER: { 20834fccf43aSdan nAdvance = 9; 20844fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) ){ 20854fccf43aSdan sqlite3_int64 iVal = sessionGetI64(&pCsr[1]); 20864fccf43aSdan if( eType==SQLITE_INTEGER ){ 20874fccf43aSdan if( iVal==sqlite3_column_int64(pStmt, i) ) break; 20884fccf43aSdan }else{ 20894fccf43aSdan double dVal; 20904fccf43aSdan memcpy(&dVal, &iVal, 8); 20914fccf43aSdan if( dVal==sqlite3_column_double(pStmt, i) ) break; 20924fccf43aSdan } 20934fccf43aSdan } 209437f133ecSdan bChanged = 1; 20954fccf43aSdan break; 20964fccf43aSdan } 20974fccf43aSdan 2098e5754eecSdan default: { 2099895decf6Sdan int n; 2100895decf6Sdan int nHdr = 1 + sessionVarintGet(&pCsr[1], &n); 2101e5754eecSdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 2102895decf6Sdan nAdvance = nHdr + n; 21034fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) 2104895decf6Sdan && n==sqlite3_column_bytes(pStmt, i) 2105895decf6Sdan && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n)) 21064fccf43aSdan ){ 21074fccf43aSdan break; 21084fccf43aSdan } 210937f133ecSdan bChanged = 1; 21104fccf43aSdan } 21114fccf43aSdan } 21124fccf43aSdan 211373b3c055Sdan /* If at least one field has been modified, this is not a no-op. */ 211473b3c055Sdan if( bChanged ) bNoop = 0; 211573b3c055Sdan 211673b3c055Sdan /* Add a field to the old.* record. This is omitted if this modules is 211773b3c055Sdan ** currently generating a patchset. */ 211873b3c055Sdan if( bPatchset==0 ){ 211937f133ecSdan if( bChanged || abPK[i] ){ 212080fe2d93Sdan sessionAppendBlob(pBuf, pCsr, nAdvance, &rc); 21214fccf43aSdan }else{ 212280fe2d93Sdan sessionAppendByte(pBuf, 0, &rc); 212337f133ecSdan } 212473b3c055Sdan } 212537f133ecSdan 212673b3c055Sdan /* Add a field to the new.* record. Or the only record if currently 212773b3c055Sdan ** generating a patchset. */ 212873b3c055Sdan if( bChanged || (bPatchset && abPK[i]) ){ 212980fe2d93Sdan sessionAppendCol(&buf2, pStmt, i, &rc); 213037f133ecSdan }else{ 213180fe2d93Sdan sessionAppendByte(&buf2, 0, &rc); 21324fccf43aSdan } 213337f133ecSdan 21344fccf43aSdan pCsr += nAdvance; 21354fccf43aSdan } 21364fccf43aSdan 21374fccf43aSdan if( bNoop ){ 21381f34f8ccSdan pBuf->nBuf = nRewind; 21394fccf43aSdan }else{ 214080fe2d93Sdan sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc); 21414fccf43aSdan } 21421f34f8ccSdan sqlite3_free(buf2.aBuf); 214380fe2d93Sdan 214480fe2d93Sdan return rc; 2145d5f0767cSdan } 21464fccf43aSdan 2147a71d2371Sdan /* 2148a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use 2149a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset 2150a71d2371Sdan ** format otherwise. 2151a71d2371Sdan */ 215273b3c055Sdan static int sessionAppendDelete( 215373b3c055Sdan SessionBuffer *pBuf, /* Buffer to append to */ 215473b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 215573b3c055Sdan SessionChange *p, /* Object containing old values */ 2156a71d2371Sdan int nCol, /* Number of columns in table */ 215773b3c055Sdan u8 *abPK /* Boolean array - true for PK columns */ 215873b3c055Sdan ){ 215973b3c055Sdan int rc = SQLITE_OK; 216073b3c055Sdan 216173b3c055Sdan sessionAppendByte(pBuf, SQLITE_DELETE, &rc); 216273b3c055Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 216373b3c055Sdan 216473b3c055Sdan if( bPatchset==0 ){ 216573b3c055Sdan sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc); 216673b3c055Sdan }else{ 216773b3c055Sdan int i; 216873b3c055Sdan u8 *a = p->aRecord; 216973b3c055Sdan for(i=0; i<nCol; i++){ 217073b3c055Sdan u8 *pStart = a; 217173b3c055Sdan int eType = *a++; 217273b3c055Sdan 217373b3c055Sdan switch( eType ){ 217473b3c055Sdan case 0: 217573b3c055Sdan case SQLITE_NULL: 217673b3c055Sdan assert( abPK[i]==0 ); 217773b3c055Sdan break; 217873b3c055Sdan 217973b3c055Sdan case SQLITE_FLOAT: 218073b3c055Sdan case SQLITE_INTEGER: 218173b3c055Sdan a += 8; 218273b3c055Sdan break; 218373b3c055Sdan 218473b3c055Sdan default: { 218573b3c055Sdan int n; 218673b3c055Sdan a += sessionVarintGet(a, &n); 218773b3c055Sdan a += n; 218873b3c055Sdan break; 218973b3c055Sdan } 219073b3c055Sdan } 219173b3c055Sdan if( abPK[i] ){ 2192f5ab08c7Sdrh sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc); 219373b3c055Sdan } 219473b3c055Sdan } 219573b3c055Sdan assert( (a - p->aRecord)==p->nRecord ); 219673b3c055Sdan } 219773b3c055Sdan 219873b3c055Sdan return rc; 219973b3c055Sdan } 220073b3c055Sdan 220177fc1d5bSdan /* 220277fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table 220377fc1d5bSdan ** zTab in database zDb based on its primary key. i.e. 220477fc1d5bSdan ** 220577fc1d5bSdan ** SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ... 220677fc1d5bSdan */ 2207e8d5648eSdan static int sessionSelectStmt( 2208e8d5648eSdan sqlite3 *db, /* Database handle */ 2209d7fb7d24Sdan const char *zDb, /* Database name */ 2210e8d5648eSdan const char *zTab, /* Table name */ 221177fc1d5bSdan int nCol, /* Number of columns in table */ 221277fc1d5bSdan const char **azCol, /* Names of table columns */ 221377fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 221477fc1d5bSdan sqlite3_stmt **ppStmt /* OUT: Prepared SELECT statement */ 2215d5f0767cSdan ){ 2216e8d5648eSdan int rc = SQLITE_OK; 22173739f298Sdan char *zSql = 0; 22183739f298Sdan int nSql = -1; 22193739f298Sdan 22203739f298Sdan if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){ 22213739f298Sdan zSql = sqlite3_mprintf( 22223739f298Sdan "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND " 22233739f298Sdan "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb 22243739f298Sdan ); 2225bd45374cSdan if( zSql==0 ) rc = SQLITE_NOMEM; 22263739f298Sdan }else{ 2227d5f0767cSdan int i; 2228e8d5648eSdan const char *zSep = ""; 2229e8d5648eSdan SessionBuffer buf = {0, 0, 0}; 2230d5f0767cSdan 2231e8d5648eSdan sessionAppendStr(&buf, "SELECT * FROM ", &rc); 2232d7fb7d24Sdan sessionAppendIdent(&buf, zDb, &rc); 2233d7fb7d24Sdan sessionAppendStr(&buf, ".", &rc); 2234e8d5648eSdan sessionAppendIdent(&buf, zTab, &rc); 2235e8d5648eSdan sessionAppendStr(&buf, " WHERE ", &rc); 2236e8d5648eSdan for(i=0; i<nCol; i++){ 2237e8d5648eSdan if( abPK[i] ){ 2238e8d5648eSdan sessionAppendStr(&buf, zSep, &rc); 2239e8d5648eSdan sessionAppendIdent(&buf, azCol[i], &rc); 22403739f298Sdan sessionAppendStr(&buf, " IS ?", &rc); 2241e8d5648eSdan sessionAppendInteger(&buf, i+1, &rc); 2242e8d5648eSdan zSep = " AND "; 2243d5f0767cSdan } 2244d5f0767cSdan } 22453739f298Sdan zSql = (char*)buf.aBuf; 22463739f298Sdan nSql = buf.nBuf; 2247d5f0767cSdan } 22483739f298Sdan 22493739f298Sdan if( rc==SQLITE_OK ){ 22503739f298Sdan rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0); 22513739f298Sdan } 22523739f298Sdan sqlite3_free(zSql); 2253e8d5648eSdan return rc; 2254d5f0767cSdan } 2255d5f0767cSdan 225677fc1d5bSdan /* 225777fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange 225877fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement 225977fc1d5bSdan ** is as prepared by function sessionSelectStmt(). 226077fc1d5bSdan ** 226177fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite 226277fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise. 226377fc1d5bSdan */ 2264e8d5648eSdan static int sessionSelectBind( 226577fc1d5bSdan sqlite3_stmt *pSelect, /* SELECT from sessionSelectStmt() */ 226677fc1d5bSdan int nCol, /* Number of columns in table */ 226777fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 226877fc1d5bSdan SessionChange *pChange /* Change structure */ 2269e8d5648eSdan ){ 2270e8d5648eSdan int i; 2271e8d5648eSdan int rc = SQLITE_OK; 2272e5754eecSdan u8 *a = pChange->aRecord; 2273d5f0767cSdan 2274e8d5648eSdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2275e8d5648eSdan int eType = *a++; 2276e8d5648eSdan 2277e8d5648eSdan switch( eType ){ 227880fe2d93Sdan case 0: 2279e8d5648eSdan case SQLITE_NULL: 22801611e5a3Sdan assert( abPK[i]==0 ); 2281e8d5648eSdan break; 2282e8d5648eSdan 2283e8d5648eSdan case SQLITE_INTEGER: { 2284e8d5648eSdan if( abPK[i] ){ 2285e8d5648eSdan i64 iVal = sessionGetI64(a); 2286e8d5648eSdan rc = sqlite3_bind_int64(pSelect, i+1, iVal); 2287e8d5648eSdan } 2288e8d5648eSdan a += 8; 2289e8d5648eSdan break; 2290d5f0767cSdan } 2291296c7658Sdan 2292e8d5648eSdan case SQLITE_FLOAT: { 2293e8d5648eSdan if( abPK[i] ){ 2294e8d5648eSdan double rVal; 2295e8d5648eSdan i64 iVal = sessionGetI64(a); 2296e8d5648eSdan memcpy(&rVal, &iVal, 8); 22974e895da1Sdan rc = sqlite3_bind_double(pSelect, i+1, rVal); 2298d5f0767cSdan } 2299e8d5648eSdan a += 8; 2300e8d5648eSdan break; 2301e8d5648eSdan } 2302e8d5648eSdan 2303e8d5648eSdan case SQLITE_TEXT: { 2304e8d5648eSdan int n; 2305e8d5648eSdan a += sessionVarintGet(a, &n); 2306e8d5648eSdan if( abPK[i] ){ 2307e8d5648eSdan rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT); 2308e8d5648eSdan } 2309e8d5648eSdan a += n; 2310e8d5648eSdan break; 2311e8d5648eSdan } 2312e8d5648eSdan 2313e5754eecSdan default: { 2314e8d5648eSdan int n; 2315e5754eecSdan assert( eType==SQLITE_BLOB ); 2316e8d5648eSdan a += sessionVarintGet(a, &n); 2317e8d5648eSdan if( abPK[i] ){ 2318e8d5648eSdan rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT); 2319e8d5648eSdan } 2320e8d5648eSdan a += n; 2321e8d5648eSdan break; 2322e8d5648eSdan } 2323e8d5648eSdan } 2324e8d5648eSdan } 2325e8d5648eSdan 2326d5f0767cSdan return rc; 23274fccf43aSdan } 23284fccf43aSdan 232977fc1d5bSdan /* 233077fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it 233177fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary 233277fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an 233377fc1d5bSdan ** SQLite error code before returning. 233477fc1d5bSdan */ 23355d607a6eSdan static void sessionAppendTableHdr( 2336a71d2371Sdan SessionBuffer *pBuf, /* Append header to this buffer */ 2337a71d2371Sdan int bPatchset, /* Use the patchset format if true */ 2338a71d2371Sdan SessionTable *pTab, /* Table object to append header for */ 2339a71d2371Sdan int *pRc /* IN/OUT: Error code */ 23405d607a6eSdan ){ 23415d607a6eSdan /* Write a table header */ 234273b3c055Sdan sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc); 23435d607a6eSdan sessionAppendVarint(pBuf, pTab->nCol, pRc); 23445d607a6eSdan sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc); 23454f528042Sdan sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc); 23465d607a6eSdan } 23475d607a6eSdan 2348a71d2371Sdan /* 2349a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset 2350a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object 2351a71d2371Sdan ** passed as the first argument. 2352a71d2371Sdan ** 2353a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset 2354a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error 2355a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set 2356a71d2371Sdan ** to 0. 2357a71d2371Sdan */ 2358adf3bf58Sdrh static int sessionGenerateChangeset( 23594fccf43aSdan sqlite3_session *pSession, /* Session object */ 236073b3c055Sdan int bPatchset, /* True for patchset, false for changeset */ 2361ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2362ef7a6304Sdan void *pOut, /* First argument for xOutput */ 23634fccf43aSdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 23644fccf43aSdan void **ppChangeset /* OUT: Buffer containing changeset */ 23654fccf43aSdan ){ 2366296c7658Sdan sqlite3 *db = pSession->db; /* Source database handle */ 2367296c7658Sdan SessionTable *pTab; /* Used to iterate through attached tables */ 2368296c7658Sdan SessionBuffer buf = {0,0,0}; /* Buffer in which to accumlate changeset */ 2369296c7658Sdan int rc; /* Return code */ 23704fccf43aSdan 2371ef7a6304Sdan assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) ); 2372ef7a6304Sdan 2373296c7658Sdan /* Zero the output variables in case an error occurs. If this session 2374296c7658Sdan ** object is already in the error state (sqlite3_session.rc != SQLITE_OK), 2375296c7658Sdan ** this call will be a no-op. */ 2376ef7a6304Sdan if( xOutput==0 ){ 23774fccf43aSdan *pnChangeset = 0; 23784fccf43aSdan *ppChangeset = 0; 2379ef7a6304Sdan } 2380e5754eecSdan 2381e5754eecSdan if( pSession->rc ) return pSession->rc; 2382e5754eecSdan rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0); 2383e5754eecSdan if( rc!=SQLITE_OK ) return rc; 2384e5754eecSdan 2385e5754eecSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 23864fccf43aSdan 23874fccf43aSdan for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 23884fccf43aSdan if( pTab->nEntry ){ 2389d7fb7d24Sdan const char *zName = pTab->zName; 2390a9605b91Sdan int nCol; /* Number of columns in table */ 2391a9605b91Sdan u8 *abPK; /* Primary key array */ 2392a9605b91Sdan const char **azCol = 0; /* Table columns */ 23931f34f8ccSdan int i; /* Used to iterate through hash buckets */ 23941f34f8ccSdan sqlite3_stmt *pSel = 0; /* SELECT statement to query table pTab */ 23951f34f8ccSdan int nRewind = buf.nBuf; /* Initial size of write buffer */ 23961f34f8ccSdan int nNoop; /* Size of buffer after writing tbl header */ 23974fccf43aSdan 2398a9605b91Sdan /* Check the table schema is still Ok. */ 2399a9605b91Sdan rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK); 2400a9605b91Sdan if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){ 2401a9605b91Sdan rc = SQLITE_SCHEMA; 2402a9605b91Sdan } 2403a9605b91Sdan 24044fccf43aSdan /* Write a table header */ 240573b3c055Sdan sessionAppendTableHdr(&buf, bPatchset, pTab, &rc); 24064fccf43aSdan 24074fccf43aSdan /* Build and compile a statement to execute: */ 24084fccf43aSdan if( rc==SQLITE_OK ){ 2409d7fb7d24Sdan rc = sessionSelectStmt( 2410a9605b91Sdan db, pSession->zDb, zName, nCol, azCol, abPK, &pSel); 24114fccf43aSdan } 24124fccf43aSdan 24131f34f8ccSdan nNoop = buf.nBuf; 241412ca0b56Sdan for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){ 2415e8d5648eSdan SessionChange *p; /* Used to iterate through changes */ 2416e8d5648eSdan 24174fccf43aSdan for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){ 2418e5754eecSdan rc = sessionSelectBind(pSel, nCol, abPK, p); 241980fe2d93Sdan if( rc!=SQLITE_OK ) continue; 24201f34f8ccSdan if( sqlite3_step(pSel)==SQLITE_ROW ){ 2421798693b2Sdan if( p->op==SQLITE_INSERT ){ 24224fccf43aSdan int iCol; 24234fccf43aSdan sessionAppendByte(&buf, SQLITE_INSERT, &rc); 2424b4480e94Sdan sessionAppendByte(&buf, p->bIndirect, &rc); 2425e8d5648eSdan for(iCol=0; iCol<nCol; iCol++){ 24261f34f8ccSdan sessionAppendCol(&buf, pSel, iCol, &rc); 24274fccf43aSdan } 2428e8d5648eSdan }else{ 242973b3c055Sdan rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK); 24304fccf43aSdan } 2431798693b2Sdan }else if( p->op!=SQLITE_INSERT ){ 2432a71d2371Sdan rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK); 24334fccf43aSdan } 243412ca0b56Sdan if( rc==SQLITE_OK ){ 24351f34f8ccSdan rc = sqlite3_reset(pSel); 24364fccf43aSdan } 2437ef7a6304Sdan 24381f48e67dSdan /* If the buffer is now larger than sessions_strm_chunk_size, pass 2439ef7a6304Sdan ** its contents to the xOutput() callback. */ 2440ef7a6304Sdan if( xOutput 2441ef7a6304Sdan && rc==SQLITE_OK 2442ef7a6304Sdan && buf.nBuf>nNoop 24431f48e67dSdan && buf.nBuf>sessions_strm_chunk_size 2444ef7a6304Sdan ){ 2445ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2446ef7a6304Sdan nNoop = -1; 2447ef7a6304Sdan buf.nBuf = 0; 2448ef7a6304Sdan } 2449ef7a6304Sdan 24504fccf43aSdan } 2451e8d5648eSdan } 24524fccf43aSdan 24531f34f8ccSdan sqlite3_finalize(pSel); 24541f34f8ccSdan if( buf.nBuf==nNoop ){ 24554fccf43aSdan buf.nBuf = nRewind; 24564fccf43aSdan } 2457cfdbde21Sdrh sqlite3_free((char*)azCol); /* cast works around VC++ bug */ 24584fccf43aSdan } 24594fccf43aSdan } 24604fccf43aSdan 24614fccf43aSdan if( rc==SQLITE_OK ){ 2462ef7a6304Sdan if( xOutput==0 ){ 24634fccf43aSdan *pnChangeset = buf.nBuf; 24644fccf43aSdan *ppChangeset = buf.aBuf; 2465ef7a6304Sdan buf.aBuf = 0; 2466ef7a6304Sdan }else if( buf.nBuf>0 ){ 2467ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2468ef7a6304Sdan } 24694fccf43aSdan } 24704c220252Sdan 2471ef7a6304Sdan sqlite3_free(buf.aBuf); 2472e5754eecSdan sqlite3_exec(db, "RELEASE changeset", 0, 0, 0); 24734c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 24744fccf43aSdan return rc; 24754fccf43aSdan } 24764fccf43aSdan 2477296c7658Sdan /* 247873b3c055Sdan ** Obtain a changeset object containing all changes recorded by the 247973b3c055Sdan ** session object passed as the first argument. 248073b3c055Sdan ** 248173b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 248273b3c055Sdan ** using sqlite3_free(). 248373b3c055Sdan */ 248473b3c055Sdan int sqlite3session_changeset( 248573b3c055Sdan sqlite3_session *pSession, /* Session object */ 248673b3c055Sdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 248773b3c055Sdan void **ppChangeset /* OUT: Buffer containing changeset */ 248873b3c055Sdan ){ 2489ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset); 2490ef7a6304Sdan } 2491ef7a6304Sdan 2492ef7a6304Sdan /* 2493ef7a6304Sdan ** Streaming version of sqlite3session_changeset(). 2494ef7a6304Sdan */ 2495f1a08ad8Sdrh int sqlite3session_changeset_strm( 2496ef7a6304Sdan sqlite3_session *pSession, 2497ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2498ef7a6304Sdan void *pOut 2499ef7a6304Sdan ){ 2500ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0); 2501ef7a6304Sdan } 2502ef7a6304Sdan 2503ef7a6304Sdan /* 2504ef7a6304Sdan ** Streaming version of sqlite3session_patchset(). 2505ef7a6304Sdan */ 2506f1a08ad8Sdrh int sqlite3session_patchset_strm( 2507ef7a6304Sdan sqlite3_session *pSession, 2508ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2509ef7a6304Sdan void *pOut 2510ef7a6304Sdan ){ 2511ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0); 251273b3c055Sdan } 251373b3c055Sdan 251473b3c055Sdan /* 251573b3c055Sdan ** Obtain a patchset object containing all changes recorded by the 251673b3c055Sdan ** session object passed as the first argument. 251773b3c055Sdan ** 251873b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 251973b3c055Sdan ** using sqlite3_free(). 252073b3c055Sdan */ 252173b3c055Sdan int sqlite3session_patchset( 252273b3c055Sdan sqlite3_session *pSession, /* Session object */ 252373b3c055Sdan int *pnPatchset, /* OUT: Size of buffer at *ppChangeset */ 252473b3c055Sdan void **ppPatchset /* OUT: Buffer containing changeset */ 252573b3c055Sdan ){ 2526ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset); 252773b3c055Sdan } 252873b3c055Sdan 252973b3c055Sdan /* 2530296c7658Sdan ** Enable or disable the session object passed as the first argument. 2531296c7658Sdan */ 25324fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){ 25334c220252Sdan int ret; 25344c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2535296c7658Sdan if( bEnable>=0 ){ 2536296c7658Sdan pSession->bEnable = bEnable; 25374fccf43aSdan } 25384c220252Sdan ret = pSession->bEnable; 25394c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 25404c220252Sdan return ret; 2541296c7658Sdan } 25424fccf43aSdan 25434fccf43aSdan /* 2544b4480e94Sdan ** Enable or disable the session object passed as the first argument. 2545b4480e94Sdan */ 2546b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){ 2547b4480e94Sdan int ret; 2548b4480e94Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2549b4480e94Sdan if( bIndirect>=0 ){ 2550b4480e94Sdan pSession->bIndirect = bIndirect; 2551b4480e94Sdan } 2552b4480e94Sdan ret = pSession->bIndirect; 2553b4480e94Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2554b4480e94Sdan return ret; 2555b4480e94Sdan } 2556b4480e94Sdan 2557b4480e94Sdan /* 2558b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded 2559b69ec348Sdan ** by the session object passed as the only argument. 2560b69ec348Sdan */ 2561b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){ 2562b69ec348Sdan int ret = 0; 2563b69ec348Sdan SessionTable *pTab; 2564b69ec348Sdan 2565b69ec348Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2566b69ec348Sdan for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){ 2567b69ec348Sdan ret = (pTab->nEntry>0); 2568b69ec348Sdan } 2569b69ec348Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2570b69ec348Sdan 2571ff530326Sdan return (ret==0); 2572b69ec348Sdan } 2573b69ec348Sdan 2574b69ec348Sdan /* 2575f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm(). 25764fccf43aSdan */ 2577adf3bf58Sdrh static int sessionChangesetStart( 2578296c7658Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2579ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2580ef7a6304Sdan void *pIn, 2581296c7658Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 258244748f27Sdan void *pChangeset, /* Pointer to buffer containing changeset */ 258344748f27Sdan int bInvert /* True to invert changeset */ 25844fccf43aSdan ){ 25854fccf43aSdan sqlite3_changeset_iter *pRet; /* Iterator to return */ 25864fccf43aSdan int nByte; /* Number of bytes to allocate for iterator */ 25874fccf43aSdan 2588ef7a6304Sdan assert( xInput==0 || (pChangeset==0 && nChangeset==0) ); 2589ef7a6304Sdan 2590296c7658Sdan /* Zero the output variable in case an error occurs. */ 2591296c7658Sdan *pp = 0; 25924fccf43aSdan 2593296c7658Sdan /* Allocate and initialize the iterator structure. */ 25944fccf43aSdan nByte = sizeof(sqlite3_changeset_iter); 25954fccf43aSdan pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte); 25964fccf43aSdan if( !pRet ) return SQLITE_NOMEM; 25974fccf43aSdan memset(pRet, 0, sizeof(sqlite3_changeset_iter)); 25984757c658Sdan pRet->in.aData = (u8 *)pChangeset; 25994757c658Sdan pRet->in.nData = nChangeset; 2600ef7a6304Sdan pRet->in.xInput = xInput; 2601ef7a6304Sdan pRet->in.pIn = pIn; 2602ef7a6304Sdan pRet->in.bEof = (xInput ? 0 : 1); 260344748f27Sdan pRet->bInvert = bInvert; 26044fccf43aSdan 2605296c7658Sdan /* Populate the output variable and return success. */ 2606296c7658Sdan *pp = pRet; 26074fccf43aSdan return SQLITE_OK; 26084fccf43aSdan } 26094fccf43aSdan 2610296c7658Sdan /* 2611ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset. 2612ef7a6304Sdan */ 2613ef7a6304Sdan int sqlite3changeset_start( 2614ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2615ef7a6304Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2616ef7a6304Sdan void *pChangeset /* Pointer to buffer containing changeset */ 2617ef7a6304Sdan ){ 261844748f27Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, 0); 2619ef7a6304Sdan } 262046de0728Sdan int sqlite3changeset_start_v2( 262146de0728Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 262246de0728Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 262346de0728Sdan void *pChangeset, /* Pointer to buffer containing changeset */ 262446de0728Sdan int flags 262546de0728Sdan ){ 262646de0728Sdan int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT); 262746de0728Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, bInvert); 262846de0728Sdan } 2629ef7a6304Sdan 2630ef7a6304Sdan /* 2631ef7a6304Sdan ** Streaming version of sqlite3changeset_start(). 2632ef7a6304Sdan */ 2633f1a08ad8Sdrh int sqlite3changeset_start_strm( 2634ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2635ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2636ef7a6304Sdan void *pIn 2637ef7a6304Sdan ){ 263844748f27Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0, 0); 2639ef7a6304Sdan } 264046de0728Sdan int sqlite3changeset_start_v2_strm( 264146de0728Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 264246de0728Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 264346de0728Sdan void *pIn, 264446de0728Sdan int flags 264546de0728Sdan ){ 264646de0728Sdan int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT); 264746de0728Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0, bInvert); 264846de0728Sdan } 2649ef7a6304Sdan 2650ef7a6304Sdan /* 2651d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming 2652d9151526Sdan ** object and the buffer is full, discard some data to free up space. 2653d9151526Sdan */ 2654d9151526Sdan static void sessionDiscardData(SessionInput *pIn){ 26551f48e67dSdan if( pIn->xInput && pIn->iNext>=sessions_strm_chunk_size ){ 2656d9151526Sdan int nMove = pIn->buf.nBuf - pIn->iNext; 2657d9151526Sdan assert( nMove>=0 ); 2658d9151526Sdan if( nMove>0 ){ 2659d9151526Sdan memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove); 2660d9151526Sdan } 2661d9151526Sdan pIn->buf.nBuf -= pIn->iNext; 2662d9151526Sdan pIn->iNext = 0; 2663d9151526Sdan pIn->nData = pIn->buf.nBuf; 2664d9151526Sdan } 2665d9151526Sdan } 2666d9151526Sdan 2667d9151526Sdan /* 2668ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or, 2669ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available 2670ef7a6304Sdan ** data is in the buffer. 2671ef7a6304Sdan ** 2672ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise. 2673ef7a6304Sdan */ 26744757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){ 2675ef7a6304Sdan int rc = SQLITE_OK; 26764757c658Sdan if( pIn->xInput ){ 26774757c658Sdan while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){ 26781f48e67dSdan int nNew = sessions_strm_chunk_size; 26794757c658Sdan 2680d9151526Sdan if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn); 26814757c658Sdan if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){ 26824757c658Sdan rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew); 26834757c658Sdan if( nNew==0 ){ 26844757c658Sdan pIn->bEof = 1; 26854757c658Sdan }else{ 26864757c658Sdan pIn->buf.nBuf += nNew; 26874757c658Sdan } 26884757c658Sdan } 26894757c658Sdan 26904757c658Sdan pIn->aData = pIn->buf.aBuf; 26914757c658Sdan pIn->nData = pIn->buf.nBuf; 26924757c658Sdan } 2693ef7a6304Sdan } 2694ef7a6304Sdan return rc; 2695ef7a6304Sdan } 2696ef7a6304Sdan 2697ef7a6304Sdan /* 2698ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record 2699ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec 2700ef7a6304Sdan ** until it points to the byte immediately following that record. 2701ef7a6304Sdan */ 2702ef7a6304Sdan static void sessionSkipRecord( 2703ef7a6304Sdan u8 **ppRec, /* IN/OUT: Record pointer */ 2704ef7a6304Sdan int nCol /* Number of values in record */ 2705ef7a6304Sdan ){ 2706ef7a6304Sdan u8 *aRec = *ppRec; 2707ef7a6304Sdan int i; 2708ef7a6304Sdan for(i=0; i<nCol; i++){ 2709ef7a6304Sdan int eType = *aRec++; 2710ef7a6304Sdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2711ef7a6304Sdan int nByte; 2712ef7a6304Sdan aRec += sessionVarintGet((u8*)aRec, &nByte); 2713ef7a6304Sdan aRec += nByte; 2714ef7a6304Sdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2715ef7a6304Sdan aRec += 8; 2716ef7a6304Sdan } 2717ef7a6304Sdan } 2718ef7a6304Sdan 2719ef7a6304Sdan *ppRec = aRec; 2720ef7a6304Sdan } 2721ef7a6304Sdan 2722ef7a6304Sdan /* 27234757c658Sdan ** This function sets the value of the sqlite3_value object passed as the 27244757c658Sdan ** first argument to a copy of the string or blob held in the aData[] 27254757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM 27264757c658Sdan ** error occurs. 27274757c658Sdan */ 27284757c658Sdan static int sessionValueSetStr( 27294757c658Sdan sqlite3_value *pVal, /* Set the value of this object */ 27304757c658Sdan u8 *aData, /* Buffer containing string or blob data */ 27314757c658Sdan int nData, /* Size of buffer aData[] in bytes */ 27324757c658Sdan u8 enc /* String encoding (0 for blobs) */ 27334757c658Sdan ){ 273416228167Sdan /* In theory this code could just pass SQLITE_TRANSIENT as the final 273516228167Sdan ** argument to sqlite3ValueSetStr() and have the copy created 273616228167Sdan ** automatically. But doing so makes it difficult to detect any OOM 273716228167Sdan ** error. Hence the code to create the copy externally. */ 27382d77d80aSdrh u8 *aCopy = sqlite3_malloc64((sqlite3_int64)nData+1); 27394757c658Sdan if( aCopy==0 ) return SQLITE_NOMEM; 27404757c658Sdan memcpy(aCopy, aData, nData); 27414757c658Sdan sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free); 27424757c658Sdan return SQLITE_OK; 27434757c658Sdan } 27444757c658Sdan 27454757c658Sdan /* 2746296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT" 2747296c7658Sdan ** for details. 2748296c7658Sdan ** 2749296c7658Sdan ** When this function is called, *paChange points to the start of the record 2750296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to 2751296c7658Sdan ** one byte after the end of the same record before this function returns. 2752a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or, 2753a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields 2754a71d2371Sdan ** (in other words, it is a patchset DELETE record). 2755296c7658Sdan ** 2756296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller) 2757296c7658Sdan ** is set to point to an sqlite3_value object containing the value read 2758296c7658Sdan ** from the corresponding position in the record. If that value is not 2759296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change 2760296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is 2761296c7658Sdan ** set to NULL. 2762296c7658Sdan ** 2763296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures 2764296c7658Sdan ** using sqlite3_free(). 2765296c7658Sdan ** 2766296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned. 2767296c7658Sdan ** The apOut[] array may have been partially populated in this case. 2768296c7658Sdan */ 27694fccf43aSdan static int sessionReadRecord( 2770ef7a6304Sdan SessionInput *pIn, /* Input data */ 27714fccf43aSdan int nCol, /* Number of values in record */ 277273b3c055Sdan u8 *abPK, /* Array of primary key flags, or NULL */ 27734fccf43aSdan sqlite3_value **apOut /* Write values to this array */ 27744fccf43aSdan ){ 2775296c7658Sdan int i; /* Used to iterate through columns */ 2776ef7a6304Sdan int rc = SQLITE_OK; 27774fccf43aSdan 2778ef7a6304Sdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2779ef7a6304Sdan int eType = 0; /* Type of value (SQLITE_NULL, TEXT etc.) */ 278073b3c055Sdan if( abPK && abPK[i]==0 ) continue; 2781ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2782ef7a6304Sdan if( rc==SQLITE_OK ){ 2783e341ec69Sdan if( pIn->iNext>=pIn->nData ){ 2784e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2785e341ec69Sdan }else{ 27864757c658Sdan eType = pIn->aData[pIn->iNext++]; 2787e8fa8c96Sdan assert( apOut[i]==0 ); 27884fccf43aSdan if( eType ){ 27894fccf43aSdan apOut[i] = sqlite3ValueNew(0); 2790ef7a6304Sdan if( !apOut[i] ) rc = SQLITE_NOMEM; 2791ef7a6304Sdan } 2792dd8a4af8Sdan } 2793e341ec69Sdan } 27944fccf43aSdan 2795ef7a6304Sdan if( rc==SQLITE_OK ){ 27964757c658Sdan u8 *aVal = &pIn->aData[pIn->iNext]; 27974fccf43aSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 27984fccf43aSdan int nByte; 2799ef7a6304Sdan pIn->iNext += sessionVarintGet(aVal, &nByte); 2800ef7a6304Sdan rc = sessionInputBuffer(pIn, nByte); 2801e8fa8c96Sdan if( rc==SQLITE_OK ){ 2802e341ec69Sdan if( nByte<0 || nByte>pIn->nData-pIn->iNext ){ 2803e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2804e341ec69Sdan }else{ 28056734007dSdan u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0); 28064757c658Sdan rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc); 2807ef7a6304Sdan pIn->iNext += nByte; 28084fccf43aSdan } 2809e341ec69Sdan } 2810e341ec69Sdan } 28114fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2812ef7a6304Sdan sqlite3_int64 v = sessionGetI64(aVal); 28134fccf43aSdan if( eType==SQLITE_INTEGER ){ 28144fccf43aSdan sqlite3VdbeMemSetInt64(apOut[i], v); 28154fccf43aSdan }else{ 28164fccf43aSdan double d; 28174e895da1Sdan memcpy(&d, &v, 8); 28184fccf43aSdan sqlite3VdbeMemSetDouble(apOut[i], d); 28194fccf43aSdan } 2820ef7a6304Sdan pIn->iNext += 8; 282191ddd559Sdan } 28224fccf43aSdan } 28234fccf43aSdan } 28244fccf43aSdan 2825ef7a6304Sdan return rc; 2826ef7a6304Sdan } 2827ef7a6304Sdan 2828ef7a6304Sdan /* 2829ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2830ef7a6304Sdan ** Specifically, to the following: 2831ef7a6304Sdan ** 2832ef7a6304Sdan ** + number of columns in table (varint) 2833ef7a6304Sdan ** + array of PK flags (1 byte per column), 2834ef7a6304Sdan ** + table name (nul terminated). 2835ef7a6304Sdan ** 2836ef7a6304Sdan ** This function ensures that all of the above is present in the input 2837ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()). 2838ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code. 2839ef7a6304Sdan ** The input pointer is not moved. 2840ef7a6304Sdan */ 2841ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){ 2842ef7a6304Sdan int rc = SQLITE_OK; 2843ef7a6304Sdan int nCol = 0; 28444757c658Sdan int nRead = 0; 2845ef7a6304Sdan 2846ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2847ef7a6304Sdan if( rc==SQLITE_OK ){ 28484757c658Sdan nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol); 28496344eddaSdan /* The hard upper limit for the number of columns in an SQLite 28506344eddaSdan ** database table is, according to sqliteLimit.h, 32676. So 28516344eddaSdan ** consider any table-header that purports to have more than 65536 28526344eddaSdan ** columns to be corrupt. This is convenient because otherwise, 28536344eddaSdan ** if the (nCol>65536) condition below were omitted, a sufficiently 28546344eddaSdan ** large value for nCol may cause nRead to wrap around and become 28556344eddaSdan ** negative. Leading to a crash. */ 28566344eddaSdan if( nCol<0 || nCol>65536 ){ 2857e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2858e341ec69Sdan }else{ 28594757c658Sdan rc = sessionInputBuffer(pIn, nRead+nCol+100); 28604757c658Sdan nRead += nCol; 2861ef7a6304Sdan } 2862e341ec69Sdan } 28634757c658Sdan 2864ef7a6304Sdan while( rc==SQLITE_OK ){ 28654757c658Sdan while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){ 28664757c658Sdan nRead++; 2867ef7a6304Sdan } 2868e8fa8c96Sdan if( (pIn->iNext + nRead)<pIn->nData ) break; 28694757c658Sdan rc = sessionInputBuffer(pIn, nRead + 100); 28704757c658Sdan } 2871e8fa8c96Sdan *pnByte = nRead+1; 2872ef7a6304Sdan return rc; 2873ef7a6304Sdan } 2874ef7a6304Sdan 2875ef7a6304Sdan /* 2876fa122adaSdan ** The input pointer currently points to the first byte of the first field 2877fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire 287816228167Sdan ** record is buffered. It does not move the input pointer. 287916228167Sdan ** 288016228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of 288116228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The 288216228167Sdan ** final value of *pnByte is undefined in this case. 2883fa122adaSdan */ 2884fa122adaSdan static int sessionChangesetBufferRecord( 288516228167Sdan SessionInput *pIn, /* Input data */ 288616228167Sdan int nCol, /* Number of columns in record */ 288716228167Sdan int *pnByte /* OUT: Size of record in bytes */ 2888fa122adaSdan ){ 2889fa122adaSdan int rc = SQLITE_OK; 2890fa122adaSdan int nByte = 0; 2891fa122adaSdan int i; 2892fa122adaSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 2893fa122adaSdan int eType; 2894fa122adaSdan rc = sessionInputBuffer(pIn, nByte + 10); 2895fa122adaSdan if( rc==SQLITE_OK ){ 2896fa122adaSdan eType = pIn->aData[pIn->iNext + nByte++]; 2897fa122adaSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2898fa122adaSdan int n; 2899fa122adaSdan nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n); 2900fa122adaSdan nByte += n; 2901fa122adaSdan rc = sessionInputBuffer(pIn, nByte); 2902fa122adaSdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2903fa122adaSdan nByte += 8; 2904fa122adaSdan } 2905fa122adaSdan } 2906fa122adaSdan } 2907fa122adaSdan *pnByte = nByte; 2908fa122adaSdan return rc; 2909fa122adaSdan } 2910fa122adaSdan 2911fa122adaSdan /* 2912ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2913ef7a6304Sdan ** Specifically, to the following: 2914ef7a6304Sdan ** 2915ef7a6304Sdan ** + number of columns in table (varint) 2916ef7a6304Sdan ** + array of PK flags (1 byte per column), 2917ef7a6304Sdan ** + table name (nul terminated). 291816228167Sdan ** 291916228167Sdan ** This function decodes the table-header and populates the p->nCol, 292016228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is 292116228167Sdan ** also allocated or resized according to the new value of p->nCol. The 292216228167Sdan ** input pointer is left pointing to the byte following the table header. 292316228167Sdan ** 292416228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code 292516228167Sdan ** is returned and the final values of the various fields enumerated above 292616228167Sdan ** are undefined. 2927ef7a6304Sdan */ 2928ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){ 2929ef7a6304Sdan int rc; 2930ef7a6304Sdan int nCopy; 2931ef7a6304Sdan assert( p->rc==SQLITE_OK ); 2932ef7a6304Sdan 2933ef7a6304Sdan rc = sessionChangesetBufferTblhdr(&p->in, &nCopy); 2934ef7a6304Sdan if( rc==SQLITE_OK ){ 2935ef7a6304Sdan int nByte; 2936ef7a6304Sdan int nVarint; 29374757c658Sdan nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol); 2938dd8a4af8Sdan if( p->nCol>0 ){ 2939ef7a6304Sdan nCopy -= nVarint; 2940ef7a6304Sdan p->in.iNext += nVarint; 2941ef7a6304Sdan nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy; 2942ef7a6304Sdan p->tblhdr.nBuf = 0; 2943ef7a6304Sdan sessionBufferGrow(&p->tblhdr, nByte, &rc); 2944dd8a4af8Sdan }else{ 2945e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2946dd8a4af8Sdan } 2947ef7a6304Sdan } 2948ef7a6304Sdan 2949ef7a6304Sdan if( rc==SQLITE_OK ){ 2950f6ad201aSdrh size_t iPK = sizeof(sqlite3_value*)*p->nCol*2; 2951ef7a6304Sdan memset(p->tblhdr.aBuf, 0, iPK); 29524757c658Sdan memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy); 2953ef7a6304Sdan p->in.iNext += nCopy; 2954ef7a6304Sdan } 2955ef7a6304Sdan 2956ef7a6304Sdan p->apValue = (sqlite3_value**)p->tblhdr.aBuf; 2957ef7a6304Sdan p->abPK = (u8*)&p->apValue[p->nCol*2]; 2958ef7a6304Sdan p->zTab = (char*)&p->abPK[p->nCol]; 2959ef7a6304Sdan return (p->rc = rc); 29604fccf43aSdan } 29614fccf43aSdan 296277fc1d5bSdan /* 296377fc1d5bSdan ** Advance the changeset iterator to the next change. 296477fc1d5bSdan ** 296577fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public 296677fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the 296777fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values. 296877fc1d5bSdan ** 296977fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change 297077fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in 297177fc1d5bSdan ** the record to *pnRec. 297277fc1d5bSdan ** 297377fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is 297477fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite 297577fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further 297677fc1d5bSdan ** changes in the changeset. 297777fc1d5bSdan */ 29785d607a6eSdan static int sessionChangesetNext( 297977fc1d5bSdan sqlite3_changeset_iter *p, /* Changeset iterator */ 298077fc1d5bSdan u8 **paRec, /* If non-NULL, store record pointer here */ 2981c0a499eaSdan int *pnRec, /* If non-NULL, store size of record here */ 2982c0a499eaSdan int *pbNew /* If non-NULL, true if new table */ 29835d607a6eSdan ){ 29844fccf43aSdan int i; 2985ef7a6304Sdan u8 op; 29864fccf43aSdan 29875d607a6eSdan assert( (paRec==0 && pnRec==0) || (paRec && pnRec) ); 29885d607a6eSdan 2989296c7658Sdan /* If the iterator is in the error-state, return immediately. */ 29904fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29914fccf43aSdan 29925d607a6eSdan /* Free the current contents of p->apValue[], if any. */ 29934fccf43aSdan if( p->apValue ){ 29944fccf43aSdan for(i=0; i<p->nCol*2; i++){ 29954fccf43aSdan sqlite3ValueFree(p->apValue[i]); 29964fccf43aSdan } 29974fccf43aSdan memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2); 29984fccf43aSdan } 29994fccf43aSdan 3000ef7a6304Sdan /* Make sure the buffer contains at least 10 bytes of input data, or all 3001ef7a6304Sdan ** remaining data if there are less than 10 bytes available. This is 3002ef7a6304Sdan ** sufficient either for the 'T' or 'P' byte and the varint that follows 3003ef7a6304Sdan ** it, or for the two single byte values otherwise. */ 3004ef7a6304Sdan p->rc = sessionInputBuffer(&p->in, 2); 3005ef7a6304Sdan if( p->rc!=SQLITE_OK ) return p->rc; 3006ef7a6304Sdan 30074fccf43aSdan /* If the iterator is already at the end of the changeset, return DONE. */ 30084757c658Sdan if( p->in.iNext>=p->in.nData ){ 30094fccf43aSdan return SQLITE_DONE; 30104fccf43aSdan } 30114fccf43aSdan 3012d9151526Sdan sessionDiscardData(&p->in); 3013d9151526Sdan p->in.iCurrent = p->in.iNext; 3014d9151526Sdan 30154757c658Sdan op = p->in.aData[p->in.iNext++]; 301607d0f15eSdan while( op=='T' || op=='P' ){ 3017c0a499eaSdan if( pbNew ) *pbNew = 1; 3018ef7a6304Sdan p->bPatchset = (op=='P'); 3019ef7a6304Sdan if( sessionChangesetReadTblhdr(p) ) return p->rc; 3020ef7a6304Sdan if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc; 3021d9151526Sdan p->in.iCurrent = p->in.iNext; 302207d0f15eSdan if( p->in.iNext>=p->in.nData ) return SQLITE_DONE; 30234757c658Sdan op = p->in.aData[p->in.iNext++]; 30245d607a6eSdan } 30255d607a6eSdan 302644748f27Sdan if( p->zTab==0 || (p->bPatchset && p->bInvert) ){ 3027dd8a4af8Sdan /* The first record in the changeset is not a table header. Must be a 3028dd8a4af8Sdan ** corrupt changeset. */ 302944748f27Sdan assert( p->in.iNext==1 || p->zTab ); 3030dd8a4af8Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 3031dd8a4af8Sdan } 3032dd8a4af8Sdan 3033ef7a6304Sdan p->op = op; 30344757c658Sdan p->bIndirect = p->in.aData[p->in.iNext++]; 30354fccf43aSdan if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){ 30364757c658Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 30374fccf43aSdan } 30384fccf43aSdan 3039cbf6d2d2Sdan if( paRec ){ 3040cbf6d2d2Sdan int nVal; /* Number of values to buffer */ 3041cbf6d2d2Sdan if( p->bPatchset==0 && op==SQLITE_UPDATE ){ 3042cbf6d2d2Sdan nVal = p->nCol * 2; 3043cbf6d2d2Sdan }else if( p->bPatchset && op==SQLITE_DELETE ){ 3044cbf6d2d2Sdan nVal = 0; 3045cbf6d2d2Sdan for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++; 3046cbf6d2d2Sdan }else{ 3047cbf6d2d2Sdan nVal = p->nCol; 3048cbf6d2d2Sdan } 3049cbf6d2d2Sdan p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec); 3050cbf6d2d2Sdan if( p->rc!=SQLITE_OK ) return p->rc; 3051cbf6d2d2Sdan *paRec = &p->in.aData[p->in.iNext]; 3052cbf6d2d2Sdan p->in.iNext += *pnRec; 3053cbf6d2d2Sdan }else{ 305444748f27Sdan sqlite3_value **apOld = (p->bInvert ? &p->apValue[p->nCol] : p->apValue); 305544748f27Sdan sqlite3_value **apNew = (p->bInvert ? p->apValue : &p->apValue[p->nCol]); 30565d607a6eSdan 30574fccf43aSdan /* If this is an UPDATE or DELETE, read the old.* record. */ 305873b3c055Sdan if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){ 305973b3c055Sdan u8 *abPK = p->bPatchset ? p->abPK : 0; 306044748f27Sdan p->rc = sessionReadRecord(&p->in, p->nCol, abPK, apOld); 30614fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30624fccf43aSdan } 30634fccf43aSdan 30644fccf43aSdan /* If this is an INSERT or UPDATE, read the new.* record. */ 30654fccf43aSdan if( p->op!=SQLITE_DELETE ){ 306644748f27Sdan p->rc = sessionReadRecord(&p->in, p->nCol, 0, apNew); 30674fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30684fccf43aSdan } 30694fccf43aSdan 307044748f27Sdan if( (p->bPatchset || p->bInvert) && p->op==SQLITE_UPDATE ){ 307173b3c055Sdan /* If this is an UPDATE that is part of a patchset, then all PK and 307273b3c055Sdan ** modified fields are present in the new.* record. The old.* record 307373b3c055Sdan ** is currently completely empty. This block shifts the PK fields from 307473b3c055Sdan ** new.* to old.*, to accommodate the code that reads these arrays. */ 307573b3c055Sdan for(i=0; i<p->nCol; i++){ 307644748f27Sdan assert( p->bPatchset==0 || p->apValue[i]==0 ); 307773b3c055Sdan if( p->abPK[i] ){ 307844748f27Sdan assert( p->apValue[i]==0 ); 307973b3c055Sdan p->apValue[i] = p->apValue[i+p->nCol]; 3080e341ec69Sdan if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT); 308173b3c055Sdan p->apValue[i+p->nCol] = 0; 308273b3c055Sdan } 308373b3c055Sdan } 308444748f27Sdan }else if( p->bInvert ){ 308544748f27Sdan if( p->op==SQLITE_INSERT ) p->op = SQLITE_DELETE; 308644748f27Sdan else if( p->op==SQLITE_DELETE ) p->op = SQLITE_INSERT; 308773b3c055Sdan } 3088cbf6d2d2Sdan } 3089ef7a6304Sdan 30904fccf43aSdan return SQLITE_ROW; 30914fccf43aSdan } 30924fccf43aSdan 30934fccf43aSdan /* 30945d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next 30955d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE 30965d607a6eSdan ** or SQLITE_CORRUPT. 30975d607a6eSdan ** 30985d607a6eSdan ** This function may not be called on iterators passed to a conflict handler 30995d607a6eSdan ** callback by changeset_apply(). 31005d607a6eSdan */ 31015d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){ 3102c0a499eaSdan return sessionChangesetNext(p, 0, 0, 0); 31035d607a6eSdan } 31045d607a6eSdan 31055d607a6eSdan /* 3106244593c8Sdan ** The following function extracts information on the current change 310777fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next() 31084fccf43aSdan ** has returned SQLITE_ROW. 31094fccf43aSdan */ 31104fccf43aSdan int sqlite3changeset_op( 3111296c7658Sdan sqlite3_changeset_iter *pIter, /* Iterator handle */ 31124fccf43aSdan const char **pzTab, /* OUT: Pointer to table name */ 31134fccf43aSdan int *pnCol, /* OUT: Number of columns in table */ 3114b4480e94Sdan int *pOp, /* OUT: SQLITE_INSERT, DELETE or UPDATE */ 3115b4480e94Sdan int *pbIndirect /* OUT: True if change is indirect */ 31164fccf43aSdan ){ 31174fccf43aSdan *pOp = pIter->op; 31184fccf43aSdan *pnCol = pIter->nCol; 31194fccf43aSdan *pzTab = pIter->zTab; 3120b4480e94Sdan if( pbIndirect ) *pbIndirect = pIter->bIndirect; 31214fccf43aSdan return SQLITE_OK; 31224fccf43aSdan } 31234fccf43aSdan 312477fc1d5bSdan /* 312577fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in 312677fc1d5bSdan ** the database table affected by the change that pIter currently points 312777fc1d5bSdan ** to. This function may only be called after changeset_next() returns 312877fc1d5bSdan ** SQLITE_ROW. 312977fc1d5bSdan */ 3130244593c8Sdan int sqlite3changeset_pk( 3131244593c8Sdan sqlite3_changeset_iter *pIter, /* Iterator object */ 3132244593c8Sdan unsigned char **pabPK, /* OUT: Array of boolean - true for PK cols */ 3133244593c8Sdan int *pnCol /* OUT: Number of entries in output array */ 3134244593c8Sdan ){ 3135244593c8Sdan *pabPK = pIter->abPK; 3136244593c8Sdan if( pnCol ) *pnCol = pIter->nCol; 3137244593c8Sdan return SQLITE_OK; 3138244593c8Sdan } 3139244593c8Sdan 3140296c7658Sdan /* 3141296c7658Sdan ** This function may only be called while the iterator is pointing to an 3142296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()). 3143296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3144296c7658Sdan ** 3145296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3146296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not 3147296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3148296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL. 3149296c7658Sdan ** 3150296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3151296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3152296c7658Sdan */ 31534fccf43aSdan int sqlite3changeset_old( 3154296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3155296c7658Sdan int iVal, /* Index of old.* value to retrieve */ 31564fccf43aSdan sqlite3_value **ppValue /* OUT: Old value (or NULL pointer) */ 31574fccf43aSdan ){ 3158d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){ 3159d5f0767cSdan return SQLITE_MISUSE; 3160d5f0767cSdan } 31614fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 31624fccf43aSdan return SQLITE_RANGE; 31634fccf43aSdan } 31644fccf43aSdan *ppValue = pIter->apValue[iVal]; 31654fccf43aSdan return SQLITE_OK; 31664fccf43aSdan } 31674fccf43aSdan 3168296c7658Sdan /* 3169296c7658Sdan ** This function may only be called while the iterator is pointing to an 3170296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()). 3171296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3172296c7658Sdan ** 3173296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3174296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not 3175296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3176296c7658Sdan ** was not modified), set *ppValue to NULL. 3177296c7658Sdan ** 3178296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3179296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3180296c7658Sdan */ 31814fccf43aSdan int sqlite3changeset_new( 3182296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3183296c7658Sdan int iVal, /* Index of new.* value to retrieve */ 31844fccf43aSdan sqlite3_value **ppValue /* OUT: New value (or NULL pointer) */ 31854fccf43aSdan ){ 3186d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){ 3187d5f0767cSdan return SQLITE_MISUSE; 3188d5f0767cSdan } 31894fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 31904fccf43aSdan return SQLITE_RANGE; 31914fccf43aSdan } 31924fccf43aSdan *ppValue = pIter->apValue[pIter->nCol+iVal]; 31934fccf43aSdan return SQLITE_OK; 31944fccf43aSdan } 31954fccf43aSdan 3196296c7658Sdan /* 31977aa469cdSdan ** The following two macros are used internally. They are similar to the 31987aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that 31997aa469cdSdan ** they omit all error checking and return a pointer to the requested value. 32007aa469cdSdan */ 32017aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)] 32027aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)] 32037aa469cdSdan 32047aa469cdSdan /* 3205296c7658Sdan ** This function may only be called with a changeset iterator that has been 3206296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT 3207296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned. 3208296c7658Sdan ** 3209296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure 3210296c7658Sdan ** containing the iVal'th value of the conflicting record. 3211296c7658Sdan ** 3212296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error 3213296c7658Sdan ** code is returned. Otherwise, SQLITE_OK. 3214296c7658Sdan */ 3215d5f0767cSdan int sqlite3changeset_conflict( 3216296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3217296c7658Sdan int iVal, /* Index of conflict record value to fetch */ 3218d5f0767cSdan sqlite3_value **ppValue /* OUT: Value from conflicting row */ 3219d5f0767cSdan ){ 3220d5f0767cSdan if( !pIter->pConflict ){ 3221d5f0767cSdan return SQLITE_MISUSE; 3222d5f0767cSdan } 3223ff677b20Sdan if( iVal<0 || iVal>=pIter->nCol ){ 3224d5f0767cSdan return SQLITE_RANGE; 3225d5f0767cSdan } 3226d5f0767cSdan *ppValue = sqlite3_column_value(pIter->pConflict, iVal); 3227d5f0767cSdan return SQLITE_OK; 3228d5f0767cSdan } 3229d5f0767cSdan 32304fccf43aSdan /* 3231cb3e4b79Sdan ** This function may only be called with an iterator passed to an 3232cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case 3233cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key 3234cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK. 3235cb3e4b79Sdan ** 3236cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE. 3237cb3e4b79Sdan */ 3238cb3e4b79Sdan int sqlite3changeset_fk_conflicts( 3239cb3e4b79Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3240cb3e4b79Sdan int *pnOut /* OUT: Number of FK violations */ 3241cb3e4b79Sdan ){ 3242cb3e4b79Sdan if( pIter->pConflict || pIter->apValue ){ 3243cb3e4b79Sdan return SQLITE_MISUSE; 3244cb3e4b79Sdan } 3245cb3e4b79Sdan *pnOut = pIter->nCol; 3246cb3e4b79Sdan return SQLITE_OK; 3247cb3e4b79Sdan } 3248cb3e4b79Sdan 3249cb3e4b79Sdan 3250cb3e4b79Sdan /* 32514fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start(). 32524fccf43aSdan ** 32534fccf43aSdan ** This function may not be called on iterators passed to a conflict handler 32544fccf43aSdan ** callback by changeset_apply(). 32554fccf43aSdan */ 32564fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){ 3257cbf6d2d2Sdan int rc = SQLITE_OK; 3258cbf6d2d2Sdan if( p ){ 3259296c7658Sdan int i; /* Used to iterate through p->apValue[] */ 3260cbf6d2d2Sdan rc = p->rc; 326112ca0b56Sdan if( p->apValue ){ 32624fccf43aSdan for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]); 326312ca0b56Sdan } 3264ef7a6304Sdan sqlite3_free(p->tblhdr.aBuf); 32654757c658Sdan sqlite3_free(p->in.buf.aBuf); 32664fccf43aSdan sqlite3_free(p); 3267cbf6d2d2Sdan } 32684fccf43aSdan return rc; 32694fccf43aSdan } 32704fccf43aSdan 3271fa122adaSdan static int sessionChangesetInvert( 3272fa122adaSdan SessionInput *pInput, /* Input changeset */ 3273fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3274fa122adaSdan void *pOut, 327591ddd559Sdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 327691ddd559Sdan void **ppInverted /* OUT: Inverse of pChangeset */ 327791ddd559Sdan ){ 3278cfec7eeeSdan int rc = SQLITE_OK; /* Return value */ 3279fa122adaSdan SessionBuffer sOut; /* Output buffer */ 3280cfec7eeeSdan int nCol = 0; /* Number of cols in current table */ 3281cfec7eeeSdan u8 *abPK = 0; /* PK array for current table */ 3282cfec7eeeSdan sqlite3_value **apVal = 0; /* Space for values for UPDATE inversion */ 3283ef7a6304Sdan SessionBuffer sPK = {0, 0, 0}; /* PK array for current table */ 328491ddd559Sdan 3285fa122adaSdan /* Initialize the output buffer */ 3286fa122adaSdan memset(&sOut, 0, sizeof(SessionBuffer)); 3287fa122adaSdan 328891ddd559Sdan /* Zero the output variables in case an error occurs. */ 3289fa122adaSdan if( ppInverted ){ 329091ddd559Sdan *ppInverted = 0; 329191ddd559Sdan *pnInverted = 0; 3292fa122adaSdan } 329391ddd559Sdan 3294fa122adaSdan while( 1 ){ 3295ef7a6304Sdan u8 eType; 3296fa122adaSdan 3297fa122adaSdan /* Test for EOF. */ 3298fa122adaSdan if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert; 3299fa122adaSdan if( pInput->iNext>=pInput->nData ) break; 3300fa122adaSdan eType = pInput->aData[pInput->iNext]; 3301fa122adaSdan 330291ddd559Sdan switch( eType ){ 330391ddd559Sdan case 'T': { 3304244593c8Sdan /* A 'table' record consists of: 3305244593c8Sdan ** 3306244593c8Sdan ** * A constant 'T' character, 3307244593c8Sdan ** * Number of columns in said table (a varint), 3308ef7a6304Sdan ** * An array of nCol bytes (sPK), 3309244593c8Sdan ** * A nul-terminated table name. 3310244593c8Sdan */ 3311ef7a6304Sdan int nByte; 3312fa122adaSdan int nVar; 3313fa122adaSdan pInput->iNext++; 3314fa122adaSdan if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){ 3315ef7a6304Sdan goto finished_invert; 3316ef7a6304Sdan } 3317fa122adaSdan nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol); 3318ef7a6304Sdan sPK.nBuf = 0; 3319fa122adaSdan sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc); 3320fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3321fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3322ef7a6304Sdan if( rc ) goto finished_invert; 3323fa122adaSdan 3324fa122adaSdan pInput->iNext += nByte; 3325cfec7eeeSdan sqlite3_free(apVal); 3326cfec7eeeSdan apVal = 0; 3327ef7a6304Sdan abPK = sPK.aBuf; 332891ddd559Sdan break; 332991ddd559Sdan } 333091ddd559Sdan 333191ddd559Sdan case SQLITE_INSERT: 333291ddd559Sdan case SQLITE_DELETE: { 333391ddd559Sdan int nByte; 3334fa122adaSdan int bIndirect = pInput->aData[pInput->iNext+1]; 3335fa122adaSdan int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE); 3336fa122adaSdan pInput->iNext += 2; 3337fa122adaSdan assert( rc==SQLITE_OK ); 3338fa122adaSdan rc = sessionChangesetBufferRecord(pInput, nCol, &nByte); 3339fa122adaSdan sessionAppendByte(&sOut, eType2, &rc); 3340fa122adaSdan sessionAppendByte(&sOut, bIndirect, &rc); 3341fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3342fa122adaSdan pInput->iNext += nByte; 3343fa122adaSdan if( rc ) goto finished_invert; 334491ddd559Sdan break; 334591ddd559Sdan } 334691ddd559Sdan 334791ddd559Sdan case SQLITE_UPDATE: { 3348cfec7eeeSdan int iCol; 334991ddd559Sdan 3350cfec7eeeSdan if( 0==apVal ){ 33512d77d80aSdrh apVal = (sqlite3_value **)sqlite3_malloc64(sizeof(apVal[0])*nCol*2); 3352cfec7eeeSdan if( 0==apVal ){ 3353cfec7eeeSdan rc = SQLITE_NOMEM; 3354cfec7eeeSdan goto finished_invert; 3355cfec7eeeSdan } 3356cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3357cfec7eeeSdan } 335891ddd559Sdan 3359cfec7eeeSdan /* Write the header for the new UPDATE change. Same as the original. */ 3360fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3361fa122adaSdan sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc); 336291ddd559Sdan 3363ef7a6304Sdan /* Read the old.* and new.* records for the update change. */ 3364fa122adaSdan pInput->iNext += 2; 3365fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]); 3366ef7a6304Sdan if( rc==SQLITE_OK ){ 3367fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]); 3368ef7a6304Sdan } 3369ef7a6304Sdan 3370cfec7eeeSdan /* Write the new old.* record. Consists of the PK columns from the 3371cfec7eeeSdan ** original old.* record, and the other values from the original 3372cfec7eeeSdan ** new.* record. */ 3373e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3374cfec7eeeSdan sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)]; 3375fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3376cfec7eeeSdan } 3377cfec7eeeSdan 3378cfec7eeeSdan /* Write the new new.* record. Consists of a copy of all values 3379cfec7eeeSdan ** from the original old.* record, except for the PK columns, which 3380cfec7eeeSdan ** are set to "undefined". */ 3381e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3382cfec7eeeSdan sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]); 3383fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3384cfec7eeeSdan } 3385cfec7eeeSdan 3386cfec7eeeSdan for(iCol=0; iCol<nCol*2; iCol++){ 3387cfec7eeeSdan sqlite3ValueFree(apVal[iCol]); 3388cfec7eeeSdan } 3389cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3390cfec7eeeSdan if( rc!=SQLITE_OK ){ 3391cfec7eeeSdan goto finished_invert; 3392cfec7eeeSdan } 3393cfec7eeeSdan 339491ddd559Sdan break; 339591ddd559Sdan } 339691ddd559Sdan 339791ddd559Sdan default: 33984757c658Sdan rc = SQLITE_CORRUPT_BKPT; 3399cfec7eeeSdan goto finished_invert; 340091ddd559Sdan } 3401fa122adaSdan 3402fa122adaSdan assert( rc==SQLITE_OK ); 34031f48e67dSdan if( xOutput && sOut.nBuf>=sessions_strm_chunk_size ){ 3404fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3405fa122adaSdan sOut.nBuf = 0; 3406fa122adaSdan if( rc!=SQLITE_OK ) goto finished_invert; 3407fa122adaSdan } 340891ddd559Sdan } 340991ddd559Sdan 3410cfec7eeeSdan assert( rc==SQLITE_OK ); 3411fa122adaSdan if( pnInverted ){ 3412fa122adaSdan *pnInverted = sOut.nBuf; 3413fa122adaSdan *ppInverted = sOut.aBuf; 3414fa122adaSdan sOut.aBuf = 0; 3415fa122adaSdan }else if( sOut.nBuf>0 ){ 3416fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3417fa122adaSdan } 3418cfec7eeeSdan 3419cfec7eeeSdan finished_invert: 3420fa122adaSdan sqlite3_free(sOut.aBuf); 3421cfec7eeeSdan sqlite3_free(apVal); 3422ef7a6304Sdan sqlite3_free(sPK.aBuf); 3423cfec7eeeSdan return rc; 342491ddd559Sdan } 342591ddd559Sdan 3426fa122adaSdan 3427fa122adaSdan /* 3428fa122adaSdan ** Invert a changeset object. 3429fa122adaSdan */ 3430fa122adaSdan int sqlite3changeset_invert( 3431fa122adaSdan int nChangeset, /* Number of bytes in input */ 3432fa122adaSdan const void *pChangeset, /* Input changeset */ 3433fa122adaSdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 3434fa122adaSdan void **ppInverted /* OUT: Inverse of pChangeset */ 3435fa122adaSdan ){ 3436fa122adaSdan SessionInput sInput; 3437fa122adaSdan 3438fa122adaSdan /* Set up the input stream */ 3439fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3440fa122adaSdan sInput.nData = nChangeset; 3441fa122adaSdan sInput.aData = (u8*)pChangeset; 3442fa122adaSdan 3443fa122adaSdan return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted); 3444fa122adaSdan } 3445fa122adaSdan 3446fa122adaSdan /* 3447fa122adaSdan ** Streaming version of sqlite3changeset_invert(). 3448fa122adaSdan */ 3449f1a08ad8Sdrh int sqlite3changeset_invert_strm( 3450fa122adaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 3451fa122adaSdan void *pIn, 3452fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3453fa122adaSdan void *pOut 3454fa122adaSdan ){ 3455fa122adaSdan SessionInput sInput; 3456fa122adaSdan int rc; 3457fa122adaSdan 3458fa122adaSdan /* Set up the input stream */ 3459fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3460fa122adaSdan sInput.xInput = xInput; 3461fa122adaSdan sInput.pIn = pIn; 3462fa122adaSdan 3463fa122adaSdan rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0); 3464fa122adaSdan sqlite3_free(sInput.buf.aBuf); 3465fa122adaSdan return rc; 3466fa122adaSdan } 3467fa122adaSdan 34680c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx; 34690c698471Sdan struct SessionApplyCtx { 34700c698471Sdan sqlite3 *db; 34710c698471Sdan sqlite3_stmt *pDelete; /* DELETE statement */ 3472cfec7eeeSdan sqlite3_stmt *pUpdate; /* UPDATE statement */ 34730c698471Sdan sqlite3_stmt *pInsert; /* INSERT statement */ 34740c698471Sdan sqlite3_stmt *pSelect; /* SELECT statement */ 34750c698471Sdan int nCol; /* Size of azCol[] and abPK[] arrays */ 34760c698471Sdan const char **azCol; /* Array of column names */ 34770c698471Sdan u8 *abPK; /* Boolean array - true if column is in PK */ 3478d1cccf19Sdan int bStat1; /* True if table is sqlite_stat1 */ 3479d9151526Sdan int bDeferConstraints; /* True to defer constraints */ 3480d9151526Sdan SessionBuffer constraints; /* Deferred constraints are stored here */ 3481a38e6c57Sdan SessionBuffer rebase; /* Rebase information (if any) here */ 3482dbe7d37aSdan u8 bRebaseStarted; /* If table header is already in rebase */ 3483dbe7d37aSdan u8 bRebase; /* True to collect rebase information */ 34840c698471Sdan }; 34850c698471Sdan 3486d5f0767cSdan /* 3487d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table 3488d5f0767cSdan ** structure like this: 3489d5f0767cSdan ** 3490d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3491d5f0767cSdan ** 3492d5f0767cSdan ** The DELETE statement looks like this: 3493d5f0767cSdan ** 3494db04571cSdan ** DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4) 3495d5f0767cSdan ** 3496d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require 3497d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the 3498d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE. 3499296c7658Sdan ** 3500296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left 3501296c7658Sdan ** pointing to the prepared version of the SQL statement. 3502d5f0767cSdan */ 3503d5f0767cSdan static int sessionDeleteRow( 3504d5f0767cSdan sqlite3 *db, /* Database handle */ 3505d5f0767cSdan const char *zTab, /* Table name */ 35060c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3507d5f0767cSdan ){ 3508296c7658Sdan int i; 3509296c7658Sdan const char *zSep = ""; 3510d5f0767cSdan int rc = SQLITE_OK; 3511d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 35127cf7df7dSdan int nPk = 0; 3513d5f0767cSdan 3514d5f0767cSdan sessionAppendStr(&buf, "DELETE FROM ", &rc); 3515d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3516296c7658Sdan sessionAppendStr(&buf, " WHERE ", &rc); 3517296c7658Sdan 3518296c7658Sdan for(i=0; i<p->nCol; i++){ 3519296c7658Sdan if( p->abPK[i] ){ 35207cf7df7dSdan nPk++; 3521296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3522296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3523296c7658Sdan sessionAppendStr(&buf, " = ?", &rc); 3524296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3525296c7658Sdan zSep = " AND "; 3526296c7658Sdan } 3527296c7658Sdan } 3528296c7658Sdan 35297cf7df7dSdan if( nPk<p->nCol ){ 3530296c7658Sdan sessionAppendStr(&buf, " AND (?", &rc); 3531296c7658Sdan sessionAppendInteger(&buf, p->nCol+1, &rc); 3532296c7658Sdan sessionAppendStr(&buf, " OR ", &rc); 3533296c7658Sdan 3534296c7658Sdan zSep = ""; 3535296c7658Sdan for(i=0; i<p->nCol; i++){ 3536296c7658Sdan if( !p->abPK[i] ){ 3537296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3538296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3539296c7658Sdan sessionAppendStr(&buf, " IS ?", &rc); 3540296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3541296c7658Sdan zSep = "AND "; 3542296c7658Sdan } 3543296c7658Sdan } 3544296c7658Sdan sessionAppendStr(&buf, ")", &rc); 35457cf7df7dSdan } 3546d5f0767cSdan 3547d5f0767cSdan if( rc==SQLITE_OK ){ 35480c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0); 3549d5f0767cSdan } 3550d5f0767cSdan sqlite3_free(buf.aBuf); 3551d5f0767cSdan 3552d5f0767cSdan return rc; 3553d5f0767cSdan } 3554d5f0767cSdan 3555d5f0767cSdan /* 3556d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db. 3557d5f0767cSdan ** Assuming a table structure like this: 3558d5f0767cSdan ** 3559d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3560d5f0767cSdan ** 3561d5f0767cSdan ** The UPDATE statement looks like this: 3562d5f0767cSdan ** 3563d5f0767cSdan ** UPDATE x SET 3564d5f0767cSdan ** a = CASE WHEN ?2 THEN ?3 ELSE a END, 3565964cbd46Sdan ** b = CASE WHEN ?5 THEN ?6 ELSE b END, 3566964cbd46Sdan ** c = CASE WHEN ?8 THEN ?9 ELSE c END, 3567964cbd46Sdan ** d = CASE WHEN ?11 THEN ?12 ELSE d END 3568d5f0767cSdan ** WHERE a = ?1 AND c = ?7 AND (?13 OR 3569964cbd46Sdan ** (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND 3570d5f0767cSdan ** ) 3571d5f0767cSdan ** 3572d5f0767cSdan ** For each column in the table, there are three variables to bind: 3573d5f0767cSdan ** 3574d5f0767cSdan ** ?(i*3+1) The old.* value of the column, if any. 3575d5f0767cSdan ** ?(i*3+2) A boolean flag indicating that the value is being modified. 3576d5f0767cSdan ** ?(i*3+3) The new.* value of the column, if any. 3577d5f0767cSdan ** 3578d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update 3579d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the 3580d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns 3581d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)". 3582d5f0767cSdan ** 3583296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left 3584296c7658Sdan ** pointing to the prepared version of the SQL statement. 3585d5f0767cSdan */ 3586d5f0767cSdan static int sessionUpdateRow( 3587d5f0767cSdan sqlite3 *db, /* Database handle */ 3588d5f0767cSdan const char *zTab, /* Table name */ 35890c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3590d5f0767cSdan ){ 3591d5f0767cSdan int rc = SQLITE_OK; 3592d5f0767cSdan int i; 3593d5f0767cSdan const char *zSep = ""; 3594d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 3595d5f0767cSdan 3596d5f0767cSdan /* Append "UPDATE tbl SET " */ 3597d5f0767cSdan sessionAppendStr(&buf, "UPDATE ", &rc); 3598d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3599d5f0767cSdan sessionAppendStr(&buf, " SET ", &rc); 3600d5f0767cSdan 3601d5f0767cSdan /* Append the assignments */ 36020c698471Sdan for(i=0; i<p->nCol; i++){ 3603d5f0767cSdan sessionAppendStr(&buf, zSep, &rc); 36040c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3605d5f0767cSdan sessionAppendStr(&buf, " = CASE WHEN ?", &rc); 3606d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3607d5f0767cSdan sessionAppendStr(&buf, " THEN ?", &rc); 3608d5f0767cSdan sessionAppendInteger(&buf, i*3+3, &rc); 3609d5f0767cSdan sessionAppendStr(&buf, " ELSE ", &rc); 36100c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3611d5f0767cSdan sessionAppendStr(&buf, " END", &rc); 3612d5f0767cSdan zSep = ", "; 3613d5f0767cSdan } 3614d5f0767cSdan 3615d5f0767cSdan /* Append the PK part of the WHERE clause */ 3616d5f0767cSdan sessionAppendStr(&buf, " WHERE ", &rc); 36170c698471Sdan for(i=0; i<p->nCol; i++){ 36180c698471Sdan if( p->abPK[i] ){ 36190c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3620d5f0767cSdan sessionAppendStr(&buf, " = ?", &rc); 3621d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3622d5f0767cSdan sessionAppendStr(&buf, " AND ", &rc); 3623d5f0767cSdan } 3624d5f0767cSdan } 3625d5f0767cSdan 3626d5f0767cSdan /* Append the non-PK part of the WHERE clause */ 3627d5f0767cSdan sessionAppendStr(&buf, " (?", &rc); 36280c698471Sdan sessionAppendInteger(&buf, p->nCol*3+1, &rc); 3629d5f0767cSdan sessionAppendStr(&buf, " OR 1", &rc); 36300c698471Sdan for(i=0; i<p->nCol; i++){ 36310c698471Sdan if( !p->abPK[i] ){ 3632d5f0767cSdan sessionAppendStr(&buf, " AND (?", &rc); 3633d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3634d5f0767cSdan sessionAppendStr(&buf, "=0 OR ", &rc); 36350c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3636d5f0767cSdan sessionAppendStr(&buf, " IS ?", &rc); 3637d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3638d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3639d5f0767cSdan } 3640d5f0767cSdan } 3641d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3642d5f0767cSdan 3643d5f0767cSdan if( rc==SQLITE_OK ){ 36440c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0); 3645d5f0767cSdan } 3646d5f0767cSdan sqlite3_free(buf.aBuf); 3647d5f0767cSdan 3648d5f0767cSdan return rc; 3649d5f0767cSdan } 3650d5f0767cSdan 36513739f298Sdan 3652296c7658Sdan /* 3653296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary 3654296c7658Sdan ** key. Assuming the following table structure: 3655296c7658Sdan ** 3656296c7658Sdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3657296c7658Sdan ** 3658296c7658Sdan ** The SELECT statement looks like this: 3659296c7658Sdan ** 3660296c7658Sdan ** SELECT * FROM x WHERE a = ?1 AND c = ?3 3661296c7658Sdan ** 3662296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left 3663296c7658Sdan ** pointing to the prepared version of the SQL statement. 3664296c7658Sdan */ 3665d5f0767cSdan static int sessionSelectRow( 3666d5f0767cSdan sqlite3 *db, /* Database handle */ 3667d5f0767cSdan const char *zTab, /* Table name */ 36680c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3669d5f0767cSdan ){ 3670d7fb7d24Sdan return sessionSelectStmt( 3671d7fb7d24Sdan db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect); 3672d5f0767cSdan } 3673d5f0767cSdan 3674296c7658Sdan /* 3675296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab. 3676296c7658Sdan ** For example: 3677296c7658Sdan ** 3678296c7658Sdan ** INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...); 3679296c7658Sdan ** 3680296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left 3681296c7658Sdan ** pointing to the prepared version of the SQL statement. 3682296c7658Sdan */ 36830c698471Sdan static int sessionInsertRow( 36840c698471Sdan sqlite3 *db, /* Database handle */ 36850c698471Sdan const char *zTab, /* Table name */ 36860c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 36870c698471Sdan ){ 36880c698471Sdan int rc = SQLITE_OK; 36890c698471Sdan int i; 36900c698471Sdan SessionBuffer buf = {0, 0, 0}; 36910c698471Sdan 36920c698471Sdan sessionAppendStr(&buf, "INSERT INTO main.", &rc); 36930c698471Sdan sessionAppendIdent(&buf, zTab, &rc); 3694ff677b20Sdan sessionAppendStr(&buf, "(", &rc); 3695ff677b20Sdan for(i=0; i<p->nCol; i++){ 3696ff677b20Sdan if( i!=0 ) sessionAppendStr(&buf, ", ", &rc); 3697ff677b20Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3698ff677b20Sdan } 3699ff677b20Sdan 3700ff677b20Sdan sessionAppendStr(&buf, ") VALUES(?", &rc); 37010c698471Sdan for(i=1; i<p->nCol; i++){ 37020c698471Sdan sessionAppendStr(&buf, ", ?", &rc); 37030c698471Sdan } 37040c698471Sdan sessionAppendStr(&buf, ")", &rc); 37050c698471Sdan 37060c698471Sdan if( rc==SQLITE_OK ){ 37070c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0); 37080c698471Sdan } 37090c698471Sdan sqlite3_free(buf.aBuf); 37100c698471Sdan return rc; 37110c698471Sdan } 37120c698471Sdan 37133739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){ 37143739f298Sdan return sqlite3_prepare_v2(db, zSql, -1, pp, 0); 37153739f298Sdan } 37163739f298Sdan 37173739f298Sdan /* 37183739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table. 37193739f298Sdan ** These are similar to those created by sessionSelectRow(), 37203739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for 37213739f298Sdan ** other tables. 37223739f298Sdan */ 37233739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){ 37243739f298Sdan int rc = sessionSelectRow(db, "sqlite_stat1", p); 37253739f298Sdan if( rc==SQLITE_OK ){ 37263739f298Sdan rc = sessionPrepare(db, &p->pInsert, 37273739f298Sdan "INSERT INTO main.sqlite_stat1 VALUES(?1, " 37283739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, " 37293739f298Sdan "?3)" 37303739f298Sdan ); 37313739f298Sdan } 37323739f298Sdan if( rc==SQLITE_OK ){ 37333739f298Sdan rc = sessionPrepare(db, &p->pUpdate, 37343739f298Sdan "UPDATE main.sqlite_stat1 SET " 37353739f298Sdan "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, " 37363739f298Sdan "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, " 37373739f298Sdan "stat = CASE WHEN ?8 THEN ?9 ELSE stat END " 37383739f298Sdan "WHERE tbl=?1 AND idx IS " 37393739f298Sdan "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END " 37403739f298Sdan "AND (?10 OR ?8=0 OR stat IS ?7)" 37413739f298Sdan ); 37423739f298Sdan } 37433739f298Sdan if( rc==SQLITE_OK ){ 37443739f298Sdan rc = sessionPrepare(db, &p->pDelete, 37453739f298Sdan "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS " 37463739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END " 37473739f298Sdan "AND (?4 OR stat IS ?3)" 37483739f298Sdan ); 37493739f298Sdan } 37503739f298Sdan return rc; 37513739f298Sdan } 37523739f298Sdan 3753296c7658Sdan /* 37547aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem. 37557aa469cdSdan ** See comments in the body of this function for details. 37567aa469cdSdan */ 37577aa469cdSdan static int sessionBindValue( 37587aa469cdSdan sqlite3_stmt *pStmt, /* Statement to bind value to */ 37597aa469cdSdan int i, /* Parameter number to bind to */ 37607aa469cdSdan sqlite3_value *pVal /* Value to bind */ 37617aa469cdSdan ){ 37625671ef69Sdrh int eType = sqlite3_value_type(pVal); 3763082c96dfSdan /* COVERAGE: The (pVal->z==0) branch is never true using current versions 3764082c96dfSdan ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either 3765082c96dfSdan ** the (pVal->z) variable remains as it was or the type of the value is 3766082c96dfSdan ** set to SQLITE_NULL. */ 37675671ef69Sdrh if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){ 37687aa469cdSdan /* This condition occurs when an earlier OOM in a call to 37697aa469cdSdan ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within 3770082c96dfSdan ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */ 37717aa469cdSdan return SQLITE_NOMEM; 37727aa469cdSdan } 37737aa469cdSdan return sqlite3_bind_value(pStmt, i, pVal); 37747aa469cdSdan } 37757aa469cdSdan 37767aa469cdSdan /* 3777db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function 3778db04571cSdan ** transfers new.* values from the current iterator entry to statement 3779db04571cSdan ** pStmt. The table being inserted into has nCol columns. 3780db04571cSdan ** 3781d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of 3782db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1) 3783db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points 3784db04571cSdan ** to an array nCol elements in size. In this case only those values for 3785db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the 3786db04571cSdan ** statement. 3787db04571cSdan ** 3788db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK. 3789db04571cSdan */ 37907aa469cdSdan static int sessionBindRow( 3791db04571cSdan sqlite3_changeset_iter *pIter, /* Iterator to read values from */ 37927aa469cdSdan int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **), 3793db04571cSdan int nCol, /* Number of columns */ 3794db04571cSdan u8 *abPK, /* If not NULL, bind only if true */ 3795db04571cSdan sqlite3_stmt *pStmt /* Bind values to this statement */ 3796db04571cSdan ){ 3797db04571cSdan int i; 3798db04571cSdan int rc = SQLITE_OK; 37997aa469cdSdan 38007aa469cdSdan /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the 38017aa469cdSdan ** argument iterator points to a suitable entry. Make sure that xValue 38027aa469cdSdan ** is one of these to guarantee that it is safe to ignore the return 38037aa469cdSdan ** in the code below. */ 38047aa469cdSdan assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new ); 38057aa469cdSdan 3806db04571cSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 3807db04571cSdan if( !abPK || abPK[i] ){ 3808db04571cSdan sqlite3_value *pVal; 38097aa469cdSdan (void)xValue(pIter, i, &pVal); 3810dd8a4af8Sdan if( pVal==0 ){ 3811dd8a4af8Sdan /* The value in the changeset was "undefined". This indicates a 3812dd8a4af8Sdan ** corrupt changeset blob. */ 3813e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 3814dd8a4af8Sdan }else{ 38157aa469cdSdan rc = sessionBindValue(pStmt, i+1, pVal); 3816db04571cSdan } 3817db04571cSdan } 3818dd8a4af8Sdan } 3819db04571cSdan return rc; 3820db04571cSdan } 3821db04571cSdan 3822db04571cSdan /* 3823296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function. 3824296c7658Sdan ** This function binds the primary key values from the change that changeset 3825296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table 3826296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row 3827296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error 3828296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an 38297aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned. 38307aa469cdSdan ** 38317aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset() 38327aa469cdSdan ** statement pSelect. If any other value is returned, the statement does 38337aa469cdSdan ** not require a reset(). 3834296c7658Sdan ** 3835296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the 3836db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or 3837db04571cSdan ** UPDATE, bind values from the old.* record. 3838296c7658Sdan */ 38390c698471Sdan static int sessionSeekToRow( 384037f133ecSdan sqlite3 *db, /* Database handle */ 384137f133ecSdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 384237f133ecSdan u8 *abPK, /* Primary key flags array */ 38430c698471Sdan sqlite3_stmt *pSelect /* SELECT statement from sessionSelectRow() */ 384437f133ecSdan ){ 38457aa469cdSdan int rc; /* Return code */ 3846296c7658Sdan int nCol; /* Number of columns in table */ 3847296c7658Sdan int op; /* Changset operation (SQLITE_UPDATE etc.) */ 3848296c7658Sdan const char *zDummy; /* Unused */ 384937f133ecSdan 3850b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 38517aa469cdSdan rc = sessionBindRow(pIter, 3852db04571cSdan op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old, 3853db04571cSdan nCol, abPK, pSelect 3854db04571cSdan ); 38550c698471Sdan 38560c698471Sdan if( rc==SQLITE_OK ){ 38570c698471Sdan rc = sqlite3_step(pSelect); 38580c698471Sdan if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect); 38590c698471Sdan } 38600c698471Sdan 38610c698471Sdan return rc; 38620c698471Sdan } 38630c698471Sdan 3864b880a7b1Sdan /* 3865a920b209Sdrh ** This function is called from within sqlite3changeset_apply_v2() when 3866b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution 3867b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE).. 3868b880a7b1Sdan ** It adds a conflict resolution record to the buffer in 3869b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller 3870b880a7b1Sdan ** of apply_v2() as the "rebase" buffer. 3871b880a7b1Sdan ** 3872b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise. 3873b880a7b1Sdan */ 3874a38e6c57Sdan static int sessionRebaseAdd( 3875b880a7b1Sdan SessionApplyCtx *p, /* Apply context */ 3876b880a7b1Sdan int eType, /* Conflict resolution (OMIT or REPLACE) */ 3877b880a7b1Sdan sqlite3_changeset_iter *pIter /* Iterator pointing at current change */ 3878a38e6c57Sdan ){ 3879a38e6c57Sdan int rc = SQLITE_OK; 3880dbe7d37aSdan if( p->bRebase ){ 3881a38e6c57Sdan int i; 3882a38e6c57Sdan int eOp = pIter->op; 3883a38e6c57Sdan if( p->bRebaseStarted==0 ){ 3884a38e6c57Sdan /* Append a table-header to the rebase buffer */ 3885a38e6c57Sdan const char *zTab = pIter->zTab; 3886a38e6c57Sdan sessionAppendByte(&p->rebase, 'T', &rc); 3887a38e6c57Sdan sessionAppendVarint(&p->rebase, p->nCol, &rc); 3888a38e6c57Sdan sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc); 3889a38e6c57Sdan sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc); 3890a38e6c57Sdan p->bRebaseStarted = 1; 3891a38e6c57Sdan } 3892a38e6c57Sdan 3893a38e6c57Sdan assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT ); 3894a38e6c57Sdan assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE ); 3895a38e6c57Sdan 3896f01d3a7eSdan sessionAppendByte(&p->rebase, 3897f01d3a7eSdan (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc 3898f01d3a7eSdan ); 3899f01d3a7eSdan sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc); 3900a38e6c57Sdan for(i=0; i<p->nCol; i++){ 3901a38e6c57Sdan sqlite3_value *pVal = 0; 3902f01d3a7eSdan if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){ 3903a38e6c57Sdan sqlite3changeset_old(pIter, i, &pVal); 3904a38e6c57Sdan }else{ 3905a38e6c57Sdan sqlite3changeset_new(pIter, i, &pVal); 3906a38e6c57Sdan } 3907a38e6c57Sdan sessionAppendValue(&p->rebase, pVal, &rc); 3908a38e6c57Sdan } 3909dbe7d37aSdan } 3910a38e6c57Sdan return rc; 3911a38e6c57Sdan } 3912a38e6c57Sdan 3913296c7658Sdan /* 3914296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator 3915296c7658Sdan ** currently points to. 3916296c7658Sdan ** 3917296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT. 3918296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked 3919296c7658Sdan ** depends solely on eType, as follows: 3920296c7658Sdan ** 3921296c7658Sdan ** eType value Value passed to xConflict 3922296c7658Sdan ** ------------------------------------------------- 3923296c7658Sdan ** CHANGESET_DATA CHANGESET_NOTFOUND 3924296c7658Sdan ** CHANGESET_CONFLICT CHANGESET_CONSTRAINT 3925296c7658Sdan ** 3926296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing 3927296c7658Sdan ** record with the same primary key as the record about to be deleted, updated 3928296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict 3929296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict 3930296c7658Sdan ** handler invoked is as follows: 3931296c7658Sdan ** 3932296c7658Sdan ** eType value PK Record found? Value passed to xConflict 3933296c7658Sdan ** ---------------------------------------------------------------- 3934296c7658Sdan ** CHANGESET_DATA Yes CHANGESET_DATA 3935296c7658Sdan ** CHANGESET_DATA No CHANGESET_NOTFOUND 3936296c7658Sdan ** CHANGESET_CONFLICT Yes CHANGESET_CONFLICT 3937296c7658Sdan ** CHANGESET_CONFLICT No CHANGESET_CONSTRAINT 3938296c7658Sdan ** 3939296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and 3940296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace 3941296c7658Sdan ** is set to non-zero before returning SQLITE_OK. 3942296c7658Sdan ** 3943296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is 3944296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value, 3945296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT, 3946296c7658Sdan ** this function returns SQLITE_OK. 3947296c7658Sdan */ 39480c698471Sdan static int sessionConflictHandler( 3949296c7658Sdan int eType, /* Either CHANGESET_DATA or CONFLICT */ 3950296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 39510c698471Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 39520c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter*), 3953296c7658Sdan void *pCtx, /* First argument for conflict handler */ 3954296c7658Sdan int *pbReplace /* OUT: Set to true if PK row is found */ 39550c698471Sdan ){ 395674f598b6Smistachkin int res = 0; /* Value returned by conflict handler */ 39570c698471Sdan int rc; 39580c698471Sdan int nCol; 39590c698471Sdan int op; 39600c698471Sdan const char *zDummy; 39610c698471Sdan 3962b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 39630c698471Sdan 39640c698471Sdan assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA ); 39650c698471Sdan assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT ); 39660c698471Sdan assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND ); 396737f133ecSdan 396837f133ecSdan /* Bind the new.* PRIMARY KEY values to the SELECT statement. */ 39690c698471Sdan if( pbReplace ){ 39700c698471Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 39710c698471Sdan }else{ 3972db04571cSdan rc = SQLITE_OK; 39730c698471Sdan } 39740c698471Sdan 39750c698471Sdan if( rc==SQLITE_ROW ){ 39760c698471Sdan /* There exists another row with the new.* primary key. */ 39770c698471Sdan pIter->pConflict = p->pSelect; 39780c698471Sdan res = xConflict(pCtx, eType, pIter); 39790c698471Sdan pIter->pConflict = 0; 39800c698471Sdan rc = sqlite3_reset(p->pSelect); 3981db04571cSdan }else if( rc==SQLITE_OK ){ 3982d9151526Sdan if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){ 3983d9151526Sdan /* Instead of invoking the conflict handler, append the change blob 3984d9151526Sdan ** to the SessionApplyCtx.constraints buffer. */ 3985d9151526Sdan u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent]; 3986d9151526Sdan int nBlob = pIter->in.iNext - pIter->in.iCurrent; 3987d9151526Sdan sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc); 3988a38e6c57Sdan return SQLITE_OK; 3989d9151526Sdan }else{ 39900c698471Sdan /* No other row with the new.* primary key. */ 39910c698471Sdan res = xConflict(pCtx, eType+1, pIter); 39920c698471Sdan if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE; 399337f133ecSdan } 3994d9151526Sdan } 399537f133ecSdan 399637f133ecSdan if( rc==SQLITE_OK ){ 39970c698471Sdan switch( res ){ 39980c698471Sdan case SQLITE_CHANGESET_REPLACE: 3999f51e5f6cSdan assert( pbReplace ); 4000f51e5f6cSdan *pbReplace = 1; 40010c698471Sdan break; 40020c698471Sdan 40030c698471Sdan case SQLITE_CHANGESET_OMIT: 40040c698471Sdan break; 40050c698471Sdan 40060c698471Sdan case SQLITE_CHANGESET_ABORT: 40070c698471Sdan rc = SQLITE_ABORT; 40080c698471Sdan break; 40090c698471Sdan 40100c698471Sdan default: 40110c698471Sdan rc = SQLITE_MISUSE; 40120c698471Sdan break; 40130c698471Sdan } 4014a38e6c57Sdan if( rc==SQLITE_OK ){ 4015a38e6c57Sdan rc = sessionRebaseAdd(p, res, pIter); 4016a38e6c57Sdan } 40170c698471Sdan } 40180c698471Sdan 40190c698471Sdan return rc; 40200c698471Sdan } 40210c698471Sdan 4022296c7658Sdan /* 4023296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument 4024296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke 4025296c7658Sdan ** the conflict handler callback. 4026296c7658Sdan ** 4027296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If 4028296c7658Sdan ** one is encountered, update or delete the row with the matching primary key 4029296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs, 4030296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry 4031296c7658Sdan ** to true before returning. In this case the caller will invoke this function 4032296c7658Sdan ** again, this time with pbRetry set to NULL. 4033296c7658Sdan ** 4034296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is 4035296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead. 4036296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such 4037296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true 4038296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting 4039296c7658Sdan ** row before invoking this function again, this time with pbReplace set 4040296c7658Sdan ** to NULL. 4041296c7658Sdan ** 4042296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function 4043296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is 4044296c7658Sdan ** returned. 4045296c7658Sdan */ 40460c698471Sdan static int sessionApplyOneOp( 4047296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 4048296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 40490c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter *), 4050296c7658Sdan void *pCtx, /* First argument for the conflict handler */ 4051296c7658Sdan int *pbReplace, /* OUT: True to remove PK row and retry */ 4052296c7658Sdan int *pbRetry /* OUT: True to retry. */ 40530c698471Sdan ){ 40540c698471Sdan const char *zDummy; 40550c698471Sdan int op; 40560c698471Sdan int nCol; 40570c698471Sdan int rc = SQLITE_OK; 40580c698471Sdan 40590c698471Sdan assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect ); 40600c698471Sdan assert( p->azCol && p->abPK ); 40610c698471Sdan assert( !pbReplace || *pbReplace==0 ); 40620c698471Sdan 4063b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 40640c698471Sdan 40650c698471Sdan if( op==SQLITE_DELETE ){ 40660c698471Sdan 406773b3c055Sdan /* Bind values to the DELETE statement. If conflict handling is required, 406873b3c055Sdan ** bind values for all columns and set bound variable (nCol+1) to true. 406973b3c055Sdan ** Or, if conflict handling is not required, bind just the PK column 407073b3c055Sdan ** values and, if it exists, set (nCol+1) to false. Conflict handling 407173b3c055Sdan ** is not required if: 407273b3c055Sdan ** 407373b3c055Sdan ** * this is a patchset, or 407473b3c055Sdan ** * (pbRetry==0), or 407573b3c055Sdan ** * all columns of the table are PK columns (in this case there is 407673b3c055Sdan ** no (nCol+1) variable to bind to). 407773b3c055Sdan */ 407873b3c055Sdan u8 *abPK = (pIter->bPatchset ? p->abPK : 0); 407973b3c055Sdan rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete); 40807cf7df7dSdan if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){ 408173b3c055Sdan rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK)); 40827cf7df7dSdan } 40830c698471Sdan if( rc!=SQLITE_OK ) return rc; 40840c698471Sdan 40850c698471Sdan sqlite3_step(p->pDelete); 40860c698471Sdan rc = sqlite3_reset(p->pDelete); 40870c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 40880c698471Sdan rc = sessionConflictHandler( 40890c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 40900c698471Sdan ); 409135e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 40920c698471Sdan rc = sessionConflictHandler( 40930c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 40940c698471Sdan ); 40950c698471Sdan } 40960c698471Sdan 40970c698471Sdan }else if( op==SQLITE_UPDATE ){ 40980c698471Sdan int i; 40990c698471Sdan 41000c698471Sdan /* Bind values to the UPDATE statement. */ 41010c698471Sdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 41027aa469cdSdan sqlite3_value *pOld = sessionChangesetOld(pIter, i); 41037aa469cdSdan sqlite3_value *pNew = sessionChangesetNew(pIter, i); 41047aa469cdSdan 41050c698471Sdan sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew); 41067aa469cdSdan if( pOld ){ 41077aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+1, pOld); 41087aa469cdSdan } 41097aa469cdSdan if( rc==SQLITE_OK && pNew ){ 41107aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+3, pNew); 41110c698471Sdan } 41120c698471Sdan } 411373b3c055Sdan if( rc==SQLITE_OK ){ 411473b3c055Sdan sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset); 411573b3c055Sdan } 41160c698471Sdan if( rc!=SQLITE_OK ) return rc; 41170c698471Sdan 41180c698471Sdan /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict, 41190c698471Sdan ** the result will be SQLITE_OK with 0 rows modified. */ 41200c698471Sdan sqlite3_step(p->pUpdate); 41210c698471Sdan rc = sqlite3_reset(p->pUpdate); 41220c698471Sdan 41230c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 41240c698471Sdan /* A NOTFOUND or DATA error. Search the table to see if it contains 41250c698471Sdan ** a row with a matching primary key. If so, this is a DATA conflict. 41260c698471Sdan ** Otherwise, if there is no primary key match, it is a NOTFOUND. */ 41270c698471Sdan 41280c698471Sdan rc = sessionConflictHandler( 41290c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 41300c698471Sdan ); 41310c698471Sdan 413235e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 4133db04571cSdan /* This is always a CONSTRAINT conflict. */ 4134db04571cSdan rc = sessionConflictHandler( 4135db04571cSdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 41360c698471Sdan ); 41370c698471Sdan } 41380c698471Sdan 41390c698471Sdan }else{ 41400c698471Sdan assert( op==SQLITE_INSERT ); 4141d1cccf19Sdan if( p->bStat1 ){ 4142d1cccf19Sdan /* Check if there is a conflicting row. For sqlite_stat1, this needs 4143d1cccf19Sdan ** to be done using a SELECT, as there is no PRIMARY KEY in the 4144d1cccf19Sdan ** database schema to throw an exception if a duplicate is inserted. */ 4145d1cccf19Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 4146d1cccf19Sdan if( rc==SQLITE_ROW ){ 4147d1cccf19Sdan rc = SQLITE_CONSTRAINT; 4148d1cccf19Sdan sqlite3_reset(p->pSelect); 4149d1cccf19Sdan } 4150d1cccf19Sdan } 4151d1cccf19Sdan 4152d1cccf19Sdan if( rc==SQLITE_OK ){ 41537aa469cdSdan rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert); 41540c698471Sdan if( rc!=SQLITE_OK ) return rc; 41550c698471Sdan 41560c698471Sdan sqlite3_step(p->pInsert); 41570c698471Sdan rc = sqlite3_reset(p->pInsert); 4158d1cccf19Sdan } 4159d1cccf19Sdan 416035e2858eSdan if( (rc&0xff)==SQLITE_CONSTRAINT ){ 41610c698471Sdan rc = sessionConflictHandler( 41620c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace 41630c698471Sdan ); 416437f133ecSdan } 416537f133ecSdan } 416637f133ecSdan 416737f133ecSdan return rc; 416837f133ecSdan } 416937f133ecSdan 41705f5663dcSdan /* 41715f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument 41725f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke 41735f5663dcSdan ** the conflict handler callback. 41745f5663dcSdan ** 41755f5663dcSdan ** The difference between this function and sessionApplyOne() is that this 41765f5663dcSdan ** function handles the case where the conflict-handler is invoked and 41775f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be 41785f5663dcSdan ** retried in some manner. 41795f5663dcSdan */ 4180d9151526Sdan static int sessionApplyOneWithRetry( 4181d9151526Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4182d9151526Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator to read change from */ 4183d9151526Sdan SessionApplyCtx *pApply, /* Apply context */ 4184d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4185d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4186d9151526Sdan ){ 4187d9151526Sdan int bReplace = 0; 4188d9151526Sdan int bRetry = 0; 4189d9151526Sdan int rc; 4190d9151526Sdan 4191d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry); 4192a38e6c57Sdan if( rc==SQLITE_OK ){ 41935f5663dcSdan /* If the bRetry flag is set, the change has not been applied due to an 41945f5663dcSdan ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and 41955f5663dcSdan ** a row with the correct PK is present in the db, but one or more other 41965f5663dcSdan ** fields do not contain the expected values) and the conflict handler 41975f5663dcSdan ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation, 41985f5663dcSdan ** but pass NULL as the final argument so that sessionApplyOneOp() ignores 41995f5663dcSdan ** the SQLITE_CHANGESET_DATA problem. */ 42005f5663dcSdan if( bRetry ){ 42015f5663dcSdan assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE ); 42025f5663dcSdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4203d9151526Sdan } 4204d9151526Sdan 42055f5663dcSdan /* If the bReplace flag is set, the change is an INSERT that has not 42065f5663dcSdan ** been performed because the database already contains a row with the 42075f5663dcSdan ** specified primary key and the conflict handler returned 42085f5663dcSdan ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row 42095f5663dcSdan ** before reattempting the INSERT. */ 42105f5663dcSdan else if( bReplace ){ 4211d9151526Sdan assert( pIter->op==SQLITE_INSERT ); 4212d9151526Sdan rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0); 4213d9151526Sdan if( rc==SQLITE_OK ){ 4214d9151526Sdan rc = sessionBindRow(pIter, 4215d9151526Sdan sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete); 4216d9151526Sdan sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1); 4217d9151526Sdan } 4218d9151526Sdan if( rc==SQLITE_OK ){ 4219d9151526Sdan sqlite3_step(pApply->pDelete); 4220d9151526Sdan rc = sqlite3_reset(pApply->pDelete); 4221d9151526Sdan } 4222d9151526Sdan if( rc==SQLITE_OK ){ 4223d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4224d9151526Sdan } 4225d9151526Sdan if( rc==SQLITE_OK ){ 4226d9151526Sdan rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0); 4227d9151526Sdan } 4228d9151526Sdan } 4229a38e6c57Sdan } 4230d9151526Sdan 4231d9151526Sdan return rc; 4232d9151526Sdan } 4233d9151526Sdan 4234d9151526Sdan /* 4235d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer. 4236d9151526Sdan */ 4237d9151526Sdan static int sessionRetryConstraints( 4238d9151526Sdan sqlite3 *db, 4239d9151526Sdan int bPatchset, 4240d9151526Sdan const char *zTab, 4241d9151526Sdan SessionApplyCtx *pApply, 4242d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4243d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4244d9151526Sdan ){ 4245d9151526Sdan int rc = SQLITE_OK; 4246d9151526Sdan 4247d9151526Sdan while( pApply->constraints.nBuf ){ 4248d9151526Sdan sqlite3_changeset_iter *pIter2 = 0; 4249d9151526Sdan SessionBuffer cons = pApply->constraints; 4250d9151526Sdan memset(&pApply->constraints, 0, sizeof(SessionBuffer)); 4251d9151526Sdan 425244748f27Sdan rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf, 0); 4253d9151526Sdan if( rc==SQLITE_OK ){ 4254f6ad201aSdrh size_t nByte = 2*pApply->nCol*sizeof(sqlite3_value*); 4255d9151526Sdan int rc2; 4256d9151526Sdan pIter2->bPatchset = bPatchset; 4257d9151526Sdan pIter2->zTab = (char*)zTab; 4258d9151526Sdan pIter2->nCol = pApply->nCol; 4259d9151526Sdan pIter2->abPK = pApply->abPK; 4260d9151526Sdan sessionBufferGrow(&pIter2->tblhdr, nByte, &rc); 4261d9151526Sdan pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf; 4262d9151526Sdan if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte); 4263d9151526Sdan 4264d9151526Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){ 4265d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx); 4266d9151526Sdan } 4267d9151526Sdan 4268d9151526Sdan rc2 = sqlite3changeset_finalize(pIter2); 42697e0765a9Sdrh if( rc==SQLITE_OK ) rc = rc2; 4270d9151526Sdan } 4271d9151526Sdan assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 ); 4272d9151526Sdan 4273d9151526Sdan sqlite3_free(cons.aBuf); 4274d9151526Sdan if( rc!=SQLITE_OK ) break; 4275d9151526Sdan if( pApply->constraints.nBuf>=cons.nBuf ){ 4276d9151526Sdan /* No progress was made on the last round. */ 4277d9151526Sdan pApply->bDeferConstraints = 0; 4278d9151526Sdan } 4279d9151526Sdan } 4280d9151526Sdan 4281d9151526Sdan return rc; 4282d9151526Sdan } 4283d9151526Sdan 4284296c7658Sdan /* 42854757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but 42864757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the 42874757c658Sdan ** changeset to the main database attached to handle "db". The supplied 42884757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered 42894757c658Sdan ** while applying the change. 4290296c7658Sdan */ 42914757c658Sdan static int sessionChangesetApply( 4292296c7658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 42934757c658Sdan sqlite3_changeset_iter *pIter, /* Changeset to apply */ 429440368988Sdan int(*xFilter)( 429540368988Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 429640368988Sdan const char *zTab /* Table name */ 429740368988Sdan ), 4298d5f0767cSdan int(*xConflict)( 4299d5f0767cSdan void *pCtx, /* Copy of fifth arg to _apply() */ 4300d5f0767cSdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4301d5f0767cSdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4302d5f0767cSdan ), 4303a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4304fe55da38Sdan void **ppRebase, int *pnRebase, /* OUT: Rebase information */ 4305fe55da38Sdan int flags /* SESSION_APPLY_XXX flags */ 4306d5f0767cSdan ){ 4307ca62ad57Sdan int schemaMismatch = 0; 4308fe55da38Sdan int rc = SQLITE_OK; /* Return code */ 4309d5f0767cSdan const char *zTab = 0; /* Name of current table */ 4310cfdbde21Sdrh int nTab = 0; /* Result of sqlite3Strlen30(zTab) */ 4311296c7658Sdan SessionApplyCtx sApply; /* changeset_apply() context object */ 43125f5663dcSdan int bPatchset; 4313d5f0767cSdan 4314082c96dfSdan assert( xConflict!=0 ); 4315082c96dfSdan 4316d9151526Sdan pIter->in.bNoDiscard = 1; 43170c698471Sdan memset(&sApply, 0, sizeof(sApply)); 4318dbe7d37aSdan sApply.bRebase = (ppRebase && pnRebase); 43194c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 4320fe55da38Sdan if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){ 43210c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0); 4322fe55da38Sdan } 4323cb3e4b79Sdan if( rc==SQLITE_OK ){ 4324cb3e4b79Sdan rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0); 4325cb3e4b79Sdan } 43260c698471Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){ 43270c698471Sdan int nCol; 4328d5f0767cSdan int op; 43290c698471Sdan const char *zNew; 4330ca62ad57Sdan 4331b4480e94Sdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0); 4332d5f0767cSdan 43330c698471Sdan if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){ 4334ca62ad57Sdan u8 *abPK; 4335ca62ad57Sdan 4336d9151526Sdan rc = sessionRetryConstraints( 4337d9151526Sdan db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx 4338d9151526Sdan ); 4339d9151526Sdan if( rc!=SQLITE_OK ) break; 4340d9151526Sdan 4341cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 43420c698471Sdan sqlite3_finalize(sApply.pDelete); 43430c698471Sdan sqlite3_finalize(sApply.pUpdate); 43440c698471Sdan sqlite3_finalize(sApply.pInsert); 43450c698471Sdan sqlite3_finalize(sApply.pSelect); 43460c698471Sdan sApply.db = db; 4347f1b40e83Sdan sApply.pDelete = 0; 4348f1b40e83Sdan sApply.pUpdate = 0; 4349f1b40e83Sdan sApply.pInsert = 0; 4350f1b40e83Sdan sApply.pSelect = 0; 4351f1b40e83Sdan sApply.nCol = 0; 4352f1b40e83Sdan sApply.azCol = 0; 4353f1b40e83Sdan sApply.abPK = 0; 4354f1b40e83Sdan sApply.bStat1 = 0; 4355d9151526Sdan sApply.bDeferConstraints = 1; 4356a38e6c57Sdan sApply.bRebaseStarted = 0; 4357f1b40e83Sdan memset(&sApply.constraints, 0, sizeof(SessionBuffer)); 435837f133ecSdan 435940368988Sdan /* If an xFilter() callback was specified, invoke it now. If the 436040368988Sdan ** xFilter callback returns zero, skip this table. If it returns 436140368988Sdan ** non-zero, proceed. */ 436240368988Sdan schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew))); 436340368988Sdan if( schemaMismatch ){ 436440368988Sdan zTab = sqlite3_mprintf("%s", zNew); 4365f05ac112Sdan if( zTab==0 ){ 4366f05ac112Sdan rc = SQLITE_NOMEM; 4367f05ac112Sdan break; 4368f05ac112Sdan } 43694f528042Sdan nTab = (int)strlen(zTab); 437040368988Sdan sApply.azCol = (const char **)zTab; 437140368988Sdan }else{ 4372ff677b20Sdan int nMinCol = 0; 4373ff677b20Sdan int i; 4374ff677b20Sdan 4375ca62ad57Sdan sqlite3changeset_pk(pIter, &abPK, 0); 4376296c7658Sdan rc = sessionTableInfo( 4377ca62ad57Sdan db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK 4378ca62ad57Sdan ); 4379ca62ad57Sdan if( rc!=SQLITE_OK ) break; 4380ff677b20Sdan for(i=0; i<sApply.nCol; i++){ 4381ff677b20Sdan if( sApply.abPK[i] ) nMinCol = i+1; 4382ff677b20Sdan } 43830c698471Sdan 4384ca62ad57Sdan if( sApply.nCol==0 ){ 4385ca62ad57Sdan schemaMismatch = 1; 4386ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4387ca62ad57Sdan "sqlite3changeset_apply(): no such table: %s", zTab 4388ca62ad57Sdan ); 4389ca62ad57Sdan } 4390ff677b20Sdan else if( sApply.nCol<nCol ){ 4391ca62ad57Sdan schemaMismatch = 1; 4392ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4393ff677b20Sdan "sqlite3changeset_apply(): table %s has %d columns, " 4394ff677b20Sdan "expected %d or more", 4395ca62ad57Sdan zTab, sApply.nCol, nCol 4396ca62ad57Sdan ); 4397ca62ad57Sdan } 4398ff677b20Sdan else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){ 4399ca62ad57Sdan schemaMismatch = 1; 440040368988Sdan sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): " 440140368988Sdan "primary key mismatch for table %s", zTab 4402ca62ad57Sdan ); 4403ca62ad57Sdan } 4404ff677b20Sdan else{ 4405ff677b20Sdan sApply.nCol = nCol; 44063739f298Sdan if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){ 44073739f298Sdan if( (rc = sessionStat1Sql(db, &sApply) ) ){ 44083739f298Sdan break; 44093739f298Sdan } 4410d1cccf19Sdan sApply.bStat1 = 1; 4411d1cccf19Sdan }else{ 4412ff677b20Sdan if((rc = sessionSelectRow(db, zTab, &sApply)) 44130c698471Sdan || (rc = sessionUpdateRow(db, zTab, &sApply)) 44140c698471Sdan || (rc = sessionDeleteRow(db, zTab, &sApply)) 44150c698471Sdan || (rc = sessionInsertRow(db, zTab, &sApply)) 441637f133ecSdan ){ 441737f133ecSdan break; 441837f133ecSdan } 4419d1cccf19Sdan sApply.bStat1 = 0; 4420d1cccf19Sdan } 4421ff677b20Sdan } 4422cfdbde21Sdrh nTab = sqlite3Strlen30(zTab); 4423d5f0767cSdan } 442440368988Sdan } 4425d5f0767cSdan 4426ca62ad57Sdan /* If there is a schema mismatch on the current table, proceed to the 4427ca62ad57Sdan ** next change. A log message has already been issued. */ 4428ca62ad57Sdan if( schemaMismatch ) continue; 4429ca62ad57Sdan 4430d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx); 44310c698471Sdan } 44320c698471Sdan 44335f5663dcSdan bPatchset = pIter->bPatchset; 4434296c7658Sdan if( rc==SQLITE_OK ){ 4435296c7658Sdan rc = sqlite3changeset_finalize(pIter); 4436296c7658Sdan }else{ 4437296c7658Sdan sqlite3changeset_finalize(pIter); 4438296c7658Sdan } 4439d5f0767cSdan 4440d5f0767cSdan if( rc==SQLITE_OK ){ 44415f5663dcSdan rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx); 44425f5663dcSdan } 44435f5663dcSdan 44445f5663dcSdan if( rc==SQLITE_OK ){ 444507001c45Sdrh int nFk, notUsed; 444607001c45Sdrh sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, ¬Used, 0); 444707001c45Sdrh if( nFk!=0 ){ 4448cb3e4b79Sdan int res = SQLITE_CHANGESET_ABORT; 4449cb3e4b79Sdan sqlite3_changeset_iter sIter; 4450cb3e4b79Sdan memset(&sIter, 0, sizeof(sIter)); 4451cb3e4b79Sdan sIter.nCol = nFk; 4452cb3e4b79Sdan res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter); 4453cb3e4b79Sdan if( res!=SQLITE_CHANGESET_OMIT ){ 4454cb3e4b79Sdan rc = SQLITE_CONSTRAINT; 4455cb3e4b79Sdan } 4456cb3e4b79Sdan } 4457cb3e4b79Sdan } 4458cb3e4b79Sdan sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0); 4459cb3e4b79Sdan 4460fe55da38Sdan if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){ 4461cb3e4b79Sdan if( rc==SQLITE_OK ){ 4462d5f0767cSdan rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4463d5f0767cSdan }else{ 4464d5f0767cSdan sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0); 4465d5f0767cSdan sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4466d5f0767cSdan } 4467fe55da38Sdan } 4468d5f0767cSdan 4469dbe7d37aSdan assert( sApply.bRebase || sApply.rebase.nBuf==0 ); 4470dbe7d37aSdan if( rc==SQLITE_OK && bPatchset==0 && sApply.bRebase ){ 4471a38e6c57Sdan *ppRebase = (void*)sApply.rebase.aBuf; 4472a38e6c57Sdan *pnRebase = sApply.rebase.nBuf; 4473a38e6c57Sdan sApply.rebase.aBuf = 0; 4474a38e6c57Sdan } 44750c698471Sdan sqlite3_finalize(sApply.pInsert); 44760c698471Sdan sqlite3_finalize(sApply.pDelete); 44770c698471Sdan sqlite3_finalize(sApply.pUpdate); 44780c698471Sdan sqlite3_finalize(sApply.pSelect); 4479cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 4480d9151526Sdan sqlite3_free((char*)sApply.constraints.aBuf); 4481a38e6c57Sdan sqlite3_free((char*)sApply.rebase.aBuf); 44824c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 4483d5f0767cSdan return rc; 4484d5f0767cSdan } 448591ddd559Sdan 4486b880a7b1Sdan /* 4487b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main 4488b880a7b1Sdan ** database attached to handle "db". 4489b880a7b1Sdan */ 4490a38e6c57Sdan int sqlite3changeset_apply_v2( 4491a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4492a38e6c57Sdan int nChangeset, /* Size of changeset in bytes */ 4493a38e6c57Sdan void *pChangeset, /* Changeset blob */ 4494a38e6c57Sdan int(*xFilter)( 4495a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4496a38e6c57Sdan const char *zTab /* Table name */ 4497a38e6c57Sdan ), 4498a38e6c57Sdan int(*xConflict)( 4499a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4500a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4501a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4502a38e6c57Sdan ), 4503a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4504fe55da38Sdan void **ppRebase, int *pnRebase, 4505fe55da38Sdan int flags 4506a38e6c57Sdan ){ 4507a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 450844748f27Sdan int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT); 450944748f27Sdan int rc = sessionChangesetStart(&pIter, 0, 0, nChangeset, pChangeset,bInverse); 4510a38e6c57Sdan if( rc==SQLITE_OK ){ 4511a38e6c57Sdan rc = sessionChangesetApply( 4512fe55da38Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags 4513a38e6c57Sdan ); 4514a38e6c57Sdan } 4515a38e6c57Sdan return rc; 4516a38e6c57Sdan } 4517a38e6c57Sdan 451877fc1d5bSdan /* 45194757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database 45204757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 45214757c658Sdan ** to resolve any conflicts encountered while applying the change. 45224757c658Sdan */ 45234757c658Sdan int sqlite3changeset_apply( 45244757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 45254757c658Sdan int nChangeset, /* Size of changeset in bytes */ 45264757c658Sdan void *pChangeset, /* Changeset blob */ 45274757c658Sdan int(*xFilter)( 45284757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45294757c658Sdan const char *zTab /* Table name */ 45304757c658Sdan ), 45314757c658Sdan int(*xConflict)( 45324757c658Sdan void *pCtx, /* Copy of fifth arg to _apply() */ 45334757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 45344757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 45354757c658Sdan ), 45364757c658Sdan void *pCtx /* First argument passed to xConflict */ 45374757c658Sdan ){ 4538a38e6c57Sdan return sqlite3changeset_apply_v2( 4539fe55da38Sdan db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0, 0 4540a38e6c57Sdan ); 45414757c658Sdan } 45424757c658Sdan 45434757c658Sdan /* 45444757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database 45454757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 45464757c658Sdan ** to resolve any conflicts encountered while applying the change. 45474757c658Sdan */ 4548a38e6c57Sdan int sqlite3changeset_apply_v2_strm( 4549a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4550a38e6c57Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 4551a38e6c57Sdan void *pIn, /* First arg for xInput */ 4552a38e6c57Sdan int(*xFilter)( 4553a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4554a38e6c57Sdan const char *zTab /* Table name */ 4555a38e6c57Sdan ), 4556a38e6c57Sdan int(*xConflict)( 4557a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4558a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4559a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4560a38e6c57Sdan ), 4561a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4562fe55da38Sdan void **ppRebase, int *pnRebase, 4563fe55da38Sdan int flags 4564a38e6c57Sdan ){ 4565a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 456644748f27Sdan int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT); 456744748f27Sdan int rc = sessionChangesetStart(&pIter, xInput, pIn, 0, 0, bInverse); 4568a38e6c57Sdan if( rc==SQLITE_OK ){ 4569a38e6c57Sdan rc = sessionChangesetApply( 4570fe55da38Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags 4571a38e6c57Sdan ); 4572a38e6c57Sdan } 4573a38e6c57Sdan return rc; 4574a38e6c57Sdan } 4575f1a08ad8Sdrh int sqlite3changeset_apply_strm( 45764757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 45774757c658Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 45784757c658Sdan void *pIn, /* First arg for xInput */ 45794757c658Sdan int(*xFilter)( 45804757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45814757c658Sdan const char *zTab /* Table name */ 45824757c658Sdan ), 45834757c658Sdan int(*xConflict)( 45844757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45854757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 45864757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 45874757c658Sdan ), 45884757c658Sdan void *pCtx /* First argument passed to xConflict */ 45894757c658Sdan ){ 4590a38e6c57Sdan return sqlite3changeset_apply_v2_strm( 4591fe55da38Sdan db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0, 0 4592a38e6c57Sdan ); 45934757c658Sdan } 45944757c658Sdan 45954757c658Sdan /* 45965898ad69Sdan ** sqlite3_changegroup handle. 45975898ad69Sdan */ 45985898ad69Sdan struct sqlite3_changegroup { 45995898ad69Sdan int rc; /* Error code */ 46005898ad69Sdan int bPatch; /* True to accumulate patchsets */ 46015898ad69Sdan SessionTable *pList; /* List of tables in current patch */ 46025898ad69Sdan }; 46035898ad69Sdan 46045898ad69Sdan /* 460577fc1d5bSdan ** This function is called to merge two changes to the same row together as 460677fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is 460777fc1d5bSdan ** allocated and a pointer to it stored in *ppNew. 460877fc1d5bSdan */ 46095d607a6eSdan static int sessionChangeMerge( 461077fc1d5bSdan SessionTable *pTab, /* Table structure */ 4611c0a499eaSdan int bRebase, /* True for a rebase hash-table */ 461264277f4aSdan int bPatchset, /* True for patchsets */ 461377fc1d5bSdan SessionChange *pExist, /* Existing change */ 461477fc1d5bSdan int op2, /* Second change operation */ 461577fc1d5bSdan int bIndirect, /* True if second change is indirect */ 461677fc1d5bSdan u8 *aRec, /* Second change record */ 461777fc1d5bSdan int nRec, /* Number of bytes in aRec */ 461877fc1d5bSdan SessionChange **ppNew /* OUT: Merged change */ 46195d607a6eSdan ){ 46205d607a6eSdan SessionChange *pNew = 0; 462124a0c453Sdan int rc = SQLITE_OK; 46225d607a6eSdan 46235d607a6eSdan if( !pExist ){ 46242d77d80aSdrh pNew = (SessionChange *)sqlite3_malloc64(sizeof(SessionChange) + nRec); 46255d607a6eSdan if( !pNew ){ 46265d607a6eSdan return SQLITE_NOMEM; 46275d607a6eSdan } 46285d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 4629798693b2Sdan pNew->op = op2; 46305d607a6eSdan pNew->bIndirect = bIndirect; 4631cbf6d2d2Sdan pNew->aRecord = (u8*)&pNew[1]; 463224a0c453Sdan if( bIndirect==0 || bRebase==0 ){ 463324a0c453Sdan pNew->nRecord = nRec; 4634cbf6d2d2Sdan memcpy(pNew->aRecord, aRec, nRec); 463524a0c453Sdan }else{ 463624a0c453Sdan int i; 463724a0c453Sdan u8 *pIn = aRec; 463824a0c453Sdan u8 *pOut = pNew->aRecord; 463924a0c453Sdan for(i=0; i<pTab->nCol; i++){ 464024a0c453Sdan int nIn = sessionSerialLen(pIn); 464124a0c453Sdan if( *pIn==0 ){ 464224a0c453Sdan *pOut++ = 0; 464324a0c453Sdan }else if( pTab->abPK[i]==0 ){ 464424a0c453Sdan *pOut++ = 0xFF; 464524a0c453Sdan }else{ 464624a0c453Sdan memcpy(pOut, pIn, nIn); 464724a0c453Sdan pOut += nIn; 464824a0c453Sdan } 464924a0c453Sdan pIn += nIn; 465024a0c453Sdan } 465124a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 465224a0c453Sdan } 4653c0a499eaSdan }else if( bRebase ){ 465424a0c453Sdan if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){ 465524a0c453Sdan *ppNew = pExist; 465624a0c453Sdan }else{ 46572d77d80aSdrh sqlite3_int64 nByte = nRec + pExist->nRecord + sizeof(SessionChange); 46582d77d80aSdrh pNew = (SessionChange*)sqlite3_malloc64(nByte); 465924a0c453Sdan if( pNew==0 ){ 466024a0c453Sdan rc = SQLITE_NOMEM; 466124a0c453Sdan }else{ 466224a0c453Sdan int i; 466324a0c453Sdan u8 *a1 = pExist->aRecord; 466424a0c453Sdan u8 *a2 = aRec; 466524a0c453Sdan u8 *pOut; 466624a0c453Sdan 466724a0c453Sdan memset(pNew, 0, nByte); 466824a0c453Sdan pNew->bIndirect = bIndirect || pExist->bIndirect; 466924a0c453Sdan pNew->op = op2; 467024a0c453Sdan pOut = pNew->aRecord = (u8*)&pNew[1]; 467124a0c453Sdan 467224a0c453Sdan for(i=0; i<pTab->nCol; i++){ 467324a0c453Sdan int n1 = sessionSerialLen(a1); 467424a0c453Sdan int n2 = sessionSerialLen(a2); 4675f231e18cSdan if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){ 467624a0c453Sdan *pOut++ = 0xFF; 467724a0c453Sdan }else if( *a2==0 ){ 467824a0c453Sdan memcpy(pOut, a1, n1); 467924a0c453Sdan pOut += n1; 468024a0c453Sdan }else{ 468124a0c453Sdan memcpy(pOut, a2, n2); 468224a0c453Sdan pOut += n2; 468324a0c453Sdan } 468424a0c453Sdan a1 += n1; 468524a0c453Sdan a2 += n2; 468624a0c453Sdan } 468724a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 468824a0c453Sdan } 468924a0c453Sdan sqlite3_free(pExist); 469024a0c453Sdan } 46915d607a6eSdan }else{ 4692798693b2Sdan int op1 = pExist->op; 46935d607a6eSdan 46945d607a6eSdan /* 46955d607a6eSdan ** op1=INSERT, op2=INSERT -> Unsupported. Discard op2. 46965d607a6eSdan ** op1=INSERT, op2=UPDATE -> INSERT. 46975d607a6eSdan ** op1=INSERT, op2=DELETE -> (none) 46985d607a6eSdan ** 46995d607a6eSdan ** op1=UPDATE, op2=INSERT -> Unsupported. Discard op2. 47005d607a6eSdan ** op1=UPDATE, op2=UPDATE -> UPDATE. 47015d607a6eSdan ** op1=UPDATE, op2=DELETE -> DELETE. 47025d607a6eSdan ** 47035d607a6eSdan ** op1=DELETE, op2=INSERT -> UPDATE. 47045d607a6eSdan ** op1=DELETE, op2=UPDATE -> Unsupported. Discard op2. 47055d607a6eSdan ** op1=DELETE, op2=DELETE -> Unsupported. Discard op2. 47065d607a6eSdan */ 47075d607a6eSdan if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT) 47085d607a6eSdan || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT) 47095d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE) 47105d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_DELETE) 47115d607a6eSdan ){ 47125d607a6eSdan pNew = pExist; 47135d607a6eSdan }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){ 47145d607a6eSdan sqlite3_free(pExist); 47155d607a6eSdan assert( pNew==0 ); 47165d607a6eSdan }else{ 471764277f4aSdan u8 *aExist = pExist->aRecord; 47182d77d80aSdrh sqlite3_int64 nByte; 47195d607a6eSdan u8 *aCsr; 47205d607a6eSdan 472164277f4aSdan /* Allocate a new SessionChange object. Ensure that the aRecord[] 472264277f4aSdan ** buffer of the new object is large enough to hold any record that 472364277f4aSdan ** may be generated by combining the input records. */ 47245d607a6eSdan nByte = sizeof(SessionChange) + pExist->nRecord + nRec; 47252d77d80aSdrh pNew = (SessionChange *)sqlite3_malloc64(nByte); 47265d607a6eSdan if( !pNew ){ 47271756ae10Sdan sqlite3_free(pExist); 47285d607a6eSdan return SQLITE_NOMEM; 47295d607a6eSdan } 47305d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 47315d607a6eSdan pNew->bIndirect = (bIndirect && pExist->bIndirect); 47325d607a6eSdan aCsr = pNew->aRecord = (u8 *)&pNew[1]; 47335d607a6eSdan 4734b08a1efaSdan if( op1==SQLITE_INSERT ){ /* INSERT + UPDATE */ 47355d607a6eSdan u8 *a1 = aRec; 4736b08a1efaSdan assert( op2==SQLITE_UPDATE ); 4737798693b2Sdan pNew->op = SQLITE_INSERT; 4738ef7a6304Sdan if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol); 473964277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1); 4740b08a1efaSdan }else if( op1==SQLITE_DELETE ){ /* DELETE + INSERT */ 4741b08a1efaSdan assert( op2==SQLITE_INSERT ); 4742798693b2Sdan pNew->op = SQLITE_UPDATE; 4743fa29ecc4Sdan if( bPatchset ){ 4744fa29ecc4Sdan memcpy(aCsr, aRec, nRec); 4745fa29ecc4Sdan aCsr += nRec; 4746fa29ecc4Sdan }else{ 474764277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){ 4748b08a1efaSdan sqlite3_free(pNew); 4749b08a1efaSdan pNew = 0; 47505d607a6eSdan } 4751fa29ecc4Sdan } 4752b08a1efaSdan }else if( op2==SQLITE_UPDATE ){ /* UPDATE + UPDATE */ 475364277f4aSdan u8 *a1 = aExist; 47545d607a6eSdan u8 *a2 = aRec; 4755cfec7eeeSdan assert( op1==SQLITE_UPDATE ); 475664277f4aSdan if( bPatchset==0 ){ 4757ef7a6304Sdan sessionSkipRecord(&a1, pTab->nCol); 4758ef7a6304Sdan sessionSkipRecord(&a2, pTab->nCol); 475964277f4aSdan } 4760798693b2Sdan pNew->op = SQLITE_UPDATE; 476164277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){ 47625d607a6eSdan sqlite3_free(pNew); 47635d607a6eSdan pNew = 0; 47645d607a6eSdan } 4765b08a1efaSdan }else{ /* UPDATE + DELETE */ 4766b08a1efaSdan assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE ); 4767798693b2Sdan pNew->op = SQLITE_DELETE; 476864277f4aSdan if( bPatchset ){ 476964277f4aSdan memcpy(aCsr, aRec, nRec); 477064277f4aSdan aCsr += nRec; 477164277f4aSdan }else{ 477264277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist); 477364277f4aSdan } 47745d607a6eSdan } 47755d607a6eSdan 47765d607a6eSdan if( pNew ){ 47774f528042Sdan pNew->nRecord = (int)(aCsr - pNew->aRecord); 47785d607a6eSdan } 47795d607a6eSdan sqlite3_free(pExist); 47805d607a6eSdan } 47815d607a6eSdan } 47825d607a6eSdan 47835d607a6eSdan *ppNew = pNew; 478424a0c453Sdan return rc; 47855d607a6eSdan } 47865d607a6eSdan 478777fc1d5bSdan /* 47885898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as 47895898ad69Sdan ** the first argument to the changegroup hash tables. 479077fc1d5bSdan */ 479116228167Sdan static int sessionChangesetToHash( 4792cbf6d2d2Sdan sqlite3_changeset_iter *pIter, /* Iterator to read from */ 4793c0a499eaSdan sqlite3_changegroup *pGrp, /* Changegroup object to add changeset to */ 4794c0a499eaSdan int bRebase /* True if hash table is for rebasing */ 47955d607a6eSdan ){ 47965d607a6eSdan u8 *aRec; 47975d607a6eSdan int nRec; 4798cbf6d2d2Sdan int rc = SQLITE_OK; 47995d607a6eSdan SessionTable *pTab = 0; 48005d607a6eSdan 4801c0a499eaSdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){ 48025d607a6eSdan const char *zNew; 48035d607a6eSdan int nCol; 48045d607a6eSdan int op; 48055d607a6eSdan int iHash; 48065d607a6eSdan int bIndirect; 48075d607a6eSdan SessionChange *pChange; 48085d607a6eSdan SessionChange *pExist = 0; 48095d607a6eSdan SessionChange **pp; 48105d607a6eSdan 48115898ad69Sdan if( pGrp->pList==0 ){ 48125898ad69Sdan pGrp->bPatch = pIter->bPatchset; 48135898ad69Sdan }else if( pIter->bPatchset!=pGrp->bPatch ){ 48145898ad69Sdan rc = SQLITE_ERROR; 48155898ad69Sdan break; 48165898ad69Sdan } 48175898ad69Sdan 48185d607a6eSdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect); 4819ef7a6304Sdan if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){ 48205d607a6eSdan /* Search the list for a matching table */ 48214f528042Sdan int nNew = (int)strlen(zNew); 4822f29123b5Sdan u8 *abPK; 4823f29123b5Sdan 4824f29123b5Sdan sqlite3changeset_pk(pIter, &abPK, 0); 48255898ad69Sdan for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){ 48265d607a6eSdan if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break; 48275d607a6eSdan } 48285d607a6eSdan if( !pTab ){ 48296c39e6a8Sdan SessionTable **ppTab; 48306c39e6a8Sdan 48312d77d80aSdrh pTab = sqlite3_malloc64(sizeof(SessionTable) + nCol + nNew+1); 4832f29123b5Sdan if( !pTab ){ 4833f29123b5Sdan rc = SQLITE_NOMEM; 4834f29123b5Sdan break; 4835f29123b5Sdan } 48365d607a6eSdan memset(pTab, 0, sizeof(SessionTable)); 4837f29123b5Sdan pTab->nCol = nCol; 4838ef7a6304Sdan pTab->abPK = (u8*)&pTab[1]; 4839ef7a6304Sdan memcpy(pTab->abPK, abPK, nCol); 4840ef7a6304Sdan pTab->zName = (char*)&pTab->abPK[nCol]; 4841ef7a6304Sdan memcpy(pTab->zName, zNew, nNew+1); 48426c39e6a8Sdan 48436c39e6a8Sdan /* The new object must be linked on to the end of the list, not 48446c39e6a8Sdan ** simply added to the start of it. This is to ensure that the 48456c39e6a8Sdan ** tables within the output of sqlite3changegroup_output() are in 48466c39e6a8Sdan ** the right order. */ 48476c39e6a8Sdan for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext); 48486c39e6a8Sdan *ppTab = pTab; 4849f29123b5Sdan }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){ 4850f29123b5Sdan rc = SQLITE_SCHEMA; 4851f29123b5Sdan break; 48525d607a6eSdan } 48535d607a6eSdan } 48545d607a6eSdan 4855cbf6d2d2Sdan if( sessionGrowHash(pIter->bPatchset, pTab) ){ 48561756ae10Sdan rc = SQLITE_NOMEM; 48571756ae10Sdan break; 48581756ae10Sdan } 485964277f4aSdan iHash = sessionChangeHash( 4860cbf6d2d2Sdan pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange 486164277f4aSdan ); 48625d607a6eSdan 48635d607a6eSdan /* Search for existing entry. If found, remove it from the hash table. 48645d607a6eSdan ** Code below may link it back in. 48655d607a6eSdan */ 48665d607a6eSdan for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){ 486764277f4aSdan int bPkOnly1 = 0; 486864277f4aSdan int bPkOnly2 = 0; 4869cbf6d2d2Sdan if( pIter->bPatchset ){ 487064277f4aSdan bPkOnly1 = (*pp)->op==SQLITE_DELETE; 487164277f4aSdan bPkOnly2 = op==SQLITE_DELETE; 487264277f4aSdan } 487364277f4aSdan if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){ 48745d607a6eSdan pExist = *pp; 48755d607a6eSdan *pp = (*pp)->pNext; 48765d607a6eSdan pTab->nEntry--; 48775d607a6eSdan break; 48785d607a6eSdan } 48795d607a6eSdan } 48805d607a6eSdan 4881c0a499eaSdan rc = sessionChangeMerge(pTab, bRebase, 4882cbf6d2d2Sdan pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange 488364277f4aSdan ); 48845d607a6eSdan if( rc ) break; 48855d607a6eSdan if( pChange ){ 48865d607a6eSdan pChange->pNext = pTab->apChange[iHash]; 48875d607a6eSdan pTab->apChange[iHash] = pChange; 48885d607a6eSdan pTab->nEntry++; 48895d607a6eSdan } 48905d607a6eSdan } 48915d607a6eSdan 4892cbf6d2d2Sdan if( rc==SQLITE_OK ) rc = pIter->rc; 48935d607a6eSdan return rc; 48945d607a6eSdan } 48955d607a6eSdan 48965d607a6eSdan /* 48975898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets) 48985898ad69Sdan ** added to the changegroup object passed as the first argument. 48995d607a6eSdan ** 49005898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the 49015898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming 49025898ad69Sdan ** interfaces. 49035d607a6eSdan ** 49045898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a 49055898ad69Sdan ** buffer containing the output changeset before this function returns. In 49065898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It 49075898ad69Sdan ** is the responsibility of the caller to free the output buffer using 49085898ad69Sdan ** sqlite3_free() when it is no longer required. 49095898ad69Sdan ** 49105898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite 49115898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut) 49125898ad69Sdan ** are both set to 0 before returning. 49135d607a6eSdan */ 49145898ad69Sdan static int sessionChangegroupOutput( 49155898ad69Sdan sqlite3_changegroup *pGrp, 4916cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4917cbf6d2d2Sdan void *pOut, 4918cbf6d2d2Sdan int *pnOut, 4919cbf6d2d2Sdan void **ppOut 49205d607a6eSdan ){ 49215898ad69Sdan int rc = SQLITE_OK; 4922e8fa8c96Sdan SessionBuffer buf = {0, 0, 0}; 49235898ad69Sdan SessionTable *pTab; 4924cbf6d2d2Sdan assert( xOutput==0 || (ppOut==0 && pnOut==0) ); 49255d607a6eSdan 49265d607a6eSdan /* Create the serialized output changeset based on the contents of the 49275898ad69Sdan ** hash tables attached to the SessionTable objects in list p->pList. 49285d607a6eSdan */ 49295898ad69Sdan for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 49305d607a6eSdan int i; 49315d607a6eSdan if( pTab->nEntry==0 ) continue; 49325d607a6eSdan 49335898ad69Sdan sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc); 49345d607a6eSdan for(i=0; i<pTab->nChange; i++){ 49355d607a6eSdan SessionChange *p; 49365d607a6eSdan for(p=pTab->apChange[i]; p; p=p->pNext){ 4937798693b2Sdan sessionAppendByte(&buf, p->op, &rc); 49385d607a6eSdan sessionAppendByte(&buf, p->bIndirect, &rc); 49395d607a6eSdan sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc); 49401f48e67dSdan if( rc==SQLITE_OK && xOutput && buf.nBuf>=sessions_strm_chunk_size ){ 4941cbf6d2d2Sdan rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4942cbf6d2d2Sdan buf.nBuf = 0; 4943cbf6d2d2Sdan } 49445d607a6eSdan } 4945f8248c42Sdan } 4946f8248c42Sdan } 49475d607a6eSdan 49485d607a6eSdan if( rc==SQLITE_OK ){ 4949cbf6d2d2Sdan if( xOutput ){ 4950cbf6d2d2Sdan if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4951cbf6d2d2Sdan }else{ 49525d607a6eSdan *ppOut = buf.aBuf; 49535d607a6eSdan *pnOut = buf.nBuf; 4954cbf6d2d2Sdan buf.aBuf = 0; 49555d607a6eSdan } 49565d607a6eSdan } 4957cbf6d2d2Sdan sqlite3_free(buf.aBuf); 49585d607a6eSdan 49595d607a6eSdan return rc; 49605d607a6eSdan } 49615d607a6eSdan 4962cbf6d2d2Sdan /* 49635898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup. 49645898ad69Sdan */ 49655898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){ 49665898ad69Sdan int rc = SQLITE_OK; /* Return code */ 49675898ad69Sdan sqlite3_changegroup *p; /* New object */ 49685898ad69Sdan p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup)); 49695898ad69Sdan if( p==0 ){ 49705898ad69Sdan rc = SQLITE_NOMEM; 49715898ad69Sdan }else{ 49725898ad69Sdan memset(p, 0, sizeof(sqlite3_changegroup)); 49735898ad69Sdan } 49745898ad69Sdan *pp = p; 49755898ad69Sdan return rc; 49765898ad69Sdan } 49775898ad69Sdan 49785898ad69Sdan /* 49795898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes, 49805898ad69Sdan ** to changeset-group p. 49815898ad69Sdan */ 49825898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){ 49835898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 49845898ad69Sdan int rc; /* Return code */ 49855898ad69Sdan 49865898ad69Sdan rc = sqlite3changeset_start(&pIter, nData, pData); 49875898ad69Sdan if( rc==SQLITE_OK ){ 4988c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 49895898ad69Sdan } 49905898ad69Sdan sqlite3changeset_finalize(pIter); 49915898ad69Sdan return rc; 49925898ad69Sdan } 49935898ad69Sdan 49945898ad69Sdan /* 49955898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation 49965898ad69Sdan ** of all changesets added to the group so far. 49975898ad69Sdan */ 49985898ad69Sdan int sqlite3changegroup_output( 49995898ad69Sdan sqlite3_changegroup *pGrp, 50005898ad69Sdan int *pnData, 50015898ad69Sdan void **ppData 50025898ad69Sdan ){ 50035898ad69Sdan return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData); 50045898ad69Sdan } 50055898ad69Sdan 50065898ad69Sdan /* 50075898ad69Sdan ** Streaming versions of changegroup_add(). 50085898ad69Sdan */ 50095898ad69Sdan int sqlite3changegroup_add_strm( 50105898ad69Sdan sqlite3_changegroup *pGrp, 50115898ad69Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 50125898ad69Sdan void *pIn 50135898ad69Sdan ){ 50145898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 50155898ad69Sdan int rc; /* Return code */ 50165898ad69Sdan 50175898ad69Sdan rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 50185898ad69Sdan if( rc==SQLITE_OK ){ 5019c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 50205898ad69Sdan } 50215898ad69Sdan sqlite3changeset_finalize(pIter); 50225898ad69Sdan return rc; 50235898ad69Sdan } 50245898ad69Sdan 50255898ad69Sdan /* 50265898ad69Sdan ** Streaming versions of changegroup_output(). 50275898ad69Sdan */ 50285898ad69Sdan int sqlite3changegroup_output_strm( 50295898ad69Sdan sqlite3_changegroup *pGrp, 50305898ad69Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 50315898ad69Sdan void *pOut 50325898ad69Sdan ){ 50335898ad69Sdan return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0); 50345898ad69Sdan } 50355898ad69Sdan 50365898ad69Sdan /* 50375898ad69Sdan ** Delete a changegroup object. 50385898ad69Sdan */ 50395898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){ 50405898ad69Sdan if( pGrp ){ 50415898ad69Sdan sessionDeleteTable(pGrp->pList); 50425898ad69Sdan sqlite3_free(pGrp); 50435898ad69Sdan } 50445898ad69Sdan } 50455898ad69Sdan 50465898ad69Sdan /* 5047cbf6d2d2Sdan ** Combine two changesets together. 5048cbf6d2d2Sdan */ 5049cbf6d2d2Sdan int sqlite3changeset_concat( 5050cbf6d2d2Sdan int nLeft, /* Number of bytes in lhs input */ 5051cbf6d2d2Sdan void *pLeft, /* Lhs input changeset */ 5052cbf6d2d2Sdan int nRight /* Number of bytes in rhs input */, 5053cbf6d2d2Sdan void *pRight, /* Rhs input changeset */ 5054cbf6d2d2Sdan int *pnOut, /* OUT: Number of bytes in output changeset */ 5055cbf6d2d2Sdan void **ppOut /* OUT: changeset (left <concat> right) */ 5056cbf6d2d2Sdan ){ 50575898ad69Sdan sqlite3_changegroup *pGrp; 5058cbf6d2d2Sdan int rc; 5059cbf6d2d2Sdan 50605898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 5061cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50625898ad69Sdan rc = sqlite3changegroup_add(pGrp, nLeft, pLeft); 5063cbf6d2d2Sdan } 5064cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50655898ad69Sdan rc = sqlite3changegroup_add(pGrp, nRight, pRight); 5066cbf6d2d2Sdan } 50675898ad69Sdan if( rc==SQLITE_OK ){ 50685898ad69Sdan rc = sqlite3changegroup_output(pGrp, pnOut, ppOut); 50695898ad69Sdan } 50705898ad69Sdan sqlite3changegroup_delete(pGrp); 5071cbf6d2d2Sdan 5072cbf6d2d2Sdan return rc; 5073cbf6d2d2Sdan } 5074cbf6d2d2Sdan 5075cbf6d2d2Sdan /* 5076cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat(). 5077cbf6d2d2Sdan */ 5078f1a08ad8Sdrh int sqlite3changeset_concat_strm( 5079cbf6d2d2Sdan int (*xInputA)(void *pIn, void *pData, int *pnData), 5080cbf6d2d2Sdan void *pInA, 5081cbf6d2d2Sdan int (*xInputB)(void *pIn, void *pData, int *pnData), 5082cbf6d2d2Sdan void *pInB, 5083cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 5084cbf6d2d2Sdan void *pOut 5085cbf6d2d2Sdan ){ 50865898ad69Sdan sqlite3_changegroup *pGrp; 5087cbf6d2d2Sdan int rc; 5088cbf6d2d2Sdan 50895898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 5090cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50915898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA); 5092cbf6d2d2Sdan } 5093cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50945898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB); 5095cbf6d2d2Sdan } 50965898ad69Sdan if( rc==SQLITE_OK ){ 50975898ad69Sdan rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut); 50985898ad69Sdan } 50995898ad69Sdan sqlite3changegroup_delete(pGrp); 5100cbf6d2d2Sdan 5101cbf6d2d2Sdan return rc; 5102cbf6d2d2Sdan } 5103cbf6d2d2Sdan 51043fa5463cSdan /* 51053fa5463cSdan ** Changeset rebaser handle. 51063fa5463cSdan */ 5107c0a499eaSdan struct sqlite3_rebaser { 5108c0a499eaSdan sqlite3_changegroup grp; /* Hash table */ 5109c0a499eaSdan }; 5110c0a499eaSdan 5111c0a499eaSdan /* 5112c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol 5113c0a499eaSdan ** fields in size. This function appends an nCol sessions module 5114b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for 5115b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[]. 5116c0a499eaSdan */ 5117c0a499eaSdan static void sessionAppendRecordMerge( 5118b880a7b1Sdan SessionBuffer *pBuf, /* Buffer to append to */ 5119b880a7b1Sdan int nCol, /* Number of columns in each record */ 5120b880a7b1Sdan u8 *a1, int n1, /* Record 1 */ 5121b880a7b1Sdan u8 *a2, int n2, /* Record 2 */ 5122b880a7b1Sdan int *pRc /* IN/OUT: error code */ 5123c0a499eaSdan ){ 5124c0a499eaSdan sessionBufferGrow(pBuf, n1+n2, pRc); 5125c0a499eaSdan if( *pRc==SQLITE_OK ){ 5126c0a499eaSdan int i; 5127c0a499eaSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5128c0a499eaSdan for(i=0; i<nCol; i++){ 5129c0a499eaSdan int nn1 = sessionSerialLen(a1); 5130c0a499eaSdan int nn2 = sessionSerialLen(a2); 513124a0c453Sdan if( *a1==0 || *a1==0xFF ){ 5132c0a499eaSdan memcpy(pOut, a2, nn2); 5133c0a499eaSdan pOut += nn2; 5134c0a499eaSdan }else{ 5135c0a499eaSdan memcpy(pOut, a1, nn1); 5136c0a499eaSdan pOut += nn1; 5137c0a499eaSdan } 5138f1b40e83Sdan a1 += nn1; 5139f1b40e83Sdan a2 += nn2; 5140c0a499eaSdan } 5141f1b40e83Sdan 5142f1b40e83Sdan pBuf->nBuf = pOut-pBuf->aBuf; 5143f1b40e83Sdan assert( pBuf->nBuf<=pBuf->nAlloc ); 5144c0a499eaSdan } 5145c0a499eaSdan } 5146c0a499eaSdan 51473fa5463cSdan /* 51483fa5463cSdan ** This function is called when rebasing a local UPDATE change against one 51493fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current 51503fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single 51513fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer 51523fa5463cSdan ** pBuf. 51533fa5463cSdan ** 51543fa5463cSdan ** Rebasing the UPDATE involves: 51553fa5463cSdan ** 51563fa5463cSdan ** * Removing any changes to fields for which the corresponding field 51573fa5463cSdan ** in the rebase buffer is set to "replaced" (type 0xFF). If this 51583fa5463cSdan ** means the UPDATE change updates no fields, nothing is appended 51593fa5463cSdan ** to the output buffer. 51603fa5463cSdan ** 51613fa5463cSdan ** * For each field modified by the local change for which the 51623fa5463cSdan ** corresponding field in the rebase buffer is not "undefined" (0x00) 51633fa5463cSdan ** or "replaced" (0xFF), the old.* value is replaced by the value 51643fa5463cSdan ** in the rebase buffer. 51653fa5463cSdan */ 5166f01d3a7eSdan static void sessionAppendPartialUpdate( 516724a0c453Sdan SessionBuffer *pBuf, /* Append record here */ 516824a0c453Sdan sqlite3_changeset_iter *pIter, /* Iterator pointed at local change */ 516924a0c453Sdan u8 *aRec, int nRec, /* Local change */ 517024a0c453Sdan u8 *aChange, int nChange, /* Record to rebase against */ 517124a0c453Sdan int *pRc /* IN/OUT: Return Code */ 5172f01d3a7eSdan ){ 5173f01d3a7eSdan sessionBufferGrow(pBuf, 2+nRec+nChange, pRc); 5174f01d3a7eSdan if( *pRc==SQLITE_OK ){ 5175f01d3a7eSdan int bData = 0; 5176f01d3a7eSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5177f01d3a7eSdan int i; 5178f01d3a7eSdan u8 *a1 = aRec; 5179f01d3a7eSdan u8 *a2 = aChange; 5180f01d3a7eSdan 5181f01d3a7eSdan *pOut++ = SQLITE_UPDATE; 5182f01d3a7eSdan *pOut++ = pIter->bIndirect; 5183f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5184f01d3a7eSdan int n1 = sessionSerialLen(a1); 5185f01d3a7eSdan int n2 = sessionSerialLen(a2); 5186f01d3a7eSdan if( pIter->abPK[i] || a2[0]==0 ){ 5187f01d3a7eSdan if( !pIter->abPK[i] ) bData = 1; 5188f01d3a7eSdan memcpy(pOut, a1, n1); 5189f01d3a7eSdan pOut += n1; 519024a0c453Sdan }else if( a2[0]!=0xFF ){ 519124a0c453Sdan bData = 1; 519224a0c453Sdan memcpy(pOut, a2, n2); 519324a0c453Sdan pOut += n2; 5194f01d3a7eSdan }else{ 5195f01d3a7eSdan *pOut++ = '\0'; 5196f01d3a7eSdan } 5197f01d3a7eSdan a1 += n1; 5198f01d3a7eSdan a2 += n2; 5199f01d3a7eSdan } 5200f01d3a7eSdan if( bData ){ 5201f01d3a7eSdan a2 = aChange; 5202f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5203f01d3a7eSdan int n1 = sessionSerialLen(a1); 5204f01d3a7eSdan int n2 = sessionSerialLen(a2); 520524a0c453Sdan if( pIter->abPK[i] || a2[0]!=0xFF ){ 5206f01d3a7eSdan memcpy(pOut, a1, n1); 5207f01d3a7eSdan pOut += n1; 5208f01d3a7eSdan }else{ 5209f01d3a7eSdan *pOut++ = '\0'; 5210f01d3a7eSdan } 5211f01d3a7eSdan a1 += n1; 5212f01d3a7eSdan a2 += n2; 5213f01d3a7eSdan } 5214f01d3a7eSdan pBuf->nBuf = (pOut - pBuf->aBuf); 5215f01d3a7eSdan } 5216f01d3a7eSdan } 5217f01d3a7eSdan } 5218f01d3a7eSdan 52193fa5463cSdan /* 52203fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases 52213fa5463cSdan ** that changeset according to the current configuration of the rebaser 52223fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput 52233fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking 52243fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL, 52253fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset 52263fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of 52273fa5463cSdan ** the buffer in bytes. It is the responsibility of the caller to eventually 52283fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free(). 52293fa5463cSdan ** 52303fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and 52313fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before 52323fa5463cSdan ** returning. 52333fa5463cSdan */ 5234c0a499eaSdan static int sessionRebase( 5235c0a499eaSdan sqlite3_rebaser *p, /* Rebaser hash table */ 5236c0a499eaSdan sqlite3_changeset_iter *pIter, /* Input data */ 5237c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5238c0a499eaSdan void *pOut, /* Context for xOutput callback */ 5239c0a499eaSdan int *pnOut, /* OUT: Number of bytes in output changeset */ 5240c0a499eaSdan void **ppOut /* OUT: Inverse of pChangeset */ 5241c0a499eaSdan ){ 5242c0a499eaSdan int rc = SQLITE_OK; 5243c0a499eaSdan u8 *aRec = 0; 5244c0a499eaSdan int nRec = 0; 5245c0a499eaSdan int bNew = 0; 5246c0a499eaSdan SessionTable *pTab = 0; 5247c0a499eaSdan SessionBuffer sOut = {0,0,0}; 5248c0a499eaSdan 5249f1b40e83Sdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){ 5250c0a499eaSdan SessionChange *pChange = 0; 5251f01d3a7eSdan int bDone = 0; 5252c0a499eaSdan 5253c0a499eaSdan if( bNew ){ 5254c0a499eaSdan const char *zTab = pIter->zTab; 5255c0a499eaSdan for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){ 5256c0a499eaSdan if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break; 5257c0a499eaSdan } 5258c0a499eaSdan bNew = 0; 5259c0a499eaSdan 526095ccb6dcSdan /* A patchset may not be rebased */ 526195ccb6dcSdan if( pIter->bPatchset ){ 526295ccb6dcSdan rc = SQLITE_ERROR; 526395ccb6dcSdan } 526495ccb6dcSdan 5265c0a499eaSdan /* Append a table header to the output for this new table */ 5266c0a499eaSdan sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc); 5267c0a499eaSdan sessionAppendVarint(&sOut, pIter->nCol, &rc); 5268c0a499eaSdan sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc); 52692c42c34dSmistachkin sessionAppendBlob(&sOut,(u8*)pIter->zTab,(int)strlen(pIter->zTab)+1,&rc); 5270c0a499eaSdan } 5271c0a499eaSdan 52728cb83beeSdan if( pTab && rc==SQLITE_OK ){ 52738cb83beeSdan int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange); 5274c0a499eaSdan 5275c0a499eaSdan for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){ 52768cb83beeSdan if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){ 5277c0a499eaSdan break; 5278c0a499eaSdan } 5279c0a499eaSdan } 5280c0a499eaSdan } 5281c0a499eaSdan 5282c0a499eaSdan if( pChange ){ 5283c0a499eaSdan assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT ); 5284f01d3a7eSdan switch( pIter->op ){ 5285f01d3a7eSdan case SQLITE_INSERT: 5286f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5287f01d3a7eSdan bDone = 1; 5288f01d3a7eSdan if( pChange->bIndirect==0 ){ 5289f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_UPDATE, &rc); 5290f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5291f01d3a7eSdan sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc); 5292f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5293f01d3a7eSdan } 5294f01d3a7eSdan } 5295f01d3a7eSdan break; 5296f01d3a7eSdan 5297f01d3a7eSdan case SQLITE_UPDATE: 5298f01d3a7eSdan bDone = 1; 5299f01d3a7eSdan if( pChange->op==SQLITE_DELETE ){ 5300f01d3a7eSdan if( pChange->bIndirect==0 ){ 5301f01d3a7eSdan u8 *pCsr = aRec; 5302f01d3a7eSdan sessionSkipRecord(&pCsr, pIter->nCol); 5303f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_INSERT, &rc); 5304f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5305b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5306f01d3a7eSdan pCsr, nRec-(pCsr-aRec), 5307f01d3a7eSdan pChange->aRecord, pChange->nRecord, &rc 5308f01d3a7eSdan ); 5309f01d3a7eSdan } 5310f01d3a7eSdan }else{ 5311f01d3a7eSdan sessionAppendPartialUpdate(&sOut, pIter, 5312f01d3a7eSdan aRec, nRec, pChange->aRecord, pChange->nRecord, &rc 5313f01d3a7eSdan ); 5314f01d3a7eSdan } 5315f01d3a7eSdan break; 5316f01d3a7eSdan 5317f01d3a7eSdan default: 5318f01d3a7eSdan assert( pIter->op==SQLITE_DELETE ); 5319f01d3a7eSdan bDone = 1; 5320f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5321f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_DELETE, &rc); 5322f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5323b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5324f01d3a7eSdan pChange->aRecord, pChange->nRecord, aRec, nRec, &rc 5325f01d3a7eSdan ); 5326f01d3a7eSdan } 5327f01d3a7eSdan break; 5328f01d3a7eSdan } 5329f01d3a7eSdan } 5330f01d3a7eSdan 5331f01d3a7eSdan if( bDone==0 ){ 5332f01d3a7eSdan sessionAppendByte(&sOut, pIter->op, &rc); 5333f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5334f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5335f01d3a7eSdan } 53361f48e67dSdan if( rc==SQLITE_OK && xOutput && sOut.nBuf>sessions_strm_chunk_size ){ 5337c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5338c0a499eaSdan sOut.nBuf = 0; 5339c0a499eaSdan } 5340c0a499eaSdan if( rc ) break; 5341c0a499eaSdan } 5342c0a499eaSdan 5343c0a499eaSdan if( rc!=SQLITE_OK ){ 5344c0a499eaSdan sqlite3_free(sOut.aBuf); 5345c0a499eaSdan memset(&sOut, 0, sizeof(sOut)); 5346c0a499eaSdan } 5347c0a499eaSdan 5348c0a499eaSdan if( rc==SQLITE_OK ){ 5349c0a499eaSdan if( xOutput ){ 5350c0a499eaSdan if( sOut.nBuf>0 ){ 5351c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5352c0a499eaSdan } 5353c0a499eaSdan }else{ 5354c0a499eaSdan *ppOut = (void*)sOut.aBuf; 5355c0a499eaSdan *pnOut = sOut.nBuf; 5356c0a499eaSdan sOut.aBuf = 0; 5357c0a499eaSdan } 5358c0a499eaSdan } 5359c0a499eaSdan sqlite3_free(sOut.aBuf); 5360c0a499eaSdan return rc; 5361c0a499eaSdan } 5362c0a499eaSdan 5363c0a499eaSdan /* 5364c0a499eaSdan ** Create a new rebaser object. 5365c0a499eaSdan */ 5366c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){ 5367c0a499eaSdan int rc = SQLITE_OK; 5368c0a499eaSdan sqlite3_rebaser *pNew; 5369c0a499eaSdan 5370c0a499eaSdan pNew = sqlite3_malloc(sizeof(sqlite3_rebaser)); 5371c0a499eaSdan if( pNew==0 ){ 5372c0a499eaSdan rc = SQLITE_NOMEM; 5373f1b40e83Sdan }else{ 5374f1b40e83Sdan memset(pNew, 0, sizeof(sqlite3_rebaser)); 5375c0a499eaSdan } 5376c0a499eaSdan *ppNew = pNew; 5377c0a499eaSdan return rc; 5378c0a499eaSdan } 5379c0a499eaSdan 5380c0a499eaSdan /* 5381c0a499eaSdan ** Call this one or more times to configure a rebaser. 5382c0a499eaSdan */ 5383c0a499eaSdan int sqlite3rebaser_configure( 5384c0a499eaSdan sqlite3_rebaser *p, 5385c0a499eaSdan int nRebase, const void *pRebase 5386c0a499eaSdan ){ 5387c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator opened on pData/nData */ 5388c0a499eaSdan int rc; /* Return code */ 5389c0a499eaSdan rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase); 5390c0a499eaSdan if( rc==SQLITE_OK ){ 5391c0a499eaSdan rc = sessionChangesetToHash(pIter, &p->grp, 1); 5392c0a499eaSdan } 5393c0a499eaSdan sqlite3changeset_finalize(pIter); 5394c0a499eaSdan return rc; 5395c0a499eaSdan } 5396c0a499eaSdan 5397c0a499eaSdan /* 5398c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5399c0a499eaSdan */ 5400c0a499eaSdan int sqlite3rebaser_rebase( 5401c0a499eaSdan sqlite3_rebaser *p, 5402c0a499eaSdan int nIn, const void *pIn, 5403c0a499eaSdan int *pnOut, void **ppOut 5404c0a499eaSdan ){ 5405c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5406c0a499eaSdan int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn); 5407c0a499eaSdan 5408c0a499eaSdan if( rc==SQLITE_OK ){ 5409c0a499eaSdan rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut); 5410c0a499eaSdan sqlite3changeset_finalize(pIter); 5411c0a499eaSdan } 5412c0a499eaSdan 5413c0a499eaSdan return rc; 5414c0a499eaSdan } 5415c0a499eaSdan 5416c0a499eaSdan /* 5417c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5418c0a499eaSdan */ 5419c0a499eaSdan int sqlite3rebaser_rebase_strm( 5420c0a499eaSdan sqlite3_rebaser *p, 5421c0a499eaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 5422c0a499eaSdan void *pIn, 5423c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5424c0a499eaSdan void *pOut 5425c0a499eaSdan ){ 5426c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5427c0a499eaSdan int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 5428c0a499eaSdan 5429c0a499eaSdan if( rc==SQLITE_OK ){ 5430c0a499eaSdan rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0); 5431c0a499eaSdan sqlite3changeset_finalize(pIter); 5432c0a499eaSdan } 5433c0a499eaSdan 5434c0a499eaSdan return rc; 5435c0a499eaSdan } 5436c0a499eaSdan 5437c0a499eaSdan /* 5438c0a499eaSdan ** Destroy a rebaser object 5439c0a499eaSdan */ 5440f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){ 5441c0a499eaSdan if( p ){ 5442c0a499eaSdan sessionDeleteTable(p->grp.pList); 5443c0a499eaSdan sqlite3_free(p); 5444c0a499eaSdan } 5445c0a499eaSdan } 5446c0a499eaSdan 54471f48e67dSdan /* 54481f48e67dSdan ** Global configuration 54491f48e67dSdan */ 54501f48e67dSdan int sqlite3session_config(int op, void *pArg){ 54511f48e67dSdan int rc = SQLITE_OK; 54521f48e67dSdan switch( op ){ 54531f48e67dSdan case SQLITE_SESSION_CONFIG_STRMSIZE: { 54541f48e67dSdan int *pInt = (int*)pArg; 54551f48e67dSdan if( *pInt>0 ){ 54561f48e67dSdan sessions_strm_chunk_size = *pInt; 54571f48e67dSdan } 54581f48e67dSdan *pInt = sessions_strm_chunk_size; 54591f48e67dSdan break; 54601f48e67dSdan } 54611f48e67dSdan default: 54621f48e67dSdan rc = SQLITE_MISUSE; 54631f48e67dSdan break; 54641f48e67dSdan } 54651f48e67dSdan return rc; 54661f48e67dSdan } 54671f48e67dSdan 54689b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */ 5469