Retrieve quotas from SQL database

@@ -17,7 +17,7 @@
 
 #include <mail.local.h>
 
-#ifdef USE_DBM
+#if defined (USE_DBM) || defined (USE_SQL)
 
 #define DEFRETVAL MQUOTA_UNLIMITED
 
@@ -34,14 +34,17 @@ get_size (char *str, size_t *size, char **endp)
     {
     case 0:
       break;
+      
     case 'k':
     case 'K':
       s *= 1024;
       break;
+      
     case 'm':
     case 'M':
       s *= 1024*1024;
       break;
+      
     default:
       *endp = str;
       return -1;
@@ -50,18 +53,29 @@ get_size (char *str, size_t *size, char **endp)
   return 0;
 }
 
+#define RETR_OK        0
+#define RETR_UNLIMITED -1
+#define RETR_FAILURE   1
+
 int
-check_quota (char *name, size_t size, size_t *rest)
+fail_retrieve_quota (char *name, size_t *quota)
+{
+  mu_error (_("No quota retrieving mechanism"));
+  return RETR_FAILURE;
+}
+
+#ifdef USE_DBM
+int
+dbm_retrieve_quota (char *name, size_t *quota)
 {
   DBM_FILE db;
   DBM_DATUM named, contentd;
-  size_t quota;
   char buffer[64];
   int unlimited = 0;
   int rc;
 
   if (!quotadbname || mu_dbm_open (quotadbname, &db, MU_STREAM_READ, 0600)) 
-    return DEFRETVAL;
+    return RETR_FAILURE;
 
   memset (&named, 0, sizeof named);
   memset (&contentd, 0, sizeof contentd);
@@ -78,10 +92,10 @@ check_quota (char *name, size_t size, size_t *rest)
       if (rc)
 	{
 	  /*mu_error (_("can't fetch data: %s"), strerror (rc));*/
-	  return DEFRETVAL;
+	  return RETR_FAILURE;
 	}
       if (!MU_DATUM_PTR (contentd))
-	return DEFRETVAL;
+	return RETR_FAILURE;
     }
 
   if (strncasecmp("none",
@@ -92,7 +106,7 @@ check_quota (char *name, size_t size, size_t *rest)
     {
       mu_error (_("mailbox quota for `%s' is too big: %d digits"),
 		name, MU_DATUM_SIZE (contentd));
-      quota = groupquota;
+      *quota = groupquota;
     }
   else
     {
@@ -100,24 +114,171 @@ check_quota (char *name, size_t size, size_t *rest)
 		
       strncpy(buffer, MU_DATUM_PTR (contentd), MU_DATUM_SIZE (contentd));
       buffer[MU_DATUM_SIZE (contentd)] = 0;
-      quota = strtoul (buffer, &p, 0);
-      if (get_size (buffer, &quota, &p))
+      *quota = strtoul (buffer, &p, 0);
+      if (get_size (buffer, quota, &p))
 	{
 	  mu_error (_("bogus mailbox quota for `%s' (near `%s')"), name, p);
-	  quota = groupquota;
+	  *quota = groupquota;
 	}
     }
 
   mu_dbm_close (db);
-  if (unlimited) 
+  
+  return unlimited ? RETR_UNLIMITED : RETR_OK;
+}
+
+# define default_retrieve_quota dbm_retrieve_quota
+#else
+# define default_retrieve_quota fail_retrieve_quota
+#endif
+
+#ifdef USE_SQL
+#include <mailutils/sql.h>
+
+/* FIXME: defined in auth/sql.c */
+extern int sql_interface;
+extern char *mu_sql_host;
+extern char *mu_sql_user;
+extern char *mu_sql_passwd;
+extern char *mu_sql_db;
+extern int  mu_sql_port;
+extern char *mu_sql_expand_query (const char *query, const char *ustr);
+
+int
+sql_retrieve_quota (char *name, size_t *quota)
+{
+  mu_sql_connection_t conn;
+  char *query_str;
+  int rc, status;
+  char *tmp;
+  size_t n;
+  
+  query_str = mu_sql_expand_query (quota_query, name);
+  if (!query_str)
+    return RETR_FAILURE;
+
+  status = mu_sql_connection_init (&conn,
+				   sql_interface,
+				   mu_sql_host,
+				   mu_sql_port,
+				   mu_sql_user,
+				   mu_sql_passwd,
+				   mu_sql_db);
+  
+  if (status)
+    {
+      mu_error ("%s. SQL error: %s",
+		mu_strerror (status), mu_sql_strerror (conn));
+      mu_sql_connection_destroy (&conn);
+      free (query_str);
+      return RETR_FAILURE;
+    }
+
+  status = mu_sql_connect (conn);
+
+  if (status)
+    {
+      mu_error ("%s. SQL error: %s",
+		mu_strerror (status), mu_sql_strerror (conn));
+      mu_sql_connection_destroy (&conn);
+      free (query_str);
+      return  RETR_FAILURE;
+    }
+  
+  status = mu_sql_query (conn, query_str);
+  free (query_str);
+  
+  if (status)
+    {
+      mu_error (_("SQL Query failed: %s"),
+		(status == MU_ERR_SQL) ?  mu_sql_strerror (conn) :
+	 	                          mu_strerror (status));
+      mu_sql_connection_destroy (&conn);
+      return  RETR_FAILURE;
+    }
+
+  status = mu_sql_store_result (conn);
+
+  if (status)
+    {
+      mu_error (_("cannot store SQL result: %s"),
+		(status == MU_ERR_SQL) ?  mu_sql_strerror (conn) :
+	 	                          mu_strerror (status));
+      mu_sql_connection_destroy (&conn);
+      return RETR_FAILURE;
+    }
+
+  mu_sql_num_tuples (conn, &n);
+  if (n == 0)
+    {
+      rc = RETR_FAILURE;
+    }
+  else
+    {
+      rc = RETR_OK;
+      status = mu_sql_get_column (conn, 0, 0, &tmp);
+      if (status)
+	{
+	  mu_error (_("cannot retrieve SQL quota: %s"),
+		    (status == MU_ERR_SQL) ?  mu_sql_strerror (conn) :
+		    mu_strerror (status));
+	  rc = RETR_FAILURE;
+	}
+      else if (strcasecmp (tmp, "none") == 0)
+	rc = RETR_UNLIMITED;
+      else
+	{
+	  char *p;
+	  
+	  if (get_size (tmp, quota, &p))
+	    {
+	      mu_error (_("bogus mailbox quota for `%s' (near `%s')"), name, p);
+	      *quota = groupquota;
+	    }
+	}
+    }
+  
+  mu_sql_release_result (conn);
+  mu_sql_disconnect (conn);
+  mu_sql_connection_destroy (&conn);
+  return rc;
+}
+#endif
+
+
+static int
+retrieve_quota (char *name, size_t *quota)
+{
+#ifdef USE_SQL
+  if (quota_query)
+    return sql_retrieve_quota (name, quota);
+#endif
+  return default_retrieve_quota (name, quota);
+}
+
+int
+check_quota (char *name, size_t size, size_t *rest)
+{
+  size_t quota;
+  
+  switch (retrieve_quota (name, &quota))
+    {
+    case RETR_FAILURE:
+      return DEFRETVAL;
+      
+    case RETR_UNLIMITED:
       return MQUOTA_UNLIMITED;
-  else if (quota < size)  /* Mailbox full */
+
+    case RETR_OK:
+      if (quota < size)  /* Mailbox full */
 	return MQUOTA_EXCEEDED;
 	
       if (rest)
 	*rest = quota - size;
+    }
   
   return MQUOTA_OK;
+      
 }
   
-#endif
+#endif /* USE_MAIL_QUOTA */