__init_security.c 723 字节
Newer Older
1 2 3 4 5 6 7 8 9 10
#include <stddef.h>
#include <elf.h>
#include <poll.h>
#include <fcntl.h>
#include "syscall.h"
#include "libc.h"
#include "atomic.h"

#define AUX_CNT 24

11
static void dummy(size_t *auxv)
12 13 14 15
{
}
weak_alias(dummy, __init_ssp);

16 17 18 19 20
void __init_security(size_t *auxv)
{
	size_t i, aux[AUX_CNT] = { 0 };
	struct pollfd pfd[3] = { {.fd=0}, {.fd=1}, {.fd=2} };

21 22 23
#ifndef SHARED
	__init_ssp(auxv);
#endif
24

25 26 27 28 29 30 31 32 33 34 35
	for (; auxv[0]; auxv+=2) if (auxv[0]<AUX_CNT) aux[auxv[0]] = auxv[1];
	if (aux[AT_UID]==aux[AT_EUID] && aux[AT_GID]==aux[AT_EGID]
		&& !aux[AT_SECURE]) return;

	__syscall(SYS_poll, pfd, 3, 0);
	for (i=0; i<3; i++)
		if (pfd[i].revents&POLLNVAL)
			if (__syscall(SYS_open, "/dev/null", O_RDWR)<0)
				a_crash();
	libc.secure = 1;
}