core/smn_datapacks.cpp
author Borja Ferrer <faluco@alliedmods.net>
Wed Feb 14 11:03:03 2007 +0000 (2007-02-14)
changeset 488 daf1bd692ef3
parent 485 73bad6a148b9
child 490 02193490a838
permissions -rw-r--r--
improved data pack type checking
added readmemory member
faluco@484
     1
#include "sm_globals.h"
faluco@484
     2
#include "HandleSys.h"
faluco@484
     3
#include "CDataPack.h"
faluco@484
     4
faluco@484
     5
HandleType_t g_DataPackType;
faluco@484
     6
faluco@484
     7
class DataPackNatives : 
faluco@484
     8
	public SMGlobalClass,
faluco@484
     9
	public IHandleTypeDispatch
faluco@484
    10
{
faluco@484
    11
public:
faluco@484
    12
	void OnSourceModAllInitialized()
faluco@484
    13
	{
faluco@484
    14
		g_DataPackType = g_HandleSys.CreateType("DataPack", this, 0, NULL, NULL, g_pCoreIdent, NULL);
faluco@484
    15
	}
faluco@484
    16
	void OnSourceModShutdown()
faluco@484
    17
	{
faluco@484
    18
		g_HandleSys.RemoveType(g_DataPackType, g_pCoreIdent);
faluco@484
    19
		g_DataPackType = 0;
faluco@484
    20
	}
faluco@484
    21
	void OnHandleDestroy(HandleType_t type, void *object)
faluco@484
    22
	{
faluco@484
    23
		g_SourceMod.FreeDataPack(reinterpret_cast<IDataPack *>(object));
faluco@484
    24
	}
faluco@484
    25
};
faluco@484
    26
faluco@484
    27
static cell_t smn_CreateDataPack(IPluginContext *pContext, const cell_t *params)
faluco@484
    28
{
faluco@484
    29
	IDataPack *pDataPack = g_SourceMod.CreateDataPack();
faluco@484
    30
faluco@484
    31
	if (!pDataPack)
faluco@484
    32
	{
faluco@484
    33
		return 0;
faluco@484
    34
	}
faluco@484
    35
faluco@484
    36
	return g_HandleSys.CreateHandle(g_DataPackType, pDataPack, pContext->GetIdentity(), g_pCoreIdent, NULL);
faluco@484
    37
}
faluco@484
    38
faluco@484
    39
static cell_t smn_WritePackCell(IPluginContext *pContext, const cell_t *params)
faluco@484
    40
{
faluco@484
    41
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
    42
	HandleError herr;
faluco@484
    43
	HandleSecurity sec;
faluco@484
    44
	IDataPack *pDataPack;
faluco@484
    45
faluco@484
    46
	sec.pOwner = NULL;
faluco@484
    47
	sec.pIdentity = g_pCoreIdent;
faluco@484
    48
faluco@484
    49
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
    50
		!= HandleError_None)
faluco@484
    51
	{
faluco@484
    52
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
    53
	}
faluco@484
    54
faluco@484
    55
	pDataPack->PackCell(params[2]);
faluco@484
    56
faluco@484
    57
	return 1;
faluco@484
    58
}
faluco@484
    59
faluco@484
    60
static cell_t smn_WritePackFloat(IPluginContext *pContext, const cell_t *params)
faluco@484
    61
{
faluco@484
    62
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
    63
	HandleError herr;
faluco@484
    64
	HandleSecurity sec;
faluco@484
    65
	IDataPack *pDataPack;
faluco@484
    66
faluco@484
    67
	sec.pOwner = NULL;
faluco@484
    68
	sec.pIdentity = g_pCoreIdent;
faluco@484
    69
faluco@484
    70
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
    71
		!= HandleError_None)
faluco@484
    72
	{
faluco@484
    73
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
    74
	}
faluco@484
    75
faluco@484
    76
	pDataPack->PackFloat(sp_ctof(params[2]));
faluco@484
    77
faluco@484
    78
	return 1;
faluco@484
    79
}
faluco@484
    80
faluco@484
    81
static cell_t smn_WritePackString(IPluginContext *pContext, const cell_t *params)
faluco@484
    82
{
faluco@484
    83
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
    84
	HandleError herr;
faluco@484
    85
	HandleSecurity sec;
faluco@484
    86
	IDataPack *pDataPack;
faluco@484
    87
	int err;
faluco@484
    88
faluco@484
    89
	sec.pOwner = NULL;
faluco@484
    90
	sec.pIdentity = g_pCoreIdent;
faluco@484
    91
faluco@484
    92
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
    93
		!= HandleError_None)
