Added abandoned KDE3 version of K3B

git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/k3b@1084400 283d02a7-25f6-0310-bc7c-ecb5cbfe19da

1 files changed, 307 insertions, 0 deletions

diff --git a/libk3b/tools/k3blibdvdcss.cpp b/libk3b/tools/k3blibdvdcss.cpp
new file mode 100644
index 0000000..0a8d1f0
--- /dev/null
+++ b/libk3b/tools/k3blibdvdcss.cpp
@@ -0,0 +1,307 @@
+/* 
+ *
+ * $Id: k3blibdvdcss.cpp 619556 2007-01-03 17:38:12Z trueg $
+ * Copyright (C) 2004 Sebastian Trueg <[email protected]>
+ *
+ * This file is part of the K3b project.
+ * Copyright (C) 1998-2007 Sebastian Trueg <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ * See the file "COPYING" for the exact licensing terms.
+ */
+
+#include <config.h>
+
+#include "k3blibdvdcss.h"
+
+#include <k3bdevice.h>
+#include <k3biso9660.h>
+#include <k3biso9660backend.h>
+
+#include <qfile.h>
+#include <qcstring.h>
+#include <qvaluevector.h>
+#include <qpair.h>
+
+#include <dlfcn.h>
+
+
+void* K3bLibDvdCss::s_libDvdCss = 0;
+int K3bLibDvdCss::s_counter = 0;
+
+
+extern "C" {
+  struct dvdcss_s;
+  typedef struct dvdcss_s* dvdcss_t;
+
+  dvdcss_t (*k3b_dvdcss_open)(char*);
+  int (*k3b_dvdcss_close)( dvdcss_t );
+  int (*k3b_dvdcss_seek)( dvdcss_t, int, int );
+  int (*k3b_dvdcss_read)( dvdcss_t, void*, int, int );
+}
+
+
+
+class K3bLibDvdCss::Private
+{
+public:
+  Private()
+    :dvd(0) {
+  }
+
+  dvdcss_t dvd;
+  K3bDevice::Device* device;
+  QValueVector< QPair<int,int> > titleOffsets;
+  int currentSector;
+  bool currentSectorInTitle;
+};
+
+K3bLibDvdCss::K3bLibDvdCss()
+{
+  d = new Private();
+  s_counter++;
+}
+
+
+K3bLibDvdCss::~K3bLibDvdCss()
+{
+  close();
+  delete d;
+  s_counter--;
+  if( s_counter == 0 ) {
+    dlclose( s_libDvdCss );
+    s_libDvdCss = 0;
+  }
+}
+
+
+bool K3bLibDvdCss::open( K3bDevice::Device* dev )
+{
+  d->device = dev;
+  dev->close();
+  d->dvd = k3b_dvdcss_open( const_cast<char*>( QFile::encodeName(dev->blockDeviceName()).data() ) );
+  d->currentSector = 0;
+  d->currentSectorInTitle = false;
+  return ( d->dvd != 0 );
+}
+
+
+void K3bLibDvdCss::close()
+{
+  if( d->dvd )
+    k3b_dvdcss_close( d->dvd );
+  d->dvd = 0;
+}
+
+
+int K3bLibDvdCss::seek( int sector, int flags )
+{
+  return k3b_dvdcss_seek( d->dvd, sector, flags );
+}
+
+
+int K3bLibDvdCss::read( void* buffer, int sectors, int flags )
+{
+  return k3b_dvdcss_read( d->dvd, buffer, sectors, flags );
+}
+
+
+int K3bLibDvdCss::readWrapped( void* buffer, int firstSector, int sectors )
+{
+  // 1. are we in a title?
+  // 2. does a new title start in the read sector area?
+  //    - see below, set title if firstSector is the first sector of a new title
+  // 3. does a title end in the read sector area?
+  //    3.1 does a previous title end
+  //    3.2 does the title from 2. already end
+
+  // we need to seek to the first sector. Otherwise we get faulty data.
+  bool needToSeek = ( firstSector != d->currentSector || firstSector == 0 );
+  bool inTitle = false;
+  bool startOfTitle = false;
+
+  //
+  // Make sure we never read encrypted and unencrypted data at once since libdvdcss
+  // only decrypts the whole area of read sectors or nothing at all.
+  //
+  for( unsigned int i = 0; i < d->titleOffsets.count(); ++i ) {
+    int titleStart = d->titleOffsets[i].first;
+    int titleEnd = titleStart + d->titleOffsets[i].second - 1;
+
+    // update key when entrering a new title
+    // FIXME: we also need this if we seek into a new title (not only the start of the title)
+    if( titleStart == firstSector )
+      startOfTitle = needToSeek = inTitle = true;
+    
+    // check if a new title or non-title area starts inside the read sector range
+    if( firstSector < titleStart && firstSector+sectors > titleStart ) {
+      kdDebug() << "(K3bLibDvdCss) title start inside of sector range (" 
+		<< firstSector << "-" << (firstSector+sectors-1) 
+		<< "). only reading " << (titleStart - firstSector) << " sectors up to title offset "
+		<< (titleStart-1) << endl;
+      sectors = titleStart - firstSector;
+    }
+    
+    if( firstSector < titleEnd && firstSector+sectors > titleEnd ) {
+      kdDebug() << "(K3bLibDvdCss) title end inside of sector range (" 
+		<< firstSector << "-" << (firstSector+sectors-1) 
+		<< "). only reading " << (titleEnd - firstSector + 1) << " sectors up to title offset "
+		<< titleEnd << endl;
+      sectors = titleEnd - firstSector + 1;
+      inTitle = true;
+    }
+
+    // is our read range part of one title
+    if( firstSector >= titleStart && firstSector+sectors-1 <= titleEnd )
+      inTitle = true;
+  }
+
+  if( needToSeek ) {
+    int flags = DVDCSS_NOFLAGS;
+    if( startOfTitle )
+      flags = DVDCSS_SEEK_KEY;
+    else if( inTitle )
+      flags = DVDCSS_SEEK_MPEG;
+
+    kdDebug() << "(K3bLibDvdCss) need to seek from " << d->currentSector << " to " << firstSector << " with " << flags << endl;
+
+    d->currentSector = seek( firstSector, flags );
+    if( d->currentSector != firstSector ) {
+      kdDebug() << "(K3bLibDvdCss) seek failed: " << d->currentSector << endl;
+      return -1;
+    }
+
+    kdDebug() << "(K3bLibDvdCss) seek done: " << d->currentSector << endl;
+  }
+
+
+  int flags = DVDCSS_NOFLAGS;
+  if( inTitle )
+    flags = DVDCSS_READ_DECRYPT;
+
+  int ret = read( buffer, sectors, flags );
+  if( ret >= 0 )
+    d->currentSector += ret;
+  else
+    d->currentSector = 0; // force a seek the next time
+
+  return ret;
+}
+
+
+bool K3bLibDvdCss::crackAllKeys()
+{
+  //
+  // Loop over all titles and crack the keys (inspired by libdvdread)
+  //
+  kdDebug() << "(K3bLibDvdCss) cracking all keys." << endl;
+
+  d->titleOffsets.clear();
+
+  K3bIso9660 iso( new K3bIso9660DeviceBackend( d->device ) );
+  iso.setPlainIso9660( true );
+  if( !iso.open() ) {
+    kdDebug() << "(K3bLibDvdCss) could not open iso9660 fs." << endl;
+    return false;
+  }
+
+#ifdef K3B_DEBUG
+  iso.debug();
+#endif
+
+  const K3bIso9660Directory* dir = iso.firstIsoDirEntry();
+
+  int title = 0;
+  for( ; title < 100; ++title ) {
+    QString filename;
+
+    // first we get the menu vob
+    if( title == 0 )
+      filename.sprintf( "VIDEO_TS/VIDEO_TS.VOB" );
+    else
+      filename.sprintf( "VIDEO_TS/VTS_%02d_%d.VOB", title, 0 );
+
+    const K3bIso9660File* file = dynamic_cast<const K3bIso9660File*>( dir->entry( filename ) );
+    if( file && file->size() > 0 ) {
+      d->titleOffsets.append( qMakePair( (int)file->startSector(), (int)(file->size() / 2048U) ) );
+      kdDebug() << "(K3bLibDvdCss) Get key for /" << filename << " at " << file->startSector() << endl;
+      if( seek( (int)file->startSector(), DVDCSS_SEEK_KEY ) < 0 ) {
+	kdDebug() << "(K3bLibDvdCss) unable to seek to " << file->startSector() << endl;
+	return false;
+      }
+    }
+
+    if( title > 0 ) {
+      QPair<int,int> p;
+      int vob = 1;
+      for( ; vob < 100; ++vob ) {
+	filename.sprintf( "VIDEO_TS/VTS_%02d_%d.VOB", title, vob );
+	file = dynamic_cast<const K3bIso9660File*>( dir->entry( filename ) );
+	if( file ) {
+	  if( file->size() % 2048 )
+	    kdError() << "(K3bLibDvdCss) FILESIZE % 2048 != 0!!!" << endl;
+	  if( vob == 1 ) {
+	    p.first = file->startSector();
+	    p.second = file->size() / 2048;
+	    kdDebug() << "(K3bLibDvdCss) Get key for /" << filename << " at " << file->startSector() << endl;
+	    if( seek( (int)file->startSector(), DVDCSS_SEEK_KEY ) < 0 ) {
+	      kdDebug() << "(K3bLibDvdCss) unable to seek to " << file->startSector() << endl;
+	      return false;
+	    }
+	  }
+	  else {
+	    p.second += file->size() / 2048;
+	  }
+	}
+	else {
+	  // last vob
+	  break;
+	}
+      }
+      --vob;
+
+      // last title
+      if( vob == 0 )
+	break;
+
+      kdDebug() << "(K3bLibDvdCss) Title " << title << " " << vob << " vobs with length " << p.second << endl;
+      d->titleOffsets.append( p );
+    }
+  }
+
+  --title;
+
+  kdDebug() << "(K3bLibDvdCss) found " << title << " titles." << endl;
+
+  return (title > 0);
+}
+
+
+K3bLibDvdCss* K3bLibDvdCss::create()
+{
+  if( s_libDvdCss == 0 ) {
+    s_libDvdCss = dlopen( "libdvdcss.so.2", RTLD_LAZY|RTLD_GLOBAL );
+    if( s_libDvdCss ) {
+      k3b_dvdcss_open = (dvdcss_t (*)(char*))dlsym( s_libDvdCss, "dvdcss_open" );
+      k3b_dvdcss_close = (int (*)( dvdcss_t ))dlsym( s_libDvdCss, "dvdcss_close" );
+      k3b_dvdcss_seek = (int (*)( dvdcss_t, int, int ))dlsym( s_libDvdCss, "dvdcss_seek" );
+      k3b_dvdcss_read = (int (*)( dvdcss_t, void*, int, int ))dlsym( s_libDvdCss, "dvdcss_read" );
+
+      if( !k3b_dvdcss_open || !k3b_dvdcss_close || !k3b_dvdcss_seek || !k3b_dvdcss_read ) {
+	kdDebug() << "(K3bLibDvdCss) unable to resolve libdvdcss." << endl;
+	dlclose( s_libDvdCss );
+	s_libDvdCss = 0;
+      }
+    }
+    else
+      kdDebug() << "(K3bLibDvdCss) unable to load libdvdcss." << endl;
+  }
+
+  if( s_libDvdCss )
+    return new K3bLibDvdCss();
+  else
+    return 0;
+}