ObjectOutputStream.java 36.9 KB
Newer Older
Tom Tromey committed
1
/* ObjectOutputStream.java -- Class used to write serialized objects
2
   Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003 Free Software Foundation, Inc.
Tom Tromey committed
3 4 5 6 7 8 9

This file is part of GNU Classpath.

GNU Classpath 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, or (at your option)
any later version.
10

Tom Tromey committed
11 12 13 14 15 16 17 18 19 20
GNU Classpath 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.

You should have received a copy of the GNU General Public License
along with GNU Classpath; see the file COPYING.  If not, write to the
Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
02111-1307 USA.

21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
Linking this library statically or dynamically with other modules is
making a combined work based on this library.  Thus, the terms and
conditions of the GNU General Public License cover the whole
combination.

As a special exception, the copyright holders of this library give you
permission to link this library with independent modules to produce an
executable, regardless of the license terms of these independent
modules, and to copy and distribute the resulting executable under
terms of your choice, provided that you also meet, for each linked
independent module, the terms and conditions of the license of that
module.  An independent module is a module which is not derived from
or based on this library.  If you modify this library, you may extend
this exception to your version of the library, but you are not
obligated to do so.  If you do not wish to do so, delete this
exception statement from your version. */
Tom Tromey committed
37 38 39 40 41 42 43


package java.io;

import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
44
import java.lang.reflect.InvocationTargetException;
Tom Tromey committed
45 46 47 48
import java.util.Hashtable;

import gnu.java.io.ObjectIdentityWrapper;
import gnu.java.lang.reflect.TypeSignature;
49
import gnu.classpath.Configuration;
Tom Tromey committed
50 51

/**
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
 * An <code>ObjectOutputStream</code> can be used to write objects
 * as well as primitive data in a platform-independent manner to an
 * <code>OutputStream</code>.
 *
 * The data produced by an <code>ObjectOutputStream</code> can be read
 * and reconstituted by an <code>ObjectInputStream</code>.
 *
 * <code>writeObject (Object)</code> is used to write Objects, the
 * <code>write&lt;type&gt;</code> methods are used to write primitive
 * data (as in <code>DataOutputStream</code>). Strings can be written
 * as objects or as primitive data.
 *
 * Not all objects can be written out using an
 * <code>ObjectOutputStream</code>.  Only those objects that are an
 * instance of <code>java.io.Serializable</code> can be written.
 *
 * Using default serialization, information about the class of an
 * object is written, all of the non-transient, non-static fields of
 * the object are written, if any of these fields are objects, they are
 * written out in the same manner.
 *
 * An object is only written out the first time it is encountered.  If
 * the object is encountered later, a reference to it is written to
 * the underlying stream.  Thus writing circular object graphs
 * does not present a problem, nor are relationships between objects
 * in a graph lost.
 *
 * Example usage:
 * <pre>
 * Hashtable map = new Hashtable ();
 * map.put ("one", new Integer (1));
 * map.put ("two", new Integer (2));
 *
 * ObjectOutputStream oos =
 * new ObjectOutputStream (new FileOutputStream ("numbers"));
 * oos.writeObject (map);
 * oos.close ();
 *
 * ObjectInputStream ois =
 * new ObjectInputStream (new FileInputStream ("numbers"));
 * Hashtable newmap = (Hashtable)ois.readObject ();
 *
 * System.out.println (newmap);
 * </pre>
 *
 * The default serialization can be overriden in two ways.
 *
 * By defining a method <code>private void
 * writeObject (ObjectOutputStream)</code>, a class can dictate exactly
 * how information about itself is written.
 * <code>defaultWriteObject ()</code> may be called from this method to
 * carry out default serialization.  This method is not
 * responsible for dealing with fields of super-classes or subclasses.
 *
 * By implementing <code>java.io.Externalizable</code>.  This gives
 * the class complete control over the way it is written to the
 * stream.  If this approach is used the burden of writing superclass
 * and subclass data is transfered to the class implementing
 * <code>java.io.Externalizable</code>.
 *
 * @see java.io.DataOutputStream
 * @see java.io.Externalizable
 * @see java.io.ObjectInputStream
 * @see java.io.Serializable
 */
