structure Domain :> DOMAIN = struct
+structure SM = DataStructures.StringMap
+structure SS = DataStructures.StringSet
+
+val nodes = map #1 Config.nodeIps
+val nodeMap = foldl (fn ((node, ip), mp) => SM.insert (mp, node, ip))
+ SM.empty Config.nodeIps
+
+val usr = ref ""
+fun setUser ur = usr := ur
+fun getUser () = !usr
+
+val your_doms = ref SS.empty
+fun your_domains () = !your_doms
+
fun validIp s =
case map Int.fromString (String.fields (fn ch => ch = #".") s) of
[SOME n1, SOME n2, SOME n3, SOME n4] =>
size s > 0 andalso size s < 100
andalso List.all validHost (String.fields (fn ch => ch = #".") s)
-fun validNode s = List.exists (fn s' => s = s') Config.nodes
+fun validNode s = List.exists (fn s' => s = s') nodes
+
+fun yourDomain s = SS.member (your_domains (), s)
val _ = Env.type_one "ip"
Env.string
Env.string
validDomain
+val _ = Env.type_one "your_domain"
+ Env.string
+ yourDomain
+
val _ = Env.type_one "node"
Env.string
validNode
handle OS.SysErr _ => OS.FileSys.mkDir tmpPath
end
in
- app doNode Config.nodes;
+ app doNode nodes;
elems
end) [] toks
in
TextIO.closeOut outf
end
- fun saveNamed (kind, soa : soa) node =
+ fun saveNamed (kind, soa : soa, masterIp) node =
let
val outf = domainFile {node = node, name = "named.conf"}
in
TextIO.output (outf, ".zone\";\n");
case kind of
"master" => TextIO.output (outf, "\tallow-update { none; };\n")
- | _ => TextIO.output (outf, "\tmasters { 1.2.3.4; };\n");
- TextIO.output (outf, "}\n");
+ | _ => (TextIO.output (outf, "\tmasters { ");
+ TextIO.output (outf, masterIp);
+ TextIO.output (outf, " };\n"));
+ TextIO.output (outf, "};\n");
TextIO.closeOut outf
end
-
- fun saveBoth ks = (saveSoa ks; saveNamed ks)
in
case kind of
NoDns => masterNode := NONE
| UseDns dns =>
- (app (saveSoa ("slave", #soa dns)) (#slaves dns);
- app (saveNamed ("slave", #soa dns)) (#slaves dns);
- case #master dns of
- InternalMaster node =>
- (masterNode := SOME node;
- saveSoa ("master", #soa dns) node;
- saveNamed ("master", #soa dns) node)
- | _ => masterNode := NONE);
- !befores dom
+ let
+ val masterIp =
+ case #master dns of
+ InternalMaster node => valOf (SM.find (nodeMap, node))
+ | ExternalMaster ip => ip
+ in
+ app (saveSoa ("slave", #soa dns)) (#slaves dns);
+ app (saveNamed ("slave", #soa dns, masterIp)) (#slaves dns);
+ case #master dns of
+ InternalMaster node =>
+ (masterNode := SOME node;
+ saveSoa ("master", #soa dns) node;
+ saveNamed ("master", #soa dns, masterIp) node)
+ | _ => masterNode := NONE;
+ !befores dom
+ end
end,
fn () => !afters (!current))
+val () = Env.registerPreTycheck (fn () => (setUser Config.testUser;
+ Acl.read Config.aclFile;
+ your_doms := Acl.class {user = getUser (),
+ class = "domain"}))
+
val () = Env.registerPre (fn () => (ignore (Slave.shellF ([Config.rm, " -rf ", Config.tmpDir, ""],
fn cl => "Temp file cleanup failed: " ^ cl));
OS.FileSys.mkDir Config.tmpDir;
app (fn node => OS.FileSys.mkDir
(OS.Path.joinDirFile {dir = Config.tmpDir,
file = node}))
- Config.nodes;
+ nodes;
app (fn node => OS.FileSys.mkDir
(OS.Path.joinDirFile {dir = Config.resultRoot,
file = node})
handle OS.SysErr _ => ())
- Config.nodes))
+ nodes))
val () = Env.registerPost (fn () =>
let
if !ErrorMsg.anyErrors then
()
else
- Slave.handleChanges (map #2 diffs);
+ Slave.handleChanges (map #2 diffs)(*;
ignore (Slave.shellF ([Config.rm, " -rf ", Config.tmpDir, ""],
- fn cl => "Temp file cleanup failed: " ^ cl))
+ fn cl => "Temp file cleanup failed: " ^ cl))*)
end)