Commit 657d02fb authored by Daniel Stenberg's avatar Daniel Stenberg
Browse files

rtsp.c: converted to C

Trimmed the newlines to be LF-only. Converted the source to plain C, to
use curl style indents, to compile warning-free with picky options and
fixed the minor fprintf() bug on line 245. Added to makefile.
parent 8e2de867
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ check_PROGRAMS = 10-at-a-time anyauthput cookie_interface debug fileupload \
  https multi-app multi-debugcallback multi-double multi-post multi-single \
  persistant post-callback postit2 sepheaders simple simplepost simplessl  \
  sendrecv httpcustomheader certinfo chkspeed ftpgetinfo ftp-wildcard \
  smtp-multi simplesmtp smtp-tls
  smtp-multi simplesmtp smtp-tls rtsp

# These examples require external dependencies that may not be commonly
# available on POSIX systems, so don't bother attempting to compile them here.
+271 −253
Original line number Diff line number Diff line
@@ -33,18 +33,20 @@
#include <string.h>

#if defined (WIN32)
#  include <conio.h>  // _getch()
#  include <conio.h>  /* _getch() */
#else
#  include <termios.h>
#  include <unistd.h>

  int _getch(void) {
static int _getch(void)
{
  struct termios oldt, newt;
  int ch;
  tcgetattr( STDIN_FILENO, &oldt );
  newt = oldt;
  newt.c_lflag &= ~( ICANON | ECHO );
  tcsetattr( STDIN_FILENO, TCSANOW, &newt );
    int ch = getchar();
  ch = getchar();
  tcsetattr( STDIN_FILENO, TCSANOW, &oldt );
  return ch;
}
@@ -54,18 +56,20 @@

#define VERSION_STR  "V1.0"

// error handling macros
/* error handling macros */
#define my_curl_easy_setopt(A, B, C) \
  if ((res = curl_easy_setopt((A), (B), (C))) != CURLE_OK) \
    fprintf(stderr, "curl_easy_setopt(%s, %s, %s) failed: %d\n", #A, #B, #C, res);
    fprintf(stderr, "curl_easy_setopt(%s, %s, %s) failed: %d\n", \
            #A, #B, #C, res);

#define my_curl_easy_perform(A) \
  if ((res = curl_easy_perform((A))) != CURLE_OK) \
    fprintf(stderr, "curl_easy_perform(%s) failed: %d\n", #A, res);


// send RTSP OPTIONS request
void rtsp_options(CURL *curl, const char *uri) {
/* send RTSP OPTIONS request */
static void rtsp_options(CURL *curl, const char *uri)
{
  CURLcode res = CURLE_OK;
  printf("\nRTSP: OPTIONS %s\n", uri);
  my_curl_easy_setopt(curl, CURLOPT_RTSP_STREAM_URI, uri);
@@ -74,15 +78,18 @@ void rtsp_options(CURL *curl, const char *uri) {
}


// send RTSP DESCRIBE request and write sdp response to a file
void rtsp_describe(CURL *curl, const char *uri, const char *sdp_filename) {
/* send RTSP DESCRIBE request and write sdp response to a file */
static void rtsp_describe(CURL *curl, const char *uri,
                          const char *sdp_filename)
{
  CURLcode res = CURLE_OK;
    printf("\nRTSP: DESCRIBE %s\n", uri);
  FILE *sdp_fp = fopen(sdp_filename, "wt");
  printf("\nRTSP: DESCRIBE %s\n", uri);
  if (sdp_fp == NULL) {
    fprintf(stderr, "Could not open '%s' for writing\n", sdp_filename);
    sdp_fp = stdout;
    } else {
  }
  else {
    printf("Writing SDP to '%s'\n", sdp_filename);
  }
  my_curl_easy_setopt(curl, CURLOPT_WRITEDATA, sdp_fp);
@@ -94,9 +101,9 @@ void rtsp_describe(CURL *curl, const char *uri, const char *sdp_filename) {
  }
}


// send RTSP SETUP request
void rtsp_setup(CURL *curl, const char *uri, const char *transport) {
/* send RTSP SETUP request */
static void rtsp_setup(CURL *curl, const char *uri, const char *transport)
{
  CURLcode res = CURLE_OK;
  printf("\nRTSP: SETUP %s\n", uri);
  printf("      TRANSPORT %s\n", transport);
@@ -107,8 +114,9 @@ void rtsp_setup(CURL *curl, const char *uri, const char *transport) {
}


// send RTSP PLAY request
void rtsp_play(CURL *curl, const char *uri, const char *range) {
/* send RTSP PLAY request */
static void rtsp_play(CURL *curl, const char *uri, const char *range)
{
  CURLcode res = CURLE_OK;
  printf("\nRTSP: PLAY %s\n", uri);
  my_curl_easy_setopt(curl, CURLOPT_RTSP_STREAM_URI, uri);
@@ -118,8 +126,9 @@ void rtsp_play(CURL *curl, const char *uri, const char *range) {
}


// send RTSP TEARDOWN request
void rtsp_teardown(CURL *curl, const char *uri) {
/* send RTSP TEARDOWN request */
static void rtsp_teardown(CURL *curl, const char *uri)
{
  CURLcode res = CURLE_OK;
  printf("\nRTSP: TEARDOWN %s\n", uri);
  my_curl_easy_setopt(curl, CURLOPT_RTSP_REQUEST, CURL_RTSPREQ_TEARDOWN);
@@ -127,10 +136,11 @@ void rtsp_teardown(CURL *curl, const char *uri) {
}


// convert url into an sdp filename
void get_sdp_filename(const char *url, char *sdp_filename) {
    strcpy(sdp_filename, "video.sdp");
/* convert url into an sdp filename */
static void get_sdp_filename(const char *url, char *sdp_filename)
{
  const char *s = strrchr(url, '/');
  strcpy(sdp_filename, "video.sdp");
  if (s != NULL) {
    s++;
    if (s[0] != '\0') {
@@ -140,35 +150,41 @@ void get_sdp_filename(const char *url, char *sdp_filename) {
}


// scan sdp file for media control attribute
void get_media_control_attribute(const char *sdp_filename, char *control) {
    control[0] = '\0';
/* scan sdp file for media control attribute */
static void get_media_control_attribute(const char *sdp_filename,
                                        char *control)
{
  int max_len = 256;
    char *s = new char[max_len];
  char *s = malloc(max_len);
  FILE *sdp_fp = fopen(sdp_filename, "rt");
  control[0] = '\0';
  if (sdp_fp != NULL) {
    while (fgets(s, max_len - 2, sdp_fp) != NULL) {
      sscanf(s, " a = control: %s", control);
    }
    fclose(sdp_fp);
  }
    delete []s;
  free(s);
}


// main app
int main(int argc, char * const argv[]) {
    const char *transport = "RTP/AVP;unicast;client_port=1234-1235";  // UDP		
//    const char *transport = "RTP/AVP/TCP;unicast;client_port=1234-1235";  // TCP
/* main app */
int main(int argc, char * const argv[])
{
#if 1
  const char *transport = "RTP/AVP;unicast;client_port=1234-1235";  /* UDP */
#else
  const char *transport = "RTP/AVP/TCP;unicast;client_port=1234-1235";  /* TCP */
#endif
  const char *range = "0.000-";
  int rc = EXIT_SUCCESS;
  char *basename = NULL;

  printf("\nRTSP request %s\n", VERSION_STR);
  printf("    Project web site: http://code.google.com/p/rtsprequest/\n");
  printf("    Requires cURL V7.20 or greater\n\n");

    // check command line
    char *basename = NULL;
  /* check command line */
  if ((argc != 2) && (argc != 3)) {
    basename = strrchr(argv[0], '/');
    if (basename == NULL) {
@@ -187,54 +203,55 @@ int main(int argc, char * const argv[]) {
    rc = EXIT_FAILURE;
  } else {
    const char *url = argv[1];
        char *uri = new char[strlen(url) + 32];
        char *sdp_filename = new char[strlen(url) + 32];
        char *control = new char[strlen(url) + 32];
    char *uri = malloc(strlen(url) + 32);
    char *sdp_filename = malloc(strlen(url) + 32);
    char *control = malloc(strlen(url) + 32);
    CURLcode res;
    get_sdp_filename(url, sdp_filename);
    if (argc == 3) {
      transport = argv[2];
    }

        // initialize curl
        CURLcode res = CURLE_OK;
    /* initialize curl */
    res = curl_global_init(CURL_GLOBAL_ALL);
    if (res == CURLE_OK) {
      curl_version_info_data *data = curl_version_info(CURLVERSION_NOW);
      CURL *curl;
      fprintf(stderr, "    cURL V%s loaded\n", data->version);

            // initialize this curl session
            CURL *curl = curl_easy_init();
      /* initialize this curl session */
      curl = curl_easy_init();
      if (curl != NULL) {
        my_curl_easy_setopt(curl, CURLOPT_VERBOSE, 0L);
        my_curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1L);
        my_curl_easy_setopt(curl, CURLOPT_WRITEHEADER, stdout);
        my_curl_easy_setopt(curl, CURLOPT_URL, url);

                // request server options
        /* request server options */
        sprintf(uri, "%s", url);
        rtsp_options(curl, uri);

                // request session description and write response to sdp file
        /* request session description and write response to sdp file */
        rtsp_describe(curl, uri, sdp_filename);

                // get media control attribute from sdp file
        /* get media control attribute from sdp file */
        get_media_control_attribute(sdp_filename, control);

                // setup media stream
        /* setup media stream */
        sprintf(uri, "%s/%s", url, control);
        rtsp_setup(curl, uri, transport);

                // start playing media stream
        /* start playing media stream */
        sprintf(uri, "%s/", url);
        rtsp_play(curl, uri, range);
        printf("Playing video, press any key to stop ...");
        _getch();
        printf("\n");

                // teardown session
        /* teardown session */
        rtsp_teardown(curl, uri);

                // cleanup
        /* cleanup */
        curl_easy_cleanup(curl);
        curl = NULL;
      } else {
@@ -242,11 +259,12 @@ int main(int argc, char * const argv[]) {
      }
      curl_global_cleanup();
    } else {
            fprintf(stderr, "curl_global_init(%s) failed\n", "CURL_GLOBAL_ALL", res);
      fprintf(stderr, "curl_global_init(%s) failed: %d\n",
              "CURL_GLOBAL_ALL", res);
    }
        delete []control;
        delete []sdp_filename;
        delete []uri;
    free(control);
    free(sdp_filename);
    free(uri);
  }

  return rc;