Tom Tromey committed
117 118 119 120
public class ObjectOutputStream extends OutputStream
  implements ObjectOutput, ObjectStreamConstants
{
  /**
121 122 123 124 125 126 127 128 129 130
   * Creates a new <code>ObjectOutputStream</code> that will do all of
   * its writing onto <code>out</code>.  This method also initializes
   * the stream by writing the header information (stream magic number
   * and stream version).
   *
   * @exception IOException Writing stream header to underlying
   * stream cannot be completed.
   *
   * @see #writeStreamHeader()
   */
Tom Tromey committed
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
  public ObjectOutputStream (OutputStream out) throws IOException
  {
    realOutput = new DataOutputStream (out);
    blockData = new byte[ BUFFER_SIZE ];
    blockDataCount = 0;
    blockDataOutput = new DataOutputStream (this);
    setBlockDataMode (true);
    replacementEnabled = false;
    isSerializing = false;
    nextOID = baseWireHandle;
    OIDLookupTable = new Hashtable ();
    protocolVersion = defaultProtocolVersion;
    useSubclassMethod = false;
    writeStreamHeader ();
  }

  /**
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
   * Writes a representation of <code>obj</code> to the underlying
   * output stream by writing out information about its class, then
   * writing out each of the objects non-transient, non-static
   * fields.  If any of these fields are other objects,
   * they are written out in the same manner.
   *
   * This method can be overriden by a class by implementing
   * <code>private void writeObject (ObjectOutputStream)</code>.
   *
   * If an exception is thrown from this method, the stream is left in
   * an undefined state.
   *
   * @exception NotSerializableException An attempt was made to
   * serialize an <code>Object</code> that is not serializable.
   *
   * @exception IOException Exception from underlying
   * <code>OutputStream</code>.
   */
Tom Tromey committed
166 167 168
  public final void writeObject (Object obj) throws IOException
  {
    if (useSubclassMethod)
169 170 171 172
      {
	writeObjectOverride (obj);
	return;
      }
Tom Tromey committed
173 174

    boolean was_serializing = isSerializing;
175
    boolean old_mode = setBlockDataMode (false);
Tom Tromey committed
176 177
    try
      {
178 179
	isSerializing = true;
	boolean replaceDone = false;
Tom Tromey committed
180
	Object replacedObject = null;
181 182 183 184 185 186 187 188
	
	while (true)
	  {
	    if (obj == null)
	      {
		realOutput.writeByte (TC_NULL);
		break;
	      }
189

190 191 192 193 194 195 196
	    Integer handle = findHandle (obj);
	    if (handle != null)
	      {
		realOutput.writeByte (TC_REFERENCE);
		realOutput.writeInt (handle.intValue ());
		break;
	      }
197

198 199 200
	    if (obj instanceof Class)
	      {
		Class cl = (Class)obj;
201
		ObjectStreamClass osc = ObjectStreamClass.lookupForClassObject (cl);
202 203 204 205
		assignNewHandle (obj);
		realOutput.writeByte (TC_CLASS);
		if (!osc.isProxyClass)
		  {
206
		    writeObject (osc);
207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
		  }
		else
		  {
		    realOutput.writeByte (TC_PROXYCLASSDESC);
		    Class[] intfs = cl.getInterfaces();
		    realOutput.writeInt(intfs.length);
		    for (int i = 0; i < intfs.length; i++)
		      realOutput.writeUTF(intfs[i].getName());
		    
		    boolean oldmode = setBlockDataMode (true);
		    annotateProxyClass(cl);
		    setBlockDataMode (oldmode);
		    realOutput.writeByte(TC_ENDBLOCKDATA);
		    
		    writeObject (osc.getSuper());
		  }
		break;
	      }
225

226 227 228 229 230 231 232
	    if (obj instanceof ObjectStreamClass)
	      {
		ObjectStreamClass osc = (ObjectStreamClass)obj;
		realOutput.writeByte (TC_CLASSDESC);
		realOutput.writeUTF (osc.getName ());
		realOutput.writeLong (osc.getSerialVersionUID ());
		assignNewHandle (obj);
233

234
		int flags = osc.getFlags ();
235

236 237 238
		if (protocolVersion == PROTOCOL_VERSION_2
		    && osc.isExternalizable ())
		  flags |= SC_BLOCK_DATA;
239

240
		realOutput.writeByte (flags);
241

242 243
		ObjectStreamField[] fields = osc.fields;
		realOutput.writeShort (fields.length);
244

245 246 247 248 249 250
		ObjectStreamField field;
		for (int i=0; i < fields.length; i++)
		  {
		    field = fields[i];
		    realOutput.writeByte (field.getTypeCode ());
		    realOutput.writeUTF (field.getName ());
251

252 253 254
		    if (! field.isPrimitive ())
		      writeObject (field.getTypeString ());
		  }
255

256 257 258 259
		boolean oldmode = setBlockDataMode (true);
		annotateClass (osc.forClass ());
		setBlockDataMode (oldmode);
		realOutput.writeByte (TC_ENDBLOCKDATA);
260

261 262 263 264 265 266
		if (osc.isSerializable ())
		  writeObject (osc.getSuper ());
		else
		  writeObject (null);
		break;
	      }
267

268 269
	    if ((replacementEnabled || obj instanceof Serializable)
		&& ! replaceDone)
270
	      {
271
		replacedObject = obj;
272

273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
		if (obj instanceof Serializable)
		  {
		    Method m = null;
		    try
		      {
			Class classArgs[] = {};
			m = obj.getClass ().getDeclaredMethod ("writeReplace",
							       classArgs);
			// m can't be null by definition since an exception would
			// have been thrown so a check for null is not needed.
			obj = m.invoke (obj, new Object[] {});
		      }
		    catch (NoSuchMethodException ignore)
		      {
		      }
		    catch (IllegalAccessException ignore)
		      {
		      }
		    catch (InvocationTargetException ignore)
		      {
		      }
		  }
295

296 297
		if (replacementEnabled)
		  obj = replaceObject (obj);
298

299 300
		replaceDone = true;
		continue;
301
	      }
302

303
	    if (obj instanceof String)
304
	      {
305 306 307 308
		realOutput.writeByte (TC_STRING);
		assignNewHandle (obj);
		realOutput.writeUTF ((String)obj);
		break;
309
	      }
310

311 312 313 314
	    Class clazz = obj.getClass ();
	    ObjectStreamClass osc = ObjectStreamClass.lookupForClassObject (clazz);
	    if (osc == null)
	      throw new NotSerializableException (clazz.getName ());
315

316
	    if (clazz.isArray ())
317
	      {
318 319 320 321 322
		realOutput.writeByte (TC_ARRAY);
		writeObject (osc);
		assignNewHandle (obj);
		writeArraySizeAndElements (obj, clazz.getComponentType ());
		break;
323
	      }
324

325 326
	    realOutput.writeByte (TC_OBJECT);
	    writeObject (osc);
327

328 329 330 331
	    if (replaceDone)
	      assignNewHandle (replacedObject);
	    else
	      assignNewHandle (obj);
332

333
	    if (obj instanceof Externalizable)
334
	      {
335 336
		if (protocolVersion == PROTOCOL_VERSION_2)
		  setBlockDataMode (true);
337

338
		((Externalizable)obj).writeExternal (this);
339

340 341 342 343 344
		if (protocolVersion == PROTOCOL_VERSION_2)
		  {
		    setBlockDataMode (false);
		    realOutput.writeByte (TC_ENDBLOCKDATA);
		  }
345

346
		break;
347
	      }
348

349 350 351 352 353
	    if (obj instanceof Serializable)
	      {
		currentObject = obj;
		ObjectStreamClass[] hierarchy =
		  ObjectStreamClass.getObjectStreamClasses (clazz);
354

355 356 357
		for (int i=0; i < hierarchy.length; i++)
		  {
		    currentObjectStreamClass = hierarchy[i];
358

359
		    fieldsAlreadyWritten = false;
360 361 362 363 364 365 366 367 368
		    if (currentObjectStreamClass.hasWriteMethod ())
		      {
			setBlockDataMode (true);
			callWriteMethod (obj, currentObjectStreamClass);
			setBlockDataMode (false);
			realOutput.writeByte (TC_ENDBLOCKDATA);
		      }
		    else
		      writeFields (obj, currentObjectStreamClass);
369
		  }
370

371 372 373 374 375
		currentObject = null;
		currentObjectStreamClass = null;
		currentPutField = null;
		break;
	      }
376

377 378 379 380 381 382 383 384 385 386 387 388 389 390 391
	    throw new NotSerializableException (clazz.getName ());
	  } // end pseudo-loop
      }
    catch (ObjectStreamException ose)
      {
	// Rethrow these are fatal.
	throw ose;
      }
    catch (IOException e)
      {
	realOutput.writeByte (TC_EXCEPTION);
	reset (true);

	setBlockDataMode (false);
	try
Tom Tromey committed
392
	  {
393
	    writeObject (e);
Tom Tromey committed
394
	  }
395
	catch (IOException ioe)
Tom Tromey committed
396
	  {
397
	    throw new StreamCorruptedException ("Exception " + ioe + " thrown while exception was being written to stream.");
Tom Tromey committed
398
	  }
399

400
	reset (true);
Tom Tromey committed
401 402
      }
    finally
403 404 405 406
      {
	isSerializing = was_serializing;
	setBlockDataMode (old_mode);
      }
Tom Tromey committed
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428
  }

  /**
     Writes the current objects non-transient, non-static fields from
     the current class to the underlying output stream.

     This method is intended to be called from within a object's
     <code>private void writeObject (ObjectOutputStream)</code>
     method.

     @exception NotActiveException This method was called from a
     context other than from the current object's and current class's
     <code>private void writeObject (ObjectOutputStream)</code>
     method.

     @exception IOException Exception from underlying
     <code>OutputStream</code>.
  */
  public void defaultWriteObject ()
    throws IOException, NotActiveException
  {
    markFieldsWritten ();
429
    writeFields (currentObject, currentObjectStreamClass);
Tom Tromey committed
430 431 432 433 434 435 436 437 438
  }


  private void markFieldsWritten () throws IOException
  {
    if (currentObject == null || currentObjectStreamClass == null)
      throw new NotActiveException ("defaultWriteObject called by non-active class and/or object");

    if (fieldsAlreadyWritten)
439
      throw new IOException ("Only one of putFields and defaultWriteObject may be called, and it may only be called once");
Tom Tromey committed
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465

    fieldsAlreadyWritten = true;
  }


  /**
     Resets stream to state equivalent to the state just after it was
     constructed.

     Causes all objects previously written to the stream to be
     forgotten.  A notification of this reset is also written to the
     underlying stream.

     @exception IOException Exception from underlying
     <code>OutputStream</code> or reset called while serialization is
     in progress.
  */
  public void reset () throws IOException
  {
    reset (false);
  }


  private void reset (boolean internal) throws IOException
  {
    if (!internal)
466 467 468
      {
	if (isSerializing)
	  throw new IOException ("Reset called while serialization in progress");
Tom Tromey committed
469

470 471
	realOutput.writeByte (TC_RESET);
      }
Tom Tromey committed
472 473 474 475 476 477 478 479 480 481

    clearHandles ();
  }


  /**
     Informs this <code>ObjectOutputStream</code> to write data
     according to the specified protocol.  There are currently two
     different protocols, specified by <code>PROTOCOL_VERSION_1</code>
     and <code>PROTOCOL_VERSION_2</code>.  This implementation writes
482 483
     data using <code>PROTOCOL_VERSION_2</code> by default, as is done
     by the JDK 1.2.
Tom Tromey committed
484 485 486 487 488 489 490 491 492 493 494

     A non-portable method, <code>setDefaultProtocolVersion (int
     version)</code> is provided to change the default protocol
     version.

     For an explination of the differences beween the two protocols
     see XXX: the Java ObjectSerialization Specification.

     @exception IOException if <code>version</code> is not a valid
     protocol

495
     @see #setDefaultProtocolVersion(int)
Tom Tromey committed
496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517
  */
  public void useProtocolVersion (int version) throws IOException
  {
    if (version != PROTOCOL_VERSION_1 && version != PROTOCOL_VERSION_2)
      throw new IOException ("Invalid protocol version requested.");

    protocolVersion = version;
  }


  /**
     <em>GNU $classpath specific</em>

     Changes the default stream protocol used by all
     <code>ObjectOutputStream</code>s.  There are currently two
     different protocols, specified by <code>PROTOCOL_VERSION_1</code>
     and <code>PROTOCOL_VERSION_2</code>.  The default default is
     <code>PROTOCOL_VERSION_1</code>.

     @exception IOException if <code>version</code> is not a valid
     protocol

518
     @see #useProtocolVersion(int)
Tom Tromey committed
519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538
  */
  public static void setDefaultProtocolVersion (int version)
    throws IOException
  {
    if (version != PROTOCOL_VERSION_1 && version != PROTOCOL_VERSION_2)
      throw new IOException ("Invalid protocol version requested.");

    defaultProtocolVersion = version;
  }


  /**
     An empty hook that allows subclasses to write extra information
     about classes to the stream.  This method is called the first
     time each class is seen, and after all of the standard
     information about the class has been written.

     @exception IOException Exception from underlying
     <code>OutputStream</code>.

539
     @see ObjectInputStream#resolveClass(java.io.ObjectStreamClass)
Tom Tromey committed
540 541 542 543
  */
  protected void annotateClass (Class cl) throws IOException
  {}

544 545
  protected void annotateProxyClass(Class cl) throws IOException
  {}
Tom Tromey committed
546 547 548 549 550 551 552 553 554 555 556 557 558

  /**
     Allows subclasses to replace objects that are written to the
     stream with other objects to be written in their place.  This
     method is called the first time each object is encountered
     (modulo reseting of the stream).

     This method must be enabled before it will be called in the
     serialization process.

     @exception IOException Exception from underlying
     <code>OutputStream</code>.

559
     @see #enableReplaceObject(boolean)
Tom Tromey committed
560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578
  */
  protected Object replaceObject (Object obj) throws IOException
  {
    return obj;
  }


  /**
     If <code>enable</code> is <code>true</code> and this object is
     trusted, then <code>replaceObject (Object)</code> will be called
     in subsequent calls to <code>writeObject (Object)</code>.
     Otherwise, <code>replaceObject (Object)</code> will not be called.

     @exception SecurityException This class is not trusted.
  */
  protected boolean enableReplaceObject (boolean enable)
    throws SecurityException
  {
    if (enable)
579 580 581 582 583
      {
	SecurityManager sm = System.getSecurityManager ();
	if (sm != null)
	  sm.checkPermission (new SerializablePermission ("enableSubstitution"));
      }
Tom Tromey committed
584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606

    boolean old_val = replacementEnabled;
    replacementEnabled = enable;
    return old_val;
  }


  /**
     Writes stream magic and stream version information to the
     underlying stream.

     @exception IOException Exception from underlying
     <code>OutputStream</code>.
  */
  protected void writeStreamHeader () throws IOException
  {
    realOutput.writeShort (STREAM_MAGIC);
    realOutput.writeShort (STREAM_VERSION);
  }



  /**
607 608 609 610 611 612 613 614 615 616
   * Protected constructor that allows subclasses to override
   * serialization.  This constructor should be called by subclasses
   * that wish to override <code>writeObject (Object)</code>.  This
   * method does a security check <i>NOTE: currently not
   * implemented</i>, then sets a flag that informs
   * <code>writeObject (Object)</code> to call the subclasses
   * <code>writeObjectOverride (Object)</code> method.
   *
   * @see #writeObjectOverride(Object)
   */
Tom Tromey committed
617 618 619 620 621 622 623 624 625 626
  protected ObjectOutputStream () throws IOException, SecurityException
  {
    SecurityManager sec_man = System.getSecurityManager ();
    if (sec_man != null)
      sec_man.checkPermission (SUBCLASS_IMPLEMENTATION_PERMISSION);
    useSubclassMethod = true;
  }


  /**
627 628 629 630 631 632 633 634 635 636 637
   * This method allows subclasses to override the default
   * serialization mechanism provided by
   * <code>ObjectOutputStream</code>.  To make this method be used for
   * writing objects, subclasses must invoke the 0-argument
   * constructor on this class from there constructor.
   *
   * @see #ObjectOutputStream()
   *
   * @exception NotActiveException Subclass has arranged for this
   * method to be called, but did not implement this method.
   */
Tom Tromey committed
638 639 640 641 642 643 644 645
  protected void writeObjectOverride (Object obj) throws NotActiveException,
    IOException
  {
    throw new NotActiveException ("Subclass of ObjectOutputStream must implement writeObjectOverride");
  }


  /**
646 647
   * @see DataOutputStream#write(int)
   */
Tom Tromey committed
648 649 650
  public void write (int data) throws IOException
  {
    if (writeDataAsBlocks)
651 652 653
      {
	if (blockDataCount == BUFFER_SIZE)
	  drain ();
Tom Tromey committed
654

655 656
	blockData[ blockDataCount++ ] = (byte)data;
      }
Tom Tromey committed
657 658 659 660 661 662
    else
      realOutput.write (data);
  }


  /**
663 664
   * @see DataOutputStream#write(byte[])
   */
665
  public void write (byte[] b) throws IOException
Tom Tromey committed
666 667 668 669 670 671
  {
    write (b, 0, b.length);
  }


  /**
672 673
   * @see DataOutputStream#write(byte[],int,int)
   */
674
  public void write (byte[] b, int off, int len) throws IOException
Tom Tromey committed
675 676 677
  {
    if (writeDataAsBlocks)
      {
678 679 680 681 682 683 684 685 686 687 688 689 690 691
	if (len < 0)
	  throw new IndexOutOfBoundsException ();

	if (blockDataCount + len < BUFFER_SIZE)
	  {
	    System.arraycopy (b, off, blockData, blockDataCount, len);
	    blockDataCount += len;
	  }
	else
	  {
	    drain ();
	    writeBlockDataHeader (len);
	    realOutput.write (b, off, len);
	  }
Tom Tromey committed
692 693 694 695 696 697 698
      }
    else
      realOutput.write (b, off, len);
  }


  /**
699 700
   * @see DataOutputStream#flush()
   */
Tom Tromey committed
701 702 703 704 705 706 707 708
  public void flush () throws IOException
  {
    drain ();
    realOutput.flush ();
  }


  /**
709 710 711 712 713 714
   * Causes the block-data buffer to be written to the underlying
   * stream, but does not flush underlying stream.
   *
   * @exception IOException Exception from underlying
   * <code>OutputStream</code>.
   */
Tom Tromey committed
715 716 717 718 719
  protected void drain () throws IOException
  {
    if (blockDataCount == 0)
      return;

720 721
    if (writeDataAsBlocks)
      writeBlockDataHeader (blockDataCount);
Tom Tromey committed
722 723 724 725 726 727
    realOutput.write (blockData, 0, blockDataCount);
    blockDataCount = 0;
  }


  /**
728 729
   * @see java.io.DataOutputStream#close ()
   */
Tom Tromey committed
730 731
  public void close () throws IOException
  {
732
    flush ();
Tom Tromey committed
733 734 735 736 737
    realOutput.close ();
  }


  /**
738 739
   * @see java.io.DataOutputStream#writeBoolean (boolean)
   */
Tom Tromey committed
740 741
  public void writeBoolean (boolean data) throws IOException
  {
742
    blockDataOutput.writeBoolean (data);
Tom Tromey committed
743 744 745 746
  }


  /**
747 748
   * @see java.io.DataOutputStream#writeByte (int)
   */
Tom Tromey committed
749 750
  public void writeByte (int data) throws IOException
  {
751
    blockDataOutput.writeByte (data);
Tom Tromey committed
752 753 754 755
  }


  /**
756 757
   * @see java.io.DataOutputStream#writeShort (int)
   */
Tom Tromey committed
758 759
  public void writeShort (int data) throws IOException
  {
760
    blockDataOutput.writeShort (data);
Tom Tromey committed
761 762 763 764
  }


  /**
765 766
   * @see java.io.DataOutputStream#writeChar (int)
   */
Tom Tromey committed
767 768
  public void writeChar (int data) throws IOException
  {
769
    blockDataOutput.writeChar (data);
Tom Tromey committed
770 771 772 773
  }


  /**
774 775
   * @see java.io.DataOutputStream#writeInt (int)
   */
Tom Tromey committed
776 777
  public void writeInt (int data) throws IOException
  {
778
    blockDataOutput.writeInt (data);
Tom Tromey committed
779 780 781 782
  }


  /**
783 784
   * @see java.io.DataOutputStream#writeLong (long)
   */
Tom Tromey committed
785 786
  public void writeLong (long data) throws IOException
  {
787
    blockDataOutput.writeLong (data);
Tom Tromey committed
788 789 790 791
  }


  /**
792 793
   * @see java.io.DataOutputStream#writeFloat (float)
   */
Tom Tromey committed
794 795
  public void writeFloat (float data) throws IOException
  {
796
    blockDataOutput.writeFloat (data);
Tom Tromey committed
797 798 799 800
  }


  /**
801 802
   * @see java.io.DataOutputStream#writeDouble (double)
   */
Tom Tromey committed
803 804
  public void writeDouble (double data) throws IOException
  {
805
    blockDataOutput.writeDouble (data);
Tom Tromey committed
806 807 808 809
  }


  /**
810 811
   * @see java.io.DataOutputStream#writeBytes (java.lang.String)
   */
Tom Tromey committed
812 813
  public void writeBytes (String data) throws IOException
  {
814
    blockDataOutput.writeBytes (data);
Tom Tromey committed
815 816 817 818
  }


  /**
819 820
   * @see java.io.DataOutputStream#writeChars (java.lang.String)
   */
Tom Tromey committed
821 822 823 824 825 826 827
  public void writeChars (String data) throws IOException
  {
    dataOutput.writeChars (data);
  }


  /**
828 829
   * @see java.io.DataOutputStream#writeUTF (java.lang.String)
   */
Tom Tromey committed
830 831 832 833 834 835 836
  public void writeUTF (String data) throws IOException
  {
    dataOutput.writeUTF (data);
  }


  /**
837 838 839 840 841
   * This class allows a class to specify exactly which fields should
   * be written, and what values should be written for these fields.
   *
   * XXX: finish up comments
   */
Tom Tromey committed
842 843
  public static abstract class PutField
  {
844 845 846 847 848 849 850 851 852 853 854 855 856
    public abstract void put (String name, boolean value);
    public abstract void put (String name, byte value);
    public abstract void put (String name, char value);
    public abstract void put (String name, double value);
    public abstract void put (String name, float value);
    public abstract void put (String name, int value);
    public abstract void put (String name, long value);
    public abstract void put (String name, short value);
    public abstract void put (String name, Object value);

    /**
     * @deprecated
     */
Tom Tromey committed
857 858 859 860 861 862 863 864 865 866
    public abstract void write (ObjectOutput out) throws IOException;
  }

  public PutField putFields () throws IOException
  {
    markFieldsWritten ();

    currentPutField = new PutField ()
      {
	private byte[] prim_field_data
867
	= new byte[currentObjectStreamClass.primFieldSize];
Tom Tromey committed
868
	private Object[] objs
869
	= new Object[currentObjectStreamClass.objectFieldCount];
Tom Tromey committed
870 871

	public void put (String name, boolean value)
872 873 874 875 876 877
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'Z');
	  prim_field_data[field.getOffset ()] = (byte)(value ? 1 : 0);
	}
Tom Tromey committed
878 879

	public void put (String name, byte value)
880 881 882 883 884 885
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'B');
	  prim_field_data[field.getOffset ()] = value;
	}
