12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529 |
- #import "FMDatabase.h"
- #import <unistd.h>
- #import <objc/runtime.h>
- #if FMDB_SQLITE_STANDALONE
- #import <sqlite3/sqlite3.h>
- #else
- #import <sqlite3.h>
- #endif
- // MARK: - FMDatabase Private Extension
- NS_ASSUME_NONNULL_BEGIN
- @interface FMDatabase () {
- void* _db;
- BOOL _isExecutingStatement;
- NSTimeInterval _startBusyRetryTime;
-
- NSMutableSet *_openResultSets;
- NSMutableSet *_openFunctions;
-
- NSDateFormatter *_dateFormat;
- }
- - (FMResultSet * _Nullable)executeQuery:(NSString *)sql withArgumentsInArray:(NSArray * _Nullable)arrayArgs orDictionary:(NSDictionary * _Nullable)dictionaryArgs orVAList:(va_list)args shouldBind:(BOOL)shouldBind;
- - (BOOL)executeUpdate:(NSString *)sql error:(NSError * _Nullable __autoreleasing *)outErr withArgumentsInArray:(NSArray * _Nullable)arrayArgs orDictionary:(NSDictionary * _Nullable)dictionaryArgs orVAList:(va_list)args;
- @end
- // MARK: - FMResultSet Private Extension
- @interface FMResultSet ()
- - (int)internalStepWithError:(NSError * _Nullable __autoreleasing *)outErr;
- + (instancetype)resultSetWithStatement:(FMStatement *)statement usingParentDatabase:(FMDatabase*)aDB shouldAutoClose:(BOOL)shouldAutoClose;
- @end
- NS_ASSUME_NONNULL_END
- // MARK: - FMDatabase
- @implementation FMDatabase
- // Because these two properties have all of their accessor methods implemented,
- // we have to synthesize them to get the corresponding ivars. The rest of the
- // properties have their ivars synthesized automatically for us.
- @synthesize shouldCacheStatements = _shouldCacheStatements;
- @synthesize maxBusyRetryTimeInterval = _maxBusyRetryTimeInterval;
- #pragma mark FMDatabase instantiation and deallocation
- + (instancetype)databaseWithPath:(NSString *)aPath {
- return FMDBReturnAutoreleased([[self alloc] initWithPath:aPath]);
- }
- + (instancetype)databaseWithURL:(NSURL *)url {
- return FMDBReturnAutoreleased([[self alloc] initWithURL:url]);
- }
- - (instancetype)init {
- return [self initWithPath:nil];
- }
- - (instancetype)initWithURL:(NSURL *)url {
- return [self initWithPath:url.path];
- }
- - (instancetype)initWithPath:(NSString *)path {
-
- assert(sqlite3_threadsafe()); // whoa there big boy- gotta make sure sqlite it happy with what we're going to do.
-
- self = [super init];
-
- if (self) {
- _databasePath = [path copy];
- _openResultSets = [[NSMutableSet alloc] init];
- _db = nil;
- _logsErrors = YES;
- _crashOnErrors = NO;
- _maxBusyRetryTimeInterval = 2;
- _isOpen = NO;
- }
-
- return self;
- }
- #if ! __has_feature(objc_arc)
- - (void)finalize {
- [self close];
- [super finalize];
- }
- #endif
- - (void)dealloc {
- [self close];
- FMDBRelease(_openResultSets);
- FMDBRelease(_cachedStatements);
- FMDBRelease(_dateFormat);
- FMDBRelease(_databasePath);
- FMDBRelease(_openFunctions);
-
- #if ! __has_feature(objc_arc)
- [super dealloc];
- #endif
- }
- - (NSURL *)databaseURL {
- return _databasePath ? [NSURL fileURLWithPath:_databasePath] : nil;
- }
- + (NSString*)FMDBUserVersion {
- return @"2.7.8";
- }
- + (SInt32)FMDBVersion {
-
- // we go through these hoops so that we only have to change the version number in a single spot.
- static dispatch_once_t once;
- static SInt32 FMDBVersionVal = 0;
-
- dispatch_once(&once, ^{
- NSString *prodVersion = [self FMDBUserVersion];
-
- while ([[prodVersion componentsSeparatedByString:@"."] count] < 3) {
- prodVersion = [prodVersion stringByAppendingString:@".0"];
- }
- NSArray *components = [prodVersion componentsSeparatedByString:@"."];
- for (NSUInteger i = 0; i < 3; i++) {
- SInt32 component = [components[i] intValue];
- if (component > 15) {
- NSLog(@"FMDBVersion is invalid: Please use FMDBUserVersion instead.");
- component = 15;
- }
- FMDBVersionVal = FMDBVersionVal << 4 | component;
- }
- });
-
- return FMDBVersionVal;
- }
- #pragma mark SQLite information
- + (NSString*)sqliteLibVersion {
- return [NSString stringWithFormat:@"%s", sqlite3_libversion()];
- }
- + (BOOL)isSQLiteThreadSafe {
- // make sure to read the sqlite headers on this guy!
- return sqlite3_threadsafe() != 0;
- }
- - (void*)sqliteHandle {
- return _db;
- }
- - (const char*)sqlitePath {
-
- if (!_databasePath) {
- return ":memory:";
- }
-
- if ([_databasePath length] == 0) {
- return ""; // this creates a temporary database (it's an sqlite thing).
- }
-
- return [_databasePath fileSystemRepresentation];
-
- }
- - (int)limitFor:(int)type value:(int)newLimit {
- return sqlite3_limit(_db, type, newLimit);
- }
- #pragma mark Open and close database
- - (BOOL)open {
- if (_isOpen) {
- return YES;
- }
-
- // if we previously tried to open and it failed, make sure to close it before we try again
-
- if (_db) {
- [self close];
- }
-
- // now open database
- int err = sqlite3_open([self sqlitePath], (sqlite3**)&_db );
- if(err != SQLITE_OK) {
- NSLog(@"error opening!: %d", err);
- return NO;
- }
-
- if (_maxBusyRetryTimeInterval > 0.0) {
- // set the handler
- [self setMaxBusyRetryTimeInterval:_maxBusyRetryTimeInterval];
- }
-
- _isOpen = YES;
-
- return YES;
- }
- - (BOOL)openWithFlags:(int)flags {
- return [self openWithFlags:flags vfs:nil];
- }
- - (BOOL)openWithFlags:(int)flags vfs:(NSString *)vfsName {
- #if SQLITE_VERSION_NUMBER >= 3005000
- if (_isOpen) {
- return YES;
- }
-
- // if we previously tried to open and it failed, make sure to close it before we try again
-
- if (_db) {
- [self close];
- }
-
- // now open database
-
- int err = sqlite3_open_v2([self sqlitePath], (sqlite3**)&_db, flags, [vfsName UTF8String]);
- if(err != SQLITE_OK) {
- NSLog(@"error opening!: %d", err);
- return NO;
- }
-
- if (_maxBusyRetryTimeInterval > 0.0) {
- // set the handler
- [self setMaxBusyRetryTimeInterval:_maxBusyRetryTimeInterval];
- }
-
- _isOpen = YES;
-
- return YES;
- #else
- NSLog(@"openWithFlags requires SQLite 3.5");
- return NO;
- #endif
- }
- - (BOOL)close {
-
- [self clearCachedStatements];
- [self closeOpenResultSets];
-
- if (!_db) {
- return YES;
- }
-
- int rc;
- BOOL retry;
- BOOL triedFinalizingOpenStatements = NO;
-
- do {
- retry = NO;
- rc = sqlite3_close(_db);
- if (SQLITE_BUSY == rc || SQLITE_LOCKED == rc) {
- if (!triedFinalizingOpenStatements) {
- triedFinalizingOpenStatements = YES;
- sqlite3_stmt *pStmt;
- while ((pStmt = sqlite3_next_stmt(_db, nil)) !=0) {
- NSLog(@"Closing leaked statement");
- sqlite3_finalize(pStmt);
- pStmt = 0x00;
- retry = YES;
- }
- }
- }
- else if (SQLITE_OK != rc) {
- NSLog(@"error closing!: %d", rc);
- }
- }
- while (retry);
-
- _db = nil;
- _isOpen = false;
-
- return YES;
- }
- #pragma mark Busy handler routines
- // NOTE: appledoc seems to choke on this function for some reason;
- // so when generating documentation, you might want to ignore the
- // .m files so that it only documents the public interfaces outlined
- // in the .h files.
- //
- // This is a known appledoc bug that it has problems with C functions
- // within a class implementation, but for some reason, only this
- // C function causes problems; the rest don't. Anyway, ignoring the .m
- // files with appledoc will prevent this problem from occurring.
- static int FMDBDatabaseBusyHandler(void *f, int count) {
- FMDatabase *self = (__bridge FMDatabase*)f;
-
- if (count == 0) {
- self->_startBusyRetryTime = [NSDate timeIntervalSinceReferenceDate];
- return 1;
- }
-
- NSTimeInterval delta = [NSDate timeIntervalSinceReferenceDate] - (self->_startBusyRetryTime);
-
- if (delta < [self maxBusyRetryTimeInterval]) {
- int requestedSleepInMillseconds = (int) arc4random_uniform(50) + 50;
- int actualSleepInMilliseconds = sqlite3_sleep(requestedSleepInMillseconds);
- if (actualSleepInMilliseconds != requestedSleepInMillseconds) {
- NSLog(@"WARNING: Requested sleep of %i milliseconds, but SQLite returned %i. Maybe SQLite wasn't built with HAVE_USLEEP=1?", requestedSleepInMillseconds, actualSleepInMilliseconds);
- }
- return 1;
- }
-
- return 0;
- }
- - (void)setMaxBusyRetryTimeInterval:(NSTimeInterval)timeout {
-
- _maxBusyRetryTimeInterval = timeout;
-
- if (!_db) {
- return;
- }
-
- if (timeout > 0) {
- sqlite3_busy_handler(_db, &FMDBDatabaseBusyHandler, (__bridge void *)(self));
- }
- else {
- // turn it off otherwise
- sqlite3_busy_handler(_db, nil, nil);
- }
- }
- - (NSTimeInterval)maxBusyRetryTimeInterval {
- return _maxBusyRetryTimeInterval;
- }
- // we no longer make busyRetryTimeout public
- // but for folks who don't bother noticing that the interface to FMDatabase changed,
- // we'll still implement the method so they don't get suprise crashes
- - (int)busyRetryTimeout {
- NSLog(@"%s:%d", __FUNCTION__, __LINE__);
- NSLog(@"FMDB: busyRetryTimeout no longer works, please use maxBusyRetryTimeInterval");
- return -1;
- }
- - (void)setBusyRetryTimeout:(int)i {
- #pragma unused(i)
- NSLog(@"%s:%d", __FUNCTION__, __LINE__);
- NSLog(@"FMDB: setBusyRetryTimeout does nothing, please use setMaxBusyRetryTimeInterval:");
- }
- #pragma mark Result set functions
- - (BOOL)hasOpenResultSets {
- return [_openResultSets count] > 0;
- }
- - (void)closeOpenResultSets {
-
- //Copy the set so we don't get mutation errors
- NSSet *openSetCopy = FMDBReturnAutoreleased([_openResultSets copy]);
- for (NSValue *rsInWrappedInATastyValueMeal in openSetCopy) {
- FMResultSet *rs = (FMResultSet *)[rsInWrappedInATastyValueMeal pointerValue];
-
- [rs setParentDB:nil];
- [rs close];
-
- [_openResultSets removeObject:rsInWrappedInATastyValueMeal];
- }
- }
- - (void)resultSetDidClose:(FMResultSet *)resultSet {
- NSValue *setValue = [NSValue valueWithNonretainedObject:resultSet];
-
- [_openResultSets removeObject:setValue];
- }
- #pragma mark Cached statements
- - (void)clearCachedStatements {
-
- for (NSMutableSet *statements in [_cachedStatements objectEnumerator]) {
- for (FMStatement *statement in [statements allObjects]) {
- [statement close];
- }
- }
-
- [_cachedStatements removeAllObjects];
- }
- - (FMStatement*)cachedStatementForQuery:(NSString*)query {
-
- NSMutableSet* statements = [_cachedStatements objectForKey:query];
-
- return [[statements objectsPassingTest:^BOOL(FMStatement* statement, BOOL *stop) {
-
- *stop = ![statement inUse];
- return *stop;
-
- }] anyObject];
- }
- - (void)setCachedStatement:(FMStatement*)statement forQuery:(NSString*)query {
- NSParameterAssert(query);
- if (!query) {
- NSLog(@"API misuse, -[FMDatabase setCachedStatement:forQuery:] query must not be nil");
- return;
- }
-
- query = [query copy]; // in case we got handed in a mutable string...
- [statement setQuery:query];
-
- NSMutableSet* statements = [_cachedStatements objectForKey:query];
- if (!statements) {
- statements = [NSMutableSet set];
- }
-
- [statements addObject:statement];
-
- [_cachedStatements setObject:statements forKey:query];
-
- FMDBRelease(query);
- }
- #pragma mark Key routines
- - (BOOL)rekey:(NSString*)key {
- NSData *keyData = [NSData dataWithBytes:(void *)[key UTF8String] length:(NSUInteger)strlen([key UTF8String])];
-
- return [self rekeyWithData:keyData];
- }
- - (BOOL)rekeyWithData:(NSData *)keyData {
- #ifdef SQLITE_HAS_CODEC
- if (!keyData) {
- return NO;
- }
-
- int rc = sqlite3_rekey(_db, [keyData bytes], (int)[keyData length]);
-
- if (rc != SQLITE_OK) {
- NSLog(@"error on rekey: %d", rc);
- NSLog(@"%@", [self lastErrorMessage]);
- }
-
- return (rc == SQLITE_OK);
- #else
- #pragma unused(keyData)
- return NO;
- #endif
- }
- - (BOOL)setKey:(NSString*)key {
- NSData *keyData = [NSData dataWithBytes:[key UTF8String] length:(NSUInteger)strlen([key UTF8String])];
-
- return [self setKeyWithData:keyData];
- }
- - (BOOL)setKeyWithData:(NSData *)keyData {
- #ifdef SQLITE_HAS_CODEC
- if (!keyData) {
- return NO;
- }
-
- int rc = sqlite3_key(_db, [keyData bytes], (int)[keyData length]);
-
- return (rc == SQLITE_OK);
- #else
- #pragma unused(keyData)
- return NO;
- #endif
- }
- #pragma mark Date routines
- + (NSDateFormatter *)storeableDateFormat:(NSString *)format {
-
- NSDateFormatter *result = FMDBReturnAutoreleased([[NSDateFormatter alloc] init]);
- result.dateFormat = format;
- result.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
- result.locale = FMDBReturnAutoreleased([[NSLocale alloc] initWithLocaleIdentifier:@"en_US"]);
- return result;
- }
- - (BOOL)hasDateFormatter {
- return _dateFormat != nil;
- }
- - (void)setDateFormat:(NSDateFormatter *)format {
- FMDBAutorelease(_dateFormat);
- _dateFormat = FMDBReturnRetained(format);
- }
- - (NSDate *)dateFromString:(NSString *)s {
- return [_dateFormat dateFromString:s];
- }
- - (NSString *)stringFromDate:(NSDate *)date {
- return [_dateFormat stringFromDate:date];
- }
- #pragma mark State of database
- - (BOOL)goodConnection {
-
- if (!_isOpen) {
- return NO;
- }
-
- #ifdef SQLCIPHER_CRYPTO
- // Starting with Xcode8 / iOS 10 we check to make sure we really are linked with
- // SQLCipher because there is no longer a linker error if we accidently link
- // with unencrypted sqlite library.
- //
- // https://discuss.zetetic.net/t/important-advisory-sqlcipher-with-xcode-8-and-new-sdks/1688
-
- FMResultSet *rs = [self executeQuery:@"PRAGMA cipher_version"];
- if ([rs next]) {
- NSLog(@"SQLCipher version: %@", rs.resultDictionary[@"cipher_version"]);
-
- [rs close];
- return YES;
- }
- #else
- FMResultSet *rs = [self executeQuery:@"select name from sqlite_master where type='table'"];
-
- if (rs) {
- [rs close];
- return YES;
- }
- #endif
-
- return NO;
- }
- - (void)warnInUse {
- NSLog(@"The FMDatabase %@ is currently in use.", self);
-
- #ifndef NS_BLOCK_ASSERTIONS
- if (_crashOnErrors) {
- NSAssert(false, @"The FMDatabase %@ is currently in use.", self);
- abort();
- }
- #endif
- }
- - (BOOL)databaseExists {
-
- if (!_isOpen) {
-
- NSLog(@"The FMDatabase %@ is not open.", self);
-
- #ifndef NS_BLOCK_ASSERTIONS
- if (_crashOnErrors) {
- NSAssert(false, @"The FMDatabase %@ is not open.", self);
- abort();
- }
- #endif
-
- return NO;
- }
-
- return YES;
- }
- #pragma mark Error routines
- - (NSString *)lastErrorMessage {
- return [NSString stringWithUTF8String:sqlite3_errmsg(_db)];
- }
- - (BOOL)hadError {
- int lastErrCode = [self lastErrorCode];
-
- return (lastErrCode > SQLITE_OK && lastErrCode < SQLITE_ROW);
- }
- - (int)lastErrorCode {
- return sqlite3_errcode(_db);
- }
- - (int)lastExtendedErrorCode {
- return sqlite3_extended_errcode(_db);
- }
- - (NSError*)errorWithMessage:(NSString *)message {
- NSDictionary* errorMessage = [NSDictionary dictionaryWithObject:message forKey:NSLocalizedDescriptionKey];
-
- return [NSError errorWithDomain:@"FMDatabase" code:sqlite3_errcode(_db) userInfo:errorMessage];
- }
- - (NSError*)lastError {
- return [self errorWithMessage:[self lastErrorMessage]];
- }
- #pragma mark Update information routines
- - (sqlite_int64)lastInsertRowId {
-
- if (_isExecutingStatement) {
- [self warnInUse];
- return NO;
- }
-
- _isExecutingStatement = YES;
-
- sqlite_int64 ret = sqlite3_last_insert_rowid(_db);
-
- _isExecutingStatement = NO;
-
- return ret;
- }
- - (int)changes {
- if (_isExecutingStatement) {
- [self warnInUse];
- return 0;
- }
-
- _isExecutingStatement = YES;
-
- int ret = sqlite3_changes(_db);
-
- _isExecutingStatement = NO;
-
- return ret;
- }
- #pragma mark SQL manipulation
- - (int)bindObject:(id)obj toColumn:(int)idx inStatement:(sqlite3_stmt*)pStmt {
-
- if ((!obj) || ((NSNull *)obj == [NSNull null])) {
- return sqlite3_bind_null(pStmt, idx);
- }
-
- // FIXME - someday check the return codes on these binds.
- else if ([obj isKindOfClass:[NSData class]]) {
- const void *bytes = [obj bytes];
- if (!bytes) {
- // it's an empty NSData object, aka [NSData data].
- // Don't pass a NULL pointer, or sqlite will bind a SQL null instead of a blob.
- bytes = "";
- }
- return sqlite3_bind_blob(pStmt, idx, bytes, (int)[obj length], SQLITE_TRANSIENT);
- }
- else if ([obj isKindOfClass:[NSDate class]]) {
- if (self.hasDateFormatter)
- return sqlite3_bind_text(pStmt, idx, [[self stringFromDate:obj] UTF8String], -1, SQLITE_TRANSIENT);
- else
- return sqlite3_bind_double(pStmt, idx, [obj timeIntervalSince1970]);
- }
- else if ([obj isKindOfClass:[NSNumber class]]) {
-
- if (strcmp([obj objCType], @encode(char)) == 0) {
- return sqlite3_bind_int(pStmt, idx, [obj charValue]);
- }
- else if (strcmp([obj objCType], @encode(unsigned char)) == 0) {
- return sqlite3_bind_int(pStmt, idx, [obj unsignedCharValue]);
- }
- else if (strcmp([obj objCType], @encode(short)) == 0) {
- return sqlite3_bind_int(pStmt, idx, [obj shortValue]);
- }
- else if (strcmp([obj objCType], @encode(unsigned short)) == 0) {
- return sqlite3_bind_int(pStmt, idx, [obj unsignedShortValue]);
- }
- else if (strcmp([obj objCType], @encode(int)) == 0) {
- return sqlite3_bind_int(pStmt, idx, [obj intValue]);
- }
- else if (strcmp([obj objCType], @encode(unsigned int)) == 0) {
- return sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedIntValue]);
- }
- else if (strcmp([obj objCType], @encode(long)) == 0) {
- return sqlite3_bind_int64(pStmt, idx, [obj longValue]);
- }
- else if (strcmp([obj objCType], @encode(unsigned long)) == 0) {
- return sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedLongValue]);
- }
- else if (strcmp([obj objCType], @encode(long long)) == 0) {
- return sqlite3_bind_int64(pStmt, idx, [obj longLongValue]);
- }
- else if (strcmp([obj objCType], @encode(unsigned long long)) == 0) {
- return sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedLongLongValue]);
- }
- else if (strcmp([obj objCType], @encode(float)) == 0) {
- return sqlite3_bind_double(pStmt, idx, [obj floatValue]);
- }
- else if (strcmp([obj objCType], @encode(double)) == 0) {
- return sqlite3_bind_double(pStmt, idx, [obj doubleValue]);
- }
- else if (strcmp([obj objCType], @encode(BOOL)) == 0) {
- return sqlite3_bind_int(pStmt, idx, ([obj boolValue] ? 1 : 0));
- }
- else {
- return sqlite3_bind_text(pStmt, idx, [[obj description] UTF8String], -1, SQLITE_TRANSIENT);
- }
- }
- return sqlite3_bind_text(pStmt, idx, [[obj description] UTF8String], -1, SQLITE_TRANSIENT);
- }
- - (void)extractSQL:(NSString *)sql argumentsList:(va_list)args intoString:(NSMutableString *)cleanedSQL arguments:(NSMutableArray *)arguments {
-
- NSUInteger length = [sql length];
- unichar last = '\0';
- for (NSUInteger i = 0; i < length; ++i) {
- id arg = nil;
- unichar current = [sql characterAtIndex:i];
- unichar add = current;
- if (last == '%') {
- switch (current) {
- case '@':
- arg = va_arg(args, id);
- break;
- case 'c':
- // warning: second argument to 'va_arg' is of promotable type 'char'; this va_arg has undefined behavior because arguments will be promoted to 'int'
- arg = [NSString stringWithFormat:@"%c", va_arg(args, int)];
- break;
- case 's':
- arg = [NSString stringWithUTF8String:va_arg(args, char*)];
- break;
- case 'd':
- case 'D':
- case 'i':
- arg = [NSNumber numberWithInt:va_arg(args, int)];
- break;
- case 'u':
- case 'U':
- arg = [NSNumber numberWithUnsignedInt:va_arg(args, unsigned int)];
- break;
- case 'h':
- i++;
- if (i < length && [sql characterAtIndex:i] == 'i') {
- // warning: second argument to 'va_arg' is of promotable type 'short'; this va_arg has undefined behavior because arguments will be promoted to 'int'
- arg = [NSNumber numberWithShort:(short)(va_arg(args, int))];
- }
- else if (i < length && [sql characterAtIndex:i] == 'u') {
- // warning: second argument to 'va_arg' is of promotable type 'unsigned short'; this va_arg has undefined behavior because arguments will be promoted to 'int'
- arg = [NSNumber numberWithUnsignedShort:(unsigned short)(va_arg(args, uint))];
- }
- else {
- i--;
- }
- break;
- case 'q':
- i++;
- if (i < length && [sql characterAtIndex:i] == 'i') {
- arg = [NSNumber numberWithLongLong:va_arg(args, long long)];
- }
- else if (i < length && [sql characterAtIndex:i] == 'u') {
- arg = [NSNumber numberWithUnsignedLongLong:va_arg(args, unsigned long long)];
- }
- else {
- i--;
- }
- break;
- case 'f':
- arg = [NSNumber numberWithDouble:va_arg(args, double)];
- break;
- case 'g':
- // warning: second argument to 'va_arg' is of promotable type 'float'; this va_arg has undefined behavior because arguments will be promoted to 'double'
- arg = [NSNumber numberWithFloat:(float)(va_arg(args, double))];
- break;
- case 'l':
- i++;
- if (i < length) {
- unichar next = [sql characterAtIndex:i];
- if (next == 'l') {
- i++;
- if (i < length && [sql characterAtIndex:i] == 'd') {
- //%lld
- arg = [NSNumber numberWithLongLong:va_arg(args, long long)];
- }
- else if (i < length && [sql characterAtIndex:i] == 'u') {
- //%llu
- arg = [NSNumber numberWithUnsignedLongLong:va_arg(args, unsigned long long)];
- }
- else {
- i--;
- }
- }
- else if (next == 'd') {
- //%ld
- arg = [NSNumber numberWithLong:va_arg(args, long)];
- }
- else if (next == 'u') {
- //%lu
- arg = [NSNumber numberWithUnsignedLong:va_arg(args, unsigned long)];
- }
- else {
- i--;
- }
- }
- else {
- i--;
- }
- break;
- default:
- // something else that we can't interpret. just pass it on through like normal
- break;
- }
- }
- else if (current == '%') {
- // percent sign; skip this character
- add = '\0';
- }
-
- if (arg != nil) {
- [cleanedSQL appendString:@"?"];
- [arguments addObject:arg];
- }
- else if (add == (unichar)'@' && last == (unichar) '%') {
- [cleanedSQL appendFormat:@"NULL"];
- }
- else if (add != '\0') {
- [cleanedSQL appendFormat:@"%C", add];
- }
- last = current;
- }
- }
- #pragma mark Execute queries
- - (FMResultSet *)executeQuery:(NSString *)sql withParameterDictionary:(NSDictionary *)arguments {
- return [self executeQuery:sql withArgumentsInArray:nil orDictionary:arguments orVAList:nil shouldBind:true];
- }
- - (FMResultSet *)executeQuery:(NSString *)sql withArgumentsInArray:(NSArray*)arrayArgs orDictionary:(NSDictionary *)dictionaryArgs orVAList:(va_list)args shouldBind:(BOOL)shouldBind {
- if (![self databaseExists]) {
- return 0x00;
- }
-
- if (_isExecutingStatement) {
- [self warnInUse];
- return 0x00;
- }
-
- _isExecutingStatement = YES;
-
- int rc = 0x00;
- sqlite3_stmt *pStmt = 0x00;
- FMStatement *statement = 0x00;
- FMResultSet *rs = 0x00;
-
- if (_traceExecution && sql) {
- NSLog(@"%@ executeQuery: %@", self, sql);
- }
-
- if (_shouldCacheStatements) {
- statement = [self cachedStatementForQuery:sql];
- pStmt = statement ? [statement statement] : 0x00;
- [statement reset];
- }
-
- if (!pStmt) {
- rc = sqlite3_prepare_v2(_db, [sql UTF8String], -1, &pStmt, 0);
-
- if (SQLITE_OK != rc) {
- if (_logsErrors) {
- NSLog(@"DB Error: %d \"%@\"", [self lastErrorCode], [self lastErrorMessage]);
- NSLog(@"DB Query: %@", sql);
- NSLog(@"DB Path: %@", _databasePath);
- }
-
- if (_crashOnErrors) {
- NSAssert(false, @"DB Error: %d \"%@\"", [self lastErrorCode], [self lastErrorMessage]);
- abort();
- }
-
- sqlite3_finalize(pStmt);
- pStmt = 0x00;
- _isExecutingStatement = NO;
- return nil;
- }
- }
- if (shouldBind) {
- BOOL success = [self bindStatement:pStmt WithArgumentsInArray:arrayArgs orDictionary:dictionaryArgs orVAList:args];
- if (!success) {
- return nil;
- }
- }
- FMDBRetain(statement); // to balance the release below
-
- if (!statement) {
- statement = [[FMStatement alloc] init];
- [statement setStatement:pStmt];
-
- if (_shouldCacheStatements && sql) {
- [self setCachedStatement:statement forQuery:sql];
- }
- }
-
- // the statement gets closed in rs's dealloc or [rs close];
- // we should only autoclose if we're binding automatically when the statement is prepared
- rs = [FMResultSet resultSetWithStatement:statement usingParentDatabase:self shouldAutoClose:shouldBind];
- [rs setQuery:sql];
-
- NSValue *openResultSet = [NSValue valueWithNonretainedObject:rs];
- [_openResultSets addObject:openResultSet];
-
- [statement setUseCount:[statement useCount] + 1];
-
- FMDBRelease(statement);
-
- _isExecutingStatement = NO;
-
- return rs;
- }
- - (BOOL)bindStatement:(sqlite3_stmt *)pStmt WithArgumentsInArray:(NSArray*)arrayArgs orDictionary:(NSDictionary *)dictionaryArgs orVAList:(va_list)args {
- id obj;
- int idx = 0;
- int queryCount = sqlite3_bind_parameter_count(pStmt); // pointed out by Dominic Yu (thanks!)
- // If dictionaryArgs is passed in, that means we are using sqlite's named parameter support
- if (dictionaryArgs) {
- for (NSString *dictionaryKey in [dictionaryArgs allKeys]) {
- // Prefix the key with a colon.
- NSString *parameterName = [[NSString alloc] initWithFormat:@":%@", dictionaryKey];
- if (_traceExecution) {
- NSLog(@"%@ = %@", parameterName, [dictionaryArgs objectForKey:dictionaryKey]);
- }
- // Get the index for the parameter name.
- int namedIdx = sqlite3_bind_parameter_index(pStmt, [parameterName UTF8String]);
- FMDBRelease(parameterName);
- if (namedIdx > 0) {
- // Standard binding from here.
- int rc = [self bindObject:[dictionaryArgs objectForKey:dictionaryKey] toColumn:namedIdx inStatement:pStmt];
- if (rc != SQLITE_OK) {
- NSLog(@"Error: unable to bind (%d, %s", rc, sqlite3_errmsg(_db));
- sqlite3_finalize(pStmt);
- pStmt = 0x00;
- _isExecutingStatement = NO;
- return false;
- }
- // increment the binding count, so our check below works out
- idx++;
- }
- else {
- NSLog(@"Could not find index for %@", dictionaryKey);
- }
- }
- }
- else {
- while (idx < queryCount) {
- if (arrayArgs && idx < (int)[arrayArgs count]) {
- obj = [arrayArgs objectAtIndex:(NSUInteger)idx];
- }
- else if (args) {
- obj = va_arg(args, id);
- }
- else {
- //We ran out of arguments
- break;
- }
- if (_traceExecution) {
- if ([obj isKindOfClass:[NSData class]]) {
- NSLog(@"data: %ld bytes", (unsigned long)[(NSData*)obj length]);
- }
- else {
- NSLog(@"obj: %@", obj);
- }
- }
- idx++;
- int rc = [self bindObject:obj toColumn:idx inStatement:pStmt];
- if (rc != SQLITE_OK) {
- NSLog(@"Error: unable to bind (%d, %s", rc, sqlite3_errmsg(_db));
- sqlite3_finalize(pStmt);
- pStmt = 0x00;
- _isExecutingStatement = NO;
- return false;
- }
- }
- }
- if (idx != queryCount) {
- NSLog(@"Error: the bind count is not correct for the # of variables (executeQuery)");
- sqlite3_finalize(pStmt);
- pStmt = 0x00;
- _isExecutingStatement = NO;
- return false;
- }
- return true;
- }
- - (FMResultSet *)executeQuery:(NSString*)sql, ... {
- va_list args;
- va_start(args, sql);
-
- id result = [self executeQuery:sql withArgumentsInArray:nil orDictionary:nil orVAList:args shouldBind:true];
-
- va_end(args);
- return result;
- }
- - (FMResultSet *)executeQueryWithFormat:(NSString*)format, ... {
- va_list args;
- va_start(args, format);
-
- NSMutableString *sql = [NSMutableString stringWithCapacity:[format length]];
- NSMutableArray *arguments = [NSMutableArray array];
- [self extractSQL:format argumentsList:args intoString:sql arguments:arguments];
-
- va_end(args);
-
- return [self executeQuery:sql withArgumentsInArray:arguments];
- }
- - (FMResultSet *)executeQuery:(NSString *)sql withArgumentsInArray:(NSArray *)arguments {
- return [self executeQuery:sql withArgumentsInArray:arguments orDictionary:nil orVAList:nil shouldBind:true];
- }
- - (FMResultSet *)executeQuery:(NSString *)sql values:(NSArray *)values error:(NSError * __autoreleasing *)error {
- FMResultSet *rs = [self executeQuery:sql withArgumentsInArray:values orDictionary:nil orVAList:nil shouldBind:true];
- if (!rs && error) {
- *error = [self lastError];
- }
- return rs;
- }
- - (FMResultSet *)executeQuery:(NSString*)sql withVAList:(va_list)args {
- return [self executeQuery:sql withArgumentsInArray:nil orDictionary:nil orVAList:args shouldBind:true];
- }
- #pragma mark Execute updates
- - (BOOL)executeUpdate:(NSString*)sql error:(NSError * _Nullable __autoreleasing *)outErr withArgumentsInArray:(NSArray*)arrayArgs orDictionary:(NSDictionary *)dictionaryArgs orVAList:(va_list)args {
- FMResultSet *rs = [self executeQuery:sql withArgumentsInArray:arrayArgs orDictionary:dictionaryArgs orVAList:args shouldBind:true];
- if (!rs) {
- if (outErr) {
- *outErr = [self lastError];
- }
- return false;
- }
- return [rs internalStepWithError:outErr] == SQLITE_DONE;
- }
- - (BOOL)executeUpdate:(NSString*)sql, ... {
- va_list args;
- va_start(args, sql);
-
- BOOL result = [self executeUpdate:sql error:nil withArgumentsInArray:nil orDictionary:nil orVAList:args];
-
- va_end(args);
- return result;
- }
- - (BOOL)executeUpdate:(NSString*)sql withArgumentsInArray:(NSArray *)arguments {
- return [self executeUpdate:sql error:nil withArgumentsInArray:arguments orDictionary:nil orVAList:nil];
- }
- - (BOOL)executeUpdate:(NSString*)sql values:(NSArray *)values error:(NSError * __autoreleasing *)error {
- return [self executeUpdate:sql error:error withArgumentsInArray:values orDictionary:nil orVAList:nil];
- }
- - (BOOL)executeUpdate:(NSString*)sql withParameterDictionary:(NSDictionary *)arguments {
- return [self executeUpdate:sql error:nil withArgumentsInArray:nil orDictionary:arguments orVAList:nil];
- }
- - (BOOL)executeUpdate:(NSString*)sql withVAList:(va_list)args {
- return [self executeUpdate:sql error:nil withArgumentsInArray:nil orDictionary:nil orVAList:args];
- }
- - (BOOL)executeUpdateWithFormat:(NSString*)format, ... {
- va_list args;
- va_start(args, format);
-
- NSMutableString *sql = [NSMutableString stringWithCapacity:[format length]];
- NSMutableArray *arguments = [NSMutableArray array];
-
- [self extractSQL:format argumentsList:args intoString:sql arguments:arguments];
-
- va_end(args);
-
- return [self executeUpdate:sql withArgumentsInArray:arguments];
- }
- int FMDBExecuteBulkSQLCallback(void *theBlockAsVoid, int columns, char **values, char **names); // shhh clang.
- int FMDBExecuteBulkSQLCallback(void *theBlockAsVoid, int columns, char **values, char **names) {
-
- if (!theBlockAsVoid) {
- return SQLITE_OK;
- }
-
- int (^execCallbackBlock)(NSDictionary *resultsDictionary) = (__bridge int (^)(NSDictionary *__strong))(theBlockAsVoid);
-
- NSMutableDictionary *dictionary = [NSMutableDictionary dictionaryWithCapacity:(NSUInteger)columns];
-
- for (NSInteger i = 0; i < columns; i++) {
- NSString *key = [NSString stringWithUTF8String:names[i]];
- id value = values[i] ? [NSString stringWithUTF8String:values[i]] : [NSNull null];
- value = value ? value : [NSNull null];
- [dictionary setObject:value forKey:key];
- }
-
- return execCallbackBlock(dictionary);
- }
- - (BOOL)executeStatements:(NSString *)sql {
- return [self executeStatements:sql withResultBlock:nil];
- }
- - (BOOL)executeStatements:(NSString *)sql withResultBlock:(__attribute__((noescape)) FMDBExecuteStatementsCallbackBlock)block {
-
- int rc;
- char *errmsg = nil;
-
- rc = sqlite3_exec([self sqliteHandle], [sql UTF8String], block ? FMDBExecuteBulkSQLCallback : nil, (__bridge void *)(block), &errmsg);
-
- if (errmsg && [self logsErrors]) {
- NSLog(@"Error inserting batch: %s", errmsg);
- }
- if (errmsg) {
- sqlite3_free(errmsg);
- }
-
- return (rc == SQLITE_OK);
- }
- - (BOOL)executeUpdate:(NSString*)sql withErrorAndBindings:(NSError * _Nullable __autoreleasing *)outErr, ... {
-
- va_list args;
- va_start(args, outErr);
-
- BOOL result = [self executeUpdate:sql error:outErr withArgumentsInArray:nil orDictionary:nil orVAList:args];
-
- va_end(args);
- return result;
- }
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wdeprecated-implementations"
- - (BOOL)update:(NSString*)sql withErrorAndBindings:(NSError * _Nullable __autoreleasing *)outErr, ... {
- va_list args;
- va_start(args, outErr);
-
- BOOL result = [self executeUpdate:sql error:outErr withArgumentsInArray:nil orDictionary:nil orVAList:args];
-
- va_end(args);
- return result;
- }
- #pragma clang diagnostic pop
- #pragma mark Prepare
- - (FMResultSet *)prepare:(NSString *)sql {
- return [self executeQuery:sql withArgumentsInArray:nil orDictionary:nil orVAList:nil shouldBind:false];
- }
- #pragma mark Transactions
- - (BOOL)rollback {
- BOOL b = [self executeUpdate:@"rollback transaction"];
-
- if (b) {
- _isInTransaction = NO;
- }
-
- return b;
- }
- - (BOOL)commit {
- BOOL b = [self executeUpdate:@"commit transaction"];
-
- if (b) {
- _isInTransaction = NO;
- }
-
- return b;
- }
- - (BOOL)beginTransaction {
-
- BOOL b = [self executeUpdate:@"begin exclusive transaction"];
- if (b) {
- _isInTransaction = YES;
- }
-
- return b;
- }
- - (BOOL)beginDeferredTransaction {
-
- BOOL b = [self executeUpdate:@"begin deferred transaction"];
- if (b) {
- _isInTransaction = YES;
- }
-
- return b;
- }
- - (BOOL)beginImmediateTransaction {
-
- BOOL b = [self executeUpdate:@"begin immediate transaction"];
- if (b) {
- _isInTransaction = YES;
- }
-
- return b;
- }
- - (BOOL)beginExclusiveTransaction {
-
- BOOL b = [self executeUpdate:@"begin exclusive transaction"];
- if (b) {
- _isInTransaction = YES;
- }
-
- return b;
- }
- - (BOOL)inTransaction {
- return _isInTransaction;
- }
- - (BOOL)interrupt
- {
- if (_db) {
- sqlite3_interrupt([self sqliteHandle]);
- return YES;
- }
- return NO;
- }
- static NSString *FMDBEscapeSavePointName(NSString *savepointName) {
- return [savepointName stringByReplacingOccurrencesOfString:@"'" withString:@"''"];
- }
- - (BOOL)startSavePointWithName:(NSString*)name error:(NSError * _Nullable __autoreleasing *)outErr {
- #if SQLITE_VERSION_NUMBER >= 3007000
- NSParameterAssert(name);
-
- NSString *sql = [NSString stringWithFormat:@"savepoint '%@';", FMDBEscapeSavePointName(name)];
-
- return [self executeUpdate:sql error:outErr withArgumentsInArray:nil orDictionary:nil orVAList:nil];
- #else
- NSString *errorMessage = NSLocalizedStringFromTable(@"Save point functions require SQLite 3.7", @"FMDB", nil);
- if (self.logsErrors) NSLog(@"%@", errorMessage);
- return NO;
- #endif
- }
- - (BOOL)releaseSavePointWithName:(NSString*)name error:(NSError * _Nullable __autoreleasing *)outErr {
- #if SQLITE_VERSION_NUMBER >= 3007000
- NSParameterAssert(name);
-
- NSString *sql = [NSString stringWithFormat:@"release savepoint '%@';", FMDBEscapeSavePointName(name)];
- return [self executeUpdate:sql error:outErr withArgumentsInArray:nil orDictionary:nil orVAList:nil];
- #else
- NSString *errorMessage = NSLocalizedStringFromTable(@"Save point functions require SQLite 3.7", @"FMDB", nil);
- if (self.logsErrors) NSLog(@"%@", errorMessage);
- return NO;
- #endif
- }
- - (BOOL)rollbackToSavePointWithName:(NSString*)name error:(NSError * _Nullable __autoreleasing *)outErr {
- #if SQLITE_VERSION_NUMBER >= 3007000
- NSParameterAssert(name);
-
- NSString *sql = [NSString stringWithFormat:@"rollback transaction to savepoint '%@';", FMDBEscapeSavePointName(name)];
- return [self executeUpdate:sql error:outErr withArgumentsInArray:nil orDictionary:nil orVAList:nil];
- #else
- NSString *errorMessage = NSLocalizedStringFromTable(@"Save point functions require SQLite 3.7", @"FMDB", nil);
- if (self.logsErrors) NSLog(@"%@", errorMessage);
- return NO;
- #endif
- }
- - (NSError*)inSavePoint:(__attribute__((noescape)) void (^)(BOOL *rollback))block {
- #if SQLITE_VERSION_NUMBER >= 3007000
- static unsigned long savePointIdx = 0;
-
- NSString *name = [NSString stringWithFormat:@"dbSavePoint%ld", savePointIdx++];
-
- BOOL shouldRollback = NO;
-
- NSError *err = 0x00;
-
- if (![self startSavePointWithName:name error:&err]) {
- return err;
- }
-
- if (block) {
- block(&shouldRollback);
- }
-
- if (shouldRollback) {
- // We need to rollback and release this savepoint to remove it
- [self rollbackToSavePointWithName:name error:&err];
- }
- [self releaseSavePointWithName:name error:&err];
-
- return err;
- #else
- NSString *errorMessage = NSLocalizedStringFromTable(@"Save point functions require SQLite 3.7", @"FMDB", nil);
- if (self.logsErrors) NSLog(@"%@", errorMessage);
- return [NSError errorWithDomain:@"FMDatabase" code:0 userInfo:@{NSLocalizedDescriptionKey : errorMessage}];
- #endif
- }
- - (BOOL)checkpoint:(FMDBCheckpointMode)checkpointMode error:(NSError * __autoreleasing *)error {
- return [self checkpoint:checkpointMode name:nil logFrameCount:NULL checkpointCount:NULL error:error];
- }
- - (BOOL)checkpoint:(FMDBCheckpointMode)checkpointMode name:(NSString *)name error:(NSError * __autoreleasing *)error {
- return [self checkpoint:checkpointMode name:name logFrameCount:NULL checkpointCount:NULL error:error];
- }
- - (BOOL)checkpoint:(FMDBCheckpointMode)checkpointMode name:(NSString *)name logFrameCount:(int *)logFrameCount checkpointCount:(int *)checkpointCount error:(NSError * __autoreleasing *)error
- {
- const char* dbName = [name UTF8String];
- #if SQLITE_VERSION_NUMBER >= 3007006
- int err = sqlite3_wal_checkpoint_v2(_db, dbName, checkpointMode, logFrameCount, checkpointCount);
- #else
- NSLog(@"sqlite3_wal_checkpoint_v2 unavailable before sqlite 3.7.6. Ignoring checkpoint mode: %d", mode);
- int err = sqlite3_wal_checkpoint(_db, dbName);
- #endif
- if(err != SQLITE_OK) {
- if (error) {
- *error = [self lastError];
- }
- if (self.logsErrors) NSLog(@"%@", [self lastErrorMessage]);
- if (self.crashOnErrors) {
- NSAssert(false, @"%@", [self lastErrorMessage]);
- abort();
- }
- return NO;
- } else {
- return YES;
- }
- }
- #pragma mark Cache statements
- - (BOOL)shouldCacheStatements {
- return _shouldCacheStatements;
- }
- - (void)setShouldCacheStatements:(BOOL)value {
-
- _shouldCacheStatements = value;
-
- if (_shouldCacheStatements && !_cachedStatements) {
- [self setCachedStatements:[NSMutableDictionary dictionary]];
- }
-
- if (!_shouldCacheStatements) {
- [self setCachedStatements:nil];
- }
- }
- #pragma mark Callback function
- void FMDBBlockSQLiteCallBackFunction(sqlite3_context *context, int argc, sqlite3_value **argv); // -Wmissing-prototypes
- void FMDBBlockSQLiteCallBackFunction(sqlite3_context *context, int argc, sqlite3_value **argv) {
- #if ! __has_feature(objc_arc)
- void (^block)(sqlite3_context *context, int argc, sqlite3_value **argv) = (id)sqlite3_user_data(context);
- #else
- void (^block)(sqlite3_context *context, int argc, sqlite3_value **argv) = (__bridge id)sqlite3_user_data(context);
- #endif
- if (block) {
- @autoreleasepool {
- block(context, argc, argv);
- }
- }
- }
- // deprecated because "arguments" parameter is not maximum argument count, but actual argument count.
- - (void)makeFunctionNamed:(NSString *)name maximumArguments:(int)arguments withBlock:(void (^)(void *context, int argc, void **argv))block {
- [self makeFunctionNamed:name arguments:arguments block:block];
- }
- - (void)makeFunctionNamed:(NSString *)name arguments:(int)arguments block:(void (^)(void *context, int argc, void **argv))block {
-
- if (!_openFunctions) {
- _openFunctions = [NSMutableSet new];
- }
-
- id b = FMDBReturnAutoreleased([block copy]);
-
- [_openFunctions addObject:b];
-
- /* I tried adding custom functions to release the block when the connection is destroyed- but they seemed to never be called, so we use _openFunctions to store the values instead. */
- #if ! __has_feature(objc_arc)
- sqlite3_create_function([self sqliteHandle], [name UTF8String], arguments, SQLITE_UTF8, (void*)b, &FMDBBlockSQLiteCallBackFunction, 0x00, 0x00);
- #else
- sqlite3_create_function([self sqliteHandle], [name UTF8String], arguments, SQLITE_UTF8, (__bridge void*)b, &FMDBBlockSQLiteCallBackFunction, 0x00, 0x00);
- #endif
- }
- - (SqliteValueType)valueType:(void *)value {
- return sqlite3_value_type(value);
- }
- - (int)valueInt:(void *)value {
- return sqlite3_value_int(value);
- }
- - (long long)valueLong:(void *)value {
- return sqlite3_value_int64(value);
- }
- - (double)valueDouble:(void *)value {
- return sqlite3_value_double(value);
- }
- - (NSData *)valueData:(void *)value {
- const void *bytes = sqlite3_value_blob(value);
- int length = sqlite3_value_bytes(value);
- return bytes ? [NSData dataWithBytes:bytes length:(NSUInteger)length] : nil;
- }
- - (NSString *)valueString:(void *)value {
- const char *cString = (const char *)sqlite3_value_text(value);
- return cString ? [NSString stringWithUTF8String:cString] : nil;
- }
- - (void)resultNullInContext:(void *)context {
- sqlite3_result_null(context);
- }
- - (void)resultInt:(int) value context:(void *)context {
- sqlite3_result_int(context, value);
- }
- - (void)resultLong:(long long)value context:(void *)context {
- sqlite3_result_int64(context, value);
- }
- - (void)resultDouble:(double)value context:(void *)context {
- sqlite3_result_double(context, value);
- }
- - (void)resultData:(NSData *)data context:(void *)context {
- sqlite3_result_blob(context, data.bytes, (int)data.length, SQLITE_TRANSIENT);
- }
- - (void)resultString:(NSString *)value context:(void *)context {
- sqlite3_result_text(context, [value UTF8String], -1, SQLITE_TRANSIENT);
- }
- - (void)resultError:(NSString *)error context:(void *)context {
- sqlite3_result_error(context, [error UTF8String], -1);
- }
- - (void)resultErrorCode:(int)errorCode context:(void *)context {
- sqlite3_result_error_code(context, errorCode);
- }
- - (void)resultErrorNoMemoryInContext:(void *)context {
- sqlite3_result_error_nomem(context);
- }
- - (void)resultErrorTooBigInContext:(void *)context {
- sqlite3_result_error_toobig(context);
- }
- @end
- // MARK: - FMStatement
- @implementation FMStatement
- #if ! __has_feature(objc_arc)
- - (void)finalize {
- [self close];
- [super finalize];
- }
- #endif
- - (void)dealloc {
- [self close];
- FMDBRelease(_query);
- #if ! __has_feature(objc_arc)
- [super dealloc];
- #endif
- }
- - (void)close {
- if (_statement) {
- sqlite3_finalize(_statement);
- _statement = 0x00;
- }
-
- _inUse = NO;
- }
- - (void)reset {
- if (_statement) {
- sqlite3_reset(_statement);
- }
-
- _inUse = NO;
- }
- - (NSString*)description {
- return [NSString stringWithFormat:@"%@ %ld hit(s) for query %@", [super description], _useCount, _query];
- }
- @end
|