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 28296c7658Sdan /* 29296c7658Sdan ** Session handle structure. 30296c7658Sdan */ 314fccf43aSdan struct sqlite3_session { 324fccf43aSdan sqlite3 *db; /* Database handle session is attached to */ 334fccf43aSdan char *zDb; /* Name of database session is attached to */ 34296c7658Sdan int bEnable; /* True if currently recording */ 35b4480e94Sdan int bIndirect; /* True if all changes are indirect */ 36ff4d0f41Sdan int bAutoAttach; /* True to auto-attach tables */ 374fccf43aSdan int rc; /* Non-zero if an error has occurred */ 387531a5a3Sdan void *pFilterCtx; /* First argument to pass to xTableFilter */ 397531a5a3Sdan int (*xTableFilter)(void *pCtx, const char *zTab); 404fccf43aSdan sqlite3_session *pNext; /* Next session object on same db. */ 414fccf43aSdan SessionTable *pTable; /* List of attached tables */ 424fccf43aSdan }; 434fccf43aSdan 444fccf43aSdan /* 45ef7a6304Sdan ** Instances of this structure are used to build strings or binary records. 46ef7a6304Sdan */ 47ef7a6304Sdan struct SessionBuffer { 48ef7a6304Sdan u8 *aBuf; /* Pointer to changeset buffer */ 49ef7a6304Sdan int nBuf; /* Size of buffer aBuf */ 50ef7a6304Sdan int nAlloc; /* Size of allocation containing aBuf */ 51ef7a6304Sdan }; 52ef7a6304Sdan 53ef7a6304Sdan /* 5416228167Sdan ** An object of this type is used internally as an abstraction for 5516228167Sdan ** input data. Input data may be supplied either as a single large buffer 5616228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g. 57f1a08ad8Sdrh ** sqlite3changeset_start_strm()). 58ef7a6304Sdan */ 59ef7a6304Sdan struct SessionInput { 604757c658Sdan int iNext; /* Offset in aData[] of next change */ 614757c658Sdan u8 *aData; /* Pointer to buffer containing changeset */ 624757c658Sdan int nData; /* Number of bytes in aData */ 634757c658Sdan 64ef7a6304Sdan SessionBuffer buf; /* Current read buffer */ 65ef7a6304Sdan int (*xInput)(void*, void*, int*); /* Input stream call (or NULL) */ 66ef7a6304Sdan void *pIn; /* First argument to xInput */ 67ef7a6304Sdan int bEof; /* Set to true after xInput finished */ 68ef7a6304Sdan }; 69ef7a6304Sdan 70ef7a6304Sdan /* 71296c7658Sdan ** Structure for changeset iterators. 72296c7658Sdan */ 73296c7658Sdan struct sqlite3_changeset_iter { 74ef7a6304Sdan SessionInput in; /* Input buffer or stream */ 75ef7a6304Sdan SessionBuffer tblhdr; /* Buffer to hold apValue/zTab/abPK/ */ 7673b3c055Sdan int bPatchset; /* True if this is a patchset */ 77296c7658Sdan int rc; /* Iterator error code */ 78296c7658Sdan sqlite3_stmt *pConflict; /* Points to conflicting row, if any */ 79296c7658Sdan char *zTab; /* Current table */ 80296c7658Sdan int nCol; /* Number of columns in zTab */ 81296c7658Sdan int op; /* Current operation */ 82b4480e94Sdan int bIndirect; /* True if current change was indirect */ 83244593c8Sdan u8 *abPK; /* Primary key array */ 84296c7658Sdan sqlite3_value **apValue; /* old.* and new.* values */ 85296c7658Sdan }; 86296c7658Sdan 87296c7658Sdan /* 884fccf43aSdan ** Each session object maintains a set of the following structures, one 894fccf43aSdan ** for each table the session object is monitoring. The structures are 904fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable. 914fccf43aSdan ** 924fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have 934fccf43aSdan ** been modified in any way since the session object was attached to the 944fccf43aSdan ** table. 954fccf43aSdan ** 964fccf43aSdan ** The data associated with each hash-table entry is a structure containing 974fccf43aSdan ** a subset of the initial values that the modified row contained at the 984fccf43aSdan ** start of the session. Or no initial values if the row was inserted. 994fccf43aSdan */ 1004fccf43aSdan struct SessionTable { 1014fccf43aSdan SessionTable *pNext; 1024fccf43aSdan char *zName; /* Local name of table */ 1034fccf43aSdan int nCol; /* Number of columns in table zName */ 104e8d5648eSdan const char **azCol; /* Column names */ 105e8d5648eSdan u8 *abPK; /* Array of primary key flags */ 106296c7658Sdan int nEntry; /* Total number of entries in hash table */ 1074fccf43aSdan int nChange; /* Size of apChange[] array */ 1084fccf43aSdan SessionChange **apChange; /* Hash table buckets */ 1094fccf43aSdan }; 1104fccf43aSdan 1114fccf43aSdan /* 1124fccf43aSdan ** RECORD FORMAT: 1134fccf43aSdan ** 1144fccf43aSdan ** The following record format is similar to (but not compatible with) that 1154fccf43aSdan ** used in SQLite database files. This format is used as part of the 1164fccf43aSdan ** change-set binary format, and so must be architecture independent. 1174fccf43aSdan ** 1184fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained - 1194fccf43aSdan ** there is no separation of header and data. Each field begins with a 1204fccf43aSdan ** single byte describing its type, as follows: 1214fccf43aSdan ** 1224fccf43aSdan ** 0x00: Undefined value. 1234fccf43aSdan ** 0x01: Integer value. 1244fccf43aSdan ** 0x02: Real value. 1254fccf43aSdan ** 0x03: Text value. 1264fccf43aSdan ** 0x04: Blob value. 1274fccf43aSdan ** 0x05: SQL NULL value. 1284fccf43aSdan ** 1294fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT 1304fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists 1314fccf43aSdan ** only of the single type byte. For other types of values, the type byte 1324fccf43aSdan ** is followed by: 1334fccf43aSdan ** 1344fccf43aSdan ** Text values: 1354fccf43aSdan ** A varint containing the number of bytes in the value (encoded using 1364fccf43aSdan ** UTF-8). Followed by a buffer containing the UTF-8 representation 1374fccf43aSdan ** of the text value. There is no nul terminator. 1384fccf43aSdan ** 1394fccf43aSdan ** Blob values: 1404fccf43aSdan ** A varint containing the number of bytes in the value, followed by 1414fccf43aSdan ** a buffer containing the value itself. 1424fccf43aSdan ** 1434fccf43aSdan ** Integer values: 1444fccf43aSdan ** An 8-byte big-endian integer value. 1454fccf43aSdan ** 1464fccf43aSdan ** Real values: 1474fccf43aSdan ** An 8-byte big-endian IEEE 754-2008 real value. 1484fccf43aSdan ** 1494fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite 1504fccf43aSdan ** record format. 1514fccf43aSdan ** 1524fccf43aSdan ** CHANGESET FORMAT: 1534fccf43aSdan ** 1544fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on 1554fccf43aSdan ** one or more tables. Operations on a single table are grouped together, 1564fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all 1574fccf43aSdan ** mixed together). 1584fccf43aSdan ** 1594fccf43aSdan ** Each group of changes begins with a table header: 1604fccf43aSdan ** 1614fccf43aSdan ** 1 byte: Constant 0x54 (capital 'T') 162*730bb805Sdan ** Varint: Number of columns in the table. 16373b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 1644fccf43aSdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 1654fccf43aSdan ** 1664fccf43aSdan ** Followed by one or more changes to the table. 1674fccf43aSdan ** 1684fccf43aSdan ** 1 byte: Either SQLITE_INSERT, UPDATE or DELETE. 1695d607a6eSdan ** 1 byte: The "indirect-change" flag. 1704fccf43aSdan ** old.* record: (delete and update only) 1714fccf43aSdan ** new.* record: (insert and update only) 17273b3c055Sdan ** 173*730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the 174*730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of 175*730bb805Sdan ** columns in the table. The i'th field of each record is associated with 176*730bb805Sdan ** the i'th column of the table, counting from left to right in the order 177*730bb805Sdan ** in which columns were declared in the CREATE TABLE statement. 178*730bb805Sdan ** 179*730bb805Sdan ** The new.* record that is part of each INSERT change contains the values 180*730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each 181*730bb805Sdan ** DELETE change contains the values that made up the row that was deleted 182*730bb805Sdan ** from the database. In the changeset format, the records that are part 183*730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00) 184*730bb805Sdan ** fields. 185*730bb805Sdan ** 186*730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields 187*730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are 188*730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields 189*730bb805Sdan ** are set to the values that made up the row before the UPDATE that the 190*730bb805Sdan ** change records took place. Within the new.* record, fields associated 191*730bb805Sdan ** with table columns modified by the UPDATE change contain the new 192*730bb805Sdan ** values. Fields associated with table columns that are not modified 193*730bb805Sdan ** are set to "undefined". 194*730bb805Sdan ** 19573b3c055Sdan ** PATCHSET FORMAT: 19673b3c055Sdan ** 19773b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset, 198*730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict 199*730bb805Sdan ** resolution is required when applying the changeset. 20073b3c055Sdan ** 20173b3c055Sdan ** Each group of changes begins with a table header: 20273b3c055Sdan ** 20373b3c055Sdan ** 1 byte: Constant 0x50 (capital 'P') 204*730bb805Sdan ** Varint: Number of columns in the table. 20573b3c055Sdan ** nCol bytes: 0x01 for PK columns, 0x00 otherwise. 20673b3c055Sdan ** N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated. 20773b3c055Sdan ** 20873b3c055Sdan ** Followed by one or more changes to the table. 20973b3c055Sdan ** 21073b3c055Sdan ** 1 byte: Either SQLITE_INSERT, UPDATE or DELETE. 21173b3c055Sdan ** 1 byte: The "indirect-change" flag. 212*730bb805Sdan ** single record: (PK fields for DELETE, PK and modified fields for UPDATE, 213*730bb805Sdan ** full record for INSERT). 214*730bb805Sdan ** 215*730bb805Sdan ** As in the changeset format, each field of the single record that is part 216*730bb805Sdan ** of a patchset change is associated with the correspondingly positioned 217*730bb805Sdan ** table column, counting from left to right within the CREATE TABLE 218*730bb805Sdan ** statement. 219*730bb805Sdan ** 220*730bb805Sdan ** For a DELETE change, all fields within the record except those associated 221*730bb805Sdan ** with PRIMARY KEY columns are set to "undefined". The PRIMARY KEY fields 222*730bb805Sdan ** contain the values identifying the row to delete. 223*730bb805Sdan ** 224*730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY 225*730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined". 226*730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update, 227*730bb805Sdan ** and fields associated with modified columns contain the new column values. 228*730bb805Sdan ** 229*730bb805Sdan ** The records associated with INSERT changes are in the same format as for 230*730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT 231*730bb805Sdan ** change to contain a field set to "undefined". 2324fccf43aSdan */ 2334fccf43aSdan 2344fccf43aSdan /* 2354fccf43aSdan ** For each row modified during a session, there exists a single instance of 2364fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table. 2374fccf43aSdan */ 2384fccf43aSdan struct SessionChange { 239798693b2Sdan int op; /* One of UPDATE, DELETE, INSERT */ 240b4480e94Sdan int bIndirect; /* True if this change is "indirect" */ 2414fccf43aSdan int nRecord; /* Number of bytes in buffer aRecord[] */ 2424fccf43aSdan u8 *aRecord; /* Buffer containing old.* record */ 2434fccf43aSdan SessionChange *pNext; /* For hash-table collisions */ 2444fccf43aSdan }; 2454fccf43aSdan 246296c7658Sdan /* 247296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the 248296c7658Sdan ** number of bytes written. 249296c7658Sdan */ 250296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){ 251296c7658Sdan return putVarint32(aBuf, iVal); 2524fccf43aSdan } 2534fccf43aSdan 254296c7658Sdan /* 255296c7658Sdan ** Return the number of bytes required to store value iVal as a varint. 256296c7658Sdan */ 257296c7658Sdan static int sessionVarintLen(int iVal){ 258296c7658Sdan return sqlite3VarintLen(iVal); 259296c7658Sdan } 260296c7658Sdan 261296c7658Sdan /* 262296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of 263296c7658Sdan ** bytes read. 264296c7658Sdan */ 2654fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){ 266296c7658Sdan return getVarint32(aBuf, *piVal); 2674fccf43aSdan } 2684fccf43aSdan 269296c7658Sdan /* 270296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return 271296c7658Sdan ** the value read. 272296c7658Sdan */ 2734fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){ 2744fccf43aSdan return (((sqlite3_int64)aRec[0]) << 56) 2754fccf43aSdan + (((sqlite3_int64)aRec[1]) << 48) 2764fccf43aSdan + (((sqlite3_int64)aRec[2]) << 40) 2774fccf43aSdan + (((sqlite3_int64)aRec[3]) << 32) 2784fccf43aSdan + (((sqlite3_int64)aRec[4]) << 24) 2794fccf43aSdan + (((sqlite3_int64)aRec[5]) << 16) 2804fccf43aSdan + (((sqlite3_int64)aRec[6]) << 8) 2814fccf43aSdan + (((sqlite3_int64)aRec[7]) << 0); 2824fccf43aSdan } 2834fccf43aSdan 2844fccf43aSdan /* 285296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[]. 286296c7658Sdan */ 287296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){ 288296c7658Sdan aBuf[0] = (i>>56) & 0xFF; 289296c7658Sdan aBuf[1] = (i>>48) & 0xFF; 290296c7658Sdan aBuf[2] = (i>>40) & 0xFF; 291296c7658Sdan aBuf[3] = (i>>32) & 0xFF; 292296c7658Sdan aBuf[4] = (i>>24) & 0xFF; 293296c7658Sdan aBuf[5] = (i>>16) & 0xFF; 294296c7658Sdan aBuf[6] = (i>> 8) & 0xFF; 295296c7658Sdan aBuf[7] = (i>> 0) & 0xFF; 296296c7658Sdan } 297296c7658Sdan 298296c7658Sdan /* 2994fccf43aSdan ** This function is used to serialize the contents of value pValue (see 3004fccf43aSdan ** comment titled "RECORD FORMAT" above). 3014fccf43aSdan ** 3024fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to 3034fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before 3044fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is 3054fccf43aSdan ** set *pnWrite. 3064fccf43aSdan ** 3074fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs 3084fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16)) 3094fccf43aSdan ** SQLITE_NOMEM is returned. 3104fccf43aSdan */ 3114fccf43aSdan static int sessionSerializeValue( 3124fccf43aSdan u8 *aBuf, /* If non-NULL, write serialized value here */ 3134fccf43aSdan sqlite3_value *pValue, /* Value to serialize */ 3144fccf43aSdan int *pnWrite /* IN/OUT: Increment by bytes written */ 3154fccf43aSdan ){ 316296c7658Sdan int nByte; /* Size of serialized value in bytes */ 3174fccf43aSdan 31880fe2d93Sdan if( pValue ){ 31980fe2d93Sdan int eType; /* Value type (SQLITE_NULL, TEXT etc.) */ 32080fe2d93Sdan 3214fccf43aSdan eType = sqlite3_value_type(pValue); 3224fccf43aSdan if( aBuf ) aBuf[0] = eType; 3234fccf43aSdan 3244fccf43aSdan switch( eType ){ 3254fccf43aSdan case SQLITE_NULL: 3264fccf43aSdan nByte = 1; 3274fccf43aSdan break; 3284fccf43aSdan 3294fccf43aSdan case SQLITE_INTEGER: 3304fccf43aSdan case SQLITE_FLOAT: 3314fccf43aSdan if( aBuf ){ 3324fccf43aSdan /* TODO: SQLite does something special to deal with mixed-endian 3334fccf43aSdan ** floating point values (e.g. ARM7). This code probably should 3344fccf43aSdan ** too. */ 3354fccf43aSdan u64 i; 3364fccf43aSdan if( eType==SQLITE_INTEGER ){ 3374fccf43aSdan i = (u64)sqlite3_value_int64(pValue); 3384fccf43aSdan }else{ 3394fccf43aSdan double r; 3404fccf43aSdan assert( sizeof(double)==8 && sizeof(u64)==8 ); 3414fccf43aSdan r = sqlite3_value_double(pValue); 3424fccf43aSdan memcpy(&i, &r, 8); 3434fccf43aSdan } 344296c7658Sdan sessionPutI64(&aBuf[1], i); 3454fccf43aSdan } 3464fccf43aSdan nByte = 9; 3474fccf43aSdan break; 3484fccf43aSdan 3494e895da1Sdan default: { 35080fe2d93Sdan u8 *z; 35180fe2d93Sdan int n; 35280fe2d93Sdan int nVarint; 35380fe2d93Sdan 3544e895da1Sdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 35580fe2d93Sdan if( eType==SQLITE_TEXT ){ 35680fe2d93Sdan z = (u8 *)sqlite3_value_text(pValue); 35780fe2d93Sdan }else{ 35880fe2d93Sdan z = (u8 *)sqlite3_value_blob(pValue); 35980fe2d93Sdan } 36080fe2d93Sdan n = sqlite3_value_bytes(pValue); 3613cc89d95Sdan if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 36280fe2d93Sdan nVarint = sessionVarintLen(n); 36380fe2d93Sdan 3644fccf43aSdan if( aBuf ){ 3654fccf43aSdan sessionVarintPut(&aBuf[1], n); 3664fccf43aSdan memcpy(&aBuf[nVarint + 1], eType==SQLITE_TEXT ? 3674fccf43aSdan sqlite3_value_text(pValue) : sqlite3_value_blob(pValue), n 3684fccf43aSdan ); 3694fccf43aSdan } 3704fccf43aSdan 3714fccf43aSdan nByte = 1 + nVarint + n; 3724fccf43aSdan break; 3734fccf43aSdan } 3744fccf43aSdan } 37580fe2d93Sdan }else{ 37680fe2d93Sdan nByte = 1; 37780fe2d93Sdan if( aBuf ) aBuf[0] = '\0'; 37880fe2d93Sdan } 3794fccf43aSdan 380fa122adaSdan if( pnWrite ) *pnWrite += nByte; 3814fccf43aSdan return SQLITE_OK; 3824fccf43aSdan } 3834fccf43aSdan 384fa122adaSdan 385798693b2Sdan /* 386798693b2Sdan ** This macro is used to calculate hash key values for data structures. In 387798693b2Sdan ** order to use this macro, the entire data structure must be represented 388798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value 389798693b2Sdan ** for a data structure represented as three such integers, the macro may 390798693b2Sdan ** then be used as follows: 391798693b2Sdan ** 392798693b2Sdan ** int hash_key_value; 393798693b2Sdan ** hash_key_value = HASH_APPEND(0, <value 1>); 394798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 2>); 395798693b2Sdan ** hash_key_value = HASH_APPEND(hash_key_value, <value 3>); 396798693b2Sdan ** 397798693b2Sdan ** In practice, the data structures this macro is used for are the primary 398798693b2Sdan ** key values of modified rows. 399798693b2Sdan */ 4004131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add) 401798693b2Sdan 402798693b2Sdan /* 403798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the 404798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 405798693b2Sdan */ 4064131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){ 407e8d5648eSdan h = HASH_APPEND(h, i & 0xFFFFFFFF); 408e8d5648eSdan return HASH_APPEND(h, (i>>32)&0xFFFFFFFF); 409e8d5648eSdan } 410798693b2Sdan 411798693b2Sdan /* 412798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to 413798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value. 414798693b2Sdan */ 4154131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){ 416e8d5648eSdan int i; 417e8d5648eSdan for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]); 418e8d5648eSdan return h; 419e8d5648eSdan } 420e8d5648eSdan 4214fccf43aSdan /* 422798693b2Sdan ** Append the hash of the data type passed as the second argument to the 423798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value. 424798693b2Sdan */ 425798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){ 426798693b2Sdan return HASH_APPEND(h, eType); 427798693b2Sdan } 428798693b2Sdan 429798693b2Sdan /* 4304131639cSdan ** This function may only be called from within a pre-update callback. 4314131639cSdan ** It calculates a hash based on the primary key values of the old.* or 432798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to 433798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL 434798693b2Sdan ** values, *pbNullPK is set to true before returning. 435798693b2Sdan ** 436798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values 437798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned 438798693b2Sdan ** and the output variables are set as described above. 4394fccf43aSdan */ 440798693b2Sdan static int sessionPreupdateHash( 441e8d5648eSdan sqlite3 *db, /* Database handle */ 442e8d5648eSdan SessionTable *pTab, /* Session table handle */ 443e8d5648eSdan int bNew, /* True to hash the new.* PK */ 44427453faeSdan int *piHash, /* OUT: Hash value */ 445798693b2Sdan int *pbNullPK /* OUT: True if there are NULL values in PK */ 446e8d5648eSdan ){ 4474131639cSdan unsigned int h = 0; /* Hash value to return */ 4484131639cSdan int i; /* Used to iterate through columns */ 449e8d5648eSdan 45027453faeSdan assert( *pbNullPK==0 ); 451e8d5648eSdan assert( pTab->nCol==sqlite3_preupdate_count(db) ); 452e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 453e8d5648eSdan if( pTab->abPK[i] ){ 454e8d5648eSdan int rc; 455e8d5648eSdan int eType; 456e8d5648eSdan sqlite3_value *pVal; 457e8d5648eSdan 458e8d5648eSdan if( bNew ){ 459e8d5648eSdan rc = sqlite3_preupdate_new(db, i, &pVal); 460e8d5648eSdan }else{ 461e8d5648eSdan rc = sqlite3_preupdate_old(db, i, &pVal); 462e8d5648eSdan } 46312ca0b56Sdan if( rc!=SQLITE_OK ) return rc; 464e8d5648eSdan 465e8d5648eSdan eType = sqlite3_value_type(pVal); 466798693b2Sdan h = sessionHashAppendType(h, eType); 4676734007dSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 468e8d5648eSdan i64 iVal; 469e8d5648eSdan if( eType==SQLITE_INTEGER ){ 470e8d5648eSdan iVal = sqlite3_value_int64(pVal); 471e8d5648eSdan }else{ 472e8d5648eSdan double rVal = sqlite3_value_double(pVal); 473e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 474e8d5648eSdan memcpy(&iVal, &rVal, 8); 475e8d5648eSdan } 476e8d5648eSdan h = sessionHashAppendI64(h, iVal); 4776734007dSdan }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 4786734007dSdan const u8 *z; 4793cc89d95Sdan int n; 4806734007dSdan if( eType==SQLITE_TEXT ){ 4816734007dSdan z = (const u8 *)sqlite3_value_text(pVal); 4826734007dSdan }else{ 4836734007dSdan z = (const u8 *)sqlite3_value_blob(pVal); 484e8d5648eSdan } 4853cc89d95Sdan n = sqlite3_value_bytes(pVal); 4863cc89d95Sdan if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM; 4873cc89d95Sdan h = sessionHashAppendBlob(h, n, z); 4886734007dSdan }else{ 48927453faeSdan assert( eType==SQLITE_NULL ); 49027453faeSdan *pbNullPK = 1; 491e8d5648eSdan } 492e8d5648eSdan } 493e8d5648eSdan } 494e8d5648eSdan 495e8d5648eSdan *piHash = (h % pTab->nChange); 496e8d5648eSdan return SQLITE_OK; 497e8d5648eSdan } 498e8d5648eSdan 4994131639cSdan /* 5006cda207fSdan ** The buffer that the argument points to contains a serialized SQL value. 5016cda207fSdan ** Return the number of bytes of space occupied by the value (including 5026cda207fSdan ** the type byte). 5036cda207fSdan */ 5046cda207fSdan static int sessionSerialLen(u8 *a){ 5056cda207fSdan int e = *a; 5066cda207fSdan int n; 5076cda207fSdan if( e==0 ) return 1; 5086cda207fSdan if( e==SQLITE_NULL ) return 1; 5096cda207fSdan if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9; 5106cda207fSdan return sessionVarintGet(&a[1], &n) + 1 + n; 5116cda207fSdan } 5126cda207fSdan 5136cda207fSdan /* 5145d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a 515798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys 5164131639cSdan ** calculated by this function are compatible with those calculated by 5174131639cSdan ** sessionPreupdateHash(). 51864277f4aSdan ** 51964277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from 52064277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely. 5214131639cSdan */ 5224131639cSdan static unsigned int sessionChangeHash( 5234131639cSdan SessionTable *pTab, /* Table handle */ 52464277f4aSdan int bPkOnly, /* Record consists of PK fields only */ 5255d607a6eSdan u8 *aRecord, /* Change record */ 5264131639cSdan int nBucket /* Assume this many buckets in hash table */ 527e8d5648eSdan ){ 5284131639cSdan unsigned int h = 0; /* Value to return */ 5294131639cSdan int i; /* Used to iterate through columns */ 5305d607a6eSdan u8 *a = aRecord; /* Used to iterate through change record */ 531e8d5648eSdan 532e8d5648eSdan for(i=0; i<pTab->nCol; i++){ 5336cda207fSdan int eType = *a; 534e8d5648eSdan int isPK = pTab->abPK[i]; 53564277f4aSdan if( bPkOnly && isPK==0 ) continue; 536e8d5648eSdan 53727453faeSdan /* It is not possible for eType to be SQLITE_NULL here. The session 53827453faeSdan ** module does not record changes for rows with NULL values stored in 53927453faeSdan ** primary key columns. */ 54027453faeSdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 54127453faeSdan || eType==SQLITE_TEXT || eType==SQLITE_BLOB 5426cda207fSdan || eType==SQLITE_NULL || eType==0 54327453faeSdan ); 5446cda207fSdan assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) ); 54527453faeSdan 5466cda207fSdan if( isPK ){ 5476cda207fSdan a++; 548798693b2Sdan h = sessionHashAppendType(h, eType); 54927453faeSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 5506734007dSdan h = sessionHashAppendI64(h, sessionGetI64(a)); 551e8d5648eSdan a += 8; 55227453faeSdan }else{ 553e8d5648eSdan int n; 554e8d5648eSdan a += sessionVarintGet(a, &n); 5556734007dSdan h = sessionHashAppendBlob(h, n, a); 556e8d5648eSdan a += n; 557e8d5648eSdan } 5586cda207fSdan }else{ 5596cda207fSdan a += sessionSerialLen(a); 5606cda207fSdan } 561e8d5648eSdan } 562e8d5648eSdan return (h % nBucket); 563e8d5648eSdan } 564e8d5648eSdan 565798693b2Sdan /* 566798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab. 567798693b2Sdan ** This function returns true if the two records apply to the same row (i.e. 568798693b2Sdan ** have the same values stored in the primary key columns), or false 569798693b2Sdan ** otherwise. 570798693b2Sdan */ 5715d607a6eSdan static int sessionChangeEqual( 572798693b2Sdan SessionTable *pTab, /* Table used for PK definition */ 573a71d2371Sdan int bLeftPkOnly, /* True if aLeft[] contains PK fields only */ 5745d607a6eSdan u8 *aLeft, /* Change record */ 575a71d2371Sdan int bRightPkOnly, /* True if aRight[] contains PK fields only */ 5765d607a6eSdan u8 *aRight /* Change record */ 5775d607a6eSdan ){ 578798693b2Sdan u8 *a1 = aLeft; /* Cursor to iterate through aLeft */ 579798693b2Sdan u8 *a2 = aRight; /* Cursor to iterate through aRight */ 580798693b2Sdan int iCol; /* Used to iterate through table columns */ 5815d607a6eSdan 582798693b2Sdan for(iCol=0; iCol<pTab->nCol; iCol++){ 5835d607a6eSdan int n1 = sessionSerialLen(a1); 5845d607a6eSdan int n2 = sessionSerialLen(a2); 5855d607a6eSdan 586798693b2Sdan if( pTab->abPK[iCol] && (n1!=n2 || memcmp(a1, a2, n1)) ){ 5875d607a6eSdan return 0; 5885d607a6eSdan } 58964277f4aSdan if( pTab->abPK[iCol] || bLeftPkOnly==0 ) a1 += n1; 59064277f4aSdan if( pTab->abPK[iCol] || bRightPkOnly==0 ) a2 += n2; 5915d607a6eSdan } 5925d607a6eSdan 5935d607a6eSdan return 1; 5945d607a6eSdan } 5955d607a6eSdan 596798693b2Sdan /* 597798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change 598798693b2Sdan ** records with nCol columns. This function "merges" the two records into 599798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is 600798693b2Sdan ** then set to point to one byte after the last byte written before 601798693b2Sdan ** returning. 602798693b2Sdan ** 603798693b2Sdan ** The merging of records is done as follows: For each column, if the 604798693b2Sdan ** aRight record contains a value for the column, copy the value from 605798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither 606798693b2Sdan ** record contains a value for a given column, then neither does the 607798693b2Sdan ** output record. 608798693b2Sdan */ 6095d607a6eSdan static void sessionMergeRecord( 6105d607a6eSdan u8 **paOut, 611798693b2Sdan int nCol, 6125d607a6eSdan u8 *aLeft, 6135d607a6eSdan u8 *aRight 6145d607a6eSdan ){ 615798693b2Sdan u8 *a1 = aLeft; /* Cursor used to iterate through aLeft */ 616798693b2Sdan u8 *a2 = aRight; /* Cursor used to iterate through aRight */ 617798693b2Sdan u8 *aOut = *paOut; /* Output cursor */ 618798693b2Sdan int iCol; /* Used to iterate from 0 to nCol */ 6195d607a6eSdan 620798693b2Sdan for(iCol=0; iCol<nCol; iCol++){ 6215d607a6eSdan int n1 = sessionSerialLen(a1); 6225d607a6eSdan int n2 = sessionSerialLen(a2); 6235d607a6eSdan if( *a2 ){ 6245d607a6eSdan memcpy(aOut, a2, n2); 6255d607a6eSdan aOut += n2; 6265d607a6eSdan }else{ 6275d607a6eSdan memcpy(aOut, a1, n1); 6285d607a6eSdan aOut += n1; 6295d607a6eSdan } 6305d607a6eSdan a1 += n1; 6315d607a6eSdan a2 += n2; 6325d607a6eSdan } 6335d607a6eSdan 6345d607a6eSdan *paOut = aOut; 6355d607a6eSdan } 6365d607a6eSdan 637798693b2Sdan /* 638798693b2Sdan ** This is a helper function used by sessionMergeUpdate(). 639798693b2Sdan ** 640798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value 641798693b2Sdan ** within a change record. Before it returns, both have been advanced so 642798693b2Sdan ** as to point to the next value in the record. 643798693b2Sdan ** 644798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e. 645fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo 646798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the 647798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal 648798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points 649fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words: 650fa29ecc4Sdan ** 651fa29ecc4Sdan ** if( *paTwo is valid ) return *paTwo; 652fa29ecc4Sdan ** return *paOne; 653fa29ecc4Sdan ** 654798693b2Sdan */ 6555d607a6eSdan static u8 *sessionMergeValue( 656798693b2Sdan u8 **paOne, /* IN/OUT: Left-hand buffer pointer */ 657798693b2Sdan u8 **paTwo, /* IN/OUT: Right-hand buffer pointer */ 658798693b2Sdan int *pnVal /* OUT: Bytes in returned value */ 6595d607a6eSdan ){ 6605d607a6eSdan u8 *a1 = *paOne; 6615d607a6eSdan u8 *a2 = *paTwo; 6625d607a6eSdan u8 *pRet = 0; 6635d607a6eSdan int n1; 6645d607a6eSdan 6655d607a6eSdan assert( a1 ); 6665d607a6eSdan if( a2 ){ 6675d607a6eSdan int n2 = sessionSerialLen(a2); 6685d607a6eSdan if( *a2 ){ 6695d607a6eSdan *pnVal = n2; 6705d607a6eSdan pRet = a2; 6715d607a6eSdan } 6725d607a6eSdan *paTwo = &a2[n2]; 6735d607a6eSdan } 6745d607a6eSdan 6755d607a6eSdan n1 = sessionSerialLen(a1); 6765d607a6eSdan if( pRet==0 ){ 6775d607a6eSdan *pnVal = n1; 6785d607a6eSdan pRet = a1; 6795d607a6eSdan } 6805d607a6eSdan *paOne = &a1[n1]; 6815d607a6eSdan 6825d607a6eSdan return pRet; 6835d607a6eSdan } 6845d607a6eSdan 685798693b2Sdan /* 686798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes 687798693b2Sdan ** on the same row. 688798693b2Sdan */ 6895d607a6eSdan static int sessionMergeUpdate( 690798693b2Sdan u8 **paOut, /* IN/OUT: Pointer to output buffer */ 691798693b2Sdan SessionTable *pTab, /* Table change pertains to */ 692a71d2371Sdan int bPatchset, /* True if records are patchset records */ 693798693b2Sdan u8 *aOldRecord1, /* old.* record for first change */ 694798693b2Sdan u8 *aOldRecord2, /* old.* record for second change */ 695798693b2Sdan u8 *aNewRecord1, /* new.* record for first change */ 696798693b2Sdan u8 *aNewRecord2 /* new.* record for second change */ 6975d607a6eSdan ){ 6985d607a6eSdan u8 *aOld1 = aOldRecord1; 6995d607a6eSdan u8 *aOld2 = aOldRecord2; 7005d607a6eSdan u8 *aNew1 = aNewRecord1; 7015d607a6eSdan u8 *aNew2 = aNewRecord2; 7025d607a6eSdan 7035d607a6eSdan u8 *aOut = *paOut; 7045d607a6eSdan int i; 70564277f4aSdan 70664277f4aSdan if( bPatchset==0 ){ 7075d607a6eSdan int bRequired = 0; 7085d607a6eSdan 7095d607a6eSdan assert( aOldRecord1 && aNewRecord1 ); 7105d607a6eSdan 7115d607a6eSdan /* Write the old.* vector first. */ 7125d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7135d607a6eSdan int nOld; 7145d607a6eSdan u8 *aOld; 7155d607a6eSdan int nNew; 7165d607a6eSdan u8 *aNew; 7175d607a6eSdan 7185d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 7195d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 7205d607a6eSdan if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){ 7215d607a6eSdan if( pTab->abPK[i]==0 ) bRequired = 1; 7225d607a6eSdan memcpy(aOut, aOld, nOld); 7235d607a6eSdan aOut += nOld; 7245d607a6eSdan }else{ 7255d607a6eSdan *(aOut++) = '\0'; 7265d607a6eSdan } 7275d607a6eSdan } 7285d607a6eSdan 7295d607a6eSdan if( !bRequired ) return 0; 73064277f4aSdan } 7315d607a6eSdan 7325d607a6eSdan /* Write the new.* vector */ 7335d607a6eSdan aOld1 = aOldRecord1; 7345d607a6eSdan aOld2 = aOldRecord2; 7355d607a6eSdan aNew1 = aNewRecord1; 7365d607a6eSdan aNew2 = aNewRecord2; 7375d607a6eSdan for(i=0; i<pTab->nCol; i++){ 7385d607a6eSdan int nOld; 7395d607a6eSdan u8 *aOld; 7405d607a6eSdan int nNew; 7415d607a6eSdan u8 *aNew; 7425d607a6eSdan 7435d607a6eSdan aOld = sessionMergeValue(&aOld1, &aOld2, &nOld); 7445d607a6eSdan aNew = sessionMergeValue(&aNew1, &aNew2, &nNew); 74564277f4aSdan if( bPatchset==0 74664277f4aSdan && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew))) 74764277f4aSdan ){ 7485d607a6eSdan *(aOut++) = '\0'; 7495d607a6eSdan }else{ 7505d607a6eSdan memcpy(aOut, aNew, nNew); 7515d607a6eSdan aOut += nNew; 7525d607a6eSdan } 7535d607a6eSdan } 7545d607a6eSdan 7555d607a6eSdan *paOut = aOut; 7565d607a6eSdan return 1; 7575d607a6eSdan } 7585d607a6eSdan 75977fc1d5bSdan /* 76077fc1d5bSdan ** This function is only called from within a pre-update-hook callback. 76177fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row 76277fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise 76377fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns 76477fc1d5bSdan ** false. 76577fc1d5bSdan */ 76677fc1d5bSdan static int sessionPreupdateEqual( 76777fc1d5bSdan sqlite3 *db, /* Database handle */ 76877fc1d5bSdan SessionTable *pTab, /* Table associated with change */ 76977fc1d5bSdan SessionChange *pChange, /* Change to compare to */ 77077fc1d5bSdan int op /* Current pre-update operation */ 771e8d5648eSdan ){ 77277fc1d5bSdan int iCol; /* Used to iterate through columns */ 77377fc1d5bSdan u8 *a = pChange->aRecord; /* Cursor used to scan change record */ 774e8d5648eSdan 77577fc1d5bSdan assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE ); 77677fc1d5bSdan for(iCol=0; iCol<pTab->nCol; iCol++){ 77777fc1d5bSdan if( !pTab->abPK[iCol] ){ 778798693b2Sdan a += sessionSerialLen(a); 779e8d5648eSdan }else{ 7806734007dSdan sqlite3_value *pVal; /* Value returned by preupdate_new/old */ 7816734007dSdan int rc; /* Error code from preupdate_new/old */ 782798693b2Sdan int eType = *a++; /* Type of value from change record */ 7836734007dSdan 7846734007dSdan /* The following calls to preupdate_new() and preupdate_old() can not 7856734007dSdan ** fail. This is because they cache their return values, and by the 7866734007dSdan ** time control flows to here they have already been called once from 7876734007dSdan ** within sessionPreupdateHash(). The first two asserts below verify 7886734007dSdan ** this (that the method has already been called). */ 78977fc1d5bSdan if( op==SQLITE_INSERT ){ 7906734007dSdan assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); 79177fc1d5bSdan rc = sqlite3_preupdate_new(db, iCol, &pVal); 792e8d5648eSdan }else{ 7936734007dSdan assert( db->pPreUpdate->pUnpacked ); 79477fc1d5bSdan rc = sqlite3_preupdate_old(db, iCol, &pVal); 795e8d5648eSdan } 7966734007dSdan assert( rc==SQLITE_OK ); 79777fc1d5bSdan if( sqlite3_value_type(pVal)!=eType ) return 0; 798e8d5648eSdan 79912ca0b56Sdan /* A SessionChange object never has a NULL value in a PK column */ 80012ca0b56Sdan assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT 80112ca0b56Sdan || eType==SQLITE_BLOB || eType==SQLITE_TEXT 80212ca0b56Sdan ); 80312ca0b56Sdan 80412ca0b56Sdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 805e8d5648eSdan i64 iVal = sessionGetI64(a); 806e8d5648eSdan a += 8; 807e8d5648eSdan if( eType==SQLITE_INTEGER ){ 80877fc1d5bSdan if( sqlite3_value_int64(pVal)!=iVal ) return 0; 809e8d5648eSdan }else{ 810e8d5648eSdan double rVal; 811e8d5648eSdan assert( sizeof(iVal)==8 && sizeof(rVal)==8 ); 812e8d5648eSdan memcpy(&rVal, &iVal, 8); 81377fc1d5bSdan if( sqlite3_value_double(pVal)!=rVal ) return 0; 814e8d5648eSdan } 81512ca0b56Sdan }else{ 816e8d5648eSdan int n; 817e8d5648eSdan const u8 *z; 818e8d5648eSdan a += sessionVarintGet(a, &n); 81977fc1d5bSdan if( sqlite3_value_bytes(pVal)!=n ) return 0; 82012ca0b56Sdan if( eType==SQLITE_TEXT ){ 82112ca0b56Sdan z = sqlite3_value_text(pVal); 82212ca0b56Sdan }else{ 82312ca0b56Sdan z = sqlite3_value_blob(pVal); 82412ca0b56Sdan } 82577fc1d5bSdan if( memcmp(a, z, n) ) return 0; 826e8d5648eSdan a += n; 827e8d5648eSdan break; 828e8d5648eSdan } 829e8d5648eSdan } 830e8d5648eSdan } 831e8d5648eSdan 83277fc1d5bSdan return 1; 8334fccf43aSdan } 8344fccf43aSdan 8354fccf43aSdan /* 8364fccf43aSdan ** If required, grow the hash table used to store changes on table pTab 8374fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the 8384fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return 8394fccf43aSdan ** SQLITE_OK. 8404fccf43aSdan ** 8414fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In 8424fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway. 8434fccf43aSdan ** Growing the hash table in this case is a performance optimization only, 8444fccf43aSdan ** it is not required for correct operation. 8454fccf43aSdan */ 84664277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){ 8474fccf43aSdan if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){ 8484fccf43aSdan int i; 8494fccf43aSdan SessionChange **apNew; 8504fccf43aSdan int nNew = (pTab->nChange ? pTab->nChange : 128) * 2; 8514fccf43aSdan 8524fccf43aSdan apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew); 8534fccf43aSdan if( apNew==0 ){ 8544fccf43aSdan if( pTab->nChange==0 ){ 8554fccf43aSdan return SQLITE_ERROR; 8564fccf43aSdan } 8574fccf43aSdan return SQLITE_OK; 8584fccf43aSdan } 8594fccf43aSdan memset(apNew, 0, sizeof(SessionChange *) * nNew); 8604fccf43aSdan 8614fccf43aSdan for(i=0; i<pTab->nChange; i++){ 8624fccf43aSdan SessionChange *p; 8634fccf43aSdan SessionChange *pNext; 8644fccf43aSdan for(p=pTab->apChange[i]; p; p=pNext){ 86564277f4aSdan int bPkOnly = (p->op==SQLITE_DELETE && bPatchset); 86664277f4aSdan int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew); 8674fccf43aSdan pNext = p->pNext; 8684fccf43aSdan p->pNext = apNew[iHash]; 8694fccf43aSdan apNew[iHash] = p; 8704fccf43aSdan } 8714fccf43aSdan } 8724fccf43aSdan 8734fccf43aSdan sqlite3_free(pTab->apChange); 8744fccf43aSdan pTab->nChange = nNew; 8754fccf43aSdan pTab->apChange = apNew; 8764fccf43aSdan } 8774fccf43aSdan 8784fccf43aSdan return SQLITE_OK; 8794fccf43aSdan } 8804fccf43aSdan 881296c7658Sdan /* 882e8d5648eSdan ** This function queries the database for the names of the columns of table 883e8d5648eSdan ** zThis, in schema zDb. It is expected that the table has nCol columns. If 884e8d5648eSdan ** not, SQLITE_SCHEMA is returned and none of the output variables are 885e8d5648eSdan ** populated. 886e8d5648eSdan ** 88777fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number 88877fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a 889e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to 890e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not 891e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding 892e8d5648eSdan ** column is part of the primary key. 893e8d5648eSdan ** 894e8d5648eSdan ** For example, if the table is declared as: 895e8d5648eSdan ** 896e8d5648eSdan ** CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z)); 897e8d5648eSdan ** 89877fc1d5bSdan ** Then the four output variables are populated as follows: 899e8d5648eSdan ** 90077fc1d5bSdan ** *pnCol = 4 901e8d5648eSdan ** *pzTab = "tbl1" 902e8d5648eSdan ** *pazCol = {"w", "x", "y", "z"} 903e8d5648eSdan ** *pabPK = {1, 0, 0, 1} 904e8d5648eSdan ** 905e8d5648eSdan ** All returned buffers are part of the same single allocation, which must 906e8d5648eSdan ** be freed using sqlite3_free() by the caller. If pazCol was not NULL, then 907e8d5648eSdan ** pointer *pazCol should be freed to release all memory. Otherwise, pointer 908e8d5648eSdan ** *pabPK. It is illegal for both pazCol and pabPK to be NULL. 909e8d5648eSdan */ 910e8d5648eSdan static int sessionTableInfo( 911e8d5648eSdan sqlite3 *db, /* Database connection */ 912e8d5648eSdan const char *zDb, /* Name of attached database (e.g. "main") */ 913e8d5648eSdan const char *zThis, /* Table name */ 914ca62ad57Sdan int *pnCol, /* OUT: number of columns */ 915e8d5648eSdan const char **pzTab, /* OUT: Copy of zThis */ 916e8d5648eSdan const char ***pazCol, /* OUT: Array of column names for table */ 917e8d5648eSdan u8 **pabPK /* OUT: Array of booleans - true for PK col */ 918e8d5648eSdan ){ 919e8d5648eSdan char *zPragma; 920e8d5648eSdan sqlite3_stmt *pStmt; 921e8d5648eSdan int rc; 922e8d5648eSdan int nByte; 923e8d5648eSdan int nDbCol = 0; 924e8d5648eSdan int nThis; 925e8d5648eSdan int i; 926e8d5648eSdan u8 *pAlloc; 927db04571cSdan char **azCol = 0; 928e8d5648eSdan u8 *abPK; 929e8d5648eSdan 930db04571cSdan assert( pazCol && pabPK ); 931e8d5648eSdan 932cfdbde21Sdrh nThis = sqlite3Strlen30(zThis); 933e8d5648eSdan zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis); 934e8d5648eSdan if( !zPragma ) return SQLITE_NOMEM; 935e8d5648eSdan 936e8d5648eSdan rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0); 937e8d5648eSdan sqlite3_free(zPragma); 938e8d5648eSdan if( rc!=SQLITE_OK ) return rc; 939e8d5648eSdan 940e8d5648eSdan nByte = nThis + 1; 941e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 942e8d5648eSdan nByte += sqlite3_column_bytes(pStmt, 1); 943e8d5648eSdan nDbCol++; 944e8d5648eSdan } 945e8d5648eSdan rc = sqlite3_reset(pStmt); 946e8d5648eSdan 947e8d5648eSdan if( rc==SQLITE_OK ){ 948e8d5648eSdan nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1); 949e8d5648eSdan pAlloc = sqlite3_malloc(nByte); 950e8d5648eSdan if( pAlloc==0 ){ 951e8d5648eSdan rc = SQLITE_NOMEM; 952e8d5648eSdan } 953e8d5648eSdan } 954e8d5648eSdan if( rc==SQLITE_OK ){ 955e8d5648eSdan azCol = (char **)pAlloc; 956ca62ad57Sdan pAlloc = (u8 *)&azCol[nDbCol]; 957e8d5648eSdan abPK = (u8 *)pAlloc; 958ca62ad57Sdan pAlloc = &abPK[nDbCol]; 959e8d5648eSdan if( pzTab ){ 960e8d5648eSdan memcpy(pAlloc, zThis, nThis+1); 961e8d5648eSdan *pzTab = (char *)pAlloc; 962e8d5648eSdan pAlloc += nThis+1; 963e8d5648eSdan } 964e8d5648eSdan 965e8d5648eSdan i = 0; 966e8d5648eSdan while( SQLITE_ROW==sqlite3_step(pStmt) ){ 967e8d5648eSdan int nName = sqlite3_column_bytes(pStmt, 1); 968e8d5648eSdan const unsigned char *zName = sqlite3_column_text(pStmt, 1); 969e8d5648eSdan if( zName==0 ) break; 970e8d5648eSdan memcpy(pAlloc, zName, nName+1); 971e8d5648eSdan azCol[i] = (char *)pAlloc; 972e8d5648eSdan pAlloc += nName+1; 973db04571cSdan abPK[i] = sqlite3_column_int(pStmt, 5); 974e8d5648eSdan i++; 975e8d5648eSdan } 976e8d5648eSdan rc = sqlite3_reset(pStmt); 977e8d5648eSdan 978e8d5648eSdan } 979e8d5648eSdan 980e8d5648eSdan /* If successful, populate the output variables. Otherwise, zero them and 981e8d5648eSdan ** free any allocation made. An error code will be returned in this case. 982e8d5648eSdan */ 983e8d5648eSdan if( rc==SQLITE_OK ){ 984db04571cSdan *pazCol = (const char **)azCol; 985db04571cSdan *pabPK = abPK; 986ca62ad57Sdan *pnCol = nDbCol; 987e8d5648eSdan }else{ 988db04571cSdan *pazCol = 0; 989db04571cSdan *pabPK = 0; 990ca62ad57Sdan *pnCol = 0; 991e8d5648eSdan if( pzTab ) *pzTab = 0; 992db04571cSdan sqlite3_free(azCol); 993e8d5648eSdan } 994e8d5648eSdan sqlite3_finalize(pStmt); 995e8d5648eSdan return rc; 996e8d5648eSdan } 997e8d5648eSdan 998e8d5648eSdan /* 999296c7658Sdan ** This function is only called from within a pre-update handler for a 1000296c7658Sdan ** write to table pTab, part of session pSession. If this is the first 1001296c7658Sdan ** write to this table, set the SessionTable.nCol variable to the number 1002296c7658Sdan ** of columns in the table. 1003296c7658Sdan ** 1004296c7658Sdan ** Otherwise, if this is not the first time this table has been written 1005296c7658Sdan ** to, check that the number of columns in the table has not changed. If 1006296c7658Sdan ** it has not, return zero. 1007296c7658Sdan ** 1008296c7658Sdan ** If the number of columns in the table has changed since the last write 1009296c7658Sdan ** was recorded, set the session error-code to SQLITE_SCHEMA and return 1010296c7658Sdan ** non-zero. Users are not allowed to change the number of columns in a table 1011296c7658Sdan ** for which changes are being recorded by the session module. If they do so, 1012296c7658Sdan ** it is an error. 1013296c7658Sdan */ 10144fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){ 10154fccf43aSdan if( pTab->nCol==0 ){ 1016e8d5648eSdan assert( pTab->azCol==0 || pTab->abPK==0 ); 1017e8d5648eSdan pSession->rc = sessionTableInfo(pSession->db, pSession->zDb, 1018ca62ad57Sdan pTab->zName, &pTab->nCol, 0, &pTab->azCol, &pTab->abPK 1019e8d5648eSdan ); 1020ca62ad57Sdan } 1021ca62ad57Sdan if( pSession->rc==SQLITE_OK 1022ca62ad57Sdan && pTab->nCol!=sqlite3_preupdate_count(pSession->db) 1023ca62ad57Sdan ){ 10244fccf43aSdan pSession->rc = SQLITE_SCHEMA; 10254fccf43aSdan } 1026e8d5648eSdan return pSession->rc; 1027e8d5648eSdan } 1028e8d5648eSdan 102977fc1d5bSdan /* 103077fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a 103177fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change 103277fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument. 103377fc1d5bSdan ** 103477fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added 103577fc1d5bSdan ** to the changed-rows hash table associated with table pTab. 103677fc1d5bSdan */ 1037e8d5648eSdan static void sessionPreupdateOneChange( 103877fc1d5bSdan int op, /* One of SQLITE_UPDATE, INSERT, DELETE */ 103977fc1d5bSdan sqlite3_session *pSession, /* Session object pTab is attached to */ 104077fc1d5bSdan SessionTable *pTab /* Table that change applies to */ 1041e8d5648eSdan ){ 1042e8d5648eSdan sqlite3 *db = pSession->db; 1043e8d5648eSdan int iHash; 104427453faeSdan int bNullPk = 0; 1045e8d5648eSdan int rc = SQLITE_OK; 1046e8d5648eSdan 1047e8d5648eSdan if( pSession->rc ) return; 1048e8d5648eSdan 1049e8d5648eSdan /* Load table details if required */ 1050e8d5648eSdan if( sessionInitTable(pSession, pTab) ) return; 1051e8d5648eSdan 1052e8d5648eSdan /* Grow the hash table if required */ 105364277f4aSdan if( sessionGrowHash(0, pTab) ){ 10545d607a6eSdan pSession->rc = SQLITE_NOMEM; 10555d607a6eSdan return; 10565d607a6eSdan } 1057e8d5648eSdan 105880fe2d93Sdan /* Calculate the hash-key for this change. If the primary key of the row 105980fe2d93Sdan ** includes a NULL value, exit early. Such changes are ignored by the 106080fe2d93Sdan ** session module. */ 106127453faeSdan rc = sessionPreupdateHash(db, pTab, op==SQLITE_INSERT, &iHash, &bNullPk); 106280fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 106380fe2d93Sdan 106480fe2d93Sdan if( bNullPk==0 ){ 106580fe2d93Sdan /* Search the hash table for an existing record for this row. */ 1066b4480e94Sdan SessionChange *pC; 10676734007dSdan for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){ 106877fc1d5bSdan if( sessionPreupdateEqual(db, pTab, pC, op) ) break; 1069e8d5648eSdan } 107080fe2d93Sdan 1071e8d5648eSdan if( pC==0 ){ 1072e8d5648eSdan /* Create a new change object containing all the old values (if 1073e8d5648eSdan ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK 1074e8d5648eSdan ** values (if this is an INSERT). */ 1075b4480e94Sdan SessionChange *pChange; /* New change object */ 1076e8d5648eSdan int nByte; /* Number of bytes to allocate */ 1077e8d5648eSdan int i; /* Used to iterate through columns */ 1078e8d5648eSdan 1079b4480e94Sdan assert( rc==SQLITE_OK ); 1080e8d5648eSdan pTab->nEntry++; 1081e8d5648eSdan 1082e8d5648eSdan /* Figure out how large an allocation is required */ 1083e8d5648eSdan nByte = sizeof(SessionChange); 108480fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1085e8d5648eSdan sqlite3_value *p = 0; 1086e8d5648eSdan if( op!=SQLITE_INSERT ){ 108780fe2d93Sdan TESTONLY(int trc = ) sqlite3_preupdate_old(pSession->db, i, &p); 108880fe2d93Sdan assert( trc==SQLITE_OK ); 108980fe2d93Sdan }else if( pTab->abPK[i] ){ 109080fe2d93Sdan TESTONLY(int trc = ) sqlite3_preupdate_new(pSession->db, i, &p); 109180fe2d93Sdan assert( trc==SQLITE_OK ); 1092e8d5648eSdan } 109380fe2d93Sdan 109480fe2d93Sdan /* This may fail if SQLite value p contains a utf-16 string that must 109580fe2d93Sdan ** be converted to utf-8 and an OOM error occurs while doing so. */ 1096e8d5648eSdan rc = sessionSerializeValue(0, p, &nByte); 109780fe2d93Sdan if( rc!=SQLITE_OK ) goto error_out; 1098e8d5648eSdan } 1099e8d5648eSdan 1100e8d5648eSdan /* Allocate the change object */ 1101e8d5648eSdan pChange = (SessionChange *)sqlite3_malloc(nByte); 1102e8d5648eSdan if( !pChange ){ 1103e8d5648eSdan rc = SQLITE_NOMEM; 110480fe2d93Sdan goto error_out; 1105e8d5648eSdan }else{ 1106e8d5648eSdan memset(pChange, 0, sizeof(SessionChange)); 1107e8d5648eSdan pChange->aRecord = (u8 *)&pChange[1]; 1108e8d5648eSdan } 1109e8d5648eSdan 111080fe2d93Sdan /* Populate the change object. None of the preupdate_old(), 111180fe2d93Sdan ** preupdate_new() or SerializeValue() calls below may fail as all 111280fe2d93Sdan ** required values and encodings have already been cached in memory. 111380fe2d93Sdan ** It is not possible for an OOM to occur in this block. */ 1114e8d5648eSdan nByte = 0; 111580fe2d93Sdan for(i=0; i<pTab->nCol; i++){ 1116e8d5648eSdan sqlite3_value *p = 0; 1117e8d5648eSdan if( op!=SQLITE_INSERT ){ 111880fe2d93Sdan sqlite3_preupdate_old(pSession->db, i, &p); 111980fe2d93Sdan }else if( pTab->abPK[i] ){ 112080fe2d93Sdan sqlite3_preupdate_new(pSession->db, i, &p); 1121e8d5648eSdan } 112280fe2d93Sdan sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte); 1123e8d5648eSdan } 112480fe2d93Sdan 112580fe2d93Sdan /* Add the change to the hash-table */ 1126b4480e94Sdan if( pSession->bIndirect || sqlite3_preupdate_depth(pSession->db) ){ 1127b4480e94Sdan pChange->bIndirect = 1; 1128b4480e94Sdan } 112912ca0b56Sdan pChange->nRecord = nByte; 1130798693b2Sdan pChange->op = op; 1131e8d5648eSdan pChange->pNext = pTab->apChange[iHash]; 1132e8d5648eSdan pTab->apChange[iHash] = pChange; 113380fe2d93Sdan 113480fe2d93Sdan }else if( pC->bIndirect ){ 1135b4480e94Sdan /* If the existing change is considered "indirect", but this current 1136b4480e94Sdan ** change is "direct", mark the change object as direct. */ 1137b4480e94Sdan if( sqlite3_preupdate_depth(pSession->db)==0 && pSession->bIndirect==0 ){ 1138b4480e94Sdan pC->bIndirect = 0; 1139b4480e94Sdan } 1140e8d5648eSdan } 11414fccf43aSdan } 114212ca0b56Sdan 114312ca0b56Sdan /* If an error has occurred, mark the session object as failed. */ 114480fe2d93Sdan error_out: 114512ca0b56Sdan if( rc!=SQLITE_OK ){ 114612ca0b56Sdan pSession->rc = rc; 114712ca0b56Sdan } 114827453faeSdan } 11494fccf43aSdan 11504fccf43aSdan /* 11514fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases. 11524fccf43aSdan */ 11534fccf43aSdan static void xPreUpdate( 11544fccf43aSdan void *pCtx, /* Copy of third arg to preupdate_hook() */ 11554fccf43aSdan sqlite3 *db, /* Database handle */ 11564fccf43aSdan int op, /* SQLITE_UPDATE, DELETE or INSERT */ 11574fccf43aSdan char const *zDb, /* Database name */ 11584fccf43aSdan char const *zName, /* Table name */ 11594fccf43aSdan sqlite3_int64 iKey1, /* Rowid of row about to be deleted/updated */ 11604fccf43aSdan sqlite3_int64 iKey2 /* New rowid value (for a rowid UPDATE) */ 11614fccf43aSdan ){ 11624fccf43aSdan sqlite3_session *pSession; 1163cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); 1164cfdbde21Sdrh int nName = sqlite3Strlen30(zName); 11654fccf43aSdan 11664c220252Sdan assert( sqlite3_mutex_held(db->mutex) ); 11674c220252Sdan 11684fccf43aSdan for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){ 11694fccf43aSdan SessionTable *pTab; 1170296c7658Sdan 1171e8d5648eSdan /* If this session is attached to a different database ("main", "temp" 1172e8d5648eSdan ** etc.), or if it is not currently enabled, there is nothing to do. Skip 1173e8d5648eSdan ** to the next session object attached to this database. */ 1174296c7658Sdan if( pSession->bEnable==0 ) continue; 11754fccf43aSdan if( pSession->rc ) continue; 11764fccf43aSdan if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue; 1177296c7658Sdan 1178ff4d0f41Sdan for(pTab=pSession->pTable; pTab || pSession->bAutoAttach; pTab=pTab->pNext){ 1179ff4d0f41Sdan if( !pTab ){ 1180ff4d0f41Sdan /* This branch is taken if table zName has not yet been attached to 1181ff4d0f41Sdan ** this session and the auto-attach flag is set. */ 11827531a5a3Sdan 11837531a5a3Sdan /* If there is a table-filter configured, invoke it. If it returns 0, 11847531a5a3Sdan ** this change will not be recorded. Break out of the loop early in 11857531a5a3Sdan ** this case. */ 11867531a5a3Sdan if( pSession->xTableFilter 11877531a5a3Sdan && pSession->xTableFilter(pSession->pFilterCtx, zName)==0 11887531a5a3Sdan ){ 11897531a5a3Sdan break; 11907531a5a3Sdan } 11917531a5a3Sdan 1192ff4d0f41Sdan pSession->rc = sqlite3session_attach(pSession,zName); 1193245b49b2Sdan if( pSession->rc ) break; 1194ff4d0f41Sdan pTab = pSession->pTable; 1195ff4d0f41Sdan assert( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ); 1196ff4d0f41Sdan } 1197ff4d0f41Sdan 11984fccf43aSdan if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ){ 1199e8d5648eSdan sessionPreupdateOneChange(op, pSession, pTab); 1200e8d5648eSdan if( op==SQLITE_UPDATE ){ 1201e8d5648eSdan sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab); 12024fccf43aSdan } 12034fccf43aSdan break; 12044fccf43aSdan } 12054fccf43aSdan } 12064fccf43aSdan } 1207296c7658Sdan } 12084fccf43aSdan 12094fccf43aSdan /* 12104fccf43aSdan ** Create a session object. This session object will record changes to 12114fccf43aSdan ** database zDb attached to connection db. 12124fccf43aSdan */ 12134fccf43aSdan int sqlite3session_create( 12144fccf43aSdan sqlite3 *db, /* Database handle */ 12154fccf43aSdan const char *zDb, /* Name of db (e.g. "main") */ 12164fccf43aSdan sqlite3_session **ppSession /* OUT: New session object */ 12174fccf43aSdan ){ 1218296c7658Sdan sqlite3_session *pNew; /* Newly allocated session object */ 1219296c7658Sdan sqlite3_session *pOld; /* Session object already attached to db */ 1220cfdbde21Sdrh int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */ 12214fccf43aSdan 1222296c7658Sdan /* Zero the output value in case an error occurs. */ 12234fccf43aSdan *ppSession = 0; 12244fccf43aSdan 12254fccf43aSdan /* Allocate and populate the new session object. */ 12264fccf43aSdan pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1); 12274fccf43aSdan if( !pNew ) return SQLITE_NOMEM; 12284fccf43aSdan memset(pNew, 0, sizeof(sqlite3_session)); 12294fccf43aSdan pNew->db = db; 12304fccf43aSdan pNew->zDb = (char *)&pNew[1]; 1231296c7658Sdan pNew->bEnable = 1; 12324fccf43aSdan memcpy(pNew->zDb, zDb, nDb+1); 12334fccf43aSdan 12344fccf43aSdan /* Add the new session object to the linked list of session objects 12354fccf43aSdan ** attached to database handle $db. Do this under the cover of the db 12364fccf43aSdan ** handle mutex. */ 12374fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 12384fccf43aSdan pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew); 12394fccf43aSdan pNew->pNext = pOld; 12404fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 12414fccf43aSdan 12424fccf43aSdan *ppSession = pNew; 12434fccf43aSdan return SQLITE_OK; 12444fccf43aSdan } 12454fccf43aSdan 124677fc1d5bSdan /* 124777fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents 124877fc1d5bSdan ** of the changed-rows hash tables are also deleted. 124977fc1d5bSdan */ 12501ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){ 12515d607a6eSdan SessionTable *pNext; 12525d607a6eSdan SessionTable *pTab; 12535d607a6eSdan 12545d607a6eSdan for(pTab=pList; pTab; pTab=pNext){ 12555d607a6eSdan int i; 12565d607a6eSdan pNext = pTab->pNext; 12575d607a6eSdan for(i=0; i<pTab->nChange; i++){ 12585d607a6eSdan SessionChange *p; 12595d607a6eSdan SessionChange *pNext; 12605d607a6eSdan for(p=pTab->apChange[i]; p; p=pNext){ 12615d607a6eSdan pNext = p->pNext; 12625d607a6eSdan sqlite3_free(p); 12635d607a6eSdan } 12645d607a6eSdan } 12655d607a6eSdan sqlite3_free((char*)pTab->azCol); /* cast works around VC++ bug */ 12665d607a6eSdan sqlite3_free(pTab->apChange); 12675d607a6eSdan sqlite3_free(pTab); 12685d607a6eSdan } 12695d607a6eSdan } 12705d607a6eSdan 12714fccf43aSdan /* 12724fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create(). 12734fccf43aSdan */ 12744fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){ 12754fccf43aSdan sqlite3 *db = pSession->db; 12764fccf43aSdan sqlite3_session *pHead; 12774fccf43aSdan sqlite3_session **pp; 12784fccf43aSdan 1279296c7658Sdan /* Unlink the session from the linked list of sessions attached to the 1280296c7658Sdan ** database handle. Hold the db mutex while doing so. */ 12814fccf43aSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 12824fccf43aSdan pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0); 12834fccf43aSdan for(pp=&pHead; (*pp)!=pSession; pp=&((*pp)->pNext)); 12844fccf43aSdan *pp = (*pp)->pNext; 12854fccf43aSdan if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void *)pHead); 12864fccf43aSdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 12874fccf43aSdan 1288296c7658Sdan /* Delete all attached table objects. And the contents of their 1289296c7658Sdan ** associated hash-tables. */ 12905d607a6eSdan sessionDeleteTable(pSession->pTable); 12914fccf43aSdan 1292296c7658Sdan /* Free the session object itself. */ 12934fccf43aSdan sqlite3_free(pSession); 12944fccf43aSdan } 12954fccf43aSdan 12964fccf43aSdan /* 12977531a5a3Sdan ** Set a table filter on a Session Object. 12987531a5a3Sdan */ 12997531a5a3Sdan void sqlite3session_table_filter( 13007531a5a3Sdan sqlite3_session *pSession, 13017531a5a3Sdan int(*xFilter)(void*, const char*), 13027531a5a3Sdan void *pCtx /* First argument passed to xFilter */ 13037531a5a3Sdan ){ 13047531a5a3Sdan pSession->bAutoAttach = 1; 13057531a5a3Sdan pSession->pFilterCtx = pCtx; 13067531a5a3Sdan pSession->xTableFilter = xFilter; 13077531a5a3Sdan } 13087531a5a3Sdan 13097531a5a3Sdan /* 13104fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table 13114fccf43aSdan ** while the session object is enabled will be recorded. 13124fccf43aSdan ** 13134fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does 13144fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias) 13154fccf43aSdan ** or not. 13164fccf43aSdan */ 13174fccf43aSdan int sqlite3session_attach( 13184fccf43aSdan sqlite3_session *pSession, /* Session object */ 13194fccf43aSdan const char *zName /* Table name */ 13204fccf43aSdan ){ 1321ff4d0f41Sdan int rc = SQLITE_OK; 1322ff4d0f41Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 1323ff4d0f41Sdan 1324ff4d0f41Sdan if( !zName ){ 1325ff4d0f41Sdan pSession->bAutoAttach = 1; 1326ff4d0f41Sdan }else{ 1327296c7658Sdan SessionTable *pTab; /* New table object (if required) */ 1328296c7658Sdan int nName; /* Number of bytes in string zName */ 13294fccf43aSdan 13304fccf43aSdan /* First search for an existing entry. If one is found, this call is 13314fccf43aSdan ** a no-op. Return early. */ 1332cfdbde21Sdrh nName = sqlite3Strlen30(zName); 13334fccf43aSdan for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){ 13344c220252Sdan if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break; 13354fccf43aSdan } 13364fccf43aSdan 13374c220252Sdan if( !pTab ){ 13384fccf43aSdan /* Allocate new SessionTable object. */ 13394fccf43aSdan pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1); 13404c220252Sdan if( !pTab ){ 13414c220252Sdan rc = SQLITE_NOMEM; 13424c220252Sdan }else{ 13434fccf43aSdan /* Populate the new SessionTable object and link it into the list. */ 13444fccf43aSdan memset(pTab, 0, sizeof(SessionTable)); 13454fccf43aSdan pTab->zName = (char *)&pTab[1]; 13464fccf43aSdan memcpy(pTab->zName, zName, nName+1); 13474fccf43aSdan pTab->pNext = pSession->pTable; 13484fccf43aSdan pSession->pTable = pTab; 13494c220252Sdan } 13504c220252Sdan } 1351ff4d0f41Sdan } 13524fccf43aSdan 13534c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 13544c220252Sdan return rc; 13554fccf43aSdan } 13564fccf43aSdan 1357296c7658Sdan /* 1358296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data. 1359296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is. 1360296c7658Sdan ** 1361296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered, 1362296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero. 1363296c7658Sdan */ 13644fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){ 136580fe2d93Sdan if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){ 13664fccf43aSdan u8 *aNew; 13674fccf43aSdan int nNew = p->nAlloc ? p->nAlloc : 128; 13684fccf43aSdan do { 13694fccf43aSdan nNew = nNew*2; 1370ef7a6304Sdan }while( nNew<(p->nBuf+nByte) ); 13714fccf43aSdan 13724fccf43aSdan aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew); 13734fccf43aSdan if( 0==aNew ){ 13744fccf43aSdan *pRc = SQLITE_NOMEM; 137580fe2d93Sdan }else{ 13764fccf43aSdan p->aBuf = aNew; 13774fccf43aSdan p->nAlloc = nNew; 13784fccf43aSdan } 137980fe2d93Sdan } 138080fe2d93Sdan return (*pRc!=SQLITE_OK); 13814fccf43aSdan } 13824fccf43aSdan 1383296c7658Sdan /* 1384fa122adaSdan ** Append the value passed as the second argument to the buffer passed 1385fa122adaSdan ** as the first. 1386fa122adaSdan ** 1387fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called. 1388fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code 1389fa122adaSdan ** before returning. 1390fa122adaSdan */ 1391fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){ 1392fa122adaSdan int rc = *pRc; 1393fa122adaSdan if( rc==SQLITE_OK ){ 1394fa122adaSdan int nByte = 0; 1395e8fa8c96Sdan rc = sessionSerializeValue(0, pVal, &nByte); 1396fa122adaSdan sessionBufferGrow(p, nByte, &rc); 1397fa122adaSdan if( rc==SQLITE_OK ){ 1398fa122adaSdan rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0); 1399fa122adaSdan p->nBuf += nByte; 1400fa122adaSdan }else{ 1401fa122adaSdan *pRc = rc; 1402fa122adaSdan } 1403fa122adaSdan } 1404fa122adaSdan } 1405fa122adaSdan 1406fa122adaSdan /* 1407296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1408296c7658Sdan ** called. Otherwise, append a single byte to the buffer. 1409296c7658Sdan ** 1410296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1411296c7658Sdan ** returning. 1412296c7658Sdan */ 14134fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){ 141480fe2d93Sdan if( 0==sessionBufferGrow(p, 1, pRc) ){ 14154fccf43aSdan p->aBuf[p->nBuf++] = v; 14164fccf43aSdan } 14174fccf43aSdan } 14184fccf43aSdan 1419296c7658Sdan /* 1420296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1421296c7658Sdan ** called. Otherwise, append a single varint to the buffer. 1422296c7658Sdan ** 1423296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1424296c7658Sdan ** returning. 1425296c7658Sdan */ 1426cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){ 142780fe2d93Sdan if( 0==sessionBufferGrow(p, 9, pRc) ){ 14284fccf43aSdan p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v); 14294fccf43aSdan } 14304fccf43aSdan } 14314fccf43aSdan 1432296c7658Sdan /* 1433296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1434296c7658Sdan ** called. Otherwise, append a blob of data to the buffer. 1435296c7658Sdan ** 1436296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1437296c7658Sdan ** returning. 1438296c7658Sdan */ 14394fccf43aSdan static void sessionAppendBlob( 14404fccf43aSdan SessionBuffer *p, 14414fccf43aSdan const u8 *aBlob, 14424fccf43aSdan int nBlob, 14434fccf43aSdan int *pRc 14444fccf43aSdan ){ 144580fe2d93Sdan if( 0==sessionBufferGrow(p, nBlob, pRc) ){ 14464fccf43aSdan memcpy(&p->aBuf[p->nBuf], aBlob, nBlob); 14474fccf43aSdan p->nBuf += nBlob; 14484fccf43aSdan } 14494fccf43aSdan } 14504fccf43aSdan 1451296c7658Sdan /* 1452296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1453296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string 1454296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer. 1455296c7658Sdan ** 1456296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1457296c7658Sdan ** returning. 1458296c7658Sdan */ 1459d5f0767cSdan static void sessionAppendStr( 1460d5f0767cSdan SessionBuffer *p, 1461d5f0767cSdan const char *zStr, 1462d5f0767cSdan int *pRc 1463d5f0767cSdan ){ 1464cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr); 146580fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1466d5f0767cSdan memcpy(&p->aBuf[p->nBuf], zStr, nStr); 1467d5f0767cSdan p->nBuf += nStr; 1468d5f0767cSdan } 1469d5f0767cSdan } 1470d5f0767cSdan 1471296c7658Sdan /* 1472296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1473296c7658Sdan ** called. Otherwise, append the string representation of integer iVal 1474296c7658Sdan ** to the buffer. No nul-terminator is written. 1475296c7658Sdan ** 1476296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1477296c7658Sdan ** returning. 1478296c7658Sdan */ 1479d5f0767cSdan static void sessionAppendInteger( 1480296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1481296c7658Sdan int iVal, /* Value to write the string rep. of */ 1482296c7658Sdan int *pRc /* IN/OUT: Error code */ 1483d5f0767cSdan ){ 1484d5f0767cSdan char aBuf[24]; 1485d5f0767cSdan sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal); 1486d5f0767cSdan sessionAppendStr(p, aBuf, pRc); 1487d5f0767cSdan } 1488d5f0767cSdan 1489296c7658Sdan /* 1490296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1491296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and 1492296c7658Sdan ** with any embedded quote characters escaped to the buffer. No 1493296c7658Sdan ** nul-terminator byte is written. 1494296c7658Sdan ** 1495296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before 1496296c7658Sdan ** returning. 1497296c7658Sdan */ 1498d5f0767cSdan static void sessionAppendIdent( 1499296c7658Sdan SessionBuffer *p, /* Buffer to a append to */ 1500296c7658Sdan const char *zStr, /* String to quote, escape and append */ 1501296c7658Sdan int *pRc /* IN/OUT: Error code */ 1502d5f0767cSdan ){ 1503cfdbde21Sdrh int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1; 150480fe2d93Sdan if( 0==sessionBufferGrow(p, nStr, pRc) ){ 1505d5f0767cSdan char *zOut = (char *)&p->aBuf[p->nBuf]; 1506d5f0767cSdan const char *zIn = zStr; 1507d5f0767cSdan *zOut++ = '"'; 1508d5f0767cSdan while( *zIn ){ 1509d5f0767cSdan if( *zIn=='"' ) *zOut++ = '"'; 1510d5f0767cSdan *zOut++ = *(zIn++); 1511d5f0767cSdan } 1512d5f0767cSdan *zOut++ = '"'; 1513cfdbde21Sdrh p->nBuf = (int)((u8 *)zOut - p->aBuf); 1514d5f0767cSdan } 1515d5f0767cSdan } 1516d5f0767cSdan 1517296c7658Sdan /* 1518296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is 1519296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored 1520296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points 1521296c7658Sdan ** to to the buffer. 1522296c7658Sdan */ 15234fccf43aSdan static void sessionAppendCol( 1524296c7658Sdan SessionBuffer *p, /* Buffer to append to */ 1525296c7658Sdan sqlite3_stmt *pStmt, /* Handle pointing to row containing value */ 1526296c7658Sdan int iCol, /* Column to read value from */ 1527296c7658Sdan int *pRc /* IN/OUT: Error code */ 15284fccf43aSdan ){ 15294fccf43aSdan if( *pRc==SQLITE_OK ){ 15304fccf43aSdan int eType = sqlite3_column_type(pStmt, iCol); 15314fccf43aSdan sessionAppendByte(p, (u8)eType, pRc); 15324fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 15334fccf43aSdan sqlite3_int64 i; 15344fccf43aSdan u8 aBuf[8]; 15354fccf43aSdan if( eType==SQLITE_INTEGER ){ 15364fccf43aSdan i = sqlite3_column_int64(pStmt, iCol); 15374fccf43aSdan }else{ 15384fccf43aSdan double r = sqlite3_column_double(pStmt, iCol); 15394fccf43aSdan memcpy(&i, &r, 8); 15404fccf43aSdan } 1541296c7658Sdan sessionPutI64(aBuf, i); 15424fccf43aSdan sessionAppendBlob(p, aBuf, 8, pRc); 15434fccf43aSdan } 15444fccf43aSdan if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){ 15456734007dSdan u8 *z; 15463cc89d95Sdan int nByte; 15476734007dSdan if( eType==SQLITE_BLOB ){ 15486734007dSdan z = (u8 *)sqlite3_column_blob(pStmt, iCol); 15496734007dSdan }else{ 15506734007dSdan z = (u8 *)sqlite3_column_text(pStmt, iCol); 15516734007dSdan } 15523cc89d95Sdan nByte = sqlite3_column_bytes(pStmt, iCol); 15533cc89d95Sdan if( z || (eType==SQLITE_BLOB && nByte==0) ){ 15544fccf43aSdan sessionAppendVarint(p, nByte, pRc); 15556734007dSdan sessionAppendBlob(p, z, nByte, pRc); 15566734007dSdan }else{ 15576734007dSdan *pRc = SQLITE_NOMEM; 15586734007dSdan } 15594fccf43aSdan } 15604fccf43aSdan } 15614fccf43aSdan } 15624fccf43aSdan 1563296c7658Sdan /* 1564296c7658Sdan ** 156580fe2d93Sdan ** This function appends an update change to the buffer (see the comments 156680fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change 156780fe2d93Sdan ** consists of: 1568296c7658Sdan ** 1569296c7658Sdan ** 1 byte: SQLITE_UPDATE (0x17) 1570296c7658Sdan ** n bytes: old.* record (see RECORD FORMAT) 1571296c7658Sdan ** m bytes: new.* record (see RECORD FORMAT) 1572296c7658Sdan ** 1573296c7658Sdan ** The SessionChange object passed as the third argument contains the 1574296c7658Sdan ** values that were stored in the row when the session began (the old.* 1575296c7658Sdan ** values). The statement handle passed as the second argument points 1576296c7658Sdan ** at the current version of the row (the new.* values). 1577296c7658Sdan ** 1578296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value 1579296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer. 1580296c7658Sdan ** 1581296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the 1582296c7658Sdan ** original values of any fields that have been modified. The new.* record 1583296c7658Sdan ** contains the new values of only those fields that have been modified. 1584296c7658Sdan */ 158580fe2d93Sdan static int sessionAppendUpdate( 1586296c7658Sdan SessionBuffer *pBuf, /* Buffer to append to */ 158773b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 1588296c7658Sdan sqlite3_stmt *pStmt, /* Statement handle pointing at new row */ 1589296c7658Sdan SessionChange *p, /* Object containing old values */ 159080fe2d93Sdan u8 *abPK /* Boolean array - true for PK columns */ 15914fccf43aSdan ){ 159280fe2d93Sdan int rc = SQLITE_OK; 1593296c7658Sdan SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */ 1594296c7658Sdan int bNoop = 1; /* Set to zero if any values are modified */ 15951f34f8ccSdan int nRewind = pBuf->nBuf; /* Set to zero if any values are modified */ 1596296c7658Sdan int i; /* Used to iterate through columns */ 1597296c7658Sdan u8 *pCsr = p->aRecord; /* Used to iterate through old.* values */ 1598296c7658Sdan 159980fe2d93Sdan sessionAppendByte(pBuf, SQLITE_UPDATE, &rc); 160080fe2d93Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 16014fccf43aSdan for(i=0; i<sqlite3_column_count(pStmt); i++){ 160237f133ecSdan int bChanged = 0; 16034fccf43aSdan int nAdvance; 16044fccf43aSdan int eType = *pCsr; 16054fccf43aSdan switch( eType ){ 16064fccf43aSdan case SQLITE_NULL: 16074fccf43aSdan nAdvance = 1; 16084fccf43aSdan if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){ 160937f133ecSdan bChanged = 1; 16104fccf43aSdan } 16114fccf43aSdan break; 16124fccf43aSdan 16134fccf43aSdan case SQLITE_FLOAT: 16144fccf43aSdan case SQLITE_INTEGER: { 16154fccf43aSdan nAdvance = 9; 16164fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) ){ 16174fccf43aSdan sqlite3_int64 iVal = sessionGetI64(&pCsr[1]); 16184fccf43aSdan if( eType==SQLITE_INTEGER ){ 16194fccf43aSdan if( iVal==sqlite3_column_int64(pStmt, i) ) break; 16204fccf43aSdan }else{ 16214fccf43aSdan double dVal; 16224fccf43aSdan memcpy(&dVal, &iVal, 8); 16234fccf43aSdan if( dVal==sqlite3_column_double(pStmt, i) ) break; 16244fccf43aSdan } 16254fccf43aSdan } 162637f133ecSdan bChanged = 1; 16274fccf43aSdan break; 16284fccf43aSdan } 16294fccf43aSdan 1630e5754eecSdan default: { 16314fccf43aSdan int nByte; 16324fccf43aSdan int nHdr = 1 + sessionVarintGet(&pCsr[1], &nByte); 1633e5754eecSdan assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB ); 16344fccf43aSdan nAdvance = nHdr + nByte; 16354fccf43aSdan if( eType==sqlite3_column_type(pStmt, i) 16364fccf43aSdan && nByte==sqlite3_column_bytes(pStmt, i) 16374fccf43aSdan && 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), nByte) 16384fccf43aSdan ){ 16394fccf43aSdan break; 16404fccf43aSdan } 164137f133ecSdan bChanged = 1; 16424fccf43aSdan } 16434fccf43aSdan } 16444fccf43aSdan 164573b3c055Sdan /* If at least one field has been modified, this is not a no-op. */ 164673b3c055Sdan if( bChanged ) bNoop = 0; 164773b3c055Sdan 164873b3c055Sdan /* Add a field to the old.* record. This is omitted if this modules is 164973b3c055Sdan ** currently generating a patchset. */ 165073b3c055Sdan if( bPatchset==0 ){ 165137f133ecSdan if( bChanged || abPK[i] ){ 165280fe2d93Sdan sessionAppendBlob(pBuf, pCsr, nAdvance, &rc); 16534fccf43aSdan }else{ 165480fe2d93Sdan sessionAppendByte(pBuf, 0, &rc); 165537f133ecSdan } 165673b3c055Sdan } 165737f133ecSdan 165873b3c055Sdan /* Add a field to the new.* record. Or the only record if currently 165973b3c055Sdan ** generating a patchset. */ 166073b3c055Sdan if( bChanged || (bPatchset && abPK[i]) ){ 166180fe2d93Sdan sessionAppendCol(&buf2, pStmt, i, &rc); 166237f133ecSdan }else{ 166380fe2d93Sdan sessionAppendByte(&buf2, 0, &rc); 16644fccf43aSdan } 166537f133ecSdan 16664fccf43aSdan pCsr += nAdvance; 16674fccf43aSdan } 16684fccf43aSdan 16694fccf43aSdan if( bNoop ){ 16701f34f8ccSdan pBuf->nBuf = nRewind; 16714fccf43aSdan }else{ 167280fe2d93Sdan sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc); 16734fccf43aSdan } 16741f34f8ccSdan sqlite3_free(buf2.aBuf); 167580fe2d93Sdan 167680fe2d93Sdan return rc; 1677d5f0767cSdan } 16784fccf43aSdan 1679a71d2371Sdan /* 1680a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use 1681a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset 1682a71d2371Sdan ** format otherwise. 1683a71d2371Sdan */ 168473b3c055Sdan static int sessionAppendDelete( 168573b3c055Sdan SessionBuffer *pBuf, /* Buffer to append to */ 168673b3c055Sdan int bPatchset, /* True for "patchset", 0 for "changeset" */ 168773b3c055Sdan SessionChange *p, /* Object containing old values */ 1688a71d2371Sdan int nCol, /* Number of columns in table */ 168973b3c055Sdan u8 *abPK /* Boolean array - true for PK columns */ 169073b3c055Sdan ){ 169173b3c055Sdan int rc = SQLITE_OK; 169273b3c055Sdan 169373b3c055Sdan sessionAppendByte(pBuf, SQLITE_DELETE, &rc); 169473b3c055Sdan sessionAppendByte(pBuf, p->bIndirect, &rc); 169573b3c055Sdan 169673b3c055Sdan if( bPatchset==0 ){ 169773b3c055Sdan sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc); 169873b3c055Sdan }else{ 169973b3c055Sdan int i; 170073b3c055Sdan u8 *a = p->aRecord; 170173b3c055Sdan for(i=0; i<nCol; i++){ 170273b3c055Sdan u8 *pStart = a; 170373b3c055Sdan int eType = *a++; 170473b3c055Sdan 170573b3c055Sdan switch( eType ){ 170673b3c055Sdan case 0: 170773b3c055Sdan case SQLITE_NULL: 170873b3c055Sdan assert( abPK[i]==0 ); 170973b3c055Sdan break; 171073b3c055Sdan 171173b3c055Sdan case SQLITE_FLOAT: 171273b3c055Sdan case SQLITE_INTEGER: 171373b3c055Sdan a += 8; 171473b3c055Sdan break; 171573b3c055Sdan 171673b3c055Sdan default: { 171773b3c055Sdan int n; 171873b3c055Sdan a += sessionVarintGet(a, &n); 171973b3c055Sdan a += n; 172073b3c055Sdan break; 172173b3c055Sdan } 172273b3c055Sdan } 172373b3c055Sdan if( abPK[i] ){ 1724f5ab08c7Sdrh sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc); 172573b3c055Sdan } 172673b3c055Sdan } 172773b3c055Sdan assert( (a - p->aRecord)==p->nRecord ); 172873b3c055Sdan } 172973b3c055Sdan 173073b3c055Sdan return rc; 173173b3c055Sdan } 173273b3c055Sdan 173377fc1d5bSdan /* 173477fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table 173577fc1d5bSdan ** zTab in database zDb based on its primary key. i.e. 173677fc1d5bSdan ** 173777fc1d5bSdan ** SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ... 173877fc1d5bSdan */ 1739e8d5648eSdan static int sessionSelectStmt( 1740e8d5648eSdan sqlite3 *db, /* Database handle */ 1741d7fb7d24Sdan const char *zDb, /* Database name */ 1742e8d5648eSdan const char *zTab, /* Table name */ 174377fc1d5bSdan int nCol, /* Number of columns in table */ 174477fc1d5bSdan const char **azCol, /* Names of table columns */ 174577fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 174677fc1d5bSdan sqlite3_stmt **ppStmt /* OUT: Prepared SELECT statement */ 1747d5f0767cSdan ){ 1748e8d5648eSdan int rc = SQLITE_OK; 1749d5f0767cSdan int i; 1750e8d5648eSdan const char *zSep = ""; 1751e8d5648eSdan SessionBuffer buf = {0, 0, 0}; 1752d5f0767cSdan 1753e8d5648eSdan sessionAppendStr(&buf, "SELECT * FROM ", &rc); 1754d7fb7d24Sdan sessionAppendIdent(&buf, zDb, &rc); 1755d7fb7d24Sdan sessionAppendStr(&buf, ".", &rc); 1756e8d5648eSdan sessionAppendIdent(&buf, zTab, &rc); 1757e8d5648eSdan sessionAppendStr(&buf, " WHERE ", &rc); 1758e8d5648eSdan for(i=0; i<nCol; i++){ 1759e8d5648eSdan if( abPK[i] ){ 1760e8d5648eSdan sessionAppendStr(&buf, zSep, &rc); 1761e8d5648eSdan sessionAppendIdent(&buf, azCol[i], &rc); 1762e8d5648eSdan sessionAppendStr(&buf, " = ?", &rc); 1763e8d5648eSdan sessionAppendInteger(&buf, i+1, &rc); 1764e8d5648eSdan zSep = " AND "; 1765d5f0767cSdan } 1766d5f0767cSdan } 1767d5f0767cSdan if( rc==SQLITE_OK ){ 1768e8d5648eSdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, ppStmt, 0); 1769d5f0767cSdan } 1770e8d5648eSdan sqlite3_free(buf.aBuf); 1771e8d5648eSdan return rc; 1772d5f0767cSdan } 1773d5f0767cSdan 177477fc1d5bSdan /* 177577fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange 177677fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement 177777fc1d5bSdan ** is as prepared by function sessionSelectStmt(). 177877fc1d5bSdan ** 177977fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite 178077fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise. 178177fc1d5bSdan */ 1782e8d5648eSdan static int sessionSelectBind( 178377fc1d5bSdan sqlite3_stmt *pSelect, /* SELECT from sessionSelectStmt() */ 178477fc1d5bSdan int nCol, /* Number of columns in table */ 178577fc1d5bSdan u8 *abPK, /* PRIMARY KEY array */ 178677fc1d5bSdan SessionChange *pChange /* Change structure */ 1787e8d5648eSdan ){ 1788e8d5648eSdan int i; 1789e8d5648eSdan int rc = SQLITE_OK; 1790e5754eecSdan u8 *a = pChange->aRecord; 1791d5f0767cSdan 1792e8d5648eSdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 1793e8d5648eSdan int eType = *a++; 1794e8d5648eSdan 1795e8d5648eSdan switch( eType ){ 179680fe2d93Sdan case 0: 1797e8d5648eSdan case SQLITE_NULL: 1798e5754eecSdan assert( abPK[i]==0 ); 1799e8d5648eSdan break; 1800e8d5648eSdan 1801e8d5648eSdan case SQLITE_INTEGER: { 1802e8d5648eSdan if( abPK[i] ){ 1803e8d5648eSdan i64 iVal = sessionGetI64(a); 1804e8d5648eSdan rc = sqlite3_bind_int64(pSelect, i+1, iVal); 1805e8d5648eSdan } 1806e8d5648eSdan a += 8; 1807e8d5648eSdan break; 1808d5f0767cSdan } 1809296c7658Sdan 1810e8d5648eSdan case SQLITE_FLOAT: { 1811e8d5648eSdan if( abPK[i] ){ 1812e8d5648eSdan double rVal; 1813e8d5648eSdan i64 iVal = sessionGetI64(a); 1814e8d5648eSdan memcpy(&rVal, &iVal, 8); 18154e895da1Sdan rc = sqlite3_bind_double(pSelect, i+1, rVal); 1816d5f0767cSdan } 1817e8d5648eSdan a += 8; 1818e8d5648eSdan break; 1819e8d5648eSdan } 1820e8d5648eSdan 1821e8d5648eSdan case SQLITE_TEXT: { 1822e8d5648eSdan int n; 1823e8d5648eSdan a += sessionVarintGet(a, &n); 1824e8d5648eSdan if( abPK[i] ){ 1825e8d5648eSdan rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT); 1826e8d5648eSdan } 1827e8d5648eSdan a += n; 1828e8d5648eSdan break; 1829e8d5648eSdan } 1830e8d5648eSdan 1831e5754eecSdan default: { 1832e8d5648eSdan int n; 1833e5754eecSdan assert( eType==SQLITE_BLOB ); 1834e8d5648eSdan a += sessionVarintGet(a, &n); 1835e8d5648eSdan if( abPK[i] ){ 1836e8d5648eSdan rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT); 1837e8d5648eSdan } 1838e8d5648eSdan a += n; 1839e8d5648eSdan break; 1840e8d5648eSdan } 1841e8d5648eSdan } 1842e8d5648eSdan } 1843e8d5648eSdan 1844d5f0767cSdan return rc; 18454fccf43aSdan } 18464fccf43aSdan 184777fc1d5bSdan /* 184877fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it 184977fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary 185077fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an 185177fc1d5bSdan ** SQLite error code before returning. 185277fc1d5bSdan */ 18535d607a6eSdan static void sessionAppendTableHdr( 1854a71d2371Sdan SessionBuffer *pBuf, /* Append header to this buffer */ 1855a71d2371Sdan int bPatchset, /* Use the patchset format if true */ 1856a71d2371Sdan SessionTable *pTab, /* Table object to append header for */ 1857a71d2371Sdan int *pRc /* IN/OUT: Error code */ 18585d607a6eSdan ){ 18595d607a6eSdan /* Write a table header */ 186073b3c055Sdan sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc); 18615d607a6eSdan sessionAppendVarint(pBuf, pTab->nCol, pRc); 18625d607a6eSdan sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc); 18634f528042Sdan sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc); 18645d607a6eSdan } 18655d607a6eSdan 1866a71d2371Sdan /* 1867a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset 1868a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object 1869a71d2371Sdan ** passed as the first argument. 1870a71d2371Sdan ** 1871a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset 1872a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error 1873a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set 1874a71d2371Sdan ** to 0. 1875a71d2371Sdan */ 187673b3c055Sdan int sessionGenerateChangeset( 18774fccf43aSdan sqlite3_session *pSession, /* Session object */ 187873b3c055Sdan int bPatchset, /* True for patchset, false for changeset */ 1879ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 1880ef7a6304Sdan void *pOut, /* First argument for xOutput */ 18814fccf43aSdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 18824fccf43aSdan void **ppChangeset /* OUT: Buffer containing changeset */ 18834fccf43aSdan ){ 1884296c7658Sdan sqlite3 *db = pSession->db; /* Source database handle */ 1885296c7658Sdan SessionTable *pTab; /* Used to iterate through attached tables */ 1886296c7658Sdan SessionBuffer buf = {0,0,0}; /* Buffer in which to accumlate changeset */ 1887296c7658Sdan int rc; /* Return code */ 18884fccf43aSdan 1889ef7a6304Sdan assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) ); 1890ef7a6304Sdan 1891296c7658Sdan /* Zero the output variables in case an error occurs. If this session 1892296c7658Sdan ** object is already in the error state (sqlite3_session.rc != SQLITE_OK), 1893296c7658Sdan ** this call will be a no-op. */ 1894ef7a6304Sdan if( xOutput==0 ){ 18954fccf43aSdan *pnChangeset = 0; 18964fccf43aSdan *ppChangeset = 0; 1897ef7a6304Sdan } 1898e5754eecSdan 1899e5754eecSdan if( pSession->rc ) return pSession->rc; 1900e5754eecSdan rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0); 1901e5754eecSdan if( rc!=SQLITE_OK ) return rc; 1902e5754eecSdan 1903e5754eecSdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 19044fccf43aSdan 19054fccf43aSdan for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 19064fccf43aSdan if( pTab->nEntry ){ 1907d7fb7d24Sdan const char *zName = pTab->zName; 1908a9605b91Sdan int nCol; /* Number of columns in table */ 1909a9605b91Sdan u8 *abPK; /* Primary key array */ 1910a9605b91Sdan const char **azCol = 0; /* Table columns */ 19111f34f8ccSdan int i; /* Used to iterate through hash buckets */ 19121f34f8ccSdan sqlite3_stmt *pSel = 0; /* SELECT statement to query table pTab */ 19131f34f8ccSdan int nRewind = buf.nBuf; /* Initial size of write buffer */ 19141f34f8ccSdan int nNoop; /* Size of buffer after writing tbl header */ 19154fccf43aSdan 1916a9605b91Sdan /* Check the table schema is still Ok. */ 1917a9605b91Sdan rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK); 1918a9605b91Sdan if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){ 1919a9605b91Sdan rc = SQLITE_SCHEMA; 1920a9605b91Sdan } 1921a9605b91Sdan 19224fccf43aSdan /* Write a table header */ 192373b3c055Sdan sessionAppendTableHdr(&buf, bPatchset, pTab, &rc); 19244fccf43aSdan 19254fccf43aSdan /* Build and compile a statement to execute: */ 19264fccf43aSdan if( rc==SQLITE_OK ){ 1927d7fb7d24Sdan rc = sessionSelectStmt( 1928a9605b91Sdan db, pSession->zDb, zName, nCol, azCol, abPK, &pSel); 19294fccf43aSdan } 19304fccf43aSdan 19311f34f8ccSdan nNoop = buf.nBuf; 193212ca0b56Sdan for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){ 1933e8d5648eSdan SessionChange *p; /* Used to iterate through changes */ 1934e8d5648eSdan 19354fccf43aSdan for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){ 1936e5754eecSdan rc = sessionSelectBind(pSel, nCol, abPK, p); 193780fe2d93Sdan if( rc!=SQLITE_OK ) continue; 19381f34f8ccSdan if( sqlite3_step(pSel)==SQLITE_ROW ){ 1939798693b2Sdan if( p->op==SQLITE_INSERT ){ 19404fccf43aSdan int iCol; 19414fccf43aSdan sessionAppendByte(&buf, SQLITE_INSERT, &rc); 1942b4480e94Sdan sessionAppendByte(&buf, p->bIndirect, &rc); 1943e8d5648eSdan for(iCol=0; iCol<nCol; iCol++){ 19441f34f8ccSdan sessionAppendCol(&buf, pSel, iCol, &rc); 19454fccf43aSdan } 1946e8d5648eSdan }else{ 194773b3c055Sdan rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK); 19484fccf43aSdan } 1949798693b2Sdan }else if( p->op!=SQLITE_INSERT ){ 1950a71d2371Sdan rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK); 19514fccf43aSdan } 195212ca0b56Sdan if( rc==SQLITE_OK ){ 19531f34f8ccSdan rc = sqlite3_reset(pSel); 19544fccf43aSdan } 1955ef7a6304Sdan 1956f1a08ad8Sdrh /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass 1957ef7a6304Sdan ** its contents to the xOutput() callback. */ 1958ef7a6304Sdan if( xOutput 1959ef7a6304Sdan && rc==SQLITE_OK 1960ef7a6304Sdan && buf.nBuf>nNoop 1961f1a08ad8Sdrh && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE 1962ef7a6304Sdan ){ 1963ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 1964ef7a6304Sdan nNoop = -1; 1965ef7a6304Sdan buf.nBuf = 0; 1966ef7a6304Sdan } 1967ef7a6304Sdan 19684fccf43aSdan } 1969e8d5648eSdan } 19704fccf43aSdan 19711f34f8ccSdan sqlite3_finalize(pSel); 19721f34f8ccSdan if( buf.nBuf==nNoop ){ 19734fccf43aSdan buf.nBuf = nRewind; 19744fccf43aSdan } 1975cfdbde21Sdrh sqlite3_free((char*)azCol); /* cast works around VC++ bug */ 19764fccf43aSdan } 19774fccf43aSdan } 19784fccf43aSdan 19794fccf43aSdan if( rc==SQLITE_OK ){ 1980ef7a6304Sdan if( xOutput==0 ){ 19814fccf43aSdan *pnChangeset = buf.nBuf; 19824fccf43aSdan *ppChangeset = buf.aBuf; 1983ef7a6304Sdan buf.aBuf = 0; 1984ef7a6304Sdan }else if( buf.nBuf>0 ){ 1985ef7a6304Sdan rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf); 1986ef7a6304Sdan } 19874fccf43aSdan } 19884c220252Sdan 1989ef7a6304Sdan sqlite3_free(buf.aBuf); 1990e5754eecSdan sqlite3_exec(db, "RELEASE changeset", 0, 0, 0); 19914c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 19924fccf43aSdan return rc; 19934fccf43aSdan } 19944fccf43aSdan 1995296c7658Sdan /* 199673b3c055Sdan ** Obtain a changeset object containing all changes recorded by the 199773b3c055Sdan ** session object passed as the first argument. 199873b3c055Sdan ** 199973b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 200073b3c055Sdan ** using sqlite3_free(). 200173b3c055Sdan */ 200273b3c055Sdan int sqlite3session_changeset( 200373b3c055Sdan sqlite3_session *pSession, /* Session object */ 200473b3c055Sdan int *pnChangeset, /* OUT: Size of buffer at *ppChangeset */ 200573b3c055Sdan void **ppChangeset /* OUT: Buffer containing changeset */ 200673b3c055Sdan ){ 2007ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset); 2008ef7a6304Sdan } 2009ef7a6304Sdan 2010ef7a6304Sdan /* 2011ef7a6304Sdan ** Streaming version of sqlite3session_changeset(). 2012ef7a6304Sdan */ 2013f1a08ad8Sdrh int sqlite3session_changeset_strm( 2014ef7a6304Sdan sqlite3_session *pSession, 2015ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2016ef7a6304Sdan void *pOut 2017ef7a6304Sdan ){ 2018ef7a6304Sdan return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0); 2019ef7a6304Sdan } 2020ef7a6304Sdan 2021ef7a6304Sdan /* 2022ef7a6304Sdan ** Streaming version of sqlite3session_patchset(). 2023ef7a6304Sdan */ 2024f1a08ad8Sdrh int sqlite3session_patchset_strm( 2025ef7a6304Sdan sqlite3_session *pSession, 2026ef7a6304Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 2027ef7a6304Sdan void *pOut 2028ef7a6304Sdan ){ 2029ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0); 203073b3c055Sdan } 203173b3c055Sdan 203273b3c055Sdan /* 203373b3c055Sdan ** Obtain a patchset object containing all changes recorded by the 203473b3c055Sdan ** session object passed as the first argument. 203573b3c055Sdan ** 203673b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer 203773b3c055Sdan ** using sqlite3_free(). 203873b3c055Sdan */ 203973b3c055Sdan int sqlite3session_patchset( 204073b3c055Sdan sqlite3_session *pSession, /* Session object */ 204173b3c055Sdan int *pnPatchset, /* OUT: Size of buffer at *ppChangeset */ 204273b3c055Sdan void **ppPatchset /* OUT: Buffer containing changeset */ 204373b3c055Sdan ){ 2044ef7a6304Sdan return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset); 204573b3c055Sdan } 204673b3c055Sdan 204773b3c055Sdan /* 2048296c7658Sdan ** Enable or disable the session object passed as the first argument. 2049296c7658Sdan */ 20504fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){ 20514c220252Sdan int ret; 20524c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2053296c7658Sdan if( bEnable>=0 ){ 2054296c7658Sdan pSession->bEnable = bEnable; 20554fccf43aSdan } 20564c220252Sdan ret = pSession->bEnable; 20574c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 20584c220252Sdan return ret; 2059296c7658Sdan } 20604fccf43aSdan 20614fccf43aSdan /* 2062b4480e94Sdan ** Enable or disable the session object passed as the first argument. 2063b4480e94Sdan */ 2064b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){ 2065b4480e94Sdan int ret; 2066b4480e94Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2067b4480e94Sdan if( bIndirect>=0 ){ 2068b4480e94Sdan pSession->bIndirect = bIndirect; 2069b4480e94Sdan } 2070b4480e94Sdan ret = pSession->bIndirect; 2071b4480e94Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2072b4480e94Sdan return ret; 2073b4480e94Sdan } 2074b4480e94Sdan 2075b4480e94Sdan /* 2076b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded 2077b69ec348Sdan ** by the session object passed as the only argument. 2078b69ec348Sdan */ 2079b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){ 2080b69ec348Sdan int ret = 0; 2081b69ec348Sdan SessionTable *pTab; 2082b69ec348Sdan 2083b69ec348Sdan sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db)); 2084b69ec348Sdan for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){ 2085b69ec348Sdan ret = (pTab->nEntry>0); 2086b69ec348Sdan } 2087b69ec348Sdan sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db)); 2088b69ec348Sdan 2089ff530326Sdan return (ret==0); 2090b69ec348Sdan } 2091b69ec348Sdan 2092b69ec348Sdan /* 2093f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm(). 20944fccf43aSdan */ 2095ef7a6304Sdan int sessionChangesetStart( 2096296c7658Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2097ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2098ef7a6304Sdan void *pIn, 2099296c7658Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2100296c7658Sdan void *pChangeset /* Pointer to buffer containing changeset */ 21014fccf43aSdan ){ 21024fccf43aSdan sqlite3_changeset_iter *pRet; /* Iterator to return */ 21034fccf43aSdan int nByte; /* Number of bytes to allocate for iterator */ 21044fccf43aSdan 2105ef7a6304Sdan assert( xInput==0 || (pChangeset==0 && nChangeset==0) ); 2106ef7a6304Sdan 2107296c7658Sdan /* Zero the output variable in case an error occurs. */ 2108296c7658Sdan *pp = 0; 21094fccf43aSdan 2110296c7658Sdan /* Allocate and initialize the iterator structure. */ 21114fccf43aSdan nByte = sizeof(sqlite3_changeset_iter); 21124fccf43aSdan pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte); 21134fccf43aSdan if( !pRet ) return SQLITE_NOMEM; 21144fccf43aSdan memset(pRet, 0, sizeof(sqlite3_changeset_iter)); 21154757c658Sdan pRet->in.aData = (u8 *)pChangeset; 21164757c658Sdan pRet->in.nData = nChangeset; 2117ef7a6304Sdan pRet->in.xInput = xInput; 2118ef7a6304Sdan pRet->in.pIn = pIn; 2119ef7a6304Sdan pRet->in.iNext = 0; 2120ef7a6304Sdan pRet->in.bEof = (xInput ? 0 : 1); 21214fccf43aSdan 2122296c7658Sdan /* Populate the output variable and return success. */ 2123296c7658Sdan *pp = pRet; 21244fccf43aSdan return SQLITE_OK; 21254fccf43aSdan } 21264fccf43aSdan 2127296c7658Sdan /* 2128ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset. 2129ef7a6304Sdan */ 2130ef7a6304Sdan int sqlite3changeset_start( 2131ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2132ef7a6304Sdan int nChangeset, /* Size of buffer pChangeset in bytes */ 2133ef7a6304Sdan void *pChangeset /* Pointer to buffer containing changeset */ 2134ef7a6304Sdan ){ 2135ef7a6304Sdan return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset); 2136ef7a6304Sdan } 2137ef7a6304Sdan 2138ef7a6304Sdan /* 2139ef7a6304Sdan ** Streaming version of sqlite3changeset_start(). 2140ef7a6304Sdan */ 2141f1a08ad8Sdrh int sqlite3changeset_start_strm( 2142ef7a6304Sdan sqlite3_changeset_iter **pp, /* OUT: Changeset iterator handle */ 2143ef7a6304Sdan int (*xInput)(void *pIn, void *pData, int *pnData), 2144ef7a6304Sdan void *pIn 2145ef7a6304Sdan ){ 2146ef7a6304Sdan return sessionChangesetStart(pp, xInput, pIn, 0, 0); 2147ef7a6304Sdan } 2148ef7a6304Sdan 2149ef7a6304Sdan /* 2150ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or, 2151ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available 2152ef7a6304Sdan ** data is in the buffer. 2153ef7a6304Sdan ** 2154ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise. 2155ef7a6304Sdan */ 21564757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){ 2157ef7a6304Sdan int rc = SQLITE_OK; 21584757c658Sdan if( pIn->xInput ){ 21594757c658Sdan while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){ 2160f1a08ad8Sdrh int nNew = SESSIONS_STRM_CHUNK_SIZE; 21614757c658Sdan 2162f1a08ad8Sdrh if( pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){ 21634757c658Sdan int nMove = pIn->buf.nBuf - pIn->iNext; 21644757c658Sdan memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove); 21654757c658Sdan pIn->buf.nBuf -= pIn->iNext; 21664757c658Sdan pIn->iNext = 0; 21674757c658Sdan } 21684757c658Sdan 21694757c658Sdan if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){ 21704757c658Sdan rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew); 21714757c658Sdan if( nNew==0 ){ 21724757c658Sdan pIn->bEof = 1; 21734757c658Sdan }else{ 21744757c658Sdan pIn->buf.nBuf += nNew; 21754757c658Sdan } 21764757c658Sdan } 21774757c658Sdan 21784757c658Sdan pIn->aData = pIn->buf.aBuf; 21794757c658Sdan pIn->nData = pIn->buf.nBuf; 21804757c658Sdan } 2181ef7a6304Sdan } 2182ef7a6304Sdan return rc; 2183ef7a6304Sdan } 2184ef7a6304Sdan 2185ef7a6304Sdan /* 2186ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record 2187ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec 2188ef7a6304Sdan ** until it points to the byte immediately following that record. 2189ef7a6304Sdan */ 2190ef7a6304Sdan static void sessionSkipRecord( 2191ef7a6304Sdan u8 **ppRec, /* IN/OUT: Record pointer */ 2192ef7a6304Sdan int nCol /* Number of values in record */ 2193ef7a6304Sdan ){ 2194ef7a6304Sdan u8 *aRec = *ppRec; 2195ef7a6304Sdan int i; 2196ef7a6304Sdan for(i=0; i<nCol; i++){ 2197ef7a6304Sdan int eType = *aRec++; 2198ef7a6304Sdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2199ef7a6304Sdan int nByte; 2200ef7a6304Sdan aRec += sessionVarintGet((u8*)aRec, &nByte); 2201ef7a6304Sdan aRec += nByte; 2202ef7a6304Sdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2203ef7a6304Sdan aRec += 8; 2204ef7a6304Sdan } 2205ef7a6304Sdan } 2206ef7a6304Sdan 2207ef7a6304Sdan *ppRec = aRec; 2208ef7a6304Sdan } 2209ef7a6304Sdan 2210ef7a6304Sdan /* 22114757c658Sdan ** This function sets the value of the sqlite3_value object passed as the 22124757c658Sdan ** first argument to a copy of the string or blob held in the aData[] 22134757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM 22144757c658Sdan ** error occurs. 22154757c658Sdan */ 22164757c658Sdan static int sessionValueSetStr( 22174757c658Sdan sqlite3_value *pVal, /* Set the value of this object */ 22184757c658Sdan u8 *aData, /* Buffer containing string or blob data */ 22194757c658Sdan int nData, /* Size of buffer aData[] in bytes */ 22204757c658Sdan u8 enc /* String encoding (0 for blobs) */ 22214757c658Sdan ){ 222216228167Sdan /* In theory this code could just pass SQLITE_TRANSIENT as the final 222316228167Sdan ** argument to sqlite3ValueSetStr() and have the copy created 222416228167Sdan ** automatically. But doing so makes it difficult to detect any OOM 222516228167Sdan ** error. Hence the code to create the copy externally. */ 22263cc89d95Sdan u8 *aCopy = sqlite3_malloc(nData+1); 22274757c658Sdan if( aCopy==0 ) return SQLITE_NOMEM; 22284757c658Sdan memcpy(aCopy, aData, nData); 22294757c658Sdan sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free); 22304757c658Sdan return SQLITE_OK; 22314757c658Sdan } 22324757c658Sdan 22334757c658Sdan /* 2234296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT" 2235296c7658Sdan ** for details. 2236296c7658Sdan ** 2237296c7658Sdan ** When this function is called, *paChange points to the start of the record 2238296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to 2239296c7658Sdan ** one byte after the end of the same record before this function returns. 2240a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or, 2241a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields 2242a71d2371Sdan ** (in other words, it is a patchset DELETE record). 2243296c7658Sdan ** 2244296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller) 2245296c7658Sdan ** is set to point to an sqlite3_value object containing the value read 2246296c7658Sdan ** from the corresponding position in the record. If that value is not 2247296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change 2248296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is 2249296c7658Sdan ** set to NULL. 2250296c7658Sdan ** 2251296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures 2252296c7658Sdan ** using sqlite3_free(). 2253296c7658Sdan ** 2254296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned. 2255296c7658Sdan ** The apOut[] array may have been partially populated in this case. 2256296c7658Sdan */ 22574fccf43aSdan static int sessionReadRecord( 2258ef7a6304Sdan SessionInput *pIn, /* Input data */ 22594fccf43aSdan int nCol, /* Number of values in record */ 226073b3c055Sdan u8 *abPK, /* Array of primary key flags, or NULL */ 22614fccf43aSdan sqlite3_value **apOut /* Write values to this array */ 22624fccf43aSdan ){ 2263296c7658Sdan int i; /* Used to iterate through columns */ 2264ef7a6304Sdan int rc = SQLITE_OK; 22654fccf43aSdan 2266ef7a6304Sdan for(i=0; i<nCol && rc==SQLITE_OK; i++){ 2267ef7a6304Sdan int eType = 0; /* Type of value (SQLITE_NULL, TEXT etc.) */ 226873b3c055Sdan if( abPK && abPK[i]==0 ) continue; 2269ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2270ef7a6304Sdan if( rc==SQLITE_OK ){ 22714757c658Sdan eType = pIn->aData[pIn->iNext++]; 2272ef7a6304Sdan } 2273ef7a6304Sdan 2274e8fa8c96Sdan assert( apOut[i]==0 ); 22754fccf43aSdan if( eType ){ 22764fccf43aSdan apOut[i] = sqlite3ValueNew(0); 2277ef7a6304Sdan if( !apOut[i] ) rc = SQLITE_NOMEM; 2278ef7a6304Sdan } 22794fccf43aSdan 2280ef7a6304Sdan if( rc==SQLITE_OK ){ 22814757c658Sdan u8 *aVal = &pIn->aData[pIn->iNext]; 22824fccf43aSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 22834fccf43aSdan int nByte; 2284ef7a6304Sdan pIn->iNext += sessionVarintGet(aVal, &nByte); 2285ef7a6304Sdan rc = sessionInputBuffer(pIn, nByte); 2286e8fa8c96Sdan if( rc==SQLITE_OK ){ 22876734007dSdan u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0); 22884757c658Sdan rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc); 228991ddd559Sdan } 2290ef7a6304Sdan pIn->iNext += nByte; 22914fccf43aSdan } 22924fccf43aSdan if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2293ef7a6304Sdan sqlite3_int64 v = sessionGetI64(aVal); 22944fccf43aSdan if( eType==SQLITE_INTEGER ){ 22954fccf43aSdan sqlite3VdbeMemSetInt64(apOut[i], v); 22964fccf43aSdan }else{ 22974fccf43aSdan double d; 22984e895da1Sdan memcpy(&d, &v, 8); 22994fccf43aSdan sqlite3VdbeMemSetDouble(apOut[i], d); 23004fccf43aSdan } 2301ef7a6304Sdan pIn->iNext += 8; 230291ddd559Sdan } 23034fccf43aSdan } 23044fccf43aSdan } 23054fccf43aSdan 2306ef7a6304Sdan return rc; 2307ef7a6304Sdan } 2308ef7a6304Sdan 2309ef7a6304Sdan /* 2310ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2311ef7a6304Sdan ** Specifically, to the following: 2312ef7a6304Sdan ** 2313ef7a6304Sdan ** + number of columns in table (varint) 2314ef7a6304Sdan ** + array of PK flags (1 byte per column), 2315ef7a6304Sdan ** + table name (nul terminated). 2316ef7a6304Sdan ** 2317ef7a6304Sdan ** This function ensures that all of the above is present in the input 2318ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()). 2319ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code. 2320ef7a6304Sdan ** The input pointer is not moved. 2321ef7a6304Sdan */ 2322ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){ 2323ef7a6304Sdan int rc = SQLITE_OK; 2324ef7a6304Sdan int nCol = 0; 23254757c658Sdan int nRead = 0; 2326ef7a6304Sdan 2327ef7a6304Sdan rc = sessionInputBuffer(pIn, 9); 2328ef7a6304Sdan if( rc==SQLITE_OK ){ 23294757c658Sdan nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol); 23304757c658Sdan rc = sessionInputBuffer(pIn, nRead+nCol+100); 23314757c658Sdan nRead += nCol; 2332ef7a6304Sdan } 23334757c658Sdan 2334ef7a6304Sdan while( rc==SQLITE_OK ){ 23354757c658Sdan while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){ 23364757c658Sdan nRead++; 2337ef7a6304Sdan } 2338e8fa8c96Sdan if( (pIn->iNext + nRead)<pIn->nData ) break; 23394757c658Sdan rc = sessionInputBuffer(pIn, nRead + 100); 23404757c658Sdan } 2341e8fa8c96Sdan *pnByte = nRead+1; 2342ef7a6304Sdan return rc; 2343ef7a6304Sdan } 2344ef7a6304Sdan 2345ef7a6304Sdan /* 2346fa122adaSdan ** The input pointer currently points to the first byte of the first field 2347fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire 234816228167Sdan ** record is buffered. It does not move the input pointer. 234916228167Sdan ** 235016228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of 235116228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The 235216228167Sdan ** final value of *pnByte is undefined in this case. 2353fa122adaSdan */ 2354fa122adaSdan static int sessionChangesetBufferRecord( 235516228167Sdan SessionInput *pIn, /* Input data */ 235616228167Sdan int nCol, /* Number of columns in record */ 235716228167Sdan int *pnByte /* OUT: Size of record in bytes */ 2358fa122adaSdan ){ 2359fa122adaSdan int rc = SQLITE_OK; 2360fa122adaSdan int nByte = 0; 2361fa122adaSdan int i; 2362fa122adaSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 2363fa122adaSdan int eType; 2364fa122adaSdan rc = sessionInputBuffer(pIn, nByte + 10); 2365fa122adaSdan if( rc==SQLITE_OK ){ 2366fa122adaSdan eType = pIn->aData[pIn->iNext + nByte++]; 2367fa122adaSdan if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){ 2368fa122adaSdan int n; 2369fa122adaSdan nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n); 2370fa122adaSdan nByte += n; 2371fa122adaSdan rc = sessionInputBuffer(pIn, nByte); 2372fa122adaSdan }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){ 2373fa122adaSdan nByte += 8; 2374fa122adaSdan } 2375fa122adaSdan } 2376fa122adaSdan } 2377fa122adaSdan *pnByte = nByte; 2378fa122adaSdan return rc; 2379fa122adaSdan } 2380fa122adaSdan 2381fa122adaSdan /* 2382ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header. 2383ef7a6304Sdan ** Specifically, to the following: 2384ef7a6304Sdan ** 2385ef7a6304Sdan ** + number of columns in table (varint) 2386ef7a6304Sdan ** + array of PK flags (1 byte per column), 2387ef7a6304Sdan ** + table name (nul terminated). 238816228167Sdan ** 238916228167Sdan ** This function decodes the table-header and populates the p->nCol, 239016228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is 239116228167Sdan ** also allocated or resized according to the new value of p->nCol. The 239216228167Sdan ** input pointer is left pointing to the byte following the table header. 239316228167Sdan ** 239416228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code 239516228167Sdan ** is returned and the final values of the various fields enumerated above 239616228167Sdan ** are undefined. 2397ef7a6304Sdan */ 2398ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){ 2399ef7a6304Sdan int rc; 2400ef7a6304Sdan int nCopy; 2401ef7a6304Sdan assert( p->rc==SQLITE_OK ); 2402ef7a6304Sdan 2403ef7a6304Sdan rc = sessionChangesetBufferTblhdr(&p->in, &nCopy); 2404ef7a6304Sdan if( rc==SQLITE_OK ){ 2405ef7a6304Sdan int nByte; 2406ef7a6304Sdan int nVarint; 24074757c658Sdan nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol); 2408ef7a6304Sdan nCopy -= nVarint; 2409ef7a6304Sdan p->in.iNext += nVarint; 2410ef7a6304Sdan nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy; 2411ef7a6304Sdan p->tblhdr.nBuf = 0; 2412ef7a6304Sdan sessionBufferGrow(&p->tblhdr, nByte, &rc); 2413ef7a6304Sdan } 2414ef7a6304Sdan 2415ef7a6304Sdan if( rc==SQLITE_OK ){ 2416ef7a6304Sdan int iPK = sizeof(sqlite3_value*)*p->nCol*2; 2417ef7a6304Sdan memset(p->tblhdr.aBuf, 0, iPK); 24184757c658Sdan memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy); 2419ef7a6304Sdan p->in.iNext += nCopy; 2420ef7a6304Sdan } 2421ef7a6304Sdan 2422ef7a6304Sdan p->apValue = (sqlite3_value**)p->tblhdr.aBuf; 2423ef7a6304Sdan p->abPK = (u8*)&p->apValue[p->nCol*2]; 2424ef7a6304Sdan p->zTab = (char*)&p->abPK[p->nCol]; 2425ef7a6304Sdan return (p->rc = rc); 24264fccf43aSdan } 24274fccf43aSdan 242877fc1d5bSdan /* 242977fc1d5bSdan ** Advance the changeset iterator to the next change. 243077fc1d5bSdan ** 243177fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public 243277fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the 243377fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values. 243477fc1d5bSdan ** 243577fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change 243677fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in 243777fc1d5bSdan ** the record to *pnRec. 243877fc1d5bSdan ** 243977fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is 244077fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite 244177fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further 244277fc1d5bSdan ** changes in the changeset. 244377fc1d5bSdan */ 24445d607a6eSdan static int sessionChangesetNext( 244577fc1d5bSdan sqlite3_changeset_iter *p, /* Changeset iterator */ 244677fc1d5bSdan u8 **paRec, /* If non-NULL, store record pointer here */ 244777fc1d5bSdan int *pnRec /* If non-NULL, store size of record here */ 24485d607a6eSdan ){ 24494fccf43aSdan int i; 2450ef7a6304Sdan u8 op; 24514fccf43aSdan 24525d607a6eSdan assert( (paRec==0 && pnRec==0) || (paRec && pnRec) ); 24535d607a6eSdan 2454296c7658Sdan /* If the iterator is in the error-state, return immediately. */ 24554fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 24564fccf43aSdan 24575d607a6eSdan /* Free the current contents of p->apValue[], if any. */ 24584fccf43aSdan if( p->apValue ){ 24594fccf43aSdan for(i=0; i<p->nCol*2; i++){ 24604fccf43aSdan sqlite3ValueFree(p->apValue[i]); 24614fccf43aSdan } 24624fccf43aSdan memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2); 24634fccf43aSdan } 24644fccf43aSdan 2465ef7a6304Sdan /* Make sure the buffer contains at least 10 bytes of input data, or all 2466ef7a6304Sdan ** remaining data if there are less than 10 bytes available. This is 2467ef7a6304Sdan ** sufficient either for the 'T' or 'P' byte and the varint that follows 2468ef7a6304Sdan ** it, or for the two single byte values otherwise. */ 2469ef7a6304Sdan p->rc = sessionInputBuffer(&p->in, 2); 2470ef7a6304Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2471ef7a6304Sdan 24724fccf43aSdan /* If the iterator is already at the end of the changeset, return DONE. */ 24734757c658Sdan if( p->in.iNext>=p->in.nData ){ 24744fccf43aSdan return SQLITE_DONE; 24754fccf43aSdan } 24764fccf43aSdan 24774757c658Sdan op = p->in.aData[p->in.iNext++]; 2478ef7a6304Sdan if( op=='T' || op=='P' ){ 2479ef7a6304Sdan p->bPatchset = (op=='P'); 2480ef7a6304Sdan if( sessionChangesetReadTblhdr(p) ) return p->rc; 2481ef7a6304Sdan if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc; 24824757c658Sdan op = p->in.aData[p->in.iNext++]; 24835d607a6eSdan } 24845d607a6eSdan 2485ef7a6304Sdan p->op = op; 24864757c658Sdan p->bIndirect = p->in.aData[p->in.iNext++]; 24874fccf43aSdan if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){ 24884757c658Sdan return (p->rc = SQLITE_CORRUPT_BKPT); 24894fccf43aSdan } 24904fccf43aSdan 2491cbf6d2d2Sdan if( paRec ){ 2492cbf6d2d2Sdan int nVal; /* Number of values to buffer */ 2493cbf6d2d2Sdan if( p->bPatchset==0 && op==SQLITE_UPDATE ){ 2494cbf6d2d2Sdan nVal = p->nCol * 2; 2495cbf6d2d2Sdan }else if( p->bPatchset && op==SQLITE_DELETE ){ 2496cbf6d2d2Sdan nVal = 0; 2497cbf6d2d2Sdan for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++; 2498cbf6d2d2Sdan }else{ 2499cbf6d2d2Sdan nVal = p->nCol; 2500cbf6d2d2Sdan } 2501cbf6d2d2Sdan p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec); 2502cbf6d2d2Sdan if( p->rc!=SQLITE_OK ) return p->rc; 2503cbf6d2d2Sdan *paRec = &p->in.aData[p->in.iNext]; 2504cbf6d2d2Sdan p->in.iNext += *pnRec; 2505cbf6d2d2Sdan }else{ 25065d607a6eSdan 25074fccf43aSdan /* If this is an UPDATE or DELETE, read the old.* record. */ 250873b3c055Sdan if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){ 250973b3c055Sdan u8 *abPK = p->bPatchset ? p->abPK : 0; 2510cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue); 25114fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 25124fccf43aSdan } 25134fccf43aSdan 25144fccf43aSdan /* If this is an INSERT or UPDATE, read the new.* record. */ 25154fccf43aSdan if( p->op!=SQLITE_DELETE ){ 2516cbf6d2d2Sdan p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]); 25174fccf43aSdan if( p->rc!=SQLITE_OK ) return p->rc; 25184fccf43aSdan } 25194fccf43aSdan 2520cbf6d2d2Sdan if( p->bPatchset && p->op==SQLITE_UPDATE ){ 252173b3c055Sdan /* If this is an UPDATE that is part of a patchset, then all PK and 252273b3c055Sdan ** modified fields are present in the new.* record. The old.* record 252373b3c055Sdan ** is currently completely empty. This block shifts the PK fields from 252473b3c055Sdan ** new.* to old.*, to accommodate the code that reads these arrays. */ 252573b3c055Sdan int i; 252673b3c055Sdan for(i=0; i<p->nCol; i++){ 252773b3c055Sdan assert( p->apValue[i]==0 ); 252873b3c055Sdan assert( p->abPK[i]==0 || p->apValue[i+p->nCol] ); 252973b3c055Sdan if( p->abPK[i] ){ 253073b3c055Sdan p->apValue[i] = p->apValue[i+p->nCol]; 253173b3c055Sdan p->apValue[i+p->nCol] = 0; 253273b3c055Sdan } 253373b3c055Sdan } 253473b3c055Sdan } 2535cbf6d2d2Sdan } 2536ef7a6304Sdan 25374fccf43aSdan return SQLITE_ROW; 25384fccf43aSdan } 25394fccf43aSdan 25404fccf43aSdan /* 25415d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next 25425d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE 25435d607a6eSdan ** or SQLITE_CORRUPT. 25445d607a6eSdan ** 25455d607a6eSdan ** This function may not be called on iterators passed to a conflict handler 25465d607a6eSdan ** callback by changeset_apply(). 25475d607a6eSdan */ 25485d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){ 25495d607a6eSdan return sessionChangesetNext(p, 0, 0); 25505d607a6eSdan } 25515d607a6eSdan 25525d607a6eSdan /* 2553244593c8Sdan ** The following function extracts information on the current change 255477fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next() 25554fccf43aSdan ** has returned SQLITE_ROW. 25564fccf43aSdan */ 25574fccf43aSdan int sqlite3changeset_op( 2558296c7658Sdan sqlite3_changeset_iter *pIter, /* Iterator handle */ 25594fccf43aSdan const char **pzTab, /* OUT: Pointer to table name */ 25604fccf43aSdan int *pnCol, /* OUT: Number of columns in table */ 2561b4480e94Sdan int *pOp, /* OUT: SQLITE_INSERT, DELETE or UPDATE */ 2562b4480e94Sdan int *pbIndirect /* OUT: True if change is indirect */ 25634fccf43aSdan ){ 25644fccf43aSdan *pOp = pIter->op; 25654fccf43aSdan *pnCol = pIter->nCol; 25664fccf43aSdan *pzTab = pIter->zTab; 2567b4480e94Sdan if( pbIndirect ) *pbIndirect = pIter->bIndirect; 25684fccf43aSdan return SQLITE_OK; 25694fccf43aSdan } 25704fccf43aSdan 257177fc1d5bSdan /* 257277fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in 257377fc1d5bSdan ** the database table affected by the change that pIter currently points 257477fc1d5bSdan ** to. This function may only be called after changeset_next() returns 257577fc1d5bSdan ** SQLITE_ROW. 257677fc1d5bSdan */ 2577244593c8Sdan int sqlite3changeset_pk( 2578244593c8Sdan sqlite3_changeset_iter *pIter, /* Iterator object */ 2579244593c8Sdan unsigned char **pabPK, /* OUT: Array of boolean - true for PK cols */ 2580244593c8Sdan int *pnCol /* OUT: Number of entries in output array */ 2581244593c8Sdan ){ 2582244593c8Sdan *pabPK = pIter->abPK; 2583244593c8Sdan if( pnCol ) *pnCol = pIter->nCol; 2584244593c8Sdan return SQLITE_OK; 2585244593c8Sdan } 2586244593c8Sdan 2587296c7658Sdan /* 2588296c7658Sdan ** This function may only be called while the iterator is pointing to an 2589296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()). 2590296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 2591296c7658Sdan ** 2592296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 2593296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not 2594296c7658Sdan ** included in the record (because the change is an UPDATE and the field 2595296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL. 2596296c7658Sdan ** 2597296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 2598296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 2599296c7658Sdan */ 26004fccf43aSdan int sqlite3changeset_old( 2601296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 2602296c7658Sdan int iVal, /* Index of old.* value to retrieve */ 26034fccf43aSdan sqlite3_value **ppValue /* OUT: Old value (or NULL pointer) */ 26044fccf43aSdan ){ 2605d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){ 2606d5f0767cSdan return SQLITE_MISUSE; 2607d5f0767cSdan } 26084fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 26094fccf43aSdan return SQLITE_RANGE; 26104fccf43aSdan } 26114fccf43aSdan *ppValue = pIter->apValue[iVal]; 26124fccf43aSdan return SQLITE_OK; 26134fccf43aSdan } 26144fccf43aSdan 2615296c7658Sdan /* 2616296c7658Sdan ** This function may only be called while the iterator is pointing to an 2617296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()). 2618296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned. 2619296c7658Sdan ** 2620296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the 2621296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not 2622296c7658Sdan ** included in the record (because the change is an UPDATE and the field 2623296c7658Sdan ** was not modified), set *ppValue to NULL. 2624296c7658Sdan ** 2625296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is 2626296c7658Sdan ** not modified. Otherwise, SQLITE_OK. 2627296c7658Sdan */ 26284fccf43aSdan int sqlite3changeset_new( 2629296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 2630296c7658Sdan int iVal, /* Index of new.* value to retrieve */ 26314fccf43aSdan sqlite3_value **ppValue /* OUT: New value (or NULL pointer) */ 26324fccf43aSdan ){ 2633d5f0767cSdan if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){ 2634d5f0767cSdan return SQLITE_MISUSE; 2635d5f0767cSdan } 26364fccf43aSdan if( iVal<0 || iVal>=pIter->nCol ){ 26374fccf43aSdan return SQLITE_RANGE; 26384fccf43aSdan } 26394fccf43aSdan *ppValue = pIter->apValue[pIter->nCol+iVal]; 26404fccf43aSdan return SQLITE_OK; 26414fccf43aSdan } 26424fccf43aSdan 2643296c7658Sdan /* 26447aa469cdSdan ** The following two macros are used internally. They are similar to the 26457aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that 26467aa469cdSdan ** they omit all error checking and return a pointer to the requested value. 26477aa469cdSdan */ 26487aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)] 26497aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)] 26507aa469cdSdan 26517aa469cdSdan /* 2652296c7658Sdan ** This function may only be called with a changeset iterator that has been 2653296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT 2654296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned. 2655296c7658Sdan ** 2656296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure 2657296c7658Sdan ** containing the iVal'th value of the conflicting record. 2658296c7658Sdan ** 2659296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error 2660296c7658Sdan ** code is returned. Otherwise, SQLITE_OK. 2661296c7658Sdan */ 2662d5f0767cSdan int sqlite3changeset_conflict( 2663296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 2664296c7658Sdan int iVal, /* Index of conflict record value to fetch */ 2665d5f0767cSdan sqlite3_value **ppValue /* OUT: Value from conflicting row */ 2666d5f0767cSdan ){ 2667d5f0767cSdan if( !pIter->pConflict ){ 2668d5f0767cSdan return SQLITE_MISUSE; 2669d5f0767cSdan } 2670d5f0767cSdan if( iVal<0 || iVal>=sqlite3_column_count(pIter->pConflict) ){ 2671d5f0767cSdan return SQLITE_RANGE; 2672d5f0767cSdan } 2673d5f0767cSdan *ppValue = sqlite3_column_value(pIter->pConflict, iVal); 2674d5f0767cSdan return SQLITE_OK; 2675d5f0767cSdan } 2676d5f0767cSdan 26774fccf43aSdan /* 2678cb3e4b79Sdan ** This function may only be called with an iterator passed to an 2679cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case 2680cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key 2681cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK. 2682cb3e4b79Sdan ** 2683cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE. 2684cb3e4b79Sdan */ 2685cb3e4b79Sdan int sqlite3changeset_fk_conflicts( 2686cb3e4b79Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 2687cb3e4b79Sdan int *pnOut /* OUT: Number of FK violations */ 2688cb3e4b79Sdan ){ 2689cb3e4b79Sdan if( pIter->pConflict || pIter->apValue ){ 2690cb3e4b79Sdan return SQLITE_MISUSE; 2691cb3e4b79Sdan } 2692cb3e4b79Sdan *pnOut = pIter->nCol; 2693cb3e4b79Sdan return SQLITE_OK; 2694cb3e4b79Sdan } 2695cb3e4b79Sdan 2696cb3e4b79Sdan 2697cb3e4b79Sdan /* 26984fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start(). 26994fccf43aSdan ** 27004fccf43aSdan ** This function may not be called on iterators passed to a conflict handler 27014fccf43aSdan ** callback by changeset_apply(). 27024fccf43aSdan */ 27034fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){ 2704cbf6d2d2Sdan int rc = SQLITE_OK; 2705cbf6d2d2Sdan if( p ){ 2706296c7658Sdan int i; /* Used to iterate through p->apValue[] */ 2707cbf6d2d2Sdan rc = p->rc; 270812ca0b56Sdan if( p->apValue ){ 27094fccf43aSdan for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]); 271012ca0b56Sdan } 2711ef7a6304Sdan sqlite3_free(p->tblhdr.aBuf); 27124757c658Sdan sqlite3_free(p->in.buf.aBuf); 27134fccf43aSdan sqlite3_free(p); 2714cbf6d2d2Sdan } 27154fccf43aSdan return rc; 27164fccf43aSdan } 27174fccf43aSdan 2718fa122adaSdan static int sessionChangesetInvert( 2719fa122adaSdan SessionInput *pInput, /* Input changeset */ 2720fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 2721fa122adaSdan void *pOut, 272291ddd559Sdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 272391ddd559Sdan void **ppInverted /* OUT: Inverse of pChangeset */ 272491ddd559Sdan ){ 2725cfec7eeeSdan int rc = SQLITE_OK; /* Return value */ 2726fa122adaSdan SessionBuffer sOut; /* Output buffer */ 2727cfec7eeeSdan int nCol = 0; /* Number of cols in current table */ 2728cfec7eeeSdan u8 *abPK = 0; /* PK array for current table */ 2729cfec7eeeSdan sqlite3_value **apVal = 0; /* Space for values for UPDATE inversion */ 2730ef7a6304Sdan SessionBuffer sPK = {0, 0, 0}; /* PK array for current table */ 273191ddd559Sdan 2732fa122adaSdan /* Initialize the output buffer */ 2733fa122adaSdan memset(&sOut, 0, sizeof(SessionBuffer)); 2734fa122adaSdan 273591ddd559Sdan /* Zero the output variables in case an error occurs. */ 2736fa122adaSdan if( ppInverted ){ 273791ddd559Sdan *ppInverted = 0; 273891ddd559Sdan *pnInverted = 0; 2739fa122adaSdan } 274091ddd559Sdan 2741fa122adaSdan while( 1 ){ 2742ef7a6304Sdan u8 eType; 2743fa122adaSdan 2744fa122adaSdan /* Test for EOF. */ 2745fa122adaSdan if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert; 2746fa122adaSdan if( pInput->iNext>=pInput->nData ) break; 2747fa122adaSdan eType = pInput->aData[pInput->iNext]; 2748fa122adaSdan 274991ddd559Sdan switch( eType ){ 275091ddd559Sdan case 'T': { 2751244593c8Sdan /* A 'table' record consists of: 2752244593c8Sdan ** 2753244593c8Sdan ** * A constant 'T' character, 2754244593c8Sdan ** * Number of columns in said table (a varint), 2755ef7a6304Sdan ** * An array of nCol bytes (sPK), 2756244593c8Sdan ** * A nul-terminated table name. 2757244593c8Sdan */ 2758ef7a6304Sdan int nByte; 2759fa122adaSdan int nVar; 2760fa122adaSdan pInput->iNext++; 2761fa122adaSdan if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){ 2762ef7a6304Sdan goto finished_invert; 2763ef7a6304Sdan } 2764fa122adaSdan nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol); 2765ef7a6304Sdan sPK.nBuf = 0; 2766fa122adaSdan sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc); 2767fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 2768fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 2769ef7a6304Sdan if( rc ) goto finished_invert; 2770fa122adaSdan 2771fa122adaSdan pInput->iNext += nByte; 2772cfec7eeeSdan sqlite3_free(apVal); 2773cfec7eeeSdan apVal = 0; 2774ef7a6304Sdan abPK = sPK.aBuf; 277591ddd559Sdan break; 277691ddd559Sdan } 277791ddd559Sdan 277891ddd559Sdan case SQLITE_INSERT: 277991ddd559Sdan case SQLITE_DELETE: { 278091ddd559Sdan int nByte; 2781fa122adaSdan int bIndirect = pInput->aData[pInput->iNext+1]; 2782fa122adaSdan int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE); 2783fa122adaSdan pInput->iNext += 2; 2784fa122adaSdan assert( rc==SQLITE_OK ); 2785fa122adaSdan rc = sessionChangesetBufferRecord(pInput, nCol, &nByte); 2786fa122adaSdan sessionAppendByte(&sOut, eType2, &rc); 2787fa122adaSdan sessionAppendByte(&sOut, bIndirect, &rc); 2788fa122adaSdan sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc); 2789fa122adaSdan pInput->iNext += nByte; 2790fa122adaSdan if( rc ) goto finished_invert; 279191ddd559Sdan break; 279291ddd559Sdan } 279391ddd559Sdan 279491ddd559Sdan case SQLITE_UPDATE: { 2795cfec7eeeSdan int iCol; 279691ddd559Sdan 2797cfec7eeeSdan if( 0==apVal ){ 2798cfec7eeeSdan apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2); 2799cfec7eeeSdan if( 0==apVal ){ 2800cfec7eeeSdan rc = SQLITE_NOMEM; 2801cfec7eeeSdan goto finished_invert; 2802cfec7eeeSdan } 2803cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 2804cfec7eeeSdan } 280591ddd559Sdan 2806cfec7eeeSdan /* Write the header for the new UPDATE change. Same as the original. */ 2807fa122adaSdan sessionAppendByte(&sOut, eType, &rc); 2808fa122adaSdan sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc); 280991ddd559Sdan 2810ef7a6304Sdan /* Read the old.* and new.* records for the update change. */ 2811fa122adaSdan pInput->iNext += 2; 2812fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]); 2813ef7a6304Sdan if( rc==SQLITE_OK ){ 2814fa122adaSdan rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]); 2815ef7a6304Sdan } 2816ef7a6304Sdan 2817cfec7eeeSdan /* Write the new old.* record. Consists of the PK columns from the 2818cfec7eeeSdan ** original old.* record, and the other values from the original 2819cfec7eeeSdan ** new.* record. */ 2820e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 2821cfec7eeeSdan sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)]; 2822fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 2823cfec7eeeSdan } 2824cfec7eeeSdan 2825cfec7eeeSdan /* Write the new new.* record. Consists of a copy of all values 2826cfec7eeeSdan ** from the original old.* record, except for the PK columns, which 2827cfec7eeeSdan ** are set to "undefined". */ 2828e8fa8c96Sdan for(iCol=0; iCol<nCol; iCol++){ 2829cfec7eeeSdan sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]); 2830fa122adaSdan sessionAppendValue(&sOut, pVal, &rc); 2831cfec7eeeSdan } 2832cfec7eeeSdan 2833cfec7eeeSdan for(iCol=0; iCol<nCol*2; iCol++){ 2834cfec7eeeSdan sqlite3ValueFree(apVal[iCol]); 2835cfec7eeeSdan } 2836cfec7eeeSdan memset(apVal, 0, sizeof(apVal[0])*nCol*2); 2837cfec7eeeSdan if( rc!=SQLITE_OK ){ 2838cfec7eeeSdan goto finished_invert; 2839cfec7eeeSdan } 2840cfec7eeeSdan 284191ddd559Sdan break; 284291ddd559Sdan } 284391ddd559Sdan 284491ddd559Sdan default: 28454757c658Sdan rc = SQLITE_CORRUPT_BKPT; 2846cfec7eeeSdan goto finished_invert; 284791ddd559Sdan } 2848fa122adaSdan 2849fa122adaSdan assert( rc==SQLITE_OK ); 2850f1a08ad8Sdrh if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 2851fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 2852fa122adaSdan sOut.nBuf = 0; 2853fa122adaSdan if( rc!=SQLITE_OK ) goto finished_invert; 2854fa122adaSdan } 285591ddd559Sdan } 285691ddd559Sdan 2857cfec7eeeSdan assert( rc==SQLITE_OK ); 2858fa122adaSdan if( pnInverted ){ 2859fa122adaSdan *pnInverted = sOut.nBuf; 2860fa122adaSdan *ppInverted = sOut.aBuf; 2861fa122adaSdan sOut.aBuf = 0; 2862fa122adaSdan }else if( sOut.nBuf>0 ){ 2863fa122adaSdan rc = xOutput(pOut, sOut.aBuf, sOut.nBuf); 2864fa122adaSdan } 2865cfec7eeeSdan 2866cfec7eeeSdan finished_invert: 2867fa122adaSdan sqlite3_free(sOut.aBuf); 2868cfec7eeeSdan sqlite3_free(apVal); 2869ef7a6304Sdan sqlite3_free(sPK.aBuf); 2870cfec7eeeSdan return rc; 287191ddd559Sdan } 287291ddd559Sdan 2873fa122adaSdan 2874fa122adaSdan /* 2875fa122adaSdan ** Invert a changeset object. 2876fa122adaSdan */ 2877fa122adaSdan int sqlite3changeset_invert( 2878fa122adaSdan int nChangeset, /* Number of bytes in input */ 2879fa122adaSdan const void *pChangeset, /* Input changeset */ 2880fa122adaSdan int *pnInverted, /* OUT: Number of bytes in output changeset */ 2881fa122adaSdan void **ppInverted /* OUT: Inverse of pChangeset */ 2882fa122adaSdan ){ 2883fa122adaSdan SessionInput sInput; 2884fa122adaSdan 2885fa122adaSdan /* Set up the input stream */ 2886fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 2887fa122adaSdan sInput.nData = nChangeset; 2888fa122adaSdan sInput.aData = (u8*)pChangeset; 2889fa122adaSdan 2890fa122adaSdan return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted); 2891fa122adaSdan } 2892fa122adaSdan 2893fa122adaSdan /* 2894fa122adaSdan ** Streaming version of sqlite3changeset_invert(). 2895fa122adaSdan */ 2896f1a08ad8Sdrh int sqlite3changeset_invert_strm( 2897fa122adaSdan int (*xInput)(void *pIn, void *pData, int *pnData), 2898fa122adaSdan void *pIn, 2899fa122adaSdan int (*xOutput)(void *pOut, const void *pData, int nData), 2900fa122adaSdan void *pOut 2901fa122adaSdan ){ 2902fa122adaSdan SessionInput sInput; 2903fa122adaSdan int rc; 2904fa122adaSdan 2905fa122adaSdan /* Set up the input stream */ 2906fa122adaSdan memset(&sInput, 0, sizeof(SessionInput)); 2907fa122adaSdan sInput.xInput = xInput; 2908fa122adaSdan sInput.pIn = pIn; 2909fa122adaSdan 2910fa122adaSdan rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0); 2911fa122adaSdan sqlite3_free(sInput.buf.aBuf); 2912fa122adaSdan return rc; 2913fa122adaSdan } 2914fa122adaSdan 29150c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx; 29160c698471Sdan struct SessionApplyCtx { 29170c698471Sdan sqlite3 *db; 29180c698471Sdan sqlite3_stmt *pDelete; /* DELETE statement */ 2919cfec7eeeSdan sqlite3_stmt *pUpdate; /* UPDATE statement */ 29200c698471Sdan sqlite3_stmt *pInsert; /* INSERT statement */ 29210c698471Sdan sqlite3_stmt *pSelect; /* SELECT statement */ 29220c698471Sdan int nCol; /* Size of azCol[] and abPK[] arrays */ 29230c698471Sdan const char **azCol; /* Array of column names */ 29240c698471Sdan u8 *abPK; /* Boolean array - true if column is in PK */ 29250c698471Sdan }; 29260c698471Sdan 2927d5f0767cSdan /* 2928d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table 2929d5f0767cSdan ** structure like this: 2930d5f0767cSdan ** 2931d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 2932d5f0767cSdan ** 2933d5f0767cSdan ** The DELETE statement looks like this: 2934d5f0767cSdan ** 2935db04571cSdan ** DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4) 2936d5f0767cSdan ** 2937d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require 2938d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the 2939d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE. 2940296c7658Sdan ** 2941296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left 2942296c7658Sdan ** pointing to the prepared version of the SQL statement. 2943d5f0767cSdan */ 2944d5f0767cSdan static int sessionDeleteRow( 2945d5f0767cSdan sqlite3 *db, /* Database handle */ 2946d5f0767cSdan const char *zTab, /* Table name */ 29470c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 2948d5f0767cSdan ){ 2949296c7658Sdan int i; 2950296c7658Sdan const char *zSep = ""; 2951d5f0767cSdan int rc = SQLITE_OK; 2952d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 29537cf7df7dSdan int nPk = 0; 2954d5f0767cSdan 2955d5f0767cSdan sessionAppendStr(&buf, "DELETE FROM ", &rc); 2956d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 2957296c7658Sdan sessionAppendStr(&buf, " WHERE ", &rc); 2958296c7658Sdan 2959296c7658Sdan for(i=0; i<p->nCol; i++){ 2960296c7658Sdan if( p->abPK[i] ){ 29617cf7df7dSdan nPk++; 2962296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 2963296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 2964296c7658Sdan sessionAppendStr(&buf, " = ?", &rc); 2965296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 2966296c7658Sdan zSep = " AND "; 2967296c7658Sdan } 2968296c7658Sdan } 2969296c7658Sdan 29707cf7df7dSdan if( nPk<p->nCol ){ 2971296c7658Sdan sessionAppendStr(&buf, " AND (?", &rc); 2972296c7658Sdan sessionAppendInteger(&buf, p->nCol+1, &rc); 2973296c7658Sdan sessionAppendStr(&buf, " OR ", &rc); 2974296c7658Sdan 2975296c7658Sdan zSep = ""; 2976296c7658Sdan for(i=0; i<p->nCol; i++){ 2977296c7658Sdan if( !p->abPK[i] ){ 2978296c7658Sdan sessionAppendStr(&buf, zSep, &rc); 2979296c7658Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 2980296c7658Sdan sessionAppendStr(&buf, " IS ?", &rc); 2981296c7658Sdan sessionAppendInteger(&buf, i+1, &rc); 2982296c7658Sdan zSep = "AND "; 2983296c7658Sdan } 2984296c7658Sdan } 2985296c7658Sdan sessionAppendStr(&buf, ")", &rc); 29867cf7df7dSdan } 2987d5f0767cSdan 2988d5f0767cSdan if( rc==SQLITE_OK ){ 29890c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0); 2990d5f0767cSdan } 2991d5f0767cSdan sqlite3_free(buf.aBuf); 2992d5f0767cSdan 2993d5f0767cSdan return rc; 2994d5f0767cSdan } 2995d5f0767cSdan 2996d5f0767cSdan /* 2997d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db. 2998d5f0767cSdan ** Assuming a table structure like this: 2999d5f0767cSdan ** 3000d5f0767cSdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3001d5f0767cSdan ** 3002d5f0767cSdan ** The UPDATE statement looks like this: 3003d5f0767cSdan ** 3004d5f0767cSdan ** UPDATE x SET 3005d5f0767cSdan ** a = CASE WHEN ?2 THEN ?3 ELSE a END, 3006964cbd46Sdan ** b = CASE WHEN ?5 THEN ?6 ELSE b END, 3007964cbd46Sdan ** c = CASE WHEN ?8 THEN ?9 ELSE c END, 3008964cbd46Sdan ** d = CASE WHEN ?11 THEN ?12 ELSE d END 3009d5f0767cSdan ** WHERE a = ?1 AND c = ?7 AND (?13 OR 3010964cbd46Sdan ** (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND 3011d5f0767cSdan ** ) 3012d5f0767cSdan ** 3013d5f0767cSdan ** For each column in the table, there are three variables to bind: 3014d5f0767cSdan ** 3015d5f0767cSdan ** ?(i*3+1) The old.* value of the column, if any. 3016d5f0767cSdan ** ?(i*3+2) A boolean flag indicating that the value is being modified. 3017d5f0767cSdan ** ?(i*3+3) The new.* value of the column, if any. 3018d5f0767cSdan ** 3019d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update 3020d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the 3021d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns 3022d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)". 3023d5f0767cSdan ** 3024296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left 3025296c7658Sdan ** pointing to the prepared version of the SQL statement. 3026d5f0767cSdan */ 3027d5f0767cSdan static int sessionUpdateRow( 3028d5f0767cSdan sqlite3 *db, /* Database handle */ 3029d5f0767cSdan const char *zTab, /* Table name */ 30300c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3031d5f0767cSdan ){ 3032d5f0767cSdan int rc = SQLITE_OK; 3033d5f0767cSdan int i; 3034d5f0767cSdan const char *zSep = ""; 3035d5f0767cSdan SessionBuffer buf = {0, 0, 0}; 3036d5f0767cSdan 3037d5f0767cSdan /* Append "UPDATE tbl SET " */ 3038d5f0767cSdan sessionAppendStr(&buf, "UPDATE ", &rc); 3039d5f0767cSdan sessionAppendIdent(&buf, zTab, &rc); 3040d5f0767cSdan sessionAppendStr(&buf, " SET ", &rc); 3041d5f0767cSdan 3042d5f0767cSdan /* Append the assignments */ 30430c698471Sdan for(i=0; i<p->nCol; i++){ 3044d5f0767cSdan sessionAppendStr(&buf, zSep, &rc); 30450c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3046d5f0767cSdan sessionAppendStr(&buf, " = CASE WHEN ?", &rc); 3047d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3048d5f0767cSdan sessionAppendStr(&buf, " THEN ?", &rc); 3049d5f0767cSdan sessionAppendInteger(&buf, i*3+3, &rc); 3050d5f0767cSdan sessionAppendStr(&buf, " ELSE ", &rc); 30510c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3052d5f0767cSdan sessionAppendStr(&buf, " END", &rc); 3053d5f0767cSdan zSep = ", "; 3054d5f0767cSdan } 3055d5f0767cSdan 3056d5f0767cSdan /* Append the PK part of the WHERE clause */ 3057d5f0767cSdan sessionAppendStr(&buf, " WHERE ", &rc); 30580c698471Sdan for(i=0; i<p->nCol; i++){ 30590c698471Sdan if( p->abPK[i] ){ 30600c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3061d5f0767cSdan sessionAppendStr(&buf, " = ?", &rc); 3062d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3063d5f0767cSdan sessionAppendStr(&buf, " AND ", &rc); 3064d5f0767cSdan } 3065d5f0767cSdan } 3066d5f0767cSdan 3067d5f0767cSdan /* Append the non-PK part of the WHERE clause */ 3068d5f0767cSdan sessionAppendStr(&buf, " (?", &rc); 30690c698471Sdan sessionAppendInteger(&buf, p->nCol*3+1, &rc); 3070d5f0767cSdan sessionAppendStr(&buf, " OR 1", &rc); 30710c698471Sdan for(i=0; i<p->nCol; i++){ 30720c698471Sdan if( !p->abPK[i] ){ 3073d5f0767cSdan sessionAppendStr(&buf, " AND (?", &rc); 3074d5f0767cSdan sessionAppendInteger(&buf, i*3+2, &rc); 3075d5f0767cSdan sessionAppendStr(&buf, "=0 OR ", &rc); 30760c698471Sdan sessionAppendIdent(&buf, p->azCol[i], &rc); 3077d5f0767cSdan sessionAppendStr(&buf, " IS ?", &rc); 3078d5f0767cSdan sessionAppendInteger(&buf, i*3+1, &rc); 3079d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3080d5f0767cSdan } 3081d5f0767cSdan } 3082d5f0767cSdan sessionAppendStr(&buf, ")", &rc); 3083d5f0767cSdan 3084d5f0767cSdan if( rc==SQLITE_OK ){ 30850c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0); 3086d5f0767cSdan } 3087d5f0767cSdan sqlite3_free(buf.aBuf); 3088d5f0767cSdan 3089d5f0767cSdan return rc; 3090d5f0767cSdan } 3091d5f0767cSdan 3092296c7658Sdan /* 3093296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary 3094296c7658Sdan ** key. Assuming the following table structure: 3095296c7658Sdan ** 3096296c7658Sdan ** CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c)); 3097296c7658Sdan ** 3098296c7658Sdan ** The SELECT statement looks like this: 3099296c7658Sdan ** 3100296c7658Sdan ** SELECT * FROM x WHERE a = ?1 AND c = ?3 3101296c7658Sdan ** 3102296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left 3103296c7658Sdan ** pointing to the prepared version of the SQL statement. 3104296c7658Sdan */ 3105d5f0767cSdan static int sessionSelectRow( 3106d5f0767cSdan sqlite3 *db, /* Database handle */ 3107d5f0767cSdan const char *zTab, /* Table name */ 31080c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 3109d5f0767cSdan ){ 3110d7fb7d24Sdan return sessionSelectStmt( 3111d7fb7d24Sdan db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect); 3112d5f0767cSdan } 3113d5f0767cSdan 3114296c7658Sdan /* 3115296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab. 3116296c7658Sdan ** For example: 3117296c7658Sdan ** 3118296c7658Sdan ** INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...); 3119296c7658Sdan ** 3120296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left 3121296c7658Sdan ** pointing to the prepared version of the SQL statement. 3122296c7658Sdan */ 31230c698471Sdan static int sessionInsertRow( 31240c698471Sdan sqlite3 *db, /* Database handle */ 31250c698471Sdan const char *zTab, /* Table name */ 31260c698471Sdan SessionApplyCtx *p /* Session changeset-apply context */ 31270c698471Sdan ){ 31280c698471Sdan int rc = SQLITE_OK; 31290c698471Sdan int i; 31300c698471Sdan SessionBuffer buf = {0, 0, 0}; 31310c698471Sdan 31320c698471Sdan sessionAppendStr(&buf, "INSERT INTO main.", &rc); 31330c698471Sdan sessionAppendIdent(&buf, zTab, &rc); 31340c698471Sdan sessionAppendStr(&buf, " VALUES(?", &rc); 31350c698471Sdan for(i=1; i<p->nCol; i++){ 31360c698471Sdan sessionAppendStr(&buf, ", ?", &rc); 31370c698471Sdan } 31380c698471Sdan sessionAppendStr(&buf, ")", &rc); 31390c698471Sdan 31400c698471Sdan if( rc==SQLITE_OK ){ 31410c698471Sdan rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0); 31420c698471Sdan } 31430c698471Sdan sqlite3_free(buf.aBuf); 31440c698471Sdan return rc; 31450c698471Sdan } 31460c698471Sdan 3147296c7658Sdan /* 31487aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem. 31497aa469cdSdan ** See comments in the body of this function for details. 31507aa469cdSdan */ 31517aa469cdSdan static int sessionBindValue( 31527aa469cdSdan sqlite3_stmt *pStmt, /* Statement to bind value to */ 31537aa469cdSdan int i, /* Parameter number to bind to */ 31547aa469cdSdan sqlite3_value *pVal /* Value to bind */ 31557aa469cdSdan ){ 31565671ef69Sdrh int eType = sqlite3_value_type(pVal); 3157082c96dfSdan /* COVERAGE: The (pVal->z==0) branch is never true using current versions 3158082c96dfSdan ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either 3159082c96dfSdan ** the (pVal->z) variable remains as it was or the type of the value is 3160082c96dfSdan ** set to SQLITE_NULL. */ 31615671ef69Sdrh if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){ 31627aa469cdSdan /* This condition occurs when an earlier OOM in a call to 31637aa469cdSdan ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within 3164082c96dfSdan ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */ 31657aa469cdSdan return SQLITE_NOMEM; 31667aa469cdSdan } 31677aa469cdSdan return sqlite3_bind_value(pStmt, i, pVal); 31687aa469cdSdan } 31697aa469cdSdan 31707aa469cdSdan /* 3171db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function 3172db04571cSdan ** transfers new.* values from the current iterator entry to statement 3173db04571cSdan ** pStmt. The table being inserted into has nCol columns. 3174db04571cSdan ** 3175db04571cSdan ** New.* value $i 0 from the iterator is bound to variable ($i+1) of 3176db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1) 3177db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points 3178db04571cSdan ** to an array nCol elements in size. In this case only those values for 3179db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the 3180db04571cSdan ** statement. 3181db04571cSdan ** 3182db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK. 3183db04571cSdan */ 31847aa469cdSdan static int sessionBindRow( 3185db04571cSdan sqlite3_changeset_iter *pIter, /* Iterator to read values from */ 31867aa469cdSdan int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **), 3187db04571cSdan int nCol, /* Number of columns */ 3188db04571cSdan u8 *abPK, /* If not NULL, bind only if true */ 3189db04571cSdan sqlite3_stmt *pStmt /* Bind values to this statement */ 3190db04571cSdan ){ 3191db04571cSdan int i; 3192db04571cSdan int rc = SQLITE_OK; 31937aa469cdSdan 31947aa469cdSdan /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the 31957aa469cdSdan ** argument iterator points to a suitable entry. Make sure that xValue 31967aa469cdSdan ** is one of these to guarantee that it is safe to ignore the return 31977aa469cdSdan ** in the code below. */ 31987aa469cdSdan assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new ); 31997aa469cdSdan 3200db04571cSdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 3201db04571cSdan if( !abPK || abPK[i] ){ 3202db04571cSdan sqlite3_value *pVal; 32037aa469cdSdan (void)xValue(pIter, i, &pVal); 32047aa469cdSdan rc = sessionBindValue(pStmt, i+1, pVal); 3205db04571cSdan } 3206db04571cSdan } 3207db04571cSdan return rc; 3208db04571cSdan } 3209db04571cSdan 3210db04571cSdan /* 3211296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function. 3212296c7658Sdan ** This function binds the primary key values from the change that changeset 3213296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table 3214296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row 3215296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error 3216296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an 32177aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned. 32187aa469cdSdan ** 32197aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset() 32207aa469cdSdan ** statement pSelect. If any other value is returned, the statement does 32217aa469cdSdan ** not require a reset(). 3222296c7658Sdan ** 3223296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the 3224db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or 3225db04571cSdan ** UPDATE, bind values from the old.* record. 3226296c7658Sdan */ 32270c698471Sdan static int sessionSeekToRow( 322837f133ecSdan sqlite3 *db, /* Database handle */ 322937f133ecSdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 323037f133ecSdan u8 *abPK, /* Primary key flags array */ 32310c698471Sdan sqlite3_stmt *pSelect /* SELECT statement from sessionSelectRow() */ 323237f133ecSdan ){ 32337aa469cdSdan int rc; /* Return code */ 3234296c7658Sdan int nCol; /* Number of columns in table */ 3235296c7658Sdan int op; /* Changset operation (SQLITE_UPDATE etc.) */ 3236296c7658Sdan const char *zDummy; /* Unused */ 323737f133ecSdan 3238b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 32397aa469cdSdan rc = sessionBindRow(pIter, 3240db04571cSdan op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old, 3241db04571cSdan nCol, abPK, pSelect 3242db04571cSdan ); 32430c698471Sdan 32440c698471Sdan if( rc==SQLITE_OK ){ 32450c698471Sdan rc = sqlite3_step(pSelect); 32460c698471Sdan if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect); 32470c698471Sdan } 32480c698471Sdan 32490c698471Sdan return rc; 32500c698471Sdan } 32510c698471Sdan 3252296c7658Sdan /* 3253296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator 3254296c7658Sdan ** currently points to. 3255296c7658Sdan ** 3256296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT. 3257296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked 3258296c7658Sdan ** depends solely on eType, as follows: 3259296c7658Sdan ** 3260296c7658Sdan ** eType value Value passed to xConflict 3261296c7658Sdan ** ------------------------------------------------- 3262296c7658Sdan ** CHANGESET_DATA CHANGESET_NOTFOUND 3263296c7658Sdan ** CHANGESET_CONFLICT CHANGESET_CONSTRAINT 3264296c7658Sdan ** 3265296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing 3266296c7658Sdan ** record with the same primary key as the record about to be deleted, updated 3267296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict 3268296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict 3269296c7658Sdan ** handler invoked is as follows: 3270296c7658Sdan ** 3271296c7658Sdan ** eType value PK Record found? Value passed to xConflict 3272296c7658Sdan ** ---------------------------------------------------------------- 3273296c7658Sdan ** CHANGESET_DATA Yes CHANGESET_DATA 3274296c7658Sdan ** CHANGESET_DATA No CHANGESET_NOTFOUND 3275296c7658Sdan ** CHANGESET_CONFLICT Yes CHANGESET_CONFLICT 3276296c7658Sdan ** CHANGESET_CONFLICT No CHANGESET_CONSTRAINT 3277296c7658Sdan ** 3278296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and 3279296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace 3280296c7658Sdan ** is set to non-zero before returning SQLITE_OK. 3281296c7658Sdan ** 3282296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is 3283296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value, 3284296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT, 3285296c7658Sdan ** this function returns SQLITE_OK. 3286296c7658Sdan */ 32870c698471Sdan static int sessionConflictHandler( 3288296c7658Sdan int eType, /* Either CHANGESET_DATA or CONFLICT */ 3289296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 32900c698471Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 32910c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter*), 3292296c7658Sdan void *pCtx, /* First argument for conflict handler */ 3293296c7658Sdan int *pbReplace /* OUT: Set to true if PK row is found */ 32940c698471Sdan ){ 3295296c7658Sdan int res; /* Value returned by conflict handler */ 32960c698471Sdan int rc; 32970c698471Sdan int nCol; 32980c698471Sdan int op; 32990c698471Sdan const char *zDummy; 33000c698471Sdan 3301b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 33020c698471Sdan 33030c698471Sdan assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA ); 33040c698471Sdan assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT ); 33050c698471Sdan assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND ); 330637f133ecSdan 330737f133ecSdan /* Bind the new.* PRIMARY KEY values to the SELECT statement. */ 33080c698471Sdan if( pbReplace ){ 33090c698471Sdan rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect); 33100c698471Sdan }else{ 3311db04571cSdan rc = SQLITE_OK; 33120c698471Sdan } 33130c698471Sdan 33140c698471Sdan if( rc==SQLITE_ROW ){ 33150c698471Sdan /* There exists another row with the new.* primary key. */ 33160c698471Sdan pIter->pConflict = p->pSelect; 33170c698471Sdan res = xConflict(pCtx, eType, pIter); 33180c698471Sdan pIter->pConflict = 0; 33190c698471Sdan rc = sqlite3_reset(p->pSelect); 3320db04571cSdan }else if( rc==SQLITE_OK ){ 33210c698471Sdan /* No other row with the new.* primary key. */ 33220c698471Sdan res = xConflict(pCtx, eType+1, pIter); 33230c698471Sdan if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE; 332437f133ecSdan } 332537f133ecSdan 332637f133ecSdan if( rc==SQLITE_OK ){ 33270c698471Sdan switch( res ){ 33280c698471Sdan case SQLITE_CHANGESET_REPLACE: 3329f51e5f6cSdan assert( pbReplace ); 3330f51e5f6cSdan *pbReplace = 1; 33310c698471Sdan break; 33320c698471Sdan 33330c698471Sdan case SQLITE_CHANGESET_OMIT: 33340c698471Sdan break; 33350c698471Sdan 33360c698471Sdan case SQLITE_CHANGESET_ABORT: 33370c698471Sdan rc = SQLITE_ABORT; 33380c698471Sdan break; 33390c698471Sdan 33400c698471Sdan default: 33410c698471Sdan rc = SQLITE_MISUSE; 33420c698471Sdan break; 33430c698471Sdan } 33440c698471Sdan } 33450c698471Sdan 33460c698471Sdan return rc; 33470c698471Sdan } 33480c698471Sdan 3349296c7658Sdan /* 3350296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument 3351296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke 3352296c7658Sdan ** the conflict handler callback. 3353296c7658Sdan ** 3354296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If 3355296c7658Sdan ** one is encountered, update or delete the row with the matching primary key 3356296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs, 3357296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry 3358296c7658Sdan ** to true before returning. In this case the caller will invoke this function 3359296c7658Sdan ** again, this time with pbRetry set to NULL. 3360296c7658Sdan ** 3361296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is 3362296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead. 3363296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such 3364296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true 3365296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting 3366296c7658Sdan ** row before invoking this function again, this time with pbReplace set 3367296c7658Sdan ** to NULL. 3368296c7658Sdan ** 3369296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function 3370296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is 3371296c7658Sdan ** returned. 3372296c7658Sdan */ 33730c698471Sdan static int sessionApplyOneOp( 3374296c7658Sdan sqlite3_changeset_iter *pIter, /* Changeset iterator */ 3375296c7658Sdan SessionApplyCtx *p, /* changeset_apply() context */ 33760c698471Sdan int(*xConflict)(void *, int, sqlite3_changeset_iter *), 3377296c7658Sdan void *pCtx, /* First argument for the conflict handler */ 3378296c7658Sdan int *pbReplace, /* OUT: True to remove PK row and retry */ 3379296c7658Sdan int *pbRetry /* OUT: True to retry. */ 33800c698471Sdan ){ 33810c698471Sdan const char *zDummy; 33820c698471Sdan int op; 33830c698471Sdan int nCol; 33840c698471Sdan int rc = SQLITE_OK; 33850c698471Sdan 33860c698471Sdan assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect ); 33870c698471Sdan assert( p->azCol && p->abPK ); 33880c698471Sdan assert( !pbReplace || *pbReplace==0 ); 33890c698471Sdan 3390b4480e94Sdan sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0); 33910c698471Sdan 33920c698471Sdan if( op==SQLITE_DELETE ){ 33930c698471Sdan 339473b3c055Sdan /* Bind values to the DELETE statement. If conflict handling is required, 339573b3c055Sdan ** bind values for all columns and set bound variable (nCol+1) to true. 339673b3c055Sdan ** Or, if conflict handling is not required, bind just the PK column 339773b3c055Sdan ** values and, if it exists, set (nCol+1) to false. Conflict handling 339873b3c055Sdan ** is not required if: 339973b3c055Sdan ** 340073b3c055Sdan ** * this is a patchset, or 340173b3c055Sdan ** * (pbRetry==0), or 340273b3c055Sdan ** * all columns of the table are PK columns (in this case there is 340373b3c055Sdan ** no (nCol+1) variable to bind to). 340473b3c055Sdan */ 340573b3c055Sdan u8 *abPK = (pIter->bPatchset ? p->abPK : 0); 340673b3c055Sdan rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete); 34077cf7df7dSdan if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){ 340873b3c055Sdan rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK)); 34097cf7df7dSdan } 34100c698471Sdan if( rc!=SQLITE_OK ) return rc; 34110c698471Sdan 34120c698471Sdan sqlite3_step(p->pDelete); 34130c698471Sdan rc = sqlite3_reset(p->pDelete); 34140c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 34150c698471Sdan rc = sessionConflictHandler( 34160c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 34170c698471Sdan ); 341835e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 34190c698471Sdan rc = sessionConflictHandler( 34200c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 34210c698471Sdan ); 34220c698471Sdan } 34230c698471Sdan 34240c698471Sdan }else if( op==SQLITE_UPDATE ){ 34250c698471Sdan int i; 34260c698471Sdan 34270c698471Sdan /* Bind values to the UPDATE statement. */ 34280c698471Sdan for(i=0; rc==SQLITE_OK && i<nCol; i++){ 34297aa469cdSdan sqlite3_value *pOld = sessionChangesetOld(pIter, i); 34307aa469cdSdan sqlite3_value *pNew = sessionChangesetNew(pIter, i); 34317aa469cdSdan 34320c698471Sdan sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew); 34337aa469cdSdan if( pOld ){ 34347aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+1, pOld); 34357aa469cdSdan } 34367aa469cdSdan if( rc==SQLITE_OK && pNew ){ 34377aa469cdSdan rc = sessionBindValue(p->pUpdate, i*3+3, pNew); 34380c698471Sdan } 34390c698471Sdan } 344073b3c055Sdan if( rc==SQLITE_OK ){ 344173b3c055Sdan sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset); 344273b3c055Sdan } 34430c698471Sdan if( rc!=SQLITE_OK ) return rc; 34440c698471Sdan 34450c698471Sdan /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict, 34460c698471Sdan ** the result will be SQLITE_OK with 0 rows modified. */ 34470c698471Sdan sqlite3_step(p->pUpdate); 34480c698471Sdan rc = sqlite3_reset(p->pUpdate); 34490c698471Sdan 34500c698471Sdan if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){ 34510c698471Sdan /* A NOTFOUND or DATA error. Search the table to see if it contains 34520c698471Sdan ** a row with a matching primary key. If so, this is a DATA conflict. 34530c698471Sdan ** Otherwise, if there is no primary key match, it is a NOTFOUND. */ 34540c698471Sdan 34550c698471Sdan rc = sessionConflictHandler( 34560c698471Sdan SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry 34570c698471Sdan ); 34580c698471Sdan 345935e2858eSdan }else if( (rc&0xff)==SQLITE_CONSTRAINT ){ 3460db04571cSdan /* This is always a CONSTRAINT conflict. */ 3461db04571cSdan rc = sessionConflictHandler( 3462db04571cSdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0 34630c698471Sdan ); 34640c698471Sdan } 34650c698471Sdan 34660c698471Sdan }else{ 34670c698471Sdan assert( op==SQLITE_INSERT ); 34687aa469cdSdan rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert); 34690c698471Sdan if( rc!=SQLITE_OK ) return rc; 34700c698471Sdan 34710c698471Sdan sqlite3_step(p->pInsert); 34720c698471Sdan rc = sqlite3_reset(p->pInsert); 347335e2858eSdan if( (rc&0xff)==SQLITE_CONSTRAINT ){ 34740c698471Sdan rc = sessionConflictHandler( 34750c698471Sdan SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace 34760c698471Sdan ); 347737f133ecSdan } 347837f133ecSdan } 347937f133ecSdan 348037f133ecSdan return rc; 348137f133ecSdan } 348237f133ecSdan 3483296c7658Sdan /* 34844757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but 34854757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the 34864757c658Sdan ** changeset to the main database attached to handle "db". The supplied 34874757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered 34884757c658Sdan ** while applying the change. 3489296c7658Sdan */ 34904757c658Sdan static int sessionChangesetApply( 3491296c7658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 34924757c658Sdan sqlite3_changeset_iter *pIter, /* Changeset to apply */ 349340368988Sdan int(*xFilter)( 349440368988Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 349540368988Sdan const char *zTab /* Table name */ 349640368988Sdan ), 3497d5f0767cSdan int(*xConflict)( 3498d5f0767cSdan void *pCtx, /* Copy of fifth arg to _apply() */ 3499d5f0767cSdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 3500d5f0767cSdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 3501d5f0767cSdan ), 3502296c7658Sdan void *pCtx /* First argument passed to xConflict */ 3503d5f0767cSdan ){ 3504ca62ad57Sdan int schemaMismatch = 0; 3505296c7658Sdan int rc; /* Return code */ 3506d5f0767cSdan const char *zTab = 0; /* Name of current table */ 3507cfdbde21Sdrh int nTab = 0; /* Result of sqlite3Strlen30(zTab) */ 3508296c7658Sdan SessionApplyCtx sApply; /* changeset_apply() context object */ 3509d5f0767cSdan 3510082c96dfSdan assert( xConflict!=0 ); 3511082c96dfSdan 35120c698471Sdan memset(&sApply, 0, sizeof(sApply)); 35134c220252Sdan sqlite3_mutex_enter(sqlite3_db_mutex(db)); 35140c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0); 3515cb3e4b79Sdan if( rc==SQLITE_OK ){ 3516cb3e4b79Sdan rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0); 3517cb3e4b79Sdan } 35180c698471Sdan while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){ 35190c698471Sdan int nCol; 3520d5f0767cSdan int op; 35210c698471Sdan int bReplace = 0; 35220c698471Sdan int bRetry = 0; 35230c698471Sdan const char *zNew; 3524ca62ad57Sdan 3525b4480e94Sdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0); 3526d5f0767cSdan 35270c698471Sdan if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){ 3528ca62ad57Sdan u8 *abPK; 3529ca62ad57Sdan 3530cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 35310c698471Sdan sqlite3_finalize(sApply.pDelete); 35320c698471Sdan sqlite3_finalize(sApply.pUpdate); 35330c698471Sdan sqlite3_finalize(sApply.pInsert); 35340c698471Sdan sqlite3_finalize(sApply.pSelect); 35350c698471Sdan memset(&sApply, 0, sizeof(sApply)); 35360c698471Sdan sApply.db = db; 353737f133ecSdan 353840368988Sdan /* If an xFilter() callback was specified, invoke it now. If the 353940368988Sdan ** xFilter callback returns zero, skip this table. If it returns 354040368988Sdan ** non-zero, proceed. */ 354140368988Sdan schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew))); 354240368988Sdan if( schemaMismatch ){ 354340368988Sdan zTab = sqlite3_mprintf("%s", zNew); 3544f05ac112Sdan if( zTab==0 ){ 3545f05ac112Sdan rc = SQLITE_NOMEM; 3546f05ac112Sdan break; 3547f05ac112Sdan } 35484f528042Sdan nTab = (int)strlen(zTab); 354940368988Sdan sApply.azCol = (const char **)zTab; 355040368988Sdan }else{ 3551ca62ad57Sdan sqlite3changeset_pk(pIter, &abPK, 0); 3552296c7658Sdan rc = sessionTableInfo( 3553ca62ad57Sdan db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK 3554ca62ad57Sdan ); 3555ca62ad57Sdan if( rc!=SQLITE_OK ) break; 35560c698471Sdan 3557ca62ad57Sdan if( sApply.nCol==0 ){ 3558ca62ad57Sdan schemaMismatch = 1; 3559ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 3560ca62ad57Sdan "sqlite3changeset_apply(): no such table: %s", zTab 3561ca62ad57Sdan ); 3562ca62ad57Sdan } 3563ca62ad57Sdan else if( sApply.nCol!=nCol ){ 3564ca62ad57Sdan schemaMismatch = 1; 3565ca62ad57Sdan sqlite3_log(SQLITE_SCHEMA, 3566ca62ad57Sdan "sqlite3changeset_apply(): table %s has %d columns, expected %d", 3567ca62ad57Sdan zTab, sApply.nCol, nCol 3568ca62ad57Sdan ); 3569ca62ad57Sdan } 3570ca62ad57Sdan else if( memcmp(sApply.abPK, abPK, nCol)!=0 ){ 3571ca62ad57Sdan schemaMismatch = 1; 357240368988Sdan sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): " 357340368988Sdan "primary key mismatch for table %s", zTab 3574ca62ad57Sdan ); 3575ca62ad57Sdan } 3576ca62ad57Sdan else if( 3577ca62ad57Sdan (rc = sessionSelectRow(db, zTab, &sApply)) 35780c698471Sdan || (rc = sessionUpdateRow(db, zTab, &sApply)) 35790c698471Sdan || (rc = sessionDeleteRow(db, zTab, &sApply)) 35800c698471Sdan || (rc = sessionInsertRow(db, zTab, &sApply)) 358137f133ecSdan ){ 358237f133ecSdan break; 358337f133ecSdan } 3584cfdbde21Sdrh nTab = sqlite3Strlen30(zTab); 3585d5f0767cSdan } 358640368988Sdan } 3587d5f0767cSdan 3588ca62ad57Sdan /* If there is a schema mismatch on the current table, proceed to the 3589ca62ad57Sdan ** next change. A log message has already been issued. */ 3590ca62ad57Sdan if( schemaMismatch ) continue; 3591ca62ad57Sdan 35920c698471Sdan rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, &bRetry); 35930c698471Sdan 35940c698471Sdan if( rc==SQLITE_OK && bRetry ){ 35950c698471Sdan rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, 0); 35960c698471Sdan } 35970c698471Sdan 35980c698471Sdan if( bReplace ){ 3599db04571cSdan assert( pIter->op==SQLITE_INSERT ); 36000c698471Sdan rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0); 36010c698471Sdan if( rc==SQLITE_OK ){ 36027aa469cdSdan rc = sessionBindRow(pIter, 3603db04571cSdan sqlite3changeset_new, sApply.nCol, sApply.abPK, sApply.pDelete); 36040c698471Sdan sqlite3_bind_int(sApply.pDelete, sApply.nCol+1, 1); 36050c698471Sdan } 36060c698471Sdan if( rc==SQLITE_OK ){ 36070c698471Sdan sqlite3_step(sApply.pDelete); 36080c698471Sdan rc = sqlite3_reset(sApply.pDelete); 36090c698471Sdan } 36100c698471Sdan if( rc==SQLITE_OK ){ 36110c698471Sdan rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, 0, 0); 36120c698471Sdan } 36130c698471Sdan if( rc==SQLITE_OK ){ 36140c698471Sdan rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0); 36150c698471Sdan } 36160c698471Sdan } 36170c698471Sdan } 3618d5f0767cSdan 3619296c7658Sdan if( rc==SQLITE_OK ){ 3620296c7658Sdan rc = sqlite3changeset_finalize(pIter); 3621296c7658Sdan }else{ 3622296c7658Sdan sqlite3changeset_finalize(pIter); 3623296c7658Sdan } 3624d5f0767cSdan 3625d5f0767cSdan if( rc==SQLITE_OK ){ 362607001c45Sdrh int nFk, notUsed; 362707001c45Sdrh sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, ¬Used, 0); 362807001c45Sdrh if( nFk!=0 ){ 3629cb3e4b79Sdan int res = SQLITE_CHANGESET_ABORT; 3630cb3e4b79Sdan sqlite3_changeset_iter sIter; 3631cb3e4b79Sdan memset(&sIter, 0, sizeof(sIter)); 3632cb3e4b79Sdan sIter.nCol = nFk; 3633cb3e4b79Sdan res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter); 3634cb3e4b79Sdan if( res!=SQLITE_CHANGESET_OMIT ){ 3635cb3e4b79Sdan rc = SQLITE_CONSTRAINT; 3636cb3e4b79Sdan } 3637cb3e4b79Sdan } 3638cb3e4b79Sdan } 3639cb3e4b79Sdan sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0); 3640cb3e4b79Sdan 3641cb3e4b79Sdan if( rc==SQLITE_OK ){ 3642d5f0767cSdan rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 3643d5f0767cSdan }else{ 3644d5f0767cSdan sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0); 3645d5f0767cSdan sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0); 3646d5f0767cSdan } 3647d5f0767cSdan 36480c698471Sdan sqlite3_finalize(sApply.pInsert); 36490c698471Sdan sqlite3_finalize(sApply.pDelete); 36500c698471Sdan sqlite3_finalize(sApply.pUpdate); 36510c698471Sdan sqlite3_finalize(sApply.pSelect); 3652cfdbde21Sdrh sqlite3_free((char*)sApply.azCol); /* cast works around VC++ bug */ 36534c220252Sdan sqlite3_mutex_leave(sqlite3_db_mutex(db)); 3654d5f0767cSdan return rc; 3655d5f0767cSdan } 365691ddd559Sdan 365777fc1d5bSdan /* 36584757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database 36594757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 36604757c658Sdan ** to resolve any conflicts encountered while applying the change. 36614757c658Sdan */ 36624757c658Sdan int sqlite3changeset_apply( 36634757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 36644757c658Sdan int nChangeset, /* Size of changeset in bytes */ 36654757c658Sdan void *pChangeset, /* Changeset blob */ 36664757c658Sdan int(*xFilter)( 36674757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 36684757c658Sdan const char *zTab /* Table name */ 36694757c658Sdan ), 36704757c658Sdan int(*xConflict)( 36714757c658Sdan void *pCtx, /* Copy of fifth arg to _apply() */ 36724757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 36734757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 36744757c658Sdan ), 36754757c658Sdan void *pCtx /* First argument passed to xConflict */ 36764757c658Sdan ){ 36774757c658Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 36784757c658Sdan int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset); 36794757c658Sdan if( rc==SQLITE_OK ){ 36804757c658Sdan rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx); 36814757c658Sdan } 36824757c658Sdan return rc; 36834757c658Sdan } 36844757c658Sdan 36854757c658Sdan /* 36864757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database 36874757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback 36884757c658Sdan ** to resolve any conflicts encountered while applying the change. 36894757c658Sdan */ 3690f1a08ad8Sdrh int sqlite3changeset_apply_strm( 36914757c658Sdan sqlite3 *db, /* Apply change to "main" db of this handle */ 36924757c658Sdan int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */ 36934757c658Sdan void *pIn, /* First arg for xInput */ 36944757c658Sdan int(*xFilter)( 36954757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 36964757c658Sdan const char *zTab /* Table name */ 36974757c658Sdan ), 36984757c658Sdan int(*xConflict)( 36994757c658Sdan void *pCtx, /* Copy of sixth arg to _apply() */ 37004757c658Sdan int eConflict, /* DATA, MISSING, CONFLICT, CONSTRAINT */ 37014757c658Sdan sqlite3_changeset_iter *p /* Handle describing change and conflict */ 37024757c658Sdan ), 37034757c658Sdan void *pCtx /* First argument passed to xConflict */ 37044757c658Sdan ){ 37054757c658Sdan sqlite3_changeset_iter *pIter; /* Iterator to skip through changeset */ 3706f1a08ad8Sdrh int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn); 37074757c658Sdan if( rc==SQLITE_OK ){ 37084757c658Sdan rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx); 37094757c658Sdan } 37104757c658Sdan return rc; 37114757c658Sdan } 37124757c658Sdan 37134757c658Sdan /* 371477fc1d5bSdan ** This function is called to merge two changes to the same row together as 371577fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is 371677fc1d5bSdan ** allocated and a pointer to it stored in *ppNew. 371777fc1d5bSdan */ 37185d607a6eSdan static int sessionChangeMerge( 371977fc1d5bSdan SessionTable *pTab, /* Table structure */ 372064277f4aSdan int bPatchset, /* True for patchsets */ 372177fc1d5bSdan SessionChange *pExist, /* Existing change */ 372277fc1d5bSdan int op2, /* Second change operation */ 372377fc1d5bSdan int bIndirect, /* True if second change is indirect */ 372477fc1d5bSdan u8 *aRec, /* Second change record */ 372577fc1d5bSdan int nRec, /* Number of bytes in aRec */ 372677fc1d5bSdan SessionChange **ppNew /* OUT: Merged change */ 37275d607a6eSdan ){ 37285d607a6eSdan SessionChange *pNew = 0; 37295d607a6eSdan 37305d607a6eSdan if( !pExist ){ 3731cbf6d2d2Sdan pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec); 37325d607a6eSdan if( !pNew ){ 37335d607a6eSdan return SQLITE_NOMEM; 37345d607a6eSdan } 37355d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 3736798693b2Sdan pNew->op = op2; 37375d607a6eSdan pNew->bIndirect = bIndirect; 37385d607a6eSdan pNew->nRecord = nRec; 3739cbf6d2d2Sdan pNew->aRecord = (u8*)&pNew[1]; 3740cbf6d2d2Sdan memcpy(pNew->aRecord, aRec, nRec); 37415d607a6eSdan }else{ 3742798693b2Sdan int op1 = pExist->op; 37435d607a6eSdan 37445d607a6eSdan /* 37455d607a6eSdan ** op1=INSERT, op2=INSERT -> Unsupported. Discard op2. 37465d607a6eSdan ** op1=INSERT, op2=UPDATE -> INSERT. 37475d607a6eSdan ** op1=INSERT, op2=DELETE -> (none) 37485d607a6eSdan ** 37495d607a6eSdan ** op1=UPDATE, op2=INSERT -> Unsupported. Discard op2. 37505d607a6eSdan ** op1=UPDATE, op2=UPDATE -> UPDATE. 37515d607a6eSdan ** op1=UPDATE, op2=DELETE -> DELETE. 37525d607a6eSdan ** 37535d607a6eSdan ** op1=DELETE, op2=INSERT -> UPDATE. 37545d607a6eSdan ** op1=DELETE, op2=UPDATE -> Unsupported. Discard op2. 37555d607a6eSdan ** op1=DELETE, op2=DELETE -> Unsupported. Discard op2. 37565d607a6eSdan */ 37575d607a6eSdan if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT) 37585d607a6eSdan || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT) 37595d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE) 37605d607a6eSdan || (op1==SQLITE_DELETE && op2==SQLITE_DELETE) 37615d607a6eSdan ){ 37625d607a6eSdan pNew = pExist; 37635d607a6eSdan }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){ 37645d607a6eSdan sqlite3_free(pExist); 37655d607a6eSdan assert( pNew==0 ); 37665d607a6eSdan }else{ 376764277f4aSdan u8 *aExist = pExist->aRecord; 37685d607a6eSdan int nByte; 37695d607a6eSdan u8 *aCsr; 37705d607a6eSdan 377164277f4aSdan /* Allocate a new SessionChange object. Ensure that the aRecord[] 377264277f4aSdan ** buffer of the new object is large enough to hold any record that 377364277f4aSdan ** may be generated by combining the input records. */ 37745d607a6eSdan nByte = sizeof(SessionChange) + pExist->nRecord + nRec; 37755d607a6eSdan pNew = (SessionChange *)sqlite3_malloc(nByte); 37765d607a6eSdan if( !pNew ){ 37771756ae10Sdan sqlite3_free(pExist); 37785d607a6eSdan return SQLITE_NOMEM; 37795d607a6eSdan } 37805d607a6eSdan memset(pNew, 0, sizeof(SessionChange)); 37815d607a6eSdan pNew->bIndirect = (bIndirect && pExist->bIndirect); 37825d607a6eSdan aCsr = pNew->aRecord = (u8 *)&pNew[1]; 37835d607a6eSdan 3784b08a1efaSdan if( op1==SQLITE_INSERT ){ /* INSERT + UPDATE */ 37855d607a6eSdan u8 *a1 = aRec; 3786b08a1efaSdan assert( op2==SQLITE_UPDATE ); 3787798693b2Sdan pNew->op = SQLITE_INSERT; 3788ef7a6304Sdan if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol); 378964277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1); 3790b08a1efaSdan }else if( op1==SQLITE_DELETE ){ /* DELETE + INSERT */ 3791b08a1efaSdan assert( op2==SQLITE_INSERT ); 3792798693b2Sdan pNew->op = SQLITE_UPDATE; 3793fa29ecc4Sdan if( bPatchset ){ 3794fa29ecc4Sdan memcpy(aCsr, aRec, nRec); 3795fa29ecc4Sdan aCsr += nRec; 3796fa29ecc4Sdan }else{ 379764277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){ 3798b08a1efaSdan sqlite3_free(pNew); 3799b08a1efaSdan pNew = 0; 38005d607a6eSdan } 3801fa29ecc4Sdan } 3802b08a1efaSdan }else if( op2==SQLITE_UPDATE ){ /* UPDATE + UPDATE */ 380364277f4aSdan u8 *a1 = aExist; 38045d607a6eSdan u8 *a2 = aRec; 3805cfec7eeeSdan assert( op1==SQLITE_UPDATE ); 380664277f4aSdan if( bPatchset==0 ){ 3807ef7a6304Sdan sessionSkipRecord(&a1, pTab->nCol); 3808ef7a6304Sdan sessionSkipRecord(&a2, pTab->nCol); 380964277f4aSdan } 3810798693b2Sdan pNew->op = SQLITE_UPDATE; 381164277f4aSdan if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){ 38125d607a6eSdan sqlite3_free(pNew); 38135d607a6eSdan pNew = 0; 38145d607a6eSdan } 3815b08a1efaSdan }else{ /* UPDATE + DELETE */ 3816b08a1efaSdan assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE ); 3817798693b2Sdan pNew->op = SQLITE_DELETE; 381864277f4aSdan if( bPatchset ){ 381964277f4aSdan memcpy(aCsr, aRec, nRec); 382064277f4aSdan aCsr += nRec; 382164277f4aSdan }else{ 382264277f4aSdan sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist); 382364277f4aSdan } 38245d607a6eSdan } 38255d607a6eSdan 38265d607a6eSdan if( pNew ){ 38274f528042Sdan pNew->nRecord = (int)(aCsr - pNew->aRecord); 38285d607a6eSdan } 38295d607a6eSdan sqlite3_free(pExist); 38305d607a6eSdan } 38315d607a6eSdan } 38325d607a6eSdan 38335d607a6eSdan *ppNew = pNew; 38345d607a6eSdan return SQLITE_OK; 38355d607a6eSdan } 38365d607a6eSdan 383777fc1d5bSdan /* 383877fc1d5bSdan ** Add all changes in the changeset passed via the first two arguments to 383977fc1d5bSdan ** hash tables. 384077fc1d5bSdan */ 384116228167Sdan static int sessionChangesetToHash( 3842cbf6d2d2Sdan sqlite3_changeset_iter *pIter, /* Iterator to read from */ 384377fc1d5bSdan SessionTable **ppTabList /* IN/OUT: List of table objects */ 38445d607a6eSdan ){ 38455d607a6eSdan u8 *aRec; 38465d607a6eSdan int nRec; 3847cbf6d2d2Sdan int rc = SQLITE_OK; 38485d607a6eSdan SessionTable *pTab = 0; 38495d607a6eSdan 38505d607a6eSdan while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec) ){ 38515d607a6eSdan const char *zNew; 38525d607a6eSdan int nCol; 38535d607a6eSdan int op; 38545d607a6eSdan int iHash; 38555d607a6eSdan int bIndirect; 38565d607a6eSdan SessionChange *pChange; 38575d607a6eSdan SessionChange *pExist = 0; 38585d607a6eSdan SessionChange **pp; 38595d607a6eSdan 38605d607a6eSdan sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect); 3861ef7a6304Sdan if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){ 38625d607a6eSdan /* Search the list for a matching table */ 38634f528042Sdan int nNew = (int)strlen(zNew); 3864f29123b5Sdan u8 *abPK; 3865f29123b5Sdan 3866f29123b5Sdan sqlite3changeset_pk(pIter, &abPK, 0); 38675d607a6eSdan for(pTab = *ppTabList; pTab; pTab=pTab->pNext){ 38685d607a6eSdan if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break; 38695d607a6eSdan } 38705d607a6eSdan if( !pTab ){ 3871ef7a6304Sdan pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1); 3872f29123b5Sdan if( !pTab ){ 3873f29123b5Sdan rc = SQLITE_NOMEM; 3874f29123b5Sdan break; 3875f29123b5Sdan } 38765d607a6eSdan memset(pTab, 0, sizeof(SessionTable)); 38775d607a6eSdan pTab->pNext = *ppTabList; 3878f29123b5Sdan pTab->nCol = nCol; 3879ef7a6304Sdan pTab->abPK = (u8*)&pTab[1]; 3880ef7a6304Sdan memcpy(pTab->abPK, abPK, nCol); 3881ef7a6304Sdan pTab->zName = (char*)&pTab->abPK[nCol]; 3882ef7a6304Sdan memcpy(pTab->zName, zNew, nNew+1); 38835d607a6eSdan *ppTabList = pTab; 3884f29123b5Sdan }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){ 3885f29123b5Sdan rc = SQLITE_SCHEMA; 3886f29123b5Sdan break; 38875d607a6eSdan } 38885d607a6eSdan } 38895d607a6eSdan 3890cbf6d2d2Sdan if( sessionGrowHash(pIter->bPatchset, pTab) ){ 38911756ae10Sdan rc = SQLITE_NOMEM; 38921756ae10Sdan break; 38931756ae10Sdan } 389464277f4aSdan iHash = sessionChangeHash( 3895cbf6d2d2Sdan pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange 389664277f4aSdan ); 38975d607a6eSdan 38985d607a6eSdan /* Search for existing entry. If found, remove it from the hash table. 38995d607a6eSdan ** Code below may link it back in. 39005d607a6eSdan */ 39015d607a6eSdan for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){ 390264277f4aSdan int bPkOnly1 = 0; 390364277f4aSdan int bPkOnly2 = 0; 3904cbf6d2d2Sdan if( pIter->bPatchset ){ 390564277f4aSdan bPkOnly1 = (*pp)->op==SQLITE_DELETE; 390664277f4aSdan bPkOnly2 = op==SQLITE_DELETE; 390764277f4aSdan } 390864277f4aSdan if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){ 39095d607a6eSdan pExist = *pp; 39105d607a6eSdan *pp = (*pp)->pNext; 39115d607a6eSdan pTab->nEntry--; 39125d607a6eSdan break; 39135d607a6eSdan } 39145d607a6eSdan } 39155d607a6eSdan 391664277f4aSdan rc = sessionChangeMerge(pTab, 3917cbf6d2d2Sdan pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange 391864277f4aSdan ); 39195d607a6eSdan if( rc ) break; 39205d607a6eSdan if( pChange ){ 39215d607a6eSdan pChange->pNext = pTab->apChange[iHash]; 39225d607a6eSdan pTab->apChange[iHash] = pChange; 39235d607a6eSdan pTab->nEntry++; 39245d607a6eSdan } 39255d607a6eSdan } 39265d607a6eSdan 3927cbf6d2d2Sdan if( rc==SQLITE_OK ) rc = pIter->rc; 39285d607a6eSdan return rc; 39295d607a6eSdan } 39305d607a6eSdan 39315d607a6eSdan /* 39325d607a6eSdan ** 1. Iterate through the left-hand changeset. Add an entry to a table 39335d607a6eSdan ** specific hash table for each change in the changeset. The hash table 39345d607a6eSdan ** key is the PK of the row affected by the change. 39355d607a6eSdan ** 39365d607a6eSdan ** 2. Then interate through the right-hand changeset. Attempt to add an 39375d607a6eSdan ** entry to a hash table for each component change. If a change already 39385d607a6eSdan ** exists with the same PK values, combine the two into a single change. 39395d607a6eSdan ** 39405d607a6eSdan ** 3. Write an output changeset based on the contents of the hash table. 39415d607a6eSdan */ 3942cbf6d2d2Sdan int sessionChangesetConcat( 3943cbf6d2d2Sdan sqlite3_changeset_iter *pLeft, 3944cbf6d2d2Sdan sqlite3_changeset_iter *pRight, 3945cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 3946cbf6d2d2Sdan void *pOut, 3947cbf6d2d2Sdan int *pnOut, 3948cbf6d2d2Sdan void **ppOut 39495d607a6eSdan ){ 39505d607a6eSdan SessionTable *pList = 0; /* List of SessionTable objects */ 39515d607a6eSdan int rc; /* Return code */ 395264277f4aSdan int bPatch; /* True for a patchset */ 3953e8fa8c96Sdan SessionTable *pTab; 3954e8fa8c96Sdan SessionBuffer buf = {0, 0, 0}; 39555d607a6eSdan 3956cbf6d2d2Sdan assert( xOutput==0 || (ppOut==0 && pnOut==0) ); 39575d607a6eSdan 3958abae0c4fSdan assert( pLeft->zTab==0 && pRight->zTab==0 ); 395916228167Sdan rc = sessionChangesetToHash(pLeft, &pList); 3960abae0c4fSdan assert( pLeft->zTab || pList==0 ); 39615d607a6eSdan if( rc==SQLITE_OK ){ 396216228167Sdan rc = sessionChangesetToHash(pRight, &pList); 39635d607a6eSdan } 3964cbf6d2d2Sdan bPatch = pLeft->bPatchset || pRight->bPatchset; 39655d607a6eSdan 3966abae0c4fSdan if( pLeft->zTab && pRight->zTab && pLeft->bPatchset!=pRight->bPatchset ){ 3967abae0c4fSdan rc = SQLITE_ERROR; 3968abae0c4fSdan } 3969abae0c4fSdan 39705d607a6eSdan /* Create the serialized output changeset based on the contents of the 39715d607a6eSdan ** hash tables attached to the SessionTable objects in list pList. 39725d607a6eSdan */ 3973e8fa8c96Sdan for(pTab=pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){ 39745d607a6eSdan int i; 39755d607a6eSdan if( pTab->nEntry==0 ) continue; 39765d607a6eSdan 397764277f4aSdan sessionAppendTableHdr(&buf, bPatch, pTab, &rc); 39785d607a6eSdan for(i=0; i<pTab->nChange; i++){ 39795d607a6eSdan SessionChange *p; 39805d607a6eSdan for(p=pTab->apChange[i]; p; p=p->pNext){ 3981798693b2Sdan sessionAppendByte(&buf, p->op, &rc); 39825d607a6eSdan sessionAppendByte(&buf, p->bIndirect, &rc); 39835d607a6eSdan sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc); 39845d607a6eSdan } 39855d607a6eSdan } 3986cbf6d2d2Sdan 3987f1a08ad8Sdrh if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){ 3988cbf6d2d2Sdan rc = xOutput(pOut, buf.aBuf, buf.nBuf); 3989cbf6d2d2Sdan buf.nBuf = 0; 3990cbf6d2d2Sdan } 39915d607a6eSdan } 39925d607a6eSdan 39935d607a6eSdan if( rc==SQLITE_OK ){ 3994cbf6d2d2Sdan if( xOutput ){ 3995cbf6d2d2Sdan if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf); 3996cbf6d2d2Sdan }else{ 39975d607a6eSdan *ppOut = buf.aBuf; 39985d607a6eSdan *pnOut = buf.nBuf; 3999cbf6d2d2Sdan buf.aBuf = 0; 40005d607a6eSdan } 40015d607a6eSdan } 4002cbf6d2d2Sdan sqlite3_free(buf.aBuf); 40035d607a6eSdan 40045d607a6eSdan sessionDeleteTable(pList); 40055d607a6eSdan return rc; 40065d607a6eSdan } 40075d607a6eSdan 4008cbf6d2d2Sdan /* 4009cbf6d2d2Sdan ** Combine two changesets together. 4010cbf6d2d2Sdan */ 4011cbf6d2d2Sdan int sqlite3changeset_concat( 4012cbf6d2d2Sdan int nLeft, /* Number of bytes in lhs input */ 4013cbf6d2d2Sdan void *pLeft, /* Lhs input changeset */ 4014cbf6d2d2Sdan int nRight /* Number of bytes in rhs input */, 4015cbf6d2d2Sdan void *pRight, /* Rhs input changeset */ 4016cbf6d2d2Sdan int *pnOut, /* OUT: Number of bytes in output changeset */ 4017cbf6d2d2Sdan void **ppOut /* OUT: changeset (left <concat> right) */ 4018cbf6d2d2Sdan ){ 4019cbf6d2d2Sdan sqlite3_changeset_iter *pIter1 = 0; 4020cbf6d2d2Sdan sqlite3_changeset_iter *pIter2 = 0; 4021cbf6d2d2Sdan int rc; 4022cbf6d2d2Sdan 4023cbf6d2d2Sdan *pnOut = 0; 4024cbf6d2d2Sdan *ppOut = 0; 4025cbf6d2d2Sdan rc = sqlite3changeset_start(&pIter1, nLeft, pLeft); 4026cbf6d2d2Sdan if( rc==SQLITE_OK ){ 4027cbf6d2d2Sdan rc = sqlite3changeset_start(&pIter2, nRight, pRight); 4028cbf6d2d2Sdan } 4029cbf6d2d2Sdan if( rc==SQLITE_OK ){ 4030cbf6d2d2Sdan rc = sessionChangesetConcat(pIter1, pIter2, 0, 0, pnOut, ppOut); 4031cbf6d2d2Sdan } 4032cbf6d2d2Sdan 4033cbf6d2d2Sdan sqlite3changeset_finalize(pIter1); 4034cbf6d2d2Sdan sqlite3changeset_finalize(pIter2); 4035cbf6d2d2Sdan return rc; 4036cbf6d2d2Sdan } 4037cbf6d2d2Sdan 4038cbf6d2d2Sdan /* 4039cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat(). 4040cbf6d2d2Sdan */ 4041f1a08ad8Sdrh int sqlite3changeset_concat_strm( 4042cbf6d2d2Sdan int (*xInputA)(void *pIn, void *pData, int *pnData), 4043cbf6d2d2Sdan void *pInA, 4044cbf6d2d2Sdan int (*xInputB)(void *pIn, void *pData, int *pnData), 4045cbf6d2d2Sdan void *pInB, 4046cbf6d2d2Sdan int (*xOutput)(void *pOut, const void *pData, int nData), 4047cbf6d2d2Sdan void *pOut 4048cbf6d2d2Sdan ){ 4049cbf6d2d2Sdan sqlite3_changeset_iter *pIter1 = 0; 4050cbf6d2d2Sdan sqlite3_changeset_iter *pIter2 = 0; 4051cbf6d2d2Sdan int rc; 4052cbf6d2d2Sdan 4053f1a08ad8Sdrh rc = sqlite3changeset_start_strm(&pIter1, xInputA, pInA); 4054cbf6d2d2Sdan if( rc==SQLITE_OK ){ 4055f1a08ad8Sdrh rc = sqlite3changeset_start_strm(&pIter2, xInputB, pInB); 4056cbf6d2d2Sdan } 4057cbf6d2d2Sdan if( rc==SQLITE_OK ){ 4058cbf6d2d2Sdan rc = sessionChangesetConcat(pIter1, pIter2, xOutput, pOut, 0, 0); 4059cbf6d2d2Sdan } 4060cbf6d2d2Sdan 4061cbf6d2d2Sdan sqlite3changeset_finalize(pIter1); 4062cbf6d2d2Sdan sqlite3changeset_finalize(pIter2); 4063cbf6d2d2Sdan return rc; 4064cbf6d2d2Sdan } 4065cbf6d2d2Sdan 40669b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */ 4067