226 lines
5.6 KiB
C
226 lines
5.6 KiB
C
|
/**
|
||
|
* WinPR: Windows Portable Runtime
|
||
|
* WinPR Logger
|
||
|
*
|
||
|
* Copyright © 2015 Thincast Technologies GmbH
|
||
|
* Copyright © 2015 David FORT <contact@hardening-consulting.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.
|
||
|
*/
|
||
|
|
||
|
#ifdef HAVE_CONFIG_H
|
||
|
#include "config.h"
|
||
|
#endif
|
||
|
|
||
|
#include <winpr/crt.h>
|
||
|
#include <winpr/environment.h>
|
||
|
#include <winpr/winsock.h>
|
||
|
|
||
|
#include "wlog.h"
|
||
|
|
||
|
struct _wLogUdpAppender
|
||
|
{
|
||
|
WLOG_APPENDER_COMMON();
|
||
|
char* host;
|
||
|
struct sockaddr targetAddr;
|
||
|
int targetAddrLen;
|
||
|
SOCKET sock;
|
||
|
};
|
||
|
typedef struct _wLogUdpAppender wLogUdpAppender;
|
||
|
|
||
|
static BOOL WLog_UdpAppender_Open(wLog* log, wLogAppender* appender)
|
||
|
{
|
||
|
wLogUdpAppender* udpAppender;
|
||
|
char addressString[256];
|
||
|
struct addrinfo hints;
|
||
|
struct addrinfo* result;
|
||
|
int status, addrLen;
|
||
|
char* colonPos;
|
||
|
|
||
|
if (!appender)
|
||
|
return FALSE;
|
||
|
|
||
|
udpAppender = (wLogUdpAppender*)appender;
|
||
|
|
||
|
if (udpAppender->targetAddrLen) /* already opened */
|
||
|
return TRUE;
|
||
|
|
||
|
colonPos = strchr(udpAppender->host, ':');
|
||
|
|
||
|
if (!colonPos)
|
||
|
return FALSE;
|
||
|
|
||
|
addrLen = (int)(colonPos - udpAppender->host);
|
||
|
memcpy(addressString, udpAppender->host, addrLen);
|
||
|
addressString[addrLen] = '\0';
|
||
|
ZeroMemory(&hints, sizeof(hints));
|
||
|
hints.ai_family = AF_INET;
|
||
|
hints.ai_socktype = SOCK_DGRAM;
|
||
|
status = getaddrinfo(addressString, colonPos + 1, &hints, &result);
|
||
|
|
||
|
if (status != 0)
|
||
|
return FALSE;
|
||
|
|
||
|
if (result->ai_addrlen > sizeof(udpAppender->targetAddr))
|
||
|
{
|
||
|
freeaddrinfo(result);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
memcpy(&udpAppender->targetAddr, result->ai_addr, result->ai_addrlen);
|
||
|
udpAppender->targetAddrLen = (int)result->ai_addrlen;
|
||
|
freeaddrinfo(result);
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL WLog_UdpAppender_Close(wLog* log, wLogAppender* appender)
|
||
|
{
|
||
|
if (!log || !appender)
|
||
|
return FALSE;
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL WLog_UdpAppender_WriteMessage(wLog* log, wLogAppender* appender, wLogMessage* message)
|
||
|
{
|
||
|
char prefix[WLOG_MAX_PREFIX_SIZE];
|
||
|
wLogUdpAppender* udpAppender;
|
||
|
|
||
|
if (!log || !appender || !message)
|
||
|
return FALSE;
|
||
|
|
||
|
udpAppender = (wLogUdpAppender*)appender;
|
||
|
message->PrefixString = prefix;
|
||
|
WLog_Layout_GetMessagePrefix(log, appender->Layout, message);
|
||
|
_sendto(udpAppender->sock, message->PrefixString, (int)strnlen(message->PrefixString, INT_MAX),
|
||
|
0, &udpAppender->targetAddr, udpAppender->targetAddrLen);
|
||
|
_sendto(udpAppender->sock, message->TextString, (int)strnlen(message->TextString, INT_MAX), 0,
|
||
|
&udpAppender->targetAddr, udpAppender->targetAddrLen);
|
||
|
_sendto(udpAppender->sock, "\n", 1, 0, &udpAppender->targetAddr, udpAppender->targetAddrLen);
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL WLog_UdpAppender_WriteDataMessage(wLog* log, wLogAppender* appender,
|
||
|
wLogMessage* message)
|
||
|
{
|
||
|
if (!log || !appender || !message)
|
||
|
return FALSE;
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL WLog_UdpAppender_WriteImageMessage(wLog* log, wLogAppender* appender,
|
||
|
wLogMessage* message)
|
||
|
{
|
||
|
if (!log || !appender || !message)
|
||
|
return FALSE;
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL WLog_UdpAppender_Set(wLogAppender* appender, const char* setting, void* value)
|
||
|
{
|
||
|
const char target[] = "target";
|
||
|
wLogUdpAppender* udpAppender = (wLogUdpAppender*)appender;
|
||
|
|
||
|
/* Just check the value string is not empty */
|
||
|
if (!value || (strnlen(value, 2) == 0))
|
||
|
return FALSE;
|
||
|
|
||
|
if (strncmp(target, setting, sizeof(target)))
|
||
|
return FALSE;
|
||
|
|
||
|
udpAppender->targetAddrLen = 0;
|
||
|
|
||
|
if (udpAppender->host)
|
||
|
free(udpAppender->host);
|
||
|
|
||
|
udpAppender->host = _strdup((const char*)value);
|
||
|
return (udpAppender->host != NULL) && WLog_UdpAppender_Open(NULL, appender);
|
||
|
}
|
||
|
|
||
|
static void WLog_UdpAppender_Free(wLogAppender* appender)
|
||
|
{
|
||
|
wLogUdpAppender* udpAppender;
|
||
|
|
||
|
if (appender)
|
||
|
{
|
||
|
udpAppender = (wLogUdpAppender*)appender;
|
||
|
|
||
|
if (udpAppender->sock != INVALID_SOCKET)
|
||
|
{
|
||
|
closesocket(udpAppender->sock);
|
||
|
udpAppender->sock = INVALID_SOCKET;
|
||
|
}
|
||
|
|
||
|
free(udpAppender->host);
|
||
|
free(udpAppender);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
wLogAppender* WLog_UdpAppender_New(wLog* log)
|
||
|
{
|
||
|
wLogUdpAppender* appender;
|
||
|
DWORD nSize;
|
||
|
LPCSTR name;
|
||
|
appender = (wLogUdpAppender*)calloc(1, sizeof(wLogUdpAppender));
|
||
|
|
||
|
if (!appender)
|
||
|
return NULL;
|
||
|
|
||
|
appender->Type = WLOG_APPENDER_UDP;
|
||
|
appender->Open = WLog_UdpAppender_Open;
|
||
|
appender->Close = WLog_UdpAppender_Close;
|
||
|
appender->WriteMessage = WLog_UdpAppender_WriteMessage;
|
||
|
appender->WriteDataMessage = WLog_UdpAppender_WriteDataMessage;
|
||
|
appender->WriteImageMessage = WLog_UdpAppender_WriteImageMessage;
|
||
|
appender->Free = WLog_UdpAppender_Free;
|
||
|
appender->Set = WLog_UdpAppender_Set;
|
||
|
appender->sock = _socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||
|
|
||
|
if (appender->sock == INVALID_SOCKET)
|
||
|
goto error_sock;
|
||
|
|
||
|
name = "WLOG_UDP_TARGET";
|
||
|
nSize = GetEnvironmentVariableA(name, NULL, 0);
|
||
|
|
||
|
if (nSize)
|
||
|
{
|
||
|
appender->host = (LPSTR)malloc(nSize);
|
||
|
|
||
|
if (!appender->host)
|
||
|
goto error_open;
|
||
|
|
||
|
if (GetEnvironmentVariableA(name, appender->host, nSize) != nSize - 1)
|
||
|
goto error_open;
|
||
|
|
||
|
if (!WLog_UdpAppender_Open(log, (wLogAppender*)appender))
|
||
|
goto error_open;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
appender->host = _strdup("127.0.0.1:20000");
|
||
|
|
||
|
if (!appender->host)
|
||
|
goto error_open;
|
||
|
}
|
||
|
|
||
|
return (wLogAppender*)appender;
|
||
|
error_open:
|
||
|
free(appender->host);
|
||
|
closesocket(appender->sock);
|
||
|
error_sock:
|
||
|
free(appender);
|
||
|
return NULL;
|
||
|
}
|