smb_net.c revision 8d7e41661dc4633488e93b13363137523ce59977
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License (the "License").
* You may not use this file except in compliance with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2008 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#include <sys/types.h>
#include <sys/param.h>
#include <sys/cpuvar.h>
#include <sys/ddi.h>
#include <sys/sunddi.h>
#include <sys/time.h>
#include <sys/varargs.h>
#include <sys/modctl.h>
#include <sys/pathname.h>
#include <sys/fs/snode.h>
#include <sys/fs/dv_node.h>
#include <sys/vnode.h>
#undef mem_free /* XXX Remove this after we convert everything to kmem_alloc */
#include <smbsrv/smb_vops.h>
#include <smbsrv/smb.h>
#include <smbsrv/smb_kproto.h>
#include <smbsrv/smb_kstat.h>
static kmem_cache_t *smb_txr_cache = NULL;
/*
* smb_net_init
*
* This function initializes the resources necessary to access the
* network. It assumes it won't be called simultaneously by multiple
* threads.
*
* Return Value
*
* 0 Initialization successful
* ENOMEM Initialization failed
*/
int
smb_net_init(void)
{
int rc = 0;
ASSERT(smb_txr_cache == NULL);
smb_txr_cache = kmem_cache_create(SMBSRV_KSTAT_TXRCACHE,
sizeof (smb_txreq_t), 8, NULL, NULL, NULL, NULL, NULL, 0);
if (smb_txr_cache == NULL)
rc = ENOMEM;
return (rc);
}
/*
* smb_net_fini
*
* This function releases the resources allocated by smb_net_init(). It
* assumes it won't be called simultaneously by multiple threads.
* This function can safely be called even if smb_net_init() hasn't been
* called previously.
*
* Return Value
*
* None
*/
void
smb_net_fini(void)
{
if (smb_txr_cache) {
kmem_cache_destroy(smb_txr_cache);
smb_txr_cache = NULL;
}
}
/*
* SMB Network Socket API
*
* smb_socreate: Creates an socket based on domain/type.
* smb_soshutdown: Disconnect a socket created with smb_socreate
* smb_sodestroy: Release resources associated with a socket
* smb_sosend: Send the contents of a buffer on a socket
* smb_sorecv: Receive data into a buffer from a socket
* smb_iov_sosend: Send the contents of an iovec on a socket
* smb_iov_sorecv: Receive data into an iovec from a socket
*/
struct sonode *
smb_socreate(int domain, int type, int protocol)
{
vnode_t *dvp = NULL;
vnode_t *vp = NULL;
struct snode *csp = NULL;
int err = 0;
major_t maj;
if ((vp = solookup(domain, type, protocol, NULL, &err)) == NULL) {
/*
* solookup calls sogetvp if the vp is not found in the cache.
* Since the call to sogetvp is hardwired to use USERSPACE
* and declared static we'll do the work here instead.
*/
err = lookupname(type == SOCK_STREAM ? "/dev/tcp" : "/dev/udp",
UIO_SYSSPACE, FOLLOW, NULLVPP, &vp);
if (err)
return (NULL);
/* Check that it is the correct vnode */
if (vp->v_type != VCHR) {
VN_RELE(vp);
return (NULL);
}
csp = VTOS(VTOS(vp)->s_commonvp);
if (!(csp->s_flag & SDIPSET)) {
char *pathname = kmem_alloc(MAXPATHLEN, KM_SLEEP);
err = ddi_dev_pathname(vp->v_rdev, S_IFCHR,
pathname);
if (err == 0) {
err = devfs_lookupname(pathname, NULLVPP,
&dvp);
}
VN_RELE(vp);
kmem_free(pathname, MAXPATHLEN);
if (err != 0) {
return (NULL);
}
vp = dvp;
}
maj = getmajor(vp->v_rdev);
if (!STREAMSTAB(maj)) {
VN_RELE(vp);
return (NULL);
}
}
return (socreate(vp, domain, type, protocol, SOV_DEFAULT, NULL, &err));
}
/*
* smb_soshutdown will disconnect the socket and prevent subsequent PDU
* reception and transmission. The sonode still exists but its state
* gets modified to indicate it is no longer connected. Calls to
* smb_sorecv/smb_iov_sorecv will return so smb_soshutdown can be used
* regain control of a thread stuck in smb_sorecv.
*/
void
smb_soshutdown(struct sonode *so)
{
(void) soshutdown(so, SHUT_RDWR);
}
/*
* smb_sodestroy releases all resources associated with a socket previously
* created with smb_socreate. The socket must be shutdown using smb_soshutdown
* before the socket is destroyed with smb_sodestroy, otherwise undefined
* behavior will result.
*/
void
smb_sodestroy(struct sonode *so)
{
vnode_t *vp = SOTOV(so);
(void) VOP_CLOSE(vp, 0, 1, 0, kcred, NULL);
VN_RELE(vp);
}
int
smb_sorecv(struct sonode *so, void *msg, size_t len)
{
iovec_t iov;
int err;
ASSERT(so != NULL);
ASSERT(len != 0);
/*
* Fill in iovec and receive data
*/
iov.iov_base = msg;
iov.iov_len = len;
if ((err = smb_iov_sorecv(so, &iov, 1, len)) != 0) {
return (err);
}
/* Successful receive */
return (0);
}
/*
* smb_iov_sorecv - Receives an iovec from a connection
*
* This function gets the data asked for from the socket. It will return
* only when all the requested data has been retrieved or if an error
* occurs.
*
* Returns 0 for success, the socket errno value if sorecvmsg fails, and
* -1 if sorecvmsg returns success but uio_resid != 0
*/
int
smb_iov_sorecv(struct sonode *so, iovec_t *iop, int iovlen, size_t total_len)
{
struct msghdr msg;
struct uio uio;
int error;
ASSERT(iop != NULL);
/* Initialization of the message header. */
bzero(&msg, sizeof (msg));
msg.msg_iov = iop;
msg.msg_flags = MSG_WAITALL;
msg.msg_iovlen = iovlen;
/* Initialization of the uio structure. */
bzero(&uio, sizeof (uio));
uio.uio_iov = iop;
uio.uio_iovcnt = iovlen;
uio.uio_segflg = UIO_SYSSPACE;
uio.uio_resid = total_len;
if ((error = sorecvmsg(so, &msg, &uio)) == 0) {
/* Received data */
if (uio.uio_resid == 0) {
/* All requested data received. Success */
return (0);
} else {
/* Not all data was sent. Failure */
return (-1);
}
}
/* Receive failed */
return (error);
}
/*
* smb_net_txl_constructor
*
* Transmit list constructor
*/
void
smb_net_txl_constructor(smb_txlst_t *txl)
{
ASSERT(txl->tl_magic != SMB_TXLST_MAGIC);
mutex_init(&txl->tl_mutex, NULL, MUTEX_DEFAULT, NULL);
list_create(&txl->tl_list, sizeof (smb_txreq_t),
offsetof(smb_txreq_t, tr_lnd));
txl->tl_active = B_FALSE;
txl->tl_magic = SMB_TXLST_MAGIC;
}
/*
* smb_net_txl_destructor
*
* Transmit list destructor
*/
void
smb_net_txl_destructor(smb_txlst_t *txl)
{
ASSERT(txl->tl_magic == SMB_TXLST_MAGIC);
txl->tl_magic = 0;
list_destroy(&txl->tl_list);
mutex_destroy(&txl->tl_mutex);
}
/*
* smb_net_txr_alloc
*
* Transmit buffer allocator
*/
smb_txreq_t *
smb_net_txr_alloc(void)
{
smb_txreq_t *txr;
txr = kmem_cache_alloc(smb_txr_cache, KM_SLEEP);
txr->tr_len = 0;
bzero(&txr->tr_lnd, sizeof (txr->tr_lnd));
txr->tr_magic = SMB_TXREQ_MAGIC;
return (txr);
}
/*
* smb_net_txr_free
*
* Transmit buffer deallocator
*/
void
smb_net_txr_free(smb_txreq_t *txr)
{
ASSERT(txr->tr_magic == SMB_TXREQ_MAGIC);
ASSERT(!list_link_active(&txr->tr_lnd));
txr->tr_magic = 0;
kmem_cache_free(smb_txr_cache, txr);
}
/*
* smb_net_txr_send
*
* This routine puts the transmit buffer passed in on the wire. If another
* thread is already draining the transmit list, the transmit buffer is
* queued and the routine returns immediately.
*/
int
smb_net_txr_send(struct sonode *so, smb_txlst_t *txl, smb_txreq_t *txr)
{
list_t local;
int rc = 0;
iovec_t iov;
struct msghdr msg;
struct uio uio;
ASSERT(txl->tl_magic == SMB_TXLST_MAGIC);
mutex_enter(&txl->tl_mutex);
list_insert_tail(&txl->tl_list, txr);
if (txl->tl_active) {
mutex_exit(&txl->tl_mutex);
return (0);
}
txl->tl_active = B_TRUE;
list_create(&local, sizeof (smb_txreq_t),
offsetof(smb_txreq_t, tr_lnd));
while (!list_is_empty(&txl->tl_list)) {
list_move_tail(&local, &txl->tl_list);
mutex_exit(&txl->tl_mutex);
while ((txr = list_head(&local)) != NULL) {
ASSERT(txr->tr_magic == SMB_TXREQ_MAGIC);
list_remove(&local, txr);
iov.iov_base = (void *)txr->tr_buf;
iov.iov_len = txr->tr_len;
bzero(&msg, sizeof (msg));
msg.msg_iov = &iov;
msg.msg_flags = MSG_WAITALL;
msg.msg_iovlen = 1;
bzero(&uio, sizeof (uio));
uio.uio_iov = &iov;
uio.uio_iovcnt = 1;
uio.uio_segflg = UIO_SYSSPACE;
uio.uio_resid = txr->tr_len;
rc = sosendmsg(so, &msg, &uio);
smb_net_txr_free(txr);
if ((rc == 0) && (uio.uio_resid == 0))
continue;
if (rc == 0)
rc = -1;
while ((txr = list_head(&local)) != NULL) {
ASSERT(txr->tr_magic == SMB_TXREQ_MAGIC);
list_remove(&local, txr);
smb_net_txr_free(txr);
}
break;
}
mutex_enter(&txl->tl_mutex);
if (rc == 0)
continue;
while ((txr = list_head(&txl->tl_list)) != NULL) {
ASSERT(txr->tr_magic == SMB_TXREQ_MAGIC);
list_remove(&txl->tl_list, txr);
smb_net_txr_free(txr);
}
break;
}
txl->tl_active = B_FALSE;
mutex_exit(&txl->tl_mutex);
return (rc);
}