78 lines
1.5 KiB
C++
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;
|
|
}
|
|
|