services/console-proxy-rdp/rdpconsole/src/main/java/vncclient/vnc/VncInitializer.java

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
package vncclient.vnc;

import streamer.ByteBuffer;
import streamer.Element;
import streamer.Link;
import streamer.OneTimeSwitch;
import streamer.Pipeline;
import streamer.PipelineImpl;
import streamer.debug.MockSink;
import streamer.debug.MockSource;
import common.ScreenDescription;

public class VncInitializer extends OneTimeSwitch {

    // Pad names
    public static final String CLIENT_SUPPORTED_ENCODINGS_ADAPTER_PAD = "encodings";
    public static final String CLIENT_PIXEL_FORMAT_ADAPTER_PAD = "pixel_format";

    protected byte sharedFlag = RfbConstants.EXCLUSIVE_ACCESS;

    /**
     * Properties of remote screen .
     */
    protected ScreenDescription screen;

    public VncInitializer(String id, boolean shared, ScreenDescription screen) {
        super(id);

        setSharedFlag(shared);
        this.screen = screen;

        declarePads();
    }

    @Override
    protected void declarePads() {
        super.declarePads();
        outputPads.put(CLIENT_SUPPORTED_ENCODINGS_ADAPTER_PAD, null);
        outputPads.put(CLIENT_PIXEL_FORMAT_ADAPTER_PAD, null);
    }

    public ScreenDescription getScreen() {
        return screen;
    }

    public void setScreen(ScreenDescription screen) {
        this.screen = screen;
    }

    public void setSharedFlag(boolean shared) {
        if (shared)
            sharedFlag = RfbConstants.SHARED_ACCESS;
        else
            sharedFlag = RfbConstants.EXCLUSIVE_ACCESS;
    }

    @Override
    protected void handleOneTimeData(ByteBuffer buf, Link link) {
        if (verbose)
            System.out.println("[" + this + "] INFO: Data received: " + buf + ".");

        // Server initialization message is at least 24 bytes long + length of
        // desktop name
        if (!cap(buf, 24, UNLIMITED, link, false))
            return;

        // Read server initialization message
        // Read frame buffer size
        int framebufferWidth = buf.readUnsignedShort();
        int framebufferHeight = buf.readUnsignedShort();

        // Read pixel format
        int bitsPerPixel = buf.readUnsignedByte();
        int depth = buf.readUnsignedByte();

        int bigEndianFlag = buf.readUnsignedByte();
        int trueColorFlag = buf.readUnsignedByte();

        int redMax = buf.readUnsignedShort();
        int greenMax = buf.readUnsignedShort();
        int blueMax = buf.readUnsignedShort();

        int redShift = buf.readUnsignedByte();
        int greenShift = buf.readUnsignedByte();
        int blueShift = buf.readUnsignedByte();

        // Skip padding
        buf.skipBytes(3);

        // Read desktop name
        int length = buf.readSignedInt();

        // Consume exactly $length bytes, push back any extra bytes
        if (!cap(buf, length, length, link, true))
            return;

        String desktopName = buf.readString(length, RfbConstants.US_ASCII_CHARSET);
        buf.unref();
        if (verbose)
            System.out.println("[" + this + "] INFO: Desktop name: \"" + desktopName + "\", bpp: " + bitsPerPixel + ", depth: " + depth + ", screen size: "
                    + framebufferWidth + "x" + framebufferHeight + ".");

        // Set screen properties
        screen.setFramebufferSize(framebufferWidth, framebufferHeight);
        screen.setPixelFormat(bitsPerPixel, depth, bigEndianFlag != RfbConstants.LITTLE_ENDIAN, trueColorFlag == RfbConstants.TRUE_COLOR, redMax, greenMax, blueMax, redShift,
                greenShift, blueShift);
        screen.setDesktopName(desktopName);

        // If sever screen has different parameters than ours, then change it
        if (!screen.isRGB888_32_LE()) {
            // Send client pixel format
            sendClientPixelFormat();
        }

        // Send encodings supported by client
        sendSupportedEncodings();

        switchOff();

    }

