From 59a9ad4b3deb9e4025556db430765629d590116f Mon Sep 17 00:00:00 2001
From: Raghavendra G <raghavendra@gluster.com>
Date: Sun, 28 Mar 2010 23:44:23 +0000
Subject: transport/socket: guess an appropriate address family when it is not
 specified in configuration.

Signed-off-by: Raghavendra G <raghavendra@gluster.com>
Signed-off-by: Anand V. Avati <avati@dev.gluster.com>

BUG: 561 (with address family set to 'inet-sdp', transport/socket doesn't work correctly)
URL: http://bugs.gluster.com/cgi-bin/bugzilla3/show_bug.cgi?id=561
---
 transport/socket/src/name.c   | 33 ++++++++++++++++++++++++++++-----
 transport/socket/src/name.h   | 14 ++++----------
 transport/socket/src/socket.c | 14 ++------------
 3 files changed, 34 insertions(+), 27 deletions(-)

diff --git a/transport/socket/src/name.c b/transport/socket/src/name.c
index b8d9664adb0..120a669c87b 100644
--- a/transport/socket/src/name.c
+++ b/transport/socket/src/name.c
@@ -462,18 +462,28 @@ client_bind (transport_t *this,
 }
 
 int32_t
-socket_client_get_remote_sockaddr (transport_t *this, 
+socket_client_get_remote_sockaddr (transport_t *this,
                                    struct sockaddr *sockaddr, 
-                                   socklen_t *sockaddr_len)
+                                   socklen_t *sockaddr_len,
+                                   sa_family_t *sa_family)
 {
         int32_t ret = 0;
 
+        if ((sockaddr == NULL) || (sockaddr_len == NULL)
+            || (sa_family == NULL)) {
+                ret = -1;
+                goto err;
+        }
+
+
         ret = client_fill_address_family (this, &sockaddr->sa_family);
         if (ret) {
                 ret = -1;
                 goto err;
         }
  
+        *sa_family = sockaddr->sa_family;
+
         switch (sockaddr->sa_family)
         {
         case AF_INET_SDP:
@@ -497,6 +507,10 @@ socket_client_get_remote_sockaddr (transport_t *this,
                 ret = -1;
         }
   
+        if (*sa_family == AF_UNSPEC) {
+                *sa_family = sockaddr->sa_family;
+        }
+
 err:
         return ret;
 }
@@ -547,17 +561,22 @@ out:
 
 
 int32_t
-socket_server_get_local_sockaddr (transport_t *this, 
-                                  struct sockaddr *addr, 
-                                  socklen_t *addr_len)
+socket_server_get_local_sockaddr (transport_t *this, struct sockaddr *addr, 
+                                  socklen_t *addr_len, sa_family_t *sa_family)
 {
         int32_t ret = -1;
 
+        if ((addr == NULL) || (addr_len == NULL) || (sa_family == NULL)) {
+                goto err;
+        }
+
         ret = server_fill_address_family (this, &addr->sa_family);
         if (ret == -1) {
                 goto err;
         }
 
+        *sa_family = addr->sa_family;
+
         switch (addr->sa_family)
         {
         case AF_INET_SDP:
@@ -574,6 +593,10 @@ socket_server_get_local_sockaddr (transport_t *this,
                 break;
         }
 
+        if (*sa_family == AF_UNSPEC) {
+                *sa_family = addr->sa_family;
+        }
+
 err:
         return ret;
 }
diff --git a/transport/socket/src/name.h b/transport/socket/src/name.h
index ace94ebccf7..f50a7b7f4dd 100644
--- a/transport/socket/src/name.h
+++ b/transport/socket/src/name.h
@@ -28,21 +28,15 @@ client_bind (transport_t *this,
              socklen_t *sockaddr_len, 
              int sock);
 
-int32_t
-client_fill_address_family (transport_t *this, sa_family_t *sa_family);
-
-int32_t
-server_fill_address_family (transport_t *this, sa_family_t *sa_family);
-
 int32_t
 socket_client_get_remote_sockaddr (transport_t *this, 
                                    struct sockaddr *sockaddr, 
-                                   socklen_t *sockaddr_len);
+                                   socklen_t *sockaddr_len,
+                                   sa_family_t *sa_family);
 
 int32_t
-socket_server_get_local_sockaddr (transport_t *this, 
-                                  struct sockaddr *addr, 
-                                  socklen_t *addr_len);
+socket_server_get_local_sockaddr (transport_t *this, struct sockaddr *addr, 
+                                  socklen_t *addr_len, sa_family_t *sa_family);
 
 int32_t
 get_transport_identifiers (transport_t *this);
diff --git a/transport/socket/src/socket.c b/transport/socket/src/socket.c
index e2691500b9f..28b828f29a2 100644
--- a/transport/socket/src/socket.c
+++ b/transport/socket/src/socket.c
@@ -997,13 +997,8 @@ socket_connect (transport_t *this)
                 goto err;
         }
 
-        ret = client_fill_address_family (this, &sa_family);
-        if (ret == -1) {
-                goto err;
-        }
-
         ret = socket_client_get_remote_sockaddr (this, SA (&sockaddr),
-                                                 &sockaddr_len);
+                                                 &sockaddr_len, &sa_family);
         if (ret == -1) {
                 /* logged inside client_get_remote_sockaddr */
                 goto err;
@@ -1142,13 +1137,8 @@ socket_listen (transport_t *this)
                 return ret;
         }
 
-        ret = server_fill_address_family (this, &sa_family);
-        if (ret == -1) {
-                return ret;
-        }
-
         ret = socket_server_get_local_sockaddr (this, SA (&sockaddr), 
-                                                &sockaddr_len);
+                                                &sockaddr_len, &sa_family);
         if (ret == -1) {
                 return ret;
         }
-- 
cgit