2 * Amanda, The Advanced Maryland Automatic Network Disk Archiver
3 * Copyright (c) 1991-1999 University of Maryland at College Park
6 * Permission to use, copy, modify, distribute, and sell this software and its
7 * documentation for any purpose is hereby granted without fee, provided that
8 * the above copyright notice appear in all copies and that both that
9 * copyright notice and this permission notice appear in supporting
10 * documentation, and that the name of U.M. not be used in advertising or
11 * publicity pertaining to distribution of the software without specific,
12 * written prior permission. U.M. makes no representations about the
13 * suitability of this software for any purpose. It is provided "as is"
14 * without express or implied warranty.
16 * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
18 * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
20 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
21 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
23 * Author: James da Silva, Systems Design and Analysis Group
24 * Computer Science Department
25 * University of Maryland at College Park
28 * $Id: dgram.c,v 1.32 2006/07/05 19:54:20 martinea Exp $
30 * library routines to marshall/send, recv/unmarshall UDP packets
42 if(socket < 0 || socket >= FD_SETSIZE) {
43 error("dgram_socket: socket %d out of range (0 .. %d)\n",
44 socket, FD_SETSIZE-1);
47 dgram->socket = socket;
58 struct sockaddr_in name;
60 #if defined(USE_REUSEADDR)
65 *portp = (in_port_t)0;
66 if((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
68 dbprintf(("%s: dgram_bind: socket() failed: %s\n",
70 strerror(save_errno)));
74 if(s < 0 || s >= (int)FD_SETSIZE) {
75 dbprintf(("%s: dgram_bind: socket out of range: %d\n",
79 errno = EMFILE; /* out of range */
83 memset(&name, 0, SIZEOF(name));
84 name.sin_family = (sa_family_t)AF_INET;
85 name.sin_addr.s_addr = INADDR_ANY;
88 r = setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
89 (void *)&on, (socklen_t)sizeof(on));
91 dbprintf(("%s: dgram_bind: setsockopt(SO_REUSEADDR) failed: %s\n",
98 * If a port range was specified, we try to get a port in that
99 * range first. Next, we try to get a reserved port. If that
100 * fails, we just go for any port.
102 * In all cases, not to use port that's assigned to other services.
104 * It is up to the caller to make sure we have the proper permissions
105 * to get the desired port, and to make sure we return a port that
106 * is within the range it requires.
108 for (retries = 0; ; retries++) {
110 if (bind_portrange(s, &name, UDPPORTRANGE, "udp") == 0)
112 dbprintf(("%s: dgram_bind: Could to bind to port in range: %d - %d.\n",
113 debug_prefix(NULL), UDPPORTRANGE));
116 if (bind_portrange(s, &name, (in_port_t)512,
117 (in_port_t)(IPPORT_RESERVED - 1), "udp") == 0)
119 dbprintf(("%s: dgram_bind: Could to bind to port in range: 512 - %d.\n",
120 debug_prefix(NULL), IPPORT_RESERVED - 1));
122 name.sin_port = INADDR_ANY;
123 if (bind(s, (struct sockaddr *)&name, (socklen_t)sizeof(name)) == 0)
125 dbprintf(("%s: dgram_bind: Could to bind to any port: %s\n",
126 debug_prefix(NULL), strerror(errno)));
128 if (retries >= BIND_CYCLE_RETRIES) {
129 dbprintf(("%s: dgram_bind: Giving up...\n", debug_prefix(NULL)));
133 dbprintf(("%s: dgram_bind: Retrying entire range after 10 second delay.\n",
134 debug_prefix(NULL)));
139 dbprintf(("%s: dgram_bind: bind(INADDR_ANY) failed: %s\n",
141 strerror(save_errno)));
147 /* find out what name was actually used */
149 len = (socklen_t)sizeof(name);
150 if(getsockname(s, (struct sockaddr *)&name, &len) == -1) {
152 dbprintf(("%s: dgram_bind: getsockname() failed: %s\n",
154 strerror(save_errno)));
159 *portp = (in_port_t)ntohs(name.sin_port);
162 dbprintf(("%s: dgram_bind: socket bound to %s.%d\n",
163 debug_prefix_time(NULL),
164 inet_ntoa(name.sin_addr),
172 struct sockaddr_in addr,
177 struct sockaddr_in addr_save;
181 #if defined(USE_REUSEADDR)
186 dbprintf(("%s: dgram_send_addr(addr=%p, dgram=%p)\n",
187 debug_prefix(NULL), &addr, dgram));
188 dump_sockaddr(&addr);
189 dbprintf(("%s: dgram_send_addr: %p->socket = %d\n",
190 debug_prefix(NULL), dgram, dgram->socket));
191 if(dgram->socket != -1) {
195 if((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
197 dbprintf(("%s: dgram_send_addr: socket() failed: %s\n",
199 strerror(save_errno)));
205 r = setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
206 (void *)&on, (socklen_t)sizeof(on));
208 dbprintf(("%s: dgram_send_addr: setsockopt(SO_REUSEADDR) failed: %s\n",
215 memcpy(&addr_save, &addr, SIZEOF(addr));
216 if(s < 0 || s >= FD_SETSIZE) {
217 dbprintf(("%s: dgram_send_addr: socket out of range: %d\n",
220 errno = EMFILE; /* out of range */
223 max_wait = 300 / 5; /* five minutes */
230 (struct sockaddr *)&addr,
231 (int)sizeof(struct sockaddr_in)) == -1) {
233 if(errno == ECONNREFUSED && wait_count++ < max_wait) {
235 dbprintf(("%s: dgram_send_addr: sendto(%s.%hu): retry %d after ECONNREFUSED\n",
236 debug_prefix_time(NULL),
237 inet_ntoa(addr_save.sin_addr),
238 (in_port_t)ntohs(addr.sin_port),
244 if(errno == EAGAIN && wait_count++ < max_wait) {
246 dbprintf(("%s: dgram_send_addr: sendto(%s.%hu): retry %d after EAGAIN\n",
247 debug_prefix_time(NULL),
248 inet_ntoa(addr_save.sin_addr),
249 (in_port_t)ntohs(addr.sin_port),
255 dbprintf(("%s: dgram_send_addr: sendto(%s.%d) failed: %s \n",
256 debug_prefix_time(NULL),
257 inet_ntoa(addr_save.sin_addr),
258 (int) ntohs(addr.sin_port),
259 strerror(save_errno)));
269 dbprintf(("%s: dgram_send_addr: close(%s.%d): failed: %s\n",
271 inet_ntoa(addr_save.sin_addr),
272 (int) ntohs(addr.sin_port),
275 * Calling function should not care that the close failed.
276 * It does care about the send results though.
292 struct sockaddr_in name;
296 if((hp = gethostbyname(hostname)) == 0) {
298 dbprintf(("%s: dgram_send: gethostbyname(%s) failed\n",
299 debug_prefix_time(NULL),
304 memcpy(&name.sin_addr, hp->h_addr, (size_t)hp->h_length);
305 name.sin_family = (sa_family_t)AF_INET;
306 name.sin_port = (in_port_t)htons(port);
308 return dgram_send_addr(name, dgram);
316 struct sockaddr_in *fromaddr)
318 SELECT_ARG_TYPE ready;
326 sock = dgram->socket;
329 FD_SET(sock, &ready);
333 dbprintf(("%s: dgram_recv(dgram=%p, timeout=%u, fromaddr=%p)\n",
334 debug_prefix_time(NULL), timeout, fromaddr));
335 dump_sockaddr(fromaddr);
337 nfound = (ssize_t)select(sock+1, &ready, NULL, NULL, &to);
338 if(nfound <= 0 || !FD_ISSET(sock, &ready)) {
341 dbprintf(("%s: dgram_recv: select() failed: %s\n",
342 debug_prefix_time(NULL),
343 strerror(save_errno)));
344 } else if(nfound == 0) {
345 dbprintf(("%s: dgram_recv: timeout after %d second%s\n",
346 debug_prefix_time(NULL),
348 (timeout == 1) ? "" : "s"));
350 } else if (!FD_ISSET(sock, &ready)) {
353 for(i = 0; i < sock + 1; i++) {
354 if(FD_ISSET(i, &ready)) {
355 dbprintf(("%s: dgram_recv: got fd %d instead of %d\n",
356 debug_prefix_time(NULL),
368 addrlen = (socklen_t)sizeof(struct sockaddr_in);
369 size = recvfrom(sock, dgram->data, (size_t)MAX_DGRAM, 0,
370 (struct sockaddr *)fromaddr, &addrlen);
373 dbprintf(("%s: dgram_recv: recvfrom() failed: %s\n",
375 strerror(save_errno)));
379 dgram->len = (size_t)size;
380 dgram->data[size] = '\0';
381 dgram->cur = dgram->data;
390 dgram->cur = dgram->data;
392 *(dgram->cur) = '\0';
395 printf_arglist_function1(int dgram_cat, dgram_t *, dgram, const char *, fmt)
401 assert(dgram != NULL);
404 assert(dgram->len == (size_t)(dgram->cur - dgram->data));
405 assert(dgram->len < SIZEOF(dgram->data));
407 bufsize = (ssize_t)(sizeof(dgram->data) - dgram->len);
411 arglist_start(argp, fmt);
412 len = vsnprintf(dgram->cur, (size_t)bufsize, fmt, argp);
414 if((ssize_t)len > bufsize) {
415 dgram->len = sizeof(dgram->data);
416 dgram->cur = dgram->data + dgram->len;
420 arglist_start(argp, fmt);
421 dgram->len += vsnprintf(dgram->cur, (size_t)bufsize, fmt, argp);
423 dgram->cur = dgram->data + dgram->len;
432 char *p = dgram->cur;
433 char *end = dgram->data + dgram->len;
435 while(p < end && *p && *p != '\n')