main.sh

#!/bin/bash

os=$(grep -oP '(?<=^ID=).+' /etc/os-release | tr -d '"')
work_dir=$PWD

Require() {
	case $os in
		debian|ubuntu)
		packages=(
			["hdparm"]="hdparm"
			["sysstat"]="sysstat"
			["tmux"]="tmux"
			["iptables"]="iptables"
			["nasm"]="nasm"
            ["gcc"]="gcc"
            ["make"]="make"
            ["zfsutils-linux"] ="zfsutils-linux"
            ["btrfs-progs"] ="btrfs-progs"
            ["fio"] ="fio"
            ["xfsprogs"] ="xfsprogs"
		)
		for package in "${!packages[@]}"; do
			if ! dpkg -s "${packages[$package]}" > /dev/null 2>&1; then
				sudo apt-get install -y "${packages[$package]}"
			else
				echo "$package уже установлен."
			fi
		done
		;;
		
		arch)
		declare -A packages=(
            ["zfsutils-linux"] ="zfsutils-linux"
			["hdparm"]="hdparm"
            ["gcc"]="gcc"
			["sysstat"]="sysstat"
			["tmux"]="tmux"
			["iptables"]="iptables"
			["nasm"]="nasm"
            ["btrfs-progs"] ="btrfs-progs"
            ["xfsprogs"] ="xfsprogs"
            ["fio"] ="fio"
            ["duperemove"] ="duperemove"
		)
		for package in "${!packages[@]}"; do
			if ! pacman -Qs "${packages[$package]}" > /dev/null 2>&1; then
				sudo pacman -S "${packages[$package]}"
			else
				echo "$package уже установлен."
			fi
		done
		
		if ! pacman -Qs yaourt > /dev/null 2>&1; then
			current_user=$(logname)
			sudo pacman -S --needed base-devel git wget yajl
			cd /tmp
			git clone https://aur.archlinux.org/package-query.git
			cd package-query/
			makepkg -si && cd /tmp/
			git clone https://aur.archlinux.org/yaourt.git
			cd yaourt/
			makepkg -si
		else
			echo "yaourt уже установлен."
		fi
		
		for package in "${!packages[@]}"; do
			if ! yaourt -Q "${packages[$package]}" > /dev/null 2>&1; then
				yaourt -S "${packages[$package]}"
			else
				echo "$package уже установлен."
			fi
		done
		
		sleep 20
		;;
		*)
		echo "Не удалось определить менеджер пакетов для этой операционной системы."
		
		;;
	esac
}

FSchecker() {
    echo "Создание временных файлов и разделов для тестирования..."
    declare -A images=(
        ["btrfs"]="/tmp/btrfs.img"
        ["zfs"]="/tmp/zfs.img"
        ["xfs"]="/tmp/xfs.img"
        ["ext4"]="/tmp/ext4.img"
    )
    for img in "${images[@]}"; do
        dd if=/dev/zero of=$img bs=1G count=2
    done
    echo "Форматирование образов в выбранные ФС..."
    sudo mkfs.btrfs ${images["btrfs"]}
    sudo mkfs.xfs ${images["xfs"]}
    sudo mkfs.ext4 ${images["ext4"]}

    echo "Подключение файловых систем"
    mkdir -p /mnt/{btrfs,zfs,xfs,ext4}
    sudo mount -o loop ${images["btrfs"]} /mnt/btrfs
    sudo mount -o loop ${images["xfs"]} /mnt/xfs
    sudo mount -o loop ${images["ext4"]} /mnt/ext4
    sudo zpool create testpool ${images["zfs"]}
    sudo zfs set mountpoint=/mnt/zfs testpool

    # Ассоциативный массив для хранения результатов
    declare -A results

    echo "Тестирование с помощью fio и запись результатов..."
    logfile="fs_test_results.log"
    echo "FS Test Results" > $logfile
    for fs in "${!images[@]}"; do
        echo "Testing $fs..."
        total_iops=0
        for i in {1..5}; do
            result=$(sudo fio --randrepeat=1 --ioengine=libaio --direct=1 --gtod_reduce=1 --name=test --filename=/mnt/$fs/testfile --bs=4k --iodepth=64 --size=1G --readwrite=randrw --rwmixread=75 | grep -E "read: IOPS=|write: IOPS=")
            read_iops=$(echo $result | grep -oP 'read: IOPS=\K\d+')
            write_iops=$(echo $result | grep -oP 'write: IOPS=\K\d+')
            total_iops=$((total_iops + read_iops + write_iops))
            echo "Iteration $i - Read IOPS: $read_iops, Write IOPS: $write_iops, Total IOPS: $total_iops" >> $logfile
        done
        avg_iops=$((total_iops / 5))
        results[$fs]=$avg_iops
        echo "$fs Average IOPS: $avg_iops" | tee -a $logfile
    done

    echo "Определение лучшей ФС..."
    best_fs=""
    best_score=0
    for fs in "${!results[@]}"; do
        if [[ ${results[$fs]} -gt $best_score ]]; then
            best_score=${results[$fs]}
            best_fs=$fs
        fi
    done

    echo "Best FS based on IOPS: $best_fs" | tee -a $logfile
    sleep 3
    echo "Отключение ФС и очистка"
    sudo umount /mnt/btrfs
    sudo umount /mnt/xfs
    sudo umount /mnt/ext4
    sudo zpool destroy testpool
    for img in "${images[@]}"; do
        rm -rf $img
    done
    cd $workdir
}

