client_sample.c

/**
 * Sample to show how to consume HTTPS-notif protocol using the library
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>

#include "../src/unyte_https_collector.h"
#include "../src/unyte_https_utils.h"

#define MAX_TO_RECEIVE 10
#define SERVERKEYFILE "private.key"      // Should be generated before run this sample
#define SERVERCERTFILE "certificate.pem" // Should be generated before run this sample

/**
 * Read file and return bytes
 */
char *load_file(const char *file_path)
{
  FILE *infile;
  char *buffer;
  long numbytes;

  infile = fopen(file_path, "r");

  if (infile == NULL)
    return NULL;

  fseek(infile, 0L, SEEK_END);
  numbytes = ftell(infile);
  rewind(infile);

  buffer = (char *)malloc(sizeof(char) * (numbytes + 1));

  if (buffer == NULL)
    return NULL;

  fread(buffer, sizeof(char), numbytes, infile);
  fclose(infile);
  buffer[numbytes] = '\0';
  return buffer;
}

int main(int argc, char *argv[])
{
  if (argc != 3)
  {
    printf("Error: arguments not valid\n");
    printf("Usage: ./client_sample <ip> <port>\n");
    exit(1);
  }

  char *key_pem = load_file(SERVERKEYFILE);
  char *cert_pem = load_file(SERVERCERTFILE);

  if ((key_pem == NULL) || (cert_pem == NULL))
  {
    printf("The key/certificate files could not be read.\n");
    return 1;
  }

  unyte_https_options_t options = {0};
  options.address = argv[1];
  options.port = argv[2];
  options.cert_pem = cert_pem;
  options.key_pem = key_pem;
  options.disable_xml_encoding = true;
  options.disable_json_encoding = false;

  unyte_https_collector_t *collector = unyte_https_start_collector(&options);
  printf("Starting collector on %s:%s\n", options.address, options.port);

  uint count = 0;
  while (count < MAX_TO_RECEIVE)
  {
    void *res = unyte_https_queue_read(collector->queue);
    unyte_https_msg_met_t *msg = (unyte_https_msg_met_t *)res;

    print_https_notif_msg(msg, stdout);

    // printf("unyte_https_get_src: %s\n", unyte_https_get_src(msg)->ss_family == AF_INET ? "IPv4" : "IPv6");
    // printf("unyte_https_get_payload: %s\n", unyte_https_get_payload(msg));
    // printf("unyte_https_get_payload_length: %lu\n", unyte_https_get_payload_length(msg));
    // printf("unyte_https_get_content_type: %s\n", unyte_https_get_content_type(msg));

    fflush(stdout);

    // Freeing struct
    unyte_https_free_msg(msg);
    count++;
  }

  // Stopping the collector and the https server
  unyte_https_stop_collector(collector);
  // Freeing all dynamic mallocs
  unyte_https_free_collector(collector);

  free(key_pem);
  free(cert_pem);
  return 0;
}