diff options
Diffstat (limited to 'src/lib')
-rw-r--r-- | src/lib/CMakeLists.txt | 7 | ||||
-rw-r--r-- | src/lib/cdap.c | 430 | ||||
-rw-r--r-- | src/lib/cdap.proto | 18 | ||||
-rw-r--r-- | src/lib/ipcp.c | 8 | ||||
-rw-r--r-- | src/lib/irm.c | 1 | ||||
-rw-r--r-- | src/lib/shm_du_map.c | 1 | ||||
-rw-r--r-- | src/lib/sockets.c | 7 |
7 files changed, 464 insertions, 8 deletions
diff --git a/src/lib/CMakeLists.txt b/src/lib/CMakeLists.txt index ae46f5bc..57f44f15 100644 --- a/src/lib/CMakeLists.txt +++ b/src/lib/CMakeLists.txt @@ -9,7 +9,9 @@ include_directories(${PROTOBUF_INCLUDE_DIRS}) protobuf_generate_c(IRM_PROTO_SRCS IRM_PROTO_HDRS irmd_messages.proto) protobuf_generate_c(IPCP_PROTO_SRCS IPCP_PROTO_HDRS ipcpd_messages.proto) protobuf_generate_c(DIF_CONFIG_PROTO_SRCS DIF_CONFIG_PROTO_HDRS - dif_config.proto) + dif_config.proto) +protobuf_generate_c(CDAP_PROTO_SRCS CDAP_PROTO_HDRS + cdap.proto) find_library(LIBRT_LIBRARIES rt) if(NOT LIBRT_LIBRARIES) @@ -38,7 +40,8 @@ set(SOURCE_FILES ) add_library(ouroboros SHARED ${SOURCE_FILES} - ${IRM_PROTO_SRCS} ${IPCP_PROTO_SRCS} ${DIF_CONFIG_PROTO_SRCS}) + ${IRM_PROTO_SRCS} ${IPCP_PROTO_SRCS} + ${DIF_CONFIG_PROTO_SRCS} ${CDAP_PROTO_SRCS}) target_link_libraries(ouroboros ${LIBRT_LIBRARIES} ${LIBPTHREAD_LIBRARIES} ${PROTOBUF_C_LIBRARY}) diff --git a/src/lib/cdap.c b/src/lib/cdap.c index e69de29b..8967c8bd 100644 --- a/src/lib/cdap.c +++ b/src/lib/cdap.c @@ -0,0 +1,430 @@ +/* + * Ouroboros - Copyright (C) 2016 + * + * The Common Distributed Application Protocol + * + * Sander Vrijders <sander.vrijders@intec.ugent.be> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + */ + +#include <ouroboros/cdap.h> +#include <ouroboros/bitmap.h> +#include <ouroboros/common.h> +#include <ouroboros/dev.h> + +#include <stdlib.h> +#include <pthread.h> + +#include "cdap.pb-c.h" +typedef Cdap cdap_t; +typedef Opcode opcode_t; + +#define IDS_SIZE 256 +#define BUF_SIZE 2048 + +struct cdap { + int fd; + struct bmp * ids; + pthread_mutex_t ids_lock; + pthread_t reader; + struct cdap_ops * ops; +}; + + +static ssize_t cdap_msg_to_buffer(cdap_t * msg, + buffer_t ** val) +{ + int i; + size_t len; + + len = msg->n_value; + + *val = malloc(len * sizeof(**val)); + if (*val == NULL) { + return -1; + } + + for (i = 0; i < len; i++) { + if (msg->value[i].data == NULL) { + free(*val); + return -1; + } + + (*val)[i].data = msg->value[i].data; + (*val)[i].len = msg->value[i].len; + } + + return len; +} + + +static void * sdu_reader(void * o) +{ + struct cdap * instance = (struct cdap *) o; + cdap_t * msg; + uint8_t buf[BUF_SIZE]; + size_t len; + ssize_t length; + buffer_t * val; + + while (true) { + len = flow_read(instance->fd, buf, BUF_SIZE); + if (len < 0) { + return (void *) -1; + } + + msg = cdap__unpack(NULL, len, buf); + if (msg == NULL) { + continue; + } + + switch (msg->opcode) { + case OPCODE__READ: + if (msg->name != NULL) + instance->ops->cdap_read(instance, + msg->name); + break; + case OPCODE__WRITE: + length = cdap_msg_to_buffer(msg, &val); + if (msg->name != NULL && + msg->value != NULL && + len > 0) { + instance->ops->cdap_write(instance, + msg->name, + val, + length, + msg->flags); + free(val); + } + break; + case OPCODE__CREATE: + length = cdap_msg_to_buffer(msg, &val); + if (msg->name != NULL && + length == 1) { + instance->ops->cdap_create(instance, + msg->name, + val[0]); + free(val); + } + break; + case OPCODE__DELETE: + length = cdap_msg_to_buffer(msg, &val); + if (msg->name != NULL && + length == 1) { + instance->ops->cdap_create(instance, + msg->name, + val[0]); + free(val); + } + break; + case OPCODE__START: + if (msg->name != NULL) + instance->ops->cdap_start(instance, + msg->name); + break; + case OPCODE__STOP: + if (msg->name != NULL) + instance->ops->cdap_stop(instance, + msg->name); + break; + case OPCODE__REPLY: + length = cdap_msg_to_buffer(msg, &val); + if (msg->name != NULL && + length > 0) { + instance->ops->cdap_reply(instance, + msg->invoke_id, + msg->result, + val, + length); + free(val); + } + break; + default: + break; + } + + cdap__free_unpacked(msg, NULL); + } + + return (void *) 0; +} + +struct cdap * cdap_create(struct cdap_ops * ops, + int fd) +{ + struct cdap * instance = NULL; + + if (ops == NULL || fd < 0 || + ops->cdap_reply == NULL || + ops->cdap_read == NULL || + ops->cdap_write == NULL || + ops->cdap_create == NULL || + ops->cdap_delete == NULL || + ops->cdap_start == NULL || + ops->cdap_stop == NULL) + return NULL; + + instance = malloc(sizeof(*instance)); + if (instance == NULL) + return NULL; + + if (pthread_mutex_init(&instance->ids_lock, NULL)) { + free(instance); + return NULL; + } + + instance->ops = ops; + + instance->ids = bmp_create(IDS_SIZE, 0); + if (instance->ids == NULL) { + free(instance); + return NULL; + } + + pthread_create(&instance->reader, + NULL, + sdu_reader, + (void *) instance); + + return instance; +} + +int cdap_destroy(struct cdap * instance) +{ + if (instance == NULL) + return -1; + + pthread_cancel(instance->reader); + + pthread_mutex_lock(&instance->ids_lock); + + bmp_destroy(instance->ids); + + pthread_mutex_unlock(&instance->ids_lock); + + pthread_join(instance->reader, + NULL); + + free(instance); + + return 0; +} + +static int next_invoke_id(struct cdap * instance) +{ + int ret; + + pthread_mutex_lock(&instance->ids_lock); + ret = bmp_allocate(instance->ids); + pthread_mutex_unlock(&instance->ids_lock); + + return ret; +} + +static int release_invoke_id(struct cdap * instance, + int id) +{ + int ret; + + pthread_mutex_lock(&instance->ids_lock); + ret = bmp_release(instance->ids, id); + pthread_mutex_unlock(&instance->ids_lock); + + return ret; +} + +static int write_msg(struct cdap * instance, + cdap_t * msg) +{ + buffer_t buf; + + buf.len = cdap__get_packed_size(msg); + if (buf.len == 0) { + return -1; + } + + cdap__pack(msg, buf.data); + + return flow_write(instance->fd, buf.data, buf.len); +} + +static int buffer_to_cdap_msg(cdap_t * msg, + buffer_t * val, + size_t len) +{ + int i; + + msg->value = malloc(len * sizeof(*msg->value)); + if (msg->value == NULL) { + return -1; + } + + msg->n_value = len; + for (i = 0; i < len; i++) { + if (val[i].data == NULL) { + free(msg->value); + return -1; + } + + msg->value[i].data = val[i].data; + msg->value[i].len = val[i].len; + } + + return 0; +} + +static int send_read_or_start_or_stop(struct cdap * instance, + char * name, + opcode_t code) +{ + int id; + cdap_t msg = CDAP__INIT; + + if (instance == NULL || name == NULL) + return -1; + + id = next_invoke_id(instance); + if (!bmp_is_id_valid(instance->ids, id)) + return -1; + + msg.opcode = code; + msg.invoke_id = id; + msg.name = name; + + return write_msg(instance, &msg); +} + +static int send_create_or_delete(struct cdap * instance, + char * name, + buffer_t val, + opcode_t code) +{ + int id; + cdap_t msg = CDAP__INIT; + int ret; + + if (instance == NULL || name == NULL) + return -1; + + id = next_invoke_id(instance); + if (!bmp_is_id_valid(instance->ids, id)) + return -1; + + msg.opcode = code; + msg.name = name; + msg.invoke_id = id; + + if (buffer_to_cdap_msg(&msg, &val, 1)) { + release_invoke_id(instance, id); + return -1; + } + + ret = write_msg(instance, &msg); + + free(msg.value); + + return ret; +} + +int cdap_send_read(struct cdap * instance, + char * name) +{ + return send_read_or_start_or_stop(instance, name, OPCODE__READ); +} + +int cdap_send_write(struct cdap * instance, + char * name, + buffer_t * val, + size_t len, + uint32_t flags) +{ + int id; + int ret; + cdap_t msg = CDAP__INIT; + + if (instance == NULL || name == NULL || + val == NULL || len < 1) + return -1; + + id = next_invoke_id(instance); + if (!bmp_is_id_valid(instance->ids, id)) + return -1; + + msg.opcode = OPCODE__WRITE; + msg.name = name; + msg.has_flags = true; + msg.flags = flags; + msg.invoke_id = id; + + if (buffer_to_cdap_msg(&msg, val, len)) { + release_invoke_id(instance, id); + return -1; + } + + ret = write_msg(instance, &msg); + + free(msg.value); + + return ret; +} + +int cdap_send_create(struct cdap * instance, + char * name, + buffer_t val) +{ + return send_create_or_delete(instance, name, val, OPCODE__CREATE); +} + +int cdap_send_delete(struct cdap * instance, + char * name, + buffer_t val) +{ + return send_create_or_delete(instance, name, val, OPCODE__DELETE); +} + +int cdap_send_start(struct cdap * instance, + char * name) +{ + return send_read_or_start_or_stop(instance, name, OPCODE__START); +} + +int cdap_send_stop(struct cdap * instance, + char * name) +{ + return send_read_or_start_or_stop(instance, name, OPCODE__STOP); +} + +int cdap_send_reply(struct cdap * instance, + int invoke_id, + int result, + buffer_t * val, + size_t len) +{ + cdap_t msg = CDAP__INIT; + + if (instance == NULL || val == NULL) + return -1; + + msg.invoke_id = invoke_id; + msg.has_result = true; + msg.result = result; + + if (buffer_to_cdap_msg(&msg, val, len)) { + return -1; + } + + return write_msg(instance, &msg); +} diff --git a/src/lib/cdap.proto b/src/lib/cdap.proto new file mode 100644 index 00000000..a5e0306d --- /dev/null +++ b/src/lib/cdap.proto @@ -0,0 +1,18 @@ +enum opcode { + CREATE = 1; + DELETE = 2; + READ = 3; + WRITE = 4; + START = 5; + STOP = 6; + REPLY = 7; +} + +message cdap { + required opcode opcode = 1; + required uint32 invoke_id = 2; + optional uint32 flags = 3; + optional string name = 4; + repeated bytes value = 5; + optional int32 result = 6; +} diff --git a/src/lib/ipcp.c b/src/lib/ipcp.c index 2b6b6825..b336155e 100644 --- a/src/lib/ipcp.c +++ b/src/lib/ipcp.c @@ -32,6 +32,8 @@ #include <stdlib.h> #include <string.h> #include <signal.h> +#include <errno.h> +#include <stdbool.h> #include <sys/types.h> #include <sys/wait.h> @@ -54,8 +56,8 @@ static ipcp_msg_t * send_recv_ipcp_msg(pid_t api, return NULL; } - buf.size = ipcp_msg__get_packed_size(msg); - if (buf.size == 0) { + buf.len = ipcp_msg__get_packed_size(msg); + if (buf.len == 0) { close(sockfd); free(sock_path); return NULL; @@ -70,7 +72,7 @@ static ipcp_msg_t * send_recv_ipcp_msg(pid_t api, ipcp_msg__pack(msg, buf.data); - if (write(sockfd, buf.data, buf.size) == -1) { + if (write(sockfd, buf.data, buf.len) == -1) { free(sock_path); free(buf.data); close(sockfd); diff --git a/src/lib/irm.c b/src/lib/irm.c index 8f475e38..ca2111ca 100644 --- a/src/lib/irm.c +++ b/src/lib/irm.c @@ -29,6 +29,7 @@ #include <ouroboros/logs.h> #include <ouroboros/sockets.h> +#include <stdbool.h> #include <string.h> #include <stdlib.h> #include <sys/stat.h> diff --git a/src/lib/shm_du_map.c b/src/lib/shm_du_map.c index 31fcca8e..cf0bad19 100644 --- a/src/lib/shm_du_map.c +++ b/src/lib/shm_du_map.c @@ -25,6 +25,7 @@ #include <ouroboros/shm_du_map.h> #include <ouroboros/shm_ap_rbuff.h> #include <ouroboros/time_utils.h> + #include <pthread.h> #include <sys/mman.h> #include <fcntl.h> diff --git a/src/lib/sockets.c b/src/lib/sockets.c index 9bfbad5e..d60ea91a 100644 --- a/src/lib/sockets.c +++ b/src/lib/sockets.c @@ -32,6 +32,7 @@ #include <sys/un.h> #include <string.h> #include <stdlib.h> +#include <errno.h> int client_socket_open(char * file_name) { @@ -109,8 +110,8 @@ irm_msg_t * send_recv_irm_msg(irm_msg_t * msg) if (sockfd < 0) return NULL; - buf.size = irm_msg__get_packed_size(msg); - if (buf.size == 0) { + buf.len = irm_msg__get_packed_size(msg); + if (buf.len == 0) { close(sockfd); return NULL; } @@ -123,7 +124,7 @@ irm_msg_t * send_recv_irm_msg(irm_msg_t * msg) irm_msg__pack(msg, buf.data); - if (write(sockfd, buf.data, buf.size) == -1) { + if (write(sockfd, buf.data, buf.len) == -1) { free(buf.data); close(sockfd); return NULL; |