127 lines
2.9 KiB
C++
127 lines
2.9 KiB
C++
// 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
|
|
|
|
//
|
|
// MySQL Module
|
|
//
|
|
|
|
#include "amxxmodule.h"
|
|
#include "MysqlDriver.h"
|
|
#include "MysqlDatabase.h"
|
|
#include <amtl/am-string.h>
|
|
|
|
using namespace SourceMod;
|
|
|
|
bool MysqlDriver::IsCompatDriver(const char *namestring)
|
|
{
|
|
return (strncasecmp(namestring, "mysql", 5) == 0);
|
|
}
|
|
|
|
const char *MysqlDriver::NameString()
|
|
{
|
|
return "mysql";
|
|
}
|
|
|
|
IDatabase *MysqlDriver::Connect(DatabaseInfo *info, int *errcode, char *error, size_t maxlength)
|
|
{
|
|
return _Connect(info, errcode, error, maxlength, false);
|
|
}
|
|
|
|
IDatabase *MysqlDriver::Connect2(DatabaseInfo *info, int *errcode, char *error, size_t maxlength)
|
|
{
|
|
return _Connect(info, errcode, error, maxlength, true);
|
|
}
|
|
|
|
IDatabase *MysqlDriver::_Connect(DatabaseInfo *info, int *errcode, char *error, size_t maxlength, bool do_timeout)
|
|
{
|
|
MYSQL *mysql = mysql_init(NULL);
|
|
|
|
if (!mysql)
|
|
{
|
|
if (errcode)
|
|
*errcode = -1;
|
|
if (error && maxlength)
|
|
{
|
|
ke::SafeSprintf(error, maxlength, "Initialization failed");
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
decltype(info->max_timeout) timeout = atoi(LOCALINFO("mysql_timeout"));
|
|
|
|
if (timeout <= 0)
|
|
{
|
|
timeout = 60;
|
|
}
|
|
|
|
if (do_timeout && info->max_timeout > 0)
|
|
{
|
|
timeout = info->max_timeout;
|
|
}
|
|
|
|
mysql_options(mysql, MYSQL_OPT_CONNECT_TIMEOUT, (const char *)&timeout);
|
|
mysql_options(mysql, MYSQL_OPT_READ_TIMEOUT, (const char *)&timeout);
|
|
mysql_options(mysql, MYSQL_OPT_WRITE_TIMEOUT, (const char *)&timeout);
|
|
|
|
/** Have MySQL automatically reconnect if it times out or loses connection.
|
|
* This will prevent "MySQL server has gone away" errors after a while.
|
|
*/
|
|
my_bool my_true = true;
|
|
mysql_options(mysql, MYSQL_OPT_RECONNECT, (const char *)&my_true);
|
|
|
|
if (mysql_real_connect(mysql,
|
|
info->host,
|
|
info->user,
|
|
info->pass,
|
|
info->database,
|
|
info->port,
|
|
NULL,
|
|
CLIENT_MULTI_STATEMENTS) == NULL)
|
|
{
|
|
if (errcode)
|
|
{
|
|
*errcode = mysql_errno(mysql);
|
|
}
|
|
if (error && maxlength)
|
|
{
|
|
ke::SafeSprintf(error, maxlength, "%s", mysql_error(mysql));
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
MysqlDatabase *pMysql = new MysqlDatabase(mysql, this);
|
|
|
|
if (info->charset && *info->charset)
|
|
{
|
|
pMysql->SetCharacterSet(info->charset);
|
|
}
|
|
|
|
return static_cast<IDatabase *>(pMysql);
|
|
}
|
|
|
|
int MysqlDriver::QuoteString(const char *str, char buffer[], size_t maxlen, size_t *newsize)
|
|
{
|
|
unsigned long size = static_cast<unsigned long>(strlen(str));
|
|
unsigned long needed = size*2 + 1;
|
|
|
|
if (maxlen < needed)
|
|
{
|
|
return (int)needed;
|
|
}
|
|
|
|
needed = mysql_escape_string(buffer, str, size);
|
|
if (newsize)
|
|
{
|
|
*newsize = static_cast<size_t>(needed);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|