helper.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.
 */
 
/* DEBUG: section 84    Helper process maintenance */
 
#ifndef SQUID_HELPER_H
#define SQUID_HELPER_H
 
#include "base/AsyncCall.h"
#include "base/InstanceId.h"
#include "cbdata.h"
#include "comm/forward.h"
#include "dlink.h"
#include "helper/ChildConfig.h"
#include "helper/forward.h"
#include "helper/Reply.h"
#include "helper/Request.h"
#include "helper/ReservationId.h"
#include "ip/Address.h"
#include "sbuf/SBuf.h"
 
#include <list>
#include <map>
#include <queue>
#include <unordered_map>
 
class Packable;
class wordlist;
 
namespace Helper
{
class Xaction {
    MEMPROXY_CLASS(Helper::Xaction);
public:
    Xaction(HLPCB *c, void *d, const char *b): request(c, d, b) {}
    Helper::Request request;
    Helper::Reply reply;
};
}
 
class HelperServerBase;
class helper
{
    CBDATA_CLASS(helper);
 
public:
    inline helper(const char *name) :
        cmdline(nullptr),
        id_name(name),
        ipc_type(0),
        droppedRequests(0),
        overloadStart(0),
        last_queue_warn(0),
        last_restart(0),
        timeout(0),
        retryTimedOut(false),
        retryBrokenHelper(false),
        eom('\n') {
        memset(&stats, 0, sizeof(stats));
    }
    ~helper();
 
    Helper::Xaction *nextRequest();
 
    bool trySubmit(const char *buf, HLPCB * callback, void *data);
 
    void submitRequest(Helper::Xaction *r);
 
    void packStatsInto(Packable *p, const char *label = nullptr) const;
    bool willOverload() const;
 
    void handleKilledServer(HelperServerBase *srv, bool &needsNewServers);
 
public:
    wordlist *cmdline;
    dlink_list servers;
    std::queue<Helper::Xaction *> queue;
    const char *id_name;
    Helper::ChildConfig childs;    
    int ipc_type;
    Ip::Address addr;
    unsigned int droppedRequests; 
    time_t overloadStart; 
    time_t last_queue_warn;
    time_t last_restart;
    time_t timeout; 
    bool retryTimedOut; 
    bool retryBrokenHelper; 
    SBuf onTimedOutResponse; 
    char eom;   
 
    struct _stats {
        int requests;
        int replies;
        int timedout;
        int queue_size;
        int avg_svc_time;
    } stats;
 
protected:
    friend void helperSubmit(helper * hlp, const char *buf, HLPCB * callback, void *data);
    bool queueFull() const;
    bool overloaded() const;
    void syncQueueStats();
    bool prepSubmit();
    void submit(const char *buf, HLPCB * callback, void *data);
};
 
class statefulhelper : public helper
{
    CBDATA_CLASS(statefulhelper);
 
public:
    typedef std::unordered_map<Helper::ReservationId, helper_stateful_server *> Reservations;
 
    inline statefulhelper(const char *name) : helper(name) {}
    inline ~statefulhelper() {}
 
public:
    void reserveServer(helper_stateful_server * srv);
 
    void cancelReservation(const Helper::ReservationId reservation);
 
private:
    friend void helperStatefulSubmit(statefulhelper * hlp, const char *buf, HLPCB * callback, void *data, const Helper::ReservationId & reservation);
 
    helper_stateful_server *findServer(const Helper::ReservationId & reservation);
 
    void submit(const char *buf, HLPCB * callback, void *data, const Helper::ReservationId & reservation);
    bool trySubmit(const char *buf, HLPCB * callback, void *data, const Helper::ReservationId & reservation);
 
    Reservations reservations;
};
 
class HelperServerBase: public CbdataParent
{
public:
    ~HelperServerBase() override;
    void closePipesSafely(const char *name);
 
    void closeWritePipeSafely(const char *name);
 
    // TODO: Teach each child to report its child-specific state instead.
    virtual bool reserved() = 0;
 
    virtual void dropQueued();
 
    virtual helper *getParent() const = 0;
 
public:
    const InstanceId<HelperServerBase> index;
    int pid;
    Ip::Address addr;
    Comm::ConnectionPointer readPipe;
    Comm::ConnectionPointer writePipe;
    void *hIpc;
 
    char *rbuf;
    size_t rbuf_sz;
    size_t roffset;
 
    struct timeval dispatch_time;
    struct timeval answer_time;
 
    dlink_node link;
 
    struct _helper_flags {
        bool writing;
        bool closing;
        bool shutdown;
    } flags;
 
    typedef std::list<Helper::Xaction *> Requests;
    Requests requests; 
 
    struct {
        uint64_t uses;     //< requests sent to this helper
        uint64_t replies;  //< replies received from this helper
        uint64_t pending;  //< queued lookups waiting to be sent to this helper
        uint64_t releases; //< times release() has been called on this helper (if stateful)
        uint64_t timedout; //< requests which timed-out
    } stats;
    void initStats();
};
 
class MemBuf;
class CommTimeoutCbParams;
 
// TODO: Rename to StatelessHelperServer and rename HelperServerBase to HelperServer.
class helper_server : public HelperServerBase
{
    CBDATA_CHILD(helper_server);
 
public:
    uint64_t nextRequestId;
 
    MemBuf *wqueue;
    MemBuf *writebuf;
 
    helper *parent;
 
    Helper::Xaction *replyXaction;
 
    bool ignoreToEom;
 
    // STL says storing std::list iterators is safe when changing the list
    typedef std::map<uint64_t, Requests::iterator> RequestIndex;
    RequestIndex requestsIndex; 
 
    ~helper_server() override;
    Helper::Xaction *popRequest(int requestId);
 
    void checkForTimedOutRequests(bool const retry);
 
    /*HelperServerBase API*/
    bool reserved() override {return false;}
    void dropQueued() override;
    helper *getParent() const override {return parent;}
 
    static void requestTimeout(const CommTimeoutCbParams &io);
 
    static void HelperServerClosed(helper_server *srv);
};
 
// TODO: Rename to StatefulHelperServer and rename HelperServerBase to HelperServer.
class helper_stateful_server : public HelperServerBase
{
    CBDATA_CHILD(helper_stateful_server);
 
public:
    ~helper_stateful_server() override;
    void reserve();
    void clearReservation();
 
    /* HelperServerBase API */
    bool reserved() override {return reservationId.reserved();}
    helper *getParent() const override {return parent;}
 
    static void HelperServerClosed(helper_stateful_server *srv);
 
    statefulhelper *parent;
 
    // Reservations temporary lock the server for an exclusive "client" use. The
    // client keeps the reservation ID as a proof of her reservation. If a
    // reservation expires, and the server is reserved for another client, then
    // the reservation ID presented by the late client will not match ours.
    Helper::ReservationId reservationId; 
    time_t reservationStart; 
};
 
/* helper.c */
void helperOpenServers(helper * hlp);
void helperStatefulOpenServers(statefulhelper * hlp);
void helperSubmit(helper * hlp, const char *buf, HLPCB * callback, void *data);
void helperStatefulSubmit(statefulhelper * hlp, const char *buf, HLPCB * callback, void *data, uint64_t reservation);
void helperShutdown(helper * hlp);
void helperStatefulShutdown(statefulhelper * hlp);
 
#endif /* SQUID_HELPER_H */