/* * Ouroboros - Copyright (C) 2016 * * The sockets layer to communicate between daemons * * 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. */ #define OUROBOROS_PREFIX "libouroboros-sockets" #include <ouroboros/logs.h> #include <ouroboros/common.h> #include <ouroboros/sockets.h> #include <ouroboros/utils.h> #include <sys/socket.h> #include <sys/un.h> #include <sys/stat.h> #include <string.h> #include <malloc.h> int client_socket_open(char * file_name) { int sockfd; struct sockaddr_un serv_addr; sockfd = socket(AF_UNIX, SOCK_STREAM, 0); if (sockfd < 0) { LOG_ERR("Failed to open socket"); return -1; } serv_addr.sun_family = AF_UNIX; sprintf(serv_addr.sun_path, "%s", file_name); if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr))) { LOG_ERR("Failed to connect to daemon"); return -1; } return sockfd; } int server_socket_open(char * file_name) { int sockfd; struct sockaddr_un serv_addr; struct stat sb; if (!stat(file_name, &sb)) { /* File exists */ if (unlink(file_name)) { LOG_ERR("Failed to unlink filename: %s", strerror(errno)); return -1; } } sockfd = socket(AF_UNIX, SOCK_STREAM, 0); if (sockfd < 0) { LOG_ERR("Failed to open socket"); return -1; } serv_addr.sun_family = AF_UNIX; sprintf(serv_addr.sun_path, "%s", file_name); if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr))) { LOG_ERR("Failed to bind socket"); return -1; } if (listen(sockfd, 0)) { LOG_ERR("Failed to listen to socket"); return -1; } return sockfd; } char * ipcp_sock_path(pid_t pid) { char * full_name = NULL; char * pid_string = NULL; size_t len = 0; char * delim = "-"; len = n_digits(pid); pid_string = malloc(len + 1); if (pid_string == NULL) return NULL; sprintf(pid_string, "%d", pid); len += strlen(IPCP_SOCK_PATH_PREFIX); len += strlen(delim); full_name = malloc(len + 1); if (full_name == NULL) { free(pid_string); return NULL; } strcpy(full_name, IPCP_SOCK_PATH_PREFIX); strcat(full_name, delim); strcat(full_name, pid_string); free(pid_string); return full_name; } static int serialized_string_len(uint8_t * data) { uint8_t * seek = data; while (*seek != '\0') seek++; return (seek - data) + 1; } static void ser_copy_value(size_t flen, void * dst, void * src, int * offset) { memcpy(dst + *offset, src, flen); *offset += flen; } static void ser_copy_name(rina_name_t * name, uint8_t * data, int * offset) { ser_copy_value(strlen(name->ap_name) + 1, data, name->ap_name, offset); ser_copy_value(sizeof(int), data, &name->api_id, offset); ser_copy_value(strlen(name->ae_name) + 1, data, name->ae_name, offset); ser_copy_value(sizeof(int), data, &name->aei_id, offset); } static void deser_copy_value(size_t flen, void * dst, void * src, int * offset) { memcpy(dst, src + *offset, flen); *offset += flen; } static int deser_copy_string(uint8_t * data, char ** dst, int * offset) { size_t flen; flen = serialized_string_len(data + *offset); *dst = malloc(sizeof(**dst) * (flen + 1)); if (*dst == NULL) return -1; deser_copy_value(flen, *dst, data, offset); return 0; } static void deser_copy_int(uint8_t * data, unsigned int * dst, int * offset) { *dst = 0; deser_copy_value(sizeof(int), dst, data, offset); } static void deser_copy_size_t(uint8_t * data, size_t * dst, int * offset) { *dst = 0; deser_copy_value(sizeof(size_t), dst, data, offset); } static rina_name_t * deser_copy_name(uint8_t * data, int * offset) { rina_name_t * name; name = name_create(); if (name == NULL) { LOG_ERR("Failed to alloc memory"); return NULL; } if (deser_copy_string(data, &name->ap_name, offset)) { name_destroy(name); return NULL; } deser_copy_int(data, &name->api_id, offset); if (deser_copy_string(data, &name->ae_name, offset)) { name_destroy(name); return NULL; } deser_copy_int(data, &name->aei_id, offset); return name; } /* Move these to a separate file? */ static buffer_t * buffer_create() { buffer_t * buf; buf = malloc(sizeof(*buf)); if (buf == NULL) return NULL; buf->data = malloc(IRM_MSG_BUF_SIZE); if (buf->data == NULL) { free(buf); return NULL; } return buf; } static void buffer_destroy(buffer_t * buf) { if (buf->data != NULL) free(buf->data); if (buf != NULL) free(buf); } buffer_t * serialize_irm_msg(struct irm_msg * msg) { buffer_t * buf; uint8_t * data; int offset = 0; int i; char ** pos; if (msg == NULL) return NULL; buf = buffer_create(); if (buf == NULL) return NULL; data = buf->data; ser_copy_value(sizeof(enum irm_msg_code), data, &msg->code, &offset); if (!name_is_ok(msg->name)) { LOG_ERR("Null pointer passed"); buffer_destroy(buf); return NULL; } ser_copy_name(msg->name, data, &offset); switch (msg->code) { case IRM_CREATE_IPCP: if (!msg->ipcp_type) { LOG_ERR("Null pointer passed"); buffer_destroy(buf); return NULL; } ser_copy_value(strlen(msg->ipcp_type) + 1, data, msg->ipcp_type, &offset); break; case IRM_DESTROY_IPCP: break; case IRM_BOOTSTRAP_IPCP: /* FIXME: Fields missing, need to define dif_conf properly */ break; case IRM_ENROLL_IPCP: if (msg->dif_name == NULL) { buffer_destroy(buf); return NULL; } ser_copy_value(strlen(msg->dif_name) + 1, data, msg->dif_name, &offset); break; case IRM_REG_IPCP: case IRM_UNREG_IPCP: if (msg->difs == NULL || msg->difs[0] == NULL) { buffer_destroy(buf); return NULL; } ser_copy_value(sizeof(size_t), data, &msg->difs_size, &offset); pos = msg->difs; for (i = 0; i < msg->difs_size; i++) { ser_copy_value(strlen(*pos) + 1, data, *pos, &offset); pos++; } break; default: LOG_ERR("Don't know that code"); buffer_destroy(buf); return NULL; } buf->size = offset; return buf; } struct irm_msg * deserialize_irm_msg(buffer_t * data) { struct irm_msg * msg; int i, j; int offset = 0; size_t difs_size; if (data == NULL || data->data == NULL) { LOG_ERR("Got a null pointer"); return NULL; } msg = malloc(sizeof(*msg)); if (msg == NULL) { LOG_ERR("Failed to allocate memory"); return NULL; } deser_copy_value(sizeof(enum irm_msg_code), &msg->code, data->data, &offset); msg->name = deser_copy_name(data->data, &offset); if (msg->name == NULL) { LOG_ERR("Failed to reconstruct name"); free(msg); return NULL; } switch (msg->code) { case IRM_CREATE_IPCP: if (deser_copy_string(data->data, &msg->ipcp_type, &offset)) { name_destroy(msg->name); free(msg); return NULL; } break; case IRM_DESTROY_IPCP: break; case IRM_BOOTSTRAP_IPCP: break; case IRM_ENROLL_IPCP: if (deser_copy_string(data->data, &msg->dif_name, &offset)) { name_destroy(msg->name); free(msg); return NULL; } break; case IRM_REG_IPCP: case IRM_UNREG_IPCP: deser_copy_size_t(data->data, &difs_size, &offset); msg->difs_size = difs_size; msg->difs = malloc(sizeof(*(msg->difs)) * difs_size); if (msg->difs == NULL) { name_destroy(msg->name); free(msg); return NULL; } for (i = 0; i < difs_size; i++) { if (deser_copy_string(data->data, &msg->difs[i], &offset)) { for (j = 0; j < i; j++) free(msg->difs[j]); free(msg->difs); name_destroy(msg->name); free(msg); return NULL; } } break; default: LOG_ERR("Don't know that code"); free(msg); return NULL; } return msg; } buffer_t * serialize_ipcp_msg(struct ipcp_msg * msg) { buffer_t * buf = NULL; uint8_t * data = NULL; int offset = 0; char ** pos = NULL; int i = 0; if (msg == NULL) return NULL; buf = buffer_create(); if (buf == NULL) return NULL; data = buf->data; ser_copy_value(sizeof(enum ipcp_msg_code), data, &msg->code, &offset); switch (msg->code) { case IPCP_BOOTSTRAP: break; case IPCP_ENROLL: if (msg->dif_name == NULL) { buffer_destroy(buf); return NULL; } ser_copy_value(strlen(msg->dif_name) + 1, data, msg->dif_name, &offset); if (!name_is_ok(msg->member)) { LOG_ERR("Null pointer passed"); buffer_destroy(buf); return NULL; } ser_copy_name(msg->member, data, &offset); /* All these operations end with a list of DIFs */ case IPCP_REG: case IPCP_UNREG: if (msg->difs == NULL || msg->difs[0] == NULL) { buffer_destroy(buf); return NULL; } ser_copy_value(sizeof(size_t), data, &msg->difs_size, &offset); pos = msg->difs; for (i = 0; i < msg->difs_size; i++) { ser_copy_value(strlen(*pos) + 1, data, *pos, &offset); pos++; } break; default: LOG_ERR("Don't know that code"); buffer_destroy(buf); return NULL; } buf->size = offset; return buf; } struct ipcp_msg * deserialize_ipcp_msg(buffer_t * data) { struct ipcp_msg * msg; int i, j; int offset = 0; size_t difs_size; if (data == NULL || data->data == NULL) { LOG_ERR("Got a null pointer"); return NULL; } msg = malloc(sizeof(*msg)); if (msg == NULL) { LOG_ERR("Failed to allocate memory"); return NULL; } deser_copy_value(sizeof(enum ipcp_msg_code), &msg->code, data->data, &offset); switch (msg->code) { case IPCP_BOOTSTRAP: break; case IPCP_ENROLL: if (deser_copy_string(data->data, &msg->dif_name, &offset)) { free(msg); return NULL; } msg->member = deser_copy_name(data->data, &offset); if (msg->member == NULL) { LOG_ERR("Failed to reconstruct name"); free(msg->dif_name); free(msg); return NULL; } deser_copy_size_t(data->data, &difs_size, &offset); msg->difs_size = difs_size; msg->difs = malloc(sizeof(*(msg->difs)) * difs_size); if (msg->difs == NULL) { free(msg->dif_name); name_destroy(msg->member); free(msg); return NULL; } for (i = 0; i < difs_size; i++) { if (deser_copy_string(data->data, &msg->difs[i], &offset)) { for (j = 0; j < i; j++) free(msg->difs[j]); free(msg->dif_name); free(msg->difs); name_destroy(msg->member); free(msg); return NULL; } } break; case IPCP_REG: case IPCP_UNREG: deser_copy_size_t(data->data, &difs_size, &offset); msg->difs_size = difs_size; msg->difs = malloc(sizeof(*(msg->difs)) * difs_size); if (msg->difs == NULL) { free(msg); return NULL; } for (i = 0; i < difs_size; i++) { if (deser_copy_string(data->data, &msg->difs[i], &offset)) { for (j = 0; j < i; j++) free(msg->difs[j]); free(msg->difs); free(msg); return NULL; } } break; default: LOG_ERR("Don't know that code"); free(msg); return NULL; } return msg; }