AllocTests() {
    cd $work_dir/AllocTests

    echo "Выберите файл для компиляции и запуска:"
    echo "1. Aligned_Alloc.c"
    echo "2. Calloc.c"
    echo "3. Malloc.c"
    read -p "Введите номер выбранного файла: " file_number

    case $file_number in
        1) file_name="Aligned_Alloc";;
        2) file_name="Calloc";;
        3) file_name="Malloc";;
        *) echo "Некорректный ввод"; return;;
    esac

    gcc -o $file_name $file_name.c

    read -p "У вас есть возможность запустить программу. 
        Запуск - 'y', отказ от запуска - 'n': " response
    if [ "$response" == "y" ]; then 
        echo "Запуск..."
        sleep 2
        ./$file_name
    elif [ "$response" == "n" ]; then
        echo "Отменяем запуск..."
    else
        echo "Некорректный ввод"
    fi

    sleep 5
    rm $file_name
    cd $work_dir
	clear
    echo "Был создан memory_allocation_log"
    main
}

Scheduler() {
    sudo modprobe bfq
    sudo modprobe kyber-iosched
    cd $work_dir/Sched
    chmod +x sched.sh
    sudo ./sched.sh
}

LinPack() {
        git clone https://github.com/ereyes01/linpack.git
        cd linpack && make
        clear
        
        restore_default() {
            local param_name="$1"
            local default_value="$2"
            echo "$default_value" > "/proc/sys/kernel/sched_$param_name"
        }

        cd /proc/sys/kernel/
        for file in sched_*; do
            param_name="${file#sched_}"
            current_value=$(cat "$file")
            echo "sched_$param_name: $current_value"
        done

        param_list=$(ls sched_* | sed 's/sched_//')
        echo "Введите имя параметра для изменения (или 'q' для выхода) без 'sched_':"
        read -p "" chosen_param

        while [[ "$chosen_param" != "q" ]]; do
            if [[ "$param_list" =~ (^|[[:space:]])"$chosen_param"($|[[:space:]]) ]]; then
                current_value=$(cat "sched_$chosen_param")
                echo "Текущее значение '$chosen_param': $current_value. Введите новое значение:" 
                read -p "" new_value
                sudo echo "$new_value" > "sched_$chosen_param"
                echo "Значение параметра '$chosen_param' изменено на '$new_value'."
            else
                echo "Неверное имя параметра. Попробуйте еще раз."
            fi

            echo "Введите имя параметра для изменения (или 'q' для выхода):"
            read -p "" chosen_param
        done

        echo "Вы хотите восстановить значения по умолчанию? (y/n):"
        read -p "" restore_default_choice

        if [[ "$restore_default_choice" == "y" ]]; then
            for param in $param_list; do
                default_value=$(cat "sched_$param")
                restore_default "$param" "$default_value"
                echo "Значение параметра '$param' восстановлено по умолчанию: $default_value"
            done
        fi

        cd $BASEDIR
        cd linpack

        echo "Дефолт запуск"
        for ((i=1; i<=2; i++)); do
            ./linpack
        done
        echo "nice -n -19 запуск"
        for ((i=1; i<=2; i++)); do
            nice -n -19 ./linpack
        done
        echo "nice -n 20 запуск"
        for ((i=1; i<=2; i++)); do
            nice -n 20 ./linpack
        done
        echo "taskset -c 0 запуск и taskset -c 1,2 "
        for ((i=1; i<=2; i++)); do
            taskset -c 0 ./linpack
            taskset -c 1,2 ./linpack
        done
        echo "Дефолт запуск"
        for ((i=1; i<=2; i++)); do
            ./linpack
        done

        read -p "Тестирование завершено. Проверьте результаты в './logs'. Нажмите Enter, чтобы удалить все тестовые файлы и очистить консоль."
        rm -rf $BASEDIR/linpack
        cd $workdir
}

