blob: e3bbfe31f31c11eccba9d0efce555463f31ae8a6 (
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
|
// SPDX-License-Identifier: GPL-3.0-or-later
#include "proto/agent/v1/connection.pb.h"
#include "proto/agent/v1/disconnect.pb.h"
#include "connection.h"
#include "schema_wrapper_utils.h"
#include <sys/time.h>
#include <stdlib.h>
using namespace agent::v1;
char *generate_update_agent_connection(size_t *len, const update_agent_connection_t *data)
{
UpdateAgentConnection connupd;
connupd.set_claim_id(data->claim_id);
connupd.set_reachable(data->reachable);
connupd.set_session_id(data->session_id);
connupd.set_update_source((data->lwt) ? CONNECTION_UPDATE_SOURCE_LWT : CONNECTION_UPDATE_SOURCE_AGENT);
struct timeval tv;
gettimeofday(&tv, NULL);
google::protobuf::Timestamp *timestamp = connupd.mutable_updated_at();
timestamp->set_seconds(tv.tv_sec);
timestamp->set_nanos(tv.tv_usec * 1000);
*len = PROTO_COMPAT_MSG_SIZE(connupd);
char *msg = (char*)malloc(*len);
if (msg)
connupd.SerializeToArray(msg, *len);
return msg;
}
struct disconnect_cmd *parse_disconnect_cmd(const char *data, size_t len) {
DisconnectReq req;
struct disconnect_cmd *res;
if (!req.ParseFromArray(data, len))
return NULL;
res = (struct disconnect_cmd *)calloc(1, sizeof(struct disconnect_cmd));
if (!res)
return NULL;
res->reconnect_after_s = req.reconnect_after_seconds();
res->permaban = req.permaban();
res->error_code = req.error_code();
if (req.error_description().c_str()) {
res->error_description = strdup(req.error_description().c_str());
if (!res->error_description) {
free(res);
return NULL;
}
}
return res;
}
|
