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 ){ 1627595d9f5fSdan if( pzErrMsg ){ 1628b9db9099Sdan *pzErrMsg = sqlite3_mprintf("table schemas do not match"); 1629595d9f5fSdan } 1630b9db9099Sdan rc = SQLITE_SCHEMA; 1631b9db9099Sdan } 1632b9db9099Sdan if( bHasPk==0 ){ 1633b9db9099Sdan /* Ignore tables with no primary keys */ 1634b9db9099Sdan goto diff_out; 1635b9db9099Sdan } 1636cf8e9144Sdan } 1637cf8e9144Sdan 1638cf8e9144Sdan if( rc==SQLITE_OK ){ 1639cf8e9144Sdan zExpr = sessionExprComparePK(pTo->nCol, 1640cf8e9144Sdan zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK 1641cf8e9144Sdan ); 1642cf8e9144Sdan } 1643cf8e9144Sdan 1644cf8e9144Sdan /* Find new rows */ 1645cf8e9144Sdan if( rc==SQLITE_OK ){ 1646cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr); 1647cf8e9144Sdan } 1648cf8e9144Sdan 1649cf8e9144Sdan /* Find old rows */ 1650cf8e9144Sdan if( rc==SQLITE_OK ){ 1651cf8e9144Sdan rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr); 1652cf8e9144Sdan } 1653cf8e9144Sdan 1654cf8e9144Sdan /* Find modified rows */ 1655cf8e9144Sdan if( rc==SQLITE_OK ){ 1656cf8e9144Sdan rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr); 1657cf8e9144Sdan } 1658cf8e9144Sdan 1659cf8e9144Sdan sqlite3_free(zExpr); 1660cf8e9144Sdan } 1661cf8e9144Sdan 1662cf8e9144Sdan diff_out: 1663cf8e9144Sdan sessionPreupdateHooks(pSession); 166410dc553cSdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 1665cf8e9144Sdan return rc; 1666296c7658Sdan } 16674fccf43aSdan 16684fccf43aSdan /* 16694fccf43aSdan ** Create a session object. This session object will record changes to 16704fccf43aSdan ** database zDb attached to connection db. 16714fccf43aSdan */ 16724fccf43aSdan int sqlite3session_create( 16734fccf43aSdan sqlite3 *db, /* Database handle */ 16744fccf43aSdan const char *zDb, /* Name of db (e.g. "main") */ 16754fccf43aSdan sqlite3_session **ppSession /* OUT: New session object */ 16764fccf43aSdan ){ 1677296c7658Sdan sqlite3_session *pNew; /* Newly allocated session object */ 1678296c7658Sdan sqlite3_session *pOld; /* Session object already attached to db */ 1679cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */ 16804fccf43aSdan 1681296c7658Sdan /* Zero the output value in case an error occurs. */ 16824fccf43aSdan *ppSession = 0; 16834fccf43aSdan 16844fccf43aSdan /* Allocate and populate the new session object. */ 16852d77d80aSdrh pNew = (sqlite3_session *)sqlite3_malloc64(sizeof(sqlite3_session) + nDb + 1); 16864fccf43aSdan if( !pNew ) return SQLITE_NOMEM; 16874fccf43aSdan memset(pNew, 0, sizeof(sqlite3_session)); 16884fccf43aSdan pNew->db = db; 16894fccf43aSdan pNew->zDb = (char *)&pNew[1]; 1690296c7658Sdan pNew->bEnable = 1; 16914fccf43aSdan memcpy(pNew->zDb, zDb, nDb+1); 1692cf8e9144Sdan sessionPreupdateHooks(pNew); 16934fccf43aSdan 16944fccf43aSdan /* Add the new session object to the linked list of session objects 16954fccf43aSdan ** attached to database handle $db. Do this under the cover of the db 16964fccf43aSdan ** handle mutex. */ 16974fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 16984fccf43aSdan pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew); 16994fccf43aSdan pNew->pNext = pOld; 17004fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 17014fccf43aSdan 17024fccf43aSdan *ppSession = pNew; 17034fccf43aSdan return SQLITE_OK; 17044fccf43aSdan } 17054fccf43aSdan 170677fc1d5bSdan /* 170777fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents 170877fc1d5bSdan ** of the changed-rows hash tables are also deleted. 170977fc1d5bSdan */ 17101ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){ 17115d607a6eSdan SessionTable *pNext; 17125d607a6eSdan SessionTable *pTab; 17135d607a6eSdan 17145d607a6eSdan for(pTab=pList; pTab; pTab=pNext){ 17155d607a6eSdan int i; 17165d607a6eSdan pNext = pTab->pNext; 17175d607a6eSdan for(i=0; i<pTab->nChange; i++){ 17185d607a6eSdan SessionChange *p; 171902d436b1Smistachkin SessionChange *pNextChange; 172002d436b1Smistachkin for(p=pTab->apChange[i]; p; p=pNextChange){ 172102d436b1Smistachkin pNextChange = p->pNext; 17225d607a6eSdan sqlite3_free(p); 17235d607a6eSdan } 17245d607a6eSdan } 17255d607a6eSdan sqlite3_free((char*)pTab->azCol); /* cast works around VC++ bug */ 17265d607a6eSdan sqlite3_free(pTab->apChange); 17275d607a6eSdan sqlite3_free(pTab); 17285d607a6eSdan } 17295d607a6eSdan } 17305d607a6eSdan 17314fccf43aSdan /* 17324fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create(). 17334fccf43aSdan */ 17344fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){ 17354fccf43aSdan sqlite3 *db = pSession->db; 17364fccf43aSdan sqlite3_session *pHead; 17374fccf43aSdan sqlite3_session **pp; 17384fccf43aSdan 1739296c7658Sdan /* Unlink the session from the linked list of sessions attached to the 1740296c7658Sdan ** database handle. Hold the db mutex while doing so. */ 17414fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 17424fccf43aSdan pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0); 174350d348b1Sdrh for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){ 174450d348b1Sdrh if( (*pp)==pSession ){ 17454fccf43aSdan *pp = (*pp)->pNext; 17464fccf43aSdan if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead); 174750d348b1Sdrh break; 174850d348b1Sdrh } 174950d348b1Sdrh } 17504fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 17511611e5a3Sdan sqlite3ValueFree(pSession->pZeroBlob); 17524fccf43aSdan 1753296c7658Sdan /* Delete all attached table objects. And the contents of their 1754296c7658Sdan ** associated hash-tables. */ 17555d607a6eSdan sessionDeleteTable(pSession->pTable); 17564fccf43aSdan 1757296c7658Sdan /* Free the session object itself. */ 17584fccf43aSdan sqlite3_free(pSession); 17594fccf43aSdan } 17604fccf43aSdan 17614fccf43aSdan /* 17627531a5a3Sdan ** Set a table filter on a Session Object. 17637531a5a3Sdan */ 17647531a5a3Sdan void sqlite3session_table_filter( 17657531a5a3Sdan sqlite3_session *pSession, 17667531a5a3Sdan int(*xFilter)(void*, const char*), 17677531a5a3Sdan void *pCtx /* First argument passed to xFilter */ 17687531a5a3Sdan ){ 17697531a5a3Sdan pSession->bAutoAttach = 1; 17707531a5a3Sdan pSession->pFilterCtx = pCtx; 17717531a5a3Sdan pSession->xTableFilter = xFilter; 17727531a5a3Sdan } 17737531a5a3Sdan 17747531a5a3Sdan /* 17754fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table 17764fccf43aSdan ** while the session object is enabled will be recorded. 17774fccf43aSdan ** 17784fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does 17794fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias) 17804fccf43aSdan ** or not. 17814fccf43aSdan */ 17824fccf43aSdan int sqlite3session_attach( 17834fccf43aSdan sqlite3_session *pSession, /* Session object */ 17844fccf43aSdan const char *zName /* Table name */ 17854fccf43aSdan ){ 1786ff4d0f41Sdan int rc = SQLITE_OK; 1787ff4d0f41Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1788ff4d0f41Sdan 1789ff4d0f41Sdan if( !zName ){ 1790ff4d0f41Sdan pSession->bAutoAttach = 1; 1791ff4d0f41Sdan }else{ 1792296c7658Sdan SessionTable *pTab; /* New table object (if required) */ 1793296c7658Sdan int nName; /* Number of bytes in string zName */ 17944fccf43aSdan 17954fccf43aSdan /* First search for an existing entry. If one is found, this call is 17964fccf43aSdan ** a no-op. Return early. */ 1797cfdbde21Sdrh nName = sqlite3Strlen30(zName); 17984fccf43aSdan for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){ 17994c220252Sdan if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break; 18004fccf43aSdan } 18014fccf43aSdan 18024c220252Sdan if( !pTab ){ 18034fccf43aSdan /* Allocate new SessionTable object. */ 18042d77d80aSdrh pTab = (SessionTable *)sqlite3_malloc64(sizeof(SessionTable) + nName + 1); 18054c220252Sdan if( !pTab ){ 18064c220252Sdan rc = SQLITE_NOMEM; 18074c220252Sdan }else{ 18086c39e6a8Sdan /* Populate the new SessionTable object and link it into the list. 18096c39e6a8Sdan ** The new object must be linked onto the end of the list, not 18106c39e6a8Sdan ** simply added to the start of it in order to ensure that tables 18116c39e6a8Sdan ** appear in the correct order when a changeset or patchset is 18126c39e6a8Sdan ** eventually generated. */ 18136c39e6a8Sdan SessionTable **ppTab; 18144fccf43aSdan memset(pTab, 0, sizeof(SessionTable)); 18154fccf43aSdan pTab->zName = (char *)&pTab[1]; 18164fccf43aSdan memcpy(pTab->zName, zName, nName+1); 18176c39e6a8Sdan for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext); 18186c39e6a8Sdan *ppTab = pTab; 18194c220252Sdan } 18204c220252Sdan } 1821ff4d0f41Sdan } 18224fccf43aSdan 18234c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 18244c220252Sdan return rc; 18254fccf43aSdan } 18264fccf43aSdan 1827296c7658Sdan /* 1828296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data. 1829296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is. 1830296c7658Sdan ** 1831296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered, 1832296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero. 1833296c7658Sdan */ 1834f6ad201aSdrh static int sessionBufferGrow(SessionBuffer *p, size_t nByte, int *pRc){ 18353196abb7Smistachkin if( *pRc==SQLITE_OK && (size_t)(p->nAlloc-p->nBuf)<nByte ){ 18364fccf43aSdan u8 *aNew; 18379c18ef09Sdan i64 nNew = p->nAlloc ? p->nAlloc : 128; 18384fccf43aSdan do { 18394fccf43aSdan nNew = nNew*2; 1840ac713403Smistachkin }while( (size_t)(nNew-p->nBuf)<nByte ); 18414fccf43aSdan 18429c18ef09Sdan aNew = (u8 *)sqlite3_realloc64(p->aBuf, nNew); 18434fccf43aSdan if( 0==aNew ){ 18444fccf43aSdan *pRc = SQLITE_NOMEM; 184580fe2d93Sdan }else{ 18464fccf43aSdan p->aBuf = aNew; 18474fccf43aSdan p->nAlloc = nNew; 18484fccf43aSdan } 184980fe2d93Sdan } 185080fe2d93Sdan return (*pRc!=SQLITE_OK); 18514fccf43aSdan } 18524fccf43aSdan 1853296c7658Sdan /* 1854fa122adaSdan ** Append the value passed as the second argument to the buffer passed 1855fa122adaSdan ** as the first. 1856fa122adaSdan ** 1857fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called. 1858fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code 1859fa122adaSdan ** before returning. 1860fa122adaSdan */ 1861fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){ 1862fa122adaSdan int rc = *pRc; 1863fa122adaSdan if( rc==SQLITE_OK ){ 18642d77d80aSdrh sqlite3_int64 nByte = 0; 1865e8fa8c96Sdan rc = sessionSerializeValue(0, pVal, &nByte); 1866fa122adaSdan sessionBufferGrow(p, nByte, &rc); 1867fa122adaSdan if( rc==SQLITE_OK ){ 1868fa122adaSdan rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0); 1869fa122adaSdan p->nBuf += nByte; 1870fa122adaSdan }else{ 1871fa122adaSdan *pRc = rc; 1872fa122adaSdan } 1873fa122adaSdan } 1874fa122adaSdan } 1875fa122adaSdan 1876fa122adaSdan /* 1877296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1878296c7658Sdan ** called. Otherwise, append a single byte to the buffer. 1879296c7658Sdan ** 1880296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1881296c7658Sdan ** returning. 1882296c7658Sdan */ 18834fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){ 188480fe2d93Sdan if( 0==sessionBufferGrow(p, 1, pRc) ){ 18854fccf43aSdan p->aBuf[p->nBuf++] = v; 18864fccf43aSdan } 18874fccf43aSdan } 18884fccf43aSdan 1889296c7658Sdan /* 1890296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1891296c7658Sdan ** called. Otherwise, append a single varint to the buffer. 1892296c7658Sdan ** 1893296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1894296c7658Sdan ** returning. 1895296c7658Sdan */ 1896cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){ 189780fe2d93Sdan if( 0==sessionBufferGrow(p, 9, pRc) ){ 18984fccf43aSdan p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v); 18994fccf43aSdan } 19004fccf43aSdan } 19014fccf43aSdan 1902296c7658Sdan /* 1903296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1904296c7658Sdan ** called. Otherwise, append a blob of data to the buffer. 1905296c7658Sdan ** 1906296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1907296c7658Sdan ** returning. 1908296c7658Sdan */ 19094fccf43aSdan static void sessionAppendBlob( 19104fccf43aSdan SessionBuffer *p, 19114fccf43aSdan const u8 *aBlob, 19124fccf43aSdan int nBlob, 19134fccf43aSdan int *pRc 19144fccf43aSdan ){ 1915895decf6Sdan if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){ 19164fccf43aSdan memcpy(&p->aBuf[p->nBuf], aBlob, nBlob); 19174fccf43aSdan p->nBuf += nBlob; 19184fccf43aSdan } 19194fccf43aSdan } 19204fccf43aSdan 1921296c7658Sdan /* 1922296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1923296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string 1924296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer. 1925296c7658Sdan ** 1926296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1927296c7658Sdan ** returning. 1928296c7658Sdan */ 1929d5f0767cSdan static void sessionAppendStr( 1930d5f0767cSdan SessionBuffer *p, 1931d5f0767cSdan const char *zStr, 1932d5f0767cSdan int *pRc 1933d5f0767cSdan ){ 1934cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr); 193580fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1936d5f0767cSdan memcpy(&p->aBuf[p->nBuf], zStr, nStr); 1937d5f0767cSdan p->nBuf += nStr; 1938d5f0767cSdan } 1939d5f0767cSdan } 1940d5f0767cSdan 1941296c7658Sdan /* 1942296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1943296c7658Sdan ** called. Otherwise, append the string representation of integer iVal 1944296c7658Sdan ** to the buffer. No nul-terminator is written. 1945296c7658Sdan ** 1946296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1947296c7658Sdan ** returning. 1948296c7658Sdan */ 1949d5f0767cSdan static void sessionAppendInteger( 1950296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1951296c7658Sdan int iVal, /* Value to write the string rep. of */ 1952296c7658Sdan int *pRc /* IN/OUT: Error code */ 1953d5f0767cSdan ){ 1954d5f0767cSdan char aBuf[24]; 1955d5f0767cSdan sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal); 1956d5f0767cSdan sessionAppendStr(p, aBuf, pRc); 1957d5f0767cSdan } 1958d5f0767cSdan 1959296c7658Sdan /* 1960296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1961296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and 1962296c7658Sdan ** with any embedded quote characters escaped to the buffer. No 1963296c7658Sdan ** nul-terminator byte is written. 1964296c7658Sdan ** 1965296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1966296c7658Sdan ** returning. 1967296c7658Sdan */ 1968d5f0767cSdan static void sessionAppendIdent( 1969296c7658Sdan SessionBuffer *p, /* Buffer to a append to */ 1970296c7658Sdan const char *zStr, /* String to quote, escape and append */ 1971296c7658Sdan int *pRc /* IN/OUT: Error code */ 1972d5f0767cSdan ){ 1973cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1; 197480fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1975d5f0767cSdan char *zOut = (char *)&p->aBuf[p->nBuf]; 1976d5f0767cSdan const char *zIn = zStr; 1977d5f0767cSdan *zOut++ = '"'; 1978d5f0767cSdan while( *zIn ){ 1979d5f0767cSdan if( *zIn=='"' ) *zOut++ = '"'; 1980d5f0767cSdan *zOut++ = *(zIn++); 1981d5f0767cSdan } 1982d5f0767cSdan *zOut++ = '"'; 1983cfdbde21Sdrh p->nBuf = (int)((u8 *)zOut - p->aBuf); 1984d5f0767cSdan } 1985d5f0767cSdan } 1986d5f0767cSdan 1987296c7658Sdan /* 1988296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1989296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored 1990296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points 1991296c7658Sdan ** to to the buffer. 1992296c7658Sdan */ 19934fccf43aSdan static void sessionAppendCol( 1994296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1995296c7658Sdan sqlite3_stmt *pStmt, /* Handle pointing to row containing value */ 1996296c7658Sdan int iCol, /* Column to read value from */ 1997296c7658Sdan int *pRc /* IN/OUT: Error code */ 19984fccf43aSdan ){ 19994fccf43aSdan if( *pRc==SQLITE_OK ){ 20004fccf43aSdan int eType = sqlite3_column_type(pStmt, iCol); 20014fccf43aSdan sessionAppendByte(p, (u8)eType, pRc); 20024fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 20034fccf43aSdan sqlite3_int64 i; 20044fccf43aSdan u8 aBuf[8]; 20054fccf43aSdan if( eType==SQLITE_INTEGER ){ 20064fccf43aSdan i = sqlite3_column_int64(pStmt, iCol); 20074fccf43aSdan }else{ 20084fccf43aSdan double r = sqlite3_column_double(pStmt, iCol); 20094fccf43aSdan memcpy(&i, &r, 8); 20104fccf43aSdan } 2011296c7658Sdan sessionPutI64(aBuf, i); 20124fccf43aSdan sessionAppendBlob(p, aBuf, 8, pRc); 20134fccf43aSdan } 20144fccf43aSdan if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){ 20156734007dSdan u8 *z; 20163cc89d95Sdan int nByte; 20176734007dSdan if( eType==SQLITE_BLOB ){ 20186734007dSdan z = (u8 *)sqlite3_column_blob(pStmt, iCol); 20196734007dSdan }else{ 20206734007dSdan z = (u8 *)sqlite3_column_text(pStmt, iCol); 20216734007dSdan } 20223cc89d95Sdan nByte = sqlite3_column_bytes(pStmt, iCol); 20233cc89d95Sdan if( z || (eType==SQLITE_BLOB && nByte==0) ){ 20244fccf43aSdan sessionAppendVarint(p, nByte, pRc); 20256734007dSdan sessionAppendBlob(p, z, nByte, pRc); 20266734007dSdan }else{ 20276734007dSdan *pRc = SQLITE_NOMEM; 20286734007dSdan } 20294fccf43aSdan } 20304fccf43aSdan } 20314fccf43aSdan } 20324fccf43aSdan 2033296c7658Sdan /* 2034296c7658Sdan ** 203580fe2d93Sdan ** This function appends an update change to the buffer (see the comments 203680fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change 203780fe2d93Sdan ** consists of: 2038296c7658Sdan ** 2039296c7658Sdan ** 1 byte: SQLITE_UPDATE (0x17) 2040296c7658Sdan ** n bytes: old.* record (see RECORD FORMAT) 2041296c7658Sdan ** m bytes: new.* record (see RECORD FORMAT) 2042296c7658Sdan ** 2043296c7658Sdan ** The SessionChange object passed as the third argument contains the 2044296c7658Sdan ** values that were stored in the row when the session began (the old.* 2045296c7658Sdan ** values). The statement handle passed as the second argument points 2046296c7658Sdan ** at the current version of the row (the new.* values). 2047296c7658Sdan ** 2048296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value 2049296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer. 2050296c7658Sdan ** 2051296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the 2052296c7658Sdan ** original values of any fields that have been modified. The new.* record 2053296c7658Sdan ** contains the new values of only those fields that have been modified. 2054296c7658Sdan */ 205580fe2d93Sdan static int sessionAppendUpdate( 2056296c7658Sdan SessionBuffer *pBuf, /* Buffer to append to */ 205773b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 2058296c7658Sdan sqlite3_stmt *pStmt, /* Statement handle pointing at new row */ 2059296c7658Sdan SessionChange *p, /* Object containing old values */ 206080fe2d93Sdan u8 *abPK /* Boolean array - true for PK columns */ 20614fccf43aSdan ){ 206280fe2d93Sdan int rc = SQLITE_OK; 2063296c7658Sdan SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */ 2064296c7658Sdan int bNoop = 1; /* Set to zero if any values are modified */ 20651f34f8ccSdan int nRewind = pBuf->nBuf; /* Set to zero if any values are modified */ 2066296c7658Sdan int i; /* Used to iterate through columns */ 2067296c7658Sdan u8 *pCsr = p->aRecord; /* Used to iterate through old.* values */ 2068296c7658Sdan 206980fe2d93Sdan sessionAppendByte(pBuf, SQLITE_UPDATE, &rc); 207080fe2d93Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 20714fccf43aSdan for(i=0; i<sqlite3_column_count(pStmt); i++){ 207237f133ecSdan int bChanged = 0; 20734fccf43aSdan int nAdvance; 20744fccf43aSdan int eType = *pCsr; 20754fccf43aSdan switch( eType ){ 20764fccf43aSdan case SQLITE_NULL: 20774fccf43aSdan nAdvance = 1; 20784fccf43aSdan if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){ 207937f133ecSdan bChanged = 1; 20804fccf43aSdan } 20814fccf43aSdan break; 20824fccf43aSdan 20834fccf43aSdan case SQLITE_FLOAT: 20844fccf43aSdan case SQLITE_INTEGER: { 20854fccf43aSdan nAdvance = 9; 20864fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) ){ 20874fccf43aSdan sqlite3_int64 iVal = sessionGetI64(&pCsr[1]); 20884fccf43aSdan if( eType==SQLITE_INTEGER ){ 20894fccf43aSdan if( iVal==sqlite3_column_int64(pStmt, i) ) break; 20904fccf43aSdan }else{ 20914fccf43aSdan double dVal; 20924fccf43aSdan memcpy(&dVal, &iVal, 8); 20934fccf43aSdan if( dVal==sqlite3_column_double(pStmt, i) ) break; 20944fccf43aSdan } 20954fccf43aSdan } 209637f133ecSdan bChanged = 1; 20974fccf43aSdan break; 20984fccf43aSdan } 20994fccf43aSdan 2100e5754eecSdan default: { 2101895decf6Sdan int n; 2102895decf6Sdan int nHdr = 1 + sessionVarintGet(&pCsr[1], &n); 2103e5754eecSdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 2104895decf6Sdan nAdvance = nHdr + n; 21054fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) 2106895decf6Sdan && n==sqlite3_column_bytes(pStmt, i) 2107895decf6Sdan && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n)) 21084fccf43aSdan ){ 21094fccf43aSdan break; 21104fccf43aSdan } 211137f133ecSdan bChanged = 1; 21124fccf43aSdan } 21134fccf43aSdan } 21144fccf43aSdan 211573b3c055Sdan /* If at least one field has been modified, this is not a no-op. */ 211673b3c055Sdan if( bChanged ) bNoop = 0; 211773b3c055Sdan 211873b3c055Sdan /* Add a field to the old.* record. This is omitted if this modules is 211973b3c055Sdan ** currently generating a patchset. */ 212073b3c055Sdan if( bPatchset==0 ){ 212137f133ecSdan if( bChanged || abPK[i] ){ 212280fe2d93Sdan sessionAppendBlob(pBuf, pCsr, nAdvance, &rc); 21234fccf43aSdan }else{ 212480fe2d93Sdan sessionAppendByte(pBuf, 0, &rc); 212537f133ecSdan } 212673b3c055Sdan } 212737f133ecSdan 212873b3c055Sdan /* Add a field to the new.* record. Or the only record if currently 212973b3c055Sdan ** generating a patchset. */ 213073b3c055Sdan if( bChanged || (bPatchset && abPK[i]) ){ 213180fe2d93Sdan sessionAppendCol(&buf2, pStmt, i, &rc); 213237f133ecSdan }else{ 213380fe2d93Sdan sessionAppendByte(&buf2, 0, &rc); 21344fccf43aSdan } 213537f133ecSdan 21364fccf43aSdan pCsr += nAdvance; 21374fccf43aSdan } 21384fccf43aSdan 21394fccf43aSdan if( bNoop ){ 21401f34f8ccSdan pBuf->nBuf = nRewind; 21414fccf43aSdan }else{ 214280fe2d93Sdan sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc); 21434fccf43aSdan } 21441f34f8ccSdan sqlite3_free(buf2.aBuf); 214580fe2d93Sdan 214680fe2d93Sdan return rc; 2147d5f0767cSdan } 21484fccf43aSdan 2149a71d2371Sdan /* 2150a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use 2151a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset 2152a71d2371Sdan ** format otherwise. 2153a71d2371Sdan */ 215473b3c055Sdan static int sessionAppendDelete( 215573b3c055Sdan SessionBuffer *pBuf, /* Buffer to append to */ 215673b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 215773b3c055Sdan SessionChange *p, /* Object containing old values */ 2158a71d2371Sdan int nCol, /* Number of columns in table */ 215973b3c055Sdan u8 *abPK /* Boolean array - true for PK columns */ 216073b3c055Sdan ){ 216173b3c055Sdan int rc = SQLITE_OK; 216273b3c055Sdan 216373b3c055Sdan sessionAppendByte(pBuf, SQLITE_DELETE, &rc); 216473b3c055Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 216573b3c055Sdan 216673b3c055Sdan if( bPatchset==0 ){ 216773b3c055Sdan sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc); 216873b3c055Sdan }else{ 216973b3c055Sdan int i; 217073b3c055Sdan u8 *a = p->aRecord; 217173b3c055Sdan for(i=0; i<nCol; i++){ 217273b3c055Sdan u8 *pStart = a; 217373b3c055Sdan int eType = *a++; 217473b3c055Sdan 217573b3c055Sdan switch( eType ){ 217673b3c055Sdan case 0: 217773b3c055Sdan case SQLITE_NULL: 217873b3c055Sdan assert( abPK[i]==0 ); 217973b3c055Sdan break; 218073b3c055Sdan 218173b3c055Sdan case SQLITE_FLOAT: 218273b3c055Sdan case SQLITE_INTEGER: 218373b3c055Sdan a += 8; 218473b3c055Sdan break; 218573b3c055Sdan 218673b3c055Sdan default: { 218773b3c055Sdan int n; 218873b3c055Sdan a += sessionVarintGet(a, &n); 218973b3c055Sdan a += n; 219073b3c055Sdan break; 219173b3c055Sdan } 219273b3c055Sdan } 219373b3c055Sdan if( abPK[i] ){ 2194f5ab08c7Sdrh sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc); 219573b3c055Sdan } 219673b3c055Sdan } 219773b3c055Sdan assert( (a - p->aRecord)==p->nRecord ); 219873b3c055Sdan } 219973b3c055Sdan 220073b3c055Sdan return rc; 220173b3c055Sdan } 220273b3c055Sdan 220377fc1d5bSdan /* 220477fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table 220577fc1d5bSdan ** zTab in database zDb based on its primary key. i.e. 220677fc1d5bSdan ** 220777fc1d5bSdan ** SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ... 220877fc1d5bSdan */ 2209e8d5648eSdan static int sessionSelectStmt( 2210e8d5648eSdan sqlite3 *db, /* Database handle */ 2211d7fb7d24Sdan const char *zDb, /* Database name */ 2212e8d5648eSdan const char *zTab, /* Table name */ 221377fc1d5bSdan int nCol, /* Number of columns in table */ 221477fc1d5bSdan const char **azCol, /* Names of table columns */ 221577fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 221677fc1d5bSdan sqlite3_stmt **ppStmt /* OUT: Prepared SELECT statement */ 2217d5f0767cSdan ){ 2218e8d5648eSdan int rc = SQLITE_OK; 22193739f298Sdan char *zSql = 0; 22203739f298Sdan int nSql = -1; 22213739f298Sdan 22223739f298Sdan if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){ 22233739f298Sdan zSql = sqlite3_mprintf( 22243739f298Sdan "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND " 22253739f298Sdan "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb 22263739f298Sdan ); 2227bd45374cSdan if( zSql==0 ) rc = SQLITE_NOMEM; 22283739f298Sdan }else{ 2229d5f0767cSdan int i; 2230e8d5648eSdan const char *zSep = ""; 2231e8d5648eSdan SessionBuffer buf = {0, 0, 0}; 2232d5f0767cSdan 2233e8d5648eSdan sessionAppendStr(&buf, "SELECT * FROM ", &rc); 2234d7fb7d24Sdan sessionAppendIdent(&buf, zDb, &rc); 2235d7fb7d24Sdan sessionAppendStr(&buf, ".", &rc); 2236e8d5648eSdan sessionAppendIdent(&buf, zTab, &rc); 2237e8d5648eSdan sessionAppendStr(&buf, " WHERE ", &rc); 2238e8d5648eSdan for(i=0; i<nCol; i++){ 2239e8d5648eSdan if( abPK[i] ){ 2240e8d5648eSdan sessionAppendStr(&buf, zSep, &rc); 2241e8d5648eSdan sessionAppendIdent(&buf, azCol[i], &rc); 22423739f298Sdan sessionAppendStr(&buf, " IS ?", &rc); 2243e8d5648eSdan sessionAppendInteger(&buf, i+1, &rc); 2244e8d5648eSdan zSep = " AND "; 2245d5f0767cSdan } 2246d5f0767cSdan } 22473739f298Sdan zSql = (char*)buf.aBuf; 22483739f298Sdan nSql = buf.nBuf; 2249d5f0767cSdan } 22503739f298Sdan 22513739f298Sdan if( rc==SQLITE_OK ){ 22523739f298Sdan rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0); 22533739f298Sdan } 22543739f298Sdan sqlite3_free(zSql); 2255e8d5648eSdan return rc; 2256d5f0767cSdan } 2257d5f0767cSdan 225877fc1d5bSdan /* 225977fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange 226077fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement 226177fc1d5bSdan ** is as prepared by function sessionSelectStmt(). 226277fc1d5bSdan ** 226377fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite 226477fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise. 226577fc1d5bSdan */ 2266e8d5648eSdan static int sessionSelectBind( 226777fc1d5bSdan sqlite3_stmt *pSelect, /* SELECT from sessionSelectStmt() */ 226877fc1d5bSdan int nCol, /* Number of columns in table */ 226977fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 227077fc1d5bSdan SessionChange *pChange /* Change structure */ 2271e8d5648eSdan ){ 2272e8d5648eSdan int i; 2273e8d5648eSdan int rc = SQLITE_OK; 2274e5754eecSdan u8 *a = pChange->aRecord; 2275d5f0767cSdan 2276e8d5648eSdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2277e8d5648eSdan int eType = *a++; 2278e8d5648eSdan 2279e8d5648eSdan switch( eType ){ 228080fe2d93Sdan case 0: 2281e8d5648eSdan case SQLITE_NULL: 22821611e5a3Sdan assert( abPK[i]==0 ); 2283e8d5648eSdan break; 2284e8d5648eSdan 2285e8d5648eSdan case SQLITE_INTEGER: { 2286e8d5648eSdan if( abPK[i] ){ 2287e8d5648eSdan i64 iVal = sessionGetI64(a); 2288e8d5648eSdan rc = sqlite3_bind_int64(pSelect, i+1, iVal); 2289e8d5648eSdan } 2290e8d5648eSdan a += 8; 2291e8d5648eSdan break; 2292d5f0767cSdan } 2293296c7658Sdan 2294e8d5648eSdan case SQLITE_FLOAT: { 2295e8d5648eSdan if( abPK[i] ){ 2296e8d5648eSdan double rVal; 2297e8d5648eSdan i64 iVal = sessionGetI64(a); 2298e8d5648eSdan memcpy(&rVal, &iVal, 8); 22994e895da1Sdan rc = sqlite3_bind_double(pSelect, i+1, rVal); 2300d5f0767cSdan } 2301e8d5648eSdan a += 8; 2302e8d5648eSdan break; 2303e8d5648eSdan } 2304e8d5648eSdan 2305e8d5648eSdan case SQLITE_TEXT: { 2306e8d5648eSdan int n; 2307e8d5648eSdan a += sessionVarintGet(a, &n); 2308e8d5648eSdan if( abPK[i] ){ 2309e8d5648eSdan rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT); 2310e8d5648eSdan } 2311e8d5648eSdan a += n; 2312e8d5648eSdan break; 2313e8d5648eSdan } 2314e8d5648eSdan 2315e5754eecSdan default: { 2316e8d5648eSdan int n; 2317e5754eecSdan assert( eType==SQLITE_BLOB ); 2318e8d5648eSdan a += sessionVarintGet(a, &n); 2319e8d5648eSdan if( abPK[i] ){ 2320e8d5648eSdan rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT); 2321e8d5648eSdan } 2322e8d5648eSdan a += n; 2323e8d5648eSdan break; 2324e8d5648eSdan } 2325e8d5648eSdan } 2326e8d5648eSdan } 2327e8d5648eSdan 2328d5f0767cSdan return rc; 23294fccf43aSdan } 23304fccf43aSdan 233177fc1d5bSdan /* 233277fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it 233377fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary 233477fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an 233577fc1d5bSdan ** SQLite error code before returning. 233677fc1d5bSdan */ 23375d607a6eSdan static void sessionAppendTableHdr( 2338a71d2371Sdan SessionBuffer *pBuf, /* Append header to this buffer */ 2339a71d2371Sdan int bPatchset, /* Use the patchset format if true */ 2340a71d2371Sdan SessionTable *pTab, /* Table object to append header for */ 2341a71d2371Sdan int *pRc /* IN/OUT: Error code */ 23425d607a6eSdan ){ 23435d607a6eSdan /* Write a table header */ 234473b3c055Sdan sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc); 23455d607a6eSdan sessionAppendVarint(pBuf, pTab->nCol, pRc); 23465d607a6eSdan sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc); 23474f528042Sdan sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc); 23485d607a6eSdan } 23495d607a6eSdan 2350a71d2371Sdan /* 2351a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset 2352a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object 2353a71d2371Sdan ** passed as the first argument. 2354a71d2371Sdan ** 2355a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset 2356a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error 2357a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set 2358a71d2371Sdan ** to 0. 2359a71d2371Sdan */ 2360adf3bf58Sdrh static int sessionGenerateChangeset( 23614fccf43aSdan sqlite3_session *pSession, /* Session object */ 236273b3c055Sdan int bPatchset, /* True for patchset, false for changeset */ 2363ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2364ef7a6304Sdan void *pOut, /* First argument for xOutput */ 23654fccf43aSdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 23664fccf43aSdan void **ppChangeset /* OUT: Buffer containing changeset */ 23674fccf43aSdan ){ 2368296c7658Sdan sqlite3 *db = pSession->db; /* Source database handle */ 2369296c7658Sdan SessionTable *pTab; /* Used to iterate through attached tables */ 2370296c7658Sdan SessionBuffer buf = {0,0,0}; /* Buffer in which to accumlate changeset */ 2371296c7658Sdan int rc; /* Return code */ 23724fccf43aSdan 2373ef7a6304Sdan assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) ); 2374ef7a6304Sdan 2375296c7658Sdan /* Zero the output variables in case an error occurs. If this session 2376296c7658Sdan ** object is already in the error state (sqlite3_session.rc != SQLITE_OK), 2377296c7658Sdan ** this call will be a no-op. */ 2378ef7a6304Sdan if( xOutput==0 ){ 23794fccf43aSdan *pnChangeset = 0; 23804fccf43aSdan *ppChangeset = 0; 2381ef7a6304Sdan } 2382e5754eecSdan 2383e5754eecSdan if( pSession->rc ) return pSession->rc; 2384e5754eecSdan rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0); 2385e5754eecSdan if( rc!=SQLITE_OK ) return rc; 2386e5754eecSdan 2387e5754eecSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 23884fccf43aSdan 23894fccf43aSdan for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 23904fccf43aSdan if( pTab->nEntry ){ 2391d7fb7d24Sdan const char *zName = pTab->zName; 2392a9605b91Sdan int nCol; /* Number of columns in table */ 2393a9605b91Sdan u8 *abPK; /* Primary key array */ 2394a9605b91Sdan const char **azCol = 0; /* Table columns */ 23951f34f8ccSdan int i; /* Used to iterate through hash buckets */ 23961f34f8ccSdan sqlite3_stmt *pSel = 0; /* SELECT statement to query table pTab */ 23971f34f8ccSdan int nRewind = buf.nBuf; /* Initial size of write buffer */ 23981f34f8ccSdan int nNoop; /* Size of buffer after writing tbl header */ 23994fccf43aSdan 2400a9605b91Sdan /* Check the table schema is still Ok. */ 2401a9605b91Sdan rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK); 2402a9605b91Sdan if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){ 2403a9605b91Sdan rc = SQLITE_SCHEMA; 2404a9605b91Sdan } 2405a9605b91Sdan 24064fccf43aSdan /* Write a table header */ 240773b3c055Sdan sessionAppendTableHdr(&buf, bPatchset, pTab, &rc); 24084fccf43aSdan 24094fccf43aSdan /* Build and compile a statement to execute: */ 24104fccf43aSdan if( rc==SQLITE_OK ){ 2411d7fb7d24Sdan rc = sessionSelectStmt( 2412a9605b91Sdan db, pSession->zDb, zName, nCol, azCol, abPK, &pSel); 24134fccf43aSdan } 24144fccf43aSdan 24151f34f8ccSdan nNoop = buf.nBuf; 241612ca0b56Sdan for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){ 2417e8d5648eSdan SessionChange *p; /* Used to iterate through changes */ 2418e8d5648eSdan 24194fccf43aSdan for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){ 2420e5754eecSdan rc = sessionSelectBind(pSel, nCol, abPK, p); 242180fe2d93Sdan if( rc!=SQLITE_OK ) continue; 24221f34f8ccSdan if( sqlite3_step(pSel)==SQLITE_ROW ){ 2423798693b2Sdan if( p->op==SQLITE_INSERT ){ 24244fccf43aSdan int iCol; 24254fccf43aSdan sessionAppendByte(&buf, SQLITE_INSERT, &rc); 2426b4480e94Sdan sessionAppendByte(&buf, p->bIndirect, &rc); 2427e8d5648eSdan for(iCol=0; iCol<nCol; iCol++){ 24281f34f8ccSdan sessionAppendCol(&buf, pSel, iCol, &rc); 24294fccf43aSdan } 2430e8d5648eSdan }else{ 243173b3c055Sdan rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK); 24324fccf43aSdan } 2433798693b2Sdan }else if( p->op!=SQLITE_INSERT ){ 2434a71d2371Sdan rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK); 24354fccf43aSdan } 243612ca0b56Sdan if( rc==SQLITE_OK ){ 24371f34f8ccSdan rc = sqlite3_reset(pSel); 24384fccf43aSdan } 2439ef7a6304Sdan 24401f48e67dSdan /* If the buffer is now larger than sessions_strm_chunk_size, pass 2441ef7a6304Sdan ** its contents to the xOutput() callback. */ 2442ef7a6304Sdan if( xOutput 2443ef7a6304Sdan && rc==SQLITE_OK 2444ef7a6304Sdan && buf.nBuf>nNoop 24451f48e67dSdan && buf.nBuf>sessions_strm_chunk_size 2446ef7a6304Sdan ){ 2447ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2448ef7a6304Sdan nNoop = -1; 2449ef7a6304Sdan buf.nBuf = 0; 2450ef7a6304Sdan } 2451ef7a6304Sdan 24524fccf43aSdan } 2453e8d5648eSdan } 24544fccf43aSdan 24551f34f8ccSdan sqlite3_finalize(pSel); 24561f34f8ccSdan if( buf.nBuf==nNoop ){ 24574fccf43aSdan buf.nBuf = nRewind; 24584fccf43aSdan } 2459cfdbde21Sdrh sqlite3_free((char*)azCol); /* cast works around VC++ bug */ 24604fccf43aSdan } 24614fccf43aSdan } 24624fccf43aSdan 24634fccf43aSdan if( rc==SQLITE_OK ){ 2464ef7a6304Sdan if( xOutput==0 ){ 24654fccf43aSdan *pnChangeset = buf.nBuf; 24664fccf43aSdan *ppChangeset = buf.aBuf; 2467ef7a6304Sdan buf.aBuf = 0; 2468ef7a6304Sdan }else if( buf.nBuf>0 ){ 2469ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 2470ef7a6304Sdan } 24714fccf43aSdan } 24724c220252Sdan 2473ef7a6304Sdan sqlite3_free(buf.aBuf); 2474e5754eecSdan sqlite3_exec(db, "RELEASE changeset", 0, 0, 0); 24754c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 24764fccf43aSdan return rc; 24774fccf43aSdan } 24784fccf43aSdan 2479296c7658Sdan /* 248073b3c055Sdan ** Obtain a changeset object containing all changes recorded by the 248173b3c055Sdan ** session object passed as the first argument. 248273b3c055Sdan ** 248373b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 248473b3c055Sdan ** using sqlite3_free(). 248573b3c055Sdan */ 248673b3c055Sdan int sqlite3session_changeset( 248773b3c055Sdan sqlite3_session *pSession, /* Session object */ 248873b3c055Sdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 248973b3c055Sdan void **ppChangeset /* OUT: Buffer containing changeset */ 249073b3c055Sdan ){ 2491ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset); 2492ef7a6304Sdan } 2493ef7a6304Sdan 2494ef7a6304Sdan /* 2495ef7a6304Sdan ** Streaming version of sqlite3session_changeset(). 2496ef7a6304Sdan */ 2497f1a08ad8Sdrh int sqlite3session_changeset_strm( 2498ef7a6304Sdan sqlite3_session *pSession, 2499ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2500ef7a6304Sdan void *pOut 2501ef7a6304Sdan ){ 2502ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0); 2503ef7a6304Sdan } 2504ef7a6304Sdan 2505ef7a6304Sdan /* 2506ef7a6304Sdan ** Streaming version of sqlite3session_patchset(). 2507ef7a6304Sdan */ 2508f1a08ad8Sdrh int sqlite3session_patchset_strm( 2509ef7a6304Sdan sqlite3_session *pSession, 2510ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2511ef7a6304Sdan void *pOut 2512ef7a6304Sdan ){ 2513ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0); 251473b3c055Sdan } 251573b3c055Sdan 251673b3c055Sdan /* 251773b3c055Sdan ** Obtain a patchset object containing all changes recorded by the 251873b3c055Sdan ** session object passed as the first argument. 251973b3c055Sdan ** 252073b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 252173b3c055Sdan ** using sqlite3_free(). 252273b3c055Sdan */ 252373b3c055Sdan int sqlite3session_patchset( 252473b3c055Sdan sqlite3_session *pSession, /* Session object */ 252573b3c055Sdan int *pnPatchset, /* OUT: Size of buffer at *ppChangeset */ 252673b3c055Sdan void **ppPatchset /* OUT: Buffer containing changeset */ 252773b3c055Sdan ){ 2528ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset); 252973b3c055Sdan } 253073b3c055Sdan 253173b3c055Sdan /* 2532296c7658Sdan ** Enable or disable the session object passed as the first argument. 2533296c7658Sdan */ 25344fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){ 25354c220252Sdan int ret; 25364c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2537296c7658Sdan if( bEnable>=0 ){ 2538296c7658Sdan pSession->bEnable = bEnable; 25394fccf43aSdan } 25404c220252Sdan ret = pSession->bEnable; 25414c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 25424c220252Sdan return ret; 2543296c7658Sdan } 25444fccf43aSdan 25454fccf43aSdan /* 2546b4480e94Sdan ** Enable or disable the session object passed as the first argument. 2547b4480e94Sdan */ 2548b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){ 2549b4480e94Sdan int ret; 2550b4480e94Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2551b4480e94Sdan if( bIndirect>=0 ){ 2552b4480e94Sdan pSession->bIndirect = bIndirect; 2553b4480e94Sdan } 2554b4480e94Sdan ret = pSession->bIndirect; 2555b4480e94Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2556b4480e94Sdan return ret; 2557b4480e94Sdan } 2558b4480e94Sdan 2559b4480e94Sdan /* 2560b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded 2561b69ec348Sdan ** by the session object passed as the only argument. 2562b69ec348Sdan */ 2563b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){ 2564b69ec348Sdan int ret = 0; 2565b69ec348Sdan SessionTable *pTab; 2566b69ec348Sdan 2567b69ec348Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2568b69ec348Sdan for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){ 2569b69ec348Sdan ret = (pTab->nEntry>0); 2570b69ec348Sdan } 2571b69ec348Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2572b69ec348Sdan 2573ff530326Sdan return (ret==0); 2574b69ec348Sdan } 2575b69ec348Sdan 2576b69ec348Sdan /* 2577f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm(). 25784fccf43aSdan */ 2579adf3bf58Sdrh static int sessionChangesetStart( 2580296c7658Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2581ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2582ef7a6304Sdan void *pIn, 2583296c7658Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 258444748f27Sdan void *pChangeset, /* Pointer to buffer containing changeset */ 258544748f27Sdan int bInvert /* True to invert changeset */ 25864fccf43aSdan ){ 25874fccf43aSdan sqlite3_changeset_iter *pRet; /* Iterator to return */ 25884fccf43aSdan int nByte; /* Number of bytes to allocate for iterator */ 25894fccf43aSdan 2590ef7a6304Sdan assert( xInput==0 || (pChangeset==0 && nChangeset==0) ); 2591ef7a6304Sdan 2592296c7658Sdan /* Zero the output variable in case an error occurs. */ 2593296c7658Sdan *pp = 0; 25944fccf43aSdan 2595296c7658Sdan /* Allocate and initialize the iterator structure. */ 25964fccf43aSdan nByte = sizeof(sqlite3_changeset_iter); 25974fccf43aSdan pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte); 25984fccf43aSdan if( !pRet ) return SQLITE_NOMEM; 25994fccf43aSdan memset(pRet, 0, sizeof(sqlite3_changeset_iter)); 26004757c658Sdan pRet->in.aData = (u8 *)pChangeset; 26014757c658Sdan pRet->in.nData = nChangeset; 2602ef7a6304Sdan pRet->in.xInput = xInput; 2603ef7a6304Sdan pRet->in.pIn = pIn; 2604ef7a6304Sdan pRet->in.bEof = (xInput ? 0 : 1); 260544748f27Sdan pRet->bInvert = bInvert; 26064fccf43aSdan 2607296c7658Sdan /* Populate the output variable and return success. */ 2608296c7658Sdan *pp = pRet; 26094fccf43aSdan return SQLITE_OK; 26104fccf43aSdan } 26114fccf43aSdan 2612296c7658Sdan /* 2613ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset. 2614ef7a6304Sdan */ 2615ef7a6304Sdan int sqlite3changeset_start( 2616ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2617ef7a6304Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2618ef7a6304Sdan void *pChangeset /* Pointer to buffer containing changeset */ 2619ef7a6304Sdan ){ 262044748f27Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, 0); 2621ef7a6304Sdan } 262246de0728Sdan int sqlite3changeset_start_v2( 262346de0728Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 262446de0728Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 262546de0728Sdan void *pChangeset, /* Pointer to buffer containing changeset */ 262646de0728Sdan int flags 262746de0728Sdan ){ 262846de0728Sdan int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT); 262946de0728Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, bInvert); 263046de0728Sdan } 2631ef7a6304Sdan 2632ef7a6304Sdan /* 2633ef7a6304Sdan ** Streaming version of sqlite3changeset_start(). 2634ef7a6304Sdan */ 2635f1a08ad8Sdrh int sqlite3changeset_start_strm( 2636ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2637ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2638ef7a6304Sdan void *pIn 2639ef7a6304Sdan ){ 264044748f27Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0, 0); 2641ef7a6304Sdan } 264246de0728Sdan int sqlite3changeset_start_v2_strm( 264346de0728Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 264446de0728Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 264546de0728Sdan void *pIn, 264646de0728Sdan int flags 264746de0728Sdan ){ 264846de0728Sdan int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT); 264946de0728Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0, bInvert); 265046de0728Sdan } 2651ef7a6304Sdan 2652ef7a6304Sdan /* 2653d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming 2654d9151526Sdan ** object and the buffer is full, discard some data to free up space. 2655d9151526Sdan */ 2656d9151526Sdan static void sessionDiscardData(SessionInput *pIn){ 26571f48e67dSdan if( pIn->xInput && pIn->iNext>=sessions_strm_chunk_size ){ 2658d9151526Sdan int nMove = pIn->buf.nBuf - pIn->iNext; 2659d9151526Sdan assert( nMove>=0 ); 2660d9151526Sdan if( nMove>0 ){ 2661d9151526Sdan memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove); 2662d9151526Sdan } 2663d9151526Sdan pIn->buf.nBuf -= pIn->iNext; 2664d9151526Sdan pIn->iNext = 0; 2665d9151526Sdan pIn->nData = pIn->buf.nBuf; 2666d9151526Sdan } 2667d9151526Sdan } 2668d9151526Sdan 2669d9151526Sdan /* 2670ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or, 2671ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available 2672ef7a6304Sdan ** data is in the buffer. 2673ef7a6304Sdan ** 2674ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise. 2675ef7a6304Sdan */ 26764757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){ 2677ef7a6304Sdan int rc = SQLITE_OK; 26784757c658Sdan if( pIn->xInput ){ 26794757c658Sdan while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){ 26801f48e67dSdan int nNew = sessions_strm_chunk_size; 26814757c658Sdan 2682d9151526Sdan if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn); 26834757c658Sdan if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){ 26844757c658Sdan rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew); 26854757c658Sdan if( nNew==0 ){ 26864757c658Sdan pIn->bEof = 1; 26874757c658Sdan }else{ 26884757c658Sdan pIn->buf.nBuf += nNew; 26894757c658Sdan } 26904757c658Sdan } 26914757c658Sdan 26924757c658Sdan pIn->aData = pIn->buf.aBuf; 26934757c658Sdan pIn->nData = pIn->buf.nBuf; 26944757c658Sdan } 2695ef7a6304Sdan } 2696ef7a6304Sdan return rc; 2697ef7a6304Sdan } 2698ef7a6304Sdan 2699ef7a6304Sdan /* 2700ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record 2701ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec 2702ef7a6304Sdan ** until it points to the byte immediately following that record. 2703ef7a6304Sdan */ 2704ef7a6304Sdan static void sessionSkipRecord( 2705ef7a6304Sdan u8 **ppRec, /* IN/OUT: Record pointer */ 2706ef7a6304Sdan int nCol /* Number of values in record */ 2707ef7a6304Sdan ){ 2708ef7a6304Sdan u8 *aRec = *ppRec; 2709ef7a6304Sdan int i; 2710ef7a6304Sdan for(i=0; i<nCol; i++){ 2711ef7a6304Sdan int eType = *aRec++; 2712ef7a6304Sdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2713ef7a6304Sdan int nByte; 2714ef7a6304Sdan aRec += sessionVarintGet((u8*)aRec, &nByte); 2715ef7a6304Sdan aRec += nByte; 2716ef7a6304Sdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2717ef7a6304Sdan aRec += 8; 2718ef7a6304Sdan } 2719ef7a6304Sdan } 2720ef7a6304Sdan 2721ef7a6304Sdan *ppRec = aRec; 2722ef7a6304Sdan } 2723ef7a6304Sdan 2724ef7a6304Sdan /* 27254757c658Sdan ** This function sets the value of the sqlite3_value object passed as the 27264757c658Sdan ** first argument to a copy of the string or blob held in the aData[] 27274757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM 27284757c658Sdan ** error occurs. 27294757c658Sdan */ 27304757c658Sdan static int sessionValueSetStr( 27314757c658Sdan sqlite3_value *pVal, /* Set the value of this object */ 27324757c658Sdan u8 *aData, /* Buffer containing string or blob data */ 27334757c658Sdan int nData, /* Size of buffer aData[] in bytes */ 27344757c658Sdan u8 enc /* String encoding (0 for blobs) */ 27354757c658Sdan ){ 273616228167Sdan /* In theory this code could just pass SQLITE_TRANSIENT as the final 273716228167Sdan ** argument to sqlite3ValueSetStr() and have the copy created 273816228167Sdan ** automatically. But doing so makes it difficult to detect any OOM 273916228167Sdan ** error. Hence the code to create the copy externally. */ 27402d77d80aSdrh u8 *aCopy = sqlite3_malloc64((sqlite3_int64)nData+1); 27414757c658Sdan if( aCopy==0 ) return SQLITE_NOMEM; 27424757c658Sdan memcpy(aCopy, aData, nData); 27434757c658Sdan sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free); 27444757c658Sdan return SQLITE_OK; 27454757c658Sdan } 27464757c658Sdan 27474757c658Sdan /* 2748296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT" 2749296c7658Sdan ** for details. 2750296c7658Sdan ** 2751296c7658Sdan ** When this function is called, *paChange points to the start of the record 2752296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to 2753296c7658Sdan ** one byte after the end of the same record before this function returns. 2754a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or, 2755a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields 2756a71d2371Sdan ** (in other words, it is a patchset DELETE record). 2757296c7658Sdan ** 2758296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller) 2759296c7658Sdan ** is set to point to an sqlite3_value object containing the value read 2760296c7658Sdan ** from the corresponding position in the record. If that value is not 2761296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change 2762296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is 2763296c7658Sdan ** set to NULL. 2764296c7658Sdan ** 2765296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures 2766296c7658Sdan ** using sqlite3_free(). 2767296c7658Sdan ** 2768296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned. 2769296c7658Sdan ** The apOut[] array may have been partially populated in this case. 2770296c7658Sdan */ 27714fccf43aSdan static int sessionReadRecord( 2772ef7a6304Sdan SessionInput *pIn, /* Input data */ 27734fccf43aSdan int nCol, /* Number of values in record */ 277473b3c055Sdan u8 *abPK, /* Array of primary key flags, or NULL */ 27754fccf43aSdan sqlite3_value **apOut /* Write values to this array */ 27764fccf43aSdan ){ 2777296c7658Sdan int i; /* Used to iterate through columns */ 2778ef7a6304Sdan int rc = SQLITE_OK; 27794fccf43aSdan 2780ef7a6304Sdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2781ef7a6304Sdan int eType = 0; /* Type of value (SQLITE_NULL, TEXT etc.) */ 278273b3c055Sdan if( abPK && abPK[i]==0 ) continue; 2783ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2784ef7a6304Sdan if( rc==SQLITE_OK ){ 2785e341ec69Sdan if( pIn->iNext>=pIn->nData ){ 2786e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2787e341ec69Sdan }else{ 27884757c658Sdan eType = pIn->aData[pIn->iNext++]; 2789e8fa8c96Sdan assert( apOut[i]==0 ); 27904fccf43aSdan if( eType ){ 27914fccf43aSdan apOut[i] = sqlite3ValueNew(0); 2792ef7a6304Sdan if( !apOut[i] ) rc = SQLITE_NOMEM; 2793ef7a6304Sdan } 2794dd8a4af8Sdan } 2795e341ec69Sdan } 27964fccf43aSdan 2797ef7a6304Sdan if( rc==SQLITE_OK ){ 27984757c658Sdan u8 *aVal = &pIn->aData[pIn->iNext]; 27994fccf43aSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 28004fccf43aSdan int nByte; 2801ef7a6304Sdan pIn->iNext += sessionVarintGet(aVal, &nByte); 2802ef7a6304Sdan rc = sessionInputBuffer(pIn, nByte); 2803e8fa8c96Sdan if( rc==SQLITE_OK ){ 2804e341ec69Sdan if( nByte<0 || nByte>pIn->nData-pIn->iNext ){ 2805e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2806e341ec69Sdan }else{ 28076734007dSdan u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0); 28084757c658Sdan rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc); 2809ef7a6304Sdan pIn->iNext += nByte; 28104fccf43aSdan } 2811e341ec69Sdan } 2812e341ec69Sdan } 28134fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2814ef7a6304Sdan sqlite3_int64 v = sessionGetI64(aVal); 28154fccf43aSdan if( eType==SQLITE_INTEGER ){ 28164fccf43aSdan sqlite3VdbeMemSetInt64(apOut[i], v); 28174fccf43aSdan }else{ 28184fccf43aSdan double d; 28194e895da1Sdan memcpy(&d, &v, 8); 28204fccf43aSdan sqlite3VdbeMemSetDouble(apOut[i], d); 28214fccf43aSdan } 2822ef7a6304Sdan pIn->iNext += 8; 282391ddd559Sdan } 28244fccf43aSdan } 28254fccf43aSdan } 28264fccf43aSdan 2827ef7a6304Sdan return rc; 2828ef7a6304Sdan } 2829ef7a6304Sdan 2830ef7a6304Sdan /* 2831ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2832ef7a6304Sdan ** Specifically, to the following: 2833ef7a6304Sdan ** 2834ef7a6304Sdan ** + number of columns in table (varint) 2835ef7a6304Sdan ** + array of PK flags (1 byte per column), 2836ef7a6304Sdan ** + table name (nul terminated). 2837ef7a6304Sdan ** 2838ef7a6304Sdan ** This function ensures that all of the above is present in the input 2839ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()). 2840ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code. 2841ef7a6304Sdan ** The input pointer is not moved. 2842ef7a6304Sdan */ 2843ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){ 2844ef7a6304Sdan int rc = SQLITE_OK; 2845ef7a6304Sdan int nCol = 0; 28464757c658Sdan int nRead = 0; 2847ef7a6304Sdan 2848ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2849ef7a6304Sdan if( rc==SQLITE_OK ){ 28504757c658Sdan nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol); 28516344eddaSdan /* The hard upper limit for the number of columns in an SQLite 28526344eddaSdan ** database table is, according to sqliteLimit.h, 32676. So 28536344eddaSdan ** consider any table-header that purports to have more than 65536 28546344eddaSdan ** columns to be corrupt. This is convenient because otherwise, 28556344eddaSdan ** if the (nCol>65536) condition below were omitted, a sufficiently 28566344eddaSdan ** large value for nCol may cause nRead to wrap around and become 28576344eddaSdan ** negative. Leading to a crash. */ 28586344eddaSdan if( nCol<0 || nCol>65536 ){ 2859e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2860e341ec69Sdan }else{ 28614757c658Sdan rc = sessionInputBuffer(pIn, nRead+nCol+100); 28624757c658Sdan nRead += nCol; 2863ef7a6304Sdan } 2864e341ec69Sdan } 28654757c658Sdan 2866ef7a6304Sdan while( rc==SQLITE_OK ){ 28674757c658Sdan while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){ 28684757c658Sdan nRead++; 2869ef7a6304Sdan } 2870e8fa8c96Sdan if( (pIn->iNext + nRead)<pIn->nData ) break; 28714757c658Sdan rc = sessionInputBuffer(pIn, nRead + 100); 28724757c658Sdan } 2873e8fa8c96Sdan *pnByte = nRead+1; 2874ef7a6304Sdan return rc; 2875ef7a6304Sdan } 2876ef7a6304Sdan 2877ef7a6304Sdan /* 2878fa122adaSdan ** The input pointer currently points to the first byte of the first field 2879fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire 288016228167Sdan ** record is buffered. It does not move the input pointer. 288116228167Sdan ** 288216228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of 288316228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The 288416228167Sdan ** final value of *pnByte is undefined in this case. 2885fa122adaSdan */ 2886fa122adaSdan static int sessionChangesetBufferRecord( 288716228167Sdan SessionInput *pIn, /* Input data */ 288816228167Sdan int nCol, /* Number of columns in record */ 288916228167Sdan int *pnByte /* OUT: Size of record in bytes */ 2890fa122adaSdan ){ 2891fa122adaSdan int rc = SQLITE_OK; 2892fa122adaSdan int nByte = 0; 2893fa122adaSdan int i; 2894fa122adaSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 2895fa122adaSdan int eType; 2896fa122adaSdan rc = sessionInputBuffer(pIn, nByte + 10); 2897fa122adaSdan if( rc==SQLITE_OK ){ 2898fa122adaSdan eType = pIn->aData[pIn->iNext + nByte++]; 2899fa122adaSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2900fa122adaSdan int n; 2901fa122adaSdan nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n); 2902fa122adaSdan nByte += n; 2903fa122adaSdan rc = sessionInputBuffer(pIn, nByte); 2904fa122adaSdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2905fa122adaSdan nByte += 8; 2906fa122adaSdan } 2907fa122adaSdan } 2908fa122adaSdan } 2909fa122adaSdan *pnByte = nByte; 2910fa122adaSdan return rc; 2911fa122adaSdan } 2912fa122adaSdan 2913fa122adaSdan /* 2914ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2915ef7a6304Sdan ** Specifically, to the following: 2916ef7a6304Sdan ** 2917ef7a6304Sdan ** + number of columns in table (varint) 2918ef7a6304Sdan ** + array of PK flags (1 byte per column), 2919ef7a6304Sdan ** + table name (nul terminated). 292016228167Sdan ** 292116228167Sdan ** This function decodes the table-header and populates the p->nCol, 292216228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is 292316228167Sdan ** also allocated or resized according to the new value of p->nCol. The 292416228167Sdan ** input pointer is left pointing to the byte following the table header. 292516228167Sdan ** 292616228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code 292716228167Sdan ** is returned and the final values of the various fields enumerated above 292816228167Sdan ** are undefined. 2929ef7a6304Sdan */ 2930ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){ 2931ef7a6304Sdan int rc; 2932ef7a6304Sdan int nCopy; 2933ef7a6304Sdan assert( p->rc==SQLITE_OK ); 2934ef7a6304Sdan 2935ef7a6304Sdan rc = sessionChangesetBufferTblhdr(&p->in, &nCopy); 2936ef7a6304Sdan if( rc==SQLITE_OK ){ 2937ef7a6304Sdan int nByte; 2938ef7a6304Sdan int nVarint; 29394757c658Sdan nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol); 2940dd8a4af8Sdan if( p->nCol>0 ){ 2941ef7a6304Sdan nCopy -= nVarint; 2942ef7a6304Sdan p->in.iNext += nVarint; 2943ef7a6304Sdan nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy; 2944ef7a6304Sdan p->tblhdr.nBuf = 0; 2945ef7a6304Sdan sessionBufferGrow(&p->tblhdr, nByte, &rc); 2946dd8a4af8Sdan }else{ 2947e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 2948dd8a4af8Sdan } 2949ef7a6304Sdan } 2950ef7a6304Sdan 2951ef7a6304Sdan if( rc==SQLITE_OK ){ 2952f6ad201aSdrh size_t iPK = sizeof(sqlite3_value*)*p->nCol*2; 2953ef7a6304Sdan memset(p->tblhdr.aBuf, 0, iPK); 29544757c658Sdan memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy); 2955ef7a6304Sdan p->in.iNext += nCopy; 2956ef7a6304Sdan } 2957ef7a6304Sdan 2958ef7a6304Sdan p->apValue = (sqlite3_value**)p->tblhdr.aBuf; 2959ef7a6304Sdan p->abPK = (u8*)&p->apValue[p->nCol*2]; 2960ef7a6304Sdan p->zTab = (char*)&p->abPK[p->nCol]; 2961ef7a6304Sdan return (p->rc = rc); 29624fccf43aSdan } 29634fccf43aSdan 296477fc1d5bSdan /* 296577fc1d5bSdan ** Advance the changeset iterator to the next change. 296677fc1d5bSdan ** 296777fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public 296877fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the 296977fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values. 297077fc1d5bSdan ** 297177fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change 297277fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in 297377fc1d5bSdan ** the record to *pnRec. 297477fc1d5bSdan ** 297577fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is 297677fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite 297777fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further 297877fc1d5bSdan ** changes in the changeset. 297977fc1d5bSdan */ 29805d607a6eSdan static int sessionChangesetNext( 298177fc1d5bSdan sqlite3_changeset_iter *p, /* Changeset iterator */ 298277fc1d5bSdan u8 **paRec, /* If non-NULL, store record pointer here */ 2983c0a499eaSdan int *pnRec, /* If non-NULL, store size of record here */ 2984c0a499eaSdan int *pbNew /* If non-NULL, true if new table */ 29855d607a6eSdan ){ 29864fccf43aSdan int i; 2987ef7a6304Sdan u8 op; 29884fccf43aSdan 29895d607a6eSdan assert( (paRec==0 && pnRec==0) || (paRec && pnRec) ); 29905d607a6eSdan 2991296c7658Sdan /* If the iterator is in the error-state, return immediately. */ 29924fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 29934fccf43aSdan 29945d607a6eSdan /* Free the current contents of p->apValue[], if any. */ 29954fccf43aSdan if( p->apValue ){ 29964fccf43aSdan for(i=0; i<p->nCol*2; i++){ 29974fccf43aSdan sqlite3ValueFree(p->apValue[i]); 29984fccf43aSdan } 29994fccf43aSdan memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2); 30004fccf43aSdan } 30014fccf43aSdan 3002ef7a6304Sdan /* Make sure the buffer contains at least 10 bytes of input data, or all 3003ef7a6304Sdan ** remaining data if there are less than 10 bytes available. This is 3004ef7a6304Sdan ** sufficient either for the 'T' or 'P' byte and the varint that follows 3005ef7a6304Sdan ** it, or for the two single byte values otherwise. */ 3006ef7a6304Sdan p->rc = sessionInputBuffer(&p->in, 2); 3007ef7a6304Sdan if( p->rc!=SQLITE_OK ) return p->rc; 3008ef7a6304Sdan 30094fccf43aSdan /* If the iterator is already at the end of the changeset, return DONE. */ 30104757c658Sdan if( p->in.iNext>=p->in.nData ){ 30114fccf43aSdan return SQLITE_DONE; 30124fccf43aSdan } 30134fccf43aSdan 3014d9151526Sdan sessionDiscardData(&p->in); 3015d9151526Sdan p->in.iCurrent = p->in.iNext; 3016d9151526Sdan 30174757c658Sdan op = p->in.aData[p->in.iNext++]; 301807d0f15eSdan while( op=='T' || op=='P' ){ 3019c0a499eaSdan if( pbNew ) *pbNew = 1; 3020ef7a6304Sdan p->bPatchset = (op=='P'); 3021ef7a6304Sdan if( sessionChangesetReadTblhdr(p) ) return p->rc; 3022ef7a6304Sdan if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc; 3023d9151526Sdan p->in.iCurrent = p->in.iNext; 302407d0f15eSdan if( p->in.iNext>=p->in.nData ) return SQLITE_DONE; 30254757c658Sdan op = p->in.aData[p->in.iNext++]; 30265d607a6eSdan } 30275d607a6eSdan 302844748f27Sdan if( p->zTab==0 || (p->bPatchset && p->bInvert) ){ 3029dd8a4af8Sdan /* The first record in the changeset is not a table header. Must be a 3030dd8a4af8Sdan ** corrupt changeset. */ 303144748f27Sdan assert( p->in.iNext==1 || p->zTab ); 3032dd8a4af8Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 3033dd8a4af8Sdan } 3034dd8a4af8Sdan 3035ef7a6304Sdan p->op = op; 30364757c658Sdan p->bIndirect = p->in.aData[p->in.iNext++]; 30374fccf43aSdan if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){ 30384757c658Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 30394fccf43aSdan } 30404fccf43aSdan 3041cbf6d2d2Sdan if( paRec ){ 3042cbf6d2d2Sdan int nVal; /* Number of values to buffer */ 3043cbf6d2d2Sdan if( p->bPatchset==0 && op==SQLITE_UPDATE ){ 3044cbf6d2d2Sdan nVal = p->nCol * 2; 3045cbf6d2d2Sdan }else if( p->bPatchset && op==SQLITE_DELETE ){ 3046cbf6d2d2Sdan nVal = 0; 3047cbf6d2d2Sdan for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++; 3048cbf6d2d2Sdan }else{ 3049cbf6d2d2Sdan nVal = p->nCol; 3050cbf6d2d2Sdan } 3051cbf6d2d2Sdan p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec); 3052cbf6d2d2Sdan if( p->rc!=SQLITE_OK ) return p->rc; 3053cbf6d2d2Sdan *paRec = &p->in.aData[p->in.iNext]; 3054cbf6d2d2Sdan p->in.iNext += *pnRec; 3055cbf6d2d2Sdan }else{ 305644748f27Sdan sqlite3_value **apOld = (p->bInvert ? &p->apValue[p->nCol] : p->apValue); 305744748f27Sdan sqlite3_value **apNew = (p->bInvert ? p->apValue : &p->apValue[p->nCol]); 30585d607a6eSdan 30594fccf43aSdan /* If this is an UPDATE or DELETE, read the old.* record. */ 306073b3c055Sdan if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){ 306173b3c055Sdan u8 *abPK = p->bPatchset ? p->abPK : 0; 306244748f27Sdan p->rc = sessionReadRecord(&p->in, p->nCol, abPK, apOld); 30634fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30644fccf43aSdan } 30654fccf43aSdan 30664fccf43aSdan /* If this is an INSERT or UPDATE, read the new.* record. */ 30674fccf43aSdan if( p->op!=SQLITE_DELETE ){ 306844748f27Sdan p->rc = sessionReadRecord(&p->in, p->nCol, 0, apNew); 30694fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 30704fccf43aSdan } 30714fccf43aSdan 307244748f27Sdan if( (p->bPatchset || p->bInvert) && p->op==SQLITE_UPDATE ){ 307373b3c055Sdan /* If this is an UPDATE that is part of a patchset, then all PK and 307473b3c055Sdan ** modified fields are present in the new.* record. The old.* record 307573b3c055Sdan ** is currently completely empty. This block shifts the PK fields from 307673b3c055Sdan ** new.* to old.*, to accommodate the code that reads these arrays. */ 307773b3c055Sdan for(i=0; i<p->nCol; i++){ 307844748f27Sdan assert( p->bPatchset==0 || p->apValue[i]==0 ); 307973b3c055Sdan if( p->abPK[i] ){ 308044748f27Sdan assert( p->apValue[i]==0 ); 308173b3c055Sdan p->apValue[i] = p->apValue[i+p->nCol]; 3082e341ec69Sdan if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT); 308373b3c055Sdan p->apValue[i+p->nCol] = 0; 308473b3c055Sdan } 308573b3c055Sdan } 308644748f27Sdan }else if( p->bInvert ){ 308744748f27Sdan if( p->op==SQLITE_INSERT ) p->op = SQLITE_DELETE; 308844748f27Sdan else if( p->op==SQLITE_DELETE ) p->op = SQLITE_INSERT; 308973b3c055Sdan } 3090cbf6d2d2Sdan } 3091ef7a6304Sdan 30924fccf43aSdan return SQLITE_ROW; 30934fccf43aSdan } 30944fccf43aSdan 30954fccf43aSdan /* 30965d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next 30975d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE 30985d607a6eSdan ** or SQLITE_CORRUPT. 30995d607a6eSdan ** 31005d607a6eSdan ** This function may not be called on iterators passed to a conflict handler 31015d607a6eSdan ** callback by changeset_apply(). 31025d607a6eSdan */ 31035d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){ 3104c0a499eaSdan return sessionChangesetNext(p, 0, 0, 0); 31055d607a6eSdan } 31065d607a6eSdan 31075d607a6eSdan /* 3108244593c8Sdan ** The following function extracts information on the current change 310977fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next() 31104fccf43aSdan ** has returned SQLITE_ROW. 31114fccf43aSdan */ 31124fccf43aSdan int sqlite3changeset_op( 3113296c7658Sdan sqlite3_changeset_iter *pIter, /* Iterator handle */ 31144fccf43aSdan const char **pzTab, /* OUT: Pointer to table name */ 31154fccf43aSdan int *pnCol, /* OUT: Number of columns in table */ 3116b4480e94Sdan int *pOp, /* OUT: SQLITE_INSERT, DELETE or UPDATE */ 3117b4480e94Sdan int *pbIndirect /* OUT: True if change is indirect */ 31184fccf43aSdan ){ 31194fccf43aSdan *pOp = pIter->op; 31204fccf43aSdan *pnCol = pIter->nCol; 31214fccf43aSdan *pzTab = pIter->zTab; 3122b4480e94Sdan if( pbIndirect ) *pbIndirect = pIter->bIndirect; 31234fccf43aSdan return SQLITE_OK; 31244fccf43aSdan } 31254fccf43aSdan 312677fc1d5bSdan /* 312777fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in 312877fc1d5bSdan ** the database table affected by the change that pIter currently points 312977fc1d5bSdan ** to. This function may only be called after changeset_next() returns 313077fc1d5bSdan ** SQLITE_ROW. 313177fc1d5bSdan */ 3132244593c8Sdan int sqlite3changeset_pk( 3133244593c8Sdan sqlite3_changeset_iter *pIter, /* Iterator object */ 3134244593c8Sdan unsigned char **pabPK, /* OUT: Array of boolean - true for PK cols */ 3135244593c8Sdan int *pnCol /* OUT: Number of entries in output array */ 3136244593c8Sdan ){ 3137244593c8Sdan *pabPK = pIter->abPK; 3138244593c8Sdan if( pnCol ) *pnCol = pIter->nCol; 3139244593c8Sdan return SQLITE_OK; 3140244593c8Sdan } 3141244593c8Sdan 3142296c7658Sdan /* 3143296c7658Sdan ** This function may only be called while the iterator is pointing to an 3144296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()). 3145296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3146296c7658Sdan ** 3147296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3148296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not 3149296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3150296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL. 3151296c7658Sdan ** 3152296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3153296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3154296c7658Sdan */ 31554fccf43aSdan int sqlite3changeset_old( 3156296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3157296c7658Sdan int iVal, /* Index of old.* value to retrieve */ 31584fccf43aSdan sqlite3_value **ppValue /* OUT: Old value (or NULL pointer) */ 31594fccf43aSdan ){ 3160d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){ 3161d5f0767cSdan return SQLITE_MISUSE; 3162d5f0767cSdan } 31634fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 31644fccf43aSdan return SQLITE_RANGE; 31654fccf43aSdan } 31664fccf43aSdan *ppValue = pIter->apValue[iVal]; 31674fccf43aSdan return SQLITE_OK; 31684fccf43aSdan } 31694fccf43aSdan 3170296c7658Sdan /* 3171296c7658Sdan ** This function may only be called while the iterator is pointing to an 3172296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()). 3173296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 3174296c7658Sdan ** 3175296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 3176296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not 3177296c7658Sdan ** included in the record (because the change is an UPDATE and the field 3178296c7658Sdan ** was not modified), set *ppValue to NULL. 3179296c7658Sdan ** 3180296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 3181296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 3182296c7658Sdan */ 31834fccf43aSdan int sqlite3changeset_new( 3184296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3185296c7658Sdan int iVal, /* Index of new.* value to retrieve */ 31864fccf43aSdan sqlite3_value **ppValue /* OUT: New value (or NULL pointer) */ 31874fccf43aSdan ){ 3188d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){ 3189d5f0767cSdan return SQLITE_MISUSE; 3190d5f0767cSdan } 31914fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 31924fccf43aSdan return SQLITE_RANGE; 31934fccf43aSdan } 31944fccf43aSdan *ppValue = pIter->apValue[pIter->nCol+iVal]; 31954fccf43aSdan return SQLITE_OK; 31964fccf43aSdan } 31974fccf43aSdan 3198296c7658Sdan /* 31997aa469cdSdan ** The following two macros are used internally. They are similar to the 32007aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that 32017aa469cdSdan ** they omit all error checking and return a pointer to the requested value. 32027aa469cdSdan */ 32037aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)] 32047aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)] 32057aa469cdSdan 32067aa469cdSdan /* 3207296c7658Sdan ** This function may only be called with a changeset iterator that has been 3208296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT 3209296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned. 3210296c7658Sdan ** 3211296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure 3212296c7658Sdan ** containing the iVal'th value of the conflicting record. 3213296c7658Sdan ** 3214296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error 3215296c7658Sdan ** code is returned. Otherwise, SQLITE_OK. 3216296c7658Sdan */ 3217d5f0767cSdan int sqlite3changeset_conflict( 3218296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3219296c7658Sdan int iVal, /* Index of conflict record value to fetch */ 3220d5f0767cSdan sqlite3_value **ppValue /* OUT: Value from conflicting row */ 3221d5f0767cSdan ){ 3222d5f0767cSdan if( !pIter->pConflict ){ 3223d5f0767cSdan return SQLITE_MISUSE; 3224d5f0767cSdan } 3225ff677b20Sdan if( iVal<0 || iVal>=pIter->nCol ){ 3226d5f0767cSdan return SQLITE_RANGE; 3227d5f0767cSdan } 3228d5f0767cSdan *ppValue = sqlite3_column_value(pIter->pConflict, iVal); 3229d5f0767cSdan return SQLITE_OK; 3230d5f0767cSdan } 3231d5f0767cSdan 32324fccf43aSdan /* 3233cb3e4b79Sdan ** This function may only be called with an iterator passed to an 3234cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case 3235cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key 3236cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK. 3237cb3e4b79Sdan ** 3238cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE. 3239cb3e4b79Sdan */ 3240cb3e4b79Sdan int sqlite3changeset_fk_conflicts( 3241cb3e4b79Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3242cb3e4b79Sdan int *pnOut /* OUT: Number of FK violations */ 3243cb3e4b79Sdan ){ 3244cb3e4b79Sdan if( pIter->pConflict || pIter->apValue ){ 3245cb3e4b79Sdan return SQLITE_MISUSE; 3246cb3e4b79Sdan } 3247cb3e4b79Sdan *pnOut = pIter->nCol; 3248cb3e4b79Sdan return SQLITE_OK; 3249cb3e4b79Sdan } 3250cb3e4b79Sdan 3251cb3e4b79Sdan 3252cb3e4b79Sdan /* 32534fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start(). 32544fccf43aSdan ** 32554fccf43aSdan ** This function may not be called on iterators passed to a conflict handler 32564fccf43aSdan ** callback by changeset_apply(). 32574fccf43aSdan */ 32584fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){ 3259cbf6d2d2Sdan int rc = SQLITE_OK; 3260cbf6d2d2Sdan if( p ){ 3261296c7658Sdan int i; /* Used to iterate through p->apValue[] */ 3262cbf6d2d2Sdan rc = p->rc; 326312ca0b56Sdan if( p->apValue ){ 32644fccf43aSdan for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]); 326512ca0b56Sdan } 3266ef7a6304Sdan sqlite3_free(p->tblhdr.aBuf); 32674757c658Sdan sqlite3_free(p->in.buf.aBuf); 32684fccf43aSdan sqlite3_free(p); 3269cbf6d2d2Sdan } 32704fccf43aSdan return rc; 32714fccf43aSdan } 32724fccf43aSdan 3273fa122adaSdan static int sessionChangesetInvert( 3274fa122adaSdan SessionInput *pInput, /* Input changeset */ 3275fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3276fa122adaSdan void *pOut, 327791ddd559Sdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 327891ddd559Sdan void **ppInverted /* OUT: Inverse of pChangeset */ 327991ddd559Sdan ){ 3280cfec7eeeSdan int rc = SQLITE_OK; /* Return value */ 3281fa122adaSdan SessionBuffer sOut; /* Output buffer */ 3282cfec7eeeSdan int nCol = 0; /* Number of cols in current table */ 3283cfec7eeeSdan u8 *abPK = 0; /* PK array for current table */ 3284cfec7eeeSdan sqlite3_value **apVal = 0; /* Space for values for UPDATE inversion */ 3285ef7a6304Sdan SessionBuffer sPK = {0, 0, 0}; /* PK array for current table */ 328691ddd559Sdan 3287fa122adaSdan /* Initialize the output buffer */ 3288fa122adaSdan memset(&sOut, 0, sizeof(SessionBuffer)); 3289fa122adaSdan 329091ddd559Sdan /* Zero the output variables in case an error occurs. */ 3291fa122adaSdan if( ppInverted ){ 329291ddd559Sdan *ppInverted = 0; 329391ddd559Sdan *pnInverted = 0; 3294fa122adaSdan } 329591ddd559Sdan 3296fa122adaSdan while( 1 ){ 3297ef7a6304Sdan u8 eType; 3298fa122adaSdan 3299fa122adaSdan /* Test for EOF. */ 3300fa122adaSdan if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert; 3301fa122adaSdan if( pInput->iNext>=pInput->nData ) break; 3302fa122adaSdan eType = pInput->aData[pInput->iNext]; 3303fa122adaSdan 330491ddd559Sdan switch( eType ){ 330591ddd559Sdan case 'T': { 3306244593c8Sdan /* A 'table' record consists of: 3307244593c8Sdan ** 3308244593c8Sdan ** * A constant 'T' character, 3309244593c8Sdan ** * Number of columns in said table (a varint), 3310ef7a6304Sdan ** * An array of nCol bytes (sPK), 3311244593c8Sdan ** * A nul-terminated table name. 3312244593c8Sdan */ 3313ef7a6304Sdan int nByte; 3314fa122adaSdan int nVar; 3315fa122adaSdan pInput->iNext++; 3316fa122adaSdan if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){ 3317ef7a6304Sdan goto finished_invert; 3318ef7a6304Sdan } 3319fa122adaSdan nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol); 3320ef7a6304Sdan sPK.nBuf = 0; 3321fa122adaSdan sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc); 3322fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3323fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3324ef7a6304Sdan if( rc ) goto finished_invert; 3325fa122adaSdan 3326fa122adaSdan pInput->iNext += nByte; 3327cfec7eeeSdan sqlite3_free(apVal); 3328cfec7eeeSdan apVal = 0; 3329ef7a6304Sdan abPK = sPK.aBuf; 333091ddd559Sdan break; 333191ddd559Sdan } 333291ddd559Sdan 333391ddd559Sdan case SQLITE_INSERT: 333491ddd559Sdan case SQLITE_DELETE: { 333591ddd559Sdan int nByte; 3336fa122adaSdan int bIndirect = pInput->aData[pInput->iNext+1]; 3337fa122adaSdan int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE); 3338fa122adaSdan pInput->iNext += 2; 3339fa122adaSdan assert( rc==SQLITE_OK ); 3340fa122adaSdan rc = sessionChangesetBufferRecord(pInput, nCol, &nByte); 3341fa122adaSdan sessionAppendByte(&sOut, eType2, &rc); 3342fa122adaSdan sessionAppendByte(&sOut, bIndirect, &rc); 3343fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 3344fa122adaSdan pInput->iNext += nByte; 3345fa122adaSdan if( rc ) goto finished_invert; 334691ddd559Sdan break; 334791ddd559Sdan } 334891ddd559Sdan 334991ddd559Sdan case SQLITE_UPDATE: { 3350cfec7eeeSdan int iCol; 335191ddd559Sdan 3352cfec7eeeSdan if( 0==apVal ){ 33532d77d80aSdrh apVal = (sqlite3_value **)sqlite3_malloc64(sizeof(apVal[0])*nCol*2); 3354cfec7eeeSdan if( 0==apVal ){ 3355cfec7eeeSdan rc = SQLITE_NOMEM; 3356cfec7eeeSdan goto finished_invert; 3357cfec7eeeSdan } 3358cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3359cfec7eeeSdan } 336091ddd559Sdan 3361cfec7eeeSdan /* Write the header for the new UPDATE change. Same as the original. */ 3362fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 3363fa122adaSdan sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc); 336491ddd559Sdan 3365ef7a6304Sdan /* Read the old.* and new.* records for the update change. */ 3366fa122adaSdan pInput->iNext += 2; 3367fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]); 3368ef7a6304Sdan if( rc==SQLITE_OK ){ 3369fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]); 3370ef7a6304Sdan } 3371ef7a6304Sdan 3372cfec7eeeSdan /* Write the new old.* record. Consists of the PK columns from the 3373cfec7eeeSdan ** original old.* record, and the other values from the original 3374cfec7eeeSdan ** new.* record. */ 3375e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3376cfec7eeeSdan sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)]; 3377fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3378cfec7eeeSdan } 3379cfec7eeeSdan 3380cfec7eeeSdan /* Write the new new.* record. Consists of a copy of all values 3381cfec7eeeSdan ** from the original old.* record, except for the PK columns, which 3382cfec7eeeSdan ** are set to "undefined". */ 3383e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 3384cfec7eeeSdan sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]); 3385fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 3386cfec7eeeSdan } 3387cfec7eeeSdan 3388cfec7eeeSdan for(iCol=0; iCol<nCol*2; iCol++){ 3389cfec7eeeSdan sqlite3ValueFree(apVal[iCol]); 3390cfec7eeeSdan } 3391cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 3392cfec7eeeSdan if( rc!=SQLITE_OK ){ 3393cfec7eeeSdan goto finished_invert; 3394cfec7eeeSdan } 3395cfec7eeeSdan 339691ddd559Sdan break; 339791ddd559Sdan } 339891ddd559Sdan 339991ddd559Sdan default: 34004757c658Sdan rc = SQLITE_CORRUPT_BKPT; 3401cfec7eeeSdan goto finished_invert; 340291ddd559Sdan } 3403fa122adaSdan 3404fa122adaSdan assert( rc==SQLITE_OK ); 34051f48e67dSdan if( xOutput && sOut.nBuf>=sessions_strm_chunk_size ){ 3406fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3407fa122adaSdan sOut.nBuf = 0; 3408fa122adaSdan if( rc!=SQLITE_OK ) goto finished_invert; 3409fa122adaSdan } 341091ddd559Sdan } 341191ddd559Sdan 3412cfec7eeeSdan assert( rc==SQLITE_OK ); 3413fa122adaSdan if( pnInverted ){ 3414fa122adaSdan *pnInverted = sOut.nBuf; 3415fa122adaSdan *ppInverted = sOut.aBuf; 3416fa122adaSdan sOut.aBuf = 0; 3417fa122adaSdan }else if( sOut.nBuf>0 ){ 3418fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 3419fa122adaSdan } 3420cfec7eeeSdan 3421cfec7eeeSdan finished_invert: 3422fa122adaSdan sqlite3_free(sOut.aBuf); 3423cfec7eeeSdan sqlite3_free(apVal); 3424ef7a6304Sdan sqlite3_free(sPK.aBuf); 3425cfec7eeeSdan return rc; 342691ddd559Sdan } 342791ddd559Sdan 3428fa122adaSdan 3429fa122adaSdan /* 3430fa122adaSdan ** Invert a changeset object. 3431fa122adaSdan */ 3432fa122adaSdan int sqlite3changeset_invert( 3433fa122adaSdan int nChangeset, /* Number of bytes in input */ 3434fa122adaSdan const void *pChangeset, /* Input changeset */ 3435fa122adaSdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 3436fa122adaSdan void **ppInverted /* OUT: Inverse of pChangeset */ 3437fa122adaSdan ){ 3438fa122adaSdan SessionInput sInput; 3439fa122adaSdan 3440fa122adaSdan /* Set up the input stream */ 3441fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3442fa122adaSdan sInput.nData = nChangeset; 3443fa122adaSdan sInput.aData = (u8*)pChangeset; 3444fa122adaSdan 3445fa122adaSdan return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted); 3446fa122adaSdan } 3447fa122adaSdan 3448fa122adaSdan /* 3449fa122adaSdan ** Streaming version of sqlite3changeset_invert(). 3450fa122adaSdan */ 3451f1a08ad8Sdrh int sqlite3changeset_invert_strm( 3452fa122adaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 3453fa122adaSdan void *pIn, 3454fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 3455fa122adaSdan void *pOut 3456fa122adaSdan ){ 3457fa122adaSdan SessionInput sInput; 3458fa122adaSdan int rc; 3459fa122adaSdan 3460fa122adaSdan /* Set up the input stream */ 3461fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 3462fa122adaSdan sInput.xInput = xInput; 3463fa122adaSdan sInput.pIn = pIn; 3464fa122adaSdan 3465fa122adaSdan rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0); 3466fa122adaSdan sqlite3_free(sInput.buf.aBuf); 3467fa122adaSdan return rc; 3468fa122adaSdan } 3469fa122adaSdan 34700c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx; 34710c698471Sdan struct SessionApplyCtx { 34720c698471Sdan sqlite3 *db; 34730c698471Sdan sqlite3_stmt *pDelete; /* DELETE statement */ 3474cfec7eeeSdan sqlite3_stmt *pUpdate; /* UPDATE statement */ 34750c698471Sdan sqlite3_stmt *pInsert; /* INSERT statement */ 34760c698471Sdan sqlite3_stmt *pSelect; /* SELECT statement */ 34770c698471Sdan int nCol; /* Size of azCol[] and abPK[] arrays */ 34780c698471Sdan const char **azCol; /* Array of column names */ 34790c698471Sdan u8 *abPK; /* Boolean array - true if column is in PK */ 3480d1cccf19Sdan int bStat1; /* True if table is sqlite_stat1 */ 3481d9151526Sdan int bDeferConstraints; /* True to defer constraints */ 3482*5d237bfaSdan int bInvertConstraints; /* Invert when iterating constraints buffer */ 3483d9151526Sdan SessionBuffer constraints; /* Deferred constraints are stored here */ 3484a38e6c57Sdan SessionBuffer rebase; /* Rebase information (if any) here */ 3485dbe7d37aSdan u8 bRebaseStarted; /* If table header is already in rebase */ 3486dbe7d37aSdan u8 bRebase; /* True to collect rebase information */ 34870c698471Sdan }; 34880c698471Sdan 3489d5f0767cSdan /* 3490d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table 3491d5f0767cSdan ** structure like this: 3492d5f0767cSdan ** 3493d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3494d5f0767cSdan ** 3495d5f0767cSdan ** The DELETE statement looks like this: 3496d5f0767cSdan ** 3497db04571cSdan ** DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4) 3498d5f0767cSdan ** 3499d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require 3500d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the 3501d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE. 3502296c7658Sdan ** 3503296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left 3504296c7658Sdan ** pointing to the prepared version of the SQL statement. 3505d5f0767cSdan */ 3506d5f0767cSdan static int sessionDeleteRow( 3507d5f0767cSdan sqlite3 *db, /* Database handle */ 3508d5f0767cSdan const char *zTab, /* Table name */ 35090c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3510d5f0767cSdan ){ 3511296c7658Sdan int i; 3512296c7658Sdan const char *zSep = ""; 3513d5f0767cSdan int rc = SQLITE_OK; 3514d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 35157cf7df7dSdan int nPk = 0; 3516d5f0767cSdan 35179e5ecdc1Sdan sessionAppendStr(&buf, "DELETE FROM main.", &rc); 3518d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3519296c7658Sdan sessionAppendStr(&buf, " WHERE ", &rc); 3520296c7658Sdan 3521296c7658Sdan for(i=0; i<p->nCol; i++){ 3522296c7658Sdan if( p->abPK[i] ){ 35237cf7df7dSdan nPk++; 3524296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3525296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3526296c7658Sdan sessionAppendStr(&buf, " = ?", &rc); 3527296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3528296c7658Sdan zSep = " AND "; 3529296c7658Sdan } 3530296c7658Sdan } 3531296c7658Sdan 35327cf7df7dSdan if( nPk<p->nCol ){ 3533296c7658Sdan sessionAppendStr(&buf, " AND (?", &rc); 3534296c7658Sdan sessionAppendInteger(&buf, p->nCol+1, &rc); 3535296c7658Sdan sessionAppendStr(&buf, " OR ", &rc); 3536296c7658Sdan 3537296c7658Sdan zSep = ""; 3538296c7658Sdan for(i=0; i<p->nCol; i++){ 3539296c7658Sdan if( !p->abPK[i] ){ 3540296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 3541296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3542296c7658Sdan sessionAppendStr(&buf, " IS ?", &rc); 3543296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 3544296c7658Sdan zSep = "AND "; 3545296c7658Sdan } 3546296c7658Sdan } 3547296c7658Sdan sessionAppendStr(&buf, ")", &rc); 35487cf7df7dSdan } 3549d5f0767cSdan 3550d5f0767cSdan if( rc==SQLITE_OK ){ 35510c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0); 3552d5f0767cSdan } 3553d5f0767cSdan sqlite3_free(buf.aBuf); 3554d5f0767cSdan 3555d5f0767cSdan return rc; 3556d5f0767cSdan } 3557d5f0767cSdan 3558d5f0767cSdan /* 3559d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db. 3560d5f0767cSdan ** Assuming a table structure like this: 3561d5f0767cSdan ** 3562d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3563d5f0767cSdan ** 3564d5f0767cSdan ** The UPDATE statement looks like this: 3565d5f0767cSdan ** 3566d5f0767cSdan ** UPDATE x SET 3567d5f0767cSdan ** a = CASE WHEN ?2 THEN ?3 ELSE a END, 3568964cbd46Sdan ** b = CASE WHEN ?5 THEN ?6 ELSE b END, 3569964cbd46Sdan ** c = CASE WHEN ?8 THEN ?9 ELSE c END, 3570964cbd46Sdan ** d = CASE WHEN ?11 THEN ?12 ELSE d END 3571d5f0767cSdan ** WHERE a = ?1 AND c = ?7 AND (?13 OR 3572964cbd46Sdan ** (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND 3573d5f0767cSdan ** ) 3574d5f0767cSdan ** 3575d5f0767cSdan ** For each column in the table, there are three variables to bind: 3576d5f0767cSdan ** 3577d5f0767cSdan ** ?(i*3+1) The old.* value of the column, if any. 3578d5f0767cSdan ** ?(i*3+2) A boolean flag indicating that the value is being modified. 3579d5f0767cSdan ** ?(i*3+3) The new.* value of the column, if any. 3580d5f0767cSdan ** 3581d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update 3582d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the 3583d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns 3584d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)". 3585d5f0767cSdan ** 3586296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left 3587296c7658Sdan ** pointing to the prepared version of the SQL statement. 3588d5f0767cSdan */ 3589d5f0767cSdan static int sessionUpdateRow( 3590d5f0767cSdan sqlite3 *db, /* Database handle */ 3591d5f0767cSdan const char *zTab, /* Table name */ 35920c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3593d5f0767cSdan ){ 3594d5f0767cSdan int rc = SQLITE_OK; 3595d5f0767cSdan int i; 3596d5f0767cSdan const char *zSep = ""; 3597d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 3598d5f0767cSdan 3599d5f0767cSdan /* Append "UPDATE tbl SET " */ 36009e5ecdc1Sdan sessionAppendStr(&buf, "UPDATE main.", &rc); 3601d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3602d5f0767cSdan sessionAppendStr(&buf, " SET ", &rc); 3603d5f0767cSdan 3604d5f0767cSdan /* Append the assignments */ 36050c698471Sdan for(i=0; i<p->nCol; i++){ 3606d5f0767cSdan sessionAppendStr(&buf, zSep, &rc); 36070c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3608d5f0767cSdan sessionAppendStr(&buf, " = CASE WHEN ?", &rc); 3609d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3610d5f0767cSdan sessionAppendStr(&buf, " THEN ?", &rc); 3611d5f0767cSdan sessionAppendInteger(&buf, i*3+3, &rc); 3612d5f0767cSdan sessionAppendStr(&buf, " ELSE ", &rc); 36130c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3614d5f0767cSdan sessionAppendStr(&buf, " END", &rc); 3615d5f0767cSdan zSep = ", "; 3616d5f0767cSdan } 3617d5f0767cSdan 3618d5f0767cSdan /* Append the PK part of the WHERE clause */ 3619d5f0767cSdan sessionAppendStr(&buf, " WHERE ", &rc); 36200c698471Sdan for(i=0; i<p->nCol; i++){ 36210c698471Sdan if( p->abPK[i] ){ 36220c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3623d5f0767cSdan sessionAppendStr(&buf, " = ?", &rc); 3624d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3625d5f0767cSdan sessionAppendStr(&buf, " AND ", &rc); 3626d5f0767cSdan } 3627d5f0767cSdan } 3628d5f0767cSdan 3629d5f0767cSdan /* Append the non-PK part of the WHERE clause */ 3630d5f0767cSdan sessionAppendStr(&buf, " (?", &rc); 36310c698471Sdan sessionAppendInteger(&buf, p->nCol*3+1, &rc); 3632d5f0767cSdan sessionAppendStr(&buf, " OR 1", &rc); 36330c698471Sdan for(i=0; i<p->nCol; i++){ 36340c698471Sdan if( !p->abPK[i] ){ 3635d5f0767cSdan sessionAppendStr(&buf, " AND (?", &rc); 3636d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3637d5f0767cSdan sessionAppendStr(&buf, "=0 OR ", &rc); 36380c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3639d5f0767cSdan sessionAppendStr(&buf, " IS ?", &rc); 3640d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3641d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3642d5f0767cSdan } 3643d5f0767cSdan } 3644d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3645d5f0767cSdan 3646d5f0767cSdan if( rc==SQLITE_OK ){ 36470c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0); 3648d5f0767cSdan } 3649d5f0767cSdan sqlite3_free(buf.aBuf); 3650d5f0767cSdan 3651d5f0767cSdan return rc; 3652d5f0767cSdan } 3653d5f0767cSdan 36543739f298Sdan 3655296c7658Sdan /* 3656296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary 3657296c7658Sdan ** key. Assuming the following table structure: 3658296c7658Sdan ** 3659296c7658Sdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3660296c7658Sdan ** 3661296c7658Sdan ** The SELECT statement looks like this: 3662296c7658Sdan ** 3663296c7658Sdan ** SELECT * FROM x WHERE a = ?1 AND c = ?3 3664296c7658Sdan ** 3665296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left 3666296c7658Sdan ** pointing to the prepared version of the SQL statement. 3667296c7658Sdan */ 3668d5f0767cSdan static int sessionSelectRow( 3669d5f0767cSdan sqlite3 *db, /* Database handle */ 3670d5f0767cSdan const char *zTab, /* Table name */ 36710c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3672d5f0767cSdan ){ 3673d7fb7d24Sdan return sessionSelectStmt( 3674d7fb7d24Sdan db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect); 3675d5f0767cSdan } 3676d5f0767cSdan 3677296c7658Sdan /* 3678296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab. 3679296c7658Sdan ** For example: 3680296c7658Sdan ** 3681296c7658Sdan ** INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...); 3682296c7658Sdan ** 3683296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left 3684296c7658Sdan ** pointing to the prepared version of the SQL statement. 3685296c7658Sdan */ 36860c698471Sdan static int sessionInsertRow( 36870c698471Sdan sqlite3 *db, /* Database handle */ 36880c698471Sdan const char *zTab, /* Table name */ 36890c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 36900c698471Sdan ){ 36910c698471Sdan int rc = SQLITE_OK; 36920c698471Sdan int i; 36930c698471Sdan SessionBuffer buf = {0, 0, 0}; 36940c698471Sdan 36950c698471Sdan sessionAppendStr(&buf, "INSERT INTO main.", &rc); 36960c698471Sdan sessionAppendIdent(&buf, zTab, &rc); 3697ff677b20Sdan sessionAppendStr(&buf, "(", &rc); 3698ff677b20Sdan for(i=0; i<p->nCol; i++){ 3699ff677b20Sdan if( i!=0 ) sessionAppendStr(&buf, ", ", &rc); 3700ff677b20Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3701ff677b20Sdan } 3702ff677b20Sdan 3703ff677b20Sdan sessionAppendStr(&buf, ") VALUES(?", &rc); 37040c698471Sdan for(i=1; i<p->nCol; i++){ 37050c698471Sdan sessionAppendStr(&buf, ", ?", &rc); 37060c698471Sdan } 37070c698471Sdan sessionAppendStr(&buf, ")", &rc); 37080c698471Sdan 37090c698471Sdan if( rc==SQLITE_OK ){ 37100c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0); 37110c698471Sdan } 37120c698471Sdan sqlite3_free(buf.aBuf); 37130c698471Sdan return rc; 37140c698471Sdan } 37150c698471Sdan 37163739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){ 37173739f298Sdan return sqlite3_prepare_v2(db, zSql, -1, pp, 0); 37183739f298Sdan } 37193739f298Sdan 37203739f298Sdan /* 37213739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table. 37223739f298Sdan ** These are similar to those created by sessionSelectRow(), 37233739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for 37243739f298Sdan ** other tables. 37253739f298Sdan */ 37263739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){ 37273739f298Sdan int rc = sessionSelectRow(db, "sqlite_stat1", p); 37283739f298Sdan if( rc==SQLITE_OK ){ 37293739f298Sdan rc = sessionPrepare(db, &p->pInsert, 37303739f298Sdan "INSERT INTO main.sqlite_stat1 VALUES(?1, " 37313739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, " 37323739f298Sdan "?3)" 37333739f298Sdan ); 37343739f298Sdan } 37353739f298Sdan if( rc==SQLITE_OK ){ 37363739f298Sdan rc = sessionPrepare(db, &p->pUpdate, 37373739f298Sdan "UPDATE main.sqlite_stat1 SET " 37383739f298Sdan "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, " 37393739f298Sdan "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, " 37403739f298Sdan "stat = CASE WHEN ?8 THEN ?9 ELSE stat END " 37413739f298Sdan "WHERE tbl=?1 AND idx IS " 37423739f298Sdan "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END " 37433739f298Sdan "AND (?10 OR ?8=0 OR stat IS ?7)" 37443739f298Sdan ); 37453739f298Sdan } 37463739f298Sdan if( rc==SQLITE_OK ){ 37473739f298Sdan rc = sessionPrepare(db, &p->pDelete, 37483739f298Sdan "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS " 37493739f298Sdan "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END " 37503739f298Sdan "AND (?4 OR stat IS ?3)" 37513739f298Sdan ); 37523739f298Sdan } 37533739f298Sdan return rc; 37543739f298Sdan } 37553739f298Sdan 3756296c7658Sdan /* 37577aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem. 37587aa469cdSdan ** See comments in the body of this function for details. 37597aa469cdSdan */ 37607aa469cdSdan static int sessionBindValue( 37617aa469cdSdan sqlite3_stmt *pStmt, /* Statement to bind value to */ 37627aa469cdSdan int i, /* Parameter number to bind to */ 37637aa469cdSdan sqlite3_value *pVal /* Value to bind */ 37647aa469cdSdan ){ 37655671ef69Sdrh int eType = sqlite3_value_type(pVal); 3766082c96dfSdan /* COVERAGE: The (pVal->z==0) branch is never true using current versions 3767082c96dfSdan ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either 3768082c96dfSdan ** the (pVal->z) variable remains as it was or the type of the value is 3769082c96dfSdan ** set to SQLITE_NULL. */ 37705671ef69Sdrh if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){ 37717aa469cdSdan /* This condition occurs when an earlier OOM in a call to 37727aa469cdSdan ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within 3773082c96dfSdan ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */ 37747aa469cdSdan return SQLITE_NOMEM; 37757aa469cdSdan } 37767aa469cdSdan return sqlite3_bind_value(pStmt, i, pVal); 37777aa469cdSdan } 37787aa469cdSdan 37797aa469cdSdan /* 3780db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function 3781db04571cSdan ** transfers new.* values from the current iterator entry to statement 3782db04571cSdan ** pStmt. The table being inserted into has nCol columns. 3783db04571cSdan ** 3784d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of 3785db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1) 3786db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points 3787db04571cSdan ** to an array nCol elements in size. In this case only those values for 3788db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the 3789db04571cSdan ** statement. 3790db04571cSdan ** 3791db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK. 3792db04571cSdan */ 37937aa469cdSdan static int sessionBindRow( 3794db04571cSdan sqlite3_changeset_iter *pIter, /* Iterator to read values from */ 37957aa469cdSdan int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **), 3796db04571cSdan int nCol, /* Number of columns */ 3797db04571cSdan u8 *abPK, /* If not NULL, bind only if true */ 3798db04571cSdan sqlite3_stmt *pStmt /* Bind values to this statement */ 3799db04571cSdan ){ 3800db04571cSdan int i; 3801db04571cSdan int rc = SQLITE_OK; 38027aa469cdSdan 38037aa469cdSdan /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the 38047aa469cdSdan ** argument iterator points to a suitable entry. Make sure that xValue 38057aa469cdSdan ** is one of these to guarantee that it is safe to ignore the return 38067aa469cdSdan ** in the code below. */ 38077aa469cdSdan assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new ); 38087aa469cdSdan 3809db04571cSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 3810db04571cSdan if( !abPK || abPK[i] ){ 3811db04571cSdan sqlite3_value *pVal; 38127aa469cdSdan (void)xValue(pIter, i, &pVal); 3813dd8a4af8Sdan if( pVal==0 ){ 3814dd8a4af8Sdan /* The value in the changeset was "undefined". This indicates a 3815dd8a4af8Sdan ** corrupt changeset blob. */ 3816e341ec69Sdan rc = SQLITE_CORRUPT_BKPT; 3817dd8a4af8Sdan }else{ 38187aa469cdSdan rc = sessionBindValue(pStmt, i+1, pVal); 3819db04571cSdan } 3820db04571cSdan } 3821dd8a4af8Sdan } 3822db04571cSdan return rc; 3823db04571cSdan } 3824db04571cSdan 3825db04571cSdan /* 3826296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function. 3827296c7658Sdan ** This function binds the primary key values from the change that changeset 3828296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table 3829296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row 3830296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error 3831296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an 38327aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned. 38337aa469cdSdan ** 38347aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset() 38357aa469cdSdan ** statement pSelect. If any other value is returned, the statement does 38367aa469cdSdan ** not require a reset(). 3837296c7658Sdan ** 3838296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the 3839db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or 3840db04571cSdan ** UPDATE, bind values from the old.* record. 3841296c7658Sdan */ 38420c698471Sdan static int sessionSeekToRow( 384337f133ecSdan sqlite3 *db, /* Database handle */ 384437f133ecSdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 384537f133ecSdan u8 *abPK, /* Primary key flags array */ 38460c698471Sdan sqlite3_stmt *pSelect /* SELECT statement from sessionSelectRow() */ 384737f133ecSdan ){ 38487aa469cdSdan int rc; /* Return code */ 3849296c7658Sdan int nCol; /* Number of columns in table */ 3850296c7658Sdan int op; /* Changset operation (SQLITE_UPDATE etc.) */ 3851296c7658Sdan const char *zDummy; /* Unused */ 385237f133ecSdan 3853b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 38547aa469cdSdan rc = sessionBindRow(pIter, 3855db04571cSdan op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old, 3856db04571cSdan nCol, abPK, pSelect 3857db04571cSdan ); 38580c698471Sdan 38590c698471Sdan if( rc==SQLITE_OK ){ 38600c698471Sdan rc = sqlite3_step(pSelect); 38610c698471Sdan if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect); 38620c698471Sdan } 38630c698471Sdan 38640c698471Sdan return rc; 38650c698471Sdan } 38660c698471Sdan 3867b880a7b1Sdan /* 3868a920b209Sdrh ** This function is called from within sqlite3changeset_apply_v2() when 3869b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution 3870b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE).. 3871b880a7b1Sdan ** It adds a conflict resolution record to the buffer in 3872b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller 3873b880a7b1Sdan ** of apply_v2() as the "rebase" buffer. 3874b880a7b1Sdan ** 3875b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise. 3876b880a7b1Sdan */ 3877a38e6c57Sdan static int sessionRebaseAdd( 3878b880a7b1Sdan SessionApplyCtx *p, /* Apply context */ 3879b880a7b1Sdan int eType, /* Conflict resolution (OMIT or REPLACE) */ 3880b880a7b1Sdan sqlite3_changeset_iter *pIter /* Iterator pointing at current change */ 3881a38e6c57Sdan ){ 3882a38e6c57Sdan int rc = SQLITE_OK; 3883dbe7d37aSdan if( p->bRebase ){ 3884a38e6c57Sdan int i; 3885a38e6c57Sdan int eOp = pIter->op; 3886a38e6c57Sdan if( p->bRebaseStarted==0 ){ 3887a38e6c57Sdan /* Append a table-header to the rebase buffer */ 3888a38e6c57Sdan const char *zTab = pIter->zTab; 3889a38e6c57Sdan sessionAppendByte(&p->rebase, 'T', &rc); 3890a38e6c57Sdan sessionAppendVarint(&p->rebase, p->nCol, &rc); 3891a38e6c57Sdan sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc); 3892a38e6c57Sdan sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc); 3893a38e6c57Sdan p->bRebaseStarted = 1; 3894a38e6c57Sdan } 3895a38e6c57Sdan 3896a38e6c57Sdan assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT ); 3897a38e6c57Sdan assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE ); 3898a38e6c57Sdan 3899f01d3a7eSdan sessionAppendByte(&p->rebase, 3900f01d3a7eSdan (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc 3901f01d3a7eSdan ); 3902f01d3a7eSdan sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc); 3903a38e6c57Sdan for(i=0; i<p->nCol; i++){ 3904a38e6c57Sdan sqlite3_value *pVal = 0; 3905f01d3a7eSdan if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){ 3906a38e6c57Sdan sqlite3changeset_old(pIter, i, &pVal); 3907a38e6c57Sdan }else{ 3908a38e6c57Sdan sqlite3changeset_new(pIter, i, &pVal); 3909a38e6c57Sdan } 3910a38e6c57Sdan sessionAppendValue(&p->rebase, pVal, &rc); 3911a38e6c57Sdan } 3912dbe7d37aSdan } 3913a38e6c57Sdan return rc; 3914a38e6c57Sdan } 3915a38e6c57Sdan 3916296c7658Sdan /* 3917296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator 3918296c7658Sdan ** currently points to. 3919296c7658Sdan ** 3920296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT. 3921296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked 3922296c7658Sdan ** depends solely on eType, as follows: 3923296c7658Sdan ** 3924296c7658Sdan ** eType value Value passed to xConflict 3925296c7658Sdan ** ------------------------------------------------- 3926296c7658Sdan ** CHANGESET_DATA CHANGESET_NOTFOUND 3927296c7658Sdan ** CHANGESET_CONFLICT CHANGESET_CONSTRAINT 3928296c7658Sdan ** 3929296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing 3930296c7658Sdan ** record with the same primary key as the record about to be deleted, updated 3931296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict 3932296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict 3933296c7658Sdan ** handler invoked is as follows: 3934296c7658Sdan ** 3935296c7658Sdan ** eType value PK Record found? Value passed to xConflict 3936296c7658Sdan ** ---------------------------------------------------------------- 3937296c7658Sdan ** CHANGESET_DATA Yes CHANGESET_DATA 3938296c7658Sdan ** CHANGESET_DATA No CHANGESET_NOTFOUND 3939296c7658Sdan ** CHANGESET_CONFLICT Yes CHANGESET_CONFLICT 3940296c7658Sdan ** CHANGESET_CONFLICT No CHANGESET_CONSTRAINT 3941296c7658Sdan ** 3942296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and 3943296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace 3944296c7658Sdan ** is set to non-zero before returning SQLITE_OK. 3945296c7658Sdan ** 3946296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is 3947296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value, 3948296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT, 3949296c7658Sdan ** this function returns SQLITE_OK. 3950296c7658Sdan */ 39510c698471Sdan static int sessionConflictHandler( 3952296c7658Sdan int eType, /* Either CHANGESET_DATA or CONFLICT */ 3953296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 39540c698471Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 39550c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter*), 3956296c7658Sdan void *pCtx, /* First argument for conflict handler */ 3957296c7658Sdan int *pbReplace /* OUT: Set to true if PK row is found */ 39580c698471Sdan ){ 395974f598b6Smistachkin int res = 0; /* Value returned by conflict handler */ 39600c698471Sdan int rc; 39610c698471Sdan int nCol; 39620c698471Sdan int op; 39630c698471Sdan const char *zDummy; 39640c698471Sdan 3965b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 39660c698471Sdan 39670c698471Sdan assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA ); 39680c698471Sdan assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT ); 39690c698471Sdan assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND ); 397037f133ecSdan 397137f133ecSdan /* Bind the new.* PRIMARY KEY values to the SELECT statement. */ 39720c698471Sdan if( pbReplace ){ 39730c698471Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 39740c698471Sdan }else{ 3975db04571cSdan rc = SQLITE_OK; 39760c698471Sdan } 39770c698471Sdan 39780c698471Sdan if( rc==SQLITE_ROW ){ 39790c698471Sdan /* There exists another row with the new.* primary key. */ 39800c698471Sdan pIter->pConflict = p->pSelect; 39810c698471Sdan res = xConflict(pCtx, eType, pIter); 39820c698471Sdan pIter->pConflict = 0; 39830c698471Sdan rc = sqlite3_reset(p->pSelect); 3984db04571cSdan }else if( rc==SQLITE_OK ){ 3985d9151526Sdan if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){ 3986d9151526Sdan /* Instead of invoking the conflict handler, append the change blob 3987d9151526Sdan ** to the SessionApplyCtx.constraints buffer. */ 3988d9151526Sdan u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent]; 3989d9151526Sdan int nBlob = pIter->in.iNext - pIter->in.iCurrent; 3990d9151526Sdan sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc); 3991a38e6c57Sdan return SQLITE_OK; 3992d9151526Sdan }else{ 39930c698471Sdan /* No other row with the new.* primary key. */ 39940c698471Sdan res = xConflict(pCtx, eType+1, pIter); 39950c698471Sdan if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE; 399637f133ecSdan } 3997d9151526Sdan } 399837f133ecSdan 399937f133ecSdan if( rc==SQLITE_OK ){ 40000c698471Sdan switch( res ){ 40010c698471Sdan case SQLITE_CHANGESET_REPLACE: 4002f51e5f6cSdan assert( pbReplace ); 4003f51e5f6cSdan *pbReplace = 1; 40040c698471Sdan break; 40050c698471Sdan 40060c698471Sdan case SQLITE_CHANGESET_OMIT: 40070c698471Sdan break; 40080c698471Sdan 40090c698471Sdan case SQLITE_CHANGESET_ABORT: 40100c698471Sdan rc = SQLITE_ABORT; 40110c698471Sdan break; 40120c698471Sdan 40130c698471Sdan default: 40140c698471Sdan rc = SQLITE_MISUSE; 40150c698471Sdan break; 40160c698471Sdan } 4017a38e6c57Sdan if( rc==SQLITE_OK ){ 4018a38e6c57Sdan rc = sessionRebaseAdd(p, res, pIter); 4019a38e6c57Sdan } 40200c698471Sdan } 40210c698471Sdan 40220c698471Sdan return rc; 40230c698471Sdan } 40240c698471Sdan 4025296c7658Sdan /* 4026296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument 4027296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke 4028296c7658Sdan ** the conflict handler callback. 4029296c7658Sdan ** 4030296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If 4031296c7658Sdan ** one is encountered, update or delete the row with the matching primary key 4032296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs, 4033296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry 4034296c7658Sdan ** to true before returning. In this case the caller will invoke this function 4035296c7658Sdan ** again, this time with pbRetry set to NULL. 4036296c7658Sdan ** 4037296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is 4038296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead. 4039296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such 4040296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true 4041296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting 4042296c7658Sdan ** row before invoking this function again, this time with pbReplace set 4043296c7658Sdan ** to NULL. 4044296c7658Sdan ** 4045296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function 4046296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is 4047296c7658Sdan ** returned. 4048296c7658Sdan */ 40490c698471Sdan static int sessionApplyOneOp( 4050296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 4051296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 40520c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter *), 4053296c7658Sdan void *pCtx, /* First argument for the conflict handler */ 4054296c7658Sdan int *pbReplace, /* OUT: True to remove PK row and retry */ 4055296c7658Sdan int *pbRetry /* OUT: True to retry. */ 40560c698471Sdan ){ 40570c698471Sdan const char *zDummy; 40580c698471Sdan int op; 40590c698471Sdan int nCol; 40600c698471Sdan int rc = SQLITE_OK; 40610c698471Sdan 40620c698471Sdan assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect ); 40630c698471Sdan assert( p->azCol && p->abPK ); 40640c698471Sdan assert( !pbReplace || *pbReplace==0 ); 40650c698471Sdan 4066b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 40670c698471Sdan 40680c698471Sdan if( op==SQLITE_DELETE ){ 40690c698471Sdan 407073b3c055Sdan /* Bind values to the DELETE statement. If conflict handling is required, 407173b3c055Sdan ** bind values for all columns and set bound variable (nCol+1) to true. 407273b3c055Sdan ** Or, if conflict handling is not required, bind just the PK column 407373b3c055Sdan ** values and, if it exists, set (nCol+1) to false. Conflict handling 407473b3c055Sdan ** is not required if: 407573b3c055Sdan ** 407673b3c055Sdan ** * this is a patchset, or 407773b3c055Sdan ** * (pbRetry==0), or 407873b3c055Sdan ** * all columns of the table are PK columns (in this case there is 407973b3c055Sdan ** no (nCol+1) variable to bind to). 408073b3c055Sdan */ 408173b3c055Sdan u8 *abPK = (pIter->bPatchset ? p->abPK : 0); 408273b3c055Sdan rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete); 40837cf7df7dSdan if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){ 408473b3c055Sdan rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK)); 40857cf7df7dSdan } 40860c698471Sdan if( rc!=SQLITE_OK ) return rc; 40870c698471Sdan 40880c698471Sdan sqlite3_step(p->pDelete); 40890c698471Sdan rc = sqlite3_reset(p->pDelete); 40900c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 40910c698471Sdan rc = sessionConflictHandler( 40920c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 40930c698471Sdan ); 409435e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 40950c698471Sdan rc = sessionConflictHandler( 40960c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 40970c698471Sdan ); 40980c698471Sdan } 40990c698471Sdan 41000c698471Sdan }else if( op==SQLITE_UPDATE ){ 41010c698471Sdan int i; 41020c698471Sdan 41030c698471Sdan /* Bind values to the UPDATE statement. */ 41040c698471Sdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 41057aa469cdSdan sqlite3_value *pOld = sessionChangesetOld(pIter, i); 41067aa469cdSdan sqlite3_value *pNew = sessionChangesetNew(pIter, i); 41077aa469cdSdan 41080c698471Sdan sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew); 41097aa469cdSdan if( pOld ){ 41107aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+1, pOld); 41117aa469cdSdan } 41127aa469cdSdan if( rc==SQLITE_OK && pNew ){ 41137aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+3, pNew); 41140c698471Sdan } 41150c698471Sdan } 411673b3c055Sdan if( rc==SQLITE_OK ){ 411773b3c055Sdan sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset); 411873b3c055Sdan } 41190c698471Sdan if( rc!=SQLITE_OK ) return rc; 41200c698471Sdan 41210c698471Sdan /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict, 41220c698471Sdan ** the result will be SQLITE_OK with 0 rows modified. */ 41230c698471Sdan sqlite3_step(p->pUpdate); 41240c698471Sdan rc = sqlite3_reset(p->pUpdate); 41250c698471Sdan 41260c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 41270c698471Sdan /* A NOTFOUND or DATA error. Search the table to see if it contains 41280c698471Sdan ** a row with a matching primary key. If so, this is a DATA conflict. 41290c698471Sdan ** Otherwise, if there is no primary key match, it is a NOTFOUND. */ 41300c698471Sdan 41310c698471Sdan rc = sessionConflictHandler( 41320c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 41330c698471Sdan ); 41340c698471Sdan 413535e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 4136db04571cSdan /* This is always a CONSTRAINT conflict. */ 4137db04571cSdan rc = sessionConflictHandler( 4138db04571cSdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 41390c698471Sdan ); 41400c698471Sdan } 41410c698471Sdan 41420c698471Sdan }else{ 41430c698471Sdan assert( op==SQLITE_INSERT ); 4144d1cccf19Sdan if( p->bStat1 ){ 4145d1cccf19Sdan /* Check if there is a conflicting row. For sqlite_stat1, this needs 4146d1cccf19Sdan ** to be done using a SELECT, as there is no PRIMARY KEY in the 4147d1cccf19Sdan ** database schema to throw an exception if a duplicate is inserted. */ 4148d1cccf19Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 4149d1cccf19Sdan if( rc==SQLITE_ROW ){ 4150d1cccf19Sdan rc = SQLITE_CONSTRAINT; 4151d1cccf19Sdan sqlite3_reset(p->pSelect); 4152d1cccf19Sdan } 4153d1cccf19Sdan } 4154d1cccf19Sdan 4155d1cccf19Sdan if( rc==SQLITE_OK ){ 41567aa469cdSdan rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert); 41570c698471Sdan if( rc!=SQLITE_OK ) return rc; 41580c698471Sdan 41590c698471Sdan sqlite3_step(p->pInsert); 41600c698471Sdan rc = sqlite3_reset(p->pInsert); 4161d1cccf19Sdan } 4162d1cccf19Sdan 416335e2858eSdan if( (rc&0xff)==SQLITE_CONSTRAINT ){ 41640c698471Sdan rc = sessionConflictHandler( 41650c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace 41660c698471Sdan ); 416737f133ecSdan } 416837f133ecSdan } 416937f133ecSdan 417037f133ecSdan return rc; 417137f133ecSdan } 417237f133ecSdan 41735f5663dcSdan /* 41745f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument 41755f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke 41765f5663dcSdan ** the conflict handler callback. 41775f5663dcSdan ** 41785f5663dcSdan ** The difference between this function and sessionApplyOne() is that this 41795f5663dcSdan ** function handles the case where the conflict-handler is invoked and 41805f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be 41815f5663dcSdan ** retried in some manner. 41825f5663dcSdan */ 4183d9151526Sdan static int sessionApplyOneWithRetry( 4184d9151526Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4185d9151526Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator to read change from */ 4186d9151526Sdan SessionApplyCtx *pApply, /* Apply context */ 4187d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4188d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4189d9151526Sdan ){ 4190d9151526Sdan int bReplace = 0; 4191d9151526Sdan int bRetry = 0; 4192d9151526Sdan int rc; 4193d9151526Sdan 4194d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry); 4195a38e6c57Sdan if( rc==SQLITE_OK ){ 41965f5663dcSdan /* If the bRetry flag is set, the change has not been applied due to an 41975f5663dcSdan ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and 41985f5663dcSdan ** a row with the correct PK is present in the db, but one or more other 41995f5663dcSdan ** fields do not contain the expected values) and the conflict handler 42005f5663dcSdan ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation, 42015f5663dcSdan ** but pass NULL as the final argument so that sessionApplyOneOp() ignores 42025f5663dcSdan ** the SQLITE_CHANGESET_DATA problem. */ 42035f5663dcSdan if( bRetry ){ 42045f5663dcSdan assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE ); 42055f5663dcSdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4206d9151526Sdan } 4207d9151526Sdan 42085f5663dcSdan /* If the bReplace flag is set, the change is an INSERT that has not 42095f5663dcSdan ** been performed because the database already contains a row with the 42105f5663dcSdan ** specified primary key and the conflict handler returned 42115f5663dcSdan ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row 42125f5663dcSdan ** before reattempting the INSERT. */ 42135f5663dcSdan else if( bReplace ){ 4214d9151526Sdan assert( pIter->op==SQLITE_INSERT ); 4215d9151526Sdan rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0); 4216d9151526Sdan if( rc==SQLITE_OK ){ 4217d9151526Sdan rc = sessionBindRow(pIter, 4218d9151526Sdan sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete); 4219d9151526Sdan sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1); 4220d9151526Sdan } 4221d9151526Sdan if( rc==SQLITE_OK ){ 4222d9151526Sdan sqlite3_step(pApply->pDelete); 4223d9151526Sdan rc = sqlite3_reset(pApply->pDelete); 4224d9151526Sdan } 4225d9151526Sdan if( rc==SQLITE_OK ){ 4226d9151526Sdan rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0); 4227d9151526Sdan } 4228d9151526Sdan if( rc==SQLITE_OK ){ 4229d9151526Sdan rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0); 4230d9151526Sdan } 4231d9151526Sdan } 4232a38e6c57Sdan } 4233d9151526Sdan 4234d9151526Sdan return rc; 4235d9151526Sdan } 4236d9151526Sdan 4237d9151526Sdan /* 4238d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer. 4239d9151526Sdan */ 4240d9151526Sdan static int sessionRetryConstraints( 4241d9151526Sdan sqlite3 *db, 4242d9151526Sdan int bPatchset, 4243d9151526Sdan const char *zTab, 4244d9151526Sdan SessionApplyCtx *pApply, 4245d9151526Sdan int(*xConflict)(void*, int, sqlite3_changeset_iter*), 4246d9151526Sdan void *pCtx /* First argument passed to xConflict */ 4247d9151526Sdan ){ 4248d9151526Sdan int rc = SQLITE_OK; 4249d9151526Sdan 4250d9151526Sdan while( pApply->constraints.nBuf ){ 4251d9151526Sdan sqlite3_changeset_iter *pIter2 = 0; 4252d9151526Sdan SessionBuffer cons = pApply->constraints; 4253d9151526Sdan memset(&pApply->constraints, 0, sizeof(SessionBuffer)); 4254d9151526Sdan 4255*5d237bfaSdan rc = sessionChangesetStart( 4256*5d237bfaSdan &pIter2, 0, 0, cons.nBuf, cons.aBuf, pApply->bInvertConstraints 4257*5d237bfaSdan ); 4258d9151526Sdan if( rc==SQLITE_OK ){ 4259f6ad201aSdrh size_t nByte = 2*pApply->nCol*sizeof(sqlite3_value*); 4260d9151526Sdan int rc2; 4261d9151526Sdan pIter2->bPatchset = bPatchset; 4262d9151526Sdan pIter2->zTab = (char*)zTab; 4263d9151526Sdan pIter2->nCol = pApply->nCol; 4264d9151526Sdan pIter2->abPK = pApply->abPK; 4265d9151526Sdan sessionBufferGrow(&pIter2->tblhdr, nByte, &rc); 4266d9151526Sdan pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf; 4267d9151526Sdan if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte); 4268d9151526Sdan 4269d9151526Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){ 4270d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx); 4271d9151526Sdan } 4272d9151526Sdan 4273d9151526Sdan rc2 = sqlite3changeset_finalize(pIter2); 42747e0765a9Sdrh if( rc==SQLITE_OK ) rc = rc2; 4275d9151526Sdan } 4276d9151526Sdan assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 ); 4277d9151526Sdan 4278d9151526Sdan sqlite3_free(cons.aBuf); 4279d9151526Sdan if( rc!=SQLITE_OK ) break; 4280d9151526Sdan if( pApply->constraints.nBuf>=cons.nBuf ){ 4281d9151526Sdan /* No progress was made on the last round. */ 4282d9151526Sdan pApply->bDeferConstraints = 0; 4283d9151526Sdan } 4284d9151526Sdan } 4285d9151526Sdan 4286d9151526Sdan return rc; 4287d9151526Sdan } 4288d9151526Sdan 4289296c7658Sdan /* 42904757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but 42914757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the 42924757c658Sdan ** changeset to the main database attached to handle "db". The supplied 42934757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered 42944757c658Sdan ** while applying the change. 4295296c7658Sdan */ 42964757c658Sdan static int sessionChangesetApply( 4297296c7658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 42984757c658Sdan sqlite3_changeset_iter *pIter, /* Changeset to apply */ 429940368988Sdan int(*xFilter)( 430040368988Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 430140368988Sdan const char *zTab /* Table name */ 430240368988Sdan ), 4303d5f0767cSdan int(*xConflict)( 4304d5f0767cSdan void *pCtx, /* Copy of fifth arg to _apply() */ 4305d5f0767cSdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4306d5f0767cSdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4307d5f0767cSdan ), 4308a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4309fe55da38Sdan void **ppRebase, int *pnRebase, /* OUT: Rebase information */ 4310fe55da38Sdan int flags /* SESSION_APPLY_XXX flags */ 4311d5f0767cSdan ){ 4312ca62ad57Sdan int schemaMismatch = 0; 4313fe55da38Sdan int rc = SQLITE_OK; /* Return code */ 4314d5f0767cSdan const char *zTab = 0; /* Name of current table */ 4315cfdbde21Sdrh int nTab = 0; /* Result of sqlite3Strlen30(zTab) */ 4316296c7658Sdan SessionApplyCtx sApply; /* changeset_apply() context object */ 43175f5663dcSdan int bPatchset; 4318d5f0767cSdan 4319082c96dfSdan assert( xConflict!=0 ); 4320082c96dfSdan 4321d9151526Sdan pIter->in.bNoDiscard = 1; 43220c698471Sdan memset(&sApply, 0, sizeof(sApply)); 4323dbe7d37aSdan sApply.bRebase = (ppRebase && pnRebase); 4324*5d237bfaSdan sApply.bInvertConstraints = !!(flags & SQLITE_CHANGESETAPPLY_INVERT); 43254c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 4326fe55da38Sdan if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){ 43270c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0); 4328fe55da38Sdan } 4329cb3e4b79Sdan if( rc==SQLITE_OK ){ 4330cb3e4b79Sdan rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0); 4331cb3e4b79Sdan } 43320c698471Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){ 43330c698471Sdan int nCol; 4334d5f0767cSdan int op; 43350c698471Sdan const char *zNew; 4336ca62ad57Sdan 4337b4480e94Sdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0); 4338d5f0767cSdan 43390c698471Sdan if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){ 4340ca62ad57Sdan u8 *abPK; 4341ca62ad57Sdan 4342d9151526Sdan rc = sessionRetryConstraints( 4343d9151526Sdan db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx 4344d9151526Sdan ); 4345d9151526Sdan if( rc!=SQLITE_OK ) break; 4346d9151526Sdan 4347cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 43480c698471Sdan sqlite3_finalize(sApply.pDelete); 43490c698471Sdan sqlite3_finalize(sApply.pUpdate); 43500c698471Sdan sqlite3_finalize(sApply.pInsert); 43510c698471Sdan sqlite3_finalize(sApply.pSelect); 43520c698471Sdan sApply.db = db; 4353f1b40e83Sdan sApply.pDelete = 0; 4354f1b40e83Sdan sApply.pUpdate = 0; 4355f1b40e83Sdan sApply.pInsert = 0; 4356f1b40e83Sdan sApply.pSelect = 0; 4357f1b40e83Sdan sApply.nCol = 0; 4358f1b40e83Sdan sApply.azCol = 0; 4359f1b40e83Sdan sApply.abPK = 0; 4360f1b40e83Sdan sApply.bStat1 = 0; 4361d9151526Sdan sApply.bDeferConstraints = 1; 4362a38e6c57Sdan sApply.bRebaseStarted = 0; 4363f1b40e83Sdan memset(&sApply.constraints, 0, sizeof(SessionBuffer)); 436437f133ecSdan 436540368988Sdan /* If an xFilter() callback was specified, invoke it now. If the 436640368988Sdan ** xFilter callback returns zero, skip this table. If it returns 436740368988Sdan ** non-zero, proceed. */ 436840368988Sdan schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew))); 436940368988Sdan if( schemaMismatch ){ 437040368988Sdan zTab = sqlite3_mprintf("%s", zNew); 4371f05ac112Sdan if( zTab==0 ){ 4372f05ac112Sdan rc = SQLITE_NOMEM; 4373f05ac112Sdan break; 4374f05ac112Sdan } 43754f528042Sdan nTab = (int)strlen(zTab); 437640368988Sdan sApply.azCol = (const char **)zTab; 437740368988Sdan }else{ 4378ff677b20Sdan int nMinCol = 0; 4379ff677b20Sdan int i; 4380ff677b20Sdan 4381ca62ad57Sdan sqlite3changeset_pk(pIter, &abPK, 0); 4382296c7658Sdan rc = sessionTableInfo( 4383ca62ad57Sdan db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK 4384ca62ad57Sdan ); 4385ca62ad57Sdan if( rc!=SQLITE_OK ) break; 4386ff677b20Sdan for(i=0; i<sApply.nCol; i++){ 4387ff677b20Sdan if( sApply.abPK[i] ) nMinCol = i+1; 4388ff677b20Sdan } 43890c698471Sdan 4390ca62ad57Sdan if( sApply.nCol==0 ){ 4391ca62ad57Sdan schemaMismatch = 1; 4392ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4393ca62ad57Sdan "sqlite3changeset_apply(): no such table: %s", zTab 4394ca62ad57Sdan ); 4395ca62ad57Sdan } 4396ff677b20Sdan else if( sApply.nCol<nCol ){ 4397ca62ad57Sdan schemaMismatch = 1; 4398ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 4399ff677b20Sdan "sqlite3changeset_apply(): table %s has %d columns, " 4400ff677b20Sdan "expected %d or more", 4401ca62ad57Sdan zTab, sApply.nCol, nCol 4402ca62ad57Sdan ); 4403ca62ad57Sdan } 4404ff677b20Sdan else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){ 4405ca62ad57Sdan schemaMismatch = 1; 440640368988Sdan sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): " 440740368988Sdan "primary key mismatch for table %s", zTab 4408ca62ad57Sdan ); 4409ca62ad57Sdan } 4410ff677b20Sdan else{ 4411ff677b20Sdan sApply.nCol = nCol; 44123739f298Sdan if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){ 44133739f298Sdan if( (rc = sessionStat1Sql(db, &sApply) ) ){ 44143739f298Sdan break; 44153739f298Sdan } 4416d1cccf19Sdan sApply.bStat1 = 1; 4417d1cccf19Sdan }else{ 4418ff677b20Sdan if((rc = sessionSelectRow(db, zTab, &sApply)) 44190c698471Sdan || (rc = sessionUpdateRow(db, zTab, &sApply)) 44200c698471Sdan || (rc = sessionDeleteRow(db, zTab, &sApply)) 44210c698471Sdan || (rc = sessionInsertRow(db, zTab, &sApply)) 442237f133ecSdan ){ 442337f133ecSdan break; 442437f133ecSdan } 4425d1cccf19Sdan sApply.bStat1 = 0; 4426d1cccf19Sdan } 4427ff677b20Sdan } 4428cfdbde21Sdrh nTab = sqlite3Strlen30(zTab); 4429d5f0767cSdan } 443040368988Sdan } 4431d5f0767cSdan 4432ca62ad57Sdan /* If there is a schema mismatch on the current table, proceed to the 4433ca62ad57Sdan ** next change. A log message has already been issued. */ 4434ca62ad57Sdan if( schemaMismatch ) continue; 4435ca62ad57Sdan 4436d9151526Sdan rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx); 44370c698471Sdan } 44380c698471Sdan 44395f5663dcSdan bPatchset = pIter->bPatchset; 4440296c7658Sdan if( rc==SQLITE_OK ){ 4441296c7658Sdan rc = sqlite3changeset_finalize(pIter); 4442296c7658Sdan }else{ 4443296c7658Sdan sqlite3changeset_finalize(pIter); 4444296c7658Sdan } 4445d5f0767cSdan 4446d5f0767cSdan if( rc==SQLITE_OK ){ 44475f5663dcSdan rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx); 44485f5663dcSdan } 44495f5663dcSdan 44505f5663dcSdan if( rc==SQLITE_OK ){ 445107001c45Sdrh int nFk, notUsed; 445207001c45Sdrh sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, ¬Used, 0); 445307001c45Sdrh if( nFk!=0 ){ 4454cb3e4b79Sdan int res = SQLITE_CHANGESET_ABORT; 4455cb3e4b79Sdan sqlite3_changeset_iter sIter; 4456cb3e4b79Sdan memset(&sIter, 0, sizeof(sIter)); 4457cb3e4b79Sdan sIter.nCol = nFk; 4458cb3e4b79Sdan res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter); 4459cb3e4b79Sdan if( res!=SQLITE_CHANGESET_OMIT ){ 4460cb3e4b79Sdan rc = SQLITE_CONSTRAINT; 4461cb3e4b79Sdan } 4462cb3e4b79Sdan } 4463cb3e4b79Sdan } 4464cb3e4b79Sdan sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0); 4465cb3e4b79Sdan 4466fe55da38Sdan if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){ 4467cb3e4b79Sdan if( rc==SQLITE_OK ){ 4468d5f0767cSdan rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4469d5f0767cSdan }else{ 4470d5f0767cSdan sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0); 4471d5f0767cSdan sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 4472d5f0767cSdan } 4473fe55da38Sdan } 4474d5f0767cSdan 4475dbe7d37aSdan assert( sApply.bRebase || sApply.rebase.nBuf==0 ); 4476dbe7d37aSdan if( rc==SQLITE_OK && bPatchset==0 && sApply.bRebase ){ 4477a38e6c57Sdan *ppRebase = (void*)sApply.rebase.aBuf; 4478a38e6c57Sdan *pnRebase = sApply.rebase.nBuf; 4479a38e6c57Sdan sApply.rebase.aBuf = 0; 4480a38e6c57Sdan } 44810c698471Sdan sqlite3_finalize(sApply.pInsert); 44820c698471Sdan sqlite3_finalize(sApply.pDelete); 44830c698471Sdan sqlite3_finalize(sApply.pUpdate); 44840c698471Sdan sqlite3_finalize(sApply.pSelect); 4485cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 4486d9151526Sdan sqlite3_free((char*)sApply.constraints.aBuf); 4487a38e6c57Sdan sqlite3_free((char*)sApply.rebase.aBuf); 44884c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 4489d5f0767cSdan return rc; 4490d5f0767cSdan } 449191ddd559Sdan 4492b880a7b1Sdan /* 4493b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main 4494b880a7b1Sdan ** database attached to handle "db". 4495b880a7b1Sdan */ 4496a38e6c57Sdan int sqlite3changeset_apply_v2( 4497a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4498a38e6c57Sdan int nChangeset, /* Size of changeset in bytes */ 4499a38e6c57Sdan void *pChangeset, /* Changeset blob */ 4500a38e6c57Sdan int(*xFilter)( 4501a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4502a38e6c57Sdan const char *zTab /* Table name */ 4503a38e6c57Sdan ), 4504a38e6c57Sdan int(*xConflict)( 4505a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4506a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4507a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4508a38e6c57Sdan ), 4509a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4510fe55da38Sdan void **ppRebase, int *pnRebase, 4511fe55da38Sdan int flags 4512a38e6c57Sdan ){ 4513a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 451444748f27Sdan int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT); 451544748f27Sdan int rc = sessionChangesetStart(&pIter, 0, 0, nChangeset, pChangeset,bInverse); 4516a38e6c57Sdan if( rc==SQLITE_OK ){ 4517a38e6c57Sdan rc = sessionChangesetApply( 4518fe55da38Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags 4519a38e6c57Sdan ); 4520a38e6c57Sdan } 4521a38e6c57Sdan return rc; 4522a38e6c57Sdan } 4523a38e6c57Sdan 452477fc1d5bSdan /* 45254757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database 45264757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 45274757c658Sdan ** to resolve any conflicts encountered while applying the change. 45284757c658Sdan */ 45294757c658Sdan int sqlite3changeset_apply( 45304757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 45314757c658Sdan int nChangeset, /* Size of changeset in bytes */ 45324757c658Sdan void *pChangeset, /* Changeset blob */ 45334757c658Sdan int(*xFilter)( 45344757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45354757c658Sdan const char *zTab /* Table name */ 45364757c658Sdan ), 45374757c658Sdan int(*xConflict)( 45384757c658Sdan void *pCtx, /* Copy of fifth arg to _apply() */ 45394757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 45404757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 45414757c658Sdan ), 45424757c658Sdan void *pCtx /* First argument passed to xConflict */ 45434757c658Sdan ){ 4544a38e6c57Sdan return sqlite3changeset_apply_v2( 4545fe55da38Sdan db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0, 0 4546a38e6c57Sdan ); 45474757c658Sdan } 45484757c658Sdan 45494757c658Sdan /* 45504757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database 45514757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 45524757c658Sdan ** to resolve any conflicts encountered while applying the change. 45534757c658Sdan */ 4554a38e6c57Sdan int sqlite3changeset_apply_v2_strm( 4555a38e6c57Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 4556a38e6c57Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 4557a38e6c57Sdan void *pIn, /* First arg for xInput */ 4558a38e6c57Sdan int(*xFilter)( 4559a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4560a38e6c57Sdan const char *zTab /* Table name */ 4561a38e6c57Sdan ), 4562a38e6c57Sdan int(*xConflict)( 4563a38e6c57Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 4564a38e6c57Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 4565a38e6c57Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 4566a38e6c57Sdan ), 4567a38e6c57Sdan void *pCtx, /* First argument passed to xConflict */ 4568fe55da38Sdan void **ppRebase, int *pnRebase, 4569fe55da38Sdan int flags 4570a38e6c57Sdan ){ 4571a38e6c57Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 457244748f27Sdan int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT); 457344748f27Sdan int rc = sessionChangesetStart(&pIter, xInput, pIn, 0, 0, bInverse); 4574a38e6c57Sdan if( rc==SQLITE_OK ){ 4575a38e6c57Sdan rc = sessionChangesetApply( 4576fe55da38Sdan db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags 4577a38e6c57Sdan ); 4578a38e6c57Sdan } 4579a38e6c57Sdan return rc; 4580a38e6c57Sdan } 4581f1a08ad8Sdrh int sqlite3changeset_apply_strm( 45824757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 45834757c658Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 45844757c658Sdan void *pIn, /* First arg for xInput */ 45854757c658Sdan int(*xFilter)( 45864757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45874757c658Sdan const char *zTab /* Table name */ 45884757c658Sdan ), 45894757c658Sdan int(*xConflict)( 45904757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 45914757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 45924757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 45934757c658Sdan ), 45944757c658Sdan void *pCtx /* First argument passed to xConflict */ 45954757c658Sdan ){ 4596a38e6c57Sdan return sqlite3changeset_apply_v2_strm( 4597fe55da38Sdan db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0, 0 4598a38e6c57Sdan ); 45994757c658Sdan } 46004757c658Sdan 46014757c658Sdan /* 46025898ad69Sdan ** sqlite3_changegroup handle. 46035898ad69Sdan */ 46045898ad69Sdan struct sqlite3_changegroup { 46055898ad69Sdan int rc; /* Error code */ 46065898ad69Sdan int bPatch; /* True to accumulate patchsets */ 46075898ad69Sdan SessionTable *pList; /* List of tables in current patch */ 46085898ad69Sdan }; 46095898ad69Sdan 46105898ad69Sdan /* 461177fc1d5bSdan ** This function is called to merge two changes to the same row together as 461277fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is 461377fc1d5bSdan ** allocated and a pointer to it stored in *ppNew. 461477fc1d5bSdan */ 46155d607a6eSdan static int sessionChangeMerge( 461677fc1d5bSdan SessionTable *pTab, /* Table structure */ 4617c0a499eaSdan int bRebase, /* True for a rebase hash-table */ 461864277f4aSdan int bPatchset, /* True for patchsets */ 461977fc1d5bSdan SessionChange *pExist, /* Existing change */ 462077fc1d5bSdan int op2, /* Second change operation */ 462177fc1d5bSdan int bIndirect, /* True if second change is indirect */ 462277fc1d5bSdan u8 *aRec, /* Second change record */ 462377fc1d5bSdan int nRec, /* Number of bytes in aRec */ 462477fc1d5bSdan SessionChange **ppNew /* OUT: Merged change */ 46255d607a6eSdan ){ 46265d607a6eSdan SessionChange *pNew = 0; 462724a0c453Sdan int rc = SQLITE_OK; 46285d607a6eSdan 46295d607a6eSdan if( !pExist ){ 46302d77d80aSdrh pNew = (SessionChange *)sqlite3_malloc64(sizeof(SessionChange) + nRec); 46315d607a6eSdan if( !pNew ){ 46325d607a6eSdan return SQLITE_NOMEM; 46335d607a6eSdan } 46345d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 4635798693b2Sdan pNew->op = op2; 46365d607a6eSdan pNew->bIndirect = bIndirect; 4637cbf6d2d2Sdan pNew->aRecord = (u8*)&pNew[1]; 463824a0c453Sdan if( bIndirect==0 || bRebase==0 ){ 463924a0c453Sdan pNew->nRecord = nRec; 4640cbf6d2d2Sdan memcpy(pNew->aRecord, aRec, nRec); 464124a0c453Sdan }else{ 464224a0c453Sdan int i; 464324a0c453Sdan u8 *pIn = aRec; 464424a0c453Sdan u8 *pOut = pNew->aRecord; 464524a0c453Sdan for(i=0; i<pTab->nCol; i++){ 464624a0c453Sdan int nIn = sessionSerialLen(pIn); 464724a0c453Sdan if( *pIn==0 ){ 464824a0c453Sdan *pOut++ = 0; 464924a0c453Sdan }else if( pTab->abPK[i]==0 ){ 465024a0c453Sdan *pOut++ = 0xFF; 465124a0c453Sdan }else{ 465224a0c453Sdan memcpy(pOut, pIn, nIn); 465324a0c453Sdan pOut += nIn; 465424a0c453Sdan } 465524a0c453Sdan pIn += nIn; 465624a0c453Sdan } 465724a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 465824a0c453Sdan } 4659c0a499eaSdan }else if( bRebase ){ 466024a0c453Sdan if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){ 466124a0c453Sdan *ppNew = pExist; 466224a0c453Sdan }else{ 46632d77d80aSdrh sqlite3_int64 nByte = nRec + pExist->nRecord + sizeof(SessionChange); 46642d77d80aSdrh pNew = (SessionChange*)sqlite3_malloc64(nByte); 466524a0c453Sdan if( pNew==0 ){ 466624a0c453Sdan rc = SQLITE_NOMEM; 466724a0c453Sdan }else{ 466824a0c453Sdan int i; 466924a0c453Sdan u8 *a1 = pExist->aRecord; 467024a0c453Sdan u8 *a2 = aRec; 467124a0c453Sdan u8 *pOut; 467224a0c453Sdan 467324a0c453Sdan memset(pNew, 0, nByte); 467424a0c453Sdan pNew->bIndirect = bIndirect || pExist->bIndirect; 467524a0c453Sdan pNew->op = op2; 467624a0c453Sdan pOut = pNew->aRecord = (u8*)&pNew[1]; 467724a0c453Sdan 467824a0c453Sdan for(i=0; i<pTab->nCol; i++){ 467924a0c453Sdan int n1 = sessionSerialLen(a1); 468024a0c453Sdan int n2 = sessionSerialLen(a2); 4681f231e18cSdan if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){ 468224a0c453Sdan *pOut++ = 0xFF; 468324a0c453Sdan }else if( *a2==0 ){ 468424a0c453Sdan memcpy(pOut, a1, n1); 468524a0c453Sdan pOut += n1; 468624a0c453Sdan }else{ 468724a0c453Sdan memcpy(pOut, a2, n2); 468824a0c453Sdan pOut += n2; 468924a0c453Sdan } 469024a0c453Sdan a1 += n1; 469124a0c453Sdan a2 += n2; 469224a0c453Sdan } 469324a0c453Sdan pNew->nRecord = pOut - pNew->aRecord; 469424a0c453Sdan } 469524a0c453Sdan sqlite3_free(pExist); 469624a0c453Sdan } 46975d607a6eSdan }else{ 4698798693b2Sdan int op1 = pExist->op; 46995d607a6eSdan 47005d607a6eSdan /* 47015d607a6eSdan ** op1=INSERT, op2=INSERT -> Unsupported. Discard op2. 47025d607a6eSdan ** op1=INSERT, op2=UPDATE -> INSERT. 47035d607a6eSdan ** op1=INSERT, op2=DELETE -> (none) 47045d607a6eSdan ** 47055d607a6eSdan ** op1=UPDATE, op2=INSERT -> Unsupported. Discard op2. 47065d607a6eSdan ** op1=UPDATE, op2=UPDATE -> UPDATE. 47075d607a6eSdan ** op1=UPDATE, op2=DELETE -> DELETE. 47085d607a6eSdan ** 47095d607a6eSdan ** op1=DELETE, op2=INSERT -> UPDATE. 47105d607a6eSdan ** op1=DELETE, op2=UPDATE -> Unsupported. Discard op2. 47115d607a6eSdan ** op1=DELETE, op2=DELETE -> Unsupported. Discard op2. 47125d607a6eSdan */ 47135d607a6eSdan if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT) 47145d607a6eSdan || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT) 47155d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE) 47165d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_DELETE) 47175d607a6eSdan ){ 47185d607a6eSdan pNew = pExist; 47195d607a6eSdan }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){ 47205d607a6eSdan sqlite3_free(pExist); 47215d607a6eSdan assert( pNew==0 ); 47225d607a6eSdan }else{ 472364277f4aSdan u8 *aExist = pExist->aRecord; 47242d77d80aSdrh sqlite3_int64 nByte; 47255d607a6eSdan u8 *aCsr; 47265d607a6eSdan 472764277f4aSdan /* Allocate a new SessionChange object. Ensure that the aRecord[] 472864277f4aSdan ** buffer of the new object is large enough to hold any record that 472964277f4aSdan ** may be generated by combining the input records. */ 47305d607a6eSdan nByte = sizeof(SessionChange) + pExist->nRecord + nRec; 47312d77d80aSdrh pNew = (SessionChange *)sqlite3_malloc64(nByte); 47325d607a6eSdan if( !pNew ){ 47331756ae10Sdan sqlite3_free(pExist); 47345d607a6eSdan return SQLITE_NOMEM; 47355d607a6eSdan } 47365d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 47375d607a6eSdan pNew->bIndirect = (bIndirect && pExist->bIndirect); 47385d607a6eSdan aCsr = pNew->aRecord = (u8 *)&pNew[1]; 47395d607a6eSdan 4740b08a1efaSdan if( op1==SQLITE_INSERT ){ /* INSERT + UPDATE */ 47415d607a6eSdan u8 *a1 = aRec; 4742b08a1efaSdan assert( op2==SQLITE_UPDATE ); 4743798693b2Sdan pNew->op = SQLITE_INSERT; 4744ef7a6304Sdan if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol); 474564277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1); 4746b08a1efaSdan }else if( op1==SQLITE_DELETE ){ /* DELETE + INSERT */ 4747b08a1efaSdan assert( op2==SQLITE_INSERT ); 4748798693b2Sdan pNew->op = SQLITE_UPDATE; 4749fa29ecc4Sdan if( bPatchset ){ 4750fa29ecc4Sdan memcpy(aCsr, aRec, nRec); 4751fa29ecc4Sdan aCsr += nRec; 4752fa29ecc4Sdan }else{ 475364277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){ 4754b08a1efaSdan sqlite3_free(pNew); 4755b08a1efaSdan pNew = 0; 47565d607a6eSdan } 4757fa29ecc4Sdan } 4758b08a1efaSdan }else if( op2==SQLITE_UPDATE ){ /* UPDATE + UPDATE */ 475964277f4aSdan u8 *a1 = aExist; 47605d607a6eSdan u8 *a2 = aRec; 4761cfec7eeeSdan assert( op1==SQLITE_UPDATE ); 476264277f4aSdan if( bPatchset==0 ){ 4763ef7a6304Sdan sessionSkipRecord(&a1, pTab->nCol); 4764ef7a6304Sdan sessionSkipRecord(&a2, pTab->nCol); 476564277f4aSdan } 4766798693b2Sdan pNew->op = SQLITE_UPDATE; 476764277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){ 47685d607a6eSdan sqlite3_free(pNew); 47695d607a6eSdan pNew = 0; 47705d607a6eSdan } 4771b08a1efaSdan }else{ /* UPDATE + DELETE */ 4772b08a1efaSdan assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE ); 4773798693b2Sdan pNew->op = SQLITE_DELETE; 477464277f4aSdan if( bPatchset ){ 477564277f4aSdan memcpy(aCsr, aRec, nRec); 477664277f4aSdan aCsr += nRec; 477764277f4aSdan }else{ 477864277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist); 477964277f4aSdan } 47805d607a6eSdan } 47815d607a6eSdan 47825d607a6eSdan if( pNew ){ 47834f528042Sdan pNew->nRecord = (int)(aCsr - pNew->aRecord); 47845d607a6eSdan } 47855d607a6eSdan sqlite3_free(pExist); 47865d607a6eSdan } 47875d607a6eSdan } 47885d607a6eSdan 47895d607a6eSdan *ppNew = pNew; 479024a0c453Sdan return rc; 47915d607a6eSdan } 47925d607a6eSdan 479377fc1d5bSdan /* 47945898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as 47955898ad69Sdan ** the first argument to the changegroup hash tables. 479677fc1d5bSdan */ 479716228167Sdan static int sessionChangesetToHash( 4798cbf6d2d2Sdan sqlite3_changeset_iter *pIter, /* Iterator to read from */ 4799c0a499eaSdan sqlite3_changegroup *pGrp, /* Changegroup object to add changeset to */ 4800c0a499eaSdan int bRebase /* True if hash table is for rebasing */ 48015d607a6eSdan ){ 48025d607a6eSdan u8 *aRec; 48035d607a6eSdan int nRec; 4804cbf6d2d2Sdan int rc = SQLITE_OK; 48055d607a6eSdan SessionTable *pTab = 0; 48065d607a6eSdan 4807c0a499eaSdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){ 48085d607a6eSdan const char *zNew; 48095d607a6eSdan int nCol; 48105d607a6eSdan int op; 48115d607a6eSdan int iHash; 48125d607a6eSdan int bIndirect; 48135d607a6eSdan SessionChange *pChange; 48145d607a6eSdan SessionChange *pExist = 0; 48155d607a6eSdan SessionChange **pp; 48165d607a6eSdan 48175898ad69Sdan if( pGrp->pList==0 ){ 48185898ad69Sdan pGrp->bPatch = pIter->bPatchset; 48195898ad69Sdan }else if( pIter->bPatchset!=pGrp->bPatch ){ 48205898ad69Sdan rc = SQLITE_ERROR; 48215898ad69Sdan break; 48225898ad69Sdan } 48235898ad69Sdan 48245d607a6eSdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect); 4825ef7a6304Sdan if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){ 48265d607a6eSdan /* Search the list for a matching table */ 48274f528042Sdan int nNew = (int)strlen(zNew); 4828f29123b5Sdan u8 *abPK; 4829f29123b5Sdan 4830f29123b5Sdan sqlite3changeset_pk(pIter, &abPK, 0); 48315898ad69Sdan for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){ 48325d607a6eSdan if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break; 48335d607a6eSdan } 48345d607a6eSdan if( !pTab ){ 48356c39e6a8Sdan SessionTable **ppTab; 48366c39e6a8Sdan 48372d77d80aSdrh pTab = sqlite3_malloc64(sizeof(SessionTable) + nCol + nNew+1); 4838f29123b5Sdan if( !pTab ){ 4839f29123b5Sdan rc = SQLITE_NOMEM; 4840f29123b5Sdan break; 4841f29123b5Sdan } 48425d607a6eSdan memset(pTab, 0, sizeof(SessionTable)); 4843f29123b5Sdan pTab->nCol = nCol; 4844ef7a6304Sdan pTab->abPK = (u8*)&pTab[1]; 4845ef7a6304Sdan memcpy(pTab->abPK, abPK, nCol); 4846ef7a6304Sdan pTab->zName = (char*)&pTab->abPK[nCol]; 4847ef7a6304Sdan memcpy(pTab->zName, zNew, nNew+1); 48486c39e6a8Sdan 48496c39e6a8Sdan /* The new object must be linked on to the end of the list, not 48506c39e6a8Sdan ** simply added to the start of it. This is to ensure that the 48516c39e6a8Sdan ** tables within the output of sqlite3changegroup_output() are in 48526c39e6a8Sdan ** the right order. */ 48536c39e6a8Sdan for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext); 48546c39e6a8Sdan *ppTab = pTab; 4855f29123b5Sdan }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){ 4856f29123b5Sdan rc = SQLITE_SCHEMA; 4857f29123b5Sdan break; 48585d607a6eSdan } 48595d607a6eSdan } 48605d607a6eSdan 4861cbf6d2d2Sdan if( sessionGrowHash(pIter->bPatchset, pTab) ){ 48621756ae10Sdan rc = SQLITE_NOMEM; 48631756ae10Sdan break; 48641756ae10Sdan } 486564277f4aSdan iHash = sessionChangeHash( 4866cbf6d2d2Sdan pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange 486764277f4aSdan ); 48685d607a6eSdan 48695d607a6eSdan /* Search for existing entry. If found, remove it from the hash table. 48705d607a6eSdan ** Code below may link it back in. 48715d607a6eSdan */ 48725d607a6eSdan for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){ 487364277f4aSdan int bPkOnly1 = 0; 487464277f4aSdan int bPkOnly2 = 0; 4875cbf6d2d2Sdan if( pIter->bPatchset ){ 487664277f4aSdan bPkOnly1 = (*pp)->op==SQLITE_DELETE; 487764277f4aSdan bPkOnly2 = op==SQLITE_DELETE; 487864277f4aSdan } 487964277f4aSdan if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){ 48805d607a6eSdan pExist = *pp; 48815d607a6eSdan *pp = (*pp)->pNext; 48825d607a6eSdan pTab->nEntry--; 48835d607a6eSdan break; 48845d607a6eSdan } 48855d607a6eSdan } 48865d607a6eSdan 4887c0a499eaSdan rc = sessionChangeMerge(pTab, bRebase, 4888cbf6d2d2Sdan pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange 488964277f4aSdan ); 48905d607a6eSdan if( rc ) break; 48915d607a6eSdan if( pChange ){ 48925d607a6eSdan pChange->pNext = pTab->apChange[iHash]; 48935d607a6eSdan pTab->apChange[iHash] = pChange; 48945d607a6eSdan pTab->nEntry++; 48955d607a6eSdan } 48965d607a6eSdan } 48975d607a6eSdan 4898cbf6d2d2Sdan if( rc==SQLITE_OK ) rc = pIter->rc; 48995d607a6eSdan return rc; 49005d607a6eSdan } 49015d607a6eSdan 49025d607a6eSdan /* 49035898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets) 49045898ad69Sdan ** added to the changegroup object passed as the first argument. 49055d607a6eSdan ** 49065898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the 49075898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming 49085898ad69Sdan ** interfaces. 49095d607a6eSdan ** 49105898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a 49115898ad69Sdan ** buffer containing the output changeset before this function returns. In 49125898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It 49135898ad69Sdan ** is the responsibility of the caller to free the output buffer using 49145898ad69Sdan ** sqlite3_free() when it is no longer required. 49155898ad69Sdan ** 49165898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite 49175898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut) 49185898ad69Sdan ** are both set to 0 before returning. 49195d607a6eSdan */ 49205898ad69Sdan static int sessionChangegroupOutput( 49215898ad69Sdan sqlite3_changegroup *pGrp, 4922cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4923cbf6d2d2Sdan void *pOut, 4924cbf6d2d2Sdan int *pnOut, 4925cbf6d2d2Sdan void **ppOut 49265d607a6eSdan ){ 49275898ad69Sdan int rc = SQLITE_OK; 4928e8fa8c96Sdan SessionBuffer buf = {0, 0, 0}; 49295898ad69Sdan SessionTable *pTab; 4930cbf6d2d2Sdan assert( xOutput==0 || (ppOut==0 && pnOut==0) ); 49315d607a6eSdan 49325d607a6eSdan /* Create the serialized output changeset based on the contents of the 49335898ad69Sdan ** hash tables attached to the SessionTable objects in list p->pList. 49345d607a6eSdan */ 49355898ad69Sdan for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 49365d607a6eSdan int i; 49375d607a6eSdan if( pTab->nEntry==0 ) continue; 49385d607a6eSdan 49395898ad69Sdan sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc); 49405d607a6eSdan for(i=0; i<pTab->nChange; i++){ 49415d607a6eSdan SessionChange *p; 49425d607a6eSdan for(p=pTab->apChange[i]; p; p=p->pNext){ 4943798693b2Sdan sessionAppendByte(&buf, p->op, &rc); 49445d607a6eSdan sessionAppendByte(&buf, p->bIndirect, &rc); 49455d607a6eSdan sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc); 49461f48e67dSdan if( rc==SQLITE_OK && xOutput && buf.nBuf>=sessions_strm_chunk_size ){ 4947cbf6d2d2Sdan rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4948cbf6d2d2Sdan buf.nBuf = 0; 4949cbf6d2d2Sdan } 49505d607a6eSdan } 4951f8248c42Sdan } 4952f8248c42Sdan } 49535d607a6eSdan 49545d607a6eSdan if( rc==SQLITE_OK ){ 4955cbf6d2d2Sdan if( xOutput ){ 4956cbf6d2d2Sdan if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf); 4957cbf6d2d2Sdan }else{ 49585d607a6eSdan *ppOut = buf.aBuf; 49595d607a6eSdan *pnOut = buf.nBuf; 4960cbf6d2d2Sdan buf.aBuf = 0; 49615d607a6eSdan } 49625d607a6eSdan } 4963cbf6d2d2Sdan sqlite3_free(buf.aBuf); 49645d607a6eSdan 49655d607a6eSdan return rc; 49665d607a6eSdan } 49675d607a6eSdan 4968cbf6d2d2Sdan /* 49695898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup. 49705898ad69Sdan */ 49715898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){ 49725898ad69Sdan int rc = SQLITE_OK; /* Return code */ 49735898ad69Sdan sqlite3_changegroup *p; /* New object */ 49745898ad69Sdan p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup)); 49755898ad69Sdan if( p==0 ){ 49765898ad69Sdan rc = SQLITE_NOMEM; 49775898ad69Sdan }else{ 49785898ad69Sdan memset(p, 0, sizeof(sqlite3_changegroup)); 49795898ad69Sdan } 49805898ad69Sdan *pp = p; 49815898ad69Sdan return rc; 49825898ad69Sdan } 49835898ad69Sdan 49845898ad69Sdan /* 49855898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes, 49865898ad69Sdan ** to changeset-group p. 49875898ad69Sdan */ 49885898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){ 49895898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 49905898ad69Sdan int rc; /* Return code */ 49915898ad69Sdan 49925898ad69Sdan rc = sqlite3changeset_start(&pIter, nData, pData); 49935898ad69Sdan if( rc==SQLITE_OK ){ 4994c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 49955898ad69Sdan } 49965898ad69Sdan sqlite3changeset_finalize(pIter); 49975898ad69Sdan return rc; 49985898ad69Sdan } 49995898ad69Sdan 50005898ad69Sdan /* 50015898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation 50025898ad69Sdan ** of all changesets added to the group so far. 50035898ad69Sdan */ 50045898ad69Sdan int sqlite3changegroup_output( 50055898ad69Sdan sqlite3_changegroup *pGrp, 50065898ad69Sdan int *pnData, 50075898ad69Sdan void **ppData 50085898ad69Sdan ){ 50095898ad69Sdan return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData); 50105898ad69Sdan } 50115898ad69Sdan 50125898ad69Sdan /* 50135898ad69Sdan ** Streaming versions of changegroup_add(). 50145898ad69Sdan */ 50155898ad69Sdan int sqlite3changegroup_add_strm( 50165898ad69Sdan sqlite3_changegroup *pGrp, 50175898ad69Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 50185898ad69Sdan void *pIn 50195898ad69Sdan ){ 50205898ad69Sdan sqlite3_changeset_iter *pIter; /* Iterator opened on pData/nData */ 50215898ad69Sdan int rc; /* Return code */ 50225898ad69Sdan 50235898ad69Sdan rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 50245898ad69Sdan if( rc==SQLITE_OK ){ 5025c0a499eaSdan rc = sessionChangesetToHash(pIter, pGrp, 0); 50265898ad69Sdan } 50275898ad69Sdan sqlite3changeset_finalize(pIter); 50285898ad69Sdan return rc; 50295898ad69Sdan } 50305898ad69Sdan 50315898ad69Sdan /* 50325898ad69Sdan ** Streaming versions of changegroup_output(). 50335898ad69Sdan */ 50345898ad69Sdan int sqlite3changegroup_output_strm( 50355898ad69Sdan sqlite3_changegroup *pGrp, 50365898ad69Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 50375898ad69Sdan void *pOut 50385898ad69Sdan ){ 50395898ad69Sdan return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0); 50405898ad69Sdan } 50415898ad69Sdan 50425898ad69Sdan /* 50435898ad69Sdan ** Delete a changegroup object. 50445898ad69Sdan */ 50455898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){ 50465898ad69Sdan if( pGrp ){ 50475898ad69Sdan sessionDeleteTable(pGrp->pList); 50485898ad69Sdan sqlite3_free(pGrp); 50495898ad69Sdan } 50505898ad69Sdan } 50515898ad69Sdan 50525898ad69Sdan /* 5053cbf6d2d2Sdan ** Combine two changesets together. 5054cbf6d2d2Sdan */ 5055cbf6d2d2Sdan int sqlite3changeset_concat( 5056cbf6d2d2Sdan int nLeft, /* Number of bytes in lhs input */ 5057cbf6d2d2Sdan void *pLeft, /* Lhs input changeset */ 5058cbf6d2d2Sdan int nRight /* Number of bytes in rhs input */, 5059cbf6d2d2Sdan void *pRight, /* Rhs input changeset */ 5060cbf6d2d2Sdan int *pnOut, /* OUT: Number of bytes in output changeset */ 5061cbf6d2d2Sdan void **ppOut /* OUT: changeset (left <concat> right) */ 5062cbf6d2d2Sdan ){ 50635898ad69Sdan sqlite3_changegroup *pGrp; 5064cbf6d2d2Sdan int rc; 5065cbf6d2d2Sdan 50665898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 5067cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50685898ad69Sdan rc = sqlite3changegroup_add(pGrp, nLeft, pLeft); 5069cbf6d2d2Sdan } 5070cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50715898ad69Sdan rc = sqlite3changegroup_add(pGrp, nRight, pRight); 5072cbf6d2d2Sdan } 50735898ad69Sdan if( rc==SQLITE_OK ){ 50745898ad69Sdan rc = sqlite3changegroup_output(pGrp, pnOut, ppOut); 50755898ad69Sdan } 50765898ad69Sdan sqlite3changegroup_delete(pGrp); 5077cbf6d2d2Sdan 5078cbf6d2d2Sdan return rc; 5079cbf6d2d2Sdan } 5080cbf6d2d2Sdan 5081cbf6d2d2Sdan /* 5082cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat(). 5083cbf6d2d2Sdan */ 5084f1a08ad8Sdrh int sqlite3changeset_concat_strm( 5085cbf6d2d2Sdan int (*xInputA)(void *pIn, void *pData, int *pnData), 5086cbf6d2d2Sdan void *pInA, 5087cbf6d2d2Sdan int (*xInputB)(void *pIn, void *pData, int *pnData), 5088cbf6d2d2Sdan void *pInB, 5089cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 5090cbf6d2d2Sdan void *pOut 5091cbf6d2d2Sdan ){ 50925898ad69Sdan sqlite3_changegroup *pGrp; 5093cbf6d2d2Sdan int rc; 5094cbf6d2d2Sdan 50955898ad69Sdan rc = sqlite3changegroup_new(&pGrp); 5096cbf6d2d2Sdan if( rc==SQLITE_OK ){ 50975898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA); 5098cbf6d2d2Sdan } 5099cbf6d2d2Sdan if( rc==SQLITE_OK ){ 51005898ad69Sdan rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB); 5101cbf6d2d2Sdan } 51025898ad69Sdan if( rc==SQLITE_OK ){ 51035898ad69Sdan rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut); 51045898ad69Sdan } 51055898ad69Sdan sqlite3changegroup_delete(pGrp); 5106cbf6d2d2Sdan 5107cbf6d2d2Sdan return rc; 5108cbf6d2d2Sdan } 5109cbf6d2d2Sdan 51103fa5463cSdan /* 51113fa5463cSdan ** Changeset rebaser handle. 51123fa5463cSdan */ 5113c0a499eaSdan struct sqlite3_rebaser { 5114c0a499eaSdan sqlite3_changegroup grp; /* Hash table */ 5115c0a499eaSdan }; 5116c0a499eaSdan 5117c0a499eaSdan /* 5118c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol 5119c0a499eaSdan ** fields in size. This function appends an nCol sessions module 5120b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for 5121b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[]. 5122c0a499eaSdan */ 5123c0a499eaSdan static void sessionAppendRecordMerge( 5124b880a7b1Sdan SessionBuffer *pBuf, /* Buffer to append to */ 5125b880a7b1Sdan int nCol, /* Number of columns in each record */ 5126b880a7b1Sdan u8 *a1, int n1, /* Record 1 */ 5127b880a7b1Sdan u8 *a2, int n2, /* Record 2 */ 5128b880a7b1Sdan int *pRc /* IN/OUT: error code */ 5129c0a499eaSdan ){ 5130c0a499eaSdan sessionBufferGrow(pBuf, n1+n2, pRc); 5131c0a499eaSdan if( *pRc==SQLITE_OK ){ 5132c0a499eaSdan int i; 5133c0a499eaSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5134c0a499eaSdan for(i=0; i<nCol; i++){ 5135c0a499eaSdan int nn1 = sessionSerialLen(a1); 5136c0a499eaSdan int nn2 = sessionSerialLen(a2); 513724a0c453Sdan if( *a1==0 || *a1==0xFF ){ 5138c0a499eaSdan memcpy(pOut, a2, nn2); 5139c0a499eaSdan pOut += nn2; 5140c0a499eaSdan }else{ 5141c0a499eaSdan memcpy(pOut, a1, nn1); 5142c0a499eaSdan pOut += nn1; 5143c0a499eaSdan } 5144f1b40e83Sdan a1 += nn1; 5145f1b40e83Sdan a2 += nn2; 5146c0a499eaSdan } 5147f1b40e83Sdan 5148f1b40e83Sdan pBuf->nBuf = pOut-pBuf->aBuf; 5149f1b40e83Sdan assert( pBuf->nBuf<=pBuf->nAlloc ); 5150c0a499eaSdan } 5151c0a499eaSdan } 5152c0a499eaSdan 51533fa5463cSdan /* 51543fa5463cSdan ** This function is called when rebasing a local UPDATE change against one 51553fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current 51563fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single 51573fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer 51583fa5463cSdan ** pBuf. 51593fa5463cSdan ** 51603fa5463cSdan ** Rebasing the UPDATE involves: 51613fa5463cSdan ** 51623fa5463cSdan ** * Removing any changes to fields for which the corresponding field 51633fa5463cSdan ** in the rebase buffer is set to "replaced" (type 0xFF). If this 51643fa5463cSdan ** means the UPDATE change updates no fields, nothing is appended 51653fa5463cSdan ** to the output buffer. 51663fa5463cSdan ** 51673fa5463cSdan ** * For each field modified by the local change for which the 51683fa5463cSdan ** corresponding field in the rebase buffer is not "undefined" (0x00) 51693fa5463cSdan ** or "replaced" (0xFF), the old.* value is replaced by the value 51703fa5463cSdan ** in the rebase buffer. 51713fa5463cSdan */ 5172f01d3a7eSdan static void sessionAppendPartialUpdate( 517324a0c453Sdan SessionBuffer *pBuf, /* Append record here */ 517424a0c453Sdan sqlite3_changeset_iter *pIter, /* Iterator pointed at local change */ 517524a0c453Sdan u8 *aRec, int nRec, /* Local change */ 517624a0c453Sdan u8 *aChange, int nChange, /* Record to rebase against */ 517724a0c453Sdan int *pRc /* IN/OUT: Return Code */ 5178f01d3a7eSdan ){ 5179f01d3a7eSdan sessionBufferGrow(pBuf, 2+nRec+nChange, pRc); 5180f01d3a7eSdan if( *pRc==SQLITE_OK ){ 5181f01d3a7eSdan int bData = 0; 5182f01d3a7eSdan u8 *pOut = &pBuf->aBuf[pBuf->nBuf]; 5183f01d3a7eSdan int i; 5184f01d3a7eSdan u8 *a1 = aRec; 5185f01d3a7eSdan u8 *a2 = aChange; 5186f01d3a7eSdan 5187f01d3a7eSdan *pOut++ = SQLITE_UPDATE; 5188f01d3a7eSdan *pOut++ = pIter->bIndirect; 5189f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5190f01d3a7eSdan int n1 = sessionSerialLen(a1); 5191f01d3a7eSdan int n2 = sessionSerialLen(a2); 5192f01d3a7eSdan if( pIter->abPK[i] || a2[0]==0 ){ 5193f01d3a7eSdan if( !pIter->abPK[i] ) bData = 1; 5194f01d3a7eSdan memcpy(pOut, a1, n1); 5195f01d3a7eSdan pOut += n1; 519624a0c453Sdan }else if( a2[0]!=0xFF ){ 519724a0c453Sdan bData = 1; 519824a0c453Sdan memcpy(pOut, a2, n2); 519924a0c453Sdan pOut += n2; 5200f01d3a7eSdan }else{ 5201f01d3a7eSdan *pOut++ = '\0'; 5202f01d3a7eSdan } 5203f01d3a7eSdan a1 += n1; 5204f01d3a7eSdan a2 += n2; 5205f01d3a7eSdan } 5206f01d3a7eSdan if( bData ){ 5207f01d3a7eSdan a2 = aChange; 5208f01d3a7eSdan for(i=0; i<pIter->nCol; i++){ 5209f01d3a7eSdan int n1 = sessionSerialLen(a1); 5210f01d3a7eSdan int n2 = sessionSerialLen(a2); 521124a0c453Sdan if( pIter->abPK[i] || a2[0]!=0xFF ){ 5212f01d3a7eSdan memcpy(pOut, a1, n1); 5213f01d3a7eSdan pOut += n1; 5214f01d3a7eSdan }else{ 5215f01d3a7eSdan *pOut++ = '\0'; 5216f01d3a7eSdan } 5217f01d3a7eSdan a1 += n1; 5218f01d3a7eSdan a2 += n2; 5219f01d3a7eSdan } 5220f01d3a7eSdan pBuf->nBuf = (pOut - pBuf->aBuf); 5221f01d3a7eSdan } 5222f01d3a7eSdan } 5223f01d3a7eSdan } 5224f01d3a7eSdan 52253fa5463cSdan /* 52263fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases 52273fa5463cSdan ** that changeset according to the current configuration of the rebaser 52283fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput 52293fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking 52303fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL, 52313fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset 52323fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of 52333fa5463cSdan ** the buffer in bytes. It is the responsibility of the caller to eventually 52343fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free(). 52353fa5463cSdan ** 52363fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and 52373fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before 52383fa5463cSdan ** returning. 52393fa5463cSdan */ 5240c0a499eaSdan static int sessionRebase( 5241c0a499eaSdan sqlite3_rebaser *p, /* Rebaser hash table */ 5242c0a499eaSdan sqlite3_changeset_iter *pIter, /* Input data */ 5243c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5244c0a499eaSdan void *pOut, /* Context for xOutput callback */ 5245c0a499eaSdan int *pnOut, /* OUT: Number of bytes in output changeset */ 5246c0a499eaSdan void **ppOut /* OUT: Inverse of pChangeset */ 5247c0a499eaSdan ){ 5248c0a499eaSdan int rc = SQLITE_OK; 5249c0a499eaSdan u8 *aRec = 0; 5250c0a499eaSdan int nRec = 0; 5251c0a499eaSdan int bNew = 0; 5252c0a499eaSdan SessionTable *pTab = 0; 5253c0a499eaSdan SessionBuffer sOut = {0,0,0}; 5254c0a499eaSdan 5255f1b40e83Sdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){ 5256c0a499eaSdan SessionChange *pChange = 0; 5257f01d3a7eSdan int bDone = 0; 5258c0a499eaSdan 5259c0a499eaSdan if( bNew ){ 5260c0a499eaSdan const char *zTab = pIter->zTab; 5261c0a499eaSdan for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){ 5262c0a499eaSdan if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break; 5263c0a499eaSdan } 5264c0a499eaSdan bNew = 0; 5265c0a499eaSdan 526695ccb6dcSdan /* A patchset may not be rebased */ 526795ccb6dcSdan if( pIter->bPatchset ){ 526895ccb6dcSdan rc = SQLITE_ERROR; 526995ccb6dcSdan } 527095ccb6dcSdan 5271c0a499eaSdan /* Append a table header to the output for this new table */ 5272c0a499eaSdan sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc); 5273c0a499eaSdan sessionAppendVarint(&sOut, pIter->nCol, &rc); 5274c0a499eaSdan sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc); 52752c42c34dSmistachkin sessionAppendBlob(&sOut,(u8*)pIter->zTab,(int)strlen(pIter->zTab)+1,&rc); 5276c0a499eaSdan } 5277c0a499eaSdan 52788cb83beeSdan if( pTab && rc==SQLITE_OK ){ 52798cb83beeSdan int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange); 5280c0a499eaSdan 5281c0a499eaSdan for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){ 52828cb83beeSdan if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){ 5283c0a499eaSdan break; 5284c0a499eaSdan } 5285c0a499eaSdan } 5286c0a499eaSdan } 5287c0a499eaSdan 5288c0a499eaSdan if( pChange ){ 5289c0a499eaSdan assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT ); 5290f01d3a7eSdan switch( pIter->op ){ 5291f01d3a7eSdan case SQLITE_INSERT: 5292f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5293f01d3a7eSdan bDone = 1; 5294f01d3a7eSdan if( pChange->bIndirect==0 ){ 5295f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_UPDATE, &rc); 5296f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5297f01d3a7eSdan sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc); 5298f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5299f01d3a7eSdan } 5300f01d3a7eSdan } 5301f01d3a7eSdan break; 5302f01d3a7eSdan 5303f01d3a7eSdan case SQLITE_UPDATE: 5304f01d3a7eSdan bDone = 1; 5305f01d3a7eSdan if( pChange->op==SQLITE_DELETE ){ 5306f01d3a7eSdan if( pChange->bIndirect==0 ){ 5307f01d3a7eSdan u8 *pCsr = aRec; 5308f01d3a7eSdan sessionSkipRecord(&pCsr, pIter->nCol); 5309f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_INSERT, &rc); 5310f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5311b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5312f01d3a7eSdan pCsr, nRec-(pCsr-aRec), 5313f01d3a7eSdan pChange->aRecord, pChange->nRecord, &rc 5314f01d3a7eSdan ); 5315f01d3a7eSdan } 5316f01d3a7eSdan }else{ 5317f01d3a7eSdan sessionAppendPartialUpdate(&sOut, pIter, 5318f01d3a7eSdan aRec, nRec, pChange->aRecord, pChange->nRecord, &rc 5319f01d3a7eSdan ); 5320f01d3a7eSdan } 5321f01d3a7eSdan break; 5322f01d3a7eSdan 5323f01d3a7eSdan default: 5324f01d3a7eSdan assert( pIter->op==SQLITE_DELETE ); 5325f01d3a7eSdan bDone = 1; 5326f01d3a7eSdan if( pChange->op==SQLITE_INSERT ){ 5327f01d3a7eSdan sessionAppendByte(&sOut, SQLITE_DELETE, &rc); 5328f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5329b880a7b1Sdan sessionAppendRecordMerge(&sOut, pIter->nCol, 5330f01d3a7eSdan pChange->aRecord, pChange->nRecord, aRec, nRec, &rc 5331f01d3a7eSdan ); 5332f01d3a7eSdan } 5333f01d3a7eSdan break; 5334f01d3a7eSdan } 5335f01d3a7eSdan } 5336f01d3a7eSdan 5337f01d3a7eSdan if( bDone==0 ){ 5338f01d3a7eSdan sessionAppendByte(&sOut, pIter->op, &rc); 5339f01d3a7eSdan sessionAppendByte(&sOut, pIter->bIndirect, &rc); 5340f01d3a7eSdan sessionAppendBlob(&sOut, aRec, nRec, &rc); 5341f01d3a7eSdan } 53421f48e67dSdan if( rc==SQLITE_OK && xOutput && sOut.nBuf>sessions_strm_chunk_size ){ 5343c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5344c0a499eaSdan sOut.nBuf = 0; 5345c0a499eaSdan } 5346c0a499eaSdan if( rc ) break; 5347c0a499eaSdan } 5348c0a499eaSdan 5349c0a499eaSdan if( rc!=SQLITE_OK ){ 5350c0a499eaSdan sqlite3_free(sOut.aBuf); 5351c0a499eaSdan memset(&sOut, 0, sizeof(sOut)); 5352c0a499eaSdan } 5353c0a499eaSdan 5354c0a499eaSdan if( rc==SQLITE_OK ){ 5355c0a499eaSdan if( xOutput ){ 5356c0a499eaSdan if( sOut.nBuf>0 ){ 5357c0a499eaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 5358c0a499eaSdan } 5359c0a499eaSdan }else{ 5360c0a499eaSdan *ppOut = (void*)sOut.aBuf; 5361c0a499eaSdan *pnOut = sOut.nBuf; 5362c0a499eaSdan sOut.aBuf = 0; 5363c0a499eaSdan } 5364c0a499eaSdan } 5365c0a499eaSdan sqlite3_free(sOut.aBuf); 5366c0a499eaSdan return rc; 5367c0a499eaSdan } 5368c0a499eaSdan 5369c0a499eaSdan /* 5370c0a499eaSdan ** Create a new rebaser object. 5371c0a499eaSdan */ 5372c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){ 5373c0a499eaSdan int rc = SQLITE_OK; 5374c0a499eaSdan sqlite3_rebaser *pNew; 5375c0a499eaSdan 5376c0a499eaSdan pNew = sqlite3_malloc(sizeof(sqlite3_rebaser)); 5377c0a499eaSdan if( pNew==0 ){ 5378c0a499eaSdan rc = SQLITE_NOMEM; 5379f1b40e83Sdan }else{ 5380f1b40e83Sdan memset(pNew, 0, sizeof(sqlite3_rebaser)); 5381c0a499eaSdan } 5382c0a499eaSdan *ppNew = pNew; 5383c0a499eaSdan return rc; 5384c0a499eaSdan } 5385c0a499eaSdan 5386c0a499eaSdan /* 5387c0a499eaSdan ** Call this one or more times to configure a rebaser. 5388c0a499eaSdan */ 5389c0a499eaSdan int sqlite3rebaser_configure( 5390c0a499eaSdan sqlite3_rebaser *p, 5391c0a499eaSdan int nRebase, const void *pRebase 5392c0a499eaSdan ){ 5393c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator opened on pData/nData */ 5394c0a499eaSdan int rc; /* Return code */ 5395c0a499eaSdan rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase); 5396c0a499eaSdan if( rc==SQLITE_OK ){ 5397c0a499eaSdan rc = sessionChangesetToHash(pIter, &p->grp, 1); 5398c0a499eaSdan } 5399c0a499eaSdan sqlite3changeset_finalize(pIter); 5400c0a499eaSdan return rc; 5401c0a499eaSdan } 5402c0a499eaSdan 5403c0a499eaSdan /* 5404c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5405c0a499eaSdan */ 5406c0a499eaSdan int sqlite3rebaser_rebase( 5407c0a499eaSdan sqlite3_rebaser *p, 5408c0a499eaSdan int nIn, const void *pIn, 5409c0a499eaSdan int *pnOut, void **ppOut 5410c0a499eaSdan ){ 5411c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5412c0a499eaSdan int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn); 5413c0a499eaSdan 5414c0a499eaSdan if( rc==SQLITE_OK ){ 5415c0a499eaSdan rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut); 5416c0a499eaSdan sqlite3changeset_finalize(pIter); 5417c0a499eaSdan } 5418c0a499eaSdan 5419c0a499eaSdan return rc; 5420c0a499eaSdan } 5421c0a499eaSdan 5422c0a499eaSdan /* 5423c0a499eaSdan ** Rebase a changeset according to current rebaser configuration 5424c0a499eaSdan */ 5425c0a499eaSdan int sqlite3rebaser_rebase_strm( 5426c0a499eaSdan sqlite3_rebaser *p, 5427c0a499eaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 5428c0a499eaSdan void *pIn, 5429c0a499eaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 5430c0a499eaSdan void *pOut 5431c0a499eaSdan ){ 5432c0a499eaSdan sqlite3_changeset_iter *pIter = 0; /* Iterator to skip through input */ 5433c0a499eaSdan int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 5434c0a499eaSdan 5435c0a499eaSdan if( rc==SQLITE_OK ){ 5436c0a499eaSdan rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0); 5437c0a499eaSdan sqlite3changeset_finalize(pIter); 5438c0a499eaSdan } 5439c0a499eaSdan 5440c0a499eaSdan return rc; 5441c0a499eaSdan } 5442c0a499eaSdan 5443c0a499eaSdan /* 5444c0a499eaSdan ** Destroy a rebaser object 5445c0a499eaSdan */ 5446f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){ 5447c0a499eaSdan if( p ){ 5448c0a499eaSdan sessionDeleteTable(p->grp.pList); 5449c0a499eaSdan sqlite3_free(p); 5450c0a499eaSdan } 5451c0a499eaSdan } 5452c0a499eaSdan 54531f48e67dSdan /* 54541f48e67dSdan ** Global configuration 54551f48e67dSdan */ 54561f48e67dSdan int sqlite3session_config(int op, void *pArg){ 54571f48e67dSdan int rc = SQLITE_OK; 54581f48e67dSdan switch( op ){ 54591f48e67dSdan case SQLITE_SESSION_CONFIG_STRMSIZE: { 54601f48e67dSdan int *pInt = (int*)pArg; 54611f48e67dSdan if( *pInt>0 ){ 54621f48e67dSdan sessions_strm_chunk_size = *pInt; 54631f48e67dSdan } 54641f48e67dSdan *pInt = sessions_strm_chunk_size; 54651f48e67dSdan break; 54661f48e67dSdan } 54671f48e67dSdan default: 54681f48e67dSdan rc = SQLITE_MISUSE; 54691f48e67dSdan break; 54701f48e67dSdan } 54711f48e67dSdan return rc; 54721f48e67dSdan } 54731f48e67dSdan 54749b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */ 5475