faluco@484
    94
	{
faluco@484
    95
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
    96
	}
faluco@484
    97
faluco@484
    98
	char *str;
faluco@484
    99
	if ((err=pContext->LocalToString(params[2], &str)) != SP_ERROR_NONE)
faluco@484
   100
	{
faluco@484
   101
		pContext->ThrowNativeErrorEx(err, NULL);
faluco@484
   102
		return 0;
faluco@484
   103
	}
faluco@484
   104
faluco@484
   105
	pDataPack->PackString(str);
faluco@484
   106
faluco@484
   107
	return 1;
faluco@484
   108
}
faluco@484
   109
faluco@484
   110
static cell_t smn_ReadPackCell(IPluginContext *pContext, const cell_t *params)
faluco@484
   111
{
faluco@484
   112
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   113
	HandleError herr;
faluco@484
   114
	HandleSecurity sec;
faluco@484
   115
	IDataPack *pDataPack;
faluco@484
   116
faluco@484
   117
	sec.pOwner = NULL;
faluco@484
   118
	sec.pIdentity = g_pCoreIdent;
faluco@484
   119
faluco@484
   120
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   121
		!= HandleError_None)
faluco@484
   122
	{
faluco@484
   123
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   124
	}
faluco@484
   125
faluco@488
   126
	if (!pDataPack->IsReadable(sizeof(size_t) + sizeof(cell_t)))
faluco@484
   127
	{
faluco@484
   128
		return pContext->ThrowNativeError("DataPack operation is out of bounds.");
faluco@484
   129
	}
faluco@484
   130
faluco@484
   131
	return pDataPack->ReadCell();
faluco@484
   132
}
faluco@484
   133
faluco@484
   134
static cell_t smn_ReadPackFloat(IPluginContext *pContext, const cell_t *params)
faluco@484
   135
{
faluco@484
   136
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   137
	HandleError herr;
faluco@484
   138
	HandleSecurity sec;
faluco@484
   139
	IDataPack *pDataPack;
faluco@484
   140
faluco@484
   141
	sec.pOwner = NULL;
faluco@484
   142
	sec.pIdentity = g_pCoreIdent;
faluco@484
   143
faluco@484
   144
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   145
		!= HandleError_None)
faluco@484
   146
	{
faluco@484
   147
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   148
	}
faluco@484
   149
faluco@488
   150
	if (!pDataPack->IsReadable(sizeof(size_t) + sizeof(float)))
faluco@484
   151
	{
faluco@484
   152
		return pContext->ThrowNativeError("DataPack operation is out of bounds.");
faluco@484
   153
	}
faluco@484
   154
faluco@484
   155
	return sp_ftoc(pDataPack->ReadFloat());
faluco@484
   156
}
faluco@484
   157
faluco@484
   158
static cell_t smn_ReadPackString(IPluginContext *pContext, const cell_t *params)
faluco@484
   159
{
faluco@484
   160
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   161
	HandleError herr;
faluco@484
   162
	HandleSecurity sec;
faluco@484
   163
	IDataPack *pDataPack;
faluco@484
   164
faluco@484
   165
	sec.pOwner = NULL;
faluco@484
   166
	sec.pIdentity = g_pCoreIdent;
faluco@484
   167
faluco@484
   168
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   169
		!= HandleError_None)
faluco@484
   170
	{
faluco@484
   171
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   172
	}
faluco@484
   173
faluco@484
   174
	const char *str;
faluco@484
   175
	if (!(str=pDataPack->ReadString(NULL)))
faluco@484
   176
	{
faluco@484
   177
		return pContext->ThrowNativeError("DataPack operation is out of bounds.");
faluco@484
   178
	}
faluco@484
   179
faluco@484
   180
	pContext->StringToLocal(params[2], params[3], str);
faluco@484
   181
faluco@484
   182
	return 1;
faluco@484
   183
}
faluco@484
   184
faluco@484
   185
static cell_t smn_ResetPack(IPluginContext *pContext, const cell_t *params)
faluco@484
   186
{
faluco@484
   187
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   188
	HandleError herr;
faluco@484
   189
	HandleSecurity sec;
faluco@484
   190
	IDataPack *pDataPack;
faluco@484
   191
faluco@484
   192
	sec.pOwner = NULL;
faluco@484
   193
	sec.pIdentity = g_pCoreIdent;
faluco@484
   194
faluco@484
   195
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   196
		!= HandleError_None)
