yp.c 26.5 KB
Newer Older
1 2 3 4 5
/* Icecast
 *
 * This program is distributed under the GNU General Public License, version 2.
 * A copy of this license is included with this source.
 *
6
 * Copyright 2000-2004, Jack Moffitt <jack@xiph.org,
7 8 9 10 11 12
 *                      Michael Smith <msmith@xiph.org>,
 *                      oddsock <oddsock@xiph.org>,
 *                      Karl Heyes <karl@xiph.org>
 *                      and others (see AUTHORS for details).
 */

13
/* -*- c-basic-offset: 4; indent-tabs-mode: nil; -*- */
14 15 16 17
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

18 19
#include <stdio.h>
#include <string.h>
20
#include <stdlib.h>
21

22
#include "common/thread/thread.h"
23

24
#include "yp.h"
25
#include "global.h"
26
#include "curl.h"
27 28
#include "logging.h"
#include "source.h"
29
#include "cfgfile.h"
30
#include "stats.h"
31 32 33 34

#ifdef WIN32
#define snprintf _snprintf
#endif
35

36
#define CATMODULE "yp"
37

38 39 40
struct yp_server
{
    char        *url;
41
    char        *server_id;
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
    unsigned    url_timeout;
    unsigned    touch_interval;
    int         remove;

    CURL *curl;
    struct ypdata_tag *mounts, *pending_mounts;
    struct yp_server *next;
    char curl_error[CURL_ERROR_SIZE];
};



typedef struct ypdata_tag
{
    int remove;
57
    int release;
58 59 60 61 62 63 64 65 66 67 68 69 70 71
    int cmd_ok;

    char *sid;
    char *mount;
    char *url;
    char *listen_url;
    char *server_name;
    char *server_desc;
    char *server_genre;
    char *cluster_password;
    char *bitrate;
    char *audio_info;
    char *server_type;
    char *current_song;
72
    char *subtype;
73 74

    struct yp_server *server;
75 76 77 78
    time_t next_update;
    unsigned touch_interval;
    char *error_msg;
    int (*process)(struct ypdata_tag *yp, char *s, unsigned len);
79 80 81 82 83 84 85 86

    struct ypdata_tag *next;
} ypdata_t;


static rwlock_t yp_lock;
static mutex_t yp_pending_lock;

Karl Heyes's avatar
Karl Heyes committed
87
static volatile struct yp_server *active_yps = NULL, *pending_yps = NULL;
88
static volatile int yp_update = 0;
89 90 91
static int yp_running;
static time_t now;
static thread_type *yp_thread;
Karl Heyes's avatar
Karl Heyes committed
92
static volatile unsigned client_limit = 0;
93
static volatile char *server_version = NULL;
94 95

static void *yp_update_thread(void *arg);
96 97 98 99
static void add_yp_info(ypdata_t *yp, void *info, int type);
static int do_yp_remove(ypdata_t *yp, char *s, unsigned len);
static int do_yp_add(ypdata_t *yp, char *s, unsigned len);
static int do_yp_touch(ypdata_t *yp, char *s, unsigned len);
100 101
static void add_pending_yp (struct yp_server *server);
static void delete_marked_yp(struct yp_server *server);
102 103 104 105
static void yp_destroy_ypdata(ypdata_t *ypdata);


