-/*

- Copyright (c) 2008-2009 Gluster, Inc. <http://www.gluster.com>

- This file is part of GlusterFS.

-

- GlusterFS is free software; you can redistribute it and/or modify

- it under the terms of the GNU General Public License as published

- by the Free Software Foundation; either version 3 of the License,

- or (at your option) any later version.

-

- GlusterFS is distributed in the hope that it will be useful, but

- WITHOUT ANY WARRANTY; without even the implied warranty of

- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

- General Public License for more details.

-

- You should have received a copy of the GNU General Public License

- along with this program. If not, see

- <http://www.gnu.org/licenses/>.

-*/

-

-

-#ifndef _CONFIG_H

-#define _CONFIG_H

-#include "config.h"

-#endif

-

-#include "socket.h"

-#include "name.h"

-#include "dict.h"

-#include "transport.h"

-#include "logging.h"

-#include "xlator.h"

-#include "byte-order.h"

-#include "common-utils.h"

-#include "compat-errno.h"

-

-#include <fcntl.h>

-#include <errno.h>

-#include <netinet/tcp.h>

-

-

-#define GF_LOG_ERRNO(errno) ((errno == ENOTCONN) ? GF_LOG_DEBUG : GF_LOG_ERROR)

-#define SA(ptr) ((struct sockaddr *)ptr)

-

-int socket_init (transport_t *this);

-

-/*

- * return value:

- * 0 = success (completed)

- * -1 = error

- * > 0 = incomplete

- */

-

-int

-__socket_rwv (transport_t *this, struct iovec *vector, int count,

- struct iovec **pending_vector, int *pending_count,

- int write)

-{

- socket_private_t *priv = NULL;

- int sock = -1;

- int ret = -1;

- struct iovec *opvector = NULL;

- int opcount = 0;

- int moved = 0;

-

- priv = this->private;

- sock = priv->sock;

-

- opvector = vector;

- opcount = count;

-

- while (opcount) {

- if (write) {

- ret = writev (sock, opvector, opcount);

-

- if (ret == 0 || (ret == -1 && errno == EAGAIN)) {

- /* done for now */

- break;

- }

- } else {

- ret = readv (sock, opvector, opcount);

-

- if (ret == -1 && errno == EAGAIN) {

- /* done for now */

- break;

- }

- }

-

- if (ret == 0) {

- /* Mostly due to 'umount' in client */

- gf_log (this->xl->name, GF_LOG_TRACE,

- "EOF from peer %s", this->peerinfo.identifier);

- opcount = -1;

- errno = ENOTCONN;

- break;

- }

-

- if (ret == -1) {

- if (errno == EINTR)

- continue;

-

- gf_log (this->xl->name, GF_LOG_TRACE,

- "%s failed (%s)", write ? "writev" : "readv",

- strerror (errno));

- opcount = -1;

- break;

- }

-

- moved = 0;

-

- while (moved < ret) {

- if ((ret - moved) >= opvector[0].iov_len) {

- moved += opvector[0].iov_len;

- opvector++;

- opcount--;

- } else {

- opvector[0].iov_len -= (ret - moved);

- opvector[0].iov_base += (ret - moved);

- moved += (ret - moved);

- }

- while (opcount && !opvector[0].iov_len) {

- opvector++;

- opcount--;

- }

- }

- }

-

- if (pending_vector)

- *pending_vector = opvector;

-

- if (pending_count)

- *pending_count = opcount;

-

- return opcount;

-}

-

-

-int

-__socket_readv (transport_t *this, struct iovec *vector, int count,

- struct iovec **pending_vector, int *pending_count)

-{

- int ret = -1;

-

- ret = __socket_rwv (this, vector, count,

- pending_vector, pending_count, 0);

-

- return ret;

-}

-

-

-int

-__socket_writev (transport_t *this, struct iovec *vector, int count,

- struct iovec **pending_vector, int *pending_count)

-{

- int ret = -1;

-

- ret = __socket_rwv (this, vector, count,

- pending_vector, pending_count, 1);

-

- return ret;

-}

-

-

-int

-__socket_disconnect (transport_t *this)

-{

- socket_private_t *priv = NULL;

- int ret = -1;

-

- priv = this->private;

-

- if (priv->sock != -1) {

- ret = shutdown (priv->sock, SHUT_RDWR);

- priv->connected = -1;

- gf_log (this->xl->name, GF_LOG_TRACE,

- "shutdown() returned %d. set connection state to -1",

- ret);

- }

-

- return ret;

-}

-

-

-int

-__socket_server_bind (transport_t *this)

-{

- socket_private_t *priv = NULL;

- int ret = -1;

- int opt = 1;

-

- priv = this->private;

-

- ret = setsockopt (priv->sock, SOL_SOCKET, SO_REUSEADDR,

- &opt, sizeof (opt));

-

- if (ret == -1) {

- gf_log (this->xl->name, GF_LOG_ERROR,

- "setsockopt() for SO_REUSEADDR failed (%s)",

- strerror (errno));

- }

-

- ret = bind (priv->sock, (struct sockaddr *)&this->myinfo.sockaddr,

- this->myinfo.sockaddr_len);

-

- if (ret == -1) {

- gf_log (this->xl->name, GF_LOG_ERROR,

- "binding to %s failed: %s",

- this->myinfo.identifier, strerror (errno));

- if (errno == EADDRINUSE) {

- gf_log (this->xl->name, GF_LOG_ERROR,

- "Port is already in use");

- }

- }

-

- return ret;

-}

-

-

-int

-__socket_nonblock (int fd)

-{

- int flags = 0;

- int ret = -1;

-

- flags = fcntl (fd, F_GETFL);

-

- if (flags != -1)

- ret = fcntl (fd, F_SETFL, flags | O_NONBLOCK);

-

- return ret;

-}

-

-

-int

-__socket_nodelay (int fd)

-{

- int on = 1;

- int ret = -1;

-

- ret = setsockopt (fd, IPPROTO_TCP, TCP_NODELAY,

- &on, sizeof (on));

- if (!ret)

- gf_log ("", GF_LOG_TRACE,

- "NODELAY enabled for socket %d", fd);

-

- return ret;

-}

-

-

-int

-__socket_keepalive (int fd, int keepalive_intvl)

-{

- int on = 1;

- int ret = -1;

-

- ret = setsockopt (fd, SOL_SOCKET, SO_KEEPALIVE, &on, sizeof (on));

- if (ret == -1)

- goto err;

-

- if (keepalive_intvl == GF_USE_DEFAULT_KEEPALIVE)

- goto done;

-

-#ifndef GF_LINUX_HOST_OS

- ret = setsockopt (fd, IPPROTO_TCP, TCP_KEEPALIVE, &keepalive_intvl,

- sizeof (keepalive_intvl));

- if (ret == -1)

- goto err;

-#else

- ret = setsockopt (fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepalive_intvl,

- sizeof (keepalive_intvl));

- if (ret == -1)