diff options
author | David Bremner <david@tethera.net> | 2019-03-30 10:03:01 -0300 |
---|---|---|
committer | David Bremner <david@tethera.net> | 2019-05-03 07:48:38 -0300 |
commit | 98b3eebc37eba8b86e888af5dc57dd28ca923e24 (patch) | |
tree | 54652595afa07e9797e8bd7c9a13719bb1fad193 /util | |
parent | 2f33afd159980361d3fcc390223b490b68d92530 (diff) |
util/gmime-extra: add g_mime_stream_gzfile_{new, open}
These are usable as standard GMime streams, and transparently
decompress gzipped files.
Diffstat (limited to 'util')
-rw-r--r-- | util/gmime-extra.c | 57 | ||||
-rw-r--r-- | util/gmime-extra.h | 8 |
2 files changed, 65 insertions, 0 deletions
diff --git a/util/gmime-extra.c b/util/gmime-extra.c index 4b577a5b..7562d906 100644 --- a/util/gmime-extra.c +++ b/util/gmime-extra.c @@ -1,6 +1,63 @@ #include "gmime-extra.h" #include <string.h> +static +GMimeStream * +_gzfile_maybe_filter (GMimeStream *file_stream) { + char buf[4]; + int bytes_read; + + if ((bytes_read = g_mime_stream_read (file_stream, buf, sizeof (buf))) < 0) + return NULL; + + if (g_mime_stream_reset (file_stream)) + return NULL; + + /* check for gzipped input */ + if (bytes_read >= 2 && buf[0] == 0x1f && (unsigned char)buf[1] == 0x8b) { + GMimeStream *gzstream; + GMimeFilter *gzfilter; + + gzfilter = g_mime_filter_gzip_new (GMIME_FILTER_GZIP_MODE_UNZIP, 0); + if (! gzfilter) + return NULL; + + gzstream = g_mime_stream_filter_new (file_stream); + if (! gzstream) + return NULL; + + /* ignore filter id */ + (void)g_mime_stream_filter_add ((GMimeStreamFilter *)gzstream, gzfilter); + return gzstream; + } else { + return file_stream; + } +} + +GMimeStream * +g_mime_stream_gzfile_new (int fd) +{ + GMimeStream *file_stream; + + file_stream = g_mime_stream_fs_new (fd); + if (! file_stream) + return NULL; + + return _gzfile_maybe_filter (file_stream); +} + +GMimeStream * +g_mime_stream_gzfile_open (const char *filename) +{ + GMimeStream *file_stream; + + file_stream = g_mime_stream_fs_open (filename, 0, 0, NULL); + if (! file_stream) + return NULL; + + return _gzfile_maybe_filter (file_stream); +} + GMimeStream * g_mime_stream_stdout_new() { diff --git a/util/gmime-extra.h b/util/gmime-extra.h index a42e43e7..b0c8d3d8 100644 --- a/util/gmime-extra.h +++ b/util/gmime-extra.h @@ -9,6 +9,14 @@ extern "C" { GMimeStream *g_mime_stream_stdout_new(void); +/* Return a GMime stream for this open file descriptor, un-gzipping if + * necessary */ +GMimeStream *g_mime_stream_gzfile_new (int fd); + +/* Return a GMime stream for this path, un-gzipping if + * necessary */ +GMimeStream *g_mime_stream_gzfile_open (const char *filename); + /** * Get last 16 hex digits of fingerprint ("keyid") */ |