version 1.1.1.1.8.2, 2010/09/10 12:38:26
|
version 1.10.6.4, 2012/05/23 13:53:00
|
Line 1
|
Line 1
|
/************************************************************************* |
/************************************************************************* |
* (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com> | * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org> |
* by Michael Pounov <misho@openbsd-bg.org> | * by Michael Pounov <misho@elwix.org> |
* |
* |
* $Author$ |
* $Author$ |
* $Id$ |
* $Id$ |
* |
* |
*************************************************************************/ | ************************************************************************** |
| The ELWIX and AITNET software is distributed under the following |
| terms: |
| |
| All of the documentation and software included in the ELWIX and AITNET |
| Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org> |
| |
| Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 |
| by Michael Pounov <misho@elwix.org>. All rights reserved. |
| |
| Redistribution and use in source and binary forms, with or without |
| modification, are permitted provided that the following conditions |
| are met: |
| 1. Redistributions of source code must retain the above copyright |
| notice, this list of conditions and the following disclaimer. |
| 2. Redistributions in binary form must reproduce the above copyright |
| notice, this list of conditions and the following disclaimer in the |
| documentation and/or other materials provided with the distribution. |
| 3. All advertising materials mentioning features or use of this software |
| must display the following acknowledgement: |
| This product includes software developed by Michael Pounov <misho@elwix.org> |
| ELWIX - Embedded LightWeight unIX and its contributors. |
| 4. Neither the name of AITNET nor the names of its contributors |
| may be used to endorse or promote products derived from this software |
| without specific prior written permission. |
| |
| THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND |
| ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
| FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
| OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
| OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| SUCH DAMAGE. |
| */ |
#include "global.h" |
#include "global.h" |
|
|
|
|
int io_Debug; |
int io_Debug; |
|
mpool_t *io_mpool; |
|
|
|
|
|
/* Memory management */ |
|
|
|
void *(*io_malloc)(size_t) = malloc; |
|
void *(*io_calloc)(size_t, size_t) = calloc; |
|
void *(*io_realloc)(void*, size_t) = realloc; |
|
char *(*io_strdup)(const char*) = strdup; |
|
void (*io_free)(void*) = free; |
|
|
|
|
#pragma GCC visibility push(hidden) |
#pragma GCC visibility push(hidden) |
|
|
|
int use_mm; |
|
|
int io_Errno; |
int io_Errno; |
char io_Error[STRSIZ]; |
char io_Error[STRSIZ]; |
|
|
Line 21 char io_Error[STRSIZ];
|
Line 70 char io_Error[STRSIZ];
|
|
|
|
|
// io_GetErrno() Get error code of last operation |
// io_GetErrno() Get error code of last operation |
inline int io_GetErrno() | inline int |
| io_GetErrno() |
{ |
{ |
return io_Errno; |
return io_Errno; |
} |
} |
|
|
// io_GetError() Get error text of last operation |
// io_GetError() Get error text of last operation |
inline const char *io_GetError() | inline const char * |
| io_GetError() |
{ |
{ |
return io_Error; |
return io_Error; |
} |
} |
|
|
// io_SetErr() Set error to variables for internal use!!! |
// io_SetErr() Set error to variables for internal use!!! |
inline void io_SetErr(int eno, char *estr, ...) | inline void |
| io_SetErr(int eno, char *estr, ...) |
{ |
{ |
va_list lst; |
va_list lst; |
|
|
Line 44 inline void io_SetErr(int eno, char *estr, ...)
|
Line 96 inline void io_SetErr(int eno, char *estr, ...)
|
va_end(lst); |
va_end(lst); |
} |
} |
|
|
|
// io_mm_inuse() Check for memory management model |
|
inline int |
|
io_mm_inuse() |
|
{ |
|
return use_mm & IO_MPOOL; |
|
} |
|
|
|
|
|
// init libaitio routine |
|
void |
|
_init() |
|
{ |
|
#ifdef USE_MPOOL |
|
ioLibInit(IO_MPOOL, 0); |
|
#else |
|
ioLibInit(IO_SYSM, 0); |
|
#endif |
|
} |
|
|
|
// fini libaitio routine |
|
void |
|
_fini() |
|
{ |
|
ioLibFini(); |
|
} |
|
|
/* |
/* |
* ioPromptRead() Read data from input h[0] with prompt to output h[1] | * ioLibInit() - Init libaitio library memory management |
| * |
| * @mm = memory management (IO_SYSM or IO_MPOOL) |
| * @maxmem = memory limit |
| * return: -1 error or !=-1 used memory management model |
| */ |
| inline int |
| ioLibInit(int mm, u_long maxmem) |
| { |
| switch (mm) { |
| case IO_MPOOL: /* mpool */ |
| io_mpool = mpool_init(maxmem); |
| if (io_mpool) { |
| io_malloc = mpool_xmalloc; |
| io_calloc = mpool_xcalloc; |
| io_realloc = mpool_xrealloc; |
| io_strdup = mpool_xstrdup; |
| io_free = mpool_xfree; |
| use_mm = mm; |
| break; |
| } else { |
| #undef USE_MPOOL |
| } |
| case IO_SYSM: /* system */ |
| io_malloc = malloc; |
| io_calloc = calloc; |
| io_realloc = realloc; |
| io_strdup = strdup; |
| io_free = free; |
| use_mm = mm; |
| break; |
| default: /* not supported */ |
| io_SetErr(EINVAL, "Not supported memory management"); |
| return -1; |
| } |
| |
| return use_mm; |
| } |
| |
| /* |
| * ioLibFini() - Finish libaitio library memory management |
| * |
| * return: none |
| */ |
| inline void |
| ioLibFini() |
| { |
| switch (use_mm) { |
| case 1: |
| mpool_destroy(&io_mpool); |
| |
| io_malloc = malloc; |
| io_calloc = calloc; |
| io_realloc = realloc; |
| io_strdup = strdup; |
| io_free = free; |
| use_mm = IO_SYSM; |
| break; |
| } |
| } |
| |
| |
| /* |
| * ioPromptRead() - Read data from input h[0] with prompt to output h[1] |
| * |
* @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout |
* @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout |
* @csPrompt = Prompt before input, may be NULL |
* @csPrompt = Prompt before input, may be NULL |
* @psData = Readed data |
* @psData = Readed data |
* @dataLen = Length of data |
* @dataLen = Length of data |
* return: 0 EOF; -1 error:: can`t read; >0 count of readed chars |
* return: 0 EOF; -1 error:: can`t read; >0 count of readed chars |
*/ |
*/ |
int ioPromptRead(int *h, const char *csPrompt, char * __restrict psData, int dataLen) | int |
| ioPromptRead(int *h, const char *csPrompt, char * __restrict psData, int dataLen) |
{ |
{ |
int ok = 0; |
int ok = 0; |
FILE *inp, *out; |
FILE *inp, *out; |
Line 98 int ioPromptRead(int *h, const char *csPrompt, char *
|
Line 240 int ioPromptRead(int *h, const char *csPrompt, char *
|
} |
} |
|
|
/* |
/* |
* ioPromptPassword() Read password from input h[0] with prompt to output h[1] | * ioPromptPassword() - Read password from input h[0] with prompt to output h[1] |
| * |
* @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout |
* @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout |
* @csPrompt = Prompt before input, may be NULL |
* @csPrompt = Prompt before input, may be NULL |
* @psPass = Readed password |
* @psPass = Readed password |
Line 106 int ioPromptRead(int *h, const char *csPrompt, char *
|
Line 249 int ioPromptRead(int *h, const char *csPrompt, char *
|
* @confirm = Confirm password, 0 - get password, !=0 Ask for confirmation |
* @confirm = Confirm password, 0 - get password, !=0 Ask for confirmation |
* return: 0 EOF; -1 error:: can`t read; >0 count of readed chars |
* return: 0 EOF; -1 error:: can`t read; >0 count of readed chars |
*/ |
*/ |
int ioPromptPassword(int *h, const char *csPrompt, char * __restrict psPass, int passLen, int confirm) | int |
| ioPromptPassword(int *h, const char *csPrompt, char * __restrict psPass, int passLen, int confirm) |
{ |
{ |
int ret, ok = 0; |
int ret, ok = 0; |
FILE *inp, *out; |
FILE *inp, *out; |
Line 183 next:
|
Line 327 next:
|
} |
} |
|
|
/* |
/* |
* ioRegexVerify() Function for verify data match in regex expression | * ioRegexVerify() - Function for verify data match in regex expression |
| * |
* @csRegex = Regulare expression pattern |
* @csRegex = Regulare expression pattern |
* @csData = Data for check and verify |
* @csData = Data for check and verify |
* @startPos = Return start positions |
* @startPos = Return start positions |
* @endPos = Return end positions |
* @endPos = Return end positions |
* return: NULL not match or error; !=NULL begin of matched data |
* return: NULL not match or error; !=NULL begin of matched data |
*/ |
*/ |
const char *ioRegexVerify(const char *csRegex, const char *csData, int *startPos, int *endPos) | const char * |
| ioRegexVerify(const char *csRegex, const char *csData, int *startPos, int *endPos) |
{ |
{ |
regex_t re; |
regex_t re; |
regmatch_t match; |
regmatch_t match; |
Line 203 const char *ioRegexVerify(const char *csRegex, const c
|
Line 349 const char *ioRegexVerify(const char *csRegex, const c
|
|
|
if ((ret = regcomp(&re, csRegex, REG_EXTENDED))) { |
if ((ret = regcomp(&re, csRegex, REG_EXTENDED))) { |
regerror(ret, &re, szErr, STRSIZ); |
regerror(ret, &re, szErr, STRSIZ); |
io_SetErr(ret, "Error:: %s\n", szErr); | io_SetErr(ret, "%s", szErr); |
regfree(&re); |
regfree(&re); |
return NULL; |
return NULL; |
} |
} |
Line 221 const char *ioRegexVerify(const char *csRegex, const c
|
Line 367 const char *ioRegexVerify(const char *csRegex, const c
|
|
|
if (ret) { |
if (ret) { |
regerror(ret, &re, szErr, STRSIZ); |
regerror(ret, &re, szErr, STRSIZ); |
io_SetErr(ret, "Error:: %s\n", szErr); | io_SetErr(ret, "%s", szErr); |
pos = NULL; |
pos = NULL; |
} |
} |
|
|
Line 230 const char *ioRegexVerify(const char *csRegex, const c
|
Line 376 const char *ioRegexVerify(const char *csRegex, const c
|
} |
} |
|
|
/* |
/* |
* ioRegexGet() Function for get data match in regex expression | * ioRegexGet() - Function for get data match in regex expression |
| * |
* @csRegex = Regulare expression pattern |
* @csRegex = Regulare expression pattern |
* @csData = Data from get |
* @csData = Data from get |
* @psString = Returned string if match |
* @psString = Returned string if match |
* @strLen = Length of string |
* @strLen = Length of string |
* return: 0 not match; >0 count of returned chars |
* return: 0 not match; >0 count of returned chars |
*/ |
*/ |
int ioRegexGet(const char *csRegex, const char *csData, char * __restrict psString, int strLen) | int |
| ioRegexGet(const char *csRegex, const char *csData, char * __restrict psString, int strLen) |
{ |
{ |
int sp, ep, len; |
int sp, ep, len; |
const char *str; |
const char *str; |
Line 259 int ioRegexGet(const char *csRegex, const char *csData
|
Line 407 int ioRegexGet(const char *csRegex, const char *csData
|
} |
} |
|
|
/* |
/* |
* ioRegexReplace() Function for replace data match in regex expression with newdata | * ioRegexReplace() - Function for replace data match in regex expression with newdata |
| * |
* @csRegex = Regulare expression pattern |
* @csRegex = Regulare expression pattern |
* @csData = Source data |
* @csData = Source data |
* @csNew = Data for replace |
* @csNew = Data for replace |
* return: NULL not match or error; !=NULL allocated new string, must be free after use! | * return: NULL not match or error; !=NULL allocated new string, must be xfree after use! |
*/ |
*/ |
char *ioRegexReplace(const char *csRegex, const char *csData, const char *csNew) | char * |
| ioRegexReplace(const char *csRegex, const char *csData, const char *csNew) |
{ |
{ |
int sp, ep, len; |
int sp, ep, len; |
char *str = NULL; |
char *str = NULL; |
Line 278 char *ioRegexReplace(const char *csRegex, const char *
|
Line 428 char *ioRegexReplace(const char *csRegex, const char *
|
|
|
// ___ before match |
// ___ before match |
len = sp + 1; |
len = sp + 1; |
str = malloc(len); | str = xmalloc(len); |
if (!str) { |
if (!str) { |
LOGERR; |
LOGERR; |
return NULL; |
return NULL; |
Line 287 char *ioRegexReplace(const char *csRegex, const char *
|
Line 437 char *ioRegexReplace(const char *csRegex, const char *
|
// * replace match * |
// * replace match * |
if (csNew) { |
if (csNew) { |
len += strlen(csNew); |
len += strlen(csNew); |
str = realloc(str, len); | str = xrealloc(str, len); |
if (!str) { |
if (!str) { |
LOGERR; |
LOGERR; |
return NULL; |
return NULL; |
Line 296 char *ioRegexReplace(const char *csRegex, const char *
|
Line 446 char *ioRegexReplace(const char *csRegex, const char *
|
} |
} |
// after match ___ |
// after match ___ |
len += strlen(csData) - ep; |
len += strlen(csData) - ep; |
str = realloc(str, len); | str = xrealloc(str, len); |
if (!str) { |
if (!str) { |
LOGERR; |
LOGERR; |
return NULL; |
return NULL; |
Line 306 char *ioRegexReplace(const char *csRegex, const char *
|
Line 456 char *ioRegexReplace(const char *csRegex, const char *
|
return str; |
return str; |
} |
} |
|
|
|
/* |
|
* ioStrAst() - Function for evaluate string like asterisk variable "{text[:[-]#[:#]]}" |
|
* |
|
* @csString = Input string |
|
* return: NULL error, !=NULL Allocated new string evaluated from input string, must be xfree() |
|
*/ |
|
char * |
|
ioStrAst(const char *csString) |
|
{ |
|
char *ext, *str, *out = NULL; |
|
int e[2] = { 0 }; |
|
|
|
if (!csString) |
|
return NULL; |
|
|
|
if (!strchr(csString, '{') || !strrchr(csString, '}')) { |
|
memset(io_Error, 0, STRSIZ); |
|
snprintf(io_Error, STRSIZ, "Invalid input string format ... " |
|
"must be like {text[:[-]#[:#]]}"); |
|
io_Errno = EINVAL; |
|
return NULL; |
|
} else { |
|
str = strdup(strchr(csString, '{') + 1); |
|
*strrchr(str, '}') = 0; |
|
} |
|
|
|
if ((ext = strchr(str, ':'))) { |
|
*ext++ = 0; |
|
e[0] = strtol(ext, NULL, 0); |
|
if ((ext = strchr(ext, ':'))) |
|
e[1] = strtol(++ext, NULL, 0); |
|
|
|
/* make cut prefix */ |
|
if (e[0] >= 0) |
|
ext = str + e[0]; |
|
else |
|
ext = str + strlen(str) + e[0]; |
|
/* make cut suffix */ |
|
if (e[1] > 0) |
|
*(ext + e[1]) = 0; |
|
} else |
|
/* ok, clear show */ |
|
ext = str; |
|
|
|
out = strdup(ext); |
|
xfree(str); |
|
|
|
return out; |
|
} |
|
|
|
|
/* |
/* |
* ioMkDir() Function for racursive directory creation and validation | * ioMkDir() - Function for racursive directory creation and validation |
| * |
* @csDir = Full directory path |
* @csDir = Full directory path |
* @mode = Mode for directory creation if missing dir |
* @mode = Mode for directory creation if missing dir |
* return: -1 error, 0 directory path exist, >0 created missing dirs |
* return: -1 error, 0 directory path exist, >0 created missing dirs |
Line 350 ioMkDir(const char *csDir, int mode)
|
Line 551 ioMkDir(const char *csDir, int mode)
|
} |
} |
end: |
end: |
chdir(szOld); |
chdir(szOld); |
free(str); | xfree(str); |
return cx; |
return cx; |
} |
} |
|
|
|
/* |
|
* ioWatchDirLoop() - Function for watching changes in directory and fire callback |
|
* |
|
* @csDir = Full directory path |
|
* @callback = Callback if raise event! nOp -1 delete, 0 change/move, 1 create |
|
* return: -1 error, !=-1 ok, number of total signaled events |
|
*/ |
|
int |
|
ioWatchDirLoop(const char *csDir, int (*callback)(const char *csName, int nOp)) |
|
{ |
|
glob_t g[2] = {{ 0 }, { 0 }}; |
|
int d, kq, n = 0; |
|
register int j, i; |
|
struct kevent req, chg; |
|
char wrk[MAXPATHLEN * 2], str[MAXPATHLEN] = { 0 }; |
|
|
|
if (!csDir || !callback) |
|
return 0; |
|
|
|
strlcpy(str, csDir, MAXPATHLEN); |
|
strlcat(str, "/*", MAXPATHLEN); |
|
|
|
kq = kqueue(); |
|
if (kq == -1) { |
|
LOGERR; |
|
return -1; |
|
} |
|
d = open(csDir, O_RDONLY); |
|
if (d == -1) { |
|
LOGERR; |
|
close(kq); |
|
return -1; |
|
} |
|
|
|
EV_SET(&req, d, EVFILT_VNODE, EV_ADD | EV_CLEAR, NOTE_WRITE, 0, 0); |
|
|
|
if ((n = glob(str, GLOB_NOCHECK, NULL, &g[0]))) { |
|
LOGERR; |
|
close(d); |
|
close(kq); |
|
return -1; |
|
} /*else |
|
ioDEBUG(3, "Start files %d in %s\n", g[0].gl_matchc, str);*/ |
|
|
|
while (kevent(kq, &req, 1, &chg, 1, NULL) > 0) { |
|
/*ioDEBUG(1, "Event:: req=0x%x -> chg=0x%x data=%x\n", req.fflags, chg.fflags, chg.data);*/ |
|
|
|
if (!glob(str, GLOB_NOCHECK, NULL, &g[1])) { |
|
/*ioDEBUG(3, "Diffs %d <> %d\n", g[0].gl_matchc, g[1].gl_matchc);*/ |
|
|
|
if (g[0].gl_matchc != g[1].gl_matchc) { |
|
/* find new items */ |
|
for (j = 0; j < g[1].gl_matchc; j++) { |
|
for (i = 0; i < g[0].gl_matchc; i++) |
|
if (!strcmp(g[0].gl_pathv[i], g[1].gl_pathv[j])) |
|
break; |
|
if (i == g[0].gl_matchc) { |
|
if (callback(g[1].gl_pathv[j], 1) < 0) |
|
break; |
|
else |
|
n++; |
|
} |
|
} |
|
/* find del items */ |
|
for (j = 0; j < g[0].gl_matchc; j++) { |
|
for (i = 0; i < g[1].gl_matchc; i++) |
|
if (!strcmp(g[1].gl_pathv[i], g[0].gl_pathv[j])) |
|
break; |
|
if (i == g[1].gl_matchc) { |
|
if (callback(g[0].gl_pathv[j], -1) < 0) |
|
break; |
|
else |
|
n++; |
|
} |
|
} |
|
} else { |
|
/* find chg from items */ |
|
for (j = 0; j < g[0].gl_matchc; j++) { |
|
for (i = 0; i < g[1].gl_matchc; i++) |
|
if (!strcmp(g[1].gl_pathv[i], g[0].gl_pathv[j])) |
|
break; |
|
if (i == g[1].gl_matchc) { |
|
strlcpy(wrk, g[0].gl_pathv[j], sizeof wrk); |
|
strlcat(wrk, ":", sizeof wrk); |
|
} |
|
} |
|
/* find chg to items */ |
|
for (j = 0; j < g[1].gl_matchc; j++) { |
|
for (i = 0; i < g[0].gl_matchc; i++) |
|
if (!strcmp(g[0].gl_pathv[i], g[1].gl_pathv[j])) |
|
break; |
|
if (i == g[0].gl_matchc) { |
|
strlcat(wrk, g[1].gl_pathv[j], sizeof wrk); |
|
if (callback(wrk, 0) < 0) |
|
break; |
|
else |
|
n++; |
|
} |
|
} |
|
} |
|
|
|
globfree(&g[0]); |
|
g[0] = g[1]; |
|
} |
|
} |
|
|
|
globfree(&g[0]); |
|
close(d); |
|
close(kq); |
|
return n; |
|
} |
|
|
|
/* |
|
* ioCreatePIDFile() - Create PID file |
|
* |
|
* @csName = PID filename |
|
* @ifExists = !=0 if filename exists return error |
|
* return: -1 error or 0 ok |
|
*/ |
|
inline int |
|
ioCreatePIDFile(const char *csName, int ifExists) |
|
{ |
|
int fd; |
|
char str[STRSIZ] = { 0 }; |
|
|
|
if (!csName) |
|
return -1; |
|
|
|
fd = open(csName, O_WRONLY | O_CREAT | (ifExists ? O_EXCL : 0), 0644); |
|
if (fd == -1) { |
|
LOGERR; |
|
return -1; |
|
} |
|
snprintf(str, sizeof str, "%d", getpid()); |
|
write(fd, str, strlen(str)); |
|
close(fd); |
|
return 0; |
|
} |
|
|
|
|
|
/* |
|
* ioSendFile() - AITNET sendfile() userland implementation, not dependant from OS |
|
* |
|
* @s = socket |
|
* @csFile = file for send |
|
* @sendLen = bytes to send, if 0 send all data |
|
* @offset = start file offset |
|
* @sndbuf = SO_SNDBUF value, if 0 use default |
|
* return: 0 error, >0 ok, sended bytes |
|
*/ |
|
size_t |
|
ioSendFile(int s, const char *csFile, size_t sendLen, off_t offset, int sndbuf) |
|
{ |
|
void *addr; |
|
int fd; |
|
size_t len = 0; |
|
register size_t off = 0; |
|
|
|
if (!csFile) |
|
return 0; |
|
|
|
if (sndbuf) |
|
if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof sndbuf) == -1) { |
|
LOGERR; |
|
return 0; |
|
} |
|
|
|
fd = open(csFile, O_RDONLY); |
|
if (fd == -1) { |
|
LOGERR; |
|
return 0; |
|
} |
|
if (!sendLen) { |
|
sendLen = lseek(fd, 0, SEEK_END); |
|
if (sendLen == -1) { |
|
LOGERR; |
|
close(fd); |
|
return 0; |
|
} |
|
} |
|
addr = mmap(NULL, sendLen, PROT_READ, MAP_SHARED, fd, offset); |
|
if (addr == MAP_FAILED) { |
|
LOGERR; |
|
close(fd); |
|
return 0; |
|
} else |
|
close(fd); |
|
|
|
while (off < sendLen && (len = write(s, addr + off, sendLen - off)) != -1) |
|
off += len; |
|
if (len == -1) { |
|
LOGERR; |
|
munmap(addr, sendLen); |
|
return 0; |
|
} else |
|
len = off; |
|
|
|
if (len != sendLen) { |
|
io_SetErr(ECANCELED, "Different sizes - request %u bytes, actually sended %u bytes\n", |
|
sendLen, len); |
|
len ^= len; |
|
} |
|
|
|
munmap(addr, sendLen); |
|
return len; |
|
} |
|
|
|
/* |
|
* ioRecvFile() - Receive file from socket, fastest (zero-copy) way |
|
* |
|
* @s = socket |
|
* @csFile = file for receive |
|
* @recvLen = receive bytes |
|
* @over = overwrite file if exists with mode like 0644 |
|
* @rcvbuf = SO_RCVBUF value, if 0 use default |
|
* return: 0 error, >0 ok, received bytes |
|
*/ |
|
size_t |
|
ioRecvFile(int s, const char *csFile, size_t recvLen, int over, int rcvbuf) |
|
{ |
|
void *addr; |
|
int fd; |
|
size_t len = 0; |
|
register size_t off = 0; |
|
struct pollfd pfd = { s, POLLIN | POLLPRI, 0 }; |
|
|
|
if (!csFile || !recvLen) |
|
return 0; |
|
if (!over && !access(csFile, F_OK)) |
|
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); |
|
if (fd == -1) { |
|
LOGERR; |
|
unlink(csFile); |
|
return 0; |
|
} |
|
if (ftruncate(fd, recvLen) == -1) { |
|
LOGERR; |
|
close(fd); |
|
unlink(csFile); |
|
return 0; |
|
} |
|
addr = mmap(NULL, recvLen, PROT_WRITE, MAP_SHARED, fd, 0); |
|
if (addr == MAP_FAILED) { |
|
LOGERR; |
|
close(fd); |
|
unlink(csFile); |
|
return 0; |
|
} else |
|
close(fd); |
|
|
|
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) { |
|
LOGERR; |
|
munmap(addr, recvLen); |
|
unlink(csFile); |
|
return 0; |
|
} else |
|
len = off; |
|
|
|
if (len != recvLen) |
|
io_SetErr(EAGAIN, "Different sizes - request %u bytes, actually received %u bytes\n", |
|
recvLen, len); |
|
|
|
munmap(addr, recvLen); |
|
return len; |
|
} |