Ejemplo n.º 1
0
        static void Main(string[] args)
        {
            Directory.SetCurrentDirectory(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location));
            mainDirectory = Environment.CurrentDirectory + "/";

            pathList    = new string[3];
            pathList[0] = mainDirectory;
            pathList[1] = mainDirectory + "asm/work/";
            pathList[2] = mainDirectory + "asm/";             // this is for compatibility with old patches.

            if (!Asar.init())
            {
                Console.WriteLine("Could not initialize or find asar.dll");
                Console.WriteLine("Please redownload the program.");
                Pause();
                return;
            }

            if (args.Length == 0 || args.Length > 2)
            {
                Console.WriteLine("Usage: UberASMTool [code list] [ROM]");
                Console.WriteLine("If ROM is not specified, UberASM Tool will search for the one in the code list.");
                Console.WriteLine("If code list is not specified, UberASM Tool will try loading 'list.txt'.");
                Console.WriteLine();

                if (args.Length > 2)
                {
                    Pause();
                    return;
                }
            }

            UberConfigParser parser = new UberConfigParser();

            if (args.Length == 2)
            {
                parser.OverrideROM = args[1];

                try
                {
                    parser.LoadListFile(args[0]);
                }
                catch (Exception ex)
                {
                    Console.WriteLine("Can't read {0}: {1}.", args[0], ex.Message);
                    Pause();
                    return;
                }
            }
            else if (args.Length == 1)
            {
                parser.OverrideROM = null;

                try
                {
                    parser.LoadListFile(args[0]);
                }
                catch (Exception ex)
                {
                    Console.WriteLine("Can't read {0}: {1}.", args[0], ex.Message);
                    Pause();
                    return;
                }
            }
            else if (File.Exists("list.txt"))
            {
                parser.LoadListFile("list.txt");
            }
            else
            {
                Console.WriteLine("list.txt not found.");
                Pause();
                return;
            }

            Console.WriteLine();

            if (!parser.ParseList())
            {
                Console.WriteLine("Could not parse list file!");
                Console.WriteLine(parser.GetLogs());
                Pause();
                return;
            }

            config = parser.Build();

            if (config.ROMPath == null)
            {
                Console.WriteLine("ROM file not given.");
                Pause();
                return;
            }

            try
            {
                rom = new ROM(config.ROMPath);
                rom.Init();
                SNES.DetectAddressing(rom.romType & 255);
            }
            catch (Exception ex)
            {
                Console.WriteLine("Error: {0}", ex.Message);
                Pause();
                return;
            }

            CleanPreviousRun();

            if (error)
            {
                Pause();
                return;
            }

            BuildLibrary();

            if (error)
            {
                Pause();
                return;
            }

            GenerateLibrary();

            BuildAsm("overworld", 7, 1, "OW_init_table", "OW_asm_table", "OW_nmi_table", null);
            BuildAsm("gamemode", 256, 2, "gamemode_init_table", "gamemode_main_table", "gamemode_nmi_table", null);
            BuildAsm("level", 512, 0, "level_init_table", "level_asm_table", "level_nmi_table", "level_load_table");
            BuildOther();

            if (error)
            {
                Pause();
                return;
            }

            if (config.VerboseMode)
            {
                Console.WriteLine("Total files insert size: {0} (0x{0:X4}) bytes", totalInsertSize);
            }

            if (Asar.patch(mainDirectory + "asm/work/main.asm", ref rom.romData, pathList))
            {
                PrintWarningsAndErrors("");

                var  prints  = Asar.getprints();
                bool printed = false;

                for (int i = 0; i < prints.Length; ++i)
                {
                    if (i + 1 != prints.Length)
                    {
                        Console.WriteLine(prints[i]);
                    }
                    else if (int.TryParse(prints[i], out int insertSize) && config.VerboseMode)
                    {
                        Console.WriteLine("Main patch insert size: {0} (0x{0:X4}) bytes", insertSize);
                        Console.WriteLine();
                        Console.WriteLine("Total: {0} (0x{0:X4}) bytes", insertSize + totalInsertSize);
                        Console.WriteLine();
                        printed = true;
                    }
                }

                if (config.VerboseMode && !printed)
                {
                    Console.WriteLine();
                }

                Console.WriteLine("Codes inserted successfully.");
                rom.Save();

                WriteRestoreComment();
            }
            else
            {
                PrintWarningsAndErrors("");

                Console.WriteLine("Some errors occured while applying main patch. Process aborted.");
                Console.WriteLine("Your ROM wasn't modified.");
            }

            Pause();
        }
