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 Tycheck
.checkFile
G (Defaults
.tInit ()) prog
41 val dir
= Posix
.FileSys
.opendir Config
.libRoot
44 case Posix
.FileSys
.readdir dir
of
45 NONE
=> (Posix
.FileSys
.closedir dir
;
48 if String.isSuffix
".dtl" fname
then
49 loop (OS
.Path
.joinDirFile
{dir
= Config
.libRoot
,
56 val (_
, files
) = Order
.order NONE files
58 if !ErrorMsg
.anyErrors
then
61 (Tycheck
.allowExterns ();
62 foldl (fn (fname
, G
) => check
' G fname
) Env
.empty files
63 before Tycheck
.disallowExterns ())
68 val _
= ErrorMsg
.reset ()
69 val _
= Env
.preTycheck ()
73 if !ErrorMsg
.anyErrors
then
77 val _
= Tycheck
.disallowExterns ()
78 val _
= ErrorMsg
.reset ()
79 val prog
= Parse
.parse fname
81 if !ErrorMsg
.anyErrors
then
85 val G
' = Tycheck
.checkFile
b (Defaults
.tInit ()) prog
87 if !ErrorMsg
.anyErrors
then
96 String.sub (s
, 0) <> #
"."
97 andalso CharVector
.all (fn ch
=> Char.isAlphaNum ch
orelse ch
= #
"." orelse ch
= #
"_" orelse ch
= #
"-") s
102 case Posix
.ProcEnv
.getenv
"DOMTOOL_USER" of
105 val uid
= Posix
.ProcEnv
.getuid ()
107 Posix
.SysDB
.Passwd
.name (Posix
.SysDB
.getpwuid uid
)
111 Acl
.read Config
.aclFile
;
116 fun checkDir
' dname
=
120 val dir
= Posix
.FileSys
.opendir dname
123 case Posix
.FileSys
.readdir dir
of
124 NONE
=> (Posix
.FileSys
.closedir dir
;
128 loop (OS
.Path
.joinDirFile
{dir
= dname
,
135 val (_
, files
) = Order
.order (SOME b
) files
137 if !ErrorMsg
.anyErrors
then
140 (foldl (fn (fname
, G
) => check
' G fname
) b files
;
141 if !ErrorMsg
.anyErrors
then
153 val (G
, body
) = check fname
155 if !ErrorMsg
.anyErrors
then
161 val body
' = Reduce
.reduceExp G body
163 (*printd (PD
.hovBox (PD
.PPS
.Rel
0,
164 [PD
.string "Result:",
175 if !ErrorMsg
.anyErrors
then
178 Eval
.exec (Defaults
.eInit ()) body
'
179 | NONE
=> raise ErrorMsg
.Error
184 if !ErrorMsg
.anyErrors
then
187 ignore (Eval
.exec
' (Defaults
.eInit ()) body
')
188 | NONE
=> raise ErrorMsg
.Error
191 Config
.dispatcher ^
":" ^
Int.toString Config
.dispatcherPort
194 "localhost:" ^
Int.toString Config
.slavePort
197 (OpenSSL
.context
false x
)
198 handle e
as OpenSSL
.OpenSSL s
=>
199 (print
"Couldn't find your certificate.\nYou probably haven't been given any Domtool privileges.\n";
200 print ("I looked in: " ^ #
1 x ^
"\n");
201 print ("Additional information: " ^ s ^
"\n");
204 fun requestContext f
=
206 val user
= setupUser ()
210 val context
= context (Config
.certDir ^
"/" ^ user ^
".pem",
211 Config
.keyDir ^
"/" ^ user ^
"/key.pem",
219 val (user
, context
) = requestContext f
221 (user
, OpenSSL
.connect (context
, dispatcher
))
224 fun requestSlaveBio () =
226 val (user
, context
) = requestContext (fn () => ())
228 (user
, OpenSSL
.connect (context
, self
))
233 val (user
, bio
) = requestBio (fn () => ignore (check fname
))
235 val inf
= TextIO.openIn fname
238 case TextIO.inputLine inf
of
239 NONE
=> String.concat (List.rev lines
)
240 | SOME line
=> loop (line
:: lines
)
245 Msg
.send (bio
, MsgConfig code
);
247 NONE
=> print
"Server closed connection unexpectedly.\n"
250 MsgOk
=> print
"Configuration succeeded.\n"
251 | MsgError s
=> print ("Configuration failed: " ^ s ^
"\n")
252 | _
=> print
"Unexpected server reply.\n";
255 handle ErrorMsg
.Error
=> ()
257 fun requestDir dname
=
259 val _
= if Posix
.FileSys
.access (dname
, []) then
262 (print ("Can't access " ^ dname ^
".\n");
263 print
"Did you mean to run domtool on a specific file, instead of asking for all\n";
264 print
"files in your ~/.domtool directory?\n";
265 OS
.Process
.exit OS
.Process
.failure
)
267 val _
= ErrorMsg
.reset ()
269 val (user
, bio
) = requestBio (fn () => checkDir
' dname
)
273 val dir
= Posix
.FileSys
.opendir dname
276 case Posix
.FileSys
.readdir dir
of
277 NONE
=> (Posix
.FileSys
.closedir dir
;
281 loop (OS
.Path
.joinDirFile
{dir
= dname
,
288 val (_
, files
) = Order
.order (SOME b
) files
290 val _
= if !ErrorMsg
.anyErrors
then
295 val codes
= map (fn fname
=>
297 val inf
= TextIO.openIn fname
300 case TextIO.inputLine inf
of
301 NONE
=> String.concat (rev lines
)
302 | SOME line
=> loop (line
:: lines
)
305 before TextIO.closeIn inf
308 if !ErrorMsg
.anyErrors
then
311 (Msg
.send (bio
, MsgMultiConfig codes
);
313 NONE
=> print
"Server closed connection unexpectedly.\n"
316 MsgOk
=> print
"Configuration succeeded.\n"
317 | MsgError s
=> print ("Configuration failed: " ^ s ^
"\n")
318 | _
=> print
"Unexpected server reply.\n";
321 handle ErrorMsg
.Error
=> ()
325 val (_
, bio
) = requestBio (fn () => ())
330 handle _
=> OS
.Process
.failure
332 fun requestShutdown () =
334 val (_
, bio
) = requestBio (fn () => ())
336 Msg
.send (bio
, MsgShutdown
);
338 NONE
=> print
"Server closed connection unexpectedly.\n"
341 MsgOk
=> print
"Shutdown begun.\n"
342 | MsgError s
=> print ("Shutdown failed: " ^ s ^
"\n")
343 | _
=> print
"Unexpected server reply.\n";
347 fun requestSlavePing () =
349 val (_
, bio
) = requestSlaveBio ()
354 handle _
=> OS
.Process
.failure
356 fun requestSlaveShutdown () =
358 val (_
, bio
) = requestSlaveBio ()
360 Msg
.send (bio
, MsgShutdown
);
362 NONE
=> print
"Server closed connection unexpectedly.\n"
365 MsgOk
=> print
"Shutdown begun.\n"
366 | MsgError s
=> print ("Shutdown failed: " ^ s ^
"\n")
367 | _
=> print
"Unexpected server reply.\n";
371 fun requestGrant acl
=
373 val (user
, bio
) = requestBio (fn () => ())
375 Msg
.send (bio
, MsgGrant acl
);
377 NONE
=> print
"Server closed connection unexpectedly.\n"
380 MsgOk
=> print
"Grant succeeded.\n"
381 | MsgError s
=> print ("Grant failed: " ^ s ^
"\n")
382 | _
=> print
"Unexpected server reply.\n";
386 fun requestRevoke acl
=
388 val (user
, bio
) = requestBio (fn () => ())
390 Msg
.send (bio
, MsgRevoke acl
);
392 NONE
=> print
"Server closed connection unexpectedly.\n"
395 MsgOk
=> print
"Revoke succeeded.\n"
396 | MsgError s
=> print ("Revoke failed: " ^ s ^
"\n")
397 | _
=> print
"Unexpected server reply.\n";
401 fun requestListPerms user
=
403 val (_
, bio
) = requestBio (fn () => ())
405 Msg
.send (bio
, MsgListPerms user
);
406 (case Msg
.recv bio
of
407 NONE
=> (print
"Server closed connection unexpectedly.\n";
411 MsgPerms perms
=> SOME perms
412 | MsgError s
=> (print ("Listing failed: " ^ s ^
"\n");
414 | _
=> (print
"Unexpected server reply.\n";
416 before OpenSSL
.close bio
419 fun requestWhoHas perm
=
421 val (_
, bio
) = requestBio (fn () => ())
423 Msg
.send (bio
, MsgWhoHas perm
);
424 (case Msg
.recv bio
of
425 NONE
=> (print
"Server closed connection unexpectedly.\n";
429 MsgWhoHasResponse users
=> SOME users
430 | MsgError s
=> (print ("whohas failed: " ^ s ^
"\n");
432 | _
=> (print
"Unexpected server reply.\n";
434 before OpenSSL
.close bio
437 fun requestRegen () =
439 val (_
, bio
) = requestBio (fn () => ())
441 Msg
.send (bio
, MsgRegenerate
);
443 NONE
=> print
"Server closed connection unexpectedly.\n"
446 MsgOk
=> print
"Regeneration succeeded.\n"
447 | MsgError s
=> print ("Regeneration failed: " ^ s ^
"\n")
448 | _
=> print
"Unexpected server reply.\n";
452 fun requestRegenTc () =
454 val (_
, bio
) = requestBio (fn () => ())
456 Msg
.send (bio
, MsgRegenerateTc
);
458 NONE
=> print
"Server closed connection unexpectedly.\n"
461 MsgOk
=> print
"All configuration validated.\n"
462 | MsgError s
=> print ("Configuration validation failed: " ^ s ^
"\n")
463 | _
=> print
"Unexpected server reply.\n";
467 fun requestRmdom dom
=
469 val (_
, bio
) = requestBio (fn () => ())
471 Msg
.send (bio
, MsgRmdom dom
);
473 NONE
=> print
"Server closed connection unexpectedly.\n"
476 MsgOk
=> print
"Removal succeeded.\n"
477 | MsgError s
=> print ("Removal failed: " ^ s ^
"\n")
478 | _
=> print
"Unexpected server reply.\n";
482 fun requestRmuser user
=
484 val (_
, bio
) = requestBio (fn () => ())
486 Msg
.send (bio
, MsgRmuser user
);
488 NONE
=> print
"Server closed connection unexpectedly.\n"
491 MsgOk
=> print
"Removal succeeded.\n"
492 | MsgError s
=> print ("Removal failed: " ^ s ^
"\n")
493 | _
=> print
"Unexpected server reply.\n";
497 fun requestDbUser dbtype
=
499 val (_
, bio
) = requestBio (fn () => ())
501 Msg
.send (bio
, MsgCreateDbUser dbtype
);
503 NONE
=> print
"Server closed connection unexpectedly.\n"
506 MsgOk
=> print
"Your user has been created.\n"
507 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
508 | _
=> print
"Unexpected server reply.\n";
512 fun requestDbPasswd rc
=
514 val (_
, bio
) = requestBio (fn () => ())
516 Msg
.send (bio
, MsgDbPasswd rc
);
518 NONE
=> print
"Server closed connection unexpectedly.\n"
521 MsgOk
=> print
"Your password has been changed.\n"
522 | MsgError s
=> print ("Password set failed: " ^ s ^
"\n")
523 | _
=> print
"Unexpected server reply.\n";
527 fun requestDbTable p
=
529 val (user
, bio
) = requestBio (fn () => ())
531 Msg
.send (bio
, MsgCreateDb p
);
533 NONE
=> print
"Server closed connection unexpectedly.\n"
536 MsgOk
=> print ("Your database " ^ user ^
"_" ^ #dbname p ^
" has been created.\n")
537 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
538 | _
=> print
"Unexpected server reply.\n";
542 fun requestDbDrop p
=
544 val (user
, bio
) = requestBio (fn () => ())
546 Msg
.send (bio
, MsgDropDb p
);
548 NONE
=> print
"Server closed connection unexpectedly.\n"
551 MsgOk
=> print ("Your database " ^ user ^
"_" ^ #dbname p ^
" has been dropped.\n")
552 | MsgError s
=> print ("Drop failed: " ^ s ^
"\n")
553 | _
=> print
"Unexpected server reply.\n";
557 fun requestDbGrant p
=
559 val (user
, bio
) = requestBio (fn () => ())
561 Msg
.send (bio
, MsgGrantDb p
);
563 NONE
=> print
"Server closed connection unexpectedly.\n"
566 MsgOk
=> print ("You've been granted all allowed privileges to database " ^ user ^
"_" ^ #dbname p ^
".\n")
567 | MsgError s
=> print ("Grant failed: " ^ s ^
"\n")
568 | _
=> print
"Unexpected server reply.\n";
572 fun requestListMailboxes domain
=
574 val (_
, bio
) = requestBio (fn () => ())
576 Msg
.send (bio
, MsgListMailboxes domain
);
577 (case Msg
.recv bio
of
578 NONE
=> Vmail
.Error
"Server closed connection unexpectedly."
581 MsgMailboxes users
=> (Msg
.send (bio
, MsgOk
);
583 | MsgError s
=> Vmail
.Error ("Listing failed: " ^ s
)
584 | _
=> Vmail
.Error
"Unexpected server reply.")
585 before OpenSSL
.close bio
588 fun requestNewMailbox p
=
590 val (_
, bio
) = requestBio (fn () => ())
592 Msg
.send (bio
, MsgNewMailbox p
);
594 NONE
=> print
"Server closed connection unexpectedly.\n"
597 MsgOk
=> print ("A mapping for " ^ #user p ^
"@" ^ #domain p ^
" has been created.\n")
598 | MsgError s
=> print ("Creation failed: " ^ s ^
"\n")
599 | _
=> print
"Unexpected server reply.\n";
603 fun requestPasswdMailbox p
=
605 val (_
, bio
) = requestBio (fn () => ())
607 Msg
.send (bio
, MsgPasswdMailbox p
);
609 NONE
=> print
"Server closed connection unexpectedly.\n"
612 MsgOk
=> print ("The password for " ^ #user p ^
"@" ^ #domain p ^
" has been changed.\n")
613 | MsgError s
=> print ("Set failed: " ^ s ^
"\n")
614 | _
=> print
"Unexpected server reply.\n";
618 fun requestRmMailbox p
=
620 val (_
, bio
) = requestBio (fn () => ())
622 Msg
.send (bio
, MsgRmMailbox p
);
624 NONE
=> print
"Server closed connection unexpectedly.\n"
627 MsgOk
=> print ("The mapping for mailbox " ^ #user p ^
"@" ^ #domain p ^
" has been deleted.\n")
628 | MsgError s
=> print ("Remove failed: " ^ s ^
"\n")
629 | _
=> print
"Unexpected server reply.\n";
633 fun requestSaQuery addr
=
635 val (_
, bio
) = requestBio (fn () => ())
637 Msg
.send (bio
, MsgSaQuery addr
);
638 (case Msg
.recv bio
of
639 NONE
=> print
"Server closed connection unexpectedly.\n"
642 MsgSaStatus b
=> (print ("SpamAssassin filtering for " ^ addr ^
" is "
643 ^
(if b
then "ON" else "OFF") ^
".\n");
644 Msg
.send (bio
, MsgOk
))
645 | MsgError s
=> print ("Query failed: " ^ s ^
"\n")
646 | _
=> print
"Unexpected server reply.\n")
647 before OpenSSL
.close bio
652 val (_
, bio
) = requestBio (fn () => ())
654 Msg
.send (bio
, MsgSaSet p
);
656 NONE
=> print
"Server closed connection unexpectedly.\n"
659 MsgOk
=> print ("SpamAssassin filtering for " ^ #
1 p ^
" is now "
660 ^
(if #
2 p
then "ON" else "OFF") ^
".\n")
661 | MsgError s
=> print ("Set failed: " ^ s ^
"\n")
662 | _
=> print
"Unexpected server reply.\n";
666 fun requestSmtpLog domain
=
668 val (_
, bio
) = requestBio (fn () => ())
670 val _
= Msg
.send (bio
, MsgSmtpLogReq domain
)
674 NONE
=> print
"Server closed connection unexpectedly.\n"
678 | MsgSmtpLogRes line
=> (print line
;
680 | MsgError s
=> print ("Log search failed: " ^ s ^
"\n")
681 | _
=> print
"Unexpected server reply.\n"
687 fun requestMysqlFixperms () =
689 val (_
, bio
) = requestBio (fn () => ())
691 Msg
.send (bio
, MsgMysqlFixperms
);
693 NONE
=> print
"Server closed connection unexpectedly.\n"
696 MsgOk
=> print
"Permissions granted.\n"
697 | MsgError s
=> print ("Failed: " ^ s ^
"\n")
698 | _
=> print
"Unexpected server reply.\n";
702 fun requestApt
{node
, pkg
} =
704 val (user
, context
) = requestContext (fn () => ())
705 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
708 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
710 val _
= Msg
.send (bio
, MsgQuery (QApt pkg
))
714 NONE
=> (print
"Server closed connection unexpectedly.\n";
718 MsgYes
=> (print
"Package is installed.\n";
720 | MsgNo
=> (print
"Package is not installed.\n";
722 | MsgError s
=> (print ("APT query failed: " ^ s ^
"\n");
724 | _
=> (print
"Unexpected server reply.\n";
728 before OpenSSL
.close bio
731 fun requestCron
{node
, uname
} =
733 val (user
, context
) = requestContext (fn () => ())
734 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
737 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
739 val _
= Msg
.send (bio
, MsgQuery (QCron uname
))
743 NONE
=> (print
"Server closed connection unexpectedly.\n";
747 MsgYes
=> (print
"User has cron permissions.\n";
749 | MsgNo
=> (print
"User does not have cron permissions.\n";
751 | MsgError s
=> (print ("Cron query failed: " ^ s ^
"\n");
753 | _
=> (print
"Unexpected server reply.\n";
757 before OpenSSL
.close bio
760 fun requestFtp
{node
, uname
} =
762 val (user
, context
) = requestContext (fn () => ())
763 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
766 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
768 val _
= Msg
.send (bio
, MsgQuery (QFtp uname
))
772 NONE
=> (print
"Server closed connection unexpectedly.\n";
776 MsgYes
=> (print
"User has FTP permissions.\n";
778 | MsgNo
=> (print
"User does not have FTP permissions.\n";
780 | MsgError s
=> (print ("FTP query failed: " ^ s ^
"\n");
782 | _
=> (print
"Unexpected server reply.\n";
786 before OpenSSL
.close bio
789 fun requestTrustedPath
{node
, uname
} =
791 val (user
, context
) = requestContext (fn () => ())
792 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
795 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
797 val _
= Msg
.send (bio
, MsgQuery (QTrustedPath uname
))
801 NONE
=> (print
"Server closed connection unexpectedly.\n";
805 MsgYes
=> (print
"User has trusted path restriction.\n";
807 | MsgNo
=> (print
"User does not have trusted path restriction.\n";
809 | MsgError s
=> (print ("Trusted path query failed: " ^ s ^
"\n");
811 | _
=> (print
"Unexpected server reply.\n";
815 before OpenSSL
.close bio
818 fun requestSocketPerm
{node
, uname
} =
820 val (user
, context
) = requestContext (fn () => ())
821 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
824 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
826 val _
= Msg
.send (bio
, MsgQuery (QSocket uname
))
830 NONE
=> (print
"Server closed connection unexpectedly.\n";
834 MsgSocket p
=> (case p
of
836 | Client
=> print
"Client\n"
837 | Server
=> print
"Server\n"
838 | Nada
=> print
"Nada\n";
840 | MsgError s
=> (print ("Socket permission query failed: " ^ s ^
"\n");
842 | _
=> (print
"Unexpected server reply.\n";
846 before OpenSSL
.close bio
849 fun requestFirewall
{node
, uname
} =
851 val (user
, context
) = requestContext (fn () => ())
852 val bio
= OpenSSL
.connect (context
, if node
= Config
.masterNode
then
855 Domain
.nodeIp node ^
":" ^
Int.toString Config
.slavePort
)
857 val _
= Msg
.send (bio
, MsgQuery (QFirewall uname
))
861 NONE
=> (print
"Server closed connection unexpectedly.\n";
865 MsgFirewall ls
=> (app (fn s
=> (print s
; print
"\n")) ls
;
867 | MsgError s
=> (print ("Firewall query failed: " ^ s ^
"\n");
869 | _
=> (print
"Unexpected server reply.\n";
873 before OpenSSL
.close bio
876 fun regenerateEither tc checker context
=
884 val _
= ErrorMsg
.reset ()
887 val () = Tycheck
.disallowExterns ()
889 val () = ifReal Domain
.resetGlobal
893 fun contactNode (node
, ip
) =
894 if node
= Config
.defaultNode
then
897 val bio
= OpenSSL
.connect (context
,
900 ^
Int.toString Config
.slavePort
)
902 Msg
.send (bio
, MsgRegenerate
);
904 NONE
=> print
"Slave closed connection unexpectedly\n"
907 MsgOk
=> print ("Slave " ^ node ^
" pre-regeneration finished\n")
908 | MsgError s
=> print ("Slave " ^ node
909 ^
" returned error: " ^
911 | _
=> print ("Slave " ^ node
912 ^
" returned unexpected command\n");
915 handle OpenSSL
.OpenSSL s
=> print ("OpenSSL error: " ^ s ^
"\n")
919 val _
= Domain
.setUser user
920 val _
= ErrorMsg
.reset ()
922 val dname
= Config
.domtoolDir user
924 if Posix
.FileSys
.access (dname
, []) then
926 val dir
= Posix
.FileSys
.opendir dname
929 case Posix
.FileSys
.readdir dir
of
930 NONE
=> (Posix
.FileSys
.closedir dir
;
934 loop (OS
.Path
.joinDirFile
{dir
= dname
,
941 val (_
, files
) = Order
.order (SOME b
) files
943 if !ErrorMsg
.anyErrors
then
945 print ("User " ^ user ^
"'s configuration has errors!\n"))
952 handle IO
.Io
{name
, function
, ...} =>
953 (print ("IO error processing user " ^ user ^
": " ^ function ^
": " ^ name ^
"\n");
955 | exn
as OS
.SysErr (s
, _
) => (print ("System error processing user " ^ user ^
": " ^ s ^
"\n");
957 | ErrorMsg
.Error
=> (ErrorMsg
.reset ();
958 print ("User " ^ user ^
" had a compilation error.\n");
960 | _
=> (print
"Unknown exception during regeneration!\n";
963 ifReal (fn () => (app contactNode Config
.nodeIps
;
965 app
doUser (Acl
.users ());
970 val regenerate
= regenerateEither
false eval
'
971 val regenerateTc
= regenerateEither
true (ignore
o check
)
975 val doms
= Acl
.class
{user
= user
, class
= "domain"}
976 val doms
= List.filter (fn dom
=>
977 case Acl
.whoHas
{class
= "domain", value
= dom
} of
979 | _
=> false) (StringSet
.listItems doms
)
985 fun now () = Date
.toString (Date
.fromTimeUniv (Time
.now ()))
989 QApt pkg
=> if Apt
.installed pkg
then MsgYes
else MsgNo
990 | QCron user
=> if Cron
.allowed user
then MsgYes
else MsgNo
991 | QFtp user
=> if Ftp
.allowed user
then MsgYes
else MsgNo
992 | QTrustedPath user
=> if TrustedPath
.query user
then MsgYes
else MsgNo
993 | QSocket user
=> MsgSocket (SocketPerm
.query user
)
994 | QFirewall user
=> MsgFirewall (Firewall
.query user
)
996 fun describeQuery q
=
998 QApt pkg
=> "Requested installation status of package " ^ pkg
999 | QCron user
=> "Asked about cron permissions for user " ^ user
1000 | QFtp user
=> "Asked about FTP permissions for user " ^ user
1001 | QTrustedPath user
=> "Asked about trusted path settings for user " ^ user
1002 | QSocket user
=> "Asked about socket permissions for user " ^ user
1003 | QFirewall user
=> "Asked about firewall rules for user " ^ user
1007 val () = Acl
.read Config
.aclFile
1009 val context
= context (Config
.serverCert
,
1012 val _
= Domain
.set_context context
1014 val sock
= OpenSSL
.listen (context
, Config
.dispatcherPort
)
1017 (case OpenSSL
.accept sock
of
1021 val user
= OpenSSL
.peerCN bio
1022 val () = print ("\nConnection from " ^ user ^
" at " ^
now () ^
"\n")
1023 val () = Domain
.setUser user
1025 fun doIt f cleanup
=
1027 (msgLocal
, SOME msgRemote
) =>
1030 Msg
.send (bio
, MsgError msgRemote
))
1031 |
(msgLocal
, NONE
) =>
1034 Msg
.send (bio
, MsgOk
)))
1035 handle e
as (OpenSSL
.OpenSSL s
) =>
1036 (print ("OpenSSL error: " ^ s ^
"\n");
1037 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1038 Msg
.send (bio
, MsgError ("OpenSSL error: " ^ s
))
1039 handle OpenSSL
.OpenSSL _
=> ())
1040 | OS
.SysErr (s
, _
) =>
1041 (print
"System error: ";
1044 Msg
.send (bio
, MsgError ("System error: " ^ s
))
1045 handle OpenSSL
.OpenSSL _
=> ())
1050 Msg
.send (bio
, MsgError ("Failure: " ^ s
))
1051 handle OpenSSL
.OpenSSL _
=> ())
1053 (print
"Compilation error\n";
1054 Msg
.send (bio
, MsgError
"Error during configuration evaluation")
1055 handle OpenSSL
.OpenSSL _
=> ());
1057 ignore (OpenSSL
.readChar bio
);
1059 handle OpenSSL
.OpenSSL _
=> ();
1062 fun doConfig codes
=
1064 val _
= print
"Configuration:\n"
1065 val _
= app (fn s
=> (print s
; print
"\n")) codes
1068 val outname
= OS
.FileSys
.tmpName ()
1072 val outf
= TextIO.openOut outname
1074 TextIO.output (outf
, code
);
1075 TextIO.closeOut outf
;
1079 doIt (fn () => (Env
.pre ();
1082 Msg
.send (bio
, MsgOk
);
1083 ("Configuration complete.", NONE
)))
1084 (fn () => OS
.FileSys
.remove outname
)
1088 case String.fields (fn ch
=> ch
= #
"@") s
of
1090 if user
= user
' then
1094 |
[user
', domain
] =>
1095 if Domain
.validEmailUser user
' andalso Domain
.yourDomain domain
then
1096 SOME (SetSA
.Email s
)
1102 case Msg
.recv bio
of
1103 NONE
=> (OpenSSL
.close bio
1104 handle OpenSSL
.OpenSSL _
=> ();
1108 MsgConfig code
=> doConfig
[code
]
1109 | MsgMultiConfig codes
=> doConfig codes
1112 if Acl
.query
{user
= user
, class
= "priv", value
= "all"}
1113 orelse Acl
.query
{user
= user
, class
= "priv", value
= "shutdown"} then
1114 print ("Domtool dispatcher shutting down at " ^
now () ^
"\n\n")
1116 (print
"Unauthorized shutdown command!\n";
1118 handle OpenSSL
.OpenSSL _
=> ();
1123 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1125 Acl
.write Config
.aclFile
;
1126 ("Granted permission " ^ #value acl ^
" to " ^ #user acl ^
" in " ^ #class acl ^
".",
1129 ("Unauthorized user asked to grant a permission!",
1130 SOME
"Not authorized to grant privileges"))
1135 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1137 Acl
.write Config
.aclFile
;
1138 ("Revoked permission " ^ #value acl ^
" from " ^ #user acl ^
" in " ^ #class acl ^
".",
1141 ("Unauthorized user asked to revoke a permission!",
1142 SOME
"Not authorized to revoke privileges"))
1145 | MsgListPerms user
=>
1147 (Msg
.send (bio
, MsgPerms (Acl
.queryAll user
));
1148 ("Sent permission list for user " ^ user ^
".",
1154 (Msg
.send (bio
, MsgWhoHasResponse (Acl
.whoHas perm
));
1155 ("Sent whohas response for " ^ #class perm ^
" / " ^ #value perm ^
".",
1161 if Acl
.query
{user
= user
, class
= "priv", value
= "all"}
1162 orelse List.all (fn dom
=> Acl
.query
{user
= user
, class
= "domain", value
= dom
}) doms
then
1165 Acl
.revokeFromAll
{class
= "domain", value
= dom
}) doms
;
1166 Acl
.write Config
.aclFile
;
1167 ("Removed domains" ^
foldl (fn (d
, s
) => s ^
" " ^ d
) "" doms ^
".",
1170 ("Unauthorized user asked to remove a domain!",
1171 SOME
"Not authorized to remove that domain"))
1176 if Acl
.query
{user
= user
, class
= "priv", value
= "regen"}
1177 orelse Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1178 (if regenerate context
then
1179 ("Regenerated all configuration.",
1182 ("Error regenerating configuration!",
1183 SOME
"Error regenerating configuration! Consult /var/log/domtool.log."))
1185 ("Unauthorized user asked to regenerate!",
1186 SOME
"Not authorized to regenerate"))
1189 | MsgRegenerateTc
=>
1191 if Acl
.query
{user
= user
, class
= "priv", value
= "regen"}
1192 orelse Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1193 (if regenerateTc context
then
1194 ("Checked all configuration.",
1197 ("Found a compilation error!",
1198 SOME
"Found a compilation error! Consult /var/log/domtool.log."))
1200 ("Unauthorized user asked to regenerate -tc!",
1201 SOME
"Not authorized to regenerate -tc"))
1204 | MsgRmuser user
' =>
1206 if Acl
.query
{user
= user
, class
= "priv", value
= "all"} then
1208 Acl
.write Config
.aclFile
;
1209 ("Removed user " ^ user
' ^
".",
1212 ("Unauthorized user asked to remove a user!",
1213 SOME
"Not authorized to remove users"))
1216 | MsgCreateDbUser
{dbtype
, passwd
} =>
1218 case Dbms
.lookup dbtype
of
1219 NONE
=> ("Database user creation request with unknown datatype type " ^ dbtype
,
1220 SOME ("Unknown database type " ^ dbtype
))
1222 case #adduser handler
{user
= user
, passwd
= passwd
} of
1223 NONE
=> ("Added " ^ dbtype ^
" user " ^ user ^
".",
1226 ("Error adding a " ^ dbtype ^
" user " ^ user ^
": " ^ msg
,
1227 SOME ("Error adding user: " ^ msg
)))
1230 | MsgDbPasswd
{dbtype
, passwd
} =>
1232 case Dbms
.lookup dbtype
of
1233 NONE
=> ("Database passwd request with unknown datatype type " ^ dbtype
,
1234 SOME ("Unknown database type " ^ dbtype
))
1236 case #passwd handler
{user
= user
, passwd
= passwd
} of
1237 NONE
=> ("Changed " ^ dbtype ^
" password of user " ^ user ^
".",
1240 ("Error setting " ^ dbtype ^
" password of user " ^ user ^
": " ^ msg
,
1241 SOME ("Error adding user: " ^ msg
)))
1244 | MsgCreateDb
{dbtype
, dbname
} =>
1246 if Dbms
.validDbname dbname
then
1247 case Dbms
.lookup dbtype
of
1248 NONE
=> ("Database creation request with unknown datatype type " ^ dbtype
,
1249 SOME ("Unknown database type " ^ dbtype
))
1251 case #createdb handler
{user
= user
, dbname
= dbname
} of
1252 NONE
=> ("Created database " ^ user ^
"_" ^ dbname ^
".",
1254 | SOME msg
=> ("Error creating database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1255 SOME ("Error creating database: " ^ msg
))
1257 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1258 SOME ("Invalid database name " ^ dbname
)))
1261 | MsgDropDb
{dbtype
, dbname
} =>
1263 if Dbms
.validDbname dbname
then
1264 case Dbms
.lookup dbtype
of
1265 NONE
=> ("Database drop request with unknown datatype type " ^ dbtype
,
1266 SOME ("Unknown database type " ^ dbtype
))
1268 case #dropdb handler
{user
= user
, dbname
= dbname
} of
1269 NONE
=> ("Drop database " ^ user ^
"_" ^ dbname ^
".",
1271 | SOME msg
=> ("Error dropping database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1272 SOME ("Error dropping database: " ^ msg
))
1274 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1275 SOME ("Invalid database name " ^ dbname
)))
1278 | MsgGrantDb
{dbtype
, dbname
} =>
1280 if Dbms
.validDbname dbname
then
1281 case Dbms
.lookup dbtype
of
1282 NONE
=> ("Database drop request with unknown datatype type " ^ dbtype
,
1283 SOME ("Unknown database type " ^ dbtype
))
1285 case #grant handler
{user
= user
, dbname
= dbname
} of
1286 NONE
=> ("Grant permissions to database " ^ user ^
"_" ^ dbname ^
".",
1288 | SOME msg
=> ("Error granting permissions to database " ^ user ^
"_" ^ dbname ^
": " ^ msg
,
1289 SOME ("Error granting permissions to database: " ^ msg
))
1291 ("Invalid database name " ^ user ^
"_" ^ dbname
,
1292 SOME ("Invalid database name " ^ dbname
)))
1295 | MsgListMailboxes domain
=>
1297 if not (Domain
.yourDomain domain
) then
1298 ("User wasn't authorized to list mailboxes for " ^ domain
,
1299 SOME
"You're not authorized to configure that domain.")
1301 case Vmail
.list domain
of
1302 Vmail
.Listing users
=> (Msg
.send (bio
, MsgMailboxes users
);
1303 ("Sent mailbox list for " ^ domain
,
1305 | Vmail
.Error msg
=> ("Error listing mailboxes for " ^ domain ^
": " ^ msg
,
1309 | MsgNewMailbox
{domain
, user
= emailUser
, passwd
, mailbox
} =>
1311 if not (Domain
.yourDomain domain
) then
1312 ("User wasn't authorized to add a mailbox to " ^ domain
,
1313 SOME
"You're not authorized to configure that domain.")
1314 else if not (Domain
.validEmailUser emailUser
) then
1315 ("Invalid e-mail username " ^ emailUser
,
1316 SOME
"Invalid e-mail username")
1317 else if not (CharVector
.all
Char.isGraph passwd
) then
1318 ("Invalid password",
1319 SOME
"Invalid password; may only contain printable, non-space characters")
1320 else if not (Domain
.yourPath mailbox
) then
1321 ("User wasn't authorized to add a mailbox at " ^ mailbox
,
1322 SOME ("You're not authorized to use that mailbox location. ("
1325 case Vmail
.add
{requester
= user
,
1326 domain
= domain
, user
= emailUser
,
1327 passwd
= passwd
, mailbox
= mailbox
} of
1328 NONE
=> ("Added mailbox " ^ emailUser ^
"@" ^ domain ^
" at " ^ mailbox
,
1330 | SOME msg
=> ("Error adding mailbox " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1334 | MsgPasswdMailbox
{domain
, user
= emailUser
, passwd
} =>
1336 if not (Domain
.yourDomain domain
) then
1337 ("User wasn't authorized to change password of a mailbox for " ^ domain
,
1338 SOME
"You're not authorized to configure that domain.")
1339 else if not (Domain
.validEmailUser emailUser
) then
1340 ("Invalid e-mail username " ^ emailUser
,
1341 SOME
"Invalid e-mail username")
1342 else if not (CharVector
.all
Char.isGraph passwd
) then
1343 ("Invalid password",
1344 SOME
"Invalid password; may only contain printable, non-space characters")
1346 case Vmail
.passwd
{domain
= domain
, user
= emailUser
,
1348 NONE
=> ("Changed password of mailbox " ^ emailUser ^
"@" ^ domain
,
1350 | SOME msg
=> ("Error changing mailbox password for " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1354 | MsgRmMailbox
{domain
, user
= emailUser
} =>
1356 if not (Domain
.yourDomain domain
) then
1357 ("User wasn't authorized to change password of a mailbox for " ^ domain
,
1358 SOME
"You're not authorized to configure that domain.")
1359 else if not (Domain
.validEmailUser emailUser
) then
1360 ("Invalid e-mail username " ^ emailUser
,
1361 SOME
"Invalid e-mail username")
1363 case Vmail
.rm
{domain
= domain
, user
= emailUser
} of
1364 NONE
=> ("Deleted mailbox " ^ emailUser ^
"@" ^ domain
,
1366 | SOME msg
=> ("Error deleting mailbox " ^ emailUser ^
"@" ^ domain ^
": " ^ msg
,
1370 | MsgSaQuery addr
=>
1372 case checkAddr addr
of
1373 NONE
=> ("User tried to query SA filtering for " ^ addr
,
1374 SOME
"You aren't allowed to configure SA filtering for that recipient.")
1375 | SOME addr
' => (Msg
.send (bio
, MsgSaStatus (SetSA
.query addr
'));
1376 ("Queried SA filtering status for " ^ addr
,
1380 |
MsgSaSet (addr
, b
) =>
1382 case checkAddr addr
of
1383 NONE
=> ("User tried to set SA filtering for " ^ addr
,
1384 SOME
"You aren't allowed to configure SA filtering for that recipient.")
1385 | SOME addr
' => (SetSA
.set (addr
', b
);
1386 Msg
.send (bio
, MsgOk
);
1387 ("Set SA filtering status for " ^ addr ^
" to "
1388 ^
(if b
then "ON" else "OFF"),
1392 | MsgSmtpLogReq domain
=>
1394 if not (Domain
.yourDomain domain
) then
1395 ("Unauthorized user tried to request SMTP logs for " ^ domain
,
1396 SOME
"You aren't authorized to configure that domain.")
1398 (SmtpLog
.search (fn line
=> Msg
.send (bio
, MsgSmtpLogRes line
))
1400 ("Requested SMTP logs for " ^ domain
,
1405 doIt (fn () => (Msg
.send (bio
, answerQuery q
);
1410 | MsgMysqlFixperms
=>
1411 doIt (fn () => if OS
.Process
.isSuccess
1412 (OS
.Process
.system
"/usr/bin/sudo -H /afs/hcoop.net/common/etc/scripts/mysql-grant-table-drop") then
1413 ("Requested mysql-fixperms",
1416 ("Requested mysql-fixperms, but execution failed!",
1417 SOME
"Script execution failed."))
1421 doIt (fn () => ("Unexpected command",
1422 SOME
"Unexpected command"))
1427 handle e
as (OpenSSL
.OpenSSL s
) =>
1428 (print ("OpenSSL error: " ^ s ^
"\n");
1429 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1431 handle OpenSSL
.OpenSSL _
=> ();
1433 | OS
.SysErr (s
, _
) =>
1434 (print ("System error: " ^ s ^
"\n");
1436 handle OpenSSL
.OpenSSL _
=> ();
1438 | IO
.Io
{name
, function
, cause
} =>
1439 (print ("IO error: " ^ function ^
" for " ^ name ^
"\n");
1440 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory cause
);
1442 handle OpenSSL
.OpenSSL _
=> ();
1444 | OS
.Path
.InvalidArc
=>
1445 (print
"Invalid arc\n";
1447 handle OpenSSL
.OpenSSL _
=> ();
1450 (print
"Unknown exception in main loop!\n";
1451 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1453 handle OpenSSL
.OpenSSL _
=> ();
1455 handle e
as (OpenSSL
.OpenSSL s
) =>
1456 (print ("OpenSSL error: " ^ s ^
"\n");
1457 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1459 | OS
.SysErr (s
, _
) =>
1460 (print ("System error: " ^ s ^
"\n");
1462 | IO
.Io
{name
, function
, cause
} =>
1463 (print ("IO error: " ^ function ^
" for " ^ name ^
"\n");
1464 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory cause
);
1467 (print
"Unknown exception in main loop!\n";
1468 app (fn x
=> print (x ^
"\n")) (SMLofNJ
.exnHistory e
);
1471 print ("Domtool dispatcher starting up at " ^
now () ^
"\n");
1472 print
"Listening for connections....\n";
1474 OpenSSL
.shutdown sock
1479 val host
= Slave
.hostname ()
1481 val context
= context (Config
.certDir ^
"/" ^ host ^
".pem",
1482 Config
.keyDir ^
"/" ^ host ^
"/key.pem",
1485 val sock
= OpenSSL
.listen (context
, Config
.slavePort
)
1487 val _
= print ("Slave server starting at " ^
now () ^
"\n")
1490 case OpenSSL
.accept sock
of
1494 val peer
= OpenSSL
.peerCN bio
1495 val () = print ("\nConnection from " ^ peer ^
" at " ^
now () ^
"\n")
1497 if peer
= Config
.dispatcherName
then let
1499 case Msg
.recv bio
of
1500 NONE
=> print
"Dispatcher closed connection unexpectedly\n"
1503 MsgFile file
=> loop
' (file
:: files
)
1504 | MsgDoFiles
=> (Slave
.handleChanges files
;
1505 Msg
.send (bio
, MsgOk
))
1506 | MsgRegenerate
=> (Domain
.resetLocal ();
1507 Msg
.send (bio
, MsgOk
))
1508 | _
=> (print
"Dispatcher sent unexpected command\n";
1509 Msg
.send (bio
, MsgError
"Unexpected command"))
1512 ignore (OpenSSL
.readChar bio
);
1516 else if peer
= "domtool" then
1517 case Msg
.recv bio
of
1518 SOME MsgShutdown
=> (OpenSSL
.close bio
;
1519 print ("Shutting down at " ^
now () ^
"\n\n"))
1520 | _
=> (OpenSSL
.close bio
;
1523 case Msg
.recv bio
of
1524 SOME (MsgQuery q
) => (print (describeQuery q ^
"\n");
1525 Msg
.send (bio
, answerQuery q
);
1526 ignore (OpenSSL
.readChar bio
);
1529 | _
=> (OpenSSL
.close bio
;
1531 end handle OpenSSL
.OpenSSL s
=>
1532 (print ("OpenSSL error: "^ s ^
"\n");
1534 handle OpenSSL
.OpenSSL _
=> ();
1536 | e
as OS
.SysErr (s
, _
) =>
1537 (app (fn s
=> print (s ^
"\n")) (SMLofNJ
.exnHistory e
);
1538 print ("System error: "^ s ^
"\n");
1540 handle OpenSSL
.OpenSSL _
=> ();
1544 OpenSSL
.shutdown sock
1549 val dir
= Posix
.FileSys
.opendir Config
.libRoot
1552 case Posix
.FileSys
.readdir dir
of
1553 NONE
=> (Posix
.FileSys
.closedir dir
;
1556 if String.isSuffix
".dtl" fname
then
1557 loop (OS
.Path
.joinDirFile
{dir
= Config
.libRoot
,
1566 fun autodocBasis outdir
=
1567 Autodoc
.autodoc
{outdir
= outdir
, infiles
= listBasis ()}