faluco@484
   197
	{
faluco@484
   198
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   199
	}
faluco@484
   200
faluco@484
   201
	pDataPack->Reset();
faluco@484
   202
	if (params[2])
faluco@484
   203
	{
faluco@484
   204
		pDataPack->ResetSize();
faluco@484
   205
	}
faluco@484
   206
faluco@484
   207
	return 1;
faluco@484
   208
}
faluco@484
   209
faluco@484
   210
static cell_t smn_GetPackPosition(IPluginContext *pContext, const cell_t *params)
faluco@484
   211
{
faluco@484
   212
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   213
	HandleError herr;
faluco@484
   214
	HandleSecurity sec;
faluco@484
   215
	IDataPack *pDataPack;
faluco@484
   216
faluco@484
   217
	sec.pOwner = NULL;
faluco@484
   218
	sec.pIdentity = g_pCoreIdent;
faluco@484
   219
faluco@484
   220
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   221
		!= HandleError_None)
faluco@484
   222
	{
faluco@484
   223
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   224
	}
faluco@484
   225
faluco@484
   226
	return static_cast<cell_t>(pDataPack->GetPosition());
faluco@484
   227
}
faluco@484
   228
faluco@484
   229
static cell_t smn_SetPackPosition(IPluginContext *pContext, const cell_t *params)
faluco@484
   230
{
faluco@484
   231
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   232
	HandleError herr;
faluco@484
   233
	HandleSecurity sec;
faluco@484
   234
	IDataPack *pDataPack;
faluco@484
   235
faluco@484
   236
	sec.pOwner = NULL;
faluco@484
   237
	sec.pIdentity = g_pCoreIdent;
faluco@484
   238
faluco@484
   239
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   240
		!= HandleError_None)
faluco@484
   241
	{
faluco@484
   242
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   243
	}
faluco@484
   244
faluco@484
   245
	if (!pDataPack->SetPosition(params[2]))
faluco@484
   246
	{
faluco@484
   247
		return pContext->ThrowNativeError("Invalid DataPack position, %d is out of bounds", params[2]);
faluco@484
   248
	}
faluco@484
   249
faluco@484
   250
	return 1;
faluco@484
   251
}
faluco@484
   252
faluco@484
   253
static cell_t smn_IsPackReadable(IPluginContext *pContext, const cell_t *params)
faluco@484
   254
{
faluco@484
   255
	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@484
   256
	HandleError herr;
faluco@484
   257
	HandleSecurity sec;
faluco@484
   258
	IDataPack *pDataPack;
faluco@484
   259
faluco@484
   260
	sec.pOwner = NULL;
faluco@484
   261
	sec.pIdentity = g_pCoreIdent;
faluco@484
   262
faluco@484
   263
	if ((herr=g_HandleSys.ReadHandle(hndl, g_DataPackType, &sec, (void **)&pDataPack))
faluco@484
   264
		!= HandleError_None)
faluco@484
   265
	{
faluco@484
   266
		return pContext->ThrowNativeError("Invalid data pack handle %x (error %d)", hndl, herr);
faluco@484
   267
	}
faluco@484
   268
faluco@484
   269
	return pDataPack->IsReadable(params[2]) ? 1 : 0;
faluco@484
   270
}
faluco@484
   271
faluco@484
   272
static DataPackNatives s_DataPackNatives;
faluco@484
   273
faluco@484
   274
REGISTER_NATIVES(datapacknatives)
faluco@484
   275
{
faluco@484
   276
	{"CreateDataPack",			smn_CreateDataPack},
faluco@484
   277
	{"WritePackCell",			smn_WritePackCell},
faluco@484
   278
	{"WritePackFloat",			smn_WritePackFloat},
faluco@484
   279
	{"WritePackString",			smn_WritePackString},
faluco@484
   280
	{"ReadPackCell",			smn_ReadPackCell},
faluco@484
   281
	{"ReadPackFloat",			smn_ReadPackFloat},
faluco@484
   282
	{"ReadPackString",			smn_ReadPackString},
faluco@484
   283
	{"ResetPack",				smn_ResetPack},
faluco@484
   284
	{"GetPackPosition",			smn_GetPackPosition},
faluco@484
   285
	{"SetPackPosition",			smn_SetPackPosition},
faluco@484
   286
	{"IsPackReadable",			smn_IsPackReadable},
faluco@484
   287
	{NULL,						NULL}
faluco@485
   288
};