14fccf43aSdan 
29b1c62d4Sdrh #if defined(SQLITE_ENABLE_SESSION) && defined(SQLITE_ENABLE_PREUPDATE_HOOK)
34fccf43aSdan #include "sqlite3session.h"
44fccf43aSdan #include <assert.h>
54fccf43aSdan #include <string.h>
64fccf43aSdan 
75d8a2984Sdan #ifndef SQLITE_AMALGAMATION
84fccf43aSdan # include "sqliteInt.h"
94fccf43aSdan # include "vdbeInt.h"
105d8a2984Sdan #endif
114fccf43aSdan 
124fccf43aSdan typedef struct SessionTable SessionTable;
134fccf43aSdan typedef struct SessionChange SessionChange;
14296c7658Sdan typedef struct SessionBuffer SessionBuffer;
15ef7a6304Sdan typedef struct SessionInput SessionInput;
16ef7a6304Sdan 
17ef7a6304Sdan /*
18ef7a6304Sdan ** Minimum chunk size used by streaming versions of functions.
19ef7a6304Sdan */
20f1a08ad8Sdrh #ifndef SESSIONS_STRM_CHUNK_SIZE
214757c658Sdan # ifdef SQLITE_TEST
22f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 64
234757c658Sdan # else
24f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 1024
25f1a08ad8Sdrh # endif
264757c658Sdan #endif
274fccf43aSdan 
28cf8e9144Sdan typedef struct SessionHook SessionHook;
29cf8e9144Sdan struct SessionHook {
30cf8e9144Sdan   void *pCtx;
31cf8e9144Sdan   int (*xOld)(void*,int,sqlite3_value**);
32cf8e9144Sdan   int (*xNew)(void*,int,sqlite3_value**);
33cf8e9144Sdan   int (*xCount)(void*);
34cf8e9144Sdan   int (*xDepth)(void*);
35cf8e9144Sdan };
36cf8e9144Sdan 
37296c7658Sdan /*
38296c7658Sdan ** Session handle structure.
39296c7658Sdan */
404fccf43aSdan struct sqlite3_session {
414fccf43aSdan   sqlite3 *db;                    /* Database handle session is attached to */
424fccf43aSdan   char *zDb;                      /* Name of database session is attached to */
43296c7658Sdan   int bEnable;                    /* True if currently recording */
44b4480e94Sdan   int bIndirect;                  /* True if all changes are indirect */
45ff4d0f41Sdan   int bAutoAttach;                /* True to auto-attach tables */
464fccf43aSdan   int rc;                         /* Non-zero if an error has occurred */
477531a5a3Sdan   void *pFilterCtx;               /* First argument to pass to xTableFilter */
487531a5a3Sdan   int (*xTableFilter)(void *pCtx, const char *zTab);
494fccf43aSdan   sqlite3_session *pNext;         /* Next session object on same db. */
504fccf43aSdan   SessionTable *pTable;           /* List of attached tables */
51cf8e9144Sdan   SessionHook hook;               /* APIs to grab new and old data with */
524fccf43aSdan };
534fccf43aSdan 
544fccf43aSdan /*
55ef7a6304Sdan ** Instances of this structure are used to build strings or binary records.
56ef7a6304Sdan */
57ef7a6304Sdan struct SessionBuffer {
58ef7a6304Sdan   u8 *aBuf;                       /* Pointer to changeset buffer */
59ef7a6304Sdan   int nBuf;                       /* Size of buffer aBuf */
60ef7a6304Sdan   int nAlloc;                     /* Size of allocation containing aBuf */
61ef7a6304Sdan };
62ef7a6304Sdan 
63ef7a6304Sdan /*
6416228167Sdan ** An object of this type is used internally as an abstraction for
6516228167Sdan ** input data. Input data may be supplied either as a single large buffer
6616228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g.
67f1a08ad8Sdrh **  sqlite3changeset_start_strm()).
68ef7a6304Sdan */
69ef7a6304Sdan struct SessionInput {
704757c658Sdan   int iNext;                      /* Offset in aData[] of next change */
714757c658Sdan   u8 *aData;                      /* Pointer to buffer containing changeset */
724757c658Sdan   int nData;                      /* Number of bytes in aData */
734757c658Sdan 
74ef7a6304Sdan   SessionBuffer buf;              /* Current read buffer */
75ef7a6304Sdan   int (*xInput)(void*, void*, int*);        /* Input stream call (or NULL) */
76ef7a6304Sdan   void *pIn;                                /* First argument to xInput */
77ef7a6304Sdan   int bEof;                       /* Set to true after xInput finished */
78ef7a6304Sdan };
79ef7a6304Sdan 
80ef7a6304Sdan /*
81296c7658Sdan ** Structure for changeset iterators.
82296c7658Sdan */
83296c7658Sdan struct sqlite3_changeset_iter {
84ef7a6304Sdan   SessionInput in;                /* Input buffer or stream */
85ef7a6304Sdan   SessionBuffer tblhdr;           /* Buffer to hold apValue/zTab/abPK/ */
8673b3c055Sdan   int bPatchset;                  /* True if this is a patchset */
87296c7658Sdan   int rc;                         /* Iterator error code */
88296c7658Sdan   sqlite3_stmt *pConflict;        /* Points to conflicting row, if any */
89296c7658Sdan   char *zTab;                     /* Current table */
90296c7658Sdan   int nCol;                       /* Number of columns in zTab */
91296c7658Sdan   int op;                         /* Current operation */
92b4480e94Sdan   int bIndirect;                  /* True if current change was indirect */
93244593c8Sdan   u8 *abPK;                       /* Primary key array */
94296c7658Sdan   sqlite3_value **apValue;        /* old.* and new.* values */
95296c7658Sdan };
96296c7658Sdan 
97296c7658Sdan /*
984fccf43aSdan ** Each session object maintains a set of the following structures, one
994fccf43aSdan ** for each table the session object is monitoring. The structures are
1004fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable.
1014fccf43aSdan **
1024fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have
1034fccf43aSdan ** been modified in any way since the session object was attached to the
1044fccf43aSdan ** table.
1054fccf43aSdan **
1064fccf43aSdan ** The data associated with each hash-table entry is a structure containing
1074fccf43aSdan ** a subset of the initial values that the modified row contained at the
1084fccf43aSdan ** start of the session. Or no initial values if the row was inserted.
1094fccf43aSdan */
1104fccf43aSdan struct SessionTable {
1114fccf43aSdan   SessionTable *pNext;
1124fccf43aSdan   char *zName;                    /* Local name of table */
1134fccf43aSdan   int nCol;                       /* Number of columns in table zName */
114e8d5648eSdan   const char **azCol;             /* Column names */
115e8d5648eSdan   u8 *abPK;                       /* Array of primary key flags */
116296c7658Sdan   int nEntry;                     /* Total number of entries in hash table */
1174fccf43aSdan   int nChange;                    /* Size of apChange[] array */
1184fccf43aSdan   SessionChange **apChange;       /* Hash table buckets */
1194fccf43aSdan };
1204fccf43aSdan 
1214fccf43aSdan /*
1224fccf43aSdan ** RECORD FORMAT:
1234fccf43aSdan **
1244fccf43aSdan ** The following record format is similar to (but not compatible with) that
1254fccf43aSdan ** used in SQLite database files. This format is used as part of the
1264fccf43aSdan ** change-set binary format, and so must be architecture independent.
1274fccf43aSdan **
1284fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained -
1294fccf43aSdan ** there is no separation of header and data. Each field begins with a
1304fccf43aSdan ** single byte describing its type, as follows:
1314fccf43aSdan **
1324fccf43aSdan **       0x00: Undefined value.
1334fccf43aSdan **       0x01: Integer value.
1344fccf43aSdan **       0x02: Real value.
1354fccf43aSdan **       0x03: Text value.
1364fccf43aSdan **       0x04: Blob value.
1374fccf43aSdan **       0x05: SQL NULL value.
1384fccf43aSdan **
1394fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT
1404fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists
1414fccf43aSdan ** only of the single type byte. For other types of values, the type byte
1424fccf43aSdan ** is followed by:
1434fccf43aSdan **
1444fccf43aSdan **   Text values:
1454fccf43aSdan **     A varint containing the number of bytes in the value (encoded using
1464fccf43aSdan **     UTF-8). Followed by a buffer containing the UTF-8 representation
1474fccf43aSdan **     of the text value. There is no nul terminator.
1484fccf43aSdan **
1494fccf43aSdan **   Blob values:
1504fccf43aSdan **     A varint containing the number of bytes in the value, followed by
1514fccf43aSdan **     a buffer containing the value itself.
1524fccf43aSdan **
1534fccf43aSdan **   Integer values:
1544fccf43aSdan **     An 8-byte big-endian integer value.
1554fccf43aSdan **
1564fccf43aSdan **   Real values:
1574fccf43aSdan **     An 8-byte big-endian IEEE 754-2008 real value.
1584fccf43aSdan **
1594fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite
1604fccf43aSdan ** record format.
1614fccf43aSdan **
1624fccf43aSdan ** CHANGESET FORMAT:
1634fccf43aSdan **
1644fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on
1654fccf43aSdan ** one or more tables. Operations on a single table are grouped together,
1664fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all
1674fccf43aSdan ** mixed together).
1684fccf43aSdan **
1694fccf43aSdan ** Each group of changes begins with a table header:
1704fccf43aSdan **
1714fccf43aSdan **   1 byte: Constant 0x54 (capital 'T')
172730bb805Sdan **   Varint: Number of columns in the table.
17373b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
1744fccf43aSdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
1754fccf43aSdan **
1764fccf43aSdan ** Followed by one or more changes to the table.
1774fccf43aSdan **
1784fccf43aSdan **   1 byte: Either SQLITE_INSERT, UPDATE or DELETE.
1795d607a6eSdan **   1 byte: The "indirect-change" flag.
1804fccf43aSdan **   old.* record: (delete and update only)
1814fccf43aSdan **   new.* record: (insert and update only)
18273b3c055Sdan **
183730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the
184730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of
185730bb805Sdan ** columns in the table. The i'th field of each record is associated with
186730bb805Sdan ** the i'th column of the table, counting from left to right in the order
187730bb805Sdan ** in which columns were declared in the CREATE TABLE statement.
188730bb805Sdan **
189730bb805Sdan ** The new.* record that is part of each INSERT change contains the values
190730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each
191730bb805Sdan ** DELETE change contains the values that made up the row that was deleted
192730bb805Sdan ** from the database. In the changeset format, the records that are part
193730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00)
194730bb805Sdan ** fields.
195730bb805Sdan **
196730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields
197730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are
198730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields
199730bb805Sdan ** are set to the values that made up the row before the UPDATE that the
200730bb805Sdan ** change records took place. Within the new.* record, fields associated
201730bb805Sdan ** with table columns modified by the UPDATE change contain the new
202730bb805Sdan ** values. Fields associated with table columns that are not modified
203730bb805Sdan ** are set to "undefined".
204730bb805Sdan **
20573b3c055Sdan ** PATCHSET FORMAT:
20673b3c055Sdan **
20773b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset,
208730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict
209730bb805Sdan ** resolution is required when applying the changeset.
21073b3c055Sdan **
21173b3c055Sdan ** Each group of changes begins with a table header:
21273b3c055Sdan **
21373b3c055Sdan **   1 byte: Constant 0x50 (capital 'P')
214730bb805Sdan **   Varint: Number of columns in the table.
21573b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
21673b3c055Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
21773b3c055Sdan **
21873b3c055Sdan ** Followed by one or more changes to the table.
21973b3c055Sdan **
22073b3c055Sdan **   1 byte: Either SQLITE_INSERT, UPDATE or DELETE.
22173b3c055Sdan **   1 byte: The "indirect-change" flag.
222730bb805Sdan **   single record: (PK fields for DELETE, PK and modified fields for UPDATE,
223730bb805Sdan **                   full record for INSERT).
224730bb805Sdan **
225730bb805Sdan ** As in the changeset format, each field of the single record that is part
226730bb805Sdan ** of a patchset change is associated with the correspondingly positioned
227730bb805Sdan ** table column, counting from left to right within the CREATE TABLE
228730bb805Sdan ** statement.
229730bb805Sdan **
230730bb805Sdan ** For a DELETE change, all fields within the record except those associated
231730bb805Sdan ** with PRIMARY KEY columns are set to "undefined". The PRIMARY KEY fields
232730bb805Sdan ** contain the values identifying the row to delete.
233730bb805Sdan **
234730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY
235730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined".
236730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update,
237730bb805Sdan ** and fields associated with modified columns contain the new column values.
238730bb805Sdan **
239730bb805Sdan ** The records associated with INSERT changes are in the same format as for
240730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT
241730bb805Sdan ** change to contain a field set to "undefined".
2424fccf43aSdan */
2434fccf43aSdan 
2444fccf43aSdan /*
2454fccf43aSdan ** For each row modified during a session, there exists a single instance of
2464fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table.
2474fccf43aSdan */
2484fccf43aSdan struct SessionChange {
249798693b2Sdan   int op;                         /* One of UPDATE, DELETE, INSERT */
250b4480e94Sdan   int bIndirect;                  /* True if this change is "indirect" */
2514fccf43aSdan   int nRecord;                    /* Number of bytes in buffer aRecord[] */
2524fccf43aSdan   u8 *aRecord;                    /* Buffer containing old.* record */
2534fccf43aSdan   SessionChange *pNext;           /* For hash-table collisions */
2544fccf43aSdan };
2554fccf43aSdan 
256296c7658Sdan /*
257296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the
258296c7658Sdan ** number of bytes written.
259296c7658Sdan */
260296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){
261296c7658Sdan   return putVarint32(aBuf, iVal);
2624fccf43aSdan }
2634fccf43aSdan 
264296c7658Sdan /*
265296c7658Sdan ** Return the number of bytes required to store value iVal as a varint.
266296c7658Sdan */
267296c7658Sdan static int sessionVarintLen(int iVal){
268296c7658Sdan   return sqlite3VarintLen(iVal);
269296c7658Sdan }
270296c7658Sdan 
271296c7658Sdan /*
272296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of
273296c7658Sdan ** bytes read.
274296c7658Sdan */
2754fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){
276296c7658Sdan   return getVarint32(aBuf, *piVal);
2774fccf43aSdan }
2784fccf43aSdan 
279296c7658Sdan /*
280296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return
281296c7658Sdan ** the value read.
282296c7658Sdan */
2834fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){
2844fccf43aSdan   return (((sqlite3_int64)aRec[0]) << 56)
2854fccf43aSdan        + (((sqlite3_int64)aRec[1]) << 48)
2864fccf43aSdan        + (((sqlite3_int64)aRec[2]) << 40)
2874fccf43aSdan        + (((sqlite3_int64)aRec[3]) << 32)
2884fccf43aSdan        + (((sqlite3_int64)aRec[4]) << 24)
2894fccf43aSdan        + (((sqlite3_int64)aRec[5]) << 16)
2904fccf43aSdan        + (((sqlite3_int64)aRec[6]) <<  8)
2914fccf43aSdan        + (((sqlite3_int64)aRec[7]) <<  0);
2924fccf43aSdan }
2934fccf43aSdan 
2944fccf43aSdan /*
295296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[].
296296c7658Sdan */
297296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){
298296c7658Sdan   aBuf[0] = (i>>56) & 0xFF;
299296c7658Sdan   aBuf[1] = (i>>48) & 0xFF;
300296c7658Sdan   aBuf[2] = (i>>40) & 0xFF;
301296c7658Sdan   aBuf[3] = (i>>32) & 0xFF;
302296c7658Sdan   aBuf[4] = (i>>24) & 0xFF;
303296c7658Sdan   aBuf[5] = (i>>16) & 0xFF;
304296c7658Sdan   aBuf[6] = (i>> 8) & 0xFF;
305296c7658Sdan   aBuf[7] = (i>> 0) & 0xFF;
306296c7658Sdan }
307296c7658Sdan 
308296c7658Sdan /*
3094fccf43aSdan ** This function is used to serialize the contents of value pValue (see
3104fccf43aSdan ** comment titled "RECORD FORMAT" above).
3114fccf43aSdan **
3124fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to
3134fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before
3144fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is
3154fccf43aSdan ** set *pnWrite.
3164fccf43aSdan **
3174fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs
3184fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16))
3194fccf43aSdan ** SQLITE_NOMEM is returned.
3204fccf43aSdan */
3214fccf43aSdan static int sessionSerializeValue(
3224fccf43aSdan   u8 *aBuf,                       /* If non-NULL, write serialized value here */
3234fccf43aSdan   sqlite3_value *pValue,          /* Value to serialize */
3244fccf43aSdan   int *pnWrite                    /* IN/OUT: Increment by bytes written */
3254fccf43aSdan ){
326296c7658Sdan   int nByte;                      /* Size of serialized value in bytes */
3274fccf43aSdan 
32880fe2d93Sdan   if( pValue ){
32980fe2d93Sdan     int eType;                    /* Value type (SQLITE_NULL, TEXT etc.) */
33080fe2d93Sdan 
3314fccf43aSdan     eType = sqlite3_value_type(pValue);
3324fccf43aSdan     if( aBuf ) aBuf[0] = eType;
3334fccf43aSdan 
3344fccf43aSdan     switch( eType ){
3354fccf43aSdan       case SQLITE_NULL:
3364fccf43aSdan         nByte = 1;
3374fccf43aSdan         break;
3384fccf43aSdan 
3394fccf43aSdan       case SQLITE_INTEGER:
3404fccf43aSdan       case SQLITE_FLOAT:
3414fccf43aSdan         if( aBuf ){
3424fccf43aSdan           /* TODO: SQLite does something special to deal with mixed-endian
3434fccf43aSdan           ** floating point values (e.g. ARM7). This code probably should
3444fccf43aSdan           ** too.  */
3454fccf43aSdan           u64 i;
3464fccf43aSdan           if( eType==SQLITE_INTEGER ){
3474fccf43aSdan             i = (u64)sqlite3_value_int64(pValue);
3484fccf43aSdan           }else{
3494fccf43aSdan             double r;
3504fccf43aSdan             assert( sizeof(double)==8 && sizeof(u64)==8 );
3514fccf43aSdan             r = sqlite3_value_double(pValue);
3524fccf43aSdan             memcpy(&i, &r, 8);
3534fccf43aSdan           }
354296c7658Sdan           sessionPutI64(&aBuf[1], i);
3554fccf43aSdan         }
3564fccf43aSdan         nByte = 9;
3574fccf43aSdan         break;
3584fccf43aSdan 
3594e895da1Sdan       default: {
36080fe2d93Sdan         u8 *z;
36180fe2d93Sdan         int n;
36280fe2d93Sdan         int nVarint;
36380fe2d93Sdan 
3644e895da1Sdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
36580fe2d93Sdan         if( eType==SQLITE_TEXT ){
36680fe2d93Sdan           z = (u8 *)sqlite3_value_text(pValue);
36780fe2d93Sdan         }else{
36880fe2d93Sdan           z = (u8 *)sqlite3_value_blob(pValue);
36980fe2d93Sdan         }
37080fe2d93Sdan         n = sqlite3_value_bytes(pValue);
3713cc89d95Sdan         if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
37280fe2d93Sdan         nVarint = sessionVarintLen(n);
37380fe2d93Sdan 
3744fccf43aSdan         if( aBuf ){
3754fccf43aSdan           sessionVarintPut(&aBuf[1], n);
3764fccf43aSdan           memcpy(&aBuf[nVarint + 1], eType==SQLITE_TEXT ?
3774fccf43aSdan               sqlite3_value_text(pValue) : sqlite3_value_blob(pValue), n
3784fccf43aSdan           );
3794fccf43aSdan         }
3804fccf43aSdan 
3814fccf43aSdan         nByte = 1 + nVarint + n;
3824fccf43aSdan         break;
3834fccf43aSdan       }
3844fccf43aSdan     }
38580fe2d93Sdan   }else{
38680fe2d93Sdan     nByte = 1;
38780fe2d93Sdan     if( aBuf ) aBuf[0] = '\0';
38880fe2d93Sdan   }
3894fccf43aSdan 
390fa122adaSdan   if( pnWrite ) *pnWrite += nByte;
3914fccf43aSdan   return SQLITE_OK;
3924fccf43aSdan }
3934fccf43aSdan 
394fa122adaSdan 
395798693b2Sdan /*
396798693b2Sdan ** This macro is used to calculate hash key values for data structures. In
397798693b2Sdan ** order to use this macro, the entire data structure must be represented
398798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value
399798693b2Sdan ** for a data structure represented as three such integers, the macro may
400798693b2Sdan ** then be used as follows:
401798693b2Sdan **
402798693b2Sdan **    int hash_key_value;
403798693b2Sdan **    hash_key_value = HASH_APPEND(0, <value 1>);
404798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 2>);
405798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 3>);
406798693b2Sdan **
407798693b2Sdan ** In practice, the data structures this macro is used for are the primary
408798693b2Sdan ** key values of modified rows.
409798693b2Sdan */
4104131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add)
411798693b2Sdan 
412798693b2Sdan /*
413798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the
414798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
415798693b2Sdan */
4164131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){
417e8d5648eSdan   h = HASH_APPEND(h, i & 0xFFFFFFFF);
418e8d5648eSdan   return HASH_APPEND(h, (i>>32)&0xFFFFFFFF);
419e8d5648eSdan }
420798693b2Sdan 
421798693b2Sdan /*
422798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to
423798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value.
424798693b2Sdan */
4254131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){
426e8d5648eSdan   int i;
427e8d5648eSdan   for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]);
428e8d5648eSdan   return h;
429e8d5648eSdan }
430e8d5648eSdan 
4314fccf43aSdan /*
432798693b2Sdan ** Append the hash of the data type passed as the second argument to the
433798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
434798693b2Sdan */
435798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){
436798693b2Sdan   return HASH_APPEND(h, eType);
437798693b2Sdan }
438798693b2Sdan 
439798693b2Sdan /*
4404131639cSdan ** This function may only be called from within a pre-update callback.
4414131639cSdan ** It calculates a hash based on the primary key values of the old.* or
442798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to
443798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL
444798693b2Sdan ** values, *pbNullPK is set to true before returning.
445798693b2Sdan **
446798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values
447798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned
448798693b2Sdan ** and the output variables are set as described above.
4494fccf43aSdan */
450798693b2Sdan static int sessionPreupdateHash(
451cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns pTab */
452e8d5648eSdan   SessionTable *pTab,             /* Session table handle */
453e8d5648eSdan   int bNew,                       /* True to hash the new.* PK */
45427453faeSdan   int *piHash,                    /* OUT: Hash value */
455798693b2Sdan   int *pbNullPK                   /* OUT: True if there are NULL values in PK */
456e8d5648eSdan ){
4574131639cSdan   unsigned int h = 0;             /* Hash value to return */
4584131639cSdan   int i;                          /* Used to iterate through columns */
459e8d5648eSdan 
46027453faeSdan   assert( *pbNullPK==0 );
461cf8e9144Sdan   assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) );
462e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
463e8d5648eSdan     if( pTab->abPK[i] ){
464e8d5648eSdan       int rc;
465e8d5648eSdan       int eType;
466e8d5648eSdan       sqlite3_value *pVal;
467e8d5648eSdan 
468e8d5648eSdan       if( bNew ){
469cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal);
470e8d5648eSdan       }else{
471cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal);
472e8d5648eSdan       }
47312ca0b56Sdan       if( rc!=SQLITE_OK ) return rc;
474e8d5648eSdan 
475e8d5648eSdan       eType = sqlite3_value_type(pVal);
476798693b2Sdan       h = sessionHashAppendType(h, eType);
4776734007dSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
478e8d5648eSdan         i64 iVal;
479e8d5648eSdan         if( eType==SQLITE_INTEGER ){
480e8d5648eSdan           iVal = sqlite3_value_int64(pVal);
481e8d5648eSdan         }else{
482e8d5648eSdan           double rVal = sqlite3_value_double(pVal);
483e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
484e8d5648eSdan           memcpy(&iVal, &rVal, 8);
485e8d5648eSdan         }
486e8d5648eSdan         h = sessionHashAppendI64(h, iVal);
4876734007dSdan       }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
4886734007dSdan         const u8 *z;
4893cc89d95Sdan         int n;
4906734007dSdan         if( eType==SQLITE_TEXT ){
4916734007dSdan           z = (const u8 *)sqlite3_value_text(pVal);
4926734007dSdan         }else{
4936734007dSdan           z = (const u8 *)sqlite3_value_blob(pVal);
494e8d5648eSdan         }
4953cc89d95Sdan         n = sqlite3_value_bytes(pVal);
4963cc89d95Sdan         if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
4973cc89d95Sdan         h = sessionHashAppendBlob(h, n, z);
4986734007dSdan       }else{
49927453faeSdan         assert( eType==SQLITE_NULL );
50027453faeSdan         *pbNullPK = 1;
501e8d5648eSdan       }
502e8d5648eSdan     }
503e8d5648eSdan   }
504e8d5648eSdan 
505e8d5648eSdan   *piHash = (h % pTab->nChange);
506e8d5648eSdan   return SQLITE_OK;
507e8d5648eSdan }
508e8d5648eSdan 
5094131639cSdan /*
5106cda207fSdan ** The buffer that the argument points to contains a serialized SQL value.
5116cda207fSdan ** Return the number of bytes of space occupied by the value (including
5126cda207fSdan ** the type byte).
5136cda207fSdan */
5146cda207fSdan static int sessionSerialLen(u8 *a){
5156cda207fSdan   int e = *a;
5166cda207fSdan   int n;
5176cda207fSdan   if( e==0 ) return 1;
5186cda207fSdan   if( e==SQLITE_NULL ) return 1;
5196cda207fSdan   if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9;
5206cda207fSdan   return sessionVarintGet(&a[1], &n) + 1 + n;
5216cda207fSdan }
5226cda207fSdan 
5236cda207fSdan /*
5245d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a
525798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys
5264131639cSdan ** calculated by this function are compatible with those calculated by
5274131639cSdan ** sessionPreupdateHash().
52864277f4aSdan **
52964277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from
53064277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely.
5314131639cSdan */
5324131639cSdan static unsigned int sessionChangeHash(
5334131639cSdan   SessionTable *pTab,             /* Table handle */
53464277f4aSdan   int bPkOnly,                    /* Record consists of PK fields only */
5355d607a6eSdan   u8 *aRecord,                    /* Change record */
5364131639cSdan   int nBucket                     /* Assume this many buckets in hash table */
537e8d5648eSdan ){
5384131639cSdan   unsigned int h = 0;             /* Value to return */
5394131639cSdan   int i;                          /* Used to iterate through columns */
5405d607a6eSdan   u8 *a = aRecord;                /* Used to iterate through change record */
541e8d5648eSdan 
542e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
5436cda207fSdan     int eType = *a;
544e8d5648eSdan     int isPK = pTab->abPK[i];
54564277f4aSdan     if( bPkOnly && isPK==0 ) continue;
546e8d5648eSdan 
54727453faeSdan     /* It is not possible for eType to be SQLITE_NULL here. The session
54827453faeSdan     ** module does not record changes for rows with NULL values stored in
54927453faeSdan     ** primary key columns. */
55027453faeSdan     assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
55127453faeSdan          || eType==SQLITE_TEXT || eType==SQLITE_BLOB
5526cda207fSdan          || eType==SQLITE_NULL || eType==0
55327453faeSdan     );
5546cda207fSdan     assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) );
55527453faeSdan 
5566cda207fSdan     if( isPK ){
5576cda207fSdan       a++;
558798693b2Sdan       h = sessionHashAppendType(h, eType);
55927453faeSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
5606734007dSdan         h = sessionHashAppendI64(h, sessionGetI64(a));
561e8d5648eSdan         a += 8;
56227453faeSdan       }else{
563e8d5648eSdan         int n;
564e8d5648eSdan         a += sessionVarintGet(a, &n);
5656734007dSdan         h = sessionHashAppendBlob(h, n, a);
566e8d5648eSdan         a += n;
567e8d5648eSdan       }
5686cda207fSdan     }else{
5696cda207fSdan       a += sessionSerialLen(a);
5706cda207fSdan     }
571e8d5648eSdan   }
572e8d5648eSdan   return (h % nBucket);
573e8d5648eSdan }
574e8d5648eSdan 
575798693b2Sdan /*
576798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab.
577798693b2Sdan ** This function returns true if the two records apply to the same row (i.e.
578798693b2Sdan ** have the same values stored in the primary key columns), or false
579798693b2Sdan ** otherwise.
580798693b2Sdan */
5815d607a6eSdan static int sessionChangeEqual(
582798693b2Sdan   SessionTable *pTab,             /* Table used for PK definition */
583a71d2371Sdan   int bLeftPkOnly,                /* True if aLeft[] contains PK fields only */
5845d607a6eSdan   u8 *aLeft,                      /* Change record */
585a71d2371Sdan   int bRightPkOnly,               /* True if aRight[] contains PK fields only */
5865d607a6eSdan   u8 *aRight                      /* Change record */
5875d607a6eSdan ){
588798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor to iterate through aLeft */
589798693b2Sdan   u8 *a2 = aRight;                /* Cursor to iterate through aRight */
590798693b2Sdan   int iCol;                       /* Used to iterate through table columns */
5915d607a6eSdan 
592798693b2Sdan   for(iCol=0; iCol<pTab->nCol; iCol++){
5935d607a6eSdan     int n1 = sessionSerialLen(a1);
5945d607a6eSdan     int n2 = sessionSerialLen(a2);
5955d607a6eSdan 
596798693b2Sdan     if( pTab->abPK[iCol] && (n1!=n2 || memcmp(a1, a2, n1)) ){
5975d607a6eSdan       return 0;
5985d607a6eSdan     }
59964277f4aSdan     if( pTab->abPK[iCol] || bLeftPkOnly==0 ) a1 += n1;
60064277f4aSdan     if( pTab->abPK[iCol] || bRightPkOnly==0 ) a2 += n2;
6015d607a6eSdan   }
6025d607a6eSdan 
6035d607a6eSdan   return 1;
6045d607a6eSdan }
6055d607a6eSdan 
606798693b2Sdan /*
607798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change
608798693b2Sdan ** records with nCol columns. This function "merges" the two records into
609798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is
610798693b2Sdan ** then set to point to one byte after the last byte written before
611798693b2Sdan ** returning.
612798693b2Sdan **
613798693b2Sdan ** The merging of records is done as follows: For each column, if the
614798693b2Sdan ** aRight record contains a value for the column, copy the value from
615798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither
616798693b2Sdan ** record contains a value for a given column, then neither does the
617798693b2Sdan ** output record.
618798693b2Sdan */
6195d607a6eSdan static void sessionMergeRecord(
6205d607a6eSdan   u8 **paOut,
621798693b2Sdan   int nCol,
6225d607a6eSdan   u8 *aLeft,
6235d607a6eSdan   u8 *aRight
6245d607a6eSdan ){
625798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor used to iterate through aLeft */
626798693b2Sdan   u8 *a2 = aRight;                /* Cursor used to iterate through aRight */
627798693b2Sdan   u8 *aOut = *paOut;              /* Output cursor */
628798693b2Sdan   int iCol;                       /* Used to iterate from 0 to nCol */
6295d607a6eSdan 
630798693b2Sdan   for(iCol=0; iCol<nCol; iCol++){
6315d607a6eSdan     int n1 = sessionSerialLen(a1);
6325d607a6eSdan     int n2 = sessionSerialLen(a2);
6335d607a6eSdan     if( *a2 ){
6345d607a6eSdan       memcpy(aOut, a2, n2);
6355d607a6eSdan       aOut += n2;
6365d607a6eSdan     }else{
6375d607a6eSdan       memcpy(aOut, a1, n1);
6385d607a6eSdan       aOut += n1;
6395d607a6eSdan     }
6405d607a6eSdan     a1 += n1;
6415d607a6eSdan     a2 += n2;
6425d607a6eSdan   }
6435d607a6eSdan 
6445d607a6eSdan   *paOut = aOut;
6455d607a6eSdan }
6465d607a6eSdan 
647798693b2Sdan /*
648798693b2Sdan ** This is a helper function used by sessionMergeUpdate().
649798693b2Sdan **
650798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value
651798693b2Sdan ** within a change record. Before it returns, both have been advanced so
652798693b2Sdan ** as to point to the next value in the record.
653798693b2Sdan **
654798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e.
655fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo
656798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the
657798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal
658798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points
659fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words:
660fa29ecc4Sdan **
661fa29ecc4Sdan **   if( *paTwo is valid ) return *paTwo;
662fa29ecc4Sdan **   return *paOne;
663fa29ecc4Sdan **
664798693b2Sdan */
6655d607a6eSdan static u8 *sessionMergeValue(
666798693b2Sdan   u8 **paOne,                     /* IN/OUT: Left-hand buffer pointer */
667798693b2Sdan   u8 **paTwo,                     /* IN/OUT: Right-hand buffer pointer */
668798693b2Sdan   int *pnVal                      /* OUT: Bytes in returned value */
6695d607a6eSdan ){
6705d607a6eSdan   u8 *a1 = *paOne;
6715d607a6eSdan   u8 *a2 = *paTwo;
6725d607a6eSdan   u8 *pRet = 0;
6735d607a6eSdan   int n1;
6745d607a6eSdan 
6755d607a6eSdan   assert( a1 );
6765d607a6eSdan   if( a2 ){
6775d607a6eSdan     int n2 = sessionSerialLen(a2);
6785d607a6eSdan     if( *a2 ){
6795d607a6eSdan       *pnVal = n2;
6805d607a6eSdan       pRet = a2;
6815d607a6eSdan     }
6825d607a6eSdan     *paTwo = &a2[n2];
6835d607a6eSdan   }
6845d607a6eSdan 
6855d607a6eSdan   n1 = sessionSerialLen(a1);
6865d607a6eSdan   if( pRet==0 ){
6875d607a6eSdan     *pnVal = n1;
6885d607a6eSdan     pRet = a1;
6895d607a6eSdan   }
6905d607a6eSdan   *paOne = &a1[n1];
6915d607a6eSdan 
6925d607a6eSdan   return pRet;
6935d607a6eSdan }
6945d607a6eSdan 
695798693b2Sdan /*
696798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes
697798693b2Sdan ** on the same row.
698798693b2Sdan */
6995d607a6eSdan static int sessionMergeUpdate(
700798693b2Sdan   u8 **paOut,                     /* IN/OUT: Pointer to output buffer */
701798693b2Sdan   SessionTable *pTab,             /* Table change pertains to */
702a71d2371Sdan   int bPatchset,                  /* True if records are patchset records */
703798693b2Sdan   u8 *aOldRecord1,                /* old.* record for first change */
704798693b2Sdan   u8 *aOldRecord2,                /* old.* record for second change */
705798693b2Sdan   u8 *aNewRecord1,                /* new.* record for first change */
706798693b2Sdan   u8 *aNewRecord2                 /* new.* record for second change */
7075d607a6eSdan ){
7085d607a6eSdan   u8 *aOld1 = aOldRecord1;
7095d607a6eSdan   u8 *aOld2 = aOldRecord2;
7105d607a6eSdan   u8 *aNew1 = aNewRecord1;
7115d607a6eSdan   u8 *aNew2 = aNewRecord2;
7125d607a6eSdan 
7135d607a6eSdan   u8 *aOut = *paOut;
7145d607a6eSdan   int i;
71564277f4aSdan 
71664277f4aSdan   if( bPatchset==0 ){
7175d607a6eSdan     int bRequired = 0;
7185d607a6eSdan 
7195d607a6eSdan     assert( aOldRecord1 && aNewRecord1 );
7205d607a6eSdan 
7215d607a6eSdan     /* Write the old.* vector first. */
7225d607a6eSdan     for(i=0; i<pTab->nCol; i++){
7235d607a6eSdan       int nOld;
7245d607a6eSdan       u8 *aOld;
7255d607a6eSdan       int nNew;
7265d607a6eSdan       u8 *aNew;
7275d607a6eSdan 
7285d607a6eSdan       aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7295d607a6eSdan       aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
7305d607a6eSdan       if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){
7315d607a6eSdan         if( pTab->abPK[i]==0 ) bRequired = 1;
7325d607a6eSdan         memcpy(aOut, aOld, nOld);
7335d607a6eSdan         aOut += nOld;
7345d607a6eSdan       }else{
7355d607a6eSdan         *(aOut++) = '\0';
7365d607a6eSdan       }
7375d607a6eSdan     }
7385d607a6eSdan 
7395d607a6eSdan     if( !bRequired ) return 0;
74064277f4aSdan   }
7415d607a6eSdan 
7425d607a6eSdan   /* Write the new.* vector */
7435d607a6eSdan   aOld1 = aOldRecord1;
7445d607a6eSdan   aOld2 = aOldRecord2;
7455d607a6eSdan   aNew1 = aNewRecord1;
7465d607a6eSdan   aNew2 = aNewRecord2;
7475d607a6eSdan   for(i=0; i<pTab->nCol; i++){
7485d607a6eSdan     int nOld;
7495d607a6eSdan     u8 *aOld;
7505d607a6eSdan     int nNew;
7515d607a6eSdan     u8 *aNew;
7525d607a6eSdan 
7535d607a6eSdan     aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7545d607a6eSdan     aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
75564277f4aSdan     if( bPatchset==0
75664277f4aSdan      && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew)))
75764277f4aSdan     ){
7585d607a6eSdan       *(aOut++) = '\0';
7595d607a6eSdan     }else{
7605d607a6eSdan       memcpy(aOut, aNew, nNew);
7615d607a6eSdan       aOut += nNew;
7625d607a6eSdan     }
7635d607a6eSdan   }
7645d607a6eSdan 
7655d607a6eSdan   *paOut = aOut;
7665d607a6eSdan   return 1;
7675d607a6eSdan }
7685d607a6eSdan 
76977fc1d5bSdan /*
77077fc1d5bSdan ** This function is only called from within a pre-update-hook callback.
77177fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row
77277fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise
77377fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns
77477fc1d5bSdan ** false.
77577fc1d5bSdan */
77677fc1d5bSdan static int sessionPreupdateEqual(
777cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns SessionTable */
77877fc1d5bSdan   SessionTable *pTab,             /* Table associated with change */
77977fc1d5bSdan   SessionChange *pChange,         /* Change to compare to */
78077fc1d5bSdan   int op                          /* Current pre-update operation */
781e8d5648eSdan ){
78277fc1d5bSdan   int iCol;                       /* Used to iterate through columns */
78377fc1d5bSdan   u8 *a = pChange->aRecord;       /* Cursor used to scan change record */
784e8d5648eSdan 
78577fc1d5bSdan   assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
78677fc1d5bSdan   for(iCol=0; iCol<pTab->nCol; iCol++){
78777fc1d5bSdan     if( !pTab->abPK[iCol] ){
788798693b2Sdan       a += sessionSerialLen(a);
789e8d5648eSdan     }else{
7906734007dSdan       sqlite3_value *pVal;        /* Value returned by preupdate_new/old */
7916734007dSdan       int rc;                     /* Error code from preupdate_new/old */
792798693b2Sdan       int eType = *a++;           /* Type of value from change record */
7936734007dSdan 
7946734007dSdan       /* The following calls to preupdate_new() and preupdate_old() can not
7956734007dSdan       ** fail. This is because they cache their return values, and by the
7966734007dSdan       ** time control flows to here they have already been called once from
7976734007dSdan       ** within sessionPreupdateHash(). The first two asserts below verify
7986734007dSdan       ** this (that the method has already been called). */
79977fc1d5bSdan       if( op==SQLITE_INSERT ){
800cf8e9144Sdan         /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */
801cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal);
802e8d5648eSdan       }else{
803cf8e9144Sdan         /* assert( db->pPreUpdate->pUnpacked ); */
804cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal);
805e8d5648eSdan       }
8066734007dSdan       assert( rc==SQLITE_OK );
80777fc1d5bSdan       if( sqlite3_value_type(pVal)!=eType ) return 0;
808e8d5648eSdan 
80912ca0b56Sdan       /* A SessionChange object never has a NULL value in a PK column */
81012ca0b56Sdan       assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
81112ca0b56Sdan            || eType==SQLITE_BLOB    || eType==SQLITE_TEXT
81212ca0b56Sdan       );
81312ca0b56Sdan 
81412ca0b56Sdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
815e8d5648eSdan         i64 iVal = sessionGetI64(a);
816e8d5648eSdan         a += 8;
817e8d5648eSdan         if( eType==SQLITE_INTEGER ){
81877fc1d5bSdan           if( sqlite3_value_int64(pVal)!=iVal ) return 0;
819e8d5648eSdan         }else{
820e8d5648eSdan           double rVal;
821e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
822e8d5648eSdan           memcpy(&rVal, &iVal, 8);
82377fc1d5bSdan           if( sqlite3_value_double(pVal)!=rVal ) return 0;
824e8d5648eSdan         }
82512ca0b56Sdan       }else{
826e8d5648eSdan         int n;
827e8d5648eSdan         const u8 *z;
828e8d5648eSdan         a += sessionVarintGet(a, &n);
82977fc1d5bSdan         if( sqlite3_value_bytes(pVal)!=n ) return 0;
83012ca0b56Sdan         if( eType==SQLITE_TEXT ){
83112ca0b56Sdan           z = sqlite3_value_text(pVal);
83212ca0b56Sdan         }else{
83312ca0b56Sdan           z = sqlite3_value_blob(pVal);
83412ca0b56Sdan         }
83577fc1d5bSdan         if( memcmp(a, z, n) ) return 0;
836e8d5648eSdan         a += n;
837e8d5648eSdan         break;
838e8d5648eSdan       }
839e8d5648eSdan     }
840e8d5648eSdan   }
841e8d5648eSdan 
84277fc1d5bSdan   return 1;
8434fccf43aSdan }
8444fccf43aSdan 
8454fccf43aSdan /*
8464fccf43aSdan ** If required, grow the hash table used to store changes on table pTab
8474fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the
8484fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return
8494fccf43aSdan ** SQLITE_OK.
8504fccf43aSdan **
8514fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In
8524fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway.
8534fccf43aSdan ** Growing the hash table in this case is a performance optimization only,
8544fccf43aSdan ** it is not required for correct operation.
8554fccf43aSdan */
85664277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){
8574fccf43aSdan   if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){
8584fccf43aSdan     int i;
8594fccf43aSdan     SessionChange **apNew;
8604fccf43aSdan     int nNew = (pTab->nChange ? pTab->nChange : 128) * 2;
8614fccf43aSdan 
8624fccf43aSdan     apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew);
8634fccf43aSdan     if( apNew==0 ){
8644fccf43aSdan       if( pTab->nChange==0 ){
8654fccf43aSdan         return SQLITE_ERROR;
8664fccf43aSdan       }
8674fccf43aSdan       return SQLITE_OK;
8684fccf43aSdan     }
8694fccf43aSdan     memset(apNew, 0, sizeof(SessionChange *) * nNew);
8704fccf43aSdan 
8714fccf43aSdan     for(i=0; i<pTab->nChange; i++){
8724fccf43aSdan       SessionChange *p;
8734fccf43aSdan       SessionChange *pNext;
8744fccf43aSdan       for(p=pTab->apChange[i]; p; p=pNext){
87564277f4aSdan         int bPkOnly = (p->op==SQLITE_DELETE && bPatchset);
87664277f4aSdan         int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew);
8774fccf43aSdan         pNext = p->pNext;
8784fccf43aSdan         p->pNext = apNew[iHash];
8794fccf43aSdan         apNew[iHash] = p;
8804fccf43aSdan       }
8814fccf43aSdan     }
8824fccf43aSdan 
8834fccf43aSdan     sqlite3_free(pTab->apChange);
8844fccf43aSdan     pTab->nChange = nNew;
8854fccf43aSdan     pTab->apChange = apNew;
8864fccf43aSdan   }
8874fccf43aSdan 
8884fccf43aSdan   return SQLITE_OK;
8894fccf43aSdan }
8904fccf43aSdan 
891296c7658Sdan /*
892e8d5648eSdan ** This function queries the database for the names of the columns of table
893e8d5648eSdan ** zThis, in schema zDb. It is expected that the table has nCol columns. If
894e8d5648eSdan ** not, SQLITE_SCHEMA is returned and none of the output variables are
895e8d5648eSdan ** populated.
896e8d5648eSdan **
89777fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number
89877fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a
899e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to
900e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not
901e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding
902e8d5648eSdan ** column is part of the primary key.
903e8d5648eSdan **
904e8d5648eSdan ** For example, if the table is declared as:
905e8d5648eSdan **
906e8d5648eSdan **     CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z));
907e8d5648eSdan **
90877fc1d5bSdan ** Then the four output variables are populated as follows:
909e8d5648eSdan **
91077fc1d5bSdan **     *pnCol  = 4
911e8d5648eSdan **     *pzTab  = "tbl1"
912e8d5648eSdan **     *pazCol = {"w", "x", "y", "z"}
913e8d5648eSdan **     *pabPK  = {1, 0, 0, 1}
914e8d5648eSdan **
915e8d5648eSdan ** All returned buffers are part of the same single allocation, which must
916e8d5648eSdan ** be freed using sqlite3_free() by the caller. If pazCol was not NULL, then
917e8d5648eSdan ** pointer *pazCol should be freed to release all memory. Otherwise, pointer
918e8d5648eSdan ** *pabPK. It is illegal for both pazCol and pabPK to be NULL.
919e8d5648eSdan */
920e8d5648eSdan static int sessionTableInfo(
921e8d5648eSdan   sqlite3 *db,                    /* Database connection */
922e8d5648eSdan   const char *zDb,                /* Name of attached database (e.g. "main") */
923e8d5648eSdan   const char *zThis,              /* Table name */
924ca62ad57Sdan   int *pnCol,                     /* OUT: number of columns */
925e8d5648eSdan   const char **pzTab,             /* OUT: Copy of zThis */
926e8d5648eSdan   const char ***pazCol,           /* OUT: Array of column names for table */
927e8d5648eSdan   u8 **pabPK                      /* OUT: Array of booleans - true for PK col */
928e8d5648eSdan ){
929e8d5648eSdan   char *zPragma;
930e8d5648eSdan   sqlite3_stmt *pStmt;
931e8d5648eSdan   int rc;
932e8d5648eSdan   int nByte;
933e8d5648eSdan   int nDbCol = 0;
934e8d5648eSdan   int nThis;
935e8d5648eSdan   int i;
936e8d5648eSdan   u8 *pAlloc;
937db04571cSdan   char **azCol = 0;
938e8d5648eSdan   u8 *abPK;
939e8d5648eSdan 
940db04571cSdan   assert( pazCol && pabPK );
941e8d5648eSdan 
942cfdbde21Sdrh   nThis = sqlite3Strlen30(zThis);
943e8d5648eSdan   zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis);
944e8d5648eSdan   if( !zPragma ) return SQLITE_NOMEM;
945e8d5648eSdan 
946e8d5648eSdan   rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0);
947e8d5648eSdan   sqlite3_free(zPragma);
948e8d5648eSdan   if( rc!=SQLITE_OK ) return rc;
949e8d5648eSdan 
950e8d5648eSdan   nByte = nThis + 1;
951e8d5648eSdan   while( SQLITE_ROW==sqlite3_step(pStmt) ){
952e8d5648eSdan     nByte += sqlite3_column_bytes(pStmt, 1);
953e8d5648eSdan     nDbCol++;
954e8d5648eSdan   }
955e8d5648eSdan   rc = sqlite3_reset(pStmt);
956e8d5648eSdan 
957e8d5648eSdan   if( rc==SQLITE_OK ){
958e8d5648eSdan     nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1);
959e8d5648eSdan     pAlloc = sqlite3_malloc(nByte);
960e8d5648eSdan     if( pAlloc==0 ){
961e8d5648eSdan       rc = SQLITE_NOMEM;
962e8d5648eSdan     }
963e8d5648eSdan   }
964e8d5648eSdan   if( rc==SQLITE_OK ){
965e8d5648eSdan     azCol = (char **)pAlloc;
966ca62ad57Sdan     pAlloc = (u8 *)&azCol[nDbCol];
967e8d5648eSdan     abPK = (u8 *)pAlloc;
968ca62ad57Sdan     pAlloc = &abPK[nDbCol];
969e8d5648eSdan     if( pzTab ){
970e8d5648eSdan       memcpy(pAlloc, zThis, nThis+1);
971e8d5648eSdan       *pzTab = (char *)pAlloc;
972e8d5648eSdan       pAlloc += nThis+1;
973e8d5648eSdan     }
974e8d5648eSdan 
975e8d5648eSdan     i = 0;
976e8d5648eSdan     while( SQLITE_ROW==sqlite3_step(pStmt) ){
977e8d5648eSdan       int nName = sqlite3_column_bytes(pStmt, 1);
978e8d5648eSdan       const unsigned char *zName = sqlite3_column_text(pStmt, 1);
979e8d5648eSdan       if( zName==0 ) break;
980e8d5648eSdan       memcpy(pAlloc, zName, nName+1);
981e8d5648eSdan       azCol[i] = (char *)pAlloc;
982e8d5648eSdan       pAlloc += nName+1;
983db04571cSdan       abPK[i] = sqlite3_column_int(pStmt, 5);
984e8d5648eSdan       i++;
985e8d5648eSdan     }
986e8d5648eSdan     rc = sqlite3_reset(pStmt);
987e8d5648eSdan 
988e8d5648eSdan   }
989e8d5648eSdan 
990e8d5648eSdan   /* If successful, populate the output variables. Otherwise, zero them and
991e8d5648eSdan   ** free any allocation made. An error code will be returned in this case.
992e8d5648eSdan   */
993e8d5648eSdan   if( rc==SQLITE_OK ){
994db04571cSdan     *pazCol = (const char **)azCol;
995db04571cSdan     *pabPK = abPK;
996ca62ad57Sdan     *pnCol = nDbCol;
997e8d5648eSdan   }else{
998db04571cSdan     *pazCol = 0;
999db04571cSdan     *pabPK = 0;
1000ca62ad57Sdan     *pnCol = 0;
1001e8d5648eSdan     if( pzTab ) *pzTab = 0;
1002db04571cSdan     sqlite3_free(azCol);
1003e8d5648eSdan   }
1004e8d5648eSdan   sqlite3_finalize(pStmt);
1005e8d5648eSdan   return rc;
1006e8d5648eSdan }
1007e8d5648eSdan 
1008e8d5648eSdan /*
1009296c7658Sdan ** This function is only called from within a pre-update handler for a
1010296c7658Sdan ** write to table pTab, part of session pSession. If this is the first
1011296c7658Sdan ** write to this table, set the SessionTable.nCol variable to the number
1012296c7658Sdan ** of columns in the table.
1013296c7658Sdan **
1014296c7658Sdan ** Otherwise, if this is not the first time this table has been written
1015296c7658Sdan ** to, check that the number of columns in the table has not changed. If
1016296c7658Sdan ** it has not, return zero.
1017296c7658Sdan **
1018296c7658Sdan ** If the number of columns in the table has changed since the last write
1019296c7658Sdan ** was recorded, set the session error-code to SQLITE_SCHEMA and return
1020296c7658Sdan ** non-zero. Users are not allowed to change the number of columns in a table
1021296c7658Sdan ** for which changes are being recorded by the session module. If they do so,
1022296c7658Sdan ** it is an error.
1023296c7658Sdan */
10244fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){
10254fccf43aSdan   if( pTab->nCol==0 ){
1026e8d5648eSdan     assert( pTab->azCol==0 || pTab->abPK==0 );
1027e8d5648eSdan     pSession->rc = sessionTableInfo(pSession->db, pSession->zDb,
1028ca62ad57Sdan         pTab->zName, &pTab->nCol, 0, &pTab->azCol, &pTab->abPK
1029e8d5648eSdan     );
1030ca62ad57Sdan   }
1031ca62ad57Sdan   if( pSession->rc==SQLITE_OK
1032cf8e9144Sdan    && pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx)
1033ca62ad57Sdan   ){
10344fccf43aSdan     pSession->rc = SQLITE_SCHEMA;
10354fccf43aSdan   }
1036e8d5648eSdan   return pSession->rc;
1037e8d5648eSdan }
1038e8d5648eSdan 
103977fc1d5bSdan /*
104077fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a
104177fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change
104277fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument.
104377fc1d5bSdan **
104477fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added
104577fc1d5bSdan ** to the changed-rows hash table associated with table pTab.
104677fc1d5bSdan */
1047e8d5648eSdan static void sessionPreupdateOneChange(
104877fc1d5bSdan   int op,                         /* One of SQLITE_UPDATE, INSERT, DELETE */
104977fc1d5bSdan   sqlite3_session *pSession,      /* Session object pTab is attached to */
105077fc1d5bSdan   SessionTable *pTab              /* Table that change applies to */
1051e8d5648eSdan ){
1052e8d5648eSdan   int iHash;
1053cf8e9144Sdan   int bNull = 0;
1054e8d5648eSdan   int rc = SQLITE_OK;
1055e8d5648eSdan 
1056e8d5648eSdan   if( pSession->rc ) return;
1057e8d5648eSdan 
1058e8d5648eSdan   /* Load table details if required */
1059e8d5648eSdan   if( sessionInitTable(pSession, pTab) ) return;
1060e8d5648eSdan 
1061e8d5648eSdan   /* Grow the hash table if required */
106264277f4aSdan   if( sessionGrowHash(0, pTab) ){
10635d607a6eSdan     pSession->rc = SQLITE_NOMEM;
10645d607a6eSdan     return;
10655d607a6eSdan   }
1066e8d5648eSdan 
106780fe2d93Sdan   /* Calculate the hash-key for this change. If the primary key of the row
106880fe2d93Sdan   ** includes a NULL value, exit early. Such changes are ignored by the
106980fe2d93Sdan   ** session module. */
1070cf8e9144Sdan   rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull);
107180fe2d93Sdan   if( rc!=SQLITE_OK ) goto error_out;
107280fe2d93Sdan 
1073cf8e9144Sdan   if( bNull==0 ){
107480fe2d93Sdan     /* Search the hash table for an existing record for this row. */
1075b4480e94Sdan     SessionChange *pC;
10766734007dSdan     for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){
1077cf8e9144Sdan       if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break;
1078e8d5648eSdan     }
107980fe2d93Sdan 
1080e8d5648eSdan     if( pC==0 ){
1081e8d5648eSdan       /* Create a new change object containing all the old values (if
1082e8d5648eSdan       ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK
1083e8d5648eSdan       ** values (if this is an INSERT). */
1084b4480e94Sdan       SessionChange *pChange; /* New change object */
1085e8d5648eSdan       int nByte;              /* Number of bytes to allocate */
1086e8d5648eSdan       int i;                  /* Used to iterate through columns */
1087e8d5648eSdan 
1088b4480e94Sdan       assert( rc==SQLITE_OK );
1089e8d5648eSdan       pTab->nEntry++;
1090e8d5648eSdan 
1091e8d5648eSdan       /* Figure out how large an allocation is required */
1092e8d5648eSdan       nByte = sizeof(SessionChange);
109380fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1094e8d5648eSdan         sqlite3_value *p = 0;
1095e8d5648eSdan         if( op!=SQLITE_INSERT ){
1096cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p);
109780fe2d93Sdan           assert( trc==SQLITE_OK );
109880fe2d93Sdan         }else if( pTab->abPK[i] ){
1099cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p);
110080fe2d93Sdan           assert( trc==SQLITE_OK );
1101e8d5648eSdan         }
110280fe2d93Sdan 
110380fe2d93Sdan         /* This may fail if SQLite value p contains a utf-16 string that must
110480fe2d93Sdan         ** be converted to utf-8 and an OOM error occurs while doing so. */
1105e8d5648eSdan         rc = sessionSerializeValue(0, p, &nByte);
110680fe2d93Sdan         if( rc!=SQLITE_OK ) goto error_out;
1107e8d5648eSdan       }
1108e8d5648eSdan 
1109e8d5648eSdan       /* Allocate the change object */
1110e8d5648eSdan       pChange = (SessionChange *)sqlite3_malloc(nByte);
1111e8d5648eSdan       if( !pChange ){
1112e8d5648eSdan         rc = SQLITE_NOMEM;
111380fe2d93Sdan         goto error_out;
1114e8d5648eSdan       }else{
1115e8d5648eSdan         memset(pChange, 0, sizeof(SessionChange));
1116e8d5648eSdan         pChange->aRecord = (u8 *)&pChange[1];
1117e8d5648eSdan       }
1118e8d5648eSdan 
111980fe2d93Sdan       /* Populate the change object. None of the preupdate_old(),
112080fe2d93Sdan       ** preupdate_new() or SerializeValue() calls below may fail as all
112180fe2d93Sdan       ** required values and encodings have already been cached in memory.
112280fe2d93Sdan       ** It is not possible for an OOM to occur in this block. */
1123e8d5648eSdan       nByte = 0;
112480fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1125e8d5648eSdan         sqlite3_value *p = 0;
1126e8d5648eSdan         if( op!=SQLITE_INSERT ){
1127cf8e9144Sdan           pSession->hook.xOld(pSession->hook.pCtx, i, &p);
112880fe2d93Sdan         }else if( pTab->abPK[i] ){
1129cf8e9144Sdan           pSession->hook.xNew(pSession->hook.pCtx, i, &p);
1130e8d5648eSdan         }
113180fe2d93Sdan         sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte);
1132e8d5648eSdan       }
113380fe2d93Sdan 
113480fe2d93Sdan       /* Add the change to the hash-table */
1135cf8e9144Sdan       if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){
1136b4480e94Sdan         pChange->bIndirect = 1;
1137b4480e94Sdan       }
113812ca0b56Sdan       pChange->nRecord = nByte;
1139798693b2Sdan       pChange->op = op;
1140e8d5648eSdan       pChange->pNext = pTab->apChange[iHash];
1141e8d5648eSdan       pTab->apChange[iHash] = pChange;
114280fe2d93Sdan 
114380fe2d93Sdan     }else if( pC->bIndirect ){
1144b4480e94Sdan       /* If the existing change is considered "indirect", but this current
1145b4480e94Sdan       ** change is "direct", mark the change object as direct. */
1146cf8e9144Sdan       if( pSession->hook.xDepth(pSession->hook.pCtx)==0
1147cf8e9144Sdan        && pSession->bIndirect==0
1148cf8e9144Sdan       ){
1149b4480e94Sdan         pC->bIndirect = 0;
1150b4480e94Sdan       }
1151e8d5648eSdan     }
11524fccf43aSdan   }
115312ca0b56Sdan 
115412ca0b56Sdan   /* If an error has occurred, mark the session object as failed. */
115580fe2d93Sdan  error_out:
115612ca0b56Sdan   if( rc!=SQLITE_OK ){
115712ca0b56Sdan     pSession->rc = rc;
115812ca0b56Sdan   }
115927453faeSdan }
11604fccf43aSdan 
1161cf8e9144Sdan static int sessionFindTable(
1162cf8e9144Sdan   sqlite3_session *pSession,
1163cf8e9144Sdan   const char *zName,
1164cf8e9144Sdan   SessionTable **ppTab
1165cf8e9144Sdan ){
1166cf8e9144Sdan   int rc = SQLITE_OK;
1167cf8e9144Sdan   int nName = sqlite3Strlen30(zName);
1168cf8e9144Sdan   SessionTable *pRet;
1169cf8e9144Sdan 
1170cf8e9144Sdan   /* Search for an existing table */
1171cf8e9144Sdan   for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){
1172cf8e9144Sdan     if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break;
1173cf8e9144Sdan   }
1174cf8e9144Sdan 
1175cf8e9144Sdan   if( pRet==0 && pSession->bAutoAttach ){
1176cf8e9144Sdan     /* If there is a table-filter configured, invoke it. If it returns 0,
1177cf8e9144Sdan     ** do not automatically add the new table. */
1178cf8e9144Sdan     if( pSession->xTableFilter==0
1179cf8e9144Sdan      || pSession->xTableFilter(pSession->pFilterCtx, zName)
1180cf8e9144Sdan     ){
1181cf8e9144Sdan       rc = sqlite3session_attach(pSession, zName);
1182cf8e9144Sdan       if( rc==SQLITE_OK ){
1183cf8e9144Sdan         pRet = pSession->pTable;
1184cf8e9144Sdan         assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) );
1185cf8e9144Sdan       }
1186cf8e9144Sdan     }
1187cf8e9144Sdan   }
1188cf8e9144Sdan 
1189cf8e9144Sdan   assert( rc==SQLITE_OK || pRet==0 );
1190cf8e9144Sdan   *ppTab = pRet;
1191cf8e9144Sdan   return rc;
1192cf8e9144Sdan }
1193cf8e9144Sdan 
11944fccf43aSdan /*
11954fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases.
11964fccf43aSdan */
11974fccf43aSdan static void xPreUpdate(
11984fccf43aSdan   void *pCtx,                     /* Copy of third arg to preupdate_hook() */
11994fccf43aSdan   sqlite3 *db,                    /* Database handle */
12004fccf43aSdan   int op,                         /* SQLITE_UPDATE, DELETE or INSERT */
12014fccf43aSdan   char const *zDb,                /* Database name */
12024fccf43aSdan   char const *zName,              /* Table name */
12034fccf43aSdan   sqlite3_int64 iKey1,            /* Rowid of row about to be deleted/updated */
12044fccf43aSdan   sqlite3_int64 iKey2             /* New rowid value (for a rowid UPDATE) */
12054fccf43aSdan ){
12064fccf43aSdan   sqlite3_session *pSession;
1207cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb);
12084fccf43aSdan 
12094c220252Sdan   assert( sqlite3_mutex_held(db->mutex) );
12104c220252Sdan 
12114fccf43aSdan   for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){
12124fccf43aSdan     SessionTable *pTab;
1213296c7658Sdan 
1214e8d5648eSdan     /* If this session is attached to a different database ("main", "temp"
1215e8d5648eSdan     ** etc.), or if it is not currently enabled, there is nothing to do. Skip
1216e8d5648eSdan     ** to the next session object attached to this database. */
1217296c7658Sdan     if( pSession->bEnable==0 ) continue;
12184fccf43aSdan     if( pSession->rc ) continue;
12194fccf43aSdan     if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue;
1220296c7658Sdan 
1221cf8e9144Sdan     pSession->rc = sessionFindTable(pSession, zName, &pTab);
1222cf8e9144Sdan     if( pTab ){
1223cf8e9144Sdan       assert( pSession->rc==SQLITE_OK );
1224e8d5648eSdan       sessionPreupdateOneChange(op, pSession, pTab);
1225e8d5648eSdan       if( op==SQLITE_UPDATE ){
1226e8d5648eSdan         sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab);
12274fccf43aSdan       }
12284fccf43aSdan     }
12294fccf43aSdan   }
12304fccf43aSdan }
1231cf8e9144Sdan 
1232cf8e9144Sdan /*
1233cf8e9144Sdan ** The pre-update hook implementations.
1234cf8e9144Sdan */
1235cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1236cf8e9144Sdan   return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal);
1237cf8e9144Sdan }
1238cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1239cf8e9144Sdan   return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal);
1240cf8e9144Sdan }
1241cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){
1242cf8e9144Sdan   return sqlite3_preupdate_count((sqlite3*)pCtx);
1243cf8e9144Sdan }
1244cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){
1245cf8e9144Sdan   return sqlite3_preupdate_depth((sqlite3*)pCtx);
1246cf8e9144Sdan }
1247cf8e9144Sdan 
1248cf8e9144Sdan /*
1249cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only
1250cf8e9144Sdan ** argument.
1251cf8e9144Sdan */
1252cf8e9144Sdan static void sessionPreupdateHooks(
1253cf8e9144Sdan   sqlite3_session *pSession
1254cf8e9144Sdan ){
1255cf8e9144Sdan   pSession->hook.pCtx = (void*)pSession->db;
1256cf8e9144Sdan   pSession->hook.xOld = sessionPreupdateOld;
1257cf8e9144Sdan   pSession->hook.xNew = sessionPreupdateNew;
1258cf8e9144Sdan   pSession->hook.xCount = sessionPreupdateCount;
1259cf8e9144Sdan   pSession->hook.xDepth = sessionPreupdateDepth;
1260cf8e9144Sdan }
1261cf8e9144Sdan 
1262cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx;
1263cf8e9144Sdan struct SessionDiffCtx {
1264cf8e9144Sdan   sqlite3_stmt *pStmt;
1265cf8e9144Sdan   int nOldOff;
1266cf8e9144Sdan };
1267cf8e9144Sdan 
1268cf8e9144Sdan /*
1269cf8e9144Sdan ** The diff hook implementations.
1270cf8e9144Sdan */
1271cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1272cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1273cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff);
1274cf8e9144Sdan   return SQLITE_OK;
1275cf8e9144Sdan }
1276cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1277cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1278cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal);
1279cf8e9144Sdan    return SQLITE_OK;
1280cf8e9144Sdan }
1281cf8e9144Sdan static int sessionDiffCount(void *pCtx){
1282cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1283cf8e9144Sdan   return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt);
1284cf8e9144Sdan }
1285cf8e9144Sdan static int sessionDiffDepth(void *pCtx){
1286cf8e9144Sdan   return 0;
1287cf8e9144Sdan }
1288cf8e9144Sdan 
1289cf8e9144Sdan /*
1290cf8e9144Sdan ** Install the diff hooks on the session object passed as the only
1291cf8e9144Sdan ** argument.
1292cf8e9144Sdan */
1293cf8e9144Sdan static void sessionDiffHooks(
1294cf8e9144Sdan   sqlite3_session *pSession,
1295cf8e9144Sdan   SessionDiffCtx *pDiffCtx
1296cf8e9144Sdan ){
1297cf8e9144Sdan   pSession->hook.pCtx = (void*)pDiffCtx;
1298cf8e9144Sdan   pSession->hook.xOld = sessionDiffOld;
1299cf8e9144Sdan   pSession->hook.xNew = sessionDiffNew;
1300cf8e9144Sdan   pSession->hook.xCount = sessionDiffCount;
1301cf8e9144Sdan   pSession->hook.xDepth = sessionDiffDepth;
1302cf8e9144Sdan }
1303cf8e9144Sdan 
1304cf8e9144Sdan static char *sessionExprComparePK(
1305cf8e9144Sdan   int nCol,
1306cf8e9144Sdan   const char *zDb1, const char *zDb2,
1307cf8e9144Sdan   const char *zTab,
1308cf8e9144Sdan   const char **azCol, u8 *abPK
1309cf8e9144Sdan ){
1310cf8e9144Sdan   int i;
1311cf8e9144Sdan   const char *zSep = "";
1312cf8e9144Sdan   char *zRet = 0;
1313cf8e9144Sdan 
1314cf8e9144Sdan   for(i=0; i<nCol; i++){
1315cf8e9144Sdan     if( abPK[i] ){
1316cf8e9144Sdan       zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"",
1317cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1318cf8e9144Sdan       );
1319cf8e9144Sdan       zSep = " AND ";
1320cf8e9144Sdan       if( zRet==0 ) break;
1321cf8e9144Sdan     }
1322cf8e9144Sdan   }
1323cf8e9144Sdan 
1324cf8e9144Sdan   return zRet;
1325cf8e9144Sdan }
1326cf8e9144Sdan 
1327cf8e9144Sdan static char *sessionExprCompareOther(
1328cf8e9144Sdan   int nCol,
1329cf8e9144Sdan   const char *zDb1, const char *zDb2,
1330cf8e9144Sdan   const char *zTab,
1331cf8e9144Sdan   const char **azCol, u8 *abPK
1332cf8e9144Sdan ){
1333cf8e9144Sdan   int i;
1334cf8e9144Sdan   const char *zSep = "";
1335cf8e9144Sdan   char *zRet = 0;
1336cf8e9144Sdan   int bHave = 0;
1337cf8e9144Sdan 
1338cf8e9144Sdan   for(i=0; i<nCol; i++){
1339cf8e9144Sdan     if( abPK[i]==0 ){
1340cf8e9144Sdan       bHave = 1;
1341cf8e9144Sdan       zRet = sqlite3_mprintf(
1342cf8e9144Sdan           "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"",
1343cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1344cf8e9144Sdan       );
1345cf8e9144Sdan       zSep = " OR ";
1346cf8e9144Sdan       if( zRet==0 ) break;
1347cf8e9144Sdan     }
1348cf8e9144Sdan   }
1349cf8e9144Sdan 
1350cf8e9144Sdan   if( bHave==0 ){
1351cf8e9144Sdan     assert( zRet==0 );
1352cf8e9144Sdan     zRet = sqlite3_mprintf("0");
1353cf8e9144Sdan   }
1354cf8e9144Sdan 
1355cf8e9144Sdan   return zRet;
1356cf8e9144Sdan }
1357cf8e9144Sdan 
1358cf8e9144Sdan static char *sessionSelectFindNew(
1359cf8e9144Sdan   int nCol,
1360cf8e9144Sdan   const char *zDb1,      /* Pick rows in this db only */
1361cf8e9144Sdan   const char *zDb2,      /* But not in this one */
1362cf8e9144Sdan   const char *zTbl,      /* Table name */
1363cf8e9144Sdan   const char *zExpr
1364cf8e9144Sdan ){
1365cf8e9144Sdan   char *zRet = sqlite3_mprintf(
1366cf8e9144Sdan       "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS ("
1367cf8e9144Sdan       "  SELECT 1 FROM \"%w\".\"%w\" WHERE %s"
1368cf8e9144Sdan       ")",
1369cf8e9144Sdan       zDb1, zTbl, zDb2, zTbl, zExpr
1370cf8e9144Sdan   );
1371cf8e9144Sdan   return zRet;
1372cf8e9144Sdan }
1373cf8e9144Sdan 
1374cf8e9144Sdan static int sessionDiffFindNew(
1375cf8e9144Sdan   int op,
1376cf8e9144Sdan   sqlite3_session *pSession,
1377cf8e9144Sdan   SessionTable *pTab,
1378cf8e9144Sdan   const char *zDb1,
1379cf8e9144Sdan   const char *zDb2,
1380cf8e9144Sdan   char *zExpr
1381cf8e9144Sdan ){
1382cf8e9144Sdan   int rc = SQLITE_OK;
1383cf8e9144Sdan   char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr);
1384cf8e9144Sdan 
1385cf8e9144Sdan   if( zStmt==0 ){
1386cf8e9144Sdan     rc = SQLITE_NOMEM;
1387cf8e9144Sdan   }else{
1388cf8e9144Sdan     sqlite3_stmt *pStmt;
1389cf8e9144Sdan     rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1390cf8e9144Sdan     if( rc==SQLITE_OK ){
1391cf8e9144Sdan       SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1392cf8e9144Sdan       pDiffCtx->pStmt = pStmt;
1393cf8e9144Sdan       pDiffCtx->nOldOff = 0;
1394cf8e9144Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
1395cf8e9144Sdan         sessionPreupdateOneChange(op, pSession, pTab);
1396cf8e9144Sdan       }
1397cf8e9144Sdan       rc = sqlite3_finalize(pStmt);
1398cf8e9144Sdan     }
1399cf8e9144Sdan     sqlite3_free(zStmt);
1400cf8e9144Sdan   }
1401cf8e9144Sdan 
1402cf8e9144Sdan   return rc;
1403cf8e9144Sdan }
1404cf8e9144Sdan 
1405cf8e9144Sdan static int sessionDiffFindModified(
1406cf8e9144Sdan   sqlite3_session *pSession,
1407cf8e9144Sdan   SessionTable *pTab,
1408cf8e9144Sdan   const char *zFrom,
1409cf8e9144Sdan   const char *zExpr
1410cf8e9144Sdan ){
1411cf8e9144Sdan   int rc = SQLITE_OK;
1412cf8e9144Sdan 
1413cf8e9144Sdan   char *zExpr2 = sessionExprCompareOther(pTab->nCol,
1414cf8e9144Sdan       pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK
1415cf8e9144Sdan   );
1416cf8e9144Sdan   if( zExpr2==0 ){
1417cf8e9144Sdan     rc = SQLITE_NOMEM;
1418cf8e9144Sdan   }else{
1419cf8e9144Sdan     char *zStmt = sqlite3_mprintf(
1420*dd009f83Sdan         "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)",
1421cf8e9144Sdan         pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2
1422cf8e9144Sdan     );
1423cf8e9144Sdan     if( zStmt==0 ){
1424cf8e9144Sdan       rc = SQLITE_NOMEM;
1425cf8e9144Sdan     }else{
1426cf8e9144Sdan       sqlite3_stmt *pStmt;
1427cf8e9144Sdan       rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1428cf8e9144Sdan 
1429cf8e9144Sdan       if( rc==SQLITE_OK ){
1430cf8e9144Sdan         SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1431cf8e9144Sdan         pDiffCtx->pStmt = pStmt;
1432cf8e9144Sdan         pDiffCtx->nOldOff = pTab->nCol;
1433cf8e9144Sdan         while( SQLITE_ROW==sqlite3_step(pStmt) ){
1434cf8e9144Sdan           sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab);
1435cf8e9144Sdan         }
1436cf8e9144Sdan         rc = sqlite3_finalize(pStmt);
1437cf8e9144Sdan       }
1438cf8e9144Sdan       sqlite3_free(zStmt);
1439cf8e9144Sdan     }
1440cf8e9144Sdan   }
1441cf8e9144Sdan 
1442cf8e9144Sdan   return rc;
1443cf8e9144Sdan }
1444cf8e9144Sdan 
1445cf8e9144Sdan int sqlite3session_diff(
1446cf8e9144Sdan   sqlite3_session *pSession,
1447cf8e9144Sdan   const char *zFrom,
1448cf8e9144Sdan   const char *zTbl,
1449cf8e9144Sdan   char **pzErrMsg
1450cf8e9144Sdan ){
1451cf8e9144Sdan   const char *zDb = pSession->zDb;
1452cf8e9144Sdan   int rc = pSession->rc;
1453cf8e9144Sdan   SessionDiffCtx d;
1454cf8e9144Sdan 
1455cf8e9144Sdan   memset(&d, 0, sizeof(d));
1456cf8e9144Sdan   sessionDiffHooks(pSession, &d);
1457cf8e9144Sdan 
145810dc553cSdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1459cf8e9144Sdan   if( pzErrMsg ) *pzErrMsg = 0;
1460cf8e9144Sdan   if( rc==SQLITE_OK ){
1461cf8e9144Sdan     char *zExpr = 0;
1462cf8e9144Sdan     sqlite3 *db = pSession->db;
1463cf8e9144Sdan     SessionTable *pTo;            /* Table zTbl */
1464cf8e9144Sdan 
1465cf8e9144Sdan     /* Locate and if necessary initialize the target table object */
1466cf8e9144Sdan     rc = sessionFindTable(pSession, zTbl, &pTo);
1467cf8e9144Sdan     if( pTo==0 ) goto diff_out;
1468cf8e9144Sdan     if( pTo->nCol==0 ){
1469cf8e9144Sdan       rc = pSession->rc = sessionTableInfo(db, zDb,
1470cf8e9144Sdan           pTo->zName, &pTo->nCol, 0, &pTo->azCol, &pTo->abPK
1471cf8e9144Sdan       );
1472cf8e9144Sdan     }
1473cf8e9144Sdan 
1474cf8e9144Sdan     /* Check the table schemas match */
1475cf8e9144Sdan     if( rc==SQLITE_OK ){
1476cf8e9144Sdan       int nCol;                   /* Columns in zFrom.zTbl */
1477cf8e9144Sdan       u8 *abPK;
1478cf8e9144Sdan       const char **azCol = 0;
1479cf8e9144Sdan       rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK);
1480cf8e9144Sdan       if( rc==SQLITE_OK ){
1481cf8e9144Sdan         int bMismatch = 0;
1482cf8e9144Sdan         if( pTo->nCol!=nCol || memcmp(pTo->abPK, abPK, nCol) ){
1483cf8e9144Sdan           bMismatch = 1;
1484cf8e9144Sdan         }else{
1485cf8e9144Sdan           int i;
1486cf8e9144Sdan           for(i=0; i<nCol; i++){
1487cf8e9144Sdan             if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1;
1488cf8e9144Sdan           }
1489cf8e9144Sdan         }
1490cf8e9144Sdan 
1491cf8e9144Sdan         if( bMismatch ){
1492cf8e9144Sdan           *pzErrMsg = sqlite3_mprintf("table schemas do not match");
1493cf8e9144Sdan           rc = SQLITE_ERROR;
1494cf8e9144Sdan         }
1495cf8e9144Sdan       }
1496cf8e9144Sdan       sqlite3_free(azCol);
1497cf8e9144Sdan     }
1498cf8e9144Sdan 
1499cf8e9144Sdan     if( rc==SQLITE_OK ){
1500cf8e9144Sdan       zExpr = sessionExprComparePK(pTo->nCol,
1501cf8e9144Sdan           zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK
1502cf8e9144Sdan       );
1503cf8e9144Sdan     }
1504cf8e9144Sdan 
1505cf8e9144Sdan     /* Find new rows */
1506cf8e9144Sdan     if( rc==SQLITE_OK ){
1507cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr);
1508cf8e9144Sdan     }
1509cf8e9144Sdan 
1510cf8e9144Sdan     /* Find old rows */
1511cf8e9144Sdan     if( rc==SQLITE_OK ){
1512cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr);
1513cf8e9144Sdan     }
1514cf8e9144Sdan 
1515cf8e9144Sdan     /* Find modified rows */
1516cf8e9144Sdan     if( rc==SQLITE_OK ){
1517cf8e9144Sdan       rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr);
1518cf8e9144Sdan     }
1519cf8e9144Sdan 
1520cf8e9144Sdan     sqlite3_free(zExpr);
1521cf8e9144Sdan   }
1522cf8e9144Sdan 
1523cf8e9144Sdan  diff_out:
1524cf8e9144Sdan   sessionPreupdateHooks(pSession);
152510dc553cSdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
1526cf8e9144Sdan   return rc;
1527296c7658Sdan }
15284fccf43aSdan 
15294fccf43aSdan /*
15304fccf43aSdan ** Create a session object. This session object will record changes to
15314fccf43aSdan ** database zDb attached to connection db.
15324fccf43aSdan */
15334fccf43aSdan int sqlite3session_create(
15344fccf43aSdan   sqlite3 *db,                    /* Database handle */
15354fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
15364fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
15374fccf43aSdan ){
1538296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1539296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1540cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
15414fccf43aSdan 
1542296c7658Sdan   /* Zero the output value in case an error occurs. */
15434fccf43aSdan   *ppSession = 0;
15444fccf43aSdan 
15454fccf43aSdan   /* Allocate and populate the new session object. */
15464fccf43aSdan   pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1);
15474fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
15484fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
15494fccf43aSdan   pNew->db = db;
15504fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1551296c7658Sdan   pNew->bEnable = 1;
15524fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
1553cf8e9144Sdan   sessionPreupdateHooks(pNew);
15544fccf43aSdan 
15554fccf43aSdan   /* Add the new session object to the linked list of session objects
15564fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
15574fccf43aSdan   ** handle mutex.  */
15584fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
15594fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
15604fccf43aSdan   pNew->pNext = pOld;
15614fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
15624fccf43aSdan 
15634fccf43aSdan   *ppSession = pNew;
15644fccf43aSdan   return SQLITE_OK;
15654fccf43aSdan }
15664fccf43aSdan 
156777fc1d5bSdan /*
156877fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
156977fc1d5bSdan ** of the changed-rows hash tables are also deleted.
157077fc1d5bSdan */
15711ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
15725d607a6eSdan   SessionTable *pNext;
15735d607a6eSdan   SessionTable *pTab;
15745d607a6eSdan 
15755d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
15765d607a6eSdan     int i;
15775d607a6eSdan     pNext = pTab->pNext;
15785d607a6eSdan     for(i=0; i<pTab->nChange; i++){
15795d607a6eSdan       SessionChange *p;
15805d607a6eSdan       SessionChange *pNext;
15815d607a6eSdan       for(p=pTab->apChange[i]; p; p=pNext){
15825d607a6eSdan         pNext = p->pNext;
15835d607a6eSdan         sqlite3_free(p);
15845d607a6eSdan       }
15855d607a6eSdan     }
15865d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
15875d607a6eSdan     sqlite3_free(pTab->apChange);
15885d607a6eSdan     sqlite3_free(pTab);
15895d607a6eSdan   }
15905d607a6eSdan }
15915d607a6eSdan 
15924fccf43aSdan /*
15934fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
15944fccf43aSdan */
15954fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
15964fccf43aSdan   sqlite3 *db = pSession->db;
15974fccf43aSdan   sqlite3_session *pHead;
15984fccf43aSdan   sqlite3_session **pp;
15994fccf43aSdan 
1600296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1601296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
16024fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
16034fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
16044fccf43aSdan   for(pp=&pHead; (*pp)!=pSession; pp=&((*pp)->pNext));
16054fccf43aSdan   *pp = (*pp)->pNext;
16064fccf43aSdan   if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void *)pHead);
16074fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
16084fccf43aSdan 
1609296c7658Sdan   /* Delete all attached table objects. And the contents of their
1610296c7658Sdan   ** associated hash-tables. */
16115d607a6eSdan   sessionDeleteTable(pSession->pTable);
16124fccf43aSdan 
1613296c7658Sdan   /* Free the session object itself. */
16144fccf43aSdan   sqlite3_free(pSession);
16154fccf43aSdan }
16164fccf43aSdan 
16174fccf43aSdan /*
16187531a5a3Sdan ** Set a table filter on a Session Object.
16197531a5a3Sdan */
16207531a5a3Sdan void sqlite3session_table_filter(
16217531a5a3Sdan   sqlite3_session *pSession,
16227531a5a3Sdan   int(*xFilter)(void*, const char*),
16237531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
16247531a5a3Sdan ){
16257531a5a3Sdan   pSession->bAutoAttach = 1;
16267531a5a3Sdan   pSession->pFilterCtx = pCtx;
16277531a5a3Sdan   pSession->xTableFilter = xFilter;
16287531a5a3Sdan }
16297531a5a3Sdan 
16307531a5a3Sdan /*
16314fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
16324fccf43aSdan ** while the session object is enabled will be recorded.
16334fccf43aSdan **
16344fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
16354fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
16364fccf43aSdan ** or not.
16374fccf43aSdan */
16384fccf43aSdan int sqlite3session_attach(
16394fccf43aSdan   sqlite3_session *pSession,      /* Session object */
16404fccf43aSdan   const char *zName               /* Table name */
16414fccf43aSdan ){
1642ff4d0f41Sdan   int rc = SQLITE_OK;
1643ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1644ff4d0f41Sdan 
1645ff4d0f41Sdan   if( !zName ){
1646ff4d0f41Sdan     pSession->bAutoAttach = 1;
1647ff4d0f41Sdan   }else{
1648296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1649296c7658Sdan     int nName;                    /* Number of bytes in string zName */
16504fccf43aSdan 
16514fccf43aSdan     /* First search for an existing entry. If one is found, this call is
16524fccf43aSdan     ** a no-op. Return early. */
1653cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
16544fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
16554c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
16564fccf43aSdan     }
16574fccf43aSdan 
16584c220252Sdan     if( !pTab ){
16594fccf43aSdan       /* Allocate new SessionTable object. */
16604fccf43aSdan       pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1);
16614c220252Sdan       if( !pTab ){
16624c220252Sdan         rc = SQLITE_NOMEM;
16634c220252Sdan       }else{
16644fccf43aSdan         /* Populate the new SessionTable object and link it into the list. */
16654fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
16664fccf43aSdan         pTab->zName = (char *)&pTab[1];
16674fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
16684fccf43aSdan         pTab->pNext = pSession->pTable;
16694fccf43aSdan         pSession->pTable = pTab;
16704c220252Sdan       }
16714c220252Sdan     }
1672ff4d0f41Sdan   }
16734fccf43aSdan 
16744c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
16754c220252Sdan   return rc;
16764fccf43aSdan }
16774fccf43aSdan 
1678296c7658Sdan /*
1679296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1680296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1681296c7658Sdan **
1682296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1683296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1684296c7658Sdan */
16854fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){
168680fe2d93Sdan   if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){
16874fccf43aSdan     u8 *aNew;
16884fccf43aSdan     int nNew = p->nAlloc ? p->nAlloc : 128;
16894fccf43aSdan     do {
16904fccf43aSdan       nNew = nNew*2;
1691ef7a6304Sdan     }while( nNew<(p->nBuf+nByte) );
16924fccf43aSdan 
16934fccf43aSdan     aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew);
16944fccf43aSdan     if( 0==aNew ){
16954fccf43aSdan       *pRc = SQLITE_NOMEM;
169680fe2d93Sdan     }else{
16974fccf43aSdan       p->aBuf = aNew;
16984fccf43aSdan       p->nAlloc = nNew;
16994fccf43aSdan     }
170080fe2d93Sdan   }
170180fe2d93Sdan   return (*pRc!=SQLITE_OK);
17024fccf43aSdan }
17034fccf43aSdan 
1704296c7658Sdan /*
1705fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1706fa122adaSdan ** as the first.
1707fa122adaSdan **
1708fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1709fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1710fa122adaSdan ** before returning.
1711fa122adaSdan */
1712fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1713fa122adaSdan   int rc = *pRc;
1714fa122adaSdan   if( rc==SQLITE_OK ){
1715fa122adaSdan     int nByte = 0;
1716e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1717fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1718fa122adaSdan     if( rc==SQLITE_OK ){
1719fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1720fa122adaSdan       p->nBuf += nByte;
1721fa122adaSdan     }else{
1722fa122adaSdan       *pRc = rc;
1723fa122adaSdan     }
1724fa122adaSdan   }
1725fa122adaSdan }
1726fa122adaSdan 
1727fa122adaSdan /*
1728296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1729296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1730296c7658Sdan **
1731296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1732296c7658Sdan ** returning.
1733296c7658Sdan */
17344fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
173580fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
17364fccf43aSdan     p->aBuf[p->nBuf++] = v;
17374fccf43aSdan   }
17384fccf43aSdan }
17394fccf43aSdan 
1740296c7658Sdan /*
1741296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1742296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1743296c7658Sdan **
1744296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1745296c7658Sdan ** returning.
1746296c7658Sdan */
1747cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
174880fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
17494fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
17504fccf43aSdan   }
17514fccf43aSdan }
17524fccf43aSdan 
1753296c7658Sdan /*
1754296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1755296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1756296c7658Sdan **
1757296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1758296c7658Sdan ** returning.
1759296c7658Sdan */
17604fccf43aSdan static void sessionAppendBlob(
17614fccf43aSdan   SessionBuffer *p,
17624fccf43aSdan   const u8 *aBlob,
17634fccf43aSdan   int nBlob,
17644fccf43aSdan   int *pRc
17654fccf43aSdan ){
176680fe2d93Sdan   if( 0==sessionBufferGrow(p, nBlob, pRc) ){
17674fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
17684fccf43aSdan     p->nBuf += nBlob;
17694fccf43aSdan   }
17704fccf43aSdan }
17714fccf43aSdan 
1772296c7658Sdan /*
1773296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1774296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1775296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1776296c7658Sdan **
1777296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1778296c7658Sdan ** returning.
1779296c7658Sdan */
1780d5f0767cSdan static void sessionAppendStr(
1781d5f0767cSdan   SessionBuffer *p,
1782d5f0767cSdan   const char *zStr,
1783d5f0767cSdan   int *pRc
1784d5f0767cSdan ){
1785cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
178680fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1787d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1788d5f0767cSdan     p->nBuf += nStr;
1789d5f0767cSdan   }
1790d5f0767cSdan }
1791d5f0767cSdan 
1792296c7658Sdan /*
1793296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1794296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1795296c7658Sdan ** to the buffer. No nul-terminator is written.
1796296c7658Sdan **
1797296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1798296c7658Sdan ** returning.
1799296c7658Sdan */
1800d5f0767cSdan static void sessionAppendInteger(
1801296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1802296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1803296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1804d5f0767cSdan ){
1805d5f0767cSdan   char aBuf[24];
1806d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1807d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1808d5f0767cSdan }
1809d5f0767cSdan 
1810296c7658Sdan /*
1811296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1812296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1813296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1814296c7658Sdan ** nul-terminator byte is written.
1815296c7658Sdan **
1816296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1817296c7658Sdan ** returning.
1818296c7658Sdan */
1819d5f0767cSdan static void sessionAppendIdent(
1820296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1821296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1822296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1823d5f0767cSdan ){
1824cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
182580fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1826d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1827d5f0767cSdan     const char *zIn = zStr;
1828d5f0767cSdan     *zOut++ = '"';
1829d5f0767cSdan     while( *zIn ){
1830d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1831d5f0767cSdan       *zOut++ = *(zIn++);
1832d5f0767cSdan     }
1833d5f0767cSdan     *zOut++ = '"';
1834cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1835d5f0767cSdan   }
1836d5f0767cSdan }
1837d5f0767cSdan 
1838296c7658Sdan /*
1839296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1840296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1841296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1842296c7658Sdan ** to to the buffer.
1843296c7658Sdan */
18444fccf43aSdan static void sessionAppendCol(
1845296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1846296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1847296c7658Sdan   int iCol,                       /* Column to read value from */
1848296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
18494fccf43aSdan ){
18504fccf43aSdan   if( *pRc==SQLITE_OK ){
18514fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
18524fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
18534fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
18544fccf43aSdan       sqlite3_int64 i;
18554fccf43aSdan       u8 aBuf[8];
18564fccf43aSdan       if( eType==SQLITE_INTEGER ){
18574fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
18584fccf43aSdan       }else{
18594fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
18604fccf43aSdan         memcpy(&i, &r, 8);
18614fccf43aSdan       }
1862296c7658Sdan       sessionPutI64(aBuf, i);
18634fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
18644fccf43aSdan     }
18654fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
18666734007dSdan       u8 *z;
18673cc89d95Sdan       int nByte;
18686734007dSdan       if( eType==SQLITE_BLOB ){
18696734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
18706734007dSdan       }else{
18716734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
18726734007dSdan       }
18733cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
18743cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
18754fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
18766734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
18776734007dSdan       }else{
18786734007dSdan         *pRc = SQLITE_NOMEM;
18796734007dSdan       }
18804fccf43aSdan     }
18814fccf43aSdan   }
18824fccf43aSdan }
18834fccf43aSdan 
1884296c7658Sdan /*
1885296c7658Sdan **
188680fe2d93Sdan ** This function appends an update change to the buffer (see the comments
188780fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
188880fe2d93Sdan ** consists of:
1889296c7658Sdan **
1890296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
1891296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
1892296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
1893296c7658Sdan **
1894296c7658Sdan ** The SessionChange object passed as the third argument contains the
1895296c7658Sdan ** values that were stored in the row when the session began (the old.*
1896296c7658Sdan ** values). The statement handle passed as the second argument points
1897296c7658Sdan ** at the current version of the row (the new.* values).
1898296c7658Sdan **
1899296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
1900296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
1901296c7658Sdan **
1902296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
1903296c7658Sdan ** original values of any fields that have been modified. The new.* record
1904296c7658Sdan ** contains the new values of only those fields that have been modified.
1905296c7658Sdan */
190680fe2d93Sdan static int sessionAppendUpdate(
1907296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
190873b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
1909296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
1910296c7658Sdan   SessionChange *p,               /* Object containing old values */
191180fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
19124fccf43aSdan ){
191380fe2d93Sdan   int rc = SQLITE_OK;
1914296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
1915296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
19161f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
1917296c7658Sdan   int i;                        /* Used to iterate through columns */
1918296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
1919296c7658Sdan 
192080fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
192180fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
19224fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
192337f133ecSdan     int bChanged = 0;
19244fccf43aSdan     int nAdvance;
19254fccf43aSdan     int eType = *pCsr;
19264fccf43aSdan     switch( eType ){
19274fccf43aSdan       case SQLITE_NULL:
19284fccf43aSdan         nAdvance = 1;
19294fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
193037f133ecSdan           bChanged = 1;
19314fccf43aSdan         }
19324fccf43aSdan         break;
19334fccf43aSdan 
19344fccf43aSdan       case SQLITE_FLOAT:
19354fccf43aSdan       case SQLITE_INTEGER: {
19364fccf43aSdan         nAdvance = 9;
19374fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
19384fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
19394fccf43aSdan           if( eType==SQLITE_INTEGER ){
19404fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
19414fccf43aSdan           }else{
19424fccf43aSdan             double dVal;
19434fccf43aSdan             memcpy(&dVal, &iVal, 8);
19444fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
19454fccf43aSdan           }
19464fccf43aSdan         }
194737f133ecSdan         bChanged = 1;
19484fccf43aSdan         break;
19494fccf43aSdan       }
19504fccf43aSdan 
1951e5754eecSdan       default: {
19524fccf43aSdan         int nByte;
19534fccf43aSdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &nByte);
1954e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
19554fccf43aSdan         nAdvance = nHdr + nByte;
19564fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
19574fccf43aSdan          && nByte==sqlite3_column_bytes(pStmt, i)
19584fccf43aSdan          && 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), nByte)
19594fccf43aSdan         ){
19604fccf43aSdan           break;
19614fccf43aSdan         }
196237f133ecSdan         bChanged = 1;
19634fccf43aSdan       }
19644fccf43aSdan     }
19654fccf43aSdan 
196673b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
196773b3c055Sdan     if( bChanged ) bNoop = 0;
196873b3c055Sdan 
196973b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
197073b3c055Sdan     ** currently generating a patchset. */
197173b3c055Sdan     if( bPatchset==0 ){
197237f133ecSdan       if( bChanged || abPK[i] ){
197380fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
19744fccf43aSdan       }else{
197580fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
197637f133ecSdan       }
197773b3c055Sdan     }
197837f133ecSdan 
197973b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
198073b3c055Sdan     ** generating a patchset.  */
198173b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
198280fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
198337f133ecSdan     }else{
198480fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
19854fccf43aSdan     }
198637f133ecSdan 
19874fccf43aSdan     pCsr += nAdvance;
19884fccf43aSdan   }
19894fccf43aSdan 
19904fccf43aSdan   if( bNoop ){
19911f34f8ccSdan     pBuf->nBuf = nRewind;
19924fccf43aSdan   }else{
199380fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
19944fccf43aSdan   }
19951f34f8ccSdan   sqlite3_free(buf2.aBuf);
199680fe2d93Sdan 
199780fe2d93Sdan   return rc;
1998d5f0767cSdan }
19994fccf43aSdan 
2000a71d2371Sdan /*
2001a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
2002a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
2003a71d2371Sdan ** format otherwise.
2004a71d2371Sdan */
200573b3c055Sdan static int sessionAppendDelete(
200673b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
200773b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
200873b3c055Sdan   SessionChange *p,               /* Object containing old values */
2009a71d2371Sdan   int nCol,                       /* Number of columns in table */
201073b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
201173b3c055Sdan ){
201273b3c055Sdan   int rc = SQLITE_OK;
201373b3c055Sdan 
201473b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
201573b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
201673b3c055Sdan 
201773b3c055Sdan   if( bPatchset==0 ){
201873b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
201973b3c055Sdan   }else{
202073b3c055Sdan     int i;
202173b3c055Sdan     u8 *a = p->aRecord;
202273b3c055Sdan     for(i=0; i<nCol; i++){
202373b3c055Sdan       u8 *pStart = a;
202473b3c055Sdan       int eType = *a++;
202573b3c055Sdan 
202673b3c055Sdan       switch( eType ){
202773b3c055Sdan         case 0:
202873b3c055Sdan         case SQLITE_NULL:
202973b3c055Sdan           assert( abPK[i]==0 );
203073b3c055Sdan           break;
203173b3c055Sdan 
203273b3c055Sdan         case SQLITE_FLOAT:
203373b3c055Sdan         case SQLITE_INTEGER:
203473b3c055Sdan           a += 8;
203573b3c055Sdan           break;
203673b3c055Sdan 
203773b3c055Sdan         default: {
203873b3c055Sdan           int n;
203973b3c055Sdan           a += sessionVarintGet(a, &n);
204073b3c055Sdan           a += n;
204173b3c055Sdan           break;
204273b3c055Sdan         }
204373b3c055Sdan       }
204473b3c055Sdan       if( abPK[i] ){
2045f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
204673b3c055Sdan       }
204773b3c055Sdan     }
204873b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
204973b3c055Sdan   }
205073b3c055Sdan 
205173b3c055Sdan   return rc;
205273b3c055Sdan }
205373b3c055Sdan 
205477fc1d5bSdan /*
205577fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
205677fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
205777fc1d5bSdan **
205877fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
205977fc1d5bSdan */
2060e8d5648eSdan static int sessionSelectStmt(
2061e8d5648eSdan   sqlite3 *db,                    /* Database handle */
2062d7fb7d24Sdan   const char *zDb,                /* Database name */
2063e8d5648eSdan   const char *zTab,               /* Table name */
206477fc1d5bSdan   int nCol,                       /* Number of columns in table */
206577fc1d5bSdan   const char **azCol,             /* Names of table columns */
206677fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
206777fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
2068d5f0767cSdan ){
2069e8d5648eSdan   int rc = SQLITE_OK;
2070d5f0767cSdan   int i;
2071e8d5648eSdan   const char *zSep = "";
2072e8d5648eSdan   SessionBuffer buf = {0, 0, 0};
2073d5f0767cSdan 
2074e8d5648eSdan   sessionAppendStr(&buf, "SELECT * FROM ", &rc);
2075d7fb7d24Sdan   sessionAppendIdent(&buf, zDb, &rc);
2076d7fb7d24Sdan   sessionAppendStr(&buf, ".", &rc);
2077e8d5648eSdan   sessionAppendIdent(&buf, zTab, &rc);
2078e8d5648eSdan   sessionAppendStr(&buf, " WHERE ", &rc);
2079e8d5648eSdan   for(i=0; i<nCol; i++){
2080e8d5648eSdan     if( abPK[i] ){
2081e8d5648eSdan       sessionAppendStr(&buf, zSep, &rc);
2082e8d5648eSdan       sessionAppendIdent(&buf, azCol[i], &rc);
2083e8d5648eSdan       sessionAppendStr(&buf, " = ?", &rc);
2084e8d5648eSdan       sessionAppendInteger(&buf, i+1, &rc);
2085e8d5648eSdan       zSep = " AND ";
2086d5f0767cSdan     }
2087d5f0767cSdan   }
2088d5f0767cSdan   if( rc==SQLITE_OK ){
2089e8d5648eSdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, ppStmt, 0);
2090d5f0767cSdan   }
2091e8d5648eSdan   sqlite3_free(buf.aBuf);
2092e8d5648eSdan   return rc;
2093d5f0767cSdan }
2094d5f0767cSdan 
209577fc1d5bSdan /*
209677fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
209777fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
209877fc1d5bSdan ** is as prepared by function sessionSelectStmt().
209977fc1d5bSdan **
210077fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
210177fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
210277fc1d5bSdan */
2103e8d5648eSdan static int sessionSelectBind(
210477fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
210577fc1d5bSdan   int nCol,                       /* Number of columns in table */
210677fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
210777fc1d5bSdan   SessionChange *pChange          /* Change structure */
2108e8d5648eSdan ){
2109e8d5648eSdan   int i;
2110e8d5648eSdan   int rc = SQLITE_OK;
2111e5754eecSdan   u8 *a = pChange->aRecord;
2112d5f0767cSdan 
2113e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2114e8d5648eSdan     int eType = *a++;
2115e8d5648eSdan 
2116e8d5648eSdan     switch( eType ){
211780fe2d93Sdan       case 0:
2118e8d5648eSdan       case SQLITE_NULL:
2119e5754eecSdan         assert( abPK[i]==0 );
2120e8d5648eSdan         break;
2121e8d5648eSdan 
2122e8d5648eSdan       case SQLITE_INTEGER: {
2123e8d5648eSdan         if( abPK[i] ){
2124e8d5648eSdan           i64 iVal = sessionGetI64(a);
2125e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
2126e8d5648eSdan         }
2127e8d5648eSdan         a += 8;
2128e8d5648eSdan         break;
2129d5f0767cSdan       }
2130296c7658Sdan 
2131e8d5648eSdan       case SQLITE_FLOAT: {
2132e8d5648eSdan         if( abPK[i] ){
2133e8d5648eSdan           double rVal;
2134e8d5648eSdan           i64 iVal = sessionGetI64(a);
2135e8d5648eSdan           memcpy(&rVal, &iVal, 8);
21364e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
2137d5f0767cSdan         }
2138e8d5648eSdan         a += 8;
2139e8d5648eSdan         break;
2140e8d5648eSdan       }
2141e8d5648eSdan 
2142e8d5648eSdan       case SQLITE_TEXT: {
2143e8d5648eSdan         int n;
2144e8d5648eSdan         a += sessionVarintGet(a, &n);
2145e8d5648eSdan         if( abPK[i] ){
2146e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
2147e8d5648eSdan         }
2148e8d5648eSdan         a += n;
2149e8d5648eSdan         break;
2150e8d5648eSdan       }
2151e8d5648eSdan 
2152e5754eecSdan       default: {
2153e8d5648eSdan         int n;
2154e5754eecSdan         assert( eType==SQLITE_BLOB );
2155e8d5648eSdan         a += sessionVarintGet(a, &n);
2156e8d5648eSdan         if( abPK[i] ){
2157e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
2158e8d5648eSdan         }
2159e8d5648eSdan         a += n;
2160e8d5648eSdan         break;
2161e8d5648eSdan       }
2162e8d5648eSdan     }
2163e8d5648eSdan   }
2164e8d5648eSdan 
2165d5f0767cSdan   return rc;
21664fccf43aSdan }
21674fccf43aSdan 
216877fc1d5bSdan /*
216977fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
217077fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
217177fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
217277fc1d5bSdan ** SQLite error code before returning.
217377fc1d5bSdan */
21745d607a6eSdan static void sessionAppendTableHdr(
2175a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
2176a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
2177a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
2178a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
21795d607a6eSdan ){
21805d607a6eSdan   /* Write a table header */
218173b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
21825d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
21835d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
21844f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
21855d607a6eSdan }
21865d607a6eSdan 
2187a71d2371Sdan /*
2188a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
2189a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
2190a71d2371Sdan ** passed as the first argument.
2191a71d2371Sdan **
2192a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
2193a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
2194a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
2195a71d2371Sdan ** to 0.
2196a71d2371Sdan */
219773b3c055Sdan int sessionGenerateChangeset(
21984fccf43aSdan   sqlite3_session *pSession,      /* Session object */
219973b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
2200ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2201ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
22024fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
22034fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
22044fccf43aSdan ){
2205296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
2206296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
2207296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
2208296c7658Sdan   int rc;                         /* Return code */
22094fccf43aSdan 
2210ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
2211ef7a6304Sdan 
2212296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
2213296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
2214296c7658Sdan   ** this call will be a no-op.  */
2215ef7a6304Sdan   if( xOutput==0 ){
22164fccf43aSdan     *pnChangeset = 0;
22174fccf43aSdan     *ppChangeset = 0;
2218ef7a6304Sdan   }
2219e5754eecSdan 
2220e5754eecSdan   if( pSession->rc ) return pSession->rc;
2221e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
2222e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
2223e5754eecSdan 
2224e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
22254fccf43aSdan 
22264fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
22274fccf43aSdan     if( pTab->nEntry ){
2228d7fb7d24Sdan       const char *zName = pTab->zName;
2229a9605b91Sdan       int nCol;                   /* Number of columns in table */
2230a9605b91Sdan       u8 *abPK;                   /* Primary key array */
2231a9605b91Sdan       const char **azCol = 0;     /* Table columns */
22321f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
22331f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
22341f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
22351f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
22364fccf43aSdan 
2237a9605b91Sdan       /* Check the table schema is still Ok. */
2238a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
2239a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
2240a9605b91Sdan         rc = SQLITE_SCHEMA;
2241a9605b91Sdan       }
2242a9605b91Sdan 
22434fccf43aSdan       /* Write a table header */
224473b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
22454fccf43aSdan 
22464fccf43aSdan       /* Build and compile a statement to execute: */
22474fccf43aSdan       if( rc==SQLITE_OK ){
2248d7fb7d24Sdan         rc = sessionSelectStmt(
2249a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
22504fccf43aSdan       }
22514fccf43aSdan 
22521f34f8ccSdan       nNoop = buf.nBuf;
225312ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
2254e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
2255e8d5648eSdan 
22564fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
2257e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
225880fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
22591f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
2260798693b2Sdan             if( p->op==SQLITE_INSERT ){
22614fccf43aSdan               int iCol;
22624fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
2263b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
2264e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
22651f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
22664fccf43aSdan               }
2267e8d5648eSdan             }else{
226873b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
22694fccf43aSdan             }
2270798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
2271a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
22724fccf43aSdan           }
227312ca0b56Sdan           if( rc==SQLITE_OK ){
22741f34f8ccSdan             rc = sqlite3_reset(pSel);
22754fccf43aSdan           }
2276ef7a6304Sdan 
2277f1a08ad8Sdrh           /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass
2278ef7a6304Sdan           ** its contents to the xOutput() callback. */
2279ef7a6304Sdan           if( xOutput
2280ef7a6304Sdan            && rc==SQLITE_OK
2281ef7a6304Sdan            && buf.nBuf>nNoop
2282f1a08ad8Sdrh            && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE
2283ef7a6304Sdan           ){
2284ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2285ef7a6304Sdan             nNoop = -1;
2286ef7a6304Sdan             buf.nBuf = 0;
2287ef7a6304Sdan           }
2288ef7a6304Sdan 
22894fccf43aSdan         }
2290e8d5648eSdan       }
22914fccf43aSdan 
22921f34f8ccSdan       sqlite3_finalize(pSel);
22931f34f8ccSdan       if( buf.nBuf==nNoop ){
22944fccf43aSdan         buf.nBuf = nRewind;
22954fccf43aSdan       }
2296cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
22974fccf43aSdan     }
22984fccf43aSdan   }
22994fccf43aSdan 
23004fccf43aSdan   if( rc==SQLITE_OK ){
2301ef7a6304Sdan     if( xOutput==0 ){
23024fccf43aSdan       *pnChangeset = buf.nBuf;
23034fccf43aSdan       *ppChangeset = buf.aBuf;
2304ef7a6304Sdan       buf.aBuf = 0;
2305ef7a6304Sdan     }else if( buf.nBuf>0 ){
2306ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2307ef7a6304Sdan     }
23084fccf43aSdan   }
23094c220252Sdan 
2310ef7a6304Sdan   sqlite3_free(buf.aBuf);
2311e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
23124c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
23134fccf43aSdan   return rc;
23144fccf43aSdan }
23154fccf43aSdan 
2316296c7658Sdan /*
231773b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
231873b3c055Sdan ** session object passed as the first argument.
231973b3c055Sdan **
232073b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
232173b3c055Sdan ** using sqlite3_free().
232273b3c055Sdan */
232373b3c055Sdan int sqlite3session_changeset(
232473b3c055Sdan   sqlite3_session *pSession,      /* Session object */
232573b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
232673b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
232773b3c055Sdan ){
2328ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2329ef7a6304Sdan }
2330ef7a6304Sdan 
2331ef7a6304Sdan /*
2332ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2333ef7a6304Sdan */
2334f1a08ad8Sdrh int sqlite3session_changeset_strm(
2335ef7a6304Sdan   sqlite3_session *pSession,
2336ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2337ef7a6304Sdan   void *pOut
2338ef7a6304Sdan ){
2339ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2340ef7a6304Sdan }
2341ef7a6304Sdan 
2342ef7a6304Sdan /*
2343ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2344ef7a6304Sdan */
2345f1a08ad8Sdrh int sqlite3session_patchset_strm(
2346ef7a6304Sdan   sqlite3_session *pSession,
2347ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2348ef7a6304Sdan   void *pOut
2349ef7a6304Sdan ){
2350ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
235173b3c055Sdan }
235273b3c055Sdan 
235373b3c055Sdan /*
235473b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
235573b3c055Sdan ** session object passed as the first argument.
235673b3c055Sdan **
235773b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
235873b3c055Sdan ** using sqlite3_free().
235973b3c055Sdan */
236073b3c055Sdan int sqlite3session_patchset(
236173b3c055Sdan   sqlite3_session *pSession,      /* Session object */
236273b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
236373b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
236473b3c055Sdan ){
2365ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
236673b3c055Sdan }
236773b3c055Sdan 
236873b3c055Sdan /*
2369296c7658Sdan ** Enable or disable the session object passed as the first argument.
2370296c7658Sdan */
23714fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
23724c220252Sdan   int ret;
23734c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2374296c7658Sdan   if( bEnable>=0 ){
2375296c7658Sdan     pSession->bEnable = bEnable;
23764fccf43aSdan   }
23774c220252Sdan   ret = pSession->bEnable;
23784c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
23794c220252Sdan   return ret;
2380296c7658Sdan }
23814fccf43aSdan 
23824fccf43aSdan /*
2383b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2384b4480e94Sdan */
2385b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2386b4480e94Sdan   int ret;
2387b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2388b4480e94Sdan   if( bIndirect>=0 ){
2389b4480e94Sdan     pSession->bIndirect = bIndirect;
2390b4480e94Sdan   }
2391b4480e94Sdan   ret = pSession->bIndirect;
2392b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2393b4480e94Sdan   return ret;
2394b4480e94Sdan }
2395b4480e94Sdan 
2396b4480e94Sdan /*
2397b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2398b69ec348Sdan ** by the session object passed as the only argument.
2399b69ec348Sdan */
2400b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2401b69ec348Sdan   int ret = 0;
2402b69ec348Sdan   SessionTable *pTab;
2403b69ec348Sdan 
2404b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2405b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2406b69ec348Sdan     ret = (pTab->nEntry>0);
2407b69ec348Sdan   }
2408b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2409b69ec348Sdan 
2410ff530326Sdan   return (ret==0);
2411b69ec348Sdan }
2412b69ec348Sdan 
2413b69ec348Sdan /*
2414f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
24154fccf43aSdan */
2416ef7a6304Sdan int sessionChangesetStart(
2417296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2418ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2419ef7a6304Sdan   void *pIn,
2420296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2421296c7658Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
24224fccf43aSdan ){
24234fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
24244fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
24254fccf43aSdan 
2426ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2427ef7a6304Sdan 
2428296c7658Sdan   /* Zero the output variable in case an error occurs. */
2429296c7658Sdan   *pp = 0;
24304fccf43aSdan 
2431296c7658Sdan   /* Allocate and initialize the iterator structure. */
24324fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
24334fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
24344fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
24354fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
24364757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
24374757c658Sdan   pRet->in.nData = nChangeset;
2438ef7a6304Sdan   pRet->in.xInput = xInput;
2439ef7a6304Sdan   pRet->in.pIn = pIn;
2440ef7a6304Sdan   pRet->in.iNext = 0;
2441ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
24424fccf43aSdan 
2443296c7658Sdan   /* Populate the output variable and return success. */
2444296c7658Sdan   *pp = pRet;
24454fccf43aSdan   return SQLITE_OK;
24464fccf43aSdan }
24474fccf43aSdan 
2448296c7658Sdan /*
2449ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2450ef7a6304Sdan */
2451ef7a6304Sdan int sqlite3changeset_start(
2452ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2453ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2454ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2455ef7a6304Sdan ){
2456ef7a6304Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset);
2457ef7a6304Sdan }
2458ef7a6304Sdan 
2459ef7a6304Sdan /*
2460ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2461ef7a6304Sdan */
2462f1a08ad8Sdrh int sqlite3changeset_start_strm(
2463ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2464ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2465ef7a6304Sdan   void *pIn
2466ef7a6304Sdan ){
2467ef7a6304Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0);
2468ef7a6304Sdan }
2469ef7a6304Sdan 
2470ef7a6304Sdan /*
2471ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2472ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2473ef7a6304Sdan ** data is in the buffer.
2474ef7a6304Sdan **
2475ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2476ef7a6304Sdan */
24774757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2478ef7a6304Sdan   int rc = SQLITE_OK;
24794757c658Sdan   if( pIn->xInput ){
24804757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
2481f1a08ad8Sdrh       int nNew = SESSIONS_STRM_CHUNK_SIZE;
24824757c658Sdan 
2483f1a08ad8Sdrh       if( pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){
24844757c658Sdan         int nMove = pIn->buf.nBuf - pIn->iNext;
24854757c658Sdan         memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
24864757c658Sdan         pIn->buf.nBuf -= pIn->iNext;
24874757c658Sdan         pIn->iNext = 0;
24884757c658Sdan       }
24894757c658Sdan 
24904757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
24914757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
24924757c658Sdan         if( nNew==0 ){
24934757c658Sdan           pIn->bEof = 1;
24944757c658Sdan         }else{
24954757c658Sdan           pIn->buf.nBuf += nNew;
24964757c658Sdan         }
24974757c658Sdan       }
24984757c658Sdan 
24994757c658Sdan       pIn->aData = pIn->buf.aBuf;
25004757c658Sdan       pIn->nData = pIn->buf.nBuf;
25014757c658Sdan     }
2502ef7a6304Sdan   }
2503ef7a6304Sdan   return rc;
2504ef7a6304Sdan }
2505ef7a6304Sdan 
2506ef7a6304Sdan /*
2507ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2508ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2509ef7a6304Sdan ** until it points to the byte immediately following that record.
2510ef7a6304Sdan */
2511ef7a6304Sdan static void sessionSkipRecord(
2512ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2513ef7a6304Sdan   int nCol                        /* Number of values in record */
2514ef7a6304Sdan ){
2515ef7a6304Sdan   u8 *aRec = *ppRec;
2516ef7a6304Sdan   int i;
2517ef7a6304Sdan   for(i=0; i<nCol; i++){
2518ef7a6304Sdan     int eType = *aRec++;
2519ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2520ef7a6304Sdan       int nByte;
2521ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2522ef7a6304Sdan       aRec += nByte;
2523ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2524ef7a6304Sdan       aRec += 8;
2525ef7a6304Sdan     }
2526ef7a6304Sdan   }
2527ef7a6304Sdan 
2528ef7a6304Sdan   *ppRec = aRec;
2529ef7a6304Sdan }
2530ef7a6304Sdan 
2531ef7a6304Sdan /*
25324757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
25334757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
25344757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
25354757c658Sdan ** error occurs.
25364757c658Sdan */
25374757c658Sdan static int sessionValueSetStr(
25384757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
25394757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
25404757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
25414757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
25424757c658Sdan ){
254316228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
254416228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
254516228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
254616228167Sdan   ** error. Hence the code to create the copy externally. */
25473cc89d95Sdan   u8 *aCopy = sqlite3_malloc(nData+1);
25484757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
25494757c658Sdan   memcpy(aCopy, aData, nData);
25504757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
25514757c658Sdan   return SQLITE_OK;
25524757c658Sdan }
25534757c658Sdan 
25544757c658Sdan /*
2555296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2556296c7658Sdan ** for details.
2557296c7658Sdan **
2558296c7658Sdan ** When this function is called, *paChange points to the start of the record
2559296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2560296c7658Sdan ** one byte after the end of the same record before this function returns.
2561a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2562a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2563a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2564296c7658Sdan **
2565296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2566296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2567296c7658Sdan ** from the corresponding position in the record. If that value is not
2568296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2569296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2570296c7658Sdan ** set to NULL.
2571296c7658Sdan **
2572296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2573296c7658Sdan ** using sqlite3_free().
2574296c7658Sdan **
2575296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2576296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2577296c7658Sdan */
25784fccf43aSdan static int sessionReadRecord(
2579ef7a6304Sdan   SessionInput *pIn,              /* Input data */
25804fccf43aSdan   int nCol,                       /* Number of values in record */
258173b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
25824fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
25834fccf43aSdan ){
2584296c7658Sdan   int i;                          /* Used to iterate through columns */
2585ef7a6304Sdan   int rc = SQLITE_OK;
25864fccf43aSdan 
2587ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2588ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
258973b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2590ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2591ef7a6304Sdan     if( rc==SQLITE_OK ){
25924757c658Sdan       eType = pIn->aData[pIn->iNext++];
2593ef7a6304Sdan     }
2594ef7a6304Sdan 
2595e8fa8c96Sdan     assert( apOut[i]==0 );
25964fccf43aSdan     if( eType ){
25974fccf43aSdan       apOut[i] = sqlite3ValueNew(0);
2598ef7a6304Sdan       if( !apOut[i] ) rc = SQLITE_NOMEM;
2599ef7a6304Sdan     }
26004fccf43aSdan 
2601ef7a6304Sdan     if( rc==SQLITE_OK ){
26024757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
26034fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
26044fccf43aSdan         int nByte;
2605ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2606ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2607e8fa8c96Sdan         if( rc==SQLITE_OK ){
26086734007dSdan           u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
26094757c658Sdan           rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
261091ddd559Sdan         }
2611ef7a6304Sdan         pIn->iNext += nByte;
26124fccf43aSdan       }
26134fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2614ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
26154fccf43aSdan         if( eType==SQLITE_INTEGER ){
26164fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
26174fccf43aSdan         }else{
26184fccf43aSdan           double d;
26194e895da1Sdan           memcpy(&d, &v, 8);
26204fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
26214fccf43aSdan         }
2622ef7a6304Sdan         pIn->iNext += 8;
262391ddd559Sdan       }
26244fccf43aSdan     }
26254fccf43aSdan   }
26264fccf43aSdan 
2627ef7a6304Sdan   return rc;
2628ef7a6304Sdan }
2629ef7a6304Sdan 
2630ef7a6304Sdan /*
2631ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2632ef7a6304Sdan ** Specifically, to the following:
2633ef7a6304Sdan **
2634ef7a6304Sdan **   + number of columns in table (varint)
2635ef7a6304Sdan **   + array of PK flags (1 byte per column),
2636ef7a6304Sdan **   + table name (nul terminated).
2637ef7a6304Sdan **
2638ef7a6304Sdan ** This function ensures that all of the above is present in the input
2639ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2640ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2641ef7a6304Sdan ** The input pointer is not moved.
2642ef7a6304Sdan */
2643ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2644ef7a6304Sdan   int rc = SQLITE_OK;
2645ef7a6304Sdan   int nCol = 0;
26464757c658Sdan   int nRead = 0;
2647ef7a6304Sdan 
2648ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2649ef7a6304Sdan   if( rc==SQLITE_OK ){
26504757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
26514757c658Sdan     rc = sessionInputBuffer(pIn, nRead+nCol+100);
26524757c658Sdan     nRead += nCol;
2653ef7a6304Sdan   }
26544757c658Sdan 
2655ef7a6304Sdan   while( rc==SQLITE_OK ){
26564757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
26574757c658Sdan       nRead++;
2658ef7a6304Sdan     }
2659e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
26604757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
26614757c658Sdan   }
2662e8fa8c96Sdan   *pnByte = nRead+1;
2663ef7a6304Sdan   return rc;
2664ef7a6304Sdan }
2665ef7a6304Sdan 
2666ef7a6304Sdan /*
2667fa122adaSdan ** The input pointer currently points to the first byte of the first field
2668fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
266916228167Sdan ** record is buffered. It does not move the input pointer.
267016228167Sdan **
267116228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
267216228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
267316228167Sdan ** final value of *pnByte is undefined in this case.
2674fa122adaSdan */
2675fa122adaSdan static int sessionChangesetBufferRecord(
267616228167Sdan   SessionInput *pIn,              /* Input data */
267716228167Sdan   int nCol,                       /* Number of columns in record */
267816228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2679fa122adaSdan ){
2680fa122adaSdan   int rc = SQLITE_OK;
2681fa122adaSdan   int nByte = 0;
2682fa122adaSdan   int i;
2683fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2684fa122adaSdan     int eType;
2685fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2686fa122adaSdan     if( rc==SQLITE_OK ){
2687fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2688fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2689fa122adaSdan         int n;
2690fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2691fa122adaSdan         nByte += n;
2692fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2693fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2694fa122adaSdan         nByte += 8;
2695fa122adaSdan       }
2696fa122adaSdan     }
2697fa122adaSdan   }
2698fa122adaSdan   *pnByte = nByte;
2699fa122adaSdan   return rc;
2700fa122adaSdan }
2701fa122adaSdan 
2702fa122adaSdan /*
2703ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2704ef7a6304Sdan ** Specifically, to the following:
2705ef7a6304Sdan **
2706ef7a6304Sdan **   + number of columns in table (varint)
2707ef7a6304Sdan **   + array of PK flags (1 byte per column),
2708ef7a6304Sdan **   + table name (nul terminated).
270916228167Sdan **
271016228167Sdan ** This function decodes the table-header and populates the p->nCol,
271116228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
271216228167Sdan ** also allocated or resized according to the new value of p->nCol. The
271316228167Sdan ** input pointer is left pointing to the byte following the table header.
271416228167Sdan **
271516228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
271616228167Sdan ** is returned and the final values of the various fields enumerated above
271716228167Sdan ** are undefined.
2718ef7a6304Sdan */
2719ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2720ef7a6304Sdan   int rc;
2721ef7a6304Sdan   int nCopy;
2722ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2723ef7a6304Sdan 
2724ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2725ef7a6304Sdan   if( rc==SQLITE_OK ){
2726ef7a6304Sdan     int nByte;
2727ef7a6304Sdan     int nVarint;
27284757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2729ef7a6304Sdan     nCopy -= nVarint;
2730ef7a6304Sdan     p->in.iNext += nVarint;
2731ef7a6304Sdan     nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2732ef7a6304Sdan     p->tblhdr.nBuf = 0;
2733ef7a6304Sdan     sessionBufferGrow(&p->tblhdr, nByte, &rc);
2734ef7a6304Sdan   }
2735ef7a6304Sdan 
2736ef7a6304Sdan   if( rc==SQLITE_OK ){
2737ef7a6304Sdan     int iPK = sizeof(sqlite3_value*)*p->nCol*2;
2738ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
27394757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2740ef7a6304Sdan     p->in.iNext += nCopy;
2741ef7a6304Sdan   }
2742ef7a6304Sdan 
2743ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2744ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2745ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2746ef7a6304Sdan   return (p->rc = rc);
27474fccf43aSdan }
27484fccf43aSdan 
274977fc1d5bSdan /*
275077fc1d5bSdan ** Advance the changeset iterator to the next change.
275177fc1d5bSdan **
275277fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
275377fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
275477fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
275577fc1d5bSdan **
275677fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
275777fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
275877fc1d5bSdan ** the record to *pnRec.
275977fc1d5bSdan **
276077fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
276177fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
276277fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
276377fc1d5bSdan ** changes in the changeset.
276477fc1d5bSdan */
27655d607a6eSdan static int sessionChangesetNext(
276677fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
276777fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
276877fc1d5bSdan   int *pnRec                      /* If non-NULL, store size of record here */
27695d607a6eSdan ){
27704fccf43aSdan   int i;
2771ef7a6304Sdan   u8 op;
27724fccf43aSdan 
27735d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
27745d607a6eSdan 
2775296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
27764fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
27774fccf43aSdan 
27785d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
27794fccf43aSdan   if( p->apValue ){
27804fccf43aSdan     for(i=0; i<p->nCol*2; i++){
27814fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
27824fccf43aSdan     }
27834fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
27844fccf43aSdan   }
27854fccf43aSdan 
2786ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
2787ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
2788ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
2789ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
2790ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
2791ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
2792ef7a6304Sdan 
27934fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
27944757c658Sdan   if( p->in.iNext>=p->in.nData ){
27954fccf43aSdan     return SQLITE_DONE;
27964fccf43aSdan   }
27974fccf43aSdan 
27984757c658Sdan   op = p->in.aData[p->in.iNext++];
2799ef7a6304Sdan   if( op=='T' || op=='P' ){
2800ef7a6304Sdan     p->bPatchset = (op=='P');
2801ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
2802ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
28034757c658Sdan     op = p->in.aData[p->in.iNext++];
28045d607a6eSdan   }
28055d607a6eSdan 
2806ef7a6304Sdan   p->op = op;
28074757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
28084fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
28094757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
28104fccf43aSdan   }
28114fccf43aSdan 
2812cbf6d2d2Sdan   if( paRec ){
2813cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
2814cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
2815cbf6d2d2Sdan       nVal = p->nCol * 2;
2816cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
2817cbf6d2d2Sdan       nVal = 0;
2818cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
2819cbf6d2d2Sdan     }else{
2820cbf6d2d2Sdan       nVal = p->nCol;
2821cbf6d2d2Sdan     }
2822cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
2823cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
2824cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
2825cbf6d2d2Sdan     p->in.iNext += *pnRec;
2826cbf6d2d2Sdan   }else{
28275d607a6eSdan 
28284fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
282973b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
283073b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
2831cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue);
28324fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
28334fccf43aSdan     }
28344fccf43aSdan 
28354fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
28364fccf43aSdan     if( p->op!=SQLITE_DELETE ){
2837cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]);
28384fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
28394fccf43aSdan     }
28404fccf43aSdan 
2841cbf6d2d2Sdan     if( p->bPatchset && p->op==SQLITE_UPDATE ){
284273b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
284373b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
284473b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
284573b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
284673b3c055Sdan       int i;
284773b3c055Sdan       for(i=0; i<p->nCol; i++){
284873b3c055Sdan         assert( p->apValue[i]==0 );
284973b3c055Sdan         assert( p->abPK[i]==0 || p->apValue[i+p->nCol] );
285073b3c055Sdan         if( p->abPK[i] ){
285173b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
285273b3c055Sdan           p->apValue[i+p->nCol] = 0;
285373b3c055Sdan         }
285473b3c055Sdan       }
285573b3c055Sdan     }
2856cbf6d2d2Sdan   }
2857ef7a6304Sdan 
28584fccf43aSdan   return SQLITE_ROW;
28594fccf43aSdan }
28604fccf43aSdan 
28614fccf43aSdan /*
28625d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
28635d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
28645d607a6eSdan ** or SQLITE_CORRUPT.
28655d607a6eSdan **
28665d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
28675d607a6eSdan ** callback by changeset_apply().
28685d607a6eSdan */
28695d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
28705d607a6eSdan   return sessionChangesetNext(p, 0, 0);
28715d607a6eSdan }
28725d607a6eSdan 
28735d607a6eSdan /*
2874244593c8Sdan ** The following function extracts information on the current change
287577fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
28764fccf43aSdan ** has returned SQLITE_ROW.
28774fccf43aSdan */
28784fccf43aSdan int sqlite3changeset_op(
2879296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
28804fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
28814fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
2882b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
2883b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
28844fccf43aSdan ){
28854fccf43aSdan   *pOp = pIter->op;
28864fccf43aSdan   *pnCol = pIter->nCol;
28874fccf43aSdan   *pzTab = pIter->zTab;
2888b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
28894fccf43aSdan   return SQLITE_OK;
28904fccf43aSdan }
28914fccf43aSdan 
289277fc1d5bSdan /*
289377fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
289477fc1d5bSdan ** the database table affected by the change that pIter currently points
289577fc1d5bSdan ** to. This function may only be called after changeset_next() returns
289677fc1d5bSdan ** SQLITE_ROW.
289777fc1d5bSdan */
2898244593c8Sdan int sqlite3changeset_pk(
2899244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
2900244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
2901244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
2902244593c8Sdan ){
2903244593c8Sdan   *pabPK = pIter->abPK;
2904244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
2905244593c8Sdan   return SQLITE_OK;
2906244593c8Sdan }
2907244593c8Sdan 
2908296c7658Sdan /*
2909296c7658Sdan ** This function may only be called while the iterator is pointing to an
2910296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
2911296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
2912296c7658Sdan **
2913296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
2914296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
2915296c7658Sdan ** included in the record (because the change is an UPDATE and the field
2916296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
2917296c7658Sdan **
2918296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
2919296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
2920296c7658Sdan */
29214fccf43aSdan int sqlite3changeset_old(
2922296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2923296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
29244fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
29254fccf43aSdan ){
2926d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
2927d5f0767cSdan     return SQLITE_MISUSE;
2928d5f0767cSdan   }
29294fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
29304fccf43aSdan     return SQLITE_RANGE;
29314fccf43aSdan   }
29324fccf43aSdan   *ppValue = pIter->apValue[iVal];
29334fccf43aSdan   return SQLITE_OK;
29344fccf43aSdan }
29354fccf43aSdan 
2936296c7658Sdan /*
2937296c7658Sdan ** This function may only be called while the iterator is pointing to an
2938296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
2939296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
2940296c7658Sdan **
2941296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
2942296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
2943296c7658Sdan ** included in the record (because the change is an UPDATE and the field
2944296c7658Sdan ** was not modified), set *ppValue to NULL.
2945296c7658Sdan **
2946296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
2947296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
2948296c7658Sdan */
29494fccf43aSdan int sqlite3changeset_new(
2950296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2951296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
29524fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
29534fccf43aSdan ){
2954d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
2955d5f0767cSdan     return SQLITE_MISUSE;
2956d5f0767cSdan   }
29574fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
29584fccf43aSdan     return SQLITE_RANGE;
29594fccf43aSdan   }
29604fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
29614fccf43aSdan   return SQLITE_OK;
29624fccf43aSdan }
29634fccf43aSdan 
2964296c7658Sdan /*
29657aa469cdSdan ** The following two macros are used internally. They are similar to the
29667aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
29677aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
29687aa469cdSdan */
29697aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
29707aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
29717aa469cdSdan 
29727aa469cdSdan /*
2973296c7658Sdan ** This function may only be called with a changeset iterator that has been
2974296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
2975296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
2976296c7658Sdan **
2977296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
2978296c7658Sdan ** containing the iVal'th value of the conflicting record.
2979296c7658Sdan **
2980296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
2981296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
2982296c7658Sdan */
2983d5f0767cSdan int sqlite3changeset_conflict(
2984296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2985296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
2986d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
2987d5f0767cSdan ){
2988d5f0767cSdan   if( !pIter->pConflict ){
2989d5f0767cSdan     return SQLITE_MISUSE;
2990d5f0767cSdan   }
2991d5f0767cSdan   if( iVal<0 || iVal>=sqlite3_column_count(pIter->pConflict) ){
2992d5f0767cSdan     return SQLITE_RANGE;
2993d5f0767cSdan   }
2994d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
2995d5f0767cSdan   return SQLITE_OK;
2996d5f0767cSdan }
2997d5f0767cSdan 
29984fccf43aSdan /*
2999cb3e4b79Sdan ** This function may only be called with an iterator passed to an
3000cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
3001cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
3002cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
3003cb3e4b79Sdan **
3004cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
3005cb3e4b79Sdan */
3006cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
3007cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3008cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
3009cb3e4b79Sdan ){
3010cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
3011cb3e4b79Sdan     return SQLITE_MISUSE;
3012cb3e4b79Sdan   }
3013cb3e4b79Sdan   *pnOut = pIter->nCol;
3014cb3e4b79Sdan   return SQLITE_OK;
3015cb3e4b79Sdan }
3016cb3e4b79Sdan 
3017cb3e4b79Sdan 
3018cb3e4b79Sdan /*
30194fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
30204fccf43aSdan **
30214fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
30224fccf43aSdan ** callback by changeset_apply().
30234fccf43aSdan */
30244fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
3025cbf6d2d2Sdan   int rc = SQLITE_OK;
3026cbf6d2d2Sdan   if( p ){
3027296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
3028cbf6d2d2Sdan     rc = p->rc;
302912ca0b56Sdan     if( p->apValue ){
30304fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
303112ca0b56Sdan     }
3032ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
30334757c658Sdan     sqlite3_free(p->in.buf.aBuf);
30344fccf43aSdan     sqlite3_free(p);
3035cbf6d2d2Sdan   }
30364fccf43aSdan   return rc;
30374fccf43aSdan }
30384fccf43aSdan 
3039fa122adaSdan static int sessionChangesetInvert(
3040fa122adaSdan   SessionInput *pInput,           /* Input changeset */
3041fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3042fa122adaSdan   void *pOut,
304391ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
304491ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
304591ddd559Sdan ){
3046cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
3047fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
3048cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
3049cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
3050cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
3051ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
305291ddd559Sdan 
3053fa122adaSdan   /* Initialize the output buffer */
3054fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
3055fa122adaSdan 
305691ddd559Sdan   /* Zero the output variables in case an error occurs. */
3057fa122adaSdan   if( ppInverted ){
305891ddd559Sdan     *ppInverted = 0;
305991ddd559Sdan     *pnInverted = 0;
3060fa122adaSdan   }
306191ddd559Sdan 
3062fa122adaSdan   while( 1 ){
3063ef7a6304Sdan     u8 eType;
3064fa122adaSdan 
3065fa122adaSdan     /* Test for EOF. */
3066fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
3067fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
3068fa122adaSdan     eType = pInput->aData[pInput->iNext];
3069fa122adaSdan 
307091ddd559Sdan     switch( eType ){
307191ddd559Sdan       case 'T': {
3072244593c8Sdan         /* A 'table' record consists of:
3073244593c8Sdan         **
3074244593c8Sdan         **   * A constant 'T' character,
3075244593c8Sdan         **   * Number of columns in said table (a varint),
3076ef7a6304Sdan         **   * An array of nCol bytes (sPK),
3077244593c8Sdan         **   * A nul-terminated table name.
3078244593c8Sdan         */
3079ef7a6304Sdan         int nByte;
3080fa122adaSdan         int nVar;
3081fa122adaSdan         pInput->iNext++;
3082fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
3083ef7a6304Sdan           goto finished_invert;
3084ef7a6304Sdan         }
3085fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
3086ef7a6304Sdan         sPK.nBuf = 0;
3087fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
3088fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3089fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3090ef7a6304Sdan         if( rc ) goto finished_invert;
3091fa122adaSdan 
3092fa122adaSdan         pInput->iNext += nByte;
3093cfec7eeeSdan         sqlite3_free(apVal);
3094cfec7eeeSdan         apVal = 0;
3095ef7a6304Sdan         abPK = sPK.aBuf;
309691ddd559Sdan         break;
309791ddd559Sdan       }
309891ddd559Sdan 
309991ddd559Sdan       case SQLITE_INSERT:
310091ddd559Sdan       case SQLITE_DELETE: {
310191ddd559Sdan         int nByte;
3102fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
3103fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
3104fa122adaSdan         pInput->iNext += 2;
3105fa122adaSdan         assert( rc==SQLITE_OK );
3106fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
3107fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
3108fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
3109fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3110fa122adaSdan         pInput->iNext += nByte;
3111fa122adaSdan         if( rc ) goto finished_invert;
311291ddd559Sdan         break;
311391ddd559Sdan       }
311491ddd559Sdan 
311591ddd559Sdan       case SQLITE_UPDATE: {
3116cfec7eeeSdan         int iCol;
311791ddd559Sdan 
3118cfec7eeeSdan         if( 0==apVal ){
3119cfec7eeeSdan           apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2);
3120cfec7eeeSdan           if( 0==apVal ){
3121cfec7eeeSdan             rc = SQLITE_NOMEM;
3122cfec7eeeSdan             goto finished_invert;
3123cfec7eeeSdan           }
3124cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3125cfec7eeeSdan         }
312691ddd559Sdan 
3127cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
3128fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3129fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
313091ddd559Sdan 
3131ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
3132fa122adaSdan         pInput->iNext += 2;
3133fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
3134ef7a6304Sdan         if( rc==SQLITE_OK ){
3135fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
3136ef7a6304Sdan         }
3137ef7a6304Sdan 
3138cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
3139cfec7eeeSdan         ** original old.* record, and the other values from the original
3140cfec7eeeSdan         ** new.* record. */
3141e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3142cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
3143fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3144cfec7eeeSdan         }
3145cfec7eeeSdan 
3146cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
3147cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
3148cfec7eeeSdan         ** are set to "undefined". */
3149e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3150cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
3151fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3152cfec7eeeSdan         }
3153cfec7eeeSdan 
3154cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
3155cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
3156cfec7eeeSdan         }
3157cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3158cfec7eeeSdan         if( rc!=SQLITE_OK ){
3159cfec7eeeSdan           goto finished_invert;
3160cfec7eeeSdan         }
3161cfec7eeeSdan 
316291ddd559Sdan         break;
316391ddd559Sdan       }
316491ddd559Sdan 
316591ddd559Sdan       default:
31664757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
3167cfec7eeeSdan         goto finished_invert;
316891ddd559Sdan     }
3169fa122adaSdan 
3170fa122adaSdan     assert( rc==SQLITE_OK );
3171f1a08ad8Sdrh     if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
3172fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3173fa122adaSdan       sOut.nBuf = 0;
3174fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
3175fa122adaSdan     }
317691ddd559Sdan   }
317791ddd559Sdan 
3178cfec7eeeSdan   assert( rc==SQLITE_OK );
3179fa122adaSdan   if( pnInverted ){
3180fa122adaSdan     *pnInverted = sOut.nBuf;
3181fa122adaSdan     *ppInverted = sOut.aBuf;
3182fa122adaSdan     sOut.aBuf = 0;
3183fa122adaSdan   }else if( sOut.nBuf>0 ){
3184fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3185fa122adaSdan   }
3186cfec7eeeSdan 
3187cfec7eeeSdan  finished_invert:
3188fa122adaSdan   sqlite3_free(sOut.aBuf);
3189cfec7eeeSdan   sqlite3_free(apVal);
3190ef7a6304Sdan   sqlite3_free(sPK.aBuf);
3191cfec7eeeSdan   return rc;
319291ddd559Sdan }
319391ddd559Sdan 
3194fa122adaSdan 
3195fa122adaSdan /*
3196fa122adaSdan ** Invert a changeset object.
3197fa122adaSdan */
3198fa122adaSdan int sqlite3changeset_invert(
3199fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
3200fa122adaSdan   const void *pChangeset,         /* Input changeset */
3201fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
3202fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
3203fa122adaSdan ){
3204fa122adaSdan   SessionInput sInput;
3205fa122adaSdan 
3206fa122adaSdan   /* Set up the input stream */
3207fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3208fa122adaSdan   sInput.nData = nChangeset;
3209fa122adaSdan   sInput.aData = (u8*)pChangeset;
3210fa122adaSdan 
3211fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
3212fa122adaSdan }
3213fa122adaSdan 
3214fa122adaSdan /*
3215fa122adaSdan ** Streaming version of sqlite3changeset_invert().
3216fa122adaSdan */
3217f1a08ad8Sdrh int sqlite3changeset_invert_strm(
3218fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
3219fa122adaSdan   void *pIn,
3220fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3221fa122adaSdan   void *pOut
3222fa122adaSdan ){
3223fa122adaSdan   SessionInput sInput;
3224fa122adaSdan   int rc;
3225fa122adaSdan 
3226fa122adaSdan   /* Set up the input stream */
3227fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3228fa122adaSdan   sInput.xInput = xInput;
3229fa122adaSdan   sInput.pIn = pIn;
3230fa122adaSdan 
3231fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
3232fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
3233fa122adaSdan   return rc;
3234fa122adaSdan }
3235fa122adaSdan 
32360c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
32370c698471Sdan struct SessionApplyCtx {
32380c698471Sdan   sqlite3 *db;
32390c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
3240cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
32410c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
32420c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
32430c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
32440c698471Sdan   const char **azCol;             /* Array of column names */
32450c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
32460c698471Sdan };
32470c698471Sdan 
3248d5f0767cSdan /*
3249d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
3250d5f0767cSdan ** structure like this:
3251d5f0767cSdan **
3252d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3253d5f0767cSdan **
3254d5f0767cSdan ** The DELETE statement looks like this:
3255d5f0767cSdan **
3256db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
3257d5f0767cSdan **
3258d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
3259d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
3260d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
3261296c7658Sdan **
3262296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
3263296c7658Sdan ** pointing to the prepared version of the SQL statement.
3264d5f0767cSdan */
3265d5f0767cSdan static int sessionDeleteRow(
3266d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3267d5f0767cSdan   const char *zTab,               /* Table name */
32680c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3269d5f0767cSdan ){
3270296c7658Sdan   int i;
3271296c7658Sdan   const char *zSep = "";
3272d5f0767cSdan   int rc = SQLITE_OK;
3273d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
32747cf7df7dSdan   int nPk = 0;
3275d5f0767cSdan 
3276d5f0767cSdan   sessionAppendStr(&buf, "DELETE FROM ", &rc);
3277d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3278296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
3279296c7658Sdan 
3280296c7658Sdan   for(i=0; i<p->nCol; i++){
3281296c7658Sdan     if( p->abPK[i] ){
32827cf7df7dSdan       nPk++;
3283296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
3284296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3285296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
3286296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
3287296c7658Sdan       zSep = " AND ";
3288296c7658Sdan     }
3289296c7658Sdan   }
3290296c7658Sdan 
32917cf7df7dSdan   if( nPk<p->nCol ){
3292296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
3293296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
3294296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
3295296c7658Sdan 
3296296c7658Sdan     zSep = "";
3297296c7658Sdan     for(i=0; i<p->nCol; i++){
3298296c7658Sdan       if( !p->abPK[i] ){
3299296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
3300296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
3301296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
3302296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
3303296c7658Sdan         zSep = "AND ";
3304296c7658Sdan       }
3305296c7658Sdan     }
3306296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
33077cf7df7dSdan   }
3308d5f0767cSdan 
3309d5f0767cSdan   if( rc==SQLITE_OK ){
33100c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
3311d5f0767cSdan   }
3312d5f0767cSdan   sqlite3_free(buf.aBuf);
3313d5f0767cSdan 
3314d5f0767cSdan   return rc;
3315d5f0767cSdan }
3316d5f0767cSdan 
3317d5f0767cSdan /*
3318d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
3319d5f0767cSdan ** Assuming a table structure like this:
3320d5f0767cSdan **
3321d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3322d5f0767cSdan **
3323d5f0767cSdan ** The UPDATE statement looks like this:
3324d5f0767cSdan **
3325d5f0767cSdan **     UPDATE x SET
3326d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3327964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3328964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3329964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3330d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3331964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3332d5f0767cSdan **     )
3333d5f0767cSdan **
3334d5f0767cSdan ** For each column in the table, there are three variables to bind:
3335d5f0767cSdan **
3336d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3337d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3338d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3339d5f0767cSdan **
3340d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3341d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3342d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3343d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3344d5f0767cSdan **
3345296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3346296c7658Sdan ** pointing to the prepared version of the SQL statement.
3347d5f0767cSdan */
3348d5f0767cSdan static int sessionUpdateRow(
3349d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3350d5f0767cSdan   const char *zTab,               /* Table name */
33510c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3352d5f0767cSdan ){
3353d5f0767cSdan   int rc = SQLITE_OK;
3354d5f0767cSdan   int i;
3355d5f0767cSdan   const char *zSep = "";
3356d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3357d5f0767cSdan 
3358d5f0767cSdan   /* Append "UPDATE tbl SET " */
3359d5f0767cSdan   sessionAppendStr(&buf, "UPDATE ", &rc);
3360d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3361d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3362d5f0767cSdan 
3363d5f0767cSdan   /* Append the assignments */
33640c698471Sdan   for(i=0; i<p->nCol; i++){
3365d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
33660c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3367d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3368d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3369d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3370d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3371d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
33720c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3373d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3374d5f0767cSdan     zSep = ", ";
3375d5f0767cSdan   }
3376d5f0767cSdan 
3377d5f0767cSdan   /* Append the PK part of the WHERE clause */
3378d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
33790c698471Sdan   for(i=0; i<p->nCol; i++){
33800c698471Sdan     if( p->abPK[i] ){
33810c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3382d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3383d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3384d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3385d5f0767cSdan     }
3386d5f0767cSdan   }
3387d5f0767cSdan 
3388d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3389d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
33900c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3391d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
33920c698471Sdan   for(i=0; i<p->nCol; i++){
33930c698471Sdan     if( !p->abPK[i] ){
3394d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3395d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3396d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
33970c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3398d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3399d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3400d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3401d5f0767cSdan     }
3402d5f0767cSdan   }
3403d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3404d5f0767cSdan 
3405d5f0767cSdan   if( rc==SQLITE_OK ){
34060c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3407d5f0767cSdan   }
3408d5f0767cSdan   sqlite3_free(buf.aBuf);
3409d5f0767cSdan 
3410d5f0767cSdan   return rc;
3411d5f0767cSdan }
3412d5f0767cSdan 
3413296c7658Sdan /*
3414296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3415296c7658Sdan ** key. Assuming the following table structure:
3416296c7658Sdan **
3417296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3418296c7658Sdan **
3419296c7658Sdan ** The SELECT statement looks like this:
3420296c7658Sdan **
3421296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3422296c7658Sdan **
3423296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3424296c7658Sdan ** pointing to the prepared version of the SQL statement.
3425296c7658Sdan */
3426d5f0767cSdan static int sessionSelectRow(
3427d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3428d5f0767cSdan   const char *zTab,               /* Table name */
34290c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3430d5f0767cSdan ){
3431d7fb7d24Sdan   return sessionSelectStmt(
3432d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3433d5f0767cSdan }
3434d5f0767cSdan 
3435296c7658Sdan /*
3436296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3437296c7658Sdan ** For example:
3438296c7658Sdan **
3439296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3440296c7658Sdan **
3441296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3442296c7658Sdan ** pointing to the prepared version of the SQL statement.
3443296c7658Sdan */
34440c698471Sdan static int sessionInsertRow(
34450c698471Sdan   sqlite3 *db,                    /* Database handle */
34460c698471Sdan   const char *zTab,               /* Table name */
34470c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
34480c698471Sdan ){
34490c698471Sdan   int rc = SQLITE_OK;
34500c698471Sdan   int i;
34510c698471Sdan   SessionBuffer buf = {0, 0, 0};
34520c698471Sdan 
34530c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
34540c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
34550c698471Sdan   sessionAppendStr(&buf, " VALUES(?", &rc);
34560c698471Sdan   for(i=1; i<p->nCol; i++){
34570c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
34580c698471Sdan   }
34590c698471Sdan   sessionAppendStr(&buf, ")", &rc);
34600c698471Sdan 
34610c698471Sdan   if( rc==SQLITE_OK ){
34620c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
34630c698471Sdan   }
34640c698471Sdan   sqlite3_free(buf.aBuf);
34650c698471Sdan   return rc;
34660c698471Sdan }
34670c698471Sdan 
3468296c7658Sdan /*
34697aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
34707aa469cdSdan ** See comments in the body of this function for details.
34717aa469cdSdan */
34727aa469cdSdan static int sessionBindValue(
34737aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
34747aa469cdSdan   int i,                          /* Parameter number to bind to */
34757aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
34767aa469cdSdan ){
34775671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3478082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3479082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3480082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3481082c96dfSdan   ** set to SQLITE_NULL.  */
34825671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
34837aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
34847aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3485082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
34867aa469cdSdan     return SQLITE_NOMEM;
34877aa469cdSdan   }
34887aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
34897aa469cdSdan }
34907aa469cdSdan 
34917aa469cdSdan /*
3492db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3493db04571cSdan ** transfers new.* values from the current iterator entry to statement
3494db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3495db04571cSdan **
3496db04571cSdan ** New.* value $i 0 from the iterator is bound to variable ($i+1) of
3497db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3498db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3499db04571cSdan ** to an array nCol elements in size. In this case only those values for
3500db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3501db04571cSdan ** statement.
3502db04571cSdan **
3503db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3504db04571cSdan */
35057aa469cdSdan static int sessionBindRow(
3506db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
35077aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3508db04571cSdan   int nCol,                       /* Number of columns */
3509db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3510db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3511db04571cSdan ){
3512db04571cSdan   int i;
3513db04571cSdan   int rc = SQLITE_OK;
35147aa469cdSdan 
35157aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
35167aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
35177aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
35187aa469cdSdan   ** in the code below. */
35197aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
35207aa469cdSdan 
3521db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3522db04571cSdan     if( !abPK || abPK[i] ){
3523db04571cSdan       sqlite3_value *pVal;
35247aa469cdSdan       (void)xValue(pIter, i, &pVal);
35257aa469cdSdan       rc = sessionBindValue(pStmt, i+1, pVal);
3526db04571cSdan     }
3527db04571cSdan   }
3528db04571cSdan   return rc;
3529db04571cSdan }
3530db04571cSdan 
3531db04571cSdan /*
3532296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3533296c7658Sdan ** This function binds the primary key values from the change that changeset
3534296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3535296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3536296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3537296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
35387aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
35397aa469cdSdan **
35407aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
35417aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
35427aa469cdSdan ** not require a reset().
3543296c7658Sdan **
3544296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3545db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3546db04571cSdan ** UPDATE, bind values from the old.* record.
3547296c7658Sdan */
35480c698471Sdan static int sessionSeekToRow(
354937f133ecSdan   sqlite3 *db,                    /* Database handle */
355037f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
355137f133ecSdan   u8 *abPK,                       /* Primary key flags array */
35520c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
355337f133ecSdan ){
35547aa469cdSdan   int rc;                         /* Return code */
3555296c7658Sdan   int nCol;                       /* Number of columns in table */
3556296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3557296c7658Sdan   const char *zDummy;             /* Unused */
355837f133ecSdan 
3559b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
35607aa469cdSdan   rc = sessionBindRow(pIter,
3561db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3562db04571cSdan       nCol, abPK, pSelect
3563db04571cSdan   );
35640c698471Sdan 
35650c698471Sdan   if( rc==SQLITE_OK ){
35660c698471Sdan     rc = sqlite3_step(pSelect);
35670c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
35680c698471Sdan   }
35690c698471Sdan 
35700c698471Sdan   return rc;
35710c698471Sdan }
35720c698471Sdan 
3573296c7658Sdan /*
3574296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3575296c7658Sdan ** currently points to.
3576296c7658Sdan **
3577296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3578296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3579296c7658Sdan ** depends solely on eType, as follows:
3580296c7658Sdan **
3581296c7658Sdan **    eType value                 Value passed to xConflict
3582296c7658Sdan **    -------------------------------------------------
3583296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3584296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3585296c7658Sdan **
3586296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3587296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3588296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3589296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3590296c7658Sdan ** handler invoked is as follows:
3591296c7658Sdan **
3592296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3593296c7658Sdan **    ----------------------------------------------------------------
3594296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3595296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3596296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3597296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3598296c7658Sdan **
3599296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3600296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3601296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3602296c7658Sdan **
3603296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3604296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3605296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3606296c7658Sdan ** this function returns SQLITE_OK.
3607296c7658Sdan */
36080c698471Sdan static int sessionConflictHandler(
3609296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3610296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
36110c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
36120c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3613296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3614296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
36150c698471Sdan ){
3616296c7658Sdan   int res;                        /* Value returned by conflict handler */
36170c698471Sdan   int rc;
36180c698471Sdan   int nCol;
36190c698471Sdan   int op;
36200c698471Sdan   const char *zDummy;
36210c698471Sdan 
3622b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
36230c698471Sdan 
36240c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
36250c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
36260c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
362737f133ecSdan 
362837f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
36290c698471Sdan   if( pbReplace ){
36300c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
36310c698471Sdan   }else{
3632db04571cSdan     rc = SQLITE_OK;
36330c698471Sdan   }
36340c698471Sdan 
36350c698471Sdan   if( rc==SQLITE_ROW ){
36360c698471Sdan     /* There exists another row with the new.* primary key. */
36370c698471Sdan     pIter->pConflict = p->pSelect;
36380c698471Sdan     res = xConflict(pCtx, eType, pIter);
36390c698471Sdan     pIter->pConflict = 0;
36400c698471Sdan     rc = sqlite3_reset(p->pSelect);
3641db04571cSdan   }else if( rc==SQLITE_OK ){
36420c698471Sdan     /* No other row with the new.* primary key. */
36430c698471Sdan     res = xConflict(pCtx, eType+1, pIter);
36440c698471Sdan     if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
364537f133ecSdan   }
364637f133ecSdan 
364737f133ecSdan   if( rc==SQLITE_OK ){
36480c698471Sdan     switch( res ){
36490c698471Sdan       case SQLITE_CHANGESET_REPLACE:
3650f51e5f6cSdan         assert( pbReplace );
3651f51e5f6cSdan         *pbReplace = 1;
36520c698471Sdan         break;
36530c698471Sdan 
36540c698471Sdan       case SQLITE_CHANGESET_OMIT:
36550c698471Sdan         break;
36560c698471Sdan 
36570c698471Sdan       case SQLITE_CHANGESET_ABORT:
36580c698471Sdan         rc = SQLITE_ABORT;
36590c698471Sdan         break;
36600c698471Sdan 
36610c698471Sdan       default:
36620c698471Sdan         rc = SQLITE_MISUSE;
36630c698471Sdan         break;
36640c698471Sdan     }
36650c698471Sdan   }
36660c698471Sdan 
36670c698471Sdan   return rc;
36680c698471Sdan }
36690c698471Sdan 
3670296c7658Sdan /*
3671296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
3672296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
3673296c7658Sdan ** the conflict handler callback.
3674296c7658Sdan **
3675296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
3676296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
3677296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
3678296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
3679296c7658Sdan ** to true before returning. In this case the caller will invoke this function
3680296c7658Sdan ** again, this time with pbRetry set to NULL.
3681296c7658Sdan **
3682296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
3683296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
3684296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
3685296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
3686296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
3687296c7658Sdan ** row before invoking this function again, this time with pbReplace set
3688296c7658Sdan ** to NULL.
3689296c7658Sdan **
3690296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
3691296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
3692296c7658Sdan ** returned.
3693296c7658Sdan */
36940c698471Sdan static int sessionApplyOneOp(
3695296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3696296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
36970c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
3698296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
3699296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
3700296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
37010c698471Sdan ){
37020c698471Sdan   const char *zDummy;
37030c698471Sdan   int op;
37040c698471Sdan   int nCol;
37050c698471Sdan   int rc = SQLITE_OK;
37060c698471Sdan 
37070c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
37080c698471Sdan   assert( p->azCol && p->abPK );
37090c698471Sdan   assert( !pbReplace || *pbReplace==0 );
37100c698471Sdan 
3711b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
37120c698471Sdan 
37130c698471Sdan   if( op==SQLITE_DELETE ){
37140c698471Sdan 
371573b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
371673b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
371773b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
371873b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
371973b3c055Sdan     ** is not required if:
372073b3c055Sdan     **
372173b3c055Sdan     **   * this is a patchset, or
372273b3c055Sdan     **   * (pbRetry==0), or
372373b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
372473b3c055Sdan     **     no (nCol+1) variable to bind to).
372573b3c055Sdan     */
372673b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
372773b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
37287cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
372973b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
37307cf7df7dSdan     }
37310c698471Sdan     if( rc!=SQLITE_OK ) return rc;
37320c698471Sdan 
37330c698471Sdan     sqlite3_step(p->pDelete);
37340c698471Sdan     rc = sqlite3_reset(p->pDelete);
37350c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
37360c698471Sdan       rc = sessionConflictHandler(
37370c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
37380c698471Sdan       );
373935e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
37400c698471Sdan       rc = sessionConflictHandler(
37410c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
37420c698471Sdan       );
37430c698471Sdan     }
37440c698471Sdan 
37450c698471Sdan   }else if( op==SQLITE_UPDATE ){
37460c698471Sdan     int i;
37470c698471Sdan 
37480c698471Sdan     /* Bind values to the UPDATE statement. */
37490c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
37507aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
37517aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
37527aa469cdSdan 
37530c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
37547aa469cdSdan       if( pOld ){
37557aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
37567aa469cdSdan       }
37577aa469cdSdan       if( rc==SQLITE_OK && pNew ){
37587aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
37590c698471Sdan       }
37600c698471Sdan     }
376173b3c055Sdan     if( rc==SQLITE_OK ){
376273b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
376373b3c055Sdan     }
37640c698471Sdan     if( rc!=SQLITE_OK ) return rc;
37650c698471Sdan 
37660c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
37670c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
37680c698471Sdan     sqlite3_step(p->pUpdate);
37690c698471Sdan     rc = sqlite3_reset(p->pUpdate);
37700c698471Sdan 
37710c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
37720c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
37730c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
37740c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
37750c698471Sdan 
37760c698471Sdan       rc = sessionConflictHandler(
37770c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
37780c698471Sdan       );
37790c698471Sdan 
378035e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
3781db04571cSdan       /* This is always a CONSTRAINT conflict. */
3782db04571cSdan       rc = sessionConflictHandler(
3783db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
37840c698471Sdan       );
37850c698471Sdan     }
37860c698471Sdan 
37870c698471Sdan   }else{
37880c698471Sdan     assert( op==SQLITE_INSERT );
37897aa469cdSdan     rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
37900c698471Sdan     if( rc!=SQLITE_OK ) return rc;
37910c698471Sdan 
37920c698471Sdan     sqlite3_step(p->pInsert);
37930c698471Sdan     rc = sqlite3_reset(p->pInsert);
379435e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
37950c698471Sdan       rc = sessionConflictHandler(
37960c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
37970c698471Sdan       );
379837f133ecSdan     }
379937f133ecSdan   }
380037f133ecSdan 
380137f133ecSdan   return rc;
380237f133ecSdan }
380337f133ecSdan 
3804296c7658Sdan /*
38054757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
38064757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
38074757c658Sdan ** changeset to the main database attached to handle "db". The supplied
38084757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
38094757c658Sdan ** while applying the change.
3810296c7658Sdan */
38114757c658Sdan static int sessionChangesetApply(
3812296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
38134757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
381440368988Sdan   int(*xFilter)(
381540368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
381640368988Sdan     const char *zTab              /* Table name */
381740368988Sdan   ),
3818d5f0767cSdan   int(*xConflict)(
3819d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
3820d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
3821d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
3822d5f0767cSdan   ),
3823296c7658Sdan   void *pCtx                      /* First argument passed to xConflict */
3824d5f0767cSdan ){
3825ca62ad57Sdan   int schemaMismatch = 0;
3826296c7658Sdan   int rc;                         /* Return code */
3827d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
3828cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
3829296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
3830d5f0767cSdan 
3831082c96dfSdan   assert( xConflict!=0 );
3832082c96dfSdan 
38330c698471Sdan   memset(&sApply, 0, sizeof(sApply));
38344c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
38350c698471Sdan   rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
3836cb3e4b79Sdan   if( rc==SQLITE_OK ){
3837cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
3838cb3e4b79Sdan   }
38390c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
38400c698471Sdan     int nCol;
3841d5f0767cSdan     int op;
38420c698471Sdan     int bReplace = 0;
38430c698471Sdan     int bRetry = 0;
38440c698471Sdan     const char *zNew;
3845ca62ad57Sdan 
3846b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
3847d5f0767cSdan 
38480c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
3849ca62ad57Sdan       u8 *abPK;
3850ca62ad57Sdan 
3851cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
38520c698471Sdan       sqlite3_finalize(sApply.pDelete);
38530c698471Sdan       sqlite3_finalize(sApply.pUpdate);
38540c698471Sdan       sqlite3_finalize(sApply.pInsert);
38550c698471Sdan       sqlite3_finalize(sApply.pSelect);
38560c698471Sdan       memset(&sApply, 0, sizeof(sApply));
38570c698471Sdan       sApply.db = db;
385837f133ecSdan 
385940368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
386040368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
386140368988Sdan       ** non-zero, proceed. */
386240368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
386340368988Sdan       if( schemaMismatch ){
386440368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
3865f05ac112Sdan         if( zTab==0 ){
3866f05ac112Sdan           rc = SQLITE_NOMEM;
3867f05ac112Sdan           break;
3868f05ac112Sdan         }
38694f528042Sdan         nTab = (int)strlen(zTab);
387040368988Sdan         sApply.azCol = (const char **)zTab;
387140368988Sdan       }else{
3872ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
3873296c7658Sdan         rc = sessionTableInfo(
3874ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
3875ca62ad57Sdan         );
3876ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
38770c698471Sdan 
3878ca62ad57Sdan         if( sApply.nCol==0 ){
3879ca62ad57Sdan           schemaMismatch = 1;
3880ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
3881ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
3882ca62ad57Sdan           );
3883ca62ad57Sdan         }
3884ca62ad57Sdan         else if( sApply.nCol!=nCol ){
3885ca62ad57Sdan           schemaMismatch = 1;
3886ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
3887ca62ad57Sdan               "sqlite3changeset_apply(): table %s has %d columns, expected %d",
3888ca62ad57Sdan               zTab, sApply.nCol, nCol
3889ca62ad57Sdan           );
3890ca62ad57Sdan         }
3891ca62ad57Sdan         else if( memcmp(sApply.abPK, abPK, nCol)!=0 ){
3892ca62ad57Sdan           schemaMismatch = 1;
389340368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
389440368988Sdan               "primary key mismatch for table %s", zTab
3895ca62ad57Sdan           );
3896ca62ad57Sdan         }
3897ca62ad57Sdan         else if(
3898ca62ad57Sdan             (rc = sessionSelectRow(db, zTab, &sApply))
38990c698471Sdan          || (rc = sessionUpdateRow(db, zTab, &sApply))
39000c698471Sdan          || (rc = sessionDeleteRow(db, zTab, &sApply))
39010c698471Sdan          || (rc = sessionInsertRow(db, zTab, &sApply))
390237f133ecSdan         ){
390337f133ecSdan           break;
390437f133ecSdan         }
3905cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
3906d5f0767cSdan       }
390740368988Sdan     }
3908d5f0767cSdan 
3909ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
3910ca62ad57Sdan     ** next change. A log message has already been issued. */
3911ca62ad57Sdan     if( schemaMismatch ) continue;
3912ca62ad57Sdan 
39130c698471Sdan     rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, &bRetry);
39140c698471Sdan 
39150c698471Sdan     if( rc==SQLITE_OK && bRetry ){
39160c698471Sdan       rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, 0);
39170c698471Sdan     }
39180c698471Sdan 
39190c698471Sdan     if( bReplace ){
3920db04571cSdan       assert( pIter->op==SQLITE_INSERT );
39210c698471Sdan       rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
39220c698471Sdan       if( rc==SQLITE_OK ){
39237aa469cdSdan         rc = sessionBindRow(pIter,
3924db04571cSdan             sqlite3changeset_new, sApply.nCol, sApply.abPK, sApply.pDelete);
39250c698471Sdan         sqlite3_bind_int(sApply.pDelete, sApply.nCol+1, 1);
39260c698471Sdan       }
39270c698471Sdan       if( rc==SQLITE_OK ){
39280c698471Sdan         sqlite3_step(sApply.pDelete);
39290c698471Sdan         rc = sqlite3_reset(sApply.pDelete);
39300c698471Sdan       }
39310c698471Sdan       if( rc==SQLITE_OK ){
39320c698471Sdan         rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, 0, 0);
39330c698471Sdan       }
39340c698471Sdan       if( rc==SQLITE_OK ){
39350c698471Sdan         rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
39360c698471Sdan       }
39370c698471Sdan     }
39380c698471Sdan   }
3939d5f0767cSdan 
3940296c7658Sdan   if( rc==SQLITE_OK ){
3941296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
3942296c7658Sdan   }else{
3943296c7658Sdan     sqlite3changeset_finalize(pIter);
3944296c7658Sdan   }
3945d5f0767cSdan 
3946d5f0767cSdan   if( rc==SQLITE_OK ){
394707001c45Sdrh     int nFk, notUsed;
394807001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
394907001c45Sdrh     if( nFk!=0 ){
3950cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
3951cb3e4b79Sdan       sqlite3_changeset_iter sIter;
3952cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
3953cb3e4b79Sdan       sIter.nCol = nFk;
3954cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
3955cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
3956cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
3957cb3e4b79Sdan       }
3958cb3e4b79Sdan     }
3959cb3e4b79Sdan   }
3960cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
3961cb3e4b79Sdan 
3962cb3e4b79Sdan   if( rc==SQLITE_OK ){
3963d5f0767cSdan     rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
3964d5f0767cSdan   }else{
3965d5f0767cSdan     sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
3966d5f0767cSdan     sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
3967d5f0767cSdan   }
3968d5f0767cSdan 
39690c698471Sdan   sqlite3_finalize(sApply.pInsert);
39700c698471Sdan   sqlite3_finalize(sApply.pDelete);
39710c698471Sdan   sqlite3_finalize(sApply.pUpdate);
39720c698471Sdan   sqlite3_finalize(sApply.pSelect);
3973cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
39744c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
3975d5f0767cSdan   return rc;
3976d5f0767cSdan }
397791ddd559Sdan 
397877fc1d5bSdan /*
39794757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
39804757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
39814757c658Sdan ** to resolve any conflicts encountered while applying the change.
39824757c658Sdan */
39834757c658Sdan int sqlite3changeset_apply(
39844757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
39854757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
39864757c658Sdan   void *pChangeset,               /* Changeset blob */
39874757c658Sdan   int(*xFilter)(
39884757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
39894757c658Sdan     const char *zTab              /* Table name */
39904757c658Sdan   ),
39914757c658Sdan   int(*xConflict)(
39924757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
39934757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
39944757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
39954757c658Sdan   ),
39964757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
39974757c658Sdan ){
39984757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
39994757c658Sdan   int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset);
40004757c658Sdan   if( rc==SQLITE_OK ){
40014757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
40024757c658Sdan   }
40034757c658Sdan   return rc;
40044757c658Sdan }
40054757c658Sdan 
40064757c658Sdan /*
40074757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
40084757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
40094757c658Sdan ** to resolve any conflicts encountered while applying the change.
40104757c658Sdan */
4011f1a08ad8Sdrh int sqlite3changeset_apply_strm(
40124757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
40134757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
40144757c658Sdan   void *pIn,                                          /* First arg for xInput */
40154757c658Sdan   int(*xFilter)(
40164757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
40174757c658Sdan     const char *zTab              /* Table name */
40184757c658Sdan   ),
40194757c658Sdan   int(*xConflict)(
40204757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
40214757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
40224757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
40234757c658Sdan   ),
40244757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
40254757c658Sdan ){
40264757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
4027f1a08ad8Sdrh   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
40284757c658Sdan   if( rc==SQLITE_OK ){
40294757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
40304757c658Sdan   }
40314757c658Sdan   return rc;
40324757c658Sdan }
40334757c658Sdan 
40344757c658Sdan /*
403577fc1d5bSdan ** This function is called to merge two changes to the same row together as
403677fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
403777fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
403877fc1d5bSdan */
40395d607a6eSdan static int sessionChangeMerge(
404077fc1d5bSdan   SessionTable *pTab,             /* Table structure */
404164277f4aSdan   int bPatchset,                  /* True for patchsets */
404277fc1d5bSdan   SessionChange *pExist,          /* Existing change */
404377fc1d5bSdan   int op2,                        /* Second change operation */
404477fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
404577fc1d5bSdan   u8 *aRec,                       /* Second change record */
404677fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
404777fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
40485d607a6eSdan ){
40495d607a6eSdan   SessionChange *pNew = 0;
40505d607a6eSdan 
40515d607a6eSdan   if( !pExist ){
4052cbf6d2d2Sdan     pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec);
40535d607a6eSdan     if( !pNew ){
40545d607a6eSdan       return SQLITE_NOMEM;
40555d607a6eSdan     }
40565d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
4057798693b2Sdan     pNew->op = op2;
40585d607a6eSdan     pNew->bIndirect = bIndirect;
40595d607a6eSdan     pNew->nRecord = nRec;
4060cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
4061cbf6d2d2Sdan     memcpy(pNew->aRecord, aRec, nRec);
40625d607a6eSdan   }else{
4063798693b2Sdan     int op1 = pExist->op;
40645d607a6eSdan 
40655d607a6eSdan     /*
40665d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
40675d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
40685d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
40695d607a6eSdan     **
40705d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
40715d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
40725d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
40735d607a6eSdan     **
40745d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
40755d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
40765d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
40775d607a6eSdan     */
40785d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
40795d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
40805d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
40815d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
40825d607a6eSdan     ){
40835d607a6eSdan       pNew = pExist;
40845d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
40855d607a6eSdan       sqlite3_free(pExist);
40865d607a6eSdan       assert( pNew==0 );
40875d607a6eSdan     }else{
408864277f4aSdan       u8 *aExist = pExist->aRecord;
40895d607a6eSdan       int nByte;
40905d607a6eSdan       u8 *aCsr;
40915d607a6eSdan 
409264277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
409364277f4aSdan       ** buffer of the new object is large enough to hold any record that
409464277f4aSdan       ** may be generated by combining the input records.  */
40955d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
40965d607a6eSdan       pNew = (SessionChange *)sqlite3_malloc(nByte);
40975d607a6eSdan       if( !pNew ){
40981756ae10Sdan         sqlite3_free(pExist);
40995d607a6eSdan         return SQLITE_NOMEM;
41005d607a6eSdan       }
41015d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
41025d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
41035d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
41045d607a6eSdan 
4105b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
41065d607a6eSdan         u8 *a1 = aRec;
4107b08a1efaSdan         assert( op2==SQLITE_UPDATE );
4108798693b2Sdan         pNew->op = SQLITE_INSERT;
4109ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
411064277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
4111b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
4112b08a1efaSdan         assert( op2==SQLITE_INSERT );
4113798693b2Sdan         pNew->op = SQLITE_UPDATE;
4114fa29ecc4Sdan         if( bPatchset ){
4115fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
4116fa29ecc4Sdan           aCsr += nRec;
4117fa29ecc4Sdan         }else{
411864277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
4119b08a1efaSdan             sqlite3_free(pNew);
4120b08a1efaSdan             pNew = 0;
41215d607a6eSdan           }
4122fa29ecc4Sdan         }
4123b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
412464277f4aSdan         u8 *a1 = aExist;
41255d607a6eSdan         u8 *a2 = aRec;
4126cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
412764277f4aSdan         if( bPatchset==0 ){
4128ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
4129ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
413064277f4aSdan         }
4131798693b2Sdan         pNew->op = SQLITE_UPDATE;
413264277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
41335d607a6eSdan           sqlite3_free(pNew);
41345d607a6eSdan           pNew = 0;
41355d607a6eSdan         }
4136b08a1efaSdan       }else{                                /* UPDATE + DELETE */
4137b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
4138798693b2Sdan         pNew->op = SQLITE_DELETE;
413964277f4aSdan         if( bPatchset ){
414064277f4aSdan           memcpy(aCsr, aRec, nRec);
414164277f4aSdan           aCsr += nRec;
414264277f4aSdan         }else{
414364277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
414464277f4aSdan         }
41455d607a6eSdan       }
41465d607a6eSdan 
41475d607a6eSdan       if( pNew ){
41484f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
41495d607a6eSdan       }
41505d607a6eSdan       sqlite3_free(pExist);
41515d607a6eSdan     }
41525d607a6eSdan   }
41535d607a6eSdan 
41545d607a6eSdan   *ppNew = pNew;
41555d607a6eSdan   return SQLITE_OK;
41565d607a6eSdan }
41575d607a6eSdan 
415877fc1d5bSdan /*
415977fc1d5bSdan ** Add all changes in the changeset passed via the first two arguments to
416077fc1d5bSdan ** hash tables.
416177fc1d5bSdan */
416216228167Sdan static int sessionChangesetToHash(
4163cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
416477fc1d5bSdan   SessionTable **ppTabList        /* IN/OUT: List of table objects */
41655d607a6eSdan ){
41665d607a6eSdan   u8 *aRec;
41675d607a6eSdan   int nRec;
4168cbf6d2d2Sdan   int rc = SQLITE_OK;
41695d607a6eSdan   SessionTable *pTab = 0;
41705d607a6eSdan 
41715d607a6eSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec) ){
41725d607a6eSdan     const char *zNew;
41735d607a6eSdan     int nCol;
41745d607a6eSdan     int op;
41755d607a6eSdan     int iHash;
41765d607a6eSdan     int bIndirect;
41775d607a6eSdan     SessionChange *pChange;
41785d607a6eSdan     SessionChange *pExist = 0;
41795d607a6eSdan     SessionChange **pp;
41805d607a6eSdan 
41815d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
4182ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
41835d607a6eSdan       /* Search the list for a matching table */
41844f528042Sdan       int nNew = (int)strlen(zNew);
4185f29123b5Sdan       u8 *abPK;
4186f29123b5Sdan 
4187f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
41885d607a6eSdan       for(pTab = *ppTabList; pTab; pTab=pTab->pNext){
41895d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
41905d607a6eSdan       }
41915d607a6eSdan       if( !pTab ){
4192ef7a6304Sdan         pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1);
4193f29123b5Sdan         if( !pTab ){
4194f29123b5Sdan           rc = SQLITE_NOMEM;
4195f29123b5Sdan           break;
4196f29123b5Sdan         }
41975d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
41985d607a6eSdan         pTab->pNext = *ppTabList;
4199f29123b5Sdan         pTab->nCol = nCol;
4200ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
4201ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
4202ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
4203ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
42045d607a6eSdan         *ppTabList = pTab;
4205f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
4206f29123b5Sdan         rc = SQLITE_SCHEMA;
4207f29123b5Sdan         break;
42085d607a6eSdan       }
42095d607a6eSdan     }
42105d607a6eSdan 
4211cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
42121756ae10Sdan       rc = SQLITE_NOMEM;
42131756ae10Sdan       break;
42141756ae10Sdan     }
421564277f4aSdan     iHash = sessionChangeHash(
4216cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
421764277f4aSdan     );
42185d607a6eSdan 
42195d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
42205d607a6eSdan     ** Code below may link it back in.
42215d607a6eSdan     */
42225d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
422364277f4aSdan       int bPkOnly1 = 0;
422464277f4aSdan       int bPkOnly2 = 0;
4225cbf6d2d2Sdan       if( pIter->bPatchset ){
422664277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
422764277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
422864277f4aSdan       }
422964277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
42305d607a6eSdan         pExist = *pp;
42315d607a6eSdan         *pp = (*pp)->pNext;
42325d607a6eSdan         pTab->nEntry--;
42335d607a6eSdan         break;
42345d607a6eSdan       }
42355d607a6eSdan     }
42365d607a6eSdan 
423764277f4aSdan     rc = sessionChangeMerge(pTab,
4238cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
423964277f4aSdan     );
42405d607a6eSdan     if( rc ) break;
42415d607a6eSdan     if( pChange ){
42425d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
42435d607a6eSdan       pTab->apChange[iHash] = pChange;
42445d607a6eSdan       pTab->nEntry++;
42455d607a6eSdan     }
42465d607a6eSdan   }
42475d607a6eSdan 
4248cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
42495d607a6eSdan   return rc;
42505d607a6eSdan }
42515d607a6eSdan 
42525d607a6eSdan /*
42535d607a6eSdan ** 1. Iterate through the left-hand changeset. Add an entry to a table
42545d607a6eSdan **    specific hash table for each change in the changeset. The hash table
42555d607a6eSdan **    key is the PK of the row affected by the change.
42565d607a6eSdan **
42575d607a6eSdan ** 2. Then interate through the right-hand changeset. Attempt to add an
42585d607a6eSdan **    entry to a hash table for each component change. If a change already
42595d607a6eSdan **    exists with the same PK values, combine the two into a single change.
42605d607a6eSdan **
42615d607a6eSdan ** 3. Write an output changeset based on the contents of the hash table.
42625d607a6eSdan */
4263cbf6d2d2Sdan int sessionChangesetConcat(
4264cbf6d2d2Sdan   sqlite3_changeset_iter *pLeft,
4265cbf6d2d2Sdan   sqlite3_changeset_iter *pRight,
4266cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4267cbf6d2d2Sdan   void *pOut,
4268cbf6d2d2Sdan   int *pnOut,
4269cbf6d2d2Sdan   void **ppOut
42705d607a6eSdan ){
42715d607a6eSdan   SessionTable *pList = 0;        /* List of SessionTable objects */
42725d607a6eSdan   int rc;                         /* Return code */
427364277f4aSdan   int bPatch;                     /* True for a patchset */
4274e8fa8c96Sdan   SessionTable *pTab;
4275e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
42765d607a6eSdan 
4277cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
42785d607a6eSdan 
4279abae0c4fSdan   assert( pLeft->zTab==0 && pRight->zTab==0 );
428016228167Sdan   rc = sessionChangesetToHash(pLeft, &pList);
4281abae0c4fSdan   assert( pLeft->zTab || pList==0 );
42825d607a6eSdan   if( rc==SQLITE_OK ){
428316228167Sdan     rc = sessionChangesetToHash(pRight, &pList);
42845d607a6eSdan   }
4285cbf6d2d2Sdan   bPatch = pLeft->bPatchset || pRight->bPatchset;
42865d607a6eSdan 
4287abae0c4fSdan   if( pLeft->zTab && pRight->zTab && pLeft->bPatchset!=pRight->bPatchset ){
4288abae0c4fSdan     rc = SQLITE_ERROR;
4289abae0c4fSdan   }
4290abae0c4fSdan 
42915d607a6eSdan   /* Create the serialized output changeset based on the contents of the
42925d607a6eSdan   ** hash tables attached to the SessionTable objects in list pList.
42935d607a6eSdan   */
4294e8fa8c96Sdan   for(pTab=pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
42955d607a6eSdan     int i;
42965d607a6eSdan     if( pTab->nEntry==0 ) continue;
42975d607a6eSdan 
429864277f4aSdan     sessionAppendTableHdr(&buf, bPatch, pTab, &rc);
42995d607a6eSdan     for(i=0; i<pTab->nChange; i++){
43005d607a6eSdan       SessionChange *p;
43015d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
4302798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
43035d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
43045d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
43055d607a6eSdan       }
43065d607a6eSdan     }
4307cbf6d2d2Sdan 
4308f1a08ad8Sdrh     if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
4309cbf6d2d2Sdan       rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4310cbf6d2d2Sdan       buf.nBuf = 0;
4311cbf6d2d2Sdan     }
43125d607a6eSdan   }
43135d607a6eSdan 
43145d607a6eSdan   if( rc==SQLITE_OK ){
4315cbf6d2d2Sdan     if( xOutput ){
4316cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4317cbf6d2d2Sdan     }else{
43185d607a6eSdan       *ppOut = buf.aBuf;
43195d607a6eSdan       *pnOut = buf.nBuf;
4320cbf6d2d2Sdan       buf.aBuf = 0;
43215d607a6eSdan     }
43225d607a6eSdan   }
4323cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
43245d607a6eSdan 
43255d607a6eSdan   sessionDeleteTable(pList);
43265d607a6eSdan   return rc;
43275d607a6eSdan }
43285d607a6eSdan 
4329cbf6d2d2Sdan /*
4330cbf6d2d2Sdan ** Combine two changesets together.
4331cbf6d2d2Sdan */
4332cbf6d2d2Sdan int sqlite3changeset_concat(
4333cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
4334cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
4335cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
4336cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
4337cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
4338cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
4339cbf6d2d2Sdan ){
4340cbf6d2d2Sdan   sqlite3_changeset_iter *pIter1 = 0;
4341cbf6d2d2Sdan   sqlite3_changeset_iter *pIter2 = 0;
4342cbf6d2d2Sdan   int rc;
4343cbf6d2d2Sdan 
4344cbf6d2d2Sdan   *pnOut = 0;
4345cbf6d2d2Sdan   *ppOut = 0;
4346cbf6d2d2Sdan   rc = sqlite3changeset_start(&pIter1, nLeft, pLeft);
4347cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4348cbf6d2d2Sdan     rc = sqlite3changeset_start(&pIter2, nRight, pRight);
4349cbf6d2d2Sdan   }
4350cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4351cbf6d2d2Sdan     rc = sessionChangesetConcat(pIter1, pIter2, 0, 0, pnOut, ppOut);
4352cbf6d2d2Sdan   }
4353cbf6d2d2Sdan 
4354cbf6d2d2Sdan   sqlite3changeset_finalize(pIter1);
4355cbf6d2d2Sdan   sqlite3changeset_finalize(pIter2);
4356cbf6d2d2Sdan   return rc;
4357cbf6d2d2Sdan }
4358cbf6d2d2Sdan 
4359cbf6d2d2Sdan /*
4360cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
4361cbf6d2d2Sdan */
4362f1a08ad8Sdrh int sqlite3changeset_concat_strm(
4363cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
4364cbf6d2d2Sdan   void *pInA,
4365cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
4366cbf6d2d2Sdan   void *pInB,
4367cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4368cbf6d2d2Sdan   void *pOut
4369cbf6d2d2Sdan ){
4370cbf6d2d2Sdan   sqlite3_changeset_iter *pIter1 = 0;
4371cbf6d2d2Sdan   sqlite3_changeset_iter *pIter2 = 0;
4372cbf6d2d2Sdan   int rc;
4373cbf6d2d2Sdan 
4374f1a08ad8Sdrh   rc = sqlite3changeset_start_strm(&pIter1, xInputA, pInA);
4375cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4376f1a08ad8Sdrh     rc = sqlite3changeset_start_strm(&pIter2, xInputB, pInB);
4377cbf6d2d2Sdan   }
4378cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4379cbf6d2d2Sdan     rc = sessionChangesetConcat(pIter1, pIter2, xOutput, pOut, 0, 0);
4380cbf6d2d2Sdan   }
4381cbf6d2d2Sdan 
4382cbf6d2d2Sdan   sqlite3changeset_finalize(pIter1);
4383cbf6d2d2Sdan   sqlite3changeset_finalize(pIter2);
4384cbf6d2d2Sdan   return rc;
4385cbf6d2d2Sdan }
4386cbf6d2d2Sdan 
43879b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
4388