Ejemplo n.º 2
0
        private static bool PatchSingleFile(string data, string baseFolder, string originalFile, string originFolder,
                                            bool library, out int startPc, out int endPc)
        {
            startPc = -1;
            endPc   = -1;
            string realFile    = mainDirectory + baseFolder + "__temp.asm";
            string compileFile = originalFile;

            if (!FileUtils.ForceCreate(realFile, GenerateBasefile(data, library, realFile, originFolder)))
            {
                Console.WriteLine($"  Error: access denied while creating temporary file {realFile}.");
                error = true;
                return(false);
            }

            bool status = Asar.patch(realFile, ref rom.romData, pathList);

            if (!FileUtils.ForceDelete(realFile))
            {
                Console.WriteLine($"  Warning: could not delete temporary file {realFile}.");
            }

            PrintWarningsAndErrors(compileFile);

            if (!status)
            {
                Console.WriteLine();
                Console.WriteLine("Some errors occured while running UberASM tool. Process aborted.");
                Console.WriteLine("Your ROM wasn't modified.");
                error = true;
                return(false);
            }

            // process prints. prot requests, internal things, tec.
            string[] prints = Asar.getprints();
            bool     endl   = false;
            bool     startl = false;

            foreach (string print in prints)
            {
                string[] split   = print.Trim().Split(new[] { " " }, StringSplitOptions.RemoveEmptyEntries);
                string   command = split.Length > 0 ? split[0].ToLower() : "";
                string   value   = split.Length > 1 ? string.Join(" ", split, 1, split.Length - 1) : "";

                if (endl)
                {
                    Console.WriteLine("  {0}: error: unexpected print after end of the file print mark.", originalFile);
                    error = true;
                    return(false);
                }

                if (!startl && command != "_startl")
                {
                    Console.WriteLine("  {0}: error: unexpected print before STARTL command.", originalFile);
                    error = true;
                    return(false);
                }

                switch (command)
                {
                case "_prot":
                    try {
                        int ptr = Convert.ToInt32(value, 16);
                        if (!protPointerList.Contains(ptr))
                        {
                            protPointerList.Add(ptr);
                        }
                        break;
                    } catch {
                        Console.WriteLine("  {0}: error: invalid PRINT PROT value.", originalFile);
                        error = true;
                        return(false);
                    }

                case "_startl":
                    try {
                        startPc = Convert.ToInt32(value, 16);
                        startl  = true;
                        break;
                    } catch {
                        Console.WriteLine("  {0}: error: invalid PRINT STARTL value.", originalFile);
                        error = true;
                        return(false);
                    }

                case "_endl":
                    try {
                        endPc = Convert.ToInt32(value, 16);
                        endl  = true;
                        break;
                    } catch {
                        Console.WriteLine("  {0}: error: invalid PRINT ENDL value.", originalFile);
                        error = true;
                        return(false);
                    }

                default:
                    Console.WriteLine(print);
                    break;
                }
            }

            if (startPc == endPc)
            {
                Console.WriteLine("  {0}: error: empty assembled file.", originalFile);
                error = true;
                return(false);
            }

            return(true);
        }
