9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * This file and its contents are supplied under the terms of the
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Common Development and Distribution License ("CDDL"), version 1.0.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * You may only use this file in accordance with the terms of version
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * 1.0 of the CDDL.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi *
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * A full copy of the text of the CDDL should have accompanied this
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * source. A copy of the CDDL is also available via the Internet at
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * http://www.illumos.org/license/CDDL.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Copyright (c) 2015, Joyent, Inc.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Tests to make sure that a parent and child do not get the same arc4random
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * state across a fork from a signal handler. This source file is used to make
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * two tests. One which initializes the data in advance, one of which does not.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <stdlib.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <sys/mman.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <assert.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <errno.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <sys/types.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <unistd.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <sys/wait.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <signal.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <strings.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchitypedef struct arc4_fork {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi uint32_t af_parent;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi uint32_t af_child;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi uint8_t af_pbuf[4096];
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi uint8_t af_cbuf[4096];
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi} arc4_fork_t;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchiarc4_fork_t *fork_data;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchistatic pid_t pid;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*ARGSUSED*/
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchistatic void
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchisiguser_fork(int sig, siginfo_t *sip, void *ucp)
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi{
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi pid = fork();
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi}
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchiint
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchimain(void)
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi{
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi int e, i, ret;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi pid_t child;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi struct sigaction sact;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi bzero(&sact, sizeof (struct sigaction));
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi sact.sa_flags = SA_SIGINFO;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi sact.sa_sigaction = siguser_fork;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi (void) sigemptyset(&sact.sa_mask);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi ret = sigaction(SIGUSR1, &sact, NULL);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ret == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#ifdef ARC4_PREINIT
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi (void) arc4random();
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#endif
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi fork_data = (arc4_fork_t *)mmap(NULL, sizeof (arc4_fork_t),
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANON, -1, 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(fork_data != MAP_FAILED);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi ret = raise(SIGUSR1);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ret == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(pid != -1);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi if (pid == 0) {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi fork_data->af_child = arc4random();
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi arc4random_buf(fork_data->af_cbuf, sizeof (fork_data->af_cbuf));
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi exit(0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi }
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi fork_data->af_parent = arc4random();
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi arc4random_buf(fork_data->af_pbuf, sizeof (fork_data->af_pbuf));
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi do {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi child = wait(&e);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi } while (child == -1 && errno == EINTR);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(child == pid);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi /* Now verify our data doesn't match */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(fork_data->af_parent != fork_data->af_child);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi /*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * For the buffer here, we're mostly concerned that they aren't somehow
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * getting the same stream.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi for (i = 0; i < sizeof (fork_data->af_pbuf); i++) {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi if (fork_data->af_pbuf[i] != fork_data->af_cbuf[i])
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi break;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi }
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(i != sizeof (fork_data->af_pbuf));
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi return (0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi}