2 * Generating Objects from Buckets
4 * (c) 2004, Robert Spalek <robert@ucw.cz>
5 * (c) 2004, Martin Mares <mj@ucw.cz>
9 #include "lib/unaligned.h"
10 #include "lib/mempool.h"
11 #include "lib/fastbuf.h"
12 #include "lib/unicode.h"
13 #include "lib/object.h"
14 #include "lib/bucket.h"
15 #include "lib/lizard.h"
22 #define RET_ERR(num) ({ errno = num; return -1; })
27 struct lizard_buffer *lizard;
33 struct buck2obj_buf *buf = xmalloc(sizeof(struct buck2obj_buf));
35 buf->lizard = lizard_alloc();
40 buck2obj_free(struct buck2obj_buf *buf)
42 lizard_free(buf->lizard);
48 decode_attributes(byte *ptr, byte *end, struct odes *o, uns can_overwrite)
50 if (can_overwrite >= 2)
60 obj_add_attr_ref(o, type, ptr);
73 byte *dup = mp_alloc_fast_noalign(o->pool, len+1);
74 memcpy(dup, ptr, len);
76 obj_add_attr_ref(o, type, dup);
84 buck2obj_parse(struct buck2obj_buf *buf, uns buck_type, uns buck_len, struct fastbuf *body, struct odes *o_hdr, uns *body_start, struct odes *o_body)
86 if (buck_type == BUCKET_TYPE_PLAIN)
90 obj_read_multi(body, o_hdr); // ignore empty lines, read until EOF or NUL
92 else if (buck_type == BUCKET_TYPE_V30)
94 sh_off_t start = btell(body);
95 obj_read(body, o_hdr); // end on EOF or the first empty line
97 *body_start = btell(body) - start;
100 obj_read(body, o_body);
104 else if (buck_type == BUCKET_TYPE_V33 || buck_type == BUCKET_TYPE_V33_LIZARD)
106 /* Read all the bucket into 1 buffer, 0-copy if possible. */
108 uns len = bdirect_read_prepare(body, &ptr);
111 || (body->can_overwrite_buffer < 2 && buck_type == BUCKET_TYPE_V33))
113 /* Copy if the original buffer is too small.
114 * If it is write-protected, copy it also if it is uncompressed. */
115 bb_grow(&buf->bb, buck_len);
116 len = bread(body, buf->bb.ptr, buck_len);
123 ptr = decode_attributes(ptr, end, o_hdr, 0); // header
126 *body_start = ptr - start;
128 bdirect_read_commit(body, ptr);
131 if (buck_type == BUCKET_TYPE_V33_LIZARD) // decompression
137 uns adler = GET_U32(ptr);
139 byte *new_ptr = lizard_decompress_safe(ptr, buf->lizard, len);
142 if (adler32(new_ptr, len) != adler)
145 bdirect_read_commit(body, end);
150 ptr = decode_attributes(ptr, end, o_body, 2); // body
154 bdirect_read_commit_modified(body, ptr);
162 obj_read_bucket(struct buck2obj_buf *buf, struct mempool *pool, uns buck_type, uns buck_len, struct fastbuf *body, uns *body_start)
164 struct odes *o = obj_new(pool);
165 if (buck2obj_parse(buf, buck_type, buck_len, body, o, body_start, o) < 0)
172 obj_read(struct fastbuf *f, struct odes *o)
174 byte buf[MAX_ATTR_SIZE];
176 while (bgets(f, buf, sizeof(buf)))
180 obj_add_attr(o, buf[0], buf+1);
186 obj_read_multi(struct fastbuf *f, struct odes *o)
188 /* Read a multi-part object ending with either EOF or a NUL character */
189 byte buf[MAX_ATTR_SIZE];
190 while (bpeekc(f) > 0 && bgets(f, buf, sizeof(buf)))
192 obj_add_attr(o, buf[0], buf+1);