1

     2

#include "log.h"

     3

#include "sock.h"

     4

#include "irc_conn.h"

     6

     7

#include <stdlib.h>

     8

#include <stdbool.h>

     9

#include <stdio.h>

    10

#include <getopt.h>

    11

    12

#include <event2/event.h>

    13

    14

    15

#define DEFAULT_HOST "irc.fixme.fi"

    16

#define DEFAULT_PORT "6667"

    17

#define DEFAULT_PORT_SSL "6697"

    18

    26

static struct option options[] = {

    34

};

    35

    36

void usage (const char *exe) 

    37

{

    38

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

    39

    printf("\n");

    40

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

    41

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

    42

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

    43

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

    46

}

    47

    48

int main (int argc, char **argv) 

    49

{

    50

    int opt, option_index;

    51

    struct event_base *ev_base;

    52

    struct sock_stream *sock;

    53

    struct irc_conn *conn;

    54

    struct error_info err;

    55

    56

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

    57

    bool ssl = 0;

    58

    struct irc_conn_config conn_config = {

    59

        .nickname       = "SpBotDev",

    60

        .username       = "spbot-dev",

    61

        .realname       = "SpBot (development version)",

    62

    };

    64

    65

    bool port_default = true;

    66

    67

    // parse options

    68

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

    69

        switch (opt) {

    70

            case 'h':

    71

                usage(argv[0]);

    72

                return EXIT_SUCCESS;

    73

    74

            case 'H':

    75

                hostname = optarg;

    76

                break;

    77

    78

            case 'P':

    79

                portname = optarg;

    80

                port_default = false;

    81

                break;

    82

    83

            case 'S':

    84

                ssl = true;

    85

    86

                if (port_default)

    87

                    portname = DEFAULT_PORT_SSL;

    88

    89

                break;

    98

    99

            case '?':

   100

                usage(argv[0]);

   101

                return EXIT_FAILURE;

   102

        }

   103

    }

   104

   105

    // initialize libevent

   106

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

   107

        FATAL("event_base_new");

   108

   109

    // initialize sock module

   110

    if (sock_init(ev_base, &err))

   111

        FATAL_ERROR(&err, "sock_init");

   112

   113

    // over-simplified connect

   114

    if (ssl) {

   115

        log_info("SSL connecting to [%s]:%s", hostname, portname);

   116

   117

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

   118

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

   119

   120

    } else {

   121

        log_info("connecting to [%s]:%s", hostname, portname);

   122

   123

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

   124

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

   125

   126

    }

   127

   128

    log_info("connected, registering");

   129

   130

    // create the irc connection state

   131

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

   132

        FATAL_ERROR(&err, "irc_conn_create");

   139

   140

    // run event loop

   141

    if (event_base_dispatch(ev_base))

   142

        FATAL("event_base_dispatch");

   143

   144

    // ok, no cleanup

   145

    return 0;

   146

}

   147