blob: dcc43999f920cd7297cc7f36629073aacebde7a5 [file] [log] [blame]
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -08001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
2/*
3 * Copyright (c) 2012 University of California, Los Angeles
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation;
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
18 * Author: Zhenkai Zhu <zhenkai@cs.ucla.edu>
19 * 卞超轶 Chaoyi Bian <bcy@pku.edu.cn>
20 * Alexander Afanasyev <alexander.afanasyev@ucla.edu>
21 */
22
23#include "sync-full-state.h"
24
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080025#ifndef STANDALONE
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080026#include "ns3/simulator.h"
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080027#endif // STANDALONE
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080028
29#include <boost/make_shared.hpp>
30#include <boost/lambda/lambda.hpp>
31#include <boost/lambda/bind.hpp>
Alexander Afanasyevd94542d2012-03-05 08:41:46 -080032#include <boost/foreach.hpp>
33#include <boost/assert.hpp>
34
35#include "sync-full-leaf.h"
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080036
37using namespace boost;
38namespace ll = boost::lambda;
39
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080040namespace Sync {
41
42
43FullState::FullState ()
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080044// m_lastUpdated is initialized to "not_a_date_time" in STANDALONE mode and to "0" time in NS-3 mode
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080045{
46}
47
48FullState::~FullState ()
49{
50}
51
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080052TimeDurationType
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080053FullState::getTimeFromLastUpdate () const
54{
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080055#ifndef STANDALONE
Alexander Afanasyev017784c2012-03-02 11:44:13 -080056 return ns3::Simulator::Now () - m_lastUpdated;
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080057#else
58 return boost::posix_time::second_clock::universal_time () - m_lastUpdated;
59#endif // STANDALONE
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080060}
Alexander Afanasyevd94542d2012-03-05 08:41:46 -080061
62DigestConstPtr
63FullState::getDigest ()
64{
65 if (m_digest == 0)
66 {
67 m_digest = make_shared<Digest> ();
68 BOOST_FOREACH (LeafConstPtr leaf, m_leaves)
69 {
70 FullLeafConstPtr fullLeaf = dynamic_pointer_cast<const FullLeaf> (leaf);
71 BOOST_ASSERT (fullLeaf != 0);
72 *m_digest << fullLeaf->getDigest ();
73 }
74 }
75
76 return m_digest;
77}
78
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080079// from State
80void
81FullState::update (NameInfoConstPtr info, const SeqNo &seq)
82{
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080083#ifndef STANDALONE
Alexander Afanasyev017784c2012-03-02 11:44:13 -080084 m_lastUpdated = ns3::Simulator::Now ();
Alexander Afanasyev146a51b2012-03-05 10:47:35 -080085#else
86 m_lastUpdated = boost::posix_time::second_clock::universal_time ();
87#endif // STANDALONE
88
Alexander Afanasyevd94542d2012-03-05 08:41:46 -080089 m_digest.reset ();
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080090
Alexander Afanasyevb5547e32012-03-01 21:59:38 -080091 LeafContainer::iterator item = m_leaves.find (*info);
92 if (item == m_leaves.end ())
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080093 {
Alexander Afanasyevb5547e32012-03-01 21:59:38 -080094 m_leaves.insert (make_shared<Leaf> (info, cref (seq)));
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080095 }
96 else
97 {
Alexander Afanasyevb5547e32012-03-01 21:59:38 -080098 m_leaves.modify (item, ll::bind (&Leaf::setSeq, *ll::_1, seq));
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -080099 }
100}
101
102void
103FullState::remove (NameInfoConstPtr info)
104{
Alexander Afanasyev146a51b2012-03-05 10:47:35 -0800105#ifndef STANDALONE
Alexander Afanasyev017784c2012-03-02 11:44:13 -0800106 m_lastUpdated = ns3::Simulator::Now ();
Alexander Afanasyev146a51b2012-03-05 10:47:35 -0800107#else
108 m_lastUpdated = boost::posix_time::second_clock::universal_time ();
109#endif // STANDALONE
110
Alexander Afanasyevd94542d2012-03-05 08:41:46 -0800111 m_digest.reset ();
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -0800112
Alexander Afanasyevb5547e32012-03-01 21:59:38 -0800113 m_leaves.erase (*info);
Alexander Afanasyev7a696fb2012-03-01 17:17:22 -0800114}
115
116
117} // Sync