/**

 * WinPR: Windows Portable Runtime

 * Security Accounts Manager (SAM)

 *

 * Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 *

 * Licensed 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.

 */

#include <winpr/config.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <winpr/wtypes.h>

#include <winpr/crt.h>

#include <winpr/sam.h>

#include <winpr/print.h>

#include <winpr/file.h>

#include "../log.h"

#ifdef WINPR_HAVE_UNISTD_H

#include <unistd.h>

#endif

#ifdef _WIN32

#define WINPR_SAM_FILE "C:\\SAM"

#else

#define WINPR_SAM_FILE "/etc/winpr/SAM"

#endif

#define TAG WINPR_TAG("utils")

struct winpr_sam

{

  FILE* fp;

  char* line;

  char* buffer;

  char* context;

  BOOL readOnly;

};

static WINPR_SAM_ENTRY* SamEntryFromDataA(LPCSTR User, DWORD UserLength, LPCSTR Domain,

                                          DWORD DomainLength)

{

  WINPR_SAM_ENTRY* entry = calloc(1, sizeof(WINPR_SAM_ENTRY));

  if (!entry)

    return NULL;

  if (User && (UserLength > 0))

    entry->User = _strdup(User);

  entry->UserLength = UserLength;

  if (Domain && (DomainLength > 0))

    entry->Domain = _strdup(Domain);

  entry->DomainLength = DomainLength;

  return entry;

}

static BOOL SamAreEntriesEqual(const WINPR_SAM_ENTRY* a, const WINPR_SAM_ENTRY* b)

{

  if (!a || !b)

    return FALSE;

  if (a->UserLength != b->UserLength)

    return FALSE;

  if (a->DomainLength != b->DomainLength)

    return FALSE;

  if (strncmp(a->User, b->User, a->UserLength) != 0)

    return FALSE;

  if (strncmp(a->Domain, b->Domain, a->DomainLength) != 0)

    return FALSE;

  return TRUE;

}

WINPR_SAM* SamOpen(const char* filename, BOOL readOnly)

{

  FILE* fp = NULL;

  WINPR_SAM* sam = NULL;

  if (!filename)

    filename = WINPR_SAM_FILE;

  if (readOnly)

    fp = winpr_fopen(filename, "r");

  else

  {

    fp = winpr_fopen(filename, "r+");

    if (!fp)

      fp = winpr_fopen(filename, "w+");

  }

  if (fp)

  {

    sam = (WINPR_SAM*)calloc(1, sizeof(WINPR_SAM));

    if (!sam)

    {

      fclose(fp);

      return NULL;

    }

    sam->readOnly = readOnly;

    sam->fp = fp;

  }

  else

  {

    WLog_DBG(TAG, "Could not open SAM file!");

    return NULL;

  }

  return sam;

}

static BOOL SamLookupStart(WINPR_SAM* sam)

{

  size_t readSize;

  INT64 fileSize;

  if (!sam || !sam->fp)

    return FALSE;

  _fseeki64(sam->fp, 0, SEEK_END);

  fileSize = _ftelli64(sam->fp);

  _fseeki64(sam->fp, 0, SEEK_SET);

  if (fileSize < 1)

    return FALSE;

  sam->context = NULL;

  sam->buffer = (char*)calloc((size_t)fileSize + 2, 1);

  if (!sam->buffer)

    return FALSE;

  readSize = fread(sam->buffer, (size_t)fileSize, 1, sam->fp);

  if (!readSize)

  {

    if (!ferror(sam->fp))

      readSize = (size_t)fileSize;

  }

  if (readSize < 1)

  {

    free(sam->buffer);

    sam->buffer = NULL;

    return FALSE;

  }

  sam->buffer[fileSize] = '\n';

  sam->buffer[fileSize + 1] = '\0';

  sam->line = strtok_s(sam->buffer, "\n", &sam->context);

  return TRUE;

}

static void SamLookupFinish(WINPR_SAM* sam)

{

  free(sam->buffer);

  sam->buffer = NULL;

  sam->line = NULL;

}

static BOOL SamReadEntry(WINPR_SAM* sam, WINPR_SAM_ENTRY* entry)

