diff options
Diffstat (limited to 'daemon/regmaild/regmaild.c')
| -rw-r--r-- | daemon/regmaild/regmaild.c | 377 |
1 files changed, 377 insertions, 0 deletions
diff --git a/daemon/regmaild/regmaild.c b/daemon/regmaild/regmaild.c new file mode 100644 index 00000000..4d887868 --- /dev/null +++ b/daemon/regmaild/regmaild.c @@ -0,0 +1,377 @@ +// Register Mail Daemon +// $Id$ +// +// sqlite portion from wens +// Create: Hung-Te Lin <piaip@csie.ntu.edu.tw> +// Initial Date: 2009/06/19 +// +// Copyright (C) 2009, Hung-Te Lin <piaip@csie.ntu.edu.tw> +// All rights reserved + +// TODO: + +#include <stdio.h> +#include <unistd.h> +#include <signal.h> +#include <ctype.h> +#include <sys/time.h> +#include <sys/types.h> +#include <sys/resource.h> +#include <sys/wait.h> + +#include <event.h> +#include <sqlite3.h> + +#include "bbs.h" +#include "daemons.h" + +// local definiions +#ifndef EMAILDB_PATH +#define EMAILDB_PATH BBSHOME "/emaildb.db" +#endif + +sqlite3 *g_Db = NULL; + +static int +regmaildb_open(sqlite3 **Db, const char *fpath) { + int rc; + + assert(fpath); + if ((rc = sqlite3_open(fpath, 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; +} + +int +regmaildb_check_email(char * email, int email_len, char *myid) +{ + int count = -1; + sqlite3 *Db = NULL; + sqlite3_stmt *Stmt = NULL; + + if (g_Db) + Db =g_Db; + else if (regmaildb_open(&Db, EMAILDB_PATH) != 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, myid) == 0) + continue; + + // force update + u.email[0] = 0; + + if (passwd_load_user(result, &u)) + if (strcasecmp(email, u.email) == 0) + count++; + } + +end: + if (Stmt != NULL) + if (sqlite3_finalize(Stmt) != SQLITE_OK) + count = -1; + + if (Db != NULL && !g_Db) + sqlite3_close(Db); + + return count; +} + +int +regmaildb_update_email(char * userid, int userid_len, char * email, int email_len) +{ + int ret = -1; + + sqlite3 *Db = NULL; + sqlite3_stmt *Stmt = NULL; + + if (g_Db) + Db =g_Db; + else if (regmaildb_open(&Db, EMAILDB_PATH) != SQLITE_OK) + goto end; + if (regmaildb_open(&Db, EMAILDB_PATH) != SQLITE_OK) + goto end; + + if (strcmp(email, "x") == 0) + { + if (sqlite3_prepare(Db, "DELETE FROM emaildb WHERE userid like lower(?);", + -1, &Stmt, NULL) != SQLITE_OK) + goto end; + + if (sqlite3_bind_text(Stmt, 1, userid, userid_len, SQLITE_STATIC) != SQLITE_OK) + goto end; + + } else { + + 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 && !g_Db) + sqlite3_close(Db); + + return ret; +} + +/////////////////////////////////////////////////////////////////////// +// Callbacks + +// Command syntax: +// operation\n +// userid\n +// email\n +// +// operation: set,count + +static void +client_cb(int fd, short event, void *arg) +{ + struct event *ev = (struct event*) arg; + regmaildb_req req = {0}; + int ret = -1; + + assert(ev); + if ( (event & EV_TIMEOUT) || + !(event & EV_READ) || + toread(fd, &req, sizeof(req)) != sizeof(req) || + req.cb != sizeof(req)) + { + fprintf(stderr, "error: corrupted request.\r\n"); + close(fd); + free(ev); + return; + } + + switch(req.operation) + { + case REGMAILDB_REQ_COUNT: + ret = regmaildb_check_email(req.email, strlen(req.email), req.userid); + fprintf(stderr, "%s check mail[%s]: ret=%d\r\n", req.userid, req.email, ret); + if (towrite(fd, &ret, sizeof(ret)) != sizeof(ret)) + { + fprintf(stderr, " error: cannot write response...\r\n"); + } + break; + + case REGMAILDB_REQ_SET: + ret = regmaildb_update_email(req.userid, strlen(req.userid), + req.email, strlen(req.email)); + if (towrite(fd, &ret, sizeof(ret)) != sizeof(ret)) + { + fprintf(stderr, " error: cannot write response...\r\n"); + } + break; + + default: + fprintf(stderr, "error: invalid operation: %d.\r\n", req.operation); + close(fd); + free(ev); + return; + } + + // close connection anyway + close(fd); + free(ev); +} + +/////////////////////////////////////////////////////////////////////// +// Main + +#ifdef INIT_MAIN + +// standalone initialize builder + +#define TRANSCATION_PERIOD (4096) +int main(int argc, char *argv[]) +{ + 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; + const char *fpath = EMAILDB_PATH; + + // 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); + + if (argc > 1) + fpath = argv[1]; + + // init emaildb + if (regmaildb_open(&Db, fpath) != SQLITE_OK) + { + fprintf(stderr, "cannot initialize emaildb: %s.\n", fpath); + 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 + +/////////////////////////////////////////////////////////////////////// +// Event callbacks + +/////////////////////////////////////////////////////////////////////// +// Main + +struct timeval tv = {60, 0}; +static struct event ev_listen; + +static void listen_cb(int fd, short event, void *arg) +{ + int cfd; + + if ((cfd = accept(fd, NULL, NULL)) < 0 ) + return; + + struct event *ev = malloc(sizeof(struct event)); + + event_set(ev, cfd, EV_READ, client_cb, ev); + event_add(ev, &tv); +} + +int main(int argc, char *argv[]) +{ + int ch, sfd; + char *iface_ip = REGMAILD_ADDR; + + Signal(SIGPIPE, SIG_IGN); + + chdir(BBSHOME); + attach_SHM(); + + /* Give up root privileges: no way back from here */ + setgid(BBSGID); + setuid(BBSUID); + + while ( (ch = getopt(argc, argv, "i:h")) != -1 ) + switch( ch ){ + case 'i': + iface_ip = optarg; + break; + case 'h': + default: + fprintf(stderr, "usage: %s [-i [interface_ip]:port]\n", argv[0]); + return 1; + } + + if ( (sfd = tobind(iface_ip)) < 0 ) + return 1; + + // daemonize(BBSHOME "/run/regmaild.pid", NULL); + regmaildb_open(&g_Db, EMAILDB_PATH); + + event_init(); + event_set(&ev_listen, sfd, EV_READ | EV_PERSIST, listen_cb, &ev_listen); + event_add(&ev_listen, NULL); + fprintf(stderr, "start dispatch.\r\n"); + event_dispatch(); + + return 0; +} +// vim:et |