varun-r-mallya/pylibbpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/pylibbpf.git synced 2025-12-31 20:36:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Redesign BpfMap

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-18 14:51:11 +05:30
parent 5c74be041e
commit 54acc2c15d
2 changed files with 13 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/core/bpf_map.h
View File

@ -1,54 +1,44 @@
#ifndef PYLIBBPF_MAPS_H
#define PYLIBBPF_MAPS_H
#ifndef PYLIBBPF_BPF_MAP_H
#define PYLIBBPF_BPF_MAP_H
#include <libbpf.h>
#include <pybind11/pybind11.h>
#include <vector>
#include <string>
#include "bpf_program.h"
class BpfObject;
namespace py = pybind11;
class BpfMap {
private:
std::weak_ptr<BpfObject> parent_obj_;
struct bpf_map *map_;
int map_fd = -1;
//TODO: turn below into a shared pointer and ref count it so that there is no resource leakage
BpfProgram *bpf_program;
int map_fd_;
std::string map_name_;
public:
BpfMap(BpfProgram *program_, const py::object &map_from_python);
BpfMap(std::shared_ptr<BpfObject>, struct bpf_map *raw_map, const std::string &map_name);
~BpfMap() = default;
[[nodiscard]] py::object lookup(const py::object &key) const;
void update(const py::object &key, const py::object &value) const;
void delete_elem(const py::object &key) const;
py::list get_next_key(const py::object &key = py::none()) const;
py::dict items() const;
py::list keys() const;
py::list values() const;
[[nodiscard]] std::string get_name() const;
int get_type() const;
int get_key_size() const;
int get_value_size() const;
int get_max_entries() const;
[[nodiscard]] std::string get_name() const { return map_name_; }
[[nodiscard]] int get_fd() const { return map_fd_; }
[[nodiscard]] int get_type() const;
[[nodiscard]] int get_key_size() const;
[[nodiscard]] int get_value_size() const;
[[nodiscard]] int get_max_entries() const;
private:
static std::vector<uint8_t> python_to_bytes(const py::object &obj, size_t size);
static py::object bytes_to_python(const std::vector<uint8_t> &data);
};

1
src/core/bpf_program.h
View File

@ -2,7 +2,6 @@
#define PYLIBBPF_BPF_PROGRAM_H
#include <libbpf.h>
#include <pybind11/stl.h>
#include <memory>
#include <string>