ref: 28f4567ba615c4036c079021392825757ef2a008
dir: /sys/src/cmd/ip/imap4d/copy.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <auth.h> #include <libsec.h> #include "imap4d.h" static int saveMsg(char *dst, char *digest, int flags, char *head, int nhead, Biobuf *b, long n); static int saveb(int fd, DigestState *dstate, char *buf, int nr, int nw); static long appSpool(Biobuf *bout, Biobuf *bin, long n); /* * check if the message exists */ int copyCheck(Box *box, Msg *m, int uids, void *v) { int fd; USED(box); USED(uids); USED(v); if(m->expunged) return 0; fd = msgFile(m, "raw"); if(fd < 0){ msgDead(m); return 0; } close(fd); return 1; } int copySave(Box *box, Msg *m, int uids, void *vs) { Dir *d; Biobuf b; vlong length; char *head; int ok, hfd, bfd, nhead; USED(box); USED(uids); if(m->expunged) return 0; hfd = msgFile(m, "unixheader"); if(hfd < 0){ msgDead(m); return 0; } head = readFile(hfd); if(head == nil){ close(hfd); return 0; } /* * clear out the header if it doesn't end in a newline, * since it is a runt and the "header" will show up in the raw file. */ nhead = strlen(head); if(nhead > 0 && head[nhead-1] != '\n') nhead = 0; bfd = msgFile(m, "raw"); close(hfd); if(bfd < 0){ msgDead(m); return 0; } d = dirfstat(bfd); if(d == nil){ close(bfd); return 0; } length = d->length; free(d); Binit(&b, bfd, OREAD); ok = saveMsg(vs, m->info[IDigest], m->flags, head, nhead, &b, length); Bterm(&b); close(bfd); return ok; } /* * first spool the input into a temorary file, * and massage the input in the process. * then save to real box. */ int appendSave(char *mbox, int flags, char *head, Biobuf *b, long n) { Biobuf btmp; int fd, ok; fd = imapTmp(); if(fd < 0) return 0; Bprint(&bout, "+ Ready for literal data\r\n"); if(Bflush(&bout) < 0) writeErr(); Binit(&btmp, fd, OWRITE); n = appSpool(&btmp, b, n); Bterm(&btmp); if(n < 0){ close(fd); return 0; } seek(fd, 0, 0); Binit(&btmp, fd, OREAD); ok = saveMsg(mbox, nil, flags, head, strlen(head), &btmp, n); Bterm(&btmp); close(fd); return ok; } /* * copy from bin to bout, * mapping "\r\n" to "\n" and "\nFrom " to "\n From " * return the number of bytes in the mapped file. * * exactly n bytes must be read from the input, * unless an input error occurs. */ static long appSpool(Biobuf *bout, Biobuf *bin, long n) { int i, c; c = '\n'; while(n > 0){ if(c == '\n' && n >= STRLEN("From ")){ for(i = 0; i < STRLEN("From "); i++){ c = Bgetc(bin); if(c != "From "[i]){ if(c < 0) return -1; Bungetc(bin); break; } n--; } if(i == STRLEN("From ")) Bputc(bout, ' '); Bwrite(bout, "From ", i); } c = Bgetc(bin); n--; if(c == '\r' && n-- > 0){ c = Bgetc(bin); if(c != '\n') Bputc(bout, '\r'); } if(c < 0) return -1; if(Bputc(bout, c) < 0) return -1; } if(c != '\n') Bputc(bout, '\n'); if(Bflush(bout) < 0) return -1; return Boffset(bout); } static int saveMsg(char *dst, char *digest, int flags, char *head, int nhead, Biobuf *b, long n) { DigestState *dstate; MbLock *ml; uchar shadig[SHA1dlen]; char buf[BufSize + 1], digbuf[NDigest + 1]; int i, fd, nr, nw, ok; ml = mbLock(); if(ml == nil) return 0; fd = openLocked(mboxDir, dst, OWRITE); if(fd < 0){ mbUnlock(ml); return 0; } seek(fd, 0, 2); dstate = nil; if(digest == nil) dstate = sha1(nil, 0, nil, nil); if(!saveb(fd, dstate, head, nhead, nhead)){ if(dstate != nil) sha1(nil, 0, shadig, dstate); mbUnlock(ml); close(fd); return 0; } ok = 1; if(n == 0) ok = saveb(fd, dstate, "\n", 0, 1); while(n > 0){ nr = n; if(nr > BufSize) nr = BufSize; nr = Bread(b, buf, nr); if(nr <= 0){ saveb(fd, dstate, "\n\n", 0, 2); ok = 0; break; } n -= nr; nw = nr; if(n == 0){ if(buf[nw - 1] != '\n') buf[nw++] = '\n'; buf[nw++] = '\n'; } if(!saveb(fd, dstate, buf, nr, nw)){ ok = 0; break; } mbLockRefresh(ml); } close(fd); if(dstate != nil){ digest = digbuf; sha1(nil, 0, shadig, dstate); for(i = 0; i < SHA1dlen; i++) snprint(digest+2*i, NDigest+1-2*i, "%2.2ux", shadig[i]); } if(ok){ fd = cdOpen(mboxDir, impName(dst), OWRITE); if(fd < 0) fd = emptyImp(dst); if(fd >= 0){ seek(fd, 0, 2); wrImpFlags(buf, flags, 1); fprint(fd, "%.*s %.*lud %s\n", NDigest, digest, NUid, 0UL, buf); close(fd); } } mbUnlock(ml); return 1; } static int saveb(int fd, DigestState *dstate, char *buf, int nr, int nw) { if(dstate != nil) sha1((uchar*)buf, nr, nil, dstate); if(write(fd, buf, nw) != nw) return 0; return 1; }