14fccf43aSdan 
29b1c62d4Sdrh #if defined(SQLITE_ENABLE_SESSION) && defined(SQLITE_ENABLE_PREUPDATE_HOOK)
34fccf43aSdan #include "sqlite3session.h"
44fccf43aSdan #include <assert.h>
54fccf43aSdan #include <string.h>
64fccf43aSdan 
75d8a2984Sdan #ifndef SQLITE_AMALGAMATION
84fccf43aSdan # include "sqliteInt.h"
94fccf43aSdan # include "vdbeInt.h"
105d8a2984Sdan #endif
114fccf43aSdan 
124fccf43aSdan typedef struct SessionTable SessionTable;
134fccf43aSdan typedef struct SessionChange SessionChange;
14296c7658Sdan typedef struct SessionBuffer SessionBuffer;
15ef7a6304Sdan typedef struct SessionInput SessionInput;
16ef7a6304Sdan 
17ef7a6304Sdan /*
18ef7a6304Sdan ** Minimum chunk size used by streaming versions of functions.
19ef7a6304Sdan */
20f1a08ad8Sdrh #ifndef SESSIONS_STRM_CHUNK_SIZE
214757c658Sdan # ifdef SQLITE_TEST
22f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 64
234757c658Sdan # else
24f1a08ad8Sdrh #   define SESSIONS_STRM_CHUNK_SIZE 1024
25f1a08ad8Sdrh # endif
264757c658Sdan #endif
274fccf43aSdan 
28cf8e9144Sdan typedef struct SessionHook SessionHook;
29cf8e9144Sdan struct SessionHook {
30cf8e9144Sdan   void *pCtx;
31cf8e9144Sdan   int (*xOld)(void*,int,sqlite3_value**);
32cf8e9144Sdan   int (*xNew)(void*,int,sqlite3_value**);
33cf8e9144Sdan   int (*xCount)(void*);
34cf8e9144Sdan   int (*xDepth)(void*);
35cf8e9144Sdan };
36cf8e9144Sdan 
37296c7658Sdan /*
38296c7658Sdan ** Session handle structure.
39296c7658Sdan */
404fccf43aSdan struct sqlite3_session {
414fccf43aSdan   sqlite3 *db;                    /* Database handle session is attached to */
424fccf43aSdan   char *zDb;                      /* Name of database session is attached to */
43296c7658Sdan   int bEnable;                    /* True if currently recording */
44b4480e94Sdan   int bIndirect;                  /* True if all changes are indirect */
45ff4d0f41Sdan   int bAutoAttach;                /* True to auto-attach tables */
464fccf43aSdan   int rc;                         /* Non-zero if an error has occurred */
477531a5a3Sdan   void *pFilterCtx;               /* First argument to pass to xTableFilter */
487531a5a3Sdan   int (*xTableFilter)(void *pCtx, const char *zTab);
491611e5a3Sdan   sqlite3_value *pZeroBlob;       /* Value containing X'' */
504fccf43aSdan   sqlite3_session *pNext;         /* Next session object on same db. */
514fccf43aSdan   SessionTable *pTable;           /* List of attached tables */
52cf8e9144Sdan   SessionHook hook;               /* APIs to grab new and old data with */
534fccf43aSdan };
544fccf43aSdan 
554fccf43aSdan /*
56ef7a6304Sdan ** Instances of this structure are used to build strings or binary records.
57ef7a6304Sdan */
58ef7a6304Sdan struct SessionBuffer {
59ef7a6304Sdan   u8 *aBuf;                       /* Pointer to changeset buffer */
60ef7a6304Sdan   int nBuf;                       /* Size of buffer aBuf */
61ef7a6304Sdan   int nAlloc;                     /* Size of allocation containing aBuf */
62ef7a6304Sdan };
63ef7a6304Sdan 
64ef7a6304Sdan /*
6516228167Sdan ** An object of this type is used internally as an abstraction for
6616228167Sdan ** input data. Input data may be supplied either as a single large buffer
6716228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g.
68f1a08ad8Sdrh **  sqlite3changeset_start_strm()).
69ef7a6304Sdan */
70ef7a6304Sdan struct SessionInput {
71d9151526Sdan   int bNoDiscard;                 /* If true, discard no data */
72d9151526Sdan   int iCurrent;                   /* Offset in aData[] of current change */
734757c658Sdan   int iNext;                      /* Offset in aData[] of next change */
744757c658Sdan   u8 *aData;                      /* Pointer to buffer containing changeset */
754757c658Sdan   int nData;                      /* Number of bytes in aData */
764757c658Sdan 
77ef7a6304Sdan   SessionBuffer buf;              /* Current read buffer */
78ef7a6304Sdan   int (*xInput)(void*, void*, int*);        /* Input stream call (or NULL) */
79ef7a6304Sdan   void *pIn;                                /* First argument to xInput */
80ef7a6304Sdan   int bEof;                       /* Set to true after xInput finished */
81ef7a6304Sdan };
82ef7a6304Sdan 
83ef7a6304Sdan /*
84296c7658Sdan ** Structure for changeset iterators.
85296c7658Sdan */
86296c7658Sdan struct sqlite3_changeset_iter {
87ef7a6304Sdan   SessionInput in;                /* Input buffer or stream */
88ef7a6304Sdan   SessionBuffer tblhdr;           /* Buffer to hold apValue/zTab/abPK/ */
8973b3c055Sdan   int bPatchset;                  /* True if this is a patchset */
90296c7658Sdan   int rc;                         /* Iterator error code */
91296c7658Sdan   sqlite3_stmt *pConflict;        /* Points to conflicting row, if any */
92296c7658Sdan   char *zTab;                     /* Current table */
93296c7658Sdan   int nCol;                       /* Number of columns in zTab */
94296c7658Sdan   int op;                         /* Current operation */
95b4480e94Sdan   int bIndirect;                  /* True if current change was indirect */
96244593c8Sdan   u8 *abPK;                       /* Primary key array */
97296c7658Sdan   sqlite3_value **apValue;        /* old.* and new.* values */
98296c7658Sdan };
99296c7658Sdan 
100296c7658Sdan /*
1014fccf43aSdan ** Each session object maintains a set of the following structures, one
1024fccf43aSdan ** for each table the session object is monitoring. The structures are
1034fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable.
1044fccf43aSdan **
1054fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have
1064fccf43aSdan ** been modified in any way since the session object was attached to the
1074fccf43aSdan ** table.
1084fccf43aSdan **
1094fccf43aSdan ** The data associated with each hash-table entry is a structure containing
1104fccf43aSdan ** a subset of the initial values that the modified row contained at the
1114fccf43aSdan ** start of the session. Or no initial values if the row was inserted.
1124fccf43aSdan */
1134fccf43aSdan struct SessionTable {
1144fccf43aSdan   SessionTable *pNext;
1154fccf43aSdan   char *zName;                    /* Local name of table */
1164fccf43aSdan   int nCol;                       /* Number of columns in table zName */
1173739f298Sdan   int bStat1;                     /* True if this is sqlite_stat1 */
118e8d5648eSdan   const char **azCol;             /* Column names */
119e8d5648eSdan   u8 *abPK;                       /* Array of primary key flags */
120296c7658Sdan   int nEntry;                     /* Total number of entries in hash table */
1214fccf43aSdan   int nChange;                    /* Size of apChange[] array */
1224fccf43aSdan   SessionChange **apChange;       /* Hash table buckets */
1234fccf43aSdan };
1244fccf43aSdan 
1254fccf43aSdan /*
1264fccf43aSdan ** RECORD FORMAT:
1274fccf43aSdan **
1284fccf43aSdan ** The following record format is similar to (but not compatible with) that
1294fccf43aSdan ** used in SQLite database files. This format is used as part of the
1304fccf43aSdan ** change-set binary format, and so must be architecture independent.
1314fccf43aSdan **
1324fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained -
1334fccf43aSdan ** there is no separation of header and data. Each field begins with a
1344fccf43aSdan ** single byte describing its type, as follows:
1354fccf43aSdan **
1364fccf43aSdan **       0x00: Undefined value.
1374fccf43aSdan **       0x01: Integer value.
1384fccf43aSdan **       0x02: Real value.
1394fccf43aSdan **       0x03: Text value.
1404fccf43aSdan **       0x04: Blob value.
1414fccf43aSdan **       0x05: SQL NULL value.
1424fccf43aSdan **
1434fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT
1444fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists
1454fccf43aSdan ** only of the single type byte. For other types of values, the type byte
1464fccf43aSdan ** is followed by:
1474fccf43aSdan **
1484fccf43aSdan **   Text values:
1494fccf43aSdan **     A varint containing the number of bytes in the value (encoded using
1504fccf43aSdan **     UTF-8). Followed by a buffer containing the UTF-8 representation
1514fccf43aSdan **     of the text value. There is no nul terminator.
1524fccf43aSdan **
1534fccf43aSdan **   Blob values:
1544fccf43aSdan **     A varint containing the number of bytes in the value, followed by
1554fccf43aSdan **     a buffer containing the value itself.
1564fccf43aSdan **
1574fccf43aSdan **   Integer values:
1584fccf43aSdan **     An 8-byte big-endian integer value.
1594fccf43aSdan **
1604fccf43aSdan **   Real values:
1614fccf43aSdan **     An 8-byte big-endian IEEE 754-2008 real value.
1624fccf43aSdan **
1634fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite
1644fccf43aSdan ** record format.
1654fccf43aSdan **
1664fccf43aSdan ** CHANGESET FORMAT:
1674fccf43aSdan **
1684fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on
1694fccf43aSdan ** one or more tables. Operations on a single table are grouped together,
1704fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all
1714fccf43aSdan ** mixed together).
1724fccf43aSdan **
1734fccf43aSdan ** Each group of changes begins with a table header:
1744fccf43aSdan **
1754fccf43aSdan **   1 byte: Constant 0x54 (capital 'T')
176730bb805Sdan **   Varint: Number of columns in the table.
17773b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
1784fccf43aSdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
1794fccf43aSdan **
1804fccf43aSdan ** Followed by one or more changes to the table.
1814fccf43aSdan **
182c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
1835d607a6eSdan **   1 byte: The "indirect-change" flag.
1844fccf43aSdan **   old.* record: (delete and update only)
1854fccf43aSdan **   new.* record: (insert and update only)
18673b3c055Sdan **
187730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the
188730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of
189730bb805Sdan ** columns in the table. The i'th field of each record is associated with
190730bb805Sdan ** the i'th column of the table, counting from left to right in the order
191730bb805Sdan ** in which columns were declared in the CREATE TABLE statement.
192730bb805Sdan **
193730bb805Sdan ** The new.* record that is part of each INSERT change contains the values
194730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each
195730bb805Sdan ** DELETE change contains the values that made up the row that was deleted
196730bb805Sdan ** from the database. In the changeset format, the records that are part
197730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00)
198730bb805Sdan ** fields.
199730bb805Sdan **
200730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields
201730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are
202730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields
203730bb805Sdan ** are set to the values that made up the row before the UPDATE that the
204730bb805Sdan ** change records took place. Within the new.* record, fields associated
205730bb805Sdan ** with table columns modified by the UPDATE change contain the new
206730bb805Sdan ** values. Fields associated with table columns that are not modified
207730bb805Sdan ** are set to "undefined".
208730bb805Sdan **
20973b3c055Sdan ** PATCHSET FORMAT:
21073b3c055Sdan **
21173b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset,
212730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict
213730bb805Sdan ** resolution is required when applying the changeset.
21473b3c055Sdan **
21573b3c055Sdan ** Each group of changes begins with a table header:
21673b3c055Sdan **
21773b3c055Sdan **   1 byte: Constant 0x50 (capital 'P')
218730bb805Sdan **   Varint: Number of columns in the table.
21973b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
22073b3c055Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
22173b3c055Sdan **
22273b3c055Sdan ** Followed by one or more changes to the table.
22373b3c055Sdan **
224c8be6437Sdrh **   1 byte: Either SQLITE_INSERT (0x12), UPDATE (0x17) or DELETE (0x09).
22573b3c055Sdan **   1 byte: The "indirect-change" flag.
226730bb805Sdan **   single record: (PK fields for DELETE, PK and modified fields for UPDATE,
227730bb805Sdan **                   full record for INSERT).
228730bb805Sdan **
229730bb805Sdan ** As in the changeset format, each field of the single record that is part
230730bb805Sdan ** of a patchset change is associated with the correspondingly positioned
231730bb805Sdan ** table column, counting from left to right within the CREATE TABLE
232730bb805Sdan ** statement.
233730bb805Sdan **
234730bb805Sdan ** For a DELETE change, all fields within the record except those associated
235f01d3a7eSdan ** with PRIMARY KEY columns are omitted. The PRIMARY KEY fields contain the
236f01d3a7eSdan ** values identifying the row to delete.
237730bb805Sdan **
238730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY
239730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined".
240730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update,
241730bb805Sdan ** and fields associated with modified columns contain the new column values.
242730bb805Sdan **
243730bb805Sdan ** The records associated with INSERT changes are in the same format as for
244730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT
245730bb805Sdan ** change to contain a field set to "undefined".
2464fccf43aSdan */
2474fccf43aSdan 
2484fccf43aSdan /*
2494fccf43aSdan ** For each row modified during a session, there exists a single instance of
2504fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table.
2514fccf43aSdan */
2524fccf43aSdan struct SessionChange {
253798693b2Sdan   int op;                         /* One of UPDATE, DELETE, INSERT */
254b4480e94Sdan   int bIndirect;                  /* True if this change is "indirect" */
2554fccf43aSdan   int nRecord;                    /* Number of bytes in buffer aRecord[] */
2564fccf43aSdan   u8 *aRecord;                    /* Buffer containing old.* record */
2574fccf43aSdan   SessionChange *pNext;           /* For hash-table collisions */
2584fccf43aSdan };
2594fccf43aSdan 
260296c7658Sdan /*
261296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the
262296c7658Sdan ** number of bytes written.
263296c7658Sdan */
264296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){
265296c7658Sdan   return putVarint32(aBuf, iVal);
2664fccf43aSdan }
2674fccf43aSdan 
268296c7658Sdan /*
269296c7658Sdan ** Return the number of bytes required to store value iVal as a varint.
270296c7658Sdan */
271296c7658Sdan static int sessionVarintLen(int iVal){
272296c7658Sdan   return sqlite3VarintLen(iVal);
273296c7658Sdan }
274296c7658Sdan 
275296c7658Sdan /*
276296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of
277296c7658Sdan ** bytes read.
278296c7658Sdan */
2794fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){
280296c7658Sdan   return getVarint32(aBuf, *piVal);
2814fccf43aSdan }
2824fccf43aSdan 
28348cd59a5Sdrh /* Load an unaligned and unsigned 32-bit integer */
28448cd59a5Sdrh #define SESSION_UINT32(x) (((u32)(x)[0]<<24)|((x)[1]<<16)|((x)[2]<<8)|(x)[3])
28548cd59a5Sdrh 
286296c7658Sdan /*
287296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return
288296c7658Sdan ** the value read.
289296c7658Sdan */
2904fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){
29148cd59a5Sdrh   u64 x = SESSION_UINT32(aRec);
29248cd59a5Sdrh   u32 y = SESSION_UINT32(aRec+4);
29348cd59a5Sdrh   x = (x<<32) + y;
29448cd59a5Sdrh   return (sqlite3_int64)x;
2954fccf43aSdan }
2964fccf43aSdan 
2974fccf43aSdan /*
298296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[].
299296c7658Sdan */
300296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){
301296c7658Sdan   aBuf[0] = (i>>56) & 0xFF;
302296c7658Sdan   aBuf[1] = (i>>48) & 0xFF;
303296c7658Sdan   aBuf[2] = (i>>40) & 0xFF;
304296c7658Sdan   aBuf[3] = (i>>32) & 0xFF;
305296c7658Sdan   aBuf[4] = (i>>24) & 0xFF;
306296c7658Sdan   aBuf[5] = (i>>16) & 0xFF;
307296c7658Sdan   aBuf[6] = (i>> 8) & 0xFF;
308296c7658Sdan   aBuf[7] = (i>> 0) & 0xFF;
309296c7658Sdan }
310296c7658Sdan 
311296c7658Sdan /*
3124fccf43aSdan ** This function is used to serialize the contents of value pValue (see
3134fccf43aSdan ** comment titled "RECORD FORMAT" above).
3144fccf43aSdan **
3154fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to
3164fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before
3174fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is
3184fccf43aSdan ** set *pnWrite.
3194fccf43aSdan **
3204fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs
3214fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16))
3224fccf43aSdan ** SQLITE_NOMEM is returned.
3234fccf43aSdan */
3244fccf43aSdan static int sessionSerializeValue(
3254fccf43aSdan   u8 *aBuf,                       /* If non-NULL, write serialized value here */
3264fccf43aSdan   sqlite3_value *pValue,          /* Value to serialize */
3274fccf43aSdan   int *pnWrite                    /* IN/OUT: Increment by bytes written */
3284fccf43aSdan ){
329296c7658Sdan   int nByte;                      /* Size of serialized value in bytes */
3304fccf43aSdan 
33180fe2d93Sdan   if( pValue ){
33280fe2d93Sdan     int eType;                    /* Value type (SQLITE_NULL, TEXT etc.) */
33380fe2d93Sdan 
3344fccf43aSdan     eType = sqlite3_value_type(pValue);
3354fccf43aSdan     if( aBuf ) aBuf[0] = eType;
3364fccf43aSdan 
3374fccf43aSdan     switch( eType ){
3384fccf43aSdan       case SQLITE_NULL:
3394fccf43aSdan         nByte = 1;
3404fccf43aSdan         break;
3414fccf43aSdan 
3424fccf43aSdan       case SQLITE_INTEGER:
3434fccf43aSdan       case SQLITE_FLOAT:
3444fccf43aSdan         if( aBuf ){
3454fccf43aSdan           /* TODO: SQLite does something special to deal with mixed-endian
3464fccf43aSdan           ** floating point values (e.g. ARM7). This code probably should
3474fccf43aSdan           ** too.  */
3484fccf43aSdan           u64 i;
3494fccf43aSdan           if( eType==SQLITE_INTEGER ){
3504fccf43aSdan             i = (u64)sqlite3_value_int64(pValue);
3514fccf43aSdan           }else{
3524fccf43aSdan             double r;
3534fccf43aSdan             assert( sizeof(double)==8 && sizeof(u64)==8 );
3544fccf43aSdan             r = sqlite3_value_double(pValue);
3554fccf43aSdan             memcpy(&i, &r, 8);
3564fccf43aSdan           }
357296c7658Sdan           sessionPutI64(&aBuf[1], i);
3584fccf43aSdan         }
3594fccf43aSdan         nByte = 9;
3604fccf43aSdan         break;
3614fccf43aSdan 
3624e895da1Sdan       default: {
36380fe2d93Sdan         u8 *z;
36480fe2d93Sdan         int n;
36580fe2d93Sdan         int nVarint;
36680fe2d93Sdan 
3674e895da1Sdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
36880fe2d93Sdan         if( eType==SQLITE_TEXT ){
36980fe2d93Sdan           z = (u8 *)sqlite3_value_text(pValue);
37080fe2d93Sdan         }else{
37180fe2d93Sdan           z = (u8 *)sqlite3_value_blob(pValue);
37280fe2d93Sdan         }
37380fe2d93Sdan         n = sqlite3_value_bytes(pValue);
3743cc89d95Sdan         if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
37580fe2d93Sdan         nVarint = sessionVarintLen(n);
37680fe2d93Sdan 
3774fccf43aSdan         if( aBuf ){
3784fccf43aSdan           sessionVarintPut(&aBuf[1], n);
379895decf6Sdan           if( n ) memcpy(&aBuf[nVarint + 1], z, n);
3804fccf43aSdan         }
3814fccf43aSdan 
3824fccf43aSdan         nByte = 1 + nVarint + n;
3834fccf43aSdan         break;
3844fccf43aSdan       }
3854fccf43aSdan     }
38680fe2d93Sdan   }else{
38780fe2d93Sdan     nByte = 1;
38880fe2d93Sdan     if( aBuf ) aBuf[0] = '\0';
38980fe2d93Sdan   }
3904fccf43aSdan 
391fa122adaSdan   if( pnWrite ) *pnWrite += nByte;
3924fccf43aSdan   return SQLITE_OK;
3934fccf43aSdan }
3944fccf43aSdan 
395fa122adaSdan 
396798693b2Sdan /*
397798693b2Sdan ** This macro is used to calculate hash key values for data structures. In
398798693b2Sdan ** order to use this macro, the entire data structure must be represented
399798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value
400798693b2Sdan ** for a data structure represented as three such integers, the macro may
401798693b2Sdan ** then be used as follows:
402798693b2Sdan **
403798693b2Sdan **    int hash_key_value;
404798693b2Sdan **    hash_key_value = HASH_APPEND(0, <value 1>);
405798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 2>);
406798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 3>);
407798693b2Sdan **
408798693b2Sdan ** In practice, the data structures this macro is used for are the primary
409798693b2Sdan ** key values of modified rows.
410798693b2Sdan */
4114131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add)
412798693b2Sdan 
413798693b2Sdan /*
414798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the
415798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
416798693b2Sdan */
4174131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){
418e8d5648eSdan   h = HASH_APPEND(h, i & 0xFFFFFFFF);
419e8d5648eSdan   return HASH_APPEND(h, (i>>32)&0xFFFFFFFF);
420e8d5648eSdan }
421798693b2Sdan 
422798693b2Sdan /*
423798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to
424798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value.
425798693b2Sdan */
4264131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){
427e8d5648eSdan   int i;
428e8d5648eSdan   for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]);
429e8d5648eSdan   return h;
430e8d5648eSdan }
431e8d5648eSdan 
4324fccf43aSdan /*
433798693b2Sdan ** Append the hash of the data type passed as the second argument to the
434798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
435798693b2Sdan */
436798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){
437798693b2Sdan   return HASH_APPEND(h, eType);
438798693b2Sdan }
439798693b2Sdan 
440798693b2Sdan /*
4414131639cSdan ** This function may only be called from within a pre-update callback.
4424131639cSdan ** It calculates a hash based on the primary key values of the old.* or
443798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to
444798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL
445798693b2Sdan ** values, *pbNullPK is set to true before returning.
446798693b2Sdan **
447798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values
448798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned
449798693b2Sdan ** and the output variables are set as described above.
4504fccf43aSdan */
451798693b2Sdan static int sessionPreupdateHash(
452cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns pTab */
453e8d5648eSdan   SessionTable *pTab,             /* Session table handle */
454e8d5648eSdan   int bNew,                       /* True to hash the new.* PK */
45527453faeSdan   int *piHash,                    /* OUT: Hash value */
456798693b2Sdan   int *pbNullPK                   /* OUT: True if there are NULL values in PK */
457e8d5648eSdan ){
4584131639cSdan   unsigned int h = 0;             /* Hash value to return */
4594131639cSdan   int i;                          /* Used to iterate through columns */
460e8d5648eSdan 
46127453faeSdan   assert( *pbNullPK==0 );
462cf8e9144Sdan   assert( pTab->nCol==pSession->hook.xCount(pSession->hook.pCtx) );
463e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
464e8d5648eSdan     if( pTab->abPK[i] ){
465e8d5648eSdan       int rc;
466e8d5648eSdan       int eType;
467e8d5648eSdan       sqlite3_value *pVal;
468e8d5648eSdan 
469e8d5648eSdan       if( bNew ){
470cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, i, &pVal);
471e8d5648eSdan       }else{
472cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, i, &pVal);
473e8d5648eSdan       }
47412ca0b56Sdan       if( rc!=SQLITE_OK ) return rc;
475e8d5648eSdan 
476e8d5648eSdan       eType = sqlite3_value_type(pVal);
477798693b2Sdan       h = sessionHashAppendType(h, eType);
4786734007dSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
479e8d5648eSdan         i64 iVal;
480e8d5648eSdan         if( eType==SQLITE_INTEGER ){
481e8d5648eSdan           iVal = sqlite3_value_int64(pVal);
482e8d5648eSdan         }else{
483e8d5648eSdan           double rVal = sqlite3_value_double(pVal);
484e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
485e8d5648eSdan           memcpy(&iVal, &rVal, 8);
486e8d5648eSdan         }
487e8d5648eSdan         h = sessionHashAppendI64(h, iVal);
4886734007dSdan       }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
4896734007dSdan         const u8 *z;
4903cc89d95Sdan         int n;
4916734007dSdan         if( eType==SQLITE_TEXT ){
4926734007dSdan           z = (const u8 *)sqlite3_value_text(pVal);
4936734007dSdan         }else{
4946734007dSdan           z = (const u8 *)sqlite3_value_blob(pVal);
495e8d5648eSdan         }
4963cc89d95Sdan         n = sqlite3_value_bytes(pVal);
4973cc89d95Sdan         if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
4983cc89d95Sdan         h = sessionHashAppendBlob(h, n, z);
4996734007dSdan       }else{
50027453faeSdan         assert( eType==SQLITE_NULL );
5011611e5a3Sdan         assert( pTab->bStat1==0 || i!=1 );
50227453faeSdan         *pbNullPK = 1;
503e8d5648eSdan       }
504e8d5648eSdan     }
505e8d5648eSdan   }
506e8d5648eSdan 
507e8d5648eSdan   *piHash = (h % pTab->nChange);
508e8d5648eSdan   return SQLITE_OK;
509e8d5648eSdan }
510e8d5648eSdan 
5114131639cSdan /*
5126cda207fSdan ** The buffer that the argument points to contains a serialized SQL value.
5136cda207fSdan ** Return the number of bytes of space occupied by the value (including
5146cda207fSdan ** the type byte).
5156cda207fSdan */
5166cda207fSdan static int sessionSerialLen(u8 *a){
5176cda207fSdan   int e = *a;
5186cda207fSdan   int n;
51924a0c453Sdan   if( e==0 || e==0xFF ) return 1;
5206cda207fSdan   if( e==SQLITE_NULL ) return 1;
5216cda207fSdan   if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9;
5226cda207fSdan   return sessionVarintGet(&a[1], &n) + 1 + n;
5236cda207fSdan }
5246cda207fSdan 
5256cda207fSdan /*
5265d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a
527798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys
5284131639cSdan ** calculated by this function are compatible with those calculated by
5294131639cSdan ** sessionPreupdateHash().
53064277f4aSdan **
53164277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from
53264277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely.
5334131639cSdan */
5344131639cSdan static unsigned int sessionChangeHash(
5354131639cSdan   SessionTable *pTab,             /* Table handle */
53664277f4aSdan   int bPkOnly,                    /* Record consists of PK fields only */
5375d607a6eSdan   u8 *aRecord,                    /* Change record */
5384131639cSdan   int nBucket                     /* Assume this many buckets in hash table */
539e8d5648eSdan ){
5404131639cSdan   unsigned int h = 0;             /* Value to return */
5414131639cSdan   int i;                          /* Used to iterate through columns */
5425d607a6eSdan   u8 *a = aRecord;                /* Used to iterate through change record */
543e8d5648eSdan 
544e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
5456cda207fSdan     int eType = *a;
546e8d5648eSdan     int isPK = pTab->abPK[i];
54764277f4aSdan     if( bPkOnly && isPK==0 ) continue;
548e8d5648eSdan 
54927453faeSdan     /* It is not possible for eType to be SQLITE_NULL here. The session
55027453faeSdan     ** module does not record changes for rows with NULL values stored in
55127453faeSdan     ** primary key columns. */
55227453faeSdan     assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
55327453faeSdan          || eType==SQLITE_TEXT || eType==SQLITE_BLOB
5546cda207fSdan          || eType==SQLITE_NULL || eType==0
55527453faeSdan     );
5561611e5a3Sdan     assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) );
55727453faeSdan 
5586cda207fSdan     if( isPK ){
5596cda207fSdan       a++;
560798693b2Sdan       h = sessionHashAppendType(h, eType);
56127453faeSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
5626734007dSdan         h = sessionHashAppendI64(h, sessionGetI64(a));
563e8d5648eSdan         a += 8;
5641611e5a3Sdan       }else{
565e8d5648eSdan         int n;
566e8d5648eSdan         a += sessionVarintGet(a, &n);
5676734007dSdan         h = sessionHashAppendBlob(h, n, a);
568e8d5648eSdan         a += n;
569e8d5648eSdan       }
5706cda207fSdan     }else{
5716cda207fSdan       a += sessionSerialLen(a);
5726cda207fSdan     }
573e8d5648eSdan   }
574e8d5648eSdan   return (h % nBucket);
575e8d5648eSdan }
576e8d5648eSdan 
577798693b2Sdan /*
578798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab.
579798693b2Sdan ** This function returns true if the two records apply to the same row (i.e.
580798693b2Sdan ** have the same values stored in the primary key columns), or false
581798693b2Sdan ** otherwise.
582798693b2Sdan */
5835d607a6eSdan static int sessionChangeEqual(
584798693b2Sdan   SessionTable *pTab,             /* Table used for PK definition */
585a71d2371Sdan   int bLeftPkOnly,                /* True if aLeft[] contains PK fields only */
5865d607a6eSdan   u8 *aLeft,                      /* Change record */
587a71d2371Sdan   int bRightPkOnly,               /* True if aRight[] contains PK fields only */
5885d607a6eSdan   u8 *aRight                      /* Change record */
5895d607a6eSdan ){
590798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor to iterate through aLeft */
591798693b2Sdan   u8 *a2 = aRight;                /* Cursor to iterate through aRight */
592798693b2Sdan   int iCol;                       /* Used to iterate through table columns */
5935d607a6eSdan 
594798693b2Sdan   for(iCol=0; iCol<pTab->nCol; iCol++){
59514faa061Sdan     if( pTab->abPK[iCol] ){
5965d607a6eSdan       int n1 = sessionSerialLen(a1);
5975d607a6eSdan       int n2 = sessionSerialLen(a2);
5985d607a6eSdan 
599bd45374cSdan       if( n1!=n2 || memcmp(a1, a2, n1) ){
6005d607a6eSdan         return 0;
6015d607a6eSdan       }
60214faa061Sdan       a1 += n1;
60314faa061Sdan       a2 += n2;
60414faa061Sdan     }else{
60514faa061Sdan       if( bLeftPkOnly==0 ) a1 += sessionSerialLen(a1);
60614faa061Sdan       if( bRightPkOnly==0 ) a2 += sessionSerialLen(a2);
60714faa061Sdan     }
6085d607a6eSdan   }
6095d607a6eSdan 
6105d607a6eSdan   return 1;
6115d607a6eSdan }
6125d607a6eSdan 
613798693b2Sdan /*
614798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change
615798693b2Sdan ** records with nCol columns. This function "merges" the two records into
616798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is
617798693b2Sdan ** then set to point to one byte after the last byte written before
618798693b2Sdan ** returning.
619798693b2Sdan **
620798693b2Sdan ** The merging of records is done as follows: For each column, if the
621798693b2Sdan ** aRight record contains a value for the column, copy the value from
622798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither
623798693b2Sdan ** record contains a value for a given column, then neither does the
624798693b2Sdan ** output record.
625798693b2Sdan */
6265d607a6eSdan static void sessionMergeRecord(
6275d607a6eSdan   u8 **paOut,
628798693b2Sdan   int nCol,
6295d607a6eSdan   u8 *aLeft,
6305d607a6eSdan   u8 *aRight
6315d607a6eSdan ){
632798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor used to iterate through aLeft */
633798693b2Sdan   u8 *a2 = aRight;                /* Cursor used to iterate through aRight */
634798693b2Sdan   u8 *aOut = *paOut;              /* Output cursor */
635798693b2Sdan   int iCol;                       /* Used to iterate from 0 to nCol */
6365d607a6eSdan 
637798693b2Sdan   for(iCol=0; iCol<nCol; iCol++){
6385d607a6eSdan     int n1 = sessionSerialLen(a1);
6395d607a6eSdan     int n2 = sessionSerialLen(a2);
6405d607a6eSdan     if( *a2 ){
6415d607a6eSdan       memcpy(aOut, a2, n2);
6425d607a6eSdan       aOut += n2;
6435d607a6eSdan     }else{
6445d607a6eSdan       memcpy(aOut, a1, n1);
6455d607a6eSdan       aOut += n1;
6465d607a6eSdan     }
6475d607a6eSdan     a1 += n1;
6485d607a6eSdan     a2 += n2;
6495d607a6eSdan   }
6505d607a6eSdan 
6515d607a6eSdan   *paOut = aOut;
6525d607a6eSdan }
6535d607a6eSdan 
654798693b2Sdan /*
655798693b2Sdan ** This is a helper function used by sessionMergeUpdate().
656798693b2Sdan **
657798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value
658798693b2Sdan ** within a change record. Before it returns, both have been advanced so
659798693b2Sdan ** as to point to the next value in the record.
660798693b2Sdan **
661798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e.
662fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo
663798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the
664798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal
665798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points
666fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words:
667fa29ecc4Sdan **
668fa29ecc4Sdan **   if( *paTwo is valid ) return *paTwo;
669fa29ecc4Sdan **   return *paOne;
670fa29ecc4Sdan **
671798693b2Sdan */
6725d607a6eSdan static u8 *sessionMergeValue(
673798693b2Sdan   u8 **paOne,                     /* IN/OUT: Left-hand buffer pointer */
674798693b2Sdan   u8 **paTwo,                     /* IN/OUT: Right-hand buffer pointer */
675798693b2Sdan   int *pnVal                      /* OUT: Bytes in returned value */
6765d607a6eSdan ){
6775d607a6eSdan   u8 *a1 = *paOne;
6785d607a6eSdan   u8 *a2 = *paTwo;
6795d607a6eSdan   u8 *pRet = 0;
6805d607a6eSdan   int n1;
6815d607a6eSdan 
6825d607a6eSdan   assert( a1 );
6835d607a6eSdan   if( a2 ){
6845d607a6eSdan     int n2 = sessionSerialLen(a2);
6855d607a6eSdan     if( *a2 ){
6865d607a6eSdan       *pnVal = n2;
6875d607a6eSdan       pRet = a2;
6885d607a6eSdan     }
6895d607a6eSdan     *paTwo = &a2[n2];
6905d607a6eSdan   }
6915d607a6eSdan 
6925d607a6eSdan   n1 = sessionSerialLen(a1);
6935d607a6eSdan   if( pRet==0 ){
6945d607a6eSdan     *pnVal = n1;
6955d607a6eSdan     pRet = a1;
6965d607a6eSdan   }
6975d607a6eSdan   *paOne = &a1[n1];
6985d607a6eSdan 
6995d607a6eSdan   return pRet;
7005d607a6eSdan }
7015d607a6eSdan 
702798693b2Sdan /*
703798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes
704798693b2Sdan ** on the same row.
705798693b2Sdan */
7065d607a6eSdan static int sessionMergeUpdate(
707798693b2Sdan   u8 **paOut,                     /* IN/OUT: Pointer to output buffer */
708798693b2Sdan   SessionTable *pTab,             /* Table change pertains to */
709a71d2371Sdan   int bPatchset,                  /* True if records are patchset records */
710798693b2Sdan   u8 *aOldRecord1,                /* old.* record for first change */
711798693b2Sdan   u8 *aOldRecord2,                /* old.* record for second change */
712798693b2Sdan   u8 *aNewRecord1,                /* new.* record for first change */
713798693b2Sdan   u8 *aNewRecord2                 /* new.* record for second change */
7145d607a6eSdan ){
7155d607a6eSdan   u8 *aOld1 = aOldRecord1;
7165d607a6eSdan   u8 *aOld2 = aOldRecord2;
7175d607a6eSdan   u8 *aNew1 = aNewRecord1;
7185d607a6eSdan   u8 *aNew2 = aNewRecord2;
7195d607a6eSdan 
7205d607a6eSdan   u8 *aOut = *paOut;
7215d607a6eSdan   int i;
72264277f4aSdan 
72364277f4aSdan   if( bPatchset==0 ){
7245d607a6eSdan     int bRequired = 0;
7255d607a6eSdan 
7265d607a6eSdan     assert( aOldRecord1 && aNewRecord1 );
7275d607a6eSdan 
7285d607a6eSdan     /* Write the old.* vector first. */
7295d607a6eSdan     for(i=0; i<pTab->nCol; i++){
7305d607a6eSdan       int nOld;
7315d607a6eSdan       u8 *aOld;
7325d607a6eSdan       int nNew;
7335d607a6eSdan       u8 *aNew;
7345d607a6eSdan 
7355d607a6eSdan       aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7365d607a6eSdan       aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
7375d607a6eSdan       if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){
7385d607a6eSdan         if( pTab->abPK[i]==0 ) bRequired = 1;
7395d607a6eSdan         memcpy(aOut, aOld, nOld);
7405d607a6eSdan         aOut += nOld;
7415d607a6eSdan       }else{
7425d607a6eSdan         *(aOut++) = '\0';
7435d607a6eSdan       }
7445d607a6eSdan     }
7455d607a6eSdan 
7465d607a6eSdan     if( !bRequired ) return 0;
74764277f4aSdan   }
7485d607a6eSdan 
7495d607a6eSdan   /* Write the new.* vector */
7505d607a6eSdan   aOld1 = aOldRecord1;
7515d607a6eSdan   aOld2 = aOldRecord2;
7525d607a6eSdan   aNew1 = aNewRecord1;
7535d607a6eSdan   aNew2 = aNewRecord2;
7545d607a6eSdan   for(i=0; i<pTab->nCol; i++){
7555d607a6eSdan     int nOld;
7565d607a6eSdan     u8 *aOld;
7575d607a6eSdan     int nNew;
7585d607a6eSdan     u8 *aNew;
7595d607a6eSdan 
7605d607a6eSdan     aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7615d607a6eSdan     aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
76264277f4aSdan     if( bPatchset==0
76364277f4aSdan      && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew)))
76464277f4aSdan     ){
7655d607a6eSdan       *(aOut++) = '\0';
7665d607a6eSdan     }else{
7675d607a6eSdan       memcpy(aOut, aNew, nNew);
7685d607a6eSdan       aOut += nNew;
7695d607a6eSdan     }
7705d607a6eSdan   }
7715d607a6eSdan 
7725d607a6eSdan   *paOut = aOut;
7735d607a6eSdan   return 1;
7745d607a6eSdan }
7755d607a6eSdan 
77677fc1d5bSdan /*
77777fc1d5bSdan ** This function is only called from within a pre-update-hook callback.
77877fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row
77977fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise
78077fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns
78177fc1d5bSdan ** false.
78277fc1d5bSdan */
78377fc1d5bSdan static int sessionPreupdateEqual(
784cf8e9144Sdan   sqlite3_session *pSession,      /* Session object that owns SessionTable */
78577fc1d5bSdan   SessionTable *pTab,             /* Table associated with change */
78677fc1d5bSdan   SessionChange *pChange,         /* Change to compare to */
78777fc1d5bSdan   int op                          /* Current pre-update operation */
788e8d5648eSdan ){
78977fc1d5bSdan   int iCol;                       /* Used to iterate through columns */
79077fc1d5bSdan   u8 *a = pChange->aRecord;       /* Cursor used to scan change record */
791e8d5648eSdan 
79277fc1d5bSdan   assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
79377fc1d5bSdan   for(iCol=0; iCol<pTab->nCol; iCol++){
79477fc1d5bSdan     if( !pTab->abPK[iCol] ){
795798693b2Sdan       a += sessionSerialLen(a);
796e8d5648eSdan     }else{
7976734007dSdan       sqlite3_value *pVal;        /* Value returned by preupdate_new/old */
7986734007dSdan       int rc;                     /* Error code from preupdate_new/old */
799798693b2Sdan       int eType = *a++;           /* Type of value from change record */
8006734007dSdan 
8016734007dSdan       /* The following calls to preupdate_new() and preupdate_old() can not
8026734007dSdan       ** fail. This is because they cache their return values, and by the
8036734007dSdan       ** time control flows to here they have already been called once from
8046734007dSdan       ** within sessionPreupdateHash(). The first two asserts below verify
8056734007dSdan       ** this (that the method has already been called). */
80677fc1d5bSdan       if( op==SQLITE_INSERT ){
807cf8e9144Sdan         /* assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew ); */
808cf8e9144Sdan         rc = pSession->hook.xNew(pSession->hook.pCtx, iCol, &pVal);
809e8d5648eSdan       }else{
810cf8e9144Sdan         /* assert( db->pPreUpdate->pUnpacked ); */
811cf8e9144Sdan         rc = pSession->hook.xOld(pSession->hook.pCtx, iCol, &pVal);
812e8d5648eSdan       }
8136734007dSdan       assert( rc==SQLITE_OK );
8141611e5a3Sdan       if( sqlite3_value_type(pVal)!=eType ) return 0;
815e8d5648eSdan 
81612ca0b56Sdan       /* A SessionChange object never has a NULL value in a PK column */
81712ca0b56Sdan       assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
81812ca0b56Sdan            || eType==SQLITE_BLOB    || eType==SQLITE_TEXT
81912ca0b56Sdan       );
82012ca0b56Sdan 
82112ca0b56Sdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
822e8d5648eSdan         i64 iVal = sessionGetI64(a);
823e8d5648eSdan         a += 8;
824e8d5648eSdan         if( eType==SQLITE_INTEGER ){
82577fc1d5bSdan           if( sqlite3_value_int64(pVal)!=iVal ) return 0;
826e8d5648eSdan         }else{
827e8d5648eSdan           double rVal;
828e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
829e8d5648eSdan           memcpy(&rVal, &iVal, 8);
83077fc1d5bSdan           if( sqlite3_value_double(pVal)!=rVal ) return 0;
831e8d5648eSdan         }
83212ca0b56Sdan       }else{
833e8d5648eSdan         int n;
834e8d5648eSdan         const u8 *z;
835e8d5648eSdan         a += sessionVarintGet(a, &n);
83677fc1d5bSdan         if( sqlite3_value_bytes(pVal)!=n ) return 0;
83712ca0b56Sdan         if( eType==SQLITE_TEXT ){
83812ca0b56Sdan           z = sqlite3_value_text(pVal);
83912ca0b56Sdan         }else{
84012ca0b56Sdan           z = sqlite3_value_blob(pVal);
84112ca0b56Sdan         }
842*3f2cebb6Sdan         if( n>0 && memcmp(a, z, n) ) return 0;
843e8d5648eSdan         a += n;
844e8d5648eSdan       }
845e8d5648eSdan     }
846e8d5648eSdan   }
847e8d5648eSdan 
84877fc1d5bSdan   return 1;
8494fccf43aSdan }
8504fccf43aSdan 
8514fccf43aSdan /*
8524fccf43aSdan ** If required, grow the hash table used to store changes on table pTab
8534fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the
8544fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return
8554fccf43aSdan ** SQLITE_OK.
8564fccf43aSdan **
8574fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In
8584fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway.
8594fccf43aSdan ** Growing the hash table in this case is a performance optimization only,
8604fccf43aSdan ** it is not required for correct operation.
8614fccf43aSdan */
86264277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){
8634fccf43aSdan   if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){
8644fccf43aSdan     int i;
8654fccf43aSdan     SessionChange **apNew;
8664fccf43aSdan     int nNew = (pTab->nChange ? pTab->nChange : 128) * 2;
8674fccf43aSdan 
8684fccf43aSdan     apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew);
8694fccf43aSdan     if( apNew==0 ){
8704fccf43aSdan       if( pTab->nChange==0 ){
8714fccf43aSdan         return SQLITE_ERROR;
8724fccf43aSdan       }
8734fccf43aSdan       return SQLITE_OK;
8744fccf43aSdan     }
8754fccf43aSdan     memset(apNew, 0, sizeof(SessionChange *) * nNew);
8764fccf43aSdan 
8774fccf43aSdan     for(i=0; i<pTab->nChange; i++){
8784fccf43aSdan       SessionChange *p;
8794fccf43aSdan       SessionChange *pNext;
8804fccf43aSdan       for(p=pTab->apChange[i]; p; p=pNext){
88164277f4aSdan         int bPkOnly = (p->op==SQLITE_DELETE && bPatchset);
88264277f4aSdan         int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew);
8834fccf43aSdan         pNext = p->pNext;
8844fccf43aSdan         p->pNext = apNew[iHash];
8854fccf43aSdan         apNew[iHash] = p;
8864fccf43aSdan       }
8874fccf43aSdan     }
8884fccf43aSdan 
8894fccf43aSdan     sqlite3_free(pTab->apChange);
8904fccf43aSdan     pTab->nChange = nNew;
8914fccf43aSdan     pTab->apChange = apNew;
8924fccf43aSdan   }
8934fccf43aSdan 
8944fccf43aSdan   return SQLITE_OK;
8954fccf43aSdan }
8964fccf43aSdan 
897296c7658Sdan /*
898e8d5648eSdan ** This function queries the database for the names of the columns of table
8993f975373Sdrh ** zThis, in schema zDb.
900e8d5648eSdan **
90177fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number
90277fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a
903e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to
904e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not
905e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding
906e8d5648eSdan ** column is part of the primary key.
907e8d5648eSdan **
908e8d5648eSdan ** For example, if the table is declared as:
909e8d5648eSdan **
910e8d5648eSdan **     CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z));
911e8d5648eSdan **
91277fc1d5bSdan ** Then the four output variables are populated as follows:
913e8d5648eSdan **
91477fc1d5bSdan **     *pnCol  = 4
915e8d5648eSdan **     *pzTab  = "tbl1"
916e8d5648eSdan **     *pazCol = {"w", "x", "y", "z"}
917e8d5648eSdan **     *pabPK  = {1, 0, 0, 1}
918e8d5648eSdan **
919e8d5648eSdan ** All returned buffers are part of the same single allocation, which must
9203f975373Sdrh ** be freed using sqlite3_free() by the caller
921e8d5648eSdan */
922e8d5648eSdan static int sessionTableInfo(
923e8d5648eSdan   sqlite3 *db,                    /* Database connection */
924e8d5648eSdan   const char *zDb,                /* Name of attached database (e.g. "main") */
925e8d5648eSdan   const char *zThis,              /* Table name */
926ca62ad57Sdan   int *pnCol,                     /* OUT: number of columns */
927e8d5648eSdan   const char **pzTab,             /* OUT: Copy of zThis */
928e8d5648eSdan   const char ***pazCol,           /* OUT: Array of column names for table */
929e8d5648eSdan   u8 **pabPK                      /* OUT: Array of booleans - true for PK col */
930e8d5648eSdan ){
931e8d5648eSdan   char *zPragma;
932e8d5648eSdan   sqlite3_stmt *pStmt;
933e8d5648eSdan   int rc;
934e8d5648eSdan   int nByte;
935e8d5648eSdan   int nDbCol = 0;
936e8d5648eSdan   int nThis;
937e8d5648eSdan   int i;
93874f598b6Smistachkin   u8 *pAlloc = 0;
939db04571cSdan   char **azCol = 0;
94074f598b6Smistachkin   u8 *abPK = 0;
941e8d5648eSdan 
942db04571cSdan   assert( pazCol && pabPK );
943e8d5648eSdan 
944cfdbde21Sdrh   nThis = sqlite3Strlen30(zThis);
945614efe2bSdan   if( nThis==12 && 0==sqlite3_stricmp("sqlite_stat1", zThis) ){
94658713184Sdan     rc = sqlite3_table_column_metadata(db, zDb, zThis, 0, 0, 0, 0, 0, 0);
94758713184Sdan     if( rc==SQLITE_OK ){
948614efe2bSdan       /* For sqlite_stat1, pretend that (tbl,idx) is the PRIMARY KEY. */
949614efe2bSdan       zPragma = sqlite3_mprintf(
950614efe2bSdan           "SELECT 0, 'tbl',  '', 0, '', 1     UNION ALL "
951614efe2bSdan           "SELECT 1, 'idx',  '', 0, '', 2     UNION ALL "
952614efe2bSdan           "SELECT 2, 'stat', '', 0, '', 0"
953614efe2bSdan       );
95458713184Sdan     }else if( rc==SQLITE_ERROR ){
95558713184Sdan       zPragma = sqlite3_mprintf("");
95658713184Sdan     }else{
95758713184Sdan       return rc;
95858713184Sdan     }
959614efe2bSdan   }else{
960e8d5648eSdan     zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis);
961614efe2bSdan   }
962e8d5648eSdan   if( !zPragma ) return SQLITE_NOMEM;
963e8d5648eSdan 
964e8d5648eSdan   rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0);
965e8d5648eSdan   sqlite3_free(zPragma);
966e8d5648eSdan   if( rc!=SQLITE_OK ) return rc;
967e8d5648eSdan 
968e8d5648eSdan   nByte = nThis + 1;
969e8d5648eSdan   while( SQLITE_ROW==sqlite3_step(pStmt) ){
970e8d5648eSdan     nByte += sqlite3_column_bytes(pStmt, 1);
971e8d5648eSdan     nDbCol++;
972e8d5648eSdan   }
973e8d5648eSdan   rc = sqlite3_reset(pStmt);
974e8d5648eSdan 
975e8d5648eSdan   if( rc==SQLITE_OK ){
976e8d5648eSdan     nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1);
977e8d5648eSdan     pAlloc = sqlite3_malloc(nByte);
978e8d5648eSdan     if( pAlloc==0 ){
979e8d5648eSdan       rc = SQLITE_NOMEM;
980e8d5648eSdan     }
981e8d5648eSdan   }
982e8d5648eSdan   if( rc==SQLITE_OK ){
983e8d5648eSdan     azCol = (char **)pAlloc;
984ca62ad57Sdan     pAlloc = (u8 *)&azCol[nDbCol];
985e8d5648eSdan     abPK = (u8 *)pAlloc;
986ca62ad57Sdan     pAlloc = &abPK[nDbCol];
987e8d5648eSdan     if( pzTab ){
988e8d5648eSdan       memcpy(pAlloc, zThis, nThis+1);
989e8d5648eSdan       *pzTab = (char *)pAlloc;
990e8d5648eSdan       pAlloc += nThis+1;
991e8d5648eSdan     }
992e8d5648eSdan 
993e8d5648eSdan     i = 0;
994e8d5648eSdan     while( SQLITE_ROW==sqlite3_step(pStmt) ){
995e8d5648eSdan       int nName = sqlite3_column_bytes(pStmt, 1);
996e8d5648eSdan       const unsigned char *zName = sqlite3_column_text(pStmt, 1);
997e8d5648eSdan       if( zName==0 ) break;
998e8d5648eSdan       memcpy(pAlloc, zName, nName+1);
999e8d5648eSdan       azCol[i] = (char *)pAlloc;
1000e8d5648eSdan       pAlloc += nName+1;
1001db04571cSdan       abPK[i] = sqlite3_column_int(pStmt, 5);
1002e8d5648eSdan       i++;
1003e8d5648eSdan     }
1004e8d5648eSdan     rc = sqlite3_reset(pStmt);
1005e8d5648eSdan 
1006e8d5648eSdan   }
1007e8d5648eSdan 
1008e8d5648eSdan   /* If successful, populate the output variables. Otherwise, zero them and
1009e8d5648eSdan   ** free any allocation made. An error code will be returned in this case.
1010e8d5648eSdan   */
1011e8d5648eSdan   if( rc==SQLITE_OK ){
1012db04571cSdan     *pazCol = (const char **)azCol;
1013db04571cSdan     *pabPK = abPK;
1014ca62ad57Sdan     *pnCol = nDbCol;
1015e8d5648eSdan   }else{
1016db04571cSdan     *pazCol = 0;
1017db04571cSdan     *pabPK = 0;
1018ca62ad57Sdan     *pnCol = 0;
1019e8d5648eSdan     if( pzTab ) *pzTab = 0;
1020db04571cSdan     sqlite3_free(azCol);
1021e8d5648eSdan   }
1022e8d5648eSdan   sqlite3_finalize(pStmt);
1023e8d5648eSdan   return rc;
1024e8d5648eSdan }
1025e8d5648eSdan 
1026e8d5648eSdan /*
1027296c7658Sdan ** This function is only called from within a pre-update handler for a
1028296c7658Sdan ** write to table pTab, part of session pSession. If this is the first
10296dc29e60Sdan ** write to this table, initalize the SessionTable.nCol, azCol[] and
10306dc29e60Sdan ** abPK[] arrays accordingly.
1031296c7658Sdan **
10326dc29e60Sdan ** If an error occurs, an error code is stored in sqlite3_session.rc and
10336dc29e60Sdan ** non-zero returned. Or, if no error occurs but the table has no primary
10346dc29e60Sdan ** key, sqlite3_session.rc is left set to SQLITE_OK and non-zero returned to
10356dc29e60Sdan ** indicate that updates on this table should be ignored. SessionTable.abPK
10366dc29e60Sdan ** is set to NULL in this case.
1037296c7658Sdan */
10384fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){
10394fccf43aSdan   if( pTab->nCol==0 ){
10406dc29e60Sdan     u8 *abPK;
1041e8d5648eSdan     assert( pTab->azCol==0 || pTab->abPK==0 );
1042e8d5648eSdan     pSession->rc = sessionTableInfo(pSession->db, pSession->zDb,
10436dc29e60Sdan         pTab->zName, &pTab->nCol, 0, &pTab->azCol, &abPK
1044e8d5648eSdan     );
10456dc29e60Sdan     if( pSession->rc==SQLITE_OK ){
10466dc29e60Sdan       int i;
10476dc29e60Sdan       for(i=0; i<pTab->nCol; i++){
10486dc29e60Sdan         if( abPK[i] ){
10496dc29e60Sdan           pTab->abPK = abPK;
10506dc29e60Sdan           break;
1051ca62ad57Sdan         }
10524fccf43aSdan       }
10533739f298Sdan       if( 0==sqlite3_stricmp("sqlite_stat1", pTab->zName) ){
10543739f298Sdan         pTab->bStat1 = 1;
10553739f298Sdan       }
10566dc29e60Sdan     }
10576dc29e60Sdan   }
10586dc29e60Sdan   return (pSession->rc || pTab->abPK==0);
1059e8d5648eSdan }
1060e8d5648eSdan 
106177fc1d5bSdan /*
10621611e5a3Sdan ** Versions of the four methods in object SessionHook for use with the
10631611e5a3Sdan ** sqlite_stat1 table. The purpose of this is to substitute a zero-length
10641611e5a3Sdan ** blob each time a NULL value is read from the "idx" column of the
10651611e5a3Sdan ** sqlite_stat1 table.
10661611e5a3Sdan */
10671611e5a3Sdan typedef struct SessionStat1Ctx SessionStat1Ctx;
10681611e5a3Sdan struct SessionStat1Ctx {
10691611e5a3Sdan   SessionHook hook;
10701611e5a3Sdan   sqlite3_session *pSession;
10711611e5a3Sdan };
10721611e5a3Sdan static int sessionStat1Old(void *pCtx, int iCol, sqlite3_value **ppVal){
10731611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10741611e5a3Sdan   sqlite3_value *pVal = 0;
10751611e5a3Sdan   int rc = p->hook.xOld(p->hook.pCtx, iCol, &pVal);
10761611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
10771611e5a3Sdan     pVal = p->pSession->pZeroBlob;
10781611e5a3Sdan   }
10791611e5a3Sdan   *ppVal = pVal;
10801611e5a3Sdan   return rc;
10811611e5a3Sdan }
10821611e5a3Sdan static int sessionStat1New(void *pCtx, int iCol, sqlite3_value **ppVal){
10831611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10841611e5a3Sdan   sqlite3_value *pVal = 0;
10851611e5a3Sdan   int rc = p->hook.xNew(p->hook.pCtx, iCol, &pVal);
10861611e5a3Sdan   if( rc==SQLITE_OK && iCol==1 && sqlite3_value_type(pVal)==SQLITE_NULL ){
10871611e5a3Sdan     pVal = p->pSession->pZeroBlob;
10881611e5a3Sdan   }
10891611e5a3Sdan   *ppVal = pVal;
10901611e5a3Sdan   return rc;
10911611e5a3Sdan }
10921611e5a3Sdan static int sessionStat1Count(void *pCtx){
10931611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10941611e5a3Sdan   return p->hook.xCount(p->hook.pCtx);
10951611e5a3Sdan }
10961611e5a3Sdan static int sessionStat1Depth(void *pCtx){
10971611e5a3Sdan   SessionStat1Ctx *p = (SessionStat1Ctx*)pCtx;
10981611e5a3Sdan   return p->hook.xDepth(p->hook.pCtx);
10991611e5a3Sdan }
11001611e5a3Sdan 
11011611e5a3Sdan 
11021611e5a3Sdan /*
110377fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a
110477fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change
110577fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument.
110677fc1d5bSdan **
110777fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added
110877fc1d5bSdan ** to the changed-rows hash table associated with table pTab.
110977fc1d5bSdan */
1110e8d5648eSdan static void sessionPreupdateOneChange(
111177fc1d5bSdan   int op,                         /* One of SQLITE_UPDATE, INSERT, DELETE */
111277fc1d5bSdan   sqlite3_session *pSession,      /* Session object pTab is attached to */
111377fc1d5bSdan   SessionTable *pTab              /* Table that change applies to */
1114e8d5648eSdan ){
1115e8d5648eSdan   int iHash;
1116cf8e9144Sdan   int bNull = 0;
1117e8d5648eSdan   int rc = SQLITE_OK;
1118fa3d7855Smistachkin   SessionStat1Ctx stat1 = {0};
1119e8d5648eSdan 
1120e8d5648eSdan   if( pSession->rc ) return;
1121e8d5648eSdan 
1122e8d5648eSdan   /* Load table details if required */
1123e8d5648eSdan   if( sessionInitTable(pSession, pTab) ) return;
1124e8d5648eSdan 
11256dc29e60Sdan   /* Check the number of columns in this xPreUpdate call matches the
11266dc29e60Sdan   ** number of columns in the table.  */
11276dc29e60Sdan   if( pTab->nCol!=pSession->hook.xCount(pSession->hook.pCtx) ){
11286dc29e60Sdan     pSession->rc = SQLITE_SCHEMA;
11296dc29e60Sdan     return;
11306dc29e60Sdan   }
11316dc29e60Sdan 
1132e8d5648eSdan   /* Grow the hash table if required */
113364277f4aSdan   if( sessionGrowHash(0, pTab) ){
11345d607a6eSdan     pSession->rc = SQLITE_NOMEM;
11355d607a6eSdan     return;
11365d607a6eSdan   }
1137e8d5648eSdan 
11381611e5a3Sdan   if( pTab->bStat1 ){
11391611e5a3Sdan     stat1.hook = pSession->hook;
11401611e5a3Sdan     stat1.pSession = pSession;
11411611e5a3Sdan     pSession->hook.pCtx = (void*)&stat1;
11421611e5a3Sdan     pSession->hook.xNew = sessionStat1New;
11431611e5a3Sdan     pSession->hook.xOld = sessionStat1Old;
11441611e5a3Sdan     pSession->hook.xCount = sessionStat1Count;
11451611e5a3Sdan     pSession->hook.xDepth = sessionStat1Depth;
11461611e5a3Sdan     if( pSession->pZeroBlob==0 ){
11471611e5a3Sdan       sqlite3_value *p = sqlite3ValueNew(0);
11481611e5a3Sdan       if( p==0 ){
11491611e5a3Sdan         rc = SQLITE_NOMEM;
11501611e5a3Sdan         goto error_out;
11511611e5a3Sdan       }
11521611e5a3Sdan       sqlite3ValueSetStr(p, 0, "", 0, SQLITE_STATIC);
11531611e5a3Sdan       pSession->pZeroBlob = p;
11541611e5a3Sdan     }
11551611e5a3Sdan   }
11561611e5a3Sdan 
115780fe2d93Sdan   /* Calculate the hash-key for this change. If the primary key of the row
115880fe2d93Sdan   ** includes a NULL value, exit early. Such changes are ignored by the
115980fe2d93Sdan   ** session module. */
1160cf8e9144Sdan   rc = sessionPreupdateHash(pSession, pTab, op==SQLITE_INSERT, &iHash, &bNull);
116180fe2d93Sdan   if( rc!=SQLITE_OK ) goto error_out;
116280fe2d93Sdan 
11631611e5a3Sdan   if( bNull==0 ){
116480fe2d93Sdan     /* Search the hash table for an existing record for this row. */
1165b4480e94Sdan     SessionChange *pC;
11666734007dSdan     for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){
1167cf8e9144Sdan       if( sessionPreupdateEqual(pSession, pTab, pC, op) ) break;
1168e8d5648eSdan     }
116980fe2d93Sdan 
1170e8d5648eSdan     if( pC==0 ){
1171e8d5648eSdan       /* Create a new change object containing all the old values (if
1172e8d5648eSdan       ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK
1173e8d5648eSdan       ** values (if this is an INSERT). */
1174b4480e94Sdan       SessionChange *pChange; /* New change object */
1175e8d5648eSdan       int nByte;              /* Number of bytes to allocate */
1176e8d5648eSdan       int i;                  /* Used to iterate through columns */
1177e8d5648eSdan 
1178b4480e94Sdan       assert( rc==SQLITE_OK );
1179e8d5648eSdan       pTab->nEntry++;
1180e8d5648eSdan 
1181e8d5648eSdan       /* Figure out how large an allocation is required */
1182e8d5648eSdan       nByte = sizeof(SessionChange);
118380fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1184e8d5648eSdan         sqlite3_value *p = 0;
1185e8d5648eSdan         if( op!=SQLITE_INSERT ){
1186cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xOld(pSession->hook.pCtx, i, &p);
118780fe2d93Sdan           assert( trc==SQLITE_OK );
118880fe2d93Sdan         }else if( pTab->abPK[i] ){
1189cf8e9144Sdan           TESTONLY(int trc = ) pSession->hook.xNew(pSession->hook.pCtx, i, &p);
119080fe2d93Sdan           assert( trc==SQLITE_OK );
1191e8d5648eSdan         }
119280fe2d93Sdan 
119380fe2d93Sdan         /* This may fail if SQLite value p contains a utf-16 string that must
119480fe2d93Sdan         ** be converted to utf-8 and an OOM error occurs while doing so. */
1195e8d5648eSdan         rc = sessionSerializeValue(0, p, &nByte);
119680fe2d93Sdan         if( rc!=SQLITE_OK ) goto error_out;
1197e8d5648eSdan       }
1198e8d5648eSdan 
1199e8d5648eSdan       /* Allocate the change object */
1200e8d5648eSdan       pChange = (SessionChange *)sqlite3_malloc(nByte);
1201e8d5648eSdan       if( !pChange ){
1202e8d5648eSdan         rc = SQLITE_NOMEM;
120380fe2d93Sdan         goto error_out;
1204e8d5648eSdan       }else{
1205e8d5648eSdan         memset(pChange, 0, sizeof(SessionChange));
1206e8d5648eSdan         pChange->aRecord = (u8 *)&pChange[1];
1207e8d5648eSdan       }
1208e8d5648eSdan 
120980fe2d93Sdan       /* Populate the change object. None of the preupdate_old(),
121080fe2d93Sdan       ** preupdate_new() or SerializeValue() calls below may fail as all
121180fe2d93Sdan       ** required values and encodings have already been cached in memory.
121280fe2d93Sdan       ** It is not possible for an OOM to occur in this block. */
1213e8d5648eSdan       nByte = 0;
121480fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1215e8d5648eSdan         sqlite3_value *p = 0;
1216e8d5648eSdan         if( op!=SQLITE_INSERT ){
1217cf8e9144Sdan           pSession->hook.xOld(pSession->hook.pCtx, i, &p);
121880fe2d93Sdan         }else if( pTab->abPK[i] ){
1219cf8e9144Sdan           pSession->hook.xNew(pSession->hook.pCtx, i, &p);
1220e8d5648eSdan         }
122180fe2d93Sdan         sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte);
1222e8d5648eSdan       }
122380fe2d93Sdan 
122480fe2d93Sdan       /* Add the change to the hash-table */
1225cf8e9144Sdan       if( pSession->bIndirect || pSession->hook.xDepth(pSession->hook.pCtx) ){
1226b4480e94Sdan         pChange->bIndirect = 1;
1227b4480e94Sdan       }
122812ca0b56Sdan       pChange->nRecord = nByte;
1229798693b2Sdan       pChange->op = op;
1230e8d5648eSdan       pChange->pNext = pTab->apChange[iHash];
1231e8d5648eSdan       pTab->apChange[iHash] = pChange;
123280fe2d93Sdan 
123380fe2d93Sdan     }else if( pC->bIndirect ){
1234b4480e94Sdan       /* If the existing change is considered "indirect", but this current
1235b4480e94Sdan       ** change is "direct", mark the change object as direct. */
1236cf8e9144Sdan       if( pSession->hook.xDepth(pSession->hook.pCtx)==0
1237cf8e9144Sdan        && pSession->bIndirect==0
1238cf8e9144Sdan       ){
1239b4480e94Sdan         pC->bIndirect = 0;
1240b4480e94Sdan       }
1241e8d5648eSdan     }
12424fccf43aSdan   }
124312ca0b56Sdan 
124412ca0b56Sdan   /* If an error has occurred, mark the session object as failed. */
124580fe2d93Sdan  error_out:
12461611e5a3Sdan   if( pTab->bStat1 ){
12471611e5a3Sdan     pSession->hook = stat1.hook;
12481611e5a3Sdan   }
124912ca0b56Sdan   if( rc!=SQLITE_OK ){
125012ca0b56Sdan     pSession->rc = rc;
125112ca0b56Sdan   }
125227453faeSdan }
12534fccf43aSdan 
1254cf8e9144Sdan static int sessionFindTable(
1255cf8e9144Sdan   sqlite3_session *pSession,
1256cf8e9144Sdan   const char *zName,
1257cf8e9144Sdan   SessionTable **ppTab
1258cf8e9144Sdan ){
1259cf8e9144Sdan   int rc = SQLITE_OK;
1260cf8e9144Sdan   int nName = sqlite3Strlen30(zName);
1261cf8e9144Sdan   SessionTable *pRet;
1262cf8e9144Sdan 
1263cf8e9144Sdan   /* Search for an existing table */
1264cf8e9144Sdan   for(pRet=pSession->pTable; pRet; pRet=pRet->pNext){
1265cf8e9144Sdan     if( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) ) break;
1266cf8e9144Sdan   }
1267cf8e9144Sdan 
1268cf8e9144Sdan   if( pRet==0 && pSession->bAutoAttach ){
1269cf8e9144Sdan     /* If there is a table-filter configured, invoke it. If it returns 0,
1270cf8e9144Sdan     ** do not automatically add the new table. */
1271cf8e9144Sdan     if( pSession->xTableFilter==0
1272cf8e9144Sdan      || pSession->xTableFilter(pSession->pFilterCtx, zName)
1273cf8e9144Sdan     ){
1274cf8e9144Sdan       rc = sqlite3session_attach(pSession, zName);
1275cf8e9144Sdan       if( rc==SQLITE_OK ){
12766c39e6a8Sdan         for(pRet=pSession->pTable; pRet->pNext; pRet=pRet->pNext);
1277cf8e9144Sdan         assert( 0==sqlite3_strnicmp(pRet->zName, zName, nName+1) );
1278cf8e9144Sdan       }
1279cf8e9144Sdan     }
1280cf8e9144Sdan   }
1281cf8e9144Sdan 
1282cf8e9144Sdan   assert( rc==SQLITE_OK || pRet==0 );
1283cf8e9144Sdan   *ppTab = pRet;
1284cf8e9144Sdan   return rc;
1285cf8e9144Sdan }
1286cf8e9144Sdan 
12874fccf43aSdan /*
12884fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases.
12894fccf43aSdan */
12904fccf43aSdan static void xPreUpdate(
12914fccf43aSdan   void *pCtx,                     /* Copy of third arg to preupdate_hook() */
12924fccf43aSdan   sqlite3 *db,                    /* Database handle */
12934fccf43aSdan   int op,                         /* SQLITE_UPDATE, DELETE or INSERT */
12944fccf43aSdan   char const *zDb,                /* Database name */
12954fccf43aSdan   char const *zName,              /* Table name */
12964fccf43aSdan   sqlite3_int64 iKey1,            /* Rowid of row about to be deleted/updated */
12974fccf43aSdan   sqlite3_int64 iKey2             /* New rowid value (for a rowid UPDATE) */
12984fccf43aSdan ){
12994fccf43aSdan   sqlite3_session *pSession;
1300cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb);
13014fccf43aSdan 
13024c220252Sdan   assert( sqlite3_mutex_held(db->mutex) );
13034c220252Sdan 
13044fccf43aSdan   for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){
13054fccf43aSdan     SessionTable *pTab;
1306296c7658Sdan 
1307e8d5648eSdan     /* If this session is attached to a different database ("main", "temp"
1308e8d5648eSdan     ** etc.), or if it is not currently enabled, there is nothing to do. Skip
1309e8d5648eSdan     ** to the next session object attached to this database. */
1310296c7658Sdan     if( pSession->bEnable==0 ) continue;
13114fccf43aSdan     if( pSession->rc ) continue;
13124fccf43aSdan     if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue;
1313296c7658Sdan 
1314cf8e9144Sdan     pSession->rc = sessionFindTable(pSession, zName, &pTab);
1315cf8e9144Sdan     if( pTab ){
1316cf8e9144Sdan       assert( pSession->rc==SQLITE_OK );
1317e8d5648eSdan       sessionPreupdateOneChange(op, pSession, pTab);
1318e8d5648eSdan       if( op==SQLITE_UPDATE ){
1319e8d5648eSdan         sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab);
13204fccf43aSdan       }
13214fccf43aSdan     }
13224fccf43aSdan   }
13234fccf43aSdan }
1324cf8e9144Sdan 
1325cf8e9144Sdan /*
1326cf8e9144Sdan ** The pre-update hook implementations.
1327cf8e9144Sdan */
1328cf8e9144Sdan static int sessionPreupdateOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1329cf8e9144Sdan   return sqlite3_preupdate_old((sqlite3*)pCtx, iVal, ppVal);
1330cf8e9144Sdan }
1331cf8e9144Sdan static int sessionPreupdateNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1332cf8e9144Sdan   return sqlite3_preupdate_new((sqlite3*)pCtx, iVal, ppVal);
1333cf8e9144Sdan }
1334cf8e9144Sdan static int sessionPreupdateCount(void *pCtx){
1335cf8e9144Sdan   return sqlite3_preupdate_count((sqlite3*)pCtx);
1336cf8e9144Sdan }
1337cf8e9144Sdan static int sessionPreupdateDepth(void *pCtx){
1338cf8e9144Sdan   return sqlite3_preupdate_depth((sqlite3*)pCtx);
1339cf8e9144Sdan }
1340cf8e9144Sdan 
1341cf8e9144Sdan /*
1342cf8e9144Sdan ** Install the pre-update hooks on the session object passed as the only
1343cf8e9144Sdan ** argument.
1344cf8e9144Sdan */
1345cf8e9144Sdan static void sessionPreupdateHooks(
1346cf8e9144Sdan   sqlite3_session *pSession
1347cf8e9144Sdan ){
1348cf8e9144Sdan   pSession->hook.pCtx = (void*)pSession->db;
1349cf8e9144Sdan   pSession->hook.xOld = sessionPreupdateOld;
1350cf8e9144Sdan   pSession->hook.xNew = sessionPreupdateNew;
1351cf8e9144Sdan   pSession->hook.xCount = sessionPreupdateCount;
1352cf8e9144Sdan   pSession->hook.xDepth = sessionPreupdateDepth;
1353cf8e9144Sdan }
1354cf8e9144Sdan 
1355cf8e9144Sdan typedef struct SessionDiffCtx SessionDiffCtx;
1356cf8e9144Sdan struct SessionDiffCtx {
1357cf8e9144Sdan   sqlite3_stmt *pStmt;
1358cf8e9144Sdan   int nOldOff;
1359cf8e9144Sdan };
1360cf8e9144Sdan 
1361cf8e9144Sdan /*
1362cf8e9144Sdan ** The diff hook implementations.
1363cf8e9144Sdan */
1364cf8e9144Sdan static int sessionDiffOld(void *pCtx, int iVal, sqlite3_value **ppVal){
1365cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1366cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal+p->nOldOff);
1367cf8e9144Sdan   return SQLITE_OK;
1368cf8e9144Sdan }
1369cf8e9144Sdan static int sessionDiffNew(void *pCtx, int iVal, sqlite3_value **ppVal){
1370cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1371cf8e9144Sdan   *ppVal = sqlite3_column_value(p->pStmt, iVal);
1372cf8e9144Sdan    return SQLITE_OK;
1373cf8e9144Sdan }
1374cf8e9144Sdan static int sessionDiffCount(void *pCtx){
1375cf8e9144Sdan   SessionDiffCtx *p = (SessionDiffCtx*)pCtx;
1376cf8e9144Sdan   return p->nOldOff ? p->nOldOff : sqlite3_column_count(p->pStmt);
1377cf8e9144Sdan }
1378cf8e9144Sdan static int sessionDiffDepth(void *pCtx){
1379cf8e9144Sdan   return 0;
1380cf8e9144Sdan }
1381cf8e9144Sdan 
1382cf8e9144Sdan /*
1383cf8e9144Sdan ** Install the diff hooks on the session object passed as the only
1384cf8e9144Sdan ** argument.
1385cf8e9144Sdan */
1386cf8e9144Sdan static void sessionDiffHooks(
1387cf8e9144Sdan   sqlite3_session *pSession,
1388cf8e9144Sdan   SessionDiffCtx *pDiffCtx
1389cf8e9144Sdan ){
1390cf8e9144Sdan   pSession->hook.pCtx = (void*)pDiffCtx;
1391cf8e9144Sdan   pSession->hook.xOld = sessionDiffOld;
1392cf8e9144Sdan   pSession->hook.xNew = sessionDiffNew;
1393cf8e9144Sdan   pSession->hook.xCount = sessionDiffCount;
1394cf8e9144Sdan   pSession->hook.xDepth = sessionDiffDepth;
1395cf8e9144Sdan }
1396cf8e9144Sdan 
1397cf8e9144Sdan static char *sessionExprComparePK(
1398cf8e9144Sdan   int nCol,
1399cf8e9144Sdan   const char *zDb1, const char *zDb2,
1400cf8e9144Sdan   const char *zTab,
1401cf8e9144Sdan   const char **azCol, u8 *abPK
1402cf8e9144Sdan ){
1403cf8e9144Sdan   int i;
1404cf8e9144Sdan   const char *zSep = "";
1405cf8e9144Sdan   char *zRet = 0;
1406cf8e9144Sdan 
1407cf8e9144Sdan   for(i=0; i<nCol; i++){
1408cf8e9144Sdan     if( abPK[i] ){
1409cf8e9144Sdan       zRet = sqlite3_mprintf("%z%s\"%w\".\"%w\".\"%w\"=\"%w\".\"%w\".\"%w\"",
1410cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1411cf8e9144Sdan       );
1412cf8e9144Sdan       zSep = " AND ";
1413cf8e9144Sdan       if( zRet==0 ) break;
1414cf8e9144Sdan     }
1415cf8e9144Sdan   }
1416cf8e9144Sdan 
1417cf8e9144Sdan   return zRet;
1418cf8e9144Sdan }
1419cf8e9144Sdan 
1420cf8e9144Sdan static char *sessionExprCompareOther(
1421cf8e9144Sdan   int nCol,
1422cf8e9144Sdan   const char *zDb1, const char *zDb2,
1423cf8e9144Sdan   const char *zTab,
1424cf8e9144Sdan   const char **azCol, u8 *abPK
1425cf8e9144Sdan ){
1426cf8e9144Sdan   int i;
1427cf8e9144Sdan   const char *zSep = "";
1428cf8e9144Sdan   char *zRet = 0;
1429cf8e9144Sdan   int bHave = 0;
1430cf8e9144Sdan 
1431cf8e9144Sdan   for(i=0; i<nCol; i++){
1432cf8e9144Sdan     if( abPK[i]==0 ){
1433cf8e9144Sdan       bHave = 1;
1434cf8e9144Sdan       zRet = sqlite3_mprintf(
1435cf8e9144Sdan           "%z%s\"%w\".\"%w\".\"%w\" IS NOT \"%w\".\"%w\".\"%w\"",
1436cf8e9144Sdan           zRet, zSep, zDb1, zTab, azCol[i], zDb2, zTab, azCol[i]
1437cf8e9144Sdan       );
1438cf8e9144Sdan       zSep = " OR ";
1439cf8e9144Sdan       if( zRet==0 ) break;
1440cf8e9144Sdan     }
1441cf8e9144Sdan   }
1442cf8e9144Sdan 
1443cf8e9144Sdan   if( bHave==0 ){
1444cf8e9144Sdan     assert( zRet==0 );
1445cf8e9144Sdan     zRet = sqlite3_mprintf("0");
1446cf8e9144Sdan   }
1447cf8e9144Sdan 
1448cf8e9144Sdan   return zRet;
1449cf8e9144Sdan }
1450cf8e9144Sdan 
1451cf8e9144Sdan static char *sessionSelectFindNew(
1452cf8e9144Sdan   int nCol,
1453cf8e9144Sdan   const char *zDb1,      /* Pick rows in this db only */
1454cf8e9144Sdan   const char *zDb2,      /* But not in this one */
1455cf8e9144Sdan   const char *zTbl,      /* Table name */
1456cf8e9144Sdan   const char *zExpr
1457cf8e9144Sdan ){
1458cf8e9144Sdan   char *zRet = sqlite3_mprintf(
1459cf8e9144Sdan       "SELECT * FROM \"%w\".\"%w\" WHERE NOT EXISTS ("
1460cf8e9144Sdan       "  SELECT 1 FROM \"%w\".\"%w\" WHERE %s"
1461cf8e9144Sdan       ")",
1462cf8e9144Sdan       zDb1, zTbl, zDb2, zTbl, zExpr
1463cf8e9144Sdan   );
1464cf8e9144Sdan   return zRet;
1465cf8e9144Sdan }
1466cf8e9144Sdan 
1467cf8e9144Sdan static int sessionDiffFindNew(
1468cf8e9144Sdan   int op,
1469cf8e9144Sdan   sqlite3_session *pSession,
1470cf8e9144Sdan   SessionTable *pTab,
1471cf8e9144Sdan   const char *zDb1,
1472cf8e9144Sdan   const char *zDb2,
1473cf8e9144Sdan   char *zExpr
1474cf8e9144Sdan ){
1475cf8e9144Sdan   int rc = SQLITE_OK;
1476cf8e9144Sdan   char *zStmt = sessionSelectFindNew(pTab->nCol, zDb1, zDb2, pTab->zName,zExpr);
1477cf8e9144Sdan 
1478cf8e9144Sdan   if( zStmt==0 ){
1479cf8e9144Sdan     rc = SQLITE_NOMEM;
1480cf8e9144Sdan   }else{
1481cf8e9144Sdan     sqlite3_stmt *pStmt;
1482cf8e9144Sdan     rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1483cf8e9144Sdan     if( rc==SQLITE_OK ){
1484cf8e9144Sdan       SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1485cf8e9144Sdan       pDiffCtx->pStmt = pStmt;
1486cf8e9144Sdan       pDiffCtx->nOldOff = 0;
1487cf8e9144Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
1488cf8e9144Sdan         sessionPreupdateOneChange(op, pSession, pTab);
1489cf8e9144Sdan       }
1490cf8e9144Sdan       rc = sqlite3_finalize(pStmt);
1491cf8e9144Sdan     }
1492cf8e9144Sdan     sqlite3_free(zStmt);
1493cf8e9144Sdan   }
1494cf8e9144Sdan 
1495cf8e9144Sdan   return rc;
1496cf8e9144Sdan }
1497cf8e9144Sdan 
1498cf8e9144Sdan static int sessionDiffFindModified(
1499cf8e9144Sdan   sqlite3_session *pSession,
1500cf8e9144Sdan   SessionTable *pTab,
1501cf8e9144Sdan   const char *zFrom,
1502cf8e9144Sdan   const char *zExpr
1503cf8e9144Sdan ){
1504cf8e9144Sdan   int rc = SQLITE_OK;
1505cf8e9144Sdan 
1506cf8e9144Sdan   char *zExpr2 = sessionExprCompareOther(pTab->nCol,
1507cf8e9144Sdan       pSession->zDb, zFrom, pTab->zName, pTab->azCol, pTab->abPK
1508cf8e9144Sdan   );
1509cf8e9144Sdan   if( zExpr2==0 ){
1510cf8e9144Sdan     rc = SQLITE_NOMEM;
1511cf8e9144Sdan   }else{
1512cf8e9144Sdan     char *zStmt = sqlite3_mprintf(
1513dd009f83Sdan         "SELECT * FROM \"%w\".\"%w\", \"%w\".\"%w\" WHERE %s AND (%z)",
1514cf8e9144Sdan         pSession->zDb, pTab->zName, zFrom, pTab->zName, zExpr, zExpr2
1515cf8e9144Sdan     );
1516cf8e9144Sdan     if( zStmt==0 ){
1517cf8e9144Sdan       rc = SQLITE_NOMEM;
1518cf8e9144Sdan     }else{
1519cf8e9144Sdan       sqlite3_stmt *pStmt;
1520cf8e9144Sdan       rc = sqlite3_prepare(pSession->db, zStmt, -1, &pStmt, 0);
1521cf8e9144Sdan 
1522cf8e9144Sdan       if( rc==SQLITE_OK ){
1523cf8e9144Sdan         SessionDiffCtx *pDiffCtx = (SessionDiffCtx*)pSession->hook.pCtx;
1524cf8e9144Sdan         pDiffCtx->pStmt = pStmt;
1525cf8e9144Sdan         pDiffCtx->nOldOff = pTab->nCol;
1526cf8e9144Sdan         while( SQLITE_ROW==sqlite3_step(pStmt) ){
1527cf8e9144Sdan           sessionPreupdateOneChange(SQLITE_UPDATE, pSession, pTab);
1528cf8e9144Sdan         }
1529cf8e9144Sdan         rc = sqlite3_finalize(pStmt);
1530cf8e9144Sdan       }
1531cf8e9144Sdan       sqlite3_free(zStmt);
1532cf8e9144Sdan     }
1533cf8e9144Sdan   }
1534cf8e9144Sdan 
1535cf8e9144Sdan   return rc;
1536cf8e9144Sdan }
1537cf8e9144Sdan 
1538cf8e9144Sdan int sqlite3session_diff(
1539cf8e9144Sdan   sqlite3_session *pSession,
1540cf8e9144Sdan   const char *zFrom,
1541cf8e9144Sdan   const char *zTbl,
1542cf8e9144Sdan   char **pzErrMsg
1543cf8e9144Sdan ){
1544cf8e9144Sdan   const char *zDb = pSession->zDb;
1545cf8e9144Sdan   int rc = pSession->rc;
1546cf8e9144Sdan   SessionDiffCtx d;
1547cf8e9144Sdan 
1548cf8e9144Sdan   memset(&d, 0, sizeof(d));
1549cf8e9144Sdan   sessionDiffHooks(pSession, &d);
1550cf8e9144Sdan 
155110dc553cSdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1552cf8e9144Sdan   if( pzErrMsg ) *pzErrMsg = 0;
1553cf8e9144Sdan   if( rc==SQLITE_OK ){
1554cf8e9144Sdan     char *zExpr = 0;
1555cf8e9144Sdan     sqlite3 *db = pSession->db;
1556cf8e9144Sdan     SessionTable *pTo;            /* Table zTbl */
1557cf8e9144Sdan 
1558cf8e9144Sdan     /* Locate and if necessary initialize the target table object */
1559cf8e9144Sdan     rc = sessionFindTable(pSession, zTbl, &pTo);
1560cf8e9144Sdan     if( pTo==0 ) goto diff_out;
15616dc29e60Sdan     if( sessionInitTable(pSession, pTo) ){
15626dc29e60Sdan       rc = pSession->rc;
15636dc29e60Sdan       goto diff_out;
1564cf8e9144Sdan     }
1565cf8e9144Sdan 
1566cf8e9144Sdan     /* Check the table schemas match */
1567cf8e9144Sdan     if( rc==SQLITE_OK ){
15684cc923e3Sdan       int bHasPk = 0;
1569b9db9099Sdan       int bMismatch = 0;
1570cf8e9144Sdan       int nCol;                   /* Columns in zFrom.zTbl */
1571cf8e9144Sdan       u8 *abPK;
1572cf8e9144Sdan       const char **azCol = 0;
1573cf8e9144Sdan       rc = sessionTableInfo(db, zFrom, zTbl, &nCol, 0, &azCol, &abPK);
1574cf8e9144Sdan       if( rc==SQLITE_OK ){
15754cc923e3Sdan         if( pTo->nCol!=nCol ){
1576cf8e9144Sdan           bMismatch = 1;
1577cf8e9144Sdan         }else{
1578cf8e9144Sdan           int i;
1579cf8e9144Sdan           for(i=0; i<nCol; i++){
15804cc923e3Sdan             if( pTo->abPK[i]!=abPK[i] ) bMismatch = 1;
1581cf8e9144Sdan             if( sqlite3_stricmp(azCol[i], pTo->azCol[i]) ) bMismatch = 1;
15824cc923e3Sdan             if( abPK[i] ) bHasPk = 1;
1583cf8e9144Sdan           }
1584cf8e9144Sdan         }
1585cf8e9144Sdan       }
1586dbbd8160Sdrh       sqlite3_free((char*)azCol);
1587b9db9099Sdan       if( bMismatch ){
1588b9db9099Sdan         *pzErrMsg = sqlite3_mprintf("table schemas do not match");
1589b9db9099Sdan         rc = SQLITE_SCHEMA;
1590b9db9099Sdan       }
1591b9db9099Sdan       if( bHasPk==0 ){
1592b9db9099Sdan         /* Ignore tables with no primary keys */
1593b9db9099Sdan         goto diff_out;
1594b9db9099Sdan       }
1595cf8e9144Sdan     }
1596cf8e9144Sdan 
1597cf8e9144Sdan     if( rc==SQLITE_OK ){
1598cf8e9144Sdan       zExpr = sessionExprComparePK(pTo->nCol,
1599cf8e9144Sdan           zDb, zFrom, pTo->zName, pTo->azCol, pTo->abPK
1600cf8e9144Sdan       );
1601cf8e9144Sdan     }
1602cf8e9144Sdan 
1603cf8e9144Sdan     /* Find new rows */
1604cf8e9144Sdan     if( rc==SQLITE_OK ){
1605cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_INSERT, pSession, pTo, zDb, zFrom, zExpr);
1606cf8e9144Sdan     }
1607cf8e9144Sdan 
1608cf8e9144Sdan     /* Find old rows */
1609cf8e9144Sdan     if( rc==SQLITE_OK ){
1610cf8e9144Sdan       rc = sessionDiffFindNew(SQLITE_DELETE, pSession, pTo, zFrom, zDb, zExpr);
1611cf8e9144Sdan     }
1612cf8e9144Sdan 
1613cf8e9144Sdan     /* Find modified rows */
1614cf8e9144Sdan     if( rc==SQLITE_OK ){
1615cf8e9144Sdan       rc = sessionDiffFindModified(pSession, pTo, zFrom, zExpr);
1616cf8e9144Sdan     }
1617cf8e9144Sdan 
1618cf8e9144Sdan     sqlite3_free(zExpr);
1619cf8e9144Sdan   }
1620cf8e9144Sdan 
1621cf8e9144Sdan  diff_out:
1622cf8e9144Sdan   sessionPreupdateHooks(pSession);
162310dc553cSdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
1624cf8e9144Sdan   return rc;
1625296c7658Sdan }
16264fccf43aSdan 
16274fccf43aSdan /*
16284fccf43aSdan ** Create a session object. This session object will record changes to
16294fccf43aSdan ** database zDb attached to connection db.
16304fccf43aSdan */
16314fccf43aSdan int sqlite3session_create(
16324fccf43aSdan   sqlite3 *db,                    /* Database handle */
16334fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
16344fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
16354fccf43aSdan ){
1636296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1637296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1638cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
16394fccf43aSdan 
1640296c7658Sdan   /* Zero the output value in case an error occurs. */
16414fccf43aSdan   *ppSession = 0;
16424fccf43aSdan 
16434fccf43aSdan   /* Allocate and populate the new session object. */
16444fccf43aSdan   pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1);
16454fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
16464fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
16474fccf43aSdan   pNew->db = db;
16484fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1649296c7658Sdan   pNew->bEnable = 1;
16504fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
1651cf8e9144Sdan   sessionPreupdateHooks(pNew);
16524fccf43aSdan 
16534fccf43aSdan   /* Add the new session object to the linked list of session objects
16544fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
16554fccf43aSdan   ** handle mutex.  */
16564fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
16574fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
16584fccf43aSdan   pNew->pNext = pOld;
16594fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
16604fccf43aSdan 
16614fccf43aSdan   *ppSession = pNew;
16624fccf43aSdan   return SQLITE_OK;
16634fccf43aSdan }
16644fccf43aSdan 
166577fc1d5bSdan /*
166677fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
166777fc1d5bSdan ** of the changed-rows hash tables are also deleted.
166877fc1d5bSdan */
16691ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
16705d607a6eSdan   SessionTable *pNext;
16715d607a6eSdan   SessionTable *pTab;
16725d607a6eSdan 
16735d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
16745d607a6eSdan     int i;
16755d607a6eSdan     pNext = pTab->pNext;
16765d607a6eSdan     for(i=0; i<pTab->nChange; i++){
16775d607a6eSdan       SessionChange *p;
167802d436b1Smistachkin       SessionChange *pNextChange;
167902d436b1Smistachkin       for(p=pTab->apChange[i]; p; p=pNextChange){
168002d436b1Smistachkin         pNextChange = p->pNext;
16815d607a6eSdan         sqlite3_free(p);
16825d607a6eSdan       }
16835d607a6eSdan     }
16845d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
16855d607a6eSdan     sqlite3_free(pTab->apChange);
16865d607a6eSdan     sqlite3_free(pTab);
16875d607a6eSdan   }
16885d607a6eSdan }
16895d607a6eSdan 
16904fccf43aSdan /*
16914fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
16924fccf43aSdan */
16934fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
16944fccf43aSdan   sqlite3 *db = pSession->db;
16954fccf43aSdan   sqlite3_session *pHead;
16964fccf43aSdan   sqlite3_session **pp;
16974fccf43aSdan 
1698296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1699296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
17004fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
17014fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
170250d348b1Sdrh   for(pp=&pHead; ALWAYS((*pp)!=0); pp=&((*pp)->pNext)){
170350d348b1Sdrh     if( (*pp)==pSession ){
17044fccf43aSdan       *pp = (*pp)->pNext;
17054fccf43aSdan       if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void*)pHead);
170650d348b1Sdrh       break;
170750d348b1Sdrh     }
170850d348b1Sdrh   }
17094fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
17101611e5a3Sdan   sqlite3ValueFree(pSession->pZeroBlob);
17114fccf43aSdan 
1712296c7658Sdan   /* Delete all attached table objects. And the contents of their
1713296c7658Sdan   ** associated hash-tables. */
17145d607a6eSdan   sessionDeleteTable(pSession->pTable);
17154fccf43aSdan 
1716296c7658Sdan   /* Free the session object itself. */
17174fccf43aSdan   sqlite3_free(pSession);
17184fccf43aSdan }
17194fccf43aSdan 
17204fccf43aSdan /*
17217531a5a3Sdan ** Set a table filter on a Session Object.
17227531a5a3Sdan */
17237531a5a3Sdan void sqlite3session_table_filter(
17247531a5a3Sdan   sqlite3_session *pSession,
17257531a5a3Sdan   int(*xFilter)(void*, const char*),
17267531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
17277531a5a3Sdan ){
17287531a5a3Sdan   pSession->bAutoAttach = 1;
17297531a5a3Sdan   pSession->pFilterCtx = pCtx;
17307531a5a3Sdan   pSession->xTableFilter = xFilter;
17317531a5a3Sdan }
17327531a5a3Sdan 
17337531a5a3Sdan /*
17344fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
17354fccf43aSdan ** while the session object is enabled will be recorded.
17364fccf43aSdan **
17374fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
17384fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
17394fccf43aSdan ** or not.
17404fccf43aSdan */
17414fccf43aSdan int sqlite3session_attach(
17424fccf43aSdan   sqlite3_session *pSession,      /* Session object */
17434fccf43aSdan   const char *zName               /* Table name */
17444fccf43aSdan ){
1745ff4d0f41Sdan   int rc = SQLITE_OK;
1746ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1747ff4d0f41Sdan 
1748ff4d0f41Sdan   if( !zName ){
1749ff4d0f41Sdan     pSession->bAutoAttach = 1;
1750ff4d0f41Sdan   }else{
1751296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1752296c7658Sdan     int nName;                    /* Number of bytes in string zName */
17534fccf43aSdan 
17544fccf43aSdan     /* First search for an existing entry. If one is found, this call is
17554fccf43aSdan     ** a no-op. Return early. */
1756cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
17574fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
17584c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
17594fccf43aSdan     }
17604fccf43aSdan 
17614c220252Sdan     if( !pTab ){
17624fccf43aSdan       /* Allocate new SessionTable object. */
17634fccf43aSdan       pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1);
17644c220252Sdan       if( !pTab ){
17654c220252Sdan         rc = SQLITE_NOMEM;
17664c220252Sdan       }else{
17676c39e6a8Sdan         /* Populate the new SessionTable object and link it into the list.
17686c39e6a8Sdan         ** The new object must be linked onto the end of the list, not
17696c39e6a8Sdan         ** simply added to the start of it in order to ensure that tables
17706c39e6a8Sdan         ** appear in the correct order when a changeset or patchset is
17716c39e6a8Sdan         ** eventually generated. */
17726c39e6a8Sdan         SessionTable **ppTab;
17734fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
17744fccf43aSdan         pTab->zName = (char *)&pTab[1];
17754fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
17766c39e6a8Sdan         for(ppTab=&pSession->pTable; *ppTab; ppTab=&(*ppTab)->pNext);
17776c39e6a8Sdan         *ppTab = pTab;
17784c220252Sdan       }
17794c220252Sdan     }
1780ff4d0f41Sdan   }
17814fccf43aSdan 
17824c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
17834c220252Sdan   return rc;
17844fccf43aSdan }
17854fccf43aSdan 
1786296c7658Sdan /*
1787296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1788296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1789296c7658Sdan **
1790296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1791296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1792296c7658Sdan */
17934fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){
179480fe2d93Sdan   if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){
17954fccf43aSdan     u8 *aNew;
17964fccf43aSdan     int nNew = p->nAlloc ? p->nAlloc : 128;
17974fccf43aSdan     do {
17984fccf43aSdan       nNew = nNew*2;
1799ef7a6304Sdan     }while( nNew<(p->nBuf+nByte) );
18004fccf43aSdan 
18014fccf43aSdan     aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew);
18024fccf43aSdan     if( 0==aNew ){
18034fccf43aSdan       *pRc = SQLITE_NOMEM;
180480fe2d93Sdan     }else{
18054fccf43aSdan       p->aBuf = aNew;
18064fccf43aSdan       p->nAlloc = nNew;
18074fccf43aSdan     }
180880fe2d93Sdan   }
180980fe2d93Sdan   return (*pRc!=SQLITE_OK);
18104fccf43aSdan }
18114fccf43aSdan 
1812296c7658Sdan /*
1813fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1814fa122adaSdan ** as the first.
1815fa122adaSdan **
1816fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1817fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1818fa122adaSdan ** before returning.
1819fa122adaSdan */
1820fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1821fa122adaSdan   int rc = *pRc;
1822fa122adaSdan   if( rc==SQLITE_OK ){
1823fa122adaSdan     int nByte = 0;
1824e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1825fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1826fa122adaSdan     if( rc==SQLITE_OK ){
1827fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1828fa122adaSdan       p->nBuf += nByte;
1829fa122adaSdan     }else{
1830fa122adaSdan       *pRc = rc;
1831fa122adaSdan     }
1832fa122adaSdan   }
1833fa122adaSdan }
1834fa122adaSdan 
1835fa122adaSdan /*
1836296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1837296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1838296c7658Sdan **
1839296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1840296c7658Sdan ** returning.
1841296c7658Sdan */
18424fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
184380fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
18444fccf43aSdan     p->aBuf[p->nBuf++] = v;
18454fccf43aSdan   }
18464fccf43aSdan }
18474fccf43aSdan 
1848296c7658Sdan /*
1849296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1850296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1851296c7658Sdan **
1852296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1853296c7658Sdan ** returning.
1854296c7658Sdan */
1855cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
185680fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
18574fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
18584fccf43aSdan   }
18594fccf43aSdan }
18604fccf43aSdan 
1861296c7658Sdan /*
1862296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1863296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1864296c7658Sdan **
1865296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1866296c7658Sdan ** returning.
1867296c7658Sdan */
18684fccf43aSdan static void sessionAppendBlob(
18694fccf43aSdan   SessionBuffer *p,
18704fccf43aSdan   const u8 *aBlob,
18714fccf43aSdan   int nBlob,
18724fccf43aSdan   int *pRc
18734fccf43aSdan ){
1874895decf6Sdan   if( nBlob>0 && 0==sessionBufferGrow(p, nBlob, pRc) ){
18754fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
18764fccf43aSdan     p->nBuf += nBlob;
18774fccf43aSdan   }
18784fccf43aSdan }
18794fccf43aSdan 
1880296c7658Sdan /*
1881296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1882296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1883296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1884296c7658Sdan **
1885296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1886296c7658Sdan ** returning.
1887296c7658Sdan */
1888d5f0767cSdan static void sessionAppendStr(
1889d5f0767cSdan   SessionBuffer *p,
1890d5f0767cSdan   const char *zStr,
1891d5f0767cSdan   int *pRc
1892d5f0767cSdan ){
1893cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
189480fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1895d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1896d5f0767cSdan     p->nBuf += nStr;
1897d5f0767cSdan   }
1898d5f0767cSdan }
1899d5f0767cSdan 
1900296c7658Sdan /*
1901296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1902296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1903296c7658Sdan ** to the buffer. No nul-terminator is written.
1904296c7658Sdan **
1905296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1906296c7658Sdan ** returning.
1907296c7658Sdan */
1908d5f0767cSdan static void sessionAppendInteger(
1909296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1910296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1911296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1912d5f0767cSdan ){
1913d5f0767cSdan   char aBuf[24];
1914d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1915d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1916d5f0767cSdan }
1917d5f0767cSdan 
1918296c7658Sdan /*
1919296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1920296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1921296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1922296c7658Sdan ** nul-terminator byte is written.
1923296c7658Sdan **
1924296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1925296c7658Sdan ** returning.
1926296c7658Sdan */
1927d5f0767cSdan static void sessionAppendIdent(
1928296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1929296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1930296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1931d5f0767cSdan ){
1932cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
193380fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1934d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1935d5f0767cSdan     const char *zIn = zStr;
1936d5f0767cSdan     *zOut++ = '"';
1937d5f0767cSdan     while( *zIn ){
1938d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1939d5f0767cSdan       *zOut++ = *(zIn++);
1940d5f0767cSdan     }
1941d5f0767cSdan     *zOut++ = '"';
1942cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1943d5f0767cSdan   }
1944d5f0767cSdan }
1945d5f0767cSdan 
1946296c7658Sdan /*
1947296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1948296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1949296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1950296c7658Sdan ** to to the buffer.
1951296c7658Sdan */
19524fccf43aSdan static void sessionAppendCol(
1953296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1954296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1955296c7658Sdan   int iCol,                       /* Column to read value from */
1956296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
19574fccf43aSdan ){
19584fccf43aSdan   if( *pRc==SQLITE_OK ){
19594fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
19604fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
19614fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
19624fccf43aSdan       sqlite3_int64 i;
19634fccf43aSdan       u8 aBuf[8];
19644fccf43aSdan       if( eType==SQLITE_INTEGER ){
19654fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
19664fccf43aSdan       }else{
19674fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
19684fccf43aSdan         memcpy(&i, &r, 8);
19694fccf43aSdan       }
1970296c7658Sdan       sessionPutI64(aBuf, i);
19714fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
19724fccf43aSdan     }
19734fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
19746734007dSdan       u8 *z;
19753cc89d95Sdan       int nByte;
19766734007dSdan       if( eType==SQLITE_BLOB ){
19776734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
19786734007dSdan       }else{
19796734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
19806734007dSdan       }
19813cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
19823cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
19834fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
19846734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
19856734007dSdan       }else{
19866734007dSdan         *pRc = SQLITE_NOMEM;
19876734007dSdan       }
19884fccf43aSdan     }
19894fccf43aSdan   }
19904fccf43aSdan }
19914fccf43aSdan 
1992296c7658Sdan /*
1993296c7658Sdan **
199480fe2d93Sdan ** This function appends an update change to the buffer (see the comments
199580fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
199680fe2d93Sdan ** consists of:
1997296c7658Sdan **
1998296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
1999296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
2000296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
2001296c7658Sdan **
2002296c7658Sdan ** The SessionChange object passed as the third argument contains the
2003296c7658Sdan ** values that were stored in the row when the session began (the old.*
2004296c7658Sdan ** values). The statement handle passed as the second argument points
2005296c7658Sdan ** at the current version of the row (the new.* values).
2006296c7658Sdan **
2007296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
2008296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
2009296c7658Sdan **
2010296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
2011296c7658Sdan ** original values of any fields that have been modified. The new.* record
2012296c7658Sdan ** contains the new values of only those fields that have been modified.
2013296c7658Sdan */
201480fe2d93Sdan static int sessionAppendUpdate(
2015296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
201673b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
2017296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
2018296c7658Sdan   SessionChange *p,               /* Object containing old values */
201980fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
20204fccf43aSdan ){
202180fe2d93Sdan   int rc = SQLITE_OK;
2022296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
2023296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
20241f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
2025296c7658Sdan   int i;                        /* Used to iterate through columns */
2026296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
2027296c7658Sdan 
202880fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
202980fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
20304fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
203137f133ecSdan     int bChanged = 0;
20324fccf43aSdan     int nAdvance;
20334fccf43aSdan     int eType = *pCsr;
20344fccf43aSdan     switch( eType ){
20354fccf43aSdan       case SQLITE_NULL:
20364fccf43aSdan         nAdvance = 1;
20374fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
203837f133ecSdan           bChanged = 1;
20394fccf43aSdan         }
20404fccf43aSdan         break;
20414fccf43aSdan 
20424fccf43aSdan       case SQLITE_FLOAT:
20434fccf43aSdan       case SQLITE_INTEGER: {
20444fccf43aSdan         nAdvance = 9;
20454fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
20464fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
20474fccf43aSdan           if( eType==SQLITE_INTEGER ){
20484fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
20494fccf43aSdan           }else{
20504fccf43aSdan             double dVal;
20514fccf43aSdan             memcpy(&dVal, &iVal, 8);
20524fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
20534fccf43aSdan           }
20544fccf43aSdan         }
205537f133ecSdan         bChanged = 1;
20564fccf43aSdan         break;
20574fccf43aSdan       }
20584fccf43aSdan 
2059e5754eecSdan       default: {
2060895decf6Sdan         int n;
2061895decf6Sdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &n);
2062e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
2063895decf6Sdan         nAdvance = nHdr + n;
20644fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
2065895decf6Sdan          && n==sqlite3_column_bytes(pStmt, i)
2066895decf6Sdan          && (n==0 || 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), n))
20674fccf43aSdan         ){
20684fccf43aSdan           break;
20694fccf43aSdan         }
207037f133ecSdan         bChanged = 1;
20714fccf43aSdan       }
20724fccf43aSdan     }
20734fccf43aSdan 
207473b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
207573b3c055Sdan     if( bChanged ) bNoop = 0;
207673b3c055Sdan 
207773b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
207873b3c055Sdan     ** currently generating a patchset. */
207973b3c055Sdan     if( bPatchset==0 ){
208037f133ecSdan       if( bChanged || abPK[i] ){
208180fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
20824fccf43aSdan       }else{
208380fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
208437f133ecSdan       }
208573b3c055Sdan     }
208637f133ecSdan 
208773b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
208873b3c055Sdan     ** generating a patchset.  */
208973b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
209080fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
209137f133ecSdan     }else{
209280fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
20934fccf43aSdan     }
209437f133ecSdan 
20954fccf43aSdan     pCsr += nAdvance;
20964fccf43aSdan   }
20974fccf43aSdan 
20984fccf43aSdan   if( bNoop ){
20991f34f8ccSdan     pBuf->nBuf = nRewind;
21004fccf43aSdan   }else{
210180fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
21024fccf43aSdan   }
21031f34f8ccSdan   sqlite3_free(buf2.aBuf);
210480fe2d93Sdan 
210580fe2d93Sdan   return rc;
2106d5f0767cSdan }
21074fccf43aSdan 
2108a71d2371Sdan /*
2109a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
2110a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
2111a71d2371Sdan ** format otherwise.
2112a71d2371Sdan */
211373b3c055Sdan static int sessionAppendDelete(
211473b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
211573b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
211673b3c055Sdan   SessionChange *p,               /* Object containing old values */
2117a71d2371Sdan   int nCol,                       /* Number of columns in table */
211873b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
211973b3c055Sdan ){
212073b3c055Sdan   int rc = SQLITE_OK;
212173b3c055Sdan 
212273b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
212373b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
212473b3c055Sdan 
212573b3c055Sdan   if( bPatchset==0 ){
212673b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
212773b3c055Sdan   }else{
212873b3c055Sdan     int i;
212973b3c055Sdan     u8 *a = p->aRecord;
213073b3c055Sdan     for(i=0; i<nCol; i++){
213173b3c055Sdan       u8 *pStart = a;
213273b3c055Sdan       int eType = *a++;
213373b3c055Sdan 
213473b3c055Sdan       switch( eType ){
213573b3c055Sdan         case 0:
213673b3c055Sdan         case SQLITE_NULL:
213773b3c055Sdan           assert( abPK[i]==0 );
213873b3c055Sdan           break;
213973b3c055Sdan 
214073b3c055Sdan         case SQLITE_FLOAT:
214173b3c055Sdan         case SQLITE_INTEGER:
214273b3c055Sdan           a += 8;
214373b3c055Sdan           break;
214473b3c055Sdan 
214573b3c055Sdan         default: {
214673b3c055Sdan           int n;
214773b3c055Sdan           a += sessionVarintGet(a, &n);
214873b3c055Sdan           a += n;
214973b3c055Sdan           break;
215073b3c055Sdan         }
215173b3c055Sdan       }
215273b3c055Sdan       if( abPK[i] ){
2153f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
215473b3c055Sdan       }
215573b3c055Sdan     }
215673b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
215773b3c055Sdan   }
215873b3c055Sdan 
215973b3c055Sdan   return rc;
216073b3c055Sdan }
216173b3c055Sdan 
216277fc1d5bSdan /*
216377fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
216477fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
216577fc1d5bSdan **
216677fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
216777fc1d5bSdan */
2168e8d5648eSdan static int sessionSelectStmt(
2169e8d5648eSdan   sqlite3 *db,                    /* Database handle */
2170d7fb7d24Sdan   const char *zDb,                /* Database name */
2171e8d5648eSdan   const char *zTab,               /* Table name */
217277fc1d5bSdan   int nCol,                       /* Number of columns in table */
217377fc1d5bSdan   const char **azCol,             /* Names of table columns */
217477fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
217577fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
2176d5f0767cSdan ){
2177e8d5648eSdan   int rc = SQLITE_OK;
21783739f298Sdan   char *zSql = 0;
21793739f298Sdan   int nSql = -1;
21803739f298Sdan 
21813739f298Sdan   if( 0==sqlite3_stricmp("sqlite_stat1", zTab) ){
21823739f298Sdan     zSql = sqlite3_mprintf(
21833739f298Sdan         "SELECT tbl, ?2, stat FROM %Q.sqlite_stat1 WHERE tbl IS ?1 AND "
21843739f298Sdan         "idx IS (CASE WHEN ?2=X'' THEN NULL ELSE ?2 END)", zDb
21853739f298Sdan     );
2186bd45374cSdan     if( zSql==0 ) rc = SQLITE_NOMEM;
21873739f298Sdan   }else{
2188d5f0767cSdan     int i;
2189e8d5648eSdan     const char *zSep = "";
2190e8d5648eSdan     SessionBuffer buf = {0, 0, 0};
2191d5f0767cSdan 
2192e8d5648eSdan     sessionAppendStr(&buf, "SELECT * FROM ", &rc);
2193d7fb7d24Sdan     sessionAppendIdent(&buf, zDb, &rc);
2194d7fb7d24Sdan     sessionAppendStr(&buf, ".", &rc);
2195e8d5648eSdan     sessionAppendIdent(&buf, zTab, &rc);
2196e8d5648eSdan     sessionAppendStr(&buf, " WHERE ", &rc);
2197e8d5648eSdan     for(i=0; i<nCol; i++){
2198e8d5648eSdan       if( abPK[i] ){
2199e8d5648eSdan         sessionAppendStr(&buf, zSep, &rc);
2200e8d5648eSdan         sessionAppendIdent(&buf, azCol[i], &rc);
22013739f298Sdan         sessionAppendStr(&buf, " IS ?", &rc);
2202e8d5648eSdan         sessionAppendInteger(&buf, i+1, &rc);
2203e8d5648eSdan         zSep = " AND ";
2204d5f0767cSdan       }
2205d5f0767cSdan     }
22063739f298Sdan     zSql = (char*)buf.aBuf;
22073739f298Sdan     nSql = buf.nBuf;
2208d5f0767cSdan   }
22093739f298Sdan 
22103739f298Sdan   if( rc==SQLITE_OK ){
22113739f298Sdan     rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, 0);
22123739f298Sdan   }
22133739f298Sdan   sqlite3_free(zSql);
2214e8d5648eSdan   return rc;
2215d5f0767cSdan }
2216d5f0767cSdan 
221777fc1d5bSdan /*
221877fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
221977fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
222077fc1d5bSdan ** is as prepared by function sessionSelectStmt().
222177fc1d5bSdan **
222277fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
222377fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
222477fc1d5bSdan */
2225e8d5648eSdan static int sessionSelectBind(
222677fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
222777fc1d5bSdan   int nCol,                       /* Number of columns in table */
222877fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
222977fc1d5bSdan   SessionChange *pChange          /* Change structure */
2230e8d5648eSdan ){
2231e8d5648eSdan   int i;
2232e8d5648eSdan   int rc = SQLITE_OK;
2233e5754eecSdan   u8 *a = pChange->aRecord;
2234d5f0767cSdan 
2235e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2236e8d5648eSdan     int eType = *a++;
2237e8d5648eSdan 
2238e8d5648eSdan     switch( eType ){
223980fe2d93Sdan       case 0:
2240e8d5648eSdan       case SQLITE_NULL:
22411611e5a3Sdan         assert( abPK[i]==0 );
2242e8d5648eSdan         break;
2243e8d5648eSdan 
2244e8d5648eSdan       case SQLITE_INTEGER: {
2245e8d5648eSdan         if( abPK[i] ){
2246e8d5648eSdan           i64 iVal = sessionGetI64(a);
2247e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
2248e8d5648eSdan         }
2249e8d5648eSdan         a += 8;
2250e8d5648eSdan         break;
2251d5f0767cSdan       }
2252296c7658Sdan 
2253e8d5648eSdan       case SQLITE_FLOAT: {
2254e8d5648eSdan         if( abPK[i] ){
2255e8d5648eSdan           double rVal;
2256e8d5648eSdan           i64 iVal = sessionGetI64(a);
2257e8d5648eSdan           memcpy(&rVal, &iVal, 8);
22584e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
2259d5f0767cSdan         }
2260e8d5648eSdan         a += 8;
2261e8d5648eSdan         break;
2262e8d5648eSdan       }
2263e8d5648eSdan 
2264e8d5648eSdan       case SQLITE_TEXT: {
2265e8d5648eSdan         int n;
2266e8d5648eSdan         a += sessionVarintGet(a, &n);
2267e8d5648eSdan         if( abPK[i] ){
2268e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
2269e8d5648eSdan         }
2270e8d5648eSdan         a += n;
2271e8d5648eSdan         break;
2272e8d5648eSdan       }
2273e8d5648eSdan 
2274e5754eecSdan       default: {
2275e8d5648eSdan         int n;
2276e5754eecSdan         assert( eType==SQLITE_BLOB );
2277e8d5648eSdan         a += sessionVarintGet(a, &n);
2278e8d5648eSdan         if( abPK[i] ){
2279e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
2280e8d5648eSdan         }
2281e8d5648eSdan         a += n;
2282e8d5648eSdan         break;
2283e8d5648eSdan       }
2284e8d5648eSdan     }
2285e8d5648eSdan   }
2286e8d5648eSdan 
2287d5f0767cSdan   return rc;
22884fccf43aSdan }
22894fccf43aSdan 
229077fc1d5bSdan /*
229177fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
229277fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
229377fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
229477fc1d5bSdan ** SQLite error code before returning.
229577fc1d5bSdan */
22965d607a6eSdan static void sessionAppendTableHdr(
2297a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
2298a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
2299a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
2300a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
23015d607a6eSdan ){
23025d607a6eSdan   /* Write a table header */
230373b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
23045d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
23055d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
23064f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
23075d607a6eSdan }
23085d607a6eSdan 
2309a71d2371Sdan /*
2310a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
2311a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
2312a71d2371Sdan ** passed as the first argument.
2313a71d2371Sdan **
2314a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
2315a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
2316a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
2317a71d2371Sdan ** to 0.
2318a71d2371Sdan */
2319adf3bf58Sdrh static int sessionGenerateChangeset(
23204fccf43aSdan   sqlite3_session *pSession,      /* Session object */
232173b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
2322ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2323ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
23244fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
23254fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
23264fccf43aSdan ){
2327296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
2328296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
2329296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
2330296c7658Sdan   int rc;                         /* Return code */
23314fccf43aSdan 
2332ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
2333ef7a6304Sdan 
2334296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
2335296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
2336296c7658Sdan   ** this call will be a no-op.  */
2337ef7a6304Sdan   if( xOutput==0 ){
23384fccf43aSdan     *pnChangeset = 0;
23394fccf43aSdan     *ppChangeset = 0;
2340ef7a6304Sdan   }
2341e5754eecSdan 
2342e5754eecSdan   if( pSession->rc ) return pSession->rc;
2343e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
2344e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
2345e5754eecSdan 
2346e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
23474fccf43aSdan 
23484fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
23494fccf43aSdan     if( pTab->nEntry ){
2350d7fb7d24Sdan       const char *zName = pTab->zName;
2351a9605b91Sdan       int nCol;                   /* Number of columns in table */
2352a9605b91Sdan       u8 *abPK;                   /* Primary key array */
2353a9605b91Sdan       const char **azCol = 0;     /* Table columns */
23541f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
23551f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
23561f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
23571f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
23584fccf43aSdan 
2359a9605b91Sdan       /* Check the table schema is still Ok. */
2360a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
2361a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
2362a9605b91Sdan         rc = SQLITE_SCHEMA;
2363a9605b91Sdan       }
2364a9605b91Sdan 
23654fccf43aSdan       /* Write a table header */
236673b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
23674fccf43aSdan 
23684fccf43aSdan       /* Build and compile a statement to execute: */
23694fccf43aSdan       if( rc==SQLITE_OK ){
2370d7fb7d24Sdan         rc = sessionSelectStmt(
2371a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
23724fccf43aSdan       }
23734fccf43aSdan 
23741f34f8ccSdan       nNoop = buf.nBuf;
237512ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
2376e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
2377e8d5648eSdan 
23784fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
2379e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
238080fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
23811f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
2382798693b2Sdan             if( p->op==SQLITE_INSERT ){
23834fccf43aSdan               int iCol;
23844fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
2385b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
2386e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
23871f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
23884fccf43aSdan               }
2389e8d5648eSdan             }else{
239073b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
23914fccf43aSdan             }
2392798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
2393a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
23944fccf43aSdan           }
239512ca0b56Sdan           if( rc==SQLITE_OK ){
23961f34f8ccSdan             rc = sqlite3_reset(pSel);
23974fccf43aSdan           }
2398ef7a6304Sdan 
2399f1a08ad8Sdrh           /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass
2400ef7a6304Sdan           ** its contents to the xOutput() callback. */
2401ef7a6304Sdan           if( xOutput
2402ef7a6304Sdan            && rc==SQLITE_OK
2403ef7a6304Sdan            && buf.nBuf>nNoop
2404f1a08ad8Sdrh            && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE
2405ef7a6304Sdan           ){
2406ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2407ef7a6304Sdan             nNoop = -1;
2408ef7a6304Sdan             buf.nBuf = 0;
2409ef7a6304Sdan           }
2410ef7a6304Sdan 
24114fccf43aSdan         }
2412e8d5648eSdan       }
24134fccf43aSdan 
24141f34f8ccSdan       sqlite3_finalize(pSel);
24151f34f8ccSdan       if( buf.nBuf==nNoop ){
24164fccf43aSdan         buf.nBuf = nRewind;
24174fccf43aSdan       }
2418cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
24194fccf43aSdan     }
24204fccf43aSdan   }
24214fccf43aSdan 
24224fccf43aSdan   if( rc==SQLITE_OK ){
2423ef7a6304Sdan     if( xOutput==0 ){
24244fccf43aSdan       *pnChangeset = buf.nBuf;
24254fccf43aSdan       *ppChangeset = buf.aBuf;
2426ef7a6304Sdan       buf.aBuf = 0;
2427ef7a6304Sdan     }else if( buf.nBuf>0 ){
2428ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
2429ef7a6304Sdan     }
24304fccf43aSdan   }
24314c220252Sdan 
2432ef7a6304Sdan   sqlite3_free(buf.aBuf);
2433e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
24344c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
24354fccf43aSdan   return rc;
24364fccf43aSdan }
24374fccf43aSdan 
2438296c7658Sdan /*
243973b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
244073b3c055Sdan ** session object passed as the first argument.
244173b3c055Sdan **
244273b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
244373b3c055Sdan ** using sqlite3_free().
244473b3c055Sdan */
244573b3c055Sdan int sqlite3session_changeset(
244673b3c055Sdan   sqlite3_session *pSession,      /* Session object */
244773b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
244873b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
244973b3c055Sdan ){
2450ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2451ef7a6304Sdan }
2452ef7a6304Sdan 
2453ef7a6304Sdan /*
2454ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2455ef7a6304Sdan */
2456f1a08ad8Sdrh int sqlite3session_changeset_strm(
2457ef7a6304Sdan   sqlite3_session *pSession,
2458ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2459ef7a6304Sdan   void *pOut
2460ef7a6304Sdan ){
2461ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2462ef7a6304Sdan }
2463ef7a6304Sdan 
2464ef7a6304Sdan /*
2465ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2466ef7a6304Sdan */
2467f1a08ad8Sdrh int sqlite3session_patchset_strm(
2468ef7a6304Sdan   sqlite3_session *pSession,
2469ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2470ef7a6304Sdan   void *pOut
2471ef7a6304Sdan ){
2472ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
247373b3c055Sdan }
247473b3c055Sdan 
247573b3c055Sdan /*
247673b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
247773b3c055Sdan ** session object passed as the first argument.
247873b3c055Sdan **
247973b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
248073b3c055Sdan ** using sqlite3_free().
248173b3c055Sdan */
248273b3c055Sdan int sqlite3session_patchset(
248373b3c055Sdan   sqlite3_session *pSession,      /* Session object */
248473b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
248573b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
248673b3c055Sdan ){
2487ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
248873b3c055Sdan }
248973b3c055Sdan 
249073b3c055Sdan /*
2491296c7658Sdan ** Enable or disable the session object passed as the first argument.
2492296c7658Sdan */
24934fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
24944c220252Sdan   int ret;
24954c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2496296c7658Sdan   if( bEnable>=0 ){
2497296c7658Sdan     pSession->bEnable = bEnable;
24984fccf43aSdan   }
24994c220252Sdan   ret = pSession->bEnable;
25004c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
25014c220252Sdan   return ret;
2502296c7658Sdan }
25034fccf43aSdan 
25044fccf43aSdan /*
2505b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2506b4480e94Sdan */
2507b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2508b4480e94Sdan   int ret;
2509b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2510b4480e94Sdan   if( bIndirect>=0 ){
2511b4480e94Sdan     pSession->bIndirect = bIndirect;
2512b4480e94Sdan   }
2513b4480e94Sdan   ret = pSession->bIndirect;
2514b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2515b4480e94Sdan   return ret;
2516b4480e94Sdan }
2517b4480e94Sdan 
2518b4480e94Sdan /*
2519b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2520b69ec348Sdan ** by the session object passed as the only argument.
2521b69ec348Sdan */
2522b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2523b69ec348Sdan   int ret = 0;
2524b69ec348Sdan   SessionTable *pTab;
2525b69ec348Sdan 
2526b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2527b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2528b69ec348Sdan     ret = (pTab->nEntry>0);
2529b69ec348Sdan   }
2530b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2531b69ec348Sdan 
2532ff530326Sdan   return (ret==0);
2533b69ec348Sdan }
2534b69ec348Sdan 
2535b69ec348Sdan /*
2536f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
25374fccf43aSdan */
2538adf3bf58Sdrh static int sessionChangesetStart(
2539296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2540ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2541ef7a6304Sdan   void *pIn,
2542296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2543296c7658Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
25444fccf43aSdan ){
25454fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
25464fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
25474fccf43aSdan 
2548ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2549ef7a6304Sdan 
2550296c7658Sdan   /* Zero the output variable in case an error occurs. */
2551296c7658Sdan   *pp = 0;
25524fccf43aSdan 
2553296c7658Sdan   /* Allocate and initialize the iterator structure. */
25544fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
25554fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
25564fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
25574fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
25584757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
25594757c658Sdan   pRet->in.nData = nChangeset;
2560ef7a6304Sdan   pRet->in.xInput = xInput;
2561ef7a6304Sdan   pRet->in.pIn = pIn;
2562ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
25634fccf43aSdan 
2564296c7658Sdan   /* Populate the output variable and return success. */
2565296c7658Sdan   *pp = pRet;
25664fccf43aSdan   return SQLITE_OK;
25674fccf43aSdan }
25684fccf43aSdan 
2569296c7658Sdan /*
2570ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2571ef7a6304Sdan */
2572ef7a6304Sdan int sqlite3changeset_start(
2573ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2574ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2575ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2576ef7a6304Sdan ){
2577ef7a6304Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset);
2578ef7a6304Sdan }
2579ef7a6304Sdan 
2580ef7a6304Sdan /*
2581ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2582ef7a6304Sdan */
2583f1a08ad8Sdrh int sqlite3changeset_start_strm(
2584ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2585ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2586ef7a6304Sdan   void *pIn
2587ef7a6304Sdan ){
2588ef7a6304Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0);
2589ef7a6304Sdan }
2590ef7a6304Sdan 
2591ef7a6304Sdan /*
2592d9151526Sdan ** If the SessionInput object passed as the only argument is a streaming
2593d9151526Sdan ** object and the buffer is full, discard some data to free up space.
2594d9151526Sdan */
2595d9151526Sdan static void sessionDiscardData(SessionInput *pIn){
2596d9151526Sdan   if( pIn->bEof && pIn->xInput && pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){
2597d9151526Sdan     int nMove = pIn->buf.nBuf - pIn->iNext;
2598d9151526Sdan     assert( nMove>=0 );
2599d9151526Sdan     if( nMove>0 ){
2600d9151526Sdan       memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
2601d9151526Sdan     }
2602d9151526Sdan     pIn->buf.nBuf -= pIn->iNext;
2603d9151526Sdan     pIn->iNext = 0;
2604d9151526Sdan     pIn->nData = pIn->buf.nBuf;
2605d9151526Sdan   }
2606d9151526Sdan }
2607d9151526Sdan 
2608d9151526Sdan /*
2609ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2610ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2611ef7a6304Sdan ** data is in the buffer.
2612ef7a6304Sdan **
2613ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2614ef7a6304Sdan */
26154757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2616ef7a6304Sdan   int rc = SQLITE_OK;
26174757c658Sdan   if( pIn->xInput ){
26184757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
2619f1a08ad8Sdrh       int nNew = SESSIONS_STRM_CHUNK_SIZE;
26204757c658Sdan 
2621d9151526Sdan       if( pIn->bNoDiscard==0 ) sessionDiscardData(pIn);
26224757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
26234757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
26244757c658Sdan         if( nNew==0 ){
26254757c658Sdan           pIn->bEof = 1;
26264757c658Sdan         }else{
26274757c658Sdan           pIn->buf.nBuf += nNew;
26284757c658Sdan         }
26294757c658Sdan       }
26304757c658Sdan 
26314757c658Sdan       pIn->aData = pIn->buf.aBuf;
26324757c658Sdan       pIn->nData = pIn->buf.nBuf;
26334757c658Sdan     }
2634ef7a6304Sdan   }
2635ef7a6304Sdan   return rc;
2636ef7a6304Sdan }
2637ef7a6304Sdan 
2638ef7a6304Sdan /*
2639ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2640ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2641ef7a6304Sdan ** until it points to the byte immediately following that record.
2642ef7a6304Sdan */
2643ef7a6304Sdan static void sessionSkipRecord(
2644ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2645ef7a6304Sdan   int nCol                        /* Number of values in record */
2646ef7a6304Sdan ){
2647ef7a6304Sdan   u8 *aRec = *ppRec;
2648ef7a6304Sdan   int i;
2649ef7a6304Sdan   for(i=0; i<nCol; i++){
2650ef7a6304Sdan     int eType = *aRec++;
2651ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2652ef7a6304Sdan       int nByte;
2653ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2654ef7a6304Sdan       aRec += nByte;
2655ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2656ef7a6304Sdan       aRec += 8;
2657ef7a6304Sdan     }
2658ef7a6304Sdan   }
2659ef7a6304Sdan 
2660ef7a6304Sdan   *ppRec = aRec;
2661ef7a6304Sdan }
2662ef7a6304Sdan 
2663ef7a6304Sdan /*
26644757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
26654757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
26664757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
26674757c658Sdan ** error occurs.
26684757c658Sdan */
26694757c658Sdan static int sessionValueSetStr(
26704757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
26714757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
26724757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
26734757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
26744757c658Sdan ){
267516228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
267616228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
267716228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
267816228167Sdan   ** error. Hence the code to create the copy externally. */
26793cc89d95Sdan   u8 *aCopy = sqlite3_malloc(nData+1);
26804757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
26814757c658Sdan   memcpy(aCopy, aData, nData);
26824757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
26834757c658Sdan   return SQLITE_OK;
26844757c658Sdan }
26854757c658Sdan 
26864757c658Sdan /*
2687296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2688296c7658Sdan ** for details.
2689296c7658Sdan **
2690296c7658Sdan ** When this function is called, *paChange points to the start of the record
2691296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2692296c7658Sdan ** one byte after the end of the same record before this function returns.
2693a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2694a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2695a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2696296c7658Sdan **
2697296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2698296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2699296c7658Sdan ** from the corresponding position in the record. If that value is not
2700296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2701296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2702296c7658Sdan ** set to NULL.
2703296c7658Sdan **
2704296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2705296c7658Sdan ** using sqlite3_free().
2706296c7658Sdan **
2707296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2708296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2709296c7658Sdan */
27104fccf43aSdan static int sessionReadRecord(
2711ef7a6304Sdan   SessionInput *pIn,              /* Input data */
27124fccf43aSdan   int nCol,                       /* Number of values in record */
271373b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
27144fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
27154fccf43aSdan ){
2716296c7658Sdan   int i;                          /* Used to iterate through columns */
2717ef7a6304Sdan   int rc = SQLITE_OK;
27184fccf43aSdan 
2719ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2720ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
272173b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2722ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2723ef7a6304Sdan     if( rc==SQLITE_OK ){
2724e341ec69Sdan       if( pIn->iNext>=pIn->nData ){
2725e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
2726e341ec69Sdan       }else{
27274757c658Sdan         eType = pIn->aData[pIn->iNext++];
2728e8fa8c96Sdan         assert( apOut[i]==0 );
27294fccf43aSdan         if( eType ){
27304fccf43aSdan           apOut[i] = sqlite3ValueNew(0);
2731ef7a6304Sdan           if( !apOut[i] ) rc = SQLITE_NOMEM;
2732ef7a6304Sdan         }
2733dd8a4af8Sdan       }
2734e341ec69Sdan     }
27354fccf43aSdan 
2736ef7a6304Sdan     if( rc==SQLITE_OK ){
27374757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
27384fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
27394fccf43aSdan         int nByte;
2740ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2741ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2742e8fa8c96Sdan         if( rc==SQLITE_OK ){
2743e341ec69Sdan           if( nByte<0 || nByte>pIn->nData-pIn->iNext ){
2744e341ec69Sdan             rc = SQLITE_CORRUPT_BKPT;
2745e341ec69Sdan           }else{
27466734007dSdan             u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
27474757c658Sdan             rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
2748ef7a6304Sdan             pIn->iNext += nByte;
27494fccf43aSdan           }
2750e341ec69Sdan         }
2751e341ec69Sdan       }
27524fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2753ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
27544fccf43aSdan         if( eType==SQLITE_INTEGER ){
27554fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
27564fccf43aSdan         }else{
27574fccf43aSdan           double d;
27584e895da1Sdan           memcpy(&d, &v, 8);
27594fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
27604fccf43aSdan         }
2761ef7a6304Sdan         pIn->iNext += 8;
276291ddd559Sdan       }
27634fccf43aSdan     }
27644fccf43aSdan   }
27654fccf43aSdan 
2766ef7a6304Sdan   return rc;
2767ef7a6304Sdan }
2768ef7a6304Sdan 
2769ef7a6304Sdan /*
2770ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2771ef7a6304Sdan ** Specifically, to the following:
2772ef7a6304Sdan **
2773ef7a6304Sdan **   + number of columns in table (varint)
2774ef7a6304Sdan **   + array of PK flags (1 byte per column),
2775ef7a6304Sdan **   + table name (nul terminated).
2776ef7a6304Sdan **
2777ef7a6304Sdan ** This function ensures that all of the above is present in the input
2778ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2779ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2780ef7a6304Sdan ** The input pointer is not moved.
2781ef7a6304Sdan */
2782ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2783ef7a6304Sdan   int rc = SQLITE_OK;
2784ef7a6304Sdan   int nCol = 0;
27854757c658Sdan   int nRead = 0;
2786ef7a6304Sdan 
2787ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2788ef7a6304Sdan   if( rc==SQLITE_OK ){
27894757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
27906344eddaSdan     /* The hard upper limit for the number of columns in an SQLite
27916344eddaSdan     ** database table is, according to sqliteLimit.h, 32676. So
27926344eddaSdan     ** consider any table-header that purports to have more than 65536
27936344eddaSdan     ** columns to be corrupt. This is convenient because otherwise,
27946344eddaSdan     ** if the (nCol>65536) condition below were omitted, a sufficiently
27956344eddaSdan     ** large value for nCol may cause nRead to wrap around and become
27966344eddaSdan     ** negative. Leading to a crash. */
27976344eddaSdan     if( nCol<0 || nCol>65536 ){
2798e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2799e341ec69Sdan     }else{
28004757c658Sdan       rc = sessionInputBuffer(pIn, nRead+nCol+100);
28014757c658Sdan       nRead += nCol;
2802ef7a6304Sdan     }
2803e341ec69Sdan   }
28044757c658Sdan 
2805ef7a6304Sdan   while( rc==SQLITE_OK ){
28064757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
28074757c658Sdan       nRead++;
2808ef7a6304Sdan     }
2809e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
28104757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
28114757c658Sdan   }
2812e8fa8c96Sdan   *pnByte = nRead+1;
2813ef7a6304Sdan   return rc;
2814ef7a6304Sdan }
2815ef7a6304Sdan 
2816ef7a6304Sdan /*
2817fa122adaSdan ** The input pointer currently points to the first byte of the first field
2818fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
281916228167Sdan ** record is buffered. It does not move the input pointer.
282016228167Sdan **
282116228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
282216228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
282316228167Sdan ** final value of *pnByte is undefined in this case.
2824fa122adaSdan */
2825fa122adaSdan static int sessionChangesetBufferRecord(
282616228167Sdan   SessionInput *pIn,              /* Input data */
282716228167Sdan   int nCol,                       /* Number of columns in record */
282816228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2829fa122adaSdan ){
2830fa122adaSdan   int rc = SQLITE_OK;
2831fa122adaSdan   int nByte = 0;
2832fa122adaSdan   int i;
2833fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2834fa122adaSdan     int eType;
2835fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2836fa122adaSdan     if( rc==SQLITE_OK ){
2837fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2838fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2839fa122adaSdan         int n;
2840fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2841fa122adaSdan         nByte += n;
2842fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2843fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2844fa122adaSdan         nByte += 8;
2845fa122adaSdan       }
2846fa122adaSdan     }
2847fa122adaSdan   }
2848fa122adaSdan   *pnByte = nByte;
2849fa122adaSdan   return rc;
2850fa122adaSdan }
2851fa122adaSdan 
2852fa122adaSdan /*
2853ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2854ef7a6304Sdan ** Specifically, to the following:
2855ef7a6304Sdan **
2856ef7a6304Sdan **   + number of columns in table (varint)
2857ef7a6304Sdan **   + array of PK flags (1 byte per column),
2858ef7a6304Sdan **   + table name (nul terminated).
285916228167Sdan **
286016228167Sdan ** This function decodes the table-header and populates the p->nCol,
286116228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
286216228167Sdan ** also allocated or resized according to the new value of p->nCol. The
286316228167Sdan ** input pointer is left pointing to the byte following the table header.
286416228167Sdan **
286516228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
286616228167Sdan ** is returned and the final values of the various fields enumerated above
286716228167Sdan ** are undefined.
2868ef7a6304Sdan */
2869ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2870ef7a6304Sdan   int rc;
2871ef7a6304Sdan   int nCopy;
2872ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2873ef7a6304Sdan 
2874ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2875ef7a6304Sdan   if( rc==SQLITE_OK ){
2876ef7a6304Sdan     int nByte;
2877ef7a6304Sdan     int nVarint;
28784757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2879dd8a4af8Sdan     if( p->nCol>0 ){
2880ef7a6304Sdan       nCopy -= nVarint;
2881ef7a6304Sdan       p->in.iNext += nVarint;
2882ef7a6304Sdan       nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2883ef7a6304Sdan       p->tblhdr.nBuf = 0;
2884ef7a6304Sdan       sessionBufferGrow(&p->tblhdr, nByte, &rc);
2885dd8a4af8Sdan     }else{
2886e341ec69Sdan       rc = SQLITE_CORRUPT_BKPT;
2887dd8a4af8Sdan     }
2888ef7a6304Sdan   }
2889ef7a6304Sdan 
2890ef7a6304Sdan   if( rc==SQLITE_OK ){
2891ef7a6304Sdan     int iPK = sizeof(sqlite3_value*)*p->nCol*2;
2892ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
28934757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2894ef7a6304Sdan     p->in.iNext += nCopy;
2895ef7a6304Sdan   }
2896ef7a6304Sdan 
2897ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2898ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2899ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2900ef7a6304Sdan   return (p->rc = rc);
29014fccf43aSdan }
29024fccf43aSdan 
290377fc1d5bSdan /*
290477fc1d5bSdan ** Advance the changeset iterator to the next change.
290577fc1d5bSdan **
290677fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
290777fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
290877fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
290977fc1d5bSdan **
291077fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
291177fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
291277fc1d5bSdan ** the record to *pnRec.
291377fc1d5bSdan **
291477fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
291577fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
291677fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
291777fc1d5bSdan ** changes in the changeset.
291877fc1d5bSdan */
29195d607a6eSdan static int sessionChangesetNext(
292077fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
292177fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
2922c0a499eaSdan   int *pnRec,                     /* If non-NULL, store size of record here */
2923c0a499eaSdan   int *pbNew                      /* If non-NULL, true if new table */
29245d607a6eSdan ){
29254fccf43aSdan   int i;
2926ef7a6304Sdan   u8 op;
29274fccf43aSdan 
29285d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
29295d607a6eSdan 
2930296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
29314fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
29324fccf43aSdan 
29335d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
29344fccf43aSdan   if( p->apValue ){
29354fccf43aSdan     for(i=0; i<p->nCol*2; i++){
29364fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
29374fccf43aSdan     }
29384fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
29394fccf43aSdan   }
29404fccf43aSdan 
2941ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
2942ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
2943ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
2944ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
2945ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
2946ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
2947ef7a6304Sdan 
29484fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
29494757c658Sdan   if( p->in.iNext>=p->in.nData ){
29504fccf43aSdan     return SQLITE_DONE;
29514fccf43aSdan   }
29524fccf43aSdan 
2953d9151526Sdan   sessionDiscardData(&p->in);
2954d9151526Sdan   p->in.iCurrent = p->in.iNext;
2955d9151526Sdan 
29564757c658Sdan   op = p->in.aData[p->in.iNext++];
295707d0f15eSdan   while( op=='T' || op=='P' ){
2958c0a499eaSdan     if( pbNew ) *pbNew = 1;
2959ef7a6304Sdan     p->bPatchset = (op=='P');
2960ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
2961ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
2962d9151526Sdan     p->in.iCurrent = p->in.iNext;
296307d0f15eSdan     if( p->in.iNext>=p->in.nData ) return SQLITE_DONE;
29644757c658Sdan     op = p->in.aData[p->in.iNext++];
29655d607a6eSdan   }
29665d607a6eSdan 
2967dd8a4af8Sdan   if( p->zTab==0 ){
2968dd8a4af8Sdan     /* The first record in the changeset is not a table header. Must be a
2969dd8a4af8Sdan     ** corrupt changeset. */
2970dd8a4af8Sdan     assert( p->in.iNext==1 );
2971dd8a4af8Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
2972dd8a4af8Sdan   }
2973dd8a4af8Sdan 
2974ef7a6304Sdan   p->op = op;
29754757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
29764fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
29774757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
29784fccf43aSdan   }
29794fccf43aSdan 
2980cbf6d2d2Sdan   if( paRec ){
2981cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
2982cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
2983cbf6d2d2Sdan       nVal = p->nCol * 2;
2984cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
2985cbf6d2d2Sdan       nVal = 0;
2986cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
2987cbf6d2d2Sdan     }else{
2988cbf6d2d2Sdan       nVal = p->nCol;
2989cbf6d2d2Sdan     }
2990cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
2991cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
2992cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
2993cbf6d2d2Sdan     p->in.iNext += *pnRec;
2994cbf6d2d2Sdan   }else{
29955d607a6eSdan 
29964fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
299773b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
299873b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
2999cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue);
30004fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30014fccf43aSdan     }
30024fccf43aSdan 
30034fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
30044fccf43aSdan     if( p->op!=SQLITE_DELETE ){
3005cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]);
30064fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
30074fccf43aSdan     }
30084fccf43aSdan 
3009cbf6d2d2Sdan     if( p->bPatchset && p->op==SQLITE_UPDATE ){
301073b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
301173b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
301273b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
301373b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
301473b3c055Sdan       for(i=0; i<p->nCol; i++){
301573b3c055Sdan         assert( p->apValue[i]==0 );
301673b3c055Sdan         if( p->abPK[i] ){
301773b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
3018e341ec69Sdan           if( p->apValue[i]==0 ) return (p->rc = SQLITE_CORRUPT_BKPT);
301973b3c055Sdan           p->apValue[i+p->nCol] = 0;
302073b3c055Sdan         }
302173b3c055Sdan       }
302273b3c055Sdan     }
3023cbf6d2d2Sdan   }
3024ef7a6304Sdan 
30254fccf43aSdan   return SQLITE_ROW;
30264fccf43aSdan }
30274fccf43aSdan 
30284fccf43aSdan /*
30295d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
30305d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
30315d607a6eSdan ** or SQLITE_CORRUPT.
30325d607a6eSdan **
30335d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
30345d607a6eSdan ** callback by changeset_apply().
30355d607a6eSdan */
30365d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
3037c0a499eaSdan   return sessionChangesetNext(p, 0, 0, 0);
30385d607a6eSdan }
30395d607a6eSdan 
30405d607a6eSdan /*
3041244593c8Sdan ** The following function extracts information on the current change
304277fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
30434fccf43aSdan ** has returned SQLITE_ROW.
30444fccf43aSdan */
30454fccf43aSdan int sqlite3changeset_op(
3046296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
30474fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
30484fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
3049b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
3050b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
30514fccf43aSdan ){
30524fccf43aSdan   *pOp = pIter->op;
30534fccf43aSdan   *pnCol = pIter->nCol;
30544fccf43aSdan   *pzTab = pIter->zTab;
3055b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
30564fccf43aSdan   return SQLITE_OK;
30574fccf43aSdan }
30584fccf43aSdan 
305977fc1d5bSdan /*
306077fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
306177fc1d5bSdan ** the database table affected by the change that pIter currently points
306277fc1d5bSdan ** to. This function may only be called after changeset_next() returns
306377fc1d5bSdan ** SQLITE_ROW.
306477fc1d5bSdan */
3065244593c8Sdan int sqlite3changeset_pk(
3066244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
3067244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
3068244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
3069244593c8Sdan ){
3070244593c8Sdan   *pabPK = pIter->abPK;
3071244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
3072244593c8Sdan   return SQLITE_OK;
3073244593c8Sdan }
3074244593c8Sdan 
3075296c7658Sdan /*
3076296c7658Sdan ** This function may only be called while the iterator is pointing to an
3077296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
3078296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3079296c7658Sdan **
3080296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3081296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
3082296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3083296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
3084296c7658Sdan **
3085296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3086296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3087296c7658Sdan */
30884fccf43aSdan int sqlite3changeset_old(
3089296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3090296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
30914fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
30924fccf43aSdan ){
3093d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
3094d5f0767cSdan     return SQLITE_MISUSE;
3095d5f0767cSdan   }
30964fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
30974fccf43aSdan     return SQLITE_RANGE;
30984fccf43aSdan   }
30994fccf43aSdan   *ppValue = pIter->apValue[iVal];
31004fccf43aSdan   return SQLITE_OK;
31014fccf43aSdan }
31024fccf43aSdan 
3103296c7658Sdan /*
3104296c7658Sdan ** This function may only be called while the iterator is pointing to an
3105296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
3106296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
3107296c7658Sdan **
3108296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
3109296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
3110296c7658Sdan ** included in the record (because the change is an UPDATE and the field
3111296c7658Sdan ** was not modified), set *ppValue to NULL.
3112296c7658Sdan **
3113296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
3114296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
3115296c7658Sdan */
31164fccf43aSdan int sqlite3changeset_new(
3117296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3118296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
31194fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
31204fccf43aSdan ){
3121d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
3122d5f0767cSdan     return SQLITE_MISUSE;
3123d5f0767cSdan   }
31244fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
31254fccf43aSdan     return SQLITE_RANGE;
31264fccf43aSdan   }
31274fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
31284fccf43aSdan   return SQLITE_OK;
31294fccf43aSdan }
31304fccf43aSdan 
3131296c7658Sdan /*
31327aa469cdSdan ** The following two macros are used internally. They are similar to the
31337aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
31347aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
31357aa469cdSdan */
31367aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
31377aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
31387aa469cdSdan 
31397aa469cdSdan /*
3140296c7658Sdan ** This function may only be called with a changeset iterator that has been
3141296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
3142296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
3143296c7658Sdan **
3144296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
3145296c7658Sdan ** containing the iVal'th value of the conflicting record.
3146296c7658Sdan **
3147296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
3148296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
3149296c7658Sdan */
3150d5f0767cSdan int sqlite3changeset_conflict(
3151296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3152296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
3153d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
3154d5f0767cSdan ){
3155d5f0767cSdan   if( !pIter->pConflict ){
3156d5f0767cSdan     return SQLITE_MISUSE;
3157d5f0767cSdan   }
3158ff677b20Sdan   if( iVal<0 || iVal>=pIter->nCol ){
3159d5f0767cSdan     return SQLITE_RANGE;
3160d5f0767cSdan   }
3161d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
3162d5f0767cSdan   return SQLITE_OK;
3163d5f0767cSdan }
3164d5f0767cSdan 
31654fccf43aSdan /*
3166cb3e4b79Sdan ** This function may only be called with an iterator passed to an
3167cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
3168cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
3169cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
3170cb3e4b79Sdan **
3171cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
3172cb3e4b79Sdan */
3173cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
3174cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3175cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
3176cb3e4b79Sdan ){
3177cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
3178cb3e4b79Sdan     return SQLITE_MISUSE;
3179cb3e4b79Sdan   }
3180cb3e4b79Sdan   *pnOut = pIter->nCol;
3181cb3e4b79Sdan   return SQLITE_OK;
3182cb3e4b79Sdan }
3183cb3e4b79Sdan 
3184cb3e4b79Sdan 
3185cb3e4b79Sdan /*
31864fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
31874fccf43aSdan **
31884fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
31894fccf43aSdan ** callback by changeset_apply().
31904fccf43aSdan */
31914fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
3192cbf6d2d2Sdan   int rc = SQLITE_OK;
3193cbf6d2d2Sdan   if( p ){
3194296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
3195cbf6d2d2Sdan     rc = p->rc;
319612ca0b56Sdan     if( p->apValue ){
31974fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
319812ca0b56Sdan     }
3199ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
32004757c658Sdan     sqlite3_free(p->in.buf.aBuf);
32014fccf43aSdan     sqlite3_free(p);
3202cbf6d2d2Sdan   }
32034fccf43aSdan   return rc;
32044fccf43aSdan }
32054fccf43aSdan 
3206fa122adaSdan static int sessionChangesetInvert(
3207fa122adaSdan   SessionInput *pInput,           /* Input changeset */
3208fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3209fa122adaSdan   void *pOut,
321091ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
321191ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
321291ddd559Sdan ){
3213cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
3214fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
3215cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
3216cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
3217cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
3218ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
321991ddd559Sdan 
3220fa122adaSdan   /* Initialize the output buffer */
3221fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
3222fa122adaSdan 
322391ddd559Sdan   /* Zero the output variables in case an error occurs. */
3224fa122adaSdan   if( ppInverted ){
322591ddd559Sdan     *ppInverted = 0;
322691ddd559Sdan     *pnInverted = 0;
3227fa122adaSdan   }
322891ddd559Sdan 
3229fa122adaSdan   while( 1 ){
3230ef7a6304Sdan     u8 eType;
3231fa122adaSdan 
3232fa122adaSdan     /* Test for EOF. */
3233fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
3234fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
3235fa122adaSdan     eType = pInput->aData[pInput->iNext];
3236fa122adaSdan 
323791ddd559Sdan     switch( eType ){
323891ddd559Sdan       case 'T': {
3239244593c8Sdan         /* A 'table' record consists of:
3240244593c8Sdan         **
3241244593c8Sdan         **   * A constant 'T' character,
3242244593c8Sdan         **   * Number of columns in said table (a varint),
3243ef7a6304Sdan         **   * An array of nCol bytes (sPK),
3244244593c8Sdan         **   * A nul-terminated table name.
3245244593c8Sdan         */
3246ef7a6304Sdan         int nByte;
3247fa122adaSdan         int nVar;
3248fa122adaSdan         pInput->iNext++;
3249fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
3250ef7a6304Sdan           goto finished_invert;
3251ef7a6304Sdan         }
3252fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
3253ef7a6304Sdan         sPK.nBuf = 0;
3254fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
3255fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3256fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3257ef7a6304Sdan         if( rc ) goto finished_invert;
3258fa122adaSdan 
3259fa122adaSdan         pInput->iNext += nByte;
3260cfec7eeeSdan         sqlite3_free(apVal);
3261cfec7eeeSdan         apVal = 0;
3262ef7a6304Sdan         abPK = sPK.aBuf;
326391ddd559Sdan         break;
326491ddd559Sdan       }
326591ddd559Sdan 
326691ddd559Sdan       case SQLITE_INSERT:
326791ddd559Sdan       case SQLITE_DELETE: {
326891ddd559Sdan         int nByte;
3269fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
3270fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
3271fa122adaSdan         pInput->iNext += 2;
3272fa122adaSdan         assert( rc==SQLITE_OK );
3273fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
3274fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
3275fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
3276fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
3277fa122adaSdan         pInput->iNext += nByte;
3278fa122adaSdan         if( rc ) goto finished_invert;
327991ddd559Sdan         break;
328091ddd559Sdan       }
328191ddd559Sdan 
328291ddd559Sdan       case SQLITE_UPDATE: {
3283cfec7eeeSdan         int iCol;
328491ddd559Sdan 
3285cfec7eeeSdan         if( 0==apVal ){
3286cfec7eeeSdan           apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2);
3287cfec7eeeSdan           if( 0==apVal ){
3288cfec7eeeSdan             rc = SQLITE_NOMEM;
3289cfec7eeeSdan             goto finished_invert;
3290cfec7eeeSdan           }
3291cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3292cfec7eeeSdan         }
329391ddd559Sdan 
3294cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
3295fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
3296fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
329791ddd559Sdan 
3298ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
3299fa122adaSdan         pInput->iNext += 2;
3300fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
3301ef7a6304Sdan         if( rc==SQLITE_OK ){
3302fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
3303ef7a6304Sdan         }
3304ef7a6304Sdan 
3305cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
3306cfec7eeeSdan         ** original old.* record, and the other values from the original
3307cfec7eeeSdan         ** new.* record. */
3308e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3309cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
3310fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3311cfec7eeeSdan         }
3312cfec7eeeSdan 
3313cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
3314cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
3315cfec7eeeSdan         ** are set to "undefined". */
3316e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
3317cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
3318fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
3319cfec7eeeSdan         }
3320cfec7eeeSdan 
3321cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
3322cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
3323cfec7eeeSdan         }
3324cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
3325cfec7eeeSdan         if( rc!=SQLITE_OK ){
3326cfec7eeeSdan           goto finished_invert;
3327cfec7eeeSdan         }
3328cfec7eeeSdan 
332991ddd559Sdan         break;
333091ddd559Sdan       }
333191ddd559Sdan 
333291ddd559Sdan       default:
33334757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
3334cfec7eeeSdan         goto finished_invert;
333591ddd559Sdan     }
3336fa122adaSdan 
3337fa122adaSdan     assert( rc==SQLITE_OK );
3338f1a08ad8Sdrh     if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
3339fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3340fa122adaSdan       sOut.nBuf = 0;
3341fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
3342fa122adaSdan     }
334391ddd559Sdan   }
334491ddd559Sdan 
3345cfec7eeeSdan   assert( rc==SQLITE_OK );
3346fa122adaSdan   if( pnInverted ){
3347fa122adaSdan     *pnInverted = sOut.nBuf;
3348fa122adaSdan     *ppInverted = sOut.aBuf;
3349fa122adaSdan     sOut.aBuf = 0;
3350fa122adaSdan   }else if( sOut.nBuf>0 ){
3351fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
3352fa122adaSdan   }
3353cfec7eeeSdan 
3354cfec7eeeSdan  finished_invert:
3355fa122adaSdan   sqlite3_free(sOut.aBuf);
3356cfec7eeeSdan   sqlite3_free(apVal);
3357ef7a6304Sdan   sqlite3_free(sPK.aBuf);
3358cfec7eeeSdan   return rc;
335991ddd559Sdan }
336091ddd559Sdan 
3361fa122adaSdan 
3362fa122adaSdan /*
3363fa122adaSdan ** Invert a changeset object.
3364fa122adaSdan */
3365fa122adaSdan int sqlite3changeset_invert(
3366fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
3367fa122adaSdan   const void *pChangeset,         /* Input changeset */
3368fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
3369fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
3370fa122adaSdan ){
3371fa122adaSdan   SessionInput sInput;
3372fa122adaSdan 
3373fa122adaSdan   /* Set up the input stream */
3374fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3375fa122adaSdan   sInput.nData = nChangeset;
3376fa122adaSdan   sInput.aData = (u8*)pChangeset;
3377fa122adaSdan 
3378fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
3379fa122adaSdan }
3380fa122adaSdan 
3381fa122adaSdan /*
3382fa122adaSdan ** Streaming version of sqlite3changeset_invert().
3383fa122adaSdan */
3384f1a08ad8Sdrh int sqlite3changeset_invert_strm(
3385fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
3386fa122adaSdan   void *pIn,
3387fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3388fa122adaSdan   void *pOut
3389fa122adaSdan ){
3390fa122adaSdan   SessionInput sInput;
3391fa122adaSdan   int rc;
3392fa122adaSdan 
3393fa122adaSdan   /* Set up the input stream */
3394fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
3395fa122adaSdan   sInput.xInput = xInput;
3396fa122adaSdan   sInput.pIn = pIn;
3397fa122adaSdan 
3398fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
3399fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
3400fa122adaSdan   return rc;
3401fa122adaSdan }
3402fa122adaSdan 
34030c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
34040c698471Sdan struct SessionApplyCtx {
34050c698471Sdan   sqlite3 *db;
34060c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
3407cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
34080c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
34090c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
34100c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
34110c698471Sdan   const char **azCol;             /* Array of column names */
34120c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
3413d1cccf19Sdan   int bStat1;                     /* True if table is sqlite_stat1 */
3414d9151526Sdan   int bDeferConstraints;          /* True to defer constraints */
3415d9151526Sdan   SessionBuffer constraints;      /* Deferred constraints are stored here */
3416a38e6c57Sdan   SessionBuffer rebase;           /* Rebase information (if any) here */
3417a38e6c57Sdan   int bRebaseStarted;             /* If table header is already in rebase */
34180c698471Sdan };
34190c698471Sdan 
3420d5f0767cSdan /*
3421d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
3422d5f0767cSdan ** structure like this:
3423d5f0767cSdan **
3424d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3425d5f0767cSdan **
3426d5f0767cSdan ** The DELETE statement looks like this:
3427d5f0767cSdan **
3428db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
3429d5f0767cSdan **
3430d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
3431d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
3432d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
3433296c7658Sdan **
3434296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
3435296c7658Sdan ** pointing to the prepared version of the SQL statement.
3436d5f0767cSdan */
3437d5f0767cSdan static int sessionDeleteRow(
3438d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3439d5f0767cSdan   const char *zTab,               /* Table name */
34400c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3441d5f0767cSdan ){
3442296c7658Sdan   int i;
3443296c7658Sdan   const char *zSep = "";
3444d5f0767cSdan   int rc = SQLITE_OK;
3445d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
34467cf7df7dSdan   int nPk = 0;
3447d5f0767cSdan 
3448d5f0767cSdan   sessionAppendStr(&buf, "DELETE FROM ", &rc);
3449d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3450296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
3451296c7658Sdan 
3452296c7658Sdan   for(i=0; i<p->nCol; i++){
3453296c7658Sdan     if( p->abPK[i] ){
34547cf7df7dSdan       nPk++;
3455296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
3456296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3457296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
3458296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
3459296c7658Sdan       zSep = " AND ";
3460296c7658Sdan     }
3461296c7658Sdan   }
3462296c7658Sdan 
34637cf7df7dSdan   if( nPk<p->nCol ){
3464296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
3465296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
3466296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
3467296c7658Sdan 
3468296c7658Sdan     zSep = "";
3469296c7658Sdan     for(i=0; i<p->nCol; i++){
3470296c7658Sdan       if( !p->abPK[i] ){
3471296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
3472296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
3473296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
3474296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
3475296c7658Sdan         zSep = "AND ";
3476296c7658Sdan       }
3477296c7658Sdan     }
3478296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
34797cf7df7dSdan   }
3480d5f0767cSdan 
3481d5f0767cSdan   if( rc==SQLITE_OK ){
34820c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
3483d5f0767cSdan   }
3484d5f0767cSdan   sqlite3_free(buf.aBuf);
3485d5f0767cSdan 
3486d5f0767cSdan   return rc;
3487d5f0767cSdan }
3488d5f0767cSdan 
3489d5f0767cSdan /*
3490d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
3491d5f0767cSdan ** Assuming a table structure like this:
3492d5f0767cSdan **
3493d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3494d5f0767cSdan **
3495d5f0767cSdan ** The UPDATE statement looks like this:
3496d5f0767cSdan **
3497d5f0767cSdan **     UPDATE x SET
3498d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3499964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3500964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3501964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3502d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3503964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3504d5f0767cSdan **     )
3505d5f0767cSdan **
3506d5f0767cSdan ** For each column in the table, there are three variables to bind:
3507d5f0767cSdan **
3508d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3509d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3510d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3511d5f0767cSdan **
3512d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3513d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3514d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3515d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3516d5f0767cSdan **
3517296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3518296c7658Sdan ** pointing to the prepared version of the SQL statement.
3519d5f0767cSdan */
3520d5f0767cSdan static int sessionUpdateRow(
3521d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3522d5f0767cSdan   const char *zTab,               /* Table name */
35230c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3524d5f0767cSdan ){
3525d5f0767cSdan   int rc = SQLITE_OK;
3526d5f0767cSdan   int i;
3527d5f0767cSdan   const char *zSep = "";
3528d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3529d5f0767cSdan 
3530d5f0767cSdan   /* Append "UPDATE tbl SET " */
3531d5f0767cSdan   sessionAppendStr(&buf, "UPDATE ", &rc);
3532d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3533d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3534d5f0767cSdan 
3535d5f0767cSdan   /* Append the assignments */
35360c698471Sdan   for(i=0; i<p->nCol; i++){
3537d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
35380c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3539d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3540d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3541d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3542d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3543d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
35440c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3545d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3546d5f0767cSdan     zSep = ", ";
3547d5f0767cSdan   }
3548d5f0767cSdan 
3549d5f0767cSdan   /* Append the PK part of the WHERE clause */
3550d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
35510c698471Sdan   for(i=0; i<p->nCol; i++){
35520c698471Sdan     if( p->abPK[i] ){
35530c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3554d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3555d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3556d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3557d5f0767cSdan     }
3558d5f0767cSdan   }
3559d5f0767cSdan 
3560d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3561d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
35620c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3563d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
35640c698471Sdan   for(i=0; i<p->nCol; i++){
35650c698471Sdan     if( !p->abPK[i] ){
3566d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3567d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3568d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
35690c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3570d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3571d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3572d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3573d5f0767cSdan     }
3574d5f0767cSdan   }
3575d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3576d5f0767cSdan 
3577d5f0767cSdan   if( rc==SQLITE_OK ){
35780c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3579d5f0767cSdan   }
3580d5f0767cSdan   sqlite3_free(buf.aBuf);
3581d5f0767cSdan 
3582d5f0767cSdan   return rc;
3583d5f0767cSdan }
3584d5f0767cSdan 
35853739f298Sdan 
3586296c7658Sdan /*
3587296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3588296c7658Sdan ** key. Assuming the following table structure:
3589296c7658Sdan **
3590296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3591296c7658Sdan **
3592296c7658Sdan ** The SELECT statement looks like this:
3593296c7658Sdan **
3594296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3595296c7658Sdan **
3596296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3597296c7658Sdan ** pointing to the prepared version of the SQL statement.
3598296c7658Sdan */
3599d5f0767cSdan static int sessionSelectRow(
3600d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3601d5f0767cSdan   const char *zTab,               /* Table name */
36020c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3603d5f0767cSdan ){
3604d7fb7d24Sdan   return sessionSelectStmt(
3605d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3606d5f0767cSdan }
3607d5f0767cSdan 
3608296c7658Sdan /*
3609296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3610296c7658Sdan ** For example:
3611296c7658Sdan **
3612296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3613296c7658Sdan **
3614296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3615296c7658Sdan ** pointing to the prepared version of the SQL statement.
3616296c7658Sdan */
36170c698471Sdan static int sessionInsertRow(
36180c698471Sdan   sqlite3 *db,                    /* Database handle */
36190c698471Sdan   const char *zTab,               /* Table name */
36200c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
36210c698471Sdan ){
36220c698471Sdan   int rc = SQLITE_OK;
36230c698471Sdan   int i;
36240c698471Sdan   SessionBuffer buf = {0, 0, 0};
36250c698471Sdan 
36260c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
36270c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
3628ff677b20Sdan   sessionAppendStr(&buf, "(", &rc);
3629ff677b20Sdan   for(i=0; i<p->nCol; i++){
3630ff677b20Sdan     if( i!=0 ) sessionAppendStr(&buf, ", ", &rc);
3631ff677b20Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3632ff677b20Sdan   }
3633ff677b20Sdan 
3634ff677b20Sdan   sessionAppendStr(&buf, ") VALUES(?", &rc);
36350c698471Sdan   for(i=1; i<p->nCol; i++){
36360c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
36370c698471Sdan   }
36380c698471Sdan   sessionAppendStr(&buf, ")", &rc);
36390c698471Sdan 
36400c698471Sdan   if( rc==SQLITE_OK ){
36410c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
36420c698471Sdan   }
36430c698471Sdan   sqlite3_free(buf.aBuf);
36440c698471Sdan   return rc;
36450c698471Sdan }
36460c698471Sdan 
36473739f298Sdan static int sessionPrepare(sqlite3 *db, sqlite3_stmt **pp, const char *zSql){
36483739f298Sdan   return sqlite3_prepare_v2(db, zSql, -1, pp, 0);
36493739f298Sdan }
36503739f298Sdan 
36513739f298Sdan /*
36523739f298Sdan ** Prepare statements for applying changes to the sqlite_stat1 table.
36533739f298Sdan ** These are similar to those created by sessionSelectRow(),
36543739f298Sdan ** sessionInsertRow(), sessionUpdateRow() and sessionDeleteRow() for
36553739f298Sdan ** other tables.
36563739f298Sdan */
36573739f298Sdan static int sessionStat1Sql(sqlite3 *db, SessionApplyCtx *p){
36583739f298Sdan   int rc = sessionSelectRow(db, "sqlite_stat1", p);
36593739f298Sdan   if( rc==SQLITE_OK ){
36603739f298Sdan     rc = sessionPrepare(db, &p->pInsert,
36613739f298Sdan         "INSERT INTO main.sqlite_stat1 VALUES(?1, "
36623739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END, "
36633739f298Sdan         "?3)"
36643739f298Sdan     );
36653739f298Sdan   }
36663739f298Sdan   if( rc==SQLITE_OK ){
36673739f298Sdan     rc = sessionPrepare(db, &p->pUpdate,
36683739f298Sdan         "UPDATE main.sqlite_stat1 SET "
36693739f298Sdan         "tbl = CASE WHEN ?2 THEN ?3 ELSE tbl END, "
36703739f298Sdan         "idx = CASE WHEN ?5 THEN ?6 ELSE idx END, "
36713739f298Sdan         "stat = CASE WHEN ?8 THEN ?9 ELSE stat END  "
36723739f298Sdan         "WHERE tbl=?1 AND idx IS "
36733739f298Sdan         "CASE WHEN length(?4)=0 AND typeof(?4)='blob' THEN NULL ELSE ?4 END "
36743739f298Sdan         "AND (?10 OR ?8=0 OR stat IS ?7)"
36753739f298Sdan     );
36763739f298Sdan   }
36773739f298Sdan   if( rc==SQLITE_OK ){
36783739f298Sdan     rc = sessionPrepare(db, &p->pDelete,
36793739f298Sdan         "DELETE FROM main.sqlite_stat1 WHERE tbl=?1 AND idx IS "
36803739f298Sdan         "CASE WHEN length(?2)=0 AND typeof(?2)='blob' THEN NULL ELSE ?2 END "
36813739f298Sdan         "AND (?4 OR stat IS ?3)"
36823739f298Sdan     );
36833739f298Sdan   }
36843739f298Sdan   return rc;
36853739f298Sdan }
36863739f298Sdan 
3687296c7658Sdan /*
36887aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
36897aa469cdSdan ** See comments in the body of this function for details.
36907aa469cdSdan */
36917aa469cdSdan static int sessionBindValue(
36927aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
36937aa469cdSdan   int i,                          /* Parameter number to bind to */
36947aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
36957aa469cdSdan ){
36965671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3697082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3698082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3699082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3700082c96dfSdan   ** set to SQLITE_NULL.  */
37015671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
37027aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
37037aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3704082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
37057aa469cdSdan     return SQLITE_NOMEM;
37067aa469cdSdan   }
37077aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
37087aa469cdSdan }
37097aa469cdSdan 
37107aa469cdSdan /*
3711db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3712db04571cSdan ** transfers new.* values from the current iterator entry to statement
3713db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3714db04571cSdan **
3715d9151526Sdan ** New.* value $i from the iterator is bound to variable ($i+1) of
3716db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3717db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3718db04571cSdan ** to an array nCol elements in size. In this case only those values for
3719db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3720db04571cSdan ** statement.
3721db04571cSdan **
3722db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3723db04571cSdan */
37247aa469cdSdan static int sessionBindRow(
3725db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
37267aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3727db04571cSdan   int nCol,                       /* Number of columns */
3728db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3729db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3730db04571cSdan ){
3731db04571cSdan   int i;
3732db04571cSdan   int rc = SQLITE_OK;
37337aa469cdSdan 
37347aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
37357aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
37367aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
37377aa469cdSdan   ** in the code below. */
37387aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
37397aa469cdSdan 
3740db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3741db04571cSdan     if( !abPK || abPK[i] ){
3742db04571cSdan       sqlite3_value *pVal;
37437aa469cdSdan       (void)xValue(pIter, i, &pVal);
3744dd8a4af8Sdan       if( pVal==0 ){
3745dd8a4af8Sdan         /* The value in the changeset was "undefined". This indicates a
3746dd8a4af8Sdan         ** corrupt changeset blob.  */
3747e341ec69Sdan         rc = SQLITE_CORRUPT_BKPT;
3748dd8a4af8Sdan       }else{
37497aa469cdSdan         rc = sessionBindValue(pStmt, i+1, pVal);
3750db04571cSdan       }
3751db04571cSdan     }
3752dd8a4af8Sdan   }
3753db04571cSdan   return rc;
3754db04571cSdan }
3755db04571cSdan 
3756db04571cSdan /*
3757296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3758296c7658Sdan ** This function binds the primary key values from the change that changeset
3759296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3760296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3761296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3762296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
37637aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
37647aa469cdSdan **
37657aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
37667aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
37677aa469cdSdan ** not require a reset().
3768296c7658Sdan **
3769296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3770db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3771db04571cSdan ** UPDATE, bind values from the old.* record.
3772296c7658Sdan */
37730c698471Sdan static int sessionSeekToRow(
377437f133ecSdan   sqlite3 *db,                    /* Database handle */
377537f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
377637f133ecSdan   u8 *abPK,                       /* Primary key flags array */
37770c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
377837f133ecSdan ){
37797aa469cdSdan   int rc;                         /* Return code */
3780296c7658Sdan   int nCol;                       /* Number of columns in table */
3781296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3782296c7658Sdan   const char *zDummy;             /* Unused */
378337f133ecSdan 
3784b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
37857aa469cdSdan   rc = sessionBindRow(pIter,
3786db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3787db04571cSdan       nCol, abPK, pSelect
3788db04571cSdan   );
37890c698471Sdan 
37900c698471Sdan   if( rc==SQLITE_OK ){
37910c698471Sdan     rc = sqlite3_step(pSelect);
37920c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
37930c698471Sdan   }
37940c698471Sdan 
37950c698471Sdan   return rc;
37960c698471Sdan }
37970c698471Sdan 
3798b880a7b1Sdan /*
3799b880a7b1Sdan ** This function is called from within sqlite3changset_apply_v2() when
3800b880a7b1Sdan ** a conflict is encountered and resolved using conflict resolution
3801b880a7b1Sdan ** mode eType (either SQLITE_CHANGESET_OMIT or SQLITE_CHANGESET_REPLACE)..
3802b880a7b1Sdan ** It adds a conflict resolution record to the buffer in
3803b880a7b1Sdan ** SessionApplyCtx.rebase, which will eventually be returned to the caller
3804b880a7b1Sdan ** of apply_v2() as the "rebase" buffer.
3805b880a7b1Sdan **
3806b880a7b1Sdan ** Return SQLITE_OK if successful, or an SQLite error code otherwise.
3807b880a7b1Sdan */
3808a38e6c57Sdan static int sessionRebaseAdd(
3809b880a7b1Sdan   SessionApplyCtx *p,             /* Apply context */
3810b880a7b1Sdan   int eType,                      /* Conflict resolution (OMIT or REPLACE) */
3811b880a7b1Sdan   sqlite3_changeset_iter *pIter   /* Iterator pointing at current change */
3812a38e6c57Sdan ){
3813a38e6c57Sdan   int rc = SQLITE_OK;
3814a38e6c57Sdan   int i;
3815a38e6c57Sdan   int eOp = pIter->op;
3816a38e6c57Sdan   if( p->bRebaseStarted==0 ){
3817a38e6c57Sdan     /* Append a table-header to the rebase buffer */
3818a38e6c57Sdan     const char *zTab = pIter->zTab;
3819a38e6c57Sdan     sessionAppendByte(&p->rebase, 'T', &rc);
3820a38e6c57Sdan     sessionAppendVarint(&p->rebase, p->nCol, &rc);
3821a38e6c57Sdan     sessionAppendBlob(&p->rebase, p->abPK, p->nCol, &rc);
3822a38e6c57Sdan     sessionAppendBlob(&p->rebase, (u8*)zTab, (int)strlen(zTab)+1, &rc);
3823a38e6c57Sdan     p->bRebaseStarted = 1;
3824a38e6c57Sdan   }
3825a38e6c57Sdan 
3826a38e6c57Sdan   assert( eType==SQLITE_CHANGESET_REPLACE||eType==SQLITE_CHANGESET_OMIT );
3827a38e6c57Sdan   assert( eOp==SQLITE_DELETE || eOp==SQLITE_INSERT || eOp==SQLITE_UPDATE );
3828a38e6c57Sdan 
3829f01d3a7eSdan   sessionAppendByte(&p->rebase,
3830f01d3a7eSdan       (eOp==SQLITE_DELETE ? SQLITE_DELETE : SQLITE_INSERT), &rc
3831f01d3a7eSdan   );
3832f01d3a7eSdan   sessionAppendByte(&p->rebase, (eType==SQLITE_CHANGESET_REPLACE), &rc);
3833a38e6c57Sdan   for(i=0; i<p->nCol; i++){
3834a38e6c57Sdan     sqlite3_value *pVal = 0;
3835f01d3a7eSdan     if( eOp==SQLITE_DELETE || (eOp==SQLITE_UPDATE && p->abPK[i]) ){
3836a38e6c57Sdan       sqlite3changeset_old(pIter, i, &pVal);
3837a38e6c57Sdan     }else{
3838a38e6c57Sdan       sqlite3changeset_new(pIter, i, &pVal);
3839a38e6c57Sdan     }
3840a38e6c57Sdan     sessionAppendValue(&p->rebase, pVal, &rc);
3841a38e6c57Sdan   }
3842a38e6c57Sdan 
3843a38e6c57Sdan   return rc;
3844a38e6c57Sdan }
3845a38e6c57Sdan 
3846296c7658Sdan /*
3847296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3848296c7658Sdan ** currently points to.
3849296c7658Sdan **
3850296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3851296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3852296c7658Sdan ** depends solely on eType, as follows:
3853296c7658Sdan **
3854296c7658Sdan **    eType value                 Value passed to xConflict
3855296c7658Sdan **    -------------------------------------------------
3856296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3857296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3858296c7658Sdan **
3859296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3860296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3861296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3862296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3863296c7658Sdan ** handler invoked is as follows:
3864296c7658Sdan **
3865296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3866296c7658Sdan **    ----------------------------------------------------------------
3867296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3868296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3869296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3870296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3871296c7658Sdan **
3872296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3873296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3874296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3875296c7658Sdan **
3876296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3877296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3878296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3879296c7658Sdan ** this function returns SQLITE_OK.
3880296c7658Sdan */
38810c698471Sdan static int sessionConflictHandler(
3882296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3883296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
38840c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
38850c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3886296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3887296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
38880c698471Sdan ){
388974f598b6Smistachkin   int res = 0;                    /* Value returned by conflict handler */
38900c698471Sdan   int rc;
38910c698471Sdan   int nCol;
38920c698471Sdan   int op;
38930c698471Sdan   const char *zDummy;
38940c698471Sdan 
3895b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
38960c698471Sdan 
38970c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
38980c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
38990c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
390037f133ecSdan 
390137f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
39020c698471Sdan   if( pbReplace ){
39030c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
39040c698471Sdan   }else{
3905db04571cSdan     rc = SQLITE_OK;
39060c698471Sdan   }
39070c698471Sdan 
39080c698471Sdan   if( rc==SQLITE_ROW ){
39090c698471Sdan     /* There exists another row with the new.* primary key. */
39100c698471Sdan     pIter->pConflict = p->pSelect;
39110c698471Sdan     res = xConflict(pCtx, eType, pIter);
39120c698471Sdan     pIter->pConflict = 0;
39130c698471Sdan     rc = sqlite3_reset(p->pSelect);
3914db04571cSdan   }else if( rc==SQLITE_OK ){
3915d9151526Sdan     if( p->bDeferConstraints && eType==SQLITE_CHANGESET_CONFLICT ){
3916d9151526Sdan       /* Instead of invoking the conflict handler, append the change blob
3917d9151526Sdan       ** to the SessionApplyCtx.constraints buffer. */
3918d9151526Sdan       u8 *aBlob = &pIter->in.aData[pIter->in.iCurrent];
3919d9151526Sdan       int nBlob = pIter->in.iNext - pIter->in.iCurrent;
3920d9151526Sdan       sessionAppendBlob(&p->constraints, aBlob, nBlob, &rc);
3921a38e6c57Sdan       return SQLITE_OK;
3922d9151526Sdan     }else{
39230c698471Sdan       /* No other row with the new.* primary key. */
39240c698471Sdan       res = xConflict(pCtx, eType+1, pIter);
39250c698471Sdan       if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
392637f133ecSdan     }
3927d9151526Sdan   }
392837f133ecSdan 
392937f133ecSdan   if( rc==SQLITE_OK ){
39300c698471Sdan     switch( res ){
39310c698471Sdan       case SQLITE_CHANGESET_REPLACE:
3932f51e5f6cSdan         assert( pbReplace );
3933f51e5f6cSdan         *pbReplace = 1;
39340c698471Sdan         break;
39350c698471Sdan 
39360c698471Sdan       case SQLITE_CHANGESET_OMIT:
39370c698471Sdan         break;
39380c698471Sdan 
39390c698471Sdan       case SQLITE_CHANGESET_ABORT:
39400c698471Sdan         rc = SQLITE_ABORT;
39410c698471Sdan         break;
39420c698471Sdan 
39430c698471Sdan       default:
39440c698471Sdan         rc = SQLITE_MISUSE;
39450c698471Sdan         break;
39460c698471Sdan     }
3947a38e6c57Sdan     if( rc==SQLITE_OK ){
3948a38e6c57Sdan       rc = sessionRebaseAdd(p, res, pIter);
3949a38e6c57Sdan     }
39500c698471Sdan   }
39510c698471Sdan 
39520c698471Sdan   return rc;
39530c698471Sdan }
39540c698471Sdan 
3955296c7658Sdan /*
3956296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
3957296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
3958296c7658Sdan ** the conflict handler callback.
3959296c7658Sdan **
3960296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
3961296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
3962296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
3963296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
3964296c7658Sdan ** to true before returning. In this case the caller will invoke this function
3965296c7658Sdan ** again, this time with pbRetry set to NULL.
3966296c7658Sdan **
3967296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
3968296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
3969296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
3970296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
3971296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
3972296c7658Sdan ** row before invoking this function again, this time with pbReplace set
3973296c7658Sdan ** to NULL.
3974296c7658Sdan **
3975296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
3976296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
3977296c7658Sdan ** returned.
3978296c7658Sdan */
39790c698471Sdan static int sessionApplyOneOp(
3980296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3981296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
39820c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
3983296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
3984296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
3985296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
39860c698471Sdan ){
39870c698471Sdan   const char *zDummy;
39880c698471Sdan   int op;
39890c698471Sdan   int nCol;
39900c698471Sdan   int rc = SQLITE_OK;
39910c698471Sdan 
39920c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
39930c698471Sdan   assert( p->azCol && p->abPK );
39940c698471Sdan   assert( !pbReplace || *pbReplace==0 );
39950c698471Sdan 
3996b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
39970c698471Sdan 
39980c698471Sdan   if( op==SQLITE_DELETE ){
39990c698471Sdan 
400073b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
400173b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
400273b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
400373b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
400473b3c055Sdan     ** is not required if:
400573b3c055Sdan     **
400673b3c055Sdan     **   * this is a patchset, or
400773b3c055Sdan     **   * (pbRetry==0), or
400873b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
400973b3c055Sdan     **     no (nCol+1) variable to bind to).
401073b3c055Sdan     */
401173b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
401273b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
40137cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
401473b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
40157cf7df7dSdan     }
40160c698471Sdan     if( rc!=SQLITE_OK ) return rc;
40170c698471Sdan 
40180c698471Sdan     sqlite3_step(p->pDelete);
40190c698471Sdan     rc = sqlite3_reset(p->pDelete);
40200c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
40210c698471Sdan       rc = sessionConflictHandler(
40220c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
40230c698471Sdan       );
402435e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
40250c698471Sdan       rc = sessionConflictHandler(
40260c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
40270c698471Sdan       );
40280c698471Sdan     }
40290c698471Sdan 
40300c698471Sdan   }else if( op==SQLITE_UPDATE ){
40310c698471Sdan     int i;
40320c698471Sdan 
40330c698471Sdan     /* Bind values to the UPDATE statement. */
40340c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
40357aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
40367aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
40377aa469cdSdan 
40380c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
40397aa469cdSdan       if( pOld ){
40407aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
40417aa469cdSdan       }
40427aa469cdSdan       if( rc==SQLITE_OK && pNew ){
40437aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
40440c698471Sdan       }
40450c698471Sdan     }
404673b3c055Sdan     if( rc==SQLITE_OK ){
404773b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
404873b3c055Sdan     }
40490c698471Sdan     if( rc!=SQLITE_OK ) return rc;
40500c698471Sdan 
40510c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
40520c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
40530c698471Sdan     sqlite3_step(p->pUpdate);
40540c698471Sdan     rc = sqlite3_reset(p->pUpdate);
40550c698471Sdan 
40560c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
40570c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
40580c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
40590c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
40600c698471Sdan 
40610c698471Sdan       rc = sessionConflictHandler(
40620c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
40630c698471Sdan       );
40640c698471Sdan 
406535e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
4066db04571cSdan       /* This is always a CONSTRAINT conflict. */
4067db04571cSdan       rc = sessionConflictHandler(
4068db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
40690c698471Sdan       );
40700c698471Sdan     }
40710c698471Sdan 
40720c698471Sdan   }else{
40730c698471Sdan     assert( op==SQLITE_INSERT );
4074d1cccf19Sdan     if( p->bStat1 ){
4075d1cccf19Sdan       /* Check if there is a conflicting row. For sqlite_stat1, this needs
4076d1cccf19Sdan       ** to be done using a SELECT, as there is no PRIMARY KEY in the
4077d1cccf19Sdan       ** database schema to throw an exception if a duplicate is inserted.  */
4078d1cccf19Sdan       rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
4079d1cccf19Sdan       if( rc==SQLITE_ROW ){
4080d1cccf19Sdan         rc = SQLITE_CONSTRAINT;
4081d1cccf19Sdan         sqlite3_reset(p->pSelect);
4082d1cccf19Sdan       }
4083d1cccf19Sdan     }
4084d1cccf19Sdan 
4085d1cccf19Sdan     if( rc==SQLITE_OK ){
40867aa469cdSdan       rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
40870c698471Sdan       if( rc!=SQLITE_OK ) return rc;
40880c698471Sdan 
40890c698471Sdan       sqlite3_step(p->pInsert);
40900c698471Sdan       rc = sqlite3_reset(p->pInsert);
4091d1cccf19Sdan     }
4092d1cccf19Sdan 
409335e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
40940c698471Sdan       rc = sessionConflictHandler(
40950c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
40960c698471Sdan       );
409737f133ecSdan     }
409837f133ecSdan   }
409937f133ecSdan 
410037f133ecSdan   return rc;
410137f133ecSdan }
410237f133ecSdan 
41035f5663dcSdan /*
41045f5663dcSdan ** Attempt to apply the change that the iterator passed as the first argument
41055f5663dcSdan ** currently points to to the database. If a conflict is encountered, invoke
41065f5663dcSdan ** the conflict handler callback.
41075f5663dcSdan **
41085f5663dcSdan ** The difference between this function and sessionApplyOne() is that this
41095f5663dcSdan ** function handles the case where the conflict-handler is invoked and
41105f5663dcSdan ** returns SQLITE_CHANGESET_REPLACE - indicating that the change should be
41115f5663dcSdan ** retried in some manner.
41125f5663dcSdan */
4113d9151526Sdan static int sessionApplyOneWithRetry(
4114d9151526Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4115d9151526Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator to read change from */
4116d9151526Sdan   SessionApplyCtx *pApply,        /* Apply context */
4117d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4118d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4119d9151526Sdan ){
4120d9151526Sdan   int bReplace = 0;
4121d9151526Sdan   int bRetry = 0;
4122d9151526Sdan   int rc;
4123d9151526Sdan 
4124d9151526Sdan   rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, &bReplace, &bRetry);
4125a38e6c57Sdan   if( rc==SQLITE_OK ){
41265f5663dcSdan     /* If the bRetry flag is set, the change has not been applied due to an
41275f5663dcSdan     ** SQLITE_CHANGESET_DATA problem (i.e. this is an UPDATE or DELETE and
41285f5663dcSdan     ** a row with the correct PK is present in the db, but one or more other
41295f5663dcSdan     ** fields do not contain the expected values) and the conflict handler
41305f5663dcSdan     ** returned SQLITE_CHANGESET_REPLACE. In this case retry the operation,
41315f5663dcSdan     ** but pass NULL as the final argument so that sessionApplyOneOp() ignores
41325f5663dcSdan     ** the SQLITE_CHANGESET_DATA problem.  */
41335f5663dcSdan     if( bRetry ){
41345f5663dcSdan       assert( pIter->op==SQLITE_UPDATE || pIter->op==SQLITE_DELETE );
41355f5663dcSdan       rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4136d9151526Sdan     }
4137d9151526Sdan 
41385f5663dcSdan     /* If the bReplace flag is set, the change is an INSERT that has not
41395f5663dcSdan     ** been performed because the database already contains a row with the
41405f5663dcSdan     ** specified primary key and the conflict handler returned
41415f5663dcSdan     ** SQLITE_CHANGESET_REPLACE. In this case remove the conflicting row
41425f5663dcSdan     ** before reattempting the INSERT.  */
41435f5663dcSdan     else if( bReplace ){
4144d9151526Sdan       assert( pIter->op==SQLITE_INSERT );
4145d9151526Sdan       rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
4146d9151526Sdan       if( rc==SQLITE_OK ){
4147d9151526Sdan         rc = sessionBindRow(pIter,
4148d9151526Sdan             sqlite3changeset_new, pApply->nCol, pApply->abPK, pApply->pDelete);
4149d9151526Sdan         sqlite3_bind_int(pApply->pDelete, pApply->nCol+1, 1);
4150d9151526Sdan       }
4151d9151526Sdan       if( rc==SQLITE_OK ){
4152d9151526Sdan         sqlite3_step(pApply->pDelete);
4153d9151526Sdan         rc = sqlite3_reset(pApply->pDelete);
4154d9151526Sdan       }
4155d9151526Sdan       if( rc==SQLITE_OK ){
4156d9151526Sdan         rc = sessionApplyOneOp(pIter, pApply, xConflict, pCtx, 0, 0);
4157d9151526Sdan       }
4158d9151526Sdan       if( rc==SQLITE_OK ){
4159d9151526Sdan         rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
4160d9151526Sdan       }
4161d9151526Sdan     }
4162a38e6c57Sdan   }
4163d9151526Sdan 
4164d9151526Sdan   return rc;
4165d9151526Sdan }
4166d9151526Sdan 
4167d9151526Sdan /*
4168d9151526Sdan ** Retry the changes accumulated in the pApply->constraints buffer.
4169d9151526Sdan */
4170d9151526Sdan static int sessionRetryConstraints(
4171d9151526Sdan   sqlite3 *db,
4172d9151526Sdan   int bPatchset,
4173d9151526Sdan   const char *zTab,
4174d9151526Sdan   SessionApplyCtx *pApply,
4175d9151526Sdan   int(*xConflict)(void*, int, sqlite3_changeset_iter*),
4176d9151526Sdan   void *pCtx                      /* First argument passed to xConflict */
4177d9151526Sdan ){
4178d9151526Sdan   int rc = SQLITE_OK;
4179d9151526Sdan 
4180d9151526Sdan   while( pApply->constraints.nBuf ){
4181d9151526Sdan     sqlite3_changeset_iter *pIter2 = 0;
4182d9151526Sdan     SessionBuffer cons = pApply->constraints;
4183d9151526Sdan     memset(&pApply->constraints, 0, sizeof(SessionBuffer));
4184d9151526Sdan 
4185d9151526Sdan     rc = sessionChangesetStart(&pIter2, 0, 0, cons.nBuf, cons.aBuf);
4186d9151526Sdan     if( rc==SQLITE_OK ){
4187d9151526Sdan       int nByte = 2*pApply->nCol*sizeof(sqlite3_value*);
4188d9151526Sdan       int rc2;
4189d9151526Sdan       pIter2->bPatchset = bPatchset;
4190d9151526Sdan       pIter2->zTab = (char*)zTab;
4191d9151526Sdan       pIter2->nCol = pApply->nCol;
4192d9151526Sdan       pIter2->abPK = pApply->abPK;
4193d9151526Sdan       sessionBufferGrow(&pIter2->tblhdr, nByte, &rc);
4194d9151526Sdan       pIter2->apValue = (sqlite3_value**)pIter2->tblhdr.aBuf;
4195d9151526Sdan       if( rc==SQLITE_OK ) memset(pIter2->apValue, 0, nByte);
4196d9151526Sdan 
4197d9151526Sdan       while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter2) ){
4198d9151526Sdan         rc = sessionApplyOneWithRetry(db, pIter2, pApply, xConflict, pCtx);
4199d9151526Sdan       }
4200d9151526Sdan 
4201d9151526Sdan       rc2 = sqlite3changeset_finalize(pIter2);
42027e0765a9Sdrh       if( rc==SQLITE_OK ) rc = rc2;
4203d9151526Sdan     }
4204d9151526Sdan     assert( pApply->bDeferConstraints || pApply->constraints.nBuf==0 );
4205d9151526Sdan 
4206d9151526Sdan     sqlite3_free(cons.aBuf);
4207d9151526Sdan     if( rc!=SQLITE_OK ) break;
4208d9151526Sdan     if( pApply->constraints.nBuf>=cons.nBuf ){
4209d9151526Sdan       /* No progress was made on the last round. */
4210d9151526Sdan       pApply->bDeferConstraints = 0;
4211d9151526Sdan     }
4212d9151526Sdan   }
4213d9151526Sdan 
4214d9151526Sdan   return rc;
4215d9151526Sdan }
4216d9151526Sdan 
4217296c7658Sdan /*
42184757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
42194757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
42204757c658Sdan ** changeset to the main database attached to handle "db". The supplied
42214757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
42224757c658Sdan ** while applying the change.
4223296c7658Sdan */
42244757c658Sdan static int sessionChangesetApply(
4225296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
42264757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
422740368988Sdan   int(*xFilter)(
422840368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
422940368988Sdan     const char *zTab              /* Table name */
423040368988Sdan   ),
4231d5f0767cSdan   int(*xConflict)(
4232d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
4233d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4234d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4235d5f0767cSdan   ),
4236a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4237a38e6c57Sdan   void **ppRebase, int *pnRebase  /* OUT: Rebase information */
4238d5f0767cSdan ){
4239ca62ad57Sdan   int schemaMismatch = 0;
4240296c7658Sdan   int rc;                         /* Return code */
4241d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
4242cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
4243296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
42445f5663dcSdan   int bPatchset;
4245d5f0767cSdan 
4246082c96dfSdan   assert( xConflict!=0 );
4247082c96dfSdan 
4248d9151526Sdan   pIter->in.bNoDiscard = 1;
42490c698471Sdan   memset(&sApply, 0, sizeof(sApply));
42504c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
42510c698471Sdan   rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
4252cb3e4b79Sdan   if( rc==SQLITE_OK ){
4253cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
4254cb3e4b79Sdan   }
42550c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
42560c698471Sdan     int nCol;
4257d5f0767cSdan     int op;
42580c698471Sdan     const char *zNew;
4259ca62ad57Sdan 
4260b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
4261d5f0767cSdan 
42620c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
4263ca62ad57Sdan       u8 *abPK;
4264ca62ad57Sdan 
4265d9151526Sdan       rc = sessionRetryConstraints(
4266d9151526Sdan           db, pIter->bPatchset, zTab, &sApply, xConflict, pCtx
4267d9151526Sdan       );
4268d9151526Sdan       if( rc!=SQLITE_OK ) break;
4269d9151526Sdan 
4270cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
42710c698471Sdan       sqlite3_finalize(sApply.pDelete);
42720c698471Sdan       sqlite3_finalize(sApply.pUpdate);
42730c698471Sdan       sqlite3_finalize(sApply.pInsert);
42740c698471Sdan       sqlite3_finalize(sApply.pSelect);
42750c698471Sdan       sApply.db = db;
4276f1b40e83Sdan       sApply.pDelete = 0;
4277f1b40e83Sdan       sApply.pUpdate = 0;
4278f1b40e83Sdan       sApply.pInsert = 0;
4279f1b40e83Sdan       sApply.pSelect = 0;
4280f1b40e83Sdan       sApply.nCol = 0;
4281f1b40e83Sdan       sApply.azCol = 0;
4282f1b40e83Sdan       sApply.abPK = 0;
4283f1b40e83Sdan       sApply.bStat1 = 0;
4284d9151526Sdan       sApply.bDeferConstraints = 1;
4285a38e6c57Sdan       sApply.bRebaseStarted = 0;
4286f1b40e83Sdan       memset(&sApply.constraints, 0, sizeof(SessionBuffer));
428737f133ecSdan 
428840368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
428940368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
429040368988Sdan       ** non-zero, proceed. */
429140368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
429240368988Sdan       if( schemaMismatch ){
429340368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
4294f05ac112Sdan         if( zTab==0 ){
4295f05ac112Sdan           rc = SQLITE_NOMEM;
4296f05ac112Sdan           break;
4297f05ac112Sdan         }
42984f528042Sdan         nTab = (int)strlen(zTab);
429940368988Sdan         sApply.azCol = (const char **)zTab;
430040368988Sdan       }else{
4301ff677b20Sdan         int nMinCol = 0;
4302ff677b20Sdan         int i;
4303ff677b20Sdan 
4304ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
4305296c7658Sdan         rc = sessionTableInfo(
4306ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
4307ca62ad57Sdan         );
4308ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
4309ff677b20Sdan         for(i=0; i<sApply.nCol; i++){
4310ff677b20Sdan           if( sApply.abPK[i] ) nMinCol = i+1;
4311ff677b20Sdan         }
43120c698471Sdan 
4313ca62ad57Sdan         if( sApply.nCol==0 ){
4314ca62ad57Sdan           schemaMismatch = 1;
4315ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4316ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
4317ca62ad57Sdan           );
4318ca62ad57Sdan         }
4319ff677b20Sdan         else if( sApply.nCol<nCol ){
4320ca62ad57Sdan           schemaMismatch = 1;
4321ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
4322ff677b20Sdan               "sqlite3changeset_apply(): table %s has %d columns, "
4323ff677b20Sdan               "expected %d or more",
4324ca62ad57Sdan               zTab, sApply.nCol, nCol
4325ca62ad57Sdan           );
4326ca62ad57Sdan         }
4327ff677b20Sdan         else if( nCol<nMinCol || memcmp(sApply.abPK, abPK, nCol)!=0 ){
4328ca62ad57Sdan           schemaMismatch = 1;
432940368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
433040368988Sdan               "primary key mismatch for table %s", zTab
4331ca62ad57Sdan           );
4332ca62ad57Sdan         }
4333ff677b20Sdan         else{
4334ff677b20Sdan           sApply.nCol = nCol;
43353739f298Sdan           if( 0==sqlite3_stricmp(zTab, "sqlite_stat1") ){
43363739f298Sdan             if( (rc = sessionStat1Sql(db, &sApply) ) ){
43373739f298Sdan               break;
43383739f298Sdan             }
4339d1cccf19Sdan             sApply.bStat1 = 1;
4340d1cccf19Sdan           }else{
4341ff677b20Sdan             if((rc = sessionSelectRow(db, zTab, &sApply))
43420c698471Sdan                 || (rc = sessionUpdateRow(db, zTab, &sApply))
43430c698471Sdan                 || (rc = sessionDeleteRow(db, zTab, &sApply))
43440c698471Sdan                 || (rc = sessionInsertRow(db, zTab, &sApply))
434537f133ecSdan               ){
434637f133ecSdan               break;
434737f133ecSdan             }
4348d1cccf19Sdan             sApply.bStat1 = 0;
4349d1cccf19Sdan           }
4350ff677b20Sdan         }
4351cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
4352d5f0767cSdan       }
435340368988Sdan     }
4354d5f0767cSdan 
4355ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
4356ca62ad57Sdan     ** next change. A log message has already been issued. */
4357ca62ad57Sdan     if( schemaMismatch ) continue;
4358ca62ad57Sdan 
4359d9151526Sdan     rc = sessionApplyOneWithRetry(db, pIter, &sApply, xConflict, pCtx);
43600c698471Sdan   }
43610c698471Sdan 
43625f5663dcSdan   bPatchset = pIter->bPatchset;
4363296c7658Sdan   if( rc==SQLITE_OK ){
4364296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
4365296c7658Sdan   }else{
4366296c7658Sdan     sqlite3changeset_finalize(pIter);
4367296c7658Sdan   }
4368d5f0767cSdan 
4369d5f0767cSdan   if( rc==SQLITE_OK ){
43705f5663dcSdan     rc = sessionRetryConstraints(db, bPatchset, zTab, &sApply, xConflict, pCtx);
43715f5663dcSdan   }
43725f5663dcSdan 
43735f5663dcSdan   if( rc==SQLITE_OK ){
437407001c45Sdrh     int nFk, notUsed;
437507001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
437607001c45Sdrh     if( nFk!=0 ){
4377cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
4378cb3e4b79Sdan       sqlite3_changeset_iter sIter;
4379cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
4380cb3e4b79Sdan       sIter.nCol = nFk;
4381cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
4382cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
4383cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
4384cb3e4b79Sdan       }
4385cb3e4b79Sdan     }
4386cb3e4b79Sdan   }
4387cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
4388cb3e4b79Sdan 
4389cb3e4b79Sdan   if( rc==SQLITE_OK ){
4390d5f0767cSdan     rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4391d5f0767cSdan   }else{
4392d5f0767cSdan     sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
4393d5f0767cSdan     sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
4394d5f0767cSdan   }
4395d5f0767cSdan 
439695ccb6dcSdan   if( rc==SQLITE_OK && bPatchset==0 && ppRebase && pnRebase ){
4397a38e6c57Sdan     *ppRebase = (void*)sApply.rebase.aBuf;
4398a38e6c57Sdan     *pnRebase = sApply.rebase.nBuf;
4399a38e6c57Sdan     sApply.rebase.aBuf = 0;
4400a38e6c57Sdan   }
44010c698471Sdan   sqlite3_finalize(sApply.pInsert);
44020c698471Sdan   sqlite3_finalize(sApply.pDelete);
44030c698471Sdan   sqlite3_finalize(sApply.pUpdate);
44040c698471Sdan   sqlite3_finalize(sApply.pSelect);
4405cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
4406d9151526Sdan   sqlite3_free((char*)sApply.constraints.aBuf);
4407a38e6c57Sdan   sqlite3_free((char*)sApply.rebase.aBuf);
44084c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
4409d5f0767cSdan   return rc;
4410d5f0767cSdan }
441191ddd559Sdan 
4412b880a7b1Sdan /*
4413b880a7b1Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main
4414b880a7b1Sdan ** database attached to handle "db".
4415b880a7b1Sdan */
4416a38e6c57Sdan int sqlite3changeset_apply_v2(
4417a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4418a38e6c57Sdan   int nChangeset,                 /* Size of changeset in bytes */
4419a38e6c57Sdan   void *pChangeset,               /* Changeset blob */
4420a38e6c57Sdan   int(*xFilter)(
4421a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4422a38e6c57Sdan     const char *zTab              /* Table name */
4423a38e6c57Sdan   ),
4424a38e6c57Sdan   int(*xConflict)(
4425a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4426a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4427a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4428a38e6c57Sdan   ),
4429a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4430a38e6c57Sdan   void **ppRebase, int *pnRebase
4431a38e6c57Sdan ){
4432a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
4433a38e6c57Sdan   int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset);
4434a38e6c57Sdan   if( rc==SQLITE_OK ){
4435a38e6c57Sdan     rc = sessionChangesetApply(
4436a38e6c57Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase
4437a38e6c57Sdan     );
4438a38e6c57Sdan   }
4439a38e6c57Sdan   return rc;
4440a38e6c57Sdan }
4441a38e6c57Sdan 
444277fc1d5bSdan /*
44434757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
44444757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
44454757c658Sdan ** to resolve any conflicts encountered while applying the change.
44464757c658Sdan */
44474757c658Sdan int sqlite3changeset_apply(
44484757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
44494757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
44504757c658Sdan   void *pChangeset,               /* Changeset blob */
44514757c658Sdan   int(*xFilter)(
44524757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
44534757c658Sdan     const char *zTab              /* Table name */
44544757c658Sdan   ),
44554757c658Sdan   int(*xConflict)(
44564757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
44574757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
44584757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
44594757c658Sdan   ),
44604757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
44614757c658Sdan ){
4462a38e6c57Sdan   return sqlite3changeset_apply_v2(
4463a38e6c57Sdan       db, nChangeset, pChangeset, xFilter, xConflict, pCtx, 0, 0
4464a38e6c57Sdan   );
44654757c658Sdan }
44664757c658Sdan 
44674757c658Sdan /*
44684757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
44694757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
44704757c658Sdan ** to resolve any conflicts encountered while applying the change.
44714757c658Sdan */
4472a38e6c57Sdan int sqlite3changeset_apply_v2_strm(
4473a38e6c57Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
4474a38e6c57Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
4475a38e6c57Sdan   void *pIn,                                          /* First arg for xInput */
4476a38e6c57Sdan   int(*xFilter)(
4477a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4478a38e6c57Sdan     const char *zTab              /* Table name */
4479a38e6c57Sdan   ),
4480a38e6c57Sdan   int(*xConflict)(
4481a38e6c57Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
4482a38e6c57Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
4483a38e6c57Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
4484a38e6c57Sdan   ),
4485a38e6c57Sdan   void *pCtx,                     /* First argument passed to xConflict */
4486a38e6c57Sdan   void **ppRebase, int *pnRebase
4487a38e6c57Sdan ){
4488a38e6c57Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
4489a38e6c57Sdan   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
4490a38e6c57Sdan   if( rc==SQLITE_OK ){
4491a38e6c57Sdan     rc = sessionChangesetApply(
4492a38e6c57Sdan         db, pIter, xFilter, xConflict, pCtx, ppRebase, pnRebase
4493a38e6c57Sdan     );
4494a38e6c57Sdan   }
4495a38e6c57Sdan   return rc;
4496a38e6c57Sdan }
4497f1a08ad8Sdrh int sqlite3changeset_apply_strm(
44984757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
44994757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
45004757c658Sdan   void *pIn,                                          /* First arg for xInput */
45014757c658Sdan   int(*xFilter)(
45024757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45034757c658Sdan     const char *zTab              /* Table name */
45044757c658Sdan   ),
45054757c658Sdan   int(*xConflict)(
45064757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
45074757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
45084757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
45094757c658Sdan   ),
45104757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
45114757c658Sdan ){
4512a38e6c57Sdan   return sqlite3changeset_apply_v2_strm(
4513a38e6c57Sdan       db, xInput, pIn, xFilter, xConflict, pCtx, 0, 0
4514a38e6c57Sdan   );
45154757c658Sdan }
45164757c658Sdan 
45174757c658Sdan /*
45185898ad69Sdan ** sqlite3_changegroup handle.
45195898ad69Sdan */
45205898ad69Sdan struct sqlite3_changegroup {
45215898ad69Sdan   int rc;                         /* Error code */
45225898ad69Sdan   int bPatch;                     /* True to accumulate patchsets */
45235898ad69Sdan   SessionTable *pList;            /* List of tables in current patch */
45245898ad69Sdan };
45255898ad69Sdan 
45265898ad69Sdan /*
452777fc1d5bSdan ** This function is called to merge two changes to the same row together as
452877fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
452977fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
453077fc1d5bSdan */
45315d607a6eSdan static int sessionChangeMerge(
453277fc1d5bSdan   SessionTable *pTab,             /* Table structure */
4533c0a499eaSdan   int bRebase,                    /* True for a rebase hash-table */
453464277f4aSdan   int bPatchset,                  /* True for patchsets */
453577fc1d5bSdan   SessionChange *pExist,          /* Existing change */
453677fc1d5bSdan   int op2,                        /* Second change operation */
453777fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
453877fc1d5bSdan   u8 *aRec,                       /* Second change record */
453977fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
454077fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
45415d607a6eSdan ){
45425d607a6eSdan   SessionChange *pNew = 0;
454324a0c453Sdan   int rc = SQLITE_OK;
45445d607a6eSdan 
45455d607a6eSdan   if( !pExist ){
4546cbf6d2d2Sdan     pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec);
45475d607a6eSdan     if( !pNew ){
45485d607a6eSdan       return SQLITE_NOMEM;
45495d607a6eSdan     }
45505d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
4551798693b2Sdan     pNew->op = op2;
45525d607a6eSdan     pNew->bIndirect = bIndirect;
4553cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
455424a0c453Sdan     if( bIndirect==0 || bRebase==0 ){
455524a0c453Sdan       pNew->nRecord = nRec;
4556cbf6d2d2Sdan       memcpy(pNew->aRecord, aRec, nRec);
455724a0c453Sdan     }else{
455824a0c453Sdan       int i;
455924a0c453Sdan       u8 *pIn = aRec;
456024a0c453Sdan       u8 *pOut = pNew->aRecord;
456124a0c453Sdan       for(i=0; i<pTab->nCol; i++){
456224a0c453Sdan         int nIn = sessionSerialLen(pIn);
456324a0c453Sdan         if( *pIn==0 ){
456424a0c453Sdan           *pOut++ = 0;
456524a0c453Sdan         }else if( pTab->abPK[i]==0 ){
456624a0c453Sdan           *pOut++ = 0xFF;
456724a0c453Sdan         }else{
456824a0c453Sdan           memcpy(pOut, pIn, nIn);
456924a0c453Sdan           pOut += nIn;
457024a0c453Sdan         }
457124a0c453Sdan         pIn += nIn;
457224a0c453Sdan       }
457324a0c453Sdan       pNew->nRecord = pOut - pNew->aRecord;
457424a0c453Sdan     }
4575c0a499eaSdan   }else if( bRebase ){
457624a0c453Sdan     if( pExist->op==SQLITE_DELETE && pExist->bIndirect ){
457724a0c453Sdan       *ppNew = pExist;
457824a0c453Sdan     }else{
457924a0c453Sdan       int nByte = nRec + pExist->nRecord + sizeof(SessionChange);
458024a0c453Sdan       pNew = (SessionChange*)sqlite3_malloc(nByte);
458124a0c453Sdan       if( pNew==0 ){
458224a0c453Sdan         rc = SQLITE_NOMEM;
458324a0c453Sdan       }else{
458424a0c453Sdan         int i;
458524a0c453Sdan         u8 *a1 = pExist->aRecord;
458624a0c453Sdan         u8 *a2 = aRec;
458724a0c453Sdan         u8 *pOut;
458824a0c453Sdan 
458924a0c453Sdan         memset(pNew, 0, nByte);
459024a0c453Sdan         pNew->bIndirect = bIndirect || pExist->bIndirect;
459124a0c453Sdan         pNew->op = op2;
459224a0c453Sdan         pOut = pNew->aRecord = (u8*)&pNew[1];
459324a0c453Sdan 
459424a0c453Sdan         for(i=0; i<pTab->nCol; i++){
459524a0c453Sdan           int n1 = sessionSerialLen(a1);
459624a0c453Sdan           int n2 = sessionSerialLen(a2);
4597f231e18cSdan           if( *a1==0xFF || (pTab->abPK[i]==0 && bIndirect) ){
459824a0c453Sdan             *pOut++ = 0xFF;
459924a0c453Sdan           }else if( *a2==0 ){
460024a0c453Sdan             memcpy(pOut, a1, n1);
460124a0c453Sdan             pOut += n1;
460224a0c453Sdan           }else{
460324a0c453Sdan             memcpy(pOut, a2, n2);
460424a0c453Sdan             pOut += n2;
460524a0c453Sdan           }
460624a0c453Sdan           a1 += n1;
460724a0c453Sdan           a2 += n2;
460824a0c453Sdan         }
460924a0c453Sdan         pNew->nRecord = pOut - pNew->aRecord;
461024a0c453Sdan       }
461124a0c453Sdan       sqlite3_free(pExist);
461224a0c453Sdan     }
46135d607a6eSdan   }else{
4614798693b2Sdan     int op1 = pExist->op;
46155d607a6eSdan 
46165d607a6eSdan     /*
46175d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
46185d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
46195d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
46205d607a6eSdan     **
46215d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
46225d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
46235d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
46245d607a6eSdan     **
46255d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
46265d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
46275d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
46285d607a6eSdan     */
46295d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
46305d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
46315d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
46325d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
46335d607a6eSdan     ){
46345d607a6eSdan       pNew = pExist;
46355d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
46365d607a6eSdan       sqlite3_free(pExist);
46375d607a6eSdan       assert( pNew==0 );
46385d607a6eSdan     }else{
463964277f4aSdan       u8 *aExist = pExist->aRecord;
46405d607a6eSdan       int nByte;
46415d607a6eSdan       u8 *aCsr;
46425d607a6eSdan 
464364277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
464464277f4aSdan       ** buffer of the new object is large enough to hold any record that
464564277f4aSdan       ** may be generated by combining the input records.  */
46465d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
46475d607a6eSdan       pNew = (SessionChange *)sqlite3_malloc(nByte);
46485d607a6eSdan       if( !pNew ){
46491756ae10Sdan         sqlite3_free(pExist);
46505d607a6eSdan         return SQLITE_NOMEM;
46515d607a6eSdan       }
46525d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
46535d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
46545d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
46555d607a6eSdan 
4656b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
46575d607a6eSdan         u8 *a1 = aRec;
4658b08a1efaSdan         assert( op2==SQLITE_UPDATE );
4659798693b2Sdan         pNew->op = SQLITE_INSERT;
4660ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
466164277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
4662b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
4663b08a1efaSdan         assert( op2==SQLITE_INSERT );
4664798693b2Sdan         pNew->op = SQLITE_UPDATE;
4665fa29ecc4Sdan         if( bPatchset ){
4666fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
4667fa29ecc4Sdan           aCsr += nRec;
4668fa29ecc4Sdan         }else{
466964277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
4670b08a1efaSdan             sqlite3_free(pNew);
4671b08a1efaSdan             pNew = 0;
46725d607a6eSdan           }
4673fa29ecc4Sdan         }
4674b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
467564277f4aSdan         u8 *a1 = aExist;
46765d607a6eSdan         u8 *a2 = aRec;
4677cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
467864277f4aSdan         if( bPatchset==0 ){
4679ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
4680ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
468164277f4aSdan         }
4682798693b2Sdan         pNew->op = SQLITE_UPDATE;
468364277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
46845d607a6eSdan           sqlite3_free(pNew);
46855d607a6eSdan           pNew = 0;
46865d607a6eSdan         }
4687b08a1efaSdan       }else{                                /* UPDATE + DELETE */
4688b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
4689798693b2Sdan         pNew->op = SQLITE_DELETE;
469064277f4aSdan         if( bPatchset ){
469164277f4aSdan           memcpy(aCsr, aRec, nRec);
469264277f4aSdan           aCsr += nRec;
469364277f4aSdan         }else{
469464277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
469564277f4aSdan         }
46965d607a6eSdan       }
46975d607a6eSdan 
46985d607a6eSdan       if( pNew ){
46994f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
47005d607a6eSdan       }
47015d607a6eSdan       sqlite3_free(pExist);
47025d607a6eSdan     }
47035d607a6eSdan   }
47045d607a6eSdan 
47055d607a6eSdan   *ppNew = pNew;
470624a0c453Sdan   return rc;
47075d607a6eSdan }
47085d607a6eSdan 
470977fc1d5bSdan /*
47105898ad69Sdan ** Add all changes in the changeset traversed by the iterator passed as
47115898ad69Sdan ** the first argument to the changegroup hash tables.
471277fc1d5bSdan */
471316228167Sdan static int sessionChangesetToHash(
4714cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
4715c0a499eaSdan   sqlite3_changegroup *pGrp,       /* Changegroup object to add changeset to */
4716c0a499eaSdan   int bRebase                      /* True if hash table is for rebasing */
47175d607a6eSdan ){
47185d607a6eSdan   u8 *aRec;
47195d607a6eSdan   int nRec;
4720cbf6d2d2Sdan   int rc = SQLITE_OK;
47215d607a6eSdan   SessionTable *pTab = 0;
47225d607a6eSdan 
4723c0a499eaSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, 0) ){
47245d607a6eSdan     const char *zNew;
47255d607a6eSdan     int nCol;
47265d607a6eSdan     int op;
47275d607a6eSdan     int iHash;
47285d607a6eSdan     int bIndirect;
47295d607a6eSdan     SessionChange *pChange;
47305d607a6eSdan     SessionChange *pExist = 0;
47315d607a6eSdan     SessionChange **pp;
47325d607a6eSdan 
47335898ad69Sdan     if( pGrp->pList==0 ){
47345898ad69Sdan       pGrp->bPatch = pIter->bPatchset;
47355898ad69Sdan     }else if( pIter->bPatchset!=pGrp->bPatch ){
47365898ad69Sdan       rc = SQLITE_ERROR;
47375898ad69Sdan       break;
47385898ad69Sdan     }
47395898ad69Sdan 
47405d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
4741ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
47425d607a6eSdan       /* Search the list for a matching table */
47434f528042Sdan       int nNew = (int)strlen(zNew);
4744f29123b5Sdan       u8 *abPK;
4745f29123b5Sdan 
4746f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
47475898ad69Sdan       for(pTab = pGrp->pList; pTab; pTab=pTab->pNext){
47485d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
47495d607a6eSdan       }
47505d607a6eSdan       if( !pTab ){
47516c39e6a8Sdan         SessionTable **ppTab;
47526c39e6a8Sdan 
4753ef7a6304Sdan         pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1);
4754f29123b5Sdan         if( !pTab ){
4755f29123b5Sdan           rc = SQLITE_NOMEM;
4756f29123b5Sdan           break;
4757f29123b5Sdan         }
47585d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
4759f29123b5Sdan         pTab->nCol = nCol;
4760ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
4761ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
4762ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
4763ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
47646c39e6a8Sdan 
47656c39e6a8Sdan         /* The new object must be linked on to the end of the list, not
47666c39e6a8Sdan         ** simply added to the start of it. This is to ensure that the
47676c39e6a8Sdan         ** tables within the output of sqlite3changegroup_output() are in
47686c39e6a8Sdan         ** the right order.  */
47696c39e6a8Sdan         for(ppTab=&pGrp->pList; *ppTab; ppTab=&(*ppTab)->pNext);
47706c39e6a8Sdan         *ppTab = pTab;
4771f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
4772f29123b5Sdan         rc = SQLITE_SCHEMA;
4773f29123b5Sdan         break;
47745d607a6eSdan       }
47755d607a6eSdan     }
47765d607a6eSdan 
4777cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
47781756ae10Sdan       rc = SQLITE_NOMEM;
47791756ae10Sdan       break;
47801756ae10Sdan     }
478164277f4aSdan     iHash = sessionChangeHash(
4782cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
478364277f4aSdan     );
47845d607a6eSdan 
47855d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
47865d607a6eSdan     ** Code below may link it back in.
47875d607a6eSdan     */
47885d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
478964277f4aSdan       int bPkOnly1 = 0;
479064277f4aSdan       int bPkOnly2 = 0;
4791cbf6d2d2Sdan       if( pIter->bPatchset ){
479264277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
479364277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
479464277f4aSdan       }
479564277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
47965d607a6eSdan         pExist = *pp;
47975d607a6eSdan         *pp = (*pp)->pNext;
47985d607a6eSdan         pTab->nEntry--;
47995d607a6eSdan         break;
48005d607a6eSdan       }
48015d607a6eSdan     }
48025d607a6eSdan 
4803c0a499eaSdan     rc = sessionChangeMerge(pTab, bRebase,
4804cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
480564277f4aSdan     );
48065d607a6eSdan     if( rc ) break;
48075d607a6eSdan     if( pChange ){
48085d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
48095d607a6eSdan       pTab->apChange[iHash] = pChange;
48105d607a6eSdan       pTab->nEntry++;
48115d607a6eSdan     }
48125d607a6eSdan   }
48135d607a6eSdan 
4814cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
48155d607a6eSdan   return rc;
48165d607a6eSdan }
48175d607a6eSdan 
48185d607a6eSdan /*
48195898ad69Sdan ** Serialize a changeset (or patchset) based on all changesets (or patchsets)
48205898ad69Sdan ** added to the changegroup object passed as the first argument.
48215d607a6eSdan **
48225898ad69Sdan ** If xOutput is not NULL, then the changeset/patchset is returned to the
48235898ad69Sdan ** user via one or more calls to xOutput, as with the other streaming
48245898ad69Sdan ** interfaces.
48255d607a6eSdan **
48265898ad69Sdan ** Or, if xOutput is NULL, then (*ppOut) is populated with a pointer to a
48275898ad69Sdan ** buffer containing the output changeset before this function returns. In
48285898ad69Sdan ** this case (*pnOut) is set to the size of the output buffer in bytes. It
48295898ad69Sdan ** is the responsibility of the caller to free the output buffer using
48305898ad69Sdan ** sqlite3_free() when it is no longer required.
48315898ad69Sdan **
48325898ad69Sdan ** If successful, SQLITE_OK is returned. Or, if an error occurs, an SQLite
48335898ad69Sdan ** error code. If an error occurs and xOutput is NULL, (*ppOut) and (*pnOut)
48345898ad69Sdan ** are both set to 0 before returning.
48355d607a6eSdan */
48365898ad69Sdan static int sessionChangegroupOutput(
48375898ad69Sdan   sqlite3_changegroup *pGrp,
4838cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4839cbf6d2d2Sdan   void *pOut,
4840cbf6d2d2Sdan   int *pnOut,
4841cbf6d2d2Sdan   void **ppOut
48425d607a6eSdan ){
48435898ad69Sdan   int rc = SQLITE_OK;
4844e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
48455898ad69Sdan   SessionTable *pTab;
4846cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
48475d607a6eSdan 
48485d607a6eSdan   /* Create the serialized output changeset based on the contents of the
48495898ad69Sdan   ** hash tables attached to the SessionTable objects in list p->pList.
48505d607a6eSdan   */
48515898ad69Sdan   for(pTab=pGrp->pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
48525d607a6eSdan     int i;
48535d607a6eSdan     if( pTab->nEntry==0 ) continue;
48545d607a6eSdan 
48555898ad69Sdan     sessionAppendTableHdr(&buf, pGrp->bPatch, pTab, &rc);
48565d607a6eSdan     for(i=0; i<pTab->nChange; i++){
48575d607a6eSdan       SessionChange *p;
48585d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
4859798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
48605d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
48615d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
48625d607a6eSdan       }
48635d607a6eSdan     }
4864cbf6d2d2Sdan 
4865f1a08ad8Sdrh     if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
4866cbf6d2d2Sdan       rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4867cbf6d2d2Sdan       buf.nBuf = 0;
4868cbf6d2d2Sdan     }
48695d607a6eSdan   }
48705d607a6eSdan 
48715d607a6eSdan   if( rc==SQLITE_OK ){
4872cbf6d2d2Sdan     if( xOutput ){
4873cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
4874cbf6d2d2Sdan     }else{
48755d607a6eSdan       *ppOut = buf.aBuf;
48765d607a6eSdan       *pnOut = buf.nBuf;
4877cbf6d2d2Sdan       buf.aBuf = 0;
48785d607a6eSdan     }
48795d607a6eSdan   }
4880cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
48815d607a6eSdan 
48825d607a6eSdan   return rc;
48835d607a6eSdan }
48845d607a6eSdan 
4885cbf6d2d2Sdan /*
48865898ad69Sdan ** Allocate a new, empty, sqlite3_changegroup.
48875898ad69Sdan */
48885898ad69Sdan int sqlite3changegroup_new(sqlite3_changegroup **pp){
48895898ad69Sdan   int rc = SQLITE_OK;             /* Return code */
48905898ad69Sdan   sqlite3_changegroup *p;         /* New object */
48915898ad69Sdan   p = (sqlite3_changegroup*)sqlite3_malloc(sizeof(sqlite3_changegroup));
48925898ad69Sdan   if( p==0 ){
48935898ad69Sdan     rc = SQLITE_NOMEM;
48945898ad69Sdan   }else{
48955898ad69Sdan     memset(p, 0, sizeof(sqlite3_changegroup));
48965898ad69Sdan   }
48975898ad69Sdan   *pp = p;
48985898ad69Sdan   return rc;
48995898ad69Sdan }
49005898ad69Sdan 
49015898ad69Sdan /*
49025898ad69Sdan ** Add the changeset currently stored in buffer pData, size nData bytes,
49035898ad69Sdan ** to changeset-group p.
49045898ad69Sdan */
49055898ad69Sdan int sqlite3changegroup_add(sqlite3_changegroup *pGrp, int nData, void *pData){
49065898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
49075898ad69Sdan   int rc;                         /* Return code */
49085898ad69Sdan 
49095898ad69Sdan   rc = sqlite3changeset_start(&pIter, nData, pData);
49105898ad69Sdan   if( rc==SQLITE_OK ){
4911c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
49125898ad69Sdan   }
49135898ad69Sdan   sqlite3changeset_finalize(pIter);
49145898ad69Sdan   return rc;
49155898ad69Sdan }
49165898ad69Sdan 
49175898ad69Sdan /*
49185898ad69Sdan ** Obtain a buffer containing a changeset representing the concatenation
49195898ad69Sdan ** of all changesets added to the group so far.
49205898ad69Sdan */
49215898ad69Sdan int sqlite3changegroup_output(
49225898ad69Sdan     sqlite3_changegroup *pGrp,
49235898ad69Sdan     int *pnData,
49245898ad69Sdan     void **ppData
49255898ad69Sdan ){
49265898ad69Sdan   return sessionChangegroupOutput(pGrp, 0, 0, pnData, ppData);
49275898ad69Sdan }
49285898ad69Sdan 
49295898ad69Sdan /*
49305898ad69Sdan ** Streaming versions of changegroup_add().
49315898ad69Sdan */
49325898ad69Sdan int sqlite3changegroup_add_strm(
49335898ad69Sdan   sqlite3_changegroup *pGrp,
49345898ad69Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
49355898ad69Sdan   void *pIn
49365898ad69Sdan ){
49375898ad69Sdan   sqlite3_changeset_iter *pIter;  /* Iterator opened on pData/nData */
49385898ad69Sdan   int rc;                         /* Return code */
49395898ad69Sdan 
49405898ad69Sdan   rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
49415898ad69Sdan   if( rc==SQLITE_OK ){
4942c0a499eaSdan     rc = sessionChangesetToHash(pIter, pGrp, 0);
49435898ad69Sdan   }
49445898ad69Sdan   sqlite3changeset_finalize(pIter);
49455898ad69Sdan   return rc;
49465898ad69Sdan }
49475898ad69Sdan 
49485898ad69Sdan /*
49495898ad69Sdan ** Streaming versions of changegroup_output().
49505898ad69Sdan */
49515898ad69Sdan int sqlite3changegroup_output_strm(
49525898ad69Sdan   sqlite3_changegroup *pGrp,
49535898ad69Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
49545898ad69Sdan   void *pOut
49555898ad69Sdan ){
49565898ad69Sdan   return sessionChangegroupOutput(pGrp, xOutput, pOut, 0, 0);
49575898ad69Sdan }
49585898ad69Sdan 
49595898ad69Sdan /*
49605898ad69Sdan ** Delete a changegroup object.
49615898ad69Sdan */
49625898ad69Sdan void sqlite3changegroup_delete(sqlite3_changegroup *pGrp){
49635898ad69Sdan   if( pGrp ){
49645898ad69Sdan     sessionDeleteTable(pGrp->pList);
49655898ad69Sdan     sqlite3_free(pGrp);
49665898ad69Sdan   }
49675898ad69Sdan }
49685898ad69Sdan 
49695898ad69Sdan /*
4970cbf6d2d2Sdan ** Combine two changesets together.
4971cbf6d2d2Sdan */
4972cbf6d2d2Sdan int sqlite3changeset_concat(
4973cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
4974cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
4975cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
4976cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
4977cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
4978cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
4979cbf6d2d2Sdan ){
49805898ad69Sdan   sqlite3_changegroup *pGrp;
4981cbf6d2d2Sdan   int rc;
4982cbf6d2d2Sdan 
49835898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
4984cbf6d2d2Sdan   if( rc==SQLITE_OK ){
49855898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nLeft, pLeft);
4986cbf6d2d2Sdan   }
4987cbf6d2d2Sdan   if( rc==SQLITE_OK ){
49885898ad69Sdan     rc = sqlite3changegroup_add(pGrp, nRight, pRight);
4989cbf6d2d2Sdan   }
49905898ad69Sdan   if( rc==SQLITE_OK ){
49915898ad69Sdan     rc = sqlite3changegroup_output(pGrp, pnOut, ppOut);
49925898ad69Sdan   }
49935898ad69Sdan   sqlite3changegroup_delete(pGrp);
4994cbf6d2d2Sdan 
4995cbf6d2d2Sdan   return rc;
4996cbf6d2d2Sdan }
4997cbf6d2d2Sdan 
4998cbf6d2d2Sdan /*
4999cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
5000cbf6d2d2Sdan */
5001f1a08ad8Sdrh int sqlite3changeset_concat_strm(
5002cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
5003cbf6d2d2Sdan   void *pInA,
5004cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
5005cbf6d2d2Sdan   void *pInB,
5006cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5007cbf6d2d2Sdan   void *pOut
5008cbf6d2d2Sdan ){
50095898ad69Sdan   sqlite3_changegroup *pGrp;
5010cbf6d2d2Sdan   int rc;
5011cbf6d2d2Sdan 
50125898ad69Sdan   rc = sqlite3changegroup_new(&pGrp);
5013cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50145898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputA, pInA);
5015cbf6d2d2Sdan   }
5016cbf6d2d2Sdan   if( rc==SQLITE_OK ){
50175898ad69Sdan     rc = sqlite3changegroup_add_strm(pGrp, xInputB, pInB);
5018cbf6d2d2Sdan   }
50195898ad69Sdan   if( rc==SQLITE_OK ){
50205898ad69Sdan     rc = sqlite3changegroup_output_strm(pGrp, xOutput, pOut);
50215898ad69Sdan   }
50225898ad69Sdan   sqlite3changegroup_delete(pGrp);
5023cbf6d2d2Sdan 
5024cbf6d2d2Sdan   return rc;
5025cbf6d2d2Sdan }
5026cbf6d2d2Sdan 
50273fa5463cSdan /*
50283fa5463cSdan ** Changeset rebaser handle.
50293fa5463cSdan */
5030c0a499eaSdan struct sqlite3_rebaser {
5031c0a499eaSdan   sqlite3_changegroup grp;        /* Hash table */
5032c0a499eaSdan };
5033c0a499eaSdan 
5034c0a499eaSdan /*
5035c0a499eaSdan ** Buffers a1 and a2 must both contain a sessions module record nCol
5036c0a499eaSdan ** fields in size. This function appends an nCol sessions module
5037b880a7b1Sdan ** record to buffer pBuf that is a copy of a1, except that for
5038b880a7b1Sdan ** each field that is undefined in a1[], swap in the field from a2[].
5039c0a499eaSdan */
5040c0a499eaSdan static void sessionAppendRecordMerge(
5041b880a7b1Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
5042b880a7b1Sdan   int nCol,                       /* Number of columns in each record */
5043b880a7b1Sdan   u8 *a1, int n1,                 /* Record 1 */
5044b880a7b1Sdan   u8 *a2, int n2,                 /* Record 2 */
5045b880a7b1Sdan   int *pRc                        /* IN/OUT: error code */
5046c0a499eaSdan ){
5047c0a499eaSdan   sessionBufferGrow(pBuf, n1+n2, pRc);
5048c0a499eaSdan   if( *pRc==SQLITE_OK ){
5049c0a499eaSdan     int i;
5050c0a499eaSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5051c0a499eaSdan     for(i=0; i<nCol; i++){
5052c0a499eaSdan       int nn1 = sessionSerialLen(a1);
5053c0a499eaSdan       int nn2 = sessionSerialLen(a2);
505424a0c453Sdan       if( *a1==0 || *a1==0xFF ){
5055c0a499eaSdan         memcpy(pOut, a2, nn2);
5056c0a499eaSdan         pOut += nn2;
5057c0a499eaSdan       }else{
5058c0a499eaSdan         memcpy(pOut, a1, nn1);
5059c0a499eaSdan         pOut += nn1;
5060c0a499eaSdan       }
5061f1b40e83Sdan       a1 += nn1;
5062f1b40e83Sdan       a2 += nn2;
5063c0a499eaSdan     }
5064f1b40e83Sdan 
5065f1b40e83Sdan     pBuf->nBuf = pOut-pBuf->aBuf;
5066f1b40e83Sdan     assert( pBuf->nBuf<=pBuf->nAlloc );
5067c0a499eaSdan   }
5068c0a499eaSdan }
5069c0a499eaSdan 
50703fa5463cSdan /*
50713fa5463cSdan ** This function is called when rebasing a local UPDATE change against one
50723fa5463cSdan ** or more remote UPDATE changes. The aRec/nRec buffer contains the current
50733fa5463cSdan ** old.* and new.* records for the change. The rebase buffer (a single
50743fa5463cSdan ** record) is in aChange/nChange. The rebased change is appended to buffer
50753fa5463cSdan ** pBuf.
50763fa5463cSdan **
50773fa5463cSdan ** Rebasing the UPDATE involves:
50783fa5463cSdan **
50793fa5463cSdan **   * Removing any changes to fields for which the corresponding field
50803fa5463cSdan **     in the rebase buffer is set to "replaced" (type 0xFF). If this
50813fa5463cSdan **     means the UPDATE change updates no fields, nothing is appended
50823fa5463cSdan **     to the output buffer.
50833fa5463cSdan **
50843fa5463cSdan **   * For each field modified by the local change for which the
50853fa5463cSdan **     corresponding field in the rebase buffer is not "undefined" (0x00)
50863fa5463cSdan **     or "replaced" (0xFF), the old.* value is replaced by the value
50873fa5463cSdan **     in the rebase buffer.
50883fa5463cSdan */
5089f01d3a7eSdan static void sessionAppendPartialUpdate(
509024a0c453Sdan   SessionBuffer *pBuf,            /* Append record here */
509124a0c453Sdan   sqlite3_changeset_iter *pIter,  /* Iterator pointed at local change */
509224a0c453Sdan   u8 *aRec, int nRec,             /* Local change */
509324a0c453Sdan   u8 *aChange, int nChange,       /* Record to rebase against */
509424a0c453Sdan   int *pRc                        /* IN/OUT: Return Code */
5095f01d3a7eSdan ){
5096f01d3a7eSdan   sessionBufferGrow(pBuf, 2+nRec+nChange, pRc);
5097f01d3a7eSdan   if( *pRc==SQLITE_OK ){
5098f01d3a7eSdan     int bData = 0;
5099f01d3a7eSdan     u8 *pOut = &pBuf->aBuf[pBuf->nBuf];
5100f01d3a7eSdan     int i;
5101f01d3a7eSdan     u8 *a1 = aRec;
5102f01d3a7eSdan     u8 *a2 = aChange;
5103f01d3a7eSdan 
5104f01d3a7eSdan     *pOut++ = SQLITE_UPDATE;
5105f01d3a7eSdan     *pOut++ = pIter->bIndirect;
5106f01d3a7eSdan     for(i=0; i<pIter->nCol; i++){
5107f01d3a7eSdan       int n1 = sessionSerialLen(a1);
5108f01d3a7eSdan       int n2 = sessionSerialLen(a2);
5109f01d3a7eSdan       if( pIter->abPK[i] || a2[0]==0 ){
5110f01d3a7eSdan         if( !pIter->abPK[i] ) bData = 1;
5111f01d3a7eSdan         memcpy(pOut, a1, n1);
5112f01d3a7eSdan         pOut += n1;
511324a0c453Sdan       }else if( a2[0]!=0xFF ){
511424a0c453Sdan         bData = 1;
511524a0c453Sdan         memcpy(pOut, a2, n2);
511624a0c453Sdan         pOut += n2;
5117f01d3a7eSdan       }else{
5118f01d3a7eSdan         *pOut++ = '\0';
5119f01d3a7eSdan       }
5120f01d3a7eSdan       a1 += n1;
5121f01d3a7eSdan       a2 += n2;
5122f01d3a7eSdan     }
5123f01d3a7eSdan     if( bData ){
5124f01d3a7eSdan       a2 = aChange;
5125f01d3a7eSdan       for(i=0; i<pIter->nCol; i++){
5126f01d3a7eSdan         int n1 = sessionSerialLen(a1);
5127f01d3a7eSdan         int n2 = sessionSerialLen(a2);
512824a0c453Sdan         if( pIter->abPK[i] || a2[0]!=0xFF ){
5129f01d3a7eSdan           memcpy(pOut, a1, n1);
5130f01d3a7eSdan           pOut += n1;
5131f01d3a7eSdan         }else{
5132f01d3a7eSdan           *pOut++ = '\0';
5133f01d3a7eSdan         }
5134f01d3a7eSdan         a1 += n1;
5135f01d3a7eSdan         a2 += n2;
5136f01d3a7eSdan       }
5137f01d3a7eSdan       pBuf->nBuf = (pOut - pBuf->aBuf);
5138f01d3a7eSdan     }
5139f01d3a7eSdan   }
5140f01d3a7eSdan }
5141f01d3a7eSdan 
51423fa5463cSdan /*
51433fa5463cSdan ** pIter is configured to iterate through a changeset. This function rebases
51443fa5463cSdan ** that changeset according to the current configuration of the rebaser
51453fa5463cSdan ** object passed as the first argument. If no error occurs and argument xOutput
51463fa5463cSdan ** is not NULL, then the changeset is returned to the caller by invoking
51473fa5463cSdan ** xOutput zero or more times and SQLITE_OK returned. Or, if xOutput is NULL,
51483fa5463cSdan ** then (*ppOut) is set to point to a buffer containing the rebased changeset
51493fa5463cSdan ** before this function returns. In this case (*pnOut) is set to the size of
51503fa5463cSdan ** the buffer in bytes.  It is the responsibility of the caller to eventually
51513fa5463cSdan ** free the (*ppOut) buffer using sqlite3_free().
51523fa5463cSdan **
51533fa5463cSdan ** If an error occurs, an SQLite error code is returned. If ppOut and
51543fa5463cSdan ** pnOut are not NULL, then the two output parameters are set to 0 before
51553fa5463cSdan ** returning.
51563fa5463cSdan */
5157c0a499eaSdan static int sessionRebase(
5158c0a499eaSdan   sqlite3_rebaser *p,             /* Rebaser hash table */
5159c0a499eaSdan   sqlite3_changeset_iter *pIter,  /* Input data */
5160c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5161c0a499eaSdan   void *pOut,                     /* Context for xOutput callback */
5162c0a499eaSdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
5163c0a499eaSdan   void **ppOut                    /* OUT: Inverse of pChangeset */
5164c0a499eaSdan ){
5165c0a499eaSdan   int rc = SQLITE_OK;
5166c0a499eaSdan   u8 *aRec = 0;
5167c0a499eaSdan   int nRec = 0;
5168c0a499eaSdan   int bNew = 0;
5169c0a499eaSdan   SessionTable *pTab = 0;
5170c0a499eaSdan   SessionBuffer sOut = {0,0,0};
5171c0a499eaSdan 
5172f1b40e83Sdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec, &bNew) ){
5173c0a499eaSdan     SessionChange *pChange = 0;
5174f01d3a7eSdan     int bDone = 0;
5175c0a499eaSdan 
5176c0a499eaSdan     if( bNew ){
5177c0a499eaSdan       const char *zTab = pIter->zTab;
5178c0a499eaSdan       for(pTab=p->grp.pList; pTab; pTab=pTab->pNext){
5179c0a499eaSdan         if( 0==sqlite3_stricmp(pTab->zName, zTab) ) break;
5180c0a499eaSdan       }
5181c0a499eaSdan       bNew = 0;
5182c0a499eaSdan 
518395ccb6dcSdan       /* A patchset may not be rebased */
518495ccb6dcSdan       if( pIter->bPatchset ){
518595ccb6dcSdan         rc = SQLITE_ERROR;
518695ccb6dcSdan       }
518795ccb6dcSdan 
5188c0a499eaSdan       /* Append a table header to the output for this new table */
5189c0a499eaSdan       sessionAppendByte(&sOut, pIter->bPatchset ? 'P' : 'T', &rc);
5190c0a499eaSdan       sessionAppendVarint(&sOut, pIter->nCol, &rc);
5191c0a499eaSdan       sessionAppendBlob(&sOut, pIter->abPK, pIter->nCol, &rc);
5192c0a499eaSdan       sessionAppendBlob(&sOut, (u8*)pIter->zTab, strlen(pIter->zTab)+1, &rc);
5193c0a499eaSdan     }
5194c0a499eaSdan 
51958cb83beeSdan     if( pTab && rc==SQLITE_OK ){
51968cb83beeSdan       int iHash = sessionChangeHash(pTab, 0, aRec, pTab->nChange);
5197c0a499eaSdan 
5198c0a499eaSdan       for(pChange=pTab->apChange[iHash]; pChange; pChange=pChange->pNext){
51998cb83beeSdan         if( sessionChangeEqual(pTab, 0, aRec, 0, pChange->aRecord) ){
5200c0a499eaSdan           break;
5201c0a499eaSdan         }
5202c0a499eaSdan       }
5203c0a499eaSdan     }
5204c0a499eaSdan 
5205c0a499eaSdan     if( pChange ){
5206c0a499eaSdan       assert( pChange->op==SQLITE_DELETE || pChange->op==SQLITE_INSERT );
5207f01d3a7eSdan       switch( pIter->op ){
5208f01d3a7eSdan         case SQLITE_INSERT:
5209f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5210f01d3a7eSdan             bDone = 1;
5211f01d3a7eSdan             if( pChange->bIndirect==0 ){
5212f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_UPDATE, &rc);
5213f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5214f01d3a7eSdan               sessionAppendBlob(&sOut, pChange->aRecord, pChange->nRecord, &rc);
5215f01d3a7eSdan               sessionAppendBlob(&sOut, aRec, nRec, &rc);
5216f01d3a7eSdan             }
5217f01d3a7eSdan           }
5218f01d3a7eSdan           break;
5219f01d3a7eSdan 
5220f01d3a7eSdan         case SQLITE_UPDATE:
5221f01d3a7eSdan           bDone = 1;
5222f01d3a7eSdan           if( pChange->op==SQLITE_DELETE ){
5223f01d3a7eSdan             if( pChange->bIndirect==0 ){
5224f01d3a7eSdan               u8 *pCsr = aRec;
5225f01d3a7eSdan               sessionSkipRecord(&pCsr, pIter->nCol);
5226f01d3a7eSdan               sessionAppendByte(&sOut, SQLITE_INSERT, &rc);
5227f01d3a7eSdan               sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5228b880a7b1Sdan               sessionAppendRecordMerge(&sOut, pIter->nCol,
5229f01d3a7eSdan                   pCsr, nRec-(pCsr-aRec),
5230f01d3a7eSdan                   pChange->aRecord, pChange->nRecord, &rc
5231f01d3a7eSdan               );
5232f01d3a7eSdan             }
5233f01d3a7eSdan           }else{
5234f01d3a7eSdan             sessionAppendPartialUpdate(&sOut, pIter,
5235f01d3a7eSdan                 aRec, nRec, pChange->aRecord, pChange->nRecord, &rc
5236f01d3a7eSdan             );
5237f01d3a7eSdan           }
5238f01d3a7eSdan           break;
5239f01d3a7eSdan 
5240f01d3a7eSdan         default:
5241f01d3a7eSdan           assert( pIter->op==SQLITE_DELETE );
5242f01d3a7eSdan           bDone = 1;
5243f01d3a7eSdan           if( pChange->op==SQLITE_INSERT ){
5244f01d3a7eSdan             sessionAppendByte(&sOut, SQLITE_DELETE, &rc);
5245f01d3a7eSdan             sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5246b880a7b1Sdan             sessionAppendRecordMerge(&sOut, pIter->nCol,
5247f01d3a7eSdan                 pChange->aRecord, pChange->nRecord, aRec, nRec, &rc
5248f01d3a7eSdan             );
5249f01d3a7eSdan           }
5250f01d3a7eSdan           break;
5251f01d3a7eSdan       }
5252f01d3a7eSdan     }
5253f01d3a7eSdan 
5254f01d3a7eSdan     if( bDone==0 ){
5255f01d3a7eSdan       sessionAppendByte(&sOut, pIter->op, &rc);
5256f01d3a7eSdan       sessionAppendByte(&sOut, pIter->bIndirect, &rc);
5257f01d3a7eSdan       sessionAppendBlob(&sOut, aRec, nRec, &rc);
5258f01d3a7eSdan     }
5259c0a499eaSdan     if( rc==SQLITE_OK && xOutput && sOut.nBuf>SESSIONS_STRM_CHUNK_SIZE ){
5260c0a499eaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5261c0a499eaSdan       sOut.nBuf = 0;
5262c0a499eaSdan     }
5263c0a499eaSdan     if( rc ) break;
5264c0a499eaSdan   }
5265c0a499eaSdan 
5266c0a499eaSdan   if( rc!=SQLITE_OK ){
5267c0a499eaSdan     sqlite3_free(sOut.aBuf);
5268c0a499eaSdan     memset(&sOut, 0, sizeof(sOut));
5269c0a499eaSdan   }
5270c0a499eaSdan 
5271c0a499eaSdan   if( rc==SQLITE_OK ){
5272c0a499eaSdan     if( xOutput ){
5273c0a499eaSdan       if( sOut.nBuf>0 ){
5274c0a499eaSdan         rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
5275c0a499eaSdan       }
5276c0a499eaSdan     }else{
5277c0a499eaSdan       *ppOut = (void*)sOut.aBuf;
5278c0a499eaSdan       *pnOut = sOut.nBuf;
5279c0a499eaSdan       sOut.aBuf = 0;
5280c0a499eaSdan     }
5281c0a499eaSdan   }
5282c0a499eaSdan   sqlite3_free(sOut.aBuf);
5283c0a499eaSdan   return rc;
5284c0a499eaSdan }
5285c0a499eaSdan 
5286c0a499eaSdan /*
5287c0a499eaSdan ** Create a new rebaser object.
5288c0a499eaSdan */
5289c0a499eaSdan int sqlite3rebaser_create(sqlite3_rebaser **ppNew){
5290c0a499eaSdan   int rc = SQLITE_OK;
5291c0a499eaSdan   sqlite3_rebaser *pNew;
5292c0a499eaSdan 
5293c0a499eaSdan   pNew = sqlite3_malloc(sizeof(sqlite3_rebaser));
5294c0a499eaSdan   if( pNew==0 ){
5295c0a499eaSdan     rc = SQLITE_NOMEM;
5296f1b40e83Sdan   }else{
5297f1b40e83Sdan     memset(pNew, 0, sizeof(sqlite3_rebaser));
5298c0a499eaSdan   }
5299c0a499eaSdan   *ppNew = pNew;
5300c0a499eaSdan   return rc;
5301c0a499eaSdan }
5302c0a499eaSdan 
5303c0a499eaSdan /*
5304c0a499eaSdan ** Call this one or more times to configure a rebaser.
5305c0a499eaSdan */
5306c0a499eaSdan int sqlite3rebaser_configure(
5307c0a499eaSdan   sqlite3_rebaser *p,
5308c0a499eaSdan   int nRebase, const void *pRebase
5309c0a499eaSdan ){
5310c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator opened on pData/nData */
5311c0a499eaSdan   int rc;                              /* Return code */
5312c0a499eaSdan   rc = sqlite3changeset_start(&pIter, nRebase, (void*)pRebase);
5313c0a499eaSdan   if( rc==SQLITE_OK ){
5314c0a499eaSdan     rc = sessionChangesetToHash(pIter, &p->grp, 1);
5315c0a499eaSdan   }
5316c0a499eaSdan   sqlite3changeset_finalize(pIter);
5317c0a499eaSdan   return rc;
5318c0a499eaSdan }
5319c0a499eaSdan 
5320c0a499eaSdan /*
5321c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5322c0a499eaSdan */
5323c0a499eaSdan int sqlite3rebaser_rebase(
5324c0a499eaSdan   sqlite3_rebaser *p,
5325c0a499eaSdan   int nIn, const void *pIn,
5326c0a499eaSdan   int *pnOut, void **ppOut
5327c0a499eaSdan ){
5328c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5329c0a499eaSdan   int rc = sqlite3changeset_start(&pIter, nIn, (void*)pIn);
5330c0a499eaSdan 
5331c0a499eaSdan   if( rc==SQLITE_OK ){
5332c0a499eaSdan     rc = sessionRebase(p, pIter, 0, 0, pnOut, ppOut);
5333c0a499eaSdan     sqlite3changeset_finalize(pIter);
5334c0a499eaSdan   }
5335c0a499eaSdan 
5336c0a499eaSdan   return rc;
5337c0a499eaSdan }
5338c0a499eaSdan 
5339c0a499eaSdan /*
5340c0a499eaSdan ** Rebase a changeset according to current rebaser configuration
5341c0a499eaSdan */
5342c0a499eaSdan int sqlite3rebaser_rebase_strm(
5343c0a499eaSdan   sqlite3_rebaser *p,
5344c0a499eaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
5345c0a499eaSdan   void *pIn,
5346c0a499eaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
5347c0a499eaSdan   void *pOut
5348c0a499eaSdan ){
5349c0a499eaSdan   sqlite3_changeset_iter *pIter = 0;   /* Iterator to skip through input */
5350c0a499eaSdan   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
5351c0a499eaSdan 
5352c0a499eaSdan   if( rc==SQLITE_OK ){
5353c0a499eaSdan     rc = sessionRebase(p, pIter, xOutput, pOut, 0, 0);
5354c0a499eaSdan     sqlite3changeset_finalize(pIter);
5355c0a499eaSdan   }
5356c0a499eaSdan 
5357c0a499eaSdan   return rc;
5358c0a499eaSdan }
5359c0a499eaSdan 
5360c0a499eaSdan /*
5361c0a499eaSdan ** Destroy a rebaser object
5362c0a499eaSdan */
5363f1b40e83Sdan void sqlite3rebaser_delete(sqlite3_rebaser *p){
5364c0a499eaSdan   if( p ){
5365c0a499eaSdan     sessionDeleteTable(p->grp.pList);
5366c0a499eaSdan     sqlite3_free(p);
5367c0a499eaSdan   }
5368c0a499eaSdan }
5369c0a499eaSdan 
53709b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
5371