1 /* 2 ** 2001 September 15 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 is the implementation of the page cache subsystem or "pager". 13 ** 14 ** The pager is used to access a database disk file. It implements 15 ** atomic commit and rollback through the use of a journal file that 16 ** is separate from the database file. The pager also implements file 17 ** locking to prevent two processes from writing the same database 18 ** file simultaneously, or one process from reading the database while 19 ** another is writing. 20 ** 21 ** @(#) $Id: pager.c,v 1.282 2007/01/05 02:00:47 drh Exp $ 22 */ 23 #ifndef SQLITE_OMIT_DISKIO 24 #include "sqliteInt.h" 25 #include "os.h" 26 #include "pager.h" 27 #include <assert.h> 28 #include <string.h> 29 30 /* 31 ** Macros for troubleshooting. Normally turned off 32 */ 33 #if 0 34 #define sqlite3DebugPrintf printf 35 #define TRACE1(X) sqlite3DebugPrintf(X) 36 #define TRACE2(X,Y) sqlite3DebugPrintf(X,Y) 37 #define TRACE3(X,Y,Z) sqlite3DebugPrintf(X,Y,Z) 38 #define TRACE4(X,Y,Z,W) sqlite3DebugPrintf(X,Y,Z,W) 39 #define TRACE5(X,Y,Z,W,V) sqlite3DebugPrintf(X,Y,Z,W,V) 40 #else 41 #define TRACE1(X) 42 #define TRACE2(X,Y) 43 #define TRACE3(X,Y,Z) 44 #define TRACE4(X,Y,Z,W) 45 #define TRACE5(X,Y,Z,W,V) 46 #endif 47 48 /* 49 ** The following two macros are used within the TRACEX() macros above 50 ** to print out file-descriptors. 51 ** 52 ** PAGERID() takes a pointer to a Pager struct as it's argument. The 53 ** associated file-descriptor is returned. FILEHANDLEID() takes an OsFile 54 ** struct as it's argument. 55 */ 56 #define PAGERID(p) ((int)(p->fd)) 57 #define FILEHANDLEID(fd) ((int)fd) 58 59 /* 60 ** The page cache as a whole is always in one of the following 61 ** states: 62 ** 63 ** PAGER_UNLOCK The page cache is not currently reading or 64 ** writing the database file. There is no 65 ** data held in memory. This is the initial 66 ** state. 67 ** 68 ** PAGER_SHARED The page cache is reading the database. 69 ** Writing is not permitted. There can be 70 ** multiple readers accessing the same database 71 ** file at the same time. 72 ** 73 ** PAGER_RESERVED This process has reserved the database for writing 74 ** but has not yet made any changes. Only one process 75 ** at a time can reserve the database. The original 76 ** database file has not been modified so other 77 ** processes may still be reading the on-disk 78 ** database file. 79 ** 80 ** PAGER_EXCLUSIVE The page cache is writing the database. 81 ** Access is exclusive. No other processes or 82 ** threads can be reading or writing while one 83 ** process is writing. 84 ** 85 ** PAGER_SYNCED The pager moves to this state from PAGER_EXCLUSIVE 86 ** after all dirty pages have been written to the 87 ** database file and the file has been synced to 88 ** disk. All that remains to do is to remove the 89 ** journal file and the transaction will be 90 ** committed. 91 ** 92 ** The page cache comes up in PAGER_UNLOCK. The first time a 93 ** sqlite3pager_get() occurs, the state transitions to PAGER_SHARED. 94 ** After all pages have been released using sqlite_page_unref(), 95 ** the state transitions back to PAGER_UNLOCK. The first time 96 ** that sqlite3pager_write() is called, the state transitions to 97 ** PAGER_RESERVED. (Note that sqlite_page_write() can only be 98 ** called on an outstanding page which means that the pager must 99 ** be in PAGER_SHARED before it transitions to PAGER_RESERVED.) 100 ** The transition to PAGER_EXCLUSIVE occurs when before any changes 101 ** are made to the database file. After an sqlite3pager_rollback() 102 ** or sqlite_pager_commit(), the state goes back to PAGER_SHARED. 103 */ 104 #define PAGER_UNLOCK 0 105 #define PAGER_SHARED 1 /* same as SHARED_LOCK */ 106 #define PAGER_RESERVED 2 /* same as RESERVED_LOCK */ 107 #define PAGER_EXCLUSIVE 4 /* same as EXCLUSIVE_LOCK */ 108 #define PAGER_SYNCED 5 109 110 /* 111 ** If the SQLITE_BUSY_RESERVED_LOCK macro is set to true at compile-time, 112 ** then failed attempts to get a reserved lock will invoke the busy callback. 113 ** This is off by default. To see why, consider the following scenario: 114 ** 115 ** Suppose thread A already has a shared lock and wants a reserved lock. 116 ** Thread B already has a reserved lock and wants an exclusive lock. If 117 ** both threads are using their busy callbacks, it might be a long time 118 ** be for one of the threads give up and allows the other to proceed. 119 ** But if the thread trying to get the reserved lock gives up quickly 120 ** (if it never invokes its busy callback) then the contention will be 121 ** resolved quickly. 122 */ 123 #ifndef SQLITE_BUSY_RESERVED_LOCK 124 # define SQLITE_BUSY_RESERVED_LOCK 0 125 #endif 126 127 /* 128 ** This macro rounds values up so that if the value is an address it 129 ** is guaranteed to be an address that is aligned to an 8-byte boundary. 130 */ 131 #define FORCE_ALIGNMENT(X) (((X)+7)&~7) 132 133 /* 134 ** Each in-memory image of a page begins with the following header. 135 ** This header is only visible to this pager module. The client 136 ** code that calls pager sees only the data that follows the header. 137 ** 138 ** Client code should call sqlite3pager_write() on a page prior to making 139 ** any modifications to that page. The first time sqlite3pager_write() 140 ** is called, the original page contents are written into the rollback 141 ** journal and PgHdr.inJournal and PgHdr.needSync are set. Later, once 142 ** the journal page has made it onto the disk surface, PgHdr.needSync 143 ** is cleared. The modified page cannot be written back into the original 144 ** database file until the journal pages has been synced to disk and the 145 ** PgHdr.needSync has been cleared. 146 ** 147 ** The PgHdr.dirty flag is set when sqlite3pager_write() is called and 148 ** is cleared again when the page content is written back to the original 149 ** database file. 150 */ 151 typedef struct PgHdr PgHdr; 152 struct PgHdr { 153 Pager *pPager; /* The pager to which this page belongs */ 154 Pgno pgno; /* The page number for this page */ 155 PgHdr *pNextHash, *pPrevHash; /* Hash collision chain for PgHdr.pgno */ 156 PgHdr *pNextFree, *pPrevFree; /* Freelist of pages where nRef==0 */ 157 PgHdr *pNextAll; /* A list of all pages */ 158 PgHdr *pNextStmt, *pPrevStmt; /* List of pages in the statement journal */ 159 u8 inJournal; /* TRUE if has been written to journal */ 160 u8 inStmt; /* TRUE if in the statement subjournal */ 161 u8 dirty; /* TRUE if we need to write back changes */ 162 u8 needSync; /* Sync journal before writing this page */ 163 u8 alwaysRollback; /* Disable dont_rollback() for this page */ 164 short int nRef; /* Number of users of this page */ 165 PgHdr *pDirty, *pPrevDirty; /* Dirty pages */ 166 u32 notUsed; /* Buffer space */ 167 #ifdef SQLITE_CHECK_PAGES 168 u32 pageHash; 169 #endif 170 /* pPager->pageSize bytes of page data follow this header */ 171 /* Pager.nExtra bytes of local data follow the page data */ 172 }; 173 174 /* 175 ** For an in-memory only database, some extra information is recorded about 176 ** each page so that changes can be rolled back. (Journal files are not 177 ** used for in-memory databases.) The following information is added to 178 ** the end of every EXTRA block for in-memory databases. 179 ** 180 ** This information could have been added directly to the PgHdr structure. 181 ** But then it would take up an extra 8 bytes of storage on every PgHdr 182 ** even for disk-based databases. Splitting it out saves 8 bytes. This 183 ** is only a savings of 0.8% but those percentages add up. 184 */ 185 typedef struct PgHistory PgHistory; 186 struct PgHistory { 187 u8 *pOrig; /* Original page text. Restore to this on a full rollback */ 188 u8 *pStmt; /* Text as it was at the beginning of the current statement */ 189 }; 190 191 /* 192 ** A macro used for invoking the codec if there is one 193 */ 194 #ifdef SQLITE_HAS_CODEC 195 # define CODEC1(P,D,N,X) if( P->xCodec!=0 ){ P->xCodec(P->pCodecArg,D,N,X); } 196 # define CODEC2(P,D,N,X) ((char*)(P->xCodec!=0?P->xCodec(P->pCodecArg,D,N,X):D)) 197 #else 198 # define CODEC1(P,D,N,X) /* NO-OP */ 199 # define CODEC2(P,D,N,X) ((char*)D) 200 #endif 201 202 /* 203 ** Convert a pointer to a PgHdr into a pointer to its data 204 ** and back again. 205 */ 206 #define PGHDR_TO_DATA(P) ((void*)(&(P)[1])) 207 #define DATA_TO_PGHDR(D) (&((PgHdr*)(D))[-1]) 208 #define PGHDR_TO_EXTRA(G,P) ((void*)&((char*)(&(G)[1]))[(P)->pageSize]) 209 #define PGHDR_TO_HIST(P,PGR) \ 210 ((PgHistory*)&((char*)(&(P)[1]))[(PGR)->pageSize+(PGR)->nExtra]) 211 212 /* 213 ** A open page cache is an instance of the following structure. 214 ** 215 ** Pager.errCode may be set to SQLITE_IOERR, SQLITE_CORRUPT, SQLITE_PROTOCOL 216 ** or SQLITE_FULL. Once one of the first three errors occurs, it persists 217 ** and is returned as the result of every major pager API call. The 218 ** SQLITE_FULL return code is slightly different. It persists only until the 219 ** next successful rollback is performed on the pager cache. Also, 220 ** SQLITE_FULL does not affect the sqlite3pager_get() and sqlite3pager_lookup() 221 ** APIs, they may still be used successfully. 222 */ 223 struct Pager { 224 u8 journalOpen; /* True if journal file descriptors is valid */ 225 u8 journalStarted; /* True if header of journal is synced */ 226 u8 useJournal; /* Use a rollback journal on this file */ 227 u8 noReadlock; /* Do not bother to obtain readlocks */ 228 u8 stmtOpen; /* True if the statement subjournal is open */ 229 u8 stmtInUse; /* True we are in a statement subtransaction */ 230 u8 stmtAutoopen; /* Open stmt journal when main journal is opened*/ 231 u8 noSync; /* Do not sync the journal if true */ 232 u8 fullSync; /* Do extra syncs of the journal for robustness */ 233 u8 full_fsync; /* Use F_FULLFSYNC when available */ 234 u8 state; /* PAGER_UNLOCK, _SHARED, _RESERVED, etc. */ 235 u8 tempFile; /* zFilename is a temporary file */ 236 u8 readOnly; /* True for a read-only database */ 237 u8 needSync; /* True if an fsync() is needed on the journal */ 238 u8 dirtyCache; /* True if cached pages have changed */ 239 u8 alwaysRollback; /* Disable dont_rollback() for all pages */ 240 u8 memDb; /* True to inhibit all file I/O */ 241 u8 setMaster; /* True if a m-j name has been written to jrnl */ 242 int errCode; /* One of several kinds of errors */ 243 int dbSize; /* Number of pages in the file */ 244 int origDbSize; /* dbSize before the current change */ 245 int stmtSize; /* Size of database (in pages) at stmt_begin() */ 246 int nRec; /* Number of pages written to the journal */ 247 u32 cksumInit; /* Quasi-random value added to every checksum */ 248 int stmtNRec; /* Number of records in stmt subjournal */ 249 int nExtra; /* Add this many bytes to each in-memory page */ 250 int pageSize; /* Number of bytes in a page */ 251 int nPage; /* Total number of in-memory pages */ 252 int nMaxPage; /* High water mark of nPage */ 253 int nRef; /* Number of in-memory pages with PgHdr.nRef>0 */ 254 int mxPage; /* Maximum number of pages to hold in cache */ 255 u8 *aInJournal; /* One bit for each page in the database file */ 256 u8 *aInStmt; /* One bit for each page in the database */ 257 char *zFilename; /* Name of the database file */ 258 char *zJournal; /* Name of the journal file */ 259 char *zDirectory; /* Directory hold database and journal files */ 260 OsFile *fd, *jfd; /* File descriptors for database and journal */ 261 OsFile *stfd; /* File descriptor for the statement subjournal*/ 262 BusyHandler *pBusyHandler; /* Pointer to sqlite.busyHandler */ 263 PgHdr *pFirst, *pLast; /* List of free pages */ 264 PgHdr *pFirstSynced; /* First free page with PgHdr.needSync==0 */ 265 PgHdr *pAll; /* List of all pages */ 266 PgHdr *pStmt; /* List of pages in the statement subjournal */ 267 PgHdr *pDirty; /* List of all dirty pages */ 268 i64 journalOff; /* Current byte offset in the journal file */ 269 i64 journalHdr; /* Byte offset to previous journal header */ 270 i64 stmtHdrOff; /* First journal header written this statement */ 271 i64 stmtCksum; /* cksumInit when statement was started */ 272 i64 stmtJSize; /* Size of journal at stmt_begin() */ 273 int sectorSize; /* Assumed sector size during rollback */ 274 #ifdef SQLITE_TEST 275 int nHit, nMiss, nOvfl; /* Cache hits, missing, and LRU overflows */ 276 int nRead,nWrite; /* Database pages read/written */ 277 #endif 278 void (*xDestructor)(void*,int); /* Call this routine when freeing pages */ 279 void (*xReiniter)(void*,int); /* Call this routine when reloading pages */ 280 void *(*xCodec)(void*,void*,Pgno,int); /* Routine for en/decoding data */ 281 void *pCodecArg; /* First argument to xCodec() */ 282 int nHash; /* Size of the pager hash table */ 283 PgHdr **aHash; /* Hash table to map page number to PgHdr */ 284 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 285 Pager *pNext; /* Linked list of pagers in this thread */ 286 #endif 287 }; 288 289 /* 290 ** If SQLITE_TEST is defined then increment the variable given in 291 ** the argument 292 */ 293 #ifdef SQLITE_TEST 294 # define TEST_INCR(x) x++ 295 #else 296 # define TEST_INCR(x) 297 #endif 298 299 /* 300 ** Journal files begin with the following magic string. The data 301 ** was obtained from /dev/random. It is used only as a sanity check. 302 ** 303 ** Since version 2.8.0, the journal format contains additional sanity 304 ** checking information. If the power fails while the journal is begin 305 ** written, semi-random garbage data might appear in the journal 306 ** file after power is restored. If an attempt is then made 307 ** to roll the journal back, the database could be corrupted. The additional 308 ** sanity checking data is an attempt to discover the garbage in the 309 ** journal and ignore it. 310 ** 311 ** The sanity checking information for the new journal format consists 312 ** of a 32-bit checksum on each page of data. The checksum covers both 313 ** the page number and the pPager->pageSize bytes of data for the page. 314 ** This cksum is initialized to a 32-bit random value that appears in the 315 ** journal file right after the header. The random initializer is important, 316 ** because garbage data that appears at the end of a journal is likely 317 ** data that was once in other files that have now been deleted. If the 318 ** garbage data came from an obsolete journal file, the checksums might 319 ** be correct. But by initializing the checksum to random value which 320 ** is different for every journal, we minimize that risk. 321 */ 322 static const unsigned char aJournalMagic[] = { 323 0xd9, 0xd5, 0x05, 0xf9, 0x20, 0xa1, 0x63, 0xd7, 324 }; 325 326 /* 327 ** The size of the header and of each page in the journal is determined 328 ** by the following macros. 329 */ 330 #define JOURNAL_PG_SZ(pPager) ((pPager->pageSize) + 8) 331 332 /* 333 ** The journal header size for this pager. In the future, this could be 334 ** set to some value read from the disk controller. The important 335 ** characteristic is that it is the same size as a disk sector. 336 */ 337 #define JOURNAL_HDR_SZ(pPager) (pPager->sectorSize) 338 339 /* 340 ** The macro MEMDB is true if we are dealing with an in-memory database. 341 ** We do this as a macro so that if the SQLITE_OMIT_MEMORYDB macro is set, 342 ** the value of MEMDB will be a constant and the compiler will optimize 343 ** out code that would never execute. 344 */ 345 #ifdef SQLITE_OMIT_MEMORYDB 346 # define MEMDB 0 347 #else 348 # define MEMDB pPager->memDb 349 #endif 350 351 /* 352 ** The default size of a disk sector 353 */ 354 #ifndef PAGER_SECTOR_SIZE 355 # define PAGER_SECTOR_SIZE 512 356 #endif 357 358 /* 359 ** Page number PAGER_MJ_PGNO is never used in an SQLite database (it is 360 ** reserved for working around a windows/posix incompatibility). It is 361 ** used in the journal to signify that the remainder of the journal file 362 ** is devoted to storing a master journal name - there are no more pages to 363 ** roll back. See comments for function writeMasterJournal() for details. 364 */ 365 /* #define PAGER_MJ_PGNO(x) (PENDING_BYTE/((x)->pageSize)) */ 366 #define PAGER_MJ_PGNO(x) ((PENDING_BYTE/((x)->pageSize))+1) 367 368 /* 369 ** The maximum legal page number is (2^31 - 1). 370 */ 371 #define PAGER_MAX_PGNO 2147483647 372 373 /* 374 ** Enable reference count tracking (for debugging) here: 375 */ 376 #ifdef SQLITE_TEST 377 int pager3_refinfo_enable = 0; 378 static void pager_refinfo(PgHdr *p){ 379 static int cnt = 0; 380 if( !pager3_refinfo_enable ) return; 381 sqlite3DebugPrintf( 382 "REFCNT: %4d addr=%p nRef=%-3d total=%d\n", 383 p->pgno, PGHDR_TO_DATA(p), p->nRef, p->pPager->nRef 384 ); 385 cnt++; /* Something to set a breakpoint on */ 386 } 387 # define REFINFO(X) pager_refinfo(X) 388 #else 389 # define REFINFO(X) 390 #endif 391 392 393 /* 394 ** Change the size of the pager hash table to N. N must be a power 395 ** of two. 396 */ 397 static void pager_resize_hash_table(Pager *pPager, int N){ 398 PgHdr **aHash, *pPg; 399 assert( N>0 && (N&(N-1))==0 ); 400 aHash = sqliteMalloc( sizeof(aHash[0])*N ); 401 if( aHash==0 ){ 402 /* Failure to rehash is not an error. It is only a performance hit. */ 403 return; 404 } 405 sqliteFree(pPager->aHash); 406 pPager->nHash = N; 407 pPager->aHash = aHash; 408 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 409 int h; 410 if( pPg->pgno==0 ){ 411 assert( pPg->pNextHash==0 && pPg->pPrevHash==0 ); 412 continue; 413 } 414 h = pPg->pgno & (N-1); 415 pPg->pNextHash = aHash[h]; 416 if( aHash[h] ){ 417 aHash[h]->pPrevHash = pPg; 418 } 419 aHash[h] = pPg; 420 pPg->pPrevHash = 0; 421 } 422 } 423 424 /* 425 ** Read a 32-bit integer from the given file descriptor. Store the integer 426 ** that is read in *pRes. Return SQLITE_OK if everything worked, or an 427 ** error code is something goes wrong. 428 ** 429 ** All values are stored on disk as big-endian. 430 */ 431 static int read32bits(OsFile *fd, u32 *pRes){ 432 unsigned char ac[4]; 433 int rc = sqlite3OsRead(fd, ac, sizeof(ac)); 434 if( rc==SQLITE_OK ){ 435 *pRes = (ac[0]<<24) | (ac[1]<<16) | (ac[2]<<8) | ac[3]; 436 } 437 return rc; 438 } 439 440 /* 441 ** Write a 32-bit integer into a string buffer in big-endian byte order. 442 */ 443 static void put32bits(char *ac, u32 val){ 444 ac[0] = (val>>24) & 0xff; 445 ac[1] = (val>>16) & 0xff; 446 ac[2] = (val>>8) & 0xff; 447 ac[3] = val & 0xff; 448 } 449 450 /* 451 ** Write a 32-bit integer into the given file descriptor. Return SQLITE_OK 452 ** on success or an error code is something goes wrong. 453 */ 454 static int write32bits(OsFile *fd, u32 val){ 455 char ac[4]; 456 put32bits(ac, val); 457 return sqlite3OsWrite(fd, ac, 4); 458 } 459 460 /* 461 ** Read a 32-bit integer at offset 'offset' from the page identified by 462 ** page header 'p'. 463 */ 464 static u32 retrieve32bits(PgHdr *p, int offset){ 465 unsigned char *ac; 466 ac = &((unsigned char*)PGHDR_TO_DATA(p))[offset]; 467 return (ac[0]<<24) | (ac[1]<<16) | (ac[2]<<8) | ac[3]; 468 } 469 470 471 /* 472 ** This function should be called when an error occurs within the pager 473 ** code. The first argument is a pointer to the pager structure, the 474 ** second the error-code about to be returned by a pager API function. 475 ** The value returned is a copy of the second argument to this function. 476 ** 477 ** If the second argument is SQLITE_IOERR, SQLITE_CORRUPT or SQLITE_PROTOCOL, 478 ** the error becomes persistent. All subsequent API calls on this Pager 479 ** will immediately return the same error code. 480 */ 481 static int pager_error(Pager *pPager, int rc){ 482 int rc2 = rc & 0xff; 483 assert( pPager->errCode==SQLITE_FULL || pPager->errCode==SQLITE_OK ); 484 if( 485 rc2==SQLITE_FULL || 486 rc2==SQLITE_IOERR || 487 rc2==SQLITE_CORRUPT || 488 rc2==SQLITE_PROTOCOL 489 ){ 490 pPager->errCode = rc; 491 } 492 return rc; 493 } 494 495 #ifdef SQLITE_CHECK_PAGES 496 /* 497 ** Return a 32-bit hash of the page data for pPage. 498 */ 499 static u32 pager_pagehash(PgHdr *pPage){ 500 u32 hash = 0; 501 int i; 502 unsigned char *pData = (unsigned char *)PGHDR_TO_DATA(pPage); 503 for(i=0; i<pPage->pPager->pageSize; i++){ 504 hash = (hash+i)^pData[i]; 505 } 506 return hash; 507 } 508 509 /* 510 ** The CHECK_PAGE macro takes a PgHdr* as an argument. If SQLITE_CHECK_PAGES 511 ** is defined, and NDEBUG is not defined, an assert() statement checks 512 ** that the page is either dirty or still matches the calculated page-hash. 513 */ 514 #define CHECK_PAGE(x) checkPage(x) 515 static void checkPage(PgHdr *pPg){ 516 Pager *pPager = pPg->pPager; 517 assert( !pPg->pageHash || pPager->errCode || MEMDB || pPg->dirty || 518 pPg->pageHash==pager_pagehash(pPg) ); 519 } 520 521 #else 522 #define CHECK_PAGE(x) 523 #endif 524 525 /* 526 ** When this is called the journal file for pager pPager must be open. 527 ** The master journal file name is read from the end of the file and 528 ** written into memory obtained from sqliteMalloc(). *pzMaster is 529 ** set to point at the memory and SQLITE_OK returned. The caller must 530 ** sqliteFree() *pzMaster. 531 ** 532 ** If no master journal file name is present *pzMaster is set to 0 and 533 ** SQLITE_OK returned. 534 */ 535 static int readMasterJournal(OsFile *pJrnl, char **pzMaster){ 536 int rc; 537 u32 len; 538 i64 szJ; 539 u32 cksum; 540 int i; 541 unsigned char aMagic[8]; /* A buffer to hold the magic header */ 542 543 *pzMaster = 0; 544 545 rc = sqlite3OsFileSize(pJrnl, &szJ); 546 if( rc!=SQLITE_OK || szJ<16 ) return rc; 547 548 rc = sqlite3OsSeek(pJrnl, szJ-16); 549 if( rc!=SQLITE_OK ) return rc; 550 551 rc = read32bits(pJrnl, &len); 552 if( rc!=SQLITE_OK ) return rc; 553 554 rc = read32bits(pJrnl, &cksum); 555 if( rc!=SQLITE_OK ) return rc; 556 557 rc = sqlite3OsRead(pJrnl, aMagic, 8); 558 if( rc!=SQLITE_OK || memcmp(aMagic, aJournalMagic, 8) ) return rc; 559 560 rc = sqlite3OsSeek(pJrnl, szJ-16-len); 561 if( rc!=SQLITE_OK ) return rc; 562 563 *pzMaster = (char *)sqliteMalloc(len+1); 564 if( !*pzMaster ){ 565 return SQLITE_NOMEM; 566 } 567 rc = sqlite3OsRead(pJrnl, *pzMaster, len); 568 if( rc!=SQLITE_OK ){ 569 sqliteFree(*pzMaster); 570 *pzMaster = 0; 571 return rc; 572 } 573 574 /* See if the checksum matches the master journal name */ 575 for(i=0; i<len; i++){ 576 cksum -= (*pzMaster)[i]; 577 } 578 if( cksum ){ 579 /* If the checksum doesn't add up, then one or more of the disk sectors 580 ** containing the master journal filename is corrupted. This means 581 ** definitely roll back, so just return SQLITE_OK and report a (nul) 582 ** master-journal filename. 583 */ 584 sqliteFree(*pzMaster); 585 *pzMaster = 0; 586 }else{ 587 (*pzMaster)[len] = '\0'; 588 } 589 590 return SQLITE_OK; 591 } 592 593 /* 594 ** Seek the journal file descriptor to the next sector boundary where a 595 ** journal header may be read or written. Pager.journalOff is updated with 596 ** the new seek offset. 597 ** 598 ** i.e for a sector size of 512: 599 ** 600 ** Input Offset Output Offset 601 ** --------------------------------------- 602 ** 0 0 603 ** 512 512 604 ** 100 512 605 ** 2000 2048 606 ** 607 */ 608 static int seekJournalHdr(Pager *pPager){ 609 i64 offset = 0; 610 i64 c = pPager->journalOff; 611 if( c ){ 612 offset = ((c-1)/JOURNAL_HDR_SZ(pPager) + 1) * JOURNAL_HDR_SZ(pPager); 613 } 614 assert( offset%JOURNAL_HDR_SZ(pPager)==0 ); 615 assert( offset>=c ); 616 assert( (offset-c)<JOURNAL_HDR_SZ(pPager) ); 617 pPager->journalOff = offset; 618 return sqlite3OsSeek(pPager->jfd, pPager->journalOff); 619 } 620 621 /* 622 ** The journal file must be open when this routine is called. A journal 623 ** header (JOURNAL_HDR_SZ bytes) is written into the journal file at the 624 ** current location. 625 ** 626 ** The format for the journal header is as follows: 627 ** - 8 bytes: Magic identifying journal format. 628 ** - 4 bytes: Number of records in journal, or -1 no-sync mode is on. 629 ** - 4 bytes: Random number used for page hash. 630 ** - 4 bytes: Initial database page count. 631 ** - 4 bytes: Sector size used by the process that wrote this journal. 632 ** 633 ** Followed by (JOURNAL_HDR_SZ - 24) bytes of unused space. 634 */ 635 static int writeJournalHdr(Pager *pPager){ 636 char zHeader[sizeof(aJournalMagic)+16]; 637 638 int rc = seekJournalHdr(pPager); 639 if( rc ) return rc; 640 641 pPager->journalHdr = pPager->journalOff; 642 if( pPager->stmtHdrOff==0 ){ 643 pPager->stmtHdrOff = pPager->journalHdr; 644 } 645 pPager->journalOff += JOURNAL_HDR_SZ(pPager); 646 647 /* FIX ME: 648 ** 649 ** Possibly for a pager not in no-sync mode, the journal magic should not 650 ** be written until nRec is filled in as part of next syncJournal(). 651 ** 652 ** Actually maybe the whole journal header should be delayed until that 653 ** point. Think about this. 654 */ 655 memcpy(zHeader, aJournalMagic, sizeof(aJournalMagic)); 656 /* The nRec Field. 0xFFFFFFFF for no-sync journals. */ 657 put32bits(&zHeader[sizeof(aJournalMagic)], pPager->noSync ? 0xffffffff : 0); 658 /* The random check-hash initialiser */ 659 sqlite3Randomness(sizeof(pPager->cksumInit), &pPager->cksumInit); 660 put32bits(&zHeader[sizeof(aJournalMagic)+4], pPager->cksumInit); 661 /* The initial database size */ 662 put32bits(&zHeader[sizeof(aJournalMagic)+8], pPager->dbSize); 663 /* The assumed sector size for this process */ 664 put32bits(&zHeader[sizeof(aJournalMagic)+12], pPager->sectorSize); 665 rc = sqlite3OsWrite(pPager->jfd, zHeader, sizeof(zHeader)); 666 667 /* The journal header has been written successfully. Seek the journal 668 ** file descriptor to the end of the journal header sector. 669 */ 670 if( rc==SQLITE_OK ){ 671 rc = sqlite3OsSeek(pPager->jfd, pPager->journalOff-1); 672 if( rc==SQLITE_OK ){ 673 rc = sqlite3OsWrite(pPager->jfd, "\000", 1); 674 } 675 } 676 return rc; 677 } 678 679 /* 680 ** The journal file must be open when this is called. A journal header file 681 ** (JOURNAL_HDR_SZ bytes) is read from the current location in the journal 682 ** file. See comments above function writeJournalHdr() for a description of 683 ** the journal header format. 684 ** 685 ** If the header is read successfully, *nRec is set to the number of 686 ** page records following this header and *dbSize is set to the size of the 687 ** database before the transaction began, in pages. Also, pPager->cksumInit 688 ** is set to the value read from the journal header. SQLITE_OK is returned 689 ** in this case. 690 ** 691 ** If the journal header file appears to be corrupted, SQLITE_DONE is 692 ** returned and *nRec and *dbSize are not set. If JOURNAL_HDR_SZ bytes 693 ** cannot be read from the journal file an error code is returned. 694 */ 695 static int readJournalHdr( 696 Pager *pPager, 697 i64 journalSize, 698 u32 *pNRec, 699 u32 *pDbSize 700 ){ 701 int rc; 702 unsigned char aMagic[8]; /* A buffer to hold the magic header */ 703 704 rc = seekJournalHdr(pPager); 705 if( rc ) return rc; 706 707 if( pPager->journalOff+JOURNAL_HDR_SZ(pPager) > journalSize ){ 708 return SQLITE_DONE; 709 } 710 711 rc = sqlite3OsRead(pPager->jfd, aMagic, sizeof(aMagic)); 712 if( rc ) return rc; 713 714 if( memcmp(aMagic, aJournalMagic, sizeof(aMagic))!=0 ){ 715 return SQLITE_DONE; 716 } 717 718 rc = read32bits(pPager->jfd, pNRec); 719 if( rc ) return rc; 720 721 rc = read32bits(pPager->jfd, &pPager->cksumInit); 722 if( rc ) return rc; 723 724 rc = read32bits(pPager->jfd, pDbSize); 725 if( rc ) return rc; 726 727 /* Update the assumed sector-size to match the value used by 728 ** the process that created this journal. If this journal was 729 ** created by a process other than this one, then this routine 730 ** is being called from within pager_playback(). The local value 731 ** of Pager.sectorSize is restored at the end of that routine. 732 */ 733 rc = read32bits(pPager->jfd, (u32 *)&pPager->sectorSize); 734 if( rc ) return rc; 735 736 pPager->journalOff += JOURNAL_HDR_SZ(pPager); 737 rc = sqlite3OsSeek(pPager->jfd, pPager->journalOff); 738 return rc; 739 } 740 741 742 /* 743 ** Write the supplied master journal name into the journal file for pager 744 ** pPager at the current location. The master journal name must be the last 745 ** thing written to a journal file. If the pager is in full-sync mode, the 746 ** journal file descriptor is advanced to the next sector boundary before 747 ** anything is written. The format is: 748 ** 749 ** + 4 bytes: PAGER_MJ_PGNO. 750 ** + N bytes: length of master journal name. 751 ** + 4 bytes: N 752 ** + 4 bytes: Master journal name checksum. 753 ** + 8 bytes: aJournalMagic[]. 754 ** 755 ** The master journal page checksum is the sum of the bytes in the master 756 ** journal name. 757 ** 758 ** If zMaster is a NULL pointer (occurs for a single database transaction), 759 ** this call is a no-op. 760 */ 761 static int writeMasterJournal(Pager *pPager, const char *zMaster){ 762 int rc; 763 int len; 764 int i; 765 u32 cksum = 0; 766 char zBuf[sizeof(aJournalMagic)+2*4]; 767 768 if( !zMaster || pPager->setMaster) return SQLITE_OK; 769 pPager->setMaster = 1; 770 771 len = strlen(zMaster); 772 for(i=0; i<len; i++){ 773 cksum += zMaster[i]; 774 } 775 776 /* If in full-sync mode, advance to the next disk sector before writing 777 ** the master journal name. This is in case the previous page written to 778 ** the journal has already been synced. 779 */ 780 if( pPager->fullSync ){ 781 rc = seekJournalHdr(pPager); 782 if( rc!=SQLITE_OK ) return rc; 783 } 784 pPager->journalOff += (len+20); 785 786 rc = write32bits(pPager->jfd, PAGER_MJ_PGNO(pPager)); 787 if( rc!=SQLITE_OK ) return rc; 788 789 rc = sqlite3OsWrite(pPager->jfd, zMaster, len); 790 if( rc!=SQLITE_OK ) return rc; 791 792 put32bits(zBuf, len); 793 put32bits(&zBuf[4], cksum); 794 memcpy(&zBuf[8], aJournalMagic, sizeof(aJournalMagic)); 795 rc = sqlite3OsWrite(pPager->jfd, zBuf, 8+sizeof(aJournalMagic)); 796 pPager->needSync = !pPager->noSync; 797 return rc; 798 } 799 800 /* 801 ** Add or remove a page from the list of all pages that are in the 802 ** statement journal. 803 ** 804 ** The Pager keeps a separate list of pages that are currently in 805 ** the statement journal. This helps the sqlite3pager_stmt_commit() 806 ** routine run MUCH faster for the common case where there are many 807 ** pages in memory but only a few are in the statement journal. 808 */ 809 static void page_add_to_stmt_list(PgHdr *pPg){ 810 Pager *pPager = pPg->pPager; 811 if( pPg->inStmt ) return; 812 assert( pPg->pPrevStmt==0 && pPg->pNextStmt==0 ); 813 pPg->pPrevStmt = 0; 814 if( pPager->pStmt ){ 815 pPager->pStmt->pPrevStmt = pPg; 816 } 817 pPg->pNextStmt = pPager->pStmt; 818 pPager->pStmt = pPg; 819 pPg->inStmt = 1; 820 } 821 static void page_remove_from_stmt_list(PgHdr *pPg){ 822 if( !pPg->inStmt ) return; 823 if( pPg->pPrevStmt ){ 824 assert( pPg->pPrevStmt->pNextStmt==pPg ); 825 pPg->pPrevStmt->pNextStmt = pPg->pNextStmt; 826 }else{ 827 assert( pPg->pPager->pStmt==pPg ); 828 pPg->pPager->pStmt = pPg->pNextStmt; 829 } 830 if( pPg->pNextStmt ){ 831 assert( pPg->pNextStmt->pPrevStmt==pPg ); 832 pPg->pNextStmt->pPrevStmt = pPg->pPrevStmt; 833 } 834 pPg->pNextStmt = 0; 835 pPg->pPrevStmt = 0; 836 pPg->inStmt = 0; 837 } 838 839 /* 840 ** Find a page in the hash table given its page number. Return 841 ** a pointer to the page or NULL if not found. 842 */ 843 static PgHdr *pager_lookup(Pager *pPager, Pgno pgno){ 844 PgHdr *p; 845 if( pPager->aHash==0 ) return 0; 846 p = pPager->aHash[pgno & (pPager->nHash-1)]; 847 while( p && p->pgno!=pgno ){ 848 p = p->pNextHash; 849 } 850 return p; 851 } 852 853 /* 854 ** Unlock the database file. 855 ** 856 ** Once all locks have been removed from the database file, other 857 ** processes or threads might change the file. So make sure all of 858 ** our internal cache is invalidated. 859 */ 860 static void pager_unlock(Pager *pPager){ 861 if( !MEMDB ){ 862 sqlite3OsUnlock(pPager->fd, NO_LOCK); 863 pPager->dbSize = -1; 864 } 865 pPager->state = PAGER_UNLOCK; 866 assert( pPager->pAll==0 ); 867 } 868 869 870 /* 871 ** Unlock the database and clear the in-memory cache. This routine 872 ** sets the state of the pager back to what it was when it was first 873 ** opened. Any outstanding pages are invalidated and subsequent attempts 874 ** to access those pages will likely result in a coredump. 875 */ 876 static void pager_reset(Pager *pPager){ 877 PgHdr *pPg, *pNext; 878 if( pPager->errCode ) return; 879 for(pPg=pPager->pAll; pPg; pPg=pNext){ 880 pNext = pPg->pNextAll; 881 sqliteFree(pPg); 882 } 883 pPager->pFirst = 0; 884 pPager->pFirstSynced = 0; 885 pPager->pLast = 0; 886 pPager->pAll = 0; 887 pPager->nHash = 0; 888 sqliteFree(pPager->aHash); 889 pPager->nPage = 0; 890 pPager->aHash = 0; 891 if( pPager->state>=PAGER_RESERVED ){ 892 sqlite3pager_rollback(pPager); 893 } 894 pager_unlock(pPager); 895 pPager->nRef = 0; 896 assert( pPager->errCode || (pPager->journalOpen==0 && pPager->stmtOpen==0) ); 897 } 898 899 /* 900 ** When this routine is called, the pager has the journal file open and 901 ** a RESERVED or EXCLUSIVE lock on the database. This routine releases 902 ** the database lock and acquires a SHARED lock in its place. The journal 903 ** file is deleted and closed. 904 ** 905 ** TODO: Consider keeping the journal file open for temporary databases. 906 ** This might give a performance improvement on windows where opening 907 ** a file is an expensive operation. 908 */ 909 static int pager_unwritelock(Pager *pPager){ 910 PgHdr *pPg; 911 int rc; 912 assert( !MEMDB ); 913 if( pPager->state<PAGER_RESERVED ){ 914 return SQLITE_OK; 915 } 916 sqlite3pager_stmt_commit(pPager); 917 if( pPager->stmtOpen ){ 918 sqlite3OsClose(&pPager->stfd); 919 pPager->stmtOpen = 0; 920 } 921 if( pPager->journalOpen ){ 922 sqlite3OsClose(&pPager->jfd); 923 pPager->journalOpen = 0; 924 sqlite3OsDelete(pPager->zJournal); 925 sqliteFree( pPager->aInJournal ); 926 pPager->aInJournal = 0; 927 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 928 pPg->inJournal = 0; 929 pPg->dirty = 0; 930 pPg->needSync = 0; 931 #ifdef SQLITE_CHECK_PAGES 932 pPg->pageHash = pager_pagehash(pPg); 933 #endif 934 } 935 pPager->pDirty = 0; 936 pPager->dirtyCache = 0; 937 pPager->nRec = 0; 938 }else{ 939 assert( pPager->aInJournal==0 ); 940 assert( pPager->dirtyCache==0 || pPager->useJournal==0 ); 941 } 942 rc = sqlite3OsUnlock(pPager->fd, SHARED_LOCK); 943 pPager->state = PAGER_SHARED; 944 pPager->origDbSize = 0; 945 pPager->setMaster = 0; 946 pPager->needSync = 0; 947 pPager->pFirstSynced = pPager->pFirst; 948 pPager->dbSize = -1; 949 return rc; 950 } 951 952 /* 953 ** Compute and return a checksum for the page of data. 954 ** 955 ** This is not a real checksum. It is really just the sum of the 956 ** random initial value and the page number. We experimented with 957 ** a checksum of the entire data, but that was found to be too slow. 958 ** 959 ** Note that the page number is stored at the beginning of data and 960 ** the checksum is stored at the end. This is important. If journal 961 ** corruption occurs due to a power failure, the most likely scenario 962 ** is that one end or the other of the record will be changed. It is 963 ** much less likely that the two ends of the journal record will be 964 ** correct and the middle be corrupt. Thus, this "checksum" scheme, 965 ** though fast and simple, catches the mostly likely kind of corruption. 966 ** 967 ** FIX ME: Consider adding every 200th (or so) byte of the data to the 968 ** checksum. That way if a single page spans 3 or more disk sectors and 969 ** only the middle sector is corrupt, we will still have a reasonable 970 ** chance of failing the checksum and thus detecting the problem. 971 */ 972 static u32 pager_cksum(Pager *pPager, const u8 *aData){ 973 u32 cksum = pPager->cksumInit; 974 int i = pPager->pageSize-200; 975 while( i>0 ){ 976 cksum += aData[i]; 977 i -= 200; 978 } 979 return cksum; 980 } 981 982 /* Forward declaration */ 983 static void makeClean(PgHdr*); 984 985 /* 986 ** Read a single page from the journal file opened on file descriptor 987 ** jfd. Playback this one page. 988 ** 989 ** If useCksum==0 it means this journal does not use checksums. Checksums 990 ** are not used in statement journals because statement journals do not 991 ** need to survive power failures. 992 */ 993 static int pager_playback_one_page(Pager *pPager, OsFile *jfd, int useCksum){ 994 int rc; 995 PgHdr *pPg; /* An existing page in the cache */ 996 Pgno pgno; /* The page number of a page in journal */ 997 u32 cksum; /* Checksum used for sanity checking */ 998 u8 aData[SQLITE_MAX_PAGE_SIZE]; /* Temp storage for a page */ 999 1000 /* useCksum should be true for the main journal and false for 1001 ** statement journals. Verify that this is always the case 1002 */ 1003 assert( jfd == (useCksum ? pPager->jfd : pPager->stfd) ); 1004 1005 1006 rc = read32bits(jfd, &pgno); 1007 if( rc!=SQLITE_OK ) return rc; 1008 rc = sqlite3OsRead(jfd, &aData, pPager->pageSize); 1009 if( rc!=SQLITE_OK ) return rc; 1010 pPager->journalOff += pPager->pageSize + 4; 1011 1012 /* Sanity checking on the page. This is more important that I originally 1013 ** thought. If a power failure occurs while the journal is being written, 1014 ** it could cause invalid data to be written into the journal. We need to 1015 ** detect this invalid data (with high probability) and ignore it. 1016 */ 1017 if( pgno==0 || pgno==PAGER_MJ_PGNO(pPager) ){ 1018 return SQLITE_DONE; 1019 } 1020 if( pgno>(unsigned)pPager->dbSize ){ 1021 return SQLITE_OK; 1022 } 1023 if( useCksum ){ 1024 rc = read32bits(jfd, &cksum); 1025 if( rc ) return rc; 1026 pPager->journalOff += 4; 1027 if( pager_cksum(pPager, aData)!=cksum ){ 1028 return SQLITE_DONE; 1029 } 1030 } 1031 1032 assert( pPager->state==PAGER_RESERVED || pPager->state>=PAGER_EXCLUSIVE ); 1033 1034 /* If the pager is in RESERVED state, then there must be a copy of this 1035 ** page in the pager cache. In this case just update the pager cache, 1036 ** not the database file. The page is left marked dirty in this case. 1037 ** 1038 ** If in EXCLUSIVE state, then we update the pager cache if it exists 1039 ** and the main file. The page is then marked not dirty. 1040 ** 1041 ** Ticket #1171: The statement journal might contain page content that is 1042 ** different from the page content at the start of the transaction. 1043 ** This occurs when a page is changed prior to the start of a statement 1044 ** then changed again within the statement. When rolling back such a 1045 ** statement we must not write to the original database unless we know 1046 ** for certain that original page contents are in the main rollback 1047 ** journal. Otherwise, if a full ROLLBACK occurs after the statement 1048 ** rollback the full ROLLBACK will not restore the page to its original 1049 ** content. Two conditions must be met before writing to the database 1050 ** files. (1) the database must be locked. (2) we know that the original 1051 ** page content is in the main journal either because the page is not in 1052 ** cache or else it is marked as needSync==0. 1053 */ 1054 pPg = pager_lookup(pPager, pgno); 1055 assert( pPager->state>=PAGER_EXCLUSIVE || pPg!=0 ); 1056 TRACE3("PLAYBACK %d page %d\n", PAGERID(pPager), pgno); 1057 if( pPager->state>=PAGER_EXCLUSIVE && (pPg==0 || pPg->needSync==0) ){ 1058 rc = sqlite3OsSeek(pPager->fd, (pgno-1)*(i64)pPager->pageSize); 1059 if( rc==SQLITE_OK ){ 1060 rc = sqlite3OsWrite(pPager->fd, aData, pPager->pageSize); 1061 } 1062 if( pPg ){ 1063 makeClean(pPg); 1064 } 1065 } 1066 if( pPg ){ 1067 /* No page should ever be explicitly rolled back that is in use, except 1068 ** for page 1 which is held in use in order to keep the lock on the 1069 ** database active. However such a page may be rolled back as a result 1070 ** of an internal error resulting in an automatic call to 1071 ** sqlite3pager_rollback(). 1072 */ 1073 void *pData; 1074 /* assert( pPg->nRef==0 || pPg->pgno==1 ); */ 1075 pData = PGHDR_TO_DATA(pPg); 1076 memcpy(pData, aData, pPager->pageSize); 1077 if( pPager->xDestructor ){ /*** FIX ME: Should this be xReinit? ***/ 1078 pPager->xDestructor(pData, pPager->pageSize); 1079 } 1080 #ifdef SQLITE_CHECK_PAGES 1081 pPg->pageHash = pager_pagehash(pPg); 1082 #endif 1083 CODEC1(pPager, pData, pPg->pgno, 3); 1084 } 1085 return rc; 1086 } 1087 1088 /* 1089 ** Parameter zMaster is the name of a master journal file. A single journal 1090 ** file that referred to the master journal file has just been rolled back. 1091 ** This routine checks if it is possible to delete the master journal file, 1092 ** and does so if it is. 1093 ** 1094 ** The master journal file contains the names of all child journals. 1095 ** To tell if a master journal can be deleted, check to each of the 1096 ** children. If all children are either missing or do not refer to 1097 ** a different master journal, then this master journal can be deleted. 1098 */ 1099 static int pager_delmaster(const char *zMaster){ 1100 int rc; 1101 int master_open = 0; 1102 OsFile *master = 0; 1103 char *zMasterJournal = 0; /* Contents of master journal file */ 1104 i64 nMasterJournal; /* Size of master journal file */ 1105 1106 /* Open the master journal file exclusively in case some other process 1107 ** is running this routine also. Not that it makes too much difference. 1108 */ 1109 rc = sqlite3OsOpenReadOnly(zMaster, &master); 1110 if( rc!=SQLITE_OK ) goto delmaster_out; 1111 master_open = 1; 1112 rc = sqlite3OsFileSize(master, &nMasterJournal); 1113 if( rc!=SQLITE_OK ) goto delmaster_out; 1114 1115 if( nMasterJournal>0 ){ 1116 char *zJournal; 1117 char *zMasterPtr = 0; 1118 1119 /* Load the entire master journal file into space obtained from 1120 ** sqliteMalloc() and pointed to by zMasterJournal. 1121 */ 1122 zMasterJournal = (char *)sqliteMalloc(nMasterJournal); 1123 if( !zMasterJournal ){ 1124 rc = SQLITE_NOMEM; 1125 goto delmaster_out; 1126 } 1127 rc = sqlite3OsRead(master, zMasterJournal, nMasterJournal); 1128 if( rc!=SQLITE_OK ) goto delmaster_out; 1129 1130 zJournal = zMasterJournal; 1131 while( (zJournal-zMasterJournal)<nMasterJournal ){ 1132 if( sqlite3OsFileExists(zJournal) ){ 1133 /* One of the journals pointed to by the master journal exists. 1134 ** Open it and check if it points at the master journal. If 1135 ** so, return without deleting the master journal file. 1136 */ 1137 OsFile *journal = 0; 1138 int c; 1139 1140 rc = sqlite3OsOpenReadOnly(zJournal, &journal); 1141 if( rc!=SQLITE_OK ){ 1142 goto delmaster_out; 1143 } 1144 1145 rc = readMasterJournal(journal, &zMasterPtr); 1146 sqlite3OsClose(&journal); 1147 if( rc!=SQLITE_OK ){ 1148 goto delmaster_out; 1149 } 1150 1151 c = zMasterPtr!=0 && strcmp(zMasterPtr, zMaster)==0; 1152 sqliteFree(zMasterPtr); 1153 if( c ){ 1154 /* We have a match. Do not delete the master journal file. */ 1155 goto delmaster_out; 1156 } 1157 } 1158 zJournal += (strlen(zJournal)+1); 1159 } 1160 } 1161 1162 sqlite3OsDelete(zMaster); 1163 1164 delmaster_out: 1165 if( zMasterJournal ){ 1166 sqliteFree(zMasterJournal); 1167 } 1168 if( master_open ){ 1169 sqlite3OsClose(&master); 1170 } 1171 return rc; 1172 } 1173 1174 /* 1175 ** Make every page in the cache agree with what is on disk. In other words, 1176 ** reread the disk to reset the state of the cache. 1177 ** 1178 ** This routine is called after a rollback in which some of the dirty cache 1179 ** pages had never been written out to disk. We need to roll back the 1180 ** cache content and the easiest way to do that is to reread the old content 1181 ** back from the disk. 1182 */ 1183 static int pager_reload_cache(Pager *pPager){ 1184 PgHdr *pPg; 1185 int rc = SQLITE_OK; 1186 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 1187 char zBuf[SQLITE_MAX_PAGE_SIZE]; 1188 if( !pPg->dirty ) continue; 1189 if( (int)pPg->pgno <= pPager->origDbSize ){ 1190 rc = sqlite3OsSeek(pPager->fd, pPager->pageSize*(i64)(pPg->pgno-1)); 1191 if( rc==SQLITE_OK ){ 1192 rc = sqlite3OsRead(pPager->fd, zBuf, pPager->pageSize); 1193 } 1194 TRACE3("REFETCH %d page %d\n", PAGERID(pPager), pPg->pgno); 1195 if( rc ) break; 1196 CODEC1(pPager, zBuf, pPg->pgno, 2); 1197 }else{ 1198 memset(zBuf, 0, pPager->pageSize); 1199 } 1200 if( pPg->nRef==0 || memcmp(zBuf, PGHDR_TO_DATA(pPg), pPager->pageSize) ){ 1201 memcpy(PGHDR_TO_DATA(pPg), zBuf, pPager->pageSize); 1202 if( pPager->xReiniter ){ 1203 pPager->xReiniter(PGHDR_TO_DATA(pPg), pPager->pageSize); 1204 }else{ 1205 memset(PGHDR_TO_EXTRA(pPg, pPager), 0, pPager->nExtra); 1206 } 1207 } 1208 pPg->needSync = 0; 1209 pPg->dirty = 0; 1210 #ifdef SQLITE_CHECK_PAGES 1211 pPg->pageHash = pager_pagehash(pPg); 1212 #endif 1213 } 1214 pPager->pDirty = 0; 1215 return rc; 1216 } 1217 1218 /* 1219 ** Truncate the main file of the given pager to the number of pages 1220 ** indicated. 1221 */ 1222 static int pager_truncate(Pager *pPager, int nPage){ 1223 assert( pPager->state>=PAGER_EXCLUSIVE ); 1224 return sqlite3OsTruncate(pPager->fd, pPager->pageSize*(i64)nPage); 1225 } 1226 1227 /* 1228 ** Playback the journal and thus restore the database file to 1229 ** the state it was in before we started making changes. 1230 ** 1231 ** The journal file format is as follows: 1232 ** 1233 ** (1) 8 byte prefix. A copy of aJournalMagic[]. 1234 ** (2) 4 byte big-endian integer which is the number of valid page records 1235 ** in the journal. If this value is 0xffffffff, then compute the 1236 ** number of page records from the journal size. 1237 ** (3) 4 byte big-endian integer which is the initial value for the 1238 ** sanity checksum. 1239 ** (4) 4 byte integer which is the number of pages to truncate the 1240 ** database to during a rollback. 1241 ** (5) 4 byte integer which is the number of bytes in the master journal 1242 ** name. The value may be zero (indicate that there is no master 1243 ** journal.) 1244 ** (6) N bytes of the master journal name. The name will be nul-terminated 1245 ** and might be shorter than the value read from (5). If the first byte 1246 ** of the name is \000 then there is no master journal. The master 1247 ** journal name is stored in UTF-8. 1248 ** (7) Zero or more pages instances, each as follows: 1249 ** + 4 byte page number. 1250 ** + pPager->pageSize bytes of data. 1251 ** + 4 byte checksum 1252 ** 1253 ** When we speak of the journal header, we mean the first 6 items above. 1254 ** Each entry in the journal is an instance of the 7th item. 1255 ** 1256 ** Call the value from the second bullet "nRec". nRec is the number of 1257 ** valid page entries in the journal. In most cases, you can compute the 1258 ** value of nRec from the size of the journal file. But if a power 1259 ** failure occurred while the journal was being written, it could be the 1260 ** case that the size of the journal file had already been increased but 1261 ** the extra entries had not yet made it safely to disk. In such a case, 1262 ** the value of nRec computed from the file size would be too large. For 1263 ** that reason, we always use the nRec value in the header. 1264 ** 1265 ** If the nRec value is 0xffffffff it means that nRec should be computed 1266 ** from the file size. This value is used when the user selects the 1267 ** no-sync option for the journal. A power failure could lead to corruption 1268 ** in this case. But for things like temporary table (which will be 1269 ** deleted when the power is restored) we don't care. 1270 ** 1271 ** If the file opened as the journal file is not a well-formed 1272 ** journal file then all pages up to the first corrupted page are rolled 1273 ** back (or no pages if the journal header is corrupted). The journal file 1274 ** is then deleted and SQLITE_OK returned, just as if no corruption had 1275 ** been encountered. 1276 ** 1277 ** If an I/O or malloc() error occurs, the journal-file is not deleted 1278 ** and an error code is returned. 1279 */ 1280 static int pager_playback(Pager *pPager){ 1281 i64 szJ; /* Size of the journal file in bytes */ 1282 u32 nRec; /* Number of Records in the journal */ 1283 int i; /* Loop counter */ 1284 Pgno mxPg = 0; /* Size of the original file in pages */ 1285 int rc; /* Result code of a subroutine */ 1286 char *zMaster = 0; /* Name of master journal file if any */ 1287 1288 /* Figure out how many records are in the journal. Abort early if 1289 ** the journal is empty. 1290 */ 1291 assert( pPager->journalOpen ); 1292 rc = sqlite3OsFileSize(pPager->jfd, &szJ); 1293 if( rc!=SQLITE_OK ){ 1294 goto end_playback; 1295 } 1296 1297 /* Read the master journal name from the journal, if it is present. 1298 ** If a master journal file name is specified, but the file is not 1299 ** present on disk, then the journal is not hot and does not need to be 1300 ** played back. 1301 */ 1302 rc = readMasterJournal(pPager->jfd, &zMaster); 1303 assert( rc!=SQLITE_DONE ); 1304 if( rc!=SQLITE_OK || (zMaster && !sqlite3OsFileExists(zMaster)) ){ 1305 sqliteFree(zMaster); 1306 zMaster = 0; 1307 if( rc==SQLITE_DONE ) rc = SQLITE_OK; 1308 goto end_playback; 1309 } 1310 sqlite3OsSeek(pPager->jfd, 0); 1311 pPager->journalOff = 0; 1312 1313 /* This loop terminates either when the readJournalHdr() call returns 1314 ** SQLITE_DONE or an IO error occurs. */ 1315 while( 1 ){ 1316 1317 /* Read the next journal header from the journal file. If there are 1318 ** not enough bytes left in the journal file for a complete header, or 1319 ** it is corrupted, then a process must of failed while writing it. 1320 ** This indicates nothing more needs to be rolled back. 1321 */ 1322 rc = readJournalHdr(pPager, szJ, &nRec, &mxPg); 1323 if( rc!=SQLITE_OK ){ 1324 if( rc==SQLITE_DONE ){ 1325 rc = SQLITE_OK; 1326 } 1327 goto end_playback; 1328 } 1329 1330 /* If nRec is 0xffffffff, then this journal was created by a process 1331 ** working in no-sync mode. This means that the rest of the journal 1332 ** file consists of pages, there are no more journal headers. Compute 1333 ** the value of nRec based on this assumption. 1334 */ 1335 if( nRec==0xffffffff ){ 1336 assert( pPager->journalOff==JOURNAL_HDR_SZ(pPager) ); 1337 nRec = (szJ - JOURNAL_HDR_SZ(pPager))/JOURNAL_PG_SZ(pPager); 1338 } 1339 1340 /* If this is the first header read from the journal, truncate the 1341 ** database file back to it's original size. 1342 */ 1343 if( pPager->state>=PAGER_EXCLUSIVE && 1344 pPager->journalOff==JOURNAL_HDR_SZ(pPager) ){ 1345 assert( pPager->origDbSize==0 || pPager->origDbSize==mxPg ); 1346 rc = pager_truncate(pPager, mxPg); 1347 if( rc!=SQLITE_OK ){ 1348 goto end_playback; 1349 } 1350 pPager->dbSize = mxPg; 1351 } 1352 1353 /* Copy original pages out of the journal and back into the database file. 1354 */ 1355 for(i=0; i<nRec; i++){ 1356 rc = pager_playback_one_page(pPager, pPager->jfd, 1); 1357 if( rc!=SQLITE_OK ){ 1358 if( rc==SQLITE_DONE ){ 1359 rc = SQLITE_OK; 1360 pPager->journalOff = szJ; 1361 break; 1362 }else{ 1363 /* If we are unable to rollback a hot journal, then the database 1364 ** is probably not recoverable. Return CORRUPT. 1365 */ 1366 rc = SQLITE_CORRUPT; 1367 goto end_playback; 1368 } 1369 } 1370 } 1371 } 1372 /*NOTREACHED*/ 1373 assert( 0 ); 1374 1375 end_playback: 1376 if( rc==SQLITE_OK ){ 1377 rc = pager_unwritelock(pPager); 1378 } 1379 if( zMaster ){ 1380 /* If there was a master journal and this routine will return true, 1381 ** see if it is possible to delete the master journal. 1382 */ 1383 if( rc==SQLITE_OK ){ 1384 rc = pager_delmaster(zMaster); 1385 } 1386 sqliteFree(zMaster); 1387 } 1388 1389 /* The Pager.sectorSize variable may have been updated while rolling 1390 ** back a journal created by a process with a different PAGER_SECTOR_SIZE 1391 ** value. Reset it to the correct value for this process. 1392 */ 1393 pPager->sectorSize = PAGER_SECTOR_SIZE; 1394 return rc; 1395 } 1396 1397 /* 1398 ** Playback the statement journal. 1399 ** 1400 ** This is similar to playing back the transaction journal but with 1401 ** a few extra twists. 1402 ** 1403 ** (1) The number of pages in the database file at the start of 1404 ** the statement is stored in pPager->stmtSize, not in the 1405 ** journal file itself. 1406 ** 1407 ** (2) In addition to playing back the statement journal, also 1408 ** playback all pages of the transaction journal beginning 1409 ** at offset pPager->stmtJSize. 1410 */ 1411 static int pager_stmt_playback(Pager *pPager){ 1412 i64 szJ; /* Size of the full journal */ 1413 i64 hdrOff; 1414 int nRec; /* Number of Records */ 1415 int i; /* Loop counter */ 1416 int rc; 1417 1418 szJ = pPager->journalOff; 1419 #ifndef NDEBUG 1420 { 1421 i64 os_szJ; 1422 rc = sqlite3OsFileSize(pPager->jfd, &os_szJ); 1423 if( rc!=SQLITE_OK ) return rc; 1424 assert( szJ==os_szJ ); 1425 } 1426 #endif 1427 1428 /* Set hdrOff to be the offset to the first journal header written 1429 ** this statement transaction, or the end of the file if no journal 1430 ** header was written. 1431 */ 1432 hdrOff = pPager->stmtHdrOff; 1433 assert( pPager->fullSync || !hdrOff ); 1434 if( !hdrOff ){ 1435 hdrOff = szJ; 1436 } 1437 1438 /* Truncate the database back to its original size. 1439 */ 1440 if( pPager->state>=PAGER_EXCLUSIVE ){ 1441 rc = pager_truncate(pPager, pPager->stmtSize); 1442 } 1443 assert( pPager->state>=PAGER_SHARED ); 1444 pPager->dbSize = pPager->stmtSize; 1445 1446 /* Figure out how many records are in the statement journal. 1447 */ 1448 assert( pPager->stmtInUse && pPager->journalOpen ); 1449 sqlite3OsSeek(pPager->stfd, 0); 1450 nRec = pPager->stmtNRec; 1451 1452 /* Copy original pages out of the statement journal and back into the 1453 ** database file. Note that the statement journal omits checksums from 1454 ** each record since power-failure recovery is not important to statement 1455 ** journals. 1456 */ 1457 for(i=nRec-1; i>=0; i--){ 1458 rc = pager_playback_one_page(pPager, pPager->stfd, 0); 1459 assert( rc!=SQLITE_DONE ); 1460 if( rc!=SQLITE_OK ) goto end_stmt_playback; 1461 } 1462 1463 /* Now roll some pages back from the transaction journal. Pager.stmtJSize 1464 ** was the size of the journal file when this statement was started, so 1465 ** everything after that needs to be rolled back, either into the 1466 ** database, the memory cache, or both. 1467 ** 1468 ** If it is not zero, then Pager.stmtHdrOff is the offset to the start 1469 ** of the first journal header written during this statement transaction. 1470 */ 1471 rc = sqlite3OsSeek(pPager->jfd, pPager->stmtJSize); 1472 if( rc!=SQLITE_OK ){ 1473 goto end_stmt_playback; 1474 } 1475 pPager->journalOff = pPager->stmtJSize; 1476 pPager->cksumInit = pPager->stmtCksum; 1477 assert( JOURNAL_HDR_SZ(pPager)<(pPager->pageSize+8) ); 1478 while( pPager->journalOff <= (hdrOff-(pPager->pageSize+8)) ){ 1479 rc = pager_playback_one_page(pPager, pPager->jfd, 1); 1480 assert( rc!=SQLITE_DONE ); 1481 if( rc!=SQLITE_OK ) goto end_stmt_playback; 1482 } 1483 1484 while( pPager->journalOff < szJ ){ 1485 u32 nJRec; /* Number of Journal Records */ 1486 u32 dummy; 1487 rc = readJournalHdr(pPager, szJ, &nJRec, &dummy); 1488 if( rc!=SQLITE_OK ){ 1489 assert( rc!=SQLITE_DONE ); 1490 goto end_stmt_playback; 1491 } 1492 if( nJRec==0 ){ 1493 nJRec = (szJ - pPager->journalOff) / (pPager->pageSize+8); 1494 } 1495 for(i=nJRec-1; i>=0 && pPager->journalOff < szJ; i--){ 1496 rc = pager_playback_one_page(pPager, pPager->jfd, 1); 1497 assert( rc!=SQLITE_DONE ); 1498 if( rc!=SQLITE_OK ) goto end_stmt_playback; 1499 } 1500 } 1501 1502 pPager->journalOff = szJ; 1503 1504 end_stmt_playback: 1505 if( rc==SQLITE_OK) { 1506 pPager->journalOff = szJ; 1507 /* pager_reload_cache(pPager); */ 1508 } 1509 return rc; 1510 } 1511 1512 /* 1513 ** Change the maximum number of in-memory pages that are allowed. 1514 */ 1515 void sqlite3pager_set_cachesize(Pager *pPager, int mxPage){ 1516 if( mxPage>10 ){ 1517 pPager->mxPage = mxPage; 1518 }else{ 1519 pPager->mxPage = 10; 1520 } 1521 } 1522 1523 /* 1524 ** Adjust the robustness of the database to damage due to OS crashes 1525 ** or power failures by changing the number of syncs()s when writing 1526 ** the rollback journal. There are three levels: 1527 ** 1528 ** OFF sqlite3OsSync() is never called. This is the default 1529 ** for temporary and transient files. 1530 ** 1531 ** NORMAL The journal is synced once before writes begin on the 1532 ** database. This is normally adequate protection, but 1533 ** it is theoretically possible, though very unlikely, 1534 ** that an inopertune power failure could leave the journal 1535 ** in a state which would cause damage to the database 1536 ** when it is rolled back. 1537 ** 1538 ** FULL The journal is synced twice before writes begin on the 1539 ** database (with some additional information - the nRec field 1540 ** of the journal header - being written in between the two 1541 ** syncs). If we assume that writing a 1542 ** single disk sector is atomic, then this mode provides 1543 ** assurance that the journal will not be corrupted to the 1544 ** point of causing damage to the database during rollback. 1545 ** 1546 ** Numeric values associated with these states are OFF==1, NORMAL=2, 1547 ** and FULL=3. 1548 */ 1549 #ifndef SQLITE_OMIT_PAGER_PRAGMAS 1550 void sqlite3pager_set_safety_level(Pager *pPager, int level, int full_fsync){ 1551 pPager->noSync = level==1 || pPager->tempFile; 1552 pPager->fullSync = level==3 && !pPager->tempFile; 1553 pPager->full_fsync = full_fsync; 1554 if( pPager->noSync ) pPager->needSync = 0; 1555 } 1556 #endif 1557 1558 /* 1559 ** The following global variable is incremented whenever the library 1560 ** attempts to open a temporary file. This information is used for 1561 ** testing and analysis only. 1562 */ 1563 #ifdef SQLITE_TEST 1564 int sqlite3_opentemp_count = 0; 1565 #endif 1566 1567 /* 1568 ** Open a temporary file. Write the name of the file into zFile 1569 ** (zFile must be at least SQLITE_TEMPNAME_SIZE bytes long.) Write 1570 ** the file descriptor into *fd. Return SQLITE_OK on success or some 1571 ** other error code if we fail. 1572 ** 1573 ** The OS will automatically delete the temporary file when it is 1574 ** closed. 1575 */ 1576 static int sqlite3pager_opentemp(char *zFile, OsFile **pFd){ 1577 int cnt = 8; 1578 int rc; 1579 #ifdef SQLITE_TEST 1580 sqlite3_opentemp_count++; /* Used for testing and analysis only */ 1581 #endif 1582 do{ 1583 cnt--; 1584 sqlite3OsTempFileName(zFile); 1585 rc = sqlite3OsOpenExclusive(zFile, pFd, 1); 1586 }while( cnt>0 && rc!=SQLITE_OK && rc!=SQLITE_NOMEM ); 1587 return rc; 1588 } 1589 1590 /* 1591 ** Create a new page cache and put a pointer to the page cache in *ppPager. 1592 ** The file to be cached need not exist. The file is not locked until 1593 ** the first call to sqlite3pager_get() and is only held open until the 1594 ** last page is released using sqlite3pager_unref(). 1595 ** 1596 ** If zFilename is NULL then a randomly-named temporary file is created 1597 ** and used as the file to be cached. The file will be deleted 1598 ** automatically when it is closed. 1599 ** 1600 ** If zFilename is ":memory:" then all information is held in cache. 1601 ** It is never written to disk. This can be used to implement an 1602 ** in-memory database. 1603 */ 1604 int sqlite3pager_open( 1605 Pager **ppPager, /* Return the Pager structure here */ 1606 const char *zFilename, /* Name of the database file to open */ 1607 int nExtra, /* Extra bytes append to each in-memory page */ 1608 int flags /* flags controlling this file */ 1609 ){ 1610 Pager *pPager = 0; 1611 char *zFullPathname = 0; 1612 int nameLen; /* Compiler is wrong. This is always initialized before use */ 1613 OsFile *fd; 1614 int rc = SQLITE_OK; 1615 int i; 1616 int tempFile = 0; 1617 int memDb = 0; 1618 int readOnly = 0; 1619 int useJournal = (flags & PAGER_OMIT_JOURNAL)==0; 1620 int noReadlock = (flags & PAGER_NO_READLOCK)!=0; 1621 char zTemp[SQLITE_TEMPNAME_SIZE]; 1622 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 1623 /* A malloc() cannot fail in sqlite3ThreadData() as one or more calls to 1624 ** malloc() must have already been made by this thread before it gets 1625 ** to this point. This means the ThreadData must have been allocated already 1626 ** so that ThreadData.nAlloc can be set. It would be nice to assert 1627 ** that ThreadData.nAlloc is non-zero, but alas this breaks test cases 1628 ** written to invoke the pager directly. 1629 */ 1630 ThreadData *pTsd = sqlite3ThreadData(); 1631 assert( pTsd ); 1632 #endif 1633 1634 /* If malloc() has already failed return SQLITE_NOMEM. Before even 1635 ** testing for this, set *ppPager to NULL so the caller knows the pager 1636 ** structure was never allocated. 1637 */ 1638 *ppPager = 0; 1639 if( sqlite3MallocFailed() ){ 1640 return SQLITE_NOMEM; 1641 } 1642 memset(&fd, 0, sizeof(fd)); 1643 1644 /* Open the pager file and set zFullPathname to point at malloc()ed 1645 ** memory containing the complete filename (i.e. including the directory). 1646 */ 1647 if( zFilename && zFilename[0] ){ 1648 #ifndef SQLITE_OMIT_MEMORYDB 1649 if( strcmp(zFilename,":memory:")==0 ){ 1650 memDb = 1; 1651 zFullPathname = sqliteStrDup(""); 1652 }else 1653 #endif 1654 { 1655 zFullPathname = sqlite3OsFullPathname(zFilename); 1656 if( zFullPathname ){ 1657 rc = sqlite3OsOpenReadWrite(zFullPathname, &fd, &readOnly); 1658 } 1659 } 1660 }else{ 1661 rc = sqlite3pager_opentemp(zTemp, &fd); 1662 zFilename = zTemp; 1663 zFullPathname = sqlite3OsFullPathname(zFilename); 1664 if( rc==SQLITE_OK ){ 1665 tempFile = 1; 1666 } 1667 } 1668 1669 /* Allocate the Pager structure. As part of the same allocation, allocate 1670 ** space for the full paths of the file, directory and journal 1671 ** (Pager.zFilename, Pager.zDirectory and Pager.zJournal). 1672 */ 1673 if( zFullPathname ){ 1674 nameLen = strlen(zFullPathname); 1675 pPager = sqliteMalloc( sizeof(*pPager) + nameLen*3 + 30 ); 1676 } 1677 1678 /* If an error occured in either of the blocks above, free the memory 1679 ** pointed to by zFullPathname, free the Pager structure and close the 1680 ** file. Since the pager is not allocated there is no need to set 1681 ** any Pager.errMask variables. 1682 */ 1683 if( !pPager || !zFullPathname || rc!=SQLITE_OK ){ 1684 sqlite3OsClose(&fd); 1685 sqliteFree(zFullPathname); 1686 sqliteFree(pPager); 1687 return ((rc==SQLITE_OK)?SQLITE_NOMEM:rc); 1688 } 1689 1690 TRACE3("OPEN %d %s\n", FILEHANDLEID(fd), zFullPathname); 1691 pPager->zFilename = (char*)&pPager[1]; 1692 pPager->zDirectory = &pPager->zFilename[nameLen+1]; 1693 pPager->zJournal = &pPager->zDirectory[nameLen+1]; 1694 strcpy(pPager->zFilename, zFullPathname); 1695 strcpy(pPager->zDirectory, zFullPathname); 1696 1697 for(i=nameLen; i>0 && pPager->zDirectory[i-1]!='/'; i--){} 1698 if( i>0 ) pPager->zDirectory[i-1] = 0; 1699 strcpy(pPager->zJournal, zFullPathname); 1700 sqliteFree(zFullPathname); 1701 strcpy(&pPager->zJournal[nameLen], "-journal"); 1702 pPager->fd = fd; 1703 /* pPager->journalOpen = 0; */ 1704 pPager->useJournal = useJournal && !memDb; 1705 pPager->noReadlock = noReadlock && readOnly; 1706 /* pPager->stmtOpen = 0; */ 1707 /* pPager->stmtInUse = 0; */ 1708 /* pPager->nRef = 0; */ 1709 pPager->dbSize = memDb-1; 1710 pPager->pageSize = SQLITE_DEFAULT_PAGE_SIZE; 1711 /* pPager->stmtSize = 0; */ 1712 /* pPager->stmtJSize = 0; */ 1713 /* pPager->nPage = 0; */ 1714 /* pPager->nMaxPage = 0; */ 1715 pPager->mxPage = 100; 1716 assert( PAGER_UNLOCK==0 ); 1717 /* pPager->state = PAGER_UNLOCK; */ 1718 /* pPager->errMask = 0; */ 1719 pPager->tempFile = tempFile; 1720 pPager->memDb = memDb; 1721 pPager->readOnly = readOnly; 1722 /* pPager->needSync = 0; */ 1723 pPager->noSync = pPager->tempFile || !useJournal; 1724 pPager->fullSync = (pPager->noSync?0:1); 1725 /* pPager->pFirst = 0; */ 1726 /* pPager->pFirstSynced = 0; */ 1727 /* pPager->pLast = 0; */ 1728 pPager->nExtra = FORCE_ALIGNMENT(nExtra); 1729 pPager->sectorSize = PAGER_SECTOR_SIZE; 1730 /* pPager->pBusyHandler = 0; */ 1731 /* memset(pPager->aHash, 0, sizeof(pPager->aHash)); */ 1732 *ppPager = pPager; 1733 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 1734 pPager->pNext = pTsd->pPager; 1735 pTsd->pPager = pPager; 1736 #endif 1737 return SQLITE_OK; 1738 } 1739 1740 /* 1741 ** Set the busy handler function. 1742 */ 1743 void sqlite3pager_set_busyhandler(Pager *pPager, BusyHandler *pBusyHandler){ 1744 pPager->pBusyHandler = pBusyHandler; 1745 } 1746 1747 /* 1748 ** Set the destructor for this pager. If not NULL, the destructor is called 1749 ** when the reference count on each page reaches zero. The destructor can 1750 ** be used to clean up information in the extra segment appended to each page. 1751 ** 1752 ** The destructor is not called as a result sqlite3pager_close(). 1753 ** Destructors are only called by sqlite3pager_unref(). 1754 */ 1755 void sqlite3pager_set_destructor(Pager *pPager, void (*xDesc)(void*,int)){ 1756 pPager->xDestructor = xDesc; 1757 } 1758 1759 /* 1760 ** Set the reinitializer for this pager. If not NULL, the reinitializer 1761 ** is called when the content of a page in cache is restored to its original 1762 ** value as a result of a rollback. The callback gives higher-level code 1763 ** an opportunity to restore the EXTRA section to agree with the restored 1764 ** page data. 1765 */ 1766 void sqlite3pager_set_reiniter(Pager *pPager, void (*xReinit)(void*,int)){ 1767 pPager->xReiniter = xReinit; 1768 } 1769 1770 /* 1771 ** Set the page size. Return the new size. If the suggest new page 1772 ** size is inappropriate, then an alternative page size is selected 1773 ** and returned. 1774 */ 1775 int sqlite3pager_set_pagesize(Pager *pPager, int pageSize){ 1776 assert( pageSize>=512 && pageSize<=SQLITE_MAX_PAGE_SIZE ); 1777 if( !pPager->memDb ){ 1778 pPager->pageSize = pageSize; 1779 } 1780 return pPager->pageSize; 1781 } 1782 1783 /* 1784 ** The following set of routines are used to disable the simulated 1785 ** I/O error mechanism. These routines are used to avoid simulated 1786 ** errors in places where we do not care about errors. 1787 ** 1788 ** Unless -DSQLITE_TEST=1 is used, these routines are all no-ops 1789 ** and generate no code. 1790 */ 1791 #ifdef SQLITE_TEST 1792 extern int sqlite3_io_error_pending; 1793 extern int sqlite3_io_error_hit; 1794 static int saved_cnt; 1795 void clear_simulated_io_error(){ 1796 sqlite3_io_error_hit = 0; 1797 } 1798 void disable_simulated_io_errors(void){ 1799 saved_cnt = sqlite3_io_error_pending; 1800 sqlite3_io_error_pending = -1; 1801 } 1802 void enable_simulated_io_errors(void){ 1803 sqlite3_io_error_pending = saved_cnt; 1804 } 1805 #else 1806 # define clear_simulated_io_error() 1807 # define disable_simulated_io_errors() 1808 # define enable_simulated_io_errors() 1809 #endif 1810 1811 /* 1812 ** Read the first N bytes from the beginning of the file into memory 1813 ** that pDest points to. 1814 ** 1815 ** No error checking is done. The rational for this is that this function 1816 ** may be called even if the file does not exist or contain a header. In 1817 ** these cases sqlite3OsRead() will return an error, to which the correct 1818 ** response is to zero the memory at pDest and continue. A real IO error 1819 ** will presumably recur and be picked up later (Todo: Think about this). 1820 */ 1821 int sqlite3pager_read_fileheader(Pager *pPager, int N, unsigned char *pDest){ 1822 int rc = SQLITE_OK; 1823 memset(pDest, 0, N); 1824 if( MEMDB==0 ){ 1825 disable_simulated_io_errors(); 1826 sqlite3OsSeek(pPager->fd, 0); 1827 enable_simulated_io_errors(); 1828 rc = sqlite3OsRead(pPager->fd, pDest, N); 1829 if( rc==SQLITE_IOERR_SHORT_READ ){ 1830 rc = SQLITE_OK; 1831 } 1832 } 1833 return rc; 1834 } 1835 1836 /* 1837 ** Return the total number of pages in the disk file associated with 1838 ** pPager. 1839 ** 1840 ** If the PENDING_BYTE lies on the page directly after the end of the 1841 ** file, then consider this page part of the file too. For example, if 1842 ** PENDING_BYTE is byte 4096 (the first byte of page 5) and the size of the 1843 ** file is 4096 bytes, 5 is returned instead of 4. 1844 */ 1845 int sqlite3pager_pagecount(Pager *pPager){ 1846 i64 n; 1847 int rc; 1848 assert( pPager!=0 ); 1849 if( pPager->dbSize>=0 ){ 1850 n = pPager->dbSize; 1851 } else { 1852 if( (rc = sqlite3OsFileSize(pPager->fd, &n))!=SQLITE_OK ){ 1853 pager_error(pPager, rc); 1854 return 0; 1855 } 1856 if( n>0 && n<pPager->pageSize ){ 1857 n = 1; 1858 }else{ 1859 n /= pPager->pageSize; 1860 } 1861 if( pPager->state!=PAGER_UNLOCK ){ 1862 pPager->dbSize = n; 1863 } 1864 } 1865 if( n==(PENDING_BYTE/pPager->pageSize) ){ 1866 n++; 1867 } 1868 return n; 1869 } 1870 1871 1872 #ifndef SQLITE_OMIT_MEMORYDB 1873 /* 1874 ** Clear a PgHistory block 1875 */ 1876 static void clearHistory(PgHistory *pHist){ 1877 sqliteFree(pHist->pOrig); 1878 sqliteFree(pHist->pStmt); 1879 pHist->pOrig = 0; 1880 pHist->pStmt = 0; 1881 } 1882 #else 1883 #define clearHistory(x) 1884 #endif 1885 1886 /* 1887 ** Forward declaration 1888 */ 1889 static int syncJournal(Pager*); 1890 1891 /* 1892 ** Unlink pPg from it's hash chain. Also set the page number to 0 to indicate 1893 ** that the page is not part of any hash chain. This is required because the 1894 ** sqlite3pager_movepage() routine can leave a page in the 1895 ** pNextFree/pPrevFree list that is not a part of any hash-chain. 1896 */ 1897 static void unlinkHashChain(Pager *pPager, PgHdr *pPg){ 1898 if( pPg->pgno==0 ){ 1899 assert( pPg->pNextHash==0 && pPg->pPrevHash==0 ); 1900 return; 1901 } 1902 if( pPg->pNextHash ){ 1903 pPg->pNextHash->pPrevHash = pPg->pPrevHash; 1904 } 1905 if( pPg->pPrevHash ){ 1906 assert( pPager->aHash[pPg->pgno & (pPager->nHash-1)]!=pPg ); 1907 pPg->pPrevHash->pNextHash = pPg->pNextHash; 1908 }else{ 1909 int h = pPg->pgno & (pPager->nHash-1); 1910 pPager->aHash[h] = pPg->pNextHash; 1911 } 1912 if( MEMDB ){ 1913 clearHistory(PGHDR_TO_HIST(pPg, pPager)); 1914 } 1915 pPg->pgno = 0; 1916 pPg->pNextHash = pPg->pPrevHash = 0; 1917 } 1918 1919 /* 1920 ** Unlink a page from the free list (the list of all pages where nRef==0) 1921 ** and from its hash collision chain. 1922 */ 1923 static void unlinkPage(PgHdr *pPg){ 1924 Pager *pPager = pPg->pPager; 1925 1926 /* Keep the pFirstSynced pointer pointing at the first synchronized page */ 1927 if( pPg==pPager->pFirstSynced ){ 1928 PgHdr *p = pPg->pNextFree; 1929 while( p && p->needSync ){ p = p->pNextFree; } 1930 pPager->pFirstSynced = p; 1931 } 1932 1933 /* Unlink from the freelist */ 1934 if( pPg->pPrevFree ){ 1935 pPg->pPrevFree->pNextFree = pPg->pNextFree; 1936 }else{ 1937 assert( pPager->pFirst==pPg ); 1938 pPager->pFirst = pPg->pNextFree; 1939 } 1940 if( pPg->pNextFree ){ 1941 pPg->pNextFree->pPrevFree = pPg->pPrevFree; 1942 }else{ 1943 assert( pPager->pLast==pPg ); 1944 pPager->pLast = pPg->pPrevFree; 1945 } 1946 pPg->pNextFree = pPg->pPrevFree = 0; 1947 1948 /* Unlink from the pgno hash table */ 1949 unlinkHashChain(pPager, pPg); 1950 } 1951 1952 #ifndef SQLITE_OMIT_MEMORYDB 1953 /* 1954 ** This routine is used to truncate an in-memory database. Delete 1955 ** all pages whose pgno is larger than pPager->dbSize and is unreferenced. 1956 ** Referenced pages larger than pPager->dbSize are zeroed. 1957 */ 1958 static void memoryTruncate(Pager *pPager){ 1959 PgHdr *pPg; 1960 PgHdr **ppPg; 1961 int dbSize = pPager->dbSize; 1962 1963 ppPg = &pPager->pAll; 1964 while( (pPg = *ppPg)!=0 ){ 1965 if( pPg->pgno<=dbSize ){ 1966 ppPg = &pPg->pNextAll; 1967 }else if( pPg->nRef>0 ){ 1968 memset(PGHDR_TO_DATA(pPg), 0, pPager->pageSize); 1969 ppPg = &pPg->pNextAll; 1970 }else{ 1971 *ppPg = pPg->pNextAll; 1972 unlinkPage(pPg); 1973 makeClean(pPg); 1974 sqliteFree(pPg); 1975 pPager->nPage--; 1976 } 1977 } 1978 } 1979 #else 1980 #define memoryTruncate(p) 1981 #endif 1982 1983 /* 1984 ** Try to obtain a lock on a file. Invoke the busy callback if the lock 1985 ** is currently not available. Repeat until the busy callback returns 1986 ** false or until the lock succeeds. 1987 ** 1988 ** Return SQLITE_OK on success and an error code if we cannot obtain 1989 ** the lock. 1990 */ 1991 static int pager_wait_on_lock(Pager *pPager, int locktype){ 1992 int rc; 1993 1994 /* The OS lock values must be the same as the Pager lock values */ 1995 assert( PAGER_SHARED==SHARED_LOCK ); 1996 assert( PAGER_RESERVED==RESERVED_LOCK ); 1997 assert( PAGER_EXCLUSIVE==EXCLUSIVE_LOCK ); 1998 1999 /* If the file is currently unlocked then the size must be unknown */ 2000 assert( pPager->state>=PAGER_SHARED || pPager->dbSize<0 || MEMDB ); 2001 2002 if( pPager->state>=locktype ){ 2003 rc = SQLITE_OK; 2004 }else{ 2005 do { 2006 rc = sqlite3OsLock(pPager->fd, locktype); 2007 }while( rc==SQLITE_BUSY && sqlite3InvokeBusyHandler(pPager->pBusyHandler) ); 2008 if( rc==SQLITE_OK ){ 2009 pPager->state = locktype; 2010 } 2011 } 2012 return rc; 2013 } 2014 2015 /* 2016 ** Truncate the file to the number of pages specified. 2017 */ 2018 int sqlite3pager_truncate(Pager *pPager, Pgno nPage){ 2019 int rc; 2020 assert( pPager->state>=PAGER_SHARED || MEMDB ); 2021 sqlite3pager_pagecount(pPager); 2022 if( pPager->errCode ){ 2023 rc = pPager->errCode; 2024 return rc; 2025 } 2026 if( nPage>=(unsigned)pPager->dbSize ){ 2027 return SQLITE_OK; 2028 } 2029 if( MEMDB ){ 2030 pPager->dbSize = nPage; 2031 memoryTruncate(pPager); 2032 return SQLITE_OK; 2033 } 2034 rc = syncJournal(pPager); 2035 if( rc!=SQLITE_OK ){ 2036 return rc; 2037 } 2038 2039 /* Get an exclusive lock on the database before truncating. */ 2040 rc = pager_wait_on_lock(pPager, EXCLUSIVE_LOCK); 2041 if( rc!=SQLITE_OK ){ 2042 return rc; 2043 } 2044 2045 rc = pager_truncate(pPager, nPage); 2046 if( rc==SQLITE_OK ){ 2047 pPager->dbSize = nPage; 2048 } 2049 return rc; 2050 } 2051 2052 /* 2053 ** Shutdown the page cache. Free all memory and close all files. 2054 ** 2055 ** If a transaction was in progress when this routine is called, that 2056 ** transaction is rolled back. All outstanding pages are invalidated 2057 ** and their memory is freed. Any attempt to use a page associated 2058 ** with this page cache after this function returns will likely 2059 ** result in a coredump. 2060 ** 2061 ** This function always succeeds. If a transaction is active an attempt 2062 ** is made to roll it back. If an error occurs during the rollback 2063 ** a hot journal may be left in the filesystem but no error is returned 2064 ** to the caller. 2065 */ 2066 int sqlite3pager_close(Pager *pPager){ 2067 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 2068 /* A malloc() cannot fail in sqlite3ThreadData() as one or more calls to 2069 ** malloc() must have already been made by this thread before it gets 2070 ** to this point. This means the ThreadData must have been allocated already 2071 ** so that ThreadData.nAlloc can be set. 2072 */ 2073 ThreadData *pTsd = sqlite3ThreadData(); 2074 assert( pPager ); 2075 assert( pTsd && pTsd->nAlloc ); 2076 #endif 2077 2078 disable_simulated_io_errors(); 2079 pPager->errCode = 0; 2080 pager_reset(pPager); 2081 enable_simulated_io_errors(); 2082 TRACE2("CLOSE %d\n", PAGERID(pPager)); 2083 assert( pPager->errCode || (pPager->journalOpen==0 && pPager->stmtOpen==0) ); 2084 if( pPager->journalOpen ){ 2085 sqlite3OsClose(&pPager->jfd); 2086 } 2087 sqliteFree(pPager->aInJournal); 2088 if( pPager->stmtOpen ){ 2089 sqlite3OsClose(&pPager->stfd); 2090 } 2091 sqlite3OsClose(&pPager->fd); 2092 /* Temp files are automatically deleted by the OS 2093 ** if( pPager->tempFile ){ 2094 ** sqlite3OsDelete(pPager->zFilename); 2095 ** } 2096 */ 2097 2098 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 2099 /* Remove the pager from the linked list of pagers starting at 2100 ** ThreadData.pPager if memory-management is enabled. 2101 */ 2102 if( pPager==pTsd->pPager ){ 2103 pTsd->pPager = pPager->pNext; 2104 }else{ 2105 Pager *pTmp; 2106 for(pTmp = pTsd->pPager; pTmp->pNext!=pPager; pTmp=pTmp->pNext){} 2107 pTmp->pNext = pPager->pNext; 2108 } 2109 #endif 2110 sqliteFree(pPager->aHash); 2111 sqliteFree(pPager); 2112 return SQLITE_OK; 2113 } 2114 2115 /* 2116 ** Return the page number for the given page data. 2117 */ 2118 Pgno sqlite3pager_pagenumber(void *pData){ 2119 PgHdr *p = DATA_TO_PGHDR(pData); 2120 return p->pgno; 2121 } 2122 2123 /* 2124 ** The page_ref() function increments the reference count for a page. 2125 ** If the page is currently on the freelist (the reference count is zero) then 2126 ** remove it from the freelist. 2127 ** 2128 ** For non-test systems, page_ref() is a macro that calls _page_ref() 2129 ** online of the reference count is zero. For test systems, page_ref() 2130 ** is a real function so that we can set breakpoints and trace it. 2131 */ 2132 static void _page_ref(PgHdr *pPg){ 2133 if( pPg->nRef==0 ){ 2134 /* The page is currently on the freelist. Remove it. */ 2135 if( pPg==pPg->pPager->pFirstSynced ){ 2136 PgHdr *p = pPg->pNextFree; 2137 while( p && p->needSync ){ p = p->pNextFree; } 2138 pPg->pPager->pFirstSynced = p; 2139 } 2140 if( pPg->pPrevFree ){ 2141 pPg->pPrevFree->pNextFree = pPg->pNextFree; 2142 }else{ 2143 pPg->pPager->pFirst = pPg->pNextFree; 2144 } 2145 if( pPg->pNextFree ){ 2146 pPg->pNextFree->pPrevFree = pPg->pPrevFree; 2147 }else{ 2148 pPg->pPager->pLast = pPg->pPrevFree; 2149 } 2150 pPg->pPager->nRef++; 2151 } 2152 pPg->nRef++; 2153 REFINFO(pPg); 2154 } 2155 #ifdef SQLITE_DEBUG 2156 static void page_ref(PgHdr *pPg){ 2157 if( pPg->nRef==0 ){ 2158 _page_ref(pPg); 2159 }else{ 2160 pPg->nRef++; 2161 REFINFO(pPg); 2162 } 2163 } 2164 #else 2165 # define page_ref(P) ((P)->nRef==0?_page_ref(P):(void)(P)->nRef++) 2166 #endif 2167 2168 /* 2169 ** Increment the reference count for a page. The input pointer is 2170 ** a reference to the page data. 2171 */ 2172 int sqlite3pager_ref(void *pData){ 2173 PgHdr *pPg = DATA_TO_PGHDR(pData); 2174 page_ref(pPg); 2175 return SQLITE_OK; 2176 } 2177 2178 /* 2179 ** Sync the journal. In other words, make sure all the pages that have 2180 ** been written to the journal have actually reached the surface of the 2181 ** disk. It is not safe to modify the original database file until after 2182 ** the journal has been synced. If the original database is modified before 2183 ** the journal is synced and a power failure occurs, the unsynced journal 2184 ** data would be lost and we would be unable to completely rollback the 2185 ** database changes. Database corruption would occur. 2186 ** 2187 ** This routine also updates the nRec field in the header of the journal. 2188 ** (See comments on the pager_playback() routine for additional information.) 2189 ** If the sync mode is FULL, two syncs will occur. First the whole journal 2190 ** is synced, then the nRec field is updated, then a second sync occurs. 2191 ** 2192 ** For temporary databases, we do not care if we are able to rollback 2193 ** after a power failure, so sync occurs. 2194 ** 2195 ** This routine clears the needSync field of every page current held in 2196 ** memory. 2197 */ 2198 static int syncJournal(Pager *pPager){ 2199 PgHdr *pPg; 2200 int rc = SQLITE_OK; 2201 2202 /* Sync the journal before modifying the main database 2203 ** (assuming there is a journal and it needs to be synced.) 2204 */ 2205 if( pPager->needSync ){ 2206 if( !pPager->tempFile ){ 2207 assert( pPager->journalOpen ); 2208 /* assert( !pPager->noSync ); // noSync might be set if synchronous 2209 ** was turned off after the transaction was started. Ticket #615 */ 2210 #ifndef NDEBUG 2211 { 2212 /* Make sure the pPager->nRec counter we are keeping agrees 2213 ** with the nRec computed from the size of the journal file. 2214 */ 2215 i64 jSz; 2216 rc = sqlite3OsFileSize(pPager->jfd, &jSz); 2217 if( rc!=0 ) return rc; 2218 assert( pPager->journalOff==jSz ); 2219 } 2220 #endif 2221 { 2222 /* Write the nRec value into the journal file header. If in 2223 ** full-synchronous mode, sync the journal first. This ensures that 2224 ** all data has really hit the disk before nRec is updated to mark 2225 ** it as a candidate for rollback. 2226 */ 2227 if( pPager->fullSync ){ 2228 TRACE2("SYNC journal of %d\n", PAGERID(pPager)); 2229 rc = sqlite3OsSync(pPager->jfd, 0); 2230 if( rc!=0 ) return rc; 2231 } 2232 rc = sqlite3OsSeek(pPager->jfd, 2233 pPager->journalHdr + sizeof(aJournalMagic)); 2234 if( rc ) return rc; 2235 rc = write32bits(pPager->jfd, pPager->nRec); 2236 if( rc ) return rc; 2237 2238 rc = sqlite3OsSeek(pPager->jfd, pPager->journalOff); 2239 if( rc ) return rc; 2240 } 2241 TRACE2("SYNC journal of %d\n", PAGERID(pPager)); 2242 rc = sqlite3OsSync(pPager->jfd, pPager->full_fsync); 2243 if( rc!=0 ) return rc; 2244 pPager->journalStarted = 1; 2245 } 2246 pPager->needSync = 0; 2247 2248 /* Erase the needSync flag from every page. 2249 */ 2250 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 2251 pPg->needSync = 0; 2252 } 2253 pPager->pFirstSynced = pPager->pFirst; 2254 } 2255 2256 #ifndef NDEBUG 2257 /* If the Pager.needSync flag is clear then the PgHdr.needSync 2258 ** flag must also be clear for all pages. Verify that this 2259 ** invariant is true. 2260 */ 2261 else{ 2262 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 2263 assert( pPg->needSync==0 ); 2264 } 2265 assert( pPager->pFirstSynced==pPager->pFirst ); 2266 } 2267 #endif 2268 2269 return rc; 2270 } 2271 2272 /* 2273 ** Merge two lists of pages connected by pDirty and in pgno order. 2274 ** Do not both fixing the pPrevDirty pointers. 2275 */ 2276 static PgHdr *merge_pagelist(PgHdr *pA, PgHdr *pB){ 2277 PgHdr result, *pTail; 2278 pTail = &result; 2279 while( pA && pB ){ 2280 if( pA->pgno<pB->pgno ){ 2281 pTail->pDirty = pA; 2282 pTail = pA; 2283 pA = pA->pDirty; 2284 }else{ 2285 pTail->pDirty = pB; 2286 pTail = pB; 2287 pB = pB->pDirty; 2288 } 2289 } 2290 if( pA ){ 2291 pTail->pDirty = pA; 2292 }else if( pB ){ 2293 pTail->pDirty = pB; 2294 }else{ 2295 pTail->pDirty = 0; 2296 } 2297 return result.pDirty; 2298 } 2299 2300 /* 2301 ** Sort the list of pages in accending order by pgno. Pages are 2302 ** connected by pDirty pointers. The pPrevDirty pointers are 2303 ** corrupted by this sort. 2304 */ 2305 #define N_SORT_BUCKET 25 2306 static PgHdr *sort_pagelist(PgHdr *pIn){ 2307 PgHdr *a[N_SORT_BUCKET], *p; 2308 int i; 2309 memset(a, 0, sizeof(a)); 2310 while( pIn ){ 2311 p = pIn; 2312 pIn = p->pDirty; 2313 p->pDirty = 0; 2314 for(i=0; i<N_SORT_BUCKET-1; i++){ 2315 if( a[i]==0 ){ 2316 a[i] = p; 2317 break; 2318 }else{ 2319 p = merge_pagelist(a[i], p); 2320 a[i] = 0; 2321 } 2322 } 2323 if( i==N_SORT_BUCKET-1 ){ 2324 a[i] = merge_pagelist(a[i], p); 2325 } 2326 } 2327 p = a[0]; 2328 for(i=1; i<N_SORT_BUCKET; i++){ 2329 p = merge_pagelist(p, a[i]); 2330 } 2331 return p; 2332 } 2333 2334 /* 2335 ** Given a list of pages (connected by the PgHdr.pDirty pointer) write 2336 ** every one of those pages out to the database file and mark them all 2337 ** as clean. 2338 */ 2339 static int pager_write_pagelist(PgHdr *pList){ 2340 Pager *pPager; 2341 int rc; 2342 2343 if( pList==0 ) return SQLITE_OK; 2344 pPager = pList->pPager; 2345 2346 /* At this point there may be either a RESERVED or EXCLUSIVE lock on the 2347 ** database file. If there is already an EXCLUSIVE lock, the following 2348 ** calls to sqlite3OsLock() are no-ops. 2349 ** 2350 ** Moving the lock from RESERVED to EXCLUSIVE actually involves going 2351 ** through an intermediate state PENDING. A PENDING lock prevents new 2352 ** readers from attaching to the database but is unsufficient for us to 2353 ** write. The idea of a PENDING lock is to prevent new readers from 2354 ** coming in while we wait for existing readers to clear. 2355 ** 2356 ** While the pager is in the RESERVED state, the original database file 2357 ** is unchanged and we can rollback without having to playback the 2358 ** journal into the original database file. Once we transition to 2359 ** EXCLUSIVE, it means the database file has been changed and any rollback 2360 ** will require a journal playback. 2361 */ 2362 rc = pager_wait_on_lock(pPager, EXCLUSIVE_LOCK); 2363 if( rc!=SQLITE_OK ){ 2364 return rc; 2365 } 2366 2367 pList = sort_pagelist(pList); 2368 while( pList ){ 2369 assert( pList->dirty ); 2370 rc = sqlite3OsSeek(pPager->fd, (pList->pgno-1)*(i64)pPager->pageSize); 2371 if( rc ) return rc; 2372 /* If there are dirty pages in the page cache with page numbers greater 2373 ** than Pager.dbSize, this means sqlite3pager_truncate() was called to 2374 ** make the file smaller (presumably by auto-vacuum code). Do not write 2375 ** any such pages to the file. 2376 */ 2377 if( pList->pgno<=pPager->dbSize ){ 2378 char *pData = CODEC2(pPager, PGHDR_TO_DATA(pList), pList->pgno, 6); 2379 TRACE3("STORE %d page %d\n", PAGERID(pPager), pList->pgno); 2380 rc = sqlite3OsWrite(pPager->fd, pData, pPager->pageSize); 2381 TEST_INCR(pPager->nWrite); 2382 } 2383 #ifndef NDEBUG 2384 else{ 2385 TRACE3("NOSTORE %d page %d\n", PAGERID(pPager), pList->pgno); 2386 } 2387 #endif 2388 if( rc ) return rc; 2389 pList->dirty = 0; 2390 #ifdef SQLITE_CHECK_PAGES 2391 pList->pageHash = pager_pagehash(pList); 2392 #endif 2393 pList = pList->pDirty; 2394 } 2395 return SQLITE_OK; 2396 } 2397 2398 /* 2399 ** Collect every dirty page into a dirty list and 2400 ** return a pointer to the head of that list. All pages are 2401 ** collected even if they are still in use. 2402 */ 2403 static PgHdr *pager_get_all_dirty_pages(Pager *pPager){ 2404 return pPager->pDirty; 2405 } 2406 2407 /* 2408 ** Return TRUE if there is a hot journal on the given pager. 2409 ** A hot journal is one that needs to be played back. 2410 ** 2411 ** If the current size of the database file is 0 but a journal file 2412 ** exists, that is probably an old journal left over from a prior 2413 ** database with the same name. Just delete the journal. 2414 */ 2415 static int hasHotJournal(Pager *pPager){ 2416 if( !pPager->useJournal ) return 0; 2417 if( !sqlite3OsFileExists(pPager->zJournal) ) return 0; 2418 if( sqlite3OsCheckReservedLock(pPager->fd) ) return 0; 2419 if( sqlite3pager_pagecount(pPager)==0 ){ 2420 sqlite3OsDelete(pPager->zJournal); 2421 return 0; 2422 }else{ 2423 return 1; 2424 } 2425 } 2426 2427 /* 2428 ** Try to find a page in the cache that can be recycled. 2429 ** 2430 ** This routine may return SQLITE_IOERR, SQLITE_FULL or SQLITE_OK. It 2431 ** does not set the pPager->errCode variable. 2432 */ 2433 static int pager_recycle(Pager *pPager, int syncOk, PgHdr **ppPg){ 2434 PgHdr *pPg; 2435 *ppPg = 0; 2436 2437 /* Find a page to recycle. Try to locate a page that does not 2438 ** require us to do an fsync() on the journal. 2439 */ 2440 pPg = pPager->pFirstSynced; 2441 2442 /* If we could not find a page that does not require an fsync() 2443 ** on the journal file then fsync the journal file. This is a 2444 ** very slow operation, so we work hard to avoid it. But sometimes 2445 ** it can't be helped. 2446 */ 2447 if( pPg==0 && pPager->pFirst && syncOk && !MEMDB){ 2448 int rc = syncJournal(pPager); 2449 if( rc!=0 ){ 2450 return rc; 2451 } 2452 if( pPager->fullSync ){ 2453 /* If in full-sync mode, write a new journal header into the 2454 ** journal file. This is done to avoid ever modifying a journal 2455 ** header that is involved in the rollback of pages that have 2456 ** already been written to the database (in case the header is 2457 ** trashed when the nRec field is updated). 2458 */ 2459 pPager->nRec = 0; 2460 assert( pPager->journalOff > 0 ); 2461 rc = writeJournalHdr(pPager); 2462 if( rc!=0 ){ 2463 return rc; 2464 } 2465 } 2466 pPg = pPager->pFirst; 2467 } 2468 if( pPg==0 ){ 2469 return SQLITE_OK; 2470 } 2471 2472 assert( pPg->nRef==0 ); 2473 2474 /* Write the page to the database file if it is dirty. 2475 */ 2476 if( pPg->dirty ){ 2477 int rc; 2478 assert( pPg->needSync==0 ); 2479 makeClean(pPg); 2480 pPg->dirty = 1; 2481 pPg->pDirty = 0; 2482 rc = pager_write_pagelist( pPg ); 2483 if( rc!=SQLITE_OK ){ 2484 return rc; 2485 } 2486 } 2487 assert( pPg->dirty==0 ); 2488 2489 /* If the page we are recycling is marked as alwaysRollback, then 2490 ** set the global alwaysRollback flag, thus disabling the 2491 ** sqlite_dont_rollback() optimization for the rest of this transaction. 2492 ** It is necessary to do this because the page marked alwaysRollback 2493 ** might be reloaded at a later time but at that point we won't remember 2494 ** that is was marked alwaysRollback. This means that all pages must 2495 ** be marked as alwaysRollback from here on out. 2496 */ 2497 if( pPg->alwaysRollback ){ 2498 pPager->alwaysRollback = 1; 2499 } 2500 2501 /* Unlink the old page from the free list and the hash table 2502 */ 2503 unlinkPage(pPg); 2504 TEST_INCR(pPager->nOvfl); 2505 2506 *ppPg = pPg; 2507 return SQLITE_OK; 2508 } 2509 2510 /* 2511 ** This function is called to free superfluous dynamically allocated memory 2512 ** held by the pager system. Memory in use by any SQLite pager allocated 2513 ** by the current thread may be sqliteFree()ed. 2514 ** 2515 ** nReq is the number of bytes of memory required. Once this much has 2516 ** been released, the function returns. A negative value for nReq means 2517 ** free as much memory as possible. The return value is the total number 2518 ** of bytes of memory released. 2519 */ 2520 #ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT 2521 int sqlite3pager_release_memory(int nReq){ 2522 const ThreadData *pTsdro = sqlite3ThreadDataReadOnly(); 2523 Pager *p; 2524 int nReleased = 0; 2525 int i; 2526 2527 /* If the the global mutex is held, this subroutine becomes a 2528 ** o-op; zero bytes of memory are freed. This is because 2529 ** some of the code invoked by this function may also 2530 ** try to obtain the mutex, resulting in a deadlock. 2531 */ 2532 if( sqlite3OsInMutex(0) ){ 2533 return 0; 2534 } 2535 2536 /* Outermost loop runs for at most two iterations. First iteration we 2537 ** try to find memory that can be released without calling fsync(). Second 2538 ** iteration (which only runs if the first failed to free nReq bytes of 2539 ** memory) is permitted to call fsync(). This is of course much more 2540 ** expensive. 2541 */ 2542 for(i=0; i<=1; i++){ 2543 2544 /* Loop through all the SQLite pagers opened by the current thread. */ 2545 for(p=pTsdro->pPager; p && (nReq<0 || nReleased<nReq); p=p->pNext){ 2546 PgHdr *pPg; 2547 int rc; 2548 2549 /* For each pager, try to free as many pages as possible (without 2550 ** calling fsync() if this is the first iteration of the outermost 2551 ** loop). 2552 */ 2553 while( SQLITE_OK==(rc = pager_recycle(p, i, &pPg)) && pPg) { 2554 /* We've found a page to free. At this point the page has been 2555 ** removed from the page hash-table, free-list and synced-list 2556 ** (pFirstSynced). It is still in the all pages (pAll) list. 2557 ** Remove it from this list before freeing. 2558 ** 2559 ** Todo: Check the Pager.pStmt list to make sure this is Ok. It 2560 ** probably is though. 2561 */ 2562 PgHdr *pTmp; 2563 assert( pPg ); 2564 page_remove_from_stmt_list(pPg); 2565 if( pPg==p->pAll ){ 2566 p->pAll = pPg->pNextAll; 2567 }else{ 2568 for( pTmp=p->pAll; pTmp->pNextAll!=pPg; pTmp=pTmp->pNextAll ){} 2569 pTmp->pNextAll = pPg->pNextAll; 2570 } 2571 nReleased += sqliteAllocSize(pPg); 2572 sqliteFree(pPg); 2573 } 2574 2575 if( rc!=SQLITE_OK ){ 2576 /* An error occured whilst writing to the database file or 2577 ** journal in pager_recycle(). The error is not returned to the 2578 ** caller of this function. Instead, set the Pager.errCode variable. 2579 ** The error will be returned to the user (or users, in the case 2580 ** of a shared pager cache) of the pager for which the error occured. 2581 */ 2582 assert( (rc&0xff)==SQLITE_IOERR || rc==SQLITE_FULL ); 2583 assert( p->state>=PAGER_RESERVED ); 2584 pager_error(p, rc); 2585 } 2586 } 2587 } 2588 2589 return nReleased; 2590 } 2591 #endif /* SQLITE_ENABLE_MEMORY_MANAGEMENT */ 2592 2593 /* 2594 ** Acquire a page. 2595 ** 2596 ** A read lock on the disk file is obtained when the first page is acquired. 2597 ** This read lock is dropped when the last page is released. 2598 ** 2599 ** A _get works for any page number greater than 0. If the database 2600 ** file is smaller than the requested page, then no actual disk 2601 ** read occurs and the memory image of the page is initialized to 2602 ** all zeros. The extra data appended to a page is always initialized 2603 ** to zeros the first time a page is loaded into memory. 2604 ** 2605 ** The acquisition might fail for several reasons. In all cases, 2606 ** an appropriate error code is returned and *ppPage is set to NULL. 2607 ** 2608 ** See also sqlite3pager_lookup(). Both this routine and _lookup() attempt 2609 ** to find a page in the in-memory cache first. If the page is not already 2610 ** in memory, this routine goes to disk to read it in whereas _lookup() 2611 ** just returns 0. This routine acquires a read-lock the first time it 2612 ** has to go to disk, and could also playback an old journal if necessary. 2613 ** Since _lookup() never goes to disk, it never has to deal with locks 2614 ** or journal files. 2615 */ 2616 int sqlite3pager_get(Pager *pPager, Pgno pgno, void **ppPage){ 2617 PgHdr *pPg; 2618 int rc; 2619 2620 /* The maximum page number is 2^31. Return SQLITE_CORRUPT if a page 2621 ** number greater than this, or zero, is requested. 2622 */ 2623 if( pgno>PAGER_MAX_PGNO || pgno==0 || pgno==PAGER_MJ_PGNO(pPager) ){ 2624 return SQLITE_CORRUPT_BKPT; 2625 } 2626 2627 /* Make sure we have not hit any critical errors. 2628 */ 2629 assert( pPager!=0 ); 2630 *ppPage = 0; 2631 if( pPager->errCode && pPager->errCode!=SQLITE_FULL ){ 2632 return pPager->errCode; 2633 } 2634 2635 /* If this is the first page accessed, then get a SHARED lock 2636 ** on the database file. 2637 */ 2638 if( pPager->nRef==0 && !MEMDB ){ 2639 if( !pPager->noReadlock ){ 2640 rc = pager_wait_on_lock(pPager, SHARED_LOCK); 2641 if( rc!=SQLITE_OK ){ 2642 return pager_error(pPager, rc); 2643 } 2644 } 2645 2646 /* If a journal file exists, and there is no RESERVED lock on the 2647 ** database file, then it either needs to be played back or deleted. 2648 */ 2649 if( hasHotJournal(pPager) ){ 2650 /* Get an EXCLUSIVE lock on the database file. At this point it is 2651 ** important that a RESERVED lock is not obtained on the way to the 2652 ** EXCLUSIVE lock. If it were, another process might open the 2653 ** database file, detect the RESERVED lock, and conclude that the 2654 ** database is safe to read while this process is still rolling it 2655 ** back. 2656 ** 2657 ** Because the intermediate RESERVED lock is not requested, the 2658 ** second process will get to this point in the code and fail to 2659 ** obtain it's own EXCLUSIVE lock on the database file. 2660 */ 2661 rc = sqlite3OsLock(pPager->fd, EXCLUSIVE_LOCK); 2662 if( rc!=SQLITE_OK ){ 2663 pager_unlock(pPager); 2664 return pager_error(pPager, rc); 2665 } 2666 pPager->state = PAGER_EXCLUSIVE; 2667 2668 /* Open the journal for reading only. Return SQLITE_BUSY if 2669 ** we are unable to open the journal file. 2670 ** 2671 ** The journal file does not need to be locked itself. The 2672 ** journal file is never open unless the main database file holds 2673 ** a write lock, so there is never any chance of two or more 2674 ** processes opening the journal at the same time. 2675 */ 2676 rc = sqlite3OsOpenReadOnly(pPager->zJournal, &pPager->jfd); 2677 if( rc!=SQLITE_OK ){ 2678 pager_unlock(pPager); 2679 return SQLITE_BUSY; 2680 } 2681 pPager->journalOpen = 1; 2682 pPager->journalStarted = 0; 2683 pPager->journalOff = 0; 2684 pPager->setMaster = 0; 2685 pPager->journalHdr = 0; 2686 2687 /* Playback and delete the journal. Drop the database write 2688 ** lock and reacquire the read lock. 2689 */ 2690 rc = pager_playback(pPager); 2691 if( rc!=SQLITE_OK ){ 2692 return pager_error(pPager, rc); 2693 } 2694 } 2695 pPg = 0; 2696 }else{ 2697 /* Search for page in cache */ 2698 pPg = pager_lookup(pPager, pgno); 2699 if( MEMDB && pPager->state==PAGER_UNLOCK ){ 2700 pPager->state = PAGER_SHARED; 2701 } 2702 } 2703 if( pPg==0 ){ 2704 /* The requested page is not in the page cache. */ 2705 int h; 2706 TEST_INCR(pPager->nMiss); 2707 if( pPager->nPage<pPager->mxPage || pPager->pFirst==0 || MEMDB ){ 2708 /* Create a new page */ 2709 if( pPager->nPage>=pPager->nHash ){ 2710 pager_resize_hash_table(pPager, 2711 pPager->nHash<256 ? 256 : pPager->nHash*2); 2712 if( pPager->nHash==0 ){ 2713 return SQLITE_NOMEM; 2714 } 2715 } 2716 pPg = sqliteMallocRaw( sizeof(*pPg) + pPager->pageSize 2717 + sizeof(u32) + pPager->nExtra 2718 + MEMDB*sizeof(PgHistory) ); 2719 if( pPg==0 ){ 2720 return SQLITE_NOMEM; 2721 } 2722 memset(pPg, 0, sizeof(*pPg)); 2723 if( MEMDB ){ 2724 memset(PGHDR_TO_HIST(pPg, pPager), 0, sizeof(PgHistory)); 2725 } 2726 pPg->pPager = pPager; 2727 pPg->pNextAll = pPager->pAll; 2728 pPager->pAll = pPg; 2729 pPager->nPage++; 2730 if( pPager->nPage>pPager->nMaxPage ){ 2731 assert( pPager->nMaxPage==(pPager->nPage-1) ); 2732 pPager->nMaxPage++; 2733 } 2734 }else{ 2735 rc = pager_recycle(pPager, 1, &pPg); 2736 if( rc!=SQLITE_OK ){ 2737 return rc; 2738 } 2739 assert(pPg) ; 2740 } 2741 pPg->pgno = pgno; 2742 if( pPager->aInJournal && (int)pgno<=pPager->origDbSize ){ 2743 sqlite3CheckMemory(pPager->aInJournal, pgno/8); 2744 assert( pPager->journalOpen ); 2745 pPg->inJournal = (pPager->aInJournal[pgno/8] & (1<<(pgno&7)))!=0; 2746 pPg->needSync = 0; 2747 }else{ 2748 pPg->inJournal = 0; 2749 pPg->needSync = 0; 2750 } 2751 if( pPager->aInStmt && (int)pgno<=pPager->stmtSize 2752 && (pPager->aInStmt[pgno/8] & (1<<(pgno&7)))!=0 ){ 2753 page_add_to_stmt_list(pPg); 2754 }else{ 2755 page_remove_from_stmt_list(pPg); 2756 } 2757 makeClean(pPg); 2758 pPg->nRef = 1; 2759 REFINFO(pPg); 2760 2761 pPager->nRef++; 2762 if( pPager->nExtra>0 ){ 2763 memset(PGHDR_TO_EXTRA(pPg, pPager), 0, pPager->nExtra); 2764 } 2765 if( pPager->errCode ){ 2766 sqlite3pager_unref(PGHDR_TO_DATA(pPg)); 2767 rc = pPager->errCode; 2768 return rc; 2769 } 2770 2771 /* Populate the page with data, either by reading from the database 2772 ** file, or by setting the entire page to zero. 2773 */ 2774 if( sqlite3pager_pagecount(pPager)<(int)pgno || MEMDB ){ 2775 memset(PGHDR_TO_DATA(pPg), 0, pPager->pageSize); 2776 }else{ 2777 assert( MEMDB==0 ); 2778 rc = sqlite3OsSeek(pPager->fd, (pgno-1)*(i64)pPager->pageSize); 2779 if( rc==SQLITE_OK ){ 2780 rc = sqlite3OsRead(pPager->fd, PGHDR_TO_DATA(pPg), 2781 pPager->pageSize); 2782 } 2783 TRACE3("FETCH %d page %d\n", PAGERID(pPager), pPg->pgno); 2784 CODEC1(pPager, PGHDR_TO_DATA(pPg), pPg->pgno, 3); 2785 if( rc!=SQLITE_OK && rc!=SQLITE_IOERR_SHORT_READ ){ 2786 pPg->pgno = 0; 2787 sqlite3pager_unref(PGHDR_TO_DATA(pPg)); 2788 return rc; 2789 }else{ 2790 TEST_INCR(pPager->nRead); 2791 } 2792 } 2793 2794 /* Link the page into the page hash table */ 2795 h = pgno & (pPager->nHash-1); 2796 assert( pgno!=0 ); 2797 pPg->pNextHash = pPager->aHash[h]; 2798 pPager->aHash[h] = pPg; 2799 if( pPg->pNextHash ){ 2800 assert( pPg->pNextHash->pPrevHash==0 ); 2801 pPg->pNextHash->pPrevHash = pPg; 2802 } 2803 2804 #ifdef SQLITE_CHECK_PAGES 2805 pPg->pageHash = pager_pagehash(pPg); 2806 #endif 2807 }else{ 2808 /* The requested page is in the page cache. */ 2809 TEST_INCR(pPager->nHit); 2810 page_ref(pPg); 2811 } 2812 *ppPage = PGHDR_TO_DATA(pPg); 2813 return SQLITE_OK; 2814 } 2815 2816 /* 2817 ** Acquire a page if it is already in the in-memory cache. Do 2818 ** not read the page from disk. Return a pointer to the page, 2819 ** or 0 if the page is not in cache. 2820 ** 2821 ** See also sqlite3pager_get(). The difference between this routine 2822 ** and sqlite3pager_get() is that _get() will go to the disk and read 2823 ** in the page if the page is not already in cache. This routine 2824 ** returns NULL if the page is not in cache or if a disk I/O error 2825 ** has ever happened. 2826 */ 2827 void *sqlite3pager_lookup(Pager *pPager, Pgno pgno){ 2828 PgHdr *pPg; 2829 2830 assert( pPager!=0 ); 2831 assert( pgno!=0 ); 2832 if( pPager->errCode && pPager->errCode!=SQLITE_FULL ){ 2833 return 0; 2834 } 2835 pPg = pager_lookup(pPager, pgno); 2836 if( pPg==0 ) return 0; 2837 page_ref(pPg); 2838 return PGHDR_TO_DATA(pPg); 2839 } 2840 2841 /* 2842 ** Release a page. 2843 ** 2844 ** If the number of references to the page drop to zero, then the 2845 ** page is added to the LRU list. When all references to all pages 2846 ** are released, a rollback occurs and the lock on the database is 2847 ** removed. 2848 */ 2849 int sqlite3pager_unref(void *pData){ 2850 PgHdr *pPg; 2851 2852 /* Decrement the reference count for this page 2853 */ 2854 pPg = DATA_TO_PGHDR(pData); 2855 assert( pPg->nRef>0 ); 2856 pPg->nRef--; 2857 REFINFO(pPg); 2858 2859 CHECK_PAGE(pPg); 2860 2861 /* When the number of references to a page reach 0, call the 2862 ** destructor and add the page to the freelist. 2863 */ 2864 if( pPg->nRef==0 ){ 2865 Pager *pPager; 2866 pPager = pPg->pPager; 2867 pPg->pNextFree = 0; 2868 pPg->pPrevFree = pPager->pLast; 2869 pPager->pLast = pPg; 2870 if( pPg->pPrevFree ){ 2871 pPg->pPrevFree->pNextFree = pPg; 2872 }else{ 2873 pPager->pFirst = pPg; 2874 } 2875 if( pPg->needSync==0 && pPager->pFirstSynced==0 ){ 2876 pPager->pFirstSynced = pPg; 2877 } 2878 if( pPager->xDestructor ){ 2879 pPager->xDestructor(pData, pPager->pageSize); 2880 } 2881 2882 /* When all pages reach the freelist, drop the read lock from 2883 ** the database file. 2884 */ 2885 pPager->nRef--; 2886 assert( pPager->nRef>=0 ); 2887 if( pPager->nRef==0 && !MEMDB ){ 2888 pager_reset(pPager); 2889 } 2890 } 2891 return SQLITE_OK; 2892 } 2893 2894 /* 2895 ** Create a journal file for pPager. There should already be a RESERVED 2896 ** or EXCLUSIVE lock on the database file when this routine is called. 2897 ** 2898 ** Return SQLITE_OK if everything. Return an error code and release the 2899 ** write lock if anything goes wrong. 2900 */ 2901 static int pager_open_journal(Pager *pPager){ 2902 int rc; 2903 assert( !MEMDB ); 2904 assert( pPager->state>=PAGER_RESERVED ); 2905 assert( pPager->journalOpen==0 ); 2906 assert( pPager->useJournal ); 2907 assert( pPager->aInJournal==0 ); 2908 sqlite3pager_pagecount(pPager); 2909 pPager->aInJournal = sqliteMalloc( pPager->dbSize/8 + 1 ); 2910 if( pPager->aInJournal==0 ){ 2911 rc = SQLITE_NOMEM; 2912 goto failed_to_open_journal; 2913 } 2914 rc = sqlite3OsOpenExclusive(pPager->zJournal, &pPager->jfd, 2915 pPager->tempFile); 2916 pPager->journalOff = 0; 2917 pPager->setMaster = 0; 2918 pPager->journalHdr = 0; 2919 if( rc!=SQLITE_OK ){ 2920 goto failed_to_open_journal; 2921 } 2922 sqlite3OsSetFullSync(pPager->jfd, pPager->full_fsync); 2923 sqlite3OsSetFullSync(pPager->fd, pPager->full_fsync); 2924 sqlite3OsOpenDirectory(pPager->jfd, pPager->zDirectory); 2925 pPager->journalOpen = 1; 2926 pPager->journalStarted = 0; 2927 pPager->needSync = 0; 2928 pPager->alwaysRollback = 0; 2929 pPager->nRec = 0; 2930 if( pPager->errCode ){ 2931 rc = pPager->errCode; 2932 goto failed_to_open_journal; 2933 } 2934 pPager->origDbSize = pPager->dbSize; 2935 2936 rc = writeJournalHdr(pPager); 2937 2938 if( pPager->stmtAutoopen && rc==SQLITE_OK ){ 2939 rc = sqlite3pager_stmt_begin(pPager); 2940 } 2941 if( rc!=SQLITE_OK && rc!=SQLITE_NOMEM ){ 2942 rc = pager_unwritelock(pPager); 2943 if( rc==SQLITE_OK ){ 2944 rc = SQLITE_FULL; 2945 } 2946 } 2947 return rc; 2948 2949 failed_to_open_journal: 2950 sqliteFree(pPager->aInJournal); 2951 pPager->aInJournal = 0; 2952 if( rc==SQLITE_NOMEM ){ 2953 /* If this was a malloc() failure, then we will not be closing the pager 2954 ** file. So delete any journal file we may have just created. Otherwise, 2955 ** the system will get confused, we have a read-lock on the file and a 2956 ** mysterious journal has appeared in the filesystem. 2957 */ 2958 sqlite3OsDelete(pPager->zJournal); 2959 }else{ 2960 pager_reset(pPager); 2961 } 2962 return rc; 2963 } 2964 2965 /* 2966 ** Acquire a write-lock on the database. The lock is removed when 2967 ** the any of the following happen: 2968 ** 2969 ** * sqlite3pager_commit() is called. 2970 ** * sqlite3pager_rollback() is called. 2971 ** * sqlite3pager_close() is called. 2972 ** * sqlite3pager_unref() is called to on every outstanding page. 2973 ** 2974 ** The first parameter to this routine is a pointer to any open page of the 2975 ** database file. Nothing changes about the page - it is used merely to 2976 ** acquire a pointer to the Pager structure and as proof that there is 2977 ** already a read-lock on the database. 2978 ** 2979 ** The second parameter indicates how much space in bytes to reserve for a 2980 ** master journal file-name at the start of the journal when it is created. 2981 ** 2982 ** A journal file is opened if this is not a temporary file. For temporary 2983 ** files, the opening of the journal file is deferred until there is an 2984 ** actual need to write to the journal. 2985 ** 2986 ** If the database is already reserved for writing, this routine is a no-op. 2987 ** 2988 ** If exFlag is true, go ahead and get an EXCLUSIVE lock on the file 2989 ** immediately instead of waiting until we try to flush the cache. The 2990 ** exFlag is ignored if a transaction is already active. 2991 */ 2992 int sqlite3pager_begin(void *pData, int exFlag){ 2993 PgHdr *pPg = DATA_TO_PGHDR(pData); 2994 Pager *pPager = pPg->pPager; 2995 int rc = SQLITE_OK; 2996 assert( pPg->nRef>0 ); 2997 assert( pPager->state!=PAGER_UNLOCK ); 2998 if( pPager->state==PAGER_SHARED ){ 2999 assert( pPager->aInJournal==0 ); 3000 if( MEMDB ){ 3001 pPager->state = PAGER_EXCLUSIVE; 3002 pPager->origDbSize = pPager->dbSize; 3003 }else{ 3004 rc = sqlite3OsLock(pPager->fd, RESERVED_LOCK); 3005 if( rc==SQLITE_OK ){ 3006 pPager->state = PAGER_RESERVED; 3007 if( exFlag ){ 3008 rc = pager_wait_on_lock(pPager, EXCLUSIVE_LOCK); 3009 } 3010 } 3011 if( rc!=SQLITE_OK ){ 3012 return rc; 3013 } 3014 pPager->dirtyCache = 0; 3015 TRACE2("TRANSACTION %d\n", PAGERID(pPager)); 3016 if( pPager->useJournal && !pPager->tempFile ){ 3017 rc = pager_open_journal(pPager); 3018 } 3019 } 3020 } 3021 return rc; 3022 } 3023 3024 /* 3025 ** Make a page dirty. Set its dirty flag and add it to the dirty 3026 ** page list. 3027 */ 3028 static void makeDirty(PgHdr *pPg){ 3029 if( pPg->dirty==0 ){ 3030 Pager *pPager = pPg->pPager; 3031 pPg->dirty = 1; 3032 pPg->pDirty = pPager->pDirty; 3033 if( pPager->pDirty ){ 3034 pPager->pDirty->pPrevDirty = pPg; 3035 } 3036 pPg->pPrevDirty = 0; 3037 pPager->pDirty = pPg; 3038 } 3039 } 3040 3041 /* 3042 ** Make a page clean. Clear its dirty bit and remove it from the 3043 ** dirty page list. 3044 */ 3045 static void makeClean(PgHdr *pPg){ 3046 if( pPg->dirty ){ 3047 pPg->dirty = 0; 3048 if( pPg->pDirty ){ 3049 pPg->pDirty->pPrevDirty = pPg->pPrevDirty; 3050 } 3051 if( pPg->pPrevDirty ){ 3052 pPg->pPrevDirty->pDirty = pPg->pDirty; 3053 }else{ 3054 pPg->pPager->pDirty = pPg->pDirty; 3055 } 3056 } 3057 } 3058 3059 3060 /* 3061 ** Mark a data page as writeable. The page is written into the journal 3062 ** if it is not there already. This routine must be called before making 3063 ** changes to a page. 3064 ** 3065 ** The first time this routine is called, the pager creates a new 3066 ** journal and acquires a RESERVED lock on the database. If the RESERVED 3067 ** lock could not be acquired, this routine returns SQLITE_BUSY. The 3068 ** calling routine must check for that return value and be careful not to 3069 ** change any page data until this routine returns SQLITE_OK. 3070 ** 3071 ** If the journal file could not be written because the disk is full, 3072 ** then this routine returns SQLITE_FULL and does an immediate rollback. 3073 ** All subsequent write attempts also return SQLITE_FULL until there 3074 ** is a call to sqlite3pager_commit() or sqlite3pager_rollback() to 3075 ** reset. 3076 */ 3077 int sqlite3pager_write(void *pData){ 3078 PgHdr *pPg = DATA_TO_PGHDR(pData); 3079 Pager *pPager = pPg->pPager; 3080 int rc = SQLITE_OK; 3081 3082 /* Check for errors 3083 */ 3084 if( pPager->errCode ){ 3085 return pPager->errCode; 3086 } 3087 if( pPager->readOnly ){ 3088 return SQLITE_PERM; 3089 } 3090 3091 assert( !pPager->setMaster ); 3092 3093 CHECK_PAGE(pPg); 3094 3095 /* Mark the page as dirty. If the page has already been written 3096 ** to the journal then we can return right away. 3097 */ 3098 makeDirty(pPg); 3099 if( pPg->inJournal && (pPg->inStmt || pPager->stmtInUse==0) ){ 3100 pPager->dirtyCache = 1; 3101 }else{ 3102 3103 /* If we get this far, it means that the page needs to be 3104 ** written to the transaction journal or the ckeckpoint journal 3105 ** or both. 3106 ** 3107 ** First check to see that the transaction journal exists and 3108 ** create it if it does not. 3109 */ 3110 assert( pPager->state!=PAGER_UNLOCK ); 3111 rc = sqlite3pager_begin(pData, 0); 3112 if( rc!=SQLITE_OK ){ 3113 return rc; 3114 } 3115 assert( pPager->state>=PAGER_RESERVED ); 3116 if( !pPager->journalOpen && pPager->useJournal ){ 3117 rc = pager_open_journal(pPager); 3118 if( rc!=SQLITE_OK ) return rc; 3119 } 3120 assert( pPager->journalOpen || !pPager->useJournal ); 3121 pPager->dirtyCache = 1; 3122 3123 /* The transaction journal now exists and we have a RESERVED or an 3124 ** EXCLUSIVE lock on the main database file. Write the current page to 3125 ** the transaction journal if it is not there already. 3126 */ 3127 if( !pPg->inJournal && (pPager->useJournal || MEMDB) ){ 3128 if( (int)pPg->pgno <= pPager->origDbSize ){ 3129 int szPg; 3130 if( MEMDB ){ 3131 PgHistory *pHist = PGHDR_TO_HIST(pPg, pPager); 3132 TRACE3("JOURNAL %d page %d\n", PAGERID(pPager), pPg->pgno); 3133 assert( pHist->pOrig==0 ); 3134 pHist->pOrig = sqliteMallocRaw( pPager->pageSize ); 3135 if( pHist->pOrig ){ 3136 memcpy(pHist->pOrig, PGHDR_TO_DATA(pPg), pPager->pageSize); 3137 } 3138 }else{ 3139 u32 cksum, saved; 3140 char *pData2, *pEnd; 3141 /* We should never write to the journal file the page that 3142 ** contains the database locks. The following assert verifies 3143 ** that we do not. */ 3144 assert( pPg->pgno!=PAGER_MJ_PGNO(pPager) ); 3145 pData2 = CODEC2(pPager, pData, pPg->pgno, 7); 3146 cksum = pager_cksum(pPager, (u8*)pData2); 3147 pEnd = pData2 + pPager->pageSize; 3148 pData2 -= 4; 3149 saved = *(u32*)pEnd; 3150 put32bits(pEnd, cksum); 3151 szPg = pPager->pageSize+8; 3152 put32bits(pData2, pPg->pgno); 3153 rc = sqlite3OsWrite(pPager->jfd, pData2, szPg); 3154 pPager->journalOff += szPg; 3155 TRACE4("JOURNAL %d page %d needSync=%d\n", 3156 PAGERID(pPager), pPg->pgno, pPg->needSync); 3157 *(u32*)pEnd = saved; 3158 3159 /* An error has occured writing to the journal file. The 3160 ** transaction will be rolled back by the layer above. 3161 */ 3162 if( rc!=SQLITE_OK ){ 3163 return rc; 3164 } 3165 3166 pPager->nRec++; 3167 assert( pPager->aInJournal!=0 ); 3168 pPager->aInJournal[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3169 pPg->needSync = !pPager->noSync; 3170 if( pPager->stmtInUse ){ 3171 pPager->aInStmt[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3172 page_add_to_stmt_list(pPg); 3173 } 3174 } 3175 }else{ 3176 pPg->needSync = !pPager->journalStarted && !pPager->noSync; 3177 TRACE4("APPEND %d page %d needSync=%d\n", 3178 PAGERID(pPager), pPg->pgno, pPg->needSync); 3179 } 3180 if( pPg->needSync ){ 3181 pPager->needSync = 1; 3182 } 3183 pPg->inJournal = 1; 3184 } 3185 3186 /* If the statement journal is open and the page is not in it, 3187 ** then write the current page to the statement journal. Note that 3188 ** the statement journal format differs from the standard journal format 3189 ** in that it omits the checksums and the header. 3190 */ 3191 if( pPager->stmtInUse && !pPg->inStmt && (int)pPg->pgno<=pPager->stmtSize ){ 3192 assert( pPg->inJournal || (int)pPg->pgno>pPager->origDbSize ); 3193 if( MEMDB ){ 3194 PgHistory *pHist = PGHDR_TO_HIST(pPg, pPager); 3195 assert( pHist->pStmt==0 ); 3196 pHist->pStmt = sqliteMallocRaw( pPager->pageSize ); 3197 if( pHist->pStmt ){ 3198 memcpy(pHist->pStmt, PGHDR_TO_DATA(pPg), pPager->pageSize); 3199 } 3200 TRACE3("STMT-JOURNAL %d page %d\n", PAGERID(pPager), pPg->pgno); 3201 }else{ 3202 char *pData2 = CODEC2(pPager, pData, pPg->pgno, 7)-4; 3203 put32bits(pData2, pPg->pgno); 3204 rc = sqlite3OsWrite(pPager->stfd, pData2, pPager->pageSize+4); 3205 TRACE3("STMT-JOURNAL %d page %d\n", PAGERID(pPager), pPg->pgno); 3206 if( rc!=SQLITE_OK ){ 3207 return rc; 3208 } 3209 pPager->stmtNRec++; 3210 assert( pPager->aInStmt!=0 ); 3211 pPager->aInStmt[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3212 } 3213 page_add_to_stmt_list(pPg); 3214 } 3215 } 3216 3217 /* Update the database size and return. 3218 */ 3219 assert( pPager->state>=PAGER_SHARED ); 3220 if( pPager->dbSize<(int)pPg->pgno ){ 3221 pPager->dbSize = pPg->pgno; 3222 if( !MEMDB && pPager->dbSize==PENDING_BYTE/pPager->pageSize ){ 3223 pPager->dbSize++; 3224 } 3225 } 3226 return rc; 3227 } 3228 3229 /* 3230 ** Return TRUE if the page given in the argument was previously passed 3231 ** to sqlite3pager_write(). In other words, return TRUE if it is ok 3232 ** to change the content of the page. 3233 */ 3234 #ifndef NDEBUG 3235 int sqlite3pager_iswriteable(void *pData){ 3236 PgHdr *pPg = DATA_TO_PGHDR(pData); 3237 return pPg->dirty; 3238 } 3239 #endif 3240 3241 #ifndef SQLITE_OMIT_VACUUM 3242 /* 3243 ** Replace the content of a single page with the information in the third 3244 ** argument. 3245 */ 3246 int sqlite3pager_overwrite(Pager *pPager, Pgno pgno, void *pData){ 3247 void *pPage; 3248 int rc; 3249 3250 rc = sqlite3pager_get(pPager, pgno, &pPage); 3251 if( rc==SQLITE_OK ){ 3252 rc = sqlite3pager_write(pPage); 3253 if( rc==SQLITE_OK ){ 3254 memcpy(pPage, pData, pPager->pageSize); 3255 } 3256 sqlite3pager_unref(pPage); 3257 } 3258 return rc; 3259 } 3260 #endif 3261 3262 /* 3263 ** A call to this routine tells the pager that it is not necessary to 3264 ** write the information on page "pgno" back to the disk, even though 3265 ** that page might be marked as dirty. 3266 ** 3267 ** The overlying software layer calls this routine when all of the data 3268 ** on the given page is unused. The pager marks the page as clean so 3269 ** that it does not get written to disk. 3270 ** 3271 ** Tests show that this optimization, together with the 3272 ** sqlite3pager_dont_rollback() below, more than double the speed 3273 ** of large INSERT operations and quadruple the speed of large DELETEs. 3274 ** 3275 ** When this routine is called, set the alwaysRollback flag to true. 3276 ** Subsequent calls to sqlite3pager_dont_rollback() for the same page 3277 ** will thereafter be ignored. This is necessary to avoid a problem 3278 ** where a page with data is added to the freelist during one part of 3279 ** a transaction then removed from the freelist during a later part 3280 ** of the same transaction and reused for some other purpose. When it 3281 ** is first added to the freelist, this routine is called. When reused, 3282 ** the dont_rollback() routine is called. But because the page contains 3283 ** critical data, we still need to be sure it gets rolled back in spite 3284 ** of the dont_rollback() call. 3285 */ 3286 void sqlite3pager_dont_write(Pager *pPager, Pgno pgno){ 3287 PgHdr *pPg; 3288 3289 if( MEMDB ) return; 3290 3291 pPg = pager_lookup(pPager, pgno); 3292 assert( pPg!=0 ); /* We never call _dont_write unless the page is in mem */ 3293 pPg->alwaysRollback = 1; 3294 if( pPg->dirty && !pPager->stmtInUse ){ 3295 assert( pPager->state>=PAGER_SHARED ); 3296 if( pPager->dbSize==(int)pPg->pgno && pPager->origDbSize<pPager->dbSize ){ 3297 /* If this pages is the last page in the file and the file has grown 3298 ** during the current transaction, then do NOT mark the page as clean. 3299 ** When the database file grows, we must make sure that the last page 3300 ** gets written at least once so that the disk file will be the correct 3301 ** size. If you do not write this page and the size of the file 3302 ** on the disk ends up being too small, that can lead to database 3303 ** corruption during the next transaction. 3304 */ 3305 }else{ 3306 TRACE3("DONT_WRITE page %d of %d\n", pgno, PAGERID(pPager)); 3307 makeClean(pPg); 3308 #ifdef SQLITE_CHECK_PAGES 3309 pPg->pageHash = pager_pagehash(pPg); 3310 #endif 3311 } 3312 } 3313 } 3314 3315 /* 3316 ** A call to this routine tells the pager that if a rollback occurs, 3317 ** it is not necessary to restore the data on the given page. This 3318 ** means that the pager does not have to record the given page in the 3319 ** rollback journal. 3320 */ 3321 void sqlite3pager_dont_rollback(void *pData){ 3322 PgHdr *pPg = DATA_TO_PGHDR(pData); 3323 Pager *pPager = pPg->pPager; 3324 3325 assert( pPager->state>=PAGER_RESERVED ); 3326 if( pPager->journalOpen==0 ) return; 3327 if( pPg->alwaysRollback || pPager->alwaysRollback || MEMDB ) return; 3328 if( !pPg->inJournal && (int)pPg->pgno <= pPager->origDbSize ){ 3329 assert( pPager->aInJournal!=0 ); 3330 pPager->aInJournal[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3331 pPg->inJournal = 1; 3332 if( pPager->stmtInUse ){ 3333 pPager->aInStmt[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3334 page_add_to_stmt_list(pPg); 3335 } 3336 TRACE3("DONT_ROLLBACK page %d of %d\n", pPg->pgno, PAGERID(pPager)); 3337 } 3338 if( pPager->stmtInUse && !pPg->inStmt && (int)pPg->pgno<=pPager->stmtSize ){ 3339 assert( pPg->inJournal || (int)pPg->pgno>pPager->origDbSize ); 3340 assert( pPager->aInStmt!=0 ); 3341 pPager->aInStmt[pPg->pgno/8] |= 1<<(pPg->pgno&7); 3342 page_add_to_stmt_list(pPg); 3343 } 3344 } 3345 3346 3347 /* 3348 ** Commit all changes to the database and release the write lock. 3349 ** 3350 ** If the commit fails for any reason, a rollback attempt is made 3351 ** and an error code is returned. If the commit worked, SQLITE_OK 3352 ** is returned. 3353 */ 3354 int sqlite3pager_commit(Pager *pPager){ 3355 int rc; 3356 PgHdr *pPg; 3357 3358 if( pPager->errCode ){ 3359 return pPager->errCode; 3360 } 3361 if( pPager->state<PAGER_RESERVED ){ 3362 return SQLITE_ERROR; 3363 } 3364 TRACE2("COMMIT %d\n", PAGERID(pPager)); 3365 if( MEMDB ){ 3366 pPg = pager_get_all_dirty_pages(pPager); 3367 while( pPg ){ 3368 clearHistory(PGHDR_TO_HIST(pPg, pPager)); 3369 pPg->dirty = 0; 3370 pPg->inJournal = 0; 3371 pPg->inStmt = 0; 3372 pPg->needSync = 0; 3373 pPg->pPrevStmt = pPg->pNextStmt = 0; 3374 pPg = pPg->pDirty; 3375 } 3376 pPager->pDirty = 0; 3377 #ifndef NDEBUG 3378 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 3379 PgHistory *pHist = PGHDR_TO_HIST(pPg, pPager); 3380 assert( !pPg->alwaysRollback ); 3381 assert( !pHist->pOrig ); 3382 assert( !pHist->pStmt ); 3383 } 3384 #endif 3385 pPager->pStmt = 0; 3386 pPager->state = PAGER_SHARED; 3387 return SQLITE_OK; 3388 } 3389 if( pPager->dirtyCache==0 ){ 3390 /* Exit early (without doing the time-consuming sqlite3OsSync() calls) 3391 ** if there have been no changes to the database file. */ 3392 assert( pPager->needSync==0 ); 3393 rc = pager_unwritelock(pPager); 3394 return rc; 3395 } 3396 assert( pPager->journalOpen ); 3397 rc = sqlite3pager_sync(pPager, 0, 0); 3398 if( rc==SQLITE_OK ){ 3399 rc = pager_unwritelock(pPager); 3400 } 3401 return rc; 3402 } 3403 3404 /* 3405 ** Rollback all changes. The database falls back to PAGER_SHARED mode. 3406 ** All in-memory cache pages revert to their original data contents. 3407 ** The journal is deleted. 3408 ** 3409 ** This routine cannot fail unless some other process is not following 3410 ** the correct locking protocol (SQLITE_PROTOCOL) or unless some other 3411 ** process is writing trash into the journal file (SQLITE_CORRUPT) or 3412 ** unless a prior malloc() failed (SQLITE_NOMEM). Appropriate error 3413 ** codes are returned for all these occasions. Otherwise, 3414 ** SQLITE_OK is returned. 3415 */ 3416 int sqlite3pager_rollback(Pager *pPager){ 3417 int rc; 3418 TRACE2("ROLLBACK %d\n", PAGERID(pPager)); 3419 if( MEMDB ){ 3420 PgHdr *p; 3421 for(p=pPager->pAll; p; p=p->pNextAll){ 3422 PgHistory *pHist; 3423 assert( !p->alwaysRollback ); 3424 if( !p->dirty ){ 3425 assert( !((PgHistory *)PGHDR_TO_HIST(p, pPager))->pOrig ); 3426 assert( !((PgHistory *)PGHDR_TO_HIST(p, pPager))->pStmt ); 3427 continue; 3428 } 3429 3430 pHist = PGHDR_TO_HIST(p, pPager); 3431 if( pHist->pOrig ){ 3432 memcpy(PGHDR_TO_DATA(p), pHist->pOrig, pPager->pageSize); 3433 TRACE3("ROLLBACK-PAGE %d of %d\n", p->pgno, PAGERID(pPager)); 3434 }else{ 3435 TRACE3("PAGE %d is clean on %d\n", p->pgno, PAGERID(pPager)); 3436 } 3437 clearHistory(pHist); 3438 p->dirty = 0; 3439 p->inJournal = 0; 3440 p->inStmt = 0; 3441 p->pPrevStmt = p->pNextStmt = 0; 3442 if( pPager->xReiniter ){ 3443 pPager->xReiniter(PGHDR_TO_DATA(p), pPager->pageSize); 3444 } 3445 } 3446 pPager->pDirty = 0; 3447 pPager->pStmt = 0; 3448 pPager->dbSize = pPager->origDbSize; 3449 memoryTruncate(pPager); 3450 pPager->stmtInUse = 0; 3451 pPager->state = PAGER_SHARED; 3452 return SQLITE_OK; 3453 } 3454 3455 if( !pPager->dirtyCache || !pPager->journalOpen ){ 3456 rc = pager_unwritelock(pPager); 3457 return rc; 3458 } 3459 3460 if( pPager->errCode && pPager->errCode!=SQLITE_FULL ){ 3461 if( pPager->state>=PAGER_EXCLUSIVE ){ 3462 pager_playback(pPager); 3463 } 3464 return pPager->errCode; 3465 } 3466 if( pPager->state==PAGER_RESERVED ){ 3467 int rc2; 3468 rc = pager_reload_cache(pPager); 3469 rc2 = pager_unwritelock(pPager); 3470 if( rc==SQLITE_OK ){ 3471 rc = rc2; 3472 } 3473 }else{ 3474 rc = pager_playback(pPager); 3475 } 3476 pPager->dbSize = -1; 3477 3478 /* If an error occurs during a ROLLBACK, we can no longer trust the pager 3479 ** cache. So call pager_error() on the way out to make any error 3480 ** persistent. 3481 */ 3482 return pager_error(pPager, rc); 3483 } 3484 3485 /* 3486 ** Return TRUE if the database file is opened read-only. Return FALSE 3487 ** if the database is (in theory) writable. 3488 */ 3489 int sqlite3pager_isreadonly(Pager *pPager){ 3490 return pPager->readOnly; 3491 } 3492 3493 /* 3494 ** Return the number of references to the pager. 3495 */ 3496 int sqlite3pager_refcount(Pager *pPager){ 3497 return pPager->nRef; 3498 } 3499 3500 #ifdef SQLITE_TEST 3501 /* 3502 ** This routine is used for testing and analysis only. 3503 */ 3504 int *sqlite3pager_stats(Pager *pPager){ 3505 static int a[11]; 3506 a[0] = pPager->nRef; 3507 a[1] = pPager->nPage; 3508 a[2] = pPager->mxPage; 3509 a[3] = pPager->dbSize; 3510 a[4] = pPager->state; 3511 a[5] = pPager->errCode; 3512 a[6] = pPager->nHit; 3513 a[7] = pPager->nMiss; 3514 a[8] = pPager->nOvfl; 3515 a[9] = pPager->nRead; 3516 a[10] = pPager->nWrite; 3517 return a; 3518 } 3519 #endif 3520 3521 /* 3522 ** Set the statement rollback point. 3523 ** 3524 ** This routine should be called with the transaction journal already 3525 ** open. A new statement journal is created that can be used to rollback 3526 ** changes of a single SQL command within a larger transaction. 3527 */ 3528 int sqlite3pager_stmt_begin(Pager *pPager){ 3529 int rc; 3530 char zTemp[SQLITE_TEMPNAME_SIZE]; 3531 assert( !pPager->stmtInUse ); 3532 assert( pPager->state>=PAGER_SHARED ); 3533 assert( pPager->dbSize>=0 ); 3534 TRACE2("STMT-BEGIN %d\n", PAGERID(pPager)); 3535 if( MEMDB ){ 3536 pPager->stmtInUse = 1; 3537 pPager->stmtSize = pPager->dbSize; 3538 return SQLITE_OK; 3539 } 3540 if( !pPager->journalOpen ){ 3541 pPager->stmtAutoopen = 1; 3542 return SQLITE_OK; 3543 } 3544 assert( pPager->journalOpen ); 3545 pPager->aInStmt = sqliteMalloc( pPager->dbSize/8 + 1 ); 3546 if( pPager->aInStmt==0 ){ 3547 /* sqlite3OsLock(pPager->fd, SHARED_LOCK); */ 3548 return SQLITE_NOMEM; 3549 } 3550 #ifndef NDEBUG 3551 rc = sqlite3OsFileSize(pPager->jfd, &pPager->stmtJSize); 3552 if( rc ) goto stmt_begin_failed; 3553 assert( pPager->stmtJSize == pPager->journalOff ); 3554 #endif 3555 pPager->stmtJSize = pPager->journalOff; 3556 pPager->stmtSize = pPager->dbSize; 3557 pPager->stmtHdrOff = 0; 3558 pPager->stmtCksum = pPager->cksumInit; 3559 if( !pPager->stmtOpen ){ 3560 rc = sqlite3pager_opentemp(zTemp, &pPager->stfd); 3561 if( rc ) goto stmt_begin_failed; 3562 pPager->stmtOpen = 1; 3563 pPager->stmtNRec = 0; 3564 } 3565 pPager->stmtInUse = 1; 3566 return SQLITE_OK; 3567 3568 stmt_begin_failed: 3569 if( pPager->aInStmt ){ 3570 sqliteFree(pPager->aInStmt); 3571 pPager->aInStmt = 0; 3572 } 3573 return rc; 3574 } 3575 3576 /* 3577 ** Commit a statement. 3578 */ 3579 int sqlite3pager_stmt_commit(Pager *pPager){ 3580 if( pPager->stmtInUse ){ 3581 PgHdr *pPg, *pNext; 3582 TRACE2("STMT-COMMIT %d\n", PAGERID(pPager)); 3583 if( !MEMDB ){ 3584 sqlite3OsSeek(pPager->stfd, 0); 3585 /* sqlite3OsTruncate(pPager->stfd, 0); */ 3586 sqliteFree( pPager->aInStmt ); 3587 pPager->aInStmt = 0; 3588 } 3589 for(pPg=pPager->pStmt; pPg; pPg=pNext){ 3590 pNext = pPg->pNextStmt; 3591 assert( pPg->inStmt ); 3592 pPg->inStmt = 0; 3593 pPg->pPrevStmt = pPg->pNextStmt = 0; 3594 if( MEMDB ){ 3595 PgHistory *pHist = PGHDR_TO_HIST(pPg, pPager); 3596 sqliteFree(pHist->pStmt); 3597 pHist->pStmt = 0; 3598 } 3599 } 3600 pPager->stmtNRec = 0; 3601 pPager->stmtInUse = 0; 3602 pPager->pStmt = 0; 3603 } 3604 pPager->stmtAutoopen = 0; 3605 return SQLITE_OK; 3606 } 3607 3608 /* 3609 ** Rollback a statement. 3610 */ 3611 int sqlite3pager_stmt_rollback(Pager *pPager){ 3612 int rc; 3613 if( pPager->stmtInUse ){ 3614 TRACE2("STMT-ROLLBACK %d\n", PAGERID(pPager)); 3615 if( MEMDB ){ 3616 PgHdr *pPg; 3617 for(pPg=pPager->pStmt; pPg; pPg=pPg->pNextStmt){ 3618 PgHistory *pHist = PGHDR_TO_HIST(pPg, pPager); 3619 if( pHist->pStmt ){ 3620 memcpy(PGHDR_TO_DATA(pPg), pHist->pStmt, pPager->pageSize); 3621 sqliteFree(pHist->pStmt); 3622 pHist->pStmt = 0; 3623 } 3624 } 3625 pPager->dbSize = pPager->stmtSize; 3626 memoryTruncate(pPager); 3627 rc = SQLITE_OK; 3628 }else{ 3629 rc = pager_stmt_playback(pPager); 3630 } 3631 sqlite3pager_stmt_commit(pPager); 3632 }else{ 3633 rc = SQLITE_OK; 3634 } 3635 pPager->stmtAutoopen = 0; 3636 return rc; 3637 } 3638 3639 /* 3640 ** Return the full pathname of the database file. 3641 */ 3642 const char *sqlite3pager_filename(Pager *pPager){ 3643 return pPager->zFilename; 3644 } 3645 3646 /* 3647 ** Return the directory of the database file. 3648 */ 3649 const char *sqlite3pager_dirname(Pager *pPager){ 3650 return pPager->zDirectory; 3651 } 3652 3653 /* 3654 ** Return the full pathname of the journal file. 3655 */ 3656 const char *sqlite3pager_journalname(Pager *pPager){ 3657 return pPager->zJournal; 3658 } 3659 3660 /* 3661 ** Return true if fsync() calls are disabled for this pager. Return FALSE 3662 ** if fsync()s are executed normally. 3663 */ 3664 int sqlite3pager_nosync(Pager *pPager){ 3665 return pPager->noSync; 3666 } 3667 3668 /* 3669 ** Set the codec for this pager 3670 */ 3671 void sqlite3pager_set_codec( 3672 Pager *pPager, 3673 void *(*xCodec)(void*,void*,Pgno,int), 3674 void *pCodecArg 3675 ){ 3676 pPager->xCodec = xCodec; 3677 pPager->pCodecArg = pCodecArg; 3678 } 3679 3680 /* 3681 ** This routine is called to increment the database file change-counter, 3682 ** stored at byte 24 of the pager file. 3683 */ 3684 static int pager_incr_changecounter(Pager *pPager){ 3685 void *pPage; 3686 PgHdr *pPgHdr; 3687 u32 change_counter; 3688 int rc; 3689 3690 /* Open page 1 of the file for writing. */ 3691 rc = sqlite3pager_get(pPager, 1, &pPage); 3692 if( rc!=SQLITE_OK ) return rc; 3693 rc = sqlite3pager_write(pPage); 3694 if( rc!=SQLITE_OK ) return rc; 3695 3696 /* Read the current value at byte 24. */ 3697 pPgHdr = DATA_TO_PGHDR(pPage); 3698 change_counter = retrieve32bits(pPgHdr, 24); 3699 3700 /* Increment the value just read and write it back to byte 24. */ 3701 change_counter++; 3702 put32bits(((char*)PGHDR_TO_DATA(pPgHdr))+24, change_counter); 3703 3704 /* Release the page reference. */ 3705 sqlite3pager_unref(pPage); 3706 return SQLITE_OK; 3707 } 3708 3709 /* 3710 ** Sync the database file for the pager pPager. zMaster points to the name 3711 ** of a master journal file that should be written into the individual 3712 ** journal file. zMaster may be NULL, which is interpreted as no master 3713 ** journal (a single database transaction). 3714 ** 3715 ** This routine ensures that the journal is synced, all dirty pages written 3716 ** to the database file and the database file synced. The only thing that 3717 ** remains to commit the transaction is to delete the journal file (or 3718 ** master journal file if specified). 3719 ** 3720 ** Note that if zMaster==NULL, this does not overwrite a previous value 3721 ** passed to an sqlite3pager_sync() call. 3722 ** 3723 ** If parameter nTrunc is non-zero, then the pager file is truncated to 3724 ** nTrunc pages (this is used by auto-vacuum databases). 3725 */ 3726 int sqlite3pager_sync(Pager *pPager, const char *zMaster, Pgno nTrunc){ 3727 int rc = SQLITE_OK; 3728 3729 TRACE4("DATABASE SYNC: File=%s zMaster=%s nTrunc=%d\n", 3730 pPager->zFilename, zMaster, nTrunc); 3731 3732 /* If this is an in-memory db, or no pages have been written to, or this 3733 ** function has already been called, it is a no-op. 3734 */ 3735 if( pPager->state!=PAGER_SYNCED && !MEMDB && pPager->dirtyCache ){ 3736 PgHdr *pPg; 3737 assert( pPager->journalOpen ); 3738 3739 /* If a master journal file name has already been written to the 3740 ** journal file, then no sync is required. This happens when it is 3741 ** written, then the process fails to upgrade from a RESERVED to an 3742 ** EXCLUSIVE lock. The next time the process tries to commit the 3743 ** transaction the m-j name will have already been written. 3744 */ 3745 if( !pPager->setMaster ){ 3746 rc = pager_incr_changecounter(pPager); 3747 if( rc!=SQLITE_OK ) goto sync_exit; 3748 #ifndef SQLITE_OMIT_AUTOVACUUM 3749 if( nTrunc!=0 ){ 3750 /* If this transaction has made the database smaller, then all pages 3751 ** being discarded by the truncation must be written to the journal 3752 ** file. 3753 */ 3754 Pgno i; 3755 void *pPage; 3756 int iSkip = PAGER_MJ_PGNO(pPager); 3757 for( i=nTrunc+1; i<=pPager->origDbSize; i++ ){ 3758 if( !(pPager->aInJournal[i/8] & (1<<(i&7))) && i!=iSkip ){ 3759 rc = sqlite3pager_get(pPager, i, &pPage); 3760 if( rc!=SQLITE_OK ) goto sync_exit; 3761 rc = sqlite3pager_write(pPage); 3762 sqlite3pager_unref(pPage); 3763 if( rc!=SQLITE_OK ) goto sync_exit; 3764 } 3765 } 3766 } 3767 #endif 3768 rc = writeMasterJournal(pPager, zMaster); 3769 if( rc!=SQLITE_OK ) goto sync_exit; 3770 rc = syncJournal(pPager); 3771 if( rc!=SQLITE_OK ) goto sync_exit; 3772 } 3773 3774 #ifndef SQLITE_OMIT_AUTOVACUUM 3775 if( nTrunc!=0 ){ 3776 rc = sqlite3pager_truncate(pPager, nTrunc); 3777 if( rc!=SQLITE_OK ) goto sync_exit; 3778 } 3779 #endif 3780 3781 /* Write all dirty pages to the database file */ 3782 pPg = pager_get_all_dirty_pages(pPager); 3783 rc = pager_write_pagelist(pPg); 3784 if( rc!=SQLITE_OK ) goto sync_exit; 3785 3786 /* Sync the database file. */ 3787 if( !pPager->noSync ){ 3788 rc = sqlite3OsSync(pPager->fd, 0); 3789 } 3790 3791 pPager->state = PAGER_SYNCED; 3792 }else if( MEMDB && nTrunc!=0 ){ 3793 rc = sqlite3pager_truncate(pPager, nTrunc); 3794 } 3795 3796 sync_exit: 3797 return rc; 3798 } 3799 3800 #ifndef SQLITE_OMIT_AUTOVACUUM 3801 /* 3802 ** Move the page identified by pData to location pgno in the file. 3803 ** 3804 ** There must be no references to the current page pgno. If current page 3805 ** pgno is not already in the rollback journal, it is not written there by 3806 ** by this routine. The same applies to the page pData refers to on entry to 3807 ** this routine. 3808 ** 3809 ** References to the page refered to by pData remain valid. Updating any 3810 ** meta-data associated with page pData (i.e. data stored in the nExtra bytes 3811 ** allocated along with the page) is the responsibility of the caller. 3812 ** 3813 ** A transaction must be active when this routine is called. It used to be 3814 ** required that a statement transaction was not active, but this restriction 3815 ** has been removed (CREATE INDEX needs to move a page when a statement 3816 ** transaction is active). 3817 */ 3818 int sqlite3pager_movepage(Pager *pPager, void *pData, Pgno pgno){ 3819 PgHdr *pPg = DATA_TO_PGHDR(pData); 3820 PgHdr *pPgOld; 3821 int h; 3822 Pgno needSyncPgno = 0; 3823 3824 assert( pPg->nRef>0 ); 3825 3826 TRACE5("MOVE %d page %d (needSync=%d) moves to %d\n", 3827 PAGERID(pPager), pPg->pgno, pPg->needSync, pgno); 3828 3829 if( pPg->needSync ){ 3830 needSyncPgno = pPg->pgno; 3831 assert( pPg->inJournal ); 3832 assert( pPg->dirty ); 3833 assert( pPager->needSync ); 3834 } 3835 3836 /* Unlink pPg from it's hash-chain */ 3837 unlinkHashChain(pPager, pPg); 3838 3839 /* If the cache contains a page with page-number pgno, remove it 3840 ** from it's hash chain. Also, if the PgHdr.needSync was set for 3841 ** page pgno before the 'move' operation, it needs to be retained 3842 ** for the page moved there. 3843 */ 3844 pPgOld = pager_lookup(pPager, pgno); 3845 if( pPgOld ){ 3846 assert( pPgOld->nRef==0 ); 3847 unlinkHashChain(pPager, pPgOld); 3848 makeClean(pPgOld); 3849 if( pPgOld->needSync ){ 3850 assert( pPgOld->inJournal ); 3851 pPg->inJournal = 1; 3852 pPg->needSync = 1; 3853 assert( pPager->needSync ); 3854 } 3855 } 3856 3857 /* Change the page number for pPg and insert it into the new hash-chain. */ 3858 assert( pgno!=0 ); 3859 pPg->pgno = pgno; 3860 h = pgno & (pPager->nHash-1); 3861 if( pPager->aHash[h] ){ 3862 assert( pPager->aHash[h]->pPrevHash==0 ); 3863 pPager->aHash[h]->pPrevHash = pPg; 3864 } 3865 pPg->pNextHash = pPager->aHash[h]; 3866 pPager->aHash[h] = pPg; 3867 pPg->pPrevHash = 0; 3868 3869 makeDirty(pPg); 3870 pPager->dirtyCache = 1; 3871 3872 if( needSyncPgno ){ 3873 /* If needSyncPgno is non-zero, then the journal file needs to be 3874 ** sync()ed before any data is written to database file page needSyncPgno. 3875 ** Currently, no such page exists in the page-cache and the 3876 ** Pager.aInJournal bit has been set. This needs to be remedied by loading 3877 ** the page into the pager-cache and setting the PgHdr.needSync flag. 3878 ** 3879 ** The sqlite3pager_get() call may cause the journal to sync. So make 3880 ** sure the Pager.needSync flag is set too. 3881 */ 3882 int rc; 3883 void *pNeedSync; 3884 assert( pPager->needSync ); 3885 rc = sqlite3pager_get(pPager, needSyncPgno, &pNeedSync); 3886 if( rc!=SQLITE_OK ) return rc; 3887 pPager->needSync = 1; 3888 DATA_TO_PGHDR(pNeedSync)->needSync = 1; 3889 DATA_TO_PGHDR(pNeedSync)->inJournal = 1; 3890 makeDirty(DATA_TO_PGHDR(pNeedSync)); 3891 sqlite3pager_unref(pNeedSync); 3892 } 3893 3894 return SQLITE_OK; 3895 } 3896 #endif 3897 3898 #if defined(SQLITE_DEBUG) || defined(SQLITE_TEST) 3899 /* 3900 ** Return the current state of the file lock for the given pager. 3901 ** The return value is one of NO_LOCK, SHARED_LOCK, RESERVED_LOCK, 3902 ** PENDING_LOCK, or EXCLUSIVE_LOCK. 3903 */ 3904 int sqlite3pager_lockstate(Pager *pPager){ 3905 return sqlite3OsLockState(pPager->fd); 3906 } 3907 #endif 3908 3909 #ifdef SQLITE_DEBUG 3910 /* 3911 ** Print a listing of all referenced pages and their ref count. 3912 */ 3913 void sqlite3pager_refdump(Pager *pPager){ 3914 PgHdr *pPg; 3915 for(pPg=pPager->pAll; pPg; pPg=pPg->pNextAll){ 3916 if( pPg->nRef<=0 ) continue; 3917 sqlite3DebugPrintf("PAGE %3d addr=%p nRef=%d\n", 3918 pPg->pgno, PGHDR_TO_DATA(pPg), pPg->nRef); 3919 } 3920 } 3921 #endif 3922 3923 #endif /* SQLITE_OMIT_DISKIO */ 3924