author | unc0rr |
Sat, 14 Jan 2017 22:30:09 +0300 | |
changeset 12137 | 193dfdcb0620 |
parent 12136 | e25a82ce2374 |
child 12140 | 7e259e47eb0d |
permissions | -rw-r--r-- |
12133 | 1 |
use nom::*; |
2 |
||
3 |
use std::str; |
|
4 |
use std::str::FromStr; |
|
5 |
use super::messages::HWProtocolMessage; |
|
6 |
use super::messages::HWProtocolMessage::*; |
|
7 |
||
8 |
named!(end_of_message, tag!("\n\n")); |
|
9 |
named!(a_line<&[u8], &str>, map_res!(not_line_ending, str::from_utf8)); |
|
12135 | 10 |
named!( u8_line<&[u8], u8>, map_res!(a_line, FromStr::from_str)); |
11 |
named!(u32_line<&[u8], u32>, map_res!(a_line, FromStr::from_str)); |
|
12134 | 12 |
named!(opt_param<&[u8], Option<&str> >, opt!(flat_map!(preceded!(eol, a_line), non_empty))); |
13 |
||
12133 | 14 |
named!(basic_message<&[u8], HWProtocolMessage>, alt!( |
12135 | 15 |
do_parse!(tag!("PING") >> (Ping)) |
12133 | 16 |
| do_parse!(tag!("PONG") >> (Pong)) |
17 |
| do_parse!(tag!("LIST") >> (List)) |
|
12135 | 18 |
| do_parse!(tag!("BANLIST") >> (BanList)) |
12133 | 19 |
| do_parse!(tag!("GET_SERVER_VAR") >> (GetServerVar)) |
12135 | 20 |
| do_parse!(tag!("TOGGLE_READY") >> (ToggleReady)) |
21 |
| do_parse!(tag!("START_GAME") >> (StartGame)) |
|
22 |
| do_parse!(tag!("ROUNDFINISHED") >> (RoundFinished)) |
|
23 |
| do_parse!(tag!("TOGGLE_RESTRICT_JOINS") >> (ToggleRestrictJoin)) |
|
24 |
| do_parse!(tag!("TOGGLE_RESTRICT_TEAMS") >> (ToggleRestrictTeams)) |
|
12133 | 25 |
| do_parse!(tag!("TOGGLE_REGISTERED_ONLY") >> (ToggleRegisteredOnly)) |
26 |
)); |
|
27 |
||
28 |
named!(one_param_message<&[u8], HWProtocolMessage>, alt!( |
|
12135 | 29 |
do_parse!(tag!("NICK") >> eol >> n: a_line >> (Nick(n))) |
30 |
| do_parse!(tag!("INFO") >> eol >> n: a_line >> (Info(n))) |
|
31 |
| do_parse!(tag!("CHAT") >> eol >> m: a_line >> (Chat(m))) |
|
32 |
| do_parse!(tag!("FOLLOW") >> eol >> n: a_line >> (Follow(n))) |
|
33 |
| do_parse!(tag!("KICK") >> eol >> n: a_line >> (Kick(n))) |
|
34 |
| do_parse!(tag!("UNBAN") >> eol >> n: a_line >> (Unban(n))) |
|
35 |
| do_parse!(tag!("EM") >> eol >> m: a_line >> (EngineMessage(m))) |
|
36 |
| do_parse!(tag!("TEAMCHAT") >> eol >> m: a_line >> (TeamChat(m))) |
|
37 |
| do_parse!(tag!("ROOM_NAME") >> eol >> n: a_line >> (RoomName(n))) |
|
38 |
| do_parse!(tag!("REMOVE_TEAM") >> eol >> n: a_line >> (RemoveTeam(n))) |
|
39 |
||
40 |
| do_parse!(tag!("PROTO") >> eol >> d: u32_line >> (Proto(d))) |
|
41 |
||
42 |
| do_parse!(tag!("QUIT") >> msg: opt_param >> (Quit(msg))) |
|
43 |
)); |
|
44 |
||
45 |
named!(cmd_message<&[u8], HWProtocolMessage>, preceded!(tag!("CMD\n"), alt!( |
|
46 |
do_parse!(tag_no_case!("STATS") >> (Stats)) |
|
47 |
| do_parse!(tag_no_case!("FIX") >> (Fix)) |
|
48 |
| do_parse!(tag_no_case!("UNFIX") >> (Unfix)) |
|
49 |
| do_parse!(tag_no_case!("RESTART_SERVER") >> eol >> tag!("YES") >> (RestartServer)) |
|
50 |
| do_parse!(tag_no_case!("REGISTERED_ONLY") >> (ToggleServerRegisteredOnly)) |
|
51 |
| do_parse!(tag_no_case!("SUPER_POWER") >> (SuperPower)) |
|
52 |
| do_parse!(tag_no_case!("PART") >> eol >> m: opt_param >> (Quit(m))) |
|
53 |
| do_parse!(tag_no_case!("QUIT") >> eol >> m: opt_param >> (Part(m))) |
|
54 |
| do_parse!(tag_no_case!("DELEGATE") >> eol >> n: a_line >> (Delegate(n))) |
|
55 |
| do_parse!(tag_no_case!("SAVEROOM") >> eol >> r: a_line >> (SaveRoom(r))) |
|
56 |
| do_parse!(tag_no_case!("LOADROOM") >> eol >> r: a_line >> (LoadRoom(r))) |
|
57 |
| do_parse!(tag_no_case!("DELETE") >> eol >> r: a_line >> (Delete(r))) |
|
58 |
| do_parse!(tag_no_case!("GLOBAL") >> eol >> m: a_line >> (Global(m))) |
|
59 |
| do_parse!(tag_no_case!("WATCH") >> eol >> i: a_line >> (Watch(i))) |
|
60 |
| do_parse!(tag_no_case!("GREETING") >> eol >> m: a_line >> (Greeting(m))) |
|
61 |
| do_parse!(tag_no_case!("VOTE") >> eol >> m: a_line >> (Vote(m))) |
|
62 |
| do_parse!(tag_no_case!("FORCE") >> eol >> m: a_line >> (ForceVote(m))) |
|
63 |
| do_parse!(tag_no_case!("INFO") >> eol >> n: a_line >> (Info(n))) |
|
64 |
| do_parse!(tag_no_case!("MAXTEAMS") >> eol >> n: u8_line >> (MaxTeams(n))) |
|
65 |
))); |
|
66 |
||
67 |
named!(complex_message<&[u8], HWProtocolMessage>, alt!( |
|
68 |
do_parse!(tag!("PASSWORD") >> eol >> |
|
69 |
p: a_line >> eol >> |
|
70 |
s: a_line >> |
|
71 |
(Password(p, s))) |
|
72 |
| do_parse!(tag!("CHECKER") >> eol >> |
|
73 |
i: u32_line >> eol >> |
|
74 |
n: a_line >> eol >> |
|
75 |
p: a_line >> |
|
76 |
(Checker(i, n, p))) |
|
77 |
| do_parse!(tag!("CREATE_ROOM") >> eol >> |
|
78 |
n: a_line >> |
|
79 |
p: opt_param >> |
|
80 |
(CreateRoom(n, p))) |
|
81 |
| do_parse!(tag!("JOIN") >> eol >> |
|
82 |
n: a_line >> |
|
83 |
p: opt_param >> |
|
84 |
(Join(n, p))) |
|
85 |
| do_parse!(tag!("BAN") >> eol >> |
|
86 |
n: a_line >> eol >> |
|
87 |
r: a_line >> eol >> |
|
88 |
t: u32_line >> |
|
89 |
(Ban(n, r, t))) |
|
90 |
| do_parse!(tag!("BAN_IP") >> eol >> |
|
91 |
n: a_line >> eol >> |
|
92 |
r: a_line >> eol >> |
|
93 |
t: u32_line >> |
|
94 |
(BanIP(n, r, t))) |
|
95 |
| do_parse!(tag!("BAN_NICK") >> eol >> |
|
96 |
n: a_line >> eol >> |
|
97 |
r: a_line >> eol >> |
|
98 |
t: u32_line >> |
|
99 |
(BanNick(n, r, t))) |
|
12133 | 100 |
)); |
101 |
||
12137
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
102 |
named!(malformed_message<&[u8], HWProtocolMessage>, |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
103 |
do_parse!(separated_list!(eol, a_line) >> (Malformed))); |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
104 |
|
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
105 |
named!(empty_message<&[u8], HWProtocolMessage>, |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
106 |
do_parse!(alt!(end_of_message | eol) >> (Empty))); |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
107 |
|
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
108 |
named!(message<&[u8], HWProtocolMessage>, alt!(terminated!( |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
109 |
alt!( |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
110 |
basic_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
111 |
| one_param_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
112 |
| cmd_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
113 |
| complex_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
114 |
| malformed_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
115 |
), end_of_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
116 |
) |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
117 |
| empty_message |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
118 |
) |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
119 |
); |
12133 | 120 |
|
12136 | 121 |
named!(pub extract_messages<&[u8], Vec<HWProtocolMessage> >, many0!(complete!(message))); |
12133 | 122 |
|
123 |
#[test] |
|
124 |
fn parse_test() { |
|
12134 | 125 |
assert_eq!(message(b"PING\n\n"), IResult::Done(&b""[..], Ping)); |
126 |
assert_eq!(message(b"START_GAME\n\n"), IResult::Done(&b""[..], StartGame)); |
|
12133 | 127 |
assert_eq!(message(b"NICK\nit's me\n\n"), IResult::Done(&b""[..], Nick("it's me"))); |
12134 | 128 |
assert_eq!(message(b"PROTO\n51\n\n"), IResult::Done(&b""[..], Proto(51))); |
129 |
assert_eq!(message(b"QUIT\nbye-bye\n\n"), IResult::Done(&b""[..], Quit(Some("bye-bye")))); |
|
130 |
assert_eq!(message(b"QUIT\n\n"), IResult::Done(&b""[..], Quit(None))); |
|
12135 | 131 |
assert_eq!(message(b"CMD\nwatch\ndemo\n\n"), IResult::Done(&b""[..], Watch("demo"))); |
132 |
assert_eq!(message(b"BAN\nme\nbad\n77\n\n"), IResult::Done(&b""[..], Ban("me", "bad", 77))); |
|
12136 | 133 |
|
12137
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
134 |
assert_eq!(extract_messages(b"PING\n\nPING\n\nP"), IResult::Done(&b"P"[..], vec![Ping, Ping])); |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
135 |
assert_eq!(extract_messages(b"SING\n\nPING\n\n"), IResult::Done(&b""[..], vec![Malformed, Ping])); |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
136 |
assert_eq!(extract_messages(b"\n\n\n\nPING\n\n"), IResult::Done(&b""[..], vec![Empty, Empty, Ping])); |
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
137 |
assert_eq!(extract_messages(b"\n\n\nPING\n\n"), IResult::Done(&b""[..], vec![Empty, Empty, Ping])); |
12133 | 138 |
} |