1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
#include "config.h"
#include "flatpak-zstd-decompressor-private.h"
#include <errno.h>
#include <string.h>
#ifdef HAVE_ZSTD
#include <zstd.h>
#endif
static void flatpak_zstd_decompressor_iface_init (GConverterIface *iface);
struct _FlatpakZstdDecompressor
{
GObject parent_instance;
#ifdef HAVE_ZSTD
ZSTD_DStream *dstream;
#endif
};
G_DEFINE_TYPE_WITH_CODE (FlatpakZstdDecompressor, flatpak_zstd_decompressor, G_TYPE_OBJECT,
G_IMPLEMENT_INTERFACE (G_TYPE_CONVERTER,
flatpak_zstd_decompressor_iface_init))
static void
flatpak_zstd_decompressor_finalize (GObject *object)
{
#ifdef HAVE_ZSTD
FlatpakZstdDecompressor *decompressor;
decompressor = FLATPAK_ZSTD_DECOMPRESSOR (object);
ZSTD_freeDStream (decompressor->dstream);
#endif
G_OBJECT_CLASS (flatpak_zstd_decompressor_parent_class)->finalize (object);
}
static void
flatpak_zstd_decompressor_init (FlatpakZstdDecompressor *decompressor)
{
#ifdef HAVE_ZSTD
decompressor->dstream = ZSTD_createDStream ();
#endif
}
static void
flatpak_zstd_decompressor_class_init (FlatpakZstdDecompressorClass *klass)
{
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
gobject_class->finalize = flatpak_zstd_decompressor_finalize;
}
FlatpakZstdDecompressor *
flatpak_zstd_decompressor_new (void)
{
FlatpakZstdDecompressor *decompressor;
decompressor = g_object_new (FLATPAK_TYPE_ZSTD_DECOMPRESSOR, NULL);
return decompressor;
}
static void
flatpak_zstd_decompressor_reset (GConverter *converter)
{
#ifdef HAVE_ZSTD
FlatpakZstdDecompressor *decompressor = FLATPAK_ZSTD_DECOMPRESSOR (converter);
ZSTD_initDStream (decompressor->dstream);
#endif
}
static GConverterResult
flatpak_zstd_decompressor_convert (GConverter *converter,
const void *inbuf,
gsize inbuf_size,
void *outbuf,
gsize outbuf_size,
GConverterFlags flags,
gsize *bytes_read,
gsize *bytes_written,
GError **error)
{
#ifdef HAVE_ZSTD
FlatpakZstdDecompressor *decompressor;
ZSTD_inBuffer input = { inbuf, inbuf_size, 0 };
ZSTD_outBuffer output = {outbuf, outbuf_size, 0 };
size_t res;
decompressor = FLATPAK_ZSTD_DECOMPRESSOR (converter);
if (decompressor->dstream == NULL)
{
g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED,
"Failed to initialize libzstd");
return G_CONVERTER_ERROR;
}
res = ZSTD_decompressStream(decompressor->dstream, &output , &input);
if (ZSTD_isError (res))
{
g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_DATA,
"Zstd decompression error: %s", ZSTD_getErrorName (res));
return G_CONVERTER_ERROR;
}
*bytes_read = input.pos;
*bytes_written = output.pos;
if (res == 0)
return G_CONVERTER_FINISHED;
if (input.pos == 0 && output.pos == 0)
{
/* Did nothing, need more input? */
if (flags & G_CONVERTER_INPUT_AT_END)
g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_INVALID_DATA, "Zstd failed");
else
g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PARTIAL_INPUT, "Need more zstd data");
return G_CONVERTER_ERROR;
}
return G_CONVERTER_CONVERTED;
#else
g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED,
"libzstd not available");
return G_CONVERTER_ERROR;
#endif
}
static void
flatpak_zstd_decompressor_iface_init (GConverterIface *iface)
{
iface->convert = flatpak_zstd_decompressor_convert;
iface->reset = flatpak_zstd_decompressor_reset;
}
|