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