1 files changed, 43 insertions, 4 deletions

diff --git a/remoting/protocol/rtp_reader.cc b/remoting/protocol/rtp_reader.cc
index 7e21250..f36dc8b 100644
--- a/remoting/protocol/rtp_reader.cc
+++ b/remoting/protocol/rtp_reader.cc
@@ -10,12 +10,25 @@
 namespace remoting {
 namespace protocol {
 
-RtpPacket::RtpPacket() {}
-RtpPacket::~RtpPacket() {}
+namespace {
+// Recomended values from RTP spec.
+const int kMaxDropout = 3000;
+const int kMaxMisorder = 100;
+}  // namespace
+
+RtpPacket::RtpPacket() { }
+
+RtpPacket::~RtpPacket() { }
 
 // RtpReader class.
-RtpReader::RtpReader() {}
-RtpReader::~RtpReader() {}
+RtpReader::RtpReader()
+    : max_sequence_number_(0),
+      wrap_around_count_(0),
+      started_(false) {
+}
+
+RtpReader::~RtpReader() {
+}
 
 void RtpReader::Init(net::Socket* socket,
                      OnMessageCallback* on_message_callback) {
@@ -44,6 +57,32 @@ void RtpReader::OnDataReceived(net::IOBuffer* buffer, int data_size) {
       buffer, buffer->data() + header_size + descriptor_size,
       data_size - header_size - descriptor_size);
 
+  uint16 sequence_number = packet->header().sequence_number;
+
+  // Reset |max_sequence_number_| after we've received first packet.
+  if (!started_) {
+    started_ = true;
+    max_sequence_number_ = sequence_number;
+  }
+
+  int16 delta = sequence_number - max_sequence_number_;
+  if (delta <= -kMaxMisorder || delta > kMaxDropout) {
+    // TODO(sergeyu): Do we need to handle restarted trasmission?
+    LOG(WARNING) << "Received RTP packet with invalid sequence number.";
+    delete packet;
+    return;
+  }
+
+  packet->set_extended_sequence_number(
+      (wrap_around_count_ << 16) + max_sequence_number_ + delta);
+
+  if (delta > 0 && delta < kMaxDropout) {
+    if (sequence_number < max_sequence_number_) {
+      wrap_around_count_++;
+    }
+    max_sequence_number_ = sequence_number;
+  }
+
   on_message_callback_->Run(packet);
 }