115 lines
2.5 KiB
C++
115 lines
2.5 KiB
C++
|
#include "sm_globals.h"
|
||
|
|
#include "HandleSys.h"
|
||
|
|
#include "LibrarySys.h"
|
||
|
|
|
||
|
|
HandleType_t g_FileType;
|
||
|
|
HandleType_t g_DirType;
|
||
|
|
|
||
|
|
class FileNatives :
|
||
|
|
public SMGlobalClass,
|
||
|
|
public IHandleTypeDispatch
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
virtual void OnSourceModAllInitialized()
|
||
|
|
{
|
||
|
|
HandleSecurity sec;
|
||
|
|
sec.owner = g_pCoreIdent;
|
||
|
|
sec.access[HandleAccess_Inherit] = false;
|
||
|
|
sec.access[HandleAccess_Create] = false;
|
||
|
|
|
||
|
|
g_FileType = g_HandleSys.CreateTypeEx("File", this, 0, &sec, NULL);
|
||
|
|
g_DirType = g_HandleSys.CreateTypeEx("Directory", this, 0, &sec, NULL);
|
||
|
|
}
|
||
|
|
virtual void OnSourceModShutdown()
|
||
|
|
{
|
||
|
|
g_HandleSys.RemoveType(g_DirType, g_pCoreIdent);
|
||
|
|
g_HandleSys.RemoveType(g_FileType, g_pCoreIdent);
|
||
|
|
g_DirType = 0;
|
||
|
|
g_FileType = 0;
|
||
|
|
}
|
||
|
|
virtual void OnHandleDestroy(HandleType_t type, void *object)
|
||
|
|
{
|
||
|
|
if (type == g_FileType)
|
||
|
|
{
|
||
|
|
FILE *fp = (FILE *)object;
|
||
|
|
fclose(fp);
|
||
|
|
} else if (type == g_DirType) {
|
||
|
|
IDirectory *pDir = (IDirectory *)object;
|
||
|
|
g_LibSys.CloseDirectory(pDir);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
cell_t sm_OpenDirectory(IPluginContext *pContext, const cell_t *params)
|
||
|
|
{
|
||
|
|
char *path;
|
||
|
|
int err;
|
||
|
|
if ((err=pContext->LocalToString(params[1], &path)) != SP_ERROR_NONE)
|
||
|
|
{
|
||
|
|
pContext->ThrowNativeErrorEx(err, NULL);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
IDirectory *pDir = g_LibSys.OpenDirectory(path);
|
||
|
|
if (!pDir)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return g_HandleSys.CreateScriptHandle(g_DirType, pDir, pContext, g_pCoreIdent);
|
||
|
|
}
|
||
|
|
|
||
|
|
cell_t sm_ReadDirEntry(IPluginContext *pContext, const cell_t *params)
|
||
|
|
{
|
||
|
|
Handle_t hndl = static_cast<Handle_t>(params[1]);
|
||
|
|
|
||
|
|
IDirectory *pDir;
|
||
|
|
HandleError herr;
|
||
|
|
int err;
|
||
|
|
if ((herr=g_HandleSys.ReadHandle(hndl, g_DirType, g_pCoreIdent, (void **)&pDir))
|
||
|
|
!= HandleError_None)
|
||
|
|
{
|
||
|
|
return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!pDir->MoreFiles())
|
||
|
|
{
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
cell_t *filetype;
|
||
|
|
if ((err=pContext->LocalToPhysAddr(params[4], &filetype)) != SP_ERROR_NONE)
|
||
|
|
{
|
||
|
|
pContext->ThrowNativeErrorEx(err, NULL);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pDir->IsEntryDirectory())
|
||
|
|
{
|
||
|
|
*filetype = 1;
|
||
|
|
} else if (pDir->IsEntryFile()) {
|
||
|
|
*filetype = 2;
|
||
|
|
} else {
|
||
|
|
*filetype = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
const char *path = pDir->GetEntryName();
|
||
|
|
if ((err=pContext->StringToLocalUTF8(params[2], params[3],path, NULL))
|
||
|
|
!= SP_ERROR_NONE)
|
||
|
|
{
|
||
|
|
pContext->ThrowNativeErrorEx(err, NULL);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
REGISTER_NATIVES(filesystem)
|
||
|
|
{
|
||
|
|
{"OpenDirectory", sm_OpenDirectory},
|
||
|
|
{"ReadDirEntry", sm_ReadDirEntry},
|
||
|
|
{NULL, NULL},
|
||
|
|
};
|