summaryrefslogtreecommitdiff
path: root/src/lib/cep.c
blob: ba2380234ffc6713b310a37e0d94d5c81dee9464 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
 * Ouroboros - Copyright (C) 2016 - 2024
 *
 * The Ouroboros Connection Establishment Protocol
 *
 *    Dimitri Staessens <dimitri@ouroboros.rocks>
 *    Sander Vrijders   <sander@ouroboros.rocks>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * version 2.1 as published by the Free Software Foundation.
 *
 * This library 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 this library; if not, write to the Free Software
 * Foundation, Inc., http://www.fsf.org/about/contact/.
 */

#define _POSIX_C_SOURCE 199309L

#include <ouroboros/cep.h>
#include <ouroboros/dev.h>
#include <ouroboros/errno.h>

#include <stdlib.h>
#include <string.h>

#include "cep.pb-c.h"
typedef CepMsg cep_msg_t;

#define BUF_SIZE 128

static int read_msg(int                fd,
                    struct conn_info * info)
{
        uint8_t     buf[BUF_SIZE];
        cep_msg_t * msg;
        ssize_t     len;

        len = flow_read(fd, buf, BUF_SIZE);
        if (len < 0)
                return (int) len;

        msg = cep_msg__unpack(NULL, len, buf);
        if (msg == NULL)
                return -1;

        if (strlen(msg->comp_name) > OCEP_BUF_STRLEN) {
                cep_msg__free_unpacked(msg, NULL);
                return -1;
        }

        strcpy(info->comp_name, msg->comp_name);
        strcpy(info->protocol, msg->protocol);

        info->pref_version = msg->pref_version;
        info->pref_syntax  = msg->pref_syntax;
        info->addr         = msg->address;

        cep_msg__free_unpacked(msg, NULL);

        return 0;
}

static int send_msg(int                      fd,
                    const struct conn_info * info)
{
        cep_msg_t msg = CEP_MSG__INIT;
        uint8_t * data = NULL;
        size_t    len  = 0;

        msg.comp_name    = (char *) info->comp_name;
        msg.protocol     = (char *) info->protocol;
        msg.address      = info->addr;
        msg.pref_version = info->pref_version;
        msg.pref_syntax  = info->pref_syntax;
        if (msg.pref_syntax < 0)
                return -1;

        len = cep_msg__get_packed_size(&msg);
        if (len == 0)
                return -1;

        data = malloc(len);
        if (data == NULL)
                return -ENOMEM;

        cep_msg__pack(&msg, data);

        if (flow_write(fd, data, len) < 0) {
                free(data);
                return -1;
        }

        free(data);

        return 0;
}

int cep_snd(int                      fd,
            const struct conn_info * in)
{
        if (in == NULL)
                return -EINVAL;

        return send_msg(fd, in);
}

int cep_rcv(int                fd,
            struct conn_info * out)
{
        if (out == NULL)
                return -EINVAL;

        return read_msg(fd, out);
}