14fccf43aSdan 
29b1c62d4Sdrh #if defined(SQLITE_ENABLE_SESSION) && defined(SQLITE_ENABLE_PREUPDATE_HOOK)
34fccf43aSdan #include "sqlite3session.h"
44fccf43aSdan #include <assert.h>
54fccf43aSdan #include <string.h>
64fccf43aSdan 
75d8a2984Sdan #ifndef SQLITE_AMALGAMATION
84fccf43aSdan # include "sqliteInt.h"
94fccf43aSdan # include "vdbeInt.h"
105d8a2984Sdan #endif
114fccf43aSdan 
124fccf43aSdan typedef struct SessionTable SessionTable;
134fccf43aSdan typedef struct SessionChange SessionChange;
14296c7658Sdan typedef struct SessionBuffer SessionBuffer;
15ef7a6304Sdan typedef struct SessionInput SessionInput;
16ef7a6304Sdan 
17ef7a6304Sdan /*
18ef7a6304Sdan ** Minimum chunk size used by streaming versions of functions.
19ef7a6304Sdan */
20f1a08ad8Sdrh #ifndef SESSIONS_STRM_CHUNK_SIZE
214757c658Sdan # ifdef SQLITE_TEST
22f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 64
234757c658Sdan # else
24f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 1024
25f1a08ad8Sdrh # endif
264757c658Sdan #endif
274fccf43aSdan 
28cf8e9144Sdan typedef struct SessionHook SessionHook;
29cf8e9144Sdan struct SessionHook {
30cf8e9144Sdan   void *pCtx;
31cf8e9144Sdan   int (*xOld)(void*,int,sqlite3_value**);
32cf8e9144Sdan   int (*xNew)(void*,int,sqlite3_value**);
33cf8e9144Sdan   int (*xCount)(void*);
34cf8e9144Sdan   int (*xDepth)(void*);
35cf8e9144Sdan };
36cf8e9144Sdan 
37296c7658Sdan /*
38296c7658Sdan ** Session handle structure.
39296c7658Sdan */
404fccf43aSdan struct sqlite3_session {
414fccf43aSdan   sqlite3 *db;                    /* Database handle session is attached to */
424fccf43aSdan   char *zDb;                      /* Name of database session is attached to */
43296c7658Sdan   int bEnable;                    /* True if currently recording */
44b4480e94Sdan   int bIndirect;                  /* True if all changes are indirect */
45ff4d0f41Sdan   int bAutoAttach;                /* True to auto-attach tables */
464fccf43aSdan   int rc;                         /* Non-zero if an error has occurred */
477531a5a3Sdan   void *pFilterCtx;               /* First argument to pass to xTableFilter */
487531a5a3Sdan   int (*xTableFilter)(void *pCtx, const char *zTab);
491611e5a3Sdan   sqlite3_value *pZeroBlob;       /* Value containing X'' */
504fccf43aSdan   sqlite3_session *pNext;         /* Next session object on same db. */
514fccf43aSdan   SessionTable *pTable;           /* List of attached tables */
52cf8e9144Sdan   SessionHook hook;               /* APIs to grab new and old data with */
534fccf43aSdan };
544fccf43aSdan 
554fccf43aSdan /*
56ef7a6304Sdan ** Instances of this structure are used to build strings or binary records.
57ef7a6304Sdan */
58ef7a6304Sdan struct SessionBuffer {
59ef7a6304Sdan   u8 *aBuf;                       /* Pointer to changeset buffer */
60ef7a6304Sdan   int nBuf;                       /* Size of buffer aBuf */
61ef7a6304Sdan   int nAlloc;                     /* Size of allocation containing aBuf */
62ef7a6304Sdan };
63ef7a6304Sdan 
64ef7a6304Sdan /*
6516228167Sdan ** An object of this type is used internally as an abstraction for
6616228167Sdan ** input data. Input data may be supplied either as a single large buffer
6716228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g.
68f1a08ad8Sdrh **  sqlite3changeset_start_strm()).
69ef7a6304Sdan */
70ef7a6304Sdan struct SessionInput {
71d9151526Sdan   int bNoDiscard;                 /* If true, discard no data */
72d9151526Sdan   int iCurrent;                   /* Offset in aData[] of current change */
734757c658Sdan   int iNext;                      /* Offset in aData[] of next change */
744757c658Sdan   u8 *aData;                      /* Pointer to buffer containing changeset */
754757c658Sdan   int nData;                      /* Number of bytes in aData */
764757c658Sdan 
77ef7a6304Sdan   SessionBuffer buf;              /* Current read buffer */
78ef7a6304Sdan   int (*xInput)(void*, void*, int*);        /* Input stream call (or NULL) */
79ef7a6304Sdan   void *pIn;                                /* First argument to xInput */
80ef7a6304Sdan   int bEof;                       /* Set to true after xInput finished */
81ef7a6304Sdan };
82ef7a6304Sdan 
83ef7a6304Sdan /*
84296c7658Sdan ** Structure for changeset iterators.
85296c7658Sdan */
86296c7658Sdan struct sqlite3_changeset_iter {
87ef7a6304Sdan   SessionInput in;                /* Input buffer or stream */
88ef7a6304Sdan   SessionBuffer tblhdr;           /* Buffer to hold apValue/zTab/abPK/ */
8973b3c055Sdan   int bPatchset;                  /* True if this is a patchset */
90296c7658Sdan   int rc;                         /* Iterator error code */
91296c7658Sdan   sqlite3_stmt *pConflict;        /* Points to conflicting row, if any */
92296c7658Sdan   char *zTab;                     /* Current table */
93296c7658Sdan   int nCol;                       /* Number of columns in zTab */
94296c7658Sdan   int op;                         /* Current operation */
95b4480e94Sdan   int bIndirect;                  /* True if current change was indirect */
96244593c8Sdan   u8 *abPK;                       /* Primary key array */
97296c7658Sdan   sqlite3_value **apValue;        /* old.* and new.* values */
98296c7658Sdan };
99296c7658Sdan 
100296c7658Sdan /*
1014fccf43aSdan ** Each session object maintains a set of the following structures, one
1024fccf43aSdan ** for each table the session object is monitoring. The structures are
1034fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable.
1044fccf43aSdan **
1054fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have
1064fccf43aSdan ** been modified in any way since the session object was attached to the
1074fccf43aSdan ** table.
1084fccf43aSdan **
1094fccf43aSdan ** The data associated with each hash-table entry is a structure containing
1104fccf43aSdan ** a subset of the initial values that the modified row contained at the
1114fccf43aSdan ** start of the session. Or no initial values if the row was inserted.
1124fccf43aSdan */
1134fccf43aSdan struct SessionTable {
1144fccf43aSdan   SessionTable *pNext;
1154fccf43aSdan   char *zName;                    /* Local name of table */
1164fccf43aSdan   int nCol;                       /* Number of columns in table zName */
1173739f298Sdan   int bStat1;                     /* True if this is sqlite_stat1 */
118e8d5648eSdan   const char **azCol;             /* Column names */
119e8d5648eSdan   u8 *abPK;                       /* Array of primary key flags */
120296c7658Sdan   int nEntry;                     /* Total number of entries in hash table */
1214fccf43aSdan   int nChange;                    /* Size of apChange[] array */
1224fccf43aSdan   SessionChange **apChange;       /* Hash table buckets */
1234fccf43aSdan };
1244fccf43aSdan 
1254fccf43aSdan /*
1264fccf43aSdan ** RECORD FORMAT:
1274fccf43aSdan **
1284fccf43aSdan ** The following record format is similar to (but not compatible with) that
1294fccf43aSdan ** used in SQLite database files. This format is used as part of the
1304fccf43aSdan ** change-set binary format, and so must be architecture independent.
1314fccf43aSdan **
1324fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained -
1334fccf43aSdan ** there is no separation of header and data. Each field begins with a
1344fccf43aSdan ** single byte describing its type, as follows:
1354fccf43aSdan **
1364fccf43aSdan **       0x00: Undefined value.
1374fccf43aSdan **       0x01: Integer value.
1384fccf43aSdan **       0x02: Real value.
1394fccf43aSdan **       0x03: Text value.
1404fccf43aSdan **       0x04: Blob value.
1414fccf43aSdan **       0x05: SQL NULL value.
1424fccf43aSdan **
1434fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT
1444fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists
1454fccf43aSdan ** only of the single type byte. For other types of values, the type byte
1464fccf43aSdan ** is followed by:
1474fccf43aSdan **
1484fccf43aSdan **   Text values:
1494fccf43aSdan **     A varint containing the number of bytes in the value (encoded using
1504fccf43aSdan **     UTF-8). Followed by a buffer containing the UTF-8 representation
1514fccf43aSdan **     of the text value. There is no nul terminator.
1524fccf43aSdan **
1534fccf43aSdan **   Blob values:
1544fccf43aSdan **     A varint containing the number of bytes in the value, followed by
1554fccf43aSdan **     a buffer containing the value itself.
1564fccf43aSdan **
1574fccf43aSdan **   Integer values:
1584fccf43aSdan **     An 8-byte big-endian integer value.
1594fccf43aSdan **
1604fccf43aSdan **   Real values:
1614fccf43aSdan **     An 8-byte big-endian IEEE 754-2008 real value.
1624fccf43aSdan **
1634fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite
1644fccf43aSdan ** record format.
1654fccf43aSdan **
1664fccf43aSdan ** CHANGESET FORMAT:
1674fccf43aSdan **
1684fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on
1694fccf43aSdan ** one or more tables. Operations on a single table are grouped together,
1704fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all
1714fccf43aSdan ** mixed together).
1724fccf43aSdan **
1734fccf43aSdan ** Each group of changes begins with a table header:
1744fccf43aSdan **
1754fccf43aSdan **   1 byte: Constant 0x54 (capital 'T')
176730bb805Sdan **   Varint: Number of columns in the table.
17773b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
1784fccf43aSdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
1794fccf43aSdan **
1804fccf43aSdan ** Followed by one or more changes to the table.
1814fccf43aSdan **
182c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
1835d607a6eSdan **   1 byte: The "indirect-change" flag.
1844fccf43aSdan **   old.* record: (delete and update only)
1854fccf43aSdan **   new.* record: (insert and update only)
18673b3c055Sdan **
187730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the
188730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of
189730bb805Sdan ** columns in the table. The i'th field of each record is associated with
190730bb805Sdan ** the i'th column of the table, counting from left to right in the order
191730bb805Sdan ** in which columns were declared in the CREATE TABLE statement.
192730bb805Sdan **
193730bb805Sdan ** The new.* record that is part of each INSERT change contains the values
194730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each
195730bb805Sdan ** DELETE change contains the values that made up the row that was deleted
196730bb805Sdan ** from the database. In the changeset format, the records that are part
197730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00)
198730bb805Sdan ** fields.
199730bb805Sdan **
200730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields
201730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are
202730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields
203730bb805Sdan ** are set to the values that made up the row before the UPDATE that the
204730bb805Sdan ** change records took place. Within the new.* record, fields associated
205730bb805Sdan ** with table columns modified by the UPDATE change contain the new
206730bb805Sdan ** values. Fields associated with table columns that are not modified
207730bb805Sdan ** are set to "undefined".
208730bb805Sdan **
20973b3c055Sdan ** PATCHSET FORMAT:
21073b3c055Sdan **
21173b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset,
212730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict
213730bb805Sdan ** resolution is required when applying the changeset.
21473b3c055Sdan **
21573b3c055Sdan ** Each group of changes begins with a table header:
21673b3c055Sdan **
21773b3c055Sdan **   1 byte: Constant 0x50 (capital 'P')
218730bb805Sdan **   Varint: Number of columns in the table.
21973b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
22073b3c055Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
22173b3c055Sdan **
22273b3c055Sdan ** Followed by one or more changes to the table.
22373b3c055Sdan **
224c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
22573b3c055Sdan **   1 byte: The "indirect-change" flag.
226730bb805Sdan **   single record: (PK fields for DELETE, PK and modified fields for UPDATE,
227730bb805Sdan **                   full record for INSERT).
228730bb805Sdan **
229730bb805Sdan ** As in the changeset format, each field of the single record that is part
230730bb805Sdan ** of a patchset change is associated with the correspondingly positioned
231730bb805Sdan ** table column, counting from left to right within the CREATE TABLE
232730bb805Sdan ** statement.
233730bb805Sdan **
234730bb805Sdan ** For a DELETE change, all fields within the record except those associated
235730bb805Sdan ** with PRIMARY KEY columns are set to "undefined". The PRIMARY KEY fields
236730bb805Sdan ** contain the values identifying the row to delete.
237730bb805Sdan **
238730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY
239730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined".
240730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update,
241730bb805Sdan ** and fields associated with modified columns contain the new column values.
242730bb805Sdan **
243730bb805Sdan ** The records associated with INSERT changes are in the same format as for
244730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT
245730bb805Sdan ** change to contain a field set to "undefined".
2464fccf43aSdan */
2474fccf43aSdan 
2484fccf43aSdan /*
2494fccf43aSdan ** For each row modified during a session, there exists a single instance of
2504fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table.
2514fccf43aSdan */
2524fccf43aSdan struct SessionChange {
253798693b2Sdan   int op;                         /* One of UPDATE, DELETE, INSERT */
254b4480e94Sdan   int bIndirect;                  /* True if this change is "indirect" */
2554fccf43aSdan   int nRecord;                    /* Number of bytes in buffer aRecord[] */
2564fccf43aSdan   u8 *aRecord;                    /* Buffer containing old.* record */
2574fccf43aSdan   SessionChange *pNext;           /* For hash-table collisions */
2584fccf43aSdan };
2594fccf43aSdan 
260296c7658Sdan /*
261296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the
262296c7658Sdan ** number of bytes written.
263296c7658Sdan */
264296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){
265296c7658Sdan   return putVarint32(aBuf, iVal);
2664fccf43aSdan }
2674fccf43aSdan 
268296c7658Sdan /*
269296c7658Sdan ** Return the number of bytes required to store value iVal as a varint.
270296c7658Sdan */
271296c7658Sdan static int sessionVarintLen(int iVal){
272296c7658Sdan   return sqlite3VarintLen(iVal);
273296c7658Sdan }
274296c7658Sdan 
275296c7658Sdan /*
276296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of
277296c7658Sdan ** bytes read.
278296c7658Sdan */
2794fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){
280296c7658Sdan   return getVarint32(aBuf, *piVal);
2814fccf43aSdan }
2824fccf43aSdan 
28348cd59a5Sdrh /* Load an unaligned and unsigned 32-bit integer */
28448cd59a5Sdrh #define SESSION_UINT32(x) (((u32)(x)[0]<<24)|((x)[1]<<16)|((x)[2]<<8)|(x)[3])
28548cd59a5Sdrh 
286296c7658Sdan /*
287296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return
288296c7658Sdan ** the value read.
289296c7658Sdan */
2904fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){
29148cd59a5Sdrh   u64 x = SESSION_UINT32(aRec);
29248cd59a5Sdrh   u32 y = SESSION_UINT32(aRec+4);
29348cd59a5Sdrh   x = (x<<32) + y;
29448cd59a5Sdrh   return (sqlite3_int64)x;
2954fccf43aSdan }
2964fccf43aSdan 
2974fccf43aSdan /*
298296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[].
299296c7658Sdan */
300296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){
301296c7658Sdan   aBuf[0] = (i>>56) & 0xFF;
302296c7658Sdan   aBuf[1] = (i>>48) & 0xFF;
303296c7658Sdan   aBuf[2] = (i>>40) & 0xFF;
304296c7658Sdan   aBuf[3] = (i>>32) & 0xFF;
305296c7658Sdan   aBuf[4] = (i>>24) & 0xFF;
306296c7658Sdan   aBuf[5] = (i>>16) & 0xFF;
307296c7658Sdan   aBuf[6] = (i>> 8) & 0xFF;
308296c7658Sdan   aBuf[7] = (i>> 0) & 0xFF;
309296c7658Sdan }
310296c7658Sdan 
311296c7658Sdan /*
3124fccf43aSdan ** This function is used to serialize the contents of value pValue (see
3134fccf43aSdan ** comment titled "RECORD FORMAT" above).
3144fccf43aSdan **
3154fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to
3164fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before
3174fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is
3184fccf43aSdan ** set *pnWrite.
3194fccf43aSdan **
3204fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs
3214fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16))
3224fccf43aSdan ** SQLITE_NOMEM is returned.
3234fccf43aSdan */
3244fccf43aSdan static int sessionSerializeValue(
3254fccf43aSdan   u8 *aBuf,                       /* If non-NULL, write serialized value here */
3264fccf43aSdan   sqlite3_value *pValue,          /* Value to serialize */
3274fccf43aSdan   int *pnWrite                    /* IN/OUT: Increment by bytes written */
3284fccf43aSdan ){
329296c7658Sdan   int nByte;                      /* Size of serialized value in bytes */
3304fccf43aSdan 
33180fe2d93Sdan   if( pValue ){
33280fe2d93Sdan     int eType;                    /* Value type (SQLITE_NULL, TEXT etc.) */
33380fe2d93Sdan 
3344fccf43aSdan     eType = sqlite3_value_type(pValue);
3354fccf43aSdan     if( aBuf ) aBuf[0] = eType;
3364fccf43aSdan 
3374fccf43aSdan     switch( eType ){
3384fccf43aSdan       case SQLITE_NULL:
3394fccf43aSdan         nByte = 1;
3404fccf43aSdan         break;
3414fccf43aSdan 
3424fccf43aSdan       case SQLITE_INTEGER:
3434fccf43aSdan       case SQLITE_FLOAT:
3444fccf43aSdan         if( aBuf ){
3454fccf43aSdan           /* TODO: SQLite does something special to deal with mixed-endian
3464fccf43aSdan           ** floating point values (e.g. ARM7). This code probably should
3474fccf43aSdan           ** too.  */
3484fccf43aSdan           u64 i;
3494fccf43aSdan           if( eType==SQLITE_INTEGER ){
3504fccf43aSdan             i = (u64)sqlite3_value_int64(pValue);
3514fccf43aSdan           }else{
3524fccf43aSdan             double r;
3534fccf43aSdan             assert( sizeof(double)==8 && sizeof(u64)==8 );
3544fccf43aSdan             r = sqlite3_value_double(pValue);
3554fccf43aSdan             memcpy(&i, &r, 8);
3564fccf43aSdan           }
357296c7658Sdan           sessionPutI64(&aBuf[1], i);
3584fccf43aSdan         }
3594fccf43aSdan         nByte = 9;
3604fccf43aSdan         break;
3614fccf43aSdan 
3624e895da1Sdan       default: {
36380fe2d93Sdan         u8 *z;
36480fe2d93Sdan         int n;
36580fe2d93Sdan         int nVarint;
36680fe2d93Sdan 
3674e895da1Sdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
36880fe2d93Sdan         if( eType==SQLITE_TEXT ){
36980fe2d93Sdan           z = (u8 *)sqlite3_value_text(pValue);
37080fe2d93Sdan         }else{
37180fe2d93Sdan           z = (u8 *)sqlite3_value_blob(pValue);
37280fe2d93Sdan         }
37380fe2d93Sdan         n = sqlite3_value_bytes(pValue);
3743cc89d95Sdan         if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
37580fe2d93Sdan         nVarint = sessionVarintLen(n);
37680fe2d93Sdan 
3774fccf43aSdan         if( aBuf ){
3784fccf43aSdan           sessionVarintPut(&aBuf[1], n);
379895decf6Sdan           if( n ) memcpy(&aBuf[nVarint + 1], z, n);
3804fccf43aSdan         }
3814fccf43aSdan 
3824fccf43aSdan         nByte = 1 + nVarint + n;
3834fccf43aSdan         break;
3844fccf43aSdan       }
3854fccf43aSdan     }
38680fe2d93Sdan   }else{
38780fe2d93Sdan     nByte = 1;
38880fe2d93Sdan     if( aBuf ) aBuf[0] = '\0';
38980fe2d93Sdan   }
3904fccf43aSdan 
391fa122adaSdan   if( pnWrite ) *pnWrite += nByte;
3924fccf43aSdan   return SQLITE_OK;
3934fccf43aSdan }
3944fccf43aSdan 
395fa122adaSdan 
396798693b2Sdan /*
397798693b2Sdan ** This macro is used to calculate hash key values for data structures. In
398798693b2Sdan ** order to use this macro, the entire data structure must be represented
399798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value
400798693b2Sdan ** for a data structure represented as three such integers, the macro may
401798693b2Sdan ** then be used as follows:
402798693b2Sdan **
403798693b2Sdan **    int hash_key_value;
404798693b2Sdan **    hash_key_value = HASH_APPEND(0, <value 1>);
405798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 2>);
406798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 3>);
407798693b2Sdan **
408798693b2Sdan ** In practice, the data structures this macro is used for are the primary
409798693b2Sdan ** key values of modified rows.
410798693b2Sdan */
4114131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add)
412798693b2Sdan 
413798693b2Sdan /*
414798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the
415798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
416798693b2Sdan */
4174131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){
418e8d5648eSdan   h = HASH_APPEND(h, i & 0xFFFFFFFF);
419e8d5648eSdan   return HASH_APPEND(h, (i>>32)&0xFFFFFFFF);
420e8d5648eSdan }
421798693b2Sdan 
422798693b2Sdan /*
423798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to
424798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value.
425798693b2Sdan */
4264131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){
427e8d5648eSdan   int i;
428e8d5648eSdan   for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]);
429e8d5648eSdan   return h;
430e8d5648eSdan }
431e8d5648eSdan 
4324fccf43aSdan /*
433798693b2Sdan ** Append the hash of the data type passed as the second argument to the
434798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
435798693b2Sdan */
436798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){
437798693b2Sdan   return HASH_APPEND(h, eType);
438798693b2Sdan }
439798693b2Sdan 
440798693b2Sdan /*
4414131639cSdan ** This function may only be called from within a pre-update callback.
4424131639cSdan ** It calculates a hash based on the primary key values of the old.* or
443798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to
444798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL
445798693b2Sdan ** values, *pbNullPK is set to true before returning.
446798693b2Sdan **
447798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values
448798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned
449798693b2Sdan ** and the output variables are set as described above.
4504fccf43aSdan */
451798693b2Sdan static int sessionPreupdateHash(
452cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns pTab */
453e8d5648eSdan   SessionTable *pTab,             /* Session table handle */
454e8d5648eSdan   int bNew,                       /* True to hash the new.* PK */
45527453faeSdan   int *piHash,                    /* OUT: Hash value */
456798693b2Sdan   int *pbNullPK                   /* OUT: True if there are NULL values in PK */
457e8d5648eSdan ){
4584131639cSdan   unsigned int h = 0;             /* Hash value to return */
4594131639cSdan   int i;                          /* Used to iterate through columns */
460e8d5648eSdan 
46127453faeSdan   assert( *pbNullPK==0 );
462cf8e9144Sdan   assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) );
463e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
464e8d5648eSdan     if( pTab->abPK[i] ){
465e8d5648eSdan       int rc;
466e8d5648eSdan       int eType;
467e8d5648eSdan       sqlite3_value *pVal;
468e8d5648eSdan 
469e8d5648eSdan       if( bNew ){
470cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal);
471e8d5648eSdan       }else{
472cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal);
473e8d5648eSdan       }
47412ca0b56Sdan       if( rc!=SQLITE_OK ) return rc;
475e8d5648eSdan 
476e8d5648eSdan       eType = sqlite3_value_type(pVal);
477798693b2Sdan       h = sessionHashAppendType(h, eType);
4786734007dSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
479e8d5648eSdan         i64 iVal;
480e8d5648eSdan         if( eType==SQLITE_INTEGER ){
481e8d5648eSdan           iVal = sqlite3_value_int64(pVal);
482e8d5648eSdan         }else{
483e8d5648eSdan           double rVal = sqlite3_value_double(pVal);
484e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
485e8d5648eSdan           memcpy(&iVal, &rVal, 8);
486e8d5648eSdan         }
487e8d5648eSdan         h = sessionHashAppendI64(h, iVal);
4886734007dSdan       }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
4896734007dSdan         const u8 *z;
4903cc89d95Sdan         int n;
4916734007dSdan         if( eType==SQLITE_TEXT ){
4926734007dSdan           z = (const u8 *)sqlite3_value_text(pVal);
4936734007dSdan         }else{
4946734007dSdan           z = (const u8 *)sqlite3_value_blob(pVal);
495e8d5648eSdan         }
4963cc89d95Sdan         n = sqlite3_value_bytes(pVal);
4973cc89d95Sdan         if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
4983cc89d95Sdan         h = sessionHashAppendBlob(h, n, z);
4996734007dSdan       }else{
50027453faeSdan         assert( eType==SQLITE_NULL );
5011611e5a3Sdan         assert( pTab->bStat1==0 || i!=1 );
50227453faeSdan         *pbNullPK = 1;
503e8d5648eSdan       }
504e8d5648eSdan     }
505e8d5648eSdan   }
506e8d5648eSdan 
507e8d5648eSdan   *piHash = (h % pTab->nChange);
508e8d5648eSdan   return SQLITE_OK;
509e8d5648eSdan }
510e8d5648eSdan 
5114131639cSdan /*
5126cda207fSdan ** The buffer that the argument points to contains a serialized SQL value.
5136cda207fSdan ** Return the number of bytes of space occupied by the value (including
5146cda207fSdan ** the type byte).
5156cda207fSdan */
5166cda207fSdan static int sessionSerialLen(u8 *a){
5176cda207fSdan   int e = *a;
5186cda207fSdan   int n;
5196cda207fSdan   if( e==0 ) return 1;
5206cda207fSdan   if( e==SQLITE_NULL ) return 1;
5216cda207fSdan   if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9;
5226cda207fSdan   return sessionVarintGet(&a[1], &n) + 1 + n;
5236cda207fSdan }
5246cda207fSdan 
5256cda207fSdan /*
5265d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a
527798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys
5284131639cSdan ** calculated by this function are compatible with those calculated by
5294131639cSdan ** sessionPreupdateHash().
53064277f4aSdan **
53164277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from
53264277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely.
5334131639cSdan */
5344131639cSdan static unsigned int sessionChangeHash(
5354131639cSdan   SessionTable *pTab,             /* Table handle */
53664277f4aSdan   int bPkOnly,                    /* Record consists of PK fields only */
5375d607a6eSdan   u8 *aRecord,                    /* Change record */
5384131639cSdan   int nBucket                     /* Assume this many buckets in hash table */
539e8d5648eSdan ){
5404131639cSdan   unsigned int h = 0;             /* Value to return */
5414131639cSdan   int i;                          /* Used to iterate through columns */
5425d607a6eSdan   u8 *a = aRecord;                /* Used to iterate through change record */
543e8d5648eSdan 
544e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
5456cda207fSdan     int eType = *a;
546e8d5648eSdan     int isPK = pTab->abPK[i];
54764277f4aSdan     if( bPkOnly && isPK==0 ) continue;
548e8d5648eSdan 
54927453faeSdan     /* It is not possible for eType to be SQLITE_NULL here. The session
55027453faeSdan     ** module does not record changes for rows with NULL values stored in
55127453faeSdan     ** primary key columns. */
55227453faeSdan     assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
55327453faeSdan          || eType==SQLITE_TEXT || eType==SQLITE_BLOB
5546cda207fSdan          || eType==SQLITE_NULL || eType==0
55527453faeSdan     );
5561611e5a3Sdan     assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) );
55727453faeSdan 
5586cda207fSdan     if( isPK ){
5596cda207fSdan       a++;
560798693b2Sdan       h = sessionHashAppendType(h, eType);
56127453faeSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
5626734007dSdan         h = sessionHashAppendI64(h, sessionGetI64(a));
563e8d5648eSdan         a += 8;
5641611e5a3Sdan       }else{
565e8d5648eSdan         int n;
566e8d5648eSdan         a += sessionVarintGet(a, &n);
5676734007dSdan         h = sessionHashAppendBlob(h, n, a);
568e8d5648eSdan         a += n;
569e8d5648eSdan       }
5706cda207fSdan     }else{
5716cda207fSdan       a += sessionSerialLen(a);
5726cda207fSdan     }
573e8d5648eSdan   }
574e8d5648eSdan   return (h % nBucket);
575e8d5648eSdan }
576e8d5648eSdan 
577798693b2Sdan /*
578798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab.
579798693b2Sdan ** This function returns true if the two records apply to the same row (i.e.
580798693b2Sdan ** have the same values stored in the primary key columns), or false
581798693b2Sdan ** otherwise.
582798693b2Sdan */
5835d607a6eSdan static int sessionChangeEqual(
584798693b2Sdan   SessionTable *pTab,             /* Table used for PK definition */
585a71d2371Sdan   int bLeftPkOnly,                /* True if aLeft[] contains PK fields only */
5865d607a6eSdan   u8 *aLeft,                      /* Change record */
587a71d2371Sdan   int bRightPkOnly,               /* True if aRight[] contains PK fields only */
5885d607a6eSdan   u8 *aRight                      /* Change record */
5895d607a6eSdan ){
590798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor to iterate through aLeft */
591798693b2Sdan   u8 *a2 = aRight;                /* Cursor to iterate through aRight */
592798693b2Sdan   int iCol;                       /* Used to iterate through table columns */
5935d607a6eSdan 
594798693b2Sdan   for(iCol=0; iCol<pTab->nCol; iCol++){
59514faa061Sdan     if( pTab->abPK[iCol] ){
5965d607a6eSdan       int n1 = sessionSerialLen(a1);
5975d607a6eSdan       int n2 = sessionSerialLen(a2);
5985d607a6eSdan 
599798693b2Sdan       if( pTab->abPK[iCol] && (n1!=n2 || memcmp(a1, a2, n1)) ){
6005d607a6eSdan         return 0;
6015d607a6eSdan       }
60214faa061Sdan       a1 += n1;
60314faa061Sdan       a2 += n2;
60414faa061Sdan     }else{
60514faa061Sdan       if( bLeftPkOnly==0 ) a1 += sessionSerialLen(a1);
60614faa061Sdan       if( bRightPkOnly==0 ) a2 += sessionSerialLen(a2);
60714faa061Sdan     }
6085d607a6eSdan   }
6095d607a6eSdan 
6105d607a6eSdan   return 1;
6115d607a6eSdan }
6125d607a6eSdan 
613798693b2Sdan /*
614798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change
615798693b2Sdan ** records with nCol columns. This function "merges" the two records into
616798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is
617798693b2Sdan ** then set to point to one byte after the last byte written before
618798693b2Sdan ** returning.
619798693b2Sdan **
620798693b2Sdan ** The merging of records is done as follows: For each column, if the
621798693b2Sdan ** aRight record contains a value for the column, copy the value from
622798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither
623798693b2Sdan ** record contains a value for a given column, then neither does the
624798693b2Sdan ** output record.
625798693b2Sdan */
6265d607a6eSdan static void sessionMergeRecord(
6275d607a6eSdan   u8 **paOut,
628798693b2Sdan   int nCol,
6295d607a6eSdan   u8 *aLeft,
6305d607a6eSdan   u8 *aRight
6315d607a6eSdan ){
632798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor used to iterate through aLeft */
633798693b2Sdan   u8 *a2 = aRight;                /* Cursor used to iterate through aRight */
634798693b2Sdan   u8 *aOut = *paOut;              /* Output cursor */
635798693b2Sdan   int iCol;                       /* Used to iterate from 0 to nCol */
6365d607a6eSdan 
637798693b2Sdan   for(iCol=0; iCol<nCol; iCol++){
6385d607a6eSdan     int n1 = sessionSerialLen(a1);
6395d607a6eSdan     int n2 = sessionSerialLen(a2);
6405d607a6eSdan     if( *a2 ){
6415d607a6eSdan       memcpy(aOut, a2, n2);
6425d607a6eSdan       aOut += n2;
6435d607a6eSdan     }else{
6445d607a6eSdan       memcpy(aOut, a1, n1);
6455d607a6eSdan       aOut += n1;
6465d607a6eSdan     }
6475d607a6eSdan     a1 += n1;
6485d607a6eSdan     a2 += n2;
6495d607a6eSdan   }
6505d607a6eSdan 
6515d607a6eSdan   *paOut = aOut;
6525d607a6eSdan }
6535d607a6eSdan 
654798693b2Sdan /*
655798693b2Sdan ** This is a helper function used by sessionMergeUpdate().
656798693b2Sdan **
657798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value
658798693b2Sdan ** within a change record. Before it returns, both have been advanced so
659798693b2Sdan ** as to point to the next value in the record.
660798693b2Sdan **
661798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e.
662fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo
663798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the
664798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal
665798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points
666fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words:
667fa29ecc4Sdan **
668fa29ecc4Sdan **   if( *paTwo is valid ) return *paTwo;
669fa29ecc4Sdan **   return *paOne;
670fa29ecc4Sdan **
671798693b2Sdan */
6725d607a6eSdan static u8 *sessionMergeValue(
673798693b2Sdan   u8 **paOne,                     /* IN/OUT: Left-hand buffer pointer */
674798693b2Sdan   u8 **paTwo,                     /* IN/OUT: Right-hand buffer pointer */
675798693b2Sdan   int *pnVal                      /* OUT: Bytes in returned value */
6765d607a6eSdan ){
6775d607a6eSdan   u8 *a1 = *paOne;
6785d607a6eSdan   u8 *a2 = *paTwo;
6795d607a6eSdan   u8 *pRet = 0;
6805d607a6eSdan   int n1;
6815d607a6eSdan 
6825d607a6eSdan   assert( a1 );
6835d607a6eSdan   if( a2 ){
6845d607a6eSdan     int n2 = sessionSerialLen(a2);
6855d607a6eSdan     if( *a2 ){
6865d607a6eSdan       *pnVal = n2;
6875d607a6eSdan       pRet = a2;
6885d607a6eSdan     }
6895d607a6eSdan     *paTwo = &a2[n2];
6905d607a6eSdan   }
6915d607a6eSdan 
6925d607a6eSdan   n1 = sessionSerialLen(a1);
6935d607a6eSdan   if( pRet==0 ){
6945d607a6eSdan     *pnVal = n1;
6955d607a6eSdan     pRet = a1;
6965d607a6eSdan   }
6975d607a6eSdan   *paOne = &a1[n1];
6985d607a6eSdan 
6995d607a6eSdan   return pRet;
7005d607a6eSdan }
7015d607a6eSdan 
702798693b2Sdan /*
703798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes
704798693b2Sdan ** on the same row.
705798693b2Sdan */
7065d607a6eSdan static int sessionMergeUpdate(
707798693b2Sdan   u8 **paOut,                     /* IN/OUT: Pointer to output buffer */
708798693b2Sdan   SessionTable *pTab,             /* Table change pertains to */
709a71d2371Sdan   int bPatchset,                  /* True if records are patchset records */
710798693b2Sdan   u8 *aOldRecord1,                /* old.* record for first change */
711798693b2Sdan   u8 *aOldRecord2,                /* old.* record for second change */
712798693b2Sdan   u8 *aNewRecord1,                /* new.* record for first change */
713798693b2Sdan   u8 *aNewRecord2                 /* new.* record for second change */
7145d607a6eSdan ){
7155d607a6eSdan   u8 *aOld1 = aOldRecord1;
7165d607a6eSdan   u8 *aOld2 = aOldRecord2;
7175d607a6eSdan   u8 *aNew1 = aNewRecord1;
7185d607a6eSdan   u8 *aNew2 = aNewRecord2;
7195d607a6eSdan 
7205d607a6eSdan   u8 *aOut = *paOut;
7215d607a6eSdan   int i;
72264277f4aSdan 
72364277f4aSdan   if( bPatchset==0 ){
7245d607a6eSdan     int bRequired = 0;
7255d607a6eSdan 
7265d607a6eSdan     assert( aOldRecord1 && aNewRecord1 );
7275d607a6eSdan 
7285d607a6eSdan     /* Write the old.* vector first. */
7295d607a6eSdan     for(i=0; i<pTab->nCol; i++){
7305d607a6eSdan       int nOld;
7315d607a6eSdan       u8 *aOld;
7325d607a6eSdan       int nNew;
7335d607a6eSdan       u8 *aNew;
7345d607a6eSdan 
7355d607a6eSdan       aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7365d607a6eSdan       aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
7375d607a6eSdan       if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){
7385d607a6eSdan         if( pTab->abPK[i]==0 ) bRequired = 1;
7395d607a6eSdan         memcpy(aOut, aOld, nOld);
7405d607a6eSdan         aOut += nOld;
7415d607a6eSdan       }else{
7425d607a6eSdan         *(aOut++) = '\0';
7435d607a6eSdan       }
7445d607a6eSdan     }
7455d607a6eSdan 
7465d607a6eSdan     if( !bRequired ) return 0;
74764277f4aSdan   }
7485d607a6eSdan 
7495d607a6eSdan   /* Write the new.* vector */
7505d607a6eSdan   aOld1 = aOldRecord1;
7515d607a6eSdan   aOld2 = aOldRecord2;
7525d607a6eSdan   aNew1 = aNewRecord1;
7535d607a6eSdan   aNew2 = aNewRecord2;
7545d607a6eSdan   for(i=0; i<pTab->nCol; i++){
7555d607a6eSdan     int nOld;
7565d607a6eSdan     u8 *aOld;
7575d607a6eSdan     int nNew;
7585d607a6eSdan     u8 *aNew;
7595d607a6eSdan 
7605d607a6eSdan     aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7615d607a6eSdan     aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
76264277f4aSdan     if( bPatchset==0
76364277f4aSdan      && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew)))
76464277f4aSdan     ){
7655d607a6eSdan       *(aOut++) = '\0';
7665d607a6eSdan     }else{
7675d607a6eSdan       memcpy(aOut, aNew, nNew);
7685d607a6eSdan       aOut += nNew;
7695d607a6eSdan     }
7705d607a6eSdan   }
7715d607a6eSdan 
7725d607a6eSdan   *paOut = aOut;
7735d607a6eSdan   return 1;
7745d607a6eSdan }
7755d607a6eSdan 
77677fc1d5bSdan /*
77777fc1d5bSdan ** This function is only called from within a pre-update-hook callback.
77877fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row
77977fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise
78077fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns
78177fc1d5bSdan ** false.
78277fc1d5bSdan */
78377fc1d5bSdan static int sessionPreupdateEqual(
784cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns SessionTable */
78577fc1d5bSdan   SessionTable *pTab,             /* Table associated with change */
78677fc1d5bSdan   SessionChange *pChange,         /* Change to compare to */
78777fc1d5bSdan   int op                          /* Current pre-update operation */
788e8d5648eSdan ){
78977fc1d5bSdan   int iCol;                       /* Used to iterate through columns */
79077fc1d5bSdan   u8 *a = pChange->aRecord;       /* Cursor used to scan change record */
791e8d5648eSdan 
79277fc1d5bSdan   assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
79377fc1d5bSdan   for(iCol=0; iCol<pTab->nCol; iCol++){
79477fc1d5bSdan     if( !pTab->abPK[iCol] ){
795798693b2Sdan       a += sessionSerialLen(a);
796e8d5648eSdan     }else{
7976734007dSdan       sqlite3_value *pVal;        /* Value returned by preupdate_new/old */
7986734007dSdan       int rc;                     /* Error code from preupdate_new/old */
799798693b2Sdan       int eType = *a++;           /* Type of value from change record */
8006734007dSdan 
8016734007dSdan       /* The following calls to preupdate_new() and preupdate_old() can not
8026734007dSdan       ** fail. This is because they cache their return values, and by the
8036734007dSdan       ** time control flows to here they have already been called once from
8046734007dSdan       ** within sessionPreupdateHash(). The first two asserts below verify
8056734007dSdan       ** this (that the method has already been called). */
80677fc1d5bSdan       if( op==SQLITE_INSERT ){
807cf8e9144Sdan         /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */
808cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal);
809e8d5648eSdan       }else{
810cf8e9144Sdan         /* assert( db->pPreUpdate->pUnpacked ); */
811cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal);
812e8d5648eSdan       }
8136734007dSdan       assert( rc==SQLITE_OK );
8141611e5a3Sdan       if( sqlite3_value_type(pVal)!=eType ) return 0;
815e8d5648eSdan 
81612ca0b56Sdan       /* A SessionChange object never has a NULL value in a PK column */
81712ca0b56Sdan       assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
81812ca0b56Sdan            || eType==SQLITE_BLOB    || eType==SQLITE_TEXT
81912ca0b56Sdan       );
82012ca0b56Sdan 
82112ca0b56Sdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
822e8d5648eSdan         i64 iVal = sessionGetI64(a);
823e8d5648eSdan         a += 8;
824e8d5648eSdan         if( eType==SQLITE_INTEGER ){
82577fc1d5bSdan           if( sqlite3_value_int64(pVal)!=iVal ) return 0;
826e8d5648eSdan         }else{
827e8d5648eSdan           double rVal;
828e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
829e8d5648eSdan           memcpy(&rVal, &iVal, 8);
83077fc1d5bSdan           if( sqlite3_value_double(pVal)!=rVal ) return 0;
831e8d5648eSdan         }
83212ca0b56Sdan       }else{
833e8d5648eSdan         int n;
834e8d5648eSdan         const u8 *z;
835e8d5648eSdan         a += sessionVarintGet(a, &n);
83677fc1d5bSdan         if( sqlite3_value_bytes(pVal)!=n ) return 0;
83712ca0b56Sdan         if( eType==SQLITE_TEXT ){
83812ca0b56Sdan           z = sqlite3_value_text(pVal);
83912ca0b56Sdan         }else{
84012ca0b56Sdan           z = sqlite3_value_blob(pVal);
84112ca0b56Sdan         }
84277fc1d5bSdan         if( memcmp(a, z, n) ) return 0;
843e8d5648eSdan         a += n;
844e8d5648eSdan       }
845e8d5648eSdan     }
846e8d5648eSdan   }
847e8d5648eSdan 
84877fc1d5bSdan   return 1;
8494fccf43aSdan }
8504fccf43aSdan 
8514fccf43aSdan /*
8524fccf43aSdan ** If required, grow the hash table used to store changes on table pTab
8534fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the
8544fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return
8554fccf43aSdan ** SQLITE_OK.
8564fccf43aSdan **
8574fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In
8584fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway.
8594fccf43aSdan ** Growing the hash table in this case is a performance optimization only,
8604fccf43aSdan ** it is not required for correct operation.
8614fccf43aSdan */
86264277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){
8634fccf43aSdan   if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){
8644fccf43aSdan     int i;
8654fccf43aSdan     SessionChange **apNew;
8664fccf43aSdan     int nNew = (pTab->nChange ? pTab->nChange : 128) * 2;
8674fccf43aSdan 
8684fccf43aSdan     apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew);
8694fccf43aSdan     if( apNew==0 ){
8704fccf43aSdan       if( pTab->nChange==0 ){
8714fccf43aSdan         return SQLITE_ERROR;
8724fccf43aSdan       }
8734fccf43aSdan       return SQLITE_OK;
8744fccf43aSdan     }
8754fccf43aSdan     memset(apNew, 0, sizeof(SessionChange *) * nNew);
8764fccf43aSdan 
8774fccf43aSdan     for(i=0; i<pTab->nChange; i++){
8784fccf43aSdan       SessionChange *p;
8794fccf43aSdan       SessionChange *pNext;
8804fccf43aSdan       for(p=pTab->apChange[i]; p; p=pNext){
88164277f4aSdan         int bPkOnly = (p->op==SQLITE_DELETE && bPatchset);
88264277f4aSdan         int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew);
8834fccf43aSdan         pNext = p->pNext;
8844fccf43aSdan         p->pNext = apNew[iHash];
8854fccf43aSdan         apNew[iHash] = p;
8864fccf43aSdan       }
8874fccf43aSdan     }
8884fccf43aSdan 
8894fccf43aSdan     sqlite3_free(pTab->apChange);
8904fccf43aSdan     pTab->nChange = nNew;
8914fccf43aSdan     pTab->apChange = apNew;
8924fccf43aSdan   }
8934fccf43aSdan 
8944fccf43aSdan   return SQLITE_OK;
8954fccf43aSdan }
8964fccf43aSdan 
897296c7658Sdan /*
898e8d5648eSdan ** This function queries the database for the names of the columns of table
8993f975373Sdrh ** zThis, in schema zDb.
900e8d5648eSdan **
90177fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number
90277fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a
903e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to
904e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not
905e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding
906e8d5648eSdan ** column is part of the primary key.
907e8d5648eSdan **
908e8d5648eSdan ** For example, if the table is declared as:
909e8d5648eSdan **
910e8d5648eSdan **     CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z));
911e8d5648eSdan **
91277fc1d5bSdan ** Then the four output variables are populated as follows:
913e8d5648eSdan **
91477fc1d5bSdan **     *pnCol  = 4
915e8d5648eSdan **     *pzTab  = "tbl1"
916e8d5648eSdan **     *pazCol = {"w", "x", "y", "z"}
917e8d5648eSdan **     *pabPK  = {1, 0, 0, 1}
918e8d5648eSdan **
919e8d5648eSdan ** All returned buffers are part of the same single allocation, which must
9203f975373Sdrh ** be freed using sqlite3_free() by the caller
921e8d5648eSdan */
922e8d5648eSdan static int sessionTableInfo(
923e8d5648eSdan   sqlite3 *db,                    /* Database connection */
924e8d5648eSdan   const char *zDb,                /* Name of attached database (e.g. "main") */
925e8d5648eSdan   const char *zThis,              /* Table name */
926ca62ad57Sdan   int *pnCol,                     /* OUT: number of columns */
927e8d5648eSdan   const char **pzTab,             /* OUT: Copy of zThis */
928e8d5648eSdan   const char ***pazCol,           /* OUT: Array of column names for table */
929e8d5648eSdan   u8 **pabPK                      /* OUT: Array of booleans - true for PK col */
930e8d5648eSdan ){
931e8d5648eSdan   char *zPragma;
932e8d5648eSdan   sqlite3_stmt *pStmt;
933e8d5648eSdan   int rc;
934e8d5648eSdan   int nByte;
935e8d5648eSdan   int nDbCol = 0;
936e8d5648eSdan   int nThis;
937e8d5648eSdan   int i;
93874f598b6Smistachkin   u8 *pAlloc = 0;
939db04571cSdan   char **azCol = 0;
94074f598b6Smistachkin   u8 *abPK = 0;
941e8d5648eSdan 
942db04571cSdan   assert( pazCol && pabPK );
943e8d5648eSdan 
944cfdbde21Sdrh   nThis = sqlite3Strlen30(zThis);
945614efe2bSdan   if( nThis==12 && 0==sqlite3_stricmp("sqlite_stat1", zThis) ){
94658713184Sdan     rc = sqlite3_table_column_metadata(db, zDb, zThis, 0, 0, 0, 0, 0, 0);
94758713184Sdan     if( rc==SQLITE_OK ){
948614efe2bSdan       /* For sqlite_stat1, pretend that (tbl,idx) is the PRIMARY KEY. */
949614efe2bSdan       zPragma = sqlite3_mprintf(
950614efe2bSdan           "SELECT 0, 'tbl',  '', 0, '', 1     UNION ALL "
951614efe2bSdan           "SELECT 1, 'idx',  '', 0, '', 2     UNION ALL "
952614efe2bSdan           "SELECT 2, 'stat', '', 0, '', 0"
953614efe2bSdan       );
95458713184Sdan     }else if( rc==SQLITE_ERROR ){
95558713184Sdan       zPragma = sqlite3_mprintf("");
95658713184Sdan     }else{
95758713184Sdan       return rc;
95858713184Sdan     }
959614efe2bSdan   }else{
960e8d5648eSdan     zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis);
961614efe2bSdan   }
962e8d5648eSdan   if( !zPragma ) return SQLITE_NOMEM;
963e8d5648eSdan 
964e8d5648eSdan   rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0);
965e8d5648eSdan   sqlite3_free(zPragma);
966e8d5648eSdan   if( rc!=SQLITE_OK ) return rc;
967e8d5648eSdan 
968e8d5648eSdan   nByte = nThis + 1;
969e8d5648eSdan   while( SQLITE_ROW==sqlite3_step(pStmt) ){
970e8d5648eSdan     nByte += sqlite3_column_bytes(pStmt, 1);
971e8d5648eSdan     nDbCol++;
972e8d5648eSdan   }
973e8d5648eSdan   rc = sqlite3_reset(pStmt);
974e8d5648eSdan 
975e8d5648eSdan   if( rc==SQLITE_OK ){
976e8d5648eSdan     nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1);
977e8d5648eSdan     pAlloc = sqlite3_malloc(nByte);
978e8d5648eSdan     if( pAlloc==0 ){
979e8d5648eSdan       rc = SQLITE_NOMEM;
980e8d5648eSdan     }
981e8d5648eSdan   }
982e8d5648eSdan   if( rc==SQLITE_OK ){
983e8d5648eSdan     azCol = (char **)pAlloc;
984ca62ad57Sdan     pAlloc = (u8 *)&azCol[nDbCol];
985e8d5648eSdan     abPK = (u8 *)pAlloc;
986ca62ad57Sdan     pAlloc = &abPK[nDbCol];
987e8d5648eSdan     if( pzTab ){
988e8d5648eSdan       memcpy(pAlloc, zThis, nThis+1);
989e8d5648eSdan       *pzTab = (char *)pAlloc;
990e8d5648eSdan       pAlloc += nThis+1;
991e8d5648eSdan     }
992e8d5648eSdan 
993e8d5648eSdan     i = 0;
994e8d5648eSdan     while( SQLITE_ROW==sqlite3_step(pStmt) ){
995e8d5648eSdan       int nName = sqlite3_column_bytes(pStmt, 1);
996e8d5648eSdan       const unsigned char *zName = sqlite3_column_text(pStmt, 1);
997e8d5648eSdan       if( zName==0 ) break;
998e8d5648eSdan       memcpy(pAlloc, zName, nName+1);
999e8d5648eSdan       azCol[i] = (char *)pAlloc;
1000e8d5648eSdan       pAlloc += nName+1;
1001db04571cSdan       abPK[i] = sqlite3_column_int(pStmt, 5);
1002e8d5648eSdan       i++;
1003e8d5648eSdan     }
1004e8d5648eSdan     rc = sqlite3_reset(pStmt);
1005e8d5648eSdan 
1006e8d5648eSdan   }
1007e8d5648eSdan 
1008e8d5648eSdan   /* If successful, populate the output variables. Otherwise, zero them and
1009e8d5648eSdan   ** free any allocation made. An error code will be returned in this case.
1010e8d5648eSdan   */
1011e8d5648eSdan   if( rc==SQLITE_OK ){
1012db04571cSdan     *pazCol = (const char **)azCol;
1013db04571cSdan     *pabPK = abPK;
1014ca62ad57Sdan     *pnCol = nDbCol;
1015e8d5648eSdan   }else{
1016db04571cSdan     *pazCol = 0;
1017db04571cSdan     *pabPK = 0;
1018ca62ad57Sdan     *pnCol = 0;
1019e8d5648eSdan     if( pzTab ) *pzTab = 0;
1020db04571cSdan     sqlite3_free(azCol);
1021e8d5648eSdan   }
1022e8d5648eSdan   sqlite3_finalize(pStmt);
1023e8d5648eSdan   return rc;
1024e8d5648eSdan }
1025e8d5648eSdan 
1026e8d5648eSdan /*
1027296c7658Sdan ** This function is only called from within a pre-update handler for a
1028296c7658Sdan ** write to table pTab, part of session pSession. If this is the first
10296dc29e60Sdan ** write to this table, initalize the SessionTable.nCol, azCol[] and
10306dc29e60Sdan ** abPK[] arrays accordingly.
1031296c7658Sdan **
10326dc29e60Sdan ** If an error occurs, an error code is stored in sqlite3_session.rc and
10336dc29e60Sdan ** non-zero returned. Or, if no error occurs but the table has no primary
10346dc29e60Sdan ** key, sqlite3_session.rc is left set to SQLITE_OK and non-zero returned to
10356dc29e60Sdan ** indicate that updates on this table should be ignored. SessionTable.abPK
10366dc29e60Sdan ** is set to NULL in this case.
1037296c7658Sdan */
10384fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){
10394fccf43aSdan   if( pTab->nCol==0 ){
10406dc29e60Sdan     u8 *abPK;
1041e8d5648eSdan     assert( pTab->azCol==0 || pTab->abPK==0 );
1042e8d5648eSdan     pSession->rc = sessionTableInfo(pSession->db, pSession->zDb,
10436dc29e60Sdan         pTab->zName, &pTab->nCol, 0, &pTab->azCol, &abPK
1044e8d5648eSdan     );
10456dc29e60Sdan     if( pSession->rc==SQLITE_OK ){
10466dc29e60Sdan       int i;
10476dc29e60Sdan       for(i=0; i<pTab->nCol; i++){
10486dc29e60Sdan         if( abPK[i] ){
10496dc29e60Sdan           pTab->abPK = abPK;
10506dc29e60Sdan           break;
1051ca62ad57Sdan         }
10524fccf43aSdan       }
10533739f298Sdan       if( 0==sqlite3_stricmp("sqlite_stat1", pTab->zName) ){
10543739f298Sdan         pTab->bStat1 = 1;
10553739f298Sdan       }
10566dc29e60Sdan     }
10576dc29e60Sdan   }
10586dc29e60Sdan   return (pSession->rc || pTab->abPK==0);
1059e8d5648eSdan }
1060e8d5648eSdan 
106177fc1d5bSdan /*
10621611e5a3Sdan ** Versions of the four methods in object SessionHook for use with the
10631611e5a3Sdan ** sqlite_stat1 table. The purpose of this is to substitute a zero-length
10641611e5a3Sdan ** blob each time a NULL value is read from the "idx" column of the
10651611e5a3Sdan ** sqlite_stat1 table.
10661611e5a3Sdan */
10671611e5a3Sdan typedef struct SessionStat1Ctx SessionStat1Ctx;
10681611e5a3Sdan struct SessionStat1Ctx {
10691611e5a3Sdan   SessionHook hook;
10701611e5a3Sdan   sqlite3_session *pSession;
10711611e5a3Sdan };
10721611e5a3Sdan static int sessionStat1Old(void *pCtx, int iCol, sqlite3_value **ppVal){
10731611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10741611e5a3Sdan   sqlite3_value *pVal = 0;
10751611e5a3Sdan   int rc = p->hook.xOld(p->hook.pCtx, iCol, &pVal);
10761611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
10771611e5a3Sdan     pVal = p->pSession->pZeroBlob;
10781611e5a3Sdan   }
10791611e5a3Sdan   *ppVal = pVal;
10801611e5a3Sdan   return rc;
10811611e5a3Sdan }
10821611e5a3Sdan static int sessionStat1New(void *pCtx, int iCol, sqlite3_value **ppVal){
10831611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10841611e5a3Sdan   sqlite3_value *pVal = 0;
10851611e5a3Sdan   int rc = p->hook.xNew(p->hook.pCtx, iCol, &pVal);
10861611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
10871611e5a3Sdan     pVal = p->pSession->pZeroBlob;
10881611e5a3Sdan   }
10891611e5a3Sdan   *ppVal = pVal;
10901611e5a3Sdan   return rc;
10911611e5a3Sdan }
10921611e5a3Sdan static int sessionStat1Count(void *pCtx){
10931611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10941611e5a3Sdan   return p->hook.xCount(p->hook.pCtx);
10951611e5a3Sdan }
10961611e5a3Sdan static int sessionStat1Depth(void *pCtx){
10971611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10981611e5a3Sdan   return p->hook.xDepth(p->hook.pCtx);
10991611e5a3Sdan }
11001611e5a3Sdan 
11011611e5a3Sdan 
11021611e5a3Sdan /*
110377fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a
110477fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change
110577fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument.
110677fc1d5bSdan **
110777fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added
110877fc1d5bSdan ** to the changed-rows hash table associated with table pTab.
110977fc1d5bSdan */
1110e8d5648eSdan static void sessionPreupdateOneChange(
111177fc1d5bSdan   int op,                         /* One of SQLITE_UPDATE, INSERT, DELETE */
111277fc1d5bSdan   sqlite3_session *pSession,      /* Session object pTab is attached to */
111377fc1d5bSdan   SessionTable *pTab              /* Table that change applies to */
1114e8d5648eSdan ){
1115e8d5648eSdan   int iHash;
1116cf8e9144Sdan   int bNull = 0;
1117e8d5648eSdan   int rc = SQLITE_OK;
1118*fa3d7855Smistachkin   SessionStat1Ctx stat1 = {0};
1119e8d5648eSdan 
1120e8d5648eSdan   if( pSession->rc ) return;
1121e8d5648eSdan 
1122e8d5648eSdan   /* Load table details if required */
1123e8d5648eSdan   if( sessionInitTable(pSession, pTab) ) return;
1124e8d5648eSdan 
11256dc29e60Sdan   /* Check the number of columns in this xPreUpdate call matches the
11266dc29e60Sdan   ** number of columns in the table.  */
11276dc29e60Sdan   if( pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx) ){
11286dc29e60Sdan     pSession->rc = SQLITE_SCHEMA;
11296dc29e60Sdan     return;
11306dc29e60Sdan   }
11316dc29e60Sdan 
1132e8d5648eSdan   /* Grow the hash table if required */
113364277f4aSdan   if( sessionGrowHash(0, pTab) ){
11345d607a6eSdan     pSession->rc = SQLITE_NOMEM;
11355d607a6eSdan     return;
11365d607a6eSdan   }
1137e8d5648eSdan 
11381611e5a3Sdan   if( pTab->bStat1 ){
11391611e5a3Sdan     stat1.hook = pSession->hook;
11401611e5a3Sdan     stat1.pSession = pSession;
11411611e5a3Sdan     pSession->hook.pCtx = (void*)&stat1;
11421611e5a3Sdan     pSession->hook.xNew = sessionStat1New;
11431611e5a3Sdan     pSession->hook.xOld = sessionStat1Old;
11441611e5a3Sdan     pSession->hook.xCount = sessionStat1Count;
11451611e5a3Sdan     pSession->hook.xDepth = sessionStat1Depth;
11461611e5a3Sdan     if( pSession->pZeroBlob==0 ){
11471611e5a3Sdan       sqlite3_value *p = sqlite3ValueNew(0);
11481611e5a3Sdan       if( p==0 ){
11491611e5a3Sdan         rc = SQLITE_NOMEM;
11501611e5a3Sdan         goto error_out;
11511611e5a3Sdan       }
11521611e5a3Sdan       sqlite3ValueSetStr(p, 0, "", 0, SQLITE_STATIC);
11531611e5a3Sdan       pSession->pZeroBlob = p;
11541611e5a3Sdan     }
11551611e5a3Sdan   }
11561611e5a3Sdan 
115780fe2d93Sdan   /* Calculate the hash-key for this change. If the primary key of the row
115880fe2d93Sdan   ** includes a NULL value, exit early. Such changes are ignored by the
115980fe2d93Sdan   ** session module. */
1160cf8e9144Sdan   rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull);
116180fe2d93Sdan   if( rc!=SQLITE_OK ) goto error_out;
116280fe2d93Sdan 
11631611e5a3Sdan   if( bNull==0 ){
116480fe2d93Sdan     /* Search the hash table for an existing record for this row. */
1165b4480e94Sdan     SessionChange *pC;
11666734007dSdan     for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){
1167cf8e9144Sdan       if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break;
1168e8d5648eSdan     }
116980fe2d93Sdan 
1170e8d5648eSdan     if( pC==0 ){
1171e8d5648eSdan       /* Create a new change object containing all the old values (if
1172e8d5648eSdan       ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK
1173e8d5648eSdan       ** values (if this is an INSERT). */
1174b4480e94Sdan       SessionChange *pChange; /* New change object */
1175e8d5648eSdan       int nByte;              /* Number of bytes to allocate */
1176e8d5648eSdan       int i;                  /* Used to iterate through columns */
1177e8d5648eSdan 
1178b4480e94Sdan       assert( rc==SQLITE_OK );
1179e8d5648eSdan       pTab->nEntry++;
1180e8d5648eSdan 
1181e8d5648eSdan       /* Figure out how large an allocation is required */
1182e8d5648eSdan       nByte = sizeof(SessionChange);
118380fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1184e8d5648eSdan         sqlite3_value *p = 0;
1185e8d5648eSdan         if( op!=SQLITE_INSERT ){
1186cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p);
118780fe2d93Sdan           assert( trc==SQLITE_OK );
118880fe2d93Sdan         }else if( pTab->abPK[i] ){
1189cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p);
119080fe2d93Sdan           assert( trc==SQLITE_OK );
1191e8d5648eSdan         }
119280fe2d93Sdan 
119380fe2d93Sdan         /* This may fail if SQLite value p contains a utf-16 string that must
119480fe2d93Sdan         ** be converted to utf-8 and an OOM error occurs while doing so. */
1195e8d5648eSdan         rc = sessionSerializeValue(0, p, &nByte);
119680fe2d93Sdan         if( rc!=SQLITE_OK ) goto error_out;
1197e8d5648eSdan       }
1198e8d5648eSdan 
1199e8d5648eSdan       /* Allocate the change object */
1200e8d5648eSdan       pChange = (SessionChange *)sqlite3_malloc(nByte);
1201e8d5648eSdan       if( !pChange ){
1202e8d5648eSdan         rc = SQLITE_NOMEM;
120380fe2d93Sdan         goto error_out;
1204e8d5648eSdan       }else{
1205e8d5648eSdan         memset(pChange, 0, sizeof(SessionChange));
1206e8d5648eSdan         pChange->aRecord = (u8 *)&pChange[1];
1207e8d5648eSdan       }
1208e8d5648eSdan 
120980fe2d93Sdan       /* Populate the change object. None of the preupdate_old(),
121080fe2d93Sdan       ** preupdate_new() or SerializeValue() calls below may fail as all
121180fe2d93Sdan       ** required values and encodings have already been cached in memory.
121280fe2d93Sdan       ** It is not possible for an OOM to occur in this block. */
1213e8d5648eSdan       nByte = 0;
121480fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1215e8d5648eSdan         sqlite3_value *p = 0;
1216e8d5648eSdan         if( op!=SQLITE_INSERT ){
1217cf8e9144Sdan           pSession->hook.xOld(pSession->hook.pCtx, i, &p);
121880fe2d93Sdan         }else if( pTab->abPK[i] ){
1219cf8e9144Sdan           pSession->hook.xNew(pSession->hook.pCtx, i, &p);
1220e8d5648eSdan         }
122180fe2d93Sdan         sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte);
1222e8d5648eSdan       }
122380fe2d93Sdan 
122480fe2d93Sdan       /* Add the change to the hash-table */
1225cf8e9144Sdan       if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){
1226b4480e94Sdan         pChange->bIndirect = 1;
1227b4480e94Sdan       }
122812ca0b56Sdan       pChange->nRecord = nByte;
1229798693b2Sdan       pChange->op = op;
1230e8d5648eSdan       pChange->pNext = pTab->apChange[iHash];
1231e8d5648eSdan       pTab->apChange[iHash] = pChange;
123280fe2d93Sdan 
123380fe2d93Sdan     }else if( pC->bIndirect ){
1234b4480e94Sdan       /* If the existing change is considered "indirect", but this current
1235b4480e94Sdan       ** change is "direct", mark the change object as direct. */
1236cf8e9144Sdan       if( pSession->hook.xDepth(pSession->hook.pCtx)==0
1237cf8e9144Sdan        && pSession->bIndirect==0
1238cf8e9144Sdan       ){
1239b4480e94Sdan         pC->bIndirect = 0;
1240b4480e94Sdan       }
1241e8d5648eSdan     }
12424fccf43aSdan   }
124312ca0b56Sdan 
124412ca0b56Sdan   /* If an error has occurred, mark the session object as failed. */
124580fe2d93Sdan  error_out:
12461611e5a3Sdan   if( pTab->bStat1 ){
12471611e5a3Sdan     pSession->hook = stat1.hook;
12481611e5a3Sdan   }
124912ca0b56Sdan   if( rc!=SQLITE_OK ){
125012ca0b56Sdan     pSession->rc = rc;
125112ca0b56Sdan   }
125227453faeSdan }
12534fccf43aSdan 
1254cf8e9144Sdan static int sessionFindTable(
1255cf8e9144Sdan   sqlite3_session *pSession,
1256cf8e9144Sdan   const char *zName,
1257cf8e9144Sdan   SessionTable **ppTab
1258cf8e9144Sdan ){
1259cf8e9144Sdan   int rc = SQLITE_OK;
1260cf8e9144Sdan   int nName = sqlite3Strlen30(zName);
1261cf8e9144Sdan   SessionTable *pRet;
1262cf8e9144Sdan 
1263cf8e9144Sdan   /* Search for an existing table */
1264cf8e9144Sdan   for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){
1265cf8e9144Sdan     if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break;
1266cf8e9144Sdan   }
1267cf8e9144Sdan 
1268cf8e9144Sdan   if( pRet==0 && pSession->bAutoAttach ){
1269cf8e9144Sdan     /* If there is a table-filter configured, invoke it. If it returns 0,
1270cf8e9144Sdan     ** do not automatically add the new table. */
1271cf8e9144Sdan     if( pSession->xTableFilter==0
1272cf8e9144Sdan      || pSession->xTableFilter(pSession->pFilterCtx, zName)
1273cf8e9144Sdan     ){
1274cf8e9144Sdan       rc = sqlite3session_attach(pSession, zName);
1275cf8e9144Sdan       if( rc==SQLITE_OK ){
12766c39e6a8Sdan         for(pRet=pSession->pTable; pRet->pNext; pRet=pRet->pNext);
1277cf8e9144Sdan         assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) );
1278cf8e9144Sdan       }
1279cf8e9144Sdan     }
1280cf8e9144Sdan   }
1281cf8e9144Sdan 
1282cf8e9144Sdan   assert( rc==SQLITE_OK || pRet==0 );
1283cf8e9144Sdan   *ppTab = pRet;
1284cf8e9144Sdan   return rc;
1285cf8e9144Sdan }
1286cf8e9144Sdan 
12874fccf43aSdan /*
12884fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases.
12894fccf43aSdan */
12904fccf43aSdan static void xPreUpdate(
12914fccf43aSdan   void *pCtx,                     /* Copy of third arg to preupdate_hook() */
12924fccf43aSdan   sqlite3 *db,                    /* Database handle */
12934fccf43aSdan   int op,                         /* SQLITE_UPDATE, DELETE or INSERT */
12944fccf43aSdan   char const *zDb,                /* Database name */
12954fccf43aSdan   char const *zName,              /* Table name */
12964fccf43aSdan   sqlite3_int64 iKey1,            /* Rowid of row about to be deleted/updated */
12974fccf43aSdan   sqlite3_int64 iKey2             /* New rowid value (for a rowid UPDATE) */
12984fccf43aSdan ){
12994fccf43aSdan   sqlite3_session *pSession;
1300cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb);
13014fccf43aSdan 
13024c220252Sdan   assert( sqlite3_mutex_held(db->mutex) );
13034c220252Sdan 
13044fccf43aSdan   for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){
13054fccf43aSdan     SessionTable *pTab;
1306296c7658Sdan 
1307e8d5648eSdan     /* If this session is attached to a different database ("main", "temp"
1308e8d5648eSdan     ** etc.), or if it is not currently enabled, there is nothing to do. Skip
1309e8d5648eSdan     ** to the next session object attached to this database. */
1310296c7658Sdan     if( pSession->bEnable==0 ) continue;
13114fccf43aSdan     if( pSession->rc ) continue;
13124fccf43aSdan     if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue;
1313296c7658Sdan 
1314cf8e9144Sdan     pSession->rc = sessionFindTable(pSession, zName, &pTab);
1315cf8e9144Sdan     if( pTab ){
1316cf8e9144Sdan       assert( pSession->rc==SQLITE_OK );
1317e8d5648eSdan       sessionPreupdateOneChange(op, pSession, pTab);
1318e8d5648eSdan       if( op==SQLITE_UPDATE ){
1319e8d5648eSdan         sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab);
13204fccf43aSdan       }
13214fccf43aSdan     }
13224fccf43aSdan   }
13234fccf43aSdan }
1324cf8e9144Sdan 
1325cf8e9144Sdan /*
1326cf8e9144Sdan ** The pre-update hook implementations.
1327cf8e9144Sdan */
1328cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1329cf8e9144Sdan   return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal);
1330cf8e9144Sdan }
1331cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1332cf8e9144Sdan   return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal);
1333cf8e9144Sdan }
1334cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){
1335cf8e9144Sdan   return sqlite3_preupdate_count((sqlite3*)pCtx);
1336cf8e9144Sdan }
1337cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){
1338cf8e9144Sdan   return sqlite3_preupdate_depth((sqlite3*)pCtx);
1339cf8e9144Sdan }
1340cf8e9144Sdan 
1341cf8e9144Sdan /*
1342cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only
1343cf8e9144Sdan ** argument.
1344cf8e9144Sdan */
1345cf8e9144Sdan static void sessionPreupdateHooks(
1346cf8e9144Sdan   sqlite3_session *pSession
1347cf8e9144Sdan ){
1348cf8e9144Sdan   pSession->hook.pCtx = (void*)pSession->db;
1349cf8e9144Sdan   pSession->hook.xOld = sessionPreupdateOld;
1350cf8e9144Sdan   pSession->hook.xNew = sessionPreupdateNew;
1351cf8e9144Sdan   pSession->hook.xCount = sessionPreupdateCount;
1352cf8e9144Sdan   pSession->hook.xDepth = sessionPreupdateDepth;
1353cf8e9144Sdan }
1354cf8e9144Sdan 
1355cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx;
1356cf8e9144Sdan struct SessionDiffCtx {
1357cf8e9144Sdan   sqlite3_stmt *pStmt;
1358cf8e9144Sdan   int nOldOff;
1359cf8e9144Sdan };
1360cf8e9144Sdan 
1361cf8e9144Sdan /*
1362cf8e9144Sdan ** The diff hook implementations.
1363cf8e9144Sdan */
1364cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1365cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1366cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff);
1367cf8e9144Sdan   return SQLITE_OK;
1368cf8e9144Sdan }
1369cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1370cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1371cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal);
1372cf8e9144Sdan    return SQLITE_OK;
1373cf8e9144Sdan }
1374cf8e9144Sdan static int sessionDiffCount(void *pCtx){
1375cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1376cf8e9144Sdan   return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt);
1377cf8e9144Sdan }
1378cf8e9144Sdan static int sessionDiffDepth(void *pCtx){
1379cf8e9144Sdan   return 0;
1380cf8e9144Sdan }
1381cf8e9144Sdan 
1382cf8e9144Sdan /*
1383cf8e9144Sdan ** Install the diff hooks on the session object passed as the only
1384cf8e9144Sdan ** argument.
1385cf8e9144Sdan */
1386cf8e9144Sdan static void sessionDiffHooks(
1387cf8e9144Sdan   sqlite3_session *pSession,
1388cf8e9144Sdan   SessionDiffCtx *pDiffCtx
1389cf8e9144Sdan ){
1390cf8e9144Sdan   pSession->hook.pCtx = (void*)pDiffCtx;
1391cf8e9144Sdan   pSession->hook.xOld = sessionDiffOld;
1392cf8e9144Sdan   pSession->hook.xNew = sessionDiffNew;
1393cf8e9144Sdan   pSession->hook.xCount = sessionDiffCount;
1394cf8e9144Sdan   pSession->hook.xDepth = sessionDiffDepth;
1395cf8e9144Sdan }
1396cf8e9144Sdan 
1397cf8e9144Sdan static char *sessionExprComparePK(
1398cf8e9144Sdan   int nCol,
1399cf8e9144Sdan   const char *zDb1, const char *zDb2,
1400cf8e9144Sdan   const char *zTab,
1401cf8e9144Sdan   const char **azCol, u8 *abPK
1402cf8e9144Sdan ){
1403cf8e9144Sdan   int i;
1404cf8e9144Sdan   const char *zSep = "";
1405cf8e9144Sdan   char *zRet = 0;
1406cf8e9144Sdan 
1407cf8e9144Sdan   for(i=0; i<nCol; i++){
1408cf8e9144Sdan     if( abPK[i] ){
1409cf8e9144Sdan       zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"",
1410cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1411cf8e9144Sdan       );
1412cf8e9144Sdan       zSep = " AND ";
1413cf8e9144Sdan       if( zRet==0 ) break;
1414cf8e9144Sdan     }
1415cf8e9144Sdan   }
1416cf8e9144Sdan 
1417cf8e9144Sdan   return zRet;
1418cf8e9144Sdan }
1419cf8e9144Sdan 
1420cf8e9144Sdan static char *sessionExprCompareOther(
1421cf8e9144Sdan   int nCol,
1422cf8e9144Sdan   const char *zDb1, const char *zDb2,
1423cf8e9144Sdan   const char *zTab,
1424cf8e9144Sdan   const char **azCol, u8 *abPK
1425cf8e9144Sdan ){
1426cf8e9144Sdan   int i;
1427cf8e9144Sdan   const char *zSep = "";
1428cf8e9144Sdan   char *zRet = 0;
1429cf8e9144Sdan   int bHave = 0;
1430cf8e9144Sdan 
1431cf8e9144Sdan   for(i=0; i<nCol; i++){
1432cf8e9144Sdan     if( abPK[i]==0 ){
1433cf8e9144Sdan       bHave = 1;
1434cf8e9144Sdan       zRet = sqlite3_mprintf(
1435cf8e9144Sdan           "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"",
1436cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1437cf8e9144Sdan       );
1438cf8e9144Sdan       zSep = " OR ";
1439cf8e9144Sdan       if( zRet==0 ) break;
1440cf8e9144Sdan     }
1441cf8e9144Sdan   }
1442cf8e9144Sdan 
1443cf8e9144Sdan   if( bHave==0 ){
1444cf8e9144Sdan     assert( zRet==0 );
1445cf8e9144Sdan     zRet = sqlite3_mprintf("0");
1446cf8e9144Sdan   }
1447cf8e9144Sdan 
1448cf8e9144Sdan   return zRet;
1449cf8e9144Sdan }
1450cf8e9144Sdan 
1451cf8e9144Sdan static char *sessionSelectFindNew(
1452cf8e9144Sdan   int nCol,
1453cf8e9144Sdan   const char *zDb1,      /* Pick rows in this db only */
1454cf8e9144Sdan   const char *zDb2,      /* But not in this one */
1455cf8e9144Sdan   const char *zTbl,      /* Table name */
1456cf8e9144Sdan   const char *zExpr
1457cf8e9144Sdan ){
1458cf8e9144Sdan   char *zRet = sqlite3_mprintf(
1459cf8e9144Sdan       "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS ("
1460cf8e9144Sdan       "  SELECT 1 FROM \"%w\".\"%w\" WHERE %s"
1461cf8e9144Sdan       ")",
1462cf8e9144Sdan       zDb1, zTbl, zDb2, zTbl, zExpr
1463cf8e9144Sdan   );
1464cf8e9144Sdan   return zRet;
1465cf8e9144Sdan }
1466cf8e9144Sdan 
1467cf8e9144Sdan static int sessionDiffFindNew(
1468cf8e9144Sdan   int op,
1469cf8e9144Sdan   sqlite3_session *pSession,
1470cf8e9144Sdan   SessionTable *pTab,
1471cf8e9144Sdan   const char *zDb1,
1472cf8e9144Sdan   const char *zDb2,
1473cf8e9144Sdan   char *zExpr
1474cf8e9144Sdan ){
1475cf8e9144Sdan   int rc = SQLITE_OK;
1476cf8e9144Sdan   char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr);
1477cf8e9144Sdan 
1478cf8e9144Sdan   if( zStmt==0 ){
1479cf8e9144Sdan     rc = SQLITE_NOMEM;
1480cf8e9144Sdan   }else{
1481cf8e9144Sdan     sqlite3_stmt *pStmt;
1482cf8e9144Sdan     rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1483cf8e9144Sdan     if( rc==SQLITE_OK ){
1484cf8e9144Sdan       SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1485cf8e9144Sdan       pDiffCtx->pStmt = pStmt;
1486cf8e9144Sdan       pDiffCtx->nOldOff = 0;
1487cf8e9144Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
1488cf8e9144Sdan         sessionPreupdateOneChange(op, pSession, pTab);
1489cf8e9144Sdan       }
1490cf8e9144Sdan       rc = sqlite3_finalize(pStmt);
1491cf8e9144Sdan     }
1492cf8e9144Sdan     sqlite3_free(zStmt);
1493cf8e9144Sdan   }
1494cf8e9144Sdan 
1495cf8e9144Sdan   return rc;
1496cf8e9144Sdan }
1497cf8e9144Sdan 
1498cf8e9144Sdan static int sessionDiffFindModified(
1499cf8e9144Sdan   sqlite3_session *pSession,
1500cf8e9144Sdan   SessionTable *pTab,
1501cf8e9144Sdan   const char *zFrom,
1502cf8e9144Sdan   const char *zExpr
1503cf8e9144Sdan ){
1504cf8e9144Sdan   int rc = SQLITE_OK;
1505cf8e9144Sdan 
1506cf8e9144Sdan   char *zExpr2 = sessionExprCompareOther(pTab->nCol,
1507cf8e9144Sdan       pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK
1508cf8e9144Sdan   );
1509cf8e9144Sdan   if( zExpr2==0 ){
1510cf8e9144Sdan     rc = SQLITE_NOMEM;
1511cf8e9144Sdan   }else{
1512cf8e9144Sdan     char *zStmt = sqlite3_mprintf(
1513dd009f83Sdan         "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)",
1514cf8e9144Sdan         pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2
1515cf8e9144Sdan     );
1516cf8e9144Sdan     if( zStmt==0 ){
1517cf8e9144Sdan       rc = SQLITE_NOMEM;
1518cf8e9144Sdan     }else{
1519cf8e9144Sdan       sqlite3_stmt *pStmt;
1520cf8e9144Sdan       rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1521cf8e9144Sdan 
1522cf8e9144Sdan       if( rc==SQLITE_OK ){
1523cf8e9144Sdan         SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1524cf8e9144Sdan         pDiffCtx->pStmt = pStmt;
1525cf8e9144Sdan         pDiffCtx->nOldOff = pTab->nCol;
1526cf8e9144Sdan         while( SQLITE_ROW==sqlite3_step(pStmt) ){
1527cf8e9144Sdan           sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab);
1528cf8e9144Sdan         }
1529cf8e9144Sdan         rc = sqlite3_finalize(pStmt);
1530cf8e9144Sdan       }
1531cf8e9144Sdan       sqlite3_free(zStmt);
1532cf8e9144Sdan     }
1533cf8e9144Sdan   }
1534cf8e9144Sdan 
1535cf8e9144Sdan   return rc;
1536cf8e9144Sdan }
1537cf8e9144Sdan 
1538cf8e9144Sdan int sqlite3session_diff(
1539cf8e9144Sdan   sqlite3_session *pSession,
1540cf8e9144Sdan   const char *zFrom,
1541cf8e9144Sdan   const char *zTbl,
1542cf8e9144Sdan   char **pzErrMsg
1543cf8e9144Sdan ){
1544cf8e9144Sdan   const char *zDb = pSession->zDb;
1545cf8e9144Sdan   int rc = pSession->rc;
1546cf8e9144Sdan   SessionDiffCtx d;
1547cf8e9144Sdan 
1548cf8e9144Sdan   memset(&d, 0, sizeof(d));
1549cf8e9144Sdan   sessionDiffHooks(pSession, &d);
1550cf8e9144Sdan 
155110dc553cSdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1552cf8e9144Sdan   if( pzErrMsg ) *pzErrMsg = 0;
1553cf8e9144Sdan   if( rc==SQLITE_OK ){
1554cf8e9144Sdan     char *zExpr = 0;
1555cf8e9144Sdan     sqlite3 *db = pSession->db;
1556cf8e9144Sdan     SessionTable *pTo;            /* Table zTbl */
1557cf8e9144Sdan 
1558cf8e9144Sdan     /* Locate and if necessary initialize the target table object */
1559cf8e9144Sdan     rc = sessionFindTable(pSession, zTbl, &pTo);
1560cf8e9144Sdan     if( pTo==0 ) goto diff_out;
15616dc29e60Sdan     if( sessionInitTable(pSession, pTo) ){
15626dc29e60Sdan       rc = pSession->rc;
15636dc29e60Sdan       goto diff_out;
1564cf8e9144Sdan     }
1565cf8e9144Sdan 
1566cf8e9144Sdan     /* Check the table schemas match */
1567cf8e9144Sdan     if( rc==SQLITE_OK ){
15684cc923e3Sdan       int bHasPk = 0;
1569b9db9099Sdan       int bMismatch = 0;
1570cf8e9144Sdan       int nCol;                   /* Columns in zFrom.zTbl */
1571cf8e9144Sdan       u8 *abPK;
1572cf8e9144Sdan       const char **azCol = 0;
1573cf8e9144Sdan       rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK);
1574cf8e9144Sdan       if( rc==SQLITE_OK ){
15754cc923e3Sdan         if( pTo->nCol!=nCol ){
1576cf8e9144Sdan           bMismatch = 1;
1577cf8e9144Sdan         }else{
1578cf8e9144Sdan           int i;
1579cf8e9144Sdan           for(i=0; i<nCol; i++){
15804cc923e3Sdan             if( pTo->abPK[i]!=abPK[i] ) bMismatch = 1;
1581cf8e9144Sdan             if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1;
15824cc923e3Sdan             if( abPK[i] ) bHasPk = 1;
1583cf8e9144Sdan           }
1584cf8e9144Sdan         }
1585cf8e9144Sdan       }
1586dbbd8160Sdrh       sqlite3_free((char*)azCol);
1587b9db9099Sdan       if( bMismatch ){
1588b9db9099Sdan         *pzErrMsg = sqlite3_mprintf("table schemas do not match");
1589b9db9099Sdan         rc = SQLITE_SCHEMA;
1590b9db9099Sdan       }
1591b9db9099Sdan       if( bHasPk==0 ){
1592b9db9099Sdan         /* Ignore tables with no primary keys */
1593b9db9099Sdan         goto diff_out;
1594b9db9099Sdan       }
1595cf8e9144Sdan     }
1596cf8e9144Sdan 
1597cf8e9144Sdan     if( rc==SQLITE_OK ){
1598cf8e9144Sdan       zExpr = sessionExprComparePK(pTo->nCol,
1599cf8e9144Sdan           zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK
1600cf8e9144Sdan       );
1601cf8e9144Sdan     }
1602cf8e9144Sdan 
1603cf8e9144Sdan     /* Find new rows */
1604cf8e9144Sdan     if( rc==SQLITE_OK ){
1605cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr);
1606cf8e9144Sdan     }
1607cf8e9144Sdan 
1608cf8e9144Sdan     /* Find old rows */
1609cf8e9144Sdan     if( rc==SQLITE_OK ){
1610cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr);
1611cf8e9144Sdan     }
1612cf8e9144Sdan 
1613cf8e9144Sdan     /* Find modified rows */
1614cf8e9144Sdan     if( rc==SQLITE_OK ){
1615cf8e9144Sdan       rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr);
1616cf8e9144Sdan     }
1617cf8e9144Sdan 
1618cf8e9144Sdan     sqlite3_free(zExpr);
1619cf8e9144Sdan   }
1620cf8e9144Sdan 
1621cf8e9144Sdan  diff_out:
1622cf8e9144Sdan   sessionPreupdateHooks(pSession);
162310dc553cSdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
1624cf8e9144Sdan   return rc;
1625296c7658Sdan }
16264fccf43aSdan 
16274fccf43aSdan /*
16284fccf43aSdan ** Create a session object. This session object will record changes to
16294fccf43aSdan ** database zDb attached to connection db.
16304fccf43aSdan */
16314fccf43aSdan int sqlite3session_create(
16324fccf43aSdan   sqlite3 *db,                    /* Database handle */
16334fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
16344fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
16354fccf43aSdan ){
1636296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1637296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1638cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
16394fccf43aSdan 
1640296c7658Sdan   /* Zero the output value in case an error occurs. */
16414fccf43aSdan   *ppSession = 0;
16424fccf43aSdan 
16434fccf43aSdan   /* Allocate and populate the new session object. */
16444fccf43aSdan   pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1);
16454fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
16464fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
16474fccf43aSdan   pNew->db = db;
16484fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1649296c7658Sdan   pNew->bEnable = 1;
16504fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
1651cf8e9144Sdan   sessionPreupdateHooks(pNew);
16524fccf43aSdan 
16534fccf43aSdan   /* Add the new session object to the linked list of session objects
16544fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
16554fccf43aSdan   ** handle mutex.  */
16564fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
16574fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
16584fccf43aSdan   pNew->pNext = pOld;
16594fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
16604fccf43aSdan 
16614fccf43aSdan   *ppSession = pNew;
16624fccf43aSdan   return SQLITE_OK;
16634fccf43aSdan }
16644fccf43aSdan 
166577fc1d5bSdan /*
166677fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
166777fc1d5bSdan ** of the changed-rows hash tables are also deleted.
166877fc1d5bSdan */
16691ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
16705d607a6eSdan   SessionTable *pNext;
16715d607a6eSdan   SessionTable *pTab;
16725d607a6eSdan 
16735d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
16745d607a6eSdan     int i;
16755d607a6eSdan     pNext = pTab->pNext;
16765d607a6eSdan     for(i=0; i<pTab->nChange; i++){
16775d607a6eSdan       SessionChange *p;
167802d436b1Smistachkin       SessionChange *pNextChange;
167902d436b1Smistachkin       for(p=pTab->apChange[i]; p; p=pNextChange){
168002d436b1Smistachkin         pNextChange = p->pNext;
16815d607a6eSdan         sqlite3_free(p);
16825d607a6eSdan       }
16835d607a6eSdan     }
16845d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
16855d607a6eSdan     sqlite3_free(pTab->apChange);
16865d607a6eSdan     sqlite3_free(pTab);
16875d607a6eSdan   }
16885d607a6eSdan }
16895d607a6eSdan 
16904fccf43aSdan /*
16914fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
16924fccf43aSdan */
16934fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
16944fccf43aSdan   sqlite3 *db = pSession->db;
16954fccf43aSdan   sqlite3_session *pHead;
16964fccf43aSdan   sqlite3_session **pp;
16974fccf43aSdan 
1698296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1699296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
17004fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
17014fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
170250d348b1Sdrh   for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){
170350d348b1Sdrh     if( (*pp)==pSession ){
17044fccf43aSdan       *pp = (*pp)->pNext;
17054fccf43aSdan       if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead);
170650d348b1Sdrh       break;
170750d348b1Sdrh     }
170850d348b1Sdrh   }
17094fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
17101611e5a3Sdan   sqlite3ValueFree(pSession->pZeroBlob);
17114fccf43aSdan 
1712296c7658Sdan   /* Delete all attached table objects. And the contents of their
1713296c7658Sdan   ** associated hash-tables. */
17145d607a6eSdan   sessionDeleteTable(pSession->pTable);
17154fccf43aSdan 
1716296c7658Sdan   /* Free the session object itself. */
17174fccf43aSdan   sqlite3_free(pSession);
17184fccf43aSdan }
17194fccf43aSdan 
17204fccf43aSdan /*
17217531a5a3Sdan ** Set a table filter on a Session Object.
17227531a5a3Sdan */
17237531a5a3Sdan void sqlite3session_table_filter(
17247531a5a3Sdan   sqlite3_session *pSession,
17257531a5a3Sdan   int(*xFilter)(void*, const char*),
17267531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
17277531a5a3Sdan ){
17287531a5a3Sdan   pSession->bAutoAttach = 1;
17297531a5a3Sdan   pSession->pFilterCtx = pCtx;
17307531a5a3Sdan   pSession->xTableFilter = xFilter;
17317531a5a3Sdan }
17327531a5a3Sdan 
17337531a5a3Sdan /*
17344fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
17354fccf43aSdan ** while the session object is enabled will be recorded.
17364fccf43aSdan **
17374fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
17384fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
17394fccf43aSdan ** or not.
17404fccf43aSdan */
17414fccf43aSdan int sqlite3session_attach(
17424fccf43aSdan   sqlite3_session *pSession,      /* Session object */
17434fccf43aSdan   const char *zName               /* Table name */
17444fccf43aSdan ){
1745ff4d0f41Sdan   int rc = SQLITE_OK;
1746ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1747ff4d0f41Sdan 
1748ff4d0f41Sdan   if( !zName ){
1749ff4d0f41Sdan     pSession->bAutoAttach = 1;
1750ff4d0f41Sdan   }else{
1751296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1752296c7658Sdan     int nName;                    /* Number of bytes in string zName */
17534fccf43aSdan 
17544fccf43aSdan     /* First search for an existing entry. If one is found, this call is
17554fccf43aSdan     ** a no-op. Return early. */
1756cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
17574fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
17584c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
17594fccf43aSdan     }
17604fccf43aSdan 
17614c220252Sdan     if( !pTab ){
17624fccf43aSdan       /* Allocate new SessionTable object. */
17634fccf43aSdan       pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1);
17644c220252Sdan       if( !pTab ){
17654c220252Sdan         rc = SQLITE_NOMEM;
17664c220252Sdan       }else{
17676c39e6a8Sdan         /* Populate the new SessionTable object and link it into the list.
17686c39e6a8Sdan         ** The new object must be linked onto the end of the list, not
17696c39e6a8Sdan         ** simply added to the start of it in order to ensure that tables
17706c39e6a8Sdan         ** appear in the correct order when a changeset or patchset is
17716c39e6a8Sdan         ** eventually generated. */
17726c39e6a8Sdan         SessionTable **ppTab;
17734fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
17744fccf43aSdan         pTab->zName = (char *)&pTab[1];
17754fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
17766c39e6a8Sdan         for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext);
17776c39e6a8Sdan         *ppTab = pTab;
17784c220252Sdan       }
17794c220252Sdan     }
1780ff4d0f41Sdan   }
17814fccf43aSdan 
17824c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
17834c220252Sdan   return rc;
17844fccf43aSdan }
17854fccf43aSdan 
1786296c7658Sdan /*
1787296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1788296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1789296c7658Sdan **
1790296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1791296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1792296c7658Sdan */
17934fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){
179480fe2d93Sdan   if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){
17954fccf43aSdan     u8 *aNew;
17964fccf43aSdan     int nNew = p->nAlloc ? p->nAlloc : 128;
17974fccf43aSdan     do {
17984fccf43aSdan       nNew = nNew*2;
1799ef7a6304Sdan     }while( nNew<(p->nBuf+nByte) );
18004fccf43aSdan 
18014fccf43aSdan     aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew);
18024fccf43aSdan     if( 0==aNew ){
18034fccf43aSdan       *pRc = SQLITE_NOMEM;
180480fe2d93Sdan     }else{
18054fccf43aSdan       p->aBuf = aNew;
18064fccf43aSdan       p->nAlloc = nNew;
18074fccf43aSdan     }
180880fe2d93Sdan   }
180980fe2d93Sdan   return (*pRc!=SQLITE_OK);
18104fccf43aSdan }
18114fccf43aSdan 
1812296c7658Sdan /*
1813fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1814fa122adaSdan ** as the first.
1815fa122adaSdan **
1816fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1817fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1818fa122adaSdan ** before returning.
1819fa122adaSdan */
1820fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1821fa122adaSdan   int rc = *pRc;
1822fa122adaSdan   if( rc==SQLITE_OK ){
1823fa122adaSdan     int nByte = 0;
1824e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1825fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1826fa122adaSdan     if( rc==SQLITE_OK ){
1827fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1828fa122adaSdan       p->nBuf += nByte;
1829fa122adaSdan     }else{
1830fa122adaSdan       *pRc = rc;
1831fa122adaSdan     }
1832fa122adaSdan   }
1833fa122adaSdan }
1834fa122adaSdan 
1835fa122adaSdan /*
1836296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1837296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1838296c7658Sdan **
1839296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1840296c7658Sdan ** returning.
1841296c7658Sdan */
18424fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
184380fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
18444fccf43aSdan     p->aBuf[p->nBuf++] = v;
18454fccf43aSdan   }
18464fccf43aSdan }
18474fccf43aSdan 
1848296c7658Sdan /*
1849296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1850296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1851296c7658Sdan **
1852296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1853296c7658Sdan ** returning.
1854296c7658Sdan */
1855cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
185680fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
18574fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
18584fccf43aSdan   }
18594fccf43aSdan }
18604fccf43aSdan 
1861296c7658Sdan /*
1862296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1863296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1864296c7658Sdan **
1865296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1866296c7658Sdan ** returning.
1867296c7658Sdan */
18684fccf43aSdan static void sessionAppendBlob(
18694fccf43aSdan   SessionBuffer *p,
18704fccf43aSdan   const u8 *aBlob,
18714fccf43aSdan   int nBlob,
18724fccf43aSdan   int *pRc
18734fccf43aSdan ){
1874895decf6Sdan   if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){
18754fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
18764fccf43aSdan     p->nBuf += nBlob;
18774fccf43aSdan   }
18784fccf43aSdan }
18794fccf43aSdan 
1880296c7658Sdan /*
1881296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1882296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1883296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1884296c7658Sdan **
1885296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1886296c7658Sdan ** returning.
1887296c7658Sdan */
1888d5f0767cSdan static void sessionAppendStr(
1889d5f0767cSdan   SessionBuffer *p,
1890d5f0767cSdan   const char *zStr,
1891d5f0767cSdan   int *pRc
1892d5f0767cSdan ){
1893cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
189480fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1895d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1896d5f0767cSdan     p->nBuf += nStr;
1897d5f0767cSdan   }
1898d5f0767cSdan }
1899d5f0767cSdan 
1900296c7658Sdan /*
1901296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1902296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1903296c7658Sdan ** to the buffer. No nul-terminator is written.
1904296c7658Sdan **
1905296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1906296c7658Sdan ** returning.
1907296c7658Sdan */
1908d5f0767cSdan static void sessionAppendInteger(
1909296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1910296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1911296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1912d5f0767cSdan ){
1913d5f0767cSdan   char aBuf[24];
1914d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1915d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1916d5f0767cSdan }
1917d5f0767cSdan 
1918296c7658Sdan /*
1919296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1920296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1921296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1922296c7658Sdan ** nul-terminator byte is written.
1923296c7658Sdan **
1924296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1925296c7658Sdan ** returning.
1926296c7658Sdan */
1927d5f0767cSdan static void sessionAppendIdent(
1928296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1929296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1930296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1931d5f0767cSdan ){
1932cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
193380fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1934d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1935d5f0767cSdan     const char *zIn = zStr;
1936d5f0767cSdan     *zOut++ = '"';
1937d5f0767cSdan     while( *zIn ){
1938d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1939d5f0767cSdan       *zOut++ = *(zIn++);
1940d5f0767cSdan     }
1941d5f0767cSdan     *zOut++ = '"';
1942cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1943d5f0767cSdan   }
1944d5f0767cSdan }
1945d5f0767cSdan 
1946296c7658Sdan /*
1947296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1948296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1949296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1950296c7658Sdan ** to to the buffer.
1951296c7658Sdan */
19524fccf43aSdan static void sessionAppendCol(
1953296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1954296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1955296c7658Sdan   int iCol,                       /* Column to read value from */
1956296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
19574fccf43aSdan ){
19584fccf43aSdan   if( *pRc==SQLITE_OK ){
19594fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
19604fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
19614fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
19624fccf43aSdan       sqlite3_int64 i;
19634fccf43aSdan       u8 aBuf[8];
19644fccf43aSdan       if( eType==SQLITE_INTEGER ){
19654fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
19664fccf43aSdan       }else{
19674fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
19684fccf43aSdan         memcpy(&i, &r, 8);
19694fccf43aSdan       }
1970296c7658Sdan       sessionPutI64(aBuf, i);
19714fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
19724fccf43aSdan     }
19734fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
19746734007dSdan       u8 *z;
19753cc89d95Sdan       int nByte;
19766734007dSdan       if( eType==SQLITE_BLOB ){
19776734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
19786734007dSdan       }else{
19796734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
19806734007dSdan       }
19813cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
19823cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
19834fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
19846734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
19856734007dSdan       }else{
19866734007dSdan         *pRc = SQLITE_NOMEM;
19876734007dSdan       }
19884fccf43aSdan     }
19894fccf43aSdan   }
19904fccf43aSdan }
19914fccf43aSdan 
1992296c7658Sdan /*
1993296c7658Sdan **
199480fe2d93Sdan ** This function appends an update change to the buffer (see the comments
199580fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
199680fe2d93Sdan ** consists of:
1997296c7658Sdan **
1998296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
1999296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
2000296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
2001296c7658Sdan **
2002296c7658Sdan ** The SessionChange object passed as the third argument contains the
2003296c7658Sdan ** values that were stored in the row when the session began (the old.*
2004296c7658Sdan ** values). The statement handle passed as the second argument points
2005296c7658Sdan ** at the current version of the row (the new.* values).
2006296c7658Sdan **
2007296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
2008296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
2009296c7658Sdan **
2010296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
2011296c7658Sdan ** original values of any fields that have been modified. The new.* record
2012296c7658Sdan ** contains the new values of only those fields that have been modified.
2013296c7658Sdan */
201480fe2d93Sdan static int sessionAppendUpdate(
2015296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
201673b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
2017296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
2018296c7658Sdan   SessionChange *p,               /* Object containing old values */
201980fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
20204fccf43aSdan ){
202180fe2d93Sdan   int rc = SQLITE_OK;
2022296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
2023296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
20241f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
2025296c7658Sdan   int i;                        /* Used to iterate through columns */
2026296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
2027296c7658Sdan 
202880fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
202980fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
20304fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
203137f133ecSdan     int bChanged = 0;
20324fccf43aSdan     int nAdvance;
20334fccf43aSdan     int eType = *pCsr;
20344fccf43aSdan     switch( eType ){
20354fccf43aSdan       case SQLITE_NULL:
20364fccf43aSdan         nAdvance = 1;
20374fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
203837f133ecSdan           bChanged = 1;
20394fccf43aSdan         }
20404fccf43aSdan         break;
20414fccf43aSdan 
20424fccf43aSdan       case SQLITE_FLOAT:
20434fccf43aSdan       case SQLITE_INTEGER: {
20444fccf43aSdan         nAdvance = 9;
20454fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
20464fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
20474fccf43aSdan           if( eType==SQLITE_INTEGER ){
20484fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
20494fccf43aSdan           }else{
20504fccf43aSdan             double dVal;
20514fccf43aSdan             memcpy(&dVal, &iVal, 8);
20524fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
20534fccf43aSdan           }
20544fccf43aSdan         }
205537f133ecSdan         bChanged = 1;
20564fccf43aSdan         break;
20574fccf43aSdan       }
20584fccf43aSdan 
2059e5754eecSdan       default: {
2060895decf6Sdan         int n;
2061895decf6Sdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &n);
2062e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
2063895decf6Sdan         nAdvance = nHdr + n;
20644fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
2065895decf6Sdan          && n==sqlite3_column_bytes(pStmt, i)
2066895decf6Sdan          && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n))
20674fccf43aSdan         ){
20684fccf43aSdan           break;
20694fccf43aSdan         }
207037f133ecSdan         bChanged = 1;
20714fccf43aSdan       }
20724fccf43aSdan     }
20734fccf43aSdan 
207473b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
207573b3c055Sdan     if( bChanged ) bNoop = 0;
207673b3c055Sdan 
207773b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
207873b3c055Sdan     ** currently generating a patchset. */
207973b3c055Sdan     if( bPatchset==0 ){
208037f133ecSdan       if( bChanged || abPK[i] ){
208180fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
20824fccf43aSdan       }else{
208380fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
208437f133ecSdan       }
208573b3c055Sdan     }
208637f133ecSdan 
208773b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
208873b3c055Sdan     ** generating a patchset.  */
208973b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
209080fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
209137f133ecSdan     }else{
209280fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
20934fccf43aSdan     }
209437f133ecSdan 
20954fccf43aSdan     pCsr += nAdvance;
20964fccf43aSdan   }
20974fccf43aSdan 
20984fccf43aSdan   if( bNoop ){
20991f34f8ccSdan     pBuf->nBuf = nRewind;
21004fccf43aSdan   }else{
210180fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
21024fccf43aSdan   }
21031f34f8ccSdan   sqlite3_free(buf2.aBuf);
210480fe2d93Sdan 
210580fe2d93Sdan   return rc;
2106d5f0767cSdan }
21074fccf43aSdan 
2108a71d2371Sdan /*
2109a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
2110a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
2111a71d2371Sdan ** format otherwise.
2112a71d2371Sdan */
211373b3c055Sdan static int sessionAppendDelete(
211473b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
211573b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
211673b3c055Sdan   SessionChange *p,               /* Object containing old values */
2117a71d2371Sdan   int nCol,                       /* Number of columns in table */
211873b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
211973b3c055Sdan ){
212073b3c055Sdan   int rc = SQLITE_OK;
212173b3c055Sdan 
212273b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
212373b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
212473b3c055Sdan 
212573b3c055Sdan   if( bPatchset==0 ){
212673b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
212773b3c055Sdan   }else{
212873b3c055Sdan     int i;
212973b3c055Sdan     u8 *a = p->aRecord;
213073b3c055Sdan     for(i=0; i<nCol; i++){
213173b3c055Sdan       u8 *pStart = a;
213273b3c055Sdan       int eType = *a++;
213373b3c055Sdan 
213473b3c055Sdan       switch( eType ){
213573b3c055Sdan         case 0:
213673b3c055Sdan         case SQLITE_NULL:
213773b3c055Sdan           assert( abPK[i]==0 );
213873b3c055Sdan           break;
213973b3c055Sdan 
214073b3c055Sdan         case SQLITE_FLOAT:
214173b3c055Sdan         case SQLITE_INTEGER:
214273b3c055Sdan           a += 8;
214373b3c055Sdan           break;
214473b3c055Sdan 
214573b3c055Sdan         default: {
214673b3c055Sdan           int n;
214773b3c055Sdan           a += sessionVarintGet(a, &n);
214873b3c055Sdan           a += n;
214973b3c055Sdan           break;
215073b3c055Sdan         }
215173b3c055Sdan       }
215273b3c055Sdan       if( abPK[i] ){
2153f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
215473b3c055Sdan       }
215573b3c055Sdan     }
215673b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
215773b3c055Sdan   }
215873b3c055Sdan 
215973b3c055Sdan   return rc;
216073b3c055Sdan }
216173b3c055Sdan 
216277fc1d5bSdan /*
216377fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
216477fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
216577fc1d5bSdan **
216677fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
216777fc1d5bSdan */
2168e8d5648eSdan static int sessionSelectStmt(
2169e8d5648eSdan   sqlite3 *db,                    /* Database handle */
2170d7fb7d24Sdan   const char *zDb,                /* Database name */
2171e8d5648eSdan   const char *zTab,               /* Table name */
217277fc1d5bSdan   int nCol,                       /* Number of columns in table */
217377fc1d5bSdan   const char **azCol,             /* Names of table columns */
217477fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
217577fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
2176d5f0767cSdan ){
2177e8d5648eSdan   int rc = SQLITE_OK;
21783739f298Sdan   char *zSql = 0;
21793739f298Sdan   int nSql = -1;
21803739f298Sdan 
21813739f298Sdan   if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){
21823739f298Sdan     zSql = sqlite3_mprintf(
21833739f298Sdan         "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND "
21843739f298Sdan         "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb
21853739f298Sdan     );
21863739f298Sdan   }else{
2187d5f0767cSdan     int i;
2188e8d5648eSdan     const char *zSep = "";
2189e8d5648eSdan     SessionBuffer buf = {0, 0, 0};
2190d5f0767cSdan 
2191e8d5648eSdan     sessionAppendStr(&buf, "SELECT * FROM ", &rc);
2192d7fb7d24Sdan     sessionAppendIdent(&buf, zDb, &rc);
2193d7fb7d24Sdan     sessionAppendStr(&buf, ".", &rc);
2194e8d5648eSdan     sessionAppendIdent(&buf, zTab, &rc);
2195e8d5648eSdan     sessionAppendStr(&buf, " WHERE ", &rc);
2196e8d5648eSdan     for(i=0; i<nCol; i++){
2197e8d5648eSdan       if( abPK[i] ){
2198e8d5648eSdan         sessionAppendStr(&buf, zSep, &rc);
2199e8d5648eSdan         sessionAppendIdent(&buf, azCol[i], &rc);
22003739f298Sdan         sessionAppendStr(&buf, " IS ?", &rc);
2201e8d5648eSdan         sessionAppendInteger(&buf, i+1, &rc);
2202e8d5648eSdan         zSep = " AND ";
2203d5f0767cSdan       }
2204d5f0767cSdan     }
22053739f298Sdan     zSql = (char*)buf.aBuf;
22063739f298Sdan     nSql = buf.nBuf;
2207d5f0767cSdan   }
22083739f298Sdan 
22093739f298Sdan   if( rc==SQLITE_OK ){
22103739f298Sdan     rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0);
22113739f298Sdan   }
22123739f298Sdan   sqlite3_free(zSql);
2213e8d5648eSdan   return rc;
2214d5f0767cSdan }
2215d5f0767cSdan 
221677fc1d5bSdan /*
221777fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
221877fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
221977fc1d5bSdan ** is as prepared by function sessionSelectStmt().
222077fc1d5bSdan **
222177fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
222277fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
222377fc1d5bSdan */
2224e8d5648eSdan static int sessionSelectBind(
222577fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
222677fc1d5bSdan   int nCol,                       /* Number of columns in table */
222777fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
222877fc1d5bSdan   SessionChange *pChange          /* Change structure */
2229e8d5648eSdan ){
2230e8d5648eSdan   int i;
2231e8d5648eSdan   int rc = SQLITE_OK;
2232e5754eecSdan   u8 *a = pChange->aRecord;
2233d5f0767cSdan 
2234e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2235e8d5648eSdan     int eType = *a++;
2236e8d5648eSdan 
2237e8d5648eSdan     switch( eType ){
223880fe2d93Sdan       case 0:
2239e8d5648eSdan       case SQLITE_NULL:
22401611e5a3Sdan         assert( abPK[i]==0 );
2241e8d5648eSdan         break;
2242e8d5648eSdan 
2243e8d5648eSdan       case SQLITE_INTEGER: {
2244e8d5648eSdan         if( abPK[i] ){
2245e8d5648eSdan           i64 iVal = sessionGetI64(a);
2246e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
2247e8d5648eSdan         }
2248e8d5648eSdan         a += 8;
2249e8d5648eSdan         break;
2250d5f0767cSdan       }
2251296c7658Sdan 
2252e8d5648eSdan       case SQLITE_FLOAT: {
2253e8d5648eSdan         if( abPK[i] ){
2254e8d5648eSdan           double rVal;
2255e8d5648eSdan           i64 iVal = sessionGetI64(a);
2256e8d5648eSdan           memcpy(&rVal, &iVal, 8);
22574e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
2258d5f0767cSdan         }
2259e8d5648eSdan         a += 8;
2260e8d5648eSdan         break;
2261e8d5648eSdan       }
2262e8d5648eSdan 
2263e8d5648eSdan       case SQLITE_TEXT: {
2264e8d5648eSdan         int n;
2265e8d5648eSdan         a += sessionVarintGet(a, &n);
2266e8d5648eSdan         if( abPK[i] ){
2267e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
2268e8d5648eSdan         }
2269e8d5648eSdan         a += n;
2270e8d5648eSdan         break;
2271e8d5648eSdan       }
2272e8d5648eSdan 
2273e5754eecSdan       default: {
2274e8d5648eSdan         int n;
2275e5754eecSdan         assert( eType==SQLITE_BLOB );
2276e8d5648eSdan         a += sessionVarintGet(a, &n);
2277e8d5648eSdan         if( abPK[i] ){
2278e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
2279e8d5648eSdan         }
2280e8d5648eSdan         a += n;
2281e8d5648eSdan         break;
2282e8d5648eSdan       }
2283e8d5648eSdan     }
2284e8d5648eSdan   }
2285e8d5648eSdan 
2286d5f0767cSdan   return rc;
22874fccf43aSdan }
22884fccf43aSdan 
228977fc1d5bSdan /*
229077fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
229177fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
229277fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
229377fc1d5bSdan ** SQLite error code before returning.
229477fc1d5bSdan */
22955d607a6eSdan static void sessionAppendTableHdr(
2296a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
2297a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
2298a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
2299a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
23005d607a6eSdan ){
23015d607a6eSdan   /* Write a table header */
230273b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
23035d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
23045d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
23054f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
23065d607a6eSdan }
23075d607a6eSdan 
2308a71d2371Sdan /*
2309a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
2310a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
2311a71d2371Sdan ** passed as the first argument.
2312a71d2371Sdan **
2313a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
2314a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
2315a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
2316a71d2371Sdan ** to 0.
2317a71d2371Sdan */
2318adf3bf58Sdrh static int sessionGenerateChangeset(
23194fccf43aSdan   sqlite3_session *pSession,      /* Session object */
232073b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
2321ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2322ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
23234fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
23244fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
23254fccf43aSdan ){
2326296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
2327296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
2328296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
2329296c7658Sdan   int rc;                         /* Return code */
23304fccf43aSdan 
2331ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
2332ef7a6304Sdan 
2333296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
2334296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
2335296c7658Sdan   ** this call will be a no-op.  */
2336ef7a6304Sdan   if( xOutput==0 ){
23374fccf43aSdan     *pnChangeset = 0;
23384fccf43aSdan     *ppChangeset = 0;
2339ef7a6304Sdan   }
2340e5754eecSdan 
2341e5754eecSdan   if( pSession->rc ) return pSession->rc;
2342e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
2343e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
2344e5754eecSdan 
2345e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
23464fccf43aSdan 
23474fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
23484fccf43aSdan     if( pTab->nEntry ){
2349d7fb7d24Sdan       const char *zName = pTab->zName;
2350a9605b91Sdan       int nCol;                   /* Number of columns in table */
2351a9605b91Sdan       u8 *abPK;                   /* Primary key array */
2352a9605b91Sdan       const char **azCol = 0;     /* Table columns */
23531f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
23541f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
23551f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
23561f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
23574fccf43aSdan 
2358a9605b91Sdan       /* Check the table schema is still Ok. */
2359a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
2360a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
2361a9605b91Sdan         rc = SQLITE_SCHEMA;
2362a9605b91Sdan       }
2363a9605b91Sdan 
23644fccf43aSdan       /* Write a table header */
236573b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
23664fccf43aSdan 
23674fccf43aSdan       /* Build and compile a statement to execute: */
23684fccf43aSdan       if( rc==SQLITE_OK ){
2369d7fb7d24Sdan         rc = sessionSelectStmt(
2370a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
23714fccf43aSdan       }
23724fccf43aSdan 
23731f34f8ccSdan       nNoop = buf.nBuf;
237412ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
2375e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
2376e8d5648eSdan 
23774fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
2378e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
237980fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
23801f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
2381798693b2Sdan             if( p->op==SQLITE_INSERT ){
23824fccf43aSdan               int iCol;
23834fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
2384b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
2385e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
23861f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
23874fccf43aSdan               }
2388e8d5648eSdan             }else{
238973b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
23904fccf43aSdan             }
2391798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
2392a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
23934fccf43aSdan           }
239412ca0b56Sdan           if( rc==SQLITE_OK ){
23951f34f8ccSdan             rc = sqlite3_reset(pSel);
23964fccf43aSdan           }
2397ef7a6304Sdan 
2398f1a08ad8Sdrh           /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass
2399ef7a6304Sdan           ** its contents to the xOutput() callback. */
2400ef7a6304Sdan           if( xOutput
2401ef7a6304Sdan            && rc==SQLITE_OK
2402ef7a6304Sdan            && buf.nBuf>nNoop
2403f1a08ad8Sdrh            && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE
2404ef7a6304Sdan           ){
2405ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2406ef7a6304Sdan             nNoop = -1;
2407ef7a6304Sdan             buf.nBuf = 0;
2408ef7a6304Sdan           }
2409ef7a6304Sdan 
24104fccf43aSdan         }
2411e8d5648eSdan       }
24124fccf43aSdan 
24131f34f8ccSdan       sqlite3_finalize(pSel);
24141f34f8ccSdan       if( buf.nBuf==nNoop ){
24154fccf43aSdan         buf.nBuf = nRewind;
24164fccf43aSdan       }
2417cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
24184fccf43aSdan     }
24194fccf43aSdan   }
24204fccf43aSdan 
24214fccf43aSdan   if( rc==SQLITE_OK ){
2422ef7a6304Sdan     if( xOutput==0 ){
24234fccf43aSdan       *pnChangeset = buf.nBuf;
24244fccf43aSdan       *ppChangeset = buf.aBuf;
2425ef7a6304Sdan       buf.aBuf = 0;
2426ef7a6304Sdan     }else if( buf.nBuf>0 ){
2427ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2428ef7a6304Sdan     }
24294fccf43aSdan   }
24304c220252Sdan 
2431ef7a6304Sdan   sqlite3_free(buf.aBuf);
2432e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
24334c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
24344fccf43aSdan   return rc;
24354fccf43aSdan }
24364fccf43aSdan 
2437296c7658Sdan /*
243873b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
243973b3c055Sdan ** session object passed as the first argument.
244073b3c055Sdan **
244173b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
244273b3c055Sdan ** using sqlite3_free().
244373b3c055Sdan */
244473b3c055Sdan int sqlite3session_changeset(
244573b3c055Sdan   sqlite3_session *pSession,      /* Session object */
244673b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
244773b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
244873b3c055Sdan ){
2449ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2450ef7a6304Sdan }
2451ef7a6304Sdan 
2452ef7a6304Sdan /*
2453ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2454ef7a6304Sdan */
2455f1a08ad8Sdrh int sqlite3session_changeset_strm(
2456ef7a6304Sdan   sqlite3_session *pSession,
2457ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2458ef7a6304Sdan   void *pOut
2459ef7a6304Sdan ){
2460ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2461ef7a6304Sdan }
2462ef7a6304Sdan 
2463ef7a6304Sdan /*
2464ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2465ef7a6304Sdan */
2466f1a08ad8Sdrh int sqlite3session_patchset_strm(
2467ef7a6304Sdan   sqlite3_session *pSession,
2468ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2469ef7a6304Sdan   void *pOut
2470ef7a6304Sdan ){
2471ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
247273b3c055Sdan }
247373b3c055Sdan 
247473b3c055Sdan /*
247573b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
247673b3c055Sdan ** session object passed as the first argument.
247773b3c055Sdan **
247873b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
247973b3c055Sdan ** using sqlite3_free().
248073b3c055Sdan */
248173b3c055Sdan int sqlite3session_patchset(
248273b3c055Sdan   sqlite3_session *pSession,      /* Session object */
248373b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
248473b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
248573b3c055Sdan ){
2486ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
248773b3c055Sdan }
248873b3c055Sdan 
248973b3c055Sdan /*
2490296c7658Sdan ** Enable or disable the session object passed as the first argument.
2491296c7658Sdan */
24924fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
24934c220252Sdan   int ret;
24944c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2495296c7658Sdan   if( bEnable>=0 ){
2496296c7658Sdan     pSession->bEnable = bEnable;
24974fccf43aSdan   }
24984c220252Sdan   ret = pSession->bEnable;
24994c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
25004c220252Sdan   return ret;
2501296c7658Sdan }
25024fccf43aSdan 
25034fccf43aSdan /*
2504b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2505b4480e94Sdan */
2506b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2507b4480e94Sdan   int ret;
2508b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2509b4480e94Sdan   if( bIndirect>=0 ){
2510b4480e94Sdan     pSession->bIndirect = bIndirect;
2511b4480e94Sdan   }
2512b4480e94Sdan   ret = pSession->bIndirect;
2513b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2514b4480e94Sdan   return ret;
2515b4480e94Sdan }
2516b4480e94Sdan 
2517b4480e94Sdan /*
2518b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2519b69ec348Sdan ** by the session object passed as the only argument.
2520b69ec348Sdan */
2521b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2522b69ec348Sdan   int ret = 0;
2523b69ec348Sdan   SessionTable *pTab;
2524b69ec348Sdan 
2525b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2526b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2527b69ec348Sdan     ret = (pTab->nEntry>0);
2528b69ec348Sdan   }
2529b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2530b69ec348Sdan 
2531ff530326Sdan   return (ret==0);
2532b69ec348Sdan }
2533b69ec348Sdan 
2534b69ec348Sdan /*
2535f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
25364fccf43aSdan */
2537adf3bf58Sdrh static int sessionChangesetStart(
2538296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2539ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2540ef7a6304Sdan   void *pIn,
2541296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2542296c7658Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
25434fccf43aSdan ){
25444fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
25454fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
25464fccf43aSdan 
2547ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2548ef7a6304Sdan 
2549296c7658Sdan   /* Zero the output variable in case an error occurs. */
2550296c7658Sdan   *pp = 0;
25514fccf43aSdan 
2552296c7658Sdan   /* Allocate and initialize the iterator structure. */
25534fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
25544fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
25554fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
25564fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
25574757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
25584757c658Sdan   pRet->in.nData = nChangeset;
2559ef7a6304Sdan   pRet->in.xInput = xInput;
2560ef7a6304Sdan   pRet->in.pIn = pIn;
2561ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
25624fccf43aSdan 
2563296c7658Sdan   /* Populate the output variable and return success. */
2564296c7658Sdan   *pp = pRet;
25654fccf43aSdan   return SQLITE_OK;
25664fccf43aSdan }
25674fccf43aSdan 
2568296c7658Sdan /*
2569ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2570ef7a6304Sdan */
2571ef7a6304Sdan int sqlite3changeset_start(
2572ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2573ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2574ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2575ef7a6304Sdan ){
2576ef7a6304Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset);
2577ef7a6304Sdan }
2578ef7a6304Sdan 
2579ef7a6304Sdan /*
2580ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2581ef7a6304Sdan */
2582f1a08ad8Sdrh int sqlite3changeset_start_strm(
2583ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2584ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2585ef7a6304Sdan   void *pIn
2586ef7a6304Sdan ){
2587ef7a6304Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0);
2588ef7a6304Sdan }
2589ef7a6304Sdan 
2590ef7a6304Sdan /*
2591d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming
2592d9151526Sdan ** object and the buffer is full, discard some data to free up space.
2593d9151526Sdan */
2594d9151526Sdan static void sessionDiscardData(SessionInput *pIn){
2595d9151526Sdan   if( pIn->bEof && pIn->xInput && pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){
2596d9151526Sdan     int nMove = pIn->buf.nBuf - pIn->iNext;
2597d9151526Sdan     assert( nMove>=0 );
2598d9151526Sdan     if( nMove>0 ){
2599d9151526Sdan       memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
2600d9151526Sdan     }
2601d9151526Sdan     pIn->buf.nBuf -= pIn->iNext;
2602d9151526Sdan     pIn->iNext = 0;
2603d9151526Sdan     pIn->nData = pIn->buf.nBuf;
2604d9151526Sdan   }
2605d9151526Sdan }
2606d9151526Sdan 
2607d9151526Sdan /*
2608ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2609ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2610ef7a6304Sdan ** data is in the buffer.
2611ef7a6304Sdan **
2612ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2613ef7a6304Sdan */
26144757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2615ef7a6304Sdan   int rc = SQLITE_OK;
26164757c658Sdan   if( pIn->xInput ){
26174757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
2618f1a08ad8Sdrh       int nNew = SESSIONS_STRM_CHUNK_SIZE;
26194757c658Sdan 
2620d9151526Sdan       if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn);
26214757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
26224757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
26234757c658Sdan         if( nNew==0 ){
26244757c658Sdan           pIn->bEof = 1;
26254757c658Sdan         }else{
26264757c658Sdan           pIn->buf.nBuf += nNew;
26274757c658Sdan         }
26284757c658Sdan       }
26294757c658Sdan 
26304757c658Sdan       pIn->aData = pIn->buf.aBuf;
26314757c658Sdan       pIn->nData = pIn->buf.nBuf;
26324757c658Sdan     }
2633ef7a6304Sdan   }
2634ef7a6304Sdan   return rc;
2635ef7a6304Sdan }
2636ef7a6304Sdan 
2637ef7a6304Sdan /*
2638ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2639ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2640ef7a6304Sdan ** until it points to the byte immediately following that record.
2641ef7a6304Sdan */
2642ef7a6304Sdan static void sessionSkipRecord(
2643ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2644ef7a6304Sdan   int nCol                        /* Number of values in record */
2645ef7a6304Sdan ){
2646ef7a6304Sdan   u8 *aRec = *ppRec;
2647ef7a6304Sdan   int i;
2648ef7a6304Sdan   for(i=0; i<nCol; i++){
2649ef7a6304Sdan     int eType = *aRec++;
2650ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2651ef7a6304Sdan       int nByte;
2652ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2653ef7a6304Sdan       aRec += nByte;
2654ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2655ef7a6304Sdan       aRec += 8;
2656ef7a6304Sdan     }
2657ef7a6304Sdan   }
2658ef7a6304Sdan 
2659ef7a6304Sdan   *ppRec = aRec;
2660ef7a6304Sdan }
2661ef7a6304Sdan 
2662ef7a6304Sdan /*
26634757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
26644757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
26654757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
26664757c658Sdan ** error occurs.
26674757c658Sdan */
26684757c658Sdan static int sessionValueSetStr(
26694757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
26704757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
26714757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
26724757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
26734757c658Sdan ){
267416228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
267516228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
267616228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
267716228167Sdan   ** error. Hence the code to create the copy externally. */
26783cc89d95Sdan   u8 *aCopy = sqlite3_malloc(nData+1);
26794757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
26804757c658Sdan   memcpy(aCopy, aData, nData);
26814757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
26824757c658Sdan   return SQLITE_OK;
26834757c658Sdan }
26844757c658Sdan 
26854757c658Sdan /*
2686296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2687296c7658Sdan ** for details.
2688296c7658Sdan **
2689296c7658Sdan ** When this function is called, *paChange points to the start of the record
2690296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2691296c7658Sdan ** one byte after the end of the same record before this function returns.
2692a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2693a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2694a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2695296c7658Sdan **
2696296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2697296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2698296c7658Sdan ** from the corresponding position in the record. If that value is not
2699296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2700296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2701296c7658Sdan ** set to NULL.
2702296c7658Sdan **
2703296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2704296c7658Sdan ** using sqlite3_free().
2705296c7658Sdan **
2706296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2707296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2708296c7658Sdan */
27094fccf43aSdan static int sessionReadRecord(
2710ef7a6304Sdan   SessionInput *pIn,              /* Input data */
27114fccf43aSdan   int nCol,                       /* Number of values in record */
271273b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
27134fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
27144fccf43aSdan ){
2715296c7658Sdan   int i;                          /* Used to iterate through columns */
2716ef7a6304Sdan   int rc = SQLITE_OK;
27174fccf43aSdan 
2718ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2719ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
272073b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2721ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2722ef7a6304Sdan     if( rc==SQLITE_OK ){
27234757c658Sdan       eType = pIn->aData[pIn->iNext++];
2724ef7a6304Sdan     }
2725ef7a6304Sdan 
2726e8fa8c96Sdan     assert( apOut[i]==0 );
27274fccf43aSdan     if( eType ){
27284fccf43aSdan       apOut[i] = sqlite3ValueNew(0);
2729ef7a6304Sdan       if( !apOut[i] ) rc = SQLITE_NOMEM;
2730ef7a6304Sdan     }
27314fccf43aSdan 
2732ef7a6304Sdan     if( rc==SQLITE_OK ){
27334757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
27344fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
27354fccf43aSdan         int nByte;
2736ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2737ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2738e8fa8c96Sdan         if( rc==SQLITE_OK ){
27396734007dSdan           u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
27404757c658Sdan           rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
274191ddd559Sdan         }
2742ef7a6304Sdan         pIn->iNext += nByte;
27434fccf43aSdan       }
27444fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2745ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
27464fccf43aSdan         if( eType==SQLITE_INTEGER ){
27474fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
27484fccf43aSdan         }else{
27494fccf43aSdan           double d;
27504e895da1Sdan           memcpy(&d, &v, 8);
27514fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
27524fccf43aSdan         }
2753ef7a6304Sdan         pIn->iNext += 8;
275491ddd559Sdan       }
27554fccf43aSdan     }
27564fccf43aSdan   }
27574fccf43aSdan 
2758ef7a6304Sdan   return rc;
2759ef7a6304Sdan }
2760ef7a6304Sdan 
2761ef7a6304Sdan /*
2762ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2763ef7a6304Sdan ** Specifically, to the following:
2764ef7a6304Sdan **
2765ef7a6304Sdan **   + number of columns in table (varint)
2766ef7a6304Sdan **   + array of PK flags (1 byte per column),
2767ef7a6304Sdan **   + table name (nul terminated).
2768ef7a6304Sdan **
2769ef7a6304Sdan ** This function ensures that all of the above is present in the input
2770ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2771ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2772ef7a6304Sdan ** The input pointer is not moved.
2773ef7a6304Sdan */
2774ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2775ef7a6304Sdan   int rc = SQLITE_OK;
2776ef7a6304Sdan   int nCol = 0;
27774757c658Sdan   int nRead = 0;
2778ef7a6304Sdan 
2779ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2780ef7a6304Sdan   if( rc==SQLITE_OK ){
27814757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
27824757c658Sdan     rc = sessionInputBuffer(pIn, nRead+nCol+100);
27834757c658Sdan     nRead += nCol;
2784ef7a6304Sdan   }
27854757c658Sdan 
2786ef7a6304Sdan   while( rc==SQLITE_OK ){
27874757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
27884757c658Sdan       nRead++;
2789ef7a6304Sdan     }
2790e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
27914757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
27924757c658Sdan   }
2793e8fa8c96Sdan   *pnByte = nRead+1;
2794ef7a6304Sdan   return rc;
2795ef7a6304Sdan }
2796ef7a6304Sdan 
2797ef7a6304Sdan /*
2798fa122adaSdan ** The input pointer currently points to the first byte of the first field
2799fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
280016228167Sdan ** record is buffered. It does not move the input pointer.
280116228167Sdan **
280216228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
280316228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
280416228167Sdan ** final value of *pnByte is undefined in this case.
2805fa122adaSdan */
2806fa122adaSdan static int sessionChangesetBufferRecord(
280716228167Sdan   SessionInput *pIn,              /* Input data */
280816228167Sdan   int nCol,                       /* Number of columns in record */
280916228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2810fa122adaSdan ){
2811fa122adaSdan   int rc = SQLITE_OK;
2812fa122adaSdan   int nByte = 0;
2813fa122adaSdan   int i;
2814fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2815fa122adaSdan     int eType;
2816fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2817fa122adaSdan     if( rc==SQLITE_OK ){
2818fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2819fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2820fa122adaSdan         int n;
2821fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2822fa122adaSdan         nByte += n;
2823fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2824fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2825fa122adaSdan         nByte += 8;
2826fa122adaSdan       }
2827fa122adaSdan     }
2828fa122adaSdan   }
2829fa122adaSdan   *pnByte = nByte;
2830fa122adaSdan   return rc;
2831fa122adaSdan }
2832fa122adaSdan 
2833fa122adaSdan /*
2834ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2835ef7a6304Sdan ** Specifically, to the following:
2836ef7a6304Sdan **
2837ef7a6304Sdan **   + number of columns in table (varint)
2838ef7a6304Sdan **   + array of PK flags (1 byte per column),
2839ef7a6304Sdan **   + table name (nul terminated).
284016228167Sdan **
284116228167Sdan ** This function decodes the table-header and populates the p->nCol,
284216228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
284316228167Sdan ** also allocated or resized according to the new value of p->nCol. The
284416228167Sdan ** input pointer is left pointing to the byte following the table header.
284516228167Sdan **
284616228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
284716228167Sdan ** is returned and the final values of the various fields enumerated above
284816228167Sdan ** are undefined.
2849ef7a6304Sdan */
2850ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2851ef7a6304Sdan   int rc;
2852ef7a6304Sdan   int nCopy;
2853ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2854ef7a6304Sdan 
2855ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2856ef7a6304Sdan   if( rc==SQLITE_OK ){
2857ef7a6304Sdan     int nByte;
2858ef7a6304Sdan     int nVarint;
28594757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2860ef7a6304Sdan     nCopy -= nVarint;
2861ef7a6304Sdan     p->in.iNext += nVarint;
2862ef7a6304Sdan     nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2863ef7a6304Sdan     p->tblhdr.nBuf = 0;
2864ef7a6304Sdan     sessionBufferGrow(&p->tblhdr, nByte, &rc);
2865ef7a6304Sdan   }
2866ef7a6304Sdan 
2867ef7a6304Sdan   if( rc==SQLITE_OK ){
2868ef7a6304Sdan     int iPK = sizeof(sqlite3_value*)*p->nCol*2;
2869ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
28704757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2871ef7a6304Sdan     p->in.iNext += nCopy;
2872ef7a6304Sdan   }
2873ef7a6304Sdan 
2874ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2875ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2876ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2877ef7a6304Sdan   return (p->rc = rc);
28784fccf43aSdan }
28794fccf43aSdan 
288077fc1d5bSdan /*
288177fc1d5bSdan ** Advance the changeset iterator to the next change.
288277fc1d5bSdan **
288377fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
288477fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
288577fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
288677fc1d5bSdan **
288777fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
288877fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
288977fc1d5bSdan ** the record to *pnRec.
289077fc1d5bSdan **
289177fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
289277fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
289377fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
289477fc1d5bSdan ** changes in the changeset.
289577fc1d5bSdan */
28965d607a6eSdan static int sessionChangesetNext(
289777fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
289877fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
289977fc1d5bSdan   int *pnRec                      /* If non-NULL, store size of record here */
29005d607a6eSdan ){
29014fccf43aSdan   int i;
2902ef7a6304Sdan   u8 op;
29034fccf43aSdan 
29045d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
29055d607a6eSdan 
2906296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
29074fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
29084fccf43aSdan 
29095d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
29104fccf43aSdan   if( p->apValue ){
29114fccf43aSdan     for(i=0; i<p->nCol*2; i++){
29124fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
29134fccf43aSdan     }
29144fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
29154fccf43aSdan   }
29164fccf43aSdan 
2917ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
2918ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
2919ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
2920ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
2921ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
2922ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
2923ef7a6304Sdan 
29244fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
29254757c658Sdan   if( p->in.iNext>=p->in.nData ){
29264fccf43aSdan     return SQLITE_DONE;
29274fccf43aSdan   }
29284fccf43aSdan 
2929d9151526Sdan   sessionDiscardData(&p->in);
2930d9151526Sdan   p->in.iCurrent = p->in.iNext;
2931d9151526Sdan 
29324757c658Sdan   op = p->in.aData[p->in.iNext++];
293307d0f15eSdan   while( op=='T' || op=='P' ){
2934ef7a6304Sdan     p->bPatchset = (op=='P');
2935ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
2936ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
2937d9151526Sdan     p->in.iCurrent = p->in.iNext;
293807d0f15eSdan     if( p->in.iNext>=p->in.nData ) return SQLITE_DONE;
29394757c658Sdan     op = p->in.aData[p->in.iNext++];
29405d607a6eSdan   }
29415d607a6eSdan 
2942ef7a6304Sdan   p->op = op;
29434757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
29444fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
29454757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
29464fccf43aSdan   }
29474fccf43aSdan 
2948cbf6d2d2Sdan   if( paRec ){
2949cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
2950cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
2951cbf6d2d2Sdan       nVal = p->nCol * 2;
2952cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
2953cbf6d2d2Sdan       nVal = 0;
2954cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
2955cbf6d2d2Sdan     }else{
2956cbf6d2d2Sdan       nVal = p->nCol;
2957cbf6d2d2Sdan     }
2958cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
2959cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
2960cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
2961cbf6d2d2Sdan     p->in.iNext += *pnRec;
2962cbf6d2d2Sdan   }else{
29635d607a6eSdan 
29644fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
296573b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
296673b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
2967cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue);
29684fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
29694fccf43aSdan     }
29704fccf43aSdan 
29714fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
29724fccf43aSdan     if( p->op!=SQLITE_DELETE ){
2973cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]);
29744fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
29754fccf43aSdan     }
29764fccf43aSdan 
2977cbf6d2d2Sdan     if( p->bPatchset && p->op==SQLITE_UPDATE ){
297873b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
297973b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
298073b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
298173b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
298273b3c055Sdan       for(i=0; i<p->nCol; i++){
298373b3c055Sdan         assert( p->apValue[i]==0 );
298473b3c055Sdan         assert( p->abPK[i]==0 || p->apValue[i+p->nCol] );
298573b3c055Sdan         if( p->abPK[i] ){
298673b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
298773b3c055Sdan           p->apValue[i+p->nCol] = 0;
298873b3c055Sdan         }
298973b3c055Sdan       }
299073b3c055Sdan     }
2991cbf6d2d2Sdan   }
2992ef7a6304Sdan 
29934fccf43aSdan   return SQLITE_ROW;
29944fccf43aSdan }
29954fccf43aSdan 
29964fccf43aSdan /*
29975d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
29985d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
29995d607a6eSdan ** or SQLITE_CORRUPT.
30005d607a6eSdan **
30015d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
30025d607a6eSdan ** callback by changeset_apply().
30035d607a6eSdan */
30045d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
30055d607a6eSdan   return sessionChangesetNext(p, 0, 0);
30065d607a6eSdan }
30075d607a6eSdan 
30085d607a6eSdan /*
3009244593c8Sdan ** The following function extracts information on the current change
301077fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
30114fccf43aSdan ** has returned SQLITE_ROW.
30124fccf43aSdan */
30134fccf43aSdan int sqlite3changeset_op(
3014296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
30154fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
30164fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
3017b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
3018b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
30194fccf43aSdan ){
30204fccf43aSdan   *pOp = pIter->op;
30214fccf43aSdan   *pnCol = pIter->nCol;
30224fccf43aSdan   *pzTab = pIter->zTab;
3023b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
30244fccf43aSdan   return SQLITE_OK;
30254fccf43aSdan }
30264fccf43aSdan 
302777fc1d5bSdan /*
302877fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
302977fc1d5bSdan ** the database table affected by the change that pIter currently points
303077fc1d5bSdan ** to. This function may only be called after changeset_next() returns
303177fc1d5bSdan ** SQLITE_ROW.
303277fc1d5bSdan */
3033244593c8Sdan int sqlite3changeset_pk(
3034244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
3035244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
3036244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
3037244593c8Sdan ){
3038244593c8Sdan   *pabPK = pIter->abPK;
3039244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
3040244593c8Sdan   return SQLITE_OK;
3041244593c8Sdan }
3042244593c8Sdan 
3043296c7658Sdan /*
3044296c7658Sdan ** This function may only be called while the iterator is pointing to an
3045296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
3046296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3047296c7658Sdan **
3048296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3049296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
3050296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3051296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
3052296c7658Sdan **
3053296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3054296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3055296c7658Sdan */
30564fccf43aSdan int sqlite3changeset_old(
3057296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3058296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
30594fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
30604fccf43aSdan ){
3061d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
3062d5f0767cSdan     return SQLITE_MISUSE;
3063d5f0767cSdan   }
30644fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
30654fccf43aSdan     return SQLITE_RANGE;
30664fccf43aSdan   }
30674fccf43aSdan   *ppValue = pIter->apValue[iVal];
30684fccf43aSdan   return SQLITE_OK;
30694fccf43aSdan }
30704fccf43aSdan 
3071296c7658Sdan /*
3072296c7658Sdan ** This function may only be called while the iterator is pointing to an
3073296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
3074296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3075296c7658Sdan **
3076296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3077296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
3078296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3079296c7658Sdan ** was not modified), set *ppValue to NULL.
3080296c7658Sdan **
3081296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3082296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3083296c7658Sdan */
30844fccf43aSdan int sqlite3changeset_new(
3085296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3086296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
30874fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
30884fccf43aSdan ){
3089d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
3090d5f0767cSdan     return SQLITE_MISUSE;
3091d5f0767cSdan   }
30924fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
30934fccf43aSdan     return SQLITE_RANGE;
30944fccf43aSdan   }
30954fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
30964fccf43aSdan   return SQLITE_OK;
30974fccf43aSdan }
30984fccf43aSdan 
3099296c7658Sdan /*
31007aa469cdSdan ** The following two macros are used internally. They are similar to the
31017aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
31027aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
31037aa469cdSdan */
31047aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
31057aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
31067aa469cdSdan 
31077aa469cdSdan /*
3108296c7658Sdan ** This function may only be called with a changeset iterator that has been
3109296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
3110296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
3111296c7658Sdan **
3112296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
3113296c7658Sdan ** containing the iVal'th value of the conflicting record.
3114296c7658Sdan **
3115296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
3116296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
3117296c7658Sdan */
3118d5f0767cSdan int sqlite3changeset_conflict(
3119296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3120296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
3121d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
3122d5f0767cSdan ){
3123d5f0767cSdan   if( !pIter->pConflict ){
3124d5f0767cSdan     return SQLITE_MISUSE;
3125d5f0767cSdan   }
3126ff677b20Sdan   if( iVal<0 || iVal>=pIter->nCol ){
3127d5f0767cSdan     return SQLITE_RANGE;
3128d5f0767cSdan   }
3129d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
3130d5f0767cSdan   return SQLITE_OK;
3131d5f0767cSdan }
3132d5f0767cSdan 
31334fccf43aSdan /*
3134cb3e4b79Sdan ** This function may only be called with an iterator passed to an
3135cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
3136cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
3137cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
3138cb3e4b79Sdan **
3139cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
3140cb3e4b79Sdan */
3141cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
3142cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3143cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
3144cb3e4b79Sdan ){
3145cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
3146cb3e4b79Sdan     return SQLITE_MISUSE;
3147cb3e4b79Sdan   }
3148cb3e4b79Sdan   *pnOut = pIter->nCol;
3149cb3e4b79Sdan   return SQLITE_OK;
3150cb3e4b79Sdan }
3151cb3e4b79Sdan 
3152cb3e4b79Sdan 
3153cb3e4b79Sdan /*
31544fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
31554fccf43aSdan **
31564fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
31574fccf43aSdan ** callback by changeset_apply().
31584fccf43aSdan */
31594fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
3160cbf6d2d2Sdan   int rc = SQLITE_OK;
3161cbf6d2d2Sdan   if( p ){
3162296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
3163cbf6d2d2Sdan     rc = p->rc;
316412ca0b56Sdan     if( p->apValue ){
31654fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
316612ca0b56Sdan     }
3167ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
31684757c658Sdan     sqlite3_free(p->in.buf.aBuf);
31694fccf43aSdan     sqlite3_free(p);
3170cbf6d2d2Sdan   }
31714fccf43aSdan   return rc;
31724fccf43aSdan }
31734fccf43aSdan 
3174fa122adaSdan static int sessionChangesetInvert(
3175fa122adaSdan   SessionInput *pInput,           /* Input changeset */
3176fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3177fa122adaSdan   void *pOut,
317891ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
317991ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
318091ddd559Sdan ){
3181cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
3182fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
3183cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
3184cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
3185cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
3186ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
318791ddd559Sdan 
3188fa122adaSdan   /* Initialize the output buffer */
3189fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
3190fa122adaSdan 
319191ddd559Sdan   /* Zero the output variables in case an error occurs. */
3192fa122adaSdan   if( ppInverted ){
319391ddd559Sdan     *ppInverted = 0;
319491ddd559Sdan     *pnInverted = 0;
3195fa122adaSdan   }
319691ddd559Sdan 
3197fa122adaSdan   while( 1 ){
3198ef7a6304Sdan     u8 eType;
3199fa122adaSdan 
3200fa122adaSdan     /* Test for EOF. */
3201fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
3202fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
3203fa122adaSdan     eType = pInput->aData[pInput->iNext];
3204fa122adaSdan 
320591ddd559Sdan     switch( eType ){
320691ddd559Sdan       case 'T': {
3207244593c8Sdan         /* A 'table' record consists of:
3208244593c8Sdan         **
3209244593c8Sdan         **   * A constant 'T' character,
3210244593c8Sdan         **   * Number of columns in said table (a varint),
3211ef7a6304Sdan         **   * An array of nCol bytes (sPK),
3212244593c8Sdan         **   * A nul-terminated table name.
3213244593c8Sdan         */
3214ef7a6304Sdan         int nByte;
3215fa122adaSdan         int nVar;
3216fa122adaSdan         pInput->iNext++;
3217fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
3218ef7a6304Sdan           goto finished_invert;
3219ef7a6304Sdan         }
3220fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
3221ef7a6304Sdan         sPK.nBuf = 0;
3222fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
3223fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3224fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3225ef7a6304Sdan         if( rc ) goto finished_invert;
3226fa122adaSdan 
3227fa122adaSdan         pInput->iNext += nByte;
3228cfec7eeeSdan         sqlite3_free(apVal);
3229cfec7eeeSdan         apVal = 0;
3230ef7a6304Sdan         abPK = sPK.aBuf;
323191ddd559Sdan         break;
323291ddd559Sdan       }
323391ddd559Sdan 
323491ddd559Sdan       case SQLITE_INSERT:
323591ddd559Sdan       case SQLITE_DELETE: {
323691ddd559Sdan         int nByte;
3237fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
3238fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
3239fa122adaSdan         pInput->iNext += 2;
3240fa122adaSdan         assert( rc==SQLITE_OK );
3241fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
3242fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
3243fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
3244fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3245fa122adaSdan         pInput->iNext += nByte;
3246fa122adaSdan         if( rc ) goto finished_invert;
324791ddd559Sdan         break;
324891ddd559Sdan       }
324991ddd559Sdan 
325091ddd559Sdan       case SQLITE_UPDATE: {
3251cfec7eeeSdan         int iCol;
325291ddd559Sdan 
3253cfec7eeeSdan         if( 0==apVal ){
3254cfec7eeeSdan           apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2);
3255cfec7eeeSdan           if( 0==apVal ){
3256cfec7eeeSdan             rc = SQLITE_NOMEM;
3257cfec7eeeSdan             goto finished_invert;
3258cfec7eeeSdan           }
3259cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3260cfec7eeeSdan         }
326191ddd559Sdan 
3262cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
3263fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3264fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
326591ddd559Sdan 
3266ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
3267fa122adaSdan         pInput->iNext += 2;
3268fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
3269ef7a6304Sdan         if( rc==SQLITE_OK ){
3270fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
3271ef7a6304Sdan         }
3272ef7a6304Sdan 
3273cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
3274cfec7eeeSdan         ** original old.* record, and the other values from the original
3275cfec7eeeSdan         ** new.* record. */
3276e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3277cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
3278fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3279cfec7eeeSdan         }
3280cfec7eeeSdan 
3281cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
3282cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
3283cfec7eeeSdan         ** are set to "undefined". */
3284e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3285cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
3286fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3287cfec7eeeSdan         }
3288cfec7eeeSdan 
3289cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
3290cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
3291cfec7eeeSdan         }
3292cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3293cfec7eeeSdan         if( rc!=SQLITE_OK ){
3294cfec7eeeSdan           goto finished_invert;
3295cfec7eeeSdan         }
3296cfec7eeeSdan 
329791ddd559Sdan         break;
329891ddd559Sdan       }
329991ddd559Sdan 
330091ddd559Sdan       default:
33014757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
3302cfec7eeeSdan         goto finished_invert;
330391ddd559Sdan     }
3304fa122adaSdan 
3305fa122adaSdan     assert( rc==SQLITE_OK );
3306f1a08ad8Sdrh     if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
3307fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3308fa122adaSdan       sOut.nBuf = 0;
3309fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
3310fa122adaSdan     }
331191ddd559Sdan   }
331291ddd559Sdan 
3313cfec7eeeSdan   assert( rc==SQLITE_OK );
3314fa122adaSdan   if( pnInverted ){
3315fa122adaSdan     *pnInverted = sOut.nBuf;
3316fa122adaSdan     *ppInverted = sOut.aBuf;
3317fa122adaSdan     sOut.aBuf = 0;
3318fa122adaSdan   }else if( sOut.nBuf>0 ){
3319fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3320fa122adaSdan   }
3321cfec7eeeSdan 
3322cfec7eeeSdan  finished_invert:
3323fa122adaSdan   sqlite3_free(sOut.aBuf);
3324cfec7eeeSdan   sqlite3_free(apVal);
3325ef7a6304Sdan   sqlite3_free(sPK.aBuf);
3326cfec7eeeSdan   return rc;
332791ddd559Sdan }
332891ddd559Sdan 
3329fa122adaSdan 
3330fa122adaSdan /*
3331fa122adaSdan ** Invert a changeset object.
3332fa122adaSdan */
3333fa122adaSdan int sqlite3changeset_invert(
3334fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
3335fa122adaSdan   const void *pChangeset,         /* Input changeset */
3336fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
3337fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
3338fa122adaSdan ){
3339fa122adaSdan   SessionInput sInput;
3340fa122adaSdan 
3341fa122adaSdan   /* Set up the input stream */
3342fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3343fa122adaSdan   sInput.nData = nChangeset;
3344fa122adaSdan   sInput.aData = (u8*)pChangeset;
3345fa122adaSdan 
3346fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
3347fa122adaSdan }
3348fa122adaSdan 
3349fa122adaSdan /*
3350fa122adaSdan ** Streaming version of sqlite3changeset_invert().
3351fa122adaSdan */
3352f1a08ad8Sdrh int sqlite3changeset_invert_strm(
3353fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
3354fa122adaSdan   void *pIn,
3355fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3356fa122adaSdan   void *pOut
3357fa122adaSdan ){
3358fa122adaSdan   SessionInput sInput;
3359fa122adaSdan   int rc;
3360fa122adaSdan 
3361fa122adaSdan   /* Set up the input stream */
3362fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3363fa122adaSdan   sInput.xInput = xInput;
3364fa122adaSdan   sInput.pIn = pIn;
3365fa122adaSdan 
3366fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
3367fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
3368fa122adaSdan   return rc;
3369fa122adaSdan }
3370fa122adaSdan 
33710c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
33720c698471Sdan struct SessionApplyCtx {
33730c698471Sdan   sqlite3 *db;
33740c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
3375cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
33760c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
33770c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
33780c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
33790c698471Sdan   const char **azCol;             /* Array of column names */
33800c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
3381d1cccf19Sdan   int bStat1;                     /* True if table is sqlite_stat1 */
3382d9151526Sdan   int bDeferConstraints;          /* True to defer constraints */
3383d9151526Sdan   SessionBuffer constraints;      /* Deferred constraints are stored here */
33840c698471Sdan };
33850c698471Sdan 
3386d5f0767cSdan /*
3387d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
3388d5f0767cSdan ** structure like this:
3389d5f0767cSdan **
3390d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3391d5f0767cSdan **
3392d5f0767cSdan ** The DELETE statement looks like this:
3393d5f0767cSdan **
3394db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
3395d5f0767cSdan **
3396d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
3397d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
3398d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
3399296c7658Sdan **
3400296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
3401296c7658Sdan ** pointing to the prepared version of the SQL statement.
3402d5f0767cSdan */
3403d5f0767cSdan static int sessionDeleteRow(
3404d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3405d5f0767cSdan   const char *zTab,               /* Table name */
34060c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3407d5f0767cSdan ){
3408296c7658Sdan   int i;
3409296c7658Sdan   const char *zSep = "";
3410d5f0767cSdan   int rc = SQLITE_OK;
3411d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
34127cf7df7dSdan   int nPk = 0;
3413d5f0767cSdan 
3414d5f0767cSdan   sessionAppendStr(&buf, "DELETE FROM ", &rc);
3415d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3416296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
3417296c7658Sdan 
3418296c7658Sdan   for(i=0; i<p->nCol; i++){
3419296c7658Sdan     if( p->abPK[i] ){
34207cf7df7dSdan       nPk++;
3421296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
3422296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3423296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
3424296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
3425296c7658Sdan       zSep = " AND ";
3426296c7658Sdan     }
3427296c7658Sdan   }
3428296c7658Sdan 
34297cf7df7dSdan   if( nPk<p->nCol ){
3430296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
3431296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
3432296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
3433296c7658Sdan 
3434296c7658Sdan     zSep = "";
3435296c7658Sdan     for(i=0; i<p->nCol; i++){
3436296c7658Sdan       if( !p->abPK[i] ){
3437296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
3438296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
3439296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
3440296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
3441296c7658Sdan         zSep = "AND ";
3442296c7658Sdan       }
3443296c7658Sdan     }
3444296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
34457cf7df7dSdan   }
3446d5f0767cSdan 
3447d5f0767cSdan   if( rc==SQLITE_OK ){
34480c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
3449d5f0767cSdan   }
3450d5f0767cSdan   sqlite3_free(buf.aBuf);
3451d5f0767cSdan 
3452d5f0767cSdan   return rc;
3453d5f0767cSdan }
3454d5f0767cSdan 
3455d5f0767cSdan /*
3456d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
3457d5f0767cSdan ** Assuming a table structure like this:
3458d5f0767cSdan **
3459d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3460d5f0767cSdan **
3461d5f0767cSdan ** The UPDATE statement looks like this:
3462d5f0767cSdan **
3463d5f0767cSdan **     UPDATE x SET
3464d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3465964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3466964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3467964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3468d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3469964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3470d5f0767cSdan **     )
3471d5f0767cSdan **
3472d5f0767cSdan ** For each column in the table, there are three variables to bind:
3473d5f0767cSdan **
3474d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3475d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3476d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3477d5f0767cSdan **
3478d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3479d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3480d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3481d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3482d5f0767cSdan **
3483296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3484296c7658Sdan ** pointing to the prepared version of the SQL statement.
3485d5f0767cSdan */
3486d5f0767cSdan static int sessionUpdateRow(
3487d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3488d5f0767cSdan   const char *zTab,               /* Table name */
34890c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3490d5f0767cSdan ){
3491d5f0767cSdan   int rc = SQLITE_OK;
3492d5f0767cSdan   int i;
3493d5f0767cSdan   const char *zSep = "";
3494d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3495d5f0767cSdan 
3496d5f0767cSdan   /* Append "UPDATE tbl SET " */
3497d5f0767cSdan   sessionAppendStr(&buf, "UPDATE ", &rc);
3498d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3499d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3500d5f0767cSdan 
3501d5f0767cSdan   /* Append the assignments */
35020c698471Sdan   for(i=0; i<p->nCol; i++){
3503d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
35040c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3505d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3506d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3507d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3508d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3509d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
35100c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3511d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3512d5f0767cSdan     zSep = ", ";
3513d5f0767cSdan   }
3514d5f0767cSdan 
3515d5f0767cSdan   /* Append the PK part of the WHERE clause */
3516d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
35170c698471Sdan   for(i=0; i<p->nCol; i++){
35180c698471Sdan     if( p->abPK[i] ){
35190c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3520d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3521d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3522d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3523d5f0767cSdan     }
3524d5f0767cSdan   }
3525d5f0767cSdan 
3526d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3527d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
35280c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3529d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
35300c698471Sdan   for(i=0; i<p->nCol; i++){
35310c698471Sdan     if( !p->abPK[i] ){
3532d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3533d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3534d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
35350c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3536d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3537d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3538d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3539d5f0767cSdan     }
3540d5f0767cSdan   }
3541d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3542d5f0767cSdan 
3543d5f0767cSdan   if( rc==SQLITE_OK ){
35440c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3545d5f0767cSdan   }
3546d5f0767cSdan   sqlite3_free(buf.aBuf);
3547d5f0767cSdan 
3548d5f0767cSdan   return rc;
3549d5f0767cSdan }
3550d5f0767cSdan 
35513739f298Sdan 
3552296c7658Sdan /*
3553296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3554296c7658Sdan ** key. Assuming the following table structure:
3555296c7658Sdan **
3556296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3557296c7658Sdan **
3558296c7658Sdan ** The SELECT statement looks like this:
3559296c7658Sdan **
3560296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3561296c7658Sdan **
3562296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3563296c7658Sdan ** pointing to the prepared version of the SQL statement.
3564296c7658Sdan */
3565d5f0767cSdan static int sessionSelectRow(
3566d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3567d5f0767cSdan   const char *zTab,               /* Table name */
35680c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3569d5f0767cSdan ){
3570d7fb7d24Sdan   return sessionSelectStmt(
3571d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3572d5f0767cSdan }
3573d5f0767cSdan 
3574296c7658Sdan /*
3575296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3576296c7658Sdan ** For example:
3577296c7658Sdan **
3578296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3579296c7658Sdan **
3580296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3581296c7658Sdan ** pointing to the prepared version of the SQL statement.
3582296c7658Sdan */
35830c698471Sdan static int sessionInsertRow(
35840c698471Sdan   sqlite3 *db,                    /* Database handle */
35850c698471Sdan   const char *zTab,               /* Table name */
35860c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
35870c698471Sdan ){
35880c698471Sdan   int rc = SQLITE_OK;
35890c698471Sdan   int i;
35900c698471Sdan   SessionBuffer buf = {0, 0, 0};
35910c698471Sdan 
35920c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
35930c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
3594ff677b20Sdan   sessionAppendStr(&buf, "(", &rc);
3595ff677b20Sdan   for(i=0; i<p->nCol; i++){
3596ff677b20Sdan     if( i!=0 ) sessionAppendStr(&buf, ", ", &rc);
3597ff677b20Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3598ff677b20Sdan   }
3599ff677b20Sdan 
3600ff677b20Sdan   sessionAppendStr(&buf, ") VALUES(?", &rc);
36010c698471Sdan   for(i=1; i<p->nCol; i++){
36020c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
36030c698471Sdan   }
36040c698471Sdan   sessionAppendStr(&buf, ")", &rc);
36050c698471Sdan 
36060c698471Sdan   if( rc==SQLITE_OK ){
36070c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
36080c698471Sdan   }
36090c698471Sdan   sqlite3_free(buf.aBuf);
36100c698471Sdan   return rc;
36110c698471Sdan }
36120c698471Sdan 
36133739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){
36143739f298Sdan   return sqlite3_prepare_v2(db, zSql, -1, pp, 0);
36153739f298Sdan }
36163739f298Sdan 
36173739f298Sdan /*
36183739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table.
36193739f298Sdan ** These are similar to those created by sessionSelectRow(),
36203739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for
36213739f298Sdan ** other tables.
36223739f298Sdan */
36233739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){
36243739f298Sdan   int rc = sessionSelectRow(db, "sqlite_stat1", p);
36253739f298Sdan   if( rc==SQLITE_OK ){
36263739f298Sdan     rc = sessionPrepare(db, &p->pInsert,
36273739f298Sdan         "INSERT INTO main.sqlite_stat1 VALUES(?1, "
36283739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, "
36293739f298Sdan         "?3)"
36303739f298Sdan     );
36313739f298Sdan   }
36323739f298Sdan   if( rc==SQLITE_OK ){
36333739f298Sdan     rc = sessionPrepare(db, &p->pUpdate,
36343739f298Sdan         "UPDATE main.sqlite_stat1 SET "
36353739f298Sdan         "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, "
36363739f298Sdan         "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, "
36373739f298Sdan         "stat = CASE WHEN ?8 THEN ?9 ELSE stat END  "
36383739f298Sdan         "WHERE tbl=?1 AND idx IS "
36393739f298Sdan         "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END "
36403739f298Sdan         "AND (?10 OR ?8=0 OR stat IS ?7)"
36413739f298Sdan     );
36423739f298Sdan   }
36433739f298Sdan   if( rc==SQLITE_OK ){
36443739f298Sdan     rc = sessionPrepare(db, &p->pDelete,
36453739f298Sdan         "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS "
36463739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END "
36473739f298Sdan         "AND (?4 OR stat IS ?3)"
36483739f298Sdan     );
36493739f298Sdan   }
36503739f298Sdan   assert( rc==SQLITE_OK );
36513739f298Sdan   return rc;
36523739f298Sdan }
36533739f298Sdan 
3654296c7658Sdan /*
36557aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
36567aa469cdSdan ** See comments in the body of this function for details.
36577aa469cdSdan */
36587aa469cdSdan static int sessionBindValue(
36597aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
36607aa469cdSdan   int i,                          /* Parameter number to bind to */
36617aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
36627aa469cdSdan ){
36635671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3664082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3665082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3666082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3667082c96dfSdan   ** set to SQLITE_NULL.  */
36685671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
36697aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
36707aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3671082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
36727aa469cdSdan     return SQLITE_NOMEM;
36737aa469cdSdan   }
36747aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
36757aa469cdSdan }
36767aa469cdSdan 
36777aa469cdSdan /*
3678db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3679db04571cSdan ** transfers new.* values from the current iterator entry to statement
3680db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3681db04571cSdan **
3682d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of
3683db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3684db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3685db04571cSdan ** to an array nCol elements in size. In this case only those values for
3686db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3687db04571cSdan ** statement.
3688db04571cSdan **
3689db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3690db04571cSdan */
36917aa469cdSdan static int sessionBindRow(
3692db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
36937aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3694db04571cSdan   int nCol,                       /* Number of columns */
3695db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3696db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3697db04571cSdan ){
3698db04571cSdan   int i;
3699db04571cSdan   int rc = SQLITE_OK;
37007aa469cdSdan 
37017aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
37027aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
37037aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
37047aa469cdSdan   ** in the code below. */
37057aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
37067aa469cdSdan 
3707db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3708db04571cSdan     if( !abPK || abPK[i] ){
3709db04571cSdan       sqlite3_value *pVal;
37107aa469cdSdan       (void)xValue(pIter, i, &pVal);
37117aa469cdSdan       rc = sessionBindValue(pStmt, i+1, pVal);
3712db04571cSdan     }
3713db04571cSdan   }
3714db04571cSdan   return rc;
3715db04571cSdan }
3716db04571cSdan 
3717db04571cSdan /*
3718296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3719296c7658Sdan ** This function binds the primary key values from the change that changeset
3720296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3721296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3722296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3723296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
37247aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
37257aa469cdSdan **
37267aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
37277aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
37287aa469cdSdan ** not require a reset().
3729296c7658Sdan **
3730296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3731db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3732db04571cSdan ** UPDATE, bind values from the old.* record.
3733296c7658Sdan */
37340c698471Sdan static int sessionSeekToRow(
373537f133ecSdan   sqlite3 *db,                    /* Database handle */
373637f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
373737f133ecSdan   u8 *abPK,                       /* Primary key flags array */
37380c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
373937f133ecSdan ){
37407aa469cdSdan   int rc;                         /* Return code */
3741296c7658Sdan   int nCol;                       /* Number of columns in table */
3742296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3743296c7658Sdan   const char *zDummy;             /* Unused */
374437f133ecSdan 
3745b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
37467aa469cdSdan   rc = sessionBindRow(pIter,
3747db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3748db04571cSdan       nCol, abPK, pSelect
3749db04571cSdan   );
37500c698471Sdan 
37510c698471Sdan   if( rc==SQLITE_OK ){
37520c698471Sdan     rc = sqlite3_step(pSelect);
37530c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
37540c698471Sdan   }
37550c698471Sdan 
37560c698471Sdan   return rc;
37570c698471Sdan }
37580c698471Sdan 
3759296c7658Sdan /*
3760296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3761296c7658Sdan ** currently points to.
3762296c7658Sdan **
3763296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3764296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3765296c7658Sdan ** depends solely on eType, as follows:
3766296c7658Sdan **
3767296c7658Sdan **    eType value                 Value passed to xConflict
3768296c7658Sdan **    -------------------------------------------------
3769296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3770296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3771296c7658Sdan **
3772296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3773296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3774296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3775296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3776296c7658Sdan ** handler invoked is as follows:
3777296c7658Sdan **
3778296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3779296c7658Sdan **    ----------------------------------------------------------------
3780296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3781296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3782296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3783296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3784296c7658Sdan **
3785296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3786296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3787296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3788296c7658Sdan **
3789296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3790296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3791296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3792296c7658Sdan ** this function returns SQLITE_OK.
3793296c7658Sdan */
37940c698471Sdan static int sessionConflictHandler(
3795296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3796296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
37970c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
37980c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3799296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3800296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
38010c698471Sdan ){
380274f598b6Smistachkin   int res = 0;                    /* Value returned by conflict handler */
38030c698471Sdan   int rc;
38040c698471Sdan   int nCol;
38050c698471Sdan   int op;
38060c698471Sdan   const char *zDummy;
38070c698471Sdan 
3808b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
38090c698471Sdan 
38100c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
38110c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
38120c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
381337f133ecSdan 
381437f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
38150c698471Sdan   if( pbReplace ){
38160c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
38170c698471Sdan   }else{
3818db04571cSdan     rc = SQLITE_OK;
38190c698471Sdan   }
38200c698471Sdan 
38210c698471Sdan   if( rc==SQLITE_ROW ){
38220c698471Sdan     /* There exists another row with the new.* primary key. */
38230c698471Sdan     pIter->pConflict = p->pSelect;
38240c698471Sdan     res = xConflict(pCtx, eType, pIter);
38250c698471Sdan     pIter->pConflict = 0;
38260c698471Sdan     rc = sqlite3_reset(p->pSelect);
3827db04571cSdan   }else if( rc==SQLITE_OK ){
3828d9151526Sdan     if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){
3829d9151526Sdan       /* Instead of invoking the conflict handler, append the change blob
3830d9151526Sdan       ** to the SessionApplyCtx.constraints buffer. */
3831d9151526Sdan       u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent];
3832d9151526Sdan       int nBlob = pIter->in.iNext - pIter->in.iCurrent;
3833d9151526Sdan       sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc);
3834d9151526Sdan       res = SQLITE_CHANGESET_OMIT;
3835d9151526Sdan     }else{
38360c698471Sdan       /* No other row with the new.* primary key. */
38370c698471Sdan       res = xConflict(pCtx, eType+1, pIter);
38380c698471Sdan       if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
383937f133ecSdan     }
3840d9151526Sdan   }
384137f133ecSdan 
384237f133ecSdan   if( rc==SQLITE_OK ){
38430c698471Sdan     switch( res ){
38440c698471Sdan       case SQLITE_CHANGESET_REPLACE:
3845f51e5f6cSdan         assert( pbReplace );
3846f51e5f6cSdan         *pbReplace = 1;
38470c698471Sdan         break;
38480c698471Sdan 
38490c698471Sdan       case SQLITE_CHANGESET_OMIT:
38500c698471Sdan         break;
38510c698471Sdan 
38520c698471Sdan       case SQLITE_CHANGESET_ABORT:
38530c698471Sdan         rc = SQLITE_ABORT;
38540c698471Sdan         break;
38550c698471Sdan 
38560c698471Sdan       default:
38570c698471Sdan         rc = SQLITE_MISUSE;
38580c698471Sdan         break;
38590c698471Sdan     }
38600c698471Sdan   }
38610c698471Sdan 
38620c698471Sdan   return rc;
38630c698471Sdan }
38640c698471Sdan 
3865296c7658Sdan /*
3866296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
3867296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
3868296c7658Sdan ** the conflict handler callback.
3869296c7658Sdan **
3870296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
3871296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
3872296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
3873296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
3874296c7658Sdan ** to true before returning. In this case the caller will invoke this function
3875296c7658Sdan ** again, this time with pbRetry set to NULL.
3876296c7658Sdan **
3877296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
3878296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
3879296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
3880296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
3881296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
3882296c7658Sdan ** row before invoking this function again, this time with pbReplace set
3883296c7658Sdan ** to NULL.
3884296c7658Sdan **
3885296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
3886296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
3887296c7658Sdan ** returned.
3888296c7658Sdan */
38890c698471Sdan static int sessionApplyOneOp(
3890296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3891296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
38920c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
3893296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
3894296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
3895296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
38960c698471Sdan ){
38970c698471Sdan   const char *zDummy;
38980c698471Sdan   int op;
38990c698471Sdan   int nCol;
39000c698471Sdan   int rc = SQLITE_OK;
39010c698471Sdan 
39020c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
39030c698471Sdan   assert( p->azCol && p->abPK );
39040c698471Sdan   assert( !pbReplace || *pbReplace==0 );
39050c698471Sdan 
3906b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
39070c698471Sdan 
39080c698471Sdan   if( op==SQLITE_DELETE ){
39090c698471Sdan 
391073b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
391173b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
391273b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
391373b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
391473b3c055Sdan     ** is not required if:
391573b3c055Sdan     **
391673b3c055Sdan     **   * this is a patchset, or
391773b3c055Sdan     **   * (pbRetry==0), or
391873b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
391973b3c055Sdan     **     no (nCol+1) variable to bind to).
392073b3c055Sdan     */
392173b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
392273b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
39237cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
392473b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
39257cf7df7dSdan     }
39260c698471Sdan     if( rc!=SQLITE_OK ) return rc;
39270c698471Sdan 
39280c698471Sdan     sqlite3_step(p->pDelete);
39290c698471Sdan     rc = sqlite3_reset(p->pDelete);
39300c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
39310c698471Sdan       rc = sessionConflictHandler(
39320c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
39330c698471Sdan       );
393435e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
39350c698471Sdan       rc = sessionConflictHandler(
39360c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
39370c698471Sdan       );
39380c698471Sdan     }
39390c698471Sdan 
39400c698471Sdan   }else if( op==SQLITE_UPDATE ){
39410c698471Sdan     int i;
39420c698471Sdan 
39430c698471Sdan     /* Bind values to the UPDATE statement. */
39440c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
39457aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
39467aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
39477aa469cdSdan 
39480c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
39497aa469cdSdan       if( pOld ){
39507aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
39517aa469cdSdan       }
39527aa469cdSdan       if( rc==SQLITE_OK && pNew ){
39537aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
39540c698471Sdan       }
39550c698471Sdan     }
395673b3c055Sdan     if( rc==SQLITE_OK ){
395773b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
395873b3c055Sdan     }
39590c698471Sdan     if( rc!=SQLITE_OK ) return rc;
39600c698471Sdan 
39610c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
39620c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
39630c698471Sdan     sqlite3_step(p->pUpdate);
39640c698471Sdan     rc = sqlite3_reset(p->pUpdate);
39650c698471Sdan 
39660c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
39670c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
39680c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
39690c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
39700c698471Sdan 
39710c698471Sdan       rc = sessionConflictHandler(
39720c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
39730c698471Sdan       );
39740c698471Sdan 
397535e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
3976db04571cSdan       /* This is always a CONSTRAINT conflict. */
3977db04571cSdan       rc = sessionConflictHandler(
3978db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
39790c698471Sdan       );
39800c698471Sdan     }
39810c698471Sdan 
39820c698471Sdan   }else{
39830c698471Sdan     assert( op==SQLITE_INSERT );
3984d1cccf19Sdan     if( p->bStat1 ){
3985d1cccf19Sdan       /* Check if there is a conflicting row. For sqlite_stat1, this needs
3986d1cccf19Sdan       ** to be done using a SELECT, as there is no PRIMARY KEY in the
3987d1cccf19Sdan       ** database schema to throw an exception if a duplicate is inserted.  */
3988d1cccf19Sdan       rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
3989d1cccf19Sdan       if( rc==SQLITE_ROW ){
3990d1cccf19Sdan         rc = SQLITE_CONSTRAINT;
3991d1cccf19Sdan         sqlite3_reset(p->pSelect);
3992d1cccf19Sdan       }
3993d1cccf19Sdan     }
3994d1cccf19Sdan 
3995d1cccf19Sdan     if( rc==SQLITE_OK ){
39967aa469cdSdan       rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
39970c698471Sdan       if( rc!=SQLITE_OK ) return rc;
39980c698471Sdan 
39990c698471Sdan       sqlite3_step(p->pInsert);
40000c698471Sdan       rc = sqlite3_reset(p->pInsert);
4001d1cccf19Sdan     }
4002d1cccf19Sdan 
400335e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
40040c698471Sdan       rc = sessionConflictHandler(
40050c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
40060c698471Sdan       );
400737f133ecSdan     }
400837f133ecSdan   }
400937f133ecSdan 
401037f133ecSdan   return rc;
401137f133ecSdan }
401237f133ecSdan 
40135f5663dcSdan /*
40145f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument
40155f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke
40165f5663dcSdan ** the conflict handler callback.
40175f5663dcSdan **
40185f5663dcSdan ** The difference between this function and sessionApplyOne() is that this
40195f5663dcSdan ** function handles the case where the conflict-handler is invoked and
40205f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be
40215f5663dcSdan ** retried in some manner.
40225f5663dcSdan */
4023d9151526Sdan static int sessionApplyOneWithRetry(
4024d9151526Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4025d9151526Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator to read change from */
4026d9151526Sdan   SessionApplyCtx *pApply,        /* Apply context */
4027d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4028d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4029d9151526Sdan ){
4030d9151526Sdan   int bReplace = 0;
4031d9151526Sdan   int bRetry = 0;
4032d9151526Sdan   int rc;
4033d9151526Sdan 
4034d9151526Sdan   rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry);
40355f5663dcSdan   assert( rc==SQLITE_OK || (bRetry==0 && bReplace==0) );
4036d9151526Sdan 
40375f5663dcSdan   /* If the bRetry flag is set, the change has not been applied due to an
40385f5663dcSdan   ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and
40395f5663dcSdan   ** a row with the correct PK is present in the db, but one or more other
40405f5663dcSdan   ** fields do not contain the expected values) and the conflict handler
40415f5663dcSdan   ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation,
40425f5663dcSdan   ** but pass NULL as the final argument so that sessionApplyOneOp() ignores
40435f5663dcSdan   ** the SQLITE_CHANGESET_DATA problem.  */
40445f5663dcSdan   if( bRetry ){
40455f5663dcSdan     assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE );
40465f5663dcSdan     rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4047d9151526Sdan   }
4048d9151526Sdan 
40495f5663dcSdan   /* If the bReplace flag is set, the change is an INSERT that has not
40505f5663dcSdan   ** been performed because the database already contains a row with the
40515f5663dcSdan   ** specified primary key and the conflict handler returned
40525f5663dcSdan   ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row
40535f5663dcSdan   ** before reattempting the INSERT.  */
40545f5663dcSdan   else if( bReplace ){
4055d9151526Sdan     assert( pIter->op==SQLITE_INSERT );
4056d9151526Sdan     rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
4057d9151526Sdan     if( rc==SQLITE_OK ){
4058d9151526Sdan       rc = sessionBindRow(pIter,
4059d9151526Sdan           sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete);
4060d9151526Sdan       sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1);
4061d9151526Sdan     }
4062d9151526Sdan     if( rc==SQLITE_OK ){
4063d9151526Sdan       sqlite3_step(pApply->pDelete);
4064d9151526Sdan       rc = sqlite3_reset(pApply->pDelete);
4065d9151526Sdan     }
4066d9151526Sdan     if( rc==SQLITE_OK ){
4067d9151526Sdan       rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4068d9151526Sdan     }
4069d9151526Sdan     if( rc==SQLITE_OK ){
4070d9151526Sdan       rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
4071d9151526Sdan     }
4072d9151526Sdan   }
4073d9151526Sdan 
4074d9151526Sdan   return rc;
4075d9151526Sdan }
4076d9151526Sdan 
4077d9151526Sdan /*
4078d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer.
4079d9151526Sdan */
4080d9151526Sdan static int sessionRetryConstraints(
4081d9151526Sdan   sqlite3 *db,
4082d9151526Sdan   int bPatchset,
4083d9151526Sdan   const char *zTab,
4084d9151526Sdan   SessionApplyCtx *pApply,
4085d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4086d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4087d9151526Sdan ){
4088d9151526Sdan   int rc = SQLITE_OK;
4089d9151526Sdan 
4090d9151526Sdan   while( pApply->constraints.nBuf ){
4091d9151526Sdan     sqlite3_changeset_iter *pIter2 = 0;
4092d9151526Sdan     SessionBuffer cons = pApply->constraints;
4093d9151526Sdan     memset(&pApply->constraints, 0, sizeof(SessionBuffer));
4094d9151526Sdan 
4095d9151526Sdan     rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf);
4096d9151526Sdan     if( rc==SQLITE_OK ){
4097d9151526Sdan       int nByte = 2*pApply->nCol*sizeof(sqlite3_value*);
4098d9151526Sdan       int rc2;
4099d9151526Sdan       pIter2->bPatchset = bPatchset;
4100d9151526Sdan       pIter2->zTab = (char*)zTab;
4101d9151526Sdan       pIter2->nCol = pApply->nCol;
4102d9151526Sdan       pIter2->abPK = pApply->abPK;
4103d9151526Sdan       sessionBufferGrow(&pIter2->tblhdr, nByte, &rc);
4104d9151526Sdan       pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf;
4105d9151526Sdan       if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte);
4106d9151526Sdan 
4107d9151526Sdan       while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){
4108d9151526Sdan         rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx);
4109d9151526Sdan       }
4110d9151526Sdan 
4111d9151526Sdan       rc2 = sqlite3changeset_finalize(pIter2);
41127e0765a9Sdrh       if( rc==SQLITE_OK ) rc = rc2;
4113d9151526Sdan     }
4114d9151526Sdan     assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 );
4115d9151526Sdan 
4116d9151526Sdan     sqlite3_free(cons.aBuf);
4117d9151526Sdan     if( rc!=SQLITE_OK ) break;
4118d9151526Sdan     if( pApply->constraints.nBuf>=cons.nBuf ){
4119d9151526Sdan       /* No progress was made on the last round. */
4120d9151526Sdan       pApply->bDeferConstraints = 0;
4121d9151526Sdan     }
4122d9151526Sdan   }
4123d9151526Sdan 
4124d9151526Sdan   return rc;
4125d9151526Sdan }
4126d9151526Sdan 
4127296c7658Sdan /*
41284757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
41294757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
41304757c658Sdan ** changeset to the main database attached to handle "db". The supplied
41314757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
41324757c658Sdan ** while applying the change.
4133296c7658Sdan */
41344757c658Sdan static int sessionChangesetApply(
4135296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
41364757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
413740368988Sdan   int(*xFilter)(
413840368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
413940368988Sdan     const char *zTab              /* Table name */
414040368988Sdan   ),
4141d5f0767cSdan   int(*xConflict)(
4142d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
4143d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4144d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4145d5f0767cSdan   ),
4146296c7658Sdan   void *pCtx                      /* First argument passed to xConflict */
4147d5f0767cSdan ){
4148ca62ad57Sdan   int schemaMismatch = 0;
4149296c7658Sdan   int rc;                         /* Return code */
4150d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
4151cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
4152296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
41535f5663dcSdan   int bPatchset;
4154d5f0767cSdan 
4155082c96dfSdan   assert( xConflict!=0 );
4156082c96dfSdan 
4157d9151526Sdan   pIter->in.bNoDiscard = 1;
41580c698471Sdan   memset(&sApply, 0, sizeof(sApply));
41594c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
41600c698471Sdan   rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
4161cb3e4b79Sdan   if( rc==SQLITE_OK ){
4162cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
4163cb3e4b79Sdan   }
41640c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
41650c698471Sdan     int nCol;
4166d5f0767cSdan     int op;
41670c698471Sdan     const char *zNew;
4168ca62ad57Sdan 
4169b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
4170d5f0767cSdan 
41710c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
4172ca62ad57Sdan       u8 *abPK;
4173ca62ad57Sdan 
4174d9151526Sdan       rc = sessionRetryConstraints(
4175d9151526Sdan           db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx
4176d9151526Sdan       );
4177d9151526Sdan       if( rc!=SQLITE_OK ) break;
4178d9151526Sdan 
4179cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
41800c698471Sdan       sqlite3_finalize(sApply.pDelete);
41810c698471Sdan       sqlite3_finalize(sApply.pUpdate);
41820c698471Sdan       sqlite3_finalize(sApply.pInsert);
41830c698471Sdan       sqlite3_finalize(sApply.pSelect);
41840c698471Sdan       memset(&sApply, 0, sizeof(sApply));
41850c698471Sdan       sApply.db = db;
4186d9151526Sdan       sApply.bDeferConstraints = 1;
418737f133ecSdan 
418840368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
418940368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
419040368988Sdan       ** non-zero, proceed. */
419140368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
419240368988Sdan       if( schemaMismatch ){
419340368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
4194f05ac112Sdan         if( zTab==0 ){
4195f05ac112Sdan           rc = SQLITE_NOMEM;
4196f05ac112Sdan           break;
4197f05ac112Sdan         }
41984f528042Sdan         nTab = (int)strlen(zTab);
419940368988Sdan         sApply.azCol = (const char **)zTab;
420040368988Sdan       }else{
4201ff677b20Sdan         int nMinCol = 0;
4202ff677b20Sdan         int i;
4203ff677b20Sdan 
4204ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
4205296c7658Sdan         rc = sessionTableInfo(
4206ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
4207ca62ad57Sdan         );
4208ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
4209ff677b20Sdan         for(i=0; i<sApply.nCol; i++){
4210ff677b20Sdan           if( sApply.abPK[i] ) nMinCol = i+1;
4211ff677b20Sdan         }
42120c698471Sdan 
4213ca62ad57Sdan         if( sApply.nCol==0 ){
4214ca62ad57Sdan           schemaMismatch = 1;
4215ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4216ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
4217ca62ad57Sdan           );
4218ca62ad57Sdan         }
4219ff677b20Sdan         else if( sApply.nCol<nCol ){
4220ca62ad57Sdan           schemaMismatch = 1;
4221ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4222ff677b20Sdan               "sqlite3changeset_apply(): table %s has %d columns, "
4223ff677b20Sdan               "expected %d or more",
4224ca62ad57Sdan               zTab, sApply.nCol, nCol
4225ca62ad57Sdan           );
4226ca62ad57Sdan         }
4227ff677b20Sdan         else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){
4228ca62ad57Sdan           schemaMismatch = 1;
422940368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
423040368988Sdan               "primary key mismatch for table %s", zTab
4231ca62ad57Sdan           );
4232ca62ad57Sdan         }
4233ff677b20Sdan         else{
4234ff677b20Sdan           sApply.nCol = nCol;
42353739f298Sdan           if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){
42363739f298Sdan             if( (rc = sessionStat1Sql(db, &sApply) ) ){
42373739f298Sdan               break;
42383739f298Sdan             }
4239d1cccf19Sdan             sApply.bStat1 = 1;
4240d1cccf19Sdan           }else{
4241ff677b20Sdan             if((rc = sessionSelectRow(db, zTab, &sApply))
42420c698471Sdan                 || (rc = sessionUpdateRow(db, zTab, &sApply))
42430c698471Sdan                 || (rc = sessionDeleteRow(db, zTab, &sApply))
42440c698471Sdan                 || (rc = sessionInsertRow(db, zTab, &sApply))
424537f133ecSdan               ){
424637f133ecSdan               break;
424737f133ecSdan             }
4248d1cccf19Sdan             sApply.bStat1 = 0;
4249d1cccf19Sdan           }
4250ff677b20Sdan         }
4251cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
4252d5f0767cSdan       }
425340368988Sdan     }
4254d5f0767cSdan 
4255ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
4256ca62ad57Sdan     ** next change. A log message has already been issued. */
4257ca62ad57Sdan     if( schemaMismatch ) continue;
4258ca62ad57Sdan 
4259d9151526Sdan     rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx);
42600c698471Sdan   }
42610c698471Sdan 
42625f5663dcSdan   bPatchset = pIter->bPatchset;
4263296c7658Sdan   if( rc==SQLITE_OK ){
4264296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
4265296c7658Sdan   }else{
4266296c7658Sdan     sqlite3changeset_finalize(pIter);
4267296c7658Sdan   }
4268d5f0767cSdan 
4269d5f0767cSdan   if( rc==SQLITE_OK ){
42705f5663dcSdan     rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx);
42715f5663dcSdan   }
42725f5663dcSdan 
42735f5663dcSdan   if( rc==SQLITE_OK ){
427407001c45Sdrh     int nFk, notUsed;
427507001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
427607001c45Sdrh     if( nFk!=0 ){
4277cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
4278cb3e4b79Sdan       sqlite3_changeset_iter sIter;
4279cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
4280cb3e4b79Sdan       sIter.nCol = nFk;
4281cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
4282cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
4283cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
4284cb3e4b79Sdan       }
4285cb3e4b79Sdan     }
4286cb3e4b79Sdan   }
4287cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
4288cb3e4b79Sdan 
4289cb3e4b79Sdan   if( rc==SQLITE_OK ){
4290d5f0767cSdan     rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4291d5f0767cSdan   }else{
4292d5f0767cSdan     sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
4293d5f0767cSdan     sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4294d5f0767cSdan   }
4295d5f0767cSdan 
42960c698471Sdan   sqlite3_finalize(sApply.pInsert);
42970c698471Sdan   sqlite3_finalize(sApply.pDelete);
42980c698471Sdan   sqlite3_finalize(sApply.pUpdate);
42990c698471Sdan   sqlite3_finalize(sApply.pSelect);
4300cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
4301d9151526Sdan   sqlite3_free((char*)sApply.constraints.aBuf);
43024c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
4303d5f0767cSdan   return rc;
4304d5f0767cSdan }
430591ddd559Sdan 
430677fc1d5bSdan /*
43074757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
43084757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
43094757c658Sdan ** to resolve any conflicts encountered while applying the change.
43104757c658Sdan */
43114757c658Sdan int sqlite3changeset_apply(
43124757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
43134757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
43144757c658Sdan   void *pChangeset,               /* Changeset blob */
43154757c658Sdan   int(*xFilter)(
43164757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
43174757c658Sdan     const char *zTab              /* Table name */
43184757c658Sdan   ),
43194757c658Sdan   int(*xConflict)(
43204757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
43214757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
43224757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
43234757c658Sdan   ),
43244757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
43254757c658Sdan ){
43264757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
43274757c658Sdan   int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset);
43284757c658Sdan   if( rc==SQLITE_OK ){
43294757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
43304757c658Sdan   }
43314757c658Sdan   return rc;
43324757c658Sdan }
43334757c658Sdan 
43344757c658Sdan /*
43354757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
43364757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
43374757c658Sdan ** to resolve any conflicts encountered while applying the change.
43384757c658Sdan */
4339f1a08ad8Sdrh int sqlite3changeset_apply_strm(
43404757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
43414757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
43424757c658Sdan   void *pIn,                                          /* First arg for xInput */
43434757c658Sdan   int(*xFilter)(
43444757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
43454757c658Sdan     const char *zTab              /* Table name */
43464757c658Sdan   ),
43474757c658Sdan   int(*xConflict)(
43484757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
43494757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
43504757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
43514757c658Sdan   ),
43524757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
43534757c658Sdan ){
43544757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
4355f1a08ad8Sdrh   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
43564757c658Sdan   if( rc==SQLITE_OK ){
43574757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
43584757c658Sdan   }
43594757c658Sdan   return rc;
43604757c658Sdan }
43614757c658Sdan 
43624757c658Sdan /*
43635898ad69Sdan ** sqlite3_changegroup handle.
43645898ad69Sdan */
43655898ad69Sdan struct sqlite3_changegroup {
43665898ad69Sdan   int rc;                         /* Error code */
43675898ad69Sdan   int bPatch;                     /* True to accumulate patchsets */
43685898ad69Sdan   SessionTable *pList;            /* List of tables in current patch */
43695898ad69Sdan };
43705898ad69Sdan 
43715898ad69Sdan /*
437277fc1d5bSdan ** This function is called to merge two changes to the same row together as
437377fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
437477fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
437577fc1d5bSdan */
43765d607a6eSdan static int sessionChangeMerge(
437777fc1d5bSdan   SessionTable *pTab,             /* Table structure */
437864277f4aSdan   int bPatchset,                  /* True for patchsets */
437977fc1d5bSdan   SessionChange *pExist,          /* Existing change */
438077fc1d5bSdan   int op2,                        /* Second change operation */
438177fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
438277fc1d5bSdan   u8 *aRec,                       /* Second change record */
438377fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
438477fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
43855d607a6eSdan ){
43865d607a6eSdan   SessionChange *pNew = 0;
43875d607a6eSdan 
43885d607a6eSdan   if( !pExist ){
4389cbf6d2d2Sdan     pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec);
43905d607a6eSdan     if( !pNew ){
43915d607a6eSdan       return SQLITE_NOMEM;
43925d607a6eSdan     }
43935d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
4394798693b2Sdan     pNew->op = op2;
43955d607a6eSdan     pNew->bIndirect = bIndirect;
43965d607a6eSdan     pNew->nRecord = nRec;
4397cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
4398cbf6d2d2Sdan     memcpy(pNew->aRecord, aRec, nRec);
43995d607a6eSdan   }else{
4400798693b2Sdan     int op1 = pExist->op;
44015d607a6eSdan 
44025d607a6eSdan     /*
44035d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
44045d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
44055d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
44065d607a6eSdan     **
44075d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
44085d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
44095d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
44105d607a6eSdan     **
44115d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
44125d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
44135d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
44145d607a6eSdan     */
44155d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
44165d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
44175d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
44185d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
44195d607a6eSdan     ){
44205d607a6eSdan       pNew = pExist;
44215d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
44225d607a6eSdan       sqlite3_free(pExist);
44235d607a6eSdan       assert( pNew==0 );
44245d607a6eSdan     }else{
442564277f4aSdan       u8 *aExist = pExist->aRecord;
44265d607a6eSdan       int nByte;
44275d607a6eSdan       u8 *aCsr;
44285d607a6eSdan 
442964277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
443064277f4aSdan       ** buffer of the new object is large enough to hold any record that
443164277f4aSdan       ** may be generated by combining the input records.  */
44325d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
44335d607a6eSdan       pNew = (SessionChange *)sqlite3_malloc(nByte);
44345d607a6eSdan       if( !pNew ){
44351756ae10Sdan         sqlite3_free(pExist);
44365d607a6eSdan         return SQLITE_NOMEM;
44375d607a6eSdan       }
44385d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
44395d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
44405d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
44415d607a6eSdan 
4442b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
44435d607a6eSdan         u8 *a1 = aRec;
4444b08a1efaSdan         assert( op2==SQLITE_UPDATE );
4445798693b2Sdan         pNew->op = SQLITE_INSERT;
4446ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
444764277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
4448b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
4449b08a1efaSdan         assert( op2==SQLITE_INSERT );
4450798693b2Sdan         pNew->op = SQLITE_UPDATE;
4451fa29ecc4Sdan         if( bPatchset ){
4452fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
4453fa29ecc4Sdan           aCsr += nRec;
4454fa29ecc4Sdan         }else{
445564277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
4456b08a1efaSdan             sqlite3_free(pNew);
4457b08a1efaSdan             pNew = 0;
44585d607a6eSdan           }
4459fa29ecc4Sdan         }
4460b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
446164277f4aSdan         u8 *a1 = aExist;
44625d607a6eSdan         u8 *a2 = aRec;
4463cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
446464277f4aSdan         if( bPatchset==0 ){
4465ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
4466ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
446764277f4aSdan         }
4468798693b2Sdan         pNew->op = SQLITE_UPDATE;
446964277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
44705d607a6eSdan           sqlite3_free(pNew);
44715d607a6eSdan           pNew = 0;
44725d607a6eSdan         }
4473b08a1efaSdan       }else{                                /* UPDATE + DELETE */
4474b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
4475798693b2Sdan         pNew->op = SQLITE_DELETE;
447664277f4aSdan         if( bPatchset ){
447764277f4aSdan           memcpy(aCsr, aRec, nRec);
447864277f4aSdan           aCsr += nRec;
447964277f4aSdan         }else{
448064277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
448164277f4aSdan         }
44825d607a6eSdan       }
44835d607a6eSdan 
44845d607a6eSdan       if( pNew ){
44854f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
44865d607a6eSdan       }
44875d607a6eSdan       sqlite3_free(pExist);
44885d607a6eSdan     }
44895d607a6eSdan   }
44905d607a6eSdan 
44915d607a6eSdan   *ppNew = pNew;
44925d607a6eSdan   return SQLITE_OK;
44935d607a6eSdan }
44945d607a6eSdan 
449577fc1d5bSdan /*
44965898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as
44975898ad69Sdan ** the first argument to the changegroup hash tables.
449877fc1d5bSdan */
449916228167Sdan static int sessionChangesetToHash(
4500cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
45015898ad69Sdan   sqlite3_changegroup *pGrp        /* Changegroup object to add changeset to */
45025d607a6eSdan ){
45035d607a6eSdan   u8 *aRec;
45045d607a6eSdan   int nRec;
4505cbf6d2d2Sdan   int rc = SQLITE_OK;
45065d607a6eSdan   SessionTable *pTab = 0;
45075d607a6eSdan 
45085898ad69Sdan 
45095d607a6eSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec) ){
45105d607a6eSdan     const char *zNew;
45115d607a6eSdan     int nCol;
45125d607a6eSdan     int op;
45135d607a6eSdan     int iHash;
45145d607a6eSdan     int bIndirect;
45155d607a6eSdan     SessionChange *pChange;
45165d607a6eSdan     SessionChange *pExist = 0;
45175d607a6eSdan     SessionChange **pp;
45185d607a6eSdan 
45195898ad69Sdan     if( pGrp->pList==0 ){
45205898ad69Sdan       pGrp->bPatch = pIter->bPatchset;
45215898ad69Sdan     }else if( pIter->bPatchset!=pGrp->bPatch ){
45225898ad69Sdan       rc = SQLITE_ERROR;
45235898ad69Sdan       break;
45245898ad69Sdan     }
45255898ad69Sdan 
45265d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
4527ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
45285d607a6eSdan       /* Search the list for a matching table */
45294f528042Sdan       int nNew = (int)strlen(zNew);
4530f29123b5Sdan       u8 *abPK;
4531f29123b5Sdan 
4532f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
45335898ad69Sdan       for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){
45345d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
45355d607a6eSdan       }
45365d607a6eSdan       if( !pTab ){
45376c39e6a8Sdan         SessionTable **ppTab;
45386c39e6a8Sdan 
4539ef7a6304Sdan         pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1);
4540f29123b5Sdan         if( !pTab ){
4541f29123b5Sdan           rc = SQLITE_NOMEM;
4542f29123b5Sdan           break;
4543f29123b5Sdan         }
45445d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
4545f29123b5Sdan         pTab->nCol = nCol;
4546ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
4547ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
4548ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
4549ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
45506c39e6a8Sdan 
45516c39e6a8Sdan         /* The new object must be linked on to the end of the list, not
45526c39e6a8Sdan         ** simply added to the start of it. This is to ensure that the
45536c39e6a8Sdan         ** tables within the output of sqlite3changegroup_output() are in
45546c39e6a8Sdan         ** the right order.  */
45556c39e6a8Sdan         for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext);
45566c39e6a8Sdan         *ppTab = pTab;
4557f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
4558f29123b5Sdan         rc = SQLITE_SCHEMA;
4559f29123b5Sdan         break;
45605d607a6eSdan       }
45615d607a6eSdan     }
45625d607a6eSdan 
4563cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
45641756ae10Sdan       rc = SQLITE_NOMEM;
45651756ae10Sdan       break;
45661756ae10Sdan     }
456764277f4aSdan     iHash = sessionChangeHash(
4568cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
456964277f4aSdan     );
45705d607a6eSdan 
45715d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
45725d607a6eSdan     ** Code below may link it back in.
45735d607a6eSdan     */
45745d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
457564277f4aSdan       int bPkOnly1 = 0;
457664277f4aSdan       int bPkOnly2 = 0;
4577cbf6d2d2Sdan       if( pIter->bPatchset ){
457864277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
457964277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
458064277f4aSdan       }
458164277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
45825d607a6eSdan         pExist = *pp;
45835d607a6eSdan         *pp = (*pp)->pNext;
45845d607a6eSdan         pTab->nEntry--;
45855d607a6eSdan         break;
45865d607a6eSdan       }
45875d607a6eSdan     }
45885d607a6eSdan 
458964277f4aSdan     rc = sessionChangeMerge(pTab,
4590cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
459164277f4aSdan     );
45925d607a6eSdan     if( rc ) break;
45935d607a6eSdan     if( pChange ){
45945d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
45955d607a6eSdan       pTab->apChange[iHash] = pChange;
45965d607a6eSdan       pTab->nEntry++;
45975d607a6eSdan     }
45985d607a6eSdan   }
45995d607a6eSdan 
4600cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
46015d607a6eSdan   return rc;
46025d607a6eSdan }
46035d607a6eSdan 
46045d607a6eSdan /*
46055898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets)
46065898ad69Sdan ** added to the changegroup object passed as the first argument.
46075d607a6eSdan **
46085898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the
46095898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming
46105898ad69Sdan ** interfaces.
46115d607a6eSdan **
46125898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a
46135898ad69Sdan ** buffer containing the output changeset before this function returns. In
46145898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It
46155898ad69Sdan ** is the responsibility of the caller to free the output buffer using
46165898ad69Sdan ** sqlite3_free() when it is no longer required.
46175898ad69Sdan **
46185898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite
46195898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut)
46205898ad69Sdan ** are both set to 0 before returning.
46215d607a6eSdan */
46225898ad69Sdan static int sessionChangegroupOutput(
46235898ad69Sdan   sqlite3_changegroup *pGrp,
4624cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4625cbf6d2d2Sdan   void *pOut,
4626cbf6d2d2Sdan   int *pnOut,
4627cbf6d2d2Sdan   void **ppOut
46285d607a6eSdan ){
46295898ad69Sdan   int rc = SQLITE_OK;
4630e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
46315898ad69Sdan   SessionTable *pTab;
4632cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
46335d607a6eSdan 
46345d607a6eSdan   /* Create the serialized output changeset based on the contents of the
46355898ad69Sdan   ** hash tables attached to the SessionTable objects in list p->pList.
46365d607a6eSdan   */
46375898ad69Sdan   for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
46385d607a6eSdan     int i;
46395d607a6eSdan     if( pTab->nEntry==0 ) continue;
46405d607a6eSdan 
46415898ad69Sdan     sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc);
46425d607a6eSdan     for(i=0; i<pTab->nChange; i++){
46435d607a6eSdan       SessionChange *p;
46445d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
4645798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
46465d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
46475d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
46485d607a6eSdan       }
46495d607a6eSdan     }
4650cbf6d2d2Sdan 
4651f1a08ad8Sdrh     if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
4652cbf6d2d2Sdan       rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4653cbf6d2d2Sdan       buf.nBuf = 0;
4654cbf6d2d2Sdan     }
46555d607a6eSdan   }
46565d607a6eSdan 
46575d607a6eSdan   if( rc==SQLITE_OK ){
4658cbf6d2d2Sdan     if( xOutput ){
4659cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4660cbf6d2d2Sdan     }else{
46615d607a6eSdan       *ppOut = buf.aBuf;
46625d607a6eSdan       *pnOut = buf.nBuf;
4663cbf6d2d2Sdan       buf.aBuf = 0;
46645d607a6eSdan     }
46655d607a6eSdan   }
4666cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
46675d607a6eSdan 
46685d607a6eSdan   return rc;
46695d607a6eSdan }
46705d607a6eSdan 
4671cbf6d2d2Sdan /*
46725898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup.
46735898ad69Sdan */
46745898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){
46755898ad69Sdan   int rc = SQLITE_OK;             /* Return code */
46765898ad69Sdan   sqlite3_changegroup *p;         /* New object */
46775898ad69Sdan   p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup));
46785898ad69Sdan   if( p==0 ){
46795898ad69Sdan     rc = SQLITE_NOMEM;
46805898ad69Sdan   }else{
46815898ad69Sdan     memset(p, 0, sizeof(sqlite3_changegroup));
46825898ad69Sdan   }
46835898ad69Sdan   *pp = p;
46845898ad69Sdan   return rc;
46855898ad69Sdan }
46865898ad69Sdan 
46875898ad69Sdan /*
46885898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes,
46895898ad69Sdan ** to changeset-group p.
46905898ad69Sdan */
46915898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){
46925898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
46935898ad69Sdan   int rc;                         /* Return code */
46945898ad69Sdan 
46955898ad69Sdan   rc = sqlite3changeset_start(&pIter, nData, pData);
46965898ad69Sdan   if( rc==SQLITE_OK ){
46975898ad69Sdan     rc = sessionChangesetToHash(pIter, pGrp);
46985898ad69Sdan   }
46995898ad69Sdan   sqlite3changeset_finalize(pIter);
47005898ad69Sdan   return rc;
47015898ad69Sdan }
47025898ad69Sdan 
47035898ad69Sdan /*
47045898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation
47055898ad69Sdan ** of all changesets added to the group so far.
47065898ad69Sdan */
47075898ad69Sdan int sqlite3changegroup_output(
47085898ad69Sdan     sqlite3_changegroup *pGrp,
47095898ad69Sdan     int *pnData,
47105898ad69Sdan     void **ppData
47115898ad69Sdan ){
47125898ad69Sdan   return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData);
47135898ad69Sdan }
47145898ad69Sdan 
47155898ad69Sdan /*
47165898ad69Sdan ** Streaming versions of changegroup_add().
47175898ad69Sdan */
47185898ad69Sdan int sqlite3changegroup_add_strm(
47195898ad69Sdan   sqlite3_changegroup *pGrp,
47205898ad69Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
47215898ad69Sdan   void *pIn
47225898ad69Sdan ){
47235898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
47245898ad69Sdan   int rc;                         /* Return code */
47255898ad69Sdan 
47265898ad69Sdan   rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
47275898ad69Sdan   if( rc==SQLITE_OK ){
47285898ad69Sdan     rc = sessionChangesetToHash(pIter, pGrp);
47295898ad69Sdan   }
47305898ad69Sdan   sqlite3changeset_finalize(pIter);
47315898ad69Sdan   return rc;
47325898ad69Sdan }
47335898ad69Sdan 
47345898ad69Sdan /*
47355898ad69Sdan ** Streaming versions of changegroup_output().
47365898ad69Sdan */
47375898ad69Sdan int sqlite3changegroup_output_strm(
47385898ad69Sdan   sqlite3_changegroup *pGrp,
47395898ad69Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
47405898ad69Sdan   void *pOut
47415898ad69Sdan ){
47425898ad69Sdan   return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0);
47435898ad69Sdan }
47445898ad69Sdan 
47455898ad69Sdan /*
47465898ad69Sdan ** Delete a changegroup object.
47475898ad69Sdan */
47485898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){
47495898ad69Sdan   if( pGrp ){
47505898ad69Sdan     sessionDeleteTable(pGrp->pList);
47515898ad69Sdan     sqlite3_free(pGrp);
47525898ad69Sdan   }
47535898ad69Sdan }
47545898ad69Sdan 
47555898ad69Sdan /*
4756cbf6d2d2Sdan ** Combine two changesets together.
4757cbf6d2d2Sdan */
4758cbf6d2d2Sdan int sqlite3changeset_concat(
4759cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
4760cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
4761cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
4762cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
4763cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
4764cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
4765cbf6d2d2Sdan ){
47665898ad69Sdan   sqlite3_changegroup *pGrp;
4767cbf6d2d2Sdan   int rc;
4768cbf6d2d2Sdan 
47695898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
4770cbf6d2d2Sdan   if( rc==SQLITE_OK ){
47715898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nLeft, pLeft);
4772cbf6d2d2Sdan   }
4773cbf6d2d2Sdan   if( rc==SQLITE_OK ){
47745898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nRight, pRight);
4775cbf6d2d2Sdan   }
47765898ad69Sdan   if( rc==SQLITE_OK ){
47775898ad69Sdan     rc = sqlite3changegroup_output(pGrp, pnOut, ppOut);
47785898ad69Sdan   }
47795898ad69Sdan   sqlite3changegroup_delete(pGrp);
4780cbf6d2d2Sdan 
4781cbf6d2d2Sdan   return rc;
4782cbf6d2d2Sdan }
4783cbf6d2d2Sdan 
4784cbf6d2d2Sdan /*
4785cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
4786cbf6d2d2Sdan */
4787f1a08ad8Sdrh int sqlite3changeset_concat_strm(
4788cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
4789cbf6d2d2Sdan   void *pInA,
4790cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
4791cbf6d2d2Sdan   void *pInB,
4792cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4793cbf6d2d2Sdan   void *pOut
4794cbf6d2d2Sdan ){
47955898ad69Sdan   sqlite3_changegroup *pGrp;
4796cbf6d2d2Sdan   int rc;
4797cbf6d2d2Sdan 
47985898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
4799cbf6d2d2Sdan   if( rc==SQLITE_OK ){
48005898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA);
4801cbf6d2d2Sdan   }
4802cbf6d2d2Sdan   if( rc==SQLITE_OK ){
48035898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB);
4804cbf6d2d2Sdan   }
48055898ad69Sdan   if( rc==SQLITE_OK ){
48065898ad69Sdan     rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut);
48075898ad69Sdan   }
48085898ad69Sdan   sqlite3changegroup_delete(pGrp);
4809cbf6d2d2Sdan 
4810cbf6d2d2Sdan   return rc;
4811cbf6d2d2Sdan }
4812cbf6d2d2Sdan 
48139b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
4814