/* OpenIDA - Outil d'analyse de fichiers binaires
 * extstr.c - extension des fonctions relatives aux chaînes
 *
 * Copyright (C) 2008 Cyrille Bagard
 *
 *  This file is part of OpenIDA.
 *
 *  OpenIDA is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  OpenIDA is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with Foobar.  If not, see <http://www.gnu.org/licenses/>.
 */


#include "extstr.h"


#include <malloc.h>
#include <regex.h>
#include <string.h>



/******************************************************************************
*                                                                             *
*  Paramètres  : str1 = chaîne de caractères à compléter.                     *
*                str2 = chaîne de caractères à ajouter.                       *
*                                                                             *
*  Description : Complète une chaîne de caractères avec une autre.            *
*                                                                             *
*  Retour      : Chaîne de caractères complétée, à libérer de la mémoire.     *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

char *stradd(char *str1, const char *str2)
{
    char *result;                           /* Chaîne à renvoyer           */

    result = (char *)realloc(str1, (strlen(str1) + strlen(str2) + 1) * sizeof(char));

    strcat(result, str2);

    return result;

}


/******************************************************************************
*                                                                             *
*  Paramètres  : str1 = chaîne de caractères à compléter.                     *
*                str2 = chaîne de caractères à ajouter.                       *
*                                                                             *
*  Description : Fait précéder une chaîne de caractères par une autre.        *
*                                                                             *
*  Retour      : Chaîne de caractères complétée, à libérer de la mémoire.     *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

char *strprep(char *str1, const char *str2)
{
    char *result;                           /* Chaîne à renvoyer           */
    size_t len2;                            /* Taille de la seconde chaîne */

    result = (char *)realloc(str1, (strlen(str1) + strlen(str2) + 1) * sizeof(char));

    len2 = strlen(str2);

    memmove(&result[len2], result, strlen(result) + 1);
    memcpy(result, str2, len2);

    return result;

}


/******************************************************************************
*                                                                             *
*  Paramètres  : input = chaîne de caractères à traiter.                      *
*                                                                             *
*  Description : S'assure qu'une chaîne de caractère tient sur une ligne.     *
*                                                                             *
*  Retour      : Adresse de la chaîne de caractères modifiée.                 *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

char *escape_crlf(char *input)
{
    size_t inlen;
    regex_t preg;
    int ret;
    size_t curpos;
    regmatch_t pmatch[2];

    inlen = strlen(input);

    ret = regcomp(&preg, "(\t|\n|\r)", REG_EXTENDED | REG_ICASE);
    /* TODO: ret */

    for (curpos = 0; regexec(&preg, &input[curpos], 2, pmatch, 0) != REG_NOMATCH; )
    {
        inlen += 1 + 1;
        input = (char *)realloc(input, inlen * sizeof(char *));

        memmove(&input[curpos + pmatch[1].rm_eo + 1], &input[curpos + pmatch[1].rm_eo], inlen - 1 - curpos - pmatch[1].rm_eo);

        switch (input[curpos + pmatch[1].rm_so])
        {
            case '\t':
                memcpy(&input[curpos + pmatch[1].rm_so], "\\t", 2);
                break;
            case '\n':
                memcpy(&input[curpos + pmatch[1].rm_so], "\\n", 2);
                break;
            case '\r':
                memcpy(&input[curpos + pmatch[1].rm_so], "\\r", 2);
                break;
        }

        curpos += pmatch[1].rm_eo + 1;

    }

    return input;

}