09dadc09a3b074d16bed28a262c44187c1cf3015
[sip-router] / pass_fd.c
1 /*
2  * $Id$
3  *
4  * Copyright (C) 2001-2003 Fhg Fokus
5  *
6  * This file is part of ser, a free SIP server.
7  *
8  * ser is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version
12  *
13  * For a license to use the ser software under conditions
14  * other than those described here, or to purchase support for this
15  * software, please contact iptel.org by e-mail at the following addresses:
16  *    info@iptel.org
17  *
18  * ser is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
26  */
27  /*
28   * History:
29   * --------
30   *  2002-11-29  created by andrei
31   *  2003-02-20  added solaris support (! HAVE_MSGHDR_MSG_CONTROL) (andrei)
32   *  2003-11-03  added send_all, recv_all  and updated send/get_fd
33   *               to handle signals  (andrei)
34   */
35
36 #ifdef USE_TCP
37
38 #include <sys/types.h>
39 #include <sys/socket.h>
40 #include <sys/uio.h>
41 #include <stdlib.h> /* for NULL definition on openbsd */
42 #include <errno.h>
43 #include <string.h>
44
45 #include "dprint.h"
46
47
48
49 /* receive all the data or returns error (handles EINTR etc.)
50  * returns: bytes read or error (<0)
51  * can return < data_len if EOF */
52 int recv_all(int socket, void* data, int data_len)
53 {
54         int b_read;
55         int n;
56         
57         b_read=0;
58         do{
59                 n=recv(socket, (char*)data+b_read, data_len-b_read, MSG_WAITALL);
60                 if (n<0){
61                         /* error */
62                         if (errno==EINTR) continue; /* signal, try again */
63                         LOG(L_CRIT, "ERROR: recv_all: recv on %d failed: %s\n",
64                                         socket, strerror(errno));
65                         return n;
66                 }
67                 b_read+=n;
68         }while( (b_read!=data_len) && (n));
69         return b_read;
70 }
71
72
73 /* sends all data (takes care of signals) (assumes blocking fd)
74  * returns number of bytes sent or < 0 for an error */
75 int send_all(int socket, void* data, int data_len)
76 {
77         int n;
78         
79 again:
80         n=send(socket, data, data_len, 0);
81         if (n<0){
82                         /* error */
83                 if (errno==EINTR) goto again; /* signal, try again */
84                 LOG(L_CRIT, "ERROR: send_all: send on %d failed: %s\n",
85                                         socket, strerror(errno));
86         }
87         return n;
88 }
89
90
91 /* at least 1 byte must be sent! */
92 int send_fd(int unix_socket, void* data, int data_len, int fd)
93 {
94         struct msghdr msg;
95         struct iovec iov[1];
96         int ret;
97 #ifdef HAVE_MSGHDR_MSG_CONTROL
98         struct cmsghdr* cmsg;
99         union {
100                 struct cmsghdr cm;
101                 char control[CMSG_SPACE(sizeof(fd))];
102         }control_un;
103         
104         msg.msg_control=control_un.control;
105         msg.msg_controllen=sizeof(control_un.control);
106         
107         cmsg=CMSG_FIRSTHDR(&msg);
108         cmsg->cmsg_level = SOL_SOCKET;
109         cmsg->cmsg_type = SCM_RIGHTS;
110         cmsg->cmsg_len = CMSG_LEN(sizeof(fd));
111         *(int*)CMSG_DATA(cmsg)=fd;
112         msg.msg_flags=0;
113 #else
114         msg.msg_accrights=(caddr_t) &fd;
115         msg.msg_accrightslen=sizeof(fd);
116 #endif
117         
118         msg.msg_name=0;
119         msg.msg_namelen=0;
120         
121         iov[0].iov_base=data;
122         iov[0].iov_len=data_len;
123         msg.msg_iov=iov;
124         msg.msg_iovlen=1;
125         
126 again:
127         ret=sendmsg(unix_socket, &msg, 0);
128         if (ret<0){
129                 if (errno==EINTR) goto again;
130                 LOG(L_CRIT, "ERROR: send_fd: sendmsg failed on %d: %s\n",
131                                 unix_socket, strerror(errno));
132         }
133         
134         return ret;
135 }
136
137
138
139 int receive_fd(int unix_socket, void* data, int data_len, int* fd)
140 {
141         struct msghdr msg;
142         struct iovec iov[1];
143         int new_fd;
144         int ret;
145         int n;
146 #ifdef HAVE_MSGHDR_MSG_CONTROL
147         struct cmsghdr* cmsg;
148         union{
149                 struct cmsghdr cm;
150                 char control[CMSG_SPACE(sizeof(new_fd))];
151         }control_un;
152         
153         msg.msg_control=control_un.control;
154         msg.msg_controllen=sizeof(control_un.control);
155 #else
156         msg.msg_accrights=(caddr_t) &new_fd;
157         msg.msg_accrightslen=sizeof(int);
158 #endif
159         
160         msg.msg_name=0;
161         msg.msg_namelen=0;
162         
163         iov[0].iov_base=data;
164         iov[0].iov_len=data_len;
165         msg.msg_iov=iov;
166         msg.msg_iovlen=1;
167         
168 again:
169         ret=recvmsg(unix_socket, &msg, MSG_WAITALL);
170         if (ret<0){
171                 if (errno==EINTR) goto again;
172                 LOG(L_CRIT, "ERROR: receive_fd: recvmsg on %d failed: %s\n",
173                                 unix_socket, strerror(errno));
174                 goto error;
175         }
176         if (ret==0){
177                 /* EOF */
178                 LOG(L_CRIT, "ERROR: receive_fd: EOF on %d\n", unix_socket);
179                 goto error;
180         }
181         if (ret<data_len){
182                 LOG(L_WARN, "WARNING: receive_fd: too few bytes read (%d from %d)"
183                                     "trying to fix...\n", ret, data_len);
184                 n=recv_all(unix_socket, (char*)data+ret, data_len-ret);
185                 if (n>=0) ret+=n;
186                 else{
187                         ret=n;
188                         goto error;
189                 }
190         }
191         
192 #ifdef HAVE_MSGHDR_MSG_CONTROL
193         cmsg=CMSG_FIRSTHDR(&msg);
194         if ((cmsg!=0) && (cmsg->cmsg_len==CMSG_LEN(sizeof(new_fd)))){
195                 if (cmsg->cmsg_type!= SCM_RIGHTS){
196                         LOG(L_ERR, "ERROR: receive_fd: msg control type != SCM_RIGHTS\n");
197                         ret=-1;
198                         goto error;
199                 }
200                 if (cmsg->cmsg_level!= SOL_SOCKET){
201                         LOG(L_ERR, "ERROR: receive_fd: msg level != SOL_SOCKET\n");
202                         ret=-1;
203                         goto error;
204                 }
205                 *fd=*((int*) CMSG_DATA(cmsg));
206         }else{
207                 LOG(L_ERR, "ERROR: receive_fd: no descriptor passed, cmsg=%p,"
208                                 "len=%d\n", cmsg, cmsg->cmsg_len);
209                 *fd=-1;
210                 /* it's not really an error */
211         }
212 #else
213         if (msg.msg_accrightslen==sizeof(int)){
214                 *fd=new_fd;
215         }else{
216                 LOG(L_ERR, "ERROR: receive_fd: no descriptor passed,"
217                                 " accrightslen=%d\n", msg.msg_accrightslen);
218                 *fd=-1;
219         }
220 #endif
221         
222 error:
223         return ret;
224 }
225 #endif