efl/src/lib/evil/evil_fcntl.c

139 lines
3.0 KiB
C

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif /* HAVE_CONFIG_H */
#include <stdio.h>
#include <sys/locking.h>
#include <winsock2.h> /* for ioctlsocket */
#include <io.h>
#include "evil_private.h"
/* SOCKET is defined as a uintptr_t, so passing a fd (int) is not a problem */
static int
_is_socket(SOCKET s)
{
fd_set rfds;
struct timeval tv;
tv.tv_sec = 0;
tv.tv_usec = 100;
FD_ZERO(&rfds);
FD_SET(s, &rfds);
return select(1, &rfds, NULL, NULL, &tv) != SOCKET_ERROR;
}
/*
* port of fcntl function
*
*/
EVIL_API int
fcntl(int fd, int cmd, ...)
{
va_list va;
int res = -1;
va_start (va, cmd);
if (cmd == F_GETFD)
{
HANDLE h;
DWORD flag;
h = _is_socket(fd) ? (HANDLE)(uintptr_t)fd : (HANDLE)_get_osfhandle(fd);
if (h == INVALID_HANDLE_VALUE)
return -1;
if (GetHandleInformation(h, &flag))
{
if (flag == HANDLE_FLAG_INHERIT)
return FD_CLOEXEC;
return 0;
}
return -1;
}
if (cmd == F_SETFD)
{
HANDLE h;
long flag;
h = _is_socket(fd) ? (HANDLE)(uintptr_t)fd : (HANDLE)_get_osfhandle(fd);
if (h == INVALID_HANDLE_VALUE)
return -1;
flag = va_arg(va, long);
if (flag == FD_CLOEXEC)
{
if (SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0))
return 0;
}
}
else if (cmd == F_GETFL)
{
/* does nothing*/
}
else if (cmd == F_SETFL)
{
long flag;
flag = va_arg(va, long);
if (flag & O_NONBLOCK)
{
u_long arg = 1;
int type;
int len;
int ret;
len = (int)sizeof(int);
ret = getsockopt((SOCKET)fd, SOL_SOCKET, SO_TYPE, (char *)&type, &len);
if (!ret && (type == SOCK_STREAM))
{
if (ioctlsocket((SOCKET)fd, FIONBIO, &arg) != SOCKET_ERROR)
res = 0;
}
}
}
else if ((cmd == F_SETLK) || (cmd == F_SETLKW))
{
struct flock *fl;
off_t length = 0;
long pos;
fl = va_arg(va, struct flock *);
if (fl->l_len == 0)
{
length = _lseek(fd, 0L, SEEK_END);
if (length != -1L)
res = 0;
}
fl->l_len = length - fl->l_start;
pos = _lseek(fd, fl->l_start, fl->l_whence);
if (pos != -1L)
res = 0;
if ((fl->l_type == F_RDLCK) || (fl->l_type == F_WRLCK))
{
if (cmd == F_SETLK)
res = _locking(fd, _LK_NBLCK, fl->l_len); /* if cannot be locked, we return immediatly */
else /* F_SETLKW */
res = _locking(fd, _LK_LOCK, fl->l_len); /* otherwise, we try several times */
}
if (fl->l_type == F_UNLCK)
res = _locking(fd, _LK_UNLCK, fl->l_len);
}
va_end(va);
return res;
}