Tom Tromey committed
886 887

	public void put (String name, char value)
888 889 890 891 892 893 894 895
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'C');
	  int off = field.getOffset ();
	  prim_field_data[off++] = (byte)(value >>> 8);
	  prim_field_data[off] = (byte)value;
	}
Tom Tromey committed
896 897

	public void put (String name, double value)
898 899 900 901 902 903 904 905 906 907 908 909 910 911 912
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'D');
	  int off = field.getOffset ();
	  long l_value = Double.doubleToLongBits (value);
	  prim_field_data[off++] = (byte)(l_value >>> 52);
	  prim_field_data[off++] = (byte)(l_value >>> 48);
	  prim_field_data[off++] = (byte)(l_value >>> 40);
	  prim_field_data[off++] = (byte)(l_value >>> 32);
	  prim_field_data[off++] = (byte)(l_value >>> 24);
	  prim_field_data[off++] = (byte)(l_value >>> 16);
	  prim_field_data[off++] = (byte)(l_value >>> 8);
	  prim_field_data[off] = (byte)l_value;
	}
Tom Tromey committed
913 914

	public void put (String name, float value)
915 916 917 918 919 920 921 922 923 924 925
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'F');
	  int off = field.getOffset ();
	  int i_value = Float.floatToIntBits (value);
	  prim_field_data[off++] = (byte)(i_value >>> 24);
	  prim_field_data[off++] = (byte)(i_value >>> 16);
	  prim_field_data[off++] = (byte)(i_value >>> 8);
	  prim_field_data[off] = (byte)i_value;
	}
