aboutsummaryrefslogtreecommitdiffstats
path: root/lib/sender.c
blob: 7d2cd046811b5e8ec1f9c792e2062fc11cc176c4 (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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
/* PipeWire AGL Cluster IPC
 *
 * Copyright © 2021 Collabora Ltd.
 *    @author Julian Bouzas <julian.bouzas@collabora.com>
 *
 * SPDX-License-Identifier: MIT
 */

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <string.h>
#include <errno.h>
#include <assert.h>

#include "private.h"
#include "sender.h"

#define MAX_ASYNC_TASKS 128

struct icipc_sender_task {
  icipc_sender_reply_func_t func;
  void *data;
};

struct icipc_sender {
  struct sockaddr_un addr;
  int socket_fd;

  uint8_t *buffer_read;
  size_t buffer_size;

  struct epoll_thread epoll_thread;
  bool is_connected;

  icipc_sender_lost_conn_func_t lost_func;
  void *lost_data;

  struct icipc_sender_task async_tasks[MAX_ASYNC_TASKS];

  /* for subclasses */
  void *user_data;
};

static int
push_sync_task (struct icipc_sender *self,
                icipc_sender_reply_func_t func,
                void *data)
{
  size_t i;
  for (i = MAX_ASYNC_TASKS; i > 1; i--) {
    struct icipc_sender_task *curr = self->async_tasks + i - 1;
    struct icipc_sender_task *next = self->async_tasks + i - 2;
    if (next->func != NULL && curr->func == NULL) {
      curr->func = func;
      curr->data = data;
      return i - 1;
    } else if (i - 2 == 0 && next->func == NULL) {
      /* empty queue */
      next->func = func;
      next->data = data;
      return 0;
    }
  }
  return -1;
}

static void
pop_sync_task (struct icipc_sender *self,
               bool trigger,
               const uint8_t *buffer,
               size_t size)
{
  size_t i;
  for (i = 0; i < MAX_ASYNC_TASKS; i++) {
    struct icipc_sender_task *task = self->async_tasks + i;
    if (task->func != NULL) {
      if (trigger)
        task->func (self, buffer, size, task->data);
      task->func = NULL;
      return;
    }
  }
}

static void
socket_event_received (struct epoll_thread *t, int fd, void *data)
{
  struct icipc_sender *self = data;

  /* receiver sends a reply, read it trigger corresponding task */
  ssize_t size = icipc_socket_read (fd, &self->buffer_read, &self->buffer_size);
  if (size < 0) {
    icipc_log_error ("sender: could not read reply: %s", strerror(errno));
    return;
  }

  if (size == 0) {
    if (self->lost_func)
      self->lost_func (self, fd, self->lost_data);
    return;
  }

  /* trigger async task */
  pop_sync_task (self, true, self->buffer_read, size);
  return;
}

/* API */

struct icipc_sender *
icipc_sender_new (const char *path,
                  size_t buffer_size,
                  icipc_sender_lost_conn_func_t lost_func,
                  void *lost_data,
                  size_t user_size)
{
  struct icipc_sender *self;
  int res;

  if (path == NULL)
    return NULL;

  self = calloc (1, sizeof (struct icipc_sender) + user_size);
  if (self == NULL)
    return NULL;

  self->socket_fd = -1;

  /* set address */
  self->addr.sun_family = AF_LOCAL;
  res = icipc_construct_socket_path (path, self->addr.sun_path, sizeof(self->addr.sun_path));
  if (res < 0)
    goto error;

  /* create socket */
  self->socket_fd =
      socket(PF_LOCAL, SOCK_STREAM | SOCK_CLOEXEC| SOCK_NONBLOCK, 0);
  if (self->socket_fd < 0)
    goto error;

  /* alloc buffer read */
  self->buffer_size = buffer_size;
  self->buffer_read = calloc (buffer_size, sizeof (uint8_t));
  if (self->buffer_read == NULL)
    goto error;

  /* init epoll thread */
  if (!icipc_epoll_thread_init (&self->epoll_thread, self->socket_fd,
      socket_event_received, NULL, self))
    goto error;

  self->lost_func = lost_func;
  self->lost_data = lost_data;
  if (user_size > 0)
    self->user_data = (void *)((uint8_t *)self + sizeof (struct icipc_sender));

  return self;

error:
  if (self->buffer_read)
    free (self->buffer_read);
  if (self->socket_fd != -1)
    close (self->socket_fd);
  free (self);
  return NULL;
}

void
icipc_sender_free (struct icipc_sender *self)
{
  icipc_sender_disconnect (self);

  icipc_epoll_thread_destroy (&self->epoll_thread);
  free (self->buffer_read);
  close (self->socket_fd);
  free (self);
}

bool
icipc_sender_connect (struct icipc_sender *self)
{
  if (icipc_sender_is_connected (self))
    return true;

  if (connect(self->socket_fd, (struct sockaddr *)&self->addr,
        sizeof(self->addr)) == 0 &&
      icipc_epoll_thread_start (&self->epoll_thread)) {
    self->is_connected = true;
    return true;
  }

  return false;
}

void
icipc_sender_disconnect (struct icipc_sender *self)
{
  if (icipc_sender_is_connected (self)) {
    icipc_epoll_thread_stop (&self->epoll_thread);
    shutdown(self->socket_fd, SHUT_RDWR);
    self->is_connected = false;
  }
}

bool
icipc_sender_is_connected (struct icipc_sender *self)
{
  return self->is_connected;
}

bool
icipc_sender_send (struct icipc_sender *self,
                   const uint8_t *buffer,
                   size_t size,
                   icipc_sender_reply_func_t func,
                   void *data)
{
  int id = -1;

  if (buffer == NULL || size == 0)
    return false;

  if (!icipc_sender_is_connected (self))
    return false;

  /* add the task in the queue */
  if (func) {
    id = push_sync_task (self, func, data);
    if (id == -1)
      return false;
  }

  /* write buffer and remove task if it fails */
  if (icipc_socket_write (self->socket_fd, buffer, size) <= 0) {
    if (id != -1)
      self->async_tasks[id].func = NULL;
    return false;
  }

  return true;
}

void *
icipc_sender_get_user_data (struct icipc_sender *self)
{
  return self->user_data;
}