| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 ** 2007 August 22 | |
| 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 ** | |
| 13 ** This file implements a special kind of sqlite3_file object used | |
| 14 ** by SQLite to create journal files if the atomic-write optimization | |
| 15 ** is enabled. | |
| 16 ** | |
| 17 ** The distinctive characteristic of this sqlite3_file is that the | |
| 18 ** actual on disk file is created lazily. When the file is created, | |
| 19 ** the caller specifies a buffer size for an in-memory buffer to | |
| 20 ** be used to service read() and write() requests. The actual file | |
| 21 ** on disk is not created or populated until either: | |
| 22 ** | |
| 23 ** 1) The in-memory representation grows too large for the allocated | |
| 24 ** buffer, or | |
| 25 ** 2) The sqlite3JournalCreate() function is called. | |
| 26 */ | |
| 27 #ifdef SQLITE_ENABLE_ATOMIC_WRITE | |
| 28 #include "sqliteInt.h" | |
| 29 | |
| 30 | |
| 31 /* | |
| 32 ** A JournalFile object is a subclass of sqlite3_file used by | |
| 33 ** as an open file handle for journal files. | |
| 34 */ | |
| 35 struct JournalFile { | |
| 36 sqlite3_io_methods *pMethod; /* I/O methods on journal files */ | |
| 37 int nBuf; /* Size of zBuf[] in bytes */ | |
| 38 char *zBuf; /* Space to buffer journal writes */ | |
| 39 int iSize; /* Amount of zBuf[] currently used */ | |
| 40 int flags; /* xOpen flags */ | |
| 41 sqlite3_vfs *pVfs; /* The "real" underlying VFS */ | |
| 42 sqlite3_file *pReal; /* The "real" underlying file descriptor */ | |
| 43 const char *zJournal; /* Name of the journal file */ | |
| 44 }; | |
| 45 typedef struct JournalFile JournalFile; | |
| 46 | |
| 47 /* | |
| 48 ** If it does not already exists, create and populate the on-disk file | |
| 49 ** for JournalFile p. | |
| 50 */ | |
| 51 static int createFile(JournalFile *p){ | |
| 52 int rc = SQLITE_OK; | |
| 53 if( !p->pReal ){ | |
| 54 sqlite3_file *pReal = (sqlite3_file *)&p[1]; | |
| 55 rc = sqlite3OsOpen(p->pVfs, p->zJournal, pReal, p->flags, 0); | |
| 56 if( rc==SQLITE_OK ){ | |
| 57 p->pReal = pReal; | |
| 58 if( p->iSize>0 ){ | |
| 59 assert(p->iSize<=p->nBuf); | |
| 60 rc = sqlite3OsWrite(p->pReal, p->zBuf, p->iSize, 0); | |
| 61 } | |
| 62 if( rc!=SQLITE_OK ){ | |
| 63 /* If an error occurred while writing to the file, close it before | |
| 64 ** returning. This way, SQLite uses the in-memory journal data to | |
| 65 ** roll back changes made to the internal page-cache before this | |
| 66 ** function was called. */ | |
| 67 sqlite3OsClose(pReal); | |
| 68 p->pReal = 0; | |
| 69 } | |
| 70 } | |
| 71 } | |
| 72 return rc; | |
| 73 } | |
| 74 | |
| 75 /* | |
| 76 ** Close the file. | |
| 77 */ | |
| 78 static int jrnlClose(sqlite3_file *pJfd){ | |
| 79 JournalFile *p = (JournalFile *)pJfd; | |
| 80 if( p->pReal ){ | |
| 81 sqlite3OsClose(p->pReal); | |
| 82 } | |
| 83 sqlite3_free(p->zBuf); | |
| 84 return SQLITE_OK; | |
| 85 } | |
| 86 | |
| 87 /* | |
| 88 ** Read data from the file. | |
| 89 */ | |
| 90 static int jrnlRead( | |
| 91 sqlite3_file *pJfd, /* The journal file from which to read */ | |
| 92 void *zBuf, /* Put the results here */ | |
| 93 int iAmt, /* Number of bytes to read */ | |
| 94 sqlite_int64 iOfst /* Begin reading at this offset */ | |
| 95 ){ | |
| 96 int rc = SQLITE_OK; | |
| 97 JournalFile *p = (JournalFile *)pJfd; | |
| 98 if( p->pReal ){ | |
| 99 rc = sqlite3OsRead(p->pReal, zBuf, iAmt, iOfst); | |
| 100 }else if( (iAmt+iOfst)>p->iSize ){ | |
| 101 rc = SQLITE_IOERR_SHORT_READ; | |
| 102 }else{ | |
| 103 memcpy(zBuf, &p->zBuf[iOfst], iAmt); | |
| 104 } | |
| 105 return rc; | |
| 106 } | |
| 107 | |
| 108 /* | |
| 109 ** Write data to the file. | |
| 110 */ | |
| 111 static int jrnlWrite( | |
| 112 sqlite3_file *pJfd, /* The journal file into which to write */ | |
| 113 const void *zBuf, /* Take data to be written from here */ | |
| 114 int iAmt, /* Number of bytes to write */ | |
| 115 sqlite_int64 iOfst /* Begin writing at this offset into the file */ | |
| 116 ){ | |
| 117 int rc = SQLITE_OK; | |
| 118 JournalFile *p = (JournalFile *)pJfd; | |
| 119 if( !p->pReal && (iOfst+iAmt)>p->nBuf ){ | |
| 120 rc = createFile(p); | |
| 121 } | |
| 122 if( rc==SQLITE_OK ){ | |
| 123 if( p->pReal ){ | |
| 124 rc = sqlite3OsWrite(p->pReal, zBuf, iAmt, iOfst); | |
| 125 }else{ | |
| 126 memcpy(&p->zBuf[iOfst], zBuf, iAmt); | |
| 127 if( p->iSize<(iOfst+iAmt) ){ | |
| 128 p->iSize = (iOfst+iAmt); | |
| 129 } | |
| 130 } | |
| 131 } | |
| 132 return rc; | |
| 133 } | |
| 134 | |
| 135 /* | |
| 136 ** Truncate the file. | |
| 137 */ | |
| 138 static int jrnlTruncate(sqlite3_file *pJfd, sqlite_int64 size){ | |
| 139 int rc = SQLITE_OK; | |
| 140 JournalFile *p = (JournalFile *)pJfd; | |
| 141 if( p->pReal ){ | |
| 142 rc = sqlite3OsTruncate(p->pReal, size); | |
| 143 }else if( size<p->iSize ){ | |
| 144 p->iSize = size; | |
| 145 } | |
| 146 return rc; | |
| 147 } | |
| 148 | |
| 149 /* | |
| 150 ** Sync the file. | |
| 151 */ | |
| 152 static int jrnlSync(sqlite3_file *pJfd, int flags){ | |
| 153 int rc; | |
| 154 JournalFile *p = (JournalFile *)pJfd; | |
| 155 if( p->pReal ){ | |
| 156 rc = sqlite3OsSync(p->pReal, flags); | |
| 157 }else{ | |
| 158 rc = SQLITE_OK; | |
| 159 } | |
| 160 return rc; | |
| 161 } | |
| 162 | |
| 163 /* | |
| 164 ** Query the size of the file in bytes. | |
| 165 */ | |
| 166 static int jrnlFileSize(sqlite3_file *pJfd, sqlite_int64 *pSize){ | |
| 167 int rc = SQLITE_OK; | |
| 168 JournalFile *p = (JournalFile *)pJfd; | |
| 169 if( p->pReal ){ | |
| 170 rc = sqlite3OsFileSize(p->pReal, pSize); | |
| 171 }else{ | |
| 172 *pSize = (sqlite_int64) p->iSize; | |
| 173 } | |
| 174 return rc; | |
| 175 } | |
| 176 | |
| 177 /* | |
| 178 ** Table of methods for JournalFile sqlite3_file object. | |
| 179 */ | |
| 180 static struct sqlite3_io_methods JournalFileMethods = { | |
| 181 1, /* iVersion */ | |
| 182 jrnlClose, /* xClose */ | |
| 183 jrnlRead, /* xRead */ | |
| 184 jrnlWrite, /* xWrite */ | |
| 185 jrnlTruncate, /* xTruncate */ | |
| 186 jrnlSync, /* xSync */ | |
| 187 jrnlFileSize, /* xFileSize */ | |
| 188 0, /* xLock */ | |
| 189 0, /* xUnlock */ | |
| 190 0, /* xCheckReservedLock */ | |
| 191 0, /* xFileControl */ | |
| 192 0, /* xSectorSize */ | |
| 193 0, /* xDeviceCharacteristics */ | |
| 194 0, /* xShmMap */ | |
| 195 0, /* xShmLock */ | |
| 196 0, /* xShmBarrier */ | |
| 197 0 /* xShmUnmap */ | |
| 198 }; | |
| 199 | |
| 200 /* | |
| 201 ** Open a journal file. | |
| 202 */ | |
| 203 int sqlite3JournalOpen( | |
| 204 sqlite3_vfs *pVfs, /* The VFS to use for actual file I/O */ | |
| 205 const char *zName, /* Name of the journal file */ | |
| 206 sqlite3_file *pJfd, /* Preallocated, blank file handle */ | |
| 207 int flags, /* Opening flags */ | |
| 208 int nBuf /* Bytes buffered before opening the file */ | |
| 209 ){ | |
| 210 JournalFile *p = (JournalFile *)pJfd; | |
| 211 memset(p, 0, sqlite3JournalSize(pVfs)); | |
| 212 if( nBuf>0 ){ | |
| 213 p->zBuf = sqlite3MallocZero(nBuf); | |
| 214 if( !p->zBuf ){ | |
| 215 return SQLITE_NOMEM; | |
| 216 } | |
| 217 }else{ | |
| 218 return sqlite3OsOpen(pVfs, zName, pJfd, flags, 0); | |
| 219 } | |
| 220 p->pMethod = &JournalFileMethods; | |
| 221 p->nBuf = nBuf; | |
| 222 p->flags = flags; | |
| 223 p->zJournal = zName; | |
| 224 p->pVfs = pVfs; | |
| 225 return SQLITE_OK; | |
| 226 } | |
| 227 | |
| 228 /* | |
| 229 ** If the argument p points to a JournalFile structure, and the underlying | |
| 230 ** file has not yet been created, create it now. | |
| 231 */ | |
| 232 int sqlite3JournalCreate(sqlite3_file *p){ | |
| 233 if( p->pMethods!=&JournalFileMethods ){ | |
| 234 return SQLITE_OK; | |
| 235 } | |
| 236 return createFile((JournalFile *)p); | |
| 237 } | |
| 238 | |
| 239 /* | |
| 240 ** The file-handle passed as the only argument is guaranteed to be an open | |
| 241 ** file. It may or may not be of class JournalFile. If the file is a | |
| 242 ** JournalFile, and the underlying file on disk has not yet been opened, | |
| 243 ** return 0. Otherwise, return 1. | |
| 244 */ | |
| 245 int sqlite3JournalExists(sqlite3_file *p){ | |
| 246 return (p->pMethods!=&JournalFileMethods || ((JournalFile *)p)->pReal!=0); | |
| 247 } | |
| 248 | |
| 249 /* | |
| 250 ** Return the number of bytes required to store a JournalFile that uses vfs | |
| 251 ** pVfs to create the underlying on-disk files. | |
| 252 */ | |
| 253 int sqlite3JournalSize(sqlite3_vfs *pVfs){ | |
| 254 return (pVfs->szOsFile+sizeof(JournalFile)); | |
| 255 } | |
| 256 #endif | |
| OLD | NEW |