Import ezmlm-idx 0.40

[ezmlm] / sub_pgsql / putsubs.c

diff --git a/sub_pgsql/putsubs.c b/sub_pgsql/putsubs.c
new file mode 100644 (file)
index 0000000..c53213c
--- /dev/null
+++ b/sub_pgsql/putsubs.c
@@ -0,0 +1,128 @@
+#include "error.h"
+#include "strerr.h"
+#include "readwrite.h"
+#include "str.h"
+#include "fmt.h"
+#include "stralloc.h"
+#include "open.h"
+#include "substdio.h"
+#include "case.h"
+#include "errtxt.h"
+#include "subscribe.h"
+#include "qmail.h"
+#include <unistd.h>
+#include <libpq-fe.h>
+
+static substdio ssin;
+static char inbuf[512];
+char strnum[FMT_ULONG];
+static stralloc line = {0};
+static stralloc fn = {0};
+
+static void die_nomem(fatal)
+char *fatal;
+{
+  strerr_die2x(111,fatal,ERR_NOMEM);
+}
+
+static void die_write(fatal)
+char *fatal;
+{
+  strerr_die3x(111,fatal,ERR_WRITE,"stdout");
+}
+
+unsigned long putsubs(dbname,hash_lo,hash_hi,
+       subwrite,flagsql,fatal)
+/* Outputs all userhostesses in 'dbname' to stdout. If userhost is not null */
+/* that userhost is excluded. 'dbname' is the base directory name. For the  */
+/* mysql version, dbname is the directory where the file "sql" with mysql   */
+/* access info is found. If this file is not present or if flagmysql is not */
+/* set, the routine falls back to the old database style. subwrite must be a*/
+/* function returning >=0 on success, -1 on error, and taking arguments     */
+/* (char* string, unsigned int length). It will be called once per address  */
+/* and should take care of newline or whatever needed for the output form.  */
+
+char *dbname;          /* database base dir */
+unsigned long hash_lo;
+unsigned long hash_hi;
+int subwrite();                /* write function. */
+int flagsql;
+char *fatal;           /* fatal error string */
+
+{
+  PGresult *result;
+  int row_nr;
+  int length;
+  char *row;
+  char *table = (char *) 0;
+
+  unsigned int i;
+  int fd;
+  unsigned long no = 0L;
+  int match;
+  unsigned int hashpos;
+  char *ret = (char *) 0;
+
+  if (!flagsql || (ret = opensql(dbname,&table))) {
+    if (flagsql && *ret) strerr_die2x(111,fatal,ret);
+                                               /* fallback to local db */
+    if (!stralloc_copys(&fn,dbname)) die_nomem(fatal);
+    if (!stralloc_catb(&fn,"/subscribers/?",15)) die_nomem(fatal);
+                               /* NOTE: Also copies terminal '\0' */
+    hashpos = fn.len - 2;
+    if (hash_lo > 52) hash_lo = 52;
+    if (hash_hi > 52) hash_hi = 52;
+    if (hash_hi < hash_lo) hash_hi = hash_lo;
+
+    for (i = hash_lo;i <= hash_hi;++i) {
+      fn.s[hashpos] = 64 + i;  /* hash range 0-52 */
+      fd = open_read(fn.s);
+      if (fd == -1) {
+        if (errno != error_noent)
+         strerr_die4sys(111,fatal,ERR_READ,fn.s,": ");
+      } else {
+        substdio_fdbuf(&ssin,read,fd,inbuf,sizeof(inbuf));
+        for (;;) {
+          if (getln(&ssin,&line,&match,'\0') == -1)
+            strerr_die4sys(111,fatal,ERR_READ,fn.s,": ");
+          if (!match)
+            break;
+          if (subwrite(line.s + 1,line.len - 2) == -1) die_write(fatal);
+          no++;
+        }
+        close(fd);
+      }
+    }
+    return no;
+
+  } else {                                     /* SQL Version */
+
+                                               /* main query */
+    if (!stralloc_copys(&line,"SELECT address FROM "))
+               die_nomem(fatal);
+    if (!stralloc_cats(&line,table)) die_nomem(fatal);
+    if (!stralloc_cats(&line," WHERE hash BETWEEN ")) die_nomem(fatal);
+    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_lo)))
+               die_nomem(fatal);
+    if (!stralloc_cats(&line," AND ")) die_nomem(fatal);
+    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_hi)))
+      die_nomem(fatal);
+    if (!stralloc_0(&line)) die_nomem(fatal);
+    result = PQexec(psql,line.s);
+    if (result == NULL)
+      strerr_die2x(111,fatal,PQerrorMessage(psql));
+    if (PQresultStatus(result) != PGRES_TUPLES_OK)
+      strerr_die2x(111,fatal,PQresultErrorMessage(result));
+
+    no = 0;
+    for (row_nr=0;row_nr<PQntuples(result);row_nr++) {
+      /* this is safe even if someone messes with the address field def */
+      length = PQgetlength(result,row_nr,0);
+      row = PQgetvalue(result,row_nr,0);
+      if (subwrite(row,length) == -1) die_write(fatal);
+      no++;                                    /* count for list-list fxn */
+    }
+    PQclear(result);
+    return no;
+  }
+}