ze-client-test.c

Go to the documentation of this file.

/*
 *
 * ze-filter - Mail Server Filter for sendmail
 *
 * Copyright (c) 2001-2018 - Jose-Marcio Martins da Cruz
 *
 *  Auteur     : Jose Marcio Martins da Cruz
 *               jose.marcio.mc@gmail.org
 *
 *  Historique :
 *  Creation     : janvier 2002
 *
 * This program is free software, but with restricted license :
 *
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * More details about ze-filter license can be found at ze-filter
 * web site : http://foss.jose-marcio.org
 */


#include <ze-sys.h>
#include <ze-filter.h>


/* ****************************************************************************
 *                                                                            *
 *                                                                            *
 ******************************************************************************/
#define             DEF_TUPLE         "NET,HOST,FULL"
#define             DEF_TIMES         "0,1000,0,1000"

struct conf_T {
  char               *workdir;
  int                 count;

  char               *times;
  char               *tuples;
};

typedef struct conf_T conf_T;

#define CONF_INITIALIZER   {"/tmp/ze-grey-bench", 10000, DEF_TIMES, DEF_TUPLE}

static char        *workdir = "/var/ze-greyd";

static char        *ntuple = NULL;

static char        *tconst = NULL;

static int          do_test(conf_T * cargs);

static void         usage(char *);


int
main(argc, argv)
     int                 argc;
     char              **argv;
{
  int                 i;

  conf_T              cargs = CONF_INITIALIZER;

  zeLog_SetOutput(TRUE, TRUE);

  ze_logLevel = 7;

  /*
   ** 1. Read configuration parameters
   */
  {
    const char         *args = "ht:dvw:n:l:c:";
    int                 c;

    while ((c = getopt(argc, argv, args)) != -1) {
      switch (c) {
          /*
           * help 
           */
        case 'h':
          usage(argv[0]);
          exit(0);
          break;

          /*
           * time constants 
           */
        case 't':
          tconst = optarg;
          break;

          /*
           * verbose 
           */
        case 'v':
          ze_logLevel++;
          break;

          /*
           * group 
           */
        case 'l':
          ze_logLevel = atoi(optarg);
          break;

        case 'c':
          cargs.count = atoi(optarg);
          break;

          /*
           * Work directory 
           */
        case 'w':
          workdir = optarg;
          break;

        case 'n':
          ntuple = optarg;
          break;

        default:
          usage(argv[0]);
          printf("Error ... \n");
          exit(0);
      }
    }

  }

  configure("ze-client-test", conf_file, FALSE);

  {
    client_T            client = CLIENT_INITIALIZER;
    char                buf[1024];
    int                 i;
    int                 fd;
    char                fname[512];
    int                 tfd;
    FILE               *fin;

    printf("* Connect \n");
    if (!client_connect(&client, "inet:2012@127.0.0.1", 10))
      goto fin;

    printf("* Receive \n");
    memset(buf, 0, sizeof (buf));
    if (!client_recv(&client, buf, sizeof (buf)));

    printf("* Send \n");
    strlcpy(buf, "download pending\r\n", sizeof (buf));
    if (!client_send(&client, buf, strlen(buf)));

    printf("* Receive \n");
    strlcpy(fname, "pending.XXXXXX", sizeof (fname));
    fd = mkstemp(fname);
    for (;;) {
      memset(buf, 0, sizeof (buf));
      if (!client_recv(&client, buf, sizeof (buf)))
        break;
      printf("%s", buf);
      write(fd, buf, strlen(buf));
    }
    close(fd);

    printf("* Send \n");
    strlcpy(buf, "download valid\r\n", sizeof (buf));
    if (!client_send(&client, buf, strlen(buf)));

    printf("* Receive \n");
    strlcpy(fname, "valid.XXXXXX", sizeof (fname));
    fd = mkstemp(fname);
    for (;;) {
      memset(buf, 0, sizeof (buf));
#if 1
      if (!client_readln(&client, buf, sizeof (buf)))
        break;
      printf("%s\n", buf);
      write(fd, buf, strlen(buf));
      write(fd, "\n", 1);
#else
      if (!client_recv(&client, buf, sizeof (buf)))
        break;
      printf("%s", buf);
      write(fd, buf, strlen(buf));
#endif
    }
    close(fd);

    printf("* Send \n");
    strlcpy(buf, "download white\r\n", sizeof (buf));
    if (!client_send(&client, buf, strlen(buf)));

    printf("* Receive \n");
    strlcpy(fname, "white.XXXXXX", sizeof (fname));
    fd = mkstemp(fname);
    for (;;) {
      memset(buf, 0, sizeof (buf));
      if (!client_recv(&client, buf, sizeof (buf)))
        break;
      printf("%s", buf);
      write(fd, buf, strlen(buf));
    }
    close(fd);

    printf("* Send \n");
    strlcpy(buf, "quit\r\n", sizeof (buf));
    if (!client_send(&client, buf, strlen(buf)));

  fin:
    client_disconnect(&client, FALSE);
  }

  return 0;
}



/* ****************************************************************************
 *                                                                            *
 *                                                                            *
 ******************************************************************************/
static int
do_test(conf_T * cargs)
{
  char               *ip = "", *from = "", *to = "", *hostname = "";
  int                 n = 0;
  char                buf[256];

  time_t              ti, tf;

  ip = "1.1.1.1";
  from = "nobody@foss.jose-marcio.org";
  hostname = "nowhere.foss.jose-marcio.org";
  to = "grey@foss.jose-marcio.org";

  ti = zeTime_ms();
  for (n = 0; n < cargs->count; n++) {
    int                 r = GREY_OK;
    char               *s = NULL;
    bool                new = FALSE;
    bool                can_validate = TRUE;

    snprintf(buf, sizeof (buf), "grey-%d@foss.jose-marcio.org", n);
    to = buf;

    r = remote_grey_check(ip, from, to, hostname);
    switch (r) {
      case GREY_OK:
        s = "OK";
        break;
      case GREY_WAIT:
        s = "WAIT";
        break;
      case GREY_ERROR:
        s = "ERROR";
        break;
      case GREY_REJECT:
        s = "REJECT";
        break;
    }
  }
  tf = zeTime_ms();

  ZE_MessageInfo(1, "Entries checked : %5d\n", n);
  ZE_MessageInfo(1, "Time elapsed    : %5d ms", tf - ti);

  return 0;
}



/* ****************************************************************************
 *                                                                            *
 *                                                                            *
 ******************************************************************************/
void
usage(arg)
     char               *arg;
{
  printf("Usage : %s options\n"
         "  %s\n" "  Compiled on %s\n", arg, PACKAGE, __DATE__ " " __TIME__);
}