// should be called only after setStatus, addHeader and addContent
void MTD_FLASHMEM HTTPResponse::flush()
{
// status line
m_httpHandler->getSocket()->writeFmt(FSTR("HTTP/1.1 %s\r\n"), m_status);
// HTTPResponse headers
addHeader(FSTR("Connection"), FSTR("close"));
// user headers
for (uint32_t i = 0; i != m_headers.getItemsCount(); ++i)
{
Fields::Item* item = m_headers[i];
m_httpHandler->getSocket()->writeFmt(FSTR("%s: %s\r\n"), APtr<char>(t_strdup(item->key)).get(), APtr<char>(t_strdup(item->value)).get());
}
// content length header
m_httpHandler->getSocket()->writeFmt(FSTR("%s: %d\r\n\r\n"), STR_Content_Length, m_content.getItemsCount());
// actual content
if (m_content.getItemsCount() > 0)
{
CharChunksIterator iter = m_content.getIterator();
CharChunkBase* chunk = iter.getCurrentChunk();
while (chunk)
{
m_httpHandler->getSocket()->write((uint8_t const*)chunk->data, chunk->getItems());
chunk = iter.moveToNextChunk();
}
m_content.clear();
}
}
// should be called only after setStatus, addHeader and addContent
void MTD_FLASHMEM HTTPResponse::flush() {
// headers
flushHeaders(m_content.getItemsCount());
// actual content
if (m_content.getItemsCount() > 0) {
CharChunksIterator iter = m_content.getIterator();
CharChunkBase *chunk = iter.getCurrentChunk();
while (chunk) {
if (m_httpHandler->getSocket()->write((uint8_t const *)chunk->data, chunk->getItems()) < 0)
break;
chunk = iter.moveToNextChunk();
}
m_content.clear();
}
}
CharChunksIterator MTD_FLASHMEM
HTTPHandler::extractURLEncodedFields(CharChunksIterator begin, CharChunksIterator end, Fields *fields) {
fields->setUrlDecode(true);
CharChunksIterator curc = begin;
CharChunksIterator key = curc;
CharChunksIterator value;
while (curc != end) {
if (*curc == '=') {
*curc = 0; // ends key
value = curc;
++value; // bypass '='
} else if (*curc == '&' || *curc == ' ' || curc.isLast()) {
bool endLoop = (*curc == ' ' || curc.isLast());
*curc++ = 0; // zero-ends value
if (key.isValid() && value.isValid()) {
fields->add(key, value); // store parameter
key = value = CharChunksIterator(); // reset
}
if (endLoop)
break;
key = curc;
} else
++curc;
}
return curc;
}
CharChunksIterator MTD_FLASHMEM
HTTPHandler::extractHeaders(CharChunksIterator begin, CharChunksIterator end, Fields *fields) {
CharChunksIterator curc = begin;
CharChunksIterator key;
CharChunksIterator value;
while (curc != end) {
if (*curc == 0x0D && key.isValid() && value.isValid()) // CR?
{
*curc = 0; // ends key
// store header
fields->add(key, value);
key = value = CharChunksIterator(); // reset
} else if (!isspace(*curc) && !key.isValid()) {
// bookmark "key"
key = curc;
} else if (!value.isValid() && *curc == ':') {
*curc++ = 0; // ends value
// bypass spaces
while (curc != end && isspace(*curc))
++curc;
// bookmark value
value = curc;
}
++curc;
}
return curc;
}
void MTD_FLASHMEM HTTPHandler::processXWWWFormUrlEncoded(CharChunksIterator headerEnd, int32_t contentLength) {
// look for data (maybe POST data)
if (contentLength > 0) {
// download additional content
int32_t missingBytes = headerEnd.getPosition() + contentLength - m_receivedData.getItemsCount();
while (getSocket()->isConnected() && missingBytes > 0) {
int32_t bytesToRead = (CHUNK_CAPACITY < missingBytes ? CHUNK_CAPACITY : missingBytes);
CharChunkBase *chunk = m_receivedData.addChunk(bytesToRead);
int32_t bytesRecv = getSocket()->read(chunk->data, bytesToRead);
if (bytesRecv <= 0)
break;
chunk->setItems(bytesRecv);
missingBytes -= chunk->getItems();
}
m_receivedData.append(0); // add additional terminating "0"
CharChunksIterator contentStart = m_receivedData.getIterator(); // cannot use directly headerEnd because added data
contentStart += headerEnd.getPosition();
extractURLEncodedFields(contentStart, CharChunksIterator(), &m_request.form);
}
dispatch();
}
void MTD_FLASHMEM LinkedCharChunks::dump()
{
for (CharChunksIterator i = getIterator(); i.isValid(); ++i)
debug(getChar(&*i));
}
bool MTD_FLASHMEM HTTPHandler::processRequest() {
// look for 0x0D 0x0A 0x0D 0x0A
CharChunksIterator headerEnd =
t_strstr(m_receivedData.getIterator(), CharChunksIterator(), CharIterator(FSTR("\x0D\x0A\x0D\x0A")));
if (headerEnd.isValid()) {
// move header end after CRLFCRLF
headerEnd += 4;
CharChunksIterator curc = m_receivedData.getIterator();
// extract method (GET, POST, etc..)
CharChunksIterator method = curc;
while (curc != headerEnd && *curc != ' ')
++curc;
*curc++ = 0; // ends method
if (t_strcmp(method, CharIterator(FSTR("GET"))) == 0)
m_request.method = Get;
else if (t_strcmp(method, CharIterator(FSTR("POST"))) == 0)
m_request.method = Post;
else if (t_strcmp(method, CharIterator(FSTR("HEAD"))) == 0)
m_request.method = Head;
else
m_request.method = Unsupported;
// extract requested page and query parameters
m_request.requestedPage = curc;
while (curc != headerEnd) {
if (*curc == '?') {
*curc++ = 0; // ends requestedPage
curc = extractURLEncodedFields(curc, headerEnd, &m_request.query);
break;
} else if (*curc == ' ') {
*curc++ = 0; // ends requestedPage
break;
}
++curc;
}
// bypass HTTP version
while (curc != headerEnd && *curc != 0x0D)
++curc;
// extract headers
curc = extractHeaders(curc, headerEnd, &m_request.headers);
// get content length (may be NULL)
char const *contentLengthStr = m_request.headers[STR_Content_Length];
int32_t contentLength = contentLengthStr ? strtol(contentLengthStr, NULL, 10) : 0;
if (m_request.method == Post) {
// check content type (POST)
char const *contentType = m_request.headers[STR_Content_Type];
if (contentType && f_strstr(contentType, FSTR("multipart/form-data"))) {
//// content type is multipart/form-data
processMultipartFormData(headerEnd, contentLength, contentType);
} else if (contentType == NULL ||
(contentType && f_strstr(contentType, FSTR("application/x-www-form-urlencoded")))) {
//// content type is application/x-www-form-urlencoded
processXWWWFormUrlEncoded(headerEnd, contentLength);
}
} else
dispatch();
return true;
} else {
// header is not complete
return false;
}
}
