summaryrefslogtreecommitdiffstats
path: root/tests/client-server.c
blob: 7d1cfc63c75d40c752b191771f97407dab65b4e1 (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
/* PipeWire AGL Cluster IPC
 *
 * Copyright © 2021 Collabora Ltd.
 *    @author Julian Bouzas <julian.bouzas@collabora.com>
 *
 * SPDX-License-Identifier: MIT
 */

#include <glib.h>
#include <spa/pod/builder.h>
#include <spa/pod/parser.h>
#include <icipc.h>
#include <unistd.h>

static bool
increment_request_handler (struct icipc_server *self, int client_fd,
    const char *name, const struct spa_pod *args, void *data)
{
  int32_t val = 0;
  g_assert_true (spa_pod_is_int (args));
  g_assert_true (spa_pod_get_int (args, &val) == 0);
  struct spa_pod_int res = SPA_POD_INIT_Int (val + 1);
  return icipc_server_reply_ok (self, client_fd, (struct spa_pod *)&res);
}

static bool
error_request_handler (struct icipc_server *self, int client_fd,
    const char *name, const struct spa_pod *args, void *data)
{
  return icipc_server_reply_error (self, client_fd, "error message");
}

struct reply_data {
  int32_t incremented;
  const char *error;
  int n_replies;
  GMutex mutex;
  GCond cond;
};

static void
wait_for_reply (struct reply_data *data, int n_replies)
{
  g_mutex_lock (&data->mutex);
  while (data->n_replies < n_replies)
    g_cond_wait (&data->cond, &data->mutex);
  g_mutex_unlock (&data->mutex);
}

static void
reply_handler (struct icipc_sender *self, const uint8_t *buffer, size_t size, void *p)
{
  struct reply_data *data = p;
  g_assert_nonnull (data);

  g_mutex_lock (&data->mutex);

  const struct spa_pod *pod = icipc_client_send_request_finish (self, buffer, size, &data->error);
  if (pod) {
    g_assert_true (spa_pod_is_int (pod));
    g_assert_true (spa_pod_get_int (pod, &data->incremented) == 0);
  }
  data->n_replies++;
  g_cond_signal (&data->cond);

  g_mutex_unlock (&data->mutex);
}

static void
test_icipc_server_client ()
{
  g_autofree gchar *address = g_strdup_printf ("%s/icipc-test-%d-%d",
          g_get_tmp_dir(), getpid(), g_random_int ());
  struct icipc_server *s = icipc_server_new (address, true);
  g_assert_nonnull (s);
  struct icipc_client *c = icipc_client_new (address, true);
  g_assert_nonnull (c);
  struct reply_data data;
  g_mutex_init (&data.mutex);
  g_cond_init (&data.cond);

  /* add request handlers */
  g_assert_true (icipc_server_set_request_handler (s, "INCREMENT", increment_request_handler, NULL));
  g_assert_true (icipc_server_set_request_handler (s, "ERROR", error_request_handler, NULL));

  /* send an INCREMENT request of 3, and make sure the returned value is 4 */
  data.incremented = -1;
  data.error = NULL;
  data.n_replies = 0;
  struct spa_pod_int i = SPA_POD_INIT_Int (3);
  g_assert_true (icipc_client_send_request (c, "INCREMENT", (struct spa_pod *)&i, reply_handler, &data));
  wait_for_reply (&data, 1);
  g_assert_null (data.error);
  g_assert_cmpint (data.incremented, ==, 4);

  /* send an ERROR request, and make sure the returned value is an error */
  data.error = NULL;
  data.n_replies = 0;
  g_assert_true (icipc_client_send_request (c, "ERROR", NULL, reply_handler, &data));
  wait_for_reply (&data, 1);
  g_assert_cmpstr (data.error, ==, "error message");

  /* send an unhandled request, and make sure the server replies with an error */
  data.error = NULL;
  data.n_replies = 0;
  g_assert_true (icipc_client_send_request (c, "UNHANDLED-REQUEST", NULL, reply_handler, &data));
  wait_for_reply (&data, 1);
  g_assert_cmpstr (data.error, ==, "request handler not found");

  /* clean up */
  g_cond_clear (&data.cond);
  g_mutex_clear (&data.mutex);
  icipc_client_free (c);
  icipc_server_free (s);
}

gint
main (gint argc, gchar *argv[])
{
  g_test_init (&argc, &argv, NULL);

  g_test_add_func ("/icipc/icipc-server-client", test_icipc_server_client);

  return g_test_run ();
}