src/pstr.c

   1 /*
   2  Copyright (C) 2011-2014 jeanfi@gmail.com
   3
   4  This program is free software; you can redistribute it and/or
   5  modify it under the terms of the GNU General Public License as
   6  published by the Free Software Foundation; either version 2 of the
   7  License, or (at your option) any later version.
   8
   9  This program is distributed in the hope that it will be useful, but
  10  WITHOUT ANY WARRANTY; without even the implied warranty of
  11  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  General Public License for more details.
  13
  14  You should have received a copy of the GNU General Public License
  15  along with this program; if not, write to the Free Software
  16  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  17  02110-1301 USA
  18  */
  19
  20 #include <string.h>
  21
  22 #include <pstr.h>
  23
  24 char *strrep(char *str, const char *old, const char *new)
  25 {
  26         char *p, *res;
  27         int pos;
  28
  29         if (!str)
  30                 return NULL;
  31
  32         if (!*str || !old || !*old || !new || !strcmp(old, new))
  33                 return str;
  34
  35         p = strstr(str, old);
  36
  37         if (!p)
  38                 return str;
  39
  40         res = malloc(strlen(str) + (new ? strlen(new) : 0) - strlen(old) + 1);
  41
  42         pos = p - str;
  43
  44         strncpy(res, str, pos);
  45         res[pos] = '\0';
  46
  47         if (new)
  48                 strcat(res + pos, new);
  49
  50         strcat(res, str + pos + strlen(old));
  51
  52         return res;
  53 }
  54
  55 /*
  56   Derivated from http://stackoverflow.com/questions/779875/what-is-the-function-to-replace-string-in-c
  57 */
  58 char *strrepg(char *orig, const char *rep, const char *with) {
  59     char *result; // the return string
  60     char *ins;    // the next insert point
  61     char *tmp;    // varies
  62     int len_rep;  // length of rep
  63     int len_with; // length of with
  64     int len_front; // distance between rep and end of last rep
  65     int count;    // number of replacements
  66
  67     if (!orig)
  68         return NULL;
  69     if (!rep)
  70         rep = "";
  71     len_rep = strlen(rep);
  72     if (!with)
  73         with = "";
  74     len_with = strlen(with);
  75
  76     ins = orig;
  77     for(count = 0; (tmp = strstr(ins, rep)); count++) {
  78         ins = tmp + len_rep;
  79     }
  80
  81     // first time through the loop, all the variable are set correctly
  82     // from here on,
  83     //    tmp points to the end of the result string
  84     //    ins points to the next occurrence of rep in orig
  85     //    orig points to the remainder of orig after "end of rep"
  86     tmp = result = malloc(strlen(orig) + (len_with - len_rep) * count + 1);
  87
  88     if (!result)
  89         return NULL;
  90
  91     while (count--) {
  92         ins = strstr(orig, rep);
  93         len_front = ins - orig;
  94         tmp = strncpy(tmp, orig, len_front) + len_front;
  95         tmp = strcpy(tmp, with) + len_with;
  96         orig += len_front + len_rep; // move to next "end of rep"
  97     }
  98     strcpy(tmp, orig);
  99     return result;
 100 }