classes.py

import enum


class DataSource(enum.Enum):
    BlockInputOutputEvents = {
        "read_ops",
        "write_ops",
        "read_kbs",
        "write_kbs",
        "avgrq_sz",
        "avg_queue",
        "await",
        "r_await",
        "w_await",
        "svctm",
        "util",
    }
    FileSystem = {
        "seconds",
        "connectivity",
        "block:block_bio_backmerge",
        "block:block_bio_bounce",
        "block:block_bio_complete",
        "block:block_bio_frontmerge",
        "block:block_bio_queue",
        "block:block_bio_remap",
        "block:block_dirty_buffer",
        "block:block_getrq",
        "block:block_plug",
        "block:block_rq_complete",
        "block:block_rq_insert",
        "block:block_rq_issue",
        "block:block_rq_remap",
        "block:block_rq_requeue",
        "block:block_sleeprq",
        "block:block_split",
        "block:block_touch_buffer",
        "block:block_unplug",
        "ext4:ext4_alloc_da_blocks",
        "ext4:ext4_allocate_blocks",
        "ext4:ext4_allocate_inode",
        "ext4:ext4_begin_ordered_truncate",
        "ext4:ext4_collapse_range",
        "ext4:ext4_da_release_space",
        "ext4:ext4_da_reserve_space",
        "ext4:ext4_da_update_reserve_space",
        "ext4:ext4_da_write_begin",
        "ext4:ext4_da_write_end",
        "ext4:ext4_da_write_pages",
        "ext4:ext4_da_write_pages_extent",
        "ext4:ext4_direct_IO_enter",
        "ext4:ext4_direct_IO_exit",
        "ext4:ext4_discard_blocks",
        "ext4:ext4_discard_preallocations",
        "ext4:ext4_drop_inode",
        "ext4:ext4_error",
        "ext4:ext4_es_cache_extent",
        "ext4:ext4_es_find_extent_range_enter",
        "ext4:ext4_es_find_extent_range_exit",
        "ext4:ext4_es_insert_delayed_block",
        "ext4:ext4_es_insert_extent",
        "ext4:ext4_es_lookup_extent_enter",
        "ext4:ext4_es_lookup_extent_exit",
        "ext4:ext4_es_remove_extent",
        "ext4:ext4_es_shrink",
        "ext4:ext4_es_shrink_count",
        "ext4:ext4_es_shrink_scan_enter",
        "ext4:ext4_es_shrink_scan_exit",
        "ext4:ext4_evict_inode",
        "ext4:ext4_ext_convert_to_initialized_enter",
        "ext4:ext4_ext_convert_to_initialized_fastpath",
        "ext4:ext4_ext_handle_unwritten_extents",
        "ext4:ext4_ext_in_cache",
        "ext4:ext4_ext_load_extent",
        "ext4:ext4_ext_map_blocks_enter",
        "ext4:ext4_ext_map_blocks_exit",
        "ext4:ext4_ext_put_in_cache",
        "ext4:ext4_ext_remove_space",
        "ext4:ext4_ext_remove_space_done",
        "ext4:ext4_ext_rm_idx",
        "ext4:ext4_ext_rm_leaf",
        "ext4:ext4_ext_show_extent",
        "ext4:ext4_fallocate_enter",
        "ext4:ext4_fallocate_exit",
        "ext4:ext4_find_delalloc_range",
        "ext4:ext4_forget",
        "ext4:ext4_free_blocks",
        "ext4:ext4_free_inode",
        "ext4:ext4_fsmap_high_key",
        "ext4:ext4_fsmap_low_key",
        "ext4:ext4_fsmap_mapping",
        "ext4:ext4_get_implied_cluster_alloc_exit",
        "ext4:ext4_get_reserved_cluster_alloc",
        "ext4:ext4_getfsmap_high_key",
        "ext4:ext4_getfsmap_low_key",
        "ext4:ext4_getfsmap_mapping",
        "ext4:ext4_ind_map_blocks_enter",
        "ext4:ext4_ind_map_blocks_exit",
        "ext4:ext4_insert_range",
        "ext4:ext4_invalidatepage",
        "ext4:ext4_journal_start",
        "ext4:ext4_journal_start_reserved",
        "ext4:ext4_journalled_invalidatepage",
        "ext4:ext4_journalled_write_end",
        "ext4:ext4_load_inode",
        "ext4:ext4_load_inode_bitmap",
        "ext4:ext4_mark_inode_dirty",
        "ext4:ext4_mb_bitmap_load",
        "ext4:ext4_mb_buddy_bitmap_load",
        "ext4:ext4_mb_discard_preallocations",
        "ext4:ext4_mb_new_group_pa",
        "ext4:ext4_mb_new_inode_pa",
        "ext4:ext4_mb_release_group_pa",
        "ext4:ext4_mb_release_inode_pa",
        "ext4:ext4_mballoc_alloc",
        "ext4:ext4_mballoc_discard",
        "ext4:ext4_mballoc_free",
        "ext4:ext4_mballoc_prealloc",
        "ext4:ext4_nfs_commit_metadata",
        "ext4:ext4_other_inode_update_time",
        "ext4:ext4_punch_hole",
        "ext4:ext4_read_block_bitmap_load",
        "ext4:ext4_readpage",
        "ext4:ext4_releasepage",
        "ext4:ext4_remove_blocks",
        "ext4:ext4_request_blocks",
        "ext4:ext4_request_inode",
        "ext4:ext4_shutdown",
        "ext4:ext4_sync_file_enter",
        "ext4:ext4_sync_file_exit",
        "ext4:ext4_sync_fs",
        "ext4:ext4_trim_all_free",
        "ext4:ext4_trim_extent",
        "ext4:ext4_truncate_enter",
        "ext4:ext4_truncate_exit",
        "ext4:ext4_unlink_enter",
        "ext4:ext4_unlink_exit",
        "ext4:ext4_write_begin",
        "ext4:ext4_write_end",
        "ext4:ext4_writepage",
        "ext4:ext4_writepages",
        "ext4:ext4_writepages_result",
        "ext4:ext4_zero_range",
        "filemap:file_check_and_advance_wb_err",
        "filemap:filemap_set_wb_err",
        "filemap:mm_filemap_add_to_page_cache",
        "filemap:mm_filemap_delete_from_page_cache",
        "jbd2:jbd2_checkpoint",
        "jbd2:jbd2_checkpoint_stats",
        "jbd2:jbd2_commit_flushing",
        "jbd2:jbd2_commit_locking",
        "jbd2:jbd2_commit_logging",
        "jbd2:jbd2_drop_transaction",
        "jbd2:jbd2_end_commit",
        "jbd2:jbd2_handle_extend",
        "jbd2:jbd2_handle_start",
        "jbd2:jbd2_handle_stats",
        "jbd2:jbd2_lock_buffer_stall",
        "jbd2:jbd2_run_stats",
        "jbd2:jbd2_start_commit",
        "jbd2:jbd2_submit_inode_data",
        "jbd2:jbd2_update_log_tail",
        "jbd2:jbd2_write_superblock",
        "writeback:balance_dirty_pages",
        "writeback:bdi_dirty_ratelimit",
        "writeback:flush_foreign",
        "writeback:global_dirty_state",
        "writeback:inode_foreign_history",
        "writeback:inode_switch_wbs",
        "writeback:sb_clear_inode_writeback",
        "writeback:sb_mark_inode_writeback",
        "writeback:track_foreign_dirty",
        "writeback:wait_on_page_writeback",
        "writeback:wbc_writepage",
        "writeback:writeback_bdi_register",
        "writeback:writeback_congestion_wait",
        "writeback:writeback_dirty_inode",
        "writeback:writeback_dirty_inode_enqueue",
        "writeback:writeback_dirty_inode_start",
        "writeback:writeback_dirty_page",
        "writeback:writeback_exec",
        "writeback:writeback_lazytime",
        "writeback:writeback_lazytime_iput",
        "writeback:writeback_mark_inode_dirty",
        "writeback:writeback_pages_written",
        "writeback:writeback_queue",
        "writeback:writeback_queue_io",
        "writeback:writeback_sb_inodes_requeue",
        "writeback:writeback_single_inode",
        "writeback:writeback_single_inode_start",
        "writeback:writeback_start",
        "writeback:writeback_wait",
        "writeback:writeback_wait_iff_congested",
        "writeback:writeback_wake_background",
        "writeback:writeback_write_inode",
        "writeback:writeback_write_inode_start",
        "writeback:writeback_written\n",
    }
    KernelEvents = {
        "seconds",
        "connectivity",
        "alarmtimer:alarmtimer_fired",
        "alarmtimer:alarmtimer_start",
        "block:block_bio_backmerge",
        "block:block_bio_remap",
        "block:block_dirty_buffer",
        "block:block_getrq",
        "block:block_touch_buffer",
        "block:block_unplug",
        "cachefiles:cachefiles_create",
        "cachefiles:cachefiles_lookup",
        "cachefiles:cachefiles_mark_active",
        "clk:clk_set_rate",
        "cpu-migrations",
        "cs",
        "dma_fence:dma_fence_init",
        "fib:fib_table_lookup",
        "filemap:mm_filemap_add_to_page_cache",
        "gpio:gpio_value",
        "ipi:ipi_raise",
        "irq:irq_handler_entry",
        "irq:softirq_entry",
        "jbd2:jbd2_handle_start",
        "jbd2:jbd2_start_commit",
        "kmem:kfree",
        "kmem:kmalloc",
        "kmem:kmem_cache_alloc",
        "kmem:kmem_cache_free",
        "kmem:mm_page_alloc",
        "kmem:mm_page_alloc_zone_locked",
        "kmem:mm_page_free",
        "kmem:mm_page_pcpu_drain",
        "mmc:mmc_request_start",
        "net:net_dev_queue",
        "net:net_dev_xmit",
        "net:netif_rx",
        "page-faults",
        "pagemap:mm_lru_insertion",
        "preemptirq:irq_enable",
        "qdisc:qdisc_dequeue",
        "qdisc:qdisc_dequeue.1",
        "random:get_random_bytes",
        "random:mix_pool_bytes_nolock",
        "random:urandom_read",
        "raw_syscalls:sys_enter",
        "raw_syscalls:sys_exit",
        "rpm:rpm_resume",
        "rpm:rpm_suspend",
        "sched:sched_process_exec",
        "sched:sched_process_free",
        "sched:sched_process_wait",
        "sched:sched_switch",
        "sched:sched_wakeup",
        "signal:signal_deliver",
        "signal:signal_generate",
        "skb:consume_skb",
        "skb:consume_skb.1",
        "skb:kfree_skb",
        "skb:kfree_skb.1",
        "skb:skb_copy_datagram_iovec",
        "sock:inet_sock_set_state",
        "task:task_newtask",
        "tcp:tcp_destroy_sock",
        "tcp:tcp_probe",
        "timer:hrtimer_start",
        "timer:timer_start",
        "udp:udp_fail_queue_rcv_skb",
        "workqueue:workqueue_activate_work",
        "writeback:global_dirty_state",
        "writeback:sb_clear_inode_writeback",
        "writeback:wbc_writepage",
        "writeback:writeback_dirty_inode",
        "writeback:writeback_dirty_inode_enqueue",
        "writeback:writeback_dirty_page",
        "writeback:writeback_mark_inode_dirty",
        "writeback:writeback_pages_written",
        "writeback:writeback_single_inode",
        "writeback:writeback_write_inode",
        "writeback:writeback_written\n",
    }
    NetRecords = {"Protocol", "SourceIP", "SourcePort", "DestIP", "DestPort", "Length"}
    ResourceUsageEvents = {
        "cpu",
        "seconds",
        "ioread",
        "iowrite",
        "ioreadbytes",
        "iowritebytes",
        "ioreadtime",
        "iowritetime",
        "iobusytime",
        "read_merge",
        "write_merge",
        "memory",
        "net_in",
        "net_out",
        "pkt_in",
        "pkt_out",
        "err_in",
        "err_out",
        "drop_in",
        "drop_out",
        "timer:tick_stop",
        "sched:sched_process_exec",
        "sched:sched_waking",
        "task:task_newtask",
        "sched:sched_stat_runtime",
        "timer:timer_cancel",
        "timer:timer_init",
        "timer:timer_start",
        "workqueue:workqueue_execute_start",
        "branch-instructions",
        "branch-misses",
        "bus-cycles",
        "cache-misses",
        "cache-references",
        "cpu-cycles",
        "instructions",
        "context-switches",
        "cpu-migrations",
        "minor-faults",
        "page-faults",
        "L1-dcache-load-misses",
        "L1-dcache-loads",
        "L1-dcache-store-misses",
        "L1-dcache-stores",
        "L1-icache-load-misses",
        "L1-icache-loads",
        "LLC-load-misses",
        "LLC-loads",
        "LLC-store-misses",
        "LLC-stores",
        "branch-load-misses",
        "branch-loads",
        "dTLB-load-misses",
        "dTLB-store-misses",
        "iTLB-load-misses",
        "armv7_cortex_a7/br_immed_retired/",
        "armv7_cortex_a7/br_mis_pred/",
        "armv7_cortex_a7/br_pred/",
        "armv7_cortex_a7/bus_cycles/",
        "armv7_cortex_a7/cpu_cycles/",
        "armv7_cortex_a7/exc_return/",
        "armv7_cortex_a7/exc_taken/",
        "armv7_cortex_a7/inst_retired/",
        "armv7_cortex_a7/l1d_cache/",
        "armv7_cortex_a7/l1d_cache_refill/",
        "armv7_cortex_a7/l1d_cache_wb/",
        "armv7_cortex_a7/l1d_tlb_refill/",
        "armv7_cortex_a7/l1i_cache/",
        "armv7_cortex_a7/l1i_cache_refill/",
        "armv7_cortex_a7/l1i_tlb_refill/",
        "armv7_cortex_a7/l2d_cache/",
        "armv7_cortex_a7/l2d_cache_wb/",
        "armv7_cortex_a7/ld_retired/",
        "armv7_cortex_a7/mem_access/",
        "armv7_cortex_a7/pc_write_retired/",
        "armv7_cortex_a7/st_retired/",
        "armv7_cortex_a7/unaligned_ldst_retired/",
        "armv7_cortex_a7/cid_write_retired/",
        "armv7_cortex_a7/bus_cycles/.1",
        "block:block_bio_frontmerge",
        "block:block_dirty_buffer",
        "block:block_split",
        "block:block_touch_buffer",
        "ext4:ext4_es_lookup_extent_enter",
        "ext4:ext4_ext_load_extent",
        "ext4:ext4_writepages_result",
        "ext4:ext4_journal_start",
        "filemap:mm_filemap_add_to_page_cache",
        "jbd2:jbd2_handle_stats",
        "ext4:ext4_da_update_reserve_space",
        "ext4:ext4_sync_file_enter",
        "jbd2:jbd2_checkpoint_stats",
        "ext4:ext4_free_inode",
        "ext4:ext4_evict_inode",
        "ext4:ext4_releasepage",
        "ext4:ext4_unlink_enter",
        "block:block_bio_remap",
        "filemap:mm_filemap_delete_from_page_cache",
        "kmem:kfree",
        "kmem:kmem_cache_alloc",
        "kmem:mm_page_alloc_zone_locked",
        "kmem:mm_page_free",
        "mmc:mmc_request_done",
        "writeback:global_dirty_state",
        "writeback:sb_clear_inode_writeback",
        "writeback:wait_on_page_writeback",
        "napi:napi_poll",
        "tcp:tcp_probe",
        "net:netif_rx",
        "gpio:gpio_value",
        "irq:softirq_exit",
        "pagemap:mm_lru_activate",
        "rpm:rpm_return_int",
        "fib:fib_table_lookup",
        "raw_syscalls:sys_enter",
        "random:credit_entropy_bits",
    }
    ResourceUsageEvents_RP4 = {
        "cpu",
        "seconds",
        "ioread",
        "iowrite",
        "ioreadbytes",
        "iowritebytes",
        "ioreadtime",
        "iowritetime",
        "iobusytime",
        "read_merge",
        "write_merge",
        "memory",
        "net_in",
        "net_out",
        "pkt_in",
        "pkt_out",
        "err_in",
        "err_out",
        "drop_in",
        "drop_out",
        "timer:tick_stop",
        "sched:sched_process_exec",
        "sched:sched_waking",
        "task:task_newtask",
        "sched:sched_stat_runtime",
        "timer:timer_cancel",
        "timer:timer_init",
        "timer:timer_start",
        "workqueue:workqueue_execute_start",
        "branch-instructions",
        "branch-misses",
        "bus-cycles",
        "cache-misses",
        "cache-references",
        "cpu-cycles",
        "instructions",
        "context-switches",
        "cpu-migrations",
        "minor-faults",
        "page-faults",
        "L1-dcache-load-misses",
        "L1-dcache-loads",
        "L1-dcache-store-misses",
        "L1-dcache-stores",
        "L1-icache-load-misses",
        "L1-icache-loads",
        "LLC-load-misses",
        "LLC-loads",
        "LLC-store-misses",
        "LLC-stores",
        "branch-load-misses",
        "branch-loads",
        "dTLB-load-misses",
        "dTLB-store-misses",
        "iTLB-load-misses",
        "armv7_cortex_a15/br_immed_retired/",
        "armv7_cortex_a15/br_mis_pred/",
        "armv7_cortex_a15/br_pred/",
        "armv7_cortex_a15/bus_cycles/",
        "armv7_cortex_a15/cpu_cycles/",
        "armv7_cortex_a15/exc_return/",
        "armv7_cortex_a15/exc_taken/",
        "armv7_cortex_a15/inst_retired/",
        "armv7_cortex_a15/l1d_cache/",
        "armv7_cortex_a15/l1d_cache_refill/",
        "armv7_cortex_a15/l1d_cache_wb/",
        "armv7_cortex_a15/l1d_tlb_refill/",
        "armv7_cortex_a15/l1i_cache/",
        "armv7_cortex_a15/l1i_cache_refill/",
        "armv7_cortex_a15/l1i_tlb_refill/",
        "armv7_cortex_a15/l2d_cache/",
        "armv7_cortex_a15/l2d_cache_wb/",
        "armv7_cortex_a15/ld_retired/",
        "armv7_cortex_a15/mem_access/",
        "armv7_cortex_a15/pc_write_retired/",
        "armv7_cortex_a15/st_retired/",
        "armv7_cortex_a15/unaligned_ldst_retired/",
        "armv7_cortex_a15/cid_write_retired/",
        "armv7_cortex_a15/bus_cycles/.1",
        "block:block_bio_frontmerge",
        "block:block_dirty_buffer",
        "block:block_split",
        "block:block_touch_buffer",
        "ext4:ext4_es_lookup_extent_enter",
        "ext4:ext4_ext_load_extent",
        "ext4:ext4_writepages_result",
        "ext4:ext4_journal_start",
        "filemap:mm_filemap_add_to_page_cache",
        "jbd2:jbd2_handle_stats",
        "ext4:ext4_da_update_reserve_space",
        "ext4:ext4_sync_file_enter",
        "jbd2:jbd2_checkpoint_stats",
        "ext4:ext4_free_inode",
        "ext4:ext4_evict_inode",
        "ext4:ext4_releasepage",
        "ext4:ext4_unlink_enter",
        "block:block_bio_remap",
        "filemap:mm_filemap_delete_from_page_cache",
        "kmem:kfree",
        "kmem:kmem_cache_alloc",
        "kmem:mm_page_alloc_zone_locked",
        "kmem:mm_page_free",
        "mmc:mmc_request_done",
        "writeback:global_dirty_state",
        "writeback:sb_clear_inode_writeback",
        "writeback:wait_on_page_writeback",
        "napi:napi_poll",
        "tcp:tcp_probe",
        "net:netif_rx",
        "gpio:gpio_value",
        "irq:softirq_exit",
        "pagemap:mm_lru_activate",
        "rpm:rpm_return_int",
        "fib:fib_table_lookup",
        "raw_syscalls:sys_enter",
        "random:credit_entropy_bits",
    }
    SystemCalls = {}
    Entropy = {"file_path", "action", "entropy"}

    @classmethod
    def from_columns(cls, columns):
        cols_set = set(columns)

        if "index" in cols_set:
            cols_set.remove("index")

        for data_type in cls:
            if cols_set == data_type.value:
                return data_type
        raise ValueError(f"Unknown data type for columns: {columns}")

    @classmethod
    def from_single_column(cls, column: str):
        for data_type in cls:
            if column in data_type.value:
                return data_type
        raise ValueError(f"Unknown data type for column: {column}")

    @classmethod
    def from_string_name(cls, data_source: str):
        match data_source:
            case "SYS_data":
                return cls.SystemCalls
            case "KERN_data":
                return cls.KernelEvents
            case "FLS_data":
                return cls.FileSystem
            case "RES_data":
                return cls.ResourceUsageEvents
            case "block_data":
                return cls.BlockInputOutputEvents
            case "network_data":
                return cls.NetRecords
            case "entropy_data":
                return cls.Entropy
        raise ValueError(f"Unknown data type for string name: {data_source}")