| // 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 rdpclient.rdp; |
| |
| 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; |
| |
| /** |
| * Server response to MCS Attach User request. |
| * |
| * Once the User Channel ID has been extracted, the client MUST send an MCS |
| * Channel Join Request PDU for the user channel. |
| * |
| * @see http://msdn.microsoft.com/en-us/library/cc240685.aspx |
| */ |
| public class ServerMCSAttachUserConfirmPDU extends OneTimeSwitch { |
| |
| public static final int MCS_ATTACH_USER_CONFIRM_PDU = 0xb; |
| |
| public static final int INITIATOR_PRESENT = 0x2; |
| |
| protected RdpState state; |
| |
| public ServerMCSAttachUserConfirmPDU(String id, RdpState state) { |
| super(id); |
| this.state = state; |
| } |
| |
| @Override |
| protected void handleOneTimeData(ByteBuffer buf, Link link) { |
| if (verbose) |
| System.out.println("[" + this + "] INFO: Data received: " + buf + "."); |
| |
| int typeAndFlags = buf.readUnsignedByte(); |
| int type = typeAndFlags >> 2; |
| int flags = typeAndFlags & 0x3; |
| |
| if (type != MCS_ATTACH_USER_CONFIRM_PDU) |
| throw new RuntimeException("[" + this + "] ERROR: Incorrect type of MCS AttachUserConfirm PDU. Expected value: 11, actual value: " + type + ", data: " + buf + "."); |
| |
| if (flags != INITIATOR_PRESENT) |
| throw new RuntimeException("Initator field is not present in MCS AttachUserConfirm PDU. Data: " + buf + "."); |
| |
| int rtSuccess = buf.readUnsignedByte() >> 4; |
| if (rtSuccess != 0) |
| throw new RuntimeException("[" + this + "] ERROR: Cannot attach user: request failed. Error code: " + rtSuccess + ", data: " + buf + "."); |
| |
| // If the initiator field is present, the client stores the value of the |
| // initiator in the User Channel ID store , because the initiator specifies |
| // the User Channel ID. |
| state.serverUserChannelId = buf.readUnsignedShort() + 1001; |
| |
| buf.unref(); |
| |
| // Next: client MCS Channel Join Request PDU (s) |
| switchOff(); |
| } |
| |
| /** |
| * Example. |
| */ |
| /** |
| * Example. |
| * |
| * @see http://msdn.microsoft.com/en-us/library/cc240842.aspx |
| * @see http://msdn.microsoft.com/en-us/library/cc240500.aspx |
| */ |
| public static void main(String args[]) { |
| // System.setProperty("streamer.Link.debug", "true"); |
| System.setProperty("streamer.Element.debug", "true"); |
| // System.setProperty("streamer.Pipeline.debug", "true"); |
| |
| byte[] packet = new byte[] {(byte)0x2E, // MCS user confirm (001011.., |
| // 0xb), InitiatorPresent: 1 |
| // (......01, 0x1) |
| (byte)0x00, // RT successfull (0000...., 0x0) |
| // Initiator: 1001+3 = 1004 |
| (byte)0x00, (byte)0x03,}; |
| |
| RdpState rdpState = new RdpState(); |
| MockSource source = new MockSource("source", ByteBuffer.convertByteArraysToByteBuffers(packet, new byte[] {1, 2, 3})); |
| Element atachUserConfirm = new ServerMCSAttachUserConfirmPDU("attach_user_confirm", rdpState); |
| Element sink = new MockSink("sink"); |
| Element mainSink = new MockSink("mainSink", ByteBuffer.convertByteArraysToByteBuffers(new byte[] {1, 2, 3})); |
| |
| Pipeline pipeline = new PipelineImpl("test"); |
| pipeline.add(source, atachUserConfirm, sink, mainSink); |
| pipeline.link("source", "attach_user_confirm", "mainSink"); |
| pipeline.link("attach_user_confirm >" + OTOUT, "sink"); |
| pipeline.runMainLoop("source", STDOUT, false, false); |
| |
| if (rdpState.serverUserChannelId != 1004) |
| System.err.println("Incorrect user channel ID. Expected value: 1004, actual value: " + rdpState.serverUserChannelId + "."); |
| } |
| |
| } |