aniani
/
ezstream
mirror of https://gitlab.xiph.org/xiph/ezstream.git
1
0
Fork 0
Code
ezstream/src/ezstream.c

859 lines
20 KiB
C
Raw Permalink Blame History

/*
* ezstream - source client for Icecast with external en-/decoder support
* Copyright (C) 2003, 2004, 2005, 2006 Ed Zaleski <oddsock@oddsock.org>
* Copyright (C) 2007, 2009, 2015, 2017 Moritz Grimm <mgrimm@mrsserver.net>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
*
* 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.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "compat.h"
#include "ezstream.h"
#include <signal.h>
#include "cfg.h"
#include "cmdline.h"
#include "log.h"
#include "mdata.h"
#include "playlist.h"
#include "stream.h"
#include "util.h"
#include "xalloc.h"
#define STREAM_DONE 0
#define STREAM_CONT 1
#define STREAM_SKIP 2
#define STREAM_SERVERR 3
#define STREAM_UPDMDATA 4
stream_t main_stream;
playlist_t playlist;
int playlistMode;
unsigned int resource_errors;
const int ezstream_signals[] = {
SIGTERM, SIGINT, SIGHUP, SIGUSR1, SIGUSR2
};
volatile sig_atomic_t rereadPlaylist;
volatile sig_atomic_t rereadPlaylist_notify;
volatile sig_atomic_t skipTrack;
volatile sig_atomic_t queryMetadata;
volatile sig_atomic_t quit;
void sig_handler(int);
static char * _build_reencode_cmd(const char *, const char *, cfg_stream_t,
mdata_t);
static FILE * openResource(stream_t, const char *, int *, mdata_t *,
int *, long *);
int reconnect(stream_t);
const char * getTimeString(long);
int sendStream(stream_t, FILE *, const char *, int, const char *,
struct timespec *);
int streamFile(stream_t, const char *);
int streamPlaylist(stream_t);
int ez_shutdown(int);
void
sig_handler(int sig)
{
switch (sig) {
case SIGTERM:
case SIGINT:
quit = 1;
break;
case SIGHUP:
rereadPlaylist = 1;
rereadPlaylist_notify = 1;
break;
case SIGUSR1:
skipTrack = 1;
break;
case SIGUSR2:
queryMetadata = 1;
break;
default:
break;
}
}
static char *
_build_reencode_cmd(const char *extension, const char *filename,
cfg_stream_t cfg_stream, mdata_t md)
{
cfg_decoder_t decoder;
cfg_encoder_t encoder;
char *artist, *album, *title, *songinfo, *tmp;
char *filename_quoted;
char *custom_songinfo;
struct util_dict dicts[6];
char *dec_str;
char *cmd_str;
size_t cmd_str_size;
decoder = cfg_decoder_list_findext(cfg_get_decoders(), extension);
if (!decoder) {
log_error("cannot decode: %s: unsupported file extension %s",
filename, extension);
return (NULL);
}
encoder = cfg_encoder_list_find(cfg_get_encoders(),
cfg_stream_get_encoder(cfg_stream));
if (!encoder) {
log_error("cannot encode: %s: unknown encoder",
cfg_stream_get_encoder(cfg_stream));
return (NULL);
}
tmp = util_utf82char(mdata_get_artist(md));
artist = util_shellquote(tmp, 0);
xfree(tmp);
tmp = util_utf82char(mdata_get_album(md));
album = util_shellquote(tmp, 0);
xfree(tmp);
tmp = util_utf82char(mdata_get_title(md));
title = util_shellquote(tmp, 0);
xfree(tmp);
tmp = util_utf82char(mdata_get_songinfo(md));
songinfo = util_shellquote(tmp, 0);
xfree(tmp);
filename_quoted = util_shellquote(filename, 0);
/*
* if (prog && format)
* metatoformat
* else
* if (!prog && title)
* emptymeta
* else
* replacemeta
*/
if (cfg_get_metadata_program() &&
cfg_get_metadata_format_str()) {
char buf[BUFSIZ];
char *unquoted;
mdata_strformat(md, buf, sizeof(buf),
cfg_get_metadata_format_str());
unquoted = util_utf82char(buf);
custom_songinfo = util_shellquote(unquoted, 0);
xfree(unquoted);
} else {
if (!cfg_get_metadata_program() &&
strstr(cfg_decoder_get_program(decoder),
PLACEHOLDER_TITLE) != NULL) {
custom_songinfo = xstrdup("");
} else {
custom_songinfo = xstrdup(songinfo);
}
}
xfree(songinfo);
memset(dicts, 0, sizeof(dicts));
dicts[0].from = PLACEHOLDER_ARTIST;
dicts[0].to = artist;
dicts[1].from = PLACEHOLDER_ALBUM;
dicts[1].to = album;
dicts[2].from = PLACEHOLDER_TITLE;
dicts[2].to = title;
dicts[3].from = PLACEHOLDER_TRACK;
dicts[3].to = filename_quoted;
dicts[4].from = PLACEHOLDER_METADATA;
dicts[4].to = custom_songinfo;
if (!cfg_get_metadata_program() &&
strstr(cfg_encoder_get_program(encoder),
PLACEHOLDER_TITLE) != NULL) {
xfree(custom_songinfo);
dicts[4].to = custom_songinfo = xstrdup("");
}
dec_str = util_expand_words(cfg_decoder_get_program(decoder), dicts);
cmd_str_size = strlen(dec_str) + 1;
if (cfg_encoder_get_program(encoder)) {
char *enc_str;
enc_str = util_expand_words(cfg_encoder_get_program(encoder),
dicts);
cmd_str_size += strlen(" | ") + strlen(enc_str);
cmd_str = xcalloc(cmd_str_size, sizeof(char));
snprintf(cmd_str, cmd_str_size, "%s | %s", dec_str, enc_str);
xfree(enc_str);
xfree(dec_str);
} else {
cmd_str = xcalloc(cmd_str_size, sizeof(char));
snprintf(cmd_str, cmd_str_size, "%s", dec_str);
xfree(dec_str);
}
xfree(artist);
xfree(album);
xfree(title);
xfree(filename_quoted);
xfree(custom_songinfo);
return (cmd_str);
}
static FILE *
openResource(stream_t stream, const char *filename, int *popenFlag,
mdata_t *md_p, int *isStdin, long *songLen)
{
FILE *filep = NULL;
char extension[25];
char *p = NULL;
char *pCommandString = NULL;
mdata_t md;
cfg_stream_t cfg_stream = stream_get_cfg_stream(stream);
if (md_p != NULL)
*md_p = NULL;
if (songLen != NULL)
*songLen = 0;
if ((isStdin && *isStdin) ||
strcasecmp(filename, "stdin") == 0) {
if (cfg_get_metadata_program()) {
md = mdata_create();
if (0 > mdata_run_program(md, cfg_get_metadata_program()) ||
0 > stream_set_metadata(stream, md, NULL)) {
mdata_destroy(&md);
return (NULL);
}
if (md_p != NULL)
*md_p = md;
else
mdata_destroy(&md);
}
if (isStdin != NULL)
*isStdin = 1;
filep = stdin;
return (filep);
}
if (isStdin != NULL)
*isStdin = 0;
extension[0] = '\0';
p = strrchr(filename, '.');
if (p != NULL)
strlcpy(extension, p, sizeof(extension));
for (p = extension; *p != '\0'; p++)
*p = (char)tolower((int)*p);
if (strlen(extension) == 0) {
log_error("%s: cannot determine file type", filename);
return (filep);
}
md = mdata_create();
if (cfg_get_metadata_program()) {
if (0 > mdata_run_program(md, cfg_get_metadata_program()))
mdata_destroy(&md);
} else {
if (0 > mdata_parse_file(md, filename))
mdata_destroy(&md);
}
if (NULL == md)
return (NULL);
if (songLen != NULL)
*songLen = mdata_get_length(md);
*popenFlag = 0;
if (cfg_stream_get_encoder(cfg_stream)) {
int stderr_fd = -1;
pCommandString = _build_reencode_cmd(extension, filename,
cfg_stream, md);
if (NULL == pCommandString) {
mdata_destroy(&md);
return (NULL);
}
if (md_p != NULL)
*md_p = md;
else
mdata_destroy(&md);
log_info("running command: %s", pCommandString);
if (cfg_get_program_quiet_stderr()) {
int fd;
stderr_fd = dup(fileno(stderr));
if (0 > stderr_fd) {
log_alert("dup: %s", strerror(errno));
exit(1);
}
if ((fd = open(_PATH_DEVNULL, O_RDWR, 0)) == -1) {
log_alert("%s: %s", _PATH_DEVNULL,
strerror(errno));
exit(1);
}
dup2(fd, fileno(stderr));
close(fd);
}
fflush(NULL);
errno = 0;
if ((filep = popen(pCommandString, "r")) == NULL) {
/* popen() does not set errno reliably ... */
if (errno)
log_error("execution error: %s: %s",
pCommandString, strerror(errno));
else
log_error("execution error: %s",
pCommandString);
} else {
*popenFlag = 1;
}
xfree(pCommandString);
if (cfg_get_program_quiet_stderr())
dup2(stderr_fd, fileno(stderr));
if (stderr_fd != -1)
close(stderr_fd);
return (filep);
}
if (md_p != NULL)
*md_p = md;
else
mdata_destroy(&md);
if ((filep = fopen(filename, "rb")) == NULL) {
log_error("%s: %s", filename, strerror(errno));
return (NULL);
}
return (filep);
}
int
reconnect(stream_t stream)
{
unsigned int i;
cfg_server_t cfg_server = stream_get_cfg_server(stream);
i = 0;
while (++i) {
if (cfg_server_get_reconnect_attempts(cfg_server) > 0)
log_notice("reconnect: %s: attempt #%u/%u ...",
cfg_server_get_hostname(cfg_server), i,
cfg_server_get_reconnect_attempts(cfg_server));
else
log_notice("reconnect: %s: attempt #%u ...",
cfg_server_get_hostname(cfg_server), i);
stream_disconnect(stream);
if (0 == stream_connect(stream)) {
log_notice("reconnect: %s: success",
cfg_server_get_hostname(cfg_server));
return (0);
}
if (cfg_server_get_reconnect_attempts(cfg_server) > 0 &&
i >= cfg_server_get_reconnect_attempts(cfg_server))
break;
if (quit)
return (-1);
else
sleep(5);
};
log_warning("reconnect failed: giving up");
return (-1);
}
const char *
getTimeString(long seconds)
{
static char str[25];
long secs, mins, hours;
if (seconds < 0)
return (NULL);
secs = seconds;
hours = secs / 3600;
secs %= 3600;
mins = secs / 60;
secs %= 60;
snprintf(str, sizeof(str), "%ldh%02ldm%02lds", hours, mins, secs);
return ((const char *)str);
}
int
sendStream(stream_t stream, FILE *filepstream, const char *fileName,
int isStdin, const char *songLenStr, struct timespec *tv)
{
char buff[4096];
size_t bytes_read, total, oldTotal;
int ret;
double kbps = -1.0;
struct timespec timeStamp, *startTime = tv;
struct timespec callTime, currentTime;
cfg_server_t cfg_server = stream_get_cfg_server(stream);
cfg_intake_t cfg_intake = stream_get_cfg_intake(stream);
clock_gettime(CLOCK_MONOTONIC, &callTime);
timeStamp.tv_sec = startTime->tv_sec;
timeStamp.tv_nsec = startTime->tv_nsec;
total = oldTotal = 0;
ret = STREAM_DONE;
while ((bytes_read = fread(buff, 1, sizeof(buff), filepstream)) > 0) {
if (!stream_get_connected(stream)) {
log_warning("%s: connection lost",
cfg_server_get_hostname(cfg_server));
if (0 > reconnect(stream)) {
ret = STREAM_SERVERR;
break;
}
}
stream_sync(stream);
if (0 > stream_send(stream, buff, bytes_read)) {
if (0 > reconnect(stream))
ret = STREAM_SERVERR;
break;
}
if (quit)
break;
if (rereadPlaylist_notify) {
rereadPlaylist_notify = 0;
if (CFG_INTAKE_PLAYLIST == cfg_intake_get_type(cfg_intake))
log_notice("HUP signal received: playlist re-read scheduled");
}
if (skipTrack) {
skipTrack = 0;
ret = STREAM_SKIP;
break;
}
clock_gettime(CLOCK_MONOTONIC, &currentTime);
if (queryMetadata ||
(0 <= cfg_get_metadata_refresh_interval() &&
(currentTime.tv_sec - callTime.tv_sec >=
cfg_get_metadata_refresh_interval()))) {
queryMetadata = 0;
if (cfg_get_metadata_program()) {
ret = STREAM_UPDMDATA;
break;
}
}
total += bytes_read;
if (cfg_get_program_rtstatus_output()) {
double oldTime, newTime;
if (!isStdin && playlistMode) {
if (CFG_INTAKE_PROGRAM == cfg_intake_get_type(cfg_intake)) {
char *tmp = xstrdup(cfg_intake_get_filename(cfg_intake));
printf(" [%s]", basename(tmp));
xfree(tmp);
} else
printf(" [%4lu/%-4lu]",
playlist_get_position(playlist),
playlist_get_num_items(playlist));
}
oldTime = (double)timeStamp.tv_sec
+ (double)timeStamp.tv_nsec / 1000000000.0;
newTime = (double)currentTime.tv_sec
+ (double)currentTime.tv_nsec / 1000000000.0;
if (songLenStr == NULL)
printf(" [ %s]",
getTimeString(currentTime.tv_sec -
startTime->tv_sec));
else
printf(" [ %s/%s]",
getTimeString(currentTime.tv_sec -
startTime->tv_sec),
songLenStr);
if (newTime - oldTime >= 1.0) {
kbps = (((double)(total - oldTotal)
/ (newTime - oldTime)) * 8.0) / 1000.0;
timeStamp.tv_sec = currentTime.tv_sec;
timeStamp.tv_nsec = currentTime.tv_nsec;
oldTotal = total;
}
if (kbps < 0)
printf(" ");
else
printf(" [%8.2f kbps]", kbps);
printf(" \r");
fflush(stdout);
}
}
if (ferror(filepstream)) {
if (errno == EINTR) {
clearerr(filepstream);
ret = STREAM_CONT;
} else if (errno == EBADF && isStdin)
log_notice("no (more) data available on standard input");
else
log_error("sendStream: %s: %s", fileName,
strerror(errno));
}
return (ret);
}
int
streamFile(stream_t stream, const char *fileName)
{
FILE *filepstream = NULL;
int popenFlag = 0;
char *songLenStr = NULL;
int ret, retval = 0;
long songLen;
mdata_t md = NULL;
struct timespec startTime;
cfg_stream_t cfg_stream = stream_get_cfg_stream(stream);
cfg_intake_t cfg_intake = stream_get_cfg_intake(stream);
int isStdin = cfg_intake_get_type(cfg_intake) == CFG_INTAKE_STDIN;
if ((filepstream = openResource(stream, fileName, &popenFlag, &md, &isStdin, &songLen))
== NULL) {
mdata_destroy(&md);
if (++resource_errors > 100) {
log_error("too many errors; giving up");
return (0);
}
/* Continue with next resource on failure: */
return (1);
}
resource_errors = 0;
if (md != NULL) {
const char *tmp;
char *metaData;
tmp = mdata_get_songinfo(md) ?
mdata_get_songinfo(md) : mdata_get_name(md);
metaData = util_utf82char(tmp);
log_notice("streaming: %s (%s)", metaData,
isStdin ? "stdin" : fileName);
xfree(metaData);
/* MP3 streams are special, so set the metadata explicitly: */
if (CFG_STREAM_MP3 == cfg_stream_get_format(cfg_stream))
stream_set_metadata(stream, md, NULL);
mdata_destroy(&md);
} else if (isStdin)
log_notice("streaming: standard input");
if (songLen > 0)
songLenStr = xstrdup(getTimeString(songLen));
clock_gettime(CLOCK_MONOTONIC, &startTime);
do {
ret = sendStream(stream, filepstream, fileName, isStdin,
songLenStr, &startTime);
if (quit)
break;
if (ret != STREAM_DONE) {
if ((skipTrack && rereadPlaylist) ||
(skipTrack && queryMetadata)) {
skipTrack = 0;
ret = STREAM_CONT;
}
if (queryMetadata && rereadPlaylist) {
queryMetadata = 0;
ret = STREAM_CONT;
}
if (ret == STREAM_SKIP || skipTrack) {
skipTrack = 0;
if (!isStdin)
log_notice("USR1 signal received: skipping current track");
retval = 1;
ret = STREAM_DONE;
}
if (ret == STREAM_UPDMDATA || queryMetadata) {
queryMetadata = 0;
if (cfg_get_metadata_no_updates())
continue;
if (cfg_get_metadata_program()) {
char *mdataStr = NULL;
log_info("running metadata program: %s",
cfg_get_metadata_program());
md = mdata_create();
if (0 > mdata_run_program(md, cfg_get_metadata_program()) ||
0 > stream_set_metadata(stream, md, &mdataStr)) {
mdata_destroy(&md);
retval = 0;
ret = STREAM_DONE;
continue;
}
mdata_destroy(&md);
log_info("new metadata: %s", mdataStr);
xfree(mdataStr);
}
}
if (ret == STREAM_SERVERR) {
retval = 0;
ret = STREAM_DONE;
}
} else
retval = 1;
} while (ret != STREAM_DONE);
if (popenFlag)
pclose(filepstream);
else if (!isStdin)
fclose(filepstream);
if (songLenStr != NULL)
xfree(songLenStr);
return (retval);
}
int
streamPlaylist(stream_t stream)
{
int ret;
const char *song_next;
char song_prev[PATH_MAX];
char *song_path;
char tmp_path[2 * PATH_MAX];
cfg_intake_t cfg_intake = stream_get_cfg_intake(stream);
if (playlist == NULL) {
switch (cfg_intake_get_type(cfg_intake)) {
case CFG_INTAKE_PROGRAM:
if ((playlist = playlist_program(cfg_intake_get_filename(cfg_intake))) == NULL)
return (0);
break;
case CFG_INTAKE_STDIN:
if ((playlist = playlist_read(NULL)) == NULL)
return (0);
break;
default:
if ((playlist = playlist_read(cfg_intake_get_filename(cfg_intake))) == NULL)
return (0);
if (playlist_get_num_items(playlist) == 0)
log_warning("%s: playlist empty",
cfg_intake_get_filename(cfg_intake));
break;
}
} else {
/*
* XXX: This preserves traditional behavior, however,
* rereading the playlist after each walkthrough seems a
* bit more logical.
*/
playlist_rewind(playlist);
}
if (CFG_INTAKE_PROGRAM != cfg_intake_get_type(cfg_intake) &&
cfg_intake_get_shuffle(cfg_intake))
playlist_shuffle(playlist);
while ((song_next = playlist_get_next(playlist)) != NULL) {
strlcpy(song_prev, song_next, sizeof(song_prev));
if ('/' == song_next[0] ||
!cfg_intake_get_rel_to_list(cfg_intake))
(void)snprintf(tmp_path, sizeof(tmp_path), "%s",
song_next);
else
(void)snprintf(tmp_path, sizeof(tmp_path), "%s/%s",
playlist_get_location(playlist), song_next);
song_path = realpath(tmp_path, NULL);
if (NULL == song_path) {
log_warning("%s: %s", song_next, strerror(errno));
return (0);
}
ret = streamFile(stream, song_path);
free(song_path);
if (!ret)
return (0);
if (quit)
break;
if (rereadPlaylist) {
rereadPlaylist = rereadPlaylist_notify = 0;
if (CFG_INTAKE_PROGRAM == cfg_intake_get_type(cfg_intake))
continue;
log_notice("rereading playlist");
if (!playlist_reread(&playlist))
return (0);
if (cfg_intake_get_shuffle(cfg_intake))
playlist_shuffle(playlist);
else {
playlist_goto_entry(playlist, song_prev);
playlist_skip_next(playlist);
}
continue;
}
}
return (1);
}
int
ez_shutdown(int exitval)
{
if (main_stream)
stream_destroy(&main_stream);
stream_exit();
playlist_exit();
log_exit();
cfg_exit();
return (exitval);
}
int
main(int argc, char *argv[])
{
int ret, cont;
const char *errstr;
extern char *optarg;
extern int optind;
struct sigaction act;
unsigned int i;
cfg_server_t cfg_server;
cfg_stream_t cfg_stream;
cfg_intake_t cfg_intake;
ret = 1;
if (0 > cfg_init() ||
0 > cmdline_parse(argc, argv, &ret) ||
0 > log_init(cfg_get_program_name()) ||
0 > playlist_init() ||
0 > cfg_file_reload() ||
0 > stream_init())
return (ez_shutdown(ret));
log_set_verbosity(cfg_get_program_verbosity());
if (0 > cfg_check(&errstr)) {
log_error("%s: %s", cfg_get_program_config_file(), errstr);
return (ez_shutdown(2));
}
main_stream = stream_create(CFG_DEFAULT);
if (0 > stream_configure(main_stream)) {
stream_destroy(&main_stream);
return (ez_shutdown(1));
}
cfg_server = stream_get_cfg_server(main_stream);
cfg_stream = stream_get_cfg_stream(main_stream);
cfg_intake = stream_get_cfg_intake(main_stream);
memset(&act, 0, sizeof(act));
act.sa_handler = sig_handler;
#ifdef SA_RESTART
act.sa_flags = SA_RESTART;
#endif
for (i = 0; i < sizeof(ezstream_signals) / sizeof(int); i++) {
if (sigaction(ezstream_signals[i], &act, NULL) == -1) {
log_syserr(ERROR, errno, "sigaction");
stream_destroy(&main_stream);
return (ez_shutdown(1));
}
}
/*
* Ignore SIGPIPE, which has been seen to give a long-running ezstream
* process trouble. EOF and/or EPIPE are also easier to handle.
*/
#ifndef SIG_IGN
# define SIG_IGN (void (*)(int))1
#endif /* !SIG_IGN */
act.sa_handler = SIG_IGN;
if (sigaction(SIGPIPE, &act, NULL) == -1) {
log_syserr(ERROR, errno, "sigaction");
stream_destroy(&main_stream);
return (ez_shutdown(1));
}
if (0 > util_write_pid_file(cfg_get_program_pid_file()))
log_syserr(WARNING, errno, cfg_get_program_pid_file());
if (0 > stream_connect(main_stream)) {
log_error("initial server connection failed");
stream_destroy(&main_stream);
return (ez_shutdown(1));
}
log_notice("connected: %s://%s:%u%s",
cfg_server_get_protocol_str(cfg_server),
cfg_server_get_hostname(cfg_server),
cfg_server_get_port(cfg_server),
cfg_stream_get_mountpoint(cfg_stream));
if (CFG_INTAKE_PROGRAM == cfg_intake_get_type(cfg_intake) ||
CFG_INTAKE_PLAYLIST == cfg_intake_get_type(cfg_intake) ||
(CFG_INTAKE_AUTODETECT == cfg_intake_get_type(cfg_intake) &&
(util_strrcasecmp(cfg_intake_get_filename(cfg_intake), ".m3u") == 0 ||
util_strrcasecmp(cfg_intake_get_filename(cfg_intake), ".txt") == 0)))
playlistMode = 1;
else
playlistMode = 0;
do {
if (playlistMode) {
cont = streamPlaylist(main_stream);
} else {
cont = streamFile(main_stream,
cfg_intake_get_filename(cfg_intake));
}
if (quit)
break;
if (cfg_intake_get_stream_once(cfg_intake))
break;
} while (cont);
stream_disconnect(main_stream);
stream_destroy(&main_stream);
if (quit) {
if (cfg_get_program_quiet_stderr() &&
cfg_get_program_verbosity())
printf("\r");
log_notice("INT or TERM signal received");
}
log_info("exiting");
playlist_free(&playlist);
return (ez_shutdown(0));
}
View Git Blame Copy Permalink