Packets/80221Bundle: fixed MIH message validation unit test @ ubuntu hardy

[senf.git] / senf / Packets / 80221Bundle / TLVParser.cc

diff --git a/senf/Packets/80221Bundle/TLVParser.cc b/senf/Packets/80221Bundle/TLVParser.cc
index 78c43c0..5d6a20c 100644 (file)
--- a/senf/Packets/80221Bundle/TLVParser.cc
+++ b/senf/Packets/80221Bundle/TLVParser.cc
@@ -27,19 +27,34 @@
 //#include "TLVParser.ih"
 
 // Custom includes
-#include <iomanip>
 #include <senf/Utils/hexdump.hh>
 #include <senf/Utils/Format.hh>
+#include <senf/Utils/String.hh>
 
 #define prefix_
-///////////////////////////////cc.p////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 
 SENF_PACKET_TLV_REGISTRY_REGISTER( senf::MIHFSrcIdTLVParser );
 SENF_PACKET_TLV_REGISTRY_REGISTER( senf::MIHFDstIdTLVParser );
 SENF_PACKET_TLV_REGISTRY_REGISTER( senf::MIHStatusTLVParser );
 SENF_PACKET_TLV_REGISTRY_REGISTER( senf::MIHValidTimeIntervalTLVParser );
 
-///////////////////////////////////////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
+// MIHBaseTLVParser
+
+prefix_ std::pair<bool, std::string> senf::MIHBaseTLVParser::validateTL(boost::uint8_t expectedType, MIHTLVLengthParser::value_type expectedLength)
+    const
+{
+    if (! check( 1 + senf::bytes(length_()) + length()) )
+        return std::make_pair(false, "truncated TLV. Type: " + senf::str(type()));
+    if (type() != expectedType)
+        return std::make_pair(false, "invalid TLV type: " + senf::str(type()));
+    if (length() != expectedLength)
+        return std::make_pair(false, "invalid length in TLV. Type: " + senf::str(type()));
+    return std::make_pair(true, "");
+}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHBaseListTLVParser
 
 prefix_ void senf::MIHBaseListTLVParser::maxListSize(MIHTLVLengthParser::value_type maxl)
@@ -49,7 +64,8 @@ prefix_ void senf::MIHBaseListTLVParser::maxListSize(MIHTLVLengthParser::value_t
     maxLength( maxl + senf::bytes(listSize_()));
 }
 
-///////////////////////////////////////////////////////////////////////////
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHFIdTLVParser
 
 prefix_ void senf::MIHFIdTLVParser::dump(std::ostream & os)
@@ -156,7 +172,7 @@ prefix_ senf::MIHFId senf::MIHFIdTLVParser::valueAs(MIHFId::Type type)
 }
 
 
-///////////////////////////////////////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHFSrcIdTLVParser
 
 prefix_ void senf::MIHFSrcIdTLVParser::dump(std::ostream & os)
@@ -167,7 +183,18 @@ prefix_ void senf::MIHFSrcIdTLVParser::dump(std::ostream & os)
     MIHFIdTLVParser::dump(os);
 }
 
-///////////////////////////////////////////////////////////////////////////
+prefix_ std::pair<bool, std::string> senf::MIHFSrcIdTLVParser::validate()
+    const
+{
+    if (! check( 1 + senf::bytes(length_()) + length()) )
+        return std::make_pair(false, "truncated TLV. Type: " + senf::str(type()));
+    if (type() != typeId)
+        return std::make_pair(false, "invalid TLV type: " + senf::str(type()));
+    return std::make_pair(true, "");
+}
+
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHFDstIdTLVParser
 
 prefix_ void senf::MIHFDstIdTLVParser::dump(std::ostream & os)
@@ -178,7 +205,17 @@ prefix_ void senf::MIHFDstIdTLVParser::dump(std::ostream & os)
     MIHFIdTLVParser::dump(os);
 }
 
-///////////////////////////////////////////////////////////////////////////
+prefix_ std::pair<bool, std::string> senf::MIHFDstIdTLVParser::validate()
+    const
+{
+    if (! check( 1 + senf::bytes(length_()) + length()) )
+        return std::make_pair(false, "truncated TLV. Type: " + senf::str(type()));
+    if (type() != typeId)
+        return std::make_pair(false, "invalid TLV type: " + senf::str(type()));
+    return std::make_pair(true, "");
+}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHStatusTLVParser
 
 prefix_ void senf::MIHStatusTLVParser::dump(std::ostream & os)
@@ -210,7 +247,17 @@ prefix_ void senf::MIHStatusTLVParser::dump(std::ostream & os)
     os << " (???; invalid value!)" << std::endl;
 }
 
-///////////////////////////////////////////////////////////////////////////
+prefix_ std::pair<bool, std::string> senf::MIHStatusTLVParser::validate()
+    const
+{
+    if (! validateTL( typeId, 1).first)
+        return validateTL( typeId, 1);
+    if (value() >= 4)
+        return std::make_pair(false, "invalid value in MIHStatusTLV " + senf::str(value()));
+    return std::make_pair(true, "");
+}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHRegisterReqCodeTLVParser
 
 prefix_ void senf::MIHRegisterReqCodeTLVParser::dump(std::ostream & os)
@@ -233,7 +280,17 @@ prefix_ void senf::MIHRegisterReqCodeTLVParser::dump(std::ostream & os)
     os << " (???; invalid value!)" << std::endl;
 }
 
-///////////////////////////////////////////////////////////////////////////
+prefix_ std::pair<bool, std::string> senf::MIHRegisterReqCodeTLVParser::validate()
+    const
+{
+    if (! validateTL( typeId, 1).first)
+        return validateTL( typeId, 1);
+    if (value() >= 2)
+        return std::make_pair(false, "invalid value in MIHRegisterReqCodeTLV " + senf::str(value()));
+    return std::make_pair(true, "");
+}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHValidTimeIntervalTLVParser
 
 prefix_ void senf::MIHValidTimeIntervalTLVParser::dump(std::ostream & os)
@@ -248,7 +305,14 @@ prefix_ void senf::MIHValidTimeIntervalTLVParser::dump(std::ostream & os)
        << ( value()==0 ? " (infinite)" : " seconds") << std::endl;
 }
 
-///////////////////////////////////////////////////////////////////////////
+prefix_ std::pair<bool, std::string> senf::MIHValidTimeIntervalTLVParser::validate()
+    const
+{
+    if (! validateTL( typeId, 4).first) return validateTL( typeId, 4);
+    return std::make_pair(true, "");
+}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::MIHTLVLengthParser
 
 prefix_ senf::MIHTLVLengthParser::value_type senf::MIHTLVLengthParser::value() const
@@ -385,7 +449,7 @@ prefix_ void senf::MIHTLVLengthParser::resize_(size_type size)
 }
 
 
-///////////////////////////////cc.e////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 #undef prefix_
 
 \f