slave.c 5.92 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
/* slave.c
 * by Ciaran Anscomb <ciaran.anscomb@6809.org.uk>
 *
 * Periodically requests a list of streams from a master server
 * and creates source threads for any it doesn't already have.
 * */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <sys/types.h>

#ifndef _WIN32
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#else
#include <winsock2.h>
#define snprintf _snprintf
#define strcasecmp stricmp
#define strncasecmp strnicmp
#endif

#include "os.h"

#include "thread.h"
#include "avl.h"
#include "sock.h"
#include "log.h"
#include "httpp.h"

#include "config.h"
#include "global.h"
#include "util.h"
#include "connection.h"
#include "refbuf.h"
#include "client.h"
#include "stats.h"
#include "logging.h"
41
#include "geturl.h"
42
#include "source.h"
Michael Smith's avatar
Michael Smith committed
43
#include "format.h"
44 45 46 47

#define CATMODULE "slave"

static void *_slave_thread(void *arg);
48
thread_type *_slave_thread_id;
49 50 51 52 53
static int _initialized = 0;

void slave_initialize(void) {
	if (_initialized) return;
    /* Don't create a slave thread if it isn't configured */
Michael Smith's avatar
Michael Smith committed
54 55
    if (config_get_config()->master_server == NULL && 
            config_get_config()->relay == NULL)
56 57 58 59 60 61 62 63 64 65 66 67
        return;

	_initialized = 1;
	_slave_thread_id = thread_create("Slave Thread", _slave_thread, NULL, THREAD_ATTACHED);
}

void slave_shutdown(void) {
	if (!_initialized) return;
	_initialized = 0;
	thread_join(_slave_thread_id);
}

68
static void create_relay_stream(char *server, int port, 
69
        char *remotemount, char *localmount, int mp3)
Michael Smith's avatar
Michael Smith committed
70 71
{
    sock_t streamsock;
72 73 74
	char header[4096];
	connection_t *con;
	http_parser_t *parser;
75
    client_t *client;
Michael Smith's avatar
Michael Smith committed
76

77 78 79 80
    if(!localmount)
        localmount = remotemount;

    DEBUG1("Adding source at mountpoint \"%s\"", localmount);
Michael Smith's avatar
Michael Smith committed
81 82 83

	streamsock = sock_connect_wto(server, port, 0);
	if (streamsock == SOCK_ERROR) {
84
        WARN2("Failed to relay stream from master server, couldn't connect to http://%s:%d", server, port);
Michael Smith's avatar
Michael Smith committed
85 86 87
        return;
	}
	con = create_connection(streamsock, NULL);
88
    if(mp3) {
89 90 91 92 93 94 95
        /* Some mp3 servers are bitchy, send a user-agent string to make them
         * send the right response.
         */
    	sock_write(streamsock, "GET %s HTTP/1.0\r\n"
                               "User-Agent: " ICECAST_VERSION_STRING "\r\n"
                               "Icy-MetaData: 1\r\n"
                               "\r\n", 
96 97 98 99 100
                remotemount);
    }
    else {
    	sock_write(streamsock, "GET %s HTTP/1.0\r\n\r\n", remotemount);
    }
Michael Smith's avatar
Michael Smith committed
101 102
	memset(header, 0, sizeof(header));
	if (util_read_header(con->sock, header, 4096) == 0) {
103
        WARN0("Header read failed");
Michael Smith's avatar
Michael Smith committed
104 105 106 107 108
		connection_close(con);
		return;
	}
	parser = httpp_create_parser();
	httpp_initialize(parser, NULL);
109
	if(!httpp_parse_response(parser, header, strlen(header), localmount)) {
Michael Smith's avatar
Michael Smith committed
110 111 112 113 114 115 116 117 118 119 120 121 122 123
        if(httpp_getvar(parser, HTTPP_VAR_ERROR_MESSAGE)) {
            ERROR1("Error parsing relay request: %s", 
                    httpp_getvar(parser, HTTPP_VAR_ERROR_MESSAGE));
        }
        else
            ERROR0("Error parsing relay request");
		connection_close(con);
        httpp_destroy(parser);
        return;
    }

    client = client_create(con, parser);
	if (!connection_create_source(client, con, parser, 
                httpp_getvar(parser, HTTPP_VAR_URI))) {
124
        DEBUG0("Failed to create source");
Michael Smith's avatar
Michael Smith committed
125 126
        client_destroy(client);
	}
127

Michael Smith's avatar
Michael Smith committed
128 129 130 131 132 133
    return;
}

