diff --git a/README.md b/README.md index 4d2d3d1..7d40b16 100644 --- a/README.md +++ b/README.md @@ -6,21 +6,27 @@ This is a system utility written in C that allows one to quickly black hole host # How it Works +The program "blackholes" hosts by setting their entries in /etc/hosts to be +`0.0.0.0` so that when one tries to resolve the host in the browser the host will +resolve to `0.0.0.0`. + The program will wake up in a configurable timeframe continuously, reading a config file into memory and making appropriate changes to the /etc/hosts based on what it reads from the config. +Starting the program like this allows for the program to run as a daemon: + +```bash +sudo hb -period -config -daemon +``` + +The config file at the moment is just a newline separated list of hosts that will +be added to the /etc/hosts file. + # Motivations I would like to get better with C, and will be using C much more in the near future. In addition, I have issues with browsing to sites that hurt my productivity and am always on the command line. # TODO -1. Get delete working for my /etc/hosts editor. -2. Daemonize this program to be started up by systemd. - - As part of daemonization, remember and remove added hosts on teardown. - - Can do this with an atexit handler, and a routine in main. -3. Log dns requests that look at the hosts file -- To see how many go where -4. As part of daemonization -- only block hosts within a time limit. - - Sleep, wakeup, check if time has passed or not, block/unblock host, - sleep again. -5. Read hosts to block from a config file (to not pollute the /etc/hosts file) -6. Design a unified interface for reading a hosts file. +- Debug logging +- Systemd unit file for running as a daemon + diff --git a/main.c b/main.c index ae1ab9c..7dd0129 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,23 +70,22 @@ void modifyHostsFile(char *oldhost, char *newhost, int deleteHost) { newHostsFile[strlen(newHostsFile)] = EOF; // // Seek to 0 -// fseek(hostsFile, 0, SEEK_SET); - fclose(hostsFile); + 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); + fclose(hostsFile); } /** * 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);