mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-25 22:45:46 +03:00
e8c1f5c8d3
Merge in DNS/adguard-home from 2508-ip-conversion to master Updates #2508. Squashed commit of the following: commit 3f64709fbc73ef74c11b910997be1e9bc337193c Merge: 5ac7faaaa0d67aa251Author: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 16:21:34 2021 +0300 Merge branch 'master' into 2508-ip-conversion commit 5ac7faaaa9dda570fdb872acad5d13d078f46b64 Author: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 12:00:11 2021 +0300 all: replace conditions with appropriate functions in tests commit 9e3fa9a115ed23024c57dd5192d5173477ddbf71 Merge: db992a42abba74859eAuthor: Eugene Burkov <e.burkov@adguard.com> Date: Wed Jan 13 10:47:10 2021 +0300 Merge branch 'master' into 2508-ip-conversion commit db992a42a2c6f315421e78a6a0492e2bfb3ce89d Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 18:55:53 2021 +0300 sysutil: fix linux tests commit f629b15d62349323ce2da05e68dc9cc0b5f6e194 Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 18:41:20 2021 +0300 all: improve code quality commit 3bf03a75524040738562298bd1de6db536af130f Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 17:33:26 2021 +0300 sysutil: fix linux net.IP conversion commit 5d5b6994916923636e635588631b63b7e7b74e5f Author: Eugene Burkov <e.burkov@adguard.com> Date: Tue Jan 12 14:57:26 2021 +0300 dnsforward: remove redundant net.IP <-> string conversion commit 0b955d99b7fad40942f21d1dd8734adb99126195 Author: Eugene Burkov <e.burkov@adguard.com> Date: Mon Jan 11 18:04:25 2021 +0300 dhcpd: remove net.IP <-> string conversion
348 lines
8.8 KiB
Go
348 lines
8.8 KiB
Go
package querylog
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
"io/ioutil"
|
|
"math"
|
|
"net"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestQLogFileEmpty(t *testing.T) {
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
testFile := prepareTestFile(testDir, 0)
|
|
|
|
// create the new QLogFile instance
|
|
q, err := NewQLogFile(testFile)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, q)
|
|
defer q.Close()
|
|
|
|
// seek to the start
|
|
pos, err := q.SeekStart()
|
|
assert.Nil(t, err)
|
|
assert.EqualValues(t, 0, pos)
|
|
|
|
// try reading anyway
|
|
line, err := q.ReadNext()
|
|
assert.Equal(t, io.EOF, err)
|
|
assert.Empty(t, line)
|
|
}
|
|
|
|
func TestQLogFileLarge(t *testing.T) {
|
|
// should be large enough
|
|
count := 50000
|
|
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
testFile := prepareTestFile(testDir, count)
|
|
|
|
// create the new QLogFile instance
|
|
q, err := NewQLogFile(testFile)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, q)
|
|
defer q.Close()
|
|
|
|
// seek to the start
|
|
pos, err := q.SeekStart()
|
|
assert.Nil(t, err)
|
|
assert.NotEqualValues(t, 0, pos)
|
|
|
|
read := 0
|
|
var line string
|
|
for err == nil {
|
|
line, err = q.ReadNext()
|
|
if err == nil {
|
|
assert.NotZero(t, len(line))
|
|
read++
|
|
}
|
|
}
|
|
|
|
assert.Equal(t, count, read)
|
|
assert.Equal(t, io.EOF, err)
|
|
}
|
|
|
|
func TestQLogFileSeekLargeFile(t *testing.T) {
|
|
// more or less big file
|
|
count := 10000
|
|
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
testFile := prepareTestFile(testDir, count)
|
|
|
|
// create the new QLogFile instance
|
|
q, err := NewQLogFile(testFile)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, q)
|
|
defer q.Close()
|
|
|
|
// CASE 1: NOT TOO OLD LINE
|
|
testSeekLineQLogFile(t, q, 300)
|
|
|
|
// CASE 2: OLD LINE
|
|
testSeekLineQLogFile(t, q, count-300)
|
|
|
|
// CASE 3: FIRST LINE
|
|
testSeekLineQLogFile(t, q, 0)
|
|
|
|
// CASE 4: LAST LINE
|
|
testSeekLineQLogFile(t, q, count)
|
|
|
|
// CASE 5: Seek non-existent (too low)
|
|
_, _, err = q.SeekTS(123)
|
|
assert.NotNil(t, err)
|
|
|
|
// CASE 6: Seek non-existent (too high)
|
|
ts, _ := time.Parse(time.RFC3339, "2100-01-02T15:04:05Z07:00")
|
|
_, _, err = q.SeekTS(ts.UnixNano())
|
|
assert.NotNil(t, err)
|
|
|
|
// CASE 7: "Almost" found
|
|
line, err := getQLogFileLine(q, count/2)
|
|
assert.Nil(t, err)
|
|
// ALMOST the record we need
|
|
timestamp := readQLogTimestamp(line) - 1
|
|
assert.NotEqualValues(t, 0, timestamp)
|
|
_, depth, err := q.SeekTS(timestamp)
|
|
assert.NotNil(t, err)
|
|
assert.LessOrEqual(t, depth, int(math.Log2(float64(count))+3))
|
|
}
|
|
|
|
func TestQLogFileSeekSmallFile(t *testing.T) {
|
|
// more or less big file
|
|
count := 10
|
|
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
testFile := prepareTestFile(testDir, count)
|
|
|
|
// create the new QLogFile instance
|
|
q, err := NewQLogFile(testFile)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, q)
|
|
defer q.Close()
|
|
|
|
// CASE 1: NOT TOO OLD LINE
|
|
testSeekLineQLogFile(t, q, 2)
|
|
|
|
// CASE 2: OLD LINE
|
|
testSeekLineQLogFile(t, q, count-2)
|
|
|
|
// CASE 3: FIRST LINE
|
|
testSeekLineQLogFile(t, q, 0)
|
|
|
|
// CASE 4: LAST LINE
|
|
testSeekLineQLogFile(t, q, count)
|
|
|
|
// CASE 5: Seek non-existent (too low)
|
|
_, _, err = q.SeekTS(123)
|
|
assert.NotNil(t, err)
|
|
|
|
// CASE 6: Seek non-existent (too high)
|
|
ts, _ := time.Parse(time.RFC3339, "2100-01-02T15:04:05Z07:00")
|
|
_, _, err = q.SeekTS(ts.UnixNano())
|
|
assert.NotNil(t, err)
|
|
|
|
// CASE 7: "Almost" found
|
|
line, err := getQLogFileLine(q, count/2)
|
|
assert.Nil(t, err)
|
|
// ALMOST the record we need
|
|
timestamp := readQLogTimestamp(line) - 1
|
|
assert.NotEqualValues(t, 0, timestamp)
|
|
_, depth, err := q.SeekTS(timestamp)
|
|
assert.NotNil(t, err)
|
|
assert.LessOrEqual(t, depth, int(math.Log2(float64(count))+3))
|
|
}
|
|
|
|
func testSeekLineQLogFile(t *testing.T, q *QLogFile, lineNumber int) {
|
|
line, err := getQLogFileLine(q, lineNumber)
|
|
assert.Nil(t, err)
|
|
ts := readQLogTimestamp(line)
|
|
assert.NotEqualValues(t, 0, ts)
|
|
|
|
// try seeking to that line now
|
|
pos, _, err := q.SeekTS(ts)
|
|
assert.Nil(t, err)
|
|
assert.NotEqualValues(t, 0, pos)
|
|
|
|
testLine, err := q.ReadNext()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, line, testLine)
|
|
}
|
|
|
|
func getQLogFileLine(q *QLogFile, lineNumber int) (string, error) {
|
|
_, err := q.SeekStart()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
for i := 1; i < lineNumber; i++ {
|
|
_, err := q.ReadNext()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
return q.ReadNext()
|
|
}
|
|
|
|
// Check adding and loading (with filtering) entries from disk and memory
|
|
func TestQLogFile(t *testing.T) {
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
testFile := prepareTestFile(testDir, 2)
|
|
|
|
// create the new QLogFile instance
|
|
q, err := NewQLogFile(testFile)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, q)
|
|
defer q.Close()
|
|
|
|
// seek to the start
|
|
pos, err := q.SeekStart()
|
|
assert.Nil(t, err)
|
|
assert.Greater(t, pos, int64(0))
|
|
|
|
// read first line
|
|
line, err := q.ReadNext()
|
|
assert.Nil(t, err)
|
|
assert.Contains(t, line, "0.0.0.2")
|
|
assert.True(t, strings.HasPrefix(line, "{"), line)
|
|
assert.True(t, strings.HasSuffix(line, "}"), line)
|
|
|
|
// read second line
|
|
line, err = q.ReadNext()
|
|
assert.Nil(t, err)
|
|
assert.EqualValues(t, 0, q.position)
|
|
assert.Contains(t, line, "0.0.0.1")
|
|
assert.True(t, strings.HasPrefix(line, "{"), line)
|
|
assert.True(t, strings.HasSuffix(line, "}"), line)
|
|
|
|
// try reading again (there's nothing to read anymore)
|
|
line, err = q.ReadNext()
|
|
assert.Equal(t, io.EOF, err)
|
|
assert.Empty(t, line)
|
|
}
|
|
|
|
// prepareTestFile - prepares a test query log file with the specified number of lines
|
|
func prepareTestFile(dir string, linesCount int) string {
|
|
return prepareTestFiles(dir, 1, linesCount)[0]
|
|
}
|
|
|
|
// prepareTestFiles - prepares several test query log files
|
|
// each of them -- with the specified linesCount
|
|
func prepareTestFiles(dir string, filesCount, linesCount int) []string {
|
|
format := `{"IP":"${IP}","T":"${TIMESTAMP}","QH":"example.org","QT":"A","QC":"IN","Answer":"AAAAAAABAAEAAAAAB2V4YW1wbGUDb3JnAAABAAEHZXhhbXBsZQNvcmcAAAEAAQAAAAAABAECAwQ=","Result":{},"Elapsed":0,"Upstream":"upstream"}`
|
|
|
|
lineTime, _ := time.Parse(time.RFC3339Nano, "2020-02-18T22:36:35.920973+03:00")
|
|
lineIP := uint32(0)
|
|
|
|
files := make([]string, filesCount)
|
|
for j := 0; j < filesCount; j++ {
|
|
f, _ := ioutil.TempFile(dir, "*.txt")
|
|
files[filesCount-j-1] = f.Name()
|
|
|
|
for i := 0; i < linesCount; i++ {
|
|
lineIP++
|
|
lineTime = lineTime.Add(time.Second)
|
|
|
|
ip := make(net.IP, 4)
|
|
binary.BigEndian.PutUint32(ip, lineIP)
|
|
|
|
line := format
|
|
line = strings.ReplaceAll(line, "${IP}", ip.String())
|
|
line = strings.ReplaceAll(line, "${TIMESTAMP}", lineTime.Format(time.RFC3339Nano))
|
|
|
|
_, _ = f.WriteString(line)
|
|
_, _ = f.WriteString("\n")
|
|
}
|
|
}
|
|
|
|
return files
|
|
}
|
|
|
|
func TestQLogSeek(t *testing.T) {
|
|
testDir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(testDir) }()
|
|
|
|
d := `{"T":"2020-08-31T18:44:23.911246629+03:00","QH":"wfqvjymurpwegyv","QT":"A","QC":"IN","CP":"","Answer":"","Result":{},"Elapsed":66286385,"Upstream":"tls://dns-unfiltered.adguard.com:853"}
|
|
{"T":"2020-08-31T18:44:25.376690873+03:00"}
|
|
{"T":"2020-08-31T18:44:25.382540454+03:00"}`
|
|
f, _ := ioutil.TempFile(testDir, "*.txt")
|
|
_, _ = f.WriteString(d)
|
|
defer f.Close()
|
|
|
|
q, err := NewQLogFile(f.Name())
|
|
assert.Nil(t, err)
|
|
defer q.Close()
|
|
|
|
target, _ := time.Parse(time.RFC3339, "2020-08-31T18:44:25.376690873+03:00")
|
|
|
|
_, depth, err := q.SeekTS(target.UnixNano())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, depth)
|
|
}
|
|
|
|
func TestQLogSeek_ErrTSTooLate(t *testing.T) {
|
|
testDir := prepareTestDir()
|
|
t.Cleanup(func() {
|
|
_ = os.RemoveAll(testDir)
|
|
})
|
|
|
|
d := `{"T":"2020-08-31T18:44:23.911246629+03:00","QH":"wfqvjymurpwegyv","QT":"A","QC":"IN","CP":"","Answer":"","Result":{},"Elapsed":66286385,"Upstream":"tls://dns-unfiltered.adguard.com:853"}
|
|
{"T":"2020-08-31T18:44:25.376690873+03:00"}
|
|
{"T":"2020-08-31T18:44:25.382540454+03:00"}
|
|
`
|
|
f, err := ioutil.TempFile(testDir, "*.txt")
|
|
assert.Nil(t, err)
|
|
defer f.Close()
|
|
|
|
_, err = f.WriteString(d)
|
|
assert.Nil(t, err)
|
|
|
|
q, err := NewQLogFile(f.Name())
|
|
assert.Nil(t, err)
|
|
defer q.Close()
|
|
|
|
target, err := time.Parse(time.RFC3339, "2020-08-31T18:44:25.382540454+03:00")
|
|
assert.Nil(t, err)
|
|
|
|
_, depth, err := q.SeekTS(target.UnixNano() + int64(time.Second))
|
|
assert.Equal(t, ErrTSTooLate, err)
|
|
assert.Equal(t, 2, depth)
|
|
}
|
|
|
|
func TestQLogSeek_ErrTSTooEarly(t *testing.T) {
|
|
testDir := prepareTestDir()
|
|
t.Cleanup(func() {
|
|
_ = os.RemoveAll(testDir)
|
|
})
|
|
|
|
d := `{"T":"2020-08-31T18:44:23.911246629+03:00","QH":"wfqvjymurpwegyv","QT":"A","QC":"IN","CP":"","Answer":"","Result":{},"Elapsed":66286385,"Upstream":"tls://dns-unfiltered.adguard.com:853"}
|
|
{"T":"2020-08-31T18:44:25.376690873+03:00"}
|
|
{"T":"2020-08-31T18:44:25.382540454+03:00"}
|
|
`
|
|
f, err := ioutil.TempFile(testDir, "*.txt")
|
|
assert.Nil(t, err)
|
|
defer f.Close()
|
|
|
|
_, err = f.WriteString(d)
|
|
assert.Nil(t, err)
|
|
|
|
q, err := NewQLogFile(f.Name())
|
|
assert.Nil(t, err)
|
|
defer q.Close()
|
|
|
|
target, err := time.Parse(time.RFC3339, "2020-08-31T18:44:23.911246629+03:00")
|
|
assert.Nil(t, err)
|
|
|
|
_, depth, err := q.SeekTS(target.UnixNano() - int64(time.Second))
|
|
assert.Equal(t, ErrTSTooEarly, err)
|
|
assert.Equal(t, 1, depth)
|
|
}
|