xref: /onnv-gate/usr/src/lib/libwanboot/common/socket_inet.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  * Copyright 2002-2003 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  * Use is subject to license terms.
25*0Sstevel@tonic-gate  */
26*0Sstevel@tonic-gate 
27*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
28*0Sstevel@tonic-gate 
29*0Sstevel@tonic-gate #include <sys/types.h>
30*0Sstevel@tonic-gate #include <sys/socket.h>
31*0Sstevel@tonic-gate #include <netinet/in.h>
32*0Sstevel@tonic-gate #include <unistd.h>
33*0Sstevel@tonic-gate #include <poll.h>
34*0Sstevel@tonic-gate #include <errno.h>
35*0Sstevel@tonic-gate 
36*0Sstevel@tonic-gate #include <socket_inet.h>
37*0Sstevel@tonic-gate 
38*0Sstevel@tonic-gate /*
39*0Sstevel@tonic-gate  * Name:	socket_read
40*0Sstevel@tonic-gate  * Description:	Use recv in non-secure sockets.
41*0Sstevel@tonic-gate  * Scope:	private
42*0Sstevel@tonic-gate  * Arguments:	fildes		- Socket file descriptor.
43*0Sstevel@tonic-gate  *		buf		- Buffer to read data into.
44*0Sstevel@tonic-gate  *		nbyte		- Number of bytes to read.
45*0Sstevel@tonic-gate  *              read_timeout    - Timeout value in seconds.
46*0Sstevel@tonic-gate  * Returns:	n		- Number of bytes read. -1 on error.
47*0Sstevel@tonic-gate  */
48*0Sstevel@tonic-gate int
socket_read(int fildes,void * buf,size_t nbyte,int read_timeout)49*0Sstevel@tonic-gate socket_read(int fildes, void *buf, size_t nbyte, int read_timeout)
50*0Sstevel@tonic-gate {
51*0Sstevel@tonic-gate 	struct pollfd pfd;
52*0Sstevel@tonic-gate 
53*0Sstevel@tonic-gate 	pfd.fd = fildes;
54*0Sstevel@tonic-gate 	pfd.events = POLLIN;
55*0Sstevel@tonic-gate 
56*0Sstevel@tonic-gate 	switch (poll(&pfd, 1, read_timeout * 1000)) {
57*0Sstevel@tonic-gate 	case 0:
58*0Sstevel@tonic-gate 		errno = EINTR;
59*0Sstevel@tonic-gate 		return (-1);
60*0Sstevel@tonic-gate 	case -1:
61*0Sstevel@tonic-gate 		return (-1);
62*0Sstevel@tonic-gate 	default:
63*0Sstevel@tonic-gate 		break;
64*0Sstevel@tonic-gate 	}
65*0Sstevel@tonic-gate 
66*0Sstevel@tonic-gate 	return (recv(fildes, buf, nbyte, 0));
67*0Sstevel@tonic-gate }
68*0Sstevel@tonic-gate 
69*0Sstevel@tonic-gate /*
70*0Sstevel@tonic-gate  * Name:	socket_write
71*0Sstevel@tonic-gate  * Description:	Use sendto for non-secure connections.
72*0Sstevel@tonic-gate  * Scope:	private
73*0Sstevel@tonic-gate  * Arguments:	fildes		- Socket file descriptor.
74*0Sstevel@tonic-gate  *		buf		- Buffer containing data to be written.
75*0Sstevel@tonic-gate  *		nbyte		- Number of bytes to write.
76*0Sstevel@tonic-gate  *              addr            - Connection address
77*0Sstevel@tonic-gate  * Returns:	n		- Number of bytes written. -1 on error.
78*0Sstevel@tonic-gate  */
79*0Sstevel@tonic-gate int
socket_write(int fildes,const void * buf,size_t nbyte,struct sockaddr_in * addr)80*0Sstevel@tonic-gate socket_write(int fildes, const void *buf, size_t nbyte,
81*0Sstevel@tonic-gate     struct sockaddr_in *addr)
82*0Sstevel@tonic-gate {
83*0Sstevel@tonic-gate 	return (sendto(fildes, buf, nbyte, 0, (struct sockaddr *)addr,
84*0Sstevel@tonic-gate 	    sizeof (*addr)));
85*0Sstevel@tonic-gate }
86*0Sstevel@tonic-gate 
87*0Sstevel@tonic-gate int
socket_close(int fildes)88*0Sstevel@tonic-gate socket_close(int fildes)
89*0Sstevel@tonic-gate {
90*0Sstevel@tonic-gate 	return (close(fildes));
91*0Sstevel@tonic-gate }
92