srsLTE/srsepc/hdr/spgw/gtpc.h

107 lines
3.7 KiB
C++

/**
*
* \section COPYRIGHT
*
* Copyright 2013-2021 Software Radio Systems Limited
*
* By using this file, you agree to the terms and conditions set
* forth in the LICENSE file which can be found at the top level of
* the distribution.
*
*/
#ifndef SRSEPC_GTPC_H
#define SRSEPC_GTPC_H
#include "srsepc/hdr/spgw/spgw.h"
#include "srsran/asn1/gtpc.h"
#include "srsran/common/standard_streams.h"
#include "srsran/interfaces/epc_interfaces.h"
#include "srsran/srslog/srslog.h"
#include <set>
#include <sys/socket.h>
#include <sys/un.h>
namespace srsepc {
class spgw::gtpc : public gtpc_interface_gtpu
{
public:
gtpc();
virtual ~gtpc();
int init(spgw_args_t* args, spgw* spgw, gtpu_interface_gtpc* gtpu, const std::map<std::string, uint64_t>& ip_to_imsi);
void stop();
int init_s11(spgw_args_t* args);
int init_ue_ip(spgw_args_t* args, const std::map<std::string, uint64_t>& ip_to_imsi);
int get_s11();
uint64_t get_new_ctrl_teid();
uint64_t get_new_user_teid();
in_addr_t get_new_ue_ipv4(uint64_t imsi);
void handle_s11_pdu(srsran::byte_buffer_t* msg);
bool send_s11_pdu(const srsran::gtpc_pdu& pdu);
void handle_create_session_request(const srsran::gtpc_create_session_request& cs_req);
void handle_modify_bearer_request(const srsran::gtpc_header& mb_req_hdr,
const srsran::gtpc_modify_bearer_request& mb_req);
void handle_delete_session_request(const srsran::gtpc_header& header,
const srsran::gtpc_delete_session_request& del_req);
void handle_release_access_bearers_request(const srsran::gtpc_header& header,
const srsran::gtpc_release_access_bearers_request& rel_req);
void
handle_downlink_data_notification_acknowledge(const srsran::gtpc_header& header,
const srsran::gtpc_downlink_data_notification_acknowledge& not_ack);
void handle_downlink_data_notification_failure_indication(
const srsran::gtpc_header& header,
const srsran::gtpc_downlink_data_notification_failure_indication& not_fail);
virtual bool queue_downlink_packet(uint32_t spgw_ctr_teid, srsran::unique_byte_buffer_t msg) override;
virtual bool send_downlink_data_notification(uint32_t spgw_ctr_teid) override;
spgw_tunnel_ctx_t* create_gtpc_ctx(const srsran::gtpc_create_session_request& cs_req);
bool delete_gtpc_ctx(uint32_t ctrl_teid);
bool free_all_queued_packets(spgw_tunnel_ctx_t* tunnel_ctx);
spgw* m_spgw;
gtpu_interface_gtpc* m_gtpu;
int m_s11;
struct sockaddr_un m_spgw_addr, m_mme_addr;
uint32_t m_h_next_ue_ip;
uint64_t m_next_ctrl_teid;
uint64_t m_next_user_teid;
uint32_t m_max_paging_queue;
std::map<uint64_t, uint32_t> m_imsi_to_ctr_teid; // IMSI to control TEID map. Important to check if UE
// is previously connected
std::map<uint32_t, spgw_tunnel_ctx*> m_teid_to_tunnel_ctx; // Map control TEID to tunnel ctx. Usefull to get
// reply ctrl TEID, UE IP, etc.
std::set<uint32_t> m_ue_ip_addr_pool;
std::map<uint64_t, struct in_addr> m_imsi_to_ip;
srslog::basic_logger& m_logger = srslog::fetch_basic_logger("SPGW GTPC");
};
inline int spgw::gtpc::get_s11()
{
return m_s11;
}
inline uint64_t spgw::gtpc::get_new_ctrl_teid()
{
return m_next_ctrl_teid++;
}
inline uint64_t spgw::gtpc::get_new_user_teid()
{
return m_next_user_teid++;
}
} // namespace srsepc
#endif // SRSEPC_SPGW_H