#ifndef QPID_ACL_ACLDATA_H
#define QPID_ACL_ACLDATA_H


/*
 *
 * Copyright (c) 2006 The Apache Software Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * 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
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#include "qpid/broker/AclModule.h"
#include "qpid/AclHost.h"
#include "AclTopicMatch.h"
#include "qpid/log/Statement.h"
#include "boost/shared_ptr.hpp"
#include <vector>
#include <sstream>

namespace qpid {
namespace acl {

/** A rule for tracking black/white host connection settings.
 * When a connection is attempted, the remote host is verified
 * against lists of these rules. When the remote host is in
 * the range specified by this aclHost then the AclResult is
 * applied as allow/deny.
 */
class AclBWHostRule {
public:
    AclBWHostRule(AclResult r, std::string h) :
        aclResult(r), aclHost(h) {}

    std::string toString () const {
        std::ostringstream ruleStr;
        ruleStr << "[ruleMode = " << AclHelper::getAclResultStr(aclResult)
                << " {" << aclHost.str() << "}";
        return ruleStr.str();
    }
    const AclHost& getAclHost() const { return aclHost; }
    const AclResult& getAclResult() const { return aclResult; }

private:
    AclResult   aclResult;
    AclHost     aclHost;
};


class AclData {


public:

    typedef std::map<qpid::acl::Property, std::string> propertyMap;
    typedef propertyMap::const_iterator                propertyMapItr;

    typedef std::map<qpid::acl::SpecProperty, std::string> specPropertyMap;
    typedef specPropertyMap::const_iterator                specPropertyMapItr;

    //
    // Rule
    //
    // Created by AclReader and stored in a ruleSet vector for subsequent
    //  run-time lookup matching and allow/deny decisions.
    // RuleSet vectors are indexed by Action-Object-actorId so these
    //  attributes are not part of a rule.
    // A single ACL file entry may create many rule entries in
    //  many ruleset vectors.
    //
    struct Rule {
        typedef broker::TopicExchange::TopicExchangeTester topicTester;

        int                   rawRuleNum;   // rule number in ACL file
        qpid::acl::AclResult  ruleMode;     // combined allow/deny log/nolog
        specPropertyMap       props;        // properties to be matched
                                            // pubXxx for publish exchange fastpath
        bool                  pubRoutingKeyInRule;
        std::string           pubRoutingKey;
        boost::shared_ptr<topicTester> pTTest;
        bool                  pubExchNameInRule;
        bool                  pubExchNameMatchesBlank;
        std::string           pubExchName;
        std::vector<bool>     ruleHasUserSub;
        std::string           lookupSource;
        std::string           lookupHelp;

        Rule (int ruleNum, qpid::acl::AclResult res, specPropertyMap& p) :
            rawRuleNum(ruleNum),
            ruleMode(res),
            props(p),
            pubRoutingKeyInRule(false),
            pubRoutingKey(),
            pTTest(boost::shared_ptr<topicTester>(new topicTester())),
            pubExchNameInRule(false),
            pubExchNameMatchesBlank(false),
            pubExchName(),
            ruleHasUserSub(PROPERTYSIZE, false)
            {}

        // Variation of Rule for tracking PropertyDefs
        // for AclValidation.
        Rule (int ruleNum, qpid::acl::AclResult res, specPropertyMap& p,
              const std::string& ls, const std::string& lh
        ) :
            rawRuleNum(ruleNum),
            ruleMode(res),
            props(p),
            pubRoutingKeyInRule(false),
            pubRoutingKey(),
            pubExchNameInRule(false),
            pubExchNameMatchesBlank(false),
            pubExchName(),
            ruleHasUserSub(PROPERTYSIZE, false),
            lookupSource(ls),
            lookupHelp(lh)
            {}


        std::string toString () const {
            std::ostringstream ruleStr;
            ruleStr << "[rule " << rawRuleNum
                    << " ruleMode = " << AclHelper::getAclResultStr(ruleMode)
                    << " props{";
            for (specPropertyMapItr pMItr  = props.begin();
                                    pMItr != props.end();
                                    pMItr++) {
                ruleStr << " "
                        << AclHelper::getPropertyStr((SpecProperty) pMItr-> first)
                        << "=" << pMItr->second;
            }
            ruleStr << " }]";
            return ruleStr.str();
        }

        void addTopicTest(const std::string& pattern) {
            pTTest->addBindingKey(broker::TopicExchange::normalize(pattern));
        }

        // Topic Exchange tester
        // return true if any bindings match 'pattern'
        bool matchRoutingKey(const std::string& pattern) const
        {
            topicTester::BindingVec bv;
            return pTTest->findMatches(pattern, bv);
        }
    };

    typedef  std::vector<Rule>               ruleSet;
    typedef  ruleSet::const_iterator         ruleSetItr;
    typedef  std::map<std::string, ruleSet > actionObject; // user
    typedef  actionObject::iterator          actObjItr;
    typedef  actionObject*                   aclAction;
    typedef  std::map<std::string, uint16_t> quotaRuleSet; // <username, N>
    typedef  quotaRuleSet::const_iterator    quotaRuleSetItr;
    typedef  std::vector<AclBWHostRule>      bwHostRuleSet; // allow/deny hosts-vector
    typedef  bwHostRuleSet::const_iterator   bwHostRuleSetItr;
    typedef  std::map<std::string, bwHostRuleSet> bwHostUserRuleMap; //<username, hosts-vector>
    typedef  bwHostUserRuleMap::const_iterator    bwHostUserRuleMapItr;

