1

     2

#include <stdlib.h>

     3

#include <stdbool.h>

     4

#include <err.h>

     5

#include <stdio.h>

     6

#include <getopt.h>

     7

     8

#include <event2/event.h>

     9

    10

#include "sock.h"

    12

    16

    17

static struct option options[] = {

    18

    {"help",            0,  NULL,   'h' },

    19

    {"hostname",        1,  NULL,   'H' },

    20

    {"port",            1,  NULL,   'P' },

    21

    {"ssl",             0,  NULL,   'S' },

    22

    {0,                 0,  0,      0   },

    23

};

    24

    25

void usage (const char *exe) 

    26

{

    27

    printf("Usage: %s [OPTIONS]\n", exe);

    28

    printf("\n");

    29

    printf(" --help / -h            display this message\n");

    30

    printf(" --hostname / -H HOST   set hostname to connect to\n");

    31

    printf(" --port / -P PORT       set service port to connect to\n");

    32

    printf(" --ssl / -S             use SSL\n");

    33

}

    34

    35

int main (int argc, char **argv) 

    36

{

    37

    int opt, option_index;

    38

    struct event_base *ev_base;

    39

    struct sock_stream *sock;

    41

    struct error_info _err;

    42

    44

    bool ssl = 0;

    52

    53

    // parse options

    54

    while ((opt = getopt_long(argc, argv, "hH:P:S", options, &option_index)) != -1) {

    55

        switch (opt) {

    56

            case 'h':

    57

                usage(argv[0]);

    58

                return EXIT_SUCCESS;

    59

    60

            case 'H':

    61

                hostname = optarg;

    62

                break;

    63

    64

            case 'P':

    65

                portname = optarg;

    67

                break;

    68

    69

            case 'S':

    70

                ssl = true;

    75

                break;

    76

    77

            case '?':

    78

                usage(argv[0]);

    79

                return EXIT_FAILURE;

    80

        }

    81

    }

    82

    83

    // initialize libevent

    84

    if ((ev_base = event_base_new()) == NULL)

    85

        err(1, "event_base_new");

    86

    87

    // initialize sock module

    88

    if (sock_init(ev_base, &_err))

    89

        errx(1, "sock_init: %s", error_msg(&_err));

    90

    91

    // over-simplified connect

    92

    if (ssl) {

    93

        if (sock_ssl_connect(&sock, hostname, portname, &_err))

    94

            errx(1, "sock_ssl_connect: %s", error_msg(&_err));

    95

    96

    } else {

    97

        if (sock_tcp_connect(&sock, hostname, portname, &_err))

    98

            errx(1, "sock_tcp_connect: %s", error_msg(&_err));

    99

   100

    }

   101

   105

   106

    // run event loop

   107

    if (event_base_dispatch(ev_base))

   108

        errx(1, "event_base_dispatch");

   109

   110

    // ok, no cleanup

   111

    return 0;

   112

}

   113