Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
branch: master
158 lines (132 sloc) 4.223 kb
#ifndef _MULTIPART_READER_H_
#define _MULTIPART_READER_H_
#include <map>
#include <utility>
#include "MultipartParser.h"
class MultipartHeaders: public std::multimap<std::string, std::string> {
private:
std::string empty;
public:
const std::string &operator[](const std::string &key) const {
const_iterator it = find(key);
if (it == end()) {
return empty;
} else {
return it->second;
}
}
};
class MultipartReader {
public:
typedef void (*PartBeginCallback)(const MultipartHeaders &headers, void *userData);
typedef void (*PartDataCallback)(const char *buffer, size_t size, void *userData);
typedef void (*Callback)(void *userData);
private:
MultipartParser parser;
bool headersProcessed;
MultipartHeaders currentHeaders;
std::string currentHeaderName, currentHeaderValue;
void resetReaderCallbacks() {
onPartBegin = NULL;
onPartData = NULL;
onPartEnd = NULL;
onEnd = NULL;
userData = NULL;
}
void setParserCallbacks() {
parser.onPartBegin = cbPartBegin;
parser.onHeaderField = cbHeaderField;
parser.onHeaderValue = cbHeaderValue;
parser.onHeaderEnd = cbHeaderEnd;
parser.onHeadersEnd = cbHeadersEnd;
parser.onPartData = cbPartData;
parser.onPartEnd = cbPartEnd;
parser.onEnd = cbEnd;
parser.userData = this;
}
static void cbPartBegin(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
self->headersProcessed = false;
self->currentHeaders.clear();
self->currentHeaderName.clear();
self->currentHeaderValue.clear();
}
static void cbHeaderField(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
self->currentHeaderName.append(buffer + start, end - start);
}
static void cbHeaderValue(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
self->currentHeaderValue.append(buffer + start, end - start);
}
static void cbHeaderEnd(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
self->currentHeaders.insert(std::make_pair(self->currentHeaderName,
self->currentHeaderValue));
self->currentHeaderName.clear();
self->currentHeaderValue.clear();
}
static void cbHeadersEnd(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
if (self->onPartBegin != NULL) {
self->onPartBegin(self->currentHeaders, self->userData);
}
self->currentHeaders.clear();
self->currentHeaderName.clear();
self->currentHeaderValue.clear();
}
static void cbPartData(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
if (self->onPartData != NULL) {
self->onPartData(buffer + start, end - start, self->userData);
}
}
static void cbPartEnd(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
if (self->onPartEnd != NULL) {
self->onPartEnd(self->userData);
}
}
static void cbEnd(const char *buffer, size_t start, size_t end, void *userData) {
MultipartReader *self = (MultipartReader *) userData;
if (self->onEnd != NULL) {
self->onEnd(self->userData);
}
}
public:
PartBeginCallback onPartBegin;
PartDataCallback onPartData;
Callback onPartEnd;
Callback onEnd;
void *userData;
MultipartReader() {
resetReaderCallbacks();
setParserCallbacks();
}
MultipartReader(const std::string &boundary): parser(boundary) {
resetReaderCallbacks();
setParserCallbacks();
}
void reset() {
parser.reset();
}
void setBoundary(const std::string &boundary) {
parser.setBoundary(boundary);
}
size_t feed(const char *buffer, size_t len) {
return parser.feed(buffer, len);
}
bool succeeded() const {
return parser.succeeded();
}
bool hasError() const {
return parser.hasError();
}
bool stopped() const {
return parser.stopped();
}
const char *getErrorMessage() const {
return parser.getErrorMessage();
}
};
#endif /* _MULTIPART_READER_H_ */
Jump to Line
Something went wrong with that request. Please try again.