scottk/nDPId-2
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/nDPId-2.git synced 2025-10-29 17:32:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed some SonarCloud complaints.

Browse Source 
Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
This commit is contained in:
Toni Uhlig
2023-11-03 23:23:44 +01:00
parent 93498fff02
commit 84b12cd02c
4 changed files with 81 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@@ -1,8 +1,10 @@
FROM ubuntu:22.04 as builder
FROM ubuntu:22.04 AS builder
WORKDIR /root
RUN env DEBIAN_FRONTEND=noninteractive apt-get -y update && apt-get install -y --no-install-recommends autoconf automake build-essential ca-certificates wget unzip git make cmake pkg-config libpcap-dev autoconf libtool
RUN env DEBIAN_FRONTEND=noninteractive apt-get clean
RUN env DEBIAN_FRONTEND=noninteractive apt-get autoclean
RUN git clone https://github.com/utoni/nDPId.git
RUN cd nDPId && mkdir -p build && cd build && cmake .. -DBUILD_NDPI=ON && make
@@ -14,6 +16,8 @@ COPY --from=builder /root/nDPId/build/nDPId /usr/sbin/nDPId
COPY --from=builder /root/nDPId/build/nDPIsrvd /usr/bin/nDPIsrvd
RUN env DEBIAN_FRONTEND=noninteractive apt-get -y update && apt-get install -y --no-install-recommends libpcap-dev
RUN env DEBIAN_FRONTEND=noninteractive apt-get clean
RUN env DEBIAN_FRONTEND=noninteractive apt-get autoclean
USER nobody
RUN /usr/bin/nDPIsrvd -h || { RC=$?; test ${RC} -eq 1; }

14
examples/py-machine-learning/keras-autoencoder.py
View File

