/*
 * Copyright (c) 2010 Sun Microsystems, Inc. All rights reserved.
 *
 * Sun Microsystems, Inc. has intellectual property rights relating to technology embodied in the product that is
 * described in this document. In particular, and without limitation, these intellectual property rights may include one
 * or more of the U.S. patents listed at http://www.sun.com/patents and one or more additional patents or pending patent
 * applications in the U.S. and in other countries.
 *
 * U.S. Government Rights - Commercial software. Government users are subject to the Sun Microsystems, Inc. standard
 * license agreement and applicable provisions of the FAR and its supplements.
 *
 * Use is subject to license terms. Sun, Sun Microsystems, the Sun logo, Java and Solaris are trademarks or registered
 * trademarks of Sun Microsystems, Inc. in the U.S. and other countries. All SPARC trademarks are used under license and
 * are trademarks or registered trademarks of SPARC International, Inc. in the U.S. and other countries.
 *
 * UNIX is a registered trademark in the U.S. and other countries, exclusively licensed through X/Open Company, Ltd.
 */
package com.sun.hotspot.c1x.server;

import java.io.*;
import java.lang.reflect.Proxy;
import java.net.*;
import java.rmi.registry.*;

import javax.net.*;

import com.sun.cri.ci.*;
import com.sun.cri.ci.CiDebugInfo.Frame;
import com.sun.cri.ri.*;
import com.sun.hotspot.c1x.*;
import com.sun.hotspot.c1x.Compiler;
import com.sun.hotspot.c1x.logging.*;

/**
 * Server side of the client/server compilation model.
 *
 * @author Lukas Stadler
 */
public class CompilationServer {

    private Registry registry;
    private ServerSocket serverSocket;
    private Socket socket;
    private ObjectOutputStream output;
    private ObjectInputStream input;

    public static void main(String[] args) throws Exception {
        new CompilationServer().run();
    }

    public static class Container implements Serializable {

        public final Class<?> clazz;
        public final Object[] values;

        public Container(Class<?> clazz, Object... values) {
            this.clazz = clazz;
            this.values = values;
        }
    }

    /**
     * Replaces certain cir objects that cannot easily be made Serializable.
     */
    public static class ReplacingOutputStream extends ObjectOutputStream {

        public ReplacingOutputStream(OutputStream out) throws IOException {
            super(out);
            enableReplaceObject(true);
        }

        @Override
        protected Object replaceObject(Object obj) throws IOException {
            Class<? extends Object> clazz = obj.getClass();
            if (clazz == CiConstant.class) {
                CiConstant o = (CiConstant) obj;
                return new Container(clazz, o.kind, o.boxedValue());
            } else if (clazz == CiDebugInfo.class) {
                CiDebugInfo o = (CiDebugInfo)obj;
                return new Container(clazz, o.codePos, o.frame, o.registerRefMap, o.frameRefMap);
            } else if(clazz == CiCodePos.class) {
                CiCodePos o = (CiCodePos)obj;
                return new Container(clazz, o.caller, o.method, o.bci);
            }
            return obj;
        }
    }

    /**
     * Replaces certain cir objects that cannot easily be made Serializable.
     */
    public static class ReplacingInputStream extends ObjectInputStream {

        public ReplacingInputStream(InputStream in) throws IOException {
            super(in);
            enableResolveObject(true);
        }

        @Override
        protected Object resolveObject(Object obj) throws IOException {
            if (obj instanceof Container) {
                Container c = (Container)obj;
                if( c.clazz == CiConstant.class) {
                    return CiConstant.forBoxed((CiKind) c.values[0], c.values[1]);
                } else if (c.clazz == CiDebugInfo.class) {
                    return new CiDebugInfo((CiCodePos)c.values[0], (Frame)c.values[1], (byte[])c.values[2], (byte[])c.values[3]);
                } else if (c.clazz == CiCodePos.class) {
                    return new CiCodePos((CiCodePos)c.values[0], (RiMethod)c.values[1], (Integer)c.values[2]);
                }
                throw new RuntimeException("unexpected container class");
            }
            return obj;
        }
    }

    private void run() throws IOException, ClassNotFoundException {
        serverSocket = ServerSocketFactory.getDefault().createServerSocket(1199);
        while (true) {
            try {
                Logger.log("Compilation server ready, waiting for client to connect...");
                socket = serverSocket.accept();
                Logger.log("Connected to " + socket.getRemoteSocketAddress());
                output = new ReplacingOutputStream(socket.getOutputStream());
                input = new ReplacingInputStream(socket.getInputStream());

                InvocationSocket invocation = new InvocationSocket(output, input);
                VMEntries entries = (VMEntries) Proxy.newProxyInstance(VMEntries.class.getClassLoader(), new Class<?>[] { VMEntries.class}, invocation);
                VMExits exits = Compiler.initializeServer(entries);
                invocation.setDelegate(exits);

                invocation.waitForResult();
            } catch (IOException e) {
                e.printStackTrace();
                socket.close();
            }
        }
    }
}