This commit is contained in:
Lior Halphon 2020-09-19 19:31:31 +03:00
commit 7c92845882
2 changed files with 20 additions and 3 deletions

View File

@ -4,6 +4,7 @@
#include <stdlib.h> #include <stdlib.h>
#include <unistd.h> #include <unistd.h>
#include <assert.h> #include <assert.h>
#include <err.h>
void opts(uint8_t byte, uint8_t *options) void opts(uint8_t byte, uint8_t *options)
{ {
@ -13,6 +14,17 @@ void opts(uint8_t byte, uint8_t *options)
*(options++) = byte & (byte >> 1); *(options++) = byte & (byte >> 1);
} }
void write_all(int fd, const void *buf, size_t count) {
while (count) {
ssize_t written = write(fd, buf, count);
if (written < 0) {
err(1, "write");
}
count -= written;
buf += written;
}
}
int main() int main()
{ {
static uint8_t source[0x4000]; static uint8_t source[0x4000];
@ -76,15 +88,15 @@ int main()
if (bits >= 8) { if (bits >= 8) {
uint8_t outctl = control >> (bits - 8); uint8_t outctl = control >> (bits - 8);
assert(outctl != 1); assert(outctl != 1);
write(STDOUT_FILENO, &outctl, 1); write_all(STDOUT_FILENO, &outctl, 1);
write(STDOUT_FILENO, literals, literals_size); write_all(STDOUT_FILENO, literals, literals_size);
bits -= 8; bits -= 8;
control &= (1 << bits) - 1; control &= (1 << bits) - 1;
literals_size = 0; literals_size = 0;
} }
} }
uint8_t end_byte = 1; uint8_t end_byte = 1;
write(STDOUT_FILENO, &end_byte, 1); write_all(STDOUT_FILENO, &end_byte, 1);
return 0; return 0;
} }

View File

@ -106,6 +106,11 @@ ifeq ($(shell $(CC) -x c -c $(NULL) -o $(NULL) -Werror -Wpartial-availability 2>
WARNINGS += -Wpartial-availability WARNINGS += -Wpartial-availability
endif endif
# GCC's implementation of this warning has false positives, so we skip it
ifneq ($(shell $(CC) --version 2>&1 | grep "gcc"), )
WARNINGS += -Wno-maybe-uninitialized
endif
CFLAGS += $(WARNINGS) CFLAGS += $(WARNINGS)
CFLAGS += -std=gnu11 -D_GNU_SOURCE -DVERSION="$(VERSION)" -I. -D_USE_MATH_DEFINES CFLAGS += -std=gnu11 -D_GNU_SOURCE -DVERSION="$(VERSION)" -I. -D_USE_MATH_DEFINES