KSG-Framework/Datei.cpp

#include "Datei.h"
#include "Text.h"
#include "Zeit.h"
#include "Key.h"
#ifdef WIN32
#include <direct.h>
#include <Shlwapi.h>
#pragma comment(lib, "Shlwapi.lib")
#else
#include <dirent.h>
#include <sys/stat.h>
#include <stdio.h>
#endif

using namespace Framework;
using namespace Encryption;

// Inhalt der Datei Klasse aus Datei.h
// Konstruktor 
Datei::Datei()
	: ReferenceCounter(),
	stream(0),
	pfad(0),
	gr(0),
	tmpLByte(0),
	tmpLBPos(7),
	tmpSByte(0),
	tmpSBPos(-1),
	key(0)
{}

//! Konstruktor 
Datei::Datei(const char* pfad)
	: Datei()
{
	setDatei(pfad);
}

//! Konstruktor 
Datei::Datei(Text* pfad)
	: Datei()
{
	setDatei(pfad);
}

// Destruktor 
Datei::~Datei()
{
	if (key)
		key->release();
	if (stream)
		delete stream;
	if (pfad)
		pfad->release();
}

// nicht constant 
void Datei::setDatei(const char* pfad) // setzt die Datei
{
	if (istOffen())
		close();
	if (!this->pfad)
		this->pfad = new Text();
	this->pfad->setText(pfad);
	gr = 0;
}

void Datei::setDatei(Text* pfad)
{
	if (istOffen())
		close();
	if (!this->pfad)
		this->pfad = new Text();
	this->pfad->setText(pfad);
	gr = 0;
}

bool Datei::umbenennen(const char* pfad) // benennt die Datei um und verschiebt sie eventuell
{
	if (!pfad)
		return 0;
	if (DateiUmbenennen(this->pfad->getText(), pfad))
	{
		this->pfad->setText(pfad);
		return 1;
	}
	return 0;
}

bool Datei::umbenennen(Text* pfad)
{
	if (!this->pfad)
	{
		pfad->release();
		return 0;
	}
	if (DateiUmbenennen(this->pfad->getText(), pfad->getText()))
	{
		this->pfad->setText(pfad);
		return 1;
	}
	pfad->release();
	return 0;
}

bool Datei::remove() // löscht die Datei
{
	if (!pfad)
		return 0;
	return DateiRemove(dynamic_cast<Text*>(pfad->getThis()));
}

bool Datei::erstellen() // erstellt die Datei
{
	if (!pfad)
		return 0;
	return DateiPfadErstellen(dynamic_cast<Text*>(pfad->getThis()));
}

bool Datei::open(int style) // öffnet die Datei
{
	if (!pfad)
		return 0;
	if (stream)
		delete stream;
	stream = new std::fstream();
	std::ios_base::openmode om = std::ios::binary;
	if ((style | Style::lesen) == style)
		om |= std::ios::in;
	if ((style | Style::schreiben) == style)
		om |= std::ios::out;
	stream->open(pfad->getText(), om);
	if ((style | Style::ende) == style)
	{
		if ((style | Style::lesen) == style)
			stream->seekg(0, std::ios::end);
		if ((style | Style::schreiben) == style)
			stream->seekp(0, std::ios::end);
	}
	if (!stream->is_open() || !stream->good())
	{
		delete stream;
		stream = 0;
		return 0;
	}
	tmpLBPos = 7;
	tmpSBPos = -1;
	return 1;
}

void Datei::setLPosition(__int64 pos, bool ende) // setzt die Leseposition
{
	if (!pfad)
		return;
	if (stream)
	{
		if (ende)
			stream->seekg(pos, std::ios::end);
		else
			stream->seekg(pos, std::ios::beg);
	}
	tmpLBPos = 7;
}

void Datei::setSPosition(__int64 pos, bool ende) // setzt die Schreibeposition
{
	if (!pfad)
		return;
	if (stream)
	{
		if (ende)
			stream->seekp(pos, std::ios::end);
		else
			stream->seekp(pos, std::ios::beg);
	}
	tmpSBPos = -1;
}

