Skip to content

updated dnscommon for NS, PTR, MX records and enhancement for 'AAAA' and 'TXT' records #166

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
pranavnyu wants to merge 11 commits into
base: master
Choose a base branch
from
Open

updated dnscommon for NS, PTR, MX records and enhancement for 'AAAA' and 'TXT' records #166

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
35cc77c
updated dnscommon for NS and PTR
pranavnyu May 5, 2015
ba00027
fix for 'MX' and support for 'AAAA' records
pranavnyu May 12, 2015
7f73846
updating support for 'TXT' queries
pranavnyu May 12, 2015
71d1fd1
dnsquery update
pranavnyu May 15, 2015
34aac48
Update dnsquery.r2py
pranavnyu May 15, 2015
c2421fd
updated dnscommon
pranavnyu May 21, 2015
49e888f
updated dnsquery and dnscommon
pranavnyu May 27, 2015
9ea31f2
updated dnscommon
pranavnyu Jun 2, 2015
4a965e0
updated
pranavnyu Jun 2, 2015
14f78f0
updated dnsquery with exception in gethostbyname function
pranavnyu Jun 25, 2015
d4c2ef7
updated
pranavnyu Jun 25, 2015
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
68 changes: 62 additions & 6 deletions dnscommon.r2py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,7 @@ _type_lookup_table = {
14 : 'MINFO', # Mailbox Information
15 : 'MX', # Mail Exchange
16 : 'TXT', # Text Strings
28 : 'AAAA', # IPv6 address
252 : 'AXFR', # Request for transfer of zone
253 : 'MAILB', # Request for mailbox-related RRs
254 : 'MAILA', # Request for mail agent RRs (Obsolete)
Expand All @@ -130,6 +131,7 @@ _type_lookup_table = {
'MINFO' : 14,
'MX' : 15,
'TXT' : 16,
'AAAA' : 28,
'AXFR' : 252,
'MAILB' : 253,
'MAILA' : 254,
Expand Down Expand Up @@ -494,17 +496,16 @@ def _read_single_answer(answer_index, dns_query_data):
"""
# Parse answer address.
read_index, answer_name = _parse_address(answer_index, dns_query_data)

# Read the type.
answer_type = _type_lookup_table[
_charpair_to_int16(dns_query_data[read_index:read_index + 2])]
read_index += 2

read_index += 2
# Read the class.
answer_class = _class_lookup_table[
_charpair_to_int16(dns_query_data[read_index:read_index + 2])]
read_index += 2

# Some math magic with the TTL.
time_to_live = _charpair_to_int16(dns_query_data[read_index:read_index + 2]) * 2 ** 16
read_index += 2
Expand All @@ -519,9 +520,26 @@ def _read_single_answer(answer_index, dns_query_data):

# There are a lot of different types of queries. We can parse all of these ones
# in the same way, though.
simple_answers = ['A', 'NS', 'CNAME', 'MD', 'MB', 'MF', 'MG', 'MR', 'MX', 'PTR']
simple_answers = ['A', 'CNAME', 'MD', 'MB', 'MF', 'MG', 'MR']
if answer_type in simple_answers:
read_index, resource_data['address'] = _parse_answer_address(read_index, dns_query_data)
elif answer_type == 'NS':
read_index, resource_data['nameserver'] = _parse_address(read_index,dns_query_data)
Copy link
Contributor

@aaaaalbert aaaaalbert May 18, 2015

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please combine all the answer types that are treated the same way under a single elif branch.

Copy link
Author

@pranavnyu pranavnyu May 21, 2015

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We should not combine all because when you query for MX or reverse or AAAA etc then you have additional section containing extra data. We don't need that but if we don't separate then packet dictionary will contain mixed details. So by separating, packet dictionary will contain all information with their respective query types.
for eg.

;; QUESTION SECTION:
;google.com.            IN  MX

;; ANSWER SECTION:
google.com.     600 IN  MX  20 alt1.aspmx.l.google.com.
google.com.     600 IN  MX  30 alt2.aspmx.l.google.com.
google.com.     600 IN  MX  40 alt3.aspmx.l.google.com.
google.com.     600 IN  MX  10 aspmx.l.google.com.
google.com.     600 IN  MX  50 alt4.aspmx.l.google.com.

;; ADDITIONAL SECTION:
alt1.aspmx.l.google.com. 156    IN  A   64.233.186.27

or

;google.com.            IN  NS

;; ANSWER SECTION:
google.com.     14400   IN  NS  ns4.google.com.
google.com.     14400   IN  NS  ns1.google.com.
google.com.     14400   IN  NS  ns2.google.com.
google.com.     14400   IN  NS  ns3.google.com.

;; ADDITIONAL SECTION:
ns1.google.com.     13328   IN  A   216.239.32.10
ns2.google.com.     13956   IN  A   216.239.34.10
ns3.google.com.     2225    IN  A   216.239.36.10
ns4.google.com.     2315    IN  A   216.239.38.10

elif answer_type == 'PTR':
read_index, resource_data['domain'] = _parse_address(read_index, dns_query_data)
elif answer_type == 'MX':
read_index += 1
resource_data['preference'] = str(ord(dns_query_data[read_index]))
read_index += 1
read_index, resource_data['record'] = _parse_address(read_index, dns_query_data)
elif answer_type == 'AAAA':
read_index, resource_data['address'] = _parse_aaaa_address(read_index, dns_query_data)
elif answer_type == 'TXT':
txt_length = len(dns_query_data)-read_index
resource_text = ''
resource_text += dns_query_data[read_index:read_index+txt_length]
read_index += txt_length
resource_data['text'] = resource_text
elif answer_type == 'SOA':
read_index, mname = _parse_address(read_index, dns_query_data)
resource_data['mname'] = mname
Expand Down Expand Up @@ -552,7 +570,6 @@ def _read_single_answer(answer_index, dns_query_data):
resource_data['retry'] = retry
resource_data['expire'] = expire
resource_data['minimum'] = minimum

answer_dict = {
'name' : answer_name,
'type' : answer_type,
Expand All @@ -564,7 +581,46 @@ def _read_single_answer(answer_index, dns_query_data):
return read_index, answer_dict


def _parse_aaaa_address(address_index, dns_query_data):
"""
<Purpose>
This method parses a IPv6 directly out of an answer
section. This method is needed because answer IPs are not delimited,
since their labels are not of variable length.

<Arguments>
address_index
The integer index at which the address starts.
dns_query_data
The string containing raw query data

<Exceptions>
If the address is improperly formatted, this will produce an integer
parsing failure.

<Side Effects>
None

<Returns>
A tuple containing the new read index and IPv6 Address.
"""
#converts packet data to hex
add = []
address = ''
for i in range(0,16):
add.append(str(hex(ord(dns_query_data[address_index]))))
address_index += 1
#converts hex data to readable IPv6
for i in add:
i = i[2:]
if len(i) == 2:
address += i
elif len(i)==1:
i = '0'+i
address += i
address = ':'.join([address[x:x+4] for x in range(0, len(address)-1, 4)])

return address_index, address

def _parse_answer_address(address_index, dns_query_data):
"""
Expand Down Expand Up @@ -600,7 +656,7 @@ def _parse_answer_address(address_index, dns_query_data):
address_index += 1
address += str(ord(dns_query_data[address_index]))
address_index += 1

return address_index, address


Expand Down
Loading