Exemplo n.º 1
0
        /// <summary>
        /// Read the Pk2 structure recursively.
        /// </summary>
        private void Read(long Position, string RootPath)
        {
            BinaryReader reader = new BinaryReader(m_FileStream);

            reader.BaseStream.Position = Position;
            List <Pk2Folder> folders    = new List <Pk2Folder>();
            sPk2EntryBlock   entryBlock = (sPk2EntryBlock)BufferToStruct(m_Blowfish.Decode(reader.ReadBytes(Marshal.SizeOf(typeof(sPk2EntryBlock)))), typeof(sPk2EntryBlock));

            for (int i = 0; i < 20; i++)
            {
                sPk2Entry entry = entryBlock.Entries[i];                 //.....
                switch (entry.Type)
                {
                case 0:                         //Null Entry

                    break;

                case 1:                         //Folder
                    if (entry.Name != "." && entry.Name != "..")
                    {
                        Pk2Folder folder = new Pk2Folder();
                        folder.Name     = entry.Name;
                        folder.Position = BitConverter.ToInt64(entry.g_Position, 0);
                        folders.Add(folder);
                        m_Folders[(RootPath + entry.Name).ToUpper()] = folder;
                        m_CurrentFolder.SubFolders.Add(folder);
                    }
                    break;

                case 2:                         //File
                    Pk2File file = new Pk2File();
                    file.Position     = entry.Position;
                    file.Name         = entry.Name;
                    file.Size         = entry.Size;
                    file.ParentFolder = m_CurrentFolder;
                    m_Files[(RootPath + entry.Name).ToUpper()] = file;
                    m_CurrentFolder.Files.Add(file);
                    break;
                }
            }
            if (entryBlock.Entries[19].NextChain != 0)
            {
                Read(entryBlock.Entries[19].NextChain, RootPath);
            }

            foreach (Pk2Folder folder in folders)
            {
                m_CurrentFolder = folder;
                if (folder.Files == null)
                {
                    folder.Files = new List <Pk2File>();
                }
                if (folder.SubFolders == null)
                {
                    folder.SubFolders = new List <Pk2Folder>();
                }
                Read(folder.Position, RootPath + folder.Name + "\\");
            }
        }
Exemplo n.º 2
0
 public void Dispose()
 {
     m_CurrentFolder = null;
     m_Files         = null;
     m_FileStream    = null;
     m_Folders       = null;
     Key             = null;
     ASCIIKey        = null;
     m_MainFolder    = null;
     Size            = 0;
 }
Exemplo n.º 3
0
        /// <summary>
        /// Check if a file exists in the Pk2 path specified.
        /// </summary>
        public bool FileExists(string FileName, string Path)
        {
            Pk2Folder folder = null;

            m_Folders.TryGetValue(Path, out folder);

            if (folder != null)
            {
                return(folder.Files.Exists(file => file.Name.Equals(FileName, StringComparison.OrdinalIgnoreCase)));
            }
            return(false);
        }
Exemplo n.º 4
0
        /// <summary>
        /// Get folders from Pk2 path specified.
        /// </summary>
        public List <Pk2Folder> GetSubFolders(string Path)
        {
            List <Pk2Folder> folders = new List <Pk2Folder>();

            Pk2Folder folder = null;

            m_Folders.TryGetValue(Path, out folder);

            if (folder != null)
            {
                folders.AddRange(folder.SubFolders);
            }
            return(folders);
        }
Exemplo n.º 5
0
        /// <summary>
        /// Get files from Pk2 path specified.
        /// </summary>
        public List <Pk2File> GetFiles(string Path)
        {
            List <Pk2File> files = new List <Pk2File>();

            Pk2Folder folder = null;

            m_Folders.TryGetValue(Path, out folder);

            if (folder != null)
            {
                files.AddRange(folder.Files);
            }
            return(files);
        }
Exemplo n.º 6
0
        public Pk2Folder GetFolder(string FolderPath)
        {
            if (FolderPath == "")
            {
                return(null);
            }

            // Normalize to the same dictionary key path format
            FolderPath = FolderPath.ToUpper();
            FolderPath = FolderPath.Replace("/", "\\");
            if (FolderPath.EndsWith("\\"))
            {
                FolderPath = FolderPath.Substring(0, FolderPath.Length - 1);
            }

            Pk2Folder folder = null;

            m_Folders.TryGetValue(FolderPath, out folder);
            return(folder);
        }
Exemplo n.º 7
0
        public Pk2Reader(string FilePath, string BlowfishKey)
        {
            if (!File.Exists(FilePath))
            {
                throw new Exception("File not found");
            }
            else
            {
                this.FullPath = Path.GetFullPath(FilePath);
                // Set default key for most clients
                if (BlowfishKey == "")
                {
                    ASCIIKey = "169841";
                }
                else
                {
                    ASCIIKey = BlowfishKey;
                }
                Key = GenerateFinalBlowfishKey(ASCIIKey);

                m_FileStream = new FileStream(FilePath, FileMode.Open, FileAccess.Read, FileShare.Read);
                Size         = m_FileStream.Length;

                m_Blowfish.Initialize(Key);
                BinaryReader reader = new BinaryReader(m_FileStream);
                Header = (sPk2Header)BufferToStruct(reader.ReadBytes(256), typeof(sPk2Header));

                m_CurrentFolder            = new Pk2Folder();
                m_CurrentFolder.Name       = FilePath;
                m_CurrentFolder.Files      = new List <Pk2File>();
                m_CurrentFolder.SubFolders = new List <Pk2Folder>();

                m_MainFolder = m_CurrentFolder;
                Read(reader.BaseStream.Position, "");
            }
        }