Skip to content
http_codec.hh 2.51 KiB
Newer Older
YannGarcia's avatar
YannGarcia committed
#pragma once

#include <memory>

#include "codec_gen.hh"
#include "params.hh"

class Base_Type;
class Record_Type;
class TTCN_Typedescriptor_t;
class TTCN_Buffer;

namespace LibHttp__TypesAndValues {
  class HttpMessage;
  class Request;
  class Response;
  class Headers;
  class Header;
} // namespace LibHttp__TypesAndValues
namespace LibHttp__MessageBodyTypes {
  class HttpMessageBody;
}

struct encoding_context {
  unsigned int  length;
  unsigned char is_content_length_present;

  encoding_context() { reset(); };
  void reset() {
    length                    = -1;
    is_content_length_present = 0x00;
  };
};

struct decoding_context {
  unsigned int  length;
  unsigned char is_binary;
  bool          chunked;

  decoding_context() { reset(); };
  void reset() {
    length    = -1;
    is_binary = 0x00;
    chunked   = false;
  };
};

class http_codec : public codec_gen<LibHttp__TypesAndValues::HttpMessage, LibHttp__TypesAndValues::HttpMessage> {
  encoding_context                                                        _ec;
  decoding_context                                                        _dc;
  std::map<std::string, std::unique_ptr<codec_gen<Record_Type, Record_Type>>> _codecs;

public:
  explicit http_codec() : codec_gen<LibHttp__TypesAndValues::HttpMessage, LibHttp__TypesAndValues::HttpMessage>(), _ec(), _dc(), _codecs(){};
  virtual ~http_codec(){};

  virtual int encode(const LibHttp__TypesAndValues::HttpMessage &, OCTETSTRING &data);
  virtual int decode(const OCTETSTRING &data, LibHttp__TypesAndValues::HttpMessage &, params *params = NULL);

  void set_payload_codecs(const std::string &p_codecs);

private:
  int encode_request(const LibHttp__TypesAndValues::Request &p_request, TTCN_Buffer &p_encoding_buffer);
  int encode_response(const LibHttp__TypesAndValues::Response &p_response, TTCN_Buffer &p_encoding_buffer);
  int encode_body(const LibHttp__MessageBodyTypes::HttpMessageBody &p_message_body, OCTETSTRING &p_encoding_buffer, const std::string &p_content_type);

  int decode_headers(TTCN_Buffer &decoding_buffer, LibHttp__TypesAndValues::Headers &headers, std::string &p_content_type);
  int decode_header(CHARSTRING &header_line, LibHttp__TypesAndValues::Header &header);
  int decode_body(TTCN_Buffer &decoding_buffer, LibHttp__MessageBodyTypes::HttpMessageBody &message_body, const std::string &p_content_type);
  int get_line(TTCN_Buffer &buffer, CHARSTRING &to, const bool concatenate_header_lines = false);

}; // End of class http_codec