/* * This file is a part of Winix * and is not publicly distributed * * Copyright (c) 2008-2011, Tomasz Sowa * All rights reserved. * */ #ifndef headerfile_winix_core_postmultiparser #define headerfile_winix_core_postmultiparser #include #include #include #include "error.h" #include "requesttypes.h" #include "config.h" #include "misc.h" // 2 MB #define WINIX_POSTMULTI_INPUT_BUFFER 2097152 #define WINIX_POSTMULTI_OUTPUT_BUFFER 2097152 class PostMultiParser { public: PostMultiParser(); PostMultiParser(const PostMultiParser &); PostMultiParser & operator=(const PostMultiParser &); ~PostMultiParser(); void SetConfig(Config * pconfig); Error Parse(FCGX_Stream * in_, PostTab & post_tab_, PostFileTab & post_file_tab_); private: Config * config; FCGX_Stream * in; unsigned char * in_buffer; std::ofstream tmp_file; std::wstring tmp_filename; std::string atmp_filename; int tmp_filename_postfix; size_t in_buffer_ind; size_t in_buffer_len; PostTab * post_tab; PostFileTab * post_file_tab; int last; // last read character int var_index; // used as a postfix to the same name (is auto increment) bool line_end_dos; std::string boundary; std::string content; size_t content_len; std::string header_name, header_value; Error err; std::string name, filename; std::wstring namew, contentw; PostFile post_file_temp; void LogFirst(const std::string & to_log, size_t len); void ConvStr(const std::string & src, std::wstring & dst); bool IsWhite(int c); void SkipWhite(); void AddNormalPostVar(); void AddFilePostVar(); void AddPostVar(); void ReadBoundary(); bool IsHeader(); void ReadHeaderName(); void ReadHeaderValue(); void ReadPartHeader(); void CreateTmpFile(); bool HasBoundary(); void ReadContentSkipBoundary(bool has_boundary); void ReadContentToFileLoop(); void ReadContentToFile(); void ReadContentLoop(); void ReadContent(); void CheckBoundaryEnd(); void ReadPart(); void ReadChar(); template bool InsertPostVar(Container & container, std::wstring & key, const Value & value); }; template bool PostMultiParser::InsertPostVar(Container & container, std::wstring & key, const Value & value) { bool added; std::pair res; res = container.insert( std::make_pair(key, value) ); added = res.second; if( !added ) { key += L"_inc"; key += Toa(var_index); res = container.insert( std::make_pair(key, value) ); added = res.second; var_index += 1; } return added; } #endif