]> git.sur5r.net Git - openldap/blob - libraries/libldap/os-local.c
01335503991606e04b8371b61768714b2eb185a1
[openldap] / libraries / libldap / os-local.c
1 /* $OpenLDAP$ */
2 /*
3  * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
4  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5  */
6 /*  Portions
7  *  Copyright (c) 1995 Regents of the University of Michigan.
8  *  All rights reserved.
9  *  Copyright (c) 1999 PADL Software Pty Ltd.
10  *  os-ip.c -- platform-specific domain socket code
11  */
12
13
14 #include "portable.h"
15
16 #ifdef LDAP_PF_LOCAL
17
18 #include <stdio.h>
19
20 #include <ac/stdlib.h>
21
22 #include <ac/errno.h>
23 #include <ac/socket.h>
24 #include <ac/string.h>
25 #include <ac/time.h>
26 #include <ac/unistd.h>
27
28 /* XXX non-portable */
29 #include <sys/stat.h>
30
31 #ifdef HAVE_IO_H
32 #include <io.h>
33 #endif /* HAVE_IO_H */
34
35 #include "ldap-int.h"
36 #include "ldap_defaults.h"
37
38 /* int ldap_int_tblsize = 0; */
39
40 #define oslocal_debug(ld,fmt,arg1,arg2,arg3) \
41 do { \
42         ldap_log_printf(ld, LDAP_DEBUG_TRACE, fmt, arg1, arg2, arg3); \
43 } while(0)
44
45 static void
46 ldap_pvt_set_errno(int err)
47 {
48         errno = err;
49 }
50
51 static int
52 ldap_pvt_ndelay_on(LDAP *ld, int fd)
53 {
54         oslocal_debug(ld, "ldap_ndelay_on: %d\n",fd,0,0);
55         return ber_pvt_socket_set_nonblock( fd, 1 );
56 }
57    
58 static int
59 ldap_pvt_ndelay_off(LDAP *ld, int fd)
60 {
61         oslocal_debug(ld, "ldap_ndelay_off: %d\n",fd,0,0);
62         return ber_pvt_socket_set_nonblock( fd, 0 );
63 }
64
65 static ber_socket_t
66 ldap_pvt_socket(LDAP *ld)
67 {
68         ber_socket_t s = socket(PF_LOCAL, SOCK_STREAM, 0);
69         oslocal_debug(ld, "ldap_new_socket: %d\n",s,0,0);
70         return ( s );
71 }
72
73 static int
74 ldap_pvt_close_socket(LDAP *ld, int s)
75 {
76         oslocal_debug(ld, "ldap_close_socket: %d\n",s,0,0);
77         return tcp_close(s);
78 }
79
80 #undef TRACE
81 #define TRACE do { \
82         oslocal_debug(ld, \
83                 "ldap_is_socket_ready: errror on socket %d: errno: %d (%s)\n", \
84                 s, \
85                 errno, \
86                 STRERROR(errno) ); \
87 } while( 0 )
88
89 /*
90  * check the socket for errors after select returned.
91  */
92 static int
93 ldap_pvt_is_socket_ready(LDAP *ld, int s)
94 {
95         oslocal_debug(ld, "ldap_is_sock_ready: %d\n",s,0,0);
96
97 #if defined( notyet ) /* && defined( SO_ERROR ) */
98 {
99         int so_errno;
100         socklen_t dummy = sizeof(so_errno);
101         if ( getsockopt( s, SOL_SOCKET, SO_ERROR, &so_errno, &dummy )
102                 == AC_SOCKET_ERROR )
103         {
104                 return -1;
105         }
106         if ( so_errno ) {
107                 ldap_pvt_set_errno(so_errno);
108                 TRACE;
109                 return -1;
110         }
111         return 0;
112 }
113 #else
114 {
115         /* error slippery */
116         struct sockaddr_un sa;
117         char ch;
118         socklen_t dummy = sizeof(sa);
119         if ( getpeername( s, (struct sockaddr *) &sa, &dummy )
120                 == AC_SOCKET_ERROR )
121         {
122                 /* XXX: needs to be replace with ber_stream_read() */
123                 read(s, &ch, 1);
124                 TRACE;
125                 return -1;
126         }
127         return 0;
128 }
129 #endif
130         return -1;
131 }
132 #undef TRACE
133
134 #if !defined(HAVE_GETPEEREID) && !defined(SO_PEERCRED) && !defined(LOCAL_PEERCRED) && defined(HAVE_SENDMSG)
135 #define DO_SENDMSG
136 #endif
137
138 static int
139 ldap_pvt_connect(LDAP *ld, ber_socket_t s, struct sockaddr_un *sa, int async)
140 {
141         int rc;
142         struct timeval  tv, *opt_tv=NULL;
143         fd_set          wfds, *z=NULL;
144
145         if ( (opt_tv = ld->ld_options.ldo_tm_net) != NULL ) {
146                 tv.tv_usec = opt_tv->tv_usec;
147                 tv.tv_sec = opt_tv->tv_sec;
148         }
149
150         oslocal_debug(ld, "ldap_connect_timeout: fd: %d tm: %ld async: %d\n",
151                         s, opt_tv ? tv.tv_sec : -1L, async);
152
153         if ( ldap_pvt_ndelay_on(ld, s) == -1 )
154                 return ( -1 );
155
156         if ( connect(s, (struct sockaddr *) sa, sizeof(struct sockaddr_un))
157                 != AC_SOCKET_ERROR )
158         {
159                 if ( ldap_pvt_ndelay_off(ld, s) == -1 ) {
160                         return ( -1 );
161                 }
162 #ifdef DO_SENDMSG
163         /* Send a dummy message with access rights. Remote side will
164          * obtain our uid/gid by fstat'ing this descriptor.
165          */
166 sendcred:        {
167                         int fds[2], rc;
168                         /* Abandon, noop, has no reply */
169                         char txt[] = {LDAP_TAG_MESSAGE, 6, LDAP_TAG_MSGID, 1, 0,  LDAP_REQ_ABANDON, 1, 0};
170                         struct iovec iov = {txt, sizeof(txt)};
171                         struct msghdr msg = {0};
172                         if (pipe(fds) == 0) {
173                                 msg.msg_iov = &iov;
174                                 msg.msg_iovlen = 1;
175                                 msg.msg_accrights = (char *)fds;
176                                 msg.msg_accrightslen = sizeof(int);
177                                 rc = sendmsg( s, &msg, 0 );
178                                 if (rc < 0) rc = errno;
179                                 close(fds[0]);
180                                 close(fds[1]);
181                         }
182                 }
183 #endif
184                 return ( 0 );
185         }
186
187         if ( errno != EINPROGRESS && errno != EWOULDBLOCK ) {
188                 return ( -1 );
189         }
190         
191 #ifdef notyet
192         if ( async ) return ( -2 );
193 #endif
194
195         FD_ZERO(&wfds);
196         FD_SET(s, &wfds );
197
198         do { 
199                 rc = select(ldap_int_tblsize, z, &wfds, z, opt_tv ? &tv : NULL);
200         } while( rc == AC_SOCKET_ERROR && errno == EINTR &&
201                 LDAP_BOOL_GET(&ld->ld_options, LDAP_BOOL_RESTART ));
202
203         if( rc == AC_SOCKET_ERROR ) return rc;
204
205         if ( FD_ISSET(s, &wfds) ) {
206                 if ( ldap_pvt_is_socket_ready(ld, s) == -1 )
207                         return ( -1 );
208                 if ( ldap_pvt_ndelay_off(ld, s) == -1 )
209                         return ( -1 );
210 #ifdef DO_SENDMSG
211                 goto sendcred;
212 #else
213                 return ( 0 );
214 #endif
215         }
216         oslocal_debug(ld, "ldap_connect_timeout: timed out\n",0,0,0);
217         ldap_pvt_set_errno( ETIMEDOUT );
218         return ( -1 );
219 }
220
221 int
222 ldap_connect_to_path(LDAP *ld, Sockbuf *sb, const char *path, int async)
223 {
224         struct sockaddr_un      server;
225         ber_socket_t            s;
226         int                     rc;
227
228         oslocal_debug(ld, "ldap_connect_to_path\n",0,0,0);
229
230         s = ldap_pvt_socket( ld );
231         if ( s == AC_SOCKET_INVALID ) {
232                 return -1;
233         }
234
235         if ( path == NULL || path[0] == '\0' ) {
236                 path = LDAPI_SOCK;
237         } else {
238                 if ( strlen(path) > (sizeof( server.sun_path ) - 1) ) {
239                         ldap_pvt_set_errno( ENAMETOOLONG );
240                         return -1;
241                 }
242         }
243
244         oslocal_debug(ld, "ldap_connect_to_path: Trying %s\n", path, 0, 0);
245
246         memset( &server, '\0', sizeof(server) );
247         server.sun_family = AF_LOCAL;
248         strcpy( server.sun_path, path );
249
250         rc = ldap_pvt_connect(ld, s, &server, async);
251
252         if (rc == 0) {
253                 ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_FD, (void *)&s );
254         } else {
255                 ldap_pvt_close_socket(ld, s);
256         }
257         return rc;
258 }
259 #else
260 static int dummy;
261 #endif /* LDAP_PF_LOCAL */