1 /*	$OpenBSD: uthread_connect.c,v 1.5 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_connect.c,v 1.6 1999/08/28 00:03:28 peter Exp $
34  */
35 #include <errno.h>
36 #include <sys/types.h>
37 #include <sys/socket.h>
38 #include <fcntl.h>
39 #ifdef _THREAD_SAFE
40 #include <pthread.h>
41 #include "pthread_private.h"
42 
43 int
connect(int fd,const struct sockaddr * name,socklen_t namelen)44 connect(int fd, const struct sockaddr * name, socklen_t namelen)
45 {
46 	struct pthread	*curthread = _get_curthread();
47 	struct sockaddr tmpname;
48 	int             errnolen, ret, tmpnamelen;
49 
50 	/* This is a cancellation point: */
51 	_thread_enter_cancellation_point();
52 
53 	if ((ret = _FD_LOCK(fd, FD_RDWR, NULL)) == 0) {
54 		if ((ret = _thread_sys_connect(fd, name, namelen)) < 0) {
55 			if (!(_thread_fd_table[fd]->flags & O_NONBLOCK) &&
56 			((errno == EWOULDBLOCK) || (errno == EINPROGRESS) ||
57 			 (errno == EALREADY) || (errno == EAGAIN))) {
58 				curthread->data.fd.fd = fd;
59 
60 				/* Set the timeout: */
61 				_thread_kern_set_timeout(NULL);
62 				_thread_kern_sched_state(PS_FDW_WAIT, __FILE__, __LINE__);
63 
64 				tmpnamelen = sizeof(tmpname);
65 				/* 0 now lets see if it really worked */
66 				if (((ret = _thread_sys_getpeername(fd, &tmpname, &tmpnamelen)) < 0) && (errno == ENOTCONN)) {
67 
68 					/*
69 					 * Get the error, this function
70 					 * should not fail
71 					 */
72 					errnolen = sizeof(errno);
73 					_thread_sys_getsockopt(fd, SOL_SOCKET, SO_ERROR, &errno, &errnolen);
74 				}
75 			} else {
76 				ret = -1;
77 			}
78 		}
79 		_FD_UNLOCK(fd, FD_RDWR);
80 	}
81 
82 	/* No longer in a cancellation point: */
83 	_thread_leave_cancellation_point();
84 
85 	return (ret);
86 }
87 #endif
88