X-Git-Url: https://git.distorted.org.uk/~mdw/ezmlm/blobdiff_plain/5b62e993b0af39700031c2875d7f6654e6a02850..f8beb284087c279acfb30506f5bb32baa4949b44:/sub_pgsql/tagmsg.c

diff --git a/sub_pgsql/tagmsg.c b/sub_pgsql/tagmsg.c
new file mode 100644
index 0000000..a99797a
--- /dev/null
+++ b/sub_pgsql/tagmsg.c
@@ -0,0 +1,115 @@
+/*$Id: tagmsg.c,v 1.2 1999/10/07 23:31:01 lindberg Exp $*/
+/*$Name: ezmlm-idx-040 $*/
+#include "stralloc.h"
+#include "scan.h"
+#include "fmt.h"
+#include "strerr.h"
+#include "cookie.h"
+#include "slurp.h"
+#include "errtxt.h"
+#include "subscribe.h"
+#include "makehash.h"
+#include <unistd.h>
+#include <libpq-fe.h>
+
+static stralloc line = {0};
+static stralloc key = {0};
+static char hash[COOKIE];
+static char strnum[FMT_ULONG];	/* message number as sz */
+
+static void die_nomem(fatal)
+char *fatal;
+{
+  strerr_die2x(100,fatal,ERR_NOMEM);
+}
+
+void tagmsg(dir,msgnum,seed,action,hashout,bodysize,chunk,fatal)
+/* This routine creates a cookie from num,seed and the */
+/* list key and returns that cookie in hashout. The use of sender/num and */
+/* first char of action is used to make cookie differ between messages,   */
+/* the key is the secret list key. The cookie will be inserted into       */
+/* table_cookie where table and other data is taken from dir/sql. We log  */
+/* arrival of the message (done=0). */
+
+char *dir;			/* db base dir */
+unsigned long msgnum;		/* number of this message */
+char *seed;			/* seed. NULL ok, but less entropy */
+char *action;			/* to make it certain the cookie differs from*/
+				/* one used for a digest */
+char *hashout;			/* calculated hash goes here */
+unsigned long bodysize;
+unsigned long chunk;
+char *fatal;
+{
+  PGresult *result;
+  PGresult *result2; /* Need for dupicate check */
+  char *table = (char *) 0;
+  char *ret;
+  unsigned int i;
+
+  strnum[fmt_ulong(strnum,msgnum)] = '\0';	/* message nr ->string*/
+
+    switch(slurp("key",&key,32)) {
+      case -1:
+	strerr_die3sys(111,fatal,ERR_READ,"key: ");
+      case 0:
+	strerr_die3x(100,fatal,"key",ERR_NOEXIST);
+    }
+    cookie(hash,key.s,key.len,strnum,seed,action);
+    for (i = 0; i < COOKIE; i++)
+      hashout[i] = hash[i];
+
+  if ((ret = opensql(dir,&table))) {
+    if (*ret) strerr_die2x(111,fatal,ret);
+    return;				/* no sql => success */
+
+  } else {
+    if (chunk >= 53L) chunk = 0L;	/* sanity */
+
+	/* INSERT INTO table_cookie (msgnum,cookie) VALUES (num,cookie) */
+	/* (we may have tried message before, but failed to complete, so */
+	/* ER_DUP_ENTRY is ok) */
+    if (!stralloc_copys(&line,"INSERT INTO ")) die_nomem(fatal);
+    if (!stralloc_cats(&line,table)) die_nomem(fatal);
+    if (!stralloc_cats(&line,"_cookie (msgnum,cookie,bodysize,chunk) VALUES ("))
+      die_nomem(fatal);
+    if (!stralloc_cats(&line,strnum)) die_nomem(fatal);
+    if (!stralloc_cats(&line,",'")) die_nomem(fatal);
+    if (!stralloc_catb(&line,hash,COOKIE)) die_nomem(fatal);
+    if (!stralloc_cats(&line,"',")) die_nomem(fatal);
+    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,bodysize)))
+      die_nomem(fatal);
+    if (!stralloc_cats(&line,",")) die_nomem(fatal);
+    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,chunk))) die_nomem(fatal);
+    if (!stralloc_cats(&line,")")) 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_COMMAND_OK) { /* Possible tuplicate */
+      if (!stralloc_copys(&line,"SELECT msgnum FROM ")) die_nomem(fatal);
+      if (!stralloc_cats(&line,table)) die_nomem(fatal);	  
+      if (!stralloc_cats(&line,"_cookie WHERE msgnum = ")) die_nomem(fatal);
+      if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,msgnum))) 
+	die_nomem(fatal);
+      /* Query */
+      if (!stralloc_0(&line)) die_nomem(fatal);
+      result2 = PQexec(psql,line.s);
+      if (result2 == NULL)
+	strerr_die2x(111,fatal,PQerrorMessage(psql));
+      if (PQresultStatus(result2) != PGRES_TUPLES_OK)
+	strerr_die2x(111,fatal,PQresultErrorMessage(result2));
+      /* No duplicate, return ERROR from first query */
+      if (PQntuples(result2)<1) 
+	strerr_die2x(111,fatal,PQresultErrorMessage(result));
+      PQclear(result2);
+    }
+    PQclear(result);
+
+    if (! (ret = logmsg(dir,msgnum,0L,0L,1))) return;	/* log done=1*/
+    if (*ret) strerr_die2x(111,fatal,ret);
+  }
+
+  return;
+}