Luaのdofileで使うファイルシステムを置き換える

Luaはdofile等でスクリプトを読み込む時にfopenを使います。しかし、AndroidのAPKのassetsフォルダはfopenでアクセスできません。独自のファイルシステムを構築したい場合もあります。

そういう時は、以下のようにして、標準のdofileを上書きしてファイルシステムを置き換えます。

	int aflDoFileForReplace(lua_State* L)
	{
	const char* fileName = lua_tostring(L, -1);
	char* img = (char*)LoadFile(fileName);
	if (!img) {
	luaL_error(L, "aflDoFile: could not load file %s", fileName);
	return false;
	}
	bool ok = true;
	if (luaL_loadbuffer(L, img, strlen(img), fileName)) {
	luaL_error(L, "luaL_loadbuffer failed!\n%s", lua_tostring(L, -1));
	lua_pop(L, 1);
	ok = false;
	}
	free(img);
	if (ok && lua_pcall(L, 0, LUA_MULTRET, 0)) {
	luaL_error(L, "lua_pcall failed!\n%s", lua_tostring(L, -1));
	lua_pop(L, 1);
	ok = false;
	}
	return lua_gettop(L) - 1;
	}
	void ReplaceDoFile(lua_State* L)
	{
	lua_register(L, "dofile", aflDoFileForReplace);
	}

view raw replace_dofile.cpp hosted with ❤ by GitHub

LoadFile関数はファイル名を受け取ってmallocで確保したメモリブロックにファイルの内容を格納して返すようにしました。例えばAndroidの場合はassetsフォルダからファイルを持ってくるようにします。このエントリを参考にしてください。

luaL_loadbuffer関数にスクリプトの内容とファイル名を渡します。ファイル名は万一スクリプト内でエラーが発生した場合にLuaがどのファイルの何行目なのかを知らせてくれる為に重要です。

lua_pcallは、読み込んだスクリプトを一度「実行」します。Cのような言語と違い、Luaは関数や変数の定義も「実行」することで行われますので、これをしないとスクリプトを読み込まなかったのと同じことになります。

luaL_loadbuffer又はlua_pcallでエラーが発生した場合、Luaスタックの最上位にエラーメッセージが入ります。ここでは、適当にメッセージを加工してluaL_errorに渡します。これによって、dofileの呼び出し元のスクリプト側のエラーとして処理されます。

最後のreturn lua_gettop(L) - 1;lua_pcallがスタック何かを積んだ場合、その個数を呼び出し元に返します。-1するのはdofileが受け取ったファイル名がスタックを一つ使っているのでそれを除外するためです。説明より実例のほうがわかりやすいと思うので、以下に書いてみました。例えば、以下のような事ができるようになります。

return "A message from callee!"

view raw callee.lua hosted with ❤ by GitHub

	local a = dofile("callee.lua")
	print(a)

view raw caller.lua hosted with ❤ by GitHub

呼びだされたcallee.luaの最後の行にreturn文があり、その戻り値をcaller.luaで受け取っています。こういった使い方が必要ない場合は、return lua_gettop(L) - 1;はreturn 0;に置き換えても不都合はありません。ただし、caller.luaはdofileから戻り値を受け取れなくなります。