1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
"""
@copyright: 2009 Bastian Blank <waldi@debian.org>
@license: GNU GPL-3
"""
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from ctypes import CDLL, CFUNCTYPE, c_char_p, c_int, c_void_p, create_string_buffer
try:
_libdvdcss = CDLL('libdvdcss.so.2')
except OSError:
raise ImportError
_dvdcss_open = CFUNCTYPE(c_void_p, c_char_p)(('dvdcss_open', _libdvdcss))
_dvdcss_close = CFUNCTYPE(c_int, c_void_p)(('dvdcss_open', _libdvdcss))
_dvdcss_seek = CFUNCTYPE(c_int, c_void_p, c_int, c_int)(('dvdcss_seek', _libdvdcss))
_dvdcss_read = CFUNCTYPE(c_int, c_void_p, c_void_p, c_int, c_int)(('dvdcss_read', _libdvdcss))
class DvdCssFile:
def __init__(self, filename):
self._handle = _dvdcss_open(filename.encode('utf-8'))
if not self._handle:
raise RuntimeError
self._cur = 0
def read(self, count):
return self.read_sector(count // 2048)
def read_sector(self, count, encrypted=False):
buf = create_string_buffer(count * 2048)
ret = _dvdcss_read(self._handle, buf, count, encrypted and 1 or 0)
if ret < 0:
raise IOError
self._cur += ret
if ret != count:
# Short read, libdvdcss does a weird seek in this case, fix it
self.seek_sector(self._cur)
return buf[:ret * 2048]
return buf
def seek(self, offset):
self.seek_sector(offset // 2048)
def seek_sector(self, offset, start_encrypted=False):
ret = _dvdcss_seek(self._handle, offset, start_encrypted and 1 or 0)
if start_encrypted and ret < 0:
# Retry without key
ret = _dvdcss_seek(self._handle, offset, 0)
if ret < 0 or ret != offset:
raise RuntimeError
self._cur = ret
def tell(self, offset):
return self.tell_sector() * 2048
def tell_sector(self):
return self._cur
|
