// 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; } }