payload.hpp

Go to the documentation of this file.

#ifndef __FACADE_PAYLOAD_HPP
#define __FACADE_PAYLOAD_HPP
 
 
#include <facade/png.hpp>
#include <facade/ico.hpp>
 
namespace facade
{
   class
   EXPORT
   PNGPayload : public png::Image
   {
   public:
      PNGPayload() : png::Image() {}
      PNGPayload(const void *ptr, std::size_t size) : png::Image(ptr, size) {}
      PNGPayload(const std::vector<std::uint8_t> &vec) : png::Image(vec) {}
      PNGPayload(const std::string &filename) : png::Image(filename) {}
      PNGPayload(const PNGPayload &other) : png::Image(other) {}
 
      png::Text &add_text_payload(const std::string &keyword, const void *ptr, std::size_t size);
 
      png::Text &add_text_payload(const std::string &keyword, const std::vector<std::uint8_t> &data);
 
      void remove_text_payload(const png::Text &payload);
 
      std::vector<png::Text> get_text_payloads(const std::string &keyword) const;
 
      std::vector<std::vector<std::uint8_t>> extract_text_payloads(const std::string &keyword) const;
 
      png::ZText &add_ztext_payload(const std::string &keyword, const void *ptr, std::size_t size);
 
      png::ZText &add_ztext_payload(const std::string &keyword, const std::vector<std::uint8_t> &data);
 
      void remove_ztext_payload(const png::ZText &payload);
 
      std::vector<png::ZText> get_ztext_payloads(const std::string &keyword) const;
 
      std::vector<std::vector<std::uint8_t>> extract_ztext_payloads(const std::string &keyword) const;
 
      std::vector<std::uint8_t> read_stego_data(std::size_t bit_offset, std::size_t size) const;
      void write_stego_data(const void *ptr, std::size_t size, std::size_t bit_offset);
      void write_stego_data(const std::vector<std::uint8_t> &data, std::size_t bit_offset);
 
      bool has_stego_payload() const;
      PNGPayload create_stego_payload(const void *ptr, std::size_t size) const;
      PNGPayload create_stego_payload(const std::vector<std::uint8_t> &data) const;
      std::vector<std::uint8_t> extract_stego_payload() const;
   };
 
   class
   EXPORT
   ICOPayload : public ico::Icon
   {
      std::optional<std::size_t> _index;
      std::optional<PNGPayload> _payload;
      
   public:
      ICOPayload() : ico::Icon() {}
      ICOPayload(const void *ptr, std::size_t size) : ico::Icon(ptr, size) { this->find_png(); }
      ICOPayload(const std::vector<std::uint8_t> &vec) : ico::Icon(vec) { this->find_png(); }
      ICOPayload(const std::string &filename) : ico::Icon(filename) { this->find_png(); }
      ICOPayload(const ICOPayload &other) : _index(other._index), _payload(other._payload), ico::Icon(other) {}
 
      ICOPayload &operator=(const ICOPayload &other);
      PNGPayload *operator->(void);
      PNGPayload &operator*(void);
 
      PNGPayload &png_payload(void);
      const PNGPayload &png_payload(void) const;
 
      void find_png(void);
      void reset_png(void);
      void set_png(void);
   };
}
 
#endif