tagmsg.c

Go to the documentation of this file.

#include "stralloc.h"
#include "scan.h"
#include "fmt.h"
#include "logmsg.h"
#include "cookie.h"
#include "slurp.h"
#include "errtxt.h"
#include "subscribe.h"
#include "makehash.h"
#include "idx.h"
#include <unistd.h>
#include <libpq-fe.h>
 
#define WHO "tagmsg"

 
extern PGconn *pgsql;
 
static stralloc line = {0};
static stralloc key = {0};
static char hash[COOKIE];
static char strnum[FMT_ULONG];                      /* message number as sz */

 
void tagmsg(const char *dir,unsigned long msgnum,const char *seed,const char *action,
            char *hashout,unsigned long bodysize,unsigned long chunk)
{
  PGresult *result;
  PGresult *result2;                             /* Need for dupicate check */
  const char *table = (char *) 0;
  const char *ret;
  unsigned int i;
 
  strnum[fmt_ulong(strnum,msgnum)] = '\0';       /* message nr ->string*/
 
  switch (openreadclose("key",&key,32)) {
    case -1: logmsg(WHO,111,FATAL,B("ERR_READ :","key: "));
    case  0: logmsg(WHO,100,FATAL,B("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) logmsg(WHO,2x(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();
    if (!stralloc_cats(&line,table)) die_nomem();
    if (!stralloc_cats(&line,"_cookie (msgnum,cookie,bodysize,chunk) VALUES (")) die_nomem();
    if (!stralloc_cats(&line,strnum)) die_nomem();
    if (!stralloc_cats(&line,",'")) die_nomem();
    if (!stralloc_catb(&line,hash,COOKIE)) die_nomem();
    if (!stralloc_cats(&line,"',")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,bodysize))) die_nomem();
    if (!stralloc_cats(&line,",")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,chunk))) die_nomem();
    if (!stralloc_cats(&line,")")) die_nomem();
    if (!stralloc_0(&line)) die_nomem();
 
    result = PQexec(pgsql,line.s);
    if (result == NULL)
      logmsg(WHO,111,FATAL,PQerrorMessage(pgsql));
 
    if (PQresultStatus(result) != PGRES_COMMAND_OK) {      /* Possible duplicate */
      if (!stralloc_copys(&line,"SELECT msgnum FROM ")) die_nomem();
      if (!stralloc_cats(&line,table)) die_nomem();      
      if (!stralloc_cats(&line,"_cookie WHERE msgnum = ")) die_nomem();
      if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,msgnum))) die_nomem();
      if (!stralloc_0(&line)) die_nomem();
 
      /* Query */
 
      result2 = PQexec(pgsql,line.s);
      if (result2 == NULL)
        logmsg(WHO,111,FATAL,PQerrorMessage(pgsql));
      if (PQresultStatus(result2) != PGRES_TUPLES_OK)
        logmsg(WHO,111,FATAL,PQresultErrorMessage(result2));
 
      /* No duplicate, return ERROR from first query */
 
      if (PQntuples(result2) < 1) 
        logmsg(WHO,111,FATAL,PQresultErrorMessage(result));
 
      PQclear(result2);
    }
    PQclear(result);
 
    if (! (ret = logmsg(dir,msgnum,0L,0L,1))) return;      /* log done=1*/
    if (*ret) logmsg(WHO,111,FATAL,ret);
  }
 
  return;
}