From cb36f1790a7b57eca22a9f1db1c6bb43769eff35 Mon Sep 17 00:00:00 2001 From: Jacob Date: Sat, 18 Jan 2020 07:32:50 -0500 Subject: [PATCH] More refactorings, pretty much done! --- main.c | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/main.c b/main.c index ae1ab9c..354a13a 100644 --- a/main.c +++ b/main.c @@ -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);