slave.c 6.61 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
static int _initialized = 0;

void slave_initialize(void) {
Michael Smith's avatar
Michael Smith committed
52
    ice_config_t *config;
53
	if (_initialized) return;
Michael Smith's avatar
Michael Smith committed
54
55

    config = config_get_config();
56
    /* Don't create a slave thread if it isn't configured */
Michael Smith's avatar
Michael Smith committed
57
58
59
60
    if (config->master_server == NULL && 
            config->relay == NULL)
    {
        config_release_config();
61
        return;
Michael Smith's avatar
Michael Smith committed
62
63
    }
    config_release_config();
64
65
66
67
68
69
70
71
72
73
74

	_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);
}

75
static void create_relay_stream(char *server, int port, 
76
        char *remotemount, char *localmount, int mp3)
Michael Smith's avatar
Michael Smith committed
77
78
{
    sock_t streamsock;
79
80
81
	char header[4096];
	connection_t *con;
	http_parser_t *parser;
82
    client_t *client;
Michael Smith's avatar
Michael Smith committed
83

84
85
86
87
    if(!localmount)
        localmount = remotemount;

    DEBUG1("Adding source at mountpoint \"%s\"", localmount);
Michael Smith's avatar
Michael Smith committed
88
89
90

	streamsock = sock_connect_wto(server, port, 0);
	if (streamsock == SOCK_ERROR) {
91
        WARN2("Failed to relay stream from master server, couldn't connect to http://%s:%d", server, port);
Michael Smith's avatar
Michael Smith committed
92
93
94
        return;
	}
	con = create_connection(streamsock, NULL);
95
    if(mp3) {
96
97
98
99
100
101
102
        /* 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", 
103
104
105
                remotemount);
    }
    else {
106
107
108
109
    	sock_write(streamsock, "GET %s HTTP/1.0\r\n"
                               "User-Agent: " ICECAST_VERSION_STRING "\r\n"
                               "\r\n",
                remotemount);
110
    }
Michael Smith's avatar
Michael Smith committed
111
112
	memset(header, 0, sizeof(header));
	if (util_read_header(con->sock, header, 4096) == 0) {
113
        WARN0("Header read failed");
Michael Smith's avatar
Michael Smith committed
114
115
116
117
118
		connection_close(con);
		return;
	}
	parser = httpp_create_parser();
	httpp_initialize(parser, NULL);
119
	if(!httpp_parse_response(parser, header, strlen(header), localmount)) {
Michael Smith's avatar
Michael Smith committed
120
121
122
123
124
125
126
127
128
129
130
131
132
133
        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))) {
134
        DEBUG0("Failed to create source");
Michael Smith's avatar
Michael Smith committed
135
136
        client_destroy(client);
	}
137

Michael Smith's avatar
Michael Smith committed
138
139
140
141
142
143
    return;
}

static void *_slave_thread(void *arg) {
	sock_t mastersock;
	char buf[256];
Michael Smith's avatar
Michael Smith committed
144
    int interval;
145
146
147
    char *authheader, *data;
    int len;
    char *username = "relay";
Michael Smith's avatar
Michael Smith committed
148
149
    char *password;
    int max_interval;
Michael Smith's avatar
Michael Smith committed
150
    relay_server *relay;
Michael Smith's avatar
Michael Smith committed
151
152
153
154
155
156
    ice_config_t *config;
    
    config = config_get_config();

    password = config->master_password;
    interval = max_interval = config->master_update_interval;
157
158

    if(password == NULL)
Michael Smith's avatar
Michael Smith committed
159
160
161
        password = config->source_password;

    config_release_config();
162

163
164

	while (_initialized) {
Michael Smith's avatar
Michael Smith committed
165
        if (max_interval > ++interval) {
166
167
168
		    thread_sleep(1000000);
            continue;
        }
Michael Smith's avatar
Michael Smith committed
169
170
171
172
173
174
        else {
            /* In case it's been reconfigured */
            config = config_get_config();
            max_interval = config->master_update_interval;
            config_release_config();

175
            interval = 0;
Michael Smith's avatar
Michael Smith committed
176
177
178
179
180
181
182
183
184
        }

        config = config_get_config();
        if(config->master_server != NULL) {
            char *server = config->master_server;
            int port = config->master_server_port;
            config_release_config();

		    mastersock = sock_connect_wto(server, port, 0);
185

Michael Smith's avatar
Michael Smith committed
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
    		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;
            }
207

Michael Smith's avatar
Michael Smith committed
208
209
210
211
212
	    	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);

Michael Smith's avatar
Michael Smith committed
213
                    create_relay_stream(server, port, buf, NULL, 0);
Michael Smith's avatar
Michael Smith committed
214
215
216
217
218
219
    			} 
                else
    	    		avl_tree_unlock(global.source_tree);
    		}
	    	sock_close(mastersock);
        }
Michael Smith's avatar
Michael Smith committed
220
221
222
        else {
            config_release_config();
        }
Michael Smith's avatar
Michael Smith committed
223
224

        /* And now, we process the individual mounts... */
Michael Smith's avatar
Michael Smith committed
225
226
227
228
229
        config = config_get_config();
        relay = config->relay;
        thread_mutex_lock(&(config_locks()->relay_lock));
        config_release_config();

Michael Smith's avatar
Michael Smith committed
230
231
        while(relay) {
            avl_tree_rlock(global.source_tree);
232
            if(!source_find_mount(relay->localmount)) {
Michael Smith's avatar
Michael Smith committed
233
234
                avl_tree_unlock(global.source_tree);

235
                create_relay_stream(relay->server, relay->port, relay->mount,
236
                        relay->localmount, relay->mp3metadata);
Michael Smith's avatar
Michael Smith committed
237
238
239
            }
            else
                avl_tree_unlock(global.source_tree);
Michael Smith's avatar
Michael Smith committed
240
            relay = relay->next;
Michael Smith's avatar
Michael Smith committed
241
        }
Michael Smith's avatar
Michael Smith committed
242
243

        thread_mutex_unlock(&(config_locks()->relay_lock));
244
245
246
247
	}
	thread_exit(0);
	return NULL;
}
Michael Smith's avatar
Michael Smith committed
248