Merge remote branch 't-nelson/automake'
authorRonnie Sahlberg <ronniesahlberg@gmail.com>
Sat, 2 Jul 2011 01:10:36 +0000 (11:10 +1000)
committerRonnie Sahlberg <ronniesahlberg@gmail.com>
Sat, 2 Jul 2011 01:10:36 +0000 (11:10 +1000)
1  2 
lib/socket.c

diff --combined lib/socket.c
index 0bed7515561ce09ac8d68baa31077c47744c4307,96bc695c60c105f518d0a56663f0f55cac19be5f..dcbd1c177fd9be1d998399630ad8787be8deb441
@@@ -15,6 -15,9 +15,9 @@@
     along with this program; if not, see <http://www.gnu.org/licenses/>.
  */
  
+ #ifdef HAVE_CONFIG_H
+ #include "config.h"
+ #endif
  #include <stdio.h>
  #include <stdlib.h>
  #include <unistd.h>
  #include <rpc/rpc.h>
  #include <rpc/xdr.h>
  #include <arpa/inet.h>
+ #ifdef HAVE_SYS_FILIO_H
+ #include <sys/filio.h>
+ #endif
  #include <sys/ioctl.h>
 +#include <sys/types.h>
 +#include <sys/socket.h>
 +#include <netdb.h>
  #include "libnfs.h"
  #include "libnfs-raw.h"
  #include "libnfs-private.h"
@@@ -50,11 -53,6 +56,11 @@@ int rpc_which_events(struct rpc_contex
  {
        int events = rpc->is_connected ? POLLIN : POLLOUT;
  
 +      if (rpc->is_udp != 0) {
 +              /* for udp sockets we only wait for pollin */
 +              return POLLIN;
 +      }
 +
        if (rpc->outqueue) {
                events |= POLLOUT;
        }
@@@ -114,29 -112,6 +120,29 @@@ static int rpc_read_from_socket(struct 
                return -1;
        }
  
 +      if (rpc->is_udp) {
 +              char *buf;
 +              socklen_t socklen = sizeof(rpc->udp_src);
 +
 +              buf = malloc(available);
 +              if (buf == NULL) {
 +                      rpc_set_error(rpc, "Failed to malloc buffer for recvfrom");
 +                      return -1;
 +              }
 +              count = recvfrom(rpc->fd, buf, available, MSG_DONTWAIT, (struct sockaddr *)&rpc->udp_src, &socklen);
 +              if (count < 0) {
 +                      rpc_set_error(rpc, "Failed recvfrom: %s", strerror(errno));
 +                      free(buf);
 +              }
 +              if (rpc_process_pdu(rpc, buf, count) != 0) {
 +                      rpc_set_error(rpc, "Invalid/garbage pdu received from server. Ignoring PDU");
 +                      free(buf);
 +                      return -1;
 +              }
 +              free(buf);
 +              return 0;
 +      }
 +
        /* read record marker, 4 bytes at the beginning of every pdu */
        if (rpc->inbuf == NULL) {
                rpc->insize = 4;
@@@ -290,11 -265,6 +296,11 @@@ int rpc_connect_async(struct rpc_contex
                return -1;
        }
  
 +      if (rpc->is_udp != 0) {
 +              rpc_set_error(rpc, "Trying to connect on UDP socket");
 +              return -1;
 +      }
 +
        sin->sin_family = AF_INET;
        sin->sin_port   = htons(port);
        if (inet_pton(AF_INET, server, &sin->sin_addr) != 1) {
@@@ -343,87 -313,3 +349,87 @@@ int rpc_disconnect(struct rpc_context *
  
        return 0;
  }
 +
 +
 +int rpc_bind_udp(struct rpc_context *rpc, char *addr, int port)
 +{
 +      struct addrinfo *ai = NULL;
 +      char service[6];
 +
 +      if (rpc->is_udp == 0) {
 +              rpc_set_error(rpc, "Cant not bind UDP. Not UDP context");
 +              return -1;
 +      }
 +
 +      snprintf(service, 6, "%d", port);
 +      if (getaddrinfo(addr, service, NULL, &ai) != 0) {
 +              rpc_set_error(rpc, "Invalid address:%s. "
 +                      "Can not resolv into IPv4/v6 structure.");
 +              return -1;
 +      }
 +
 +      switch(ai->ai_family) {
 +      case AF_INET:
 +              rpc->fd = socket(ai->ai_family, SOCK_DGRAM, 0);
 +              if (rpc->fd == -1) {
 +                      rpc_set_error(rpc, "Failed to create UDP socket: %s", strerror(errno)); 
 +                      freeaddrinfo(ai);
 +                      return -1;
 +              }
 +
 +              if (bind(rpc->fd, (struct sockaddr *)ai->ai_addr, sizeof(struct sockaddr_in)) != 0) {
 +                      rpc_set_error(rpc, "Failed to bind to UDP socket: %s",strerror(errno)); 
 +                      freeaddrinfo(ai);
 +                      return -1;
 +              }
 +              break;
 +      default:
 +              rpc_set_error(rpc, "Can not handle UPD sockets of family %d yet", ai->ai_family);
 +              freeaddrinfo(ai);
 +              return -1;
 +      }
 +
 +      freeaddrinfo(ai);
 +
 +      return 0;
 +}
 +
 +int rpc_set_udp_destination(struct rpc_context *rpc, char *addr, int port, int is_broadcast)
 +{
 +      struct addrinfo *ai = NULL;
 +      char service[6];
 +
 +      if (rpc->is_udp == 0) {
 +              rpc_set_error(rpc, "Can not set destination sockaddr. Not UDP context");
 +              return -1;
 +      }
 +
 +      snprintf(service, 6, "%d", port);
 +      if (getaddrinfo(addr, service, NULL, &ai) != 0) {
 +              rpc_set_error(rpc, "Invalid address:%s. "
 +                      "Can not resolv into IPv4/v6 structure.");
 +              return -1;
 +      }
 +
 +      if (rpc->udp_dest) {
 +              free(rpc->udp_dest);
 +              rpc->udp_dest = NULL;
 +      }
 +      rpc->udp_dest = malloc(ai->ai_addrlen);
 +      if (rpc->udp_dest == NULL) {
 +              rpc_set_error(rpc, "Out of memory. Failed to allocate sockaddr structure");
 +              return -1;
 +      }
 +      memcpy(rpc->udp_dest, ai->ai_addr, ai->ai_addrlen);
 +      freeaddrinfo(ai);
 +
 +      rpc->is_broadcast = is_broadcast;
 +      setsockopt(rpc->fd, SOL_SOCKET, SO_BROADCAST, &is_broadcast, sizeof(is_broadcast));
 +
 +      return 0;
 +}
 +
 +struct sockaddr *rpc_get_recv_sockaddr(struct rpc_context *rpc)
 +{
 +      return (struct sockaddr *)&rpc->udp_src;
 +}