/* curl callback used to parse headers coming back from the YP server */
106
static size_t handle_returned_header (void *ptr, size_t size, size_t nmemb, void *stream)
107 108
{
    ypdata_t *yp = stream;
109
    size_t bytes = size * nmemb;
110

111
    /* ICECAST_LOG_DEBUG("header from YP is \"%.*s\"", bytes, ptr); */
112
    if (strncasecmp (ptr, "YPResponse: 1", 13) == 0)
113 114
        yp->cmd_ok = 1;

115
    if (strncasecmp (ptr, "YPMessage: ", 11) == 0)
116
    {
117
        size_t len = bytes - 11;
118 119 120
        free (yp->error_msg);
        yp->error_msg = calloc (1, len);
        if (yp->error_msg)
121
            sscanf (ptr + 11, "%[^\r\n]", yp->error_msg);
122 123 124 125
    }

    if (yp->process == do_yp_add)
    {
126
        if (strncasecmp (ptr, "SID: ", 5) == 0)
127
        {
128
            size_t len = bytes - 5;
129 130 131
            free (yp->sid);
            yp->sid = calloc (1, len);
            if (yp->sid)
132
                sscanf (ptr + 5, "%[^\r\n]", yp->sid);
133
        }
134
    }
135
    if (strncasecmp (ptr, "TouchFreq: ", 11) == 0)
136 137
    {
        unsigned secs;
138 139
        if ( sscanf (ptr + 11, "%u", &secs) != 1 )
            secs = 0;
140 141
        if (secs < 30)
            secs = 30;
142
        ICECAST_LOG_DEBUG("server touch interval is %u", secs);
143
        yp->touch_interval = secs;
144
    }
145
    return (size_t)bytes;
146 147 148 149 150 151 152 153
}


/* search the active and pending YP server lists */
static struct yp_server *find_yp_server (const char *url)
{
    struct yp_server *server;

154
    server = (struct yp_server *)active_yps;
155 156 157 158 159 160
    while (server)
    {
        if (strcmp (server->url, url) == 0)
            return server;
        server = server->next;
    }
161
    server = (struct yp_server *)pending_yps;
162 163 164 165 166 167 168 169 170 171 172 173
    while (server)
    {
        if (strcmp (server->url, url) == 0)
            break;
        server = server->next;
    }
    return server;
}


static void destroy_yp_server (struct yp_server *server)
{
174 175
    ypdata_t *yp;

176 177
    if (server == NULL)
        return;
178
    ICECAST_LOG_DEBUG("Removing YP server entry for %s", server->url);
179 180 181 182 183 184 185 186 187 188 189 190 191 192

    /* delete yps:
     * first move all pendings into main queue.
     * then mark all main queue entries for deleting.
     * then remove all marked entries.
     */
    add_pending_yp(server);
    yp = server->mounts;
    while (yp) {
        yp->remove = 1;
        yp = yp->next;
    }
    delete_marked_yp(server);

193
    icecast_curl_free(server->curl);
194 195
    if (server->mounts) ICECAST_LOG_WARN("active ypdata not freed");
    if (server->pending_mounts) ICECAST_LOG_WARN("pending ypdata not freed");
196
    free (server->url);
197
    free (server->server_id);
198 199 200 201 202 203
    free (server);
}



/* search for a ypdata entry corresponding to a specific mountpoint */
204
static ypdata_t *find_yp_mount (ypdata_t *mounts, const char *mount)
205
{
206
    ypdata_t *yp = mounts;
207 208 209 210 211 212 213 214 215
    while (yp)
    {
        if (strcmp (yp->mount, mount) == 0)
            break;
        yp = yp->next;
    }
    return yp;
}

216 217 218

void yp_recheck_config (ice_config_t *config)
{
219 220 221
    int i;
    struct yp_server *server;

222
    ICECAST_LOG_DEBUG("Updating YP configuration");
223 224
    thread_rwlock_rlock (&yp_lock);

225
    server = (struct yp_server *)active_yps;
226 227 228 229 230
    while (server)
    {
        server->remove = 1;
        server = server->next;
    }
Karl Heyes's avatar
Karl Heyes committed
231
    client_limit = config->client_limit;
232 233
    free ((char*)server_version);
    server_version = strdup (config->server_id);
234
    /* for each yp url in config, check to see if one exists
235 236 237 238 239 240 241 242 243 244 245 246 247
       if not, then add it. */
    for (i=0 ; i < config->num_yp_directories; i++)
    {
        server = find_yp_server (config->yp_url[i]);
        if (server == NULL)
        {
            server = calloc (1, sizeof (struct yp_server));

            if (server == NULL)
            {
                destroy_yp_server (server);
                break;
            }
248
            server->server_id = strdup ((char *)server_version);
249 250
            server->url = strdup (config->yp_url[i]);
            server->url_timeout = config->yp_url_timeout[i];
251
            server->touch_interval = config->yp_touch_interval[i];
252
            server->curl = icecast_curl_new(server->url, &(server->curl_error[0]));
253 254 255 256 257
            if (server->curl == NULL)
            {
                destroy_yp_server (server);
                break;
            }
258 259
            if (server->url_timeout > 10 || server->url_timeout < 1)
                server->url_timeout = 6;
260 261 262
            if (server->touch_interval < 30)
                server->touch_interval = 30;
            curl_easy_setopt (server->curl, CURLOPT_HEADERFUNCTION, handle_returned_header);
263
            server->next = (struct yp_server *)pending_yps;
264
            pending_yps = server;
265
            ICECAST_LOG_INFO("Adding new YP server \"%s\" (timeout %ds, default interval %ds)",
266 267 268 269 270 271 272 273 274
                    server->url, server->url_timeout, server->touch_interval);
        }
        else
        {
            server->remove = 0;
        }
    }
    thread_rwlock_unlock (&yp_lock);
    yp_update = 1;
275 276
}