Tom Tromey committed
926 927

	public void put (String name, int value)
928 929 930 931 932 933 934 935 936 937
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'I');
	  int off = field.getOffset ();
	  prim_field_data[off++] = (byte)(value >>> 24);
	  prim_field_data[off++] = (byte)(value >>> 16);
	  prim_field_data[off++] = (byte)(value >>> 8);
	  prim_field_data[off] = (byte)value;
	}
Tom Tromey committed
938 939

	public void put (String name, long value)
940 941 942 943 944 945 946 947 948 949 950 951 952 953
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'J');
	  int off = field.getOffset ();
	  prim_field_data[off++] = (byte)(value >>> 52);
	  prim_field_data[off++] = (byte)(value >>> 48);
	  prim_field_data[off++] = (byte)(value >>> 40);
	  prim_field_data[off++] = (byte)(value >>> 32);
	  prim_field_data[off++] = (byte)(value >>> 24);
	  prim_field_data[off++] = (byte)(value >>> 16);
	  prim_field_data[off++] = (byte)(value >>> 8);
	  prim_field_data[off] = (byte)value;
	}
Tom Tromey committed
954 955

	public void put (String name, short value)
956 957 958 959 960 961 962 963
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  checkType (field, 'S');
	  int off = field.getOffset ();
	  prim_field_data[off++] = (byte)(value >>> 8);
	  prim_field_data[off] = (byte)value;
	}
