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
|
|
| #include <stdio.h>
| #include <stdlib.h>
| #include <string.h>
| #include <unistd.h>
|
| #include <nng/nng.h>
| #include <nng/protocol/bus0/bus.h>
|
| void
| fatal(const char *func, int rv)
| {
| fprintf(stderr, "%s: %s\n", func, nng_strerror(rv));
| exit(1);
| }
|
| int
| node(int argc, char **argv)
| {
| nng_socket sock;
| int rv;
| size_t sz;
|
| if ((rv = nng_bus0_open(&sock)) != 0) {
| fatal("nng_bus0_open", rv);
| }
| if ((rv = nng_listen(sock, argv[2], NULL, 0)) != 0) {
| fatal("nng_listen", rv);
| }
|
| sleep(1); // wait for peers to bind
| if (argc >= 3) {
| for (int x = 3; x < argc; x++) {
| if ((rv = nng_dial(sock, argv[x], NULL, 0)) != 0) {
| fatal("nng_dial", rv);
| }
| }
| }
|
| sleep(1); // wait for connects to establish
|
| // SEND
| sz = strlen(argv[1]) + 1; // '\0' too
| printf("%s: SENDING '%s' ONTO BUS\n", argv[1], argv[1]);
| if ((rv = nng_send(sock, argv[1], sz, 0)) != 0) {
| fatal("nng_send", rv);
| }
|
| // RECV
| for (;;) {
| char *buf = NULL;
| size_t sz;
| if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) !=0) {
| if (rv == NNG_ETIMEDOUT) {
| fatal("nng_recv", rv);
| }
| }
| printf("%s: RECEIVED '%s' FROM BUS\n", argv[1], buf);
| nng_free(buf, sz);
| }
| nng_close(sock);
| return (0);
| }
|
| int
| main(int argc, char **argv)
| {
| if (argc >= 3) {
| return (node(argc, argv));
| }
| fprintf(stderr, "Usage: bus <NODE_NAME> <URL> <URL> ...\n");
| return 1;
| }
|
|
