int main(int argc, char **argv)
{
int opt;
char *url, *out = "nemesi.dump";
FILE *outfile = NULL;
rtsp_ctrl *ctl;
rtsp_session *sess;
rtsp_medium *med;
nms_rtsp_hints rtsp_hints = { -1 };
sdp_attr *attr;
RTSP_Error reply;
if (argc < 2) {
fprintf(stderr, "\tPlease specify at least an url.\n");
fprintf(stderr,
"\tUsage: %s [-f outputfile ] [-p rtp_port] [-t url\n",
argv[0]);
exit(1);
}
#ifndef WIN32
while ((opt = getopt(argc, argv, "df:p:v:")) != -1) {
switch (opt) {
/* Set output file */
case 'f':
out = strdup(optarg);
break;
/* Set rtp port */
case 'p':
rtsp_hints.first_rtp_port = atoi(optarg);
break;
/* Set verbosity */
case 'v':
nms_verbosity_set(atoi(optarg));
break;
/* Unknown option */
case '?':
fprintf(stderr, "\n Unknown option `-%c'.\n", optopt);
fprintf(stderr,
"\tUsage: %s [-f outputfile ] [-d] url\n\n",
argv[0]);
return 1;
}
}
#endif
outfile = fopen(out, "rb");
if (outfile == NULL)
outfile = stderr;
url = argv[argc - 1];
fprintf(stderr, "URL %s.\n", url);
/* initialize the rtsp state machine, starts the rtsp and the rtp threads
* the hints available are just one:
* - the first port to use (instead of picking one at random)
*/
if ((ctl = rtsp_init(&rtsp_hints)) == NULL) {
fprintf(stderr, "Cannot init rtsp.\n");
return 1;
}
if (rtsp_open(ctl, url)) {
fprintf(stderr, "rtsp_open failed.\n");
// die
return 1;
}
// you must call rtsp_wait after issuing any command
reply = rtsp_wait(ctl);
printf("OPEN: Received reply from server: %s\n", reply.message.reply_str);
// Get the session information
sess = ctl->rtsp_queue;
if (!sess) {
fprintf(stderr, "No session available.\n");
return 1;
}
while (sess) { // foreach session...
fprintf(outfile, "\tSession %s\n", sess->pathname);
fprintf(outfile, "\tSession Duration %s\n", sess->info->t);
for (attr = sess->info->attr_list;
attr;
attr = attr->next) {
fprintf(outfile, "\t* %s %s\n", attr->name, attr->value);
}
med = sess->media_queue;
while (med) { //... foreach medium
switch (med->medium_info->media_type) {
// Just care about audio and video
case 'A':
case 'V':
fprintf(outfile, "\tTransport %s\n",
med->medium_info->transport);
fprintf(outfile, "\tMedia Type %s\n",
(med->medium_info->media_type == 'A'?
"Audio" : "Video"));
fprintf(outfile, "\tMedia format %s\n",
med->medium_info->fmts);
break;
default:
//do nothing
break;
}
// attributes are already parsed, get them from the list
for (attr = med->medium_info->attr_list;
attr;
attr = attr->next) {
fprintf(outfile, "\t* %s %s\n", attr->name, attr->value);
}
med = med->next;
}
sess = sess->next;
}
/*
* Close the rtsp connection, we are polite
*/
rtsp_close(ctl);
reply = rtsp_wait(ctl);
printf("CLOSE: Received reply from server: %s\n", reply.message.reply_str);
/*
* Kill the threads, dealloc everything.
*/
rtsp_uninit(ctl);
return 0;
}
static demuxer_t* demux_open_rtp(demuxer_t* demuxer)
{
nms_rtsp_hints hints;
char * url = demuxer->stream->streaming_ctrl->url->url;
rtsp_ctrl * ctl;
RTSP_Error reply;
rtsp_medium * media;
Nemesi_DemuxerStreamData * ndsd = calloc(1, sizeof(Nemesi_DemuxerStreamData));
memset(&hints,0,sizeof(hints));
if (rtsp_port) hints.first_rtp_port = rtsp_port;
if (rtsp_transport_tcp) {
hints.pref_rtsp_proto = TCP;
hints.pref_rtp_proto = TCP;
}
if (rtsp_transport_sctp) {
hints.pref_rtsp_proto = SCTP;
hints.pref_rtp_proto = SCTP;
}
mp_msg(MSGT_DEMUX, MSGL_INFO, "Initializing libNemesi\n");
if ((ctl = rtsp_init(&hints)) == NULL) {
free(ndsd);
return STREAM_ERROR;
}
ndsd->rtsp = ctl;
demuxer->priv = ndsd;
//nms_verbosity_set(1);
mp_msg(MSGT_DEMUX, MSGL_INFO, "Opening: %s\n", url);
if (rtsp_open(ctl, url)) {
mp_msg(MSGT_DEMUX, MSGL_ERR, "rtsp_open failed.\n");
return demuxer;
}
reply = rtsp_wait(ctl);
if (reply.got_error) {
mp_msg(MSGT_DEMUX, MSGL_ERR,
"OPEN Error from the server: %s\n",
reply.message.reply_str);
return demuxer;
}
rtsp_play(ctl, 0, 0);
reply = rtsp_wait(ctl);
if (reply.got_error) {
mp_msg(MSGT_DEMUX, MSGL_ERR,
"PLAY Error from the server: %s\n",
reply.message.reply_str);
return demuxer;
}
if (!ctl->rtsp_queue)
return demuxer;
media = ctl->rtsp_queue->media_queue;
for (; media; media=media->next) {
sdp_medium_info * info = media->medium_info;
rtp_session * sess = media->rtp_sess;
rtp_buff buff;
int media_format = atoi(info->fmts);
rtp_pt * ptinfo = rtp_get_pt_info(sess, media_format);
char const * format_name = ptinfo ? ptinfo->name : NULL;
memset(&buff, 0, sizeof(rtp_buff));
if (sess->parsers[media_format] == NULL) {
mp_msg(MSGT_DEMUX, MSGL_ERR,
"libNemesi unsupported media format: %s\n",
format_name ? format_name : info->fmts);
continue;
}
else {
mp_msg(MSGT_DEMUX, MSGL_INFO,
"libNemesi supported media: %s\n",
format_name);
}
if (ptinfo->type == AU) {
if (ndsd->session[NEMESI_SESSION_AUDIO] == NULL) {
sh_audio_t* sh_audio = new_sh_audio(demuxer,0, NULL);
WAVEFORMATEX* wf;
demux_stream_t* d_audio = demuxer->audio;
demuxer->audio->id = 0;
mp_msg(MSGT_DEMUX, MSGL_INFO, "Detected as AUDIO stream...\n");
link_session_and_fetch_conf(ndsd, NEMESI_SESSION_AUDIO,
sess, &buff, NULL);
wf = calloc(1,sizeof(*wf)+buff.len);
wf->cbSize = buff.len;
memcpy(wf+1, buff.data, buff.len);
sh_audio->wf = wf;
d_audio->sh = sh_audio;
wf->nSamplesPerSec = 0;
wf->wFormatTag =
sh_audio->format = get4CC(supported_audio, format_name);
if ( !(wf->wFormatTag) )
mp_msg(MSGT_DEMUX, MSGL_WARN,
"Unknown MPlayer format code for MIME"
" type \"audio/%s\"\n", format_name);
} else {
mp_msg(MSGT_DEMUX, MSGL_ERR,
"There is already an audio session registered,"
" ignoring...\n");
}
} else if (ptinfo->type == VI) {
if (ndsd->session[NEMESI_SESSION_VIDEO] == NULL) {
sh_video_t* sh_video;
BITMAPINFOHEADER* bih;
demux_stream_t* d_video;
int fps = 0;
mp_msg(MSGT_DEMUX, MSGL_INFO, "Detected as VIDEO stream...\n");
link_session_and_fetch_conf(ndsd, NEMESI_SESSION_VIDEO,
sess, &buff, &fps);
bih = calloc(1,sizeof(*bih)+buff.len);
bih->biSize = sizeof(*bih)+buff.len;
memcpy(bih+1, buff.data, buff.len);
sh_video = new_sh_video(demuxer,0);
sh_video->bih = bih;
d_video = demuxer->video;
d_video->sh = sh_video;
if (fps) {
sh_video->fps = fps;
sh_video->frametime = 1.0/fps;
}
bih->biCompression =
sh_video->format = get4CC(supported_video, format_name);
if ( !(bih->biCompression) ) {
mp_msg(MSGT_DEMUX, MSGL_WARN,
"Unknown MPlayer format code for MIME"
" type \"video/%s\"\n", format_name);
}
} else {
mp_msg(MSGT_DEMUX, MSGL_ERR,
"There is already a video session registered,"
" ignoring...\n");
}
} else {
mp_msg(MSGT_DEMUX, MSGL_ERR, "Unsupported media type\n");
}
}
demuxer->stream->eof = 0;
return demuxer;
}
int main(int argc, char **argv)
{
int do_fork = 1;
char *config_fn = DEFAULT_CONFIG_FILE;
char *pid_fn = DEFAULT_PID_FILE;
int c;
evloop = EV_DEFAULT;
ev_started_at = ev_time();
opterr = 0;
while ((c = getopt (argc, argv, "vhdfc:p:s:l:")) != -1) {
switch (c) {
case 'v':
verbose++;
break;
case 'h':
usage(0);
return 0;
case 'd':
debug = 1;
do_fork = 0;
break;
case 'f':
do_fork = 0;
break;
case 'c':
config_fn = optarg;
break;
case 'p':
pid_fn = optarg;
break;
case 's':
server_str_id = strdup(optarg);
break;
case 'l':
server_listen_port = atoi(optarg);
if (server_listen_port < 1 || server_listen_port > 65534){
fprintf (stderr, "Invalid listen port: %s.\n", optarg);
return 1;
}
break;
case '?':
if ((optopt == 'c') || (optopt == 'p') || (optopt == 's') || (optopt == 'l'))
fprintf (stderr, "Option -%c requires an argument.\n", optopt);
else if (isprint (optopt))
fprintf (stderr, "Unknown option `-%c'.\n", optopt);
else
fprintf (stderr, "Unknown option character `\\x%x'.\n", optopt);
return 1;
default:
usage();
return 1;
}
}
if (optind < argc) {
fprintf (stderr, "Unknown option: %s.\n", argv[optind]);
return 1;
}
rtsp_server = rtsp_alloc();
if (!rtsp_server){
return 0;
}
log_to_stderr = !do_fork;
if (!log_to_stderr)
openlog(PROGRAM_NAME, LOG_PID, LOG_DAEMON);
if (load_config(config_fn)) {
fprintf (stderr, "Error loading config file.\n");
return 2;
}
my_random_init();
if (do_fork){
log_debug("forking...");
if (daemon(0, 0)){
log_error("daemonize failed: %s", strerror(errno));
return -1;
}
if (write_pid_file(pid_fn)){
log_warning("failed to create pid file");
}
ev_loop_fork(evloop);
}
if (!do_fork) {
ev_signal_init (&signal_watcher1, sig_term_cb, SIGINT);
ev_signal_start (evloop, &signal_watcher1);
}
ev_signal_init (&signal_watcher2, sig_term_cb, SIGTERM);
ev_signal_start (evloop, &signal_watcher2);
ev_signal_init (&signal_watcher3, sig_ignore_cb, SIGHUP);
ev_signal_start (evloop, &signal_watcher3);
ev_signal_init (&signal_watcher4, sig_ignore_cb, SIGPIPE);
ev_signal_start (evloop, &signal_watcher4);
if (csconv_init()){
return 0;
}
if (rtsp_init(rtsp_server)){
csconv_cleanup();
return 0;
}
elevate_privileges();
setup_io_prio();
drop_privileges();
log_info("starting...");
ev_run(evloop, 0);
log_info("cleaning up...");
rtsp_cleanup(rtsp_server);
csconv_cleanup();
if (server_str_id)
free(server_str_id);
log_info("exiting...");
if (!log_to_stderr)
closelog();
return 0;
}