Tom Tromey committed
964 965

	public void put (String name, Object value)
966 967 968 969 970 971 972 973 974 975
	{
	  ObjectStreamField field
	    = currentObjectStreamClass.getField (name);
	  if (field == null)
	    throw new IllegalArgumentException ();
	  if (value != null &&
	      ! field.getType ().isAssignableFrom (value.getClass ()))
	    throw new IllegalArgumentException ();
	  objs[field.getOffset ()] = value;
	}
Tom Tromey committed
976 977

	public void write (ObjectOutput out) throws IOException
978 979 980 981 982 983 984 985 986 987
	{
	  // Apparently Block data is not used with PutField as per
	  // empirical evidence against JDK 1.2.  Also see Mauve test
	  // java.io.ObjectInputOutput.Test.GetPutField.
	  boolean oldmode = setBlockDataMode (false);
	  out.write (prim_field_data);
	  for (int i = 0; i < objs.length; ++ i)
	    out.writeObject (objs[i]);
	  setBlockDataMode (oldmode);
	}
Tom Tromey committed
988 989 990

	private void checkType (ObjectStreamField field, char type)
	  throws IllegalArgumentException
991 992 993 994
	{
	  if (TypeSignature.getEncodingOfClass (field.getType ()).charAt (0) != type)
	    throw new IllegalArgumentException ();
	}
