lua 的 WriteFile 和 ReadFile

时间:2022-07-31 19:10:57
lua 的 ReadFile 和 WriteFile

注意:
lua 代码中的 string 可以包含二进制数据 string.len(str) 不会被 '\0' 截断
从 lua 传入 c 时: 需要传入包含二进制数据的 string 和数据的 size, size 用 string.len(str) 来计算
从 c 传入 lua 时: 应该使用 lua_pushlstring 而不是 lua_pushstring, lua_pushstring 内部使用 strlen() 来计算数据的长度,而 lua_pushlstring 需要传入一个 size int LuaFileSystem::ReadFile( lua_State* luaState )
{
if (!lua_isstring(luaState, ))
{
return ;
}
const char* strPath = luaL_checkstring(luaState, );
std::wstring wstrPath;
TransCode::UTF8_to_Unicode(strPath, strlen(strPath), wstrPath);
HANDLE hFile = ::CreateFile(wstrPath.c_str(), GENERIC_READ, , NULL, OPEN_EXISTING, NULL, NULL);
if (hFile != INVALID_HANDLE_VALUE)
{
DWORD nFileSize = ::GetFileSize(hFile, NULL);
char* lpBuffer = new char[nFileSize];
DWORD nNumberOfBytesRead;
BOOL bRet = ::ReadFile(hFile, lpBuffer, nFileSize, &nNumberOfBytesRead, NULL); lua_pushlstring(luaState, lpBuffer, nNumberOfBytesRead);
delete[] lpBuffer;
CloseHandle(hFile);
return ;
}
return ;
} int LuaFileSystem::WriteFile( lua_State* luaState )
{
if (!lua_isstring(luaState, ) || !lua_isstring(luaState, ))
{
return ;
}
const char* strPath = luaL_checkstring(luaState, );
const char* strContent = luaL_checkstring(luaState, );
int nContentSize = strlen(strContent);
if (lua_isnumber(luaState, ))
{
nContentSize = luaL_checkint(luaState, );
}
std::wstring wstrPath;
TransCode::UTF8_to_Unicode(strPath, strlen(strPath), wstrPath);
HANDLE hFile = ::CreateFile(wstrPath.c_str(), GENERIC_WRITE, , NULL, CREATE_ALWAYS, NULL, NULL);
if (hFile != INVALID_HANDLE_VALUE)
{
DWORD nNumberOfBytesWritten;
::WriteFile(hFile, strContent, nContentSize, &nNumberOfBytesWritten, NULL);
CloseHandle(hFile);
}
return ;
}