putsubs.c

Go to the documentation of this file.

#include <unistd.h>
#include "getln.h"
#include "error.h"
#include "logmsg.h"
#include "readwrite.h"
#include "str.h"
#include "fmt.h"
#include "stralloc.h"
#include "open.h"
#include "buffer.h"
#include "case.h"
#include "errtxt.h"
#include "subscribe.h"
#include "qmail.h"
#include "idx.h"
#include <mysql.h>
 
 
extern MYSQL *mysql;
 
static buffer bi;
static char inbuf[512];
static char strnum[FMT_ULONG];
static stralloc line = {0};
static stralloc fn = {0};
 
static void die_write(void) { logmsg(WHO,111,FATAL,B("ERR_WRITE :","stdout"); }

 
unsigned long putsubs(const char *dbname,unsigned long hash_lo,unsigned long hash_hi,int subwrite(),int flagsql)
{
  MYSQL_RES *result;
  MYSQL_ROW row;
  const char *table = (char *) 0;
  unsigned long *lengths;
 
  unsigned int i;
  int fd;
  unsigned long no = 0L;
  int match;
  unsigned int hashpos;
  const char *ret = (char *) 0;
 
  if (!flagsql || (ret = opensql(dbname,&table))) {
    if (flagsql && *ret) logmsg(WHO,111,FATAL,ret);        /* fallback to local db */
    if (!stralloc_copys(&fn,dbname)) die_nomem();
    if (!stralloc_catb(&fn,"/subscribers/?",15)) 
      die_nomem();                                         /* 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)
      logmsg(WHO,111,FATAL,B("ERR_READ :",fn.s));
      } else {
        buffer_init(&bi,buffer_unixread,fd,inbuf,sizeof(inbuf));
        for (;;) {
          if (getln(&bi,&line,&match,'\0') == -1)
            logmsg(WHO,111,FATAL,B("ERR_READ :",fn.s));
          if (!match)
            break;
          if (subwrite(line.s + 1,line.len - 2) == -1) die_write();
          no++;
        }
        close(fd);
      }
    }
    return no;
  } else {                                                  /* SQL Version: main query */
    if (!stralloc_copys(&line,"SELECT address FROM ")) die_nomem();
    if (!stralloc_cats(&line,table)) die_nomem();
    if (!stralloc_cats(&line," WHERE hash BETWEEN ")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_lo))) die_nomem();
    if (!stralloc_cats(&line," AND ")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_hi))) die_nomem();
 
    if (mysql_real_query(mysql,line.s,line.len))           /* query */
      logmsg(WHO,111,FATAL,mysql_error(mysql));
    
    if (!(result = mysql_use_result(mysql)))
      logmsg(WHO,111,FATAL,mysql_error(mysql));
    no = 0;
 
    /* Next statements are safe; even if someone messes with the address field def */
 
    while ((row = mysql_fetch_row(result))) {
      if (!(lengths = mysql_fetch_lengths(result)))
        logmsg(WHO,111,FATAL,mysql_error(mysql));
      if (subwrite(row[0],lengths[0]) == -1) die_write();
      no++;                                                /* count for list-list fxn */
    }
    if (!mysql_eof(result))
      logmsg(WHO,111,FATAL,mysql_error(mysql));
 
    mysql_free_result(result);
 
    return no;
  }
}