App/script/CoreScript.cpp - main - BirdLeon/ROBLOX2016

App/script/CoreScript.cpp
116 строк3 KB
PatoFlamejanteTV
full source code
19 дек 2024, 19:11
19 дек 2024, 19:1105db15d
#include "stdafx.h"

#include "Script/CoreScript.h"
#include "script/ScriptContext.h"

#include "V8DataModel/DataModel.h"
#include "v8datamodel/ContentProvider.h"
#include "Lua/Lua.hpp"
#include "util/FileSystem.h"
#include "rbx/Log.h"

#include "StringConv.h"

namespace RBX
{
	const char* const sCoreScript = "CoreScript";

	CoreScript::CoreScript()
		:Super()
	{
		setName(sCoreScript);
		setRobloxLocked(true);
	}

	// used to make sure corescripts not in the workspace still get removed properly
	void CoreScript::onServiceProvider(ServiceProvider* oldProvider, ServiceProvider* newProvider)
	{
		if(oldProvider && !workspace)
		{
			RBX::ScriptContext* sc = ServiceProvider::find<ScriptContext>(oldProvider);
			RBXASSERT(sc);

			RBXASSERT(sc->hasScript(this));
			sc->removeScript(weak_from(this));	
		}
		Super::onServiceProvider(oldProvider, newProvider);
	}

    boost::optional<ProtectedString> CoreScript::fetchSource(const std::string& name)
    {
        if (LuaVM::canCompileScripts())
        {
            std::string path = BaseScript::hasCoreScriptReplacements() ? BaseScript::adminScriptsPath : ContentProvider::assetFolder() + "scripts";
            std::string file = path + "/" + name + ".lua";

            std::ifstream in(utf8_decode(file).c_str(), std::ios::in | std::ios::binary);
            
            if (!in)
                return boost::optional<ProtectedString>();
            
            std::stringstream ss;
            ss << in.rdbuf();
            
            return ProtectedString::fromTrustedSource(ss.str());
        }
        else
        {
            std::string bytecode = LuaVM::getBytecodeCore(name);
            
            if (bytecode.empty())
                return boost::optional<ProtectedString>();
            else
                return ProtectedString::fromBytecode(bytecode);
        }
    }

	BaseScript::Code CoreScript::requestCode(ScriptInformationProvider* scriptInfoProvider)
	{
        if (boost::optional<ProtectedString> source = fetchSource(getName()))
        {
            return Code(boost::flyweight<ProtectedString>(*source));
        }
        else
        {
            throw RBX::runtime_error("Error loading core script %s", getName().c_str());
        }
    }

	// log the error in a special file if it's from a CoreScript
	void CoreScript::extraErrorReporting(lua_State *thread)
	{
		std::stringstream out;
		
		// format and collect error information
		const char* error = RBX::Lua::safe_lua_tostring(thread, -1);
		if (error && strlen(error)>0)
			out << error;
		else
			out << "Error occurred, no output from Lua.";
			
		int line;
		shared_ptr<BaseScript> source;
		out << "\n\nCallStack:\n" << ScriptContext::extractCallStack(thread, source, line);
		RBXASSERT(source.get() == this);

		if(DataModel* dataModel = DataModel::get(this))
			out << "\nPlaceID: " << dataModel->getPlaceIDOrZeroInStudio();
		else
			out << "\nError finding PlaceID!";
		std::string errorReport = out.str();
		
		// create cse file, later to be uploaded to s3 (same place as .dmp)
		std::stringstream fileOut;
		fileOut << RBX::FileSystem::getUserDirectory(true, RBX::DirAppData, "logs") << this->getName() << "_ln" << line << "_" << ".cse";
		std::string filename = fileOut.str();
		
		std::ofstream errorLog;
		errorLog.open(filename.c_str());
		if(errorLog.is_open())
		{
			RBX::Log::timeStamp(errorLog, true);
			errorLog << errorReport;
		}
		errorLog.close();
	}
}