示例#1
0
        protected async Task InternalCreateIRD(string savePath, CancellationToken cancellation, string isoPath = null)
        {
            byte[] buffer = new byte[Utilities.SectorSize];

            List <FileHash> fileList = new List <FileHash>();

            try
            {
                // First, get the partition table of this iso
                using (Stream fs = Open())
                {
                    if (fs == null)
                    {
                        return;
                    }
                    PS3CDReader reader = new PS3CDReader(fs);
                    List <DirectoryMemberInformation> files = reader.Members.Where(d => d.IsFile).Distinct().ToList();
                    StartOfDataSector = files.First().StartSector;
                    EndOfDataSector   = files.Last().StartSector + files.Last().TotalSectors;

                    var updateFile = files.FirstOrDefault(
                        d => d.Path.Equals("/PS3_UPDATE/PS3UPDAT.PUP", StringComparison.OrdinalIgnoreCase));
                    long updateOffset = updateFile != null ? updateFile.StartSector * Utilities.SectorSize : 0;

                    fileList.AddRange(files.Select(d => new FileHash(d.StartSector, d.Length)));

                    if (cancellation.IsCancellationRequested)
                    {
                        return;
                    }

                    IrdFile irdFile = IrdFile.New();

                    using (Stream s = reader.OpenFile("PS3_GAME\\PARAM.SFO", FileMode.Open, FileAccess.Read))
                    {
                        ParamSfo sfo = ParamSfo.Load(s);
                        irdFile.AppVersion  = sfo.GetStringValue("APP_VER");
                        irdFile.GameVersion = sfo.GetStringValue("VERSION");
                        irdFile.GameID      = sfo.GetStringValue("TITLE_ID");
                        irdFile.GameName    = sfo.GetStringValue("TITLE");
                    }

                    Interaction.Instance.ReportMessage("Processing " + (string.IsNullOrEmpty(irdFile.GameName)
                                                                            ? irdFile.GameID
                                                                            : irdFile.GameName));

                    irdFile.UpdateVersion = Utilities.FindUpdateVersion(fs, updateOffset);

                    using (FileStream isoStream = string.IsNullOrEmpty(isoPath)
                                                      ? null
                                                      : new FileStream(isoPath, FileMode.Create, FileAccess.Write))
                    {
                        if (isoStream != null)
                        {
                            isoStream.SetLength(TotalSectors * Utilities.SectorSize);
                        }

                        using (IOStream ioStream = new IOStream(fs, isoStream))
                        {
                            ioStream.Seek(0, SeekOrigin.Begin);

                            // Read the header, until StartOfDataSector (in sectors)
                            irdFile.Header = new MemoryStream();
                            for (int i = 0; i < StartOfDataSector; i++)
                            {
                                int bytesRead = ioStream.Read(buffer, 0, buffer.Length);
                                irdFile.Header.Write(buffer, 0, bytesRead);
                                Interaction.Instance.ReportProgress(i);
                                if (cancellation.IsCancellationRequested)
                                {
                                    return;
                                }
                            }
                            irdFile.ExtractAuthData();

                            // Fix the regions for the actual interesting data
                            Regions.First().Start   = (uint)StartOfDataSector;
                            Regions.First().Length -= (uint)StartOfDataSector;
                            Regions.Last().Length   = (uint)EndOfDataSector - Regions.Last().Start;

                            // Now, we should calculate the md5 sums of all regions
                            Interaction.Instance.ReportMessage("Calculating hashes for " + Regions.Length + " regions.");
                            for (int i = 0; i < Regions.Length; i++)
                            {
                                // Calculate the hash
                                Interaction.Instance.ReportMessage(
                                    "Calculate hash for region " + i + " (" + Regions[i].Start.ToString("X2") + "-" +
                                    Regions[i].End.ToString("X2") + ")");

                                ioStream.Seek(Regions[i].Start * Utilities.SectorSize, SeekOrigin.Begin);

                                using (FileHashStream fhs = new FileHashStream(fileList, Regions[i].Start))
                                {
                                    await Regions[i].CopyRegion(cancellation, ioStream, fhs);
                                    if (cancellation.IsCancellationRequested)
                                    {
                                        return;
                                    }
                                    irdFile.RegionHashes.Add(Regions[i].SourceHash);
                                    Interaction.Instance.ReportMessage("Region " + i + " hash: " +
                                                                       Regions[i].SourceHash.AsString());
                                }
                            }

                            ioStream.Seek(EndOfDataSector * Utilities.SectorSize, SeekOrigin.Begin);
                            irdFile.Footer = new MemoryStream();
                            for (long i = EndOfDataSector; i < TotalSectors; i++)
                            {
                                int bytesRead = ioStream.Read(buffer, 0, buffer.Length);
                                irdFile.Footer.Write(buffer, 0, bytesRead);
                                Interaction.Instance.ReportProgress((int)(ioStream.Position / Utilities.SectorSize));
                                if (cancellation.IsCancellationRequested)
                                {
                                    return;
                                }
                            }
                        }
                    }
                    irdFile.FileHashes = fileList.ToDictionary(t => t.StartSector, t => t.Hash);
                    irdFile.Save(savePath);
                    Interaction.Instance.ReportMessage("All done, IRD file saved to " + savePath);
                }
            }
            catch (BadReadException e)
            {
                Interaction.Instance.ReportMessage(e.Message, ReportType.Fail);
            }
            catch (AuthenticationException e)
            {
                Interaction.Instance.ReportMessage(e.Message, ReportType.Fail);
            }
        }
