HTTP server: Greatly improved HTTP request parsing

src/webui/httpconnection.cpp

@@ -78,26 +78,41 @@ void HttpConnection::handleDownloadFailure(const QString& url,
 }
 
 void HttpConnection::read() {
-  static QByteArray input;
+  QByteArray input = m_socket->readAll();
 
-  input.append(m_socket->readAll());
-  if(input.size() > 100000) {
-    qDebug("Request too big");
-    input.clear();
+  // Parse HTTP request header
+  int header_end = input.indexOf("\r\n\r\n");
+  QByteArray header = input.left(header_end);
+  m_parser.writeHeader(header);
+  if (m_parser.isError()) {
+    qDebug() << Q_FUNC_INFO << "parsing error";
     m_generator.setStatusLine(400, "Bad Request");
     write();
     return;
   }
 
-  if (!input.endsWith("\r\n")) {
-    // incomplete, let wait for more data
-    qDebug() << Q_FUNC_INFO << "Incomplete HTTP request, let's wait for more data...";
+  // Parse HTTP request message
+  if (m_parser.header().hasContentLength())  {
+    QByteArray message = input.mid(header_end + 4);
+    int expected_length = m_parser.header().contentLength();
+
+    if (expected_length > 100000) {
+      m_generator.setStatusLine(400, "Bad Request");
+      write();
       return;
     }
 
-  // HTTP Request is complete, let's parse it
-  m_parser.write(input);
-  input.clear();
+    bool is_reading = true;
+    while (message.length() < expected_length && is_reading) {
+      disconnect(m_socket, SIGNAL(readyRead()), this, SLOT(read()));
+      is_reading = m_socket->waitForReadyRead(2000);
+      if (is_reading) {
+        message.append(m_socket->readAll());
+      }
+      connect(m_socket, SIGNAL(readyRead()), this, SLOT(read()));
+    }
+    m_parser.writeMessage(message);
+  }
 
   if(m_parser.isError()) {
     qDebug() << Q_FUNC_INFO << "parsing error";

src/webui/httprequestparser.cpp

@@ -65,16 +65,9 @@ const QByteArray& HttpRequestParser::torrent() const {
   return m_torrentContent;
 }
 
-void HttpRequestParser::write(const QByteArray& ba) {
-  int end_of_header = ba.indexOf("\r\n\r\n");
-  if (end_of_header < 0) {
-    qWarning() << "Could not find HTTP header: " << ba.constData();
-    m_error = true;
-    return;
-  }
-
+void HttpRequestParser::writeHeader(const QByteArray& ba) {
   // Parse header
-  m_header = QHttpRequestHeader(ba.left(end_of_header));
+  m_header = QHttpRequestHeader(ba);
   QUrl url = QUrl::fromEncoded(m_header.path().toAscii());
   m_path = url.path();
 
@@ -84,12 +77,13 @@ void HttpRequestParser::write(const QByteArray& ba) {
     QPair<QString, QString> pair = i.next();
     m_getMap[pair.first] = pair.second;
   }
+}
 
+void HttpRequestParser::writeMessage(const QByteArray& ba) {
   // Parse message content
-  if (m_header.hasContentLength()) {
-    qDebug() << Q_FUNC_INFO << "ba.size(): " << ba.size();
-    qDebug() << Q_FUNC_INFO << "\n\n" << "header: " <<  ba.left(end_of_header) << "\n\n";
-    m_data = ba.mid(end_of_header + 4, m_header.contentLength()); // +4 to skip "\r\n\r\n"
+  Q_ASSERT (m_header.hasContentLength());
+
+  m_data = ba;
   qDebug() << "m_data.size(): " << m_data.size();
 
   // Parse POST data
@@ -140,4 +134,3 @@ Submit Query
     }
   }
 }
-}

src/webui/httprequestparser.h

@@ -45,7 +45,8 @@ public:
   QString get(const QString& key) const;
   QString post(const QString& key) const;
   const QByteArray& torrent() const;
-  void write(const QByteArray& ba);
+  void writeHeader(const QByteArray& ba);
+  void writeMessage(const QByteArray& ba);
   inline QHttpRequestHeader& header() { return m_header; }
 
 private: