nstool/lib/libnx/include/nx/NsoHeader.h

#pragma once
#include <nx/nso.h>
#include <fnd/MemoryBlob.h>
#include <fnd/List.h>
#include <fnd/ISerialiseableBinary.h>

namespace nx
{
	class NsoHeader : 
		public fnd::ISerialiseableBinary
	{
	public:
		struct sModuleId
		{
			byte_t data[nso::kModuleIdSize];

			void operator=(const sModuleId& other)
			{
				memcpy(data, other.data, nso::kModuleIdSize);
			}

			bool operator==(const sModuleId& other) const
			{
				return memcmp(data, other.data, nso::kModuleIdSize) == 0;
			}

			bool operator!=(const sModuleId& other) const
			{
				return !(*this == other);
			}
		};

		struct sLayout
		{
			uint32_t offset;
			uint32_t size;

			void operator=(const sLayout& other)
			{
				offset = other.offset;
				size = other.size;
			}

			bool operator==(const sLayout& other) const
			{
				return (offset == other.offset) \
					&& (size == other.size);
			}

			bool operator!=(const sLayout& other) const
			{
				return !(*this == other);
			}
		};

		struct sCodeSegment
		{
			sLayout file_layout;
			sLayout memory_layout;
			bool is_compressed;
			bool is_hashed;
			crypto::sha::sSha256Hash hash;

			void operator=(const sCodeSegment& other)
			{
				file_layout = other.file_layout;
				memory_layout = other.memory_layout;
				is_compressed = other.is_compressed;
				is_hashed = other.is_hashed;
				hash = other.hash;
			}

			bool operator==(const sCodeSegment& other) const
			{
				return (file_layout == other.file_layout) \
					&& (memory_layout == other.memory_layout) \
					&& (is_compressed == other.is_compressed) \
					&& (is_hashed == other.is_hashed) \
					&& (hash == other.hash);
			}

			bool operator!=(const sCodeSegment& other) const
			{
				return !(*this == other);
			}
		};

		NsoHeader();
		NsoHeader(const NsoHeader& other);
		NsoHeader(const byte_t* bytes, size_t len);

		bool operator==(const NsoHeader& other) const;
		bool operator!=(const NsoHeader& other) const;
		void operator=(const NsoHeader& other);

		// to be used after export
		const byte_t* getBytes() const;
		size_t getSize() const;

		// export/import binary
		void exportBinary();
		void importBinary(const byte_t* bytes, size_t len);

		// variables
		void clear();
		
		const sModuleId& getModuleId() const;
		void setModuleId(const sModuleId& id);

		uint32_t getBssSize() const;
		void setBssSize(uint32_t size);

		const sCodeSegment& getTextSegmentInfo() const;
		void setTextSegmentInfo(const sCodeSegment& info);

		const sCodeSegment& getRoSegmentInfo() const;
		void setRoSegmentInfo(const sCodeSegment& info);

		const sCodeSegment& getDataSegmentInfo() const;
		void setDataSegmentInfo(const sCodeSegment& info);

		const sLayout& getModuleNameInfo() const;
		void setModuleNameInfo(const sLayout& info);

		const sLayout& getRoEmbeddedInfo() const;
		void setRoEmbeddedInfo(const sLayout& info);

		const sLayout& getRoDynStrInfo() const;
		void setRoDynStrInfo(const sLayout& info);

		const sLayout& getRoDynSymInfo() const;
		void setRoDynSymInfo(const sLayout& info);
	private:
		const std::string kModuleName = "NSO_HEADER";

		// binary
		fnd::MemoryBlob mBinaryBlob;

		// data
		sModuleId mModuleId;
		uint32_t mBssSize;
		sCodeSegment mTextSegmentInfo;
		sCodeSegment mRoSegmentInfo;
		sCodeSegment mDataSegmentInfo;
		sLayout mModuleNameInfo;
		sLayout mRoEmbeddedInfo;
		sLayout mRoDynStrInfo;
		sLayout mRoDynSymInfo;
		
		// helpers
		bool isEqual(const NsoHeader& other) const;
		void copyFrom(const NsoHeader& other);
	};

}