jacob.windle/hb
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

Working on host replacement method

Browse Source
This commit is contained in:
Jacob Windle 2019-12-12 16:08:27 -05:00
parent 7989bf9278
commit 323fd19544
2 changed files with 62 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt Normal file
View File

@ -0,0 +1,2 @@
project(HostBlocker)
add_executable(hb main.c)

68
main.c
View File

@ -11,15 +11,27 @@
* block sites within a certain time limit. * block sites within a certain time limit.
*/ */
const char *blockString = "0.0.0.0 "; const char *blockString = "0.0.0.0 ";
static const char *HOSTFILE = "/etc/hosts";
void replacehost(char *oldhost, char *newhost, FILE *hostsFile);
/**
* Return an open handle to the hosts file.
*
* @param mode whichever mode host file you want to open
* @return FILE *
*/
FILE *fopenHostsFile(int mode) FILE *fopenHostsFile(int mode)
{ {
switch(mode) switch(mode)
{ {
case 0: case 0:
return fopen("/etc/hosts", "r"); return fopen(HOSTFILE, "r");
case 1: case 1:
return fopen("/etc/hosts", "a"); return fopen(HOSTFILE, "a");
case 2:
return fopen(HOSTFILE, "r+");
} }
} }
@ -69,18 +81,23 @@ void readToHost(char *host, FILE *hostsFile)
char buf[256]; char buf[256];
char *ptr, *f; char *ptr, *f;
while ((ptr = fgets(buf, 256, hostsFile)) != NULL) printf("Starting to read!\n");
printf("Looking for host:%s\n", host);
fseek(hostsFile, 0, SEEK_SET);
while (fgets(buf, 256, hostsFile) != NULL)
{ {
/** /**
printf("%s\n", ptr); printf("%s\n", ptr);
printf("%s\n", host); printf("%s\n", host);
**/ **/
f = strcasestr(host, ptr); printf("Reading!\n");
f = strcasestr(buf, host);
if (f != NULL) { if (f != NULL) {
printf("FOUND IT\n"); printf("FOUND IT\n");
break; break;
} }
memset(buf, 0, sizeof(buf));
} }
} }
@ -100,16 +117,17 @@ int main(int argc, char **argv)
{ {
if (strcmp(argv[i], "add") == 0) if (strcmp(argv[i], "add") == 0)
{ {
if (argc < 3) { if (argc < 3) {
printf("Please provide a host!\n"); printf("Please provide a host!\n");
exit(1); exit(1);
} }
hostsFile = fopenHostsFile(1); hostsFile = fopenHostsFile(1);
blockHost(hostsFile, argv[i+1]); blockHost(hostsFile, argv[i+1]);
} }
else if (strcmp(argv[i], "edit") == 0) else if (strcmp(argv[i], "edit") == 0)
{ {
readToHost(argv[i+1], hostsFile); hostsFile = fopenHostsFile(2);
replacehost(argv[i+1], argv[i+2], hostsFile);
} }
else if (strcmp(argv[i], "delete") == 0) else if (strcmp(argv[i], "delete") == 0)
{ {
@ -129,3 +147,37 @@ int main(int argc, char **argv)
if (hostsFile != NULL) if (hostsFile != NULL)
fclose(hostsFile); fclose(hostsFile);
} }
void replacehost(char *oldhost, char *newhost, FILE *hostsFile) {
char buf[256];
char *ptr, *f;
char newHostsFile[4096];
printf("Starting to read!\n");
printf("Looking for host:%s\n", oldhost);
fseek(hostsFile, 0, SEEK_SET);
while (fgets(buf, 256, hostsFile) != NULL)
{
/**
printf("%s\n", ptr);
printf("%s\n", host);
**/
printf("Reading!\n");
f = strcasestr(buf, oldhost);
if (f != NULL) {
printf("FOUND IT\n");
memset(buf, 0, sizeof(buf));
sprintf(buf, "%s %s\n", blockString, newhost);
printf("New buf: %s", buf);
}
strcat(newHostsFile, buf);
memset(buf, 0, sizeof(buf));
}
if(0 >= fwrite(newHostsFile, sizeof(char), sizeof(newHostsFile), hostsFile)) {
fprintf(stderr, "Did not write anything!\n");
}
}