From d51409221d5c7171cf3fd02cbc7625140ef7b089 Mon Sep 17 00:00:00 2001
From: "alexey.stratulat" <alexey.stratulat@demlabs.net>
Date: Thu, 19 Dec 2019 21:05:55 +0700
Subject: [PATCH] [+] Added security guard
---
include/dap_server.h | 147 +++++++++++++++++++++++++++++++++++++++++++
1 file changed, 147 insertions(+)
create mode 100755 include/dap_server.h
diff --git a/include/dap_server.h b/include/dap_server.h
new file mode 100755
index 00000000..a226961b
--- /dev/null
+++ b/include/dap_server.h
@@ -0,0 +1,147 @@
+/*
+ Copyright (c) 2017-2018 (c) Project "DeM Labs Inc" https://github.com/demlabsinc
+ All rights reserved.
+
+ This file is part of DAP (Deus Applications Prototypes) the open source project
+
+ DAP (Deus Applicaions Prototypes) is free software: you can redistribute it and/or modify
+ it under the terms of the GNU Lesser General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ DAP is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public License
+ along with any DAP based project. If not, see <http://www.gnu.org/licenses/>.
+*/
+#pragma once
+
+#ifndef _WIN32
+#include <netinet/in.h>
+#include <stdint.h>
+#include <sys/epoll.h>
+#include <sys/timerfd.h>
+#define EPOLL_HANDLE int
+#else
+#define EPOLL_HANDLE HANDLE
+#define MSG_DONTWAIT 0
+#define MSG_NOSIGNAL 0
+#include "wepoll.h"
+#endif
+
+#include <pthread.h>
+#include "uthash.h"
+#include "utlist.h"
+
+#include "dap_cpu_monitor.h"
+#include "dap_client_remote.h"
+
+typedef enum dap_server_type {DAP_SERVER_TCP} dap_server_type_t;
+
+#define BIT( x ) ( 1 << x )
+
+#define DAP_SOCK_READY_TO_READ BIT( 0 )
+#define DAP_SOCK_READY_TO_WRITE BIT( 1 )
+#define DAP_SOCK_SIGNAL_CLOSE BIT( 2 )
+#define DAP_SOCK_ACTIVE BIT( 3 )
+
+typedef struct dap_server_thread_s {
+
+ EPOLL_HANDLE epoll_fd;
+
+ uint32_t thread_num;
+ uint32_t connections_count;
+ uint32_t to_kill_count;
+
+ struct epoll_event *epoll_events;
+ dap_client_remote_t *dap_remote_clients;
+ dap_client_remote_t *hclients; // Hashmap of clients
+ dap_client_remote_t *dap_clients_to_kill;
+
+ pthread_mutex_t mutex_dlist_add_remove;
+ pthread_mutex_t mutex_on_hash;
+
+} dap_server_thread_t;
+
+struct dap_server;
+
+typedef void (*dap_server_callback_t)( struct dap_server *,void * arg ); // Callback for specific server's operations
+
+typedef struct dap_server {
+
+ dap_server_type_t type; // Server's type
+ uint16_t port; // Listen port
+ char *address; // Listen address
+
+ int32_t socket_listener; // Socket for listener
+ EPOLL_HANDLE epoll_fd; // Epoll fd
+
+ struct sockaddr_in listener_addr; // Kernel structure for listener's binded address
+
+ void *_inheritor; // Pointer to the internal data, HTTP for example
+
+ dap_cpu_stats_t cpu_stats;
+
+ dap_server_callback_t server_delete_callback;
+
+ dap_server_client_callback_t client_new_callback; // Create new client callback
+ dap_server_client_callback_t client_delete_callback; // Delete client callback
+ dap_server_client_callback_t client_read_callback; // Read function
+ dap_server_client_callback_t client_write_callback; // Write function
+ dap_server_client_callback_t client_error_callback; // Error processing function
+
+} dap_server_t;
+
+int32_t dap_server_init( uint32_t count_threads ); // Init server module
+void dap_server_deinit( void ); // Deinit server module
+
+dap_server_t *dap_server_listen( const char *addr, uint16_t port, dap_server_type_t type );
+
+int32_t dap_server_loop( dap_server_t *d_server );
+
+#define DL_LIST_REMOVE_NODE( head, obj, _prev_, _next_, total ) \
+ \
+ if ( obj->_next_ ) { \
+ \
+ if ( obj->_prev_ ) \
+ obj->_next_->_prev_ = obj->_prev_; \
+ else { \
+ \
+ obj->_next_->_prev_ = NULL; \
+ head = obj->_next_; \
+ } \
+ } \
+ \
+ if ( obj->_prev_ ) { \
+ \
+ if ( obj->_next_ ) \
+ obj->_prev_->_next_ = obj->_next_; \
+ else { \
+ \
+ obj->_prev_->_next_ = NULL; \
+ } \
+ } \
+ -- total;
+
+#define DL_LIST_ADD_NODE_HEAD( head, obj, _prev_, _next_, total )\
+ \
+ if ( !total ) { \
+ \
+ obj->_prev_ = NULL; \
+ obj->_next_ = NULL; \
+ \
+ head = obj; \
+ } \
+ else { \
+ \
+ head->_prev_ = obj; \
+ \
+ obj->_prev_ = NULL; \
+ obj->_next_ = head; \
+ \
+ head = obj; \
+ } \
+ ++ total;
--
GitLab