19 |
|
|
20 |
#define _GNU_SOURCE |
#define _GNU_SOURCE |
21 |
|
|
22 |
|
#include <config.h> |
23 |
|
|
24 |
#include "stream.h" |
#include "stream.h" |
25 |
|
|
26 |
#include <stdarg.h> |
#include <stdarg.h> |
41 |
} |
} |
42 |
|
|
43 |
void Stream_write(Stream * stream, const void * p, size_t size) { |
void Stream_write(Stream * stream, const void * p, size_t size) { |
44 |
if (SIZE_MAX - size < stream->length) { |
size_t stream_length = stream->length; |
45 |
fatal("out of memory"); |
size_t stream_capacity = stream->capacity; |
46 |
} |
if (stream_capacity - stream_length < size) { |
47 |
|
if (SIZE_MAX - size < stream_length) { |
48 |
|
fatal("out of memory"); |
49 |
|
} |
50 |
|
|
51 |
size_t new_length = stream->length + size; |
if (SIZE_MAX / 2 < stream_capacity) { |
52 |
size_t new_capacity = stream->capacity; |
stream_capacity = SIZE_MAX; |
|
if (new_capacity < new_length) { |
|
|
if (SIZE_MAX / 2 < new_capacity) { |
|
|
new_capacity = SIZE_MAX; |
|
53 |
} |
} |
54 |
else { |
else { |
55 |
new_capacity *= 2; |
stream_capacity *= 2; |
56 |
} |
} |
57 |
|
|
58 |
if (new_capacity < new_length) { |
size_t new_length = stream_length + size; |
59 |
new_capacity = new_length; |
if (stream_capacity < new_length) { |
60 |
|
stream_capacity = new_length; |
61 |
} |
} |
|
} |
|
62 |
|
|
63 |
if (new_capacity != stream->capacity) { |
stream->data = xrealloc(stream->data, stream_capacity); |
64 |
stream->data = xrealloc(stream->data, stream->capacity); |
stream->capacity = stream_capacity; |
65 |
stream->capacity = new_capacity; |
memcpy(stream->data + stream_length, p, size); |
66 |
|
stream->length = new_length; |
67 |
|
} |
68 |
|
else { |
69 |
|
memcpy(stream->data + stream_length, p, size); |
70 |
|
stream->length = stream_length + size; |
71 |
} |
} |
|
|
|
|
memcpy(stream->data + stream->length, p, size); |
|
|
stream->length = new_length; |
|
72 |
} |
} |
73 |
|
|
74 |
void Stream_write_string(Stream * stream, const char * s) { |
void Stream_write_string(Stream * stream, const char * s) { |
107 |
free(stream->data); |
free(stream->data); |
108 |
free(stream); |
free(stream); |
109 |
} |
} |
110 |
|
|
111 |
|
void Stream_reset(Stream * stream) { |
112 |
|
stream->length = 0; |
113 |
|
} |