14fccf43aSdan 
29b1c62d4Sdrh #if defined(SQLITE_ENABLE_SESSION) && defined(SQLITE_ENABLE_PREUPDATE_HOOK)
34fccf43aSdan #include "sqlite3session.h"
44fccf43aSdan #include <assert.h>
54fccf43aSdan #include <string.h>
64fccf43aSdan 
75d8a2984Sdan #ifndef SQLITE_AMALGAMATION
84fccf43aSdan # include "sqliteInt.h"
94fccf43aSdan # include "vdbeInt.h"
105d8a2984Sdan #endif
114fccf43aSdan 
124fccf43aSdan typedef struct SessionTable SessionTable;
134fccf43aSdan typedef struct SessionChange SessionChange;
14296c7658Sdan typedef struct SessionBuffer SessionBuffer;
15ef7a6304Sdan typedef struct SessionInput SessionInput;
16ef7a6304Sdan 
17ef7a6304Sdan /*
18ef7a6304Sdan ** Minimum chunk size used by streaming versions of functions.
19ef7a6304Sdan */
20f1a08ad8Sdrh #ifndef SESSIONS_STRM_CHUNK_SIZE
214757c658Sdan # ifdef SQLITE_TEST
22f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 64
234757c658Sdan # else
24f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 1024
25f1a08ad8Sdrh # endif
264757c658Sdan #endif
274fccf43aSdan 
281f48e67dSdan static int sessions_strm_chunk_size = SESSIONS_STRM_CHUNK_SIZE;
291f48e67dSdan 
30cf8e9144Sdan typedef struct SessionHook SessionHook;
31cf8e9144Sdan struct SessionHook {
32cf8e9144Sdan   void *pCtx;
33cf8e9144Sdan   int (*xOld)(void*,int,sqlite3_value**);
34cf8e9144Sdan   int (*xNew)(void*,int,sqlite3_value**);
35cf8e9144Sdan   int (*xCount)(void*);
36cf8e9144Sdan   int (*xDepth)(void*);
37cf8e9144Sdan };
38cf8e9144Sdan 
39296c7658Sdan /*
40296c7658Sdan ** Session handle structure.
41296c7658Sdan */
424fccf43aSdan struct sqlite3_session {
434fccf43aSdan   sqlite3 *db;                    /* Database handle session is attached to */
444fccf43aSdan   char *zDb;                      /* Name of database session is attached to */
45296c7658Sdan   int bEnable;                    /* True if currently recording */
46b4480e94Sdan   int bIndirect;                  /* True if all changes are indirect */
47ff4d0f41Sdan   int bAutoAttach;                /* True to auto-attach tables */
484fccf43aSdan   int rc;                         /* Non-zero if an error has occurred */
497531a5a3Sdan   void *pFilterCtx;               /* First argument to pass to xTableFilter */
507531a5a3Sdan   int (*xTableFilter)(void *pCtx, const char *zTab);
511611e5a3Sdan   sqlite3_value *pZeroBlob;       /* Value containing X'' */
524fccf43aSdan   sqlite3_session *pNext;         /* Next session object on same db. */
534fccf43aSdan   SessionTable *pTable;           /* List of attached tables */
54cf8e9144Sdan   SessionHook hook;               /* APIs to grab new and old data with */
554fccf43aSdan };
564fccf43aSdan 
574fccf43aSdan /*
58ef7a6304Sdan ** Instances of this structure are used to build strings or binary records.
59ef7a6304Sdan */
60ef7a6304Sdan struct SessionBuffer {
61ef7a6304Sdan   u8 *aBuf;                       /* Pointer to changeset buffer */
62ef7a6304Sdan   int nBuf;                       /* Size of buffer aBuf */
63ef7a6304Sdan   int nAlloc;                     /* Size of allocation containing aBuf */
64ef7a6304Sdan };
65ef7a6304Sdan 
66ef7a6304Sdan /*
6716228167Sdan ** An object of this type is used internally as an abstraction for
6816228167Sdan ** input data. Input data may be supplied either as a single large buffer
6916228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g.
70f1a08ad8Sdrh **  sqlite3changeset_start_strm()).
71ef7a6304Sdan */
72ef7a6304Sdan struct SessionInput {
733e259bcdSdan   int bNoDiscard;                 /* If true, do not discard in InputBuffer() */
74d9151526Sdan   int iCurrent;                   /* Offset in aData[] of current change */
754757c658Sdan   int iNext;                      /* Offset in aData[] of next change */
764757c658Sdan   u8 *aData;                      /* Pointer to buffer containing changeset */
774757c658Sdan   int nData;                      /* Number of bytes in aData */
784757c658Sdan 
79ef7a6304Sdan   SessionBuffer buf;              /* Current read buffer */
80ef7a6304Sdan   int (*xInput)(void*, void*, int*);        /* Input stream call (or NULL) */
81ef7a6304Sdan   void *pIn;                                /* First argument to xInput */
82ef7a6304Sdan   int bEof;                       /* Set to true after xInput finished */
83ef7a6304Sdan };
84ef7a6304Sdan 
85ef7a6304Sdan /*
86296c7658Sdan ** Structure for changeset iterators.
87296c7658Sdan */
88296c7658Sdan struct sqlite3_changeset_iter {
89ef7a6304Sdan   SessionInput in;                /* Input buffer or stream */
90ef7a6304Sdan   SessionBuffer tblhdr;           /* Buffer to hold apValue/zTab/abPK/ */
9173b3c055Sdan   int bPatchset;                  /* True if this is a patchset */
9244748f27Sdan   int bInvert;                    /* True to invert changeset */
93296c7658Sdan   int rc;                         /* Iterator error code */
94296c7658Sdan   sqlite3_stmt *pConflict;        /* Points to conflicting row, if any */
95296c7658Sdan   char *zTab;                     /* Current table */
96296c7658Sdan   int nCol;                       /* Number of columns in zTab */
97296c7658Sdan   int op;                         /* Current operation */
98b4480e94Sdan   int bIndirect;                  /* True if current change was indirect */
99244593c8Sdan   u8 *abPK;                       /* Primary key array */
100296c7658Sdan   sqlite3_value **apValue;        /* old.* and new.* values */
101296c7658Sdan };
102296c7658Sdan 
103296c7658Sdan /*
1044fccf43aSdan ** Each session object maintains a set of the following structures, one
1054fccf43aSdan ** for each table the session object is monitoring. The structures are
1064fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable.
1074fccf43aSdan **
1084fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have
1094fccf43aSdan ** been modified in any way since the session object was attached to the
1104fccf43aSdan ** table.
1114fccf43aSdan **
1124fccf43aSdan ** The data associated with each hash-table entry is a structure containing
1134fccf43aSdan ** a subset of the initial values that the modified row contained at the
1144fccf43aSdan ** start of the session. Or no initial values if the row was inserted.
1154fccf43aSdan */
1164fccf43aSdan struct SessionTable {
1174fccf43aSdan   SessionTable *pNext;
1184fccf43aSdan   char *zName;                    /* Local name of table */
1194fccf43aSdan   int nCol;                       /* Number of columns in table zName */
1203739f298Sdan   int bStat1;                     /* True if this is sqlite_stat1 */
121e8d5648eSdan   const char **azCol;             /* Column names */
122e8d5648eSdan   u8 *abPK;                       /* Array of primary key flags */
123296c7658Sdan   int nEntry;                     /* Total number of entries in hash table */
1244fccf43aSdan   int nChange;                    /* Size of apChange[] array */
1254fccf43aSdan   SessionChange **apChange;       /* Hash table buckets */
1264fccf43aSdan };
1274fccf43aSdan 
1284fccf43aSdan /*
1294fccf43aSdan ** RECORD FORMAT:
1304fccf43aSdan **
1314fccf43aSdan ** The following record format is similar to (but not compatible with) that
1324fccf43aSdan ** used in SQLite database files. This format is used as part of the
1334fccf43aSdan ** change-set binary format, and so must be architecture independent.
1344fccf43aSdan **
1354fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained -
1364fccf43aSdan ** there is no separation of header and data. Each field begins with a
1374fccf43aSdan ** single byte describing its type, as follows:
1384fccf43aSdan **
1394fccf43aSdan **       0x00: Undefined value.
1404fccf43aSdan **       0x01: Integer value.
1414fccf43aSdan **       0x02: Real value.
1424fccf43aSdan **       0x03: Text value.
1434fccf43aSdan **       0x04: Blob value.
1444fccf43aSdan **       0x05: SQL NULL value.
1454fccf43aSdan **
1464fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT
1474fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists
1484fccf43aSdan ** only of the single type byte. For other types of values, the type byte
1494fccf43aSdan ** is followed by:
1504fccf43aSdan **
1514fccf43aSdan **   Text values:
1524fccf43aSdan **     A varint containing the number of bytes in the value (encoded using
1534fccf43aSdan **     UTF-8). Followed by a buffer containing the UTF-8 representation
1544fccf43aSdan **     of the text value. There is no nul terminator.
1554fccf43aSdan **
1564fccf43aSdan **   Blob values:
1574fccf43aSdan **     A varint containing the number of bytes in the value, followed by
1584fccf43aSdan **     a buffer containing the value itself.
1594fccf43aSdan **
1604fccf43aSdan **   Integer values:
1614fccf43aSdan **     An 8-byte big-endian integer value.
1624fccf43aSdan **
1634fccf43aSdan **   Real values:
1644fccf43aSdan **     An 8-byte big-endian IEEE 754-2008 real value.
1654fccf43aSdan **
1664fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite
1674fccf43aSdan ** record format.
1684fccf43aSdan **
1694fccf43aSdan ** CHANGESET FORMAT:
1704fccf43aSdan **
1714fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on
1724fccf43aSdan ** one or more tables. Operations on a single table are grouped together,
1734fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all
1744fccf43aSdan ** mixed together).
1754fccf43aSdan **
1764fccf43aSdan ** Each group of changes begins with a table header:
1774fccf43aSdan **
1784fccf43aSdan **   1 byte: Constant 0x54 (capital 'T')
179730bb805Sdan **   Varint: Number of columns in the table.
18073b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
1814fccf43aSdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
1824fccf43aSdan **
1834fccf43aSdan ** Followed by one or more changes to the table.
1844fccf43aSdan **
185c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
1865d607a6eSdan **   1 byte: The "indirect-change" flag.
1874fccf43aSdan **   old.* record: (delete and update only)
1884fccf43aSdan **   new.* record: (insert and update only)
18973b3c055Sdan **
190730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the
191730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of
192730bb805Sdan ** columns in the table. The i'th field of each record is associated with
193730bb805Sdan ** the i'th column of the table, counting from left to right in the order
194730bb805Sdan ** in which columns were declared in the CREATE TABLE statement.
195730bb805Sdan **
196730bb805Sdan ** The new.* record that is part of each INSERT change contains the values
197730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each
198730bb805Sdan ** DELETE change contains the values that made up the row that was deleted
199730bb805Sdan ** from the database. In the changeset format, the records that are part
200730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00)
201730bb805Sdan ** fields.
202730bb805Sdan **
203730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields
204730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are
205730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields
206730bb805Sdan ** are set to the values that made up the row before the UPDATE that the
207730bb805Sdan ** change records took place. Within the new.* record, fields associated
208730bb805Sdan ** with table columns modified by the UPDATE change contain the new
209730bb805Sdan ** values. Fields associated with table columns that are not modified
210730bb805Sdan ** are set to "undefined".
211730bb805Sdan **
21273b3c055Sdan ** PATCHSET FORMAT:
21373b3c055Sdan **
21473b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset,
215730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict
216730bb805Sdan ** resolution is required when applying the changeset.
21773b3c055Sdan **
21873b3c055Sdan ** Each group of changes begins with a table header:
21973b3c055Sdan **
22073b3c055Sdan **   1 byte: Constant 0x50 (capital 'P')
221730bb805Sdan **   Varint: Number of columns in the table.
22273b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
22373b3c055Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
22473b3c055Sdan **
22573b3c055Sdan ** Followed by one or more changes to the table.
22673b3c055Sdan **
227c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
22873b3c055Sdan **   1 byte: The "indirect-change" flag.
229730bb805Sdan **   single record: (PK fields for DELETE, PK and modified fields for UPDATE,
230730bb805Sdan **                   full record for INSERT).
231730bb805Sdan **
232730bb805Sdan ** As in the changeset format, each field of the single record that is part
233730bb805Sdan ** of a patchset change is associated with the correspondingly positioned
234730bb805Sdan ** table column, counting from left to right within the CREATE TABLE
235730bb805Sdan ** statement.
236730bb805Sdan **
237730bb805Sdan ** For a DELETE change, all fields within the record except those associated
238f01d3a7eSdan ** with PRIMARY KEY columns are omitted. The PRIMARY KEY fields contain the
239f01d3a7eSdan ** values identifying the row to delete.
240730bb805Sdan **
241730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY
242730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined".
243730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update,
244730bb805Sdan ** and fields associated with modified columns contain the new column values.
245730bb805Sdan **
246730bb805Sdan ** The records associated with INSERT changes are in the same format as for
247730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT
248730bb805Sdan ** change to contain a field set to "undefined".
249b74cf4b6Sdan **
250b74cf4b6Sdan ** REBASE BLOB FORMAT:
251b74cf4b6Sdan **
252b74cf4b6Sdan ** A rebase blob may be output by sqlite3changeset_apply_v2() and its
253b74cf4b6Sdan ** streaming equivalent for use with the sqlite3_rebaser APIs to rebase
254b74cf4b6Sdan ** existing changesets. A rebase blob contains one entry for each conflict
255b74cf4b6Sdan ** resolved using either the OMIT or REPLACE strategies within the apply_v2()
256b74cf4b6Sdan ** call.
257b74cf4b6Sdan **
258b74cf4b6Sdan ** The format used for a rebase blob is very similar to that used for
259b74cf4b6Sdan ** changesets. All entries related to a single table are grouped together.
260b74cf4b6Sdan **
261b74cf4b6Sdan ** Each group of entries begins with a table header in changeset format:
262b74cf4b6Sdan **
263b74cf4b6Sdan **   1 byte: Constant 0x54 (capital 'T')
264b74cf4b6Sdan **   Varint: Number of columns in the table.
265b74cf4b6Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
266b74cf4b6Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
267b74cf4b6Sdan **
268b74cf4b6Sdan ** Followed by one or more entries associated with the table.
269b74cf4b6Sdan **
270b74cf4b6Sdan **   1 byte: Either SQLITE_INSERT (0x12), DELETE (0x09).
271b74cf4b6Sdan **   1 byte: Flag. 0x01 for REPLACE, 0x00 for OMIT.
272b74cf4b6Sdan **   record: (in the record format defined above).
273b74cf4b6Sdan **
274b74cf4b6Sdan ** In a rebase blob, the first field is set to SQLITE_INSERT if the change
275b74cf4b6Sdan ** that caused the conflict was an INSERT or UPDATE, or to SQLITE_DELETE if
276b74cf4b6Sdan ** it was a DELETE. The second field is set to 0x01 if the conflict
277b74cf4b6Sdan ** resolution strategy was REPLACE, or 0x00 if it was OMIT.
278b74cf4b6Sdan **
279b74cf4b6Sdan ** If the change that caused the conflict was a DELETE, then the single
280b74cf4b6Sdan ** record is a copy of the old.* record from the original changeset. If it
281b74cf4b6Sdan ** was an INSERT, then the single record is a copy of the new.* record. If
282b74cf4b6Sdan ** the conflicting change was an UPDATE, then the single record is a copy
283b74cf4b6Sdan ** of the new.* record with the PK fields filled in based on the original
284b74cf4b6Sdan ** old.* record.
2854fccf43aSdan */
2864fccf43aSdan 
2874fccf43aSdan /*
2884fccf43aSdan ** For each row modified during a session, there exists a single instance of
2894fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table.
2904fccf43aSdan */
2914fccf43aSdan struct SessionChange {
292798693b2Sdan   int op;                         /* One of UPDATE, DELETE, INSERT */
293b4480e94Sdan   int bIndirect;                  /* True if this change is "indirect" */
2944fccf43aSdan   int nRecord;                    /* Number of bytes in buffer aRecord[] */
2954fccf43aSdan   u8 *aRecord;                    /* Buffer containing old.* record */
2964fccf43aSdan   SessionChange *pNext;           /* For hash-table collisions */
2974fccf43aSdan };
2984fccf43aSdan 
299296c7658Sdan /*
300296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the
301296c7658Sdan ** number of bytes written.
302296c7658Sdan */
303296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){
304296c7658Sdan   return putVarint32(aBuf, iVal);
3054fccf43aSdan }
3064fccf43aSdan 
307296c7658Sdan /*
308296c7658Sdan ** Return the number of bytes required to store value iVal as a varint.
309296c7658Sdan */
310296c7658Sdan static int sessionVarintLen(int iVal){
311296c7658Sdan   return sqlite3VarintLen(iVal);
312296c7658Sdan }
313296c7658Sdan 
314296c7658Sdan /*
315296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of
316296c7658Sdan ** bytes read.
317296c7658Sdan */
3184fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){
319296c7658Sdan   return getVarint32(aBuf, *piVal);
3204fccf43aSdan }
3214fccf43aSdan 
32248cd59a5Sdrh /* Load an unaligned and unsigned 32-bit integer */
32348cd59a5Sdrh #define SESSION_UINT32(x) (((u32)(x)[0]<<24)|((x)[1]<<16)|((x)[2]<<8)|(x)[3])
32448cd59a5Sdrh 
325296c7658Sdan /*
326296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return
327296c7658Sdan ** the value read.
328296c7658Sdan */
3294fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){
33048cd59a5Sdrh   u64 x = SESSION_UINT32(aRec);
33148cd59a5Sdrh   u32 y = SESSION_UINT32(aRec+4);
33248cd59a5Sdrh   x = (x<<32) + y;
33348cd59a5Sdrh   return (sqlite3_int64)x;
3344fccf43aSdan }
3354fccf43aSdan 
3364fccf43aSdan /*
337296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[].
338296c7658Sdan */
339296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){
340296c7658Sdan   aBuf[0] = (i>>56) & 0xFF;
341296c7658Sdan   aBuf[1] = (i>>48) & 0xFF;
342296c7658Sdan   aBuf[2] = (i>>40) & 0xFF;
343296c7658Sdan   aBuf[3] = (i>>32) & 0xFF;
344296c7658Sdan   aBuf[4] = (i>>24) & 0xFF;
345296c7658Sdan   aBuf[5] = (i>>16) & 0xFF;
346296c7658Sdan   aBuf[6] = (i>> 8) & 0xFF;
347296c7658Sdan   aBuf[7] = (i>> 0) & 0xFF;
348296c7658Sdan }
349296c7658Sdan 
350296c7658Sdan /*
3514fccf43aSdan ** This function is used to serialize the contents of value pValue (see
3524fccf43aSdan ** comment titled "RECORD FORMAT" above).
3534fccf43aSdan **
3544fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to
3554fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before
3564fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is
3574fccf43aSdan ** set *pnWrite.
3584fccf43aSdan **
3594fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs
3604fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16))
3614fccf43aSdan ** SQLITE_NOMEM is returned.
3624fccf43aSdan */
3634fccf43aSdan static int sessionSerializeValue(
3644fccf43aSdan   u8 *aBuf,                       /* If non-NULL, write serialized value here */
3654fccf43aSdan   sqlite3_value *pValue,          /* Value to serialize */
3662d77d80aSdrh   sqlite3_int64 *pnWrite          /* IN/OUT: Increment by bytes written */
3674fccf43aSdan ){
368296c7658Sdan   int nByte;                      /* Size of serialized value in bytes */
3694fccf43aSdan 
37080fe2d93Sdan   if( pValue ){
37180fe2d93Sdan     int eType;                    /* Value type (SQLITE_NULL, TEXT etc.) */
37280fe2d93Sdan 
3734fccf43aSdan     eType = sqlite3_value_type(pValue);
3744fccf43aSdan     if( aBuf ) aBuf[0] = eType;
3754fccf43aSdan 
3764fccf43aSdan     switch( eType ){
3774fccf43aSdan       case SQLITE_NULL:
3784fccf43aSdan         nByte = 1;
3794fccf43aSdan         break;
3804fccf43aSdan 
3814fccf43aSdan       case SQLITE_INTEGER:
3824fccf43aSdan       case SQLITE_FLOAT:
3834fccf43aSdan         if( aBuf ){
3844fccf43aSdan           /* TODO: SQLite does something special to deal with mixed-endian
3854fccf43aSdan           ** floating point values (e.g. ARM7). This code probably should
3864fccf43aSdan           ** too.  */
3874fccf43aSdan           u64 i;
3884fccf43aSdan           if( eType==SQLITE_INTEGER ){
3894fccf43aSdan             i = (u64)sqlite3_value_int64(pValue);
3904fccf43aSdan           }else{
3914fccf43aSdan             double r;
3924fccf43aSdan             assert( sizeof(double)==8 && sizeof(u64)==8 );
3934fccf43aSdan             r = sqlite3_value_double(pValue);
3944fccf43aSdan             memcpy(&i, &r, 8);
3954fccf43aSdan           }
396296c7658Sdan           sessionPutI64(&aBuf[1], i);
3974fccf43aSdan         }
3984fccf43aSdan         nByte = 9;
3994fccf43aSdan         break;
4004fccf43aSdan 
4014e895da1Sdan       default: {
40280fe2d93Sdan         u8 *z;
40380fe2d93Sdan         int n;
40480fe2d93Sdan         int nVarint;
40580fe2d93Sdan 
4064e895da1Sdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
40780fe2d93Sdan         if( eType==SQLITE_TEXT ){
40880fe2d93Sdan           z = (u8 *)sqlite3_value_text(pValue);
40980fe2d93Sdan         }else{
41080fe2d93Sdan           z = (u8 *)sqlite3_value_blob(pValue);
41180fe2d93Sdan         }
41280fe2d93Sdan         n = sqlite3_value_bytes(pValue);
4133cc89d95Sdan         if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
41480fe2d93Sdan         nVarint = sessionVarintLen(n);
41580fe2d93Sdan 
4164fccf43aSdan         if( aBuf ){
4174fccf43aSdan           sessionVarintPut(&aBuf[1], n);
418895decf6Sdan           if( n ) memcpy(&aBuf[nVarint + 1], z, n);
4194fccf43aSdan         }
4204fccf43aSdan 
4214fccf43aSdan         nByte = 1 + nVarint + n;
4224fccf43aSdan         break;
4234fccf43aSdan       }
4244fccf43aSdan     }
42580fe2d93Sdan   }else{
42680fe2d93Sdan     nByte = 1;
42780fe2d93Sdan     if( aBuf ) aBuf[0] = '\0';
42880fe2d93Sdan   }
4294fccf43aSdan 
430fa122adaSdan   if( pnWrite ) *pnWrite += nByte;
4314fccf43aSdan   return SQLITE_OK;
4324fccf43aSdan }
4334fccf43aSdan 
434fa122adaSdan 
435798693b2Sdan /*
436798693b2Sdan ** This macro is used to calculate hash key values for data structures. In
437798693b2Sdan ** order to use this macro, the entire data structure must be represented
438798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value
439798693b2Sdan ** for a data structure represented as three such integers, the macro may
440798693b2Sdan ** then be used as follows:
441798693b2Sdan **
442798693b2Sdan **    int hash_key_value;
443798693b2Sdan **    hash_key_value = HASH_APPEND(0, <value 1>);
444798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 2>);
445798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 3>);
446798693b2Sdan **
447798693b2Sdan ** In practice, the data structures this macro is used for are the primary
448798693b2Sdan ** key values of modified rows.
449798693b2Sdan */
4504131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add)
451798693b2Sdan 
452798693b2Sdan /*
453798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the
454798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
455798693b2Sdan */
4564131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){
457e8d5648eSdan   h = HASH_APPEND(h, i & 0xFFFFFFFF);
458e8d5648eSdan   return HASH_APPEND(h, (i>>32)&0xFFFFFFFF);
459e8d5648eSdan }
460798693b2Sdan 
461798693b2Sdan /*
462798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to
463798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value.
464798693b2Sdan */
4654131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){
466e8d5648eSdan   int i;
467e8d5648eSdan   for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]);
468e8d5648eSdan   return h;
469e8d5648eSdan }
470e8d5648eSdan 
4714fccf43aSdan /*
472798693b2Sdan ** Append the hash of the data type passed as the second argument to the
473798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
474798693b2Sdan */
475798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){
476798693b2Sdan   return HASH_APPEND(h, eType);
477798693b2Sdan }
478798693b2Sdan 
479798693b2Sdan /*
4804131639cSdan ** This function may only be called from within a pre-update callback.
4814131639cSdan ** It calculates a hash based on the primary key values of the old.* or
482798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to
483798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL
484798693b2Sdan ** values, *pbNullPK is set to true before returning.
485798693b2Sdan **
486798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values
487798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned
488798693b2Sdan ** and the output variables are set as described above.
4894fccf43aSdan */
490798693b2Sdan static int sessionPreupdateHash(
491cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns pTab */
492e8d5648eSdan   SessionTable *pTab,             /* Session table handle */
493e8d5648eSdan   int bNew,                       /* True to hash the new.* PK */
49427453faeSdan   int *piHash,                    /* OUT: Hash value */
495798693b2Sdan   int *pbNullPK                   /* OUT: True if there are NULL values in PK */
496e8d5648eSdan ){
4974131639cSdan   unsigned int h = 0;             /* Hash value to return */
4984131639cSdan   int i;                          /* Used to iterate through columns */
499e8d5648eSdan 
50027453faeSdan   assert( *pbNullPK==0 );
501cf8e9144Sdan   assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) );
502e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
503e8d5648eSdan     if( pTab->abPK[i] ){
504e8d5648eSdan       int rc;
505e8d5648eSdan       int eType;
506e8d5648eSdan       sqlite3_value *pVal;
507e8d5648eSdan 
508e8d5648eSdan       if( bNew ){
509cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal);
510e8d5648eSdan       }else{
511cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal);
512e8d5648eSdan       }
51312ca0b56Sdan       if( rc!=SQLITE_OK ) return rc;
514e8d5648eSdan 
515e8d5648eSdan       eType = sqlite3_value_type(pVal);
516798693b2Sdan       h = sessionHashAppendType(h, eType);
5176734007dSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
518e8d5648eSdan         i64 iVal;
519e8d5648eSdan         if( eType==SQLITE_INTEGER ){
520e8d5648eSdan           iVal = sqlite3_value_int64(pVal);
521e8d5648eSdan         }else{
522e8d5648eSdan           double rVal = sqlite3_value_double(pVal);
523e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
524e8d5648eSdan           memcpy(&iVal, &rVal, 8);
525e8d5648eSdan         }
526e8d5648eSdan         h = sessionHashAppendI64(h, iVal);
5276734007dSdan       }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
5286734007dSdan         const u8 *z;
5293cc89d95Sdan         int n;
5306734007dSdan         if( eType==SQLITE_TEXT ){
5316734007dSdan           z = (const u8 *)sqlite3_value_text(pVal);
5326734007dSdan         }else{
5336734007dSdan           z = (const u8 *)sqlite3_value_blob(pVal);
534e8d5648eSdan         }
5353cc89d95Sdan         n = sqlite3_value_bytes(pVal);
5363cc89d95Sdan         if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
5373cc89d95Sdan         h = sessionHashAppendBlob(h, n, z);
5386734007dSdan       }else{
53927453faeSdan         assert( eType==SQLITE_NULL );
5401611e5a3Sdan         assert( pTab->bStat1==0 || i!=1 );
54127453faeSdan         *pbNullPK = 1;
542e8d5648eSdan       }
543e8d5648eSdan     }
544e8d5648eSdan   }
545e8d5648eSdan 
546e8d5648eSdan   *piHash = (h % pTab->nChange);
547e8d5648eSdan   return SQLITE_OK;
548e8d5648eSdan }
549e8d5648eSdan 
5504131639cSdan /*
5516cda207fSdan ** The buffer that the argument points to contains a serialized SQL value.
5526cda207fSdan ** Return the number of bytes of space occupied by the value (including
5536cda207fSdan ** the type byte).
5546cda207fSdan */
5556cda207fSdan static int sessionSerialLen(u8 *a){
5566cda207fSdan   int e = *a;
5576cda207fSdan   int n;
55824a0c453Sdan   if( e==0 || e==0xFF ) return 1;
5596cda207fSdan   if( e==SQLITE_NULL ) return 1;
5606cda207fSdan   if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9;
5616cda207fSdan   return sessionVarintGet(&a[1], &n) + 1 + n;
5626cda207fSdan }
5636cda207fSdan 
5646cda207fSdan /*
5655d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a
566798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys
5674131639cSdan ** calculated by this function are compatible with those calculated by
5684131639cSdan ** sessionPreupdateHash().
56964277f4aSdan **
57064277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from
57164277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely.
5724131639cSdan */
5734131639cSdan static unsigned int sessionChangeHash(
5744131639cSdan   SessionTable *pTab,             /* Table handle */
57564277f4aSdan   int bPkOnly,                    /* Record consists of PK fields only */
5765d607a6eSdan   u8 *aRecord,                    /* Change record */
5774131639cSdan   int nBucket                     /* Assume this many buckets in hash table */
578e8d5648eSdan ){
5794131639cSdan   unsigned int h = 0;             /* Value to return */
5804131639cSdan   int i;                          /* Used to iterate through columns */
5815d607a6eSdan   u8 *a = aRecord;                /* Used to iterate through change record */
582e8d5648eSdan 
583e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
5846cda207fSdan     int eType = *a;
585e8d5648eSdan     int isPK = pTab->abPK[i];
58664277f4aSdan     if( bPkOnly && isPK==0 ) continue;
587e8d5648eSdan 
58827453faeSdan     /* It is not possible for eType to be SQLITE_NULL here. The session
58927453faeSdan     ** module does not record changes for rows with NULL values stored in
59027453faeSdan     ** primary key columns. */
59127453faeSdan     assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
59227453faeSdan          || eType==SQLITE_TEXT || eType==SQLITE_BLOB
5936cda207fSdan          || eType==SQLITE_NULL || eType==0
59427453faeSdan     );
5951611e5a3Sdan     assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) );
59627453faeSdan 
5976cda207fSdan     if( isPK ){
5986cda207fSdan       a++;
599798693b2Sdan       h = sessionHashAppendType(h, eType);
60027453faeSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
6016734007dSdan         h = sessionHashAppendI64(h, sessionGetI64(a));
602e8d5648eSdan         a += 8;
6031611e5a3Sdan       }else{
604e8d5648eSdan         int n;
605e8d5648eSdan         a += sessionVarintGet(a, &n);
6066734007dSdan         h = sessionHashAppendBlob(h, n, a);
607e8d5648eSdan         a += n;
608e8d5648eSdan       }
6096cda207fSdan     }else{
6106cda207fSdan       a += sessionSerialLen(a);
6116cda207fSdan     }
612e8d5648eSdan   }
613e8d5648eSdan   return (h % nBucket);
614e8d5648eSdan }
615e8d5648eSdan 
616798693b2Sdan /*
617798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab.
618798693b2Sdan ** This function returns true if the two records apply to the same row (i.e.
619798693b2Sdan ** have the same values stored in the primary key columns), or false
620798693b2Sdan ** otherwise.
621798693b2Sdan */
6225d607a6eSdan static int sessionChangeEqual(
623798693b2Sdan   SessionTable *pTab,             /* Table used for PK definition */
624a71d2371Sdan   int bLeftPkOnly,                /* True if aLeft[] contains PK fields only */
6255d607a6eSdan   u8 *aLeft,                      /* Change record */
626a71d2371Sdan   int bRightPkOnly,               /* True if aRight[] contains PK fields only */
6275d607a6eSdan   u8 *aRight                      /* Change record */
6285d607a6eSdan ){
629798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor to iterate through aLeft */
630798693b2Sdan   u8 *a2 = aRight;                /* Cursor to iterate through aRight */
631798693b2Sdan   int iCol;                       /* Used to iterate through table columns */
6325d607a6eSdan 
633798693b2Sdan   for(iCol=0; iCol<pTab->nCol; iCol++){
63414faa061Sdan     if( pTab->abPK[iCol] ){
6355d607a6eSdan       int n1 = sessionSerialLen(a1);
6365d607a6eSdan       int n2 = sessionSerialLen(a2);
6375d607a6eSdan 
638bd45374cSdan       if( n1!=n2 || memcmp(a1, a2, n1) ){
6395d607a6eSdan         return 0;
6405d607a6eSdan       }
64114faa061Sdan       a1 += n1;
64214faa061Sdan       a2 += n2;
64314faa061Sdan     }else{
64414faa061Sdan       if( bLeftPkOnly==0 ) a1 += sessionSerialLen(a1);
64514faa061Sdan       if( bRightPkOnly==0 ) a2 += sessionSerialLen(a2);
64614faa061Sdan     }
6475d607a6eSdan   }
6485d607a6eSdan 
6495d607a6eSdan   return 1;
6505d607a6eSdan }
6515d607a6eSdan 
652798693b2Sdan /*
653798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change
654798693b2Sdan ** records with nCol columns. This function "merges" the two records into
655798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is
656798693b2Sdan ** then set to point to one byte after the last byte written before
657798693b2Sdan ** returning.
658798693b2Sdan **
659798693b2Sdan ** The merging of records is done as follows: For each column, if the
660798693b2Sdan ** aRight record contains a value for the column, copy the value from
661798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither
662798693b2Sdan ** record contains a value for a given column, then neither does the
663798693b2Sdan ** output record.
664798693b2Sdan */
6655d607a6eSdan static void sessionMergeRecord(
6665d607a6eSdan   u8 **paOut,
667798693b2Sdan   int nCol,
6685d607a6eSdan   u8 *aLeft,
6695d607a6eSdan   u8 *aRight
6705d607a6eSdan ){
671798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor used to iterate through aLeft */
672798693b2Sdan   u8 *a2 = aRight;                /* Cursor used to iterate through aRight */
673798693b2Sdan   u8 *aOut = *paOut;              /* Output cursor */
674798693b2Sdan   int iCol;                       /* Used to iterate from 0 to nCol */
6755d607a6eSdan 
676798693b2Sdan   for(iCol=0; iCol<nCol; iCol++){
6775d607a6eSdan     int n1 = sessionSerialLen(a1);
6785d607a6eSdan     int n2 = sessionSerialLen(a2);
6795d607a6eSdan     if( *a2 ){
6805d607a6eSdan       memcpy(aOut, a2, n2);
6815d607a6eSdan       aOut += n2;
6825d607a6eSdan     }else{
6835d607a6eSdan       memcpy(aOut, a1, n1);
6845d607a6eSdan       aOut += n1;
6855d607a6eSdan     }
6865d607a6eSdan     a1 += n1;
6875d607a6eSdan     a2 += n2;
6885d607a6eSdan   }
6895d607a6eSdan 
6905d607a6eSdan   *paOut = aOut;
6915d607a6eSdan }
6925d607a6eSdan 
693798693b2Sdan /*
694798693b2Sdan ** This is a helper function used by sessionMergeUpdate().
695798693b2Sdan **
696798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value
697798693b2Sdan ** within a change record. Before it returns, both have been advanced so
698798693b2Sdan ** as to point to the next value in the record.
699798693b2Sdan **
700798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e.
701fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo
702798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the
703798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal
704798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points
705fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words:
706fa29ecc4Sdan **
707fa29ecc4Sdan **   if( *paTwo is valid ) return *paTwo;
708fa29ecc4Sdan **   return *paOne;
709fa29ecc4Sdan **
710798693b2Sdan */
7115d607a6eSdan static u8 *sessionMergeValue(
712798693b2Sdan   u8 **paOne,                     /* IN/OUT: Left-hand buffer pointer */
713798693b2Sdan   u8 **paTwo,                     /* IN/OUT: Right-hand buffer pointer */
714798693b2Sdan   int *pnVal                      /* OUT: Bytes in returned value */
7155d607a6eSdan ){
7165d607a6eSdan   u8 *a1 = *paOne;
7175d607a6eSdan   u8 *a2 = *paTwo;
7185d607a6eSdan   u8 *pRet = 0;
7195d607a6eSdan   int n1;
7205d607a6eSdan 
7215d607a6eSdan   assert( a1 );
7225d607a6eSdan   if( a2 ){
7235d607a6eSdan     int n2 = sessionSerialLen(a2);
7245d607a6eSdan     if( *a2 ){
7255d607a6eSdan       *pnVal = n2;
7265d607a6eSdan       pRet = a2;
7275d607a6eSdan     }
7285d607a6eSdan     *paTwo = &a2[n2];
7295d607a6eSdan   }
7305d607a6eSdan 
7315d607a6eSdan   n1 = sessionSerialLen(a1);
7325d607a6eSdan   if( pRet==0 ){
7335d607a6eSdan     *pnVal = n1;
7345d607a6eSdan     pRet = a1;
7355d607a6eSdan   }
7365d607a6eSdan   *paOne = &a1[n1];
7375d607a6eSdan 
7385d607a6eSdan   return pRet;
7395d607a6eSdan }
7405d607a6eSdan 
741798693b2Sdan /*
742798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes
743798693b2Sdan ** on the same row.
744798693b2Sdan */
7455d607a6eSdan static int sessionMergeUpdate(
746798693b2Sdan   u8 **paOut,                     /* IN/OUT: Pointer to output buffer */
747798693b2Sdan   SessionTable *pTab,             /* Table change pertains to */
748a71d2371Sdan   int bPatchset,                  /* True if records are patchset records */
749798693b2Sdan   u8 *aOldRecord1,                /* old.* record for first change */
750798693b2Sdan   u8 *aOldRecord2,                /* old.* record for second change */
751798693b2Sdan   u8 *aNewRecord1,                /* new.* record for first change */
752798693b2Sdan   u8 *aNewRecord2                 /* new.* record for second change */
7535d607a6eSdan ){
7545d607a6eSdan   u8 *aOld1 = aOldRecord1;
7555d607a6eSdan   u8 *aOld2 = aOldRecord2;
7565d607a6eSdan   u8 *aNew1 = aNewRecord1;
7575d607a6eSdan   u8 *aNew2 = aNewRecord2;
7585d607a6eSdan 
7595d607a6eSdan   u8 *aOut = *paOut;
7605d607a6eSdan   int i;
76164277f4aSdan 
76264277f4aSdan   if( bPatchset==0 ){
7635d607a6eSdan     int bRequired = 0;
7645d607a6eSdan 
7655d607a6eSdan     assert( aOldRecord1 && aNewRecord1 );
7665d607a6eSdan 
7675d607a6eSdan     /* Write the old.* vector first. */
7685d607a6eSdan     for(i=0; i<pTab->nCol; i++){
7695d607a6eSdan       int nOld;
7705d607a6eSdan       u8 *aOld;
7715d607a6eSdan       int nNew;
7725d607a6eSdan       u8 *aNew;
7735d607a6eSdan 
7745d607a6eSdan       aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7755d607a6eSdan       aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
7765d607a6eSdan       if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){
7775d607a6eSdan         if( pTab->abPK[i]==0 ) bRequired = 1;
7785d607a6eSdan         memcpy(aOut, aOld, nOld);
7795d607a6eSdan         aOut += nOld;
7805d607a6eSdan       }else{
7815d607a6eSdan         *(aOut++) = '\0';
7825d607a6eSdan       }
7835d607a6eSdan     }
7845d607a6eSdan 
7855d607a6eSdan     if( !bRequired ) return 0;
78664277f4aSdan   }
7875d607a6eSdan 
7885d607a6eSdan   /* Write the new.* vector */
7895d607a6eSdan   aOld1 = aOldRecord1;
7905d607a6eSdan   aOld2 = aOldRecord2;
7915d607a6eSdan   aNew1 = aNewRecord1;
7925d607a6eSdan   aNew2 = aNewRecord2;
7935d607a6eSdan   for(i=0; i<pTab->nCol; i++){
7945d607a6eSdan     int nOld;
7955d607a6eSdan     u8 *aOld;
7965d607a6eSdan     int nNew;
7975d607a6eSdan     u8 *aNew;
7985d607a6eSdan 
7995d607a6eSdan     aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
8005d607a6eSdan     aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
80164277f4aSdan     if( bPatchset==0
80264277f4aSdan      && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew)))
80364277f4aSdan     ){
8045d607a6eSdan       *(aOut++) = '\0';
8055d607a6eSdan     }else{
8065d607a6eSdan       memcpy(aOut, aNew, nNew);
8075d607a6eSdan       aOut += nNew;
8085d607a6eSdan     }
8095d607a6eSdan   }
8105d607a6eSdan 
8115d607a6eSdan   *paOut = aOut;
8125d607a6eSdan   return 1;
8135d607a6eSdan }
8145d607a6eSdan 
81577fc1d5bSdan /*
81677fc1d5bSdan ** This function is only called from within a pre-update-hook callback.
81777fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row
81877fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise
81977fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns
82077fc1d5bSdan ** false.
82177fc1d5bSdan */
82277fc1d5bSdan static int sessionPreupdateEqual(
823cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns SessionTable */
82477fc1d5bSdan   SessionTable *pTab,             /* Table associated with change */
82577fc1d5bSdan   SessionChange *pChange,         /* Change to compare to */
82677fc1d5bSdan   int op                          /* Current pre-update operation */
827e8d5648eSdan ){
82877fc1d5bSdan   int iCol;                       /* Used to iterate through columns */
82977fc1d5bSdan   u8 *a = pChange->aRecord;       /* Cursor used to scan change record */
830e8d5648eSdan 
83177fc1d5bSdan   assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
83277fc1d5bSdan   for(iCol=0; iCol<pTab->nCol; iCol++){
83377fc1d5bSdan     if( !pTab->abPK[iCol] ){
834798693b2Sdan       a += sessionSerialLen(a);
835e8d5648eSdan     }else{
8366734007dSdan       sqlite3_value *pVal;        /* Value returned by preupdate_new/old */
8376734007dSdan       int rc;                     /* Error code from preupdate_new/old */
838798693b2Sdan       int eType = *a++;           /* Type of value from change record */
8396734007dSdan 
8406734007dSdan       /* The following calls to preupdate_new() and preupdate_old() can not
8416734007dSdan       ** fail. This is because they cache their return values, and by the
8426734007dSdan       ** time control flows to here they have already been called once from
8436734007dSdan       ** within sessionPreupdateHash(). The first two asserts below verify
8446734007dSdan       ** this (that the method has already been called). */
84577fc1d5bSdan       if( op==SQLITE_INSERT ){
846cf8e9144Sdan         /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */
847cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal);
848e8d5648eSdan       }else{
849cf8e9144Sdan         /* assert( db->pPreUpdate->pUnpacked ); */
850cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal);
851e8d5648eSdan       }
8526734007dSdan       assert( rc==SQLITE_OK );
8531611e5a3Sdan       if( sqlite3_value_type(pVal)!=eType ) return 0;
854e8d5648eSdan 
85512ca0b56Sdan       /* A SessionChange object never has a NULL value in a PK column */
85612ca0b56Sdan       assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
85712ca0b56Sdan            || eType==SQLITE_BLOB    || eType==SQLITE_TEXT
85812ca0b56Sdan       );
85912ca0b56Sdan 
86012ca0b56Sdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
861e8d5648eSdan         i64 iVal = sessionGetI64(a);
862e8d5648eSdan         a += 8;
863e8d5648eSdan         if( eType==SQLITE_INTEGER ){
86477fc1d5bSdan           if( sqlite3_value_int64(pVal)!=iVal ) return 0;
865e8d5648eSdan         }else{
866e8d5648eSdan           double rVal;
867e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
868e8d5648eSdan           memcpy(&rVal, &iVal, 8);
86977fc1d5bSdan           if( sqlite3_value_double(pVal)!=rVal ) return 0;
870e8d5648eSdan         }
87112ca0b56Sdan       }else{
872e8d5648eSdan         int n;
873e8d5648eSdan         const u8 *z;
874e8d5648eSdan         a += sessionVarintGet(a, &n);
87577fc1d5bSdan         if( sqlite3_value_bytes(pVal)!=n ) return 0;
87612ca0b56Sdan         if( eType==SQLITE_TEXT ){
87712ca0b56Sdan           z = sqlite3_value_text(pVal);
87812ca0b56Sdan         }else{
87912ca0b56Sdan           z = sqlite3_value_blob(pVal);
88012ca0b56Sdan         }
8813f2cebb6Sdan         if( n>0 && memcmp(a, z, n) ) return 0;
882e8d5648eSdan         a += n;
883e8d5648eSdan       }
884e8d5648eSdan     }
885e8d5648eSdan   }
886e8d5648eSdan 
88777fc1d5bSdan   return 1;
8884fccf43aSdan }
8894fccf43aSdan 
8904fccf43aSdan /*
8914fccf43aSdan ** If required, grow the hash table used to store changes on table pTab
8924fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the
8934fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return
8944fccf43aSdan ** SQLITE_OK.
8954fccf43aSdan **
8964fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In
8974fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway.
8984fccf43aSdan ** Growing the hash table in this case is a performance optimization only,
8994fccf43aSdan ** it is not required for correct operation.
9004fccf43aSdan */
90164277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){
9024fccf43aSdan   if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){
9034fccf43aSdan     int i;
9044fccf43aSdan     SessionChange **apNew;
905f6ad201aSdrh     sqlite3_int64 nNew = 2*(sqlite3_int64)(pTab->nChange ? pTab->nChange : 128);
9064fccf43aSdan 
9072d77d80aSdrh     apNew = (SessionChange **)sqlite3_malloc64(sizeof(SessionChange *) * nNew);
9084fccf43aSdan     if( apNew==0 ){
9094fccf43aSdan       if( pTab->nChange==0 ){
9104fccf43aSdan         return SQLITE_ERROR;
9114fccf43aSdan       }
9124fccf43aSdan       return SQLITE_OK;
9134fccf43aSdan     }
9144fccf43aSdan     memset(apNew, 0, sizeof(SessionChange *) * nNew);
9154fccf43aSdan 
9164fccf43aSdan     for(i=0; i<pTab->nChange; i++){
9174fccf43aSdan       SessionChange *p;
9184fccf43aSdan       SessionChange *pNext;
9194fccf43aSdan       for(p=pTab->apChange[i]; p; p=pNext){
92064277f4aSdan         int bPkOnly = (p->op==SQLITE_DELETE && bPatchset);
92164277f4aSdan         int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew);
9224fccf43aSdan         pNext = p->pNext;
9234fccf43aSdan         p->pNext = apNew[iHash];
9244fccf43aSdan         apNew[iHash] = p;
9254fccf43aSdan       }
9264fccf43aSdan     }
9274fccf43aSdan 
9284fccf43aSdan     sqlite3_free(pTab->apChange);
9294fccf43aSdan     pTab->nChange = nNew;
9304fccf43aSdan     pTab->apChange = apNew;
9314fccf43aSdan   }
9324fccf43aSdan 
9334fccf43aSdan   return SQLITE_OK;
9344fccf43aSdan }
9354fccf43aSdan 
936296c7658Sdan /*
937e8d5648eSdan ** This function queries the database for the names of the columns of table
9383f975373Sdrh ** zThis, in schema zDb.
939e8d5648eSdan **
94077fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number
94177fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a
942e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to
943e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not
944e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding
945e8d5648eSdan ** column is part of the primary key.
946e8d5648eSdan **
947e8d5648eSdan ** For example, if the table is declared as:
948e8d5648eSdan **
949e8d5648eSdan **     CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z));
950e8d5648eSdan **
95177fc1d5bSdan ** Then the four output variables are populated as follows:
952e8d5648eSdan **
95377fc1d5bSdan **     *pnCol  = 4
954e8d5648eSdan **     *pzTab  = "tbl1"
955e8d5648eSdan **     *pazCol = {"w", "x", "y", "z"}
956e8d5648eSdan **     *pabPK  = {1, 0, 0, 1}
957e8d5648eSdan **
958e8d5648eSdan ** All returned buffers are part of the same single allocation, which must
9593f975373Sdrh ** be freed using sqlite3_free() by the caller
960e8d5648eSdan */
961e8d5648eSdan static int sessionTableInfo(
962e8d5648eSdan   sqlite3 *db,                    /* Database connection */
963e8d5648eSdan   const char *zDb,                /* Name of attached database (e.g. "main") */
964e8d5648eSdan   const char *zThis,              /* Table name */
965ca62ad57Sdan   int *pnCol,                     /* OUT: number of columns */
966e8d5648eSdan   const char **pzTab,             /* OUT: Copy of zThis */
967e8d5648eSdan   const char ***pazCol,           /* OUT: Array of column names for table */
968e8d5648eSdan   u8 **pabPK                      /* OUT: Array of booleans - true for PK col */
969e8d5648eSdan ){
970e8d5648eSdan   char *zPragma;
971e8d5648eSdan   sqlite3_stmt *pStmt;
972e8d5648eSdan   int rc;
9732d77d80aSdrh   sqlite3_int64 nByte;
974e8d5648eSdan   int nDbCol = 0;
975e8d5648eSdan   int nThis;
976e8d5648eSdan   int i;
97774f598b6Smistachkin   u8 *pAlloc = 0;
978db04571cSdan   char **azCol = 0;
97974f598b6Smistachkin   u8 *abPK = 0;
980e8d5648eSdan 
981db04571cSdan   assert( pazCol && pabPK );
982e8d5648eSdan 
983cfdbde21Sdrh   nThis = sqlite3Strlen30(zThis);
984614efe2bSdan   if( nThis==12 && 0==sqlite3_stricmp("sqlite_stat1", zThis) ){
98558713184Sdan     rc = sqlite3_table_column_metadata(db, zDb, zThis, 0, 0, 0, 0, 0, 0);
98658713184Sdan     if( rc==SQLITE_OK ){
987614efe2bSdan       /* For sqlite_stat1, pretend that (tbl,idx) is the PRIMARY KEY. */
988614efe2bSdan       zPragma = sqlite3_mprintf(
989614efe2bSdan           "SELECT 0, 'tbl',  '', 0, '', 1     UNION ALL "
990614efe2bSdan           "SELECT 1, 'idx',  '', 0, '', 2     UNION ALL "
991614efe2bSdan           "SELECT 2, 'stat', '', 0, '', 0"
992614efe2bSdan       );
99358713184Sdan     }else if( rc==SQLITE_ERROR ){
99458713184Sdan       zPragma = sqlite3_mprintf("");
99558713184Sdan     }else{
99658713184Sdan       return rc;
99758713184Sdan     }
998614efe2bSdan   }else{
999e8d5648eSdan     zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis);
1000614efe2bSdan   }
1001e8d5648eSdan   if( !zPragma ) return SQLITE_NOMEM;
1002e8d5648eSdan 
1003e8d5648eSdan   rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0);
1004e8d5648eSdan   sqlite3_free(zPragma);
1005e8d5648eSdan   if( rc!=SQLITE_OK ) return rc;
1006e8d5648eSdan 
1007e8d5648eSdan   nByte = nThis + 1;
1008e8d5648eSdan   while( SQLITE_ROW==sqlite3_step(pStmt) ){
1009e8d5648eSdan     nByte += sqlite3_column_bytes(pStmt, 1);
1010e8d5648eSdan     nDbCol++;
1011e8d5648eSdan   }
1012e8d5648eSdan   rc = sqlite3_reset(pStmt);
1013e8d5648eSdan 
1014e8d5648eSdan   if( rc==SQLITE_OK ){
1015e8d5648eSdan     nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1);
10162d77d80aSdrh     pAlloc = sqlite3_malloc64(nByte);
1017e8d5648eSdan     if( pAlloc==0 ){
1018e8d5648eSdan       rc = SQLITE_NOMEM;
1019e8d5648eSdan     }
1020e8d5648eSdan   }
1021e8d5648eSdan   if( rc==SQLITE_OK ){
1022e8d5648eSdan     azCol = (char **)pAlloc;
1023ca62ad57Sdan     pAlloc = (u8 *)&azCol[nDbCol];
1024e8d5648eSdan     abPK = (u8 *)pAlloc;
1025ca62ad57Sdan     pAlloc = &abPK[nDbCol];
1026e8d5648eSdan     if( pzTab ){
1027e8d5648eSdan       memcpy(pAlloc, zThis, nThis+1);
1028e8d5648eSdan       *pzTab = (char *)pAlloc;
1029e8d5648eSdan       pAlloc += nThis+1;
1030e8d5648eSdan     }
1031e8d5648eSdan 
1032e8d5648eSdan     i = 0;
1033e8d5648eSdan     while( SQLITE_ROW==sqlite3_step(pStmt) ){
1034e8d5648eSdan       int nName = sqlite3_column_bytes(pStmt, 1);
1035e8d5648eSdan       const unsigned char *zName = sqlite3_column_text(pStmt, 1);
1036e8d5648eSdan       if( zName==0 ) break;
1037e8d5648eSdan       memcpy(pAlloc, zName, nName+1);
1038e8d5648eSdan       azCol[i] = (char *)pAlloc;
1039e8d5648eSdan       pAlloc += nName+1;
1040db04571cSdan       abPK[i] = sqlite3_column_int(pStmt, 5);
1041e8d5648eSdan       i++;
1042e8d5648eSdan     }
1043e8d5648eSdan     rc = sqlite3_reset(pStmt);
1044e8d5648eSdan 
1045e8d5648eSdan   }
1046e8d5648eSdan 
1047e8d5648eSdan   /* If successful, populate the output variables. Otherwise, zero them and
1048e8d5648eSdan   ** free any allocation made. An error code will be returned in this case.
1049e8d5648eSdan   */
1050e8d5648eSdan   if( rc==SQLITE_OK ){
1051db04571cSdan     *pazCol = (const char **)azCol;
1052db04571cSdan     *pabPK = abPK;
1053ca62ad57Sdan     *pnCol = nDbCol;
1054e8d5648eSdan   }else{
1055db04571cSdan     *pazCol = 0;
1056db04571cSdan     *pabPK = 0;
1057ca62ad57Sdan     *pnCol = 0;
1058e8d5648eSdan     if( pzTab ) *pzTab = 0;
1059db04571cSdan     sqlite3_free(azCol);
1060e8d5648eSdan   }
1061e8d5648eSdan   sqlite3_finalize(pStmt);
1062e8d5648eSdan   return rc;
1063e8d5648eSdan }
1064e8d5648eSdan 
1065e8d5648eSdan /*
1066296c7658Sdan ** This function is only called from within a pre-update handler for a
1067296c7658Sdan ** write to table pTab, part of session pSession. If this is the first
10686dc29e60Sdan ** write to this table, initalize the SessionTable.nCol, azCol[] and
10696dc29e60Sdan ** abPK[] arrays accordingly.
1070296c7658Sdan **
10716dc29e60Sdan ** If an error occurs, an error code is stored in sqlite3_session.rc and
10726dc29e60Sdan ** non-zero returned. Or, if no error occurs but the table has no primary
10736dc29e60Sdan ** key, sqlite3_session.rc is left set to SQLITE_OK and non-zero returned to
10746dc29e60Sdan ** indicate that updates on this table should be ignored. SessionTable.abPK
10756dc29e60Sdan ** is set to NULL in this case.
1076296c7658Sdan */
10774fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){
10784fccf43aSdan   if( pTab->nCol==0 ){
10796dc29e60Sdan     u8 *abPK;
1080e8d5648eSdan     assert( pTab->azCol==0 || pTab->abPK==0 );
1081e8d5648eSdan     pSession->rc = sessionTableInfo(pSession->db, pSession->zDb,
10826dc29e60Sdan         pTab->zName, &pTab->nCol, 0, &pTab->azCol, &abPK
1083e8d5648eSdan     );
10846dc29e60Sdan     if( pSession->rc==SQLITE_OK ){
10856dc29e60Sdan       int i;
10866dc29e60Sdan       for(i=0; i<pTab->nCol; i++){
10876dc29e60Sdan         if( abPK[i] ){
10886dc29e60Sdan           pTab->abPK = abPK;
10896dc29e60Sdan           break;
1090ca62ad57Sdan         }
10914fccf43aSdan       }
10923739f298Sdan       if( 0==sqlite3_stricmp("sqlite_stat1", pTab->zName) ){
10933739f298Sdan         pTab->bStat1 = 1;
10943739f298Sdan       }
10956dc29e60Sdan     }
10966dc29e60Sdan   }
10976dc29e60Sdan   return (pSession->rc || pTab->abPK==0);
1098e8d5648eSdan }
1099e8d5648eSdan 
110077fc1d5bSdan /*
11011611e5a3Sdan ** Versions of the four methods in object SessionHook for use with the
11021611e5a3Sdan ** sqlite_stat1 table. The purpose of this is to substitute a zero-length
11031611e5a3Sdan ** blob each time a NULL value is read from the "idx" column of the
11041611e5a3Sdan ** sqlite_stat1 table.
11051611e5a3Sdan */
11061611e5a3Sdan typedef struct SessionStat1Ctx SessionStat1Ctx;
11071611e5a3Sdan struct SessionStat1Ctx {
11081611e5a3Sdan   SessionHook hook;
11091611e5a3Sdan   sqlite3_session *pSession;
11101611e5a3Sdan };
11111611e5a3Sdan static int sessionStat1Old(void *pCtx, int iCol, sqlite3_value **ppVal){
11121611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
11131611e5a3Sdan   sqlite3_value *pVal = 0;
11141611e5a3Sdan   int rc = p->hook.xOld(p->hook.pCtx, iCol, &pVal);
11151611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
11161611e5a3Sdan     pVal = p->pSession->pZeroBlob;
11171611e5a3Sdan   }
11181611e5a3Sdan   *ppVal = pVal;
11191611e5a3Sdan   return rc;
11201611e5a3Sdan }
11211611e5a3Sdan static int sessionStat1New(void *pCtx, int iCol, sqlite3_value **ppVal){
11221611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
11231611e5a3Sdan   sqlite3_value *pVal = 0;
11241611e5a3Sdan   int rc = p->hook.xNew(p->hook.pCtx, iCol, &pVal);
11251611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
11261611e5a3Sdan     pVal = p->pSession->pZeroBlob;
11271611e5a3Sdan   }
11281611e5a3Sdan   *ppVal = pVal;
11291611e5a3Sdan   return rc;
11301611e5a3Sdan }
11311611e5a3Sdan static int sessionStat1Count(void *pCtx){
11321611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
11331611e5a3Sdan   return p->hook.xCount(p->hook.pCtx);
11341611e5a3Sdan }
11351611e5a3Sdan static int sessionStat1Depth(void *pCtx){
11361611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
11371611e5a3Sdan   return p->hook.xDepth(p->hook.pCtx);
11381611e5a3Sdan }
11391611e5a3Sdan 
11401611e5a3Sdan 
11411611e5a3Sdan /*
114277fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a
114377fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change
114477fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument.
114577fc1d5bSdan **
114677fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added
114777fc1d5bSdan ** to the changed-rows hash table associated with table pTab.
114877fc1d5bSdan */
1149e8d5648eSdan static void sessionPreupdateOneChange(
115077fc1d5bSdan   int op,                         /* One of SQLITE_UPDATE, INSERT, DELETE */
115177fc1d5bSdan   sqlite3_session *pSession,      /* Session object pTab is attached to */
115277fc1d5bSdan   SessionTable *pTab              /* Table that change applies to */
1153e8d5648eSdan ){
1154e8d5648eSdan   int iHash;
1155cf8e9144Sdan   int bNull = 0;
1156e8d5648eSdan   int rc = SQLITE_OK;
1157fa94d492Sdrh   SessionStat1Ctx stat1 = {{0,0,0,0,0},0};
1158e8d5648eSdan 
1159e8d5648eSdan   if( pSession->rc ) return;
1160e8d5648eSdan 
1161e8d5648eSdan   /* Load table details if required */
1162e8d5648eSdan   if( sessionInitTable(pSession, pTab) ) return;
1163e8d5648eSdan 
11646dc29e60Sdan   /* Check the number of columns in this xPreUpdate call matches the
11656dc29e60Sdan   ** number of columns in the table.  */
11666dc29e60Sdan   if( pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx) ){
11676dc29e60Sdan     pSession->rc = SQLITE_SCHEMA;
11686dc29e60Sdan     return;
11696dc29e60Sdan   }
11706dc29e60Sdan 
1171e8d5648eSdan   /* Grow the hash table if required */
117264277f4aSdan   if( sessionGrowHash(0, pTab) ){
11735d607a6eSdan     pSession->rc = SQLITE_NOMEM;
11745d607a6eSdan     return;
11755d607a6eSdan   }
1176e8d5648eSdan 
11771611e5a3Sdan   if( pTab->bStat1 ){
11781611e5a3Sdan     stat1.hook = pSession->hook;
11791611e5a3Sdan     stat1.pSession = pSession;
11801611e5a3Sdan     pSession->hook.pCtx = (void*)&stat1;
11811611e5a3Sdan     pSession->hook.xNew = sessionStat1New;
11821611e5a3Sdan     pSession->hook.xOld = sessionStat1Old;
11831611e5a3Sdan     pSession->hook.xCount = sessionStat1Count;
11841611e5a3Sdan     pSession->hook.xDepth = sessionStat1Depth;
11851611e5a3Sdan     if( pSession->pZeroBlob==0 ){
11861611e5a3Sdan       sqlite3_value *p = sqlite3ValueNew(0);
11871611e5a3Sdan       if( p==0 ){
11881611e5a3Sdan         rc = SQLITE_NOMEM;
11891611e5a3Sdan         goto error_out;
11901611e5a3Sdan       }
11911611e5a3Sdan       sqlite3ValueSetStr(p, 0, "", 0, SQLITE_STATIC);
11921611e5a3Sdan       pSession->pZeroBlob = p;
11931611e5a3Sdan     }
11941611e5a3Sdan   }
11951611e5a3Sdan 
119680fe2d93Sdan   /* Calculate the hash-key for this change. If the primary key of the row
119780fe2d93Sdan   ** includes a NULL value, exit early. Such changes are ignored by the
119880fe2d93Sdan   ** session module. */
1199cf8e9144Sdan   rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull);
120080fe2d93Sdan   if( rc!=SQLITE_OK ) goto error_out;
120180fe2d93Sdan 
12021611e5a3Sdan   if( bNull==0 ){
120380fe2d93Sdan     /* Search the hash table for an existing record for this row. */
1204b4480e94Sdan     SessionChange *pC;
12056734007dSdan     for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){
1206cf8e9144Sdan       if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break;
1207e8d5648eSdan     }
120880fe2d93Sdan 
1209e8d5648eSdan     if( pC==0 ){
1210e8d5648eSdan       /* Create a new change object containing all the old values (if
1211e8d5648eSdan       ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK
1212e8d5648eSdan       ** values (if this is an INSERT). */
1213b4480e94Sdan       SessionChange *pChange; /* New change object */
12142d77d80aSdrh       sqlite3_int64 nByte;    /* Number of bytes to allocate */
1215e8d5648eSdan       int i;                  /* Used to iterate through columns */
1216e8d5648eSdan 
1217b4480e94Sdan       assert( rc==SQLITE_OK );
1218e8d5648eSdan       pTab->nEntry++;
1219e8d5648eSdan 
1220e8d5648eSdan       /* Figure out how large an allocation is required */
1221e8d5648eSdan       nByte = sizeof(SessionChange);
122280fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1223e8d5648eSdan         sqlite3_value *p = 0;
1224e8d5648eSdan         if( op!=SQLITE_INSERT ){
1225cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p);
122680fe2d93Sdan           assert( trc==SQLITE_OK );
122780fe2d93Sdan         }else if( pTab->abPK[i] ){
1228cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p);
122980fe2d93Sdan           assert( trc==SQLITE_OK );
1230e8d5648eSdan         }
123180fe2d93Sdan 
123280fe2d93Sdan         /* This may fail if SQLite value p contains a utf-16 string that must
123380fe2d93Sdan         ** be converted to utf-8 and an OOM error occurs while doing so. */
1234e8d5648eSdan         rc = sessionSerializeValue(0, p, &nByte);
123580fe2d93Sdan         if( rc!=SQLITE_OK ) goto error_out;
1236e8d5648eSdan       }
1237e8d5648eSdan 
1238e8d5648eSdan       /* Allocate the change object */
12392d77d80aSdrh       pChange = (SessionChange *)sqlite3_malloc64(nByte);
1240e8d5648eSdan       if( !pChange ){
1241e8d5648eSdan         rc = SQLITE_NOMEM;
124280fe2d93Sdan         goto error_out;
1243e8d5648eSdan       }else{
1244e8d5648eSdan         memset(pChange, 0, sizeof(SessionChange));
1245e8d5648eSdan         pChange->aRecord = (u8 *)&pChange[1];
1246e8d5648eSdan       }
1247e8d5648eSdan 
124880fe2d93Sdan       /* Populate the change object. None of the preupdate_old(),
124980fe2d93Sdan       ** preupdate_new() or SerializeValue() calls below may fail as all
125080fe2d93Sdan       ** required values and encodings have already been cached in memory.
125180fe2d93Sdan       ** It is not possible for an OOM to occur in this block. */
1252e8d5648eSdan       nByte = 0;
125380fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1254e8d5648eSdan         sqlite3_value *p = 0;
1255e8d5648eSdan         if( op!=SQLITE_INSERT ){
1256cf8e9144Sdan           pSession->hook.xOld(pSession->hook.pCtx, i, &p);
125780fe2d93Sdan         }else if( pTab->abPK[i] ){
1258cf8e9144Sdan           pSession->hook.xNew(pSession->hook.pCtx, i, &p);
1259e8d5648eSdan         }
126080fe2d93Sdan         sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte);
1261e8d5648eSdan       }
126280fe2d93Sdan 
126380fe2d93Sdan       /* Add the change to the hash-table */
1264cf8e9144Sdan       if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){
1265b4480e94Sdan         pChange->bIndirect = 1;
1266b4480e94Sdan       }
126712ca0b56Sdan       pChange->nRecord = nByte;
1268798693b2Sdan       pChange->op = op;
1269e8d5648eSdan       pChange->pNext = pTab->apChange[iHash];
1270e8d5648eSdan       pTab->apChange[iHash] = pChange;
127180fe2d93Sdan 
127280fe2d93Sdan     }else if( pC->bIndirect ){
1273b4480e94Sdan       /* If the existing change is considered "indirect", but this current
1274b4480e94Sdan       ** change is "direct", mark the change object as direct. */
1275cf8e9144Sdan       if( pSession->hook.xDepth(pSession->hook.pCtx)==0
1276cf8e9144Sdan        && pSession->bIndirect==0
1277cf8e9144Sdan       ){
1278b4480e94Sdan         pC->bIndirect = 0;
1279b4480e94Sdan       }
1280e8d5648eSdan     }
12814fccf43aSdan   }
128212ca0b56Sdan 
128312ca0b56Sdan   /* If an error has occurred, mark the session object as failed. */
128480fe2d93Sdan  error_out:
12851611e5a3Sdan   if( pTab->bStat1 ){
12861611e5a3Sdan     pSession->hook = stat1.hook;
12871611e5a3Sdan   }
128812ca0b56Sdan   if( rc!=SQLITE_OK ){
128912ca0b56Sdan     pSession->rc = rc;
129012ca0b56Sdan   }
129127453faeSdan }
12924fccf43aSdan 
1293cf8e9144Sdan static int sessionFindTable(
1294cf8e9144Sdan   sqlite3_session *pSession,
1295cf8e9144Sdan   const char *zName,
1296cf8e9144Sdan   SessionTable **ppTab
1297cf8e9144Sdan ){
1298cf8e9144Sdan   int rc = SQLITE_OK;
1299cf8e9144Sdan   int nName = sqlite3Strlen30(zName);
1300cf8e9144Sdan   SessionTable *pRet;
1301cf8e9144Sdan 
1302cf8e9144Sdan   /* Search for an existing table */
1303cf8e9144Sdan   for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){
1304cf8e9144Sdan     if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break;
1305cf8e9144Sdan   }
1306cf8e9144Sdan 
1307cf8e9144Sdan   if( pRet==0 && pSession->bAutoAttach ){
1308cf8e9144Sdan     /* If there is a table-filter configured, invoke it. If it returns 0,
1309cf8e9144Sdan     ** do not automatically add the new table. */
1310cf8e9144Sdan     if( pSession->xTableFilter==0
1311cf8e9144Sdan      || pSession->xTableFilter(pSession->pFilterCtx, zName)
1312cf8e9144Sdan     ){
1313cf8e9144Sdan       rc = sqlite3session_attach(pSession, zName);
1314cf8e9144Sdan       if( rc==SQLITE_OK ){
13156c39e6a8Sdan         for(pRet=pSession->pTable; pRet->pNext; pRet=pRet->pNext);
1316cf8e9144Sdan         assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) );
1317cf8e9144Sdan       }
1318cf8e9144Sdan     }
1319cf8e9144Sdan   }
1320cf8e9144Sdan 
1321cf8e9144Sdan   assert( rc==SQLITE_OK || pRet==0 );
1322cf8e9144Sdan   *ppTab = pRet;
1323cf8e9144Sdan   return rc;
1324cf8e9144Sdan }
1325cf8e9144Sdan 
13264fccf43aSdan /*
13274fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases.
13284fccf43aSdan */
13294fccf43aSdan static void xPreUpdate(
13304fccf43aSdan   void *pCtx,                     /* Copy of third arg to preupdate_hook() */
13314fccf43aSdan   sqlite3 *db,                    /* Database handle */
13324fccf43aSdan   int op,                         /* SQLITE_UPDATE, DELETE or INSERT */
13334fccf43aSdan   char const *zDb,                /* Database name */
13344fccf43aSdan   char const *zName,              /* Table name */
13354fccf43aSdan   sqlite3_int64 iKey1,            /* Rowid of row about to be deleted/updated */
13364fccf43aSdan   sqlite3_int64 iKey2             /* New rowid value (for a rowid UPDATE) */
13374fccf43aSdan ){
13384fccf43aSdan   sqlite3_session *pSession;
1339cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb);
13404fccf43aSdan 
13414c220252Sdan   assert( sqlite3_mutex_held(db->mutex) );
13424c220252Sdan 
13434fccf43aSdan   for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){
13444fccf43aSdan     SessionTable *pTab;
1345296c7658Sdan 
1346e8d5648eSdan     /* If this session is attached to a different database ("main", "temp"
1347e8d5648eSdan     ** etc.), or if it is not currently enabled, there is nothing to do. Skip
1348e8d5648eSdan     ** to the next session object attached to this database. */
1349296c7658Sdan     if( pSession->bEnable==0 ) continue;
13504fccf43aSdan     if( pSession->rc ) continue;
13514fccf43aSdan     if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue;
1352296c7658Sdan 
1353cf8e9144Sdan     pSession->rc = sessionFindTable(pSession, zName, &pTab);
1354cf8e9144Sdan     if( pTab ){
1355cf8e9144Sdan       assert( pSession->rc==SQLITE_OK );
1356e8d5648eSdan       sessionPreupdateOneChange(op, pSession, pTab);
1357e8d5648eSdan       if( op==SQLITE_UPDATE ){
1358e8d5648eSdan         sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab);
13594fccf43aSdan       }
13604fccf43aSdan     }
13614fccf43aSdan   }
13624fccf43aSdan }
1363cf8e9144Sdan 
1364cf8e9144Sdan /*
1365cf8e9144Sdan ** The pre-update hook implementations.
1366cf8e9144Sdan */
1367cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1368cf8e9144Sdan   return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal);
1369cf8e9144Sdan }
1370cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1371cf8e9144Sdan   return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal);
1372cf8e9144Sdan }
1373cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){
1374cf8e9144Sdan   return sqlite3_preupdate_count((sqlite3*)pCtx);
1375cf8e9144Sdan }
1376cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){
1377cf8e9144Sdan   return sqlite3_preupdate_depth((sqlite3*)pCtx);
1378cf8e9144Sdan }
1379cf8e9144Sdan 
1380cf8e9144Sdan /*
1381cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only
1382cf8e9144Sdan ** argument.
1383cf8e9144Sdan */
1384cf8e9144Sdan static void sessionPreupdateHooks(
1385cf8e9144Sdan   sqlite3_session *pSession
1386cf8e9144Sdan ){
1387cf8e9144Sdan   pSession->hook.pCtx = (void*)pSession->db;
1388cf8e9144Sdan   pSession->hook.xOld = sessionPreupdateOld;
1389cf8e9144Sdan   pSession->hook.xNew = sessionPreupdateNew;
1390cf8e9144Sdan   pSession->hook.xCount = sessionPreupdateCount;
1391cf8e9144Sdan   pSession->hook.xDepth = sessionPreupdateDepth;
1392cf8e9144Sdan }
1393cf8e9144Sdan 
1394cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx;
1395cf8e9144Sdan struct SessionDiffCtx {
1396cf8e9144Sdan   sqlite3_stmt *pStmt;
1397cf8e9144Sdan   int nOldOff;
1398cf8e9144Sdan };
1399cf8e9144Sdan 
1400cf8e9144Sdan /*
1401cf8e9144Sdan ** The diff hook implementations.
1402cf8e9144Sdan */
1403cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1404cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1405cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff);
1406cf8e9144Sdan   return SQLITE_OK;
1407cf8e9144Sdan }
1408cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1409cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1410cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal);
1411cf8e9144Sdan    return SQLITE_OK;
1412cf8e9144Sdan }
1413cf8e9144Sdan static int sessionDiffCount(void *pCtx){
1414cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1415cf8e9144Sdan   return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt);
1416cf8e9144Sdan }
1417cf8e9144Sdan static int sessionDiffDepth(void *pCtx){
1418cf8e9144Sdan   return 0;
1419cf8e9144Sdan }
1420cf8e9144Sdan 
1421cf8e9144Sdan /*
1422cf8e9144Sdan ** Install the diff hooks on the session object passed as the only
1423cf8e9144Sdan ** argument.
1424cf8e9144Sdan */
1425cf8e9144Sdan static void sessionDiffHooks(
1426cf8e9144Sdan   sqlite3_session *pSession,
1427cf8e9144Sdan   SessionDiffCtx *pDiffCtx
1428cf8e9144Sdan ){
1429cf8e9144Sdan   pSession->hook.pCtx = (void*)pDiffCtx;
1430cf8e9144Sdan   pSession->hook.xOld = sessionDiffOld;
1431cf8e9144Sdan   pSession->hook.xNew = sessionDiffNew;
1432cf8e9144Sdan   pSession->hook.xCount = sessionDiffCount;
1433cf8e9144Sdan   pSession->hook.xDepth = sessionDiffDepth;
1434cf8e9144Sdan }
1435cf8e9144Sdan 
1436cf8e9144Sdan static char *sessionExprComparePK(
1437cf8e9144Sdan   int nCol,
1438cf8e9144Sdan   const char *zDb1, const char *zDb2,
1439cf8e9144Sdan   const char *zTab,
1440cf8e9144Sdan   const char **azCol, u8 *abPK
1441cf8e9144Sdan ){
1442cf8e9144Sdan   int i;
1443cf8e9144Sdan   const char *zSep = "";
1444cf8e9144Sdan   char *zRet = 0;
1445cf8e9144Sdan 
1446cf8e9144Sdan   for(i=0; i<nCol; i++){
1447cf8e9144Sdan     if( abPK[i] ){
1448cf8e9144Sdan       zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"",
1449cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1450cf8e9144Sdan       );
1451cf8e9144Sdan       zSep = " AND ";
1452cf8e9144Sdan       if( zRet==0 ) break;
1453cf8e9144Sdan     }
1454cf8e9144Sdan   }
1455cf8e9144Sdan 
1456cf8e9144Sdan   return zRet;
1457cf8e9144Sdan }
1458cf8e9144Sdan 
1459cf8e9144Sdan static char *sessionExprCompareOther(
1460cf8e9144Sdan   int nCol,
1461cf8e9144Sdan   const char *zDb1, const char *zDb2,
1462cf8e9144Sdan   const char *zTab,
1463cf8e9144Sdan   const char **azCol, u8 *abPK
1464cf8e9144Sdan ){
1465cf8e9144Sdan   int i;
1466cf8e9144Sdan   const char *zSep = "";
1467cf8e9144Sdan   char *zRet = 0;
1468cf8e9144Sdan   int bHave = 0;
1469cf8e9144Sdan 
1470cf8e9144Sdan   for(i=0; i<nCol; i++){
1471cf8e9144Sdan     if( abPK[i]==0 ){
1472cf8e9144Sdan       bHave = 1;
1473cf8e9144Sdan       zRet = sqlite3_mprintf(
1474cf8e9144Sdan           "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"",
1475cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1476cf8e9144Sdan       );
1477cf8e9144Sdan       zSep = " OR ";
1478cf8e9144Sdan       if( zRet==0 ) break;
1479cf8e9144Sdan     }
1480cf8e9144Sdan   }
1481cf8e9144Sdan 
1482cf8e9144Sdan   if( bHave==0 ){
1483cf8e9144Sdan     assert( zRet==0 );
1484cf8e9144Sdan     zRet = sqlite3_mprintf("0");
1485cf8e9144Sdan   }
1486cf8e9144Sdan 
1487cf8e9144Sdan   return zRet;
1488cf8e9144Sdan }
1489cf8e9144Sdan 
1490cf8e9144Sdan static char *sessionSelectFindNew(
1491cf8e9144Sdan   int nCol,
1492cf8e9144Sdan   const char *zDb1,      /* Pick rows in this db only */
1493cf8e9144Sdan   const char *zDb2,      /* But not in this one */
1494cf8e9144Sdan   const char *zTbl,      /* Table name */
1495cf8e9144Sdan   const char *zExpr
1496cf8e9144Sdan ){
1497cf8e9144Sdan   char *zRet = sqlite3_mprintf(
1498cf8e9144Sdan       "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS ("
1499cf8e9144Sdan       "  SELECT 1 FROM \"%w\".\"%w\" WHERE %s"
1500cf8e9144Sdan       ")",
1501cf8e9144Sdan       zDb1, zTbl, zDb2, zTbl, zExpr
1502cf8e9144Sdan   );
1503cf8e9144Sdan   return zRet;
1504cf8e9144Sdan }
1505cf8e9144Sdan 
1506cf8e9144Sdan static int sessionDiffFindNew(
1507cf8e9144Sdan   int op,
1508cf8e9144Sdan   sqlite3_session *pSession,
1509cf8e9144Sdan   SessionTable *pTab,
1510cf8e9144Sdan   const char *zDb1,
1511cf8e9144Sdan   const char *zDb2,
1512cf8e9144Sdan   char *zExpr
1513cf8e9144Sdan ){
1514cf8e9144Sdan   int rc = SQLITE_OK;
1515cf8e9144Sdan   char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr);
1516cf8e9144Sdan 
1517cf8e9144Sdan   if( zStmt==0 ){
1518cf8e9144Sdan     rc = SQLITE_NOMEM;
1519cf8e9144Sdan   }else{
1520cf8e9144Sdan     sqlite3_stmt *pStmt;
1521cf8e9144Sdan     rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1522cf8e9144Sdan     if( rc==SQLITE_OK ){
1523cf8e9144Sdan       SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1524cf8e9144Sdan       pDiffCtx->pStmt = pStmt;
1525cf8e9144Sdan       pDiffCtx->nOldOff = 0;
1526cf8e9144Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
1527cf8e9144Sdan         sessionPreupdateOneChange(op, pSession, pTab);
1528cf8e9144Sdan       }
1529cf8e9144Sdan       rc = sqlite3_finalize(pStmt);
1530cf8e9144Sdan     }
1531cf8e9144Sdan     sqlite3_free(zStmt);
1532cf8e9144Sdan   }
1533cf8e9144Sdan 
1534cf8e9144Sdan   return rc;
1535cf8e9144Sdan }
1536cf8e9144Sdan 
1537cf8e9144Sdan static int sessionDiffFindModified(
1538cf8e9144Sdan   sqlite3_session *pSession,
1539cf8e9144Sdan   SessionTable *pTab,
1540cf8e9144Sdan   const char *zFrom,
1541cf8e9144Sdan   const char *zExpr
1542cf8e9144Sdan ){
1543cf8e9144Sdan   int rc = SQLITE_OK;
1544cf8e9144Sdan 
1545cf8e9144Sdan   char *zExpr2 = sessionExprCompareOther(pTab->nCol,
1546cf8e9144Sdan       pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK
1547cf8e9144Sdan   );
1548cf8e9144Sdan   if( zExpr2==0 ){
1549cf8e9144Sdan     rc = SQLITE_NOMEM;
1550cf8e9144Sdan   }else{
1551cf8e9144Sdan     char *zStmt = sqlite3_mprintf(
1552dd009f83Sdan         "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)",
1553cf8e9144Sdan         pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2
1554cf8e9144Sdan     );
1555cf8e9144Sdan     if( zStmt==0 ){
1556cf8e9144Sdan       rc = SQLITE_NOMEM;
1557cf8e9144Sdan     }else{
1558cf8e9144Sdan       sqlite3_stmt *pStmt;
1559cf8e9144Sdan       rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1560cf8e9144Sdan 
1561cf8e9144Sdan       if( rc==SQLITE_OK ){
1562cf8e9144Sdan         SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1563cf8e9144Sdan         pDiffCtx->pStmt = pStmt;
1564cf8e9144Sdan         pDiffCtx->nOldOff = pTab->nCol;
1565cf8e9144Sdan         while( SQLITE_ROW==sqlite3_step(pStmt) ){
1566cf8e9144Sdan           sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab);
1567cf8e9144Sdan         }
1568cf8e9144Sdan         rc = sqlite3_finalize(pStmt);
1569cf8e9144Sdan       }
1570cf8e9144Sdan       sqlite3_free(zStmt);
1571cf8e9144Sdan     }
1572cf8e9144Sdan   }
1573cf8e9144Sdan 
1574cf8e9144Sdan   return rc;
1575cf8e9144Sdan }
1576cf8e9144Sdan 
1577cf8e9144Sdan int sqlite3session_diff(
1578cf8e9144Sdan   sqlite3_session *pSession,
1579cf8e9144Sdan   const char *zFrom,
1580cf8e9144Sdan   const char *zTbl,
1581cf8e9144Sdan   char **pzErrMsg
1582cf8e9144Sdan ){
1583cf8e9144Sdan   const char *zDb = pSession->zDb;
1584cf8e9144Sdan   int rc = pSession->rc;
1585cf8e9144Sdan   SessionDiffCtx d;
1586cf8e9144Sdan 
1587cf8e9144Sdan   memset(&d, 0, sizeof(d));
1588cf8e9144Sdan   sessionDiffHooks(pSession, &d);
1589cf8e9144Sdan 
159010dc553cSdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1591cf8e9144Sdan   if( pzErrMsg ) *pzErrMsg = 0;
1592cf8e9144Sdan   if( rc==SQLITE_OK ){
1593cf8e9144Sdan     char *zExpr = 0;
1594cf8e9144Sdan     sqlite3 *db = pSession->db;
1595cf8e9144Sdan     SessionTable *pTo;            /* Table zTbl */
1596cf8e9144Sdan 
1597cf8e9144Sdan     /* Locate and if necessary initialize the target table object */
1598cf8e9144Sdan     rc = sessionFindTable(pSession, zTbl, &pTo);
1599cf8e9144Sdan     if( pTo==0 ) goto diff_out;
16006dc29e60Sdan     if( sessionInitTable(pSession, pTo) ){
16016dc29e60Sdan       rc = pSession->rc;
16026dc29e60Sdan       goto diff_out;
1603cf8e9144Sdan     }
1604cf8e9144Sdan 
1605cf8e9144Sdan     /* Check the table schemas match */
1606cf8e9144Sdan     if( rc==SQLITE_OK ){
16074cc923e3Sdan       int bHasPk = 0;
1608b9db9099Sdan       int bMismatch = 0;
1609cf8e9144Sdan       int nCol;                   /* Columns in zFrom.zTbl */
1610cf8e9144Sdan       u8 *abPK;
1611cf8e9144Sdan       const char **azCol = 0;
1612cf8e9144Sdan       rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK);
1613cf8e9144Sdan       if( rc==SQLITE_OK ){
16144cc923e3Sdan         if( pTo->nCol!=nCol ){
1615cf8e9144Sdan           bMismatch = 1;
1616cf8e9144Sdan         }else{
1617cf8e9144Sdan           int i;
1618cf8e9144Sdan           for(i=0; i<nCol; i++){
16194cc923e3Sdan             if( pTo->abPK[i]!=abPK[i] ) bMismatch = 1;
1620cf8e9144Sdan             if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1;
16214cc923e3Sdan             if( abPK[i] ) bHasPk = 1;
1622cf8e9144Sdan           }
1623cf8e9144Sdan         }
1624cf8e9144Sdan       }
1625dbbd8160Sdrh       sqlite3_free((char*)azCol);
1626b9db9099Sdan       if( bMismatch ){
1627595d9f5fSdan         if( pzErrMsg ){
1628b9db9099Sdan           *pzErrMsg = sqlite3_mprintf("table schemas do not match");
1629595d9f5fSdan         }
1630b9db9099Sdan         rc = SQLITE_SCHEMA;
1631b9db9099Sdan       }
1632b9db9099Sdan       if( bHasPk==0 ){
1633b9db9099Sdan         /* Ignore tables with no primary keys */
1634b9db9099Sdan         goto diff_out;
1635b9db9099Sdan       }
1636cf8e9144Sdan     }
1637cf8e9144Sdan 
1638cf8e9144Sdan     if( rc==SQLITE_OK ){
1639cf8e9144Sdan       zExpr = sessionExprComparePK(pTo->nCol,
1640cf8e9144Sdan           zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK
1641cf8e9144Sdan       );
1642cf8e9144Sdan     }
1643cf8e9144Sdan 
1644cf8e9144Sdan     /* Find new rows */
1645cf8e9144Sdan     if( rc==SQLITE_OK ){
1646cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr);
1647cf8e9144Sdan     }
1648cf8e9144Sdan 
1649cf8e9144Sdan     /* Find old rows */
1650cf8e9144Sdan     if( rc==SQLITE_OK ){
1651cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr);
1652cf8e9144Sdan     }
1653cf8e9144Sdan 
1654cf8e9144Sdan     /* Find modified rows */
1655cf8e9144Sdan     if( rc==SQLITE_OK ){
1656cf8e9144Sdan       rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr);
1657cf8e9144Sdan     }
1658cf8e9144Sdan 
1659cf8e9144Sdan     sqlite3_free(zExpr);
1660cf8e9144Sdan   }
1661cf8e9144Sdan 
1662cf8e9144Sdan  diff_out:
1663cf8e9144Sdan   sessionPreupdateHooks(pSession);
166410dc553cSdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
1665cf8e9144Sdan   return rc;
1666296c7658Sdan }
16674fccf43aSdan 
16684fccf43aSdan /*
16694fccf43aSdan ** Create a session object. This session object will record changes to
16704fccf43aSdan ** database zDb attached to connection db.
16714fccf43aSdan */
16724fccf43aSdan int sqlite3session_create(
16734fccf43aSdan   sqlite3 *db,                    /* Database handle */
16744fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
16754fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
16764fccf43aSdan ){
1677296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1678296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1679cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
16804fccf43aSdan 
1681296c7658Sdan   /* Zero the output value in case an error occurs. */
16824fccf43aSdan   *ppSession = 0;
16834fccf43aSdan 
16844fccf43aSdan   /* Allocate and populate the new session object. */
16852d77d80aSdrh   pNew = (sqlite3_session *)sqlite3_malloc64(sizeof(sqlite3_session) + nDb + 1);
16864fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
16874fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
16884fccf43aSdan   pNew->db = db;
16894fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1690296c7658Sdan   pNew->bEnable = 1;
16914fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
1692cf8e9144Sdan   sessionPreupdateHooks(pNew);
16934fccf43aSdan 
16944fccf43aSdan   /* Add the new session object to the linked list of session objects
16954fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
16964fccf43aSdan   ** handle mutex.  */
16974fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
16984fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
16994fccf43aSdan   pNew->pNext = pOld;
17004fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
17014fccf43aSdan 
17024fccf43aSdan   *ppSession = pNew;
17034fccf43aSdan   return SQLITE_OK;
17044fccf43aSdan }
17054fccf43aSdan 
170677fc1d5bSdan /*
170777fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
170877fc1d5bSdan ** of the changed-rows hash tables are also deleted.
170977fc1d5bSdan */
17101ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
17115d607a6eSdan   SessionTable *pNext;
17125d607a6eSdan   SessionTable *pTab;
17135d607a6eSdan 
17145d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
17155d607a6eSdan     int i;
17165d607a6eSdan     pNext = pTab->pNext;
17175d607a6eSdan     for(i=0; i<pTab->nChange; i++){
17185d607a6eSdan       SessionChange *p;
171902d436b1Smistachkin       SessionChange *pNextChange;
172002d436b1Smistachkin       for(p=pTab->apChange[i]; p; p=pNextChange){
172102d436b1Smistachkin         pNextChange = p->pNext;
17225d607a6eSdan         sqlite3_free(p);
17235d607a6eSdan       }
17245d607a6eSdan     }
17255d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
17265d607a6eSdan     sqlite3_free(pTab->apChange);
17275d607a6eSdan     sqlite3_free(pTab);
17285d607a6eSdan   }
17295d607a6eSdan }
17305d607a6eSdan 
17314fccf43aSdan /*
17324fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
17334fccf43aSdan */
17344fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
17354fccf43aSdan   sqlite3 *db = pSession->db;
17364fccf43aSdan   sqlite3_session *pHead;
17374fccf43aSdan   sqlite3_session **pp;
17384fccf43aSdan 
1739296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1740296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
17414fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
17424fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
174350d348b1Sdrh   for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){
174450d348b1Sdrh     if( (*pp)==pSession ){
17454fccf43aSdan       *pp = (*pp)->pNext;
17464fccf43aSdan       if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead);
174750d348b1Sdrh       break;
174850d348b1Sdrh     }
174950d348b1Sdrh   }
17504fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
17511611e5a3Sdan   sqlite3ValueFree(pSession->pZeroBlob);
17524fccf43aSdan 
1753296c7658Sdan   /* Delete all attached table objects. And the contents of their
1754296c7658Sdan   ** associated hash-tables. */
17555d607a6eSdan   sessionDeleteTable(pSession->pTable);
17564fccf43aSdan 
1757296c7658Sdan   /* Free the session object itself. */
17584fccf43aSdan   sqlite3_free(pSession);
17594fccf43aSdan }
17604fccf43aSdan 
17614fccf43aSdan /*
17627531a5a3Sdan ** Set a table filter on a Session Object.
17637531a5a3Sdan */
17647531a5a3Sdan void sqlite3session_table_filter(
17657531a5a3Sdan   sqlite3_session *pSession,
17667531a5a3Sdan   int(*xFilter)(void*, const char*),
17677531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
17687531a5a3Sdan ){
17697531a5a3Sdan   pSession->bAutoAttach = 1;
17707531a5a3Sdan   pSession->pFilterCtx = pCtx;
17717531a5a3Sdan   pSession->xTableFilter = xFilter;
17727531a5a3Sdan }
17737531a5a3Sdan 
17747531a5a3Sdan /*
17754fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
17764fccf43aSdan ** while the session object is enabled will be recorded.
17774fccf43aSdan **
17784fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
17794fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
17804fccf43aSdan ** or not.
17814fccf43aSdan */
17824fccf43aSdan int sqlite3session_attach(
17834fccf43aSdan   sqlite3_session *pSession,      /* Session object */
17844fccf43aSdan   const char *zName               /* Table name */
17854fccf43aSdan ){
1786ff4d0f41Sdan   int rc = SQLITE_OK;
1787ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1788ff4d0f41Sdan 
1789ff4d0f41Sdan   if( !zName ){
1790ff4d0f41Sdan     pSession->bAutoAttach = 1;
1791ff4d0f41Sdan   }else{
1792296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1793296c7658Sdan     int nName;                    /* Number of bytes in string zName */
17944fccf43aSdan 
17954fccf43aSdan     /* First search for an existing entry. If one is found, this call is
17964fccf43aSdan     ** a no-op. Return early. */
1797cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
17984fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
17994c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
18004fccf43aSdan     }
18014fccf43aSdan 
18024c220252Sdan     if( !pTab ){
18034fccf43aSdan       /* Allocate new SessionTable object. */
18042d77d80aSdrh       pTab = (SessionTable *)sqlite3_malloc64(sizeof(SessionTable) + nName + 1);
18054c220252Sdan       if( !pTab ){
18064c220252Sdan         rc = SQLITE_NOMEM;
18074c220252Sdan       }else{
18086c39e6a8Sdan         /* Populate the new SessionTable object and link it into the list.
18096c39e6a8Sdan         ** The new object must be linked onto the end of the list, not
18106c39e6a8Sdan         ** simply added to the start of it in order to ensure that tables
18116c39e6a8Sdan         ** appear in the correct order when a changeset or patchset is
18126c39e6a8Sdan         ** eventually generated. */
18136c39e6a8Sdan         SessionTable **ppTab;
18144fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
18154fccf43aSdan         pTab->zName = (char *)&pTab[1];
18164fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
18176c39e6a8Sdan         for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext);
18186c39e6a8Sdan         *ppTab = pTab;
18194c220252Sdan       }
18204c220252Sdan     }
1821ff4d0f41Sdan   }
18224fccf43aSdan 
18234c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
18244c220252Sdan   return rc;
18254fccf43aSdan }
18264fccf43aSdan 
1827296c7658Sdan /*
1828296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1829296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1830296c7658Sdan **
1831296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1832296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1833296c7658Sdan */
1834f6ad201aSdrh static int sessionBufferGrow(SessionBuffer *p, size_t nByte, int *pRc){
18353196abb7Smistachkin   if( *pRc==SQLITE_OK && (size_t)(p->nAlloc-p->nBuf)<nByte ){
18364fccf43aSdan     u8 *aNew;
18379c18ef09Sdan     i64 nNew = p->nAlloc ? p->nAlloc : 128;
18384fccf43aSdan     do {
18394fccf43aSdan       nNew = nNew*2;
1840ac713403Smistachkin     }while( (size_t)(nNew-p->nBuf)<nByte );
18414fccf43aSdan 
18429c18ef09Sdan     aNew = (u8 *)sqlite3_realloc64(p->aBuf, nNew);
18434fccf43aSdan     if( 0==aNew ){
18444fccf43aSdan       *pRc = SQLITE_NOMEM;
184580fe2d93Sdan     }else{
18464fccf43aSdan       p->aBuf = aNew;
18474fccf43aSdan       p->nAlloc = nNew;
18484fccf43aSdan     }
184980fe2d93Sdan   }
185080fe2d93Sdan   return (*pRc!=SQLITE_OK);
18514fccf43aSdan }
18524fccf43aSdan 
1853296c7658Sdan /*
1854fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1855fa122adaSdan ** as the first.
1856fa122adaSdan **
1857fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1858fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1859fa122adaSdan ** before returning.
1860fa122adaSdan */
1861fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1862fa122adaSdan   int rc = *pRc;
1863fa122adaSdan   if( rc==SQLITE_OK ){
18642d77d80aSdrh     sqlite3_int64 nByte = 0;
1865e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1866fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1867fa122adaSdan     if( rc==SQLITE_OK ){
1868fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1869fa122adaSdan       p->nBuf += nByte;
1870fa122adaSdan     }else{
1871fa122adaSdan       *pRc = rc;
1872fa122adaSdan     }
1873fa122adaSdan   }
1874fa122adaSdan }
1875fa122adaSdan 
1876fa122adaSdan /*
1877296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1878296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1879296c7658Sdan **
1880296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1881296c7658Sdan ** returning.
1882296c7658Sdan */
18834fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
188480fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
18854fccf43aSdan     p->aBuf[p->nBuf++] = v;
18864fccf43aSdan   }
18874fccf43aSdan }
18884fccf43aSdan 
1889296c7658Sdan /*
1890296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1891296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1892296c7658Sdan **
1893296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1894296c7658Sdan ** returning.
1895296c7658Sdan */
1896cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
189780fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
18984fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
18994fccf43aSdan   }
19004fccf43aSdan }
19014fccf43aSdan 
1902296c7658Sdan /*
1903296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1904296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1905296c7658Sdan **
1906296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1907296c7658Sdan ** returning.
1908296c7658Sdan */
19094fccf43aSdan static void sessionAppendBlob(
19104fccf43aSdan   SessionBuffer *p,
19114fccf43aSdan   const u8 *aBlob,
19124fccf43aSdan   int nBlob,
19134fccf43aSdan   int *pRc
19144fccf43aSdan ){
1915895decf6Sdan   if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){
19164fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
19174fccf43aSdan     p->nBuf += nBlob;
19184fccf43aSdan   }
19194fccf43aSdan }
19204fccf43aSdan 
1921296c7658Sdan /*
1922296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1923296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1924296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1925296c7658Sdan **
1926296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1927296c7658Sdan ** returning.
1928296c7658Sdan */
1929d5f0767cSdan static void sessionAppendStr(
1930d5f0767cSdan   SessionBuffer *p,
1931d5f0767cSdan   const char *zStr,
1932d5f0767cSdan   int *pRc
1933d5f0767cSdan ){
1934cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
193580fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1936d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1937d5f0767cSdan     p->nBuf += nStr;
1938d5f0767cSdan   }
1939d5f0767cSdan }
1940d5f0767cSdan 
1941296c7658Sdan /*
1942296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1943296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1944296c7658Sdan ** to the buffer. No nul-terminator is written.
1945296c7658Sdan **
1946296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1947296c7658Sdan ** returning.
1948296c7658Sdan */
1949d5f0767cSdan static void sessionAppendInteger(
1950296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1951296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1952296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1953d5f0767cSdan ){
1954d5f0767cSdan   char aBuf[24];
1955d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1956d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1957d5f0767cSdan }
1958d5f0767cSdan 
1959296c7658Sdan /*
1960296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1961296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1962296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1963296c7658Sdan ** nul-terminator byte is written.
1964296c7658Sdan **
1965296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1966296c7658Sdan ** returning.
1967296c7658Sdan */
1968d5f0767cSdan static void sessionAppendIdent(
1969296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1970296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1971296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1972d5f0767cSdan ){
1973cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
197480fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1975d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1976d5f0767cSdan     const char *zIn = zStr;
1977d5f0767cSdan     *zOut++ = '"';
1978d5f0767cSdan     while( *zIn ){
1979d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1980d5f0767cSdan       *zOut++ = *(zIn++);
1981d5f0767cSdan     }
1982d5f0767cSdan     *zOut++ = '"';
1983cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1984d5f0767cSdan   }
1985d5f0767cSdan }
1986d5f0767cSdan 
1987296c7658Sdan /*
1988296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1989296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1990296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1991296c7658Sdan ** to to the buffer.
1992296c7658Sdan */
19934fccf43aSdan static void sessionAppendCol(
1994296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1995296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1996296c7658Sdan   int iCol,                       /* Column to read value from */
1997296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
19984fccf43aSdan ){
19994fccf43aSdan   if( *pRc==SQLITE_OK ){
20004fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
20014fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
20024fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
20034fccf43aSdan       sqlite3_int64 i;
20044fccf43aSdan       u8 aBuf[8];
20054fccf43aSdan       if( eType==SQLITE_INTEGER ){
20064fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
20074fccf43aSdan       }else{
20084fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
20094fccf43aSdan         memcpy(&i, &r, 8);
20104fccf43aSdan       }
2011296c7658Sdan       sessionPutI64(aBuf, i);
20124fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
20134fccf43aSdan     }
20144fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
20156734007dSdan       u8 *z;
20163cc89d95Sdan       int nByte;
20176734007dSdan       if( eType==SQLITE_BLOB ){
20186734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
20196734007dSdan       }else{
20206734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
20216734007dSdan       }
20223cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
20233cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
20244fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
20256734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
20266734007dSdan       }else{
20276734007dSdan         *pRc = SQLITE_NOMEM;
20286734007dSdan       }
20294fccf43aSdan     }
20304fccf43aSdan   }
20314fccf43aSdan }
20324fccf43aSdan 
2033296c7658Sdan /*
2034296c7658Sdan **
203580fe2d93Sdan ** This function appends an update change to the buffer (see the comments
203680fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
203780fe2d93Sdan ** consists of:
2038296c7658Sdan **
2039296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
2040296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
2041296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
2042296c7658Sdan **
2043296c7658Sdan ** The SessionChange object passed as the third argument contains the
2044296c7658Sdan ** values that were stored in the row when the session began (the old.*
2045296c7658Sdan ** values). The statement handle passed as the second argument points
2046296c7658Sdan ** at the current version of the row (the new.* values).
2047296c7658Sdan **
2048296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
2049296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
2050296c7658Sdan **
2051296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
2052296c7658Sdan ** original values of any fields that have been modified. The new.* record
2053296c7658Sdan ** contains the new values of only those fields that have been modified.
2054296c7658Sdan */
205580fe2d93Sdan static int sessionAppendUpdate(
2056296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
205773b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
2058296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
2059296c7658Sdan   SessionChange *p,               /* Object containing old values */
206080fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
20614fccf43aSdan ){
206280fe2d93Sdan   int rc = SQLITE_OK;
2063296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
2064296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
20651f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
2066296c7658Sdan   int i;                        /* Used to iterate through columns */
2067296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
2068296c7658Sdan 
206980fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
207080fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
20714fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
207237f133ecSdan     int bChanged = 0;
20734fccf43aSdan     int nAdvance;
20744fccf43aSdan     int eType = *pCsr;
20754fccf43aSdan     switch( eType ){
20764fccf43aSdan       case SQLITE_NULL:
20774fccf43aSdan         nAdvance = 1;
20784fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
207937f133ecSdan           bChanged = 1;
20804fccf43aSdan         }
20814fccf43aSdan         break;
20824fccf43aSdan 
20834fccf43aSdan       case SQLITE_FLOAT:
20844fccf43aSdan       case SQLITE_INTEGER: {
20854fccf43aSdan         nAdvance = 9;
20864fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
20874fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
20884fccf43aSdan           if( eType==SQLITE_INTEGER ){
20894fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
20904fccf43aSdan           }else{
20914fccf43aSdan             double dVal;
20924fccf43aSdan             memcpy(&dVal, &iVal, 8);
20934fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
20944fccf43aSdan           }
20954fccf43aSdan         }
209637f133ecSdan         bChanged = 1;
20974fccf43aSdan         break;
20984fccf43aSdan       }
20994fccf43aSdan 
2100e5754eecSdan       default: {
2101895decf6Sdan         int n;
2102895decf6Sdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &n);
2103e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
2104895decf6Sdan         nAdvance = nHdr + n;
21054fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
2106895decf6Sdan          && n==sqlite3_column_bytes(pStmt, i)
2107895decf6Sdan          && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n))
21084fccf43aSdan         ){
21094fccf43aSdan           break;
21104fccf43aSdan         }
211137f133ecSdan         bChanged = 1;
21124fccf43aSdan       }
21134fccf43aSdan     }
21144fccf43aSdan 
211573b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
211673b3c055Sdan     if( bChanged ) bNoop = 0;
211773b3c055Sdan 
211873b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
211973b3c055Sdan     ** currently generating a patchset. */
212073b3c055Sdan     if( bPatchset==0 ){
212137f133ecSdan       if( bChanged || abPK[i] ){
212280fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
21234fccf43aSdan       }else{
212480fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
212537f133ecSdan       }
212673b3c055Sdan     }
212737f133ecSdan 
212873b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
212973b3c055Sdan     ** generating a patchset.  */
213073b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
213180fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
213237f133ecSdan     }else{
213380fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
21344fccf43aSdan     }
213537f133ecSdan 
21364fccf43aSdan     pCsr += nAdvance;
21374fccf43aSdan   }
21384fccf43aSdan 
21394fccf43aSdan   if( bNoop ){
21401f34f8ccSdan     pBuf->nBuf = nRewind;
21414fccf43aSdan   }else{
214280fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
21434fccf43aSdan   }
21441f34f8ccSdan   sqlite3_free(buf2.aBuf);
214580fe2d93Sdan 
214680fe2d93Sdan   return rc;
2147d5f0767cSdan }
21484fccf43aSdan 
2149a71d2371Sdan /*
2150a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
2151a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
2152a71d2371Sdan ** format otherwise.
2153a71d2371Sdan */
215473b3c055Sdan static int sessionAppendDelete(
215573b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
215673b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
215773b3c055Sdan   SessionChange *p,               /* Object containing old values */
2158a71d2371Sdan   int nCol,                       /* Number of columns in table */
215973b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
216073b3c055Sdan ){
216173b3c055Sdan   int rc = SQLITE_OK;
216273b3c055Sdan 
216373b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
216473b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
216573b3c055Sdan 
216673b3c055Sdan   if( bPatchset==0 ){
216773b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
216873b3c055Sdan   }else{
216973b3c055Sdan     int i;
217073b3c055Sdan     u8 *a = p->aRecord;
217173b3c055Sdan     for(i=0; i<nCol; i++){
217273b3c055Sdan       u8 *pStart = a;
217373b3c055Sdan       int eType = *a++;
217473b3c055Sdan 
217573b3c055Sdan       switch( eType ){
217673b3c055Sdan         case 0:
217773b3c055Sdan         case SQLITE_NULL:
217873b3c055Sdan           assert( abPK[i]==0 );
217973b3c055Sdan           break;
218073b3c055Sdan 
218173b3c055Sdan         case SQLITE_FLOAT:
218273b3c055Sdan         case SQLITE_INTEGER:
218373b3c055Sdan           a += 8;
218473b3c055Sdan           break;
218573b3c055Sdan 
218673b3c055Sdan         default: {
218773b3c055Sdan           int n;
218873b3c055Sdan           a += sessionVarintGet(a, &n);
218973b3c055Sdan           a += n;
219073b3c055Sdan           break;
219173b3c055Sdan         }
219273b3c055Sdan       }
219373b3c055Sdan       if( abPK[i] ){
2194f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
219573b3c055Sdan       }
219673b3c055Sdan     }
219773b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
219873b3c055Sdan   }
219973b3c055Sdan 
220073b3c055Sdan   return rc;
220173b3c055Sdan }
220273b3c055Sdan 
220377fc1d5bSdan /*
220477fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
220577fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
220677fc1d5bSdan **
220777fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
220877fc1d5bSdan */
2209e8d5648eSdan static int sessionSelectStmt(
2210e8d5648eSdan   sqlite3 *db,                    /* Database handle */
2211d7fb7d24Sdan   const char *zDb,                /* Database name */
2212e8d5648eSdan   const char *zTab,               /* Table name */
221377fc1d5bSdan   int nCol,                       /* Number of columns in table */
221477fc1d5bSdan   const char **azCol,             /* Names of table columns */
221577fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
221677fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
2217d5f0767cSdan ){
2218e8d5648eSdan   int rc = SQLITE_OK;
22193739f298Sdan   char *zSql = 0;
22203739f298Sdan   int nSql = -1;
22213739f298Sdan 
22223739f298Sdan   if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){
22233739f298Sdan     zSql = sqlite3_mprintf(
22243739f298Sdan         "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND "
22253739f298Sdan         "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb
22263739f298Sdan     );
2227bd45374cSdan     if( zSql==0 ) rc = SQLITE_NOMEM;
22283739f298Sdan   }else{
2229d5f0767cSdan     int i;
2230e8d5648eSdan     const char *zSep = "";
2231e8d5648eSdan     SessionBuffer buf = {0, 0, 0};
2232d5f0767cSdan 
2233e8d5648eSdan     sessionAppendStr(&buf, "SELECT * FROM ", &rc);
2234d7fb7d24Sdan     sessionAppendIdent(&buf, zDb, &rc);
2235d7fb7d24Sdan     sessionAppendStr(&buf, ".", &rc);
2236e8d5648eSdan     sessionAppendIdent(&buf, zTab, &rc);
2237e8d5648eSdan     sessionAppendStr(&buf, " WHERE ", &rc);
2238e8d5648eSdan     for(i=0; i<nCol; i++){
2239e8d5648eSdan       if( abPK[i] ){
2240e8d5648eSdan         sessionAppendStr(&buf, zSep, &rc);
2241e8d5648eSdan         sessionAppendIdent(&buf, azCol[i], &rc);
22423739f298Sdan         sessionAppendStr(&buf, " IS ?", &rc);
2243e8d5648eSdan         sessionAppendInteger(&buf, i+1, &rc);
2244e8d5648eSdan         zSep = " AND ";
2245d5f0767cSdan       }
2246d5f0767cSdan     }
22473739f298Sdan     zSql = (char*)buf.aBuf;
22483739f298Sdan     nSql = buf.nBuf;
2249d5f0767cSdan   }
22503739f298Sdan 
22513739f298Sdan   if( rc==SQLITE_OK ){
22523739f298Sdan     rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0);
22533739f298Sdan   }
22543739f298Sdan   sqlite3_free(zSql);
2255e8d5648eSdan   return rc;
2256d5f0767cSdan }
2257d5f0767cSdan 
225877fc1d5bSdan /*
225977fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
226077fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
226177fc1d5bSdan ** is as prepared by function sessionSelectStmt().
226277fc1d5bSdan **
226377fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
226477fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
226577fc1d5bSdan */
2266e8d5648eSdan static int sessionSelectBind(
226777fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
226877fc1d5bSdan   int nCol,                       /* Number of columns in table */
226977fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
227077fc1d5bSdan   SessionChange *pChange          /* Change structure */
2271e8d5648eSdan ){
2272e8d5648eSdan   int i;
2273e8d5648eSdan   int rc = SQLITE_OK;
2274e5754eecSdan   u8 *a = pChange->aRecord;
2275d5f0767cSdan 
2276e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2277e8d5648eSdan     int eType = *a++;
2278e8d5648eSdan 
2279e8d5648eSdan     switch( eType ){
228080fe2d93Sdan       case 0:
2281e8d5648eSdan       case SQLITE_NULL:
22821611e5a3Sdan         assert( abPK[i]==0 );
2283e8d5648eSdan         break;
2284e8d5648eSdan 
2285e8d5648eSdan       case SQLITE_INTEGER: {
2286e8d5648eSdan         if( abPK[i] ){
2287e8d5648eSdan           i64 iVal = sessionGetI64(a);
2288e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
2289e8d5648eSdan         }
2290e8d5648eSdan         a += 8;
2291e8d5648eSdan         break;
2292d5f0767cSdan       }
2293296c7658Sdan 
2294e8d5648eSdan       case SQLITE_FLOAT: {
2295e8d5648eSdan         if( abPK[i] ){
2296e8d5648eSdan           double rVal;
2297e8d5648eSdan           i64 iVal = sessionGetI64(a);
2298e8d5648eSdan           memcpy(&rVal, &iVal, 8);
22994e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
2300d5f0767cSdan         }
2301e8d5648eSdan         a += 8;
2302e8d5648eSdan         break;
2303e8d5648eSdan       }
2304e8d5648eSdan 
2305e8d5648eSdan       case SQLITE_TEXT: {
2306e8d5648eSdan         int n;
2307e8d5648eSdan         a += sessionVarintGet(a, &n);
2308e8d5648eSdan         if( abPK[i] ){
2309e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
2310e8d5648eSdan         }
2311e8d5648eSdan         a += n;
2312e8d5648eSdan         break;
2313e8d5648eSdan       }
2314e8d5648eSdan 
2315e5754eecSdan       default: {
2316e8d5648eSdan         int n;
2317e5754eecSdan         assert( eType==SQLITE_BLOB );
2318e8d5648eSdan         a += sessionVarintGet(a, &n);
2319e8d5648eSdan         if( abPK[i] ){
2320e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
2321e8d5648eSdan         }
2322e8d5648eSdan         a += n;
2323e8d5648eSdan         break;
2324e8d5648eSdan       }
2325e8d5648eSdan     }
2326e8d5648eSdan   }
2327e8d5648eSdan 
2328d5f0767cSdan   return rc;
23294fccf43aSdan }
23304fccf43aSdan 
233177fc1d5bSdan /*
233277fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
233377fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
233477fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
233577fc1d5bSdan ** SQLite error code before returning.
233677fc1d5bSdan */
23375d607a6eSdan static void sessionAppendTableHdr(
2338a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
2339a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
2340a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
2341a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
23425d607a6eSdan ){
23435d607a6eSdan   /* Write a table header */
234473b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
23455d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
23465d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
23474f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
23485d607a6eSdan }
23495d607a6eSdan 
2350a71d2371Sdan /*
2351a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
2352a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
2353a71d2371Sdan ** passed as the first argument.
2354a71d2371Sdan **
2355a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
2356a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
2357a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
2358a71d2371Sdan ** to 0.
2359a71d2371Sdan */
2360adf3bf58Sdrh static int sessionGenerateChangeset(
23614fccf43aSdan   sqlite3_session *pSession,      /* Session object */
236273b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
2363ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2364ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
23654fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
23664fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
23674fccf43aSdan ){
2368296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
2369296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
2370296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
2371296c7658Sdan   int rc;                         /* Return code */
23724fccf43aSdan 
2373ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
2374ef7a6304Sdan 
2375296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
2376296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
2377296c7658Sdan   ** this call will be a no-op.  */
2378ef7a6304Sdan   if( xOutput==0 ){
23794fccf43aSdan     *pnChangeset = 0;
23804fccf43aSdan     *ppChangeset = 0;
2381ef7a6304Sdan   }
2382e5754eecSdan 
2383e5754eecSdan   if( pSession->rc ) return pSession->rc;
2384e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
2385e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
2386e5754eecSdan 
2387e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
23884fccf43aSdan 
23894fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
23904fccf43aSdan     if( pTab->nEntry ){
2391d7fb7d24Sdan       const char *zName = pTab->zName;
2392a9605b91Sdan       int nCol;                   /* Number of columns in table */
2393a9605b91Sdan       u8 *abPK;                   /* Primary key array */
2394a9605b91Sdan       const char **azCol = 0;     /* Table columns */
23951f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
23961f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
23971f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
23981f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
23994fccf43aSdan 
2400a9605b91Sdan       /* Check the table schema is still Ok. */
2401a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
2402a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
2403a9605b91Sdan         rc = SQLITE_SCHEMA;
2404a9605b91Sdan       }
2405a9605b91Sdan 
24064fccf43aSdan       /* Write a table header */
240773b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
24084fccf43aSdan 
24094fccf43aSdan       /* Build and compile a statement to execute: */
24104fccf43aSdan       if( rc==SQLITE_OK ){
2411d7fb7d24Sdan         rc = sessionSelectStmt(
2412a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
24134fccf43aSdan       }
24144fccf43aSdan 
24151f34f8ccSdan       nNoop = buf.nBuf;
241612ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
2417e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
2418e8d5648eSdan 
24194fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
2420e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
242180fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
24221f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
2423798693b2Sdan             if( p->op==SQLITE_INSERT ){
24244fccf43aSdan               int iCol;
24254fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
2426b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
2427e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
24281f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
24294fccf43aSdan               }
2430e8d5648eSdan             }else{
243173b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
24324fccf43aSdan             }
2433798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
2434a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
24354fccf43aSdan           }
243612ca0b56Sdan           if( rc==SQLITE_OK ){
24371f34f8ccSdan             rc = sqlite3_reset(pSel);
24384fccf43aSdan           }
2439ef7a6304Sdan 
24401f48e67dSdan           /* If the buffer is now larger than sessions_strm_chunk_size, pass
2441ef7a6304Sdan           ** its contents to the xOutput() callback. */
2442ef7a6304Sdan           if( xOutput
2443ef7a6304Sdan            && rc==SQLITE_OK
2444ef7a6304Sdan            && buf.nBuf>nNoop
24451f48e67dSdan            && buf.nBuf>sessions_strm_chunk_size
2446ef7a6304Sdan           ){
2447ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2448ef7a6304Sdan             nNoop = -1;
2449ef7a6304Sdan             buf.nBuf = 0;
2450ef7a6304Sdan           }
2451ef7a6304Sdan 
24524fccf43aSdan         }
2453e8d5648eSdan       }
24544fccf43aSdan 
24551f34f8ccSdan       sqlite3_finalize(pSel);
24561f34f8ccSdan       if( buf.nBuf==nNoop ){
24574fccf43aSdan         buf.nBuf = nRewind;
24584fccf43aSdan       }
2459cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
24604fccf43aSdan     }
24614fccf43aSdan   }
24624fccf43aSdan 
24634fccf43aSdan   if( rc==SQLITE_OK ){
2464ef7a6304Sdan     if( xOutput==0 ){
24654fccf43aSdan       *pnChangeset = buf.nBuf;
24664fccf43aSdan       *ppChangeset = buf.aBuf;
2467ef7a6304Sdan       buf.aBuf = 0;
2468ef7a6304Sdan     }else if( buf.nBuf>0 ){
2469ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2470ef7a6304Sdan     }
24714fccf43aSdan   }
24724c220252Sdan 
2473ef7a6304Sdan   sqlite3_free(buf.aBuf);
2474e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
24754c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
24764fccf43aSdan   return rc;
24774fccf43aSdan }
24784fccf43aSdan 
2479296c7658Sdan /*
248073b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
248173b3c055Sdan ** session object passed as the first argument.
248273b3c055Sdan **
248373b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
248473b3c055Sdan ** using sqlite3_free().
248573b3c055Sdan */
248673b3c055Sdan int sqlite3session_changeset(
248773b3c055Sdan   sqlite3_session *pSession,      /* Session object */
248873b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
248973b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
249073b3c055Sdan ){
2491ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2492ef7a6304Sdan }
2493ef7a6304Sdan 
2494ef7a6304Sdan /*
2495ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2496ef7a6304Sdan */
2497f1a08ad8Sdrh int sqlite3session_changeset_strm(
2498ef7a6304Sdan   sqlite3_session *pSession,
2499ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2500ef7a6304Sdan   void *pOut
2501ef7a6304Sdan ){
2502ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2503ef7a6304Sdan }
2504ef7a6304Sdan 
2505ef7a6304Sdan /*
2506ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2507ef7a6304Sdan */
2508f1a08ad8Sdrh int sqlite3session_patchset_strm(
2509ef7a6304Sdan   sqlite3_session *pSession,
2510ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2511ef7a6304Sdan   void *pOut
2512ef7a6304Sdan ){
2513ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
251473b3c055Sdan }
251573b3c055Sdan 
251673b3c055Sdan /*
251773b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
251873b3c055Sdan ** session object passed as the first argument.
251973b3c055Sdan **
252073b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
252173b3c055Sdan ** using sqlite3_free().
252273b3c055Sdan */
252373b3c055Sdan int sqlite3session_patchset(
252473b3c055Sdan   sqlite3_session *pSession,      /* Session object */
252573b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
252673b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
252773b3c055Sdan ){
2528ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
252973b3c055Sdan }
253073b3c055Sdan 
253173b3c055Sdan /*
2532296c7658Sdan ** Enable or disable the session object passed as the first argument.
2533296c7658Sdan */
25344fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
25354c220252Sdan   int ret;
25364c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2537296c7658Sdan   if( bEnable>=0 ){
2538296c7658Sdan     pSession->bEnable = bEnable;
25394fccf43aSdan   }
25404c220252Sdan   ret = pSession->bEnable;
25414c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
25424c220252Sdan   return ret;
2543296c7658Sdan }
25444fccf43aSdan 
25454fccf43aSdan /*
2546b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2547b4480e94Sdan */
2548b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2549b4480e94Sdan   int ret;
2550b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2551b4480e94Sdan   if( bIndirect>=0 ){
2552b4480e94Sdan     pSession->bIndirect = bIndirect;
2553b4480e94Sdan   }
2554b4480e94Sdan   ret = pSession->bIndirect;
2555b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2556b4480e94Sdan   return ret;
2557b4480e94Sdan }
2558b4480e94Sdan 
2559b4480e94Sdan /*
2560b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2561b69ec348Sdan ** by the session object passed as the only argument.
2562b69ec348Sdan */
2563b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2564b69ec348Sdan   int ret = 0;
2565b69ec348Sdan   SessionTable *pTab;
2566b69ec348Sdan 
2567b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2568b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2569b69ec348Sdan     ret = (pTab->nEntry>0);
2570b69ec348Sdan   }
2571b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2572b69ec348Sdan 
2573ff530326Sdan   return (ret==0);
2574b69ec348Sdan }
2575b69ec348Sdan 
2576b69ec348Sdan /*
2577f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
25784fccf43aSdan */
2579adf3bf58Sdrh static int sessionChangesetStart(
2580296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2581ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2582ef7a6304Sdan   void *pIn,
2583296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
258444748f27Sdan   void *pChangeset,               /* Pointer to buffer containing changeset */
258544748f27Sdan   int bInvert                     /* True to invert changeset */
25864fccf43aSdan ){
25874fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
25884fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
25894fccf43aSdan 
2590ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2591ef7a6304Sdan 
2592296c7658Sdan   /* Zero the output variable in case an error occurs. */
2593296c7658Sdan   *pp = 0;
25944fccf43aSdan 
2595296c7658Sdan   /* Allocate and initialize the iterator structure. */
25964fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
25974fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
25984fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
25994fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
26004757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
26014757c658Sdan   pRet->in.nData = nChangeset;
2602ef7a6304Sdan   pRet->in.xInput = xInput;
2603ef7a6304Sdan   pRet->in.pIn = pIn;
2604ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
260544748f27Sdan   pRet->bInvert = bInvert;
26064fccf43aSdan 
2607296c7658Sdan   /* Populate the output variable and return success. */
2608296c7658Sdan   *pp = pRet;
26094fccf43aSdan   return SQLITE_OK;
26104fccf43aSdan }
26114fccf43aSdan 
2612296c7658Sdan /*
2613ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2614ef7a6304Sdan */
2615ef7a6304Sdan int sqlite3changeset_start(
2616ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2617ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2618ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2619ef7a6304Sdan ){
262044748f27Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, 0);
2621ef7a6304Sdan }
262246de0728Sdan int sqlite3changeset_start_v2(
262346de0728Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
262446de0728Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
262546de0728Sdan   void *pChangeset,               /* Pointer to buffer containing changeset */
262646de0728Sdan   int flags
262746de0728Sdan ){
262846de0728Sdan   int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT);
262946de0728Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, bInvert);
263046de0728Sdan }
2631ef7a6304Sdan 
2632ef7a6304Sdan /*
2633ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2634ef7a6304Sdan */
2635f1a08ad8Sdrh int sqlite3changeset_start_strm(
2636ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2637ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2638ef7a6304Sdan   void *pIn
2639ef7a6304Sdan ){
264044748f27Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0, 0);
2641ef7a6304Sdan }
264246de0728Sdan int sqlite3changeset_start_v2_strm(
264346de0728Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
264446de0728Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
264546de0728Sdan   void *pIn,
264646de0728Sdan   int flags
264746de0728Sdan ){
264846de0728Sdan   int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT);
264946de0728Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0, bInvert);
265046de0728Sdan }
2651ef7a6304Sdan 
2652ef7a6304Sdan /*
2653d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming
2654d9151526Sdan ** object and the buffer is full, discard some data to free up space.
2655d9151526Sdan */
2656d9151526Sdan static void sessionDiscardData(SessionInput *pIn){
26571f48e67dSdan   if( pIn->xInput && pIn->iNext>=sessions_strm_chunk_size ){
2658d9151526Sdan     int nMove = pIn->buf.nBuf - pIn->iNext;
2659d9151526Sdan     assert( nMove>=0 );
2660d9151526Sdan     if( nMove>0 ){
2661d9151526Sdan       memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
2662d9151526Sdan     }
2663d9151526Sdan     pIn->buf.nBuf -= pIn->iNext;
2664d9151526Sdan     pIn->iNext = 0;
2665d9151526Sdan     pIn->nData = pIn->buf.nBuf;
2666d9151526Sdan   }
2667d9151526Sdan }
2668d9151526Sdan 
2669d9151526Sdan /*
2670ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2671ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2672ef7a6304Sdan ** data is in the buffer.
2673ef7a6304Sdan **
2674ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2675ef7a6304Sdan */
26764757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2677ef7a6304Sdan   int rc = SQLITE_OK;
26784757c658Sdan   if( pIn->xInput ){
26794757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
26801f48e67dSdan       int nNew = sessions_strm_chunk_size;
26814757c658Sdan 
2682d9151526Sdan       if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn);
26834757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
26844757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
26854757c658Sdan         if( nNew==0 ){
26864757c658Sdan           pIn->bEof = 1;
26874757c658Sdan         }else{
26884757c658Sdan           pIn->buf.nBuf += nNew;
26894757c658Sdan         }
26904757c658Sdan       }
26914757c658Sdan 
26924757c658Sdan       pIn->aData = pIn->buf.aBuf;
26934757c658Sdan       pIn->nData = pIn->buf.nBuf;
26944757c658Sdan     }
2695ef7a6304Sdan   }
2696ef7a6304Sdan   return rc;
2697ef7a6304Sdan }
2698ef7a6304Sdan 
2699ef7a6304Sdan /*
2700ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2701ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2702ef7a6304Sdan ** until it points to the byte immediately following that record.
2703ef7a6304Sdan */
2704ef7a6304Sdan static void sessionSkipRecord(
2705ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2706ef7a6304Sdan   int nCol                        /* Number of values in record */
2707ef7a6304Sdan ){
2708ef7a6304Sdan   u8 *aRec = *ppRec;
2709ef7a6304Sdan   int i;
2710ef7a6304Sdan   for(i=0; i<nCol; i++){
2711ef7a6304Sdan     int eType = *aRec++;
2712ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2713ef7a6304Sdan       int nByte;
2714ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2715ef7a6304Sdan       aRec += nByte;
2716ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2717ef7a6304Sdan       aRec += 8;
2718ef7a6304Sdan     }
2719ef7a6304Sdan   }
2720ef7a6304Sdan 
2721ef7a6304Sdan   *ppRec = aRec;
2722ef7a6304Sdan }
2723ef7a6304Sdan 
2724ef7a6304Sdan /*
27254757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
27264757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
27274757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
27284757c658Sdan ** error occurs.
27294757c658Sdan */
27304757c658Sdan static int sessionValueSetStr(
27314757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
27324757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
27334757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
27344757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
27354757c658Sdan ){
273616228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
273716228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
273816228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
273916228167Sdan   ** error. Hence the code to create the copy externally. */
27402d77d80aSdrh   u8 *aCopy = sqlite3_malloc64((sqlite3_int64)nData+1);
27414757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
27424757c658Sdan   memcpy(aCopy, aData, nData);
27434757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
27444757c658Sdan   return SQLITE_OK;
27454757c658Sdan }
27464757c658Sdan 
27474757c658Sdan /*
2748296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2749296c7658Sdan ** for details.
2750296c7658Sdan **
2751296c7658Sdan ** When this function is called, *paChange points to the start of the record
2752296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2753296c7658Sdan ** one byte after the end of the same record before this function returns.
2754a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2755a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2756a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2757296c7658Sdan **
2758296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2759296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2760296c7658Sdan ** from the corresponding position in the record. If that value is not
2761296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2762296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2763296c7658Sdan ** set to NULL.
2764296c7658Sdan **
2765296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2766296c7658Sdan ** using sqlite3_free().
2767296c7658Sdan **
2768296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2769296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2770296c7658Sdan */
27714fccf43aSdan static int sessionReadRecord(
2772ef7a6304Sdan   SessionInput *pIn,              /* Input data */
27734fccf43aSdan   int nCol,                       /* Number of values in record */
277473b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
27754fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
27764fccf43aSdan ){
2777296c7658Sdan   int i;                          /* Used to iterate through columns */
2778ef7a6304Sdan   int rc = SQLITE_OK;
27794fccf43aSdan 
2780ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2781ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
278273b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2783ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2784ef7a6304Sdan     if( rc==SQLITE_OK ){
2785e341ec69Sdan       if( pIn->iNext>=pIn->nData ){
2786e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
2787e341ec69Sdan       }else{
27884757c658Sdan         eType = pIn->aData[pIn->iNext++];
2789e8fa8c96Sdan         assert( apOut[i]==0 );
27904fccf43aSdan         if( eType ){
27914fccf43aSdan           apOut[i] = sqlite3ValueNew(0);
2792ef7a6304Sdan           if( !apOut[i] ) rc = SQLITE_NOMEM;
2793ef7a6304Sdan         }
2794dd8a4af8Sdan       }
2795e341ec69Sdan     }
27964fccf43aSdan 
2797ef7a6304Sdan     if( rc==SQLITE_OK ){
27984757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
27994fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
28004fccf43aSdan         int nByte;
2801ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2802ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2803e8fa8c96Sdan         if( rc==SQLITE_OK ){
2804e341ec69Sdan           if( nByte<0 || nByte>pIn->nData-pIn->iNext ){
2805e341ec69Sdan             rc = SQLITE_CORRUPT_BKPT;
2806e341ec69Sdan           }else{
28076734007dSdan             u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
28084757c658Sdan             rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
2809ef7a6304Sdan             pIn->iNext += nByte;
28104fccf43aSdan           }
2811e341ec69Sdan         }
2812e341ec69Sdan       }
28134fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2814ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
28154fccf43aSdan         if( eType==SQLITE_INTEGER ){
28164fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
28174fccf43aSdan         }else{
28184fccf43aSdan           double d;
28194e895da1Sdan           memcpy(&d, &v, 8);
28204fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
28214fccf43aSdan         }
2822ef7a6304Sdan         pIn->iNext += 8;
282391ddd559Sdan       }
28244fccf43aSdan     }
28254fccf43aSdan   }
28264fccf43aSdan 
2827ef7a6304Sdan   return rc;
2828ef7a6304Sdan }
2829ef7a6304Sdan 
2830ef7a6304Sdan /*
2831ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2832ef7a6304Sdan ** Specifically, to the following:
2833ef7a6304Sdan **
2834ef7a6304Sdan **   + number of columns in table (varint)
2835ef7a6304Sdan **   + array of PK flags (1 byte per column),
2836ef7a6304Sdan **   + table name (nul terminated).
2837ef7a6304Sdan **
2838ef7a6304Sdan ** This function ensures that all of the above is present in the input
2839ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2840ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2841ef7a6304Sdan ** The input pointer is not moved.
2842ef7a6304Sdan */
2843ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2844ef7a6304Sdan   int rc = SQLITE_OK;
2845ef7a6304Sdan   int nCol = 0;
28464757c658Sdan   int nRead = 0;
2847ef7a6304Sdan 
2848ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2849ef7a6304Sdan   if( rc==SQLITE_OK ){
28504757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
28516344eddaSdan     /* The hard upper limit for the number of columns in an SQLite
28526344eddaSdan     ** database table is, according to sqliteLimit.h, 32676. So
28536344eddaSdan     ** consider any table-header that purports to have more than 65536
28546344eddaSdan     ** columns to be corrupt. This is convenient because otherwise,
28556344eddaSdan     ** if the (nCol>65536) condition below were omitted, a sufficiently
28566344eddaSdan     ** large value for nCol may cause nRead to wrap around and become
28576344eddaSdan     ** negative. Leading to a crash. */
28586344eddaSdan     if( nCol<0 || nCol>65536 ){
2859e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2860e341ec69Sdan     }else{
28614757c658Sdan       rc = sessionInputBuffer(pIn, nRead+nCol+100);
28624757c658Sdan       nRead += nCol;
2863ef7a6304Sdan     }
2864e341ec69Sdan   }
28654757c658Sdan 
2866ef7a6304Sdan   while( rc==SQLITE_OK ){
28674757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
28684757c658Sdan       nRead++;
2869ef7a6304Sdan     }
2870e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
28714757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
28724757c658Sdan   }
2873e8fa8c96Sdan   *pnByte = nRead+1;
2874ef7a6304Sdan   return rc;
2875ef7a6304Sdan }
2876ef7a6304Sdan 
2877ef7a6304Sdan /*
2878fa122adaSdan ** The input pointer currently points to the first byte of the first field
2879fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
288016228167Sdan ** record is buffered. It does not move the input pointer.
288116228167Sdan **
288216228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
288316228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
288416228167Sdan ** final value of *pnByte is undefined in this case.
2885fa122adaSdan */
2886fa122adaSdan static int sessionChangesetBufferRecord(
288716228167Sdan   SessionInput *pIn,              /* Input data */
288816228167Sdan   int nCol,                       /* Number of columns in record */
288916228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2890fa122adaSdan ){
2891fa122adaSdan   int rc = SQLITE_OK;
2892fa122adaSdan   int nByte = 0;
2893fa122adaSdan   int i;
2894fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2895fa122adaSdan     int eType;
2896fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2897fa122adaSdan     if( rc==SQLITE_OK ){
2898fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2899fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2900fa122adaSdan         int n;
2901fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2902fa122adaSdan         nByte += n;
2903fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2904fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2905fa122adaSdan         nByte += 8;
2906fa122adaSdan       }
2907fa122adaSdan     }
2908fa122adaSdan   }
2909fa122adaSdan   *pnByte = nByte;
2910fa122adaSdan   return rc;
2911fa122adaSdan }
2912fa122adaSdan 
2913fa122adaSdan /*
2914ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2915ef7a6304Sdan ** Specifically, to the following:
2916ef7a6304Sdan **
2917ef7a6304Sdan **   + number of columns in table (varint)
2918ef7a6304Sdan **   + array of PK flags (1 byte per column),
2919ef7a6304Sdan **   + table name (nul terminated).
292016228167Sdan **
292116228167Sdan ** This function decodes the table-header and populates the p->nCol,
292216228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
292316228167Sdan ** also allocated or resized according to the new value of p->nCol. The
292416228167Sdan ** input pointer is left pointing to the byte following the table header.
292516228167Sdan **
292616228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
292716228167Sdan ** is returned and the final values of the various fields enumerated above
292816228167Sdan ** are undefined.
2929ef7a6304Sdan */
2930ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2931ef7a6304Sdan   int rc;
2932ef7a6304Sdan   int nCopy;
2933ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2934ef7a6304Sdan 
2935ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2936ef7a6304Sdan   if( rc==SQLITE_OK ){
2937ef7a6304Sdan     int nByte;
2938ef7a6304Sdan     int nVarint;
29394757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2940dd8a4af8Sdan     if( p->nCol>0 ){
2941ef7a6304Sdan       nCopy -= nVarint;
2942ef7a6304Sdan       p->in.iNext += nVarint;
2943ef7a6304Sdan       nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2944ef7a6304Sdan       p->tblhdr.nBuf = 0;
2945ef7a6304Sdan       sessionBufferGrow(&p->tblhdr, nByte, &rc);
2946dd8a4af8Sdan     }else{
2947e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2948dd8a4af8Sdan     }
2949ef7a6304Sdan   }
2950ef7a6304Sdan 
2951ef7a6304Sdan   if( rc==SQLITE_OK ){
2952f6ad201aSdrh     size_t iPK = sizeof(sqlite3_value*)*p->nCol*2;
2953ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
29544757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2955ef7a6304Sdan     p->in.iNext += nCopy;
2956ef7a6304Sdan   }
2957ef7a6304Sdan 
2958ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2959ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2960ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2961ef7a6304Sdan   return (p->rc = rc);
29624fccf43aSdan }
29634fccf43aSdan 
296477fc1d5bSdan /*
296577fc1d5bSdan ** Advance the changeset iterator to the next change.
296677fc1d5bSdan **
296777fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
296877fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
296977fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
297077fc1d5bSdan **
297177fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
297277fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
297377fc1d5bSdan ** the record to *pnRec.
297477fc1d5bSdan **
297577fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
297677fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
297777fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
297877fc1d5bSdan ** changes in the changeset.
297977fc1d5bSdan */
29805d607a6eSdan static int sessionChangesetNext(
298177fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
298277fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
2983c0a499eaSdan   int *pnRec,                     /* If non-NULL, store size of record here */
2984c0a499eaSdan   int *pbNew                      /* If non-NULL, true if new table */
29855d607a6eSdan ){
29864fccf43aSdan   int i;
2987ef7a6304Sdan   u8 op;
29884fccf43aSdan 
29895d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
29905d607a6eSdan 
2991296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
29924fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
29934fccf43aSdan 
29945d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
29954fccf43aSdan   if( p->apValue ){
29964fccf43aSdan     for(i=0; i<p->nCol*2; i++){
29974fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
29984fccf43aSdan     }
29994fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
30004fccf43aSdan   }
30014fccf43aSdan 
3002ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
3003ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
3004ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
3005ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
3006ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
3007ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
3008ef7a6304Sdan 
30094fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
30104757c658Sdan   if( p->in.iNext>=p->in.nData ){
30114fccf43aSdan     return SQLITE_DONE;
30124fccf43aSdan   }
30134fccf43aSdan 
3014d9151526Sdan   sessionDiscardData(&p->in);
3015d9151526Sdan   p->in.iCurrent = p->in.iNext;
3016d9151526Sdan 
30174757c658Sdan   op = p->in.aData[p->in.iNext++];
301807d0f15eSdan   while( op=='T' || op=='P' ){
3019c0a499eaSdan     if( pbNew ) *pbNew = 1;
3020ef7a6304Sdan     p->bPatchset = (op=='P');
3021ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
3022ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
3023d9151526Sdan     p->in.iCurrent = p->in.iNext;
302407d0f15eSdan     if( p->in.iNext>=p->in.nData ) return SQLITE_DONE;
30254757c658Sdan     op = p->in.aData[p->in.iNext++];
30265d607a6eSdan   }
30275d607a6eSdan 
302844748f27Sdan   if( p->zTab==0 || (p->bPatchset && p->bInvert) ){
3029dd8a4af8Sdan     /* The first record in the changeset is not a table header. Must be a
3030dd8a4af8Sdan     ** corrupt changeset. */
303144748f27Sdan     assert( p->in.iNext==1 || p->zTab );
3032dd8a4af8Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
3033dd8a4af8Sdan   }
3034dd8a4af8Sdan 
3035ef7a6304Sdan   p->op = op;
30364757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
30374fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
30384757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
30394fccf43aSdan   }
30404fccf43aSdan 
3041cbf6d2d2Sdan   if( paRec ){
3042cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
3043cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
3044cbf6d2d2Sdan       nVal = p->nCol * 2;
3045cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
3046cbf6d2d2Sdan       nVal = 0;
3047cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
3048cbf6d2d2Sdan     }else{
3049cbf6d2d2Sdan       nVal = p->nCol;
3050cbf6d2d2Sdan     }
3051cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
3052cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
3053cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
3054cbf6d2d2Sdan     p->in.iNext += *pnRec;
3055cbf6d2d2Sdan   }else{
305644748f27Sdan     sqlite3_value **apOld = (p->bInvert ? &p->apValue[p->nCol] : p->apValue);
305744748f27Sdan     sqlite3_value **apNew = (p->bInvert ? p->apValue : &p->apValue[p->nCol]);
30585d607a6eSdan 
30594fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
306073b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
306173b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
306244748f27Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, apOld);
30634fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30644fccf43aSdan     }
30654fccf43aSdan 
30664fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
30674fccf43aSdan     if( p->op!=SQLITE_DELETE ){
306844748f27Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, apNew);
30694fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30704fccf43aSdan     }
30714fccf43aSdan 
307244748f27Sdan     if( (p->bPatchset || p->bInvert) && p->op==SQLITE_UPDATE ){
307373b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
307473b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
307573b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
307673b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
307773b3c055Sdan       for(i=0; i<p->nCol; i++){
307844748f27Sdan         assert( p->bPatchset==0 || p->apValue[i]==0 );
307973b3c055Sdan         if( p->abPK[i] ){
308044748f27Sdan           assert( p->apValue[i]==0 );
308173b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
3082e341ec69Sdan           if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT);
308373b3c055Sdan           p->apValue[i+p->nCol] = 0;
308473b3c055Sdan         }
308573b3c055Sdan       }
308644748f27Sdan     }else if( p->bInvert ){
308744748f27Sdan       if( p->op==SQLITE_INSERT ) p->op = SQLITE_DELETE;
308844748f27Sdan       else if( p->op==SQLITE_DELETE ) p->op = SQLITE_INSERT;
308973b3c055Sdan     }
3090cbf6d2d2Sdan   }
3091ef7a6304Sdan 
30924fccf43aSdan   return SQLITE_ROW;
30934fccf43aSdan }
30944fccf43aSdan 
30954fccf43aSdan /*
30965d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
30975d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
30985d607a6eSdan ** or SQLITE_CORRUPT.
30995d607a6eSdan **
31005d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
31015d607a6eSdan ** callback by changeset_apply().
31025d607a6eSdan */
31035d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
3104c0a499eaSdan   return sessionChangesetNext(p, 0, 0, 0);
31055d607a6eSdan }
31065d607a6eSdan 
31075d607a6eSdan /*
3108244593c8Sdan ** The following function extracts information on the current change
310977fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
31104fccf43aSdan ** has returned SQLITE_ROW.
31114fccf43aSdan */
31124fccf43aSdan int sqlite3changeset_op(
3113296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
31144fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
31154fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
3116b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
3117b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
31184fccf43aSdan ){
31194fccf43aSdan   *pOp = pIter->op;
31204fccf43aSdan   *pnCol = pIter->nCol;
31214fccf43aSdan   *pzTab = pIter->zTab;
3122b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
31234fccf43aSdan   return SQLITE_OK;
31244fccf43aSdan }
31254fccf43aSdan 
312677fc1d5bSdan /*
312777fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
312877fc1d5bSdan ** the database table affected by the change that pIter currently points
312977fc1d5bSdan ** to. This function may only be called after changeset_next() returns
313077fc1d5bSdan ** SQLITE_ROW.
313177fc1d5bSdan */
3132244593c8Sdan int sqlite3changeset_pk(
3133244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
3134244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
3135244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
3136244593c8Sdan ){
3137244593c8Sdan   *pabPK = pIter->abPK;
3138244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
3139244593c8Sdan   return SQLITE_OK;
3140244593c8Sdan }
3141244593c8Sdan 
3142296c7658Sdan /*
3143296c7658Sdan ** This function may only be called while the iterator is pointing to an
3144296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
3145296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3146296c7658Sdan **
3147296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3148296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
3149296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3150296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
3151296c7658Sdan **
3152296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3153296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3154296c7658Sdan */
31554fccf43aSdan int sqlite3changeset_old(
3156296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3157296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
31584fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
31594fccf43aSdan ){
3160d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
3161d5f0767cSdan     return SQLITE_MISUSE;
3162d5f0767cSdan   }
31634fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
31644fccf43aSdan     return SQLITE_RANGE;
31654fccf43aSdan   }
31664fccf43aSdan   *ppValue = pIter->apValue[iVal];
31674fccf43aSdan   return SQLITE_OK;
31684fccf43aSdan }
31694fccf43aSdan 
3170296c7658Sdan /*
3171296c7658Sdan ** This function may only be called while the iterator is pointing to an
3172296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
3173296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3174296c7658Sdan **
3175296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3176296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
3177296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3178296c7658Sdan ** was not modified), set *ppValue to NULL.
3179296c7658Sdan **
3180296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3181296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3182296c7658Sdan */
31834fccf43aSdan int sqlite3changeset_new(
3184296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3185296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
31864fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
31874fccf43aSdan ){
3188d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
3189d5f0767cSdan     return SQLITE_MISUSE;
3190d5f0767cSdan   }
31914fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
31924fccf43aSdan     return SQLITE_RANGE;
31934fccf43aSdan   }
31944fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
31954fccf43aSdan   return SQLITE_OK;
31964fccf43aSdan }
31974fccf43aSdan 
3198296c7658Sdan /*
31997aa469cdSdan ** The following two macros are used internally. They are similar to the
32007aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
32017aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
32027aa469cdSdan */
32037aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
32047aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
32057aa469cdSdan 
32067aa469cdSdan /*
3207296c7658Sdan ** This function may only be called with a changeset iterator that has been
3208296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
3209296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
3210296c7658Sdan **
3211296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
3212296c7658Sdan ** containing the iVal'th value of the conflicting record.
3213296c7658Sdan **
3214296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
3215296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
3216296c7658Sdan */
3217d5f0767cSdan int sqlite3changeset_conflict(
3218296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3219296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
3220d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
3221d5f0767cSdan ){
3222d5f0767cSdan   if( !pIter->pConflict ){
3223d5f0767cSdan     return SQLITE_MISUSE;
3224d5f0767cSdan   }
3225ff677b20Sdan   if( iVal<0 || iVal>=pIter->nCol ){
3226d5f0767cSdan     return SQLITE_RANGE;
3227d5f0767cSdan   }
3228d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
3229d5f0767cSdan   return SQLITE_OK;
3230d5f0767cSdan }
3231d5f0767cSdan 
32324fccf43aSdan /*
3233cb3e4b79Sdan ** This function may only be called with an iterator passed to an
3234cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
3235cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
3236cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
3237cb3e4b79Sdan **
3238cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
3239cb3e4b79Sdan */
3240cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
3241cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3242cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
3243cb3e4b79Sdan ){
3244cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
3245cb3e4b79Sdan     return SQLITE_MISUSE;
3246cb3e4b79Sdan   }
3247cb3e4b79Sdan   *pnOut = pIter->nCol;
3248cb3e4b79Sdan   return SQLITE_OK;
3249cb3e4b79Sdan }
3250cb3e4b79Sdan 
3251cb3e4b79Sdan 
3252cb3e4b79Sdan /*
32534fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
32544fccf43aSdan **
32554fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
32564fccf43aSdan ** callback by changeset_apply().
32574fccf43aSdan */
32584fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
3259cbf6d2d2Sdan   int rc = SQLITE_OK;
3260cbf6d2d2Sdan   if( p ){
3261296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
3262cbf6d2d2Sdan     rc = p->rc;
326312ca0b56Sdan     if( p->apValue ){
32644fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
326512ca0b56Sdan     }
3266ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
32674757c658Sdan     sqlite3_free(p->in.buf.aBuf);
32684fccf43aSdan     sqlite3_free(p);
3269cbf6d2d2Sdan   }
32704fccf43aSdan   return rc;
32714fccf43aSdan }
32724fccf43aSdan 
3273fa122adaSdan static int sessionChangesetInvert(
3274fa122adaSdan   SessionInput *pInput,           /* Input changeset */
3275fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3276fa122adaSdan   void *pOut,
327791ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
327891ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
327991ddd559Sdan ){
3280cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
3281fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
3282cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
3283cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
3284cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
3285ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
328691ddd559Sdan 
3287fa122adaSdan   /* Initialize the output buffer */
3288fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
3289fa122adaSdan 
329091ddd559Sdan   /* Zero the output variables in case an error occurs. */
3291fa122adaSdan   if( ppInverted ){
329291ddd559Sdan     *ppInverted = 0;
329391ddd559Sdan     *pnInverted = 0;
3294fa122adaSdan   }
329591ddd559Sdan 
3296fa122adaSdan   while( 1 ){
3297ef7a6304Sdan     u8 eType;
3298fa122adaSdan 
3299fa122adaSdan     /* Test for EOF. */
3300fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
3301fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
3302fa122adaSdan     eType = pInput->aData[pInput->iNext];
3303fa122adaSdan 
330491ddd559Sdan     switch( eType ){
330591ddd559Sdan       case 'T': {
3306244593c8Sdan         /* A 'table' record consists of:
3307244593c8Sdan         **
3308244593c8Sdan         **   * A constant 'T' character,
3309244593c8Sdan         **   * Number of columns in said table (a varint),
3310ef7a6304Sdan         **   * An array of nCol bytes (sPK),
3311244593c8Sdan         **   * A nul-terminated table name.
3312244593c8Sdan         */
3313ef7a6304Sdan         int nByte;
3314fa122adaSdan         int nVar;
3315fa122adaSdan         pInput->iNext++;
3316fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
3317ef7a6304Sdan           goto finished_invert;
3318ef7a6304Sdan         }
3319fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
3320ef7a6304Sdan         sPK.nBuf = 0;
3321fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
3322fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3323fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3324ef7a6304Sdan         if( rc ) goto finished_invert;
3325fa122adaSdan 
3326fa122adaSdan         pInput->iNext += nByte;
3327cfec7eeeSdan         sqlite3_free(apVal);
3328cfec7eeeSdan         apVal = 0;
3329ef7a6304Sdan         abPK = sPK.aBuf;
333091ddd559Sdan         break;
333191ddd559Sdan       }
333291ddd559Sdan 
333391ddd559Sdan       case SQLITE_INSERT:
333491ddd559Sdan       case SQLITE_DELETE: {
333591ddd559Sdan         int nByte;
3336fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
3337fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
3338fa122adaSdan         pInput->iNext += 2;
3339fa122adaSdan         assert( rc==SQLITE_OK );
3340fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
3341fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
3342fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
3343fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3344fa122adaSdan         pInput->iNext += nByte;
3345fa122adaSdan         if( rc ) goto finished_invert;
334691ddd559Sdan         break;
334791ddd559Sdan       }
334891ddd559Sdan 
334991ddd559Sdan       case SQLITE_UPDATE: {
3350cfec7eeeSdan         int iCol;
335191ddd559Sdan 
3352cfec7eeeSdan         if( 0==apVal ){
33532d77d80aSdrh           apVal = (sqlite3_value **)sqlite3_malloc64(sizeof(apVal[0])*nCol*2);
3354cfec7eeeSdan           if( 0==apVal ){
3355cfec7eeeSdan             rc = SQLITE_NOMEM;
3356cfec7eeeSdan             goto finished_invert;
3357cfec7eeeSdan           }
3358cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3359cfec7eeeSdan         }
336091ddd559Sdan 
3361cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
3362fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3363fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
336491ddd559Sdan 
3365ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
3366fa122adaSdan         pInput->iNext += 2;
3367fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
3368ef7a6304Sdan         if( rc==SQLITE_OK ){
3369fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
3370ef7a6304Sdan         }
3371ef7a6304Sdan 
3372cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
3373cfec7eeeSdan         ** original old.* record, and the other values from the original
3374cfec7eeeSdan         ** new.* record. */
3375e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3376cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
3377fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3378cfec7eeeSdan         }
3379cfec7eeeSdan 
3380cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
3381cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
3382cfec7eeeSdan         ** are set to "undefined". */
3383e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3384cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
3385fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3386cfec7eeeSdan         }
3387cfec7eeeSdan 
3388cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
3389cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
3390cfec7eeeSdan         }
3391cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3392cfec7eeeSdan         if( rc!=SQLITE_OK ){
3393cfec7eeeSdan           goto finished_invert;
3394cfec7eeeSdan         }
3395cfec7eeeSdan 
339691ddd559Sdan         break;
339791ddd559Sdan       }
339891ddd559Sdan 
339991ddd559Sdan       default:
34004757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
3401cfec7eeeSdan         goto finished_invert;
340291ddd559Sdan     }
3403fa122adaSdan 
3404fa122adaSdan     assert( rc==SQLITE_OK );
34051f48e67dSdan     if( xOutput && sOut.nBuf>=sessions_strm_chunk_size ){
3406fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3407fa122adaSdan       sOut.nBuf = 0;
3408fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
3409fa122adaSdan     }
341091ddd559Sdan   }
341191ddd559Sdan 
3412cfec7eeeSdan   assert( rc==SQLITE_OK );
3413fa122adaSdan   if( pnInverted ){
3414fa122adaSdan     *pnInverted = sOut.nBuf;
3415fa122adaSdan     *ppInverted = sOut.aBuf;
3416fa122adaSdan     sOut.aBuf = 0;
3417fa122adaSdan   }else if( sOut.nBuf>0 ){
3418fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3419fa122adaSdan   }
3420cfec7eeeSdan 
3421cfec7eeeSdan  finished_invert:
3422fa122adaSdan   sqlite3_free(sOut.aBuf);
3423cfec7eeeSdan   sqlite3_free(apVal);
3424ef7a6304Sdan   sqlite3_free(sPK.aBuf);
3425cfec7eeeSdan   return rc;
342691ddd559Sdan }
342791ddd559Sdan 
3428fa122adaSdan 
3429fa122adaSdan /*
3430fa122adaSdan ** Invert a changeset object.
3431fa122adaSdan */
3432fa122adaSdan int sqlite3changeset_invert(
3433fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
3434fa122adaSdan   const void *pChangeset,         /* Input changeset */
3435fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
3436fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
3437fa122adaSdan ){
3438fa122adaSdan   SessionInput sInput;
3439fa122adaSdan 
3440fa122adaSdan   /* Set up the input stream */
3441fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3442fa122adaSdan   sInput.nData = nChangeset;
3443fa122adaSdan   sInput.aData = (u8*)pChangeset;
3444fa122adaSdan 
3445fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
3446fa122adaSdan }
3447fa122adaSdan 
3448fa122adaSdan /*
3449fa122adaSdan ** Streaming version of sqlite3changeset_invert().
3450fa122adaSdan */
3451f1a08ad8Sdrh int sqlite3changeset_invert_strm(
3452fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
3453fa122adaSdan   void *pIn,
3454fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3455fa122adaSdan   void *pOut
3456fa122adaSdan ){
3457fa122adaSdan   SessionInput sInput;
3458fa122adaSdan   int rc;
3459fa122adaSdan 
3460fa122adaSdan   /* Set up the input stream */
3461fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3462fa122adaSdan   sInput.xInput = xInput;
3463fa122adaSdan   sInput.pIn = pIn;
3464fa122adaSdan 
3465fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
3466fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
3467fa122adaSdan   return rc;
3468fa122adaSdan }
3469fa122adaSdan 
34700c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
34710c698471Sdan struct SessionApplyCtx {
34720c698471Sdan   sqlite3 *db;
34730c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
3474cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
34750c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
34760c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
34770c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
34780c698471Sdan   const char **azCol;             /* Array of column names */
34790c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
3480d1cccf19Sdan   int bStat1;                     /* True if table is sqlite_stat1 */
3481d9151526Sdan   int bDeferConstraints;          /* True to defer constraints */
3482*5d237bfaSdan   int bInvertConstraints;         /* Invert when iterating constraints buffer */
3483d9151526Sdan   SessionBuffer constraints;      /* Deferred constraints are stored here */
3484a38e6c57Sdan   SessionBuffer rebase;           /* Rebase information (if any) here */
3485dbe7d37aSdan   u8 bRebaseStarted;              /* If table header is already in rebase */
3486dbe7d37aSdan   u8 bRebase;                     /* True to collect rebase information */
34870c698471Sdan };
34880c698471Sdan 
3489d5f0767cSdan /*
3490d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
3491d5f0767cSdan ** structure like this:
3492d5f0767cSdan **
3493d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3494d5f0767cSdan **
3495d5f0767cSdan ** The DELETE statement looks like this:
3496d5f0767cSdan **
3497db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
3498d5f0767cSdan **
3499d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
3500d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
3501d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
3502296c7658Sdan **
3503296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
3504296c7658Sdan ** pointing to the prepared version of the SQL statement.
3505d5f0767cSdan */
3506d5f0767cSdan static int sessionDeleteRow(
3507d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3508d5f0767cSdan   const char *zTab,               /* Table name */
35090c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3510d5f0767cSdan ){
3511296c7658Sdan   int i;
3512296c7658Sdan   const char *zSep = "";
3513d5f0767cSdan   int rc = SQLITE_OK;
3514d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
35157cf7df7dSdan   int nPk = 0;
3516d5f0767cSdan 
35179e5ecdc1Sdan   sessionAppendStr(&buf, "DELETE FROM main.", &rc);
3518d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3519296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
3520296c7658Sdan 
3521296c7658Sdan   for(i=0; i<p->nCol; i++){
3522296c7658Sdan     if( p->abPK[i] ){
35237cf7df7dSdan       nPk++;
3524296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
3525296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3526296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
3527296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
3528296c7658Sdan       zSep = " AND ";
3529296c7658Sdan     }
3530296c7658Sdan   }
3531296c7658Sdan 
35327cf7df7dSdan   if( nPk<p->nCol ){
3533296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
3534296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
3535296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
3536296c7658Sdan 
3537296c7658Sdan     zSep = "";
3538296c7658Sdan     for(i=0; i<p->nCol; i++){
3539296c7658Sdan       if( !p->abPK[i] ){
3540296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
3541296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
3542296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
3543296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
3544296c7658Sdan         zSep = "AND ";
3545296c7658Sdan       }
3546296c7658Sdan     }
3547296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
35487cf7df7dSdan   }
3549d5f0767cSdan 
3550d5f0767cSdan   if( rc==SQLITE_OK ){
35510c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
3552d5f0767cSdan   }
3553d5f0767cSdan   sqlite3_free(buf.aBuf);
3554d5f0767cSdan 
3555d5f0767cSdan   return rc;
3556d5f0767cSdan }
3557d5f0767cSdan 
3558d5f0767cSdan /*
3559d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
3560d5f0767cSdan ** Assuming a table structure like this:
3561d5f0767cSdan **
3562d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3563d5f0767cSdan **
3564d5f0767cSdan ** The UPDATE statement looks like this:
3565d5f0767cSdan **
3566d5f0767cSdan **     UPDATE x SET
3567d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3568964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3569964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3570964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3571d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3572964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3573d5f0767cSdan **     )
3574d5f0767cSdan **
3575d5f0767cSdan ** For each column in the table, there are three variables to bind:
3576d5f0767cSdan **
3577d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3578d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3579d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3580d5f0767cSdan **
3581d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3582d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3583d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3584d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3585d5f0767cSdan **
3586296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3587296c7658Sdan ** pointing to the prepared version of the SQL statement.
3588d5f0767cSdan */
3589d5f0767cSdan static int sessionUpdateRow(
3590d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3591d5f0767cSdan   const char *zTab,               /* Table name */
35920c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3593d5f0767cSdan ){
3594d5f0767cSdan   int rc = SQLITE_OK;
3595d5f0767cSdan   int i;
3596d5f0767cSdan   const char *zSep = "";
3597d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3598d5f0767cSdan 
3599d5f0767cSdan   /* Append "UPDATE tbl SET " */
36009e5ecdc1Sdan   sessionAppendStr(&buf, "UPDATE main.", &rc);
3601d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3602d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3603d5f0767cSdan 
3604d5f0767cSdan   /* Append the assignments */
36050c698471Sdan   for(i=0; i<p->nCol; i++){
3606d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
36070c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3608d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3609d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3610d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3611d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3612d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
36130c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3614d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3615d5f0767cSdan     zSep = ", ";
3616d5f0767cSdan   }
3617d5f0767cSdan 
3618d5f0767cSdan   /* Append the PK part of the WHERE clause */
3619d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
36200c698471Sdan   for(i=0; i<p->nCol; i++){
36210c698471Sdan     if( p->abPK[i] ){
36220c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3623d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3624d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3625d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3626d5f0767cSdan     }
3627d5f0767cSdan   }
3628d5f0767cSdan 
3629d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3630d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
36310c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3632d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
36330c698471Sdan   for(i=0; i<p->nCol; i++){
36340c698471Sdan     if( !p->abPK[i] ){
3635d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3636d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3637d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
36380c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3639d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3640d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3641d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3642d5f0767cSdan     }
3643d5f0767cSdan   }
3644d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3645d5f0767cSdan 
3646d5f0767cSdan   if( rc==SQLITE_OK ){
36470c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3648d5f0767cSdan   }
3649d5f0767cSdan   sqlite3_free(buf.aBuf);
3650d5f0767cSdan 
3651d5f0767cSdan   return rc;
3652d5f0767cSdan }
3653d5f0767cSdan 
36543739f298Sdan 
3655296c7658Sdan /*
3656296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3657296c7658Sdan ** key. Assuming the following table structure:
3658296c7658Sdan **
3659296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3660296c7658Sdan **
3661296c7658Sdan ** The SELECT statement looks like this:
3662296c7658Sdan **
3663296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3664296c7658Sdan **
3665296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3666296c7658Sdan ** pointing to the prepared version of the SQL statement.
3667296c7658Sdan */
3668d5f0767cSdan static int sessionSelectRow(
3669d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3670d5f0767cSdan   const char *zTab,               /* Table name */
36710c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3672d5f0767cSdan ){
3673d7fb7d24Sdan   return sessionSelectStmt(
3674d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3675d5f0767cSdan }
3676d5f0767cSdan 
3677296c7658Sdan /*
3678296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3679296c7658Sdan ** For example:
3680296c7658Sdan **
3681296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3682296c7658Sdan **
3683296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3684296c7658Sdan ** pointing to the prepared version of the SQL statement.
3685296c7658Sdan */
36860c698471Sdan static int sessionInsertRow(
36870c698471Sdan   sqlite3 *db,                    /* Database handle */
36880c698471Sdan   const char *zTab,               /* Table name */
36890c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
36900c698471Sdan ){
36910c698471Sdan   int rc = SQLITE_OK;
36920c698471Sdan   int i;
36930c698471Sdan   SessionBuffer buf = {0, 0, 0};
36940c698471Sdan 
36950c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
36960c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
3697ff677b20Sdan   sessionAppendStr(&buf, "(", &rc);
3698ff677b20Sdan   for(i=0; i<p->nCol; i++){
3699ff677b20Sdan     if( i!=0 ) sessionAppendStr(&buf, ", ", &rc);
3700ff677b20Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3701ff677b20Sdan   }
3702ff677b20Sdan 
3703ff677b20Sdan   sessionAppendStr(&buf, ") VALUES(?", &rc);
37040c698471Sdan   for(i=1; i<p->nCol; i++){
37050c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
37060c698471Sdan   }
37070c698471Sdan   sessionAppendStr(&buf, ")", &rc);
37080c698471Sdan 
37090c698471Sdan   if( rc==SQLITE_OK ){
37100c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
37110c698471Sdan   }
37120c698471Sdan   sqlite3_free(buf.aBuf);
37130c698471Sdan   return rc;
37140c698471Sdan }
37150c698471Sdan 
37163739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){
37173739f298Sdan   return sqlite3_prepare_v2(db, zSql, -1, pp, 0);
37183739f298Sdan }
37193739f298Sdan 
37203739f298Sdan /*
37213739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table.
37223739f298Sdan ** These are similar to those created by sessionSelectRow(),
37233739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for
37243739f298Sdan ** other tables.
37253739f298Sdan */
37263739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){
37273739f298Sdan   int rc = sessionSelectRow(db, "sqlite_stat1", p);
37283739f298Sdan   if( rc==SQLITE_OK ){
37293739f298Sdan     rc = sessionPrepare(db, &p->pInsert,
37303739f298Sdan         "INSERT INTO main.sqlite_stat1 VALUES(?1, "
37313739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, "
37323739f298Sdan         "?3)"
37333739f298Sdan     );
37343739f298Sdan   }
37353739f298Sdan   if( rc==SQLITE_OK ){
37363739f298Sdan     rc = sessionPrepare(db, &p->pUpdate,
37373739f298Sdan         "UPDATE main.sqlite_stat1 SET "
37383739f298Sdan         "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, "
37393739f298Sdan         "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, "
37403739f298Sdan         "stat = CASE WHEN ?8 THEN ?9 ELSE stat END  "
37413739f298Sdan         "WHERE tbl=?1 AND idx IS "
37423739f298Sdan         "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END "
37433739f298Sdan         "AND (?10 OR ?8=0 OR stat IS ?7)"
37443739f298Sdan     );
37453739f298Sdan   }
37463739f298Sdan   if( rc==SQLITE_OK ){
37473739f298Sdan     rc = sessionPrepare(db, &p->pDelete,
37483739f298Sdan         "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS "
37493739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END "
37503739f298Sdan         "AND (?4 OR stat IS ?3)"
37513739f298Sdan     );
37523739f298Sdan   }
37533739f298Sdan   return rc;
37543739f298Sdan }
37553739f298Sdan 
3756296c7658Sdan /*
37577aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
37587aa469cdSdan ** See comments in the body of this function for details.
37597aa469cdSdan */
37607aa469cdSdan static int sessionBindValue(
37617aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
37627aa469cdSdan   int i,                          /* Parameter number to bind to */
37637aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
37647aa469cdSdan ){
37655671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3766082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3767082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3768082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3769082c96dfSdan   ** set to SQLITE_NULL.  */
37705671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
37717aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
37727aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3773082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
37747aa469cdSdan     return SQLITE_NOMEM;
37757aa469cdSdan   }
37767aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
37777aa469cdSdan }
37787aa469cdSdan 
37797aa469cdSdan /*
3780db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3781db04571cSdan ** transfers new.* values from the current iterator entry to statement
3782db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3783db04571cSdan **
3784d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of
3785db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3786db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3787db04571cSdan ** to an array nCol elements in size. In this case only those values for
3788db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3789db04571cSdan ** statement.
3790db04571cSdan **
3791db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3792db04571cSdan */
37937aa469cdSdan static int sessionBindRow(
3794db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
37957aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3796db04571cSdan   int nCol,                       /* Number of columns */
3797db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3798db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3799db04571cSdan ){
3800db04571cSdan   int i;
3801db04571cSdan   int rc = SQLITE_OK;
38027aa469cdSdan 
38037aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
38047aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
38057aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
38067aa469cdSdan   ** in the code below. */
38077aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
38087aa469cdSdan 
3809db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3810db04571cSdan     if( !abPK || abPK[i] ){
3811db04571cSdan       sqlite3_value *pVal;
38127aa469cdSdan       (void)xValue(pIter, i, &pVal);
3813dd8a4af8Sdan       if( pVal==0 ){
3814dd8a4af8Sdan         /* The value in the changeset was "undefined". This indicates a
3815dd8a4af8Sdan         ** corrupt changeset blob.  */
3816e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
3817dd8a4af8Sdan       }else{
38187aa469cdSdan         rc = sessionBindValue(pStmt, i+1, pVal);
3819db04571cSdan       }
3820db04571cSdan     }
3821dd8a4af8Sdan   }
3822db04571cSdan   return rc;
3823db04571cSdan }
3824db04571cSdan 
3825db04571cSdan /*
3826296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3827296c7658Sdan ** This function binds the primary key values from the change that changeset
3828296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3829296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3830296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3831296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
38327aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
38337aa469cdSdan **
38347aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
38357aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
38367aa469cdSdan ** not require a reset().
3837296c7658Sdan **
3838296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3839db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3840db04571cSdan ** UPDATE, bind values from the old.* record.
3841296c7658Sdan */
38420c698471Sdan static int sessionSeekToRow(
384337f133ecSdan   sqlite3 *db,                    /* Database handle */
384437f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
384537f133ecSdan   u8 *abPK,                       /* Primary key flags array */
38460c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
384737f133ecSdan ){
38487aa469cdSdan   int rc;                         /* Return code */
3849296c7658Sdan   int nCol;                       /* Number of columns in table */
3850296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3851296c7658Sdan   const char *zDummy;             /* Unused */
385237f133ecSdan 
3853b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
38547aa469cdSdan   rc = sessionBindRow(pIter,
3855db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3856db04571cSdan       nCol, abPK, pSelect
3857db04571cSdan   );
38580c698471Sdan 
38590c698471Sdan   if( rc==SQLITE_OK ){
38600c698471Sdan     rc = sqlite3_step(pSelect);
38610c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
38620c698471Sdan   }
38630c698471Sdan 
38640c698471Sdan   return rc;
38650c698471Sdan }
38660c698471Sdan 
3867b880a7b1Sdan /*
3868a920b209Sdrh ** This function is called from within sqlite3changeset_apply_v2() when
3869b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution
3870b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE)..
3871b880a7b1Sdan ** It adds a conflict resolution record to the buffer in
3872b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller
3873b880a7b1Sdan ** of apply_v2() as the "rebase" buffer.
3874b880a7b1Sdan **
3875b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise.
3876b880a7b1Sdan */
3877a38e6c57Sdan static int sessionRebaseAdd(
3878b880a7b1Sdan   SessionApplyCtx *p,             /* Apply context */
3879b880a7b1Sdan   int eType,                      /* Conflict resolution (OMIT or REPLACE) */
3880b880a7b1Sdan   sqlite3_changeset_iter *pIter   /* Iterator pointing at current change */
3881a38e6c57Sdan ){
3882a38e6c57Sdan   int rc = SQLITE_OK;
3883dbe7d37aSdan   if( p->bRebase ){
3884a38e6c57Sdan     int i;
3885a38e6c57Sdan     int eOp = pIter->op;
3886a38e6c57Sdan     if( p->bRebaseStarted==0 ){
3887a38e6c57Sdan       /* Append a table-header to the rebase buffer */
3888a38e6c57Sdan       const char *zTab = pIter->zTab;
3889a38e6c57Sdan       sessionAppendByte(&p->rebase, 'T', &rc);
3890a38e6c57Sdan       sessionAppendVarint(&p->rebase, p->nCol, &rc);
3891a38e6c57Sdan       sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc);
3892a38e6c57Sdan       sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc);
3893a38e6c57Sdan       p->bRebaseStarted = 1;
3894a38e6c57Sdan     }
3895a38e6c57Sdan 
3896a38e6c57Sdan     assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT );
3897a38e6c57Sdan     assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE );
3898a38e6c57Sdan 
3899f01d3a7eSdan     sessionAppendByte(&p->rebase,
3900f01d3a7eSdan         (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc
3901f01d3a7eSdan         );
3902f01d3a7eSdan     sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc);
3903a38e6c57Sdan     for(i=0; i<p->nCol; i++){
3904a38e6c57Sdan       sqlite3_value *pVal = 0;
3905f01d3a7eSdan       if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){
3906a38e6c57Sdan         sqlite3changeset_old(pIter, i, &pVal);
3907a38e6c57Sdan       }else{
3908a38e6c57Sdan         sqlite3changeset_new(pIter, i, &pVal);
3909a38e6c57Sdan       }
3910a38e6c57Sdan       sessionAppendValue(&p->rebase, pVal, &rc);
3911a38e6c57Sdan     }
3912dbe7d37aSdan   }
3913a38e6c57Sdan   return rc;
3914a38e6c57Sdan }
3915a38e6c57Sdan 
3916296c7658Sdan /*
3917296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3918296c7658Sdan ** currently points to.
3919296c7658Sdan **
3920296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3921296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3922296c7658Sdan ** depends solely on eType, as follows:
3923296c7658Sdan **
3924296c7658Sdan **    eType value                 Value passed to xConflict
3925296c7658Sdan **    -------------------------------------------------
3926296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3927296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3928296c7658Sdan **
3929296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3930296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3931296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3932296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3933296c7658Sdan ** handler invoked is as follows:
3934296c7658Sdan **
3935296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3936296c7658Sdan **    ----------------------------------------------------------------
3937296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3938296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3939296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3940296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3941296c7658Sdan **
3942296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3943296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3944296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3945296c7658Sdan **
3946296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3947296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3948296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3949296c7658Sdan ** this function returns SQLITE_OK.
3950296c7658Sdan */
39510c698471Sdan static int sessionConflictHandler(
3952296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3953296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
39540c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
39550c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3956296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3957296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
39580c698471Sdan ){
395974f598b6Smistachkin   int res = 0;                    /* Value returned by conflict handler */
39600c698471Sdan   int rc;
39610c698471Sdan   int nCol;
39620c698471Sdan   int op;
39630c698471Sdan   const char *zDummy;
39640c698471Sdan 
3965b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
39660c698471Sdan 
39670c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
39680c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
39690c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
397037f133ecSdan 
397137f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
39720c698471Sdan   if( pbReplace ){
39730c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
39740c698471Sdan   }else{
3975db04571cSdan     rc = SQLITE_OK;
39760c698471Sdan   }
39770c698471Sdan 
39780c698471Sdan   if( rc==SQLITE_ROW ){
39790c698471Sdan     /* There exists another row with the new.* primary key. */
39800c698471Sdan     pIter->pConflict = p->pSelect;
39810c698471Sdan     res = xConflict(pCtx, eType, pIter);
39820c698471Sdan     pIter->pConflict = 0;
39830c698471Sdan     rc = sqlite3_reset(p->pSelect);
3984db04571cSdan   }else if( rc==SQLITE_OK ){
3985d9151526Sdan     if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){
3986d9151526Sdan       /* Instead of invoking the conflict handler, append the change blob
3987d9151526Sdan       ** to the SessionApplyCtx.constraints buffer. */
3988d9151526Sdan       u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent];
3989d9151526Sdan       int nBlob = pIter->in.iNext - pIter->in.iCurrent;
3990d9151526Sdan       sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc);
3991a38e6c57Sdan       return SQLITE_OK;
3992d9151526Sdan     }else{
39930c698471Sdan       /* No other row with the new.* primary key. */
39940c698471Sdan       res = xConflict(pCtx, eType+1, pIter);
39950c698471Sdan       if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
399637f133ecSdan     }
3997d9151526Sdan   }
399837f133ecSdan 
399937f133ecSdan   if( rc==SQLITE_OK ){
40000c698471Sdan     switch( res ){
40010c698471Sdan       case SQLITE_CHANGESET_REPLACE:
4002f51e5f6cSdan         assert( pbReplace );
4003f51e5f6cSdan         *pbReplace = 1;
40040c698471Sdan         break;
40050c698471Sdan 
40060c698471Sdan       case SQLITE_CHANGESET_OMIT:
40070c698471Sdan         break;
40080c698471Sdan 
40090c698471Sdan       case SQLITE_CHANGESET_ABORT:
40100c698471Sdan         rc = SQLITE_ABORT;
40110c698471Sdan         break;
40120c698471Sdan 
40130c698471Sdan       default:
40140c698471Sdan         rc = SQLITE_MISUSE;
40150c698471Sdan         break;
40160c698471Sdan     }
4017a38e6c57Sdan     if( rc==SQLITE_OK ){
4018a38e6c57Sdan       rc = sessionRebaseAdd(p, res, pIter);
4019a38e6c57Sdan     }
40200c698471Sdan   }
40210c698471Sdan 
40220c698471Sdan   return rc;
40230c698471Sdan }
40240c698471Sdan 
4025296c7658Sdan /*
4026296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
4027296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
4028296c7658Sdan ** the conflict handler callback.
4029296c7658Sdan **
4030296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
4031296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
4032296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
4033296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
4034296c7658Sdan ** to true before returning. In this case the caller will invoke this function
4035296c7658Sdan ** again, this time with pbRetry set to NULL.
4036296c7658Sdan **
4037296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
4038296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
4039296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
4040296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
4041296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
4042296c7658Sdan ** row before invoking this function again, this time with pbReplace set
4043296c7658Sdan ** to NULL.
4044296c7658Sdan **
4045296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
4046296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
4047296c7658Sdan ** returned.
4048296c7658Sdan */
40490c698471Sdan static int sessionApplyOneOp(
4050296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
4051296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
40520c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
4053296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
4054296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
4055296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
40560c698471Sdan ){
40570c698471Sdan   const char *zDummy;
40580c698471Sdan   int op;
40590c698471Sdan   int nCol;
40600c698471Sdan   int rc = SQLITE_OK;
40610c698471Sdan 
40620c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
40630c698471Sdan   assert( p->azCol && p->abPK );
40640c698471Sdan   assert( !pbReplace || *pbReplace==0 );
40650c698471Sdan 
4066b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
40670c698471Sdan 
40680c698471Sdan   if( op==SQLITE_DELETE ){
40690c698471Sdan 
407073b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
407173b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
407273b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
407373b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
407473b3c055Sdan     ** is not required if:
407573b3c055Sdan     **
407673b3c055Sdan     **   * this is a patchset, or
407773b3c055Sdan     **   * (pbRetry==0), or
407873b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
407973b3c055Sdan     **     no (nCol+1) variable to bind to).
408073b3c055Sdan     */
408173b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
408273b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
40837cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
408473b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
40857cf7df7dSdan     }
40860c698471Sdan     if( rc!=SQLITE_OK ) return rc;
40870c698471Sdan 
40880c698471Sdan     sqlite3_step(p->pDelete);
40890c698471Sdan     rc = sqlite3_reset(p->pDelete);
40900c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
40910c698471Sdan       rc = sessionConflictHandler(
40920c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
40930c698471Sdan       );
409435e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
40950c698471Sdan       rc = sessionConflictHandler(
40960c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
40970c698471Sdan       );
40980c698471Sdan     }
40990c698471Sdan 
41000c698471Sdan   }else if( op==SQLITE_UPDATE ){
41010c698471Sdan     int i;
41020c698471Sdan 
41030c698471Sdan     /* Bind values to the UPDATE statement. */
41040c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
41057aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
41067aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
41077aa469cdSdan 
41080c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
41097aa469cdSdan       if( pOld ){
41107aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
41117aa469cdSdan       }
41127aa469cdSdan       if( rc==SQLITE_OK && pNew ){
41137aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
41140c698471Sdan       }
41150c698471Sdan     }
411673b3c055Sdan     if( rc==SQLITE_OK ){
411773b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
411873b3c055Sdan     }
41190c698471Sdan     if( rc!=SQLITE_OK ) return rc;
41200c698471Sdan 
41210c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
41220c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
41230c698471Sdan     sqlite3_step(p->pUpdate);
41240c698471Sdan     rc = sqlite3_reset(p->pUpdate);
41250c698471Sdan 
41260c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
41270c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
41280c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
41290c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
41300c698471Sdan 
41310c698471Sdan       rc = sessionConflictHandler(
41320c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
41330c698471Sdan       );
41340c698471Sdan 
413535e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
4136db04571cSdan       /* This is always a CONSTRAINT conflict. */
4137db04571cSdan       rc = sessionConflictHandler(
4138db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
41390c698471Sdan       );
41400c698471Sdan     }
41410c698471Sdan 
41420c698471Sdan   }else{
41430c698471Sdan     assert( op==SQLITE_INSERT );
4144d1cccf19Sdan     if( p->bStat1 ){
4145d1cccf19Sdan       /* Check if there is a conflicting row. For sqlite_stat1, this needs
4146d1cccf19Sdan       ** to be done using a SELECT, as there is no PRIMARY KEY in the
4147d1cccf19Sdan       ** database schema to throw an exception if a duplicate is inserted.  */
4148d1cccf19Sdan       rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
4149d1cccf19Sdan       if( rc==SQLITE_ROW ){
4150d1cccf19Sdan         rc = SQLITE_CONSTRAINT;
4151d1cccf19Sdan         sqlite3_reset(p->pSelect);
4152d1cccf19Sdan       }
4153d1cccf19Sdan     }
4154d1cccf19Sdan 
4155d1cccf19Sdan     if( rc==SQLITE_OK ){
41567aa469cdSdan       rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
41570c698471Sdan       if( rc!=SQLITE_OK ) return rc;
41580c698471Sdan 
41590c698471Sdan       sqlite3_step(p->pInsert);
41600c698471Sdan       rc = sqlite3_reset(p->pInsert);
4161d1cccf19Sdan     }
4162d1cccf19Sdan 
416335e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
41640c698471Sdan       rc = sessionConflictHandler(
41650c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
41660c698471Sdan       );
416737f133ecSdan     }
416837f133ecSdan   }
416937f133ecSdan 
417037f133ecSdan   return rc;
417137f133ecSdan }
417237f133ecSdan 
41735f5663dcSdan /*
41745f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument
41755f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke
41765f5663dcSdan ** the conflict handler callback.
41775f5663dcSdan **
41785f5663dcSdan ** The difference between this function and sessionApplyOne() is that this
41795f5663dcSdan ** function handles the case where the conflict-handler is invoked and
41805f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be
41815f5663dcSdan ** retried in some manner.
41825f5663dcSdan */
4183d9151526Sdan static int sessionApplyOneWithRetry(
4184d9151526Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4185d9151526Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator to read change from */
4186d9151526Sdan   SessionApplyCtx *pApply,        /* Apply context */
4187d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4188d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4189d9151526Sdan ){
4190d9151526Sdan   int bReplace = 0;
4191d9151526Sdan   int bRetry = 0;
4192d9151526Sdan   int rc;
4193d9151526Sdan 
4194d9151526Sdan   rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry);
4195a38e6c57Sdan   if( rc==SQLITE_OK ){
41965f5663dcSdan     /* If the bRetry flag is set, the change has not been applied due to an
41975f5663dcSdan     ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and
41985f5663dcSdan     ** a row with the correct PK is present in the db, but one or more other
41995f5663dcSdan     ** fields do not contain the expected values) and the conflict handler
42005f5663dcSdan     ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation,
42015f5663dcSdan     ** but pass NULL as the final argument so that sessionApplyOneOp() ignores
42025f5663dcSdan     ** the SQLITE_CHANGESET_DATA problem.  */
42035f5663dcSdan     if( bRetry ){
42045f5663dcSdan       assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE );
42055f5663dcSdan       rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4206d9151526Sdan     }
4207d9151526Sdan 
42085f5663dcSdan     /* If the bReplace flag is set, the change is an INSERT that has not
42095f5663dcSdan     ** been performed because the database already contains a row with the
42105f5663dcSdan     ** specified primary key and the conflict handler returned
42115f5663dcSdan     ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row
42125f5663dcSdan     ** before reattempting the INSERT.  */
42135f5663dcSdan     else if( bReplace ){
4214d9151526Sdan       assert( pIter->op==SQLITE_INSERT );
4215d9151526Sdan       rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
4216d9151526Sdan       if( rc==SQLITE_OK ){
4217d9151526Sdan         rc = sessionBindRow(pIter,
4218d9151526Sdan             sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete);
4219d9151526Sdan         sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1);
4220d9151526Sdan       }
4221d9151526Sdan       if( rc==SQLITE_OK ){
4222d9151526Sdan         sqlite3_step(pApply->pDelete);
4223d9151526Sdan         rc = sqlite3_reset(pApply->pDelete);
4224d9151526Sdan       }
4225d9151526Sdan       if( rc==SQLITE_OK ){
4226d9151526Sdan         rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4227d9151526Sdan       }
4228d9151526Sdan       if( rc==SQLITE_OK ){
4229d9151526Sdan         rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
4230d9151526Sdan       }
4231d9151526Sdan     }
4232a38e6c57Sdan   }
4233d9151526Sdan 
4234d9151526Sdan   return rc;
4235d9151526Sdan }
4236d9151526Sdan 
4237d9151526Sdan /*
4238d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer.
4239d9151526Sdan */
4240d9151526Sdan static int sessionRetryConstraints(
4241d9151526Sdan   sqlite3 *db,
4242d9151526Sdan   int bPatchset,
4243d9151526Sdan   const char *zTab,
4244d9151526Sdan   SessionApplyCtx *pApply,
4245d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4246d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4247d9151526Sdan ){
4248d9151526Sdan   int rc = SQLITE_OK;
4249d9151526Sdan 
4250d9151526Sdan   while( pApply->constraints.nBuf ){
4251d9151526Sdan     sqlite3_changeset_iter *pIter2 = 0;
4252d9151526Sdan     SessionBuffer cons = pApply->constraints;
4253d9151526Sdan     memset(&pApply->constraints, 0, sizeof(SessionBuffer));
4254d9151526Sdan 
4255*5d237bfaSdan     rc = sessionChangesetStart(
4256*5d237bfaSdan         &pIter2, 0, 0, cons.nBuf, cons.aBuf, pApply->bInvertConstraints
4257*5d237bfaSdan     );
4258d9151526Sdan     if( rc==SQLITE_OK ){
4259f6ad201aSdrh       size_t nByte = 2*pApply->nCol*sizeof(sqlite3_value*);
4260d9151526Sdan       int rc2;
4261d9151526Sdan       pIter2->bPatchset = bPatchset;
4262d9151526Sdan       pIter2->zTab = (char*)zTab;
4263d9151526Sdan       pIter2->nCol = pApply->nCol;
4264d9151526Sdan       pIter2->abPK = pApply->abPK;
4265d9151526Sdan       sessionBufferGrow(&pIter2->tblhdr, nByte, &rc);
4266d9151526Sdan       pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf;
4267d9151526Sdan       if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte);
4268d9151526Sdan 
4269d9151526Sdan       while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){
4270d9151526Sdan         rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx);
4271d9151526Sdan       }
4272d9151526Sdan 
4273d9151526Sdan       rc2 = sqlite3changeset_finalize(pIter2);
42747e0765a9Sdrh       if( rc==SQLITE_OK ) rc = rc2;
4275d9151526Sdan     }
4276d9151526Sdan     assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 );
4277d9151526Sdan 
4278d9151526Sdan     sqlite3_free(cons.aBuf);
4279d9151526Sdan     if( rc!=SQLITE_OK ) break;
4280d9151526Sdan     if( pApply->constraints.nBuf>=cons.nBuf ){
4281d9151526Sdan       /* No progress was made on the last round. */
4282d9151526Sdan       pApply->bDeferConstraints = 0;
4283d9151526Sdan     }
4284d9151526Sdan   }
4285d9151526Sdan 
4286d9151526Sdan   return rc;
4287d9151526Sdan }
4288d9151526Sdan 
4289296c7658Sdan /*
42904757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
42914757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
42924757c658Sdan ** changeset to the main database attached to handle "db". The supplied
42934757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
42944757c658Sdan ** while applying the change.
4295296c7658Sdan */
42964757c658Sdan static int sessionChangesetApply(
4297296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
42984757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
429940368988Sdan   int(*xFilter)(
430040368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
430140368988Sdan     const char *zTab              /* Table name */
430240368988Sdan   ),
4303d5f0767cSdan   int(*xConflict)(
4304d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
4305d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4306d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4307d5f0767cSdan   ),
4308a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4309fe55da38Sdan   void **ppRebase, int *pnRebase, /* OUT: Rebase information */
4310fe55da38Sdan   int flags                       /* SESSION_APPLY_XXX flags */
4311d5f0767cSdan ){
4312ca62ad57Sdan   int schemaMismatch = 0;
4313fe55da38Sdan   int rc = SQLITE_OK;             /* Return code */
4314d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
4315cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
4316296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
43175f5663dcSdan   int bPatchset;
4318d5f0767cSdan 
4319082c96dfSdan   assert( xConflict!=0 );
4320082c96dfSdan 
4321d9151526Sdan   pIter->in.bNoDiscard = 1;
43220c698471Sdan   memset(&sApply, 0, sizeof(sApply));
4323dbe7d37aSdan   sApply.bRebase = (ppRebase && pnRebase);
4324*5d237bfaSdan   sApply.bInvertConstraints = !!(flags & SQLITE_CHANGESETAPPLY_INVERT);
43254c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
4326fe55da38Sdan   if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){
43270c698471Sdan     rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
4328fe55da38Sdan   }
4329cb3e4b79Sdan   if( rc==SQLITE_OK ){
4330cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
4331cb3e4b79Sdan   }
43320c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
43330c698471Sdan     int nCol;
4334d5f0767cSdan     int op;
43350c698471Sdan     const char *zNew;
4336ca62ad57Sdan 
4337b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
4338d5f0767cSdan 
43390c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
4340ca62ad57Sdan       u8 *abPK;
4341ca62ad57Sdan 
4342d9151526Sdan       rc = sessionRetryConstraints(
4343d9151526Sdan           db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx
4344d9151526Sdan       );
4345d9151526Sdan       if( rc!=SQLITE_OK ) break;
4346d9151526Sdan 
4347cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
43480c698471Sdan       sqlite3_finalize(sApply.pDelete);
43490c698471Sdan       sqlite3_finalize(sApply.pUpdate);
43500c698471Sdan       sqlite3_finalize(sApply.pInsert);
43510c698471Sdan       sqlite3_finalize(sApply.pSelect);
43520c698471Sdan       sApply.db = db;
4353f1b40e83Sdan       sApply.pDelete = 0;
4354f1b40e83Sdan       sApply.pUpdate = 0;
4355f1b40e83Sdan       sApply.pInsert = 0;
4356f1b40e83Sdan       sApply.pSelect = 0;
4357f1b40e83Sdan       sApply.nCol = 0;
4358f1b40e83Sdan       sApply.azCol = 0;
4359f1b40e83Sdan       sApply.abPK = 0;
4360f1b40e83Sdan       sApply.bStat1 = 0;
4361d9151526Sdan       sApply.bDeferConstraints = 1;
4362a38e6c57Sdan       sApply.bRebaseStarted = 0;
4363f1b40e83Sdan       memset(&sApply.constraints, 0, sizeof(SessionBuffer));
436437f133ecSdan 
436540368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
436640368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
436740368988Sdan       ** non-zero, proceed. */
436840368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
436940368988Sdan       if( schemaMismatch ){
437040368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
4371f05ac112Sdan         if( zTab==0 ){
4372f05ac112Sdan           rc = SQLITE_NOMEM;
4373f05ac112Sdan           break;
4374f05ac112Sdan         }
43754f528042Sdan         nTab = (int)strlen(zTab);
437640368988Sdan         sApply.azCol = (const char **)zTab;
437740368988Sdan       }else{
4378ff677b20Sdan         int nMinCol = 0;
4379ff677b20Sdan         int i;
4380ff677b20Sdan 
4381ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
4382296c7658Sdan         rc = sessionTableInfo(
4383ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
4384ca62ad57Sdan         );
4385ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
4386ff677b20Sdan         for(i=0; i<sApply.nCol; i++){
4387ff677b20Sdan           if( sApply.abPK[i] ) nMinCol = i+1;
4388ff677b20Sdan         }
43890c698471Sdan 
4390ca62ad57Sdan         if( sApply.nCol==0 ){
4391ca62ad57Sdan           schemaMismatch = 1;
4392ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4393ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
4394ca62ad57Sdan           );
4395ca62ad57Sdan         }
4396ff677b20Sdan         else if( sApply.nCol<nCol ){
4397ca62ad57Sdan           schemaMismatch = 1;
4398ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4399ff677b20Sdan               "sqlite3changeset_apply(): table %s has %d columns, "
4400ff677b20Sdan               "expected %d or more",
4401ca62ad57Sdan               zTab, sApply.nCol, nCol
4402ca62ad57Sdan           );
4403ca62ad57Sdan         }
4404ff677b20Sdan         else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){
4405ca62ad57Sdan           schemaMismatch = 1;
440640368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
440740368988Sdan               "primary key mismatch for table %s", zTab
4408ca62ad57Sdan           );
4409ca62ad57Sdan         }
4410ff677b20Sdan         else{
4411ff677b20Sdan           sApply.nCol = nCol;
44123739f298Sdan           if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){
44133739f298Sdan             if( (rc = sessionStat1Sql(db, &sApply) ) ){
44143739f298Sdan               break;
44153739f298Sdan             }
4416d1cccf19Sdan             sApply.bStat1 = 1;
4417d1cccf19Sdan           }else{
4418ff677b20Sdan             if((rc = sessionSelectRow(db, zTab, &sApply))
44190c698471Sdan                 || (rc = sessionUpdateRow(db, zTab, &sApply))
44200c698471Sdan                 || (rc = sessionDeleteRow(db, zTab, &sApply))
44210c698471Sdan                 || (rc = sessionInsertRow(db, zTab, &sApply))
442237f133ecSdan               ){
442337f133ecSdan               break;
442437f133ecSdan             }
4425d1cccf19Sdan             sApply.bStat1 = 0;
4426d1cccf19Sdan           }
4427ff677b20Sdan         }
4428cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
4429d5f0767cSdan       }
443040368988Sdan     }
4431d5f0767cSdan 
4432ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
4433ca62ad57Sdan     ** next change. A log message has already been issued. */
4434ca62ad57Sdan     if( schemaMismatch ) continue;
4435ca62ad57Sdan 
4436d9151526Sdan     rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx);
44370c698471Sdan   }
44380c698471Sdan 
44395f5663dcSdan   bPatchset = pIter->bPatchset;
4440296c7658Sdan   if( rc==SQLITE_OK ){
4441296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
4442296c7658Sdan   }else{
4443296c7658Sdan     sqlite3changeset_finalize(pIter);
4444296c7658Sdan   }
4445d5f0767cSdan 
4446d5f0767cSdan   if( rc==SQLITE_OK ){
44475f5663dcSdan     rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx);
44485f5663dcSdan   }
44495f5663dcSdan 
44505f5663dcSdan   if( rc==SQLITE_OK ){
445107001c45Sdrh     int nFk, notUsed;
445207001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
445307001c45Sdrh     if( nFk!=0 ){
4454cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
4455cb3e4b79Sdan       sqlite3_changeset_iter sIter;
4456cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
4457cb3e4b79Sdan       sIter.nCol = nFk;
4458cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
4459cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
4460cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
4461cb3e4b79Sdan       }
4462cb3e4b79Sdan     }
4463cb3e4b79Sdan   }
4464cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
4465cb3e4b79Sdan 
4466fe55da38Sdan   if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){
4467cb3e4b79Sdan     if( rc==SQLITE_OK ){
4468d5f0767cSdan       rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4469d5f0767cSdan     }else{
4470d5f0767cSdan       sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
4471d5f0767cSdan       sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4472d5f0767cSdan     }
4473fe55da38Sdan   }
4474d5f0767cSdan 
4475dbe7d37aSdan   assert( sApply.bRebase || sApply.rebase.nBuf==0 );
4476dbe7d37aSdan   if( rc==SQLITE_OK && bPatchset==0 && sApply.bRebase ){
4477a38e6c57Sdan     *ppRebase = (void*)sApply.rebase.aBuf;
4478a38e6c57Sdan     *pnRebase = sApply.rebase.nBuf;
4479a38e6c57Sdan     sApply.rebase.aBuf = 0;
4480a38e6c57Sdan   }
44810c698471Sdan   sqlite3_finalize(sApply.pInsert);
44820c698471Sdan   sqlite3_finalize(sApply.pDelete);
44830c698471Sdan   sqlite3_finalize(sApply.pUpdate);
44840c698471Sdan   sqlite3_finalize(sApply.pSelect);
4485cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
4486d9151526Sdan   sqlite3_free((char*)sApply.constraints.aBuf);
4487a38e6c57Sdan   sqlite3_free((char*)sApply.rebase.aBuf);
44884c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
4489d5f0767cSdan   return rc;
4490d5f0767cSdan }
449191ddd559Sdan 
4492b880a7b1Sdan /*
4493b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main
4494b880a7b1Sdan ** database attached to handle "db".
4495b880a7b1Sdan */
4496a38e6c57Sdan int sqlite3changeset_apply_v2(
4497a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4498a38e6c57Sdan   int nChangeset,                 /* Size of changeset in bytes */
4499a38e6c57Sdan   void *pChangeset,               /* Changeset blob */
4500a38e6c57Sdan   int(*xFilter)(
4501a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4502a38e6c57Sdan     const char *zTab              /* Table name */
4503a38e6c57Sdan   ),
4504a38e6c57Sdan   int(*xConflict)(
4505a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4506a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4507a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4508a38e6c57Sdan   ),
4509a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4510fe55da38Sdan   void **ppRebase, int *pnRebase,
4511fe55da38Sdan   int flags
4512a38e6c57Sdan ){
4513a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
451444748f27Sdan   int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT);
451544748f27Sdan   int rc = sessionChangesetStart(&pIter, 0, 0, nChangeset, pChangeset,bInverse);
4516a38e6c57Sdan   if( rc==SQLITE_OK ){
4517a38e6c57Sdan     rc = sessionChangesetApply(
4518fe55da38Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags
4519a38e6c57Sdan     );
4520a38e6c57Sdan   }
4521a38e6c57Sdan   return rc;
4522a38e6c57Sdan }
4523a38e6c57Sdan 
452477fc1d5bSdan /*
45254757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
45264757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
45274757c658Sdan ** to resolve any conflicts encountered while applying the change.
45284757c658Sdan */
45294757c658Sdan int sqlite3changeset_apply(
45304757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
45314757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
45324757c658Sdan   void *pChangeset,               /* Changeset blob */
45334757c658Sdan   int(*xFilter)(
45344757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45354757c658Sdan     const char *zTab              /* Table name */
45364757c658Sdan   ),
45374757c658Sdan   int(*xConflict)(
45384757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
45394757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
45404757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
45414757c658Sdan   ),
45424757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
45434757c658Sdan ){
4544a38e6c57Sdan   return sqlite3changeset_apply_v2(
4545fe55da38Sdan       db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0, 0
4546a38e6c57Sdan   );
45474757c658Sdan }
45484757c658Sdan 
45494757c658Sdan /*
45504757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
45514757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
45524757c658Sdan ** to resolve any conflicts encountered while applying the change.
45534757c658Sdan */
4554a38e6c57Sdan int sqlite3changeset_apply_v2_strm(
4555a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4556a38e6c57Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
4557a38e6c57Sdan   void *pIn,                                          /* First arg for xInput */
4558a38e6c57Sdan   int(*xFilter)(
4559a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4560a38e6c57Sdan     const char *zTab              /* Table name */
4561a38e6c57Sdan   ),
4562a38e6c57Sdan   int(*xConflict)(
4563a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4564a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4565a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4566a38e6c57Sdan   ),
4567a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4568fe55da38Sdan   void **ppRebase, int *pnRebase,
4569fe55da38Sdan   int flags
4570a38e6c57Sdan ){
4571a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
457244748f27Sdan   int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT);
457344748f27Sdan   int rc = sessionChangesetStart(&pIter, xInput, pIn, 0, 0, bInverse);
4574a38e6c57Sdan   if( rc==SQLITE_OK ){
4575a38e6c57Sdan     rc = sessionChangesetApply(
4576fe55da38Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags
4577a38e6c57Sdan     );
4578a38e6c57Sdan   }
4579a38e6c57Sdan   return rc;
4580a38e6c57Sdan }
4581f1a08ad8Sdrh int sqlite3changeset_apply_strm(
45824757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
45834757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
45844757c658Sdan   void *pIn,                                          /* First arg for xInput */
45854757c658Sdan   int(*xFilter)(
45864757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45874757c658Sdan     const char *zTab              /* Table name */
45884757c658Sdan   ),
45894757c658Sdan   int(*xConflict)(
45904757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45914757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
45924757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
45934757c658Sdan   ),
45944757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
45954757c658Sdan ){
4596a38e6c57Sdan   return sqlite3changeset_apply_v2_strm(
4597fe55da38Sdan       db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0, 0
4598a38e6c57Sdan   );
45994757c658Sdan }
46004757c658Sdan 
46014757c658Sdan /*
46025898ad69Sdan ** sqlite3_changegroup handle.
46035898ad69Sdan */
46045898ad69Sdan struct sqlite3_changegroup {
46055898ad69Sdan   int rc;                         /* Error code */
46065898ad69Sdan   int bPatch;                     /* True to accumulate patchsets */
46075898ad69Sdan   SessionTable *pList;            /* List of tables in current patch */
46085898ad69Sdan };
46095898ad69Sdan 
46105898ad69Sdan /*
461177fc1d5bSdan ** This function is called to merge two changes to the same row together as
461277fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
461377fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
461477fc1d5bSdan */
46155d607a6eSdan static int sessionChangeMerge(
461677fc1d5bSdan   SessionTable *pTab,             /* Table structure */
4617c0a499eaSdan   int bRebase,                    /* True for a rebase hash-table */
461864277f4aSdan   int bPatchset,                  /* True for patchsets */
461977fc1d5bSdan   SessionChange *pExist,          /* Existing change */
462077fc1d5bSdan   int op2,                        /* Second change operation */
462177fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
462277fc1d5bSdan   u8 *aRec,                       /* Second change record */
462377fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
462477fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
46255d607a6eSdan ){
46265d607a6eSdan   SessionChange *pNew = 0;
462724a0c453Sdan   int rc = SQLITE_OK;
46285d607a6eSdan 
46295d607a6eSdan   if( !pExist ){
46302d77d80aSdrh     pNew = (SessionChange *)sqlite3_malloc64(sizeof(SessionChange) + nRec);
46315d607a6eSdan     if( !pNew ){
46325d607a6eSdan       return SQLITE_NOMEM;
46335d607a6eSdan     }
46345d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
4635798693b2Sdan     pNew->op = op2;
46365d607a6eSdan     pNew->bIndirect = bIndirect;
4637cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
463824a0c453Sdan     if( bIndirect==0 || bRebase==0 ){
463924a0c453Sdan       pNew->nRecord = nRec;
4640cbf6d2d2Sdan       memcpy(pNew->aRecord, aRec, nRec);
464124a0c453Sdan     }else{
464224a0c453Sdan       int i;
464324a0c453Sdan       u8 *pIn = aRec;
464424a0c453Sdan       u8 *pOut = pNew->aRecord;
464524a0c453Sdan       for(i=0; i<pTab->nCol; i++){
464624a0c453Sdan         int nIn = sessionSerialLen(pIn);
464724a0c453Sdan         if( *pIn==0 ){
464824a0c453Sdan           *pOut++ = 0;
464924a0c453Sdan         }else if( pTab->abPK[i]==0 ){
465024a0c453Sdan           *pOut++ = 0xFF;
465124a0c453Sdan         }else{
465224a0c453Sdan           memcpy(pOut, pIn, nIn);
465324a0c453Sdan           pOut += nIn;
465424a0c453Sdan         }
465524a0c453Sdan         pIn += nIn;
465624a0c453Sdan       }
465724a0c453Sdan       pNew->nRecord = pOut - pNew->aRecord;
465824a0c453Sdan     }
4659c0a499eaSdan   }else if( bRebase ){
466024a0c453Sdan     if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){
466124a0c453Sdan       *ppNew = pExist;
466224a0c453Sdan     }else{
46632d77d80aSdrh       sqlite3_int64 nByte = nRec + pExist->nRecord + sizeof(SessionChange);
46642d77d80aSdrh       pNew = (SessionChange*)sqlite3_malloc64(nByte);
466524a0c453Sdan       if( pNew==0 ){
466624a0c453Sdan         rc = SQLITE_NOMEM;
466724a0c453Sdan       }else{
466824a0c453Sdan         int i;
466924a0c453Sdan         u8 *a1 = pExist->aRecord;
467024a0c453Sdan         u8 *a2 = aRec;
467124a0c453Sdan         u8 *pOut;
467224a0c453Sdan 
467324a0c453Sdan         memset(pNew, 0, nByte);
467424a0c453Sdan         pNew->bIndirect = bIndirect || pExist->bIndirect;
467524a0c453Sdan         pNew->op = op2;
467624a0c453Sdan         pOut = pNew->aRecord = (u8*)&pNew[1];
467724a0c453Sdan 
467824a0c453Sdan         for(i=0; i<pTab->nCol; i++){
467924a0c453Sdan           int n1 = sessionSerialLen(a1);
468024a0c453Sdan           int n2 = sessionSerialLen(a2);
4681f231e18cSdan           if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){
468224a0c453Sdan             *pOut++ = 0xFF;
468324a0c453Sdan           }else if( *a2==0 ){
468424a0c453Sdan             memcpy(pOut, a1, n1);
468524a0c453Sdan             pOut += n1;
468624a0c453Sdan           }else{
468724a0c453Sdan             memcpy(pOut, a2, n2);
468824a0c453Sdan             pOut += n2;
468924a0c453Sdan           }
469024a0c453Sdan           a1 += n1;
469124a0c453Sdan           a2 += n2;
469224a0c453Sdan         }
469324a0c453Sdan         pNew->nRecord = pOut - pNew->aRecord;
469424a0c453Sdan       }
469524a0c453Sdan       sqlite3_free(pExist);
469624a0c453Sdan     }
46975d607a6eSdan   }else{
4698798693b2Sdan     int op1 = pExist->op;
46995d607a6eSdan 
47005d607a6eSdan     /*
47015d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
47025d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
47035d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
47045d607a6eSdan     **
47055d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
47065d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
47075d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
47085d607a6eSdan     **
47095d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
47105d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
47115d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
47125d607a6eSdan     */
47135d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
47145d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
47155d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
47165d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
47175d607a6eSdan     ){
47185d607a6eSdan       pNew = pExist;
47195d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
47205d607a6eSdan       sqlite3_free(pExist);
47215d607a6eSdan       assert( pNew==0 );
47225d607a6eSdan     }else{
472364277f4aSdan       u8 *aExist = pExist->aRecord;
47242d77d80aSdrh       sqlite3_int64 nByte;
47255d607a6eSdan       u8 *aCsr;
47265d607a6eSdan 
472764277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
472864277f4aSdan       ** buffer of the new object is large enough to hold any record that
472964277f4aSdan       ** may be generated by combining the input records.  */
47305d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
47312d77d80aSdrh       pNew = (SessionChange *)sqlite3_malloc64(nByte);
47325d607a6eSdan       if( !pNew ){
47331756ae10Sdan         sqlite3_free(pExist);
47345d607a6eSdan         return SQLITE_NOMEM;
47355d607a6eSdan       }
47365d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
47375d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
47385d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
47395d607a6eSdan 
4740b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
47415d607a6eSdan         u8 *a1 = aRec;
4742b08a1efaSdan         assert( op2==SQLITE_UPDATE );
4743798693b2Sdan         pNew->op = SQLITE_INSERT;
4744ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
474564277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
4746b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
4747b08a1efaSdan         assert( op2==SQLITE_INSERT );
4748798693b2Sdan         pNew->op = SQLITE_UPDATE;
4749fa29ecc4Sdan         if( bPatchset ){
4750fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
4751fa29ecc4Sdan           aCsr += nRec;
4752fa29ecc4Sdan         }else{
475364277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
4754b08a1efaSdan             sqlite3_free(pNew);
4755b08a1efaSdan             pNew = 0;
47565d607a6eSdan           }
4757fa29ecc4Sdan         }
4758b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
475964277f4aSdan         u8 *a1 = aExist;
47605d607a6eSdan         u8 *a2 = aRec;
4761cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
476264277f4aSdan         if( bPatchset==0 ){
4763ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
4764ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
476564277f4aSdan         }
4766798693b2Sdan         pNew->op = SQLITE_UPDATE;
476764277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
47685d607a6eSdan           sqlite3_free(pNew);
47695d607a6eSdan           pNew = 0;
47705d607a6eSdan         }
4771b08a1efaSdan       }else{                                /* UPDATE + DELETE */
4772b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
4773798693b2Sdan         pNew->op = SQLITE_DELETE;
477464277f4aSdan         if( bPatchset ){
477564277f4aSdan           memcpy(aCsr, aRec, nRec);
477664277f4aSdan           aCsr += nRec;
477764277f4aSdan         }else{
477864277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
477964277f4aSdan         }
47805d607a6eSdan       }
47815d607a6eSdan 
47825d607a6eSdan       if( pNew ){
47834f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
47845d607a6eSdan       }
47855d607a6eSdan       sqlite3_free(pExist);
47865d607a6eSdan     }
47875d607a6eSdan   }
47885d607a6eSdan 
47895d607a6eSdan   *ppNew = pNew;
479024a0c453Sdan   return rc;
47915d607a6eSdan }
47925d607a6eSdan 
479377fc1d5bSdan /*
47945898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as
47955898ad69Sdan ** the first argument to the changegroup hash tables.
479677fc1d5bSdan */
479716228167Sdan static int sessionChangesetToHash(
4798cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
4799c0a499eaSdan   sqlite3_changegroup *pGrp,       /* Changegroup object to add changeset to */
4800c0a499eaSdan   int bRebase                      /* True if hash table is for rebasing */
48015d607a6eSdan ){
48025d607a6eSdan   u8 *aRec;
48035d607a6eSdan   int nRec;
4804cbf6d2d2Sdan   int rc = SQLITE_OK;
48055d607a6eSdan   SessionTable *pTab = 0;
48065d607a6eSdan 
4807c0a499eaSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){
48085d607a6eSdan     const char *zNew;
48095d607a6eSdan     int nCol;
48105d607a6eSdan     int op;
48115d607a6eSdan     int iHash;
48125d607a6eSdan     int bIndirect;
48135d607a6eSdan     SessionChange *pChange;
48145d607a6eSdan     SessionChange *pExist = 0;
48155d607a6eSdan     SessionChange **pp;
48165d607a6eSdan 
48175898ad69Sdan     if( pGrp->pList==0 ){
48185898ad69Sdan       pGrp->bPatch = pIter->bPatchset;
48195898ad69Sdan     }else if( pIter->bPatchset!=pGrp->bPatch ){
48205898ad69Sdan       rc = SQLITE_ERROR;
48215898ad69Sdan       break;
48225898ad69Sdan     }
48235898ad69Sdan 
48245d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
4825ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
48265d607a6eSdan       /* Search the list for a matching table */
48274f528042Sdan       int nNew = (int)strlen(zNew);
4828f29123b5Sdan       u8 *abPK;
4829f29123b5Sdan 
4830f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
48315898ad69Sdan       for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){
48325d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
48335d607a6eSdan       }
48345d607a6eSdan       if( !pTab ){
48356c39e6a8Sdan         SessionTable **ppTab;
48366c39e6a8Sdan 
48372d77d80aSdrh         pTab = sqlite3_malloc64(sizeof(SessionTable) + nCol + nNew+1);
4838f29123b5Sdan         if( !pTab ){
4839f29123b5Sdan           rc = SQLITE_NOMEM;
4840f29123b5Sdan           break;
4841f29123b5Sdan         }
48425d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
4843f29123b5Sdan         pTab->nCol = nCol;
4844ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
4845ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
4846ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
4847ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
48486c39e6a8Sdan 
48496c39e6a8Sdan         /* The new object must be linked on to the end of the list, not
48506c39e6a8Sdan         ** simply added to the start of it. This is to ensure that the
48516c39e6a8Sdan         ** tables within the output of sqlite3changegroup_output() are in
48526c39e6a8Sdan         ** the right order.  */
48536c39e6a8Sdan         for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext);
48546c39e6a8Sdan         *ppTab = pTab;
4855f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
4856f29123b5Sdan         rc = SQLITE_SCHEMA;
4857f29123b5Sdan         break;
48585d607a6eSdan       }
48595d607a6eSdan     }
48605d607a6eSdan 
4861cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
48621756ae10Sdan       rc = SQLITE_NOMEM;
48631756ae10Sdan       break;
48641756ae10Sdan     }
486564277f4aSdan     iHash = sessionChangeHash(
4866cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
486764277f4aSdan     );
48685d607a6eSdan 
48695d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
48705d607a6eSdan     ** Code below may link it back in.
48715d607a6eSdan     */
48725d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
487364277f4aSdan       int bPkOnly1 = 0;
487464277f4aSdan       int bPkOnly2 = 0;
4875cbf6d2d2Sdan       if( pIter->bPatchset ){
487664277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
487764277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
487864277f4aSdan       }
487964277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
48805d607a6eSdan         pExist = *pp;
48815d607a6eSdan         *pp = (*pp)->pNext;
48825d607a6eSdan         pTab->nEntry--;
48835d607a6eSdan         break;
48845d607a6eSdan       }
48855d607a6eSdan     }
48865d607a6eSdan 
4887c0a499eaSdan     rc = sessionChangeMerge(pTab, bRebase,
4888cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
488964277f4aSdan     );
48905d607a6eSdan     if( rc ) break;
48915d607a6eSdan     if( pChange ){
48925d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
48935d607a6eSdan       pTab->apChange[iHash] = pChange;
48945d607a6eSdan       pTab->nEntry++;
48955d607a6eSdan     }
48965d607a6eSdan   }
48975d607a6eSdan 
4898cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
48995d607a6eSdan   return rc;
49005d607a6eSdan }
49015d607a6eSdan 
49025d607a6eSdan /*
49035898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets)
49045898ad69Sdan ** added to the changegroup object passed as the first argument.
49055d607a6eSdan **
49065898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the
49075898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming
49085898ad69Sdan ** interfaces.
49095d607a6eSdan **
49105898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a
49115898ad69Sdan ** buffer containing the output changeset before this function returns. In
49125898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It
49135898ad69Sdan ** is the responsibility of the caller to free the output buffer using
49145898ad69Sdan ** sqlite3_free() when it is no longer required.
49155898ad69Sdan **
49165898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite
49175898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut)
49185898ad69Sdan ** are both set to 0 before returning.
49195d607a6eSdan */
49205898ad69Sdan static int sessionChangegroupOutput(
49215898ad69Sdan   sqlite3_changegroup *pGrp,
4922cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4923cbf6d2d2Sdan   void *pOut,
4924cbf6d2d2Sdan   int *pnOut,
4925cbf6d2d2Sdan   void **ppOut
49265d607a6eSdan ){
49275898ad69Sdan   int rc = SQLITE_OK;
4928e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
49295898ad69Sdan   SessionTable *pTab;
4930cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
49315d607a6eSdan 
49325d607a6eSdan   /* Create the serialized output changeset based on the contents of the
49335898ad69Sdan   ** hash tables attached to the SessionTable objects in list p->pList.
49345d607a6eSdan   */
49355898ad69Sdan   for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
49365d607a6eSdan     int i;
49375d607a6eSdan     if( pTab->nEntry==0 ) continue;
49385d607a6eSdan 
49395898ad69Sdan     sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc);
49405d607a6eSdan     for(i=0; i<pTab->nChange; i++){
49415d607a6eSdan       SessionChange *p;
49425d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
4943798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
49445d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
49455d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
49461f48e67dSdan         if( rc==SQLITE_OK && xOutput && buf.nBuf>=sessions_strm_chunk_size ){
4947cbf6d2d2Sdan           rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4948cbf6d2d2Sdan           buf.nBuf = 0;
4949cbf6d2d2Sdan         }
49505d607a6eSdan       }
4951f8248c42Sdan     }
4952f8248c42Sdan   }
49535d607a6eSdan 
49545d607a6eSdan   if( rc==SQLITE_OK ){
4955cbf6d2d2Sdan     if( xOutput ){
4956cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4957cbf6d2d2Sdan     }else{
49585d607a6eSdan       *ppOut = buf.aBuf;
49595d607a6eSdan       *pnOut = buf.nBuf;
4960cbf6d2d2Sdan       buf.aBuf = 0;
49615d607a6eSdan     }
49625d607a6eSdan   }
4963cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
49645d607a6eSdan 
49655d607a6eSdan   return rc;
49665d607a6eSdan }
49675d607a6eSdan 
4968cbf6d2d2Sdan /*
49695898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup.
49705898ad69Sdan */
49715898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){
49725898ad69Sdan   int rc = SQLITE_OK;             /* Return code */
49735898ad69Sdan   sqlite3_changegroup *p;         /* New object */
49745898ad69Sdan   p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup));
49755898ad69Sdan   if( p==0 ){
49765898ad69Sdan     rc = SQLITE_NOMEM;
49775898ad69Sdan   }else{
49785898ad69Sdan     memset(p, 0, sizeof(sqlite3_changegroup));
49795898ad69Sdan   }
49805898ad69Sdan   *pp = p;
49815898ad69Sdan   return rc;
49825898ad69Sdan }
49835898ad69Sdan 
49845898ad69Sdan /*
49855898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes,
49865898ad69Sdan ** to changeset-group p.
49875898ad69Sdan */
49885898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){
49895898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
49905898ad69Sdan   int rc;                         /* Return code */
49915898ad69Sdan 
49925898ad69Sdan   rc = sqlite3changeset_start(&pIter, nData, pData);
49935898ad69Sdan   if( rc==SQLITE_OK ){
4994c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
49955898ad69Sdan   }
49965898ad69Sdan   sqlite3changeset_finalize(pIter);
49975898ad69Sdan   return rc;
49985898ad69Sdan }
49995898ad69Sdan 
50005898ad69Sdan /*
50015898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation
50025898ad69Sdan ** of all changesets added to the group so far.
50035898ad69Sdan */
50045898ad69Sdan int sqlite3changegroup_output(
50055898ad69Sdan     sqlite3_changegroup *pGrp,
50065898ad69Sdan     int *pnData,
50075898ad69Sdan     void **ppData
50085898ad69Sdan ){
50095898ad69Sdan   return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData);
50105898ad69Sdan }
50115898ad69Sdan 
50125898ad69Sdan /*
50135898ad69Sdan ** Streaming versions of changegroup_add().
50145898ad69Sdan */
50155898ad69Sdan int sqlite3changegroup_add_strm(
50165898ad69Sdan   sqlite3_changegroup *pGrp,
50175898ad69Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
50185898ad69Sdan   void *pIn
50195898ad69Sdan ){
50205898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
50215898ad69Sdan   int rc;                         /* Return code */
50225898ad69Sdan 
50235898ad69Sdan   rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
50245898ad69Sdan   if( rc==SQLITE_OK ){
5025c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
50265898ad69Sdan   }
50275898ad69Sdan   sqlite3changeset_finalize(pIter);
50285898ad69Sdan   return rc;
50295898ad69Sdan }
50305898ad69Sdan 
50315898ad69Sdan /*
50325898ad69Sdan ** Streaming versions of changegroup_output().
50335898ad69Sdan */
50345898ad69Sdan int sqlite3changegroup_output_strm(
50355898ad69Sdan   sqlite3_changegroup *pGrp,
50365898ad69Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
50375898ad69Sdan   void *pOut
50385898ad69Sdan ){
50395898ad69Sdan   return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0);
50405898ad69Sdan }
50415898ad69Sdan 
50425898ad69Sdan /*
50435898ad69Sdan ** Delete a changegroup object.
50445898ad69Sdan */
50455898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){
50465898ad69Sdan   if( pGrp ){
50475898ad69Sdan     sessionDeleteTable(pGrp->pList);
50485898ad69Sdan     sqlite3_free(pGrp);
50495898ad69Sdan   }
50505898ad69Sdan }
50515898ad69Sdan 
50525898ad69Sdan /*
5053cbf6d2d2Sdan ** Combine two changesets together.
5054cbf6d2d2Sdan */
5055cbf6d2d2Sdan int sqlite3changeset_concat(
5056cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
5057cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
5058cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
5059cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
5060cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
5061cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
5062cbf6d2d2Sdan ){
50635898ad69Sdan   sqlite3_changegroup *pGrp;
5064cbf6d2d2Sdan   int rc;
5065cbf6d2d2Sdan 
50665898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
5067cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50685898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nLeft, pLeft);
5069cbf6d2d2Sdan   }
5070cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50715898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nRight, pRight);
5072cbf6d2d2Sdan   }
50735898ad69Sdan   if( rc==SQLITE_OK ){
50745898ad69Sdan     rc = sqlite3changegroup_output(pGrp, pnOut, ppOut);
50755898ad69Sdan   }
50765898ad69Sdan   sqlite3changegroup_delete(pGrp);
5077cbf6d2d2Sdan 
5078cbf6d2d2Sdan   return rc;
5079cbf6d2d2Sdan }
5080cbf6d2d2Sdan 
5081cbf6d2d2Sdan /*
5082cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
5083cbf6d2d2Sdan */
5084f1a08ad8Sdrh int sqlite3changeset_concat_strm(
5085cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
5086cbf6d2d2Sdan   void *pInA,
5087cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
5088cbf6d2d2Sdan   void *pInB,
5089cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5090cbf6d2d2Sdan   void *pOut
5091cbf6d2d2Sdan ){
50925898ad69Sdan   sqlite3_changegroup *pGrp;
5093cbf6d2d2Sdan   int rc;
5094cbf6d2d2Sdan 
50955898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
5096cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50975898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA);
5098cbf6d2d2Sdan   }
5099cbf6d2d2Sdan   if( rc==SQLITE_OK ){
51005898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB);
5101cbf6d2d2Sdan   }
51025898ad69Sdan   if( rc==SQLITE_OK ){
51035898ad69Sdan     rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut);
51045898ad69Sdan   }
51055898ad69Sdan   sqlite3changegroup_delete(pGrp);
5106cbf6d2d2Sdan 
5107cbf6d2d2Sdan   return rc;
5108cbf6d2d2Sdan }
5109cbf6d2d2Sdan 
51103fa5463cSdan /*
51113fa5463cSdan ** Changeset rebaser handle.
51123fa5463cSdan */
5113c0a499eaSdan struct sqlite3_rebaser {
5114c0a499eaSdan   sqlite3_changegroup grp;        /* Hash table */
5115c0a499eaSdan };
5116c0a499eaSdan 
5117c0a499eaSdan /*
5118c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol
5119c0a499eaSdan ** fields in size. This function appends an nCol sessions module
5120b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for
5121b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[].
5122c0a499eaSdan */
5123c0a499eaSdan static void sessionAppendRecordMerge(
5124b880a7b1Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
5125b880a7b1Sdan   int nCol,                       /* Number of columns in each record */
5126b880a7b1Sdan   u8 *a1, int n1,                 /* Record 1 */
5127b880a7b1Sdan   u8 *a2, int n2,                 /* Record 2 */
5128b880a7b1Sdan   int *pRc                        /* IN/OUT: error code */
5129c0a499eaSdan ){
5130c0a499eaSdan   sessionBufferGrow(pBuf, n1+n2, pRc);
5131c0a499eaSdan   if( *pRc==SQLITE_OK ){
5132c0a499eaSdan     int i;
5133c0a499eaSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5134c0a499eaSdan     for(i=0; i<nCol; i++){
5135c0a499eaSdan       int nn1 = sessionSerialLen(a1);
5136c0a499eaSdan       int nn2 = sessionSerialLen(a2);
513724a0c453Sdan       if( *a1==0 || *a1==0xFF ){
5138c0a499eaSdan         memcpy(pOut, a2, nn2);
5139c0a499eaSdan         pOut += nn2;
5140c0a499eaSdan       }else{
5141c0a499eaSdan         memcpy(pOut, a1, nn1);
5142c0a499eaSdan         pOut += nn1;
5143c0a499eaSdan       }
5144f1b40e83Sdan       a1 += nn1;
5145f1b40e83Sdan       a2 += nn2;
5146c0a499eaSdan     }
5147f1b40e83Sdan 
5148f1b40e83Sdan     pBuf->nBuf = pOut-pBuf->aBuf;
5149f1b40e83Sdan     assert( pBuf->nBuf<=pBuf->nAlloc );
5150c0a499eaSdan   }
5151c0a499eaSdan }
5152c0a499eaSdan 
51533fa5463cSdan /*
51543fa5463cSdan ** This function is called when rebasing a local UPDATE change against one
51553fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current
51563fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single
51573fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer
51583fa5463cSdan ** pBuf.
51593fa5463cSdan **
51603fa5463cSdan ** Rebasing the UPDATE involves:
51613fa5463cSdan **
51623fa5463cSdan **   * Removing any changes to fields for which the corresponding field
51633fa5463cSdan **     in the rebase buffer is set to "replaced" (type 0xFF). If this
51643fa5463cSdan **     means the UPDATE change updates no fields, nothing is appended
51653fa5463cSdan **     to the output buffer.
51663fa5463cSdan **
51673fa5463cSdan **   * For each field modified by the local change for which the
51683fa5463cSdan **     corresponding field in the rebase buffer is not "undefined" (0x00)
51693fa5463cSdan **     or "replaced" (0xFF), the old.* value is replaced by the value
51703fa5463cSdan **     in the rebase buffer.
51713fa5463cSdan */
5172f01d3a7eSdan static void sessionAppendPartialUpdate(
517324a0c453Sdan   SessionBuffer *pBuf,            /* Append record here */
517424a0c453Sdan   sqlite3_changeset_iter *pIter,  /* Iterator pointed at local change */
517524a0c453Sdan   u8 *aRec, int nRec,             /* Local change */
517624a0c453Sdan   u8 *aChange, int nChange,       /* Record to rebase against */
517724a0c453Sdan   int *pRc                        /* IN/OUT: Return Code */
5178f01d3a7eSdan ){
5179f01d3a7eSdan   sessionBufferGrow(pBuf, 2+nRec+nChange, pRc);
5180f01d3a7eSdan   if( *pRc==SQLITE_OK ){
5181f01d3a7eSdan     int bData = 0;
5182f01d3a7eSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5183f01d3a7eSdan     int i;
5184f01d3a7eSdan     u8 *a1 = aRec;
5185f01d3a7eSdan     u8 *a2 = aChange;
5186f01d3a7eSdan 
5187f01d3a7eSdan     *pOut++ = SQLITE_UPDATE;
5188f01d3a7eSdan     *pOut++ = pIter->bIndirect;
5189f01d3a7eSdan     for(i=0; i<pIter->nCol; i++){
5190f01d3a7eSdan       int n1 = sessionSerialLen(a1);
5191f01d3a7eSdan       int n2 = sessionSerialLen(a2);
5192f01d3a7eSdan       if( pIter->abPK[i] || a2[0]==0 ){
5193f01d3a7eSdan         if( !pIter->abPK[i] ) bData = 1;
5194f01d3a7eSdan         memcpy(pOut, a1, n1);
5195f01d3a7eSdan         pOut += n1;
519624a0c453Sdan       }else if( a2[0]!=0xFF ){
519724a0c453Sdan         bData = 1;
519824a0c453Sdan         memcpy(pOut, a2, n2);
519924a0c453Sdan         pOut += n2;
5200f01d3a7eSdan       }else{
5201f01d3a7eSdan         *pOut++ = '\0';
5202f01d3a7eSdan       }
5203f01d3a7eSdan       a1 += n1;
5204f01d3a7eSdan       a2 += n2;
5205f01d3a7eSdan     }
5206f01d3a7eSdan     if( bData ){
5207f01d3a7eSdan       a2 = aChange;
5208f01d3a7eSdan       for(i=0; i<pIter->nCol; i++){
5209f01d3a7eSdan         int n1 = sessionSerialLen(a1);
5210f01d3a7eSdan         int n2 = sessionSerialLen(a2);
521124a0c453Sdan         if( pIter->abPK[i] || a2[0]!=0xFF ){
5212f01d3a7eSdan           memcpy(pOut, a1, n1);
5213f01d3a7eSdan           pOut += n1;
5214f01d3a7eSdan         }else{
5215f01d3a7eSdan           *pOut++ = '\0';
5216f01d3a7eSdan         }
5217f01d3a7eSdan         a1 += n1;
5218f01d3a7eSdan         a2 += n2;
5219f01d3a7eSdan       }
5220f01d3a7eSdan       pBuf->nBuf = (pOut - pBuf->aBuf);
5221f01d3a7eSdan     }
5222f01d3a7eSdan   }
5223f01d3a7eSdan }
5224f01d3a7eSdan 
52253fa5463cSdan /*
52263fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases
52273fa5463cSdan ** that changeset according to the current configuration of the rebaser
52283fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput
52293fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking
52303fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL,
52313fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset
52323fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of
52333fa5463cSdan ** the buffer in bytes.  It is the responsibility of the caller to eventually
52343fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free().
52353fa5463cSdan **
52363fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and
52373fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before
52383fa5463cSdan ** returning.
52393fa5463cSdan */
5240c0a499eaSdan static int sessionRebase(
5241c0a499eaSdan   sqlite3_rebaser *p,             /* Rebaser hash table */
5242c0a499eaSdan   sqlite3_changeset_iter *pIter,  /* Input data */
5243c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5244c0a499eaSdan   void *pOut,                     /* Context for xOutput callback */
5245c0a499eaSdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
5246c0a499eaSdan   void **ppOut                    /* OUT: Inverse of pChangeset */
5247c0a499eaSdan ){
5248c0a499eaSdan   int rc = SQLITE_OK;
5249c0a499eaSdan   u8 *aRec = 0;
5250c0a499eaSdan   int nRec = 0;
5251c0a499eaSdan   int bNew = 0;
5252c0a499eaSdan   SessionTable *pTab = 0;
5253c0a499eaSdan   SessionBuffer sOut = {0,0,0};
5254c0a499eaSdan 
5255f1b40e83Sdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){
5256c0a499eaSdan     SessionChange *pChange = 0;
5257f01d3a7eSdan     int bDone = 0;
5258c0a499eaSdan 
5259c0a499eaSdan     if( bNew ){
5260c0a499eaSdan       const char *zTab = pIter->zTab;
5261c0a499eaSdan       for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){
5262c0a499eaSdan         if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break;
5263c0a499eaSdan       }
5264c0a499eaSdan       bNew = 0;
5265c0a499eaSdan 
526695ccb6dcSdan       /* A patchset may not be rebased */
526795ccb6dcSdan       if( pIter->bPatchset ){
526895ccb6dcSdan         rc = SQLITE_ERROR;
526995ccb6dcSdan       }
527095ccb6dcSdan 
5271c0a499eaSdan       /* Append a table header to the output for this new table */
5272c0a499eaSdan       sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc);
5273c0a499eaSdan       sessionAppendVarint(&sOut, pIter->nCol, &rc);
5274c0a499eaSdan       sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc);
52752c42c34dSmistachkin       sessionAppendBlob(&sOut,(u8*)pIter->zTab,(int)strlen(pIter->zTab)+1,&rc);
5276c0a499eaSdan     }
5277c0a499eaSdan 
52788cb83beeSdan     if( pTab && rc==SQLITE_OK ){
52798cb83beeSdan       int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange);
5280c0a499eaSdan 
5281c0a499eaSdan       for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){
52828cb83beeSdan         if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){
5283c0a499eaSdan           break;
5284c0a499eaSdan         }
5285c0a499eaSdan       }
5286c0a499eaSdan     }
5287c0a499eaSdan 
5288c0a499eaSdan     if( pChange ){
5289c0a499eaSdan       assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT );
5290f01d3a7eSdan       switch( pIter->op ){
5291f01d3a7eSdan         case SQLITE_INSERT:
5292f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5293f01d3a7eSdan             bDone = 1;
5294f01d3a7eSdan             if( pChange->bIndirect==0 ){
5295f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_UPDATE, &rc);
5296f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5297f01d3a7eSdan               sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc);
5298f01d3a7eSdan               sessionAppendBlob(&sOut, aRec, nRec, &rc);
5299f01d3a7eSdan             }
5300f01d3a7eSdan           }
5301f01d3a7eSdan           break;
5302f01d3a7eSdan 
5303f01d3a7eSdan         case SQLITE_UPDATE:
5304f01d3a7eSdan           bDone = 1;
5305f01d3a7eSdan           if( pChange->op==SQLITE_DELETE ){
5306f01d3a7eSdan             if( pChange->bIndirect==0 ){
5307f01d3a7eSdan               u8 *pCsr = aRec;
5308f01d3a7eSdan               sessionSkipRecord(&pCsr, pIter->nCol);
5309f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_INSERT, &rc);
5310f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5311b880a7b1Sdan               sessionAppendRecordMerge(&sOut, pIter->nCol,
5312f01d3a7eSdan                   pCsr, nRec-(pCsr-aRec),
5313f01d3a7eSdan                   pChange->aRecord, pChange->nRecord, &rc
5314f01d3a7eSdan               );
5315f01d3a7eSdan             }
5316f01d3a7eSdan           }else{
5317f01d3a7eSdan             sessionAppendPartialUpdate(&sOut, pIter,
5318f01d3a7eSdan                 aRec, nRec, pChange->aRecord, pChange->nRecord, &rc
5319f01d3a7eSdan             );
5320f01d3a7eSdan           }
5321f01d3a7eSdan           break;
5322f01d3a7eSdan 
5323f01d3a7eSdan         default:
5324f01d3a7eSdan           assert( pIter->op==SQLITE_DELETE );
5325f01d3a7eSdan           bDone = 1;
5326f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5327f01d3a7eSdan             sessionAppendByte(&sOut, SQLITE_DELETE, &rc);
5328f01d3a7eSdan             sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5329b880a7b1Sdan             sessionAppendRecordMerge(&sOut, pIter->nCol,
5330f01d3a7eSdan                 pChange->aRecord, pChange->nRecord, aRec, nRec, &rc
5331f01d3a7eSdan             );
5332f01d3a7eSdan           }
5333f01d3a7eSdan           break;
5334f01d3a7eSdan       }
5335f01d3a7eSdan     }
5336f01d3a7eSdan 
5337f01d3a7eSdan     if( bDone==0 ){
5338f01d3a7eSdan       sessionAppendByte(&sOut, pIter->op, &rc);
5339f01d3a7eSdan       sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5340f01d3a7eSdan       sessionAppendBlob(&sOut, aRec, nRec, &rc);
5341f01d3a7eSdan     }
53421f48e67dSdan     if( rc==SQLITE_OK && xOutput && sOut.nBuf>sessions_strm_chunk_size ){
5343c0a499eaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5344c0a499eaSdan       sOut.nBuf = 0;
5345c0a499eaSdan     }
5346c0a499eaSdan     if( rc ) break;
5347c0a499eaSdan   }
5348c0a499eaSdan 
5349c0a499eaSdan   if( rc!=SQLITE_OK ){
5350c0a499eaSdan     sqlite3_free(sOut.aBuf);
5351c0a499eaSdan     memset(&sOut, 0, sizeof(sOut));
5352c0a499eaSdan   }
5353c0a499eaSdan 
5354c0a499eaSdan   if( rc==SQLITE_OK ){
5355c0a499eaSdan     if( xOutput ){
5356c0a499eaSdan       if( sOut.nBuf>0 ){
5357c0a499eaSdan         rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5358c0a499eaSdan       }
5359c0a499eaSdan     }else{
5360c0a499eaSdan       *ppOut = (void*)sOut.aBuf;
5361c0a499eaSdan       *pnOut = sOut.nBuf;
5362c0a499eaSdan       sOut.aBuf = 0;
5363c0a499eaSdan     }
5364c0a499eaSdan   }
5365c0a499eaSdan   sqlite3_free(sOut.aBuf);
5366c0a499eaSdan   return rc;
5367c0a499eaSdan }
5368c0a499eaSdan 
5369c0a499eaSdan /*
5370c0a499eaSdan ** Create a new rebaser object.
5371c0a499eaSdan */
5372c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){
5373c0a499eaSdan   int rc = SQLITE_OK;
5374c0a499eaSdan   sqlite3_rebaser *pNew;
5375c0a499eaSdan 
5376c0a499eaSdan   pNew = sqlite3_malloc(sizeof(sqlite3_rebaser));
5377c0a499eaSdan   if( pNew==0 ){
5378c0a499eaSdan     rc = SQLITE_NOMEM;
5379f1b40e83Sdan   }else{
5380f1b40e83Sdan     memset(pNew, 0, sizeof(sqlite3_rebaser));
5381c0a499eaSdan   }
5382c0a499eaSdan   *ppNew = pNew;
5383c0a499eaSdan   return rc;
5384c0a499eaSdan }
5385c0a499eaSdan 
5386c0a499eaSdan /*
5387c0a499eaSdan ** Call this one or more times to configure a rebaser.
5388c0a499eaSdan */
5389c0a499eaSdan int sqlite3rebaser_configure(
5390c0a499eaSdan   sqlite3_rebaser *p,
5391c0a499eaSdan   int nRebase, const void *pRebase
5392c0a499eaSdan ){
5393c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator opened on pData/nData */
5394c0a499eaSdan   int rc;                              /* Return code */
5395c0a499eaSdan   rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase);
5396c0a499eaSdan   if( rc==SQLITE_OK ){
5397c0a499eaSdan     rc = sessionChangesetToHash(pIter, &p->grp, 1);
5398c0a499eaSdan   }
5399c0a499eaSdan   sqlite3changeset_finalize(pIter);
5400c0a499eaSdan   return rc;
5401c0a499eaSdan }
5402c0a499eaSdan 
5403c0a499eaSdan /*
5404c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5405c0a499eaSdan */
5406c0a499eaSdan int sqlite3rebaser_rebase(
5407c0a499eaSdan   sqlite3_rebaser *p,
5408c0a499eaSdan   int nIn, const void *pIn,
5409c0a499eaSdan   int *pnOut, void **ppOut
5410c0a499eaSdan ){
5411c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5412c0a499eaSdan   int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn);
5413c0a499eaSdan 
5414c0a499eaSdan   if( rc==SQLITE_OK ){
5415c0a499eaSdan     rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut);
5416c0a499eaSdan     sqlite3changeset_finalize(pIter);
5417c0a499eaSdan   }
5418c0a499eaSdan 
5419c0a499eaSdan   return rc;
5420c0a499eaSdan }
5421c0a499eaSdan 
5422c0a499eaSdan /*
5423c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5424c0a499eaSdan */
5425c0a499eaSdan int sqlite3rebaser_rebase_strm(
5426c0a499eaSdan   sqlite3_rebaser *p,
5427c0a499eaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
5428c0a499eaSdan   void *pIn,
5429c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5430c0a499eaSdan   void *pOut
5431c0a499eaSdan ){
5432c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5433c0a499eaSdan   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
5434c0a499eaSdan 
5435c0a499eaSdan   if( rc==SQLITE_OK ){
5436c0a499eaSdan     rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0);
5437c0a499eaSdan     sqlite3changeset_finalize(pIter);
5438c0a499eaSdan   }
5439c0a499eaSdan 
5440c0a499eaSdan   return rc;
5441c0a499eaSdan }
5442c0a499eaSdan 
5443c0a499eaSdan /*
5444c0a499eaSdan ** Destroy a rebaser object
5445c0a499eaSdan */
5446f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){
5447c0a499eaSdan   if( p ){
5448c0a499eaSdan     sessionDeleteTable(p->grp.pList);
5449c0a499eaSdan     sqlite3_free(p);
5450c0a499eaSdan   }
5451c0a499eaSdan }
5452c0a499eaSdan 
54531f48e67dSdan /*
54541f48e67dSdan ** Global configuration
54551f48e67dSdan */
54561f48e67dSdan int sqlite3session_config(int op, void *pArg){
54571f48e67dSdan   int rc = SQLITE_OK;
54581f48e67dSdan   switch( op ){
54591f48e67dSdan     case SQLITE_SESSION_CONFIG_STRMSIZE: {
54601f48e67dSdan       int *pInt = (int*)pArg;
54611f48e67dSdan       if( *pInt>0 ){
54621f48e67dSdan         sessions_strm_chunk_size = *pInt;
54631f48e67dSdan       }
54641f48e67dSdan       *pInt = sessions_strm_chunk_size;
54651f48e67dSdan       break;
54661f48e67dSdan     }
54671f48e67dSdan     default:
54681f48e67dSdan       rc = SQLITE_MISUSE;
54691f48e67dSdan       break;
54701f48e67dSdan   }
54711f48e67dSdan   return rc;
54721f48e67dSdan }
54731f48e67dSdan 
54749b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
5475