You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
1.9 KiB
93 lines
1.9 KiB
6 years ago
|
#include <common.h>
|
||
6 years ago
|
|
||
|
|
void *mkstack(size_t size)
|
||
|
|
{
|
||
|
|
const size_t pgsize = 1 << 12;
|
||
|
|
size += pgsize - 1;
|
||
|
|
size &= ~(pgsize - 1);
|
||
|
|
char *ret = mmap(NULL, size, PROT_READ | PROT_WRITE,
|
||
|
|
MAP_ANON | MAP_PRIVATE, -1, 0);
|
||
|
|
if (!ret)
|
||
|
|
errx(-1, "couldn't mmap");
|
||
|
|
return ret + size;
|
||
|
|
}
|
||
|
|
|
||
|
|
void loop(char *msg, const int iters)
|
||
|
|
{
|
||
|
|
int j;
|
||
|
|
for (j = 0; j < iters; j++)
|
||
|
|
// appease gcc
|
||
|
|
printf("%s", msg);
|
||
|
|
}
|
||
|
|
|
||
|
|
__thread long tlvar;
|
||
|
|
|
||
|
|
long child(void *msg)
|
||
|
|
{
|
||
|
|
printf("thread var is: %ld\n", tlvar);
|
||
|
|
printf("child loop (msg: %s)\n", (char *)msg);
|
||
|
|
char buf[10];
|
||
|
|
snprintf(buf, sizeof(buf), "%ld ", tlvar);
|
||
|
|
loop(buf, 30);
|
||
|
|
printf("child exit\n");
|
||
|
|
return tlvar;
|
||
|
|
}
|
||
|
|
|
||
|
|
static long tls1;
|
||
|
|
static long tls2;
|
||
|
|
static void *tcbs[] = {(char *)&tls1 + 8, (char *)&tls2 + 8};
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
printf("start thread tests\n");
|
||
|
|
|
||
|
|
//int pid = fork();
|
||
|
|
//if (pid == 0) {
|
||
|
|
// threxit(1001);
|
||
|
|
// errx(-1, "why no exit");
|
||
|
|
// return -2;
|
||
|
|
//}
|
||
|
|
//int status;
|
||
|
|
//if (wait(&status) != pid)
|
||
|
|
// errx(-1, "wrong pid: %d\n", pid);
|
||
|
|
//if (status != 0)
|
||
|
|
// errx(-1, "posix is sad");
|
||
|
|
|
||
|
|
struct tfork_t tf;
|
||
|
|
|
||
|
|
tf.tf_tcb = &tcbs[0];
|
||
|
|
tf.tf_tid = &tls1;
|
||
|
|
tf.tf_stack = mkstack(4096*1);
|
||
|
|
int tid = tfork_thread(&tf, child, "child 1!");
|
||
|
|
if (tid < 0)
|
||
|
|
err(tid, "tfork");
|
||
|
|
if (tid != tls1)
|
||
|
|
errx(-1, "tid mismatch");
|
||
|
|
|
||
|
|
tf.tf_tcb = &tcbs[1];
|
||
|
|
tf.tf_tid = &tls2;
|
||
|
|
tf.tf_stack = mkstack(4096*1);
|
||
|
|
tid = tfork_thread(&tf, child, "child 2!");
|
||
|
|
if (tid < 0)
|
||
|
|
err(tid, "tfork");
|
||
|
|
if (tid != tls2)
|
||
|
|
errx(-1, "tid mismatch");
|
||
|
|
|
||
|
|
printf("parent loop\n");
|
||
|
|
loop("p ", 10);
|
||
|
|
|
||
|
|
printf("waiting for child threads\n");
|
||
|
|
long status;
|
||
|
|
if ((tid = thrwait(tls1, &status)) != tls1)
|
||
|
|
err(tid, "thrwait1");
|
||
|
|
if (status != tls1)
|
||
|
|
errx(-1, "bad status %ld %ld", status, tls1);
|
||
|
|
if ((tid = thrwait(tls2, &status)) != tls2)
|
||
|
|
err(tid, "thrwait2");
|
||
|
|
if (status != tls2)
|
||
|
|
errx(-1, "bad status %ld %ld", status, tls2);
|
||
|
|
printf("parent exit\n");
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|