← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/pbms/src/cslib/CSHTTPStream.h

  • Committer: Monty Taylor
  • Date: 2008-09-16 00:00:48 UTC
  • mto: This revision was merged to the branch mainline in revision 391.
  • Revision ID: monty@inaugust.com-20080916000048-3rvrv3gv9l0ad3gs
Fixed copyright headers in drizzled/

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/pbms/src/cslib/CSHTTPStream.h
1
 
/* Copyright (C) 2008 PrimeBase Technologies GmbH, Germany
2
 
 *
3
 
 * PrimeBase Media Stream for MySQL
4
 
 *
5
 
 * This program is free software; you can redistribute it and/or modify
6
 
 * it under the terms of the GNU General Public License as published by
7
 
 * the Free Software Foundation; either version 2 of the License, or
8
 
 * (at your option) any later version.
9
 
 *
10
 
 * This program is distributed in the hope that it will be useful,
11
 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
 * GNU General Public License for more details.
14
 
 *
15
 
 * You should have received a copy of the GNU General Public License
16
 
 * along with this program; if not, write to the Free Software
17
 
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18
 
 *
19
 
 * Original author: Paul McCullagh (H&G2JCtL)
20
 
 * Continued development: Barry Leslie
21
 
 *
22
 
 * 2007-06-10
23
 
 *
24
 
 *
25
 
 */
26
 
 
27
 
#ifndef __CSHTTPSTREAM_H__
28
 
#define __CSHTTPSTREAM_H__
29
 
 
30
 
#include "CSDefs.h"
31
 
#include "CSStorage.h"
32
 
#include "CSStream.h"
33
 
 
34
 
// The http tags before and after the exception message text must be well defined so that
35
 
// the client api can parse the error reply and get the error text out.
36
 
#define EXCEPTION_REPLY_MESSAGE_PREFIX_TAG "<P><B>"
37
 
#define EXCEPTION_REPLY_MESSAGE_SUFFIX_TAG "</B></P><PRE>"
38
 
#define EXCEPTION_REPLY_STACK_TRACE_SUFFIX_TAG "</PRE><P><font size=-1>"
39
 
 
40
 
class CSHeader : public CSRefObject {
41
 
public:
42
 
        CSHeader():iName(NULL), iValue(NULL) { }
43
 
        virtual ~CSHeader();
44
 
 
45
 
        void setName(const char *name);
46
 
        void setName(const char *name, uint32_t len);
47
 
        void setName(CSString *name);
48
 
 
49
 
        void setValue(const char *value);
50
 
        void setValue(const char *value, uint32_t len);
51
 
        void setValue(CSString *value);
52
 
 
53
 
        const char *getNameCString() { return iName ? iName->getCString() : ""; }
54
 
        const char *getValueCString() { return iValue ? iValue->getCString() : ""; }
55
 
 
56
 
        void write(CSOutputStream *out, bool trace);
57
 
        
58
 
        friend class CSHTTPHeaders;
59
 
private:
60
 
        CSString *getName() { return iName; } // Return a none referenced object!!
61
 
        CSString *getValue() { return iValue; }// Return a none referenced object!!
62
 
 
63
 
        CSString        *iName;
64
 
        CSString        *iValue;
65
 
};
66
 
 
67
 
class CSHTTPHeaders {
68
 
public:
69
 
        CSHTTPHeaders(CSVector *list = NULL): iHeaders(list), iKeepAlive(false), iExpect100Continue(false), iUnknownEpectHeader(false) { }
70
 
        virtual ~CSHTTPHeaders() { if (iHeaders) iHeaders->release();}
71
 
 
72
 
        void clearHeaders();
73
 
        CSVector * takeHeaders();
74
 
        void setHeaders(CSVector *headers);
75
 
        void addHeaders(CSHTTPHeaders *h);
76
 
        void addHeader(CSHeader *h);
77
 
        void addHeader(const char *name, const char *value);
78
 
        void addHeader(const char *name, uint32_t nlen, const char *value, uint32_t vlen);
79
 
        void addHeader(CSString *name, CSString *value);
80
 
        void addHeader(const char *name, CSString *value);
81
 
        void addHeader(const char *name, uint64_t value);
82
 
        void removeHeader(const char *name);
83
 
        void removeHeader(CSString *name);
84
 
        CSString *getHeaderValue(const char *name);
85
 
        const char *getHeaderCStringValue(const char *name);
86
 
        void writeHeader(CSOutputStream *out, bool trace);
87
 
        bool keepAlive();
88
 
        bool expect100Continue();
89
 
        bool unknownEpectHeader();
90
 
 
91
 
        uint32_t numHeaders() { return (iHeaders)?iHeaders->size(): 0; }
92
 
        CSHeader *getHeader(uint32_t idx) 
93
 
        { 
94
 
                CSHeader *header = NULL;
95
 
                
96
 
                if (iHeaders) 
97
 
                        header = (CSHeader *)(iHeaders->get(idx));
98
 
                        
99
 
                if (header)
100
 
                        header->retain();
101
 
                return header;
102
 
        }
103
 
        
104
 
private:
105
 
        CSVector *iHeaders;
106
 
        bool iKeepAlive;
107
 
