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

Merge branch 'feature/delete-host' of jacob.windle/hb into master

master
Jacob Windle 2020-01-18 12:43:37 +00:00 committed by Gitea
parent 76feb40f1a 91c01dd192
commit b35703d936
2 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
README.md
View File

@ -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 <wakeup period> -config <path/to/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

18
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,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);