示例#2
0
        private async void LoadIrd(string filename)
        {
            IrdFile irdFile = await TaskEx.Run(() => IrdFile.Load(filename));

            lblVersion.Text         = irdFile.Version.ToString(CultureInfo.InvariantCulture);
            lblUpdateVersion.Text   = irdFile.UpdateVersion;
            lblGameId.Text          = irdFile.GameID;
            lblGameName.Text        = irdFile.GameName;
            lblGameVersion.Text     = irdFile.GameVersion;
            lblAppVersion.Text      = irdFile.AppVersion;
            lblAmountOfRegions.Text = irdFile.RegionHashes.Count.ToString(CultureInfo.InvariantCulture);

            Region[] regions = await TaskEx.Run(() => irdFile.Header.GetRegions());

            for (int i = 0; i < irdFile.RegionHashes.Count && i < regions.Length; i++)
            {
                ListViewItem itm = new ListViewItem(i.ToString(CultureInfo.InvariantCulture));
                itm.SubItems.Add(regions[i].Start.ToString("X2"));
                itm.SubItems.Add(regions[i].End.ToString("X2"));
                itm.SubItems.Add(irdFile.RegionHashes[i].AsString());
                lvHashes.Items.Add(itm);
            }

            lblData1.Text = irdFile.Data1.AsString();
            lblData2.Text = irdFile.Data2.AsString();
            picHex.Stream = new MemoryStream(irdFile.PIC); // Shouldn't I dispose this one?
            d2            = irdFile.Data2;

            lblHeaderLength.Text =
                (irdFile.Header.Length / Utilities.SectorSize).ToString(CultureInfo.InvariantCulture);
            lblFooterLength.Text =
                (irdFile.Footer.Length / Utilities.SectorSize).ToString(CultureInfo.InvariantCulture);

            headerHex.Stream = irdFile.Header;
            footerHex.Stream = irdFile.Footer;

            lblCrc.Text     = BitConverter.GetBytes(irdFile.Crc).AsString();
            lblIRDHash.Text = IrdFile.GetHash(irdFile.FullPath);

            tabCtrl.Visible   = true;
            pbLoading.Visible = false;

            lvFiles.BeginUpdate();

            // Load files
            PS3CDReader cdReader = await TaskEx.Run(() => new PS3CDReader(irdFile.Header));

            foreach (var dmi in cdReader.Members.Where(d => d.IsFile).Distinct())
            {
                ListViewItem itm = new ListViewItem(dmi.Path);
                KeyValuePair <long, byte[]> h = irdFile.FileHashes.FirstOrDefault(f => f.Key == dmi.StartSector);
                itm.SubItems.Add(dmi.StartSector.ToString("X2"));
                itm.SubItems.Add(dmi.Length.ToString("X2"));
                itm.SubItems.Add(h.Value.AsString());
                lvFiles.Items.Add(itm);
            }

            lvFiles.EndUpdate();
            lblLoadingFiles.Visible = pbLoadingFiles.Visible = false;
            lvFiles.Visible         = true;
        }