Xaction.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_XACTION_H
#define SQUID_SRC_ADAPTATION_ICAP_XACTION_H
 
#include "AccessLogEntry.h"
#include "adaptation/icap/ServiceRep.h"
#include "adaptation/Initiate.h"
#include "base/JobWait.h"
#include "comm/ConnOpener.h"
#include "error/forward.h"
#include "HttpReply.h"
#include "ipcache.h"
#include "sbuf/SBuf.h"
 
class MemBuf;
 
namespace Ssl {
class IcapPeerConnector;
}
 
namespace Adaptation
{
namespace Icap
{
 
/*
 * The ICAP Xaction implements common tasks for ICAP OPTIONS, REQMOD, and
 * RESPMOD transactions. It is started by an Initiator. It terminates
 * on its own, when done. Transactions communicate with Initiator using
 * asynchronous messages because a transaction or Initiator may be gone at
 * any time.
 */
 
// Note: Xaction must be the first parent for object-unaware cbdata to work
 
class Xaction: public Adaptation::Initiate
{
 
public:
    Xaction(const char *aTypeName, ServiceRep::Pointer &aService);
    ~Xaction() override;
 
    void disableRetries();
    void disableRepeats(const char *reason);
    bool retriable() const { return isRetriable; }
    bool repeatable() const { return isRepeatable; }
 
    // comm handler wrappers, treat as private
    void noteCommConnected(const CommConnectCbParams &io);
    void noteCommWrote(const CommIoCbParams &io);
    void noteCommRead(const CommIoCbParams &io);
    void noteCommTimedout(const CommTimeoutCbParams &io);
    void noteCommClosed(const CommCloseCbParams &io);
 
    // TODO: create these only when actually sending/receiving
    HttpRequest *icapRequest; 
    HttpReply::Pointer icapReply; 
 
    int attempts;
 
protected:
    void start() override;
    void noteInitiatorAborted() override; // TODO: move to Adaptation::Initiate
 
    virtual void startShoveling() = 0;
 
    // comm hanndlers; called by comm handler wrappers
    virtual void handleCommWrote(size_t sz) = 0;
    virtual void handleCommRead(size_t sz) = 0;
 
    void handleSecuredPeer(Security::EncryptorAnswer &answer);
    virtual void detailError(const ErrorDetailPointer &) {}
 
    void openConnection();
    void closeConnection();
    bool haveConnection() const;
 
    void scheduleRead();
    void scheduleWrite(MemBuf &buf);
    void updateTimeout();
 
    void cancelRead();
 
    bool parseHttpMsg(Http::Message *msg); // true=success; false=needMore; throw=err
    bool mayReadMore() const;
 
    virtual bool doneReading() const;
    virtual bool doneWriting() const;
    bool doneWithIo() const;
    bool doneAll() const override;
 
    // called just before the 'done' transaction is deleted
    void swanSong() override;
 
    // returns a temporary string depicting transaction status, for debugging
    const char *status() const override;
    virtual void fillPendingStatus(MemBuf &buf) const;
    virtual void fillDoneStatus(MemBuf &buf) const;
 
    // useful for debugging
    virtual bool fillVirginHttpHeader(MemBuf&) const;
 
public:
    // custom exception handling and end-of-call checks
    void callException(const std::exception  &e) override;
    void callEnd() override;
    virtual void clearError() {}
    virtual AccessLogEntry::Pointer masterLogEntry();
    void dnsLookupDone(std::optional<Ip::Address>);
 
protected:
    // logging
    void setOutcome(const XactOutcome &xo);
    virtual void finalizeLogInfo();
 
public:
    ServiceRep &service();
 
private:
    void useTransportConnection(const Comm::ConnectionPointer &);
    void useIcapConnection(const Comm::ConnectionPointer &);
    void dieOnConnectionFailure();
    void tellQueryAborted();
    void maybeLog();
 
protected:
    Adaptation::Icap::ServiceRep::Pointer theService;
 
    SBuf readBuf;
    bool commEof;
    bool reuseConnection;
    bool isRetriable;  
    bool isRepeatable; 
    bool ignoreLastWrite;
    bool waitingForDns; 
 
    AsyncCall::Pointer reader;
    AsyncCall::Pointer writer;
 
    AccessLogEntry::Pointer alep; 
    AccessLogEntry &al; 
 
    timeval icap_tr_start;     /*time when the ICAP transaction was created */
    timeval icap_tio_start;    /*time when the first ICAP request byte was scheduled for sending*/
    timeval icap_tio_finish;   /*time when the last byte of the ICAP responsewas received*/
 
private:
    JobWait<Comm::ConnOpener> transportWait;
 
    JobWait<Ssl::IcapPeerConnector> encryptionWait;
 
    Comm::ConnectionPointer connection;
 
    AsyncCall::Pointer closer;
};
 
} // namespace Icap
} // namespace Adaptation
 
#endif /* SQUID_SRC_ADAPTATION_ICAP_XACTION_H */