mirror of
https://github.com/pi-hole/pi-hole.git
synced 2024-12-04 20:13:17 +00:00
179 lines
No EOL
5.3 KiB
Python
179 lines
No EOL
5.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) 2015, LCI Technology Group, LLC
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# Redistributions of source code must retain the above copyright notice, this
|
|
# list of conditions and the following disclaimer.
|
|
#
|
|
# Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions and the following disclaimer in the documentation
|
|
# and/or other materials provided with the distribution.
|
|
#
|
|
# Neither the name of LCI Technology Group, LLC nor the names of its
|
|
# contributors may be used to endorse or promote products derived from this
|
|
# software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
# Pi-hole: A black hole for Internet advertisements
|
|
# (c) 2015, 2016 by Jacob Salmela
|
|
# Network-wide ad blocking via your Raspberry Pi
|
|
# http://pi-hole.net
|
|
# Controller for all pihole scripts and functions.
|
|
#
|
|
# Pi-hole is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
|
|
import datetime
|
|
import sqlite3
|
|
import time
|
|
import sys
|
|
import re
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Compiled Regular Expressions
|
|
#-----------------------------------------------------------------------------
|
|
q_re = re.compile(r'(.*) dnsmasq\[\d+\]: query\[(.*)\] (.*) from (.*)')
|
|
f_re = re.compile(r'(.*) dnsmasq\[\d+\]: forwarded (.*) to (.*)')
|
|
r_re = re.compile(r'(.*) dnsmasq\[\d+\]: (reply|cached) (.*) is (.*)')
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Functions
|
|
#-----------------------------------------------------------------------------
|
|
def create_tables():
|
|
qt = '''
|
|
CREATE TABLE IF NOT EXISTS queries (
|
|
source text,
|
|
query_type text,
|
|
name text,
|
|
ts datetime,
|
|
piholed bool
|
|
)
|
|
'''
|
|
c.execute(qt)
|
|
conn.commit()
|
|
|
|
ft = '''
|
|
CREATE TABLE IF NOT EXISTS forwards (
|
|
resolver text,
|
|
name text,
|
|
ts datetime
|
|
)
|
|
'''
|
|
c.execute(ft)
|
|
conn.commit()
|
|
|
|
rt = '''
|
|
CREATE TABLE IF NOT EXISTS replies (
|
|
ip text,
|
|
reply_type text,
|
|
name text,
|
|
ts datetime
|
|
)
|
|
'''
|
|
c.execute(rt)
|
|
conn.commit()
|
|
|
|
|
|
def convert_date(ds):
|
|
y = str(datetime.datetime.now().year)
|
|
ltime = time.strptime('{0} {1}'.format(y, ds), '%Y %b %d %H:%M:%S')
|
|
|
|
return time.strftime('%Y-%m-%d %H:%M:%S', ltime)
|
|
|
|
|
|
def parse_query(query):
|
|
m = q_re.match(query)
|
|
if m is not None:
|
|
add_query(m.group(4), m.group(2), m.group(3), m.group(1),check_gravity(m.group(3)))
|
|
|
|
|
|
def check_gravity(txt):
|
|
for entry in gravFile:
|
|
if entry == txt:
|
|
return True
|
|
return False
|
|
|
|
|
|
def parse_forward(query):
|
|
m = f_re.match(query)
|
|
if m is not None:
|
|
add_forward(m.group(3), m.group(2), m.group(1))
|
|
|
|
|
|
def parse_reply(query):
|
|
m = r_re.match(query)
|
|
if m is not None:
|
|
add_reply(m.group(4), m.group(2), m.group(3), m.group(1))
|
|
|
|
|
|
def add_query(source, qtype, name, ts, blocked):
|
|
sql = "INSERT INTO queries (source, query_type, name, ts, piholed) VALUES(?,?,?,?,?)"
|
|
c.execute(sql, (source, qtype, name, convert_date(ts), blocked))
|
|
|
|
|
|
def add_forward(resolver, name, ts):
|
|
sql = "INSERT INTO forwards (resolver, name, ts) VALUES(?,?,?)"
|
|
c.execute(sql, (resolver, name, convert_date(ts)))
|
|
|
|
|
|
def add_reply(ip, rtype, name, ts):
|
|
sql = "INSERT INTO replies (ip, reply_type, name, ts) VALUES(?,?,?,?)"
|
|
c.execute(sql, (ip, rtype, name, convert_date(ts)))
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Main
|
|
#-----------------------------------------------------------------------------
|
|
if len(sys.argv) != 2:
|
|
print 'Usage: dnsmasq_parse.py logfile'
|
|
sys.exit()
|
|
|
|
logfile = sys.argv[1]
|
|
|
|
# Create the SQLite connection
|
|
conn = sqlite3.connect('/etc/pihole/pihole.db')
|
|
|
|
gravFile = []
|
|
|
|
with open('/etc/pihole/gravity.list', 'r') as g:
|
|
for line in g:
|
|
gravFile.append(line.rstrip().split(' ')[1])
|
|
|
|
with conn:
|
|
c = conn.cursor()
|
|
|
|
create_tables()
|
|
|
|
# Parse the log file.
|
|
with open(logfile) as f:
|
|
for line in f:
|
|
line = line.rstrip()
|
|
|
|
if ': query[' in line:
|
|
parse_query(line)
|
|
|
|
elif ': forwarded ' in line:
|
|
parse_forward(line)
|
|
|
|
elif (': reply ' in line) or (': cached ' in line):
|
|
parse_reply(line) |