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 ){
1627b9db9099Sdan         *pzErrMsg = sqlite3_mprintf("table schemas do not match");
1628b9db9099Sdan         rc = SQLITE_SCHEMA;
1629b9db9099Sdan       }
1630b9db9099Sdan       if( bHasPk==0 ){
1631b9db9099Sdan         /* Ignore tables with no primary keys */
1632b9db9099Sdan         goto diff_out;
1633b9db9099Sdan       }
1634cf8e9144Sdan     }
1635cf8e9144Sdan 
1636cf8e9144Sdan     if( rc==SQLITE_OK ){
1637cf8e9144Sdan       zExpr = sessionExprComparePK(pTo->nCol,
1638cf8e9144Sdan           zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK
1639cf8e9144Sdan       );
1640cf8e9144Sdan     }
1641cf8e9144Sdan 
1642cf8e9144Sdan     /* Find new rows */
1643cf8e9144Sdan     if( rc==SQLITE_OK ){
1644cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr);
1645cf8e9144Sdan     }
1646cf8e9144Sdan 
1647cf8e9144Sdan     /* Find old rows */
1648cf8e9144Sdan     if( rc==SQLITE_OK ){
1649cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr);
1650cf8e9144Sdan     }
1651cf8e9144Sdan 
1652cf8e9144Sdan     /* Find modified rows */
1653cf8e9144Sdan     if( rc==SQLITE_OK ){
1654cf8e9144Sdan       rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr);
1655cf8e9144Sdan     }
1656cf8e9144Sdan 
1657cf8e9144Sdan     sqlite3_free(zExpr);
1658cf8e9144Sdan   }
1659cf8e9144Sdan 
1660cf8e9144Sdan  diff_out:
1661cf8e9144Sdan   sessionPreupdateHooks(pSession);
166210dc553cSdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
1663cf8e9144Sdan   return rc;
1664296c7658Sdan }
16654fccf43aSdan 
16664fccf43aSdan /*
16674fccf43aSdan ** Create a session object. This session object will record changes to
16684fccf43aSdan ** database zDb attached to connection db.
16694fccf43aSdan */
16704fccf43aSdan int sqlite3session_create(
16714fccf43aSdan   sqlite3 *db,                    /* Database handle */
16724fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
16734fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
16744fccf43aSdan ){
1675296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1676296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1677cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
16784fccf43aSdan 
1679296c7658Sdan   /* Zero the output value in case an error occurs. */
16804fccf43aSdan   *ppSession = 0;
16814fccf43aSdan 
16824fccf43aSdan   /* Allocate and populate the new session object. */
16832d77d80aSdrh   pNew = (sqlite3_session *)sqlite3_malloc64(sizeof(sqlite3_session) + nDb + 1);
16844fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
16854fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
16864fccf43aSdan   pNew->db = db;
16874fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1688296c7658Sdan   pNew->bEnable = 1;
16894fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
1690cf8e9144Sdan   sessionPreupdateHooks(pNew);
16914fccf43aSdan 
16924fccf43aSdan   /* Add the new session object to the linked list of session objects
16934fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
16944fccf43aSdan   ** handle mutex.  */
16954fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
16964fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
16974fccf43aSdan   pNew->pNext = pOld;
16984fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
16994fccf43aSdan 
17004fccf43aSdan   *ppSession = pNew;
17014fccf43aSdan   return SQLITE_OK;
17024fccf43aSdan }
17034fccf43aSdan 
170477fc1d5bSdan /*
170577fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
170677fc1d5bSdan ** of the changed-rows hash tables are also deleted.
170777fc1d5bSdan */
17081ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
17095d607a6eSdan   SessionTable *pNext;
17105d607a6eSdan   SessionTable *pTab;
17115d607a6eSdan 
17125d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
17135d607a6eSdan     int i;
17145d607a6eSdan     pNext = pTab->pNext;
17155d607a6eSdan     for(i=0; i<pTab->nChange; i++){
17165d607a6eSdan       SessionChange *p;
171702d436b1Smistachkin       SessionChange *pNextChange;
171802d436b1Smistachkin       for(p=pTab->apChange[i]; p; p=pNextChange){
171902d436b1Smistachkin         pNextChange = p->pNext;
17205d607a6eSdan         sqlite3_free(p);
17215d607a6eSdan       }
17225d607a6eSdan     }
17235d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
17245d607a6eSdan     sqlite3_free(pTab->apChange);
17255d607a6eSdan     sqlite3_free(pTab);
17265d607a6eSdan   }
17275d607a6eSdan }
17285d607a6eSdan 
17294fccf43aSdan /*
17304fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
17314fccf43aSdan */
17324fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
17334fccf43aSdan   sqlite3 *db = pSession->db;
17344fccf43aSdan   sqlite3_session *pHead;
17354fccf43aSdan   sqlite3_session **pp;
17364fccf43aSdan 
1737296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1738296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
17394fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
17404fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
174150d348b1Sdrh   for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){
174250d348b1Sdrh     if( (*pp)==pSession ){
17434fccf43aSdan       *pp = (*pp)->pNext;
17444fccf43aSdan       if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead);
174550d348b1Sdrh       break;
174650d348b1Sdrh     }
174750d348b1Sdrh   }
17484fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
17491611e5a3Sdan   sqlite3ValueFree(pSession->pZeroBlob);
17504fccf43aSdan 
1751296c7658Sdan   /* Delete all attached table objects. And the contents of their
1752296c7658Sdan   ** associated hash-tables. */
17535d607a6eSdan   sessionDeleteTable(pSession->pTable);
17544fccf43aSdan 
1755296c7658Sdan   /* Free the session object itself. */
17564fccf43aSdan   sqlite3_free(pSession);
17574fccf43aSdan }
17584fccf43aSdan 
17594fccf43aSdan /*
17607531a5a3Sdan ** Set a table filter on a Session Object.
17617531a5a3Sdan */
17627531a5a3Sdan void sqlite3session_table_filter(
17637531a5a3Sdan   sqlite3_session *pSession,
17647531a5a3Sdan   int(*xFilter)(void*, const char*),
17657531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
17667531a5a3Sdan ){
17677531a5a3Sdan   pSession->bAutoAttach = 1;
17687531a5a3Sdan   pSession->pFilterCtx = pCtx;
17697531a5a3Sdan   pSession->xTableFilter = xFilter;
17707531a5a3Sdan }
17717531a5a3Sdan 
17727531a5a3Sdan /*
17734fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
17744fccf43aSdan ** while the session object is enabled will be recorded.
17754fccf43aSdan **
17764fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
17774fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
17784fccf43aSdan ** or not.
17794fccf43aSdan */
17804fccf43aSdan int sqlite3session_attach(
17814fccf43aSdan   sqlite3_session *pSession,      /* Session object */
17824fccf43aSdan   const char *zName               /* Table name */
17834fccf43aSdan ){
1784ff4d0f41Sdan   int rc = SQLITE_OK;
1785ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1786ff4d0f41Sdan 
1787ff4d0f41Sdan   if( !zName ){
1788ff4d0f41Sdan     pSession->bAutoAttach = 1;
1789ff4d0f41Sdan   }else{
1790296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1791296c7658Sdan     int nName;                    /* Number of bytes in string zName */
17924fccf43aSdan 
17934fccf43aSdan     /* First search for an existing entry. If one is found, this call is
17944fccf43aSdan     ** a no-op. Return early. */
1795cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
17964fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
17974c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
17984fccf43aSdan     }
17994fccf43aSdan 
18004c220252Sdan     if( !pTab ){
18014fccf43aSdan       /* Allocate new SessionTable object. */
18022d77d80aSdrh       pTab = (SessionTable *)sqlite3_malloc64(sizeof(SessionTable) + nName + 1);
18034c220252Sdan       if( !pTab ){
18044c220252Sdan         rc = SQLITE_NOMEM;
18054c220252Sdan       }else{
18066c39e6a8Sdan         /* Populate the new SessionTable object and link it into the list.
18076c39e6a8Sdan         ** The new object must be linked onto the end of the list, not
18086c39e6a8Sdan         ** simply added to the start of it in order to ensure that tables
18096c39e6a8Sdan         ** appear in the correct order when a changeset or patchset is
18106c39e6a8Sdan         ** eventually generated. */
18116c39e6a8Sdan         SessionTable **ppTab;
18124fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
18134fccf43aSdan         pTab->zName = (char *)&pTab[1];
18144fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
18156c39e6a8Sdan         for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext);
18166c39e6a8Sdan         *ppTab = pTab;
18174c220252Sdan       }
18184c220252Sdan     }
1819ff4d0f41Sdan   }
18204fccf43aSdan 
18214c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
18224c220252Sdan   return rc;
18234fccf43aSdan }
18244fccf43aSdan 
1825296c7658Sdan /*
1826296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1827296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1828296c7658Sdan **
1829296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1830296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1831296c7658Sdan */
1832f6ad201aSdrh static int sessionBufferGrow(SessionBuffer *p, size_t nByte, int *pRc){
1833*3196abb7Smistachkin   if( *pRc==SQLITE_OK && (size_t)(p->nAlloc-p->nBuf)<nByte ){
18344fccf43aSdan     u8 *aNew;
18359c18ef09Sdan     i64 nNew = p->nAlloc ? p->nAlloc : 128;
18364fccf43aSdan     do {
18374fccf43aSdan       nNew = nNew*2;
18389c18ef09Sdan     }while( (nNew-p->nBuf)<nByte );
18394fccf43aSdan 
18409c18ef09Sdan     aNew = (u8 *)sqlite3_realloc64(p->aBuf, nNew);
18414fccf43aSdan     if( 0==aNew ){
18424fccf43aSdan       *pRc = SQLITE_NOMEM;
184380fe2d93Sdan     }else{
18444fccf43aSdan       p->aBuf = aNew;
18454fccf43aSdan       p->nAlloc = nNew;
18464fccf43aSdan     }
184780fe2d93Sdan   }
184880fe2d93Sdan   return (*pRc!=SQLITE_OK);
18494fccf43aSdan }
18504fccf43aSdan 
1851296c7658Sdan /*
1852fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1853fa122adaSdan ** as the first.
1854fa122adaSdan **
1855fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1856fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1857fa122adaSdan ** before returning.
1858fa122adaSdan */
1859fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1860fa122adaSdan   int rc = *pRc;
1861fa122adaSdan   if( rc==SQLITE_OK ){
18622d77d80aSdrh     sqlite3_int64 nByte = 0;
1863e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1864fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1865fa122adaSdan     if( rc==SQLITE_OK ){
1866fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1867fa122adaSdan       p->nBuf += nByte;
1868fa122adaSdan     }else{
1869fa122adaSdan       *pRc = rc;
1870fa122adaSdan     }
1871fa122adaSdan   }
1872fa122adaSdan }
1873fa122adaSdan 
1874fa122adaSdan /*
1875296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1876296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1877296c7658Sdan **
1878296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1879296c7658Sdan ** returning.
1880296c7658Sdan */
18814fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
188280fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
18834fccf43aSdan     p->aBuf[p->nBuf++] = v;
18844fccf43aSdan   }
18854fccf43aSdan }
18864fccf43aSdan 
1887296c7658Sdan /*
1888296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1889296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1890296c7658Sdan **
1891296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1892296c7658Sdan ** returning.
1893296c7658Sdan */
1894cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
189580fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
18964fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
18974fccf43aSdan   }
18984fccf43aSdan }
18994fccf43aSdan 
1900296c7658Sdan /*
1901296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1902296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1903296c7658Sdan **
1904296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1905296c7658Sdan ** returning.
1906296c7658Sdan */
19074fccf43aSdan static void sessionAppendBlob(
19084fccf43aSdan   SessionBuffer *p,
19094fccf43aSdan   const u8 *aBlob,
19104fccf43aSdan   int nBlob,
19114fccf43aSdan   int *pRc
19124fccf43aSdan ){
1913895decf6Sdan   if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){
19144fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
19154fccf43aSdan     p->nBuf += nBlob;
19164fccf43aSdan   }
19174fccf43aSdan }
19184fccf43aSdan 
1919296c7658Sdan /*
1920296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1921296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1922296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1923296c7658Sdan **
1924296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1925296c7658Sdan ** returning.
1926296c7658Sdan */
1927d5f0767cSdan static void sessionAppendStr(
1928d5f0767cSdan   SessionBuffer *p,
1929d5f0767cSdan   const char *zStr,
1930d5f0767cSdan   int *pRc
1931d5f0767cSdan ){
1932cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
193380fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1934d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1935d5f0767cSdan     p->nBuf += nStr;
1936d5f0767cSdan   }
1937d5f0767cSdan }
1938d5f0767cSdan 
1939296c7658Sdan /*
1940296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1941296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1942296c7658Sdan ** to the buffer. No nul-terminator is written.
1943296c7658Sdan **
1944296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1945296c7658Sdan ** returning.
1946296c7658Sdan */
1947d5f0767cSdan static void sessionAppendInteger(
1948296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1949296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1950296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1951d5f0767cSdan ){
1952d5f0767cSdan   char aBuf[24];
1953d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1954d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1955d5f0767cSdan }
1956d5f0767cSdan 
1957296c7658Sdan /*
1958296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1959296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1960296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1961296c7658Sdan ** nul-terminator byte is written.
1962296c7658Sdan **
1963296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1964296c7658Sdan ** returning.
1965296c7658Sdan */
1966d5f0767cSdan static void sessionAppendIdent(
1967296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1968296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1969296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1970d5f0767cSdan ){
1971cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
197280fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1973d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1974d5f0767cSdan     const char *zIn = zStr;
1975d5f0767cSdan     *zOut++ = '"';
1976d5f0767cSdan     while( *zIn ){
1977d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1978d5f0767cSdan       *zOut++ = *(zIn++);
1979d5f0767cSdan     }
1980d5f0767cSdan     *zOut++ = '"';
1981cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1982d5f0767cSdan   }
1983d5f0767cSdan }
1984d5f0767cSdan 
1985296c7658Sdan /*
1986296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1987296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1988296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1989296c7658Sdan ** to to the buffer.
1990296c7658Sdan */
19914fccf43aSdan static void sessionAppendCol(
1992296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1993296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1994296c7658Sdan   int iCol,                       /* Column to read value from */
1995296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
19964fccf43aSdan ){
19974fccf43aSdan   if( *pRc==SQLITE_OK ){
19984fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
19994fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
20004fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
20014fccf43aSdan       sqlite3_int64 i;
20024fccf43aSdan       u8 aBuf[8];
20034fccf43aSdan       if( eType==SQLITE_INTEGER ){
20044fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
20054fccf43aSdan       }else{
20064fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
20074fccf43aSdan         memcpy(&i, &r, 8);
20084fccf43aSdan       }
2009296c7658Sdan       sessionPutI64(aBuf, i);
20104fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
20114fccf43aSdan     }
20124fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
20136734007dSdan       u8 *z;
20143cc89d95Sdan       int nByte;
20156734007dSdan       if( eType==SQLITE_BLOB ){
20166734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
20176734007dSdan       }else{
20186734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
20196734007dSdan       }
20203cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
20213cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
20224fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
20236734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
20246734007dSdan       }else{
20256734007dSdan         *pRc = SQLITE_NOMEM;
20266734007dSdan       }
20274fccf43aSdan     }
20284fccf43aSdan   }
20294fccf43aSdan }
20304fccf43aSdan 
2031296c7658Sdan /*
2032296c7658Sdan **
203380fe2d93Sdan ** This function appends an update change to the buffer (see the comments
203480fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
203580fe2d93Sdan ** consists of:
2036296c7658Sdan **
2037296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
2038296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
2039296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
2040296c7658Sdan **
2041296c7658Sdan ** The SessionChange object passed as the third argument contains the
2042296c7658Sdan ** values that were stored in the row when the session began (the old.*
2043296c7658Sdan ** values). The statement handle passed as the second argument points
2044296c7658Sdan ** at the current version of the row (the new.* values).
2045296c7658Sdan **
2046296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
2047296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
2048296c7658Sdan **
2049296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
2050296c7658Sdan ** original values of any fields that have been modified. The new.* record
2051296c7658Sdan ** contains the new values of only those fields that have been modified.
2052296c7658Sdan */
205380fe2d93Sdan static int sessionAppendUpdate(
2054296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
205573b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
2056296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
2057296c7658Sdan   SessionChange *p,               /* Object containing old values */
205880fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
20594fccf43aSdan ){
206080fe2d93Sdan   int rc = SQLITE_OK;
2061296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
2062296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
20631f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
2064296c7658Sdan   int i;                        /* Used to iterate through columns */
2065296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
2066296c7658Sdan 
206780fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
206880fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
20694fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
207037f133ecSdan     int bChanged = 0;
20714fccf43aSdan     int nAdvance;
20724fccf43aSdan     int eType = *pCsr;
20734fccf43aSdan     switch( eType ){
20744fccf43aSdan       case SQLITE_NULL:
20754fccf43aSdan         nAdvance = 1;
20764fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
207737f133ecSdan           bChanged = 1;
20784fccf43aSdan         }
20794fccf43aSdan         break;
20804fccf43aSdan 
20814fccf43aSdan       case SQLITE_FLOAT:
20824fccf43aSdan       case SQLITE_INTEGER: {
20834fccf43aSdan         nAdvance = 9;
20844fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
20854fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
20864fccf43aSdan           if( eType==SQLITE_INTEGER ){
20874fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
20884fccf43aSdan           }else{
20894fccf43aSdan             double dVal;
20904fccf43aSdan             memcpy(&dVal, &iVal, 8);
20914fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
20924fccf43aSdan           }
20934fccf43aSdan         }
209437f133ecSdan         bChanged = 1;
20954fccf43aSdan         break;
20964fccf43aSdan       }
20974fccf43aSdan 
2098e5754eecSdan       default: {
2099895decf6Sdan         int n;
2100895decf6Sdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &n);
2101e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
2102895decf6Sdan         nAdvance = nHdr + n;
21034fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
2104895decf6Sdan          && n==sqlite3_column_bytes(pStmt, i)
2105895decf6Sdan          && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n))
21064fccf43aSdan         ){
21074fccf43aSdan           break;
21084fccf43aSdan         }
210937f133ecSdan         bChanged = 1;
21104fccf43aSdan       }
21114fccf43aSdan     }
21124fccf43aSdan 
211373b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
211473b3c055Sdan     if( bChanged ) bNoop = 0;
211573b3c055Sdan 
211673b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
211773b3c055Sdan     ** currently generating a patchset. */
211873b3c055Sdan     if( bPatchset==0 ){
211937f133ecSdan       if( bChanged || abPK[i] ){
212080fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
21214fccf43aSdan       }else{
212280fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
212337f133ecSdan       }
212473b3c055Sdan     }
212537f133ecSdan 
212673b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
212773b3c055Sdan     ** generating a patchset.  */
212873b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
212980fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
213037f133ecSdan     }else{
213180fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
21324fccf43aSdan     }
213337f133ecSdan 
21344fccf43aSdan     pCsr += nAdvance;
21354fccf43aSdan   }
21364fccf43aSdan 
21374fccf43aSdan   if( bNoop ){
21381f34f8ccSdan     pBuf->nBuf = nRewind;
21394fccf43aSdan   }else{
214080fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
21414fccf43aSdan   }
21421f34f8ccSdan   sqlite3_free(buf2.aBuf);
214380fe2d93Sdan 
214480fe2d93Sdan   return rc;
2145d5f0767cSdan }
21464fccf43aSdan 
2147a71d2371Sdan /*
2148a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
2149a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
2150a71d2371Sdan ** format otherwise.
2151a71d2371Sdan */
215273b3c055Sdan static int sessionAppendDelete(
215373b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
215473b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
215573b3c055Sdan   SessionChange *p,               /* Object containing old values */
2156a71d2371Sdan   int nCol,                       /* Number of columns in table */
215773b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
215873b3c055Sdan ){
215973b3c055Sdan   int rc = SQLITE_OK;
216073b3c055Sdan 
216173b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
216273b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
216373b3c055Sdan 
216473b3c055Sdan   if( bPatchset==0 ){
216573b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
216673b3c055Sdan   }else{
216773b3c055Sdan     int i;
216873b3c055Sdan     u8 *a = p->aRecord;
216973b3c055Sdan     for(i=0; i<nCol; i++){
217073b3c055Sdan       u8 *pStart = a;
217173b3c055Sdan       int eType = *a++;
217273b3c055Sdan 
217373b3c055Sdan       switch( eType ){
217473b3c055Sdan         case 0:
217573b3c055Sdan         case SQLITE_NULL:
217673b3c055Sdan           assert( abPK[i]==0 );
217773b3c055Sdan           break;
217873b3c055Sdan 
217973b3c055Sdan         case SQLITE_FLOAT:
218073b3c055Sdan         case SQLITE_INTEGER:
218173b3c055Sdan           a += 8;
218273b3c055Sdan           break;
218373b3c055Sdan 
218473b3c055Sdan         default: {
218573b3c055Sdan           int n;
218673b3c055Sdan           a += sessionVarintGet(a, &n);
218773b3c055Sdan           a += n;
218873b3c055Sdan           break;
218973b3c055Sdan         }
219073b3c055Sdan       }
219173b3c055Sdan       if( abPK[i] ){
2192f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
219373b3c055Sdan       }
219473b3c055Sdan     }
219573b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
219673b3c055Sdan   }
219773b3c055Sdan 
219873b3c055Sdan   return rc;
219973b3c055Sdan }
220073b3c055Sdan 
220177fc1d5bSdan /*
220277fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
220377fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
220477fc1d5bSdan **
220577fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
220677fc1d5bSdan */
2207e8d5648eSdan static int sessionSelectStmt(
2208e8d5648eSdan   sqlite3 *db,                    /* Database handle */
2209d7fb7d24Sdan   const char *zDb,                /* Database name */
2210e8d5648eSdan   const char *zTab,               /* Table name */
221177fc1d5bSdan   int nCol,                       /* Number of columns in table */
221277fc1d5bSdan   const char **azCol,             /* Names of table columns */
221377fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
221477fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
2215d5f0767cSdan ){
2216e8d5648eSdan   int rc = SQLITE_OK;
22173739f298Sdan   char *zSql = 0;
22183739f298Sdan   int nSql = -1;
22193739f298Sdan 
22203739f298Sdan   if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){
22213739f298Sdan     zSql = sqlite3_mprintf(
22223739f298Sdan         "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND "
22233739f298Sdan         "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb
22243739f298Sdan     );
2225bd45374cSdan     if( zSql==0 ) rc = SQLITE_NOMEM;
22263739f298Sdan   }else{
2227d5f0767cSdan     int i;
2228e8d5648eSdan     const char *zSep = "";
2229e8d5648eSdan     SessionBuffer buf = {0, 0, 0};
2230d5f0767cSdan 
2231e8d5648eSdan     sessionAppendStr(&buf, "SELECT * FROM ", &rc);
2232d7fb7d24Sdan     sessionAppendIdent(&buf, zDb, &rc);
2233d7fb7d24Sdan     sessionAppendStr(&buf, ".", &rc);
2234e8d5648eSdan     sessionAppendIdent(&buf, zTab, &rc);
2235e8d5648eSdan     sessionAppendStr(&buf, " WHERE ", &rc);
2236e8d5648eSdan     for(i=0; i<nCol; i++){
2237e8d5648eSdan       if( abPK[i] ){
2238e8d5648eSdan         sessionAppendStr(&buf, zSep, &rc);
2239e8d5648eSdan         sessionAppendIdent(&buf, azCol[i], &rc);
22403739f298Sdan         sessionAppendStr(&buf, " IS ?", &rc);
2241e8d5648eSdan         sessionAppendInteger(&buf, i+1, &rc);
2242e8d5648eSdan         zSep = " AND ";
2243d5f0767cSdan       }
2244d5f0767cSdan     }
22453739f298Sdan     zSql = (char*)buf.aBuf;
22463739f298Sdan     nSql = buf.nBuf;
2247d5f0767cSdan   }
22483739f298Sdan 
22493739f298Sdan   if( rc==SQLITE_OK ){
22503739f298Sdan     rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0);
22513739f298Sdan   }
22523739f298Sdan   sqlite3_free(zSql);
2253e8d5648eSdan   return rc;
2254d5f0767cSdan }
2255d5f0767cSdan 
225677fc1d5bSdan /*
225777fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
225877fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
225977fc1d5bSdan ** is as prepared by function sessionSelectStmt().
226077fc1d5bSdan **
226177fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
226277fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
226377fc1d5bSdan */
2264e8d5648eSdan static int sessionSelectBind(
226577fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
226677fc1d5bSdan   int nCol,                       /* Number of columns in table */
226777fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
226877fc1d5bSdan   SessionChange *pChange          /* Change structure */
2269e8d5648eSdan ){
2270e8d5648eSdan   int i;
2271e8d5648eSdan   int rc = SQLITE_OK;
2272e5754eecSdan   u8 *a = pChange->aRecord;
2273d5f0767cSdan 
2274e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2275e8d5648eSdan     int eType = *a++;
2276e8d5648eSdan 
2277e8d5648eSdan     switch( eType ){
227880fe2d93Sdan       case 0:
2279e8d5648eSdan       case SQLITE_NULL:
22801611e5a3Sdan         assert( abPK[i]==0 );
2281e8d5648eSdan         break;
2282e8d5648eSdan 
2283e8d5648eSdan       case SQLITE_INTEGER: {
2284e8d5648eSdan         if( abPK[i] ){
2285e8d5648eSdan           i64 iVal = sessionGetI64(a);
2286e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
2287e8d5648eSdan         }
2288e8d5648eSdan         a += 8;
2289e8d5648eSdan         break;
2290d5f0767cSdan       }
2291296c7658Sdan 
2292e8d5648eSdan       case SQLITE_FLOAT: {
2293e8d5648eSdan         if( abPK[i] ){
2294e8d5648eSdan           double rVal;
2295e8d5648eSdan           i64 iVal = sessionGetI64(a);
2296e8d5648eSdan           memcpy(&rVal, &iVal, 8);
22974e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
2298d5f0767cSdan         }
2299e8d5648eSdan         a += 8;
2300e8d5648eSdan         break;
2301e8d5648eSdan       }
2302e8d5648eSdan 
2303e8d5648eSdan       case SQLITE_TEXT: {
2304e8d5648eSdan         int n;
2305e8d5648eSdan         a += sessionVarintGet(a, &n);
2306e8d5648eSdan         if( abPK[i] ){
2307e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
2308e8d5648eSdan         }
2309e8d5648eSdan         a += n;
2310e8d5648eSdan         break;
2311e8d5648eSdan       }
2312e8d5648eSdan 
2313e5754eecSdan       default: {
2314e8d5648eSdan         int n;
2315e5754eecSdan         assert( eType==SQLITE_BLOB );
2316e8d5648eSdan         a += sessionVarintGet(a, &n);
2317e8d5648eSdan         if( abPK[i] ){
2318e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
2319e8d5648eSdan         }
2320e8d5648eSdan         a += n;
2321e8d5648eSdan         break;
2322e8d5648eSdan       }
2323e8d5648eSdan     }
2324e8d5648eSdan   }
2325e8d5648eSdan 
2326d5f0767cSdan   return rc;
23274fccf43aSdan }
23284fccf43aSdan 
232977fc1d5bSdan /*
233077fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
233177fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
233277fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
233377fc1d5bSdan ** SQLite error code before returning.
233477fc1d5bSdan */
23355d607a6eSdan static void sessionAppendTableHdr(
2336a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
2337a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
2338a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
2339a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
23405d607a6eSdan ){
23415d607a6eSdan   /* Write a table header */
234273b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
23435d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
23445d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
23454f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
23465d607a6eSdan }
23475d607a6eSdan 
2348a71d2371Sdan /*
2349a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
2350a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
2351a71d2371Sdan ** passed as the first argument.
2352a71d2371Sdan **
2353a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
2354a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
2355a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
2356a71d2371Sdan ** to 0.
2357a71d2371Sdan */
2358adf3bf58Sdrh static int sessionGenerateChangeset(
23594fccf43aSdan   sqlite3_session *pSession,      /* Session object */
236073b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
2361ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2362ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
23634fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
23644fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
23654fccf43aSdan ){
2366296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
2367296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
2368296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
2369296c7658Sdan   int rc;                         /* Return code */
23704fccf43aSdan 
2371ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
2372ef7a6304Sdan 
2373296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
2374296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
2375296c7658Sdan   ** this call will be a no-op.  */
2376ef7a6304Sdan   if( xOutput==0 ){
23774fccf43aSdan     *pnChangeset = 0;
23784fccf43aSdan     *ppChangeset = 0;
2379ef7a6304Sdan   }
2380e5754eecSdan 
2381e5754eecSdan   if( pSession->rc ) return pSession->rc;
2382e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
2383e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
2384e5754eecSdan 
2385e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
23864fccf43aSdan 
23874fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
23884fccf43aSdan     if( pTab->nEntry ){
2389d7fb7d24Sdan       const char *zName = pTab->zName;
2390a9605b91Sdan       int nCol;                   /* Number of columns in table */
2391a9605b91Sdan       u8 *abPK;                   /* Primary key array */
2392a9605b91Sdan       const char **azCol = 0;     /* Table columns */
23931f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
23941f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
23951f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
23961f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
23974fccf43aSdan 
2398a9605b91Sdan       /* Check the table schema is still Ok. */
2399a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
2400a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
2401a9605b91Sdan         rc = SQLITE_SCHEMA;
2402a9605b91Sdan       }
2403a9605b91Sdan 
24044fccf43aSdan       /* Write a table header */
240573b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
24064fccf43aSdan 
24074fccf43aSdan       /* Build and compile a statement to execute: */
24084fccf43aSdan       if( rc==SQLITE_OK ){
2409d7fb7d24Sdan         rc = sessionSelectStmt(
2410a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
24114fccf43aSdan       }
24124fccf43aSdan 
24131f34f8ccSdan       nNoop = buf.nBuf;
241412ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
2415e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
2416e8d5648eSdan 
24174fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
2418e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
241980fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
24201f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
2421798693b2Sdan             if( p->op==SQLITE_INSERT ){
24224fccf43aSdan               int iCol;
24234fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
2424b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
2425e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
24261f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
24274fccf43aSdan               }
2428e8d5648eSdan             }else{
242973b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
24304fccf43aSdan             }
2431798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
2432a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
24334fccf43aSdan           }
243412ca0b56Sdan           if( rc==SQLITE_OK ){
24351f34f8ccSdan             rc = sqlite3_reset(pSel);
24364fccf43aSdan           }
2437ef7a6304Sdan 
24381f48e67dSdan           /* If the buffer is now larger than sessions_strm_chunk_size, pass
2439ef7a6304Sdan           ** its contents to the xOutput() callback. */
2440ef7a6304Sdan           if( xOutput
2441ef7a6304Sdan            && rc==SQLITE_OK
2442ef7a6304Sdan            && buf.nBuf>nNoop
24431f48e67dSdan            && buf.nBuf>sessions_strm_chunk_size
2444ef7a6304Sdan           ){
2445ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2446ef7a6304Sdan             nNoop = -1;
2447ef7a6304Sdan             buf.nBuf = 0;
2448ef7a6304Sdan           }
2449ef7a6304Sdan 
24504fccf43aSdan         }
2451e8d5648eSdan       }
24524fccf43aSdan 
24531f34f8ccSdan       sqlite3_finalize(pSel);
24541f34f8ccSdan       if( buf.nBuf==nNoop ){
24554fccf43aSdan         buf.nBuf = nRewind;
24564fccf43aSdan       }
2457cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
24584fccf43aSdan     }
24594fccf43aSdan   }
24604fccf43aSdan 
24614fccf43aSdan   if( rc==SQLITE_OK ){
2462ef7a6304Sdan     if( xOutput==0 ){
24634fccf43aSdan       *pnChangeset = buf.nBuf;
24644fccf43aSdan       *ppChangeset = buf.aBuf;
2465ef7a6304Sdan       buf.aBuf = 0;
2466ef7a6304Sdan     }else if( buf.nBuf>0 ){
2467ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2468ef7a6304Sdan     }
24694fccf43aSdan   }
24704c220252Sdan 
2471ef7a6304Sdan   sqlite3_free(buf.aBuf);
2472e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
24734c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
24744fccf43aSdan   return rc;
24754fccf43aSdan }
24764fccf43aSdan 
2477296c7658Sdan /*
247873b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
247973b3c055Sdan ** session object passed as the first argument.
248073b3c055Sdan **
248173b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
248273b3c055Sdan ** using sqlite3_free().
248373b3c055Sdan */
248473b3c055Sdan int sqlite3session_changeset(
248573b3c055Sdan   sqlite3_session *pSession,      /* Session object */
248673b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
248773b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
248873b3c055Sdan ){
2489ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2490ef7a6304Sdan }
2491ef7a6304Sdan 
2492ef7a6304Sdan /*
2493ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2494ef7a6304Sdan */
2495f1a08ad8Sdrh int sqlite3session_changeset_strm(
2496ef7a6304Sdan   sqlite3_session *pSession,
2497ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2498ef7a6304Sdan   void *pOut
2499ef7a6304Sdan ){
2500ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2501ef7a6304Sdan }
2502ef7a6304Sdan 
2503ef7a6304Sdan /*
2504ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2505ef7a6304Sdan */
2506f1a08ad8Sdrh int sqlite3session_patchset_strm(
2507ef7a6304Sdan   sqlite3_session *pSession,
2508ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2509ef7a6304Sdan   void *pOut
2510ef7a6304Sdan ){
2511ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
251273b3c055Sdan }
251373b3c055Sdan 
251473b3c055Sdan /*
251573b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
251673b3c055Sdan ** session object passed as the first argument.
251773b3c055Sdan **
251873b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
251973b3c055Sdan ** using sqlite3_free().
252073b3c055Sdan */
252173b3c055Sdan int sqlite3session_patchset(
252273b3c055Sdan   sqlite3_session *pSession,      /* Session object */
252373b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
252473b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
252573b3c055Sdan ){
2526ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
252773b3c055Sdan }
252873b3c055Sdan 
252973b3c055Sdan /*
2530296c7658Sdan ** Enable or disable the session object passed as the first argument.
2531296c7658Sdan */
25324fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
25334c220252Sdan   int ret;
25344c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2535296c7658Sdan   if( bEnable>=0 ){
2536296c7658Sdan     pSession->bEnable = bEnable;
25374fccf43aSdan   }
25384c220252Sdan   ret = pSession->bEnable;
25394c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
25404c220252Sdan   return ret;
2541296c7658Sdan }
25424fccf43aSdan 
25434fccf43aSdan /*
2544b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2545b4480e94Sdan */
2546b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2547b4480e94Sdan   int ret;
2548b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2549b4480e94Sdan   if( bIndirect>=0 ){
2550b4480e94Sdan     pSession->bIndirect = bIndirect;
2551b4480e94Sdan   }
2552b4480e94Sdan   ret = pSession->bIndirect;
2553b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2554b4480e94Sdan   return ret;
2555b4480e94Sdan }
2556b4480e94Sdan 
2557b4480e94Sdan /*
2558b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2559b69ec348Sdan ** by the session object passed as the only argument.
2560b69ec348Sdan */
2561b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2562b69ec348Sdan   int ret = 0;
2563b69ec348Sdan   SessionTable *pTab;
2564b69ec348Sdan 
2565b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2566b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2567b69ec348Sdan     ret = (pTab->nEntry>0);
2568b69ec348Sdan   }
2569b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2570b69ec348Sdan 
2571ff530326Sdan   return (ret==0);
2572b69ec348Sdan }
2573b69ec348Sdan 
2574b69ec348Sdan /*
2575f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
25764fccf43aSdan */
2577adf3bf58Sdrh static int sessionChangesetStart(
2578296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2579ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2580ef7a6304Sdan   void *pIn,
2581296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
258244748f27Sdan   void *pChangeset,               /* Pointer to buffer containing changeset */
258344748f27Sdan   int bInvert                     /* True to invert changeset */
25844fccf43aSdan ){
25854fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
25864fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
25874fccf43aSdan 
2588ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2589ef7a6304Sdan 
2590296c7658Sdan   /* Zero the output variable in case an error occurs. */
2591296c7658Sdan   *pp = 0;
25924fccf43aSdan 
2593296c7658Sdan   /* Allocate and initialize the iterator structure. */
25944fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
25954fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
25964fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
25974fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
25984757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
25994757c658Sdan   pRet->in.nData = nChangeset;
2600ef7a6304Sdan   pRet->in.xInput = xInput;
2601ef7a6304Sdan   pRet->in.pIn = pIn;
2602ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
260344748f27Sdan   pRet->bInvert = bInvert;
26044fccf43aSdan 
2605296c7658Sdan   /* Populate the output variable and return success. */
2606296c7658Sdan   *pp = pRet;
26074fccf43aSdan   return SQLITE_OK;
26084fccf43aSdan }
26094fccf43aSdan 
2610296c7658Sdan /*
2611ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2612ef7a6304Sdan */
2613ef7a6304Sdan int sqlite3changeset_start(
2614ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2615ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2616ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2617ef7a6304Sdan ){
261844748f27Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, 0);
2619ef7a6304Sdan }
262046de0728Sdan int sqlite3changeset_start_v2(
262146de0728Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
262246de0728Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
262346de0728Sdan   void *pChangeset,               /* Pointer to buffer containing changeset */
262446de0728Sdan   int flags
262546de0728Sdan ){
262646de0728Sdan   int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT);
262746de0728Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset, bInvert);
262846de0728Sdan }
2629ef7a6304Sdan 
2630ef7a6304Sdan /*
2631ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2632ef7a6304Sdan */
2633f1a08ad8Sdrh int sqlite3changeset_start_strm(
2634ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2635ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2636ef7a6304Sdan   void *pIn
2637ef7a6304Sdan ){
263844748f27Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0, 0);
2639ef7a6304Sdan }
264046de0728Sdan int sqlite3changeset_start_v2_strm(
264146de0728Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
264246de0728Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
264346de0728Sdan   void *pIn,
264446de0728Sdan   int flags
264546de0728Sdan ){
264646de0728Sdan   int bInvert = !!(flags & SQLITE_CHANGESETSTART_INVERT);
264746de0728Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0, bInvert);
264846de0728Sdan }
2649ef7a6304Sdan 
2650ef7a6304Sdan /*
2651d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming
2652d9151526Sdan ** object and the buffer is full, discard some data to free up space.
2653d9151526Sdan */
2654d9151526Sdan static void sessionDiscardData(SessionInput *pIn){
26551f48e67dSdan   if( pIn->xInput && pIn->iNext>=sessions_strm_chunk_size ){
2656d9151526Sdan     int nMove = pIn->buf.nBuf - pIn->iNext;
2657d9151526Sdan     assert( nMove>=0 );
2658d9151526Sdan     if( nMove>0 ){
2659d9151526Sdan       memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
2660d9151526Sdan     }
2661d9151526Sdan     pIn->buf.nBuf -= pIn->iNext;
2662d9151526Sdan     pIn->iNext = 0;
2663d9151526Sdan     pIn->nData = pIn->buf.nBuf;
2664d9151526Sdan   }
2665d9151526Sdan }
2666d9151526Sdan 
2667d9151526Sdan /*
2668ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2669ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2670ef7a6304Sdan ** data is in the buffer.
2671ef7a6304Sdan **
2672ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2673ef7a6304Sdan */
26744757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2675ef7a6304Sdan   int rc = SQLITE_OK;
26764757c658Sdan   if( pIn->xInput ){
26774757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
26781f48e67dSdan       int nNew = sessions_strm_chunk_size;
26794757c658Sdan 
2680d9151526Sdan       if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn);
26814757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
26824757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
26834757c658Sdan         if( nNew==0 ){
26844757c658Sdan           pIn->bEof = 1;
26854757c658Sdan         }else{
26864757c658Sdan           pIn->buf.nBuf += nNew;
26874757c658Sdan         }
26884757c658Sdan       }
26894757c658Sdan 
26904757c658Sdan       pIn->aData = pIn->buf.aBuf;
26914757c658Sdan       pIn->nData = pIn->buf.nBuf;
26924757c658Sdan     }
2693ef7a6304Sdan   }
2694ef7a6304Sdan   return rc;
2695ef7a6304Sdan }
2696ef7a6304Sdan 
2697ef7a6304Sdan /*
2698ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2699ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2700ef7a6304Sdan ** until it points to the byte immediately following that record.
2701ef7a6304Sdan */
2702ef7a6304Sdan static void sessionSkipRecord(
2703ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2704ef7a6304Sdan   int nCol                        /* Number of values in record */
2705ef7a6304Sdan ){
2706ef7a6304Sdan   u8 *aRec = *ppRec;
2707ef7a6304Sdan   int i;
2708ef7a6304Sdan   for(i=0; i<nCol; i++){
2709ef7a6304Sdan     int eType = *aRec++;
2710ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2711ef7a6304Sdan       int nByte;
2712ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2713ef7a6304Sdan       aRec += nByte;
2714ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2715ef7a6304Sdan       aRec += 8;
2716ef7a6304Sdan     }
2717ef7a6304Sdan   }
2718ef7a6304Sdan 
2719ef7a6304Sdan   *ppRec = aRec;
2720ef7a6304Sdan }
2721ef7a6304Sdan 
2722ef7a6304Sdan /*
27234757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
27244757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
27254757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
27264757c658Sdan ** error occurs.
27274757c658Sdan */
27284757c658Sdan static int sessionValueSetStr(
27294757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
27304757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
27314757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
27324757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
27334757c658Sdan ){
273416228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
273516228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
273616228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
273716228167Sdan   ** error. Hence the code to create the copy externally. */
27382d77d80aSdrh   u8 *aCopy = sqlite3_malloc64((sqlite3_int64)nData+1);
27394757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
27404757c658Sdan   memcpy(aCopy, aData, nData);
27414757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
27424757c658Sdan   return SQLITE_OK;
27434757c658Sdan }
27444757c658Sdan 
27454757c658Sdan /*
2746296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2747296c7658Sdan ** for details.
2748296c7658Sdan **
2749296c7658Sdan ** When this function is called, *paChange points to the start of the record
2750296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2751296c7658Sdan ** one byte after the end of the same record before this function returns.
2752a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2753a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2754a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2755296c7658Sdan **
2756296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2757296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2758296c7658Sdan ** from the corresponding position in the record. If that value is not
2759296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2760296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2761296c7658Sdan ** set to NULL.
2762296c7658Sdan **
2763296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2764296c7658Sdan ** using sqlite3_free().
2765296c7658Sdan **
2766296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2767296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2768296c7658Sdan */
27694fccf43aSdan static int sessionReadRecord(
2770ef7a6304Sdan   SessionInput *pIn,              /* Input data */
27714fccf43aSdan   int nCol,                       /* Number of values in record */
277273b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
27734fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
27744fccf43aSdan ){
2775296c7658Sdan   int i;                          /* Used to iterate through columns */
2776ef7a6304Sdan   int rc = SQLITE_OK;
27774fccf43aSdan 
2778ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2779ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
278073b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2781ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2782ef7a6304Sdan     if( rc==SQLITE_OK ){
2783e341ec69Sdan       if( pIn->iNext>=pIn->nData ){
2784e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
2785e341ec69Sdan       }else{
27864757c658Sdan         eType = pIn->aData[pIn->iNext++];
2787e8fa8c96Sdan         assert( apOut[i]==0 );
27884fccf43aSdan         if( eType ){
27894fccf43aSdan           apOut[i] = sqlite3ValueNew(0);
2790ef7a6304Sdan           if( !apOut[i] ) rc = SQLITE_NOMEM;
2791ef7a6304Sdan         }
2792dd8a4af8Sdan       }
2793e341ec69Sdan     }
27944fccf43aSdan 
2795ef7a6304Sdan     if( rc==SQLITE_OK ){
27964757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
27974fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
27984fccf43aSdan         int nByte;
2799ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2800ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2801e8fa8c96Sdan         if( rc==SQLITE_OK ){
2802e341ec69Sdan           if( nByte<0 || nByte>pIn->nData-pIn->iNext ){
2803e341ec69Sdan             rc = SQLITE_CORRUPT_BKPT;
2804e341ec69Sdan           }else{
28056734007dSdan             u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
28064757c658Sdan             rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
2807ef7a6304Sdan             pIn->iNext += nByte;
28084fccf43aSdan           }
2809e341ec69Sdan         }
2810e341ec69Sdan       }
28114fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2812ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
28134fccf43aSdan         if( eType==SQLITE_INTEGER ){
28144fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
28154fccf43aSdan         }else{
28164fccf43aSdan           double d;
28174e895da1Sdan           memcpy(&d, &v, 8);
28184fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
28194fccf43aSdan         }
2820ef7a6304Sdan         pIn->iNext += 8;
282191ddd559Sdan       }
28224fccf43aSdan     }
28234fccf43aSdan   }
28244fccf43aSdan 
2825ef7a6304Sdan   return rc;
2826ef7a6304Sdan }
2827ef7a6304Sdan 
2828ef7a6304Sdan /*
2829ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2830ef7a6304Sdan ** Specifically, to the following:
2831ef7a6304Sdan **
2832ef7a6304Sdan **   + number of columns in table (varint)
2833ef7a6304Sdan **   + array of PK flags (1 byte per column),
2834ef7a6304Sdan **   + table name (nul terminated).
2835ef7a6304Sdan **
2836ef7a6304Sdan ** This function ensures that all of the above is present in the input
2837ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2838ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2839ef7a6304Sdan ** The input pointer is not moved.
2840ef7a6304Sdan */
2841ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2842ef7a6304Sdan   int rc = SQLITE_OK;
2843ef7a6304Sdan   int nCol = 0;
28444757c658Sdan   int nRead = 0;
2845ef7a6304Sdan 
2846ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2847ef7a6304Sdan   if( rc==SQLITE_OK ){
28484757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
28496344eddaSdan     /* The hard upper limit for the number of columns in an SQLite
28506344eddaSdan     ** database table is, according to sqliteLimit.h, 32676. So
28516344eddaSdan     ** consider any table-header that purports to have more than 65536
28526344eddaSdan     ** columns to be corrupt. This is convenient because otherwise,
28536344eddaSdan     ** if the (nCol>65536) condition below were omitted, a sufficiently
28546344eddaSdan     ** large value for nCol may cause nRead to wrap around and become
28556344eddaSdan     ** negative. Leading to a crash. */
28566344eddaSdan     if( nCol<0 || nCol>65536 ){
2857e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2858e341ec69Sdan     }else{
28594757c658Sdan       rc = sessionInputBuffer(pIn, nRead+nCol+100);
28604757c658Sdan       nRead += nCol;
2861ef7a6304Sdan     }
2862e341ec69Sdan   }
28634757c658Sdan 
2864ef7a6304Sdan   while( rc==SQLITE_OK ){
28654757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
28664757c658Sdan       nRead++;
2867ef7a6304Sdan     }
2868e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
28694757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
28704757c658Sdan   }
2871e8fa8c96Sdan   *pnByte = nRead+1;
2872ef7a6304Sdan   return rc;
2873ef7a6304Sdan }
2874ef7a6304Sdan 
2875ef7a6304Sdan /*
2876fa122adaSdan ** The input pointer currently points to the first byte of the first field
2877fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
287816228167Sdan ** record is buffered. It does not move the input pointer.
287916228167Sdan **
288016228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
288116228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
288216228167Sdan ** final value of *pnByte is undefined in this case.
2883fa122adaSdan */
2884fa122adaSdan static int sessionChangesetBufferRecord(
288516228167Sdan   SessionInput *pIn,              /* Input data */
288616228167Sdan   int nCol,                       /* Number of columns in record */
288716228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2888fa122adaSdan ){
2889fa122adaSdan   int rc = SQLITE_OK;
2890fa122adaSdan   int nByte = 0;
2891fa122adaSdan   int i;
2892fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2893fa122adaSdan     int eType;
2894fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2895fa122adaSdan     if( rc==SQLITE_OK ){
2896fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2897fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2898fa122adaSdan         int n;
2899fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2900fa122adaSdan         nByte += n;
2901fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2902fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2903fa122adaSdan         nByte += 8;
2904fa122adaSdan       }
2905fa122adaSdan     }
2906fa122adaSdan   }
2907fa122adaSdan   *pnByte = nByte;
2908fa122adaSdan   return rc;
2909fa122adaSdan }
2910fa122adaSdan 
2911fa122adaSdan /*
2912ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2913ef7a6304Sdan ** Specifically, to the following:
2914ef7a6304Sdan **
2915ef7a6304Sdan **   + number of columns in table (varint)
2916ef7a6304Sdan **   + array of PK flags (1 byte per column),
2917ef7a6304Sdan **   + table name (nul terminated).
291816228167Sdan **
291916228167Sdan ** This function decodes the table-header and populates the p->nCol,
292016228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
292116228167Sdan ** also allocated or resized according to the new value of p->nCol. The
292216228167Sdan ** input pointer is left pointing to the byte following the table header.
292316228167Sdan **
292416228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
292516228167Sdan ** is returned and the final values of the various fields enumerated above
292616228167Sdan ** are undefined.
2927ef7a6304Sdan */
2928ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2929ef7a6304Sdan   int rc;
2930ef7a6304Sdan   int nCopy;
2931ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2932ef7a6304Sdan 
2933ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2934ef7a6304Sdan   if( rc==SQLITE_OK ){
2935ef7a6304Sdan     int nByte;
2936ef7a6304Sdan     int nVarint;
29374757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2938dd8a4af8Sdan     if( p->nCol>0 ){
2939ef7a6304Sdan       nCopy -= nVarint;
2940ef7a6304Sdan       p->in.iNext += nVarint;
2941ef7a6304Sdan       nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2942ef7a6304Sdan       p->tblhdr.nBuf = 0;
2943ef7a6304Sdan       sessionBufferGrow(&p->tblhdr, nByte, &rc);
2944dd8a4af8Sdan     }else{
2945e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2946dd8a4af8Sdan     }
2947ef7a6304Sdan   }
2948ef7a6304Sdan 
2949ef7a6304Sdan   if( rc==SQLITE_OK ){
2950f6ad201aSdrh     size_t iPK = sizeof(sqlite3_value*)*p->nCol*2;
2951ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
29524757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2953ef7a6304Sdan     p->in.iNext += nCopy;
2954ef7a6304Sdan   }
2955ef7a6304Sdan 
2956ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2957ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2958ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2959ef7a6304Sdan   return (p->rc = rc);
29604fccf43aSdan }
29614fccf43aSdan 
296277fc1d5bSdan /*
296377fc1d5bSdan ** Advance the changeset iterator to the next change.
296477fc1d5bSdan **
296577fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
296677fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
296777fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
296877fc1d5bSdan **
296977fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
297077fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
297177fc1d5bSdan ** the record to *pnRec.
297277fc1d5bSdan **
297377fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
297477fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
297577fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
297677fc1d5bSdan ** changes in the changeset.
297777fc1d5bSdan */
29785d607a6eSdan static int sessionChangesetNext(
297977fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
298077fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
2981c0a499eaSdan   int *pnRec,                     /* If non-NULL, store size of record here */
2982c0a499eaSdan   int *pbNew                      /* If non-NULL, true if new table */
29835d607a6eSdan ){
29844fccf43aSdan   int i;
2985ef7a6304Sdan   u8 op;
29864fccf43aSdan 
29875d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
29885d607a6eSdan 
2989296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
29904fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
29914fccf43aSdan 
29925d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
29934fccf43aSdan   if( p->apValue ){
29944fccf43aSdan     for(i=0; i<p->nCol*2; i++){
29954fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
29964fccf43aSdan     }
29974fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
29984fccf43aSdan   }
29994fccf43aSdan 
3000ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
3001ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
3002ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
3003ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
3004ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
3005ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
3006ef7a6304Sdan 
30074fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
30084757c658Sdan   if( p->in.iNext>=p->in.nData ){
30094fccf43aSdan     return SQLITE_DONE;
30104fccf43aSdan   }
30114fccf43aSdan 
3012d9151526Sdan   sessionDiscardData(&p->in);
3013d9151526Sdan   p->in.iCurrent = p->in.iNext;
3014d9151526Sdan 
30154757c658Sdan   op = p->in.aData[p->in.iNext++];
301607d0f15eSdan   while( op=='T' || op=='P' ){
3017c0a499eaSdan     if( pbNew ) *pbNew = 1;
3018ef7a6304Sdan     p->bPatchset = (op=='P');
3019ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
3020ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
3021d9151526Sdan     p->in.iCurrent = p->in.iNext;
302207d0f15eSdan     if( p->in.iNext>=p->in.nData ) return SQLITE_DONE;
30234757c658Sdan     op = p->in.aData[p->in.iNext++];
30245d607a6eSdan   }
30255d607a6eSdan 
302644748f27Sdan   if( p->zTab==0 || (p->bPatchset && p->bInvert) ){
3027dd8a4af8Sdan     /* The first record in the changeset is not a table header. Must be a
3028dd8a4af8Sdan     ** corrupt changeset. */
302944748f27Sdan     assert( p->in.iNext==1 || p->zTab );
3030dd8a4af8Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
3031dd8a4af8Sdan   }
3032dd8a4af8Sdan 
3033ef7a6304Sdan   p->op = op;
30344757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
30354fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
30364757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
30374fccf43aSdan   }
30384fccf43aSdan 
3039cbf6d2d2Sdan   if( paRec ){
3040cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
3041cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
3042cbf6d2d2Sdan       nVal = p->nCol * 2;
3043cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
3044cbf6d2d2Sdan       nVal = 0;
3045cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
3046cbf6d2d2Sdan     }else{
3047cbf6d2d2Sdan       nVal = p->nCol;
3048cbf6d2d2Sdan     }
3049cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
3050cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
3051cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
3052cbf6d2d2Sdan     p->in.iNext += *pnRec;
3053cbf6d2d2Sdan   }else{
305444748f27Sdan     sqlite3_value **apOld = (p->bInvert ? &p->apValue[p->nCol] : p->apValue);
305544748f27Sdan     sqlite3_value **apNew = (p->bInvert ? p->apValue : &p->apValue[p->nCol]);
30565d607a6eSdan 
30574fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
305873b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
305973b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
306044748f27Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, apOld);
30614fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30624fccf43aSdan     }
30634fccf43aSdan 
30644fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
30654fccf43aSdan     if( p->op!=SQLITE_DELETE ){
306644748f27Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, apNew);
30674fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30684fccf43aSdan     }
30694fccf43aSdan 
307044748f27Sdan     if( (p->bPatchset || p->bInvert) && p->op==SQLITE_UPDATE ){
307173b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
307273b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
307373b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
307473b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
307573b3c055Sdan       for(i=0; i<p->nCol; i++){
307644748f27Sdan         assert( p->bPatchset==0 || p->apValue[i]==0 );
307773b3c055Sdan         if( p->abPK[i] ){
307844748f27Sdan           assert( p->apValue[i]==0 );
307973b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
3080e341ec69Sdan           if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT);
308173b3c055Sdan           p->apValue[i+p->nCol] = 0;
308273b3c055Sdan         }
308373b3c055Sdan       }
308444748f27Sdan     }else if( p->bInvert ){
308544748f27Sdan       if( p->op==SQLITE_INSERT ) p->op = SQLITE_DELETE;
308644748f27Sdan       else if( p->op==SQLITE_DELETE ) p->op = SQLITE_INSERT;
308773b3c055Sdan     }
3088cbf6d2d2Sdan   }
3089ef7a6304Sdan 
30904fccf43aSdan   return SQLITE_ROW;
30914fccf43aSdan }
30924fccf43aSdan 
30934fccf43aSdan /*
30945d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
30955d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
30965d607a6eSdan ** or SQLITE_CORRUPT.
30975d607a6eSdan **
30985d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
30995d607a6eSdan ** callback by changeset_apply().
31005d607a6eSdan */
31015d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
3102c0a499eaSdan   return sessionChangesetNext(p, 0, 0, 0);
31035d607a6eSdan }
31045d607a6eSdan 
31055d607a6eSdan /*
3106244593c8Sdan ** The following function extracts information on the current change
310777fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
31084fccf43aSdan ** has returned SQLITE_ROW.
31094fccf43aSdan */
31104fccf43aSdan int sqlite3changeset_op(
3111296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
31124fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
31134fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
3114b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
3115b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
31164fccf43aSdan ){
31174fccf43aSdan   *pOp = pIter->op;
31184fccf43aSdan   *pnCol = pIter->nCol;
31194fccf43aSdan   *pzTab = pIter->zTab;
3120b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
31214fccf43aSdan   return SQLITE_OK;
31224fccf43aSdan }
31234fccf43aSdan 
312477fc1d5bSdan /*
312577fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
312677fc1d5bSdan ** the database table affected by the change that pIter currently points
312777fc1d5bSdan ** to. This function may only be called after changeset_next() returns
312877fc1d5bSdan ** SQLITE_ROW.
312977fc1d5bSdan */
3130244593c8Sdan int sqlite3changeset_pk(
3131244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
3132244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
3133244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
3134244593c8Sdan ){
3135244593c8Sdan   *pabPK = pIter->abPK;
3136244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
3137244593c8Sdan   return SQLITE_OK;
3138244593c8Sdan }
3139244593c8Sdan 
3140296c7658Sdan /*
3141296c7658Sdan ** This function may only be called while the iterator is pointing to an
3142296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
3143296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3144296c7658Sdan **
3145296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3146296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
3147296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3148296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
3149296c7658Sdan **
3150296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3151296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3152296c7658Sdan */
31534fccf43aSdan int sqlite3changeset_old(
3154296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3155296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
31564fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
31574fccf43aSdan ){
3158d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
3159d5f0767cSdan     return SQLITE_MISUSE;
3160d5f0767cSdan   }
31614fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
31624fccf43aSdan     return SQLITE_RANGE;
31634fccf43aSdan   }
31644fccf43aSdan   *ppValue = pIter->apValue[iVal];
31654fccf43aSdan   return SQLITE_OK;
31664fccf43aSdan }
31674fccf43aSdan 
3168296c7658Sdan /*
3169296c7658Sdan ** This function may only be called while the iterator is pointing to an
3170296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
3171296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3172296c7658Sdan **
3173296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3174296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
3175296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3176296c7658Sdan ** was not modified), set *ppValue to NULL.
3177296c7658Sdan **
3178296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3179296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3180296c7658Sdan */
31814fccf43aSdan int sqlite3changeset_new(
3182296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3183296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
31844fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
31854fccf43aSdan ){
3186d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
3187d5f0767cSdan     return SQLITE_MISUSE;
3188d5f0767cSdan   }
31894fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
31904fccf43aSdan     return SQLITE_RANGE;
31914fccf43aSdan   }
31924fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
31934fccf43aSdan   return SQLITE_OK;
31944fccf43aSdan }
31954fccf43aSdan 
3196296c7658Sdan /*
31977aa469cdSdan ** The following two macros are used internally. They are similar to the
31987aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
31997aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
32007aa469cdSdan */
32017aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
32027aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
32037aa469cdSdan 
32047aa469cdSdan /*
3205296c7658Sdan ** This function may only be called with a changeset iterator that has been
3206296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
3207296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
3208296c7658Sdan **
3209296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
3210296c7658Sdan ** containing the iVal'th value of the conflicting record.
3211296c7658Sdan **
3212296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
3213296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
3214296c7658Sdan */
3215d5f0767cSdan int sqlite3changeset_conflict(
3216296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3217296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
3218d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
3219d5f0767cSdan ){
3220d5f0767cSdan   if( !pIter->pConflict ){
3221d5f0767cSdan     return SQLITE_MISUSE;
3222d5f0767cSdan   }
3223ff677b20Sdan   if( iVal<0 || iVal>=pIter->nCol ){
3224d5f0767cSdan     return SQLITE_RANGE;
3225d5f0767cSdan   }
3226d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
3227d5f0767cSdan   return SQLITE_OK;
3228d5f0767cSdan }
3229d5f0767cSdan 
32304fccf43aSdan /*
3231cb3e4b79Sdan ** This function may only be called with an iterator passed to an
3232cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
3233cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
3234cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
3235cb3e4b79Sdan **
3236cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
3237cb3e4b79Sdan */
3238cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
3239cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3240cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
3241cb3e4b79Sdan ){
3242cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
3243cb3e4b79Sdan     return SQLITE_MISUSE;
3244cb3e4b79Sdan   }
3245cb3e4b79Sdan   *pnOut = pIter->nCol;
3246cb3e4b79Sdan   return SQLITE_OK;
3247cb3e4b79Sdan }
3248cb3e4b79Sdan 
3249cb3e4b79Sdan 
3250cb3e4b79Sdan /*
32514fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
32524fccf43aSdan **
32534fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
32544fccf43aSdan ** callback by changeset_apply().
32554fccf43aSdan */
32564fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
3257cbf6d2d2Sdan   int rc = SQLITE_OK;
3258cbf6d2d2Sdan   if( p ){
3259296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
3260cbf6d2d2Sdan     rc = p->rc;
326112ca0b56Sdan     if( p->apValue ){
32624fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
326312ca0b56Sdan     }
3264ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
32654757c658Sdan     sqlite3_free(p->in.buf.aBuf);
32664fccf43aSdan     sqlite3_free(p);
3267cbf6d2d2Sdan   }
32684fccf43aSdan   return rc;
32694fccf43aSdan }
32704fccf43aSdan 
3271fa122adaSdan static int sessionChangesetInvert(
3272fa122adaSdan   SessionInput *pInput,           /* Input changeset */
3273fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3274fa122adaSdan   void *pOut,
327591ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
327691ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
327791ddd559Sdan ){
3278cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
3279fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
3280cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
3281cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
3282cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
3283ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
328491ddd559Sdan 
3285fa122adaSdan   /* Initialize the output buffer */
3286fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
3287fa122adaSdan 
328891ddd559Sdan   /* Zero the output variables in case an error occurs. */
3289fa122adaSdan   if( ppInverted ){
329091ddd559Sdan     *ppInverted = 0;
329191ddd559Sdan     *pnInverted = 0;
3292fa122adaSdan   }
329391ddd559Sdan 
3294fa122adaSdan   while( 1 ){
3295ef7a6304Sdan     u8 eType;
3296fa122adaSdan 
3297fa122adaSdan     /* Test for EOF. */
3298fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
3299fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
3300fa122adaSdan     eType = pInput->aData[pInput->iNext];
3301fa122adaSdan 
330291ddd559Sdan     switch( eType ){
330391ddd559Sdan       case 'T': {
3304244593c8Sdan         /* A 'table' record consists of:
3305244593c8Sdan         **
3306244593c8Sdan         **   * A constant 'T' character,
3307244593c8Sdan         **   * Number of columns in said table (a varint),
3308ef7a6304Sdan         **   * An array of nCol bytes (sPK),
3309244593c8Sdan         **   * A nul-terminated table name.
3310244593c8Sdan         */
3311ef7a6304Sdan         int nByte;
3312fa122adaSdan         int nVar;
3313fa122adaSdan         pInput->iNext++;
3314fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
3315ef7a6304Sdan           goto finished_invert;
3316ef7a6304Sdan         }
3317fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
3318ef7a6304Sdan         sPK.nBuf = 0;
3319fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
3320fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3321fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3322ef7a6304Sdan         if( rc ) goto finished_invert;
3323fa122adaSdan 
3324fa122adaSdan         pInput->iNext += nByte;
3325cfec7eeeSdan         sqlite3_free(apVal);
3326cfec7eeeSdan         apVal = 0;
3327ef7a6304Sdan         abPK = sPK.aBuf;
332891ddd559Sdan         break;
332991ddd559Sdan       }
333091ddd559Sdan 
333191ddd559Sdan       case SQLITE_INSERT:
333291ddd559Sdan       case SQLITE_DELETE: {
333391ddd559Sdan         int nByte;
3334fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
3335fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
3336fa122adaSdan         pInput->iNext += 2;
3337fa122adaSdan         assert( rc==SQLITE_OK );
3338fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
3339fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
3340fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
3341fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3342fa122adaSdan         pInput->iNext += nByte;
3343fa122adaSdan         if( rc ) goto finished_invert;
334491ddd559Sdan         break;
334591ddd559Sdan       }
334691ddd559Sdan 
334791ddd559Sdan       case SQLITE_UPDATE: {
3348cfec7eeeSdan         int iCol;
334991ddd559Sdan 
3350cfec7eeeSdan         if( 0==apVal ){
33512d77d80aSdrh           apVal = (sqlite3_value **)sqlite3_malloc64(sizeof(apVal[0])*nCol*2);
3352cfec7eeeSdan           if( 0==apVal ){
3353cfec7eeeSdan             rc = SQLITE_NOMEM;
3354cfec7eeeSdan             goto finished_invert;
3355cfec7eeeSdan           }
3356cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3357cfec7eeeSdan         }
335891ddd559Sdan 
3359cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
3360fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3361fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
336291ddd559Sdan 
3363ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
3364fa122adaSdan         pInput->iNext += 2;
3365fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
3366ef7a6304Sdan         if( rc==SQLITE_OK ){
3367fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
3368ef7a6304Sdan         }
3369ef7a6304Sdan 
3370cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
3371cfec7eeeSdan         ** original old.* record, and the other values from the original
3372cfec7eeeSdan         ** new.* record. */
3373e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3374cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
3375fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3376cfec7eeeSdan         }
3377cfec7eeeSdan 
3378cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
3379cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
3380cfec7eeeSdan         ** are set to "undefined". */
3381e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3382cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
3383fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3384cfec7eeeSdan         }
3385cfec7eeeSdan 
3386cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
3387cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
3388cfec7eeeSdan         }
3389cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3390cfec7eeeSdan         if( rc!=SQLITE_OK ){
3391cfec7eeeSdan           goto finished_invert;
3392cfec7eeeSdan         }
3393cfec7eeeSdan 
339491ddd559Sdan         break;
339591ddd559Sdan       }
339691ddd559Sdan 
339791ddd559Sdan       default:
33984757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
3399cfec7eeeSdan         goto finished_invert;
340091ddd559Sdan     }
3401fa122adaSdan 
3402fa122adaSdan     assert( rc==SQLITE_OK );
34031f48e67dSdan     if( xOutput && sOut.nBuf>=sessions_strm_chunk_size ){
3404fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3405fa122adaSdan       sOut.nBuf = 0;
3406fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
3407fa122adaSdan     }
340891ddd559Sdan   }
340991ddd559Sdan 
3410cfec7eeeSdan   assert( rc==SQLITE_OK );
3411fa122adaSdan   if( pnInverted ){
3412fa122adaSdan     *pnInverted = sOut.nBuf;
3413fa122adaSdan     *ppInverted = sOut.aBuf;
3414fa122adaSdan     sOut.aBuf = 0;
3415fa122adaSdan   }else if( sOut.nBuf>0 ){
3416fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3417fa122adaSdan   }
3418cfec7eeeSdan 
3419cfec7eeeSdan  finished_invert:
3420fa122adaSdan   sqlite3_free(sOut.aBuf);
3421cfec7eeeSdan   sqlite3_free(apVal);
3422ef7a6304Sdan   sqlite3_free(sPK.aBuf);
3423cfec7eeeSdan   return rc;
342491ddd559Sdan }
342591ddd559Sdan 
3426fa122adaSdan 
3427fa122adaSdan /*
3428fa122adaSdan ** Invert a changeset object.
3429fa122adaSdan */
3430fa122adaSdan int sqlite3changeset_invert(
3431fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
3432fa122adaSdan   const void *pChangeset,         /* Input changeset */
3433fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
3434fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
3435fa122adaSdan ){
3436fa122adaSdan   SessionInput sInput;
3437fa122adaSdan 
3438fa122adaSdan   /* Set up the input stream */
3439fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3440fa122adaSdan   sInput.nData = nChangeset;
3441fa122adaSdan   sInput.aData = (u8*)pChangeset;
3442fa122adaSdan 
3443fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
3444fa122adaSdan }
3445fa122adaSdan 
3446fa122adaSdan /*
3447fa122adaSdan ** Streaming version of sqlite3changeset_invert().
3448fa122adaSdan */
3449f1a08ad8Sdrh int sqlite3changeset_invert_strm(
3450fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
3451fa122adaSdan   void *pIn,
3452fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3453fa122adaSdan   void *pOut
3454fa122adaSdan ){
3455fa122adaSdan   SessionInput sInput;
3456fa122adaSdan   int rc;
3457fa122adaSdan 
3458fa122adaSdan   /* Set up the input stream */
3459fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3460fa122adaSdan   sInput.xInput = xInput;
3461fa122adaSdan   sInput.pIn = pIn;
3462fa122adaSdan 
3463fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
3464fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
3465fa122adaSdan   return rc;
3466fa122adaSdan }
3467fa122adaSdan 
34680c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
34690c698471Sdan struct SessionApplyCtx {
34700c698471Sdan   sqlite3 *db;
34710c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
3472cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
34730c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
34740c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
34750c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
34760c698471Sdan   const char **azCol;             /* Array of column names */
34770c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
3478d1cccf19Sdan   int bStat1;                     /* True if table is sqlite_stat1 */
3479d9151526Sdan   int bDeferConstraints;          /* True to defer constraints */
3480d9151526Sdan   SessionBuffer constraints;      /* Deferred constraints are stored here */
3481a38e6c57Sdan   SessionBuffer rebase;           /* Rebase information (if any) here */
3482dbe7d37aSdan   u8 bRebaseStarted;              /* If table header is already in rebase */
3483dbe7d37aSdan   u8 bRebase;                     /* True to collect rebase information */
34840c698471Sdan };
34850c698471Sdan 
3486d5f0767cSdan /*
3487d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
3488d5f0767cSdan ** structure like this:
3489d5f0767cSdan **
3490d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3491d5f0767cSdan **
3492d5f0767cSdan ** The DELETE statement looks like this:
3493d5f0767cSdan **
3494db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
3495d5f0767cSdan **
3496d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
3497d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
3498d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
3499296c7658Sdan **
3500296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
3501296c7658Sdan ** pointing to the prepared version of the SQL statement.
3502d5f0767cSdan */
3503d5f0767cSdan static int sessionDeleteRow(
3504d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3505d5f0767cSdan   const char *zTab,               /* Table name */
35060c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3507d5f0767cSdan ){
3508296c7658Sdan   int i;
3509296c7658Sdan   const char *zSep = "";
3510d5f0767cSdan   int rc = SQLITE_OK;
3511d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
35127cf7df7dSdan   int nPk = 0;
3513d5f0767cSdan 
3514d5f0767cSdan   sessionAppendStr(&buf, "DELETE FROM ", &rc);
3515d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3516296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
3517296c7658Sdan 
3518296c7658Sdan   for(i=0; i<p->nCol; i++){
3519296c7658Sdan     if( p->abPK[i] ){
35207cf7df7dSdan       nPk++;
3521296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
3522296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3523296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
3524296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
3525296c7658Sdan       zSep = " AND ";
3526296c7658Sdan     }
3527296c7658Sdan   }
3528296c7658Sdan 
35297cf7df7dSdan   if( nPk<p->nCol ){
3530296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
3531296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
3532296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
3533296c7658Sdan 
3534296c7658Sdan     zSep = "";
3535296c7658Sdan     for(i=0; i<p->nCol; i++){
3536296c7658Sdan       if( !p->abPK[i] ){
3537296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
3538296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
3539296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
3540296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
3541296c7658Sdan         zSep = "AND ";
3542296c7658Sdan       }
3543296c7658Sdan     }
3544296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
35457cf7df7dSdan   }
3546d5f0767cSdan 
3547d5f0767cSdan   if( rc==SQLITE_OK ){
35480c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
3549d5f0767cSdan   }
3550d5f0767cSdan   sqlite3_free(buf.aBuf);
3551d5f0767cSdan 
3552d5f0767cSdan   return rc;
3553d5f0767cSdan }
3554d5f0767cSdan 
3555d5f0767cSdan /*
3556d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
3557d5f0767cSdan ** Assuming a table structure like this:
3558d5f0767cSdan **
3559d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3560d5f0767cSdan **
3561d5f0767cSdan ** The UPDATE statement looks like this:
3562d5f0767cSdan **
3563d5f0767cSdan **     UPDATE x SET
3564d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3565964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3566964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3567964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3568d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3569964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3570d5f0767cSdan **     )
3571d5f0767cSdan **
3572d5f0767cSdan ** For each column in the table, there are three variables to bind:
3573d5f0767cSdan **
3574d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3575d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3576d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3577d5f0767cSdan **
3578d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3579d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3580d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3581d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3582d5f0767cSdan **
3583296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3584296c7658Sdan ** pointing to the prepared version of the SQL statement.
3585d5f0767cSdan */
3586d5f0767cSdan static int sessionUpdateRow(
3587d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3588d5f0767cSdan   const char *zTab,               /* Table name */
35890c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3590d5f0767cSdan ){
3591d5f0767cSdan   int rc = SQLITE_OK;
3592d5f0767cSdan   int i;
3593d5f0767cSdan   const char *zSep = "";
3594d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3595d5f0767cSdan 
3596d5f0767cSdan   /* Append "UPDATE tbl SET " */
3597d5f0767cSdan   sessionAppendStr(&buf, "UPDATE ", &rc);
3598d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3599d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3600d5f0767cSdan 
3601d5f0767cSdan   /* Append the assignments */
36020c698471Sdan   for(i=0; i<p->nCol; i++){
3603d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
36040c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3605d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3606d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3607d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3608d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3609d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
36100c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3611d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3612d5f0767cSdan     zSep = ", ";
3613d5f0767cSdan   }
3614d5f0767cSdan 
3615d5f0767cSdan   /* Append the PK part of the WHERE clause */
3616d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
36170c698471Sdan   for(i=0; i<p->nCol; i++){
36180c698471Sdan     if( p->abPK[i] ){
36190c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3620d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3621d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3622d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3623d5f0767cSdan     }
3624d5f0767cSdan   }
3625d5f0767cSdan 
3626d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3627d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
36280c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3629d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
36300c698471Sdan   for(i=0; i<p->nCol; i++){
36310c698471Sdan     if( !p->abPK[i] ){
3632d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3633d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3634d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
36350c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3636d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3637d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3638d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3639d5f0767cSdan     }
3640d5f0767cSdan   }
3641d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3642d5f0767cSdan 
3643d5f0767cSdan   if( rc==SQLITE_OK ){
36440c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3645d5f0767cSdan   }
3646d5f0767cSdan   sqlite3_free(buf.aBuf);
3647d5f0767cSdan 
3648d5f0767cSdan   return rc;
3649d5f0767cSdan }
3650d5f0767cSdan 
36513739f298Sdan 
3652296c7658Sdan /*
3653296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3654296c7658Sdan ** key. Assuming the following table structure:
3655296c7658Sdan **
3656296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3657296c7658Sdan **
3658296c7658Sdan ** The SELECT statement looks like this:
3659296c7658Sdan **
3660296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3661296c7658Sdan **
3662296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3663296c7658Sdan ** pointing to the prepared version of the SQL statement.
3664296c7658Sdan */
3665d5f0767cSdan static int sessionSelectRow(
3666d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3667d5f0767cSdan   const char *zTab,               /* Table name */
36680c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3669d5f0767cSdan ){
3670d7fb7d24Sdan   return sessionSelectStmt(
3671d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3672d5f0767cSdan }
3673d5f0767cSdan 
3674296c7658Sdan /*
3675296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3676296c7658Sdan ** For example:
3677296c7658Sdan **
3678296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3679296c7658Sdan **
3680296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3681296c7658Sdan ** pointing to the prepared version of the SQL statement.
3682296c7658Sdan */
36830c698471Sdan static int sessionInsertRow(
36840c698471Sdan   sqlite3 *db,                    /* Database handle */
36850c698471Sdan   const char *zTab,               /* Table name */
36860c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
36870c698471Sdan ){
36880c698471Sdan   int rc = SQLITE_OK;
36890c698471Sdan   int i;
36900c698471Sdan   SessionBuffer buf = {0, 0, 0};
36910c698471Sdan 
36920c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
36930c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
3694ff677b20Sdan   sessionAppendStr(&buf, "(", &rc);
3695ff677b20Sdan   for(i=0; i<p->nCol; i++){
3696ff677b20Sdan     if( i!=0 ) sessionAppendStr(&buf, ", ", &rc);
3697ff677b20Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3698ff677b20Sdan   }
3699ff677b20Sdan 
3700ff677b20Sdan   sessionAppendStr(&buf, ") VALUES(?", &rc);
37010c698471Sdan   for(i=1; i<p->nCol; i++){
37020c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
37030c698471Sdan   }
37040c698471Sdan   sessionAppendStr(&buf, ")", &rc);
37050c698471Sdan 
37060c698471Sdan   if( rc==SQLITE_OK ){
37070c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
37080c698471Sdan   }
37090c698471Sdan   sqlite3_free(buf.aBuf);
37100c698471Sdan   return rc;
37110c698471Sdan }
37120c698471Sdan 
37133739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){
37143739f298Sdan   return sqlite3_prepare_v2(db, zSql, -1, pp, 0);
37153739f298Sdan }
37163739f298Sdan 
37173739f298Sdan /*
37183739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table.
37193739f298Sdan ** These are similar to those created by sessionSelectRow(),
37203739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for
37213739f298Sdan ** other tables.
37223739f298Sdan */
37233739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){
37243739f298Sdan   int rc = sessionSelectRow(db, "sqlite_stat1", p);
37253739f298Sdan   if( rc==SQLITE_OK ){
37263739f298Sdan     rc = sessionPrepare(db, &p->pInsert,
37273739f298Sdan         "INSERT INTO main.sqlite_stat1 VALUES(?1, "
37283739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, "
37293739f298Sdan         "?3)"
37303739f298Sdan     );
37313739f298Sdan   }
37323739f298Sdan   if( rc==SQLITE_OK ){
37333739f298Sdan     rc = sessionPrepare(db, &p->pUpdate,
37343739f298Sdan         "UPDATE main.sqlite_stat1 SET "
37353739f298Sdan         "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, "
37363739f298Sdan         "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, "
37373739f298Sdan         "stat = CASE WHEN ?8 THEN ?9 ELSE stat END  "
37383739f298Sdan         "WHERE tbl=?1 AND idx IS "
37393739f298Sdan         "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END "
37403739f298Sdan         "AND (?10 OR ?8=0 OR stat IS ?7)"
37413739f298Sdan     );
37423739f298Sdan   }
37433739f298Sdan   if( rc==SQLITE_OK ){
37443739f298Sdan     rc = sessionPrepare(db, &p->pDelete,
37453739f298Sdan         "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS "
37463739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END "
37473739f298Sdan         "AND (?4 OR stat IS ?3)"
37483739f298Sdan     );
37493739f298Sdan   }
37503739f298Sdan   return rc;
37513739f298Sdan }
37523739f298Sdan 
3753296c7658Sdan /*
37547aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
37557aa469cdSdan ** See comments in the body of this function for details.
37567aa469cdSdan */
37577aa469cdSdan static int sessionBindValue(
37587aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
37597aa469cdSdan   int i,                          /* Parameter number to bind to */
37607aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
37617aa469cdSdan ){
37625671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3763082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3764082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3765082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3766082c96dfSdan   ** set to SQLITE_NULL.  */
37675671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
37687aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
37697aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3770082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
37717aa469cdSdan     return SQLITE_NOMEM;
37727aa469cdSdan   }
37737aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
37747aa469cdSdan }
37757aa469cdSdan 
37767aa469cdSdan /*
3777db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3778db04571cSdan ** transfers new.* values from the current iterator entry to statement
3779db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3780db04571cSdan **
3781d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of
3782db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3783db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3784db04571cSdan ** to an array nCol elements in size. In this case only those values for
3785db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3786db04571cSdan ** statement.
3787db04571cSdan **
3788db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3789db04571cSdan */
37907aa469cdSdan static int sessionBindRow(
3791db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
37927aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3793db04571cSdan   int nCol,                       /* Number of columns */
3794db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3795db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3796db04571cSdan ){
3797db04571cSdan   int i;
3798db04571cSdan   int rc = SQLITE_OK;
37997aa469cdSdan 
38007aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
38017aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
38027aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
38037aa469cdSdan   ** in the code below. */
38047aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
38057aa469cdSdan 
3806db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3807db04571cSdan     if( !abPK || abPK[i] ){
3808db04571cSdan       sqlite3_value *pVal;
38097aa469cdSdan       (void)xValue(pIter, i, &pVal);
3810dd8a4af8Sdan       if( pVal==0 ){
3811dd8a4af8Sdan         /* The value in the changeset was "undefined". This indicates a
3812dd8a4af8Sdan         ** corrupt changeset blob.  */
3813e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
3814dd8a4af8Sdan       }else{
38157aa469cdSdan         rc = sessionBindValue(pStmt, i+1, pVal);
3816db04571cSdan       }
3817db04571cSdan     }
3818dd8a4af8Sdan   }
3819db04571cSdan   return rc;
3820db04571cSdan }
3821db04571cSdan 
3822db04571cSdan /*
3823296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3824296c7658Sdan ** This function binds the primary key values from the change that changeset
3825296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3826296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3827296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3828296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
38297aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
38307aa469cdSdan **
38317aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
38327aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
38337aa469cdSdan ** not require a reset().
3834296c7658Sdan **
3835296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3836db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3837db04571cSdan ** UPDATE, bind values from the old.* record.
3838296c7658Sdan */
38390c698471Sdan static int sessionSeekToRow(
384037f133ecSdan   sqlite3 *db,                    /* Database handle */
384137f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
384237f133ecSdan   u8 *abPK,                       /* Primary key flags array */
38430c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
384437f133ecSdan ){
38457aa469cdSdan   int rc;                         /* Return code */
3846296c7658Sdan   int nCol;                       /* Number of columns in table */
3847296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3848296c7658Sdan   const char *zDummy;             /* Unused */
384937f133ecSdan 
3850b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
38517aa469cdSdan   rc = sessionBindRow(pIter,
3852db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3853db04571cSdan       nCol, abPK, pSelect
3854db04571cSdan   );
38550c698471Sdan 
38560c698471Sdan   if( rc==SQLITE_OK ){
38570c698471Sdan     rc = sqlite3_step(pSelect);
38580c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
38590c698471Sdan   }
38600c698471Sdan 
38610c698471Sdan   return rc;
38620c698471Sdan }
38630c698471Sdan 
3864b880a7b1Sdan /*
3865a920b209Sdrh ** This function is called from within sqlite3changeset_apply_v2() when
3866b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution
3867b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE)..
3868b880a7b1Sdan ** It adds a conflict resolution record to the buffer in
3869b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller
3870b880a7b1Sdan ** of apply_v2() as the "rebase" buffer.
3871b880a7b1Sdan **
3872b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise.
3873b880a7b1Sdan */
3874a38e6c57Sdan static int sessionRebaseAdd(
3875b880a7b1Sdan   SessionApplyCtx *p,             /* Apply context */
3876b880a7b1Sdan   int eType,                      /* Conflict resolution (OMIT or REPLACE) */
3877b880a7b1Sdan   sqlite3_changeset_iter *pIter   /* Iterator pointing at current change */
3878a38e6c57Sdan ){
3879a38e6c57Sdan   int rc = SQLITE_OK;
3880dbe7d37aSdan   if( p->bRebase ){
3881a38e6c57Sdan     int i;
3882a38e6c57Sdan     int eOp = pIter->op;
3883a38e6c57Sdan     if( p->bRebaseStarted==0 ){
3884a38e6c57Sdan       /* Append a table-header to the rebase buffer */
3885a38e6c57Sdan       const char *zTab = pIter->zTab;
3886a38e6c57Sdan       sessionAppendByte(&p->rebase, 'T', &rc);
3887a38e6c57Sdan       sessionAppendVarint(&p->rebase, p->nCol, &rc);
3888a38e6c57Sdan       sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc);
3889a38e6c57Sdan       sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc);
3890a38e6c57Sdan       p->bRebaseStarted = 1;
3891a38e6c57Sdan     }
3892a38e6c57Sdan 
3893a38e6c57Sdan     assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT );
3894a38e6c57Sdan     assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE );
3895a38e6c57Sdan 
3896f01d3a7eSdan     sessionAppendByte(&p->rebase,
3897f01d3a7eSdan         (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc
3898f01d3a7eSdan         );
3899f01d3a7eSdan     sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc);
3900a38e6c57Sdan     for(i=0; i<p->nCol; i++){
3901a38e6c57Sdan       sqlite3_value *pVal = 0;
3902f01d3a7eSdan       if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){
3903a38e6c57Sdan         sqlite3changeset_old(pIter, i, &pVal);
3904a38e6c57Sdan       }else{
3905a38e6c57Sdan         sqlite3changeset_new(pIter, i, &pVal);
3906a38e6c57Sdan       }
3907a38e6c57Sdan       sessionAppendValue(&p->rebase, pVal, &rc);
3908a38e6c57Sdan     }
3909dbe7d37aSdan   }
3910a38e6c57Sdan   return rc;
3911a38e6c57Sdan }
3912a38e6c57Sdan 
3913296c7658Sdan /*
3914296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3915296c7658Sdan ** currently points to.
3916296c7658Sdan **
3917296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3918296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3919296c7658Sdan ** depends solely on eType, as follows:
3920296c7658Sdan **
3921296c7658Sdan **    eType value                 Value passed to xConflict
3922296c7658Sdan **    -------------------------------------------------
3923296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3924296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3925296c7658Sdan **
3926296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3927296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3928296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3929296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3930296c7658Sdan ** handler invoked is as follows:
3931296c7658Sdan **
3932296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3933296c7658Sdan **    ----------------------------------------------------------------
3934296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3935296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3936296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3937296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3938296c7658Sdan **
3939296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3940296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3941296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3942296c7658Sdan **
3943296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3944296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3945296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3946296c7658Sdan ** this function returns SQLITE_OK.
3947296c7658Sdan */
39480c698471Sdan static int sessionConflictHandler(
3949296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3950296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
39510c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
39520c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3953296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3954296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
39550c698471Sdan ){
395674f598b6Smistachkin   int res = 0;                    /* Value returned by conflict handler */
39570c698471Sdan   int rc;
39580c698471Sdan   int nCol;
39590c698471Sdan   int op;
39600c698471Sdan   const char *zDummy;
39610c698471Sdan 
3962b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
39630c698471Sdan 
39640c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
39650c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
39660c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
396737f133ecSdan 
396837f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
39690c698471Sdan   if( pbReplace ){
39700c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
39710c698471Sdan   }else{
3972db04571cSdan     rc = SQLITE_OK;
39730c698471Sdan   }
39740c698471Sdan 
39750c698471Sdan   if( rc==SQLITE_ROW ){
39760c698471Sdan     /* There exists another row with the new.* primary key. */
39770c698471Sdan     pIter->pConflict = p->pSelect;
39780c698471Sdan     res = xConflict(pCtx, eType, pIter);
39790c698471Sdan     pIter->pConflict = 0;
39800c698471Sdan     rc = sqlite3_reset(p->pSelect);
3981db04571cSdan   }else if( rc==SQLITE_OK ){
3982d9151526Sdan     if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){
3983d9151526Sdan       /* Instead of invoking the conflict handler, append the change blob
3984d9151526Sdan       ** to the SessionApplyCtx.constraints buffer. */
3985d9151526Sdan       u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent];
3986d9151526Sdan       int nBlob = pIter->in.iNext - pIter->in.iCurrent;
3987d9151526Sdan       sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc);
3988a38e6c57Sdan       return SQLITE_OK;
3989d9151526Sdan     }else{
39900c698471Sdan       /* No other row with the new.* primary key. */
39910c698471Sdan       res = xConflict(pCtx, eType+1, pIter);
39920c698471Sdan       if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
399337f133ecSdan     }
3994d9151526Sdan   }
399537f133ecSdan 
399637f133ecSdan   if( rc==SQLITE_OK ){
39970c698471Sdan     switch( res ){
39980c698471Sdan       case SQLITE_CHANGESET_REPLACE:
3999f51e5f6cSdan         assert( pbReplace );
4000f51e5f6cSdan         *pbReplace = 1;
40010c698471Sdan         break;
40020c698471Sdan 
40030c698471Sdan       case SQLITE_CHANGESET_OMIT:
40040c698471Sdan         break;
40050c698471Sdan 
40060c698471Sdan       case SQLITE_CHANGESET_ABORT:
40070c698471Sdan         rc = SQLITE_ABORT;
40080c698471Sdan         break;
40090c698471Sdan 
40100c698471Sdan       default:
40110c698471Sdan         rc = SQLITE_MISUSE;
40120c698471Sdan         break;
40130c698471Sdan     }
4014a38e6c57Sdan     if( rc==SQLITE_OK ){
4015a38e6c57Sdan       rc = sessionRebaseAdd(p, res, pIter);
4016a38e6c57Sdan     }
40170c698471Sdan   }
40180c698471Sdan 
40190c698471Sdan   return rc;
40200c698471Sdan }
40210c698471Sdan 
4022296c7658Sdan /*
4023296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
4024296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
4025296c7658Sdan ** the conflict handler callback.
4026296c7658Sdan **
4027296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
4028296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
4029296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
4030296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
4031296c7658Sdan ** to true before returning. In this case the caller will invoke this function
4032296c7658Sdan ** again, this time with pbRetry set to NULL.
4033296c7658Sdan **
4034296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
4035296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
4036296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
4037296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
4038296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
4039296c7658Sdan ** row before invoking this function again, this time with pbReplace set
4040296c7658Sdan ** to NULL.
4041296c7658Sdan **
4042296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
4043296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
4044296c7658Sdan ** returned.
4045296c7658Sdan */
40460c698471Sdan static int sessionApplyOneOp(
4047296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
4048296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
40490c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
4050296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
4051296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
4052296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
40530c698471Sdan ){
40540c698471Sdan   const char *zDummy;
40550c698471Sdan   int op;
40560c698471Sdan   int nCol;
40570c698471Sdan   int rc = SQLITE_OK;
40580c698471Sdan 
40590c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
40600c698471Sdan   assert( p->azCol && p->abPK );
40610c698471Sdan   assert( !pbReplace || *pbReplace==0 );
40620c698471Sdan 
4063b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
40640c698471Sdan 
40650c698471Sdan   if( op==SQLITE_DELETE ){
40660c698471Sdan 
406773b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
406873b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
406973b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
407073b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
407173b3c055Sdan     ** is not required if:
407273b3c055Sdan     **
407373b3c055Sdan     **   * this is a patchset, or
407473b3c055Sdan     **   * (pbRetry==0), or
407573b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
407673b3c055Sdan     **     no (nCol+1) variable to bind to).
407773b3c055Sdan     */
407873b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
407973b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
40807cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
408173b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
40827cf7df7dSdan     }
40830c698471Sdan     if( rc!=SQLITE_OK ) return rc;
40840c698471Sdan 
40850c698471Sdan     sqlite3_step(p->pDelete);
40860c698471Sdan     rc = sqlite3_reset(p->pDelete);
40870c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
40880c698471Sdan       rc = sessionConflictHandler(
40890c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
40900c698471Sdan       );
409135e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
40920c698471Sdan       rc = sessionConflictHandler(
40930c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
40940c698471Sdan       );
40950c698471Sdan     }
40960c698471Sdan 
40970c698471Sdan   }else if( op==SQLITE_UPDATE ){
40980c698471Sdan     int i;
40990c698471Sdan 
41000c698471Sdan     /* Bind values to the UPDATE statement. */
41010c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
41027aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
41037aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
41047aa469cdSdan 
41050c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
41067aa469cdSdan       if( pOld ){
41077aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
41087aa469cdSdan       }
41097aa469cdSdan       if( rc==SQLITE_OK && pNew ){
41107aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
41110c698471Sdan       }
41120c698471Sdan     }
411373b3c055Sdan     if( rc==SQLITE_OK ){
411473b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
411573b3c055Sdan     }
41160c698471Sdan     if( rc!=SQLITE_OK ) return rc;
41170c698471Sdan 
41180c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
41190c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
41200c698471Sdan     sqlite3_step(p->pUpdate);
41210c698471Sdan     rc = sqlite3_reset(p->pUpdate);
41220c698471Sdan 
41230c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
41240c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
41250c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
41260c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
41270c698471Sdan 
41280c698471Sdan       rc = sessionConflictHandler(
41290c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
41300c698471Sdan       );
41310c698471Sdan 
413235e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
4133db04571cSdan       /* This is always a CONSTRAINT conflict. */
4134db04571cSdan       rc = sessionConflictHandler(
4135db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
41360c698471Sdan       );
41370c698471Sdan     }
41380c698471Sdan 
41390c698471Sdan   }else{
41400c698471Sdan     assert( op==SQLITE_INSERT );
4141d1cccf19Sdan     if( p->bStat1 ){
4142d1cccf19Sdan       /* Check if there is a conflicting row. For sqlite_stat1, this needs
4143d1cccf19Sdan       ** to be done using a SELECT, as there is no PRIMARY KEY in the
4144d1cccf19Sdan       ** database schema to throw an exception if a duplicate is inserted.  */
4145d1cccf19Sdan       rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
4146d1cccf19Sdan       if( rc==SQLITE_ROW ){
4147d1cccf19Sdan         rc = SQLITE_CONSTRAINT;
4148d1cccf19Sdan         sqlite3_reset(p->pSelect);
4149d1cccf19Sdan       }
4150d1cccf19Sdan     }
4151d1cccf19Sdan 
4152d1cccf19Sdan     if( rc==SQLITE_OK ){
41537aa469cdSdan       rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
41540c698471Sdan       if( rc!=SQLITE_OK ) return rc;
41550c698471Sdan 
41560c698471Sdan       sqlite3_step(p->pInsert);
41570c698471Sdan       rc = sqlite3_reset(p->pInsert);
4158d1cccf19Sdan     }
4159d1cccf19Sdan 
416035e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
41610c698471Sdan       rc = sessionConflictHandler(
41620c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
41630c698471Sdan       );
416437f133ecSdan     }
416537f133ecSdan   }
416637f133ecSdan 
416737f133ecSdan   return rc;
416837f133ecSdan }
416937f133ecSdan 
41705f5663dcSdan /*
41715f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument
41725f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke
41735f5663dcSdan ** the conflict handler callback.
41745f5663dcSdan **
41755f5663dcSdan ** The difference between this function and sessionApplyOne() is that this
41765f5663dcSdan ** function handles the case where the conflict-handler is invoked and
41775f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be
41785f5663dcSdan ** retried in some manner.
41795f5663dcSdan */
4180d9151526Sdan static int sessionApplyOneWithRetry(
4181d9151526Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4182d9151526Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator to read change from */
4183d9151526Sdan   SessionApplyCtx *pApply,        /* Apply context */
4184d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4185d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4186d9151526Sdan ){
4187d9151526Sdan   int bReplace = 0;
4188d9151526Sdan   int bRetry = 0;
4189d9151526Sdan   int rc;
4190d9151526Sdan 
4191d9151526Sdan   rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry);
4192a38e6c57Sdan   if( rc==SQLITE_OK ){
41935f5663dcSdan     /* If the bRetry flag is set, the change has not been applied due to an
41945f5663dcSdan     ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and
41955f5663dcSdan     ** a row with the correct PK is present in the db, but one or more other
41965f5663dcSdan     ** fields do not contain the expected values) and the conflict handler
41975f5663dcSdan     ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation,
41985f5663dcSdan     ** but pass NULL as the final argument so that sessionApplyOneOp() ignores
41995f5663dcSdan     ** the SQLITE_CHANGESET_DATA problem.  */
42005f5663dcSdan     if( bRetry ){
42015f5663dcSdan       assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE );
42025f5663dcSdan       rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4203d9151526Sdan     }
4204d9151526Sdan 
42055f5663dcSdan     /* If the bReplace flag is set, the change is an INSERT that has not
42065f5663dcSdan     ** been performed because the database already contains a row with the
42075f5663dcSdan     ** specified primary key and the conflict handler returned
42085f5663dcSdan     ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row
42095f5663dcSdan     ** before reattempting the INSERT.  */
42105f5663dcSdan     else if( bReplace ){
4211d9151526Sdan       assert( pIter->op==SQLITE_INSERT );
4212d9151526Sdan       rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
4213d9151526Sdan       if( rc==SQLITE_OK ){
4214d9151526Sdan         rc = sessionBindRow(pIter,
4215d9151526Sdan             sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete);
4216d9151526Sdan         sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1);
4217d9151526Sdan       }
4218d9151526Sdan       if( rc==SQLITE_OK ){
4219d9151526Sdan         sqlite3_step(pApply->pDelete);
4220d9151526Sdan         rc = sqlite3_reset(pApply->pDelete);
4221d9151526Sdan       }
4222d9151526Sdan       if( rc==SQLITE_OK ){
4223d9151526Sdan         rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4224d9151526Sdan       }
4225d9151526Sdan       if( rc==SQLITE_OK ){
4226d9151526Sdan         rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
4227d9151526Sdan       }
4228d9151526Sdan     }
4229a38e6c57Sdan   }
4230d9151526Sdan 
4231d9151526Sdan   return rc;
4232d9151526Sdan }
4233d9151526Sdan 
4234d9151526Sdan /*
4235d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer.
4236d9151526Sdan */
4237d9151526Sdan static int sessionRetryConstraints(
4238d9151526Sdan   sqlite3 *db,
4239d9151526Sdan   int bPatchset,
4240d9151526Sdan   const char *zTab,
4241d9151526Sdan   SessionApplyCtx *pApply,
4242d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4243d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4244d9151526Sdan ){
4245d9151526Sdan   int rc = SQLITE_OK;
4246d9151526Sdan 
4247d9151526Sdan   while( pApply->constraints.nBuf ){
4248d9151526Sdan     sqlite3_changeset_iter *pIter2 = 0;
4249d9151526Sdan     SessionBuffer cons = pApply->constraints;
4250d9151526Sdan     memset(&pApply->constraints, 0, sizeof(SessionBuffer));
4251d9151526Sdan 
425244748f27Sdan     rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf, 0);
4253d9151526Sdan     if( rc==SQLITE_OK ){
4254f6ad201aSdrh       size_t nByte = 2*pApply->nCol*sizeof(sqlite3_value*);
4255d9151526Sdan       int rc2;
4256d9151526Sdan       pIter2->bPatchset = bPatchset;
4257d9151526Sdan       pIter2->zTab = (char*)zTab;
4258d9151526Sdan       pIter2->nCol = pApply->nCol;
4259d9151526Sdan       pIter2->abPK = pApply->abPK;
4260d9151526Sdan       sessionBufferGrow(&pIter2->tblhdr, nByte, &rc);
4261d9151526Sdan       pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf;
4262d9151526Sdan       if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte);
4263d9151526Sdan 
4264d9151526Sdan       while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){
4265d9151526Sdan         rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx);
4266d9151526Sdan       }
4267d9151526Sdan 
4268d9151526Sdan       rc2 = sqlite3changeset_finalize(pIter2);
42697e0765a9Sdrh       if( rc==SQLITE_OK ) rc = rc2;
4270d9151526Sdan     }
4271d9151526Sdan     assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 );
4272d9151526Sdan 
4273d9151526Sdan     sqlite3_free(cons.aBuf);
4274d9151526Sdan     if( rc!=SQLITE_OK ) break;
4275d9151526Sdan     if( pApply->constraints.nBuf>=cons.nBuf ){
4276d9151526Sdan       /* No progress was made on the last round. */
4277d9151526Sdan       pApply->bDeferConstraints = 0;
4278d9151526Sdan     }
4279d9151526Sdan   }
4280d9151526Sdan 
4281d9151526Sdan   return rc;
4282d9151526Sdan }
4283d9151526Sdan 
4284296c7658Sdan /*
42854757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
42864757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
42874757c658Sdan ** changeset to the main database attached to handle "db". The supplied
42884757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
42894757c658Sdan ** while applying the change.
4290296c7658Sdan */
42914757c658Sdan static int sessionChangesetApply(
4292296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
42934757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
429440368988Sdan   int(*xFilter)(
429540368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
429640368988Sdan     const char *zTab              /* Table name */
429740368988Sdan   ),
4298d5f0767cSdan   int(*xConflict)(
4299d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
4300d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4301d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4302d5f0767cSdan   ),
4303a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4304fe55da38Sdan   void **ppRebase, int *pnRebase, /* OUT: Rebase information */
4305fe55da38Sdan   int flags                       /* SESSION_APPLY_XXX flags */
4306d5f0767cSdan ){
4307ca62ad57Sdan   int schemaMismatch = 0;
4308fe55da38Sdan   int rc = SQLITE_OK;             /* Return code */
4309d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
4310cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
4311296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
43125f5663dcSdan   int bPatchset;
4313d5f0767cSdan 
4314082c96dfSdan   assert( xConflict!=0 );
4315082c96dfSdan 
4316d9151526Sdan   pIter->in.bNoDiscard = 1;
43170c698471Sdan   memset(&sApply, 0, sizeof(sApply));
4318dbe7d37aSdan   sApply.bRebase = (ppRebase && pnRebase);
43194c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
4320fe55da38Sdan   if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){
43210c698471Sdan     rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
4322fe55da38Sdan   }
4323cb3e4b79Sdan   if( rc==SQLITE_OK ){
4324cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
4325cb3e4b79Sdan   }
43260c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
43270c698471Sdan     int nCol;
4328d5f0767cSdan     int op;
43290c698471Sdan     const char *zNew;
4330ca62ad57Sdan 
4331b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
4332d5f0767cSdan 
43330c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
4334ca62ad57Sdan       u8 *abPK;
4335ca62ad57Sdan 
4336d9151526Sdan       rc = sessionRetryConstraints(
4337d9151526Sdan           db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx
4338d9151526Sdan       );
4339d9151526Sdan       if( rc!=SQLITE_OK ) break;
4340d9151526Sdan 
4341cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
43420c698471Sdan       sqlite3_finalize(sApply.pDelete);
43430c698471Sdan       sqlite3_finalize(sApply.pUpdate);
43440c698471Sdan       sqlite3_finalize(sApply.pInsert);
43450c698471Sdan       sqlite3_finalize(sApply.pSelect);
43460c698471Sdan       sApply.db = db;
4347f1b40e83Sdan       sApply.pDelete = 0;
4348f1b40e83Sdan       sApply.pUpdate = 0;
4349f1b40e83Sdan       sApply.pInsert = 0;
4350f1b40e83Sdan       sApply.pSelect = 0;
4351f1b40e83Sdan       sApply.nCol = 0;
4352f1b40e83Sdan       sApply.azCol = 0;
4353f1b40e83Sdan       sApply.abPK = 0;
4354f1b40e83Sdan       sApply.bStat1 = 0;
4355d9151526Sdan       sApply.bDeferConstraints = 1;
4356a38e6c57Sdan       sApply.bRebaseStarted = 0;
4357f1b40e83Sdan       memset(&sApply.constraints, 0, sizeof(SessionBuffer));
435837f133ecSdan 
435940368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
436040368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
436140368988Sdan       ** non-zero, proceed. */
436240368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
436340368988Sdan       if( schemaMismatch ){
436440368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
4365f05ac112Sdan         if( zTab==0 ){
4366f05ac112Sdan           rc = SQLITE_NOMEM;
4367f05ac112Sdan           break;
4368f05ac112Sdan         }
43694f528042Sdan         nTab = (int)strlen(zTab);
437040368988Sdan         sApply.azCol = (const char **)zTab;
437140368988Sdan       }else{
4372ff677b20Sdan         int nMinCol = 0;
4373ff677b20Sdan         int i;
4374ff677b20Sdan 
4375ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
4376296c7658Sdan         rc = sessionTableInfo(
4377ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
4378ca62ad57Sdan         );
4379ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
4380ff677b20Sdan         for(i=0; i<sApply.nCol; i++){
4381ff677b20Sdan           if( sApply.abPK[i] ) nMinCol = i+1;
4382ff677b20Sdan         }
43830c698471Sdan 
4384ca62ad57Sdan         if( sApply.nCol==0 ){
4385ca62ad57Sdan           schemaMismatch = 1;
4386ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4387ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
4388ca62ad57Sdan           );
4389ca62ad57Sdan         }
4390ff677b20Sdan         else if( sApply.nCol<nCol ){
4391ca62ad57Sdan           schemaMismatch = 1;
4392ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4393ff677b20Sdan               "sqlite3changeset_apply(): table %s has %d columns, "
4394ff677b20Sdan               "expected %d or more",
4395ca62ad57Sdan               zTab, sApply.nCol, nCol
4396ca62ad57Sdan           );
4397ca62ad57Sdan         }
4398ff677b20Sdan         else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){
4399ca62ad57Sdan           schemaMismatch = 1;
440040368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
440140368988Sdan               "primary key mismatch for table %s", zTab
4402ca62ad57Sdan           );
4403ca62ad57Sdan         }
4404ff677b20Sdan         else{
4405ff677b20Sdan           sApply.nCol = nCol;
44063739f298Sdan           if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){
44073739f298Sdan             if( (rc = sessionStat1Sql(db, &sApply) ) ){
44083739f298Sdan               break;
44093739f298Sdan             }
4410d1cccf19Sdan             sApply.bStat1 = 1;
4411d1cccf19Sdan           }else{
4412ff677b20Sdan             if((rc = sessionSelectRow(db, zTab, &sApply))
44130c698471Sdan                 || (rc = sessionUpdateRow(db, zTab, &sApply))
44140c698471Sdan                 || (rc = sessionDeleteRow(db, zTab, &sApply))
44150c698471Sdan                 || (rc = sessionInsertRow(db, zTab, &sApply))
441637f133ecSdan               ){
441737f133ecSdan               break;
441837f133ecSdan             }
4419d1cccf19Sdan             sApply.bStat1 = 0;
4420d1cccf19Sdan           }
4421ff677b20Sdan         }
4422cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
4423d5f0767cSdan       }
442440368988Sdan     }
4425d5f0767cSdan 
4426ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
4427ca62ad57Sdan     ** next change. A log message has already been issued. */
4428ca62ad57Sdan     if( schemaMismatch ) continue;
4429ca62ad57Sdan 
4430d9151526Sdan     rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx);
44310c698471Sdan   }
44320c698471Sdan 
44335f5663dcSdan   bPatchset = pIter->bPatchset;
4434296c7658Sdan   if( rc==SQLITE_OK ){
4435296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
4436296c7658Sdan   }else{
4437296c7658Sdan     sqlite3changeset_finalize(pIter);
4438296c7658Sdan   }
4439d5f0767cSdan 
4440d5f0767cSdan   if( rc==SQLITE_OK ){
44415f5663dcSdan     rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx);
44425f5663dcSdan   }
44435f5663dcSdan 
44445f5663dcSdan   if( rc==SQLITE_OK ){
444507001c45Sdrh     int nFk, notUsed;
444607001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
444707001c45Sdrh     if( nFk!=0 ){
4448cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
4449cb3e4b79Sdan       sqlite3_changeset_iter sIter;
4450cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
4451cb3e4b79Sdan       sIter.nCol = nFk;
4452cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
4453cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
4454cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
4455cb3e4b79Sdan       }
4456cb3e4b79Sdan     }
4457cb3e4b79Sdan   }
4458cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
4459cb3e4b79Sdan 
4460fe55da38Sdan   if( (flags & SQLITE_CHANGESETAPPLY_NOSAVEPOINT)==0 ){
4461cb3e4b79Sdan     if( rc==SQLITE_OK ){
4462d5f0767cSdan       rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4463d5f0767cSdan     }else{
4464d5f0767cSdan       sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
4465d5f0767cSdan       sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4466d5f0767cSdan     }
4467fe55da38Sdan   }
4468d5f0767cSdan 
4469dbe7d37aSdan   assert( sApply.bRebase || sApply.rebase.nBuf==0 );
4470dbe7d37aSdan   if( rc==SQLITE_OK && bPatchset==0 && sApply.bRebase ){
4471a38e6c57Sdan     *ppRebase = (void*)sApply.rebase.aBuf;
4472a38e6c57Sdan     *pnRebase = sApply.rebase.nBuf;
4473a38e6c57Sdan     sApply.rebase.aBuf = 0;
4474a38e6c57Sdan   }
44750c698471Sdan   sqlite3_finalize(sApply.pInsert);
44760c698471Sdan   sqlite3_finalize(sApply.pDelete);
44770c698471Sdan   sqlite3_finalize(sApply.pUpdate);
44780c698471Sdan   sqlite3_finalize(sApply.pSelect);
4479cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
4480d9151526Sdan   sqlite3_free((char*)sApply.constraints.aBuf);
4481a38e6c57Sdan   sqlite3_free((char*)sApply.rebase.aBuf);
44824c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
4483d5f0767cSdan   return rc;
4484d5f0767cSdan }
448591ddd559Sdan 
4486b880a7b1Sdan /*
4487b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main
4488b880a7b1Sdan ** database attached to handle "db".
4489b880a7b1Sdan */
4490a38e6c57Sdan int sqlite3changeset_apply_v2(
4491a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4492a38e6c57Sdan   int nChangeset,                 /* Size of changeset in bytes */
4493a38e6c57Sdan   void *pChangeset,               /* Changeset blob */
4494a38e6c57Sdan   int(*xFilter)(
4495a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4496a38e6c57Sdan     const char *zTab              /* Table name */
4497a38e6c57Sdan   ),
4498a38e6c57Sdan   int(*xConflict)(
4499a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4500a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4501a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4502a38e6c57Sdan   ),
4503a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4504fe55da38Sdan   void **ppRebase, int *pnRebase,
4505fe55da38Sdan   int flags
4506a38e6c57Sdan ){
4507a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
450844748f27Sdan   int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT);
450944748f27Sdan   int rc = sessionChangesetStart(&pIter, 0, 0, nChangeset, pChangeset,bInverse);
4510a38e6c57Sdan   if( rc==SQLITE_OK ){
4511a38e6c57Sdan     rc = sessionChangesetApply(
4512fe55da38Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags
4513a38e6c57Sdan     );
4514a38e6c57Sdan   }
4515a38e6c57Sdan   return rc;
4516a38e6c57Sdan }
4517a38e6c57Sdan 
451877fc1d5bSdan /*
45194757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
45204757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
45214757c658Sdan ** to resolve any conflicts encountered while applying the change.
45224757c658Sdan */
45234757c658Sdan int sqlite3changeset_apply(
45244757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
45254757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
45264757c658Sdan   void *pChangeset,               /* Changeset blob */
45274757c658Sdan   int(*xFilter)(
45284757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45294757c658Sdan     const char *zTab              /* Table name */
45304757c658Sdan   ),
45314757c658Sdan   int(*xConflict)(
45324757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
45334757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
45344757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
45354757c658Sdan   ),
45364757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
45374757c658Sdan ){
4538a38e6c57Sdan   return sqlite3changeset_apply_v2(
4539fe55da38Sdan       db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0, 0
4540a38e6c57Sdan   );
45414757c658Sdan }
45424757c658Sdan 
45434757c658Sdan /*
45444757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
45454757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
45464757c658Sdan ** to resolve any conflicts encountered while applying the change.
45474757c658Sdan */
4548a38e6c57Sdan int sqlite3changeset_apply_v2_strm(
4549a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4550a38e6c57Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
4551a38e6c57Sdan   void *pIn,                                          /* First arg for xInput */
4552a38e6c57Sdan   int(*xFilter)(
4553a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4554a38e6c57Sdan     const char *zTab              /* Table name */
4555a38e6c57Sdan   ),
4556a38e6c57Sdan   int(*xConflict)(
4557a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4558a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4559a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4560a38e6c57Sdan   ),
4561a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4562fe55da38Sdan   void **ppRebase, int *pnRebase,
4563fe55da38Sdan   int flags
4564a38e6c57Sdan ){
4565a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
456644748f27Sdan   int bInverse = !!(flags & SQLITE_CHANGESETAPPLY_INVERT);
456744748f27Sdan   int rc = sessionChangesetStart(&pIter, xInput, pIn, 0, 0, bInverse);
4568a38e6c57Sdan   if( rc==SQLITE_OK ){
4569a38e6c57Sdan     rc = sessionChangesetApply(
4570fe55da38Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase, flags
4571a38e6c57Sdan     );
4572a38e6c57Sdan   }
4573a38e6c57Sdan   return rc;
4574a38e6c57Sdan }
4575f1a08ad8Sdrh int sqlite3changeset_apply_strm(
45764757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
45774757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
45784757c658Sdan   void *pIn,                                          /* First arg for xInput */
45794757c658Sdan   int(*xFilter)(
45804757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45814757c658Sdan     const char *zTab              /* Table name */
45824757c658Sdan   ),
45834757c658Sdan   int(*xConflict)(
45844757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45854757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
45864757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
45874757c658Sdan   ),
45884757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
45894757c658Sdan ){
4590a38e6c57Sdan   return sqlite3changeset_apply_v2_strm(
4591fe55da38Sdan       db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0, 0
4592a38e6c57Sdan   );
45934757c658Sdan }
45944757c658Sdan 
45954757c658Sdan /*
45965898ad69Sdan ** sqlite3_changegroup handle.
45975898ad69Sdan */
45985898ad69Sdan struct sqlite3_changegroup {
45995898ad69Sdan   int rc;                         /* Error code */
46005898ad69Sdan   int bPatch;                     /* True to accumulate patchsets */
46015898ad69Sdan   SessionTable *pList;            /* List of tables in current patch */
46025898ad69Sdan };
46035898ad69Sdan 
46045898ad69Sdan /*
460577fc1d5bSdan ** This function is called to merge two changes to the same row together as
460677fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
460777fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
460877fc1d5bSdan */
46095d607a6eSdan static int sessionChangeMerge(
461077fc1d5bSdan   SessionTable *pTab,             /* Table structure */
4611c0a499eaSdan   int bRebase,                    /* True for a rebase hash-table */
461264277f4aSdan   int bPatchset,                  /* True for patchsets */
461377fc1d5bSdan   SessionChange *pExist,          /* Existing change */
461477fc1d5bSdan   int op2,                        /* Second change operation */
461577fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
461677fc1d5bSdan   u8 *aRec,                       /* Second change record */
461777fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
461877fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
46195d607a6eSdan ){
46205d607a6eSdan   SessionChange *pNew = 0;
462124a0c453Sdan   int rc = SQLITE_OK;
46225d607a6eSdan 
46235d607a6eSdan   if( !pExist ){
46242d77d80aSdrh     pNew = (SessionChange *)sqlite3_malloc64(sizeof(SessionChange) + nRec);
46255d607a6eSdan     if( !pNew ){
46265d607a6eSdan       return SQLITE_NOMEM;
46275d607a6eSdan     }
46285d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
4629798693b2Sdan     pNew->op = op2;
46305d607a6eSdan     pNew->bIndirect = bIndirect;
4631cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
463224a0c453Sdan     if( bIndirect==0 || bRebase==0 ){
463324a0c453Sdan       pNew->nRecord = nRec;
4634cbf6d2d2Sdan       memcpy(pNew->aRecord, aRec, nRec);
463524a0c453Sdan     }else{
463624a0c453Sdan       int i;
463724a0c453Sdan       u8 *pIn = aRec;
463824a0c453Sdan       u8 *pOut = pNew->aRecord;
463924a0c453Sdan       for(i=0; i<pTab->nCol; i++){
464024a0c453Sdan         int nIn = sessionSerialLen(pIn);
464124a0c453Sdan         if( *pIn==0 ){
464224a0c453Sdan           *pOut++ = 0;
464324a0c453Sdan         }else if( pTab->abPK[i]==0 ){
464424a0c453Sdan           *pOut++ = 0xFF;
464524a0c453Sdan         }else{
464624a0c453Sdan           memcpy(pOut, pIn, nIn);
464724a0c453Sdan           pOut += nIn;
464824a0c453Sdan         }
464924a0c453Sdan         pIn += nIn;
465024a0c453Sdan       }
465124a0c453Sdan       pNew->nRecord = pOut - pNew->aRecord;
465224a0c453Sdan     }
4653c0a499eaSdan   }else if( bRebase ){
465424a0c453Sdan     if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){
465524a0c453Sdan       *ppNew = pExist;
465624a0c453Sdan     }else{
46572d77d80aSdrh       sqlite3_int64 nByte = nRec + pExist->nRecord + sizeof(SessionChange);
46582d77d80aSdrh       pNew = (SessionChange*)sqlite3_malloc64(nByte);
465924a0c453Sdan       if( pNew==0 ){
466024a0c453Sdan         rc = SQLITE_NOMEM;
466124a0c453Sdan       }else{
466224a0c453Sdan         int i;
466324a0c453Sdan         u8 *a1 = pExist->aRecord;
466424a0c453Sdan         u8 *a2 = aRec;
466524a0c453Sdan         u8 *pOut;
466624a0c453Sdan 
466724a0c453Sdan         memset(pNew, 0, nByte);
466824a0c453Sdan         pNew->bIndirect = bIndirect || pExist->bIndirect;
466924a0c453Sdan         pNew->op = op2;
467024a0c453Sdan         pOut = pNew->aRecord = (u8*)&pNew[1];
467124a0c453Sdan 
467224a0c453Sdan         for(i=0; i<pTab->nCol; i++){
467324a0c453Sdan           int n1 = sessionSerialLen(a1);
467424a0c453Sdan           int n2 = sessionSerialLen(a2);
4675f231e18cSdan           if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){
467624a0c453Sdan             *pOut++ = 0xFF;
467724a0c453Sdan           }else if( *a2==0 ){
467824a0c453Sdan             memcpy(pOut, a1, n1);
467924a0c453Sdan             pOut += n1;
468024a0c453Sdan           }else{
468124a0c453Sdan             memcpy(pOut, a2, n2);
468224a0c453Sdan             pOut += n2;
468324a0c453Sdan           }
468424a0c453Sdan           a1 += n1;
468524a0c453Sdan           a2 += n2;
468624a0c453Sdan         }
468724a0c453Sdan         pNew->nRecord = pOut - pNew->aRecord;
468824a0c453Sdan       }
468924a0c453Sdan       sqlite3_free(pExist);
469024a0c453Sdan     }
46915d607a6eSdan   }else{
4692798693b2Sdan     int op1 = pExist->op;
46935d607a6eSdan 
46945d607a6eSdan     /*
46955d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
46965d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
46975d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
46985d607a6eSdan     **
46995d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
47005d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
47015d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
47025d607a6eSdan     **
47035d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
47045d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
47055d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
47065d607a6eSdan     */
47075d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
47085d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
47095d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
47105d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
47115d607a6eSdan     ){
47125d607a6eSdan       pNew = pExist;
47135d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
47145d607a6eSdan       sqlite3_free(pExist);
47155d607a6eSdan       assert( pNew==0 );
47165d607a6eSdan     }else{
471764277f4aSdan       u8 *aExist = pExist->aRecord;
47182d77d80aSdrh       sqlite3_int64 nByte;
47195d607a6eSdan       u8 *aCsr;
47205d607a6eSdan 
472164277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
472264277f4aSdan       ** buffer of the new object is large enough to hold any record that
472364277f4aSdan       ** may be generated by combining the input records.  */
47245d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
47252d77d80aSdrh       pNew = (SessionChange *)sqlite3_malloc64(nByte);
47265d607a6eSdan       if( !pNew ){
47271756ae10Sdan         sqlite3_free(pExist);
47285d607a6eSdan         return SQLITE_NOMEM;
47295d607a6eSdan       }
47305d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
47315d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
47325d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
47335d607a6eSdan 
4734b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
47355d607a6eSdan         u8 *a1 = aRec;
4736b08a1efaSdan         assert( op2==SQLITE_UPDATE );
4737798693b2Sdan         pNew->op = SQLITE_INSERT;
4738ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
473964277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
4740b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
4741b08a1efaSdan         assert( op2==SQLITE_INSERT );
4742798693b2Sdan         pNew->op = SQLITE_UPDATE;
4743fa29ecc4Sdan         if( bPatchset ){
4744fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
4745fa29ecc4Sdan           aCsr += nRec;
4746fa29ecc4Sdan         }else{
474764277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
4748b08a1efaSdan             sqlite3_free(pNew);
4749b08a1efaSdan             pNew = 0;
47505d607a6eSdan           }
4751fa29ecc4Sdan         }
4752b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
475364277f4aSdan         u8 *a1 = aExist;
47545d607a6eSdan         u8 *a2 = aRec;
4755cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
475664277f4aSdan         if( bPatchset==0 ){
4757ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
4758ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
475964277f4aSdan         }
4760798693b2Sdan         pNew->op = SQLITE_UPDATE;
476164277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
47625d607a6eSdan           sqlite3_free(pNew);
47635d607a6eSdan           pNew = 0;
47645d607a6eSdan         }
4765b08a1efaSdan       }else{                                /* UPDATE + DELETE */
4766b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
4767798693b2Sdan         pNew->op = SQLITE_DELETE;
476864277f4aSdan         if( bPatchset ){
476964277f4aSdan           memcpy(aCsr, aRec, nRec);
477064277f4aSdan           aCsr += nRec;
477164277f4aSdan         }else{
477264277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
477364277f4aSdan         }
47745d607a6eSdan       }
47755d607a6eSdan 
47765d607a6eSdan       if( pNew ){
47774f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
47785d607a6eSdan       }
47795d607a6eSdan       sqlite3_free(pExist);
47805d607a6eSdan     }
47815d607a6eSdan   }
47825d607a6eSdan 
47835d607a6eSdan   *ppNew = pNew;
478424a0c453Sdan   return rc;
47855d607a6eSdan }
47865d607a6eSdan 
478777fc1d5bSdan /*
47885898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as
47895898ad69Sdan ** the first argument to the changegroup hash tables.
479077fc1d5bSdan */
479116228167Sdan static int sessionChangesetToHash(
4792cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
4793c0a499eaSdan   sqlite3_changegroup *pGrp,       /* Changegroup object to add changeset to */
4794c0a499eaSdan   int bRebase                      /* True if hash table is for rebasing */
47955d607a6eSdan ){
47965d607a6eSdan   u8 *aRec;
47975d607a6eSdan   int nRec;
4798cbf6d2d2Sdan   int rc = SQLITE_OK;
47995d607a6eSdan   SessionTable *pTab = 0;
48005d607a6eSdan 
4801c0a499eaSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){
48025d607a6eSdan     const char *zNew;
48035d607a6eSdan     int nCol;
48045d607a6eSdan     int op;
48055d607a6eSdan     int iHash;
48065d607a6eSdan     int bIndirect;
48075d607a6eSdan     SessionChange *pChange;
48085d607a6eSdan     SessionChange *pExist = 0;
48095d607a6eSdan     SessionChange **pp;
48105d607a6eSdan 
48115898ad69Sdan     if( pGrp->pList==0 ){
48125898ad69Sdan       pGrp->bPatch = pIter->bPatchset;
48135898ad69Sdan     }else if( pIter->bPatchset!=pGrp->bPatch ){
48145898ad69Sdan       rc = SQLITE_ERROR;
48155898ad69Sdan       break;
48165898ad69Sdan     }
48175898ad69Sdan 
48185d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
4819ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
48205d607a6eSdan       /* Search the list for a matching table */
48214f528042Sdan       int nNew = (int)strlen(zNew);
4822f29123b5Sdan       u8 *abPK;
4823f29123b5Sdan 
4824f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
48255898ad69Sdan       for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){
48265d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
48275d607a6eSdan       }
48285d607a6eSdan       if( !pTab ){
48296c39e6a8Sdan         SessionTable **ppTab;
48306c39e6a8Sdan 
48312d77d80aSdrh         pTab = sqlite3_malloc64(sizeof(SessionTable) + nCol + nNew+1);
4832f29123b5Sdan         if( !pTab ){
4833f29123b5Sdan           rc = SQLITE_NOMEM;
4834f29123b5Sdan           break;
4835f29123b5Sdan         }
48365d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
4837f29123b5Sdan         pTab->nCol = nCol;
4838ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
4839ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
4840ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
4841ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
48426c39e6a8Sdan 
48436c39e6a8Sdan         /* The new object must be linked on to the end of the list, not
48446c39e6a8Sdan         ** simply added to the start of it. This is to ensure that the
48456c39e6a8Sdan         ** tables within the output of sqlite3changegroup_output() are in
48466c39e6a8Sdan         ** the right order.  */
48476c39e6a8Sdan         for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext);
48486c39e6a8Sdan         *ppTab = pTab;
4849f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
4850f29123b5Sdan         rc = SQLITE_SCHEMA;
4851f29123b5Sdan         break;
48525d607a6eSdan       }
48535d607a6eSdan     }
48545d607a6eSdan 
4855cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
48561756ae10Sdan       rc = SQLITE_NOMEM;
48571756ae10Sdan       break;
48581756ae10Sdan     }
485964277f4aSdan     iHash = sessionChangeHash(
4860cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
486164277f4aSdan     );
48625d607a6eSdan 
48635d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
48645d607a6eSdan     ** Code below may link it back in.
48655d607a6eSdan     */
48665d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
486764277f4aSdan       int bPkOnly1 = 0;
486864277f4aSdan       int bPkOnly2 = 0;
4869cbf6d2d2Sdan       if( pIter->bPatchset ){
487064277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
487164277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
487264277f4aSdan       }
487364277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
48745d607a6eSdan         pExist = *pp;
48755d607a6eSdan         *pp = (*pp)->pNext;
48765d607a6eSdan         pTab->nEntry--;
48775d607a6eSdan         break;
48785d607a6eSdan       }
48795d607a6eSdan     }
48805d607a6eSdan 
4881c0a499eaSdan     rc = sessionChangeMerge(pTab, bRebase,
4882cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
488364277f4aSdan     );
48845d607a6eSdan     if( rc ) break;
48855d607a6eSdan     if( pChange ){
48865d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
48875d607a6eSdan       pTab->apChange[iHash] = pChange;
48885d607a6eSdan       pTab->nEntry++;
48895d607a6eSdan     }
48905d607a6eSdan   }
48915d607a6eSdan 
4892cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
48935d607a6eSdan   return rc;
48945d607a6eSdan }
48955d607a6eSdan 
48965d607a6eSdan /*
48975898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets)
48985898ad69Sdan ** added to the changegroup object passed as the first argument.
48995d607a6eSdan **
49005898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the
49015898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming
49025898ad69Sdan ** interfaces.
49035d607a6eSdan **
49045898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a
49055898ad69Sdan ** buffer containing the output changeset before this function returns. In
49065898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It
49075898ad69Sdan ** is the responsibility of the caller to free the output buffer using
49085898ad69Sdan ** sqlite3_free() when it is no longer required.
49095898ad69Sdan **
49105898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite
49115898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut)
49125898ad69Sdan ** are both set to 0 before returning.
49135d607a6eSdan */
49145898ad69Sdan static int sessionChangegroupOutput(
49155898ad69Sdan   sqlite3_changegroup *pGrp,
4916cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4917cbf6d2d2Sdan   void *pOut,
4918cbf6d2d2Sdan   int *pnOut,
4919cbf6d2d2Sdan   void **ppOut
49205d607a6eSdan ){
49215898ad69Sdan   int rc = SQLITE_OK;
4922e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
49235898ad69Sdan   SessionTable *pTab;
4924cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
49255d607a6eSdan 
49265d607a6eSdan   /* Create the serialized output changeset based on the contents of the
49275898ad69Sdan   ** hash tables attached to the SessionTable objects in list p->pList.
49285d607a6eSdan   */
49295898ad69Sdan   for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
49305d607a6eSdan     int i;
49315d607a6eSdan     if( pTab->nEntry==0 ) continue;
49325d607a6eSdan 
49335898ad69Sdan     sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc);
49345d607a6eSdan     for(i=0; i<pTab->nChange; i++){
49355d607a6eSdan       SessionChange *p;
49365d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
4937798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
49385d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
49395d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
49401f48e67dSdan         if( rc==SQLITE_OK && xOutput && buf.nBuf>=sessions_strm_chunk_size ){
4941cbf6d2d2Sdan           rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4942cbf6d2d2Sdan           buf.nBuf = 0;
4943cbf6d2d2Sdan         }
49445d607a6eSdan       }
4945f8248c42Sdan     }
4946f8248c42Sdan   }
49475d607a6eSdan 
49485d607a6eSdan   if( rc==SQLITE_OK ){
4949cbf6d2d2Sdan     if( xOutput ){
4950cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4951cbf6d2d2Sdan     }else{
49525d607a6eSdan       *ppOut = buf.aBuf;
49535d607a6eSdan       *pnOut = buf.nBuf;
4954cbf6d2d2Sdan       buf.aBuf = 0;
49555d607a6eSdan     }
49565d607a6eSdan   }
4957cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
49585d607a6eSdan 
49595d607a6eSdan   return rc;
49605d607a6eSdan }
49615d607a6eSdan 
4962cbf6d2d2Sdan /*
49635898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup.
49645898ad69Sdan */
49655898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){
49665898ad69Sdan   int rc = SQLITE_OK;             /* Return code */
49675898ad69Sdan   sqlite3_changegroup *p;         /* New object */
49685898ad69Sdan   p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup));
49695898ad69Sdan   if( p==0 ){
49705898ad69Sdan     rc = SQLITE_NOMEM;
49715898ad69Sdan   }else{
49725898ad69Sdan     memset(p, 0, sizeof(sqlite3_changegroup));
49735898ad69Sdan   }
49745898ad69Sdan   *pp = p;
49755898ad69Sdan   return rc;
49765898ad69Sdan }
49775898ad69Sdan 
49785898ad69Sdan /*
49795898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes,
49805898ad69Sdan ** to changeset-group p.
49815898ad69Sdan */
49825898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){
49835898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
49845898ad69Sdan   int rc;                         /* Return code */
49855898ad69Sdan 
49865898ad69Sdan   rc = sqlite3changeset_start(&pIter, nData, pData);
49875898ad69Sdan   if( rc==SQLITE_OK ){
4988c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
49895898ad69Sdan   }
49905898ad69Sdan   sqlite3changeset_finalize(pIter);
49915898ad69Sdan   return rc;
49925898ad69Sdan }
49935898ad69Sdan 
49945898ad69Sdan /*
49955898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation
49965898ad69Sdan ** of all changesets added to the group so far.
49975898ad69Sdan */
49985898ad69Sdan int sqlite3changegroup_output(
49995898ad69Sdan     sqlite3_changegroup *pGrp,
50005898ad69Sdan     int *pnData,
50015898ad69Sdan     void **ppData
50025898ad69Sdan ){
50035898ad69Sdan   return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData);
50045898ad69Sdan }
50055898ad69Sdan 
50065898ad69Sdan /*
50075898ad69Sdan ** Streaming versions of changegroup_add().
50085898ad69Sdan */
50095898ad69Sdan int sqlite3changegroup_add_strm(
50105898ad69Sdan   sqlite3_changegroup *pGrp,
50115898ad69Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
50125898ad69Sdan   void *pIn
50135898ad69Sdan ){
50145898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
50155898ad69Sdan   int rc;                         /* Return code */
50165898ad69Sdan 
50175898ad69Sdan   rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
50185898ad69Sdan   if( rc==SQLITE_OK ){
5019c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
50205898ad69Sdan   }
50215898ad69Sdan   sqlite3changeset_finalize(pIter);
50225898ad69Sdan   return rc;
50235898ad69Sdan }
50245898ad69Sdan 
50255898ad69Sdan /*
50265898ad69Sdan ** Streaming versions of changegroup_output().
50275898ad69Sdan */
50285898ad69Sdan int sqlite3changegroup_output_strm(
50295898ad69Sdan   sqlite3_changegroup *pGrp,
50305898ad69Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
50315898ad69Sdan   void *pOut
50325898ad69Sdan ){
50335898ad69Sdan   return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0);
50345898ad69Sdan }
50355898ad69Sdan 
50365898ad69Sdan /*
50375898ad69Sdan ** Delete a changegroup object.
50385898ad69Sdan */
50395898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){
50405898ad69Sdan   if( pGrp ){
50415898ad69Sdan     sessionDeleteTable(pGrp->pList);
50425898ad69Sdan     sqlite3_free(pGrp);
50435898ad69Sdan   }
50445898ad69Sdan }
50455898ad69Sdan 
50465898ad69Sdan /*
5047cbf6d2d2Sdan ** Combine two changesets together.
5048cbf6d2d2Sdan */
5049cbf6d2d2Sdan int sqlite3changeset_concat(
5050cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
5051cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
5052cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
5053cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
5054cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
5055cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
5056cbf6d2d2Sdan ){
50575898ad69Sdan   sqlite3_changegroup *pGrp;
5058cbf6d2d2Sdan   int rc;
5059cbf6d2d2Sdan 
50605898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
5061cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50625898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nLeft, pLeft);
5063cbf6d2d2Sdan   }
5064cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50655898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nRight, pRight);
5066cbf6d2d2Sdan   }
50675898ad69Sdan   if( rc==SQLITE_OK ){
50685898ad69Sdan     rc = sqlite3changegroup_output(pGrp, pnOut, ppOut);
50695898ad69Sdan   }
50705898ad69Sdan   sqlite3changegroup_delete(pGrp);
5071cbf6d2d2Sdan 
5072cbf6d2d2Sdan   return rc;
5073cbf6d2d2Sdan }
5074cbf6d2d2Sdan 
5075cbf6d2d2Sdan /*
5076cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
5077cbf6d2d2Sdan */
5078f1a08ad8Sdrh int sqlite3changeset_concat_strm(
5079cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
5080cbf6d2d2Sdan   void *pInA,
5081cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
5082cbf6d2d2Sdan   void *pInB,
5083cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5084cbf6d2d2Sdan   void *pOut
5085cbf6d2d2Sdan ){
50865898ad69Sdan   sqlite3_changegroup *pGrp;
5087cbf6d2d2Sdan   int rc;
5088cbf6d2d2Sdan 
50895898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
5090cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50915898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA);
5092cbf6d2d2Sdan   }
5093cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50945898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB);
5095cbf6d2d2Sdan   }
50965898ad69Sdan   if( rc==SQLITE_OK ){
50975898ad69Sdan     rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut);
50985898ad69Sdan   }
50995898ad69Sdan   sqlite3changegroup_delete(pGrp);
5100cbf6d2d2Sdan 
5101cbf6d2d2Sdan   return rc;
5102cbf6d2d2Sdan }
5103cbf6d2d2Sdan 
51043fa5463cSdan /*
51053fa5463cSdan ** Changeset rebaser handle.
51063fa5463cSdan */
5107c0a499eaSdan struct sqlite3_rebaser {
5108c0a499eaSdan   sqlite3_changegroup grp;        /* Hash table */
5109c0a499eaSdan };
5110c0a499eaSdan 
5111c0a499eaSdan /*
5112c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol
5113c0a499eaSdan ** fields in size. This function appends an nCol sessions module
5114b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for
5115b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[].
5116c0a499eaSdan */
5117c0a499eaSdan static void sessionAppendRecordMerge(
5118b880a7b1Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
5119b880a7b1Sdan   int nCol,                       /* Number of columns in each record */
5120b880a7b1Sdan   u8 *a1, int n1,                 /* Record 1 */
5121b880a7b1Sdan   u8 *a2, int n2,                 /* Record 2 */
5122b880a7b1Sdan   int *pRc                        /* IN/OUT: error code */
5123c0a499eaSdan ){
5124c0a499eaSdan   sessionBufferGrow(pBuf, n1+n2, pRc);
5125c0a499eaSdan   if( *pRc==SQLITE_OK ){
5126c0a499eaSdan     int i;
5127c0a499eaSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5128c0a499eaSdan     for(i=0; i<nCol; i++){
5129c0a499eaSdan       int nn1 = sessionSerialLen(a1);
5130c0a499eaSdan       int nn2 = sessionSerialLen(a2);
513124a0c453Sdan       if( *a1==0 || *a1==0xFF ){
5132c0a499eaSdan         memcpy(pOut, a2, nn2);
5133c0a499eaSdan         pOut += nn2;
5134c0a499eaSdan       }else{
5135c0a499eaSdan         memcpy(pOut, a1, nn1);
5136c0a499eaSdan         pOut += nn1;
5137c0a499eaSdan       }
5138f1b40e83Sdan       a1 += nn1;
5139f1b40e83Sdan       a2 += nn2;
5140c0a499eaSdan     }
5141f1b40e83Sdan 
5142f1b40e83Sdan     pBuf->nBuf = pOut-pBuf->aBuf;
5143f1b40e83Sdan     assert( pBuf->nBuf<=pBuf->nAlloc );
5144c0a499eaSdan   }
5145c0a499eaSdan }
5146c0a499eaSdan 
51473fa5463cSdan /*
51483fa5463cSdan ** This function is called when rebasing a local UPDATE change against one
51493fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current
51503fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single
51513fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer
51523fa5463cSdan ** pBuf.
51533fa5463cSdan **
51543fa5463cSdan ** Rebasing the UPDATE involves:
51553fa5463cSdan **
51563fa5463cSdan **   * Removing any changes to fields for which the corresponding field
51573fa5463cSdan **     in the rebase buffer is set to "replaced" (type 0xFF). If this
51583fa5463cSdan **     means the UPDATE change updates no fields, nothing is appended
51593fa5463cSdan **     to the output buffer.
51603fa5463cSdan **
51613fa5463cSdan **   * For each field modified by the local change for which the
51623fa5463cSdan **     corresponding field in the rebase buffer is not "undefined" (0x00)
51633fa5463cSdan **     or "replaced" (0xFF), the old.* value is replaced by the value
51643fa5463cSdan **     in the rebase buffer.
51653fa5463cSdan */
5166f01d3a7eSdan static void sessionAppendPartialUpdate(
516724a0c453Sdan   SessionBuffer *pBuf,            /* Append record here */
516824a0c453Sdan   sqlite3_changeset_iter *pIter,  /* Iterator pointed at local change */
516924a0c453Sdan   u8 *aRec, int nRec,             /* Local change */
517024a0c453Sdan   u8 *aChange, int nChange,       /* Record to rebase against */
517124a0c453Sdan   int *pRc                        /* IN/OUT: Return Code */
5172f01d3a7eSdan ){
5173f01d3a7eSdan   sessionBufferGrow(pBuf, 2+nRec+nChange, pRc);
5174f01d3a7eSdan   if( *pRc==SQLITE_OK ){
5175f01d3a7eSdan     int bData = 0;
5176f01d3a7eSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5177f01d3a7eSdan     int i;
5178f01d3a7eSdan     u8 *a1 = aRec;
5179f01d3a7eSdan     u8 *a2 = aChange;
5180f01d3a7eSdan 
5181f01d3a7eSdan     *pOut++ = SQLITE_UPDATE;
5182f01d3a7eSdan     *pOut++ = pIter->bIndirect;
5183f01d3a7eSdan     for(i=0; i<pIter->nCol; i++){
5184f01d3a7eSdan       int n1 = sessionSerialLen(a1);
5185f01d3a7eSdan       int n2 = sessionSerialLen(a2);
5186f01d3a7eSdan       if( pIter->abPK[i] || a2[0]==0 ){
5187f01d3a7eSdan         if( !pIter->abPK[i] ) bData = 1;
5188f01d3a7eSdan         memcpy(pOut, a1, n1);
5189f01d3a7eSdan         pOut += n1;
519024a0c453Sdan       }else if( a2[0]!=0xFF ){
519124a0c453Sdan         bData = 1;
519224a0c453Sdan         memcpy(pOut, a2, n2);
519324a0c453Sdan         pOut += n2;
5194f01d3a7eSdan       }else{
5195f01d3a7eSdan         *pOut++ = '\0';
5196f01d3a7eSdan       }
5197f01d3a7eSdan       a1 += n1;
5198f01d3a7eSdan       a2 += n2;
5199f01d3a7eSdan     }
5200f01d3a7eSdan     if( bData ){
5201f01d3a7eSdan       a2 = aChange;
5202f01d3a7eSdan       for(i=0; i<pIter->nCol; i++){
5203f01d3a7eSdan         int n1 = sessionSerialLen(a1);
5204f01d3a7eSdan         int n2 = sessionSerialLen(a2);
520524a0c453Sdan         if( pIter->abPK[i] || a2[0]!=0xFF ){
5206f01d3a7eSdan           memcpy(pOut, a1, n1);
5207f01d3a7eSdan           pOut += n1;
5208f01d3a7eSdan         }else{
5209f01d3a7eSdan           *pOut++ = '\0';
5210f01d3a7eSdan         }
5211f01d3a7eSdan         a1 += n1;
5212f01d3a7eSdan         a2 += n2;
5213f01d3a7eSdan       }
5214f01d3a7eSdan       pBuf->nBuf = (pOut - pBuf->aBuf);
5215f01d3a7eSdan     }
5216f01d3a7eSdan   }
5217f01d3a7eSdan }
5218f01d3a7eSdan 
52193fa5463cSdan /*
52203fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases
52213fa5463cSdan ** that changeset according to the current configuration of the rebaser
52223fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput
52233fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking
52243fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL,
52253fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset
52263fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of
52273fa5463cSdan ** the buffer in bytes.  It is the responsibility of the caller to eventually
52283fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free().
52293fa5463cSdan **
52303fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and
52313fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before
52323fa5463cSdan ** returning.
52333fa5463cSdan */
5234c0a499eaSdan static int sessionRebase(
5235c0a499eaSdan   sqlite3_rebaser *p,             /* Rebaser hash table */
5236c0a499eaSdan   sqlite3_changeset_iter *pIter,  /* Input data */
5237c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5238c0a499eaSdan   void *pOut,                     /* Context for xOutput callback */
5239c0a499eaSdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
5240c0a499eaSdan   void **ppOut                    /* OUT: Inverse of pChangeset */
5241c0a499eaSdan ){
5242c0a499eaSdan   int rc = SQLITE_OK;
5243c0a499eaSdan   u8 *aRec = 0;
5244c0a499eaSdan   int nRec = 0;
5245c0a499eaSdan   int bNew = 0;
5246c0a499eaSdan   SessionTable *pTab = 0;
5247c0a499eaSdan   SessionBuffer sOut = {0,0,0};
5248c0a499eaSdan 
5249f1b40e83Sdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){
5250c0a499eaSdan     SessionChange *pChange = 0;
5251f01d3a7eSdan     int bDone = 0;
5252c0a499eaSdan 
5253c0a499eaSdan     if( bNew ){
5254c0a499eaSdan       const char *zTab = pIter->zTab;
5255c0a499eaSdan       for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){
5256c0a499eaSdan         if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break;
5257c0a499eaSdan       }
5258c0a499eaSdan       bNew = 0;
5259c0a499eaSdan 
526095ccb6dcSdan       /* A patchset may not be rebased */
526195ccb6dcSdan       if( pIter->bPatchset ){
526295ccb6dcSdan         rc = SQLITE_ERROR;
526395ccb6dcSdan       }
526495ccb6dcSdan 
5265c0a499eaSdan       /* Append a table header to the output for this new table */
5266c0a499eaSdan       sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc);
5267c0a499eaSdan       sessionAppendVarint(&sOut, pIter->nCol, &rc);
5268c0a499eaSdan       sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc);
52692c42c34dSmistachkin       sessionAppendBlob(&sOut,(u8*)pIter->zTab,(int)strlen(pIter->zTab)+1,&rc);
5270c0a499eaSdan     }
5271c0a499eaSdan 
52728cb83beeSdan     if( pTab && rc==SQLITE_OK ){
52738cb83beeSdan       int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange);
5274c0a499eaSdan 
5275c0a499eaSdan       for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){
52768cb83beeSdan         if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){
5277c0a499eaSdan           break;
5278c0a499eaSdan         }
5279c0a499eaSdan       }
5280c0a499eaSdan     }
5281c0a499eaSdan 
5282c0a499eaSdan     if( pChange ){
5283c0a499eaSdan       assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT );
5284f01d3a7eSdan       switch( pIter->op ){
5285f01d3a7eSdan         case SQLITE_INSERT:
5286f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5287f01d3a7eSdan             bDone = 1;
5288f01d3a7eSdan             if( pChange->bIndirect==0 ){
5289f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_UPDATE, &rc);
5290f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5291f01d3a7eSdan               sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc);
5292f01d3a7eSdan               sessionAppendBlob(&sOut, aRec, nRec, &rc);
5293f01d3a7eSdan             }
5294f01d3a7eSdan           }
5295f01d3a7eSdan           break;
5296f01d3a7eSdan 
5297f01d3a7eSdan         case SQLITE_UPDATE:
5298f01d3a7eSdan           bDone = 1;
5299f01d3a7eSdan           if( pChange->op==SQLITE_DELETE ){
5300f01d3a7eSdan             if( pChange->bIndirect==0 ){
5301f01d3a7eSdan               u8 *pCsr = aRec;
5302f01d3a7eSdan               sessionSkipRecord(&pCsr, pIter->nCol);
5303f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_INSERT, &rc);
5304f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5305b880a7b1Sdan               sessionAppendRecordMerge(&sOut, pIter->nCol,
5306f01d3a7eSdan                   pCsr, nRec-(pCsr-aRec),
5307f01d3a7eSdan                   pChange->aRecord, pChange->nRecord, &rc
5308f01d3a7eSdan               );
5309f01d3a7eSdan             }
5310f01d3a7eSdan           }else{
5311f01d3a7eSdan             sessionAppendPartialUpdate(&sOut, pIter,
5312f01d3a7eSdan                 aRec, nRec, pChange->aRecord, pChange->nRecord, &rc
5313f01d3a7eSdan             );
5314f01d3a7eSdan           }
5315f01d3a7eSdan           break;
5316f01d3a7eSdan 
5317f01d3a7eSdan         default:
5318f01d3a7eSdan           assert( pIter->op==SQLITE_DELETE );
5319f01d3a7eSdan           bDone = 1;
5320f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5321f01d3a7eSdan             sessionAppendByte(&sOut, SQLITE_DELETE, &rc);
5322f01d3a7eSdan             sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5323b880a7b1Sdan             sessionAppendRecordMerge(&sOut, pIter->nCol,
5324f01d3a7eSdan                 pChange->aRecord, pChange->nRecord, aRec, nRec, &rc
5325f01d3a7eSdan             );
5326f01d3a7eSdan           }
5327f01d3a7eSdan           break;
5328f01d3a7eSdan       }
5329f01d3a7eSdan     }
5330f01d3a7eSdan 
5331f01d3a7eSdan     if( bDone==0 ){
5332f01d3a7eSdan       sessionAppendByte(&sOut, pIter->op, &rc);
5333f01d3a7eSdan       sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5334f01d3a7eSdan       sessionAppendBlob(&sOut, aRec, nRec, &rc);
5335f01d3a7eSdan     }
53361f48e67dSdan     if( rc==SQLITE_OK && xOutput && sOut.nBuf>sessions_strm_chunk_size ){
5337c0a499eaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5338c0a499eaSdan       sOut.nBuf = 0;
5339c0a499eaSdan     }
5340c0a499eaSdan     if( rc ) break;
5341c0a499eaSdan   }
5342c0a499eaSdan 
5343c0a499eaSdan   if( rc!=SQLITE_OK ){
5344c0a499eaSdan     sqlite3_free(sOut.aBuf);
5345c0a499eaSdan     memset(&sOut, 0, sizeof(sOut));
5346c0a499eaSdan   }
5347c0a499eaSdan 
5348c0a499eaSdan   if( rc==SQLITE_OK ){
5349c0a499eaSdan     if( xOutput ){
5350c0a499eaSdan       if( sOut.nBuf>0 ){
5351c0a499eaSdan         rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5352c0a499eaSdan       }
5353c0a499eaSdan     }else{
5354c0a499eaSdan       *ppOut = (void*)sOut.aBuf;
5355c0a499eaSdan       *pnOut = sOut.nBuf;
5356c0a499eaSdan       sOut.aBuf = 0;
5357c0a499eaSdan     }
5358c0a499eaSdan   }
5359c0a499eaSdan   sqlite3_free(sOut.aBuf);
5360c0a499eaSdan   return rc;
5361c0a499eaSdan }
5362c0a499eaSdan 
5363c0a499eaSdan /*
5364c0a499eaSdan ** Create a new rebaser object.
5365c0a499eaSdan */
5366c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){
5367c0a499eaSdan   int rc = SQLITE_OK;
5368c0a499eaSdan   sqlite3_rebaser *pNew;
5369c0a499eaSdan 
5370c0a499eaSdan   pNew = sqlite3_malloc(sizeof(sqlite3_rebaser));
5371c0a499eaSdan   if( pNew==0 ){
5372c0a499eaSdan     rc = SQLITE_NOMEM;
5373f1b40e83Sdan   }else{
5374f1b40e83Sdan     memset(pNew, 0, sizeof(sqlite3_rebaser));
5375c0a499eaSdan   }
5376c0a499eaSdan   *ppNew = pNew;
5377c0a499eaSdan   return rc;
5378c0a499eaSdan }
5379c0a499eaSdan 
5380c0a499eaSdan /*
5381c0a499eaSdan ** Call this one or more times to configure a rebaser.
5382c0a499eaSdan */
5383c0a499eaSdan int sqlite3rebaser_configure(
5384c0a499eaSdan   sqlite3_rebaser *p,
5385c0a499eaSdan   int nRebase, const void *pRebase
5386c0a499eaSdan ){
5387c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator opened on pData/nData */
5388c0a499eaSdan   int rc;                              /* Return code */
5389c0a499eaSdan   rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase);
5390c0a499eaSdan   if( rc==SQLITE_OK ){
5391c0a499eaSdan     rc = sessionChangesetToHash(pIter, &p->grp, 1);
5392c0a499eaSdan   }
5393c0a499eaSdan   sqlite3changeset_finalize(pIter);
5394c0a499eaSdan   return rc;
5395c0a499eaSdan }
5396c0a499eaSdan 
5397c0a499eaSdan /*
5398c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5399c0a499eaSdan */
5400c0a499eaSdan int sqlite3rebaser_rebase(
5401c0a499eaSdan   sqlite3_rebaser *p,
5402c0a499eaSdan   int nIn, const void *pIn,
5403c0a499eaSdan   int *pnOut, void **ppOut
5404c0a499eaSdan ){
5405c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5406c0a499eaSdan   int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn);
5407c0a499eaSdan 
5408c0a499eaSdan   if( rc==SQLITE_OK ){
5409c0a499eaSdan     rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut);
5410c0a499eaSdan     sqlite3changeset_finalize(pIter);
5411c0a499eaSdan   }
5412c0a499eaSdan 
5413c0a499eaSdan   return rc;
5414c0a499eaSdan }
5415c0a499eaSdan 
5416c0a499eaSdan /*
5417c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5418c0a499eaSdan */
5419c0a499eaSdan int sqlite3rebaser_rebase_strm(
5420c0a499eaSdan   sqlite3_rebaser *p,
5421c0a499eaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
5422c0a499eaSdan   void *pIn,
5423c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5424c0a499eaSdan   void *pOut
5425c0a499eaSdan ){
5426c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5427c0a499eaSdan   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
5428c0a499eaSdan 
5429c0a499eaSdan   if( rc==SQLITE_OK ){
5430c0a499eaSdan     rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0);
5431c0a499eaSdan     sqlite3changeset_finalize(pIter);
5432c0a499eaSdan   }
5433c0a499eaSdan 
5434c0a499eaSdan   return rc;
5435c0a499eaSdan }
5436c0a499eaSdan 
5437c0a499eaSdan /*
5438c0a499eaSdan ** Destroy a rebaser object
5439c0a499eaSdan */
5440f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){
5441c0a499eaSdan   if( p ){
5442c0a499eaSdan     sessionDeleteTable(p->grp.pList);
5443c0a499eaSdan     sqlite3_free(p);
5444c0a499eaSdan   }
5445c0a499eaSdan }
5446c0a499eaSdan 
54471f48e67dSdan /*
54481f48e67dSdan ** Global configuration
54491f48e67dSdan */
54501f48e67dSdan int sqlite3session_config(int op, void *pArg){
54511f48e67dSdan   int rc = SQLITE_OK;
54521f48e67dSdan   switch( op ){
54531f48e67dSdan     case SQLITE_SESSION_CONFIG_STRMSIZE: {
54541f48e67dSdan       int *pInt = (int*)pArg;
54551f48e67dSdan       if( *pInt>0 ){
54561f48e67dSdan         sessions_strm_chunk_size = *pInt;
54571f48e67dSdan       }
54581f48e67dSdan       *pInt = sessions_strm_chunk_size;
54591f48e67dSdan       break;
54601f48e67dSdan     }
54611f48e67dSdan     default:
54621f48e67dSdan       rc = SQLITE_MISUSE;
54631f48e67dSdan       break;
54641f48e67dSdan   }
54651f48e67dSdan   return rc;
54661f48e67dSdan }
54671f48e67dSdan 
54689b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
5469