aboutsummaryrefslogtreecommitdiffstats
path: root/host/lib/transport/dpdk_zero_copy.hpp
blob: f90e73d0e9e70fd3826ab6cdbdb2c4f75e95c58f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
//
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//

#ifndef DPDK_ZERO_COPY_HPP
#define DPDK_ZERO_COPY_HPP

#include <uhd/config.hpp>
#include <uhd/transport/zero_copy.hpp>
#include <uhd/types/device_addr.hpp>
#include <uhd/utils/log.hpp>
#include <uhd/utils/static.hpp>
#include <boost/shared_ptr.hpp>
#include <mutex>
#include <string>
#include <vector>


namespace uhd { namespace transport {

class uhd_dpdk_ctx : boost::noncopyable
{
public:
    UHD_SINGLETON_FCN(uhd_dpdk_ctx, get);

    ~uhd_dpdk_ctx(void);

    /*!
     * Initialize uhd-dpdk (and do only once)
     * \param eal_args Arguments to pass to DPDK rte_eal_init() function
     * \param num_ports Size of port_thread_mapping array (also number in use)
     * \param port_thread_mapping Map NICs to threads: index=port, value=thread
     * \param num_mbufs Number of packet buffers for each port's memory pool
     * \param mbuf_cache_size Size of per-core packet buffer cache from mempool
     * \param mtu MTU of NIC ports
     */
    void init(const dict<std::string, std::string>& eal_args,
        unsigned int num_ports,
        int* port_thread_mapping,
        int num_mbufs,
        int mbuf_cache_size,
        size_t mtu);

    /*!
     * Get port ID from provided MAC address
     * \param mac_addr MAC address
     * \param port_id Int to write ID of port corresponding to MAC address
     * \return 0 if match found, else no match
     */
    int get_port_id(std::array<uint8_t, 6> mac_addr, unsigned int& port_id);

    /*!
     * Get port ID for routing packet destined for given address
     * \param addr Destination address
     * \return port ID from routing table
     */
    int get_route(const std::string& addr) const;

    /*!
     * Set IPv4 address and subnet mask of given NIC port
     * Not thread-safe. Should only be written before ports are in use.
     * \param port_id NIC port ID
     * \param ipv4_addr IPv4 address to write
     * \param netmask Subnet mask identifying network number in ipv4_addr
     * \return 0 if successful, else error
     */
    int set_ipv4_addr(unsigned int port_id, uint32_t ipv4_addr, uint32_t netmask);

    /*!
     * \return whether init() has been called
     */
    bool is_init_done(void);

private:
    uhd_dpdk_ctx(void);

    size_t _mtu;
    std::mutex _init_mutex;
    std::atomic<bool> _init_done;
};

/*!
 * A zero copy transport interface to the dpdk DMA library.
 */
class dpdk_zero_copy : public virtual zero_copy_if
{
public:
    typedef boost::shared_ptr<dpdk_zero_copy> sptr;

    static sptr make(struct uhd_dpdk_ctx& ctx,
        const unsigned int dpdk_port_id,
        const std::string& addr,
        const std::string& remote_port,
        const std::string& local_port, /* 0 = auto-assign */
        const zero_copy_xport_params& default_buff_args,
        const device_addr_t& hints);

    virtual uint16_t get_local_port(void) const = 0;

    virtual std::string get_local_addr(void) const = 0;

    virtual uint32_t get_drop_count(void) const = 0;
};

}} // namespace uhd::transport

#endif /* DPDK_ZERO_COPY_HPP */