277

278
void yp_initialize(void)
279
{
280
    ice_config_t *config = config_get_config();
281 282
    thread_rwlock_create (&yp_lock);
    thread_mutex_create (&yp_pending_lock);
283 284
    yp_recheck_config (config);
    config_release_config ();
285 286
    yp_thread = thread_create("YP Touch Thread", yp_update_thread,
                            (void *)NULL, THREAD_ATTACHED);
287
}
288 289 290



291
/* handler for curl, checks if successful handling occurred
292 293
 * return 0 for ok, -1 for this entry failed, -2 for server fail.
 * On failure case, update and process are modified
294
 */
295
static int send_to_yp (const char *cmd, ypdata_t *yp, char *post)
296
{
297 298 299
    int curlcode;
    struct yp_server *server = yp->server;

300
    /* ICECAST_LOG_DEBUG("send YP (%s):%s", cmd, post); */
301 302 303 304 305 306 307
    yp->cmd_ok = 0;
    curl_easy_setopt (server->curl, CURLOPT_POSTFIELDS, post);
    curl_easy_setopt (server->curl, CURLOPT_WRITEHEADER, yp);
    curlcode = curl_easy_perform (server->curl);
    if (curlcode)
    {
        yp->process = do_yp_add;
308
        yp->next_update = now + 1200;
309
        ICECAST_LOG_ERROR("connection to %s failed with \"%s\"", server->url, server->curl_error);
310
        return -2;
311
    }
312 313
    if (yp->cmd_ok == 0)
    {
314 315
        if (yp->error_msg == NULL)
            yp->error_msg = strdup ("no response from server");
316 317
        if (yp->process == do_yp_add)
        {
318
            ICECAST_LOG_ERROR("YP %s on %s failed: %s", cmd, server->url, yp->error_msg);
319
            yp->next_update = now + 7200;
320
        }
321
        if (yp->process == do_yp_touch)
322
        {
323 324 325 326 327 328 329
            /* At this point the touch request failed, either because they rejected our session
             * or the server isn't accessible. This means we have to wait before doing another
             * add request. We have a minimum delay but we could allow the directory server to
             * give us a wait time using the TouchFreq header. This time could be given in such
             * cases as a firewall block or incorrect listenurl.
             */
            if (yp->touch_interval < 1200)
330
                yp->next_update = now + 1200;
331
            else
332
                yp->next_update = now + yp->touch_interval;
333
            ICECAST_LOG_INFO("YP %s on %s failed: %s", cmd, server->url, yp->error_msg);
334
        }
335
        yp->process = do_yp_add;
336
        free(yp->sid);
337
        yp->sid = NULL;
338
        return -1;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
339
    }
340
    ICECAST_LOG_DEBUG("YP %s at %s succeeded", cmd, server->url);
341 342 343 344 345
    return 0;
}


