bgp_xmpp_peer_close.cc

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */

#include <boost/foreach.hpp>
#include <boost/scoped_ptr.hpp>

#include "base/task_annotations.h"
#include "bgp/bgp_factory.h"
#include "bgp/bgp_path.h"
#include "bgp/bgp_xmpp_peer_close.h"
#include "bgp/bgp_xmpp_rtarget_manager.h"
#include "bgp/peer_close_manager.h"
#include "bgp/bgp_xmpp_channel.h"
#include "xmpp/xmpp_connection.h"
#include "xmpp/xmpp_server.h"

using std::string;
using std::vector;

BgpXmppPeerClose::BgpXmppPeerClose(BgpXmppChannel *channel) :
        channel_(channel) {
}

BgpXmppPeerClose::~BgpXmppPeerClose() {
}

PeerCloseManager *BgpXmppPeerClose::GetManager() const {
    return channel_->close_manager();
}

void BgpXmppPeerClose::UpdateRouteStats(Address::Family family,
                                        const BgpPath *old_path,
                                        uint32_t path_flags) const {
    GetManager()->UpdateRouteStats(family, old_path, path_flags);
}

bool BgpXmppPeerClose::IsReady() const {
    return channel_->Peer()->IsReady();
}

IPeer *BgpXmppPeerClose::peer() const {
    return channel_->Peer();
}

int BgpXmppPeerClose::GetGracefulRestartTime() const {
    return channel_->manager()->xmpp_server()->GetGracefulRestartTime();
}

int BgpXmppPeerClose::GetLongLivedGracefulRestartTime() const {
    return channel_->manager()->xmpp_server()->
        GetLongLivedGracefulRestartTime();
}

// Mark all current subscription as 'stale'
// Concurrency: Protected with a mutex from peer close manager
void BgpXmppPeerClose::GracefulRestartStale() {
    channel_->StaleCurrentSubscriptions();
}

// Mark all current subscriptions as 'llgr_stale'
// Concurrency: Protected with a mutex from peer close manager
void BgpXmppPeerClose::LongLivedGracefulRestartStale() {
    channel_->LlgrStaleCurrentSubscriptions();
}

// Delete all current subscriptions which are still stale.
// Concurrency: Protected with a mutex from peer close manager
void BgpXmppPeerClose::GracefulRestartSweep() {
    channel_->SweepCurrentSubscriptions();
}

bool BgpXmppPeerClose::IsCloseGraceful() const {
    // Do not close gracefully if connection is already marked for deletion.
    if (channel_->channel()->connection()->IsDeleted())
        return false;
    return channel_->manager()->xmpp_server()->IsPeerCloseGraceful();
}

bool BgpXmppPeerClose::IsCloseLongLivedGraceful() const {
    return IsCloseGraceful() && GetLongLivedGracefulRestartTime() != 0;
}

// EoR from xmpp is afi independent at the moment.
void BgpXmppPeerClose::GetGracefulRestartFamilies(Families *families) const {
    families->insert(Address::UNSPEC);
}

void BgpXmppPeerClose::GetLongLivedGracefulRestartFamilies(Families *families) const {
    families->insert(Address::UNSPEC);
}

void BgpXmppPeerClose::ReceiveEndOfRIB(Address::Family family) {
    channel_->ReceiveEndOfRIB(family);
}

// Process any pending subscriptions as close manager is now no longer
// using membership manager.
void BgpXmppPeerClose::MembershipRequestCallbackComplete() {
    CHECK_CONCURRENCY(GetTaskName());
    if (channel_) {
        assert(!GetManager()->IsMembershipInUse());
        channel_->ProcessPendingSubscriptions();
    }
}

const char *BgpXmppPeerClose::GetTaskName() const {
    return "bgp::Config";
}

int BgpXmppPeerClose::GetTaskInstance() const {
    return 0;
}

void BgpXmppPeerClose::CustomClose() {
    channel_->rtarget_manager()->Close();
    channel_->ClearSubscriptions();
}

void BgpXmppPeerClose::CloseComplete() {
    assert(!channel_->channel()->connection()->IsDeleted());
    channel_->set_peer_closed(false);
    channel_->channel()->UnRegisterReceive(xmps::BGP);
}

void BgpXmppPeerClose::Delete() {
    channel_->set_delete_in_progress(true);
    channel_->set_peer_closed(true);
    channel_->manager()->increment_deleting_count();
    channel_->manager()->Enqueue(channel_);
    channel_ = NULL;
}

void BgpXmppPeerClose::Close(bool graceful) {
    if (channel_) {
        assert(channel_->peer_deleted());
        if (!IsCloseGraceful())
            graceful = false;
        GetManager()->Close(graceful);
    }
}