blob: c5688c2fc518cb42e958b8d0ea84c678ca2e93be (
plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#pragma once
#pragma warning(disable : 4200)
#include <atlbase.h>
#include <atlcoll.h>
//#include <winioctl.h> // platform sdk
#include <winddk\ntddcdvd.h>
class CDVDSession
{
protected:
HANDLE m_hDrive;
DVD_SESSION_ID m_session;
bool BeginSession();
void EndSession();
BYTE m_SessionKey[5];
bool Authenticate();
BYTE m_DiscKey[6], m_TitleKey[6];
bool GetDiscKey();
bool GetTitleKey(int lba, BYTE* pKey);
public:
CDVDSession();
virtual ~CDVDSession();
bool Open(LPCTSTR path);
void Close();
operator HANDLE() {return m_hDrive;}
operator DVD_SESSION_ID() {return m_session;}
bool SendKey(DVD_KEY_TYPE KeyType, BYTE* pKeyData);
bool ReadKey(DVD_KEY_TYPE KeyType, BYTE* pKeyData, int lba = 0);
};
class CLBAFile : private CFile
{
public:
CLBAFile();
virtual ~CLBAFile();
bool IsOpen();
bool Open(LPCTSTR path);
void Close();
int GetLength();
int GetPosition();
int Seek(int lba);
bool Read(BYTE* buff);
};
class CVobFile : public CDVDSession
{
// all files
typedef struct {CString fn; int size;} file_t;
CAtlArray<file_t> m_files;
int m_iFile;
int m_pos, m_size, m_offset;
// currently opened file
CLBAFile m_file;
// attribs
bool m_fDVD, m_fHasDiscKey, m_fHasTitleKey;
public:
CVobFile();
virtual ~CVobFile();
bool IsDVD();
bool HasDiscKey(BYTE* key);
bool HasTitleKey(BYTE* key);
bool Open(CString fn, CAtlList<CString>& files /* out */); // vts ifo
bool Open(CAtlList<CString>& files, int offset = -1); // vts vobs, video vob offset in lba
void Close();
int GetLength();
int GetPosition();
int Seek(int pos);
bool Read(BYTE* buff);
};
|
