Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
       3             :  */
       4             : 
       5             : #include "base/proto.h"
       6             : 
       7             : #include <boost/bind/bind.hpp>
       8             : 
       9             : using namespace std;
      10             : using namespace boost::placeholders;
      11             : 
      12             : namespace detail {
      13             :     bool debug_ = false;
      14             : }
      15             : 
      16             : struct ParseContext::StackFrame {
      17     2638535 :     StackFrame() : offset(0), lensize(0), size(-1), total_size(-1) {
      18     2638449 :     }
      19             :     int offset; // offset of the data pointer at present
      20             :     int lensize; //size of the length of the current element being parsed
      21             :     size_t size;
      22             :     size_t total_size;
      23             :     std::unique_ptr<ParseObject> data;
      24             : };
      25             : 
      26      195767 : ParseContext::ParseContext()
      27      195767 :     : offset_(0) {
      28      195764 : }
      29             : 
      30             : struct deleter {
      31             :     template <typename T>
      32      196148 :     void operator()(T *ptr) {
      33      196148 :         delete ptr;
      34      196149 :     }
      35             : };
      36             : 
      37      195759 : ParseContext::~ParseContext() {
      38      195759 :     for_each(stack_.begin(), stack_.end(), deleter());
      39      195757 :     stack_.clear();
      40      195760 : }
      41             : 
      42      188221 : ParseObject *ParseContext::release() {
      43      188221 :     if (stack_.empty()) {
      44           0 :         return NULL;
      45             :     }
      46      188219 :     StackFrame *current = stack_.back();
      47      188222 :     return current->data.release();
      48             : }
      49             : 
      50     2638468 : void ParseContext::Push(ParseObject *data) {
      51     2638468 :     StackFrame *frame = new StackFrame();
      52     2638443 :     frame->data.reset(data);
      53     2638342 :     stack_.push_back(frame);
      54     2638382 : }
      55             : 
      56     2630450 : ParseObject *ParseContext::Pop() {
      57     2630450 :     if (stack_.size() <= 1) {
      58      188216 :         return NULL;
      59             :     }
      60     2442224 :     StackFrame *frame = stack_.back();
      61     2442219 :     ParseObject *obj = frame->data.release();
      62     2442194 :     stack_.pop_back();
      63     2442197 :     delete frame;
      64     2442243 :     return obj;
      65             : }
      66             : 
      67      656994 : void ParseContext::ReleaseData() {
      68      656994 :     if (!stack_.empty()) {
      69      656986 :         StackFrame *frame = stack_.back();
      70      656984 :         frame->data.release();
      71             :     }
      72      656973 : }
      73     1280066 : void ParseContext::SwapData(ParseObject *obj) {
      74     1280066 :     if (!stack_.empty() && obj) {
      75     1280058 :         StackFrame *frame = stack_.back();
      76     1280043 :         frame->data.reset(obj);
      77             :     } else {
      78           1 :         delete obj;
      79             :     }
      80     1280023 : }
      81             : 
      82           0 : ParseObject *ParseContext::data() {
      83           0 :     if (stack_.empty()) {
      84           0 :         return NULL;
      85             :     }
      86           0 :     StackFrame *current = stack_.back();
      87           0 :     return current->data.get();
      88             : }
      89             : 
      90     9181490 : void ParseContext::advance(int delta) {
      91     9181490 :     if (!stack_.empty()) {
      92     8602610 :         stack_.back()->offset += delta;
      93             :     }
      94     9180569 :     offset_ += delta;
      95     9180569 : }
      96             : 
      97     2146530 : void ParseContext::set_lensize(int length) {
      98     2146530 :     if (!stack_.empty()) {
      99     2146508 :         StackFrame *current = stack_.back();
     100     2146493 :         current->lensize = length;
     101             :     }
     102     2146493 : }
     103             : 
     104     3256636 : int ParseContext::lensize() const {
     105     3256636 :     if (stack_.empty()) {
     106           0 :         return -1;
     107             :     }
     108     3256617 :     StackFrame *current = stack_.back();
     109     3256608 :     return current->lensize;
     110             : }
     111             : 
     112     7212857 : void ParseContext::set_size(size_t length) {
     113     7212857 :     if (!stack_.empty()) {
     114     6826216 :         StackFrame *current = stack_.back();
     115     6826125 :         current->size = length;
     116             :     }
     117     7212522 : }
     118             : 
     119    23408803 : size_t ParseContext::size() const {
     120    23408803 :     if (stack_.empty()) {
     121      969039 :         return -1;
     122             :     }
     123    22436986 :     StackFrame *current = stack_.back();
     124    22436088 :     return current->size;
     125             : }
     126             : 
     127     2772445 : void ParseContext::set_total_size() {
     128     2772445 :     if (!stack_.empty()) {
     129     2772423 :         StackFrame *current = stack_.back();
     130     2772424 :         current->total_size = current->size;
     131             :     }
     132     2772424 : }
     133             : 
     134       85949 : size_t ParseContext::total_size() const {
     135       85949 :     if (stack_.empty()) {
     136           0 :         return -1;
     137             :     }
     138       85948 :     StackFrame *current = stack_.back();
     139       85948 :     return current->total_size >= 0 ? current->total_size : current->size;
     140             : }
     141             : 
     142        7840 : void ParseContext::SetError(int error, int subcode, std::string type,
     143             :                             const uint8_t *data, int data_size) {
     144        7840 :     if (error_context_.error_code) {
     145             :         // Error already set. Ignore this one.
     146         300 :         return;
     147             :     }
     148        7540 :     error_context_.error_code = error;
     149        7540 :     error_context_.error_subcode = subcode;
     150        7540 :     error_context_.type_name = type;
     151        7540 :     error_context_.data = data;
     152        7540 :     error_context_.data_size = data_size;
     153             : }
     154             : 
     155             : struct EncodeContext::StackFrame {
     156             :     typedef boost::function<void(EncodeContext *)> CallbackFn;
     157             :     typedef std::vector<CallbackFn>::iterator CallbackIterator;
     158     5049277 :     StackFrame() : offset(0) {
     159     5048996 :     }
     160             :     int offset;
     161             :     std::vector<CallbackFn> callbacks;
     162             : };
     163             : 
     164      273830 : EncodeContext::EncodeContext() {
     165      273823 : }
     166             : 
     167      273834 : EncodeContext::~EncodeContext() {
     168      273834 : }
     169             : 
     170     5048774 : void EncodeContext::Push() {
     171     5048774 :     StackFrame *frame = new StackFrame();
     172     5048949 :     stack_.push_back(frame);
     173     5046891 : }
     174             : 
     175     5048306 : void EncodeContext::Pop(bool callback) {
     176     5048306 :     StackFrame *frame = &stack_.back();
     177     5047080 :     if (callback) {
     178     5046956 :         for (StackFrame::CallbackIterator iter =
     179    12515841 :              frame->callbacks.begin(); iter != frame->callbacks.end(); ++iter) {
     180     2421972 :             StackFrame::CallbackFn cb = *iter;
     181     2421895 :             (cb)(this);
     182     2421642 :         }
     183             :     }
     184             : 
     185     5047505 :     int p_offset = frame->offset;
     186     5047505 :     stack_.pop_back();                  // deletes frame
     187     5047379 :     if (!stack_.empty()) {
     188     4773558 :         StackFrame *top = &(stack_.back());
     189     4772262 :         top->offset += p_offset;
     190             :     }
     191     5045437 : }
     192             : 
     193     8550547 : void EncodeContext::advance(int delta) {
     194     8550547 :     if (stack_.empty()) {
     195           0 :         return;
     196             :     }
     197     8548033 :     StackFrame *frame = &stack_.back();
     198     8544495 :     frame->offset += delta;
     199             : }
     200             : 
     201     2185544 : int EncodeContext::length() const {
     202     2185544 :     const StackFrame *frame = &stack_.back();
     203     2185276 :     return frame->offset;
     204             : };
     205             : 
     206     2422658 : void EncodeContext::AddCallback(CallbackType cb, uint8_t *data,
     207             :                                 int element_size) {
     208     2422658 :     StackFrame *top = &(stack_.back());
     209     2422183 :     top->callbacks.push_back(
     210     4843672 :         boost::bind(cb, _1, data, top->offset, element_size));
     211     2421300 : }
     212             : 
     213      569075 : void EncodeOffsets::SaveOffset(std::string key, int offset) {
     214             :     std::pair<std::map<std::string, int>::iterator, bool > result =
     215      569075 :             offsets_.insert(make_pair(key, offset));
     216      569282 :     if (!result.second) {
     217           4 :         PROTO_DEBUG("Offset for " << key << " already set");
     218             :     }
     219      569282 : }
     220             : 
     221      569024 : void EncodeContext::SaveOffset(std::string key) {
     222      569024 :     int length = 0;
     223      569024 :     for (boost::ptr_vector<StackFrame>::iterator it = stack_.begin();
     224     2195348 :          it != stack_.end(); ++it) {
     225     1626358 :         length += it->offset;
     226             :     }
     227      568862 :     PROTO_DEBUG("Saving Offset for " << key << " at " << length);
     228      568862 :     offsets_.SaveOffset(key, length);
     229      569274 : }
     230             : 
     231      187883 : int EncodeOffsets::FindOffset(const char *key) {
     232      187883 :     std::map<std::string, int>::iterator it = offsets_.find(std::string(key));
     233      187877 :     if (it == offsets_.end())
     234           0 :         return -1;
     235      187881 :     return it->second;
     236             : }