Ejemplo n.º 3
0
        /// <summary>
        /// Picks a ROM through a <see cref="SaveFileDialog"/> if not specified (or forced) and patch multi_midway.asm.
        /// </summary>
        private void PatchRom(bool saveAs = false)
        {
            try
            {
                // Similar as in ExportAsmTable(): Compress the tables first.
                CompressTable();

                // Now we save the midway points with a fixed name.
                Midway.ExportAsm(midways, new FileStream(directory + MmpAsm, FileMode.Create));
                UnsavedChanges = false;
            }
            catch
            {
                MessageBox.Show("Cannot create multi_midway_table.asm. Please check the file's permission or whether it's already in use.", "Cannot create " + MmpAsm, MessageBoxButtons.OK, MessageBoxIcon.Error);
                return;
            }

            if (Asar.init())
            {
                Asar.reset();
                if (string.IsNullOrEmpty(romName) || saveAs)
                {
                    if (patchRomDialog.ShowDialog() != DialogResult.OK)
                    {
                        return;
                    }
                    romName = patchRomDialog.FileName;
                }

                try
                {
                    // Remember that Asar expects headerless ROMs.
                    // The code seperates the raw data from the header.
                    byte[] fullRom = File.ReadAllBytes(patchRomDialog.FileName);

                    int    lHeader = fullRom.Length & 0x7fff;
                    byte[] header  = new byte[lHeader];
                    byte[] rom     = new byte[fullRom.Length - lHeader];
                    Array.Copy(fullRom, 0, header, 0, lHeader);
                    Array.Copy(fullRom, lHeader, rom, 0, fullRom.Length - lHeader);

                    // Patching starts...
                    if (Asar.patch(directory + "multi_midway.asm", ref rom))
                    {
                        try
                        {
                            // Now it's time to merge them back.
                            fullRom = new byte[rom.Length + lHeader];
                            Array.Copy(header, 0, fullRom, 0, lHeader);
                            Array.Copy(rom, 0, fullRom, lHeader, rom.Length);

                            File.WriteAllBytes(patchRomDialog.FileName, fullRom);
                        }
                        catch (IOException ex)
                        {
                            MessageBox.Show("An error appeared when patching Multiple Midway Points: " + ex.Message, "ROM couldn't be written", MessageBoxButtons.OK, MessageBoxIcon.Error);
                            Asar.close();
                            return;
                        }
                        StringBuilder warningBuilder = new StringBuilder();
                        foreach (var warning in Asar.getwarnings())
                        {
                            warningBuilder.AppendLine(warning.Fullerrdata);
                        }
                        string warnings    = warningBuilder.ToString();
                        string fullWarning = !string.IsNullOrEmpty(warnings) ? "The following warnings appeared:\n" + warnings : "";
                        using (LongMessage message = new LongMessage("Multiple Midway Points has been inserted successfully. It uses " + Asar.getprints()[0] + " bytes of freespace.\n" + fullWarning, "Patching successful"))
                        {
                            message.ShowDialog(this);
                        }
                    }
                    else
                    {
                        MessageBox.Show("An error appeared when patching Multiple Midway Points. See mmp.log for more information.", "ROM couldn't be patched", MessageBoxButtons.OK, MessageBoxIcon.Error);
                        using (FileStream stream = new FileStream("mmp.log", FileMode.Create))
                        {
                            using (StreamWriter writer = new StreamWriter(stream))
                            {
                                foreach (var warning in Asar.getwarnings())
                                {
                                    writer.WriteLine(warning.Fullerrdata);
                                }
                                foreach (var error in Asar.geterrors())
                                {
                                    writer.WriteLine(error.Fullerrdata);
                                }
                            }
                        }
                    }
                }
                catch (IOException ex)
                {
                    MessageBox.Show("An error appeared when patching Multiple Midway Points: " + ex.Message, "ROM couldn't be opened", MessageBoxButtons.OK, MessageBoxIcon.Error);
                    return;
                }
                finally
                {
                    Asar.close();
                }
            }
            else
            {
                MessageBox.Show("Asar couldn't be started. (Perhaps asar.dll is missing or a wrong version is used?)", "Couldn't open Asar", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }