Add a eBPF map to store iface name and index
Since the kernel bpf program can only get the iface index instead of
iface name, we need a seperate map to store the iface index and name
pair in userspace so the kernel program can know what iface each
received packet is and account against the correct name.
Test: run cts -m TrafficStatsTest
Bug: 30950746
Bug: 73137611
Change-Id: I6638dc4b03db6fd18b6b38b4524ec89e25a55bc0
diff --git a/server/TrafficController.h b/server/TrafficController.h
index cbef8fc..7fdc3b7 100644
--- a/server/TrafficController.h
+++ b/server/TrafficController.h
@@ -32,11 +32,13 @@
// uid_counter_set_map: key: 4 bytes, value: 4 bytes, total:10000*4*2 bytes = 80Kbytes
// uid_stats_map: key: 16 bytes, value: 32 bytes, total:10000*16+10000*32 bytes = 480Kbytes
// tag_stats_map: key: 16 bytes, value: 32 bytes, total:10000*16+10000*32 bytes = 480Kbytes
-// total: 1200Kbytes
+// iface_index_name_map:key: 4 bytes, value: 32 bytes, total:10000*36 bytes = 360Kbytes
+// total: 1560Kbytes
constexpr const int COOKIE_UID_MAP_SIZE = 10000;
constexpr const int UID_COUNTERSET_MAP_SIZE = 10000;
constexpr const int UID_STATS_MAP_SIZE = 10000;
constexpr const int TAG_STATS_MAP_SIZE = 10000;
+constexpr const int IFACE_INDEX_NAME_MAP_SIZE = 10000;
constexpr const int UID_OWNER_MAP_SIZE = 10000;
constexpr const int COUNTERSETS_LIMIT = 2;
@@ -92,6 +94,11 @@
*/
bool checkBpfStatsEnable();
+ /*
+ * Add the interface name and index pair into the eBPF map.
+ */
+ int addInterface(const char* name, uint32_t ifaceIndex);
+
private:
/*
* mCookieTagMap: Store the corresponding tag and uid for a specific socket.
@@ -131,6 +138,13 @@
*/
base::unique_fd mTagStatsMap;
+ /*
+ * mIfaceIndexNameMap: Store the index name pair of each interface show up
+ * on the device since boot. The interface index is used by the eBPF program
+ * to correctly match the iface name when receiving a packet.
+ */
+ base::unique_fd mIfaceIndexNameMap;
+
std::unique_ptr<NetlinkListenerInterface> mSkDestroyListener;
bool ebpfSupported;