{

  char* p[5];

  size_t LmHashLength;

  size_t NtHashLength;

  size_t count = 0;

  char* cur;

  if (!sam || !entry || !sam->line)

    return FALSE;

  cur = sam->line;

  while ((cur = strchr(cur, ':')) != NULL)

  {

    count++;

    cur++;

  }

  if (count < 4)

    return FALSE;

  p[0] = sam->line;

  p[1] = strchr(p[0], ':') + 1;

  p[2] = strchr(p[1], ':') + 1;

  p[3] = strchr(p[2], ':') + 1;

  p[4] = strchr(p[3], ':') + 1;

  LmHashLength = (p[3] - p[2] - 1);

  NtHashLength = (p[4] - p[3] - 1);

  if ((LmHashLength != 0) && (LmHashLength != 32))

    return FALSE;

  if ((NtHashLength != 0) && (NtHashLength != 32))

    return FALSE;

  entry->UserLength = (UINT32)(p[1] - p[0] - 1);

  entry->User = (LPSTR)malloc(entry->UserLength + 1);

  if (!entry->User)

    return FALSE;

  entry->User[entry->UserLength] = '\0';

  entry->DomainLength = (UINT32)(p[2] - p[1] - 1);

  memcpy(entry->User, p[0], entry->UserLength);

  if (entry->DomainLength > 0)

  {

    entry->Domain = (LPSTR)malloc(entry->DomainLength + 1);

    if (!entry->Domain)

    {

      free(entry->User);

      entry->User = NULL;

      return FALSE;

    }

    memcpy(entry->Domain, p[1], entry->DomainLength);

    entry->Domain[entry->DomainLength] = '\0';

  }

  else

    entry->Domain = NULL;

  if (LmHashLength == 32)

    winpr_HexStringToBinBuffer(p[2], LmHashLength, entry->LmHash, sizeof(entry->LmHash));

  if (NtHashLength == 32)

    winpr_HexStringToBinBuffer(p[3], NtHashLength, (BYTE*)entry->NtHash, sizeof(entry->NtHash));

  return TRUE;

}

void SamFreeEntry(WINPR_SAM* sam, WINPR_SAM_ENTRY* entry)

{

  if (entry)

  {

    if (entry->UserLength > 0)

      free(entry->User);

    if (entry->DomainLength > 0)

      free(entry->Domain);

    free(entry);

  }

}

void SamResetEntry(WINPR_SAM_ENTRY* entry)

{

  if (!entry)

    return;

  if (entry->UserLength)

  {

    free(entry->User);

    entry->User = NULL;

  }

  if (entry->DomainLength)

  {

    free(entry->Domain);

    entry->Domain = NULL;

  }

  ZeroMemory(entry->LmHash, sizeof(entry->LmHash));

  ZeroMemory(entry->NtHash, sizeof(entry->NtHash));

}

WINPR_SAM_ENTRY* SamLookupUserA(WINPR_SAM* sam, LPCSTR User, UINT32 UserLength, LPCSTR Domain,

                                UINT32 DomainLength)

{

  size_t length;

  BOOL found = FALSE;

  WINPR_SAM_ENTRY* search = SamEntryFromDataA(User, UserLength, Domain, DomainLength);

  WINPR_SAM_ENTRY* entry = (WINPR_SAM_ENTRY*)calloc(1, sizeof(WINPR_SAM_ENTRY));

  if (!entry || !search)

    goto fail;

  if (!SamLookupStart(sam))

    goto fail;

  while (sam->line != NULL)

  {

    length = strlen(sam->line);

    if (length > 1)

    {

      if (sam->line[0] != '#')

      {

        if (!SamReadEntry(sam, entry))

        {

          goto out_fail;

        }

        if (SamAreEntriesEqual(entry, search))

        {

          found = 1;

          break;

        }

      }

    }

    SamResetEntry(entry);

    sam->line = strtok_s(NULL, "\n", &sam->context);

  }

out_fail:

  SamLookupFinish(sam);

fail:

  SamFreeEntry(sam, search);

  if (!found)

  {

    SamFreeEntry(sam, entry);

    return NULL;

  }

  return entry;

}

WINPR_SAM_ENTRY* SamLookupUserW(WINPR_SAM* sam, LPCWSTR User, UINT32 UserLength, LPCWSTR Domain,

                                UINT32 DomainLength)

{

  WINPR_SAM_ENTRY* entry = NULL;

  char* utfUser = NULL;

  char* utfDomain = NULL;

  size_t userCharLen = 0;

  size_t domainCharLen = 0;

  utfUser = ConvertWCharNToUtf8Alloc(User, UserLength / sizeof(WCHAR), &userCharLen);

  if (!utfUser)

    goto fail;

  if (DomainLength > 0)

  {

    utfDomain = ConvertWCharNToUtf8Alloc(Domain, DomainLength / sizeof(WCHAR), &domainCharLen);

    if (!utfDomain)

      goto fail;

  }

  entry = SamLookupUserA(sam, utfUser, (UINT32)userCharLen, utfDomain, (UINT32)domainCharLen);

fail:

  free(utfUser);

  free(utfDomain);

  return entry;

}

void SamClose(WINPR_SAM* sam)

{

  if (sam != NULL)

  {

    if (sam->fp)

      fclose(sam->fp);

    free(sam);

  }

}