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
|
/* ISC license. */
#include <stdint.h>
#include <string.h>
#include <skalibs/types.h>
#include <skalibs/sgetopt.h>
#include <skalibs/strerr2.h>
#include <skalibs/tai.h>
#include <skalibs/buffer.h>
#include <skalibs/stralloc.h>
#include "skabus-rpccctl.h"
#define USAGE "skabus-rpccctl [ -t timeout ] help|interface|interface-remove|query args..."
#define dieusage() strerr_dieusage(100, USAGE)
static tain_t deadline ;
static skabus_rpcc_t a = SKABUS_RPCC_ZERO ;
static void add_interface (char const *rpccpath, char const *ifname, char const *ifprog, char const *re)
{
if (!skabus_rpcc_start_g(&a, rpccpath, &deadline))
strerr_diefu2sys(111, "start session with skabus-rpcc instance at ", rpccpath) ;
if (!skabus_rpcc_interface_register_g(&a, ifname, ifprog, re, &deadline))
strerr_diefu6sys(111, "register interface ", ifname, " with body ", ifprog, " and regex ", re) ;
skabus_rpcc_end(&a) ;
}
static void remove_interface (char const *rpccpath, char const *ifname)
{
if (!skabus_rpcc_start_g(&a, rpccpath, &deadline))
strerr_diefu2sys(111, "start session with skabus-rpcc instance at ", rpccpath) ;
if (!skabus_rpcc_interface_unregister_g(&a, ifname, &deadline))
strerr_diefu2sys(111, "unregister interface ", ifname) ;
skabus_rpcc_end(&a) ;
}
static void query (char const *rpccpath, char const *ifname, char const *query, uint32_t timeout)
{
stralloc sa = STRALLOC_ZERO ;
if (!skabus_rpcc_start_g(&a, rpccpath, &deadline))
strerr_diefu2sys(111, "start session with skabus-rpcc instance at ", rpccpath) ;
if (!skabus_rpcc_query_g(&a, &sa, ifname, query, timeout, &deadline))
strerr_diefu2sys(111, "send query to interface ", ifname) ;
skabus_rpcc_end(&a) ;
if (buffer_putflush(buffer_1, sa.s, sa.len) < 0)
strerr_diefu1sys(111, "write to stdout") ;
stralloc_free(&sa) ;
}
static void quit (char const *rpccpath)
{
if (!skabus_rpcc_start_g(&a, rpccpath, &deadline))
strerr_diefu2sys(111, "start session with skabus-rpcc instance at ", rpccpath) ;
if (!skabus_rpcc_quit_g(&a, &deadline))
strerr_diefu1sys(111, "send quit command") ;
skabus_rpcc_end(&a) ;
}
static inline void print_help (void)
{
static char const *help =
"skabus-rpccctl help\n"
"skabus-rpccctl [ -t timeout ] interface rpccpath interface-name interface-program clientid-regex\n"
"skabus-rpccctl [ -t timeout ] interface-remove rpccpath interface-name\n"
"skabus-rpccctl [ -t timeout ] [ -T limit ] query rpccpath interface-name query\n"
"skabus-rpccctl quit\n" ;
if (buffer_putsflush(buffer_1, help) < 0)
strerr_diefu1sys(111, "write to stdout") ;
}
static inline unsigned int lookup (char const *const *table, char const *command)
{
unsigned int i = 0 ;
for (; table[i] ; i++) if (!strcmp(command, table[i])) break ;
return i ;
}
static inline unsigned int parse_command (char const *command)
{
static char const *const command_table[13] =
{
"help",
"interface",
"interface-remove",
"query",
"quit",
0
} ;
unsigned int i = lookup(command_table, command) ;
if (!command_table[i]) dieusage() ;
return i ;
}
int main (int argc, char const *const *argv)
{
unsigned int what ;
uint32_t timeout = 0 ;
PROG = "skabus-rpccctl" ;
{
unsigned int t = 0 ;
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
int opt = subgetopt_r(argc, argv, "t:T:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 't' : if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
case 'T' : if (!uint320_scan(l.arg, &timeout)) dieusage() ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
if (t) tain_from_millisecs(&deadline, t) ;
else deadline = tain_infinite_relative ;
}
if (!argc) dieusage() ;
tain_now_g() ;
tain_add_g(&deadline, &deadline) ;
what = parse_command(argv[0]) ;
switch (what)
{
case 0 :
print_help() ;
break ;
case 1 :
if (argc < 5) dieusage() ;
add_interface(argv[2], argv[3], argv[4], argv[5]) ;
break ;
case 2 :
if (argc < 3) dieusage() ;
remove_interface(argv[2], argv[3]) ;
break ;
case 3 :
if (argc < 4) dieusage() ;
query(argv[2], argv[3], argv[4], timeout) ;
break ;
case 4 :
if (argc < 2) dieusage() ;
quit(argv[2]) ;
break ;
default : dieusage() ;
}
return 0 ;
}
|
