|
version 1.7.2.1, 2011/11/02 13:13:48
|
version 1.7.2.2, 2011/11/02 16:10:25
|
|
Line 592 ioCreatePIDFile(const char *csName, int ifExists)
|
Line 592 ioCreatePIDFile(const char *csName, int ifExists)
|
| * @csFile = file for send |
* @csFile = file for send |
| * @sendLen = bytes to send, if 0 send all data |
* @sendLen = bytes to send, if 0 send all data |
| * @offset = start file offset |
* @offset = start file offset |
| |
* @sndbuf = SO_SNDBUF value, if 0 use default |
| * return: 0 error, >0 ok, sended bytes |
* return: 0 error, >0 ok, sended bytes |
| */ |
*/ |
| size_t |
size_t |
| ioSendFile(int s, const char *csFile, size_t sendLen, off_t offset) | ioSendFile(int s, const char *csFile, size_t sendLen, off_t offset, int sndbuf) |
| { |
{ |
| void *addr; |
void *addr; |
| int fd; |
int fd; |
| size_t len; | size_t len = 0; |
| | register size_t off = 0; |
| |
|
| if (!csFile) |
if (!csFile) |
| return 0; |
return 0; |
| |
|
| |
if (sndbuf) |
| |
if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof sndbuf) == -1) { |
| |
LOGERR; |
| |
return 0; |
| |
} |
| |
|
| fd = open(csFile, O_RDONLY); |
fd = open(csFile, O_RDONLY); |
| if (fd == -1) { |
if (fd == -1) { |
| LOGERR; |
LOGERR; |
|
Line 625 ioSendFile(int s, const char *csFile, size_t sendLen,
|
Line 633 ioSendFile(int s, const char *csFile, size_t sendLen,
|
| } else |
} else |
| close(fd); |
close(fd); |
| |
|
| len = write(s, addr, sendLen); | while (off < sendLen && (len = write(s, addr + off, sendLen - off)) != -1) |
| | off += len; |
| if (len == -1) { |
if (len == -1) { |
| LOGERR; |
LOGERR; |
| munmap(addr, sendLen); |
munmap(addr, sendLen); |
| return 0; |
return 0; |
| } | } else |
| | len = off; |
| |
|
| if (len != sendLen) { |
if (len != sendLen) { |
| io_SetErr(ECANCELED, "Different sizes - request %u bytes, actually sended %u bytes\n", |
io_SetErr(ECANCELED, "Different sizes - request %u bytes, actually sended %u bytes\n", |
|
Line 648 ioSendFile(int s, const char *csFile, size_t sendLen,
|
Line 658 ioSendFile(int s, const char *csFile, size_t sendLen,
|
| * @csFile = file for receive |
* @csFile = file for receive |
| * @recvLen = receive bytes |
* @recvLen = receive bytes |
| * @over = overwrite file if exists with mode like 0644 |
* @over = overwrite file if exists with mode like 0644 |
| |
* @rcvbuf = SO_RCVBUF value, if 0 use default |
| * return: 0 error, >0 ok, received bytes |
* return: 0 error, >0 ok, received bytes |
| */ |
*/ |
| size_t |
size_t |
| ioRecvFile(int s, const char *csFile, size_t recvLen, int over) | ioRecvFile(int s, const char *csFile, size_t recvLen, int over, int rcvbuf) |
| { |
{ |
| void *addr; |
void *addr; |
| int fd; |
int fd; |
| size_t len; | size_t len = 0; |
| | register size_t off = 0; |
| | struct pollfd pfd = { s, POLLIN | POLLPRI, 0 }; |
| |
|
| if (!csFile || !recvLen) |
if (!csFile || !recvLen) |
| return 0; |
return 0; |
| if (!over && !access(csFile, F_OK)) |
if (!over && !access(csFile, F_OK)) |
| return 0; |
return 0; |
| |
|
| |
if (rcvbuf) |
| |
if (setsockopt(s, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof rcvbuf) == -1) { |
| |
LOGERR; |
| |
return 0; |
| |
} |
| |
|
| fd = open(csFile, O_WRONLY | O_CREAT | O_TRUNC, over); |
fd = open(csFile, O_WRONLY | O_CREAT | O_TRUNC, over); |
| if (fd == -1) { |
if (fd == -1) { |
| LOGERR; |
LOGERR; |
|
Line 689 ioRecvFile(int s, const char *csFile, size_t recvLen,
|
Line 708 ioRecvFile(int s, const char *csFile, size_t recvLen,
|
| } else |
} else |
| close(fd); |
close(fd); |
| |
|
| len = read(s, addr, recvLen); | while (off < recvLen && poll(&pfd, 1, RECV_TIMEOUT) != -1) |
| | while (off < recvLen && (len = read(s, addr + off, recvLen - off)) != -1) |
| | off += len; |
| if (len == -1) { |
if (len == -1) { |
| LOGERR; |
LOGERR; |
| munmap(addr, recvLen); |
munmap(addr, recvLen); |
| unlink(csFile); |
unlink(csFile); |
| return 0; |
return 0; |
| } | } else |
| | len = off; |
| |
|
| if (len != recvLen) |
if (len != recvLen) |
| io_SetErr(EAGAIN, "Different sizes - request %u bytes, actually received %u bytes\n", |
io_SetErr(EAGAIN, "Different sizes - request %u bytes, actually received %u bytes\n", |