From a4b9f643b801f82b64addf39ccf16283d633703d Mon Sep 17 00:00:00 2001 From: "C. McEnroe" Date: Fri, 17 Apr 2020 11:00:28 -0400 Subject: Add decodeToFile --- archive.h | 1 + decode.c | 18 +++++++++++++++--- 2 files changed, 16 insertions(+), 3 deletions(-) diff --git a/archive.h b/archive.h index a7377ab..d0bc00d 100644 --- a/archive.h +++ b/archive.h @@ -154,6 +154,7 @@ char *templateURL(const char *template, const struct Variable vars[]); char *decodeHeader(const char *header); char *decodeToString(const struct BodyPart *part, const char *content); +int decodeToFile(FILE *file, const struct BodyPart *part, const char *content); static inline const char *pathUID(uint32_t uid, const char *type) { static char buf[PATH_MAX]; diff --git a/decode.c b/decode.c index 24de9d3..35e6dfd 100644 --- a/decode.c +++ b/decode.c @@ -244,14 +244,26 @@ char *decodeHeader(const char *header) { return bufferString(&buf); } -char *decodeToString(const struct BodyPart *part, const char *src) { - struct Buffer dst = bufferAlloc(strlen(src) + 1); +static const char *partCharset(const struct BodyPart *part) { const char *charset = NULL; for (size_t i = 0; i + 1 < part->params.len; i += 2) { const char *key = dataCheck(part->params.ptr[i], String).string; if (strcasecmp(key, "charset")) continue; charset = dataCheck(part->params.ptr[i + 1], String).string; } - decode(&dst, part->encoding, charset, src); + return charset; +} + +char *decodeToString(const struct BodyPart *part, const char *src) { + struct Buffer dst = bufferAlloc(strlen(src) + 1); + decode(&dst, part->encoding, partCharset(part), src); return bufferString(&dst); } + +int decodeToFile(FILE *file, const struct BodyPart *part, const char *src) { + struct Buffer dst = bufferAlloc(strlen(src)); + decode(&dst, part->encoding, partCharset(part), src); + size_t n = fwrite(dst.ptr, dst.len, 1, file); + free(dst.ptr); + return (n ? 0 : -1); +} -- cgit 1.4.1