annotations.hpp

//--------------------------------------------------------------------*- C++ -*-
// Kunai-static-analyzer: library for doing analysis of dalvik files
// @author Farenain <kunai.static.analysis@gmail.com>
// @author Ernesto Java <javaernesto@gmail.com>
//
// @file annotations.hpp
// @brief Manage the annotations from the DEX file
 
#ifndef KUNAI_DEX_PARSER_ANNOTATIONS_HPP
#define KUNAI_DEX_PARSER_ANNOTATIONS_HPP
 
#include "Kunai/Utils/kunaistream.hpp"
#include "Kunai/DEX/parser/fields.hpp"
#include "Kunai/DEX/parser/methods.hpp"
 
#include <memory>
#include <vector>
#include <unordered_map>
 
namespace KUNAI
{
namespace DEX
{
    class ParameterAnnotation
    {
        std::uint32_t method_idx;
        std::uint32_t annotations_off;
    public:
        ParameterAnnotation(std::uint32_t method_idx, std::uint32_t annotations_off)
            : method_idx(method_idx), annotations_off(annotations_off)
        {}
 
        ~ParameterAnnotation() = default;
 
        std::uint32_t get_method_idx() const
        {
            return method_idx;
        }
 
        std::uint32_t get_annotations_off() const
        {
            return annotations_off;
        }
    };
 
    using parameterannotation_t = std::unique_ptr<ParameterAnnotation>;
 
    class MethodAnnotation
    {
        std::uint32_t method_idx;
        std::uint32_t annotations_off;
    public:
        MethodAnnotation(std::uint32_t method_idx, std::uint32_t annotations_off)
            : method_idx(method_idx), annotations_off(annotations_off)
        {}
        ~MethodAnnotation() = default;
 
        std::uint32_t get_method_idx() const
        {
            return method_idx;
        }
        std::uint32_t get_annotations_off() const
        {
            return annotations_off;
        }
    };
 
    using methodannotation_t = std::unique_ptr<MethodAnnotation>;
 
    class FieldAnnotation
    {
        std::uint32_t field_idx;
        std::uint32_t annotations_off;
    public:
 
        FieldAnnotation(std::uint32_t field_idx, std::uint32_t annotations_off)
            : field_idx(field_idx), annotations_off(annotations_off)
        {}
 
        ~FieldAnnotation() = default;
 
        std::uint32_t get_field_idx() const
        {
            return field_idx;
        }
 
        std::uint32_t get_annotations_off() const
        {
            return annotations_off;
        }
    };
 
    using fieldannotation_t = std::unique_ptr<FieldAnnotation>;
 
    class AnnotationDirectoryItem
    {
        std::uint32_t class_annotations_off;
        std::vector<fieldannotation_t> field_annotations;
        std::unordered_map<std::uint32_t, FieldAnnotation*> field_annotations_by_id;
        std::vector<methodannotation_t> method_annotations;
        std::unordered_map<std::uint32_t, MethodAnnotation*> method_annotations_by_id;
        std::vector<parameterannotation_t> parameter_annotations;
        std::unordered_map<std::uint32_t, ParameterAnnotation*> parameter_annotations_by_id;
    public:
        AnnotationDirectoryItem() = default;
        ~AnnotationDirectoryItem() = default;
        void parse_annotation_directory_item(stream::KunaiStream* stream);
 
        const std::vector<fieldannotation_t>& get_field_annotations() const
        {
            return field_annotations;
        }
 
        std::vector<fieldannotation_t>& get_field_annotations()
        {
            return field_annotations;
        }
 
        FieldAnnotation* get_field_annotation_by_id(std::uint32_t idx);
 
        const std::vector<methodannotation_t>& get_method_annotations() const
        {
            return method_annotations;
        }
 
        std::vector<methodannotation_t>& get_method_annotations()
        {
            return method_annotations;
        }
 
        MethodAnnotation* get_method_annotation_by_id(std::uint32_t idx);
 
        const std::vector<parameterannotation_t>& get_parameter_annotations() const
        {
            return parameter_annotations;
        }
 
        std::vector<parameterannotation_t>& get_parameter_annotations()
        {
            return parameter_annotations;
        }
 
        ParameterAnnotation* get_parameter_annotation_by_id(std::uint32_t idx);
    };
} // namespace DEX
} // namespace KUNAI
 
 
#endif