__fdopen.c 1.4 KB
Newer Older
R
Rich Felker 已提交
1
#include "stdio_impl.h"
R
Rich Felker 已提交
2 3 4 5 6
#include <stdlib.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
7
#include "libc.h"
R
Rich Felker 已提交
8 9 10 11

FILE *__fdopen(int fd, const char *mode)
{
	FILE *f;
12
	struct winsize wsz;
R
Rich Felker 已提交
13 14

	/* Check for valid initial mode character */
15 16 17 18
	if (!strchr("rwa", *mode)) {
		errno = EINVAL;
		return 0;
	}
R
Rich Felker 已提交
19 20 21 22 23 24 25 26

	/* Allocate FILE+buffer or fail */
	if (!(f=malloc(sizeof *f + UNGET + BUFSIZ))) return 0;

	/* Zero-fill only the struct, not the buffer */
	memset(f, 0, sizeof *f);

	/* Impose mode restrictions */
27 28 29 30
	if (!strchr(mode, '+')) f->flags = (*mode == 'r') ? F_NOWR : F_NORD;

	/* Apply close-on-exec flag */
	if (strchr(mode, 'e')) __syscall(SYS_fcntl, fd, F_SETFD, FD_CLOEXEC);
R
Rich Felker 已提交
31 32 33

	/* Set append mode on fd if opened for append */
	if (*mode == 'a') {
34
		int flags = __syscall(SYS_fcntl, fd, F_GETFL);
35 36
		if (!(flags & O_APPEND))
			__syscall(SYS_fcntl, fd, F_SETFL, flags | O_APPEND);
37
		f->flags |= F_APP;
R
Rich Felker 已提交
38 39 40 41 42 43 44 45
	}

	f->fd = fd;
	f->buf = (unsigned char *)f + sizeof *f + UNGET;
	f->buf_size = BUFSIZ;

	/* Activate line buffered mode for terminals */
	f->lbf = EOF;
46
	if (!(f->flags & F_NOWR) && !__syscall(SYS_ioctl, fd, TIOCGWINSZ, &wsz))
R
Rich Felker 已提交
47 48 49 50 51 52 53 54
		f->lbf = '\n';

	/* Initialize op ptrs. No problem if some are unneeded. */
	f->read = __stdio_read;
	f->write = __stdio_write;
	f->seek = __stdio_seek;
	f->close = __stdio_close;

55 56
	if (!libc.threaded) f->lock = -1;

R
Rich Felker 已提交
57
	/* Add new FILE to open file list */
58
	return __ofl_add(f);
R
Rich Felker 已提交
59 60 61
}

weak_alias(__fdopen, fdopen);