1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2011 Nexenta Systems, Inc.  All rights reserved.
24  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
25  * Copyright (c) 2014, Joyent, Inc. All rights reserved.
26  */
27 
28 #include <sys/file.h>
29 #include <sys/stropts.h>
30 #include <sys/socket.h>
31 #include <sys/socketvar.h>
32 #include <sys/sysmacros.h>
33 #include <sys/filio.h>		/* FIO* ioctls */
34 #include <sys/sockio.h>		/* SIOC* ioctls */
35 #include <sys/poll_impl.h>
36 #include <sys/cmn_err.h>
37 #include <sys/ksocket.h>
38 #include <io/ksocket/ksocket_impl.h>
39 #include <fs/sockfs/sockcommon.h>
40 
41 #define	SOCKETMOD_TCP	"tcp"
42 #define	SOCKETMOD_UDP	"udp"
43 /*
44  * Kernel Sockets
45  *
46  * Mostly a wrapper around the private socket_* functions.
47  */
48 int
49 ksocket_socket(ksocket_t *ksp, int domain, int type, int protocol, int flags,
50     struct cred *cr)
51 {
52 	static const int version = SOV_DEFAULT;
53 	int error = 0;
54 	struct sonode *so;
55 	*ksp = NULL;
56 
57 	/* All Solaris components should pass a cred for this operation. */
58 	ASSERT(cr != NULL);
59 
60 	if (domain == AF_NCA)
61 		return (EAFNOSUPPORT);
62 
63 	ASSERT(flags == KSOCKET_SLEEP || flags == KSOCKET_NOSLEEP);
64 	so = socket_create(domain, type, protocol, NULL, NULL, version, flags,
65 	    cr, &error);
66 	if (so == NULL) {
67 		if (error == EAFNOSUPPORT) {
68 			char *mod = NULL;
69 
70 			/*
71 			 * Could be that root file system is not loaded or
72 			 * soconfig has not run yet.
73 			 */
74 			if (type == SOCK_STREAM && (domain == AF_INET ||
75 			    domain == AF_INET6) && (protocol == 0 ||
76 			    protocol == IPPROTO_TCP)) {
77 					mod = SOCKETMOD_TCP;
78 			} else if (type == SOCK_DGRAM && (domain == AF_INET ||
79 			    domain == AF_INET6) && (protocol == 0 ||
80 			    protocol == IPPROTO_UDP)) {
81 					mod = SOCKETMOD_UDP;
82 			} else {
83 				return (EAFNOSUPPORT);
84 			}
85 
86 			so = socket_create(domain, type, protocol, NULL,
87 			    mod, version, flags, cr, &error);
88 			if (so == NULL)
89 				return (error);
90 		} else {
91 			return (error);
92 		}
93 	}
94 
95 	so->so_mode |= SM_KERNEL;
96 
97 	*ksp = SOTOKS(so);
98 
99 	return (0);
100 }
101 int
102 ksocket_bind(ksocket_t ks, struct sockaddr *addr, socklen_t addrlen,
103     struct cred *cr)
104 {
105 	int error;
106 
107 	/* All Solaris components should pass a cred for this operation. */
108 	ASSERT(cr != NULL);
109 
110 	if (!KSOCKET_VALID(ks))
111 		return (ENOTSOCK);
112 
113 	error = socket_bind(KSTOSO(ks), addr, addrlen, _SOBIND_SOCKBSD, cr);
114 
115 	return (error);
116 }
117 
118 int
119 ksocket_listen(ksocket_t ks, int backlog, struct cred *cr)
120 {
121 	/* All Solaris components should pass a cred for this operation. */
122 	ASSERT(cr != NULL);
123 
124 	if (!KSOCKET_VALID(ks))
125 		return (ENOTSOCK);
126 
127 	return (socket_listen(KSTOSO(ks), backlog, cr));
128 }
129 
130 int
131 ksocket_accept(ksocket_t ks, struct sockaddr *addr,
132     socklen_t *addrlenp, ksocket_t *nks, struct cred *cr)
133 {
134 	int error;
135 	struct sonode *nso = NULL;
136 
137 	/* All Solaris components should pass a cred for this operation. */
138 	ASSERT(cr != NULL);
139 
140 	*nks = NULL;
141 
142 	if (!KSOCKET_VALID(ks))
143 		return (ENOTSOCK);
144 
145 	if (addr != NULL && addrlenp == NULL)
146 		return (EFAULT);
147 
148 	error = socket_accept(KSTOSO(ks), KSOCKET_FMODE(ks), cr, &nso);
149 	if (error != 0)
150 		return (error);
151 
152 	ASSERT(nso != NULL);
153 
154 	nso->so_mode |= SM_KERNEL;
155 
156 	if (addr != NULL && addrlenp != NULL) {
157 		error = socket_getpeername(nso, addr, addrlenp, B_TRUE, cr);
158 		if (error != 0) {
159 			(void) socket_close(nso, 0, cr);
160 			socket_destroy(nso);
161 			return ((error == ENOTCONN) ? ECONNABORTED : error);
162 		}
163 	}
164 
165 	*nks = SOTOKS(nso);
166 
167 	return (error);
168 }
169 
170 int
171 ksocket_connect(ksocket_t ks, struct sockaddr *addr, socklen_t addrlen,
172     struct cred *cr)
173 {
174 	/* All Solaris components should pass a cred for this operation. */
175 	ASSERT(cr != NULL);
176 
177 	if (!KSOCKET_VALID(ks))
178 		return (ENOTSOCK);
179 
180 	return (socket_connect(KSTOSO(ks), addr, addrlen,
181 	    KSOCKET_FMODE(ks), 0, cr));
182 }
183 
184 int
185 ksocket_send(ksocket_t ks, void *msg, size_t msglen, int flags,
186     size_t *sent, struct cred *cr)
187 {
188 	int error;
189 	struct nmsghdr msghdr;
190 	struct uio auio;
191 	struct iovec iov;
192 
193 	/* All Solaris components should pass a cred for this operation. */
194 	ASSERT(cr != NULL);
195 
196 	if (!KSOCKET_VALID(ks)) {
197 		if (sent != NULL)
198 			*sent = 0;
199 		return (ENOTSOCK);
200 	}
201 
202 	iov.iov_base = msg;
203 	iov.iov_len = msglen;
204 
205 	bzero(&auio, sizeof (struct uio));
206 	auio.uio_loffset = 0;
207 	auio.uio_iov = &iov;
208 	auio.uio_iovcnt = 1;
209 	auio.uio_resid = msglen;
210 	if (flags & MSG_USERSPACE)
211 		auio.uio_segflg = UIO_USERSPACE;
212 	else
213 		auio.uio_segflg = UIO_SYSSPACE;
214 	auio.uio_extflg = UIO_COPY_DEFAULT;
215 	auio.uio_limit = 0;
216 	auio.uio_fmode = KSOCKET_FMODE(ks);
217 
218 	msghdr.msg_name = NULL;
219 	msghdr.msg_namelen = 0;
220 	msghdr.msg_control = NULL;
221 	msghdr.msg_controllen = 0;
222 	msghdr.msg_flags = flags | MSG_EOR;
223 
224 	error = socket_sendmsg(KSTOSO(ks), &msghdr, &auio, cr);
225 	if (error != 0) {
226 		if (sent != NULL)
227 			*sent = 0;
228 		return (error);
229 	}
230 
231 	if (sent != NULL)
232 		*sent = msglen - auio.uio_resid;
233 	return (0);
234 }
235 
236 int
237 ksocket_sendto(ksocket_t ks, void *msg, size_t msglen, int flags,
238     struct sockaddr *name, socklen_t namelen, size_t *sent, struct cred *cr)
239 {
240 	int error;
241 	struct nmsghdr msghdr;
242 	struct uio auio;
243 	struct iovec iov;
244 
245 	/* All Solaris components should pass a cred for this operation. */
246 	ASSERT(cr != NULL);
247 
248 	if (!KSOCKET_VALID(ks)) {
249 		if (sent != NULL)
250 			*sent = 0;
251 		return (ENOTSOCK);
252 	}
253 
254 	iov.iov_base = msg;
255 	iov.iov_len = msglen;
256 
257 	bzero(&auio, sizeof (struct uio));
258 	auio.uio_loffset = 0;
259 	auio.uio_iov = &iov;
260 	auio.uio_iovcnt = 1;
261 	auio.uio_resid = msglen;
262 	if (flags & MSG_USERSPACE)
263 		auio.uio_segflg = UIO_USERSPACE;
264 	else
265 		auio.uio_segflg = UIO_SYSSPACE;
266 	auio.uio_extflg = UIO_COPY_DEFAULT;
267 	auio.uio_limit = 0;
268 	auio.uio_fmode = KSOCKET_FMODE(ks);
269 
270 	msghdr.msg_iov = &iov;
271 	msghdr.msg_iovlen = 1;
272 	msghdr.msg_name = (char *)name;
273 	msghdr.msg_namelen = namelen;
274 	msghdr.msg_control = NULL;
275 	msghdr.msg_controllen = 0;
276 	msghdr.msg_flags = flags | MSG_EOR;
277 
278 	error = socket_sendmsg(KSTOSO(ks), &msghdr, &auio, cr);
279 	if (error != 0) {
280 		if (sent != NULL)
281 			*sent = 0;
282 		return (error);
283 	}
284 	if (sent != NULL)
285 		*sent = msglen - auio.uio_resid;
286 	return (0);
287 }
288 
289 int
290 ksocket_sendmsg(ksocket_t ks, struct nmsghdr *msg, int flags,
291     size_t *sent, struct cred *cr)
292 {
293 	int error;
294 	ssize_t len;
295 	int i;
296 	struct uio auio;
297 
298 	/* All Solaris components should pass a cred for this operation. */
299 	ASSERT(cr != NULL);
300 
301 	if (!KSOCKET_VALID(ks)) {
302 		if (sent != NULL)
303 			*sent = 0;
304 		return (ENOTSOCK);
305 	}
306 
307 	bzero(&auio, sizeof (struct uio));
308 	auio.uio_loffset = 0;
309 	auio.uio_iov = msg->msg_iov;
310 	auio.uio_iovcnt = msg->msg_iovlen;
311 	if (flags & MSG_USERSPACE)
312 		auio.uio_segflg = UIO_USERSPACE;
313 	else
314 		auio.uio_segflg = UIO_SYSSPACE;
315 	auio.uio_extflg = UIO_COPY_DEFAULT;
316 	auio.uio_limit = 0;
317 	auio.uio_fmode = KSOCKET_FMODE(ks);
318 	len = 0;
319 	for (i = 0; i < msg->msg_iovlen; i++) {
320 		ssize_t iovlen;
321 		iovlen = (msg->msg_iov)[i].iov_len;
322 		len += iovlen;
323 		if (len < 0 || iovlen < 0)
324 			return (EINVAL);
325 	}
326 	auio.uio_resid = len;
327 
328 	msg->msg_flags = flags | MSG_EOR;
329 
330 	error = socket_sendmsg(KSTOSO(ks), msg, &auio, cr);
331 	if (error != 0) {
332 		if (sent != NULL)
333 			*sent = 0;
334 		return (error);
335 	}
336 
337 	if (sent != NULL)
338 		*sent = len - auio.uio_resid;
339 	return (0);
340 }
341 
342 
343 int
344 ksocket_recv(ksocket_t ks, void *msg, size_t msglen, int flags,
345     size_t *recv, struct cred *cr)
346 {
347 	int error;
348 	struct nmsghdr msghdr;
349 	struct uio auio;
350 	struct iovec iov;
351 
352 	/* All Solaris components should pass a cred for this operation. */
353 	ASSERT(cr != NULL);
354 
355 	if (!KSOCKET_VALID(ks)) {
356 		if (recv != NULL)
357 			*recv = 0;
358 		return (ENOTSOCK);
359 	}
360 
361 	iov.iov_base = msg;
362 	iov.iov_len = msglen;
363 
364 	bzero(&auio, sizeof (struct uio));
365 	auio.uio_loffset = 0;
366 	auio.uio_iov = &iov;
367 	auio.uio_iovcnt = 1;
368 	auio.uio_resid = msglen;
369 	if (flags & MSG_USERSPACE)
370 		auio.uio_segflg = UIO_USERSPACE;
371 	else
372 		auio.uio_segflg = UIO_SYSSPACE;
373 	auio.uio_extflg = UIO_COPY_DEFAULT;
374 	auio.uio_limit = 0;
375 	auio.uio_fmode = KSOCKET_FMODE(ks);
376 
377 	msghdr.msg_name = NULL;
378 	msghdr.msg_namelen = 0;
379 	msghdr.msg_control = NULL;
380 	msghdr.msg_controllen = 0;
381 	msghdr.msg_flags = flags & (MSG_OOB | MSG_PEEK | MSG_WAITALL |
382 	    MSG_DONTWAIT | MSG_USERSPACE);
383 
384 	error = socket_recvmsg(KSTOSO(ks), &msghdr, &auio, cr);
385 	if (error != 0) {
386 		if (recv != NULL)
387 			*recv = 0;
388 		return (error);
389 	}
390 
391 	if (recv != NULL)
392 		*recv = msglen - auio.uio_resid;
393 	return (0);
394 }
395 
396 int
397 ksocket_recvfrom(ksocket_t ks, void *msg, size_t msglen, int flags,
398     struct sockaddr *name, socklen_t *namelen, size_t *recv, struct cred *cr)
399 {
400 	int error;
401 	struct nmsghdr msghdr;
402 	struct uio auio;
403 	struct iovec iov;
404 
405 	/* All Solaris components should pass a cred for this operation. */
406 	ASSERT(cr != NULL);
407 
408 	if (!KSOCKET_VALID(ks)) {
409 		if (recv != NULL)
410 			*recv = 0;
411 		return (ENOTSOCK);
412 	}
413 
414 	iov.iov_base = msg;
415 	iov.iov_len = msglen;
416 
417 	bzero(&auio, sizeof (struct uio));
418 	auio.uio_loffset = 0;
419 	auio.uio_iov = &iov;
420 	auio.uio_iovcnt = 1;
421 	auio.uio_resid = msglen;
422 	if (flags & MSG_USERSPACE)
423 		auio.uio_segflg = UIO_USERSPACE;
424 	else
425 		auio.uio_segflg = UIO_SYSSPACE;
426 	auio.uio_extflg = UIO_COPY_DEFAULT;
427 	auio.uio_limit = 0;
428 	auio.uio_fmode = KSOCKET_FMODE(ks);
429 
430 	msghdr.msg_name = (char *)name;
431 	msghdr.msg_namelen = *namelen;
432 	msghdr.msg_control = NULL;
433 	msghdr.msg_controllen = 0;
434 	msghdr.msg_flags = flags & (MSG_OOB | MSG_PEEK | MSG_WAITALL |
435 	    MSG_DONTWAIT | MSG_USERSPACE);
436 
437 	error = socket_recvmsg(KSTOSO(ks), &msghdr, &auio, cr);
438 	if (error != 0) {
439 		if (recv != NULL)
440 			*recv = 0;
441 		return (error);
442 	}
443 	if (recv != NULL)
444 		*recv = msglen - auio.uio_resid;
445 
446 	bcopy(msghdr.msg_name, name, msghdr.msg_namelen);
447 	bcopy(&msghdr.msg_namelen, namelen, sizeof (msghdr.msg_namelen));
448 	return (0);
449 }
450 
451 int
452 ksocket_recvmsg(ksocket_t ks, struct nmsghdr *msg, int flags, size_t *recv,
453     struct cred *cr)
454 {
455 	int error;
456 	ssize_t len;
457 	int i;
458 	struct uio auio;
459 
460 	/* All Solaris components should pass a cred for this operation. */
461 	ASSERT(cr != NULL);
462 
463 	if (!KSOCKET_VALID(ks)) {
464 		if (recv != NULL)
465 			*recv = 0;
466 		return (ENOTSOCK);
467 	}
468 
469 	bzero(&auio, sizeof (struct uio));
470 	auio.uio_loffset = 0;
471 	auio.uio_iov = msg->msg_iov;
472 	auio.uio_iovcnt = msg->msg_iovlen;
473 	if (msg->msg_flags & MSG_USERSPACE)
474 		auio.uio_segflg = UIO_USERSPACE;
475 	else
476 		auio.uio_segflg = UIO_SYSSPACE;
477 	auio.uio_extflg = UIO_COPY_DEFAULT;
478 	auio.uio_limit = 0;
479 	auio.uio_fmode = KSOCKET_FMODE(ks);
480 	len = 0;
481 
482 	for (i = 0; i < msg->msg_iovlen; i++) {
483 		ssize_t iovlen;
484 		iovlen = (msg->msg_iov)[i].iov_len;
485 		len += iovlen;
486 		if (len < 0 || iovlen < 0)
487 			return (EINVAL);
488 	}
489 	auio.uio_resid = len;
490 
491 	msg->msg_flags = flags & (MSG_OOB | MSG_PEEK | MSG_WAITALL |
492 	    MSG_DONTWAIT | MSG_USERSPACE);
493 
494 	error = socket_recvmsg(KSTOSO(ks), msg, &auio, cr);
495 	if (error != 0) {
496 		if (recv != NULL)
497 			*recv = 0;
498 		return (error);
499 	}
500 	if (recv != NULL)
501 		*recv = len - auio.uio_resid;
502 	return (0);
503 
504 }
505 
506 int
507 ksocket_shutdown(ksocket_t ks, int how, struct cred *cr)
508 {
509 	struct sonode *so;
510 
511 	/* All Solaris components should pass a cred for this operation. */
512 	ASSERT(cr != NULL);
513 
514 	if (!KSOCKET_VALID(ks))
515 		return (ENOTSOCK);
516 
517 	so = KSTOSO(ks);
518 
519 	return (socket_shutdown(so, how, cr));
520 }
521 
522 int
523 ksocket_close(ksocket_t ks, struct cred *cr)
524 {
525 	struct sonode *so;
526 	so = KSTOSO(ks);
527 
528 	/* All Solaris components should pass a cred for this operation. */
529 	ASSERT(cr != NULL);
530 
531 	mutex_enter(&so->so_lock);
532 
533 	if (!KSOCKET_VALID(ks)) {
534 		mutex_exit(&so->so_lock);
535 		return (ENOTSOCK);
536 	}
537 
538 	so->so_state |= SS_CLOSING;
539 
540 	if (so->so_count > 1) {
541 		mutex_enter(&so->so_acceptq_lock);
542 		cv_broadcast(&so->so_acceptq_cv);
543 		mutex_exit(&so->so_acceptq_lock);
544 		cv_broadcast(&so->so_rcv_cv);
545 		cv_broadcast(&so->so_state_cv);
546 		cv_broadcast(&so->so_single_cv);
547 		cv_broadcast(&so->so_read_cv);
548 		cv_broadcast(&so->so_snd_cv);
549 		cv_broadcast(&so->so_copy_cv);
550 	}
551 	while (so->so_count > 1)
552 		cv_wait(&so->so_closing_cv, &so->so_lock);
553 
554 	mutex_exit(&so->so_lock);
555 	/* Remove callbacks, if any */
556 	(void) ksocket_setcallbacks(ks, NULL, NULL, cr);
557 
558 	(void) socket_close(so, 0, cr);
559 	socket_destroy(so);
560 
561 	return (0);
562 }
563 
564 int
565 ksocket_getsockname(ksocket_t ks, struct sockaddr *addr, socklen_t *addrlen,
566     struct cred *cr)
567 {
568 	struct sonode *so;
569 
570 	/* All Solaris components should pass a cred for this operation. */
571 	ASSERT(cr != NULL);
572 
573 	if (!KSOCKET_VALID(ks))
574 		return (ENOTSOCK);
575 
576 	so = KSTOSO(ks);
577 
578 	if (addrlen == NULL || (addr == NULL && *addrlen != 0))
579 		return (EFAULT);
580 
581 	return (socket_getsockname(so, addr, addrlen, cr));
582 }
583 
584 int
585 ksocket_getpeername(ksocket_t ks, struct sockaddr *addr, socklen_t *addrlen,
586     struct cred *cr)
587 {
588 	struct sonode *so;
589 
590 	/* All Solaris components should pass a cred for this operation. */
591 	ASSERT(cr != NULL);
592 
593 	if (!KSOCKET_VALID(ks))
594 		return (ENOTSOCK);
595 
596 	so = KSTOSO(ks);
597 
598 	if (addrlen == NULL || (addr == NULL && *addrlen != 0))
599 		return (EFAULT);
600 
601 	return (socket_getpeername(so, addr, addrlen, B_FALSE, cr));
602 }
603 
604 int
605 ksocket_getsockopt(ksocket_t ks, int level, int optname, void *optval,
606     int *optlen, struct cred *cr)
607 {
608 	struct sonode *so;
609 
610 	/* All Solaris components should pass a cred for this operation. */
611 	ASSERT(cr != NULL);
612 
613 	if (!KSOCKET_VALID(ks))
614 		return (ENOTSOCK);
615 
616 	so = KSTOSO(ks);
617 
618 	if (optlen == NULL)
619 		return (EFAULT);
620 	if (*optlen > SO_MAXARGSIZE)
621 		return (EINVAL);
622 
623 	return (socket_getsockopt(so, level, optname, optval,
624 	    (socklen_t *)optlen, 0, cr));
625 }
626 
627 int
628 ksocket_setsockopt(ksocket_t ks, int level, int optname, const void *optval,
629     int optlen, struct cred *cr)
630 {
631 	struct sonode *so;
632 
633 	/* All Solaris components should pass a cred for this operation. */
634 	ASSERT(cr != NULL);
635 
636 	if (!KSOCKET_VALID(ks))
637 		return (ENOTSOCK);
638 
639 	so = KSTOSO(ks);
640 
641 	if (optval == NULL)
642 		optlen = 0;
643 
644 	return (socket_setsockopt(so, level, optname, optval,
645 	    (t_uscalar_t)optlen, cr));
646 }
647 
648 /* ARGSUSED */
649 int
650 ksocket_setcallbacks(ksocket_t ks, ksocket_callbacks_t *cb, void *arg,
651     struct cred *cr)
652 {
653 	struct sonode *so;
654 
655 	/* All Solaris components should pass a cred for this operation. */
656 	ASSERT(cr != NULL);
657 
658 	if (!KSOCKET_VALID(ks))
659 		return (ENOTSOCK);
660 
661 	so = KSTOSO(ks);
662 
663 	if (cb == NULL && arg != NULL)
664 		return (EFAULT);
665 	if (cb == NULL) {
666 		mutex_enter(&so->so_lock);
667 		bzero(&(so->so_ksock_callbacks), sizeof (ksocket_callbacks_t));
668 		so->so_ksock_cb_arg = NULL;
669 		mutex_exit(&so->so_lock);
670 	} else {
671 		mutex_enter(&so->so_lock);
672 		SETCALLBACK(so, cb, connected, KSOCKET_CB_CONNECTED)
673 		SETCALLBACK(so, cb, connectfailed, KSOCKET_CB_CONNECTFAILED)
674 		SETCALLBACK(so, cb, disconnected, KSOCKET_CB_DISCONNECTED)
675 		SETCALLBACK(so, cb, newdata, KSOCKET_CB_NEWDATA)
676 		SETCALLBACK(so, cb, newconn, KSOCKET_CB_NEWCONN)
677 		SETCALLBACK(so, cb, cansend, KSOCKET_CB_CANSEND)
678 		SETCALLBACK(so, cb, oobdata, KSOCKET_CB_OOBDATA)
679 		SETCALLBACK(so, cb, cantsendmore, KSOCKET_CB_CANTSENDMORE)
680 		SETCALLBACK(so, cb, cantrecvmore, KSOCKET_CB_CANTRECVMORE)
681 		so->so_ksock_cb_arg = arg;
682 		mutex_exit(&so->so_lock);
683 	}
684 	return (0);
685 }
686 
687 int
688 ksocket_ioctl(ksocket_t ks, int cmd, intptr_t arg, int *rvalp, struct cred *cr)
689 {
690 	struct sonode *so;
691 	int rval;
692 
693 	/* All Solaris components should pass a cred for this operation. */
694 	ASSERT(cr != NULL);
695 
696 	if (!KSOCKET_VALID(ks))
697 		return (ENOTSOCK);
698 
699 	so = KSTOSO(ks);
700 
701 	switch (cmd) {
702 	default:
703 		/* STREAM iotcls are not supported */
704 		if ((cmd & 0xffffff00U) == STR) {
705 			rval = EOPNOTSUPP;
706 		} else {
707 			rval = socket_ioctl(so, cmd, arg,
708 			    KSOCKET_FMODE(ks) | FKIOCTL, cr, rvalp);
709 		}
710 		break;
711 	case FIOASYNC:
712 	case SIOCSPGRP:
713 	case FIOSETOWN:
714 	case SIOCGPGRP:
715 	case FIOGETOWN:
716 		rval = EOPNOTSUPP;
717 		break;
718 	}
719 
720 	return (rval);
721 }
722 
723 /*
724  * Wait for an input event, similar to t_kspoll().
725  * Ideas and code borrowed from ../devpoll.c
726  * Basically, setup just enough poll data structures so
727  * we can block on a CV until timeout or pollwakeup().
728  */
729 int
730 ksocket_spoll(ksocket_t ks, int timo, short events, short *revents,
731     struct cred *cr)
732 {
733 	struct		sonode *so;
734 	pollhead_t	*php, *php2;
735 	polldat_t	*pdp;
736 	pollcache_t	*pcp;
737 	int		error;
738 	clock_t		expires = 0;
739 	clock_t		rval;
740 
741 	/* All Solaris components should pass a cred for this operation. */
742 	ASSERT(cr != NULL);
743 	ASSERT(curthread->t_pollcache == NULL);
744 
745 	if (revents == NULL)
746 		return (EINVAL);
747 	if (!KSOCKET_VALID(ks))
748 		return (ENOTSOCK);
749 	so = KSTOSO(ks);
750 
751 	/*
752 	 * Check if there are any events already pending.
753 	 * If we're not willing to block, (timo == 0) then
754 	 * pass "anyyet">0 to socket_poll so it can skip
755 	 * some work.  Othewise pass "anyyet"=0 and if
756 	 * there are no events pending, it will fill in
757 	 * the pollhead pointer we need for pollwakeup().
758 	 *
759 	 * XXX - pollrelock() logic needs to know which
760 	 * which pollcache lock to grab. It'd be a
761 	 * cleaner solution if we could pass pcp as
762 	 * an arguement in VOP_POLL interface instead
763 	 * of implicitly passing it using thread_t
764 	 * struct. On the other hand, changing VOP_POLL
765 	 * interface will require all driver/file system
766 	 * poll routine to change. May want to revisit
767 	 * the tradeoff later.
768 	 */
769 	php = NULL;
770 	*revents = 0;
771 	pcp = pcache_alloc();
772 	pcache_create(pcp, 1);
773 
774 	mutex_enter(&pcp->pc_lock);
775 	curthread->t_pollcache = pcp;
776 	error = socket_poll(so, (short)events, (timo == 0),
777 	    revents, &php);
778 	curthread->t_pollcache = NULL;
779 	mutex_exit(&pcp->pc_lock);
780 
781 	if (error != 0 || *revents != 0 || timo == 0)
782 		goto out;
783 
784 	/*
785 	 * Need to block.  Did not get *revents, so the
786 	 * php should be non-NULL, but let's verify.
787 	 * Also compute when our sleep expires.
788 	 */
789 	if (php == NULL) {
790 		error = EIO;
791 		goto out;
792 	}
793 	if (timo > 0)
794 		expires = ddi_get_lbolt() +
795 		    MSEC_TO_TICK_ROUNDUP(timo);
796 
797 	/*
798 	 * Setup: pollhead -> polldat -> pollcache
799 	 * needed for pollwakeup()
800 	 * pdp should be freed by pcache_destroy
801 	 */
802 	pdp = kmem_zalloc(sizeof (*pdp), KM_SLEEP);
803 	pdp->pd_fd = 0;
804 	pdp->pd_events = events;
805 	pdp->pd_pcache = pcp;
806 	pcache_insert_fd(pcp, pdp, 1);
807 	pollhead_insert(php, pdp);
808 	pdp->pd_php = php;
809 
810 	mutex_enter(&pcp->pc_lock);
811 	while (!(so->so_state & SS_CLOSING)) {
812 		pcp->pc_flag = 0;
813 
814 		/* Ditto pcp comment above. */
815 		curthread->t_pollcache = pcp;
816 		error = socket_poll(so, (short)events, 0,
817 		    revents, &php2);
818 		curthread->t_pollcache = NULL;
819 		ASSERT(php2 == php);
820 
821 		if (error != 0 || *revents != 0)
822 			break;
823 
824 		if (pcp->pc_flag & PC_POLLWAKE)
825 			continue;
826 
827 		if (timo == -1) {
828 			rval = cv_wait_sig(&pcp->pc_cv, &pcp->pc_lock);
829 		} else {
830 			rval = cv_timedwait_sig(&pcp->pc_cv, &pcp->pc_lock,
831 			    expires);
832 		}
833 		if (rval <= 0) {
834 			if (rval == 0)
835 				error = EINTR;
836 			break;
837 		}
838 	}
839 	mutex_exit(&pcp->pc_lock);
840 
841 	if (pdp->pd_php != NULL) {
842 		pollhead_delete(pdp->pd_php, pdp);
843 		pdp->pd_php = NULL;
844 		pdp->pd_fd = NULL;
845 	}
846 
847 	/*
848 	 * pollwakeup() may still interact with this pollcache. Wait until
849 	 * it is done.
850 	 */
851 	mutex_enter(&pcp->pc_no_exit);
852 	ASSERT(pcp->pc_busy >= 0);
853 	while (pcp->pc_busy > 0)
854 		cv_wait(&pcp->pc_busy_cv, &pcp->pc_no_exit);
855 	mutex_exit(&pcp->pc_no_exit);
856 out:
857 	pcache_destroy(pcp);
858 	return (error);
859 }
860 
861 int
862 ksocket_sendmblk(ksocket_t ks, struct nmsghdr *msg, int flags,
863     mblk_t **mpp, cred_t *cr)
864 {
865 	struct		sonode *so;
866 	int		i_val;
867 	socklen_t	val_len;
868 	mblk_t		*mp = *mpp;
869 	int		error;
870 
871 	/* All Solaris components should pass a cred for this operation. */
872 	ASSERT(cr != NULL);
873 
874 	if (!KSOCKET_VALID(ks))
875 		return (ENOTSOCK);
876 
877 	so = KSTOSO(ks);
878 
879 	if (flags & MSG_MBLK_QUICKRELE) {
880 		error = socket_getsockopt(so, SOL_SOCKET, SO_SND_COPYAVOID,
881 		    &i_val, &val_len, 0, cr);
882 		if (error != 0)
883 			return (error);
884 
885 		/* Zero copy is not enable */
886 		if (i_val == 0)
887 			return (ECANCELED);
888 
889 		for (; mp != NULL; mp = mp->b_cont)
890 			mp->b_datap->db_struioflag |= STRUIO_ZC;
891 	}
892 
893 	error = socket_sendmblk(so, msg, flags, cr, mpp);
894 
895 	return (error);
896 }
897 
898 
899 void
900 ksocket_hold(ksocket_t ks)
901 {
902 	struct sonode *so;
903 	so = KSTOSO(ks);
904 
905 	if (!mutex_owned(&so->so_lock)) {
906 		mutex_enter(&so->so_lock);
907 		so->so_count++;
908 		mutex_exit(&so->so_lock);
909 	} else
910 		so->so_count++;
911 }
912 
913 void
914 ksocket_rele(ksocket_t ks)
915 {
916 	struct sonode *so;
917 
918 	so = KSTOSO(ks);
919 	/*
920 	 * When so_count equals 1 means no thread working on this ksocket
921 	 */
922 	if (so->so_count < 2)
923 		cmn_err(CE_PANIC, "ksocket_rele: sonode ref count 0 or 1");
924 
925 	if (!mutex_owned(&so->so_lock)) {
926 		mutex_enter(&so->so_lock);
927 		if (--so->so_count == 1)
928 			cv_signal(&so->so_closing_cv);
929 		mutex_exit(&so->so_lock);
930 	} else {
931 		if (--so->so_count == 1)
932 			cv_signal(&so->so_closing_cv);
933 	}
934 }
935