OLD | NEW |
(Empty) | |
| 1 /* |
| 2 ** 2013 Jan 11 |
| 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 ** Code for testing the virtual table interfaces. This code |
| 13 ** is not included in the SQLite library. It is used for automated |
| 14 ** testing of the SQLite library. |
| 15 ** |
| 16 ** The FS virtual table is created as follows: |
| 17 ** |
| 18 ** CREATE VIRTUAL TABLE tbl USING fs(idx); |
| 19 ** |
| 20 ** where idx is the name of a table in the db with 2 columns. The virtual |
| 21 ** table also has two columns - file path and file contents. |
| 22 ** |
| 23 ** The first column of table idx must be an IPK, and the second contains file |
| 24 ** paths. For example: |
| 25 ** |
| 26 ** CREATE TABLE idx(id INTEGER PRIMARY KEY, path TEXT); |
| 27 ** INSERT INTO idx VALUES(4, '/etc/passwd'); |
| 28 ** |
| 29 ** Adding the row to the idx table automatically creates a row in the |
| 30 ** virtual table with rowid=4, path=/etc/passwd and a text field that |
| 31 ** contains data read from file /etc/passwd on disk. |
| 32 ** |
| 33 ************************************************************************* |
| 34 ** Virtual table module "fsdir" |
| 35 ** |
| 36 ** This module is designed to be used as a read-only eponymous virtual table. |
| 37 ** Its schema is as follows: |
| 38 ** |
| 39 ** CREATE TABLE fsdir(dir TEXT, name TEXT); |
| 40 ** |
| 41 ** When queried, a WHERE term of the form "dir = $dir" must be provided. The |
| 42 ** virtual table then appears to have one row for each entry in file-system |
| 43 ** directory $dir. Column dir contains a copy of $dir, and column "name" |
| 44 ** contains the name of the directory entry. |
| 45 ** |
| 46 ** If the specified $dir cannot be opened or is not a directory, it is not |
| 47 ** an error. The virtual table appears to be empty in this case. |
| 48 ** |
| 49 ************************************************************************* |
| 50 ** Virtual table module "fstree" |
| 51 ** |
| 52 ** This module is also a read-only eponymous virtual table with the |
| 53 ** following schema: |
| 54 ** |
| 55 ** CREATE TABLE fstree(path TEXT, size INT, data BLOB); |
| 56 ** |
| 57 ** Running a "SELECT * FROM fstree" query on this table returns the entire |
| 58 ** contents of the file-system, starting at "/". To restrict the search |
| 59 ** space, the virtual table supports LIKE and GLOB constraints on the |
| 60 ** 'path' column. For example: |
| 61 ** |
| 62 ** SELECT * FROM fstree WHERE path LIKE '/home/dan/sqlite/%' |
| 63 */ |
| 64 #include "sqliteInt.h" |
| 65 #if defined(INCLUDE_SQLITE_TCL_H) |
| 66 # include "sqlite_tcl.h" |
| 67 #else |
| 68 # include "tcl.h" |
| 69 #endif |
| 70 |
| 71 #include <stdlib.h> |
| 72 #include <string.h> |
| 73 #include <sys/types.h> |
| 74 #include <sys/stat.h> |
| 75 #include <fcntl.h> |
| 76 |
| 77 #if SQLITE_OS_UNIX || defined(__MINGW_H) |
| 78 # include <unistd.h> |
| 79 # include <dirent.h> |
| 80 # ifndef DIRENT |
| 81 # define DIRENT dirent |
| 82 # endif |
| 83 #endif |
| 84 #if SQLITE_OS_WIN |
| 85 # include <io.h> |
| 86 # if !defined(__MINGW_H) |
| 87 # include "test_windirent.h" |
| 88 # endif |
| 89 # ifndef S_ISREG |
| 90 # define S_ISREG(mode) (((mode) & S_IFMT) == S_IFREG) |
| 91 # endif |
| 92 #endif |
| 93 |
| 94 #ifndef SQLITE_OMIT_VIRTUALTABLE |
| 95 |
| 96 typedef struct fs_vtab fs_vtab; |
| 97 typedef struct fs_cursor fs_cursor; |
| 98 |
| 99 /* |
| 100 ** A fs virtual-table object |
| 101 */ |
| 102 struct fs_vtab { |
| 103 sqlite3_vtab base; |
| 104 sqlite3 *db; |
| 105 char *zDb; /* Name of db containing zTbl */ |
| 106 char *zTbl; /* Name of docid->file map table */ |
| 107 }; |
| 108 |
| 109 /* A fs cursor object */ |
| 110 struct fs_cursor { |
| 111 sqlite3_vtab_cursor base; |
| 112 sqlite3_stmt *pStmt; |
| 113 char *zBuf; |
| 114 int nBuf; |
| 115 int nAlloc; |
| 116 }; |
| 117 |
| 118 /************************************************************************* |
| 119 ** Start of fsdir implementation. |
| 120 */ |
| 121 typedef struct FsdirVtab FsdirVtab; |
| 122 typedef struct FsdirCsr FsdirCsr; |
| 123 struct FsdirVtab { |
| 124 sqlite3_vtab base; |
| 125 }; |
| 126 |
| 127 struct FsdirCsr { |
| 128 sqlite3_vtab_cursor base; |
| 129 char *zDir; /* Buffer containing directory scanned */ |
| 130 DIR *pDir; /* Open directory */ |
| 131 sqlite3_int64 iRowid; |
| 132 struct DIRENT entry; /* Current entry */ |
| 133 }; |
| 134 |
| 135 /* |
| 136 ** This function is the implementation of both the xConnect and xCreate |
| 137 ** methods of the fsdir virtual table. |
| 138 ** |
| 139 ** The argv[] array contains the following: |
| 140 ** |
| 141 ** argv[0] -> module name ("fs") |
| 142 ** argv[1] -> database name |
| 143 ** argv[2] -> table name |
| 144 ** argv[...] -> other module argument fields. |
| 145 */ |
| 146 static int fsdirConnect( |
| 147 sqlite3 *db, |
| 148 void *pAux, |
| 149 int argc, const char *const*argv, |
| 150 sqlite3_vtab **ppVtab, |
| 151 char **pzErr |
| 152 ){ |
| 153 FsdirVtab *pTab; |
| 154 |
| 155 if( argc!=3 ){ |
| 156 *pzErr = sqlite3_mprintf("wrong number of arguments"); |
| 157 return SQLITE_ERROR; |
| 158 } |
| 159 |
| 160 pTab = (FsdirVtab *)sqlite3_malloc(sizeof(FsdirVtab)); |
| 161 if( !pTab ) return SQLITE_NOMEM; |
| 162 memset(pTab, 0, sizeof(FsdirVtab)); |
| 163 |
| 164 *ppVtab = &pTab->base; |
| 165 sqlite3_declare_vtab(db, "CREATE TABLE xyz(dir, name);"); |
| 166 |
| 167 return SQLITE_OK; |
| 168 } |
| 169 |
| 170 /* |
| 171 ** xDestroy/xDisconnect implementation. |
| 172 */ |
| 173 static int fsdirDisconnect(sqlite3_vtab *pVtab){ |
| 174 sqlite3_free(pVtab); |
| 175 return SQLITE_OK; |
| 176 } |
| 177 |
| 178 /* |
| 179 ** xBestIndex implementation. The only constraint supported is: |
| 180 ** |
| 181 ** (dir = ?) |
| 182 */ |
| 183 static int fsdirBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){ |
| 184 int ii; |
| 185 |
| 186 pIdxInfo->estimatedCost = 1000000000.0; |
| 187 |
| 188 for(ii=0; ii<pIdxInfo->nConstraint; ii++){ |
| 189 struct sqlite3_index_constraint const *p = &pIdxInfo->aConstraint[ii]; |
| 190 if( p->iColumn==0 && p->usable && p->op==SQLITE_INDEX_CONSTRAINT_EQ ){ |
| 191 struct sqlite3_index_constraint_usage *pUsage; |
| 192 pUsage = &pIdxInfo->aConstraintUsage[ii]; |
| 193 pUsage->omit = 1; |
| 194 pUsage->argvIndex = 1; |
| 195 pIdxInfo->idxNum = 1; |
| 196 pIdxInfo->estimatedCost = 1.0; |
| 197 break; |
| 198 } |
| 199 } |
| 200 |
| 201 return SQLITE_OK; |
| 202 } |
| 203 |
| 204 /* |
| 205 ** xOpen implementation. |
| 206 ** |
| 207 ** Open a new fsdir cursor. |
| 208 */ |
| 209 static int fsdirOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){ |
| 210 FsdirCsr *pCur; |
| 211 /* Allocate an extra 256 bytes because it is undefined how big dirent.d_name |
| 212 ** is and we need enough space. Linux provides plenty already, but |
| 213 ** Solaris only provides one byte. */ |
| 214 pCur = (FsdirCsr*)sqlite3_malloc(sizeof(FsdirCsr)+256); |
| 215 if( pCur==0 ) return SQLITE_NOMEM; |
| 216 memset(pCur, 0, sizeof(FsdirCsr)); |
| 217 *ppCursor = &pCur->base; |
| 218 return SQLITE_OK; |
| 219 } |
| 220 |
| 221 /* |
| 222 ** Close a fsdir cursor. |
| 223 */ |
| 224 static int fsdirClose(sqlite3_vtab_cursor *cur){ |
| 225 FsdirCsr *pCur = (FsdirCsr*)cur; |
| 226 if( pCur->pDir ) closedir(pCur->pDir); |
| 227 sqlite3_free(pCur->zDir); |
| 228 sqlite3_free(pCur); |
| 229 return SQLITE_OK; |
| 230 } |
| 231 |
| 232 /* |
| 233 ** Skip the cursor to the next entry. |
| 234 */ |
| 235 static int fsdirNext(sqlite3_vtab_cursor *cur){ |
| 236 FsdirCsr *pCsr = (FsdirCsr*)cur; |
| 237 |
| 238 if( pCsr->pDir ){ |
| 239 struct DIRENT *pRes = 0; |
| 240 #if defined(__MINGW_H) |
| 241 pRes = readdir(pCsr->pDir); |
| 242 if( pRes!=0 ){ |
| 243 memcpy(&pCsr->entry, pRes, sizeof(struct DIRENT)); |
| 244 } |
| 245 #else |
| 246 readdir_r(pCsr->pDir, &pCsr->entry, &pRes); |
| 247 #endif |
| 248 if( pRes==0 ){ |
| 249 closedir(pCsr->pDir); |
| 250 pCsr->pDir = 0; |
| 251 } |
| 252 pCsr->iRowid++; |
| 253 } |
| 254 |
| 255 return SQLITE_OK; |
| 256 } |
| 257 |
| 258 /* |
| 259 ** xFilter method implementation. |
| 260 */ |
| 261 static int fsdirFilter( |
| 262 sqlite3_vtab_cursor *pVtabCursor, |
| 263 int idxNum, const char *idxStr, |
| 264 int argc, sqlite3_value **argv |
| 265 ){ |
| 266 FsdirCsr *pCsr = (FsdirCsr*)pVtabCursor; |
| 267 const char *zDir; |
| 268 int nDir; |
| 269 |
| 270 |
| 271 if( idxNum!=1 || argc!=1 ){ |
| 272 return SQLITE_ERROR; |
| 273 } |
| 274 |
| 275 pCsr->iRowid = 0; |
| 276 sqlite3_free(pCsr->zDir); |
| 277 if( pCsr->pDir ){ |
| 278 closedir(pCsr->pDir); |
| 279 pCsr->pDir = 0; |
| 280 } |
| 281 |
| 282 zDir = (const char*)sqlite3_value_text(argv[0]); |
| 283 nDir = sqlite3_value_bytes(argv[0]); |
| 284 pCsr->zDir = sqlite3_malloc(nDir+1); |
| 285 if( pCsr->zDir==0 ) return SQLITE_NOMEM; |
| 286 memcpy(pCsr->zDir, zDir, nDir+1); |
| 287 |
| 288 pCsr->pDir = opendir(pCsr->zDir); |
| 289 return fsdirNext(pVtabCursor); |
| 290 } |
| 291 |
| 292 /* |
| 293 ** xEof method implementation. |
| 294 */ |
| 295 static int fsdirEof(sqlite3_vtab_cursor *cur){ |
| 296 FsdirCsr *pCsr = (FsdirCsr*)cur; |
| 297 return pCsr->pDir==0; |
| 298 } |
| 299 |
| 300 /* |
| 301 ** xColumn method implementation. |
| 302 */ |
| 303 static int fsdirColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){ |
| 304 FsdirCsr *pCsr = (FsdirCsr*)cur; |
| 305 switch( i ){ |
| 306 case 0: /* dir */ |
| 307 sqlite3_result_text(ctx, pCsr->zDir, -1, SQLITE_STATIC); |
| 308 break; |
| 309 |
| 310 case 1: /* name */ |
| 311 sqlite3_result_text(ctx, pCsr->entry.d_name, -1, SQLITE_TRANSIENT); |
| 312 break; |
| 313 |
| 314 default: |
| 315 assert( 0 ); |
| 316 } |
| 317 |
| 318 return SQLITE_OK; |
| 319 } |
| 320 |
| 321 /* |
| 322 ** xRowid method implementation. |
| 323 */ |
| 324 static int fsdirRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){ |
| 325 FsdirCsr *pCsr = (FsdirCsr*)cur; |
| 326 *pRowid = pCsr->iRowid; |
| 327 return SQLITE_OK; |
| 328 } |
| 329 /* |
| 330 ** End of fsdir implementation. |
| 331 *************************************************************************/ |
| 332 |
| 333 /************************************************************************* |
| 334 ** Start of fstree implementation. |
| 335 */ |
| 336 typedef struct FstreeVtab FstreeVtab; |
| 337 typedef struct FstreeCsr FstreeCsr; |
| 338 struct FstreeVtab { |
| 339 sqlite3_vtab base; |
| 340 sqlite3 *db; |
| 341 }; |
| 342 |
| 343 struct FstreeCsr { |
| 344 sqlite3_vtab_cursor base; |
| 345 sqlite3_stmt *pStmt; /* Statement to list paths */ |
| 346 int fd; /* File descriptor open on current path */ |
| 347 }; |
| 348 |
| 349 /* |
| 350 ** This function is the implementation of both the xConnect and xCreate |
| 351 ** methods of the fstree virtual table. |
| 352 ** |
| 353 ** The argv[] array contains the following: |
| 354 ** |
| 355 ** argv[0] -> module name ("fs") |
| 356 ** argv[1] -> database name |
| 357 ** argv[2] -> table name |
| 358 ** argv[...] -> other module argument fields. |
| 359 */ |
| 360 static int fstreeConnect( |
| 361 sqlite3 *db, |
| 362 void *pAux, |
| 363 int argc, const char *const*argv, |
| 364 sqlite3_vtab **ppVtab, |
| 365 char **pzErr |
| 366 ){ |
| 367 FstreeVtab *pTab; |
| 368 |
| 369 if( argc!=3 ){ |
| 370 *pzErr = sqlite3_mprintf("wrong number of arguments"); |
| 371 return SQLITE_ERROR; |
| 372 } |
| 373 |
| 374 pTab = (FstreeVtab *)sqlite3_malloc(sizeof(FstreeVtab)); |
| 375 if( !pTab ) return SQLITE_NOMEM; |
| 376 memset(pTab, 0, sizeof(FstreeVtab)); |
| 377 pTab->db = db; |
| 378 |
| 379 *ppVtab = &pTab->base; |
| 380 sqlite3_declare_vtab(db, "CREATE TABLE xyz(path, size, data);"); |
| 381 |
| 382 return SQLITE_OK; |
| 383 } |
| 384 |
| 385 /* |
| 386 ** xDestroy/xDisconnect implementation. |
| 387 */ |
| 388 static int fstreeDisconnect(sqlite3_vtab *pVtab){ |
| 389 sqlite3_free(pVtab); |
| 390 return SQLITE_OK; |
| 391 } |
| 392 |
| 393 /* |
| 394 ** xBestIndex implementation. The only constraint supported is: |
| 395 ** |
| 396 ** (dir = ?) |
| 397 */ |
| 398 static int fstreeBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){ |
| 399 int ii; |
| 400 |
| 401 for(ii=0; ii<pIdxInfo->nConstraint; ii++){ |
| 402 struct sqlite3_index_constraint const *p = &pIdxInfo->aConstraint[ii]; |
| 403 if( p->iColumn==0 && p->usable && ( |
| 404 p->op==SQLITE_INDEX_CONSTRAINT_GLOB |
| 405 || p->op==SQLITE_INDEX_CONSTRAINT_LIKE |
| 406 || p->op==SQLITE_INDEX_CONSTRAINT_EQ |
| 407 )){ |
| 408 struct sqlite3_index_constraint_usage *pUsage; |
| 409 pUsage = &pIdxInfo->aConstraintUsage[ii]; |
| 410 pIdxInfo->idxNum = p->op; |
| 411 pUsage->argvIndex = 1; |
| 412 pIdxInfo->estimatedCost = 100000.0; |
| 413 return SQLITE_OK; |
| 414 } |
| 415 } |
| 416 |
| 417 pIdxInfo->estimatedCost = 1000000000.0; |
| 418 return SQLITE_OK; |
| 419 } |
| 420 |
| 421 /* |
| 422 ** xOpen implementation. |
| 423 ** |
| 424 ** Open a new fstree cursor. |
| 425 */ |
| 426 static int fstreeOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){ |
| 427 FstreeCsr *pCur; |
| 428 pCur = (FstreeCsr*)sqlite3_malloc(sizeof(FstreeCsr)); |
| 429 if( pCur==0 ) return SQLITE_NOMEM; |
| 430 memset(pCur, 0, sizeof(FstreeCsr)); |
| 431 pCur->fd = -1; |
| 432 *ppCursor = &pCur->base; |
| 433 return SQLITE_OK; |
| 434 } |
| 435 |
| 436 static void fstreeCloseFd(FstreeCsr *pCsr){ |
| 437 if( pCsr->fd>=0 ){ |
| 438 close(pCsr->fd); |
| 439 pCsr->fd = -1; |
| 440 } |
| 441 } |
| 442 |
| 443 /* |
| 444 ** Close a fstree cursor. |
| 445 */ |
| 446 static int fstreeClose(sqlite3_vtab_cursor *cur){ |
| 447 FstreeCsr *pCsr = (FstreeCsr*)cur; |
| 448 sqlite3_finalize(pCsr->pStmt); |
| 449 fstreeCloseFd(pCsr); |
| 450 sqlite3_free(pCsr); |
| 451 return SQLITE_OK; |
| 452 } |
| 453 |
| 454 /* |
| 455 ** Skip the cursor to the next entry. |
| 456 */ |
| 457 static int fstreeNext(sqlite3_vtab_cursor *cur){ |
| 458 FstreeCsr *pCsr = (FstreeCsr*)cur; |
| 459 int rc; |
| 460 |
| 461 fstreeCloseFd(pCsr); |
| 462 rc = sqlite3_step(pCsr->pStmt); |
| 463 if( rc!=SQLITE_ROW ){ |
| 464 rc = sqlite3_finalize(pCsr->pStmt); |
| 465 pCsr->pStmt = 0; |
| 466 }else{ |
| 467 rc = SQLITE_OK; |
| 468 pCsr->fd = open((const char*)sqlite3_column_text(pCsr->pStmt, 0), O_RDONLY); |
| 469 } |
| 470 |
| 471 return rc; |
| 472 } |
| 473 |
| 474 /* |
| 475 ** xFilter method implementation. |
| 476 */ |
| 477 static int fstreeFilter( |
| 478 sqlite3_vtab_cursor *pVtabCursor, |
| 479 int idxNum, const char *idxStr, |
| 480 int argc, sqlite3_value **argv |
| 481 ){ |
| 482 FstreeCsr *pCsr = (FstreeCsr*)pVtabCursor; |
| 483 FstreeVtab *pTab = (FstreeVtab*)(pCsr->base.pVtab); |
| 484 int rc; |
| 485 const char *zSql = |
| 486 "WITH r(d) AS (" |
| 487 " SELECT CASE WHEN dir=?2 THEN ?3 ELSE dir END || '/' || name " |
| 488 " FROM fsdir WHERE dir=?1 AND name NOT LIKE '.%'" |
| 489 " UNION ALL" |
| 490 " SELECT dir || '/' || name FROM r, fsdir WHERE dir=d AND name NOT LIKE '.%'" |
| 491 ") SELECT d FROM r;"; |
| 492 |
| 493 char *zRoot; |
| 494 int nRoot; |
| 495 char *zPrefix; |
| 496 int nPrefix; |
| 497 const char *zDir; |
| 498 int nDir; |
| 499 char aWild[2] = { '\0', '\0' }; |
| 500 |
| 501 #if SQLITE_OS_WIN |
| 502 const char *zDrive = windirent_getenv("fstreeDrive"); |
| 503 if( zDrive==0 ){ |
| 504 zDrive = windirent_getenv("SystemDrive"); |
| 505 } |
| 506 zRoot = sqlite3_mprintf("%s%c", zDrive, '/'); |
| 507 nRoot = sqlite3Strlen30(zRoot); |
| 508 zPrefix = sqlite3_mprintf("%s", zDrive); |
| 509 nPrefix = sqlite3Strlen30(zPrefix); |
| 510 #else |
| 511 zRoot = "/"; |
| 512 nRoot = 1; |
| 513 zPrefix = ""; |
| 514 nPrefix = 0; |
| 515 #endif |
| 516 |
| 517 zDir = zRoot; |
| 518 nDir = nRoot; |
| 519 |
| 520 fstreeCloseFd(pCsr); |
| 521 sqlite3_finalize(pCsr->pStmt); |
| 522 pCsr->pStmt = 0; |
| 523 rc = sqlite3_prepare_v2(pTab->db, zSql, -1, &pCsr->pStmt, 0); |
| 524 if( rc!=SQLITE_OK ) return rc; |
| 525 |
| 526 if( idxNum ){ |
| 527 const char *zQuery = (const char*)sqlite3_value_text(argv[0]); |
| 528 switch( idxNum ){ |
| 529 case SQLITE_INDEX_CONSTRAINT_GLOB: |
| 530 aWild[0] = '*'; |
| 531 aWild[1] = '?'; |
| 532 break; |
| 533 case SQLITE_INDEX_CONSTRAINT_LIKE: |
| 534 aWild[0] = '_'; |
| 535 aWild[1] = '%'; |
| 536 break; |
| 537 } |
| 538 |
| 539 if( sqlite3_strnicmp(zQuery, zPrefix, nPrefix)==0 ){ |
| 540 int i; |
| 541 for(i=nPrefix; zQuery[i]; i++){ |
| 542 if( zQuery[i]==aWild[0] || zQuery[i]==aWild[1] ) break; |
| 543 if( zQuery[i]=='/' ) nDir = i; |
| 544 } |
| 545 zDir = zQuery; |
| 546 } |
| 547 } |
| 548 |
| 549 sqlite3_bind_text(pCsr->pStmt, 1, zDir, nDir, SQLITE_TRANSIENT); |
| 550 sqlite3_bind_text(pCsr->pStmt, 2, zRoot, nRoot, SQLITE_TRANSIENT); |
| 551 sqlite3_bind_text(pCsr->pStmt, 3, zPrefix, nPrefix, SQLITE_TRANSIENT); |
| 552 |
| 553 #if SQLITE_OS_WIN |
| 554 sqlite3_free(zPrefix); |
| 555 sqlite3_free(zRoot); |
| 556 #endif |
| 557 |
| 558 return fstreeNext(pVtabCursor); |
| 559 } |
| 560 |
| 561 /* |
| 562 ** xEof method implementation. |
| 563 */ |
| 564 static int fstreeEof(sqlite3_vtab_cursor *cur){ |
| 565 FstreeCsr *pCsr = (FstreeCsr*)cur; |
| 566 return pCsr->pStmt==0; |
| 567 } |
| 568 |
| 569 /* |
| 570 ** xColumn method implementation. |
| 571 */ |
| 572 static int fstreeColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){ |
| 573 FstreeCsr *pCsr = (FstreeCsr*)cur; |
| 574 if( i==0 ){ /* path */ |
| 575 sqlite3_result_value(ctx, sqlite3_column_value(pCsr->pStmt, 0)); |
| 576 }else{ |
| 577 struct stat sBuf; |
| 578 fstat(pCsr->fd, &sBuf); |
| 579 |
| 580 if( S_ISREG(sBuf.st_mode) ){ |
| 581 if( i==1 ){ |
| 582 sqlite3_result_int64(ctx, sBuf.st_size); |
| 583 }else{ |
| 584 int nRead; |
| 585 char *aBuf = sqlite3_malloc(sBuf.st_mode+1); |
| 586 if( !aBuf ) return SQLITE_NOMEM; |
| 587 nRead = read(pCsr->fd, aBuf, sBuf.st_mode); |
| 588 if( nRead!=sBuf.st_mode ){ |
| 589 return SQLITE_IOERR; |
| 590 } |
| 591 sqlite3_result_blob(ctx, aBuf, nRead, SQLITE_TRANSIENT); |
| 592 sqlite3_free(aBuf); |
| 593 } |
| 594 } |
| 595 } |
| 596 |
| 597 return SQLITE_OK; |
| 598 } |
| 599 |
| 600 /* |
| 601 ** xRowid method implementation. |
| 602 */ |
| 603 static int fstreeRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){ |
| 604 *pRowid = 0; |
| 605 return SQLITE_OK; |
| 606 } |
| 607 /* |
| 608 ** End of fstree implementation. |
| 609 *************************************************************************/ |
| 610 |
| 611 |
| 612 |
| 613 |
| 614 /* |
| 615 ** This function is the implementation of both the xConnect and xCreate |
| 616 ** methods of the fs virtual table. |
| 617 ** |
| 618 ** The argv[] array contains the following: |
| 619 ** |
| 620 ** argv[0] -> module name ("fs") |
| 621 ** argv[1] -> database name |
| 622 ** argv[2] -> table name |
| 623 ** argv[...] -> other module argument fields. |
| 624 */ |
| 625 static int fsConnect( |
| 626 sqlite3 *db, |
| 627 void *pAux, |
| 628 int argc, const char *const*argv, |
| 629 sqlite3_vtab **ppVtab, |
| 630 char **pzErr |
| 631 ){ |
| 632 fs_vtab *pVtab; |
| 633 int nByte; |
| 634 const char *zTbl; |
| 635 const char *zDb = argv[1]; |
| 636 |
| 637 if( argc!=4 ){ |
| 638 *pzErr = sqlite3_mprintf("wrong number of arguments"); |
| 639 return SQLITE_ERROR; |
| 640 } |
| 641 zTbl = argv[3]; |
| 642 |
| 643 nByte = sizeof(fs_vtab) + (int)strlen(zTbl) + 1 + (int)strlen(zDb) + 1; |
| 644 pVtab = (fs_vtab *)sqlite3MallocZero( nByte ); |
| 645 if( !pVtab ) return SQLITE_NOMEM; |
| 646 |
| 647 pVtab->zTbl = (char *)&pVtab[1]; |
| 648 pVtab->zDb = &pVtab->zTbl[strlen(zTbl)+1]; |
| 649 pVtab->db = db; |
| 650 memcpy(pVtab->zTbl, zTbl, strlen(zTbl)); |
| 651 memcpy(pVtab->zDb, zDb, strlen(zDb)); |
| 652 *ppVtab = &pVtab->base; |
| 653 sqlite3_declare_vtab(db, "CREATE TABLE x(path TEXT, data TEXT)"); |
| 654 |
| 655 return SQLITE_OK; |
| 656 } |
| 657 /* Note that for this virtual table, the xCreate and xConnect |
| 658 ** methods are identical. */ |
| 659 |
| 660 static int fsDisconnect(sqlite3_vtab *pVtab){ |
| 661 sqlite3_free(pVtab); |
| 662 return SQLITE_OK; |
| 663 } |
| 664 /* The xDisconnect and xDestroy methods are also the same */ |
| 665 |
| 666 /* |
| 667 ** Open a new fs cursor. |
| 668 */ |
| 669 static int fsOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){ |
| 670 fs_cursor *pCur; |
| 671 pCur = sqlite3MallocZero(sizeof(fs_cursor)); |
| 672 *ppCursor = &pCur->base; |
| 673 return SQLITE_OK; |
| 674 } |
| 675 |
| 676 /* |
| 677 ** Close a fs cursor. |
| 678 */ |
| 679 static int fsClose(sqlite3_vtab_cursor *cur){ |
| 680 fs_cursor *pCur = (fs_cursor *)cur; |
| 681 sqlite3_finalize(pCur->pStmt); |
| 682 sqlite3_free(pCur->zBuf); |
| 683 sqlite3_free(pCur); |
| 684 return SQLITE_OK; |
| 685 } |
| 686 |
| 687 static int fsNext(sqlite3_vtab_cursor *cur){ |
| 688 fs_cursor *pCur = (fs_cursor *)cur; |
| 689 int rc; |
| 690 |
| 691 rc = sqlite3_step(pCur->pStmt); |
| 692 if( rc==SQLITE_ROW || rc==SQLITE_DONE ) rc = SQLITE_OK; |
| 693 |
| 694 return rc; |
| 695 } |
| 696 |
| 697 static int fsFilter( |
| 698 sqlite3_vtab_cursor *pVtabCursor, |
| 699 int idxNum, const char *idxStr, |
| 700 int argc, sqlite3_value **argv |
| 701 ){ |
| 702 int rc; |
| 703 fs_cursor *pCur = (fs_cursor *)pVtabCursor; |
| 704 fs_vtab *p = (fs_vtab *)(pVtabCursor->pVtab); |
| 705 |
| 706 assert( (idxNum==0 && argc==0) || (idxNum==1 && argc==1) ); |
| 707 if( idxNum==1 ){ |
| 708 char *zStmt = sqlite3_mprintf( |
| 709 "SELECT * FROM %Q.%Q WHERE rowid=?", p->zDb, p->zTbl); |
| 710 if( !zStmt ) return SQLITE_NOMEM; |
| 711 rc = sqlite3_prepare_v2(p->db, zStmt, -1, &pCur->pStmt, 0); |
| 712 sqlite3_free(zStmt); |
| 713 if( rc==SQLITE_OK ){ |
| 714 sqlite3_bind_value(pCur->pStmt, 1, argv[0]); |
| 715 } |
| 716 }else{ |
| 717 char *zStmt = sqlite3_mprintf("SELECT * FROM %Q.%Q", p->zDb, p->zTbl); |
| 718 if( !zStmt ) return SQLITE_NOMEM; |
| 719 rc = sqlite3_prepare_v2(p->db, zStmt, -1, &pCur->pStmt, 0); |
| 720 sqlite3_free(zStmt); |
| 721 } |
| 722 |
| 723 if( rc==SQLITE_OK ){ |
| 724 rc = fsNext(pVtabCursor); |
| 725 } |
| 726 return rc; |
| 727 } |
| 728 |
| 729 static int fsColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){ |
| 730 fs_cursor *pCur = (fs_cursor*)cur; |
| 731 |
| 732 assert( i==0 || i==1 || i==2 ); |
| 733 if( i==0 ){ |
| 734 sqlite3_result_value(ctx, sqlite3_column_value(pCur->pStmt, 0)); |
| 735 }else{ |
| 736 const char *zFile = (const char *)sqlite3_column_text(pCur->pStmt, 1); |
| 737 struct stat sbuf; |
| 738 int fd; |
| 739 |
| 740 int n; |
| 741 fd = open(zFile, O_RDONLY); |
| 742 if( fd<0 ) return SQLITE_IOERR; |
| 743 fstat(fd, &sbuf); |
| 744 |
| 745 if( sbuf.st_size>=pCur->nAlloc ){ |
| 746 int nNew = sbuf.st_size*2; |
| 747 char *zNew; |
| 748 if( nNew<1024 ) nNew = 1024; |
| 749 |
| 750 zNew = sqlite3Realloc(pCur->zBuf, nNew); |
| 751 if( zNew==0 ){ |
| 752 close(fd); |
| 753 return SQLITE_NOMEM; |
| 754 } |
| 755 pCur->zBuf = zNew; |
| 756 pCur->nAlloc = nNew; |
| 757 } |
| 758 |
| 759 n = (int)read(fd, pCur->zBuf, sbuf.st_size); |
| 760 close(fd); |
| 761 if( n!=sbuf.st_size ) return SQLITE_ERROR; |
| 762 pCur->nBuf = sbuf.st_size; |
| 763 pCur->zBuf[pCur->nBuf] = '\0'; |
| 764 |
| 765 sqlite3_result_text(ctx, pCur->zBuf, -1, SQLITE_TRANSIENT); |
| 766 } |
| 767 return SQLITE_OK; |
| 768 } |
| 769 |
| 770 static int fsRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){ |
| 771 fs_cursor *pCur = (fs_cursor*)cur; |
| 772 *pRowid = sqlite3_column_int64(pCur->pStmt, 0); |
| 773 return SQLITE_OK; |
| 774 } |
| 775 |
| 776 static int fsEof(sqlite3_vtab_cursor *cur){ |
| 777 fs_cursor *pCur = (fs_cursor*)cur; |
| 778 return (sqlite3_data_count(pCur->pStmt)==0); |
| 779 } |
| 780 |
| 781 static int fsBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){ |
| 782 int ii; |
| 783 |
| 784 for(ii=0; ii<pIdxInfo->nConstraint; ii++){ |
| 785 struct sqlite3_index_constraint const *pCons = &pIdxInfo->aConstraint[ii]; |
| 786 if( pCons->iColumn<0 && pCons->usable |
| 787 && pCons->op==SQLITE_INDEX_CONSTRAINT_EQ ){ |
| 788 struct sqlite3_index_constraint_usage *pUsage; |
| 789 pUsage = &pIdxInfo->aConstraintUsage[ii]; |
| 790 pUsage->omit = 0; |
| 791 pUsage->argvIndex = 1; |
| 792 pIdxInfo->idxNum = 1; |
| 793 pIdxInfo->estimatedCost = 1.0; |
| 794 break; |
| 795 } |
| 796 } |
| 797 |
| 798 return SQLITE_OK; |
| 799 } |
| 800 |
| 801 /* |
| 802 ** A virtual table module that provides read-only access to a |
| 803 ** Tcl global variable namespace. |
| 804 */ |
| 805 static sqlite3_module fsModule = { |
| 806 0, /* iVersion */ |
| 807 fsConnect, |
| 808 fsConnect, |
| 809 fsBestIndex, |
| 810 fsDisconnect, |
| 811 fsDisconnect, |
| 812 fsOpen, /* xOpen - open a cursor */ |
| 813 fsClose, /* xClose - close a cursor */ |
| 814 fsFilter, /* xFilter - configure scan constraints */ |
| 815 fsNext, /* xNext - advance a cursor */ |
| 816 fsEof, /* xEof - check for end of scan */ |
| 817 fsColumn, /* xColumn - read data */ |
| 818 fsRowid, /* xRowid - read data */ |
| 819 0, /* xUpdate */ |
| 820 0, /* xBegin */ |
| 821 0, /* xSync */ |
| 822 0, /* xCommit */ |
| 823 0, /* xRollback */ |
| 824 0, /* xFindMethod */ |
| 825 0, /* xRename */ |
| 826 }; |
| 827 |
| 828 static sqlite3_module fsdirModule = { |
| 829 0, /* iVersion */ |
| 830 fsdirConnect, /* xCreate */ |
| 831 fsdirConnect, /* xConnect */ |
| 832 fsdirBestIndex, /* xBestIndex */ |
| 833 fsdirDisconnect, /* xDisconnect */ |
| 834 fsdirDisconnect, /* xDestroy */ |
| 835 fsdirOpen, /* xOpen - open a cursor */ |
| 836 fsdirClose, /* xClose - close a cursor */ |
| 837 fsdirFilter, /* xFilter - configure scan constraints */ |
| 838 fsdirNext, /* xNext - advance a cursor */ |
| 839 fsdirEof, /* xEof - check for end of scan */ |
| 840 fsdirColumn, /* xColumn - read data */ |
| 841 fsdirRowid, /* xRowid - read data */ |
| 842 0, /* xUpdate */ |
| 843 0, /* xBegin */ |
| 844 0, /* xSync */ |
| 845 0, /* xCommit */ |
| 846 0, /* xRollback */ |
| 847 0, /* xFindMethod */ |
| 848 0, /* xRename */ |
| 849 }; |
| 850 |
| 851 static sqlite3_module fstreeModule = { |
| 852 0, /* iVersion */ |
| 853 fstreeConnect, /* xCreate */ |
| 854 fstreeConnect, /* xConnect */ |
| 855 fstreeBestIndex, /* xBestIndex */ |
| 856 fstreeDisconnect, /* xDisconnect */ |
| 857 fstreeDisconnect, /* xDestroy */ |
| 858 fstreeOpen, /* xOpen - open a cursor */ |
| 859 fstreeClose, /* xClose - close a cursor */ |
| 860 fstreeFilter, /* xFilter - configure scan constraints */ |
| 861 fstreeNext, /* xNext - advance a cursor */ |
| 862 fstreeEof, /* xEof - check for end of scan */ |
| 863 fstreeColumn, /* xColumn - read data */ |
| 864 fstreeRowid, /* xRowid - read data */ |
| 865 0, /* xUpdate */ |
| 866 0, /* xBegin */ |
| 867 0, /* xSync */ |
| 868 0, /* xCommit */ |
| 869 0, /* xRollback */ |
| 870 0, /* xFindMethod */ |
| 871 0, /* xRename */ |
| 872 }; |
| 873 |
| 874 /* |
| 875 ** Decode a pointer to an sqlite3 object. |
| 876 */ |
| 877 extern int getDbPointer(Tcl_Interp *interp, const char *zA, sqlite3 **ppDb); |
| 878 |
| 879 /* |
| 880 ** Register the echo virtual table module. |
| 881 */ |
| 882 static int SQLITE_TCLAPI register_fs_module( |
| 883 ClientData clientData, /* Pointer to sqlite3_enable_XXX function */ |
| 884 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ |
| 885 int objc, /* Number of arguments */ |
| 886 Tcl_Obj *CONST objv[] /* Command arguments */ |
| 887 ){ |
| 888 sqlite3 *db; |
| 889 if( objc!=2 ){ |
| 890 Tcl_WrongNumArgs(interp, 1, objv, "DB"); |
| 891 return TCL_ERROR; |
| 892 } |
| 893 if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; |
| 894 #ifndef SQLITE_OMIT_VIRTUALTABLE |
| 895 sqlite3_create_module(db, "fs", &fsModule, (void *)interp); |
| 896 sqlite3_create_module(db, "fsdir", &fsdirModule, 0); |
| 897 sqlite3_create_module(db, "fstree", &fstreeModule, 0); |
| 898 #endif |
| 899 return TCL_OK; |
| 900 } |
| 901 |
| 902 #endif |
| 903 |
| 904 |
| 905 /* |
| 906 ** Register commands with the TCL interpreter. |
| 907 */ |
| 908 int Sqlitetestfs_Init(Tcl_Interp *interp){ |
| 909 #ifndef SQLITE_OMIT_VIRTUALTABLE |
| 910 static struct { |
| 911 char *zName; |
| 912 Tcl_ObjCmdProc *xProc; |
| 913 void *clientData; |
| 914 } aObjCmd[] = { |
| 915 { "register_fs_module", register_fs_module, 0 }, |
| 916 }; |
| 917 int i; |
| 918 for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){ |
| 919 Tcl_CreateObjCommand(interp, aObjCmd[i].zName, |
| 920 aObjCmd[i].xProc, aObjCmd[i].clientData, 0); |
| 921 } |
| 922 #endif |
| 923 return TCL_OK; |
| 924 } |
OLD | NEW |