/* $Id: record.c,v 1.14 2003/06/28 08:47:45 kcwu Exp $ */
#include "bbs.h"

#undef  HAVE_MMAP
#define BUFSIZE 512

static void
PttLock(int fd, int size, int mode)
{
    static struct flock lock_it;
    int             ret;

    lock_it.l_whence = SEEK_CUR;/* from current point */
    lock_it.l_start = 0;	/* -"- */
    lock_it.l_len = size;	/* length of data */
    lock_it.l_type = mode;	/* set exclusive/write lock */
    lock_it.l_pid = 0;		/* pid not actually interesting */
    while ((ret = fcntl(fd, F_SETLKW, &lock_it)) < 0 && errno == EINTR);
}

#define safewrite       write

int
get_num_records(char *fpath, int size)
{
    struct stat     st;
    if (stat(fpath, &st) == -1)
	return 0;
    return st.st_size / size;
}

int
get_sum_records(char *fpath, int size)
{
    struct stat     st;
    long            ans = 0;
    FILE           *fp;
    fileheader_t    fhdr;
    char            buf[200], *p;

    if (!(fp = fopen(fpath, "r")))
	return -1;

    strlcpy(buf, fpath, sizeof(buf));
    p = strrchr(buf, '/');
    assert(p);
    p++;

    while (fread(&fhdr, size, 1, fp) == 1) {
	strcpy(p, fhdr.filename);
	if (stat(buf, &st) == 0 && S_ISREG(st.st_mode) && st.st_nlink == 1)
	    ans += st.st_size;
    }
    fclose(fp);
    return ans / 1024;
}

int
get_record(char *fpath, void *rptr, int size, int id)
{
    int             fd = -1;

    if (id < 1 || (fd = open(fpath, O_RDONLY, 0)) != -1) {
	if (lseek(fd, (off_t) (size * (id - 1)), SEEK_SET) != -1) {
	    if (read(fd, rptr, size) == size) {
		close(fd);
		return 0;
	    }
	}
	close(fd);
    }
    return -1;
}

int
get_records(char *fpath, void *rptr, int size, int id, int number)
{
    int             fd;

    if (id < 1 || (fd = open(fpath, O_RDONLY, 0)) == -1)
	return -1;

	if( flock(fd, LOCK_EX) < 0 ){
	    close(fd);
	    return -1;
	}

    if (lseek(fd, (off_t) (size * (id - 1)), SEEK_SET) == -1) {
	close(fd);
	return 0;
    }
    if ((id = read(fd, rptr, size * number)) == -1) {
	close(fd);
	return -1;
    }
    close(fd);
    return id / size;
}

int
lock_substitute_record(char *fpath, void *rptr, int size, int id, int mode)
{
    static  int     fd = -1;
    switch( mode ){
    case LOCK_EX:
	if( id < 1 || (fd = open(fpath, O_RDWR | O_CREAT, 0644)) == -1 )
	    return -1;

	if( flock(fd, LOCK_EX) < 0 ){
	    close(fd);
	    return -1;
	}
	lseek(fd, (off_t) (size * (id - 1)), SEEK_SET);
	read(fd, rptr, size);
	return 0;

    case LOCK_UN:
	if( fd < 0 )
	    return -1;
	lseek(fd, (off_t) (size * (id - 1)), SEEK_SET);
	write(fd, rptr, size);
	flock(fd, LOCK_UN);
	close(fd);
	fd = -1;
	return 0;

    default:
	return -1;
    }
}

int
substitute_record(char *fpath, void *rptr, int size, int id)
{
    int             fd;

    if (id < 1 || (fd = open(fpath, O_WRONLY | O_CREAT, 0644)) == -1)
	return -1;

    lseek(fd, (off_t) (size * (id - 1)), SEEK_SET);
    PttLock(fd, size, F_WRLCK);
    safewrite(fd, rptr, size);
    PttLock(fd, size, F_UNLCK);
    close(fd);

    return 0;
}

/* rocker.011022: 避免lock檔開啟時不正常斷線,造成永久lock */
static int
force_open(char *fname)
{
    int             fd;
    time_t          expire;

    expire = now - 3600;	/* lock 存在超過一個小時就是有問題! */

    if (dasht(fname) < expire)
	return -1;
    unlink(fname);
    fd = open(fname, O_WRONLY | O_TRUNC, 0644);

    return fd;
}


