1 /* <!-- copyright */ 2 /* 3 * aria2 - The high speed download utility 4 * 5 * Copyright (C) 2012 Tatsuhiro Tsujikawa 6 * 7 * This program is free software; you can redistribute it and/or modify 8 * it under the terms of the GNU General Public License as published by 9 * the Free Software Foundation; either version 2 of the License, or 10 * (at your option) any later version. 11 * 12 * This program is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15 * GNU General Public License for more details. 16 * 17 * You should have received a copy of the GNU General Public License 18 * along with this program; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 * 21 * In addition, as a special exception, the copyright holders give 22 * permission to link the code of portions of this program with the 23 * OpenSSL library under certain conditions as described in each 24 * individual source file, and distribute linked combinations 25 * including the two. 26 * You must obey the GNU General Public License in all respects 27 * for all of the code used other than OpenSSL. If you modify 28 * file(s) with this exception, you may extend this exception to your 29 * version of the file(s), but you are not obligated to do so. If you 30 * do not wish to do so, delete this exception statement from your 31 * version. If you delete this exception statement from all source 32 * files in the program, then also delete it here. 33 */ 34 /* copyright --> */ 35 #ifndef D_HTTP_HEADER_PROCESSOR_H 36 #define D_HTTP_HEADER_PROCESSOR_H 37 38 #include "common.h" 39 40 #include <utility> 41 #include <string> 42 #include <memory> 43 44 namespace aria2 { 45 46 class HttpHeader; 47 48 class HttpHeaderProcessor { 49 public: 50 enum ParserMode { CLIENT_PARSER, SERVER_PARSER }; 51 52 HttpHeaderProcessor(ParserMode mode); 53 54 ~HttpHeaderProcessor(); 55 /** 56 * Parses incoming data. Returns true if end of header is reached. 57 * This function stops processing data when end of header is 58 * reached. 59 */ 60 bool parse(const unsigned char* data, size_t length); 61 bool parse(const std::string& data); 62 63 /** 64 * Retruns the number of bytes processed in the last invocation of 65 * parse(). 66 */ 67 size_t getLastBytesProcessed() const; 68 69 /** 70 * Processes the received header as a http response header and 71 * returns HttpHeader object. This method transfers the ownership of 72 * resulting HttpHeader to the caller. 73 */ 74 std::unique_ptr<HttpHeader> getResult(); 75 76 std::string getHeaderString() const; 77 78 /** 79 * Resets internal status and ready for next header processing. 80 */ 81 void clear(); 82 83 private: 84 ParserMode mode_; 85 int state_; 86 size_t lastBytesProcessed_; 87 std::string buf_; 88 std::string lastFieldName_; 89 int lastFieldHdKey_; 90 std::unique_ptr<HttpHeader> result_; 91 std::string headers_; 92 }; 93 94 } // namespace aria2 95 96 #endif // D_HTTP_HEADER_PROCESSOR_H 97