xiph-rtp/theorartp.c

   1 /*
   2   File: Vorbis RTP Server
   3   Authors: Phil Kerr, Luca Barbato
   4   Date: 05/01/2005
   5   Platform: Linux
   6
   7   Copyright (c) 2006, Fluendo / Xiph.Org
   8
   9   Redistribution and use in source and binary forms, with or without
  10   modification, are permitted provided that the following conditions are met:
  11
  12   * Redistributions of source code must retain the above copyright notice,
  13     this list of conditions and the following disclaimer.
  14
  15   * Redistributions in binary form must reproduce the above copyright notice,
  16     this list of conditions and the following disclaimer in the documentation
  17     and/or other materials provided with the distribution.
  18
  19   * Neither the name of the Xiph.Org nor the names of its contributors may
  20     be used to endorse or promote products derived from this software without
  21     specific prior written permission.
  22
  23   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  25   TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  26   PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
  27   BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  28   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  29   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  30   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  31   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  32   ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  33   THE POSSIBILITY OF SUCH DAMAGE.
  34
  35 */
  36
  37 /*  System includes  */
  38
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #include <unistd.h>
  43 #include <time.h>
  44
  45 /*  Codec includes  */
  46
  47 #include <theora/theora.h>
  48
  49 /*  Network includes  */
  50
  51 #include <arpa/inet.h>
  52 #include <netinet/in.h>
  53 #include <netdb.h>
  54 #include <sys/types.h>
  55 #include <sys/socket.h>
  56
  57 /* Common include  */
  58
  59 #define HAVE_THEORA
  60
  61 #include "xiph_rtp.h"
  62
  63 #define BUFFER_SIZE 4096
  64 unsigned int cfg_parse( xiph_rtp_t *xr )
  65 {
  66         oggpack_buffer opb;
  67
  68         oggpackB_readinit(&opb,xr->header[0].packet, xr->header[0].bytes);
  69
  70         oggpackB_read(&opb,8*7); // 0x80"theora"
  71         oggpackB_read(&opb,8*3); // version
  72         oggpackB_read(&opb,16); //w
  73         oggpackB_read(&opb,16); //h
  74
  75         oggpackB_read(&opb,64);
  76
  77         oggpackB_read(&opb,32);
  78         oggpackB_read(&opb,32);
  79
  80         oggpackB_read(&opb,24);
  81         oggpackB_read(&opb,24);
  82
  83         oggpackB_read(&opb,38);
  84
  85         xr->gp_shift = oggpackB_read(&opb,5);
  86
  87         printf("gp_shift %d \n", xr->gp_shift);
  88
  89         return 0; //FIXME...
  90 }
  91
  92 int main (int argc, char **argv)
  93 {
  94         xiph_rtp_t xr;
  95
  96         char *buffer;
  97         int  bytes;
  98
  99         char *filename;
 100         FILE *file;
 101
 102         ogg_sync_init (&xr.oy);
 103
 104         int i = 0;
 105         int opt;
 106
 107         char *ip = "227.0.0.1";
 108         unsigned int port = 4044;
 109         unsigned int ttl  = 1;
 110         long timestamp = 0;
 111
 112         fprintf (stderr, "Theora RTP Server (draft-barbato-avt-rtp-theora-00)\n");
 113         memset (&xr,0,sizeof(xiph_rtp_t));
 114
 115 /*  Command-line args processing  */
 116
 117     if (argc < 2) {
 118         fprintf (stderr, "\n\tNo Theora file specified.\n");
 119         fprintf (stderr, "\tUsage: %s [-i ip address] [-p port] [-t ttl] filename\n\n",argv[0]);
 120         exit (1);
 121     }
 122
 123     while ((opt = getopt (argc, argv, "i:p:t:")) != -1) {
 124         switch (opt)
 125             {
 126
 127         /*  Set IP address  */
 128             case 'i':
 129                 ip = optarg;
 130                 break;
 131
 132         /*  Set port  */
 133             case 'p':
 134                     port = atoi (optarg);
 135                 break;
 136
 137         /*  Set TTL value  */
 138             case 't':
 139                 ttl = atoi (optarg);
 140                 break;
 141
 142         /* Unknown option  */
 143             case '?':
 144                     fprintf (stderr, "\n  Unknown option `-%c'.\n", optopt);
 145                     fprintf (stderr, "\tUsage: %s [-i ip address] [-p port] [-t ttl] filename\n\n",argv[0]);
 146                 return 1;
 147         }
 148     }
 149
 150 /*  Init RTP socket  */
 151
 152     if ( createsocket (&xr, ip, port, ttl)<0) return 1;
 153
 154 /*  Print network details  */
 155
 156     fprintf (stdout, "\n  Network setup\n");
 157     fprintf (stdout, "  IP Address= %s\n", ip);
 158     fprintf (stdout, "  Port = %d\n", port);
 159     fprintf (stdout, "  TTL = %d\n", ttl);
 160     fprintf (stdout, "\n");
 161
 162 /*  Open Theora file  */
 163
 164     filename = argv [argc - 1];
 165     file = fopen (filename, "rb");
 166
 167     if (file == NULL) {
 168         fprintf (stderr, "  Could not open file %s\n", filename);
 169         exit (1);
 170     }
 171
 172     fprintf (stdout, "  Theora setup\n");
 173     fprintf (stdout, "  Filename: %s\n", filename);
 174
 175     int eos = 0;
 176
 177     buffer = ogg_sync_buffer (&xr.oy, BUFFER_SIZE);
 178
 179     bytes = fread (buffer, 1, BUFFER_SIZE, file);
 180
 181     ogg_sync_wrote (&xr.oy, bytes);
 182
 183     if (ogg_sync_pageout (&xr.oy, &xr.og) != 1) {
 184         if (bytes < BUFFER_SIZE) {
 185             fprintf (stdout, "  Done\n");
 186             exit (0);
 187         }
 188
 189         fprintf (stderr, "\n  Input does not appear to be an Ogg bitstream.\n");
 190         exit (1);
 191     }
 192
 193     ogg_stream_init (&xr.os, ogg_page_serialno (&xr.og));
 194
 195     theora_info_init (&xr.ti);
 196     theora_comment_init (&xr.tc);
 197
 198     if (ogg_stream_pagein (&xr.os, &xr.og) < 0) {
 199         fprintf (stderr, "  Error reading first page of Ogg bitstream data.\n");
 200         exit (1);
 201     }
 202
 203     if (ogg_stream_packetout (&xr.os, &xr.op) != 1) {
 204         fprintf (stderr, "  Error reading initial header packet.\n");
 205         exit (1);
 206     }
 207
 208     if (theora_decode_header (&xr.ti, &xr.tc, &xr.op) < 0) {
 209         fprintf (stderr, "  This Ogg bitstream does not contain Theora video data.\n");
 210         exit (1);
 211     }
 212
 213     ogg_copy_packet(&(xr.header[i]), &xr.op); //FIXME change it?
 214
 215 /*===========================================================================*/
 216 /*  Process comment and codebook headers                                     */
 217 /*===========================================================================*/
 218
 219     while (i < 2) {
 220         while (i < 2) {
 221             int result = ogg_sync_pageout (&xr.oy, &xr.og);
 222
 223             if (result == 0) break; /* Need more data  */
 224
 225             if (result == 1) {
 226                 ogg_stream_pagein (&xr.os, &xr.og);
 227
 228                 while(i < 2) {
 229                     result = ogg_stream_packetout (&xr.os, &xr.op);
 230
 231                     if (result == 0) break;
 232
 233                     if (result < 0) {
 234                         fprintf (stderr, "  Corrupt secondary header.  Exiting.\n");
 235                         exit (1);
 236                     }
 237
 238                     theora_decode_header (&xr.ti, &xr.tc, &xr.op);
 239                     i++;
 240
 241                     ogg_copy_packet(&(xr.header[i]), &xr.op);
 242                 }
 243             }
 244         }
 245
 246         buffer = ogg_sync_buffer (&xr.oy, BUFFER_SIZE);
 247         bytes = fread (buffer, 1, BUFFER_SIZE, file);
 248
 249         if (bytes == 0 && i < 2) {
 250             fprintf (stderr, "||  End of file before finding all Vorbis headers!\n");
 251             exit (1);
 252         }
 253
 254         ogg_sync_wrote (&xr.oy, bytes);
 255     }
 256
 257     xr.bitfield.cbident = rand ();
 258
 259     theora_decode_init (&xr.td, &xr.ti);
 260
 261 /*===========================================================================*/
 262 /*  Print details                                                            */
 263 /*===========================================================================*/
 264     fprintf(stdout,"Theora %dx%d %.02f fps video\n"
 265                     "Encoded frame content is %dx%d with %dx%d offset\n",
 266             xr.ti.width,xr.ti.height,
 267             (double)xr.ti.fps_numerator/xr.ti.fps_denominator,
 268             xr.ti.frame_width, xr.ti.frame_height,
 269             xr.ti.offset_x, xr.ti.offset_y);
 270
 271     fprintf (stdout, "Decode setup ident is 0x%06x\n", xr.bitfield.cbident);
 272     fprintf (stdout, "\n");
 273     fprintf (stdout, "Processing\n");
 274
 275 /*===========================================================================*/
 276 /*  Send the three headers inline                                            */
 277 /*===========================================================================*/
 278 {
 279 int conf_bytes = xr.header[0].bytes + xr.header[2].bytes;
 280 unsigned char *conf_packet = malloc(conf_bytes);
 281
 282     memcpy (conf_packet, xr.header[0].packet, xr.header[0].bytes);
 283     memcpy (conf_packet + xr.header[0].bytes,
 284                     xr.header[2].packet,
 285                     xr.header[2].bytes);
 286     creatertp(&xr, conf_packet, conf_bytes, 0, 0, 1, 0);
 287
 288     free(conf_packet);
 289 }
 290
 291 cfg_parse(&xr);
 292
 293 /*  Read raw data and send RTP packet  */
 294
 295     while (!eos) {
 296         while (!eos) {
 297             int result = ogg_sync_pageout (&xr.oy, &xr.og);
 298
 299             if (result == 0) break; /* need more data  */
 300
 301             if (result < 0) {
 302                 fprintf (stderr, "\n  Corrupt or missing data in bitstream; continuing....\n  ");
 303             } else {
 304                 ogg_stream_pagein (&xr.os, &xr.og);
 305
 306                 while (1) {
 307                     result = ogg_stream_packetout (&xr.os, &xr.op);
 308
 309                     if (result == 0) break; /* need more data  */
 310
 311                     if (result < 0) {
 312                         /* no reason to complain; already complained above  */
 313                     } else {
 314                         theora_decode_packetin(&xr.td,&xr.op);
 315                         if (xr.op.granulepos != -1)
 316                                 timestamp =  (xr.op.granulepos>>xr.gp_shift)+
 317                                 (xr.op.granulepos & ((1<<xr.gp_shift)-1));
 318                         else
 319                                 timestamp++ ;
 320 #ifdef DEBUG
 321                         {
 322                         oggpack_buffer opb;
 323                         int type;
 324                           oggpackB_readinit(&opb,xr.op.packet,xr.op.bytes);
 325                           oggpack_read(&opb,1); //video marker
 326
 327                           type = oggpackB_read(&opb,1);
 328                           printf("  type %d stamp %ld", type, timestamp);
 329                         }
 330                         printf("  bytes %ld bos %ld eos %ld gp %lld pno %lld\n", xr.op.bytes, xr.op.b_o_s, xr.op.e_o_s, xr.op.granulepos, xr.op.packetno);
 331 #endif
 332                         //FIXME ugly as hell and probably wrong!!!!
 333                         creatertp ( &xr, xr.op.packet, xr.op.bytes,
 334                                     timestamp*90000/((double)xr.ti.fps_numerator/xr.ti.fps_denominator), timestamp, 0, xr.op.e_o_s );
 335                     }
 336                 }
 337
 338                 if (ogg_page_eos (&xr.og)) eos = 1;
 339             }
 340         }
 341
 342         if (!eos) {
 343             buffer = ogg_sync_buffer (&xr.oy, BUFFER_SIZE);
 344             bytes = fread (buffer, 1, BUFFER_SIZE, file);
 345
 346             ogg_sync_wrote (&xr.oy, bytes);
 347
 348             if (bytes == 0) eos = 1;
 349         }
 350     }
 351 //FIXME make it a cleanup function!
 352     ogg_packet_clear (&(xr.header[0]));
 353     ogg_packet_clear (&(xr.header[1]));
 354     ogg_packet_clear (&(xr.header[2]));
 355
 356     ogg_stream_clear (&xr.os);
 357
 358     theora_clear(&xr.td);
 359     theora_comment_clear(&xr.tc);
 360     theora_info_clear(&xr.ti);
 361
 362     ogg_sync_clear (&xr.oy);
 363     fclose (file);
 364
 365     fprintf (stdout, "Done.\n");
 366     return (0);
 367 }