aswi2020horky-gitlab/demo_mh/Deserializer/src/deserialize/Grammar.java @ d7ea8c04

package deserialize;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import org.json.simple.JSONObject;

/**
 * https://docs.oracle.com/javase/7/docs/platform/serialization/spec/protocol.html
 */
public class Grammar {
	
	private final static short STREAM_MAGIC = (short) 0xACED;
	private final static short STREAM_VERSION = 5;
	private final static byte TC_NULL = 0x70;
	private final static byte TC_REFERENCE = 0x71;
	private final static byte TC_CLASSDESC = 0x72;
	private final static byte TC_OBJECT = 0x73;
	private final static byte TC_STRING = 0x74;
	private final static byte TC_ARRAY = 0x75;
	private final static byte TC_CLASS = 0x76;
	private final static byte TC_BLOCKDATA = 0x77;
	private final static byte TC_ENDBLOCKDATA = 0x78;
	private final static byte TC_RESET = 0x79;
	private final static byte TC_BLOCKDATALONG = 0x7A;
	private final static byte TC_EXCEPTION = 0x7B;
	private final static byte TC_LONGSTRING = 0x7C;
	private final static byte TC_PROXYCLASSDESC = 0x7D;
	private final static byte TC_ENUM = 0x7E;
	private final static int baseWireHandle = (int) 0x7E0000;
	
	final static byte SC_WRITE_METHOD = 0x01; // if SC_SERIALIZABLE
	final static byte SC_BLOCK_DATA = 0x08; // if SC_EXTERNALIZABLE
	final static byte SC_SERIALIZABLE = 0x02;
	final static byte SC_EXTERNALIZABLE = 0x04;
	final static byte SC_ENUM = 0x10;

	private ByteReader reader;
	
	public Grammar(byte buffer[]) {
		reader = new ByteReader(buffer);
	}
	
	
	List<ClassDescription> classes = new ArrayList<ClassDescription>();
	ClassDescription actClass = null;
	
	public JSONObject process() {
		System.out.println("Kontrola hlavicky souboru: " + (reader.readShort() == STREAM_MAGIC && reader.readShort() == STREAM_VERSION));
		
		while (!reader.end()) {
			try {
				object();
			} catch (Exception e) {
				e.printStackTrace();
				break;
			}
		}
		
		return classes.get(0).getJSONObject();
		// System.out.println(classes.get(0).toString());
	}
	
