d330d9b8 |
1 | (* HCoop Domtool (http://hcoop.sourceforge.net/) |
2 | * Copyright (c) 2006, Adam Chlipala |
3 | * |
4 | * This program is free software; you can redistribute it and/or |
5 | * modify it under the terms of the GNU General Public License |
6 | * as published by the Free Software Foundation; either version 2 |
7 | * of the License, or (at your option) any later version. |
8 | * |
9 | * This program is distributed in the hope that it will be useful, |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | * GNU General Public License for more details. |
13 | * |
14 | * You should have received a copy of the GNU General Public License |
15 | * along with this program; if not, write to the Free Software |
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
17 | *) |
18 | |
19 | (* Network messages *) |
20 | |
21 | structure Msg :> MSG = struct |
22 | |
23 | open OpenSSL MsgTypes Slave |
24 | |
25 | val a2i = fn Add => 0 |
26 | | Delete => 1 |
27 | | Modify => 2 |
28 | |
29 | val i2a = fn 0 => Add |
30 | | 1 => Delete |
31 | | 2 => Modify |
32 | | _ => raise OpenSSL.OpenSSL "Bad action number to deserialize" |
33 | |
e2130d9c |
34 | fun sendAcl (bio, {user, class, value}) = |
35 | (OpenSSL.writeString (bio, user); |
36 | OpenSSL.writeString (bio, class); |
37 | OpenSSL.writeString (bio, value)) |
38 | |
39 | fun recvAcl bio = |
40 | case (OpenSSL.readString bio, OpenSSL.readString bio, OpenSSL.readString bio) of |
41 | (SOME user, SOME class, SOME value) => SOME {user = user, class = class, value = value} |
42 | | _ => NONE |
43 | |
c8a739af |
44 | fun sendList f (bio, ls) = |
45 | (app (fn x => |
46 | (OpenSSL.writeInt (bio, 1); |
47 | f (bio, x))) ls; |
48 | OpenSSL.writeInt (bio, 0)) |
49 | |
50 | fun recvList f bio = |
51 | let |
52 | fun loop ls = |
53 | case OpenSSL.readInt bio of |
54 | SOME 0 => SOME (rev ls) |
55 | | SOME 1 => |
56 | (case f bio of |
57 | SOME x => loop (x :: ls) |
58 | | NONE => NONE) |
59 | | _ => NONE |
60 | in |
61 | loop [] |
62 | end |
63 | |
2bc895e7 |
64 | fun sendOption f (bio, opt) = |
65 | case opt of |
66 | NONE => OpenSSL.writeInt (bio, 0) |
67 | | SOME x => (OpenSSL.writeInt (bio, 1); |
68 | f (bio, x)) |
69 | |
70 | fun recvOption f bio = |
71 | case OpenSSL.readInt bio of |
72 | SOME 0 => SOME NONE |
73 | | SOME 1 => |
74 | (case f bio of |
75 | SOME x => SOME (SOME x) |
76 | | NONE => NONE) |
77 | | _ => NONE |
78 | |
bb8cc8c9 |
79 | fun sendBool (bio, b) = |
80 | if b then |
81 | OpenSSL.writeInt (bio, 1) |
82 | else |
83 | OpenSSL.writeInt (bio, 0) |
84 | |
85 | fun recvBool bio = |
86 | case OpenSSL.readInt bio of |
87 | SOME 0 => SOME false |
88 | | SOME 1 => SOME true |
89 | | _ => NONE |
90 | |
9f063ecc |
91 | fun sendSockPerm (bio, p) = |
92 | case p of |
93 | Any => OpenSSL.writeInt (bio, 0) |
94 | | Client => OpenSSL.writeInt (bio, 1) |
95 | | Server => OpenSSL.writeInt (bio, 2) |
96 | | Nada => OpenSSL.writeInt (bio, 3) |
97 | |
98 | fun recvSockPerm bio = |
99 | case OpenSSL.readInt bio of |
100 | SOME 0 => SOME Any |
101 | | SOME 1 => SOME Client |
102 | | SOME 2 => SOME Server |
103 | | SOME 3 => SOME Nada |
104 | | _ => NONE |
105 | |
e5f8cb7b |
106 | fun sendQuery (bio, q) = |
107 | case q of |
108 | QApt s => (OpenSSL.writeInt (bio, 0); |
109 | OpenSSL.writeString (bio, s)) |
891b033c |
110 | | QCron s => (OpenSSL.writeInt (bio, 1); |
111 | OpenSSL.writeString (bio, s)) |
112 | | QFtp s => (OpenSSL.writeInt (bio, 2); |
113 | OpenSSL.writeString (bio, s)) |
4ee81d53 |
114 | | QTrustedPath s => (OpenSSL.writeInt (bio, 3); |
115 | OpenSSL.writeString (bio, s)) |
9f063ecc |
116 | | QSocket s => (OpenSSL.writeInt (bio, 4); |
117 | OpenSSL.writeString (bio, s)) |
5514f5b5 |
118 | | QFirewall s => (OpenSSL.writeInt (bio, 5); |
119 | OpenSSL.writeString (bio, s)) |
e5f8cb7b |
120 | |
121 | fun recvQuery bio = |
122 | case OpenSSL.readInt bio of |
123 | SOME n => |
124 | (case n of |
125 | 0 => Option.map QApt (OpenSSL.readString bio) |
891b033c |
126 | | 1 => Option.map QCron (OpenSSL.readString bio) |
127 | | 2 => Option.map QFtp (OpenSSL.readString bio) |
4ee81d53 |
128 | | 3 => Option.map QTrustedPath (OpenSSL.readString bio) |
9f063ecc |
129 | | 4 => Option.map QSocket (OpenSSL.readString bio) |
5514f5b5 |
130 | | 5 => Option.map QFirewall (OpenSSL.readString bio) |
e5f8cb7b |
131 | | _ => NONE) |
132 | | NONE => NONE |
133 | |
d330d9b8 |
134 | fun send (bio, m) = |
135 | case m of |
136 | MsgOk => OpenSSL.writeInt (bio, 1) |
137 | | MsgError s => (OpenSSL.writeInt (bio, 2); |
138 | OpenSSL.writeString (bio, s)) |
139 | | MsgConfig s => (OpenSSL.writeInt (bio, 3); |
140 | OpenSSL.writeString (bio, s)) |
141 | | MsgFile {action, domain, dir, file} => |
142 | (OpenSSL.writeInt (bio, 4); |
143 | OpenSSL.writeInt (bio, a2i action); |
144 | OpenSSL.writeString (bio, domain); |
145 | OpenSSL.writeString (bio, dir); |
146 | OpenSSL.writeString (bio, file)) |
147 | | MsgDoFiles => OpenSSL.writeInt (bio, 5) |
e2130d9c |
148 | | MsgGrant acl => (OpenSSL.writeInt (bio, 6); |
149 | sendAcl (bio, acl)) |
d1aa6a21 |
150 | | MsgRevoke acl => (OpenSSL.writeInt (bio, 7); |
151 | sendAcl (bio, acl)) |
646381db |
152 | | MsgListPerms user => (OpenSSL.writeInt (bio, 8); |
153 | OpenSSL.writeString (bio, user)) |
154 | | MsgPerms classes => (OpenSSL.writeInt (bio, 9); |
c8a739af |
155 | sendList (fn (bio, (class, values)) => |
156 | (OpenSSL.writeString (bio, class); |
157 | sendList OpenSSL.writeString (bio, values))) |
158 | (bio, classes)) |
d0e75410 |
159 | | MsgWhoHas {class, value} => (OpenSSL.writeInt (bio, 10); |
160 | OpenSSL.writeString (bio, class); |
161 | OpenSSL.writeString (bio, value)) |
162 | | MsgWhoHasResponse users => (OpenSSL.writeInt (bio, 11); |
c8a739af |
163 | sendList OpenSSL.writeString (bio, users)) |
164 | | MsgMultiConfig codes => (OpenSSL.writeInt (bio, 12); |
165 | sendList OpenSSL.writeString (bio, codes)) |
aba1f07e |
166 | | MsgRmdom doms => (OpenSSL.writeInt (bio, 13); |
167 | sendList OpenSSL.writeString (bio, doms)) |
f92c6883 |
168 | | MsgRegenerate => OpenSSL.writeInt (bio, 14) |
aba1f07e |
169 | | MsgRmuser dom => (OpenSSL.writeInt (bio, 15); |
170 | OpenSSL.writeString (bio, dom)) |
2bc895e7 |
171 | | MsgCreateDbUser {dbtype, passwd} => (OpenSSL.writeInt (bio, 16); |
172 | OpenSSL.writeString (bio, dbtype); |
173 | sendOption OpenSSL.writeString (bio, passwd)) |
d34cbcb8 |
174 | | MsgCreateDbTable {dbtype, dbname} => (OpenSSL.writeInt (bio, 17); |
175 | OpenSSL.writeString (bio, dbtype); |
176 | OpenSSL.writeString (bio, dbname)) |
c45f1662 |
177 | | MsgNewMailbox {domain, user, passwd, mailbox} => |
178 | (OpenSSL.writeInt (bio, 18); |
179 | OpenSSL.writeString (bio, domain); |
180 | OpenSSL.writeString (bio, user); |
181 | OpenSSL.writeString (bio, passwd); |
182 | OpenSSL.writeString (bio, mailbox)) |
183 | | MsgPasswdMailbox {domain, user, passwd} => |
184 | (OpenSSL.writeInt (bio, 19); |
185 | OpenSSL.writeString (bio, domain); |
186 | OpenSSL.writeString (bio, user); |
187 | OpenSSL.writeString (bio, passwd)) |
188 | | MsgRmMailbox {domain, user} => |
189 | (OpenSSL.writeInt (bio, 20); |
190 | OpenSSL.writeString (bio, domain); |
191 | OpenSSL.writeString (bio, user)) |
0a58b2f3 |
192 | | MsgListMailboxes domain => |
193 | (OpenSSL.writeInt (bio, 21); |
194 | OpenSSL.writeString (bio, domain)) |
195 | | MsgMailboxes users => |
196 | (OpenSSL.writeInt (bio, 22); |
1850d85f |
197 | sendList (fn (bio, {user, mailbox}) => |
198 | (OpenSSL.writeString (bio, user); |
199 | OpenSSL.writeString (bio, mailbox))) |
200 | (bio, users)) |
bb8cc8c9 |
201 | | MsgSaQuery addr => (OpenSSL.writeInt (bio, 23); |
202 | OpenSSL.writeString (bio, addr)) |
203 | | MsgSaStatus b => (OpenSSL.writeInt (bio, 24); |
204 | sendBool (bio, b)) |
205 | | MsgSaSet (addr, b) => (OpenSSL.writeInt (bio, 25); |
206 | OpenSSL.writeString (bio, addr); |
207 | sendBool (bio, b)) |
c685120e |
208 | | MsgSmtpLogReq domain => (OpenSSL.writeInt (bio, 26); |
209 | OpenSSL.writeString (bio, domain)) |
210 | | MsgSmtpLogRes domain => (OpenSSL.writeInt (bio, 27); |
211 | OpenSSL.writeString (bio, domain)) |
dc6789fc |
212 | | MsgDbPasswd {dbtype, passwd} => (OpenSSL.writeInt (bio, 28); |
213 | OpenSSL.writeString (bio, dbtype); |
214 | OpenSSL.writeString (bio, passwd)) |
4c68bd31 |
215 | | MsgShutdown => OpenSSL.writeInt (bio, 29) |
8301e970 |
216 | | MsgYes => OpenSSL.writeInt (bio, 30) |
217 | | MsgNo => OpenSSL.writeInt (bio, 31) |
e5f8cb7b |
218 | | MsgQuery q => (OpenSSL.writeInt (bio, 32); |
219 | sendQuery (bio, q)) |
9f063ecc |
220 | | MsgSocket p => (OpenSSL.writeInt (bio, 33); |
221 | sendSockPerm (bio, p)) |
5514f5b5 |
222 | | MsgFirewall ls => (OpenSSL.writeInt (bio, 34); |
223 | sendList OpenSSL.writeString (bio, ls)) |
b4f5896d |
224 | | MsgRegenerateTc => OpenSSL.writeInt (bio, 35) |
a71b2f2b |
225 | | MsgDropDb {dbtype, dbname} => (OpenSSL.writeInt (bio, 36); |
226 | OpenSSL.writeString (bio, dbtype); |
227 | OpenSSL.writeString (bio, dbname)) |
d330d9b8 |
228 | |
229 | fun checkIt v = |
230 | case v of |
231 | NONE => raise OpenSSL.OpenSSL "Bad Msg format" |
232 | | _ => v |
233 | |
234 | fun recv bio = |
235 | case OpenSSL.readInt bio of |
236 | NONE => NONE |
237 | | SOME n => |
238 | checkIt (case n of |
239 | 1 => SOME MsgOk |
240 | | 2 => Option.map MsgError (OpenSSL.readString bio) |
241 | | 3 => Option.map MsgConfig (OpenSSL.readString bio) |
242 | | 4 => (case (OpenSSL.readInt bio, |
243 | OpenSSL.readString bio, |
244 | OpenSSL.readString bio, |
245 | OpenSSL.readString bio) of |
246 | (SOME action, SOME domain, SOME dir, SOME file) => |
247 | SOME (MsgFile {action = i2a action, |
248 | domain = domain, |
249 | dir = dir, |
250 | file = file}) |
251 | | _ => NONE) |
252 | | 5 => SOME MsgDoFiles |
e2130d9c |
253 | | 6 => (case recvAcl bio of |
254 | SOME acl => SOME (MsgGrant acl) |
255 | | _ => NONE) |
d1aa6a21 |
256 | | 7 => (case recvAcl bio of |
257 | SOME acl => SOME (MsgRevoke acl) |
258 | | _ => NONE) |
646381db |
259 | | 8 => (case OpenSSL.readString bio of |
260 | SOME user => SOME (MsgListPerms user) |
261 | | _ => NONE) |
c8a739af |
262 | | 9 => Option.map MsgPerms |
263 | (recvList (fn bio => |
264 | case (OpenSSL.readString bio, |
265 | recvList OpenSSL.readString bio) of |
266 | (SOME class, SOME values) => SOME (class, values) |
267 | | _ => NONE) bio) |
d0e75410 |
268 | | 10 => (case (OpenSSL.readString bio, OpenSSL.readString bio) of |
269 | (SOME class, SOME value) => SOME (MsgWhoHas {class = class, value = value}) |
270 | | _ => NONE) |
c8a739af |
271 | | 11 => Option.map MsgWhoHasResponse |
272 | (recvList OpenSSL.readString bio) |
273 | | 12 => Option.map MsgMultiConfig |
274 | (recvList OpenSSL.readString bio) |
aba1f07e |
275 | | 13 => Option.map MsgRmdom (recvList OpenSSL.readString bio) |
f92c6883 |
276 | | 14 => SOME MsgRegenerate |
aba1f07e |
277 | | 15 => Option.map MsgRmuser (OpenSSL.readString bio) |
2bc895e7 |
278 | | 16 => (case (OpenSSL.readString bio, recvOption OpenSSL.readString bio) of |
279 | (SOME dbtype, SOME passwd) => |
280 | SOME (MsgCreateDbUser {dbtype = dbtype, passwd = passwd}) |
281 | | _ => NONE) |
d34cbcb8 |
282 | | 17 => (case (OpenSSL.readString bio, OpenSSL.readString bio) of |
283 | (SOME dbtype, SOME dbname) => |
284 | SOME (MsgCreateDbTable {dbtype = dbtype, dbname = dbname}) |
285 | | _ => NONE) |
c45f1662 |
286 | | 18 => (case (OpenSSL.readString bio, OpenSSL.readString bio, |
287 | OpenSSL.readString bio, OpenSSL.readString bio) of |
288 | (SOME domain, SOME user, SOME passwd, SOME mailbox) => |
289 | SOME (MsgNewMailbox {domain = domain, user = user, |
290 | passwd = passwd, mailbox = mailbox}) |
291 | | _ => NONE) |
292 | | 19 => (case (OpenSSL.readString bio, OpenSSL.readString bio, |
293 | OpenSSL.readString bio) of |
294 | (SOME domain, SOME user, SOME passwd) => |
295 | SOME (MsgPasswdMailbox {domain = domain, user = user, |
296 | passwd = passwd}) |
297 | | _ => NONE) |
298 | | 20 => (case (OpenSSL.readString bio, OpenSSL.readString bio) of |
299 | (SOME domain, SOME user) => |
300 | SOME (MsgRmMailbox {domain = domain, user = user}) |
301 | | _ => NONE) |
0a58b2f3 |
302 | | 21 => Option.map MsgListMailboxes (OpenSSL.readString bio) |
1850d85f |
303 | | 22 => Option.map MsgMailboxes (recvList |
304 | (fn bio => |
305 | case (OpenSSL.readString bio, |
306 | OpenSSL.readString bio) of |
307 | (SOME user, SOME mailbox) => |
308 | SOME {user = user, mailbox = mailbox} |
309 | | _ => NONE) |
310 | bio) |
bb8cc8c9 |
311 | | 23 => Option.map MsgSaQuery (OpenSSL.readString bio) |
312 | | 24 => Option.map MsgSaStatus (recvBool bio) |
313 | | 25 => (case (OpenSSL.readString bio, recvBool bio) of |
314 | (SOME user, SOME b) => SOME (MsgSaSet (user, b)) |
315 | | _ => NONE) |
c685120e |
316 | | 26 => Option.map MsgSmtpLogReq (OpenSSL.readString bio) |
317 | | 27 => Option.map MsgSmtpLogRes (OpenSSL.readString bio) |
dc6789fc |
318 | | 28 => (case (OpenSSL.readString bio, OpenSSL.readString bio) of |
319 | (SOME dbtype, SOME passwd) => |
320 | SOME (MsgDbPasswd {dbtype = dbtype, passwd = passwd}) |
321 | | _ => NONE) |
4c68bd31 |
322 | | 29 => SOME MsgShutdown |
8301e970 |
323 | | 30 => SOME MsgYes |
324 | | 31 => SOME MsgNo |
e5f8cb7b |
325 | | 32 => Option.map MsgQuery (recvQuery bio) |
9f063ecc |
326 | | 33 => Option.map MsgSocket (recvSockPerm bio) |
5514f5b5 |
327 | | 34 => Option.map MsgFirewall (recvList OpenSSL.readString bio) |
b4f5896d |
328 | | 35 => SOME MsgRegenerateTc |
a71b2f2b |
329 | | 36 => (case (OpenSSL.readString bio, OpenSSL.readString bio) of |
330 | (SOME dbtype, SOME dbname) => |
331 | SOME (MsgDropDb {dbtype = dbtype, dbname = dbname}) |
332 | | _ => NONE) |
d330d9b8 |
333 | | _ => NONE) |
334 | |
335 | end |