/* routines for building and issues requests to the YP server */
346
static int do_yp_remove (ypdata_t *yp, char *s, unsigned len)
347
{
348 349
    int ret = 0;

350 351
    if (yp->sid)
    {
352
        ret = snprintf (s, len, "action=remove&sid=%s", yp->sid);
353 354 355
        if (ret >= (signed)len)
            return ret+1;

356
        ICECAST_LOG_INFO("clearing up YP entry for %s", yp->mount);
357
        ret = send_to_yp ("remove", yp, s);
358 359
        free (yp->sid);
        yp->sid = NULL;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
360
    }
361 362
    yp->remove = 1;
    yp->process = do_yp_add;
363
    yp_update = 1;
364

365
    return ret;
366 367
}

368

369
static int do_yp_add (ypdata_t *yp, char *s, unsigned len)
370
{
371 372
    int ret;
    char *value;
373 374 375 376 377 378
    ice_config_t *config;
    char *admin;

    config = config_get_config();
    admin = util_url_escape(config->admin);
    config_release_config();
379

380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
    value = stats_get_value (yp->mount, "server_type");
    add_yp_info (yp, value, YP_SERVER_TYPE);
    free (value);

    value = stats_get_value (yp->mount, "server_name");
    add_yp_info (yp, value, YP_SERVER_NAME);
    free (value);

    value = stats_get_value (yp->mount, "server_url");
    add_yp_info (yp, value, YP_SERVER_URL);
    free (value);

    value = stats_get_value (yp->mount, "genre");
    add_yp_info (yp, value, YP_SERVER_GENRE);
    free (value);

    value = stats_get_value (yp->mount, "bitrate");
397 398
    if (value == NULL)
        value = stats_get_value (yp->mount, "ice-bitrate");
399 400 401 402 403 404 405
    add_yp_info (yp, value, YP_BITRATE);
    free (value);

    value = stats_get_value (yp->mount, "server_description");
    add_yp_info (yp, value, YP_SERVER_DESC);
    free (value);

406
    value = stats_get_value (yp->mount, "subtype");
407 408 409 410 411 412
    add_yp_info (yp, value, YP_SUBTYPE);
    free (value);

    value = stats_get_value (yp->mount, "audio_info");
    add_yp_info (yp, value, YP_AUDIO_INFO);
    free (value);
413

414
    ret = snprintf (s, len, "action=add&admin=%s&sn=%s&genre=%s&cpswd=%s&desc="
415
                    "%s&url=%s&listenurl=%s&type=%s&stype=%s&b=%s&%s\r\n",
416
                    admin,
417 418
                    yp->server_name, yp->server_genre, yp->cluster_password,
                    yp->server_desc, yp->url, yp->listen_url,
419
                    yp->server_type, yp->subtype, yp->bitrate, yp->audio_info);
420 421
    free(admin);

422 423
    if (ret >= (signed)len)
        return ret+1;
424 425
    ret = send_to_yp ("add", yp, s);
    if (ret == 0)
426 427 428 429 430
    {
        yp->process = do_yp_touch;
        /* force first touch in 5 secs */
        yp->next_update = time(NULL) + 5;
    }
431
    return ret;
432 433 434
}