@@ -89,7 +89,7 @@ def get_autoencoder(load_from_file=None):
decoder_submodel = autoencoder.layers[2]
return encoder_submodel, decoder_submodel, autoencoder
def onJsonLineRecvd(json_dict, instance, current_flow, global_user_data):
def on_json_line(json_dict, instance, current_flow, global_user_data):
if 'packet_event_name' not in json_dict:
return True
@@ -138,13 +138,13 @@ def onJsonLineRecvd(json_dict, instance, current_flow, global_user_data):
return True
def nDPIsrvd_worker(address, shared_shutdown_event, shared_training_event, shared_packet_list):
def ndpisrvd_worker(address, shared_shutdown_event, shared_training_event, shared_packet_list):
nsock = nDPIsrvdSocket()
try:
nsock.connect(address)
print_dots = 1
nsock.loop(onJsonLineRecvd, None, (shared_shutdown_event, shared_training_event, shared_packet_list, print_dots))
nsock.loop(on_json_line, None, (shared_shutdown_event, shared_training_event, shared_packet_list, print_dots))
except nDPIsrvd.SocketConnectionBroken as err:
sys.stderr.write('\nnDPIsrvd-Worker Socket Error: {}\n'.format(err))
except KeyboardInterrupt:
@@ -158,11 +158,11 @@ def nDPIsrvd_worker(address, shared_shutdown_event, shared_training_event, share
def keras_worker(load_model, save_model, shared_shutdown_event, shared_training_event, shared_packet_queue, shared_plot_queue):
shared_training_event.set()
try:
encoder, decoder, autoencoder = get_autoencoder(load_model)
encoder, _, autoencoder = get_autoencoder(load_model)
except Exception as err:
sys.stderr.write('Could not load Keras model from file: {}\n'.format(str(err)))
sys.stderr.flush()
encoder, decoder, autoencoder = get_autoencoder()
encoder, _, autoencoder = get_autoencoder()
autoencoder.summary()
tensorboard = TensorBoard(log_dir=TB_LOGPATH, histogram_freq=1)
early_stopping = EarlyStopping(monitor='val_loss', min_delta=0.0001, patience=ES_PATIENCE, restore_best_weights=True, start_from_epoch=0, verbose=0, mode='auto')
@@ -213,7 +213,7 @@ def keras_worker(load_model, save_model, shared_shutdown_event, shared_training_
try:
shared_shutdown_event.set()
except:
except Exception:
pass
def plot_animate(i, shared_plot_queue, ax, xs, ys):
@@ -345,7 +345,7 @@ if __name__ == '__main__':
shared_packet_queue = mgr.JoinableQueue()
shared_plot_queue = mgr.JoinableQueue()
nDPIsrvd_job = mp.Process(target=nDPIsrvd_worker, args=(
nDPIsrvd_job = mp.Process(target=ndpisrvd_worker, args=(
address,
shared_shutdown_event,
shared_training_event,

26
nDPId-test.c
View File

@@ -168,13 +168,13 @@ static unsigned long long int nDPIsrvd_free_bytes = 0;
do \
{ \
((struct thread_return_value *)thread_arg)->val = (errno != 0 ? errno : 1); \
} while (0);
} while (0)
#define THREAD_ERROR_GOTO(thread_arg) \
do \
{ \
THREAD_ERROR(thread_arg); \
goto error; \
} while (0);
} while (0)
static void nDPIsrvd_memprof_log(char const * const format, ...)
{
@@ -384,10 +384,13 @@ static void * nDPIsrvd_mainloop_thread(void * const arg)
{
struct remote_desc * remote = (struct remote_desc *)nio_get_ptr(&io, i);
if (remote == mock_json_desc || remote == mock_test_desc ||
remote == mock_buff_desc || remote == mock_null_desc ||
remote == mock_arpa_desc)
if (remote != mock_json_desc && remote != mock_test_desc && remote != mock_buff_desc &&
remote != mock_null_desc && remote != mock_arpa_desc)
{
logger(1, "nDPIsrvd epoll returned unexpected event data: %p", remote);
THREAD_ERROR_GOTO(arg);
}
if (nio_has_error(&io, i) == NIO_SUCCESS)
{
char const * remote_desc_name;
@@ -455,13 +458,6 @@ static void * nDPIsrvd_mainloop_thread(void * const arg)
}
}
}
else
{
logger(1,
"nDPIsrvd epoll returned unexpected event data: %p", remote);
THREAD_ERROR_GOTO(arg);
}
}
}
error:
@@ -1186,8 +1182,7 @@ static void * distributor_client_mainloop_thread(void * const arg)
#endif
else
{
logger(1,
"Distributor epoll returned unexpected event data: %p", nio_get_ptr(&io, i));
logger(1, "Distributor epoll returned unexpected event data: %p", nio_get_ptr(&io, i));
THREAD_ERROR_GOTO(trv);
}
}
@@ -1621,8 +1616,7 @@ static int nio_selftest()
goto error;
}
if (nio_del_fd(&io, pipefds[0]) != NIO_SUCCESS
|| nio_del_fd(&io, pipefds[1]) != NIO_SUCCESS)
if (nio_del_fd(&io, pipefds[0]) != NIO_SUCCESS || nio_del_fd(&io, pipefds[1]) != NIO_SUCCESS)
{
logger(1, "%s", "Event delete failed");
goto error;

5
nDPId.c
View File

@@ -155,7 +155,7 @@ enum nDPId_flow_state
enum nDPId_flow_direction
{
FD_SRC2DST = 0,
FD_DST2SRC = 1,
FD_DST2SRC,
FD_COUNT
};
@@ -1066,7 +1066,8 @@ static void get_ip4_address_and_netmask(struct ifaddrs const * const ifaddr)
static int get_ip_netmask_from_pcap_dev(char const * const pcap_dev)
{
int retval = 0, found_dev = 0;
int ip4_interface_avail = 0, ip6_interface_avail = 0;
int ip4_interface_avail = 0;
int ip6_interface_avail = 0;
struct ifaddrs * ifaddrs = NULL;
struct ifaddrs * ifa;