apps/silcmap/silcmap.c

   1 /*
   2
   3   silcmap.c
   4
   5   Author: Pekka Riikonen <priikone@silcnet.org>
   6
   7   Copyright (C) 2003 Pekka Riikonen
   8
   9   This program is free software; you can redistribute it and/or modify
  10   it under the terms of the GNU General Public License as published by
  11   the Free Software Foundation; version 2 of the License.
  12
  13   This program is distributed in the hope that it will be useful,
  14   but WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16   GNU General Public License for more details.
  17
  18 */
  19
  20 #include "silcincludes.h"
  21 #include "silcclient.h"
  22 #include "silcversion.h"
  23 #include "silcmap.h"
  24
  25 /* Allocates new SilcMap context and the SilcClient in it. */
  26
  27 SilcMap silc_map_alloc(const char *conffile)
  28 {
  29   SilcMap map = silc_calloc(1, sizeof(*map));
  30   if (!map)
  31     return NULL;
  32
  33   /* Allocate client */
  34   map->client = silc_client_alloc(&silc_map_client_ops, NULL, NULL, NULL);
  35   if (!map->client) {
  36     silc_free(map);
  37     return NULL;
  38   }
  39
  40   map->client->username = strdup("silcmap");
  41   map->client->hostname = silc_net_localhost();
  42   map->client->realname = strdup("silcmap");
  43
  44   /* Init the client */
  45   if (!silc_client_init(map->client)) {
  46     silc_client_free(map->client);
  47     silc_free(map);
  48     return NULL;
  49   }
  50
  51   /* Load new key pair if it exists, create if it doesn't. */
  52   if (!silc_load_key_pair("silcmap.pub", "silcmap.prv", "",
  53                           &map->client->pkcs,
  54                           &map->client->public_key,
  55                           &map->client->private_key)) {
  56     /* The keys don't exist.  Let's generate us a key pair then!  There's
  57        nice ready routine for that too.  Let's do 1024 bit RSA key pair. */
  58     if (!silc_create_key_pair("rsa", 1024, "silcmap.pub",
  59                               "silcmap.prv", NULL, "",
  60                               &map->client->pkcs,
  61                               &map->client->public_key,
  62                               &map->client->private_key, FALSE)) {
  63       fprintf(stderr, "Could not create new key pair");
  64       silc_client_free(map->client);
  65       silc_free(map);
  66       return NULL;
  67     }
  68   }
  69
  70   map->conffile = strdup(conffile);
  71
  72   return map;
  73 }
  74
  75 /* Free the SilcMap context and all data in it. */
  76
  77 void silc_map_free(SilcMap map)
  78 {
  79   SilcMapConnection mapconn;
  80   SilcMapCommand cmd;
  81   char *h;
  82
  83   silc_free(map->conffile);
  84   silc_free(map->bitmap);
  85
  86   if (map->client) {
  87     silc_free(map->client->username);
  88     silc_free(map->client->realname);
  89     silc_free(map->client->hostname);
  90     silc_client_free(map->client);
  91   }
  92
  93   if (map->conns) {
  94     silc_dlist_start(map->conns);
  95     while ((mapconn = silc_dlist_get(map->conns)) != SILC_LIST_END) {
  96       silc_dlist_start(mapconn->hostnames);
  97       while ((h = silc_dlist_get(mapconn->hostnames)) != SILC_LIST_END)
  98         silc_free(h);
  99       silc_dlist_uninit(mapconn->hostnames);
 100
 101       silc_dlist_start(mapconn->ips);
 102       while ((h = silc_dlist_get(mapconn->ips)) != SILC_LIST_END)
 103         silc_free(h);
 104       silc_dlist_uninit(mapconn->ips);
 105
 106       silc_dlist_start(mapconn->commands);
 107       while ((cmd = silc_dlist_get(mapconn->commands)) != SILC_LIST_END) {
 108         silc_free(cmd->filename);
 109         silc_free(cmd->text);
 110         silc_free(cmd);
 111       }
 112       silc_dlist_uninit(mapconn->commands);
 113
 114       silc_free(mapconn->public_key);
 115       silc_free(mapconn->country);
 116       silc_free(mapconn->city);
 117       silc_free(mapconn->admin);
 118       silc_free(mapconn->description);
 119       silc_free(mapconn->writemaphtml_url);
 120       silc_free(mapconn->up_color);
 121       silc_free(mapconn->up_text_color);
 122       silc_free(mapconn->down_color);
 123       silc_free(mapconn->down_text_color);
 124       silc_free(mapconn->data.motd);
 125       silc_free(mapconn);
 126     }
 127     silc_dlist_uninit(map->conns);
 128   }
 129
 130   silc_free(map->writemap.filename);
 131   silc_free(map->writehtml.filename);
 132   silc_free(map->writemaphtml.filename);
 133   silc_free(map->writemaphtml.text);
 134   silc_free(map->cut.filename);
 135
 136   silc_free(map);
 137 }
 138
 139 /* Starts the actual silcmap by parsing the commands script. */
 140
 141 SILC_TASK_CALLBACK(silc_map_start)
 142 {
 143   SilcMap map = context;
 144
 145   /* Load default font */
 146   silc_map_load_font(map, "default.fnt");
 147
 148   /* Start command parsing.  Most of the commands are executed when they
 149      are parsed so most of the real magic happens here. */
 150   if (!silc_map_commands_parse(map, map->conffile)) {
 151     /* Program stops */
 152     silc_schedule_stop(map->client->schedule);
 153   }
 154 }
 155
 156 /* Long command line options */
 157 static struct option long_opts[] =
 158 {
 159   { "config-file", 1, NULL, 'f' },
 160   { "debug", 2, NULL, 'd' },
 161   { "help", 0, NULL, 'h' },
 162   { "version", 0, NULL,'V' },
 163
 164   { NULL, 0, NULL, 0 }
 165 };
 166
 167 static void silc_map_usage(void)
 168 {
 169   printf(""
 170 "Usage: silcmap [options]\n"
 171 "\n"
 172 "  Generic Options:\n"
 173 "  -f  --config-file=FILE        Alternate SILC Map configuration file\n"
 174 "  -d  --debug=string            Enable debugging\n"
 175 "  -h  --help                    Display this message and exit\n"
 176 "  -V  --version                 Display version and exit\n"
 177 "\n");
 178   exit(0);
 179 }
 180
 181 int main(int argc, char **argv)
 182 {
 183   SilcMap map;
 184   int opt, option_index;
 185   char *filename = NULL;
 186
 187   if (argc > 1) {
 188     while ((opt = getopt_long(argc, argv, "f:d:hV",
 189                               long_opts, &option_index)) != EOF) {
 190       switch(opt) {
 191         case 'h':
 192           silc_map_usage();
 193           break;
 194         case 'V':
 195           printf("SILC Map, version %s\n", silc_dist_version);
 196           printf("(c) 2003 Pekka Riikonen <priikone@silcnet.org>\n");
 197           exit(0);
 198           break;
 199         case 'd':
 200 #ifdef SILC_DEBUG
 201           silc_debug = TRUE;
 202           silc_debug_hexdump = TRUE;
 203           if (optarg)
 204             silc_log_set_debug_string(optarg);
 205           silc_log_quick = TRUE;
 206 #else
 207           fprintf(stderr,
 208                   "Run-time debugging is not enabled. To enable it recompile\n"
 209                   "the server with --enable-debug configuration option.\n");
 210 #endif
 211           break;
 212         case 'f':
 213           filename = strdup(optarg);
 214           break;
 215         default:
 216           silc_map_usage();
 217           break;
 218       }
 219     }
 220   }
 221
 222   /* Allocate map context */
 223   if (!filename)
 224     filename = strdup("silcmap.conf");
 225   map = silc_map_alloc(filename);
 226   if (!map)
 227     return 1;
 228
 229   /* Schedule for command script parsing */
 230   silc_schedule_task_add(map->client->schedule, 0,
 231                          silc_map_start, map, 0, 1,
 232                          SILC_TASK_TIMEOUT, SILC_TASK_PRI_NORMAL);
 233
 234   /* Run the silcmap client */
 235   silc_client_run(map->client);
 236
 237   /* Cleanup */
 238   silc_client_stop(map->client);
 239   silc_map_free(map);
 240   silc_free(filename);
 241
 242   return 0;
 243 }