stream/cache2.c

   1 #include "config.h"
   2
   3 // Initial draft of my new cache system...
   4 // Note it runs in 2 processes (using fork()), but doesn't requires locking!!
   5 // TODO: seeking, data consistency checking
   6
   7 #define READ_USLEEP_TIME 10000
   8 #define FILL_USLEEP_TIME 50000
   9 #define PREFILL_SLEEP_TIME 200
  10 #define CONTROL_SLEEP_TIME 0
  11
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <string.h>
  15 #include <signal.h>
  16 #include <sys/types.h>
  17 #include <unistd.h>
  18 #include <errno.h>
  19
  20 #include "osdep/shmem.h"
  21 #include "osdep/timer.h"
  22 #if defined(__MINGW32__)
  23 #include <windows.h>
  24 static void ThreadProc( void *s );
  25 #elif defined(__OS2__)
  26 #define INCL_DOS
  27 #include <os2.h>
  28 static void ThreadProc( void *s );
  29 #elif defined(PTHREAD_CACHE)
  30 #include <pthread.h>
  31 static void *ThreadProc(void *s);
  32 #else
  33 #include <sys/wait.h>
  34 #endif
  35
  36 #include "mp_msg.h"
  37 #include "help_mp.h"
  38
  39 #include "stream.h"
  40 #include "cache2.h"
  41 extern int use_gui;
  42
  43 int stream_fill_buffer(stream_t *s);
  44 int stream_seek_long(stream_t *s,off_t pos);
  45
  46 typedef struct {
  47   // constats:
  48   unsigned char *buffer;      // base pointer of the alllocated buffer memory
  49   int buffer_size; // size of the alllocated buffer memory
  50   int sector_size; // size of a single sector (2048/2324)
  51   int back_size;   // we should keep back_size amount of old bytes for backward seek
  52   int fill_limit;  // we should fill buffer only if space>=fill_limit
  53   int seek_limit;  // keep filling cache if distanse is less that seek limit
  54   // filler's pointers:
  55   int eof;
  56   off_t min_filepos; // buffer contain only a part of the file, from min-max pos
  57   off_t max_filepos;
  58   off_t offset;      // filepos <-> bufferpos  offset value (filepos of the buffer's first byte)
  59   // reader's pointers:
  60   off_t read_filepos;
  61   // commands/locking:
  62 //  int seek_lock;   // 1 if we will seek/reset buffer, 2 if we are ready for cmd
  63 //  int fifo_flag;  // 1 if we should use FIFO to notice cache about buffer reads.
  64   // callback
  65   stream_t* stream;
  66   volatile int control;
  67   volatile unsigned control_uint_arg;
  68   volatile double control_double_arg;
  69   volatile int control_res;
  70   volatile off_t control_new_pos;
  71   volatile double stream_time_length;
  72 } cache_vars_t;
  73
  74 static int min_fill=0;
  75
  76 int cache_fill_status=0;
  77
  78 void cache_stats(cache_vars_t* s){
  79   int newb=s->max_filepos-s->read_filepos; // new bytes in the buffer
  80   mp_msg(MSGT_CACHE,MSGL_INFO,"0x%06X  [0x%06X]  0x%06X   ",(int)s->min_filepos,(int)s->read_filepos,(int)s->max_filepos);
  81   mp_msg(MSGT_CACHE,MSGL_INFO,"%3d %%  (%3d%%)\n",100*newb/s->buffer_size,100*min_fill/s->buffer_size);
  82 }
  83
  84 int cache_read(cache_vars_t* s,unsigned char* buf,int size){
  85   int total=0;
  86   while(size>0){
  87     int pos,newb,len;
  88
  89   //printf("CACHE2_READ: 0x%X <= 0x%X <= 0x%X  \n",s->min_filepos,s->read_filepos,s->max_filepos);
  90
  91     if(s->read_filepos>=s->max_filepos || s->read_filepos<s->min_filepos){
  92         // eof?
  93         if(s->eof) break;
  94         // waiting for buffer fill...
  95         usec_sleep(READ_USLEEP_TIME); // 10ms
  96         continue; // try again...
  97     }
  98
  99     newb=s->max_filepos-s->read_filepos; // new bytes in the buffer
 100     if(newb<min_fill) min_fill=newb; // statistics...
 101
 102 //    printf("*** newb: %d bytes ***\n",newb);
 103
 104     pos=s->read_filepos - s->offset;
 105     if(pos<0) pos+=s->buffer_size; else
 106     if(pos>=s->buffer_size) pos-=s->buffer_size;
 107
 108     if(newb>s->buffer_size-pos) newb=s->buffer_size-pos; // handle wrap...
 109     if(newb>size) newb=size;
 110
 111     // check:
 112     if(s->read_filepos<s->min_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"Ehh. s->read_filepos<s->min_filepos !!! Report bug...\n");
 113
 114     // len=write(mem,newb)
 115     //printf("Buffer read: %d bytes\n",newb);
 116     memcpy(buf,&s->buffer[pos],newb);
 117     buf+=newb;
 118     len=newb;
 119     // ...
 120
 121     s->read_filepos+=len;
 122     size-=len;
 123     total+=len;
 124
 125   }
 126   cache_fill_status=(s->max_filepos-s->read_filepos)/(s->buffer_size / 100);
 127   return total;
 128 }
 129
 130 int cache_fill(cache_vars_t* s){
 131   int back,back2,newb,space,len,pos;
 132   off_t read=s->read_filepos;
 133
 134   if(read<s->min_filepos || read>s->max_filepos){
 135       // seek...
 136       mp_msg(MSGT_CACHE,MSGL_DBG2,"Out of boundaries... seeking to 0x%"PRIX64"  \n",(int64_t)read);
 137       // streaming: drop cache contents only if seeking backward or too much fwd:
 138       if(s->stream->type!=STREAMTYPE_STREAM ||
 139           read<s->min_filepos || read>=s->max_filepos+s->seek_limit)
 140       {
 141         s->offset= // FIXME!?
 142         s->min_filepos=s->max_filepos=read; // drop cache content :(
 143         if(s->stream->eof) stream_reset(s->stream);
 144         stream_seek(s->stream,read);
 145         mp_msg(MSGT_CACHE,MSGL_DBG2,"Seek done. new pos: 0x%"PRIX64"  \n",(int64_t)stream_tell(s->stream));
 146       }
 147   }
 148
 149   // calc number of back-bytes:
 150   back=read - s->min_filepos;
 151   if(back<0) back=0; // strange...
 152   if(back>s->back_size) back=s->back_size;
 153
 154   // calc number of new bytes:
 155   newb=s->max_filepos - read;
 156   if(newb<0) newb=0; // strange...
 157
 158   // calc free buffer space:
 159   space=s->buffer_size - (newb+back);
 160
 161   // calc bufferpos:
 162   pos=s->max_filepos - s->offset;
 163   if(pos>=s->buffer_size) pos-=s->buffer_size; // wrap-around
 164
 165   if(space<s->fill_limit){
 166 //    printf("Buffer is full (%d bytes free, limit: %d)\n",space,s->fill_limit);
 167     return 0; // no fill...
 168   }
 169
 170 //  printf("### read=0x%X  back=%d  newb=%d  space=%d  pos=%d\n",read,back,newb,space,pos);
 171
 172   // reduce space if needed:
 173   if(space>s->buffer_size-pos) space=s->buffer_size-pos;
 174
 175 //  if(space>32768) space=32768; // limit one-time block size
 176   if(space>4*s->sector_size) space=4*s->sector_size;
 177
 178 //  if(s->seek_lock) return 0; // FIXME
 179
 180 #if 1
 181   // back+newb+space <= buffer_size
 182   back2=s->buffer_size-(space+newb); // max back size
 183   if(s->min_filepos<(read-back2)) s->min_filepos=read-back2;
 184 #else
 185   s->min_filepos=read-back; // avoid seeking-back to temp area...
 186 #endif
 187
 188   // ....
 189   //printf("Buffer fill: %d bytes of %d\n",space,s->buffer_size);
 190   //len=stream_fill_buffer(s->stream);
 191   //memcpy(&s->buffer[pos],s->stream->buffer,len); // avoid this extra copy!
 192   // ....
 193   len=stream_read(s->stream,&s->buffer[pos],space);
 194   if(!len) s->eof=1;
 195
 196   s->max_filepos+=len;
 197   if(pos+len>=s->buffer_size){
 198       // wrap...
 199       s->offset+=s->buffer_size;
 200   }
 201
 202   return len;
 203
 204 }
 205
 206 static int cache_execute_control(cache_vars_t *s) {
 207   int res = 1;
 208   static unsigned last;
 209   if (!s->stream->control) {
 210     s->stream_time_length = 0;
 211     s->control_new_pos = 0;
 212     s->control_res = STREAM_UNSUPPORTED;
 213     s->control = -1;
 214     return res;
 215   }
 216   if (GetTimerMS() - last > 99) {
 217     double len;
 218     if (s->stream->control(s->stream, STREAM_CTRL_GET_TIME_LENGTH, &len) == STREAM_OK)
 219       s->stream_time_length = len;
 220     else
 221       s->stream_time_length = 0;
 222     last = GetTimerMS();
 223   }
 224   if (s->control == -1) return res;
 225   switch (s->control) {
 226     case STREAM_CTRL_GET_CURRENT_TIME:
 227     case STREAM_CTRL_SEEK_TO_TIME:
 228     case STREAM_CTRL_GET_ASPECT_RATIO:
 229       s->control_res = s->stream->control(s->stream, s->control, &s->control_double_arg);
 230       break;
 231     case STREAM_CTRL_SEEK_TO_CHAPTER:
 232     case STREAM_CTRL_GET_NUM_CHAPTERS:
 233     case STREAM_CTRL_GET_CURRENT_CHAPTER:
 234     case STREAM_CTRL_GET_NUM_ANGLES:
 235     case STREAM_CTRL_GET_ANGLE:
 236     case STREAM_CTRL_SET_ANGLE:
 237       s->control_res = s->stream->control(s->stream, s->control, &s->control_uint_arg);
 238       break;
 239     case -2:
 240       res = 0;
 241     default:
 242       s->control_res = STREAM_UNSUPPORTED;
 243       break;
 244   }
 245   s->control_new_pos = s->stream->pos;
 246   s->control = -1;
 247   return res;
 248 }
 249
 250 static cache_vars_t* cache_init(int size,int sector){
 251   int num;
 252 #if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
 253   cache_vars_t* s=shmem_alloc(sizeof(cache_vars_t));
 254 #else
 255   cache_vars_t* s=malloc(sizeof(cache_vars_t));
 256 #endif
 257   if(s==NULL) return NULL;
 258
 259   memset(s,0,sizeof(cache_vars_t));
 260   num=size/sector;
 261   if(num < 16){
 262      num = 16;
 263   }//32kb min_size
 264   s->buffer_size=num*sector;
 265   s->sector_size=sector;
 266 #if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
 267   s->buffer=shmem_alloc(s->buffer_size);
 268 #else
 269   s->buffer=malloc(s->buffer_size);
 270 #endif
 271
 272   if(s->buffer == NULL){
 273 #if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
 274     shmem_free(s,sizeof(cache_vars_t));
 275 #else
 276     free(s);
 277 #endif
 278     return NULL;
 279   }
 280
 281   s->fill_limit=8*sector;
 282   s->back_size=s->buffer_size/2;
 283   return s;
 284 }
 285
 286 void cache_uninit(stream_t *s) {
 287   cache_vars_t* c = s->cache_data;
 288   if(s->cache_pid) {
 289 #if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
 290     cache_do_control(s, -2, NULL);
 291 #else
 292     kill(s->cache_pid,SIGKILL);
 293     waitpid(s->cache_pid,NULL,0);
 294 #endif
 295     s->cache_pid = 0;
 296   }
 297   if(!c) return;
 298 #if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
 299   free(c->stream);
 300   free(c->buffer);
 301   c->buffer = NULL;
 302   free(s->cache_data);
 303 #else
 304   shmem_free(c->buffer,c->buffer_size);
 305   c->buffer = NULL;
 306   shmem_free(s->cache_data,sizeof(cache_vars_t));
 307 #endif
 308   s->cache_data = NULL;
 309 }
 310
 311 static void exit_sighandler(int x){
 312   // close stream
 313   exit(0);
 314 }
 315
 316 /**
 317  * \return 1 on success, 0 if the function was interrupted and -1 on error
 318  */
 319 int stream_enable_cache(stream_t *stream,int size,int min,int seek_limit){
 320   int ss = stream->sector_size ? stream->sector_size : STREAM_BUFFER_SIZE;
 321   int res = -1;
 322   cache_vars_t* s;
 323
 324   if (stream->flags & STREAM_NON_CACHEABLE) {
 325     mp_msg(MSGT_CACHE,MSGL_STATUS,"\rThis stream is non-cacheable\n");
 326     return 1;
 327   }
 328
 329   s=cache_init(size,ss);
 330   if(s == NULL) return -1;
 331   stream->cache_data=s;
 332   s->stream=stream; // callback
 333   s->seek_limit=seek_limit;
 334
 335
 336   //make sure that we won't wait from cache_fill
 337   //more data than it is alowed to fill
 338   if (s->seek_limit > s->buffer_size - s->fill_limit ){
 339      s->seek_limit = s->buffer_size - s->fill_limit;
 340   }
 341   if (min > s->buffer_size - s->fill_limit) {
 342      min = s->buffer_size - s->fill_limit;
 343   }
 344
 345 #if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
 346   if((stream->cache_pid=fork())){
 347     if ((pid_t)stream->cache_pid == -1)
 348       stream->cache_pid = 0;
 349 #else
 350   {
 351     stream_t* stream2=malloc(sizeof(stream_t));
 352     memcpy(stream2,s->stream,sizeof(stream_t));
 353     s->stream=stream2;
 354 #if defined(__MINGW32__)
 355     stream->cache_pid = _beginthread( ThreadProc, 0, s );
 356 #elif defined(__OS2__)
 357     stream->cache_pid = _beginthread( ThreadProc, NULL, 256 * 1024, s );
 358 #else
 359     {
 360     pthread_t tid;
 361     pthread_create(&tid, NULL, ThreadProc, s);
 362     stream->cache_pid = 1;
 363     }
 364 #endif
 365 #endif
 366     if (!stream->cache_pid) {
 367         mp_msg(MSGT_CACHE, MSGL_ERR,
 368                "Starting cache process/thread failed: %s.\n", strerror(errno));
 369         goto err_out;
 370     }
 371     // wait until cache is filled at least prefill_init %
 372     mp_msg(MSGT_CACHE,MSGL_V,"CACHE_PRE_INIT: %"PRId64" [%"PRId64"] %"PRId64"  pre:%d  eof:%d  \n",
 373         (int64_t)s->min_filepos,(int64_t)s->read_filepos,(int64_t)s->max_filepos,min,s->eof);
 374     while(s->read_filepos<s->min_filepos || s->max_filepos-s->read_filepos<min){
 375         mp_msg(MSGT_CACHE,MSGL_STATUS,MSGTR_CacheFill,
 376             100.0*(float)(s->max_filepos-s->read_filepos)/(float)(s->buffer_size),
 377             (int64_t)s->max_filepos-s->read_filepos
 378         );
 379         if(s->eof) break; // file is smaller than prefill size
 380         if(stream_check_interrupt(PREFILL_SLEEP_TIME)) {
 381           res = 0;
 382           goto err_out;
 383         }
 384     }
 385     mp_msg(MSGT_CACHE,MSGL_STATUS,"\n");
 386     return 1; // parent exits
 387
 388 err_out:
 389     cache_uninit(stream);
 390     return res;
 391   }
 392
 393 #if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
 394 }
 395 #ifdef PTHREAD_CACHE
 396 static void *ThreadProc( void *s ){
 397 #else
 398 static void ThreadProc( void *s ){
 399 #endif
 400 #endif
 401
 402 #ifdef CONFIG_GUI
 403   use_gui = 0; // mp_msg may not use gui stuff in forked code
 404 #endif
 405 // cache thread mainloop:
 406   signal(SIGTERM,exit_sighandler); // kill
 407   do {
 408     if(!cache_fill(s)){
 409          usec_sleep(FILL_USLEEP_TIME); // idle
 410     }
 411 //       cache_stats(s->cache_data);
 412   } while (cache_execute_control(s));
 413 #if defined(__MINGW32__) || defined(__OS2__)
 414   _endthread();
 415 #endif
 416 #ifdef PTHREAD_CACHE
 417   return NULL;
 418 #endif
 419   // make sure forked code never leaves this function
 420   exit(0);
 421 }
 422
 423 int cache_stream_fill_buffer(stream_t *s){
 424   int len;
 425   if(s->eof){ s->buf_pos=s->buf_len=0; return 0; }
 426   if(!s->cache_pid) return stream_fill_buffer(s);
 427
 428 //  cache_stats(s->cache_data);
 429
 430   if(s->pos!=((cache_vars_t*)s->cache_data)->read_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"!!! read_filepos differs!!! report this bug...\n");
 431
 432   len=cache_read(s->cache_data,s->buffer, ((cache_vars_t*)s->cache_data)->sector_size);
 433   //printf("cache_stream_fill_buffer->read -> %d\n",len);
 434
 435   if(len<=0){ s->eof=1; s->buf_pos=s->buf_len=0; return 0; }
 436   s->buf_pos=0;
 437   s->buf_len=len;
 438   s->pos+=len;
 439 //  printf("[%d]",len);fflush(stdout);
 440   return len;
 441
 442 }
 443
 444 int cache_stream_seek_long(stream_t *stream,off_t pos){
 445   cache_vars_t* s;
 446   off_t newpos;
 447   if(!stream->cache_pid) return stream_seek_long(stream,pos);
 448
 449   s=stream->cache_data;
 450 //  s->seek_lock=1;
 451
 452   mp_msg(MSGT_CACHE,MSGL_DBG2,"CACHE2_SEEK: 0x%"PRIX64" <= 0x%"PRIX64" (0x%"PRIX64") <= 0x%"PRIX64"  \n",s->min_filepos,pos,s->read_filepos,s->max_filepos);
 453
 454   newpos=pos/s->sector_size; newpos*=s->sector_size; // align
 455   stream->pos=s->read_filepos=newpos;
 456   s->eof=0; // !!!!!!!
 457
 458   cache_stream_fill_buffer(stream);
 459
 460   pos-=newpos;
 461   if(pos>=0 && pos<=stream->buf_len){
 462     stream->buf_pos=pos; // byte position in sector
 463     return 1;
 464   }
 465
 466 //  stream->buf_pos=stream->buf_len=0;
 467 //  return 1;
 468
 469   mp_msg(MSGT_CACHE,MSGL_V,"cache_stream_seek: WARNING! Can't seek to 0x%"PRIX64" !\n",(int64_t)(pos+newpos));
 470   return 0;
 471 }
 472
 473 int cache_do_control(stream_t *stream, int cmd, void *arg) {
 474   cache_vars_t* s = stream->cache_data;
 475   switch (cmd) {
 476     case STREAM_CTRL_SEEK_TO_TIME:
 477       s->control_double_arg = *(double *)arg;
 478       s->control = cmd;
 479       break;
 480     case STREAM_CTRL_SEEK_TO_CHAPTER:
 481     case STREAM_CTRL_SET_ANGLE:
 482       s->control_uint_arg = *(unsigned *)arg;
 483       s->control = cmd;
 484       break;
 485 // the core might call these every frame, they are too slow for this...
 486     case STREAM_CTRL_GET_TIME_LENGTH:
 487 //    case STREAM_CTRL_GET_CURRENT_TIME:
 488       *(double *)arg = s->stream_time_length;
 489       return s->stream_time_length ? STREAM_OK : STREAM_UNSUPPORTED;
 490     case STREAM_CTRL_GET_NUM_CHAPTERS:
 491     case STREAM_CTRL_GET_CURRENT_CHAPTER:
 492     case STREAM_CTRL_GET_ASPECT_RATIO:
 493     case STREAM_CTRL_GET_NUM_ANGLES:
 494     case STREAM_CTRL_GET_ANGLE:
 495     case -2:
 496       s->control = cmd;
 497       break;
 498     default:
 499       return STREAM_UNSUPPORTED;
 500   }
 501   while (s->control != -1)
 502     usec_sleep(CONTROL_SLEEP_TIME);
 503   switch (cmd) {
 504     case STREAM_CTRL_GET_TIME_LENGTH:
 505     case STREAM_CTRL_GET_CURRENT_TIME:
 506     case STREAM_CTRL_GET_ASPECT_RATIO:
 507       *(double *)arg = s->control_double_arg;
 508       break;
 509     case STREAM_CTRL_GET_NUM_CHAPTERS:
 510     case STREAM_CTRL_GET_CURRENT_CHAPTER:
 511     case STREAM_CTRL_GET_NUM_ANGLES:
 512     case STREAM_CTRL_GET_ANGLE:
 513       *(unsigned *)arg = s->control_uint_arg;
 514       break;
 515     case STREAM_CTRL_SEEK_TO_CHAPTER:
 516     case STREAM_CTRL_SEEK_TO_TIME:
 517     case STREAM_CTRL_SET_ANGLE:
 518       stream->pos = s->read_filepos = s->control_new_pos;
 519       break;
 520   }
 521   return s->control_res;
 522 }