/*
 * Java Network Programming, Second Edition
 * Merlin Hughes, Michael Shoffner, Derek Hamner
 * Manning Publications Company; ISBN 188477749X
 *
 * http://nitric.com/jnp/
 *
 * Copyright (c) 1997-1999 Merlin Hughes, Michael Shoffner, Derek Hamner;
 * all rights reserved; see license.txt for details.
 */

import java.io.*;
import java.net.*;

public class MyObjectOutputStream extends ObjectOutputStream implements MyOSConstants {
  public MyObjectOutputStream (OutputStream out) throws IOException {
    super (out);
    enableReplaceObject (true);
  }

  protected void writeStreamHeader () throws IOException {
    super.writeStreamHeader ();
    writeShort (myOSMagick);
    writeShort (myOSVersion);
  }

  protected Object replaceObject (Object object) throws IOException {
    if (object.getClass () == DatagramPacket.class)
      return new MyDatagramPacket ((DatagramPacket) object);
    else if (object.getClass ().getName ().equals ("java.io.File"))
      return new MyFile ((File) object);
    else
      return object;
  }
}