void Datei::schreibe(const char* bytes, int len) // schreibt bytes in datei
{
	if (!pfad || !stream)
		return;
	if (tmpSBPos >= 0)
	{
		tmpSBPos = -1;
		stream->write(&tmpSByte, 1);
		tmpSByte = 0;
	}
	if (key)
	{
		key->setPos(getSPosition());
		Bytes* n = new Bytes(bytes, len);
		key->codieren(dynamic_cast<Bytes*>(n->getThis()));
		stream->write(n->getBytes(), len);
		n->release();
	}
	else
		stream->write(bytes, len);
}

void Datei::lese(char* bytes, int len) // ließt bytes aus datei
{
	if (!pfad)
		return;
	if (stream)
	{
		__int64 tmp = getLPosition();
		stream->read(bytes, len);
		if (key)
		{
			key->setPos(tmp);
			Bytes* n = new Bytes();
			n->setBytesZ(bytes, len);
			key->decodieren(n);
		}
	}
	tmpLBPos = 7;
	tmpSBPos = -1;
}

Text* Datei::leseZeile() // ließt eine zeile
{
	if (!pfad || !stream)
		return 0;
	if (istEnde())
		return 0;
	Text* ret = new Text("");
	__int64 len = getSize();
	for (char c = 0; c != '\n' && stream->tellg() < len; )
	{
		__int64 tmp = getLPosition();
		stream->read(&c, 1);
		if (key)
		{
			key->setPos(tmp);
			Bytes* n = new Bytes();
			n->setBytesZ(&c, 1);
			key->decodieren(n);
		}
		if (c)
			ret->append(&c, 1);
	}
	tmpSBPos = 7;
	tmpSBPos = -1;
	return ret;
}

void Datei::close() // schließt die Datei
{
	if (!pfad || !stream)
		return;
	if (tmpSBPos >= 0)
	{
		if (key)
		{
			key->setPos(getSPosition());
			Bytes* n = new Bytes(&tmpSByte, 1);
			key->codieren(dynamic_cast<Bytes*>(n->getThis()));
			stream->write(n->getBytes(), 1);
			n->release();
		}
		else
			stream->write(&tmpSByte, 1);
	}
	stream->close();
	delete stream;
	stream = 0;
}

