123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242 |
- /*
- * mbsync - mailbox synchronizer
- * Copyright (C) 2000-2002 Michael R. Elkins <me@mutt.org>
- * Copyright (C) 2002-2004 Oswald Buddenhagen <ossi@users.sf.net>
- * Copyright (C) 2004 Theodore Y. Ts'o <tytso@mit.edu>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
- *
- * As a special exception, mbsync may be linked with the OpenSSL library,
- * despite that library's more restrictive license.
- */
- #include "isync.h"
- #include <limits.h>
- #include <stdlib.h>
- #include <string.h>
- #include <ctype.h>
- #include <dirent.h>
- #include <fcntl.h>
- #include <stdio.h>
- #include <unistd.h>
- #include <sys/stat.h>
- #include <sys/file.h>
- #include <errno.h>
- #include <time.h>
- #define USE_DB 1
- #ifdef __linux__
- # define LEGACY_FLOCK 1
- #endif
- #ifdef USE_DB
- #include <db.h>
- #endif /* USE_DB */
- typedef struct maildir_store_conf {
- store_conf_t gen;
- char *inbox;
- #ifdef USE_DB
- int alt_map;
- #endif /* USE_DB */
- } maildir_store_conf_t;
- typedef struct maildir_message {
- message_t gen;
- char *base;
- char tuid[TUIDL];
- } maildir_message_t;
- typedef struct maildir_store {
- store_t gen;
- int uvfd, uvok, nuid;
- int minuid, maxuid, nexcs, *excs;
- #ifdef USE_DB
- DB *db;
- #endif /* USE_DB */
- } maildir_store_t;
- #ifdef USE_DB
- static DBT key, value; /* no need to be reentrant, and this saves lots of memset()s */
- #endif /* USE_DB */
- static struct flock lck;
- static int MaildirCount;
- static const char Flags[] = { 'D', 'F', 'R', 'S', 'T' };
- static unsigned char
- maildir_parse_flags( const char *base )
- {
- const char *s;
- unsigned i;
- unsigned char flags;
- flags = 0;
- if ((s = strstr( base, ":2," )))
- for (s += 3, i = 0; i < as(Flags); i++)
- if (strchr( s, Flags[i] ))
- flags |= (1 << i);
- return flags;
- }
- static void maildir_close_store( store_t *gctx );
- static store_t *
- maildir_open_store( store_conf_t *conf, store_t *oldctx )
- {
- maildir_store_t *ctx;
- struct stat st;
- if (oldctx)
- maildir_close_store( oldctx );
- if (stat( conf->path, &st ) || !S_ISDIR(st.st_mode)) {
- error( "Maildir error: cannot open store %s\n", conf->path );
- return 0;
- }
- ctx = nfcalloc( sizeof(*ctx) );
- ctx->gen.conf = conf;
- ctx->uvfd = -1;
- return &ctx->gen;
- }
- static void
- free_maildir_messages( message_t *msg )
- {
- message_t *tmsg;
- for (; (tmsg = msg); msg = tmsg) {
- tmsg = msg->next;
- free( ((maildir_message_t *)msg)->base );
- free( msg );
- }
- }
- static void
- maildir_cleanup( store_t *gctx )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- free_maildir_messages( gctx->msgs );
- #ifdef USE_DB
- if (ctx->db)
- ctx->db->close( ctx->db, 0 );
- #endif /* USE_DB */
- if (gctx->path)
- free( gctx->path );
- if (ctx->excs)
- free( ctx->excs );
- if (ctx->uvfd >= 0)
- close( ctx->uvfd );
- }
- static void
- maildir_close_store( store_t *gctx )
- {
- maildir_cleanup( gctx );
- free_string_list( gctx->boxes );
- free( gctx );
- }
- static int
- maildir_list( store_t *gctx )
- {
- DIR *dir;
- struct dirent *de;
- if (!(dir = opendir( gctx->conf->path ))) {
- error( "%s: %s\n", gctx->conf->path, strerror(errno) );
- return DRV_STORE_BAD;
- }
- while ((de = readdir( dir ))) {
- struct stat st;
- char buf[PATH_MAX];
- if (*de->d_name == '.')
- continue;
- nfsnprintf( buf, sizeof(buf), "%s%s/cur", gctx->conf->path, de->d_name );
- if (stat( buf, &st ) || !S_ISDIR(st.st_mode))
- continue;
- add_string_list( &gctx->boxes, de->d_name );
- }
- closedir (dir);
- gctx->listed = 1;
- return DRV_OK;
- }
- static const char *subdirs[] = { "cur", "new", "tmp" };
- typedef struct {
- char *base;
- int size;
- unsigned uid:31, recent:1;
- char tuid[TUIDL];
- } msg_t;
- typedef struct {
- msg_t *ents;
- int nents, nalloc;
- } msglist_t;
- static void
- maildir_free_scan( msglist_t *msglist )
- {
- int i;
- if (msglist->ents) {
- for (i = 0; i < msglist->nents; i++)
- if (msglist->ents[i].base)
- free( msglist->ents[i].base );
- free( msglist->ents );
- }
- }
- #define _24_HOURS (3600 * 24)
- static int
- maildir_validate( const char *prefix, const char *box, int create )
- {
- DIR *dirp;
- struct dirent *entry;
- time_t now;
- int i, j, bl;
- struct stat st;
- char buf[_POSIX_PATH_MAX];
- bl = nfsnprintf( buf, sizeof(buf) - 4, "%s%s/", prefix, box );
- if (stat( buf, &st )) {
- if (errno == ENOENT) {
- if (create) {
- if (mkdir( buf, 0700 )) {
- error( "Maildir error: mkdir %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- return DRV_STORE_BAD;
- }
- mkdirs:
- for (i = 0; i < 3; i++) {
- memcpy( buf + bl, subdirs[i], 4 );
- if (mkdir( buf, 0700 )) {
- error( "Maildir error: mkdir %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- return DRV_BOX_BAD;
- }
- }
- } else {
- error( "Maildir error: mailbox '%s' does not exist\n", buf );
- return DRV_BOX_BAD;
- }
- } else {
- error( "Maildir error: stat %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- return DRV_BOX_BAD;
- }
- } else {
- for (i = j = 0; i < 3; i++) {
- memcpy( buf + bl, subdirs[i], 4 );
- if (!stat( buf, &st ) && S_ISDIR(st.st_mode))
- j++;
- }
- if (!j)
- goto mkdirs;
- if (j != 3) {
- error( "Maildir error: '%.*s' is no valid mailbox\n", bl, buf );
- return DRV_BOX_BAD;
- }
- memcpy( buf + bl, "tmp/", 5 );
- bl += 4;
- if (!(dirp = opendir( buf ))) {
- error( "Maildir error: opendir: %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- return DRV_BOX_BAD;
- }
- time( &now );
- while ((entry = readdir( dirp ))) {
- nfsnprintf( buf + bl, sizeof(buf) - bl, "%s", entry->d_name );
- if (stat( buf, &st ))
- error( "Maildir error: stat: %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- else if (S_ISREG(st.st_mode) && now - st.st_ctime >= _24_HOURS) {
- /* this should happen infrequently enough that it won't be
- * bothersome to the user to display when it occurs.
- */
- info( "Maildir notice: removing stale file %s\n", buf );
- if (unlink( buf ))
- error( "Maildir error: unlink: %s: %s (errno %d)\n",
- buf, strerror(errno), errno );
- }
- }
- closedir( dirp );
- }
- return DRV_OK;
- }
- #ifdef USE_DB
- static void
- make_key( DBT *tkey, char *name )
- {
- char *u = strpbrk( name, ":," );
- tkey->data = name;
- tkey->size = u ? (size_t)(u - name) : strlen( name );
- }
- static int
- maildir_set_uid( maildir_store_t *ctx, const char *name, int *uid )
- {
- int ret, uv[2];
- if (uid)
- *uid = ++ctx->nuid;
- key.data = (void *)"UIDVALIDITY";
- key.size = 11;
- uv[0] = ctx->gen.uidvalidity;
- uv[1] = ctx->nuid;
- value.data = uv;
- value.size = sizeof(uv);
- if ((ret = ctx->db->put( ctx->db, 0, &key, &value, 0 ))) {
- tbork:
- ctx->db->err( ctx->db, ret, "Maildir error: db->put()" );
- bork:
- ctx->db->close( ctx->db, 0 );
- ctx->db = 0;
- return DRV_BOX_BAD;
- }
- if (uid) {
- make_key( &key, (char *)name );
- value.data = uid;
- value.size = sizeof(*uid);
- if ((ret = ctx->db->put( ctx->db, 0, &key, &value, 0 )))
- goto tbork;
- }
- if ((ret = ctx->db->sync( ctx->db, 0 ))) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->sync()" );
- goto bork;
- }
- return DRV_OK;
- }
- #endif /* USE_DB */
- static int
- maildir_store_uid( maildir_store_t *ctx )
- {
- int n;
- char buf[128];
- n = sprintf( buf, "%d\n%d\n", ctx->gen.uidvalidity, ctx->nuid );
- lseek( ctx->uvfd, 0, SEEK_SET );
- if (write( ctx->uvfd, buf, n ) != n || ftruncate( ctx->uvfd, n )) {
- error( "Maildir error: cannot write UIDVALIDITY.\n" );
- return DRV_BOX_BAD;
- }
- return DRV_OK;
- }
- static int
- maildir_init_uid( maildir_store_t *ctx, const char *msg )
- {
- info( "Maildir notice: %s.\n", msg ? msg : "cannot read UIDVALIDITY, creating new" );
- ctx->gen.uidvalidity = time( 0 );
- ctx->nuid = 0;
- ctx->uvok = 0;
- #ifdef USE_DB
- if (ctx->db) {
- ctx->db->truncate( ctx->db, 0, 0 /* &u_int32_t_dummy */, 0 );
- return maildir_set_uid( ctx, 0, 0 );
- }
- #endif /* USE_DB */
- return maildir_store_uid( ctx );
- }
- static int
- maildir_uidval_lock( maildir_store_t *ctx )
- {
- int n;
- char buf[128];
- #ifdef LEGACY_FLOCK
- /* This is legacy only */
- if (flock( ctx->uvfd, LOCK_EX ) < 0) {
- error( "Maildir error: cannot flock UIDVALIDITY.\n" );
- return DRV_BOX_BAD;
- }
- #endif
- /* This (theoretically) works over NFS. Let's hope nobody else did
- the same in the opposite order, as we'd deadlock then. */
- #if SEEK_SET != 0
- lck.l_whence = SEEK_SET;
- #endif
- lck.l_type = F_WRLCK;
- if (fcntl( ctx->uvfd, F_SETLKW, &lck )) {
- error( "Maildir error: cannot fcntl lock UIDVALIDITY.\n" );
- return DRV_BOX_BAD;
- }
- lseek( ctx->uvfd, 0, SEEK_SET );
- if ((n = read( ctx->uvfd, buf, sizeof(buf) )) <= 0 ||
- (buf[n] = 0, sscanf( buf, "%d\n%d", &ctx->gen.uidvalidity, &ctx->nuid ) != 2)) {
- return maildir_init_uid( ctx, 0 );
- } else
- ctx->uvok = 1;
- return DRV_OK;
- }
- static void
- maildir_uidval_unlock( maildir_store_t *ctx )
- {
- lck.l_type = F_UNLCK;
- fcntl( ctx->uvfd, F_SETLK, &lck );
- #ifdef LEGACY_FLOCK
- /* This is legacy only */
- flock( ctx->uvfd, LOCK_UN );
- #endif
- }
- static int
- maildir_obtain_uid( maildir_store_t *ctx, int *uid )
- {
- *uid = ++ctx->nuid;
- return maildir_store_uid( ctx );
- }
- static int
- maildir_compare( const void *l, const void *r )
- {
- msg_t *lm = (msg_t *)l, *rm = (msg_t *)r;
- char *ldot, *rdot, *ldot2, *rdot2, *lseq, *rseq;
- int ret, llen, rlen;
- if ((ret = lm->uid - rm->uid))
- return ret;
- /* No UID, so sort by arrival date. We should not do this, but we rely
- on the suggested unique file name scheme - we have no choice. */
- /* The first field are always the seconds. Alphabetical sort should be
- faster than numeric. */
- if (!(ldot = strchr( lm->base, '.' )) || !(rdot = strchr( rm->base, '.' )))
- goto stronly; /* Should never happen ... */
- llen = ldot - lm->base, rlen = rdot - rm->base;
- /* The shorter number is smaller. Really. This won't trigger with any
- mail created after Sep 9 2001 anyway. */
- if ((ret = llen - rlen))
- return ret;
- if ((ret = memcmp( lm->base, rm->base, llen )))
- return ret;
- ldot++, rdot++;
- if ((llen = strtol( ldot, &ldot2, 10 ))) {
- if (!(rlen = strtol( rdot, &rdot2, 10 )))
- goto stronly; /* Comparing apples to oranges ... */
- /* Classical PID specs */
- if ((ret = llen - rlen)) {
- retpid:
- /* Handle PID wraparound. This works only on systems
- where PIDs are not reused too fast */
- if (ret > 20000 || ret < -20000)
- ret = -ret;
- return ret;
- }
- return (*ldot2 != '_' ? 0 : atoi( ldot2 + 1 )) -
- (*rdot2 != '_' ? 0 : atoi( rdot2 + 1 ));
- }
- if (!(ldot2 = strchr( ldot, '.' )) || !(rdot2 = strchr( rdot, '.' )))
- goto stronly; /* Should never happen ... */
- llen = ldot2 - ldot, rlen = rdot2 - rdot;
- if (((lseq = memchr( ldot, '#', llen )) && (rseq = memchr( rdot, '#', rlen ))) ||
- ((lseq = memchr( ldot, 'M', llen )) && (rseq = memchr( rdot, 'M', rlen ))))
- return atoi( lseq + 1 ) - atoi( rseq + 1 );
- if ((lseq = memchr( ldot, 'P', llen )) && (rseq = memchr( rdot, 'P', rlen ))) {
- if ((ret = atoi( lseq + 1 ) - atoi( rseq + 1 )))
- goto retpid;
- if ((lseq = memchr( ldot, 'Q', llen )) && (rseq = memchr( rdot, 'Q', rlen )))
- return atoi( lseq + 1 ) - atoi( rseq + 1 );
- }
- stronly:
- /* Fall-back, so the sort order is defined at all */
- return strcmp( lm->base, rm->base );
- }
- static int
- maildir_scan( maildir_store_t *ctx, msglist_t *msglist )
- {
- DIR *d;
- FILE *f;
- struct dirent *e;
- const char *u, *ru;
- #ifdef USE_DB
- DB *tdb;
- DBC *dbc;
- #endif /* USE_DB */
- msg_t *entry;
- int i, j, uid, bl, ml, fnl, ret;
- struct stat st;
- char buf[_POSIX_PATH_MAX], nbuf[_POSIX_PATH_MAX];
- #ifdef USE_DB
- if (!ctx->db)
- #endif /* USE_DB */
- if ((ret = maildir_uidval_lock( ctx )) != DRV_OK)
- return ret;
- again:
- msglist->ents = 0;
- msglist->nents = msglist->nalloc = 0;
- ctx->gen.count = ctx->gen.recent = 0;
- if (ctx->uvok || ctx->maxuid == INT_MAX) {
- #ifdef USE_DB
- if (ctx->db) {
- if (db_create( &tdb, 0, 0 )) {
- fputs( "Maildir error: db_create() failed\n", stderr );
- return DRV_BOX_BAD;
- }
- if (tdb->open( tdb, 0, 0, 0, DB_HASH, DB_CREATE, 0 )) {
- fputs( "Maildir error: tdb->open() failed\n", stderr );
- tdb->close( tdb, 0 );
- return DRV_BOX_BAD;
- }
- }
- #endif /* USE_DB */
- bl = nfsnprintf( buf, sizeof(buf) - 4, "%s/", ctx->gen.path );
- for (i = 0; i < 2; i++) {
- memcpy( buf + bl, subdirs[i], 4 );
- if (!(d = opendir( buf ))) {
- perror( buf );
- #ifdef USE_DB
- if (!ctx->db)
- #endif /* USE_DB */
- maildir_uidval_unlock( ctx );
- #ifdef USE_DB
- bork:
- #endif /* USE_DB */
- maildir_free_scan( msglist );
- #ifdef USE_DB
- if (ctx->db)
- tdb->close( tdb, 0 );
- #endif /* USE_DB */
- return DRV_BOX_BAD;
- }
- while ((e = readdir( d ))) {
- if (*e->d_name == '.')
- continue;
- ctx->gen.count++;
- ctx->gen.recent += i;
- #ifdef USE_DB
- if (ctx->db) {
- make_key( &key, e->d_name );
- if ((ret = ctx->db->get( ctx->db, 0, &key, &value, 0 ))) {
- if (ret != DB_NOTFOUND) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->get()" );
- ctx->db->close( ctx->db, 0 );
- ctx->db = 0;
- goto bork;
- }
- uid = INT_MAX;
- } else {
- value.size = 0;
- if ((ret = tdb->put( tdb, 0, &key, &value, 0 ))) {
- tdb->err( tdb, ret, "Maildir error: tdb->put()" );
- goto bork;
- }
- uid = *(int *)value.data;
- }
- } else
- #endif /* USE_DB */
- {
- uid = (ctx->uvok && (u = strstr( e->d_name, ",U=" ))) ? atoi( u + 3 ) : 0;
- if (!uid)
- uid = INT_MAX;
- }
- if (uid <= ctx->maxuid) {
- if (uid < ctx->minuid) {
- for (j = 0; j < ctx->nexcs; j++)
- if (ctx->excs[j] == uid)
- goto oke;
- continue;
- oke: ;
- }
- if (msglist->nalloc == msglist->nents) {
- msglist->nalloc = msglist->nalloc * 2 + 100;
- msglist->ents = nfrealloc( msglist->ents, msglist->nalloc * sizeof(msg_t) );
- }
- entry = &msglist->ents[msglist->nents++];
- entry->base = nfstrdup( e->d_name );
- entry->uid = uid;
- entry->recent = i;
- entry->size = 0;
- entry->tuid[0] = 0;
- }
- }
- closedir( d );
- }
- #ifdef USE_DB
- if (ctx->db) {
- if ((ret = ctx->db->cursor( ctx->db, 0, &dbc, 0 )))
- ctx->db->err( ctx->db, ret, "Maildir error: db->cursor()" );
- else {
- for (;;) {
- if ((ret = dbc->c_get( dbc, &key, &value, DB_NEXT ))) {
- if (ret != DB_NOTFOUND)
- ctx->db->err( ctx->db, ret, "Maildir error: db->c_get()" );
- break;
- }
- if ((key.size != 11 || memcmp( key.data, "UIDVALIDITY", 11 )) &&
- (ret = tdb->get( tdb, 0, &key, &value, 0 ))) {
- if (ret != DB_NOTFOUND) {
- tdb->err( tdb, ret, "Maildir error: tdb->get()" );
- break;
- }
- if ((ret = dbc->c_del( dbc, 0 ))) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->c_del()" );
- break;
- }
- }
- }
- dbc->c_close( dbc );
- }
- tdb->close( tdb, 0 );
- }
- #endif /* USE_DB */
- qsort( msglist->ents, msglist->nents, sizeof(msg_t), maildir_compare );
- for (uid = i = 0; i < msglist->nents; i++) {
- entry = &msglist->ents[i];
- if (entry->uid != INT_MAX) {
- if (uid == entry->uid) {
- if ((ret = maildir_init_uid( ctx, "duplicate UID; changing UIDVALIDITY" )) != DRV_OK) {
- maildir_free_scan( msglist );
- return ret;
- }
- maildir_free_scan( msglist );
- goto again;
- }
- uid = entry->uid;
- if (ctx->gen.opts & (OPEN_SIZE|OPEN_FIND))
- nfsnprintf( buf + bl, sizeof(buf) - bl, "%s/%s", subdirs[entry->recent], entry->base );
- #ifdef USE_DB
- } else if (ctx->db) {
- if ((ret = maildir_set_uid( ctx, entry->base, &uid )) != DRV_OK) {
- maildir_free_scan( msglist );
- return ret;
- }
- entry->uid = uid;
- if (ctx->gen.opts & (OPEN_SIZE|OPEN_FIND))
- nfsnprintf( buf + bl, sizeof(buf) - bl, "%s/%s", subdirs[entry->recent], entry->base );
- #endif /* USE_DB */
- } else {
- if ((ret = maildir_obtain_uid( ctx, &uid )) != DRV_OK) {
- maildir_free_scan( msglist );
- return ret;
- }
- entry->uid = uid;
- if ((u = strstr( entry->base, ",U=" )))
- for (ru = u + 3; isdigit( (unsigned char)*ru ); ru++);
- else
- u = ru = strchr( entry->base, ':' );
- if (u)
- ml = u - entry->base;
- else
- ru = "", ml = INT_MAX;
- fnl = nfsnprintf( buf + bl, sizeof(buf) - bl, "%s/%.*s,U=%d%s", subdirs[entry->recent], ml, entry->base, uid, ru ) + 1 - 4;
- memcpy( nbuf, buf, bl + 4 );
- nfsnprintf( nbuf + bl + 4, sizeof(nbuf) - bl - 4, "%s", entry->base );
- if (rename( nbuf, buf )) {
- notok:
- if (errno != ENOENT) {
- perror( buf );
- maildir_uidval_unlock( ctx );
- maildir_free_scan( msglist );
- return DRV_BOX_BAD;
- }
- maildir_free_scan( msglist );
- goto again;
- }
- free( entry->base );
- entry->base = nfmalloc( fnl );
- memcpy( entry->base, buf + bl + 4, fnl );
- }
- if (ctx->gen.opts & OPEN_SIZE) {
- if (stat( buf, &st ))
- goto notok;
- entry->size = st.st_size;
- }
- if (ctx->gen.opts & OPEN_FIND) {
- if (!(f = fopen( buf, "r" )))
- goto notok;
- while (fgets( nbuf, sizeof(nbuf), f )) {
- if (!nbuf[0] || nbuf[0] == '\n')
- break;
- if (!memcmp( nbuf, "X-TUID: ", 8 ) && nbuf[8 + TUIDL] == '\n') {
- memcpy( entry->tuid, nbuf + 8, TUIDL );
- break;
- }
- }
- fclose( f );
- }
- }
- ctx->uvok = 1;
- }
- #ifdef USE_DB
- if (!ctx->db)
- #endif /* ! USE_DB */
- maildir_uidval_unlock( ctx );
- return DRV_OK;
- }
- static void
- maildir_init_msg( maildir_store_t *ctx, maildir_message_t *msg, msg_t *entry )
- {
- msg->base = entry->base;
- entry->base = 0; /* prevent deletion */
- msg->gen.size = entry->size;
- strncpy( msg->tuid, entry->tuid, TUIDL );
- if (entry->recent)
- msg->gen.status |= M_RECENT;
- if (ctx->gen.opts & OPEN_FLAGS) {
- msg->gen.status |= M_FLAGS;
- msg->gen.flags = maildir_parse_flags( msg->base );
- } else
- msg->gen.flags = 0;
- }
- static void
- maildir_app_msg( maildir_store_t *ctx, message_t ***msgapp, msg_t *entry )
- {
- maildir_message_t *msg = nfmalloc( sizeof(*msg) );
- msg->gen.next = **msgapp;
- **msgapp = &msg->gen;
- *msgapp = &msg->gen.next;
- msg->gen.uid = entry->uid;
- msg->gen.status = 0;
- maildir_init_msg( ctx, msg, entry );
- }
- static void
- maildir_prepare_paths( store_t *gctx )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- maildir_cleanup( gctx );
- gctx->msgs = 0;
- ctx->uvfd = -1;
- #ifdef USE_DB
- ctx->db = 0;
- #endif /* USE_DB */
- if (!strcmp( gctx->name, "INBOX" ))
- gctx->path = nfstrdup( ((maildir_store_conf_t *)gctx->conf)->inbox );
- else
- nfasprintf( &gctx->path, "%s%s", gctx->conf->path, gctx->name );
- }
- static void
- maildir_prepare_opts( store_t *gctx, int opts )
- {
- if (opts & OPEN_SETFLAGS)
- opts |= OPEN_OLD;
- if (opts & OPEN_EXPUNGE)
- opts |= OPEN_OLD|OPEN_NEW|OPEN_FLAGS;
- gctx->opts = opts;
- }
- static int
- maildir_select( store_t *gctx, int minuid, int maxuid, int *excs, int nexcs )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- message_t **msgapp;
- msglist_t msglist;
- int i;
- #ifdef USE_DB
- int ret;
- #endif /* USE_DB */
- char uvpath[_POSIX_PATH_MAX];
- ctx->minuid = minuid;
- ctx->maxuid = maxuid;
- ctx->excs = nfrealloc( excs, nexcs * sizeof(int) );
- ctx->nexcs = nexcs;
- if (maildir_validate( gctx->path, "", ctx->gen.opts & OPEN_CREATE ) != DRV_OK)
- return DRV_BOX_BAD;
- nfsnprintf( uvpath, sizeof(uvpath), "%s/.uidvalidity", gctx->path );
- #ifndef USE_DB
- if ((ctx->uvfd = open( uvpath, O_RDWR|O_CREAT, 0600 )) < 0) {
- perror( uvpath );
- return DRV_BOX_BAD;
- }
- #else
- if ((ctx->uvfd = open( uvpath, O_RDWR, 0600 )) < 0) {
- nfsnprintf( uvpath, sizeof(uvpath), "%s/.isyncuidmap.db", gctx->path );
- if ((ctx->uvfd = open( uvpath, O_RDWR, 0600 )) < 0) {
- if (((maildir_store_conf_t *)gctx->conf)->alt_map) {
- if ((ctx->uvfd = open( uvpath, O_RDWR|O_CREAT, 0600 )) >= 0)
- goto dbok;
- } else {
- nfsnprintf( uvpath, sizeof(uvpath), "%s/.uidvalidity", gctx->path );
- if ((ctx->uvfd = open( uvpath, O_RDWR|O_CREAT, 0600 )) >= 0)
- goto fnok;
- }
- perror( uvpath );
- return DRV_BOX_BAD;
- }
- dbok:
- #if SEEK_SET != 0
- lck.l_whence = SEEK_SET;
- #endif
- lck.l_type = F_WRLCK;
- if (fcntl( ctx->uvfd, F_SETLKW, &lck )) {
- perror( uvpath );
- bork:
- close( ctx->uvfd );
- ctx->uvfd = -1;
- return DRV_BOX_BAD;
- }
- if (db_create( &ctx->db, 0, 0 )) {
- fputs( "Maildir error: db_create() failed\n", stderr );
- goto bork;
- }
- if ((ret = ctx->db->open( ctx->db, 0, uvpath, 0, DB_HASH, DB_CREATE, 0 ))) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->open(%s)", uvpath );
- dbork:
- ctx->db->close( ctx->db, 0 );
- goto bork;
- }
- key.data = (void *)"UIDVALIDITY";
- key.size = 11;
- if ((ret = ctx->db->get( ctx->db, 0, &key, &value, 0 ))) {
- if (ret != DB_NOTFOUND) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->get()" );
- goto dbork;
- }
- if (maildir_init_uid( ctx, 0 ) != DRV_OK)
- goto dbork;
- } else {
- ctx->gen.uidvalidity = ((int *)value.data)[0];
- ctx->nuid = ((int *)value.data)[1];
- ctx->uvok = 1;
- }
- }
- fnok:
- #endif /* USE_DB */
- if (maildir_scan( ctx, &msglist ) != DRV_OK)
- return DRV_BOX_BAD;
- msgapp = &ctx->gen.msgs;
- for (i = 0; i < msglist.nents; i++)
- maildir_app_msg( ctx, &msgapp, msglist.ents + i );
- maildir_free_scan( &msglist );
- return DRV_OK;
- }
- static int
- maildir_rescan( maildir_store_t *ctx )
- {
- message_t **msgapp;
- maildir_message_t *msg;
- msglist_t msglist;
- int i;
- if (maildir_scan( ctx, &msglist ) != DRV_OK)
- return DRV_BOX_BAD;
- ctx->gen.recent = 0;
- for (msgapp = &ctx->gen.msgs, i = 0;
- (msg = (maildir_message_t *)*msgapp) || i < msglist.nents; )
- {
- if (!msg) {
- #if 0
- debug( "adding new message %d\n", msglist.ents[i].uid );
- maildir_app_msg( ctx, &msgapp, msglist.ents + i );
- #else
- debug( "ignoring new message %d\n", msglist.ents[i].uid );
- #endif
- i++;
- } else if (i >= msglist.nents) {
- debug( "purging deleted message %d\n", msg->gen.uid );
- msg->gen.status = M_DEAD;
- msgapp = &msg->gen.next;
- } else if (msglist.ents[i].uid < msg->gen.uid) {
- /* this should not happen, actually */
- #if 0
- debug( "adding new message %d\n", msglist.ents[i].uid );
- maildir_app_msg( ctx, &msgapp, msglist.ents + i );
- #else
- debug( "ignoring new message %d\n", msglist.ents[i].uid );
- #endif
- i++;
- } else if (msglist.ents[i].uid > msg->gen.uid) {
- debug( "purging deleted message %d\n", msg->gen.uid );
- msg->gen.status = M_DEAD;
- msgapp = &msg->gen.next;
- } else {
- debug( "updating message %d\n", msg->gen.uid );
- msg->gen.status &= ~(M_FLAGS|M_RECENT);
- free( msg->base );
- maildir_init_msg( ctx, msg, msglist.ents + i );
- i++, msgapp = &msg->gen.next;
- }
- }
- maildir_free_scan( &msglist );
- return DRV_OK;
- }
- static int
- maildir_again( maildir_store_t *ctx, maildir_message_t *msg, const char *fn )
- {
- int ret;
- if (errno != ENOENT) {
- perror( fn );
- return DRV_BOX_BAD;
- }
- if ((ret = maildir_rescan( ctx )) != DRV_OK)
- return ret;
- return (msg->gen.status & M_DEAD) ? DRV_MSG_BAD : DRV_OK;
- }
- static int
- maildir_fetch_msg( store_t *gctx, message_t *gmsg, msg_data_t *data )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- maildir_message_t *msg = (maildir_message_t *)gmsg;
- int fd, ret;
- struct stat st;
- char buf[_POSIX_PATH_MAX];
- for (;;) {
- nfsnprintf( buf, sizeof(buf), "%s/%s/%s", gctx->path, subdirs[gmsg->status & M_RECENT], msg->base );
- if ((fd = open( buf, O_RDONLY )) >= 0)
- break;
- if ((ret = maildir_again( ctx, msg, buf )) != DRV_OK)
- return ret;
- }
- fstat( fd, &st );
- data->len = st.st_size;
- data->data = nfmalloc( data->len );
- if (read( fd, data->data, data->len ) != data->len) {
- perror( buf );
- close( fd );
- return DRV_MSG_BAD;
- }
- close( fd );
- if (!(gmsg->status & M_FLAGS))
- data->flags = maildir_parse_flags( msg->base );
- return DRV_OK;
- }
- static int
- maildir_make_flags( int flags, char *buf )
- {
- unsigned i, d;
- buf[0] = ':';
- buf[1] = '2';
- buf[2] = ',';
- for (d = 3, i = 0; i < as(Flags); i++)
- if (flags & (1 << i))
- buf[d++] = Flags[i];
- buf[d] = 0;
- return d;
- }
- static int
- maildir_store_msg( store_t *gctx, msg_data_t *data, int *uid )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- const char *prefix, *box;
- int ret, fd, bl;
- char buf[_POSIX_PATH_MAX], nbuf[_POSIX_PATH_MAX], fbuf[NUM_FLAGS + 3], base[128];
- bl = nfsnprintf( base, sizeof(base), "%ld.%d_%d.%s", time( 0 ), Pid, ++MaildirCount, Hostname );
- if (uid) {
- #ifdef USE_DB
- if (ctx->db) {
- if ((ret = maildir_set_uid( ctx, base, uid )) != DRV_OK) {
- free( data->data );
- return ret;
- }
- } else
- #endif /* USE_DB */
- {
- if ((ret = maildir_uidval_lock( ctx )) != DRV_OK ||
- (ret = maildir_obtain_uid( ctx, uid )) != DRV_OK)
- return ret;
- maildir_uidval_unlock( ctx );
- nfsnprintf( base + bl, sizeof(base) - bl, ",U=%d", *uid );
- }
- prefix = gctx->path;
- box = "";
- } else {
- prefix = gctx->conf->path;
- box = gctx->conf->trash;
- }
- maildir_make_flags( data->flags, fbuf );
- nfsnprintf( buf, sizeof(buf), "%s%s/tmp/%s%s", prefix, box, base, fbuf );
- if ((fd = open( buf, O_WRONLY|O_CREAT|O_EXCL, 0600 )) < 0) {
- if (errno != ENOENT) {
- perror( buf );
- free( data->data );
- return DRV_BOX_BAD;
- }
- if ((ret = maildir_validate( gctx->conf->path, gctx->conf->trash, gctx->opts & OPEN_CREATE )) != DRV_OK) {
- free( data->data );
- return ret;
- }
- if ((fd = open( buf, O_WRONLY|O_CREAT|O_EXCL, 0600 )) < 0) {
- perror( buf );
- free( data->data );
- return DRV_BOX_BAD;
- }
- }
- ret = write( fd, data->data, data->len );
- free( data->data );
- if (ret != data->len) {
- if (ret < 0)
- perror( buf );
- else
- error( "Maildir error: %s: partial write\n", buf );
- close( fd );
- return DRV_BOX_BAD;
- }
- close( fd );
- nfsnprintf( nbuf, sizeof(nbuf), "%s%s/%s/%s%s", prefix, box, subdirs[!(data->flags & F_SEEN)], base, fbuf );
- if (rename( buf, nbuf )) {
- perror( nbuf );
- return DRV_BOX_BAD;
- }
- if (uid)
- gctx->count++;
- return DRV_OK;
- }
- static int
- maildir_find_msg( store_t *gctx, const char *tuid, int *uid )
- {
- message_t *msg;
- /* using a hash table might turn out to be more appropriate ... */
- for (msg = gctx->msgs; msg; msg = msg->next)
- if (!(msg->status & M_DEAD) && !memcmp( ((maildir_message_t *)msg)->tuid, tuid, TUIDL )) {
- *uid = msg->uid;
- return DRV_OK;
- }
- return DRV_MSG_BAD;
- }
- static int
- maildir_set_flags( store_t *gctx, message_t *gmsg, int uid, int add, int del )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- maildir_message_t *msg = (maildir_message_t *)gmsg;
- char *s, *p;
- unsigned i;
- int j, ret, ol, fl, bbl, bl, tl;
- char buf[_POSIX_PATH_MAX], nbuf[_POSIX_PATH_MAX];
- (void) uid;
- bbl = nfsnprintf( buf, sizeof(buf), "%s/", gctx->path );
- memcpy( nbuf, gctx->path, bbl - 1 );
- memcpy( nbuf + bbl - 1, "/cur/", 5 );
- for (;;) {
- bl = bbl + nfsnprintf( buf + bbl, sizeof(buf) - bbl, "%s/", subdirs[gmsg->status & M_RECENT] );
- ol = strlen( msg->base );
- if ((int)sizeof(buf) - bl < ol + 3 + NUM_FLAGS)
- oob();
- memcpy( buf + bl, msg->base, ol + 1 );
- memcpy( nbuf + bl, msg->base, ol + 1 );
- if ((s = strstr( nbuf + bl, ":2," ))) {
- s += 3;
- fl = ol - (s - (nbuf + bl));
- for (i = 0; i < as(Flags); i++) {
- if ((p = strchr( s, Flags[i] ))) {
- if (del & (1 << i)) {
- memcpy( p, p + 1, fl - (p - s) );
- fl--;
- }
- } else if (add & (1 << i)) {
- for (j = 0; j < fl && Flags[i] > s[j]; j++);
- fl++;
- memmove( s + j + 1, s + j, fl - j );
- s[j] = Flags[i];
- }
- }
- tl = ol + 3 + fl;
- } else {
- tl = ol + maildir_make_flags( msg->gen.flags, nbuf + bl + ol );
- }
- if (!rename( buf, nbuf ))
- break;
- if ((ret = maildir_again( ctx, msg, buf )) != DRV_OK)
- return ret;
- }
- free( msg->base );
- msg->base = nfmalloc( tl + 1 );
- memcpy( msg->base, nbuf + bl, tl + 1 );
- msg->gen.flags |= add;
- msg->gen.flags &= ~del;
- gmsg->status &= ~M_RECENT;
- return DRV_OK;
- }
- #ifdef USE_DB
- static int
- maildir_purge_msg( maildir_store_t *ctx, const char *name )
- {
- int ret;
- make_key( &key, (char *)name );
- if ((ret = ctx->db->del( ctx->db, 0, &key, 0 ))) {
- ctx->db->err( ctx->db, ret, "Maildir error: db->del()" );
- ctx->db->close( ctx->db, 0 );
- ctx->db = 0;
- return DRV_BOX_BAD;
- }
- return DRV_OK;
- }
- #endif /* USE_DB */
- static int
- maildir_trash_msg( store_t *gctx, message_t *gmsg )
- {
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- maildir_message_t *msg = (maildir_message_t *)gmsg;
- char *s;
- int ret;
- struct stat st;
- char buf[_POSIX_PATH_MAX], nbuf[_POSIX_PATH_MAX];
- for (;;) {
- nfsnprintf( buf, sizeof(buf), "%s/%s/%s", gctx->path, subdirs[gmsg->status & M_RECENT], msg->base );
- s = strstr( msg->base, ":2," );
- nfsnprintf( nbuf, sizeof(nbuf), "%s%s/%s/%ld.%d_%d.%s%s", gctx->conf->path, gctx->conf->trash,
- subdirs[gmsg->status & M_RECENT], time( 0 ), Pid, ++MaildirCount, Hostname, s ? s : "" );
- if (!rename( buf, nbuf ))
- break;
- if (!stat( buf, &st )) {
- if ((ret = maildir_validate( gctx->conf->path, gctx->conf->trash, 1 )) != DRV_OK)
- return ret;
- if (!rename( buf, nbuf ))
- break;
- if (errno != ENOENT) {
- perror( nbuf );
- return DRV_BOX_BAD;
- }
- }
- if ((ret = maildir_again( ctx, msg, buf )) != DRV_OK)
- return ret;
- }
- gmsg->status |= M_DEAD;
- gctx->count--;
- #ifdef USE_DB
- if (ctx->db)
- return maildir_purge_msg( ctx, msg->base );
- #endif /* USE_DB */
- return DRV_OK;
- }
- static int
- maildir_close( store_t *gctx )
- {
- #ifdef USE_DB
- maildir_store_t *ctx = (maildir_store_t *)gctx;
- #endif /* USE_DB */
- message_t *msg;
- int basel, retry, ret;
- char buf[_POSIX_PATH_MAX];
- for (;;) {
- retry = 0;
- basel = nfsnprintf( buf, sizeof(buf), "%s/", gctx->path );
- for (msg = gctx->msgs; msg; msg = msg->next)
- if (!(msg->status & M_DEAD) && (msg->flags & F_DELETED)) {
- nfsnprintf( buf + basel, sizeof(buf) - basel, "%s/%s", subdirs[msg->status & M_RECENT], ((maildir_message_t *)msg)->base );
- if (unlink( buf )) {
- if (errno == ENOENT)
- retry = 1;
- else
- perror( buf );
- } else {
- msg->status |= M_DEAD;
- gctx->count--;
- #ifdef USE_DB
- if (ctx->db && (ret = maildir_purge_msg( ctx, ((maildir_message_t *)msg)->base )) != DRV_OK)
- return ret;
- #endif /* USE_DB */
- }
- }
- if (!retry)
- return DRV_OK;
- if ((ret = maildir_rescan( (maildir_store_t *)gctx )) != DRV_OK)
- return ret;
- }
- }
- static int
- maildir_check( store_t *gctx )
- {
- (void) gctx;
- return DRV_OK;
- }
- static int
- maildir_parse_store( conffile_t *cfg, store_conf_t **storep, int *err )
- {
- maildir_store_conf_t *store;
- if (strcasecmp( "MaildirStore", cfg->cmd ))
- return 0;
- store = nfcalloc( sizeof(*store) );
- store->gen.driver = &maildir_driver;
- store->gen.name = nfstrdup( cfg->val );
- while (getcline( cfg ) && cfg->cmd)
- if (!strcasecmp( "Inbox", cfg->cmd ))
- store->inbox = expand_strdup( cfg->val );
- else if (!strcasecmp( "Path", cfg->cmd ))
- store->gen.path = expand_strdup( cfg->val );
- #ifdef USE_DB
- else if (!strcasecmp( "AltMap", cfg->cmd ))
- store->alt_map = parse_bool( cfg );
- #endif /* USE_DB */
- else
- parse_generic_store( &store->gen, cfg, err );
- if (!store->inbox)
- store->inbox = expand_strdup( "~/Maildir" );
- *storep = &store->gen;
- return 1;
- }
- struct driver maildir_driver = {
- 0,
- maildir_parse_store,
- maildir_open_store,
- maildir_close_store,
- maildir_list,
- maildir_prepare_paths,
- maildir_prepare_opts,
- maildir_select,
- maildir_fetch_msg,
- maildir_store_msg,
- maildir_find_msg,
- maildir_set_flags,
- maildir_trash_msg,
- maildir_check,
- maildir_close
- };
|