        bool iExpect100Continue;
108
 
        bool iUnknownEpectHeader;
109
 
};
110
 
 
111
 
class CSRefHTTPHeaders : public CSHTTPHeaders, public CSRefObject {
112
 
public:
113
 
        CSRefHTTPHeaders(CSVector *list):CSHTTPHeaders(list){}
114
 
        ~CSRefHTTPHeaders(){}   
115
 
};
116
 
 
117
 
class CSHTTPInputStream : public CSInputStream, public CSHTTPHeaders {
118
 
public:
119
 
        CSHTTPInputStream(CSInputStream *s);
120
 
        virtual ~CSHTTPInputStream();
121
 
 
122
 
        void readHead(bool trace = false);
123
 
        void readBody();
124
 
        bool getContentLength(uint64_t *length);
125
 
        const char *getMethod();
126
 
        char *getBodyData() { return iBody.getCString(); };
127
 
        size_t getBodyLength() { return iBody.length(); };
128
 
        void setBody(CSStringBufferImpl *buf) { iBody.take(buf); }
129
 
        int getStatus() { return iStatus; }
130
 
        CSString *getStatusPhrase() { return iStatusPhrase; }
131
 
        CSString *getRequestURI() { return iRequestURI; }
132
 
        bool getRange(uint64_t *size, uint64_t *offset);
133
 
 
134
 
        virtual void close();
135
 
 
136
 
        virtual size_t read(char *b, size_t len);
137
 
 
138
 
        virtual int read();
139
 
 
140
 
        virtual int peek();
141
 
 
142
 
        virtual void reset() { iInput->reset(); }
143
 
 
144
 
        virtual const char *identify() { return iInput->identify(); }
145
 
 
146
 
        static CSHTTPInputStream *newStream(CSInputStream* i);
147
 
 
148
 
private:
149
 
        void freeHead();
150
 
 
151
 
        // Request-Line   = Method SP Request-URI SP HTTP-Version CRLF
152
 
        CSInputStream   *iInput;
153
 
        int                             iStatus;
154
 
        CSString                *iMethod;
155
 
        CSString                *iRequestURI;
156
 
        CSString                *iHTTPVersion;
157
 
        CSString                *iStatusPhrase;
158
 
        CSStringBuffer  iBody;
159
 
};
160
 
 
161
 
class CSHTTPOutputStream : public CSOutputStream, public CSHTTPHeaders {
162
 
public:
163
 
        CSHTTPOutputStream(CSOutputStream* s);
164
 
        virtual ~CSHTTPOutputStream();
165
 
 
166
 
        void setStatus(int status) { iStatus = status; }
167
 
        int getStatus() { return iStatus; }
168
 
        void setContentLength(uint64_t size) { iContentLength = size; }
169
 
        void setRange(uint64_t size, uint64_t offset, uint64_t total) { iRangeSize = size; iRangeOffset = offset; iTotalLength = total;}
170
 
 
171
 
        void writeHead(bool trace = false);     // Writes a standard HTTP header.
172
 
        void writeHeaders(bool trace = false); // Write the current headers.
173
 
 
174
 
        void clearBody();
175
 
        void writeBody();
176
 
 
177
 
        // The virtual and non virtual print() methods
178
 
        // must be kept seperate to avoid possible compiler
179
 
        // warnings about hidden methods.
180
 
        void print(const char *str, bool trace);
181
 
        void print(int32_t value, bool trace);
182
 
        void print(uint64_t value, bool trace);
183
 
 
184
 
        virtual void print(const char *str) {print(str, false);}
185
 
        virtual void print(CSString *s) {print(s->getCString(), false);}
186
 
        virtual void print(int32_t value) {print(value, false);}
187
 
        virtual void print(uint64_t value) {print(value, false);}
188
 
 
189
 
        void appendBody(const char *str);
190
 
        void appendBody(int32_t value);
191
 
        void appendBody(uint32_t value);
192
 
        void appendBody(uint64_t value);
193
 
        const char *getBodyData();
194
 
        size_t getBodyLength();
195
 
        void setBody(CSStringBufferImpl *buf);
196
 
 
197
 
        virtual void close();
198
 
 
199
 
        virtual void write(const char *b, size_t len);
200
 
 
201
 
        virtual const char *getEOL() { return "\r\n"; };
202
 
 
203
 
        virtual void flush();
204
 
 
205
 
        virtual void write(char b);
206
 
 
207
 
        virtual void reset() { iOutput->reset(); }
208
 
 
209
 
        virtual const char *identify() { return iOutput->identify(); }
210
 
 
211
 
        static const char *getReasonPhrase(int code);
212
 
 
213
 
        static CSHTTPOutputStream *newStream(CSOutputStream* i);
214
 
 
215
 
private:
216
 
        // Status-Line = HTTP-Version SP Status-Code SP Reason-Phrase CRLF
217
 
        CSOutputStream  *iOutput;
218
 
        int                             iStatus;
219
 
        uint64_t                        iContentLength;
220
 
        CSStringBuffer  iBody;
221
 
        uint64_t                        iRangeSize;
222
 
        uint64_t                        iRangeOffset;
223
 
        uint64_t                        iTotalLength;
224
 
};
225
 
 
226
 
#endif