From 621c8083c892cb9484d2e092dd46db96be18c909 Mon Sep 17 00:00:00 2001
From: DL6ER <derigs@ph1.uni-koeln.de>
Date: Mon, 12 Dec 2016 13:37:58 +0100
Subject: [PATCH 1/3] Apply API_QUERY_LOG_SHOW to API result

---
 data.php | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/data.php b/data.php
index 419ecd15..7813d4c8 100644
--- a/data.php
+++ b/data.php
@@ -141,11 +141,35 @@
     }
 
     function getAllQueries($orderBy) {
-        global $log;
+        global $log,$setupVars;
         $allQueries = array("data" => array());
         $dns_queries = getDnsQueriesAll($log);
         $hostname = trim(file_get_contents("/etc/hostname"), "\x00..\x1F");
 
+        if(isset($setupVars["API_QUERY_LOG_SHOW"]))
+        {
+            if($setupVars["API_QUERY_LOG_SHOW"] === "all")
+            {
+                $showblocked = true;
+                $showpermitted = true;
+            }
+            elseif($setupVars["API_QUERY_LOG_SHOW"] === "permittedonly")
+            {
+                $showblocked = false;
+                $showpermitted = true;
+            }
+            elseif($setupVars["API_QUERY_LOG_SHOW"] === "blockedonly")
+            {
+                $showblocked = true;
+                $showpermitted = false;
+            }
+        }
+        else
+        {
+            $showblocked = true;
+            $showpermitted = true;
+        }
+
         foreach ($dns_queries as $query) {
             $time = date_create(substr($query, 0, 16));
             $exploded = explode(" ", trim($query));
@@ -158,10 +182,10 @@
               $client = $exploded[count($exploded)-1];
 
             }
-            elseif (substr($tmp, 0, 9) == "forwarded" || $exploded[count($exploded)-3] == "pi.hole" || $exploded[count($exploded)-3] == $hostname){
+            elseif ((substr($tmp, 0, 9) == "forwarded" || $exploded[count($exploded)-3] == "pi.hole" || $exploded[count($exploded)-3] == $hostname) && $showpermitted){
               $status="OK";
             }
-            elseif (substr($tmp, strlen($tmp) - 12, 12)  == "gravity.list"  && $exploded[count($exploded)-5] != "read"){
+            elseif ((substr($tmp, strlen($tmp) - 12, 12)  == "gravity.list"  && $exploded[count($exploded)-5] != "read") && $showblocked){
               $status="Pi-holed";
             }
 

From afa490cb81de72ab16890e27519c02f12c43edec Mon Sep 17 00:00:00 2001
From: DL6ER <derigs@ph1.uni-koeln.de>
Date: Mon, 12 Dec 2016 13:38:18 +0100
Subject: [PATCH 2/3] Show information if API_QUERY_LOG_SHOW is set

---
 queries.php | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/queries.php b/queries.php
index d4a02374..153b5cbf 100644
--- a/queries.php
+++ b/queries.php
@@ -6,10 +6,30 @@ if(empty($_SESSION['token'])) {
     $_SESSION['token'] = base64_encode(openssl_random_pseudo_bytes(32));
 }
 $token = $_SESSION['token'];
+
+$showing = "";
+
+if(isset($setupVars["API_QUERY_LOG_SHOW"]))
+{
+	if($setupVars["API_QUERY_LOG_SHOW"] === "all")
+	{
+		$showing = "(showing all queries)";
+	}
+	elseif($setupVars["API_QUERY_LOG_SHOW"] === "permittedonly")
+	{
+		$showing = "(showing permitted queries only)";
+	}
+	elseif($setupVars["API_QUERY_LOG_SHOW"] === "blockedonly")
+	{
+		$showing = "(showing blocked queries only)";
+	}
+}
+
 ?>
 <!-- Send PHP info to JS -->
 <div id="token" hidden><?php echo $token ?></div>
 
+
 <!--
 <div class="row">
     <div class="col-md-12">
@@ -36,7 +56,7 @@ $token = $_SESSION['token'];
     <div class="col-md-12">
       <div class="box" id="recent-queries">
         <div class="box-header with-border">
-          <h3 class="box-title">Recent Queries</h3>
+          <h3 class="box-title">Recent Queries <?php echo $showing; ?></h3>
         </div>
         <!-- /.box-header -->
         <div class="box-body">

From 6f780ab5febce102e9e04cb00199767b003fe52c Mon Sep 17 00:00:00 2001
From: DL6ER <derigs@ph1.uni-koeln.de>
Date: Mon, 12 Dec 2016 22:15:39 +0100
Subject: [PATCH 3/3] Check for flag none

---
 data.php | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/data.php b/data.php
index 7813d4c8..9c91cc34 100644
--- a/data.php
+++ b/data.php
@@ -163,6 +163,17 @@
                 $showblocked = true;
                 $showpermitted = false;
             }
+            elseif($setupVars["API_QUERY_LOG_SHOW"] === "none")
+            {
+                $showblocked = false;
+                $showpermitted = false;
+            }
+            else
+            {
+                // Invalid settings, show everything
+                $showblocked = true;
+                $showpermitted = true;
+            }
         }
         else
         {