amxmodx/dlls/mssql/mssql.cpp

351 lines
8.1 KiB
C++
Raw Normal View History

2004-06-28 23:20:04 +00:00
/* AMX Mod X
* MSSQL Module
2004-05-28 04:24:15 +00:00
*
2004-06-28 23:20:04 +00:00
* by BAILOPAN
2004-05-28 04:24:15 +00:00
*
2004-06-28 23:20:04 +00:00
* This file is part of AMX Mod X.
2004-05-28 04:24:15 +00:00
*
*
2004-06-28 23:20:04 +00:00
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at
* your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* In addition, as a special exception, the author gives permission to
* link the code of this program with the Half-Life Game Engine ("HL
* Engine") and Modified Game Libraries ("MODs") developed by Valve,
* L.L.C ("Valve"). You must obey the GNU General Public License in all
* respects for all of the code used other than the HL Engine and MODs
* from Valve. If you modify this file, you may extend this exception
* to your version of the file, but you are not obligated to do so. If
* you do not wish to do so, delete this exception statement from your
* version.
2004-05-28 04:24:15 +00:00
*/
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
#include <vector>
#include "mssql_amx.h"
#include "amxxmodule.h"
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
using namespace std;
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
std::vector<SQLResult*> Results;
std::vector<SQL*> DBList;
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
int sql_exists(const char* host,const char* user,const char* pass,const char* dbase) {
vector<SQL*>::iterator i;
2004-05-28 01:48:22 +00:00
int id = 0;
for (i=DBList.begin(); i!=DBList.end(); i++) {
id++;
2004-06-28 23:20:04 +00:00
if (((*i)->Host.compare(host) == 0) &&
((*i)->Username.compare(user) == 0) &&
((*i)->Password.compare(pass) == 0) &&
((*i)->Database.compare(dbase) == 0) &&
(!(*i)->isFree)) {
return id;
2004-05-28 01:48:22 +00:00
}
}
return -1;
}
2004-06-28 23:20:04 +00:00
// ///////////////////////////////
// mssql natives for AMX scripting
// ///////////////////////////////
// sql = mssql_connect(host[],user[],pass[],dbname[],error[],maxlength) :
// - open connection
static cell AMX_NATIVE_CALL sql_connect(AMX *amx, cell *params) // 6 param
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
int i;
char *host = MF_GetAmxString(amx,params[1], 0, &i);
char *user = MF_GetAmxString(amx,params[2], 1, &i);
char *pass = MF_GetAmxString(amx,params[3], 2, &i);
char *dbname = MF_GetAmxString(amx,params[4], 3, &i);
i = 0;
if (!strlen(host) || !strlen(user) || !strlen(dbname)) {
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid parameter supplied");
2004-06-28 23:20:04 +00:00
return -1;
}
int id = sql_exists(host,user,pass,dbname);
if (id > 0)
2004-05-28 01:48:22 +00:00
return id;
2004-06-28 23:20:04 +00:00
SQL *c=NULL;
for (i=0; (unsigned int)i<DBList.size(); i++) {
if (DBList[i]->isFree) {
2004-05-28 01:48:22 +00:00
id = i;
break;
}
}
2004-06-28 23:20:04 +00:00
if (id>=0) {
c = DBList[id];
2004-05-28 01:48:22 +00:00
} else {
2004-06-28 23:20:04 +00:00
c = new SQL;
DBList.push_back(c);
id = (unsigned int)(DBList.size() - 1);
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
if (!c->Connect(host, user, pass, dbname))
{
MF_SetAmxString(amx, params[5], c->ErrorStr.c_str(), params[6]);
2004-05-28 01:48:22 +00:00
return -1;
}
2004-06-28 23:20:04 +00:00
MF_SetAmxString(amx,params[5],"",params[6]);
return id+1;
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
// mssql_error(sql,dest[],maxlength)
// - store maxlength characters from mssql error in current row to dest
static cell AMX_NATIVE_CALL sql_error(AMX *amx, cell *params) // 3 params
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
unsigned int id = params[1]-1;
if (id >= DBList.size() || DBList[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid connection %d", id);
2004-05-28 01:48:22 +00:00
return 0;
}
2004-06-28 23:20:04 +00:00
SQL *sql = DBList[id];
if (sql->ErrorStr.size() > 1)
{
MF_SetAmxString(amx, params[2], sql->ErrorStr.c_str(), params[3]);
sql->ErrorStr.assign("");
return 1;
} else {
if (sql->ErrorStr.size() > 1)
{
MF_SetAmxString(amx, params[2], sql->ErrorStr.c_str(), params[3]);
sql->ErrorStr.assign("");
return 1;
}
}
MF_SetAmxString(amx, params[2], "", params[3]);
return 0;
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
// mssql_query(sql,query[]) - returns 0 on success, <0 on failure, >0 on result set
static cell AMX_NATIVE_CALL sql_query(AMX *amx, cell *params) // 2 params
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
unsigned int id = params[1]-1;
if (id >= DBList.size() || DBList[id]->isFree) {
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid Database Handle %d", id);
2004-06-28 23:20:04 +00:00
return -1;
}
int len = 0;
const char *query = MF_FormatAmxString(amx, params, 2, &len);
SQL *sql = DBList[id];
return sql->Query(query); //Return the result set handle, if any
}
// mssql_nextrow(sql) :
// - read next row
// - return :
// . number of line
// . 0 at end
static cell AMX_NATIVE_CALL sql_nextrow(AMX *amx, cell *params) // 1 param
{
unsigned int id = params[1]-1;
if (id >= Results.size() || Results[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid result handle %d", id);
2004-05-28 01:48:22 +00:00
return 0;
}
2004-06-28 23:20:04 +00:00
SQLResult *Result = Results[id];
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
return Result->Nextrow();
}
// mssql_close(sql) :
// - free result
// - close connection
static cell AMX_NATIVE_CALL sql_close(AMX *amx, cell *params) // 1 param
{
unsigned int id = params[1]-1;
if (id >= DBList.size() || DBList[id]->isFree) {
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid Database Handle %d", id);
2004-06-28 23:20:04 +00:00
return 0;
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
SQL *sql = DBList[id];
sql->Disconnect();
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
return 1;
}
//Returns a field from a query result handle.
// 2 param - returns integer
// 3 param - stores float in cell byref
// 4 param - stores string
static cell AMX_NATIVE_CALL sql_getfield(AMX *amx, cell *params) // 2-4 params
{
unsigned int id = params[1]-1;
if (id >= Results.size() || Results[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid result handle %d", id);
2004-06-28 23:20:04 +00:00
return 0;
}
SQLResult *Result = Results[id];
int numParams = (*params)/sizeof(cell);
cell *fAddr = NULL;
const char *field = Result->GetField(id);
if (field == NULL)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid column %d", id);
2004-06-28 23:20:04 +00:00
return 0;
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
switch (numParams)
{
case 2:
return atoi(field);
break;
case 3:
fAddr = MF_GetAmxAddr(amx, params[3]);
*fAddr = amx_ftoc((REAL)atof(field));
2004-05-28 03:57:48 +00:00
return 1;
2004-06-28 23:20:04 +00:00
break;
case 4:
return MF_SetAmxString(amx, params[3], field?field:"", params[4]);
break;
default:
break;
2004-05-28 03:57:48 +00:00
}
2004-06-28 23:20:04 +00:00
return 0;
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
//Returns a field from a query result handle.
// 2 param - returns integer
// 3 param - stores float in cell byref
// 4 param - stores string
static cell AMX_NATIVE_CALL sql_getresult(AMX *amx, cell *params) // 4 params
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
unsigned int id = params[1]-1;
if (id >= Results.size() || Results[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid result handle %d", id);
2004-05-28 01:48:22 +00:00
return 0;
}
2004-06-28 23:20:04 +00:00
SQLResult *Result = Results[id];
int numParams = (*params)/sizeof(cell);
cell *fAddr = NULL;
int len = 0;
const char *column = MF_GetAmxString(amx, params[2], 0, &len);
const char *field = Result->GetField(id);
if (field == NULL)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid column \"%s\"", column);
2004-05-28 01:48:22 +00:00
return 0;
2004-06-28 23:20:04 +00:00
}
2004-05-28 01:48:22 +00:00
2004-06-28 23:20:04 +00:00
switch (numParams)
{
case 2:
return atoi(field);
break;
case 3:
fAddr = MF_GetAmxAddr(amx, params[3]);
*fAddr = amx_ftoc((REAL)atof(field));
2004-05-28 01:48:22 +00:00
return 1;
2004-06-28 23:20:04 +00:00
break;
case 4:
return MF_SetAmxString(amx, params[3], field?field:"", params[4]);
break;
default:
break;
2004-05-28 01:48:22 +00:00
}
return 0;
}
2004-06-28 23:20:04 +00:00
static cell AMX_NATIVE_CALL sql_free_result(AMX *amx, cell *params)
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
unsigned int id = params[1]-1;
if (id >= Results.size() || Results[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid result handle %d", id);
2004-05-28 01:48:22 +00:00
return 0;
}
2004-06-28 23:20:04 +00:00
SQLResult *Result = Results[id];
Result->FreeResult();
2004-05-28 01:48:22 +00:00
return 1;
}
2004-06-28 23:20:04 +00:00
static cell AMX_NATIVE_CALL sql_num_rows(AMX *amx, cell *params)
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
unsigned int id = params[1]-1;
if (id >= Results.size() || Results[id]->isFree)
{
MF_LogError(amx, AMX_ERR_NATIVE, "Invalid result handle %d", id);
2004-06-28 23:20:04 +00:00
return 0;
}
SQLResult *Result = Results[id];
return (cell)Result->NumRows();
2004-05-28 01:48:22 +00:00
}
2004-06-28 23:20:04 +00:00
static cell AMX_NATIVE_CALL sql_type(AMX *amx, cell *params)
2004-05-28 09:21:11 +00:00
{
return MF_SetAmxString(amx, params[1], "mssql", params[2]);
}
2004-06-28 23:20:04 +00:00
AMX_NATIVE_INFO mssql_Natives[] = {
{ "dbi_connect", sql_connect },
{ "dbi_query", sql_query },
{ "dbi_field", sql_getfield },
{ "dbi_nextrow", sql_nextrow },
{ "dbi_close", sql_close },
{ "dbi_error", sql_error },
{ "dbi_type", sql_type },
{ "dbi_free_result", sql_free_result },
{ "dbi_num_rows", sql_num_rows },
{ "dbi_result", sql_getresult },
{ NULL, NULL }
};
2004-05-28 09:21:11 +00:00
2004-05-28 01:48:22 +00:00
void OnAmxxAttach()
{
MF_AddNatives(mssql_Natives);
}
2004-06-28 23:20:04 +00:00
void OnAmxxDetach()
2004-05-28 01:48:22 +00:00
{
2004-06-28 23:20:04 +00:00
Results.clear();
DBList.clear();
2004-05-28 01:48:22 +00:00
}