niceSOFT/libmodsecurity: ModSecurity is an open source, cross platform web application firewall (WAF) engine for Apache, IIS and Nginx. It has a robust event-based programming language which provides protection from a range of attacks against web applications and allows for HTTP traffic monitoring, logging and real-time analysis.

src/variables/remote_user.cc

80 строк2 KB

gberkes

Refactor: used the init-statement to declare "pos" inside the if statement.

07 авг 2024, 14:05

07 авг 2024, 14:05ab78d4a

/*
 * ModSecurity, http://www.modsecurity.org/
 * Copyright (c) 2015 - 2023 Trustwave Holdings, Inc. (http://www.trustwave.com/)
 *
 * You may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * If any of the files related to licensing are missing or if you have any
 * other questions related to licensing please contact Trustwave Holdings, Inc.
 * directly using the email address security@modsecurity.org.
 *
 */

#include "src/variables/remote_user.h"

#include <time.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <iostream>
#include <string>
#include <vector>
#include <list>
#include <utility>
#include <memory>

#include "modsecurity/transaction.h"
#include "src/utils/base64.h"

namespace modsecurity {
namespace variables {


void RemoteUser::evaluate(Transaction *transaction,
    RuleWithActions *rule,
    std::vector<const VariableValue *> *l) {
    std::vector<const VariableValue *> l2;

    transaction->m_variableRequestHeaders.resolve("authorization", &l2);

    if (!l2.empty()) {
        const auto *v = l2[0];

        const auto &header = v->getValue();

        std::string base64;

        if (header.compare(0, 6, "Basic ") == 0) {
            base64 = std::string(header, 6, header.length());
        }

        base64 = Utils::Base64::decode(base64);

        if (const auto pos{base64.find(":")}; pos != std::string::npos) {
            transaction->m_variableRemoteUser.assign(std::string(base64, 0, pos));

            auto var = std::make_unique<VariableValue>(&v->getKeyWithCollection(),
                &transaction->m_variableRemoteUser);

            var->reserveOrigin(v->getOrigin().size());
            for (const auto &i : v->getOrigin()) {
                var->addOrigin(i);
            }
            l->push_back(var.release());
        }

        for (auto &a : l2) {
            delete a;
        }
    }
}


}  // namespace variables
}  // namespace modsecurity