summaryrefslogtreecommitdiff
path: root/usr/src/test/os-tests/tests/ucontext.c
blob: a6e43941c1f62fca744bde9849c46048bdb0f7b3 (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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2022 OmniOS Community Edition (OmniOSce) Association.
 */

#include <atomic.h>
#include <err.h>
#include <pthread.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <ucontext.h>
#include <unistd.h>
#include <sys/debug.h>
#include <sys/types.h>

static __thread uint64_t tlsvar;
static ucontext_t ctx;
static uint32_t failures;
static pthread_t tid;

#define	THREAD1_VALUE	0x1010
#define	THREAD2_VALUE	0x0202

void
report(char *tag)
{
	pthread_t ltid = pthread_self();

	printf("  %-14s: thread=%x, TLS variable=%x\n",
	    tag, (uint_t)ltid, tlsvar);
}

void
run(void)
{
	pthread_t ltid = pthread_self();

	printf("Coroutine started from second thread\n");
	report("coroutine");

	if (ltid != tid) {
		fprintf(stderr,
		    "FAIL: coroutine thread ID is %x, expected %x\n",
		    ltid, tid);
		atomic_inc_32(&failures);
	}

	if (tlsvar != THREAD2_VALUE) {
		fprintf(stderr,
		    "FAIL: coroutine TLS variable is %x, expected %x\n",
		    tlsvar, THREAD2_VALUE);
		atomic_inc_32(&failures);
	}
}

void *
thread(void *arg __unused)
{
	tlsvar = THREAD2_VALUE;

	report("second thread");
	/*
	 * setcontext() does not return if successful, checking the return
	 * value upsets smatch.
	 */
	(void) setcontext(&ctx);
	errx(EXIT_FAILURE, "setcontext() returned and should not have.");

	return (NULL);
}

int
main(void)
{
	char stk[SIGSTKSZ];
	void *status;

	tlsvar = THREAD1_VALUE;

	report("main thread");

	VERIFY0(getcontext(&ctx));
	ctx.uc_link = NULL;
	ctx.uc_stack.ss_sp = stk;
	ctx.uc_stack.ss_size = sizeof (stk);
	makecontext(&ctx, run, 0);

	VERIFY0(pthread_create(&tid, NULL, thread, NULL));
	VERIFY0(pthread_join(tid, &status));

	return (failures == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
}