#if !defined(_BBS_UTIL_C_)
/* new/old/lock file processing */
typedef struct nol_t {
    char            newfn[256];
    char            oldfn[256];
    char            lockfn[256];
}               nol_t;

static void
nolfilename(nol_t * n, char *fpath)
{
    snprintf(n->newfn, sizeof(n->newfn), "%s.new", fpath);
    snprintf(n->oldfn, sizeof(n->oldfn), "%s.old", fpath);
    snprintf(n->lockfn, sizeof(n->lockfn), "%s.lock", fpath);
}

int
delete_record(char fpath[], int size, int id)
{
    nol_t           my;
    char            abuf[BUFSIZE];
    int             fdr, fdw, fd;
    int             count;

    nolfilename(&my, fpath);
    if ((fd = open(my.lockfn, O_RDWR | O_CREAT | O_APPEND, 0644)) == -1)
	return -1;

    flock(fd, LOCK_EX);

    if ((fdr = open(fpath, O_RDONLY, 0)) == -1) {
	move(10, 10);
	outs("delete_record failed!!! (open)");
	pressanykey();
	flock(fd, LOCK_UN);
	close(fd);
	return -1;
    }
    if (
	((fdw = open(my.newfn, O_WRONLY | O_CREAT | O_EXCL, 0644)) == -1) &&
	((fdw = force_open(my.newfn)) == -1)) {
	flock(fd, LOCK_UN);
	close(fd);
	close(fdr);
	return -1;
    }
    count = 1;
    while (read(fdr, abuf, size) == size) {
	if (id != count++ && (safewrite(fdw, abuf, size) == -1)) {
	    unlink(my.newfn);
	    close(fdr);
	    close(fdw);
	    flock(fd, LOCK_UN);
	    close(fd);
	    return -1;
	}
    }
    close(fdr);
    close(fdw);
    if (Rename(fpath, my.oldfn) == -1 || Rename(my.newfn, fpath) == -1) {
	flock(fd, LOCK_UN);
	close(fd);
	return -1;
    }
    flock(fd, LOCK_UN);
    close(fd);
    return 0;
}

#if 0
static char    *
title_body(char *title)
{
    if (!strncasecmp(title, str_reply, 3)) {
	title += 3;
	if (*title == ' ')
	    title++;
    }
    return title;
}
#endif

int
delete_range(char *fpath, int id1, int id2)
{
    fileheader_t    fhdr;
    nol_t           my;
    char            fullpath[STRLEN], *t;
    int             fdr, fdw, fd;
    int             count;

    nolfilename(&my, fpath);

    if ((fd = open(my.lockfn, O_RDWR | O_CREAT | O_APPEND, 0644)) == -1)
	return -1;

    flock(fd, LOCK_EX);

    if ((fdr = open(fpath, O_RDONLY, 0)) == -1) {
	flock(fd, LOCK_UN);
	close(fd);
	return -1;
    }
    if (
	((fdw = open(my.newfn, O_WRONLY | O_CREAT | O_EXCL, 0644)) == -1) &&
	((fdw = force_open(my.newfn)) == -1)) {
	close(fdr);
	flock(fd, LOCK_UN);
	close(fd);
	return -1;
    }
    count = 1;
    strlcpy(fullpath, fpath, sizeof(fullpath));
    t = strrchr(fullpath, '/');
    assert(t);
    t++;

    while (read(fdr, &fhdr, sizeof(fileheader_t)) == sizeof(fileheader_t)) {
	strcpy(t, fhdr.filename);

	/* rocker.011018: add new tag delete */
	if (
	    (fhdr.filemode & FILE_MARKED) ||	/* 標記 */
	    (fhdr.filemode & FILE_DIGEST) ||	/* 文摘 */
	    (id1 && (count < id1 || count > id2)) ||	/* range */
	    (!id1 && Tagger(atoi(t + 2), count, TAG_NIN))) {	/* TagList */
	    if ((safewrite(fdw, &fhdr, sizeof(fileheader_t)) == -1)) {
		close(fdr);
		close(fdw);
		unlink(my.newfn);
		flock(fd, LOCK_UN);
		close(fd);
		return -1;
	    }
	} else {
	    //if (dashd(fullpath))
		unlink(fullpath);
	}
	++count;
    }
    close(fdr);
    close(fdw);
    if (Rename(fpath, my.oldfn) == -1 || Rename(my.newfn, fpath) == -1) {
	flock(fd, LOCK_UN);
	close(fd);
	return -1;
    }
    flock(fd, LOCK_UN);
    close(fd);
    return 0;
}

int
search_rec(char *dirname, int (*filecheck) ())
{
    fileheader_t    fhdr;
    FILE           *fp;
    int             ans = 0;

    if (!(fp = fopen(dirname, "r")))
	return 0;

    while (fread(&fhdr, sizeof(fhdr), 1, fp)) {
	ans++;
	if ((*filecheck) (&fhdr)) {
	    fclose(fp);
	    return ans;
	}
    }
    fclose(fp);
    return 0;
}

int             delete_files(char *dirname, int (*filecheck) (), int record){
    fileheader_t    fhdr;
    FILE           *fp, *fptmp;
    int             ans = 0;
    char            tmpfname[128];
    char            genbuf[256];
    char            deleted[256];
    fileheader_t    delfh;
    char            deletedDIR[] = "boards/d/deleted/.DIR";

    strlcpy(deleted, "boards/d/deleted", sizeof(deleted));

    if (!(fp = fopen(dirname, "r")))
	return ans;

    strlcpy(tmpfname, dirname, sizeof(tmpfname));
    strcat(tmpfname, "_tmp");

    if (!(fptmp = fopen(tmpfname, "w"))) {
	fclose(fp);
	return ans;
    }
    while (fread(&fhdr, sizeof(fhdr), 1, fp)) {
	if ((*filecheck) (&fhdr)) {
	    ans++;
	    setdirpath(genbuf, dirname, fhdr.filename);
	    if (record) {
		deleted[14] = '\0';
		stampfile(deleted, &delfh);
		strlcpy(delfh.owner, fhdr.owner, sizeof(delfh.owner));
		strlcpy(delfh.title, fhdr.title, sizeof(delfh.title));
		Link(genbuf, deleted);
		append_record(deletedDIR, &delfh, sizeof(delfh));
	    }
	    unlink(genbuf);
	} else
	    fwrite(&fhdr, sizeof(fhdr), 1, fptmp);
    }

    fclose(fp);
    fclose(fptmp);
    unlink(dirname);
    Rename(tmpfname, dirname);

    return ans;
}

int
delete_file(char *dirname, int size, int ent, int (*filecheck) ())
{
    char            abuf[BUFSIZE];
    int             fd;
    struct stat     st;
    long            numents;

    if (ent < 1 || (fd = open(dirname, O_RDWR)) == -1)
	return -1;
    flock(fd, LOCK_EX);
    fstat(fd, &st);
    numents = ((long)st.st_size) / size;
    if (((long)st.st_size) % size)
	fprintf(stderr, "align err\n");
    if (lseek(fd, (off_t) size * (ent - 1), SEEK_SET) != -1) {
	if (read(fd, abuf, size) == size) {
	    if ((*filecheck) (abuf)) {
		int             i;

		for (i = ent; i < numents; i++) {
		    if (lseek(fd, (off_t) ((i) * size), SEEK_SET) == -1 ||
			read(fd, abuf, size) != size ||
			lseek(fd, (off_t) (i - 1) * size, SEEK_SET) == -1)
			break;
		    if (safewrite(fd, abuf, size) != size)
			break;
		}
		ftruncate(fd, (off_t) size * (numents - 1));
		flock(fd, LOCK_UN);
		close(fd);
		return 0;
	    }
	}
    }
    lseek(fd, 0, SEEK_SET);
    ent = 1;
    while (read(fd, abuf, size) == size) {
	if ((*filecheck) (abuf)) {
	    int             i;

	    for (i = ent; i < numents; i++) {
		if (lseek(fd, (off_t) (i + 1) * size, SEEK_SET) == -1 ||
		    read(fd, abuf, size) != size ||
		    lseek(fd, (off_t) (i) * size, SEEK_SET) == -1 ||
		    safewrite(fd, abuf, size) != size)
		    break;
	    }
	    ftruncate(fd, (off_t) size * (numents - 1));
	    flock(fd, LOCK_UN);
	    close(fd);
	    return 0;
	}
	ent++;
    }
    flock(fd, LOCK_UN);
    close(fd);
    return -1;
}

#endif				/* !defined(_BBS_UTIL_C_) */

