jehanne/qa/lib/c/rwakeup0.c

143 lines
2.7 KiB
C

/*
* This file is part of Jehanne.
*
* Copyright (C) 2015 Giacomo Tesio <giacomo@tesio.it>
*
* Jehanne is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, version 2 of the License.
*
* Jehanne is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Jehanne. If not, see <http://www.gnu.org/licenses/>.
*/
#include <u.h>
#include <libc.h>
/* verify that rwakeup returns 0 on Rendez that has already been awaken */
Rendez r;
QLock l;
int ready;
int verbose = 0;
int killerProc; /* pid, will kill the other processes if starved */
int
handletimeout(void *v, char *s)
{
/* just not exit, please */
if(strcmp(s, "timedout") == 0){
if(verbose)
print("%d: noted: %s\n", getpid(), s);
print("FAIL: timedout\n");
exits("FAIL");
}
return 0;
}
void
killKiller(void)
{
postnote(PNPROC, killerProc, "interrupt");
}
void
stopAllAfter(int seconds)
{
int pid;
switch((pid = rfork(RFMEM|RFPROC|RFNOWAIT)))
{
case 0:
if(verbose)
print("killer proc started: pid %d\n", getpid());
sleep(seconds * 1000);
postnote(PNGROUP, killerProc, "timedout");
if(verbose)
print("killer proc timedout: pid %d\n", getpid());
exits("FAIL");
case -1:
fprint(2, "%r\n");
exits("rfork fails");
default:
killerProc = pid;
atexit(killKiller);
}
}
void
main(void)
{
int s, w;
rfork(RFNOTEG);
if (!atnotify(handletimeout, 1)){
fprint(2, "%r\n");
exits("atnotify fails");
}
r.l = &l;
stopAllAfter(30);
/* one process to sleep */
switch((s = rfork(RFMEM|RFPROC|RFNOWAIT)))
{
case 0:
ready = 1;
qlock(&l);
rsleep(&r);
qunlock(&l);
exits(nil);
break;
case -1:
print("rfork: %r\n");
exits("rfork fails");
break;
default:
while(ready == 0)
;
break;
}
/* one process to wakeup */
switch((w = rfork(RFMEM|RFPROC|RFNOWAIT)))
{
case 0:
qlock(&l);
rwakeup(&r);
ready = 2;
qunlock(&l);
exits(nil);
break;
case -1:
print("rfork: %r\n");
exits("rfork fails");
break;
default:
while(ready == 1)
;
break;
}
/* now, we try to wakeup a free Rendez */
qlock(&l);
if(rwakeup(&r) == 0){
qunlock(&l);
print("PASS\n");
exits("PASS");
}
if((s = open(smprint("/proc/%d/ctl", s), OWRITE)) >= 0){
write(s, "kill", 5);
close(s);
}
if((w = open(smprint("/proc/%d/ctl", w), OWRITE)) >= 0){
write(s, "kill", 5);
close(s);
}
print("FAIL");
exits("FAIL");
}