amxmodx/dlls/sqlite/sqlitepp/SqliteDriver.cpp

78 lines
1.5 KiB
C++

#include <string.h>
#include <stdio.h>
#include "SqliteHeaders.h"
#include "SqliteDriver.h"
#include "SqliteDatabase.h"
#if defined WIN32
#define snprintf _snprintf
#define strncasecmp strnicmp
#define WINDOWS_LEAN_AND_MEAN
#include <windows.h>
#else
#include <unistd.h>
#endif
using namespace SourceMod;
bool SqliteDriver::IsCompatDriver(const char *namestr)
{
return (strncasecmp(namestr, "sqlite", 5) == 0);
}
const char *SqliteDriver::NameString()
{
return "sqlite";
}
int busy_handler(void *unused1, int unused2)
{
#if defined __linux__
usleep(100000);
#else
Sleep(100);
#endif
return 1;
}
IDatabase *SqliteDriver::Connect2(DatabaseInfo *info, int *errcode, char *error, size_t maxlength)
{
return Connect(info, errcode, error, maxlength);
}
IDatabase *SqliteDriver::Connect(DatabaseInfo *info, int *errcode, char *error, size_t maxlength)
{
sqlite3 *pSql;
int err = sqlite3_open(info->database, &pSql);
if (err != SQLITE_OK)
{
if (errcode)
{
*errcode = sqlite3_errcode(pSql);
}
if (error)
{
snprintf(error, maxlength, "%s", sqlite3_errmsg(pSql));
}
sqlite3_close(pSql);
return NULL;
} else {
sqlite3_busy_handler(pSql, busy_handler, NULL);
SqliteDatabase *pDb = new SqliteDatabase(pSql, this);
return static_cast<IDatabase *>(pDb);
}
}
int SqliteDriver::QuoteString(const char *str, char buffer[], size_t maxlen, size_t *newsize)
{
char *res = sqlite3_snprintf(static_cast<int>(maxlen), buffer, "%q", str);
if (res != NULL && newsize != NULL)
{
*newsize = strlen(buffer);
}
return 0;
}