1 /*	$OpenBSD: uthread_accept.c,v 1.8 2003/12/23 19:31:05 brad Exp $	*/
2 /*
3  * Copyright (c) 1995-1998 John Birrell <jb@cimlogic.com.au>
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. All advertising materials mentioning features or use of this software
15  *    must display the following acknowledgement:
16  *	This product includes software developed by John Birrell.
17  * 4. Neither the name of the author nor the names of any co-contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY JOHN BIRRELL AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: uthread_accept.c,v 1.9 1999/08/28 00:03:20 peter Exp $
34  */
35 #include <errno.h>
36 #include <unistd.h>
37 #include <sys/types.h>
38 #include <sys/socket.h>
39 #include <fcntl.h>
40 #ifdef _THREAD_SAFE
41 #include <pthread.h>
42 #include "pthread_private.h"
43 
44 int
accept(int fd,struct sockaddr * name,socklen_t * namelen)45 accept(int fd, struct sockaddr * name, socklen_t *namelen)
46 {
47 	struct pthread	*curthread = _get_curthread();
48 	int             ret;
49 
50 	/* This is a cancellation point: */
51 	_thread_enter_cancellation_point();
52 
53 	/* Lock the file descriptor: */
54 	if ((ret = _FD_LOCK(fd, FD_RDWR, NULL)) == 0) {
55 		/* Enter a loop to wait for a connection request: */
56 		while ((ret = _thread_sys_accept(fd, name, namelen)) < 0) {
57 			/* Check if the socket is to block: */
58 			if ((_thread_fd_table[fd]->flags & O_NONBLOCK) == 0 &&
59 			    (errno == EWOULDBLOCK || errno == EAGAIN)) {
60 				/* Save the socket file descriptor: */
61 				curthread->data.fd.fd = fd;
62 				curthread->data.fd.fname = __FILE__;
63 				curthread->data.fd.branch = __LINE__;
64 
65 				/* Set the timeout: */
66 				_thread_kern_set_timeout(NULL);
67 				curthread->interrupted = 0;
68 
69 				/* Schedule the next thread: */
70 				_thread_kern_sched_state(PS_FDR_WAIT, __FILE__,
71 							 __LINE__);
72 
73 				/* Check if the wait was interrupted: */
74 				if (curthread->interrupted) {
75 					/* Return an error status: */
76 					errno = EINTR;
77 					ret = -1;
78 					break;
79 				}
80 			} else {
81 				/*
82 				 * Another error has occurred, so exit the
83 				 * loop here:
84 				 */
85 				break;
86 			}
87 		}
88 
89 		/*
90 		 * If no errors initialize the file descriptor table
91 		 * for the new socket.   Turn on blocking mode in the
92 		 * child if it is on in the parent.   This is done
93 		 * as _thread_fd_table_init *may* have turned the flag on.
94 		 */
95 		if (ret != -1) {
96 			if (_thread_fd_table_init(ret) != 0) {
97 				/* Quietly close the socket: */
98 				_thread_sys_close(ret);
99 				ret = -1;
100 			} else if ((_thread_fd_table[fd]->flags & O_NONBLOCK) == 0)
101 				_thread_fd_table[ret]->flags &= ~O_NONBLOCK;
102 		}
103 
104 		/* Unlock the file descriptor: */
105 		_FD_UNLOCK(fd, FD_RDWR);
106 	}
107 
108 	/* No longer in a cancellation point: */
109 	_thread_leave_cancellation_point();
110 
111 	/* Return the socket file descriptor or -1 on error: */
112 	return (ret);
113 }
114 #endif
115