hot_update.h

Go to the documentation of this file.

#ifndef HOT_UPDATE_H
#define HOT_UPDATE_H
 
#include <fstream>
#include <iostream>
#include <string>
 
#include "../json.hpp"
#include "../base64.h"
 
using json = nlohmann::json;
 
struct ebpf_update_meta_data
{
  std::string name;
  std::vector<std::string> maps_name;
  std::vector<std::string> progs_name;
  int data_sz;
  std::string data;
  std::vector<char> base64_decode_buffer;
 
  std::string to_json()
  {
    json j;
    j["name"] = name;
    j["maps"] = maps_name;
    j["progs"] = progs_name;
    j["data_sz"] = data_sz;
    j["data"] = data;
    return j.dump();
  }
  void from_json_str(const std::string& j_str)
  {
    json jj = json::parse(j_str);
    name = jj["name"];
    maps_name = jj["maps"];
    progs_name = jj["progs"];
    data_sz = jj["data_sz"];
    data = jj["data"];
  }
};
 
static std::string bpf_skeleton_encode(const struct bpf_object_skeleton *skeleton) {
  struct ebpf_update_meta_data data;
 
  data.name ==skeleton->name;
  data.data_sz = skeleton->data_sz;
  data.data = base64_encode((const unsigned char *)skeleton->data, data.data_sz);
  for (int i = 0; i < skeleton->map_cnt; i++)
  {
    data.maps_name.push_back(skeleton->maps[i].name);
  }
  for (int i = 0; i < skeleton->prog_cnt; i++)
  {
    data.progs_name.push_back(skeleton->progs[i].name);
  }
  return data.to_json();
}
 
#endif