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 type matita_file = MatitaFilesystem.matita_flag * string
  29
  30 (* user table: user id, (password, optional session id) *)
  31 type user = string * (string * session_id option)
  32
  33 let user_tbl = (ref [] : user list ref)
  34
  35 (* session table: session id, (user id, matita status, matita history *)
  36 type session = session_id * (string * MatitaEngine.status * MatitaEngine.status list)
  37
  38 let session_tbl = (ref [] : session list ref)
  39
  40 exception UsernameCollision of string
  41
  42 let lookup_user uid = List.assoc uid !user_tbl
  43
  44 let user_of_session sid = let res,_,_ = List.assoc sid !session_tbl in res
  45
  46 let create_session uid =
  47   let status = new MatitaEngine.status (Some uid) "cic:/matita" in
  48   let history = [status] in
  49   let pw,sid = List.assoc uid !user_tbl in
  50   let clean_utbl = List.remove_assoc uid !user_tbl in
  51   let new_session = Uuidm.create `V4 in
  52   user_tbl := (uid,(pw,Some new_session))::clean_utbl;
  53   let clean_stbl = match sid with
  54     | Some sid' ->
  55        List.remove_assoc sid' !session_tbl
  56     | _ -> !session_tbl
  57   in
  58   session_tbl := (new_session,(uid,status,history))::clean_stbl;
  59   new_session
  60 ;;
  61
  62 let get_session_owner sid =
  63   let uid,_,_ = List.assoc sid !session_tbl
  64   in uid
  65
  66 let get_status sid =
  67   let _,st,_ = List.assoc sid !session_tbl
  68   in st
  69
  70 let get_history sid =
  71   let _,_,hist = List.assoc sid !session_tbl
  72   in hist
  73
  74 let set_status sid st =
  75   let uid, oldst, hist = List.assoc sid !session_tbl in
  76   session_tbl := (sid,(uid,st,hist))::(List.remove_assoc sid !session_tbl)
  77
  78 let set_history sid hist =
  79   let uid, st, oldhist = List.assoc sid !session_tbl in
  80   session_tbl := (sid,(uid,st,hist))::(List.remove_assoc sid !session_tbl)
  81
  82 let logout_user sid =
  83   let uid,st,hist = List.assoc sid !session_tbl in
  84   let pw,_ = List.assoc uid !user_tbl in
  85   user_tbl := (uid,(pw,None))::List.remove_assoc uid !user_tbl;
  86   session_tbl := List.remove_assoc sid !session_tbl
  87 ;;
  88
  89 let remove_user uid =
  90   user_tbl := List.remove_assoc uid !user_tbl
  91 ;;
  92
  93 (* serialization and deserialization of the user table *)
  94 let config_path () =
  95   let path = Helm_registry.get "matita.basedir" in
  96   let dirname = Filename.dirname path in
  97   HExtlib.mkdir dirname;
  98   path
  99 ;;
 100
 101 let serialize () =
 102   let clean_utbl = List.map (fun (uid,(pw,_)) -> uid,(pw,None)) !user_tbl in
 103   let utbl_ch = open_out (config_path () ^ "/usertable.dump") in
 104   Marshal.to_channel utbl_ch clean_utbl [];
 105   close_out utbl_ch;
 106 ;;
 107
 108 let deserialize () =
 109   (try
 110     let utbl_ch = open_in (config_path () ^ "/usertable.dump") in
 111     user_tbl := Marshal.from_channel utbl_ch;
 112     close_in utbl_ch;
 113   with
 114     | Sys_error _ ->
 115        user_tbl := []; serialize());
 116   (* old_sessions are now invalid *)
 117   session_tbl := [];
 118 ;;
 119
 120 let write_ft uid ft =
 121   let ft_ch = open_out (config_path () ^ "/ft_" ^ uid ^ ".dump") in
 122   Marshal.to_channel ft_ch ft [];
 123   close_out ft_ch;
 124 ;;
 125
 126 let read_ft uid =
 127   try
 128     let ft_ch = open_in (config_path () ^ "/ft_" ^ uid ^ ".dump") in
 129     let ft = Marshal.from_channel ft_ch in
 130     close_in ft_ch;
 131     ft
 132   with
 133     | Sys_error _ ->
 134       (* this is an error, we should rebuild the table by a diff of
 135          the directory listing and svn stat *)
 136       []
 137 ;;
 138
 139 let set_file_flag uid filename flag =
 140   let filename = MatitaFilesystem.normalize_qfn filename in
 141   let ft = read_ft uid in
 142   let oldflag =
 143     try List.assoc filename ft
 144     with Not_found -> MatitaFilesystem.MSynchronized
 145   in
 146   if oldflag <> MatitaFilesystem.MConflict
 147     then
 148       let ft = (filename,flag)::
 149         List.filter (fun (fn,_) -> fn <> filename) ft in
 150       write_ft uid ft
 151     else ()
 152 ;;
 153
 154 let add_user uid pw =
 155   try
 156     let _ = lookup_user uid in
 157     raise (UsernameCollision uid)
 158   with Not_found ->
 159     let ft = MatitaFilesystem.checkout uid in
 160     user_tbl := (uid,(pw,None))::!user_tbl;
 161     write_ft uid ft;
 162     serialize ()
 163 ;;
 164
 165 let reset () =
 166   user_tbl := [];
 167   session_tbl := [];
 168   MatitaFilesystem.reset_lib ();
 169   serialize ();
 170 ;;