lib-lib/rx.c

   1 /*
   2  *  This program is free software; you can redistribute it and/or modify
   3  *  it under the terms of the GNU General Public License as published by
   4  *  the Free Software Foundation; either version 2 of the License, or (at
   5  *  your option) any later version.
   6  *
   7  *  This program is distributed in the hope that it will be useful, but
   8  *  WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10  *  General Public License for more details.
  11  *
  12  *  You should have received a copy of the GNU General Public License
  13  *  along with this program; if not, write to the Free Software
  14  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  15  *  MA 02110-1301, USA.
  16  *
  17  *  Copyright © 2006 Pierre Habouzit
  18  */
  19 /*
  20  * This file is part of mutt-ng, see http://www.muttng.org/.
  21  * It's licensed under the GNU General Public License,
  22  * please see the file GPL in the top level source directory.
  23  */
  24
  25 #include "lib-lib.h"
  26
  27 rx_t *rx_compile(const char *s, int flags)
  28 {
  29     rx_t *pp = p_new(rx_t, 1);
  30
  31     pp->pattern = m_strdup(s);
  32     pp->rx = p_new(regex_t, 1);
  33
  34     if (REGCOMP(pp->rx, NONULL(s), flags) != 0) {
  35         rx_delete(&pp);
  36     }
  37
  38     return pp;
  39 }
  40
  41 void rx_set_template(rx_t *rx, const char *tpl)
  42 {
  43     const char *p = tpl;
  44
  45     m_strreplace(&rx->template, tpl);
  46     rx->nmatch = 0;
  47
  48     while ((p = strchr(p, '%'))) {
  49         if (isdigit(*++p)) {
  50             int n = strtol(p, (char **)&p, 10);
  51             rx->nmatch = MAX(n, rx->nmatch);
  52         } else {
  53             if (*p == '%')
  54                 p++;
  55         }
  56     }
  57
  58     rx->nmatch++;     /* match 0 is always the whole expr */
  59 }
  60
  61 void rx_delete(rx_t **p)
  62 {
  63     p_delete(&(*p)->pattern);
  64     regfree((*p)->rx);
  65     p_delete(&(*p)->rx);
  66     p_delete(&(*p)->template);
  67     p_delete(p);
  68 }
  69
  70 int rx_list_match(rx_t *l, const char *pat)
  71 {
  72     if (!pat || !*pat)
  73         return 0;
  74
  75     while (l) {
  76         if (!REGEXEC(l->rx, pat))
  77             return 1;
  78         l = l->next;
  79     }
  80
  81     return 0;
  82 }
  83
  84 rx_t **rx_lookup(rx_t **l, const char *pat)
  85 {
  86     if (!pat || !*pat)
  87         return NULL;
  88
  89     while (*l) {
  90         if (!strcmp((*l)->pattern, pat))
  91             return l;
  92         l = &(*l)->next;
  93     }
  94
  95     return NULL;
  96 }
  97
  98 int rx_sanitize_string(char *dst, ssize_t n, const char *src)
  99 {
 100     while (*src) {
 101         if (n <= 1)
 102             break;
 103
 104         /* these characters must be escaped in regular expressions */
 105         if (strchr("^.[$()|*+?{\\", *src)) {
 106             if (n <= 2)
 107                 break;
 108
 109             *dst++ = '\\';
 110             n--;
 111         }
 112
 113         *dst++ = *src++;
 114         n--;
 115     }
 116
 117     *dst = '\0';
 118
 119     return *src ? -1 : 0;
 120 }