From c605e1247912ad4638acaaa23bba2ee03f75750c Mon Sep 17 00:00:00 2001
From: Mayank Agarwal <amayank@fb.com>
Date: Thu, 9 May 2013 17:33:27 -0700
Subject: [PATCH] Adding Sanity checks in Get and Iterator->value for ttl

Summary: The 2 checks added will increase reliabilty and help in debugging

Test Plan: make ttl_test;./ttl_test

Reviewers: vamsi, dhruba, sheki, haobo

Reviewed By: vamsi

Differential Revision: https://reviews.facebook.net/D10713
---
 utilities/ttl/db_ttl.cc | 25 +++++++++++++++++++++----
 utilities/ttl/db_ttl.h  |  4 ++++
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/utilities/ttl/db_ttl.cc b/utilities/ttl/db_ttl.cc
index 3af366c70..d89e08941 100644
--- a/utilities/ttl/db_ttl.cc
+++ b/utilities/ttl/db_ttl.cc
@@ -53,7 +53,7 @@ class TtlIterator : public Iterator {
   }
 
   Slice value() const {
-    assert(iter_->value().size() >= (unsigned)ts_len_);
+    assert(DBWithTTL::SanityCheckTimestamp(iter_->value().ToString()).ok());
     Slice trimmed_value = iter_->value();
     trimmed_value.size_ -= ts_len_;
     return trimmed_value;
@@ -130,6 +130,22 @@ Status DBWithTTL::AppendTS(const Slice& val, std::string& val_with_ts) {
   return st;
 }
 
+// Returns corruption if the length of the string is lesser than timestamp, or
+// timestamp refers to a time lesser than ttl-feature release time
+Status DBWithTTL::SanityCheckTimestamp(const std::string& str) {
+  if (str.length() < (unsigned)kTSLength) {
+    return Status::Corruption("Error: value's length less than timestamp's\n");
+  }
+  // Checks that TS is not lesser than kMinTimestamp
+  // Gaurds against corruption & normal database opened incorrectly in ttl mode
+  int32_t timestamp_value =
+    DecodeFixed32(str.data() + str.size() - kTSLength);
+  if (timestamp_value < kMinTimestamp){
+    return Status::Corruption("Error: Timestamp < ttl feature release time!\n");
+  }
+  return Status::OK();
+}
+
 // Checks if the string is stale or not according to TTl provided
 bool DBWithTTL::IsStale(const Slice& value, int32_t ttl) {
   if (ttl <= 0) { // Data is fresh if TTL is non-positive
@@ -151,9 +167,6 @@ bool DBWithTTL::IsStale(const Slice& value, int32_t ttl) {
 // Strips the TS from the end of the string
 Status DBWithTTL::StripTS(std::string* str) {
   Status st;
-  if (str->length() < (unsigned)kTSLength) {
-    return Status::IOError("Error: value's length less than timestamp's\n");
-  }
   // Erasing characters which hold the TS
   str->erase(str->length() - kTSLength, kTSLength);
   return st;
@@ -178,6 +191,10 @@ Status DBWithTTL::Get(const ReadOptions& options,
   if (!st.ok()) {
     return st;
   }
+  st = SanityCheckTimestamp(*value);
+  if (!st.ok()) {
+    return st;
+  }
   return StripTS(value);
 }
 
diff --git a/utilities/ttl/db_ttl.h b/utilities/ttl/db_ttl.h
index e29a37b72..0a38070c3 100644
--- a/utilities/ttl/db_ttl.h
+++ b/utilities/ttl/db_ttl.h
@@ -81,12 +81,16 @@ class DBWithTTL : public DB {
 
   static Status AppendTS(const Slice& val, std::string& val_with_ts);
 
+  static Status SanityCheckTimestamp(const std::string& str);
+
   static Status StripTS(std::string* str);
 
   static Status GetCurrentTime(int32_t& curtime);
 
   static const int32_t kTSLength = sizeof(int32_t); // size of timestamp
 
+  static const int32_t kMinTimestamp = 1368146402; // 05/09/2013:5:40PM
+
  private:
   DB* db_;
   int32_t ttl_;