xref: /sqlite-3.40.0/src/vdbesort.c (revision 2e27d28f)
1 /*
2 ** 2011-07-09
3 **
4 ** The author disclaims copyright to this source code.  In place of
5 ** a legal notice, here is a blessing:
6 **
7 **    May you do good and not evil.
8 **    May you find forgiveness for yourself and forgive others.
9 **    May you share freely, never taking more than you give.
10 **
11 *************************************************************************
12 ** This file contains code for the VdbeSorter object, used in concert with
13 ** a VdbeCursor to sort large numbers of keys for CREATE INDEX statements
14 ** or by SELECT statements with ORDER BY clauses that cannot be satisfied
15 ** using indexes and without LIMIT clauses.
16 **
17 ** The VdbeSorter object implements a multi-threaded external merge sort
18 ** algorithm that is efficient even if the number of elements being sorted
19 ** exceeds the available memory.
20 **
21 ** Here is the (internal, non-API) interface between this module and the
22 ** rest of the SQLite system:
23 **
24 **    sqlite3VdbeSorterInit()       Create a new VdbeSorter object.
25 **
26 **    sqlite3VdbeSorterWrite()      Add a single new row to the VdbeSorter
27 **                                  object.  The row is a binary blob in the
28 **                                  OP_MakeRecord format that contains both
29 **                                  the ORDER BY key columns and result columns
30 **                                  in the case of a SELECT w/ ORDER BY, or
31 **                                  the complete record for an index entry
32 **                                  in the case of a CREATE INDEX.
33 **
34 **    sqlite3VdbeSorterRewind()     Sort all content previously added.
35 **                                  Position the read cursor on the
36 **                                  first sorted element.
37 **
38 **    sqlite3VdbeSorterNext()       Advance the read cursor to the next sorted
39 **                                  element.
40 **
41 **    sqlite3VdbeSorterRowkey()     Return the complete binary blob for the
42 **                                  row currently under the read cursor.
43 **
44 **    sqlite3VdbeSorterCompare()    Compare the binary blob for the row
45 **                                  currently under the read cursor against
46 **                                  another binary blob X and report if
47 **                                  X is strictly less than the read cursor.
48 **                                  Used to enforce uniqueness in a
49 **                                  CREATE UNIQUE INDEX statement.
50 **
51 **    sqlite3VdbeSorterClose()      Close the VdbeSorter object and reclaim
52 **                                  all resources.
53 **
54 **    sqlite3VdbeSorterReset()      Refurbish the VdbeSorter for reuse.  This
55 **                                  is like Close() followed by Init() only
56 **                                  much faster.
57 **
58 ** The interfaces above must be called in a particular order.  Write() can
59 ** only occur in between Init()/Reset() and Rewind().  Next(), Rowkey(), and
60 ** Compare() can only occur in between Rewind() and Close()/Reset(). i.e.
61 **
62 **   Init()
63 **   for each record: Write()
64 **   Rewind()
65 **     Rowkey()/Compare()
66 **   Next()
67 **   Close()
68 **
69 ** Algorithm:
70 **
71 ** Records passed to the sorter via calls to Write() are initially held
72 ** unsorted in main memory. Assuming the amount of memory used never exceeds
73 ** a threshold, when Rewind() is called the set of records is sorted using
74 ** an in-memory merge sort. In this case, no temporary files are required
75 ** and subsequent calls to Rowkey(), Next() and Compare() read records
76 ** directly from main memory.
77 **
78 ** If the amount of space used to store records in main memory exceeds the
79 ** threshold, then the set of records currently in memory are sorted and
80 ** written to a temporary file in "Packed Memory Array" (PMA) format.
81 ** A PMA created at this point is known as a "level-0 PMA". Higher levels
82 ** of PMAs may be created by merging existing PMAs together - for example
83 ** merging two or more level-0 PMAs together creates a level-1 PMA.
84 **
85 ** The threshold for the amount of main memory to use before flushing
86 ** records to a PMA is roughly the same as the limit configured for the
87 ** page-cache of the main database. Specifically, the threshold is set to
88 ** the value returned by "PRAGMA main.page_size" multipled by
89 ** that returned by "PRAGMA main.cache_size", in bytes.
90 **
91 ** If the sorter is running in single-threaded mode, then all PMAs generated
92 ** are appended to a single temporary file. Or, if the sorter is running in
93 ** multi-threaded mode then up to (N+1) temporary files may be opened, where
94 ** N is the configured number of worker threads. In this case, instead of
95 ** sorting the records and writing the PMA to a temporary file itself, the
96 ** calling thread usually launches a worker thread to do so. Except, if
97 ** there are already N worker threads running, the main thread does the work
98 ** itself.
99 **
100 ** The sorter is running in multi-threaded mode if (a) the library was built
101 ** with pre-processor symbol SQLITE_MAX_WORKER_THREADS set to a value greater
102 ** than zero, and (b) worker threads have been enabled at runtime by calling
103 ** "PRAGMA threads=N" with some value of N greater than 0.
104 **
105 ** When Rewind() is called, any data remaining in memory is flushed to a
106 ** final PMA. So at this point the data is stored in some number of sorted
107 ** PMAs within temporary files on disk.
108 **
109 ** If there are fewer than SORTER_MAX_MERGE_COUNT PMAs in total and the
110 ** sorter is running in single-threaded mode, then these PMAs are merged
111 ** incrementally as keys are retreived from the sorter by the VDBE.  The
112 ** MergeEngine object, described in further detail below, performs this
113 ** merge.
114 **
115 ** Or, if running in multi-threaded mode, then a background thread is
116 ** launched to merge the existing PMAs. Once the background thread has
117 ** merged T bytes of data into a single sorted PMA, the main thread
118 ** begins reading keys from that PMA while the background thread proceeds
119 ** with merging the next T bytes of data. And so on.
120 **
121 ** Parameter T is set to half the value of the memory threshold used
122 ** by Write() above to determine when to create a new PMA.
123 **
124 ** If there are more than SORTER_MAX_MERGE_COUNT PMAs in total when
125 ** Rewind() is called, then a hierarchy of incremental-merges is used.
126 ** First, T bytes of data from the first SORTER_MAX_MERGE_COUNT PMAs on
127 ** disk are merged together. Then T bytes of data from the second set, and
128 ** so on, such that no operation ever merges more than SORTER_MAX_MERGE_COUNT
129 ** PMAs at a time. This done is to improve locality.
130 **
131 ** If running in multi-threaded mode and there are more than
132 ** SORTER_MAX_MERGE_COUNT PMAs on disk when Rewind() is called, then more
133 ** than one background thread may be created. Specifically, there may be
134 ** one background thread for each temporary file on disk, and one background
135 ** thread to merge the output of each of the others to a single PMA for
136 ** the main thread to read from.
137 */
138 #include "sqliteInt.h"
139 #include "vdbeInt.h"
140 
141 /*
142 ** If SQLITE_DEBUG_SORTER_THREADS is defined, this module outputs various
143 ** messages to stderr that may be helpful in understanding the performance
144 ** characteristics of the sorter in multi-threaded mode.
145 */
146 #if 0
147 # define SQLITE_DEBUG_SORTER_THREADS 1
148 #endif
149 
150 /*
151 ** Hard-coded maximum amount of data to accumulate in memory before flushing
152 ** to a level 0 PMA. The purpose of this limit is to prevent various integer
153 ** overflows. 512MiB.
154 */
155 #define SQLITE_MAX_PMASZ    (1<<29)
156 
157 /*
158 ** Private objects used by the sorter
159 */
160 typedef struct MergeEngine MergeEngine;     /* Merge PMAs together */
161 typedef struct PmaReader PmaReader;         /* Incrementally read one PMA */
162 typedef struct PmaWriter PmaWriter;         /* Incrementally write one PMA */
163 typedef struct SorterRecord SorterRecord;   /* A record being sorted */
164 typedef struct SortSubtask SortSubtask;     /* A sub-task in the sort process */
165 typedef struct SorterFile SorterFile;       /* Temporary file object wrapper */
166 typedef struct SorterList SorterList;       /* In-memory list of records */
167 typedef struct IncrMerger IncrMerger;       /* Read & merge multiple PMAs */
168 
169 /*
170 ** A container for a temp file handle and the current amount of data
171 ** stored in the file.
172 */
173 struct SorterFile {
174   sqlite3_file *pFd;              /* File handle */
175   i64 iEof;                       /* Bytes of data stored in pFd */
176 };
177 
178 /*
179 ** An in-memory list of objects to be sorted.
180 **
181 ** If aMemory==0 then each object is allocated separately and the objects
182 ** are connected using SorterRecord.u.pNext.  If aMemory!=0 then all objects
183 ** are stored in the aMemory[] bulk memory, one right after the other, and
184 ** are connected using SorterRecord.u.iNext.
185 */
186 struct SorterList {
187   SorterRecord *pList;            /* Linked list of records */
188   u8 *aMemory;                    /* If non-NULL, bulk memory to hold pList */
189   int szPMA;                      /* Size of pList as PMA in bytes */
190 };
191 
192 /*
193 ** The MergeEngine object is used to combine two or more smaller PMAs into
194 ** one big PMA using a merge operation.  Separate PMAs all need to be
195 ** combined into one big PMA in order to be able to step through the sorted
196 ** records in order.
197 **
198 ** The aReadr[] array contains a PmaReader object for each of the PMAs being
199 ** merged.  An aReadr[] object either points to a valid key or else is at EOF.
200 ** ("EOF" means "End Of File".  When aReadr[] is at EOF there is no more data.)
201 ** For the purposes of the paragraphs below, we assume that the array is
202 ** actually N elements in size, where N is the smallest power of 2 greater
203 ** to or equal to the number of PMAs being merged. The extra aReadr[] elements
204 ** are treated as if they are empty (always at EOF).
205 **
206 ** The aTree[] array is also N elements in size. The value of N is stored in
207 ** the MergeEngine.nTree variable.
208 **
209 ** The final (N/2) elements of aTree[] contain the results of comparing
210 ** pairs of PMA keys together. Element i contains the result of
211 ** comparing aReadr[2*i-N] and aReadr[2*i-N+1]. Whichever key is smaller, the
212 ** aTree element is set to the index of it.
213 **
214 ** For the purposes of this comparison, EOF is considered greater than any
215 ** other key value. If the keys are equal (only possible with two EOF
216 ** values), it doesn't matter which index is stored.
217 **
218 ** The (N/4) elements of aTree[] that precede the final (N/2) described
219 ** above contains the index of the smallest of each block of 4 PmaReaders
220 ** And so on. So that aTree[1] contains the index of the PmaReader that
221 ** currently points to the smallest key value. aTree[0] is unused.
222 **
223 ** Example:
224 **
225 **     aReadr[0] -> Banana
226 **     aReadr[1] -> Feijoa
227 **     aReadr[2] -> Elderberry
228 **     aReadr[3] -> Currant
229 **     aReadr[4] -> Grapefruit
230 **     aReadr[5] -> Apple
231 **     aReadr[6] -> Durian
232 **     aReadr[7] -> EOF
233 **
234 **     aTree[] = { X, 5   0, 5    0, 3, 5, 6 }
235 **
236 ** The current element is "Apple" (the value of the key indicated by
237 ** PmaReader 5). When the Next() operation is invoked, PmaReader 5 will
238 ** be advanced to the next key in its segment. Say the next key is
239 ** "Eggplant":
240 **
241 **     aReadr[5] -> Eggplant
242 **
243 ** The contents of aTree[] are updated first by comparing the new PmaReader
244 ** 5 key to the current key of PmaReader 4 (still "Grapefruit"). The PmaReader
245 ** 5 value is still smaller, so aTree[6] is set to 5. And so on up the tree.
246 ** The value of PmaReader 6 - "Durian" - is now smaller than that of PmaReader
247 ** 5, so aTree[3] is set to 6. Key 0 is smaller than key 6 (Banana<Durian),
248 ** so the value written into element 1 of the array is 0. As follows:
249 **
250 **     aTree[] = { X, 0   0, 6    0, 3, 5, 6 }
251 **
252 ** In other words, each time we advance to the next sorter element, log2(N)
253 ** key comparison operations are required, where N is the number of segments
254 ** being merged (rounded up to the next power of 2).
255 */
256 struct MergeEngine {
257   int nTree;                 /* Used size of aTree/aReadr (power of 2) */
258   SortSubtask *pTask;        /* Used by this thread only */
259   int *aTree;                /* Current state of incremental merge */
260   PmaReader *aReadr;         /* Array of PmaReaders to merge data from */
261 };
262 
263 /*
264 ** This object represents a single thread of control in a sort operation.
265 ** Exactly VdbeSorter.nTask instances of this object are allocated
266 ** as part of each VdbeSorter object. Instances are never allocated any
267 ** other way. VdbeSorter.nTask is set to the number of worker threads allowed
268 ** (see SQLITE_CONFIG_WORKER_THREADS) plus one (the main thread).  Thus for
269 ** single-threaded operation, there is exactly one instance of this object
270 ** and for multi-threaded operation there are two or more instances.
271 **
272 ** Essentially, this structure contains all those fields of the VdbeSorter
273 ** structure for which each thread requires a separate instance. For example,
274 ** each thread requries its own UnpackedRecord object to unpack records in
275 ** as part of comparison operations.
276 **
277 ** Before a background thread is launched, variable bDone is set to 0. Then,
278 ** right before it exits, the thread itself sets bDone to 1. This is used for
279 ** two purposes:
280 **
281 **   1. When flushing the contents of memory to a level-0 PMA on disk, to
282 **      attempt to select a SortSubtask for which there is not already an
283 **      active background thread (since doing so causes the main thread
284 **      to block until it finishes).
285 **
286 **   2. If SQLITE_DEBUG_SORTER_THREADS is defined, to determine if a call
287 **      to sqlite3ThreadJoin() is likely to block. Cases that are likely to
288 **      block provoke debugging output.
289 **
290 ** In both cases, the effects of the main thread seeing (bDone==0) even
291 ** after the thread has finished are not dire. So we don't worry about
292 ** memory barriers and such here.
293 */
294 typedef int (*SorterCompare)(SortSubtask*,int*,const void*,int,const void*,int);
295 struct SortSubtask {
296   SQLiteThread *pThread;          /* Background thread, if any */
297   int bDone;                      /* Set if thread is finished but not joined */
298   VdbeSorter *pSorter;            /* Sorter that owns this sub-task */
299   UnpackedRecord *pUnpacked;      /* Space to unpack a record */
300   SorterList list;                /* List for thread to write to a PMA */
301   int nPMA;                       /* Number of PMAs currently in file */
302   SorterCompare xCompare;         /* Compare function to use */
303   SorterFile file;                /* Temp file for level-0 PMAs */
304   SorterFile file2;               /* Space for other PMAs */
305 };
306 
307 
308 /*
309 ** Main sorter structure. A single instance of this is allocated for each
310 ** sorter cursor created by the VDBE.
311 **
312 ** mxKeysize:
313 **   As records are added to the sorter by calls to sqlite3VdbeSorterWrite(),
314 **   this variable is updated so as to be set to the size on disk of the
315 **   largest record in the sorter.
316 */
317 struct VdbeSorter {
318   int mnPmaSize;                  /* Minimum PMA size, in bytes */
319   int mxPmaSize;                  /* Maximum PMA size, in bytes.  0==no limit */
320   int mxKeysize;                  /* Largest serialized key seen so far */
321   int pgsz;                       /* Main database page size */
322   PmaReader *pReader;             /* Readr data from here after Rewind() */
323   MergeEngine *pMerger;           /* Or here, if bUseThreads==0 */
324   sqlite3 *db;                    /* Database connection */
325   KeyInfo *pKeyInfo;              /* How to compare records */
326   UnpackedRecord *pUnpacked;      /* Used by VdbeSorterCompare() */
327   SorterList list;                /* List of in-memory records */
328   int iMemory;                    /* Offset of free space in list.aMemory */
329   int nMemory;                    /* Size of list.aMemory allocation in bytes */
330   u8 bUsePMA;                     /* True if one or more PMAs created */
331   u8 bUseThreads;                 /* True to use background threads */
332   u8 iPrev;                       /* Previous thread used to flush PMA */
333   u8 nTask;                       /* Size of aTask[] array */
334   u8 typeMask;
335   SortSubtask aTask[1];           /* One or more subtasks */
336 };
337 
338 #define SORTER_TYPE_INTEGER 0x01
339 #define SORTER_TYPE_TEXT    0x02
340 
341 /*
342 ** An instance of the following object is used to read records out of a
343 ** PMA, in sorted order.  The next key to be read is cached in nKey/aKey.
344 ** aKey might point into aMap or into aBuffer.  If neither of those locations
345 ** contain a contiguous representation of the key, then aAlloc is allocated
346 ** and the key is copied into aAlloc and aKey is made to poitn to aAlloc.
347 **
348 ** pFd==0 at EOF.
349 */
350 struct PmaReader {
351   i64 iReadOff;               /* Current read offset */
352   i64 iEof;                   /* 1 byte past EOF for this PmaReader */
353   int nAlloc;                 /* Bytes of space at aAlloc */
354   int nKey;                   /* Number of bytes in key */
355   sqlite3_file *pFd;          /* File handle we are reading from */
356   u8 *aAlloc;                 /* Space for aKey if aBuffer and pMap wont work */
357   u8 *aKey;                   /* Pointer to current key */
358   u8 *aBuffer;                /* Current read buffer */
359   int nBuffer;                /* Size of read buffer in bytes */
360   u8 *aMap;                   /* Pointer to mapping of entire file */
361   IncrMerger *pIncr;          /* Incremental merger */
362 };
363 
364 /*
365 ** Normally, a PmaReader object iterates through an existing PMA stored
366 ** within a temp file. However, if the PmaReader.pIncr variable points to
367 ** an object of the following type, it may be used to iterate/merge through
368 ** multiple PMAs simultaneously.
369 **
370 ** There are two types of IncrMerger object - single (bUseThread==0) and
371 ** multi-threaded (bUseThread==1).
372 **
373 ** A multi-threaded IncrMerger object uses two temporary files - aFile[0]
374 ** and aFile[1]. Neither file is allowed to grow to more than mxSz bytes in
375 ** size. When the IncrMerger is initialized, it reads enough data from
376 ** pMerger to populate aFile[0]. It then sets variables within the
377 ** corresponding PmaReader object to read from that file and kicks off
378 ** a background thread to populate aFile[1] with the next mxSz bytes of
379 ** sorted record data from pMerger.
380 **
381 ** When the PmaReader reaches the end of aFile[0], it blocks until the
382 ** background thread has finished populating aFile[1]. It then exchanges
383 ** the contents of the aFile[0] and aFile[1] variables within this structure,
384 ** sets the PmaReader fields to read from the new aFile[0] and kicks off
385 ** another background thread to populate the new aFile[1]. And so on, until
386 ** the contents of pMerger are exhausted.
387 **
388 ** A single-threaded IncrMerger does not open any temporary files of its
389 ** own. Instead, it has exclusive access to mxSz bytes of space beginning
390 ** at offset iStartOff of file pTask->file2. And instead of using a
391 ** background thread to prepare data for the PmaReader, with a single
392 ** threaded IncrMerger the allocate part of pTask->file2 is "refilled" with
393 ** keys from pMerger by the calling thread whenever the PmaReader runs out
394 ** of data.
395 */
396 struct IncrMerger {
397   SortSubtask *pTask;             /* Task that owns this merger */
398   MergeEngine *pMerger;           /* Merge engine thread reads data from */
399   i64 iStartOff;                  /* Offset to start writing file at */
400   int mxSz;                       /* Maximum bytes of data to store */
401   int bEof;                       /* Set to true when merge is finished */
402   int bUseThread;                 /* True to use a bg thread for this object */
403   SorterFile aFile[2];            /* aFile[0] for reading, [1] for writing */
404 };
405 
406 /*
407 ** An instance of this object is used for writing a PMA.
408 **
409 ** The PMA is written one record at a time.  Each record is of an arbitrary
410 ** size.  But I/O is more efficient if it occurs in page-sized blocks where
411 ** each block is aligned on a page boundary.  This object caches writes to
412 ** the PMA so that aligned, page-size blocks are written.
413 */
414 struct PmaWriter {
415   int eFWErr;                     /* Non-zero if in an error state */
416   u8 *aBuffer;                    /* Pointer to write buffer */
417   int nBuffer;                    /* Size of write buffer in bytes */
418   int iBufStart;                  /* First byte of buffer to write */
419   int iBufEnd;                    /* Last byte of buffer to write */
420   i64 iWriteOff;                  /* Offset of start of buffer in file */
421   sqlite3_file *pFd;              /* File handle to write to */
422 };
423 
424 /*
425 ** This object is the header on a single record while that record is being
426 ** held in memory and prior to being written out as part of a PMA.
427 **
428 ** How the linked list is connected depends on how memory is being managed
429 ** by this module. If using a separate allocation for each in-memory record
430 ** (VdbeSorter.list.aMemory==0), then the list is always connected using the
431 ** SorterRecord.u.pNext pointers.
432 **
433 ** Or, if using the single large allocation method (VdbeSorter.list.aMemory!=0),
434 ** then while records are being accumulated the list is linked using the
435 ** SorterRecord.u.iNext offset. This is because the aMemory[] array may
436 ** be sqlite3Realloc()ed while records are being accumulated. Once the VM
437 ** has finished passing records to the sorter, or when the in-memory buffer
438 ** is full, the list is sorted. As part of the sorting process, it is
439 ** converted to use the SorterRecord.u.pNext pointers. See function
440 ** vdbeSorterSort() for details.
441 */
442 struct SorterRecord {
443   int nVal;                       /* Size of the record in bytes */
444   union {
445     SorterRecord *pNext;          /* Pointer to next record in list */
446     int iNext;                    /* Offset within aMemory of next record */
447   } u;
448   /* The data for the record immediately follows this header */
449 };
450 
451 /* Return a pointer to the buffer containing the record data for SorterRecord
452 ** object p. Should be used as if:
453 **
454 **   void *SRVAL(SorterRecord *p) { return (void*)&p[1]; }
455 */
456 #define SRVAL(p) ((void*)((SorterRecord*)(p) + 1))
457 
458 
459 /* Maximum number of PMAs that a single MergeEngine can merge */
460 #define SORTER_MAX_MERGE_COUNT 16
461 
462 static int vdbeIncrSwap(IncrMerger*);
463 static void vdbeIncrFree(IncrMerger *);
464 
465 /*
466 ** Free all memory belonging to the PmaReader object passed as the
467 ** argument. All structure fields are set to zero before returning.
468 */
469 static void vdbePmaReaderClear(PmaReader *pReadr){
470   sqlite3_free(pReadr->aAlloc);
471   sqlite3_free(pReadr->aBuffer);
472   if( pReadr->aMap ) sqlite3OsUnfetch(pReadr->pFd, 0, pReadr->aMap);
473   vdbeIncrFree(pReadr->pIncr);
474   memset(pReadr, 0, sizeof(PmaReader));
475 }
476 
477 /*
478 ** Read the next nByte bytes of data from the PMA p.
479 ** If successful, set *ppOut to point to a buffer containing the data
480 ** and return SQLITE_OK. Otherwise, if an error occurs, return an SQLite
481 ** error code.
482 **
483 ** The buffer returned in *ppOut is only valid until the
484 ** next call to this function.
485 */
486 static int vdbePmaReadBlob(
487   PmaReader *p,                   /* PmaReader from which to take the blob */
488   int nByte,                      /* Bytes of data to read */
489   u8 **ppOut                      /* OUT: Pointer to buffer containing data */
490 ){
491   int iBuf;                       /* Offset within buffer to read from */
492   int nAvail;                     /* Bytes of data available in buffer */
493 
494   if( p->aMap ){
495     *ppOut = &p->aMap[p->iReadOff];
496     p->iReadOff += nByte;
497     return SQLITE_OK;
498   }
499 
500   assert( p->aBuffer );
501 
502   /* If there is no more data to be read from the buffer, read the next
503   ** p->nBuffer bytes of data from the file into it. Or, if there are less
504   ** than p->nBuffer bytes remaining in the PMA, read all remaining data.  */
505   iBuf = p->iReadOff % p->nBuffer;
506   if( iBuf==0 ){
507     int nRead;                    /* Bytes to read from disk */
508     int rc;                       /* sqlite3OsRead() return code */
509 
510     /* Determine how many bytes of data to read. */
511     if( (p->iEof - p->iReadOff) > (i64)p->nBuffer ){
512       nRead = p->nBuffer;
513     }else{
514       nRead = (int)(p->iEof - p->iReadOff);
515     }
516     assert( nRead>0 );
517 
518     /* Readr data from the file. Return early if an error occurs. */
519     rc = sqlite3OsRead(p->pFd, p->aBuffer, nRead, p->iReadOff);
520     assert( rc!=SQLITE_IOERR_SHORT_READ );
521     if( rc!=SQLITE_OK ) return rc;
522   }
523   nAvail = p->nBuffer - iBuf;
524 
525   if( nByte<=nAvail ){
526     /* The requested data is available in the in-memory buffer. In this
527     ** case there is no need to make a copy of the data, just return a
528     ** pointer into the buffer to the caller.  */
529     *ppOut = &p->aBuffer[iBuf];
530     p->iReadOff += nByte;
531   }else{
532     /* The requested data is not all available in the in-memory buffer.
533     ** In this case, allocate space at p->aAlloc[] to copy the requested
534     ** range into. Then return a copy of pointer p->aAlloc to the caller.  */
535     int nRem;                     /* Bytes remaining to copy */
536 
537     /* Extend the p->aAlloc[] allocation if required. */
538     if( p->nAlloc<nByte ){
539       u8 *aNew;
540       int nNew = MAX(128, p->nAlloc*2);
541       while( nByte>nNew ) nNew = nNew*2;
542       aNew = sqlite3Realloc(p->aAlloc, nNew);
543       if( !aNew ) return SQLITE_NOMEM_BKPT;
544       p->nAlloc = nNew;
545       p->aAlloc = aNew;
546     }
547 
548     /* Copy as much data as is available in the buffer into the start of
549     ** p->aAlloc[].  */
550     memcpy(p->aAlloc, &p->aBuffer[iBuf], nAvail);
551     p->iReadOff += nAvail;
552     nRem = nByte - nAvail;
553 
554     /* The following loop copies up to p->nBuffer bytes per iteration into
555     ** the p->aAlloc[] buffer.  */
556     while( nRem>0 ){
557       int rc;                     /* vdbePmaReadBlob() return code */
558       int nCopy;                  /* Number of bytes to copy */
559       u8 *aNext;                  /* Pointer to buffer to copy data from */
560 
561       nCopy = nRem;
562       if( nRem>p->nBuffer ) nCopy = p->nBuffer;
563       rc = vdbePmaReadBlob(p, nCopy, &aNext);
564       if( rc!=SQLITE_OK ) return rc;
565       assert( aNext!=p->aAlloc );
566       memcpy(&p->aAlloc[nByte - nRem], aNext, nCopy);
567       nRem -= nCopy;
568     }
569 
570     *ppOut = p->aAlloc;
571   }
572 
573   return SQLITE_OK;
574 }
575 
576 /*
577 ** Read a varint from the stream of data accessed by p. Set *pnOut to
578 ** the value read.
579 */
580 static int vdbePmaReadVarint(PmaReader *p, u64 *pnOut){
581   int iBuf;
582 
583   if( p->aMap ){
584     p->iReadOff += sqlite3GetVarint(&p->aMap[p->iReadOff], pnOut);
585   }else{
586     iBuf = p->iReadOff % p->nBuffer;
587     if( iBuf && (p->nBuffer-iBuf)>=9 ){
588       p->iReadOff += sqlite3GetVarint(&p->aBuffer[iBuf], pnOut);
589     }else{
590       u8 aVarint[16], *a;
591       int i = 0, rc;
592       do{
593         rc = vdbePmaReadBlob(p, 1, &a);
594         if( rc ) return rc;
595         aVarint[(i++)&0xf] = a[0];
596       }while( (a[0]&0x80)!=0 );
597       sqlite3GetVarint(aVarint, pnOut);
598     }
599   }
600 
601   return SQLITE_OK;
602 }
603 
604 /*
605 ** Attempt to memory map file pFile. If successful, set *pp to point to the
606 ** new mapping and return SQLITE_OK. If the mapping is not attempted
607 ** (because the file is too large or the VFS layer is configured not to use
608 ** mmap), return SQLITE_OK and set *pp to NULL.
609 **
610 ** Or, if an error occurs, return an SQLite error code. The final value of
611 ** *pp is undefined in this case.
612 */
613 static int vdbeSorterMapFile(SortSubtask *pTask, SorterFile *pFile, u8 **pp){
614   int rc = SQLITE_OK;
615   if( pFile->iEof<=(i64)(pTask->pSorter->db->nMaxSorterMmap) ){
616     sqlite3_file *pFd = pFile->pFd;
617     if( pFd->pMethods->iVersion>=3 ){
618       rc = sqlite3OsFetch(pFd, 0, (int)pFile->iEof, (void**)pp);
619       testcase( rc!=SQLITE_OK );
620     }
621   }
622   return rc;
623 }
624 
625 /*
626 ** Attach PmaReader pReadr to file pFile (if it is not already attached to
627 ** that file) and seek it to offset iOff within the file.  Return SQLITE_OK
628 ** if successful, or an SQLite error code if an error occurs.
629 */
630 static int vdbePmaReaderSeek(
631   SortSubtask *pTask,             /* Task context */
632   PmaReader *pReadr,              /* Reader whose cursor is to be moved */
633   SorterFile *pFile,              /* Sorter file to read from */
634   i64 iOff                        /* Offset in pFile */
635 ){
636   int rc = SQLITE_OK;
637 
638   assert( pReadr->pIncr==0 || pReadr->pIncr->bEof==0 );
639 
640   if( sqlite3FaultSim(201) ) return SQLITE_IOERR_READ;
641   if( pReadr->aMap ){
642     sqlite3OsUnfetch(pReadr->pFd, 0, pReadr->aMap);
643     pReadr->aMap = 0;
644   }
645   pReadr->iReadOff = iOff;
646   pReadr->iEof = pFile->iEof;
647   pReadr->pFd = pFile->pFd;
648 
649   rc = vdbeSorterMapFile(pTask, pFile, &pReadr->aMap);
650   if( rc==SQLITE_OK && pReadr->aMap==0 ){
651     int pgsz = pTask->pSorter->pgsz;
652     int iBuf = pReadr->iReadOff % pgsz;
653     if( pReadr->aBuffer==0 ){
654       pReadr->aBuffer = (u8*)sqlite3Malloc(pgsz);
655       if( pReadr->aBuffer==0 ) rc = SQLITE_NOMEM_BKPT;
656       pReadr->nBuffer = pgsz;
657     }
658     if( rc==SQLITE_OK && iBuf ){
659       int nRead = pgsz - iBuf;
660       if( (pReadr->iReadOff + nRead) > pReadr->iEof ){
661         nRead = (int)(pReadr->iEof - pReadr->iReadOff);
662       }
663       rc = sqlite3OsRead(
664           pReadr->pFd, &pReadr->aBuffer[iBuf], nRead, pReadr->iReadOff
665       );
666       testcase( rc!=SQLITE_OK );
667     }
668   }
669 
670   return rc;
671 }
672 
673 /*
674 ** Advance PmaReader pReadr to the next key in its PMA. Return SQLITE_OK if
675 ** no error occurs, or an SQLite error code if one does.
676 */
677 static int vdbePmaReaderNext(PmaReader *pReadr){
678   int rc = SQLITE_OK;             /* Return Code */
679   u64 nRec = 0;                   /* Size of record in bytes */
680 
681 
682   if( pReadr->iReadOff>=pReadr->iEof ){
683     IncrMerger *pIncr = pReadr->pIncr;
684     int bEof = 1;
685     if( pIncr ){
686       rc = vdbeIncrSwap(pIncr);
687       if( rc==SQLITE_OK && pIncr->bEof==0 ){
688         rc = vdbePmaReaderSeek(
689             pIncr->pTask, pReadr, &pIncr->aFile[0], pIncr->iStartOff
690         );
691         bEof = 0;
692       }
693     }
694 
695     if( bEof ){
696       /* This is an EOF condition */
697       vdbePmaReaderClear(pReadr);
698       testcase( rc!=SQLITE_OK );
699       return rc;
700     }
701   }
702 
703   if( rc==SQLITE_OK ){
704     rc = vdbePmaReadVarint(pReadr, &nRec);
705   }
706   if( rc==SQLITE_OK ){
707     pReadr->nKey = (int)nRec;
708     rc = vdbePmaReadBlob(pReadr, (int)nRec, &pReadr->aKey);
709     testcase( rc!=SQLITE_OK );
710   }
711 
712   return rc;
713 }
714 
715 /*
716 ** Initialize PmaReader pReadr to scan through the PMA stored in file pFile
717 ** starting at offset iStart and ending at offset iEof-1. This function
718 ** leaves the PmaReader pointing to the first key in the PMA (or EOF if the
719 ** PMA is empty).
720 **
721 ** If the pnByte parameter is NULL, then it is assumed that the file
722 ** contains a single PMA, and that that PMA omits the initial length varint.
723 */
724 static int vdbePmaReaderInit(
725   SortSubtask *pTask,             /* Task context */
726   SorterFile *pFile,              /* Sorter file to read from */
727   i64 iStart,                     /* Start offset in pFile */
728   PmaReader *pReadr,              /* PmaReader to populate */
729   i64 *pnByte                     /* IN/OUT: Increment this value by PMA size */
730 ){
731   int rc;
732 
733   assert( pFile->iEof>iStart );
734   assert( pReadr->aAlloc==0 && pReadr->nAlloc==0 );
735   assert( pReadr->aBuffer==0 );
736   assert( pReadr->aMap==0 );
737 
738   rc = vdbePmaReaderSeek(pTask, pReadr, pFile, iStart);
739   if( rc==SQLITE_OK ){
740     u64 nByte = 0;                 /* Size of PMA in bytes */
741     rc = vdbePmaReadVarint(pReadr, &nByte);
742     pReadr->iEof = pReadr->iReadOff + nByte;
743     *pnByte += nByte;
744   }
745 
746   if( rc==SQLITE_OK ){
747     rc = vdbePmaReaderNext(pReadr);
748   }
749   return rc;
750 }
751 
752 /*
753 ** A version of vdbeSorterCompare() that assumes that it has already been
754 ** determined that the first field of key1 is equal to the first field of
755 ** key2.
756 */
757 static int vdbeSorterCompareTail(
758   SortSubtask *pTask,             /* Subtask context (for pKeyInfo) */
759   int *pbKey2Cached,              /* True if pTask->pUnpacked is pKey2 */
760   const void *pKey1, int nKey1,   /* Left side of comparison */
761   const void *pKey2, int nKey2    /* Right side of comparison */
762 ){
763   UnpackedRecord *r2 = pTask->pUnpacked;
764   if( *pbKey2Cached==0 ){
765     sqlite3VdbeRecordUnpack(pTask->pSorter->pKeyInfo, nKey2, pKey2, r2);
766     *pbKey2Cached = 1;
767   }
768   return sqlite3VdbeRecordCompareWithSkip(nKey1, pKey1, r2, 1);
769 }
770 
771 /*
772 ** Compare key1 (buffer pKey1, size nKey1 bytes) with key2 (buffer pKey2,
773 ** size nKey2 bytes). Use (pTask->pKeyInfo) for the collation sequences
774 ** used by the comparison. Return the result of the comparison.
775 **
776 ** If IN/OUT parameter *pbKey2Cached is true when this function is called,
777 ** it is assumed that (pTask->pUnpacked) contains the unpacked version
778 ** of key2. If it is false, (pTask->pUnpacked) is populated with the unpacked
779 ** version of key2 and *pbKey2Cached set to true before returning.
780 **
781 ** If an OOM error is encountered, (pTask->pUnpacked->error_rc) is set
782 ** to SQLITE_NOMEM.
783 */
784 static int vdbeSorterCompare(
785   SortSubtask *pTask,             /* Subtask context (for pKeyInfo) */
786   int *pbKey2Cached,              /* True if pTask->pUnpacked is pKey2 */
787   const void *pKey1, int nKey1,   /* Left side of comparison */
788   const void *pKey2, int nKey2    /* Right side of comparison */
789 ){
790   UnpackedRecord *r2 = pTask->pUnpacked;
791   if( !*pbKey2Cached ){
792     sqlite3VdbeRecordUnpack(pTask->pSorter->pKeyInfo, nKey2, pKey2, r2);
793     *pbKey2Cached = 1;
794   }
795   return sqlite3VdbeRecordCompare(nKey1, pKey1, r2);
796 }
797 
798 /*
799 ** A specially optimized version of vdbeSorterCompare() that assumes that
800 ** the first field of each key is a TEXT value and that the collation
801 ** sequence to compare them with is BINARY.
802 */
803 static int vdbeSorterCompareText(
804   SortSubtask *pTask,             /* Subtask context (for pKeyInfo) */
805   int *pbKey2Cached,              /* True if pTask->pUnpacked is pKey2 */
806   const void *pKey1, int nKey1,   /* Left side of comparison */
807   const void *pKey2, int nKey2    /* Right side of comparison */
808 ){
809   const u8 * const p1 = (const u8 * const)pKey1;
810   const u8 * const p2 = (const u8 * const)pKey2;
811   const u8 * const v1 = &p1[ p1[0] ];   /* Pointer to value 1 */
812   const u8 * const v2 = &p2[ p2[0] ];   /* Pointer to value 2 */
813 
814   int n1;
815   int n2;
816   int res;
817 
818   getVarint32(&p1[1], n1);
819   getVarint32(&p2[1], n2);
820   res = memcmp(v1, v2, (MIN(n1, n2) - 13)/2);
821   if( res==0 ){
822     res = n1 - n2;
823   }
824 
825   if( res==0 ){
826     if( pTask->pSorter->pKeyInfo->nKeyField>1 ){
827       res = vdbeSorterCompareTail(
828           pTask, pbKey2Cached, pKey1, nKey1, pKey2, nKey2
829       );
830     }
831   }else{
832     if( pTask->pSorter->pKeyInfo->aSortOrder[0] ){
833       res = res * -1;
834     }
835   }
836 
837   return res;
838 }
839 
840 /*
841 ** A specially optimized version of vdbeSorterCompare() that assumes that
842 ** the first field of each key is an INTEGER value.
843 */
844 static int vdbeSorterCompareInt(
845   SortSubtask *pTask,             /* Subtask context (for pKeyInfo) */
846   int *pbKey2Cached,              /* True if pTask->pUnpacked is pKey2 */
847   const void *pKey1, int nKey1,   /* Left side of comparison */
848   const void *pKey2, int nKey2    /* Right side of comparison */
849 ){
850   const u8 * const p1 = (const u8 * const)pKey1;
851   const u8 * const p2 = (const u8 * const)pKey2;
852   const int s1 = p1[1];                 /* Left hand serial type */
853   const int s2 = p2[1];                 /* Right hand serial type */
854   const u8 * const v1 = &p1[ p1[0] ];   /* Pointer to value 1 */
855   const u8 * const v2 = &p2[ p2[0] ];   /* Pointer to value 2 */
856   int res;                              /* Return value */
857 
858   assert( (s1>0 && s1<7) || s1==8 || s1==9 );
859   assert( (s2>0 && s2<7) || s2==8 || s2==9 );
860 
861   if( s1==s2 ){
862     /* The two values have the same sign. Compare using memcmp(). */
863     static const u8 aLen[] = {0, 1, 2, 3, 4, 6, 8, 0, 0, 0 };
864     const u8 n = aLen[s1];
865     int i;
866     res = 0;
867     for(i=0; i<n; i++){
868       if( (res = v1[i] - v2[i])!=0 ){
869         if( ((v1[0] ^ v2[0]) & 0x80)!=0 ){
870           res = v1[0] & 0x80 ? -1 : +1;
871         }
872         break;
873       }
874     }
875   }else if( s1>7 && s2>7 ){
876     res = s1 - s2;
877   }else{
878     if( s2>7 ){
879       res = +1;
880     }else if( s1>7 ){
881       res = -1;
882     }else{
883       res = s1 - s2;
884     }
885     assert( res!=0 );
886 
887     if( res>0 ){
888       if( *v1 & 0x80 ) res = -1;
889     }else{
890       if( *v2 & 0x80 ) res = +1;
891     }
892   }
893 
894   if( res==0 ){
895     if( pTask->pSorter->pKeyInfo->nKeyField>1 ){
896       res = vdbeSorterCompareTail(
897           pTask, pbKey2Cached, pKey1, nKey1, pKey2, nKey2
898       );
899     }
900   }else if( pTask->pSorter->pKeyInfo->aSortOrder[0] ){
901     res = res * -1;
902   }
903 
904   return res;
905 }
906 
907 /*
908 ** Initialize the temporary index cursor just opened as a sorter cursor.
909 **
910 ** Usually, the sorter module uses the value of (pCsr->pKeyInfo->nKeyField)
911 ** to determine the number of fields that should be compared from the
912 ** records being sorted. However, if the value passed as argument nField
913 ** is non-zero and the sorter is able to guarantee a stable sort, nField
914 ** is used instead. This is used when sorting records for a CREATE INDEX
915 ** statement. In this case, keys are always delivered to the sorter in
916 ** order of the primary key, which happens to be make up the final part
917 ** of the records being sorted. So if the sort is stable, there is never
918 ** any reason to compare PK fields and they can be ignored for a small
919 ** performance boost.
920 **
921 ** The sorter can guarantee a stable sort when running in single-threaded
922 ** mode, but not in multi-threaded mode.
923 **
924 ** SQLITE_OK is returned if successful, or an SQLite error code otherwise.
925 */
926 int sqlite3VdbeSorterInit(
927   sqlite3 *db,                    /* Database connection (for malloc()) */
928   int nField,                     /* Number of key fields in each record */
929   VdbeCursor *pCsr                /* Cursor that holds the new sorter */
930 ){
931   int pgsz;                       /* Page size of main database */
932   int i;                          /* Used to iterate through aTask[] */
933   VdbeSorter *pSorter;            /* The new sorter */
934   KeyInfo *pKeyInfo;              /* Copy of pCsr->pKeyInfo with db==0 */
935   int szKeyInfo;                  /* Size of pCsr->pKeyInfo in bytes */
936   int sz;                         /* Size of pSorter in bytes */
937   int rc = SQLITE_OK;
938 #if SQLITE_MAX_WORKER_THREADS==0
939 # define nWorker 0
940 #else
941   int nWorker;
942 #endif
943 
944   /* Initialize the upper limit on the number of worker threads */
945 #if SQLITE_MAX_WORKER_THREADS>0
946   if( sqlite3TempInMemory(db) || sqlite3GlobalConfig.bCoreMutex==0 ){
947     nWorker = 0;
948   }else{
949     nWorker = db->aLimit[SQLITE_LIMIT_WORKER_THREADS];
950   }
951 #endif
952 
953   /* Do not allow the total number of threads (main thread + all workers)
954   ** to exceed the maximum merge count */
955 #if SQLITE_MAX_WORKER_THREADS>=SORTER_MAX_MERGE_COUNT
956   if( nWorker>=SORTER_MAX_MERGE_COUNT ){
957     nWorker = SORTER_MAX_MERGE_COUNT-1;
958   }
959 #endif
960 
961   assert( pCsr->pKeyInfo && pCsr->pBtx==0 );
962   assert( pCsr->eCurType==CURTYPE_SORTER );
963   szKeyInfo = sizeof(KeyInfo) + (pCsr->pKeyInfo->nKeyField-1)*sizeof(CollSeq*);
964   sz = sizeof(VdbeSorter) + nWorker * sizeof(SortSubtask);
965 
966   pSorter = (VdbeSorter*)sqlite3DbMallocZero(db, sz + szKeyInfo);
967   pCsr->uc.pSorter = pSorter;
968   if( pSorter==0 ){
969     rc = SQLITE_NOMEM_BKPT;
970   }else{
971     pSorter->pKeyInfo = pKeyInfo = (KeyInfo*)((u8*)pSorter + sz);
972     memcpy(pKeyInfo, pCsr->pKeyInfo, szKeyInfo);
973     pKeyInfo->db = 0;
974     if( nField && nWorker==0 ){
975       pKeyInfo->nKeyField = nField;
976     }
977     pSorter->pgsz = pgsz = sqlite3BtreeGetPageSize(db->aDb[0].pBt);
978     pSorter->nTask = nWorker + 1;
979     pSorter->iPrev = (u8)(nWorker - 1);
980     pSorter->bUseThreads = (pSorter->nTask>1);
981     pSorter->db = db;
982     for(i=0; i<pSorter->nTask; i++){
983       SortSubtask *pTask = &pSorter->aTask[i];
984       pTask->pSorter = pSorter;
985     }
986 
987     if( !sqlite3TempInMemory(db) ){
988       i64 mxCache;                /* Cache size in bytes*/
989       u32 szPma = sqlite3GlobalConfig.szPma;
990       pSorter->mnPmaSize = szPma * pgsz;
991 
992       mxCache = db->aDb[0].pSchema->cache_size;
993       if( mxCache<0 ){
994         /* A negative cache-size value C indicates that the cache is abs(C)
995         ** KiB in size.  */
996         mxCache = mxCache * -1024;
997       }else{
998         mxCache = mxCache * pgsz;
999       }
1000       mxCache = MIN(mxCache, SQLITE_MAX_PMASZ);
1001       pSorter->mxPmaSize = MAX(pSorter->mnPmaSize, (int)mxCache);
1002 
1003       /* EVIDENCE-OF: R-26747-61719 When the application provides any amount of
1004       ** scratch memory using SQLITE_CONFIG_SCRATCH, SQLite avoids unnecessary
1005       ** large heap allocations.
1006       */
1007       if( sqlite3GlobalConfig.pScratch==0 ){
1008         assert( pSorter->iMemory==0 );
1009         pSorter->nMemory = pgsz;
1010         pSorter->list.aMemory = (u8*)sqlite3Malloc(pgsz);
1011         if( !pSorter->list.aMemory ) rc = SQLITE_NOMEM_BKPT;
1012       }
1013     }
1014 
1015     if( pKeyInfo->nAllField<13
1016      && (pKeyInfo->aColl[0]==0 || pKeyInfo->aColl[0]==db->pDfltColl)
1017     ){
1018       pSorter->typeMask = SORTER_TYPE_INTEGER | SORTER_TYPE_TEXT;
1019     }
1020   }
1021 
1022   return rc;
1023 }
1024 #undef nWorker   /* Defined at the top of this function */
1025 
1026 /*
1027 ** Free the list of sorted records starting at pRecord.
1028 */
1029 static void vdbeSorterRecordFree(sqlite3 *db, SorterRecord *pRecord){
1030   SorterRecord *p;
1031   SorterRecord *pNext;
1032   for(p=pRecord; p; p=pNext){
1033     pNext = p->u.pNext;
1034     sqlite3DbFree(db, p);
1035   }
1036 }
1037 
1038 /*
1039 ** Free all resources owned by the object indicated by argument pTask. All
1040 ** fields of *pTask are zeroed before returning.
1041 */
1042 static void vdbeSortSubtaskCleanup(sqlite3 *db, SortSubtask *pTask){
1043   sqlite3DbFree(db, pTask->pUnpacked);
1044 #if SQLITE_MAX_WORKER_THREADS>0
1045   /* pTask->list.aMemory can only be non-zero if it was handed memory
1046   ** from the main thread.  That only occurs SQLITE_MAX_WORKER_THREADS>0 */
1047   if( pTask->list.aMemory ){
1048     sqlite3_free(pTask->list.aMemory);
1049   }else
1050 #endif
1051   {
1052     assert( pTask->list.aMemory==0 );
1053     vdbeSorterRecordFree(0, pTask->list.pList);
1054   }
1055   if( pTask->file.pFd ){
1056     sqlite3OsCloseFree(pTask->file.pFd);
1057   }
1058   if( pTask->file2.pFd ){
1059     sqlite3OsCloseFree(pTask->file2.pFd);
1060   }
1061   memset(pTask, 0, sizeof(SortSubtask));
1062 }
1063 
1064 #ifdef SQLITE_DEBUG_SORTER_THREADS
1065 static void vdbeSorterWorkDebug(SortSubtask *pTask, const char *zEvent){
1066   i64 t;
1067   int iTask = (pTask - pTask->pSorter->aTask);
1068   sqlite3OsCurrentTimeInt64(pTask->pSorter->db->pVfs, &t);
1069   fprintf(stderr, "%lld:%d %s\n", t, iTask, zEvent);
1070 }
1071 static void vdbeSorterRewindDebug(const char *zEvent){
1072   i64 t;
1073   sqlite3OsCurrentTimeInt64(sqlite3_vfs_find(0), &t);
1074   fprintf(stderr, "%lld:X %s\n", t, zEvent);
1075 }
1076 static void vdbeSorterPopulateDebug(
1077   SortSubtask *pTask,
1078   const char *zEvent
1079 ){
1080   i64 t;
1081   int iTask = (pTask - pTask->pSorter->aTask);
1082   sqlite3OsCurrentTimeInt64(pTask->pSorter->db->pVfs, &t);
1083   fprintf(stderr, "%lld:bg%d %s\n", t, iTask, zEvent);
1084 }
1085 static void vdbeSorterBlockDebug(
1086   SortSubtask *pTask,
1087   int bBlocked,
1088   const char *zEvent
1089 ){
1090   if( bBlocked ){
1091     i64 t;
1092     sqlite3OsCurrentTimeInt64(pTask->pSorter->db->pVfs, &t);
1093     fprintf(stderr, "%lld:main %s\n", t, zEvent);
1094   }
1095 }
1096 #else
1097 # define vdbeSorterWorkDebug(x,y)
1098 # define vdbeSorterRewindDebug(y)
1099 # define vdbeSorterPopulateDebug(x,y)
1100 # define vdbeSorterBlockDebug(x,y,z)
1101 #endif
1102 
1103 #if SQLITE_MAX_WORKER_THREADS>0
1104 /*
1105 ** Join thread pTask->thread.
1106 */
1107 static int vdbeSorterJoinThread(SortSubtask *pTask){
1108   int rc = SQLITE_OK;
1109   if( pTask->pThread ){
1110 #ifdef SQLITE_DEBUG_SORTER_THREADS
1111     int bDone = pTask->bDone;
1112 #endif
1113     void *pRet = SQLITE_INT_TO_PTR(SQLITE_ERROR);
1114     vdbeSorterBlockDebug(pTask, !bDone, "enter");
1115     (void)sqlite3ThreadJoin(pTask->pThread, &pRet);
1116     vdbeSorterBlockDebug(pTask, !bDone, "exit");
1117     rc = SQLITE_PTR_TO_INT(pRet);
1118     assert( pTask->bDone==1 );
1119     pTask->bDone = 0;
1120     pTask->pThread = 0;
1121   }
1122   return rc;
1123 }
1124 
1125 /*
1126 ** Launch a background thread to run xTask(pIn).
1127 */
1128 static int vdbeSorterCreateThread(
1129   SortSubtask *pTask,             /* Thread will use this task object */
1130   void *(*xTask)(void*),          /* Routine to run in a separate thread */
1131   void *pIn                       /* Argument passed into xTask() */
1132 ){
1133   assert( pTask->pThread==0 && pTask->bDone==0 );
1134   return sqlite3ThreadCreate(&pTask->pThread, xTask, pIn);
1135 }
1136 
1137 /*
1138 ** Join all outstanding threads launched by SorterWrite() to create
1139 ** level-0 PMAs.
1140 */
1141 static int vdbeSorterJoinAll(VdbeSorter *pSorter, int rcin){
1142   int rc = rcin;
1143   int i;
1144 
1145   /* This function is always called by the main user thread.
1146   **
1147   ** If this function is being called after SorterRewind() has been called,
1148   ** it is possible that thread pSorter->aTask[pSorter->nTask-1].pThread
1149   ** is currently attempt to join one of the other threads. To avoid a race
1150   ** condition where this thread also attempts to join the same object, join
1151   ** thread pSorter->aTask[pSorter->nTask-1].pThread first. */
1152   for(i=pSorter->nTask-1; i>=0; i--){
1153     SortSubtask *pTask = &pSorter->aTask[i];
1154     int rc2 = vdbeSorterJoinThread(pTask);
1155     if( rc==SQLITE_OK ) rc = rc2;
1156   }
1157   return rc;
1158 }
1159 #else
1160 # define vdbeSorterJoinAll(x,rcin) (rcin)
1161 # define vdbeSorterJoinThread(pTask) SQLITE_OK
1162 #endif
1163 
1164 /*
1165 ** Allocate a new MergeEngine object capable of handling up to
1166 ** nReader PmaReader inputs.
1167 **
1168 ** nReader is automatically rounded up to the next power of two.
1169 ** nReader may not exceed SORTER_MAX_MERGE_COUNT even after rounding up.
1170 */
1171 static MergeEngine *vdbeMergeEngineNew(int nReader){
1172   int N = 2;                      /* Smallest power of two >= nReader */
1173   int nByte;                      /* Total bytes of space to allocate */
1174   MergeEngine *pNew;              /* Pointer to allocated object to return */
1175 
1176   assert( nReader<=SORTER_MAX_MERGE_COUNT );
1177 
1178   while( N<nReader ) N += N;
1179   nByte = sizeof(MergeEngine) + N * (sizeof(int) + sizeof(PmaReader));
1180 
1181   pNew = sqlite3FaultSim(100) ? 0 : (MergeEngine*)sqlite3MallocZero(nByte);
1182   if( pNew ){
1183     pNew->nTree = N;
1184     pNew->pTask = 0;
1185     pNew->aReadr = (PmaReader*)&pNew[1];
1186     pNew->aTree = (int*)&pNew->aReadr[N];
1187   }
1188   return pNew;
1189 }
1190 
1191 /*
1192 ** Free the MergeEngine object passed as the only argument.
1193 */
1194 static void vdbeMergeEngineFree(MergeEngine *pMerger){
1195   int i;
1196   if( pMerger ){
1197     for(i=0; i<pMerger->nTree; i++){
1198       vdbePmaReaderClear(&pMerger->aReadr[i]);
1199     }
1200   }
1201   sqlite3_free(pMerger);
1202 }
1203 
1204 /*
1205 ** Free all resources associated with the IncrMerger object indicated by
1206 ** the first argument.
1207 */
1208 static void vdbeIncrFree(IncrMerger *pIncr){
1209   if( pIncr ){
1210 #if SQLITE_MAX_WORKER_THREADS>0
1211     if( pIncr->bUseThread ){
1212       vdbeSorterJoinThread(pIncr->pTask);
1213       if( pIncr->aFile[0].pFd ) sqlite3OsCloseFree(pIncr->aFile[0].pFd);
1214       if( pIncr->aFile[1].pFd ) sqlite3OsCloseFree(pIncr->aFile[1].pFd);
1215     }
1216 #endif
1217     vdbeMergeEngineFree(pIncr->pMerger);
1218     sqlite3_free(pIncr);
1219   }
1220 }
1221 
1222 /*
1223 ** Reset a sorting cursor back to its original empty state.
1224 */
1225 void sqlite3VdbeSorterReset(sqlite3 *db, VdbeSorter *pSorter){
1226   int i;
1227   (void)vdbeSorterJoinAll(pSorter, SQLITE_OK);
1228   assert( pSorter->bUseThreads || pSorter->pReader==0 );
1229 #if SQLITE_MAX_WORKER_THREADS>0
1230   if( pSorter->pReader ){
1231     vdbePmaReaderClear(pSorter->pReader);
1232     sqlite3DbFree(db, pSorter->pReader);
1233     pSorter->pReader = 0;
1234   }
1235 #endif
1236   vdbeMergeEngineFree(pSorter->pMerger);
1237   pSorter->pMerger = 0;
1238   for(i=0; i<pSorter->nTask; i++){
1239     SortSubtask *pTask = &pSorter->aTask[i];
1240     vdbeSortSubtaskCleanup(db, pTask);
1241     pTask->pSorter = pSorter;
1242   }
1243   if( pSorter->list.aMemory==0 ){
1244     vdbeSorterRecordFree(0, pSorter->list.pList);
1245   }
1246   pSorter->list.pList = 0;
1247   pSorter->list.szPMA = 0;
1248   pSorter->bUsePMA = 0;
1249   pSorter->iMemory = 0;
1250   pSorter->mxKeysize = 0;
1251   sqlite3DbFree(db, pSorter->pUnpacked);
1252   pSorter->pUnpacked = 0;
1253 }
1254 
1255 /*
1256 ** Free any cursor components allocated by sqlite3VdbeSorterXXX routines.
1257 */
1258 void sqlite3VdbeSorterClose(sqlite3 *db, VdbeCursor *pCsr){
1259   VdbeSorter *pSorter;
1260   assert( pCsr->eCurType==CURTYPE_SORTER );
1261   pSorter = pCsr->uc.pSorter;
1262   if( pSorter ){
1263     sqlite3VdbeSorterReset(db, pSorter);
1264     sqlite3_free(pSorter->list.aMemory);
1265     sqlite3DbFree(db, pSorter);
1266     pCsr->uc.pSorter = 0;
1267   }
1268 }
1269 
1270 #if SQLITE_MAX_MMAP_SIZE>0
1271 /*
1272 ** The first argument is a file-handle open on a temporary file. The file
1273 ** is guaranteed to be nByte bytes or smaller in size. This function
1274 ** attempts to extend the file to nByte bytes in size and to ensure that
1275 ** the VFS has memory mapped it.
1276 **
1277 ** Whether or not the file does end up memory mapped of course depends on
1278 ** the specific VFS implementation.
1279 */
1280 static void vdbeSorterExtendFile(sqlite3 *db, sqlite3_file *pFd, i64 nByte){
1281   if( nByte<=(i64)(db->nMaxSorterMmap) && pFd->pMethods->iVersion>=3 ){
1282     void *p = 0;
1283     int chunksize = 4*1024;
1284     sqlite3OsFileControlHint(pFd, SQLITE_FCNTL_CHUNK_SIZE, &chunksize);
1285     sqlite3OsFileControlHint(pFd, SQLITE_FCNTL_SIZE_HINT, &nByte);
1286     sqlite3OsFetch(pFd, 0, (int)nByte, &p);
1287     sqlite3OsUnfetch(pFd, 0, p);
1288   }
1289 }
1290 #else
1291 # define vdbeSorterExtendFile(x,y,z)
1292 #endif
1293 
1294 /*
1295 ** Allocate space for a file-handle and open a temporary file. If successful,
1296 ** set *ppFd to point to the malloc'd file-handle and return SQLITE_OK.
1297 ** Otherwise, set *ppFd to 0 and return an SQLite error code.
1298 */
1299 static int vdbeSorterOpenTempFile(
1300   sqlite3 *db,                    /* Database handle doing sort */
1301   i64 nExtend,                    /* Attempt to extend file to this size */
1302   sqlite3_file **ppFd
1303 ){
1304   int rc;
1305   if( sqlite3FaultSim(202) ) return SQLITE_IOERR_ACCESS;
1306   rc = sqlite3OsOpenMalloc(db->pVfs, 0, ppFd,
1307       SQLITE_OPEN_TEMP_JOURNAL |
1308       SQLITE_OPEN_READWRITE    | SQLITE_OPEN_CREATE |
1309       SQLITE_OPEN_EXCLUSIVE    | SQLITE_OPEN_DELETEONCLOSE, &rc
1310   );
1311   if( rc==SQLITE_OK ){
1312     i64 max = SQLITE_MAX_MMAP_SIZE;
1313     sqlite3OsFileControlHint(*ppFd, SQLITE_FCNTL_MMAP_SIZE, (void*)&max);
1314     if( nExtend>0 ){
1315       vdbeSorterExtendFile(db, *ppFd, nExtend);
1316     }
1317   }
1318   return rc;
1319 }
1320 
1321 /*
1322 ** If it has not already been allocated, allocate the UnpackedRecord
1323 ** structure at pTask->pUnpacked. Return SQLITE_OK if successful (or
1324 ** if no allocation was required), or SQLITE_NOMEM otherwise.
1325 */
1326 static int vdbeSortAllocUnpacked(SortSubtask *pTask){
1327   if( pTask->pUnpacked==0 ){
1328     pTask->pUnpacked = sqlite3VdbeAllocUnpackedRecord(pTask->pSorter->pKeyInfo);
1329     if( pTask->pUnpacked==0 ) return SQLITE_NOMEM_BKPT;
1330     pTask->pUnpacked->nField = pTask->pSorter->pKeyInfo->nKeyField;
1331     pTask->pUnpacked->errCode = 0;
1332   }
1333   return SQLITE_OK;
1334 }
1335 
1336 
1337 /*
1338 ** Merge the two sorted lists p1 and p2 into a single list.
1339 */
1340 static SorterRecord *vdbeSorterMerge(
1341   SortSubtask *pTask,             /* Calling thread context */
1342   SorterRecord *p1,               /* First list to merge */
1343   SorterRecord *p2                /* Second list to merge */
1344 ){
1345   SorterRecord *pFinal = 0;
1346   SorterRecord **pp = &pFinal;
1347   int bCached = 0;
1348 
1349   assert( p1!=0 && p2!=0 );
1350   for(;;){
1351     int res;
1352     res = pTask->xCompare(
1353         pTask, &bCached, SRVAL(p1), p1->nVal, SRVAL(p2), p2->nVal
1354     );
1355 
1356     if( res<=0 ){
1357       *pp = p1;
1358       pp = &p1->u.pNext;
1359       p1 = p1->u.pNext;
1360       if( p1==0 ){
1361         *pp = p2;
1362         break;
1363       }
1364     }else{
1365       *pp = p2;
1366       pp = &p2->u.pNext;
1367       p2 = p2->u.pNext;
1368       bCached = 0;
1369       if( p2==0 ){
1370         *pp = p1;
1371         break;
1372       }
1373     }
1374   }
1375   return pFinal;
1376 }
1377 
1378 /*
1379 ** Return the SorterCompare function to compare values collected by the
1380 ** sorter object passed as the only argument.
1381 */
1382 static SorterCompare vdbeSorterGetCompare(VdbeSorter *p){
1383   if( p->typeMask==SORTER_TYPE_INTEGER ){
1384     return vdbeSorterCompareInt;
1385   }else if( p->typeMask==SORTER_TYPE_TEXT ){
1386     return vdbeSorterCompareText;
1387   }
1388   return vdbeSorterCompare;
1389 }
1390 
1391 /*
1392 ** Sort the linked list of records headed at pTask->pList. Return
1393 ** SQLITE_OK if successful, or an SQLite error code (i.e. SQLITE_NOMEM) if
1394 ** an error occurs.
1395 */
1396 static int vdbeSorterSort(SortSubtask *pTask, SorterList *pList){
1397   int i;
1398   SorterRecord **aSlot;
1399   SorterRecord *p;
1400   int rc;
1401 
1402   rc = vdbeSortAllocUnpacked(pTask);
1403   if( rc!=SQLITE_OK ) return rc;
1404 
1405   p = pList->pList;
1406   pTask->xCompare = vdbeSorterGetCompare(pTask->pSorter);
1407 
1408   aSlot = (SorterRecord **)sqlite3MallocZero(64 * sizeof(SorterRecord *));
1409   if( !aSlot ){
1410     return SQLITE_NOMEM_BKPT;
1411   }
1412 
1413   while( p ){
1414     SorterRecord *pNext;
1415     if( pList->aMemory ){
1416       if( (u8*)p==pList->aMemory ){
1417         pNext = 0;
1418       }else{
1419         assert( p->u.iNext<sqlite3MallocSize(pList->aMemory) );
1420         pNext = (SorterRecord*)&pList->aMemory[p->u.iNext];
1421       }
1422     }else{
1423       pNext = p->u.pNext;
1424     }
1425 
1426     p->u.pNext = 0;
1427     for(i=0; aSlot[i]; i++){
1428       p = vdbeSorterMerge(pTask, p, aSlot[i]);
1429       aSlot[i] = 0;
1430     }
1431     aSlot[i] = p;
1432     p = pNext;
1433   }
1434 
1435   p = 0;
1436   for(i=0; i<64; i++){
1437     if( aSlot[i]==0 ) continue;
1438     p = p ? vdbeSorterMerge(pTask, p, aSlot[i]) : aSlot[i];
1439   }
1440   pList->pList = p;
1441 
1442   sqlite3_free(aSlot);
1443   assert( pTask->pUnpacked->errCode==SQLITE_OK
1444        || pTask->pUnpacked->errCode==SQLITE_NOMEM
1445   );
1446   return pTask->pUnpacked->errCode;
1447 }
1448 
1449 /*
1450 ** Initialize a PMA-writer object.
1451 */
1452 static void vdbePmaWriterInit(
1453   sqlite3_file *pFd,              /* File handle to write to */
1454   PmaWriter *p,                   /* Object to populate */
1455   int nBuf,                       /* Buffer size */
1456   i64 iStart                      /* Offset of pFd to begin writing at */
1457 ){
1458   memset(p, 0, sizeof(PmaWriter));
1459   p->aBuffer = (u8*)sqlite3Malloc(nBuf);
1460   if( !p->aBuffer ){
1461     p->eFWErr = SQLITE_NOMEM_BKPT;
1462   }else{
1463     p->iBufEnd = p->iBufStart = (iStart % nBuf);
1464     p->iWriteOff = iStart - p->iBufStart;
1465     p->nBuffer = nBuf;
1466     p->pFd = pFd;
1467   }
1468 }
1469 
1470 /*
1471 ** Write nData bytes of data to the PMA. Return SQLITE_OK
1472 ** if successful, or an SQLite error code if an error occurs.
1473 */
1474 static void vdbePmaWriteBlob(PmaWriter *p, u8 *pData, int nData){
1475   int nRem = nData;
1476   while( nRem>0 && p->eFWErr==0 ){
1477     int nCopy = nRem;
1478     if( nCopy>(p->nBuffer - p->iBufEnd) ){
1479       nCopy = p->nBuffer - p->iBufEnd;
1480     }
1481 
1482     memcpy(&p->aBuffer[p->iBufEnd], &pData[nData-nRem], nCopy);
1483     p->iBufEnd += nCopy;
1484     if( p->iBufEnd==p->nBuffer ){
1485       p->eFWErr = sqlite3OsWrite(p->pFd,
1486           &p->aBuffer[p->iBufStart], p->iBufEnd - p->iBufStart,
1487           p->iWriteOff + p->iBufStart
1488       );
1489       p->iBufStart = p->iBufEnd = 0;
1490       p->iWriteOff += p->nBuffer;
1491     }
1492     assert( p->iBufEnd<p->nBuffer );
1493 
1494     nRem -= nCopy;
1495   }
1496 }
1497 
1498 /*
1499 ** Flush any buffered data to disk and clean up the PMA-writer object.
1500 ** The results of using the PMA-writer after this call are undefined.
1501 ** Return SQLITE_OK if flushing the buffered data succeeds or is not
1502 ** required. Otherwise, return an SQLite error code.
1503 **
1504 ** Before returning, set *piEof to the offset immediately following the
1505 ** last byte written to the file.
1506 */
1507 static int vdbePmaWriterFinish(PmaWriter *p, i64 *piEof){
1508   int rc;
1509   if( p->eFWErr==0 && ALWAYS(p->aBuffer) && p->iBufEnd>p->iBufStart ){
1510     p->eFWErr = sqlite3OsWrite(p->pFd,
1511         &p->aBuffer[p->iBufStart], p->iBufEnd - p->iBufStart,
1512         p->iWriteOff + p->iBufStart
1513     );
1514   }
1515   *piEof = (p->iWriteOff + p->iBufEnd);
1516   sqlite3_free(p->aBuffer);
1517   rc = p->eFWErr;
1518   memset(p, 0, sizeof(PmaWriter));
1519   return rc;
1520 }
1521 
1522 /*
1523 ** Write value iVal encoded as a varint to the PMA. Return
1524 ** SQLITE_OK if successful, or an SQLite error code if an error occurs.
1525 */
1526 static void vdbePmaWriteVarint(PmaWriter *p, u64 iVal){
1527   int nByte;
1528   u8 aByte[10];
1529   nByte = sqlite3PutVarint(aByte, iVal);
1530   vdbePmaWriteBlob(p, aByte, nByte);
1531 }
1532 
1533 /*
1534 ** Write the current contents of in-memory linked-list pList to a level-0
1535 ** PMA in the temp file belonging to sub-task pTask. Return SQLITE_OK if
1536 ** successful, or an SQLite error code otherwise.
1537 **
1538 ** The format of a PMA is:
1539 **
1540 **     * A varint. This varint contains the total number of bytes of content
1541 **       in the PMA (not including the varint itself).
1542 **
1543 **     * One or more records packed end-to-end in order of ascending keys.
1544 **       Each record consists of a varint followed by a blob of data (the
1545 **       key). The varint is the number of bytes in the blob of data.
1546 */
1547 static int vdbeSorterListToPMA(SortSubtask *pTask, SorterList *pList){
1548   sqlite3 *db = pTask->pSorter->db;
1549   int rc = SQLITE_OK;             /* Return code */
1550   PmaWriter writer;               /* Object used to write to the file */
1551 
1552 #ifdef SQLITE_DEBUG
1553   /* Set iSz to the expected size of file pTask->file after writing the PMA.
1554   ** This is used by an assert() statement at the end of this function.  */
1555   i64 iSz = pList->szPMA + sqlite3VarintLen(pList->szPMA) + pTask->file.iEof;
1556 #endif
1557 
1558   vdbeSorterWorkDebug(pTask, "enter");
1559   memset(&writer, 0, sizeof(PmaWriter));
1560   assert( pList->szPMA>0 );
1561 
1562   /* If the first temporary PMA file has not been opened, open it now. */
1563   if( pTask->file.pFd==0 ){
1564     rc = vdbeSorterOpenTempFile(db, 0, &pTask->file.pFd);
1565     assert( rc!=SQLITE_OK || pTask->file.pFd );
1566     assert( pTask->file.iEof==0 );
1567     assert( pTask->nPMA==0 );
1568   }
1569 
1570   /* Try to get the file to memory map */
1571   if( rc==SQLITE_OK ){
1572     vdbeSorterExtendFile(db, pTask->file.pFd, pTask->file.iEof+pList->szPMA+9);
1573   }
1574 
1575   /* Sort the list */
1576   if( rc==SQLITE_OK ){
1577     rc = vdbeSorterSort(pTask, pList);
1578   }
1579 
1580   if( rc==SQLITE_OK ){
1581     SorterRecord *p;
1582     SorterRecord *pNext = 0;
1583 
1584     vdbePmaWriterInit(pTask->file.pFd, &writer, pTask->pSorter->pgsz,
1585                       pTask->file.iEof);
1586     pTask->nPMA++;
1587     vdbePmaWriteVarint(&writer, pList->szPMA);
1588     for(p=pList->pList; p; p=pNext){
1589       pNext = p->u.pNext;
1590       vdbePmaWriteVarint(&writer, p->nVal);
1591       vdbePmaWriteBlob(&writer, SRVAL(p), p->nVal);
1592       if( pList->aMemory==0 ) sqlite3_free(p);
1593     }
1594     pList->pList = p;
1595     rc = vdbePmaWriterFinish(&writer, &pTask->file.iEof);
1596   }
1597 
1598   vdbeSorterWorkDebug(pTask, "exit");
1599   assert( rc!=SQLITE_OK || pList->pList==0 );
1600   assert( rc!=SQLITE_OK || pTask->file.iEof==iSz );
1601   return rc;
1602 }
1603 
1604 /*
1605 ** Advance the MergeEngine to its next entry.
1606 ** Set *pbEof to true there is no next entry because
1607 ** the MergeEngine has reached the end of all its inputs.
1608 **
1609 ** Return SQLITE_OK if successful or an error code if an error occurs.
1610 */
1611 static int vdbeMergeEngineStep(
1612   MergeEngine *pMerger,      /* The merge engine to advance to the next row */
1613   int *pbEof                 /* Set TRUE at EOF.  Set false for more content */
1614 ){
1615   int rc;
1616   int iPrev = pMerger->aTree[1];/* Index of PmaReader to advance */
1617   SortSubtask *pTask = pMerger->pTask;
1618 
1619   /* Advance the current PmaReader */
1620   rc = vdbePmaReaderNext(&pMerger->aReadr[iPrev]);
1621 
1622   /* Update contents of aTree[] */
1623   if( rc==SQLITE_OK ){
1624     int i;                      /* Index of aTree[] to recalculate */
1625     PmaReader *pReadr1;         /* First PmaReader to compare */
1626     PmaReader *pReadr2;         /* Second PmaReader to compare */
1627     int bCached = 0;
1628 
1629     /* Find the first two PmaReaders to compare. The one that was just
1630     ** advanced (iPrev) and the one next to it in the array.  */
1631     pReadr1 = &pMerger->aReadr[(iPrev & 0xFFFE)];
1632     pReadr2 = &pMerger->aReadr[(iPrev | 0x0001)];
1633 
1634     for(i=(pMerger->nTree+iPrev)/2; i>0; i=i/2){
1635       /* Compare pReadr1 and pReadr2. Store the result in variable iRes. */
1636       int iRes;
1637       if( pReadr1->pFd==0 ){
1638         iRes = +1;
1639       }else if( pReadr2->pFd==0 ){
1640         iRes = -1;
1641       }else{
1642         iRes = pTask->xCompare(pTask, &bCached,
1643             pReadr1->aKey, pReadr1->nKey, pReadr2->aKey, pReadr2->nKey
1644         );
1645       }
1646 
1647       /* If pReadr1 contained the smaller value, set aTree[i] to its index.
1648       ** Then set pReadr2 to the next PmaReader to compare to pReadr1. In this
1649       ** case there is no cache of pReadr2 in pTask->pUnpacked, so set
1650       ** pKey2 to point to the record belonging to pReadr2.
1651       **
1652       ** Alternatively, if pReadr2 contains the smaller of the two values,
1653       ** set aTree[i] to its index and update pReadr1. If vdbeSorterCompare()
1654       ** was actually called above, then pTask->pUnpacked now contains
1655       ** a value equivalent to pReadr2. So set pKey2 to NULL to prevent
1656       ** vdbeSorterCompare() from decoding pReadr2 again.
1657       **
1658       ** If the two values were equal, then the value from the oldest
1659       ** PMA should be considered smaller. The VdbeSorter.aReadr[] array
1660       ** is sorted from oldest to newest, so pReadr1 contains older values
1661       ** than pReadr2 iff (pReadr1<pReadr2).  */
1662       if( iRes<0 || (iRes==0 && pReadr1<pReadr2) ){
1663         pMerger->aTree[i] = (int)(pReadr1 - pMerger->aReadr);
1664         pReadr2 = &pMerger->aReadr[ pMerger->aTree[i ^ 0x0001] ];
1665         bCached = 0;
1666       }else{
1667         if( pReadr1->pFd ) bCached = 0;
1668         pMerger->aTree[i] = (int)(pReadr2 - pMerger->aReadr);
1669         pReadr1 = &pMerger->aReadr[ pMerger->aTree[i ^ 0x0001] ];
1670       }
1671     }
1672     *pbEof = (pMerger->aReadr[pMerger->aTree[1]].pFd==0);
1673   }
1674 
1675   return (rc==SQLITE_OK ? pTask->pUnpacked->errCode : rc);
1676 }
1677 
1678 #if SQLITE_MAX_WORKER_THREADS>0
1679 /*
1680 ** The main routine for background threads that write level-0 PMAs.
1681 */
1682 static void *vdbeSorterFlushThread(void *pCtx){
1683   SortSubtask *pTask = (SortSubtask*)pCtx;
1684   int rc;                         /* Return code */
1685   assert( pTask->bDone==0 );
1686   rc = vdbeSorterListToPMA(pTask, &pTask->list);
1687   pTask->bDone = 1;
1688   return SQLITE_INT_TO_PTR(rc);
1689 }
1690 #endif /* SQLITE_MAX_WORKER_THREADS>0 */
1691 
1692 /*
1693 ** Flush the current contents of VdbeSorter.list to a new PMA, possibly
1694 ** using a background thread.
1695 */
1696 static int vdbeSorterFlushPMA(VdbeSorter *pSorter){
1697 #if SQLITE_MAX_WORKER_THREADS==0
1698   pSorter->bUsePMA = 1;
1699   return vdbeSorterListToPMA(&pSorter->aTask[0], &pSorter->list);
1700 #else
1701   int rc = SQLITE_OK;
1702   int i;
1703   SortSubtask *pTask = 0;    /* Thread context used to create new PMA */
1704   int nWorker = (pSorter->nTask-1);
1705 
1706   /* Set the flag to indicate that at least one PMA has been written.
1707   ** Or will be, anyhow.  */
1708   pSorter->bUsePMA = 1;
1709 
1710   /* Select a sub-task to sort and flush the current list of in-memory
1711   ** records to disk. If the sorter is running in multi-threaded mode,
1712   ** round-robin between the first (pSorter->nTask-1) tasks. Except, if
1713   ** the background thread from a sub-tasks previous turn is still running,
1714   ** skip it. If the first (pSorter->nTask-1) sub-tasks are all still busy,
1715   ** fall back to using the final sub-task. The first (pSorter->nTask-1)
1716   ** sub-tasks are prefered as they use background threads - the final
1717   ** sub-task uses the main thread. */
1718   for(i=0; i<nWorker; i++){
1719     int iTest = (pSorter->iPrev + i + 1) % nWorker;
1720     pTask = &pSorter->aTask[iTest];
1721     if( pTask->bDone ){
1722       rc = vdbeSorterJoinThread(pTask);
1723     }
1724     if( rc!=SQLITE_OK || pTask->pThread==0 ) break;
1725   }
1726 
1727   if( rc==SQLITE_OK ){
1728     if( i==nWorker ){
1729       /* Use the foreground thread for this operation */
1730       rc = vdbeSorterListToPMA(&pSorter->aTask[nWorker], &pSorter->list);
1731     }else{
1732       /* Launch a background thread for this operation */
1733       u8 *aMem = pTask->list.aMemory;
1734       void *pCtx = (void*)pTask;
1735 
1736       assert( pTask->pThread==0 && pTask->bDone==0 );
1737       assert( pTask->list.pList==0 );
1738       assert( pTask->list.aMemory==0 || pSorter->list.aMemory!=0 );
1739 
1740       pSorter->iPrev = (u8)(pTask - pSorter->aTask);
1741       pTask->list = pSorter->list;
1742       pSorter->list.pList = 0;
1743       pSorter->list.szPMA = 0;
1744       if( aMem ){
1745         pSorter->list.aMemory = aMem;
1746         pSorter->nMemory = sqlite3MallocSize(aMem);
1747       }else if( pSorter->list.aMemory ){
1748         pSorter->list.aMemory = sqlite3Malloc(pSorter->nMemory);
1749         if( !pSorter->list.aMemory ) return SQLITE_NOMEM_BKPT;
1750       }
1751 
1752       rc = vdbeSorterCreateThread(pTask, vdbeSorterFlushThread, pCtx);
1753     }
1754   }
1755 
1756   return rc;
1757 #endif /* SQLITE_MAX_WORKER_THREADS!=0 */
1758 }
1759 
1760 /*
1761 ** Add a record to the sorter.
1762 */
1763 int sqlite3VdbeSorterWrite(
1764   const VdbeCursor *pCsr,         /* Sorter cursor */
1765   Mem *pVal                       /* Memory cell containing record */
1766 ){
1767   VdbeSorter *pSorter;
1768   int rc = SQLITE_OK;             /* Return Code */
1769   SorterRecord *pNew;             /* New list element */
1770   int bFlush;                     /* True to flush contents of memory to PMA */
1771   int nReq;                       /* Bytes of memory required */
1772   int nPMA;                       /* Bytes of PMA space required */
1773   int t;                          /* serial type of first record field */
1774 
1775   assert( pCsr->eCurType==CURTYPE_SORTER );
1776   pSorter = pCsr->uc.pSorter;
1777   getVarint32((const u8*)&pVal->z[1], t);
1778   if( t>0 && t<10 && t!=7 ){
1779     pSorter->typeMask &= SORTER_TYPE_INTEGER;
1780   }else if( t>10 && (t & 0x01) ){
1781     pSorter->typeMask &= SORTER_TYPE_TEXT;
1782   }else{
1783     pSorter->typeMask = 0;
1784   }
1785 
1786   assert( pSorter );
1787 
1788   /* Figure out whether or not the current contents of memory should be
1789   ** flushed to a PMA before continuing. If so, do so.
1790   **
1791   ** If using the single large allocation mode (pSorter->aMemory!=0), then
1792   ** flush the contents of memory to a new PMA if (a) at least one value is
1793   ** already in memory and (b) the new value will not fit in memory.
1794   **
1795   ** Or, if using separate allocations for each record, flush the contents
1796   ** of memory to a PMA if either of the following are true:
1797   **
1798   **   * The total memory allocated for the in-memory list is greater
1799   **     than (page-size * cache-size), or
1800   **
1801   **   * The total memory allocated for the in-memory list is greater
1802   **     than (page-size * 10) and sqlite3HeapNearlyFull() returns true.
1803   */
1804   nReq = pVal->n + sizeof(SorterRecord);
1805   nPMA = pVal->n + sqlite3VarintLen(pVal->n);
1806   if( pSorter->mxPmaSize ){
1807     if( pSorter->list.aMemory ){
1808       bFlush = pSorter->iMemory && (pSorter->iMemory+nReq) > pSorter->mxPmaSize;
1809     }else{
1810       bFlush = (
1811           (pSorter->list.szPMA > pSorter->mxPmaSize)
1812        || (pSorter->list.szPMA > pSorter->mnPmaSize && sqlite3HeapNearlyFull())
1813       );
1814     }
1815     if( bFlush ){
1816       rc = vdbeSorterFlushPMA(pSorter);
1817       pSorter->list.szPMA = 0;
1818       pSorter->iMemory = 0;
1819       assert( rc!=SQLITE_OK || pSorter->list.pList==0 );
1820     }
1821   }
1822 
1823   pSorter->list.szPMA += nPMA;
1824   if( nPMA>pSorter->mxKeysize ){
1825     pSorter->mxKeysize = nPMA;
1826   }
1827 
1828   if( pSorter->list.aMemory ){
1829     int nMin = pSorter->iMemory + nReq;
1830 
1831     if( nMin>pSorter->nMemory ){
1832       u8 *aNew;
1833       int iListOff = (u8*)pSorter->list.pList - pSorter->list.aMemory;
1834       int nNew = pSorter->nMemory * 2;
1835       while( nNew < nMin ) nNew = nNew*2;
1836       if( nNew > pSorter->mxPmaSize ) nNew = pSorter->mxPmaSize;
1837       if( nNew < nMin ) nNew = nMin;
1838 
1839       aNew = sqlite3Realloc(pSorter->list.aMemory, nNew);
1840       if( !aNew ) return SQLITE_NOMEM_BKPT;
1841       pSorter->list.pList = (SorterRecord*)&aNew[iListOff];
1842       pSorter->list.aMemory = aNew;
1843       pSorter->nMemory = nNew;
1844     }
1845 
1846     pNew = (SorterRecord*)&pSorter->list.aMemory[pSorter->iMemory];
1847     pSorter->iMemory += ROUND8(nReq);
1848     if( pSorter->list.pList ){
1849       pNew->u.iNext = (int)((u8*)(pSorter->list.pList) - pSorter->list.aMemory);
1850     }
1851   }else{
1852     pNew = (SorterRecord *)sqlite3Malloc(nReq);
1853     if( pNew==0 ){
1854       return SQLITE_NOMEM_BKPT;
1855     }
1856     pNew->u.pNext = pSorter->list.pList;
1857   }
1858 
1859   memcpy(SRVAL(pNew), pVal->z, pVal->n);
1860   pNew->nVal = pVal->n;
1861   pSorter->list.pList = pNew;
1862 
1863   return rc;
1864 }
1865 
1866 /*
1867 ** Read keys from pIncr->pMerger and populate pIncr->aFile[1]. The format
1868 ** of the data stored in aFile[1] is the same as that used by regular PMAs,
1869 ** except that the number-of-bytes varint is omitted from the start.
1870 */
1871 static int vdbeIncrPopulate(IncrMerger *pIncr){
1872   int rc = SQLITE_OK;
1873   int rc2;
1874   i64 iStart = pIncr->iStartOff;
1875   SorterFile *pOut = &pIncr->aFile[1];
1876   SortSubtask *pTask = pIncr->pTask;
1877   MergeEngine *pMerger = pIncr->pMerger;
1878   PmaWriter writer;
1879   assert( pIncr->bEof==0 );
1880 
1881   vdbeSorterPopulateDebug(pTask, "enter");
1882 
1883   vdbePmaWriterInit(pOut->pFd, &writer, pTask->pSorter->pgsz, iStart);
1884   while( rc==SQLITE_OK ){
1885     int dummy;
1886     PmaReader *pReader = &pMerger->aReadr[ pMerger->aTree[1] ];
1887     int nKey = pReader->nKey;
1888     i64 iEof = writer.iWriteOff + writer.iBufEnd;
1889 
1890     /* Check if the output file is full or if the input has been exhausted.
1891     ** In either case exit the loop. */
1892     if( pReader->pFd==0 ) break;
1893     if( (iEof + nKey + sqlite3VarintLen(nKey))>(iStart + pIncr->mxSz) ) break;
1894 
1895     /* Write the next key to the output. */
1896     vdbePmaWriteVarint(&writer, nKey);
1897     vdbePmaWriteBlob(&writer, pReader->aKey, nKey);
1898     assert( pIncr->pMerger->pTask==pTask );
1899     rc = vdbeMergeEngineStep(pIncr->pMerger, &dummy);
1900   }
1901 
1902   rc2 = vdbePmaWriterFinish(&writer, &pOut->iEof);
1903   if( rc==SQLITE_OK ) rc = rc2;
1904   vdbeSorterPopulateDebug(pTask, "exit");
1905   return rc;
1906 }
1907 
1908 #if SQLITE_MAX_WORKER_THREADS>0
1909 /*
1910 ** The main routine for background threads that populate aFile[1] of
1911 ** multi-threaded IncrMerger objects.
1912 */
1913 static void *vdbeIncrPopulateThread(void *pCtx){
1914   IncrMerger *pIncr = (IncrMerger*)pCtx;
1915   void *pRet = SQLITE_INT_TO_PTR( vdbeIncrPopulate(pIncr) );
1916   pIncr->pTask->bDone = 1;
1917   return pRet;
1918 }
1919 
1920 /*
1921 ** Launch a background thread to populate aFile[1] of pIncr.
1922 */
1923 static int vdbeIncrBgPopulate(IncrMerger *pIncr){
1924   void *p = (void*)pIncr;
1925   assert( pIncr->bUseThread );
1926   return vdbeSorterCreateThread(pIncr->pTask, vdbeIncrPopulateThread, p);
1927 }
1928 #endif
1929 
1930 /*
1931 ** This function is called when the PmaReader corresponding to pIncr has
1932 ** finished reading the contents of aFile[0]. Its purpose is to "refill"
1933 ** aFile[0] such that the PmaReader should start rereading it from the
1934 ** beginning.
1935 **
1936 ** For single-threaded objects, this is accomplished by literally reading
1937 ** keys from pIncr->pMerger and repopulating aFile[0].
1938 **
1939 ** For multi-threaded objects, all that is required is to wait until the
1940 ** background thread is finished (if it is not already) and then swap
1941 ** aFile[0] and aFile[1] in place. If the contents of pMerger have not
1942 ** been exhausted, this function also launches a new background thread
1943 ** to populate the new aFile[1].
1944 **
1945 ** SQLITE_OK is returned on success, or an SQLite error code otherwise.
1946 */
1947 static int vdbeIncrSwap(IncrMerger *pIncr){
1948   int rc = SQLITE_OK;
1949 
1950 #if SQLITE_MAX_WORKER_THREADS>0
1951   if( pIncr->bUseThread ){
1952     rc = vdbeSorterJoinThread(pIncr->pTask);
1953 
1954     if( rc==SQLITE_OK ){
1955       SorterFile f0 = pIncr->aFile[0];
1956       pIncr->aFile[0] = pIncr->aFile[1];
1957       pIncr->aFile[1] = f0;
1958     }
1959 
1960     if( rc==SQLITE_OK ){
1961       if( pIncr->aFile[0].iEof==pIncr->iStartOff ){
1962         pIncr->bEof = 1;
1963       }else{
1964         rc = vdbeIncrBgPopulate(pIncr);
1965       }
1966     }
1967   }else
1968 #endif
1969   {
1970     rc = vdbeIncrPopulate(pIncr);
1971     pIncr->aFile[0] = pIncr->aFile[1];
1972     if( pIncr->aFile[0].iEof==pIncr->iStartOff ){
1973       pIncr->bEof = 1;
1974     }
1975   }
1976 
1977   return rc;
1978 }
1979 
1980 /*
1981 ** Allocate and return a new IncrMerger object to read data from pMerger.
1982 **
1983 ** If an OOM condition is encountered, return NULL. In this case free the
1984 ** pMerger argument before returning.
1985 */
1986 static int vdbeIncrMergerNew(
1987   SortSubtask *pTask,     /* The thread that will be using the new IncrMerger */
1988   MergeEngine *pMerger,   /* The MergeEngine that the IncrMerger will control */
1989   IncrMerger **ppOut      /* Write the new IncrMerger here */
1990 ){
1991   int rc = SQLITE_OK;
1992   IncrMerger *pIncr = *ppOut = (IncrMerger*)
1993        (sqlite3FaultSim(100) ? 0 : sqlite3MallocZero(sizeof(*pIncr)));
1994   if( pIncr ){
1995     pIncr->pMerger = pMerger;
1996     pIncr->pTask = pTask;
1997     pIncr->mxSz = MAX(pTask->pSorter->mxKeysize+9,pTask->pSorter->mxPmaSize/2);
1998     pTask->file2.iEof += pIncr->mxSz;
1999   }else{
2000     vdbeMergeEngineFree(pMerger);
2001     rc = SQLITE_NOMEM_BKPT;
2002   }
2003   return rc;
2004 }
2005 
2006 #if SQLITE_MAX_WORKER_THREADS>0
2007 /*
2008 ** Set the "use-threads" flag on object pIncr.
2009 */
2010 static void vdbeIncrMergerSetThreads(IncrMerger *pIncr){
2011   pIncr->bUseThread = 1;
2012   pIncr->pTask->file2.iEof -= pIncr->mxSz;
2013 }
2014 #endif /* SQLITE_MAX_WORKER_THREADS>0 */
2015 
2016 
2017 
2018 /*
2019 ** Recompute pMerger->aTree[iOut] by comparing the next keys on the
2020 ** two PmaReaders that feed that entry.  Neither of the PmaReaders
2021 ** are advanced.  This routine merely does the comparison.
2022 */
2023 static void vdbeMergeEngineCompare(
2024   MergeEngine *pMerger,  /* Merge engine containing PmaReaders to compare */
2025   int iOut               /* Store the result in pMerger->aTree[iOut] */
2026 ){
2027   int i1;
2028   int i2;
2029   int iRes;
2030   PmaReader *p1;
2031   PmaReader *p2;
2032 
2033   assert( iOut<pMerger->nTree && iOut>0 );
2034 
2035   if( iOut>=(pMerger->nTree/2) ){
2036     i1 = (iOut - pMerger->nTree/2) * 2;
2037     i2 = i1 + 1;
2038   }else{
2039     i1 = pMerger->aTree[iOut*2];
2040     i2 = pMerger->aTree[iOut*2+1];
2041   }
2042 
2043   p1 = &pMerger->aReadr[i1];
2044   p2 = &pMerger->aReadr[i2];
2045 
2046   if( p1->pFd==0 ){
2047     iRes = i2;
2048   }else if( p2->pFd==0 ){
2049     iRes = i1;
2050   }else{
2051     SortSubtask *pTask = pMerger->pTask;
2052     int bCached = 0;
2053     int res;
2054     assert( pTask->pUnpacked!=0 );  /* from vdbeSortSubtaskMain() */
2055     res = pTask->xCompare(
2056         pTask, &bCached, p1->aKey, p1->nKey, p2->aKey, p2->nKey
2057     );
2058     if( res<=0 ){
2059       iRes = i1;
2060     }else{
2061       iRes = i2;
2062     }
2063   }
2064 
2065   pMerger->aTree[iOut] = iRes;
2066 }
2067 
2068 /*
2069 ** Allowed values for the eMode parameter to vdbeMergeEngineInit()
2070 ** and vdbePmaReaderIncrMergeInit().
2071 **
2072 ** Only INCRINIT_NORMAL is valid in single-threaded builds (when
2073 ** SQLITE_MAX_WORKER_THREADS==0).  The other values are only used
2074 ** when there exists one or more separate worker threads.
2075 */
2076 #define INCRINIT_NORMAL 0
2077 #define INCRINIT_TASK   1
2078 #define INCRINIT_ROOT   2
2079 
2080 /*
2081 ** Forward reference required as the vdbeIncrMergeInit() and
2082 ** vdbePmaReaderIncrInit() routines are called mutually recursively when
2083 ** building a merge tree.
2084 */
2085 static int vdbePmaReaderIncrInit(PmaReader *pReadr, int eMode);
2086 
2087 /*
2088 ** Initialize the MergeEngine object passed as the second argument. Once this
2089 ** function returns, the first key of merged data may be read from the
2090 ** MergeEngine object in the usual fashion.
2091 **
2092 ** If argument eMode is INCRINIT_ROOT, then it is assumed that any IncrMerge
2093 ** objects attached to the PmaReader objects that the merger reads from have
2094 ** already been populated, but that they have not yet populated aFile[0] and
2095 ** set the PmaReader objects up to read from it. In this case all that is
2096 ** required is to call vdbePmaReaderNext() on each PmaReader to point it at
2097 ** its first key.
2098 **
2099 ** Otherwise, if eMode is any value other than INCRINIT_ROOT, then use
2100 ** vdbePmaReaderIncrMergeInit() to initialize each PmaReader that feeds data
2101 ** to pMerger.
2102 **
2103 ** SQLITE_OK is returned if successful, or an SQLite error code otherwise.
2104 */
2105 static int vdbeMergeEngineInit(
2106   SortSubtask *pTask,             /* Thread that will run pMerger */
2107   MergeEngine *pMerger,           /* MergeEngine to initialize */
2108   int eMode                       /* One of the INCRINIT_XXX constants */
2109 ){
2110   int rc = SQLITE_OK;             /* Return code */
2111   int i;                          /* For looping over PmaReader objects */
2112   int nTree = pMerger->nTree;
2113 
2114   /* eMode is always INCRINIT_NORMAL in single-threaded mode */
2115   assert( SQLITE_MAX_WORKER_THREADS>0 || eMode==INCRINIT_NORMAL );
2116 
2117   /* Verify that the MergeEngine is assigned to a single thread */
2118   assert( pMerger->pTask==0 );
2119   pMerger->pTask = pTask;
2120 
2121   for(i=0; i<nTree; i++){
2122     if( SQLITE_MAX_WORKER_THREADS>0 && eMode==INCRINIT_ROOT ){
2123       /* PmaReaders should be normally initialized in order, as if they are
2124       ** reading from the same temp file this makes for more linear file IO.
2125       ** However, in the INCRINIT_ROOT case, if PmaReader aReadr[nTask-1] is
2126       ** in use it will block the vdbePmaReaderNext() call while it uses
2127       ** the main thread to fill its buffer. So calling PmaReaderNext()
2128       ** on this PmaReader before any of the multi-threaded PmaReaders takes
2129       ** better advantage of multi-processor hardware. */
2130       rc = vdbePmaReaderNext(&pMerger->aReadr[nTree-i-1]);
2131     }else{
2132       rc = vdbePmaReaderIncrInit(&pMerger->aReadr[i], INCRINIT_NORMAL);
2133     }
2134     if( rc!=SQLITE_OK ) return rc;
2135   }
2136 
2137   for(i=pMerger->nTree-1; i>0; i--){
2138     vdbeMergeEngineCompare(pMerger, i);
2139   }
2140   return pTask->pUnpacked->errCode;
2141 }
2142 
2143 /*
2144 ** The PmaReader passed as the first argument is guaranteed to be an
2145 ** incremental-reader (pReadr->pIncr!=0). This function serves to open
2146 ** and/or initialize the temp file related fields of the IncrMerge
2147 ** object at (pReadr->pIncr).
2148 **
2149 ** If argument eMode is set to INCRINIT_NORMAL, then all PmaReaders
2150 ** in the sub-tree headed by pReadr are also initialized. Data is then
2151 ** loaded into the buffers belonging to pReadr and it is set to point to
2152 ** the first key in its range.
2153 **
2154 ** If argument eMode is set to INCRINIT_TASK, then pReadr is guaranteed
2155 ** to be a multi-threaded PmaReader and this function is being called in a
2156 ** background thread. In this case all PmaReaders in the sub-tree are
2157 ** initialized as for INCRINIT_NORMAL and the aFile[1] buffer belonging to
2158 ** pReadr is populated. However, pReadr itself is not set up to point
2159 ** to its first key. A call to vdbePmaReaderNext() is still required to do
2160 ** that.
2161 **
2162 ** The reason this function does not call vdbePmaReaderNext() immediately
2163 ** in the INCRINIT_TASK case is that vdbePmaReaderNext() assumes that it has
2164 ** to block on thread (pTask->thread) before accessing aFile[1]. But, since
2165 ** this entire function is being run by thread (pTask->thread), that will
2166 ** lead to the current background thread attempting to join itself.
2167 **
2168 ** Finally, if argument eMode is set to INCRINIT_ROOT, it may be assumed
2169 ** that pReadr->pIncr is a multi-threaded IncrMerge objects, and that all
2170 ** child-trees have already been initialized using IncrInit(INCRINIT_TASK).
2171 ** In this case vdbePmaReaderNext() is called on all child PmaReaders and
2172 ** the current PmaReader set to point to the first key in its range.
2173 **
2174 ** SQLITE_OK is returned if successful, or an SQLite error code otherwise.
2175 */
2176 static int vdbePmaReaderIncrMergeInit(PmaReader *pReadr, int eMode){
2177   int rc = SQLITE_OK;
2178   IncrMerger *pIncr = pReadr->pIncr;
2179   SortSubtask *pTask = pIncr->pTask;
2180   sqlite3 *db = pTask->pSorter->db;
2181 
2182   /* eMode is always INCRINIT_NORMAL in single-threaded mode */
2183   assert( SQLITE_MAX_WORKER_THREADS>0 || eMode==INCRINIT_NORMAL );
2184 
2185   rc = vdbeMergeEngineInit(pTask, pIncr->pMerger, eMode);
2186 
2187   /* Set up the required files for pIncr. A multi-theaded IncrMerge object
2188   ** requires two temp files to itself, whereas a single-threaded object
2189   ** only requires a region of pTask->file2. */
2190   if( rc==SQLITE_OK ){
2191     int mxSz = pIncr->mxSz;
2192 #if SQLITE_MAX_WORKER_THREADS>0
2193     if( pIncr->bUseThread ){
2194       rc = vdbeSorterOpenTempFile(db, mxSz, &pIncr->aFile[0].pFd);
2195       if( rc==SQLITE_OK ){
2196         rc = vdbeSorterOpenTempFile(db, mxSz, &pIncr->aFile[1].pFd);
2197       }
2198     }else
2199 #endif
2200     /*if( !pIncr->bUseThread )*/{
2201       if( pTask->file2.pFd==0 ){
2202         assert( pTask->file2.iEof>0 );
2203         rc = vdbeSorterOpenTempFile(db, pTask->file2.iEof, &pTask->file2.pFd);
2204         pTask->file2.iEof = 0;
2205       }
2206       if( rc==SQLITE_OK ){
2207         pIncr->aFile[1].pFd = pTask->file2.pFd;
2208         pIncr->iStartOff = pTask->file2.iEof;
2209         pTask->file2.iEof += mxSz;
2210       }
2211     }
2212   }
2213 
2214 #if SQLITE_MAX_WORKER_THREADS>0
2215   if( rc==SQLITE_OK && pIncr->bUseThread ){
2216     /* Use the current thread to populate aFile[1], even though this
2217     ** PmaReader is multi-threaded. If this is an INCRINIT_TASK object,
2218     ** then this function is already running in background thread
2219     ** pIncr->pTask->thread.
2220     **
2221     ** If this is the INCRINIT_ROOT object, then it is running in the
2222     ** main VDBE thread. But that is Ok, as that thread cannot return
2223     ** control to the VDBE or proceed with anything useful until the
2224     ** first results are ready from this merger object anyway.
2225     */
2226     assert( eMode==INCRINIT_ROOT || eMode==INCRINIT_TASK );
2227     rc = vdbeIncrPopulate(pIncr);
2228   }
2229 #endif
2230 
2231   if( rc==SQLITE_OK && (SQLITE_MAX_WORKER_THREADS==0 || eMode!=INCRINIT_TASK) ){
2232     rc = vdbePmaReaderNext(pReadr);
2233   }
2234 
2235   return rc;
2236 }
2237 
2238 #if SQLITE_MAX_WORKER_THREADS>0
2239 /*
2240 ** The main routine for vdbePmaReaderIncrMergeInit() operations run in
2241 ** background threads.
2242 */
2243 static void *vdbePmaReaderBgIncrInit(void *pCtx){
2244   PmaReader *pReader = (PmaReader*)pCtx;
2245   void *pRet = SQLITE_INT_TO_PTR(
2246                   vdbePmaReaderIncrMergeInit(pReader,INCRINIT_TASK)
2247                );
2248   pReader->pIncr->pTask->bDone = 1;
2249   return pRet;
2250 }
2251 #endif
2252 
2253 /*
2254 ** If the PmaReader passed as the first argument is not an incremental-reader
2255 ** (if pReadr->pIncr==0), then this function is a no-op. Otherwise, it invokes
2256 ** the vdbePmaReaderIncrMergeInit() function with the parameters passed to
2257 ** this routine to initialize the incremental merge.
2258 **
2259 ** If the IncrMerger object is multi-threaded (IncrMerger.bUseThread==1),
2260 ** then a background thread is launched to call vdbePmaReaderIncrMergeInit().
2261 ** Or, if the IncrMerger is single threaded, the same function is called
2262 ** using the current thread.
2263 */
2264 static int vdbePmaReaderIncrInit(PmaReader *pReadr, int eMode){
2265   IncrMerger *pIncr = pReadr->pIncr;   /* Incremental merger */
2266   int rc = SQLITE_OK;                  /* Return code */
2267   if( pIncr ){
2268 #if SQLITE_MAX_WORKER_THREADS>0
2269     assert( pIncr->bUseThread==0 || eMode==INCRINIT_TASK );
2270     if( pIncr->bUseThread ){
2271       void *pCtx = (void*)pReadr;
2272       rc = vdbeSorterCreateThread(pIncr->pTask, vdbePmaReaderBgIncrInit, pCtx);
2273     }else
2274 #endif
2275     {
2276       rc = vdbePmaReaderIncrMergeInit(pReadr, eMode);
2277     }
2278   }
2279   return rc;
2280 }
2281 
2282 /*
2283 ** Allocate a new MergeEngine object to merge the contents of nPMA level-0
2284 ** PMAs from pTask->file. If no error occurs, set *ppOut to point to
2285 ** the new object and return SQLITE_OK. Or, if an error does occur, set *ppOut
2286 ** to NULL and return an SQLite error code.
2287 **
2288 ** When this function is called, *piOffset is set to the offset of the
2289 ** first PMA to read from pTask->file. Assuming no error occurs, it is
2290 ** set to the offset immediately following the last byte of the last
2291 ** PMA before returning. If an error does occur, then the final value of
2292 ** *piOffset is undefined.
2293 */
2294 static int vdbeMergeEngineLevel0(
2295   SortSubtask *pTask,             /* Sorter task to read from */
2296   int nPMA,                       /* Number of PMAs to read */
2297   i64 *piOffset,                  /* IN/OUT: Readr offset in pTask->file */
2298   MergeEngine **ppOut             /* OUT: New merge-engine */
2299 ){
2300   MergeEngine *pNew;              /* Merge engine to return */
2301   i64 iOff = *piOffset;
2302   int i;
2303   int rc = SQLITE_OK;
2304 
2305   *ppOut = pNew = vdbeMergeEngineNew(nPMA);
2306   if( pNew==0 ) rc = SQLITE_NOMEM_BKPT;
2307 
2308   for(i=0; i<nPMA && rc==SQLITE_OK; i++){
2309     i64 nDummy = 0;
2310     PmaReader *pReadr = &pNew->aReadr[i];
2311     rc = vdbePmaReaderInit(pTask, &pTask->file, iOff, pReadr, &nDummy);
2312     iOff = pReadr->iEof;
2313   }
2314 
2315   if( rc!=SQLITE_OK ){
2316     vdbeMergeEngineFree(pNew);
2317     *ppOut = 0;
2318   }
2319   *piOffset = iOff;
2320   return rc;
2321 }
2322 
2323 /*
2324 ** Return the depth of a tree comprising nPMA PMAs, assuming a fanout of
2325 ** SORTER_MAX_MERGE_COUNT. The returned value does not include leaf nodes.
2326 **
2327 ** i.e.
2328 **
2329 **   nPMA<=16    -> TreeDepth() == 0
2330 **   nPMA<=256   -> TreeDepth() == 1
2331 **   nPMA<=65536 -> TreeDepth() == 2
2332 */
2333 static int vdbeSorterTreeDepth(int nPMA){
2334   int nDepth = 0;
2335   i64 nDiv = SORTER_MAX_MERGE_COUNT;
2336   while( nDiv < (i64)nPMA ){
2337     nDiv = nDiv * SORTER_MAX_MERGE_COUNT;
2338     nDepth++;
2339   }
2340   return nDepth;
2341 }
2342 
2343 /*
2344 ** pRoot is the root of an incremental merge-tree with depth nDepth (according
2345 ** to vdbeSorterTreeDepth()). pLeaf is the iSeq'th leaf to be added to the
2346 ** tree, counting from zero. This function adds pLeaf to the tree.
2347 **
2348 ** If successful, SQLITE_OK is returned. If an error occurs, an SQLite error
2349 ** code is returned and pLeaf is freed.
2350 */
2351 static int vdbeSorterAddToTree(
2352   SortSubtask *pTask,             /* Task context */
2353   int nDepth,                     /* Depth of tree according to TreeDepth() */
2354   int iSeq,                       /* Sequence number of leaf within tree */
2355   MergeEngine *pRoot,             /* Root of tree */
2356   MergeEngine *pLeaf              /* Leaf to add to tree */
2357 ){
2358   int rc = SQLITE_OK;
2359   int nDiv = 1;
2360   int i;
2361   MergeEngine *p = pRoot;
2362   IncrMerger *pIncr;
2363 
2364   rc = vdbeIncrMergerNew(pTask, pLeaf, &pIncr);
2365 
2366   for(i=1; i<nDepth; i++){
2367     nDiv = nDiv * SORTER_MAX_MERGE_COUNT;
2368   }
2369 
2370   for(i=1; i<nDepth && rc==SQLITE_OK; i++){
2371     int iIter = (iSeq / nDiv) % SORTER_MAX_MERGE_COUNT;
2372     PmaReader *pReadr = &p->aReadr[iIter];
2373 
2374     if( pReadr->pIncr==0 ){
2375       MergeEngine *pNew = vdbeMergeEngineNew(SORTER_MAX_MERGE_COUNT);
2376       if( pNew==0 ){
2377         rc = SQLITE_NOMEM_BKPT;
2378       }else{
2379         rc = vdbeIncrMergerNew(pTask, pNew, &pReadr->pIncr);
2380       }
2381     }
2382     if( rc==SQLITE_OK ){
2383       p = pReadr->pIncr->pMerger;
2384       nDiv = nDiv / SORTER_MAX_MERGE_COUNT;
2385     }
2386   }
2387 
2388   if( rc==SQLITE_OK ){
2389     p->aReadr[iSeq % SORTER_MAX_MERGE_COUNT].pIncr = pIncr;
2390   }else{
2391     vdbeIncrFree(pIncr);
2392   }
2393   return rc;
2394 }
2395 
2396 /*
2397 ** This function is called as part of a SorterRewind() operation on a sorter
2398 ** that has already written two or more level-0 PMAs to one or more temp
2399 ** files. It builds a tree of MergeEngine/IncrMerger/PmaReader objects that
2400 ** can be used to incrementally merge all PMAs on disk.
2401 **
2402 ** If successful, SQLITE_OK is returned and *ppOut set to point to the
2403 ** MergeEngine object at the root of the tree before returning. Or, if an
2404 ** error occurs, an SQLite error code is returned and the final value
2405 ** of *ppOut is undefined.
2406 */
2407 static int vdbeSorterMergeTreeBuild(
2408   VdbeSorter *pSorter,       /* The VDBE cursor that implements the sort */
2409   MergeEngine **ppOut        /* Write the MergeEngine here */
2410 ){
2411   MergeEngine *pMain = 0;
2412   int rc = SQLITE_OK;
2413   int iTask;
2414 
2415 #if SQLITE_MAX_WORKER_THREADS>0
2416   /* If the sorter uses more than one task, then create the top-level
2417   ** MergeEngine here. This MergeEngine will read data from exactly
2418   ** one PmaReader per sub-task.  */
2419   assert( pSorter->bUseThreads || pSorter->nTask==1 );
2420   if( pSorter->nTask>1 ){
2421     pMain = vdbeMergeEngineNew(pSorter->nTask);
2422     if( pMain==0 ) rc = SQLITE_NOMEM_BKPT;
2423   }
2424 #endif
2425 
2426   for(iTask=0; rc==SQLITE_OK && iTask<pSorter->nTask; iTask++){
2427     SortSubtask *pTask = &pSorter->aTask[iTask];
2428     assert( pTask->nPMA>0 || SQLITE_MAX_WORKER_THREADS>0 );
2429     if( SQLITE_MAX_WORKER_THREADS==0 || pTask->nPMA ){
2430       MergeEngine *pRoot = 0;     /* Root node of tree for this task */
2431       int nDepth = vdbeSorterTreeDepth(pTask->nPMA);
2432       i64 iReadOff = 0;
2433 
2434       if( pTask->nPMA<=SORTER_MAX_MERGE_COUNT ){
2435         rc = vdbeMergeEngineLevel0(pTask, pTask->nPMA, &iReadOff, &pRoot);
2436       }else{
2437         int i;
2438         int iSeq = 0;
2439         pRoot = vdbeMergeEngineNew(SORTER_MAX_MERGE_COUNT);
2440         if( pRoot==0 ) rc = SQLITE_NOMEM_BKPT;
2441         for(i=0; i<pTask->nPMA && rc==SQLITE_OK; i += SORTER_MAX_MERGE_COUNT){
2442           MergeEngine *pMerger = 0; /* New level-0 PMA merger */
2443           int nReader;              /* Number of level-0 PMAs to merge */
2444 
2445           nReader = MIN(pTask->nPMA - i, SORTER_MAX_MERGE_COUNT);
2446           rc = vdbeMergeEngineLevel0(pTask, nReader, &iReadOff, &pMerger);
2447           if( rc==SQLITE_OK ){
2448             rc = vdbeSorterAddToTree(pTask, nDepth, iSeq++, pRoot, pMerger);
2449           }
2450         }
2451       }
2452 
2453       if( rc==SQLITE_OK ){
2454 #if SQLITE_MAX_WORKER_THREADS>0
2455         if( pMain!=0 ){
2456           rc = vdbeIncrMergerNew(pTask, pRoot, &pMain->aReadr[iTask].pIncr);
2457         }else
2458 #endif
2459         {
2460           assert( pMain==0 );
2461           pMain = pRoot;
2462         }
2463       }else{
2464         vdbeMergeEngineFree(pRoot);
2465       }
2466     }
2467   }
2468 
2469   if( rc!=SQLITE_OK ){
2470     vdbeMergeEngineFree(pMain);
2471     pMain = 0;
2472   }
2473   *ppOut = pMain;
2474   return rc;
2475 }
2476 
2477 /*
2478 ** This function is called as part of an sqlite3VdbeSorterRewind() operation
2479 ** on a sorter that has written two or more PMAs to temporary files. It sets
2480 ** up either VdbeSorter.pMerger (for single threaded sorters) or pReader
2481 ** (for multi-threaded sorters) so that it can be used to iterate through
2482 ** all records stored in the sorter.
2483 **
2484 ** SQLITE_OK is returned if successful, or an SQLite error code otherwise.
2485 */
2486 static int vdbeSorterSetupMerge(VdbeSorter *pSorter){
2487   int rc;                         /* Return code */
2488   SortSubtask *pTask0 = &pSorter->aTask[0];
2489   MergeEngine *pMain = 0;
2490 #if SQLITE_MAX_WORKER_THREADS
2491   sqlite3 *db = pTask0->pSorter->db;
2492   int i;
2493   SorterCompare xCompare = vdbeSorterGetCompare(pSorter);
2494   for(i=0; i<pSorter->nTask; i++){
2495     pSorter->aTask[i].xCompare = xCompare;
2496   }
2497 #endif
2498 
2499   rc = vdbeSorterMergeTreeBuild(pSorter, &pMain);
2500   if( rc==SQLITE_OK ){
2501 #if SQLITE_MAX_WORKER_THREADS
2502     assert( pSorter->bUseThreads==0 || pSorter->nTask>1 );
2503     if( pSorter->bUseThreads ){
2504       int iTask;
2505       PmaReader *pReadr = 0;
2506       SortSubtask *pLast = &pSorter->aTask[pSorter->nTask-1];
2507       rc = vdbeSortAllocUnpacked(pLast);
2508       if( rc==SQLITE_OK ){
2509         pReadr = (PmaReader*)sqlite3DbMallocZero(db, sizeof(PmaReader));
2510         pSorter->pReader = pReadr;
2511         if( pReadr==0 ) rc = SQLITE_NOMEM_BKPT;
2512       }
2513       if( rc==SQLITE_OK ){
2514         rc = vdbeIncrMergerNew(pLast, pMain, &pReadr->pIncr);
2515         if( rc==SQLITE_OK ){
2516           vdbeIncrMergerSetThreads(pReadr->pIncr);
2517           for(iTask=0; iTask<(pSorter->nTask-1); iTask++){
2518             IncrMerger *pIncr;
2519             if( (pIncr = pMain->aReadr[iTask].pIncr) ){
2520               vdbeIncrMergerSetThreads(pIncr);
2521               assert( pIncr->pTask!=pLast );
2522             }
2523           }
2524           for(iTask=0; rc==SQLITE_OK && iTask<pSorter->nTask; iTask++){
2525             /* Check that:
2526             **
2527             **   a) The incremental merge object is configured to use the
2528             **      right task, and
2529             **   b) If it is using task (nTask-1), it is configured to run
2530             **      in single-threaded mode. This is important, as the
2531             **      root merge (INCRINIT_ROOT) will be using the same task
2532             **      object.
2533             */
2534             PmaReader *p = &pMain->aReadr[iTask];
2535             assert( p->pIncr==0 || (
2536                 (p->pIncr->pTask==&pSorter->aTask[iTask])             /* a */
2537              && (iTask!=pSorter->nTask-1 || p->pIncr->bUseThread==0)  /* b */
2538             ));
2539             rc = vdbePmaReaderIncrInit(p, INCRINIT_TASK);
2540           }
2541         }
2542         pMain = 0;
2543       }
2544       if( rc==SQLITE_OK ){
2545         rc = vdbePmaReaderIncrMergeInit(pReadr, INCRINIT_ROOT);
2546       }
2547     }else
2548 #endif
2549     {
2550       rc = vdbeMergeEngineInit(pTask0, pMain, INCRINIT_NORMAL);
2551       pSorter->pMerger = pMain;
2552       pMain = 0;
2553     }
2554   }
2555 
2556   if( rc!=SQLITE_OK ){
2557     vdbeMergeEngineFree(pMain);
2558   }
2559   return rc;
2560 }
2561 
2562 
2563 /*
2564 ** Once the sorter has been populated by calls to sqlite3VdbeSorterWrite,
2565 ** this function is called to prepare for iterating through the records
2566 ** in sorted order.
2567 */
2568 int sqlite3VdbeSorterRewind(const VdbeCursor *pCsr, int *pbEof){
2569   VdbeSorter *pSorter;
2570   int rc = SQLITE_OK;             /* Return code */
2571 
2572   assert( pCsr->eCurType==CURTYPE_SORTER );
2573   pSorter = pCsr->uc.pSorter;
2574   assert( pSorter );
2575 
2576   /* If no data has been written to disk, then do not do so now. Instead,
2577   ** sort the VdbeSorter.pRecord list. The vdbe layer will read data directly
2578   ** from the in-memory list.  */
2579   if( pSorter->bUsePMA==0 ){
2580     if( pSorter->list.pList ){
2581       *pbEof = 0;
2582       rc = vdbeSorterSort(&pSorter->aTask[0], &pSorter->list);
2583     }else{
2584       *pbEof = 1;
2585     }
2586     return rc;
2587   }
2588 
2589   /* Write the current in-memory list to a PMA. When the VdbeSorterWrite()
2590   ** function flushes the contents of memory to disk, it immediately always
2591   ** creates a new list consisting of a single key immediately afterwards.
2592   ** So the list is never empty at this point.  */
2593   assert( pSorter->list.pList );
2594   rc = vdbeSorterFlushPMA(pSorter);
2595 
2596   /* Join all threads */
2597   rc = vdbeSorterJoinAll(pSorter, rc);
2598 
2599   vdbeSorterRewindDebug("rewind");
2600 
2601   /* Assuming no errors have occurred, set up a merger structure to
2602   ** incrementally read and merge all remaining PMAs.  */
2603   assert( pSorter->pReader==0 );
2604   if( rc==SQLITE_OK ){
2605     rc = vdbeSorterSetupMerge(pSorter);
2606     *pbEof = 0;
2607   }
2608 
2609   vdbeSorterRewindDebug("rewinddone");
2610   return rc;
2611 }
2612 
2613 /*
2614 ** Advance to the next element in the sorter.  Return value:
2615 **
2616 **    SQLITE_OK     success
2617 **    SQLITE_DONE   end of data
2618 **    otherwise     some kind of error.
2619 */
2620 int sqlite3VdbeSorterNext(sqlite3 *db, const VdbeCursor *pCsr){
2621   VdbeSorter *pSorter;
2622   int rc;                         /* Return code */
2623 
2624   assert( pCsr->eCurType==CURTYPE_SORTER );
2625   pSorter = pCsr->uc.pSorter;
2626   assert( pSorter->bUsePMA || (pSorter->pReader==0 && pSorter->pMerger==0) );
2627   if( pSorter->bUsePMA ){
2628     assert( pSorter->pReader==0 || pSorter->pMerger==0 );
2629     assert( pSorter->bUseThreads==0 || pSorter->pReader );
2630     assert( pSorter->bUseThreads==1 || pSorter->pMerger );
2631 #if SQLITE_MAX_WORKER_THREADS>0
2632     if( pSorter->bUseThreads ){
2633       rc = vdbePmaReaderNext(pSorter->pReader);
2634       if( rc==SQLITE_OK && pSorter->pReader->pFd==0 ) rc = SQLITE_DONE;
2635     }else
2636 #endif
2637     /*if( !pSorter->bUseThreads )*/ {
2638       int res = 0;
2639       assert( pSorter->pMerger!=0 );
2640       assert( pSorter->pMerger->pTask==(&pSorter->aTask[0]) );
2641       rc = vdbeMergeEngineStep(pSorter->pMerger, &res);
2642       if( rc==SQLITE_OK && res ) rc = SQLITE_DONE;
2643     }
2644   }else{
2645     SorterRecord *pFree = pSorter->list.pList;
2646     pSorter->list.pList = pFree->u.pNext;
2647     pFree->u.pNext = 0;
2648     if( pSorter->list.aMemory==0 ) vdbeSorterRecordFree(db, pFree);
2649     rc = pSorter->list.pList ? SQLITE_OK : SQLITE_DONE;
2650   }
2651   return rc;
2652 }
2653 
2654 /*
2655 ** Return a pointer to a buffer owned by the sorter that contains the
2656 ** current key.
2657 */
2658 static void *vdbeSorterRowkey(
2659   const VdbeSorter *pSorter,      /* Sorter object */
2660   int *pnKey                      /* OUT: Size of current key in bytes */
2661 ){
2662   void *pKey;
2663   if( pSorter->bUsePMA ){
2664     PmaReader *pReader;
2665 #if SQLITE_MAX_WORKER_THREADS>0
2666     if( pSorter->bUseThreads ){
2667       pReader = pSorter->pReader;
2668     }else
2669 #endif
2670     /*if( !pSorter->bUseThreads )*/{
2671       pReader = &pSorter->pMerger->aReadr[pSorter->pMerger->aTree[1]];
2672     }
2673     *pnKey = pReader->nKey;
2674     pKey = pReader->aKey;
2675   }else{
2676     *pnKey = pSorter->list.pList->nVal;
2677     pKey = SRVAL(pSorter->list.pList);
2678   }
2679   return pKey;
2680 }
2681 
2682 /*
2683 ** Copy the current sorter key into the memory cell pOut.
2684 */
2685 int sqlite3VdbeSorterRowkey(const VdbeCursor *pCsr, Mem *pOut){
2686   VdbeSorter *pSorter;
2687   void *pKey; int nKey;           /* Sorter key to copy into pOut */
2688 
2689   assert( pCsr->eCurType==CURTYPE_SORTER );
2690   pSorter = pCsr->uc.pSorter;
2691   pKey = vdbeSorterRowkey(pSorter, &nKey);
2692   if( sqlite3VdbeMemClearAndResize(pOut, nKey) ){
2693     return SQLITE_NOMEM_BKPT;
2694   }
2695   pOut->n = nKey;
2696   MemSetTypeFlag(pOut, MEM_Blob);
2697   memcpy(pOut->z, pKey, nKey);
2698 
2699   return SQLITE_OK;
2700 }
2701 
2702 /*
2703 ** Compare the key in memory cell pVal with the key that the sorter cursor
2704 ** passed as the first argument currently points to. For the purposes of
2705 ** the comparison, ignore the rowid field at the end of each record.
2706 **
2707 ** If the sorter cursor key contains any NULL values, consider it to be
2708 ** less than pVal. Even if pVal also contains NULL values.
2709 **
2710 ** If an error occurs, return an SQLite error code (i.e. SQLITE_NOMEM).
2711 ** Otherwise, set *pRes to a negative, zero or positive value if the
2712 ** key in pVal is smaller than, equal to or larger than the current sorter
2713 ** key.
2714 **
2715 ** This routine forms the core of the OP_SorterCompare opcode, which in
2716 ** turn is used to verify uniqueness when constructing a UNIQUE INDEX.
2717 */
2718 int sqlite3VdbeSorterCompare(
2719   const VdbeCursor *pCsr,         /* Sorter cursor */
2720   Mem *pVal,                      /* Value to compare to current sorter key */
2721   int nKeyCol,                    /* Compare this many columns */
2722   int *pRes                       /* OUT: Result of comparison */
2723 ){
2724   VdbeSorter *pSorter;
2725   UnpackedRecord *r2;
2726   KeyInfo *pKeyInfo;
2727   int i;
2728   void *pKey; int nKey;           /* Sorter key to compare pVal with */
2729 
2730   assert( pCsr->eCurType==CURTYPE_SORTER );
2731   pSorter = pCsr->uc.pSorter;
2732   r2 = pSorter->pUnpacked;
2733   pKeyInfo = pCsr->pKeyInfo;
2734   if( r2==0 ){
2735     r2 = pSorter->pUnpacked = sqlite3VdbeAllocUnpackedRecord(pKeyInfo);
2736     if( r2==0 ) return SQLITE_NOMEM_BKPT;
2737     r2->nField = nKeyCol;
2738   }
2739   assert( r2->nField==nKeyCol );
2740 
2741   pKey = vdbeSorterRowkey(pSorter, &nKey);
2742   sqlite3VdbeRecordUnpack(pKeyInfo, nKey, pKey, r2);
2743   for(i=0; i<nKeyCol; i++){
2744     if( r2->aMem[i].flags & MEM_Null ){
2745       *pRes = -1;
2746       return SQLITE_OK;
2747     }
2748   }
2749 
2750   *pRes = sqlite3VdbeRecordCompare(pVal->n, pVal->z, r2);
2751   return SQLITE_OK;
2752 }
2753