435
static int do_yp_touch (ypdata_t *yp, char *s, unsigned len)
436
{
Karl Heyes's avatar
Karl Heyes committed
437
    unsigned listeners = 0, max_listeners = 1;
438 439 440 441 442 443 444 445
    char *val, *artist, *title;
    int ret;

    artist = (char *)stats_get_value (yp->mount, "artist");
    title = (char *)stats_get_value (yp->mount, "title");
    if (artist || title)
    {
         char *song;
446
         const char *separator = " - ";
447 448 449 450 451 452 453 454 455 456
         if (artist == NULL)
         {
             artist = strdup("");
             separator = "";
         }
         if (title == NULL) title = strdup("");
         song = malloc (strlen (artist) + strlen (title) + strlen (separator) +1);
         if (song)
         {
             sprintf (song, "%s%s%s", artist, separator, title);
457 458
             add_yp_info(yp, song, YP_CURRENT_SONG);
             stats_event (yp->mount, "yp_currently_playing", song);
459 460 461 462 463 464 465 466 467 468 469 470
             free (song);
         }
    }
    free (artist);
    free (title);

    val = (char *)stats_get_value (yp->mount, "listeners");
    if (val)
    {
        listeners = atoi (val);
        free (val);
    }
Karl Heyes's avatar
Karl Heyes committed
471
    val = stats_get_value (yp->mount, "max_listeners");
472
    if (val == NULL || strcmp (val, "unlimited") == 0 || atoi(val) < 0)
Karl Heyes's avatar
Karl Heyes committed
473 474 475
        max_listeners = client_limit;
    else
        max_listeners = atoi (val);
476
    free (val);
Karl Heyes's avatar
Karl Heyes committed
477

478 479 480
    val = stats_get_value (yp->mount, "subtype");
    if (val)
    {
481
        add_yp_info (yp, val, YP_SUBTYPE);
482 483
        free (val);
    }
484 485

    ret = snprintf (s, len, "action=touch&sid=%s&st=%s"
Karl Heyes's avatar
Karl Heyes committed
486
            "&listeners=%u&max_listeners=%u&stype=%s\r\n",
487
            yp->sid, yp->current_song, listeners, max_listeners, yp->subtype);
488 489 490 491

    if (ret >= (signed)len)
        return ret+1; /* space required for above text and nul*/

492 493 494 495 496 497
    if (send_to_yp ("touch", yp, s) == 0)
    {
        yp->next_update = now + yp->touch_interval;
        return 0;
    }
    return -1;
498
}
499

500 501


502
static int process_ypdata(struct yp_server *server, ypdata_t *yp)
503
{
504
    unsigned len = 1024;
505
    char *s = NULL, *tmp;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
506

507
    if (now < yp->next_update)
508
        return 0;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
509

510 511 512
    /* loop just in case the memory area isn't big enough */
    while (1)
    {
513
        int ret;
514
        if ((tmp = realloc(s, len)) == NULL)
515
            return 0;
516
        s = tmp;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
517

518 519 520 521 522 523
        if (yp->release)
        {
            yp->process = do_yp_remove;
            yp->next_update = 0;
        }

524
        ret = yp->process (yp, s, len);
525
        if (ret <= 0)
526 527
        {
           free (s);
528
           return ret;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
529
        }
530
        len = ret;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
531
    }
532
    return 0;
533
}
534 535 536


static void yp_process_server (struct yp_server *server)
537
{
538
    ypdata_t *yp;
539
    int state = 0;
540

541
    /* ICECAST_LOG_DEBUG("processing yp server %s", server->url); */
542 543 544 545
    yp = server->mounts;
    while (yp)
    {
        now = time (NULL);
546 547 548 549 550
        /* if one of the streams shows that the server cannot be contacted then mark the
         * other entries for an update later. Assume YP server is dead and skip it for now
         */
        if (state == -2)
        {
551
            ICECAST_LOG_DEBUG("skipping %s on %s", yp->mount, server->url);
552 553 554 555 556
            yp->process = do_yp_add;
            yp->next_update += 900;
        }
        else
            state = process_ypdata (server, yp);
557 558 559 560 561 562
        yp = yp->next;
    }
}



563
static ypdata_t *create_yp_entry (const char *mount)
564 565
{
    ypdata_t *yp;
566
    char *s;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
567

568 569 570 571 572 573
    yp = calloc (1, sizeof (ypdata_t));
    do
    {
        unsigned len = 512;
        int ret;
        char *url;
574
        mount_proxy *mountproxy = NULL;
575 576 577 578
        ice_config_t *config;

        if (yp == NULL)
            break;
579
        yp->mount = strdup (mount);
580 581 582 583 584 585 586 587 588
        yp->server_name = strdup ("");
        yp->server_desc = strdup ("");
        yp->server_genre = strdup ("");
        yp->bitrate = strdup ("");
        yp->server_type = strdup ("");
        yp->cluster_password = strdup ("");
        yp->url = strdup ("");
        yp->current_song = strdup ("");
        yp->audio_info = strdup ("");
589
        yp->subtype = strdup ("");
590 591 592 593 594 595
        yp->process = do_yp_add;

        url = malloc (len);
        if (url == NULL)
            break;
        config = config_get_config();
596
        ret = snprintf (url, len, "http://%s:%d%s", config->hostname, config->port, mount);
597
        if (ret >= (signed)len)
598
        {
599 600
            s = realloc (url, ++ret);
            if (s) url = s;
601
            snprintf (url, ret, "http://%s:%d%s", config->hostname, config->port, mount);
602
        }
603

604
        mountproxy = config_find_mount (config, mount, MOUNT_TYPE_NORMAL);
605
        if (mountproxy && mountproxy->cluster_password)
606
            add_yp_info (yp, mountproxy->cluster_password, YP_CLUSTER_PASSWORD);
607
        config_release_config();
608

609 610 611 612 613 614 615 616 617 618 619 620 621 622
        yp->listen_url = util_url_escape (url);
        free (url);
        if (yp->listen_url == NULL)
            break;

        return yp;
    } while (0);

    yp_destroy_ypdata (yp);
    return NULL;
}


