15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/protocol/message_decoder.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/base/compound_buffer.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/proto/internal.pb.h" 115f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)#include "third_party/webrtc/base/byteorder.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace protocol { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)MessageDecoder::MessageDecoder() 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : next_payload_(0), 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_payload_known_(false) { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)MessageDecoder::~MessageDecoder() {} 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void MessageDecoder::AddData(scoped_refptr<net::IOBuffer> data, 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int data_size) { 25868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) buffer_.Append(data.get(), data_size); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CompoundBuffer* MessageDecoder::GetNextMessage() { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Determine the payload size. If we already know it then skip this part. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We may not have enough data to determine the payload size so use a 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // utility function to find out. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int next_payload = -1; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!next_payload_known_ && GetPayloadSize(&next_payload)) { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(-1, next_payload); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_payload_ = next_payload; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_payload_known_ = true; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the next payload size is still not known or we don't have enough 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data for parsing then exit. 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!next_payload_known_ || buffer_.total_bytes() < next_payload_) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompoundBuffer* message_buffer = new CompoundBuffer(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_buffer->CopyFrom(buffer_, 0, next_payload_); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_buffer->Lock(); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_.CropFront(next_payload_); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_payload_known_ = false; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return message_buffer; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool MessageDecoder::GetPayloadSize(int* size) { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The header has a size of 4 bytes. 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const int kHeaderSize = sizeof(int32); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (buffer_.total_bytes() < kHeaderSize) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompoundBuffer header_buffer; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char header[kHeaderSize]; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header_buffer.CopyFrom(buffer_, 0, kHeaderSize); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) header_buffer.CopyTo(header, kHeaderSize); 645f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) *size = rtc::GetBE32(header); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_.CropFront(kHeaderSize); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace protocol 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 71