summaryrefslogtreecommitdiffstats
path: root/lib/fd.c
blob: bcfa374a2dfa0f35db88f50761f27e1adde419e8 (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
// SPDX-FileCopyrightText: 2024, Skyler Ferrante <sjf5462@rit.edu>
// SPDX-License-Identifier: BSD-3-Clause

/**
 * To protect against file descriptor omission attacks, we open the std file
 * descriptors with /dev/null if they are not already open. Code is based on
 * fix_fds from sudo.c.
 */

#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>

#include "prototypes.h"

static void check_fd(int fd);

void
check_fds(void)
{
	/**
	 * Make sure stdin, stdout, stderr are open
	 * If they are closed, set them to /dev/null
	 */
	check_fd(STDIN_FILENO);
	check_fd(STDOUT_FILENO);
	check_fd(STDERR_FILENO);
}

static void
check_fd(int fd)
{
	int  devnull;

	if (fcntl(fd, F_GETFL, 0) != -1)
		return;

	devnull = open("/dev/null", O_RDWR);
	if (devnull != fd)
		abort();
}