	// FOR VALUES : 
	// The size and types are described by the
    // classDesc for the current object
	public Object object() {
		Object retValue;
		byte b = reader.readByte();
		switch (b) {
			case TC_OBJECT: {
					System.out.println("X1");
					
					// Note. TC_OBJECT; classDesc(); classdata()[]; // data for each class
					// Note. classDescFlags = (byte)
					
					ClassDescription desc = (ClassDescription) object();
					retValue = desc;
					
					/*byte flags = readByte();
					System.out.println("DEF: " + String.format("%02X", flags));
					boolean xxx = SC_SERIALIZABLE == desc2.flags && SC_WRITE_METHOD != flags;
					boolean xxx2 = SC_SERIALIZABLE == desc2.flags && SC_WRITE_METHOD == flags;
					boolean xxx3 = SC_EXTERNALIZABLE == desc2.flags && SC_BLOCK_DATA != flags;
					boolean xxx4 = SC_EXTERNALIZABLE == desc2.flags && SC_BLOCK_DATA == flags;
					System.out.println(xxx);
					System.out.println(xxx2);
					System.out.println(xxx3);
					System.out.println(xxx4);*/
					
					/* TODO
					classDesc(); // && (
							
							//values() // fields in order of class descriptor    // SC_SERIALIZABLE & classDescFlag &&
																				// !(SC_WRITE_METHOD & classDescFlags)
							// OR
							//values() fields in order of class descriptor; object();  // SC_SERIALIZABLE & classDescFlag &&
																	// SC_WRITE_METHOD & classDescFlags
							// OR
							//object();          // SC_EXTERNALIZABLE & classDescFlag &&
														// !(SC_BLOCKDATA  & classDescFlags
							// OR
							//object();          // SC_EXTERNALIZABLE & classDescFlag&&
														// SC_BLOCKDATA & classDescFlags
							// );*/
				}
				break;
			case TC_CLASS: {
					System.out.println("X2");
					ClassDescription desc = (ClassDescription) object();
					retValue = desc;
				}
				break;
			case TC_ARRAY: {
					System.out.println("X3");
					ClassDescription desc = (ClassDescription) object();
					// TODO int size = readInt();
					retValue = desc; // TODO + "[" + size + "]";
				}
				break;
			case TC_STRING: {
					System.out.println("X4");
					String str = reader.readUtf();
					retValue = str;
				}
				break;
			case TC_LONGSTRING: {
					System.out.println("X5");
					String str = reader.readLongUtf();
					retValue = str;
				}
				break;
			case TC_ENUM: {
					System.out.println("X6");
					ClassDescription desc = (ClassDescription) object();
					desc.isClass = false;
					retValue = desc;
					System.out.println("enumConstName: " + (String) object());
				}
				break;
				
				
			case TC_CLASSDESC: {
					System.out.println("X7");
					ClassDescription desc = new ClassDescription();
					desc.className = reader.readUtf();
					desc.serialVersionUID = reader.readLong();
					desc.flags = reader.readByte();
					desc.isClass = true;
					desc.setVariables(fields());
					
					if (actClass == null) {
						actClass = desc;
					}
					// TODO
					actClass.addParent(desc);
					
					// TODO object(null); classDesc();
					// TODO System.out.println(desc.toString());
					retValue = desc;
				}
				break;
			case TC_PROXYCLASSDESC: {
					System.out.println("X8");
					int count = reader.readInt();
					String name = reader.readUtf();
					retValue = name + "[" + +count + "]";
					// TODO object(null); classDesc();
					
				}
				break;
			case TC_NULL: {
					// TODO
					System.out.println("---NULL---");
					retValue = null;
					
					classes.add(actClass);
					actClass = null;
					while (true) {
						Variable actVar = classes.get(classes.size() - 1).getActVariable();
						if (actVar == null) {
							break;
						}
						
						switch (actVar.getType()) {
						case 'B': actVar.setValue(reader.readByte()); break;
						case 'C': actVar.setValue((char) reader.readByte()); break;
						case 'D': actVar.setValue(reader.readDouble()); break;
						case 'F': actVar.setValue(reader.readFloat()); break;
						case 'I': actVar.setValue(reader.readInt()); break;
						case 'J': actVar.setValue(reader.readLong()); break;
						case 'S': actVar.setValue(reader.readShort()); break;
						case 'Z': actVar.setValue(reader.readBoolean()); break;
						case '[':
							ClassDescription desccc = (ClassDescription) object();
							
							object(); // END BLOK
							object(); // NULL
							
							int count = reader.readInt();
							System.out.println("------------- VELIKOST POLE = " + count);
							for (int i = 0; i < count; i++) {
								System.out.println("------------- HODNOTA = " + reader.readChar());
							}
							//actVar.setValue(readByte());
							break;
						case 'L': actVar.setValue(object()); break;
						}
					}
				}
				break;
			case TC_REFERENCE: {
					System.out.println("---REFERENCE---");
					// TODO (ClassDesc) TC_REFERENCE (int)handle ?!?
					retValue = null;
				}
				break;
				
				
			case TC_EXCEPTION: {
					// TODO (Throwable)object(null);
					retValue = "(Throwable)";
				}
				break;
			case TC_RESET: {
					// TODO
					System.out.println("---RESET---");
					retValue = null;
				}
				break;
			case TC_BLOCKDATA: {
					System.out.println("X13");
					char size = reader.readUnsignedByte();
					// (unsigned byte)<size> (byte)[size];
					retValue = "byte [" + (int) size + "]";
				}
				break;
			case TC_BLOCKDATALONG: {
					System.out.println("X14");
					int size = reader.readInt();
					// (int)<size> (byte)[size];
					retValue = "byte [" + size + "]";
				}
				break;
			case TC_ENDBLOCKDATA: {
					// TODO
					System.out.println("---ENDBLOCKDATA---");
					retValue = null;
				}
				break;
			default:
				System.out.println("DEF: " + String.format("%02X", b) + " " + (char) b);
				retValue = null;
		}
		
		return retValue;
	}
	
	public List<Variable> fields() {
		List<Variable> variables = new ArrayList<Variable>();
		
		short count = reader.readShort();
		for (int i = 0; i < count; i++) {
			char type = (char) reader.readByte();
			String variable = reader.readUtf();
			String typeName = null;
			if (!DataType.isPrimaryType(type))
				typeName = (String) object();
			variables.add(new Variable(type, variable, typeName));
		}
		
		return variables;
	}
	
}