matitaB/matita/matitaAuthentication.ml

   1 (* Copyright (C) 2004-2011, HELM Team.
   2  *
   3  * This file is part of HELM, an Hypertextual, Electronic
   4  * Library of Mathematics, developed at the Computer Science
   5  * Department, University of Bologna, Italy.
   6  *
   7  * HELM is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * HELM is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with HELM; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20  * MA  02111-1307, USA.
  21  *
  22  * For details, see the HELM World-Wide-Web page,
  23  * http://helm.cs.unibo.it/
  24  *)
  25
  26 type session_id = Uuidm.t
  27
  28 (* user table: user id, (password, optional session id) *)
  29 type user = string * (string * session_id option)
  30
  31 let user_tbl = (ref [] : user list ref)
  32
  33 (* session table: (user id, session id), matita status *)
  34 type session = session_id * (MatitaEngine.status * MatitaEngine.status list)
  35
  36 let session_tbl = (ref [] : session list ref)
  37
  38 exception UsernameCollision of string
  39
  40 let lookup_user uid = List.assoc uid !user_tbl
  41
  42 let create_session uid =
  43   let status = new MatitaEngine.status (Some uid) "cic:/matita" in
  44   let history = [status] in
  45   let pw,sid = List.assoc uid !user_tbl in
  46   let clean_utbl = List.remove_assoc uid !user_tbl in
  47   let new_session = Uuidm.create `V4 in
  48   user_tbl := (uid,(pw,Some new_session))::clean_utbl;
  49   let clean_stbl = match sid with
  50     | Some sid' ->
  51        List.remove_assoc sid' !session_tbl
  52     | _ -> !session_tbl
  53   in
  54   session_tbl := (new_session,(status,history))::clean_stbl;
  55   new_session
  56 ;;
  57
  58 let get_status sid = fst (List.assoc sid !session_tbl)
  59
  60 let get_history sid = snd (List.assoc sid !session_tbl)
  61
  62 let set_status sid st =
  63   let oldst, hist = List.assoc sid !session_tbl in
  64   session_tbl := (sid,(st,hist))::(List.remove_assoc sid !session_tbl)
  65
  66 let set_history sid hist =
  67   let st, oldhist = List.assoc sid !session_tbl in
  68   session_tbl := (sid,(st,hist))::(List.remove_assoc sid !session_tbl)
  69
  70 let logout_user uid =
  71   match List.assoc uid !user_tbl with
  72   | _,None -> ()
  73   | pw, Some sid ->
  74      user_tbl := (uid,(pw,None))::List.remove_assoc uid !user_tbl;
  75      session_tbl := List.remove_assoc sid !session_tbl
  76 ;;
  77
  78 let remove_user uid =
  79   user_tbl := List.remove_assoc uid !user_tbl
  80 ;;
  81
  82 (* serialization and deserialization of the user table *)
  83 let config_path () =
  84   let path = Helm_registry.get "matita.basedir" in
  85   let dirname = Filename.dirname path in
  86   HExtlib.mkdir dirname;
  87   path
  88 ;;
  89
  90 let serialize () =
  91   let clean_utbl = List.map (fun (uid,(pw,_)) -> uid,(pw,None)) !user_tbl in
  92   let utbl_ch = open_out (config_path () ^ "/usertable.dump") in
  93   Marshal.to_channel utbl_ch clean_utbl [];
  94   close_out utbl_ch;
  95 ;;
  96
  97 let deserialize () =
  98   let utbl_ch = open_in (config_path () ^ "/usertable.dump") in
  99   user_tbl := Marshal.from_channel utbl_ch;
 100   close_in utbl_ch;
 101   (* old_sessions are now invalid *)
 102   session_tbl := [];
 103 ;;
 104
 105 let add_user uid pw =
 106   try
 107     let _ = lookup_user uid in
 108     raise (UsernameCollision uid)
 109   with Not_found ->
 110     user_tbl := (uid,(pw,None))::!user_tbl;
 111     serialize ()
 112 ;;