versaplexd/vxsqlpool.cs

   1 /*
   2  * Versaplex:
   3  *   Copyright (C)2007-2008 Versabanq Innovations Inc. and contributors.
   4  *       See the included file named LICENSE for license information.
   5  */
   6 using System;
   7 using System.Data.Common;
   8 using System.Collections;
   9 using System.Collections.Generic;
  10 using System.IO;
  11 using Wv;
  12 using Wv.Extensions;
  13
  14 public static class VxSqlPool
  15 {
  16     private static WvIni inifile;
  17
  18     public static void SetIniFile(string path)
  19     {
  20         if (path.e() || !File.Exists(path))
  21             throw new Exception
  22                 (wv.fmt("Could not find config file '{0}'", path));
  23         inifile = new WvIni(path);
  24     }
  25
  26     private static string find_connection_moniker(string connid)
  27     {
  28         // At the moment, a connection ID is just a username
  29         string dbname = inifile.get("User Map", connid);
  30         if (dbname == null)
  31             dbname = inifile.get("User Map", "*"); // try default
  32         if (dbname == null)
  33             throw new VxConfigException(
  34                 String.Format("No user '{0}' found.",
  35                 connid));
  36
  37         string cfgval = inifile.get("Connections", dbname);
  38         if (cfgval == null)
  39             throw new VxConfigException(String.Format(
  40                 "No connection found for user {0}", connid));
  41
  42         return cfgval;
  43     }
  44
  45     public static string GetUsernameForCert(string cert)
  46     {
  47         string username = inifile.get("Cert Map", cert);
  48         if (username == null)
  49             throw new VxConfigException(String.Format(
  50                 "No user found for cert with fingerprint '{0}'.", cert));
  51
  52         return username;
  53     }
  54
  55     static Dictionary<string, int> userpermissions =
  56         new Dictionary<string, int>();
  57
  58     public static int access_restrictions(string connid)
  59     {
  60         int security_level;
  61         if (!userpermissions.TryGetValue(connid, out security_level))
  62         {
  63             string iniseclevel = inifile.get("Security Level", connid);
  64
  65             // If we found *nothing* perhaps we should evaluate a judgement
  66             // on security, see if '*' has values... assuming we're  not
  67             // already looking for '*''s security.
  68             if (iniseclevel == null)
  69             {
  70                 security_level = 2;
  71                 if (connid != "*")
  72                 {
  73                     iniseclevel = inifile.get("Security Level", "*");
  74
  75                     if (iniseclevel != null &&
  76                         (iniseclevel == "0" ||
  77                         iniseclevel == "1"))
  78                         security_level = Convert.ToInt32(iniseclevel);
  79                 }
  80             }
  81             else
  82                 security_level = !(iniseclevel == "0" ||
  83                     iniseclevel == "1") ? 2 :
  84                         Convert.ToInt32(iniseclevel);
  85
  86             userpermissions[connid] = security_level;
  87         }
  88         return security_level;
  89     }
  90
  91     public static WvDbi create(string connid)
  92     {
  93         try
  94         {
  95             return WvDbi.create(find_connection_moniker(connid));
  96         }
  97         catch (DbException e)
  98         {
  99             throw new VxConfigException("Connect: " + e.Message);
 100         }
 101     }
 102 }