com.replaymod.replaystudio.launcher.ReverseLauncher.java Source code

Introduction

Here is the source code for com.replaymod.replaystudio.launcher.ReverseLauncher.java

Source

/*
 * This file is part of ReplayStudio, licensed under the MIT License (MIT).
 *
 * Copyright (c) 2016 johni0702 <https://github.com/johni0702>
 * Copyright (c) contributors
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package com.replaymod.replaystudio.launcher;

import org.apache.commons.cli.CommandLine;

import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import static com.replaymod.replaystudio.util.Utils.readInt;

/**
 * Launcher which reverses the order of all packets in the replay supplied to it and saves the result to a raw
 * packet data file (tmcpr).
 */
public class ReverseLauncher {

    public void launch(CommandLine cmd) throws Exception {
        ZipFile file = new ZipFile(cmd.getArgs()[0]);
        ZipEntry entry = file.getEntry("recording.tmcpr");
        if (entry == null) {
            throw new IOException("Input file is not a valid replay file.");
        }
        long size = entry.getSize();
        if (size == -1) {
            throw new IOException("Uncompressed size of recording.tmcpr not set.");
        }

        InputStream from = file.getInputStream(entry);
        RandomAccessFile to = new RandomAccessFile(cmd.getArgs()[1], "rw");

        to.setLength(size);
        int nRead;
        long pos = size;
        byte[] buffer = new byte[8192];

        long lastUpdate = -1;
        while (true) {
            long pct = 100 - pos * 100 / size;
            if (lastUpdate != pct) {
                System.out.print("Reversing " + size + " bytes... " + pct + "%\r");
                lastUpdate = pct;
            }
            int next = readInt(from);
            int length = readInt(from);
            if (next == -1 || length == -1) {
                break; // reached end of stream
            }
            // Increase buffer if necessary
            if (length + 8 > buffer.length) {
                buffer = new byte[length + 8];
            }
            buffer[0] = (byte) ((next >>> 24) & 0xFF);
            buffer[1] = (byte) ((next >>> 16) & 0xFF);
            buffer[2] = (byte) ((next >>> 8) & 0xFF);
            buffer[3] = (byte) (next & 0xFF);
            buffer[4] = (byte) ((length >>> 24) & 0xFF);
            buffer[5] = (byte) ((length >>> 16) & 0xFF);
            buffer[6] = (byte) ((length >>> 8) & 0xFF);
            buffer[7] = (byte) (length & 0xFF);

            nRead = 0;
            while (nRead < length) {
                nRead += from.read(buffer, 8 + nRead, length - nRead);
            }

            pos -= length + 8;
            to.seek(pos);
            to.write(buffer, 0, length + 8);
        }

        from.close();
        to.close();

        System.out.println("\nDone!");
    }
}