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

FILE *__fdopen(int fd, const char *mode)
{
	FILE *f;
	struct termios tio;

	/* 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 35
		int flags = __syscall(SYS_fcntl, fd, F_GETFL);
		__syscall(SYS_fcntl, fd, F_SETFL, flags | O_APPEND);
36
		f->flags |= F_APP;
R
Rich Felker 已提交
37 38 39 40 41 42 43 44
	}

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

	/* Activate line buffered mode for terminals */
	f->lbf = EOF;
45
	if (!(f->flags & F_NOWR) && !__syscall(SYS_ioctl, fd, TCGETS, &tio))
R
Rich Felker 已提交
46 47 48 49 50 51 52 53
		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;

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

R
Rich Felker 已提交
56 57
	/* Add new FILE to open file list */
	OFLLOCK();
58 59 60
	f->next = libc.ofl_head;
	if (libc.ofl_head) libc.ofl_head->prev = f;
	libc.ofl_head = f;
R
Rich Felker 已提交
61 62 63 64 65 66
	OFLUNLOCK();

	return f;
}

weak_alias(__fdopen, fdopen);