summaryrefslogtreecommitdiff
path: root/project2/proj2_s4498062/dns/cache.py
diff options
context:
space:
mode:
Diffstat (limited to 'project2/proj2_s4498062/dns/cache.py')
-rw-r--r--project2/proj2_s4498062/dns/cache.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/project2/proj2_s4498062/dns/cache.py b/project2/proj2_s4498062/dns/cache.py
index 9cde66f..a8b62be 100644
--- a/project2/proj2_s4498062/dns/cache.py
+++ b/project2/proj2_s4498062/dns/cache.py
@@ -53,7 +53,7 @@ class RecordCache(object):
FILE = '.dns.cache'
- def __init__(self):
+ def __init__(self, ttl):
""" Initialize the RecordCache
Args:
@@ -61,15 +61,16 @@ class RecordCache(object):
"""
self.records = []
self.read_cache_file()
+ self.ttl = ttl
def __del__(self):
self.write_cache_file()
def remove_old(self):
"""Remove entries for which the TTL has expired"""
- now = int(time.clock())
+ now = int(time.time())
for record in reversed(self.records):
- if record.ttl + record.timestamp < now:
+ if min(self.ttl, record.ttl) + record.timestamp < now:
self.records.remove(record)
def lookup(self, dname, type_, class_):