    @Override
    protected void onStart() {
        ByteBuffer buf = new ByteBuffer(new byte[] {sharedFlag});
        pushDataToOTOut(buf);
    }

    private void sendClientPixelFormat() {
        pushDataToPad(CLIENT_PIXEL_FORMAT_ADAPTER_PAD, new ByteBuffer(0));
    }

    private void sendSupportedEncodings() {
        pushDataToPad(CLIENT_SUPPORTED_ENCODINGS_ADAPTER_PAD, new ByteBuffer(0));
    }

    @Override
    public String toString() {
        return "VncInit(" + id + ")";
    }

    /**
     * Example.
     */
    public static void main(String args[]) {
        // System.setProperty("streamer.Link.debug", "true");
        System.setProperty("streamer.Element.debug", "true");
        // System.setProperty("streamer.Pipeline.debug", "true");

        final String desktopName = "test";

        Element source = new MockSource("source") {
            {
                bufs = ByteBuffer.convertByteArraysToByteBuffers(
                        // Send screen description
                        new byte[] {
                                // Framebuffer width (short)
                                0, (byte)200,
                                // Framebuffer height (short)
                                0, 100,
                                // Bits per pixel
                                32,
                                // Depth,
                                24,
                                // Endianness flag
                                RfbConstants.LITTLE_ENDIAN,
                                // Truecolor flag
                                RfbConstants.TRUE_COLOR,
                                // Red max (short)
                                0, (byte)255,
                                // Green max (short)
                                0, (byte)255,
                                // Blue max (short)
                                0, (byte)255,
                                // Red shift
                                16,
                                // Green shift
                                8,
                                // Blue shift
                                0,
                                // Padding
                                0, 0, 0,
                                // Desktop name length (int)
                                0, 0, 0, 4,
                                // Desktop name ("test", 4 bytes)
                                't', 'e', 's', 't',

                                // Tail
                                1, 2, 3

                        },
                        // Tail packet
                        new byte[] {4, 5, 6});
            }
        };

        ScreenDescription screen = new ScreenDescription();
        final VncInitializer init = new VncInitializer("init", true, screen);
        Element initSink = new MockSink("initSink") {
            {
                // Expect shared flag
                bufs = ByteBuffer.convertByteArraysToByteBuffers(new byte[] {RfbConstants.SHARED_ACCESS});
            }
        };
        Element mainSink = new MockSink("mainSink") {
            {
                // Expect two tail packets
                bufs = ByteBuffer.convertByteArraysToByteBuffers(new byte[] {1, 2, 3}, new byte[] {4, 5, 6});
            }
        };
        ByteBuffer[] emptyBuf = ByteBuffer.convertByteArraysToByteBuffers(new byte[] {});
        Element encodingsSink = new MockSink("encodings", emptyBuf);
        Element pixelFormatSink = new MockSink("pixel_format", emptyBuf);

        Pipeline pipeline = new PipelineImpl("test");
        pipeline.addAndLink(source, init, mainSink);
        pipeline.add(encodingsSink, pixelFormatSink, initSink);
        pipeline.link("init >otout", "initSink");
        pipeline.link("init >" + CLIENT_SUPPORTED_ENCODINGS_ADAPTER_PAD, "encodings");
        pipeline.link("init >" + CLIENT_PIXEL_FORMAT_ADAPTER_PAD, "pixel_format");

        pipeline.runMainLoop("source", STDOUT, false, false);

        if (!screen.isRGB888_32_LE())
            System.err.println("Screen description was read incorrectly: " + screen + ".");
        if (!desktopName.equals(screen.getDesktopName()))
            System.err.println("Screen desktop name was read incorrectly: \"" + screen.getDesktopName() + "\".");

    }
}