Max/SameBoy
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

pb12: check the return value of write

Browse Source
This commit is contained in:
Jakub Kądziołka 2020-09-17 19:56:31 +02:00
parent 2e4a638077
commit 9c50a992af
No known key found for this signature in database
GPG Key ID: E315A75846131564
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
BootROMs/pb12.c
View File

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