/*

 *  Unix SMB/CIFS implementation.

 *  Version 3.0

 *  NTLMSSP Signing routines

 *  Copyright (C) Luke Kenneth Casson Leighton 1996-2001

 *  Copyright (C) Andrew Bartlett <abartlet@samba.org> 2003-2005

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 3 of the License, or

 *  (at your option) any later version.

 *

 *  This program 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 for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, see <http://www.gnu.org/licenses/>.

 */

#include "includes.h"

#include "auth/ntlmssp/ntlmssp.h"

#include "auth/gensec/gensec.h"

#include "auth/gensec/gensec_internal.h"

#include "auth/ntlmssp/ntlmssp_private.h"

NTSTATUS gensec_ntlmssp_magic(struct gensec_security *gensec_security,

            const DATA_BLOB *first_packet)

{

  if (ntlmssp_blob_matches_magic(first_packet)) {

    return NT_STATUS_OK;

  } else {

    return NT_STATUS_INVALID_PARAMETER;

  }

}

/**

 * Return the NTLMSSP master session key

 *

 * @param ntlmssp_state NTLMSSP State

 */

NTSTATUS gensec_ntlmssp_session_key(struct gensec_security *gensec_security,

            TALLOC_CTX *mem_ctx,

            DATA_BLOB *session_key)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;

  if (ntlmssp_state->expected_state != NTLMSSP_DONE) {

    return NT_STATUS_NO_USER_SESSION_KEY;

  }

  if (!ntlmssp_state->session_key.data) {

    return NT_STATUS_NO_USER_SESSION_KEY;

  }

  *session_key = data_blob_talloc(mem_ctx, ntlmssp_state->session_key.data, ntlmssp_state->session_key.length);

  if (!session_key->data) {

    return NT_STATUS_NO_MEMORY;

  }

  return NT_STATUS_OK;

}

bool gensec_ntlmssp_have_feature(struct gensec_security *gensec_security,

         uint32_t feature)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;

  if (feature & GENSEC_FEATURE_SIGN) {

    if (!ntlmssp_state->session_key.length) {

      return false;

    }

    if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SIGN) {

      return true;

    }

  }

  if (feature & GENSEC_FEATURE_SEAL) {

    if (!ntlmssp_state->session_key.length) {

      return false;

    }

    if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SEAL) {

      return true;

    }

  }

  if (feature & GENSEC_FEATURE_SESSION_KEY) {

    if (ntlmssp_state->session_key.length) {

      return true;

    }

  }

  if (feature & GENSEC_FEATURE_DCE_STYLE) {

    return true;

  }

  if (feature & GENSEC_FEATURE_ASYNC_REPLIES) {

    if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {

      return true;

    }

  }

  if (feature & GENSEC_FEATURE_SIGN_PKT_HEADER) {

    return true;

  }

  if (feature & GENSEC_FEATURE_NEW_SPNEGO) {

    if (!ntlmssp_state->session_key.length) {

      return false;

    }

    if (!(ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SIGN)) {

      return false;

    }

    return ntlmssp_state->new_spnego;

  }

  return false;

}

NTSTATUS gensec_ntlmssp_start(struct gensec_security *gensec_security)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp;

  gensec_ntlmssp = talloc_zero(gensec_security,

             struct gensec_ntlmssp_context);

  if (!gensec_ntlmssp) {

    return NT_STATUS_NO_MEMORY;

  }

  gensec_security->private_data = gensec_ntlmssp;

  return NT_STATUS_OK;

}

NTSTATUS gensec_ntlmssp_sign_packet(struct gensec_security *gensec_security,

            TALLOC_CTX *sig_mem_ctx,

            const uint8_t *data, size_t length,

            const uint8_t *whole_pdu, size_t pdu_length,

            DATA_BLOB *sig)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  NTSTATUS nt_status;

  nt_status = ntlmssp_sign_packet(gensec_ntlmssp->ntlmssp_state,

          sig_mem_ctx,

          data, length,

          whole_pdu, pdu_length,

          sig);

  return nt_status;

}

NTSTATUS gensec_ntlmssp_check_packet(struct gensec_security *gensec_security,

             const uint8_t *data, size_t length,

             const uint8_t *whole_pdu, size_t pdu_length,

             const DATA_BLOB *sig)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  NTSTATUS nt_status;

  nt_status = ntlmssp_check_packet(gensec_ntlmssp->ntlmssp_state,

           data, length,

           whole_pdu, pdu_length,

           sig);

  return nt_status;

}

NTSTATUS gensec_ntlmssp_seal_packet(struct gensec_security *gensec_security,

            TALLOC_CTX *sig_mem_ctx,

            uint8_t *data, size_t length,

            const uint8_t *whole_pdu, size_t pdu_length,

            DATA_BLOB *sig)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  NTSTATUS nt_status;

  nt_status = ntlmssp_seal_packet(gensec_ntlmssp->ntlmssp_state,

          sig_mem_ctx,

          data, length,

          whole_pdu, pdu_length,

          sig);

  return nt_status;

}

/*

  wrappers for the ntlmssp_*() functions

*/

NTSTATUS gensec_ntlmssp_unseal_packet(struct gensec_security *gensec_security,

              uint8_t *data, size_t length,

              const uint8_t *whole_pdu, size_t pdu_length,

              const DATA_BLOB *sig)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  NTSTATUS nt_status;

  nt_status = ntlmssp_unseal_packet(gensec_ntlmssp->ntlmssp_state,

            data, length,

            whole_pdu, pdu_length,

            sig);

  return nt_status;

}

size_t gensec_ntlmssp_sig_size(struct gensec_security *gensec_security, size_t data_size)

{

  return NTLMSSP_SIG_SIZE;

}

NTSTATUS gensec_ntlmssp_wrap(struct gensec_security *gensec_security,

           TALLOC_CTX *out_mem_ctx,

           const DATA_BLOB *in,

           DATA_BLOB *out)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  return ntlmssp_wrap(gensec_ntlmssp->ntlmssp_state,

          out_mem_ctx,

          in, out);

}

NTSTATUS gensec_ntlmssp_unwrap(struct gensec_security *gensec_security,

             TALLOC_CTX *out_mem_ctx,

             const DATA_BLOB *in,

             DATA_BLOB *out)

{

  struct gensec_ntlmssp_context *gensec_ntlmssp =

    talloc_get_type_abort(gensec_security->private_data,

              struct gensec_ntlmssp_context);

  return ntlmssp_unwrap(gensec_ntlmssp->ntlmssp_state,

            out_mem_ctx,

            in, out);

}