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 Aquire Method - This is the HTTP aquire method for APT.
	7
	8	##################################################################### */
	9	/}}}/
	10
	11	#ifndef APT_HTTP_H
	12	#define APT_HTTP_H
	13
	14	#define MAXLEN 360
	15
	16	#include <iostream>
	17
	18	using std::cout;
	19	using std::endl;
	20
	21	class HttpMethod;
	22
	23	class CircleBuf
	24	{
	25	unsigned char *Buf;
	26	unsigned long Size;
	27	unsigned long InP;
	28	unsigned long OutP;
	29	string OutQueue;
	30	unsigned long StrPos;
	31	unsigned long MaxGet;
	32	struct timeval Start;
	33
	34	unsigned long LeftRead()
	35	{
	36	unsigned long Sz = Size - (InP - OutP);
	37	if (Sz > Size - (InP%Size))
	38	Sz = Size - (InP%Size);
	39	return Sz;
	40	}
	41	unsigned long LeftWrite()
	42	{
	43	unsigned long Sz = InP - OutP;
	44	if (InP > MaxGet)
	45	Sz = MaxGet - OutP;
	46	if (Sz > Size - (OutP%Size))
	47	Sz = Size - (OutP%Size);
	48	return Sz;
	49	}
	50	void FillOut();
	51
	52	public:
	53
	54	Hashes *Hash;
	55
	56	// Read data in
	57	bool Read(int Fd);
	58	bool Read(string Data);
	59
	60	// Write data out
	61	bool Write(int Fd);
	62	bool WriteTillEl(string &Data,bool Single = false);
	63
	64	// Control the write limit
	65	void Limit(long Max) {if (Max == -1) MaxGet = 0-1; else MaxGet = OutP + Max;}
	66	bool IsLimit() {return MaxGet == OutP;};
	67	void Print() {cout << MaxGet << ',' << OutP << endl;};
	68
	69	// Test for free space in the buffer
	70	bool ReadSpace() {return Size - (InP - OutP) > 0;};
	71	bool WriteSpace() {return InP - OutP > 0;};
	72
	73	// Dump everything
	74	void Reset();
	75	void Stats();
	76
	77	CircleBuf(unsigned long Size);
	78	~CircleBuf() {delete [] Buf; delete Hash;};
	79	};
	80
	81	struct ServerState
	82	{
	83	// This is the last parsed Header Line
	84	unsigned int Major;
	85	unsigned int Minor;
	86	unsigned int Result;
	87	char Code[MAXLEN];
	88
	89	// These are some statistics from the last parsed header lines
	90	unsigned long Size;
	91	signed long StartPos;
	92	time_t Date;
	93	bool HaveContent;
	94	enum {Chunked,Stream,Closes} Encoding;
	95	enum {Header, Data} State;
	96	bool Persistent;
	97
	98	// This is a Persistent attribute of the server itself.
	99	bool Pipeline;
	100
	101	HttpMethod *Owner;
	102
	103	// This is the connection itself. Output is data FROM the server
	104	CircleBuf In;
	105	CircleBuf Out;
	106	int ServerFd;
	107	URI ServerName;
	108
	109	bool HeaderLine(string Line);
	110	bool Comp(URI Other) {return Other.Host == ServerName.Host && Other.Port == ServerName.Port;};
	111	void Reset() {Major = 0; Minor = 0; Result = 0; Size = 0; StartPos = 0;
	112	Encoding = Closes; time(&Date); ServerFd = -1;
	113	Pipeline = true;};
	114	int RunHeaders();
	115	bool RunData();
	116
	117	bool Open();
	118	bool Close();
	119
	120	ServerState(URI Srv,HttpMethod *Owner);
	121	~ServerState() {Close();};
	122	};
	123
	124	class HttpMethod : public pkgAcqMethod
	125	{
	126	void SendReq(FetchItem *Itm,CircleBuf &Out);
	127	bool Go(bool ToFile,ServerState *Srv);
	128	bool Flush(ServerState *Srv);
	129	bool ServerDie(ServerState *Srv);
	130	int DealWithHeaders(FetchResult &Res,ServerState *Srv);
	131
	132	virtual bool Fetch(FetchItem *);
	133	virtual bool Configuration(string Message);
	134
	135	// In the event of a fatal signal this file will be closed and timestamped.
	136	static string FailFile;
	137	static int FailFd;
	138	static time_t FailTime;
	139	static void SigTerm(int);
	140
	141	public:
	142	friend class ServerState;
	143
	144	FileFd *File;
	145	ServerState *Server;
	146
	147	int Loop();
	148
	149	HttpMethod() : pkgAcqMethod("1.2",Pipeline \| SendConfig)
	150	{
	151	File = 0;
	152	Server = 0;
	153	};
	154	};
	155
	156	URI Proxy;
	157
	158	#endif