subscribe.c

Go to the documentation of this file.

#include <unistd.h>
#include "str.h"
#include "stralloc.h"
#include "getln.h"
#include "readwrite.h"
#include "buffer.h"
#include "logmsg.h"
#include "open.h"
#include "byte.h"
#include "case.h"
#include "lock.h"
#include "error.h"
#include "subscribe.h"
#include "uint_t.h"
#include "fmt.h"
#include "errtxt.h"
#include "log.h"
#include "idx.h"
#include <mysql.h>
#include <mysqld_error.h>

 
extern MYSQL *mysql;
 
static stralloc addr = {0};
static stralloc lcaddr = {0};
static stralloc line = {0};
static stralloc domain = {0};
static stralloc logline = {0};
static stralloc quoted = {0};
static stralloc fnnew = {0};
static stralloc fn = {0};
static stralloc fnlock = {0};
 
void die_read(void) { logmsg(WH0,111,FATAL,B("ERR_READ :",fn.s)); }
void die_write(void) { logmsg(WH0,111,FATAL,B("ERR_WRITE :",fnnew.s)); }
 
static int fd;
static buffer bi;
static char inbuf[256];
static int fdnew;
static buffer bo;
static char outbuf[256];

 
int subscribe(const char *dbname,const char *userhost,int flagadd,const char *comment,
              const char *event,int flagmysql,int forcehash,const char *tab)
{
  int fdlock;
  MYSQL_RES *result;
  MYSQL_ROW row;
  char *cpat;
  char szhash[3] = "00";
  const char *r = (char *) 0;
  const char *table = (char *) 0;
  const char **ptable = &table;
 
  unsigned int j;
  uint32 h,lch;
  unsigned char ch,lcch;
  int match;
  int flagwasthere;
 
  if (userhost[str_chr(userhost,'\n')])
    logmsg(WH0,100,FATAL,ERR_ADDR_NL);
 
  if (tab) ptable = &tab;
 
  if (!flagmysql || (r = opensql(dbname,ptable))) {
    if (r && *r) logmsg(WH0,111,FATAL,r);
 
    /* fallback to local db */
 
    if (!stralloc_copys(&addr,"T")) die_nomem();
    if (!stralloc_cats(&addr,userhost)) die_nomem();
    if (addr.len > 401)
      logmsg(WH0,100,FATAL,ERR_ADDR_LONG);
 
    j = byte_rchr(addr.s,addr.len,'@');
    if (j == addr.len)
      logmsg(WH0,100,FATAL,ERR_ADDR_AT);
    case_lowerb(addr.s + j + 1,addr.len - j - 1);
    if (!stralloc_copy(&lcaddr,&addr)) die_nomem();
    case_lowerb(lcaddr.s + 1,j - 1);   /* make all-lc version of address */
 
    if (forcehash >= 0 && forcehash <= 52) {
      ch = lcch = (unsigned char) forcehash;
    } else {
      h = 5381;
      lch = h;
      for (j = 0;j < addr.len;++j) {
        h = (h + (h << 5)) ^ (uint32) (unsigned char) addr.s[j];
        lch = (lch + (lch << 5)) ^ (uint32) (unsigned char) lcaddr.s[j];
      }
      lcch = 64 + (lch % 53);
      ch = 64 + (h % 53);
    }
 
    if (!stralloc_0(&addr)) die_nomem();
    if (!stralloc_0(&lcaddr)) die_nomem();
    if (!stralloc_copys(&fn,dbname)) die_nomem();
    if (!stralloc_copys(&fnlock,dbname)) die_nomem();
 
    if (!stralloc_cats(&fn,"/subscribers/")) die_nomem();
    if (!stralloc_catb(&fn,&lcch,1)) die_nomem();
    if (!stralloc_copy(&fnnew,&fn)) die_nomem();
    /* code later depends on fnnew = fn + 'n' */
    if (!stralloc_cats(&fnnew,"n")) die_nomem();
    if (!stralloc_cats(&fnlock,"/lock")) die_nomem();
    if (!stralloc_0(&fnnew)) die_nomem();
    if (!stralloc_0(&fn)) die_nomem();
    if (!stralloc_0(&fnlock)) die_nomem();
 
    fdlock = lockfile(fnlock.s);
 
    /* do lower case hashed version first */
 
    fdnew = open_trunc(fnnew.s);
    if (fdnew == -1) die_write();
    buffer_init(&bo,buffer_unixwrite,fdnew,outbuf,sizeof(outbuf));
 
    flagwasthere = 0;
 
    fd = open_read(fn.s);
    if (fd == -1) {
      if (errno != ENOENT) { close(fdnew); die_read(); }
    }
    else {
      buffer_init(&bi,buffer_unixread,fd,inbuf,sizeof(inbuf));
 
      for (;;) {
        if (getln(&bi,&line,&match,'\0') == -1) {
          close(fd); close(fdnew); die_read();
        }
        if (!match) break;
        if (line.len == addr.len)
          if (!case_diffb(line.s,line.len,addr.s)) {
        flagwasthere = 1;
        if (!flagadd) continue;
      }
        if (buffer_bput(&bo,line.s,line.len) == -1) {
          close(fd); close(fdnew); die_write();
        }
      }
 
      close(fd);
    }
 
    if (flagadd && !flagwasthere)
      if (buffer_bput(&bo,addr.s,addr.len) == -1) {
        close(fdnew); die_write();
      }
 
    if (buffer_flush(&bo) == -1) { close(fdnew); die_write(); }
    if (fsync(fdnew) == -1) { close(fdnew); die_write(); }
    close(fdnew);
 
    if (rename(fnnew.s,fn.s) == -1)
      logmsg(WH0,111,FATAL,B("ERR_MOVE :",fnnew.s," to : ",fn.s));
 
    if ((ch == lcch) || flagwasthere) {
      close(fdlock);
      if (flagadd ^ flagwasthere) {
        if (!stralloc_0(&addr)) die_nomem();
        logaddr(dbname,event,addr.s+1,comment);
        return 1;
      }
      return 0;
    }
 
    /* If unsub and not found and hashed differ, OR */
    /* sub and not found (so added with new hash) */
    /* do the 'case-dependent' hash */
 
    fn.s[fn.len - 2] = ch;
    fnnew.s[fnnew.len - 3] = ch;
    fdnew = open_trunc(fnnew.s);
    if (fdnew == -1) die_write();
    buffer_init(&bo,buffer_unixwrite,fdnew,outbuf,sizeof(outbuf));
 
    fd = open_read(fn.s);
    if (fd == -1) {
      if (errno != ENOENT) { close(fdnew); die_read(); }
    } else {
      buffer_init(&bi,buffer_unixread,fd,inbuf,sizeof(inbuf));
 
      for (;;) {
        if (getln(&ss,&line,&match,'\0') == -1)
          { close(fd); close(fdnew); die_read(); }
        if (!match) break;
        if (line.len == addr.len)
          if (!case_diffb(line.s,line.len,addr.s)) {
            flagwasthere = 1;
            continue;                  /* always want to remove from case-sensitive hash */
          }
        if (buffer_bput(&bo,line.s,line.len) == -1)
          { close(fd); close(fdnew); die_write(); }
      }
 
      close(fd);
    }
 
    if (buffer_flush(&bo) == -1) { close(fdnew); die_write(); }
    if (fsync(fdnew) == -1) { close(fdnew); die_write(); }
    close(fdnew);
 
    if (rename(fnnew.s,fn.s) == -1)
      logmsg(WH0,111,FATAL,B("ERR_MOVE :",fnnew.s," to :",fn.s));
 
    close(fdlock);
    if (flagadd ^ flagwasthere) {
      if (!stralloc_0(&addr)) die_nomem();
      logaddr(dbname,event,addr.s + 1,comment);
      return 1;
    }
    return 0;
 
  } else {                             /* SQL version */
    domain.len = 0;                    /* clear domain */
 
    /* lowercase and check address */
 
    if (!stralloc_copys(&addr,userhost)) die_nomem();
    if (addr.len > 255)                /* this is 401 in std ezmlm. 255 should be plenty! */
      logmsg(WH0,100,FATAL,ERR_ADDR_LONG);
 
    j = byte_rchr(addr.s,addr.len,'@');
    if (j == addr.len)
      logmsg(WH0,100,FATAL,ERR_ADDR_AT);
    cpat = addr.s + j;
 
    case_lowerb(cpat + 1,addr.len - j - 1);
    if (!stralloc_ready(&quoted,2 * addr.len + 1)) die_nomem();
    quoted.len = mysql_escape_string(quoted.s,addr.s,addr.len);
 
    /* stored unescaped, so it should be ok if quoted.len is >255, as */
    /* long as addr.len is not */
 
    if (forcehash < 0) {
      if (!stralloc_copy(&lcaddr,&addr)) die_nomem();
      case_lowerb(lcaddr.s,j);         /* make all-lc version of address */
      h = 5381;
      for (j = 0;j < lcaddr.len;++j) {
        h = (h + (h << 5)) ^ (uint32) (unsigned char) lcaddr.s[j];
      }
      ch = (h % 53);            /* 0 - 52 */
    } else
      ch = (forcehash % 100);
 
    szhash[0] = '0' + ch / 10;        /* hash for sublist split */
    szhash[1] = '0' + (ch % 10);
 
    if (flagadd) {
      if (!stralloc_copys(&line,"LOCK TABLES ")) die_nomem();
      if (!stralloc_cats(&line,table)) die_nomem();
      if (!stralloc_cats(&line," WRITE")) die_nomem();
      if (mysql_real_query(mysql,line.s,line.len))
        logmsg(WH0,111,FATAL,mysql_error(mysql));
      if (!stralloc_copys(&line,"SELECT address FROM ")) die_nomem();
      if (!stralloc_cats(&line,table)) die_nomem();
      if (!stralloc_cats(&line," WHERE address='")) die_nomem();
      if (!stralloc_cat(&line,&quoted)) die_nomem();       /* addr */
      if (!stralloc_cats(&line,"'")) die_nomem();
      if (mysql_real_query(mysql,line.s,line.len))
        logmsg(WH0,111,FATAL,mysql_error(mysql));
      if (!(result = mysql_use_result(mysql)))
        logmsg(WH0,111,FATAL,mysql_error(mysql));
 
      if ((row = mysql_fetch_row(result))) {               /* there */
        while (mysql_fetch_row(result));                   /* use'm up */
        mysql_free_result(result);
        if (mysql_query(mysql,"UNLOCK TABLES"))
          logmsg(WH0,111,"FATAL",mysql_error(mysql));
        return 0;                                          /* there */
      } else {                                             /* not there */
        mysql_free_result(result);
        if (mysql_errno(mysql))                            /* or ERROR */
          logmsg(WH0,111,FATAL,mysql_error(mysql));
        if (!stralloc_copys(&line,"INSERT INTO ")) die_nomem();
        if (!stralloc_cats(&line,table)) die_nomem();
        if (!stralloc_cats(&line," (address,hash) VALUES ('")) die_nomem();
        if (!stralloc_cat(&line,&quoted)) die_nomem();     /* addr */
        if (!stralloc_cats(&line,"',")) die_nomem();
        if (!stralloc_cats(&line,szhash)) die_nomem();     /* hash */
        if (!stralloc_cats(&line,")")) die_nomem();
        if (mysql_real_query(mysql,line.s,line.len))       /* INSERT */
          logmsg(WH0,111,FATAL,mysql_error(mysql));
        if (mysql_query(mysql,"UNLOCK TABLES"))
          logmsg(WH0,111,FATAL,mysql_error(mysql));
      }
    } else {                                               /* unsub */
      if (!stralloc_copys(&line,"DELETE FROM ")) die_nomem();
      if (!stralloc_cats(&line,table)) die_nomem();
      if (!stralloc_cats(&line," WHERE address='")) die_nomem();
      if (!stralloc_cat(&line,&quoted)) die_nomem();      /* addr */
      if (forcehash >= 0) {
        if (!stralloc_cats(&line,"' AND hash=")) die_nomem();
        if (!stralloc_cats(&line,szhash)) die_nomem();
      } else {
        if (!stralloc_cats(&line,"' AND hash BETWEEN 0 AND 52")) die_nomem();
      }
      if (mysql_real_query(mysql,line.s,line.len))
        logmsg(WH0,111,FATAL,mysql_error(mysql));
      if (mysql_affected_rows(mysql) == 0)
 
      return 0;                                            /* address wasn't there*/
    }
 
    /* log to subscriber log */
    /* INSERT INTO t_slog (address,edir,etype,fromline) */
    /* VALUES('address',{'+'|'-'},'etype','[comment]') */
 
    if (!stralloc_copys(&logline,"INSERT INTO ")) die_nomem();
    if (!stralloc_cats(&logline,table)) die_nomem();
    if (!stralloc_cats(&logline,"_slog (address,edir,etype,fromline) VALUES ('")) die_nomem();
    if (!stralloc_cat(&logline,&quoted)) die_nomem();
    if (flagadd) {                                         /* edir */
      if (!stralloc_cats(&logline,"','+','")) die_nomem();
    } else {
      if (!stralloc_cats(&logline,"','-','")) die_nomem();
    }
 
    if (*(event + 1))                                      /* ezmlm-0.53 uses '' for ezmlm-manage's work */
      if (!stralloc_catb(&logline,event + 1,1)) die_nomem(); /* etype */
    if (!stralloc_cats(&logline,"','")) die_nomem();
    if (comment && *comment) {
      j = str_len(comment);
      if (!stralloc_ready(&quoted,2*j + 1)) die_nomem();
      quoted.len = mysql_escape_string(quoted.s,comment,j);/* from */
      if (!stralloc_cat(&logline,&quoted)) die_nomem();
    }
    if (!stralloc_cats(&logline,"')")) die_nomem();
 
    if (mysql_real_query(mysql,logline.s,logline.len))
        ;                                                  /* log (ignore errors) */
    if (!stralloc_0(&addr))
        ;                                                  /* ignore errors */
    logaddr(dbname,event,addr.s,comment);                  /* also log to old log */
 
    return 1;                                              /* desired effect */
  }
}