libsam3/examples/sam3a/test00.c

/*
 * Copyright © 2023 I2P
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files
 * (the “Software”), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
 * OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 * http://git.idk.i2p/i2p-hackers/libsam3/
 */

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>

#include "../libsam3a/libsam3a.h"

////////////////////////////////////////////////////////////////////////////////
static void scbErrorClose(Sam3ASession *ses) {
  fprintf(stderr,
          "\n===============================\nSESION_ERROR: "
          "[%s]\n===============================\n",
          ses->error);
  sam3aCloseSession(ses); // it's safe here
}

static void scbNRCreated(Sam3ASession *ses) {
  fprintf(stderr, "\n===============================\nNAME RESOLVED: [%s]\n",
          ses->params);
  fprintf(stderr, "PUB: %s\n===============================\n", ses->destkey);
  sam3aCloseSession(ses); // it's safe here
}

static const Sam3ASessionCallbacks scbNR = {
    .cbError = scbErrorClose,
    .cbCreated = scbNRCreated,
    .cbDisconnected = NULL,
    .cbDatagramRead = NULL,
    .cbDestroy = NULL,
};

////////////////////////////////////////////////////////////////////////////////
static void scbKGCreated(Sam3ASession *ses) {
  fprintf(stderr, "\n===============================\nKEYS GENERATED\n");
  fprintf(stderr, "\rPRIV: %s\n", ses->privkey);
  fprintf(stderr, "\nPUB: %s\n===============================\n", ses->pubkey);
  sam3aCloseSession(ses); // it's safe here
}

static const Sam3ASessionCallbacks scbKG = {
    .cbError = scbErrorClose,
    .cbCreated = scbKGCreated,
    .cbDisconnected = NULL,
    .cbDatagramRead = NULL,
    .cbDestroy = NULL,
};

////////////////////////////////////////////////////////////////////////////////
static void scbError(Sam3ASession *ses) {
  fprintf(stderr,
          "\n===============================\nSESION_ERROR: "
          "[%s]\n===============================\n",
          ses->error);
}

static void scbCreated(Sam3ASession *ses) {
  fprintf(stderr, "\n===============================\nSESION_CREATED\n");
  fprintf(stderr, "\rPRIV: %s\n", ses->privkey);
  fprintf(stderr, "\nPUB: %s\n===============================\n", ses->pubkey);
  sam3aCancelSession(ses); // it's safe here
}

static void scbDisconnected(Sam3ASession *ses) {
  fprintf(stderr, "\n===============================\nSESION_DISCONNECTED\n===="
                  "===========================\n");
}

static void scbDGramRead(Sam3ASession *ses, const void *buf, int bufsize) {
  fprintf(stderr, "\n===============================\nSESION_DATAGRAM_READ\n==="
                  "============================\n");
}

static void scbDestroy(Sam3ASession *ses) {
  fprintf(stderr, "\n===============================\nSESION_DESTROYED\n======="
                  "========================\n");
}

/** callbacks for our SAM session */
static const Sam3ASessionCallbacks scb = {
    .cbError = scbError,
    .cbCreated = scbCreated,
    .cbDisconnected = scbDisconnected,
    .cbDatagramRead = scbDGramRead,
    .cbDestroy = scbDestroy,
};

////////////////////////////////////////////////////////////////////////////////
#define HOST SAM3A_HOST_DEFAULT
//#define HOST  "google.com"

int main(int argc, char *argv[]) {
  Sam3ASession ses, snr, skg;
  //
  // libsam3a_debug = 1;
  //
  if (sam3aCreateSession(&ses, &scb, HOST, SAM3A_PORT_DEFAULT,
                         SAM3A_DESTINATION_TRANSIENT,
                         SAM3A_SESSION_STREAM) < 0) {
    fprintf(stderr, "FATAL: can't create main session!\n");
    return 1;
  }
  // generate keys
  if (sam3aGenerateKeys(&skg, &scbKG, HOST, SAM3A_PORT_DEFAULT) < 0) {
    sam3aCloseSession(&ses);
    fprintf(stderr, "FATAL: can't create keygen session!\n");
    return 1;
  }
  // do a name lookup for zzz.i2p
  if (sam3aNameLookup(&snr, &scbNR, HOST, SAM3A_PORT_DEFAULT, "zzz.i2p") < 0) {
    sam3aCloseSession(&skg);
    sam3aCloseSession(&ses);
    fprintf(stderr, "FATAL: can't create name resolving session!\n");
    return 1;
  }
  // while we have sessions ...
  while (sam3aIsActiveSession(&ses) || sam3aIsActiveSession(&snr) ||
         sam3aIsActiveSession(&skg)) {
    fd_set rds, wrs;
    int res, maxfd = 0;
    struct timeval to;
    // set up file descriptors for select()
    FD_ZERO(&rds);
    FD_ZERO(&wrs);
    // obtain the maximum fd for select()
    if (sam3aIsActiveSession(&ses) &&
        (maxfd = sam3aAddSessionToFDS(&ses, -1, &rds, &wrs)) < 0)
      break;
    if (sam3aIsActiveSession(&snr) &&
        (maxfd = sam3aAddSessionToFDS(&snr, -1, &rds, &wrs)) < 0)
      break;
    if (sam3aIsActiveSession(&skg) &&
        (maxfd = sam3aAddSessionToFDS(&skg, -1, &rds, &wrs)) < 0)
      break;
    // set timeout to 1 second
    sam3ams2timeval(&to, 1000);
    // call select()
    res = select(maxfd + 1, &rds, &wrs, NULL, &to);
    if (res < 0) {
      if (errno == EINTR)
        continue;
      fprintf(stderr, "FATAL: select() error!\n");
      break;
    }
    if (res == 0) {
      // idle, no activity
      fprintf(stdout, ".");
      fflush(stdout);
    } else {
      // we have activity, process io
      if (sam3aIsActiveSession(&ses))
        sam3aProcessSessionIO(&ses, &rds, &wrs);
      if (sam3aIsActiveSession(&snr))
        sam3aProcessSessionIO(&snr, &rds, &wrs);
      if (sam3aIsActiveSession(&skg))
        sam3aProcessSessionIO(&skg, &rds, &wrs);
    }
  }
  // close seessions
  sam3aCloseSession(&ses);
  sam3aCloseSession(&skg);
  sam3aCloseSession(&snr);
  // exit
  return 0;
}