225 lines
4.0 KiB
C
225 lines
4.0 KiB
C
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif /* HAVE_CONFIG_H */
|
|
|
|
#ifdef HAVE_ERRNO_H
|
|
#include <errno.h>
|
|
#endif /* HAVE_ERRNO_H */
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <winsock2.h>
|
|
#undef WIN32_LEAN_AND_MEAN
|
|
|
|
#include "Evil.h"
|
|
#include "evil_private.h"
|
|
|
|
/*
|
|
* Process identifer related functions
|
|
*
|
|
*/
|
|
|
|
pid_t
|
|
getpid(void)
|
|
{
|
|
return (pid_t)GetCurrentProcessId();
|
|
}
|
|
|
|
char *
|
|
evil_getcwd(char *buffer, size_t size)
|
|
{
|
|
#if defined(__CEGCC__) || defined(__MINGW32CE__)
|
|
wchar_t wpath[PATH_MAX];
|
|
char *cpath;
|
|
char *delim;
|
|
DWORD ret = 0;
|
|
|
|
if (size <= 0)
|
|
return NULL;
|
|
|
|
ret = GetModuleFileName(GetModuleHandle(NULL), (LPWSTR)&wpath, PATH_MAX);
|
|
|
|
if (!ret)
|
|
{
|
|
_evil_error_display(__FUNCTION__, ret);
|
|
return NULL;
|
|
}
|
|
|
|
cpath = evil_wchar_to_char(wpath);
|
|
if (!cpath)
|
|
return NULL;
|
|
|
|
if (strlen(cpath) >= (size - 1))
|
|
{
|
|
free(cpath);
|
|
return NULL;
|
|
}
|
|
|
|
delim = strrchr(cpath, '\\');
|
|
if (delim)
|
|
*delim = '\0';
|
|
|
|
if (!buffer)
|
|
{
|
|
buffer = (char *)malloc(sizeof(char) * size);
|
|
if (!buffer)
|
|
{
|
|
free(cpath);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
strcpy(buffer, cpath);
|
|
free(cpath);
|
|
|
|
return buffer;
|
|
#else
|
|
return _getcwd(buffer, size);
|
|
#endif /* ! __CEGCC__ && ! __MINGW32CE__ */
|
|
}
|
|
|
|
|
|
/*
|
|
* Sockets and pipe related functions
|
|
*
|
|
*/
|
|
|
|
int
|
|
evil_sockets_init(void)
|
|
{
|
|
WSADATA wsa_data;
|
|
|
|
return (WSAStartup(MAKEWORD(2, 2), &wsa_data) == 0) ? 1 : 0;
|
|
}
|
|
|
|
void
|
|
evil_sockets_shutdown(void)
|
|
{
|
|
WSACleanup();
|
|
}
|
|
|
|
/*
|
|
* The code of the following functions has been kindly offered
|
|
* by Tor Lillqvist.
|
|
*/
|
|
int
|
|
evil_pipe(int *fds)
|
|
{
|
|
struct sockaddr_in saddr;
|
|
struct timeval tv;
|
|
SOCKET temp;
|
|
SOCKET socket1 = INVALID_SOCKET;
|
|
SOCKET socket2 = INVALID_SOCKET;
|
|
u_long arg;
|
|
fd_set read_set;
|
|
fd_set write_set;
|
|
int len;
|
|
|
|
temp = socket (AF_INET, SOCK_STREAM, 0);
|
|
|
|
if (temp == INVALID_SOCKET)
|
|
goto out0;
|
|
|
|
arg = 1;
|
|
if (ioctlsocket (temp, FIONBIO, &arg) == SOCKET_ERROR)
|
|
goto out0;
|
|
|
|
memset (&saddr, 0, sizeof (saddr));
|
|
saddr.sin_family = AF_INET;
|
|
saddr.sin_port = 0;
|
|
saddr.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
|
|
|
|
if (bind (temp, (struct sockaddr *)&saddr, sizeof (saddr)))
|
|
goto out0;
|
|
|
|
if (listen (temp, 1) == SOCKET_ERROR)
|
|
goto out0;
|
|
|
|
len = sizeof (saddr);
|
|
if (getsockname (temp, (struct sockaddr *)&saddr, &len))
|
|
goto out0;
|
|
|
|
socket1 = socket (AF_INET, SOCK_STREAM, 0);
|
|
|
|
if (socket1 == INVALID_SOCKET)
|
|
goto out0;
|
|
|
|
arg = 1;
|
|
if (ioctlsocket (socket1, FIONBIO, &arg) == SOCKET_ERROR)
|
|
goto out1;
|
|
|
|
if ((connect (socket1, (struct sockaddr *)&saddr, len) == SOCKET_ERROR) &&
|
|
(WSAGetLastError () != WSAEWOULDBLOCK))
|
|
goto out1;
|
|
|
|
FD_ZERO (&read_set);
|
|
FD_SET (temp, &read_set);
|
|
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = 0;
|
|
|
|
if (select (0, &read_set, NULL, NULL, NULL) == SOCKET_ERROR)
|
|
goto out1;
|
|
|
|
if (!FD_ISSET (temp, &read_set))
|
|
goto out1;
|
|
|
|
socket2 = accept (temp, (struct sockaddr *) &saddr, &len);
|
|
if (socket2 == INVALID_SOCKET)
|
|
goto out1;
|
|
|
|
FD_ZERO (&write_set);
|
|
FD_SET (socket1, &write_set);
|
|
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = 0;
|
|
|
|
if (select (0, NULL, &write_set, NULL, NULL) == SOCKET_ERROR)
|
|
goto out2;
|
|
|
|
if (!FD_ISSET (socket1, &write_set))
|
|
goto out2;
|
|
|
|
arg = 0;
|
|
if (ioctlsocket (socket1, FIONBIO, &arg) == SOCKET_ERROR)
|
|
goto out2;
|
|
|
|
arg = 0;
|
|
if (ioctlsocket (socket2, FIONBIO, &arg) == SOCKET_ERROR)
|
|
goto out2;
|
|
|
|
fds[0] = socket1;
|
|
fds[1] = socket2;
|
|
|
|
closesocket (temp);
|
|
|
|
return 0;
|
|
|
|
out2:
|
|
closesocket (socket2);
|
|
out1:
|
|
closesocket (socket1);
|
|
out0:
|
|
closesocket (temp);
|
|
|
|
fds[0] = INVALID_SOCKET;
|
|
fds[1] = INVALID_SOCKET;
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
/*
|
|
* Exec related functions
|
|
*
|
|
*/
|
|
|
|
#if defined (_WIN32_WCE) && ! defined (__CEGCC__)
|
|
|
|
int execvp( const char *file, char *const argv[])
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
#endif /* _WIN32_WCE && ! __CEGCC__ */
|