ServiceRep.h

Go to the documentation of this file.

/*
 * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */
 
#ifndef SQUID_SRC_ADAPTATION_ICAP_SERVICEREP_H
#define SQUID_SRC_ADAPTATION_ICAP_SERVICEREP_H
 
#include "adaptation/forward.h"
#include "adaptation/icap/Elements.h"
#include "adaptation/Initiator.h"
#include "adaptation/Service.h"
#include "base/AsyncJobCalls.h"
#include "cbdata.h"
#include "comm.h"
#include "FadingCounter.h"
#include "pconn.h"
#include <deque>
 
namespace Adaptation
{
namespace Icap
{
 
class Options;
class OptXact;
 
/* The ICAP service representative maintains information about a single ICAP
   service that Squid communicates with. The representative initiates OPTIONS
   requests to the service to keep cached options fresh. One ICAP server may
   host many ICAP services. */
 
/*
 * A service with a fresh cached OPTIONS response and without many failures
 * is an "up" service. All other services are "down". A service is "probed"
 * if we tried to get an OPTIONS response from it and succeeded or failed.
 * A probed down service is called "broken".
 *
 * The number of failures required to bring an up service down is determined
 * by icap_service_failure_limit in squid.conf.
 *
 * As a bootstrapping mechanism, ICAP transactions wait for an unprobed
 * service to get a fresh OPTIONS response (see the callWhenReady method).
 * The waiting callback is called when the OPTIONS transaction completes,
 * even if the service is now broken.
 *
 * We do not initiate ICAP transactions with a broken service, but will
 * eventually retry to fetch its options in hope to bring the service up.
 *
 * A service that should no longer be used after Squid reconfiguration is
 * treated as if it does not have a fresh cached OPTIONS response. We do
 * not try to fetch fresh options for such a service. It should be
 * auto-destroyed by refcounting when no longer used.
 */
 
class ServiceRep : public RefCountable, public Adaptation::Service,
    public Adaptation::Initiator
{
    CBDATA_CHILD(ServiceRep);
 
public:
    typedef RefCount<ServiceRep> Pointer;
 
public:
    explicit ServiceRep(const ServiceConfigPointer &aConfig);
    ~ServiceRep() override;
 
    void finalize() override;
 
    bool probed() const override; // see comments above
    bool up() const override; // see comments above
    bool availableForNew() const; 
    bool availableForOld() const; 
 
    Initiate *makeXactLauncher(Http::Message *virginHeader, HttpRequest *virginCause, AccessLogEntry::Pointer &alp) override;
 
    void callWhenAvailable(AsyncCall::Pointer &cb, bool priority = false);
    void callWhenReady(AsyncCall::Pointer &cb);
 
    // the methods below can only be called on an up() service
    bool wantsUrl(const SBuf &urlPath) const override;
    bool wantsPreview(const SBuf &urlPath, size_t &wantedSize) const;
    bool allows204() const;
    bool allows206() const;
    Comm::ConnectionPointer getIdleConnection(bool isRetriable);
    void putConnection(const Comm::ConnectionPointer &conn, bool isReusable, bool sendReset, const char *comment);
    void noteConnectionUse(const Comm::ConnectionPointer &conn);
    void noteConnectionFailed(const char *comment);
 
    void noteFailure() override; // called by transactions to report service failure
 
    void noteNewWaiter() {theAllWaiters++;} 
    void noteGoneWaiter(); 
    bool existWaiters() const {return (theAllWaiters > 0);} 
 
    //AsyncJob virtual methods
    bool doneAll() const override { return Adaptation::Initiator::doneAll() && false;}
    void callException(const std::exception &e) override;
 
    void detach() override;
    bool detached() const override;
 
public: // treat these as private, they are for callbacks only
    void noteTimeToUpdate();
    void noteTimeToNotify();
 
    // receive either an ICAP OPTIONS response header or an abort message
    void noteAdaptationAnswer(const Answer &answer) override;
 
    Security::ContextPointer sslContext;
    // TODO: Remove sslContext above when FuturePeerContext below becomes PeerContext
    Security::FuturePeerContext tlsContext;
    Security::SessionStatePointer sslSession;
 
private:
    // stores Prepare() callback info
 
    struct Client {
        Pointer service; // one for each client to preserve service
        AsyncCall::Pointer callback;
    };
 
    typedef std::vector<Client> Clients;
    // TODO: rename to theUpWaiters
    Clients theClients; // all clients waiting for a call back
 
    Options *theOptions;
    CbcPointer<Adaptation::Initiate> theOptionsFetcher; // pending ICAP OPTIONS transaction
    time_t theLastUpdate; // time the options were last updated
 
    std::deque<Client> theNotificationWaiters;
    int theBusyConns;   
    int theAllWaiters;
    int theMaxConnections; 
    // TODO: use a better type like the FadingCounter for connOverloadReported
    mutable bool connOverloadReported; 
    IdleConnList *theIdleConns; 
 
    FadingCounter theSessionFailures;
    const char *isSuspended; // also stores suspension reason for debugging
 
    bool notifying; // may be true in any state except for the initial
    bool updateScheduled; // time-based options update has been scheduled
 
private:
    ICAP::Method parseMethod(const char *) const;
    ICAP::VectPoint parseVectPoint(const char *) const;
 
    void suspend(const char *reason);
 
    bool hasOptions() const;
    bool needNewOptions() const;
    time_t optionsFetchTime() const;
 
    void scheduleUpdate(time_t when);
    void scheduleNotification();
 
    void startGettingOptions();
    void handleNewOptions(Options *newOptions);
    void changeOptions(Options *newOptions);
    void checkOptions();
 
    void announceStatusChange(const char *downPhrase, bool important) const;
 
    void setMaxConnections();
    int excessConnections() const;
    int availableConnections() const;
    void busyCheckpoint();
 
    const char *status() const override;
 
    mutable bool wasAnnouncedUp; // prevent sequential same-state announcements
    bool isDetached;
};
 
class ModXact;
class ConnWaiterDialer: public NullaryMemFunT<ModXact>
{
public:
    typedef NullaryMemFunT<ModXact> Parent;
    ServiceRep::Pointer theService;
    ConnWaiterDialer(const CbcPointer<Adaptation::Icap::ModXact> &xact, Adaptation::Icap::ConnWaiterDialer::Parent::Method aHandler);
    ConnWaiterDialer(const Adaptation::Icap::ConnWaiterDialer &aConnWaiter);
    ~ConnWaiterDialer() override;
};
 
} // namespace Icap
} // namespace Adaptation
 
#endif /* SQUID_SRC_ADAPTATION_ICAP_SERVICEREP_H */