101 lines
2.7 KiB
C++
101 lines
2.7 KiB
C++
#include "IMFMessage.h"
|
|
#include "IMFHeader.h"
|
|
#include "IMFMultipart.h"
|
|
#include "IMFPlainText.h"
|
|
#include "Log.h"
|
|
|
|
namespace coreutils {
|
|
|
|
IMFMessage::IMFMessage() : ZString() {
|
|
}
|
|
|
|
IMFMessage::IMFMessage(ZString &in) : ZString(in) {
|
|
|
|
while (!in.ifNext("\r\n")) {
|
|
headers.emplace_back(in);
|
|
}
|
|
|
|
ZString type = getHeader("Content-Type");
|
|
int len = getHeader("Content-Length").asInteger();
|
|
if(len > 0) {
|
|
ZString block = in.readBlock(len);
|
|
if(type.equals("multipart/form-data"))
|
|
body = new IMFMultipart(block, getHeaderKeyPairValue("Content-Type", "boundary"));
|
|
else if(type.equals("text/plain"))
|
|
body = new IMFPlainText(block);
|
|
else
|
|
body = new IMFBody(block);
|
|
}
|
|
}
|
|
|
|
void IMFMessage::output(std::stringstream &out) {
|
|
for(IMFHeader header: headers) {
|
|
out << header.getKey() << ": " << header.getValue() << "\r\n";
|
|
}
|
|
out << "\r\n";
|
|
}
|
|
|
|
void IMFMessage::addHeader(ZString key, ZString value) {
|
|
headers.emplace_back(key, value);
|
|
}
|
|
|
|
void IMFMessage::addHeader(const char *key, ZString value) {
|
|
headers.emplace_back(ZString(key), value);
|
|
}
|
|
|
|
ZString IMFMessage::getHeader(ZString key, bool valueOnly) {
|
|
for(IMFHeader header: headers) {
|
|
if(header.getKey().equals(key)) {
|
|
ZString value = header.getValue();
|
|
if(valueOnly) {
|
|
std::vector<ZString> split = ZString(value).split(";");
|
|
value = split[0];
|
|
}
|
|
return value;
|
|
}
|
|
}
|
|
return ZString();
|
|
}
|
|
|
|
ZString IMFMessage::getHeaderKeyPairValue(const char *headerKey, const char *key) {
|
|
return getHeaderKeyPairValue(ZString(headerKey), ZString(key));
|
|
}
|
|
|
|
ZString IMFMessage::getHeaderKeyPairValue(ZString headerKey, ZString key) {
|
|
ZString value;
|
|
ZString psource(getHeader(headerKey, false));
|
|
std::vector<ZString> sourcep = psource.split(";");
|
|
for(ZString work: sourcep) {
|
|
work.skipWhitespace();
|
|
ZString token = work.getTokenExclude("=");
|
|
if(work.ifNext((char *)"=")) {
|
|
if(token.equals(key)) {
|
|
if(work.ifNext((char *)"\"")) {
|
|
value = work.getTokenExclude((char *)"\"");
|
|
return value;
|
|
}
|
|
else {
|
|
value = work;
|
|
return value;
|
|
}
|
|
}
|
|
else
|
|
continue;
|
|
|
|
} else if (work.ifNext((char *)";") || work.ifNext((char *)" ")) {
|
|
if(token.equals(key))
|
|
return ZString((char *)"true");
|
|
else
|
|
continue;
|
|
|
|
}
|
|
}
|
|
return ZString((char *)"false");
|
|
}
|
|
|
|
IMFBody * IMFMessage::getBody() {
|
|
return body;
|
|
}
|
|
|
|
}
|