/* $Id$ */
/* 自動砍信工具程式 */
#include "bbs.h"
#define QCAST int (*)(const void *, const void *)
#define DEF_DAYS 60
#define DEF_MAXP 10000
#define DEF_MINP 9000
#define EXPIRE_CONF BBSHOME "/etc/expire.conf"
#ifdef SAFE_ARTICLE_DELETE
char safe_delete_only = 0;
#endif
extern boardheader_t *bcache;
char bpath[256];
int checkmode = 0;
typedef struct {
char bname[IDLEN + 1]; /* board ID */
int days; /* expired days */
int maxp; /* max post */
int minp; /* min post */
} life_t;
void callsystem(char *s)
{
if( checkmode )
printf("in checkmode, skip `%s`\n", s);
else
system(s);
}
void callrm(char *s)
{
if( checkmode )
printf("in checkmode, skip rm %s\n", s);
else
unlink(s);
}
void cleanSR(char *brdname)
{
DIR *dirp;
char dirf[128], fpath[PATHLEN];
struct dirent *ent;
int nDelete = 0;
sprintf(dirf, "boards/%c/%s", brdname[0], brdname);
if( (dirp = opendir(dirf)) == NULL )
return;
while( (ent = readdir(dirp)) != NULL )
if( strncmp(ent->d_name, "SR.", 3) == 0 ){
sprintf(fpath, "%s/%s", dirf, ent->d_name);
callrm(fpath);
++nDelete;
}
closedir(dirp);
printf("board %s: %d SRs are deleted.\n", brdname, nDelete);
}
void expire(life_t *brd)
{
fileheader_t head;
struct stat state;
char lockfile[128], tmpfile[128], bakfile[128], cmd[256];
char fpath[128], index[128], *fname;
int total, bid;
int fdlock, fdr, fdw = 0, done, keep;
int duetime, ftime, nKeep = 0, nDelete = 0;
printf("%s\n", brd->bname);
/* XXX: bid of cache.c's getbnum starts from 1 */
if( (bid = getbnum(brd->bname)) == 0 ||
strcmp(brd->bname, bcache[bid - 1].brdname) ){
printf("no such board?: %s\n", brd->bname);
sprintf(cmd, "mv "BBSHOME"/boards/%c/%s "BBSHOME"/boards.error/%s",
brd->bname[0], brd->bname, brd->bname);
callsystem(cmd);
return;
}
#ifdef VERBOSE
if( brd->days < 1 ){
printf(":Err: expire time must more than 1 day.\n");
return;
}
else if( brd->maxp < 100 ){
printf(":Err: maxmum posts number must more than 100.\n");
return;
}
#endif
cleanSR(brd->bname);
sprintf(index, "%s/%s/.DIR", bpath, brd->bname);
sprintf(lockfile, "%s.lock", index);
if ((fdlock = open(lockfile, O_RDWR | O_CREAT | O_APPEND, 0644)) == -1){
perror("open lock file error");
return;
}
flock(fdlock, LOCK_EX);
strcpy(fpath, index);
fname = (char *) strrchr(fpath, '.');
duetime = (int)time(NULL) - brd->days * 24 * 60 * 60;
done = 0;
if( (fdr = open(index, O_RDONLY, 0)) > 0 ){
fstat(fdr, &state);
total = state.st_size / sizeof(head);
if( !checkmode ){
sprintf(tmpfile, "%s.new", index);
unlink(tmpfile);
}
// TODO use fread/fwrite to reduce system calls
if( checkmode ||
(fdw = open(tmpfile, O_WRONLY | O_CREAT | O_EXCL, 0644)) > 0 ){
while( read(fdr, &head, sizeof(head)) == sizeof(head) ){
done = 1;
ftime = atoi(head.filename + 2);
if (head.owner[0] == '-'
#ifdef SAFE_ARTICLE_DELETE
|| strncmp(head.filename, ".delete", 7) == 0
#endif
)
keep = 0;
#ifdef SAFE_ARTICLE_DELETE
else if( safe_delete_only )
keep = 1;
#endif
else if( head.filemode & FILE_MARKED || total <= brd->minp )
keep = 1;
else if( ftime < duetime || total > brd->maxp )
keep = 0;
else
keep = 1;
if( keep ){
++nKeep;
if( !checkmode &&
write(fdw, (char *)&head, sizeof(head)) == -1 ){
done = 0;
break;
}
}
else {
++nDelete;
strcpy(fname, head.filename);
if( checkmode )
printf("\tin checkmode, skip rm %s\n", fname);
else{
unlink(fname);
printf("\t%s\n", fname);
}
total--;
}
}
if( !checkmode )
close(fdw);
}
close(fdr);
}
if( !checkmode && done ){
sprintf(bakfile, "%s.old", index);
if( rename(index, bakfile) != -1 ){
rename(tmpfile, index);
touchbtotal(bid);
}
}
printf("board %s: %d articles are kept, %d articles are deleted.\n",
brd->bname, nKeep, nDelete);
flock(fdlock, LOCK_UN);
close(fdlock);
}
int count;
life_t db, table[MAX_BOARD], *key;
void toexpire(char *brdname)
{
if( brdname[0] > ' ' && brdname[0] != '.' ){
key = NULL;
if( count )
key = (life_t *)bsearch(brdname, table, count,
sizeof(life_t), (QCAST)strcasecmp);
if( key == NULL )
key = &db;
strcpy(key->bname, brdname);
expire(key);
}
}
void visitdir(char c)
{
DIR *dirp;
struct dirent *de;
sprintf(bpath, BBSHOME "/boards/%c", c);
if (!(dirp = opendir(bpath))){
printf(":Err: unable to open %s\n", bpath);
return;
}
while( (de = readdir(dirp)) != NULL )
if( de->d_name[0] != '.' )
toexpire(de->d_name);
closedir(dirp);
}
int main(int argc, char **argv)
{
FILE *fin;
int number, i, ch;
char *ptr, *bname, buf[256];
char dirs[] = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p',
'a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l',
'z', 'x', 'c', 'v', 'b', 'n', 'm',
'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P',
'A', 'S', 'D', 'F', 'G', 'H', 'J', 'K', 'L',
'Z', 'X', 'C', 'V', 'B', 'N', 'M', 0};
chdir(BBSHOME);
/* default value */
db.days = DEF_DAYS;
db.maxp = DEF_MAXP;
db.minp = DEF_MINP;
while( (ch = getopt(argc, argv, "d:M:m:hn"
#ifdef SAFE_ARTICLE_DELETE
"D"
#endif
)) != -1 )
switch( ch ){
#ifdef SAFE_ARTICLE_DELETE
case 'D':
safe_delete_only = 1;
break;
#endif
case 'd':
db.days = atoi(optarg);
break;
case 'M':
db.maxp = atoi(optarg);
break;
case 'm':
db.minp = atoi(optarg);
break;
case 'n':
checkmode = 1;
break;
case 'h':
default:
fprintf(stderr,
"usage: expire [-m minp] [-M MAXP] [-d days] [board name...] [-n]\n"
"deletion policy:\n"
" do nothing if #articles < minp (default:%d)\n"
" delete NOT MARKED articles which were post before days \n"
" (default:%d) or #articles > MAXP (default:%d)\n",
DEF_MINP, DEF_DAYS, DEF_MAXP);
return 0;
}
argc -= optind;
argv += optind;
/* --------------- */
/* load expire.ctl */
/* --------------- */
count = 0;
if( (fin = fopen(EXPIRE_CONF, "r")) ){
while( fgets(buf, 256, fin) != NULL ){
if (buf[0] == '#')
continue;
bname = (char *) strtok(buf, " \t\r\n");
if( bname && *bname ){
ptr = (char *) strtok(NULL, " \t\r\n");
if( ptr && (number = atoi(ptr)) > 0 ){
key = &(table[count++]);
strcpy(key->bname, bname);
key->days = number;
key->maxp = db.maxp;
key->minp = db.minp;
ptr = (char *) strtok(NULL, " \t\r\n");
if( ptr && (number = atoi(ptr)) > 0 ){
key->maxp = number;
ptr = (char *) strtok(NULL, " \t\r\n");
if( ptr && (number = atoi(ptr)) > 0 ){
key->minp = number;
}
}
}
}
}
fclose(fin);
}
if( count > 1)
qsort(table, count, sizeof(life_t), (QCAST)strcasecmp);
attach_SHM();
if( argc > 0 ){
for( i = 0 ; i < argc ; ++i )
if( argv[i][1] == '*' )
visitdir(argv[i][0]);
else{
sprintf(bpath, BBSHOME "/boards/%c", argv[i][0]);
toexpire(argv[i]);
}
}
else{ // visit all boards
for( i = 0 ; dirs[i] != 0 ; ++i )
visitdir(dirs[i]);
}
return 0;
}