/* Check for changes in the YP servers configured */
623
static void check_servers (void)
624
{
625 626
    struct yp_server *server = (struct yp_server *)active_yps,
                     **server_p = (struct yp_server **)&active_yps;
627 628 629 630 631 632

    while (server)
    {
        if (server->remove)
        {
            struct yp_server *to_go = server;
633
            ICECAST_LOG_DEBUG("YP server \"%s\"removed", server->url);
634 635
            *server_p = server->next;
            server = server->next;
636
            destroy_yp_server(to_go);
637
            continue;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
638
        }
639 640 641 642 643 644 645 646
        server_p = &server->next;
        server = server->next;
    }
    /* add new server entries */
    while (pending_yps)
    {
        avl_node *node;

647
        server = (struct yp_server *)pending_yps;
648 649
        pending_yps = server->next;

650
        ICECAST_LOG_DEBUG("Add pending yps %s", server->url);
651
        server->next = (struct yp_server *)active_yps;
652 653 654 655 656 657 658 659 660 661
        active_yps = server;

        /* new YP server configured, need to populate with existing sources */
        avl_tree_rlock (global.source_tree);
        node = avl_get_first (global.source_tree);
        while (node)
        {
            ypdata_t *yp;

            source_t *source = node->key;
662
            if (source->yp_public && (yp = create_yp_entry (source->mount)) != NULL)
663
            {
664
                ICECAST_LOG_DEBUG("Adding existing mount %s", source->mount);
665 666 667 668
                yp->server = server;
                yp->touch_interval = server->touch_interval;
                yp->next = server->mounts;
                server->mounts = yp;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
669
            }
670
            node = avl_get_next (node);
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
671
        }
672
        avl_tree_unlock (global.source_tree);
673
    }
674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693
}


static void add_pending_yp (struct yp_server *server)
{
    ypdata_t *current, *yp;
    unsigned count = 0;

    if (server->pending_mounts == NULL)
        return;
    current = server->mounts;
    server->mounts = server->pending_mounts;
    server->pending_mounts = NULL;
    yp = server->mounts;
    while (1)
    {
        count++;
        if (yp->next == NULL)
            break;
        yp = yp->next;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
694
    }
695
    yp->next = current;
696
    ICECAST_LOG_DEBUG("%u YP entries added to %s", count, server->url);
697 698
}

699

700
static void delete_marked_yp(struct yp_server *server)
701 702
{
    ypdata_t *yp = server->mounts, **prev = &server->mounts;
703

704 705 706 707 708
    while (yp)
    {
        if (yp->remove)
        {
            ypdata_t *to_go = yp;
709
            ICECAST_LOG_DEBUG("removed %s from YP server %s", yp->mount, server->url);
710 711
            *prev = yp->next;
            yp = yp->next;
712
            yp_destroy_ypdata(to_go);
713 714 715 716 717
            continue;
        }
        prev = &yp->next;
        yp = yp->next;
    }
718
}
719 720 721