static void *_slave_thread(void *arg) {
	sock_t mastersock;
	char buf[256];
134
    int interval = config_get_config()->master_update_interval;
135 136 137
    char *authheader, *data;
    int len;
    char *username = "relay";
138
    char *password = config_get_config()->master_password;
Michael Smith's avatar
Michael Smith committed
139
    relay_server *relay;
140 141 142 143

    if(password == NULL)
        password = config_get_config()->source_password;

144 145 146 147 148 149 150 151 152

	while (_initialized) {
        if (config_get_config()->master_update_interval > ++interval) {
		    thread_sleep(1000000);
            continue;
        }
        else
            interval = 0;

Michael Smith's avatar
Michael Smith committed
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
        if(config_get_config()->master_server != NULL) {
		    mastersock = sock_connect_wto(config_get_config()->master_server, 
                    config_get_config()->master_server_port, 0);
    		if (mastersock == SOCK_ERROR) {
                WARN0("Relay slave failed to contact master server to fetch stream list");
		    	continue;
    		}

            len = strlen(username) + strlen(password) + 1;
            authheader = malloc(len+1);
            strcpy(authheader, username);
            strcat(authheader, ":");
            strcat(authheader, password);
            data = util_base64_encode(authheader);
		    sock_write(mastersock, 
                    "GET /admin/streamlist HTTP/1.0\r\n"
                    "Authorization: Basic %s\r\n"
                    "\r\n", data);
            free(authheader);
            free(data);
    		while (sock_read_line(mastersock, buf, sizeof(buf))) {
                if(!strlen(buf))
                    break;
            }
177

Michael Smith's avatar
Michael Smith committed
178 179 180 181 182 183 184 185
	    	while (sock_read_line(mastersock, buf, sizeof(buf))) {
		    	avl_tree_rlock(global.source_tree);
			    if (!source_find_mount(buf)) {
				    avl_tree_unlock(global.source_tree);

                    create_relay_stream(
                            config_get_config()->master_server,
                            config_get_config()->master_server_port,
186
                            buf, NULL, 0);
Michael Smith's avatar
Michael Smith committed
187 188 189 190 191 192
    			} 
                else
    	    		avl_tree_unlock(global.source_tree);
    		}
	    	sock_close(mastersock);
        }
Michael Smith's avatar
Michael Smith committed
193 194

        /* And now, we process the individual mounts... */
Michael Smith's avatar
Michael Smith committed
195
        relay = config_get_config()->relay;
Michael Smith's avatar
Michael Smith committed
196 197
        while(relay) {
            avl_tree_rlock(global.source_tree);
198
            if(!source_find_mount(relay->localmount)) {
Michael Smith's avatar
Michael Smith committed
199 200
                avl_tree_unlock(global.source_tree);

201
                create_relay_stream(relay->server, relay->port, relay->mount,
202
                        relay->localmount, relay->mp3metadata);
Michael Smith's avatar
Michael Smith committed
203 204 205
            }
            else
                avl_tree_unlock(global.source_tree);
Michael Smith's avatar
Michael Smith committed
206
            relay = relay->next;
Michael Smith's avatar
Michael Smith committed
207
        }
208 209 210 211
	}
	thread_exit(0);
	return NULL;
}
Michael Smith's avatar
Michael Smith committed
212