/*
 *  Copyright (C) 2021-2022 Savoir-faire Linux Inc.
 *
 *  Author: Olivier Dion <olivier.dion>@savoirfairelinux.com>
 *
 *  This program 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.
 *
 *  This program 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 this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.
 */

#include <cstdlib>

#include "lib/gnutls.h"
#include "lib/sip-fmt.h"

/*
 * Example that will try to overflow the SDP parser.
 */
bool
mutate_gnutls_record_send(ChanneledMessage& msg)
{
    static size_t body_repeat = 1;

    SIPFmt sip(msg.data);

    /* Only SIP request */
    if (not sip.isRequest()) {
        return false;
    }

    /* Skip none SDP Content-Type */
    if (std::string::npos == sip.getField("content-type").find("text/plain")) {
         return false;
    }

    auto body = sip.getBody();
    std::vector<uint8_t> newBody;

    newBody.reserve(body_repeat * body.size());

    for (size_t i=0; i<body_repeat; ++i) {
            for (auto it=body.cbegin(); it!=body.cend(); ++it) {
                    newBody.emplace_back(*it);
            }
    }

    sip.swapBody(newBody);

    body_repeat *= 2;

    /* Commit changes! */
    sip.swap(msg.data);

    return true;
}
