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 
28296c7658Sdan /*
29296c7658Sdan ** Session handle structure.
30296c7658Sdan */
314fccf43aSdan struct sqlite3_session {
324fccf43aSdan   sqlite3 *db;                    /* Database handle session is attached to */
334fccf43aSdan   char *zDb;                      /* Name of database session is attached to */
34296c7658Sdan   int bEnable;                    /* True if currently recording */
35b4480e94Sdan   int bIndirect;                  /* True if all changes are indirect */
36ff4d0f41Sdan   int bAutoAttach;                /* True to auto-attach tables */
374fccf43aSdan   int rc;                         /* Non-zero if an error has occurred */
387531a5a3Sdan   void *pFilterCtx;               /* First argument to pass to xTableFilter */
397531a5a3Sdan   int (*xTableFilter)(void *pCtx, const char *zTab);
404fccf43aSdan   sqlite3_session *pNext;         /* Next session object on same db. */
414fccf43aSdan   SessionTable *pTable;           /* List of attached tables */
424fccf43aSdan };
434fccf43aSdan 
444fccf43aSdan /*
45ef7a6304Sdan ** Instances of this structure are used to build strings or binary records.
46ef7a6304Sdan */
47ef7a6304Sdan struct SessionBuffer {
48ef7a6304Sdan   u8 *aBuf;                       /* Pointer to changeset buffer */
49ef7a6304Sdan   int nBuf;                       /* Size of buffer aBuf */
50ef7a6304Sdan   int nAlloc;                     /* Size of allocation containing aBuf */
51ef7a6304Sdan };
52ef7a6304Sdan 
53ef7a6304Sdan /*
5416228167Sdan ** An object of this type is used internally as an abstraction for
5516228167Sdan ** input data. Input data may be supplied either as a single large buffer
5616228167Sdan ** (e.g. sqlite3changeset_start()) or using a stream function (e.g.
57f1a08ad8Sdrh **  sqlite3changeset_start_strm()).
58ef7a6304Sdan */
59ef7a6304Sdan struct SessionInput {
604757c658Sdan   int iNext;                      /* Offset in aData[] of next change */
614757c658Sdan   u8 *aData;                      /* Pointer to buffer containing changeset */
624757c658Sdan   int nData;                      /* Number of bytes in aData */
634757c658Sdan 
64ef7a6304Sdan   SessionBuffer buf;              /* Current read buffer */
65ef7a6304Sdan   int (*xInput)(void*, void*, int*);        /* Input stream call (or NULL) */
66ef7a6304Sdan   void *pIn;                                /* First argument to xInput */
67ef7a6304Sdan   int bEof;                       /* Set to true after xInput finished */
68ef7a6304Sdan };
69ef7a6304Sdan 
70ef7a6304Sdan /*
71296c7658Sdan ** Structure for changeset iterators.
72296c7658Sdan */
73296c7658Sdan struct sqlite3_changeset_iter {
74ef7a6304Sdan   SessionInput in;                /* Input buffer or stream */
75ef7a6304Sdan   SessionBuffer tblhdr;           /* Buffer to hold apValue/zTab/abPK/ */
7673b3c055Sdan   int bPatchset;                  /* True if this is a patchset */
77296c7658Sdan   int rc;                         /* Iterator error code */
78296c7658Sdan   sqlite3_stmt *pConflict;        /* Points to conflicting row, if any */
79296c7658Sdan   char *zTab;                     /* Current table */
80296c7658Sdan   int nCol;                       /* Number of columns in zTab */
81296c7658Sdan   int op;                         /* Current operation */
82b4480e94Sdan   int bIndirect;                  /* True if current change was indirect */
83244593c8Sdan   u8 *abPK;                       /* Primary key array */
84296c7658Sdan   sqlite3_value **apValue;        /* old.* and new.* values */
85296c7658Sdan };
86296c7658Sdan 
87296c7658Sdan /*
884fccf43aSdan ** Each session object maintains a set of the following structures, one
894fccf43aSdan ** for each table the session object is monitoring. The structures are
904fccf43aSdan ** stored in a linked list starting at sqlite3_session.pTable.
914fccf43aSdan **
924fccf43aSdan ** The keys of the SessionTable.aChange[] hash table are all rows that have
934fccf43aSdan ** been modified in any way since the session object was attached to the
944fccf43aSdan ** table.
954fccf43aSdan **
964fccf43aSdan ** The data associated with each hash-table entry is a structure containing
974fccf43aSdan ** a subset of the initial values that the modified row contained at the
984fccf43aSdan ** start of the session. Or no initial values if the row was inserted.
994fccf43aSdan */
1004fccf43aSdan struct SessionTable {
1014fccf43aSdan   SessionTable *pNext;
1024fccf43aSdan   char *zName;                    /* Local name of table */
1034fccf43aSdan   int nCol;                       /* Number of columns in table zName */
104e8d5648eSdan   const char **azCol;             /* Column names */
105e8d5648eSdan   u8 *abPK;                       /* Array of primary key flags */
106296c7658Sdan   int nEntry;                     /* Total number of entries in hash table */
1074fccf43aSdan   int nChange;                    /* Size of apChange[] array */
1084fccf43aSdan   SessionChange **apChange;       /* Hash table buckets */
1094fccf43aSdan };
1104fccf43aSdan 
1114fccf43aSdan /*
1124fccf43aSdan ** RECORD FORMAT:
1134fccf43aSdan **
1144fccf43aSdan ** The following record format is similar to (but not compatible with) that
1154fccf43aSdan ** used in SQLite database files. This format is used as part of the
1164fccf43aSdan ** change-set binary format, and so must be architecture independent.
1174fccf43aSdan **
1184fccf43aSdan ** Unlike the SQLite database record format, each field is self-contained -
1194fccf43aSdan ** there is no separation of header and data. Each field begins with a
1204fccf43aSdan ** single byte describing its type, as follows:
1214fccf43aSdan **
1224fccf43aSdan **       0x00: Undefined value.
1234fccf43aSdan **       0x01: Integer value.
1244fccf43aSdan **       0x02: Real value.
1254fccf43aSdan **       0x03: Text value.
1264fccf43aSdan **       0x04: Blob value.
1274fccf43aSdan **       0x05: SQL NULL value.
1284fccf43aSdan **
1294fccf43aSdan ** Note that the above match the definitions of SQLITE_INTEGER, SQLITE_TEXT
1304fccf43aSdan ** and so on in sqlite3.h. For undefined and NULL values, the field consists
1314fccf43aSdan ** only of the single type byte. For other types of values, the type byte
1324fccf43aSdan ** is followed by:
1334fccf43aSdan **
1344fccf43aSdan **   Text values:
1354fccf43aSdan **     A varint containing the number of bytes in the value (encoded using
1364fccf43aSdan **     UTF-8). Followed by a buffer containing the UTF-8 representation
1374fccf43aSdan **     of the text value. There is no nul terminator.
1384fccf43aSdan **
1394fccf43aSdan **   Blob values:
1404fccf43aSdan **     A varint containing the number of bytes in the value, followed by
1414fccf43aSdan **     a buffer containing the value itself.
1424fccf43aSdan **
1434fccf43aSdan **   Integer values:
1444fccf43aSdan **     An 8-byte big-endian integer value.
1454fccf43aSdan **
1464fccf43aSdan **   Real values:
1474fccf43aSdan **     An 8-byte big-endian IEEE 754-2008 real value.
1484fccf43aSdan **
1494fccf43aSdan ** Varint values are encoded in the same way as varints in the SQLite
1504fccf43aSdan ** record format.
1514fccf43aSdan **
1524fccf43aSdan ** CHANGESET FORMAT:
1534fccf43aSdan **
1544fccf43aSdan ** A changeset is a collection of DELETE, UPDATE and INSERT operations on
1554fccf43aSdan ** one or more tables. Operations on a single table are grouped together,
1564fccf43aSdan ** but may occur in any order (i.e. deletes, updates and inserts are all
1574fccf43aSdan ** mixed together).
1584fccf43aSdan **
1594fccf43aSdan ** Each group of changes begins with a table header:
1604fccf43aSdan **
1614fccf43aSdan **   1 byte: Constant 0x54 (capital 'T')
162*730bb805Sdan **   Varint: Number of columns in the table.
16373b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
1644fccf43aSdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
1654fccf43aSdan **
1664fccf43aSdan ** Followed by one or more changes to the table.
1674fccf43aSdan **
1684fccf43aSdan **   1 byte: Either SQLITE_INSERT, UPDATE or DELETE.
1695d607a6eSdan **   1 byte: The "indirect-change" flag.
1704fccf43aSdan **   old.* record: (delete and update only)
1714fccf43aSdan **   new.* record: (insert and update only)
17273b3c055Sdan **
173*730bb805Sdan ** The "old.*" and "new.*" records, if present, are N field records in the
174*730bb805Sdan ** format described above under "RECORD FORMAT", where N is the number of
175*730bb805Sdan ** columns in the table. The i'th field of each record is associated with
176*730bb805Sdan ** the i'th column of the table, counting from left to right in the order
177*730bb805Sdan ** in which columns were declared in the CREATE TABLE statement.
178*730bb805Sdan **
179*730bb805Sdan ** The new.* record that is part of each INSERT change contains the values
180*730bb805Sdan ** that make up the new row. Similarly, the old.* record that is part of each
181*730bb805Sdan ** DELETE change contains the values that made up the row that was deleted
182*730bb805Sdan ** from the database. In the changeset format, the records that are part
183*730bb805Sdan ** of INSERT or DELETE changes never contain any undefined (type byte 0x00)
184*730bb805Sdan ** fields.
185*730bb805Sdan **
186*730bb805Sdan ** Within the old.* record associated with an UPDATE change, all fields
187*730bb805Sdan ** associated with table columns that are not PRIMARY KEY columns and are
188*730bb805Sdan ** not modified by the UPDATE change are set to "undefined". Other fields
189*730bb805Sdan ** are set to the values that made up the row before the UPDATE that the
190*730bb805Sdan ** change records took place. Within the new.* record, fields associated
191*730bb805Sdan ** with table columns modified by the UPDATE change contain the new
192*730bb805Sdan ** values. Fields associated with table columns that are not modified
193*730bb805Sdan ** are set to "undefined".
194*730bb805Sdan **
19573b3c055Sdan ** PATCHSET FORMAT:
19673b3c055Sdan **
19773b3c055Sdan ** A patchset is also a collection of changes. It is similar to a changeset,
198*730bb805Sdan ** but leaves undefined those fields that are not useful if no conflict
199*730bb805Sdan ** resolution is required when applying the changeset.
20073b3c055Sdan **
20173b3c055Sdan ** Each group of changes begins with a table header:
20273b3c055Sdan **
20373b3c055Sdan **   1 byte: Constant 0x50 (capital 'P')
204*730bb805Sdan **   Varint: Number of columns in the table.
20573b3c055Sdan **   nCol bytes: 0x01 for PK columns, 0x00 otherwise.
20673b3c055Sdan **   N bytes: Unqualified table name (encoded using UTF-8). Nul-terminated.
20773b3c055Sdan **
20873b3c055Sdan ** Followed by one or more changes to the table.
20973b3c055Sdan **
21073b3c055Sdan **   1 byte: Either SQLITE_INSERT, UPDATE or DELETE.
21173b3c055Sdan **   1 byte: The "indirect-change" flag.
212*730bb805Sdan **   single record: (PK fields for DELETE, PK and modified fields for UPDATE,
213*730bb805Sdan **                   full record for INSERT).
214*730bb805Sdan **
215*730bb805Sdan ** As in the changeset format, each field of the single record that is part
216*730bb805Sdan ** of a patchset change is associated with the correspondingly positioned
217*730bb805Sdan ** table column, counting from left to right within the CREATE TABLE
218*730bb805Sdan ** statement.
219*730bb805Sdan **
220*730bb805Sdan ** For a DELETE change, all fields within the record except those associated
221*730bb805Sdan ** with PRIMARY KEY columns are set to "undefined". The PRIMARY KEY fields
222*730bb805Sdan ** contain the values identifying the row to delete.
223*730bb805Sdan **
224*730bb805Sdan ** For an UPDATE change, all fields except those associated with PRIMARY KEY
225*730bb805Sdan ** columns and columns that are modified by the UPDATE are set to "undefined".
226*730bb805Sdan ** PRIMARY KEY fields contain the values identifying the table row to update,
227*730bb805Sdan ** and fields associated with modified columns contain the new column values.
228*730bb805Sdan **
229*730bb805Sdan ** The records associated with INSERT changes are in the same format as for
230*730bb805Sdan ** changesets. It is not possible for a record associated with an INSERT
231*730bb805Sdan ** change to contain a field set to "undefined".
2324fccf43aSdan */
2334fccf43aSdan 
2344fccf43aSdan /*
2354fccf43aSdan ** For each row modified during a session, there exists a single instance of
2364fccf43aSdan ** this structure stored in a SessionTable.aChange[] hash table.
2374fccf43aSdan */
2384fccf43aSdan struct SessionChange {
239798693b2Sdan   int op;                         /* One of UPDATE, DELETE, INSERT */
240b4480e94Sdan   int bIndirect;                  /* True if this change is "indirect" */
2414fccf43aSdan   int nRecord;                    /* Number of bytes in buffer aRecord[] */
2424fccf43aSdan   u8 *aRecord;                    /* Buffer containing old.* record */
2434fccf43aSdan   SessionChange *pNext;           /* For hash-table collisions */
2444fccf43aSdan };
2454fccf43aSdan 
246296c7658Sdan /*
247296c7658Sdan ** Write a varint with value iVal into the buffer at aBuf. Return the
248296c7658Sdan ** number of bytes written.
249296c7658Sdan */
250296c7658Sdan static int sessionVarintPut(u8 *aBuf, int iVal){
251296c7658Sdan   return putVarint32(aBuf, iVal);
2524fccf43aSdan }
2534fccf43aSdan 
254296c7658Sdan /*
255296c7658Sdan ** Return the number of bytes required to store value iVal as a varint.
256296c7658Sdan */
257296c7658Sdan static int sessionVarintLen(int iVal){
258296c7658Sdan   return sqlite3VarintLen(iVal);
259296c7658Sdan }
260296c7658Sdan 
261296c7658Sdan /*
262296c7658Sdan ** Read a varint value from aBuf[] into *piVal. Return the number of
263296c7658Sdan ** bytes read.
264296c7658Sdan */
2654fccf43aSdan static int sessionVarintGet(u8 *aBuf, int *piVal){
266296c7658Sdan   return getVarint32(aBuf, *piVal);
2674fccf43aSdan }
2684fccf43aSdan 
269296c7658Sdan /*
270296c7658Sdan ** Read a 64-bit big-endian integer value from buffer aRec[]. Return
271296c7658Sdan ** the value read.
272296c7658Sdan */
2734fccf43aSdan static sqlite3_int64 sessionGetI64(u8 *aRec){
2744fccf43aSdan   return (((sqlite3_int64)aRec[0]) << 56)
2754fccf43aSdan        + (((sqlite3_int64)aRec[1]) << 48)
2764fccf43aSdan        + (((sqlite3_int64)aRec[2]) << 40)
2774fccf43aSdan        + (((sqlite3_int64)aRec[3]) << 32)
2784fccf43aSdan        + (((sqlite3_int64)aRec[4]) << 24)
2794fccf43aSdan        + (((sqlite3_int64)aRec[5]) << 16)
2804fccf43aSdan        + (((sqlite3_int64)aRec[6]) <<  8)
2814fccf43aSdan        + (((sqlite3_int64)aRec[7]) <<  0);
2824fccf43aSdan }
2834fccf43aSdan 
2844fccf43aSdan /*
285296c7658Sdan ** Write a 64-bit big-endian integer value to the buffer aBuf[].
286296c7658Sdan */
287296c7658Sdan static void sessionPutI64(u8 *aBuf, sqlite3_int64 i){
288296c7658Sdan   aBuf[0] = (i>>56) & 0xFF;
289296c7658Sdan   aBuf[1] = (i>>48) & 0xFF;
290296c7658Sdan   aBuf[2] = (i>>40) & 0xFF;
291296c7658Sdan   aBuf[3] = (i>>32) & 0xFF;
292296c7658Sdan   aBuf[4] = (i>>24) & 0xFF;
293296c7658Sdan   aBuf[5] = (i>>16) & 0xFF;
294296c7658Sdan   aBuf[6] = (i>> 8) & 0xFF;
295296c7658Sdan   aBuf[7] = (i>> 0) & 0xFF;
296296c7658Sdan }
297296c7658Sdan 
298296c7658Sdan /*
2994fccf43aSdan ** This function is used to serialize the contents of value pValue (see
3004fccf43aSdan ** comment titled "RECORD FORMAT" above).
3014fccf43aSdan **
3024fccf43aSdan ** If it is non-NULL, the serialized form of the value is written to
3034fccf43aSdan ** buffer aBuf. *pnWrite is set to the number of bytes written before
3044fccf43aSdan ** returning. Or, if aBuf is NULL, the only thing this function does is
3054fccf43aSdan ** set *pnWrite.
3064fccf43aSdan **
3074fccf43aSdan ** If no error occurs, SQLITE_OK is returned. Or, if an OOM error occurs
3084fccf43aSdan ** within a call to sqlite3_value_text() (may fail if the db is utf-16))
3094fccf43aSdan ** SQLITE_NOMEM is returned.
3104fccf43aSdan */
3114fccf43aSdan static int sessionSerializeValue(
3124fccf43aSdan   u8 *aBuf,                       /* If non-NULL, write serialized value here */
3134fccf43aSdan   sqlite3_value *pValue,          /* Value to serialize */
3144fccf43aSdan   int *pnWrite                    /* IN/OUT: Increment by bytes written */
3154fccf43aSdan ){
316296c7658Sdan   int nByte;                      /* Size of serialized value in bytes */
3174fccf43aSdan 
31880fe2d93Sdan   if( pValue ){
31980fe2d93Sdan     int eType;                    /* Value type (SQLITE_NULL, TEXT etc.) */
32080fe2d93Sdan 
3214fccf43aSdan     eType = sqlite3_value_type(pValue);
3224fccf43aSdan     if( aBuf ) aBuf[0] = eType;
3234fccf43aSdan 
3244fccf43aSdan     switch( eType ){
3254fccf43aSdan       case SQLITE_NULL:
3264fccf43aSdan         nByte = 1;
3274fccf43aSdan         break;
3284fccf43aSdan 
3294fccf43aSdan       case SQLITE_INTEGER:
3304fccf43aSdan       case SQLITE_FLOAT:
3314fccf43aSdan         if( aBuf ){
3324fccf43aSdan           /* TODO: SQLite does something special to deal with mixed-endian
3334fccf43aSdan           ** floating point values (e.g. ARM7). This code probably should
3344fccf43aSdan           ** too.  */
3354fccf43aSdan           u64 i;
3364fccf43aSdan           if( eType==SQLITE_INTEGER ){
3374fccf43aSdan             i = (u64)sqlite3_value_int64(pValue);
3384fccf43aSdan           }else{
3394fccf43aSdan             double r;
3404fccf43aSdan             assert( sizeof(double)==8 && sizeof(u64)==8 );
3414fccf43aSdan             r = sqlite3_value_double(pValue);
3424fccf43aSdan             memcpy(&i, &r, 8);
3434fccf43aSdan           }
344296c7658Sdan           sessionPutI64(&aBuf[1], i);
3454fccf43aSdan         }
3464fccf43aSdan         nByte = 9;
3474fccf43aSdan         break;
3484fccf43aSdan 
3494e895da1Sdan       default: {
35080fe2d93Sdan         u8 *z;
35180fe2d93Sdan         int n;
35280fe2d93Sdan         int nVarint;
35380fe2d93Sdan 
3544e895da1Sdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
35580fe2d93Sdan         if( eType==SQLITE_TEXT ){
35680fe2d93Sdan           z = (u8 *)sqlite3_value_text(pValue);
35780fe2d93Sdan         }else{
35880fe2d93Sdan           z = (u8 *)sqlite3_value_blob(pValue);
35980fe2d93Sdan         }
36080fe2d93Sdan         n = sqlite3_value_bytes(pValue);
3613cc89d95Sdan         if( z==0 && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
36280fe2d93Sdan         nVarint = sessionVarintLen(n);
36380fe2d93Sdan 
3644fccf43aSdan         if( aBuf ){
3654fccf43aSdan           sessionVarintPut(&aBuf[1], n);
3664fccf43aSdan           memcpy(&aBuf[nVarint + 1], eType==SQLITE_TEXT ?
3674fccf43aSdan               sqlite3_value_text(pValue) : sqlite3_value_blob(pValue), n
3684fccf43aSdan           );
3694fccf43aSdan         }
3704fccf43aSdan 
3714fccf43aSdan         nByte = 1 + nVarint + n;
3724fccf43aSdan         break;
3734fccf43aSdan       }
3744fccf43aSdan     }
37580fe2d93Sdan   }else{
37680fe2d93Sdan     nByte = 1;
37780fe2d93Sdan     if( aBuf ) aBuf[0] = '\0';
37880fe2d93Sdan   }
3794fccf43aSdan 
380fa122adaSdan   if( pnWrite ) *pnWrite += nByte;
3814fccf43aSdan   return SQLITE_OK;
3824fccf43aSdan }
3834fccf43aSdan 
384fa122adaSdan 
385798693b2Sdan /*
386798693b2Sdan ** This macro is used to calculate hash key values for data structures. In
387798693b2Sdan ** order to use this macro, the entire data structure must be represented
388798693b2Sdan ** as a series of unsigned integers. In order to calculate a hash-key value
389798693b2Sdan ** for a data structure represented as three such integers, the macro may
390798693b2Sdan ** then be used as follows:
391798693b2Sdan **
392798693b2Sdan **    int hash_key_value;
393798693b2Sdan **    hash_key_value = HASH_APPEND(0, <value 1>);
394798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 2>);
395798693b2Sdan **    hash_key_value = HASH_APPEND(hash_key_value, <value 3>);
396798693b2Sdan **
397798693b2Sdan ** In practice, the data structures this macro is used for are the primary
398798693b2Sdan ** key values of modified rows.
399798693b2Sdan */
4004131639cSdan #define HASH_APPEND(hash, add) ((hash) << 3) ^ (hash) ^ (unsigned int)(add)
401798693b2Sdan 
402798693b2Sdan /*
403798693b2Sdan ** Append the hash of the 64-bit integer passed as the second argument to the
404798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
405798693b2Sdan */
4064131639cSdan static unsigned int sessionHashAppendI64(unsigned int h, i64 i){
407e8d5648eSdan   h = HASH_APPEND(h, i & 0xFFFFFFFF);
408e8d5648eSdan   return HASH_APPEND(h, (i>>32)&0xFFFFFFFF);
409e8d5648eSdan }
410798693b2Sdan 
411798693b2Sdan /*
412798693b2Sdan ** Append the hash of the blob passed via the second and third arguments to
413798693b2Sdan ** the hash-key value passed as the first. Return the new hash-key value.
414798693b2Sdan */
4154131639cSdan static unsigned int sessionHashAppendBlob(unsigned int h, int n, const u8 *z){
416e8d5648eSdan   int i;
417e8d5648eSdan   for(i=0; i<n; i++) h = HASH_APPEND(h, z[i]);
418e8d5648eSdan   return h;
419e8d5648eSdan }
420e8d5648eSdan 
4214fccf43aSdan /*
422798693b2Sdan ** Append the hash of the data type passed as the second argument to the
423798693b2Sdan ** hash-key value passed as the first. Return the new hash-key value.
424798693b2Sdan */
425798693b2Sdan static unsigned int sessionHashAppendType(unsigned int h, int eType){
426798693b2Sdan   return HASH_APPEND(h, eType);
427798693b2Sdan }
428798693b2Sdan 
429798693b2Sdan /*
4304131639cSdan ** This function may only be called from within a pre-update callback.
4314131639cSdan ** It calculates a hash based on the primary key values of the old.* or
432798693b2Sdan ** new.* row currently available and, assuming no error occurs, writes it to
433798693b2Sdan ** *piHash before returning. If the primary key contains one or more NULL
434798693b2Sdan ** values, *pbNullPK is set to true before returning.
435798693b2Sdan **
436798693b2Sdan ** If an error occurs, an SQLite error code is returned and the final values
437798693b2Sdan ** of *piHash asn *pbNullPK are undefined. Otherwise, SQLITE_OK is returned
438798693b2Sdan ** and the output variables are set as described above.
4394fccf43aSdan */
440798693b2Sdan static int sessionPreupdateHash(
441e8d5648eSdan   sqlite3 *db,                    /* Database handle */
442e8d5648eSdan   SessionTable *pTab,             /* Session table handle */
443e8d5648eSdan   int bNew,                       /* True to hash the new.* PK */
44427453faeSdan   int *piHash,                    /* OUT: Hash value */
445798693b2Sdan   int *pbNullPK                   /* OUT: True if there are NULL values in PK */
446e8d5648eSdan ){
4474131639cSdan   unsigned int h = 0;             /* Hash value to return */
4484131639cSdan   int i;                          /* Used to iterate through columns */
449e8d5648eSdan 
45027453faeSdan   assert( *pbNullPK==0 );
451e8d5648eSdan   assert( pTab->nCol==sqlite3_preupdate_count(db) );
452e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
453e8d5648eSdan     if( pTab->abPK[i] ){
454e8d5648eSdan       int rc;
455e8d5648eSdan       int eType;
456e8d5648eSdan       sqlite3_value *pVal;
457e8d5648eSdan 
458e8d5648eSdan       if( bNew ){
459e8d5648eSdan         rc = sqlite3_preupdate_new(db, i, &pVal);
460e8d5648eSdan       }else{
461e8d5648eSdan         rc = sqlite3_preupdate_old(db, i, &pVal);
462e8d5648eSdan       }
46312ca0b56Sdan       if( rc!=SQLITE_OK ) return rc;
464e8d5648eSdan 
465e8d5648eSdan       eType = sqlite3_value_type(pVal);
466798693b2Sdan       h = sessionHashAppendType(h, eType);
4676734007dSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
468e8d5648eSdan         i64 iVal;
469e8d5648eSdan         if( eType==SQLITE_INTEGER ){
470e8d5648eSdan           iVal = sqlite3_value_int64(pVal);
471e8d5648eSdan         }else{
472e8d5648eSdan           double rVal = sqlite3_value_double(pVal);
473e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
474e8d5648eSdan           memcpy(&iVal, &rVal, 8);
475e8d5648eSdan         }
476e8d5648eSdan         h = sessionHashAppendI64(h, iVal);
4776734007dSdan       }else if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
4786734007dSdan         const u8 *z;
4793cc89d95Sdan         int n;
4806734007dSdan         if( eType==SQLITE_TEXT ){
4816734007dSdan           z = (const u8 *)sqlite3_value_text(pVal);
4826734007dSdan         }else{
4836734007dSdan           z = (const u8 *)sqlite3_value_blob(pVal);
484e8d5648eSdan         }
4853cc89d95Sdan         n = sqlite3_value_bytes(pVal);
4863cc89d95Sdan         if( !z && (eType!=SQLITE_BLOB || n>0) ) return SQLITE_NOMEM;
4873cc89d95Sdan         h = sessionHashAppendBlob(h, n, z);
4886734007dSdan       }else{
48927453faeSdan         assert( eType==SQLITE_NULL );
49027453faeSdan         *pbNullPK = 1;
491e8d5648eSdan       }
492e8d5648eSdan     }
493e8d5648eSdan   }
494e8d5648eSdan 
495e8d5648eSdan   *piHash = (h % pTab->nChange);
496e8d5648eSdan   return SQLITE_OK;
497e8d5648eSdan }
498e8d5648eSdan 
4994131639cSdan /*
5006cda207fSdan ** The buffer that the argument points to contains a serialized SQL value.
5016cda207fSdan ** Return the number of bytes of space occupied by the value (including
5026cda207fSdan ** the type byte).
5036cda207fSdan */
5046cda207fSdan static int sessionSerialLen(u8 *a){
5056cda207fSdan   int e = *a;
5066cda207fSdan   int n;
5076cda207fSdan   if( e==0 ) return 1;
5086cda207fSdan   if( e==SQLITE_NULL ) return 1;
5096cda207fSdan   if( e==SQLITE_INTEGER || e==SQLITE_FLOAT ) return 9;
5106cda207fSdan   return sessionVarintGet(&a[1], &n) + 1 + n;
5116cda207fSdan }
5126cda207fSdan 
5136cda207fSdan /*
5145d607a6eSdan ** Based on the primary key values stored in change aRecord, calculate a
515798693b2Sdan ** hash key. Assume the has table has nBucket buckets. The hash keys
5164131639cSdan ** calculated by this function are compatible with those calculated by
5174131639cSdan ** sessionPreupdateHash().
51864277f4aSdan **
51964277f4aSdan ** The bPkOnly argument is non-zero if the record at aRecord[] is from
52064277f4aSdan ** a patchset DELETE. In this case the non-PK fields are omitted entirely.
5214131639cSdan */
5224131639cSdan static unsigned int sessionChangeHash(
5234131639cSdan   SessionTable *pTab,             /* Table handle */
52464277f4aSdan   int bPkOnly,                    /* Record consists of PK fields only */
5255d607a6eSdan   u8 *aRecord,                    /* Change record */
5264131639cSdan   int nBucket                     /* Assume this many buckets in hash table */
527e8d5648eSdan ){
5284131639cSdan   unsigned int h = 0;             /* Value to return */
5294131639cSdan   int i;                          /* Used to iterate through columns */
5305d607a6eSdan   u8 *a = aRecord;                /* Used to iterate through change record */
531e8d5648eSdan 
532e8d5648eSdan   for(i=0; i<pTab->nCol; i++){
5336cda207fSdan     int eType = *a;
534e8d5648eSdan     int isPK = pTab->abPK[i];
53564277f4aSdan     if( bPkOnly && isPK==0 ) continue;
536e8d5648eSdan 
53727453faeSdan     /* It is not possible for eType to be SQLITE_NULL here. The session
53827453faeSdan     ** module does not record changes for rows with NULL values stored in
53927453faeSdan     ** primary key columns. */
54027453faeSdan     assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
54127453faeSdan          || eType==SQLITE_TEXT || eType==SQLITE_BLOB
5426cda207fSdan          || eType==SQLITE_NULL || eType==0
54327453faeSdan     );
5446cda207fSdan     assert( !isPK || (eType!=0 && eType!=SQLITE_NULL) );
54527453faeSdan 
5466cda207fSdan     if( isPK ){
5476cda207fSdan       a++;
548798693b2Sdan       h = sessionHashAppendType(h, eType);
54927453faeSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
5506734007dSdan         h = sessionHashAppendI64(h, sessionGetI64(a));
551e8d5648eSdan         a += 8;
55227453faeSdan       }else{
553e8d5648eSdan         int n;
554e8d5648eSdan         a += sessionVarintGet(a, &n);
5556734007dSdan         h = sessionHashAppendBlob(h, n, a);
556e8d5648eSdan         a += n;
557e8d5648eSdan       }
5586cda207fSdan     }else{
5596cda207fSdan       a += sessionSerialLen(a);
5606cda207fSdan     }
561e8d5648eSdan   }
562e8d5648eSdan   return (h % nBucket);
563e8d5648eSdan }
564e8d5648eSdan 
565798693b2Sdan /*
566798693b2Sdan ** Arguments aLeft and aRight are pointers to change records for table pTab.
567798693b2Sdan ** This function returns true if the two records apply to the same row (i.e.
568798693b2Sdan ** have the same values stored in the primary key columns), or false
569798693b2Sdan ** otherwise.
570798693b2Sdan */
5715d607a6eSdan static int sessionChangeEqual(
572798693b2Sdan   SessionTable *pTab,             /* Table used for PK definition */
573a71d2371Sdan   int bLeftPkOnly,                /* True if aLeft[] contains PK fields only */
5745d607a6eSdan   u8 *aLeft,                      /* Change record */
575a71d2371Sdan   int bRightPkOnly,               /* True if aRight[] contains PK fields only */
5765d607a6eSdan   u8 *aRight                      /* Change record */
5775d607a6eSdan ){
578798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor to iterate through aLeft */
579798693b2Sdan   u8 *a2 = aRight;                /* Cursor to iterate through aRight */
580798693b2Sdan   int iCol;                       /* Used to iterate through table columns */
5815d607a6eSdan 
582798693b2Sdan   for(iCol=0; iCol<pTab->nCol; iCol++){
5835d607a6eSdan     int n1 = sessionSerialLen(a1);
5845d607a6eSdan     int n2 = sessionSerialLen(a2);
5855d607a6eSdan 
586798693b2Sdan     if( pTab->abPK[iCol] && (n1!=n2 || memcmp(a1, a2, n1)) ){
5875d607a6eSdan       return 0;
5885d607a6eSdan     }
58964277f4aSdan     if( pTab->abPK[iCol] || bLeftPkOnly==0 ) a1 += n1;
59064277f4aSdan     if( pTab->abPK[iCol] || bRightPkOnly==0 ) a2 += n2;
5915d607a6eSdan   }
5925d607a6eSdan 
5935d607a6eSdan   return 1;
5945d607a6eSdan }
5955d607a6eSdan 
596798693b2Sdan /*
597798693b2Sdan ** Arguments aLeft and aRight both point to buffers containing change
598798693b2Sdan ** records with nCol columns. This function "merges" the two records into
599798693b2Sdan ** a single records which is written to the buffer at *paOut. *paOut is
600798693b2Sdan ** then set to point to one byte after the last byte written before
601798693b2Sdan ** returning.
602798693b2Sdan **
603798693b2Sdan ** The merging of records is done as follows: For each column, if the
604798693b2Sdan ** aRight record contains a value for the column, copy the value from
605798693b2Sdan ** their. Otherwise, if aLeft contains a value, copy it. If neither
606798693b2Sdan ** record contains a value for a given column, then neither does the
607798693b2Sdan ** output record.
608798693b2Sdan */
6095d607a6eSdan static void sessionMergeRecord(
6105d607a6eSdan   u8 **paOut,
611798693b2Sdan   int nCol,
6125d607a6eSdan   u8 *aLeft,
6135d607a6eSdan   u8 *aRight
6145d607a6eSdan ){
615798693b2Sdan   u8 *a1 = aLeft;                 /* Cursor used to iterate through aLeft */
616798693b2Sdan   u8 *a2 = aRight;                /* Cursor used to iterate through aRight */
617798693b2Sdan   u8 *aOut = *paOut;              /* Output cursor */
618798693b2Sdan   int iCol;                       /* Used to iterate from 0 to nCol */
6195d607a6eSdan 
620798693b2Sdan   for(iCol=0; iCol<nCol; iCol++){
6215d607a6eSdan     int n1 = sessionSerialLen(a1);
6225d607a6eSdan     int n2 = sessionSerialLen(a2);
6235d607a6eSdan     if( *a2 ){
6245d607a6eSdan       memcpy(aOut, a2, n2);
6255d607a6eSdan       aOut += n2;
6265d607a6eSdan     }else{
6275d607a6eSdan       memcpy(aOut, a1, n1);
6285d607a6eSdan       aOut += n1;
6295d607a6eSdan     }
6305d607a6eSdan     a1 += n1;
6315d607a6eSdan     a2 += n2;
6325d607a6eSdan   }
6335d607a6eSdan 
6345d607a6eSdan   *paOut = aOut;
6355d607a6eSdan }
6365d607a6eSdan 
637798693b2Sdan /*
638798693b2Sdan ** This is a helper function used by sessionMergeUpdate().
639798693b2Sdan **
640798693b2Sdan ** When this function is called, both *paOne and *paTwo point to a value
641798693b2Sdan ** within a change record. Before it returns, both have been advanced so
642798693b2Sdan ** as to point to the next value in the record.
643798693b2Sdan **
644798693b2Sdan ** If, when this function is called, *paTwo points to a valid value (i.e.
645fa29ecc4Sdan ** *paTwo[0] is not 0x00 - the "no value" placeholder), a copy of the *paTwo
646798693b2Sdan ** pointer is returned and *pnVal is set to the number of bytes in the
647798693b2Sdan ** serialized value. Otherwise, a copy of *paOne is returned and *pnVal
648798693b2Sdan ** set to the number of bytes in the value at *paOne. If *paOne points
649fa29ecc4Sdan ** to the "no value" placeholder, *pnVal is set to 1. In other words:
650fa29ecc4Sdan **
651fa29ecc4Sdan **   if( *paTwo is valid ) return *paTwo;
652fa29ecc4Sdan **   return *paOne;
653fa29ecc4Sdan **
654798693b2Sdan */
6555d607a6eSdan static u8 *sessionMergeValue(
656798693b2Sdan   u8 **paOne,                     /* IN/OUT: Left-hand buffer pointer */
657798693b2Sdan   u8 **paTwo,                     /* IN/OUT: Right-hand buffer pointer */
658798693b2Sdan   int *pnVal                      /* OUT: Bytes in returned value */
6595d607a6eSdan ){
6605d607a6eSdan   u8 *a1 = *paOne;
6615d607a6eSdan   u8 *a2 = *paTwo;
6625d607a6eSdan   u8 *pRet = 0;
6635d607a6eSdan   int n1;
6645d607a6eSdan 
6655d607a6eSdan   assert( a1 );
6665d607a6eSdan   if( a2 ){
6675d607a6eSdan     int n2 = sessionSerialLen(a2);
6685d607a6eSdan     if( *a2 ){
6695d607a6eSdan       *pnVal = n2;
6705d607a6eSdan       pRet = a2;
6715d607a6eSdan     }
6725d607a6eSdan     *paTwo = &a2[n2];
6735d607a6eSdan   }
6745d607a6eSdan 
6755d607a6eSdan   n1 = sessionSerialLen(a1);
6765d607a6eSdan   if( pRet==0 ){
6775d607a6eSdan     *pnVal = n1;
6785d607a6eSdan     pRet = a1;
6795d607a6eSdan   }
6805d607a6eSdan   *paOne = &a1[n1];
6815d607a6eSdan 
6825d607a6eSdan   return pRet;
6835d607a6eSdan }
6845d607a6eSdan 
685798693b2Sdan /*
686798693b2Sdan ** This function is used by changeset_concat() to merge two UPDATE changes
687798693b2Sdan ** on the same row.
688798693b2Sdan */
6895d607a6eSdan static int sessionMergeUpdate(
690798693b2Sdan   u8 **paOut,                     /* IN/OUT: Pointer to output buffer */
691798693b2Sdan   SessionTable *pTab,             /* Table change pertains to */
692a71d2371Sdan   int bPatchset,                  /* True if records are patchset records */
693798693b2Sdan   u8 *aOldRecord1,                /* old.* record for first change */
694798693b2Sdan   u8 *aOldRecord2,                /* old.* record for second change */
695798693b2Sdan   u8 *aNewRecord1,                /* new.* record for first change */
696798693b2Sdan   u8 *aNewRecord2                 /* new.* record for second change */
6975d607a6eSdan ){
6985d607a6eSdan   u8 *aOld1 = aOldRecord1;
6995d607a6eSdan   u8 *aOld2 = aOldRecord2;
7005d607a6eSdan   u8 *aNew1 = aNewRecord1;
7015d607a6eSdan   u8 *aNew2 = aNewRecord2;
7025d607a6eSdan 
7035d607a6eSdan   u8 *aOut = *paOut;
7045d607a6eSdan   int i;
70564277f4aSdan 
70664277f4aSdan   if( bPatchset==0 ){
7075d607a6eSdan     int bRequired = 0;
7085d607a6eSdan 
7095d607a6eSdan     assert( aOldRecord1 && aNewRecord1 );
7105d607a6eSdan 
7115d607a6eSdan     /* Write the old.* vector first. */
7125d607a6eSdan     for(i=0; i<pTab->nCol; i++){
7135d607a6eSdan       int nOld;
7145d607a6eSdan       u8 *aOld;
7155d607a6eSdan       int nNew;
7165d607a6eSdan       u8 *aNew;
7175d607a6eSdan 
7185d607a6eSdan       aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7195d607a6eSdan       aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
7205d607a6eSdan       if( pTab->abPK[i] || nOld!=nNew || memcmp(aOld, aNew, nNew) ){
7215d607a6eSdan         if( pTab->abPK[i]==0 ) bRequired = 1;
7225d607a6eSdan         memcpy(aOut, aOld, nOld);
7235d607a6eSdan         aOut += nOld;
7245d607a6eSdan       }else{
7255d607a6eSdan         *(aOut++) = '\0';
7265d607a6eSdan       }
7275d607a6eSdan     }
7285d607a6eSdan 
7295d607a6eSdan     if( !bRequired ) return 0;
73064277f4aSdan   }
7315d607a6eSdan 
7325d607a6eSdan   /* Write the new.* vector */
7335d607a6eSdan   aOld1 = aOldRecord1;
7345d607a6eSdan   aOld2 = aOldRecord2;
7355d607a6eSdan   aNew1 = aNewRecord1;
7365d607a6eSdan   aNew2 = aNewRecord2;
7375d607a6eSdan   for(i=0; i<pTab->nCol; i++){
7385d607a6eSdan     int nOld;
7395d607a6eSdan     u8 *aOld;
7405d607a6eSdan     int nNew;
7415d607a6eSdan     u8 *aNew;
7425d607a6eSdan 
7435d607a6eSdan     aOld = sessionMergeValue(&aOld1, &aOld2, &nOld);
7445d607a6eSdan     aNew = sessionMergeValue(&aNew1, &aNew2, &nNew);
74564277f4aSdan     if( bPatchset==0
74664277f4aSdan      && (pTab->abPK[i] || (nOld==nNew && 0==memcmp(aOld, aNew, nNew)))
74764277f4aSdan     ){
7485d607a6eSdan       *(aOut++) = '\0';
7495d607a6eSdan     }else{
7505d607a6eSdan       memcpy(aOut, aNew, nNew);
7515d607a6eSdan       aOut += nNew;
7525d607a6eSdan     }
7535d607a6eSdan   }
7545d607a6eSdan 
7555d607a6eSdan   *paOut = aOut;
7565d607a6eSdan   return 1;
7575d607a6eSdan }
7585d607a6eSdan 
75977fc1d5bSdan /*
76077fc1d5bSdan ** This function is only called from within a pre-update-hook callback.
76177fc1d5bSdan ** It determines if the current pre-update-hook change affects the same row
76277fc1d5bSdan ** as the change stored in argument pChange. If so, it returns true. Otherwise
76377fc1d5bSdan ** if the pre-update-hook does not affect the same row as pChange, it returns
76477fc1d5bSdan ** false.
76577fc1d5bSdan */
76677fc1d5bSdan static int sessionPreupdateEqual(
76777fc1d5bSdan   sqlite3 *db,                    /* Database handle */
76877fc1d5bSdan   SessionTable *pTab,             /* Table associated with change */
76977fc1d5bSdan   SessionChange *pChange,         /* Change to compare to */
77077fc1d5bSdan   int op                          /* Current pre-update operation */
771e8d5648eSdan ){
77277fc1d5bSdan   int iCol;                       /* Used to iterate through columns */
77377fc1d5bSdan   u8 *a = pChange->aRecord;       /* Cursor used to scan change record */
774e8d5648eSdan 
77577fc1d5bSdan   assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
77677fc1d5bSdan   for(iCol=0; iCol<pTab->nCol; iCol++){
77777fc1d5bSdan     if( !pTab->abPK[iCol] ){
778798693b2Sdan       a += sessionSerialLen(a);
779e8d5648eSdan     }else{
7806734007dSdan       sqlite3_value *pVal;        /* Value returned by preupdate_new/old */
7816734007dSdan       int rc;                     /* Error code from preupdate_new/old */
782798693b2Sdan       int eType = *a++;           /* Type of value from change record */
7836734007dSdan 
7846734007dSdan       /* The following calls to preupdate_new() and preupdate_old() can not
7856734007dSdan       ** fail. This is because they cache their return values, and by the
7866734007dSdan       ** time control flows to here they have already been called once from
7876734007dSdan       ** within sessionPreupdateHash(). The first two asserts below verify
7886734007dSdan       ** this (that the method has already been called). */
78977fc1d5bSdan       if( op==SQLITE_INSERT ){
7906734007dSdan         assert( db->pPreUpdate->pNewUnpacked || db->pPreUpdate->aNew );
79177fc1d5bSdan         rc = sqlite3_preupdate_new(db, iCol, &pVal);
792e8d5648eSdan       }else{
7936734007dSdan         assert( db->pPreUpdate->pUnpacked );
79477fc1d5bSdan         rc = sqlite3_preupdate_old(db, iCol, &pVal);
795e8d5648eSdan       }
7966734007dSdan       assert( rc==SQLITE_OK );
79777fc1d5bSdan       if( sqlite3_value_type(pVal)!=eType ) return 0;
798e8d5648eSdan 
79912ca0b56Sdan       /* A SessionChange object never has a NULL value in a PK column */
80012ca0b56Sdan       assert( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT
80112ca0b56Sdan            || eType==SQLITE_BLOB    || eType==SQLITE_TEXT
80212ca0b56Sdan       );
80312ca0b56Sdan 
80412ca0b56Sdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
805e8d5648eSdan         i64 iVal = sessionGetI64(a);
806e8d5648eSdan         a += 8;
807e8d5648eSdan         if( eType==SQLITE_INTEGER ){
80877fc1d5bSdan           if( sqlite3_value_int64(pVal)!=iVal ) return 0;
809e8d5648eSdan         }else{
810e8d5648eSdan           double rVal;
811e8d5648eSdan           assert( sizeof(iVal)==8 && sizeof(rVal)==8 );
812e8d5648eSdan           memcpy(&rVal, &iVal, 8);
81377fc1d5bSdan           if( sqlite3_value_double(pVal)!=rVal ) return 0;
814e8d5648eSdan         }
81512ca0b56Sdan       }else{
816e8d5648eSdan         int n;
817e8d5648eSdan         const u8 *z;
818e8d5648eSdan         a += sessionVarintGet(a, &n);
81977fc1d5bSdan         if( sqlite3_value_bytes(pVal)!=n ) return 0;
82012ca0b56Sdan         if( eType==SQLITE_TEXT ){
82112ca0b56Sdan           z = sqlite3_value_text(pVal);
82212ca0b56Sdan         }else{
82312ca0b56Sdan           z = sqlite3_value_blob(pVal);
82412ca0b56Sdan         }
82577fc1d5bSdan         if( memcmp(a, z, n) ) return 0;
826e8d5648eSdan         a += n;
827e8d5648eSdan         break;
828e8d5648eSdan       }
829e8d5648eSdan     }
830e8d5648eSdan   }
831e8d5648eSdan 
83277fc1d5bSdan   return 1;
8334fccf43aSdan }
8344fccf43aSdan 
8354fccf43aSdan /*
8364fccf43aSdan ** If required, grow the hash table used to store changes on table pTab
8374fccf43aSdan ** (part of the session pSession). If a fatal OOM error occurs, set the
8384fccf43aSdan ** session object to failed and return SQLITE_ERROR. Otherwise, return
8394fccf43aSdan ** SQLITE_OK.
8404fccf43aSdan **
8414fccf43aSdan ** It is possible that a non-fatal OOM error occurs in this function. In
8424fccf43aSdan ** that case the hash-table does not grow, but SQLITE_OK is returned anyway.
8434fccf43aSdan ** Growing the hash table in this case is a performance optimization only,
8444fccf43aSdan ** it is not required for correct operation.
8454fccf43aSdan */
84664277f4aSdan static int sessionGrowHash(int bPatchset, SessionTable *pTab){
8474fccf43aSdan   if( pTab->nChange==0 || pTab->nEntry>=(pTab->nChange/2) ){
8484fccf43aSdan     int i;
8494fccf43aSdan     SessionChange **apNew;
8504fccf43aSdan     int nNew = (pTab->nChange ? pTab->nChange : 128) * 2;
8514fccf43aSdan 
8524fccf43aSdan     apNew = (SessionChange **)sqlite3_malloc(sizeof(SessionChange *) * nNew);
8534fccf43aSdan     if( apNew==0 ){
8544fccf43aSdan       if( pTab->nChange==0 ){
8554fccf43aSdan         return SQLITE_ERROR;
8564fccf43aSdan       }
8574fccf43aSdan       return SQLITE_OK;
8584fccf43aSdan     }
8594fccf43aSdan     memset(apNew, 0, sizeof(SessionChange *) * nNew);
8604fccf43aSdan 
8614fccf43aSdan     for(i=0; i<pTab->nChange; i++){
8624fccf43aSdan       SessionChange *p;
8634fccf43aSdan       SessionChange *pNext;
8644fccf43aSdan       for(p=pTab->apChange[i]; p; p=pNext){
86564277f4aSdan         int bPkOnly = (p->op==SQLITE_DELETE && bPatchset);
86664277f4aSdan         int iHash = sessionChangeHash(pTab, bPkOnly, p->aRecord, nNew);
8674fccf43aSdan         pNext = p->pNext;
8684fccf43aSdan         p->pNext = apNew[iHash];
8694fccf43aSdan         apNew[iHash] = p;
8704fccf43aSdan       }
8714fccf43aSdan     }
8724fccf43aSdan 
8734fccf43aSdan     sqlite3_free(pTab->apChange);
8744fccf43aSdan     pTab->nChange = nNew;
8754fccf43aSdan     pTab->apChange = apNew;
8764fccf43aSdan   }
8774fccf43aSdan 
8784fccf43aSdan   return SQLITE_OK;
8794fccf43aSdan }
8804fccf43aSdan 
881296c7658Sdan /*
882e8d5648eSdan ** This function queries the database for the names of the columns of table
883e8d5648eSdan ** zThis, in schema zDb. It is expected that the table has nCol columns. If
884e8d5648eSdan ** not, SQLITE_SCHEMA is returned and none of the output variables are
885e8d5648eSdan ** populated.
886e8d5648eSdan **
88777fc1d5bSdan ** Otherwise, if they are not NULL, variable *pnCol is set to the number
88877fc1d5bSdan ** of columns in the database table and variable *pzTab is set to point to a
889e8d5648eSdan ** nul-terminated copy of the table name. *pazCol (if not NULL) is set to
890e8d5648eSdan ** point to an array of pointers to column names. And *pabPK (again, if not
891e8d5648eSdan ** NULL) is set to point to an array of booleans - true if the corresponding
892e8d5648eSdan ** column is part of the primary key.
893e8d5648eSdan **
894e8d5648eSdan ** For example, if the table is declared as:
895e8d5648eSdan **
896e8d5648eSdan **     CREATE TABLE tbl1(w, x, y, z, PRIMARY KEY(w, z));
897e8d5648eSdan **
89877fc1d5bSdan ** Then the four output variables are populated as follows:
899e8d5648eSdan **
90077fc1d5bSdan **     *pnCol  = 4
901e8d5648eSdan **     *pzTab  = "tbl1"
902e8d5648eSdan **     *pazCol = {"w", "x", "y", "z"}
903e8d5648eSdan **     *pabPK  = {1, 0, 0, 1}
904e8d5648eSdan **
905e8d5648eSdan ** All returned buffers are part of the same single allocation, which must
906e8d5648eSdan ** be freed using sqlite3_free() by the caller. If pazCol was not NULL, then
907e8d5648eSdan ** pointer *pazCol should be freed to release all memory. Otherwise, pointer
908e8d5648eSdan ** *pabPK. It is illegal for both pazCol and pabPK to be NULL.
909e8d5648eSdan */
910e8d5648eSdan static int sessionTableInfo(
911e8d5648eSdan   sqlite3 *db,                    /* Database connection */
912e8d5648eSdan   const char *zDb,                /* Name of attached database (e.g. "main") */
913e8d5648eSdan   const char *zThis,              /* Table name */
914ca62ad57Sdan   int *pnCol,                     /* OUT: number of columns */
915e8d5648eSdan   const char **pzTab,             /* OUT: Copy of zThis */
916e8d5648eSdan   const char ***pazCol,           /* OUT: Array of column names for table */
917e8d5648eSdan   u8 **pabPK                      /* OUT: Array of booleans - true for PK col */
918e8d5648eSdan ){
919e8d5648eSdan   char *zPragma;
920e8d5648eSdan   sqlite3_stmt *pStmt;
921e8d5648eSdan   int rc;
922e8d5648eSdan   int nByte;
923e8d5648eSdan   int nDbCol = 0;
924e8d5648eSdan   int nThis;
925e8d5648eSdan   int i;
926e8d5648eSdan   u8 *pAlloc;
927db04571cSdan   char **azCol = 0;
928e8d5648eSdan   u8 *abPK;
929e8d5648eSdan 
930db04571cSdan   assert( pazCol && pabPK );
931e8d5648eSdan 
932cfdbde21Sdrh   nThis = sqlite3Strlen30(zThis);
933e8d5648eSdan   zPragma = sqlite3_mprintf("PRAGMA '%q'.table_info('%q')", zDb, zThis);
934e8d5648eSdan   if( !zPragma ) return SQLITE_NOMEM;
935e8d5648eSdan 
936e8d5648eSdan   rc = sqlite3_prepare_v2(db, zPragma, -1, &pStmt, 0);
937e8d5648eSdan   sqlite3_free(zPragma);
938e8d5648eSdan   if( rc!=SQLITE_OK ) return rc;
939e8d5648eSdan 
940e8d5648eSdan   nByte = nThis + 1;
941e8d5648eSdan   while( SQLITE_ROW==sqlite3_step(pStmt) ){
942e8d5648eSdan     nByte += sqlite3_column_bytes(pStmt, 1);
943e8d5648eSdan     nDbCol++;
944e8d5648eSdan   }
945e8d5648eSdan   rc = sqlite3_reset(pStmt);
946e8d5648eSdan 
947e8d5648eSdan   if( rc==SQLITE_OK ){
948e8d5648eSdan     nByte += nDbCol * (sizeof(const char *) + sizeof(u8) + 1);
949e8d5648eSdan     pAlloc = sqlite3_malloc(nByte);
950e8d5648eSdan     if( pAlloc==0 ){
951e8d5648eSdan       rc = SQLITE_NOMEM;
952e8d5648eSdan     }
953e8d5648eSdan   }
954e8d5648eSdan   if( rc==SQLITE_OK ){
955e8d5648eSdan     azCol = (char **)pAlloc;
956ca62ad57Sdan     pAlloc = (u8 *)&azCol[nDbCol];
957e8d5648eSdan     abPK = (u8 *)pAlloc;
958ca62ad57Sdan     pAlloc = &abPK[nDbCol];
959e8d5648eSdan     if( pzTab ){
960e8d5648eSdan       memcpy(pAlloc, zThis, nThis+1);
961e8d5648eSdan       *pzTab = (char *)pAlloc;
962e8d5648eSdan       pAlloc += nThis+1;
963e8d5648eSdan     }
964e8d5648eSdan 
965e8d5648eSdan     i = 0;
966e8d5648eSdan     while( SQLITE_ROW==sqlite3_step(pStmt) ){
967e8d5648eSdan       int nName = sqlite3_column_bytes(pStmt, 1);
968e8d5648eSdan       const unsigned char *zName = sqlite3_column_text(pStmt, 1);
969e8d5648eSdan       if( zName==0 ) break;
970e8d5648eSdan       memcpy(pAlloc, zName, nName+1);
971e8d5648eSdan       azCol[i] = (char *)pAlloc;
972e8d5648eSdan       pAlloc += nName+1;
973db04571cSdan       abPK[i] = sqlite3_column_int(pStmt, 5);
974e8d5648eSdan       i++;
975e8d5648eSdan     }
976e8d5648eSdan     rc = sqlite3_reset(pStmt);
977e8d5648eSdan 
978e8d5648eSdan   }
979e8d5648eSdan 
980e8d5648eSdan   /* If successful, populate the output variables. Otherwise, zero them and
981e8d5648eSdan   ** free any allocation made. An error code will be returned in this case.
982e8d5648eSdan   */
983e8d5648eSdan   if( rc==SQLITE_OK ){
984db04571cSdan     *pazCol = (const char **)azCol;
985db04571cSdan     *pabPK = abPK;
986ca62ad57Sdan     *pnCol = nDbCol;
987e8d5648eSdan   }else{
988db04571cSdan     *pazCol = 0;
989db04571cSdan     *pabPK = 0;
990ca62ad57Sdan     *pnCol = 0;
991e8d5648eSdan     if( pzTab ) *pzTab = 0;
992db04571cSdan     sqlite3_free(azCol);
993e8d5648eSdan   }
994e8d5648eSdan   sqlite3_finalize(pStmt);
995e8d5648eSdan   return rc;
996e8d5648eSdan }
997e8d5648eSdan 
998e8d5648eSdan /*
999296c7658Sdan ** This function is only called from within a pre-update handler for a
1000296c7658Sdan ** write to table pTab, part of session pSession. If this is the first
1001296c7658Sdan ** write to this table, set the SessionTable.nCol variable to the number
1002296c7658Sdan ** of columns in the table.
1003296c7658Sdan **
1004296c7658Sdan ** Otherwise, if this is not the first time this table has been written
1005296c7658Sdan ** to, check that the number of columns in the table has not changed. If
1006296c7658Sdan ** it has not, return zero.
1007296c7658Sdan **
1008296c7658Sdan ** If the number of columns in the table has changed since the last write
1009296c7658Sdan ** was recorded, set the session error-code to SQLITE_SCHEMA and return
1010296c7658Sdan ** non-zero. Users are not allowed to change the number of columns in a table
1011296c7658Sdan ** for which changes are being recorded by the session module. If they do so,
1012296c7658Sdan ** it is an error.
1013296c7658Sdan */
10144fccf43aSdan static int sessionInitTable(sqlite3_session *pSession, SessionTable *pTab){
10154fccf43aSdan   if( pTab->nCol==0 ){
1016e8d5648eSdan     assert( pTab->azCol==0 || pTab->abPK==0 );
1017e8d5648eSdan     pSession->rc = sessionTableInfo(pSession->db, pSession->zDb,
1018ca62ad57Sdan         pTab->zName, &pTab->nCol, 0, &pTab->azCol, &pTab->abPK
1019e8d5648eSdan     );
1020ca62ad57Sdan   }
1021ca62ad57Sdan   if( pSession->rc==SQLITE_OK
1022ca62ad57Sdan    && pTab->nCol!=sqlite3_preupdate_count(pSession->db)
1023ca62ad57Sdan   ){
10244fccf43aSdan     pSession->rc = SQLITE_SCHEMA;
10254fccf43aSdan   }
1026e8d5648eSdan   return pSession->rc;
1027e8d5648eSdan }
1028e8d5648eSdan 
102977fc1d5bSdan /*
103077fc1d5bSdan ** This function is only called from with a pre-update-hook reporting a
103177fc1d5bSdan ** change on table pTab (attached to session pSession). The type of change
103277fc1d5bSdan ** (UPDATE, INSERT, DELETE) is specified by the first argument.
103377fc1d5bSdan **
103477fc1d5bSdan ** Unless one is already present or an error occurs, an entry is added
103577fc1d5bSdan ** to the changed-rows hash table associated with table pTab.
103677fc1d5bSdan */
1037e8d5648eSdan static void sessionPreupdateOneChange(
103877fc1d5bSdan   int op,                         /* One of SQLITE_UPDATE, INSERT, DELETE */
103977fc1d5bSdan   sqlite3_session *pSession,      /* Session object pTab is attached to */
104077fc1d5bSdan   SessionTable *pTab              /* Table that change applies to */
1041e8d5648eSdan ){
1042e8d5648eSdan   sqlite3 *db = pSession->db;
1043e8d5648eSdan   int iHash;
104427453faeSdan   int bNullPk = 0;
1045e8d5648eSdan   int rc = SQLITE_OK;
1046e8d5648eSdan 
1047e8d5648eSdan   if( pSession->rc ) return;
1048e8d5648eSdan 
1049e8d5648eSdan   /* Load table details if required */
1050e8d5648eSdan   if( sessionInitTable(pSession, pTab) ) return;
1051e8d5648eSdan 
1052e8d5648eSdan   /* Grow the hash table if required */
105364277f4aSdan   if( sessionGrowHash(0, pTab) ){
10545d607a6eSdan     pSession->rc = SQLITE_NOMEM;
10555d607a6eSdan     return;
10565d607a6eSdan   }
1057e8d5648eSdan 
105880fe2d93Sdan   /* Calculate the hash-key for this change. If the primary key of the row
105980fe2d93Sdan   ** includes a NULL value, exit early. Such changes are ignored by the
106080fe2d93Sdan   ** session module. */
106127453faeSdan   rc = sessionPreupdateHash(db, pTab, op==SQLITE_INSERT, &iHash, &bNullPk);
106280fe2d93Sdan   if( rc!=SQLITE_OK ) goto error_out;
106380fe2d93Sdan 
106480fe2d93Sdan   if( bNullPk==0 ){
106580fe2d93Sdan     /* Search the hash table for an existing record for this row. */
1066b4480e94Sdan     SessionChange *pC;
10676734007dSdan     for(pC=pTab->apChange[iHash]; pC; pC=pC->pNext){
106877fc1d5bSdan       if( sessionPreupdateEqual(db, pTab, pC, op) ) break;
1069e8d5648eSdan     }
107080fe2d93Sdan 
1071e8d5648eSdan     if( pC==0 ){
1072e8d5648eSdan       /* Create a new change object containing all the old values (if
1073e8d5648eSdan       ** this is an SQLITE_UPDATE or SQLITE_DELETE), or just the PK
1074e8d5648eSdan       ** values (if this is an INSERT). */
1075b4480e94Sdan       SessionChange *pChange; /* New change object */
1076e8d5648eSdan       int nByte;              /* Number of bytes to allocate */
1077e8d5648eSdan       int i;                  /* Used to iterate through columns */
1078e8d5648eSdan 
1079b4480e94Sdan       assert( rc==SQLITE_OK );
1080e8d5648eSdan       pTab->nEntry++;
1081e8d5648eSdan 
1082e8d5648eSdan       /* Figure out how large an allocation is required */
1083e8d5648eSdan       nByte = sizeof(SessionChange);
108480fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1085e8d5648eSdan         sqlite3_value *p = 0;
1086e8d5648eSdan         if( op!=SQLITE_INSERT ){
108780fe2d93Sdan           TESTONLY(int trc = ) sqlite3_preupdate_old(pSession->db, i, &p);
108880fe2d93Sdan           assert( trc==SQLITE_OK );
108980fe2d93Sdan         }else if( pTab->abPK[i] ){
109080fe2d93Sdan           TESTONLY(int trc = ) sqlite3_preupdate_new(pSession->db, i, &p);
109180fe2d93Sdan           assert( trc==SQLITE_OK );
1092e8d5648eSdan         }
109380fe2d93Sdan 
109480fe2d93Sdan         /* This may fail if SQLite value p contains a utf-16 string that must
109580fe2d93Sdan         ** be converted to utf-8 and an OOM error occurs while doing so. */
1096e8d5648eSdan         rc = sessionSerializeValue(0, p, &nByte);
109780fe2d93Sdan         if( rc!=SQLITE_OK ) goto error_out;
1098e8d5648eSdan       }
1099e8d5648eSdan 
1100e8d5648eSdan       /* Allocate the change object */
1101e8d5648eSdan       pChange = (SessionChange *)sqlite3_malloc(nByte);
1102e8d5648eSdan       if( !pChange ){
1103e8d5648eSdan         rc = SQLITE_NOMEM;
110480fe2d93Sdan         goto error_out;
1105e8d5648eSdan       }else{
1106e8d5648eSdan         memset(pChange, 0, sizeof(SessionChange));
1107e8d5648eSdan         pChange->aRecord = (u8 *)&pChange[1];
1108e8d5648eSdan       }
1109e8d5648eSdan 
111080fe2d93Sdan       /* Populate the change object. None of the preupdate_old(),
111180fe2d93Sdan       ** preupdate_new() or SerializeValue() calls below may fail as all
111280fe2d93Sdan       ** required values and encodings have already been cached in memory.
111380fe2d93Sdan       ** It is not possible for an OOM to occur in this block. */
1114e8d5648eSdan       nByte = 0;
111580fe2d93Sdan       for(i=0; i<pTab->nCol; i++){
1116e8d5648eSdan         sqlite3_value *p = 0;
1117e8d5648eSdan         if( op!=SQLITE_INSERT ){
111880fe2d93Sdan           sqlite3_preupdate_old(pSession->db, i, &p);
111980fe2d93Sdan         }else if( pTab->abPK[i] ){
112080fe2d93Sdan           sqlite3_preupdate_new(pSession->db, i, &p);
1121e8d5648eSdan         }
112280fe2d93Sdan         sessionSerializeValue(&pChange->aRecord[nByte], p, &nByte);
1123e8d5648eSdan       }
112480fe2d93Sdan 
112580fe2d93Sdan       /* Add the change to the hash-table */
1126b4480e94Sdan       if( pSession->bIndirect || sqlite3_preupdate_depth(pSession->db) ){
1127b4480e94Sdan         pChange->bIndirect = 1;
1128b4480e94Sdan       }
112912ca0b56Sdan       pChange->nRecord = nByte;
1130798693b2Sdan       pChange->op = op;
1131e8d5648eSdan       pChange->pNext = pTab->apChange[iHash];
1132e8d5648eSdan       pTab->apChange[iHash] = pChange;
113380fe2d93Sdan 
113480fe2d93Sdan     }else if( pC->bIndirect ){
1135b4480e94Sdan       /* If the existing change is considered "indirect", but this current
1136b4480e94Sdan       ** change is "direct", mark the change object as direct. */
1137b4480e94Sdan       if( sqlite3_preupdate_depth(pSession->db)==0 && pSession->bIndirect==0 ){
1138b4480e94Sdan         pC->bIndirect = 0;
1139b4480e94Sdan       }
1140e8d5648eSdan     }
11414fccf43aSdan   }
114212ca0b56Sdan 
114312ca0b56Sdan   /* If an error has occurred, mark the session object as failed. */
114480fe2d93Sdan  error_out:
114512ca0b56Sdan   if( rc!=SQLITE_OK ){
114612ca0b56Sdan     pSession->rc = rc;
114712ca0b56Sdan   }
114827453faeSdan }
11494fccf43aSdan 
11504fccf43aSdan /*
11514fccf43aSdan ** The 'pre-update' hook registered by this module with SQLite databases.
11524fccf43aSdan */
11534fccf43aSdan static void xPreUpdate(
11544fccf43aSdan   void *pCtx,                     /* Copy of third arg to preupdate_hook() */
11554fccf43aSdan   sqlite3 *db,                    /* Database handle */
11564fccf43aSdan   int op,                         /* SQLITE_UPDATE, DELETE or INSERT */
11574fccf43aSdan   char const *zDb,                /* Database name */
11584fccf43aSdan   char const *zName,              /* Table name */
11594fccf43aSdan   sqlite3_int64 iKey1,            /* Rowid of row about to be deleted/updated */
11604fccf43aSdan   sqlite3_int64 iKey2             /* New rowid value (for a rowid UPDATE) */
11614fccf43aSdan ){
11624fccf43aSdan   sqlite3_session *pSession;
1163cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb);
1164cfdbde21Sdrh   int nName = sqlite3Strlen30(zName);
11654fccf43aSdan 
11664c220252Sdan   assert( sqlite3_mutex_held(db->mutex) );
11674c220252Sdan 
11684fccf43aSdan   for(pSession=(sqlite3_session *)pCtx; pSession; pSession=pSession->pNext){
11694fccf43aSdan     SessionTable *pTab;
1170296c7658Sdan 
1171e8d5648eSdan     /* If this session is attached to a different database ("main", "temp"
1172e8d5648eSdan     ** etc.), or if it is not currently enabled, there is nothing to do. Skip
1173e8d5648eSdan     ** to the next session object attached to this database. */
1174296c7658Sdan     if( pSession->bEnable==0 ) continue;
11754fccf43aSdan     if( pSession->rc ) continue;
11764fccf43aSdan     if( sqlite3_strnicmp(zDb, pSession->zDb, nDb+1) ) continue;
1177296c7658Sdan 
1178ff4d0f41Sdan     for(pTab=pSession->pTable; pTab || pSession->bAutoAttach; pTab=pTab->pNext){
1179ff4d0f41Sdan       if( !pTab ){
1180ff4d0f41Sdan         /* This branch is taken if table zName has not yet been attached to
1181ff4d0f41Sdan         ** this session and the auto-attach flag is set.  */
11827531a5a3Sdan 
11837531a5a3Sdan         /* If there is a table-filter configured, invoke it. If it returns 0,
11847531a5a3Sdan         ** this change will not be recorded. Break out of the loop early in
11857531a5a3Sdan         ** this case.  */
11867531a5a3Sdan         if( pSession->xTableFilter
11877531a5a3Sdan          && pSession->xTableFilter(pSession->pFilterCtx, zName)==0
11887531a5a3Sdan         ){
11897531a5a3Sdan           break;
11907531a5a3Sdan         }
11917531a5a3Sdan 
1192ff4d0f41Sdan         pSession->rc = sqlite3session_attach(pSession,zName);
1193245b49b2Sdan         if( pSession->rc ) break;
1194ff4d0f41Sdan         pTab = pSession->pTable;
1195ff4d0f41Sdan         assert( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) );
1196ff4d0f41Sdan       }
1197ff4d0f41Sdan 
11984fccf43aSdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ){
1199e8d5648eSdan         sessionPreupdateOneChange(op, pSession, pTab);
1200e8d5648eSdan         if( op==SQLITE_UPDATE ){
1201e8d5648eSdan           sessionPreupdateOneChange(SQLITE_INSERT, pSession, pTab);
12024fccf43aSdan         }
12034fccf43aSdan         break;
12044fccf43aSdan       }
12054fccf43aSdan     }
12064fccf43aSdan   }
1207296c7658Sdan }
12084fccf43aSdan 
12094fccf43aSdan /*
12104fccf43aSdan ** Create a session object. This session object will record changes to
12114fccf43aSdan ** database zDb attached to connection db.
12124fccf43aSdan */
12134fccf43aSdan int sqlite3session_create(
12144fccf43aSdan   sqlite3 *db,                    /* Database handle */
12154fccf43aSdan   const char *zDb,                /* Name of db (e.g. "main") */
12164fccf43aSdan   sqlite3_session **ppSession     /* OUT: New session object */
12174fccf43aSdan ){
1218296c7658Sdan   sqlite3_session *pNew;          /* Newly allocated session object */
1219296c7658Sdan   sqlite3_session *pOld;          /* Session object already attached to db */
1220cfdbde21Sdrh   int nDb = sqlite3Strlen30(zDb); /* Length of zDb in bytes */
12214fccf43aSdan 
1222296c7658Sdan   /* Zero the output value in case an error occurs. */
12234fccf43aSdan   *ppSession = 0;
12244fccf43aSdan 
12254fccf43aSdan   /* Allocate and populate the new session object. */
12264fccf43aSdan   pNew = (sqlite3_session *)sqlite3_malloc(sizeof(sqlite3_session) + nDb + 1);
12274fccf43aSdan   if( !pNew ) return SQLITE_NOMEM;
12284fccf43aSdan   memset(pNew, 0, sizeof(sqlite3_session));
12294fccf43aSdan   pNew->db = db;
12304fccf43aSdan   pNew->zDb = (char *)&pNew[1];
1231296c7658Sdan   pNew->bEnable = 1;
12324fccf43aSdan   memcpy(pNew->zDb, zDb, nDb+1);
12334fccf43aSdan 
12344fccf43aSdan   /* Add the new session object to the linked list of session objects
12354fccf43aSdan   ** attached to database handle $db. Do this under the cover of the db
12364fccf43aSdan   ** handle mutex.  */
12374fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
12384fccf43aSdan   pOld = (sqlite3_session*)sqlite3_preupdate_hook(db, xPreUpdate, (void*)pNew);
12394fccf43aSdan   pNew->pNext = pOld;
12404fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
12414fccf43aSdan 
12424fccf43aSdan   *ppSession = pNew;
12434fccf43aSdan   return SQLITE_OK;
12444fccf43aSdan }
12454fccf43aSdan 
124677fc1d5bSdan /*
124777fc1d5bSdan ** Free the list of table objects passed as the first argument. The contents
124877fc1d5bSdan ** of the changed-rows hash tables are also deleted.
124977fc1d5bSdan */
12501ffe7c7fSdrh static void sessionDeleteTable(SessionTable *pList){
12515d607a6eSdan   SessionTable *pNext;
12525d607a6eSdan   SessionTable *pTab;
12535d607a6eSdan 
12545d607a6eSdan   for(pTab=pList; pTab; pTab=pNext){
12555d607a6eSdan     int i;
12565d607a6eSdan     pNext = pTab->pNext;
12575d607a6eSdan     for(i=0; i<pTab->nChange; i++){
12585d607a6eSdan       SessionChange *p;
12595d607a6eSdan       SessionChange *pNext;
12605d607a6eSdan       for(p=pTab->apChange[i]; p; p=pNext){
12615d607a6eSdan         pNext = p->pNext;
12625d607a6eSdan         sqlite3_free(p);
12635d607a6eSdan       }
12645d607a6eSdan     }
12655d607a6eSdan     sqlite3_free((char*)pTab->azCol);  /* cast works around VC++ bug */
12665d607a6eSdan     sqlite3_free(pTab->apChange);
12675d607a6eSdan     sqlite3_free(pTab);
12685d607a6eSdan   }
12695d607a6eSdan }
12705d607a6eSdan 
12714fccf43aSdan /*
12724fccf43aSdan ** Delete a session object previously allocated using sqlite3session_create().
12734fccf43aSdan */
12744fccf43aSdan void sqlite3session_delete(sqlite3_session *pSession){
12754fccf43aSdan   sqlite3 *db = pSession->db;
12764fccf43aSdan   sqlite3_session *pHead;
12774fccf43aSdan   sqlite3_session **pp;
12784fccf43aSdan 
1279296c7658Sdan   /* Unlink the session from the linked list of sessions attached to the
1280296c7658Sdan   ** database handle. Hold the db mutex while doing so.  */
12814fccf43aSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
12824fccf43aSdan   pHead = (sqlite3_session*)sqlite3_preupdate_hook(db, 0, 0);
12834fccf43aSdan   for(pp=&pHead; (*pp)!=pSession; pp=&((*pp)->pNext));
12844fccf43aSdan   *pp = (*pp)->pNext;
12854fccf43aSdan   if( pHead ) sqlite3_preupdate_hook(db, xPreUpdate, (void *)pHead);
12864fccf43aSdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
12874fccf43aSdan 
1288296c7658Sdan   /* Delete all attached table objects. And the contents of their
1289296c7658Sdan   ** associated hash-tables. */
12905d607a6eSdan   sessionDeleteTable(pSession->pTable);
12914fccf43aSdan 
1292296c7658Sdan   /* Free the session object itself. */
12934fccf43aSdan   sqlite3_free(pSession);
12944fccf43aSdan }
12954fccf43aSdan 
12964fccf43aSdan /*
12977531a5a3Sdan ** Set a table filter on a Session Object.
12987531a5a3Sdan */
12997531a5a3Sdan void sqlite3session_table_filter(
13007531a5a3Sdan   sqlite3_session *pSession,
13017531a5a3Sdan   int(*xFilter)(void*, const char*),
13027531a5a3Sdan   void *pCtx                      /* First argument passed to xFilter */
13037531a5a3Sdan ){
13047531a5a3Sdan   pSession->bAutoAttach = 1;
13057531a5a3Sdan   pSession->pFilterCtx = pCtx;
13067531a5a3Sdan   pSession->xTableFilter = xFilter;
13077531a5a3Sdan }
13087531a5a3Sdan 
13097531a5a3Sdan /*
13104fccf43aSdan ** Attach a table to a session. All subsequent changes made to the table
13114fccf43aSdan ** while the session object is enabled will be recorded.
13124fccf43aSdan **
13134fccf43aSdan ** Only tables that have a PRIMARY KEY defined may be attached. It does
13144fccf43aSdan ** not matter if the PRIMARY KEY is an "INTEGER PRIMARY KEY" (rowid alias)
13154fccf43aSdan ** or not.
13164fccf43aSdan */
13174fccf43aSdan int sqlite3session_attach(
13184fccf43aSdan   sqlite3_session *pSession,      /* Session object */
13194fccf43aSdan   const char *zName               /* Table name */
13204fccf43aSdan ){
1321ff4d0f41Sdan   int rc = SQLITE_OK;
1322ff4d0f41Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
1323ff4d0f41Sdan 
1324ff4d0f41Sdan   if( !zName ){
1325ff4d0f41Sdan     pSession->bAutoAttach = 1;
1326ff4d0f41Sdan   }else{
1327296c7658Sdan     SessionTable *pTab;           /* New table object (if required) */
1328296c7658Sdan     int nName;                    /* Number of bytes in string zName */
13294fccf43aSdan 
13304fccf43aSdan     /* First search for an existing entry. If one is found, this call is
13314fccf43aSdan     ** a no-op. Return early. */
1332cfdbde21Sdrh     nName = sqlite3Strlen30(zName);
13334fccf43aSdan     for(pTab=pSession->pTable; pTab; pTab=pTab->pNext){
13344c220252Sdan       if( 0==sqlite3_strnicmp(pTab->zName, zName, nName+1) ) break;
13354fccf43aSdan     }
13364fccf43aSdan 
13374c220252Sdan     if( !pTab ){
13384fccf43aSdan       /* Allocate new SessionTable object. */
13394fccf43aSdan       pTab = (SessionTable *)sqlite3_malloc(sizeof(SessionTable) + nName + 1);
13404c220252Sdan       if( !pTab ){
13414c220252Sdan         rc = SQLITE_NOMEM;
13424c220252Sdan       }else{
13434fccf43aSdan         /* Populate the new SessionTable object and link it into the list. */
13444fccf43aSdan         memset(pTab, 0, sizeof(SessionTable));
13454fccf43aSdan         pTab->zName = (char *)&pTab[1];
13464fccf43aSdan         memcpy(pTab->zName, zName, nName+1);
13474fccf43aSdan         pTab->pNext = pSession->pTable;
13484fccf43aSdan         pSession->pTable = pTab;
13494c220252Sdan       }
13504c220252Sdan     }
1351ff4d0f41Sdan   }
13524fccf43aSdan 
13534c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
13544c220252Sdan   return rc;
13554fccf43aSdan }
13564fccf43aSdan 
1357296c7658Sdan /*
1358296c7658Sdan ** Ensure that there is room in the buffer to append nByte bytes of data.
1359296c7658Sdan ** If not, use sqlite3_realloc() to grow the buffer so that there is.
1360296c7658Sdan **
1361296c7658Sdan ** If successful, return zero. Otherwise, if an OOM condition is encountered,
1362296c7658Sdan ** set *pRc to SQLITE_NOMEM and return non-zero.
1363296c7658Sdan */
13644fccf43aSdan static int sessionBufferGrow(SessionBuffer *p, int nByte, int *pRc){
136580fe2d93Sdan   if( *pRc==SQLITE_OK && p->nAlloc-p->nBuf<nByte ){
13664fccf43aSdan     u8 *aNew;
13674fccf43aSdan     int nNew = p->nAlloc ? p->nAlloc : 128;
13684fccf43aSdan     do {
13694fccf43aSdan       nNew = nNew*2;
1370ef7a6304Sdan     }while( nNew<(p->nBuf+nByte) );
13714fccf43aSdan 
13724fccf43aSdan     aNew = (u8 *)sqlite3_realloc(p->aBuf, nNew);
13734fccf43aSdan     if( 0==aNew ){
13744fccf43aSdan       *pRc = SQLITE_NOMEM;
137580fe2d93Sdan     }else{
13764fccf43aSdan       p->aBuf = aNew;
13774fccf43aSdan       p->nAlloc = nNew;
13784fccf43aSdan     }
137980fe2d93Sdan   }
138080fe2d93Sdan   return (*pRc!=SQLITE_OK);
13814fccf43aSdan }
13824fccf43aSdan 
1383296c7658Sdan /*
1384fa122adaSdan ** Append the value passed as the second argument to the buffer passed
1385fa122adaSdan ** as the first.
1386fa122adaSdan **
1387fa122adaSdan ** This function is a no-op if *pRc is non-zero when it is called.
1388fa122adaSdan ** Otherwise, if an error occurs, *pRc is set to an SQLite error code
1389fa122adaSdan ** before returning.
1390fa122adaSdan */
1391fa122adaSdan static void sessionAppendValue(SessionBuffer *p, sqlite3_value *pVal, int *pRc){
1392fa122adaSdan   int rc = *pRc;
1393fa122adaSdan   if( rc==SQLITE_OK ){
1394fa122adaSdan     int nByte = 0;
1395e8fa8c96Sdan     rc = sessionSerializeValue(0, pVal, &nByte);
1396fa122adaSdan     sessionBufferGrow(p, nByte, &rc);
1397fa122adaSdan     if( rc==SQLITE_OK ){
1398fa122adaSdan       rc = sessionSerializeValue(&p->aBuf[p->nBuf], pVal, 0);
1399fa122adaSdan       p->nBuf += nByte;
1400fa122adaSdan     }else{
1401fa122adaSdan       *pRc = rc;
1402fa122adaSdan     }
1403fa122adaSdan   }
1404fa122adaSdan }
1405fa122adaSdan 
1406fa122adaSdan /*
1407296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1408296c7658Sdan ** called. Otherwise, append a single byte to the buffer.
1409296c7658Sdan **
1410296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1411296c7658Sdan ** returning.
1412296c7658Sdan */
14134fccf43aSdan static void sessionAppendByte(SessionBuffer *p, u8 v, int *pRc){
141480fe2d93Sdan   if( 0==sessionBufferGrow(p, 1, pRc) ){
14154fccf43aSdan     p->aBuf[p->nBuf++] = v;
14164fccf43aSdan   }
14174fccf43aSdan }
14184fccf43aSdan 
1419296c7658Sdan /*
1420296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1421296c7658Sdan ** called. Otherwise, append a single varint to the buffer.
1422296c7658Sdan **
1423296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1424296c7658Sdan ** returning.
1425296c7658Sdan */
1426cfdbde21Sdrh static void sessionAppendVarint(SessionBuffer *p, int v, int *pRc){
142780fe2d93Sdan   if( 0==sessionBufferGrow(p, 9, pRc) ){
14284fccf43aSdan     p->nBuf += sessionVarintPut(&p->aBuf[p->nBuf], v);
14294fccf43aSdan   }
14304fccf43aSdan }
14314fccf43aSdan 
1432296c7658Sdan /*
1433296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1434296c7658Sdan ** called. Otherwise, append a blob of data to the buffer.
1435296c7658Sdan **
1436296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1437296c7658Sdan ** returning.
1438296c7658Sdan */
14394fccf43aSdan static void sessionAppendBlob(
14404fccf43aSdan   SessionBuffer *p,
14414fccf43aSdan   const u8 *aBlob,
14424fccf43aSdan   int nBlob,
14434fccf43aSdan   int *pRc
14444fccf43aSdan ){
144580fe2d93Sdan   if( 0==sessionBufferGrow(p, nBlob, pRc) ){
14464fccf43aSdan     memcpy(&p->aBuf[p->nBuf], aBlob, nBlob);
14474fccf43aSdan     p->nBuf += nBlob;
14484fccf43aSdan   }
14494fccf43aSdan }
14504fccf43aSdan 
1451296c7658Sdan /*
1452296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1453296c7658Sdan ** called. Otherwise, append a string to the buffer. All bytes in the string
1454296c7658Sdan ** up to (but not including) the nul-terminator are written to the buffer.
1455296c7658Sdan **
1456296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1457296c7658Sdan ** returning.
1458296c7658Sdan */
1459d5f0767cSdan static void sessionAppendStr(
1460d5f0767cSdan   SessionBuffer *p,
1461d5f0767cSdan   const char *zStr,
1462d5f0767cSdan   int *pRc
1463d5f0767cSdan ){
1464cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr);
146580fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1466d5f0767cSdan     memcpy(&p->aBuf[p->nBuf], zStr, nStr);
1467d5f0767cSdan     p->nBuf += nStr;
1468d5f0767cSdan   }
1469d5f0767cSdan }
1470d5f0767cSdan 
1471296c7658Sdan /*
1472296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1473296c7658Sdan ** called. Otherwise, append the string representation of integer iVal
1474296c7658Sdan ** to the buffer. No nul-terminator is written.
1475296c7658Sdan **
1476296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1477296c7658Sdan ** returning.
1478296c7658Sdan */
1479d5f0767cSdan static void sessionAppendInteger(
1480296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1481296c7658Sdan   int iVal,                       /* Value to write the string rep. of */
1482296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1483d5f0767cSdan ){
1484d5f0767cSdan   char aBuf[24];
1485d5f0767cSdan   sqlite3_snprintf(sizeof(aBuf)-1, aBuf, "%d", iVal);
1486d5f0767cSdan   sessionAppendStr(p, aBuf, pRc);
1487d5f0767cSdan }
1488d5f0767cSdan 
1489296c7658Sdan /*
1490296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1491296c7658Sdan ** called. Otherwise, append the string zStr enclosed in quotes (") and
1492296c7658Sdan ** with any embedded quote characters escaped to the buffer. No
1493296c7658Sdan ** nul-terminator byte is written.
1494296c7658Sdan **
1495296c7658Sdan ** If an OOM condition is encountered, set *pRc to SQLITE_NOMEM before
1496296c7658Sdan ** returning.
1497296c7658Sdan */
1498d5f0767cSdan static void sessionAppendIdent(
1499296c7658Sdan   SessionBuffer *p,               /* Buffer to a append to */
1500296c7658Sdan   const char *zStr,               /* String to quote, escape and append */
1501296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
1502d5f0767cSdan ){
1503cfdbde21Sdrh   int nStr = sqlite3Strlen30(zStr)*2 + 2 + 1;
150480fe2d93Sdan   if( 0==sessionBufferGrow(p, nStr, pRc) ){
1505d5f0767cSdan     char *zOut = (char *)&p->aBuf[p->nBuf];
1506d5f0767cSdan     const char *zIn = zStr;
1507d5f0767cSdan     *zOut++ = '"';
1508d5f0767cSdan     while( *zIn ){
1509d5f0767cSdan       if( *zIn=='"' ) *zOut++ = '"';
1510d5f0767cSdan       *zOut++ = *(zIn++);
1511d5f0767cSdan     }
1512d5f0767cSdan     *zOut++ = '"';
1513cfdbde21Sdrh     p->nBuf = (int)((u8 *)zOut - p->aBuf);
1514d5f0767cSdan   }
1515d5f0767cSdan }
1516d5f0767cSdan 
1517296c7658Sdan /*
1518296c7658Sdan ** This function is a no-op if *pRc is other than SQLITE_OK when it is
1519296c7658Sdan ** called. Otherwse, it appends the serialized version of the value stored
1520296c7658Sdan ** in column iCol of the row that SQL statement pStmt currently points
1521296c7658Sdan ** to to the buffer.
1522296c7658Sdan */
15234fccf43aSdan static void sessionAppendCol(
1524296c7658Sdan   SessionBuffer *p,               /* Buffer to append to */
1525296c7658Sdan   sqlite3_stmt *pStmt,            /* Handle pointing to row containing value */
1526296c7658Sdan   int iCol,                       /* Column to read value from */
1527296c7658Sdan   int *pRc                        /* IN/OUT: Error code */
15284fccf43aSdan ){
15294fccf43aSdan   if( *pRc==SQLITE_OK ){
15304fccf43aSdan     int eType = sqlite3_column_type(pStmt, iCol);
15314fccf43aSdan     sessionAppendByte(p, (u8)eType, pRc);
15324fccf43aSdan     if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
15334fccf43aSdan       sqlite3_int64 i;
15344fccf43aSdan       u8 aBuf[8];
15354fccf43aSdan       if( eType==SQLITE_INTEGER ){
15364fccf43aSdan         i = sqlite3_column_int64(pStmt, iCol);
15374fccf43aSdan       }else{
15384fccf43aSdan         double r = sqlite3_column_double(pStmt, iCol);
15394fccf43aSdan         memcpy(&i, &r, 8);
15404fccf43aSdan       }
1541296c7658Sdan       sessionPutI64(aBuf, i);
15424fccf43aSdan       sessionAppendBlob(p, aBuf, 8, pRc);
15434fccf43aSdan     }
15444fccf43aSdan     if( eType==SQLITE_BLOB || eType==SQLITE_TEXT ){
15456734007dSdan       u8 *z;
15463cc89d95Sdan       int nByte;
15476734007dSdan       if( eType==SQLITE_BLOB ){
15486734007dSdan         z = (u8 *)sqlite3_column_blob(pStmt, iCol);
15496734007dSdan       }else{
15506734007dSdan         z = (u8 *)sqlite3_column_text(pStmt, iCol);
15516734007dSdan       }
15523cc89d95Sdan       nByte = sqlite3_column_bytes(pStmt, iCol);
15533cc89d95Sdan       if( z || (eType==SQLITE_BLOB && nByte==0) ){
15544fccf43aSdan         sessionAppendVarint(p, nByte, pRc);
15556734007dSdan         sessionAppendBlob(p, z, nByte, pRc);
15566734007dSdan       }else{
15576734007dSdan         *pRc = SQLITE_NOMEM;
15586734007dSdan       }
15594fccf43aSdan     }
15604fccf43aSdan   }
15614fccf43aSdan }
15624fccf43aSdan 
1563296c7658Sdan /*
1564296c7658Sdan **
156580fe2d93Sdan ** This function appends an update change to the buffer (see the comments
156680fe2d93Sdan ** under "CHANGESET FORMAT" at the top of the file). An update change
156780fe2d93Sdan ** consists of:
1568296c7658Sdan **
1569296c7658Sdan **   1 byte:  SQLITE_UPDATE (0x17)
1570296c7658Sdan **   n bytes: old.* record (see RECORD FORMAT)
1571296c7658Sdan **   m bytes: new.* record (see RECORD FORMAT)
1572296c7658Sdan **
1573296c7658Sdan ** The SessionChange object passed as the third argument contains the
1574296c7658Sdan ** values that were stored in the row when the session began (the old.*
1575296c7658Sdan ** values). The statement handle passed as the second argument points
1576296c7658Sdan ** at the current version of the row (the new.* values).
1577296c7658Sdan **
1578296c7658Sdan ** If all of the old.* values are equal to their corresponding new.* value
1579296c7658Sdan ** (i.e. nothing has changed), then no data at all is appended to the buffer.
1580296c7658Sdan **
1581296c7658Sdan ** Otherwise, the old.* record contains all primary key values and the
1582296c7658Sdan ** original values of any fields that have been modified. The new.* record
1583296c7658Sdan ** contains the new values of only those fields that have been modified.
1584296c7658Sdan */
158580fe2d93Sdan static int sessionAppendUpdate(
1586296c7658Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
158773b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
1588296c7658Sdan   sqlite3_stmt *pStmt,            /* Statement handle pointing at new row */
1589296c7658Sdan   SessionChange *p,               /* Object containing old values */
159080fe2d93Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
15914fccf43aSdan ){
159280fe2d93Sdan   int rc = SQLITE_OK;
1593296c7658Sdan   SessionBuffer buf2 = {0,0,0}; /* Buffer to accumulate new.* record in */
1594296c7658Sdan   int bNoop = 1;                /* Set to zero if any values are modified */
15951f34f8ccSdan   int nRewind = pBuf->nBuf;     /* Set to zero if any values are modified */
1596296c7658Sdan   int i;                        /* Used to iterate through columns */
1597296c7658Sdan   u8 *pCsr = p->aRecord;        /* Used to iterate through old.* values */
1598296c7658Sdan 
159980fe2d93Sdan   sessionAppendByte(pBuf, SQLITE_UPDATE, &rc);
160080fe2d93Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
16014fccf43aSdan   for(i=0; i<sqlite3_column_count(pStmt); i++){
160237f133ecSdan     int bChanged = 0;
16034fccf43aSdan     int nAdvance;
16044fccf43aSdan     int eType = *pCsr;
16054fccf43aSdan     switch( eType ){
16064fccf43aSdan       case SQLITE_NULL:
16074fccf43aSdan         nAdvance = 1;
16084fccf43aSdan         if( sqlite3_column_type(pStmt, i)!=SQLITE_NULL ){
160937f133ecSdan           bChanged = 1;
16104fccf43aSdan         }
16114fccf43aSdan         break;
16124fccf43aSdan 
16134fccf43aSdan       case SQLITE_FLOAT:
16144fccf43aSdan       case SQLITE_INTEGER: {
16154fccf43aSdan         nAdvance = 9;
16164fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i) ){
16174fccf43aSdan           sqlite3_int64 iVal = sessionGetI64(&pCsr[1]);
16184fccf43aSdan           if( eType==SQLITE_INTEGER ){
16194fccf43aSdan             if( iVal==sqlite3_column_int64(pStmt, i) ) break;
16204fccf43aSdan           }else{
16214fccf43aSdan             double dVal;
16224fccf43aSdan             memcpy(&dVal, &iVal, 8);
16234fccf43aSdan             if( dVal==sqlite3_column_double(pStmt, i) ) break;
16244fccf43aSdan           }
16254fccf43aSdan         }
162637f133ecSdan         bChanged = 1;
16274fccf43aSdan         break;
16284fccf43aSdan       }
16294fccf43aSdan 
1630e5754eecSdan       default: {
16314fccf43aSdan         int nByte;
16324fccf43aSdan         int nHdr = 1 + sessionVarintGet(&pCsr[1], &nByte);
1633e5754eecSdan         assert( eType==SQLITE_TEXT || eType==SQLITE_BLOB );
16344fccf43aSdan         nAdvance = nHdr + nByte;
16354fccf43aSdan         if( eType==sqlite3_column_type(pStmt, i)
16364fccf43aSdan          && nByte==sqlite3_column_bytes(pStmt, i)
16374fccf43aSdan          && 0==memcmp(&pCsr[nHdr], sqlite3_column_blob(pStmt, i), nByte)
16384fccf43aSdan         ){
16394fccf43aSdan           break;
16404fccf43aSdan         }
164137f133ecSdan         bChanged = 1;
16424fccf43aSdan       }
16434fccf43aSdan     }
16444fccf43aSdan 
164573b3c055Sdan     /* If at least one field has been modified, this is not a no-op. */
164673b3c055Sdan     if( bChanged ) bNoop = 0;
164773b3c055Sdan 
164873b3c055Sdan     /* Add a field to the old.* record. This is omitted if this modules is
164973b3c055Sdan     ** currently generating a patchset. */
165073b3c055Sdan     if( bPatchset==0 ){
165137f133ecSdan       if( bChanged || abPK[i] ){
165280fe2d93Sdan         sessionAppendBlob(pBuf, pCsr, nAdvance, &rc);
16534fccf43aSdan       }else{
165480fe2d93Sdan         sessionAppendByte(pBuf, 0, &rc);
165537f133ecSdan       }
165673b3c055Sdan     }
165737f133ecSdan 
165873b3c055Sdan     /* Add a field to the new.* record. Or the only record if currently
165973b3c055Sdan     ** generating a patchset.  */
166073b3c055Sdan     if( bChanged || (bPatchset && abPK[i]) ){
166180fe2d93Sdan       sessionAppendCol(&buf2, pStmt, i, &rc);
166237f133ecSdan     }else{
166380fe2d93Sdan       sessionAppendByte(&buf2, 0, &rc);
16644fccf43aSdan     }
166537f133ecSdan 
16664fccf43aSdan     pCsr += nAdvance;
16674fccf43aSdan   }
16684fccf43aSdan 
16694fccf43aSdan   if( bNoop ){
16701f34f8ccSdan     pBuf->nBuf = nRewind;
16714fccf43aSdan   }else{
167280fe2d93Sdan     sessionAppendBlob(pBuf, buf2.aBuf, buf2.nBuf, &rc);
16734fccf43aSdan   }
16741f34f8ccSdan   sqlite3_free(buf2.aBuf);
167580fe2d93Sdan 
167680fe2d93Sdan   return rc;
1677d5f0767cSdan }
16784fccf43aSdan 
1679a71d2371Sdan /*
1680a71d2371Sdan ** Append a DELETE change to the buffer passed as the first argument. Use
1681a71d2371Sdan ** the changeset format if argument bPatchset is zero, or the patchset
1682a71d2371Sdan ** format otherwise.
1683a71d2371Sdan */
168473b3c055Sdan static int sessionAppendDelete(
168573b3c055Sdan   SessionBuffer *pBuf,            /* Buffer to append to */
168673b3c055Sdan   int bPatchset,                  /* True for "patchset", 0 for "changeset" */
168773b3c055Sdan   SessionChange *p,               /* Object containing old values */
1688a71d2371Sdan   int nCol,                       /* Number of columns in table */
168973b3c055Sdan   u8 *abPK                        /* Boolean array - true for PK columns */
169073b3c055Sdan ){
169173b3c055Sdan   int rc = SQLITE_OK;
169273b3c055Sdan 
169373b3c055Sdan   sessionAppendByte(pBuf, SQLITE_DELETE, &rc);
169473b3c055Sdan   sessionAppendByte(pBuf, p->bIndirect, &rc);
169573b3c055Sdan 
169673b3c055Sdan   if( bPatchset==0 ){
169773b3c055Sdan     sessionAppendBlob(pBuf, p->aRecord, p->nRecord, &rc);
169873b3c055Sdan   }else{
169973b3c055Sdan     int i;
170073b3c055Sdan     u8 *a = p->aRecord;
170173b3c055Sdan     for(i=0; i<nCol; i++){
170273b3c055Sdan       u8 *pStart = a;
170373b3c055Sdan       int eType = *a++;
170473b3c055Sdan 
170573b3c055Sdan       switch( eType ){
170673b3c055Sdan         case 0:
170773b3c055Sdan         case SQLITE_NULL:
170873b3c055Sdan           assert( abPK[i]==0 );
170973b3c055Sdan           break;
171073b3c055Sdan 
171173b3c055Sdan         case SQLITE_FLOAT:
171273b3c055Sdan         case SQLITE_INTEGER:
171373b3c055Sdan           a += 8;
171473b3c055Sdan           break;
171573b3c055Sdan 
171673b3c055Sdan         default: {
171773b3c055Sdan           int n;
171873b3c055Sdan           a += sessionVarintGet(a, &n);
171973b3c055Sdan           a += n;
172073b3c055Sdan           break;
172173b3c055Sdan         }
172273b3c055Sdan       }
172373b3c055Sdan       if( abPK[i] ){
1724f5ab08c7Sdrh         sessionAppendBlob(pBuf, pStart, (int)(a-pStart), &rc);
172573b3c055Sdan       }
172673b3c055Sdan     }
172773b3c055Sdan     assert( (a - p->aRecord)==p->nRecord );
172873b3c055Sdan   }
172973b3c055Sdan 
173073b3c055Sdan   return rc;
173173b3c055Sdan }
173273b3c055Sdan 
173377fc1d5bSdan /*
173477fc1d5bSdan ** Formulate and prepare a SELECT statement to retrieve a row from table
173577fc1d5bSdan ** zTab in database zDb based on its primary key. i.e.
173677fc1d5bSdan **
173777fc1d5bSdan **   SELECT * FROM zDb.zTab WHERE pk1 = ? AND pk2 = ? AND ...
173877fc1d5bSdan */
1739e8d5648eSdan static int sessionSelectStmt(
1740e8d5648eSdan   sqlite3 *db,                    /* Database handle */
1741d7fb7d24Sdan   const char *zDb,                /* Database name */
1742e8d5648eSdan   const char *zTab,               /* Table name */
174377fc1d5bSdan   int nCol,                       /* Number of columns in table */
174477fc1d5bSdan   const char **azCol,             /* Names of table columns */
174577fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY  array */
174677fc1d5bSdan   sqlite3_stmt **ppStmt           /* OUT: Prepared SELECT statement */
1747d5f0767cSdan ){
1748e8d5648eSdan   int rc = SQLITE_OK;
1749d5f0767cSdan   int i;
1750e8d5648eSdan   const char *zSep = "";
1751e8d5648eSdan   SessionBuffer buf = {0, 0, 0};
1752d5f0767cSdan 
1753e8d5648eSdan   sessionAppendStr(&buf, "SELECT * FROM ", &rc);
1754d7fb7d24Sdan   sessionAppendIdent(&buf, zDb, &rc);
1755d7fb7d24Sdan   sessionAppendStr(&buf, ".", &rc);
1756e8d5648eSdan   sessionAppendIdent(&buf, zTab, &rc);
1757e8d5648eSdan   sessionAppendStr(&buf, " WHERE ", &rc);
1758e8d5648eSdan   for(i=0; i<nCol; i++){
1759e8d5648eSdan     if( abPK[i] ){
1760e8d5648eSdan       sessionAppendStr(&buf, zSep, &rc);
1761e8d5648eSdan       sessionAppendIdent(&buf, azCol[i], &rc);
1762e8d5648eSdan       sessionAppendStr(&buf, " = ?", &rc);
1763e8d5648eSdan       sessionAppendInteger(&buf, i+1, &rc);
1764e8d5648eSdan       zSep = " AND ";
1765d5f0767cSdan     }
1766d5f0767cSdan   }
1767d5f0767cSdan   if( rc==SQLITE_OK ){
1768e8d5648eSdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, ppStmt, 0);
1769d5f0767cSdan   }
1770e8d5648eSdan   sqlite3_free(buf.aBuf);
1771e8d5648eSdan   return rc;
1772d5f0767cSdan }
1773d5f0767cSdan 
177477fc1d5bSdan /*
177577fc1d5bSdan ** Bind the PRIMARY KEY values from the change passed in argument pChange
177677fc1d5bSdan ** to the SELECT statement passed as the first argument. The SELECT statement
177777fc1d5bSdan ** is as prepared by function sessionSelectStmt().
177877fc1d5bSdan **
177977fc1d5bSdan ** Return SQLITE_OK if all PK values are successfully bound, or an SQLite
178077fc1d5bSdan ** error code (e.g. SQLITE_NOMEM) otherwise.
178177fc1d5bSdan */
1782e8d5648eSdan static int sessionSelectBind(
178377fc1d5bSdan   sqlite3_stmt *pSelect,          /* SELECT from sessionSelectStmt() */
178477fc1d5bSdan   int nCol,                       /* Number of columns in table */
178577fc1d5bSdan   u8 *abPK,                       /* PRIMARY KEY array */
178677fc1d5bSdan   SessionChange *pChange          /* Change structure */
1787e8d5648eSdan ){
1788e8d5648eSdan   int i;
1789e8d5648eSdan   int rc = SQLITE_OK;
1790e5754eecSdan   u8 *a = pChange->aRecord;
1791d5f0767cSdan 
1792e8d5648eSdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
1793e8d5648eSdan     int eType = *a++;
1794e8d5648eSdan 
1795e8d5648eSdan     switch( eType ){
179680fe2d93Sdan       case 0:
1797e8d5648eSdan       case SQLITE_NULL:
1798e5754eecSdan         assert( abPK[i]==0 );
1799e8d5648eSdan         break;
1800e8d5648eSdan 
1801e8d5648eSdan       case SQLITE_INTEGER: {
1802e8d5648eSdan         if( abPK[i] ){
1803e8d5648eSdan           i64 iVal = sessionGetI64(a);
1804e8d5648eSdan           rc = sqlite3_bind_int64(pSelect, i+1, iVal);
1805e8d5648eSdan         }
1806e8d5648eSdan         a += 8;
1807e8d5648eSdan         break;
1808d5f0767cSdan       }
1809296c7658Sdan 
1810e8d5648eSdan       case SQLITE_FLOAT: {
1811e8d5648eSdan         if( abPK[i] ){
1812e8d5648eSdan           double rVal;
1813e8d5648eSdan           i64 iVal = sessionGetI64(a);
1814e8d5648eSdan           memcpy(&rVal, &iVal, 8);
18154e895da1Sdan           rc = sqlite3_bind_double(pSelect, i+1, rVal);
1816d5f0767cSdan         }
1817e8d5648eSdan         a += 8;
1818e8d5648eSdan         break;
1819e8d5648eSdan       }
1820e8d5648eSdan 
1821e8d5648eSdan       case SQLITE_TEXT: {
1822e8d5648eSdan         int n;
1823e8d5648eSdan         a += sessionVarintGet(a, &n);
1824e8d5648eSdan         if( abPK[i] ){
1825e8d5648eSdan           rc = sqlite3_bind_text(pSelect, i+1, (char *)a, n, SQLITE_TRANSIENT);
1826e8d5648eSdan         }
1827e8d5648eSdan         a += n;
1828e8d5648eSdan         break;
1829e8d5648eSdan       }
1830e8d5648eSdan 
1831e5754eecSdan       default: {
1832e8d5648eSdan         int n;
1833e5754eecSdan         assert( eType==SQLITE_BLOB );
1834e8d5648eSdan         a += sessionVarintGet(a, &n);
1835e8d5648eSdan         if( abPK[i] ){
1836e8d5648eSdan           rc = sqlite3_bind_blob(pSelect, i+1, a, n, SQLITE_TRANSIENT);
1837e8d5648eSdan         }
1838e8d5648eSdan         a += n;
1839e8d5648eSdan         break;
1840e8d5648eSdan       }
1841e8d5648eSdan     }
1842e8d5648eSdan   }
1843e8d5648eSdan 
1844d5f0767cSdan   return rc;
18454fccf43aSdan }
18464fccf43aSdan 
184777fc1d5bSdan /*
184877fc1d5bSdan ** This function is a no-op if *pRc is set to other than SQLITE_OK when it
184977fc1d5bSdan ** is called. Otherwise, append a serialized table header (part of the binary
185077fc1d5bSdan ** changeset format) to buffer *pBuf. If an error occurs, set *pRc to an
185177fc1d5bSdan ** SQLite error code before returning.
185277fc1d5bSdan */
18535d607a6eSdan static void sessionAppendTableHdr(
1854a71d2371Sdan   SessionBuffer *pBuf,            /* Append header to this buffer */
1855a71d2371Sdan   int bPatchset,                  /* Use the patchset format if true */
1856a71d2371Sdan   SessionTable *pTab,             /* Table object to append header for */
1857a71d2371Sdan   int *pRc                        /* IN/OUT: Error code */
18585d607a6eSdan ){
18595d607a6eSdan   /* Write a table header */
186073b3c055Sdan   sessionAppendByte(pBuf, (bPatchset ? 'P' : 'T'), pRc);
18615d607a6eSdan   sessionAppendVarint(pBuf, pTab->nCol, pRc);
18625d607a6eSdan   sessionAppendBlob(pBuf, pTab->abPK, pTab->nCol, pRc);
18634f528042Sdan   sessionAppendBlob(pBuf, (u8 *)pTab->zName, (int)strlen(pTab->zName)+1, pRc);
18645d607a6eSdan }
18655d607a6eSdan 
1866a71d2371Sdan /*
1867a71d2371Sdan ** Generate either a changeset (if argument bPatchset is zero) or a patchset
1868a71d2371Sdan ** (if it is non-zero) based on the current contents of the session object
1869a71d2371Sdan ** passed as the first argument.
1870a71d2371Sdan **
1871a71d2371Sdan ** If no error occurs, SQLITE_OK is returned and the new changeset/patchset
1872a71d2371Sdan ** stored in output variables *pnChangeset and *ppChangeset. Or, if an error
1873a71d2371Sdan ** occurs, an SQLite error code is returned and both output variables set
1874a71d2371Sdan ** to 0.
1875a71d2371Sdan */
187673b3c055Sdan int sessionGenerateChangeset(
18774fccf43aSdan   sqlite3_session *pSession,      /* Session object */
187873b3c055Sdan   int bPatchset,                  /* True for patchset, false for changeset */
1879ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
1880ef7a6304Sdan   void *pOut,                     /* First argument for xOutput */
18814fccf43aSdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
18824fccf43aSdan   void **ppChangeset              /* OUT: Buffer containing changeset */
18834fccf43aSdan ){
1884296c7658Sdan   sqlite3 *db = pSession->db;     /* Source database handle */
1885296c7658Sdan   SessionTable *pTab;             /* Used to iterate through attached tables */
1886296c7658Sdan   SessionBuffer buf = {0,0,0};    /* Buffer in which to accumlate changeset */
1887296c7658Sdan   int rc;                         /* Return code */
18884fccf43aSdan 
1889ef7a6304Sdan   assert( xOutput==0 || (pnChangeset==0 && ppChangeset==0 ) );
1890ef7a6304Sdan 
1891296c7658Sdan   /* Zero the output variables in case an error occurs. If this session
1892296c7658Sdan   ** object is already in the error state (sqlite3_session.rc != SQLITE_OK),
1893296c7658Sdan   ** this call will be a no-op.  */
1894ef7a6304Sdan   if( xOutput==0 ){
18954fccf43aSdan     *pnChangeset = 0;
18964fccf43aSdan     *ppChangeset = 0;
1897ef7a6304Sdan   }
1898e5754eecSdan 
1899e5754eecSdan   if( pSession->rc ) return pSession->rc;
1900e5754eecSdan   rc = sqlite3_exec(pSession->db, "SAVEPOINT changeset", 0, 0, 0);
1901e5754eecSdan   if( rc!=SQLITE_OK ) return rc;
1902e5754eecSdan 
1903e5754eecSdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
19044fccf43aSdan 
19054fccf43aSdan   for(pTab=pSession->pTable; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
19064fccf43aSdan     if( pTab->nEntry ){
1907d7fb7d24Sdan       const char *zName = pTab->zName;
1908a9605b91Sdan       int nCol;                   /* Number of columns in table */
1909a9605b91Sdan       u8 *abPK;                   /* Primary key array */
1910a9605b91Sdan       const char **azCol = 0;     /* Table columns */
19111f34f8ccSdan       int i;                      /* Used to iterate through hash buckets */
19121f34f8ccSdan       sqlite3_stmt *pSel = 0;     /* SELECT statement to query table pTab */
19131f34f8ccSdan       int nRewind = buf.nBuf;     /* Initial size of write buffer */
19141f34f8ccSdan       int nNoop;                  /* Size of buffer after writing tbl header */
19154fccf43aSdan 
1916a9605b91Sdan       /* Check the table schema is still Ok. */
1917a9605b91Sdan       rc = sessionTableInfo(db, pSession->zDb, zName, &nCol, 0, &azCol, &abPK);
1918a9605b91Sdan       if( !rc && (pTab->nCol!=nCol || memcmp(abPK, pTab->abPK, nCol)) ){
1919a9605b91Sdan         rc = SQLITE_SCHEMA;
1920a9605b91Sdan       }
1921a9605b91Sdan 
19224fccf43aSdan       /* Write a table header */
192373b3c055Sdan       sessionAppendTableHdr(&buf, bPatchset, pTab, &rc);
19244fccf43aSdan 
19254fccf43aSdan       /* Build and compile a statement to execute: */
19264fccf43aSdan       if( rc==SQLITE_OK ){
1927d7fb7d24Sdan         rc = sessionSelectStmt(
1928a9605b91Sdan             db, pSession->zDb, zName, nCol, azCol, abPK, &pSel);
19294fccf43aSdan       }
19304fccf43aSdan 
19311f34f8ccSdan       nNoop = buf.nBuf;
193212ca0b56Sdan       for(i=0; i<pTab->nChange && rc==SQLITE_OK; i++){
1933e8d5648eSdan         SessionChange *p;         /* Used to iterate through changes */
1934e8d5648eSdan 
19354fccf43aSdan         for(p=pTab->apChange[i]; rc==SQLITE_OK && p; p=p->pNext){
1936e5754eecSdan           rc = sessionSelectBind(pSel, nCol, abPK, p);
193780fe2d93Sdan           if( rc!=SQLITE_OK ) continue;
19381f34f8ccSdan           if( sqlite3_step(pSel)==SQLITE_ROW ){
1939798693b2Sdan             if( p->op==SQLITE_INSERT ){
19404fccf43aSdan               int iCol;
19414fccf43aSdan               sessionAppendByte(&buf, SQLITE_INSERT, &rc);
1942b4480e94Sdan               sessionAppendByte(&buf, p->bIndirect, &rc);
1943e8d5648eSdan               for(iCol=0; iCol<nCol; iCol++){
19441f34f8ccSdan                 sessionAppendCol(&buf, pSel, iCol, &rc);
19454fccf43aSdan               }
1946e8d5648eSdan             }else{
194773b3c055Sdan               rc = sessionAppendUpdate(&buf, bPatchset, pSel, p, abPK);
19484fccf43aSdan             }
1949798693b2Sdan           }else if( p->op!=SQLITE_INSERT ){
1950a71d2371Sdan             rc = sessionAppendDelete(&buf, bPatchset, p, nCol, abPK);
19514fccf43aSdan           }
195212ca0b56Sdan           if( rc==SQLITE_OK ){
19531f34f8ccSdan             rc = sqlite3_reset(pSel);
19544fccf43aSdan           }
1955ef7a6304Sdan 
1956f1a08ad8Sdrh           /* If the buffer is now larger than SESSIONS_STRM_CHUNK_SIZE, pass
1957ef7a6304Sdan           ** its contents to the xOutput() callback. */
1958ef7a6304Sdan           if( xOutput
1959ef7a6304Sdan            && rc==SQLITE_OK
1960ef7a6304Sdan            && buf.nBuf>nNoop
1961f1a08ad8Sdrh            && buf.nBuf>SESSIONS_STRM_CHUNK_SIZE
1962ef7a6304Sdan           ){
1963ef7a6304Sdan             rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
1964ef7a6304Sdan             nNoop = -1;
1965ef7a6304Sdan             buf.nBuf = 0;
1966ef7a6304Sdan           }
1967ef7a6304Sdan 
19684fccf43aSdan         }
1969e8d5648eSdan       }
19704fccf43aSdan 
19711f34f8ccSdan       sqlite3_finalize(pSel);
19721f34f8ccSdan       if( buf.nBuf==nNoop ){
19734fccf43aSdan         buf.nBuf = nRewind;
19744fccf43aSdan       }
1975cfdbde21Sdrh       sqlite3_free((char*)azCol);  /* cast works around VC++ bug */
19764fccf43aSdan     }
19774fccf43aSdan   }
19784fccf43aSdan 
19794fccf43aSdan   if( rc==SQLITE_OK ){
1980ef7a6304Sdan     if( xOutput==0 ){
19814fccf43aSdan       *pnChangeset = buf.nBuf;
19824fccf43aSdan       *ppChangeset = buf.aBuf;
1983ef7a6304Sdan       buf.aBuf = 0;
1984ef7a6304Sdan     }else if( buf.nBuf>0 ){
1985ef7a6304Sdan       rc = xOutput(pOut, (void*)buf.aBuf, buf.nBuf);
1986ef7a6304Sdan     }
19874fccf43aSdan   }
19884c220252Sdan 
1989ef7a6304Sdan   sqlite3_free(buf.aBuf);
1990e5754eecSdan   sqlite3_exec(db, "RELEASE changeset", 0, 0, 0);
19914c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
19924fccf43aSdan   return rc;
19934fccf43aSdan }
19944fccf43aSdan 
1995296c7658Sdan /*
199673b3c055Sdan ** Obtain a changeset object containing all changes recorded by the
199773b3c055Sdan ** session object passed as the first argument.
199873b3c055Sdan **
199973b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
200073b3c055Sdan ** using sqlite3_free().
200173b3c055Sdan */
200273b3c055Sdan int sqlite3session_changeset(
200373b3c055Sdan   sqlite3_session *pSession,      /* Session object */
200473b3c055Sdan   int *pnChangeset,               /* OUT: Size of buffer at *ppChangeset */
200573b3c055Sdan   void **ppChangeset              /* OUT: Buffer containing changeset */
200673b3c055Sdan ){
2007ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, 0, 0, pnChangeset, ppChangeset);
2008ef7a6304Sdan }
2009ef7a6304Sdan 
2010ef7a6304Sdan /*
2011ef7a6304Sdan ** Streaming version of sqlite3session_changeset().
2012ef7a6304Sdan */
2013f1a08ad8Sdrh int sqlite3session_changeset_strm(
2014ef7a6304Sdan   sqlite3_session *pSession,
2015ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2016ef7a6304Sdan   void *pOut
2017ef7a6304Sdan ){
2018ef7a6304Sdan   return sessionGenerateChangeset(pSession, 0, xOutput, pOut, 0, 0);
2019ef7a6304Sdan }
2020ef7a6304Sdan 
2021ef7a6304Sdan /*
2022ef7a6304Sdan ** Streaming version of sqlite3session_patchset().
2023ef7a6304Sdan */
2024f1a08ad8Sdrh int sqlite3session_patchset_strm(
2025ef7a6304Sdan   sqlite3_session *pSession,
2026ef7a6304Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2027ef7a6304Sdan   void *pOut
2028ef7a6304Sdan ){
2029ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, xOutput, pOut, 0, 0);
203073b3c055Sdan }
203173b3c055Sdan 
203273b3c055Sdan /*
203373b3c055Sdan ** Obtain a patchset object containing all changes recorded by the
203473b3c055Sdan ** session object passed as the first argument.
203573b3c055Sdan **
203673b3c055Sdan ** It is the responsibility of the caller to eventually free the buffer
203773b3c055Sdan ** using sqlite3_free().
203873b3c055Sdan */
203973b3c055Sdan int sqlite3session_patchset(
204073b3c055Sdan   sqlite3_session *pSession,      /* Session object */
204173b3c055Sdan   int *pnPatchset,                /* OUT: Size of buffer at *ppChangeset */
204273b3c055Sdan   void **ppPatchset               /* OUT: Buffer containing changeset */
204373b3c055Sdan ){
2044ef7a6304Sdan   return sessionGenerateChangeset(pSession, 1, 0, 0, pnPatchset, ppPatchset);
204573b3c055Sdan }
204673b3c055Sdan 
204773b3c055Sdan /*
2048296c7658Sdan ** Enable or disable the session object passed as the first argument.
2049296c7658Sdan */
20504fccf43aSdan int sqlite3session_enable(sqlite3_session *pSession, int bEnable){
20514c220252Sdan   int ret;
20524c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2053296c7658Sdan   if( bEnable>=0 ){
2054296c7658Sdan     pSession->bEnable = bEnable;
20554fccf43aSdan   }
20564c220252Sdan   ret = pSession->bEnable;
20574c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
20584c220252Sdan   return ret;
2059296c7658Sdan }
20604fccf43aSdan 
20614fccf43aSdan /*
2062b4480e94Sdan ** Enable or disable the session object passed as the first argument.
2063b4480e94Sdan */
2064b4480e94Sdan int sqlite3session_indirect(sqlite3_session *pSession, int bIndirect){
2065b4480e94Sdan   int ret;
2066b4480e94Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2067b4480e94Sdan   if( bIndirect>=0 ){
2068b4480e94Sdan     pSession->bIndirect = bIndirect;
2069b4480e94Sdan   }
2070b4480e94Sdan   ret = pSession->bIndirect;
2071b4480e94Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2072b4480e94Sdan   return ret;
2073b4480e94Sdan }
2074b4480e94Sdan 
2075b4480e94Sdan /*
2076b69ec348Sdan ** Return true if there have been no changes to monitored tables recorded
2077b69ec348Sdan ** by the session object passed as the only argument.
2078b69ec348Sdan */
2079b69ec348Sdan int sqlite3session_isempty(sqlite3_session *pSession){
2080b69ec348Sdan   int ret = 0;
2081b69ec348Sdan   SessionTable *pTab;
2082b69ec348Sdan 
2083b69ec348Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(pSession->db));
2084b69ec348Sdan   for(pTab=pSession->pTable; pTab && ret==0; pTab=pTab->pNext){
2085b69ec348Sdan     ret = (pTab->nEntry>0);
2086b69ec348Sdan   }
2087b69ec348Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(pSession->db));
2088b69ec348Sdan 
2089ff530326Sdan   return (ret==0);
2090b69ec348Sdan }
2091b69ec348Sdan 
2092b69ec348Sdan /*
2093f1a08ad8Sdrh ** Do the work for either sqlite3changeset_start() or start_strm().
20944fccf43aSdan */
2095ef7a6304Sdan int sessionChangesetStart(
2096296c7658Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2097ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2098ef7a6304Sdan   void *pIn,
2099296c7658Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2100296c7658Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
21014fccf43aSdan ){
21024fccf43aSdan   sqlite3_changeset_iter *pRet;   /* Iterator to return */
21034fccf43aSdan   int nByte;                      /* Number of bytes to allocate for iterator */
21044fccf43aSdan 
2105ef7a6304Sdan   assert( xInput==0 || (pChangeset==0 && nChangeset==0) );
2106ef7a6304Sdan 
2107296c7658Sdan   /* Zero the output variable in case an error occurs. */
2108296c7658Sdan   *pp = 0;
21094fccf43aSdan 
2110296c7658Sdan   /* Allocate and initialize the iterator structure. */
21114fccf43aSdan   nByte = sizeof(sqlite3_changeset_iter);
21124fccf43aSdan   pRet = (sqlite3_changeset_iter *)sqlite3_malloc(nByte);
21134fccf43aSdan   if( !pRet ) return SQLITE_NOMEM;
21144fccf43aSdan   memset(pRet, 0, sizeof(sqlite3_changeset_iter));
21154757c658Sdan   pRet->in.aData = (u8 *)pChangeset;
21164757c658Sdan   pRet->in.nData = nChangeset;
2117ef7a6304Sdan   pRet->in.xInput = xInput;
2118ef7a6304Sdan   pRet->in.pIn = pIn;
2119ef7a6304Sdan   pRet->in.iNext = 0;
2120ef7a6304Sdan   pRet->in.bEof = (xInput ? 0 : 1);
21214fccf43aSdan 
2122296c7658Sdan   /* Populate the output variable and return success. */
2123296c7658Sdan   *pp = pRet;
21244fccf43aSdan   return SQLITE_OK;
21254fccf43aSdan }
21264fccf43aSdan 
2127296c7658Sdan /*
2128ef7a6304Sdan ** Create an iterator used to iterate through the contents of a changeset.
2129ef7a6304Sdan */
2130ef7a6304Sdan int sqlite3changeset_start(
2131ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2132ef7a6304Sdan   int nChangeset,                 /* Size of buffer pChangeset in bytes */
2133ef7a6304Sdan   void *pChangeset                /* Pointer to buffer containing changeset */
2134ef7a6304Sdan ){
2135ef7a6304Sdan   return sessionChangesetStart(pp, 0, 0, nChangeset, pChangeset);
2136ef7a6304Sdan }
2137ef7a6304Sdan 
2138ef7a6304Sdan /*
2139ef7a6304Sdan ** Streaming version of sqlite3changeset_start().
2140ef7a6304Sdan */
2141f1a08ad8Sdrh int sqlite3changeset_start_strm(
2142ef7a6304Sdan   sqlite3_changeset_iter **pp,    /* OUT: Changeset iterator handle */
2143ef7a6304Sdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2144ef7a6304Sdan   void *pIn
2145ef7a6304Sdan ){
2146ef7a6304Sdan   return sessionChangesetStart(pp, xInput, pIn, 0, 0);
2147ef7a6304Sdan }
2148ef7a6304Sdan 
2149ef7a6304Sdan /*
2150ef7a6304Sdan ** Ensure that there are at least nByte bytes available in the buffer. Or,
2151ef7a6304Sdan ** if there are not nByte bytes remaining in the input, that all available
2152ef7a6304Sdan ** data is in the buffer.
2153ef7a6304Sdan **
2154ef7a6304Sdan ** Return an SQLite error code if an error occurs, or SQLITE_OK otherwise.
2155ef7a6304Sdan */
21564757c658Sdan static int sessionInputBuffer(SessionInput *pIn, int nByte){
2157ef7a6304Sdan   int rc = SQLITE_OK;
21584757c658Sdan   if( pIn->xInput ){
21594757c658Sdan     while( !pIn->bEof && (pIn->iNext+nByte)>=pIn->nData && rc==SQLITE_OK ){
2160f1a08ad8Sdrh       int nNew = SESSIONS_STRM_CHUNK_SIZE;
21614757c658Sdan 
2162f1a08ad8Sdrh       if( pIn->iNext>=SESSIONS_STRM_CHUNK_SIZE ){
21634757c658Sdan         int nMove = pIn->buf.nBuf - pIn->iNext;
21644757c658Sdan         memmove(pIn->buf.aBuf, &pIn->buf.aBuf[pIn->iNext], nMove);
21654757c658Sdan         pIn->buf.nBuf -= pIn->iNext;
21664757c658Sdan         pIn->iNext = 0;
21674757c658Sdan       }
21684757c658Sdan 
21694757c658Sdan       if( SQLITE_OK==sessionBufferGrow(&pIn->buf, nNew, &rc) ){
21704757c658Sdan         rc = pIn->xInput(pIn->pIn, &pIn->buf.aBuf[pIn->buf.nBuf], &nNew);
21714757c658Sdan         if( nNew==0 ){
21724757c658Sdan           pIn->bEof = 1;
21734757c658Sdan         }else{
21744757c658Sdan           pIn->buf.nBuf += nNew;
21754757c658Sdan         }
21764757c658Sdan       }
21774757c658Sdan 
21784757c658Sdan       pIn->aData = pIn->buf.aBuf;
21794757c658Sdan       pIn->nData = pIn->buf.nBuf;
21804757c658Sdan     }
2181ef7a6304Sdan   }
2182ef7a6304Sdan   return rc;
2183ef7a6304Sdan }
2184ef7a6304Sdan 
2185ef7a6304Sdan /*
2186ef7a6304Sdan ** When this function is called, *ppRec points to the start of a record
2187ef7a6304Sdan ** that contains nCol values. This function advances the pointer *ppRec
2188ef7a6304Sdan ** until it points to the byte immediately following that record.
2189ef7a6304Sdan */
2190ef7a6304Sdan static void sessionSkipRecord(
2191ef7a6304Sdan   u8 **ppRec,                     /* IN/OUT: Record pointer */
2192ef7a6304Sdan   int nCol                        /* Number of values in record */
2193ef7a6304Sdan ){
2194ef7a6304Sdan   u8 *aRec = *ppRec;
2195ef7a6304Sdan   int i;
2196ef7a6304Sdan   for(i=0; i<nCol; i++){
2197ef7a6304Sdan     int eType = *aRec++;
2198ef7a6304Sdan     if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2199ef7a6304Sdan       int nByte;
2200ef7a6304Sdan       aRec += sessionVarintGet((u8*)aRec, &nByte);
2201ef7a6304Sdan       aRec += nByte;
2202ef7a6304Sdan     }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2203ef7a6304Sdan       aRec += 8;
2204ef7a6304Sdan     }
2205ef7a6304Sdan   }
2206ef7a6304Sdan 
2207ef7a6304Sdan   *ppRec = aRec;
2208ef7a6304Sdan }
2209ef7a6304Sdan 
2210ef7a6304Sdan /*
22114757c658Sdan ** This function sets the value of the sqlite3_value object passed as the
22124757c658Sdan ** first argument to a copy of the string or blob held in the aData[]
22134757c658Sdan ** buffer. SQLITE_OK is returned if successful, or SQLITE_NOMEM if an OOM
22144757c658Sdan ** error occurs.
22154757c658Sdan */
22164757c658Sdan static int sessionValueSetStr(
22174757c658Sdan   sqlite3_value *pVal,            /* Set the value of this object */
22184757c658Sdan   u8 *aData,                      /* Buffer containing string or blob data */
22194757c658Sdan   int nData,                      /* Size of buffer aData[] in bytes */
22204757c658Sdan   u8 enc                          /* String encoding (0 for blobs) */
22214757c658Sdan ){
222216228167Sdan   /* In theory this code could just pass SQLITE_TRANSIENT as the final
222316228167Sdan   ** argument to sqlite3ValueSetStr() and have the copy created
222416228167Sdan   ** automatically. But doing so makes it difficult to detect any OOM
222516228167Sdan   ** error. Hence the code to create the copy externally. */
22263cc89d95Sdan   u8 *aCopy = sqlite3_malloc(nData+1);
22274757c658Sdan   if( aCopy==0 ) return SQLITE_NOMEM;
22284757c658Sdan   memcpy(aCopy, aData, nData);
22294757c658Sdan   sqlite3ValueSetStr(pVal, nData, (char*)aCopy, enc, sqlite3_free);
22304757c658Sdan   return SQLITE_OK;
22314757c658Sdan }
22324757c658Sdan 
22334757c658Sdan /*
2234296c7658Sdan ** Deserialize a single record from a buffer in memory. See "RECORD FORMAT"
2235296c7658Sdan ** for details.
2236296c7658Sdan **
2237296c7658Sdan ** When this function is called, *paChange points to the start of the record
2238296c7658Sdan ** to deserialize. Assuming no error occurs, *paChange is set to point to
2239296c7658Sdan ** one byte after the end of the same record before this function returns.
2240a71d2371Sdan ** If the argument abPK is NULL, then the record contains nCol values. Or,
2241a71d2371Sdan ** if abPK is other than NULL, then the record contains only the PK fields
2242a71d2371Sdan ** (in other words, it is a patchset DELETE record).
2243296c7658Sdan **
2244296c7658Sdan ** If successful, each element of the apOut[] array (allocated by the caller)
2245296c7658Sdan ** is set to point to an sqlite3_value object containing the value read
2246296c7658Sdan ** from the corresponding position in the record. If that value is not
2247296c7658Sdan ** included in the record (i.e. because the record is part of an UPDATE change
2248296c7658Sdan ** and the field was not modified), the corresponding element of apOut[] is
2249296c7658Sdan ** set to NULL.
2250296c7658Sdan **
2251296c7658Sdan ** It is the responsibility of the caller to free all sqlite_value structures
2252296c7658Sdan ** using sqlite3_free().
2253296c7658Sdan **
2254296c7658Sdan ** If an error occurs, an SQLite error code (e.g. SQLITE_NOMEM) is returned.
2255296c7658Sdan ** The apOut[] array may have been partially populated in this case.
2256296c7658Sdan */
22574fccf43aSdan static int sessionReadRecord(
2258ef7a6304Sdan   SessionInput *pIn,              /* Input data */
22594fccf43aSdan   int nCol,                       /* Number of values in record */
226073b3c055Sdan   u8 *abPK,                       /* Array of primary key flags, or NULL */
22614fccf43aSdan   sqlite3_value **apOut           /* Write values to this array */
22624fccf43aSdan ){
2263296c7658Sdan   int i;                          /* Used to iterate through columns */
2264ef7a6304Sdan   int rc = SQLITE_OK;
22654fccf43aSdan 
2266ef7a6304Sdan   for(i=0; i<nCol && rc==SQLITE_OK; i++){
2267ef7a6304Sdan     int eType = 0;                /* Type of value (SQLITE_NULL, TEXT etc.) */
226873b3c055Sdan     if( abPK && abPK[i]==0 ) continue;
2269ef7a6304Sdan     rc = sessionInputBuffer(pIn, 9);
2270ef7a6304Sdan     if( rc==SQLITE_OK ){
22714757c658Sdan       eType = pIn->aData[pIn->iNext++];
2272ef7a6304Sdan     }
2273ef7a6304Sdan 
2274e8fa8c96Sdan     assert( apOut[i]==0 );
22754fccf43aSdan     if( eType ){
22764fccf43aSdan       apOut[i] = sqlite3ValueNew(0);
2277ef7a6304Sdan       if( !apOut[i] ) rc = SQLITE_NOMEM;
2278ef7a6304Sdan     }
22794fccf43aSdan 
2280ef7a6304Sdan     if( rc==SQLITE_OK ){
22814757c658Sdan       u8 *aVal = &pIn->aData[pIn->iNext];
22824fccf43aSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
22834fccf43aSdan         int nByte;
2284ef7a6304Sdan         pIn->iNext += sessionVarintGet(aVal, &nByte);
2285ef7a6304Sdan         rc = sessionInputBuffer(pIn, nByte);
2286e8fa8c96Sdan         if( rc==SQLITE_OK ){
22876734007dSdan           u8 enc = (eType==SQLITE_TEXT ? SQLITE_UTF8 : 0);
22884757c658Sdan           rc = sessionValueSetStr(apOut[i],&pIn->aData[pIn->iNext],nByte,enc);
228991ddd559Sdan         }
2290ef7a6304Sdan         pIn->iNext += nByte;
22914fccf43aSdan       }
22924fccf43aSdan       if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2293ef7a6304Sdan         sqlite3_int64 v = sessionGetI64(aVal);
22944fccf43aSdan         if( eType==SQLITE_INTEGER ){
22954fccf43aSdan           sqlite3VdbeMemSetInt64(apOut[i], v);
22964fccf43aSdan         }else{
22974fccf43aSdan           double d;
22984e895da1Sdan           memcpy(&d, &v, 8);
22994fccf43aSdan           sqlite3VdbeMemSetDouble(apOut[i], d);
23004fccf43aSdan         }
2301ef7a6304Sdan         pIn->iNext += 8;
230291ddd559Sdan       }
23034fccf43aSdan     }
23044fccf43aSdan   }
23054fccf43aSdan 
2306ef7a6304Sdan   return rc;
2307ef7a6304Sdan }
2308ef7a6304Sdan 
2309ef7a6304Sdan /*
2310ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2311ef7a6304Sdan ** Specifically, to the following:
2312ef7a6304Sdan **
2313ef7a6304Sdan **   + number of columns in table (varint)
2314ef7a6304Sdan **   + array of PK flags (1 byte per column),
2315ef7a6304Sdan **   + table name (nul terminated).
2316ef7a6304Sdan **
2317ef7a6304Sdan ** This function ensures that all of the above is present in the input
2318ef7a6304Sdan ** buffer (i.e. that it can be accessed without any calls to xInput()).
2319ef7a6304Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code.
2320ef7a6304Sdan ** The input pointer is not moved.
2321ef7a6304Sdan */
2322ef7a6304Sdan static int sessionChangesetBufferTblhdr(SessionInput *pIn, int *pnByte){
2323ef7a6304Sdan   int rc = SQLITE_OK;
2324ef7a6304Sdan   int nCol = 0;
23254757c658Sdan   int nRead = 0;
2326ef7a6304Sdan 
2327ef7a6304Sdan   rc = sessionInputBuffer(pIn, 9);
2328ef7a6304Sdan   if( rc==SQLITE_OK ){
23294757c658Sdan     nRead += sessionVarintGet(&pIn->aData[pIn->iNext + nRead], &nCol);
23304757c658Sdan     rc = sessionInputBuffer(pIn, nRead+nCol+100);
23314757c658Sdan     nRead += nCol;
2332ef7a6304Sdan   }
23334757c658Sdan 
2334ef7a6304Sdan   while( rc==SQLITE_OK ){
23354757c658Sdan     while( (pIn->iNext + nRead)<pIn->nData && pIn->aData[pIn->iNext + nRead] ){
23364757c658Sdan       nRead++;
2337ef7a6304Sdan     }
2338e8fa8c96Sdan     if( (pIn->iNext + nRead)<pIn->nData ) break;
23394757c658Sdan     rc = sessionInputBuffer(pIn, nRead + 100);
23404757c658Sdan   }
2341e8fa8c96Sdan   *pnByte = nRead+1;
2342ef7a6304Sdan   return rc;
2343ef7a6304Sdan }
2344ef7a6304Sdan 
2345ef7a6304Sdan /*
2346fa122adaSdan ** The input pointer currently points to the first byte of the first field
2347fa122adaSdan ** of a record consisting of nCol columns. This function ensures the entire
234816228167Sdan ** record is buffered. It does not move the input pointer.
234916228167Sdan **
235016228167Sdan ** If successful, SQLITE_OK is returned and *pnByte is set to the size of
235116228167Sdan ** the record in bytes. Otherwise, an SQLite error code is returned. The
235216228167Sdan ** final value of *pnByte is undefined in this case.
2353fa122adaSdan */
2354fa122adaSdan static int sessionChangesetBufferRecord(
235516228167Sdan   SessionInput *pIn,              /* Input data */
235616228167Sdan   int nCol,                       /* Number of columns in record */
235716228167Sdan   int *pnByte                     /* OUT: Size of record in bytes */
2358fa122adaSdan ){
2359fa122adaSdan   int rc = SQLITE_OK;
2360fa122adaSdan   int nByte = 0;
2361fa122adaSdan   int i;
2362fa122adaSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
2363fa122adaSdan     int eType;
2364fa122adaSdan     rc = sessionInputBuffer(pIn, nByte + 10);
2365fa122adaSdan     if( rc==SQLITE_OK ){
2366fa122adaSdan       eType = pIn->aData[pIn->iNext + nByte++];
2367fa122adaSdan       if( eType==SQLITE_TEXT || eType==SQLITE_BLOB ){
2368fa122adaSdan         int n;
2369fa122adaSdan         nByte += sessionVarintGet(&pIn->aData[pIn->iNext+nByte], &n);
2370fa122adaSdan         nByte += n;
2371fa122adaSdan         rc = sessionInputBuffer(pIn, nByte);
2372fa122adaSdan       }else if( eType==SQLITE_INTEGER || eType==SQLITE_FLOAT ){
2373fa122adaSdan         nByte += 8;
2374fa122adaSdan       }
2375fa122adaSdan     }
2376fa122adaSdan   }
2377fa122adaSdan   *pnByte = nByte;
2378fa122adaSdan   return rc;
2379fa122adaSdan }
2380fa122adaSdan 
2381fa122adaSdan /*
2382ef7a6304Sdan ** The input pointer currently points to the second byte of a table-header.
2383ef7a6304Sdan ** Specifically, to the following:
2384ef7a6304Sdan **
2385ef7a6304Sdan **   + number of columns in table (varint)
2386ef7a6304Sdan **   + array of PK flags (1 byte per column),
2387ef7a6304Sdan **   + table name (nul terminated).
238816228167Sdan **
238916228167Sdan ** This function decodes the table-header and populates the p->nCol,
239016228167Sdan ** p->zTab and p->abPK[] variables accordingly. The p->apValue[] array is
239116228167Sdan ** also allocated or resized according to the new value of p->nCol. The
239216228167Sdan ** input pointer is left pointing to the byte following the table header.
239316228167Sdan **
239416228167Sdan ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error code
239516228167Sdan ** is returned and the final values of the various fields enumerated above
239616228167Sdan ** are undefined.
2397ef7a6304Sdan */
2398ef7a6304Sdan static int sessionChangesetReadTblhdr(sqlite3_changeset_iter *p){
2399ef7a6304Sdan   int rc;
2400ef7a6304Sdan   int nCopy;
2401ef7a6304Sdan   assert( p->rc==SQLITE_OK );
2402ef7a6304Sdan 
2403ef7a6304Sdan   rc = sessionChangesetBufferTblhdr(&p->in, &nCopy);
2404ef7a6304Sdan   if( rc==SQLITE_OK ){
2405ef7a6304Sdan     int nByte;
2406ef7a6304Sdan     int nVarint;
24074757c658Sdan     nVarint = sessionVarintGet(&p->in.aData[p->in.iNext], &p->nCol);
2408ef7a6304Sdan     nCopy -= nVarint;
2409ef7a6304Sdan     p->in.iNext += nVarint;
2410ef7a6304Sdan     nByte = p->nCol * sizeof(sqlite3_value*) * 2 + nCopy;
2411ef7a6304Sdan     p->tblhdr.nBuf = 0;
2412ef7a6304Sdan     sessionBufferGrow(&p->tblhdr, nByte, &rc);
2413ef7a6304Sdan   }
2414ef7a6304Sdan 
2415ef7a6304Sdan   if( rc==SQLITE_OK ){
2416ef7a6304Sdan     int iPK = sizeof(sqlite3_value*)*p->nCol*2;
2417ef7a6304Sdan     memset(p->tblhdr.aBuf, 0, iPK);
24184757c658Sdan     memcpy(&p->tblhdr.aBuf[iPK], &p->in.aData[p->in.iNext], nCopy);
2419ef7a6304Sdan     p->in.iNext += nCopy;
2420ef7a6304Sdan   }
2421ef7a6304Sdan 
2422ef7a6304Sdan   p->apValue = (sqlite3_value**)p->tblhdr.aBuf;
2423ef7a6304Sdan   p->abPK = (u8*)&p->apValue[p->nCol*2];
2424ef7a6304Sdan   p->zTab = (char*)&p->abPK[p->nCol];
2425ef7a6304Sdan   return (p->rc = rc);
24264fccf43aSdan }
24274fccf43aSdan 
242877fc1d5bSdan /*
242977fc1d5bSdan ** Advance the changeset iterator to the next change.
243077fc1d5bSdan **
243177fc1d5bSdan ** If both paRec and pnRec are NULL, then this function works like the public
243277fc1d5bSdan ** API sqlite3changeset_next(). If SQLITE_ROW is returned, then the
243377fc1d5bSdan ** sqlite3changeset_new() and old() APIs may be used to query for values.
243477fc1d5bSdan **
243577fc1d5bSdan ** Otherwise, if paRec and pnRec are not NULL, then a pointer to the change
243677fc1d5bSdan ** record is written to *paRec before returning and the number of bytes in
243777fc1d5bSdan ** the record to *pnRec.
243877fc1d5bSdan **
243977fc1d5bSdan ** Either way, this function returns SQLITE_ROW if the iterator is
244077fc1d5bSdan ** successfully advanced to the next change in the changeset, an SQLite
244177fc1d5bSdan ** error code if an error occurs, or SQLITE_DONE if there are no further
244277fc1d5bSdan ** changes in the changeset.
244377fc1d5bSdan */
24445d607a6eSdan static int sessionChangesetNext(
244577fc1d5bSdan   sqlite3_changeset_iter *p,      /* Changeset iterator */
244677fc1d5bSdan   u8 **paRec,                     /* If non-NULL, store record pointer here */
244777fc1d5bSdan   int *pnRec                      /* If non-NULL, store size of record here */
24485d607a6eSdan ){
24494fccf43aSdan   int i;
2450ef7a6304Sdan   u8 op;
24514fccf43aSdan 
24525d607a6eSdan   assert( (paRec==0 && pnRec==0) || (paRec && pnRec) );
24535d607a6eSdan 
2454296c7658Sdan   /* If the iterator is in the error-state, return immediately. */
24554fccf43aSdan   if( p->rc!=SQLITE_OK ) return p->rc;
24564fccf43aSdan 
24575d607a6eSdan   /* Free the current contents of p->apValue[], if any. */
24584fccf43aSdan   if( p->apValue ){
24594fccf43aSdan     for(i=0; i<p->nCol*2; i++){
24604fccf43aSdan       sqlite3ValueFree(p->apValue[i]);
24614fccf43aSdan     }
24624fccf43aSdan     memset(p->apValue, 0, sizeof(sqlite3_value*)*p->nCol*2);
24634fccf43aSdan   }
24644fccf43aSdan 
2465ef7a6304Sdan   /* Make sure the buffer contains at least 10 bytes of input data, or all
2466ef7a6304Sdan   ** remaining data if there are less than 10 bytes available. This is
2467ef7a6304Sdan   ** sufficient either for the 'T' or 'P' byte and the varint that follows
2468ef7a6304Sdan   ** it, or for the two single byte values otherwise. */
2469ef7a6304Sdan   p->rc = sessionInputBuffer(&p->in, 2);
2470ef7a6304Sdan   if( p->rc!=SQLITE_OK ) return p->rc;
2471ef7a6304Sdan 
24724fccf43aSdan   /* If the iterator is already at the end of the changeset, return DONE. */
24734757c658Sdan   if( p->in.iNext>=p->in.nData ){
24744fccf43aSdan     return SQLITE_DONE;
24754fccf43aSdan   }
24764fccf43aSdan 
24774757c658Sdan   op = p->in.aData[p->in.iNext++];
2478ef7a6304Sdan   if( op=='T' || op=='P' ){
2479ef7a6304Sdan     p->bPatchset = (op=='P');
2480ef7a6304Sdan     if( sessionChangesetReadTblhdr(p) ) return p->rc;
2481ef7a6304Sdan     if( (p->rc = sessionInputBuffer(&p->in, 2)) ) return p->rc;
24824757c658Sdan     op = p->in.aData[p->in.iNext++];
24835d607a6eSdan   }
24845d607a6eSdan 
2485ef7a6304Sdan   p->op = op;
24864757c658Sdan   p->bIndirect = p->in.aData[p->in.iNext++];
24874fccf43aSdan   if( p->op!=SQLITE_UPDATE && p->op!=SQLITE_DELETE && p->op!=SQLITE_INSERT ){
24884757c658Sdan     return (p->rc = SQLITE_CORRUPT_BKPT);
24894fccf43aSdan   }
24904fccf43aSdan 
2491cbf6d2d2Sdan   if( paRec ){
2492cbf6d2d2Sdan     int nVal;                     /* Number of values to buffer */
2493cbf6d2d2Sdan     if( p->bPatchset==0 && op==SQLITE_UPDATE ){
2494cbf6d2d2Sdan       nVal = p->nCol * 2;
2495cbf6d2d2Sdan     }else if( p->bPatchset && op==SQLITE_DELETE ){
2496cbf6d2d2Sdan       nVal = 0;
2497cbf6d2d2Sdan       for(i=0; i<p->nCol; i++) if( p->abPK[i] ) nVal++;
2498cbf6d2d2Sdan     }else{
2499cbf6d2d2Sdan       nVal = p->nCol;
2500cbf6d2d2Sdan     }
2501cbf6d2d2Sdan     p->rc = sessionChangesetBufferRecord(&p->in, nVal, pnRec);
2502cbf6d2d2Sdan     if( p->rc!=SQLITE_OK ) return p->rc;
2503cbf6d2d2Sdan     *paRec = &p->in.aData[p->in.iNext];
2504cbf6d2d2Sdan     p->in.iNext += *pnRec;
2505cbf6d2d2Sdan   }else{
25065d607a6eSdan 
25074fccf43aSdan     /* If this is an UPDATE or DELETE, read the old.* record. */
250873b3c055Sdan     if( p->op!=SQLITE_INSERT && (p->bPatchset==0 || p->op==SQLITE_DELETE) ){
250973b3c055Sdan       u8 *abPK = p->bPatchset ? p->abPK : 0;
2510cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, abPK, p->apValue);
25114fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
25124fccf43aSdan     }
25134fccf43aSdan 
25144fccf43aSdan     /* If this is an INSERT or UPDATE, read the new.* record. */
25154fccf43aSdan     if( p->op!=SQLITE_DELETE ){
2516cbf6d2d2Sdan       p->rc = sessionReadRecord(&p->in, p->nCol, 0, &p->apValue[p->nCol]);
25174fccf43aSdan       if( p->rc!=SQLITE_OK ) return p->rc;
25184fccf43aSdan     }
25194fccf43aSdan 
2520cbf6d2d2Sdan     if( p->bPatchset && p->op==SQLITE_UPDATE ){
252173b3c055Sdan       /* If this is an UPDATE that is part of a patchset, then all PK and
252273b3c055Sdan       ** modified fields are present in the new.* record. The old.* record
252373b3c055Sdan       ** is currently completely empty. This block shifts the PK fields from
252473b3c055Sdan       ** new.* to old.*, to accommodate the code that reads these arrays.  */
252573b3c055Sdan       int i;
252673b3c055Sdan       for(i=0; i<p->nCol; i++){
252773b3c055Sdan         assert( p->apValue[i]==0 );
252873b3c055Sdan         assert( p->abPK[i]==0 || p->apValue[i+p->nCol] );
252973b3c055Sdan         if( p->abPK[i] ){
253073b3c055Sdan           p->apValue[i] = p->apValue[i+p->nCol];
253173b3c055Sdan           p->apValue[i+p->nCol] = 0;
253273b3c055Sdan         }
253373b3c055Sdan       }
253473b3c055Sdan     }
2535cbf6d2d2Sdan   }
2536ef7a6304Sdan 
25374fccf43aSdan   return SQLITE_ROW;
25384fccf43aSdan }
25394fccf43aSdan 
25404fccf43aSdan /*
25415d607a6eSdan ** Advance an iterator created by sqlite3changeset_start() to the next
25425d607a6eSdan ** change in the changeset. This function may return SQLITE_ROW, SQLITE_DONE
25435d607a6eSdan ** or SQLITE_CORRUPT.
25445d607a6eSdan **
25455d607a6eSdan ** This function may not be called on iterators passed to a conflict handler
25465d607a6eSdan ** callback by changeset_apply().
25475d607a6eSdan */
25485d607a6eSdan int sqlite3changeset_next(sqlite3_changeset_iter *p){
25495d607a6eSdan   return sessionChangesetNext(p, 0, 0);
25505d607a6eSdan }
25515d607a6eSdan 
25525d607a6eSdan /*
2553244593c8Sdan ** The following function extracts information on the current change
255477fc1d5bSdan ** from a changeset iterator. It may only be called after changeset_next()
25554fccf43aSdan ** has returned SQLITE_ROW.
25564fccf43aSdan */
25574fccf43aSdan int sqlite3changeset_op(
2558296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Iterator handle */
25594fccf43aSdan   const char **pzTab,             /* OUT: Pointer to table name */
25604fccf43aSdan   int *pnCol,                     /* OUT: Number of columns in table */
2561b4480e94Sdan   int *pOp,                       /* OUT: SQLITE_INSERT, DELETE or UPDATE */
2562b4480e94Sdan   int *pbIndirect                 /* OUT: True if change is indirect */
25634fccf43aSdan ){
25644fccf43aSdan   *pOp = pIter->op;
25654fccf43aSdan   *pnCol = pIter->nCol;
25664fccf43aSdan   *pzTab = pIter->zTab;
2567b4480e94Sdan   if( pbIndirect ) *pbIndirect = pIter->bIndirect;
25684fccf43aSdan   return SQLITE_OK;
25694fccf43aSdan }
25704fccf43aSdan 
257177fc1d5bSdan /*
257277fc1d5bSdan ** Return information regarding the PRIMARY KEY and number of columns in
257377fc1d5bSdan ** the database table affected by the change that pIter currently points
257477fc1d5bSdan ** to. This function may only be called after changeset_next() returns
257577fc1d5bSdan ** SQLITE_ROW.
257677fc1d5bSdan */
2577244593c8Sdan int sqlite3changeset_pk(
2578244593c8Sdan   sqlite3_changeset_iter *pIter,  /* Iterator object */
2579244593c8Sdan   unsigned char **pabPK,          /* OUT: Array of boolean - true for PK cols */
2580244593c8Sdan   int *pnCol                      /* OUT: Number of entries in output array */
2581244593c8Sdan ){
2582244593c8Sdan   *pabPK = pIter->abPK;
2583244593c8Sdan   if( pnCol ) *pnCol = pIter->nCol;
2584244593c8Sdan   return SQLITE_OK;
2585244593c8Sdan }
2586244593c8Sdan 
2587296c7658Sdan /*
2588296c7658Sdan ** This function may only be called while the iterator is pointing to an
2589296c7658Sdan ** SQLITE_UPDATE or SQLITE_DELETE change (see sqlite3changeset_op()).
2590296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
2591296c7658Sdan **
2592296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
2593296c7658Sdan ** iVal'th value in the old.* record. Or, if that particular value is not
2594296c7658Sdan ** included in the record (because the change is an UPDATE and the field
2595296c7658Sdan ** was not modified and is not a PK column), set *ppValue to NULL.
2596296c7658Sdan **
2597296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
2598296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
2599296c7658Sdan */
26004fccf43aSdan int sqlite3changeset_old(
2601296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2602296c7658Sdan   int iVal,                       /* Index of old.* value to retrieve */
26034fccf43aSdan   sqlite3_value **ppValue         /* OUT: Old value (or NULL pointer) */
26044fccf43aSdan ){
2605d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_DELETE ){
2606d5f0767cSdan     return SQLITE_MISUSE;
2607d5f0767cSdan   }
26084fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
26094fccf43aSdan     return SQLITE_RANGE;
26104fccf43aSdan   }
26114fccf43aSdan   *ppValue = pIter->apValue[iVal];
26124fccf43aSdan   return SQLITE_OK;
26134fccf43aSdan }
26144fccf43aSdan 
2615296c7658Sdan /*
2616296c7658Sdan ** This function may only be called while the iterator is pointing to an
2617296c7658Sdan ** SQLITE_UPDATE or SQLITE_INSERT change (see sqlite3changeset_op()).
2618296c7658Sdan ** Otherwise, SQLITE_MISUSE is returned.
2619296c7658Sdan **
2620296c7658Sdan ** It sets *ppValue to point to an sqlite3_value structure containing the
2621296c7658Sdan ** iVal'th value in the new.* record. Or, if that particular value is not
2622296c7658Sdan ** included in the record (because the change is an UPDATE and the field
2623296c7658Sdan ** was not modified), set *ppValue to NULL.
2624296c7658Sdan **
2625296c7658Sdan ** If value iVal is out-of-range, SQLITE_RANGE is returned and *ppValue is
2626296c7658Sdan ** not modified. Otherwise, SQLITE_OK.
2627296c7658Sdan */
26284fccf43aSdan int sqlite3changeset_new(
2629296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2630296c7658Sdan   int iVal,                       /* Index of new.* value to retrieve */
26314fccf43aSdan   sqlite3_value **ppValue         /* OUT: New value (or NULL pointer) */
26324fccf43aSdan ){
2633d5f0767cSdan   if( pIter->op!=SQLITE_UPDATE && pIter->op!=SQLITE_INSERT ){
2634d5f0767cSdan     return SQLITE_MISUSE;
2635d5f0767cSdan   }
26364fccf43aSdan   if( iVal<0 || iVal>=pIter->nCol ){
26374fccf43aSdan     return SQLITE_RANGE;
26384fccf43aSdan   }
26394fccf43aSdan   *ppValue = pIter->apValue[pIter->nCol+iVal];
26404fccf43aSdan   return SQLITE_OK;
26414fccf43aSdan }
26424fccf43aSdan 
2643296c7658Sdan /*
26447aa469cdSdan ** The following two macros are used internally. They are similar to the
26457aa469cdSdan ** sqlite3changeset_new() and sqlite3changeset_old() functions, except that
26467aa469cdSdan ** they omit all error checking and return a pointer to the requested value.
26477aa469cdSdan */
26487aa469cdSdan #define sessionChangesetNew(pIter, iVal) (pIter)->apValue[(pIter)->nCol+(iVal)]
26497aa469cdSdan #define sessionChangesetOld(pIter, iVal) (pIter)->apValue[(iVal)]
26507aa469cdSdan 
26517aa469cdSdan /*
2652296c7658Sdan ** This function may only be called with a changeset iterator that has been
2653296c7658Sdan ** passed to an SQLITE_CHANGESET_DATA or SQLITE_CHANGESET_CONFLICT
2654296c7658Sdan ** conflict-handler function. Otherwise, SQLITE_MISUSE is returned.
2655296c7658Sdan **
2656296c7658Sdan ** If successful, *ppValue is set to point to an sqlite3_value structure
2657296c7658Sdan ** containing the iVal'th value of the conflicting record.
2658296c7658Sdan **
2659296c7658Sdan ** If value iVal is out-of-range or some other error occurs, an SQLite error
2660296c7658Sdan ** code is returned. Otherwise, SQLITE_OK.
2661296c7658Sdan */
2662d5f0767cSdan int sqlite3changeset_conflict(
2663296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2664296c7658Sdan   int iVal,                       /* Index of conflict record value to fetch */
2665d5f0767cSdan   sqlite3_value **ppValue         /* OUT: Value from conflicting row */
2666d5f0767cSdan ){
2667d5f0767cSdan   if( !pIter->pConflict ){
2668d5f0767cSdan     return SQLITE_MISUSE;
2669d5f0767cSdan   }
2670d5f0767cSdan   if( iVal<0 || iVal>=sqlite3_column_count(pIter->pConflict) ){
2671d5f0767cSdan     return SQLITE_RANGE;
2672d5f0767cSdan   }
2673d5f0767cSdan   *ppValue = sqlite3_column_value(pIter->pConflict, iVal);
2674d5f0767cSdan   return SQLITE_OK;
2675d5f0767cSdan }
2676d5f0767cSdan 
26774fccf43aSdan /*
2678cb3e4b79Sdan ** This function may only be called with an iterator passed to an
2679cb3e4b79Sdan ** SQLITE_CHANGESET_FOREIGN_KEY conflict handler callback. In this case
2680cb3e4b79Sdan ** it sets the output variable to the total number of known foreign key
2681cb3e4b79Sdan ** violations in the destination database and returns SQLITE_OK.
2682cb3e4b79Sdan **
2683cb3e4b79Sdan ** In all other cases this function returns SQLITE_MISUSE.
2684cb3e4b79Sdan */
2685cb3e4b79Sdan int sqlite3changeset_fk_conflicts(
2686cb3e4b79Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
2687cb3e4b79Sdan   int *pnOut                      /* OUT: Number of FK violations */
2688cb3e4b79Sdan ){
2689cb3e4b79Sdan   if( pIter->pConflict || pIter->apValue ){
2690cb3e4b79Sdan     return SQLITE_MISUSE;
2691cb3e4b79Sdan   }
2692cb3e4b79Sdan   *pnOut = pIter->nCol;
2693cb3e4b79Sdan   return SQLITE_OK;
2694cb3e4b79Sdan }
2695cb3e4b79Sdan 
2696cb3e4b79Sdan 
2697cb3e4b79Sdan /*
26984fccf43aSdan ** Finalize an iterator allocated with sqlite3changeset_start().
26994fccf43aSdan **
27004fccf43aSdan ** This function may not be called on iterators passed to a conflict handler
27014fccf43aSdan ** callback by changeset_apply().
27024fccf43aSdan */
27034fccf43aSdan int sqlite3changeset_finalize(sqlite3_changeset_iter *p){
2704cbf6d2d2Sdan   int rc = SQLITE_OK;
2705cbf6d2d2Sdan   if( p ){
2706296c7658Sdan     int i;                        /* Used to iterate through p->apValue[] */
2707cbf6d2d2Sdan     rc = p->rc;
270812ca0b56Sdan     if( p->apValue ){
27094fccf43aSdan       for(i=0; i<p->nCol*2; i++) sqlite3ValueFree(p->apValue[i]);
271012ca0b56Sdan     }
2711ef7a6304Sdan     sqlite3_free(p->tblhdr.aBuf);
27124757c658Sdan     sqlite3_free(p->in.buf.aBuf);
27134fccf43aSdan     sqlite3_free(p);
2714cbf6d2d2Sdan   }
27154fccf43aSdan   return rc;
27164fccf43aSdan }
27174fccf43aSdan 
2718fa122adaSdan static int sessionChangesetInvert(
2719fa122adaSdan   SessionInput *pInput,           /* Input changeset */
2720fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2721fa122adaSdan   void *pOut,
272291ddd559Sdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
272391ddd559Sdan   void **ppInverted               /* OUT: Inverse of pChangeset */
272491ddd559Sdan ){
2725cfec7eeeSdan   int rc = SQLITE_OK;             /* Return value */
2726fa122adaSdan   SessionBuffer sOut;             /* Output buffer */
2727cfec7eeeSdan   int nCol = 0;                   /* Number of cols in current table */
2728cfec7eeeSdan   u8 *abPK = 0;                   /* PK array for current table */
2729cfec7eeeSdan   sqlite3_value **apVal = 0;      /* Space for values for UPDATE inversion */
2730ef7a6304Sdan   SessionBuffer sPK = {0, 0, 0};  /* PK array for current table */
273191ddd559Sdan 
2732fa122adaSdan   /* Initialize the output buffer */
2733fa122adaSdan   memset(&sOut, 0, sizeof(SessionBuffer));
2734fa122adaSdan 
273591ddd559Sdan   /* Zero the output variables in case an error occurs. */
2736fa122adaSdan   if( ppInverted ){
273791ddd559Sdan     *ppInverted = 0;
273891ddd559Sdan     *pnInverted = 0;
2739fa122adaSdan   }
274091ddd559Sdan 
2741fa122adaSdan   while( 1 ){
2742ef7a6304Sdan     u8 eType;
2743fa122adaSdan 
2744fa122adaSdan     /* Test for EOF. */
2745fa122adaSdan     if( (rc = sessionInputBuffer(pInput, 2)) ) goto finished_invert;
2746fa122adaSdan     if( pInput->iNext>=pInput->nData ) break;
2747fa122adaSdan     eType = pInput->aData[pInput->iNext];
2748fa122adaSdan 
274991ddd559Sdan     switch( eType ){
275091ddd559Sdan       case 'T': {
2751244593c8Sdan         /* A 'table' record consists of:
2752244593c8Sdan         **
2753244593c8Sdan         **   * A constant 'T' character,
2754244593c8Sdan         **   * Number of columns in said table (a varint),
2755ef7a6304Sdan         **   * An array of nCol bytes (sPK),
2756244593c8Sdan         **   * A nul-terminated table name.
2757244593c8Sdan         */
2758ef7a6304Sdan         int nByte;
2759fa122adaSdan         int nVar;
2760fa122adaSdan         pInput->iNext++;
2761fa122adaSdan         if( (rc = sessionChangesetBufferTblhdr(pInput, &nByte)) ){
2762ef7a6304Sdan           goto finished_invert;
2763ef7a6304Sdan         }
2764fa122adaSdan         nVar = sessionVarintGet(&pInput->aData[pInput->iNext], &nCol);
2765ef7a6304Sdan         sPK.nBuf = 0;
2766fa122adaSdan         sessionAppendBlob(&sPK, &pInput->aData[pInput->iNext+nVar], nCol, &rc);
2767fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
2768fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
2769ef7a6304Sdan         if( rc ) goto finished_invert;
2770fa122adaSdan 
2771fa122adaSdan         pInput->iNext += nByte;
2772cfec7eeeSdan         sqlite3_free(apVal);
2773cfec7eeeSdan         apVal = 0;
2774ef7a6304Sdan         abPK = sPK.aBuf;
277591ddd559Sdan         break;
277691ddd559Sdan       }
277791ddd559Sdan 
277891ddd559Sdan       case SQLITE_INSERT:
277991ddd559Sdan       case SQLITE_DELETE: {
278091ddd559Sdan         int nByte;
2781fa122adaSdan         int bIndirect = pInput->aData[pInput->iNext+1];
2782fa122adaSdan         int eType2 = (eType==SQLITE_DELETE ? SQLITE_INSERT : SQLITE_DELETE);
2783fa122adaSdan         pInput->iNext += 2;
2784fa122adaSdan         assert( rc==SQLITE_OK );
2785fa122adaSdan         rc = sessionChangesetBufferRecord(pInput, nCol, &nByte);
2786fa122adaSdan         sessionAppendByte(&sOut, eType2, &rc);
2787fa122adaSdan         sessionAppendByte(&sOut, bIndirect, &rc);
2788fa122adaSdan         sessionAppendBlob(&sOut, &pInput->aData[pInput->iNext], nByte, &rc);
2789fa122adaSdan         pInput->iNext += nByte;
2790fa122adaSdan         if( rc ) goto finished_invert;
279191ddd559Sdan         break;
279291ddd559Sdan       }
279391ddd559Sdan 
279491ddd559Sdan       case SQLITE_UPDATE: {
2795cfec7eeeSdan         int iCol;
279691ddd559Sdan 
2797cfec7eeeSdan         if( 0==apVal ){
2798cfec7eeeSdan           apVal = (sqlite3_value **)sqlite3_malloc(sizeof(apVal[0])*nCol*2);
2799cfec7eeeSdan           if( 0==apVal ){
2800cfec7eeeSdan             rc = SQLITE_NOMEM;
2801cfec7eeeSdan             goto finished_invert;
2802cfec7eeeSdan           }
2803cfec7eeeSdan           memset(apVal, 0, sizeof(apVal[0])*nCol*2);
2804cfec7eeeSdan         }
280591ddd559Sdan 
2806cfec7eeeSdan         /* Write the header for the new UPDATE change. Same as the original. */
2807fa122adaSdan         sessionAppendByte(&sOut, eType, &rc);
2808fa122adaSdan         sessionAppendByte(&sOut, pInput->aData[pInput->iNext+1], &rc);
280991ddd559Sdan 
2810ef7a6304Sdan         /* Read the old.* and new.* records for the update change. */
2811fa122adaSdan         pInput->iNext += 2;
2812fa122adaSdan         rc = sessionReadRecord(pInput, nCol, 0, &apVal[0]);
2813ef7a6304Sdan         if( rc==SQLITE_OK ){
2814fa122adaSdan           rc = sessionReadRecord(pInput, nCol, 0, &apVal[nCol]);
2815ef7a6304Sdan         }
2816ef7a6304Sdan 
2817cfec7eeeSdan         /* Write the new old.* record. Consists of the PK columns from the
2818cfec7eeeSdan         ** original old.* record, and the other values from the original
2819cfec7eeeSdan         ** new.* record. */
2820e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
2821cfec7eeeSdan           sqlite3_value *pVal = apVal[iCol + (abPK[iCol] ? 0 : nCol)];
2822fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
2823cfec7eeeSdan         }
2824cfec7eeeSdan 
2825cfec7eeeSdan         /* Write the new new.* record. Consists of a copy of all values
2826cfec7eeeSdan         ** from the original old.* record, except for the PK columns, which
2827cfec7eeeSdan         ** are set to "undefined". */
2828e8fa8c96Sdan         for(iCol=0; iCol<nCol; iCol++){
2829cfec7eeeSdan           sqlite3_value *pVal = (abPK[iCol] ? 0 : apVal[iCol]);
2830fa122adaSdan           sessionAppendValue(&sOut, pVal, &rc);
2831cfec7eeeSdan         }
2832cfec7eeeSdan 
2833cfec7eeeSdan         for(iCol=0; iCol<nCol*2; iCol++){
2834cfec7eeeSdan           sqlite3ValueFree(apVal[iCol]);
2835cfec7eeeSdan         }
2836cfec7eeeSdan         memset(apVal, 0, sizeof(apVal[0])*nCol*2);
2837cfec7eeeSdan         if( rc!=SQLITE_OK ){
2838cfec7eeeSdan           goto finished_invert;
2839cfec7eeeSdan         }
2840cfec7eeeSdan 
284191ddd559Sdan         break;
284291ddd559Sdan       }
284391ddd559Sdan 
284491ddd559Sdan       default:
28454757c658Sdan         rc = SQLITE_CORRUPT_BKPT;
2846cfec7eeeSdan         goto finished_invert;
284791ddd559Sdan     }
2848fa122adaSdan 
2849fa122adaSdan     assert( rc==SQLITE_OK );
2850f1a08ad8Sdrh     if( xOutput && sOut.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
2851fa122adaSdan       rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
2852fa122adaSdan       sOut.nBuf = 0;
2853fa122adaSdan       if( rc!=SQLITE_OK ) goto finished_invert;
2854fa122adaSdan     }
285591ddd559Sdan   }
285691ddd559Sdan 
2857cfec7eeeSdan   assert( rc==SQLITE_OK );
2858fa122adaSdan   if( pnInverted ){
2859fa122adaSdan     *pnInverted = sOut.nBuf;
2860fa122adaSdan     *ppInverted = sOut.aBuf;
2861fa122adaSdan     sOut.aBuf = 0;
2862fa122adaSdan   }else if( sOut.nBuf>0 ){
2863fa122adaSdan     rc = xOutput(pOut, sOut.aBuf, sOut.nBuf);
2864fa122adaSdan   }
2865cfec7eeeSdan 
2866cfec7eeeSdan  finished_invert:
2867fa122adaSdan   sqlite3_free(sOut.aBuf);
2868cfec7eeeSdan   sqlite3_free(apVal);
2869ef7a6304Sdan   sqlite3_free(sPK.aBuf);
2870cfec7eeeSdan   return rc;
287191ddd559Sdan }
287291ddd559Sdan 
2873fa122adaSdan 
2874fa122adaSdan /*
2875fa122adaSdan ** Invert a changeset object.
2876fa122adaSdan */
2877fa122adaSdan int sqlite3changeset_invert(
2878fa122adaSdan   int nChangeset,                 /* Number of bytes in input */
2879fa122adaSdan   const void *pChangeset,         /* Input changeset */
2880fa122adaSdan   int *pnInverted,                /* OUT: Number of bytes in output changeset */
2881fa122adaSdan   void **ppInverted               /* OUT: Inverse of pChangeset */
2882fa122adaSdan ){
2883fa122adaSdan   SessionInput sInput;
2884fa122adaSdan 
2885fa122adaSdan   /* Set up the input stream */
2886fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
2887fa122adaSdan   sInput.nData = nChangeset;
2888fa122adaSdan   sInput.aData = (u8*)pChangeset;
2889fa122adaSdan 
2890fa122adaSdan   return sessionChangesetInvert(&sInput, 0, 0, pnInverted, ppInverted);
2891fa122adaSdan }
2892fa122adaSdan 
2893fa122adaSdan /*
2894fa122adaSdan ** Streaming version of sqlite3changeset_invert().
2895fa122adaSdan */
2896f1a08ad8Sdrh int sqlite3changeset_invert_strm(
2897fa122adaSdan   int (*xInput)(void *pIn, void *pData, int *pnData),
2898fa122adaSdan   void *pIn,
2899fa122adaSdan   int (*xOutput)(void *pOut, const void *pData, int nData),
2900fa122adaSdan   void *pOut
2901fa122adaSdan ){
2902fa122adaSdan   SessionInput sInput;
2903fa122adaSdan   int rc;
2904fa122adaSdan 
2905fa122adaSdan   /* Set up the input stream */
2906fa122adaSdan   memset(&sInput, 0, sizeof(SessionInput));
2907fa122adaSdan   sInput.xInput = xInput;
2908fa122adaSdan   sInput.pIn = pIn;
2909fa122adaSdan 
2910fa122adaSdan   rc = sessionChangesetInvert(&sInput, xOutput, pOut, 0, 0);
2911fa122adaSdan   sqlite3_free(sInput.buf.aBuf);
2912fa122adaSdan   return rc;
2913fa122adaSdan }
2914fa122adaSdan 
29150c698471Sdan typedef struct SessionApplyCtx SessionApplyCtx;
29160c698471Sdan struct SessionApplyCtx {
29170c698471Sdan   sqlite3 *db;
29180c698471Sdan   sqlite3_stmt *pDelete;          /* DELETE statement */
2919cfec7eeeSdan   sqlite3_stmt *pUpdate;          /* UPDATE statement */
29200c698471Sdan   sqlite3_stmt *pInsert;          /* INSERT statement */
29210c698471Sdan   sqlite3_stmt *pSelect;          /* SELECT statement */
29220c698471Sdan   int nCol;                       /* Size of azCol[] and abPK[] arrays */
29230c698471Sdan   const char **azCol;             /* Array of column names */
29240c698471Sdan   u8 *abPK;                       /* Boolean array - true if column is in PK */
29250c698471Sdan };
29260c698471Sdan 
2927d5f0767cSdan /*
2928d5f0767cSdan ** Formulate a statement to DELETE a row from database db. Assuming a table
2929d5f0767cSdan ** structure like this:
2930d5f0767cSdan **
2931d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
2932d5f0767cSdan **
2933d5f0767cSdan ** The DELETE statement looks like this:
2934d5f0767cSdan **
2935db04571cSdan **     DELETE FROM x WHERE a = :1 AND c = :3 AND (:5 OR b IS :2 AND d IS :4)
2936d5f0767cSdan **
2937d5f0767cSdan ** Variable :5 (nCol+1) is a boolean. It should be set to 0 if we require
2938d5f0767cSdan ** matching b and d values, or 1 otherwise. The second case comes up if the
2939d5f0767cSdan ** conflict handler is invoked with NOTFOUND and returns CHANGESET_REPLACE.
2940296c7658Sdan **
2941296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pDelete is left
2942296c7658Sdan ** pointing to the prepared version of the SQL statement.
2943d5f0767cSdan */
2944d5f0767cSdan static int sessionDeleteRow(
2945d5f0767cSdan   sqlite3 *db,                    /* Database handle */
2946d5f0767cSdan   const char *zTab,               /* Table name */
29470c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
2948d5f0767cSdan ){
2949296c7658Sdan   int i;
2950296c7658Sdan   const char *zSep = "";
2951d5f0767cSdan   int rc = SQLITE_OK;
2952d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
29537cf7df7dSdan   int nPk = 0;
2954d5f0767cSdan 
2955d5f0767cSdan   sessionAppendStr(&buf, "DELETE FROM ", &rc);
2956d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
2957296c7658Sdan   sessionAppendStr(&buf, " WHERE ", &rc);
2958296c7658Sdan 
2959296c7658Sdan   for(i=0; i<p->nCol; i++){
2960296c7658Sdan     if( p->abPK[i] ){
29617cf7df7dSdan       nPk++;
2962296c7658Sdan       sessionAppendStr(&buf, zSep, &rc);
2963296c7658Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
2964296c7658Sdan       sessionAppendStr(&buf, " = ?", &rc);
2965296c7658Sdan       sessionAppendInteger(&buf, i+1, &rc);
2966296c7658Sdan       zSep = " AND ";
2967296c7658Sdan     }
2968296c7658Sdan   }
2969296c7658Sdan 
29707cf7df7dSdan   if( nPk<p->nCol ){
2971296c7658Sdan     sessionAppendStr(&buf, " AND (?", &rc);
2972296c7658Sdan     sessionAppendInteger(&buf, p->nCol+1, &rc);
2973296c7658Sdan     sessionAppendStr(&buf, " OR ", &rc);
2974296c7658Sdan 
2975296c7658Sdan     zSep = "";
2976296c7658Sdan     for(i=0; i<p->nCol; i++){
2977296c7658Sdan       if( !p->abPK[i] ){
2978296c7658Sdan         sessionAppendStr(&buf, zSep, &rc);
2979296c7658Sdan         sessionAppendIdent(&buf, p->azCol[i], &rc);
2980296c7658Sdan         sessionAppendStr(&buf, " IS ?", &rc);
2981296c7658Sdan         sessionAppendInteger(&buf, i+1, &rc);
2982296c7658Sdan         zSep = "AND ";
2983296c7658Sdan       }
2984296c7658Sdan     }
2985296c7658Sdan     sessionAppendStr(&buf, ")", &rc);
29867cf7df7dSdan   }
2987d5f0767cSdan 
2988d5f0767cSdan   if( rc==SQLITE_OK ){
29890c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pDelete, 0);
2990d5f0767cSdan   }
2991d5f0767cSdan   sqlite3_free(buf.aBuf);
2992d5f0767cSdan 
2993d5f0767cSdan   return rc;
2994d5f0767cSdan }
2995d5f0767cSdan 
2996d5f0767cSdan /*
2997d5f0767cSdan ** Formulate and prepare a statement to UPDATE a row from database db.
2998d5f0767cSdan ** Assuming a table structure like this:
2999d5f0767cSdan **
3000d5f0767cSdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3001d5f0767cSdan **
3002d5f0767cSdan ** The UPDATE statement looks like this:
3003d5f0767cSdan **
3004d5f0767cSdan **     UPDATE x SET
3005d5f0767cSdan **     a = CASE WHEN ?2  THEN ?3  ELSE a END,
3006964cbd46Sdan **     b = CASE WHEN ?5  THEN ?6  ELSE b END,
3007964cbd46Sdan **     c = CASE WHEN ?8  THEN ?9  ELSE c END,
3008964cbd46Sdan **     d = CASE WHEN ?11 THEN ?12 ELSE d END
3009d5f0767cSdan **     WHERE a = ?1 AND c = ?7 AND (?13 OR
3010964cbd46Sdan **       (?5==0 OR b IS ?4) AND (?11==0 OR d IS ?10) AND
3011d5f0767cSdan **     )
3012d5f0767cSdan **
3013d5f0767cSdan ** For each column in the table, there are three variables to bind:
3014d5f0767cSdan **
3015d5f0767cSdan **     ?(i*3+1)    The old.* value of the column, if any.
3016d5f0767cSdan **     ?(i*3+2)    A boolean flag indicating that the value is being modified.
3017d5f0767cSdan **     ?(i*3+3)    The new.* value of the column, if any.
3018d5f0767cSdan **
3019d5f0767cSdan ** Also, a boolean flag that, if set to true, causes the statement to update
3020d5f0767cSdan ** a row even if the non-PK values do not match. This is required if the
3021d5f0767cSdan ** conflict-handler is invoked with CHANGESET_DATA and returns
3022d5f0767cSdan ** CHANGESET_REPLACE. This is variable "?(nCol*3+1)".
3023d5f0767cSdan **
3024296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pUpdate is left
3025296c7658Sdan ** pointing to the prepared version of the SQL statement.
3026d5f0767cSdan */
3027d5f0767cSdan static int sessionUpdateRow(
3028d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3029d5f0767cSdan   const char *zTab,               /* Table name */
30300c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3031d5f0767cSdan ){
3032d5f0767cSdan   int rc = SQLITE_OK;
3033d5f0767cSdan   int i;
3034d5f0767cSdan   const char *zSep = "";
3035d5f0767cSdan   SessionBuffer buf = {0, 0, 0};
3036d5f0767cSdan 
3037d5f0767cSdan   /* Append "UPDATE tbl SET " */
3038d5f0767cSdan   sessionAppendStr(&buf, "UPDATE ", &rc);
3039d5f0767cSdan   sessionAppendIdent(&buf, zTab, &rc);
3040d5f0767cSdan   sessionAppendStr(&buf, " SET ", &rc);
3041d5f0767cSdan 
3042d5f0767cSdan   /* Append the assignments */
30430c698471Sdan   for(i=0; i<p->nCol; i++){
3044d5f0767cSdan     sessionAppendStr(&buf, zSep, &rc);
30450c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3046d5f0767cSdan     sessionAppendStr(&buf, " = CASE WHEN ?", &rc);
3047d5f0767cSdan     sessionAppendInteger(&buf, i*3+2, &rc);
3048d5f0767cSdan     sessionAppendStr(&buf, " THEN ?", &rc);
3049d5f0767cSdan     sessionAppendInteger(&buf, i*3+3, &rc);
3050d5f0767cSdan     sessionAppendStr(&buf, " ELSE ", &rc);
30510c698471Sdan     sessionAppendIdent(&buf, p->azCol[i], &rc);
3052d5f0767cSdan     sessionAppendStr(&buf, " END", &rc);
3053d5f0767cSdan     zSep = ", ";
3054d5f0767cSdan   }
3055d5f0767cSdan 
3056d5f0767cSdan   /* Append the PK part of the WHERE clause */
3057d5f0767cSdan   sessionAppendStr(&buf, " WHERE ", &rc);
30580c698471Sdan   for(i=0; i<p->nCol; i++){
30590c698471Sdan     if( p->abPK[i] ){
30600c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3061d5f0767cSdan       sessionAppendStr(&buf, " = ?", &rc);
3062d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3063d5f0767cSdan       sessionAppendStr(&buf, " AND ", &rc);
3064d5f0767cSdan     }
3065d5f0767cSdan   }
3066d5f0767cSdan 
3067d5f0767cSdan   /* Append the non-PK part of the WHERE clause */
3068d5f0767cSdan   sessionAppendStr(&buf, " (?", &rc);
30690c698471Sdan   sessionAppendInteger(&buf, p->nCol*3+1, &rc);
3070d5f0767cSdan   sessionAppendStr(&buf, " OR 1", &rc);
30710c698471Sdan   for(i=0; i<p->nCol; i++){
30720c698471Sdan     if( !p->abPK[i] ){
3073d5f0767cSdan       sessionAppendStr(&buf, " AND (?", &rc);
3074d5f0767cSdan       sessionAppendInteger(&buf, i*3+2, &rc);
3075d5f0767cSdan       sessionAppendStr(&buf, "=0 OR ", &rc);
30760c698471Sdan       sessionAppendIdent(&buf, p->azCol[i], &rc);
3077d5f0767cSdan       sessionAppendStr(&buf, " IS ?", &rc);
3078d5f0767cSdan       sessionAppendInteger(&buf, i*3+1, &rc);
3079d5f0767cSdan       sessionAppendStr(&buf, ")", &rc);
3080d5f0767cSdan     }
3081d5f0767cSdan   }
3082d5f0767cSdan   sessionAppendStr(&buf, ")", &rc);
3083d5f0767cSdan 
3084d5f0767cSdan   if( rc==SQLITE_OK ){
30850c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pUpdate, 0);
3086d5f0767cSdan   }
3087d5f0767cSdan   sqlite3_free(buf.aBuf);
3088d5f0767cSdan 
3089d5f0767cSdan   return rc;
3090d5f0767cSdan }
3091d5f0767cSdan 
3092296c7658Sdan /*
3093296c7658Sdan ** Formulate and prepare an SQL statement to query table zTab by primary
3094296c7658Sdan ** key. Assuming the following table structure:
3095296c7658Sdan **
3096296c7658Sdan **     CREATE TABLE x(a, b, c, d, PRIMARY KEY(a, c));
3097296c7658Sdan **
3098296c7658Sdan ** The SELECT statement looks like this:
3099296c7658Sdan **
3100296c7658Sdan **     SELECT * FROM x WHERE a = ?1 AND c = ?3
3101296c7658Sdan **
3102296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pSelect is left
3103296c7658Sdan ** pointing to the prepared version of the SQL statement.
3104296c7658Sdan */
3105d5f0767cSdan static int sessionSelectRow(
3106d5f0767cSdan   sqlite3 *db,                    /* Database handle */
3107d5f0767cSdan   const char *zTab,               /* Table name */
31080c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
3109d5f0767cSdan ){
3110d7fb7d24Sdan   return sessionSelectStmt(
3111d7fb7d24Sdan       db, "main", zTab, p->nCol, p->azCol, p->abPK, &p->pSelect);
3112d5f0767cSdan }
3113d5f0767cSdan 
3114296c7658Sdan /*
3115296c7658Sdan ** Formulate and prepare an INSERT statement to add a record to table zTab.
3116296c7658Sdan ** For example:
3117296c7658Sdan **
3118296c7658Sdan **     INSERT INTO main."zTab" VALUES(?1, ?2, ?3 ...);
3119296c7658Sdan **
3120296c7658Sdan ** If successful, SQLITE_OK is returned and SessionApplyCtx.pInsert is left
3121296c7658Sdan ** pointing to the prepared version of the SQL statement.
3122296c7658Sdan */
31230c698471Sdan static int sessionInsertRow(
31240c698471Sdan   sqlite3 *db,                    /* Database handle */
31250c698471Sdan   const char *zTab,               /* Table name */
31260c698471Sdan   SessionApplyCtx *p              /* Session changeset-apply context */
31270c698471Sdan ){
31280c698471Sdan   int rc = SQLITE_OK;
31290c698471Sdan   int i;
31300c698471Sdan   SessionBuffer buf = {0, 0, 0};
31310c698471Sdan 
31320c698471Sdan   sessionAppendStr(&buf, "INSERT INTO main.", &rc);
31330c698471Sdan   sessionAppendIdent(&buf, zTab, &rc);
31340c698471Sdan   sessionAppendStr(&buf, " VALUES(?", &rc);
31350c698471Sdan   for(i=1; i<p->nCol; i++){
31360c698471Sdan     sessionAppendStr(&buf, ", ?", &rc);
31370c698471Sdan   }
31380c698471Sdan   sessionAppendStr(&buf, ")", &rc);
31390c698471Sdan 
31400c698471Sdan   if( rc==SQLITE_OK ){
31410c698471Sdan     rc = sqlite3_prepare_v2(db, (char *)buf.aBuf, buf.nBuf, &p->pInsert, 0);
31420c698471Sdan   }
31430c698471Sdan   sqlite3_free(buf.aBuf);
31440c698471Sdan   return rc;
31450c698471Sdan }
31460c698471Sdan 
3147296c7658Sdan /*
31487aa469cdSdan ** A wrapper around sqlite3_bind_value() that detects an extra problem.
31497aa469cdSdan ** See comments in the body of this function for details.
31507aa469cdSdan */
31517aa469cdSdan static int sessionBindValue(
31527aa469cdSdan   sqlite3_stmt *pStmt,            /* Statement to bind value to */
31537aa469cdSdan   int i,                          /* Parameter number to bind to */
31547aa469cdSdan   sqlite3_value *pVal             /* Value to bind */
31557aa469cdSdan ){
31565671ef69Sdrh   int eType = sqlite3_value_type(pVal);
3157082c96dfSdan   /* COVERAGE: The (pVal->z==0) branch is never true using current versions
3158082c96dfSdan   ** of SQLite. If a malloc fails in an sqlite3_value_xxx() function, either
3159082c96dfSdan   ** the (pVal->z) variable remains as it was or the type of the value is
3160082c96dfSdan   ** set to SQLITE_NULL.  */
31615671ef69Sdrh   if( (eType==SQLITE_TEXT || eType==SQLITE_BLOB) && pVal->z==0 ){
31627aa469cdSdan     /* This condition occurs when an earlier OOM in a call to
31637aa469cdSdan     ** sqlite3_value_text() or sqlite3_value_blob() (perhaps from within
3164082c96dfSdan     ** a conflict-handler) has zeroed the pVal->z pointer. Return NOMEM. */
31657aa469cdSdan     return SQLITE_NOMEM;
31667aa469cdSdan   }
31677aa469cdSdan   return sqlite3_bind_value(pStmt, i, pVal);
31687aa469cdSdan }
31697aa469cdSdan 
31707aa469cdSdan /*
3171db04571cSdan ** Iterator pIter must point to an SQLITE_INSERT entry. This function
3172db04571cSdan ** transfers new.* values from the current iterator entry to statement
3173db04571cSdan ** pStmt. The table being inserted into has nCol columns.
3174db04571cSdan **
3175db04571cSdan ** New.* value $i 0 from the iterator is bound to variable ($i+1) of
3176db04571cSdan ** statement pStmt. If parameter abPK is NULL, all values from 0 to (nCol-1)
3177db04571cSdan ** are transfered to the statement. Otherwise, if abPK is not NULL, it points
3178db04571cSdan ** to an array nCol elements in size. In this case only those values for
3179db04571cSdan ** which abPK[$i] is true are read from the iterator and bound to the
3180db04571cSdan ** statement.
3181db04571cSdan **
3182db04571cSdan ** An SQLite error code is returned if an error occurs. Otherwise, SQLITE_OK.
3183db04571cSdan */
31847aa469cdSdan static int sessionBindRow(
3185db04571cSdan   sqlite3_changeset_iter *pIter,  /* Iterator to read values from */
31867aa469cdSdan   int(*xValue)(sqlite3_changeset_iter *, int, sqlite3_value **),
3187db04571cSdan   int nCol,                       /* Number of columns */
3188db04571cSdan   u8 *abPK,                       /* If not NULL, bind only if true */
3189db04571cSdan   sqlite3_stmt *pStmt             /* Bind values to this statement */
3190db04571cSdan ){
3191db04571cSdan   int i;
3192db04571cSdan   int rc = SQLITE_OK;
31937aa469cdSdan 
31947aa469cdSdan   /* Neither sqlite3changeset_old or sqlite3changeset_new can fail if the
31957aa469cdSdan   ** argument iterator points to a suitable entry. Make sure that xValue
31967aa469cdSdan   ** is one of these to guarantee that it is safe to ignore the return
31977aa469cdSdan   ** in the code below. */
31987aa469cdSdan   assert( xValue==sqlite3changeset_old || xValue==sqlite3changeset_new );
31997aa469cdSdan 
3200db04571cSdan   for(i=0; rc==SQLITE_OK && i<nCol; i++){
3201db04571cSdan     if( !abPK || abPK[i] ){
3202db04571cSdan       sqlite3_value *pVal;
32037aa469cdSdan       (void)xValue(pIter, i, &pVal);
32047aa469cdSdan       rc = sessionBindValue(pStmt, i+1, pVal);
3205db04571cSdan     }
3206db04571cSdan   }
3207db04571cSdan   return rc;
3208db04571cSdan }
3209db04571cSdan 
3210db04571cSdan /*
3211296c7658Sdan ** SQL statement pSelect is as generated by the sessionSelectRow() function.
3212296c7658Sdan ** This function binds the primary key values from the change that changeset
3213296c7658Sdan ** iterator pIter points to to the SELECT and attempts to seek to the table
3214296c7658Sdan ** entry. If a row is found, the SELECT statement left pointing at the row
3215296c7658Sdan ** and SQLITE_ROW is returned. Otherwise, if no row is found and no error
3216296c7658Sdan ** has occured, the statement is reset and SQLITE_OK is returned. If an
32177aa469cdSdan ** error occurs, the statement is reset and an SQLite error code is returned.
32187aa469cdSdan **
32197aa469cdSdan ** If this function returns SQLITE_ROW, the caller must eventually reset()
32207aa469cdSdan ** statement pSelect. If any other value is returned, the statement does
32217aa469cdSdan ** not require a reset().
3222296c7658Sdan **
3223296c7658Sdan ** If the iterator currently points to an INSERT record, bind values from the
3224db04571cSdan ** new.* record to the SELECT statement. Or, if it points to a DELETE or
3225db04571cSdan ** UPDATE, bind values from the old.* record.
3226296c7658Sdan */
32270c698471Sdan static int sessionSeekToRow(
322837f133ecSdan   sqlite3 *db,                    /* Database handle */
322937f133ecSdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
323037f133ecSdan   u8 *abPK,                       /* Primary key flags array */
32310c698471Sdan   sqlite3_stmt *pSelect           /* SELECT statement from sessionSelectRow() */
323237f133ecSdan ){
32337aa469cdSdan   int rc;                         /* Return code */
3234296c7658Sdan   int nCol;                       /* Number of columns in table */
3235296c7658Sdan   int op;                         /* Changset operation (SQLITE_UPDATE etc.) */
3236296c7658Sdan   const char *zDummy;             /* Unused */
323737f133ecSdan 
3238b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
32397aa469cdSdan   rc = sessionBindRow(pIter,
3240db04571cSdan       op==SQLITE_INSERT ? sqlite3changeset_new : sqlite3changeset_old,
3241db04571cSdan       nCol, abPK, pSelect
3242db04571cSdan   );
32430c698471Sdan 
32440c698471Sdan   if( rc==SQLITE_OK ){
32450c698471Sdan     rc = sqlite3_step(pSelect);
32460c698471Sdan     if( rc!=SQLITE_ROW ) rc = sqlite3_reset(pSelect);
32470c698471Sdan   }
32480c698471Sdan 
32490c698471Sdan   return rc;
32500c698471Sdan }
32510c698471Sdan 
3252296c7658Sdan /*
3253296c7658Sdan ** Invoke the conflict handler for the change that the changeset iterator
3254296c7658Sdan ** currently points to.
3255296c7658Sdan **
3256296c7658Sdan ** Argument eType must be either CHANGESET_DATA or CHANGESET_CONFLICT.
3257296c7658Sdan ** If argument pbReplace is NULL, then the type of conflict handler invoked
3258296c7658Sdan ** depends solely on eType, as follows:
3259296c7658Sdan **
3260296c7658Sdan **    eType value                 Value passed to xConflict
3261296c7658Sdan **    -------------------------------------------------
3262296c7658Sdan **    CHANGESET_DATA              CHANGESET_NOTFOUND
3263296c7658Sdan **    CHANGESET_CONFLICT          CHANGESET_CONSTRAINT
3264296c7658Sdan **
3265296c7658Sdan ** Or, if pbReplace is not NULL, then an attempt is made to find an existing
3266296c7658Sdan ** record with the same primary key as the record about to be deleted, updated
3267296c7658Sdan ** or inserted. If such a record can be found, it is available to the conflict
3268296c7658Sdan ** handler as the "conflicting" record. In this case the type of conflict
3269296c7658Sdan ** handler invoked is as follows:
3270296c7658Sdan **
3271296c7658Sdan **    eType value         PK Record found?   Value passed to xConflict
3272296c7658Sdan **    ----------------------------------------------------------------
3273296c7658Sdan **    CHANGESET_DATA      Yes                CHANGESET_DATA
3274296c7658Sdan **    CHANGESET_DATA      No                 CHANGESET_NOTFOUND
3275296c7658Sdan **    CHANGESET_CONFLICT  Yes                CHANGESET_CONFLICT
3276296c7658Sdan **    CHANGESET_CONFLICT  No                 CHANGESET_CONSTRAINT
3277296c7658Sdan **
3278296c7658Sdan ** If pbReplace is not NULL, and a record with a matching PK is found, and
3279296c7658Sdan ** the conflict handler function returns SQLITE_CHANGESET_REPLACE, *pbReplace
3280296c7658Sdan ** is set to non-zero before returning SQLITE_OK.
3281296c7658Sdan **
3282296c7658Sdan ** If the conflict handler returns SQLITE_CHANGESET_ABORT, SQLITE_ABORT is
3283296c7658Sdan ** returned. Or, if the conflict handler returns an invalid value,
3284296c7658Sdan ** SQLITE_MISUSE. If the conflict handler returns SQLITE_CHANGESET_OMIT,
3285296c7658Sdan ** this function returns SQLITE_OK.
3286296c7658Sdan */
32870c698471Sdan static int sessionConflictHandler(
3288296c7658Sdan   int eType,                      /* Either CHANGESET_DATA or CONFLICT */
3289296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
32900c698471Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
32910c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter*),
3292296c7658Sdan   void *pCtx,                     /* First argument for conflict handler */
3293296c7658Sdan   int *pbReplace                  /* OUT: Set to true if PK row is found */
32940c698471Sdan ){
3295296c7658Sdan   int res;                        /* Value returned by conflict handler */
32960c698471Sdan   int rc;
32970c698471Sdan   int nCol;
32980c698471Sdan   int op;
32990c698471Sdan   const char *zDummy;
33000c698471Sdan 
3301b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
33020c698471Sdan 
33030c698471Sdan   assert( eType==SQLITE_CHANGESET_CONFLICT || eType==SQLITE_CHANGESET_DATA );
33040c698471Sdan   assert( SQLITE_CHANGESET_CONFLICT+1==SQLITE_CHANGESET_CONSTRAINT );
33050c698471Sdan   assert( SQLITE_CHANGESET_DATA+1==SQLITE_CHANGESET_NOTFOUND );
330637f133ecSdan 
330737f133ecSdan   /* Bind the new.* PRIMARY KEY values to the SELECT statement. */
33080c698471Sdan   if( pbReplace ){
33090c698471Sdan     rc = sessionSeekToRow(p->db, pIter, p->abPK, p->pSelect);
33100c698471Sdan   }else{
3311db04571cSdan     rc = SQLITE_OK;
33120c698471Sdan   }
33130c698471Sdan 
33140c698471Sdan   if( rc==SQLITE_ROW ){
33150c698471Sdan     /* There exists another row with the new.* primary key. */
33160c698471Sdan     pIter->pConflict = p->pSelect;
33170c698471Sdan     res = xConflict(pCtx, eType, pIter);
33180c698471Sdan     pIter->pConflict = 0;
33190c698471Sdan     rc = sqlite3_reset(p->pSelect);
3320db04571cSdan   }else if( rc==SQLITE_OK ){
33210c698471Sdan     /* No other row with the new.* primary key. */
33220c698471Sdan     res = xConflict(pCtx, eType+1, pIter);
33230c698471Sdan     if( res==SQLITE_CHANGESET_REPLACE ) rc = SQLITE_MISUSE;
332437f133ecSdan   }
332537f133ecSdan 
332637f133ecSdan   if( rc==SQLITE_OK ){
33270c698471Sdan     switch( res ){
33280c698471Sdan       case SQLITE_CHANGESET_REPLACE:
3329f51e5f6cSdan         assert( pbReplace );
3330f51e5f6cSdan         *pbReplace = 1;
33310c698471Sdan         break;
33320c698471Sdan 
33330c698471Sdan       case SQLITE_CHANGESET_OMIT:
33340c698471Sdan         break;
33350c698471Sdan 
33360c698471Sdan       case SQLITE_CHANGESET_ABORT:
33370c698471Sdan         rc = SQLITE_ABORT;
33380c698471Sdan         break;
33390c698471Sdan 
33400c698471Sdan       default:
33410c698471Sdan         rc = SQLITE_MISUSE;
33420c698471Sdan         break;
33430c698471Sdan     }
33440c698471Sdan   }
33450c698471Sdan 
33460c698471Sdan   return rc;
33470c698471Sdan }
33480c698471Sdan 
3349296c7658Sdan /*
3350296c7658Sdan ** Attempt to apply the change that the iterator passed as the first argument
3351296c7658Sdan ** currently points to to the database. If a conflict is encountered, invoke
3352296c7658Sdan ** the conflict handler callback.
3353296c7658Sdan **
3354296c7658Sdan ** If argument pbRetry is NULL, then ignore any CHANGESET_DATA conflict. If
3355296c7658Sdan ** one is encountered, update or delete the row with the matching primary key
3356296c7658Sdan ** instead. Or, if pbRetry is not NULL and a CHANGESET_DATA conflict occurs,
3357296c7658Sdan ** invoke the conflict handler. If it returns CHANGESET_REPLACE, set *pbRetry
3358296c7658Sdan ** to true before returning. In this case the caller will invoke this function
3359296c7658Sdan ** again, this time with pbRetry set to NULL.
3360296c7658Sdan **
3361296c7658Sdan ** If argument pbReplace is NULL and a CHANGESET_CONFLICT conflict is
3362296c7658Sdan ** encountered invoke the conflict handler with CHANGESET_CONSTRAINT instead.
3363296c7658Sdan ** Or, if pbReplace is not NULL, invoke it with CHANGESET_CONFLICT. If such
3364296c7658Sdan ** an invocation returns SQLITE_CHANGESET_REPLACE, set *pbReplace to true
3365296c7658Sdan ** before retrying. In this case the caller attempts to remove the conflicting
3366296c7658Sdan ** row before invoking this function again, this time with pbReplace set
3367296c7658Sdan ** to NULL.
3368296c7658Sdan **
3369296c7658Sdan ** If any conflict handler returns SQLITE_CHANGESET_ABORT, this function
3370296c7658Sdan ** returns SQLITE_ABORT. Otherwise, if no error occurs, SQLITE_OK is
3371296c7658Sdan ** returned.
3372296c7658Sdan */
33730c698471Sdan static int sessionApplyOneOp(
3374296c7658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset iterator */
3375296c7658Sdan   SessionApplyCtx *p,             /* changeset_apply() context */
33760c698471Sdan   int(*xConflict)(void *, int, sqlite3_changeset_iter *),
3377296c7658Sdan   void *pCtx,                     /* First argument for the conflict handler */
3378296c7658Sdan   int *pbReplace,                 /* OUT: True to remove PK row and retry */
3379296c7658Sdan   int *pbRetry                    /* OUT: True to retry. */
33800c698471Sdan ){
33810c698471Sdan   const char *zDummy;
33820c698471Sdan   int op;
33830c698471Sdan   int nCol;
33840c698471Sdan   int rc = SQLITE_OK;
33850c698471Sdan 
33860c698471Sdan   assert( p->pDelete && p->pUpdate && p->pInsert && p->pSelect );
33870c698471Sdan   assert( p->azCol && p->abPK );
33880c698471Sdan   assert( !pbReplace || *pbReplace==0 );
33890c698471Sdan 
3390b4480e94Sdan   sqlite3changeset_op(pIter, &zDummy, &nCol, &op, 0);
33910c698471Sdan 
33920c698471Sdan   if( op==SQLITE_DELETE ){
33930c698471Sdan 
339473b3c055Sdan     /* Bind values to the DELETE statement. If conflict handling is required,
339573b3c055Sdan     ** bind values for all columns and set bound variable (nCol+1) to true.
339673b3c055Sdan     ** Or, if conflict handling is not required, bind just the PK column
339773b3c055Sdan     ** values and, if it exists, set (nCol+1) to false. Conflict handling
339873b3c055Sdan     ** is not required if:
339973b3c055Sdan     **
340073b3c055Sdan     **   * this is a patchset, or
340173b3c055Sdan     **   * (pbRetry==0), or
340273b3c055Sdan     **   * all columns of the table are PK columns (in this case there is
340373b3c055Sdan     **     no (nCol+1) variable to bind to).
340473b3c055Sdan     */
340573b3c055Sdan     u8 *abPK = (pIter->bPatchset ? p->abPK : 0);
340673b3c055Sdan     rc = sessionBindRow(pIter, sqlite3changeset_old, nCol, abPK, p->pDelete);
34077cf7df7dSdan     if( rc==SQLITE_OK && sqlite3_bind_parameter_count(p->pDelete)>nCol ){
340873b3c055Sdan       rc = sqlite3_bind_int(p->pDelete, nCol+1, (pbRetry==0 || abPK));
34097cf7df7dSdan     }
34100c698471Sdan     if( rc!=SQLITE_OK ) return rc;
34110c698471Sdan 
34120c698471Sdan     sqlite3_step(p->pDelete);
34130c698471Sdan     rc = sqlite3_reset(p->pDelete);
34140c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
34150c698471Sdan       rc = sessionConflictHandler(
34160c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
34170c698471Sdan       );
341835e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
34190c698471Sdan       rc = sessionConflictHandler(
34200c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
34210c698471Sdan       );
34220c698471Sdan     }
34230c698471Sdan 
34240c698471Sdan   }else if( op==SQLITE_UPDATE ){
34250c698471Sdan     int i;
34260c698471Sdan 
34270c698471Sdan     /* Bind values to the UPDATE statement. */
34280c698471Sdan     for(i=0; rc==SQLITE_OK && i<nCol; i++){
34297aa469cdSdan       sqlite3_value *pOld = sessionChangesetOld(pIter, i);
34307aa469cdSdan       sqlite3_value *pNew = sessionChangesetNew(pIter, i);
34317aa469cdSdan 
34320c698471Sdan       sqlite3_bind_int(p->pUpdate, i*3+2, !!pNew);
34337aa469cdSdan       if( pOld ){
34347aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+1, pOld);
34357aa469cdSdan       }
34367aa469cdSdan       if( rc==SQLITE_OK && pNew ){
34377aa469cdSdan         rc = sessionBindValue(p->pUpdate, i*3+3, pNew);
34380c698471Sdan       }
34390c698471Sdan     }
344073b3c055Sdan     if( rc==SQLITE_OK ){
344173b3c055Sdan       sqlite3_bind_int(p->pUpdate, nCol*3+1, pbRetry==0 || pIter->bPatchset);
344273b3c055Sdan     }
34430c698471Sdan     if( rc!=SQLITE_OK ) return rc;
34440c698471Sdan 
34450c698471Sdan     /* Attempt the UPDATE. In the case of a NOTFOUND or DATA conflict,
34460c698471Sdan     ** the result will be SQLITE_OK with 0 rows modified. */
34470c698471Sdan     sqlite3_step(p->pUpdate);
34480c698471Sdan     rc = sqlite3_reset(p->pUpdate);
34490c698471Sdan 
34500c698471Sdan     if( rc==SQLITE_OK && sqlite3_changes(p->db)==0 ){
34510c698471Sdan       /* A NOTFOUND or DATA error. Search the table to see if it contains
34520c698471Sdan       ** a row with a matching primary key. If so, this is a DATA conflict.
34530c698471Sdan       ** Otherwise, if there is no primary key match, it is a NOTFOUND. */
34540c698471Sdan 
34550c698471Sdan       rc = sessionConflictHandler(
34560c698471Sdan           SQLITE_CHANGESET_DATA, p, pIter, xConflict, pCtx, pbRetry
34570c698471Sdan       );
34580c698471Sdan 
345935e2858eSdan     }else if( (rc&0xff)==SQLITE_CONSTRAINT ){
3460db04571cSdan       /* This is always a CONSTRAINT conflict. */
3461db04571cSdan       rc = sessionConflictHandler(
3462db04571cSdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, 0
34630c698471Sdan       );
34640c698471Sdan     }
34650c698471Sdan 
34660c698471Sdan   }else{
34670c698471Sdan     assert( op==SQLITE_INSERT );
34687aa469cdSdan     rc = sessionBindRow(pIter, sqlite3changeset_new, nCol, 0, p->pInsert);
34690c698471Sdan     if( rc!=SQLITE_OK ) return rc;
34700c698471Sdan 
34710c698471Sdan     sqlite3_step(p->pInsert);
34720c698471Sdan     rc = sqlite3_reset(p->pInsert);
347335e2858eSdan     if( (rc&0xff)==SQLITE_CONSTRAINT ){
34740c698471Sdan       rc = sessionConflictHandler(
34750c698471Sdan           SQLITE_CHANGESET_CONFLICT, p, pIter, xConflict, pCtx, pbReplace
34760c698471Sdan       );
347737f133ecSdan     }
347837f133ecSdan   }
347937f133ecSdan 
348037f133ecSdan   return rc;
348137f133ecSdan }
348237f133ecSdan 
3483296c7658Sdan /*
34844757c658Sdan ** Argument pIter is a changeset iterator that has been initialized, but
34854757c658Sdan ** not yet passed to sqlite3changeset_next(). This function applies the
34864757c658Sdan ** changeset to the main database attached to handle "db". The supplied
34874757c658Sdan ** conflict handler callback is invoked to resolve any conflicts encountered
34884757c658Sdan ** while applying the change.
3489296c7658Sdan */
34904757c658Sdan static int sessionChangesetApply(
3491296c7658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
34924757c658Sdan   sqlite3_changeset_iter *pIter,  /* Changeset to apply */
349340368988Sdan   int(*xFilter)(
349440368988Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
349540368988Sdan     const char *zTab              /* Table name */
349640368988Sdan   ),
3497d5f0767cSdan   int(*xConflict)(
3498d5f0767cSdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
3499d5f0767cSdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
3500d5f0767cSdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
3501d5f0767cSdan   ),
3502296c7658Sdan   void *pCtx                      /* First argument passed to xConflict */
3503d5f0767cSdan ){
3504ca62ad57Sdan   int schemaMismatch = 0;
3505296c7658Sdan   int rc;                         /* Return code */
3506d5f0767cSdan   const char *zTab = 0;           /* Name of current table */
3507cfdbde21Sdrh   int nTab = 0;                   /* Result of sqlite3Strlen30(zTab) */
3508296c7658Sdan   SessionApplyCtx sApply;         /* changeset_apply() context object */
3509d5f0767cSdan 
3510082c96dfSdan   assert( xConflict!=0 );
3511082c96dfSdan 
35120c698471Sdan   memset(&sApply, 0, sizeof(sApply));
35134c220252Sdan   sqlite3_mutex_enter(sqlite3_db_mutex(db));
35140c698471Sdan   rc = sqlite3_exec(db, "SAVEPOINT changeset_apply", 0, 0, 0);
3515cb3e4b79Sdan   if( rc==SQLITE_OK ){
3516cb3e4b79Sdan     rc = sqlite3_exec(db, "PRAGMA defer_foreign_keys = 1", 0, 0, 0);
3517cb3e4b79Sdan   }
35180c698471Sdan   while( rc==SQLITE_OK && SQLITE_ROW==sqlite3changeset_next(pIter) ){
35190c698471Sdan     int nCol;
3520d5f0767cSdan     int op;
35210c698471Sdan     int bReplace = 0;
35220c698471Sdan     int bRetry = 0;
35230c698471Sdan     const char *zNew;
3524ca62ad57Sdan 
3525b4480e94Sdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, 0);
3526d5f0767cSdan 
35270c698471Sdan     if( zTab==0 || sqlite3_strnicmp(zNew, zTab, nTab+1) ){
3528ca62ad57Sdan       u8 *abPK;
3529ca62ad57Sdan 
3530cfdbde21Sdrh       sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
35310c698471Sdan       sqlite3_finalize(sApply.pDelete);
35320c698471Sdan       sqlite3_finalize(sApply.pUpdate);
35330c698471Sdan       sqlite3_finalize(sApply.pInsert);
35340c698471Sdan       sqlite3_finalize(sApply.pSelect);
35350c698471Sdan       memset(&sApply, 0, sizeof(sApply));
35360c698471Sdan       sApply.db = db;
353737f133ecSdan 
353840368988Sdan       /* If an xFilter() callback was specified, invoke it now. If the
353940368988Sdan       ** xFilter callback returns zero, skip this table. If it returns
354040368988Sdan       ** non-zero, proceed. */
354140368988Sdan       schemaMismatch = (xFilter && (0==xFilter(pCtx, zNew)));
354240368988Sdan       if( schemaMismatch ){
354340368988Sdan         zTab = sqlite3_mprintf("%s", zNew);
3544f05ac112Sdan         if( zTab==0 ){
3545f05ac112Sdan           rc = SQLITE_NOMEM;
3546f05ac112Sdan           break;
3547f05ac112Sdan         }
35484f528042Sdan         nTab = (int)strlen(zTab);
354940368988Sdan         sApply.azCol = (const char **)zTab;
355040368988Sdan       }else{
3551ca62ad57Sdan         sqlite3changeset_pk(pIter, &abPK, 0);
3552296c7658Sdan         rc = sessionTableInfo(
3553ca62ad57Sdan             db, "main", zNew, &sApply.nCol, &zTab, &sApply.azCol, &sApply.abPK
3554ca62ad57Sdan         );
3555ca62ad57Sdan         if( rc!=SQLITE_OK ) break;
35560c698471Sdan 
3557ca62ad57Sdan         if( sApply.nCol==0 ){
3558ca62ad57Sdan           schemaMismatch = 1;
3559ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
3560ca62ad57Sdan               "sqlite3changeset_apply(): no such table: %s", zTab
3561ca62ad57Sdan           );
3562ca62ad57Sdan         }
3563ca62ad57Sdan         else if( sApply.nCol!=nCol ){
3564ca62ad57Sdan           schemaMismatch = 1;
3565ca62ad57Sdan           sqlite3_log(SQLITE_SCHEMA,
3566ca62ad57Sdan               "sqlite3changeset_apply(): table %s has %d columns, expected %d",
3567ca62ad57Sdan               zTab, sApply.nCol, nCol
3568ca62ad57Sdan           );
3569ca62ad57Sdan         }
3570ca62ad57Sdan         else if( memcmp(sApply.abPK, abPK, nCol)!=0 ){
3571ca62ad57Sdan           schemaMismatch = 1;
357240368988Sdan           sqlite3_log(SQLITE_SCHEMA, "sqlite3changeset_apply(): "
357340368988Sdan               "primary key mismatch for table %s", zTab
3574ca62ad57Sdan           );
3575ca62ad57Sdan         }
3576ca62ad57Sdan         else if(
3577ca62ad57Sdan             (rc = sessionSelectRow(db, zTab, &sApply))
35780c698471Sdan          || (rc = sessionUpdateRow(db, zTab, &sApply))
35790c698471Sdan          || (rc = sessionDeleteRow(db, zTab, &sApply))
35800c698471Sdan          || (rc = sessionInsertRow(db, zTab, &sApply))
358137f133ecSdan         ){
358237f133ecSdan           break;
358337f133ecSdan         }
3584cfdbde21Sdrh         nTab = sqlite3Strlen30(zTab);
3585d5f0767cSdan       }
358640368988Sdan     }
3587d5f0767cSdan 
3588ca62ad57Sdan     /* If there is a schema mismatch on the current table, proceed to the
3589ca62ad57Sdan     ** next change. A log message has already been issued. */
3590ca62ad57Sdan     if( schemaMismatch ) continue;
3591ca62ad57Sdan 
35920c698471Sdan     rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, &bRetry);
35930c698471Sdan 
35940c698471Sdan     if( rc==SQLITE_OK && bRetry ){
35950c698471Sdan       rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, &bReplace, 0);
35960c698471Sdan     }
35970c698471Sdan 
35980c698471Sdan     if( bReplace ){
3599db04571cSdan       assert( pIter->op==SQLITE_INSERT );
36000c698471Sdan       rc = sqlite3_exec(db, "SAVEPOINT replace_op", 0, 0, 0);
36010c698471Sdan       if( rc==SQLITE_OK ){
36027aa469cdSdan         rc = sessionBindRow(pIter,
3603db04571cSdan             sqlite3changeset_new, sApply.nCol, sApply.abPK, sApply.pDelete);
36040c698471Sdan         sqlite3_bind_int(sApply.pDelete, sApply.nCol+1, 1);
36050c698471Sdan       }
36060c698471Sdan       if( rc==SQLITE_OK ){
36070c698471Sdan         sqlite3_step(sApply.pDelete);
36080c698471Sdan         rc = sqlite3_reset(sApply.pDelete);
36090c698471Sdan       }
36100c698471Sdan       if( rc==SQLITE_OK ){
36110c698471Sdan         rc = sessionApplyOneOp(pIter, &sApply, xConflict, pCtx, 0, 0);
36120c698471Sdan       }
36130c698471Sdan       if( rc==SQLITE_OK ){
36140c698471Sdan         rc = sqlite3_exec(db, "RELEASE replace_op", 0, 0, 0);
36150c698471Sdan       }
36160c698471Sdan     }
36170c698471Sdan   }
3618d5f0767cSdan 
3619296c7658Sdan   if( rc==SQLITE_OK ){
3620296c7658Sdan     rc = sqlite3changeset_finalize(pIter);
3621296c7658Sdan   }else{
3622296c7658Sdan     sqlite3changeset_finalize(pIter);
3623296c7658Sdan   }
3624d5f0767cSdan 
3625d5f0767cSdan   if( rc==SQLITE_OK ){
362607001c45Sdrh     int nFk, notUsed;
362707001c45Sdrh     sqlite3_db_status(db, SQLITE_DBSTATUS_DEFERRED_FKS, &nFk, &notUsed, 0);
362807001c45Sdrh     if( nFk!=0 ){
3629cb3e4b79Sdan       int res = SQLITE_CHANGESET_ABORT;
3630cb3e4b79Sdan       sqlite3_changeset_iter sIter;
3631cb3e4b79Sdan       memset(&sIter, 0, sizeof(sIter));
3632cb3e4b79Sdan       sIter.nCol = nFk;
3633cb3e4b79Sdan       res = xConflict(pCtx, SQLITE_CHANGESET_FOREIGN_KEY, &sIter);
3634cb3e4b79Sdan       if( res!=SQLITE_CHANGESET_OMIT ){
3635cb3e4b79Sdan         rc = SQLITE_CONSTRAINT;
3636cb3e4b79Sdan       }
3637cb3e4b79Sdan     }
3638cb3e4b79Sdan   }
3639cb3e4b79Sdan   sqlite3_exec(db, "PRAGMA defer_foreign_keys = 0", 0, 0, 0);
3640cb3e4b79Sdan 
3641cb3e4b79Sdan   if( rc==SQLITE_OK ){
3642d5f0767cSdan     rc = sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
3643d5f0767cSdan   }else{
3644d5f0767cSdan     sqlite3_exec(db, "ROLLBACK TO changeset_apply", 0, 0, 0);
3645d5f0767cSdan     sqlite3_exec(db, "RELEASE changeset_apply", 0, 0, 0);
3646d5f0767cSdan   }
3647d5f0767cSdan 
36480c698471Sdan   sqlite3_finalize(sApply.pInsert);
36490c698471Sdan   sqlite3_finalize(sApply.pDelete);
36500c698471Sdan   sqlite3_finalize(sApply.pUpdate);
36510c698471Sdan   sqlite3_finalize(sApply.pSelect);
3652cfdbde21Sdrh   sqlite3_free((char*)sApply.azCol);  /* cast works around VC++ bug */
36534c220252Sdan   sqlite3_mutex_leave(sqlite3_db_mutex(db));
3654d5f0767cSdan   return rc;
3655d5f0767cSdan }
365691ddd559Sdan 
365777fc1d5bSdan /*
36584757c658Sdan ** Apply the changeset passed via pChangeset/nChangeset to the main database
36594757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
36604757c658Sdan ** to resolve any conflicts encountered while applying the change.
36614757c658Sdan */
36624757c658Sdan int sqlite3changeset_apply(
36634757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
36644757c658Sdan   int nChangeset,                 /* Size of changeset in bytes */
36654757c658Sdan   void *pChangeset,               /* Changeset blob */
36664757c658Sdan   int(*xFilter)(
36674757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
36684757c658Sdan     const char *zTab              /* Table name */
36694757c658Sdan   ),
36704757c658Sdan   int(*xConflict)(
36714757c658Sdan     void *pCtx,                   /* Copy of fifth arg to _apply() */
36724757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
36734757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
36744757c658Sdan   ),
36754757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
36764757c658Sdan ){
36774757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
36784757c658Sdan   int rc = sqlite3changeset_start(&pIter, nChangeset, pChangeset);
36794757c658Sdan   if( rc==SQLITE_OK ){
36804757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
36814757c658Sdan   }
36824757c658Sdan   return rc;
36834757c658Sdan }
36844757c658Sdan 
36854757c658Sdan /*
36864757c658Sdan ** Apply the changeset passed via xInput/pIn to the main database
36874757c658Sdan ** attached to handle "db". Invoke the supplied conflict handler callback
36884757c658Sdan ** to resolve any conflicts encountered while applying the change.
36894757c658Sdan */
3690f1a08ad8Sdrh int sqlite3changeset_apply_strm(
36914757c658Sdan   sqlite3 *db,                    /* Apply change to "main" db of this handle */
36924757c658Sdan   int (*xInput)(void *pIn, void *pData, int *pnData), /* Input function */
36934757c658Sdan   void *pIn,                                          /* First arg for xInput */
36944757c658Sdan   int(*xFilter)(
36954757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
36964757c658Sdan     const char *zTab              /* Table name */
36974757c658Sdan   ),
36984757c658Sdan   int(*xConflict)(
36994757c658Sdan     void *pCtx,                   /* Copy of sixth arg to _apply() */
37004757c658Sdan     int eConflict,                /* DATA, MISSING, CONFLICT, CONSTRAINT */
37014757c658Sdan     sqlite3_changeset_iter *p     /* Handle describing change and conflict */
37024757c658Sdan   ),
37034757c658Sdan   void *pCtx                      /* First argument passed to xConflict */
37044757c658Sdan ){
37054757c658Sdan   sqlite3_changeset_iter *pIter;  /* Iterator to skip through changeset */
3706f1a08ad8Sdrh   int rc = sqlite3changeset_start_strm(&pIter, xInput, pIn);
37074757c658Sdan   if( rc==SQLITE_OK ){
37084757c658Sdan     rc = sessionChangesetApply(db, pIter, xFilter, xConflict, pCtx);
37094757c658Sdan   }
37104757c658Sdan   return rc;
37114757c658Sdan }
37124757c658Sdan 
37134757c658Sdan /*
371477fc1d5bSdan ** This function is called to merge two changes to the same row together as
371577fc1d5bSdan ** part of an sqlite3changeset_concat() operation. A new change object is
371677fc1d5bSdan ** allocated and a pointer to it stored in *ppNew.
371777fc1d5bSdan */
37185d607a6eSdan static int sessionChangeMerge(
371977fc1d5bSdan   SessionTable *pTab,             /* Table structure */
372064277f4aSdan   int bPatchset,                  /* True for patchsets */
372177fc1d5bSdan   SessionChange *pExist,          /* Existing change */
372277fc1d5bSdan   int op2,                        /* Second change operation */
372377fc1d5bSdan   int bIndirect,                  /* True if second change is indirect */
372477fc1d5bSdan   u8 *aRec,                       /* Second change record */
372577fc1d5bSdan   int nRec,                       /* Number of bytes in aRec */
372677fc1d5bSdan   SessionChange **ppNew           /* OUT: Merged change */
37275d607a6eSdan ){
37285d607a6eSdan   SessionChange *pNew = 0;
37295d607a6eSdan 
37305d607a6eSdan   if( !pExist ){
3731cbf6d2d2Sdan     pNew = (SessionChange *)sqlite3_malloc(sizeof(SessionChange) + nRec);
37325d607a6eSdan     if( !pNew ){
37335d607a6eSdan       return SQLITE_NOMEM;
37345d607a6eSdan     }
37355d607a6eSdan     memset(pNew, 0, sizeof(SessionChange));
3736798693b2Sdan     pNew->op = op2;
37375d607a6eSdan     pNew->bIndirect = bIndirect;
37385d607a6eSdan     pNew->nRecord = nRec;
3739cbf6d2d2Sdan     pNew->aRecord = (u8*)&pNew[1];
3740cbf6d2d2Sdan     memcpy(pNew->aRecord, aRec, nRec);
37415d607a6eSdan   }else{
3742798693b2Sdan     int op1 = pExist->op;
37435d607a6eSdan 
37445d607a6eSdan     /*
37455d607a6eSdan     **   op1=INSERT, op2=INSERT      ->      Unsupported. Discard op2.
37465d607a6eSdan     **   op1=INSERT, op2=UPDATE      ->      INSERT.
37475d607a6eSdan     **   op1=INSERT, op2=DELETE      ->      (none)
37485d607a6eSdan     **
37495d607a6eSdan     **   op1=UPDATE, op2=INSERT      ->      Unsupported. Discard op2.
37505d607a6eSdan     **   op1=UPDATE, op2=UPDATE      ->      UPDATE.
37515d607a6eSdan     **   op1=UPDATE, op2=DELETE      ->      DELETE.
37525d607a6eSdan     **
37535d607a6eSdan     **   op1=DELETE, op2=INSERT      ->      UPDATE.
37545d607a6eSdan     **   op1=DELETE, op2=UPDATE      ->      Unsupported. Discard op2.
37555d607a6eSdan     **   op1=DELETE, op2=DELETE      ->      Unsupported. Discard op2.
37565d607a6eSdan     */
37575d607a6eSdan     if( (op1==SQLITE_INSERT && op2==SQLITE_INSERT)
37585d607a6eSdan      || (op1==SQLITE_UPDATE && op2==SQLITE_INSERT)
37595d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_UPDATE)
37605d607a6eSdan      || (op1==SQLITE_DELETE && op2==SQLITE_DELETE)
37615d607a6eSdan     ){
37625d607a6eSdan       pNew = pExist;
37635d607a6eSdan     }else if( op1==SQLITE_INSERT && op2==SQLITE_DELETE ){
37645d607a6eSdan       sqlite3_free(pExist);
37655d607a6eSdan       assert( pNew==0 );
37665d607a6eSdan     }else{
376764277f4aSdan       u8 *aExist = pExist->aRecord;
37685d607a6eSdan       int nByte;
37695d607a6eSdan       u8 *aCsr;
37705d607a6eSdan 
377164277f4aSdan       /* Allocate a new SessionChange object. Ensure that the aRecord[]
377264277f4aSdan       ** buffer of the new object is large enough to hold any record that
377364277f4aSdan       ** may be generated by combining the input records.  */
37745d607a6eSdan       nByte = sizeof(SessionChange) + pExist->nRecord + nRec;
37755d607a6eSdan       pNew = (SessionChange *)sqlite3_malloc(nByte);
37765d607a6eSdan       if( !pNew ){
37771756ae10Sdan         sqlite3_free(pExist);
37785d607a6eSdan         return SQLITE_NOMEM;
37795d607a6eSdan       }
37805d607a6eSdan       memset(pNew, 0, sizeof(SessionChange));
37815d607a6eSdan       pNew->bIndirect = (bIndirect && pExist->bIndirect);
37825d607a6eSdan       aCsr = pNew->aRecord = (u8 *)&pNew[1];
37835d607a6eSdan 
3784b08a1efaSdan       if( op1==SQLITE_INSERT ){             /* INSERT + UPDATE */
37855d607a6eSdan         u8 *a1 = aRec;
3786b08a1efaSdan         assert( op2==SQLITE_UPDATE );
3787798693b2Sdan         pNew->op = SQLITE_INSERT;
3788ef7a6304Sdan         if( bPatchset==0 ) sessionSkipRecord(&a1, pTab->nCol);
378964277f4aSdan         sessionMergeRecord(&aCsr, pTab->nCol, aExist, a1);
3790b08a1efaSdan       }else if( op1==SQLITE_DELETE ){       /* DELETE + INSERT */
3791b08a1efaSdan         assert( op2==SQLITE_INSERT );
3792798693b2Sdan         pNew->op = SQLITE_UPDATE;
3793fa29ecc4Sdan         if( bPatchset ){
3794fa29ecc4Sdan           memcpy(aCsr, aRec, nRec);
3795fa29ecc4Sdan           aCsr += nRec;
3796fa29ecc4Sdan         }else{
379764277f4aSdan           if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aExist, 0,aRec,0) ){
3798b08a1efaSdan             sqlite3_free(pNew);
3799b08a1efaSdan             pNew = 0;
38005d607a6eSdan           }
3801fa29ecc4Sdan         }
3802b08a1efaSdan       }else if( op2==SQLITE_UPDATE ){       /* UPDATE + UPDATE */
380364277f4aSdan         u8 *a1 = aExist;
38045d607a6eSdan         u8 *a2 = aRec;
3805cfec7eeeSdan         assert( op1==SQLITE_UPDATE );
380664277f4aSdan         if( bPatchset==0 ){
3807ef7a6304Sdan           sessionSkipRecord(&a1, pTab->nCol);
3808ef7a6304Sdan           sessionSkipRecord(&a2, pTab->nCol);
380964277f4aSdan         }
3810798693b2Sdan         pNew->op = SQLITE_UPDATE;
381164277f4aSdan         if( 0==sessionMergeUpdate(&aCsr, pTab, bPatchset, aRec, aExist,a1,a2) ){
38125d607a6eSdan           sqlite3_free(pNew);
38135d607a6eSdan           pNew = 0;
38145d607a6eSdan         }
3815b08a1efaSdan       }else{                                /* UPDATE + DELETE */
3816b08a1efaSdan         assert( op1==SQLITE_UPDATE && op2==SQLITE_DELETE );
3817798693b2Sdan         pNew->op = SQLITE_DELETE;
381864277f4aSdan         if( bPatchset ){
381964277f4aSdan           memcpy(aCsr, aRec, nRec);
382064277f4aSdan           aCsr += nRec;
382164277f4aSdan         }else{
382264277f4aSdan           sessionMergeRecord(&aCsr, pTab->nCol, aRec, aExist);
382364277f4aSdan         }
38245d607a6eSdan       }
38255d607a6eSdan 
38265d607a6eSdan       if( pNew ){
38274f528042Sdan         pNew->nRecord = (int)(aCsr - pNew->aRecord);
38285d607a6eSdan       }
38295d607a6eSdan       sqlite3_free(pExist);
38305d607a6eSdan     }
38315d607a6eSdan   }
38325d607a6eSdan 
38335d607a6eSdan   *ppNew = pNew;
38345d607a6eSdan   return SQLITE_OK;
38355d607a6eSdan }
38365d607a6eSdan 
383777fc1d5bSdan /*
383877fc1d5bSdan ** Add all changes in the changeset passed via the first two arguments to
383977fc1d5bSdan ** hash tables.
384077fc1d5bSdan */
384116228167Sdan static int sessionChangesetToHash(
3842cbf6d2d2Sdan   sqlite3_changeset_iter *pIter,   /* Iterator to read from */
384377fc1d5bSdan   SessionTable **ppTabList        /* IN/OUT: List of table objects */
38445d607a6eSdan ){
38455d607a6eSdan   u8 *aRec;
38465d607a6eSdan   int nRec;
3847cbf6d2d2Sdan   int rc = SQLITE_OK;
38485d607a6eSdan   SessionTable *pTab = 0;
38495d607a6eSdan 
38505d607a6eSdan   while( SQLITE_ROW==sessionChangesetNext(pIter, &aRec, &nRec) ){
38515d607a6eSdan     const char *zNew;
38525d607a6eSdan     int nCol;
38535d607a6eSdan     int op;
38545d607a6eSdan     int iHash;
38555d607a6eSdan     int bIndirect;
38565d607a6eSdan     SessionChange *pChange;
38575d607a6eSdan     SessionChange *pExist = 0;
38585d607a6eSdan     SessionChange **pp;
38595d607a6eSdan 
38605d607a6eSdan     sqlite3changeset_op(pIter, &zNew, &nCol, &op, &bIndirect);
3861ef7a6304Sdan     if( !pTab || sqlite3_stricmp(zNew, pTab->zName) ){
38625d607a6eSdan       /* Search the list for a matching table */
38634f528042Sdan       int nNew = (int)strlen(zNew);
3864f29123b5Sdan       u8 *abPK;
3865f29123b5Sdan 
3866f29123b5Sdan       sqlite3changeset_pk(pIter, &abPK, 0);
38675d607a6eSdan       for(pTab = *ppTabList; pTab; pTab=pTab->pNext){
38685d607a6eSdan         if( 0==sqlite3_strnicmp(pTab->zName, zNew, nNew+1) ) break;
38695d607a6eSdan       }
38705d607a6eSdan       if( !pTab ){
3871ef7a6304Sdan         pTab = sqlite3_malloc(sizeof(SessionTable) + nCol + nNew+1);
3872f29123b5Sdan         if( !pTab ){
3873f29123b5Sdan           rc = SQLITE_NOMEM;
3874f29123b5Sdan           break;
3875f29123b5Sdan         }
38765d607a6eSdan         memset(pTab, 0, sizeof(SessionTable));
38775d607a6eSdan         pTab->pNext = *ppTabList;
3878f29123b5Sdan         pTab->nCol = nCol;
3879ef7a6304Sdan         pTab->abPK = (u8*)&pTab[1];
3880ef7a6304Sdan         memcpy(pTab->abPK, abPK, nCol);
3881ef7a6304Sdan         pTab->zName = (char*)&pTab->abPK[nCol];
3882ef7a6304Sdan         memcpy(pTab->zName, zNew, nNew+1);
38835d607a6eSdan         *ppTabList = pTab;
3884f29123b5Sdan       }else if( pTab->nCol!=nCol || memcmp(pTab->abPK, abPK, nCol) ){
3885f29123b5Sdan         rc = SQLITE_SCHEMA;
3886f29123b5Sdan         break;
38875d607a6eSdan       }
38885d607a6eSdan     }
38895d607a6eSdan 
3890cbf6d2d2Sdan     if( sessionGrowHash(pIter->bPatchset, pTab) ){
38911756ae10Sdan       rc = SQLITE_NOMEM;
38921756ae10Sdan       break;
38931756ae10Sdan     }
389464277f4aSdan     iHash = sessionChangeHash(
3895cbf6d2d2Sdan         pTab, (pIter->bPatchset && op==SQLITE_DELETE), aRec, pTab->nChange
389664277f4aSdan     );
38975d607a6eSdan 
38985d607a6eSdan     /* Search for existing entry. If found, remove it from the hash table.
38995d607a6eSdan     ** Code below may link it back in.
39005d607a6eSdan     */
39015d607a6eSdan     for(pp=&pTab->apChange[iHash]; *pp; pp=&(*pp)->pNext){
390264277f4aSdan       int bPkOnly1 = 0;
390364277f4aSdan       int bPkOnly2 = 0;
3904cbf6d2d2Sdan       if( pIter->bPatchset ){
390564277f4aSdan         bPkOnly1 = (*pp)->op==SQLITE_DELETE;
390664277f4aSdan         bPkOnly2 = op==SQLITE_DELETE;
390764277f4aSdan       }
390864277f4aSdan       if( sessionChangeEqual(pTab, bPkOnly1, (*pp)->aRecord, bPkOnly2, aRec) ){
39095d607a6eSdan         pExist = *pp;
39105d607a6eSdan         *pp = (*pp)->pNext;
39115d607a6eSdan         pTab->nEntry--;
39125d607a6eSdan         break;
39135d607a6eSdan       }
39145d607a6eSdan     }
39155d607a6eSdan 
391664277f4aSdan     rc = sessionChangeMerge(pTab,
3917cbf6d2d2Sdan         pIter->bPatchset, pExist, op, bIndirect, aRec, nRec, &pChange
391864277f4aSdan     );
39195d607a6eSdan     if( rc ) break;
39205d607a6eSdan     if( pChange ){
39215d607a6eSdan       pChange->pNext = pTab->apChange[iHash];
39225d607a6eSdan       pTab->apChange[iHash] = pChange;
39235d607a6eSdan       pTab->nEntry++;
39245d607a6eSdan     }
39255d607a6eSdan   }
39265d607a6eSdan 
3927cbf6d2d2Sdan   if( rc==SQLITE_OK ) rc = pIter->rc;
39285d607a6eSdan   return rc;
39295d607a6eSdan }
39305d607a6eSdan 
39315d607a6eSdan /*
39325d607a6eSdan ** 1. Iterate through the left-hand changeset. Add an entry to a table
39335d607a6eSdan **    specific hash table for each change in the changeset. The hash table
39345d607a6eSdan **    key is the PK of the row affected by the change.
39355d607a6eSdan **
39365d607a6eSdan ** 2. Then interate through the right-hand changeset. Attempt to add an
39375d607a6eSdan **    entry to a hash table for each component change. If a change already
39385d607a6eSdan **    exists with the same PK values, combine the two into a single change.
39395d607a6eSdan **
39405d607a6eSdan ** 3. Write an output changeset based on the contents of the hash table.
39415d607a6eSdan */
3942cbf6d2d2Sdan int sessionChangesetConcat(
3943cbf6d2d2Sdan   sqlite3_changeset_iter *pLeft,
3944cbf6d2d2Sdan   sqlite3_changeset_iter *pRight,
3945cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
3946cbf6d2d2Sdan   void *pOut,
3947cbf6d2d2Sdan   int *pnOut,
3948cbf6d2d2Sdan   void **ppOut
39495d607a6eSdan ){
39505d607a6eSdan   SessionTable *pList = 0;        /* List of SessionTable objects */
39515d607a6eSdan   int rc;                         /* Return code */
395264277f4aSdan   int bPatch;                     /* True for a patchset */
3953e8fa8c96Sdan   SessionTable *pTab;
3954e8fa8c96Sdan   SessionBuffer buf = {0, 0, 0};
39555d607a6eSdan 
3956cbf6d2d2Sdan   assert( xOutput==0 || (ppOut==0 && pnOut==0) );
39575d607a6eSdan 
3958abae0c4fSdan   assert( pLeft->zTab==0 && pRight->zTab==0 );
395916228167Sdan   rc = sessionChangesetToHash(pLeft, &pList);
3960abae0c4fSdan   assert( pLeft->zTab || pList==0 );
39615d607a6eSdan   if( rc==SQLITE_OK ){
396216228167Sdan     rc = sessionChangesetToHash(pRight, &pList);
39635d607a6eSdan   }
3964cbf6d2d2Sdan   bPatch = pLeft->bPatchset || pRight->bPatchset;
39655d607a6eSdan 
3966abae0c4fSdan   if( pLeft->zTab && pRight->zTab && pLeft->bPatchset!=pRight->bPatchset ){
3967abae0c4fSdan     rc = SQLITE_ERROR;
3968abae0c4fSdan   }
3969abae0c4fSdan 
39705d607a6eSdan   /* Create the serialized output changeset based on the contents of the
39715d607a6eSdan   ** hash tables attached to the SessionTable objects in list pList.
39725d607a6eSdan   */
3973e8fa8c96Sdan   for(pTab=pList; rc==SQLITE_OK && pTab; pTab=pTab->pNext){
39745d607a6eSdan     int i;
39755d607a6eSdan     if( pTab->nEntry==0 ) continue;
39765d607a6eSdan 
397764277f4aSdan     sessionAppendTableHdr(&buf, bPatch, pTab, &rc);
39785d607a6eSdan     for(i=0; i<pTab->nChange; i++){
39795d607a6eSdan       SessionChange *p;
39805d607a6eSdan       for(p=pTab->apChange[i]; p; p=p->pNext){
3981798693b2Sdan         sessionAppendByte(&buf, p->op, &rc);
39825d607a6eSdan         sessionAppendByte(&buf, p->bIndirect, &rc);
39835d607a6eSdan         sessionAppendBlob(&buf, p->aRecord, p->nRecord, &rc);
39845d607a6eSdan       }
39855d607a6eSdan     }
3986cbf6d2d2Sdan 
3987f1a08ad8Sdrh     if( rc==SQLITE_OK && xOutput && buf.nBuf>=SESSIONS_STRM_CHUNK_SIZE ){
3988cbf6d2d2Sdan       rc = xOutput(pOut, buf.aBuf, buf.nBuf);
3989cbf6d2d2Sdan       buf.nBuf = 0;
3990cbf6d2d2Sdan     }
39915d607a6eSdan   }
39925d607a6eSdan 
39935d607a6eSdan   if( rc==SQLITE_OK ){
3994cbf6d2d2Sdan     if( xOutput ){
3995cbf6d2d2Sdan       if( buf.nBuf>0 ) rc = xOutput(pOut, buf.aBuf, buf.nBuf);
3996cbf6d2d2Sdan     }else{
39975d607a6eSdan       *ppOut = buf.aBuf;
39985d607a6eSdan       *pnOut = buf.nBuf;
3999cbf6d2d2Sdan       buf.aBuf = 0;
40005d607a6eSdan     }
40015d607a6eSdan   }
4002cbf6d2d2Sdan   sqlite3_free(buf.aBuf);
40035d607a6eSdan 
40045d607a6eSdan   sessionDeleteTable(pList);
40055d607a6eSdan   return rc;
40065d607a6eSdan }
40075d607a6eSdan 
4008cbf6d2d2Sdan /*
4009cbf6d2d2Sdan ** Combine two changesets together.
4010cbf6d2d2Sdan */
4011cbf6d2d2Sdan int sqlite3changeset_concat(
4012cbf6d2d2Sdan   int nLeft,                      /* Number of bytes in lhs input */
4013cbf6d2d2Sdan   void *pLeft,                    /* Lhs input changeset */
4014cbf6d2d2Sdan   int nRight                      /* Number of bytes in rhs input */,
4015cbf6d2d2Sdan   void *pRight,                   /* Rhs input changeset */
4016cbf6d2d2Sdan   int *pnOut,                     /* OUT: Number of bytes in output changeset */
4017cbf6d2d2Sdan   void **ppOut                    /* OUT: changeset (left <concat> right) */
4018cbf6d2d2Sdan ){
4019cbf6d2d2Sdan   sqlite3_changeset_iter *pIter1 = 0;
4020cbf6d2d2Sdan   sqlite3_changeset_iter *pIter2 = 0;
4021cbf6d2d2Sdan   int rc;
4022cbf6d2d2Sdan 
4023cbf6d2d2Sdan   *pnOut = 0;
4024cbf6d2d2Sdan   *ppOut = 0;
4025cbf6d2d2Sdan   rc = sqlite3changeset_start(&pIter1, nLeft, pLeft);
4026cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4027cbf6d2d2Sdan     rc = sqlite3changeset_start(&pIter2, nRight, pRight);
4028cbf6d2d2Sdan   }
4029cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4030cbf6d2d2Sdan     rc = sessionChangesetConcat(pIter1, pIter2, 0, 0, pnOut, ppOut);
4031cbf6d2d2Sdan   }
4032cbf6d2d2Sdan 
4033cbf6d2d2Sdan   sqlite3changeset_finalize(pIter1);
4034cbf6d2d2Sdan   sqlite3changeset_finalize(pIter2);
4035cbf6d2d2Sdan   return rc;
4036cbf6d2d2Sdan }
4037cbf6d2d2Sdan 
4038cbf6d2d2Sdan /*
4039cbf6d2d2Sdan ** Streaming version of sqlite3changeset_concat().
4040cbf6d2d2Sdan */
4041f1a08ad8Sdrh int sqlite3changeset_concat_strm(
4042cbf6d2d2Sdan   int (*xInputA)(void *pIn, void *pData, int *pnData),
4043cbf6d2d2Sdan   void *pInA,
4044cbf6d2d2Sdan   int (*xInputB)(void *pIn, void *pData, int *pnData),
4045cbf6d2d2Sdan   void *pInB,
4046cbf6d2d2Sdan   int (*xOutput)(void *pOut, const void *pData, int nData),
4047cbf6d2d2Sdan   void *pOut
4048cbf6d2d2Sdan ){
4049cbf6d2d2Sdan   sqlite3_changeset_iter *pIter1 = 0;
4050cbf6d2d2Sdan   sqlite3_changeset_iter *pIter2 = 0;
4051cbf6d2d2Sdan   int rc;
4052cbf6d2d2Sdan 
4053f1a08ad8Sdrh   rc = sqlite3changeset_start_strm(&pIter1, xInputA, pInA);
4054cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4055f1a08ad8Sdrh     rc = sqlite3changeset_start_strm(&pIter2, xInputB, pInB);
4056cbf6d2d2Sdan   }
4057cbf6d2d2Sdan   if( rc==SQLITE_OK ){
4058cbf6d2d2Sdan     rc = sessionChangesetConcat(pIter1, pIter2, xOutput, pOut, 0, 0);
4059cbf6d2d2Sdan   }
4060cbf6d2d2Sdan 
4061cbf6d2d2Sdan   sqlite3changeset_finalize(pIter1);
4062cbf6d2d2Sdan   sqlite3changeset_finalize(pIter2);
4063cbf6d2d2Sdan   return rc;
4064cbf6d2d2Sdan }
4065cbf6d2d2Sdan 
40669b1c62d4Sdrh #endif /* SQLITE_ENABLE_SESSION && SQLITE_ENABLE_PREUPDATE_HOOK */
4067