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
|
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stddef.h>
#include <unistd.h>
#include <stdint.h>
#include <errno.h>
#include <assert.h>
#include <poll.h>
#include <sys/ioctl.h>
//#include "include/uapi/kdbus/kdbus.h"
#include "../kdbus.h"
#include "kdbus-util.h"
#include "kdbus-enum.h"
static void add_match_empty(int fd)
{
struct kdbus_cmd_match cmd_match;
int ret;
memset(&cmd_match, 0, sizeof(cmd_match));
cmd_match.size = sizeof(cmd_match);
cmd_match.src_id = KDBUS_MATCH_SRC_ID_ANY;
ret = ioctl(fd, KDBUS_CMD_MATCH_ADD, &cmd_match);
if (ret < 0)
fprintf(stderr, "--- error adding conn match: %d (%m)\n", ret);
}
int main(int argc, char *argv[])
{
struct {
struct kdbus_cmd_bus_make head;
/* cgroup item */
uint64_t c_size;
uint64_t c_type;
uint64_t cgroup_id;
/* name item */
uint64_t n_size;
uint64_t n_type;
char name[64];
} bus_make;
int fdc, ret, cookie;
char *bus;
struct conn *conn_a, *conn_b;
struct pollfd fds[2];
int count;
printf("-- opening /dev/kdbus/control\n");
fdc = open("/dev/kdbus/control", O_RDWR|O_CLOEXEC);
if (fdc < 0) {
fprintf(stderr, "--- error %d (%m)\n", fdc);
return EXIT_FAILURE;
}
memset(&bus_make, 0, sizeof(bus_make));
bus_make.head.bloom_size = 64;
bus_make.cgroup_id = cgroup_systemd();
bus_make.c_type = KDBUS_MAKE_CGROUP;
bus_make.c_size = KDBUS_PART_HEADER_SIZE + sizeof(uint64_t);
snprintf(bus_make.name, sizeof(bus_make.name), "%u-testbus", getuid());
bus_make.n_type = KDBUS_MAKE_NAME;
bus_make.n_size = KDBUS_PART_HEADER_SIZE + strlen(bus_make.name) + 1;
bus_make.head.size = sizeof(struct kdbus_cmd_bus_make) +
bus_make.c_size +
bus_make.n_size;
printf("-- creating bus '%s'\n", bus_make.name);
ret = ioctl(fdc, KDBUS_CMD_BUS_MAKE, &bus_make);
if (ret) {
fprintf(stderr, "--- error %d (%m)\n", ret);
return EXIT_FAILURE;
}
if (asprintf(&bus, "/dev/kdbus/%s/bus", bus_make.name) < 0)
return EXIT_FAILURE;
conn_a = connect_to_bus(bus);
conn_b = connect_to_bus(bus);
if (!conn_a || !conn_b)
return EXIT_FAILURE;
upload_policy(conn_a->fd);
name_acquire(conn_a, "foo.bar.baz", 0);
name_acquire(conn_b, "foo.bar.baz", KDBUS_NAME_QUEUE);
name_list(conn_b);
add_match_empty(conn_a->fd);
add_match_empty(conn_b->fd);
cookie = 0;
msg_send(conn_b, NULL, 0xc0000000 | cookie, KDBUS_DST_ID_BROADCAST);
fds[0].fd = conn_a->fd;
fds[1].fd = conn_b->fd;
printf("-- entering poll loop ...\n");
for (count = 0;; count++) {
int i, nfds = sizeof(fds) / sizeof(fds[0]);
for (i = 0; i < nfds; i++) {
fds[i].events = POLLIN | POLLPRI | POLLHUP;
fds[i].revents = 0;
}
ret = poll(fds, nfds, 3000);
if (ret <= 0)
break;
if (fds[0].revents & POLLIN) {
if (count > 2)
name_release(conn_a, "foo.bar.baz");
msg_recv(conn_a);
msg_send(conn_a, NULL, 0xc0000000 | cookie++, conn_b->id);
}
if (fds[1].revents & POLLIN) {
msg_recv(conn_b);
msg_send(conn_b, NULL, 0xc0000000 | cookie++, conn_a->id);
}
name_list(conn_b);
if (count > 10)
break;
}
printf("-- closing bus connections\n");
close(conn_a->fd);
close(conn_b->fd);
free(conn_a);
free(conn_b);
printf("-- closing bus master\n");
close(fdc);
free(bus);
return EXIT_SUCCESS;
}
|
