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"

    11

#include "line_proto.h"

    13

    14

#define CONNECT_HOST "irc.fixme.fi"

    15

#define CONNECT_SERV "6697"

    16

#define LINE_LENGTH 512

    17

    18

static struct option options[] = {

    19

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

    20

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

    21

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

    22

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

    23

    {0,                 0,  0,      0   },

    24

};

    25

    26

void usage (const char *exe) 

    27

{

    28

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

    29

    printf("\n");

    30

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

    31

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

    32

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

    33

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

    34

}

    35

    37

{

    49

}

    50

    51

int main (int argc, char **argv) 

    52

{

    53

    int opt, option_index;

    54

    struct event_base *ev_base;

    55

    struct sock_stream *sock;

    56

    struct line_proto *lp;

    57

    struct error_info _err;

    58

    59

    const char *hostname = CONNECT_HOST, *portname = CONNECT_SERV;

    60

    bool ssl = 0;

    61

    62

    // parse options

    63

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

    64

        switch (opt) {

    65

            case 'h':

    66

                usage(argv[0]);

    67

                return EXIT_SUCCESS;

    68

    69

            case 'H':

    70

                hostname = optarg;

    71

                break;

    72

    73

            case 'P':

    74

                portname = optarg;

    75

                break;

    76

    77

            case 'S':

    78

                ssl = true;

    79

                break;

    80

    81

            case '?':

    82

                usage(argv[0]);

    83

                return EXIT_FAILURE;

    84

        }

    85

    }

    86

    87

    // initialize libevent

    88

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

    89

        err(1, "event_base_new");

    90

    91

    // initialize sock module

    92

    if (sock_init(ev_base, &_err))

    93

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

    94

    95

    // over-simplified connect

    96

    if (ssl) {

    97

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

    98

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

    99

   100

    } else {

   101

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

   102

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

   103

   104

    }

   105

   106

    // line protocol, with safety margin for buffer

   107

    if (line_proto_create(&lp, sock, LINE_LENGTH * 2, on_line, NULL, &_err))

   108

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

   109

   110

    // run event loop

   111

    if (event_base_dispatch(ev_base))

   112

        errx(1, "event_base_dispatch");

   113

   114

    // ok, no cleanup

   115

    return 0;

   116

}

   117