#include "config.h"

#if HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#if HAVE_UNISTD_H
#include <unistd.h>
#endif
#if HAVE_SYSLOG_H
#include <syslog.h>
#endif
#if HAVE_STRING_H
#include <string.h>
#endif

#include "smblib-priv.h"
#include "valid.h"

SMB_Handle_Type SMB_Connect_Server(void *, char *, char *);

int
Valid_User(char *username, char *password, char *server, char *backup, char *domain)
{
    int pass_is_precrypted_p = 0;
    char const *supportedDialects[] = {
        /*              "PC NETWORK PROGRAM 1.0", */
        /*              "MICROSOFT NETWORKS 1.03", */
        /*              "MICROSOFT NETWORKS 3.0", */
        "LANMAN1.0",
        "LM1.2X002",
        "Samba",
        /*              "NT LM 0.12", */
        /*              "NT LANMAN 1.0", */
        NULL
    };
    SMB_Handle_Type con;

    SMB_Init();
    con = SMB_Connect_Server(NULL, server, domain);
    if (con == NULL) {		/* Error ... */
        con = SMB_Connect_Server(NULL, backup, domain);
        if (con == NULL) {
            return (NTV_SERVER_ERROR);
        }
    }
    if (SMB_Negotiate(con, supportedDialects) < 0) {	/* An error */
        SMB_Discon(con, 0);
        return (NTV_PROTOCOL_ERROR);
    }
    /* Test for a server in share level mode do not authenticate against it */
    if (con->Security == 0) {
        SMB_Discon(con, 0);
        return (NTV_PROTOCOL_ERROR);
    }
    if (SMB_Logon_Server(con, username, password, domain, pass_is_precrypted_p) < 0) {
        SMB_Discon(con, 0);
        return (NTV_LOGON_ERROR);
    }
    SMB_Discon(con, 0);
    return (NTV_NO_ERROR);
}

void *
NTLM_Connect(char *server, char *backup, char *domain, char *nonce)
{
    char const *SMB_Prots[] = {
        /*              "PC NETWORK PROGRAM 1.0", */
        /*              "MICROSOFT NETWORKS 1.03", */
        /*              "MICROSOFT NETWORKS 3.0", */
        "LANMAN1.0",
        "LM1.2X002",
        "Samba",
        /*              "NT LM 0.12", */
        /*              "NT LANMAN 1.0", */
        NULL
    };
    SMB_Handle_Type con;

    SMB_Init();
    con = SMB_Connect_Server(NULL, server, domain);
    if (con == NULL) {		/* Error ... */
        con = SMB_Connect_Server(NULL, backup, domain);
        if (con == NULL) {
            return (NULL);
        }
    }
    if (SMB_Negotiate(con, SMB_Prots) < 0) {	/* An error */
        SMB_Discon(con, 0);
        return (NULL);
    }
    /* Test for a server in share level mode do not authenticate against it */
    if (con->Security == 0) {
        SMB_Discon(con, 0);
        return (NULL);
    }
    memcpy(nonce, con->Encrypt_Key, 8);

    return (con);
}

int
NTLM_Auth(void *handle, char *username, char *password, int flag)
{
    SMB_Handle_Type con = handle;

    if (SMB_Logon_Server(con, username, password, NULL, flag) < 0) {
        return (NTV_LOGON_ERROR);
    }
    return (NTV_NO_ERROR);
}

void
NTLM_Disconnect(void *handle)
{
    SMB_Handle_Type con = handle;
    SMB_Discon(con, 0);
}