12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252 |
- /*
- * mbsync - mailbox synchronizer
- * Copyright (C) 2000-2002 Michael R. Elkins <me@mutt.org>
- * Copyright (C) 2002-2006,2010-2013 Oswald Buddenhagen <ossi@users.sf.net>
- *
- * 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, see <http://www.gnu.org/licenses/>.
- *
- * As a special exception, mbsync may be linked with the OpenSSL library,
- * despite that library's more restrictive license.
- */
- #include "sync.h"
- #include <assert.h>
- #include <stdio.h>
- #include <limits.h>
- #include <stdlib.h>
- #include <stddef.h>
- #include <unistd.h>
- #include <time.h>
- #include <fcntl.h>
- #include <ctype.h>
- #include <string.h>
- #include <errno.h>
- #include <sys/stat.h>
- #if !defined(_POSIX_SYNCHRONIZED_IO) || _POSIX_SYNCHRONIZED_IO <= 0
- # define fdatasync fsync
- #endif
- channel_conf_t global_conf;
- channel_conf_t *channels;
- group_conf_t *groups;
- const char *str_ms[] = { "master", "slave" }, *str_hl[] = { "push", "pull" };
- static void ATTR_PRINTFLIKE(1, 2)
- debug( const char *msg, ... )
- {
- va_list va;
- va_start( va, msg );
- vdebug( DEBUG_SYNC, msg, va );
- va_end( va );
- }
- static void ATTR_PRINTFLIKE(1, 2)
- debugn( const char *msg, ... )
- {
- va_list va;
- va_start( va, msg );
- vdebugn( DEBUG_SYNC, msg, va );
- va_end( va );
- }
- static void
- Fclose( FILE *f, int safe )
- {
- if ((safe && (fflush( f ) || (UseFSync && fdatasync( fileno( f ) )))) || fclose( f ) == EOF) {
- sys_error( "Error: cannot close file" );
- exit( 1 );
- }
- }
- static void ATTR_PRINTFLIKE(2, 0)
- vFprintf( FILE *f, const char *msg, va_list va )
- {
- int r;
- r = vfprintf( f, msg, va );
- if (r < 0) {
- sys_error( "Error: cannot write file" );
- exit( 1 );
- }
- }
- static void ATTR_PRINTFLIKE(2, 3)
- Fprintf( FILE *f, const char *msg, ... )
- {
- va_list va;
- va_start( va, msg );
- vFprintf( f, msg, va );
- va_end( va );
- }
- /* Keep the mailbox driver flag definitions in sync: */
- /* grep for MAILBOX_DRIVER_FLAG */
- /* The order is according to alphabetical maildir flag sort */
- static const char Flags[] = { 'D', 'F', 'P', 'R', 'S', 'T' };
- static uchar
- parse_flags( const char *buf )
- {
- uint i, d;
- uchar flags;
- for (flags = i = d = 0; i < as(Flags); i++)
- if (buf[d] == Flags[i]) {
- flags |= (1 << i);
- d++;
- }
- return flags;
- }
- static uint
- make_flags( uchar flags, char *buf )
- {
- uint i, d;
- for (i = d = 0; i < as(Flags); i++)
- if (flags & (1 << i))
- buf[d++] = Flags[i];
- buf[d] = 0;
- return d;
- }
- // These is the (mostly) persistent status of the sync record.
- // Most of these bits are actually mutually exclusive. It is a
- // bitfield to allow for easy testing for multiple states.
- #define S_EXPIRE (1<<0) // the entry is being expired (slave message removal scheduled)
- #define S_EXPIRED (1<<1) // the entry is expired (slave message removal confirmed)
- #define S_PENDING (1<<2) // the entry is new and awaits propagation (possibly a retry)
- #define S_SKIPPED (1<<3) // the entry was not propagated (message is too big)
- #define S_DEAD (1<<7) // ephemeral: the entry was killed and should be ignored
- // Ephemeral working set.
- #define W_NEXPIRE (1<<0) // temporary: new expiration state
- #define W_DELETE (1<<1) // ephemeral: flags propagation is a deletion
- #define W_DEL(ms) (1<<(2+(ms))) // ephemeral: m/s message would be subject to expunge
- typedef struct sync_rec {
- struct sync_rec *next;
- /* string_list_t *keywords; */
- uint uid[2];
- message_t *msg[2];
- uchar status, wstate, flags, aflags[2], dflags[2];
- char tuid[TUIDL];
- } sync_rec_t;
- typedef struct {
- int t[2];
- void (*cb)( int sts, void *aux ), *aux;
- char *dname, *jname, *nname, *lname, *box_name[2];
- FILE *jfp, *nfp;
- sync_rec_t *srecs, **srecadd;
- channel_conf_t *chan;
- store_t *ctx[2];
- driver_t *drv[2];
- const char *orig_name[2];
- message_t *msgs[2], *new_msgs[2];
- uint_array_alloc_t trashed_msgs[2];
- int state[2], lfd, ret, existing, replayed;
- uint ref_count, nsrecs, opts[2];
- uint new_pending[2], flags_pending[2], trash_pending[2];
- uint maxuid[2]; // highest UID that was already propagated
- uint newmaxuid[2]; // highest UID that is currently being propagated
- uint uidval[2]; // UID validity value
- uint newuidval[2]; // UID validity obtained from driver
- uint newuid[2]; // TUID lookup makes sense only for UIDs >= this
- uint mmaxxuid; // highest expired UID on master
- uchar good_flags[2], bad_flags[2];
- } sync_vars_t;
- static void sync_ref( sync_vars_t *svars ) { ++svars->ref_count; }
- static void sync_deref( sync_vars_t *svars );
- static int check_cancel( sync_vars_t *svars );
- #define AUX &svars->t[t]
- #define INV_AUX &svars->t[1-t]
- #define DECL_SVARS \
- int t; \
- sync_vars_t *svars
- #define INIT_SVARS(aux) \
- t = *(int *)aux; \
- svars = (sync_vars_t *)(((char *)(&((int *)aux)[-t])) - offsetof(sync_vars_t, t))
- #define DECL_INIT_SVARS(aux) \
- int t = *(int *)aux; \
- sync_vars_t *svars = (sync_vars_t *)(((char *)(&((int *)aux)[-t])) - offsetof(sync_vars_t, t))
- /* operation dependencies:
- select(x): -
- load(x): select(x)
- new(M), new(S), flags(M), flags(S): load(M) & load(S)
- find_new(x): new(x)
- trash(x): flags(x)
- close(x): trash(x) & find_new(x) & new(!x) // with expunge
- cleanup: close(M) & close(S)
- */
- #define ST_LOADED (1<<0)
- #define ST_FIND_OLD (1<<1)
- #define ST_SENT_NEW (1<<2)
- #define ST_FIND_NEW (1<<3)
- #define ST_FOUND_NEW (1<<4)
- #define ST_SENT_FLAGS (1<<5)
- #define ST_SENT_TRASH (1<<6)
- #define ST_CLOSED (1<<7)
- #define ST_SENT_CANCEL (1<<8)
- #define ST_CANCELED (1<<9)
- #define ST_SELECTED (1<<10)
- #define ST_DID_EXPUNGE (1<<11)
- #define ST_CLOSING (1<<12)
- #define ST_CONFIRMED (1<<13)
- #define ST_PRESENT (1<<14)
- #define ST_SENDING_NEW (1<<15)
- static void ATTR_PRINTFLIKE(2, 3)
- jFprintf( sync_vars_t *svars, const char *msg, ... )
- {
- va_list va;
- if (JLimit && !--JLimit)
- exit( 101 );
- va_start( va, msg );
- vFprintf( svars->jfp, msg, va );
- va_end( va );
- if (JLimit && !--JLimit)
- exit( 100 );
- }
- static void
- match_tuids( sync_vars_t *svars, int t, message_t *msgs )
- {
- sync_rec_t *srec;
- message_t *tmsg, *ntmsg = NULL;
- const char *diag;
- int num_lost = 0;
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (!srec->uid[t] && srec->tuid[0]) {
- debug( " pair(%u,%u): lookup %s, TUID %." stringify(TUIDL) "s\n", srec->uid[M], srec->uid[S], str_ms[t], srec->tuid );
- for (tmsg = ntmsg; tmsg; tmsg = tmsg->next) {
- if (tmsg->status & M_DEAD)
- continue;
- if (tmsg->tuid[0] && !memcmp( tmsg->tuid, srec->tuid, TUIDL )) {
- diag = (tmsg == ntmsg) ? "adjacently" : "after gap";
- goto mfound;
- }
- }
- for (tmsg = msgs; tmsg != ntmsg; tmsg = tmsg->next) {
- if (tmsg->status & M_DEAD)
- continue;
- if (tmsg->tuid[0] && !memcmp( tmsg->tuid, srec->tuid, TUIDL )) {
- diag = "after reset";
- goto mfound;
- }
- }
- debug( " -> TUID lost\n" );
- jFprintf( svars, "& %u %u\n", srec->uid[M], srec->uid[S] );
- // Note: status remains S_PENDING.
- srec->tuid[0] = 0;
- num_lost++;
- continue;
- mfound:
- debug( " -> new UID %u %s\n", tmsg->uid, diag );
- jFprintf( svars, "%c %u %u %u\n", "<>"[t], srec->uid[M], srec->uid[S], tmsg->uid );
- tmsg->srec = srec;
- srec->msg[t] = tmsg;
- ntmsg = tmsg->next;
- srec->uid[t] = tmsg->uid;
- srec->status = 0;
- srec->tuid[0] = 0;
- }
- }
- if (num_lost)
- warn( "Warning: lost track of %d %sed message(s)\n", num_lost, str_hl[t] );
- }
- static uchar
- sanitize_flags( uchar tflags, sync_vars_t *svars, int t )
- {
- if (!(DFlags & QUIET)) {
- // We complain only once per flag per store - even though _theoretically_
- // each mailbox can support different flags according to the IMAP spec.
- uchar bflags = tflags & ~(svars->good_flags[t] | svars->bad_flags[t]);
- if (bflags) {
- char bfbuf[16];
- make_flags( bflags, bfbuf );
- notice( "Notice: %s does not support flag(s) '%s'; not propagating.\n", str_ms[t], bfbuf );
- svars->bad_flags[t] |= bflags;
- }
- }
- return tflags & svars->good_flags[t];
- }
- typedef struct copy_vars {
- void (*cb)( int sts, uint uid, struct copy_vars *vars );
- void *aux;
- sync_rec_t *srec; /* also ->tuid */
- message_t *msg;
- msg_data_t data;
- } copy_vars_t;
- static void msg_fetched( int sts, void *aux );
- static void
- copy_msg( copy_vars_t *vars )
- {
- DECL_INIT_SVARS(vars->aux);
- t ^= 1;
- vars->data.flags = vars->msg->flags;
- vars->data.date = svars->chan->use_internal_date ? -1 : 0;
- svars->drv[t]->fetch_msg( svars->ctx[t], vars->msg, &vars->data, msg_fetched, vars );
- }
- static void msg_stored( int sts, uint uid, void *aux );
- static void
- copy_msg_bytes( char **out_ptr, const char *in_buf, uint *in_idx, uint in_len, int in_cr, int out_cr )
- {
- char *out = *out_ptr;
- uint idx = *in_idx;
- if (out_cr != in_cr) {
- char c;
- if (out_cr) {
- for (; idx < in_len; idx++) {
- if ((c = in_buf[idx]) != '\r') {
- if (c == '\n')
- *out++ = '\r';
- *out++ = c;
- }
- }
- } else {
- for (; idx < in_len; idx++) {
- if ((c = in_buf[idx]) != '\r')
- *out++ = c;
- }
- }
- } else {
- memcpy( out, in_buf + idx, in_len - idx );
- out += in_len - idx;
- idx = in_len;
- }
- *out_ptr = out;
- *in_idx = idx;
- }
- static int
- copy_msg_convert( int in_cr, int out_cr, copy_vars_t *vars )
- {
- char *in_buf = vars->data.data;
- uint in_len = vars->data.len;
- uint idx = 0, sbreak = 0, ebreak = 0;
- uint lines = 0, hdr_crs = 0, bdy_crs = 0, app_cr = 0, extra = 0;
- if (vars->srec) {
- nloop: ;
- uint start = idx;
- uint line_crs = 0;
- while (idx < in_len) {
- char c = in_buf[idx++];
- if (c == '\r') {
- line_crs++;
- } else if (c == '\n') {
- if (starts_with_upper( in_buf + start, (int)(in_len - start), "X-TUID: ", 8 )) {
- extra = (sbreak = start) - (ebreak = idx);
- goto oke;
- }
- lines++;
- hdr_crs += line_crs;
- if (idx - line_crs - 1 == start) {
- sbreak = ebreak = start;
- goto oke;
- }
- goto nloop;
- }
- }
- /* invalid message */
- free( in_buf );
- return 0;
- oke:
- app_cr = out_cr && (!in_cr || hdr_crs);
- extra += 8 + TUIDL + app_cr + 1;
- }
- if (out_cr != in_cr) {
- for (; idx < in_len; idx++) {
- char c = in_buf[idx];
- if (c == '\r')
- bdy_crs++;
- else if (c == '\n')
- lines++;
- }
- extra -= hdr_crs + bdy_crs;
- if (out_cr)
- extra += lines;
- }
- vars->data.len = in_len + extra;
- char *out_buf = vars->data.data = nfmalloc( vars->data.len );
- idx = 0;
- if (vars->srec) {
- copy_msg_bytes( &out_buf, in_buf, &idx, sbreak, in_cr, out_cr );
- memcpy( out_buf, "X-TUID: ", 8 );
- out_buf += 8;
- memcpy( out_buf, vars->srec->tuid, TUIDL );
- out_buf += TUIDL;
- if (app_cr)
- *out_buf++ = '\r';
- *out_buf++ = '\n';
- idx = ebreak;
- }
- copy_msg_bytes( &out_buf, in_buf, &idx, in_len, in_cr, out_cr );
- free( in_buf );
- return 1;
- }
- static void
- msg_fetched( int sts, void *aux )
- {
- copy_vars_t *vars = (copy_vars_t *)aux;
- DECL_SVARS;
- int scr, tcr;
- switch (sts) {
- case DRV_OK:
- INIT_SVARS(vars->aux);
- if (check_cancel( svars )) {
- free( vars->data.data );
- vars->cb( SYNC_CANCELED, 0, vars );
- return;
- }
- vars->msg->flags = vars->data.flags = sanitize_flags( vars->data.flags, svars, t );
- scr = (svars->drv[1-t]->get_caps( svars->ctx[1-t] ) / DRV_CRLF) & 1;
- tcr = (svars->drv[t]->get_caps( svars->ctx[t] ) / DRV_CRLF) & 1;
- if (vars->srec || scr != tcr) {
- if (!copy_msg_convert( scr, tcr, vars )) {
- warn( "Warning: message %u from %s has incomplete header.\n",
- vars->msg->uid, str_ms[1-t] );
- vars->cb( SYNC_NOGOOD, 0, vars );
- return;
- }
- }
- svars->drv[t]->store_msg( svars->ctx[t], &vars->data, !vars->srec, msg_stored, vars );
- break;
- case DRV_CANCELED:
- vars->cb( SYNC_CANCELED, 0, vars );
- break;
- case DRV_MSG_BAD:
- vars->cb( SYNC_NOGOOD, 0, vars );
- break;
- default:
- vars->cb( SYNC_FAIL, 0, vars );
- break;
- }
- }
- static void
- msg_stored( int sts, uint uid, void *aux )
- {
- copy_vars_t *vars = (copy_vars_t *)aux;
- DECL_SVARS;
- switch (sts) {
- case DRV_OK:
- vars->cb( SYNC_OK, uid, vars );
- break;
- case DRV_CANCELED:
- vars->cb( SYNC_CANCELED, 0, vars );
- break;
- case DRV_MSG_BAD:
- INIT_SVARS(vars->aux);
- (void)svars;
- warn( "Warning: %s refuses to store message %u from %s.\n",
- str_ms[t], vars->msg->uid, str_ms[1-t] );
- vars->cb( SYNC_NOGOOD, 0, vars );
- break;
- default:
- vars->cb( SYNC_FAIL, 0, vars );
- break;
- }
- }
- static void sync_bail( sync_vars_t *svars );
- static void sync_bail2( sync_vars_t *svars );
- static void sync_bail3( sync_vars_t *svars );
- static void cancel_done( void *aux );
- static void
- cancel_sync( sync_vars_t *svars )
- {
- int t;
- for (t = 0; t < 2; t++) {
- int other_state = svars->state[1-t];
- if (svars->ret & SYNC_BAD(t)) {
- cancel_done( AUX );
- } else if (!(svars->state[t] & ST_SENT_CANCEL)) {
- /* ignore subsequent failures from in-flight commands */
- svars->state[t] |= ST_SENT_CANCEL;
- svars->drv[t]->cancel_cmds( svars->ctx[t], cancel_done, AUX );
- }
- if (other_state & ST_CANCELED)
- break;
- }
- }
- static void
- cancel_done( void *aux )
- {
- DECL_INIT_SVARS(aux);
- svars->state[t] |= ST_CANCELED;
- if (svars->state[1-t] & ST_CANCELED) {
- if (svars->nfp) {
- Fclose( svars->nfp, 0 );
- Fclose( svars->jfp, 0 );
- }
- sync_bail( svars );
- }
- }
- static void
- store_bad( void *aux )
- {
- DECL_INIT_SVARS(aux);
- svars->drv[t]->cancel_store( svars->ctx[t] );
- svars->ret |= SYNC_BAD(t);
- cancel_sync( svars );
- }
- static int
- check_cancel( sync_vars_t *svars )
- {
- return (svars->state[M] | svars->state[S]) & (ST_SENT_CANCEL | ST_CANCELED);
- }
- static int
- check_ret( int sts, void *aux )
- {
- DECL_SVARS;
- if (sts == DRV_CANCELED)
- return 1;
- INIT_SVARS(aux);
- if (sts == DRV_BOX_BAD) {
- svars->ret |= SYNC_FAIL;
- cancel_sync( svars );
- return 1;
- }
- return check_cancel( svars );
- }
- #define SVARS_CHECK_RET \
- DECL_SVARS; \
- if (check_ret( sts, aux )) \
- return; \
- INIT_SVARS(aux)
- #define SVARS_CHECK_RET_VARS(type) \
- type *vars = (type *)aux; \
- DECL_SVARS; \
- if (check_ret( sts, vars->aux )) { \
- free( vars ); \
- return; \
- } \
- INIT_SVARS(vars->aux)
- #define SVARS_CHECK_CANCEL_RET \
- DECL_SVARS; \
- if (sts == SYNC_CANCELED) { \
- free( vars ); \
- return; \
- } \
- INIT_SVARS(vars->aux)
- static char *
- clean_strdup( const char *s )
- {
- char *cs;
- uint i;
- cs = nfstrdup( s );
- for (i = 0; cs[i]; i++)
- if (cs[i] == '/')
- cs[i] = '!';
- return cs;
- }
- #define JOURNAL_VERSION "3"
- static int
- prepare_state( sync_vars_t *svars )
- {
- char *s, *cmname, *csname;
- channel_conf_t *chan;
- chan = svars->chan;
- if (!strcmp( chan->sync_state ? chan->sync_state : global_conf.sync_state, "*" )) {
- const char *path = svars->drv[S]->get_box_path( svars->ctx[S] );
- if (!path) {
- error( "Error: store '%s' does not support in-box sync state\n", chan->stores[S]->name );
- return 0;
- }
- nfasprintf( &svars->dname, "%s/." EXE "state", path );
- } else {
- csname = clean_strdup( svars->box_name[S] );
- if (chan->sync_state)
- nfasprintf( &svars->dname, "%s%s", chan->sync_state, csname );
- else {
- char c = FieldDelimiter;
- cmname = clean_strdup( svars->box_name[M] );
- nfasprintf( &svars->dname, "%s%c%s%c%s_%c%s%c%s", global_conf.sync_state,
- c, chan->stores[M]->name, c, cmname, c, chan->stores[S]->name, c, csname );
- free( cmname );
- }
- free( csname );
- if (!(s = strrchr( svars->dname, '/' ))) {
- error( "Error: invalid SyncState location '%s'\n", svars->dname );
- return 0;
- }
- *s = 0;
- if (mkdir( svars->dname, 0700 ) && errno != EEXIST) {
- sys_error( "Error: cannot create SyncState directory '%s'", svars->dname );
- return 0;
- }
- *s = '/';
- }
- nfasprintf( &svars->jname, "%s.journal", svars->dname );
- nfasprintf( &svars->nname, "%s.new", svars->dname );
- nfasprintf( &svars->lname, "%s.lock", svars->dname );
- return 1;
- }
- static int
- lock_state( sync_vars_t *svars )
- {
- struct flock lck;
- if (svars->lfd >= 0)
- return 1;
- memset( &lck, 0, sizeof(lck) );
- #if SEEK_SET != 0
- lck.l_whence = SEEK_SET;
- #endif
- #if F_WRLCK != 0
- lck.l_type = F_WRLCK;
- #endif
- if ((svars->lfd = open( svars->lname, O_WRONLY|O_CREAT, 0666 )) < 0) {
- sys_error( "Error: cannot create lock file %s", svars->lname );
- return 0;
- }
- if (fcntl( svars->lfd, F_SETLK, &lck )) {
- error( "Error: channel :%s:%s-:%s:%s is locked\n",
- svars->chan->stores[M]->name, svars->orig_name[M], svars->chan->stores[S]->name, svars->orig_name[S] );
- close( svars->lfd );
- svars->lfd = -1;
- return 0;
- }
- return 1;
- }
- static void
- save_state( sync_vars_t *svars )
- {
- sync_rec_t *srec;
- char fbuf[16]; /* enlarge when support for keywords is added */
- Fprintf( svars->nfp,
- "MasterUidValidity %u\nSlaveUidValidity %u\nMaxPulledUid %u\nMaxPushedUid %u\n",
- svars->uidval[M], svars->uidval[S], svars->maxuid[M], svars->maxuid[S] );
- if (svars->mmaxxuid)
- Fprintf( svars->nfp, "MaxExpiredMasterUid %u\n", svars->mmaxxuid );
- Fprintf( svars->nfp, "\n" );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- make_flags( srec->flags, fbuf );
- Fprintf( svars->nfp, "%u %u %s%s\n", srec->uid[M], srec->uid[S],
- (srec->status & S_SKIPPED) ? "^" : (srec->status & S_PENDING) ? "!" : (srec->status & S_EXPIRED) ? "~" : "", fbuf );
- }
- Fclose( svars->nfp, 1 );
- Fclose( svars->jfp, 0 );
- if (!(DFlags & KEEPJOURNAL)) {
- /* order is important! */
- if (rename( svars->nname, svars->dname ))
- warn( "Warning: cannot commit sync state %s\n", svars->dname );
- else if (unlink( svars->jname ))
- warn( "Warning: cannot delete journal %s\n", svars->jname );
- }
- }
- static int
- load_state( sync_vars_t *svars )
- {
- sync_rec_t *srec, *nsrec;
- char *s;
- FILE *jfp;
- uint ll;
- uint smaxxuid = 0;
- char c;
- struct stat st;
- char fbuf[16]; /* enlarge when support for keywords is added */
- char buf[128], buf1[64], buf2[64];
- if ((jfp = fopen( svars->dname, "r" ))) {
- if (!lock_state( svars ))
- goto jbail;
- debug( "reading sync state %s ...\n", svars->dname );
- int line = 0;
- while (fgets( buf, sizeof(buf), jfp )) {
- line++;
- if (!(ll = strlen( buf )) || buf[ll - 1] != '\n') {
- error( "Error: incomplete sync state header entry at %s:%d\n", svars->dname, line );
- jbail:
- fclose( jfp );
- return 0;
- }
- if (ll == 1)
- goto gothdr;
- if (line == 1 && isdigit( buf[0] )) {
- if (sscanf( buf, "%63s %63s", buf1, buf2 ) != 2 ||
- sscanf( buf1, "%u:%u", &svars->uidval[M], &svars->maxuid[M] ) < 2 ||
- sscanf( buf2, "%u:%u:%u", &svars->uidval[S], &smaxxuid, &svars->maxuid[S] ) < 3) {
- error( "Error: invalid sync state header in %s\n", svars->dname );
- goto jbail;
- }
- goto gothdr;
- }
- uint uid;
- if (sscanf( buf, "%63s %u", buf1, &uid ) != 2) {
- error( "Error: malformed sync state header entry at %s:%d\n", svars->dname, line );
- goto jbail;
- }
- if (!strcmp( buf1, "MasterUidValidity" ))
- svars->uidval[M] = uid;
- else if (!strcmp( buf1, "SlaveUidValidity" ))
- svars->uidval[S] = uid;
- else if (!strcmp( buf1, "MaxPulledUid" ))
- svars->maxuid[M] = uid;
- else if (!strcmp( buf1, "MaxPushedUid" ))
- svars->maxuid[S] = uid;
- else if (!strcmp( buf1, "MaxExpiredMasterUid" ))
- svars->mmaxxuid = uid;
- else if (!strcmp( buf1, "MaxExpiredSlaveUid" )) // Legacy
- smaxxuid = uid;
- else {
- error( "Error: unrecognized sync state header entry at %s:%d\n", svars->dname, line );
- goto jbail;
- }
- }
- error( "Error: unterminated sync state header in %s\n", svars->dname );
- goto jbail;
- gothdr:
- while (fgets( buf, sizeof(buf), jfp )) {
- line++;
- if (!(ll = strlen( buf )) || buf[--ll] != '\n') {
- error( "Error: incomplete sync state entry at %s:%d\n", svars->dname, line );
- goto jbail;
- }
- buf[ll] = 0;
- fbuf[0] = 0;
- uint t1, t2;
- if (sscanf( buf, "%u %u %15s", &t1, &t2, fbuf ) < 2) {
- error( "Error: invalid sync state entry at %s:%d\n", svars->dname, line );
- goto jbail;
- }
- srec = nfcalloc( sizeof(*srec) );
- srec->uid[M] = t1;
- srec->uid[S] = t2;
- s = fbuf;
- if (*s == '^') {
- s++;
- srec->status = S_SKIPPED;
- } else if (*s == '!') {
- s++;
- srec->status = S_PENDING;
- } else if (*s == '~' || *s == 'X' /* Pre-1.3 legacy */) {
- s++;
- srec->status = S_EXPIRE | S_EXPIRED;
- } else if (srec->uid[M] == (uint)-1) { // Pre-1.3 legacy
- srec->uid[M] = 0;
- srec->status = S_SKIPPED;
- } else if (srec->uid[M] == (uint)-2) {
- srec->uid[M] = 0;
- srec->status = S_PENDING;
- } else if (srec->uid[S] == (uint)-1) {
- srec->uid[S] = 0;
- srec->status = S_SKIPPED;
- } else if (srec->uid[S] == (uint)-2) {
- srec->uid[S] = 0;
- srec->status = S_PENDING;
- }
- srec->flags = parse_flags( s );
- debug( " entry (%u,%u,%u,%s)\n", srec->uid[M], srec->uid[S], srec->flags,
- (srec->status & S_SKIPPED) ? "SKIP" : (srec->status & S_PENDING) ? "FAIL" : (srec->status & S_EXPIRED) ? "XPIRE" : "" );
- *svars->srecadd = srec;
- svars->srecadd = &srec->next;
- svars->nsrecs++;
- }
- fclose( jfp );
- svars->existing = 1;
- } else {
- if (errno != ENOENT) {
- sys_error( "Error: cannot read sync state %s", svars->dname );
- return 0;
- }
- svars->existing = 0;
- }
- // This is legacy support for pre-1.3 sync states.
- if (smaxxuid) {
- uint minwuid = UINT_MAX;
- for (srec = svars->srecs; srec; srec = srec->next) {
- if ((srec->status & (S_DEAD | S_SKIPPED | S_PENDING)) || !srec->uid[M])
- continue;
- if (srec->status & S_EXPIRED) {
- if (!srec->uid[S]) {
- // The expired message was already gone.
- continue;
- }
- // The expired message was not expunged yet, so re-examine it.
- // This will happen en masse, so just extend the bulk fetch.
- } else {
- if (srec->uid[S] && smaxxuid >= srec->uid[S]) {
- // The non-expired message is in the generally expired range,
- // so don't make it contribute to the bulk fetch.
- continue;
- }
- // Usual non-expired message.
- }
- if (minwuid > srec->uid[M])
- minwuid = srec->uid[M];
- }
- svars->mmaxxuid = minwuid - 1;
- }
- svars->newmaxuid[M] = svars->maxuid[M];
- svars->newmaxuid[S] = svars->maxuid[S];
- int line = 0;
- if ((jfp = fopen( svars->jname, "r" ))) {
- if (!lock_state( svars ))
- goto jbail;
- if (!stat( svars->nname, &st ) && fgets( buf, sizeof(buf), jfp )) {
- debug( "recovering journal ...\n" );
- if (!(ll = strlen( buf )) || buf[--ll] != '\n') {
- error( "Error: incomplete journal header in %s\n", svars->jname );
- goto jbail;
- }
- buf[ll] = 0;
- if (!equals( buf, (int)ll, JOURNAL_VERSION, strlen(JOURNAL_VERSION) )) {
- error( "Error: incompatible journal version "
- "(got %s, expected " JOURNAL_VERSION ")\n", buf );
- goto jbail;
- }
- srec = NULL;
- line = 1;
- while (fgets( buf, sizeof(buf), jfp )) {
- line++;
- if (!(ll = strlen( buf )) || buf[--ll] != '\n') {
- error( "Error: incomplete journal entry at %s:%d\n", svars->jname, line );
- goto jbail;
- }
- buf[ll] = 0;
- int tn;
- uint t1, t2, t3;
- if ((c = buf[0]) == '#' ?
- (tn = 0, (sscanf( buf + 2, "%u %u %n", &t1, &t2, &tn ) < 2) || !tn || (ll - (uint)tn != TUIDL + 2)) :
- c == 'S' || c == '!' ?
- (sscanf( buf + 2, "%u", &t1 ) != 1) :
- c == 'F' || c == 'T' || c == '+' || c == '&' || c == '-' || c == '=' || c == '|' ?
- (sscanf( buf + 2, "%u %u", &t1, &t2 ) != 2) :
- (sscanf( buf + 2, "%u %u %u", &t1, &t2, &t3 ) != 3))
- {
- error( "Error: malformed journal entry at %s:%d\n", svars->jname, line );
- goto jbail;
- }
- if (c == 'S')
- svars->maxuid[t1] = svars->newmaxuid[t1];
- else if (c == 'F')
- svars->newuid[t1] = t2;
- else if (c == 'T')
- *uint_array_append( &svars->trashed_msgs[t1] ) = t2;
- else if (c == '!')
- svars->mmaxxuid = t1;
- else if (c == '|') {
- svars->uidval[M] = t1;
- svars->uidval[S] = t2;
- } else if (c == '+') {
- srec = nfcalloc( sizeof(*srec) );
- srec->uid[M] = t1;
- srec->uid[S] = t2;
- if (svars->newmaxuid[M] < t1)
- svars->newmaxuid[M] = t1;
- if (svars->newmaxuid[S] < t2)
- svars->newmaxuid[S] = t2;
- debug( " new entry(%u,%u)\n", t1, t2 );
- srec->status = S_PENDING;
- *svars->srecadd = srec;
- svars->srecadd = &srec->next;
- svars->nsrecs++;
- } else {
- for (nsrec = srec; srec; srec = srec->next)
- if (srec->uid[M] == t1 && srec->uid[S] == t2)
- goto syncfnd;
- for (srec = svars->srecs; srec != nsrec; srec = srec->next)
- if (srec->uid[M] == t1 && srec->uid[S] == t2)
- goto syncfnd;
- error( "Error: journal entry at %s:%d refers to non-existing sync state entry\n", svars->jname, line );
- goto jbail;
- syncfnd:
- debugn( " entry(%u,%u,%u) ", srec->uid[M], srec->uid[S], srec->flags );
- switch (c) {
- case '-':
- debug( "killed\n" );
- srec->status = S_DEAD;
- break;
- case '=':
- debug( "aborted\n" );
- svars->mmaxxuid = srec->uid[M];
- srec->status = S_DEAD;
- break;
- case '#':
- memcpy( srec->tuid, buf + tn + 2, TUIDL );
- debug( "TUID now %." stringify(TUIDL) "s\n", srec->tuid );
- break;
- case '&':
- debug( "TUID %." stringify(TUIDL) "s lost\n", srec->tuid );
- srec->tuid[0] = 0;
- break;
- case '<':
- debug( "master now %u\n", t3 );
- srec->uid[M] = t3;
- srec->status &= ~S_PENDING;
- srec->tuid[0] = 0;
- break;
- case '>':
- debug( "slave now %u\n", t3 );
- srec->uid[S] = t3;
- srec->status &= ~S_PENDING;
- srec->tuid[0] = 0;
- break;
- case '*':
- debug( "flags now %u\n", t3 );
- srec->flags = (uchar)t3;
- break;
- case '~':
- debug( "status now %#x\n", t3 );
- srec->status = (uchar)t3;
- break;
- default:
- error( "Error: unrecognized journal entry at %s:%d\n", svars->jname, line );
- goto jbail;
- }
- }
- }
- }
- fclose( jfp );
- } else {
- if (errno != ENOENT) {
- sys_error( "Error: cannot read journal %s", svars->jname );
- return 0;
- }
- }
- svars->replayed = line;
- return 1;
- }
- static void
- delete_state( sync_vars_t *svars )
- {
- unlink( svars->nname );
- unlink( svars->jname );
- if (unlink( svars->dname ) || unlink( svars->lname )) {
- sys_error( "Error: channel %s: sync state cannot be deleted", svars->chan->name );
- svars->ret = SYNC_FAIL;
- }
- }
- static void box_confirmed( int sts, uint uidvalidity, void *aux );
- static void box_confirmed2( sync_vars_t *svars, int t );
- static void box_deleted( int sts, void *aux );
- static void box_created( int sts, void *aux );
- static void box_opened( int sts, uint uidvalidity, void *aux );
- static void box_opened2( sync_vars_t *svars, int t );
- static void load_box( sync_vars_t *svars, int t, uint minwuid, uint_array_t mexcs );
- void
- sync_boxes( store_t *ctx[], const char * const names[], int present[], channel_conf_t *chan,
- void (*cb)( int sts, void *aux ), void *aux )
- {
- sync_vars_t *svars;
- int t;
- svars = nfcalloc( sizeof(*svars) );
- svars->t[1] = 1;
- svars->ref_count = 1;
- svars->cb = cb;
- svars->aux = aux;
- svars->ctx[0] = ctx[0];
- svars->ctx[1] = ctx[1];
- svars->chan = chan;
- svars->lfd = -1;
- svars->uidval[0] = svars->uidval[1] = UIDVAL_BAD;
- svars->srecadd = &svars->srecs;
- for (t = 0; t < 2; t++) {
- svars->orig_name[t] =
- (!names[t] || (ctx[t]->conf->map_inbox && !strcmp( ctx[t]->conf->map_inbox, names[t] ))) ?
- "INBOX" : names[t];
- if (!ctx[t]->conf->flat_delim[0]) {
- svars->box_name[t] = nfstrdup( svars->orig_name[t] );
- } else if (map_name( svars->orig_name[t], &svars->box_name[t], 0, "/", ctx[t]->conf->flat_delim ) < 0) {
- error( "Error: canonical mailbox name '%s' contains flattened hierarchy delimiter\n", svars->orig_name[t] );
- bail3:
- svars->ret = SYNC_FAIL;
- sync_bail3( svars );
- return;
- }
- svars->drv[t] = ctx[t]->driver;
- svars->drv[t]->set_bad_callback( ctx[t], store_bad, AUX );
- }
- /* Both boxes must be fully set up at this point, so that error exit paths
- * don't run into uninitialized variables. */
- for (t = 0; t < 2; t++) {
- switch (svars->drv[t]->select_box( ctx[t], svars->box_name[t] )) {
- case DRV_CANCELED:
- store_bad( AUX );
- return;
- case DRV_BOX_BAD:
- goto bail3;
- }
- }
- if (!prepare_state( svars )) {
- svars->ret = SYNC_FAIL;
- sync_bail2( svars );
- return;
- }
- if (!load_state( svars )) {
- svars->ret = SYNC_FAIL;
- sync_bail( svars );
- return;
- }
- sync_ref( svars );
- for (t = 0; ; t++) {
- info( "Opening %s box %s...\n", str_ms[t], svars->orig_name[t] );
- if (present[t] == BOX_ABSENT)
- box_confirmed2( svars, t );
- else
- svars->drv[t]->open_box( ctx[t], box_confirmed, AUX );
- if (t || check_cancel( svars ))
- break;
- }
- sync_deref( svars );
- }
- static void
- box_confirmed( int sts, uint uidvalidity, void *aux )
- {
- DECL_SVARS;
- if (sts == DRV_CANCELED)
- return;
- INIT_SVARS(aux);
- if (check_cancel( svars ))
- return;
- if (sts == DRV_OK) {
- svars->state[t] |= ST_PRESENT;
- svars->newuidval[t] = uidvalidity;
- }
- box_confirmed2( svars, t );
- }
- static void
- box_confirmed2( sync_vars_t *svars, int t )
- {
- svars->state[t] |= ST_CONFIRMED;
- if (!(svars->state[1-t] & ST_CONFIRMED))
- return;
- sync_ref( svars );
- for (t = 0; ; t++) {
- if (!(svars->state[t] & ST_PRESENT)) {
- if (!(svars->state[1-t] & ST_PRESENT)) {
- if (!svars->existing) {
- error( "Error: channel %s: both master %s and slave %s cannot be opened.\n",
- svars->chan->name, svars->orig_name[M], svars->orig_name[S] );
- bail:
- svars->ret = SYNC_FAIL;
- } else {
- /* This can legitimately happen if a deletion propagation was interrupted.
- * We have no place to record this transaction, so we just assume it.
- * Of course this bears the danger of clearing the state if both mailboxes
- * temorarily cannot be opened for some weird reason (while the stores can). */
- delete_state( svars );
- }
- done:
- sync_bail( svars );
- break;
- }
- if (svars->existing) {
- if (!(svars->chan->ops[1-t] & OP_REMOVE)) {
- error( "Error: channel %s: %s %s cannot be opened.\n",
- svars->chan->name, str_ms[t], svars->orig_name[t] );
- goto bail;
- }
- if (svars->drv[1-t]->confirm_box_empty( svars->ctx[1-t] ) != DRV_OK) {
- warn( "Warning: channel %s: %s %s cannot be opened and %s %s not empty.\n",
- svars->chan->name, str_ms[t], svars->orig_name[t], str_ms[1-t], svars->orig_name[1-t] );
- goto done;
- }
- info( "Deleting %s %s...\n", str_ms[1-t], svars->orig_name[1-t] );
- svars->drv[1-t]->delete_box( svars->ctx[1-t], box_deleted, INV_AUX );
- } else {
- if (!(svars->chan->ops[t] & OP_CREATE)) {
- box_opened( DRV_BOX_BAD, UIDVAL_BAD, AUX );
- } else {
- info( "Creating %s %s...\n", str_ms[t], svars->orig_name[t] );
- svars->drv[t]->create_box( svars->ctx[t], box_created, AUX );
- }
- }
- } else {
- box_opened2( svars, t );
- }
- if (t || check_cancel( svars ))
- break;
- }
- sync_deref( svars );
- }
- static void
- box_deleted( int sts, void *aux )
- {
- DECL_SVARS;
- if (check_ret( sts, aux ))
- return;
- INIT_SVARS(aux);
- delete_state( svars );
- svars->drv[t]->finish_delete_box( svars->ctx[t] );
- sync_bail( svars );
- }
- static void
- box_created( int sts, void *aux )
- {
- DECL_SVARS;
- if (check_ret( sts, aux ))
- return;
- INIT_SVARS(aux);
- svars->drv[t]->open_box( svars->ctx[t], box_opened, AUX );
- }
- static void
- box_opened( int sts, uint uidvalidity, void *aux )
- {
- DECL_SVARS;
- if (sts == DRV_CANCELED)
- return;
- INIT_SVARS(aux);
- if (check_cancel( svars ))
- return;
- if (sts == DRV_BOX_BAD) {
- error( "Error: channel %s: %s %s cannot be opened.\n",
- svars->chan->name, str_ms[t], svars->orig_name[t] );
- svars->ret = SYNC_FAIL;
- sync_bail( svars );
- } else {
- svars->newuidval[t] = uidvalidity;
- box_opened2( svars, t );
- }
- }
- static void
- box_opened2( sync_vars_t *svars, int t )
- {
- store_t *ctx[2];
- channel_conf_t *chan;
- sync_rec_t *srec;
- uint_array_alloc_t mexcs;
- uint opts[2], fails, minwuid;
- svars->state[t] |= ST_SELECTED;
- if (!(svars->state[1-t] & ST_SELECTED))
- return;
- ctx[0] = svars->ctx[0];
- ctx[1] = svars->ctx[1];
- chan = svars->chan;
- fails = 0;
- for (t = 0; t < 2; t++)
- if (svars->uidval[t] != UIDVAL_BAD && svars->uidval[t] != svars->newuidval[t])
- fails++;
- if (fails == 2) {
- error( "Error: channel %s: UIDVALIDITY of both master and slave changed\n"
- "(master got %u, expected %u; slave got %u, expected %u).\n",
- svars->chan->name,
- svars->newuidval[M], svars->uidval[M], svars->newuidval[S], svars->uidval[S] );
- bail:
- svars->ret = SYNC_FAIL;
- sync_bail( svars );
- return;
- }
- if (!lock_state( svars ))
- goto bail;
- if (!(svars->nfp = fopen( svars->nname, "w" ))) {
- sys_error( "Error: cannot create new sync state %s", svars->nname );
- goto bail;
- }
- if (!(svars->jfp = fopen( svars->jname, "a" ))) {
- sys_error( "Error: cannot create journal %s", svars->jname );
- fclose( svars->nfp );
- goto bail;
- }
- setlinebuf( svars->jfp );
- if (!svars->replayed)
- jFprintf( svars, JOURNAL_VERSION "\n" );
- opts[M] = opts[S] = 0;
- if (fails)
- opts[M] = opts[S] = OPEN_OLD|OPEN_OLD_IDS;
- for (t = 0; t < 2; t++) {
- if (chan->ops[t] & (OP_DELETE|OP_FLAGS)) {
- opts[t] |= OPEN_SETFLAGS;
- opts[1-t] |= OPEN_OLD;
- if (chan->ops[t] & OP_FLAGS)
- opts[1-t] |= OPEN_FLAGS;
- }
- if (chan->ops[t] & (OP_NEW|OP_RENEW)) {
- opts[t] |= OPEN_APPEND;
- if (chan->ops[t] & OP_RENEW)
- opts[1-t] |= OPEN_OLD;
- if (chan->ops[t] & OP_NEW)
- opts[1-t] |= OPEN_NEW;
- if (chan->ops[t] & OP_EXPUNGE)
- opts[1-t] |= OPEN_FLAGS;
- if (chan->stores[t]->max_size != UINT_MAX) {
- if (chan->ops[t] & OP_RENEW)
- opts[1-t] |= OPEN_FLAGS|OPEN_OLD_SIZE;
- if (chan->ops[t] & OP_NEW)
- opts[1-t] |= OPEN_FLAGS|OPEN_NEW_SIZE;
- }
- }
- if (chan->ops[t] & OP_EXPUNGE) {
- opts[t] |= OPEN_EXPUNGE;
- if (chan->stores[t]->trash) {
- if (!chan->stores[t]->trash_only_new)
- opts[t] |= OPEN_OLD;
- opts[t] |= OPEN_NEW|OPEN_FLAGS;
- } else if (chan->stores[1-t]->trash && chan->stores[1-t]->trash_remote_new)
- opts[t] |= OPEN_NEW|OPEN_FLAGS;
- }
- }
- if ((chan->ops[S] & (OP_NEW|OP_RENEW|OP_FLAGS)) && chan->max_messages)
- opts[S] |= OPEN_OLD|OPEN_NEW|OPEN_FLAGS;
- if (svars->replayed)
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (srec->tuid[0]) {
- if (!srec->uid[M])
- opts[M] |= OPEN_NEW|OPEN_FIND, svars->state[M] |= ST_FIND_OLD;
- else if (!srec->uid[S])
- opts[S] |= OPEN_NEW|OPEN_FIND, svars->state[S] |= ST_FIND_OLD;
- else
- warn( "Warning: sync record (%u,%u) has stray TUID. Ignoring.\n", srec->uid[M], srec->uid[S] );
- }
- }
- svars->opts[M] = svars->drv[M]->prepare_load_box( ctx[M], opts[M] );
- svars->opts[S] = svars->drv[S]->prepare_load_box( ctx[S], opts[S] );
- ARRAY_INIT( &mexcs );
- if (svars->opts[M] & OPEN_OLD) {
- if (chan->max_messages) {
- /* When messages have been expired on the slave, the master fetch is split into
- * two ranges: The bulk fetch which corresponds with the most recent messages, and an
- * exception list of messages which would have been expired if they weren't important. */
- debug( "preparing master selection - max expired master uid is %u\n", svars->mmaxxuid );
- /* First, find out the lower bound for the bulk fetch. */
- minwuid = svars->mmaxxuid + 1;
- /* Next, calculate the exception fetch. */
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (!srec->uid[M])
- continue; // No message; other state is irrelevant
- if (srec->uid[M] >= minwuid)
- continue; // Message is in non-expired range
- if ((svars->opts[M] & OPEN_NEW) && srec->uid[M] >= svars->maxuid[M])
- continue; // Message is in expired range, but new range overlaps that
- if (!srec->uid[S] && !(srec->status & S_PENDING))
- continue; // Only actually paired up messages matter
- // The pair is alive, but outside the bulk range
- *uint_array_append( &mexcs ) = srec->uid[M];
- }
- sort_uint_array( mexcs.array );
- } else {
- minwuid = 1;
- }
- } else {
- minwuid = UINT_MAX;
- }
- sync_ref( svars );
- load_box( svars, M, minwuid, mexcs.array );
- if (!check_cancel( svars ))
- load_box( svars, S, (svars->opts[S] & OPEN_OLD) ? 1 : UINT_MAX, (uint_array_t){ NULL, 0 } );
- sync_deref( svars );
- }
- static uint
- get_seenuid( sync_vars_t *svars, int t )
- {
- uint seenuid = 0;
- for (sync_rec_t *srec = svars->srecs; srec; srec = srec->next)
- if (!(srec->status & S_DEAD) && seenuid < srec->uid[t])
- seenuid = srec->uid[t];
- return seenuid;
- }
- static void box_loaded( int sts, message_t *msgs, int total_msgs, int recent_msgs, void *aux );
- static void
- load_box( sync_vars_t *svars, int t, uint minwuid, uint_array_t mexcs )
- {
- uint maxwuid, seenuid;
- if (svars->opts[t] & OPEN_NEW) {
- if (minwuid > svars->maxuid[t] + 1)
- minwuid = svars->maxuid[t] + 1;
- maxwuid = UINT_MAX;
- if (svars->opts[t] & (OPEN_OLD_IDS|OPEN_OLD_SIZE))
- seenuid = get_seenuid( svars, t );
- else
- seenuid = 0;
- } else if (svars->opts[t] & OPEN_OLD) {
- maxwuid = seenuid = get_seenuid( svars, t );
- } else
- maxwuid = seenuid = 0;
- if (seenuid < svars->maxuid[t]) {
- /* We cannot rely on the maxuid, as uni-directional syncing does not update it.
- * But if it is there, use it to avoid a possible gap in the fetched range. */
- seenuid = svars->maxuid[t];
- }
- info( "Loading %s...\n", str_ms[t] );
- svars->drv[t]->load_box( svars->ctx[t], minwuid, maxwuid, svars->newuid[t], seenuid, mexcs, box_loaded, AUX );
- }
- typedef struct {
- void *aux;
- sync_rec_t *srec;
- int aflags, dflags;
- } flag_vars_t;
- typedef struct {
- uint uid;
- sync_rec_t *srec;
- } sync_rec_map_t;
- static void flags_set( int sts, void *aux );
- static void flags_set_p2( sync_vars_t *svars, sync_rec_t *srec, int t );
- static void msgs_flags_set( sync_vars_t *svars, int t );
- static void msg_copied( int sts, uint uid, copy_vars_t *vars );
- static void msgs_copied( sync_vars_t *svars, int t );
- static void
- box_loaded( int sts, message_t *msgs, int total_msgs, int recent_msgs, void *aux )
- {
- DECL_SVARS;
- sync_rec_t *srec;
- sync_rec_map_t *srecmap;
- message_t *tmsg;
- flag_vars_t *fv;
- int no[2], del[2], alive, todel;
- uchar sflags, nflags, aflags, dflags;
- uint hashsz, idx;
- if (check_ret( sts, aux ))
- return;
- INIT_SVARS(aux);
- svars->state[t] |= ST_LOADED;
- svars->msgs[t] = msgs;
- info( "%s: %d messages, %d recent\n", str_ms[t], total_msgs, recent_msgs );
- if (svars->state[t] & ST_FIND_OLD) {
- debug( "matching previously copied messages on %s\n", str_ms[t] );
- match_tuids( svars, t, msgs );
- }
- debug( "matching messages on %s against sync records\n", str_ms[t] );
- hashsz = bucketsForSize( svars->nsrecs * 3 );
- srecmap = nfcalloc( hashsz * sizeof(*srecmap) );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- uint uid = srec->uid[t];
- if (!uid)
- continue;
- idx = (uint)(uid * 1103515245U) % hashsz;
- while (srecmap[idx].uid)
- if (++idx == hashsz)
- idx = 0;
- srecmap[idx].uid = uid;
- srecmap[idx].srec = srec;
- }
- for (tmsg = svars->msgs[t]; tmsg; tmsg = tmsg->next) {
- if (tmsg->srec) /* found by TUID */
- continue;
- uint uid = tmsg->uid;
- idx = (uint)(uid * 1103515245U) % hashsz;
- while (srecmap[idx].uid) {
- if (srecmap[idx].uid == uid) {
- srec = srecmap[idx].srec;
- goto found;
- }
- if (++idx == hashsz)
- idx = 0;
- }
- continue;
- found:
- tmsg->srec = srec;
- srec->msg[t] = tmsg;
- }
- free( srecmap );
- if (!(svars->state[1-t] & ST_LOADED))
- return;
- for (t = 0; t < 2; t++) {
- if (svars->uidval[t] != UIDVAL_BAD && svars->uidval[t] != svars->newuidval[t]) {
- unsigned need = 0, got = 0;
- debug( "trying to re-approve uid validity of %s\n", str_ms[t] );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (!srec->msg[t])
- continue; // Message disappeared.
- need++; // Present paired messages require re-validation.
- if (!srec->msg[t]->msgid)
- continue; // Messages without ID are useless for re-validation.
- if (!srec->msg[1-t])
- continue; // Partner disappeared.
- if (!srec->msg[1-t]->msgid || strcmp( srec->msg[M]->msgid, srec->msg[S]->msgid )) {
- error( "Error: channel %s, %s %s: UIDVALIDITY genuinely changed (at UID %u).\n",
- svars->chan->name, str_ms[t], svars->orig_name[t], srec->uid[t] );
- uvchg:
- svars->ret |= SYNC_FAIL;
- cancel_sync( svars );
- return;
- }
- got++;
- }
- if (got < 20 && got * 5 < need * 4) {
- // Too few confirmed messages. This is very likely in the drafts folder.
- // A proper fallback would be fetching more headers (which potentially need
- // normalization) or the message body (which should be truncated for sanity)
- // and comparing.
- error( "Error: channel %s, %s %s: Unable to recover from UIDVALIDITY change\n"
- "(got %u, expected %u).\n",
- svars->chan->name, str_ms[t], svars->orig_name[t],
- svars->newuidval[t], svars->uidval[t] );
- goto uvchg;
- }
- notice( "Notice: channel %s, %s %s: Recovered from change of UIDVALIDITY.\n",
- svars->chan->name, str_ms[t], svars->orig_name[t] );
- svars->uidval[t] = UIDVAL_BAD;
- }
- }
- if (svars->uidval[M] == UIDVAL_BAD || svars->uidval[S] == UIDVAL_BAD) {
- svars->uidval[M] = svars->newuidval[M];
- svars->uidval[S] = svars->newuidval[S];
- jFprintf( svars, "| %u %u\n", svars->uidval[M], svars->uidval[S] );
- }
- info( "Synchronizing...\n" );
- for (t = 0; t < 2; t++)
- svars->good_flags[t] = (uchar)svars->drv[t]->get_supported_flags( svars->ctx[t] );
- debug( "synchronizing old entries\n" );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- debug( "pair (%u,%u)\n", srec->uid[M], srec->uid[S] );
- assert( !srec->tuid[0] );
- // no[] means that a message is known to be not there.
- no[M] = !srec->msg[M] && (svars->opts[M] & OPEN_OLD);
- no[S] = !srec->msg[S] && (svars->opts[S] & OPEN_OLD);
- if (no[M] && no[S]) {
- // It does not matter whether one side was already known to be missing
- // (never stored [skipped or failed] or expunged [possibly expired]) -
- // now both are missing, so the entry is superfluous.
- debug( " vanished\n" );
- srec->status = S_DEAD;
- jFprintf( svars, "- %u %u\n", srec->uid[M], srec->uid[S] );
- } else {
- // del[] means that a message becomes known to have been expunged.
- del[M] = no[M] && srec->uid[M];
- del[S] = no[S] && srec->uid[S];
- for (t = 0; t < 2; t++) {
- if (srec->msg[t] && (srec->msg[t]->flags & F_DELETED))
- srec->wstate |= W_DEL(t);
- if (del[t]) {
- // The target was newly expunged, so there is nothing to update.
- // The deletion is propagated in the opposite iteration.
- } else if (!srec->uid[t]) {
- // The target was never stored, or was previously expunged, so there
- // is nothing to update.
- // Note: the opposite UID must be valid, as otherwise the entry would
- // have been pruned already.
- } else if (del[1-t]) {
- // The source was newly expunged, so possibly propagate the deletion.
- // The target may be in an unknown state (not fetched).
- if ((t == M) && (srec->status & (S_EXPIRE|S_EXPIRED))) {
- /* Don't propagate deletion resulting from expiration. */
- debug( " slave expired, orphaning master\n" );
- jFprintf( svars, "> %u %u 0\n", srec->uid[M], srec->uid[S] );
- srec->uid[S] = 0;
- } else {
- if (srec->msg[t] && (srec->msg[t]->status & M_FLAGS) && srec->msg[t]->flags != srec->flags)
- notice( "Notice: conflicting changes in (%u,%u)\n", srec->uid[M], srec->uid[S] );
- if (svars->chan->ops[t] & OP_DELETE) {
- debug( " %sing delete\n", str_hl[t] );
- srec->aflags[t] = F_DELETED;
- srec->wstate |= W_DELETE;
- } else {
- debug( " not %sing delete\n", str_hl[t] );
- }
- }
- } else if (!srec->msg[1-t]) {
- // We have no source to work with, because it was never stored,
- // it was previously expunged, or we did not fetch it.
- debug( " no %s\n", str_ms[1-t] );
- } else {
- // We have a source. The target may be in an unknown state.
- if (svars->chan->ops[t] & OP_FLAGS) {
- sflags = sanitize_flags( srec->msg[1-t]->flags, svars, t );
- if ((t == M) && (srec->status & (S_EXPIRE|S_EXPIRED))) {
- /* Don't propagate deletion resulting from expiration. */
- debug( " slave expiring\n" );
- sflags &= ~F_DELETED;
- }
- srec->aflags[t] = sflags & ~srec->flags;
- srec->dflags[t] = ~sflags & srec->flags;
- if ((DFlags & DEBUG_SYNC) && (srec->aflags[t] || srec->dflags[t])) {
- char afbuf[16], dfbuf[16]; /* enlarge when support for keywords is added */
- make_flags( srec->aflags[t], afbuf );
- make_flags( srec->dflags[t], dfbuf );
- debug( " %sing flags: +%s -%s\n", str_hl[t], afbuf, dfbuf );
- }
- }
- }
- }
- }
- }
- debug( "synchronizing new entries\n" );
- for (t = 0; t < 2; t++) {
- for (tmsg = svars->msgs[1-t]; tmsg; tmsg = tmsg->next) {
- // If new have no srec, the message is always New. If we have a srec:
- // - message is paired or expired => ignore
- // - message was skipped => ReNew
- // - message was attempted, but is still pending or failed => New
- //
- // If messages were previously ignored due to being excessive, they would now
- // appear to be newer than the messages that got actually synced, so increment
- // newmaxuid immediately to make sure we always look only at the newest ones.
- // However, committing it to maxuid must be delayed until all messages were
- // propagated, to ensure that all pending messages are still loaded next time
- // in case of interruption - in particular skipping big messages would otherwise
- // up the limit too early.
- srec = tmsg->srec;
- if (srec ? !srec->uid[t] &&
- (((srec->status & S_PENDING) && (svars->chan->ops[t] & OP_NEW)) ||
- ((srec->status & S_SKIPPED) && (svars->chan->ops[t] & OP_RENEW)))
- : svars->newmaxuid[1-t] < tmsg->uid && (svars->chan->ops[t] & OP_NEW)) {
- debug( "new message %u on %s\n", tmsg->uid, str_ms[1-t] );
- if ((svars->chan->ops[t] & OP_EXPUNGE) && (tmsg->flags & F_DELETED)) {
- debug( " -> not %sing - would be expunged anyway\n", str_hl[t] );
- } else {
- if (srec) {
- debug( " -> pair(%u,%u) exists\n", srec->uid[M], srec->uid[S] );
- } else {
- srec = nfcalloc( sizeof(*srec) );
- *svars->srecadd = srec;
- svars->srecadd = &srec->next;
- svars->nsrecs++;
- srec->status = S_PENDING;
- srec->uid[1-t] = tmsg->uid;
- srec->msg[1-t] = tmsg;
- tmsg->srec = srec;
- if (svars->newmaxuid[1-t] < tmsg->uid)
- svars->newmaxuid[1-t] = tmsg->uid;
- jFprintf( svars, "+ %u %u\n", srec->uid[M], srec->uid[S] );
- debug( " -> pair(%u,%u) created\n", srec->uid[M], srec->uid[S] );
- }
- if ((tmsg->flags & F_FLAGGED) || tmsg->size <= svars->chan->stores[t]->max_size) {
- if (srec->status != S_PENDING) {
- debug( " -> not too big any more\n" );
- srec->status = S_PENDING;
- jFprintf( svars, "~ %u %u %u\n", srec->uid[M], srec->uid[S], srec->status );
- }
- } else {
- if (srec->status == S_SKIPPED) {
- debug( " -> not %sing - still too big\n", str_hl[t] );
- } else {
- debug( " -> not %sing - too big\n", str_hl[t] );
- srec->status = S_SKIPPED;
- jFprintf( svars, "~ %u %u %u\n", srec->uid[M], srec->uid[S], srec->status );
- }
- }
- }
- }
- }
- }
- if ((svars->chan->ops[S] & (OP_NEW|OP_RENEW|OP_FLAGS)) && svars->chan->max_messages) {
- /* Note: When this branch is entered, we have loaded all slave messages. */
- /* Expire excess messages. Important (flagged, unread, or unpropagated) messages
- * older than the first not expired message are not counted towards the total. */
- debug( "preparing message expiration\n" );
- alive = 0;
- for (tmsg = svars->msgs[S]; tmsg; tmsg = tmsg->next) {
- if (tmsg->status & M_DEAD)
- continue;
- if ((srec = tmsg->srec) && srec->uid[M] &&
- ((tmsg->flags | srec->aflags[S]) & ~srec->dflags[S] & F_DELETED) &&
- !(srec->status & (S_EXPIRE|S_EXPIRED))) {
- /* Message was not propagated yet, or is deleted. */
- } else {
- alive++;
- }
- }
- for (tmsg = svars->msgs[M]; tmsg; tmsg = tmsg->next) {
- if ((srec = tmsg->srec) && (srec->status & S_PENDING) && !(tmsg->flags & F_DELETED))
- alive++;
- }
- todel = alive - svars->chan->max_messages;
- debug( "%d alive messages, %d excess - expiring\n", alive, todel );
- alive = 0;
- for (tmsg = svars->msgs[S]; tmsg; tmsg = tmsg->next) {
- if (tmsg->status & M_DEAD)
- continue;
- if (!(srec = tmsg->srec) || !srec->uid[M]) {
- /* We did not push the message, so it must be kept. */
- debug( " message %u unpropagated\n", tmsg->uid );
- todel--;
- } else {
- nflags = (tmsg->flags | srec->aflags[S]) & ~srec->dflags[S];
- if (!(nflags & F_DELETED) || (srec->status & (S_EXPIRE|S_EXPIRED))) {
- /* The message is not deleted, or is already (being) expired. */
- if ((nflags & F_FLAGGED) || !((nflags & F_SEEN) || ((void)(todel > 0 && alive++), svars->chan->expire_unread > 0))) {
- /* Important messages are always kept. */
- debug( " old pair(%u,%u) important\n", srec->uid[M], srec->uid[S] );
- todel--;
- } else if (todel > 0 ||
- ((srec->status & (S_EXPIRE|S_EXPIRED)) == (S_EXPIRE|S_EXPIRED)) ||
- ((srec->status & (S_EXPIRE|S_EXPIRED)) && (tmsg->flags & F_DELETED))) {
- /* The message is excess or was already (being) expired. */
- srec->wstate |= W_NEXPIRE;
- debug( " old pair(%u,%u) expired\n", srec->uid[M], srec->uid[S] );
- if (svars->mmaxxuid < srec->uid[M])
- svars->mmaxxuid = srec->uid[M];
- todel--;
- }
- }
- }
- }
- for (tmsg = svars->msgs[M]; tmsg; tmsg = tmsg->next) {
- if ((srec = tmsg->srec) && (srec->status & S_PENDING)) {
- nflags = tmsg->flags;
- if (!(nflags & F_DELETED)) {
- if ((nflags & F_FLAGGED) || !((nflags & F_SEEN) || ((void)(todel > 0 && alive++), svars->chan->expire_unread > 0))) {
- /* Important messages are always fetched. */
- debug( " new pair(%u,%u) important\n", srec->uid[M], srec->uid[S] );
- todel--;
- } else if (todel > 0) {
- /* The message is excess. */
- srec->wstate |= W_NEXPIRE;
- todel--;
- }
- }
- }
- }
- debug( "%d excess messages remain\n", todel );
- if (svars->chan->expire_unread < 0 && alive * 2 > svars->chan->max_messages) {
- error( "%s: %d unread messages in excess of MaxMessages (%d).\n"
- "Please set ExpireUnread to decide outcome. Skipping mailbox.\n",
- svars->orig_name[S], alive, svars->chan->max_messages );
- svars->ret |= SYNC_FAIL;
- cancel_sync( svars );
- return;
- }
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (!(srec->status & S_PENDING)) {
- if (!srec->msg[S])
- continue;
- uchar nex = (srec->wstate / W_NEXPIRE) & 1;
- if (nex != ((srec->status / S_EXPIRED) & 1)) {
- /* The record needs a state change ... */
- if (nex != ((srec->status / S_EXPIRE) & 1)) {
- /* ... and we need to start a transaction. */
- debug( " pair(%u,%u): %u (pre)\n", srec->uid[M], srec->uid[S], nex );
- srec->status = (srec->status & ~S_EXPIRE) | (nex * S_EXPIRE);
- jFprintf( svars, "~ %u %u %u\n", srec->uid[M], srec->uid[S], srec->status );
- } else {
- /* ... but the "right" transaction is already pending. */
- debug( " pair(%u,%u): %u (pending)\n", srec->uid[M], srec->uid[S], nex );
- }
- } else {
- /* Note: the "wrong" transaction may be pending here,
- * e.g.: W_NEXPIRE = 0, S_EXPIRE = 1, S_EXPIRED = 0. */
- }
- } else {
- if (srec->wstate & W_NEXPIRE) {
- jFprintf( svars, "= %u %u\n", srec->uid[M], srec->uid[S] );
- debug( " pair(%u,%u): 1 (abort)\n", srec->uid[M], srec->uid[S] );
- // If we have so many new messages that some of them are instantly expired,
- // but some are still propagated because they are important, we need to
- // ensure explicitly that the bulk fetch limit is upped.
- svars->mmaxxuid = srec->uid[M];
- srec->msg[M]->srec = NULL;
- srec->status = S_DEAD;
- }
- }
- }
- }
- sync_ref( svars );
- debug( "synchronizing flags\n" );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if ((srec->status & S_DEAD) || !srec->uid[M] || !srec->uid[S])
- continue;
- for (t = 0; t < 2; t++) {
- aflags = srec->aflags[t];
- dflags = srec->dflags[t];
- if (srec->wstate & W_DELETE) {
- if (!aflags) {
- /* This deletion propagation goes the other way round. */
- continue;
- }
- } else {
- /* The trigger is an expiration transaction being ongoing ... */
- if ((t == S) && ((shifted_bit(srec->status, S_EXPIRE, S_EXPIRED) ^ srec->status) & S_EXPIRED)) {
- /* ... but the actual action derives from the wanted state. */
- if (srec->wstate & W_NEXPIRE)
- aflags |= F_DELETED;
- else
- dflags |= F_DELETED;
- }
- }
- if ((svars->chan->ops[t] & OP_EXPUNGE) && (((srec->msg[t] ? srec->msg[t]->flags : 0) | aflags) & ~dflags & F_DELETED) &&
- (!svars->ctx[t]->conf->trash || svars->ctx[t]->conf->trash_only_new))
- {
- /* If the message is going to be expunged, don't propagate anything but the deletion. */
- srec->aflags[t] &= F_DELETED;
- aflags &= F_DELETED;
- srec->dflags[t] = dflags = 0;
- }
- if (srec->msg[t] && (srec->msg[t]->status & M_FLAGS)) {
- /* If we know the target message's state, optimize away non-changes. */
- aflags &= ~srec->msg[t]->flags;
- dflags &= srec->msg[t]->flags;
- }
- if (aflags | dflags) {
- flags_total[t]++;
- stats();
- svars->flags_pending[t]++;
- fv = nfmalloc( sizeof(*fv) );
- fv->aux = AUX;
- fv->srec = srec;
- fv->aflags = aflags;
- fv->dflags = dflags;
- svars->drv[t]->set_msg_flags( svars->ctx[t], srec->msg[t], srec->uid[t], aflags, dflags, flags_set, fv );
- if (check_cancel( svars ))
- goto out;
- } else
- flags_set_p2( svars, srec, t );
- }
- }
- for (t = 0; t < 2; t++) {
- svars->drv[t]->commit_cmds( svars->ctx[t] );
- svars->state[t] |= ST_SENT_FLAGS;
- msgs_flags_set( svars, t );
- if (check_cancel( svars ))
- goto out;
- }
- debug( "propagating new messages\n" );
- if (UseFSync)
- fdatasync( fileno( svars->jfp ) );
- for (t = 0; t < 2; t++) {
- svars->newuid[t] = svars->drv[t]->get_uidnext( svars->ctx[t] );
- jFprintf( svars, "F %d %u\n", t, svars->newuid[t] );
- svars->new_msgs[t] = svars->msgs[1-t];
- msgs_copied( svars, t );
- if (check_cancel( svars ))
- goto out;
- }
- out:
- sync_deref( svars );
- }
- static void
- msg_copied( int sts, uint uid, copy_vars_t *vars )
- {
- SVARS_CHECK_CANCEL_RET;
- sync_rec_t *srec = vars->srec;
- switch (sts) {
- case SYNC_OK:
- if (vars->msg->flags != srec->flags) {
- srec->flags = vars->msg->flags;
- debug( " -> updated flags to %u\n", srec->flags );
- jFprintf( svars, "* %u %u %u\n", srec->uid[M], srec->uid[S], srec->flags );
- }
- if (!uid) { // Stored to a non-UIDPLUS mailbox
- svars->state[t] |= ST_FIND_NEW;
- } else {
- debug( " -> new UID %u on %s\n", uid, str_ms[t] );
- jFprintf( svars, "%c %u %u %u\n", "<>"[t], srec->uid[M], srec->uid[S], uid );
- vars->srec->uid[t] = uid;
- vars->srec->status &= ~S_PENDING;
- vars->srec->tuid[0] = 0;
- }
- break;
- case SYNC_NOGOOD:
- debug( " -> killing (%u,%u)\n", srec->uid[M], srec->uid[S] );
- srec->status = S_DEAD;
- jFprintf( svars, "- %u %u\n", srec->uid[M], srec->uid[S] );
- break;
- default:
- cancel_sync( svars );
- free( vars );
- return;
- }
- free( vars );
- new_done[t]++;
- stats();
- svars->new_pending[t]--;
- msgs_copied( svars, t );
- }
- static void msgs_found_new( int sts, message_t *msgs, void *aux );
- static void msgs_new_done( sync_vars_t *svars, int t );
- static void sync_close( sync_vars_t *svars, int t );
- static void
- msgs_copied( sync_vars_t *svars, int t )
- {
- message_t *tmsg;
- sync_rec_t *srec;
- copy_vars_t *cv;
- if (svars->state[t] & ST_SENDING_NEW)
- return;
- sync_ref( svars );
- if (!(svars->state[t] & ST_SENT_NEW)) {
- for (tmsg = svars->new_msgs[t]; tmsg; tmsg = tmsg->next) {
- if ((srec = tmsg->srec) && (srec->status & S_PENDING)) {
- if (svars->drv[t]->get_memory_usage( svars->ctx[t] ) >= BufferLimit) {
- svars->new_msgs[t] = tmsg;
- goto out;
- }
- for (uint i = 0; i < TUIDL; i++) {
- uchar c = arc4_getbyte() & 0x3f;
- srec->tuid[i] = (char)(c < 26 ? c + 'A' : c < 52 ? c + 'a' - 26 : c < 62 ? c + '0' - 52 : c == 62 ? '+' : '/');
- }
- jFprintf( svars, "# %u %u %." stringify(TUIDL) "s\n", srec->uid[M], srec->uid[S], srec->tuid );
- debug( "%sing message %u, TUID %." stringify(TUIDL) "s\n", str_hl[t], tmsg->uid, srec->tuid );
- new_total[t]++;
- stats();
- svars->new_pending[t]++;
- svars->state[t] |= ST_SENDING_NEW;
- cv = nfmalloc( sizeof(*cv) );
- cv->cb = msg_copied;
- cv->aux = AUX;
- cv->srec = srec;
- cv->msg = tmsg;
- copy_msg( cv );
- svars->state[t] &= ~ST_SENDING_NEW;
- if (check_cancel( svars ))
- goto out;
- }
- }
- svars->state[t] |= ST_SENT_NEW;
- }
- if (svars->new_pending[t])
- goto out;
- if (svars->maxuid[1-t] != svars->newmaxuid[1-t]) {
- svars->maxuid[1-t] = svars->newmaxuid[1-t];
- jFprintf( svars, "S %d\n", 1-t );
- }
- sync_close( svars, 1-t );
- if (check_cancel( svars ))
- goto out;
- if (svars->state[t] & ST_FIND_NEW) {
- debug( "finding just copied messages on %s\n", str_ms[t] );
- svars->drv[t]->find_new_msgs( svars->ctx[t], svars->newuid[t], msgs_found_new, AUX );
- } else {
- msgs_new_done( svars, t );
- }
- out:
- sync_deref( svars );
- }
- static void
- msgs_found_new( int sts, message_t *msgs, void *aux )
- {
- SVARS_CHECK_RET;
- switch (sts) {
- case DRV_OK:
- debug( "matching just copied messages on %s\n", str_ms[t] );
- break;
- default:
- warn( "Warning: cannot find newly stored messages on %s.\n", str_ms[t] );
- break;
- }
- match_tuids( svars, t, msgs );
- msgs_new_done( svars, t );
- }
- static void
- msgs_new_done( sync_vars_t *svars, int t )
- {
- svars->state[t] |= ST_FOUND_NEW;
- sync_close( svars, t );
- }
- static void
- flags_set( int sts, void *aux )
- {
- SVARS_CHECK_RET_VARS(flag_vars_t);
- sync_rec_t *srec = vars->srec;
- switch (sts) {
- case DRV_OK:
- if (vars->aflags & F_DELETED)
- srec->wstate |= W_DEL(t);
- else if (vars->dflags & F_DELETED)
- srec->wstate &= ~W_DEL(t);
- flags_set_p2( svars, srec, t );
- break;
- }
- free( vars );
- flags_done[t]++;
- stats();
- svars->flags_pending[t]--;
- msgs_flags_set( svars, t );
- }
- static void
- flags_set_p2( sync_vars_t *svars, sync_rec_t *srec, int t )
- {
- if (srec->wstate & W_DELETE) {
- debug( " pair(%u,%u): resetting %s UID\n", srec->uid[M], srec->uid[S], str_ms[1-t] );
- jFprintf( svars, "%c %u %u 0\n", "><"[t], srec->uid[M], srec->uid[S] );
- srec->uid[1-t] = 0;
- } else {
- uchar nflags = (srec->flags | srec->aflags[t]) & ~srec->dflags[t];
- if (srec->flags != nflags) {
- debug( " pair(%u,%u): updating flags (%u -> %u; %sed)\n", srec->uid[M], srec->uid[S], srec->flags, nflags, str_hl[t] );
- srec->flags = nflags;
- jFprintf( svars, "* %u %u %u\n", srec->uid[M], srec->uid[S], nflags );
- }
- if (t == S) {
- uchar nex = (srec->wstate / W_NEXPIRE) & 1;
- if (nex != ((srec->status / S_EXPIRED) & 1)) {
- debug( " pair(%u,%u): expired %d (commit)\n", srec->uid[M], srec->uid[S], nex );
- srec->status = (srec->status & ~S_EXPIRED) | (nex * S_EXPIRED);
- jFprintf( svars, "~ %u %u %u\n", srec->uid[M], srec->uid[S], srec->status );
- } else if (nex != ((srec->status / S_EXPIRE) & 1)) {
- debug( " pair(%u,%u): expire %d (cancel)\n", srec->uid[M], srec->uid[S], nex );
- srec->status = (srec->status & ~S_EXPIRE) | (nex * S_EXPIRE);
- jFprintf( svars, "~ %u %u %u\n", srec->uid[M], srec->uid[S], srec->status );
- }
- }
- }
- }
- typedef struct {
- void *aux;
- message_t *msg;
- } trash_vars_t;
- static void msg_trashed( int sts, void *aux );
- static void msg_rtrashed( int sts, uint uid, copy_vars_t *vars );
- static void
- msgs_flags_set( sync_vars_t *svars, int t )
- {
- message_t *tmsg;
- trash_vars_t *tv;
- copy_vars_t *cv;
- if (!(svars->state[t] & ST_SENT_FLAGS) || svars->flags_pending[t])
- return;
- sync_ref( svars );
- if ((svars->chan->ops[t] & OP_EXPUNGE) &&
- (svars->ctx[t]->conf->trash || (svars->ctx[1-t]->conf->trash && svars->ctx[1-t]->conf->trash_remote_new))) {
- debug( "trashing in %s\n", str_ms[t] );
- for (tmsg = svars->msgs[t]; tmsg; tmsg = tmsg->next)
- if ((tmsg->flags & F_DELETED) && !find_uint_array( svars->trashed_msgs[t].array, tmsg->uid ) &&
- (t == M || !tmsg->srec || !(tmsg->srec->status & (S_EXPIRE|S_EXPIRED)))) {
- if (svars->ctx[t]->conf->trash) {
- if (!svars->ctx[t]->conf->trash_only_new || !tmsg->srec || (tmsg->srec->status & (S_PENDING | S_SKIPPED))) {
- debug( "%s: trashing message %u\n", str_ms[t], tmsg->uid );
- trash_total[t]++;
- stats();
- svars->trash_pending[t]++;
- tv = nfmalloc( sizeof(*tv) );
- tv->aux = AUX;
- tv->msg = tmsg;
- svars->drv[t]->trash_msg( svars->ctx[t], tmsg, msg_trashed, tv );
- if (check_cancel( svars ))
- goto out;
- } else
- debug( "%s: not trashing message %u - not new\n", str_ms[t], tmsg->uid );
- } else {
- if (!tmsg->srec || (tmsg->srec->status & (S_PENDING | S_SKIPPED))) {
- if (tmsg->size <= svars->ctx[1-t]->conf->max_size) {
- debug( "%s: remote trashing message %u\n", str_ms[t], tmsg->uid );
- trash_total[t]++;
- stats();
- svars->trash_pending[t]++;
- cv = nfmalloc( sizeof(*cv) );
- cv->cb = msg_rtrashed;
- cv->aux = INV_AUX;
- cv->srec = NULL;
- cv->msg = tmsg;
- copy_msg( cv );
- if (check_cancel( svars ))
- goto out;
- } else
- debug( "%s: not remote trashing message %u - too big\n", str_ms[t], tmsg->uid );
- } else
- debug( "%s: not remote trashing message %u - not new\n", str_ms[t], tmsg->uid );
- }
- }
- }
- svars->state[t] |= ST_SENT_TRASH;
- sync_close( svars, t );
- out:
- sync_deref( svars );
- }
- static void
- msg_trashed( int sts, void *aux )
- {
- trash_vars_t *vars = (trash_vars_t *)aux;
- DECL_SVARS;
- if (sts == DRV_MSG_BAD)
- sts = DRV_BOX_BAD;
- if (check_ret( sts, vars->aux ))
- return;
- INIT_SVARS(vars->aux);
- debug( " -> trashed %s %u\n", str_ms[t], vars->msg->uid );
- jFprintf( svars, "T %d %u\n", t, vars->msg->uid );
- free( vars );
- trash_done[t]++;
- stats();
- svars->trash_pending[t]--;
- sync_close( svars, t );
- }
- static void
- msg_rtrashed( int sts, uint uid ATTR_UNUSED, copy_vars_t *vars )
- {
- SVARS_CHECK_CANCEL_RET;
- switch (sts) {
- case SYNC_OK:
- case SYNC_NOGOOD: /* the message is gone or heavily busted */
- break;
- default:
- cancel_sync( svars );
- free( vars );
- return;
- }
- t ^= 1;
- debug( " -> remote trashed %s %u\n", str_ms[t], vars->msg->uid );
- jFprintf( svars, "T %d %u\n", t, vars->msg->uid );
- free( vars );
- trash_done[t]++;
- stats();
- svars->trash_pending[t]--;
- sync_close( svars, t );
- }
- static void box_closed( int sts, void *aux );
- static void box_closed_p2( sync_vars_t *svars, int t );
- static void
- sync_close( sync_vars_t *svars, int t )
- {
- if ((~svars->state[t] & (ST_FOUND_NEW|ST_SENT_TRASH)) || svars->trash_pending[t] ||
- !(svars->state[1-t] & ST_SENT_NEW) || svars->new_pending[1-t])
- return;
- if (svars->state[t] & ST_CLOSING)
- return;
- svars->state[t] |= ST_CLOSING;
- if ((svars->chan->ops[t] & OP_EXPUNGE) /*&& !(svars->state[t] & ST_TRASH_BAD)*/) {
- debug( "expunging %s\n", str_ms[t] );
- svars->drv[t]->close_box( svars->ctx[t], box_closed, AUX );
- } else {
- box_closed_p2( svars, t );
- }
- }
- static void
- box_closed( int sts, void *aux )
- {
- SVARS_CHECK_RET;
- svars->state[t] |= ST_DID_EXPUNGE;
- box_closed_p2( svars, t );
- }
- static void
- box_closed_p2( sync_vars_t *svars, int t )
- {
- sync_rec_t *srec;
- svars->state[t] |= ST_CLOSED;
- if (!(svars->state[1-t] & ST_CLOSED))
- return;
- // All the journalling done in this function is merely for the autotest -
- // the operations are idempotent, and we're about to commit the new state
- // right afterwards anyway.
- if (((svars->state[M] | svars->state[S]) & ST_DID_EXPUNGE) || svars->chan->max_messages) {
- debug( "purging obsolete entries\n" );
- for (srec = svars->srecs; srec; srec = srec->next) {
- if (srec->status & S_DEAD)
- continue;
- if (!srec->uid[S] || ((srec->wstate & W_DEL(S)) && (svars->state[S] & ST_DID_EXPUNGE))) {
- if (!srec->uid[M] || ((srec->wstate & W_DEL(M)) && (svars->state[M] & ST_DID_EXPUNGE)) ||
- ((srec->status & S_EXPIRED) && svars->maxuid[M] >= srec->uid[M] && svars->mmaxxuid >= srec->uid[M])) {
- debug( " -> killing (%u,%u)\n", srec->uid[M], srec->uid[S] );
- jFprintf( svars, "- %u %u\n", srec->uid[M], srec->uid[S] );
- srec->status = S_DEAD;
- } else if (srec->uid[S]) {
- debug( " -> orphaning (%u,[%u])\n", srec->uid[M], srec->uid[S] );
- jFprintf( svars, "> %u %u 0\n", srec->uid[M], srec->uid[S] );
- srec->uid[S] = 0;
- }
- } else if (srec->uid[M] && ((srec->wstate & W_DEL(M)) && (svars->state[M] & ST_DID_EXPUNGE))) {
- debug( " -> orphaning ([%u],%u)\n", srec->uid[M], srec->uid[S] );
- jFprintf( svars, "< %u %u 0\n", srec->uid[M], srec->uid[S] );
- srec->uid[M] = 0;
- }
- }
- }
- // This is just an optimization, so it needs no journaling of intermediate states.
- // However, doing it before the entry purge would require ensuring that the
- // exception list includes all relevant messages.
- debug( "max expired uid on master is now %u\n", svars->mmaxxuid );
- jFprintf( svars, "! %u\n", svars->mmaxxuid );
- save_state( svars );
- sync_bail( svars );
- }
- static void
- sync_bail( sync_vars_t *svars )
- {
- sync_rec_t *srec, *nsrec;
- free( svars->trashed_msgs[M].array.data );
- free( svars->trashed_msgs[S].array.data );
- for (srec = svars->srecs; srec; srec = nsrec) {
- nsrec = srec->next;
- free( srec );
- }
- if (svars->lfd >= 0) {
- unlink( svars->lname );
- close( svars->lfd );
- }
- sync_bail2( svars );
- }
- static void
- sync_bail2( sync_vars_t *svars )
- {
- free( svars->lname );
- free( svars->nname );
- free( svars->jname );
- free( svars->dname );
- sync_bail3( svars );
- }
- static void
- sync_bail3( sync_vars_t *svars )
- {
- free( svars->box_name[M] );
- free( svars->box_name[S] );
- sync_deref( svars );
- }
- static void
- sync_deref( sync_vars_t *svars )
- {
- if (!--svars->ref_count) {
- void (*cb)( int sts, void *aux ) = svars->cb;
- void *aux = svars->aux;
- int ret = svars->ret;
- free( svars );
- cb( ret, aux );
- }
- }
|