mirror of
https://github.com/opnsense/src.git
synced 2026-06-06 23:32:52 -04:00
1f155d48f8
The entire recv*() implementation set is ripe for opportunities to validate, so do what we can with what we have. Reviewed by: markj Sponsored by: Klara, Inc. Sponsored by: Stormshield Differential Revision: https://reviews.freebsd.org/D45686
432 lines
9.2 KiB
C
432 lines
9.2 KiB
C
/* @generated by `generate-fortify-tests.lua "stdlib"` */
|
|
|
|
#define _FORTIFY_SOURCE 2
|
|
#define TMPFILE_SIZE (1024 * 32)
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/random.h>
|
|
#include <sys/resource.h>
|
|
#include <sys/select.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/time.h>
|
|
#include <sys/uio.h>
|
|
#include <sys/wait.h>
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
#include <poll.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <sysexits.h>
|
|
#include <unistd.h>
|
|
#include <wchar.h>
|
|
#include <atf-c.h>
|
|
|
|
static FILE * __unused
|
|
new_fp(size_t __len)
|
|
{
|
|
static char fpbuf[LINE_MAX];
|
|
FILE *fp;
|
|
|
|
ATF_REQUIRE(__len <= sizeof(fpbuf));
|
|
|
|
memset(fpbuf, 'A', sizeof(fpbuf) - 1);
|
|
fpbuf[sizeof(fpbuf) - 1] = '\0';
|
|
|
|
fp = fmemopen(fpbuf, sizeof(fpbuf), "rb");
|
|
ATF_REQUIRE(fp != NULL);
|
|
|
|
return (fp);
|
|
}
|
|
|
|
/*
|
|
* Create a new symlink to use for readlink(2) style tests, we'll just use a
|
|
* random target name to have something interesting to look at.
|
|
*/
|
|
static const char * __unused
|
|
new_symlink(size_t __len)
|
|
{
|
|
static const char linkname[] = "link";
|
|
char target[MAXNAMLEN];
|
|
int error;
|
|
|
|
ATF_REQUIRE(__len <= sizeof(target));
|
|
|
|
arc4random_buf(target, sizeof(target));
|
|
|
|
error = unlink(linkname);
|
|
ATF_REQUIRE(error == 0 || errno == ENOENT);
|
|
|
|
error = symlink(target, linkname);
|
|
ATF_REQUIRE(error == 0);
|
|
|
|
return (linkname);
|
|
}
|
|
|
|
/*
|
|
* For our purposes, first descriptor will be the reader; we'll send both
|
|
* raw data and a control message over it so that the result can be used for
|
|
* any of our recv*() tests.
|
|
*/
|
|
static void __unused
|
|
new_socket(int sock[2])
|
|
{
|
|
unsigned char ctrl[CMSG_SPACE(sizeof(int))] = { 0 };
|
|
static char sockbuf[256];
|
|
ssize_t rv;
|
|
size_t total = 0;
|
|
struct msghdr hdr = { 0 };
|
|
struct cmsghdr *cmsg;
|
|
int error, fd;
|
|
|
|
error = socketpair(AF_UNIX, SOCK_STREAM, 0, sock);
|
|
ATF_REQUIRE(error == 0);
|
|
|
|
while (total != sizeof(sockbuf)) {
|
|
rv = send(sock[1], &sockbuf[total], sizeof(sockbuf) - total, 0);
|
|
|
|
ATF_REQUIRE_MSG(rv > 0,
|
|
"expected bytes sent, got %zd with %zu left (size %zu, total %zu)",
|
|
rv, sizeof(sockbuf) - total, sizeof(sockbuf), total);
|
|
ATF_REQUIRE_MSG(total + (size_t)rv <= sizeof(sockbuf),
|
|
"%zd exceeds total %zu", rv, sizeof(sockbuf));
|
|
total += rv;
|
|
}
|
|
|
|
hdr.msg_control = ctrl;
|
|
hdr.msg_controllen = sizeof(ctrl);
|
|
|
|
cmsg = CMSG_FIRSTHDR(&hdr);
|
|
cmsg->cmsg_level = SOL_SOCKET;
|
|
cmsg->cmsg_type = SCM_RIGHTS;
|
|
cmsg->cmsg_len = CMSG_LEN(sizeof(fd));
|
|
fd = STDIN_FILENO;
|
|
memcpy(CMSG_DATA(cmsg), &fd, sizeof(fd));
|
|
|
|
error = sendmsg(sock[1], &hdr, 0);
|
|
ATF_REQUIRE(error != -1);
|
|
}
|
|
|
|
/*
|
|
* Constructs a tmpfile that we can use for testing read(2) and friends.
|
|
*/
|
|
static int __unused
|
|
new_tmpfile(void)
|
|
{
|
|
char buf[1024];
|
|
ssize_t rv;
|
|
size_t written;
|
|
int fd;
|
|
|
|
fd = open("tmpfile", O_RDWR | O_CREAT | O_TRUNC, 0644);
|
|
ATF_REQUIRE(fd >= 0);
|
|
|
|
written = 0;
|
|
while (written < TMPFILE_SIZE) {
|
|
rv = write(fd, buf, sizeof(buf));
|
|
ATF_REQUIRE(rv > 0);
|
|
|
|
written += rv;
|
|
}
|
|
|
|
ATF_REQUIRE_EQ(0, lseek(fd, 0, SEEK_SET));
|
|
return (fd);
|
|
}
|
|
|
|
static void
|
|
disable_coredumps(void)
|
|
{
|
|
struct rlimit rl = { 0 };
|
|
|
|
if (setrlimit(RLIMIT_CORE, &rl) == -1)
|
|
_exit(EX_OSERR);
|
|
}
|
|
|
|
/*
|
|
* Replaces stdin with a file that we can actually read from, for tests where
|
|
* we want a FILE * or fd that we can get data from.
|
|
*/
|
|
static void __unused
|
|
replace_stdin(void)
|
|
{
|
|
int fd;
|
|
|
|
fd = new_tmpfile();
|
|
|
|
(void)dup2(fd, STDIN_FILENO);
|
|
if (fd != STDIN_FILENO)
|
|
close(fd);
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(arc4random_buf_before_end);
|
|
ATF_TC_BODY(arc4random_buf_before_end, tc)
|
|
{
|
|
#define BUF &__stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char __buf[42];
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(__stack.__buf);
|
|
const size_t __len = 42 - 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
arc4random_buf(__stack.__buf, __len);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(arc4random_buf_end);
|
|
ATF_TC_BODY(arc4random_buf_end, tc)
|
|
{
|
|
#define BUF &__stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char __buf[42];
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(__stack.__buf);
|
|
const size_t __len = 42;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
arc4random_buf(__stack.__buf, __len);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(arc4random_buf_heap_before_end);
|
|
ATF_TC_BODY(arc4random_buf_heap_before_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (42);
|
|
const size_t __len = 42 - 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
arc4random_buf(__stack.__buf, __len);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(arc4random_buf_heap_end);
|
|
ATF_TC_BODY(arc4random_buf_heap_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (42);
|
|
const size_t __len = 42;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
arc4random_buf(__stack.__buf, __len);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(arc4random_buf_heap_after_end);
|
|
ATF_TC_BODY(arc4random_buf_heap_after_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (42);
|
|
const size_t __len = 42 + 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
pid_t __child;
|
|
int __status;
|
|
|
|
__child = fork();
|
|
ATF_REQUIRE(__child >= 0);
|
|
if (__child > 0)
|
|
goto monitor;
|
|
|
|
/* Child */
|
|
disable_coredumps();
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
arc4random_buf(__stack.__buf, __len);
|
|
_exit(EX_SOFTWARE); /* Should have aborted. */
|
|
|
|
monitor:
|
|
while (waitpid(__child, &__status, 0) != __child) {
|
|
ATF_REQUIRE_EQ(EINTR, errno);
|
|
}
|
|
|
|
if (!WIFSIGNALED(__status)) {
|
|
switch (WEXITSTATUS(__status)) {
|
|
case EX_SOFTWARE:
|
|
atf_tc_fail("FORTIFY_SOURCE failed to abort");
|
|
break;
|
|
case EX_OSERR:
|
|
atf_tc_fail("setrlimit(2) failed");
|
|
break;
|
|
default:
|
|
atf_tc_fail("child exited with status %d",
|
|
WEXITSTATUS(__status));
|
|
}
|
|
} else {
|
|
ATF_REQUIRE_EQ(SIGABRT, WTERMSIG(__status));
|
|
}
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(realpath_before_end);
|
|
ATF_TC_BODY(realpath_before_end, tc)
|
|
{
|
|
#define BUF &__stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char __buf[PATH_MAX + 1];
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(__stack.__buf);
|
|
const size_t __len = PATH_MAX + 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
realpath(".", __stack.__buf);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(realpath_end);
|
|
ATF_TC_BODY(realpath_end, tc)
|
|
{
|
|
#define BUF &__stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char __buf[PATH_MAX];
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(__stack.__buf);
|
|
const size_t __len = PATH_MAX;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
realpath(".", __stack.__buf);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(realpath_heap_before_end);
|
|
ATF_TC_BODY(realpath_heap_before_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (PATH_MAX + 1);
|
|
const size_t __len = PATH_MAX + 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
realpath(".", __stack.__buf);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(realpath_heap_end);
|
|
ATF_TC_BODY(realpath_heap_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (PATH_MAX);
|
|
const size_t __len = PATH_MAX;
|
|
const size_t __idx __unused = __len - 1;
|
|
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
realpath(".", __stack.__buf);
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TC_WITHOUT_HEAD(realpath_heap_after_end);
|
|
ATF_TC_BODY(realpath_heap_after_end, tc)
|
|
{
|
|
#define BUF __stack.__buf
|
|
struct {
|
|
uint8_t padding_l;
|
|
unsigned char * __buf;
|
|
uint8_t padding_r;
|
|
} __stack;
|
|
const size_t __bufsz __unused = sizeof(*__stack.__buf) * (PATH_MAX - 1);
|
|
const size_t __len = PATH_MAX - 1;
|
|
const size_t __idx __unused = __len - 1;
|
|
pid_t __child;
|
|
int __status;
|
|
|
|
__child = fork();
|
|
ATF_REQUIRE(__child >= 0);
|
|
if (__child > 0)
|
|
goto monitor;
|
|
|
|
/* Child */
|
|
disable_coredumps();
|
|
__stack.__buf = malloc(__bufsz);
|
|
|
|
realpath(".", __stack.__buf);
|
|
_exit(EX_SOFTWARE); /* Should have aborted. */
|
|
|
|
monitor:
|
|
while (waitpid(__child, &__status, 0) != __child) {
|
|
ATF_REQUIRE_EQ(EINTR, errno);
|
|
}
|
|
|
|
if (!WIFSIGNALED(__status)) {
|
|
switch (WEXITSTATUS(__status)) {
|
|
case EX_SOFTWARE:
|
|
atf_tc_fail("FORTIFY_SOURCE failed to abort");
|
|
break;
|
|
case EX_OSERR:
|
|
atf_tc_fail("setrlimit(2) failed");
|
|
break;
|
|
default:
|
|
atf_tc_fail("child exited with status %d",
|
|
WEXITSTATUS(__status));
|
|
}
|
|
} else {
|
|
ATF_REQUIRE_EQ(SIGABRT, WTERMSIG(__status));
|
|
}
|
|
#undef BUF
|
|
|
|
}
|
|
|
|
ATF_TP_ADD_TCS(tp)
|
|
{
|
|
ATF_TP_ADD_TC(tp, arc4random_buf_before_end);
|
|
ATF_TP_ADD_TC(tp, arc4random_buf_end);
|
|
ATF_TP_ADD_TC(tp, arc4random_buf_heap_before_end);
|
|
ATF_TP_ADD_TC(tp, arc4random_buf_heap_end);
|
|
ATF_TP_ADD_TC(tp, arc4random_buf_heap_after_end);
|
|
ATF_TP_ADD_TC(tp, realpath_before_end);
|
|
ATF_TP_ADD_TC(tp, realpath_end);
|
|
ATF_TP_ADD_TC(tp, realpath_heap_before_end);
|
|
ATF_TP_ADD_TC(tp, realpath_heap_end);
|
|
ATF_TP_ADD_TC(tp, realpath_heap_after_end);
|
|
return (atf_no_error());
|
|
}
|