Tom Tromey committed
995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
      };
    // end PutFieldImpl

    return currentPutField;
  }


  public void writeFields () throws IOException
  {
    if (currentPutField == null)
      throw new NotActiveException ("writeFields can only be called after putFields has been called");

    currentPutField.write (this);
  }


  // write out the block-data buffer, picking the correct header
  // depending on the size of the buffer
  private void writeBlockDataHeader (int size) throws IOException
  {
    if (size < 256)
1016 1017 1018 1019
      {
	realOutput.writeByte (TC_BLOCKDATA);
	realOutput.write (size);
      }
Tom Tromey committed
1020
    else
1021 1022 1023 1024
      {
	realOutput.writeByte (TC_BLOCKDATALONG);
	realOutput.writeInt (size);
      }
Tom Tromey committed
1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060
  }


  // lookup the handle for OBJ, return null if OBJ doesn't have a
  // handle yet
  private Integer findHandle (Object obj)
  {
    return (Integer)OIDLookupTable.get (new ObjectIdentityWrapper (obj));
  }


  // assigns the next availible handle to OBJ
  private int assignNewHandle (Object obj)
  {
    OIDLookupTable.put (new ObjectIdentityWrapper (obj),
			new Integer (nextOID));
    return nextOID++;
  }


  // resets mapping from objects to handles
  private void clearHandles ()
  {
    nextOID = baseWireHandle;
    OIDLookupTable.clear ();
  }


  // write out array size followed by each element of the array
  private void writeArraySizeAndElements (Object array, Class clazz)
    throws IOException
  {
    int length = Array.getLength (array);

    if (clazz.isPrimitive ())
      {
1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123
	if (clazz == Boolean.TYPE)
	  {
	    boolean[] cast_array = (boolean[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeBoolean (cast_array[i]);
	    return;
	  }
	if (clazz == Byte.TYPE)
	  {
	    byte[] cast_array = (byte[])array;
	    realOutput.writeInt (length);
	    realOutput.write(cast_array, 0, length);
	    return;
	  }
	if (clazz == Character.TYPE)
	  {
	    char[] cast_array = (char[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeChar (cast_array[i]);
	    return;
	  }
	if (clazz == Double.TYPE)
	  {
	    double[] cast_array = (double[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeDouble (cast_array[i]);
	    return;
	  }
	if (clazz == Float.TYPE)
	  {
	    float[] cast_array = (float[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeFloat (cast_array[i]);
	    return;
	  }
	if (clazz == Integer.TYPE)
	  {
	    int[] cast_array = (int[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeInt (cast_array[i]);
	    return;
	  }
	if (clazz == Long.TYPE)
	  {
	    long[] cast_array = (long[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeLong (cast_array[i]);
	    return;
	  }
	if (clazz == Short.TYPE)
	  {
	    short[] cast_array = (short[])array;
	    realOutput.writeInt (length);
	    for (int i=0; i < length; i++)
	      realOutput.writeShort (cast_array[i]);
	    return;
	  }
Tom Tromey committed
1124
      }
1125
    else
Tom Tromey committed
1126
      {
1127
	Object[] cast_array = (Object[])array;
Tom Tromey committed
1128 1129
	realOutput.writeInt (length);
	for (int i=0; i < length; i++)
1130
	  writeObject (cast_array[i]);
Tom Tromey committed
1131 1132 1133 1134
      }
  }


1135 1136 1137 1138
  // writes out FIELDS of OBJECT for the specified ObjectStreamClass.
  // FIELDS are already in canonical order.
  private void writeFields (Object obj, ObjectStreamClass osc)
    throws IOException
Tom Tromey committed
1139
  {
1140
    ObjectStreamField[] fields = osc.fields;
1141
    boolean oldmode = setBlockDataMode (false);
Tom Tromey committed
1142 1143 1144
    String field_name;
    Class type;
    for (int i=0; i < fields.length; i++)
1145 1146 1147 1148 1149
      {
	field_name = fields[i].getName ();
	type = fields[i].getType ();

	if (type == Boolean.TYPE)
1150
	  realOutput.writeBoolean (getBooleanField (obj, osc.forClass(), field_name));
1151
	else if (type == Byte.TYPE)
1152
	  realOutput.writeByte (getByteField (obj, osc.forClass(), field_name));
1153
	else if (type == Character.TYPE)
1154
	  realOutput.writeChar (getCharField (obj, osc.forClass(), field_name));
1155
	else if (type == Double.TYPE)
1156
	  realOutput.writeDouble (getDoubleField (obj, osc.forClass(), field_name));
1157
	else if (type == Float.TYPE)
1158
	  realOutput.writeFloat (getFloatField (obj, osc.forClass(), field_name));
1159
	else if (type == Integer.TYPE)
1160
	  realOutput.writeInt (getIntField (obj, osc.forClass(), field_name));
1161
	else if (type == Long.TYPE)
1162
	  realOutput.writeLong (getLongField (obj, osc.forClass(), field_name));
1163
	else if (type == Short.TYPE)
1164
	  realOutput.writeShort (getShortField (obj, osc.forClass(), field_name));
1165
	else
1166
	  writeObject (getObjectField (obj, osc.forClass(), field_name,
1167 1168 1169
				       fields[i].getTypeString ()));
      }
    setBlockDataMode (oldmode);
Tom Tromey committed
1170 1171 1172 1173
  }


  // Toggles writing primitive data to block-data buffer.
1174
  private boolean setBlockDataMode (boolean on) throws IOException
Tom Tromey committed
1175
  {
1176 1177 1178 1179 1180
    if (on == writeDataAsBlocks)
      return on;

    drain();
    boolean oldmode = writeDataAsBlocks;
Tom Tromey committed
1181 1182 1183 1184 1185 1186
    writeDataAsBlocks = on;

    if (on)
      dataOutput = blockDataOutput;
    else
      dataOutput = realOutput;
1187 1188

    return oldmode;
Tom Tromey committed
1189 1190 1191
  }


1192
  private void callWriteMethod (Object obj, ObjectStreamClass osc) throws IOException
Tom Tromey committed
1193
  {
1194
    Class klass = osc.forClass();
Tom Tromey committed
1195 1196
    try
      {
1197
	Class classArgs[] = {ObjectOutputStream.class};
Tom Tromey committed
1198 1199 1200 1201 1202 1203
	Method m = getMethod (klass, "writeObject", classArgs);
	if (m == null)
	  return;
	Object args[] = {this};
	m.invoke (obj, args);	
      }
1204
    catch (InvocationTargetException x)
Tom Tromey committed
1205
      {
1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219
        /* Rethrow if possible. */
	Throwable exception = x.getTargetException();
	if (exception instanceof RuntimeException)
	  throw (RuntimeException) exception;
	if (exception instanceof IOException)
	  throw (IOException) exception;

	throw new IOException ("Exception thrown from writeObject() on " +
			       klass + ": " + exception.getClass().getName());
      }
    catch (Exception x)
      {
	throw new IOException ("Failure invoking writeObject() on " +
			       klass + ": " + x.getClass().getName());
Tom Tromey committed
1220 1221
      }
  }
1222

1223
  private boolean getBooleanField (Object obj, Class klass, String field_name)
1224
    throws IOException
Tom Tromey committed
1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
  {
    try
      {
	Field f = getField (klass, field_name);
	boolean b = f.getBoolean (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1238
  private byte getByteField (Object obj, Class klass, String field_name) throws IOException
Tom Tromey committed
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
  {
    try
      {
	Field f = getField (klass, field_name);
	byte b = f.getByte (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1252
  private char getCharField (Object obj, Class klass, String field_name) throws IOException
Tom Tromey committed
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
  {
    try
      {
	Field f = getField (klass, field_name);
	char b = f.getChar (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1266
  private double getDoubleField (Object obj, Class klass, String field_name)
1267
    throws IOException
Tom Tromey committed
1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280
  {
    try
      {
	Field f = getField (klass, field_name);
	double b = f.getDouble (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1281
  private float getFloatField (Object obj, Class klass, String field_name)
1282
    throws IOException
Tom Tromey committed
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295
  {
    try
      {
	Field f = getField (klass, field_name);
	float b = f.getFloat (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1296
  private int getIntField (Object obj, Class klass, String field_name) throws IOException
Tom Tromey committed
1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309
  {
    try
      {
	Field f = getField (klass, field_name);
	int b = f.getInt (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1310
  private long getLongField (Object obj, Class klass, String field_name) throws IOException
Tom Tromey committed
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323
  {
    try
      {
	Field f = getField (klass, field_name);
	long b = f.getLong (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1324
  private short getShortField (Object obj, Class klass, String field_name)
1325
    throws IOException
Tom Tromey committed
1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338
  {
    try
      {
	Field f = getField (klass, field_name);
	short b = f.getShort (obj);
	return b;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1339
  private Object getObjectField (Object obj, Class klass, String field_name,
Tom Tromey committed
1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354
				 String type_code) throws IOException
  {
    try
      {
	Field f = getField (klass, field_name);
	Object o = f.get (obj);
	// FIXME: We should check the type_code here
	return o;
      }
    catch (Exception _)
      {
	throw new IOException ();
      }    
  }

1355 1356 1357 1358 1359
  private static Field getField (Class klass, String name)
    throws java.lang.NoSuchFieldException
  {
    return klass.getDeclaredField(name);
  }
1360

1361 1362 1363 1364 1365
  private static Method getMethod (Class klass, String name, Class[] args)
    throws java.lang.NoSuchMethodException
  {
    return klass.getDeclaredMethod(name, args);
  }
1366

Tom Tromey committed
1367 1368 1369
  // this value comes from 1.2 spec, but is used in 1.1 as well
  private final static int BUFFER_SIZE = 1024;

1370
  private static int defaultProtocolVersion = PROTOCOL_VERSION_2;
Tom Tromey committed
1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387

  private DataOutputStream dataOutput;
  private boolean writeDataAsBlocks;
  private DataOutputStream realOutput;
  private DataOutputStream blockDataOutput;
  private byte[] blockData;
  private int blockDataCount;
  private Object currentObject;
  private ObjectStreamClass currentObjectStreamClass;
  private PutField currentPutField;
  private boolean fieldsAlreadyWritten;
  private boolean replacementEnabled;
  private boolean isSerializing;
  private int nextOID;
  private Hashtable OIDLookupTable;
  private int protocolVersion;
  private boolean useSubclassMethod;
1388 1389 1390 1391 1392 1393 1394 1395

  static
  {
    if (Configuration.INIT_LOAD_LIBRARY)
      {
        System.loadLibrary ("javaio");
      }
  }
Tom Tromey committed
1396
}