int             apply_record(char *fpath, int (*fptr) (), int size){
    char            abuf[BUFSIZE];
    FILE           *fp;

    if (!(fp = fopen(fpath, "r")))
	return -1;

    while (fread(abuf, 1, size, fp) == (size_t)size)
	if ((*fptr) (abuf) == QUIT) {
	    fclose(fp);
	    return QUIT;
	}
    fclose(fp);
    return 0;
}

/* mail / post 時，依據時間建立檔案，加上郵戳 */
int
stampfile(char *fpath, fileheader_t * fh)
{
    register char  *ip = fpath;
    time_t          dtime = now;
    struct tm      *ptime;
    int             fp = 0;

    if (access(fpath, X_OK | R_OK | W_OK))
	mkdir(fpath, 0755);

    while (*(++ip));
    *ip++ = '/';
    do {
	sprintf(ip, "M.%ld.A.%3.3X", ++dtime, rand() & 0xFFF);
	if (fp == -1 && errno != EEXIST)
	    return -1;
    } while ((fp = open(fpath, O_CREAT | O_EXCL | O_WRONLY, 0644)) == -1);
    close(fp);
    memset(fh, 0, sizeof(fileheader_t));
    strlcpy(fh->filename, ip, sizeof(fh->filename));
    ptime = localtime(&dtime);
    snprintf(fh->date, sizeof(fh->date),
	     "%2d/%02d", ptime->tm_mon + 1, ptime->tm_mday);
    return 0;
}

void
stampdir(char *fpath, fileheader_t * fh)
{
    register char  *ip = fpath;
    time_t          dtime = now;
    struct tm      *ptime;

    if (access(fpath, X_OK | R_OK | W_OK))
	mkdir(fpath, 0755);

    while (*(++ip));
    *ip++ = '/';
    do {
	sprintf(ip, "D%lX", ++dtime & 07777);
    } while (mkdir(fpath, 0755) == -1);
    memset(fh, 0, sizeof(fileheader_t));
    strlcpy(fh->filename, ip, sizeof(fh->filename));
    ptime = localtime(&dtime);
    snprintf(fh->date, sizeof(fh->date),
	     "%2d/%02d", ptime->tm_mon + 1, ptime->tm_mday);
}

void
stamplink(char *fpath, fileheader_t * fh)
{
    register char  *ip = fpath;
    time_t          dtime = now;
    struct tm      *ptime;

    if (access(fpath, X_OK | R_OK | W_OK))
	mkdir(fpath, 0755);

    while (*(++ip));
    *ip++ = '/';
    do {
	sprintf(ip, "S%lX", ++dtime);
    } while (symlink("temp", fpath) == -1);
    memset(fh, 0, sizeof(fileheader_t));
    strlcpy(fh->filename, ip, sizeof(fh->filename));
    ptime = localtime(&dtime);
    snprintf(fh->date, sizeof(fh->date),
	     "%2d/%02d", ptime->tm_mon + 1, ptime->tm_mday);
}

int
do_append(char *fpath, fileheader_t * record, int size)
{
    int             fd;

    if ((fd = open(fpath, O_WRONLY | O_CREAT, 0644)) == -1) {
	perror("open");
	return -1;
    }
    flock(fd, LOCK_EX);
    lseek(fd, 0, SEEK_END);

    safewrite(fd, record, size);

    flock(fd, LOCK_UN);
    close(fd);
    return 0;
}

int
append_record(char *fpath, fileheader_t * record, int size)
{
#if !defined(_BBS_UTIL_C_)
    int             m, n;
    if (get_num_records(fpath, sizeof(fileheader_t)) <= MAX_KEEPMAIL * 2) {
	FILE           *fp;
	char            buf[512], address[200];

	for (n = strlen(fpath) - 1; fpath[n] != '/' && n > 0; n--);
	strncpy(buf, fpath, n + 1);
	buf[n + 1] = 0;
	for (m = strlen(buf) - 2; buf[m] != '/' && m > 0; m--);
	strcat(buf, ".forward");
	if ((fp = fopen(buf, "r"))) {
	    fscanf(fp, "%s", address);
	    fclose(fp);
	    if (buf[0] != 0 && buf[0] != ' ') {
		buf[n + 1] = 0;
		strcat(buf, record->filename);
		do_append(fpath, record, size);
#ifndef  USE_BSMTP
		bbs_sendmail(buf, record->title, address);
#else
		bsmtp(buf, record->title, address, 0);
#endif
		return 0;
	    }
	}
    }
#endif

    do_append(fpath, record, size);

    return 0;
}