dvander@370: /**
ds@653:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@370:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
dvander@370:  *
dvander@370:  * Version: $Id$
dvander@370:  */
dvander@370: 
faluco@274: #include <sys/stat.h>
dvander@3794: #include <string.h>
dvander@3791: #include <IHandleSys.h>
dvander@3791: #include <ILibrarySys.h>
dvander@3791: #include <IPluginSys.h>
dvander@3791: #include <IForwardSys.h>
dvander@3791: #include <ISourceMod.h>
dvander@3791: #include <ITranslator.h>
dvander@3791: #include "common_logic.h"
dvander@252: 
dvander@252: HandleType_t g_FileType;
dvander@252: HandleType_t g_DirType;
dvander@1914: IChangeableForward *g_pLogHook = NULL;
dvander@252: 
dvander@252: class FileNatives : 
dvander@252: 	public SMGlobalClass,
dvander@1914: 	public IHandleTypeDispatch,
dvander@1914: 	public IPluginsListener
dvander@252: {
dvander@252: public:
dvander@1914: 	FileNatives()
dvander@1914: 	{
dvander@1914: 	}
dvander@252: 	virtual void OnSourceModAllInitialized()
dvander@252: 	{
dvander@3791: 		g_FileType = handlesys->CreateType("File", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@3791: 		g_DirType = handlesys->CreateType("Directory", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@3791: 		g_pLogHook = forwardsys->CreateForwardEx(NULL, ET_Hook, 1, NULL, Param_String);
dvander@3791: 		pluginsys->AddPluginsListener(this);
dvander@252: 	}
dvander@252: 	virtual void OnSourceModShutdown()
dvander@252: 	{
dvander@3791: 		pluginsys->RemovePluginsListener(this);
dvander@3791: 		forwardsys->ReleaseForward(g_pLogHook);
dvander@3791: 		handlesys->RemoveType(g_DirType, g_pCoreIdent);
dvander@3791: 		handlesys->RemoveType(g_FileType, g_pCoreIdent);
dvander@252: 		g_DirType = 0;
dvander@252: 		g_FileType = 0;
dvander@252: 	}
dvander@252: 	virtual void OnHandleDestroy(HandleType_t type, void *object)
dvander@252: 	{
dvander@252: 		if (type == g_FileType)
dvander@252: 		{
dvander@252: 			FILE *fp = (FILE *)object;
dvander@252: 			fclose(fp);
dvander@1914: 		}
dvander@1914: 		else if (type == g_DirType)
dvander@1914: 		{
dvander@252: 			IDirectory *pDir = (IDirectory *)object;
dvander@3791: 			libsys->CloseDirectory(pDir);
dvander@1914: 		}
dvander@1914: 	}
dvander@1914: 	virtual void AddLogHook(IPluginFunction *pFunc)
dvander@1914: 	{
dvander@1914: 		g_pLogHook->AddFunction(pFunc);
dvander@1914: 	}
dvander@1914: 	virtual void RemoveLogHook(IPluginFunction *pFunc)
dvander@1914: 	{
dvander@1914: 		g_pLogHook->RemoveFunction(pFunc);
dvander@1914: 	}
dvander@3791: 	virtual bool LogPrint(const char *msg)
dvander@1914: 	{
dvander@3791: 		cell_t result = 0;
dvander@1914: 		g_pLogHook->PushString(msg);
dvander@1914: 		g_pLogHook->Execute(&result);
dvander@3791: 		return result >= Pl_Handled;
dvander@3791: 	}
dvander@3791: } s_FileNatives;
dvander@1914: 
dvander@3791: bool OnLogPrint(const char *msg)
dvander@3791: {
dvander@3791: 	return s_FileNatives.LogPrint(msg);
dvander@3791: }
dvander@252: 
faluco@274: static cell_t sm_OpenDirectory(IPluginContext *pContext, const cell_t *params)
dvander@252: {
dvander@252: 	char *path;
dvander@252: 	int err;
dvander@252: 	if ((err=pContext->LocalToString(params[1], &path)) != SP_ERROR_NONE)
dvander@252: 	{
dvander@252: 		pContext->ThrowNativeErrorEx(err, NULL);
dvander@252: 		return 0;
dvander@252: 	}
dvander@252: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", path);
dvander@255: 
dvander@3791: 	IDirectory *pDir = libsys->OpenDirectory(realpath);
dvander@252: 	if (!pDir)
dvander@252: 	{
dvander@252: 		return 0;
dvander@252: 	}
dvander@252: 
dvander@3791: 	return handlesys->CreateHandle(g_DirType, pDir, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@252: }
dvander@252: 
faluco@274: static cell_t sm_ReadDirEntry(IPluginContext *pContext, const cell_t *params)
dvander@252: {
dvander@252: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@252: 	IDirectory *pDir;
dvander@252: 	HandleError herr;
dvander@262: 	HandleSecurity sec;
dvander@252: 	int err;
dvander@262: 
dvander@262: 	sec.pOwner = NULL;
dvander@262: 	sec.pIdentity = g_pCoreIdent;
dvander@262: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_DirType, &sec, (void **)&pDir))
dvander@252: 		!= HandleError_None)
dvander@252: 	{
dvander@252: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@252: 	}
dvander@252: 
dvander@252: 	if (!pDir->MoreFiles())
dvander@252: 	{
dvander@252: 		return false;
dvander@252: 	}
dvander@252: 
dvander@252: 	cell_t *filetype;
dvander@252: 	if ((err=pContext->LocalToPhysAddr(params[4], &filetype)) != SP_ERROR_NONE)
dvander@252: 	{
dvander@252: 		pContext->ThrowNativeErrorEx(err, NULL);
dvander@252: 		return 0;
dvander@252: 	}
dvander@252: 
dvander@252: 	if (pDir->IsEntryDirectory())
dvander@252: 	{
dvander@252: 		*filetype = 1;
dvander@252: 	} else if (pDir->IsEntryFile()) {
dvander@252: 		*filetype = 2;
dvander@252: 	} else {
dvander@252: 		*filetype = 0;
dvander@252: 	}
dvander@252: 
dvander@252: 	const char *path = pDir->GetEntryName();
dvander@255: 	if ((err=pContext->StringToLocalUTF8(params[2], params[3], path, NULL))
dvander@252: 		!= SP_ERROR_NONE)
dvander@252: 	{
dvander@252: 		pContext->ThrowNativeErrorEx(err, NULL);
dvander@252: 		return 0;
dvander@252: 	}
dvander@252: 
dvander@255: 	pDir->NextEntry();
dvander@255: 
dvander@252: 	return true;
dvander@252: }
dvander@252: 
faluco@274: static cell_t sm_OpenFile(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name, *mode;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if ((err=pContext->LocalToString(params[2], &mode)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: 
faluco@274: 	FILE *pFile = fopen(realpath, mode);
faluco@274: 	if (!pFile)
faluco@274: 	{
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
dvander@3791: 	return handlesys->CreateHandle(g_FileType, pFile, pContext->GetIdentity(), g_pCoreIdent, NULL);
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_DeleteFile(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: 
faluco@274: 	return (unlink(realpath)) ? 0 : 1;
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_ReadFileLine(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@274: 	HandleError herr;
faluco@274: 	HandleSecurity sec;
faluco@274: 	FILE *pFile;
faluco@274: 	int err;
faluco@274: 
faluco@274: 	sec.pOwner = NULL;
faluco@274: 	sec.pIdentity = g_pCoreIdent;
faluco@274: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
faluco@274: 		!= HandleError_None)
faluco@274: 	{
faluco@274: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
faluco@274: 	}
faluco@274: 
faluco@274: 	char *buf;
faluco@274: 	if ((err=pContext->LocalToString(params[2], &buf)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
dvander@638: 	if (fgets(buf, params[3], pFile) == NULL)
dvander@638: 	{
dvander@638: 		return 0;
dvander@638: 	}
faluco@274: 
faluco@274: 	return 1;
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_IsEndOfFile(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@274: 	HandleError herr;
faluco@274: 	HandleSecurity sec;
faluco@274: 	FILE *pFile;
faluco@274: 
faluco@274: 	sec.pOwner = NULL;
faluco@274: 	sec.pIdentity = g_pCoreIdent;
faluco@274: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
faluco@274: 		!= HandleError_None)
faluco@274: 	{
faluco@274: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
faluco@274: 	}
faluco@274: 
faluco@274: 	return (feof(pFile)) ? 1 : 0;
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_FileSeek(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@274: 	HandleError herr;
faluco@274: 	HandleSecurity sec;
faluco@274: 	FILE *pFile;
faluco@274: 
faluco@274: 	sec.pOwner = NULL;
faluco@274: 	sec.pIdentity = g_pCoreIdent;
faluco@274: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
faluco@274: 		!= HandleError_None)
faluco@274: 	{
faluco@274: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
faluco@274: 	}
faluco@274: 
faluco@274: 	fseek(pFile, params[2], params[3]);
faluco@274: 	
faluco@274: 	return 1;
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_FilePosition(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@274: 	HandleError herr;
faluco@274: 	HandleSecurity sec;
faluco@274: 	FILE *pFile;
faluco@274: 
faluco@274: 	sec.pOwner = NULL;
faluco@274: 	sec.pIdentity = g_pCoreIdent;
faluco@274: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
faluco@274: 		!= HandleError_None)
faluco@274: 	{
faluco@274: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
faluco@274: 	}
faluco@274: 
faluco@274: 	return ftell(pFile);
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_FileExists(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
dvander@1854: 	if (params[0] >= 2 && params[2] == 1)
dvander@1854: 	{
dvander@3791: 		return smcore.FileExists(name) ? 1 : 0;
dvander@1854: 	}
dvander@1854: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: #ifdef PLATFORM_WINDOWS
faluco@274: 	struct _stat s;
faluco@274: 	if (_stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if (s.st_mode & S_IFREG)
faluco@274: 	{
faluco@274: 		return 1;
faluco@274: 	}
faluco@274: 	return 0;
dvander@351: #elif defined PLATFORM_POSIX
faluco@274: 	struct stat s;
faluco@274: 	if (stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if (S_ISREG(s.st_mode))
faluco@274: 	{
faluco@274: 		return 1;
faluco@274: 	}
faluco@274: 	return 0;
faluco@274: #endif
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_RenameFile(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *newpath, *oldpath;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &newpath)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if ((err=pContext->LocalToString(params[2], &oldpath)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
ds@652: 	char new_realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, new_realpath, sizeof(new_realpath), "%s", newpath);
ds@652: 	char old_realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, old_realpath, sizeof(old_realpath), "%s", oldpath);
faluco@274: 
faluco@274: #ifdef PLATFORM_WINDOWS
faluco@274: 	return (MoveFileA(old_realpath, new_realpath)) ? 1 : 0;
dvander@351: #elif defined PLATFORM_POSIX
faluco@274: 	return (rename(old_realpath, new_realpath)) ? 0 : 1;
faluco@274: #endif
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_DirExists(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: #ifdef PLATFORM_WINDOWS
faluco@274: 	struct _stat s;
faluco@274: 	if (_stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if (s.st_mode & S_IFDIR)
faluco@274: 	{
faluco@274: 		return 1;
faluco@274: 	}
faluco@274: 	return 0;
dvander@351: #elif defined PLATFORM_POSIX
faluco@274: 	struct stat s;
faluco@274: 	if (stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 	if (S_ISDIR(s.st_mode))
faluco@274: 	{
faluco@274: 		return 1;
faluco@274: 	}
faluco@274: 	return 0;
faluco@274: #endif
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_FileSize(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return -1;
faluco@274: 	}
faluco@274: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: #ifdef PLATFORM_WINDOWS
faluco@274: 	struct _stat s;
faluco@274: 	if (_stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return -1;
faluco@274: 	}
faluco@274: 	if (s.st_mode & S_IFREG)
faluco@274: 	{
faluco@274: 		return static_cast<cell_t>(s.st_size);
faluco@274: 	}
faluco@274: 	return -1;
dvander@351: #elif defined PLATFORM_POSIX
faluco@274: 	struct stat s;
faluco@274: 	if (stat(realpath, &s) != 0)
faluco@274: 	{
faluco@274: 		return -1;
faluco@274: 	}
faluco@274: 	if (S_ISREG(s.st_mode))
faluco@274: 	{
faluco@274: 		return static_cast<cell_t>(s.st_size);
faluco@274: 	}
faluco@274: 	return -1;
faluco@274: #endif
faluco@274: }
faluco@274: 
dvander@2307: static cell_t sm_CreateDirectory(IPluginContext *pContext, const cell_t *params)
dvander@2307: {
dvander@2307: 	char *name;
dvander@2307: 	char realpath[PLATFORM_MAX_PATH];
dvander@2307: 
dvander@2307: 	pContext->LocalToString(params[1], &name);
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
dvander@2307: 
dvander@2307: #if defined PLATFORM_WINDOWS
dvander@2307: 	return mkdir(realpath) == 0;
dvander@2307: #else
dvander@2307: 	return mkdir(realpath, params[2]) == 0;
dvander@2307: #endif
dvander@2307: }
dvander@2307: 
faluco@274: static cell_t sm_RemoveDir(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	char *name;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
ds@652: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
faluco@274: 
faluco@274: 	return (rmdir(realpath)) ? 0 : 1;
faluco@274: }
faluco@274: 
faluco@274: static cell_t sm_WriteFileLine(IPluginContext *pContext, const cell_t *params)
faluco@274: {
faluco@274: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@274: 	HandleError herr;
faluco@274: 	HandleSecurity sec;
faluco@274: 	FILE *pFile;
faluco@274: 
faluco@274: 	sec.pOwner = NULL;
faluco@274: 	sec.pIdentity = g_pCoreIdent;
faluco@274: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
faluco@274: 		!= HandleError_None)
faluco@274: 	{
faluco@274: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
faluco@274: 	}
faluco@274: 
faluco@274: 	char *fmt;
faluco@274: 	int err;
faluco@274: 	if ((err=pContext->LocalToString(params[2], &fmt)) != SP_ERROR_NONE)
faluco@274: 	{
faluco@274: 		pContext->ThrowNativeErrorEx(err, NULL);
faluco@274: 		return 0;
faluco@274: 	}
faluco@274: 
faluco@274: 	char buffer[2048];
faluco@274: 	int arg = 3;
dvander@3791: 	smcore.atcprintf(buffer, sizeof(buffer), fmt, pContext, params, &arg);
faluco@274: 	fprintf(pFile, "%s\n", buffer);
faluco@274: 
faluco@274: 	return 1;
faluco@274: }
faluco@274: 
dvander@821: static cell_t sm_FlushFile(IPluginContext *pContext, const cell_t *params)
dvander@821: {
dvander@821: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@821: 	HandleError herr;
dvander@821: 	HandleSecurity sec;
dvander@821: 	FILE *pFile;
dvander@821: 
dvander@821: 	sec.pOwner = NULL;
dvander@821: 	sec.pIdentity = g_pCoreIdent;
dvander@821: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@821: 		!= HandleError_None)
dvander@821: 	{
dvander@821: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@821: 	}
dvander@821: 
dvander@821: 	return (fflush(pFile) == 0) ? 1 : 0;
dvander@821: }
dvander@821: 
dvander@345: static cell_t sm_BuildPath(IPluginContext *pContext, const cell_t *params)
dvander@345: {
dvander@345: 	char path[PLATFORM_MAX_PATH], *fmt, *buffer;
ds@711: 	int arg = 5;
dvander@345: 	pContext->LocalToString(params[2], &buffer);
dvander@345: 	pContext->LocalToString(params[4], &fmt);
dvander@345: 
dvander@3791: 	smcore.atcprintf(path, sizeof(path), fmt, pContext, params, &arg);
dvander@345: 
dvander@3791: 	return g_pSM->BuildPath(Path_SM_Rel, buffer, params[3], "%s", path);
dvander@345: }
dvander@345: 
dvander@347: static cell_t sm_LogToGame(IPluginContext *pContext, const cell_t *params)
dvander@347: {
dvander@3791: 	g_pSM->SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@1011: 
dvander@347: 	char buffer[1024];
dvander@3791: 	size_t len = g_pSM->FormatString(buffer, sizeof(buffer), pContext, params, 1);
dvander@347: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@1378: 	{
dvander@1378: 		return 0;
dvander@1378: 	}
dvander@1378: 
dvander@347: 	if (len >= sizeof(buffer)-2)
dvander@347: 	{
dvander@347: 		buffer[1022] = '\n';
dvander@347: 		buffer[1023] = '\0';
dvander@347: 	} else {
dvander@347: 		buffer[len++] = '\n';
dvander@347: 		buffer[len] = '\0';
dvander@347: 	}
dvander@347: 
dvander@3791: 	smcore.LogToGame(buffer);
dvander@347: 
dvander@347: 	return 1;
dvander@347: }
dvander@347: 
dvander@347: static cell_t sm_LogMessage(IPluginContext *pContext, const cell_t *params)
dvander@347: {
dvander@3791: 	g_pSM->SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@1011: 
dvander@347: 	char buffer[1024];
dvander@3791: 	g_pSM->FormatString(buffer, sizeof(buffer), pContext, params, 1);
dvander@347: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@1378: 	{
dvander@1378: 		return 0;
dvander@1378: 	}
dvander@1378: 
dvander@3791: 	IPlugin *pPlugin = pluginsys->FindPluginByContext(pContext->GetContext());
dvander@3791: 	smcore.Log("[%s] %s", pPlugin->GetFilename(), buffer);
dvander@347: 
dvander@347: 	return 1;
dvander@347: }
dvander@347: 
dvander@347: static cell_t sm_LogError(IPluginContext *pContext, const cell_t *params)
dvander@347: {
dvander@3791: 	g_pSM->SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@1011: 
dvander@347: 	char buffer[1024];
dvander@3791: 	g_pSM->FormatString(buffer, sizeof(buffer), pContext, params, 1);
dvander@347: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@1378: 	{
dvander@1378: 		return 0;
dvander@1378: 	}
dvander@1378: 
dvander@3791: 	IPlugin *pPlugin = pluginsys->FindPluginByContext(pContext->GetContext());
dvander@3791: 	smcore.LogError("[%s] %s", pPlugin->GetFilename(), buffer);
dvander@347: 
dvander@347: 	return 1;
dvander@347: }
dvander@347: 
dvander@1370: static cell_t sm_GetFileTime(IPluginContext *pContext, const cell_t *params)
dvander@1370: {
dvander@1370: 	char *name;
dvander@1370: 	int err;
dvander@1370: 	if ((err=pContext->LocalToString(params[1], &name)) != SP_ERROR_NONE)
dvander@1370: 	{
dvander@1370: 		pContext->ThrowNativeErrorEx(err, NULL);
dvander@1370: 		return 0;
dvander@1370: 	}
dvander@1370: 
dvander@1738: 	time_t time_val;
dvander@1370: 	char realpath[PLATFORM_MAX_PATH];
dvander@3791: 	g_pSM->BuildPath(Path_Game, realpath, sizeof(realpath), "%s", name);
dvander@1370: 
dvander@3791: 	if (!libsys->FileTime(realpath, (FileTimeType)params[2], &time_val))
dvander@1370: 	{
dvander@1370: 		return -1;
dvander@1370: 	}
dvander@1370: 
dvander@1738: 	return (cell_t)time_val;
dvander@1370: }
dvander@1370: 
dvander@1378: static cell_t sm_LogToOpenFile(IPluginContext *pContext, const cell_t *params)
dvander@1378: {
dvander@1378: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1378: 	HandleError herr;
dvander@1378: 	HandleSecurity sec;
dvander@1378: 	FILE *pFile;
dvander@1378: 
dvander@1378: 	sec.pOwner = NULL;
dvander@1378: 	sec.pIdentity = g_pCoreIdent;
dvander@1378: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1378: 		!= HandleError_None)
dvander@1378: 	{
dvander@1378: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1378: 	}
dvander@1378: 
dvander@1378: 	char buffer[2048];
dvander@3791: 	g_pSM->SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@3791: 	g_pSM->FormatString(buffer, sizeof(buffer), pContext, params, 2);
dvander@1378: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@1378: 	{
dvander@1378: 		return 0;
dvander@1378: 	}
dvander@1378: 
dvander@3791: 	IPlugin *pPlugin = pluginsys->FindPluginByContext(pContext->GetContext());
dvander@3791: 	smcore.LogToFile(pFile, "[%s] %s", pPlugin->GetFilename(), buffer);
dvander@1378: 
dvander@1378: 	return 1;
dvander@1378: }
dvander@1378: 
dvander@1378: static cell_t sm_LogToOpenFileEx(IPluginContext *pContext, const cell_t *params)
dvander@1378: {
dvander@1378: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1378: 	HandleError herr;
dvander@1378: 	HandleSecurity sec;
dvander@1378: 	FILE *pFile;
dvander@1378: 
dvander@1378: 	sec.pOwner = NULL;
dvander@1378: 	sec.pIdentity = g_pCoreIdent;
dvander@1378: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1378: 		!= HandleError_None)
dvander@1378: 	{
dvander@1378: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1378: 	}
dvander@1378: 
dvander@1378: 	char buffer[2048];
dvander@3791: 	g_pSM->SetGlobalTarget(SOURCEMOD_SERVER_LANGUAGE);
dvander@3791: 	g_pSM->FormatString(buffer, sizeof(buffer), pContext, params, 2);
dvander@1378: 
dvander@2193: 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)
dvander@1378: 	{
dvander@1378: 		return 0;
dvander@1378: 	}
dvander@1378: 
dvander@3791: 	smcore.LogToFile(pFile, "%s", buffer);
dvander@1378: 
dvander@1378: 	return 1;
dvander@1378: }
dvander@252: 
dvander@1796: static cell_t sm_ReadFile(IPluginContext *pContext, const cell_t *params)
dvander@1796: {
dvander@1796: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1796: 	HandleError herr;
dvander@1796: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1796: 	FILE *pFile;
dvander@1796: 	size_t read = 0;
dvander@1796: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1796: 		!= HandleError_None)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	if (params[4] != 1 && params[4] != 2 && params[4] != 4)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid size specifier (%d is not 1, 2, or 4)", params[4]);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	cell_t *data;
dvander@1796: 	pContext->LocalToPhysAddr(params[2], &data);
dvander@1796: 
dvander@1796: 	if (params[4] == 4)
dvander@1796: 	{
dvander@1796: 		read = fread(data, sizeof(cell_t), params[3], pFile);
dvander@1796: 	}
dvander@1796: 	else if (params[4] == 2)
dvander@1796: 	{
dvander@2431: 		uint16_t val;
dvander@1796: 		for (cell_t i = 0; i < params[3]; i++)
dvander@1796: 		{
dvander@2431: 			if (fread(&val, sizeof(uint16_t), 1, pFile) != 1)
dvander@1796: 			{
dvander@1796: 				break;
dvander@1796: 			}
dvander@1796: 			data[read++] = val;
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 	else if (params[4] == 1)
dvander@1796: 	{
dvander@2431: 		uint8_t val;
dvander@1796: 		for (cell_t i = 0; i < params[3]; i++)
dvander@1796: 		{
dvander@2431: 			if (fread(&val, sizeof(uint8_t), 1, pFile) != 1)
dvander@1796: 			{
dvander@1796: 				break;
dvander@1796: 			}
dvander@1796: 			data[read++] = val;
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	if (read != (size_t)params[3] && ferror(pFile) != 0)
dvander@1796: 	{
dvander@1796: 		return -1;
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	return read;
dvander@1796: }
dvander@1796: 
dvander@1796: static cell_t sm_ReadFileString(IPluginContext *pContext, const cell_t *params)
dvander@1796: {
dvander@1796: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1796: 	HandleError herr;
dvander@1796: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1796: 	FILE *pFile;
dvander@1796: 	cell_t num_read = 0;
dvander@1796: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1796: 		!= HandleError_None)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	char *buffer;
dvander@1796: 	pContext->LocalToString(params[2], &buffer);
dvander@1796: 
dvander@2445: 	if (params[4] != -1)
dvander@2445: 	{
dvander@2445: 		if (size_t(params[4]) > size_t(params[3]))
dvander@2445: 		{
dvander@2445: 			return pContext->ThrowNativeError("read_count (%u) is greater than buffer size (%u)",
dvander@2445: 				params[4],
dvander@2445: 				params[3]);
dvander@2445: 		}
dvander@2445: 
dvander@2445: 		num_read = (cell_t)fread(buffer, 1, params[4], pFile);
dvander@2445: 
dvander@2445: 		if (num_read != params[4] && ferror(pFile))
dvander@2445: 		{
dvander@2445: 			return -1;
dvander@2445: 		}
dvander@2445: 
dvander@2445: 		return num_read;
dvander@2445: 	}
dvander@2445: 
dvander@1796: 	char val;
dvander@1796: 	while (1)
dvander@1796: 	{
dvander@2445: 		if (params[3] == 0 || num_read >= params[3] - 1)
dvander@1796: 		{
dvander@1796: 			break;
dvander@1796: 		}
dvander@1796: 		if (fread(&val, sizeof(val), 1, pFile) != 1)
dvander@1796: 		{
dvander@1796: 			if (ferror(pFile))
dvander@1796: 			{
dvander@1796: 				return -1;
dvander@1796: 			}
dvander@1796: 			break;
dvander@1796: 		}
dvander@1796: 		if (val == '\0')
dvander@1796: 		{
dvander@1796: 			break;
dvander@1796: 		}
dvander@1796: 		if (params[3] > 0 && num_read < params[3] - 1)
dvander@1796: 		{
dvander@1796: 			buffer[num_read++] = val;
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	if (params[3] > 0)
dvander@1796: 	{
dvander@1796: 		buffer[num_read] = '\0';
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	return num_read;
dvander@1796: }
dvander@1796: 
dvander@1796: static cell_t sm_WriteFile(IPluginContext *pContext, const cell_t *params)
dvander@1796: {
dvander@1796: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1796: 	HandleError herr;
dvander@1796: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1796: 	FILE *pFile;
dvander@1796: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1796: 		!= HandleError_None)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	cell_t *data;
dvander@1796: 	pContext->LocalToPhysAddr(params[2], &data);
dvander@1796: 
dvander@1796: 	if (params[4] != 1 && params[4] != 2 && params[4] != 4)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid size specifier (%d is not 1, 2, or 4)", params[4]);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	/* :NOTE: This really isn't compatible with big endian but we will never have to worry about that. */
dvander@1796: 
dvander@1796: 	if (params[4] == 4)
dvander@1796: 	{
dvander@1796: 		if (fwrite(data, sizeof(cell_t), params[3], pFile) != (size_t)params[3])
dvander@1796: 		{
dvander@1796: 			return 0;
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 	else if (params[4] == 2)
dvander@1796: 	{
dvander@1796: 		for (cell_t i = 0; i < params[3]; i++)
dvander@1796: 		{
dvander@1796: 			if (fwrite(&data[i], sizeof(int16_t), 1, pFile) != 1)
dvander@1796: 			{
dvander@1796: 				return 0;
dvander@1796: 			}
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 	else if (params[4] == 1)
dvander@1796: 	{
dvander@1796: 		for (cell_t i = 0; i < params[3]; i++)
dvander@1796: 		{
dvander@1796: 			if (fwrite(&data[i], sizeof(int8_t), 1, pFile) != 1)
dvander@1796: 			{
dvander@1796: 				return 0;
dvander@1796: 			}
dvander@1796: 		}
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	return 1;
dvander@1796: }
dvander@1796: 
dvander@1796: static cell_t sm_WriteFileString(IPluginContext *pContext, const cell_t *params)
dvander@1796: {
dvander@1796: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1796: 	HandleError herr;
dvander@1796: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1796: 	FILE *pFile;
dvander@1796: 
dvander@3791: 	if ((herr=handlesys->ReadHandle(hndl, g_FileType, &sec, (void **)&pFile))
dvander@1796: 		!= HandleError_None)
dvander@1796: 	{
dvander@1796: 		return pContext->ThrowNativeError("Invalid file handle %x (error %d)", hndl, herr);
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	char *buffer;
dvander@1796: 	pContext->LocalToString(params[2], &buffer);
dvander@1796: 
dvander@1796: 	size_t len = strlen(buffer);
dvander@1796: 
dvander@1796: 	if (params[3])
dvander@1796: 	{
dvander@1796: 		len++;
dvander@1796: 	}
dvander@1796: 
dvander@1796: 	return (fwrite(buffer, sizeof(char), len, pFile) == len) ? 1 : 0;
dvander@1796: }
dvander@1796: 
dvander@1914: static cell_t sm_AddGameLogHook(IPluginContext *pContext, const cell_t *params)
dvander@1914: {
dvander@1914: 	IPluginFunction *pFunction;
dvander@1914: 
dvander@1914: 	if ((pFunction=pContext->GetFunctionById(params[1])) == NULL)
dvander@1914: 	{
dvander@1914: 		return pContext->ThrowNativeError("Function id %x is invalid", params[1]);
dvander@1914: 	}
dvander@1914: 
dvander@1914: 	s_FileNatives.AddLogHook(pFunction);
dvander@1914: 	
dvander@1914: 	return 1;
dvander@1914: }
dvander@1914: 
dvander@1914: static cell_t sm_RemoveGameLogHook(IPluginContext *pContext, const cell_t *params)
dvander@1914: {
dvander@1914: 	IPluginFunction *pFunction;
dvander@1914: 
dvander@1914: 	if ((pFunction=pContext->GetFunctionById(params[1])) == NULL)
dvander@1914: 	{
dvander@1914: 		return pContext->ThrowNativeError("Function id %x is invalid", params[1]);
dvander@1914: 	}
dvander@1914: 
dvander@1914: 	s_FileNatives.RemoveLogHook(pFunction);
dvander@1914: 
dvander@1914: 	return 1;
dvander@1914: }
dvander@1914: 
dvander@252: REGISTER_NATIVES(filesystem)
dvander@252: {
dvander@252: 	{"OpenDirectory",			sm_OpenDirectory},
dvander@252: 	{"ReadDirEntry",			sm_ReadDirEntry},
faluco@274: 	{"OpenFile",				sm_OpenFile},
faluco@274: 	{"DeleteFile",				sm_DeleteFile},
faluco@274: 	{"ReadFileLine",			sm_ReadFileLine},
faluco@274: 	{"IsEndOfFile",				sm_IsEndOfFile},
faluco@274: 	{"FileSeek",				sm_FileSeek},
faluco@274: 	{"FilePosition",			sm_FilePosition},
faluco@274: 	{"FileExists",				sm_FileExists},
faluco@274: 	{"RenameFile",				sm_RenameFile},
faluco@274: 	{"DirExists",				sm_DirExists},
faluco@274: 	{"FileSize",				sm_FileSize},
faluco@274: 	{"RemoveDir",				sm_RemoveDir},
faluco@274: 	{"WriteFileLine",			sm_WriteFileLine},
dvander@345: 	{"BuildPath",				sm_BuildPath},
dvander@636: 	{"LogToGame",				sm_LogToGame},
dvander@636: 	{"LogMessage",				sm_LogMessage},
dvander@347: 	{"LogError",				sm_LogError},
dvander@821: 	{"FlushFile",				sm_FlushFile},
dvander@1370: 	{"GetFileTime",				sm_GetFileTime},
dvander@1378: 	{"LogToOpenFile",			sm_LogToOpenFile},
dvander@1379: 	{"LogToOpenFileEx",			sm_LogToOpenFileEx},
dvander@1796: 	{"ReadFile",				sm_ReadFile},
dvander@1796: 	{"ReadFileString",			sm_ReadFileString},
dvander@1796: 	{"WriteFile",				sm_WriteFile},
dvander@1796: 	{"WriteFileString",			sm_WriteFileString},
dvander@1914: 	{"AddGameLogHook",			sm_AddGameLogHook},
dvander@1914: 	{"RemoveGameLogHook",		sm_RemoveGameLogHook},
dvander@2307: 	{"CreateDirectory",			sm_CreateDirectory},
dvander@252: 	{NULL,						NULL},
dvander@252: };