static void *yp_update_thread(void *arg)
722
{
723
    ICECAST_LOG_INFO("YP update thread started");
724 725 726 727 728 729 730 731 732 733

    yp_running = 1;
    while (yp_running)
    {
        struct yp_server *server;

        thread_sleep (200000);

        /* do the YP communication */
        thread_rwlock_rlock (&yp_lock);
734
        server = (struct yp_server *)active_yps;
735 736
        while (server)
        {
737
            /* ICECAST_LOG_DEBUG("trying %s", server->url); */
738 739
            yp_process_server (server);
            server = server->next;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
740
        }
741 742 743 744 745 746 747
        thread_rwlock_unlock (&yp_lock);

        /* update the local YP structure */
        if (yp_update)
        {
            thread_rwlock_wlock (&yp_lock);
            check_servers ();
748
            server = (struct yp_server *)active_yps;
749 750
            while (server)
            {
751
                /* ICECAST_LOG_DEBUG("Checking yps %s", server->url); */
752 753 754
                add_pending_yp (server);
                delete_marked_yp (server);
                server = server->next;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
755
            }
756 757
            yp_update = 0;
            thread_rwlock_unlock (&yp_lock);
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
758 759
        }
    }
760 761 762 763 764
    thread_rwlock_destroy (&yp_lock);
    thread_mutex_destroy (&yp_pending_lock);
    /* free server and ypdata left */
    while (active_yps)
    {
765
        struct yp_server *server = (struct yp_server *)active_yps;
766 767 768
        active_yps = server->next;
        destroy_yp_server (server);
    }
769

770
    return NULL;
771
}
772

773 774 775


static void yp_destroy_ypdata(ypdata_t *ypdata)
776
{
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
777
    if (ypdata) {
778
        if (ypdata->mount) {
779
            free(ypdata->mount);
780 781
        }
        if (ypdata->url) {
782
            free(ypdata->url);
783
        }
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810
        if (ypdata->sid) {
            free(ypdata->sid);
        }
        if (ypdata->server_name) {
            free(ypdata->server_name);
        }
        if (ypdata->server_desc) {
            free(ypdata->server_desc);
        }
        if (ypdata->server_genre) {
            free(ypdata->server_genre);
        }
        if (ypdata->cluster_password) {
            free(ypdata->cluster_password);
        }
        if (ypdata->listen_url) {
            free(ypdata->listen_url);
        }
        if (ypdata->current_song) {
            free(ypdata->current_song);
        }
        if (ypdata->bitrate) {
            free(ypdata->bitrate);
        }
        if (ypdata->server_type) {
            free(ypdata->server_type);
        }
811 812 813
        if (ypdata->audio_info) {
            free(ypdata->audio_info);
        }
814 815 816
        free(ypdata->subtype);
        free(ypdata->error_msg);
        free(ypdata);
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
817
    }
818
}
819

820
static void add_yp_info(ypdata_t *yp, void *info, int type)
821 822
{
    char *escaped;
823 824

    if (!info)
825
        return;
826

827 828 829 830
    escaped = util_url_escape(info);
    if (escaped == NULL)
        return;

831 832
    switch (type)
    {
833
        case YP_SERVER_NAME:
834 835
            free (yp->server_name);
            yp->server_name = escaped;
836
            break;
837
        case YP_SERVER_DESC:
838 839
            free (yp->server_desc);
            yp->server_desc = escaped;
840
            break;
841
        case YP_SERVER_GENRE:
842 843
            free (yp->server_genre);
            yp->server_genre = escaped;
844
            break;
845
        case YP_SERVER_URL:
846 847
            free (yp->url);
            yp->url = escaped;
848
            break;
849
        case YP_BITRATE:
850 851
            free (yp->bitrate);
            yp->bitrate = escaped;
852
            break;
853
        case YP_AUDIO_INFO:
854 855
            free (yp->audio_info);
            yp->audio_info = escaped;
856
            break;
857
        case YP_SERVER_TYPE:
858 859
            free (yp->server_type);
            yp->server_type = escaped;
860
            break;
861
        case YP_CURRENT_SONG:
862 863
            free (yp->current_song);
            yp->current_song = escaped;
864
            break;
865
        case YP_CLUSTER_PASSWORD:
866 867
            free (yp->cluster_password);
            yp->cluster_password = escaped;
868
            break;
869
        case YP_SUBTYPE:
870 871
            free (yp->subtype);
            yp->subtype = escaped;
872
            break;
873 874
        default:
            free (escaped);
875 876 877 878 879
    }
}


