109 lines
2.3 KiB
C
109 lines
2.3 KiB
C
|
/*
|
||
|
|
* This file is part of Jehanne.
|
||
|
|
*
|
||
|
|
* Copyright (C) 2016 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>
|
||
|
|
|
||
|
|
int verbose = 0;
|
||
|
|
int procNoteReceived;
|
||
|
|
int groupNoteReceived;
|
||
|
|
|
||
|
|
int
|
||
|
|
printFromProc(void *v, char *s)
|
||
|
|
{
|
||
|
|
/* just not exit, please */
|
||
|
|
if(strcmp(s, "proc") == 0){
|
||
|
|
if(procNoteReceived)
|
||
|
|
atnotify(printFromProc, 0);
|
||
|
|
procNoteReceived++;
|
||
|
|
if(verbose)
|
||
|
|
fprint(2, "%d: noted: %s at %lld\n", getpid(), s, nsec());
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
printFromGroup(void *v, char *s)
|
||
|
|
{
|
||
|
|
/* just not exit, please */
|
||
|
|
if(strcmp(s, "group") == 0){
|
||
|
|
if(groupNoteReceived)
|
||
|
|
atnotify(printFromGroup, 0);
|
||
|
|
groupNoteReceived++;
|
||
|
|
if(verbose)
|
||
|
|
fprint(2, "%d: noted: %s at %lld\n", getpid(), s, nsec());
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
fail(void *v, char *s)
|
||
|
|
{
|
||
|
|
if(groupNoteReceived == 2 || procNoteReceived == 2){
|
||
|
|
print("FAIL\n");
|
||
|
|
exits("FAIL");
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
main(void)
|
||
|
|
{
|
||
|
|
int ppid;
|
||
|
|
Waitmsg *w;
|
||
|
|
|
||
|
|
if (!atnotify(printFromProc, 1) || !atnotify(printFromGroup, 1) || !atnotify(fail, 1)){
|
||
|
|
fprint(2, "%r\n");
|
||
|
|
exits("atnotify fails");
|
||
|
|
}
|
||
|
|
|
||
|
|
rfork(RFNOTEG|RFNAMEG);
|
||
|
|
|
||
|
|
if(rfork(RFPROC) == 0){
|
||
|
|
ppid = getppid(); // this comes from exec() syscall
|
||
|
|
|
||
|
|
postnote(PNPROC, ppid, "proc");
|
||
|
|
postnote(PNGROUP, getpid(), "group");
|
||
|
|
|
||
|
|
rfork(RFNOMNT);
|
||
|
|
postnote(PNPROC, ppid, "proc");
|
||
|
|
postnote(PNGROUP, getpid(), "group");
|
||
|
|
|
||
|
|
rfork(RFNOMNT|RFCNAMEG);
|
||
|
|
if(postnote(PNPROC, ppid, "proc") != -1 ||
|
||
|
|
postnote(PNGROUP, getpid(), "group") != -1){
|
||
|
|
exits("FAIL");
|
||
|
|
}
|
||
|
|
|
||
|
|
exits(nil);
|
||
|
|
} else {
|
||
|
|
while((w = wait()) == nil)
|
||
|
|
;
|
||
|
|
if(w->msg[0]){
|
||
|
|
exits("FAIL");
|
||
|
|
}
|
||
|
|
if(procNoteReceived < 2 || groupNoteReceived < 2){
|
||
|
|
fprint(2, "FAIL: not enough notes.\n");
|
||
|
|
exits("FAIL");
|
||
|
|
}
|
||
|
|
print("PASS\n");
|
||
|
|
exits("PASS");
|
||
|
|
}
|
||
|
|
}
|