blob: f43b14881f5f04f3fb88a407835789f770c2c58f [file] [log] [blame]
dan599e9d22010-07-12 08:39:37 +00001/*
2** 2010 July 12
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******************************************************************************
drh3a0f13f2010-07-12 16:47:48 +000012**
13** This file contains an implementation of the "dbstat" virtual table.
14**
15** The dbstat virtual table is used to extract low-level formatting
16** information from an SQLite database in order to implement the
17** "sqlite3_analyzer" utility. See the ../tool/spaceanal.tcl script
18** for an example implementation.
drha46a4a62015-09-08 21:12:53 +000019**
20** Additional information is available on the "dbstat.html" page of the
21** official SQLite documentation.
dan599e9d22010-07-12 08:39:37 +000022*/
23
drh8755a5f2015-05-12 19:10:18 +000024#include "sqliteInt.h" /* Requires access to internal data structures */
drh1a4a6802015-05-04 18:31:09 +000025#if (defined(SQLITE_ENABLE_DBSTAT_VTAB) || defined(SQLITE_TEST)) \
drh59ba6e82015-05-05 10:46:02 +000026 && !defined(SQLITE_OMIT_VIRTUALTABLE)
dan6d3eb822010-07-12 18:12:41 +000027
dan599e9d22010-07-12 08:39:37 +000028/*
29** Page paths:
30**
31** The value of the 'path' column describes the path taken from the
32** root-node of the b-tree structure to each page. The value of the
33** root-node path is '/'.
34**
35** The value of the path for the left-most child page of the root of
drh3a0f13f2010-07-12 16:47:48 +000036** a b-tree is '/000/'. (Btrees store content ordered from left to right
37** so the pages to the left have smaller keys than the pages to the right.)
38** The next to left-most child of the root page is
dan599e9d22010-07-12 08:39:37 +000039** '/001', and so on, each sibling page identified by a 3-digit hex
drh3a0f13f2010-07-12 16:47:48 +000040** value. The children of the 451st left-most sibling have paths such
dan599e9d22010-07-12 08:39:37 +000041** as '/1c2/000/, '/1c2/001/' etc.
42**
43** Overflow pages are specified by appending a '+' character and a
44** six-digit hexadecimal value to the path to the cell they are linked
45** from. For example, the three overflow pages in a chain linked from
46** the left-most cell of the 450th child of the root page are identified
47** by the paths:
48**
49** '/1c2/000+000000' // First page in overflow chain
50** '/1c2/000+000001' // Second page in overflow chain
51** '/1c2/000+000002' // Third page in overflow chain
52**
53** If the paths are sorted using the BINARY collation sequence, then
54** the overflow pages associated with a cell will appear earlier in the
55** sort-order than its child page:
56**
drh3a0f13f2010-07-12 16:47:48 +000057** '/1c2/000/' // Left-most child of 451st child of root
dan599e9d22010-07-12 08:39:37 +000058*/
59#define VTAB_SCHEMA \
60 "CREATE TABLE xx( " \
61 " name STRING, /* Name of table or index */" \
62 " path INTEGER, /* Path to page from root */" \
63 " pageno INTEGER, /* Page number */" \
64 " pagetype STRING, /* 'internal', 'leaf' or 'overflow' */" \
65 " ncell INTEGER, /* Cells on page (0 for overflow) */" \
66 " payload INTEGER, /* Bytes of payload on this page */" \
67 " unused INTEGER, /* Bytes of unused space on this page */" \
drh4c9f1292011-09-28 00:50:14 +000068 " mx_payload INTEGER, /* Largest payload size of all cells */" \
69 " pgoffset INTEGER, /* Offset of page in file */" \
drha46a4a62015-09-08 21:12:53 +000070 " pgsize INTEGER, /* Size of the page */" \
71 " schema TEXT HIDDEN /* Database schema being analyzed */" \
dan599e9d22010-07-12 08:39:37 +000072 ");"
73
dan599e9d22010-07-12 08:39:37 +000074
75typedef struct StatTable StatTable;
76typedef struct StatCursor StatCursor;
77typedef struct StatPage StatPage;
78typedef struct StatCell StatCell;
79
80struct StatCell {
81 int nLocal; /* Bytes of local payload */
82 u32 iChildPg; /* Child node (or 0 if this is a leaf) */
83 int nOvfl; /* Entries in aOvfl[] */
84 u32 *aOvfl; /* Array of overflow page numbers */
85 int nLastOvfl; /* Bytes of payload on final overflow page */
86 int iOvfl; /* Iterates through aOvfl[] */
87};
88
89struct StatPage {
90 u32 iPgno;
91 DbPage *pPg;
92 int iCell;
93
94 char *zPath; /* Path to this page */
95
96 /* Variables populated by statDecodePage(): */
97 u8 flags; /* Copy of flags byte */
98 int nCell; /* Number of cells on page */
99 int nUnused; /* Number of unused bytes on page */
100 StatCell *aCell; /* Array of parsed cells */
101 u32 iRightChildPg; /* Right-child page number (or 0) */
102 int nMxPayload; /* Largest payload of any cell on this page */
103};
104
105struct StatCursor {
106 sqlite3_vtab_cursor base;
107 sqlite3_stmt *pStmt; /* Iterates through set of root pages */
108 int isEof; /* After pStmt has returned SQLITE_DONE */
drha46a4a62015-09-08 21:12:53 +0000109 int iDb; /* Schema used for this query */
dan599e9d22010-07-12 08:39:37 +0000110
111 StatPage aPage[32];
112 int iPage; /* Current entry in aPage[] */
113
114 /* Values to return. */
115 char *zName; /* Value of 'name' column */
116 char *zPath; /* Value of 'path' column */
117 u32 iPageno; /* Value of 'pageno' column */
118 char *zPagetype; /* Value of 'pagetype' column */
119 int nCell; /* Value of 'ncell' column */
120 int nPayload; /* Value of 'payload' column */
121 int nUnused; /* Value of 'unused' column */
122 int nMxPayload; /* Value of 'mx_payload' column */
drh4c9f1292011-09-28 00:50:14 +0000123 i64 iOffset; /* Value of 'pgOffset' column */
124 int szPage; /* Value of 'pgSize' column */
dan599e9d22010-07-12 08:39:37 +0000125};
126
127struct StatTable {
128 sqlite3_vtab base;
129 sqlite3 *db;
drh857df262015-05-07 14:41:56 +0000130 int iDb; /* Index of database to analyze */
dan599e9d22010-07-12 08:39:37 +0000131};
132
133#ifndef get2byte
134# define get2byte(x) ((x)[0]<<8 | (x)[1])
135#endif
136
137/*
138** Connect to or create a statvfs virtual table.
139*/
140static int statConnect(
141 sqlite3 *db,
142 void *pAux,
143 int argc, const char *const*argv,
144 sqlite3_vtab **ppVtab,
145 char **pzErr
146){
dan995f8b92015-04-27 19:53:55 +0000147 StatTable *pTab = 0;
148 int rc = SQLITE_OK;
drh857df262015-05-07 14:41:56 +0000149 int iDb;
dan599e9d22010-07-12 08:39:37 +0000150
drh857df262015-05-07 14:41:56 +0000151 if( argc>=4 ){
152 iDb = sqlite3FindDbName(db, argv[3]);
153 if( iDb<0 ){
154 *pzErr = sqlite3_mprintf("no such database: %s", argv[3]);
155 return SQLITE_ERROR;
156 }
157 }else{
158 iDb = 0;
159 }
dan995f8b92015-04-27 19:53:55 +0000160 rc = sqlite3_declare_vtab(db, VTAB_SCHEMA);
161 if( rc==SQLITE_OK ){
drhf3cdcdc2015-04-29 16:50:28 +0000162 pTab = (StatTable *)sqlite3_malloc64(sizeof(StatTable));
dan995f8b92015-04-27 19:53:55 +0000163 if( pTab==0 ) rc = SQLITE_NOMEM;
164 }
dan599e9d22010-07-12 08:39:37 +0000165
dan995f8b92015-04-27 19:53:55 +0000166 assert( rc==SQLITE_OK || pTab==0 );
167 if( rc==SQLITE_OK ){
168 memset(pTab, 0, sizeof(StatTable));
169 pTab->db = db;
drh857df262015-05-07 14:41:56 +0000170 pTab->iDb = iDb;
dan995f8b92015-04-27 19:53:55 +0000171 }
172
173 *ppVtab = (sqlite3_vtab*)pTab;
174 return rc;
dan599e9d22010-07-12 08:39:37 +0000175}
176
177/*
178** Disconnect from or destroy a statvfs virtual table.
179*/
180static int statDisconnect(sqlite3_vtab *pVtab){
181 sqlite3_free(pVtab);
182 return SQLITE_OK;
183}
184
185/*
186** There is no "best-index". This virtual table always does a linear
drha46a4a62015-09-08 21:12:53 +0000187** scan. However, a schema=? constraint should cause this table to
188** operate on a different database schema, so check for it.
189**
190** idxNum is normally 0, but will be 1 if a schema=? constraint exists.
dan599e9d22010-07-12 08:39:37 +0000191*/
192static int statBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){
drha46a4a62015-09-08 21:12:53 +0000193 int i;
194
195 pIdxInfo->estimatedCost = 1.0e6; /* Initial cost estimate */
196
197 /* Look for a valid schema=? constraint. If found, change the idxNum to
198 ** 1 and request the value of that constraint be sent to xFilter. And
199 ** lower the cost estimate to encourage the constrained version to be
200 ** used.
201 */
202 for(i=0; i<pIdxInfo->nConstraint; i++){
203 if( pIdxInfo->aConstraint[i].usable==0 ) continue;
204 if( pIdxInfo->aConstraint[i].op!=SQLITE_INDEX_CONSTRAINT_EQ ) continue;
205 if( pIdxInfo->aConstraint[i].iColumn!=10 ) continue;
206 pIdxInfo->idxNum = 1;
207 pIdxInfo->estimatedCost = 1.0;
208 pIdxInfo->aConstraintUsage[i].argvIndex = 1;
209 pIdxInfo->aConstraintUsage[i].omit = 1;
210 break;
211 }
212
dan599e9d22010-07-12 08:39:37 +0000213
214 /* Records are always returned in ascending order of (name, path).
215 ** If this will satisfy the client, set the orderByConsumed flag so that
216 ** SQLite does not do an external sort.
217 */
218 if( ( pIdxInfo->nOrderBy==1
219 && pIdxInfo->aOrderBy[0].iColumn==0
220 && pIdxInfo->aOrderBy[0].desc==0
221 ) ||
222 ( pIdxInfo->nOrderBy==2
223 && pIdxInfo->aOrderBy[0].iColumn==0
224 && pIdxInfo->aOrderBy[0].desc==0
225 && pIdxInfo->aOrderBy[1].iColumn==1
226 && pIdxInfo->aOrderBy[1].desc==0
227 )
228 ){
229 pIdxInfo->orderByConsumed = 1;
230 }
231
dan599e9d22010-07-12 08:39:37 +0000232 return SQLITE_OK;
233}
234
235/*
236** Open a new statvfs cursor.
237*/
238static int statOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){
239 StatTable *pTab = (StatTable *)pVTab;
240 StatCursor *pCsr;
dan599e9d22010-07-12 08:39:37 +0000241
drhf3cdcdc2015-04-29 16:50:28 +0000242 pCsr = (StatCursor *)sqlite3_malloc64(sizeof(StatCursor));
dan995f8b92015-04-27 19:53:55 +0000243 if( pCsr==0 ){
drha46a4a62015-09-08 21:12:53 +0000244 return SQLITE_NOMEM;
dan995f8b92015-04-27 19:53:55 +0000245 }else{
246 memset(pCsr, 0, sizeof(StatCursor));
247 pCsr->base.pVtab = pVTab;
drha46a4a62015-09-08 21:12:53 +0000248 pCsr->iDb = pTab->iDb;
dan599e9d22010-07-12 08:39:37 +0000249 }
250
251 *ppCursor = (sqlite3_vtab_cursor *)pCsr;
drha46a4a62015-09-08 21:12:53 +0000252 return SQLITE_OK;
dan599e9d22010-07-12 08:39:37 +0000253}
254
255static void statClearPage(StatPage *p){
256 int i;
dan995f8b92015-04-27 19:53:55 +0000257 if( p->aCell ){
258 for(i=0; i<p->nCell; i++){
259 sqlite3_free(p->aCell[i].aOvfl);
260 }
261 sqlite3_free(p->aCell);
dan599e9d22010-07-12 08:39:37 +0000262 }
263 sqlite3PagerUnref(p->pPg);
dan599e9d22010-07-12 08:39:37 +0000264 sqlite3_free(p->zPath);
265 memset(p, 0, sizeof(StatPage));
266}
267
268static void statResetCsr(StatCursor *pCsr){
269 int i;
270 sqlite3_reset(pCsr->pStmt);
271 for(i=0; i<ArraySize(pCsr->aPage); i++){
272 statClearPage(&pCsr->aPage[i]);
273 }
274 pCsr->iPage = 0;
275 sqlite3_free(pCsr->zPath);
276 pCsr->zPath = 0;
drha46a4a62015-09-08 21:12:53 +0000277 pCsr->isEof = 0;
dan599e9d22010-07-12 08:39:37 +0000278}
279
280/*
281** Close a statvfs cursor.
282*/
283static int statClose(sqlite3_vtab_cursor *pCursor){
284 StatCursor *pCsr = (StatCursor *)pCursor;
285 statResetCsr(pCsr);
286 sqlite3_finalize(pCsr->pStmt);
287 sqlite3_free(pCsr);
288 return SQLITE_OK;
289}
290
291static void getLocalPayload(
292 int nUsable, /* Usable bytes per page */
293 u8 flags, /* Page flags */
294 int nTotal, /* Total record (payload) size */
295 int *pnLocal /* OUT: Bytes stored locally */
296){
297 int nLocal;
298 int nMinLocal;
299 int nMaxLocal;
300
301 if( flags==0x0D ){ /* Table leaf node */
302 nMinLocal = (nUsable - 12) * 32 / 255 - 23;
303 nMaxLocal = nUsable - 35;
304 }else{ /* Index interior and leaf nodes */
305 nMinLocal = (nUsable - 12) * 32 / 255 - 23;
306 nMaxLocal = (nUsable - 12) * 64 / 255 - 23;
307 }
308
309 nLocal = nMinLocal + (nTotal - nMinLocal) % (nUsable - 4);
310 if( nLocal>nMaxLocal ) nLocal = nMinLocal;
311 *pnLocal = nLocal;
312}
313
314static int statDecodePage(Btree *pBt, StatPage *p){
315 int nUnused;
316 int iOff;
317 int nHdr;
318 int isLeaf;
drh4c9f1292011-09-28 00:50:14 +0000319 int szPage;
dan599e9d22010-07-12 08:39:37 +0000320
321 u8 *aData = sqlite3PagerGetData(p->pPg);
322 u8 *aHdr = &aData[p->iPgno==1 ? 100 : 0];
323
324 p->flags = aHdr[0];
325 p->nCell = get2byte(&aHdr[3]);
326 p->nMxPayload = 0;
327
328 isLeaf = (p->flags==0x0A || p->flags==0x0D);
329 nHdr = 12 - isLeaf*4 + (p->iPgno==1)*100;
330
331 nUnused = get2byte(&aHdr[5]) - nHdr - 2*p->nCell;
332 nUnused += (int)aHdr[7];
333 iOff = get2byte(&aHdr[1]);
334 while( iOff ){
335 nUnused += get2byte(&aData[iOff+2]);
336 iOff = get2byte(&aData[iOff]);
337 }
338 p->nUnused = nUnused;
339 p->iRightChildPg = isLeaf ? 0 : sqlite3Get4byte(&aHdr[8]);
drh4c9f1292011-09-28 00:50:14 +0000340 szPage = sqlite3BtreeGetPageSize(pBt);
dan599e9d22010-07-12 08:39:37 +0000341
342 if( p->nCell ){
343 int i; /* Used to iterate through cells */
drhad0961b2015-02-21 00:19:25 +0000344 int nUsable; /* Usable bytes per page */
dan599e9d22010-07-12 08:39:37 +0000345
drhad0961b2015-02-21 00:19:25 +0000346 sqlite3BtreeEnter(pBt);
347 nUsable = szPage - sqlite3BtreeGetReserveNoMutex(pBt);
348 sqlite3BtreeLeave(pBt);
drhf3cdcdc2015-04-29 16:50:28 +0000349 p->aCell = sqlite3_malloc64((p->nCell+1) * sizeof(StatCell));
dan995f8b92015-04-27 19:53:55 +0000350 if( p->aCell==0 ) return SQLITE_NOMEM;
dan599e9d22010-07-12 08:39:37 +0000351 memset(p->aCell, 0, (p->nCell+1) * sizeof(StatCell));
352
353 for(i=0; i<p->nCell; i++){
354 StatCell *pCell = &p->aCell[i];
355
356 iOff = get2byte(&aData[nHdr+i*2]);
357 if( !isLeaf ){
358 pCell->iChildPg = sqlite3Get4byte(&aData[iOff]);
359 iOff += 4;
360 }
361 if( p->flags==0x05 ){
362 /* A table interior node. nPayload==0. */
363 }else{
364 u32 nPayload; /* Bytes of payload total (local+overflow) */
365 int nLocal; /* Bytes of payload stored locally */
366 iOff += getVarint32(&aData[iOff], nPayload);
367 if( p->flags==0x0D ){
368 u64 dummy;
369 iOff += sqlite3GetVarint(&aData[iOff], &dummy);
370 }
drh7da5fcb2012-03-30 14:59:43 +0000371 if( nPayload>(u32)p->nMxPayload ) p->nMxPayload = nPayload;
dan599e9d22010-07-12 08:39:37 +0000372 getLocalPayload(nUsable, p->flags, nPayload, &nLocal);
373 pCell->nLocal = nLocal;
drh7da5fcb2012-03-30 14:59:43 +0000374 assert( nLocal>=0 );
drh26d8b0f2012-05-11 15:53:18 +0000375 assert( nPayload>=(u32)nLocal );
dan599e9d22010-07-12 08:39:37 +0000376 assert( nLocal<=(nUsable-35) );
drh7da5fcb2012-03-30 14:59:43 +0000377 if( nPayload>(u32)nLocal ){
dan599e9d22010-07-12 08:39:37 +0000378 int j;
379 int nOvfl = ((nPayload - nLocal) + nUsable-4 - 1) / (nUsable - 4);
380 pCell->nLastOvfl = (nPayload-nLocal) - (nOvfl-1) * (nUsable-4);
381 pCell->nOvfl = nOvfl;
drhf3cdcdc2015-04-29 16:50:28 +0000382 pCell->aOvfl = sqlite3_malloc64(sizeof(u32)*nOvfl);
dan995f8b92015-04-27 19:53:55 +0000383 if( pCell->aOvfl==0 ) return SQLITE_NOMEM;
dan599e9d22010-07-12 08:39:37 +0000384 pCell->aOvfl[0] = sqlite3Get4byte(&aData[iOff+nLocal]);
385 for(j=1; j<nOvfl; j++){
386 int rc;
387 u32 iPrev = pCell->aOvfl[j-1];
388 DbPage *pPg = 0;
389 rc = sqlite3PagerGet(sqlite3BtreePager(pBt), iPrev, &pPg);
390 if( rc!=SQLITE_OK ){
391 assert( pPg==0 );
392 return rc;
393 }
394 pCell->aOvfl[j] = sqlite3Get4byte(sqlite3PagerGetData(pPg));
395 sqlite3PagerUnref(pPg);
396 }
397 }
398 }
399 }
400 }
401
402 return SQLITE_OK;
403}
404
dan599e9d22010-07-12 08:39:37 +0000405/*
dan2cf7e0a2011-10-05 17:36:27 +0000406** Populate the pCsr->iOffset and pCsr->szPage member variables. Based on
407** the current value of pCsr->iPageno.
408*/
409static void statSizeAndOffset(StatCursor *pCsr){
410 StatTable *pTab = (StatTable *)((sqlite3_vtab_cursor *)pCsr)->pVtab;
drh857df262015-05-07 14:41:56 +0000411 Btree *pBt = pTab->db->aDb[pTab->iDb].pBt;
dan2cf7e0a2011-10-05 17:36:27 +0000412 Pager *pPager = sqlite3BtreePager(pBt);
413 sqlite3_file *fd;
414 sqlite3_int64 x[2];
415
416 /* The default page size and offset */
417 pCsr->szPage = sqlite3BtreeGetPageSize(pBt);
dan7c3210e2011-12-21 18:04:41 +0000418 pCsr->iOffset = (i64)pCsr->szPage * (pCsr->iPageno - 1);
dan2cf7e0a2011-10-05 17:36:27 +0000419
420 /* If connected to a ZIPVFS backend, override the page size and
421 ** offset with actual values obtained from ZIPVFS.
422 */
423 fd = sqlite3PagerFile(pPager);
424 x[0] = pCsr->iPageno;
drh857df262015-05-07 14:41:56 +0000425 if( fd->pMethods!=0 && sqlite3OsFileControl(fd, 230440, &x)==SQLITE_OK ){
dan2cf7e0a2011-10-05 17:36:27 +0000426 pCsr->iOffset = x[0];
drh7da5fcb2012-03-30 14:59:43 +0000427 pCsr->szPage = (int)x[1];
dan2cf7e0a2011-10-05 17:36:27 +0000428 }
429}
430
431/*
dan599e9d22010-07-12 08:39:37 +0000432** Move a statvfs cursor to the next entry in the file.
433*/
434static int statNext(sqlite3_vtab_cursor *pCursor){
435 int rc;
436 int nPayload;
drh5f36a832015-05-07 18:29:04 +0000437 char *z;
dan599e9d22010-07-12 08:39:37 +0000438 StatCursor *pCsr = (StatCursor *)pCursor;
439 StatTable *pTab = (StatTable *)pCursor->pVtab;
drha46a4a62015-09-08 21:12:53 +0000440 Btree *pBt = pTab->db->aDb[pCsr->iDb].pBt;
dan599e9d22010-07-12 08:39:37 +0000441 Pager *pPager = sqlite3BtreePager(pBt);
442
443 sqlite3_free(pCsr->zPath);
444 pCsr->zPath = 0;
445
drha4641712013-11-02 11:34:58 +0000446statNextRestart:
dan599e9d22010-07-12 08:39:37 +0000447 if( pCsr->aPage[0].pPg==0 ){
448 rc = sqlite3_step(pCsr->pStmt);
449 if( rc==SQLITE_ROW ){
dan763afe62010-08-03 06:42:39 +0000450 int nPage;
drh7da5fcb2012-03-30 14:59:43 +0000451 u32 iRoot = (u32)sqlite3_column_int64(pCsr->pStmt, 1);
dan763afe62010-08-03 06:42:39 +0000452 sqlite3PagerPagecount(pPager, &nPage);
453 if( nPage==0 ){
454 pCsr->isEof = 1;
455 return sqlite3_reset(pCsr->pStmt);
456 }
dan599e9d22010-07-12 08:39:37 +0000457 rc = sqlite3PagerGet(pPager, iRoot, &pCsr->aPage[0].pPg);
458 pCsr->aPage[0].iPgno = iRoot;
459 pCsr->aPage[0].iCell = 0;
drh5f36a832015-05-07 18:29:04 +0000460 pCsr->aPage[0].zPath = z = sqlite3_mprintf("/");
dan599e9d22010-07-12 08:39:37 +0000461 pCsr->iPage = 0;
drh5f36a832015-05-07 18:29:04 +0000462 if( z==0 ) rc = SQLITE_NOMEM;
dan599e9d22010-07-12 08:39:37 +0000463 }else{
464 pCsr->isEof = 1;
465 return sqlite3_reset(pCsr->pStmt);
466 }
467 }else{
468
469 /* Page p itself has already been visited. */
470 StatPage *p = &pCsr->aPage[pCsr->iPage];
471
472 while( p->iCell<p->nCell ){
473 StatCell *pCell = &p->aCell[p->iCell];
474 if( pCell->iOvfl<pCell->nOvfl ){
drhad0961b2015-02-21 00:19:25 +0000475 int nUsable;
476 sqlite3BtreeEnter(pBt);
477 nUsable = sqlite3BtreeGetPageSize(pBt) -
478 sqlite3BtreeGetReserveNoMutex(pBt);
479 sqlite3BtreeLeave(pBt);
dan599e9d22010-07-12 08:39:37 +0000480 pCsr->zName = (char *)sqlite3_column_text(pCsr->pStmt, 0);
481 pCsr->iPageno = pCell->aOvfl[pCell->iOvfl];
482 pCsr->zPagetype = "overflow";
483 pCsr->nCell = 0;
484 pCsr->nMxPayload = 0;
drh5f36a832015-05-07 18:29:04 +0000485 pCsr->zPath = z = sqlite3_mprintf(
dan599e9d22010-07-12 08:39:37 +0000486 "%s%.3x+%.6x", p->zPath, p->iCell, pCell->iOvfl
487 );
488 if( pCell->iOvfl<pCell->nOvfl-1 ){
489 pCsr->nUnused = 0;
490 pCsr->nPayload = nUsable - 4;
491 }else{
492 pCsr->nPayload = pCell->nLastOvfl;
493 pCsr->nUnused = nUsable - 4 - pCsr->nPayload;
494 }
495 pCell->iOvfl++;
dan2cf7e0a2011-10-05 17:36:27 +0000496 statSizeAndOffset(pCsr);
drh5f36a832015-05-07 18:29:04 +0000497 return z==0 ? SQLITE_NOMEM : SQLITE_OK;
dan599e9d22010-07-12 08:39:37 +0000498 }
499 if( p->iRightChildPg ) break;
500 p->iCell++;
501 }
502
drha4641712013-11-02 11:34:58 +0000503 if( !p->iRightChildPg || p->iCell>p->nCell ){
dan599e9d22010-07-12 08:39:37 +0000504 statClearPage(p);
505 if( pCsr->iPage==0 ) return statNext(pCursor);
506 pCsr->iPage--;
drha4641712013-11-02 11:34:58 +0000507 goto statNextRestart; /* Tail recursion */
dan599e9d22010-07-12 08:39:37 +0000508 }
509 pCsr->iPage++;
510 assert( p==&pCsr->aPage[pCsr->iPage-1] );
511
512 if( p->iCell==p->nCell ){
513 p[1].iPgno = p->iRightChildPg;
514 }else{
515 p[1].iPgno = p->aCell[p->iCell].iChildPg;
516 }
517 rc = sqlite3PagerGet(pPager, p[1].iPgno, &p[1].pPg);
518 p[1].iCell = 0;
drh5f36a832015-05-07 18:29:04 +0000519 p[1].zPath = z = sqlite3_mprintf("%s%.3x/", p->zPath, p->iCell);
dan599e9d22010-07-12 08:39:37 +0000520 p->iCell++;
drh5f36a832015-05-07 18:29:04 +0000521 if( z==0 ) rc = SQLITE_NOMEM;
dan599e9d22010-07-12 08:39:37 +0000522 }
523
524
525 /* Populate the StatCursor fields with the values to be returned
526 ** by the xColumn() and xRowid() methods.
527 */
528 if( rc==SQLITE_OK ){
529 int i;
530 StatPage *p = &pCsr->aPage[pCsr->iPage];
531 pCsr->zName = (char *)sqlite3_column_text(pCsr->pStmt, 0);
532 pCsr->iPageno = p->iPgno;
533
dan995f8b92015-04-27 19:53:55 +0000534 rc = statDecodePage(pBt, p);
535 if( rc==SQLITE_OK ){
536 statSizeAndOffset(pCsr);
drh4c9f1292011-09-28 00:50:14 +0000537
dan995f8b92015-04-27 19:53:55 +0000538 switch( p->flags ){
539 case 0x05: /* table internal */
540 case 0x02: /* index internal */
541 pCsr->zPagetype = "internal";
542 break;
543 case 0x0D: /* table leaf */
544 case 0x0A: /* index leaf */
545 pCsr->zPagetype = "leaf";
546 break;
547 default:
548 pCsr->zPagetype = "corrupted";
549 break;
550 }
551 pCsr->nCell = p->nCell;
552 pCsr->nUnused = p->nUnused;
553 pCsr->nMxPayload = p->nMxPayload;
drh5f36a832015-05-07 18:29:04 +0000554 pCsr->zPath = z = sqlite3_mprintf("%s", p->zPath);
555 if( z==0 ) rc = SQLITE_NOMEM;
dan995f8b92015-04-27 19:53:55 +0000556 nPayload = 0;
557 for(i=0; i<p->nCell; i++){
558 nPayload += p->aCell[i].nLocal;
559 }
560 pCsr->nPayload = nPayload;
dan599e9d22010-07-12 08:39:37 +0000561 }
dan599e9d22010-07-12 08:39:37 +0000562 }
563
564 return rc;
565}
566
567static int statEof(sqlite3_vtab_cursor *pCursor){
568 StatCursor *pCsr = (StatCursor *)pCursor;
569 return pCsr->isEof;
570}
571
572static int statFilter(
573 sqlite3_vtab_cursor *pCursor,
574 int idxNum, const char *idxStr,
575 int argc, sqlite3_value **argv
576){
dan599e9d22010-07-12 08:39:37 +0000577 StatCursor *pCsr = (StatCursor *)pCursor;
drha46a4a62015-09-08 21:12:53 +0000578 StatTable *pTab = (StatTable*)(pCursor->pVtab);
579 char *zSql;
580 int rc = SQLITE_OK;
581 char *zMaster;
dan599e9d22010-07-12 08:39:37 +0000582
drha46a4a62015-09-08 21:12:53 +0000583 if( idxNum==1 ){
584 const char *zDbase = (const char*)sqlite3_value_text(argv[0]);
585 pCsr->iDb = sqlite3FindDbName(pTab->db, zDbase);
586 if( pCsr->iDb<0 ){
587 sqlite3_free(pCursor->pVtab->zErrMsg);
588 pCursor->pVtab->zErrMsg = sqlite3_mprintf("no such schema: %s", zDbase);
589 return pCursor->pVtab->zErrMsg ? SQLITE_ERROR : SQLITE_NOMEM;
590 }
591 }else{
592 pCsr->iDb = pTab->iDb;
593 }
drhe5918c62010-08-14 12:42:45 +0000594 statResetCsr(pCsr);
drha46a4a62015-09-08 21:12:53 +0000595 sqlite3_finalize(pCsr->pStmt);
596 pCsr->pStmt = 0;
597 zMaster = pCsr->iDb==1 ? "sqlite_temp_master" : "sqlite_master";
598 zSql = sqlite3_mprintf(
599 "SELECT 'sqlite_master' AS name, 1 AS rootpage, 'table' AS type"
600 " UNION ALL "
601 "SELECT name, rootpage, type"
602 " FROM \"%w\".%s WHERE rootpage!=0"
603 " ORDER BY name", pTab->db->aDb[pCsr->iDb].zName, zMaster);
604 if( zSql==0 ){
605 return SQLITE_NOMEM;
606 }else{
607 rc = sqlite3_prepare_v2(pTab->db, zSql, -1, &pCsr->pStmt, 0);
608 sqlite3_free(zSql);
609 }
610
611 if( rc==SQLITE_OK ){
612 rc = statNext(pCursor);
613 }
614 return rc;
dan599e9d22010-07-12 08:39:37 +0000615}
616
617static int statColumn(
618 sqlite3_vtab_cursor *pCursor,
619 sqlite3_context *ctx,
620 int i
621){
622 StatCursor *pCsr = (StatCursor *)pCursor;
623 switch( i ){
624 case 0: /* name */
drh5f36a832015-05-07 18:29:04 +0000625 sqlite3_result_text(ctx, pCsr->zName, -1, SQLITE_TRANSIENT);
dan599e9d22010-07-12 08:39:37 +0000626 break;
627 case 1: /* path */
628 sqlite3_result_text(ctx, pCsr->zPath, -1, SQLITE_TRANSIENT);
629 break;
630 case 2: /* pageno */
631 sqlite3_result_int64(ctx, pCsr->iPageno);
632 break;
633 case 3: /* pagetype */
634 sqlite3_result_text(ctx, pCsr->zPagetype, -1, SQLITE_STATIC);
635 break;
636 case 4: /* ncell */
637 sqlite3_result_int(ctx, pCsr->nCell);
638 break;
639 case 5: /* payload */
640 sqlite3_result_int(ctx, pCsr->nPayload);
641 break;
642 case 6: /* unused */
643 sqlite3_result_int(ctx, pCsr->nUnused);
644 break;
645 case 7: /* mx_payload */
646 sqlite3_result_int(ctx, pCsr->nMxPayload);
647 break;
drh4c9f1292011-09-28 00:50:14 +0000648 case 8: /* pgoffset */
649 sqlite3_result_int64(ctx, pCsr->iOffset);
650 break;
drha46a4a62015-09-08 21:12:53 +0000651 case 9: /* pgsize */
drh4c9f1292011-09-28 00:50:14 +0000652 sqlite3_result_int(ctx, pCsr->szPage);
653 break;
drha46a4a62015-09-08 21:12:53 +0000654 default: { /* schema */
655 sqlite3 *db = sqlite3_context_db_handle(ctx);
656 int iDb = pCsr->iDb;
657 sqlite3_result_text(ctx, db->aDb[iDb].zName, -1, SQLITE_STATIC);
658 break;
659 }
dan599e9d22010-07-12 08:39:37 +0000660 }
661 return SQLITE_OK;
662}
663
664static int statRowid(sqlite3_vtab_cursor *pCursor, sqlite_int64 *pRowid){
665 StatCursor *pCsr = (StatCursor *)pCursor;
666 *pRowid = pCsr->iPageno;
667 return SQLITE_OK;
668}
669
drh1a4a6802015-05-04 18:31:09 +0000670/*
671** Invoke this routine to register the "dbstat" virtual table module
672*/
drh3e0327d2015-05-11 11:59:15 +0000673int sqlite3DbstatRegister(sqlite3 *db){
dan599e9d22010-07-12 08:39:37 +0000674 static sqlite3_module dbstat_module = {
675 0, /* iVersion */
676 statConnect, /* xCreate */
677 statConnect, /* xConnect */
678 statBestIndex, /* xBestIndex */
679 statDisconnect, /* xDisconnect */
680 statDisconnect, /* xDestroy */
681 statOpen, /* xOpen - open a cursor */
682 statClose, /* xClose - close a cursor */
683 statFilter, /* xFilter - configure scan constraints */
684 statNext, /* xNext - advance a cursor */
685 statEof, /* xEof - check for end of scan */
686 statColumn, /* xColumn - read data */
687 statRowid, /* xRowid - read data */
688 0, /* xUpdate */
689 0, /* xBegin */
690 0, /* xSync */
691 0, /* xCommit */
692 0, /* xRollback */
693 0, /* xFindMethod */
694 0, /* xRename */
695 };
dand154a432015-04-30 20:26:53 +0000696 return sqlite3_create_module(db, "dbstat", &dbstat_module, 0);
dan599e9d22010-07-12 08:39:37 +0000697}
drh10819252015-05-12 14:22:05 +0000698#elif defined(SQLITE_ENABLE_DBSTAT_VTAB)
drh6582ae52015-05-12 12:24:50 +0000699int sqlite3DbstatRegister(sqlite3 *db){ return SQLITE_OK; }
drh1a4a6802015-05-04 18:31:09 +0000700#endif /* SQLITE_ENABLE_DBSTAT_VTAB */