/* Add YP entries to active servers */
880
void yp_add (const char *mount)
881 882 883 884 885 886 887 888
{
    struct yp_server *server;

    /* make sure YP thread is not modifying the lists */
    thread_rwlock_rlock (&yp_lock);

    /* make sure we don't race against another yp_add */
    thread_mutex_lock (&yp_pending_lock);
889
    server = (struct yp_server *)active_yps;
890 891 892
    while (server)
    {
        ypdata_t *yp;
893 894 895 896

        /* on-demand relays may already have a YP entry */
        yp = find_yp_mount (server->mounts, mount);
        if (yp == NULL)
897
        {
898 899 900 901
            /* add new ypdata to each servers pending yp */
            yp = create_yp_entry (mount);
            if (yp)
            {
902
                ICECAST_LOG_DEBUG("Adding %s to %s", mount, server->url);
903 904 905
                yp->server = server;
                yp->touch_interval = server->touch_interval;
                yp->next = server->pending_mounts;
906
                yp->next_update = time(NULL) + 60;
907 908 909
                server->pending_mounts = yp;
                yp_update = 1;
            }
910
        }
911
        else
912
            ICECAST_LOG_DEBUG("YP entry %s already exists", mount);
913 914 915 916 917 918 919 920 921 922 923
        server = server->next;
    }
    thread_mutex_unlock (&yp_pending_lock);
    thread_rwlock_unlock (&yp_lock);
}



/* Mark an existing entry in the YP list as to be marked for deletion */
void yp_remove (const char *mount)
{
924
    struct yp_server *server = (struct yp_server *)active_yps;
925 926 927 928

    thread_rwlock_rlock (&yp_lock);
    while (server)
    {
929 930 931
        ypdata_t *list = server->mounts;

        while (1)
932
        {
933 934 935 936 937 938 939 940
            ypdata_t *yp = find_yp_mount (list, mount);
            if (yp == NULL)
                break;
            if (yp->release || yp->remove)
            {
                list = yp->next;
                continue;   /* search again these are old entries */
            }
941
            ICECAST_LOG_DEBUG("release %s on YP %s", mount, server->url);
942
            yp->release = 1;
943
            yp->next_update = 0;
944
        }
945
        server = server->next;
946
    }
947
    thread_rwlock_unlock (&yp_lock);
948
}
949 950 951 952 953 954


/* This is similar to yp_remove, but we force a touch
 * attempt */
void yp_touch (const char *mount)
{
955
    struct yp_server *server = (struct yp_server *)active_yps;
956
    ypdata_t *search_list = NULL;
957 958

    thread_rwlock_rlock (&yp_lock);
959 960 961
    if (server)
        search_list = server->mounts;

962 963
    while (server)
    {
964
        ypdata_t *yp = find_yp_mount (search_list, mount);
965 966
        if (yp)
        {
967
            /* we may of found old entries not purged yet, so skip them */
968
            if (yp->release != 0 || yp->remove != 0)
969 970 971 972
            {
                search_list = yp->next;
                continue;
            }
973 974 975
            /* don't update the directory if there is a touch scheduled soon */
            if (yp->process == do_yp_touch && now + yp->touch_interval - yp->next_update > 60)
                yp->next_update = now + 3;
976 977
        }
        server = server->next;
978 979
        if (server)
            search_list = server->mounts;
980 981 982 983 984
    }
    thread_rwlock_unlock (&yp_lock);
}


985
void yp_shutdown (void)
986 987 988
{
    yp_running = 0;
    yp_update = 1;
989 990
    if (yp_thread)
        thread_join (yp_thread);
991 992
    free ((char*)server_version);
    server_version = NULL;
993
    ICECAST_LOG_INFO("YP thread down");
994 995
}