379 lines
7.3 KiB
C
379 lines
7.3 KiB
C
/**
|
|
* 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 (a->UserLength > 0)
|
|
{
|
|
if (!a->User || !b->User)
|
|
return FALSE;
|
|
if (strncmp(a->User, b->User, a->UserLength) != 0)
|
|
return FALSE;
|
|
}
|
|
if (a->DomainLength > 0)
|
|
{
|
|
if (!a->Domain || !b->Domain)
|
|
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)
|
|
{
|
|
(void)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 = 0;
|
|
INT64 fileSize = 0;
|
|
|
|
if (!sam || !sam->fp)
|
|
return FALSE;
|
|
|
|
if (_fseeki64(sam->fp, 0, SEEK_END) != 0)
|
|
return FALSE;
|
|
fileSize = _ftelli64(sam->fp);
|
|
if (_fseeki64(sam->fp, 0, SEEK_SET) != 0)
|
|
return FALSE;
|
|
|
|
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 = 0;
|
|
size_t NtHashLength = 0;
|
|
size_t count = 0;
|
|
char* cur = NULL;
|
|
|
|
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 = 0;
|
|
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)
|
|
(void)fclose(sam->fp);
|
|
free(sam);
|
|
}
|
|
}
|