#ifdef WIN32
bool Datei::setLetzteÄnderung(Zeit* zeit) // setzt das änderungsdatum der Datei
{
	if (!pfad)
	{
		zeit->release();
		return 0;
	}
	HANDLE hFile = CreateFile(pfad->getText(), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
	if (hFile == INVALID_HANDLE_VALUE)
	{
		zeit->release();
		return 0;
	}
	FILETIME ftCreate, ftAccess, ftWrite;
	if (!GetFileTime(hFile, &ftCreate, &ftAccess, &ftWrite))
	{
		CloseHandle(hFile);
		zeit->release();
		return 0;
	}
	SYSTEMTIME stUTC, stLocal;
	stLocal.wMilliseconds = 0;
	stLocal.wSecond = zeit->zUhrzeit()->getSekunde();
	stLocal.wMinute = zeit->zUhrzeit()->getMinute();
	stLocal.wHour = zeit->zUhrzeit()->getStunde();
	stLocal.wDay = zeit->zDatum()->getTag();
	stLocal.wMonth = zeit->zDatum()->getMonat();
	stLocal.wYear = zeit->zDatum()->getJahr();
	zeit->release();
	if (!TzSpecificLocalTimeToSystemTime(NULL, &stLocal, &stUTC))
	{
		CloseHandle(hFile);
		return 0;
	}
	if (!SystemTimeToFileTime(&stUTC, &ftWrite))
	{
		CloseHandle(hFile);
		return 0;
	}
	if (!SetFileTime(hFile, &ftCreate, &ftAccess, &ftWrite))
	{
		CloseHandle(hFile);
		return 0;
	}
	CloseHandle(hFile);
	return 1;
}
#endif

bool Datei::getNextBit(bool& bit) // Datei Bitweise auslesen
{
	if (!pfad || !stream)
		return 0;
	if (tmpLBPos == 7)
	{
		tmpLBPos = -1;
		__int64 tmp = getLPosition();
		stream->read(&tmpLByte, 1);
		if (key)
		{
			key->setPos(tmp);
			Bytes* n = new Bytes();
			n->setBytesZ(&tmpLByte, 1);
			key->decodieren(n);
		}
	}
	tmpLBPos++;
	bit = (tmpLByte >> (7 - tmpLBPos)) & 1;
	return 1;
}

bool Datei::setNextBit(bool bit) // Datei Bitweise speichern
{
	if (!pfad || !stream)
		return 0;
	tmpSBPos++;
	tmpSByte |= (char)(((char)bit << (7 - tmpSBPos)) & (1 << (7 - tmpSBPos)));
	if (tmpSBPos == 7)
	{
		tmpSBPos = -1;
		if (key)
		{
			key->setPos(getSPosition());
			Bytes* n = new Bytes(&tmpSByte, 1);
			key->codieren(dynamic_cast<Bytes*>(n->getThis()));
			stream->write(n->getBytes(), 1);
			n->release();
		}
		else
			stream->write(&tmpSByte, 1);
		tmpSByte = 0;
	}
	return 1;
}

// Setzt den Schlüssel für die Datei
void Datei::setKey(char* s, int l)
{
	if (l == 0)
	{
		key = (Key*)key->release();
		return;
	}
	if (key)
		key->setKey(s, l);
	else
		key = new Key(s, l);
}

// constant 
bool Datei::istOrdner() const // prüft, ob die Datei ein Ordner ist
{
	if (!pfad)
		return 0;
	return DateiIstVerzeichnis(dynamic_cast<Text*>(pfad->getThis()));
}

bool Datei::istOffen() const // prüft, ob die Datei geöffnet ist
{
	if (!pfad)
		return 0;
	if (stream)
		return stream->is_open() && stream->good();
	return 0;
}

int Datei::getUnterdateiAnzahl() const // gibt die Anzahl der unterdateien an
{
#ifdef WIN32
	if (!pfad)
		return 0;
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfad->getThis())))
		return 0;
	int ret = 0;
	HANDLE fHandle;
	WIN32_FIND_DATA wfd;
	Text stxt = pfad->getText();
	stxt.ersetzen('/', '\\');
	if (stxt.positionVon('\\') == stxt.getLength() - 1)
		stxt.append("*");
	else
		stxt.append("\\*");
	fHandle = FindFirstFile(stxt.getText(), &wfd);
	FindNextFile(fHandle, &wfd);
	while (FindNextFile(fHandle, &wfd))
		++ret;
	FindClose(fHandle);
	return ret;
#else
	if (!pfad)
		return 0;
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfad->getThis())))
		return 0;
	int ret = 0;
	Text stxt = pfad->getText();
	stxt.ersetzen('\\', '/');
	if (stxt.positionVon('/') == stxt.getLength() - 1)
		stxt.remove(stxt.getLength() - 1);
	DIR* hdir;
	hdir = opendir(stxt.getText());
	for (dirent* entry = readdir(hdir); entry; entry = readdir(hdir))
	{
		if (entry && entry->d_name[0] != '.')
			++ret;
	}
	closedir(hdir);
	return ret;
#endif
}

RCArray< Text >* Datei::getDateiListe() const // gibt eine Liste mit unterdateien zurück
{
#ifdef WIN32
	if (!pfad)
		return 0;
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfad->getThis())))
		return 0;
	HANDLE fHandle;
	WIN32_FIND_DATA wfd;
	Text stxt = pfad->getText();
	stxt.ersetzen('/', '\\');
	if (stxt.positionVon('\\') == stxt.getLength() - 1)
		stxt.append("*");
	else
		stxt.append("\\*");
	fHandle = FindFirstFile(stxt.getText(), &wfd);
	FindNextFile(fHandle, &wfd);
	RCArray< Text >* ret = new RCArray< Text >();
	int count = 0;
	while (FindNextFile(fHandle, &wfd))
	{
		Text* txt = new Text(wfd.cFileName);
		ret->add(txt, count);
		++count;
	}
	FindClose(fHandle);
	return ret;
