]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/lib/bnet.c
Move more bnet functions into the BSOCK class.
[bacula/bacula] / bacula / src / lib / bnet.c
index 4fd1e59fb3407c35a3fb38c55b643c702771c3b7..b81d0bf52b6c1dc3bf4800a14251b4c4cfcb3d79 100644 (file)
@@ -79,8 +79,8 @@ int32_t read_nbytes(BSOCK * bsock, char *ptr, int32_t nbytes)
    nleft = nbytes;
    while (nleft > 0) {
       errno = 0;
-      nread = socketRead(bsock->fd, ptr, nleft);
-      if (bsock->timed_out || bsock->terminated) {
+      nread = socketRead(bsock->m_fd, ptr, nleft);
+      if (bsock->is_timed_out() || bsock->is_terminated()) {
          return nread;
       }
       if (nread == -1) {
@@ -110,13 +110,13 @@ int32_t write_nbytes(BSOCK * bsock, char *ptr, int32_t nbytes)
 {
    int32_t nleft, nwritten;
 
-   if (bsock->spool) {
-      nwritten = fwrite(ptr, 1, nbytes, bsock->spool_fd);
+   if (bsock->m_spool) {
+      nwritten = fwrite(ptr, 1, nbytes, bsock->m_spool_fd);
       if (nwritten != nbytes) {
          berrno be;
          bsock->b_errno = errno;
          Qmsg1(bsock->jcr(), M_FATAL, 0, _("Attr spool write error. ERR=%s\n"),
-               be.strerror());
+               be.bstrerror());
          Dmsg2(400, "nwritten=%d nbytes=%d.\n", nwritten, nbytes);
          errno = bsock->b_errno;
          return -1;
@@ -135,8 +135,8 @@ int32_t write_nbytes(BSOCK * bsock, char *ptr, int32_t nbytes)
    while (nleft > 0) {
       do {
          errno = 0;
-         nwritten = socketWrite(bsock->fd, ptr, nleft);
-         if (bsock->timed_out || bsock->terminated) {
+         nwritten = socketWrite(bsock->m_fd, ptr, nleft);
+         if (bsock->is_timed_out() || bsock->is_terminated()) {
             return nwritten;
          }
       } while (nwritten == -1 && errno == EINTR);
@@ -150,10 +150,10 @@ int32_t write_nbytes(BSOCK * bsock, char *ptr, int32_t nbytes)
          struct timeval tv;
 
          FD_ZERO(&fdset);
-         FD_SET((unsigned)bsock->fd, &fdset);
+         FD_SET((unsigned)bsock->m_fd, &fdset);
          tv.tv_sec = 10;
          tv.tv_usec = 0;
-         select(bsock->fd + 1, NULL, &fdset, NULL, &tv);
+         select(bsock->m_fd + 1, NULL, &fdset, NULL, &tv);
          continue;
       }
       if (nwritten <= 0) {
@@ -194,7 +194,7 @@ int32_t bnet_recv(BSOCK * bsock)
  */
 bool is_bnet_stop(BSOCK * bsock)
 {
-   return bsock->errors || bsock->terminated;
+   return bsock->errors || bsock->is_terminated();
 }
 
 /*
@@ -212,7 +212,7 @@ int is_bnet_error(BSOCK * bsock)
  */
 void bnet_suppress_error_messages(BSOCK * bsock, bool flag)
 {
-   bsock->suppress_error_msgs = flag;
+   bsock->m_suppress_error_msgs = flag;
 }
 
 /*
@@ -239,7 +239,7 @@ bool bnet_tls_server(TLS_CONTEXT *ctx, BSOCK * bsock, alist *verify_list)
 {
    TLS_CONNECTION *tls;
    
-   tls = new_tls_connection(ctx, bsock->fd);
+   tls = new_tls_connection(ctx, bsock->m_fd);
    if (!tls) {
       Qmsg0(bsock->jcr(), M_FATAL, 0, _("TLS connection initialization failed.\n"));
       return false;
@@ -278,7 +278,7 @@ bool bnet_tls_client(TLS_CONTEXT *ctx, BSOCK * bsock, alist *verify_list)
 {
    TLS_CONNECTION *tls;
 
-   tls  = new_tls_connection(ctx, bsock->fd);
+   tls  = new_tls_connection(ctx, bsock->m_fd);
    if (!tls) {
       Qmsg0(bsock->jcr(), M_FATAL, 0, _("TLS connection initialization failed.\n"));
       return false;
@@ -345,11 +345,11 @@ int bnet_wait_data(BSOCK * bsock, int sec)
    struct timeval tv;
 
    FD_ZERO(&fdset);
-   FD_SET((unsigned)bsock->fd, &fdset);
+   FD_SET((unsigned)bsock->m_fd, &fdset);
    for (;;) {
       tv.tv_sec = sec;
       tv.tv_usec = 0;
-      switch (select(bsock->fd + 1, &fdset, NULL, NULL, &tv)) {
+      switch (select(bsock->m_fd + 1, &fdset, NULL, NULL, &tv)) {
       case 0:                      /* timeout */
          bsock->b_errno = 0;
          return 0;
@@ -375,10 +375,10 @@ int bnet_wait_data_intr(BSOCK * bsock, int sec)
    struct timeval tv;
 
    FD_ZERO(&fdset);
-   FD_SET((unsigned)bsock->fd, &fdset);
+   FD_SET((unsigned)bsock->m_fd, &fdset);
    tv.tv_sec = sec;
    tv.tv_usec = 0;
-   switch (select(bsock->fd + 1, &fdset, NULL, NULL, &tv)) {
+   switch (select(bsock->m_fd + 1, &fdset, NULL, NULL, &tv)) {
    case 0:                      /* timeout */
       bsock->b_errno = 0;
       return 0;
@@ -419,7 +419,7 @@ static const char *gethost_strerror()
    berrno be;
    switch (h_errno) {
    case NETDB_INTERNAL:
-      msg = be.strerror();
+      msg = be.bstrerror();
       break;
    case NETDB_SUCCESS:
       msg = _("No problem.");
@@ -593,7 +593,7 @@ static BSOCK *bnet_open(JCR *jcr, const char *name, char *host, char *service,
          save_errno = errno;
          *fatal = 1;
          Pmsg3(000, _("Socket open error. proto=%d port=%d. ERR=%s\n"),
-            ipaddr->get_family(), ipaddr->get_port_host_order(), be.strerror());
+            ipaddr->get_family(), ipaddr->get_port_host_order(), be.bstrerror());
          continue;
       }
       /*
@@ -602,7 +602,7 @@ static BSOCK *bnet_open(JCR *jcr, const char *name, char *host, char *service,
       if (setsockopt(sockfd, SOL_SOCKET, SO_KEEPALIVE, (sockopt_val_t)&turnon, sizeof(turnon)) < 0) {
          berrno be;
          Qmsg1(jcr, M_WARNING, 0, _("Cannot set SO_KEEPALIVE on socket: %s\n"),
-               be.strerror());
+               be.bstrerror());
       }
 #if defined(TCP_KEEPIDLE)
       if (heart_beat) {
@@ -610,7 +610,7 @@ static BSOCK *bnet_open(JCR *jcr, const char *name, char *host, char *service,
          if (setsockopt(sockfd, IPPROTO_IP, TCP_KEEPIDLE, (sockopt_val_t)&opt, sizeof(opt)) < 0) {
             berrno be;
             Qmsg1(jcr, M_WARNING, 0, _("Cannot set SO_KEEPIDLE on socket: %s\n"),
-                  be.strerror());
+                  be.bstrerror());
          }
       }
 #endif
@@ -628,7 +628,7 @@ static BSOCK *bnet_open(JCR *jcr, const char *name, char *host, char *service,
 
    if (!connected) {
       free_addresses(addr_list);
-      errno = save_errno;
+      errno = save_errno | b_errno_win32;
       return NULL;
    }
    /*
@@ -638,7 +638,7 @@ static BSOCK *bnet_open(JCR *jcr, const char *name, char *host, char *service,
    if (setsockopt(sockfd, SOL_SOCKET, SO_KEEPALIVE, (sockopt_val_t)&turnon, sizeof(turnon)) < 0) {
       berrno be;
       Qmsg1(jcr, M_WARNING, 0, _("Cannot set SO_KEEPALIVE on socket: %s\n"),
-            be.strerror());
+            be.bstrerror());
    }
    BSOCK* ret =  init_bsock(jcr, sockfd, name, host, port, ipaddr->get_sockaddr());
    free_addresses(addr_list);
@@ -661,7 +661,9 @@ BSOCK *bnet_connect(JCR * jcr, int retry_interval, utime_t max_retry_time,
    btimer_t *tid = NULL;
 
    /* Try to trap out of OS call when time expires */
-   tid = start_thread_timer(pthread_self(), (uint32_t)max_retry_time);
+   if (max_retry_time) {
+      tid = start_thread_timer(pthread_self(), (uint32_t)max_retry_time);
+   }
    
    for (i = 0; (bsock = bnet_open(jcr, name, host, service, port, heart_beat, &fatal)) == NULL;
         i -= retry_interval) {
@@ -671,25 +673,25 @@ BSOCK *bnet_connect(JCR * jcr, int retry_interval, utime_t max_retry_time,
          goto bail_out;
       }
       Dmsg4(100, "Unable to connect to %s on %s:%d. ERR=%s\n",
-            name, host, port, be.strerror());
+            name, host, port, be.bstrerror());
       if (i < 0) {
          i = 60 * 5;               /* complain again in 5 minutes */
          if (verbose)
             Qmsg4(jcr, M_WARNING, 0, _(
                "Could not connect to %s on %s:%d. ERR=%s\n"
-               "Retrying ...\n"), name, host, port, be.strerror());
+               "Retrying ...\n"), name, host, port, be.bstrerror());
       }
       bmicrosleep(retry_interval, 0);
       now = time(NULL);
       if (begin_time + max_retry_time <= now) {
          Qmsg4(jcr, M_FATAL, 0, _("Unable to connect to %s on %s:%d. ERR=%s\n"),
-               name, host, port, be.strerror());
+               name, host, port, be.bstrerror());
          bsock = NULL;
          goto bail_out;
       }
    }
 
-bail_out;
+bail_out:
    if (tid) {
       stop_thread_timer(tid);
    }
@@ -707,7 +709,7 @@ const char *bnet_strerror(BSOCK * bsock)
    if (bsock->errmsg == NULL) {
       bsock->errmsg = get_pool_memory(PM_MESSAGE);
    }
-   pm_strcpy(bsock->errmsg, be.strerror(bsock->b_errno));
+   pm_strcpy(bsock->errmsg, be.bstrerror(bsock->b_errno));
    return bsock->errmsg;
 }
 
@@ -721,7 +723,7 @@ bool bnet_fsend(BSOCK * bs, const char *fmt, ...)
    va_list arg_ptr;
    int maxlen;
 
-   if (bs->errors || bs->terminated) {
+   if (bs->errors || bs->is_terminated()) {
       return false;
    }
    /* This probably won't work, but we vsnprintf, then if we
@@ -744,20 +746,9 @@ bool bnet_fsend(BSOCK * bs, const char *fmt, ...)
 
 int bnet_get_peer(BSOCK *bs, char *buf, socklen_t buflen) 
 {
-#if !defined(HAVE_WIN32)
-    if (bs->peer_addr.sin_family == 0) {
-        socklen_t salen = sizeof(bs->peer_addr);
-        int rval = (getpeername)(bs->fd, (struct sockaddr *)&bs->peer_addr, &salen);
-        if (rval < 0) return rval;
-    }
-    if (!inet_ntop(bs->peer_addr.sin_family, &bs->peer_addr.sin_addr, buf, buflen))
-        return -1;
-
-    return 0;
-#else
-    return -1;
-#endif
+   return bs->get_peer(buf, buflen);  
 }
+
 /*
  * Set the network buffer size, suggested size is in size.
  *  Actual size obtained is returned in bs->msglen
@@ -767,136 +758,25 @@ int bnet_get_peer(BSOCK *bs, char *buf, socklen_t buflen)
  */
 bool bnet_set_buffer_size(BSOCK * bs, uint32_t size, int rw)
 {
-   uint32_t dbuf_size, start_size;
-#if defined(IP_TOS) && defined(IPTOS_THROUGHPUT)
-   int opt;
-   opt = IPTOS_THROUGHPUT;
-   setsockopt(bs->fd, IPPROTO_IP, IP_TOS, (sockopt_val_t)&opt, sizeof(opt));
-#endif
-
-   if (size != 0) {
-      dbuf_size = size;
-   } else {
-      dbuf_size = DEFAULT_NETWORK_BUFFER_SIZE;
-   }
-   start_size = dbuf_size;
-   if ((bs->msg = realloc_pool_memory(bs->msg, dbuf_size + 100)) == NULL) {
-      Qmsg0(bs->jcr(), M_FATAL, 0, _("Could not malloc BSOCK data buffer\n"));
-      return false;
-   }
-   if (rw & BNET_SETBUF_READ) {
-      while ((dbuf_size > TAPE_BSIZE) && (setsockopt(bs->fd, SOL_SOCKET,
-              SO_RCVBUF, (sockopt_val_t) & dbuf_size, sizeof(dbuf_size)) < 0)) {
-         berrno be;
-         Qmsg1(bs->jcr(), M_ERROR, 0, _("sockopt error: %s\n"), be.strerror());
-         dbuf_size -= TAPE_BSIZE;
-      }
-      Dmsg1(200, "set network buffer size=%d\n", dbuf_size);
-      if (dbuf_size != start_size) {
-         Qmsg1(bs->jcr(), M_WARNING, 0,
-               _("Warning network buffer = %d bytes not max size.\n"), dbuf_size);
-      }
-      if (dbuf_size % TAPE_BSIZE != 0) {
-         Qmsg1(bs->jcr(), M_ABORT, 0,
-               _("Network buffer size %d not multiple of tape block size.\n"),
-               dbuf_size);
-      }
-   }
-   if (size != 0) {
-      dbuf_size = size;
-   } else {
-      dbuf_size = DEFAULT_NETWORK_BUFFER_SIZE;
-   }
-   start_size = dbuf_size;
-   if (rw & BNET_SETBUF_WRITE) {
-      while ((dbuf_size > TAPE_BSIZE) && (setsockopt(bs->fd, SOL_SOCKET,
-              SO_SNDBUF, (sockopt_val_t) & dbuf_size, sizeof(dbuf_size)) < 0)) {
-         berrno be;
-         Qmsg1(bs->jcr(), M_ERROR, 0, _("sockopt error: %s\n"), be.strerror());
-         dbuf_size -= TAPE_BSIZE;
-      }
-      Dmsg1(900, "set network buffer size=%d\n", dbuf_size);
-      if (dbuf_size != start_size) {
-         Qmsg1(bs->jcr(), M_WARNING, 0,
-               _("Warning network buffer = %d bytes not max size.\n"), dbuf_size);
-      }
-      if (dbuf_size % TAPE_BSIZE != 0) {
-         Qmsg1(bs->jcr(), M_ABORT, 0,
-               _("Network buffer size %d not multiple of tape block size.\n"),
-               dbuf_size);
-      }
-   }
-
-   bs->msglen = dbuf_size;
-   return true;
+   return bs->set_buffer_size(size, rw);
 }
 
 /*
  * Set socket non-blocking
  * Returns previous socket flag
  */
-int bnet_set_nonblocking (BSOCK *bsock) {
-#ifndef HAVE_WIN32
-   int oflags;
-
-   /* Get current flags */
-   if ((oflags = fcntl(bsock->fd, F_GETFL, 0)) < 0) {
-      berrno be;
-      Jmsg1(bsock->jcr(), M_ABORT, 0, _("fcntl F_GETFL error. ERR=%s\n"), be.strerror());
-   }
-
-   /* Set O_NONBLOCK flag */
-   if ((fcntl(bsock->fd, F_SETFL, oflags|O_NONBLOCK)) < 0) {
-      berrno be;
-      Jmsg1(bsock->jcr(), M_ABORT, 0, _("fcntl F_SETFL error. ERR=%s\n"), be.strerror());
-   }
-
-   bsock->blocking = 0;
-   return oflags;
-#else
-   int flags;
-   u_long ioctlArg = 1;
-
-   flags = bsock->blocking;
-   ioctlsocket(bsock->fd, FIONBIO, &ioctlArg);
-   bsock->blocking = 0;
-
-   return flags;
-#endif
+int bnet_set_nonblocking(BSOCK *bsock) 
+{
+   return bsock->set_nonblocking();
 }
 
 /*
  * Set socket blocking
  * Returns previous socket flags
  */
-int bnet_set_blocking (BSOCK *bsock) 
+int bnet_set_blocking(BSOCK *bsock) 
 {
-#ifndef HAVE_WIN32
-   int oflags;
-   /* Get current flags */
-   if ((oflags = fcntl(bsock->fd, F_GETFL, 0)) < 0) {
-      berrno be;
-      Jmsg1(bsock->jcr(), M_ABORT, 0, _("fcntl F_GETFL error. ERR=%s\n"), be.strerror());
-   }
-
-   /* Set O_NONBLOCK flag */
-   if ((fcntl(bsock->fd, F_SETFL, oflags & ~O_NONBLOCK)) < 0) {
-      berrno be;
-      Jmsg1(bsock->jcr(), M_ABORT, 0, _("fcntl F_SETFL error. ERR=%s\n"), be.strerror());
-   }
-
-   bsock->blocking = 1;
-   return oflags;
-#else
-   int flags;
-   u_long ioctlArg = 0;
-
-   flags = bsock->blocking;
-   ioctlsocket(bsock->fd, FIONBIO, &ioctlArg);
-   bsock->blocking = 1;
-
-   return flags;
-#endif
+   return bsock->set_blocking();
 }
 
 /*
@@ -904,19 +784,7 @@ int bnet_set_blocking (BSOCK *bsock)
  */
 void bnet_restore_blocking (BSOCK *bsock, int flags) 
 {
-#ifndef HAVE_WIN32
-   if ((fcntl(bsock->fd, F_SETFL, flags)) < 0) {
-      berrno be;
-      Jmsg1(bsock->jcr(), M_ABORT, 0, _("fcntl F_SETFL error. ERR=%s\n"), be.strerror());
-   }
-
-   bsock->blocking = (flags & O_NONBLOCK);
-#else
-   u_long ioctlArg = flags;
-
-   ioctlsocket(bsock->fd, FIONBIO, &ioctlArg);
-   bsock->blocking = 1;
-#endif
+   bsock->restore_blocking(flags);
 }
 
 
@@ -972,10 +840,10 @@ BSOCK *init_bsock(JCR * jcr, int sockfd, const char *who, const char *host, int
    Dmsg3(100, "who=%s host=%s port=%d\n", who, host, port);
    BSOCK *bsock = (BSOCK *)malloc(sizeof(BSOCK));
    memset(bsock, 0, sizeof(BSOCK));
-   bsock->fd = sockfd;
+   bsock->m_fd = sockfd;
    bsock->tls = NULL;
    bsock->errors = 0;
-   bsock->blocking = 1;
+   bsock->m_blocking = 1;
    bsock->msg = get_pool_memory(PM_MESSAGE);
    bsock->errmsg = get_pool_memory(PM_MESSAGE);
    bsock->set_who(bstrdup(who));
@@ -1004,7 +872,7 @@ BSOCK *dup_bsock(BSOCK * osock)
    if (osock->host()) {
       bsock->set_host(bstrdup(osock->host()));
    }
-   bsock->duped = true;
+   bsock->m_duped = true;
    return bsock;
 }