Diff for /libaitio/src/aitio.c between versions 1.10.6.4 and 1.17.8.3

version 1.10.6.4, 2012/05/23 13:53:00 version 1.17.8.3, 2016/08/10 14:19:50
Line 12  terms: Line 12  terms:
 All of the documentation and software included in the ELWIX and AITNET  All of the documentation and software included in the ELWIX and AITNET
 Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>  Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   
Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012Copyright 2004 - 2016
         by Michael Pounov <misho@elwix.org>.  All rights reserved.          by Michael Pounov <misho@elwix.org>.  All rights reserved.
   
 Redistribution and use in source and binary forms, with or without  Redistribution and use in source and binary forms, with or without
Line 46  SUCH DAMAGE. Line 46  SUCH DAMAGE.
 #include "global.h"  #include "global.h"
   
   
 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 70  char io_Error[STRSIZ]; Line 55  char io_Error[STRSIZ];
   
   
 // io_GetErrno() Get error code of last operation  // io_GetErrno() Get error code of last operation
inline intint
 io_GetErrno()  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 *const char *
 io_GetError()  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 voidvoid
 io_SetErr(int eno, char *estr, ...)  io_SetErr(int eno, char *estr, ...)
 {  {
         va_list lst;          va_list lst;
   
         io_Errno = eno;          io_Errno = eno;
        memset(io_Error, 0, STRSIZ);        memset(io_Error, 0, sizeof io_Error);
         va_start(lst, estr);          va_start(lst, estr);
        vsnprintf(io_Error, STRSIZ, estr, lst);        vsnprintf(io_Error, sizeof io_Error, estr, lst);
         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();  
 }  
   
 /*  /*
  * 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]   * 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
Line 224  ioPromptRead(int *h, const char *csPrompt, char * __re Line 121  ioPromptRead(int *h, const char *csPrompt, char * __re
                 memset(szLine, 0, BUFSIZ);                  memset(szLine, 0, BUFSIZ);
                 if (!fgets(szLine, BUFSIZ, inp)) {                  if (!fgets(szLine, BUFSIZ, inp)) {
                         clearerr(inp);                          clearerr(inp);
   #ifdef HAVE_FPURGE
                         fpurge(out);                          fpurge(out);
   #else
                           __fpurge(out);
   #endif
                         fflush(out);                          fflush(out);
                         return 0;                          return 0;
                 }                  }
Line 255  ioPromptPassword(int *h, const char *csPrompt, char *  Line 156  ioPromptPassword(int *h, const char *csPrompt, char * 
         int ret, ok = 0;          int ret, ok = 0;
         FILE *inp, *out;          FILE *inp, *out;
         char szLine[2][STRSIZ];          char szLine[2][STRSIZ];
   #ifndef __linux__
         struct sgttyb tty_state;          struct sgttyb tty_state;
   #else
           struct termios o;
   #endif
   
         if (!psPass || !passLen)          if (!psPass || !passLen)
                 return -1;                  return -1;
Line 271  ioPromptPassword(int *h, const char *csPrompt, char *  Line 176  ioPromptPassword(int *h, const char *csPrompt, char * 
                 return -1;                  return -1;
         }          }
   
   #ifndef __linux__
         if (ioctl(fileno(inp), TIOCGETP, &tty_state) == -1) {          if (ioctl(fileno(inp), TIOCGETP, &tty_state) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
Line 281  ioPromptPassword(int *h, const char *csPrompt, char *  Line 187  ioPromptPassword(int *h, const char *csPrompt, char * 
                         return -1;                          return -1;
                 }                  }
         }          }
   #else
           if (tcgetattr(fileno(inp), &o) == -1) {
                   LOGERR;
                   return -1;
           } else {
                   o.c_lflag &= ~ECHO;
                   if (tcsetattr(fileno(inp), TCSANOW, &o) == -1) {
                           LOGERR;
                           return -1;
                   }
           }
   #endif
   
         while (!ok) {          while (!ok) {
                 switch ((ret = ioPromptRead(h, (!csPrompt || !*csPrompt) ? "Password:" : csPrompt,                   switch ((ret = ioPromptRead(h, (!csPrompt || !*csPrompt) ? "Password:" : csPrompt, 
Line 317  ioPromptPassword(int *h, const char *csPrompt, char *  Line 235  ioPromptPassword(int *h, const char *csPrompt, char * 
         }          }
   
 next:  next:
   #ifndef __linux__
         tty_state.sg_flags |= ECHO;          tty_state.sg_flags |= ECHO;
         if (ioctl(fileno(inp), TIOCSETP, &tty_state) == -1) {          if (ioctl(fileno(inp), TIOCSETP, &tty_state) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
         }          }
#else
        return ok;        o.c_lflag |= ECHO;
}        if (tcsetattr(fileno(inp), TCSANOW, &o) == -1) {
                LOGERR;
/* 
 * ioRegexVerify() - Function for verify data match in regex expression 
 * 
 * @csRegex = Regulare expression pattern 
 * @csData = Data for check and verify 
 * @startPos = Return start positions 
 * @endPos = Return end positions 
 * return: NULL not match or error; !=NULL begin of matched data 
*/ 
const char * 
ioRegexVerify(const char *csRegex, const char *csData, int *startPos, int *endPos) 
{ 
        regex_t re; 
        regmatch_t match; 
        char szErr[STRSIZ]; 
        int ret, flg; 
        const char *pos; 
 
        if (!csRegex || !csData) 
                return NULL; 
 
        if ((ret = regcomp(&re, csRegex, REG_EXTENDED))) { 
                regerror(ret, &re, szErr, STRSIZ); 
                io_SetErr(ret, "%s", szErr); 
                regfree(&re); 
                return NULL; 
        } 
 
        for (ret = flg = 0, pos = csData; !(ret = regexec(&re, pos, 1, &match, flg));  
                        pos += match.rm_eo, flg = REG_NOTBOL) { 
                if (startPos) 
                        *startPos = match.rm_so; 
                if (endPos) 
                        *endPos = match.rm_eo; 
 
                pos += match.rm_so; 
                break; 
        } 
 
        if (ret) { 
                regerror(ret, &re, szErr, STRSIZ); 
                io_SetErr(ret, "%s", szErr); 
                pos = NULL; 
        } 
 
        regfree(&re); 
        return pos; 
} 
 
/* 
 * ioRegexGet() - Function for get data match in regex expression 
 * 
 * @csRegex = Regulare expression pattern 
 * @csData = Data from get 
 * @psString = Returned string if match 
 * @strLen = Length of string 
 * return: 0 not match; >0 count of returned chars 
*/ 
int 
ioRegexGet(const char *csRegex, const char *csData, char * __restrict psString, int strLen) 
{ 
        int sp, ep, len; 
        const char *str; 
 
        if (!csRegex || !csData) 
                 return -1;                  return -1;
   
         str = ioRegexVerify(csRegex, csData, &sp, &ep);  
         if (!str)  
                 return 0;  
   
         len = ep - sp;  
         if (psString && strLen) {  
                 memset(psString, 0, strLen);  
                 strncpy(psString, str, strLen <= len ? strLen - 1 : len);  
         }          }
   #endif
   
        return len;        return ok;
 }  }
   
 /*  /*
  * 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 xfree after use!  
 */  
 char *  
 ioRegexReplace(const char *csRegex, const char *csData, const char *csNew)  
 {  
         int sp, ep, len;  
         char *str = NULL;  
   
         if (!csRegex || !csData)  
                 return NULL;  
   
         if (!ioRegexVerify(csRegex, csData, &sp, &ep))  
                 return NULL;  
   
         // ___ before match  
         len = sp + 1;  
         str = xmalloc(len);  
         if (!str) {  
                 LOGERR;  
                 return NULL;  
         } else  
                 strlcpy(str, csData, len);  
         // * replace match *  
         if (csNew) {  
                 len += strlen(csNew);  
                 str = xrealloc(str, len);  
                 if (!str) {  
                         LOGERR;  
                         return NULL;  
                 } else  
                         strlcat(str, csNew, len);  
         }  
         // after match ___  
         len += strlen(csData) - ep;  
         str = xrealloc(str, len);  
         if (!str) {  
                 LOGERR;  
                 return NULL;  
         } else  
                 strlcat(str, csData + ep, len);  
   
         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
Line 523  ioMkDir(const char *csDir, int mode) Line 268  ioMkDir(const char *csDir, int mode)
         if (!csDir)          if (!csDir)
                 return cx;                  return cx;
   
        str = strdup(csDir);        str = e_strdup(csDir);
         if (!str) {          if (!str) {
                 LOGERR;                  LOGERR;
                 return cx;                  return cx;
Line 551  ioMkDir(const char *csDir, int mode) Line 296  ioMkDir(const char *csDir, int mode)
         }          }
 end:  end:
         chdir(szOld);          chdir(szOld);
        xfree(str);        e_free(str);
         return cx;          return cx;
 }  }
   
