Edit

kc3-lang/kc3/libkc3/tls_client.c

Branch :

  • libkc3/tls_client.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 <unistd.h>
#include <tls.h>
#include "kc3.h"
#include "socket.h"
#include "tls_buf.h"
#include "tls_client.h"

void kc3_tls_client_clean (s_tls_client *tls_client)
{
  s_tls_buf *tls_buf;
  buf_flush(tls_client->socket_buf->buf_rw.w);
  tls_buf = tls_client->socket_buf->buf_rw.r->user_ptr;
  tls_close(tls_buf->ctx);
  tls_buf_close(tls_client->socket_buf->buf_rw.r);
  tls_buf_close(tls_client->socket_buf->buf_rw.w);
  buf_rw_clean(&tls_client->socket_buf->buf_rw);
  close(tls_client->socket_buf->sockfd);
  alloc_free(tls_client->socket_buf);
}

s_tls_client * kc3_tls_client_init_connect (s_tls_client *tls_client,
                                            p_tls *ctx,
                                            const s_str *host,
                                            const s_str *port)
{
  bool r_open = false;
  sw r;
  bool w_open = false;
  assert(tls_client);
  assert(ctx);
  assert(*ctx);
  assert(host);
  assert(port);
  *tls_client = (s_tls_client) {0};
  tls_client->socket_buf = alloc(sizeof(s_socket_buf));
  if (! tls_client->socket_buf)
    return NULL;
  if (! socket_init_connect(&tls_client->socket_buf->sockfd, host,
                            port)) {
    alloc_free(tls_client->socket_buf);
    err_puts("kc3_tls_client_init_connect: socket_init_connect");
    assert(! "kc3_tls_client_init_connect: socket_init_connect");
    return NULL;
  }
  if (tls_connect_socket(*ctx, tls_client->socket_buf->sockfd,
                         host->ptr.p_pchar)) {
    err_write_1("kc3_tls_client_init_connect: tls_connect_socket: ");
    err_puts(tls_error(*ctx));
    assert(! "kc3_tls_client_init_connect: tls_connect_socket");
    goto clean;
  }
  if (! (tls_client->socket_buf->buf_rw.r = buf_new_alloc(BUF_SIZE))) {
    err_puts("kc3_tls_client_init_connect: buf_new_alloc r");
    assert(! "kc3_tls_client_init_connect: buf_new_alloc r");
    goto clean;
  }
  if (! (tls_client->socket_buf->buf_rw.w = buf_new_alloc(BUF_SIZE))) {
    err_puts("kc3_tls_client_init_connect: buf_new_alloc w");
    assert(! "kc3_tls_client_init_connect: buf_new_alloc w");
    goto clean;
  }
  while ((r = tls_handshake(*ctx)) == TLS_WANT_POLLIN ||
         r == TLS_WANT_POLLOUT)
    ;
  if (r) {
    err_write_1("kc3_tls_client_init_connect: tls_handshake: ");
    err_puts(tls_error(*ctx));
    assert(! "kc3_tls_client_init_connect: tls_handshake");
    goto clean;
  }
  if (! tls_buf_open_r(tls_client->socket_buf->buf_rw.r, *ctx)) {
    err_puts("kc3_tls_client_init_connect: tls_buf_open_r");
    assert(! "kc3_tls_client_init_connect: tls_buf_open_r");
    goto clean;
  }
  r_open = true;
  if (! tls_buf_open_w(tls_client->socket_buf->buf_rw.w, *ctx)) {
    err_puts("kc3_tls_client_init_connect: ");
    assert(! "kc3_tls_client_init_connect: ");
    goto clean;
  }
  w_open = true;
  return tls_client;
 clean:
  if (w_open)
    tls_buf_close(tls_client->socket_buf->buf_rw.w);
  if (r_open)
    tls_buf_close(tls_client->socket_buf->buf_rw.r);
  if (tls_client->socket_buf->buf_rw.w)
    buf_delete(tls_client->socket_buf->buf_rw.w);
  if (tls_client->socket_buf->buf_rw.r)
    buf_delete(tls_client->socket_buf->buf_rw.r);
  tls_close(*ctx);
  close(tls_client->socket_buf->sockfd);
  alloc_free(tls_client->socket_buf);
  tls_client->socket_buf = NULL;
  return NULL;
}

void kc3_tls_client_delete (s_tls_client **tls_client)
{
  assert(tls_client);
  assert(*tls_client);
  kc3_tls_client_clean(*tls_client);
  alloc_free(*tls_client);
}

s_tls_client ** kc3_tls_client_new_connect (s_tls_client **tls_client,
                                            p_tls *ctx,
                                            const s_str *host,
                                            const s_str *port)
{
  s_tls_client *tmp;
  assert(tls_client);
  tmp = alloc(sizeof(s_tls_client));
  if (! tmp)
    return NULL;
  if (! kc3_tls_client_init_connect(tmp, ctx, host, port)) {
    alloc_free(tmp);
    return NULL;
  }
  *tls_client = tmp;
  return tls_client;
}