104 lines
3.1 KiB
C++
104 lines
3.1 KiB
C++
|
|
#include <string.h>
|
|
#include <map>
|
|
|
|
#include "srslte/common/buffer_pool.h"
|
|
#include "srslte/common/log.h"
|
|
#include "upper/common_enb.h"
|
|
#include "srslte/common/threads.h"
|
|
#include "srslte/srslte.h"
|
|
#include "srslte/interfaces/enb_interfaces.h"
|
|
|
|
#ifndef GTPU_H
|
|
#define GTPU_H
|
|
|
|
namespace srsenb {
|
|
|
|
/****************************************************************************
|
|
* GTPU Header
|
|
* Ref: 3GPP TS 29.281 v10.1.0 Section 5
|
|
*
|
|
* | 8 | 7 | 6 | 5 | 4 | 3 | 2 | 1 |
|
|
*
|
|
* 1 | Version |PT | * | E | S |PN |
|
|
* 2 | Message Type |
|
|
* 3 | Length (1st Octet) |
|
|
* 4 | Length (2nd Octet) |
|
|
* 5 | TEID (1st Octet) |
|
|
* 6 | TEID (2nd Octet) |
|
|
* 7 | TEID (3rd Octet) |
|
|
* 8 | TEID (4th Octet) |
|
|
***************************************************************************/
|
|
|
|
#define GTPU_HEADER_LEN 8
|
|
|
|
typedef struct{
|
|
uint8_t flags; // Only support 0x30 - v1, PT1 (GTP), no other flags
|
|
uint8_t message_type; // Only support 0xFF - T-PDU type
|
|
uint16_t length;
|
|
uint32_t teid;
|
|
}gtpu_header_t;
|
|
|
|
class gtpu
|
|
:public gtpu_interface_rrc
|
|
,public gtpu_interface_pdcp
|
|
,public thread
|
|
{
|
|
public:
|
|
|
|
bool init(std::string gtp_bind_addr_, std::string mme_addr_, pdcp_interface_gtpu *pdcp_, srslte::log *gtpu_log_);
|
|
void stop();
|
|
|
|
// gtpu_interface_rrc
|
|
void add_bearer(uint16_t rnti, uint32_t lcid, uint32_t teid_out, uint32_t *teid_in);
|
|
void rem_bearer(uint16_t rnti, uint32_t lcid);
|
|
void rem_user(uint16_t rnti);
|
|
|
|
// gtpu_interface_pdcp
|
|
void write_pdu(uint16_t rnti, uint32_t lcid, srslte::byte_buffer_t *pdu);
|
|
|
|
|
|
private:
|
|
static const int THREAD_PRIO = 7;
|
|
static const int GTPU_PORT = 2152;
|
|
srslte::byte_buffer_pool *pool;
|
|
bool running;
|
|
bool run_enable;
|
|
|
|
std::string gtp_bind_addr;
|
|
std::string mme_addr;
|
|
srsenb::pdcp_interface_gtpu *pdcp;
|
|
srslte::log *gtpu_log;
|
|
|
|
typedef struct{
|
|
uint32_t teids_in[SRSENB_N_RADIO_BEARERS];
|
|
uint32_t teids_out[SRSENB_N_RADIO_BEARERS];
|
|
}bearer_map;
|
|
std::map<uint16_t, bearer_map> rnti_bearers;
|
|
|
|
srslte_netsink_t snk;
|
|
srslte_netsource_t src;
|
|
|
|
void run_thread();
|
|
|
|
pthread_mutex_t mutex;
|
|
|
|
/****************************************************************************
|
|
* Header pack/unpack helper functions
|
|
* Ref: 3GPP TS 29.281 v10.1.0 Section 5
|
|
***************************************************************************/
|
|
bool gtpu_write_header(gtpu_header_t *header, srslte::byte_buffer_t *pdu);
|
|
bool gtpu_read_header(srslte::byte_buffer_t *pdu, gtpu_header_t *header);
|
|
|
|
/****************************************************************************
|
|
* TEID to RNIT/LCID helper functions
|
|
***************************************************************************/
|
|
void teidin_to_rntilcid(uint32_t teidin, uint16_t *rnti, uint16_t *lcid);
|
|
void rntilcid_to_teidin(uint16_t rnti, uint16_t lcid, uint32_t *teidin);
|
|
};
|
|
|
|
|
|
} // namespace srsenb
|
|
|
|
#endif // GTPU_H
|