2019-04-24 14:23:41 +00:00
|
|
|
#pragma once
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include "evt_core.h"
|
|
|
|
#include "algo_utils.h"
|
2019-05-27 15:32:00 +00:00
|
|
|
#include "capture_traffic.h"
|
2019-04-24 14:23:41 +00:00
|
|
|
#include "url.h"
|
|
|
|
#include "utils.h"
|
|
|
|
#include "packet.h"
|
|
|
|
|
2019-05-27 15:32:00 +00:00
|
|
|
struct algo_params {
|
|
|
|
uint8_t is_waiting_bootstrap;
|
|
|
|
uint8_t is_healing;
|
|
|
|
char *algo_name, *capture_file;
|
|
|
|
int links, fresh_data, redundant_data;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct algo_ctx;
|
|
|
|
typedef void (*algo_init)(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
|
|
|
|
typedef int (*algo_ctx_on_buffer)(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
typedef int (*algo_ctx_on_event)(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
|
|
|
|
typedef void (*algo_ctx_free_misc)(void*);
|
|
|
|
|
|
|
|
struct algo_desc {
|
|
|
|
char* name;
|
|
|
|
algo_init init;
|
|
|
|
algo_ctx_on_buffer on_stream;
|
|
|
|
algo_ctx_on_buffer on_datagram;
|
|
|
|
algo_ctx_on_event on_err;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct algo_ctx {
|
|
|
|
struct algo_desc* desc;
|
|
|
|
uint8_t link_count;
|
|
|
|
uint8_t is_rdy;
|
|
|
|
struct algo_params ap;
|
|
|
|
int ref_count;
|
|
|
|
struct capture_ctx cap;
|
|
|
|
struct buffer_resources br;
|
|
|
|
void* misc; // Additional structures
|
|
|
|
algo_ctx_free_misc free_misc; // Fx ptr to free misc
|
|
|
|
};
|
|
|
|
|
2019-04-24 14:23:41 +00:00
|
|
|
void algo_naive_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
|
|
|
|
int algo_naive_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_naive_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_naive_on_err(struct evt_core_ctx *ctx, struct evt_core_fdinfo *fdinfo);
|
|
|
|
|
|
|
|
void algo_rr_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
|
|
|
|
int algo_rr_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_rr_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_rr_on_err(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
|
2019-05-06 12:23:14 +00:00
|
|
|
void algo_dup2_init(struct evt_core_ctx* ctx, struct algo_ctx* app_ctx, struct algo_params* ap);
|
|
|
|
int algo_dup2_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_dup2_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp);
|
|
|
|
int algo_dup2_on_err(struct evt_core_ctx *ctx, struct evt_core_fdinfo *fdinfo);
|
|
|
|
|
2019-04-24 14:23:41 +00:00
|
|
|
static struct algo_desc available_algo[] = {
|
|
|
|
{
|
|
|
|
.name = "naive",
|
|
|
|
.init = algo_naive_init,
|
|
|
|
.on_stream = algo_naive_on_stream,
|
|
|
|
.on_datagram = algo_naive_on_datagram,
|
|
|
|
.on_err = algo_naive_on_err
|
|
|
|
},
|
|
|
|
{
|
|
|
|
.name = "rr",
|
|
|
|
.init = algo_rr_init,
|
|
|
|
.on_stream = algo_rr_on_stream,
|
|
|
|
.on_datagram = algo_rr_on_datagram,
|
|
|
|
.on_err = algo_rr_on_err
|
|
|
|
},
|
|
|
|
{
|
|
|
|
.name = "dup2",
|
2019-05-06 12:23:14 +00:00
|
|
|
.init = algo_dup2_init,
|
|
|
|
.on_stream = algo_dup2_on_stream,
|
|
|
|
.on_datagram = algo_dup2_on_datagram,
|
|
|
|
.on_err = algo_dup2_on_err
|
2019-04-24 14:23:41 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
void algo_main_init(struct evt_core_ctx* evt, struct algo_params* ap);
|
|
|
|
|
|
|
|
int main_on_tcp_co(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_tcp_read(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_udp_read(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_tcp_write(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_udp_write (struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_timer(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|
|
|
|
int main_on_err(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo);
|