mirror of
https://github.com/ukui/apt.git
synced 2026-03-09 09:35:45 -07:00
73780d7f66
We do not want to end up in a code path while reading content from the server where we have local data left to write, which can happen if a previous read included both headers and content. Restructure Flush() to accept a new argument to allow incomplete flushs (which do not match our limit), so that it can flush as far as possible, and modify Go() and use that before and after reading from the server.
143 lines
3.9 KiB
C++
143 lines
3.9 KiB
C++
// -*- mode: cpp; mode: fold -*-
|
|
// Description /*{{{*/
|
|
/* ######################################################################
|
|
|
|
HTTP Acquire Method - This is the HTTP acquire method for APT.
|
|
|
|
##################################################################### */
|
|
/*}}}*/
|
|
|
|
#ifndef APT_HTTP_H
|
|
#define APT_HTTP_H
|
|
|
|
#include <apt-pkg/strutl.h>
|
|
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <sys/time.h>
|
|
|
|
#include "basehttp.h"
|
|
#include "connect.h"
|
|
|
|
using std::cout;
|
|
using std::endl;
|
|
|
|
class FileFd;
|
|
class HttpMethod;
|
|
class Hashes;
|
|
|
|
class CircleBuf
|
|
{
|
|
unsigned char *Buf;
|
|
unsigned long long Size;
|
|
unsigned long long InP;
|
|
unsigned long long OutP;
|
|
std::string OutQueue;
|
|
unsigned long long StrPos;
|
|
unsigned long long MaxGet;
|
|
|
|
static unsigned long long BwReadLimit;
|
|
static unsigned long long BwTickReadData;
|
|
static std::chrono::steady_clock::duration BwReadTick;
|
|
static const unsigned int BW_HZ;
|
|
|
|
unsigned long long LeftRead() const
|
|
{
|
|
unsigned long long Sz = Size - (InP - OutP);
|
|
if (Sz > Size - (InP%Size))
|
|
Sz = Size - (InP%Size);
|
|
return Sz;
|
|
}
|
|
unsigned long long LeftWrite() const
|
|
{
|
|
unsigned long long Sz = InP - OutP;
|
|
if (InP > MaxGet)
|
|
Sz = MaxGet - OutP;
|
|
if (Sz > Size - (OutP%Size))
|
|
Sz = Size - (OutP%Size);
|
|
return Sz;
|
|
}
|
|
void FillOut();
|
|
|
|
public:
|
|
Hashes *Hash;
|
|
// total amount of data that got written so far
|
|
unsigned long long TotalWriten;
|
|
|
|
// Read data in
|
|
bool Read(std::unique_ptr<MethodFd> const &Fd);
|
|
bool Read(std::string const &Data);
|
|
|
|
// Write data out
|
|
bool Write(std::unique_ptr<MethodFd> const &Fd);
|
|
bool Write(std::string &Data);
|
|
bool WriteTillEl(std::string &Data,bool Single = false);
|
|
|
|
// Control the write limit
|
|
void Limit(long long Max) {if (Max == -1) MaxGet = 0-1; else MaxGet = OutP + Max;}
|
|
bool IsLimit() const {return MaxGet == OutP;};
|
|
void Print() const {cout << MaxGet << ',' << OutP << endl;};
|
|
|
|
// Test for free space in the buffer
|
|
bool ReadSpace() const {return Size - (InP - OutP) > 0;};
|
|
bool WriteSpace() const {return InP - OutP > 0;};
|
|
|
|
void Reset();
|
|
|
|
CircleBuf(HttpMethod const * const Owner, unsigned long long Size);
|
|
~CircleBuf();
|
|
};
|
|
|
|
struct HttpServerState: public ServerState
|
|
{
|
|
// This is the connection itself. Output is data FROM the server
|
|
CircleBuf In;
|
|
CircleBuf Out;
|
|
std::unique_ptr<MethodFd> ServerFd;
|
|
|
|
protected:
|
|
virtual bool ReadHeaderLines(std::string &Data) APT_OVERRIDE;
|
|
virtual ResultState LoadNextResponse(bool const ToFile, RequestState &Req) APT_OVERRIDE;
|
|
virtual bool WriteResponse(std::string const &Data) APT_OVERRIDE;
|
|
|
|
public:
|
|
virtual void Reset() APT_OVERRIDE;
|
|
|
|
virtual ResultState RunData(RequestState &Req) APT_OVERRIDE;
|
|
virtual ResultState RunDataToDevNull(RequestState &Req) APT_OVERRIDE;
|
|
|
|
virtual ResultState Open() APT_OVERRIDE;
|
|
virtual bool IsOpen() APT_OVERRIDE;
|
|
virtual bool Close() APT_OVERRIDE;
|
|
virtual bool InitHashes(HashStringList const &ExpectedHashes) APT_OVERRIDE;
|
|
virtual Hashes * GetHashes() APT_OVERRIDE;
|
|
virtual ResultState Die(RequestState &Req) APT_OVERRIDE;
|
|
virtual bool Flush(FileFd *const File, bool MustComplete = true) APT_OVERRIDE;
|
|
virtual ResultState Go(bool ToFile, RequestState &Req) APT_OVERRIDE;
|
|
|
|
HttpServerState(URI Srv, HttpMethod *Owner);
|
|
virtual ~HttpServerState() {Close();};
|
|
};
|
|
|
|
class HttpMethod : public BaseHttpMethod
|
|
{
|
|
public:
|
|
virtual void SendReq(FetchItem *Itm) APT_OVERRIDE;
|
|
|
|
virtual std::unique_ptr<ServerState> CreateServerState(URI const &uri) APT_OVERRIDE;
|
|
virtual void RotateDNS() APT_OVERRIDE;
|
|
virtual DealWithHeadersResult DealWithHeaders(FetchResult &Res, RequestState &Req) APT_OVERRIDE;
|
|
|
|
protected:
|
|
std::string AutoDetectProxyCmd;
|
|
|
|
public:
|
|
friend struct HttpServerState;
|
|
|
|
explicit HttpMethod(std::string &&pProg);
|
|
};
|
|
|
|
#endif
|