factor out copy_msg_convert()
This commit is contained in:
parent
951b7e77f8
commit
3dffd68825
135
src/sync.c
135
src/sync.c
|
@ -331,15 +331,81 @@ copy_msg_bytes( char **out_ptr, const char *in_buf, int *in_idx, int in_len, int
|
|||
*in_idx = idx;
|
||||
}
|
||||
|
||||
static int
|
||||
copy_msg_convert( int in_cr, int out_cr, copy_vars_t *vars )
|
||||
{
|
||||
char *in_buf = vars->data.data;
|
||||
int in_len = vars->data.len;
|
||||
int idx = 0, sbreak = 0, ebreak = 0;
|
||||
int lines = 0, hdr_crs = 0, bdy_crs = 0, extra = 0;
|
||||
if (vars->srec) {
|
||||
nloop: ;
|
||||
int start = idx;
|
||||
int 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, 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:
|
||||
extra += 8 + TUIDL + 1 + (out_cr && (!in_cr || hdr_crs));
|
||||
}
|
||||
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 (out_cr && (!in_cr || hdr_crs))
|
||||
*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;
|
||||
char *fmap, *buf;
|
||||
int i, len, extra, scr, tcr, lcrs, hcrs, bcrs, lines;
|
||||
int start, sbreak = 0, ebreak = 0;
|
||||
char c;
|
||||
int scr, tcr;
|
||||
|
||||
switch (sts) {
|
||||
case DRV_OK:
|
||||
|
@ -355,71 +421,12 @@ msg_fetched( int sts, void *aux )
|
|||
scr = (svars->drv[1-t]->flags / DRV_CRLF) & 1;
|
||||
tcr = (svars->drv[t]->flags / DRV_CRLF) & 1;
|
||||
if (vars->srec || scr != tcr) {
|
||||
fmap = vars->data.data;
|
||||
len = vars->data.len;
|
||||
extra = lines = hcrs = bcrs = i = 0;
|
||||
if (vars->srec) {
|
||||
nloop:
|
||||
start = i;
|
||||
lcrs = 0;
|
||||
while (i < len) {
|
||||
c = fmap[i++];
|
||||
if (c == '\r')
|
||||
lcrs++;
|
||||
else if (c == '\n') {
|
||||
if (starts_with_upper( fmap + start, len - start, "X-TUID: ", 8 )) {
|
||||
extra = (sbreak = start) - (ebreak = i);
|
||||
goto oke;
|
||||
}
|
||||
lines++;
|
||||
hcrs += lcrs;
|
||||
if (i - lcrs - 1 == start) {
|
||||
sbreak = ebreak = start;
|
||||
goto oke;
|
||||
}
|
||||
goto nloop;
|
||||
}
|
||||
}
|
||||
/* invalid message */
|
||||
if (!copy_msg_convert( scr, tcr, vars )) {
|
||||
warn( "Warning: message %d from %s has incomplete header.\n",
|
||||
vars->msg->uid, str_ms[1-t] );
|
||||
free( fmap );
|
||||
vars->cb( SYNC_NOGOOD, 0, vars );
|
||||
return;
|
||||
oke:
|
||||
extra += 8 + TUIDL + 1 + (tcr && (!scr || hcrs));
|
||||
}
|
||||
if (tcr != scr) {
|
||||
for (; i < len; i++) {
|
||||
c = fmap[i];
|
||||
if (c == '\r')
|
||||
bcrs++;
|
||||
else if (c == '\n')
|
||||
lines++;
|
||||
}
|
||||
extra -= hcrs + bcrs;
|
||||
if (tcr)
|
||||
extra += lines;
|
||||
}
|
||||
|
||||
vars->data.len = len + extra;
|
||||
buf = vars->data.data = nfmalloc( vars->data.len );
|
||||
i = 0;
|
||||
if (vars->srec) {
|
||||
copy_msg_bytes( &buf, fmap, &i, sbreak, scr, tcr );
|
||||
|
||||
memcpy( buf, "X-TUID: ", 8 );
|
||||
buf += 8;
|
||||
memcpy( buf, vars->srec->tuid, TUIDL );
|
||||
buf += TUIDL;
|
||||
if (tcr && (!scr || hcrs))
|
||||
*buf++ = '\r';
|
||||
*buf++ = '\n';
|
||||
i = ebreak;
|
||||
}
|
||||
copy_msg_bytes( &buf, fmap, &i, len, scr, tcr );
|
||||
|
||||
free( fmap );
|
||||
}
|
||||
|
||||
svars->drv[t]->store_msg( svars->ctx[t], &vars->data, !vars->srec, msg_stored, vars );
|
||||
|
|
Loading…
Reference in New Issue
Block a user