Edit

kc3-lang/kc3/libkc3/socket_buf.c

Branch :

  • libkc3/socket_buf.c 
  • /* kc3
 * Copyright from 2022 to 2026 kmx.io <contact@kmx.io>
 *
 * Permission is hereby granted to use this software granted the above
 * copyright notice and this permission paragraph are included in all
 * copies and substantial portions of this software.
 *
 * THIS SOFTWARE IS PROVIDED "AS-IS" WITHOUT ANY GUARANTEE OF
 * PURPOSE AND PERFORMANCE. IN NO EVENT WHATSOEVER SHALL THE
 * AUTHOR BE CONSIDERED LIABLE FOR THE USE AND PERFORMANCE OF
 * THIS SOFTWARE.
 */
#include <errno.h>
#include <string.h>

#ifdef WIN32
# include <winsock2.h>
# include <ws2tcpip.h>
# undef errno
# define errno WSAGetLastError()
#else
# include <netdb.h>
# include <netinet/in.h>
# include <arpa/inet.h>
#endif

#include <unistd.h>
#include "kc3.h"
#include "mutex.h"
#include "socket.h"
#include "socket_addr.h"
#include "socket_buf.h"

void socket_buf_clean (s_socket_buf *sb)
{
  assert(sb);
  buf_rw_fd_close(&sb->buf_rw);
  if (sb->buf_rw.r || sb->buf_rw.w) {
    buf_rw_clean(&sb->buf_rw);
    sb->buf_rw.r = NULL;
    sb->buf_rw.w = NULL;
  }
  if (sb->addr_str.free.p_pvoid) {
    str_clean(&sb->addr_str);
    sb->addr_str.free.p_pvoid = NULL;
  }
  if (sb->addr) {
    socket_addr_delete(sb->addr);
    sb->addr = NULL;
  }
  if (sb->closed_mutex) {
    mutex_delete(sb->closed_mutex);
    sb->closed_mutex = NULL;
  }
}

void socket_buf_delete (s_socket_buf **sb)
{
  assert(sb);
  assert(*sb);
  socket_buf_clean(*sb);
  alloc_free(*sb);
  *sb = NULL;
}

bool socket_buf_can_close (s_socket_buf *sb)
{
  bool result;
  assert(sb);
  if (! sb->closed_mutex)
    return false;
  mutex_lock(sb->closed_mutex);
  if (sb->closed) {
    result = false;
  }
  else {
    sb->closed = true;
    result = true;
  }
  mutex_unlock(sb->closed_mutex);
  return result;
}

void socket_buf_close (s_socket_buf *sb)
{
  s32 e;
  assert(sb);
  buf_rw_fd_close(&sb->buf_rw);
  if (sb->sockfd >= 0) {
    if (shutdown(sb->sockfd, SHUT_RDWR) < 0 && errno != ENOTCONN) {
      e = errno;
      err_write_1("socket_buf_close: shutdown: ");
      err_puts(strerror(e));
    }
    if (close(sb->sockfd) < 0) {
      e = errno;
      err_write_1("socket_buf_close: close: ");
      err_puts(strerror(e));
    }
    sb->sockfd = -1;
  }
  if (sb->buf_rw.r || sb->buf_rw.w) {
    buf_rw_clean(&sb->buf_rw);
    sb->buf_rw.r = NULL;
    sb->buf_rw.w = NULL;
  }
  if (sb->addr_str.free.p_pvoid) {
    str_clean(&sb->addr_str);
    sb->addr_str.free.p_pvoid = NULL;
  }
  if (sb->addr) {
    socket_addr_delete(sb->addr);
    sb->addr = NULL;
  }
}

bool socket_buf_set_tag (s_socket_buf *sb, s_tag **tag)
{
  assert(sb);
  assert(tag);
  sb->ptag = *tag;
  return true;
}

s_pointer * socket_buf_tag (s_socket_buf *sb, s_pointer *dest)
{
  assert(sb);
  assert(dest);
  dest->target_type = &g_sym_Tag;
  dest->pointer_type = sym_target_to_pointer_type(&g_sym_Tag);
  dest->ptr.p_pvoid = sb->ptag;
  return dest;
}

s_socket_buf * socket_buf_init (s_socket_buf *sb, s64 sockfd,
                                struct sockaddr *addr,
                                u32 addr_len)
{
  assert(sb);
  assert(sockfd >= 0);
  assert(addr);
  assert(addr_len);
  *sb = (s_socket_buf) {0};
  if (! buf_rw_init_alloc(&sb->buf_rw, BUF_SIZE)) {
    err_puts("socket_buf_init: buf_rw_init_alloc");
    assert(! "socket_buf_init: buf_rw_init_alloc");
    return NULL;
  }
  sb->buf_rw.r->user_ptr = sb;
  sb->buf_rw.w->user_ptr = sb;
  sb->closed_mutex = mutex_new();
  if (! sb->closed_mutex) {
    err_puts("socket_buf_init: mutex_new");
    buf_rw_clean(&sb->buf_rw);
    return NULL;
  }
  sb->closed = false;
  if (! buf_rw_fd_open(&sb->buf_rw, sockfd)) {
    err_puts("socket_buf_init: buf_rw_fd_open");
    assert(! "socket_buf_init: buf_rw_fd_open");
    close(sockfd);
    buf_rw_clean(&sb->buf_rw);
    mutex_delete(sb->closed_mutex);
    return NULL;
  }
  sb->sockfd = sockfd;
  sb->addr = socket_addr_new_copy(addr, addr_len);
  if (! sb->addr) {
    err_puts("socket_buf_init: socket_addr_new_copy");
    assert(! "socket_buf_init: socket_addr_new_copy");
    buf_rw_fd_close(&sb->buf_rw);
    buf_rw_clean(&sb->buf_rw);
    mutex_delete(sb->closed_mutex);
    return NULL;
  }
  socket_addr_to_str(&sb->addr_str, addr, addr_len);
  sb->addr_len = addr_len;
  sb->ptag = NULL;
  return sb;
}

