diff --git a/src/lte/model/lte-enb-rrc.cc b/src/lte/model/lte-enb-rrc.cc index 0f3a29584..66e828629 100644 --- a/src/lte/model/lte-enb-rrc.cc +++ b/src/lte/model/lte-enb-rrc.cc @@ -1097,6 +1097,11 @@ LteEnbRrc::GetTypeId (void) BooleanValue (true), MakeBooleanAccessor (&LteEnbRrc::m_admitHandoverRequest), MakeBooleanChecker ()) + .AddAttribute ("AdmitRrcConnectionRequest", + "Whether to admit a connection request from a Ue", + BooleanValue (true), + MakeBooleanAccessor (&LteEnbRrc::m_admitRrcConnectionRequest), + MakeBooleanChecker ()) .AddTraceSource ("ConnectionEstablished", "trace fired upon successful RRC connection establishment", MakeTraceSourceAccessor (&LteEnbRrc::m_connectionEstablishedTrace)) @@ -1273,7 +1278,18 @@ void LteEnbRrc::DoRecvRrcConnectionRequest (uint16_t rnti, LteRrcSap::RrcConnectionRequest msg) { NS_LOG_FUNCTION (this << rnti); - GetUeManager (rnti)->RecvRrcConnectionRequest (msg); + + if (m_admitRrcConnectionRequest == true) + { + GetUeManager (rnti)->RecvRrcConnectionRequest (msg); + } + else + { + NS_LOG_INFO ("rejecting connection request to rnti " << rnti); + LteRrcSap::RrcConnectionReject rejectMsg; + rejectMsg.waitTime = 3; + m_rrcSapUser->SendRrcConnectionReject (rnti, rejectMsg); + } } void diff --git a/src/lte/model/lte-enb-rrc.h b/src/lte/model/lte-enb-rrc.h index 42f933d42..48b8597f5 100644 --- a/src/lte/model/lte-enb-rrc.h +++ b/src/lte/model/lte-enb-rrc.h @@ -574,6 +574,7 @@ public: RLC_UM_ALWAYS = 2, RLC_AM_ALWAYS = 3, PER_BASED = 4}; + private: @@ -725,6 +726,7 @@ private: bool m_reconfigureUes; bool m_admitHandoverRequest; + bool m_admitRrcConnectionRequest; // imsi cellid rnti TracedCallback m_connectionEstablishedTrace; diff --git a/src/lte/model/lte-rrc-protocol-real.cc b/src/lte/model/lte-rrc-protocol-real.cc index 5284d76f8..ad1b9acba 100644 --- a/src/lte/model/lte-rrc-protocol-real.cc +++ b/src/lte/model/lte-rrc-protocol-real.cc @@ -288,11 +288,13 @@ LteUeRrcProtocolReal::DoReceivePdcpPdu (Ptr p) RrcConnectionReestablishmentHeader rrcConnectionReestablishmentHeader; RrcConnectionReestablishmentRejectHeader rrcConnectionReestablishmentRejectHeader; RrcConnectionSetupHeader rrcConnectionSetupHeader; + RrcConnectionRejectHeader rrcConnectionRejectHeader; // Declare possible messages LteRrcSap::RrcConnectionReestablishment rrcConnectionReestablishmentMsg; LteRrcSap::RrcConnectionReestablishmentReject rrcConnectionReestablishmentRejectMsg; LteRrcSap::RrcConnectionSetup rrcConnectionSetupMsg; + LteRrcSap::RrcConnectionReject rrcConnectionRejectMsg; // Deserialize packet and call member recv function with appropiate structure switch ( rrcDlCcchMessage.GetMessageType () ) @@ -311,7 +313,9 @@ LteUeRrcProtocolReal::DoReceivePdcpPdu (Ptr p) break; case 2: // RrcConnectionReject - // ... + p->RemoveHeader (rrcConnectionReestablishmentRejectHeader); + rrcConnectionReestablishmentRejectMsg = rrcConnectionReestablishmentRejectHeader.GetMessage (); + m_ueRrcSapProvider->RecvRrcConnectionReject (rrcConnectionRejectMsg); break; case 3: // RrcConnectionSetup diff --git a/src/lte/model/lte-ue-rrc.cc b/src/lte/model/lte-ue-rrc.cc index 9b7b5531a..12cba45dc 100644 --- a/src/lte/model/lte-ue-rrc.cc +++ b/src/lte/model/lte-ue-rrc.cc @@ -693,6 +693,7 @@ void LteUeRrc::DoRecvRrcConnectionReject (LteRrcSap::RrcConnectionReject msg) { NS_LOG_FUNCTION (this); + LeaveConnectedMode(); }