nstool/lib/libfnd/source/io.cpp

#include <fnd/io.h>
#include <fnd/SimpleFile.h>
#include <fstream>
#ifdef _WIN32
#include <direct.h>
#include <cstdlib>
#else
#include <sys/stat.h>
#endif

using namespace fnd;

static const std::string kModuleName = "IO";
static const size_t kBlockSize = 0x100000;

size_t io::getFileSize(const std::string& path)
{
	std::ifstream f;
	f.open(path, std::ios_base::binary | std::ios_base::in);
	if (!f.good() || f.eof() || !f.is_open()) { return 0; }
	f.seekg(0, std::ios_base::beg);
	std::ifstream::pos_type begin_pos = f.tellg();
	f.seekg(0, std::ios_base::end);
	return static_cast<size_t>(f.tellg() - begin_pos);
}

void io::readFile(const std::string& path, MemoryBlob & blob)
{
	fnd::SimpleFile file;
	try
	{
		file.open(path, file.Read);
	}
	catch (const fnd::Exception&)
	{
		throw fnd::Exception(kModuleName, "Failed to open \"" + path + "\": does not exist");
	}

	size_t filesz, filepos;
	filesz = file.size();


	try {
		blob.alloc(filesz);
	}
	catch (const fnd::Exception& e)
	{
		throw fnd::Exception(kModuleName, "Failed to allocate memory for file: " + std::string(e.what()));
	}

	for (filepos = 0; filesz > kBlockSize; filesz -= kBlockSize, filepos += kBlockSize)
	{
		file.read(blob.getBytes() + filepos, kBlockSize);
	}

	if (filesz)
	{
		file.read(blob.getBytes() + filepos, filesz);
	}
}

void fnd::io::readFile(const std::string& path, size_t offset, size_t len, MemoryBlob& blob)
{
	fnd::SimpleFile file;
	try
	{
		file.open(path, file.Read);
	} catch (const fnd::Exception&)
	{
		throw fnd::Exception(kModuleName, "Failed to open \"" + path + "\": does not exist");
	}

	size_t filesz, filepos;
	filesz = file.size();
	file.seek(offset);

	if (filesz < len || filesz < offset || filesz < (offset + len))
	{
		throw Exception(kModuleName, "Failed to open \"" + path + "\": file to small");
	}

	try
	{
		blob.alloc(filesz);
	} catch (const fnd::Exception& e)
	{
		throw fnd::Exception(kModuleName, "Failed to allocate memory for file: " + std::string(e.what()));
	}

	for (filepos = 0; filesz > kBlockSize; filesz -= kBlockSize, filepos += kBlockSize)
	{
		file.read(blob.getBytes() + filepos, kBlockSize);
	}

	if (filesz)
	{
		file.read(blob.getBytes() + filepos, filesz);
	}
}

void io::writeFile(const std::string& path, const MemoryBlob & blob)
{
	writeFile(path, blob.getBytes(), blob.getSize());
}

void io::writeFile(const std::string & path, const byte_t * data, size_t len)
{
	fnd::SimpleFile file;
	try
	{
		file.open(path, file.Create);
	} catch (const fnd::Exception&)
	{
		throw fnd::Exception(kModuleName, "Failed to open \"" + path + "\": does not exist");
	}

	size_t filesz, filepos;

	filesz = len;


	for (filepos = 0; filesz > kBlockSize; filesz -= kBlockSize, filepos += kBlockSize)
	{
		file.write(data + filepos, kBlockSize);
	}

	if (filesz)
	{
		file.write(data + filepos, filesz);
	}
}

void io::makeDirectory(const std::string& path)
{
#ifdef _WIN32
	_mkdir(path.c_str());
#else
	mkdir(path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
#endif
}

void fnd::io::getEnvironVar(std::string & var, const std::string & key)
{
#ifdef _WIN32
	char* var_tmp = nullptr;
	size_t var_len = 0;
	_dupenv_s(&var_tmp, &var_len, key.c_str());

	if (var_len > 0)
	{
		var = std::string(var_tmp);
		free(var_tmp);
	}
#else
	char* var_tmp = nullptr;

	var_tmp = getenv(key.c_str());

	if (var_tmp != nullptr)
	{
		var = std::string(var_tmp);
	}
#endif
}

void fnd::io::makePath(std::string & out, const std::vector<std::string>& elements)
{
	out.clear();
	out = "";
	for (size_t i = 0; i < elements.size(); i++)
	{
		if (i > 0)
		{
#ifdef _WIN32
			out += "\\";
#else
			out += "/";
#endif
		}
		out += elements[i];
	}
}