#else
	if (!pfad)
		return 0;
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfad->getThis())))
		return 0;
	Text stxt = pfad->getText();
	stxt.ersetzen('\\', '/');
	if (stxt.positionVon('/') == stxt.getLength() - 1)
		stxt.remove(stxt.getLength() - 1);
	DIR* hdir;
	hdir = opendir(stxt.getText());
	if (hdir)
	{
		RCArray< Text >* ret = new RCArray< Text >();
		int count = 0;
		for (dirent* entry = readdir(hdir); entry; entry = readdir(hdir))
		{
			if (entry && entry->d_name[0] != '.')
			{
				ret->add(new Text(entry->d_name), count);
				++count;
			}
		}
		closedir(hdir);
		return ret;
	}
	return 0;
#endif
}

__int64 Datei::getSize() const // gibt die Größe der Datei zurück
{
	if (!pfad)
		return 0;
	if (gr)
		return gr;
	if (!stream || !istOffen())
	{
		std::fstream* stream = new std::fstream();
		stream->open(pfad->getText(), std::ios::binary | std::ios::in);
		__int64 tmp = stream->tellg();
		stream->seekg(0, std::ios::end);
		__int64 ret = stream->tellg();
		stream->seekg(tmp, std::ios::beg);
		stream->close();
		delete stream;
		__int64* size = (__int64*)&gr;
		*size = ret;
		return ret;
	}
	__int64 tmp = stream->tellg();
	stream->seekg(0, std::ios::end);
	__int64 ret = stream->tellg();
	stream->seekg(tmp, std::ios::beg);
	__int64* size = (__int64*)&gr;
	*size = ret;
	return ret;
}

Zeit* Datei::getLastChange() const // gibt das Datum der letzten Änderung
{
	if (!pfad)
		return 0;
#ifdef WIN32
	HANDLE hFile = CreateFile(pfad->getText(), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
	if (hFile == INVALID_HANDLE_VALUE)
		return 0;
	FILETIME ftCreate, ftAccess, ftWrite;
	SYSTEMTIME stUTC, stLocal;
	if (!GetFileTime(hFile, &ftCreate, &ftAccess, &ftWrite))
	{
		CloseHandle(hFile);
		return 0;
	}
	CloseHandle(hFile);
	if (!FileTimeToSystemTime(&ftWrite, &stUTC))
		return 0;
	if (!SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal))
		return 0;
	Zeit* ret = new Zeit();
	ret->setZeit(stLocal.wYear, stLocal.wMonth, stLocal.wDay, stLocal.wHour, stLocal.wMinute, stLocal.wSecond);
	return ret;
#else
	struct stat attrib;
	if (stat(pfad->getText(), &attrib) != 0)
		return 0;
	tm* clock = gmtime(&(attrib.st_mtime));
	Zeit* ret = new Zeit();
	ret->setZeit(clock->tm_year + 1900, clock->tm_mon + 1, clock->tm_mday, clock->tm_hour, clock->tm_min, clock->tm_sec);
	return ret;
#endif
}

bool Datei::existiert() const // prüft, ob die Datei existiert
{
	if (!pfad)
		return 0;
	return DateiExistiert(dynamic_cast<Text*>(pfad->getThis()));
}

__int64 Datei::getLPosition() const // gibt die Leseposition zurück
{
	if (!stream)
		return 0;
	return stream->tellg();
}

__int64 Datei::getSPosition() const // gibt die Schreibeposition zurück
{
	if (!stream)
		return 0;
	return stream->tellp();
}

bool Datei::istEnde() const // prüft, ob die Datei zu ende ist
{
	if (!stream || stream->tellg() < 0)
		return 1;
	__int64 i = getSize();
	return stream->tellg() >= i;
}

Text* Datei::getPfad() const // gibt den Dateipfad zurück
{
	return pfad ? dynamic_cast<Text*>(pfad->getThis()) : 0;
}

Text* Datei::zPfad() const
{
	return pfad;
}

// Datei Funktionen
void Framework::GetFreePfad(Text* zPfad) // Sucht einen unbenutzten Dateinamen
{
	Text txt = zPfad->getText();
	for (int i = 0; DateiExistiert(txt); i++)
	{
		txt = zPfad->getText();
		txt.append(i);
	}
	zPfad->setText(txt);
}

