int
libcfs_sock_listen (struct socket **sockp,
__u32 local_ip, int local_port, int backlog)
{
int fatal;
int rc;
CFS_DECL_FUNNEL_DATA;
rc = libcfs_sock_create(sockp, &fatal, local_ip, local_port);
if (rc != 0) {
if (!fatal)
CERROR("Can't create socket: port %d already in use\n",
local_port);
return rc;
}
CFS_NET_IN;
rc = solisten(*sockp, backlog);
CFS_NET_EX;
if (rc == 0)
return 0;
CERROR("Can't set listen backlog %d: %d\n", backlog, rc);
CFS_NET_IN;
soclose(*sockp);
CFS_NET_EX;
return -rc;
}
int
libcfs_sock_listen (cfs_socket_t **sockp,
__u32 local_ip, int local_port, int backlog)
{
cfs_socket_t *sock;
int fatal;
int rc;
rc = libcfs_sock_create(&sock, &fatal, local_ip, local_port);
if (rc != 0) {
if (!fatal)
CERROR("Can't create socket: port %d already in use\n",
local_port);
return rc;
}
rc = -sock_listen(C2B_SOCK(sock), backlog);
if (rc == 0) {
*sockp = sock;
return 0;
}
if (C2B_SOCK(sock) != NULL)
sock_close(C2B_SOCK(sock));
FREE(sock, M_TEMP);
return rc;
}
int
libcfs_sock_connect (cfs_socket_t **sockp, int *fatal,
__u32 local_ip, int local_port,
__u32 peer_ip, int peer_port)
{
cfs_socket_t *sock;
struct sockaddr_in srvaddr;
int rc;
rc = libcfs_sock_create(&sock, fatal, local_ip, local_port);
if (rc != 0)
return rc;
bzero(&srvaddr, sizeof(srvaddr));
srvaddr.sin_len = sizeof(struct sockaddr_in);
srvaddr.sin_family = AF_INET;
srvaddr.sin_port = htons(peer_port);
srvaddr.sin_addr.s_addr = htonl(peer_ip);
rc = -sock_connect(C2B_SOCK(sock), (struct sockaddr *)&srvaddr, 0);
if (rc == 0) {
*sockp = sock;
return 0;
}
*fatal = !(rc == -EADDRNOTAVAIL || rc == -EADDRINUSE);
CDEBUG(*fatal ? D_NETERROR : D_NET,
"Error %d connecting %u.%u.%u.%u/%d -> %u.%u.%u.%u/%d\n", rc,
HIPQUAD(local_ip), local_port, HIPQUAD(peer_ip), peer_port);
libcfs_sock_release(sock);
return rc;
}
int
libcfs_sock_listen (cfs_socket_t **sockp,
__u32 local_ip, int local_port, int backlog)
{
int rc;
int fatal;
rc = libcfs_sock_create(sockp, &fatal, local_ip, local_port);
if (rc != 0)
return rc;
if ( listen((*sockp)->s_fd, backlog) ) {
rc = -errno;
CERROR("listen() with backlog==%d failed: errno==%d\n",
backlog, errno);
goto failed;
}
return 0;
failed:
libcfs_sock_release(*sockp);
return rc;
}
/*
* XXX Liang: timeout for write is not supported yet.
*/
int
libcfs_sock_write (struct socket *sock, void *buffer, int nob, int timeout)
{
int rc;
CFS_DECL_NET_DATA;
while (nob > 0) {
struct iovec iov = {
.iov_base = buffer,
.iov_len = nob
};
struct uio suio = {
.uio_iov = &iov,
.uio_iovcnt = 1,
.uio_offset = 0,
.uio_resid = nob,
.uio_segflg = UIO_SYSSPACE,
.uio_rw = UIO_WRITE,
.uio_procp = NULL
};
CFS_NET_IN;
rc = sosend(sock, NULL, &suio, (struct mbuf *)0, (struct mbuf *)0, 0);
CFS_NET_EX;
if (rc != 0) {
if ( suio.uio_resid != nob && ( rc == ERESTART || rc == EINTR ||\
rc == EWOULDBLOCK))
rc = 0;
if ( rc != 0 )
return -rc;
rc = nob - suio.uio_resid;
buffer = ((char *)buffer) + rc;
nob = suio.uio_resid;
continue;
}
break;
}
return (0);
}
/*
* XXX Liang: timeout for read is not supported yet.
*/
int
libcfs_sock_read (struct socket *sock, void *buffer, int nob, int timeout)
{
int rc;
CFS_DECL_NET_DATA;
while (nob > 0) {
struct iovec iov = {
.iov_base = buffer,
.iov_len = nob
};
struct uio ruio = {
.uio_iov = &iov,
.uio_iovcnt = 1,
.uio_offset = 0,
.uio_resid = nob,
.uio_segflg = UIO_SYSSPACE,
.uio_rw = UIO_READ,
.uio_procp = NULL
};
CFS_NET_IN;
rc = soreceive(sock, (struct sockaddr **)0, &ruio, (struct mbuf **)0, (struct mbuf **)0, (int *)0);
CFS_NET_EX;
if (rc != 0) {
if ( ruio.uio_resid != nob && ( rc == ERESTART || rc == EINTR ||\
rc == EWOULDBLOCK))
rc = 0;
if (rc != 0)
return -rc;
rc = nob - ruio.uio_resid;
buffer = ((char *)buffer) + rc;
nob = ruio.uio_resid;
continue;
}
break;
}
return (0);
}
int
libcfs_sock_setbuf (struct socket *sock, int txbufsize, int rxbufsize)
{
struct sockopt sopt;
int rc = 0;
int option;
CFS_DECL_NET_DATA;
bzero(&sopt, sizeof sopt);
sopt.sopt_dir = SOPT_SET;
sopt.sopt_level = SOL_SOCKET;
sopt.sopt_val = &option;
sopt.sopt_valsize = sizeof(option);
if (txbufsize != 0) {
option = txbufsize;
if (option > KSOCK_MAX_BUF)
option = KSOCK_MAX_BUF;
sopt.sopt_name = SO_SNDBUF;
CFS_NET_IN;
rc = sosetopt(sock, &sopt);
CFS_NET_EX;
if (rc != 0) {
CERROR ("Can't set send buffer %d: %d\n",
option, rc);
return -rc;
}
}
if (rxbufsize != 0) {
option = rxbufsize;
sopt.sopt_name = SO_RCVBUF;
CFS_NET_IN;
rc = sosetopt(sock, &sopt);
CFS_NET_EX;
if (rc != 0) {
CERROR ("Can't set receive buffer %d: %d\n",
option, rc);
return -rc;
}
}
return 0;
}
int
libcfs_sock_getaddr (struct socket *sock, int remote, __u32 *ip, int *port)
{
struct sockaddr_in *sin;
struct sockaddr *sa = NULL;
int rc;
CFS_DECL_NET_DATA;
if (remote != 0) {
CFS_NET_IN;
rc = sock->so_proto->pr_usrreqs->pru_peeraddr(sock, &sa);
CFS_NET_EX;
if (rc != 0) {
if (sa) FREE(sa, M_SONAME);
CERROR ("Error %d getting sock peer IP\n", rc);
return -rc;
}
} else {
CFS_NET_IN;
rc = sock->so_proto->pr_usrreqs->pru_sockaddr(sock, &sa);
CFS_NET_EX;
if (rc != 0) {
if (sa) FREE(sa, M_SONAME);
CERROR ("Error %d getting sock local IP\n", rc);
return -rc;
}
}
if (sa != NULL) {
sin = (struct sockaddr_in *)sa;
if (ip != NULL)
*ip = ntohl (sin->sin_addr.s_addr);
if (port != NULL)
*port = ntohs (sin->sin_port);
if (sa)
FREE(sa, M_SONAME);
}
return 0;
}
int
libcfs_sock_getbuf (struct socket *sock, int *txbufsize, int *rxbufsize)
{
struct sockopt sopt;
int rc;
CFS_DECL_NET_DATA;
bzero(&sopt, sizeof sopt);
sopt.sopt_dir = SOPT_GET;
sopt.sopt_level = SOL_SOCKET;
if (txbufsize != NULL) {
sopt.sopt_val = txbufsize;
sopt.sopt_valsize = sizeof(*txbufsize);
sopt.sopt_name = SO_SNDBUF;
CFS_NET_IN;
rc = sogetopt(sock, &sopt);
CFS_NET_EX;
if (rc != 0) {
CERROR ("Can't get send buffer size: %d\n", rc);
return -rc;
}
}
if (rxbufsize != NULL) {
sopt.sopt_val = rxbufsize;
sopt.sopt_valsize = sizeof(*rxbufsize);
sopt.sopt_name = SO_RCVBUF;
CFS_NET_IN;
rc = sogetopt(sock, &sopt);
CFS_NET_EX;
if (rc != 0) {
CERROR ("Can't get receive buffer size: %d\n", rc);
return -rc;
}
}
return 0;
}
int
libcfs_sock_connect (struct socket **sockp, int *fatal,
__u32 local_ip, int local_port,
__u32 peer_ip, int peer_port)
{
struct sockaddr_in srvaddr;
struct socket *so;
int s;
int rc;
CFS_DECL_FUNNEL_DATA;
rc = libcfs_sock_create(sockp, fatal, local_ip, local_port);
if (rc != 0)
return rc;
so = *sockp;
bzero(&srvaddr, sizeof(srvaddr));
srvaddr.sin_len = sizeof(struct sockaddr_in);
srvaddr.sin_family = AF_INET;
srvaddr.sin_port = htons (peer_port);
srvaddr.sin_addr.s_addr = htonl (peer_ip);
CFS_NET_IN;
rc = soconnect(so, (struct sockaddr *)&srvaddr);
if (rc != 0) {
CFS_NET_EX;
if (rc != EADDRNOTAVAIL && rc != EADDRINUSE)
CDEBUG(D_NETERROR,
"Error %d connecting %u.%u.%u.%u/%d -> %u.%u.%u.%u/%d\n", rc,
HIPQUAD(local_ip), local_port, HIPQUAD(peer_ip), peer_port);
goto out;
}
s = splnet();
while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
CDEBUG(D_NET, "ksocknal sleep for waiting auto_connect.\n");
(void) tsleep((caddr_t)&so->so_timeo, PSOCK, "ksocknal_conn", hz);
}
if ((rc = so->so_error) != 0) {
so->so_error = 0;
splx(s);
CFS_NET_EX;
CDEBUG(D_NETERROR,
"Error %d connecting %u.%u.%u.%u/%d -> %u.%u.%u.%u/%d\n", rc,
HIPQUAD(local_ip), local_port, HIPQUAD(peer_ip), peer_port);
goto out;
}
LASSERT(so->so_state & SS_ISCONNECTED);
splx(s);
CFS_NET_EX;
if (sockp)
*sockp = so;
return (0);
out:
CFS_NET_IN;
soshutdown(so, 2);
soclose(so);
CFS_NET_EX;
return (-rc);
}
void
libcfs_sock_release (struct socket *sock)
{
CFS_DECL_FUNNEL_DATA;
CFS_NET_IN;
soshutdown(sock, 0);
CFS_NET_EX;
}
