vobsub.c

   1 /*
   2  * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>,
   3  * with kind permission from Gabest <gabest@freemail.hu>
   4  */
   5 #include <ctype.h>
   6 #include <errno.h>
   7 #include <limits.h>
   8 #include <stddef.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12 #include <fcntl.h>
  13 #include <unistd.h>
  14 #include <sys/stat.h>
  15 #include <sys/types.h>
  16
  17 #include "config.h"
  18 #include "version.h"
  19
  20 #include "vobsub.h"
  21 #include "spudec.h"
  22 #include "mp_msg.h"
  23 #include "unrar_exec.h"
  24 #include "libavutil/common.h"
  25
  26 extern int vobsub_id;
  27 // Record the original -vobsubid set by commandline, since vobsub_id will be
  28 // overridden if slang match any of vobsub streams.
  29 static int vobsubid = -2;
  30
  31 /**********************************************************************
  32  * RAR stream handling
  33  * The RAR file must have the same basename as the file to open
  34  **********************************************************************/
  35 #ifdef CONFIG_UNRAR_EXEC
  36 typedef struct {
  37     FILE *file;
  38     unsigned char *data;
  39     unsigned long size;
  40     unsigned long pos;
  41 } rar_stream_t;
  42 static rar_stream_t *
  43 rar_open(const char *const filename, const char *const mode)
  44 {
  45     rar_stream_t *stream;
  46     /* unrar_exec can only read */
  47     if (strcmp("r", mode) && strcmp("rb", mode)) {
  48         errno = EINVAL;
  49         return NULL;
  50     }
  51     stream = malloc(sizeof(rar_stream_t));
  52     if (stream == NULL)
  53         return NULL;
  54     /* first try normal access */
  55     stream->file = fopen(filename, mode);
  56     if (stream->file == NULL) {
  57         char *rar_filename;
  58         const char *p;
  59         int rc;
  60         /* Guess the RAR archive filename */
  61         rar_filename = NULL;
  62         p = strrchr(filename, '.');
  63         if (p) {
  64             ptrdiff_t l = p - filename;
  65             rar_filename = malloc(l + 5);
  66             if (rar_filename == NULL) {
  67                 free(stream);
  68                 return NULL;
  69             }
  70             strncpy(rar_filename, filename, l);
  71             strcpy(rar_filename + l, ".rar");
  72         }
  73         else {
  74             rar_filename = malloc(strlen(filename) + 5);
  75             if (rar_filename == NULL) {
  76                 free(stream);
  77                 return NULL;
  78             }
  79             strcpy(rar_filename, filename);
  80             strcat(rar_filename, ".rar");
  81         }
  82         /* get rid of the path if there is any */
  83         if ((p = strrchr(filename, '/')) == NULL) {
  84                 p = filename;
  85         } else {
  86                 p++;
  87         }
  88         rc = unrar_exec_get(&stream->data, &stream->size, p, rar_filename);
  89         if (!rc) {
  90             /* There is no matching filename in the archive. However, sometimes
  91              * the files we are looking for have been given arbitrary names in the archive.
  92              * Let's look for a file with an exact match in the extension only. */
  93             int i, num_files, name_len;
  94             ArchiveList_struct *list, *lp;
  95             num_files = unrar_exec_list(rar_filename, &list);
  96             if (num_files > 0) {
  97                 char *demanded_ext;
  98                 demanded_ext = strrchr (p, '.');
  99                 if (demanded_ext) {
 100                     int demanded_ext_len = strlen (demanded_ext);
 101                     for (i=0, lp=list; i<num_files; i++, lp=lp->next) {
 102                         name_len = strlen (lp->item.Name);
 103                         if (name_len >= demanded_ext_len && !strcasecmp (lp->item.Name + name_len - demanded_ext_len, demanded_ext)) {
 104                             rc = unrar_exec_get(&stream->data, &stream->size,
 105                                                 lp->item.Name, rar_filename);
 106                             if (rc) break;
 107                         }
 108                     }
 109                 }
 110                 unrar_exec_freelist(list);
 111             }
 112             if (!rc) {
 113                 free(rar_filename);
 114                 free(stream);
 115                 return NULL;
 116             }
 117         }
 118
 119         free(rar_filename);
 120         stream->pos = 0;
 121     }
 122     return stream;
 123 }
 124
 125 static int
 126 rar_close(rar_stream_t *stream)
 127 {
 128     if (stream->file)
 129         return fclose(stream->file);
 130     free(stream->data);
 131     return 0;
 132 }
 133
 134 static int
 135 rar_eof(rar_stream_t *stream)
 136 {
 137     if (stream->file)
 138         return feof(stream->file);
 139     return stream->pos >= stream->size;
 140 }
 141
 142 static long
 143 rar_tell(rar_stream_t *stream)
 144 {
 145     if (stream->file)
 146         return ftell(stream->file);
 147     return stream->pos;
 148 }
 149
 150 static int
 151 rar_seek(rar_stream_t *stream, long offset, int whence)
 152 {
 153     if (stream->file)
 154         return fseek(stream->file, offset, whence);
 155     switch (whence) {
 156     case SEEK_SET:
 157         if (offset < 0) {
 158             errno = EINVAL;
 159             return -1;
 160         }
 161         stream->pos = offset;
 162         break;
 163     case SEEK_CUR:
 164         if (offset < 0 && stream->pos < (unsigned long) -offset) {
 165             errno = EINVAL;
 166             return -1;
 167         }
 168         stream->pos += offset;
 169         break;
 170     case SEEK_END:
 171         if (offset < 0 && stream->size < (unsigned long) -offset) {
 172             errno = EINVAL;
 173             return -1;
 174         }
 175         stream->pos = stream->size + offset;
 176         break;
 177     default:
 178         errno = EINVAL;
 179         return -1;
 180     }
 181     return 0;
 182 }
 183
 184 static int
 185 rar_getc(rar_stream_t *stream)
 186 {
 187     if (stream->file)
 188         return getc(stream->file);
 189     if (rar_eof(stream))
 190         return EOF;
 191     return stream->data[stream->pos++];
 192 }
 193
 194 static size_t
 195 rar_read(void *ptr, size_t size, size_t nmemb, rar_stream_t *stream)
 196 {
 197     size_t res;
 198     unsigned long remain;
 199     if (stream->file)
 200         return fread(ptr, size, nmemb, stream->file);
 201     if (rar_eof(stream))
 202         return 0;
 203     res = size * nmemb;
 204     remain = stream->size - stream->pos;
 205     if (res > remain)
 206         res = remain / size * size;
 207     memcpy(ptr, stream->data + stream->pos, res);
 208     stream->pos += res;
 209     res /= size;
 210     return res;
 211 }
 212
 213 #else
 214 typedef FILE rar_stream_t;
 215 #define rar_open        fopen
 216 #define rar_close       fclose
 217 #define rar_eof         feof
 218 #define rar_tell        ftell
 219 #define rar_seek        fseek
 220 #define rar_getc        getc
 221 #define rar_read        fread
 222 #endif
 223
 224 /**********************************************************************/
 225
 226 static ssize_t
 227 vobsub_getline(char **lineptr, size_t *n, rar_stream_t *stream)
 228 {
 229     size_t res = 0;
 230     int c;
 231     if (*lineptr == NULL) {
 232         *lineptr = malloc(4096);
 233         if (*lineptr)
 234             *n = 4096;
 235     }
 236     else if (*n == 0) {
 237         char *tmp = realloc(*lineptr, 4096);
 238         if (tmp) {
 239             *lineptr = tmp;
 240             *n = 4096;
 241         }
 242     }
 243     if (*lineptr == NULL || *n == 0)
 244         return -1;
 245
 246     for (c = rar_getc(stream); c != EOF; c = rar_getc(stream)) {
 247         if (res + 1 >= *n) {
 248             char *tmp = realloc(*lineptr, *n * 2);
 249             if (tmp == NULL)
 250                 return -1;
 251             *lineptr = tmp;
 252             *n *= 2;
 253         }
 254         (*lineptr)[res++] = c;
 255         if (c == '\n') {
 256             (*lineptr)[res] = 0;
 257             return res;
 258         }
 259     }
 260     if (res == 0)
 261         return -1;
 262     (*lineptr)[res] = 0;
 263     return res;
 264 }
 265
 266 /**********************************************************************
 267  * MPEG parsing
 268  **********************************************************************/
 269
 270 typedef struct {
 271     rar_stream_t *stream;
 272     unsigned int pts;
 273     int aid;
 274     unsigned char *packet;
 275     unsigned int packet_reserve;
 276     unsigned int packet_size;
 277 } mpeg_t;
 278
 279 static mpeg_t *
 280 mpeg_open(const char *filename)
 281 {
 282     mpeg_t *res = malloc(sizeof(mpeg_t));
 283     int err = res == NULL;
 284     if (!err) {
 285         res->pts = 0;
 286         res->aid = -1;
 287         res->packet = NULL;
 288         res->packet_size = 0;
 289         res->packet_reserve = 0;
 290         res->stream = rar_open(filename, "rb");
 291         err = res->stream == NULL;
 292         if (err)
 293             perror("fopen Vobsub file failed");
 294         if (err)
 295             free(res);
 296     }
 297     return err ? NULL : res;
 298 }
 299
 300 static void
 301 mpeg_free(mpeg_t *mpeg)
 302 {
 303     if (mpeg->packet)
 304         free(mpeg->packet);
 305     if (mpeg->stream)
 306         rar_close(mpeg->stream);
 307     free(mpeg);
 308 }
 309
 310 static int
 311 mpeg_eof(mpeg_t *mpeg)
 312 {
 313     return rar_eof(mpeg->stream);
 314 }
 315
 316 static off_t
 317 mpeg_tell(mpeg_t *mpeg)
 318 {
 319     return rar_tell(mpeg->stream);
 320 }
 321
 322 static int
 323 mpeg_run(mpeg_t *mpeg)
 324 {
 325     unsigned int len, idx, version;
 326     int c;
 327     /* Goto start of a packet, it starts with 0x000001?? */
 328     const unsigned char wanted[] = { 0, 0, 1 };
 329     unsigned char buf[5];
 330
 331     mpeg->aid = -1;
 332     mpeg->packet_size = 0;
 333     if (rar_read(buf, 4, 1, mpeg->stream) != 1)
 334         return -1;
 335     while (memcmp(buf, wanted, sizeof(wanted)) != 0) {
 336         c = rar_getc(mpeg->stream);
 337         if (c < 0)
 338             return -1;
 339         memmove(buf, buf + 1, 3);
 340         buf[3] = c;
 341     }
 342     switch (buf[3]) {
 343     case 0xb9:                  /* System End Code */
 344         break;
 345     case 0xba:                  /* Packet start code */
 346         c = rar_getc(mpeg->stream);
 347         if (c < 0)
 348             return -1;
 349         if ((c & 0xc0) == 0x40)
 350             version = 4;
 351         else if ((c & 0xf0) == 0x20)
 352             version = 2;
 353         else {
 354             mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Unsupported MPEG version: 0x%02x\n", c);
 355             return -1;
 356         }
 357         if (version == 4) {
 358             if (rar_seek(mpeg->stream, 9, SEEK_CUR))
 359                 return -1;
 360         }
 361         else if (version == 2) {
 362             if (rar_seek(mpeg->stream, 7, SEEK_CUR))
 363                 return -1;
 364         }
 365         else
 366             abort();
 367         break;
 368     case 0xbd:                  /* packet */
 369         if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 370             return -1;
 371         len = buf[0] << 8 | buf[1];
 372         idx = mpeg_tell(mpeg);
 373         c = rar_getc(mpeg->stream);
 374         if (c < 0)
 375             return -1;
 376         if ((c & 0xC0) == 0x40) { /* skip STD scale & size */
 377             if (rar_getc(mpeg->stream) < 0)
 378                 return -1;
 379             c = rar_getc(mpeg->stream);
 380             if (c < 0)
 381                 return -1;
 382         }
 383         if ((c & 0xf0) == 0x20) { /* System-1 stream timestamp */
 384             /* Do we need this? */
 385             abort();
 386         }
 387         else if ((c & 0xf0) == 0x30) {
 388             /* Do we need this? */
 389             abort();
 390         }
 391         else if ((c & 0xc0) == 0x80) { /* System-2 (.VOB) stream */
 392             unsigned int pts_flags, hdrlen, dataidx;
 393             c = rar_getc(mpeg->stream);
 394             if (c < 0)
 395                 return -1;
 396             pts_flags = c;
 397             c = rar_getc(mpeg->stream);
 398             if (c < 0)
 399                 return -1;
 400             hdrlen = c;
 401             dataidx = mpeg_tell(mpeg) + hdrlen;
 402             if (dataidx > idx + len) {
 403                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "Invalid header length: %d (total length: %d, idx: %d, dataidx: %d)\n",
 404                         hdrlen, len, idx, dataidx);
 405                 return -1;
 406             }
 407             if ((pts_flags & 0xc0) == 0x80) {
 408                 if (rar_read(buf, 5, 1, mpeg->stream) != 1)
 409                     return -1;
 410                 if (!(((buf[0] & 0xf0) == 0x20) && (buf[0] & 1) && (buf[2] & 1) &&  (buf[4] & 1))) {
 411                     mp_msg(MSGT_VOBSUB,MSGL_ERR, "vobsub PTS error: 0x%02x %02x%02x %02x%02x \n",
 412                             buf[0], buf[1], buf[2], buf[3], buf[4]);
 413                     mpeg->pts = 0;
 414                 }
 415                 else
 416                     mpeg->pts = ((buf[0] & 0x0e) << 29 | buf[1] << 22 | (buf[2] & 0xfe) << 14
 417                         | buf[3] << 7 | (buf[4] >> 1));
 418             }
 419             else /* if ((pts_flags & 0xc0) == 0xc0) */ {
 420                 /* what's this? */
 421                 /* abort(); */
 422             }
 423             rar_seek(mpeg->stream, dataidx, SEEK_SET);
 424             mpeg->aid = rar_getc(mpeg->stream);
 425             if (mpeg->aid < 0) {
 426                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "Bogus aid %d\n", mpeg->aid);
 427                 return -1;
 428             }
 429             mpeg->packet_size = len - ((unsigned int) mpeg_tell(mpeg) - idx);
 430             if (mpeg->packet_reserve < mpeg->packet_size) {
 431                 if (mpeg->packet)
 432                     free(mpeg->packet);
 433                 mpeg->packet = malloc(mpeg->packet_size);
 434                 if (mpeg->packet)
 435                     mpeg->packet_reserve = mpeg->packet_size;
 436             }
 437             if (mpeg->packet == NULL) {
 438                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure");
 439                 mpeg->packet_reserve = 0;
 440                 mpeg->packet_size = 0;
 441                 return -1;
 442             }
 443             if (rar_read(mpeg->packet, mpeg->packet_size, 1, mpeg->stream) != 1) {
 444                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"fread failure");
 445                 mpeg->packet_size = 0;
 446                 return -1;
 447             }
 448             idx = len;
 449         }
 450         break;
 451     case 0xbe:                  /* Padding */
 452         if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 453             return -1;
 454         len = buf[0] << 8 | buf[1];
 455         if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR))
 456             return -1;
 457         break;
 458     default:
 459         if (0xc0 <= buf[3] && buf[3] < 0xf0) {
 460             /* MPEG audio or video */
 461             if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 462                 return -1;
 463             len = buf[0] << 8 | buf[1];
 464             if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR))
 465                 return -1;
 466
 467         }
 468         else {
 469             mp_msg(MSGT_VOBSUB,MSGL_ERR,"unknown header 0x%02X%02X%02X%02X\n",
 470                     buf[0], buf[1], buf[2], buf[3]);
 471             return -1;
 472         }
 473     }
 474     return 0;
 475 }
 476
 477 /**********************************************************************
 478  * Packet queue
 479  **********************************************************************/
 480
 481 typedef struct {
 482     unsigned int pts100;
 483     off_t filepos;
 484     unsigned int size;
 485     unsigned char *data;
 486 } packet_t;
 487
 488 typedef struct {
 489     char *id;
 490     packet_t *packets;
 491     unsigned int packets_reserve;
 492     unsigned int packets_size;
 493     unsigned int current_index;
 494 } packet_queue_t;
 495
 496 static void
 497 packet_construct(packet_t *pkt)
 498 {
 499     pkt->pts100 = 0;
 500     pkt->filepos = 0;
 501     pkt->size = 0;
 502     pkt->data = NULL;
 503 }
 504
 505 static void
 506 packet_destroy(packet_t *pkt)
 507 {
 508     if (pkt->data)
 509         free(pkt->data);
 510 }
 511
 512 static void
 513 packet_queue_construct(packet_queue_t *queue)
 514 {
 515     queue->id = NULL;
 516     queue->packets = NULL;
 517     queue->packets_reserve = 0;
 518     queue->packets_size = 0;
 519     queue->current_index = 0;
 520 }
 521
 522 static void
 523 packet_queue_destroy(packet_queue_t *queue)
 524 {
 525     if (queue->packets) {
 526         while (queue->packets_size--)
 527             packet_destroy(queue->packets + queue->packets_size);
 528         free(queue->packets);
 529     }
 530     return;
 531 }
 532
 533 /* Make sure there is enough room for needed_size packets in the
 534    packet queue. */
 535 static int
 536 packet_queue_ensure(packet_queue_t *queue, unsigned int needed_size)
 537 {
 538     if (queue->packets_reserve < needed_size) {
 539         if (queue->packets) {
 540             packet_t *tmp = realloc(queue->packets, 2 * queue->packets_reserve * sizeof(packet_t));
 541             if (tmp == NULL) {
 542                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"realloc failure");
 543                 return -1;
 544             }
 545             queue->packets = tmp;
 546             queue->packets_reserve *= 2;
 547         }
 548         else {
 549             queue->packets = malloc(sizeof(packet_t));
 550             if (queue->packets == NULL) {
 551                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure");
 552                 return -1;
 553             }
 554             queue->packets_reserve = 1;
 555         }
 556     }
 557     return 0;
 558 }
 559
 560 /* add one more packet */
 561 static int
 562 packet_queue_grow(packet_queue_t *queue)
 563 {
 564     if (packet_queue_ensure(queue, queue->packets_size + 1) < 0)
 565         return -1;
 566     packet_construct(queue->packets + queue->packets_size);
 567     ++queue->packets_size;
 568     return 0;
 569 }
 570
 571 /* insert a new packet, duplicating pts from the current one */
 572 static int
 573 packet_queue_insert(packet_queue_t *queue)
 574 {
 575     packet_t *pkts;
 576     if (packet_queue_ensure(queue, queue->packets_size + 1) < 0)
 577         return -1;
 578     /* XXX packet_size does not reflect the real thing here, it will be updated a bit later */
 579     memmove(queue->packets + queue->current_index + 2,
 580             queue->packets + queue->current_index + 1,
 581             sizeof(packet_t) * (queue->packets_size - queue->current_index - 1));
 582     pkts = queue->packets + queue->current_index;
 583     ++queue->packets_size;
 584     ++queue->current_index;
 585     packet_construct(pkts + 1);
 586     pkts[1].pts100 = pkts[0].pts100;
 587     pkts[1].filepos = pkts[0].filepos;
 588     return 0;
 589 }
 590
 591 /**********************************************************************
 592  * Vobsub
 593  **********************************************************************/
 594
 595 typedef struct {
 596     unsigned int palette[16];
 597     int delay;
 598     unsigned int have_palette;
 599     unsigned int orig_frame_width, orig_frame_height;
 600     unsigned int origin_x, origin_y;
 601     /* index */
 602     packet_queue_t *spu_streams;
 603     unsigned int spu_streams_size;
 604     unsigned int spu_streams_current;
 605     unsigned int spu_valid_streams_size;
 606 } vobsub_t;
 607
 608 /* Make sure that the spu stream idx exists. */
 609 static int
 610 vobsub_ensure_spu_stream(vobsub_t *vob, unsigned int index)
 611 {
 612     if (index >= vob->spu_streams_size) {
 613         /* This is a new stream */
 614         if (vob->spu_streams) {
 615             packet_queue_t *tmp = realloc(vob->spu_streams, (index + 1) * sizeof(packet_queue_t));
 616             if (tmp == NULL) {
 617                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: realloc failure");
 618                 return -1;
 619             }
 620             vob->spu_streams = tmp;
 621         }
 622         else {
 623             vob->spu_streams = malloc((index + 1) * sizeof(packet_queue_t));
 624             if (vob->spu_streams == NULL) {
 625                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: malloc failure");
 626                 return -1;
 627             }
 628         }
 629         while (vob->spu_streams_size <= index) {
 630             packet_queue_construct(vob->spu_streams + vob->spu_streams_size);
 631             ++vob->spu_streams_size;
 632         }
 633     }
 634     return 0;
 635 }
 636
 637 static int
 638 vobsub_add_id(vobsub_t *vob, const char *id, size_t idlen, const unsigned int index)
 639 {
 640     if (vobsub_ensure_spu_stream(vob, index) < 0)
 641         return -1;
 642     if (id && idlen) {
 643         if (vob->spu_streams[index].id)
 644             free(vob->spu_streams[index].id);
 645         vob->spu_streams[index].id = malloc(idlen + 1);
 646         if (vob->spu_streams[index].id == NULL) {
 647             mp_msg(MSGT_VOBSUB,MSGL_FATAL,"vobsub_add_id: malloc failure");
 648             return -1;
 649         }
 650         vob->spu_streams[index].id[idlen] = 0;
 651         memcpy(vob->spu_streams[index].id, id, idlen);
 652     }
 653     vob->spu_streams_current = index;
 654     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VOBSUB_ID=%d\n", index);
 655     if (id && idlen)
 656         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VSID_%d_LANG=%s\n", index, vob->spu_streams[index].id);
 657     mp_msg(MSGT_VOBSUB,MSGL_V,"[vobsub] subtitle (vobsubid): %d language %s\n",
 658                 index, vob->spu_streams[index].id);
 659     return 0;
 660 }
 661
 662 static int
 663 vobsub_add_timestamp(vobsub_t *vob, off_t filepos, int ms)
 664 {
 665     packet_queue_t *queue;
 666     packet_t *pkt;
 667     if (vob->spu_streams == 0) {
 668         mp_msg(MSGT_VOBSUB,MSGL_WARN,"[vobsub] warning, binning some index entries.  Check your index file\n");
 669         return -1;
 670     }
 671     queue = vob->spu_streams + vob->spu_streams_current;
 672     if (packet_queue_grow(queue) >= 0) {
 673         pkt = queue->packets + (queue->packets_size - 1);
 674         pkt->filepos = filepos;
 675         pkt->pts100 = ms < 0 ? UINT_MAX : (unsigned int)ms * 90;
 676         return 0;
 677     }
 678     return -1;
 679 }
 680
 681 static int
 682 vobsub_parse_id(vobsub_t *vob, const char *line)
 683 {
 684     // id: xx, index: n
 685     size_t idlen;
 686     const char *p, *q;
 687     p  = line;
 688     while (isspace(*p))
 689         ++p;
 690     q = p;
 691     while (isalpha(*q))
 692         ++q;
 693     idlen = q - p;
 694     if (idlen == 0)
 695         return -1;
 696     ++q;
 697     while (isspace(*q))
 698         ++q;
 699     if (strncmp("index:", q, 6))
 700         return -1;
 701     q += 6;
 702     while (isspace(*q))
 703         ++q;
 704     if (!isdigit(*q))
 705         return -1;
 706     return vobsub_add_id(vob, p, idlen, atoi(q));
 707 }
 708
 709 static int
 710 vobsub_parse_timestamp(vobsub_t *vob, const char *line)
 711 {
 712     // timestamp: HH:MM:SS.mmm, filepos: 0nnnnnnnnn
 713     const char *p;
 714     int h, m, s, ms;
 715     off_t filepos;
 716     while (isspace(*line))
 717         ++line;
 718     p = line;
 719     while (isdigit(*p))
 720         ++p;
 721     if (p - line != 2)
 722         return -1;
 723     h = atoi(line);
 724     if (*p != ':')
 725         return -1;
 726     line = ++p;
 727     while (isdigit(*p))
 728         ++p;
 729     if (p - line != 2)
 730         return -1;
 731     m = atoi(line);
 732     if (*p != ':')
 733         return -1;
 734     line = ++p;
 735     while (isdigit(*p))
 736         ++p;
 737     if (p - line != 2)
 738         return -1;
 739     s = atoi(line);
 740     if (*p != ':')
 741         return -1;
 742     line = ++p;
 743     while (isdigit(*p))
 744         ++p;
 745     if (p - line != 3)
 746         return -1;
 747     ms = atoi(line);
 748     if (*p != ',')
 749         return -1;
 750     line = p + 1;
 751     while (isspace(*line))
 752         ++line;
 753     if (strncmp("filepos:", line, 8))
 754         return -1;
 755     line += 8;
 756     while (isspace(*line))
 757         ++line;
 758     if (! isxdigit(*line))
 759         return -1;
 760     filepos = strtol(line, NULL, 16);
 761     return vobsub_add_timestamp(vob, filepos, vob->delay + ms + 1000 * (s + 60 * (m + 60 * h)));
 762 }
 763
 764 static int
 765 vobsub_parse_origin(vobsub_t *vob, const char *line)
 766 {
 767     // org: X,Y
 768     char *p;
 769     while (isspace(*line))
 770         ++line;
 771     if (!isdigit(*line))
 772         return -1;
 773     vob->origin_x = strtoul(line, &p, 10);
 774     if (*p != ',')
 775         return -1;
 776     ++p;
 777     vob->origin_y = strtoul(p, NULL, 10);
 778     return 0;
 779 }
 780
 781 unsigned int vobsub_palette_to_yuv(unsigned int pal)
 782 {
 783     int r, g, b, y, u, v;
 784     // Palette in idx file is not rgb value, it was calculated by wrong formula.
 785     // Here's reversed formula of the one used to generate palette in idx file.
 786     r = pal >> 16 & 0xff;
 787     g = pal >> 8 & 0xff;
 788     b = pal & 0xff;
 789     y = av_clip_uint8( 0.1494  * r + 0.6061 * g + 0.2445 * b);
 790     u = av_clip_uint8( 0.6066  * r - 0.4322 * g - 0.1744 * b + 128);
 791     v = av_clip_uint8(-0.08435 * r - 0.3422 * g + 0.4266 * b + 128);
 792     y = y * 219 / 255 + 16;
 793     return y << 16 | u << 8 | v;
 794 }
 795
 796 unsigned int vobsub_rgb_to_yuv(unsigned int rgb)
 797 {
 798     int r, g, b, y, u, v;
 799     r = rgb >> 16 & 0xff;
 800     g = rgb >> 8 & 0xff;
 801     b = rgb & 0xff;
 802     y = ( 0.299   * r + 0.587   * g + 0.114   * b) * 219 / 255 + 16.5;
 803     u = (-0.16874 * r - 0.33126 * g + 0.5     * b) * 224 / 255 + 128.5;
 804     v = ( 0.5     * r - 0.41869 * g - 0.08131 * b) * 224 / 255 + 128.5;
 805     return y << 16 | u << 8 | v;
 806 }
 807
 808 static int
 809 vobsub_parse_delay(vobsub_t *vob, const char *line)
 810 {
 811     int h, m, s, ms;
 812     int forward = 1;
 813     if (*(line + 7) == '+'){
 814         forward = 1;
 815         line++;
 816     }
 817     else if (*(line + 7) == '-'){
 818         forward = -1;
 819         line++;
 820     }
 821     mp_msg(MSGT_SPUDEC,MSGL_V, "forward=%d", forward);
 822     h = atoi(line + 7);
 823     mp_msg(MSGT_VOBSUB,MSGL_V, "h=%d," ,h);
 824     m = atoi(line + 10);
 825     mp_msg(MSGT_VOBSUB,MSGL_V, "m=%d,", m);
 826     s = atoi(line + 13);
 827     mp_msg(MSGT_VOBSUB,MSGL_V, "s=%d,", s);
 828     ms = atoi(line + 16);
 829     mp_msg(MSGT_VOBSUB,MSGL_V, "ms=%d", ms);
 830     vob->delay = (ms + 1000 * (s + 60 * (m + 60 * h))) * forward;
 831     return 0;
 832 }
 833
 834 static int
 835 vobsub_set_lang(const char *line)
 836 {
 837     if (vobsub_id == -1)
 838         vobsub_id = atoi(line + 8);
 839     return 0;
 840 }
 841
 842 static int
 843 vobsub_parse_one_line(vobsub_t *vob, rar_stream_t *fd,
 844                       unsigned char **extradata, unsigned int *extradata_len)
 845 {
 846     ssize_t line_size;
 847     int res = -1;
 848         size_t line_reserve = 0;
 849         char *line = NULL;
 850     do {
 851         line_size = vobsub_getline(&line, &line_reserve, fd);
 852         if (line_size < 0 || line_size > 1000000 ||
 853             *extradata_len+line_size > 10000000) {
 854             break;
 855         }
 856
 857         *extradata = realloc(*extradata, *extradata_len+line_size+1);
 858         memcpy(*extradata+*extradata_len, line, line_size);
 859         *extradata_len += line_size;
 860         (*extradata)[*extradata_len] = 0;
 861
 862         if (*line == 0 || *line == '\r' || *line == '\n' || *line == '#')
 863             continue;
 864         else if (strncmp("langidx:", line, 8) == 0)
 865             res = vobsub_set_lang(line);
 866         else if (strncmp("delay:", line, 6) == 0)
 867             res = vobsub_parse_delay(vob, line);
 868         else if (strncmp("id:", line, 3) == 0)
 869             res = vobsub_parse_id(vob, line + 3);
 870         else if (strncmp("org:", line, 4) == 0)
 871             res = vobsub_parse_origin(vob, line + 4);
 872         else if (strncmp("timestamp:", line, 10) == 0)
 873             res = vobsub_parse_timestamp(vob, line + 10);
 874         else {
 875             mp_msg(MSGT_VOBSUB,MSGL_V, "vobsub: ignoring %s", line);
 876             continue;
 877         }
 878         if (res < 0)
 879             mp_msg(MSGT_VOBSUB,MSGL_ERR,  "ERROR in %s", line);
 880         break;
 881     } while (1);
 882     if (line)
 883       free(line);
 884     return res;
 885 }
 886
 887 int
 888 vobsub_parse_ifo(void* this, const char *const name, unsigned int *palette, unsigned int *width, unsigned int *height, int force,
 889                  int sid, char *langid)
 890 {
 891     vobsub_t *vob = (vobsub_t*)this;
 892     int res = -1;
 893     rar_stream_t *fd = rar_open(name, "rb");
 894     if (fd == NULL) {
 895         if (force)
 896             mp_msg(MSGT_VOBSUB,MSGL_WARN, "VobSub: Can't open IFO file\n");
 897     } else {
 898         // parse IFO header
 899         unsigned char block[0x800];
 900         const char *const ifo_magic = "DVDVIDEO-VTS";
 901         if (rar_read(block, sizeof(block), 1, fd) != 1) {
 902             if (force)
 903                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO header\n");
 904         } else if (memcmp(block, ifo_magic, strlen(ifo_magic) + 1))
 905             mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Bad magic in IFO header\n");
 906         else {
 907             unsigned long pgci_sector = block[0xcc] << 24 | block[0xcd] << 16
 908                 | block[0xce] << 8 | block[0xcf];
 909             int standard = (block[0x200] & 0x30) >> 4;
 910             int resolution = (block[0x201] & 0x0c) >> 2;
 911             *height = standard ? 576 : 480;
 912             *width = 0;
 913             switch (resolution) {
 914             case 0x0:
 915                 *width = 720;
 916                 break;
 917             case 0x1:
 918                 *width = 704;
 919                 break;
 920             case 0x2:
 921                 *width = 352;
 922                 break;
 923             case 0x3:
 924                 *width = 352;
 925                 *height /= 2;
 926                 break;
 927             default:
 928                 mp_msg(MSGT_VOBSUB,MSGL_WARN,"Vobsub: Unknown resolution %d \n", resolution);
 929             }
 930             if (langid && 0 <= sid && sid < 32) {
 931                 unsigned char *tmp = block + 0x256 + sid * 6 + 2;
 932                 langid[0] = tmp[0];
 933                 langid[1] = tmp[1];
 934                 langid[2] = 0;
 935             }
 936             if (rar_seek(fd, pgci_sector * sizeof(block), SEEK_SET)
 937                 || rar_read(block, sizeof(block), 1, fd) != 1)
 938                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO PGCI\n");
 939             else {
 940                 unsigned long idx;
 941                 unsigned long pgc_offset = block[0xc] << 24 | block[0xd] << 16
 942                     | block[0xe] << 8 | block[0xf];
 943                 for (idx = 0; idx < 16; ++idx) {
 944                     unsigned char *p = block + pgc_offset + 0xa4 + 4 * idx;
 945                     palette[idx] = p[0] << 24 | p[1] << 16 | p[2] << 8 | p[3];
 946                 }
 947                 if(vob)
 948                   vob->have_palette = 1;
 949                 res = 0;
 950             }
 951         }
 952         rar_close(fd);
 953     }
 954     return res;
 955 }
 956
 957 void *
 958 vobsub_open(const char *const name,const char *const ifo,const int force,void** spu)
 959 {
 960     unsigned char *extradata = NULL;
 961     unsigned int extradata_len = 0;
 962     vobsub_t *vob = calloc(1, sizeof(vobsub_t));
 963     if(spu)
 964       *spu = NULL;
 965     if (vobsubid == -2)
 966       vobsubid = vobsub_id;
 967     if (vob) {
 968         char *buf;
 969         buf = malloc(strlen(name) + 5);
 970         if (buf) {
 971             rar_stream_t *fd;
 972             mpeg_t *mpg;
 973             /* read in the info file */
 974             if(!ifo) {
 975               strcpy(buf, name);
 976               strcat(buf, ".ifo");
 977               vobsub_parse_ifo(vob,buf, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL);
 978             } else
 979               vobsub_parse_ifo(vob,ifo, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL);
 980             /* read in the index */
 981             strcpy(buf, name);
 982             strcat(buf, ".idx");
 983             fd = rar_open(buf, "rb");
 984             if (fd == NULL) {
 985                 if(force)
 986                   mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open IDX file\n");
 987                 else {
 988                   free(buf);
 989                   free(vob);
 990                   return NULL;
 991                 }
 992             } else {
 993                 while (vobsub_parse_one_line(vob, fd, &extradata, &extradata_len) >= 0)
 994                     /* NOOP */ ;
 995                 rar_close(fd);
 996             }
 997             if (spu)
 998               *spu = spudec_new_scaled(vob->palette, vob->orig_frame_width, vob->orig_frame_height, extradata, extradata_len);
 999             if (extradata)
1000               free(extradata);
1001
1002             /* read the indexed mpeg_stream */
1003             strcpy(buf, name);
1004             strcat(buf, ".sub");
1005             mpg = mpeg_open(buf);
1006             if (mpg == NULL) {
1007               if(force)
1008                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open SUB file\n");
1009               else {
1010
1011                 free(buf);
1012                 free(vob);
1013                 return NULL;
1014               }
1015             } else {
1016                 long last_pts_diff = 0;
1017                 while (!mpeg_eof(mpg)) {
1018                     off_t pos = mpeg_tell(mpg);
1019                     if (mpeg_run(mpg) < 0) {
1020                         if (!mpeg_eof(mpg))
1021                             mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: mpeg_run error\n");
1022                         break;
1023                     }
1024                     if (mpg->packet_size) {
1025                         if ((mpg->aid & 0xe0) == 0x20) {
1026                             unsigned int sid = mpg->aid & 0x1f;
1027                             if (vobsub_ensure_spu_stream(vob, sid) >= 0)  {
1028                                 packet_queue_t *queue = vob->spu_streams + sid;
1029                                 /* get the packet to fill */
1030                                 if (queue->packets_size == 0 && packet_queue_grow(queue)  < 0)
1031                                   abort();
1032                                 while (queue->current_index + 1 < queue->packets_size
1033                                        && queue->packets[queue->current_index + 1].filepos <= pos)
1034                                     ++queue->current_index;
1035                                 if (queue->current_index < queue->packets_size) {
1036                                     packet_t *pkt;
1037                                     if (queue->packets[queue->current_index].data) {
1038                                         /* insert a new packet and fix the PTS ! */
1039                                         packet_queue_insert(queue);
1040                                         queue->packets[queue->current_index].pts100 =
1041                                             mpg->pts + last_pts_diff;
1042                                     }
1043                                     pkt = queue->packets + queue->current_index;
1044                                     if (pkt->pts100 != UINT_MAX) {
1045                                     if (queue->packets_size > 1)
1046                                         last_pts_diff = pkt->pts100 - mpg->pts;
1047                                     else
1048                                         pkt->pts100 = mpg->pts;
1049                                     /* FIXME: should not use mpg_sub internal informations, make a copy */
1050                                     pkt->data = mpg->packet;
1051                                     pkt->size = mpg->packet_size;
1052                                     mpg->packet = NULL;
1053                                     mpg->packet_reserve = 0;
1054                                     mpg->packet_size = 0;
1055                                     }
1056                                 }
1057                             }
1058                             else
1059                                 mp_msg(MSGT_VOBSUB,MSGL_WARN, "don't know what to do with subtitle #%u\n", sid);
1060                         }
1061                     }
1062                 }
1063                 vob->spu_streams_current = vob->spu_streams_size;
1064                 while (vob->spu_streams_current-- > 0) {
1065                     vob->spu_streams[vob->spu_streams_current].current_index = 0;
1066                     if (vobsubid == vob->spu_streams_current ||
1067                             vob->spu_streams[vob->spu_streams_current].packets_size > 0)
1068                         ++vob->spu_valid_streams_size;
1069                 }
1070                 mpeg_free(mpg);
1071             }
1072             free(buf);
1073         }
1074     }
1075     return vob;
1076 }
1077
1078 void
1079 vobsub_close(void *this)
1080 {
1081     vobsub_t *vob = (vobsub_t *)this;
1082     if (vob->spu_streams) {
1083         while (vob->spu_streams_size--)
1084             packet_queue_destroy(vob->spu_streams + vob->spu_streams_size);
1085         free(vob->spu_streams);
1086     }
1087     free(vob);
1088 }
1089
1090 unsigned int
1091 vobsub_get_indexes_count(void *vobhandle)
1092 {
1093     vobsub_t *vob = (vobsub_t *) vobhandle;
1094     return vob->spu_valid_streams_size;
1095 }
1096
1097 char *
1098 vobsub_get_id(void *vobhandle, unsigned int index)
1099 {
1100     vobsub_t *vob = (vobsub_t *) vobhandle;
1101     return (index < vob->spu_streams_size) ? vob->spu_streams[index].id : NULL;
1102 }
1103
1104 int vobsub_get_id_by_index(void *vobhandle, unsigned int index)
1105 {
1106     vobsub_t *vob = vobhandle;
1107     int i, j;
1108     if (vob == NULL)
1109         return -1;
1110     for (i = 0, j = 0; i < vob->spu_streams_size; ++i)
1111         if (i == vobsubid || vob->spu_streams[i].packets_size > 0) {
1112             if (j == index)
1113                 return i;
1114             ++j;
1115         }
1116     return -1;
1117 }
1118
1119 int vobsub_get_index_by_id(void *vobhandle, int id)
1120 {
1121     vobsub_t *vob = vobhandle;
1122     int i, j;
1123     if (vob == NULL || id < 0 || id >= vob->spu_streams_size)
1124         return -1;
1125     if (id != vobsubid && !vob->spu_streams[id].packets_size)
1126         return -1;
1127     for (i = 0, j = 0; i < id; ++i)
1128         if (i == vobsubid || vob->spu_streams[i].packets_size > 0)
1129             ++j;
1130     return j;
1131 }
1132
1133 int
1134 vobsub_set_from_lang(void *vobhandle, unsigned char * lang)
1135 {
1136     int i;
1137     vobsub_t *vob= (vobsub_t *) vobhandle;
1138     while(lang && strlen(lang) >= 2){
1139       for(i=0; i < vob->spu_streams_size; i++)
1140         if (vob->spu_streams[i].id)
1141           if ((strncmp(vob->spu_streams[i].id, lang, 2)==0)){
1142             vobsub_id=i;
1143             mp_msg(MSGT_VOBSUB, MSGL_INFO, "Selected VOBSUB language: %d language: %s\n", i, vob->spu_streams[i].id);
1144             return 0;
1145           }
1146       lang+=2;while (lang[0]==',' || lang[0]==' ') ++lang;
1147     }
1148     mp_msg(MSGT_VOBSUB, MSGL_WARN, "No matching VOBSUB language found!\n");
1149     return -1;
1150 }
1151
1152 /// make sure we seek to the first packet of packets having same pts values.
1153 static void vobsub_queue_reseek(packet_queue_t *queue, unsigned int pts100) {
1154     int reseek_count = 0;
1155     unsigned int lastpts = 0;
1156
1157     if (queue->current_index > 0
1158             && (queue->packets[queue->current_index].pts100 == UINT_MAX
1159             || queue->packets[queue->current_index].pts100 > pts100)) {
1160       // possible pts seek previous, try to check it.
1161       int i = 1;
1162       while (queue->current_index >= i
1163               && queue->packets[queue->current_index-i].pts100 == UINT_MAX)
1164           ++i;
1165       if (queue->current_index >= i
1166               && queue->packets[queue->current_index-i].pts100 > pts100)
1167         // pts seek previous confirmed, reseek from beginning
1168         queue->current_index = 0;
1169     }
1170     while (queue->current_index < queue->packets_size
1171             && queue->packets[queue->current_index].pts100 <= pts100) {
1172       lastpts = queue->packets[queue->current_index].pts100;
1173       ++queue->current_index;
1174       ++reseek_count;
1175     }
1176     while (reseek_count-- && --queue->current_index) {
1177       if (queue->packets[queue->current_index-1].pts100 != UINT_MAX &&
1178           queue->packets[queue->current_index-1].pts100 != lastpts)
1179         break;
1180     }
1181 }
1182
1183 int
1184 vobsub_get_packet(void *vobhandle, float pts,void** data, int* timestamp) {
1185   vobsub_t *vob = (vobsub_t *)vobhandle;
1186   unsigned int pts100 = 90000 * pts;
1187   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1188     packet_queue_t *queue = vob->spu_streams + vobsub_id;
1189
1190     vobsub_queue_reseek(queue, pts100);
1191
1192     while (queue->current_index < queue->packets_size) {
1193       packet_t *pkt = queue->packets + queue->current_index;
1194       if (pkt->pts100 != UINT_MAX)
1195       if (pkt->pts100 <= pts100) {
1196         ++queue->current_index;
1197         *data = pkt->data;
1198         *timestamp = pkt->pts100;
1199         return pkt->size;
1200       } else break;
1201       else
1202           ++queue->current_index;
1203     }
1204   }
1205   return -1;
1206 }
1207
1208 int
1209 vobsub_get_next_packet(void *vobhandle, void** data, int* timestamp)
1210 {
1211   vobsub_t *vob = (vobsub_t *)vobhandle;
1212   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1213     packet_queue_t *queue = vob->spu_streams + vobsub_id;
1214     if (queue->current_index < queue->packets_size) {
1215       packet_t *pkt = queue->packets + queue->current_index;
1216       ++queue->current_index;
1217       *data = pkt->data;
1218       *timestamp = pkt->pts100;
1219       return pkt->size;
1220     }
1221   }
1222   return -1;
1223 }
1224
1225 void vobsub_seek(void * vobhandle, float pts)
1226 {
1227   vobsub_t * vob = (vobsub_t *)vobhandle;
1228   packet_queue_t * queue;
1229   int seek_pts100 = pts * 90000;
1230
1231   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1232     /* do not seek if we don't know the id */
1233     if (vobsub_get_id(vob, vobsub_id) == NULL)
1234             return;
1235     queue = vob->spu_streams + vobsub_id;
1236     queue->current_index = 0;
1237     vobsub_queue_reseek(queue, seek_pts100);
1238   }
1239 }
1240
1241 void
1242 vobsub_reset(void *vobhandle)
1243 {
1244     vobsub_t *vob = (vobsub_t *)vobhandle;
1245     if (vob->spu_streams) {
1246         unsigned int n = vob->spu_streams_size;
1247         while (n-- > 0)
1248             vob->spu_streams[n].current_index = 0;
1249     }
1250 }
1251
1252 /**********************************************************************
1253  * Vobsub output
1254  **********************************************************************/
1255
1256 typedef struct {
1257     FILE *fsub;
1258     FILE *fidx;
1259     unsigned int aid;
1260 } vobsub_out_t;
1261
1262 static void
1263 create_idx(vobsub_out_t *me, const unsigned int *palette, unsigned int orig_width, unsigned int orig_height)
1264 {
1265     int i;
1266     fprintf(me->fidx,
1267             "# VobSub index file, v7 (do not modify this line!)\n"
1268             "#\n"
1269             "# Generated by MPlayer " VERSION "\n"
1270             "# See <URL:http://www.mplayerhq.hu/> for more information about MPlayer\n"
1271             "# See <URL:http://wiki.multimedia.cx/index.php?title=VOBsub> for more information about Vobsub\n"
1272             "#\n"
1273             "size: %ux%u\n",
1274             orig_width, orig_height);
1275     if (palette) {
1276         fputs("palette:", me->fidx);
1277         for (i = 0; i < 16; ++i) {
1278             const double y = palette[i] >> 16 & 0xff,
1279                 u = (palette[i] >> 8 & 0xff) - 128.0,
1280                 v = (palette[i] & 0xff) - 128.0;
1281             if (i)
1282                 putc(',', me->fidx);
1283             fprintf(me->fidx, " %02x%02x%02x",
1284                     av_clip_uint8(y + 1.4022 * u),
1285                     av_clip_uint8(y - 0.3456 * u - 0.7145 * v),
1286                     av_clip_uint8(y + 1.7710 * v));
1287         }
1288         putc('\n', me->fidx);
1289     }
1290
1291     fprintf(me->fidx,"# ON: displays only forced subtitles, OFF: shows everything\n"
1292             "forced subs: OFF\n");
1293 }
1294
1295 void *
1296 vobsub_out_open(const char *basename, const unsigned int *palette,
1297                 unsigned int orig_width, unsigned int orig_height,
1298                 const char *id, unsigned int index)
1299 {
1300     vobsub_out_t *result = NULL;
1301     char *filename;
1302     filename = malloc(strlen(basename) + 5);
1303     if (filename) {
1304         result = malloc(sizeof(vobsub_out_t));
1305         if (result) {
1306             result->aid = index;
1307             strcpy(filename, basename);
1308             strcat(filename, ".sub");
1309             result->fsub = fopen(filename, "ab");
1310             if (result->fsub == NULL)
1311                 perror("Error: vobsub_out_open subtitle file open failed");
1312             strcpy(filename, basename);
1313             strcat(filename, ".idx");
1314             result->fidx = fopen(filename, "ab");
1315             if (result->fidx) {
1316                 if (ftell(result->fidx) == 0){
1317                     create_idx(result, palette, orig_width, orig_height);
1318                     /* Make the selected language the default language */
1319                     fprintf(result->fidx, "\n# Language index in use\nlangidx: %u\n", index);
1320                 }
1321                 fprintf(result->fidx, "\nid: %s, index: %u\n", id ? id : "xx", index);
1322                 /* So that we can check the file now */
1323                 fflush(result->fidx);
1324             }
1325             else
1326                 perror("Error: vobsub_out_open index file open failed");
1327             free(filename);
1328         }
1329     }
1330     return result;
1331 }
1332
1333 void
1334 vobsub_out_close(void *me)
1335 {
1336     vobsub_out_t *vob = (vobsub_out_t*)me;
1337     if (vob->fidx)
1338         fclose(vob->fidx);
1339     if (vob->fsub)
1340         fclose(vob->fsub);
1341     free(vob);
1342 }
1343
1344 void
1345 vobsub_out_output(void *me, const unsigned char *packet, int len, double pts)
1346 {
1347     static double last_pts;
1348     static int last_pts_set = 0;
1349     vobsub_out_t *vob = (vobsub_out_t*)me;
1350     if (vob->fsub) {
1351         /*  Windows' Vobsub require that every packet is exactly 2kB long */
1352         unsigned char buffer[2048];
1353         unsigned char *p;
1354         int remain = 2048;
1355         /* Do not output twice a line with the same timestamp, this
1356            breaks Windows' Vobsub */
1357         if (vob->fidx && (!last_pts_set || last_pts != pts)) {
1358             static unsigned int last_h = 9999, last_m = 9999, last_s = 9999, last_ms = 9999;
1359             unsigned int h, m, ms;
1360             double s;
1361             s = pts;
1362             h = s / 3600;
1363             s -= h * 3600;
1364             m = s / 60;
1365             s -= m * 60;
1366             ms = (s - (unsigned int) s) * 1000;
1367             if (ms >= 1000)     /* prevent overfolws or bad float stuff */
1368                 ms = 0;
1369             if (h != last_h || m != last_m || (unsigned int) s != last_s || ms != last_ms) {
1370                 fprintf(vob->fidx, "timestamp: %02u:%02u:%02u:%03u, filepos: %09lx\n",
1371                         h, m, (unsigned int) s, ms, ftell(vob->fsub));
1372                 last_h = h;
1373                 last_m = m;
1374                 last_s = (unsigned int) s;
1375                 last_ms = ms;
1376             }
1377         }
1378         last_pts = pts;
1379         last_pts_set = 1;
1380
1381         /* Packet start code: Windows' Vobsub needs this */
1382         p = buffer;
1383         *p++ = 0;               /* 0x00 */
1384         *p++ = 0;
1385         *p++ = 1;
1386         *p++ = 0xba;
1387         *p++ = 0x40;
1388         memset(p, 0, 9);
1389         p += 9;
1390         {   /* Packet */
1391             static unsigned char last_pts[5] = { 0, 0, 0, 0, 0};
1392             unsigned char now_pts[5];
1393             int pts_len, pad_len, datalen = len;
1394             pts *= 90000;
1395             now_pts[0] = 0x21 | (((unsigned long)pts >> 29) & 0x0e);
1396             now_pts[1] = ((unsigned long)pts >> 22) & 0xff;
1397             now_pts[2] = 0x01 | (((unsigned long)pts >> 14) & 0xfe);
1398             now_pts[3] = ((unsigned long)pts >> 7) & 0xff;
1399             now_pts[4] = 0x01 | (((unsigned long)pts << 1) & 0xfe);
1400             pts_len = memcmp(last_pts, now_pts, sizeof(now_pts)) ? sizeof(now_pts) : 0;
1401             memcpy(last_pts, now_pts, sizeof(now_pts));
1402
1403             datalen += 3;       /* Version, PTS_flags, pts_len */
1404             datalen += pts_len;
1405             datalen += 1;       /* AID */
1406             pad_len = 2048 - (p - buffer) - 4 /* MPEG ID */ - 2 /* payload len */ - datalen;
1407             /* XXX - Go figure what should go here!  In any case the
1408                packet has to be completly filled.  If I can fill it
1409                with padding (0x000001be) latter I'll do that.  But if
1410                there is only room for 6 bytes then I can not write a
1411                padding packet.  So I add some padding in the PTS
1412                field.  This looks like a dirty kludge.  Oh well... */
1413             if (pad_len < 0) {
1414                 /* Packet is too big.  Let's try ommiting the PTS field */
1415                 datalen -= pts_len;
1416                 pts_len = 0;
1417                 pad_len = 0;
1418             }
1419             else if (pad_len > 6)
1420                 pad_len = 0;
1421             datalen += pad_len;
1422
1423             *p++ = 0;           /* 0x0e */
1424             *p++ = 0;
1425             *p++ = 1;
1426             *p++ = 0xbd;
1427
1428             *p++ = (datalen >> 8) & 0xff; /* length of payload */
1429             *p++ = datalen & 0xff;
1430             *p++ = 0x80;                /* System-2 (.VOB) stream */
1431             *p++ = pts_len ? 0x80 : 0x00; /* pts_flags */
1432             *p++ = pts_len + pad_len;
1433             memcpy(p, now_pts, pts_len);
1434             p += pts_len;
1435             memset(p, 0, pad_len);
1436             p += pad_len;
1437         }
1438         *p++ = 0x20 |  vob->aid; /* aid */
1439         if (fwrite(buffer, p - buffer, 1, vob->fsub) != 1
1440             || fwrite(packet, len, 1, vob->fsub) != 1)
1441             perror("ERROR: vobsub write failed");
1442         else
1443             remain -= p - buffer + len;
1444
1445         /* Padding */
1446         if (remain >= 6) {
1447             p = buffer;
1448             *p++ = 0x00;
1449             *p++ = 0x00;
1450             *p++ = 0x01;
1451             *p++ = 0xbe;
1452             *p++ = (remain - 6) >> 8;
1453             *p++ = (remain - 6) & 0xff;
1454             /* for better compression, blank this */
1455             memset(buffer + 6, 0, remain - (p - buffer));
1456             if (fwrite(buffer, remain, 1, vob->fsub) != 1)
1457                 perror("ERROR: vobsub padding write failed");
1458         }
1459         else if (remain > 0) {
1460             /* I don't know what to output.  But anyway the block
1461                needs to be 2KB big */
1462             memset(buffer, 0, remain);
1463             if (fwrite(buffer, remain, 1, vob->fsub) != 1)
1464                 perror("ERROR: vobsub blank padding write failed");
1465         }
1466         else if (remain < 0)
1467             fprintf(stderr,
1468                     "\nERROR: wrong thing happenned...\n"
1469                     "  I wrote a %i data bytes spu packet and that's too long\n", len);
1470     }
1471 }