/**
 * cc -pthread -o /tmp/singlestep /public/singlestep.c
 */

#include <unistd.h>
#include <stdio.h>
#include <sys/ptrace.h>
#include <sys/wait.h>
#include <sys/user.h>
#include <sys/types.h>
#include <signal.h>

#include <machine/reg.h>


#include <assert.h>
#include <stddef.h>

void
handler(int n)
{
	asm("nop");
}

int
child(void)
{
	assert(ptrace(PT_TRACE_ME, 0,0,0) == 0);
	signal(SIGALRM, handler);
	kill(getpid(), SIGALRM);
	return 0x23;
}

void *
getip(int pid)
{
	struct reg regs;
	ptrace(PT_GETREGS, pid, (void*)&regs, 0);

	return (void*)regs.pc;
}

int
main(void)
{
	int pid, status;

	pid = fork();
	if (!pid)
		return child();

	assert(wait(&status) == pid);
	assert(WIFSTOPPED(status) && WSTOPSIG(status) == SIGALRM);

	assert(ptrace(PT_STEP, pid, 0, SIGALRM) == 0);
	assert(wait(&status) == pid);
	assert(WIFSTOPPED(status) && WSTOPSIG(status) == SIGTRAP);
	assert((getip(pid) - (void*)handler) == 0);

	assert(ptrace(PT_STEP, pid, 0, SIGALRM) == 0);
	assert(wait(&status) == pid);
	assert(WIFSTOPPED(status) && WSTOPSIG(status) == SIGTRAP);
	assert((getip(pid) - (void*)handler) == 1);

	assert(ptrace(PT_CONTINUE, pid, 0,0) == 0);
	assert(wait(&status) == pid);
	assert(WIFEXITED(status) && WEXITSTATUS(status) == 0x23);

	return 0;
}