Mercurial > hg > Applications > TightVNC_orginal
view src/main/java/jp/ac/u_ryukyu/treevnc/server/VncProxyService.java @ 40:3c072f2f39bb
add Screen change function.
author | one |
---|---|
date | Mon, 22 Oct 2012 17:40:32 +0900 |
parents | c2f0b6907448 |
children | fe38611c4b2b |
line wrap: on
line source
package jp.ac.u_ryukyu.treevnc.server; import java.awt.event.*; import java.io.*; import java.net.Socket; import java.net.UnknownHostException; import java.util.LinkedList; import java.util.logging.Logger; import jp.ac.u_ryukyu.treevnc.AcceptThread; import jp.ac.u_ryukyu.treevnc.server.state.ChangeHost; import com.glavsoft.exceptions.AuthenticationFailedException; import com.glavsoft.exceptions.FatalException; import com.glavsoft.exceptions.TransportException; import com.glavsoft.exceptions.UnsupportedProtocolVersionException; import com.glavsoft.exceptions.UnsupportedSecurityTypeException; import com.glavsoft.rfb.IChangeSettingsListener; import com.glavsoft.rfb.IRfbSessionListener; import com.glavsoft.rfb.protocol.Protocol; import com.glavsoft.rfb.protocol.ProtocolSettings; import com.glavsoft.transport.Reader; import com.glavsoft.transport.Writer; import com.glavsoft.viewer.ConnectionManager; import com.glavsoft.viewer.Viewer; import com.glavsoft.viewer.cli.Parser; import com.glavsoft.viewer.swing.ClipboardControllerImpl; import com.glavsoft.viewer.swing.ParametersHandler; import com.glavsoft.viewer.swing.Surface; import com.glavsoft.viewer.swing.UiSettings; import com.glavsoft.viewer.swing.ParametersHandler.ConnectionParams; public class VncProxyService extends Viewer implements Runnable, IRfbSessionListener, IChangeSettingsListener { /** * */ private static final long serialVersionUID = 1L; public MyRfbProtoProxy rfb; private ConnectionManager connectionManager; private VncProxyService vps; private CreateThread createThread; public AcceptClient aClient; public int opendPort; // public AcceptClient acc; public VncProxyService(Parser parser) { this(); aClient = new AcceptClient(); ParametersHandler.completeSettingsFromCLI(parser, connectionParams, settings, uiSettings); showControls = ParametersHandler.showControls; passwordFromParams = parser.getValueFor(ParametersHandler.ARG_PASSWORD); logger.info("TightVNC Viewer version " + ver()); isApplet = false; } public VncProxyService() { connectionManager = new ConnectionManager(this, isApplet); connectionParams = new ParametersHandler.ConnectionParams(); settings = ProtocolSettings.getDefaultSettings(); uiSettings = super.uiSettings; connectionParams.hostName="localhost"; } boolean isClient = false; public static final int DEFAULT_PORT = 5900; public static Logger logger = Logger.getLogger("com.glavsoft"); private boolean forceReconnection; private String reconnectionReason; private ConnectionParams connectionParams; private final ProtocolSettings settings; private final UiSettings uiSettings; private AcceptThread acceptThread; private GetBroadCastProxy getCast; public static void main(String[] argv) { String[] mainArgs = argv; System.out.println(mainArgs.length); // input into arguments Decision Parser parser = new Parser(); ParametersHandler.completeParserOptions(parser); parser.parse(argv); if (parser.isSet(ParametersHandler.ARG_HELP)) { printUsage(parser.optionsUsage()); System.exit(0); } VncProxyService vps = new VncProxyService(parser); if (mainArgs.length != 0) { for (int i = 0; i < argv.length; i++) { vps.setArguments(mainArgs); // write later } } else { //getHostData(); } //SwingUtilities.invokeLater(myClient); /* Thread th = new Thread(vps); th.start(); try { th.join(); } catch (InterruptedException e) { e.printStackTrace(); } */ vps.run(); vps.createConnectionAndStart(); } private void setArguments(String[] mainArgs) { } @Override public void run() { rfb = new MyRfbProtoProxy(); // getHostData(); if (forceReconnection) { connectionManager.showReconnectDialog("Connection lost", reconnectionReason); forceReconnection = false; } tryAgain = true; while (tryAgain) { // connectionParams.hostName = arguments[0]; // workingSocket = connectionManager.connectToTreeHost(connectionParams, settings); workingSocket = connectionManager.connectToHost(connectionParams, settings, rfb); if (null == workingSocket) { closeApp(); break; } logger.info("Connected"); try { workingSocket.setTcpNoDelay(true); // disable Nagle algorithm Reader reader = new Reader(workingSocket.getInputStream()); Writer writer = new Writer(workingSocket.getOutputStream()); // rfb.setStream(reader,writer); workingProtocol = new Protocol(reader, writer, new PasswordChooser(passwordFromParams, connectionParams, containerFrame, this), settings); workingProtocol.handshake(); rfb.setProtocolContext(workingProtocol); threadSetAndStart(); // input into change parents ClipboardControllerImpl clipboardController = new ClipboardControllerImpl(workingProtocol, settings.getRemoteCharsetName()); clipboardController.setEnabled(settings.isAllowClipboardTransfer()); settings.addListener(clipboardController); surface = new Surface(workingProtocol, this, uiSettings.getScaleFactor()); // this method settings.addListener(this); uiSettings.addListener(surface); containerFrame = createContainer(); connectionManager.setContainerFrame(containerFrame); updateFrameTitle(); containerFrame.dispose(); containerFrame = null; workingProtocol.startNormalHandling(this, surface, clipboardController,rfb); // rfb.setInitData(workingProtocol.getInitData()); too early // workingProtocol.startNormalHandling(this, surface, clipboardController); tryAgain = false; } catch (UnsupportedProtocolVersionException e) { connectionManager.showReconnectDialog("Unsupported Protocol Version", e.getMessage()); logger.severe(e.getMessage()); } catch (UnsupportedSecurityTypeException e) { connectionManager.showReconnectDialog("Unsupported Security Type", e.getMessage()); logger.severe(e.getMessage()); } catch (AuthenticationFailedException e) { passwordFromParams = null; connectionManager.showReconnectDialog("Authentication Failed", e.getMessage()); logger.severe(e.getMessage()); } catch (TransportException e) { connectionManager.showReconnectDialog("Connection Error", "Connection Error" + ": " + e.getMessage()); logger.severe(e.getMessage()); } catch (IOException e) { connectionManager.showReconnectDialog("Connection Error", "Connection Error" + ": " + e.getMessage()); logger.severe(e.getMessage()); } catch (FatalException e) { connectionManager.showReconnectDialog("Connection Error", "Connection Error" + ": " + e.getMessage()); logger.severe(e.getMessage()); } } } public void createConnectionAndStart() { Thread thread; opendPort = rfb.selectPort(5999); acceptThread = new AcceptThread(rfb, opendPort); thread = new Thread(acceptThread); thread.start(); getCast = new GetBroadCastProxy(this, workingProtocol.getRemoteDesktopName(), this.connectionParams.hostName); thread = new Thread(getCast); thread.start(); } public void start_threads() { //rfbThread.start(); //rfb.requestThreadStart(); } public void proxyStart(String[] argv) { String[] mainArgs = argv; System.out.println(mainArgs.length); // input into arguments Decision Parser parser = new Parser(); ParametersHandler.completeParserOptions(parser); parser.parse(argv); if (parser.isSet(ParametersHandler.ARG_HELP)) { printUsage(parser.optionsUsage()); System.exit(0); } vps = new VncProxyService(parser); if (mainArgs.length != 0) { for (int i = 0; i < argv.length; i++) { vps.setArguments(mainArgs); // write later } } else { //getHostData(); } //SwingUtilities.invokeLater(myClient); Thread th = new Thread(vps); th.start(); try { th.join(); } catch (InterruptedException e) { e.printStackTrace(); } vps.createConnectionAndStart(); } @Override public void destroy() { } @Override public void windowClosing(WindowEvent e) { if (e != null && e.getComponent() != null) { e.getWindow().setVisible(false); } closeApp(); } private void threadSetAndStart() throws UnknownHostException, IOException { //aClient = new AcceptClient(); createThread = new CreateThread(aClient); Thread thread = new Thread(createThread); thread.start(); thread = new Thread(new ChangeHost(this, forceReconnection)); thread.start(); // createSocketAndSend(aClient.getList(),String.valueOf(port)); } public void setConnectionManager(ConnectionManager _connectionManager) { connectionManager = _connectionManager; } protected void socketClose() { try { workingSocket.close(); acceptThread.flag=true; getCast.setStopFlag(true); //createThread.setStopFlag(true); } catch (IOException e) {/*nop*/} } @Override public void rfbSessionStopped(final String reason) { if (workingSocket != null && workingSocket.isConnected()) { try { workingSocket.close(); } catch (IOException e) { /*nop*/ } } } private void createSocketAndSend(LinkedList<String> clientList,String port) throws UnknownHostException, IOException { int i = 0; for(String client : clientList) { if(i!=0) { Socket echoSocket; echoSocket = new Socket(client, 10001); DataOutputStream os = new DataOutputStream(echoSocket.getOutputStream()); os.writeBytes("reconnection\n"); os.writeBytes(port+"\n"); os.close(); } i++; } } public void changeVNCServer(String hostName) throws UnknownHostException, IOException { // sender and reader workingProtocol.cleanUpSession(); // run call and change workingProtocol this.connectionParams.hostName = hostName; this.run(); this.createConnectionAndStart(); createSocketAndSend(aClient.getList(),String.valueOf(opendPort)); } }