1

     2

#include "log.h"

     3

#include "sock.h"

     4

#include "irc_conn.h"

     5

     6

#include <stdlib.h>

     7

#include <stdbool.h>

     8

#include <stdio.h>

     9

#include <getopt.h>

    10

    11

#include <event2/event.h>

    12

    13

    14

#define DEFAULT_HOST "irc.fixme.fi"

    15

#define DEFAULT_PORT "6667"

    16

#define DEFAULT_PORT_SSL "6697"

    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

    36

int main (int argc, char **argv) 

    37

{

    38

    int opt, option_index;

    39

    struct event_base *ev_base;

    40

    struct sock_stream *sock;

    41

    struct irc_conn *conn;

    42

    struct error_info err;

    43

    44

    const char *hostname = DEFAULT_HOST, *portname = DEFAULT_PORT;

    45

    bool ssl = 0;

    46

    struct irc_conn_config conn_config = {

    47

        .nickname       = "SpBotDev",

    48

        .username       = "spbot-dev",

    49

        .realname       = "SpBot (development version)",

    50

    };

    51

    52

    bool port_default = true;

    53

    54

    // parse options

    55

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

    56

        switch (opt) {

    57

            case 'h':

    58

                usage(argv[0]);

    59

                return EXIT_SUCCESS;

    60

    61

            case 'H':

    62

                hostname = optarg;

    63

                break;

    64

    65

            case 'P':

    66

                portname = optarg;

    67

                port_default = false;

    68

                break;

    69

    70

            case 'S':

    71

                ssl = true;

    72

    73

                if (port_default)

    74

                    portname = DEFAULT_PORT_SSL;

    75

    76

                break;

    77

    78

            case '?':

    79

                usage(argv[0]);

    80

                return EXIT_FAILURE;

    81

        }

    82

    }

    83

    84

    // initialize libevent

    85

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

    86

        FATAL("event_base_new");

    87

    88

    // initialize sock module

    89

    if (sock_init(ev_base, &err))

    90

        FATAL_ERROR(&err, "sock_init");

    91

    92

    // over-simplified connect

    93

    if (ssl) {

    96

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

    97

            FATAL_ERROR(&err, "sock_ssl_connect(%s, %s)", hostname, portname);

    98

    99

    } else {

   102

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

   103

            FATAL_ERROR(&err, "sock_tcp_connect(%s, %s)", hostname, portname);

   104

   105

    }

   106

   109

    // create the irc connection state

   110

    if (irc_conn_create(&conn, sock, &conn_config, &err))

   111

        FATAL_ERROR(&err, "irc_conn_create");

   112

   113

    // run event loop

   114

    if (event_base_dispatch(ev_base))

   115

        FATAL("event_base_dispatch");

   116

   117

    // ok, no cleanup

   118

    return 0;

   119

}

   120