mirror of
https://github.com/pi-hole/FTL.git
synced 2024-10-26 16:52:18 +02:00
Mcat12
3a5752d908
This requires the `pihole` user to have sudo access to the `pihole` command (will require a core update). Many future API commands will have this requirement, as they too will be using the `pihole` command. Signed-off-by: Mcat12 <newtoncat12@yahoo.com>
114 lines
2.6 KiB
C
114 lines
2.6 KiB
C
/* Pi-hole: A black hole for Internet advertisements
|
|
* (c) 2017 Pi-hole, LLC (https://pi-hole.net)
|
|
* Network-wide ad blocking via your own hardware.
|
|
*
|
|
* FTL Engine
|
|
* API /dns/
|
|
*
|
|
* This file is copyright under the latest version of the EUPL.
|
|
* Please see LICENSE file for your rights under this license. */
|
|
|
|
#include "FTL.h"
|
|
#include "api.h"
|
|
#include "cJSON.h"
|
|
|
|
void getList(int *sock, char type, char list_type)
|
|
{
|
|
FILE *fp;
|
|
char *line = NULL;
|
|
size_t size = 0;
|
|
|
|
sendAPIResponseOK(*sock, type);
|
|
ssend(*sock, "\"%s\":[", list_type == WHITELIST ? "whitelist" : "blacklist");
|
|
|
|
if((fp = fopen(list_type == WHITELIST ? files.whitelist : files.blacklist, "r")) != NULL)
|
|
{
|
|
bool first = true;
|
|
|
|
while(getline(&line, &size, fp) != -1) {
|
|
// Skip empty lines
|
|
if(line[0] == '\n')
|
|
continue;
|
|
|
|
if(!first) ssend(*sock, ",");
|
|
first = false;
|
|
|
|
// Trim off the newline, if it exists
|
|
line[strcspn(line, "\r\n")] = 0;
|
|
|
|
ssend(*sock, "\"%s\"", line);
|
|
}
|
|
// Free allocated memory
|
|
if(line != NULL)
|
|
{
|
|
free(line);
|
|
line = NULL;
|
|
}
|
|
|
|
fclose(fp);
|
|
}
|
|
|
|
ssend(*sock, "]");
|
|
}
|
|
|
|
void getPiholeStatus(int *sock, char type)
|
|
{
|
|
int status = countlineswith("#addn-hosts=/etc/pihole/gravity.list", files.dnsmasqconf);
|
|
sendAPIResponseOK(*sock, type);
|
|
ssend(*sock, "\"status\":%i", status == 1 ? 0 : 1);
|
|
}
|
|
|
|
void addList(int *sock, char type, char list_type, char *data)
|
|
{
|
|
cJSON *input_root = cJSON_Parse(data);
|
|
cJSON *domain_json = cJSON_GetObjectItemCaseSensitive(input_root, "domain");
|
|
char *domain;
|
|
|
|
// Validate domain
|
|
if(cJSON_IsString(domain_json)) {
|
|
domain = domain_json->valuestring;
|
|
|
|
if(isValidDomain(domain)) {
|
|
// Valid domain
|
|
char *partial_command;
|
|
|
|
if(list_type == WHITELIST)
|
|
partial_command = "sudo pihole -w -q %s";
|
|
else if(list_type == BLACKLIST)
|
|
partial_command = "sudo pihole -b -q %s";
|
|
else {
|
|
logg("Invalid list type in addList");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
char *command = malloc((strlen(domain) + strlen(partial_command)) * sizeof(char));
|
|
sprintf(command, partial_command, domain);
|
|
int return_code = system(command);
|
|
free(command);
|
|
|
|
if(return_code == 0) {
|
|
// Successfully added to list
|
|
sendAPIResponseOK(*sock, type);
|
|
ssend(*sock, "\"status\":\"success\"");
|
|
}
|
|
else {
|
|
// Failed to add to list
|
|
sendAPIResponseInternalServerError(*sock, type);
|
|
ssend(*sock, "\"status\":\"unknown_error\"");
|
|
}
|
|
}
|
|
else {
|
|
// Invalid domain
|
|
sendAPIResponseBadRequest(*sock, type);
|
|
ssend(*sock, "\"status\":\"invalid_domain\"");
|
|
}
|
|
}
|
|
else {
|
|
// No domain
|
|
sendAPIResponseBadRequest(*sock, type);
|
|
ssend(*sock, "\"status\":\"no_domain\"");
|
|
}
|
|
|
|
cJSON_Delete(input_root);
|
|
}
|