Line 566  int Line 311  int
 ioWatchDirLoop(const char *csDir, int (*callback)(const char *csName, int nOp))  ioWatchDirLoop(const char *csDir, int (*callback)(const char *csName, int nOp))
 {  {
         glob_t g[2] = {{ 0 }, { 0 }};          glob_t g[2] = {{ 0 }, { 0 }};
        int d, kq, n = 0;        int d, n = 0;
         register int j, i;          register int j, i;
   #ifndef __linux__
           int kq;
         struct kevent req, chg;          struct kevent req, chg;
   #endif
         char wrk[MAXPATHLEN * 2], str[MAXPATHLEN] = { 0 };          char wrk[MAXPATHLEN * 2], str[MAXPATHLEN] = { 0 };
   
         if (!csDir || !callback)          if (!csDir || !callback)
Line 577  ioWatchDirLoop(const char *csDir, int (*callback)(cons Line 325  ioWatchDirLoop(const char *csDir, int (*callback)(cons
         strlcpy(str, csDir, MAXPATHLEN);          strlcpy(str, csDir, MAXPATHLEN);
         strlcat(str, "/*", MAXPATHLEN);          strlcat(str, "/*", MAXPATHLEN);
   
        kq = kqueue();        d = open(csDir, O_RDONLY);
        if (kq == -1) {        if (d == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
         }          }
        d = open(csDir, O_RDONLY);
        if (d == -1) {#ifndef __linux__
         kq = kqueue();
         if (kq == -1) {
                 LOGERR;                  LOGERR;
                close(kq);                close(d);
                 return -1;                  return -1;
         }          }
   
         EV_SET(&req, d, EVFILT_VNODE, EV_ADD | EV_CLEAR, NOTE_WRITE, 0, 0);          EV_SET(&req, d, EVFILT_VNODE, EV_ADD | EV_CLEAR, NOTE_WRITE, 0, 0);
   #endif
   
         if ((n = glob(str, GLOB_NOCHECK, NULL, &g[0]))) {          if ((n = glob(str, GLOB_NOCHECK, NULL, &g[0]))) {
                 LOGERR;                  LOGERR;
                 close(d);                  close(d);
   #ifndef __linux__
                 close(kq);                  close(kq);
   #endif
                 return -1;                  return -1;
         } /*else          } /*else
                 ioDEBUG(3, "Start files %d in %s\n", g[0].gl_matchc, str);*/                  ioDEBUG(3, "Start files %d in %s\n", g[0].gl_matchc, str);*/
   
   #ifndef __linux__
         while (kevent(kq, &req, 1, &chg, 1, NULL) > 0) {          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);*/                  /*ioDEBUG(1, "Event:: req=0x%x -> chg=0x%x data=%x\n", req.fflags, chg.fflags, chg.data);*/
   
Line 660  ioWatchDirLoop(const char *csDir, int (*callback)(cons Line 414  ioWatchDirLoop(const char *csDir, int (*callback)(cons
                         g[0] = g[1];                          g[0] = g[1];
                 }                  }
         }          }
   #endif
   
         globfree(&g[0]);          globfree(&g[0]);
        close(d);#ifndef __linux__
         close(kq);          close(kq);
   #endif
           close(d);
         return n;          return n;
 }  }
   
Line 674  ioWatchDirLoop(const char *csDir, int (*callback)(cons Line 431  ioWatchDirLoop(const char *csDir, int (*callback)(cons
  * @ifExists = !=0 if filename exists return error   * @ifExists = !=0 if filename exists return error
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
  */   */
inline intint
 ioCreatePIDFile(const char *csName, int ifExists)  ioCreatePIDFile(const char *csName, int ifExists)
 {  {
         int fd;          int fd;
Line 830  ioRecvFile(int s, const char *csFile, size_t recvLen,  Line 587  ioRecvFile(int s, const char *csFile, size_t recvLen, 
   
         munmap(addr, recvLen);          munmap(addr, recvLen);
         return len;          return len;
   }
   
   /*
    * ioRealFileName() - Get real file name
    *
    * @fname = filename
    * return: =NULL error or !=NULL real filename, should be free with e_free()
    */
   char *
   ioRealFileName(const char *fname)
   {
           char *str = NULL;
           struct stat sb;
   
           if (!fname)
                   return NULL;
   
           str = e_malloc(MAXPATHLEN);
           if (!str) {
                   io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                   return NULL;
           } else
                   memset(str, 0, MAXPATHLEN);
           if (readlink(fname, str, MAXPATHLEN) == -1) {
                   if (stat(fname, &sb) == -1) {
                           LOGERR;
                           e_free(str);
                           return NULL;
                   } else
                           strlcpy(str, fname, MAXPATHLEN);
           }
   
           return str;
 }  }

Removed from v.1.10.6.4  
changed lines
  Added in v.1.17.8.3


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>