// 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.ntlmssp;

import rdpclient.ntlmssp.asn1.TSCredentials;
import rdpclient.ntlmssp.asn1.TSPasswordCreds;
import rdpclient.ntlmssp.asn1.TSRequest;
import rdpclient.rdp.RdpConstants;
import streamer.ByteBuffer;
import streamer.Element;
import streamer.Link;
import streamer.OneTimeSwitch;

public class ClientNtlmsspUserCredentials extends OneTimeSwitch implements Element {

    protected NtlmState ntlmState;

    public ClientNtlmsspUserCredentials(String id, NtlmState ntlmState) {
        super(id);
        this.ntlmState = ntlmState;
    }

    @Override
    protected void handleOneTimeData(ByteBuffer buf, Link link) {
        if (buf == null)
            return;

        throw new RuntimeException("Unexpected packet: " + buf + ".");
    }

    @Override
    protected void onStart() {
        super.onStart();

        ByteBuffer buf = new ByteBuffer(4096, true);

        TSRequest tsRequest = new TSRequest("TSRequest");
        tsRequest.version.value = 2L;

        ByteBuffer tsCredentialsBuf = generateTSCredentials();
        tsRequest.authInfo.value = encryptTSCredentials(tsCredentialsBuf);
        tsCredentialsBuf.unref();

        tsRequest.writeTag(buf);

        // Trim buffer to actual length of data written
        buf.trimAtCursor();

        pushDataToOTOut(buf);

        switchOff();
    }

    private ByteBuffer encryptTSCredentials(ByteBuffer buf2) {
        return new ByteBuffer(ntlmState.ntlm_EncryptMessage(buf2.toByteArray()));
    }

    private ByteBuffer generateTSCredentials() {
        ByteBuffer buf = new ByteBuffer(4096);

        TSCredentials tsCredentials = new TSCredentials("authInfo");
        // 1 means that credentials field contains a TSPasswordCreds structure
        tsCredentials.credType.value = 1L;

        ByteBuffer tsPasswordCredsBuf = new ByteBuffer(4096, true);
        TSPasswordCreds tsPasswordCreds = new TSPasswordCreds("credentials");
        tsPasswordCreds.domainName.value = new ByteBuffer(ntlmState.domain.getBytes(RdpConstants.CHARSET_16));
        tsPasswordCreds.userName.value = new ByteBuffer(ntlmState.user.getBytes(RdpConstants.CHARSET_16));
        tsPasswordCreds.password.value = new ByteBuffer(ntlmState.password.getBytes(RdpConstants.CHARSET_16));
        tsPasswordCreds.writeTag(tsPasswordCredsBuf);
        tsPasswordCredsBuf.trimAtCursor();
        //* DEBUG */System.out.println("TSPasswordCreds:\n" + tsPasswordCredsBuf.dump());

        tsCredentials.credentials.value = tsPasswordCredsBuf;

        tsCredentials.writeTag(buf);
        tsPasswordCredsBuf.unref();

        // Trim buffer to actual length of data written
        buf.trimAtCursor();
        //* DEBUG */System.out.println("TSCredentials:\n" + buf.dump());

        return buf;
    }

    /**
     * @param args
     */
    public static void main(String[] args) {
        // TODO Auto-generated method stub

        /* @formatter:off */
        // TSCredentials
//  30 57 // Sequence
//  a0 03 // TAG 0
//  02 01 01 // Integer: 1 : credentials contains a TSPasswordCreds structure
//  a1 50 // TAG 1
//  04 4e // OCTETSTRING
        // TSPasswordCreds
//  30 4c // SEQUENCE
//  a0 14 // TAG 0
//  04 12 // OCTETSTRING
//  77 00 6f 00 72 00 6b 00 67 00 72 00 6f 00 75 00 70 00 // "workgroup"
//  a1 1c // TAG 1
//  04 1a // OCTETSTRING
//  41 00 64 00 6d 00 69 00 6e 00 69 00 73 00 74 00 72 00 61 00 74 00 6f 00 72 00 // "Administrator"
//  a2 16 // TAG 2
//  04 14 //
//  52 00 32 00 50 00 72 00 65 00 76 00 69 00 65 00 77 00 21 00 // "R2Preview!"
        /* @formatter:on */

    }

}