More refactorings, pretty much done!

This commit is contained in:
Jacob Windle 2020-01-18 07:32:50 -05:00
parent 31186418c3
commit cb36f1790a

14
main.c
View File

@ -31,8 +31,7 @@ static char *CONFIG;
* @param mode whichever mode host file you want to open
* @return FILE *
*/
FILE *fopenHostsFile(char *mode)
{
FILE *fopenHostsFile(char *mode) {
return fopen(HOSTFILE, mode);
}
@ -71,13 +70,13 @@ void modifyHostsFile(char *oldhost, char *newhost, int deleteHost) {
newHostsFile[strlen(newHostsFile)] = EOF;
// // Seek to 0
// fseek(hostsFile, 0, SEEK_SET);
fseek(hostsFile, 0, SEEK_SET);
fclose(hostsFile);
// If this failed, write an error message to stderr
// ONFAILED(0, (fwrite(newHostsFile, sizeof(char), sizeof(newHostsFile), hostsFile))) {
// fprintf(stderr, "Did not write anything!\n");
// }
ONFAILED(0, (fwrite(newHostsFile, sizeof(char), sizeof(newHostsFile), hostsFile))) {
fprintf(stderr, "Did not write anything!\n");
}
fprintf(stderr, "New hosts file: \n%s\n", newHostsFile);
// fclose(hostsFile);
}
@ -86,8 +85,7 @@ void modifyHostsFile(char *oldhost, char *newhost, int deleteHost) {
/**
* Block a hosts using the hostsFile in FILE
*/
void blockHost(char *host)
{
void blockHost(char *host) {
FILE *hostsFile = fopenHostsFile("a");
char blockRule[256];
sprintf(blockRule, "%s %s\n", blockString, host);