ISRI/gns3-wol-emulator
10
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

use smart pointer to avoid memory issue

Browse source
This commit is contained in:
faraphel 2024-12-23 11:09:33 +01:00
parent 9c413c0edd
commit a0060c8fa8
5 changed files with 21 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
source/gns3/GnsProject.cpp
View file

@ -23,8 +23,8 @@ std::string GnsProject::getUuid() const {
return this->uuid;
}
std::vector<GnsNode> GnsProject::getNodes() const {
std::vector<GnsNode> nodes;
std::vector<std::shared_ptr<GnsNode>> GnsProject::getNodes() const {
std::vector<std::shared_ptr<GnsNode>> nodes;
// request the API endpoint
cpr::Url endpoint = this->getApiBase() + "nodes";
@ -59,11 +59,11 @@ std::vector<GnsNode> GnsProject::getNodes() const {
}
// save the node
nodes.emplace_back(
nodes.emplace_back(std::make_shared<GnsNode>(
this,
node_data["node_id"],
ports
);
));
}
return nodes;

3
source/gns3/GnsProject.hpp
View file

@ -1,4 +1,5 @@
#pragma once
#include <memory>
#include <string>
#include <vector>
@ -34,7 +35,7 @@ public:
* Get all the nodes of the project
* @return all the nodes of the project
*/
[[nodiscard]] std::vector<GnsNode> getNodes() const;
[[nodiscard]] std::vector<std::shared_ptr<GnsNode>> getNodes() const;
private:
const GnsServer* server;

16
source/gns3/GnsServer.cpp
View file

@ -17,8 +17,8 @@ std::string GnsServer::getApiBase() const {
return "http://" + this->host + ":" + std::to_string(this->port) + "/v2/";
}
std::vector<GnsProject> GnsServer::getProjects() const {
std::vector<GnsProject> projects;
std::vector<std::shared_ptr<GnsProject>> GnsServer::getProjects() const {
std::vector<std::shared_ptr<GnsProject>> projects;
// request the API endpoint
cpr::Url endpoint = this->getApiBase() + "projects";
@ -38,20 +38,20 @@ std::vector<GnsProject> GnsServer::getProjects() const {
// deserialize the projects
projects.reserve(data.size());
for (const auto& project_data : data)
projects.emplace_back(
projects.emplace_back(std::make_shared<GnsProject>(
this,
project_data["project_id"]
);
));
return projects;
}
std::vector<GnsNode> GnsServer::getNodes() const {
std::vector<GnsNode> nodes;
std::vector<std::shared_ptr<GnsNode>> GnsServer::getNodes() const {
std::vector<std::shared_ptr<GnsNode>> nodes;
// get all the nodes from all the projects
for (const GnsProject& project : this->getProjects()) {
std::vector<GnsNode> project_nodes = project.getNodes();
for (const std::shared_ptr<GnsProject>& project : this->getProjects()) {
std::vector<std::shared_ptr<GnsNode>> project_nodes = project->getNodes();
nodes.insert(nodes.end(), project_nodes.begin(), project_nodes.end());
}

5
source/gns3/GnsServer.hpp
View file

@ -1,6 +1,7 @@
#pragma once
#include <string>
#include <cstdint>
#include <memory>
#include <vector>
#include "GnsProject.hpp"
@ -27,13 +28,13 @@ public:
* Get all the projects in the server
* @return all the projects in the server
*/
[[nodiscard]] std::vector<GnsProject> getProjects() const;
[[nodiscard]] std::vector<std::shared_ptr<GnsProject>> getProjects() const;
/**
* Get all the nodes in the server
* @return all the nodes in the server
*/
[[nodiscard]] std::vector<GnsNode> getNodes() const;
[[nodiscard]] std::vector<std::shared_ptr<GnsNode>> getNodes() const;
private:
std::string host;

8
source/main.cpp
View file

@ -50,11 +50,11 @@ void packet_wol_handler(
// TODO(Faraphel): check the 16 repetitions of the source address ?
// find the machine with the mac address
for (const gns::GnsNode& node : gnsServer->getNodes())
for (const gns::GnsPort& port : node.getPorts())
for (const std::shared_ptr<gns::GnsNode>& node : gnsServer->getNodes())
for (const gns::GnsPort& port : node->getPorts())
if (port.mac_address == mac_address_target) {
std::cout << "Matching node: " + node.getUuid() << std::endl;
node.start();
std::cout << "Matching node: " + node->getUuid() << std::endl;
node->start();
std::cout << "Node started." << std::endl;
return;
}