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
| | /*
* Copyright (C) 2012-2013, Eric Wong <normalperson@yhbt.net>
* License: GPLv3 or later (see COPYING for details)
*/
#include "cmogstored.h"
#include "mgmt.h"
/*
* only set fsck prio if we're still accepting connections, graceful
* shutdown in single-threaded mode uses normal (fair) prio
*/
static void set_prio_fsck(struct mog_mgmt *mgmt)
{
if (mgmt->svc->mgmt_mfd)
mgmt->prio = MOG_PRIO_FSCK;
}
%%{
machine mgmt_parser;
include path_parser "path_parser.rl";
eor = '\r'?'\n';
reason = ' '("fsck" @ { set_prio_fsck(mgmt); } | [a-zA-Z0-9_]+);
invalid_line := (
[ \t]*
([^ \t\r]+) > { mgmt->mark[0] = fpc - buf; }
(any-'\n')* > { mgmt->mark[1] = fpc - buf; }
'\n'
) @ {
mog_mgmt_fn_unknown(mgmt, buf);
really_done = 1;
fbreak;
};
size = (
"size "(mog_path) > { mgmt->mark[0] = fpc - buf; }
eor > { mgmt->mark[1] = fpc - buf; }
@ { mog_mgmt_fn_size(mgmt, buf); fbreak; }
);
digest = (
(
"MD5" @ { mgmt->alg = GC_MD5; }
|
"SHA-1" @ { mgmt->alg = GC_SHA1; }
)
" "
(mog_path) > { mgmt->mark[0] = fpc - buf; }
( reason? eor) > { mgmt->mark[1] = fpc - buf; }
@ { mog_mgmt_fn_digest(mgmt, buf); fbreak; }
);
watch = "watch" eor @ {
static int have_iostat = 1;
if (have_iostat)
mgmt->forward = MOG_IOSTAT;
else
mog_mgmt_fn_watch_err(mgmt);
fbreak;
};
aio_threads = (
"server aio_threads"i ' '?'='(' ')?
(digit+) > { mgmt->mark[0] = fpc - buf; }
eor > { mgmt->mark[1] = fpc - buf; }
@ { mog_mgmt_fn_aio_threads(mgmt, buf); fbreak; }
);
blank = [ \t]* eor @ { mog_mgmt_fn_blank(mgmt); fbreak; };
shutdown = "shutdown" (" "+"graceful")? eor @ {
cmogstored_quit();
fbreak;
};
command = (digest|size|watch|aio_threads|shutdown|blank);
main := command $! {
p = buf;
fhold;
fgoto invalid_line;
};
}%%
%% write data;
void mog_mgmt_reset_parser(struct mog_mgmt *mgmt)
{
int cs;
%% write init;
mgmt->cs = cs;
mgmt->mark[0] = mgmt->mark[1] = 0;
}
void mog_mgmt_init(struct mog_mgmt *mgmt, struct mog_svc *svc)
{
memset(mgmt, 0, sizeof(struct mog_mgmt));
mog_mgmt_reset_parser(mgmt);
mgmt->svc = svc;
}
enum mog_parser_state
mog_mgmt_parse(struct mog_mgmt *mgmt, char *buf, size_t len)
{
char *p, *pe, *eof = NULL;
int cs = mgmt->cs;
int really_done = 0;
size_t off = mgmt->buf_off;
uint32_t *mog_devid = &mgmt->mog_devid;
assert(mgmt->wbuf == NULL && "unwritten data in buffer");
assert(off <= len && "mgmt offset past end of buffer");
p = buf + off;
pe = buf + len;
assert((void *)(pe - p) == (void *)(len - off) &&
"pointers aren't same distance");
%% write exec;
if (really_done)
cs = mgmt_parser_first_final;
mgmt->cs = cs;
mgmt->buf_off = p - buf;
if (cs == mgmt_parser_error)
return MOG_PARSER_ERROR;
assert(p <= pe && "buffer overflow after mgmt parse");
assert(mgmt->buf_off <= len && "offset longer than len");
if (mgmt->cs == mgmt_parser_first_final) return MOG_PARSER_DONE;
return MOG_PARSER_CONTINUE;
}
|