mirror of
https://github.com/reactos/reactos.git
synced 2024-08-30 23:28:58 +00:00
161 lines
3.7 KiB
C
161 lines
3.7 KiB
C
/*
|
|
* PROJECT: ReactOS simple TCP/IP services
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/services/tcpsvcs/qotd.c
|
|
* PURPOSE: Sends a random quote to the client
|
|
* COPYRIGHT: Copyright 2005 - 2008 Ged Murphy <gedmurphy@reactos.org>
|
|
*
|
|
*/
|
|
|
|
#include "tcpsvcs.h"
|
|
|
|
#include <stdlib.h>
|
|
|
|
static WCHAR szFilePath[] = L"\\drivers\\etc\\quotes";
|
|
|
|
static BOOL
|
|
SendQuote(SOCKET sock, char* Quote)
|
|
{
|
|
INT strSize = strlen(Quote);
|
|
if (send(sock, Quote, strSize, 0) == SOCKET_ERROR)
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL
|
|
RetrieveQuote(SOCKET sock)
|
|
{
|
|
HANDLE hFile;
|
|
WCHAR szFullPath[MAX_PATH + 20];
|
|
DWORD dwBytesRead;
|
|
LPSTR lpQuotes;
|
|
LPSTR lpStr;
|
|
INT quoteNum;
|
|
INT NumQuotes = 0;
|
|
INT i;
|
|
|
|
if(!GetSystemDirectoryW(szFullPath, MAX_PATH))
|
|
{
|
|
LogEvent(L"QOTD: Getting system path failed", GetLastError(), 0, LOG_FILE);
|
|
return FALSE;
|
|
}
|
|
wcscat(szFullPath, szFilePath);
|
|
|
|
|
|
LogEvent(L"QOTD: Opening quotes file", 0, 0, LOG_FILE);
|
|
hFile = CreateFileW(szFullPath,
|
|
GENERIC_READ,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL);
|
|
if (hFile == INVALID_HANDLE_VALUE)
|
|
{
|
|
LogEvent(L"QOTD: Error opening quotes file", GetLastError(), 0, LOG_FILE);
|
|
}
|
|
else
|
|
{
|
|
DWORD dwSize = GetFileSize(hFile, NULL);
|
|
lpQuotes = (LPSTR)HeapAlloc(GetProcessHeap(), 0, dwSize + 1);
|
|
if (!lpQuotes)
|
|
{
|
|
CloseHandle(hFile);
|
|
return FALSE;
|
|
}
|
|
|
|
ReadFile(hFile,
|
|
lpQuotes,
|
|
dwSize,
|
|
&dwBytesRead,
|
|
NULL);
|
|
CloseHandle(hFile);
|
|
|
|
lpQuotes[dwSize] = 0;
|
|
|
|
if (dwBytesRead != dwSize)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, lpQuotes);
|
|
return FALSE;
|
|
}
|
|
|
|
lpStr = lpQuotes;
|
|
while (*lpStr)
|
|
{
|
|
if (*lpStr == '%')
|
|
NumQuotes++;
|
|
lpStr++;
|
|
}
|
|
|
|
/* pick a random quote */
|
|
srand((unsigned int) GetTickCount());
|
|
quoteNum = rand() % NumQuotes;
|
|
|
|
/* retrieve the full quote */
|
|
lpStr = lpQuotes;
|
|
for (i = 1; i <= quoteNum; i++)
|
|
{
|
|
/* move past preceding quote */
|
|
lpStr++;
|
|
|
|
if (i == quoteNum)
|
|
{
|
|
LPSTR lpStart = lpStr;
|
|
|
|
while (*lpStr != '%' && *lpStr != '\0')
|
|
lpStr++;
|
|
|
|
*lpStr = 0;
|
|
|
|
/* send the quote */
|
|
if (!SendQuote(sock, lpStart))
|
|
LogEvent(L"QOTD: Error sending data", 0, 0, LOG_FILE);
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
while (*lpStr != '%' && *lpStr != '\0')
|
|
lpStr++;
|
|
|
|
/* move past % and RN */
|
|
lpStr += 2;
|
|
}
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, lpQuotes);
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
DWORD WINAPI
|
|
QotdHandler(VOID* sock_)
|
|
{
|
|
SOCKET sock = (SOCKET)sock_;
|
|
DWORD retVal = 0;
|
|
|
|
if (!RetrieveQuote(sock))
|
|
{
|
|
LogEvent(L"QOTD: Error retrieving quote", 0, 0, LOG_FILE);
|
|
retVal = 1;
|
|
}
|
|
|
|
LogEvent(L"QOTD: Shutting connection down", 0, 0, LOG_FILE);
|
|
if (ShutdownConnection(sock, FALSE))
|
|
{
|
|
LogEvent(L"QOTD: Connection is down", 0, 0, LOG_FILE);
|
|
}
|
|
else
|
|
{
|
|
LogEvent(L"QOTD: Connection shutdown failed", 0, 0, LOG_FILE);
|
|
LogEvent(L"QOTD: Terminating thread", 0, 0, LOG_FILE);
|
|
retVal = 1;
|
|
}
|
|
|
|
LogEvent(L"QOTD: Terminating thread", 0, 0, LOG_FILE);
|
|
ExitThread(retVal);
|
|
}
|