--- libaitio/src/aitio.c 2012/03/27 21:37:56 1.8.6.1 +++ libaitio/src/aitio.c 2012/05/23 12:16:13 1.10.6.3 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: aitio.c,v 1.8.6.1 2012/03/27 21:37:56 misho Exp $ +* $Id: aitio.c,v 1.10.6.3 2012/05/23 12:16:13 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -51,6 +51,8 @@ int io_Debug; #pragma GCC visibility push(hidden) +int io_mpool; + int io_Errno; char io_Error[STRSIZ]; @@ -84,9 +86,34 @@ io_SetErr(int eno, char *estr, ...) va_end(lst); } +// mpool_inuse() Check for mpool usage +inline int +mpool_inuse() +{ + return io_mpool; +} + +// init libaitio routine +void +_init() +{ +#ifdef USE_MPOOL + io_mpool = 42; +#else + io_mpool = 0; +#endif +} + +// fini libaitio routine +void +_fini() +{ +} + + /* - * ioPromptRead() Read data from input h[0] with prompt to output h[1] + * 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 * @csPrompt = Prompt before input, may be NULL @@ -140,7 +167,7 @@ ioPromptRead(int *h, const char *csPrompt, char * __re } /* - * 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 * @csPrompt = Prompt before input, may be NULL @@ -227,7 +254,7 @@ next: } /* - * ioRegexVerify() Function for verify data match in regex expression + * ioRegexVerify() - Function for verify data match in regex expression * * @csRegex = Regulare expression pattern * @csData = Data for check and verify @@ -249,7 +276,7 @@ ioRegexVerify(const char *csRegex, const char *csData, if ((ret = regcomp(&re, csRegex, REG_EXTENDED))) { regerror(ret, &re, szErr, STRSIZ); - io_SetErr(ret, "Error:: %s\n", szErr); + io_SetErr(ret, "%s", szErr); regfree(&re); return NULL; } @@ -267,7 +294,7 @@ ioRegexVerify(const char *csRegex, const char *csData, if (ret) { regerror(ret, &re, szErr, STRSIZ); - io_SetErr(ret, "Error:: %s\n", szErr); + io_SetErr(ret, "%s", szErr); pos = NULL; } @@ -276,7 +303,7 @@ ioRegexVerify(const char *csRegex, const char *csData, } /* - * ioRegexGet() Function for get data match in regex expression + * ioRegexGet() - Function for get data match in regex expression * * @csRegex = Regulare expression pattern * @csData = Data from get @@ -307,12 +334,12 @@ ioRegexGet(const char *csRegex, const char *csData, ch } /* - * 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 * @csData = Source data * @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) @@ -328,7 +355,7 @@ ioRegexReplace(const char *csRegex, const char *csData // ___ before match len = sp + 1; - str = malloc(len); + str = xmalloc(len); if (!str) { LOGERR; return NULL; @@ -337,7 +364,7 @@ ioRegexReplace(const char *csRegex, const char *csData // * replace match * if (csNew) { len += strlen(csNew); - str = realloc(str, len); + str = xrealloc(str, len); if (!str) { LOGERR; return NULL; @@ -346,7 +373,7 @@ ioRegexReplace(const char *csRegex, const char *csData } // after match ___ len += strlen(csData) - ep; - str = realloc(str, len); + str = xrealloc(str, len); if (!str) { LOGERR; return NULL; @@ -357,10 +384,10 @@ ioRegexReplace(const char *csRegex, const char *csData } /* - * ioStrAst() Function for evaluate string like asterisk variable "{text[:[-]#[:#]]}" + * 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 free() + * return: NULL error, !=NULL Allocated new string evaluated from input string, must be xfree() */ char * ioStrAst(const char *csString) @@ -373,7 +400,7 @@ ioStrAst(const char *csString) if (!strchr(csString, '{') || !strrchr(csString, '}')) { memset(io_Error, 0, STRSIZ); - snprintf(io_Error, STRSIZ, "Error:: Invalid input string format ... " + snprintf(io_Error, STRSIZ, "Invalid input string format ... " "must be like {text[:[-]#[:#]]}"); io_Errno = EINVAL; return NULL; @@ -401,14 +428,14 @@ ioStrAst(const char *csString) ext = str; out = strdup(ext); - free(str); + xfree(str); return out; } /* - * ioMkDir() Function for racursive directory creation and validation + * ioMkDir() - Function for racursive directory creation and validation * * @csDir = Full directory path * @mode = Mode for directory creation if missing dir @@ -451,12 +478,12 @@ ioMkDir(const char *csDir, int mode) } end: chdir(szOld); - free(str); + xfree(str); return cx; } /* - * ioWatchDirLoop() Function for watching changes in directory and fire callback + * 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 @@ -568,7 +595,7 @@ ioWatchDirLoop(const char *csDir, int (*callback)(cons } /* - * ioCreatePIDFile() Create PID file + * ioCreatePIDFile() - Create PID file * * @csName = PID filename * @ifExists = !=0 if filename exists return error @@ -596,7 +623,7 @@ ioCreatePIDFile(const char *csName, int ifExists) /* - * ioSendFile() AITNET sendfile() userland implementation, not dependant from OS + * ioSendFile() - AITNET sendfile() userland implementation, not dependant from OS * * @s = socket * @csFile = file for send @@ -663,7 +690,7 @@ ioSendFile(int s, const char *csFile, size_t sendLen, } /* - * ioRecvFile() Receive file from socket, fastest (zero-copy) way + * ioRecvFile() - Receive file from socket, fastest (zero-copy) way * * @s = socket * @csFile = file for receive @@ -698,13 +725,7 @@ ioRecvFile(int s, const char *csFile, size_t recvLen, unlink(csFile); return 0; } - if (lseek(fd, recvLen - 1, SEEK_SET) == -1) { - LOGERR; - close(fd); - unlink(csFile); - return 0; - } - if (write(fd, "", 1) == -1) { + if (ftruncate(fd, recvLen) == -1) { LOGERR; close(fd); unlink(csFile);