/*
* Copyright (c) 2010, 2014, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code 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
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.glass.ui.monocle;
import com.sun.glass.events.MouseEvent;
import javafx.application.Platform;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.channels.WritableByteChannel;
import java.nio.charset.Charset;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.BitSet;
import java.util.HashSet;
import java.util.Set;
A headless screen that is available for remote connections using the
RFB 3.3 protocol on port 5901.
/** A headless screen that is available for remote connections using the
* RFB 3.3 protocol on port 5901.
*/
class VNCScreen extends HeadlessScreen {
private ServerSocketChannel server;
private Set<ClientConnection> clients = new HashSet<ClientConnection>();
VNCScreen() {
super(1024, 600, 32);
try {
server = ServerSocketChannel.open();
int vncPort = AccessController.doPrivileged(
(PrivilegedAction<Integer>)
() -> Integer.getInteger("vnc.port", 5901));
server.bind(new InetSocketAddress(vncPort));
Thread t = new Thread(new ConnectionAccepter());
t.setDaemon(true);
t.setName("VNC Server on port " + vncPort);
t.start();
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void shutdown() {
super.shutdown();
for (ClientConnection cc : clients) {
try {
cc.socket.close();
} catch (IOException e) { }
}
}
@Override
public void swapBuffers() {
ClientConnection[] ccs;
synchronized (clients) {
ccs = clients.toArray(new ClientConnection[clients.size()]);
}
for (ClientConnection cc : ccs) {
try {
sendBuffer(cc.socket);
} catch (IOException e) {
clients.remove(cc);
}
}
super.swapBuffers();
}
private void removeClient(ClientConnection cc, IOException e) {
synchronized (clients) {
if (clients.contains(cc)) {
System.out.format("Disconnecting %s: %s\n",
cc.descriptor, e.getMessage());
clients.remove(cc);
}
}
}
private void sendBuffer(WritableByteChannel out) throws IOException {
ByteBuffer buffer = ByteBuffer.allocate(16);
buffer.order(ByteOrder.BIG_ENDIAN);
buffer.put((byte) 0);
buffer.put((byte) 0);
buffer.putShort((short) 1); // rectangle count
buffer.putShort((short) 0); // x
buffer.putShort((short) 0); // y
buffer.putShort((short) width);
buffer.putShort((short) height);
buffer.putInt(0); // raw
buffer.flip();
out.write(buffer);
fb.write(out);
}
private class ConnectionAccepter implements Runnable {
@Override
public void run() {
ByteBuffer buffer = ByteBuffer.allocate(64);
buffer.order(ByteOrder.BIG_ENDIAN);
while (true) {
try {
SocketChannel client = server.accept();
System.out.format("Connection received from %s\n",
client.getRemoteAddress());
// Declare the server protocol version
buffer.clear();
buffer.put("RFB 003.003\n".getBytes());
buffer.flip();
client.write(buffer);
// Read the client protocol version
buffer.clear();
buffer.limit(12);
client.read(buffer);
buffer.flip();
System.out.format("Client supports %s\n",
Charset.forName("UTF-8")
.decode(buffer).toString().trim());
buffer.clear();
buffer.putInt(1); // no authentication
buffer.flip();
client.write(buffer);
buffer.clear();
buffer.limit(1);
client.read(buffer);
System.out.format("Client share request: %d\n",
buffer.get(0));
buffer.clear();
buffer.putShort((short) width);
buffer.putShort((short) height);
buffer.put((byte) depth);
buffer.put((byte) depth);
buffer.put((byte) (ByteOrder.nativeOrder().equals(ByteOrder.BIG_ENDIAN) ? 0 : 1));
buffer.put((byte) 1); // true color
if (depth == 32) {
buffer.putShort((short) 255); // red max
buffer.putShort((short) 255); // green max
buffer.putShort((short) 255); // blue max
buffer.put((byte) 16); // red offset
buffer.put((byte) 8); // blue offset
buffer.put((byte) 0); // green offset
} else {
buffer.putShort((byte) (short) 31);
buffer.putShort((byte) (short) 63);
buffer.putShort((byte) (short) 31);
buffer.put((byte) 11);
buffer.put((byte) 5);
buffer.put((byte) 0);
}
buffer.put((byte) 0); // padding
buffer.put((byte) 0);
buffer.put((byte) 0);
String name = "JavaFX on " + client.getLocalAddress();
buffer.putInt(name.length());
buffer.put(name.getBytes());
buffer.flip();
client.write(buffer);
ClientConnection cc = new ClientConnection();
cc.socket = client;
Thread t = new Thread(cc);
t.setDaemon(true);
t.setName("VNC client connection from "
+ client.getRemoteAddress());
t.start();
synchronized (clients) {
clients.add(cc);
}
sendBuffer(client);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
private class ClientConnection implements Runnable {
private SocketChannel socket;
private String descriptor;
@Override
public void run() {
ByteBuffer buffer = ByteBuffer.allocate(32);
buffer.order(ByteOrder.BIG_ENDIAN);
try {
descriptor = socket.getRemoteAddress().toString();
while (true) {
buffer.clear();
buffer.limit(4);
socket.read(buffer);
switch (buffer.get(0)) {
case 0: // SetPixelFormat
// discard the message in the next 16 bytes
buffer.clear();
buffer.limit(16);
socket.read(buffer);
break;
case 1: // FixColorMapEntries
buffer.clear();
buffer.limit(2);
socket.read(buffer);
// discard color map entries
int colorMapEntryCount = buffer.getShort(0);
for (int i = 0; i < colorMapEntryCount; i++) {
buffer.clear();
buffer.limit(6);
socket.read(buffer);
}
break;
case 2: // SetEncodings
// discard encodings
int encodingCount = buffer.getShort(2);
for (int i = 0; i < encodingCount; i++) {
buffer.clear();
buffer.limit(4);
socket.read(buffer);
}
case 3: // FramebufferUpdateRequest
buffer.clear();
buffer.limit(6);
socket.read(buffer);
Platform.runLater(() -> {
try {
if (fb.hasReceivedData()) {
// an update is in progress and will
// be sent on the next call to
// swapBuffers. No need to
// respond to this request.
} else {
sendBuffer(socket);
}
} catch (IOException e) {
removeClient(ClientConnection.this, e);
}
});
break;
case 4: // KeyEvent
buffer.clear();
buffer.limit(4);
socket.read(buffer);
break;
case 5: { // PointerEvent
int x = buffer.getShort(2);
buffer.position(1);
buffer.limit(2);
BitSet buttons = BitSet.valueOf(buffer);
buffer.clear();
buffer.limit(2);
socket.read(buffer);
int y = buffer.getShort(0);
final MouseState state = new MouseState();
state.setX(x);
state.setY(y);
if (buttons.get(0)) {
state.pressButton(MouseEvent.BUTTON_LEFT);
}
if (buttons.get(1)) {
state.pressButton(MouseEvent.BUTTON_OTHER);
}
if (buttons.get(2)) {
state.pressButton(MouseEvent.BUTTON_RIGHT);
}
Platform.runLater(() -> MouseInput.getInstance().setState(state, false));
break;
}
case 6: // ClientCutText
buffer.clear();
buffer.limit(4);
socket.read(buffer);
int textLength = buffer.getInt(0);
for (int i = 0; i < textLength; i++) {
buffer.clear();
buffer.limit(1);
socket.read(buffer);
}
break;
default:
System.err.format(
"Unknown message %d from client %s\n",
buffer.get(0), socket.getRemoteAddress());
}
}
} catch (IOException e) {
removeClient(this, e);
}
}
}
}