/* ==== test_sock_1.c ========================================================= * Copyright (c) 1993 by Chris Provenzano, proven@athena.mit.edu * * Description : Test pthread_create() and pthread_exit() calls. * * 1.00 93/08/03 proven * -Started coding this file. */ #include #include #include #include #include #include #include #include #include #include #include "test.h" struct sockaddr_in a_sout; #define MESSAGE5 "This should be message #5" #define MESSAGE6 "This should be message #6" void * sock_write(arg) void *arg; { int fd = *(int *)arg; SET_NAME("writer"); CHECKe(write(fd, MESSAGE5, sizeof(MESSAGE5))); return(NULL); } static pthread_mutex_t waiter_mutex = PTHREAD_MUTEX_INITIALIZER; void* waiter(sig) { int status; pid_t pid; SET_NAME("waiter"); CHECKr(pthread_mutex_lock(&waiter_mutex)); printf("waiting for child\n"); CHECKe(pid = wait(&status)); ASSERT(WIFEXITED(status)); ASSERT(WEXITSTATUS(status) == 0); printf("child exited\n"); CHECKr(pthread_mutex_unlock(&waiter_mutex)); return (NULL); } void * sock_accept(arg) void *arg; { pthread_t thread, wthread; struct sockaddr a_sin; int a_sin_size, a_fd, fd; u_int16_t port; char buf[1024]; pid_t pid; port = 3276; a_sout.sin_family = AF_INET; a_sout.sin_port = htons(port); a_sout.sin_addr.s_addr = INADDR_ANY; CHECKe(a_fd = socket(AF_INET, SOCK_STREAM, 0)); while(1) { if (bind(a_fd, (struct sockaddr *)&a_sout, sizeof(a_sout))==0) break; if (errno == EADDRINUSE) { a_sout.sin_port = htons((++port)); continue; } DIE(errno, "bind"); } printf("listening on port %d\n", port); CHECKe(listen(a_fd, 2)); printf("%d: This should be message #1\n", getpid()); CHECKr(pthread_mutex_init(&waiter_mutex, NULL)); CHECKr(pthread_mutex_lock(&waiter_mutex)); CHECKr(pthread_create(&wthread, NULL, waiter, NULL)); CHECKe(pid = fork()); switch(pid) { case 0: execl("test_sock_2a", "test_sock_2a", "fork okay", NULL); DIE(errno, "execl"); default: break; } CHECKr(pthread_mutex_unlock(&waiter_mutex)); pthread_yield(); a_sin_size = sizeof(a_sin); CHECKe(fd = accept(a_fd, &a_sin, &a_sin_size)); CHECKe(close(fd)); sleep(1); printf("%d: This should be message #4\n", getpid()); a_sin_size = sizeof(a_sin); memset(&a_sin, 0, sizeof(a_sin)); CHECKe(fd = accept(a_fd, &a_sin, &a_sin_size)); /* Setup a write thread */ CHECKr(pthread_create(&thread, NULL, sock_write, &fd)); CHECKe(read(fd, buf, 1024)); printf("%d: %s\n", getpid(), buf); /* message 6 */ CHECKe(close(fd)); if (pthread_mutex_trylock(&waiter_mutex) == EBUSY) { sleep(2); if (pthread_mutex_trylock(&waiter_mutex) == EBUSY) { /* forcibly kill child */ CHECKe(kill(pid, SIGKILL)); PANIC("child %d took too long to exit", pid); } } CHECKr(pthread_join(wthread, NULL)); return(NULL); } int main() { pthread_t thread; setbuf(stdout, NULL); setbuf(stderr, NULL); CHECKr(pthread_create(&thread, NULL, sock_accept, (void *)0xdeadbeaf)); CHECKr(pthread_join(thread, NULL)); SUCCEED; }