git.saurik.com Git - apt.git/blame_incremental

... / ...

Commit	Line	Data
	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/// $Id: http.h,v 1.12 2002/04/18 05:09:38 jgg Exp $
	3	// $Id: http.h,v 1.12 2002/04/18 05:09:38 jgg Exp $
	4	/* ######################################################################
	5
	6	HTTP Acquire Method - This is the HTTP acquire method for APT.
	7
	8	##################################################################### */
	9	/}}}/
	10
	11	#ifndef APT_HTTP_H
	12	#define APT_HTTP_H
	13
	14	#include <apt-pkg/strutl.h>
	15	#include <apt-pkg/acquire-method.h>
	16
	17	#include <string>
	18	#include <sys/time.h>
	19	#include <iostream>
	20
	21	#include "server.h"
	22
	23	using std::cout;
	24	using std::endl;
	25
	26	class FileFd;
	27	class HttpMethod;
	28	class Hashes;
	29
	30	class CircleBuf
	31	{
	32	unsigned char *Buf;
	33	unsigned long long Size;
	34	unsigned long long InP;
	35	unsigned long long OutP;
	36	std::string OutQueue;
	37	unsigned long long StrPos;
	38	unsigned long long MaxGet;
	39	struct timeval Start;
	40
	41	static unsigned long long BwReadLimit;
	42	static unsigned long long BwTickReadData;
	43	static struct timeval BwReadTick;
	44	static const unsigned int BW_HZ;
	45
	46	unsigned long long LeftRead() const
	47	{
	48	unsigned long long Sz = Size - (InP - OutP);
	49	if (Sz > Size - (InP%Size))
	50	Sz = Size - (InP%Size);
	51	return Sz;
	52	}
	53	unsigned long long LeftWrite() const
	54	{
	55	unsigned long long Sz = InP - OutP;
	56	if (InP > MaxGet)
	57	Sz = MaxGet - OutP;
	58	if (Sz > Size - (OutP%Size))
	59	Sz = Size - (OutP%Size);
	60	return Sz;
	61	}
	62	void FillOut();
	63
	64	public:
	65	Hashes *Hash;
	66	// total amount of data that got written so far
	67	unsigned long long TotalWriten;
	68
	69	// Read data in
	70	bool Read(int Fd);
	71	bool Read(std::string Data);
	72
	73	// Write data out
	74	bool Write(int Fd);
	75	bool WriteTillEl(std::string &Data,bool Single = false);
	76
	77	// Control the write limit
	78	void Limit(long long Max) {if (Max == -1) MaxGet = 0-1; else MaxGet = OutP + Max;}
	79	bool IsLimit() const {return MaxGet == OutP;};
	80	void Print() const {cout << MaxGet << ',' << OutP << endl;};
	81
	82	// Test for free space in the buffer
	83	bool ReadSpace() const {return Size - (InP - OutP) > 0;};
	84	bool WriteSpace() const {return InP - OutP > 0;};
	85
	86	void Reset();
	87	// Dump everything
	88	void Stats();
	89
	90	explicit CircleBuf(unsigned long long Size);
	91	~CircleBuf();
	92	};
	93
	94	struct HttpServerState: public ServerState
	95	{
	96	// This is the connection itself. Output is data FROM the server
	97	CircleBuf In;
	98	CircleBuf Out;
	99	int ServerFd;
	100
	101	protected:
	102	virtual bool ReadHeaderLines(std::string &Data) APT_OVERRIDE;
	103	virtual bool LoadNextResponse(bool const ToFile, FileFd * const File) APT_OVERRIDE;
	104	virtual bool WriteResponse(std::string const &Data) APT_OVERRIDE;
	105
	106	public:
	107	virtual void Reset() APT_OVERRIDE { ServerState::Reset(); ServerFd = -1; };
	108
	109	virtual bool RunData(FileFd * const File) APT_OVERRIDE;
	110
	111	virtual bool Open() APT_OVERRIDE;
	112	virtual bool IsOpen() APT_OVERRIDE;
	113	virtual bool Close() APT_OVERRIDE;
	114	virtual bool InitHashes(HashStringList const &ExpectedHashes) APT_OVERRIDE;
	115	virtual Hashes * GetHashes() APT_OVERRIDE;
	116	virtual bool Die(FileFd &File) APT_OVERRIDE;
	117	virtual bool Flush(FileFd * const File) APT_OVERRIDE;
	118	virtual bool Go(bool ToFile, FileFd * const File) APT_OVERRIDE;
	119
	120	HttpServerState(URI Srv, HttpMethod *Owner);
	121	virtual ~HttpServerState() {Close();};
	122	};
	123
	124	class HttpMethod : public ServerMethod
	125	{
	126	public:
	127	virtual void SendReq(FetchItem *Itm) APT_OVERRIDE;
	128
	129	virtual bool Configuration(std::string Message) APT_OVERRIDE;
	130
	131	virtual std::unique_ptr<ServerState> CreateServerState(URI const &uri) APT_OVERRIDE;
	132	virtual void RotateDNS() APT_OVERRIDE;
	133
	134	protected:
	135	std::string AutoDetectProxyCmd;
	136
	137	public:
	138	friend struct HttpServerState;
	139
	140	HttpMethod() : ServerMethod("http", "1.2",Pipeline \| SendConfig)
	141	{
	142	File = 0;
	143	Server = 0;
	144	};
	145	};
	146
	147	#endif