xv6-cs450/usertests.c

169 lines
2.6 KiB
C
Raw Normal View History

#include "types.h"
#include "stat.h"
#include "user.h"
#include "fcntl.h"
char buf[2048];
2006-06-27 16:35:53 +02:00
// simple fork and pipe read/write
2006-06-27 16:35:53 +02:00
void
2006-07-17 03:25:22 +02:00
pipe1(void)
2006-06-27 16:35:53 +02:00
{
int fds[2], pid;
int seq = 0, i, n, cc, total;
2006-06-27 16:35:53 +02:00
2006-08-29 21:06:37 +02:00
if(pipe(fds) != 0){
puts("pipe() failed\n");
exit();
}
pid = fork();
2006-06-27 16:35:53 +02:00
if(pid == 0){
close(fds[0]);
for(n = 0; n < 5; n++){
for(i = 0; i < 1033; i++)
buf[i] = seq++;
if(write(fds[1], buf, 1033) != 1033){
printf(1, "pipe1 oops 1\n");
exit();
}
}
exit();
2006-08-29 21:06:37 +02:00
} else if(pid > 0){
close(fds[1]);
total = 0;
cc = 1;
2006-07-17 03:25:22 +02:00
while((n = read(fds[0], buf, cc)) > 0){
for(i = 0; i < n; i++){
if((buf[i] & 0xff) != (seq++ & 0xff)){
printf(1, "pipe1 oops 2\n");
return;
}
}
total += n;
cc = cc * 2;
if(cc > sizeof(buf))
cc = sizeof(buf);
2006-06-27 16:35:53 +02:00
}
if(total != 5 * 1033)
2006-08-29 21:06:37 +02:00
printf(1, "pipe1 oops 3 total %d\n", total);
close(fds[0]);
wait();
2006-08-29 21:06:37 +02:00
} else {
puts("fork() failed\n");
exit();
2006-06-27 16:35:53 +02:00
}
puts("pipe1 ok\n");
}
// meant to be run w/ at most two CPUs
void
2006-07-17 03:25:22 +02:00
preempt(void)
{
int pid1, pid2, pid3;
int pfds[2];
pid1 = fork();
if(pid1 == 0)
2006-07-17 03:25:22 +02:00
for(;;)
;
pid2 = fork();
if(pid2 == 0)
2006-07-17 03:25:22 +02:00
for(;;)
;
pipe(pfds);
pid3 = fork();
if(pid3 == 0){
close(pfds[0]);
if(write(pfds[1], "x", 1) != 1)
printf(1, "preempt write error");
close(pfds[1]);
2006-07-17 03:25:22 +02:00
for(;;)
;
}
close(pfds[1]);
if(read(pfds[0], buf, sizeof(buf)) != 1){
printf(1, "preempt read error");
return;
}
close(pfds[0]);
kill(pid1);
kill(pid2);
kill(pid3);
wait();
wait();
wait();
puts("preempt ok\n");
}
// try to find any races between exit and wait
void
2006-07-17 03:25:22 +02:00
exitwait(void)
{
int i, pid;
for(i = 0; i < 100; i++){
pid = fork();
if(pid < 0){
printf(1, "fork failed\n");
return;
}
if(pid){
if(wait() != pid){
printf(1, "wait wrong pid\n");
return;
}
} else {
exit();
}
}
puts("exitwait ok\n");
}
2006-08-24 21:24:36 +02:00
void
mem(void)
{
void *m1, *m2;
2006-08-29 21:06:37 +02:00
int pid;
2006-08-24 21:24:36 +02:00
2006-08-29 21:06:37 +02:00
if((pid = fork()) == 0){
m1 = 0;
while ((m2 = malloc(10001)) != 0) {
*(char **) m2 = m1;
m1 = m2;
}
while (m1) {
m2 = *(char **)m1;
free(m1);
m1 = m2;
}
m1 = malloc(1024*20);
if (m1 == 0) {
puts("couldn't allocate mem?!!\n");
exit();
}
2006-08-24 21:24:36 +02:00
free(m1);
2006-08-29 21:06:37 +02:00
printf(1, "mem ok\n");
2006-08-24 21:24:36 +02:00
exit();
2006-08-29 21:06:37 +02:00
} else {
wait();
2006-08-24 21:24:36 +02:00
}
}
int
2006-07-29 00:33:07 +02:00
main(int argc, char *argv[])
2006-06-27 16:35:53 +02:00
{
puts("usertests starting\n");
2006-08-24 21:24:36 +02:00
mem();
2006-08-29 19:01:40 +02:00
pipe1();
preempt();
exitwait();
2006-06-27 16:35:53 +02:00
puts("usertests finished\n");
exit();
2006-06-27 16:35:53 +02:00
}