mirror of https://github.com/SuperBFG7/ympd
269 lines
8.5 KiB
C
269 lines
8.5 KiB
C
/* myMPD
|
|
(c) 2018 Juergen Mang <mail@jcgames.de>
|
|
This project's homepage is: https://github.com/jcorporation/mympd
|
|
|
|
myMPD ist fork of:
|
|
|
|
ympd
|
|
(c) 2013-2014 Andrew Karpow <andy@ndyk.de>
|
|
This project's homepage is: http://www.ympd.org
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; version 2 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sys/time.h>
|
|
#include <pwd.h>
|
|
|
|
#include "../dist/src/mongoose/mongoose.h"
|
|
#include "../dist/src/inih/ini.h"
|
|
#include "mpd_client.h"
|
|
#include "config.h"
|
|
|
|
static sig_atomic_t s_signal_received = 0;
|
|
static struct mg_serve_http_opts s_http_server_opts;
|
|
char s_redirect[250];
|
|
|
|
static void signal_handler(int sig_num) {
|
|
signal(sig_num, signal_handler); // Reinstantiate signal handler
|
|
s_signal_received = sig_num;
|
|
}
|
|
|
|
static void handle_api(struct mg_connection *nc, struct http_message *hm) {
|
|
if (!is_websocket(nc))
|
|
mg_printf(nc, "%s", "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\nContent-Type: application/json\r\n\r\n");
|
|
|
|
char buf[1000] = {0};
|
|
memcpy(buf, hm->body.p,sizeof(buf) - 1 < hm->body.len ? sizeof(buf) - 1 : hm->body.len);
|
|
struct mg_str d = {buf, strlen(buf)};
|
|
callback_mympd(nc, d);
|
|
|
|
if (!is_websocket(nc))
|
|
mg_send_http_chunk(nc, "", 0); /* Send empty chunk, the end of response */
|
|
}
|
|
|
|
static void ev_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
switch(ev) {
|
|
case MG_EV_WEBSOCKET_HANDSHAKE_DONE: {
|
|
#ifdef DEBUG
|
|
fprintf(stdout,"New Websocket connection\n");
|
|
#endif
|
|
struct mg_str d = {(char *) "{\"cmd\": \"MPD_API_WELCOME\"}", 25 };
|
|
callback_mympd(nc, d);
|
|
break;
|
|
}
|
|
case MG_EV_HTTP_REQUEST: {
|
|
struct http_message *hm = (struct http_message *) ev_data;
|
|
#ifdef DEBUG
|
|
printf("HTTP request: %.*s\n", hm->uri.len, hm->uri.p);
|
|
#endif
|
|
if (mg_vcmp(&hm->uri, "/api") == 0) {
|
|
handle_api(nc, hm);
|
|
}
|
|
else {
|
|
mg_serve_http(nc, hm, s_http_server_opts);
|
|
}
|
|
break;
|
|
}
|
|
case MG_EV_CLOSE: {
|
|
if (is_websocket(nc)) {
|
|
#ifdef DEBUG
|
|
printf("Websocket connection closed\n");
|
|
#endif
|
|
mympd_close_handler(nc);
|
|
}
|
|
else {
|
|
#ifdef DEBUG
|
|
fprintf(stdout,"HTTP Close\n");
|
|
#endif
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void ev_handler_http(struct mg_connection *nc_http, int ev, void *ev_data) {
|
|
switch(ev) {
|
|
case MG_EV_HTTP_REQUEST: {
|
|
printf("Redirecting to %s\n", s_redirect);
|
|
mg_http_send_redirect(nc_http, 301, mg_mk_str(s_redirect), mg_mk_str(NULL));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int inihandler(void* user, const char* section, const char* name, const char* value) {
|
|
configuration* p_config = (configuration*)user;
|
|
|
|
#define MATCH(n) strcmp(name, n) == 0
|
|
if (MATCH("mpdhost"))
|
|
p_config->mpdhost = strdup(value);
|
|
else if (MATCH("mpdhost"))
|
|
p_config->mpdhost = strdup(value);
|
|
else if (MATCH("mpdport"))
|
|
p_config->mpdport = atoi(value);
|
|
else if (MATCH("mpdhost"))
|
|
p_config->mpdhost = strdup(value);
|
|
else if (MATCH("mpdpass"))
|
|
p_config->mpdpass = strdup(value);
|
|
else if (MATCH("webport"))
|
|
p_config->webport = strdup(value);
|
|
else if (MATCH("ssl"))
|
|
if (strcmp(value, "true") == 0)
|
|
p_config->ssl = true;
|
|
else
|
|
p_config->ssl = false;
|
|
else if (MATCH("sslcert"))
|
|
p_config->sslcert = strdup(value);
|
|
else if (MATCH("sslkey"))
|
|
p_config->sslkey = strdup(value);
|
|
else if (MATCH("user"))
|
|
p_config->user = strdup(value);
|
|
else if (MATCH("streamport"))
|
|
p_config->streamport = atoi(value);
|
|
else if (MATCH("coverimage"))
|
|
p_config->coverimage = strdup(value);
|
|
else if (MATCH("statefile"))
|
|
p_config->statefile = strdup(value);
|
|
else
|
|
return 0; /* unknown section/name, error */
|
|
|
|
return 1;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
struct mg_mgr mgr;
|
|
struct mg_connection *nc;
|
|
struct mg_connection *nc_http;
|
|
unsigned int current_timer = 0, last_timer = 0;
|
|
struct mg_bind_opts bind_opts;
|
|
const char *err;
|
|
|
|
char hostname[1024];
|
|
hostname[1023] = '\0';
|
|
gethostname(hostname, 1023);
|
|
|
|
//defaults
|
|
config.mpdhost = "127.0.0.1";
|
|
config.mpdport = 6600;
|
|
config.mpdpass = NULL;
|
|
config.webport = "80";
|
|
config.ssl = true;
|
|
config.sslport = "443";
|
|
config.sslcert = "/etc/mympd/ssl/server.pem";
|
|
config.sslkey = "/etc/mympd/ssl/server.key";
|
|
config.user = "nobody";
|
|
config.streamport = 8000;
|
|
config.coverimage = "folder.jpg";
|
|
config.statefile = "/var/lib/mympd/mympd.state";
|
|
|
|
if (argc == 2) {
|
|
if (ini_parse(argv[1], inihandler, &config) < 0) {
|
|
printf("Can't load '%s'\n", argv[1]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
else {
|
|
fprintf(stdout, "myMPD %s\n"
|
|
"Copyright (C) 2018 Juergen Mang <mail@jcgames.de>\n"
|
|
"https://github.com/jcorporation/myMPD\n"
|
|
"Built " __DATE__ " "__TIME__"\n\n",
|
|
MYMPD_VERSION);
|
|
printf("Usage: %s /path/to/mympd.conf\n", argv[0]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
signal(SIGTERM, signal_handler);
|
|
signal(SIGINT, signal_handler);
|
|
setvbuf(stdout, NULL, _IOLBF, 0);
|
|
setvbuf(stderr, NULL, _IOLBF, 0);
|
|
|
|
mg_mgr_init(&mgr, NULL);
|
|
|
|
if (config.ssl == true) {
|
|
snprintf(s_redirect, 200, "https://%s:%s/", hostname, config.sslport);
|
|
nc_http = mg_bind(&mgr, config.webport, ev_handler_http);
|
|
if (nc_http == NULL) {
|
|
fprintf(stderr, "Error starting server on port %s\n", config.webport );
|
|
return EXIT_FAILURE;
|
|
}
|
|
memset(&bind_opts, 0, sizeof(bind_opts));
|
|
bind_opts.ssl_cert = config.sslcert;
|
|
bind_opts.ssl_key = config.sslkey;
|
|
bind_opts.error_string = &err;
|
|
nc = mg_bind_opt(&mgr, config.sslport, ev_handler, bind_opts);
|
|
if (nc == NULL) {
|
|
fprintf(stderr, "Error starting server on port %s: %s\n", config.sslport, err);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
else {
|
|
nc = mg_bind(&mgr, config.webport, ev_handler);
|
|
if (nc == NULL) {
|
|
fprintf(stderr, "Error starting server on port %s\n", config.webport );
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
if (config.user != NULL) {
|
|
printf("Droping privileges\n");
|
|
struct passwd *pw;
|
|
if ((pw = getpwnam(config.user)) == NULL) {
|
|
printf("Unknown user\n");
|
|
mg_mgr_free(&mgr);
|
|
return EXIT_FAILURE;
|
|
} else if (setgid(pw->pw_gid) != 0) {
|
|
printf("setgid() failed\n");
|
|
mg_mgr_free(&mgr);
|
|
return EXIT_FAILURE;
|
|
} else if (setuid(pw->pw_uid) != 0) {
|
|
printf("setuid() failed\n");
|
|
mg_mgr_free(&mgr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
if (getuid() == 0) {
|
|
printf("myMPD should not be run with root privileges\n");
|
|
mg_mgr_free(&mgr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (config.ssl == true)
|
|
mg_set_protocol_http_websocket(nc_http);
|
|
|
|
mg_set_protocol_http_websocket(nc);
|
|
s_http_server_opts.document_root = SRC_PATH;
|
|
s_http_server_opts.enable_directory_listing = "no";
|
|
|
|
printf("myMPD started on http port %s\n", config.webport);
|
|
if (config.ssl == true)
|
|
printf("myMPD started on ssl port %s\n", config.sslport);
|
|
|
|
while (s_signal_received == 0) {
|
|
mg_mgr_poll(&mgr, 200);
|
|
current_timer = time(NULL);
|
|
if (current_timer - last_timer) {
|
|
last_timer = current_timer;
|
|
mympd_poll(&mgr);
|
|
}
|
|
}
|
|
mg_mgr_free(&mgr);
|
|
mympd_disconnect();
|
|
return EXIT_SUCCESS;
|
|
}
|