path: root/mbbsd/emaildb.c

/* $Id$ */
#include <sqlite3.h>
#include "bbs.h"

#define EMAILDB_PATH BBSHOME "/emaildb.db"

// define FORK model to minimize memory usage.
#define FORK_MODEL

static int emaildb_open(sqlite3 **Db) {
    int rc;

    if ((rc = sqlite3_open(EMAILDB_PATH, Db)) != SQLITE_OK)
    return rc;

    // create table if it doesn't exist
    rc = sqlite3_exec(*Db, "CREATE TABLE IF NOT EXISTS emaildb (userid TEXT, email TEXT, PRIMARY KEY (userid));"
        "CREATE INDEX IF NOT EXISTS email ON emaildb (email);",
        NULL, NULL, NULL);

    return rc;
}

#ifndef INIT_MAIN
int emaildb_check_email(char * email, int email_len)
{
    int count = -1;
    pid_t pid = -1;
    sqlite3 *Db = NULL;
    sqlite3_stmt *Stmt = NULL;

#ifdef FORK_MODEL
    switch((pid = fork()))
    {
    case -1: // error
        break;

    case 0: // child
        break;

    default:
        waitpid(pid, &count, 0);
        count = WEXITSTATUS(count);
        // vmsgf(ANSI_RESET "found %d emails", count);
        return count;
    }
#endif

    if (emaildb_open(&Db) != SQLITE_OK)
    goto end;

    if (sqlite3_prepare(Db, "SELECT userid FROM emaildb WHERE email LIKE lower(?);",
        -1, &Stmt, NULL) != SQLITE_OK)
    goto end;

    if (sqlite3_bind_text(Stmt, 1, email, email_len, SQLITE_STATIC) != SQLITE_OK)
    goto end;

    count = 0;
    while (sqlite3_step(Stmt) == SQLITE_ROW) {
    char *result;
    userec_t u;

    if ((result = (char*)sqlite3_column_text(Stmt, 0)) == NULL)
        break;

    // ignore my self, because I may be the one going to
    // use mail.
    if (strcasecmp(result, cuser.userid) == 0)
        continue;

    // force update
    u.email[0] = 0;
    
    if (getuser(result, &u))
        if (strcasecmp(email, u.email) == 0)
        count++;
    }

end:
    if (Stmt != NULL)
    if (sqlite3_finalize(Stmt) != SQLITE_OK)
        count = -1;

    if (Db != NULL)
    sqlite3_close(Db);

    if (pid == 0)
    exit(count);

    return count;
}
#endif

int emaildb_update_email(char * userid, int userid_len, char * email, int email_len)
{
    int ret = -1;
    pid_t pid = -1;

    sqlite3 *Db = NULL;
    sqlite3_stmt *Stmt = NULL;

#ifdef FORK_MODEL
    switch((pid = fork()))
    {
    case -1: // error
        break;

    case 0: // child
        break;

    default:
        waitpid(pid, &ret, 0);
        ret = WEXITSTATUS(ret);
        return ret;
    }
#endif

    if (emaildb_open(&Db) != SQLITE_OK)
    goto end;

    if (sqlite3_prepare(Db, "REPLACE INTO emaildb (userid, email) VALUES (lower(?),lower(?));",
        -1, &Stmt, NULL) != SQLITE_OK)
    goto end;

    if (sqlite3_bind_text(Stmt, 1, userid, userid_len, SQLITE_STATIC) != SQLITE_OK)
    goto end;

    if (sqlite3_bind_text(Stmt, 2, email, email_len, SQLITE_STATIC) != SQLITE_OK)
    goto end;

    if (sqlite3_step(Stmt) == SQLITE_DONE)
    ret = 0;

end:
    if (Stmt != NULL)
    sqlite3_finalize(Stmt);
    if (Db != NULL)
    sqlite3_close(Db);

    if (pid == 0)
    exit(ret);

    return ret;
}

#ifdef INIT_MAIN

// standalone initialize builder

#define TRANSCATION_PERIOD (4096)
int main()
{
    int fd = 0;
    userec_t xuser;
    off_t sz = 0, i = 0, valids = 0;
    sqlite3 *Db = NULL;
    sqlite3_stmt *Stmt = NULL, *tranStart = NULL, *tranEnd = NULL;

    // init passwd
    sz = dashs(FN_PASSWD);
    fd = open(FN_PASSWD, O_RDONLY);
    if (fd < 0 || sz <= 0)
    {
    fprintf(stderr, "cannot open ~/.PASSWDS.\n");
    return 0;
    }
    sz /= sizeof(userec_t);

    // init emaildb
    if (emaildb_open(&Db) != SQLITE_OK)
    {
    fprintf(stderr, "cannot initialize emaildb.\n");
    return 0;
    }

    if (sqlite3_prepare(Db, "REPLACE INTO emaildb (userid, email) VALUES (lower(?),lower(?));",
        -1, &Stmt, NULL) != SQLITE_OK ||
    sqlite3_prepare(Db, "BEGIN TRANSACTION;", -1, &tranStart, NULL) != SQLITE_OK ||
    sqlite3_prepare(Db, "COMMIT;", -1, &tranEnd, NULL) != SQLITE_OK)
    {
    fprintf(stderr, "SQLite 3 internal error.\n");
    return 0;
    }

    sqlite3_step(tranStart);
    sqlite3_reset(tranStart);
    while (read(fd, &xuser, sizeof(xuser)) == sizeof(xuser))
    {
    i++;
    // got a record
    if (strlen(xuser.userid) < 2 || strlen(xuser.userid) > IDLEN)
        continue;
    if (strlen(xuser.email) < 5)
        continue;

    if (sqlite3_bind_text(Stmt, 1, xuser.userid, strlen(xuser.userid), 
            SQLITE_STATIC) != SQLITE_OK)
    {
        fprintf(stderr, "\ncannot prepare userid param.\n");
        break;
    }
    if (sqlite3_bind_text(Stmt, 2, xuser.email, strlen(xuser.email), 
            SQLITE_STATIC) != SQLITE_OK)
    {
        fprintf(stderr, "\ncannot prepare email param.\n");
        break;
    }

    if (sqlite3_step(Stmt) != SQLITE_DONE)
    {
        fprintf(stderr, "\ncannot execute statement.\n");
        break;
    }
    sqlite3_reset(Stmt);

    valids ++;
    if (valids % 10 == 0)
        fprintf(stderr, "%d/%d (valid: %d)\r", 
            (int)i, (int)sz, (int)valids);
    if (valids % TRANSCATION_PERIOD == 0)
    {
        sqlite3_step(tranEnd);
        sqlite3_step(tranStart);
        sqlite3_reset(tranEnd);
        sqlite3_reset(tranStart);
    }
    }

    if (valids % TRANSCATION_PERIOD)
    sqlite3_step(tranEnd);

    if (Stmt != NULL)
    sqlite3_finalize(Stmt);

    if (Db != NULL)
    sqlite3_close(Db);

    close(fd);
    return 0;
}
#endif

// vim: sw=4