Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
DNS testing tools
Remoh
Commits
2a50125d
Commit
2a50125d
authored
Mar 11, 2020
by
Alexandre
Browse files
Use a parent Request class
parent
9def6c35
Changes
1
Hide whitespace changes
Inline
Side-by-side
homer.py
View file @
2a50125d
...
...
@@ -173,23 +173,25 @@ class CustomException(Exception):
pass
class
Request
DoT
:
class
Request
:
def
__init__
(
self
,
qname
,
qtype
=
rtype
,
use_edns
=
edns
,
want_dnssec
=
dnssec
):
self
.
message
=
dns
.
message
.
make_query
(
qname
,
dns
.
rdatatype
.
from_text
(
qtype
),
use_edns
=
use_edns
,
want_dnssec
=
want_dnssec
)
self
.
message
.
flags
|=
dns
.
flags
.
AD
# Ask for validation
def
to_wire
(
self
):
self
.
data
=
self
.
message
.
to_wire
()
class
RequestDoT
(
Request
):
def
check_response
(
self
):
if
self
.
response
.
id
!=
self
.
message
.
id
:
raise
Exception
(
"The ID in the answer does not match the one in the query"
)
class
RequestDoH
:
class
RequestDoH
(
Request
)
:
def
__init__
(
self
,
qname
,
qtype
=
rtype
,
use_edns
=
edns
,
want_dnssec
=
dnssec
):
self
.
message
=
dns
.
message
.
make_query
(
qname
,
dns
.
rdatatype
.
from_text
(
q
type
)
,
use_edns
=
use_
edns
,
want_dnssec
=
want_
dnssec
)
Request
.
__init__
(
self
,
qname
,
qtype
=
r
type
,
use_edns
=
edns
,
want_dnssec
=
dnssec
)
self
.
message
.
id
=
0
# DoH requests that
self
.
message
.
flags
|=
dns
.
flags
.
AD
# Ask for validation
self
.
data
=
self
.
message
.
to_wire
()
self
.
post
=
False
self
.
head
=
False
...
...
@@ -370,6 +372,7 @@ class ConnectionDoT(Connection):
request
.
rcode
=
True
def
send_and_receive
(
self
,
request
):
request
.
to_wire
()
self
.
send_data
(
request
.
data
)
self
.
receive_data
(
request
)
...
...
@@ -469,6 +472,7 @@ class ConnectionDoH(Connection):
self
.
buffer
.
close
()
def
send_and_receive
(
self
,
request
):
request
.
to_wire
()
self
.
prepare
(
request
)
self
.
perform
()
self
.
receive
(
request
)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment