libsam3/examples/sam3a/test00.c

/* This program is free software. It comes without any warranty, to
 * the extent permitted by applicable law. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://sam.zoy.org/wtfpl/COPYING for more details.
 *
 * I2P-Bote: 5m77dFKGEq6~7jgtrfw56q3t~SmfwZubmGdyOLQOPoPp8MYwsZ~pfUCwud6LB1EmFxkm4C3CGlzq-hVs9WnhUV
 * we are the Borg. */
#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");
}


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;
  }
  //
  if (sam3aGenerateKeys(&skg, &scbKG, HOST, SAM3A_PORT_DEFAULT) < 0) {
    sam3aCloseSession(&ses);
    fprintf(stderr, "FATAL: can't create keygen session!\n");
    return 1;
  }
  //
  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;
}