summaryrefslogtreecommitdiff
path: root/lib/libc/stdio/fread_unlocked.c
blob: c80dd9c08b60b75face7eabb43c415acac479479 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#include "__stdio.h" // for __FILE, _IO_EOF, _IO_ERR

#include <stdio.h>
#include <string.h>
#include <unistd.h>

extern void __libc_init_io(void);
void *__libc_force_io_init = (void *)__libc_init_io;

size_t fread_unlocked(void *restrict ptr, size_t size, size_t nitems, FILE *restrict stream)
{
	if (size == 0 || nitems == 0)
		return 0;

	size_t total = size * nitems;
	size_t bytes_read = 0;
	char *p = ptr;

	while (total > 0) {
		if (stream->buf_pos < stream->buf_len) {
			size_t available = stream->buf_len - stream->buf_pos;
			size_t to_copy = total < available ? total : available;

			memcpy(p, stream->buf + stream->buf_pos, to_copy);
			stream->buf_pos += to_copy;
			p += to_copy;
			bytes_read += to_copy;
			total -= to_copy;
			continue;
		}

		ssize_t ret = read(stream->fd, stream->buf, stream->buf_size);
		if (ret <= 0) {
			if (ret < 0)
				stream->flags |= _IO_ERR;
			else
				stream->flags |= _IO_EOF;
			break;
		}

		stream->buf_len = ret;
		stream->buf_pos = 0;
	}

	return bytes_read / size;
}