mahmoudimus

#!/usr/bin/env python3

import os
import platform
import subprocess
import time
from concurrent.futures import ProcessPoolExecutor, as_completed

import click
from elftools.elf.elffile import ELFFile

mahmoudimus

Quick Guide For Fixing/Installing Python, PyTorch, CUDA, Triton, Sage Attention and Flash Attention For Local AI Image Generation

With all the new local image & video generation tools coming out I've been seeing a lot of posts and error threads being opened for various issues with cuda/pytorch/sage attantion/triton/flash attention. I was tired of digging links up so I initially made this as a cheat sheet for myself but expanded it with hopes that this will help some of you get your venvs and systems running smoothly.

In This Guide:

1. Check Installed Python Versions
2. Set Default Python Version by Changing PATH
3. Installing VS Build Tools
4. Check the Currently Active CUDA Version

mahmoudimus

# (c) Tim Blazytko 2021
# implementation based on the blog post "Automated Detection of Control-flow Flattening"
# https://synthesis.to/2021/03/03/flattening_detection.html
import sys

from miasm.analysis.binary import Container
from miasm.analysis.machine import Machine
from miasm.core.locationdb import LocationDB

mahmoudimus

IDA Microcode notes

References

• Ilfak's presentation at Recon 2018
• Microcode in pictures
• Hex-Rays Microcode API vs. Obfuscating Compiler
• Scripts vds10, vds11, vds12 and vds13 from Hex-Rays SDK

Implementation

mahmoudimus

Usage

In IDAPython,

execfile('<path>/cxxparser.py')
parse_file('<path>/a.cpp',[r'-I<path>\LuaJIT-2.0.5\src', '-D__NT__', '-D__X64__', '-D__EA64__'])
parse_file('<path>/malloc.c',['-target=x86_64-linux-gnu'])

mahmoudimus

import idc
import idautils
import idaapi
import ida_bytes
import ida_funcs

def is_function(ea):
    f = idaapi.get_func(ea)
    if not f:
        return False

mahmoudimus

# /etc/nsmb.conf - macOS 11.3 - 2021-04-29
#------------------------------------------------------------------------------
# SMB configuration for macOS 11.3 <-> Synology
#------------------------------------------------------------------------------
# Additional information:
# -----------------------
# https://support.apple.com/de-de/HT211927
# https://support.apple.com/en-us/HT208209
# https://apple.stackexchange.com/questions/309016/smb-share-deadlocks-since-high-sierra
# https://photographylife.com/afp-vs-nfs-vs-smb-performance

mahmoudimus

"""
On remote machine:

## windows (powershell)
$ python.exe fix-cursor-remote-ssh-connection-failure.py `
        --commit b1e87884330fc271d5eb589e368c35f14e76dec0 `
        --username ${YOUR_USERNAME_HERE} `
        --file-hash-override 84b9c6d907219bb8c2874f299540eb6a079187a0
        
## (linux/macos)

mahmoudimus

# from https://reverseengineering.stackexchange.com/a/11835/13408
# ported to python3 and IDA 8.0+ by Mahmoud Abdelkader


import idaapi
import ida_bytes
import ida_kernwin

def apply_dif_file(dif_file_name):
    print("Applying " + dif_file_name + " to database.")

mahmoudimus

"""
summary: show tabular data with icons
"""

import ida_kernwin
from ida_kernwin import Choose


# -----------------------------------------------------------------------
class chooser_handler_t(ida_kernwin.action_handler_t):