/* -*-  Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2011 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Author: Jaume Nin <jnin@cttc.cat>

 *         Nicola Baldo <nbaldo@cttc.cat>

 */

#ifndef EPC_ENB_APPLICATION_H

#define EPC_ENB_APPLICATION_H

#include <ns3/address.h>

#include <ns3/socket.h>

#include <ns3/virtual-net-device.h>

#include <ns3/traced-callback.h>

#include <ns3/callback.h>

#include <ns3/ptr.h>

#include <ns3/object.h>

#include <ns3/lte-common.h>

#include <ns3/application.h>

#include <ns3/eps-bearer.h>

#include <ns3/epc-enb-s1-sap.h>

#include <map>

namespace ns3 {

class EpcEnbS1SapUser;

class EpcEnbS1SapProvider;

/**

 * \ingroup lte

 *

 * This application is installed inside eNBs and provides the bridge functionality for user data plane packets between the radio interface and the S1-U interface.

 */

class EpcEnbApplication : public Application

{

  friend class MemberEpcEnbS1SapProvider<EpcEnbApplication>;

public:

  // inherited from Object

  static TypeId GetTypeId (void);

  /** 

   * Constructor

   * 

   * \param lteSocket the socket to be used to send/receive packets to/from the LTE radio interface

   * \param s1uSocket the socket to be used to send/receive packets

   * to/from the S1-U interface connected with the SGW 

   * \param sgwAddress the IPv4 address at which this eNB will be able to reach its SGW

   * 

   */

  EpcEnbApplication (Ptr<Socket> lteSocket, Ptr<Socket> s1uSocket, Ipv4Address sgwAddress);

  /**

   * Destructor

   * 

   */

  virtual ~EpcEnbApplication (void);

  /** 

   * Set the S1 SAP User

   * 

   * \param s the S1 SAP User

   */

  void SetS1SapUser (EpcEnbS1SapUser * s);

  /** 

   * 

   * \return the S1 SAP Provider

   */

  EpcEnbS1SapProvider* GetS1SapProvider ();

  /** 

   * This method is triggered after the eNB received

   * a S1-AP message of type E-RAB Setup Request by the MME and will

   * trigger the corresponding RadioBearer creation 

   * 

   * \param teid the Tunnel Endpoint IDentifier of the S1-bearer to be setup.

   * \param imsi the unique ID of the UE

   * \param bearer the specification of the corresponding EPS bearer

   */

  void ErabSetupRequest (uint32_t teid, uint64_t imsi, EpsBearer bearer);

  /** 

   * Method to be assigned to the recv callback of the LTE socket. It is called when the eNB receives a data packet from the radio interface that is to be forwarded to the SGW.

   * 

   * \param socket pointer to the LTE socket

   */

  void RecvFromLteSocket (Ptr<Socket> socket);

  /** 

   * Method to be assigned to the recv callback of the S1-U socket. It is called when the eNB receives a data packet from the SGW that is to be forwarded to the UE.

   * 

   * \param socket pointer to the S1-U socket

   */

  void RecvFromS1uSocket (Ptr<Socket> socket);

private:

  // S1 SAP provider methods

  void DoS1BearerSetupRequest (EpcEnbS1SapProvider::S1BearerSetupRequestParameters params);

  void DoInitialUeMessage (uint64_t imsi, uint16_t rnti);

  /** 

   * Send a packet to the UE via the LTE radio interface of the eNB

   * 

   * \param packet t

   */

  /** 

   * Send a packet to the SGW via the S1-U interface

   * 

   * \param packet packet to be sent

   * \param teid the Tunnel Enpoint IDentifier

   */

  void SendToS1uSocket (Ptr<Packet> packet, uint32_t teid);

  /** 

   * internal method used for the actual setup of the S1 Bearer

   * 

   * \param teid 

   * \param rnti 

   */

  /**

   * raw packet socket to send and receive the packets to and from the LTE radio interface

   */

  Ptr<Socket> m_lteSocket;

  /**

   * UDP socket to send and receive GTP-U the packets to and from the S1-U interface

   */

  Ptr<Socket> m_s1uSocket;

  /**

   * address of the SGW which terminates all S1-U tunnels

   */

  Ipv4Address m_sgwAddress;

  /**

   * 

   */

  /**

   * 

   */

  /**

   * UDP port to be used for GTP

   */

  uint16_t m_gtpuUdpPort;

  /**

   * Provider for the S1 SAP 

   */

  EpcEnbS1SapProvider* m_s1SapProvider;

  /**

   * User for the S1 SAP 

   */

  EpcEnbS1SapUser* m_s1SapUser;

  /**

   * UE context info

   * 

   */

  std::map<uint64_t, uint16_t> m_imsiRntiMap;

};

} //namespace ns3

#endif /* EPC_ENB_APPLICATION_H */