s_socket_buf * socket_buf_init_accept (s_socket_buf *sb, p_socket listening)
{
  struct sockaddr        *addr;
  struct sockaddr_storage addr_storage = {0};
  socklen_t               addr_len;
  s32 e;
  s64 sockfd;
  assert(sb);
  assert(listening);
  addr = (struct sockaddr *) &addr_storage;
  addr_len = sizeof(addr_storage);
  sockfd = accept(*listening, addr, &addr_len);
  if (sockfd < 0) {
    e = errno;
    err_write_1("socket_buf_init_accept: accept: ");
    err_puts(strerror(e));
    return NULL;
  }
  if (! socket_buf_init(sb, sockfd, addr, addr_len)) {
    err_puts("socket_buf_init_accept: socket_buf_init");
    assert(! "socket_buf_init_accept: socket_buf_init");
    close(sockfd);
    return NULL;
  }
  return sb;
}

s_socket_buf * socket_buf_init_connect (s_socket_buf *sb,
                                        const s_str *host,
                                        const s_str *service)
{
  struct addrinfo hints = {0};
  struct addrinfo *res;
  struct addrinfo *res0;
  struct addrinfo *res_last = NULL;
  s32 e;
  const char *error_reason = "error: ";
  char ipstr[128] = {0};
  s32 port = -1;
  s64 r = 0;
  s64 sockfd;
  s_socket_buf tmp;
  assert(sb);
  assert(host);
  assert(service);
  if (! libsocket_init())
    return NULL;
  e = getaddrinfo(host->ptr.p_pchar, service->ptr.p_pchar, &hints, &res0);
  if (e) {
    err_write_1("socket_buf_init_connect(");
    err_write_1(host->ptr.p_pchar);
    err_write_1(", ");
    err_write_1(service->ptr.p_pchar);
    err_write_1("): getaddrinfo: ");
    err_inspect_s32_decimal(e);
    err_write_1(" ");
    err_puts((const char *) gai_strerror(e));
    assert(! "socket_buf_init_connect: getaddrinfo");
    return NULL;
  }
  e = 0;
  sockfd = -1;
  res = res0;
  while (res) {
    res_last  = res;
    if (! socket_addr_port(res->ai_addr)) {
      e = errno;
      error_reason = "getaddrinfo port 0: ";
      goto next;
    }
    sockfd = socket(res->ai_family, SOCK_STREAM, res->ai_protocol);
    if (sockfd < 0) {
      e = errno;
      error_reason = "socket: ";
      goto next;
    }
    if ((r = connect(sockfd, res->ai_addr, res->ai_addrlen)) < 0) {
      e = errno;
      error_reason = "connect: ";
      goto next;
    }
    break;
  next:
    if (sockfd >= 0) {
      close(sockfd);
      sockfd = -1;
    }
    res = res->ai_next;
  }
  if (sockfd < 0) {
    err_write_1("socket_buf_init_connect(");
    err_write_1(host->ptr.p_pchar);
    err_write_1(", ");
    err_write_1(service->ptr.p_pchar);
    err_write_1("): ");
    if (res_last) {
      if (res_last->ai_family == AF_INET) {
        struct sockaddr_in *ipv4 = (struct sockaddr_in *) res_last->ai_addr;
        inet_ntop(AF_INET, &(ipv4->sin_addr), ipstr, sizeof(ipstr));
        err_write_1(ipstr);
        err_write_1(" ");
        port = socket_addr_port(res_last->ai_addr);
        err_inspect_s32_decimal(port);
        err_write_1(": ");
      }
      else if (res_last->ai_family == AF_INET6) {
        struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *) res_last->ai_addr;
        inet_ntop(AF_INET6, &(ipv6->sin6_addr), ipstr, sizeof(ipstr));
        err_write_1(ipstr);
        err_write_1(" ");
        port = socket_addr_port(res_last->ai_addr);
        err_inspect_s32_decimal(port);
        err_write_1(": ");
      }
    }
    err_write_1(error_reason);
    err_inspect_s64_decimal(r);
    err_write_1(" ");
    err_inspect_s32_decimal(e);
    err_write_1(" ");
    err_puts(strerror(e));
    assert(! "socket_buf_init_connect");
    freeaddrinfo(res0);
    return NULL;
  }
  if (! socket_buf_init(&tmp, sockfd, res->ai_addr, res->ai_addrlen)) {
    err_puts("socket_buf_init_connect: socket_buf_init");
    assert(! "socket_buf_init_connect: socket_buf_init");
    freeaddrinfo(res0);
    return NULL;
  }
  freeaddrinfo(res0);
  *sb = tmp;
  return sb;
}