OLD | NEW |
(Empty) | |
| 1 /* |
| 2 ** This utility program looks at an SQLite database and determines whether |
| 3 ** or not it is locked, the kind of lock, and who is holding this lock. |
| 4 ** |
| 5 ** This only works on unix when the posix advisory locking method is used |
| 6 ** (which is the default on unix) and when the PENDING_BYTE is in its |
| 7 ** usual place. |
| 8 */ |
| 9 #include <sys/types.h> |
| 10 #include <sys/stat.h> |
| 11 #include <unistd.h> |
| 12 #include <fcntl.h> |
| 13 #include <string.h> |
| 14 #include <stdio.h> |
| 15 #include <stdlib.h> |
| 16 #include <errno.h> |
| 17 |
| 18 static void usage(const char *argv0){ |
| 19 fprintf(stderr, "Usage: %s database\n", argv0); |
| 20 exit(1); |
| 21 } |
| 22 |
| 23 /* Check for a conflicting lock. If one is found, print an this |
| 24 ** on standard output using the format string given and return 1. |
| 25 ** If there are no conflicting locks, return 0. |
| 26 */ |
| 27 static int isLocked( |
| 28 int h, /* File descriptor to check */ |
| 29 int type, /* F_RDLCK or F_WRLCK */ |
| 30 unsigned int iOfst, /* First byte of the lock */ |
| 31 unsigned int iCnt, /* Number of bytes in the lock range */ |
| 32 const char *zType /* Type of lock */ |
| 33 ){ |
| 34 struct flock lk; |
| 35 |
| 36 memset(&lk, 0, sizeof(lk)); |
| 37 lk.l_type = type; |
| 38 lk.l_whence = SEEK_SET; |
| 39 lk.l_start = iOfst; |
| 40 lk.l_len = iCnt; |
| 41 if( fcntl(h, F_GETLK, &lk)==(-1) ){ |
| 42 fprintf(stderr, "fcntl(%d) failed: errno=%d\n", h, errno); |
| 43 exit(1); |
| 44 } |
| 45 if( lk.l_type==F_UNLCK ) return 0; |
| 46 printf("%s lock held by %d\n", zType, (int)lk.l_pid); |
| 47 return 1; |
| 48 } |
| 49 |
| 50 /* |
| 51 ** Location of locking bytes in the database file |
| 52 */ |
| 53 #define PENDING_BYTE (0x40000000) |
| 54 #define RESERVED_BYTE (PENDING_BYTE+1) |
| 55 #define SHARED_FIRST (PENDING_BYTE+2) |
| 56 #define SHARED_SIZE 510 |
| 57 |
| 58 /* |
| 59 ** Lock locations for shared-memory locks used by WAL mode. |
| 60 */ |
| 61 #define SHM_BASE 120 |
| 62 #define SHM_WRITE SHM_BASE |
| 63 #define SHM_CHECKPOINT (SHM_BASE+1) |
| 64 #define SHM_RECOVER (SHM_BASE+2) |
| 65 #define SHM_READ_FIRST (SHM_BASE+3) |
| 66 #define SHM_READ_SIZE 5 |
| 67 |
| 68 |
| 69 int main(int argc, char **argv){ |
| 70 int hDb; /* File descriptor for the open database file */ |
| 71 int hShm; /* File descriptor for WAL shared-memory file */ |
| 72 char *zShm; /* Name of the shared-memory file for WAL mode */ |
| 73 ssize_t got; /* Bytes read from header */ |
| 74 int isWal; /* True if in WAL mode */ |
| 75 int nName; /* Length of filename */ |
| 76 unsigned char aHdr[100]; /* Database header */ |
| 77 int nLock = 0; /* Number of locks held */ |
| 78 int i; /* Loop counter */ |
| 79 |
| 80 if( argc!=2 ) usage(argv[0]); |
| 81 hDb = open(argv[1], O_RDONLY, 0); |
| 82 if( hDb<0 ){ |
| 83 fprintf(stderr, "cannot open %s\n", argv[1]); |
| 84 return 1; |
| 85 } |
| 86 |
| 87 /* Make sure we are dealing with an database file */ |
| 88 got = read(hDb, aHdr, 100); |
| 89 if( got!=100 || memcmp(aHdr, "SQLite format 3",16)!=0 ){ |
| 90 fprintf(stderr, "not an SQLite database: %s\n", argv[1]); |
| 91 exit(1); |
| 92 } |
| 93 |
| 94 /* First check for an exclusive lock */ |
| 95 if( isLocked(hDb, F_RDLCK, SHARED_FIRST, SHARED_SIZE, "EXCLUSIVE") ){ |
| 96 return 0; |
| 97 } |
| 98 isWal = aHdr[18]==2; |
| 99 if( isWal==0 ){ |
| 100 /* Rollback mode */ |
| 101 if( isLocked(hDb, F_RDLCK, PENDING_BYTE, 1, "PENDING") ) return 0; |
| 102 if( isLocked(hDb, F_RDLCK, RESERVED_BYTE, 1, "RESERVED") ) return 0; |
| 103 if( isLocked(hDb, F_WRLCK, SHARED_FIRST, SHARED_SIZE, "SHARED") ){ |
| 104 return 0; |
| 105 } |
| 106 }else{ |
| 107 /* WAL mode */ |
| 108 nName = (int)strlen(argv[1]); |
| 109 zShm = malloc( nName + 100 ); |
| 110 if( zShm==0 ){ |
| 111 fprintf(stderr, "out of memory\n"); |
| 112 exit(1); |
| 113 } |
| 114 memcpy(zShm, argv[1], nName); |
| 115 memcpy(&zShm[nName], "-shm", 5); |
| 116 hShm = open(zShm, O_RDONLY, 0); |
| 117 if( hShm<0 ){ |
| 118 fprintf(stderr, "cannot open %s\n", zShm); |
| 119 return 1; |
| 120 } |
| 121 if( isLocked(hShm, F_RDLCK, SHM_RECOVER, 1, "WAL-RECOVERY") ){ |
| 122 return 0; |
| 123 } |
| 124 nLock += isLocked(hShm, F_RDLCK, SHM_CHECKPOINT, 1, "WAL-CHECKPOINT"); |
| 125 nLock += isLocked(hShm, F_RDLCK, SHM_WRITE, 1, "WAL-WRITE"); |
| 126 for(i=0; i<SHM_READ_SIZE; i++){ |
| 127 nLock += isLocked(hShm, F_WRLCK, SHM_READ_FIRST+i, 1, "WAL-READ"); |
| 128 } |
| 129 } |
| 130 if( nLock==0 ){ |
| 131 printf("file is not locked\n"); |
| 132 } |
| 133 return 0; |
| 134 } |
OLD | NEW |