annotate src/settings.sml @ 1949:d7bec4e63415

Do a bit of extra userspace copying to send HTTP response with a single system call
author Adam Chlipala <adam@chlipala.net>
date Tue, 31 Dec 2013 09:34:41 -0500
parents b6aee1d5b9b1
children cf7f7e51b0a2
rev   line source
adam@1478 1 (* Copyright (c) 2008-2011, Adam Chlipala
adamc@764 2 * All rights reserved.
adamc@764 3 *
adamc@764 4 * Redistribution and use in source and binary forms, with or without
adamc@764 5 * modification, are permitted provided that the following conditions are met:
adamc@764 6 *
adamc@764 7 * - Redistributions of source code must retain the above copyright notice,
adamc@764 8 * this list of conditions and the following disclaimer.
adamc@764 9 * - Redistributions in binary form must reproduce the above copyright notice,
adamc@764 10 * this list of conditions and the following disclaimer in the documentation
adamc@764 11 * and/or other materials provided with the distribution.
adamc@764 12 * - The names of contributors may not be used to endorse or promote products
adamc@764 13 * derived from this software without specific prior written permission.
adamc@764 14 *
adamc@764 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@764 16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@764 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@764 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
rmbruijn@1597 19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
adamc@764 20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@764 21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@764 22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@764 23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@764 24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@764 25 * POSSIBILITY OF SUCH DAMAGE.
adamc@764 26 *)
adamc@764 27
adamc@764 28 structure Settings :> SETTINGS = struct
adamc@764 29
ezyang@1739 30 val configBin = ref Config.bin
ezyang@1739 31 val configLib = ref Config.lib
ezyang@1739 32 val configSrcLib = ref Config.srclib
ezyang@1739 33 val configInclude = ref Config.includ
ezyang@1739 34 val configSitelisp = ref Config.sitelisp
ezyang@1739 35
grrwlf@1871 36 val configCCompiler = ref Config.ccompiler
grrwlf@1871 37
grrwlf@1871 38 fun getCCompiler () = !configCCompiler
grrwlf@1871 39 fun setCCompiler cc = configCCompiler := cc
grrwlf@1871 40
ezyang@1739 41 fun libUr () = OS.Path.joinDirFile {dir = !configSrcLib,
ezyang@1739 42 file = "ur"}
ezyang@1739 43 fun libC () = OS.Path.joinDirFile {dir = !configSrcLib,
ezyang@1739 44 file = "c"}
ezyang@1739 45 fun libJs () = OS.Path.joinDirFile {dir = !configSrcLib,
ezyang@1739 46 file = "js"}
ezyang@1739 47
ezyang@1739 48 fun libFile s = OS.Path.joinDirFile {dir = libUr (),
ezyang@1739 49 file = s}
ezyang@1739 50
adam@1637 51 val urlPrefixFull = ref "/"
adamc@764 52 val urlPrefix = ref "/"
adam@1370 53 val urlPrePrefix = ref ""
adamc@764 54 val timeout = ref 0
adamc@764 55 val headers = ref ([] : string list)
adamc@766 56 val scripts = ref ([] : string list)
adamc@764 57
adam@1637 58 fun getUrlPrefixFull () = !urlPrefixFull
adamc@764 59 fun getUrlPrefix () = !urlPrefix
adam@1370 60 fun getUrlPrePrefix () = !urlPrePrefix
adamc@764 61 fun setUrlPrefix p =
adam@1370 62 let
adam@1370 63 val prefix = if p = "" then
adam@1370 64 "/"
adam@1370 65 else if String.sub (p, size p - 1) <> #"/" then
adam@1370 66 p ^ "/"
adam@1370 67 else
adam@1370 68 p
adam@1370 69
adam@1470 70 fun findPrefix n =
adam@1470 71 let
adam@1470 72 val (befor, after) = Substring.splitl (fn ch => ch <> #"/") (Substring.extract (prefix, n, NONE))
adam@1470 73 in
adam@1470 74 if Substring.isEmpty after then
adam@1470 75 ("", prefix)
adam@1470 76 else
adam@1470 77 (String.substring (prefix, 0, n) ^ Substring.string befor, Substring.string after)
rmbruijn@1597 78 end
adam@1470 79
adam@1370 80 val (prepre, prefix) =
adam@1370 81 if String.isPrefix "http://" prefix then
adam@1470 82 findPrefix 7
adam@1470 83 else if String.isPrefix "https://" prefix then
adam@1470 84 findPrefix 8
adam@1370 85 else
adam@1370 86 ("", prefix)
adam@1370 87 in
adam@1637 88 urlPrefixFull := p;
adam@1370 89 urlPrePrefix := prepre;
adam@1370 90 urlPrefix := prefix
adam@1370 91 end
adamc@764 92
adamc@764 93 fun getTimeout () = !timeout
adamc@764 94 fun setTimeout n = timeout := n
adamc@764 95
adamc@764 96 fun getHeaders () = !headers
adamc@764 97 fun setHeaders ls = headers := ls
adamc@764 98
adamc@766 99 fun getScripts () = !scripts
adamc@766 100 fun setScripts ls = scripts := ls
adamc@766 101
adamc@765 102 type ffi = string * string
adamc@765 103
adamc@765 104 structure K = struct
adamc@765 105 type ord_key = ffi
adamc@765 106 fun compare ((m1, x1), (m2, x2)) =
adamc@765 107 Order.join (String.compare (m1, m2),
adamc@765 108 fn () => String.compare (x1, x2))
adamc@764 109 end
adamc@765 110
adamc@765 111 structure S = BinarySetFn(K)
adamc@765 112 structure M = BinaryMapFn(K)
adamc@765 113
adamc@765 114 fun basis x = S.addList (S.empty, map (fn x : string => ("Basis", x)) x)
adamc@765 115
adamc@765 116 val clientToServerBase = basis ["int",
adamc@765 117 "float",
adamc@765 118 "string",
adamc@765 119 "time",
adamc@765 120 "file",
adamc@765 121 "unit",
adamc@765 122 "option",
adamc@765 123 "list",
adam@1288 124 "bool",
adam@1288 125 "variant"]
adamc@765 126 val clientToServer = ref clientToServerBase
adamc@765 127 fun setClientToServer ls = clientToServer := S.addList (clientToServerBase, ls)
adamc@765 128 fun mayClientToServer x = S.member (!clientToServer, x)
adamc@765 129
adamc@779 130 val effectfulBase = basis ["dml",
adamc@779 131 "nextval",
adamc@1073 132 "setval",
adamc@779 133 "set_cookie",
adamc@1050 134 "clear_cookie",
adamc@765 135 "new_channel",
adamc@1200 136 "send",
adamc@1200 137 "htmlifyInt_w",
adamc@1200 138 "htmlifyFloat_w",
adamc@1200 139 "htmlifyString_w",
adamc@1200 140 "htmlifyBool_w",
adamc@1200 141 "htmlifyTime_w",
adamc@1200 142 "attrifyInt_w",
adamc@1200 143 "attrifyFloat_w",
adamc@1200 144 "attrifyString_w",
adamc@1200 145 "attrifyChar_w",
adamc@1200 146 "urlifyInt_w",
adamc@1200 147 "urlifyFloat_w",
adamc@1200 148 "urlifyString_w",
adamc@1200 149 "urlifyBool_w",
adamc@1200 150 "urlifyChannel_w"]
adamc@765 151
adamc@765 152 val effectful = ref effectfulBase
adamc@765 153 fun setEffectful ls = effectful := S.addList (effectfulBase, ls)
adamc@765 154 fun isEffectful x = S.member (!effectful, x)
adam@1878 155 fun addEffectful x = effectful := S.add (!effectful, x)
adamc@765 156
adamc@1171 157 val benignBase = basis ["get_cookie",
adamc@1171 158 "new_client_source",
adamc@1171 159 "get_client_source",
adamc@1171 160 "set_client_source",
adamc@1171 161 "current",
adamc@1171 162 "alert",
adam@1290 163 "confirm",
adamc@1171 164 "onError",
adamc@1171 165 "onFail",
adamc@1171 166 "onConnectFail",
adamc@1171 167 "onDisconnect",
adamc@1171 168 "onServerError",
adam@1783 169 "mouseEvent",
adam@1783 170 "keyEvent",
adamc@1250 171 "debug",
adam@1422 172 "rand",
adam@1465 173 "now",
adam@1465 174 "getHeader",
adam@1555 175 "setHeader",
adam@1555 176 "spawn",
adam@1555 177 "onClick",
adam@1555 178 "onDblclick",
adam@1555 179 "onKeydown",
adam@1555 180 "onKeypress",
adam@1555 181 "onKeyup",
adam@1555 182 "onMousedown",
adam@1791 183 "onMousemove",
adam@1791 184 "onMouseout",
adam@1791 185 "onMouseover",
adam@1556 186 "onMouseup",
adam@1559 187 "preventDefault",
adam@1559 188 "stopPropagation",
adam@1785 189 "fresh",
adam@1785 190 "giveFocus"]
adamc@1171 191
adamc@1171 192 val benign = ref benignBase
adamc@1171 193 fun setBenignEffectful ls = benign := S.addList (benignBase, ls)
adamc@1171 194 fun isBenignEffectful x = S.member (!benign, x)
adamc@1171 195
adam@1595 196 val clientBase = basis ["get_client_source",
adamc@841 197 "current",
adamc@765 198 "alert",
adam@1290 199 "confirm",
adamc@765 200 "recv",
adamc@765 201 "sleep",
adamc@765 202 "spawn",
adamc@765 203 "onError",
adamc@765 204 "onFail",
adamc@765 205 "onConnectFail",
adamc@765 206 "onDisconnect",
adamc@895 207 "onServerError",
adam@1783 208 "mouseEvent",
adam@1783 209 "keyEvent",
adam@1555 210 "onClick",
adam@1555 211 "onDblclick",
adam@1555 212 "onKeydown",
adam@1555 213 "onKeypress",
adam@1555 214 "onKeyup",
adam@1555 215 "onMousedown",
adam@1791 216 "onMousemove",
adam@1791 217 "onMouseout",
adam@1791 218 "onMouseover",
adam@1559 219 "onMouseup",
adam@1559 220 "preventDefault",
adam@1785 221 "stopPropagation",
adam@1785 222 "giveFocus"]
adamc@765 223 val client = ref clientBase
adamc@765 224 fun setClientOnly ls = client := S.addList (clientBase, ls)
adamc@765 225 fun isClientOnly x = S.member (!client, x)
adamc@765 226
adamc@765 227 val serverBase = basis ["requestHeader",
adamc@765 228 "query",
adamc@765 229 "dml",
adamc@765 230 "nextval",
adamc@1073 231 "setval",
adamc@765 232 "channel",
adam@1787 233 "send",
adam@1787 234 "fieldName",
adam@1787 235 "fieldValue",
adam@1787 236 "remainingFields",
adam@1787 237 "firstFormField"]
adamc@765 238 val server = ref serverBase
adamc@765 239 fun setServerOnly ls = server := S.addList (serverBase, ls)
adamc@765 240 fun isServerOnly x = S.member (!server, x)
adamc@765 241
adamc@765 242 val basisM = foldl (fn ((k, v : string), m) => M.insert (m, ("Basis", k), v)) M.empty
adamc@765 243
adamc@765 244 val jsFuncsBase = basisM [("alert", "alert"),
adam@1599 245 ("stringToTime", "stringToTime"),
adam@1599 246 ("stringToTime_error", "stringToTime_error"),
adam@1609 247 ("timef", "strftime"),
adam@1290 248 ("confirm", "confrm"),
adamc@765 249 ("get_client_source", "sg"),
adamc@841 250 ("current", "scur"),
adamc@765 251 ("htmlifyBool", "bs"),
adamc@765 252 ("htmlifyFloat", "ts"),
adamc@765 253 ("htmlifyInt", "ts"),
adamc@765 254 ("htmlifyString", "eh"),
adamc@765 255 ("new_client_source", "sc"),
adamc@765 256 ("set_client_source", "sv"),
adamc@838 257 ("stringToFloat", "pflo"),
adamc@838 258 ("stringToInt", "pio"),
adamc@765 259 ("stringToFloat_error", "pfl"),
adamc@765 260 ("stringToInt_error", "pi"),
adamc@765 261 ("urlifyInt", "ts"),
adamc@765 262 ("urlifyFloat", "ts"),
adam@1360 263 ("urlifyTime", "ts"),
adamc@765 264 ("urlifyString", "uf"),
adamc@912 265 ("urlifyBool", "ub"),
adamc@765 266 ("recv", "rv"),
adamc@765 267 ("strcat", "cat"),
adamc@765 268 ("intToString", "ts"),
adamc@765 269 ("floatToString", "ts"),
adamc@821 270 ("charToString", "ts"),
adamc@765 271 ("onError", "onError"),
adamc@765 272 ("onFail", "onFail"),
adamc@765 273 ("onConnectFail", "onConnectFail"),
adamc@765 274 ("onDisconnect", "onDisconnect"),
adamc@798 275 ("onServerError", "onServerError"),
adamc@1108 276 ("attrifyString", "atr"),
adamc@798 277 ("attrifyInt", "ts"),
adamc@798 278 ("attrifyFloat", "ts"),
adamc@820 279 ("attrifyBool", "bs"),
adamc@821 280 ("boolToString", "ts"),
adamc@1057 281 ("str1", "id"),
adamc@821 282 ("strsub", "sub"),
adamc@828 283 ("strsuffix", "suf"),
adamc@829 284 ("strlen", "slen"),
adamc@829 285 ("strindex", "sidx"),
adamc@829 286 ("strchr", "schr"),
adamc@831 287 ("substring", "ssub"),
adamc@895 288 ("strcspn", "sspn"),
adam@1624 289 ("strlenGe", "strlenGe"),
adam@1783 290 ("mouseEvent", "uw_mouseEvent"),
adam@1783 291 ("keyEvent", "uw_keyEvent"),
adam@1404 292 ("minTime", "0"),
adamc@1061 293
adamc@1061 294 ("islower", "isLower"),
adamc@1061 295 ("isupper", "isUpper"),
adamc@1061 296 ("isalpha", "isAlpha"),
adamc@1061 297 ("isdigit", "isDigit"),
adamc@1061 298 ("isalnum", "isAlnum"),
adamc@1061 299 ("isblank", "isBlank"),
adamc@1061 300 ("isspace", "isSpace"),
adamc@1061 301 ("isxdigit", "isXdigit"),
adam@1927 302 ("isprint", "isPrint"),
adamc@1061 303 ("tolower", "toLower"),
adamc@1323 304 ("toupper", "toUpper"),
adam@1927 305 ("ord", "ord"),
adamc@1323 306
adamc@1323 307 ("checkUrl", "checkUrl"),
adam@1366 308 ("bless", "bless"),
adam@1366 309
adam@1366 310 ("eq_time", "eq"),
adam@1366 311 ("lt_time", "lt"),
adam@1430 312 ("le_time", "le"),
adam@1430 313
adam@1625 314 ("debug", "uw_debug"),
adam@1625 315 ("naughtyDebug", "uw_debug"),
adam@1487 316
adam@1571 317 ("floatFromInt", "float"),
adam@1571 318 ("ceil", "ceil"),
adam@1571 319 ("trunc", "trunc"),
adam@1571 320 ("round", "round"),
adam@1571 321
adam@1487 322 ("now", "now"),
adam@1487 323 ("timeToString", "showTime"),
adam@1629 324 ("htmlifyTime", "showTimeHtml"),
adam@1514 325 ("toSeconds", "toSeconds"),
adam@1518 326 ("addSeconds", "addSeconds"),
adam@1555 327 ("diffInSeconds", "diffInSeconds"),
adam@1685 328 ("toMilliseconds", "toMilliseconds"),
adam@1685 329 ("diffInMilliseconds", "diffInMilliseconds"),
adam@1555 330
adam@1555 331 ("onClick", "uw_onClick"),
adam@1555 332 ("onDblclick", "uw_onDblclick"),
adam@1555 333 ("onKeydown", "uw_onKeydown"),
adam@1555 334 ("onKeypress", "uw_onKeypress"),
adam@1555 335 ("onKeyup", "uw_onKeyup"),
adam@1555 336 ("onMousedown", "uw_onMousedown"),
adam@1791 337 ("onMousemove", "uw_onMousemove"),
adam@1791 338 ("onMouseout", "uw_onMouseout"),
adam@1791 339 ("onMouseover", "uw_onMouseover"),
adam@1556 340 ("onMouseup", "uw_onMouseup"),
adam@1559 341 ("preventDefault", "uw_preventDefault"),
adam@1559 342 ("stopPropagation", "uw_stopPropagation"),
adam@1556 343
adam@1755 344 ("fresh", "fresh"),
adam@1755 345
adam@1755 346 ("atom", "atom"),
adam@1755 347 ("css_url", "css_url"),
adam@1785 348 ("property", "property"),
adam@1785 349 ("giveFocus", "giveFocus")]
adamc@765 350 val jsFuncs = ref jsFuncsBase
adamc@765 351 fun setJsFuncs ls = jsFuncs := foldl (fn ((k, v), m) => M.insert (m, k, v)) jsFuncsBase ls
adamc@765 352 fun jsFunc x = M.find (!jsFuncs, x)
adam@1433 353 fun allJsFuncs () = M.listItemsi (!jsFuncs)
adamc@765 354
adamc@768 355 datatype pattern_kind = Exact | Prefix
adamc@768 356 datatype action = Allow | Deny
adamc@768 357 type rule = { action : action, kind : pattern_kind, pattern : string }
adamc@768 358
adamc@768 359 datatype path_kind = Any | Url | Table | Sequence | View | Relation | Cookie | Style
adam@1752 360 type rewrite = { pkind : path_kind, kind : pattern_kind, from : string, to : string, hyphenate : bool }
adamc@768 361
adamc@768 362 val rewrites = ref ([] : rewrite list)
adamc@768 363
adamc@768 364 fun subsume (pk1, pk2) =
adamc@768 365 pk1 = pk2
adamc@768 366 orelse pk2 = Any
adamc@768 367 orelse pk2 = Relation andalso (pk1 = Table orelse pk1 = Sequence orelse pk1 = View)
adamc@768 368
adamc@768 369 fun setRewriteRules ls = rewrites := ls
adamc@768 370 fun rewrite pk s =
adamc@768 371 let
adamc@768 372 fun rew (ls : rewrite list) =
adamc@768 373 case ls of
adamc@768 374 [] => s
adamc@768 375 | rewr :: ls =>
adamc@768 376 let
adamc@768 377 fun match () =
adamc@768 378 case #kind rewr of
adamc@768 379 Exact => if #from rewr = s then
adamc@768 380 SOME (size s)
adamc@768 381 else
adamc@768 382 NONE
adamc@768 383 | Prefix => if String.isPrefix (#from rewr) s then
adamc@768 384 SOME (size (#from rewr))
adamc@768 385 else
adamc@768 386 NONE
adamc@768 387 in
adamc@768 388 if subsume (pk, #pkind rewr) then
adamc@768 389 case match () of
adamc@768 390 NONE => rew ls
adam@1752 391 | SOME suffixStart =>
adam@1752 392 let
adam@1752 393 val s = #to rewr ^ String.extract (s, suffixStart, NONE)
adam@1752 394 in
adam@1752 395 if #hyphenate rewr then
adam@1752 396 String.translate (fn #"_" => "-" | ch => str ch) s
adam@1752 397 else
adam@1752 398 s
adam@1752 399 end
adamc@768 400 else
adamc@768 401 rew ls
adamc@768 402 end
adamc@768 403 in
adamc@768 404 rew (!rewrites)
adamc@768 405 end
adamc@768 406
adamc@769 407 val url = ref ([] : rule list)
adamc@769 408 val mime = ref ([] : rule list)
adam@1465 409 val request = ref ([] : rule list)
adam@1465 410 val response = ref ([] : rule list)
adam@1799 411 val env = ref ([] : rule list)
adamc@769 412
adamc@769 413 fun setUrlRules ls = url := ls
adamc@769 414 fun setMimeRules ls = mime := ls
adam@1465 415 fun setRequestHeaderRules ls = request := ls
adam@1465 416 fun setResponseHeaderRules ls = response := ls
adam@1799 417 fun setEnvVarRules ls = env := ls
adamc@769 418
adamc@770 419 fun getUrlRules () = !url
adamc@770 420 fun getMimeRules () = !mime
adam@1465 421 fun getRequestHeaderRules () = !request
adam@1465 422 fun getResponseHeaderRules () = !response
adam@1799 423 fun getEnvVarRules () = !env
adamc@770 424
adamc@769 425 fun check f rules s =
adamc@769 426 let
adamc@769 427 fun chk (ls : rule list) =
adamc@769 428 case ls of
adamc@769 429 [] => false
adamc@769 430 | rule :: ls =>
adamc@769 431 let
adamc@769 432 val matches =
adamc@769 433 case #kind rule of
adamc@769 434 Exact => #pattern rule = s
adamc@769 435 | Prefix => String.isPrefix (#pattern rule) s
adamc@769 436 in
adamc@769 437 if matches then
adamc@769 438 case #action rule of
adamc@769 439 Allow => true
adamc@769 440 | Deny => false
adamc@769 441 else
adamc@769 442 chk ls
adamc@769 443 end
adamc@769 444 in
adamc@769 445 f s andalso chk (!rules)
adamc@769 446 end
adamc@769 447
adamc@769 448 val checkUrl = check (fn _ => true) url
adam@1465 449
adam@1465 450 val validMime = CharVector.all (fn ch => Char.isAlphaNum ch orelse ch = #"/" orelse ch = #"-" orelse ch = #".")
adam@1799 451 val validEnv = CharVector.all (fn ch => Char.isAlphaNum ch orelse ch = #"_" orelse ch = #".")
adam@1465 452
adam@1465 453 val checkMime = check validMime mime
adam@1465 454 val checkRequestHeader = check validMime request
adam@1465 455 val checkResponseHeader = check validMime response
adam@1799 456 val checkEnvVar = check validEnv env
adamc@769 457
adamc@855 458
adamc@855 459 type protocol = {
adamc@855 460 name : string,
adamc@1096 461 compile : string,
adamc@1095 462 linkStatic : string,
adamc@1095 463 linkDynamic : string,
adamc@1164 464 persistent : bool,
adamc@1164 465 code : unit -> Print.PD.pp_desc
adamc@855 466 }
adamc@855 467 val protocols = ref ([] : protocol list)
adamc@855 468 fun addProtocol p = protocols := p :: !protocols
adamc@855 469 fun getProtocol s = List.find (fn p => #name p = s) (!protocols)
adamc@855 470
ezyang@1739 471 fun clibFile s = OS.Path.joinDirFile {dir = libC (),
adamc@855 472 file = s}
adamc@855 473
adamc@865 474 val curProto = ref {name = "",
adamc@1096 475 compile = "",
adamc@1095 476 linkStatic = "",
adamc@1095 477 linkDynamic = "",
adamc@1164 478 persistent = false,
adamc@1164 479 code = fn () => Print.box []}
adamc@856 480 fun setProtocol name =
adamc@856 481 case getProtocol name of
adamc@856 482 NONE => raise Fail ("Unknown protocol " ^ name)
adamc@856 483 | SOME p => curProto := p
adamc@855 484 fun currentProtocol () = !curProto
adamc@855 485
adamc@857 486 val debug = ref false
adamc@857 487 fun setDebug b = debug := b
adamc@857 488 fun getDebug () = !debug
adamc@857 489
adamc@867 490 datatype sql_type =
adamc@867 491 Int
adamc@867 492 | Float
adamc@867 493 | String
adamc@1011 494 | Char
adamc@867 495 | Bool
adamc@867 496 | Time
adamc@867 497 | Blob
adamc@867 498 | Channel
adamc@867 499 | Client
adamc@867 500 | Nullable of sql_type
adamc@867 501
adamc@873 502 fun p_sql_ctype t =
adamc@867 503 let
adamc@867 504 open Print.PD
adamc@867 505 open Print
adamc@867 506 in
adamc@867 507 case t of
adamc@870 508 Int => "uw_Basis_int"
adamc@870 509 | Float => "uw_Basis_float"
adamc@870 510 | String => "uw_Basis_string"
adamc@1011 511 | Char => "uw_Basis_char"
adamc@870 512 | Bool => "uw_Basis_bool"
adamc@870 513 | Time => "uw_Basis_time"
adamc@870 514 | Blob => "uw_Basis_blob"
adamc@870 515 | Channel => "uw_Basis_channel"
adamc@870 516 | Client => "uw_Basis_client"
adamc@870 517 | Nullable String => "uw_Basis_string"
adamc@873 518 | Nullable t => p_sql_ctype t ^ "*"
adamc@867 519 end
adamc@867 520
adamc@867 521 fun isBlob Blob = true
adamc@867 522 | isBlob (Nullable t) = isBlob t
adamc@867 523 | isBlob _ = false
adamc@867 524
adamc@870 525 fun isNotNull (Nullable _) = false
adamc@870 526 | isNotNull _ = true
adamc@870 527
adam@1293 528 datatype failure_mode = Error | None
adam@1293 529
adamc@866 530 type dbms = {
adamc@866 531 name : string,
adam@1682 532 randomFunction : string,
adamc@866 533 header : string,
adamc@866 534 link : string,
adamc@873 535 p_sql_type : sql_type -> string,
adamc@870 536 init : {dbstring : string,
adamc@870 537 prepared : (string * int) list,
adamc@870 538 tables : (string * (string * sql_type) list) list,
adamc@872 539 views : (string * (string * sql_type) list) list,
adamc@870 540 sequences : string list} -> Print.PD.pp_desc,
adamc@873 541 query : {loc : ErrorMsg.span, cols : sql_type list,
adamc@880 542 doCols : ({loc : ErrorMsg.span, wontLeakStrings : bool, col : int, typ : sql_type} -> Print.PD.pp_desc)
adamc@867 543 -> Print.PD.pp_desc}
adamc@867 544 -> Print.PD.pp_desc,
adamc@867 545 queryPrepared : {loc : ErrorMsg.span, id : int, query : string,
adamc@873 546 inputs : sql_type list, cols : sql_type list,
adamc@880 547 doCols : ({loc : ErrorMsg.span, wontLeakStrings : bool, col : int,
adamc@880 548 typ : sql_type} -> Print.PD.pp_desc)
adamc@879 549 -> Print.PD.pp_desc,
adamc@879 550 nested : bool}
adamc@868 551 -> Print.PD.pp_desc,
adam@1293 552 dml : ErrorMsg.span * failure_mode -> Print.PD.pp_desc,
adamc@868 553 dmlPrepared : {loc : ErrorMsg.span, id : int, dml : string,
adam@1293 554 inputs : sql_type list, mode : failure_mode} -> Print.PD.pp_desc,
adamc@878 555 nextval : {loc : ErrorMsg.span, seqName : string option, seqE : Print.PD.pp_desc} -> Print.PD.pp_desc,
adamc@874 556 nextvalPrepared : {loc : ErrorMsg.span, id : int, query : string} -> Print.PD.pp_desc,
adamc@1073 557 setval : {loc : ErrorMsg.span, seqE : Print.PD.pp_desc, count : Print.PD.pp_desc} -> Print.PD.pp_desc,
adamc@874 558 sqlifyString : string -> string,
adamc@874 559 p_cast : string * sql_type -> string,
adamc@874 560 p_blank : int * sql_type -> string,
adamc@877 561 supportsDeleteAs : bool,
adamc@886 562 supportsUpdateAs : bool,
adamc@877 563 createSequence : string -> string,
adamc@878 564 textKeysNeedLengths : bool,
adamc@879 565 supportsNextval : bool,
adamc@882 566 supportsNestedPrepared : bool,
adamc@890 567 sqlPrefix : string,
adamc@1014 568 supportsOctetLength : bool,
adamc@1014 569 trueString : string,
adamc@1196 570 falseString : string,
adamc@1196 571 onlyUnion : bool,
adam@1777 572 nestedRelops : bool,
adam@1778 573 windowFunctions: bool
adamc@866 574 }
adamc@866 575
adamc@866 576 val dbmses = ref ([] : dbms list)
adamc@866 577 val curDb = ref ({name = "",
adam@1682 578 randomFunction = "",
adamc@866 579 header = "",
adamc@866 580 link = "",
adamc@873 581 p_sql_type = fn _ => "",
adamc@867 582 init = fn _ => Print.box [],
adamc@867 583 query = fn _ => Print.box [],
adamc@868 584 queryPrepared = fn _ => Print.box [],
adamc@868 585 dml = fn _ => Print.box [],
adamc@869 586 dmlPrepared = fn _ => Print.box [],
adamc@869 587 nextval = fn _ => Print.box [],
adamc@874 588 nextvalPrepared = fn _ => Print.box [],
adamc@1073 589 setval = fn _ => Print.box [],
adamc@874 590 sqlifyString = fn s => s,
adamc@874 591 p_cast = fn _ => "",
adamc@874 592 p_blank = fn _ => "",
adamc@877 593 supportsDeleteAs = false,
adamc@886 594 supportsUpdateAs = false,
adamc@877 595 createSequence = fn _ => "",
adamc@878 596 textKeysNeedLengths = false,
adamc@879 597 supportsNextval = false,
adamc@882 598 supportsNestedPrepared = false,
adamc@890 599 sqlPrefix = "",
adamc@1014 600 supportsOctetLength = false,
adamc@1014 601 trueString = "",
adamc@1196 602 falseString = "",
adamc@1196 603 onlyUnion = false,
adam@1777 604 nestedRelops = false,
adam@1777 605 windowFunctions = false} : dbms)
adamc@866 606
adamc@866 607 fun addDbms v = dbmses := v :: !dbmses
adamc@866 608 fun setDbms s =
adamc@866 609 case List.find (fn db => #name db = s) (!dbmses) of
adamc@866 610 NONE => raise Fail ("Unknown DBMS " ^ s)
adamc@866 611 | SOME db => curDb := db
adamc@866 612 fun currentDbms () = !curDb
adamc@866 613
adamc@891 614 val dbstring = ref (NONE : string option)
adamc@891 615 fun setDbstring so = dbstring := so
adamc@891 616 fun getDbstring () = !dbstring
adamc@891 617
adamc@891 618 val exe = ref (NONE : string option)
adamc@891 619 fun setExe so = exe := so
adamc@891 620 fun getExe () = !exe
adamc@891 621
adamc@891 622 val sql = ref (NONE : string option)
adamc@891 623 fun setSql so = sql := so
adamc@891 624 fun getSql () = !sql
adamc@891 625
adam@1820 626 val coreInline = ref 5
adamc@1016 627 fun setCoreInline n = coreInline := n
adamc@1016 628 fun getCoreInline () = !coreInline
adamc@1016 629
adam@1820 630 val monoInline = ref 5
adamc@1016 631 fun setMonoInline n = monoInline := n
adamc@1016 632 fun getMonoInline () = !monoInline
adamc@1016 633
adamc@1095 634 val staticLinking = ref false
adamc@1095 635 fun setStaticLinking b = staticLinking := b
adamc@1095 636 fun getStaticLinking () = !staticLinking
adamc@1095 637
adamc@1114 638 val deadlines = ref false
adamc@1114 639 fun setDeadlines b = deadlines := b
adamc@1114 640 fun getDeadlines () = !deadlines
adamc@1114 641
adamc@1164 642 val sigFile = ref (NONE : string option)
adamc@1164 643 fun setSigFile v = sigFile := v
adamc@1164 644 fun getSigFile () = !sigFile
adamc@1164 645
adamc@1183 646 structure SS = BinarySetFn(struct
adamc@1183 647 type ord_key = string
adamc@1183 648 val compare = String.compare
adamc@1183 649 end)
adamc@1183 650
adamc@1183 651 val safeGet = ref SS.empty
adamc@1183 652 fun setSafeGets ls = safeGet := SS.addList (SS.empty, ls)
adamc@1183 653 fun isSafeGet x = SS.member (!safeGet, x)
adamc@1183 654
adam@1294 655 val onError = ref (NONE : (string * string list * string) option)
adam@1294 656 fun setOnError x = onError := x
adam@1294 657 fun getOnError () = !onError
adam@1294 658
adam@1307 659 val limits = ["messages", "clients", "headers", "page", "heap", "script",
adam@1307 660 "inputs", "subinputs", "cleanup", "deltas", "transactionals",
adam@1308 661 "globals", "database", "time"]
adam@1307 662
adam@1307 663 val limitsList = ref ([] : (string * int) list)
adam@1307 664 fun addLimit (v as (name, _)) =
adam@1307 665 if List.exists (fn name' => name' = name) limits then
adam@1308 666 (limitsList := v :: !limitsList;
adam@1308 667 if name = "time" then
adam@1308 668 setDeadlines true
adam@1308 669 else
adam@1308 670 ())
adam@1307 671 else
adam@1307 672 raise Fail ("Unknown limit category '" ^ name ^ "'")
adam@1307 673 fun limits () = !limitsList
adam@1307 674
adam@1332 675 val minHeap = ref 0
adam@1332 676 fun setMinHeap n = if n >= 0 then minHeap := n else raise Fail "Trying to set negative minHeap"
adam@1332 677 fun getMinHeap () = !minHeap
adam@1332 678
adam@1393 679 structure SS = BinarySetFn(struct
adam@1393 680 type ord_key = string
adam@1393 681 val compare = String.compare
adam@1393 682 end)
adam@1393 683
adam@1393 684 val alwaysInline = ref SS.empty
adam@1393 685 fun addAlwaysInline s = alwaysInline := SS.add (!alwaysInline, s)
adam@1393 686 fun checkAlwaysInline s = SS.member (!alwaysInline, s)
adam@1393 687
adam@1478 688 val noXsrfProtection = ref SS.empty
adam@1478 689 fun addNoXsrfProtection s = noXsrfProtection := SS.add (!noXsrfProtection, s)
adam@1478 690 fun checkNoXsrfProtection s = SS.member (!noXsrfProtection, s)
adam@1478 691
adam@1629 692 val timeFormat = ref "%c"
adam@1629 693 fun setTimeFormat v = timeFormat := v
adam@1629 694 fun getTimeFormat () = !timeFormat
adam@1629 695
adamc@765 696 end