amxmodx/dlls/sqlite/sqlitepp/SqliteDriver.cpp

88 lines
1.9 KiB
C++
Raw Normal View History

// vim: set ts=4 sw=4 tw=99 noet:
//
// AMX Mod X, based on AMX Mod by Aleksander Naszko ("OLO").
// Copyright (C) The AMX Mod X Development Team.
//
// This software is licensed under the GNU General Public License, version 3 or higher.
// Additional exceptions apply. For full license details, see LICENSE.txt or visit:
// https://alliedmods.net/amxmodx-license
//
// SQLite Module
//
2014-08-08 14:44:37 +04:00
#include "amxxmodule.h"
2006-06-03 23:52:21 +04:00
#include "SqliteHeaders.h"
#include "SqliteDriver.h"
#include "SqliteDatabase.h"
#if defined WIN32
#define WINDOWS_LEAN_AND_MEAN
#include <windows.h>
#else
#include <unistd.h>
2006-06-03 23:52:21 +04:00
#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__ || defined __APPLE__
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);
}
2006-06-03 23:52:21 +04:00
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)
{
2014-08-08 14:44:37 +04:00
UTIL_Format(error, maxlength, "%s", sqlite3_errmsg(pSql));
2006-06-03 23:52:21 +04:00
}
sqlite3_close(pSql);
return NULL;
} else {
sqlite3_busy_handler(pSql, busy_handler, NULL);
2006-06-03 23:52:21 +04:00
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;
}