]>
Commit | Line | Data |
---|---|---|
79197521 | 1 | /* compress-debug.c - compress debug sections |
fd67aa11 | 2 | Copyright (C) 2010-2024 Free Software Foundation, Inc. |
79197521 AM |
3 | |
4 | This file is part of GAS, the GNU Assembler. | |
5 | ||
6 | GAS is free software; you can redistribute it and/or modify | |
7 | it under the terms of the GNU General Public License as published by | |
8 | the Free Software Foundation; either version 3, or (at your option) | |
9 | any later version. | |
10 | ||
11 | GAS is distributed in the hope that it will be useful, | |
12 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | GNU General Public License for more details. | |
15 | ||
16 | You should have received a copy of the GNU General Public License | |
17 | along with GAS; see the file COPYING. If not, write to the Free | |
18 | Software Foundation, 51 Franklin Street - Fifth Floor, Boston, MA | |
19 | 02110-1301, USA. */ | |
20 | ||
7362f76f | 21 | #include "config.h" |
79197521 | 22 | #include <stdio.h> |
4c6f9528 | 23 | #include <string.h> |
0138187e | 24 | #include <zlib.h> |
2cac01e3 FS |
25 | #if HAVE_ZSTD |
26 | #include <zstd.h> | |
27 | #endif | |
79197521 | 28 | #include "ansidecl.h" |
79197521 AM |
29 | #include "compress-debug.h" |
30 | ||
79197521 AM |
31 | /* Initialize the compression engine. */ |
32 | ||
2cac01e3 FS |
33 | void * |
34 | compress_init (bool use_zstd) | |
79197521 | 35 | { |
2cac01e3 FS |
36 | if (use_zstd) { |
37 | #if HAVE_ZSTD | |
38 | return ZSTD_createCCtx (); | |
39 | #endif | |
40 | } | |
41 | ||
79197521 | 42 | static struct z_stream_s strm; |
4c6f9528 | 43 | memset (&strm, 0, sizeof (strm)); |
79197521 AM |
44 | deflateInit (&strm, Z_DEFAULT_COMPRESSION); |
45 | return &strm; | |
79197521 AM |
46 | } |
47 | ||
48 | /* Stream the contents of a frag to the compression engine. Output | |
49 | from the engine goes into the current frag on the obstack. */ | |
50 | ||
51 | int | |
2cac01e3 FS |
52 | compress_data (bool use_zstd, void *ctx, const char **next_in, int *avail_in, |
53 | char **next_out, int *avail_out) | |
79197521 | 54 | { |
2cac01e3 FS |
55 | if (use_zstd) |
56 | { | |
57 | #if HAVE_ZSTD | |
58 | ZSTD_outBuffer ob = { *next_out, *avail_out, 0 }; | |
59 | ZSTD_inBuffer ib = { *next_in, *avail_in, 0 }; | |
60 | size_t ret = ZSTD_compressStream2 (ctx, &ob, &ib, ZSTD_e_continue); | |
61 | *next_in += ib.pos; | |
62 | *avail_in -= ib.pos; | |
63 | *next_out += ob.pos; | |
64 | *avail_out -= ob.pos; | |
65 | if (ZSTD_isError (ret)) | |
66 | return -1; | |
67 | return (int)ob.pos; | |
68 | #endif | |
69 | } | |
70 | ||
71 | struct z_stream_s *strm = ctx; | |
79197521 AM |
72 | |
73 | strm->next_in = (Bytef *) (*next_in); | |
74 | strm->avail_in = *avail_in; | |
75 | strm->next_out = (Bytef *) (*next_out); | |
76 | strm->avail_out = *avail_out; | |
77 | ||
2cac01e3 | 78 | int x = deflate (strm, Z_NO_FLUSH); |
79197521 AM |
79 | if (x != Z_OK) |
80 | return -1; | |
81 | ||
2cac01e3 | 82 | int out_size = *avail_out - strm->avail_out; |
79197521 AM |
83 | *next_in = (char *) (strm->next_in); |
84 | *avail_in = strm->avail_in; | |
85 | *next_out = (char *) (strm->next_out); | |
86 | *avail_out = strm->avail_out; | |
87 | ||
88 | return out_size; | |
79197521 AM |
89 | } |
90 | ||
91 | /* Finish the compression and consume the remaining compressed output. | |
92 | Returns -1 for error, 0 when done, 1 when more output buffer is | |
93 | needed. */ | |
94 | ||
95 | int | |
2cac01e3 | 96 | compress_finish (bool use_zstd, void *ctx, char **next_out, |
0138187e | 97 | int *avail_out, int *out_size) |
79197521 | 98 | { |
2cac01e3 FS |
99 | if (use_zstd) |
100 | { | |
101 | #if HAVE_ZSTD | |
102 | ZSTD_outBuffer ob = { *next_out, *avail_out, 0 }; | |
3318d800 | 103 | ZSTD_inBuffer ib = { 0, 0, 0 }; |
2cac01e3 FS |
104 | size_t ret = ZSTD_compressStream2 (ctx, &ob, &ib, ZSTD_e_end); |
105 | *out_size = ob.pos; | |
106 | *next_out += ob.pos; | |
107 | *avail_out -= ob.pos; | |
108 | if (ZSTD_isError (ret)) | |
109 | return -1; | |
110 | if (ret == 0) | |
111 | ZSTD_freeCCtx (ctx); | |
112 | return ret ? 1 : 0; | |
113 | #endif | |
114 | } | |
115 | ||
79197521 | 116 | int x; |
2cac01e3 | 117 | struct z_stream_s *strm = ctx; |
79197521 AM |
118 | |
119 | strm->avail_in = 0; | |
120 | strm->next_out = (Bytef *) (*next_out); | |
121 | strm->avail_out = *avail_out; | |
122 | ||
123 | x = deflate (strm, Z_FINISH); | |
124 | ||
125 | *out_size = *avail_out - strm->avail_out; | |
126 | *next_out = (char *) (strm->next_out); | |
127 | *avail_out = strm->avail_out; | |
128 | ||
129 | if (x == Z_STREAM_END) | |
130 | { | |
131 | deflateEnd (strm); | |
132 | return 0; | |
133 | } | |
134 | if (strm->avail_out != 0) | |
135 | return -1; | |
136 | return 1; | |
79197521 | 137 | } |