1 (* HCoop
Domtool (http
://hcoop
.sourceforge
.net
/)
2 * Copyright (c
) 2006-2007, Adam Chlipala
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
.
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
.
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
.
21 structure Main
:> MAIN
= struct
23 open Ast MsgTypes Print
25 structure SM
= StringMap
27 fun init () = Acl
.read Config
.aclFile
31 val prog
= Parse
.parse fname
33 if !ErrorMsg
.anyErrors
then
36 (Option
.app (Unused
.check G
) (#
3 prog
);
37 Tycheck
.checkFile
G (Defaults
.tInit ()) prog
)
42 val dir
= Posix
.FileSys
.opendir Config
.libRoot
45 case Posix
.FileSys
.readdir dir
of
46 NONE
=> (Posix
.FileSys
.closedir dir
;
49 if String.isSuffix
".dtl" fname
then
50 loop (OS
.Path
.joinDirFile
{dir
= Config
.libRoot
,
57 val (_
, files
) = Order
.order NONE files
59 if !ErrorMsg
.anyErrors
then
62 (Tycheck
.allowExterns ();
63 foldl (fn (fname
, G
) => check
' G fname
) Env
.empty files
64 before Tycheck
.disallowExterns ())
69 val _
= ErrorMsg
.reset ()
70 val _
= Env
.preTycheck ()
74 if !ErrorMsg
.anyErrors
then
78 val _
= Tycheck
.disallowExterns ()
79 val _
= ErrorMsg
.reset ()
80 val prog
= Parse
.parse fname
82 if !ErrorMsg
.anyErrors
then
86 val G
' = Tycheck
.checkFile
b (Defaults
.tInit ()) prog
88 if !ErrorMsg
.anyErrors
then
91 (Option
.app (Unused
.check b
) (#
3 prog
);
98 String.sub (s
, 0) <> #
"."
99 andalso CharVector
.all (fn ch
=> Char.isAlphaNum ch
orelse ch
= #
"." orelse ch
= #
"_" orelse ch
= #
"-") s
104 case Posix
.ProcEnv
.getenv
"DOMTOOL_USER" of
107 val uid
= Posix
.ProcEnv
.getuid ()
109 Posix
.SysDB
.Passwd
.name (Posix
.SysDB
.getpwuid uid
)
113 Acl
.read Config
.aclFile
;
118 fun checkDir
' dname
=
122 val dir
= Posix
.FileSys
.opendir dname
125 case Posix
.FileSys
.readdir dir
of
126 NONE
=> (Posix
.FileSys
.closedir dir
;
130 loop (OS
.Path
.joinDirFile
{dir
= dname
,
137 val (_
, files
) = Order
.order (SOME b
) files
139 if !ErrorMsg
.anyErrors
then
142 (foldl (fn (fname
, G
) => check
' G fname
) b files
;
143 if !ErrorMsg
.anyErrors
then
155 val (G
, body
) = check fname
157 if !ErrorMsg
.anyErrors
then
163 val body
' = Reduce
.reduceExp G body
165 (*printd (PD
.hovBox (PD
.PPS
.Rel
0,
166 [PD
.string "Result:",
177 if !ErrorMsg
.anyErrors
then
180 Eval
.exec (Defaults
.eInit ()) body
'
186 if !ErrorMsg
.anyErrors
then
189 ignore (Eval
.exec
' (Defaults
.eInit ()) body
')
193 Config
.dispatcher ^
":" ^
Int.toString Config
.dispatcherPort
196 "localhost:" ^
Int.toString Config
.slavePort
199 (OpenSSL
.context
false x
)
200 handle e
as OpenSSL
.OpenSSL s
=>
201 (print
"Couldn't find your certificate.\nYou probably haven't been given any Domtool privileges.\n";
202 print ("I looked in: " ^ #
1 x ^
"\n");
203 print ("Additional information: " ^ s ^
"\n");
206 fun requestContext f
=
208 val user
= setupUser ()
212 val context
= context (Config
.certDir ^
"/" ^ user ^
".pem",
213 Config
.keyDir ^
"/" ^ user ^
"/key.pem",
221 val (user
, context
) = requestContext f
223 (user
, OpenSSL
.connect (context
, dispatcher
))
226 fun requestSlaveBio () =
228 val (user
, context
) = requestContext (fn () => ())
230 (user
, OpenSSL
.connect (context
, self
))
235 val (user
, bio
) = requestBio (fn () => ignore (check fname
))
237 val inf
= TextIO.openIn fname
240 case TextIO.inputLine inf
of
241 NONE
=> String.concat (List.rev lines
)
242 | SOME line
=> loop (line
:: lines
)
247 Msg
.send (bio
, MsgConfig code
);
249 NONE
=> print
"Server closed connection unexpectedly.\n"
252 MsgOk
=> print
"Configuration succeeded.\n"
253 | MsgError s
=> print ("Configuration failed: " ^ s ^
"\n")
254 | _
=> print
"Unexpected server reply.\n";
257 handle ErrorMsg
.Error
=> ()
259 fun requestDir dname
=
261 val _
= if Posix
.FileSys
.access (dname
, []) then
264 (print ("Can't access " ^ dname ^
".\n");
265 print
"Did you mean to run domtool on a specific file, instead of asking for all\n";
266 print
"files in your ~/.domtool directory?\n";
267 OS
.Process
.exit OS
.Process
.failure
)
269 val _
= ErrorMsg
.reset ()
271 val (user
, bio
) = requestBio (fn () => checkDir
' dname
)
275 val dir
= Posix
.FileSys
.opendir dname
278 case Posix
.FileSys
.readdir dir
of
279 NONE
=> (Posix
.FileSys
.closedir dir
;
283 loop (OS
.Path
.joinDirFile
{dir
= dname
,
290 val (_
, files
) = Order
.order (SOME b
) files
292 val _
= if !ErrorMsg
.anyErrors
then
293 (print
"J\n";raise ErrorMsg
.Error
)
297 val codes
= map (fn fname
=>
299 val inf
= TextIO.openIn fname
302 case TextIO.inputLine inf
of
303 NONE
=> String.concat (rev lines
)
304 | SOME line
=> loop (line
:: lines
)
307 before TextIO.closeIn inf
310 if !ErrorMsg
.anyErrors
then
313 (Msg
.send (bio
, MsgMultiConfig codes
);
315 NONE
=> print
"Server closed connection unexpectedly.\n"
318 MsgOk
=> print
"Configuration succeeded.\n"
319 | MsgError s
=> print ("Configuration failed: " ^ s ^
"\n")
320 | _
=> print
"Unexpected server reply.\n";
323 handle ErrorMsg
.Error
=> ()
327 val (_
, bio
) = requestBio (fn () => ())
332 handle _
=> OS
.Process
.failure
334 fun requestShutdown () =
336 val (_
, bio
) = requestBio (fn () => ())
338 Msg
.send (bio
, MsgShutdown
);
340 NONE
=> print
"Server closed connection unexpectedly.\n"
343 MsgOk
=> print
"Shutdown begun.\n"
344 | MsgError s
=> print ("Shutdown failed: " ^ s ^
"\n")
345 | _
=> print
"Unexpected server reply.\n";
349 fun requestSlavePing () =
351 val (_
, bio
) = requestSlaveBio ()
356 handle _
=> OS
.Process
.failure
358 fun requestSlaveShutdown () =
360 val (_
, bio
) = requestSlaveBio ()
362 Msg
.send (bio
, MsgShutdown
);
364 NONE
=> print
"Server closed connection unexpectedly.\n"
367 MsgOk
=> print
"Shutdown begun.\n"
368 | MsgError s
=> print ("Shutdown failed: " ^ s ^
"\n")
369 | _
=> print
"Unexpected server reply.\n";
373 fun requestGrant acl
=
375 val (user
, bio
) = requestBio (fn () => ())
377 Msg
.send (bio
, MsgGrant acl
);
379 NONE
=> print
"Server closed connection unexpectedly.\n"
382 MsgOk
=> print
"Grant succeeded.\n"
383 | MsgError s
=> print ("Grant failed: " ^ s ^
"\n")
384 | _
=> print
"Unexpected server reply.\n";
388 fun requestRevoke acl
=
390 val (user
, bio
) = requestBio (fn () => ())
392 Msg
.send (bio
, MsgRevoke acl
);
394 NONE
=> print
"Server closed connection unexpectedly.\n"
397 MsgOk
=> print
"Revoke succeeded.\n"
398 | MsgError s
=> print ("Revoke failed: " ^ s ^
"\n")
399 | _
=> print
"Unexpected server reply.\n";
403 fun requestListPerms user
=
405 val (_
, bio
) = requestBio (fn () => ())
407 Msg
.send (bio
, MsgListPerms user
);
408 (case Msg
.recv bio
of
409 NONE
=> (print
"Server closed connection unexpectedly.\n";
413 MsgPerms perms
=> SOME perms
414 | MsgError s
=> (print ("Listing failed: " ^ s ^
"\n");
416 | _
=> (print
"Unexpected server reply.\n";
418 before OpenSSL
.close bio
421 fun requestWhoHas perm
=
423 val (_
, bio
) = requestBio (fn () => ())
425 Msg
.send (bio
, MsgWhoHas perm
);
426 (case Msg
.recv bio
of
427 NONE
=> (print
"Server closed connection unexpectedly.\n";
431 MsgWhoHasResponse users
=> SOME users
432 | MsgError s
=> (print ("whohas failed: " ^ s ^
"\n");
434 | _
=> (print
"Unexpected server reply.\n";
436 before OpenSSL
.close bio
439 fun requestRegen () =
441 val (_
, bio
) = requestBio (fn () => ())
443 Msg
.send (bio
, MsgRegenerate
);
445 NONE
=> print
"Server closed connection unexpectedly.\n"
448 MsgOk
=> print
"Regeneration succeeded.\n"
449 | MsgError s
=> print ("Regeneration failed: " ^ s ^
"\n")
450 | _
=> print
"Unexpected server reply.\n";
454 fun requestRegenTc () =
456 val (_
, bio
) = requestBio (fn () => ())
458 Msg
.send (bio
, MsgRegenerateTc
);
460 NONE
=> print
"Server closed connection unexpectedly.\n"
463 MsgOk
=> print
"All configuration validated.\n"
464 | MsgError s
=> print ("Configuration validation failed: " ^ s ^
"\n")
465 | _
=> print
"Unexpected server reply.\n";
469 fun requestRmdom dom
=
471 val (_
, bio
) = requestBio (fn () => ())
473 Msg
.send (bio
, MsgRmdom dom
);
475 NONE
=> print
"Server closed connection unexpectedly.\n"
478 MsgOk
=> print
"Removal succeeded.\n"
479 | MsgError s
=> print ("Removal failed: " ^ s ^
"\n")
480 | _
=> print
"Unexpected server reply.\n";
484 fun requestRmuser user
=
486 val (_
, bio
) = requestBio (fn () => ())
488 Msg
.send (bio
, MsgRmuser user
);
490 NONE
=> print
"Server closed connection unexpectedly.\n"
493 MsgOk
=> print
"Removal succeeded.\n"
494 | MsgError s
=> print ("Removal failed: " ^ s ^
"\n")
495 | _
=> print
"Unexpected server reply.\n";
499 fun requestDbUser dbtype
=
501 val (_
, bio
) = requestBio (fn () => ())
503 Msg
.send (bio
, MsgCreateDbUser dbtype
);
505 NONE
=> print
"Server closed connection unexpectedly.\n"
508 MsgOk
=> print
"Your user has been created.\n"
509 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
510 | _
=> print
"Unexpected server reply.\n";
514 fun requestDbPasswd rc
=
516 val (_
, bio
) = requestBio (fn () => ())
518 Msg
.send (bio
, MsgDbPasswd rc
);
520 NONE
=> print
"Server closed connection unexpectedly.\n"
523 MsgOk
=> print
"Your password has been changed.\n"
524 | MsgError s
=> print ("Password set failed: " ^ s ^
"\n")
525 | _
=> print
"Unexpected server reply.\n";
529 fun requestDbTable p
=
531 val (user
, bio
) = requestBio (fn () => ())
533 Msg
.send (bio
, MsgCreateDb p
);
535 NONE
=> print
"Server closed connection unexpectedly.\n"
538 MsgOk
=> print ("Your database " ^ user ^
"_" ^ #dbname p ^
" has been created.\n")
539 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
540 | _
=> print
"Unexpected server reply.\n";
544 fun requestDbDrop p
=
546 val (user
, bio
) = requestBio (fn () => ())
548 Msg
.send (bio
, MsgDropDb p
);
550 NONE
=> print
"Server closed connection unexpectedly.\n"
553 MsgOk
=> print ("Your database " ^ user ^
"_" ^ #dbname p ^
" has been dropped.\n")
554 | MsgError s
=> print ("Drop failed: " ^ s ^
"\n")
555 | _
=> print
"Unexpected server reply.\n";
559 fun requestDbGrant p
=
561 val (user
, bio
) = requestBio (fn () => ())
563 Msg
.send (bio
, MsgGrantDb p
);
565 NONE
=> print
"Server closed connection unexpectedly.\n"
568 MsgOk
=> print ("You've been granted all allowed privileges to database " ^ user ^
"_" ^ #dbname p ^
".\n")
569 | MsgError s
=> print ("Grant failed: " ^ s ^
"\n")
570 | _
=> print
"Unexpected server reply.\n";
574 fun requestListMailboxes domain
=
576 val (_
, bio
) = requestBio (fn () => ())
578 Msg
.send (bio
, MsgListMailboxes domain
);
579 (case Msg
.recv bio
of
580 NONE
=> Vmail
.Error
"Server closed connection unexpectedly."
583 MsgMailboxes users
=> (Msg
.send (bio
, MsgOk
);
585 | MsgError s
=> Vmail
.Error ("Listing failed: " ^ s
)
586 | _
=> Vmail
.Error
"Unexpected server reply.")
587 before OpenSSL
.close bio
590 fun requestNewMailbox p
=
592 val (_
, bio
) = requestBio (fn () => ())
594 Msg
.send (bio
, MsgNewMailbox p
);
596 NONE
=> print
"Server closed connection unexpectedly.\n"
599 MsgOk
=> print ("A mapping for " ^ #user p ^
"@" ^ #domain p ^
" has been created.\n")
600 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
601 | _
=> print
"Unexpected server reply.\n";
605 fun requestPasswdMailbox p
=
607 val (_
, bio
) = requestBio (fn () => ())
609 Msg
.send (bio
, MsgPasswdMailbox p
);
611 NONE
=> print
"Server closed connection unexpectedly.\n"
614 MsgOk
=> print ("The password for " ^ #user p ^
"@" ^ #domain p ^
" has been changed.\n")
615 | MsgError s
=> print ("Set failed: " ^ s ^
"\n")
616 | _
=> print
"Unexpected server reply.\n";
620 fun requestRmMailbox p
=
622 val (_
, bio
) = requestBio (fn () => ())
624 Msg
.send (bio
, MsgRmMailbox p
);
626 NONE
=> print
"Server closed connection unexpectedly.\n"
629 MsgOk
=> print ("The mapping for mailbox " ^ #user p ^
"@" ^ #domain p ^
" has been deleted.\n")
630 | MsgError s
=> print ("Remove failed: " ^ s ^
"\n")
631 | _
=> print
"Unexpected server reply.\n";
635 fun requestSaQuery addr
=
637 val (_
, bio
) = requestBio (fn () => ())
639 Msg
.send (bio
, MsgSaQuery addr
);
640 (case Msg
.recv bio
of
641 NONE
=> print
"Server closed connection unexpectedly.\n"
644 MsgSaStatus b
=> (print ("SpamAssassin filtering for " ^ addr ^
" is "
645 ^
(if b
then "ON" else "OFF") ^
".\n");
646 Msg
.send (bio
, MsgOk
))
647 | MsgError s
=> print ("Query failed: " ^ s ^
"\n")
648 | _
=> print
"Unexpected server reply.\n")
649 before OpenSSL
.close bio
654 val (_
, bio
) = requestBio (fn () => ())
656 Msg
.send (bio
, MsgSaSet p
);
658 NONE
=> print
"Server closed connection unexpectedly.\n"
661 MsgOk
=> print ("SpamAssassin filtering for " ^ #
1 p ^
" is now "
662 ^
(if #
2 p
then "ON" else "OFF") ^
".\n")
663 | MsgError s
=> print ("Set failed: " ^ s ^
"\n")
664 | _
=> print
"Unexpected server reply.\n";
668 fun requestSmtpLog domain
=
670 val (_
, bio
) = requestBio (fn () => ())
672 val _
= Msg
.send (bio
, MsgSmtpLogReq domain
)
676 NONE
=> print
"Server closed connection unexpectedly.\n"
680 | MsgSmtpLogRes line
=> (print line
;
682 | MsgError s
=> print ("Log search failed: " ^ s ^
"\n")
683 | _
=> print
"Unexpected server reply.\n"
689 fun requestMysqlFixperms () =
691 val (_
, bio
) = requestBio (fn () => ())
693 Msg
.send (bio
, MsgMysqlFixperms
);
695 NONE
=> print
"Server closed connection unexpectedly.\n"
698 MsgOk
=> print
"Permissions granted.\n"
699 | MsgError s
=> print ("Failed: " ^ s ^
"\n")
700 | _
=> print
"Unexpected server reply.\n";
704 fun requestApt
{node
, pkg
} =
706 val (user
, context
) = requestContext (fn () => ())
707 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
710 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
712 val _
= Msg
.send (bio
, MsgQuery (QApt pkg
))
716 NONE
=> (print
"Server closed connection unexpectedly.\n";
720 MsgYes
=> (print
"Package is installed.\n";
722 | MsgNo
=> (print
"Package is not installed.\n";
724 | MsgError s
=> (print ("APT query failed: " ^ s ^
"\n");
726 | _
=> (print
"Unexpected server reply.\n";
730 before OpenSSL
.close bio
733 fun requestCron
{node
, uname
} =
735 val (user
, context
) = requestContext (fn () => ())
736 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
739 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
741 val _
= Msg
.send (bio
, MsgQuery (QCron uname
))
745 NONE
=> (print
"Server closed connection unexpectedly.\n";
749 MsgYes
=> (print
"User has cron permissions.\n";
751 | MsgNo
=> (print
"User does not have cron permissions.\n";
753 | MsgError s
=> (print ("Cron query failed: " ^ s ^
"\n");
755 | _
=> (print
"Unexpected server reply.\n";
759 before OpenSSL
.close bio
762 fun requestFtp
{node
, uname
} =
764 val (user
, context
) = requestContext (fn () => ())
765 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
768 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
770 val _
= Msg
.send (bio
, MsgQuery (QFtp uname
))
774 NONE
=> (print
"Server closed connection unexpectedly.\n";
778 MsgYes
=> (print
"User has FTP permissions.\n";
780 | MsgNo
=> (print
"User does not have FTP permissions.\n";
782 | MsgError s
=> (print ("FTP query failed: " ^ s ^
"\n");
784 | _
=> (print
"Unexpected server reply.\n";
788 before OpenSSL
.close bio
791 fun requestTrustedPath
{node
, uname
} =
793 val (user
, context
) = requestContext (fn () => ())
794 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
797 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
799 val _
= Msg
.send (bio
, MsgQuery (QTrustedPath uname
))
803 NONE
=> (print
"Server closed connection unexpectedly.\n";
807 MsgYes
=> (print
"User has trusted path restriction.\n";
809 | MsgNo
=> (print
"User does not have trusted path restriction.\n";
811 | MsgError s
=> (print ("Trusted path query failed: " ^ s ^
"\n");
813 | _
=> (print
"Unexpected server reply.\n";
817 before OpenSSL
.close bio
820 fun requestSocketPerm
{node
, uname
} =
822 val (user
, context
) = requestContext (fn () => ())
823 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
826 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
828 val _
= Msg
.send (bio
, MsgQuery (QSocket uname
))
832 NONE
=> (print
"Server closed connection unexpectedly.\n";
836 MsgSocket p
=> (case p
of
838 | Client
=> print
"Client\n"
839 | Server
=> print
"Server\n"
840 | Nada
=> print
"Nada\n";
842 | MsgError s
=> (print ("Socket permission query failed: " ^ s ^
"\n");
844 | _
=> (print
"Unexpected server reply.\n";
848 before OpenSSL
.close bio
851 fun requestFirewall
{node
, uname
} =
853 val (user
, context
) = requestContext (fn () => ())
854 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
857 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
859 val _
= Msg
.send (bio
, MsgQuery (QFirewall uname
))
863 NONE
=> (print
"Server closed connection unexpectedly.\n";
867 MsgFirewall ls
=> (app (fn s
=> (print s
; print
"\n")) ls
;
869 | MsgError s
=> (print ("Firewall query failed: " ^ s ^
"\n");
871 | _
=> (print
"Unexpected server reply.\n";
875 before OpenSSL
.close bio
878 fun requestDescribe dom
=
880 val (_
, bio
) = requestBio (fn () => ())
882 Msg
.send (bio
, MsgDescribe dom
);
884 NONE
=> print
"Server closed connection unexpectedly.\n"
887 MsgDescription s
=> print s
888 | MsgError s
=> print ("Description failed: " ^ s ^
"\n")
889 | _
=> print
"Unexpected server reply.\n";
893 structure SS
= StringSet
895 fun domainList dname
=
897 val dir
= Posix
.FileSys
.opendir dname
900 case Posix
.FileSys
.readdir dir
of
904 val path
= OS
.Path
.joinDirFile
{dir
= dname
,
907 fun visitDomains (path
, bfor
, dset
) =
909 val dir
= Posix
.FileSys
.opendir path
912 case Posix
.FileSys
.readdir dir
of
916 val path
= OS
.Path
.joinDirFile
{dir
= path
,
919 if Posix
.FileSys
.ST
.isDir (Posix
.FileSys
.stat path
) then
921 val bfor
= dname
:: bfor
923 loop (visitDomains (path
, bfor
,
925 String.concatWith
"." bfor
)))
932 before Posix
.FileSys
.closedir dir
935 visitNode (visitDomains (path
, [], dset
))
939 before Posix
.FileSys
.closedir dir
942 fun regenerateEither tc checker context
=
944 val () = print
"Starting regeneration....\n"
950 domainList Config
.resultRoot
958 val _
= ErrorMsg
.reset ()
961 val () = Tycheck
.disallowExterns ()
963 val () = ifReal (fn () =>
964 (ignore (OS
.Process
.system ("rm -rf " ^ Config
.oldResultRoot ^
"/*"));
965 ignore (OS
.Process
.system ("cp -r " ^ Config
.resultRoot
966 ^
"/* " ^ Config
.oldResultRoot ^
"/"));
967 Domain
.resetGlobal ()))
971 fun contactNode (node
, ip
) =
972 if node
= Config
.defaultNode
then
975 val bio
= OpenSSL
.connect (context
,
978 ^
Int.toString Config
.slavePort
)
980 Msg
.send (bio
, MsgRegenerate
);
982 NONE
=> print
"Slave closed connection unexpectedly\n"
985 MsgOk
=> print ("Slave " ^ node ^
" pre-regeneration finished\n")
986 | MsgError s
=> print ("Slave " ^ node
987 ^
" returned error: " ^
989 | _
=> print ("Slave " ^ node
990 ^
" returned unexpected command\n");
993 handle OpenSSL
.OpenSSL s
=> print ("OpenSSL error: " ^ s ^
"\n")
997 val _
= Domain
.setUser user
998 val _
= ErrorMsg
.reset ()
1000 val dname
= Config
.domtoolDir user
1002 if Posix
.FileSys
.access (dname
, []) then
1004 val dir
= Posix
.FileSys
.opendir dname
1007 case Posix
.FileSys
.readdir dir
of
1008 NONE
=> (Posix
.FileSys
.closedir dir
;
1011 if notTmp fname
then
1012 loop (OS
.Path
.joinDirFile
{dir
= dname
,
1019 val (_
, files
) = Order
.order (SOME b
) files
1021 if !ErrorMsg
.anyErrors
then
1023 print ("User " ^ user ^
"'s configuration has errors!\n");
1032 handle IO
.Io
{name
, function
, ...} =>
1033 (print ("IO error processing user " ^ user ^
": " ^ function ^
": " ^ name ^
"\n");
1035 | exn
as OS
.SysErr (s
, _
) => (print ("System error processing user " ^ user ^
": " ^ s ^
"\n");
1037 | ErrorMsg
.Error
=> (ErrorMsg
.reset ();
1038 print ("User " ^ user ^
" had a compilation error.\n");
1040 | _
=> (print
"Unknown exception during regeneration!\n";
1043 ifReal (fn () => (app contactNode Config
.nodeIps
;
1045 app
doUser (Acl
.users ());
1048 val domainsAfter
= domainList Config
.resultRoot
1049 val domainsGone
= SS
.difference (domainsBefore
, domainsAfter
)
1051 if SS
.isEmpty domainsGone
then
1054 (print
"Domains to kill:";
1055 SS
.app (fn s
=> (print
" "; print s
)) domainsGone
;
1058 Domain
.rmdom
' Config
.oldResultRoot (SS
.listItems domainsGone
));
1065 val regenerate
= regenerateEither
false eval
'
1066 val regenerateTc
= regenerateEither
true (ignore
o check
)
1070 val doms
= Acl
.class
{user
= user
, class
= "domain"}
1071 val doms
= List.filter (fn dom
=>
1072 case Acl
.whoHas
{class
= "domain", value
= dom
} of
1074 | _
=> false) (StringSet
.listItems doms
)
1080 fun now () = Date
.toString (Date
.fromTimeUniv (Time
.now ()))
1084 QApt pkg
=> if Apt
.installed pkg
then MsgYes
else MsgNo
1085 | QCron user
=> if Cron
.allowed user
then MsgYes
else MsgNo
1086 | QFtp user
=> if Ftp
.allowed user
then MsgYes
else MsgNo
1087 | QTrustedPath user
=> if TrustedPath
.query user
then MsgYes
else MsgNo
1088 | QSocket user
=> MsgSocket (SocketPerm
.query user
)
1089 | QFirewall user
=> MsgFirewall (Firewall
.query user
)
1091 fun describeQuery q
=
1093 QApt pkg
=> "Requested installation status of package " ^ pkg
1094 | QCron user
=> "Asked about cron permissions for user " ^ user
1095 | QFtp user
=> "Asked about FTP permissions for user " ^ user
1096 | QTrustedPath user
=> "Asked about trusted path settings for user " ^ user
1097 | QSocket user
=> "Asked about socket permissions for user " ^ user
1098 | QFirewall user
=> "Asked about firewall rules for user " ^ user
1102 val () = Acl
.read Config
.aclFile
1104 val context
= context (Config
.serverCert
,
1107 val _
= Domain
.set_context context
1109 val sock
= OpenSSL
.listen (context
, Config
.dispatcherPort
)
1112 (case OpenSSL
.accept sock
of
1116 val user
= OpenSSL
.peerCN bio
1117 val () = print ("\nConnection from " ^ user ^
" at " ^
now () ^
"\n")
1118 val () = Domain
.setUser user
1120 fun doIt f cleanup
=
1122 (msgLocal
, SOME msgRemote
) =>
1125 Msg
.send (bio
, MsgError msgRemote
))
1126 |
(msgLocal
, NONE
) =>
1129 Msg
.send (bio
, MsgOk
)))
1130 handle e
as (OpenSSL
.OpenSSL s
) =>
1131 (print ("OpenSSL error: " ^ s ^
"\n");
1132 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1133 Msg
.send (bio
, MsgError ("OpenSSL error: " ^ s
))
1134 handle OpenSSL
.OpenSSL _
=> ())
1135 | OS
.SysErr (s
, _
) =>
1136 (print
"System error: ";
1139 Msg
.send (bio
, MsgError ("System error: " ^ s
))
1140 handle OpenSSL
.OpenSSL _
=> ())
1145 Msg
.send (bio
, MsgError ("Failure: " ^ s
))
1146 handle OpenSSL
.OpenSSL _
=> ())
1148 (print
"Compilation error\n";
1149 Msg
.send (bio
, MsgError
"Error during configuration evaluation")
1150 handle OpenSSL
.OpenSSL _
=> ());
1152 ignore (OpenSSL
.readChar bio
);
1154 handle OpenSSL
.OpenSSL _
=> ();
1157 fun doConfig codes
=
1159 val _
= print
"Configuration:\n"
1160 val _
= app (fn s
=> (print s
; print
"\n")) codes
1163 val outname
= OS
.FileSys
.tmpName ()
1167 val outf
= TextIO.openOut outname
1169 TextIO.output (outf
, code
);
1170 TextIO.closeOut outf
;
1174 doIt (fn () => (Env
.pre ();
1177 Msg
.send (bio
, MsgOk
);
1178 ("Configuration complete.", NONE
)))
1179 (fn () => OS
.FileSys
.remove outname
)
1183 case String.fields (fn ch
=> ch
= #
"@") s
of
1185 if user
= user
' then
1189 |
[user
', domain
] =>
1190 if Domain
.validEmailUser user
' andalso Domain
.yourDomain domain
then
1191 SOME (SetSA
.Email s
)
1197 case Msg
.recv bio
of
1198 NONE
=> (OpenSSL
.close bio
1199 handle OpenSSL
.OpenSSL _
=> ();
1203 MsgConfig code
=> doConfig
[code
]
1204 | MsgMultiConfig codes
=> doConfig codes
1207 if Acl
.query
{user
= user
, class
= "priv", value
= "all"}
1208 orelse Acl
.query
{user
= user
, class
= "priv", value
= "shutdown"} then
1209 print ("Domtool dispatcher shutting down at " ^
now () ^
"\n\n")
1211 (print
"Unauthorized shutdown command!\n";
1213 handle OpenSSL
.OpenSSL _
=> ();
1218 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1220 Acl
.write Config
.aclFile
;
1221 ("Granted permission " ^ #value acl ^
" to " ^ #user acl ^
" in " ^ #class acl ^
".",
1224 ("Unauthorized user asked to grant a permission!",
1225 SOME
"Not authorized to grant privileges"))
1230 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1232 Acl
.write Config
.aclFile
;
1233 ("Revoked permission " ^ #value acl ^
" from " ^ #user acl ^
" in " ^ #class acl ^
".",
1236 ("Unauthorized user asked to revoke a permission!",
1237 SOME
"Not authorized to revoke privileges"))
1240 | MsgListPerms user
=>
1242 (Msg
.send (bio
, MsgPerms (Acl
.queryAll user
));
1243 ("Sent permission list for user " ^ user ^
".",
1249 (Msg
.send (bio
, MsgWhoHasResponse (Acl
.whoHas perm
));
1250 ("Sent whohas response for " ^ #class perm ^
" / " ^ #value perm ^
".",
1256 if Acl
.query
{user
= user
, class
= "priv", value
= "all"}
1257 orelse List.all (fn dom
=> Acl
.query
{user
= user
, class
= "domain", value
= dom
}) doms
then
1260 Acl
.revokeFromAll
{class
= "domain", value
= dom
}) doms
;
1261 Acl
.write Config
.aclFile
;
1262 ("Removed domains" ^
foldl (fn (d
, s
) => s ^
" " ^ d
) "" doms ^
".",
1265 ("Unauthorized user asked to remove a domain!",
1266 SOME
"Not authorized to remove that domain"))
1271 if Acl
.query
{user
= user
, class
= "priv", value
= "regen"}
1272 orelse Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1273 (if regenerate context
then
1274 ("Regenerated all configuration.",
1277 ("Error regenerating configuration!",
1278 SOME
"Error regenerating configuration! Consult /var/log/domtool.log."))
1280 ("Unauthorized user asked to regenerate!",
1281 SOME
"Not authorized to regenerate"))
1284 | MsgRegenerateTc
=>
1286 if Acl
.query
{user
= user
, class
= "priv", value
= "regen"}
1287 orelse Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1288 (if regenerateTc context
then
1289 ("Checked all configuration.",
1292 ("Found a compilation error!",
1293 SOME
"Found a compilation error! Consult /var/log/domtool.log."))
1295 ("Unauthorized user asked to regenerate -tc!",
1296 SOME
"Not authorized to regenerate -tc"))
1299 | MsgRmuser user
' =>
1301 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1303 Acl
.write Config
.aclFile
;
1304 ("Removed user " ^ user
' ^
".",
1307 ("Unauthorized user asked to remove a user!",
1308 SOME
"Not authorized to remove users"))
1311 | MsgCreateDbUser
{dbtype
, passwd
} =>
1313 case Dbms
.lookup dbtype
of
1314 NONE
=> ("Database user creation request with unknown datatype type " ^ dbtype
,
1315 SOME ("Unknown database type " ^ dbtype
))
1317 case #adduser handler
{user
= user
, passwd
= passwd
} of
1318 NONE
=> ("Added " ^ dbtype ^
" user " ^ user ^
".",
1321 ("Error adding a " ^ dbtype ^
" user " ^ user ^
": " ^ msg
,
1322 SOME ("Error adding user: " ^ msg
)))
1325 | MsgDbPasswd
{dbtype
, passwd
} =>
1327 case Dbms
.lookup dbtype
of
1328 NONE
=> ("Database passwd request with unknown datatype type " ^ dbtype
,
1329 SOME ("Unknown database type " ^ dbtype
))
1331 case #passwd handler
{user
= user
, passwd
= passwd
} of
1332 NONE
=> ("Changed " ^ dbtype ^
" password of user " ^ user ^
".",
1335 ("Error setting " ^ dbtype ^
" password of user " ^ user ^
": " ^ msg
,
1336 SOME ("Error adding user: " ^ msg
)))
1339 | MsgCreateDb
{dbtype
, dbname
} =>
1341 if Dbms
.validDbname dbname
then
1342 case Dbms
.lookup dbtype
of
1343 NONE
=> ("Database creation request with unknown datatype type " ^ dbtype
,
1344 SOME ("Unknown database type " ^ dbtype
))
1346 case #createdb handler
{user
= user
, dbname
= dbname
} of
1347 NONE
=> ("Created database " ^ user ^
"_" ^ dbname ^
".",
1349 | SOME msg
=> ("Error creating database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1350 SOME ("Error creating database: " ^ msg
))
1352 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1353 SOME ("Invalid database name " ^ dbname
)))
1356 | MsgDropDb
{dbtype
, dbname
} =>
1358 if Dbms
.validDbname dbname
then
1359 case Dbms
.lookup dbtype
of
1360 NONE
=> ("Database drop request with unknown datatype type " ^ dbtype
,
1361 SOME ("Unknown database type " ^ dbtype
))
1363 case #dropdb handler
{user
= user
, dbname
= dbname
} of
1364 NONE
=> ("Drop database " ^ user ^
"_" ^ dbname ^
".",
1366 | SOME msg
=> ("Error dropping database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1367 SOME ("Error dropping database: " ^ msg
))
1369 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1370 SOME ("Invalid database name " ^ dbname
)))
1373 | MsgGrantDb
{dbtype
, dbname
} =>
1375 if Dbms
.validDbname dbname
then
1376 case Dbms
.lookup dbtype
of
1377 NONE
=> ("Database drop request with unknown datatype type " ^ dbtype
,
1378 SOME ("Unknown database type " ^ dbtype
))
1380 case #grant handler
{user
= user
, dbname
= dbname
} of
1381 NONE
=> ("Grant permissions to database " ^ user ^
"_" ^ dbname ^
".",
1383 | SOME msg
=> ("Error granting permissions to database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1384 SOME ("Error granting permissions to database: " ^ msg
))
1386 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1387 SOME ("Invalid database name " ^ dbname
)))
1390 | MsgListMailboxes domain
=>
1392 if not (Domain
.yourDomain domain
) then
1393 ("User wasn't authorized to list mailboxes for " ^ domain
,
1394 SOME
"You're not authorized to configure that domain.")
1396 case Vmail
.list domain
of
1397 Vmail
.Listing users
=> (Msg
.send (bio
, MsgMailboxes users
);
1398 ("Sent mailbox list for " ^ domain
,
1400 | Vmail
.Error msg
=> ("Error listing mailboxes for " ^ domain ^
": " ^ msg
,
1404 | MsgNewMailbox
{domain
, user
= emailUser
, passwd
, mailbox
} =>
1406 if not (Domain
.yourDomain domain
) then
1407 ("User wasn't authorized to add a mailbox to " ^ domain
,
1408 SOME
"You're not authorized to configure that domain.")
1409 else if not (Domain
.validEmailUser emailUser
) then
1410 ("Invalid e-mail username " ^ emailUser
,
1411 SOME
"Invalid e-mail username")
1412 else if not (CharVector
.all
Char.isGraph passwd
) then
1413 ("Invalid password",
1414 SOME
"Invalid password; may only contain printable, non-space characters")
1415 else if not (Domain
.yourPath mailbox
) then
1416 ("User wasn't authorized to add a mailbox at " ^ mailbox
,
1417 SOME ("You're not authorized to use that mailbox location. ("
1420 case Vmail
.add
{requester
= user
,
1421 domain
= domain
, user
= emailUser
,
1422 passwd
= passwd
, mailbox
= mailbox
} of
1423 NONE
=> ("Added mailbox " ^ emailUser ^
"@" ^ domain ^
" at " ^ mailbox
,
1425 | SOME msg
=> ("Error adding mailbox " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1429 | MsgPasswdMailbox
{domain
, user
= emailUser
, passwd
} =>
1431 if not (Domain
.yourDomain domain
) then
1432 ("User wasn't authorized to change password of a mailbox for " ^ domain
,
1433 SOME
"You're not authorized to configure that domain.")
1434 else if not (Domain
.validEmailUser emailUser
) then
1435 ("Invalid e-mail username " ^ emailUser
,
1436 SOME
"Invalid e-mail username")
1437 else if not (CharVector
.all
Char.isGraph passwd
) then
1438 ("Invalid password",
1439 SOME
"Invalid password; may only contain printable, non-space characters")
1441 case Vmail
.passwd
{domain
= domain
, user
= emailUser
,
1443 NONE
=> ("Changed password of mailbox " ^ emailUser ^
"@" ^ domain
,
1445 | SOME msg
=> ("Error changing mailbox password for " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1449 | MsgRmMailbox
{domain
, user
= emailUser
} =>
1451 if not (Domain
.yourDomain domain
) then
1452 ("User wasn't authorized to change password of a mailbox for " ^ domain
,
1453 SOME
"You're not authorized to configure that domain.")
1454 else if not (Domain
.validEmailUser emailUser
) then
1455 ("Invalid e-mail username " ^ emailUser
,
1456 SOME
"Invalid e-mail username")
1458 case Vmail
.rm
{domain
= domain
, user
= emailUser
} of
1459 NONE
=> ("Deleted mailbox " ^ emailUser ^
"@" ^ domain
,
1461 | SOME msg
=> ("Error deleting mailbox " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1465 | MsgSaQuery addr
=>
1467 case checkAddr addr
of
1468 NONE
=> ("User tried to query SA filtering for " ^ addr
,
1469 SOME
"You aren't allowed to configure SA filtering for that recipient.")
1470 | SOME addr
' => (Msg
.send (bio
, MsgSaStatus (SetSA
.query addr
'));
1471 ("Queried SA filtering status for " ^ addr
,
1475 |
MsgSaSet (addr
, b
) =>
1477 case checkAddr addr
of
1478 NONE
=> ("User tried to set SA filtering for " ^ addr
,
1479 SOME
"You aren't allowed to configure SA filtering for that recipient.")
1480 | SOME addr
' => (SetSA
.set (addr
', b
);
1481 Msg
.send (bio
, MsgOk
);
1482 ("Set SA filtering status for " ^ addr ^
" to "
1483 ^
(if b
then "ON" else "OFF"),
1487 | MsgSmtpLogReq domain
=>
1489 if not (Domain
.yourDomain domain
) then
1490 ("Unauthorized user tried to request SMTP logs for " ^ domain
,
1491 SOME
"You aren't authorized to configure that domain.")
1493 (SmtpLog
.search (fn line
=> Msg
.send (bio
, MsgSmtpLogRes line
))
1495 ("Requested SMTP logs for " ^ domain
,
1500 doIt (fn () => (Msg
.send (bio
, answerQuery q
);
1505 | MsgMysqlFixperms
=>
1506 doIt (fn () => if OS
.Process
.isSuccess
1507 (OS
.Process
.system
"/usr/bin/sudo -H /afs/hcoop.net/common/etc/scripts/mysql-grant-table-drop") then
1508 ("Requested mysql-fixperms",
1511 ("Requested mysql-fixperms, but execution failed!",
1512 SOME
"Script execution failed."))
1515 | MsgDescribe dom
=>
1516 doIt (fn () => if not (Domain
.validDomain dom
) then
1517 ("Requested description of invalid domain " ^ dom
,
1518 SOME
"Invalid domain name")
1519 else if not (Domain
.yourDomain dom
1520 orelse Acl
.query
{user
= user
, class
= "priv", value
= "all"}) then
1521 ("Requested description of " ^ dom ^
", but not allowed access",
1522 SOME
"Access denied")
1524 (Msg
.send (bio
, MsgDescription (Domain
.describe dom
));
1525 ("Sent description of domain " ^ dom
,
1530 doIt (fn () => ("Unexpected command",
1531 SOME
"Unexpected command"))
1536 handle e
as (OpenSSL
.OpenSSL s
) =>
1537 (print ("OpenSSL error: " ^ s ^
"\n");
1538 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1540 handle OpenSSL
.OpenSSL _
=> ();
1542 | OS
.SysErr (s
, _
) =>
1543 (print ("System error: " ^ s ^
"\n");
1545 handle OpenSSL
.OpenSSL _
=> ();
1547 | IO
.Io
{name
, function
, cause
} =>
1548 (print ("IO error: " ^ function ^
" for " ^ name ^
"\n");
1549 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory cause
);
1551 handle OpenSSL
.OpenSSL _
=> ();
1553 | OS
.Path
.InvalidArc
=>
1554 (print
"Invalid arc\n";
1556 handle OpenSSL
.OpenSSL _
=> ();
1559 (print
"Unknown exception in main loop!\n";
1560 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1562 handle OpenSSL
.OpenSSL _
=> ();
1564 handle e
as (OpenSSL
.OpenSSL s
) =>
1565 (print ("OpenSSL error: " ^ s ^
"\n");
1566 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1568 | OS
.SysErr (s
, _
) =>
1569 (print ("System error: " ^ s ^
"\n");
1571 | IO
.Io
{name
, function
, cause
} =>
1572 (print ("IO error: " ^ function ^
" for " ^ name ^
"\n");
1573 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory cause
);
1576 (print
"Unknown exception in main loop!\n";
1577 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1580 print ("Domtool dispatcher starting up at " ^
now () ^
"\n");
1581 print
"Listening for connections....\n";
1583 OpenSSL
.shutdown sock
1588 val host
= Slave
.hostname ()
1590 val context
= context (Config
.certDir ^
"/" ^ host ^
".pem",
1591 Config
.keyDir ^
"/" ^ host ^
"/key.pem",
1594 val sock
= OpenSSL
.listen (context
, Config
.slavePort
)
1596 val _
= print ("Slave server starting at " ^
now () ^
"\n")
1599 case OpenSSL
.accept sock
of
1603 val peer
= OpenSSL
.peerCN bio
1604 val () = print ("\nConnection from " ^ peer ^
" at " ^
now () ^
"\n")
1606 if peer
= Config
.dispatcherName
then let
1608 case Msg
.recv bio
of
1609 NONE
=> print
"Dispatcher closed connection unexpectedly\n"
1612 MsgFile file
=> loop
' (file
:: files
)
1613 | MsgDoFiles
=> (Slave
.handleChanges files
;
1614 Msg
.send (bio
, MsgOk
))
1615 | MsgRegenerate
=> (Domain
.resetLocal ();
1616 Msg
.send (bio
, MsgOk
))
1617 | _
=> (print
"Dispatcher sent unexpected command\n";
1618 Msg
.send (bio
, MsgError
"Unexpected command"))
1621 ignore (OpenSSL
.readChar bio
);
1625 else if peer
= "domtool" then
1626 case Msg
.recv bio
of
1627 SOME MsgShutdown
=> (OpenSSL
.close bio
;
1628 print ("Shutting down at " ^
now () ^
"\n\n"))
1629 | _
=> (OpenSSL
.close bio
;
1632 case Msg
.recv bio
of
1633 SOME (MsgQuery q
) => (print (describeQuery q ^
"\n");
1634 Msg
.send (bio
, answerQuery q
);
1635 ignore (OpenSSL
.readChar bio
);
1638 | _
=> (OpenSSL
.close bio
;
1640 end handle OpenSSL
.OpenSSL s
=>
1641 (print ("OpenSSL error: " ^ s ^
"\n");
1643 handle OpenSSL
.OpenSSL _
=> ();
1645 | e
as OS
.SysErr (s
, _
) =>
1646 (app (fn s
=> print (s ^
"\n")) (SMLofNJ
.exnHistory e
);
1647 print ("System error: "^ s ^
"\n");
1649 handle OpenSSL
.OpenSSL _
=> ();
1651 | IO
.Io
{function
, name
, ...} =>
1652 (print ("IO error: " ^ function ^
": " ^ name ^
"\n");
1654 handle OpenSSL
.OpenSSL _
=> ();
1657 (app (fn s
=> print (s ^
"\n")) (SMLofNJ
.exnHistory e
);
1658 print
"Uncaught exception!\n";
1660 handle OpenSSL
.OpenSSL _
=> ();
1664 OpenSSL
.shutdown sock
1669 val dir
= Posix
.FileSys
.opendir Config
.libRoot
1672 case Posix
.FileSys
.readdir dir
of
1673 NONE
=> (Posix
.FileSys
.closedir dir
;
1676 if String.isSuffix
".dtl" fname
then
1677 loop (OS
.Path
.joinDirFile
{dir
= Config
.libRoot
,
1686 fun autodocBasis outdir
=
1687 Autodoc
.autodoc
{outdir
= outdir
, infiles
= listBasis ()}