    // Action*[] -> Object*[] -> map<user, set<Rule> >
    aclAction*           actionList[qpid::acl::ACTIONSIZE];
    qpid::acl::AclResult decisionMode;  // allow/deny[-log] if no matching rule found
    bool                 transferAcl;
    std::string          aclSource;
    qpid::acl::AclResult connectionDecisionMode;

    AclResult lookup(
        const std::string&               id,        // actor id
        const Action&                    action,
        const ObjectType&                objType,
        const std::string&               name,      // object name
        std::map<Property, std::string>* params=0);

    AclResult lookup(
        const std::string&               id,        // actor id
        const Action&                    action,
        const ObjectType&                objType,
        const std::string&               ExchangeName,
        const std::string&               RoutingKey);

    boost::shared_ptr<const bwHostRuleSet> getGlobalConnectionRules() {
        return connBWHostsGlobalRules;
    }

    boost::shared_ptr<const bwHostUserRuleMap> getUserConnectionRules() {
        return connBWHostsUserRules;
    }

    bool matchProp(const std::string & src, const std::string& src1);
    void clear ();
    void printDecisionRules(int userFieldWidth);

    static const std::string ACL_KEYWORD_USER_SUBST;
    static const std::string ACL_KEYWORD_DOMAIN_SUBST;
    static const std::string ACL_KEYWORD_USERDOMAIN_SUBST;
    static const std::string ACL_KEYWORD_ALL;
    static const std::string ACL_KEYWORD_ACL;
    static const std::string ACL_KEYWORD_GROUP;
    static const std::string ACL_KEYWORD_QUOTA;
    static const std::string ACL_KEYWORD_QUOTA_CONNECTIONS;
    static const std::string ACL_KEYWORD_QUOTA_QUEUES;
    static const char        ACL_SYMBOL_WILDCARD;
    static const std::string ACL_KEYWORD_WILDCARD;
    static const char        ACL_SYMBOL_LINE_CONTINUATION;
    static const std::string ACL_KEYWORD_DEFAULT_EXCHANGE;

    void substituteString(std::string& targetString,
                          const std::string& placeholder,
                          const std::string& replacement);
    std::string normalizeUserId(const std::string& userId);
    void substituteUserId(std::string& ruleString,
                          const std::string& userId);
    void substituteKeywords(std::string& ruleString,
                            const std::string& userId);

    // Per user connection quotas extracted from acl rule file
    //   Set by reader
    void setConnQuotaRuleSettings (boost::shared_ptr<quotaRuleSet>);
    //   Get by connection approvers
    bool enforcingConnectionQuotas() const { return connQuotaRuleSettings->size() > 0; }
    bool getConnQuotaForUser(const std::string&, uint16_t*) const;

    // Per user queue quotas extracted from acl rule file
    //   Set by reader
    void setQueueQuotaRuleSettings (boost::shared_ptr<quotaRuleSet>);
    //   Get by queue approvers
    bool enforcingQueueQuotas() const { return queueQuotaRuleSettings->size() > 0; }
    bool getQueueQuotaForUser(const std::string&, uint16_t*) const;

    // Global connection Black/White list rules
    void setConnGlobalRules (boost::shared_ptr<bwHostRuleSet>);

    // Per-user connection Black/White list rules map
    void setConnUserRules (boost::shared_ptr<bwHostUserRuleMap>);

    /** getConnectMaxSpec
     * Connection quotas are held in uint16_t variables.
     * This function specifies the largest value that a user is allowed
     * to declare for a connection quota. The upper limit serves two
     * purposes: 1. It leaves room for magic numbers that may be declared
     * by keyword names in Acl files and not have those numbers conflict
     * with innocent user declared values, and 2. It makes the unsigned
     * math very close to _MAX work reliably with no risk of accidental
     * wrapping back to zero.
     */
    static uint16_t getConnectMaxSpec() {
        return 65530;
    }
    static std::string getMaxConnectSpecStr() {
        return "65530";
    }

    static uint16_t getQueueMaxSpec() {
        return 65530;
    }
    static std::string getMaxQueueSpecStr() {
        return "65530";
    }

    /**
     * isAllowedConnection
     * Return true if this user is allowed to connect to this host.
     * Return log text describing both success and failure.
     */
    AclResult isAllowedConnection(const std::string& userName,
                                  const std::string& hostName,
                                  std::string& logText);

    AclResult connectionMode() const {
        return connectionDecisionMode;
    }

    AclData();
    virtual ~AclData();

private:

    inline bool lookupMatchRule(
        const ruleSetItr&                rsItr,
        const std::string&               id,
        const std::string&               name,
        const std::map<Property, std::string>* params,
        AclResult&                       aclresult);

    inline bool lookupMatchPublishExchangeRule(
        const ruleSetItr&                      rsItr,
        const std::string&               id,
        const std::string&               name,
        const std::string&               routingKey,
        AclResult&                       aclresult);

    bool compareInt(const qpid::acl::SpecProperty theProperty,
                    const std::string             theAclValue,
                    const std::string             theLookupValue,
                    bool                          theMaxFlag);

    // Per-user connection quota
    boost::shared_ptr<quotaRuleSet> connQuotaRuleSettings;

    // Per-user queue quota
    boost::shared_ptr<quotaRuleSet> queueQuotaRuleSettings;

    // Global host connection black/white rule set
    boost::shared_ptr<bwHostRuleSet> connBWHostsGlobalRules;

    // Per-user host connection black/white rule set map
    boost::shared_ptr<bwHostUserRuleMap> connBWHostsUserRules;
};

}} // namespace qpid::acl

#endif // QPID_ACL_ACLDATA_H