MemBomb() {
	cd $work_dir/MemBomb/
	
    if [[ "$OSTYPE" == "linux-gnu"* || "$OSTYPE" == "msys"* || "$OSTYPE" == "cygwin" ]]; then
        if [[ "$OSTYPE" == "linux-gnu"* ]]; then
            gcc -o main MemBombLin.c
        else
            gcc -o main MemBombWin.c
        fi
        
        read -p "У вас есть возможность запустить программу. 
        Будьте осторожны, это может крашнуть вашу ОС. 
        В качестве альтернативы вы можете отказаться от запуска программы. 
        Запуск - 'y', отказ от запуска - 'n': " response
        if [ "$response" == "y" ]; then
            echo "Запуск..."
            sudo ./main
        elif [ "$response" == "n" ]; then
            echo "Отменяем запуск..."
        else
            echo "Некорректный ввод"
        fi
        
        read -p "Вы хотите удалить скомпилируемые файлы?(y/n): 	" response
        if [ "$response" == "y" ]; then
            find . -iname "main" -delete
        elif [ "$response" == "n" ]; then
            echo "Отменяем запуск..."
        else
            echo "Некорректный ввод"
        fi
     else
        echo "Не поддерживается ОС: $OSTYPE"
    fi

	cd $work_dir
	clear;main
}

ForkBomb() {

    cd "$BASEDIR/ForkBomb"

    if [[ "$OSTYPE" == "linux-gnu"* || "$OSTYPE" == "msys"* || "$OSTYPE" == "cygwin" ]]; then
        if [[ "$OSTYPE" == "linux-gnu"* ]]; then
            gcc -o main ForkBombLin.c
        else
            gcc -o main ForkBombWin.c
        fi

        read -p "У вас есть возможность запустить программу. 
        Будьте осторожны, это может крашнуть вашу ОС. 
        В качестве альтернативы вы можете отказаться от запуска программы. 
        Запуск - 'y', отказ от запуска - 'n': " response

        if [ "$response" == "y" ]; then
            echo "Запуск..."
            sleep 2
            ./main 
        elif [ "$response" == "n" ]; then
            echo "Отменяем запуск..."
        else
            echo "Некорректный ввод"
        fi
        read -p "Вы хотите удалить скомпилируемые файлы?(y/n): " response
        if [ "$response" == "y" ]; then
            find . -iname "main*" -delete
        elif [ "$response" == "n" ]; then
            echo "Отменяем удаление"
        else
            echo "Некорректный ввод"
        fi

        cd "$BASEDIR"
        main
    else
        echo "Не поддерживается ОС: $OSTYPE"
    fi

    cd $work_dir
	clear;main
}
VirtualCheck()
{
    cd $work_dir/VM/
    echo "Сейчас пройдет проверка на виртаульную систему..."
    ARCH=$(uname -m)
    if [ "$ARCH" = "aarch64" ]; then
        as -o aarch64.o aarch64.S
        ld -o aarch64 aarch64.o
        ./aarch64
    else
        nasm -f elf64 NASM.S -o NASM.o
        ld -o NASM NASM.o
        ./NASM
    fi
    cd $work_dir
	main
}

TCP()
{
    cd $work_dir/TCP

    read -p "Запустить TCP сервер? (y/n): " choice

    if [ "$choice" == "y" ]; then 
        echo "Запуск сервера..."
        cd $work_dir/TCP/server
        go run server.go 2>&1 | tee server.log &
        server_pid=$!
    else 
        echo "Отмена..."
        cd $work_dir
        main
    fi
    sleep 1
    read -p "Запустить TCP клиент? (y/n): " choice
    if [ "$choice" == "y" ]; then 
        echo "Запуск клиента..."
        cd $work_dir/TCP/client
        trap 'kill $server_pid; exit' INT TERM EXIT  # Добавляем trap для завершения сервера при завершении скрипта
        go run client.go 2>&1 | tee client.log
        echo "Running completed." >> client.log
    else 
        echo "Отмена..."
        cd $work_dir
        main
    fi
    cd $work_dir
    main
}

main() {
	BASEDIR=$(dirname "$(realpath "$0")")    

    echo "Выбери подпрограмму:"
    echo "1 - ForkBomb"
    echo "2 - MemBomb"
    echo "3 - LinPack"
    echo "4 - Scheduler"
    echo "5 - FSchecker"
    echo "6 - AllocTests"
    echo "7 - VirtualCheck"
    echo "8 - TCP Client/Server"
    echo "Для выхода нажми - 9"

	read -p "Введи 1-9: " Lab
	case $Lab in
		1) clear; ForkBomb ;;
        2) clear; MemBomb ;;
        3) clear; LinPack ;;
        4) BLOCK; Scheduler;;
        5) clear; FSchecker;;
        6) clear; AllocTests;;
        7) clear; VirtualCheck;;
        8) clear; TCP;;
		9) clear; exit;;
	esac
	main
}

Require;
clear; cd $work_dir; main;