amxmodx/dlls/sqlite/sqlitepp/SqliteDriver.cpp

#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;
}