bool Framework::DateiPfadErstellen(Text* pfad) // Erstellt eine Datei in dem Pfad
{
	bool ret = DateiPfadErstellen(pfad->getText());
	pfad->release();
	return ret;
}

bool Framework::DateiRemove(Text* pfad) // Löscht die angegebene Datei
{
	bool ret = DateiRemove(pfad->getText());
	pfad->release();
	return ret;
}

bool Framework::DateiUmbenennen(Text* pfad_alt, Text* pfad_neu) // Benennt die Datei um
{
	bool ret = DateiUmbenennen(pfad_alt->getText(), pfad_neu->getText());
	pfad_alt->release();
	pfad_neu->release();
	return ret;
}

bool Framework::DateiExistiert(Text* pfad) // Prüft, ob Datei existiert
{
	bool ret = DateiExistiert(pfad->getText());
	pfad->release();
	return ret;
}

bool Framework::DateiIstVerzeichnis(Text* pfad) // prüft, ob pfad ein Verzeichnis ist
{
	bool ret = DateiIstVerzeichnis(pfad->getText());
	pfad->release();
	return ret;
}

bool Framework::DateiPfadErstellen(const char* pfad) // Erstellt eine Datei in dem Pfad
{
	Text pf = pfad;
	bool erst = 1;
#ifdef WIN32
	pf.ersetzen("//", "\\"); // Pfadangaben korrigieren
	pf.ersetzen("/", "\\");
	for (int i = 0; i < pf.anzahlVon("\\"); ++i) // Jeden ordner erstellen wenn er nicht existiert
	{
		Text* t = pf.getTeilText(0, pf.positionVon("\\", i));
		if (!t || !t->getLength())
		{
			if (t)
				t->release();
			continue;
		}
		if (!DateiExistiert(dynamic_cast<Text*>(t->getThis())))
#pragma warning(suppress: 6031)
			_mkdir(t->getText());
		t->release();
		if (pf.positionVon("\\", i) == pf.getLength() - 1)
			erst = 0;
	}
#else
	pf.ersetzen("\\", "/"); // Pfadangaben korrigieren
	for (int i = 0; i < pf.anzahlVon("/"); ++i) // Jeden ordner erstellen wenn er nicht existiert
	{
		Text* t = pf.getTeilText(0, pf.positionVon("/", i));
		if (!t || !t->getLength())
		{
			if (t)
				t->release();
			continue;
		}
		if (!DateiExistiert(dynamic_cast<Text*>(t->getThis())))
			mkdir(t->getText(), 0777);
		t->release();
		if (pf.positionVon("\\", i) == pf.getLength() - 1)
			erst = 0;
	}
#endif
	if (erst)
	{
		std::ofstream f(pf, std::ios::binary); // Datei erstellen
		f.close();
	}
	return DateiExistiert(pf);
}

bool Framework::DateiRemove(const char* pfad) // Löscht die angegebene Datei
{
	Text pfa = pfad;
#ifdef WIN32
	pfa.ersetzen('\\', '/');
	bool ret = 0;
	// prüfen ob Datei existiert
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfa.getThis())))
		ret = DeleteFile(pfa.getText()) == 1; // Datei löschen
	else
	{
		ret = 1;
		Datei* dat = new Datei();
		dat->setDatei(dynamic_cast<Text*>(pfa.getThis()));
		int anz = dat->getUnterdateiAnzahl();
		RCArray< Text >* liste = dat->getDateiListe();
		for (int i = 0; i < anz; ++i)
		{
			Text* pf = new Text(pfa.getText());
			if (pf->getText()[pf->getLength() - 1] != '/')
				pf->append("/");
			pf->append(liste->get(i));
			if (ret)
				ret = DateiRemove(pf);
			else
				DateiRemove(pf);
		}
		liste->release();
		dat->release();
		if (ret)
			ret = RemoveDirectory(pfa.getText()) == 1;
		else
			RemoveDirectory(pfa.getText());
	}
	return ret;
#else
	pfa.ersetzen('\\', '/');
	bool ret = 0;
	// pruefen ob Datei existiert
	if (!DateiIstVerzeichnis(dynamic_cast<Text*>(pfa.getThis())))
		ret = std::remove(pfa.getText()) == 0; // Datei loeschen
	else
	{
		ret = 1;
		Datei* dat = new Datei();
		dat->setDatei(dynamic_cast<Text*>(pfa.getThis()));
		int anz = dat->getUnterdateiAnzahl();
		RCArray< Text >* liste = dat->getDateiListe();
		for (int i = 0; i < anz; ++i)
		{
			Text* pf = new Text(pfa.getText());
			if (pf->getText()[pf->getLength() - 1] != '/')
				pf->append("/");
			pf->append(liste->get(i));
			if (ret)
				ret = DateiRemove(pf);
			else
				DateiRemove(pf);
		}
		liste->release();
		dat->release();
		if (ret)
			ret = std::remove(pfa.getText()) == 0;
		else
			std::remove(pfa.getText());
	}
	return ret;
#endif
}

bool Framework::DateiUmbenennen(const char* pfad_alt, const char* pfad_neu) // Benennt die Datei um
{
#ifdef WIN32
	if (pfad_alt && pfad_neu && DateiExistiert(pfad_alt))
	{
		bool ret = 1;
		if (DateiIstVerzeichnis(pfad_alt))
		{
			if (!DateiExistiert(pfad_neu))
			{
				Text tmp = pfad_neu;
				tmp += "/a";
				DateiPfadErstellen(tmp);
				DateiRemove(tmp);
			}
			Datei d;
			d.setDatei(pfad_alt);
			RCArray< Text >* list = d.getDateiListe();
			int anz = list->getEintragAnzahl();
			for (int i = 0; i < anz; i++)
			{
				Text pf = pfad_neu;
				pf += "/";
				pf += list->z(i)->getText();
				Text pf_a = pfad_alt;
				pf_a += "/";
				pf_a += list->z(i)->getText();
				ret |= DateiUmbenennen(pf_a, pf);
			}
			d.remove();
		}
		else
		{
			if (DateiExistiert(pfad_neu))
				return 0;
		}
		ret |= MoveFile(pfad_alt, pfad_neu) == 1; // Datei umbenennen
		return ret;
	}
	return 0;
#else
	if (pfad_alt && pfad_neu && DateiExistiert(pfad_alt))
	{
		bool ret = 1;
		if (DateiIstVerzeichnis(pfad_alt))
		{
			if (!DateiExistiert(pfad_neu))
			{
				Text tmp = pfad_neu;
				tmp += "/a";
				DateiPfadErstellen(tmp);
				DateiRemove(tmp);
			}
			Datei d;
			d.setDatei(pfad_alt);
			RCArray< Text >* list = d.getDateiListe();
			int anz = list->getEintragAnzahl();
			for (int i = 0; i < anz; i++)
			{
				Text pf = pfad_neu;
				pf += "/";
				pf += list->z(i)->getText();
				Text pf_a = pfad_alt;
				pf_a += "/";
				pf_a += list->z(i)->getText();
				ret |= DateiUmbenennen(pf_a, pf);
			}
			d.remove();
		}
		else
		{
			if (DateiExistiert(pfad_neu))
				return 0;
		}
		ret |= rename(pfad_alt, pfad_neu) == 1; // Datei umbenennen
		return ret;
	}
	return 0;
#endif
}

bool Framework::DateiExistiert(const char* pfad) // Prüft, ob Datei existiert
{
#ifdef WIN32
	bool ret = PathFileExists(pfad) != 0;
	return ret;
#else
	std::ifstream file(pfad);
	if (file.good())
		return 1;
	return 0;
#endif
}

bool Framework::DateiIstVerzeichnis(const char* pfad) // prüft, ob pfad ein Verzeichnis ist
{
#ifdef WIN32
	WIN32_FIND_DATA wfd;
	HANDLE handle = FindFirstFile(pfad, &wfd);
	if (handle == INVALID_HANDLE_VALUE)
		return 0;
	FindClose(handle);
	return (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
#else
	struct stat path_stat;
	if (stat(pfad, &path_stat) != 0)
		return 0;
	if (S_ISDIR(path_stat.st_mode))
		return 1;
	return 0;
#endif
}