objectstreamconnection.java

MegaMek is a networked Java clone of BattleTech, a turn-based sci-fi boardgame for 2+ players. Fight

/*
 * MegaMek - Copyright (C) 2005 Ben Mazur ([e-mail])
 *
 *  This program is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License as published by the Free
 *  Software Foundation; either version 2 of the License, or (at your option)
 *  any later version.
 *
 *  This program is distributed in the hope that it will be useful, but
 *  WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 *  or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 *  for more details.
 */

package megamek.common.net;

import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.net.Socket;

/**
 * Implementation of the <code>Connection</code> that uses the
 * <code>ObjectInputStream</code> and <code>ObjectOutputStream</code>
 * to send/receive data.
 *
 */
class ObjectStreamConnection extends Connection {

    /**
     * Input stream
     */
    private ObjectInputStream in;
    
    /**
     * Output stream
     */
    private ObjectOutputStream out;

    /**
     * Creates new server connection
     * @param socket
     * @param id
     */
    public ObjectStreamConnection(Socket socket, int id) {
        super(socket, id);
    }

    /**
     * Creates new Client connection
     * @param host
     * @param port
     * @param id
     */
    public ObjectStreamConnection(String host, int port, int id) {
        super(host, port, id);
    }

    protected INetworkPacket readNetworkPacket() throws Exception {
        NetworkPacket packet = null;
        if (in == null) {
            in = new ObjectInputStream(getInputStream());
        }

        packet = (NetworkPacket)in.readObject();
        return packet;
    }

    protected void sendNetworkPacket(byte[] data, boolean zipped) throws Exception {
        if (out == null) {
            out = new ObjectOutputStream(getOutputStream());
            out.flush();
        }

        out.reset(); // write each packet fresh
        out.writeObject(new NetworkPacket(zipped, marshallingType, data));
        out.flush();
    }
    private static class NetworkPacket  implements INetworkPacket, Serializable {
        
        /**
         * Is data compressed
         */
        private boolean compressed;
        
        /**
         * Data marshalling type
         */
        private int marshallingType;
        
        /**
         * Packet data
         */
        private byte[] data;
        
        /**
         * Creates new packet
         * @param compressed
         * @param marshallingType
         * @param data
         */
        NetworkPacket(boolean compressed, int marshallingType, byte[] data) {
            this.compressed = compressed;
            this.marshallingType = marshallingType;
            this.data = data;
        }
        
        public int getMarshallingType() {
            return marshallingType;
        }
        
        public byte[] getData() {
            return data;
        }
        
        public boolean isCompressed() {
            return compressed;
        }
    };    

}