jjj

#include "stdafx.h"
#include "TemplateDLL.h"
#include <iomanip>
#include <filesystem>

#include <vector>
#include "User.h"

std::vector<User> readFile(std::wstring fileName); //read file
std::wifstream & operator >> (std::wifstream & wifs, User & user); //read each user
void tryToOpenFile(std::wstring& logText, std::wstring filePath);
void tryToCreateFile(std::wstring& logText, std::wstring filePath);
void tryToReadFile(std::wstring& logText, std::wstring filePath);
void tryToReadAllFiles(std::wstring& logText, std::wstring filePath);

namespace fs = std::experimental::filesystem;

extern "C"
{
	LPCWSTR GetName()
	{
		return m_sName;
	}

	LPCWSTR GetDescription()
	{
		return m_sDescription;
	}

	bool RunTest(std::function<void(LPCWSTR)> pFnLog)
	{
		std::wstring logText = L"log text:\n"; //test output
		// test code
		std::wstring confFileName = L"conf.ini"; //configuration file
		std::vector<User> data;  //data: users
		data = readFile(confFileName); //read conf file
		//impersonation
		HANDLE hToken; //receives a handle to a token that represents the specified user
		for (int i = 0; i < data.size(); i++) {
			std::wstring name = data[i].getUserName(); //temp variables
			std::wstring pass = data[i].getPassword();
			LogonUserW( //log on user
				(LPCWSTR)name.c_str(),
				L".", //using only the local account database
				(LPCWSTR)pass.c_str(),
				LOGON32_LOGON_INTERACTIVE,
				LOGON32_PROVIDER_DEFAULT,
				&hToken
			);
			SecureZeroMemory(&pass, sizeof(pass));
			logText += L"**********\n";
			logText += name + L"\n"; //print name of user
			ImpersonateLoggedOnUser(hToken); //impersonate user
			//try to read all files
			std::wstring folderFullPath = data[i].getFullPath();
			tryToReadAllFiles(logText,  folderFullPath);
			////try to read file FileToRead.txt
			//std::wstring fileToReadFullName = folderFullPath + L"\\FileToRead.txt";
			//tryToOpenFile(logText, data, fileToReadFullName);
			//try to create a file FileToCreate.txt
			std::wstring fileToCreateFullName = folderFullPath + L"\\FileToCreate.txt";
			tryToCreateFile(logText, fileToCreateFullName);
			//try to read FileToCreate.txt
			tryToReadFile(logText, fileToCreateFullName);
			RevertToSelf(); //session revertion
		}
		pFnLog(logText.c_str());
		return true;
	}
}

std::vector<User> readFile(std::wstring fileName) {
	std::vector<User> data; //vectror of users
	std::wifstream fin(fileName);
	while (!fin.eof())
	{
		User tmp_user;
		fin >> tmp_user;
		data.push_back(tmp_user);
	}
	return data;
}
std::wifstream & operator >> (std::wifstream & wifs, User & user) {
	wifs >> user.userName >> user.domainName >> user.password >> user.fullPath;
	return wifs;
}
void tryToOpenFile(std::wstring& logText, std::wstring filePath) {
	std::wifstream fin;
	fin.open(filePath.c_str());
	if (fin.is_open()) {
		logText += L"File is opened " + filePath + L"\n";
	}
	else {
		logText += L"Cannot open the file " + filePath + L"\n";
	}
	fin.close();
}
void tryToCreateFile(std::wstring& logText,  std::wstring filePath) {
	std::wofstream fout;
	fout.open(filePath);
	logText += L"Trying to create a file:\n";
	if (fout.is_open()) {
		logText += L"File is created " + filePath + L"\n";
		//try to write to file FileToCreate.txt
		std::wstring testText = L"Try to write text";
		fout << testText;
		logText += L"Trying to write: " + testText + L"\n";
	}
	else {
		logText += L"Cannot create the file " + filePath + L"\n";
	}
	fout.close();
}
void tryToReadFile(std::wstring& logText, std::wstring filePath) {
	std::wifstream fin;
	fin.open(filePath);
	logText += L"Trying to read file contents:\n";
	if (fin.is_open()) {
		logText += L"Reading the file " + filePath + L"\n" + L"File contains:\n";
		while (!fin.eof())
		{
			std::wstring temp;
			fin >> temp;
			logText+=temp +L" ";
		}
		logText += L"\n";
	}
	else {
		logText+=L"Cannot open " + filePath + L"\n";
	}
	fin.close();
}
void tryToReadAllFiles(std::wstring& logText, std::wstring filePath) {
	logText += L"Folder contains:\n";
	for (auto & p : fs::recursive_directory_iterator(filePath)) {
		logText += p.path().c_str();
		logText += L"\n";
	}
	logText += L"Trying to open files:\n";
	for (auto & p : fs::recursive_directory_iterator(filePath)) {
		if (!fs::is_directory(p)) {
			tryToOpenFile(logText, p.path().c_str());
		}
	}
}